push(@INC,"perlasm","../../perlasm");
require "x86asm.pl";
&asm_init($ARGV[0],"rc4-586.pl");
$x="eax";
$y="ebx";
$tx="ecx";
$ty="edx";
$in="esi";
$out="edi";
$d="ebp";
&RC4("RC4");
&asm_finish();
sub RC4_loop
{
local($n,$p,$char)=@_;
&comment("Round $n");
if ($char)
{
if ($p >= 0)
{
&mov($ty, &swtmp(2));
&cmp($ty, $in);
&jbe(&label("finished"));
&inc($in);
}
else
{
&add($ty, 8);
&inc($in);
&cmp($ty, $in);
&jb(&label("finished"));
&mov(&swtmp(2), $ty);
}
}
&add( &LB($y), &LB($tx));
&mov( $ty, &DWP(0,$d,$y,4));
&mov( &DWP(0,$d,$x,4),$ty);
&add( $ty, $tx);
&mov( &DWP(0,$d,$y,4),$tx);
&and( $ty, 0xff);
&inc( &LB($x)); &mov( $tx, &DWP(0,$d,$x,4)) if $p < 1; &mov( $ty, &DWP(0,$d,$ty,4));
if (!$char)
{
if ($p >= 1)
{
&add( $out, 8)
}
&movb( &BP($n,"esp","",0), &LB($ty));
}
else
{
&movb( &HB($ty), &BP(-1,$in,"",0));
&xorb(&LB($ty), &HB($ty));
&movb(&BP($n,$out,"",0),&LB($ty));
}
}
sub RC4
{
local($name)=@_;
&function_begin_B($name,"");
&mov($ty,&wparam(1)); &cmp($ty,0);
&jne(&label("proceed"));
&ret();
&set_label("proceed");
&comment("");
&push("ebp");
&push("ebx");
&push("esi");
&xor( $x, $x); &push("edi");
&xor( $y, $y); &mov( $d, &wparam(0)); &mov( $in, &wparam(2));
&movb( &LB($x), &BP(0,$d,"",1));
&movb( &LB($y), &BP(4,$d,"",1));
&mov( $out, &wparam(3));
&inc( &LB($x));
&stack_push(3); &add( $d, 8);
&cmp(&DWP(256,$d),-1);
&je(&label("RC4_CHAR"));
&lea( $ty, &DWP(-8,$ty,$in));
&mov( &swtmp(2), $ty); &mov( $tx, &DWP(0,$d,$x,4));
&cmp( $ty, $in);
&jb( &label("end"));
&set_label("start");
&add( $in, 8);
&RC4_loop(0,-1,0);
&RC4_loop(1,0,0);
&RC4_loop(2,0,0);
&RC4_loop(3,0,0);
&RC4_loop(4,0,0);
&RC4_loop(5,0,0);
&RC4_loop(6,0,0);
&RC4_loop(7,1,0);
&comment("apply the cipher text");
&mov( $tx, &swtmp(0));
&mov( $ty, &DWP(-8,$in,"",0));
&xor( $tx, $ty);
&mov( $ty, &DWP(-4,$in,"",0));
&mov( &DWP(-8,$out,"",0), $tx);
&mov( $tx, &swtmp(1));
&xor( $tx, $ty);
&mov( $ty, &swtmp(2)); &mov( &DWP(-4,$out,"",0), $tx);
&mov( $tx, &DWP(0,$d,$x,4));
&cmp($in, $ty);
&jbe(&label("start"));
&set_label("end");
&RC4_loop(0,-1,1);
&RC4_loop(1,0,1);
&RC4_loop(2,0,1);
&RC4_loop(3,0,1);
&RC4_loop(4,0,1);
&RC4_loop(5,0,1);
&RC4_loop(6,1,1);
&jmp(&label("finished"));
&align(16);
&set_label("RC4_CHAR");
&lea ($ty,&DWP(0,$in,$ty));
&mov (&swtmp(2),$ty);
&set_label("RC4_CHAR_loop");
&movz ($tx,&BP(0,$d,$x));
&add (&LB($y),&LB($tx));
&movz ($ty,&BP(0,$d,$y));
&movb (&BP(0,$d,$y),&LB($tx));
&movb (&BP(0,$d,$x),&LB($ty));
&add (&LB($ty),&LB($tx));
&movz ($ty,&BP(0,$d,$ty));
&xorb (&LB($ty),&BP(0,$in));
&movb (&BP(0,$out),&LB($ty));
&inc (&LB($x));
&inc ($in);
&inc ($out);
&cmp ($in,&swtmp(2));
&jb (&label("RC4_CHAR_loop"));
&set_label("finished");
&dec( $x);
&stack_pop(3);
&movb( &BP(-4,$d,"",0),&LB($y));
&movb( &BP(-8,$d,"",0),&LB($x));
&function_end($name);
}