452 lines
12 KiB
Perl
452 lines
12 KiB
Perl
|
#!/usr/bin/env perl
|
||
|
#
|
||
|
# ====================================================================
|
||
|
# Written by Andy Polyakov <appro@fy.chalmers.se> for the OpenSSL
|
||
|
# project. Rights for redistribution and usage in source and binary
|
||
|
# forms are granted according to the OpenSSL license.
|
||
|
# ====================================================================
|
||
|
#
|
||
|
# sha256/512_block procedure for x86_64.
|
||
|
#
|
||
|
# 40% improvement over compiler-generated code on Opteron. On EM64T
|
||
|
# sha256 was observed to run >80% faster and sha512 - >40%. No magical
|
||
|
# tricks, just straight implementation... I really wonder why gcc
|
||
|
# [being armed with inline assembler] fails to generate as fast code.
|
||
|
# The only thing which is cool about this module is that it's very
|
||
|
# same instruction sequence used for both SHA-256 and SHA-512. In
|
||
|
# former case the instructions operate on 32-bit operands, while in
|
||
|
# latter - on 64-bit ones. All I had to do is to get one flavor right,
|
||
|
# the other one passed the test right away:-)
|
||
|
#
|
||
|
# sha256_block runs in ~1005 cycles on Opteron, which gives you
|
||
|
# asymptotic performance of 64*1000/1005=63.7MBps times CPU clock
|
||
|
# frequency in GHz. sha512_block runs in ~1275 cycles, which results
|
||
|
# in 128*1000/1275=100MBps per GHz. Is there room for improvement?
|
||
|
# Well, if you compare it to IA-64 implementation, which maintains
|
||
|
# X[16] in register bank[!], tends to 4 instructions per CPU clock
|
||
|
# cycle and runs in 1003 cycles, 1275 is very good result for 3-way
|
||
|
# issue Opteron pipeline and X[16] maintained in memory. So that *if*
|
||
|
# there is a way to improve it, *then* the only way would be to try to
|
||
|
# offload X[16] updates to SSE unit, but that would require "deeper"
|
||
|
# loop unroll, which in turn would naturally cause size blow-up, not
|
||
|
# to mention increased complexity! And once again, only *if* it's
|
||
|
# actually possible to noticeably improve overall ILP, instruction
|
||
|
# level parallelism, on a given CPU implementation in this case.
|
||
|
#
|
||
|
# Special note on Intel EM64T. While Opteron CPU exhibits perfect
|
||
|
# perfromance ratio of 1.5 between 64- and 32-bit flavors [see above],
|
||
|
# [currently available] EM64T CPUs apparently are far from it. On the
|
||
|
# contrary, 64-bit version, sha512_block, is ~30% *slower* than 32-bit
|
||
|
# sha256_block:-( This is presumably because 64-bit shifts/rotates
|
||
|
# apparently are not atomic instructions, but implemented in microcode.
|
||
|
|
||
|
$flavour = shift;
|
||
|
$output = shift;
|
||
|
if ($flavour =~ /\./) { $output = $flavour; undef $flavour; }
|
||
|
|
||
|
$win64=0; $win64=1 if ($flavour =~ /[nm]asm|mingw64/ || $output =~ /\.asm$/);
|
||
|
|
||
|
$0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1;
|
||
|
( $xlate="${dir}x86_64-xlate.pl" and -f $xlate ) or
|
||
|
( $xlate="${dir}../../perlasm/x86_64-xlate.pl" and -f $xlate) or
|
||
|
die "can't locate x86_64-xlate.pl";
|
||
|
|
||
|
open OUT,"| \"$^X\" $xlate $flavour $output";
|
||
|
*STDOUT=*OUT;
|
||
|
|
||
|
if ($output =~ /512/) {
|
||
|
$func="sha512_block_data_order";
|
||
|
$TABLE="K512";
|
||
|
$SZ=8;
|
||
|
@ROT=($A,$B,$C,$D,$E,$F,$G,$H)=("%rax","%rbx","%rcx","%rdx",
|
||
|
"%r8", "%r9", "%r10","%r11");
|
||
|
($T1,$a0,$a1,$a2)=("%r12","%r13","%r14","%r15");
|
||
|
@Sigma0=(28,34,39);
|
||
|
@Sigma1=(14,18,41);
|
||
|
@sigma0=(1, 8, 7);
|
||
|
@sigma1=(19,61, 6);
|
||
|
$rounds=80;
|
||
|
} else {
|
||
|
$func="sha256_block_data_order";
|
||
|
$TABLE="K256";
|
||
|
$SZ=4;
|
||
|
@ROT=($A,$B,$C,$D,$E,$F,$G,$H)=("%eax","%ebx","%ecx","%edx",
|
||
|
"%r8d","%r9d","%r10d","%r11d");
|
||
|
($T1,$a0,$a1,$a2)=("%r12d","%r13d","%r14d","%r15d");
|
||
|
@Sigma0=( 2,13,22);
|
||
|
@Sigma1=( 6,11,25);
|
||
|
@sigma0=( 7,18, 3);
|
||
|
@sigma1=(17,19,10);
|
||
|
$rounds=64;
|
||
|
}
|
||
|
|
||
|
$ctx="%rdi"; # 1st arg
|
||
|
$round="%rdi"; # zaps $ctx
|
||
|
$inp="%rsi"; # 2nd arg
|
||
|
$Tbl="%rbp";
|
||
|
|
||
|
$_ctx="16*$SZ+0*8(%rsp)";
|
||
|
$_inp="16*$SZ+1*8(%rsp)";
|
||
|
$_end="16*$SZ+2*8(%rsp)";
|
||
|
$_rsp="16*$SZ+3*8(%rsp)";
|
||
|
$framesz="16*$SZ+4*8";
|
||
|
|
||
|
|
||
|
sub ROUND_00_15()
|
||
|
{ my ($i,$a,$b,$c,$d,$e,$f,$g,$h) = @_;
|
||
|
|
||
|
$code.=<<___;
|
||
|
ror \$`$Sigma1[2]-$Sigma1[1]`,$a0
|
||
|
mov $f,$a2
|
||
|
mov $T1,`$SZ*($i&0xf)`(%rsp)
|
||
|
|
||
|
ror \$`$Sigma0[2]-$Sigma0[1]`,$a1
|
||
|
xor $e,$a0
|
||
|
xor $g,$a2 # f^g
|
||
|
|
||
|
ror \$`$Sigma1[1]-$Sigma1[0]`,$a0
|
||
|
add $h,$T1 # T1+=h
|
||
|
xor $a,$a1
|
||
|
|
||
|
add ($Tbl,$round,$SZ),$T1 # T1+=K[round]
|
||
|
and $e,$a2 # (f^g)&e
|
||
|
mov $b,$h
|
||
|
|
||
|
ror \$`$Sigma0[1]-$Sigma0[0]`,$a1
|
||
|
xor $e,$a0
|
||
|
xor $g,$a2 # Ch(e,f,g)=((f^g)&e)^g
|
||
|
|
||
|
xor $c,$h # b^c
|
||
|
xor $a,$a1
|
||
|
add $a2,$T1 # T1+=Ch(e,f,g)
|
||
|
mov $b,$a2
|
||
|
|
||
|
ror \$$Sigma1[0],$a0 # Sigma1(e)
|
||
|
and $a,$h # h=(b^c)&a
|
||
|
and $c,$a2 # b&c
|
||
|
|
||
|
ror \$$Sigma0[0],$a1 # Sigma0(a)
|
||
|
add $a0,$T1 # T1+=Sigma1(e)
|
||
|
add $a2,$h # h+=b&c (completes +=Maj(a,b,c)
|
||
|
|
||
|
add $T1,$d # d+=T1
|
||
|
add $T1,$h # h+=T1
|
||
|
lea 1($round),$round # round++
|
||
|
add $a1,$h # h+=Sigma0(a)
|
||
|
|
||
|
___
|
||
|
}
|
||
|
|
||
|
sub ROUND_16_XX()
|
||
|
{ my ($i,$a,$b,$c,$d,$e,$f,$g,$h) = @_;
|
||
|
|
||
|
$code.=<<___;
|
||
|
mov `$SZ*(($i+1)&0xf)`(%rsp),$a0
|
||
|
mov `$SZ*(($i+14)&0xf)`(%rsp),$a1
|
||
|
mov $a0,$T1
|
||
|
mov $a1,$a2
|
||
|
|
||
|
ror \$`$sigma0[1]-$sigma0[0]`,$T1
|
||
|
xor $a0,$T1
|
||
|
shr \$$sigma0[2],$a0
|
||
|
|
||
|
ror \$$sigma0[0],$T1
|
||
|
xor $T1,$a0 # sigma0(X[(i+1)&0xf])
|
||
|
mov `$SZ*(($i+9)&0xf)`(%rsp),$T1
|
||
|
|
||
|
ror \$`$sigma1[1]-$sigma1[0]`,$a2
|
||
|
xor $a1,$a2
|
||
|
shr \$$sigma1[2],$a1
|
||
|
|
||
|
ror \$$sigma1[0],$a2
|
||
|
add $a0,$T1
|
||
|
xor $a2,$a1 # sigma1(X[(i+14)&0xf])
|
||
|
|
||
|
add `$SZ*($i&0xf)`(%rsp),$T1
|
||
|
mov $e,$a0
|
||
|
add $a1,$T1
|
||
|
mov $a,$a1
|
||
|
___
|
||
|
&ROUND_00_15(@_);
|
||
|
}
|
||
|
|
||
|
$code=<<___;
|
||
|
.text
|
||
|
|
||
|
.globl $func
|
||
|
.type $func,\@function,4
|
||
|
.align 16
|
||
|
$func:
|
||
|
push %rbx
|
||
|
push %rbp
|
||
|
push %r12
|
||
|
push %r13
|
||
|
push %r14
|
||
|
push %r15
|
||
|
mov %rsp,%r11 # copy %rsp
|
||
|
shl \$4,%rdx # num*16
|
||
|
sub \$$framesz,%rsp
|
||
|
lea ($inp,%rdx,$SZ),%rdx # inp+num*16*$SZ
|
||
|
and \$-64,%rsp # align stack frame
|
||
|
mov $ctx,$_ctx # save ctx, 1st arg
|
||
|
mov $inp,$_inp # save inp, 2nd arh
|
||
|
mov %rdx,$_end # save end pointer, "3rd" arg
|
||
|
mov %r11,$_rsp # save copy of %rsp
|
||
|
.Lprologue:
|
||
|
|
||
|
lea $TABLE(%rip),$Tbl
|
||
|
|
||
|
mov $SZ*0($ctx),$A
|
||
|
mov $SZ*1($ctx),$B
|
||
|
mov $SZ*2($ctx),$C
|
||
|
mov $SZ*3($ctx),$D
|
||
|
mov $SZ*4($ctx),$E
|
||
|
mov $SZ*5($ctx),$F
|
||
|
mov $SZ*6($ctx),$G
|
||
|
mov $SZ*7($ctx),$H
|
||
|
jmp .Lloop
|
||
|
|
||
|
.align 16
|
||
|
.Lloop:
|
||
|
xor $round,$round
|
||
|
___
|
||
|
for($i=0;$i<16;$i++) {
|
||
|
$code.=" mov $SZ*$i($inp),$T1\n";
|
||
|
$code.=" mov @ROT[4],$a0\n";
|
||
|
$code.=" mov @ROT[0],$a1\n";
|
||
|
$code.=" bswap $T1\n";
|
||
|
&ROUND_00_15($i,@ROT);
|
||
|
unshift(@ROT,pop(@ROT));
|
||
|
}
|
||
|
$code.=<<___;
|
||
|
jmp .Lrounds_16_xx
|
||
|
.align 16
|
||
|
.Lrounds_16_xx:
|
||
|
___
|
||
|
for(;$i<32;$i++) {
|
||
|
&ROUND_16_XX($i,@ROT);
|
||
|
unshift(@ROT,pop(@ROT));
|
||
|
}
|
||
|
|
||
|
$code.=<<___;
|
||
|
cmp \$$rounds,$round
|
||
|
jb .Lrounds_16_xx
|
||
|
|
||
|
mov $_ctx,$ctx
|
||
|
lea 16*$SZ($inp),$inp
|
||
|
|
||
|
add $SZ*0($ctx),$A
|
||
|
add $SZ*1($ctx),$B
|
||
|
add $SZ*2($ctx),$C
|
||
|
add $SZ*3($ctx),$D
|
||
|
add $SZ*4($ctx),$E
|
||
|
add $SZ*5($ctx),$F
|
||
|
add $SZ*6($ctx),$G
|
||
|
add $SZ*7($ctx),$H
|
||
|
|
||
|
cmp $_end,$inp
|
||
|
|
||
|
mov $A,$SZ*0($ctx)
|
||
|
mov $B,$SZ*1($ctx)
|
||
|
mov $C,$SZ*2($ctx)
|
||
|
mov $D,$SZ*3($ctx)
|
||
|
mov $E,$SZ*4($ctx)
|
||
|
mov $F,$SZ*5($ctx)
|
||
|
mov $G,$SZ*6($ctx)
|
||
|
mov $H,$SZ*7($ctx)
|
||
|
jb .Lloop
|
||
|
|
||
|
mov $_rsp,%rsi
|
||
|
mov (%rsi),%r15
|
||
|
mov 8(%rsi),%r14
|
||
|
mov 16(%rsi),%r13
|
||
|
mov 24(%rsi),%r12
|
||
|
mov 32(%rsi),%rbp
|
||
|
mov 40(%rsi),%rbx
|
||
|
lea 48(%rsi),%rsp
|
||
|
.Lepilogue:
|
||
|
ret
|
||
|
.size $func,.-$func
|
||
|
___
|
||
|
|
||
|
if ($SZ==4) {
|
||
|
$code.=<<___;
|
||
|
.align 64
|
||
|
.type $TABLE,\@object
|
||
|
$TABLE:
|
||
|
.long 0x428a2f98,0x71374491,0xb5c0fbcf,0xe9b5dba5
|
||
|
.long 0x3956c25b,0x59f111f1,0x923f82a4,0xab1c5ed5
|
||
|
.long 0xd807aa98,0x12835b01,0x243185be,0x550c7dc3
|
||
|
.long 0x72be5d74,0x80deb1fe,0x9bdc06a7,0xc19bf174
|
||
|
.long 0xe49b69c1,0xefbe4786,0x0fc19dc6,0x240ca1cc
|
||
|
.long 0x2de92c6f,0x4a7484aa,0x5cb0a9dc,0x76f988da
|
||
|
.long 0x983e5152,0xa831c66d,0xb00327c8,0xbf597fc7
|
||
|
.long 0xc6e00bf3,0xd5a79147,0x06ca6351,0x14292967
|
||
|
.long 0x27b70a85,0x2e1b2138,0x4d2c6dfc,0x53380d13
|
||
|
.long 0x650a7354,0x766a0abb,0x81c2c92e,0x92722c85
|
||
|
.long 0xa2bfe8a1,0xa81a664b,0xc24b8b70,0xc76c51a3
|
||
|
.long 0xd192e819,0xd6990624,0xf40e3585,0x106aa070
|
||
|
.long 0x19a4c116,0x1e376c08,0x2748774c,0x34b0bcb5
|
||
|
.long 0x391c0cb3,0x4ed8aa4a,0x5b9cca4f,0x682e6ff3
|
||
|
.long 0x748f82ee,0x78a5636f,0x84c87814,0x8cc70208
|
||
|
.long 0x90befffa,0xa4506ceb,0xbef9a3f7,0xc67178f2
|
||
|
___
|
||
|
} else {
|
||
|
$code.=<<___;
|
||
|
.align 64
|
||
|
.type $TABLE,\@object
|
||
|
$TABLE:
|
||
|
.quad 0x428a2f98d728ae22,0x7137449123ef65cd
|
||
|
.quad 0xb5c0fbcfec4d3b2f,0xe9b5dba58189dbbc
|
||
|
.quad 0x3956c25bf348b538,0x59f111f1b605d019
|
||
|
.quad 0x923f82a4af194f9b,0xab1c5ed5da6d8118
|
||
|
.quad 0xd807aa98a3030242,0x12835b0145706fbe
|
||
|
.quad 0x243185be4ee4b28c,0x550c7dc3d5ffb4e2
|
||
|
.quad 0x72be5d74f27b896f,0x80deb1fe3b1696b1
|
||
|
.quad 0x9bdc06a725c71235,0xc19bf174cf692694
|
||
|
.quad 0xe49b69c19ef14ad2,0xefbe4786384f25e3
|
||
|
.quad 0x0fc19dc68b8cd5b5,0x240ca1cc77ac9c65
|
||
|
.quad 0x2de92c6f592b0275,0x4a7484aa6ea6e483
|
||
|
.quad 0x5cb0a9dcbd41fbd4,0x76f988da831153b5
|
||
|
.quad 0x983e5152ee66dfab,0xa831c66d2db43210
|
||
|
.quad 0xb00327c898fb213f,0xbf597fc7beef0ee4
|
||
|
.quad 0xc6e00bf33da88fc2,0xd5a79147930aa725
|
||
|
.quad 0x06ca6351e003826f,0x142929670a0e6e70
|
||
|
.quad 0x27b70a8546d22ffc,0x2e1b21385c26c926
|
||
|
.quad 0x4d2c6dfc5ac42aed,0x53380d139d95b3df
|
||
|
.quad 0x650a73548baf63de,0x766a0abb3c77b2a8
|
||
|
.quad 0x81c2c92e47edaee6,0x92722c851482353b
|
||
|
.quad 0xa2bfe8a14cf10364,0xa81a664bbc423001
|
||
|
.quad 0xc24b8b70d0f89791,0xc76c51a30654be30
|
||
|
.quad 0xd192e819d6ef5218,0xd69906245565a910
|
||
|
.quad 0xf40e35855771202a,0x106aa07032bbd1b8
|
||
|
.quad 0x19a4c116b8d2d0c8,0x1e376c085141ab53
|
||
|
.quad 0x2748774cdf8eeb99,0x34b0bcb5e19b48a8
|
||
|
.quad 0x391c0cb3c5c95a63,0x4ed8aa4ae3418acb
|
||
|
.quad 0x5b9cca4f7763e373,0x682e6ff3d6b2b8a3
|
||
|
.quad 0x748f82ee5defb2fc,0x78a5636f43172f60
|
||
|
.quad 0x84c87814a1f0ab72,0x8cc702081a6439ec
|
||
|
.quad 0x90befffa23631e28,0xa4506cebde82bde9
|
||
|
.quad 0xbef9a3f7b2c67915,0xc67178f2e372532b
|
||
|
.quad 0xca273eceea26619c,0xd186b8c721c0c207
|
||
|
.quad 0xeada7dd6cde0eb1e,0xf57d4f7fee6ed178
|
||
|
.quad 0x06f067aa72176fba,0x0a637dc5a2c898a6
|
||
|
.quad 0x113f9804bef90dae,0x1b710b35131c471b
|
||
|
.quad 0x28db77f523047d84,0x32caab7b40c72493
|
||
|
.quad 0x3c9ebe0a15c9bebc,0x431d67c49c100d4c
|
||
|
.quad 0x4cc5d4becb3e42b6,0x597f299cfc657e2a
|
||
|
.quad 0x5fcb6fab3ad6faec,0x6c44198c4a475817
|
||
|
___
|
||
|
}
|
||
|
|
||
|
# EXCEPTION_DISPOSITION handler (EXCEPTION_RECORD *rec,ULONG64 frame,
|
||
|
# CONTEXT *context,DISPATCHER_CONTEXT *disp)
|
||
|
if ($win64) {
|
||
|
$rec="%rcx";
|
||
|
$frame="%rdx";
|
||
|
$context="%r8";
|
||
|
$disp="%r9";
|
||
|
|
||
|
$code.=<<___;
|
||
|
.extern __imp_RtlVirtualUnwind
|
||
|
.type se_handler,\@abi-omnipotent
|
||
|
.align 16
|
||
|
se_handler:
|
||
|
push %rsi
|
||
|
push %rdi
|
||
|
push %rbx
|
||
|
push %rbp
|
||
|
push %r12
|
||
|
push %r13
|
||
|
push %r14
|
||
|
push %r15
|
||
|
pushfq
|
||
|
sub \$64,%rsp
|
||
|
|
||
|
mov 120($context),%rax # pull context->Rax
|
||
|
mov 248($context),%rbx # pull context->Rip
|
||
|
|
||
|
lea .Lprologue(%rip),%r10
|
||
|
cmp %r10,%rbx # context->Rip<.Lprologue
|
||
|
jb .Lin_prologue
|
||
|
|
||
|
mov 152($context),%rax # pull context->Rsp
|
||
|
|
||
|
lea .Lepilogue(%rip),%r10
|
||
|
cmp %r10,%rbx # context->Rip>=.Lepilogue
|
||
|
jae .Lin_prologue
|
||
|
|
||
|
mov 16*$SZ+3*8(%rax),%rax # pull $_rsp
|
||
|
lea 48(%rax),%rax
|
||
|
|
||
|
mov -8(%rax),%rbx
|
||
|
mov -16(%rax),%rbp
|
||
|
mov -24(%rax),%r12
|
||
|
mov -32(%rax),%r13
|
||
|
mov -40(%rax),%r14
|
||
|
mov -48(%rax),%r15
|
||
|
mov %rbx,144($context) # restore context->Rbx
|
||
|
mov %rbp,160($context) # restore context->Rbp
|
||
|
mov %r12,216($context) # restore context->R12
|
||
|
mov %r13,224($context) # restore context->R13
|
||
|
mov %r14,232($context) # restore context->R14
|
||
|
mov %r15,240($context) # restore context->R15
|
||
|
|
||
|
.Lin_prologue:
|
||
|
mov 8(%rax),%rdi
|
||
|
mov 16(%rax),%rsi
|
||
|
mov %rax,152($context) # restore context->Rsp
|
||
|
mov %rsi,168($context) # restore context->Rsi
|
||
|
mov %rdi,176($context) # restore context->Rdi
|
||
|
|
||
|
mov 40($disp),%rdi # disp->ContextRecord
|
||
|
mov $context,%rsi # context
|
||
|
mov \$154,%ecx # sizeof(CONTEXT)
|
||
|
.long 0xa548f3fc # cld; rep movsq
|
||
|
|
||
|
mov $disp,%rsi
|
||
|
xor %rcx,%rcx # arg1, UNW_FLAG_NHANDLER
|
||
|
mov 8(%rsi),%rdx # arg2, disp->ImageBase
|
||
|
mov 0(%rsi),%r8 # arg3, disp->ControlPc
|
||
|
mov 16(%rsi),%r9 # arg4, disp->FunctionEntry
|
||
|
mov 40(%rsi),%r10 # disp->ContextRecord
|
||
|
lea 56(%rsi),%r11 # &disp->HandlerData
|
||
|
lea 24(%rsi),%r12 # &disp->EstablisherFrame
|
||
|
mov %r10,32(%rsp) # arg5
|
||
|
mov %r11,40(%rsp) # arg6
|
||
|
mov %r12,48(%rsp) # arg7
|
||
|
mov %rcx,56(%rsp) # arg8, (NULL)
|
||
|
call *__imp_RtlVirtualUnwind(%rip)
|
||
|
|
||
|
mov \$1,%eax # ExceptionContinueSearch
|
||
|
add \$64,%rsp
|
||
|
popfq
|
||
|
pop %r15
|
||
|
pop %r14
|
||
|
pop %r13
|
||
|
pop %r12
|
||
|
pop %rbp
|
||
|
pop %rbx
|
||
|
pop %rdi
|
||
|
pop %rsi
|
||
|
ret
|
||
|
.size se_handler,.-se_handler
|
||
|
|
||
|
.section .pdata
|
||
|
.align 4
|
||
|
.rva .LSEH_begin_$func
|
||
|
.rva .LSEH_end_$func
|
||
|
.rva .LSEH_info_$func
|
||
|
|
||
|
.section .xdata
|
||
|
.align 8
|
||
|
.LSEH_info_$func:
|
||
|
.byte 9,0,0,0
|
||
|
.rva se_handler
|
||
|
___
|
||
|
}
|
||
|
|
||
|
$code =~ s/\`([^\`]*)\`/eval $1/gem;
|
||
|
print $code;
|
||
|
close STDOUT;
|