3 # ====================================================================
4 # Written by Andy Polyakov <appro@fy.chalmers.se> for the OpenSSL
5 # project. Rights for redistribution and usage in source and binary
6 # forms are granted according to the OpenSSL license.
7 # ====================================================================
11 # This is a "teaser" code, as it can be improved in several ways...
12 # First of all non-SSE2 path should be implemented (yes, for now it
13 # performs Montgomery multiplication/convolution only on SSE2-capable
14 # CPUs such as P4, others fall down to original code). Then inner loop
15 # can be unrolled and modulo-scheduled to improve ILP and possibly
16 # moved to 128-bit XMM register bank (though it would require input
17 # rearrangement and/or increase bus bandwidth utilization). Dedicated
18 # squaring procedure should give further performance improvement...
19 # Yet, for being draft, the code improves rsa512 *sign* benchmark by
20 # 110%(!), rsa1024 one - by 70% and rsa4096 - by 20%:-)
22 push(@INC,"perlasm","../../perlasm");
25 &asm_init($ARGV[0],$0);
28 for (@ARGV) { $sse2=1 if (/-DOPENSSL_IA32_SSE2/); }
30 &external_label("OPENSSL_ia32cap_P") if ($sse2);
32 &function_begin("bn_mul_mont",$sse2?"EXTRN\t_OPENSSL_ia32cap_P:DWORD":"");
37 $rp="edi"; $bp="edi"; # overlapping variables!!!
42 $bias=2; # amount of extra words in tp
43 # (rounded up to even value)
44 $_rp=&DWP(4*($bias+0),"esp",$num,4); # stack frame layout below tp
45 $_ap=&DWP(4*($bias+1),"esp",$num,4);
46 $_bp=&DWP(4*($bias+2),"esp",$num,4);
47 $_np=&DWP(4*($bias+3),"esp",$num,4);
48 $_n0=&DWP(4*($bias+4),"esp",$num,4);
49 $_sp=&DWP(4*($bias+5),"esp",$num,4);
51 $acc0="mm0"; # mmx register bank layout
61 &picmeup("eax","OPENSSL_ia32cap_P");
62 &bt (&DWP(0,"eax"),26);
63 &mov ("eax",0); # zero signals "we did nothing"
64 &jnc (&label("non_sse2"));
66 ################################# load argument block...
67 &mov ("eax",&wparam(0)); # BN_ULONG *rp
68 &mov ("ebx",&wparam(1)); # const BN_ULONG *ap
69 &mov ("ecx",&wparam(2)); # const BN_ULONG *bp
70 &mov ("edx",&wparam(3)); # const BN_ULONG *np
71 &mov ("esi",&wparam(4)); # BN_ULONG n0
72 &mov ($num,&wparam(5)); # int num
74 &mov ("edi","esp"); # saved stack pointer!
77 &lea ("esp",&DWP(0,"esp",$num,4)); # alloca(4*(num+$bias+6))
79 &and ("esp",-1024); # minimize TLB utilization
80 &sub ($num,$bias+6); # num is restored to its original value
81 # and will remain constant from now...
83 &mov ($_rp,"eax"); # ... save a copy of argument block
88 &mov ($_sp,"edi"); # saved stack pointer!
91 &movd ($mask,"eax"); # mask 32 lower bits
93 &mov ($ap,$_ap); # load input pointers
100 &movd ($mul0,&DWP(0,$bp)); # bp[0]
101 &movd ($mul1,&DWP(0,$ap)); # ap[0]
102 &movd ($car1,&DWP(0,$np)); # np[0]
104 &pmuludq($mul1,$mul0); # ap[0]*bp[0]
106 &movq ($acc0,$mul1); # I wish movd worked for
107 &pand ($acc0,$mask); # inter-register transfers
109 &pmuludq($mul1,$_n0); # *=n0
111 &pmuludq($car1,$mul1); # "t[0]"*np[0]*n0
112 &paddq ($car1,$acc0);
119 &movd ($acc0,&DWP(0,$ap,$j,4)); # ap[j]
120 &movd ($acc1,&DWP(0,$np,$j,4)); # np[j]
121 &pmuludq($acc0,$mul0); # ap[j]*bp[0]
122 &pmuludq($acc1,$mul1); # np[j]*m1
124 &paddq ($car0,$acc0); # +=c0
128 &paddq ($car1,$acc1); # +=c1
129 &paddq ($car1,$acc0); # +=ap[j]*bp[0];
130 &movd (&DWP(-4,"esp",$j,4),$car1); # tp[j-1]=
135 &lea ($j,&DWP(1,$j));
139 &paddq ($car1,$car0);
140 &movq (&DWP(-4,"esp",$num,4),$car1);
146 &movd ($mul0,&DWP(0,$bp,$i,4)); # bp[i]
147 &movd ($mul1,&DWP(0,$ap)); # ap[0]
148 &movd ($temp,&DWP(0,"esp")); # tp[0]
149 &movd ($car1,&DWP(0,$np,$j,4)); # np[0]
150 &pmuludq($mul1,$mul0); # ap[0]*bp[i]
152 &paddq ($mul1,$temp); # +=tp[0]
157 &pmuludq($mul1,$_n0); # *=n0
159 &pmuludq($car1,$mul1);
160 &paddq ($car1,$acc0);
167 &movd ($acc0,&DWP(0,$ap,$j,4)); # ap[j]
168 &movd ($acc1,&DWP(0,$np,$j,4)); # np[j]
169 &movd ($temp,&DWP(0,"esp",$j,4)); # tp[j]
170 &pmuludq($acc0,$mul0); # ap[j]*bp[i]
171 &pmuludq($acc1,$mul1); # np[j]*m1
172 &paddq ($car0,$temp); # +=tp[j]
173 &paddq ($car0,$acc0); # +=c0
177 &paddq ($car1,$acc1); # +=c1
178 &paddq ($car1,$acc0); # +=ap[j]*bp[i]+tp[j]
179 &movd (&DWP(-4,"esp",$j,4),$car1); # tp[j-1]
184 &lea ($j,&DWP(1,$j)); # j++
186 &jl (&label("inner"));
188 &movd ($temp,&DWP(0,"esp",$num,4));
189 &paddq ($car1,$car0);
190 &paddq ($car1,$temp);
191 &movq (&DWP(-4,"esp",$num,4),$car1);
193 &lea ($i,&DWP(1,$i)); # i++
195 &jl (&label("outer"));
197 &emms (); # done with mmx bank
199 &mov ("esi",&DWP(0,"esp",$num,4)); # load upmost overflow bit
200 &mov ($rp,$_rp); # load result pointer
201 # [$ap and $bp are zapped]
203 &lea ($j,&DWP(-1,$num)); # j=num-1
204 &cmp ("esi",0); # clears CF unconditionally
205 &jnz (&label("sub"));
206 &mov ("eax",&DWP(0,"esp",$j,4));
207 &cmp ("eax",&DWP(0,$np,$j,4)); # tp[num-1]-np[num-1]?
208 &jae (&label("sub")); # if taken CF is cleared
210 &mov ("eax",&DWP(0,"esp",$j,4));
211 &mov (&DWP(0,$rp,$j,4),"eax"); # rp[i]=tp[i]
212 &mov (&DWP(0,"esp",$j,4),$j); # zap temporary vector
214 &jge (&label("copy"));
215 &jmp (&label("exit_sse2"));
218 &mov ("eax",&DWP(0,"esp",$i,4));
219 &sbb ("eax",&DWP(0,$np,$i,4));
220 &mov (&DWP(0,$rp,$i,4),"eax"); # rp[i]=tp[i]-np[i]
221 &lea ($i,&DWP(1,$i)); # i++
222 &dec ($j); # doesn't affect CF!
223 &jge (&label("sub"));
224 &lea ($j,&DWP(-1,$num)); # j=num-1
225 &sbb ("esi",0); # esi holds upmost overflow bit
226 &jc (&label("copy"));
228 &mov (&DWP(0,"esp",$j,4),$i); # zap temporary vector
230 &jge (&label("zap"));
232 &set_label("exit_sse2");
233 &mov ("esp",$_sp); # pull saved stack pointer
235 &set_label("non_sse2");
238 &function_end("bn_mul_mont");