The way you phrased your premise makes it sound like the rich would somehow be treated differently in legal situations. The only antibenefit the rich would receive in a socialist state would be having to pay more in taxes.
The common definition of socialism (abolition of private property combined with worker control of the means of production) is by its very nature more egalitarian than the alternatives.