dc3c5067cd
Even though no test could be found to trigger this, paper-n-pencil estimate suggests that x86 and ARM inner loop lazy reductions can loose a bit in H4>>*5+H0 step. Reviewed-by: Emilia Käsper <emilia@openssl.org>
1806 lines
50 KiB
Raku
Executable file
1806 lines
50 KiB
Raku
Executable file
#!/usr/bin/env perl
|
|
#
|
|
# ====================================================================
|
|
# Written by Andy Polyakov <appro@openssl.org> for the OpenSSL
|
|
# project. The module is, however, dual licensed under OpenSSL and
|
|
# CRYPTOGAMS licenses depending on where you obtain it. For further
|
|
# details see http://www.openssl.org/~appro/cryptogams/.
|
|
# ====================================================================
|
|
#
|
|
# This module implements Poly1305 hash for x86.
|
|
#
|
|
# April 2015
|
|
#
|
|
# Numbers are cycles per processed byte with poly1305_blocks alone,
|
|
# measured with rdtsc at fixed clock frequency.
|
|
#
|
|
# IALU/gcc-3.4(*) SSE2(**) AVX2
|
|
# Pentium 15.7/+80% -
|
|
# PIII 6.21/+90% -
|
|
# P4 19.8/+40% 3.24
|
|
# Core 2 4.85/+90% 1.80
|
|
# Westmere 4.58/+100% 1.43
|
|
# Sandy Bridge 3.90/+100% 1.36
|
|
# Haswell 3.88/+70% 1.18 0.72
|
|
# Silvermont 11.0/+40% 4.80
|
|
# VIA Nano 6.71/+90% 2.47
|
|
# Sledgehammer 3.51/+180% 4.27
|
|
# Bulldozer 4.53/+140% 1.31
|
|
#
|
|
# (*) gcc 4.8 for some reason generated worse code;
|
|
# (**) besides SSE2 there are floating-point and AVX options; FP
|
|
# is deemed unnecessary, because pre-SSE2 processor are too
|
|
# old to care about, while it's not the fastest option on
|
|
# SSE2-capable ones; AVX is omitted, because it doesn't give
|
|
# a lot of improvement, 5-10% depending on processor;
|
|
|
|
$0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1;
|
|
push(@INC,"${dir}","${dir}../../perlasm");
|
|
require "x86asm.pl";
|
|
|
|
$output=pop;
|
|
open STDOUT,">$output";
|
|
|
|
&asm_init($ARGV[0],"poly1305-x86.pl",$ARGV[$#ARGV] eq "386");
|
|
|
|
$sse2=$avx=0;
|
|
for (@ARGV) { $sse2=1 if (/-DOPENSSL_IA32_SSE2/); }
|
|
|
|
if ($sse2) {
|
|
&static_label("const_sse2");
|
|
&static_label("enter_blocks");
|
|
&static_label("enter_emit");
|
|
&external_label("OPENSSL_ia32cap_P");
|
|
|
|
if (`$ENV{CC} -Wa,-v -c -o /dev/null -x assembler /dev/null 2>&1`
|
|
=~ /GNU assembler version ([2-9]\.[0-9]+)/) {
|
|
$avx = ($1>=2.19) + ($1>=2.22);
|
|
}
|
|
|
|
if (!$avx && $ARGV[0] eq "win32n" &&
|
|
`nasm -v 2>&1` =~ /NASM version ([2-9]\.[0-9]+)/) {
|
|
$avx = ($1>=2.09) + ($1>=2.10);
|
|
}
|
|
|
|
if (!$avx && `$ENV{CC} -v 2>&1` =~ /(^clang version|based on LLVM) ([3-9]\.[0-9]+)/) {
|
|
$avx = ($2>=3.0) + ($2>3.0);
|
|
}
|
|
}
|
|
|
|
########################################################################
|
|
# Layout of opaque area is following.
|
|
#
|
|
# unsigned __int32 h[5]; # current hash value base 2^32
|
|
# unsigned __int32 pad; # is_base2_26 in vector context
|
|
# unsigned __int32 r[4]; # key value base 2^32
|
|
|
|
&align(64);
|
|
&function_begin("poly1305_init");
|
|
&mov ("edi",&wparam(0)); # context
|
|
&mov ("esi",&wparam(1)); # key
|
|
&mov ("ebp",&wparam(2)); # function table
|
|
|
|
&xor ("eax","eax");
|
|
&mov (&DWP(4*0,"edi"),"eax"); # zero hash value
|
|
&mov (&DWP(4*1,"edi"),"eax");
|
|
&mov (&DWP(4*2,"edi"),"eax");
|
|
&mov (&DWP(4*3,"edi"),"eax");
|
|
&mov (&DWP(4*4,"edi"),"eax");
|
|
&mov (&DWP(4*5,"edi"),"eax"); # is_base2_26
|
|
|
|
&cmp ("esi",0);
|
|
&je (&label("nokey"));
|
|
|
|
if ($sse2) {
|
|
&call (&label("pic_point"));
|
|
&set_label("pic_point");
|
|
&blindpop("ebx");
|
|
|
|
&lea ("eax",&DWP("poly1305_blocks-".&label("pic_point"),"ebx"));
|
|
&lea ("edx",&DWP("poly1305_emit-".&label("pic_point"),"ebx"));
|
|
|
|
&picmeup("edi","OPENSSL_ia32cap_P","ebx",&label("pic_point"));
|
|
&mov ("ecx",&DWP(0,"edi"));
|
|
&and ("ecx",1<<26|1<<24);
|
|
&cmp ("ecx",1<<26|1<<24); # SSE2 and XMM?
|
|
&jne (&label("no_sse2"));
|
|
|
|
&lea ("eax",&DWP("_poly1305_blocks_sse2-".&label("pic_point"),"ebx"));
|
|
&lea ("edx",&DWP("_poly1305_emit_sse2-".&label("pic_point"),"ebx"));
|
|
|
|
if ($avx>1) {
|
|
&mov ("ecx",&DWP(8,"edi"));
|
|
&test ("ecx",1<<5); # AVX2?
|
|
&jz (&label("no_sse2"));
|
|
|
|
&lea ("eax",&DWP("_poly1305_blocks_avx2-".&label("pic_point"),"ebx"));
|
|
}
|
|
&set_label("no_sse2");
|
|
&mov ("edi",&wparam(0)); # reload context
|
|
&mov (&DWP(0,"ebp"),"eax"); # fill function table
|
|
&mov (&DWP(4,"ebp"),"edx");
|
|
}
|
|
|
|
&mov ("eax",&DWP(4*0,"esi")); # load input key
|
|
&mov ("ebx",&DWP(4*1,"esi"));
|
|
&mov ("ecx",&DWP(4*2,"esi"));
|
|
&mov ("edx",&DWP(4*3,"esi"));
|
|
&and ("eax",0x0fffffff);
|
|
&and ("ebx",0x0ffffffc);
|
|
&and ("ecx",0x0ffffffc);
|
|
&and ("edx",0x0ffffffc);
|
|
&mov (&DWP(4*6,"edi"),"eax");
|
|
&mov (&DWP(4*7,"edi"),"ebx");
|
|
&mov (&DWP(4*8,"edi"),"ecx");
|
|
&mov (&DWP(4*9,"edi"),"edx");
|
|
|
|
&mov ("eax",$sse2);
|
|
&set_label("nokey");
|
|
&function_end("poly1305_init");
|
|
|
|
($h0,$h1,$h2,$h3,$h4,
|
|
$d0,$d1,$d2,$d3,
|
|
$r0,$r1,$r2,$r3,
|
|
$s1,$s2,$s3)=map(4*$_,(0..15));
|
|
|
|
&function_begin("poly1305_blocks");
|
|
&mov ("edi",&wparam(0)); # ctx
|
|
&mov ("esi",&wparam(1)); # inp
|
|
&mov ("ecx",&wparam(2)); # len
|
|
&set_label("enter_blocks");
|
|
&and ("ecx",-15);
|
|
&jz (&label("nodata"));
|
|
|
|
&stack_push(16);
|
|
&mov ("eax",&DWP(4*6,"edi")); # r0
|
|
&mov ("ebx",&DWP(4*7,"edi")); # r1
|
|
&lea ("ebp",&DWP(0,"esi","ecx")); # end of input
|
|
&mov ("ecx",&DWP(4*8,"edi")); # r2
|
|
&mov ("edx",&DWP(4*9,"edi")); # r3
|
|
|
|
&mov (&wparam(2),"ebp");
|
|
&mov ("ebp","esi");
|
|
|
|
&mov (&DWP($r0,"esp"),"eax"); # r0
|
|
&mov ("eax","ebx");
|
|
&shr ("eax",2);
|
|
&mov (&DWP($r1,"esp"),"ebx"); # r1
|
|
&add ("eax","ebx"); # s1
|
|
&mov ("ebx","ecx");
|
|
&shr ("ebx",2);
|
|
&mov (&DWP($r2,"esp"),"ecx"); # r2
|
|
&add ("ebx","ecx"); # s2
|
|
&mov ("ecx","edx");
|
|
&shr ("ecx",2);
|
|
&mov (&DWP($r3,"esp"),"edx"); # r3
|
|
&add ("ecx","edx"); # s3
|
|
&mov (&DWP($s1,"esp"),"eax"); # s1
|
|
&mov (&DWP($s2,"esp"),"ebx"); # s2
|
|
&mov (&DWP($s3,"esp"),"ecx"); # s3
|
|
|
|
&mov ("eax",&DWP(4*0,"edi")); # load hash value
|
|
&mov ("ebx",&DWP(4*1,"edi"));
|
|
&mov ("ecx",&DWP(4*2,"edi"));
|
|
&mov ("esi",&DWP(4*3,"edi"));
|
|
&mov ("edi",&DWP(4*4,"edi"));
|
|
&jmp (&label("loop"));
|
|
|
|
&set_label("loop",32);
|
|
&add ("eax",&DWP(4*0,"ebp")); # accumulate input
|
|
&adc ("ebx",&DWP(4*1,"ebp"));
|
|
&adc ("ecx",&DWP(4*2,"ebp"));
|
|
&adc ("esi",&DWP(4*3,"ebp"));
|
|
&lea ("ebp",&DWP(4*4,"ebp"));
|
|
&adc ("edi",&wparam(3)); # padbit
|
|
|
|
&mov (&DWP($h0,"esp"),"eax"); # put aside hash[+inp]
|
|
&mov (&DWP($h3,"esp"),"esi");
|
|
|
|
&mul (&DWP($r0,"esp")); # h0*r0
|
|
&mov (&DWP($h4,"esp"),"edi");
|
|
&mov ("edi","eax");
|
|
&mov ("eax","ebx"); # h1
|
|
&mov ("esi","edx");
|
|
&mul (&DWP($s3,"esp")); # h1*s3
|
|
&add ("edi","eax");
|
|
&mov ("eax","ecx"); # h2
|
|
&adc ("esi","edx");
|
|
&mul (&DWP($s2,"esp")); # h2*s2
|
|
&add ("edi","eax");
|
|
&mov ("eax",&DWP($h3,"esp"));
|
|
&adc ("esi","edx");
|
|
&mul (&DWP($s1,"esp")); # h3*s1
|
|
&add ("edi","eax");
|
|
&mov ("eax",&DWP($h0,"esp"));
|
|
&adc ("esi","edx");
|
|
|
|
&mul (&DWP($r1,"esp")); # h0*r1
|
|
&mov (&DWP($d0,"esp"),"edi");
|
|
&xor ("edi","edi");
|
|
&add ("esi","eax");
|
|
&mov ("eax","ebx"); # h1
|
|
&adc ("edi","edx");
|
|
&mul (&DWP($r0,"esp")); # h1*r0
|
|
&add ("esi","eax");
|
|
&mov ("eax","ecx"); # h2
|
|
&adc ("edi","edx");
|
|
&mul (&DWP($s3,"esp")); # h2*s3
|
|
&add ("esi","eax");
|
|
&mov ("eax",&DWP($h3,"esp"));
|
|
&adc ("edi","edx");
|
|
&mul (&DWP($s2,"esp")); # h3*s2
|
|
&add ("esi","eax");
|
|
&mov ("eax",&DWP($h4,"esp"));
|
|
&adc ("edi","edx");
|
|
&imul ("eax",&DWP($s1,"esp")); # h4*s1
|
|
&add ("esi","eax");
|
|
&mov ("eax",&DWP($h0,"esp"));
|
|
&adc ("edi",0);
|
|
|
|
&mul (&DWP($r2,"esp")); # h0*r2
|
|
&mov (&DWP($d1,"esp"),"esi");
|
|
&xor ("esi","esi");
|
|
&add ("edi","eax");
|
|
&mov ("eax","ebx"); # h1
|
|
&adc ("esi","edx");
|
|
&mul (&DWP($r1,"esp")); # h1*r1
|
|
&add ("edi","eax");
|
|
&mov ("eax","ecx"); # h2
|
|
&adc ("esi","edx");
|
|
&mul (&DWP($r0,"esp")); # h2*r0
|
|
&add ("edi","eax");
|
|
&mov ("eax",&DWP($h3,"esp"));
|
|
&adc ("esi","edx");
|
|
&mul (&DWP($s3,"esp")); # h3*s3
|
|
&add ("edi","eax");
|
|
&mov ("eax",&DWP($h4,"esp"));
|
|
&adc ("esi","edx");
|
|
&imul ("eax",&DWP($s2,"esp")); # h4*s2
|
|
&add ("edi","eax");
|
|
&mov ("eax",&DWP($h0,"esp"));
|
|
&adc ("esi",0);
|
|
|
|
&mul (&DWP($r3,"esp")); # h0*r3
|
|
&mov (&DWP($d2,"esp"),"edi");
|
|
&xor ("edi","edi");
|
|
&add ("esi","eax");
|
|
&mov ("eax","ebx"); # h1
|
|
&adc ("edi","edx");
|
|
&mul (&DWP($r2,"esp")); # h1*r2
|
|
&add ("esi","eax");
|
|
&mov ("eax","ecx"); # h2
|
|
&adc ("edi","edx");
|
|
&mul (&DWP($r1,"esp")); # h2*r1
|
|
&add ("esi","eax");
|
|
&mov ("eax",&DWP($h3,"esp"));
|
|
&adc ("edi","edx");
|
|
&mul (&DWP($r0,"esp")); # h3*r0
|
|
&add ("esi","eax");
|
|
&mov ("ecx",&DWP($h4,"esp"));
|
|
&adc ("edi","edx");
|
|
|
|
&mov ("edx","ecx");
|
|
&imul ("ecx",&DWP($s3,"esp")); # h4*s3
|
|
&add ("esi","ecx");
|
|
&mov ("eax",&DWP($d0,"esp"));
|
|
&adc ("edi",0);
|
|
|
|
&imul ("edx",&DWP($r0,"esp")); # h4*r0
|
|
&add ("edx","edi");
|
|
|
|
&mov ("ebx",&DWP($d1,"esp"));
|
|
&mov ("ecx",&DWP($d2,"esp"));
|
|
|
|
&mov ("edi","edx"); # last reduction step
|
|
&shr ("edx",2);
|
|
&and ("edi",3);
|
|
&lea ("edx",&DWP(0,"edx","edx",4)); # *5
|
|
&add ("eax","edx");
|
|
&adc ("ebx",0);
|
|
&adc ("ecx",0);
|
|
&adc ("esi",0);
|
|
&adc ("edi",0);
|
|
|
|
&cmp ("ebp",&wparam(2)); # done yet?
|
|
&jne (&label("loop"));
|
|
|
|
&mov ("edx",&wparam(0)); # ctx
|
|
&stack_pop(16);
|
|
&mov (&DWP(4*0,"edx"),"eax"); # store hash value
|
|
&mov (&DWP(4*1,"edx"),"ebx");
|
|
&mov (&DWP(4*2,"edx"),"ecx");
|
|
&mov (&DWP(4*3,"edx"),"esi");
|
|
&mov (&DWP(4*4,"edx"),"edi");
|
|
&set_label("nodata");
|
|
&function_end("poly1305_blocks");
|
|
|
|
&function_begin("poly1305_emit");
|
|
&mov ("ebp",&wparam(0)); # context
|
|
&set_label("enter_emit");
|
|
&mov ("edi",&wparam(1)); # output
|
|
&mov ("eax",&DWP(4*0,"ebp")); # load hash value
|
|
&mov ("ebx",&DWP(4*1,"ebp"));
|
|
&mov ("ecx",&DWP(4*2,"ebp"));
|
|
&mov ("edx",&DWP(4*3,"ebp"));
|
|
&mov ("esi",&DWP(4*4,"ebp"));
|
|
|
|
&add ("eax",5); # compare to modulus
|
|
&adc ("ebx",0);
|
|
&adc ("ecx",0);
|
|
&adc ("edx",0);
|
|
&adc ("esi",0);
|
|
&shr ("esi",2); # did it carry/borrow?
|
|
&neg ("esi"); # do we choose hash-modulus?
|
|
|
|
&and ("eax","esi");
|
|
&and ("ebx","esi");
|
|
&and ("ecx","esi");
|
|
&and ("edx","esi");
|
|
&mov (&DWP(4*0,"edi"),"eax");
|
|
&mov (&DWP(4*1,"edi"),"ebx");
|
|
&mov (&DWP(4*2,"edi"),"ecx");
|
|
&mov (&DWP(4*3,"edi"),"edx");
|
|
|
|
¬ ("esi"); # or original hash value?
|
|
&mov ("eax",&DWP(4*0,"ebp"));
|
|
&mov ("ebx",&DWP(4*1,"ebp"));
|
|
&mov ("ecx",&DWP(4*2,"ebp"));
|
|
&mov ("edx",&DWP(4*3,"ebp"));
|
|
&mov ("ebp",&wparam(2));
|
|
&and ("eax","esi");
|
|
&and ("ebx","esi");
|
|
&and ("ecx","esi");
|
|
&and ("edx","esi");
|
|
&or ("eax",&DWP(4*0,"edi"));
|
|
&or ("ebx",&DWP(4*1,"edi"));
|
|
&or ("ecx",&DWP(4*2,"edi"));
|
|
&or ("edx",&DWP(4*3,"edi"));
|
|
|
|
&add ("eax",&DWP(4*0,"ebp")); # accumulate key
|
|
&adc ("ebx",&DWP(4*1,"ebp"));
|
|
&adc ("ecx",&DWP(4*2,"ebp"));
|
|
&adc ("edx",&DWP(4*3,"ebp"));
|
|
|
|
&mov (&DWP(4*0,"edi"),"eax");
|
|
&mov (&DWP(4*1,"edi"),"ebx");
|
|
&mov (&DWP(4*2,"edi"),"ecx");
|
|
&mov (&DWP(4*3,"edi"),"edx");
|
|
&function_end("poly1305_emit");
|
|
|
|
if ($sse2) {
|
|
########################################################################
|
|
# Layout of opaque area is following.
|
|
#
|
|
# unsigned __int32 h[5]; # current hash value base 2^26
|
|
# unsigned __int32 is_base2_26;
|
|
# unsigned __int32 r[4]; # key value base 2^32
|
|
# unsigned __int32 pad[2];
|
|
# struct { unsigned __int32 r^4, r^3, r^2, r^1; } r[9];
|
|
#
|
|
# where r^n are base 2^26 digits of degrees of multiplier key. There are
|
|
# 5 digits, but last four are interleaved with multiples of 5, totalling
|
|
# in 9 elements: r0, r1, 5*r1, r2, 5*r2, r3, 5*r3, r4, 5*r4.
|
|
|
|
my ($D0,$D1,$D2,$D3,$D4,$T0,$T1,$T2)=map("xmm$_",(0..7));
|
|
my $MASK=$T2; # borrow and keep in mind
|
|
|
|
&align (32);
|
|
&function_begin_B("_poly1305_init_sse2");
|
|
&movdqu ($D4,&QWP(4*6,"edi")); # key base 2^32
|
|
&lea ("edi",&DWP(16*3,"edi")); # size optimization
|
|
&mov ("ebp","esp");
|
|
&sub ("esp",16*(9+5));
|
|
&and ("esp",-16);
|
|
|
|
#&pand ($D4,&QWP(96,"ebx")); # magic mask
|
|
&movq ($MASK,&QWP(64,"ebx"));
|
|
|
|
&movdqa ($D0,$D4);
|
|
&movdqa ($D1,$D4);
|
|
&movdqa ($D2,$D4);
|
|
|
|
&pand ($D0,$MASK); # -> base 2^26
|
|
&psrlq ($D1,26);
|
|
&psrldq ($D2,6);
|
|
&pand ($D1,$MASK);
|
|
&movdqa ($D3,$D2);
|
|
&psrlq ($D2,4)
|
|
&psrlq ($D3,30);
|
|
&pand ($D2,$MASK);
|
|
&pand ($D3,$MASK);
|
|
&psrldq ($D4,13);
|
|
|
|
&lea ("edx",&DWP(16*9,"esp")); # size optimization
|
|
&mov ("ecx",2);
|
|
&set_label("square");
|
|
&movdqa (&QWP(16*0,"esp"),$D0);
|
|
&movdqa (&QWP(16*1,"esp"),$D1);
|
|
&movdqa (&QWP(16*2,"esp"),$D2);
|
|
&movdqa (&QWP(16*3,"esp"),$D3);
|
|
&movdqa (&QWP(16*4,"esp"),$D4);
|
|
|
|
&movdqa ($T1,$D1);
|
|
&movdqa ($T0,$D2);
|
|
&pslld ($T1,2);
|
|
&pslld ($T0,2);
|
|
&paddd ($T1,$D1); # *5
|
|
&paddd ($T0,$D2); # *5
|
|
&movdqa (&QWP(16*5,"esp"),$T1);
|
|
&movdqa (&QWP(16*6,"esp"),$T0);
|
|
&movdqa ($T1,$D3);
|
|
&movdqa ($T0,$D4);
|
|
&pslld ($T1,2);
|
|
&pslld ($T0,2);
|
|
&paddd ($T1,$D3); # *5
|
|
&paddd ($T0,$D4); # *5
|
|
&movdqa (&QWP(16*7,"esp"),$T1);
|
|
&movdqa (&QWP(16*8,"esp"),$T0);
|
|
|
|
&pshufd ($T1,$D0,0b01000100);
|
|
&movdqa ($T0,$D1);
|
|
&pshufd ($D1,$D1,0b01000100);
|
|
&pshufd ($D2,$D2,0b01000100);
|
|
&pshufd ($D3,$D3,0b01000100);
|
|
&pshufd ($D4,$D4,0b01000100);
|
|
&movdqa (&QWP(16*0,"edx"),$T1);
|
|
&movdqa (&QWP(16*1,"edx"),$D1);
|
|
&movdqa (&QWP(16*2,"edx"),$D2);
|
|
&movdqa (&QWP(16*3,"edx"),$D3);
|
|
&movdqa (&QWP(16*4,"edx"),$D4);
|
|
|
|
################################################################
|
|
# d4 = h4*r0 + h3*r1 + h2*r2 + h1*r3 + h0*r4
|
|
# d3 = h3*r0 + h2*r1 + h1*r2 + h0*r3 + h4*5*r4
|
|
# d2 = h2*r0 + h1*r1 + h0*r2 + h4*5*r3 + h3*5*r4
|
|
# d1 = h1*r0 + h0*r1 + h4*5*r2 + h3*5*r3 + h2*5*r4
|
|
# d0 = h0*r0 + h4*5*r1 + h3*5*r2 + h2*5*r3 + h1*5*r4
|
|
|
|
&pmuludq ($D4,$D0); # h4*r0
|
|
&pmuludq ($D3,$D0); # h3*r0
|
|
&pmuludq ($D2,$D0); # h2*r0
|
|
&pmuludq ($D1,$D0); # h1*r0
|
|
&pmuludq ($D0,$T1); # h0*r0
|
|
|
|
sub pmuladd {
|
|
my $load = shift;
|
|
my $base = shift; $base = "esp" if (!defined($base));
|
|
|
|
################################################################
|
|
# As for choice to "rotate" $T0-$T2 in order to move paddq
|
|
# past next multiplication. While it makes code harder to read
|
|
# and doesn't have significant effect on most processors, it
|
|
# makes a lot of difference on Atom, up to 30% improvement.
|
|
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&QWP(16*3,$base)); # r1*h3
|
|
&movdqa ($T2,$T1);
|
|
&pmuludq ($T1,&QWP(16*2,$base)); # r1*h2
|
|
&paddq ($D4,$T0);
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&QWP(16*1,$base)); # r1*h1
|
|
&paddq ($D3,$T1);
|
|
&$load ($T1,5); # s1
|
|
&pmuludq ($T0,&QWP(16*0,$base)); # r1*h0
|
|
&paddq ($D2,$T2);
|
|
&pmuludq ($T1,&QWP(16*4,$base)); # s1*h4
|
|
&$load ($T2,2); # r2^n
|
|
&paddq ($D1,$T0);
|
|
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&QWP(16*2,$base)); # r2*h2
|
|
&paddq ($D0,$T1);
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&QWP(16*1,$base)); # r2*h1
|
|
&paddq ($D4,$T2);
|
|
&$load ($T2,6); # s2^n
|
|
&pmuludq ($T1,&QWP(16*0,$base)); # r2*h0
|
|
&paddq ($D3,$T0);
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&QWP(16*4,$base)); # s2*h4
|
|
&paddq ($D2,$T1);
|
|
&pmuludq ($T0,&QWP(16*3,$base)); # s2*h3
|
|
&$load ($T1,3); # r3^n
|
|
&paddq ($D1,$T2);
|
|
|
|
&movdqa ($T2,$T1);
|
|
&pmuludq ($T1,&QWP(16*1,$base)); # r3*h1
|
|
&paddq ($D0,$T0);
|
|
&$load ($T0,7); # s3^n
|
|
&pmuludq ($T2,&QWP(16*0,$base)); # r3*h0
|
|
&paddq ($D4,$T1);
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&QWP(16*4,$base)); # s3*h4
|
|
&paddq ($D3,$T2);
|
|
&movdqa ($T2,$T1);
|
|
&pmuludq ($T1,&QWP(16*3,$base)); # s3*h3
|
|
&paddq ($D2,$T0);
|
|
&pmuludq ($T2,&QWP(16*2,$base)); # s3*h2
|
|
&$load ($T0,4); # r4^n
|
|
&paddq ($D1,$T1);
|
|
|
|
&$load ($T1,8); # s4^n
|
|
&pmuludq ($T0,&QWP(16*0,$base)); # r4*h0
|
|
&paddq ($D0,$T2);
|
|
&movdqa ($T2,$T1);
|
|
&pmuludq ($T1,&QWP(16*4,$base)); # s4*h4
|
|
&paddq ($D4,$T0);
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&QWP(16*1,$base)); # s4*h1
|
|
&paddq ($D3,$T1);
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&QWP(16*2,$base)); # s4*h2
|
|
&paddq ($D0,$T2);
|
|
&pmuludq ($T1,&QWP(16*3,$base)); # s4*h3
|
|
&movdqa ($MASK,&QWP(64,"ebx"));
|
|
&paddq ($D1,$T0);
|
|
&paddq ($D2,$T1);
|
|
}
|
|
&pmuladd (sub { my ($reg,$i)=@_;
|
|
&movdqa ($reg,&QWP(16*$i,"esp"));
|
|
},"edx");
|
|
|
|
sub lazy_reduction {
|
|
my $extra = shift;
|
|
|
|
################################################################
|
|
# lazy reduction as discussed in "NEON crypto" by D.J. Bernstein
|
|
# and P. Schwabe
|
|
#
|
|
# [(*) see discussion in poly1305-armv4 module]
|
|
|
|
&movdqa ($T0,$D3);
|
|
&pand ($D3,$MASK);
|
|
&psrlq ($T0,26);
|
|
&$extra () if (defined($extra));
|
|
&paddq ($T0,$D4); # h3 -> h4
|
|
&movdqa ($T1,$D0);
|
|
&pand ($D0,$MASK);
|
|
&psrlq ($T1,26);
|
|
&movdqa ($D4,$T0);
|
|
&paddq ($T1,$D1); # h0 -> h1
|
|
&psrlq ($T0,26);
|
|
&pand ($D4,$MASK);
|
|
&movdqa ($D1,$T1);
|
|
&psrlq ($T1,26);
|
|
&paddd ($D0,$T0); # favour paddd when
|
|
# possible, because
|
|
# paddq is "broken"
|
|
# on Atom
|
|
&psllq ($T0,2);
|
|
&paddq ($T1,$D2); # h1 -> h2
|
|
&paddq ($T0,$D0); # h4 -> h0 (*)
|
|
&pand ($D1,$MASK);
|
|
&movdqa ($D2,$T1);
|
|
&psrlq ($T1,26);
|
|
&pand ($D2,$MASK);
|
|
&paddd ($T1,$D3); # h2 -> h3
|
|
&movdqa ($D0,$T0);
|
|
&psrlq ($T0,26);
|
|
&movdqa ($D3,$T1);
|
|
&psrlq ($T1,26);
|
|
&pand ($D0,$MASK);
|
|
&paddd ($D1,$T0); # h0 -> h1
|
|
&pand ($D3,$MASK);
|
|
&paddd ($D4,$T1); # h3 -> h4
|
|
}
|
|
&lazy_reduction ();
|
|
|
|
&dec ("ecx");
|
|
&jz (&label("square_break"));
|
|
|
|
&punpcklqdq ($D0,&QWP(16*0,"esp")); # 0:r^1:0:r^2
|
|
&punpcklqdq ($D1,&QWP(16*1,"esp"));
|
|
&punpcklqdq ($D2,&QWP(16*2,"esp"));
|
|
&punpcklqdq ($D3,&QWP(16*3,"esp"));
|
|
&punpcklqdq ($D4,&QWP(16*4,"esp"));
|
|
&jmp (&label("square"));
|
|
|
|
&set_label("square_break");
|
|
&psllq ($D0,32); # -> r^3:0:r^4:0
|
|
&psllq ($D1,32);
|
|
&psllq ($D2,32);
|
|
&psllq ($D3,32);
|
|
&psllq ($D4,32);
|
|
&por ($D0,&QWP(16*0,"esp")); # r^3:r^1:r^4:r^2
|
|
&por ($D1,&QWP(16*1,"esp"));
|
|
&por ($D2,&QWP(16*2,"esp"));
|
|
&por ($D3,&QWP(16*3,"esp"));
|
|
&por ($D4,&QWP(16*4,"esp"));
|
|
|
|
&pshufd ($D0,$D0,0b10001101); # -> r^1:r^2:r^3:r^4
|
|
&pshufd ($D1,$D1,0b10001101);
|
|
&pshufd ($D2,$D2,0b10001101);
|
|
&pshufd ($D3,$D3,0b10001101);
|
|
&pshufd ($D4,$D4,0b10001101);
|
|
|
|
&movdqu (&QWP(16*0,"edi"),$D0); # save the table
|
|
&movdqu (&QWP(16*1,"edi"),$D1);
|
|
&movdqu (&QWP(16*2,"edi"),$D2);
|
|
&movdqu (&QWP(16*3,"edi"),$D3);
|
|
&movdqu (&QWP(16*4,"edi"),$D4);
|
|
|
|
&movdqa ($T1,$D1);
|
|
&movdqa ($T0,$D2);
|
|
&pslld ($T1,2);
|
|
&pslld ($T0,2);
|
|
&paddd ($T1,$D1); # *5
|
|
&paddd ($T0,$D2); # *5
|
|
&movdqu (&QWP(16*5,"edi"),$T1);
|
|
&movdqu (&QWP(16*6,"edi"),$T0);
|
|
&movdqa ($T1,$D3);
|
|
&movdqa ($T0,$D4);
|
|
&pslld ($T1,2);
|
|
&pslld ($T0,2);
|
|
&paddd ($T1,$D3); # *5
|
|
&paddd ($T0,$D4); # *5
|
|
&movdqu (&QWP(16*7,"edi"),$T1);
|
|
&movdqu (&QWP(16*8,"edi"),$T0);
|
|
|
|
&mov ("esp","ebp");
|
|
&lea ("edi",&DWP(-16*3,"edi")); # size de-optimization
|
|
&ret ();
|
|
&function_end_B("_poly1305_init_sse2");
|
|
|
|
&align (32);
|
|
&function_begin("_poly1305_blocks_sse2");
|
|
&mov ("edi",&wparam(0)); # ctx
|
|
&mov ("esi",&wparam(1)); # inp
|
|
&mov ("ecx",&wparam(2)); # len
|
|
|
|
&mov ("eax",&DWP(4*5,"edi")); # is_base2_26
|
|
&and ("ecx",-16);
|
|
&jz (&label("nodata"));
|
|
&cmp ("ecx",64);
|
|
&jae (&label("enter_sse2"));
|
|
&test ("eax","eax"); # is_base2_26?
|
|
&jz (&label("enter_blocks"));
|
|
|
|
&set_label("enter_sse2",16);
|
|
&call (&label("pic_point"));
|
|
&set_label("pic_point");
|
|
&blindpop("ebx");
|
|
&lea ("ebx",&DWP(&label("const_sse2")."-".&label("pic_point"),"ebx"));
|
|
|
|
&test ("eax","eax"); # is_base2_26?
|
|
&jnz (&label("base2_26"));
|
|
|
|
&call ("_poly1305_init_sse2");
|
|
|
|
################################################# base 2^32 -> base 2^26
|
|
&mov ("eax",&DWP(0,"edi"));
|
|
&mov ("ecx",&DWP(3,"edi"));
|
|
&mov ("edx",&DWP(6,"edi"));
|
|
&mov ("esi",&DWP(9,"edi"));
|
|
&mov ("ebp",&DWP(13,"edi"));
|
|
&mov (&DWP(4*5,"edi"),1); # is_base2_26
|
|
|
|
&shr ("ecx",2);
|
|
&and ("eax",0x3ffffff);
|
|
&shr ("edx",4);
|
|
&and ("ecx",0x3ffffff);
|
|
&shr ("esi",6);
|
|
&and ("edx",0x3ffffff);
|
|
|
|
&movd ($D0,"eax");
|
|
&movd ($D1,"ecx");
|
|
&movd ($D2,"edx");
|
|
&movd ($D3,"esi");
|
|
&movd ($D4,"ebp");
|
|
|
|
&mov ("esi",&wparam(1)); # [reload] inp
|
|
&mov ("ecx",&wparam(2)); # [reload] len
|
|
&jmp (&label("base2_32"));
|
|
|
|
&set_label("base2_26",16);
|
|
&movd ($D0,&DWP(4*0,"edi")); # load hash value
|
|
&movd ($D1,&DWP(4*1,"edi"));
|
|
&movd ($D2,&DWP(4*2,"edi"));
|
|
&movd ($D3,&DWP(4*3,"edi"));
|
|
&movd ($D4,&DWP(4*4,"edi"));
|
|
&movdqa ($MASK,&QWP(64,"ebx"));
|
|
|
|
&set_label("base2_32");
|
|
&mov ("eax",&wparam(3)); # padbit
|
|
&mov ("ebp","esp");
|
|
|
|
&sub ("esp",16*(5+5+5+9+9));
|
|
&and ("esp",-16);
|
|
|
|
&lea ("edi",&DWP(16*3,"edi")); # size optimization
|
|
&shl ("eax",24); # padbit
|
|
|
|
&test ("ecx",31);
|
|
&jz (&label("even"));
|
|
|
|
################################################################
|
|
# process single block, with SSE2, because it's still faster
|
|
# even though half of result is discarded
|
|
|
|
&movdqu ($T1,&QWP(0,"esi")); # input
|
|
&lea ("esi",&DWP(16,"esi"));
|
|
|
|
&movdqa ($T0,$T1); # -> base 2^26 ...
|
|
&pand ($T1,$MASK);
|
|
&paddd ($D0,$T1); # ... and accumuate
|
|
|
|
&movdqa ($T1,$T0);
|
|
&psrlq ($T0,26);
|
|
&psrldq ($T1,6);
|
|
&pand ($T0,$MASK);
|
|
&paddd ($D1,$T0);
|
|
|
|
&movdqa ($T0,$T1);
|
|
&psrlq ($T1,4);
|
|
&pand ($T1,$MASK);
|
|
&paddd ($D2,$T1);
|
|
|
|
&movdqa ($T1,$T0);
|
|
&psrlq ($T0,30);
|
|
&pand ($T0,$MASK);
|
|
&psrldq ($T1,7);
|
|
&paddd ($D3,$T0);
|
|
|
|
&movd ($T0,"eax"); # padbit
|
|
&paddd ($D4,$T1);
|
|
&movd ($T1,&DWP(16*0+12,"edi")); # r0
|
|
&paddd ($D4,$T0);
|
|
|
|
&movdqa (&QWP(16*0,"esp"),$D0);
|
|
&movdqa (&QWP(16*1,"esp"),$D1);
|
|
&movdqa (&QWP(16*2,"esp"),$D2);
|
|
&movdqa (&QWP(16*3,"esp"),$D3);
|
|
&movdqa (&QWP(16*4,"esp"),$D4);
|
|
|
|
################################################################
|
|
# d4 = h4*r0 + h3*r1 + h2*r2 + h1*r3 + h0*r4
|
|
# d3 = h3*r0 + h2*r1 + h1*r2 + h0*r3 + h4*5*r4
|
|
# d2 = h2*r0 + h1*r1 + h0*r2 + h4*5*r3 + h3*5*r4
|
|
# d1 = h1*r0 + h0*r1 + h4*5*r2 + h3*5*r3 + h2*5*r4
|
|
# d0 = h0*r0 + h4*5*r1 + h3*5*r2 + h2*5*r3 + h1*5*r4
|
|
|
|
&pmuludq ($D0,$T1); # h4*r0
|
|
&pmuludq ($D1,$T1); # h3*r0
|
|
&pmuludq ($D2,$T1); # h2*r0
|
|
&movd ($T0,&DWP(16*1+12,"edi")); # r1
|
|
&pmuludq ($D3,$T1); # h1*r0
|
|
&pmuludq ($D4,$T1); # h0*r0
|
|
|
|
&pmuladd (sub { my ($reg,$i)=@_;
|
|
&movd ($reg,&DWP(16*$i+12,"edi"));
|
|
});
|
|
|
|
&lazy_reduction ();
|
|
|
|
&sub ("ecx",16);
|
|
&jz (&label("done"));
|
|
|
|
&set_label("even");
|
|
&lea ("edx",&DWP(16*(5+5+5+9),"esp"));# size optimization
|
|
&lea ("eax",&DWP(-16*2,"esi"));
|
|
&sub ("ecx",64);
|
|
|
|
################################################################
|
|
# expand and copy pre-calculated table to stack
|
|
|
|
&movdqu ($T0,&QWP(16*0,"edi")); # r^1:r^2:r^3:r^4
|
|
&pshufd ($T1,$T0,0b01000100); # duplicate r^3:r^4
|
|
&cmovb ("esi","eax");
|
|
&pshufd ($T0,$T0,0b11101110); # duplicate r^1:r^2
|
|
&movdqa (&QWP(16*0,"edx"),$T1);
|
|
&lea ("eax",&DWP(16*10,"esp"));
|
|
&movdqu ($T1,&QWP(16*1,"edi"));
|
|
&movdqa (&QWP(16*(0-9),"edx"),$T0);
|
|
&pshufd ($T0,$T1,0b01000100);
|
|
&pshufd ($T1,$T1,0b11101110);
|
|
&movdqa (&QWP(16*1,"edx"),$T0);
|
|
&movdqu ($T0,&QWP(16*2,"edi"));
|
|
&movdqa (&QWP(16*(1-9),"edx"),$T1);
|
|
&pshufd ($T1,$T0,0b01000100);
|
|
&pshufd ($T0,$T0,0b11101110);
|
|
&movdqa (&QWP(16*2,"edx"),$T1);
|
|
&movdqu ($T1,&QWP(16*3,"edi"));
|
|
&movdqa (&QWP(16*(2-9),"edx"),$T0);
|
|
&pshufd ($T0,$T1,0b01000100);
|
|
&pshufd ($T1,$T1,0b11101110);
|
|
&movdqa (&QWP(16*3,"edx"),$T0);
|
|
&movdqu ($T0,&QWP(16*4,"edi"));
|
|
&movdqa (&QWP(16*(3-9),"edx"),$T1);
|
|
&pshufd ($T1,$T0,0b01000100);
|
|
&pshufd ($T0,$T0,0b11101110);
|
|
&movdqa (&QWP(16*4,"edx"),$T1);
|
|
&movdqu ($T1,&QWP(16*5,"edi"));
|
|
&movdqa (&QWP(16*(4-9),"edx"),$T0);
|
|
&pshufd ($T0,$T1,0b01000100);
|
|
&pshufd ($T1,$T1,0b11101110);
|
|
&movdqa (&QWP(16*5,"edx"),$T0);
|
|
&movdqu ($T0,&QWP(16*6,"edi"));
|
|
&movdqa (&QWP(16*(5-9),"edx"),$T1);
|
|
&pshufd ($T1,$T0,0b01000100);
|
|
&pshufd ($T0,$T0,0b11101110);
|
|
&movdqa (&QWP(16*6,"edx"),$T1);
|
|
&movdqu ($T1,&QWP(16*7,"edi"));
|
|
&movdqa (&QWP(16*(6-9),"edx"),$T0);
|
|
&pshufd ($T0,$T1,0b01000100);
|
|
&pshufd ($T1,$T1,0b11101110);
|
|
&movdqa (&QWP(16*7,"edx"),$T0);
|
|
&movdqu ($T0,&QWP(16*8,"edi"));
|
|
&movdqa (&QWP(16*(7-9),"edx"),$T1);
|
|
&pshufd ($T1,$T0,0b01000100);
|
|
&pshufd ($T0,$T0,0b11101110);
|
|
&movdqa (&QWP(16*8,"edx"),$T1);
|
|
&movdqa (&QWP(16*(8-9),"edx"),$T0);
|
|
|
|
sub load_input {
|
|
my ($inpbase,$offbase)=@_;
|
|
|
|
&movdqu ($T0,&QWP($inpbase+0,"esi")); # load input
|
|
&movdqu ($T1,&QWP($inpbase+16,"esi"));
|
|
&lea ("esi",&DWP(16*2,"esi"));
|
|
|
|
&movdqa (&QWP($offbase+16*2,"esp"),$D2);
|
|
&movdqa (&QWP($offbase+16*3,"esp"),$D3);
|
|
&movdqa (&QWP($offbase+16*4,"esp"),$D4);
|
|
|
|
&movdqa ($D2,$T0); # splat input
|
|
&movdqa ($D3,$T1);
|
|
&psrldq ($D2,6);
|
|
&psrldq ($D3,6);
|
|
&movdqa ($D4,$T0);
|
|
&punpcklqdq ($D2,$D3); # 2:3
|
|
&punpckhqdq ($D4,$T1); # 4
|
|
&punpcklqdq ($T0,$T1); # 0:1
|
|
|
|
&movdqa ($D3,$D2);
|
|
&psrlq ($D2,4);
|
|
&psrlq ($D3,30);
|
|
&movdqa ($T1,$T0);
|
|
&psrlq ($D4,40); # 4
|
|
&psrlq ($T1,26);
|
|
&pand ($T0,$MASK); # 0
|
|
&pand ($T1,$MASK); # 1
|
|
&pand ($D2,$MASK); # 2
|
|
&pand ($D3,$MASK); # 3
|
|
&por ($D4,&QWP(0,"ebx")); # padbit, yes, always
|
|
|
|
&movdqa (&QWP($offbase+16*0,"esp"),$D0) if ($offbase);
|
|
&movdqa (&QWP($offbase+16*1,"esp"),$D1) if ($offbase);
|
|
}
|
|
&load_input (16*2,16*5);
|
|
|
|
&jbe (&label("skip_loop"));
|
|
&jmp (&label("loop"));
|
|
|
|
&set_label("loop",32);
|
|
################################################################
|
|
# ((inp[0]*r^4+inp[2]*r^2+inp[4])*r^4+inp[6]*r^2
|
|
# ((inp[1]*r^4+inp[3]*r^2+inp[5])*r^3+inp[7]*r
|
|
# \___________________/
|
|
# ((inp[0]*r^4+inp[2]*r^2+inp[4])*r^4+inp[6]*r^2+inp[8])*r^2
|
|
# ((inp[1]*r^4+inp[3]*r^2+inp[5])*r^4+inp[7]*r^2+inp[9])*r
|
|
# \___________________/ \____________________/
|
|
################################################################
|
|
|
|
&movdqa ($T2,&QWP(16*(0-9),"edx")); # r0^2
|
|
&movdqa (&QWP(16*1,"eax"),$T1);
|
|
&movdqa (&QWP(16*2,"eax"),$D2);
|
|
&movdqa (&QWP(16*3,"eax"),$D3);
|
|
&movdqa (&QWP(16*4,"eax"),$D4);
|
|
|
|
################################################################
|
|
# d4 = h4*r0 + h0*r4 + h1*r3 + h2*r2 + h3*r1
|
|
# d3 = h3*r0 + h0*r3 + h1*r2 + h2*r1 + h4*5*r4
|
|
# d2 = h2*r0 + h0*r2 + h1*r1 + h3*5*r4 + h4*5*r3
|
|
# d1 = h1*r0 + h0*r1 + h2*5*r4 + h3*5*r3 + h4*5*r2
|
|
# d0 = h0*r0 + h1*5*r4 + h2*5*r3 + h3*5*r2 + h4*5*r1
|
|
|
|
&movdqa ($D1,$T0);
|
|
&pmuludq ($T0,$T2); # h0*r0
|
|
&movdqa ($D0,$T1);
|
|
&pmuludq ($T1,$T2); # h1*r0
|
|
&pmuludq ($D2,$T2); # h2*r0
|
|
&pmuludq ($D3,$T2); # h3*r0
|
|
&pmuludq ($D4,$T2); # h4*r0
|
|
|
|
sub pmuladd_alt {
|
|
my $addr = shift;
|
|
|
|
&pmuludq ($D0,&$addr(8)); # h1*s4
|
|
&movdqa ($T2,$D1);
|
|
&pmuludq ($D1,&$addr(1)); # h0*r1
|
|
&paddq ($D0,$T0);
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&$addr(2)); # h0*r2
|
|
&paddq ($D1,$T1);
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&$addr(3)); # h0*r3
|
|
&paddq ($D2,$T2);
|
|
&movdqa ($T2,&QWP(16*1,"eax")); # pull h1
|
|
&pmuludq ($T1,&$addr(4)); # h0*r4
|
|
&paddq ($D3,$T0);
|
|
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&$addr(1)); # h1*r1
|
|
&paddq ($D4,$T1);
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&$addr(2)); # h1*r2
|
|
&paddq ($D2,$T2);
|
|
&movdqa ($T2,&QWP(16*2,"eax")); # pull h2
|
|
&pmuludq ($T1,&$addr(3)); # h1*r3
|
|
&paddq ($D3,$T0);
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&$addr(7)); # h2*s3
|
|
&paddq ($D4,$T1);
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&$addr(8)); # h2*s4
|
|
&paddq ($D0,$T2);
|
|
|
|
&movdqa ($T2,$T1);
|
|
&pmuludq ($T1,&$addr(1)); # h2*r1
|
|
&paddq ($D1,$T0);
|
|
&movdqa ($T0,&QWP(16*3,"eax")); # pull h3
|
|
&pmuludq ($T2,&$addr(2)); # h2*r2
|
|
&paddq ($D3,$T1);
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&$addr(6)); # h3*s2
|
|
&paddq ($D4,$T2);
|
|
&movdqa ($T2,$T1);
|
|
&pmuludq ($T1,&$addr(7)); # h3*s3
|
|
&paddq ($D0,$T0);
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&$addr(8)); # h3*s4
|
|
&paddq ($D1,$T1);
|
|
|
|
&movdqa ($T1,&QWP(16*4,"eax")); # pull h4
|
|
&pmuludq ($T0,&$addr(1)); # h3*r1
|
|
&paddq ($D2,$T2);
|
|
&movdqa ($T2,$T1);
|
|
&pmuludq ($T1,&$addr(8)); # h4*s4
|
|
&paddq ($D4,$T0);
|
|
&movdqa ($T0,$T2);
|
|
&pmuludq ($T2,&$addr(5)); # h4*s1
|
|
&paddq ($D3,$T1);
|
|
&movdqa ($T1,$T0);
|
|
&pmuludq ($T0,&$addr(6)); # h4*s2
|
|
&paddq ($D0,$T2);
|
|
&movdqa ($MASK,&QWP(64,"ebx"));
|
|
&pmuludq ($T1,&$addr(7)); # h4*s3
|
|
&paddq ($D1,$T0);
|
|
&paddq ($D2,$T1);
|
|
}
|
|
&pmuladd_alt (sub { my $i=shift; &QWP(16*($i-9),"edx"); });
|
|
|
|
&load_input (-16*2,0);
|
|
&lea ("eax",&DWP(-16*2,"esi"));
|
|
&sub ("ecx",64);
|
|
|
|
&paddd ($T0,&QWP(16*(5+0),"esp")); # add hash value
|
|
&paddd ($T1,&QWP(16*(5+1),"esp"));
|
|
&paddd ($D2,&QWP(16*(5+2),"esp"));
|
|
&paddd ($D3,&QWP(16*(5+3),"esp"));
|
|
&paddd ($D4,&QWP(16*(5+4),"esp"));
|
|
|
|
&cmovb ("esi","eax");
|
|
&lea ("eax",&DWP(16*10,"esp"));
|
|
|
|
&movdqa ($T2,&QWP(16*0,"edx")); # r0^4
|
|
&movdqa (&QWP(16*1,"esp"),$D1);
|
|
&movdqa (&QWP(16*1,"eax"),$T1);
|
|
&movdqa (&QWP(16*2,"eax"),$D2);
|
|
&movdqa (&QWP(16*3,"eax"),$D3);
|
|
&movdqa (&QWP(16*4,"eax"),$D4);
|
|
|
|
################################################################
|
|
# d4 += h4*r0 + h0*r4 + h1*r3 + h2*r2 + h3*r1
|
|
# d3 += h3*r0 + h0*r3 + h1*r2 + h2*r1 + h4*5*r4
|
|
# d2 += h2*r0 + h0*r2 + h1*r1 + h3*5*r4 + h4*5*r3
|
|
# d1 += h1*r0 + h0*r1 + h2*5*r4 + h3*5*r3 + h4*5*r2
|
|
# d0 += h0*r0 + h1*5*r4 + h2*5*r3 + h3*5*r2 + h4*5*r1
|
|
|
|
&movdqa ($D1,$T0);
|
|
&pmuludq ($T0,$T2); # h0*r0
|
|
&paddq ($T0,$D0);
|
|
&movdqa ($D0,$T1);
|
|
&pmuludq ($T1,$T2); # h1*r0
|
|
&pmuludq ($D2,$T2); # h2*r0
|
|
&pmuludq ($D3,$T2); # h3*r0
|
|
&pmuludq ($D4,$T2); # h4*r0
|
|
|
|
&paddq ($T1,&QWP(16*1,"esp"));
|
|
&paddq ($D2,&QWP(16*2,"esp"));
|
|
&paddq ($D3,&QWP(16*3,"esp"));
|
|
&paddq ($D4,&QWP(16*4,"esp"));
|
|
|
|
&pmuladd_alt (sub { my $i=shift; &QWP(16*$i,"edx"); });
|
|
|
|
&lazy_reduction ();
|
|
|
|
&load_input (16*2,16*5);
|
|
|
|
&ja (&label("loop"));
|
|
|
|
&set_label("skip_loop");
|
|
################################################################
|
|
# multiply (inp[0:1]+hash) or inp[2:3] by r^2:r^1
|
|
|
|
&pshufd ($T2,&QWP(16*(0-9),"edx"),0x10);# r0^n
|
|
&add ("ecx",32);
|
|
&jnz (&label("long_tail"));
|
|
|
|
&paddd ($T0,$D0); # add hash value
|
|
&paddd ($T1,$D1);
|
|
&paddd ($D2,&QWP(16*7,"esp"));
|
|
&paddd ($D3,&QWP(16*8,"esp"));
|
|
&paddd ($D4,&QWP(16*9,"esp"));
|
|
|
|
&set_label("long_tail");
|
|
|
|
&movdqa (&QWP(16*0,"eax"),$T0);
|
|
&movdqa (&QWP(16*1,"eax"),$T1);
|
|
&movdqa (&QWP(16*2,"eax"),$D2);
|
|
&movdqa (&QWP(16*3,"eax"),$D3);
|
|
&movdqa (&QWP(16*4,"eax"),$D4);
|
|
|
|
################################################################
|
|
# d4 = h4*r0 + h3*r1 + h2*r2 + h1*r3 + h0*r4
|
|
# d3 = h3*r0 + h2*r1 + h1*r2 + h0*r3 + h4*5*r4
|
|
# d2 = h2*r0 + h1*r1 + h0*r2 + h4*5*r3 + h3*5*r4
|
|
# d1 = h1*r0 + h0*r1 + h4*5*r2 + h3*5*r3 + h2*5*r4
|
|
# d0 = h0*r0 + h4*5*r1 + h3*5*r2 + h2*5*r3 + h1*5*r4
|
|
|
|
&pmuludq ($T0,$T2); # h0*r0
|
|
&pmuludq ($T1,$T2); # h1*r0
|
|
&pmuludq ($D2,$T2); # h2*r0
|
|
&movdqa ($D0,$T0);
|
|
&pshufd ($T0,&QWP(16*(1-9),"edx"),0x10);# r1^n
|
|
&pmuludq ($D3,$T2); # h3*r0
|
|
&movdqa ($D1,$T1);
|
|
&pmuludq ($D4,$T2); # h4*r0
|
|
|
|
&pmuladd (sub { my ($reg,$i)=@_;
|
|
&pshufd ($reg,&QWP(16*($i-9),"edx"),0x10);
|
|
},"eax");
|
|
|
|
&jz (&label("short_tail"));
|
|
|
|
&load_input (-16*2,0);
|
|
|
|
&pshufd ($T2,&QWP(16*0,"edx"),0x10); # r0^n
|
|
&paddd ($T0,&QWP(16*5,"esp")); # add hash value
|
|
&paddd ($T1,&QWP(16*6,"esp"));
|
|
&paddd ($D2,&QWP(16*7,"esp"));
|
|
&paddd ($D3,&QWP(16*8,"esp"));
|
|
&paddd ($D4,&QWP(16*9,"esp"));
|
|
|
|
################################################################
|
|
# multiply inp[0:1] by r^4:r^3 and accumulate
|
|
|
|
&movdqa (&QWP(16*0,"esp"),$T0);
|
|
&pmuludq ($T0,$T2); # h0*r0
|
|
&movdqa (&QWP(16*1,"esp"),$T1);
|
|
&pmuludq ($T1,$T2); # h1*r0
|
|
&paddq ($D0,$T0);
|
|
&movdqa ($T0,$D2);
|
|
&pmuludq ($D2,$T2); # h2*r0
|
|
&paddq ($D1,$T1);
|
|
&movdqa ($T1,$D3);
|
|
&pmuludq ($D3,$T2); # h3*r0
|
|
&paddq ($D2,&QWP(16*2,"esp"));
|
|
&movdqa (&QWP(16*2,"esp"),$T0);
|
|
&pshufd ($T0,&QWP(16*1,"edx"),0x10); # r1^n
|
|
&paddq ($D3,&QWP(16*3,"esp"));
|
|
&movdqa (&QWP(16*3,"esp"),$T1);
|
|
&movdqa ($T1,$D4);
|
|
&pmuludq ($D4,$T2); # h4*r0
|
|
&paddq ($D4,&QWP(16*4,"esp"));
|
|
&movdqa (&QWP(16*4,"esp"),$T1);
|
|
|
|
&pmuladd (sub { my ($reg,$i)=@_;
|
|
&pshufd ($reg,&QWP(16*$i,"edx"),0x10);
|
|
});
|
|
|
|
&set_label("short_tail");
|
|
|
|
################################################################
|
|
# horizontal addition
|
|
|
|
&pshufd ($T1,$D4,0b01001110);
|
|
&pshufd ($T0,$D3,0b01001110);
|
|
&paddq ($D4,$T1);
|
|
&paddq ($D3,$T0);
|
|
&pshufd ($T1,$D0,0b01001110);
|
|
&pshufd ($T0,$D1,0b01001110);
|
|
&paddq ($D0,$T1);
|
|
&paddq ($D1,$T0);
|
|
&pshufd ($T1,$D2,0b01001110);
|
|
#&paddq ($D2,$T1);
|
|
|
|
&lazy_reduction (sub { &paddq ($D2,$T1) });
|
|
|
|
&set_label("done");
|
|
&movd (&DWP(-16*3+4*0,"edi"),$D0); # store hash value
|
|
&movd (&DWP(-16*3+4*1,"edi"),$D1);
|
|
&movd (&DWP(-16*3+4*2,"edi"),$D2);
|
|
&movd (&DWP(-16*3+4*3,"edi"),$D3);
|
|
&movd (&DWP(-16*3+4*4,"edi"),$D4);
|
|
&mov ("esp","ebp");
|
|
&set_label("nodata");
|
|
&function_end("_poly1305_blocks_sse2");
|
|
|
|
&align (32);
|
|
&function_begin("_poly1305_emit_sse2");
|
|
&mov ("ebp",&wparam(0)); # context
|
|
|
|
&cmp (&DWP(4*5,"ebp"),0); # is_base2_26?
|
|
&je (&label("enter_emit"));
|
|
|
|
&mov ("eax",&DWP(4*0,"ebp")); # load hash value
|
|
&mov ("edi",&DWP(4*1,"ebp"));
|
|
&mov ("ecx",&DWP(4*2,"ebp"));
|
|
&mov ("edx",&DWP(4*3,"ebp"));
|
|
&mov ("esi",&DWP(4*4,"ebp"));
|
|
|
|
&mov ("ebx","edi"); # base 2^26 -> base 2^32
|
|
&shl ("edi",26);
|
|
&shr ("ebx",6);
|
|
&add ("eax","edi");
|
|
&mov ("edi","ecx");
|
|
&adc ("ebx",0);
|
|
|
|
&shl ("edi",20);
|
|
&shr ("ecx",12);
|
|
&add ("ebx","edi");
|
|
&mov ("edi","edx");
|
|
&adc ("ecx",0);
|
|
|
|
&shl ("edi",14);
|
|
&shr ("edx",18);
|
|
&add ("ecx","edi");
|
|
&mov ("edi","esi");
|
|
&adc ("edx",0);
|
|
|
|
&shl ("edi",8);
|
|
&shr ("esi",24);
|
|
&add ("edx","edi");
|
|
&adc ("esi",0); # can be partially reduced
|
|
|
|
&mov ("edi","esi"); # final reduction
|
|
&and ("esi",3);
|
|
&shr ("edi",2);
|
|
&lea ("ebp",&DWP(0,"edi","edi",4)); # *5
|
|
&mov ("edi",&wparam(1)); # output
|
|
&add ("eax","ebp");
|
|
&mov ("ebp",&wparam(2)); # key
|
|
&adc ("ebx",0);
|
|
&adc ("ecx",0);
|
|
&adc ("edx",0);
|
|
&adc ("esi",0);
|
|
|
|
&movd ($D0,"eax"); # offload original hash value
|
|
&add ("eax",5); # compare to modulus
|
|
&movd ($D1,"ebx");
|
|
&adc ("ebx",0);
|
|
&movd ($D2,"ecx");
|
|
&adc ("ecx",0);
|
|
&movd ($D3,"edx");
|
|
&adc ("edx",0);
|
|
&adc ("esi",0);
|
|
&shr ("esi",2); # did it carry/borrow?
|
|
|
|
&neg ("esi"); # do we choose (hash-modulus) ...
|
|
&and ("eax","esi");
|
|
&and ("ebx","esi");
|
|
&and ("ecx","esi");
|
|
&and ("edx","esi");
|
|
&mov (&DWP(4*0,"edi"),"eax");
|
|
&movd ("eax",$D0);
|
|
&mov (&DWP(4*1,"edi"),"ebx");
|
|
&movd ("ebx",$D1);
|
|
&mov (&DWP(4*2,"edi"),"ecx");
|
|
&movd ("ecx",$D2);
|
|
&mov (&DWP(4*3,"edi"),"edx");
|
|
&movd ("edx",$D3);
|
|
|
|
¬ ("esi"); # ... or original hash value?
|
|
&and ("eax","esi");
|
|
&and ("ebx","esi");
|
|
&or ("eax",&DWP(4*0,"edi"));
|
|
&and ("ecx","esi");
|
|
&or ("ebx",&DWP(4*1,"edi"));
|
|
&and ("edx","esi");
|
|
&or ("ecx",&DWP(4*2,"edi"));
|
|
&or ("edx",&DWP(4*3,"edi"));
|
|
|
|
&add ("eax",&DWP(4*0,"ebp")); # accumulate key
|
|
&adc ("ebx",&DWP(4*1,"ebp"));
|
|
&mov (&DWP(4*0,"edi"),"eax");
|
|
&adc ("ecx",&DWP(4*2,"ebp"));
|
|
&mov (&DWP(4*1,"edi"),"ebx");
|
|
&adc ("edx",&DWP(4*3,"ebp"));
|
|
&mov (&DWP(4*2,"edi"),"ecx");
|
|
&mov (&DWP(4*3,"edi"),"edx");
|
|
&function_end("_poly1305_emit_sse2");
|
|
|
|
if ($avx>1) {
|
|
########################################################################
|
|
# Note that poly1305_init_avx2 operates on %xmm, I could have used
|
|
# poly1305_init_sse2...
|
|
|
|
&align (32);
|
|
&function_begin_B("_poly1305_init_avx2");
|
|
&vmovdqu ($D4,&QWP(4*6,"edi")); # key base 2^32
|
|
&lea ("edi",&DWP(16*3,"edi")); # size optimization
|
|
&mov ("ebp","esp");
|
|
&sub ("esp",16*(9+5));
|
|
&and ("esp",-16);
|
|
|
|
#&vpand ($D4,$D4,&QWP(96,"ebx")); # magic mask
|
|
&vmovdqa ($MASK,&QWP(64,"ebx"));
|
|
|
|
&vpand ($D0,$D4,$MASK); # -> base 2^26
|
|
&vpsrlq ($D1,$D4,26);
|
|
&vpsrldq ($D3,$D4,6);
|
|
&vpand ($D1,$D1,$MASK);
|
|
&vpsrlq ($D2,$D3,4)
|
|
&vpsrlq ($D3,$D3,30);
|
|
&vpand ($D2,$D2,$MASK);
|
|
&vpand ($D3,$D3,$MASK);
|
|
&vpsrldq ($D4,$D4,13);
|
|
|
|
&lea ("edx",&DWP(16*9,"esp")); # size optimization
|
|
&mov ("ecx",2);
|
|
&set_label("square");
|
|
&vmovdqa (&QWP(16*0,"esp"),$D0);
|
|
&vmovdqa (&QWP(16*1,"esp"),$D1);
|
|
&vmovdqa (&QWP(16*2,"esp"),$D2);
|
|
&vmovdqa (&QWP(16*3,"esp"),$D3);
|
|
&vmovdqa (&QWP(16*4,"esp"),$D4);
|
|
|
|
&vpslld ($T1,$D1,2);
|
|
&vpslld ($T0,$D2,2);
|
|
&vpaddd ($T1,$T1,$D1); # *5
|
|
&vpaddd ($T0,$T0,$D2); # *5
|
|
&vmovdqa (&QWP(16*5,"esp"),$T1);
|
|
&vmovdqa (&QWP(16*6,"esp"),$T0);
|
|
&vpslld ($T1,$D3,2);
|
|
&vpslld ($T0,$D4,2);
|
|
&vpaddd ($T1,$T1,$D3); # *5
|
|
&vpaddd ($T0,$T0,$D4); # *5
|
|
&vmovdqa (&QWP(16*7,"esp"),$T1);
|
|
&vmovdqa (&QWP(16*8,"esp"),$T0);
|
|
|
|
&vpshufd ($T0,$D0,0b01000100);
|
|
&vmovdqa ($T1,$D1);
|
|
&vpshufd ($D1,$D1,0b01000100);
|
|
&vpshufd ($D2,$D2,0b01000100);
|
|
&vpshufd ($D3,$D3,0b01000100);
|
|
&vpshufd ($D4,$D4,0b01000100);
|
|
&vmovdqa (&QWP(16*0,"edx"),$T0);
|
|
&vmovdqa (&QWP(16*1,"edx"),$D1);
|
|
&vmovdqa (&QWP(16*2,"edx"),$D2);
|
|
&vmovdqa (&QWP(16*3,"edx"),$D3);
|
|
&vmovdqa (&QWP(16*4,"edx"),$D4);
|
|
|
|
################################################################
|
|
# d4 = h4*r0 + h3*r1 + h2*r2 + h1*r3 + h0*r4
|
|
# d3 = h3*r0 + h2*r1 + h1*r2 + h0*r3 + h4*5*r4
|
|
# d2 = h2*r0 + h1*r1 + h0*r2 + h4*5*r3 + h3*5*r4
|
|
# d1 = h1*r0 + h0*r1 + h4*5*r2 + h3*5*r3 + h2*5*r4
|
|
# d0 = h0*r0 + h4*5*r1 + h3*5*r2 + h2*5*r3 + h1*5*r4
|
|
|
|
&vpmuludq ($D4,$D4,$D0); # h4*r0
|
|
&vpmuludq ($D3,$D3,$D0); # h3*r0
|
|
&vpmuludq ($D2,$D2,$D0); # h2*r0
|
|
&vpmuludq ($D1,$D1,$D0); # h1*r0
|
|
&vpmuludq ($D0,$T0,$D0); # h0*r0
|
|
|
|
&vpmuludq ($T0,$T1,&QWP(16*3,"edx")); # r1*h3
|
|
&vpaddq ($D4,$D4,$T0);
|
|
&vpmuludq ($T2,$T1,&QWP(16*2,"edx")); # r1*h2
|
|
&vpaddq ($D3,$D3,$T2);
|
|
&vpmuludq ($T0,$T1,&QWP(16*1,"edx")); # r1*h1
|
|
&vpaddq ($D2,$D2,$T0);
|
|
&vmovdqa ($T2,&QWP(16*5,"esp")); # s1
|
|
&vpmuludq ($T1,$T1,&QWP(16*0,"edx")); # r1*h0
|
|
&vpaddq ($D1,$D1,$T1);
|
|
&vmovdqa ($T0,&QWP(16*2,"esp")); # r2
|
|
&vpmuludq ($T2,$T2,&QWP(16*4,"edx")); # s1*h4
|
|
&vpaddq ($D0,$D0,$T2);
|
|
|
|
&vpmuludq ($T1,$T0,&QWP(16*2,"edx")); # r2*h2
|
|
&vpaddq ($D4,$D4,$T1);
|
|
&vpmuludq ($T2,$T0,&QWP(16*1,"edx")); # r2*h1
|
|
&vpaddq ($D3,$D3,$T2);
|
|
&vmovdqa ($T1,&QWP(16*6,"esp")); # s2
|
|
&vpmuludq ($T0,$T0,&QWP(16*0,"edx")); # r2*h0
|
|
&vpaddq ($D2,$D2,$T0);
|
|
&vpmuludq ($T2,$T1,&QWP(16*4,"edx")); # s2*h4
|
|
&vpaddq ($D1,$D1,$T2);
|
|
&vmovdqa ($T0,&QWP(16*3,"esp")); # r3
|
|
&vpmuludq ($T1,$T1,&QWP(16*3,"edx")); # s2*h3
|
|
&vpaddq ($D0,$D0,$T1);
|
|
|
|
&vpmuludq ($T2,$T0,&QWP(16*1,"edx")); # r3*h1
|
|
&vpaddq ($D4,$D4,$T2);
|
|
&vmovdqa ($T1,&QWP(16*7,"esp")); # s3
|
|
&vpmuludq ($T0,$T0,&QWP(16*0,"edx")); # r3*h0
|
|
&vpaddq ($D3,$D3,$T0);
|
|
&vpmuludq ($T2,$T1,&QWP(16*4,"edx")); # s3*h4
|
|
&vpaddq ($D2,$D2,$T2);
|
|
&vpmuludq ($T0,$T1,&QWP(16*3,"edx")); # s3*h3
|
|
&vpaddq ($D1,$D1,$T0);
|
|
&vmovdqa ($T2,&QWP(16*4,"esp")); # r4
|
|
&vpmuludq ($T1,$T1,&QWP(16*2,"edx")); # s3*h2
|
|
&vpaddq ($D0,$D0,$T1);
|
|
|
|
&vmovdqa ($T0,&QWP(16*8,"esp")); # s4
|
|
&vpmuludq ($T2,$T2,&QWP(16*0,"edx")); # r4*h0
|
|
&vpaddq ($D4,$D4,$T2);
|
|
&vpmuludq ($T1,$T0,&QWP(16*4,"edx")); # s4*h4
|
|
&vpaddq ($D3,$D3,$T1);
|
|
&vpmuludq ($T2,$T0,&QWP(16*1,"edx")); # s4*h1
|
|
&vpaddq ($D0,$D0,$T2);
|
|
&vpmuludq ($T1,$T0,&QWP(16*2,"edx")); # s4*h2
|
|
&vpaddq ($D1,$D1,$T1);
|
|
&vmovdqa ($MASK,&QWP(64,"ebx"));
|
|
&vpmuludq ($T0,$T0,&QWP(16*3,"edx")); # s4*h3
|
|
&vpaddq ($D2,$D2,$T0);
|
|
|
|
################################################################
|
|
# lazy reduction
|
|
&vpsrlq ($T0,$D3,26);
|
|
&vpand ($D3,$D3,$MASK);
|
|
&vpsrlq ($T1,$D0,26);
|
|
&vpand ($D0,$D0,$MASK);
|
|
&vpaddq ($D4,$D4,$T0); # h3 -> h4
|
|
&vpaddq ($D1,$D1,$T1); # h0 -> h1
|
|
&vpsrlq ($T0,$D4,26);
|
|
&vpand ($D4,$D4,$MASK);
|
|
&vpsrlq ($T1,$D1,26);
|
|
&vpand ($D1,$D1,$MASK);
|
|
&vpaddq ($D2,$D2,$T1); # h1 -> h2
|
|
&vpaddd ($D0,$D0,$T0);
|
|
&vpsllq ($T0,$T0,2);
|
|
&vpsrlq ($T1,$D2,26);
|
|
&vpand ($D2,$D2,$MASK);
|
|
&vpaddd ($D0,$D0,$T0); # h4 -> h0
|
|
&vpaddd ($D3,$D3,$T1); # h2 -> h3
|
|
&vpsrlq ($T1,$D3,26);
|
|
&vpsrlq ($T0,$D0,26);
|
|
&vpand ($D0,$D0,$MASK);
|
|
&vpand ($D3,$D3,$MASK);
|
|
&vpaddd ($D1,$D1,$T0); # h0 -> h1
|
|
&vpaddd ($D4,$D4,$T1); # h3 -> h4
|
|
|
|
&dec ("ecx");
|
|
&jz (&label("square_break"));
|
|
|
|
&vpunpcklqdq ($D0,$D0,&QWP(16*0,"esp")); # 0:r^1:0:r^2
|
|
&vpunpcklqdq ($D1,$D1,&QWP(16*1,"esp"));
|
|
&vpunpcklqdq ($D2,$D2,&QWP(16*2,"esp"));
|
|
&vpunpcklqdq ($D3,$D3,&QWP(16*3,"esp"));
|
|
&vpunpcklqdq ($D4,$D4,&QWP(16*4,"esp"));
|
|
&jmp (&label("square"));
|
|
|
|
&set_label("square_break");
|
|
&vpsllq ($D0,$D0,32); # -> r^3:0:r^4:0
|
|
&vpsllq ($D1,$D1,32);
|
|
&vpsllq ($D2,$D2,32);
|
|
&vpsllq ($D3,$D3,32);
|
|
&vpsllq ($D4,$D4,32);
|
|
&vpor ($D0,$D0,&QWP(16*0,"esp")); # r^3:r^1:r^4:r^2
|
|
&vpor ($D1,$D1,&QWP(16*1,"esp"));
|
|
&vpor ($D2,$D2,&QWP(16*2,"esp"));
|
|
&vpor ($D3,$D3,&QWP(16*3,"esp"));
|
|
&vpor ($D4,$D4,&QWP(16*4,"esp"));
|
|
|
|
&vpshufd ($D0,$D0,0b10001101); # -> r^1:r^2:r^3:r^4
|
|
&vpshufd ($D1,$D1,0b10001101);
|
|
&vpshufd ($D2,$D2,0b10001101);
|
|
&vpshufd ($D3,$D3,0b10001101);
|
|
&vpshufd ($D4,$D4,0b10001101);
|
|
|
|
&vmovdqu (&QWP(16*0,"edi"),$D0); # save the table
|
|
&vmovdqu (&QWP(16*1,"edi"),$D1);
|
|
&vmovdqu (&QWP(16*2,"edi"),$D2);
|
|
&vmovdqu (&QWP(16*3,"edi"),$D3);
|
|
&vmovdqu (&QWP(16*4,"edi"),$D4);
|
|
|
|
&vpslld ($T1,$D1,2);
|
|
&vpslld ($T0,$D2,2);
|
|
&vpaddd ($T1,$T1,$D1); # *5
|
|
&vpaddd ($T0,$T0,$D2); # *5
|
|
&vmovdqu (&QWP(16*5,"edi"),$T1);
|
|
&vmovdqu (&QWP(16*6,"edi"),$T0);
|
|
&vpslld ($T1,$D3,2);
|
|
&vpslld ($T0,$D4,2);
|
|
&vpaddd ($T1,$T1,$D3); # *5
|
|
&vpaddd ($T0,$T0,$D4); # *5
|
|
&vmovdqu (&QWP(16*7,"edi"),$T1);
|
|
&vmovdqu (&QWP(16*8,"edi"),$T0);
|
|
|
|
&mov ("esp","ebp");
|
|
&lea ("edi",&DWP(-16*3,"edi")); # size de-optimization
|
|
&ret ();
|
|
&function_end_B("_poly1305_init_avx2");
|
|
|
|
########################################################################
|
|
# now it's time to switch to %ymm
|
|
|
|
my ($D0,$D1,$D2,$D3,$D4,$T0,$T1,$T2)=map("ymm$_",(0..7));
|
|
my $MASK=$T2;
|
|
|
|
sub X { my $reg=shift; $reg=~s/^ymm/xmm/; $reg; }
|
|
|
|
&align (32);
|
|
&function_begin("_poly1305_blocks_avx2");
|
|
&mov ("edi",&wparam(0)); # ctx
|
|
&mov ("esi",&wparam(1)); # inp
|
|
&mov ("ecx",&wparam(2)); # len
|
|
|
|
&mov ("eax",&DWP(4*5,"edi")); # is_base2_26
|
|
&and ("ecx",-16);
|
|
&jz (&label("nodata"));
|
|
&cmp ("ecx",64);
|
|
&jae (&label("enter_avx2"));
|
|
&test ("eax","eax"); # is_base2_26?
|
|
&jz (&label("enter_blocks"));
|
|
|
|
&set_label("enter_avx2");
|
|
&vzeroupper ();
|
|
|
|
&call (&label("pic_point"));
|
|
&set_label("pic_point");
|
|
&blindpop("ebx");
|
|
&lea ("ebx",&DWP(&label("const_sse2")."-".&label("pic_point"),"ebx"));
|
|
|
|
&test ("eax","eax"); # is_base2_26?
|
|
&jnz (&label("base2_26"));
|
|
|
|
&call ("_poly1305_init_avx2");
|
|
|
|
################################################# base 2^32 -> base 2^26
|
|
&mov ("eax",&DWP(0,"edi"));
|
|
&mov ("ecx",&DWP(3,"edi"));
|
|
&mov ("edx",&DWP(6,"edi"));
|
|
&mov ("esi",&DWP(9,"edi"));
|
|
&mov ("ebp",&DWP(13,"edi"));
|
|
|
|
&shr ("ecx",2);
|
|
&and ("eax",0x3ffffff);
|
|
&shr ("edx",4);
|
|
&and ("ecx",0x3ffffff);
|
|
&shr ("esi",6);
|
|
&and ("edx",0x3ffffff);
|
|
|
|
&mov (&DWP(4*0,"edi"),"eax");
|
|
&mov (&DWP(4*1,"edi"),"ecx");
|
|
&mov (&DWP(4*2,"edi"),"edx");
|
|
&mov (&DWP(4*3,"edi"),"esi");
|
|
&mov (&DWP(4*4,"edi"),"ebp");
|
|
&mov (&DWP(4*5,"edi"),1); # is_base2_26
|
|
|
|
&mov ("esi",&wparam(1)); # [reload] inp
|
|
&mov ("ecx",&wparam(2)); # [reload] len
|
|
|
|
&set_label("base2_26");
|
|
&mov ("eax",&wparam(3)); # padbit
|
|
&mov ("ebp","esp");
|
|
|
|
&sub ("esp",32*(5+9));
|
|
&and ("esp",-512); # ensure that frame
|
|
# doesn't cross page
|
|
# boundary, which is
|
|
# essential for
|
|
# misaligned 32-byte
|
|
# loads
|
|
|
|
################################################################
|
|
# expand and copy pre-calculated table to stack
|
|
|
|
&vmovdqu (&X($D0),&QWP(16*(3+0),"edi"));
|
|
&lea ("edx",&DWP(32*5+128,"esp")); # +128 size optimization
|
|
&vmovdqu (&X($D1),&QWP(16*(3+1),"edi"));
|
|
&vmovdqu (&X($D2),&QWP(16*(3+2),"edi"));
|
|
&vmovdqu (&X($D3),&QWP(16*(3+3),"edi"));
|
|
&vmovdqu (&X($D4),&QWP(16*(3+4),"edi"));
|
|
&lea ("edi",&DWP(16*3,"edi")); # size optimization
|
|
&vpermq ($D0,$D0,0b01000000); # 00001234 -> 12343434
|
|
&vpermq ($D1,$D1,0b01000000);
|
|
&vpermq ($D2,$D2,0b01000000);
|
|
&vpermq ($D3,$D3,0b01000000);
|
|
&vpermq ($D4,$D4,0b01000000);
|
|
&vpshufd ($D0,$D0,0b11001000); # 12343434 -> 14243444
|
|
&vpshufd ($D1,$D1,0b11001000);
|
|
&vpshufd ($D2,$D2,0b11001000);
|
|
&vpshufd ($D3,$D3,0b11001000);
|
|
&vpshufd ($D4,$D4,0b11001000);
|
|
&vmovdqa (&QWP(32*0-128,"edx"),$D0);
|
|
&vmovdqu (&X($D0),&QWP(16*5,"edi"));
|
|
&vmovdqa (&QWP(32*1-128,"edx"),$D1);
|
|
&vmovdqu (&X($D1),&QWP(16*6,"edi"));
|
|
&vmovdqa (&QWP(32*2-128,"edx"),$D2);
|
|
&vmovdqu (&X($D2),&QWP(16*7,"edi"));
|
|
&vmovdqa (&QWP(32*3-128,"edx"),$D3);
|
|
&vmovdqu (&X($D3),&QWP(16*8,"edi"));
|
|
&vmovdqa (&QWP(32*4-128,"edx"),$D4);
|
|
&vpermq ($D0,$D0,0b01000000);
|
|
&vpermq ($D1,$D1,0b01000000);
|
|
&vpermq ($D2,$D2,0b01000000);
|
|
&vpermq ($D3,$D3,0b01000000);
|
|
&vpshufd ($D0,$D0,0b11001000);
|
|
&vpshufd ($D1,$D1,0b11001000);
|
|
&vpshufd ($D2,$D2,0b11001000);
|
|
&vpshufd ($D3,$D3,0b11001000);
|
|
&vmovdqa (&QWP(32*5-128,"edx"),$D0);
|
|
&vmovd (&X($D0),&DWP(-16*3+4*0,"edi"));# load hash value
|
|
&vmovdqa (&QWP(32*6-128,"edx"),$D1);
|
|
&vmovd (&X($D1),&DWP(-16*3+4*1,"edi"));
|
|
&vmovdqa (&QWP(32*7-128,"edx"),$D2);
|
|
&vmovd (&X($D2),&DWP(-16*3+4*2,"edi"));
|
|
&vmovdqa (&QWP(32*8-128,"edx"),$D3);
|
|
&vmovd (&X($D3),&DWP(-16*3+4*3,"edi"));
|
|
&vmovd (&X($D4),&DWP(-16*3+4*4,"edi"));
|
|
&vmovdqa ($MASK,&QWP(64,"ebx"));
|
|
&neg ("eax"); # padbit
|
|
|
|
&test ("ecx",63);
|
|
&jz (&label("even"));
|
|
|
|
&mov ("edx","ecx");
|
|
&and ("ecx",-64);
|
|
&and ("edx",63);
|
|
|
|
&vmovdqu (&X($T0),&QWP(16*0,"esi"));
|
|
&cmp ("edx",32);
|
|
&jb (&label("one"));
|
|
|
|
&vmovdqu (&X($T1),&QWP(16*1,"esi"));
|
|
&je (&label("two"));
|
|
|
|
&vinserti128 ($T0,$T0,&QWP(16*2,"esi"),1);
|
|
&lea ("esi",&DWP(16*3,"esi"));
|
|
&lea ("ebx",&DWP(8,"ebx")); # three padbits
|
|
&lea ("edx",&DWP(32*5+128+8,"esp")); # --:r^1:r^2:r^3 (*)
|
|
&jmp (&label("tail"));
|
|
|
|
&set_label("two");
|
|
&lea ("esi",&DWP(16*2,"esi"));
|
|
&lea ("ebx",&DWP(16,"ebx")); # two padbits
|
|
&lea ("edx",&DWP(32*5+128+16,"esp"));# --:--:r^1:r^2 (*)
|
|
&jmp (&label("tail"));
|
|
|
|
&set_label("one");
|
|
&lea ("esi",&DWP(16*1,"esi"));
|
|
&vpxor ($T1,$T1,$T1);
|
|
&lea ("ebx",&DWP(32,"ebx","eax",8)); # one or no padbits
|
|
&lea ("edx",&DWP(32*5+128+24,"esp"));# --:--:--:r^1 (*)
|
|
&jmp (&label("tail"));
|
|
|
|
# (*) spots marked with '--' are data from next table entry, but they
|
|
# are multiplied by 0 and therefore rendered insignificant
|
|
|
|
&set_label("even",32);
|
|
&vmovdqu (&X($T0),&QWP(16*0,"esi")); # load input
|
|
&vmovdqu (&X($T1),&QWP(16*1,"esi"));
|
|
&vinserti128 ($T0,$T0,&QWP(16*2,"esi"),1);
|
|
&vinserti128 ($T1,$T1,&QWP(16*3,"esi"),1);
|
|
&lea ("esi",&DWP(16*4,"esi"));
|
|
&sub ("ecx",64);
|
|
&jz (&label("tail"));
|
|
|
|
&set_label("loop");
|
|
################################################################
|
|
# ((inp[0]*r^4+r[4])*r^4+r[8])*r^4
|
|
# ((inp[1]*r^4+r[5])*r^4+r[9])*r^3
|
|
# ((inp[2]*r^4+r[6])*r^4+r[10])*r^2
|
|
# ((inp[3]*r^4+r[7])*r^4+r[11])*r^1
|
|
# \________/ \_______/
|
|
################################################################
|
|
|
|
sub vsplat_input {
|
|
&vmovdqa (&QWP(32*2,"esp"),$D2);
|
|
&vpsrldq ($D2,$T0,6); # splat input
|
|
&vmovdqa (&QWP(32*0,"esp"),$D0);
|
|
&vpsrldq ($D0,$T1,6);
|
|
&vmovdqa (&QWP(32*1,"esp"),$D1);
|
|
&vpunpckhqdq ($D1,$T0,$T1); # 4
|
|
&vpunpcklqdq ($T0,$T0,$T1); # 0:1
|
|
&vpunpcklqdq ($D2,$D2,$D0); # 2:3
|
|
|
|
&vpsrlq ($D0,$D2,30);
|
|
&vpsrlq ($D2,$D2,4);
|
|
&vpsrlq ($T1,$T0,26);
|
|
&vpsrlq ($D1,$D1,40); # 4
|
|
&vpand ($D2,$D2,$MASK); # 2
|
|
&vpand ($T0,$T0,$MASK); # 0
|
|
&vpand ($T1,$T1,$MASK); # 1
|
|
&vpand ($D0,$D0,$MASK); # 3 (*)
|
|
&vpor ($D1,$D1,&QWP(0,"ebx")); # padbit, yes, always
|
|
|
|
# (*) note that output is counterintuitive, inp[3:4] is
|
|
# returned in $D1-2, while $D3-4 are preserved;
|
|
}
|
|
&vsplat_input ();
|
|
|
|
sub vpmuladd {
|
|
my $addr = shift;
|
|
|
|
&vpaddq ($D2,$D2,&QWP(32*2,"esp")); # add hash value
|
|
&vpaddq ($T0,$T0,&QWP(32*0,"esp"));
|
|
&vpaddq ($T1,$T1,&QWP(32*1,"esp"));
|
|
&vpaddq ($D0,$D0,$D3);
|
|
&vpaddq ($D1,$D1,$D4);
|
|
|
|
################################################################
|
|
# d3 = h2*r1 + h0*r3 + h1*r2 + h3*r0 + h4*5*r4
|
|
# d4 = h2*r2 + h0*r4 + h1*r3 + h3*r1 + h4*r0
|
|
# d0 = h2*5*r3 + h0*r0 + h1*5*r4 + h3*5*r2 + h4*5*r1
|
|
# d1 = h2*5*r4 + h0*r1 + h1*r0 + h3*5*r3 + h4*5*r2
|
|
# d2 = h2*r0 + h0*r2 + h1*r1 + h3*5*r4 + h4*5*r3
|
|
|
|
&vpmuludq ($D3,$D2,&$addr(1)); # d3 = h2*r1
|
|
&vmovdqa (QWP(32*1,"esp"),$T1);
|
|
&vpmuludq ($D4,$D2,&$addr(2)); # d4 = h2*r2
|
|
&vmovdqa (QWP(32*3,"esp"),$D0);
|
|
&vpmuludq ($D0,$D2,&$addr(7)); # d0 = h2*s3
|
|
&vmovdqa (QWP(32*4,"esp"),$D1);
|
|
&vpmuludq ($D1,$D2,&$addr(8)); # d1 = h2*s4
|
|
&vpmuludq ($D2,$D2,&$addr(0)); # d2 = h2*r0
|
|
|
|
&vpmuludq ($T2,$T0,&$addr(3)); # h0*r3
|
|
&vpaddq ($D3,$D3,$T2); # d3 += h0*r3
|
|
&vpmuludq ($T1,$T0,&$addr(4)); # h0*r4
|
|
&vpaddq ($D4,$D4,$T1); # d4 + h0*r4
|
|
&vpmuludq ($T2,$T0,&$addr(0)); # h0*r0
|
|
&vpaddq ($D0,$D0,$T2); # d0 + h0*r0
|
|
&vmovdqa ($T2,&QWP(32*1,"esp")); # h1
|
|
&vpmuludq ($T1,$T0,&$addr(1)); # h0*r1
|
|
&vpaddq ($D1,$D1,$T1); # d1 += h0*r1
|
|
&vpmuludq ($T0,$T0,&$addr(2)); # h0*r2
|
|
&vpaddq ($D2,$D2,$T0); # d2 += h0*r2
|
|
|
|
&vpmuludq ($T1,$T2,&$addr(2)); # h1*r2
|
|
&vpaddq ($D3,$D3,$T1); # d3 += h1*r2
|
|
&vpmuludq ($T0,$T2,&$addr(3)); # h1*r3
|
|
&vpaddq ($D4,$D4,$T0); # d4 += h1*r3
|
|
&vpmuludq ($T1,$T2,&$addr(8)); # h1*s4
|
|
&vpaddq ($D0,$D0,$T1); # d0 += h1*s4
|
|
&vmovdqa ($T1,&QWP(32*3,"esp")); # h3
|
|
&vpmuludq ($T0,$T2,&$addr(0)); # h1*r0
|
|
&vpaddq ($D1,$D1,$T0); # d1 += h1*r0
|
|
&vpmuludq ($T2,$T2,&$addr(1)); # h1*r1
|
|
&vpaddq ($D2,$D2,$T2); # d2 += h1*r1
|
|
|
|
&vpmuludq ($T0,$T1,&$addr(0)); # h3*r0
|
|
&vpaddq ($D3,$D3,$T0); # d3 += h3*r0
|
|
&vpmuludq ($T2,$T1,&$addr(1)); # h3*r1
|
|
&vpaddq ($D4,$D4,$T2); # d4 += h3*r1
|
|
&vpmuludq ($T0,$T1,&$addr(6)); # h3*s2
|
|
&vpaddq ($D0,$D0,$T0); # d0 += h3*s2
|
|
&vmovdqa ($T0,&QWP(32*4,"esp")); # h4
|
|
&vpmuludq ($T2,$T1,&$addr(7)); # h3*s3
|
|
&vpaddq ($D1,$D1,$T2); # d1+= h3*s3
|
|
&vpmuludq ($T1,$T1,&$addr(8)); # h3*s4
|
|
&vpaddq ($D2,$D2,$T1); # d2 += h3*s4
|
|
|
|
&vpmuludq ($T2,$T0,&$addr(8)); # h4*s4
|
|
&vpaddq ($D3,$D3,$T2); # d3 += h4*s4
|
|
&vpmuludq ($T1,$T0,&$addr(5)); # h4*s1
|
|
&vpaddq ($D0,$D0,$T1); # d0 += h4*s1
|
|
&vpmuludq ($T2,$T0,&$addr(0)); # h4*r0
|
|
&vpaddq ($D4,$D4,$T2); # d4 += h4*r0
|
|
&vmovdqa ($MASK,&QWP(64,"ebx"));
|
|
&vpmuludq ($T1,$T0,&$addr(6)); # h4*s2
|
|
&vpaddq ($D1,$D1,$T1); # d1 += h4*s2
|
|
&vpmuludq ($T0,$T0,&$addr(7)); # h4*s3
|
|
&vpaddq ($D2,$D2,$T0); # d2 += h4*s3
|
|
}
|
|
&vpmuladd (sub { my $i=shift; &QWP(32*$i-128,"edx"); });
|
|
|
|
sub vlazy_reduction {
|
|
################################################################
|
|
# lazy reduction
|
|
|
|
&vpsrlq ($T0,$D3,26);
|
|
&vpand ($D3,$D3,$MASK);
|
|
&vpsrlq ($T1,$D0,26);
|
|
&vpand ($D0,$D0,$MASK);
|
|
&vpaddq ($D4,$D4,$T0); # h3 -> h4
|
|
&vpaddq ($D1,$D1,$T1); # h0 -> h1
|
|
&vpsrlq ($T0,$D4,26);
|
|
&vpand ($D4,$D4,$MASK);
|
|
&vpsrlq ($T1,$D1,26);
|
|
&vpand ($D1,$D1,$MASK);
|
|
&vpaddq ($D2,$D2,$T1); # h1 -> h2
|
|
&vpaddq ($D0,$D0,$T0);
|
|
&vpsllq ($T0,$T0,2);
|
|
&vpsrlq ($T1,$D2,26);
|
|
&vpand ($D2,$D2,$MASK);
|
|
&vpaddq ($D0,$D0,$T0); # h4 -> h0
|
|
&vpaddq ($D3,$D3,$T1); # h2 -> h3
|
|
&vpsrlq ($T1,$D3,26);
|
|
&vpsrlq ($T0,$D0,26);
|
|
&vpand ($D0,$D0,$MASK);
|
|
&vpand ($D3,$D3,$MASK);
|
|
&vpaddq ($D1,$D1,$T0); # h0 -> h1
|
|
&vpaddq ($D4,$D4,$T1); # h3 -> h4
|
|
}
|
|
&vlazy_reduction();
|
|
|
|
&vmovdqu (&X($T0),&QWP(16*0,"esi")); # load input
|
|
&vmovdqu (&X($T1),&QWP(16*1,"esi"));
|
|
&vinserti128 ($T0,$T0,&QWP(16*2,"esi"),1);
|
|
&vinserti128 ($T1,$T1,&QWP(16*3,"esi"),1);
|
|
&lea ("esi",&DWP(16*4,"esi"));
|
|
&sub ("ecx",64);
|
|
&jnz (&label("loop"));
|
|
|
|
&set_label("tail");
|
|
&vsplat_input ();
|
|
&and ("ebx",-64); # restore pointer
|
|
|
|
&vpmuladd (sub { my $i=shift; &QWP(4+32*$i-128,"edx"); });
|
|
|
|
################################################################
|
|
# horizontal addition
|
|
|
|
&vpsrldq ($T0,$D4,8);
|
|
&vpsrldq ($T1,$D3,8);
|
|
&vpaddq ($D4,$D4,$T0);
|
|
&vpsrldq ($T0,$D0,8);
|
|
&vpaddq ($D3,$D3,$T1);
|
|
&vpsrldq ($T1,$D1,8);
|
|
&vpaddq ($D0,$D0,$T0);
|
|
&vpsrldq ($T0,$D2,8);
|
|
&vpaddq ($D1,$D1,$T1);
|
|
&vpermq ($T1,$D4,2); # keep folding
|
|
&vpaddq ($D2,$D2,$T0);
|
|
&vpermq ($T0,$D3,2);
|
|
&vpaddq ($D4,$D4,$T1);
|
|
&vpermq ($T1,$D0,2);
|
|
&vpaddq ($D3,$D3,$T0);
|
|
&vpermq ($T0,$D1,2);
|
|
&vpaddq ($D0,$D0,$T1);
|
|
&vpermq ($T1,$D2,2);
|
|
&vpaddq ($D1,$D1,$T0);
|
|
&vpaddq ($D2,$D2,$T1);
|
|
|
|
&vlazy_reduction();
|
|
|
|
&cmp ("ecx",0);
|
|
&je (&label("done"));
|
|
|
|
################################################################
|
|
# clear all but single word
|
|
|
|
&vpshufd (&X($D0),&X($D0),0b11111100);
|
|
&lea ("edx",&DWP(32*5+128,"esp")); # restore pointer
|
|
&vpshufd (&X($D1),&X($D1),0b11111100);
|
|
&vpshufd (&X($D2),&X($D2),0b11111100);
|
|
&vpshufd (&X($D3),&X($D3),0b11111100);
|
|
&vpshufd (&X($D4),&X($D4),0b11111100);
|
|
&jmp (&label("even"));
|
|
|
|
&set_label("done",16);
|
|
&vmovd (&DWP(-16*3+4*0,"edi"),&X($D0));# store hash value
|
|
&vmovd (&DWP(-16*3+4*1,"edi"),&X($D1));
|
|
&vmovd (&DWP(-16*3+4*2,"edi"),&X($D2));
|
|
&vmovd (&DWP(-16*3+4*3,"edi"),&X($D3));
|
|
&vmovd (&DWP(-16*3+4*4,"edi"),&X($D4));
|
|
&vzeroupper ();
|
|
&mov ("esp","ebp");
|
|
&set_label("nodata");
|
|
&function_end("_poly1305_blocks_avx2");
|
|
}
|
|
&set_label("const_sse2",64);
|
|
&data_word(1<<24,0, 1<<24,0, 1<<24,0, 1<<24,0);
|
|
&data_word(0,0, 0,0, 0,0, 0,0);
|
|
&data_word(0x03ffffff,0,0x03ffffff,0, 0x03ffffff,0, 0x03ffffff,0);
|
|
&data_word(0x0fffffff,0x0ffffffc,0x0ffffffc,0x0ffffffc);
|
|
}
|
|
&asciz ("Poly1305 for x86, CRYPTOGAMS by <appro\@openssl.org>");
|
|
&align (4);
|
|
|
|
&asm_finish();
|
|
|
|
close STDOUT;
|