You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

1794 lines
50 KiB

  1. #!/usr/bin/env perl
  2. #
  3. # ====================================================================
  4. # Written by Andy Polyakov <appro@openssl.org> for the OpenSSL
  5. # project. The module is, however, dual licensed under OpenSSL and
  6. # CRYPTOGAMS licenses depending on where you obtain it. For further
  7. # details see http://www.openssl.org/~appro/cryptogams/.
  8. # ====================================================================
  9. #
  10. # This module implements Poly1305 hash for x86.
  11. #
  12. # April 2015
  13. #
  14. # Numbers are cycles per processed byte with poly1305_blocks alone,
  15. # measured with rdtsc at fixed clock frequency.
  16. #
  17. # IALU/gcc-3.4(*) SSE2(**) AVX2
  18. # Pentium 15.7/+80% -
  19. # PIII 6.21/+90% -
  20. # P4 19.8/+40% 3.24
  21. # Core 2 4.85/+90% 1.80
  22. # Westmere 4.58/+100% 1.43
  23. # Sandy Bridge 3.90/+100% 1.36
  24. # Haswell 3.88/+70% 1.18 0.72
  25. # Silvermont 11.0/+40% 4.80
  26. # VIA Nano 6.71/+90% 2.47
  27. # Sledgehammer 3.51/+180% 4.27
  28. # Bulldozer 4.53/+140% 1.31
  29. #
  30. # (*) gcc 4.8 for some reason generated worse code;
  31. # (**) besides SSE2 there are floating-point and AVX options; FP
  32. # is deemed unnecessary, because pre-SSE2 processor are too
  33. # old to care about, while it's not the fastest option on
  34. # SSE2-capable ones; AVX is omitted, because it doesn't give
  35. # a lot of improvement, 5-10% depending on processor;
  36. $0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1;
  37. push(@INC,"${dir}","${dir}../../perlasm");
  38. require "x86asm.pl";
  39. $output=pop;
  40. open STDOUT,">$output";
  41. &asm_init($ARGV[0],"poly1305-x86.pl",$ARGV[$#ARGV] eq "386");
  42. $sse2=$avx=0;
  43. for (@ARGV) { $sse2=1 if (/-DOPENSSL_IA32_SSE2/); }
  44. if ($sse2) {
  45. &static_label("const_sse2");
  46. &static_label("enter_blocks");
  47. &static_label("enter_emit");
  48. &external_label("OPENSSL_ia32cap_P");
  49. # This may be set to 2, but valgrind can't do AVX2 on 32-bit. Without a
  50. # way to verify test coverage, keep it disabled.
  51. $avx = 0;
  52. }
  53. ########################################################################
  54. # Layout of opaque area is following.
  55. #
  56. # unsigned __int32 h[5]; # current hash value base 2^32
  57. # unsigned __int32 pad; # is_base2_26 in vector context
  58. # unsigned __int32 r[4]; # key value base 2^32
  59. &align(64);
  60. &function_begin("poly1305_init");
  61. &mov ("edi",&wparam(0)); # context
  62. &mov ("esi",&wparam(1)); # key
  63. &mov ("ebp",&wparam(2)); # function table
  64. &xor ("eax","eax");
  65. &mov (&DWP(4*0,"edi"),"eax"); # zero hash value
  66. &mov (&DWP(4*1,"edi"),"eax");
  67. &mov (&DWP(4*2,"edi"),"eax");
  68. &mov (&DWP(4*3,"edi"),"eax");
  69. &mov (&DWP(4*4,"edi"),"eax");
  70. &mov (&DWP(4*5,"edi"),"eax"); # is_base2_26
  71. &cmp ("esi",0);
  72. &je (&label("nokey"));
  73. if ($sse2) {
  74. &call (&label("pic_point"));
  75. &set_label("pic_point");
  76. &blindpop("ebx");
  77. &lea ("eax",&DWP("poly1305_blocks-".&label("pic_point"),"ebx"));
  78. &lea ("edx",&DWP("poly1305_emit-".&label("pic_point"),"ebx"));
  79. &picmeup("edi","OPENSSL_ia32cap_P","ebx",&label("pic_point"));
  80. &mov ("ecx",&DWP(0,"edi"));
  81. &and ("ecx",1<<26|1<<24);
  82. &cmp ("ecx",1<<26|1<<24); # SSE2 and XMM?
  83. &jne (&label("no_sse2"));
  84. &lea ("eax",&DWP("_poly1305_blocks_sse2-".&label("pic_point"),"ebx"));
  85. &lea ("edx",&DWP("_poly1305_emit_sse2-".&label("pic_point"),"ebx"));
  86. if ($avx>1) {
  87. &mov ("ecx",&DWP(8,"edi"));
  88. &test ("ecx",1<<5); # AVX2?
  89. &jz (&label("no_sse2"));
  90. &lea ("eax",&DWP("_poly1305_blocks_avx2-".&label("pic_point"),"ebx"));
  91. }
  92. &set_label("no_sse2");
  93. &mov ("edi",&wparam(0)); # reload context
  94. &mov (&DWP(0,"ebp"),"eax"); # fill function table
  95. &mov (&DWP(4,"ebp"),"edx");
  96. }
  97. &mov ("eax",&DWP(4*0,"esi")); # load input key
  98. &mov ("ebx",&DWP(4*1,"esi"));
  99. &mov ("ecx",&DWP(4*2,"esi"));
  100. &mov ("edx",&DWP(4*3,"esi"));
  101. &and ("eax",0x0fffffff);
  102. &and ("ebx",0x0ffffffc);
  103. &and ("ecx",0x0ffffffc);
  104. &and ("edx",0x0ffffffc);
  105. &mov (&DWP(4*6,"edi"),"eax");
  106. &mov (&DWP(4*7,"edi"),"ebx");
  107. &mov (&DWP(4*8,"edi"),"ecx");
  108. &mov (&DWP(4*9,"edi"),"edx");
  109. &mov ("eax",$sse2);
  110. &set_label("nokey");
  111. &function_end("poly1305_init");
  112. ($h0,$h1,$h2,$h3,$h4,
  113. $d0,$d1,$d2,$d3,
  114. $r0,$r1,$r2,$r3,
  115. $s1,$s2,$s3)=map(4*$_,(0..15));
  116. &function_begin("poly1305_blocks");
  117. &mov ("edi",&wparam(0)); # ctx
  118. &mov ("esi",&wparam(1)); # inp
  119. &mov ("ecx",&wparam(2)); # len
  120. &set_label("enter_blocks");
  121. &and ("ecx",-15);
  122. &jz (&label("nodata"));
  123. &stack_push(16);
  124. &mov ("eax",&DWP(4*6,"edi")); # r0
  125. &mov ("ebx",&DWP(4*7,"edi")); # r1
  126. &lea ("ebp",&DWP(0,"esi","ecx")); # end of input
  127. &mov ("ecx",&DWP(4*8,"edi")); # r2
  128. &mov ("edx",&DWP(4*9,"edi")); # r3
  129. &mov (&wparam(2),"ebp");
  130. &mov ("ebp","esi");
  131. &mov (&DWP($r0,"esp"),"eax"); # r0
  132. &mov ("eax","ebx");
  133. &shr ("eax",2);
  134. &mov (&DWP($r1,"esp"),"ebx"); # r1
  135. &add ("eax","ebx"); # s1
  136. &mov ("ebx","ecx");
  137. &shr ("ebx",2);
  138. &mov (&DWP($r2,"esp"),"ecx"); # r2
  139. &add ("ebx","ecx"); # s2
  140. &mov ("ecx","edx");
  141. &shr ("ecx",2);
  142. &mov (&DWP($r3,"esp"),"edx"); # r3
  143. &add ("ecx","edx"); # s3
  144. &mov (&DWP($s1,"esp"),"eax"); # s1
  145. &mov (&DWP($s2,"esp"),"ebx"); # s2
  146. &mov (&DWP($s3,"esp"),"ecx"); # s3
  147. &mov ("eax",&DWP(4*0,"edi")); # load hash value
  148. &mov ("ebx",&DWP(4*1,"edi"));
  149. &mov ("ecx",&DWP(4*2,"edi"));
  150. &mov ("esi",&DWP(4*3,"edi"));
  151. &mov ("edi",&DWP(4*4,"edi"));
  152. &jmp (&label("loop"));
  153. &set_label("loop",32);
  154. &add ("eax",&DWP(4*0,"ebp")); # accumulate input
  155. &adc ("ebx",&DWP(4*1,"ebp"));
  156. &adc ("ecx",&DWP(4*2,"ebp"));
  157. &adc ("esi",&DWP(4*3,"ebp"));
  158. &lea ("ebp",&DWP(4*4,"ebp"));
  159. &adc ("edi",&wparam(3)); # padbit
  160. &mov (&DWP($h0,"esp"),"eax"); # put aside hash[+inp]
  161. &mov (&DWP($h3,"esp"),"esi");
  162. &mul (&DWP($r0,"esp")); # h0*r0
  163. &mov (&DWP($h4,"esp"),"edi");
  164. &mov ("edi","eax");
  165. &mov ("eax","ebx"); # h1
  166. &mov ("esi","edx");
  167. &mul (&DWP($s3,"esp")); # h1*s3
  168. &add ("edi","eax");
  169. &mov ("eax","ecx"); # h2
  170. &adc ("esi","edx");
  171. &mul (&DWP($s2,"esp")); # h2*s2
  172. &add ("edi","eax");
  173. &mov ("eax",&DWP($h3,"esp"));
  174. &adc ("esi","edx");
  175. &mul (&DWP($s1,"esp")); # h3*s1
  176. &add ("edi","eax");
  177. &mov ("eax",&DWP($h0,"esp"));
  178. &adc ("esi","edx");
  179. &mul (&DWP($r1,"esp")); # h0*r1
  180. &mov (&DWP($d0,"esp"),"edi");
  181. &xor ("edi","edi");
  182. &add ("esi","eax");
  183. &mov ("eax","ebx"); # h1
  184. &adc ("edi","edx");
  185. &mul (&DWP($r0,"esp")); # h1*r0
  186. &add ("esi","eax");
  187. &mov ("eax","ecx"); # h2
  188. &adc ("edi","edx");
  189. &mul (&DWP($s3,"esp")); # h2*s3
  190. &add ("esi","eax");
  191. &mov ("eax",&DWP($h3,"esp"));
  192. &adc ("edi","edx");
  193. &mul (&DWP($s2,"esp")); # h3*s2
  194. &add ("esi","eax");
  195. &mov ("eax",&DWP($h4,"esp"));
  196. &adc ("edi","edx");
  197. &imul ("eax",&DWP($s1,"esp")); # h4*s1
  198. &add ("esi","eax");
  199. &mov ("eax",&DWP($h0,"esp"));
  200. &adc ("edi",0);
  201. &mul (&DWP($r2,"esp")); # h0*r2
  202. &mov (&DWP($d1,"esp"),"esi");
  203. &xor ("esi","esi");
  204. &add ("edi","eax");
  205. &mov ("eax","ebx"); # h1
  206. &adc ("esi","edx");
  207. &mul (&DWP($r1,"esp")); # h1*r1
  208. &add ("edi","eax");
  209. &mov ("eax","ecx"); # h2
  210. &adc ("esi","edx");
  211. &mul (&DWP($r0,"esp")); # h2*r0
  212. &add ("edi","eax");
  213. &mov ("eax",&DWP($h3,"esp"));
  214. &adc ("esi","edx");
  215. &mul (&DWP($s3,"esp")); # h3*s3
  216. &add ("edi","eax");
  217. &mov ("eax",&DWP($h4,"esp"));
  218. &adc ("esi","edx");
  219. &imul ("eax",&DWP($s2,"esp")); # h4*s2
  220. &add ("edi","eax");
  221. &mov ("eax",&DWP($h0,"esp"));
  222. &adc ("esi",0);
  223. &mul (&DWP($r3,"esp")); # h0*r3
  224. &mov (&DWP($d2,"esp"),"edi");
  225. &xor ("edi","edi");
  226. &add ("esi","eax");
  227. &mov ("eax","ebx"); # h1
  228. &adc ("edi","edx");
  229. &mul (&DWP($r2,"esp")); # h1*r2
  230. &add ("esi","eax");
  231. &mov ("eax","ecx"); # h2
  232. &adc ("edi","edx");
  233. &mul (&DWP($r1,"esp")); # h2*r1
  234. &add ("esi","eax");
  235. &mov ("eax",&DWP($h3,"esp"));
  236. &adc ("edi","edx");
  237. &mul (&DWP($r0,"esp")); # h3*r0
  238. &add ("esi","eax");
  239. &mov ("ecx",&DWP($h4,"esp"));
  240. &adc ("edi","edx");
  241. &mov ("edx","ecx");
  242. &imul ("ecx",&DWP($s3,"esp")); # h4*s3
  243. &add ("esi","ecx");
  244. &mov ("eax",&DWP($d0,"esp"));
  245. &adc ("edi",0);
  246. &imul ("edx",&DWP($r0,"esp")); # h4*r0
  247. &add ("edx","edi");
  248. &mov ("ebx",&DWP($d1,"esp"));
  249. &mov ("ecx",&DWP($d2,"esp"));
  250. &mov ("edi","edx"); # last reduction step
  251. &shr ("edx",2);
  252. &and ("edi",3);
  253. &lea ("edx",&DWP(0,"edx","edx",4)); # *5
  254. &add ("eax","edx");
  255. &adc ("ebx",0);
  256. &adc ("ecx",0);
  257. &adc ("esi",0);
  258. &cmp ("ebp",&wparam(2)); # done yet?
  259. &jne (&label("loop"));
  260. &mov ("edx",&wparam(0)); # ctx
  261. &stack_pop(16);
  262. &mov (&DWP(4*0,"edx"),"eax"); # store hash value
  263. &mov (&DWP(4*1,"edx"),"ebx");
  264. &mov (&DWP(4*2,"edx"),"ecx");
  265. &mov (&DWP(4*3,"edx"),"esi");
  266. &mov (&DWP(4*4,"edx"),"edi");
  267. &set_label("nodata");
  268. &function_end("poly1305_blocks");
  269. &function_begin("poly1305_emit");
  270. &mov ("ebp",&wparam(0)); # context
  271. &set_label("enter_emit");
  272. &mov ("edi",&wparam(1)); # output
  273. &mov ("eax",&DWP(4*0,"ebp")); # load hash value
  274. &mov ("ebx",&DWP(4*1,"ebp"));
  275. &mov ("ecx",&DWP(4*2,"ebp"));
  276. &mov ("edx",&DWP(4*3,"ebp"));
  277. &mov ("esi",&DWP(4*4,"ebp"));
  278. &add ("eax",5); # compare to modulus
  279. &adc ("ebx",0);
  280. &adc ("ecx",0);
  281. &adc ("edx",0);
  282. &adc ("esi",0);
  283. &shr ("esi",2); # did it carry/borrow?
  284. &neg ("esi"); # do we choose hash-modulus?
  285. &and ("eax","esi");
  286. &and ("ebx","esi");
  287. &and ("ecx","esi");
  288. &and ("edx","esi");
  289. &mov (&DWP(4*0,"edi"),"eax");
  290. &mov (&DWP(4*1,"edi"),"ebx");
  291. &mov (&DWP(4*2,"edi"),"ecx");
  292. &mov (&DWP(4*3,"edi"),"edx");
  293. &not ("esi"); # or original hash value?
  294. &mov ("eax",&DWP(4*0,"ebp"));
  295. &mov ("ebx",&DWP(4*1,"ebp"));
  296. &mov ("ecx",&DWP(4*2,"ebp"));
  297. &mov ("edx",&DWP(4*3,"ebp"));
  298. &mov ("ebp",&wparam(2));
  299. &and ("eax","esi");
  300. &and ("ebx","esi");
  301. &and ("ecx","esi");
  302. &and ("edx","esi");
  303. &or ("eax",&DWP(4*0,"edi"));
  304. &or ("ebx",&DWP(4*1,"edi"));
  305. &or ("ecx",&DWP(4*2,"edi"));
  306. &or ("edx",&DWP(4*3,"edi"));
  307. &add ("eax",&DWP(4*0,"ebp")); # accumulate key
  308. &adc ("ebx",&DWP(4*1,"ebp"));
  309. &adc ("ecx",&DWP(4*2,"ebp"));
  310. &adc ("edx",&DWP(4*3,"ebp"));
  311. &mov (&DWP(4*0,"edi"),"eax");
  312. &mov (&DWP(4*1,"edi"),"ebx");
  313. &mov (&DWP(4*2,"edi"),"ecx");
  314. &mov (&DWP(4*3,"edi"),"edx");
  315. &function_end("poly1305_emit");
  316. if ($sse2) {
  317. ########################################################################
  318. # Layout of opaque area is following.
  319. #
  320. # unsigned __int32 h[5]; # current hash value base 2^26
  321. # unsigned __int32 is_base2_26;
  322. # unsigned __int32 r[4]; # key value base 2^32
  323. # unsigned __int32 pad[2];
  324. # struct { unsigned __int32 r^4, r^3, r^2, r^1; } r[9];
  325. #
  326. # where r^n are base 2^26 digits of degrees of multiplier key. There are
  327. # 5 digits, but last four are interleaved with multiples of 5, totalling
  328. # in 9 elements: r0, r1, 5*r1, r2, 5*r2, r3, 5*r3, r4, 5*r4.
  329. my ($D0,$D1,$D2,$D3,$D4,$T0,$T1,$T2)=map("xmm$_",(0..7));
  330. my $MASK=$T2; # borrow and keep in mind
  331. &align (32);
  332. &function_begin_B("_poly1305_init_sse2");
  333. &movdqu ($D4,&QWP(4*6,"edi")); # key base 2^32
  334. &lea ("edi",&DWP(16*3,"edi")); # size optimization
  335. &mov ("ebp","esp");
  336. &sub ("esp",16*(9+5));
  337. &and ("esp",-16);
  338. #&pand ($D4,&QWP(96,"ebx")); # magic mask
  339. &movq ($MASK,&QWP(64,"ebx"));
  340. &movdqa ($D0,$D4);
  341. &movdqa ($D1,$D4);
  342. &movdqa ($D2,$D4);
  343. &pand ($D0,$MASK); # -> base 2^26
  344. &psrlq ($D1,26);
  345. &psrldq ($D2,6);
  346. &pand ($D1,$MASK);
  347. &movdqa ($D3,$D2);
  348. &psrlq ($D2,4)
  349. &psrlq ($D3,30);
  350. &pand ($D2,$MASK);
  351. &pand ($D3,$MASK);
  352. &psrldq ($D4,13);
  353. &lea ("edx",&DWP(16*9,"esp")); # size optimization
  354. &mov ("ecx",2);
  355. &set_label("square");
  356. &movdqa (&QWP(16*0,"esp"),$D0);
  357. &movdqa (&QWP(16*1,"esp"),$D1);
  358. &movdqa (&QWP(16*2,"esp"),$D2);
  359. &movdqa (&QWP(16*3,"esp"),$D3);
  360. &movdqa (&QWP(16*4,"esp"),$D4);
  361. &movdqa ($T1,$D1);
  362. &movdqa ($T0,$D2);
  363. &pslld ($T1,2);
  364. &pslld ($T0,2);
  365. &paddd ($T1,$D1); # *5
  366. &paddd ($T0,$D2); # *5
  367. &movdqa (&QWP(16*5,"esp"),$T1);
  368. &movdqa (&QWP(16*6,"esp"),$T0);
  369. &movdqa ($T1,$D3);
  370. &movdqa ($T0,$D4);
  371. &pslld ($T1,2);
  372. &pslld ($T0,2);
  373. &paddd ($T1,$D3); # *5
  374. &paddd ($T0,$D4); # *5
  375. &movdqa (&QWP(16*7,"esp"),$T1);
  376. &movdqa (&QWP(16*8,"esp"),$T0);
  377. &pshufd ($T1,$D0,0b01000100);
  378. &movdqa ($T0,$D1);
  379. &pshufd ($D1,$D1,0b01000100);
  380. &pshufd ($D2,$D2,0b01000100);
  381. &pshufd ($D3,$D3,0b01000100);
  382. &pshufd ($D4,$D4,0b01000100);
  383. &movdqa (&QWP(16*0,"edx"),$T1);
  384. &movdqa (&QWP(16*1,"edx"),$D1);
  385. &movdqa (&QWP(16*2,"edx"),$D2);
  386. &movdqa (&QWP(16*3,"edx"),$D3);
  387. &movdqa (&QWP(16*4,"edx"),$D4);
  388. ################################################################
  389. # d4 = h4*r0 + h3*r1 + h2*r2 + h1*r3 + h0*r4
  390. # d3 = h3*r0 + h2*r1 + h1*r2 + h0*r3 + h4*5*r4
  391. # d2 = h2*r0 + h1*r1 + h0*r2 + h4*5*r3 + h3*5*r4
  392. # d1 = h1*r0 + h0*r1 + h4*5*r2 + h3*5*r3 + h2*5*r4
  393. # d0 = h0*r0 + h4*5*r1 + h3*5*r2 + h2*5*r3 + h1*5*r4
  394. &pmuludq ($D4,$D0); # h4*r0
  395. &pmuludq ($D3,$D0); # h3*r0
  396. &pmuludq ($D2,$D0); # h2*r0
  397. &pmuludq ($D1,$D0); # h1*r0
  398. &pmuludq ($D0,$T1); # h0*r0
  399. sub pmuladd {
  400. my $load = shift;
  401. my $base = shift; $base = "esp" if (!defined($base));
  402. ################################################################
  403. # As for choice to "rotate" $T0-$T2 in order to move paddq
  404. # past next multiplication. While it makes code harder to read
  405. # and doesn't have significant effect on most processors, it
  406. # makes a lot of difference on Atom, up to 30% improvement.
  407. &movdqa ($T1,$T0);
  408. &pmuludq ($T0,&QWP(16*3,$base)); # r1*h3
  409. &movdqa ($T2,$T1);
  410. &pmuludq ($T1,&QWP(16*2,$base)); # r1*h2
  411. &paddq ($D4,$T0);
  412. &movdqa ($T0,$T2);
  413. &pmuludq ($T2,&QWP(16*1,$base)); # r1*h1
  414. &paddq ($D3,$T1);
  415. &$load ($T1,5); # s1
  416. &pmuludq ($T0,&QWP(16*0,$base)); # r1*h0
  417. &paddq ($D2,$T2);
  418. &pmuludq ($T1,&QWP(16*4,$base)); # s1*h4
  419. &$load ($T2,2); # r2^n
  420. &paddq ($D1,$T0);
  421. &movdqa ($T0,$T2);
  422. &pmuludq ($T2,&QWP(16*2,$base)); # r2*h2
  423. &paddq ($D0,$T1);
  424. &movdqa ($T1,$T0);
  425. &pmuludq ($T0,&QWP(16*1,$base)); # r2*h1
  426. &paddq ($D4,$T2);
  427. &$load ($T2,6); # s2^n
  428. &pmuludq ($T1,&QWP(16*0,$base)); # r2*h0
  429. &paddq ($D3,$T0);
  430. &movdqa ($T0,$T2);
  431. &pmuludq ($T2,&QWP(16*4,$base)); # s2*h4
  432. &paddq ($D2,$T1);
  433. &pmuludq ($T0,&QWP(16*3,$base)); # s2*h3
  434. &$load ($T1,3); # r3^n
  435. &paddq ($D1,$T2);
  436. &movdqa ($T2,$T1);
  437. &pmuludq ($T1,&QWP(16*1,$base)); # r3*h1
  438. &paddq ($D0,$T0);
  439. &$load ($T0,7); # s3^n
  440. &pmuludq ($T2,&QWP(16*0,$base)); # r3*h0
  441. &paddq ($D4,$T1);
  442. &movdqa ($T1,$T0);
  443. &pmuludq ($T0,&QWP(16*4,$base)); # s3*h4
  444. &paddq ($D3,$T2);
  445. &movdqa ($T2,$T1);
  446. &pmuludq ($T1,&QWP(16*3,$base)); # s3*h3
  447. &paddq ($D2,$T0);
  448. &pmuludq ($T2,&QWP(16*2,$base)); # s3*h2
  449. &$load ($T0,4); # r4^n
  450. &paddq ($D1,$T1);
  451. &$load ($T1,8); # s4^n
  452. &pmuludq ($T0,&QWP(16*0,$base)); # r4*h0
  453. &paddq ($D0,$T2);
  454. &movdqa ($T2,$T1);
  455. &pmuludq ($T1,&QWP(16*4,$base)); # s4*h4
  456. &paddq ($D4,$T0);
  457. &movdqa ($T0,$T2);
  458. &pmuludq ($T2,&QWP(16*1,$base)); # s4*h1
  459. &paddq ($D3,$T1);
  460. &movdqa ($T1,$T0);
  461. &pmuludq ($T0,&QWP(16*2,$base)); # s4*h2
  462. &paddq ($D0,$T2);
  463. &pmuludq ($T1,&QWP(16*3,$base)); # s4*h3
  464. &movdqa ($MASK,&QWP(64,"ebx"));
  465. &paddq ($D1,$T0);
  466. &paddq ($D2,$T1);
  467. }
  468. &pmuladd (sub { my ($reg,$i)=@_;
  469. &movdqa ($reg,&QWP(16*$i,"esp"));
  470. },"edx");
  471. sub lazy_reduction {
  472. my $extra = shift;
  473. my $paddx = defined($extra) ? paddq : paddd;
  474. ################################################################
  475. # lazy reduction as discussed in "NEON crypto" by D.J. Bernstein
  476. # and P. Schwabe
  477. &movdqa ($T0,$D3);
  478. &pand ($D3,$MASK);
  479. &psrlq ($T0,26);
  480. &$extra () if (defined($extra));
  481. &paddq ($T0,$D4); # h3 -> h4
  482. &movdqa ($T1,$D0);
  483. &pand ($D0,$MASK);
  484. &psrlq ($T1,26);
  485. &movdqa ($D4,$T0);
  486. &paddq ($T1,$D1); # h0 -> h1
  487. &psrlq ($T0,26);
  488. &pand ($D4,$MASK);
  489. &movdqa ($D1,$T1);
  490. &psrlq ($T1,26);
  491. &paddd ($D0,$T0); # favour paddd when
  492. # possible, because
  493. # paddq is "broken"
  494. # on Atom
  495. &psllq ($T0,2);
  496. &paddq ($T1,$D2); # h1 -> h2
  497. &$paddx ($T0,$D0); # h4 -> h0
  498. &pand ($D1,$MASK);
  499. &movdqa ($D2,$T1);
  500. &psrlq ($T1,26);
  501. &pand ($D2,$MASK);
  502. &paddd ($T1,$D3); # h2 -> h3
  503. &movdqa ($D0,$T0);
  504. &psrlq ($T0,26);
  505. &movdqa ($D3,$T1);
  506. &psrlq ($T1,26);
  507. &pand ($D0,$MASK);
  508. &paddd ($D1,$T0); # h0 -> h1
  509. &pand ($D3,$MASK);
  510. &paddd ($D4,$T1); # h3 -> h4
  511. }
  512. &lazy_reduction ();
  513. &dec ("ecx");
  514. &jz (&label("square_break"));
  515. &punpcklqdq ($D0,&QWP(16*0,"esp")); # 0:r^1:0:r^2
  516. &punpcklqdq ($D1,&QWP(16*1,"esp"));
  517. &punpcklqdq ($D2,&QWP(16*2,"esp"));
  518. &punpcklqdq ($D3,&QWP(16*3,"esp"));
  519. &punpcklqdq ($D4,&QWP(16*4,"esp"));
  520. &jmp (&label("square"));
  521. &set_label("square_break");
  522. &psllq ($D0,32); # -> r^3:0:r^4:0
  523. &psllq ($D1,32);
  524. &psllq ($D2,32);
  525. &psllq ($D3,32);
  526. &psllq ($D4,32);
  527. &por ($D0,&QWP(16*0,"esp")); # r^3:r^1:r^4:r^2
  528. &por ($D1,&QWP(16*1,"esp"));
  529. &por ($D2,&QWP(16*2,"esp"));
  530. &por ($D3,&QWP(16*3,"esp"));
  531. &por ($D4,&QWP(16*4,"esp"));
  532. &pshufd ($D0,$D0,0b10001101); # -> r^1:r^2:r^3:r^4
  533. &pshufd ($D1,$D1,0b10001101);
  534. &pshufd ($D2,$D2,0b10001101);
  535. &pshufd ($D3,$D3,0b10001101);
  536. &pshufd ($D4,$D4,0b10001101);
  537. &movdqu (&QWP(16*0,"edi"),$D0); # save the table
  538. &movdqu (&QWP(16*1,"edi"),$D1);
  539. &movdqu (&QWP(16*2,"edi"),$D2);
  540. &movdqu (&QWP(16*3,"edi"),$D3);
  541. &movdqu (&QWP(16*4,"edi"),$D4);
  542. &movdqa ($T1,$D1);
  543. &movdqa ($T0,$D2);
  544. &pslld ($T1,2);
  545. &pslld ($T0,2);
  546. &paddd ($T1,$D1); # *5
  547. &paddd ($T0,$D2); # *5
  548. &movdqu (&QWP(16*5,"edi"),$T1);
  549. &movdqu (&QWP(16*6,"edi"),$T0);
  550. &movdqa ($T1,$D3);
  551. &movdqa ($T0,$D4);
  552. &pslld ($T1,2);
  553. &pslld ($T0,2);
  554. &paddd ($T1,$D3); # *5
  555. &paddd ($T0,$D4); # *5
  556. &movdqu (&QWP(16*7,"edi"),$T1);
  557. &movdqu (&QWP(16*8,"edi"),$T0);
  558. &mov ("esp","ebp");
  559. &lea ("edi",&DWP(-16*3,"edi")); # size de-optimization
  560. &ret ();
  561. &function_end_B("_poly1305_init_sse2");
  562. &align (32);
  563. &function_begin("_poly1305_blocks_sse2");
  564. &mov ("edi",&wparam(0)); # ctx
  565. &mov ("esi",&wparam(1)); # inp
  566. &mov ("ecx",&wparam(2)); # len
  567. &mov ("eax",&DWP(4*5,"edi")); # is_base2_26
  568. &and ("ecx",-16);
  569. &jz (&label("nodata"));
  570. &cmp ("ecx",64);
  571. &jae (&label("enter_sse2"));
  572. &test ("eax","eax"); # is_base2_26?
  573. &jz (&label("enter_blocks"));
  574. &set_label("enter_sse2",16);
  575. &call (&label("pic_point"));
  576. &set_label("pic_point");
  577. &blindpop("ebx");
  578. &lea ("ebx",&DWP(&label("const_sse2")."-".&label("pic_point"),"ebx"));
  579. &test ("eax","eax"); # is_base2_26?
  580. &jnz (&label("base2_26"));
  581. &call ("_poly1305_init_sse2");
  582. ################################################# base 2^32 -> base 2^26
  583. &mov ("eax",&DWP(0,"edi"));
  584. &mov ("ecx",&DWP(3,"edi"));
  585. &mov ("edx",&DWP(6,"edi"));
  586. &mov ("esi",&DWP(9,"edi"));
  587. &mov ("ebp",&DWP(13,"edi"));
  588. &mov (&DWP(4*5,"edi"),1); # is_base2_26
  589. &shr ("ecx",2);
  590. &and ("eax",0x3ffffff);
  591. &shr ("edx",4);
  592. &and ("ecx",0x3ffffff);
  593. &shr ("esi",6);
  594. &and ("edx",0x3ffffff);
  595. &movd ($D0,"eax");
  596. &movd ($D1,"ecx");
  597. &movd ($D2,"edx");
  598. &movd ($D3,"esi");
  599. &movd ($D4,"ebp");
  600. &mov ("esi",&wparam(1)); # [reload] inp
  601. &mov ("ecx",&wparam(2)); # [reload] len
  602. &jmp (&label("base2_32"));
  603. &set_label("base2_26",16);
  604. &movd ($D0,&DWP(4*0,"edi")); # load hash value
  605. &movd ($D1,&DWP(4*1,"edi"));
  606. &movd ($D2,&DWP(4*2,"edi"));
  607. &movd ($D3,&DWP(4*3,"edi"));
  608. &movd ($D4,&DWP(4*4,"edi"));
  609. &movdqa ($MASK,&QWP(64,"ebx"));
  610. &set_label("base2_32");
  611. &mov ("eax",&wparam(3)); # padbit
  612. &mov ("ebp","esp");
  613. &sub ("esp",16*(5+5+5+9+9));
  614. &and ("esp",-16);
  615. &lea ("edi",&DWP(16*3,"edi")); # size optimization
  616. &shl ("eax",24); # padbit
  617. &test ("ecx",31);
  618. &jz (&label("even"));
  619. ################################################################
  620. # process single block, with SSE2, because it's still faster
  621. # even though half of result is discarded
  622. &movdqu ($T1,&QWP(0,"esi")); # input
  623. &lea ("esi",&DWP(16,"esi"));
  624. &movdqa ($T0,$T1); # -> base 2^26 ...
  625. &pand ($T1,$MASK);
  626. &paddd ($D0,$T1); # ... and accumuate
  627. &movdqa ($T1,$T0);
  628. &psrlq ($T0,26);
  629. &psrldq ($T1,6);
  630. &pand ($T0,$MASK);
  631. &paddd ($D1,$T0);
  632. &movdqa ($T0,$T1);
  633. &psrlq ($T1,4);
  634. &pand ($T1,$MASK);
  635. &paddd ($D2,$T1);
  636. &movdqa ($T1,$T0);
  637. &psrlq ($T0,30);
  638. &pand ($T0,$MASK);
  639. &psrldq ($T1,7);
  640. &paddd ($D3,$T0);
  641. &movd ($T0,"eax"); # padbit
  642. &paddd ($D4,$T1);
  643. &movd ($T1,&DWP(16*0+12,"edi")); # r0
  644. &paddd ($D4,$T0);
  645. &movdqa (&QWP(16*0,"esp"),$D0);
  646. &movdqa (&QWP(16*1,"esp"),$D1);
  647. &movdqa (&QWP(16*2,"esp"),$D2);
  648. &movdqa (&QWP(16*3,"esp"),$D3);
  649. &movdqa (&QWP(16*4,"esp"),$D4);
  650. ################################################################
  651. # d4 = h4*r0 + h3*r1 + h2*r2 + h1*r3 + h0*r4
  652. # d3 = h3*r0 + h2*r1 + h1*r2 + h0*r3 + h4*5*r4
  653. # d2 = h2*r0 + h1*r1 + h0*r2 + h4*5*r3 + h3*5*r4
  654. # d1 = h1*r0 + h0*r1 + h4*5*r2 + h3*5*r3 + h2*5*r4
  655. # d0 = h0*r0 + h4*5*r1 + h3*5*r2 + h2*5*r3 + h1*5*r4
  656. &pmuludq ($D0,$T1); # h4*r0
  657. &pmuludq ($D1,$T1); # h3*r0
  658. &pmuludq ($D2,$T1); # h2*r0
  659. &movd ($T0,&DWP(16*1+12,"edi")); # r1
  660. &pmuludq ($D3,$T1); # h1*r0
  661. &pmuludq ($D4,$T1); # h0*r0
  662. &pmuladd (sub { my ($reg,$i)=@_;
  663. &movd ($reg,&DWP(16*$i+12,"edi"));
  664. });
  665. &lazy_reduction ();
  666. &sub ("ecx",16);
  667. &jz (&label("done"));
  668. &set_label("even");
  669. &lea ("edx",&DWP(16*(5+5+5+9),"esp"));# size optimization
  670. &lea ("eax",&DWP(-16*2,"esi"));
  671. &sub ("ecx",64);
  672. ################################################################
  673. # expand and copy pre-calculated table to stack
  674. &movdqu ($T0,&QWP(16*0,"edi")); # r^1:r^2:r^3:r^4
  675. &pshufd ($T1,$T0,0b01000100); # duplicate r^3:r^4
  676. &cmovb ("esi","eax");
  677. &pshufd ($T0,$T0,0b11101110); # duplicate r^1:r^2
  678. &movdqa (&QWP(16*0,"edx"),$T1);
  679. &lea ("eax",&DWP(16*10,"esp"));
  680. &movdqu ($T1,&QWP(16*1,"edi"));
  681. &movdqa (&QWP(16*(0-9),"edx"),$T0);
  682. &pshufd ($T0,$T1,0b01000100);
  683. &pshufd ($T1,$T1,0b11101110);
  684. &movdqa (&QWP(16*1,"edx"),$T0);
  685. &movdqu ($T0,&QWP(16*2,"edi"));
  686. &movdqa (&QWP(16*(1-9),"edx"),$T1);
  687. &pshufd ($T1,$T0,0b01000100);
  688. &pshufd ($T0,$T0,0b11101110);
  689. &movdqa (&QWP(16*2,"edx"),$T1);
  690. &movdqu ($T1,&QWP(16*3,"edi"));
  691. &movdqa (&QWP(16*(2-9),"edx"),$T0);
  692. &pshufd ($T0,$T1,0b01000100);
  693. &pshufd ($T1,$T1,0b11101110);
  694. &movdqa (&QWP(16*3,"edx"),$T0);
  695. &movdqu ($T0,&QWP(16*4,"edi"));
  696. &movdqa (&QWP(16*(3-9),"edx"),$T1);
  697. &pshufd ($T1,$T0,0b01000100);
  698. &pshufd ($T0,$T0,0b11101110);
  699. &movdqa (&QWP(16*4,"edx"),$T1);
  700. &movdqu ($T1,&QWP(16*5,"edi"));
  701. &movdqa (&QWP(16*(4-9),"edx"),$T0);
  702. &pshufd ($T0,$T1,0b01000100);
  703. &pshufd ($T1,$T1,0b11101110);
  704. &movdqa (&QWP(16*5,"edx"),$T0);
  705. &movdqu ($T0,&QWP(16*6,"edi"));
  706. &movdqa (&QWP(16*(5-9),"edx"),$T1);
  707. &pshufd ($T1,$T0,0b01000100);
  708. &pshufd ($T0,$T0,0b11101110);
  709. &movdqa (&QWP(16*6,"edx"),$T1);
  710. &movdqu ($T1,&QWP(16*7,"edi"));
  711. &movdqa (&QWP(16*(6-9),"edx"),$T0);
  712. &pshufd ($T0,$T1,0b01000100);
  713. &pshufd ($T1,$T1,0b11101110);
  714. &movdqa (&QWP(16*7,"edx"),$T0);
  715. &movdqu ($T0,&QWP(16*8,"edi"));
  716. &movdqa (&QWP(16*(7-9),"edx"),$T1);
  717. &pshufd ($T1,$T0,0b01000100);
  718. &pshufd ($T0,$T0,0b11101110);
  719. &movdqa (&QWP(16*8,"edx"),$T1);
  720. &movdqa (&QWP(16*(8-9),"edx"),$T0);
  721. sub load_input {
  722. my ($inpbase,$offbase)=@_;
  723. &movdqu ($T0,&QWP($inpbase+0,"esi")); # load input
  724. &movdqu ($T1,&QWP($inpbase+16,"esi"));
  725. &lea ("esi",&DWP(16*2,"esi"));
  726. &movdqa (&QWP($offbase+16*2,"esp"),$D2);
  727. &movdqa (&QWP($offbase+16*3,"esp"),$D3);
  728. &movdqa (&QWP($offbase+16*4,"esp"),$D4);
  729. &movdqa ($D2,$T0); # splat input
  730. &movdqa ($D3,$T1);
  731. &psrldq ($D2,6);
  732. &psrldq ($D3,6);
  733. &movdqa ($D4,$T0);
  734. &punpcklqdq ($D2,$D3); # 2:3
  735. &punpckhqdq ($D4,$T1); # 4
  736. &punpcklqdq ($T0,$T1); # 0:1
  737. &movdqa ($D3,$D2);
  738. &psrlq ($D2,4);
  739. &psrlq ($D3,30);
  740. &movdqa ($T1,$T0);
  741. &psrlq ($D4,40); # 4
  742. &psrlq ($T1,26);
  743. &pand ($T0,$MASK); # 0
  744. &pand ($T1,$MASK); # 1
  745. &pand ($D2,$MASK); # 2
  746. &pand ($D3,$MASK); # 3
  747. &por ($D4,&QWP(0,"ebx")); # padbit, yes, always
  748. &movdqa (&QWP($offbase+16*0,"esp"),$D0) if ($offbase);
  749. &movdqa (&QWP($offbase+16*1,"esp"),$D1) if ($offbase);
  750. }
  751. &load_input (16*2,16*5);
  752. &jbe (&label("skip_loop"));
  753. &jmp (&label("loop"));
  754. &set_label("loop",32);
  755. ################################################################
  756. # ((inp[0]*r^4+inp[2]*r^2+inp[4])*r^4+inp[6]*r^2
  757. # ((inp[1]*r^4+inp[3]*r^2+inp[5])*r^3+inp[7]*r
  758. # \___________________/
  759. # ((inp[0]*r^4+inp[2]*r^2+inp[4])*r^4+inp[6]*r^2+inp[8])*r^2
  760. # ((inp[1]*r^4+inp[3]*r^2+inp[5])*r^4+inp[7]*r^2+inp[9])*r
  761. # \___________________/ \____________________/
  762. ################################################################
  763. &movdqa ($T2,&QWP(16*(0-9),"edx")); # r0^2
  764. &movdqa (&QWP(16*1,"eax"),$T1);
  765. &movdqa (&QWP(16*2,"eax"),$D2);
  766. &movdqa (&QWP(16*3,"eax"),$D3);
  767. &movdqa (&QWP(16*4,"eax"),$D4);
  768. ################################################################
  769. # d4 = h4*r0 + h0*r4 + h1*r3 + h2*r2 + h3*r1
  770. # d3 = h3*r0 + h0*r3 + h1*r2 + h2*r1 + h4*5*r4
  771. # d2 = h2*r0 + h0*r2 + h1*r1 + h3*5*r4 + h4*5*r3
  772. # d1 = h1*r0 + h0*r1 + h2*5*r4 + h3*5*r3 + h4*5*r2
  773. # d0 = h0*r0 + h1*5*r4 + h2*5*r3 + h3*5*r2 + h4*5*r1
  774. &movdqa ($D1,$T0);
  775. &pmuludq ($T0,$T2); # h0*r0
  776. &movdqa ($D0,$T1);
  777. &pmuludq ($T1,$T2); # h1*r0
  778. &pmuludq ($D2,$T2); # h2*r0
  779. &pmuludq ($D3,$T2); # h3*r0
  780. &pmuludq ($D4,$T2); # h4*r0
  781. sub pmuladd_alt {
  782. my $addr = shift;
  783. &pmuludq ($D0,&$addr(8)); # h1*s4
  784. &movdqa ($T2,$D1);
  785. &pmuludq ($D1,&$addr(1)); # h0*r1
  786. &paddq ($D0,$T0);
  787. &movdqa ($T0,$T2);
  788. &pmuludq ($T2,&$addr(2)); # h0*r2
  789. &paddq ($D1,$T1);
  790. &movdqa ($T1,$T0);
  791. &pmuludq ($T0,&$addr(3)); # h0*r3
  792. &paddq ($D2,$T2);
  793. &movdqa ($T2,&QWP(16*1,"eax")); # pull h1
  794. &pmuludq ($T1,&$addr(4)); # h0*r4
  795. &paddq ($D3,$T0);
  796. &movdqa ($T0,$T2);
  797. &pmuludq ($T2,&$addr(1)); # h1*r1
  798. &paddq ($D4,$T1);
  799. &movdqa ($T1,$T0);
  800. &pmuludq ($T0,&$addr(2)); # h1*r2
  801. &paddq ($D2,$T2);
  802. &movdqa ($T2,&QWP(16*2,"eax")); # pull h2
  803. &pmuludq ($T1,&$addr(3)); # h1*r3
  804. &paddq ($D3,$T0);
  805. &movdqa ($T0,$T2);
  806. &pmuludq ($T2,&$addr(7)); # h2*s3
  807. &paddq ($D4,$T1);
  808. &movdqa ($T1,$T0);
  809. &pmuludq ($T0,&$addr(8)); # h2*s4
  810. &paddq ($D0,$T2);
  811. &movdqa ($T2,$T1);
  812. &pmuludq ($T1,&$addr(1)); # h2*r1
  813. &paddq ($D1,$T0);
  814. &movdqa ($T0,&QWP(16*3,"eax")); # pull h3
  815. &pmuludq ($T2,&$addr(2)); # h2*r2
  816. &paddq ($D3,$T1);
  817. &movdqa ($T1,$T0);
  818. &pmuludq ($T0,&$addr(6)); # h3*s2
  819. &paddq ($D4,$T2);
  820. &movdqa ($T2,$T1);
  821. &pmuludq ($T1,&$addr(7)); # h3*s3
  822. &paddq ($D0,$T0);
  823. &movdqa ($T0,$T2);
  824. &pmuludq ($T2,&$addr(8)); # h3*s4
  825. &paddq ($D1,$T1);
  826. &movdqa ($T1,&QWP(16*4,"eax")); # pull h4
  827. &pmuludq ($T0,&$addr(1)); # h3*r1
  828. &paddq ($D2,$T2);
  829. &movdqa ($T2,$T1);
  830. &pmuludq ($T1,&$addr(8)); # h4*s4
  831. &paddq ($D4,$T0);
  832. &movdqa ($T0,$T2);
  833. &pmuludq ($T2,&$addr(5)); # h4*s1
  834. &paddq ($D3,$T1);
  835. &movdqa ($T1,$T0);
  836. &pmuludq ($T0,&$addr(6)); # h4*s2
  837. &paddq ($D0,$T2);
  838. &movdqa ($MASK,&QWP(64,"ebx"));
  839. &pmuludq ($T1,&$addr(7)); # h4*s3
  840. &paddq ($D1,$T0);
  841. &paddq ($D2,$T1);
  842. }
  843. &pmuladd_alt (sub { my $i=shift; &QWP(16*($i-9),"edx"); });
  844. &load_input (-16*2,0);
  845. &lea ("eax",&DWP(-16*2,"esi"));
  846. &sub ("ecx",64);
  847. &paddd ($T0,&QWP(16*(5+0),"esp")); # add hash value
  848. &paddd ($T1,&QWP(16*(5+1),"esp"));
  849. &paddd ($D2,&QWP(16*(5+2),"esp"));
  850. &paddd ($D3,&QWP(16*(5+3),"esp"));
  851. &paddd ($D4,&QWP(16*(5+4),"esp"));
  852. &cmovb ("esi","eax");
  853. &lea ("eax",&DWP(16*10,"esp"));
  854. &movdqa ($T2,&QWP(16*0,"edx")); # r0^4
  855. &movdqa (&QWP(16*1,"esp"),$D1);
  856. &movdqa (&QWP(16*1,"eax"),$T1);
  857. &movdqa (&QWP(16*2,"eax"),$D2);
  858. &movdqa (&QWP(16*3,"eax"),$D3);
  859. &movdqa (&QWP(16*4,"eax"),$D4);
  860. ################################################################
  861. # d4 += h4*r0 + h0*r4 + h1*r3 + h2*r2 + h3*r1
  862. # d3 += h3*r0 + h0*r3 + h1*r2 + h2*r1 + h4*5*r4
  863. # d2 += h2*r0 + h0*r2 + h1*r1 + h3*5*r4 + h4*5*r3
  864. # d1 += h1*r0 + h0*r1 + h2*5*r4 + h3*5*r3 + h4*5*r2
  865. # d0 += h0*r0 + h1*5*r4 + h2*5*r3 + h3*5*r2 + h4*5*r1
  866. &movdqa ($D1,$T0);
  867. &pmuludq ($T0,$T2); # h0*r0
  868. &paddq ($T0,$D0);
  869. &movdqa ($D0,$T1);
  870. &pmuludq ($T1,$T2); # h1*r0
  871. &pmuludq ($D2,$T2); # h2*r0
  872. &pmuludq ($D3,$T2); # h3*r0
  873. &pmuludq ($D4,$T2); # h4*r0
  874. &paddq ($T1,&QWP(16*1,"esp"));
  875. &paddq ($D2,&QWP(16*2,"esp"));
  876. &paddq ($D3,&QWP(16*3,"esp"));
  877. &paddq ($D4,&QWP(16*4,"esp"));
  878. &pmuladd_alt (sub { my $i=shift; &QWP(16*$i,"edx"); });
  879. &lazy_reduction ();
  880. &load_input (16*2,16*5);
  881. &ja (&label("loop"));
  882. &set_label("skip_loop");
  883. ################################################################
  884. # multiply (inp[0:1]+hash) or inp[2:3] by r^2:r^1
  885. &pshufd ($T2,&QWP(16*(0-9),"edx"),0x10);# r0^n
  886. &add ("ecx",32);
  887. &jnz (&label("long_tail"));
  888. &paddd ($T0,$D0); # add hash value
  889. &paddd ($T1,$D1);
  890. &paddd ($D2,&QWP(16*7,"esp"));
  891. &paddd ($D3,&QWP(16*8,"esp"));
  892. &paddd ($D4,&QWP(16*9,"esp"));
  893. &set_label("long_tail");
  894. &movdqa (&QWP(16*0,"eax"),$T0);
  895. &movdqa (&QWP(16*1,"eax"),$T1);
  896. &movdqa (&QWP(16*2,"eax"),$D2);
  897. &movdqa (&QWP(16*3,"eax"),$D3);
  898. &movdqa (&QWP(16*4,"eax"),$D4);
  899. ################################################################
  900. # d4 = h4*r0 + h3*r1 + h2*r2 + h1*r3 + h0*r4
  901. # d3 = h3*r0 + h2*r1 + h1*r2 + h0*r3 + h4*5*r4
  902. # d2 = h2*r0 + h1*r1 + h0*r2 + h4*5*r3 + h3*5*r4
  903. # d1 = h1*r0 + h0*r1 + h4*5*r2 + h3*5*r3 + h2*5*r4
  904. # d0 = h0*r0 + h4*5*r1 + h3*5*r2 + h2*5*r3 + h1*5*r4
  905. &pmuludq ($T0,$T2); # h0*r0
  906. &pmuludq ($T1,$T2); # h1*r0
  907. &pmuludq ($D2,$T2); # h2*r0
  908. &movdqa ($D0,$T0);
  909. &pshufd ($T0,&QWP(16*(1-9),"edx"),0x10);# r1^n
  910. &pmuludq ($D3,$T2); # h3*r0
  911. &movdqa ($D1,$T1);
  912. &pmuludq ($D4,$T2); # h4*r0
  913. &pmuladd (sub { my ($reg,$i)=@_;
  914. &pshufd ($reg,&QWP(16*($i-9),"edx"),0x10);
  915. },"eax");
  916. &jz (&label("short_tail"));
  917. &load_input (-16*2,0);
  918. &pshufd ($T2,&QWP(16*0,"edx"),0x10); # r0^n
  919. &paddd ($T0,&QWP(16*5,"esp")); # add hash value
  920. &paddd ($T1,&QWP(16*6,"esp"));
  921. &paddd ($D2,&QWP(16*7,"esp"));
  922. &paddd ($D3,&QWP(16*8,"esp"));
  923. &paddd ($D4,&QWP(16*9,"esp"));
  924. ################################################################
  925. # multiply inp[0:1] by r^4:r^3 and accumulate
  926. &movdqa (&QWP(16*0,"esp"),$T0);
  927. &pmuludq ($T0,$T2); # h0*r0
  928. &movdqa (&QWP(16*1,"esp"),$T1);
  929. &pmuludq ($T1,$T2); # h1*r0
  930. &paddq ($D0,$T0);
  931. &movdqa ($T0,$D2);
  932. &pmuludq ($D2,$T2); # h2*r0
  933. &paddq ($D1,$T1);
  934. &movdqa ($T1,$D3);
  935. &pmuludq ($D3,$T2); # h3*r0
  936. &paddq ($D2,&QWP(16*2,"esp"));
  937. &movdqa (&QWP(16*2,"esp"),$T0);
  938. &pshufd ($T0,&QWP(16*1,"edx"),0x10); # r1^n
  939. &paddq ($D3,&QWP(16*3,"esp"));
  940. &movdqa (&QWP(16*3,"esp"),$T1);
  941. &movdqa ($T1,$D4);
  942. &pmuludq ($D4,$T2); # h4*r0
  943. &paddq ($D4,&QWP(16*4,"esp"));
  944. &movdqa (&QWP(16*4,"esp"),$T1);
  945. &pmuladd (sub { my ($reg,$i)=@_;
  946. &pshufd ($reg,&QWP(16*$i,"edx"),0x10);
  947. });
  948. &set_label("short_tail");
  949. ################################################################
  950. # horizontal addition
  951. &pshufd ($T1,$D4,0b01001110);
  952. &pshufd ($T0,$D3,0b01001110);
  953. &paddq ($D4,$T1);
  954. &paddq ($D3,$T0);
  955. &pshufd ($T1,$D0,0b01001110);
  956. &pshufd ($T0,$D1,0b01001110);
  957. &paddq ($D0,$T1);
  958. &paddq ($D1,$T0);
  959. &pshufd ($T1,$D2,0b01001110);
  960. #&paddq ($D2,$T1);
  961. &lazy_reduction (sub { &paddq ($D2,$T1) });
  962. &set_label("done");
  963. &movd (&DWP(-16*3+4*0,"edi"),$D0); # store hash value
  964. &movd (&DWP(-16*3+4*1,"edi"),$D1);
  965. &movd (&DWP(-16*3+4*2,"edi"),$D2);
  966. &movd (&DWP(-16*3+4*3,"edi"),$D3);
  967. &movd (&DWP(-16*3+4*4,"edi"),$D4);
  968. &mov ("esp","ebp");
  969. &set_label("nodata");
  970. &function_end("_poly1305_blocks_sse2");
  971. &align (32);
  972. &function_begin("_poly1305_emit_sse2");
  973. &mov ("ebp",&wparam(0)); # context
  974. &cmp (&DWP(4*5,"ebp"),0); # is_base2_26?
  975. &je (&label("enter_emit"));
  976. &mov ("eax",&DWP(4*0,"ebp")); # load hash value
  977. &mov ("edi",&DWP(4*1,"ebp"));
  978. &mov ("ecx",&DWP(4*2,"ebp"));
  979. &mov ("edx",&DWP(4*3,"ebp"));
  980. &mov ("esi",&DWP(4*4,"ebp"));
  981. &mov ("ebx","edi"); # base 2^26 -> base 2^32
  982. &shl ("edi",26);
  983. &shr ("ebx",6);
  984. &add ("eax","edi");
  985. &mov ("edi","ecx");
  986. &adc ("ebx",0);
  987. &shl ("edi",20);
  988. &shr ("ecx",12);
  989. &add ("ebx","edi");
  990. &mov ("edi","edx");
  991. &adc ("ecx",0);
  992. &shl ("edi",14);
  993. &shr ("edx",18);
  994. &add ("ecx","edi");
  995. &mov ("edi","esi");
  996. &adc ("edx",0);
  997. &shl ("edi",8);
  998. &shr ("esi",24);
  999. &add ("edx","edi");
  1000. &adc ("esi",0); # can be partially reduced
  1001. &mov ("edi","esi"); # final reduction
  1002. &and ("esi",3);
  1003. &shr ("edi",2);
  1004. &lea ("ebp",&DWP(0,"edi","edi",4)); # *5
  1005. &mov ("edi",&wparam(1)); # output
  1006. add ("eax","ebp");
  1007. &mov ("ebp",&wparam(2)); # key
  1008. adc ("ebx",0);
  1009. adc ("ecx",0);
  1010. adc ("edx",0);
  1011. &movd ($D0,"eax"); # offload original hash value
  1012. &add ("eax",5); # compare to modulus
  1013. &movd ($D1,"ebx");
  1014. &adc ("ebx",0);
  1015. &movd ($D2,"ecx");
  1016. &adc ("ecx",0);
  1017. &movd ($D3,"edx");
  1018. &adc ("edx",0);
  1019. &adc ("esi",0);
  1020. &shr ("esi",2); # did it carry/borrow?
  1021. &neg ("esi"); # do we choose (hash-modulus) ...
  1022. &and ("eax","esi");
  1023. &and ("ebx","esi");
  1024. &and ("ecx","esi");
  1025. &and ("edx","esi");
  1026. &mov (&DWP(4*0,"edi"),"eax");
  1027. &movd ("eax",$D0);
  1028. &mov (&DWP(4*1,"edi"),"ebx");
  1029. &movd ("ebx",$D1);
  1030. &mov (&DWP(4*2,"edi"),"ecx");
  1031. &movd ("ecx",$D2);
  1032. &mov (&DWP(4*3,"edi"),"edx");
  1033. &movd ("edx",$D3);
  1034. &not ("esi"); # ... or original hash value?
  1035. &and ("eax","esi");
  1036. &and ("ebx","esi");
  1037. &or ("eax",&DWP(4*0,"edi"));
  1038. &and ("ecx","esi");
  1039. &or ("ebx",&DWP(4*1,"edi"));
  1040. &and ("edx","esi");
  1041. &or ("ecx",&DWP(4*2,"edi"));
  1042. &or ("edx",&DWP(4*3,"edi"));
  1043. &add ("eax",&DWP(4*0,"ebp")); # accumulate key
  1044. &adc ("ebx",&DWP(4*1,"ebp"));
  1045. &mov (&DWP(4*0,"edi"),"eax");
  1046. &adc ("ecx",&DWP(4*2,"ebp"));
  1047. &mov (&DWP(4*1,"edi"),"ebx");
  1048. &adc ("edx",&DWP(4*3,"ebp"));
  1049. &mov (&DWP(4*2,"edi"),"ecx");
  1050. &mov (&DWP(4*3,"edi"),"edx");
  1051. &function_end("_poly1305_emit_sse2");
  1052. if ($avx>1) {
  1053. ########################################################################
  1054. # Note that poly1305_init_avx2 operates on %xmm, I could have used
  1055. # poly1305_init_sse2...
  1056. &align (32);
  1057. &function_begin_B("_poly1305_init_avx2");
  1058. &vmovdqu ($D4,&QWP(4*6,"edi")); # key base 2^32
  1059. &lea ("edi",&DWP(16*3,"edi")); # size optimization
  1060. &mov ("ebp","esp");
  1061. &sub ("esp",16*(9+5));
  1062. &and ("esp",-16);
  1063. #&vpand ($D4,$D4,&QWP(96,"ebx")); # magic mask
  1064. &vmovdqa ($MASK,&QWP(64,"ebx"));
  1065. &vpand ($D0,$D4,$MASK); # -> base 2^26
  1066. &vpsrlq ($D1,$D4,26);
  1067. &vpsrldq ($D3,$D4,6);
  1068. &vpand ($D1,$D1,$MASK);
  1069. &vpsrlq ($D2,$D3,4)
  1070. &vpsrlq ($D3,$D3,30);
  1071. &vpand ($D2,$D2,$MASK);
  1072. &vpand ($D3,$D3,$MASK);
  1073. &vpsrldq ($D4,$D4,13);
  1074. &lea ("edx",&DWP(16*9,"esp")); # size optimization
  1075. &mov ("ecx",2);
  1076. &set_label("square");
  1077. &vmovdqa (&QWP(16*0,"esp"),$D0);
  1078. &vmovdqa (&QWP(16*1,"esp"),$D1);
  1079. &vmovdqa (&QWP(16*2,"esp"),$D2);
  1080. &vmovdqa (&QWP(16*3,"esp"),$D3);
  1081. &vmovdqa (&QWP(16*4,"esp"),$D4);
  1082. &vpslld ($T1,$D1,2);
  1083. &vpslld ($T0,$D2,2);
  1084. &vpaddd ($T1,$T1,$D1); # *5
  1085. &vpaddd ($T0,$T0,$D2); # *5
  1086. &vmovdqa (&QWP(16*5,"esp"),$T1);
  1087. &vmovdqa (&QWP(16*6,"esp"),$T0);
  1088. &vpslld ($T1,$D3,2);
  1089. &vpslld ($T0,$D4,2);
  1090. &vpaddd ($T1,$T1,$D3); # *5
  1091. &vpaddd ($T0,$T0,$D4); # *5
  1092. &vmovdqa (&QWP(16*7,"esp"),$T1);
  1093. &vmovdqa (&QWP(16*8,"esp"),$T0);
  1094. &vpshufd ($T0,$D0,0b01000100);
  1095. &vmovdqa ($T1,$D1);
  1096. &vpshufd ($D1,$D1,0b01000100);
  1097. &vpshufd ($D2,$D2,0b01000100);
  1098. &vpshufd ($D3,$D3,0b01000100);
  1099. &vpshufd ($D4,$D4,0b01000100);
  1100. &vmovdqa (&QWP(16*0,"edx"),$T0);
  1101. &vmovdqa (&QWP(16*1,"edx"),$D1);
  1102. &vmovdqa (&QWP(16*2,"edx"),$D2);
  1103. &vmovdqa (&QWP(16*3,"edx"),$D3);
  1104. &vmovdqa (&QWP(16*4,"edx"),$D4);
  1105. ################################################################
  1106. # d4 = h4*r0 + h3*r1 + h2*r2 + h1*r3 + h0*r4
  1107. # d3 = h3*r0 + h2*r1 + h1*r2 + h0*r3 + h4*5*r4
  1108. # d2 = h2*r0 + h1*r1 + h0*r2 + h4*5*r3 + h3*5*r4
  1109. # d1 = h1*r0 + h0*r1 + h4*5*r2 + h3*5*r3 + h2*5*r4
  1110. # d0 = h0*r0 + h4*5*r1 + h3*5*r2 + h2*5*r3 + h1*5*r4
  1111. &vpmuludq ($D4,$D4,$D0); # h4*r0
  1112. &vpmuludq ($D3,$D3,$D0); # h3*r0
  1113. &vpmuludq ($D2,$D2,$D0); # h2*r0
  1114. &vpmuludq ($D1,$D1,$D0); # h1*r0
  1115. &vpmuludq ($D0,$T0,$D0); # h0*r0
  1116. &vpmuludq ($T0,$T1,&QWP(16*3,"edx")); # r1*h3
  1117. &vpaddq ($D4,$D4,$T0);
  1118. &vpmuludq ($T2,$T1,&QWP(16*2,"edx")); # r1*h2
  1119. &vpaddq ($D3,$D3,$T2);
  1120. &vpmuludq ($T0,$T1,&QWP(16*1,"edx")); # r1*h1
  1121. &vpaddq ($D2,$D2,$T0);
  1122. &vmovdqa ($T2,&QWP(16*5,"esp")); # s1
  1123. &vpmuludq ($T1,$T1,&QWP(16*0,"edx")); # r1*h0
  1124. &vpaddq ($D1,$D1,$T1);
  1125. &vmovdqa ($T0,&QWP(16*2,"esp")); # r2
  1126. &vpmuludq ($T2,$T2,&QWP(16*4,"edx")); # s1*h4
  1127. &vpaddq ($D0,$D0,$T2);
  1128. &vpmuludq ($T1,$T0,&QWP(16*2,"edx")); # r2*h2
  1129. &vpaddq ($D4,$D4,$T1);
  1130. &vpmuludq ($T2,$T0,&QWP(16*1,"edx")); # r2*h1
  1131. &vpaddq ($D3,$D3,$T2);
  1132. &vmovdqa ($T1,&QWP(16*6,"esp")); # s2
  1133. &vpmuludq ($T0,$T0,&QWP(16*0,"edx")); # r2*h0
  1134. &vpaddq ($D2,$D2,$T0);
  1135. &vpmuludq ($T2,$T1,&QWP(16*4,"edx")); # s2*h4
  1136. &vpaddq ($D1,$D1,$T2);
  1137. &vmovdqa ($T0,&QWP(16*3,"esp")); # r3
  1138. &vpmuludq ($T1,$T1,&QWP(16*3,"edx")); # s2*h3
  1139. &vpaddq ($D0,$D0,$T1);
  1140. &vpmuludq ($T2,$T0,&QWP(16*1,"edx")); # r3*h1
  1141. &vpaddq ($D4,$D4,$T2);
  1142. &vmovdqa ($T1,&QWP(16*7,"esp")); # s3
  1143. &vpmuludq ($T0,$T0,&QWP(16*0,"edx")); # r3*h0
  1144. &vpaddq ($D3,$D3,$T0);
  1145. &vpmuludq ($T2,$T1,&QWP(16*4,"edx")); # s3*h4
  1146. &vpaddq ($D2,$D2,$T2);
  1147. &vpmuludq ($T0,$T1,&QWP(16*3,"edx")); # s3*h3
  1148. &vpaddq ($D1,$D1,$T0);
  1149. &vmovdqa ($T2,&QWP(16*4,"esp")); # r4
  1150. &vpmuludq ($T1,$T1,&QWP(16*2,"edx")); # s3*h2
  1151. &vpaddq ($D0,$D0,$T1);
  1152. &vmovdqa ($T0,&QWP(16*8,"esp")); # s4
  1153. &vpmuludq ($T2,$T2,&QWP(16*0,"edx")); # r4*h0
  1154. &vpaddq ($D4,$D4,$T2);
  1155. &vpmuludq ($T1,$T0,&QWP(16*4,"edx")); # s4*h4
  1156. &vpaddq ($D3,$D3,$T1);
  1157. &vpmuludq ($T2,$T0,&QWP(16*1,"edx")); # s4*h1
  1158. &vpaddq ($D0,$D0,$T2);
  1159. &vpmuludq ($T1,$T0,&QWP(16*2,"edx")); # s4*h2
  1160. &vpaddq ($D1,$D1,$T1);
  1161. &vmovdqa ($MASK,&QWP(64,"ebx"));
  1162. &vpmuludq ($T0,$T0,&QWP(16*3,"edx")); # s4*h3
  1163. &vpaddq ($D2,$D2,$T0);
  1164. ################################################################
  1165. # lazy reduction
  1166. &vpsrlq ($T0,$D3,26);
  1167. &vpand ($D3,$D3,$MASK);
  1168. &vpsrlq ($T1,$D0,26);
  1169. &vpand ($D0,$D0,$MASK);
  1170. &vpaddq ($D4,$D4,$T0); # h3 -> h4
  1171. &vpaddq ($D1,$D1,$T1); # h0 -> h1
  1172. &vpsrlq ($T0,$D4,26);
  1173. &vpand ($D4,$D4,$MASK);
  1174. &vpsrlq ($T1,$D1,26);
  1175. &vpand ($D1,$D1,$MASK);
  1176. &vpaddq ($D2,$D2,$T1); # h1 -> h2
  1177. &vpaddd ($D0,$D0,$T0);
  1178. &vpsllq ($T0,$T0,2);
  1179. &vpsrlq ($T1,$D2,26);
  1180. &vpand ($D2,$D2,$MASK);
  1181. &vpaddd ($D0,$D0,$T0); # h4 -> h0
  1182. &vpaddd ($D3,$D3,$T1); # h2 -> h3
  1183. &vpsrlq ($T1,$D3,26);
  1184. &vpsrlq ($T0,$D0,26);
  1185. &vpand ($D0,$D0,$MASK);
  1186. &vpand ($D3,$D3,$MASK);
  1187. &vpaddd ($D1,$D1,$T0); # h0 -> h1
  1188. &vpaddd ($D4,$D4,$T1); # h3 -> h4
  1189. &dec ("ecx");
  1190. &jz (&label("square_break"));
  1191. &vpunpcklqdq ($D0,$D0,&QWP(16*0,"esp")); # 0:r^1:0:r^2
  1192. &vpunpcklqdq ($D1,$D1,&QWP(16*1,"esp"));
  1193. &vpunpcklqdq ($D2,$D2,&QWP(16*2,"esp"));
  1194. &vpunpcklqdq ($D3,$D3,&QWP(16*3,"esp"));
  1195. &vpunpcklqdq ($D4,$D4,&QWP(16*4,"esp"));
  1196. &jmp (&label("square"));
  1197. &set_label("square_break");
  1198. &vpsllq ($D0,$D0,32); # -> r^3:0:r^4:0
  1199. &vpsllq ($D1,$D1,32);
  1200. &vpsllq ($D2,$D2,32);
  1201. &vpsllq ($D3,$D3,32);
  1202. &vpsllq ($D4,$D4,32);
  1203. &vpor ($D0,$D0,&QWP(16*0,"esp")); # r^3:r^1:r^4:r^2
  1204. &vpor ($D1,$D1,&QWP(16*1,"esp"));
  1205. &vpor ($D2,$D2,&QWP(16*2,"esp"));
  1206. &vpor ($D3,$D3,&QWP(16*3,"esp"));
  1207. &vpor ($D4,$D4,&QWP(16*4,"esp"));
  1208. &vpshufd ($D0,$D0,0b10001101); # -> r^1:r^2:r^3:r^4
  1209. &vpshufd ($D1,$D1,0b10001101);
  1210. &vpshufd ($D2,$D2,0b10001101);
  1211. &vpshufd ($D3,$D3,0b10001101);
  1212. &vpshufd ($D4,$D4,0b10001101);
  1213. &vmovdqu (&QWP(16*0,"edi"),$D0); # save the table
  1214. &vmovdqu (&QWP(16*1,"edi"),$D1);
  1215. &vmovdqu (&QWP(16*2,"edi"),$D2);
  1216. &vmovdqu (&QWP(16*3,"edi"),$D3);
  1217. &vmovdqu (&QWP(16*4,"edi"),$D4);
  1218. &vpslld ($T1,$D1,2);
  1219. &vpslld ($T0,$D2,2);
  1220. &vpaddd ($T1,$T1,$D1); # *5
  1221. &vpaddd ($T0,$T0,$D2); # *5
  1222. &vmovdqu (&QWP(16*5,"edi"),$T1);
  1223. &vmovdqu (&QWP(16*6,"edi"),$T0);
  1224. &vpslld ($T1,$D3,2);
  1225. &vpslld ($T0,$D4,2);
  1226. &vpaddd ($T1,$T1,$D3); # *5
  1227. &vpaddd ($T0,$T0,$D4); # *5
  1228. &vmovdqu (&QWP(16*7,"edi"),$T1);
  1229. &vmovdqu (&QWP(16*8,"edi"),$T0);
  1230. &mov ("esp","ebp");
  1231. &lea ("edi",&DWP(-16*3,"edi")); # size de-optimization
  1232. &ret ();
  1233. &function_end_B("_poly1305_init_avx2");
  1234. ########################################################################
  1235. # now it's time to switch to %ymm
  1236. my ($D0,$D1,$D2,$D3,$D4,$T0,$T1,$T2)=map("ymm$_",(0..7));
  1237. my $MASK=$T2;
  1238. sub X { my $reg=shift; $reg=~s/^ymm/xmm/; $reg; }
  1239. &align (32);
  1240. &function_begin("_poly1305_blocks_avx2");
  1241. &mov ("edi",&wparam(0)); # ctx
  1242. &mov ("esi",&wparam(1)); # inp
  1243. &mov ("ecx",&wparam(2)); # len
  1244. &mov ("eax",&DWP(4*5,"edi")); # is_base2_26
  1245. &and ("ecx",-16);
  1246. &jz (&label("nodata"));
  1247. &cmp ("ecx",64);
  1248. &jae (&label("enter_avx2"));
  1249. &test ("eax","eax"); # is_base2_26?
  1250. &jz (&label("enter_blocks"));
  1251. &set_label("enter_avx2");
  1252. &vzeroupper ();
  1253. &call (&label("pic_point"));
  1254. &set_label("pic_point");
  1255. &blindpop("ebx");
  1256. &lea ("ebx",&DWP(&label("const_sse2")."-".&label("pic_point"),"ebx"));
  1257. &test ("eax","eax"); # is_base2_26?
  1258. &jnz (&label("base2_26"));
  1259. &call ("_poly1305_init_avx2");
  1260. ################################################# base 2^32 -> base 2^26
  1261. &mov ("eax",&DWP(0,"edi"));
  1262. &mov ("ecx",&DWP(3,"edi"));
  1263. &mov ("edx",&DWP(6,"edi"));
  1264. &mov ("esi",&DWP(9,"edi"));
  1265. &mov ("ebp",&DWP(13,"edi"));
  1266. &shr ("ecx",2);
  1267. &and ("eax",0x3ffffff);
  1268. &shr ("edx",4);
  1269. &and ("ecx",0x3ffffff);
  1270. &shr ("esi",6);
  1271. &and ("edx",0x3ffffff);
  1272. &mov (&DWP(4*0,"edi"),"eax");
  1273. &mov (&DWP(4*1,"edi"),"ecx");
  1274. &mov (&DWP(4*2,"edi"),"edx");
  1275. &mov (&DWP(4*3,"edi"),"esi");
  1276. &mov (&DWP(4*4,"edi"),"ebp");
  1277. &mov (&DWP(4*5,"edi"),1); # is_base2_26
  1278. &mov ("esi",&wparam(1)); # [reload] inp
  1279. &mov ("ecx",&wparam(2)); # [reload] len
  1280. &set_label("base2_26");
  1281. &mov ("eax",&wparam(3)); # padbit
  1282. &mov ("ebp","esp");
  1283. &sub ("esp",32*(5+9));
  1284. &and ("esp",-512); # ensure that frame
  1285. # doesn't cross page
  1286. # boundary, which is
  1287. # essential for
  1288. # misaligned 32-byte
  1289. # loads
  1290. ################################################################
  1291. # expand and copy pre-calculated table to stack
  1292. &vmovdqu (&X($D0),&QWP(16*(3+0),"edi"));
  1293. &lea ("edx",&DWP(32*5+128,"esp")); # +128 size optimization
  1294. &vmovdqu (&X($D1),&QWP(16*(3+1),"edi"));
  1295. &vmovdqu (&X($D2),&QWP(16*(3+2),"edi"));
  1296. &vmovdqu (&X($D3),&QWP(16*(3+3),"edi"));
  1297. &vmovdqu (&X($D4),&QWP(16*(3+4),"edi"));
  1298. &lea ("edi",&DWP(16*3,"edi")); # size optimization
  1299. &vpermq ($D0,$D0,0b01000000); # 00001234 -> 12343434
  1300. &vpermq ($D1,$D1,0b01000000);
  1301. &vpermq ($D2,$D2,0b01000000);
  1302. &vpermq ($D3,$D3,0b01000000);
  1303. &vpermq ($D4,$D4,0b01000000);
  1304. &vpshufd ($D0,$D0,0b11001000); # 12343434 -> 14243444
  1305. &vpshufd ($D1,$D1,0b11001000);
  1306. &vpshufd ($D2,$D2,0b11001000);
  1307. &vpshufd ($D3,$D3,0b11001000);
  1308. &vpshufd ($D4,$D4,0b11001000);
  1309. &vmovdqa (&QWP(32*0-128,"edx"),$D0);
  1310. &vmovdqu (&X($D0),&QWP(16*5,"edi"));
  1311. &vmovdqa (&QWP(32*1-128,"edx"),$D1);
  1312. &vmovdqu (&X($D1),&QWP(16*6,"edi"));
  1313. &vmovdqa (&QWP(32*2-128,"edx"),$D2);
  1314. &vmovdqu (&X($D2),&QWP(16*7,"edi"));
  1315. &vmovdqa (&QWP(32*3-128,"edx"),$D3);
  1316. &vmovdqu (&X($D3),&QWP(16*8,"edi"));
  1317. &vmovdqa (&QWP(32*4-128,"edx"),$D4);
  1318. &vpermq ($D0,$D0,0b01000000);
  1319. &vpermq ($D1,$D1,0b01000000);
  1320. &vpermq ($D2,$D2,0b01000000);
  1321. &vpermq ($D3,$D3,0b01000000);
  1322. &vpshufd ($D0,$D0,0b11001000);
  1323. &vpshufd ($D1,$D1,0b11001000);
  1324. &vpshufd ($D2,$D2,0b11001000);
  1325. &vpshufd ($D3,$D3,0b11001000);
  1326. &vmovdqa (&QWP(32*5-128,"edx"),$D0);
  1327. &vmovd (&X($D0),&DWP(-16*3+4*0,"edi"));# load hash value
  1328. &vmovdqa (&QWP(32*6-128,"edx"),$D1);
  1329. &vmovd (&X($D1),&DWP(-16*3+4*1,"edi"));
  1330. &vmovdqa (&QWP(32*7-128,"edx"),$D2);
  1331. &vmovd (&X($D2),&DWP(-16*3+4*2,"edi"));
  1332. &vmovdqa (&QWP(32*8-128,"edx"),$D3);
  1333. &vmovd (&X($D3),&DWP(-16*3+4*3,"edi"));
  1334. &vmovd (&X($D4),&DWP(-16*3+4*4,"edi"));
  1335. &vmovdqa ($MASK,&QWP(64,"ebx"));
  1336. &neg ("eax"); # padbit
  1337. &test ("ecx",63);
  1338. &jz (&label("even"));
  1339. &mov ("edx","ecx");
  1340. &and ("ecx",-64);
  1341. &and ("edx",63);
  1342. &vmovdqu (&X($T0),&QWP(16*0,"esi"));
  1343. &cmp ("edx",32);
  1344. &jb (&label("one"));
  1345. &vmovdqu (&X($T1),&QWP(16*1,"esi"));
  1346. &je (&label("two"));
  1347. &vinserti128 ($T0,$T0,&QWP(16*2,"esi"),1);
  1348. &lea ("esi",&DWP(16*3,"esi"));
  1349. &lea ("ebx",&DWP(8,"ebx")); # three padbits
  1350. &lea ("edx",&DWP(32*5+128+8,"esp")); # --:r^1:r^2:r^3 (*)
  1351. &jmp (&label("tail"));
  1352. &set_label("two");
  1353. &lea ("esi",&DWP(16*2,"esi"));
  1354. &lea ("ebx",&DWP(16,"ebx")); # two padbits
  1355. &lea ("edx",&DWP(32*5+128+16,"esp"));# --:--:r^1:r^2 (*)
  1356. &jmp (&label("tail"));
  1357. &set_label("one");
  1358. &lea ("esi",&DWP(16*1,"esi"));
  1359. &vpxor ($T1,$T1,$T1);
  1360. &lea ("ebx",&DWP(32,"ebx","eax",8)); # one or no padbits
  1361. &lea ("edx",&DWP(32*5+128+24,"esp"));# --:--:--:r^1 (*)
  1362. &jmp (&label("tail"));
  1363. # (*) spots marked with '--' are data from next table entry, but they
  1364. # are multiplied by 0 and therefore rendered insignificant
  1365. &set_label("even",32);
  1366. &vmovdqu (&X($T0),&QWP(16*0,"esi")); # load input
  1367. &vmovdqu (&X($T1),&QWP(16*1,"esi"));
  1368. &vinserti128 ($T0,$T0,&QWP(16*2,"esi"),1);
  1369. &vinserti128 ($T1,$T1,&QWP(16*3,"esi"),1);
  1370. &lea ("esi",&DWP(16*4,"esi"));
  1371. &sub ("ecx",64);
  1372. &jz (&label("tail"));
  1373. &set_label("loop");
  1374. ################################################################
  1375. # ((inp[0]*r^4+r[4])*r^4+r[8])*r^4
  1376. # ((inp[1]*r^4+r[5])*r^4+r[9])*r^3
  1377. # ((inp[2]*r^4+r[6])*r^4+r[10])*r^2
  1378. # ((inp[3]*r^4+r[7])*r^4+r[11])*r^1
  1379. # \________/ \_______/
  1380. ################################################################
  1381. sub vsplat_input {
  1382. &vmovdqa (&QWP(32*2,"esp"),$D2);
  1383. &vpsrldq ($D2,$T0,6); # splat input
  1384. &vmovdqa (&QWP(32*0,"esp"),$D0);
  1385. &vpsrldq ($D0,$T1,6);
  1386. &vmovdqa (&QWP(32*1,"esp"),$D1);
  1387. &vpunpckhqdq ($D1,$T0,$T1); # 4
  1388. &vpunpcklqdq ($T0,$T0,$T1); # 0:1
  1389. &vpunpcklqdq ($D2,$D2,$D0); # 2:3
  1390. &vpsrlq ($D0,$D2,30);
  1391. &vpsrlq ($D2,$D2,4);
  1392. &vpsrlq ($T1,$T0,26);
  1393. &vpsrlq ($D1,$D1,40); # 4
  1394. &vpand ($D2,$D2,$MASK); # 2
  1395. &vpand ($T0,$T0,$MASK); # 0
  1396. &vpand ($T1,$T1,$MASK); # 1
  1397. &vpand ($D0,$D0,$MASK); # 3 (*)
  1398. &vpor ($D1,$D1,&QWP(0,"ebx")); # padbit, yes, always
  1399. # (*) note that output is counterintuitive, inp[3:4] is
  1400. # returned in $D1-2, while $D3-4 are preserved;
  1401. }
  1402. &vsplat_input ();
  1403. sub vpmuladd {
  1404. my $addr = shift;
  1405. &vpaddq ($D2,$D2,&QWP(32*2,"esp")); # add hash value
  1406. &vpaddq ($T0,$T0,&QWP(32*0,"esp"));
  1407. &vpaddq ($T1,$T1,&QWP(32*1,"esp"));
  1408. &vpaddq ($D0,$D0,$D3);
  1409. &vpaddq ($D1,$D1,$D4);
  1410. ################################################################
  1411. # d3 = h2*r1 + h0*r3 + h1*r2 + h3*r0 + h4*5*r4
  1412. # d4 = h2*r2 + h0*r4 + h1*r3 + h3*r1 + h4*r0
  1413. # d0 = h2*5*r3 + h0*r0 + h1*5*r4 + h3*5*r2 + h4*5*r1
  1414. # d1 = h2*5*r4 + h0*r1 + h1*r0 + h3*5*r3 + h4*5*r2
  1415. # d2 = h2*r0 + h0*r2 + h1*r1 + h3*5*r4 + h4*5*r3
  1416. &vpmuludq ($D3,$D2,&$addr(1)); # d3 = h2*r1
  1417. &vmovdqa (QWP(32*1,"esp"),$T1);
  1418. &vpmuludq ($D4,$D2,&$addr(2)); # d4 = h2*r2
  1419. &vmovdqa (QWP(32*3,"esp"),$D0);
  1420. &vpmuludq ($D0,$D2,&$addr(7)); # d0 = h2*s3
  1421. &vmovdqa (QWP(32*4,"esp"),$D1);
  1422. &vpmuludq ($D1,$D2,&$addr(8)); # d1 = h2*s4
  1423. &vpmuludq ($D2,$D2,&$addr(0)); # d2 = h2*r0
  1424. &vpmuludq ($T2,$T0,&$addr(3)); # h0*r3
  1425. &vpaddq ($D3,$D3,$T2); # d3 += h0*r3
  1426. &vpmuludq ($T1,$T0,&$addr(4)); # h0*r4
  1427. &vpaddq ($D4,$D4,$T1); # d4 + h0*r4
  1428. &vpmuludq ($T2,$T0,&$addr(0)); # h0*r0
  1429. &vpaddq ($D0,$D0,$T2); # d0 + h0*r0
  1430. &vmovdqa ($T2,&QWP(32*1,"esp")); # h1
  1431. &vpmuludq ($T1,$T0,&$addr(1)); # h0*r1
  1432. &vpaddq ($D1,$D1,$T1); # d1 += h0*r1
  1433. &vpmuludq ($T0,$T0,&$addr(2)); # h0*r2
  1434. &vpaddq ($D2,$D2,$T0); # d2 += h0*r2
  1435. &vpmuludq ($T1,$T2,&$addr(2)); # h1*r2
  1436. &vpaddq ($D3,$D3,$T1); # d3 += h1*r2
  1437. &vpmuludq ($T0,$T2,&$addr(3)); # h1*r3
  1438. &vpaddq ($D4,$D4,$T0); # d4 += h1*r3
  1439. &vpmuludq ($T1,$T2,&$addr(8)); # h1*s4
  1440. &vpaddq ($D0,$D0,$T1); # d0 += h1*s4
  1441. &vmovdqa ($T1,&QWP(32*3,"esp")); # h3
  1442. &vpmuludq ($T0,$T2,&$addr(0)); # h1*r0
  1443. &vpaddq ($D1,$D1,$T0); # d1 += h1*r0
  1444. &vpmuludq ($T2,$T2,&$addr(1)); # h1*r1
  1445. &vpaddq ($D2,$D2,$T2); # d2 += h1*r1
  1446. &vpmuludq ($T0,$T1,&$addr(0)); # h3*r0
  1447. &vpaddq ($D3,$D3,$T0); # d3 += h3*r0
  1448. &vpmuludq ($T2,$T1,&$addr(1)); # h3*r1
  1449. &vpaddq ($D4,$D4,$T2); # d4 += h3*r1
  1450. &vpmuludq ($T0,$T1,&$addr(6)); # h3*s2
  1451. &vpaddq ($D0,$D0,$T0); # d0 += h3*s2
  1452. &vmovdqa ($T0,&QWP(32*4,"esp")); # h4
  1453. &vpmuludq ($T2,$T1,&$addr(7)); # h3*s3
  1454. &vpaddq ($D1,$D1,$T2); # d1+= h3*s3
  1455. &vpmuludq ($T1,$T1,&$addr(8)); # h3*s4
  1456. &vpaddq ($D2,$D2,$T1); # d2 += h3*s4
  1457. &vpmuludq ($T2,$T0,&$addr(8)); # h4*s4
  1458. &vpaddq ($D3,$D3,$T2); # d3 += h4*s4
  1459. &vpmuludq ($T1,$T0,&$addr(5)); # h4*s1
  1460. &vpaddq ($D0,$D0,$T1); # d0 += h4*s1
  1461. &vpmuludq ($T2,$T0,&$addr(0)); # h4*r0
  1462. &vpaddq ($D4,$D4,$T2); # d4 += h4*r0
  1463. &vmovdqa ($MASK,&QWP(64,"ebx"));
  1464. &vpmuludq ($T1,$T0,&$addr(6)); # h4*s2
  1465. &vpaddq ($D1,$D1,$T1); # d1 += h4*s2
  1466. &vpmuludq ($T0,$T0,&$addr(7)); # h4*s3
  1467. &vpaddq ($D2,$D2,$T0); # d2 += h4*s3
  1468. }
  1469. &vpmuladd (sub { my $i=shift; &QWP(32*$i-128,"edx"); });
  1470. sub vlazy_reduction {
  1471. ################################################################
  1472. # lazy reduction
  1473. &vpsrlq ($T0,$D3,26);
  1474. &vpand ($D3,$D3,$MASK);
  1475. &vpsrlq ($T1,$D0,26);
  1476. &vpand ($D0,$D0,$MASK);
  1477. &vpaddq ($D4,$D4,$T0); # h3 -> h4
  1478. &vpaddq ($D1,$D1,$T1); # h0 -> h1
  1479. &vpsrlq ($T0,$D4,26);
  1480. &vpand ($D4,$D4,$MASK);
  1481. &vpsrlq ($T1,$D1,26);
  1482. &vpand ($D1,$D1,$MASK);
  1483. &vpaddq ($D2,$D2,$T1); # h1 -> h2
  1484. &vpaddq ($D0,$D0,$T0);
  1485. &vpsllq ($T0,$T0,2);
  1486. &vpsrlq ($T1,$D2,26);
  1487. &vpand ($D2,$D2,$MASK);
  1488. &vpaddq ($D0,$D0,$T0); # h4 -> h0
  1489. &vpaddq ($D3,$D3,$T1); # h2 -> h3
  1490. &vpsrlq ($T1,$D3,26);
  1491. &vpsrlq ($T0,$D0,26);
  1492. &vpand ($D0,$D0,$MASK);
  1493. &vpand ($D3,$D3,$MASK);
  1494. &vpaddq ($D1,$D1,$T0); # h0 -> h1
  1495. &vpaddq ($D4,$D4,$T1); # h3 -> h4
  1496. }
  1497. &vlazy_reduction();
  1498. &vmovdqu (&X($T0),&QWP(16*0,"esi")); # load input
  1499. &vmovdqu (&X($T1),&QWP(16*1,"esi"));
  1500. &vinserti128 ($T0,$T0,&QWP(16*2,"esi"),1);
  1501. &vinserti128 ($T1,$T1,&QWP(16*3,"esi"),1);
  1502. &lea ("esi",&DWP(16*4,"esi"));
  1503. &sub ("ecx",64);
  1504. &jnz (&label("loop"));
  1505. &set_label("tail");
  1506. &vsplat_input ();
  1507. &and ("ebx",-64); # restore pointer
  1508. &vpmuladd (sub { my $i=shift; &QWP(4+32*$i-128,"edx"); });
  1509. ################################################################
  1510. # horizontal addition
  1511. &vpsrldq ($T0,$D4,8);
  1512. &vpsrldq ($T1,$D3,8);
  1513. &vpaddq ($D4,$D4,$T0);
  1514. &vpsrldq ($T0,$D0,8);
  1515. &vpaddq ($D3,$D3,$T1);
  1516. &vpsrldq ($T1,$D1,8);
  1517. &vpaddq ($D0,$D0,$T0);
  1518. &vpsrldq ($T0,$D2,8);
  1519. &vpaddq ($D1,$D1,$T1);
  1520. &vpermq ($T1,$D4,2); # keep folding
  1521. &vpaddq ($D2,$D2,$T0);
  1522. &vpermq ($T0,$D3,2);
  1523. &vpaddq ($D4,$D4,$T1);
  1524. &vpermq ($T1,$D0,2);
  1525. &vpaddq ($D3,$D3,$T0);
  1526. &vpermq ($T0,$D1,2);
  1527. &vpaddq ($D0,$D0,$T1);
  1528. &vpermq ($T1,$D2,2);
  1529. &vpaddq ($D1,$D1,$T0);
  1530. &vpaddq ($D2,$D2,$T1);
  1531. &vlazy_reduction();
  1532. &cmp ("ecx",0);
  1533. &je (&label("done"));
  1534. ################################################################
  1535. # clear all but single word
  1536. &vpshufd (&X($D0),&X($D0),0b11111100);
  1537. &lea ("edx",&DWP(32*5+128,"esp")); # restore pointer
  1538. &vpshufd (&X($D1),&X($D1),0b11111100);
  1539. &vpshufd (&X($D2),&X($D2),0b11111100);
  1540. &vpshufd (&X($D3),&X($D3),0b11111100);
  1541. &vpshufd (&X($D4),&X($D4),0b11111100);
  1542. &jmp (&label("even"));
  1543. &set_label("done",16);
  1544. &vmovd (&DWP(-16*3+4*0,"edi"),&X($D0));# store hash value
  1545. &vmovd (&DWP(-16*3+4*1,"edi"),&X($D1));
  1546. &vmovd (&DWP(-16*3+4*2,"edi"),&X($D2));
  1547. &vmovd (&DWP(-16*3+4*3,"edi"),&X($D3));
  1548. &vmovd (&DWP(-16*3+4*4,"edi"),&X($D4));
  1549. &vzeroupper ();
  1550. &mov ("esp","ebp");
  1551. &set_label("nodata");
  1552. &function_end("_poly1305_blocks_avx2");
  1553. }
  1554. &set_label("const_sse2",64);
  1555. &data_word(1<<24,0, 1<<24,0, 1<<24,0, 1<<24,0);
  1556. &data_word(0,0, 0,0, 0,0, 0,0);
  1557. &data_word(0x03ffffff,0,0x03ffffff,0, 0x03ffffff,0, 0x03ffffff,0);
  1558. &data_word(0x0fffffff,0x0ffffffc,0x0ffffffc,0x0ffffffc);
  1559. }
  1560. &asciz ("Poly1305 for x86, CRYPTOGAMS by <appro\@openssl.org>");
  1561. &align (4);
  1562. &asm_finish();
  1563. close STDOUT;