244 lines
6.6 KiB
Raku
Executable file
244 lines
6.6 KiB
Raku
Executable file
#!/usr/bin/env perl
|
|
|
|
# ====================================================================
|
|
# Written by Andy Polyakov <appro@fy.chalmers.se> for the OpenSSL
|
|
# project. Rights for redistribution and usage in source and binary
|
|
# forms are granted according to the OpenSSL license.
|
|
# ====================================================================
|
|
|
|
# October 2005
|
|
#
|
|
# This is a "teaser" code, as it can be improved in several ways...
|
|
# First of all non-SSE2 path should be implemented (yes, for now it
|
|
# performs Montgomery multiplication/convolution only on SSE2-capable
|
|
# CPUs such as P4, others fall down to original code). Then inner loop
|
|
# can be unrolled and modulo-scheduled to improve ILP and possibly
|
|
# moved to 128-bit XMM register bank (though it would require input
|
|
# rearrangement and/or increase bus bandwidth utilization). Dedicated
|
|
# squaring procedure should give further performance improvement...
|
|
# Yet, for being draft, the code improves rsa512 *sign* benchmark by
|
|
# 110%(!), rsa1024 one - by 70% and rsa4096 - by 20%:-)
|
|
|
|
push(@INC,"perlasm","../../perlasm");
|
|
require "x86asm.pl";
|
|
|
|
&asm_init($ARGV[0],$0);
|
|
|
|
$sse2=0;
|
|
for (@ARGV) { $sse2=1 if (/-DOPENSSL_IA32_SSE2/); }
|
|
|
|
&external_label("OPENSSL_ia32cap_P") if ($sse2);
|
|
|
|
&function_begin("bn_mul_mont",$sse2?"EXTRN\t_OPENSSL_ia32cap_P:DWORD":"");
|
|
|
|
$i="ebx";
|
|
$j="ecx";
|
|
$ap="esi";
|
|
$rp="edi"; $bp="edi"; # overlapping variables!!!
|
|
$np="edx";
|
|
$num="ebp";
|
|
|
|
$_rp=&DWP(4*0,"esp"); # stack top layout
|
|
$_ap=&DWP(4*1,"esp");
|
|
$_bp=&DWP(4*2,"esp");
|
|
$_np=&DWP(4*3,"esp");
|
|
$_n0=&DWP(4*4,"esp");
|
|
$_num=&DWP(4*5,"esp");
|
|
$_sp=&DWP(4*6,"esp");
|
|
$frame=32; # size of above frame rounded up to 16n
|
|
|
|
$acc0="mm0"; # mmx register bank layout
|
|
$acc1="mm1";
|
|
$car0="mm2";
|
|
$car1="mm3";
|
|
$mul0="mm4";
|
|
$mul1="mm5";
|
|
$temp="mm6";
|
|
$mask="mm7";
|
|
|
|
if($sse2) {
|
|
&picmeup("eax","OPENSSL_ia32cap_P");
|
|
&bt (&DWP(0,"eax"),26);
|
|
&jnc (&label("non_sse2"));
|
|
|
|
################################# load argument block...
|
|
&mov ("eax",&wparam(0)); # BN_ULONG *rp
|
|
&mov ("ebx",&wparam(1)); # const BN_ULONG *ap
|
|
&mov ("ecx",&wparam(2)); # const BN_ULONG *bp
|
|
&mov ("edx",&wparam(3)); # const BN_ULONG *np
|
|
&mov ("esi",&wparam(4)); # const BN_ULONG *n0
|
|
&mov ($num,&wparam(5)); # int num
|
|
|
|
&mov ("edi","esp"); # saved stack pointer!
|
|
&add ($num,1); # extra word on top of tp
|
|
&neg ($num);
|
|
&lea ("esp",&DWP(-$frame,"esp",$num,4)); # alloca($frame+8*($num+1))
|
|
&neg ($num);
|
|
&and ("esp",-1024); # minimize TLB utilization
|
|
&sub ($num,1); # num is restored to its original value
|
|
# and will remain constant from now...
|
|
|
|
&mov ("esi",&DWP(0,"esi")); # pull n0[0]
|
|
&mov ($_rp,"eax"); # ... save a copy of argument block
|
|
&mov ($_ap,"ebx");
|
|
&mov ($_bp,"ecx");
|
|
&mov ($_np,"edx");
|
|
&mov ($_n0,"esi");
|
|
#&mov ($_num,$num); # redundant in sse2 context
|
|
&mov ($_sp,"edi"); # saved stack pointer!
|
|
|
|
&mov ("eax",-1);
|
|
&movd ($mask,"eax"); # mask 32 lower bits
|
|
|
|
&mov ($ap,$_ap); # load input pointers
|
|
&mov ($bp,$_bp);
|
|
&mov ($np,$_np);
|
|
|
|
&xor ($i,$i); # i=0
|
|
&xor ($j,$j); # j=0
|
|
|
|
&movd ($mul0,&DWP(0,$bp)); # bp[0]
|
|
&movd ($mul1,&DWP(0,$ap)); # ap[0]
|
|
&movd ($car1,&DWP(0,$np)); # np[0]
|
|
|
|
&pmuludq($mul1,$mul0); # ap[0]*bp[0]
|
|
&movq ($car0,$mul1);
|
|
&movq ($acc0,$mul1); # I wish movd worked for
|
|
&pand ($acc0,$mask); # inter-register transfers
|
|
|
|
&pmuludq($mul1,$_n0); # *=n0
|
|
|
|
&pmuludq($car1,$mul1); # "t[0]"*np[0]*n0
|
|
&paddq ($car1,$acc0);
|
|
|
|
&psrlq ($car0,32);
|
|
&psrlq ($car1,32);
|
|
|
|
&inc ($j); # j++
|
|
&set_label("1st");
|
|
&movd ($acc0,&DWP(0,$ap,$j,4)); # ap[j]
|
|
&movd ($acc1,&DWP(0,$np,$j,4)); # np[j]
|
|
&pmuludq($acc0,$mul0); # ap[j]*bp[0]
|
|
&pmuludq($acc1,$mul1); # np[j]*m1
|
|
|
|
&paddq ($car0,$acc0); # +=c0
|
|
&movq ($acc0,$car0);
|
|
&pand ($acc0,$mask);
|
|
|
|
&paddq ($car1,$acc1); # +=c1
|
|
&paddq ($car1,$acc0); # +=ap[j]*bp[0];
|
|
&movd (&DWP($frame-4,"esp",$j,4),$car1); # tp[j-1]=
|
|
|
|
&psrlq ($car0,32);
|
|
&psrlq ($car1,32);
|
|
|
|
&lea ($j,&DWP(1,$j));
|
|
&cmp ($j,$num);
|
|
&jl (&label("1st"));
|
|
|
|
&paddq ($car1,$car0);
|
|
&movq (&DWP($frame-4,"esp",$num,4),$car1);
|
|
|
|
&inc ($i); # i++
|
|
&set_label("outer");
|
|
&xor ($j,$j); # j=0
|
|
|
|
&movd ($mul0,&DWP(0,$bp,$i,4)); # bp[i]
|
|
&movd ($mul1,&DWP(0,$ap)); # ap[0]
|
|
&movd ($temp,&DWP($frame,"esp")); # tp[0]
|
|
&movd ($car1,&DWP(0,$np)); # np[0]
|
|
&pmuludq($mul1,$mul0); # ap[0]*bp[i]
|
|
|
|
&paddq ($mul1,$temp); # +=tp[0]
|
|
&movq ($acc0,$mul1);
|
|
&movq ($car0,$mul1);
|
|
&pand ($acc0,$mask);
|
|
|
|
&pmuludq($mul1,$_n0); # *=n0
|
|
|
|
&pmuludq($car1,$mul1);
|
|
&paddq ($car1,$acc0);
|
|
|
|
&psrlq ($car0,32);
|
|
&psrlq ($car1,32);
|
|
|
|
&inc ($j); # j++
|
|
&set_label("inner");
|
|
&movd ($acc0,&DWP(0,$ap,$j,4)); # ap[j]
|
|
&movd ($acc1,&DWP(0,$np,$j,4)); # np[j]
|
|
&movd ($temp,&DWP($frame,"esp",$j,4));# tp[j]
|
|
&pmuludq($acc0,$mul0); # ap[j]*bp[i]
|
|
&pmuludq($acc1,$mul1); # np[j]*m1
|
|
&paddq ($car0,$temp); # +=tp[j]
|
|
&paddq ($car0,$acc0); # +=c0
|
|
&movq ($acc0,$car0);
|
|
&pand ($acc0,$mask);
|
|
|
|
&paddq ($car1,$acc1); # +=c1
|
|
&paddq ($car1,$acc0); # +=ap[j]*bp[i]+tp[j]
|
|
&movd (&DWP($frame-4,"esp",$j,4),$car1); # tp[j-1]=
|
|
|
|
&psrlq ($car0,32);
|
|
&psrlq ($car1,32);
|
|
|
|
&lea ($j,&DWP(1,$j)); # j++
|
|
&cmp ($j,$num);
|
|
&jl (&label("inner"));
|
|
|
|
&movd ($temp,&DWP($frame,"esp",$num,4));
|
|
&paddq ($car1,$car0);
|
|
&paddq ($car1,$temp);
|
|
&movq (&DWP($frame-4,"esp",$num,4),$car1);
|
|
|
|
&lea ($i,&DWP(1,$i)); # i++
|
|
&cmp ($i,$num);
|
|
&jl (&label("outer"));
|
|
|
|
&emms (); # done with mmx bank
|
|
|
|
&mov ("esi",&DWP($frame,"esp",$num,4));# load upmost overflow bit
|
|
&mov ($rp,$_rp); # load result pointer
|
|
# [$ap and $bp are zapped]
|
|
&xor ($i,$i); # i=0
|
|
&lea ($j,&DWP(-1,$num)); # j=num-1
|
|
&cmp ("esi",0); # clears CF unconditionally
|
|
&jnz (&label("sub"));
|
|
&mov ("eax",&DWP($frame,"esp",$j,4));
|
|
&cmp ("eax",&DWP(0,$np,$j,4)); # tp[num-1]-np[num-1]?
|
|
&jae (&label("sub")); # if taken CF is cleared
|
|
&set_label("copy");
|
|
&mov ("eax",&DWP($frame,"esp",$j,4));
|
|
&mov (&DWP(0,$rp,$j,4),"eax"); # rp[i]=tp[i]
|
|
&mov (&DWP($frame,"esp",$j,4),$j); # zap temporary vector
|
|
&dec ($j);
|
|
&jge (&label("copy"));
|
|
&jmp (&label("exit_sse2"));
|
|
|
|
&set_label("sub",4);
|
|
&mov ("eax",&DWP($frame,"esp",$i,4));
|
|
&sbb ("eax",&DWP(0,$np,$i,4));
|
|
&mov (&DWP(0,$rp,$i,4),"eax"); # rp[i]=tp[i]-np[i]
|
|
&lea ($i,&DWP(1,$i)); # i++
|
|
&dec ($j); # doesn't affect CF!
|
|
&jge (&label("sub"));
|
|
&lea ($j,&DWP(-1,$num)); # j=num-1
|
|
&sbb ("esi",0); # esi holds upmost overflow bit
|
|
&jc (&label("copy"));
|
|
&set_label("zap");
|
|
&mov (&DWP($frame,"esp",$j,4),$i); # zap temporary vector
|
|
&dec ($j);
|
|
&jge (&label("zap"));
|
|
|
|
&set_label("exit_sse2");
|
|
&mov ("esp",$_sp); # pull saved stack pointer
|
|
&mov ("eax",1);
|
|
&jmp (&label("leave"));
|
|
&set_label("non_sse2");
|
|
}
|
|
|
|
&xor ("eax","eax"); # zero signals "not implemented [yet]"
|
|
|
|
&set_label("leave");
|
|
&function_end("bn_mul_mont");
|
|
|
|
&asm_finish();
|