1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804 |
- #! /usr/bin/env perl
- # Copyright 2013-2020 The OpenSSL Project Authors. All Rights Reserved.
- #
- # Licensed under the Apache License 2.0 (the "License"). You may not use
- # this file except in compliance with the License. You can obtain a copy
- # in the file LICENSE in the source distribution or at
- # https://www.openssl.org/source/license.html
- #
- # ====================================================================
- # 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/.
- # ====================================================================
- #
- # January 2013
- #
- # This is AESNI-CBC+SHA256 stitch implementation. The idea, as spelled
- # in http://download.intel.com/design/intarch/papers/323686.pdf, is
- # that since AESNI-CBC encrypt exhibit *very* low instruction-level
- # parallelism, interleaving it with another algorithm would allow to
- # utilize processor resources better and achieve better performance.
- # SHA256 instruction sequences(*) are taken from sha512-x86_64.pl and
- # AESNI code is weaved into it. As SHA256 dominates execution time,
- # stitch performance does not depend on AES key length. Below are
- # performance numbers in cycles per processed byte, less is better,
- # for standalone AESNI-CBC encrypt, standalone SHA256, and stitched
- # subroutine:
- #
- # AES-128/-192/-256+SHA256 this(**) gain
- # Sandy Bridge 5.05/6.05/7.05+11.6 13.0 +28%/36%/43%
- # Ivy Bridge 5.05/6.05/7.05+10.3 11.6 +32%/41%/50%
- # Haswell 4.43/5.29/6.19+7.80 8.79 +39%/49%/59%
- # Skylake 2.62/3.14/3.62+7.70 8.10 +27%/34%/40%
- # Bulldozer 5.77/6.89/8.00+13.7 13.7 +42%/50%/58%
- # Ryzen(***) 2.71/-/3.71+2.05 2.74/-/3.73 +74%/-/54%
- # Goldmont(***) 3.82/-/5.35+4.16 4.73/-/5.94 +69%/-/60%
- #
- # (*) there are XOP, AVX1 and AVX2 code paths, meaning that
- # Westmere is omitted from loop, this is because gain was not
- # estimated high enough to justify the effort;
- # (**) these are EVP-free results, results obtained with 'speed
- # -evp aes-256-cbc-hmac-sha256' will vary by percent or two;
- # (***) these are SHAEXT results;
- # $output is the last argument if it looks like a file (it has an extension)
- # $flavour is the first argument if it doesn't look like a file
- $output = $#ARGV >= 0 && $ARGV[$#ARGV] =~ m|\.\w+$| ? pop : undef;
- $flavour = $#ARGV >= 0 && $ARGV[0] !~ m|\.| ? shift : undef;
- $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";
- 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 && $win64 && ($flavour =~ /nasm/ || $ENV{ASM} =~ /nasm/) &&
- `nasm -v 2>&1` =~ /NASM version ([2-9]\.[0-9]+)/) {
- $avx = ($1>=2.09) + ($1>=2.10);
- }
- if (!$avx && $win64 && ($flavour =~ /masm/ || $ENV{ASM} =~ /ml64/) &&
- `ml64 2>&1` =~ /Version ([0-9]+)\./) {
- $avx = ($1>=10) + ($1>=12);
- }
- if (!$avx && `$ENV{CC} -v 2>&1` =~ /((?:clang|LLVM) version|.*based on LLVM) ([0-9]+\.[0-9]+)/) {
- $avx = ($2>=3.0) + ($2>3.0);
- }
- $shaext=$avx; ### set to zero if compiling for 1.0.1
- $avx=1 if (!$shaext && $avx);
- open OUT,"| \"$^X\" \"$xlate\" $flavour \"$output\""
- or die "can't call $xlate: $!";
- *STDOUT=*OUT;
- $func="aesni_cbc_sha256_enc";
- $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,$a3)=("%r12d","%r13d","%r14d","%r15d","%esi");
- @Sigma0=( 2,13,22);
- @Sigma1=( 6,11,25);
- @sigma0=( 7,18, 3);
- @sigma1=(17,19,10);
- $rounds=64;
- ########################################################################
- # void aesni_cbc_sha256_enc(const void *inp,
- # void *out,
- # size_t length,
- # const AES_KEY *key,
- # unsigned char *iv,
- # SHA256_CTX *ctx,
- # const void *in0);
- ($inp, $out, $len, $key, $ivp, $ctx, $in0) =
- ("%rdi","%rsi","%rdx","%rcx","%r8","%r9","%r10");
- $Tbl="%rbp";
- $_inp="16*$SZ+0*8(%rsp)";
- $_out="16*$SZ+1*8(%rsp)";
- $_end="16*$SZ+2*8(%rsp)";
- $_key="16*$SZ+3*8(%rsp)";
- $_ivp="16*$SZ+4*8(%rsp)";
- $_ctx="16*$SZ+5*8(%rsp)";
- $_in0="16*$SZ+6*8(%rsp)";
- $_rsp="`16*$SZ+7*8`(%rsp)";
- $framesz=16*$SZ+8*8;
- $code=<<___;
- .text
- .extern OPENSSL_ia32cap_P
- .globl $func
- .type $func,\@abi-omnipotent
- .align 16
- $func:
- .cfi_startproc
- ___
- if ($avx) {
- $code.=<<___;
- lea OPENSSL_ia32cap_P(%rip),%r11
- mov \$1,%eax
- cmp \$0,`$win64?"%rcx":"%rdi"`
- je .Lprobe
- mov 0(%r11),%eax
- mov 4(%r11),%r10
- ___
- $code.=<<___ if ($shaext);
- bt \$61,%r10 # check for SHA
- jc ${func}_shaext
- ___
- $code.=<<___;
- mov %r10,%r11
- shr \$32,%r11
- test \$`1<<11`,%r10d # check for XOP
- jnz ${func}_xop
- ___
- $code.=<<___ if ($avx>1);
- and \$`1<<8|1<<5|1<<3`,%r11d # check for BMI2+AVX2+BMI1
- cmp \$`1<<8|1<<5|1<<3`,%r11d
- je ${func}_avx2
- ___
- $code.=<<___;
- and \$`1<<28`,%r10d # check for AVX
- jnz ${func}_avx
- ud2
- ___
- }
- $code.=<<___;
- xor %eax,%eax
- cmp \$0,`$win64?"%rcx":"%rdi"`
- je .Lprobe
- ud2
- .Lprobe:
- ret
- .cfi_endproc
- .size $func,.-$func
- .align 64
- .type $TABLE,\@object
- $TABLE:
- .long 0x428a2f98,0x71374491,0xb5c0fbcf,0xe9b5dba5
- .long 0x428a2f98,0x71374491,0xb5c0fbcf,0xe9b5dba5
- .long 0x3956c25b,0x59f111f1,0x923f82a4,0xab1c5ed5
- .long 0x3956c25b,0x59f111f1,0x923f82a4,0xab1c5ed5
- .long 0xd807aa98,0x12835b01,0x243185be,0x550c7dc3
- .long 0xd807aa98,0x12835b01,0x243185be,0x550c7dc3
- .long 0x72be5d74,0x80deb1fe,0x9bdc06a7,0xc19bf174
- .long 0x72be5d74,0x80deb1fe,0x9bdc06a7,0xc19bf174
- .long 0xe49b69c1,0xefbe4786,0x0fc19dc6,0x240ca1cc
- .long 0xe49b69c1,0xefbe4786,0x0fc19dc6,0x240ca1cc
- .long 0x2de92c6f,0x4a7484aa,0x5cb0a9dc,0x76f988da
- .long 0x2de92c6f,0x4a7484aa,0x5cb0a9dc,0x76f988da
- .long 0x983e5152,0xa831c66d,0xb00327c8,0xbf597fc7
- .long 0x983e5152,0xa831c66d,0xb00327c8,0xbf597fc7
- .long 0xc6e00bf3,0xd5a79147,0x06ca6351,0x14292967
- .long 0xc6e00bf3,0xd5a79147,0x06ca6351,0x14292967
- .long 0x27b70a85,0x2e1b2138,0x4d2c6dfc,0x53380d13
- .long 0x27b70a85,0x2e1b2138,0x4d2c6dfc,0x53380d13
- .long 0x650a7354,0x766a0abb,0x81c2c92e,0x92722c85
- .long 0x650a7354,0x766a0abb,0x81c2c92e,0x92722c85
- .long 0xa2bfe8a1,0xa81a664b,0xc24b8b70,0xc76c51a3
- .long 0xa2bfe8a1,0xa81a664b,0xc24b8b70,0xc76c51a3
- .long 0xd192e819,0xd6990624,0xf40e3585,0x106aa070
- .long 0xd192e819,0xd6990624,0xf40e3585,0x106aa070
- .long 0x19a4c116,0x1e376c08,0x2748774c,0x34b0bcb5
- .long 0x19a4c116,0x1e376c08,0x2748774c,0x34b0bcb5
- .long 0x391c0cb3,0x4ed8aa4a,0x5b9cca4f,0x682e6ff3
- .long 0x391c0cb3,0x4ed8aa4a,0x5b9cca4f,0x682e6ff3
- .long 0x748f82ee,0x78a5636f,0x84c87814,0x8cc70208
- .long 0x748f82ee,0x78a5636f,0x84c87814,0x8cc70208
- .long 0x90befffa,0xa4506ceb,0xbef9a3f7,0xc67178f2
- .long 0x90befffa,0xa4506ceb,0xbef9a3f7,0xc67178f2
- .long 0x00010203,0x04050607,0x08090a0b,0x0c0d0e0f
- .long 0x00010203,0x04050607,0x08090a0b,0x0c0d0e0f
- .long 0,0,0,0, 0,0,0,0, -1,-1,-1,-1
- .long 0,0,0,0, 0,0,0,0
- .asciz "AESNI-CBC+SHA256 stitch for x86_64, CRYPTOGAMS by <appro\@openssl.org>"
- .align 64
- ___
- ######################################################################
- # SIMD code paths
- #
- {{{
- ($iv,$inout,$roundkey,$temp,
- $mask10,$mask12,$mask14,$offload)=map("%xmm$_",(8..15));
- $aesni_cbc_idx=0;
- @aesni_cbc_block = (
- ## &vmovdqu ($roundkey,"0x00-0x80($inp)");'
- ## &vmovdqu ($inout,($inp));
- ## &mov ($_inp,$inp);
- '&vpxor ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x10-0x80($inp)");',
- '&vpxor ($inout,$inout,$iv);',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x20-0x80($inp)");',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x30-0x80($inp)");',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x40-0x80($inp)");',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x50-0x80($inp)");',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x60-0x80($inp)");',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x70-0x80($inp)");',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x80-0x80($inp)");',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x90-0x80($inp)");',
- '&vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0xa0-0x80($inp)");',
- '&vaesenclast ($temp,$inout,$roundkey);'.
- ' &vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0xb0-0x80($inp)");',
- '&vpand ($iv,$temp,$mask10);'.
- ' &vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0xc0-0x80($inp)");',
- '&vaesenclast ($temp,$inout,$roundkey);'.
- ' &vaesenc ($inout,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0xd0-0x80($inp)");',
- '&vpand ($temp,$temp,$mask12);'.
- ' &vaesenc ($inout,$inout,$roundkey);'.
- '&vmovdqu ($roundkey,"0xe0-0x80($inp)");',
- '&vpor ($iv,$iv,$temp);'.
- ' &vaesenclast ($temp,$inout,$roundkey);'.
- ' &vmovdqu ($roundkey,"0x00-0x80($inp)");'
- ## &mov ($inp,$_inp);
- ## &mov ($out,$_out);
- ## &vpand ($temp,$temp,$mask14);
- ## &vpor ($iv,$iv,$temp);
- ## &vmovdqu ($iv,($out,$inp);
- ## &lea (inp,16($inp));
- );
- my $a4=$T1;
- my ($a,$b,$c,$d,$e,$f,$g,$h);
- sub AUTOLOAD() # thunk [simplified] 32-bit style perlasm
- { my $opcode = $AUTOLOAD; $opcode =~ s/.*:://;
- my $arg = pop;
- $arg = "\$$arg" if ($arg*1 eq $arg);
- $code .= "\t$opcode\t".join(',',$arg,reverse @_)."\n";
- }
- sub body_00_15 () {
- (
- '($a,$b,$c,$d,$e,$f,$g,$h)=@ROT;'.
- '&ror ($a0,$Sigma1[2]-$Sigma1[1])',
- '&mov ($a,$a1)',
- '&mov ($a4,$f)',
- '&xor ($a0,$e)',
- '&ror ($a1,$Sigma0[2]-$Sigma0[1])',
- '&xor ($a4,$g)', # f^g
- '&ror ($a0,$Sigma1[1]-$Sigma1[0])',
- '&xor ($a1,$a)',
- '&and ($a4,$e)', # (f^g)&e
- @aesni_cbc_block[$aesni_cbc_idx++].
- '&xor ($a0,$e)',
- '&add ($h,$SZ*($i&15)."(%rsp)")', # h+=X[i]+K[i]
- '&mov ($a2,$a)',
- '&ror ($a1,$Sigma0[1]-$Sigma0[0])',
- '&xor ($a4,$g)', # Ch(e,f,g)=((f^g)&e)^g
- '&xor ($a2,$b)', # a^b, b^c in next round
- '&ror ($a0,$Sigma1[0])', # Sigma1(e)
- '&add ($h,$a4)', # h+=Ch(e,f,g)
- '&and ($a3,$a2)', # (b^c)&(a^b)
- '&xor ($a1,$a)',
- '&add ($h,$a0)', # h+=Sigma1(e)
- '&xor ($a3,$b)', # Maj(a,b,c)=Ch(a^b,c,b)
- '&add ($d,$h)', # d+=h
- '&ror ($a1,$Sigma0[0])', # Sigma0(a)
- '&add ($h,$a3)', # h+=Maj(a,b,c)
- '&mov ($a0,$d)',
- '&add ($a1,$h);'. # h+=Sigma0(a)
- '($a2,$a3) = ($a3,$a2); unshift(@ROT,pop(@ROT)); $i++;'
- );
- }
- if ($avx) {{
- ######################################################################
- # XOP code path
- #
- $code.=<<___;
- .type ${func}_xop,\@function,6
- .align 64
- ${func}_xop:
- .cfi_startproc
- .Lxop_shortcut:
- mov `($win64?56:8)`(%rsp),$in0 # load 7th parameter
- mov %rsp,%rax # copy %rsp
- .cfi_def_cfa_register %rax
- push %rbx
- .cfi_push %rbx
- push %rbp
- .cfi_push %rbp
- push %r12
- .cfi_push %r12
- push %r13
- .cfi_push %r13
- push %r14
- .cfi_push %r14
- push %r15
- .cfi_push %r15
- sub \$`$framesz+$win64*16*10`,%rsp
- and \$-64,%rsp # align stack frame
- shl \$6,$len
- sub $inp,$out # re-bias
- sub $inp,$in0
- add $inp,$len # end of input
- #mov $inp,$_inp # saved later
- mov $out,$_out
- mov $len,$_end
- #mov $key,$_key # remains resident in $inp register
- mov $ivp,$_ivp
- mov $ctx,$_ctx
- mov $in0,$_in0
- mov %rax,$_rsp
- .cfi_cfa_expression $_rsp,deref,+8
- ___
- $code.=<<___ if ($win64);
- movaps %xmm6,`$framesz+16*0`(%rsp)
- movaps %xmm7,`$framesz+16*1`(%rsp)
- movaps %xmm8,`$framesz+16*2`(%rsp)
- movaps %xmm9,`$framesz+16*3`(%rsp)
- movaps %xmm10,`$framesz+16*4`(%rsp)
- movaps %xmm11,`$framesz+16*5`(%rsp)
- movaps %xmm12,`$framesz+16*6`(%rsp)
- movaps %xmm13,`$framesz+16*7`(%rsp)
- movaps %xmm14,`$framesz+16*8`(%rsp)
- movaps %xmm15,`$framesz+16*9`(%rsp)
- ___
- $code.=<<___;
- .Lprologue_xop:
- vzeroall
- mov $inp,%r12 # borrow $a4
- lea 0x80($key),$inp # size optimization, reassign
- lea $TABLE+`$SZ*2*$rounds+32`(%rip),%r13 # borrow $a0
- mov 0xf0-0x80($inp),%r14d # rounds, borrow $a1
- mov $ctx,%r15 # borrow $a2
- mov $in0,%rsi # borrow $a3
- vmovdqu ($ivp),$iv # load IV
- sub \$9,%r14
- mov $SZ*0(%r15),$A
- mov $SZ*1(%r15),$B
- mov $SZ*2(%r15),$C
- mov $SZ*3(%r15),$D
- mov $SZ*4(%r15),$E
- mov $SZ*5(%r15),$F
- mov $SZ*6(%r15),$G
- mov $SZ*7(%r15),$H
- vmovdqa 0x00(%r13,%r14,8),$mask14
- vmovdqa 0x10(%r13,%r14,8),$mask12
- vmovdqa 0x20(%r13,%r14,8),$mask10
- vmovdqu 0x00-0x80($inp),$roundkey
- jmp .Lloop_xop
- ___
- if ($SZ==4) { # SHA256
- my @X = map("%xmm$_",(0..3));
- my ($t0,$t1,$t2,$t3) = map("%xmm$_",(4..7));
- $code.=<<___;
- .align 16
- .Lloop_xop:
- vmovdqa $TABLE+`$SZ*2*$rounds`(%rip),$t3
- vmovdqu 0x00(%rsi,%r12),@X[0]
- vmovdqu 0x10(%rsi,%r12),@X[1]
- vmovdqu 0x20(%rsi,%r12),@X[2]
- vmovdqu 0x30(%rsi,%r12),@X[3]
- vpshufb $t3,@X[0],@X[0]
- lea $TABLE(%rip),$Tbl
- vpshufb $t3,@X[1],@X[1]
- vpshufb $t3,@X[2],@X[2]
- vpaddd 0x00($Tbl),@X[0],$t0
- vpshufb $t3,@X[3],@X[3]
- vpaddd 0x20($Tbl),@X[1],$t1
- vpaddd 0x40($Tbl),@X[2],$t2
- vpaddd 0x60($Tbl),@X[3],$t3
- vmovdqa $t0,0x00(%rsp)
- mov $A,$a1
- vmovdqa $t1,0x10(%rsp)
- mov $B,$a3
- vmovdqa $t2,0x20(%rsp)
- xor $C,$a3 # magic
- vmovdqa $t3,0x30(%rsp)
- mov $E,$a0
- jmp .Lxop_00_47
- .align 16
- .Lxop_00_47:
- sub \$-16*2*$SZ,$Tbl # size optimization
- vmovdqu (%r12),$inout # $a4
- mov %r12,$_inp # $a4
- ___
- sub XOP_256_00_47 () {
- my $j = shift;
- my $body = shift;
- my @X = @_;
- my @insns = (&$body,&$body,&$body,&$body); # 104 instructions
- &vpalignr ($t0,@X[1],@X[0],$SZ); # X[1..4]
- eval(shift(@insns));
- eval(shift(@insns));
- &vpalignr ($t3,@X[3],@X[2],$SZ); # X[9..12]
- eval(shift(@insns));
- eval(shift(@insns));
- &vprotd ($t1,$t0,8*$SZ-$sigma0[1]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vpsrld ($t0,$t0,$sigma0[2]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vpaddd (@X[0],@X[0],$t3); # X[0..3] += X[9..12]
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vprotd ($t2,$t1,$sigma0[1]-$sigma0[0]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vpxor ($t0,$t0,$t1);
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vprotd ($t3,@X[3],8*$SZ-$sigma1[1]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vpxor ($t0,$t0,$t2); # sigma0(X[1..4])
- eval(shift(@insns));
- eval(shift(@insns));
- &vpsrld ($t2,@X[3],$sigma1[2]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vpaddd (@X[0],@X[0],$t0); # X[0..3] += sigma0(X[1..4])
- eval(shift(@insns));
- eval(shift(@insns));
- &vprotd ($t1,$t3,$sigma1[1]-$sigma1[0]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vpxor ($t3,$t3,$t2);
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vpxor ($t3,$t3,$t1); # sigma1(X[14..15])
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vpsrldq ($t3,$t3,8);
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vpaddd (@X[0],@X[0],$t3); # X[0..1] += sigma1(X[14..15])
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vprotd ($t3,@X[0],8*$SZ-$sigma1[1]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vpsrld ($t2,@X[0],$sigma1[2]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vprotd ($t1,$t3,$sigma1[1]-$sigma1[0]);
- eval(shift(@insns));
- eval(shift(@insns));
- &vpxor ($t3,$t3,$t2);
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vpxor ($t3,$t3,$t1); # sigma1(X[16..17])
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vpslldq ($t3,$t3,8); # 22 instructions
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vpaddd (@X[0],@X[0],$t3); # X[2..3] += sigma1(X[16..17])
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- &vpaddd ($t2,@X[0],16*2*$j."($Tbl)");
- foreach (@insns) { eval; } # remaining instructions
- &vmovdqa (16*$j."(%rsp)",$t2);
- }
- $aesni_cbc_idx=0;
- for ($i=0,$j=0; $j<4; $j++) {
- &XOP_256_00_47($j,\&body_00_15,@X);
- push(@X,shift(@X)); # rotate(@X)
- }
- &mov ("%r12",$_inp); # borrow $a4
- &vpand ($temp,$temp,$mask14);
- &mov ("%r15",$_out); # borrow $a2
- &vpor ($iv,$iv,$temp);
- &vmovdqu ("(%r15,%r12)",$iv); # write output
- &lea ("%r12","16(%r12)"); # inp++
- &cmpb ($SZ-1+16*2*$SZ."($Tbl)",0);
- &jne (".Lxop_00_47");
- &vmovdqu ($inout,"(%r12)");
- &mov ($_inp,"%r12");
- $aesni_cbc_idx=0;
- for ($i=0; $i<16; ) {
- foreach(body_00_15()) { eval; }
- }
- }
- $code.=<<___;
- mov $_inp,%r12 # borrow $a4
- mov $_out,%r13 # borrow $a0
- mov $_ctx,%r15 # borrow $a2
- mov $_in0,%rsi # borrow $a3
- vpand $mask14,$temp,$temp
- mov $a1,$A
- vpor $temp,$iv,$iv
- vmovdqu $iv,(%r13,%r12) # write output
- lea 16(%r12),%r12 # inp++
- add $SZ*0(%r15),$A
- add $SZ*1(%r15),$B
- add $SZ*2(%r15),$C
- add $SZ*3(%r15),$D
- add $SZ*4(%r15),$E
- add $SZ*5(%r15),$F
- add $SZ*6(%r15),$G
- add $SZ*7(%r15),$H
- cmp $_end,%r12
- mov $A,$SZ*0(%r15)
- mov $B,$SZ*1(%r15)
- mov $C,$SZ*2(%r15)
- mov $D,$SZ*3(%r15)
- mov $E,$SZ*4(%r15)
- mov $F,$SZ*5(%r15)
- mov $G,$SZ*6(%r15)
- mov $H,$SZ*7(%r15)
- jb .Lloop_xop
- mov $_ivp,$ivp
- mov $_rsp,%rsi
- .cfi_def_cfa %rsi,8
- vmovdqu $iv,($ivp) # output IV
- vzeroall
- ___
- $code.=<<___ if ($win64);
- movaps `$framesz+16*0`(%rsp),%xmm6
- movaps `$framesz+16*1`(%rsp),%xmm7
- movaps `$framesz+16*2`(%rsp),%xmm8
- movaps `$framesz+16*3`(%rsp),%xmm9
- movaps `$framesz+16*4`(%rsp),%xmm10
- movaps `$framesz+16*5`(%rsp),%xmm11
- movaps `$framesz+16*6`(%rsp),%xmm12
- movaps `$framesz+16*7`(%rsp),%xmm13
- movaps `$framesz+16*8`(%rsp),%xmm14
- movaps `$framesz+16*9`(%rsp),%xmm15
- ___
- $code.=<<___;
- mov -48(%rsi),%r15
- .cfi_restore %r15
- mov -40(%rsi),%r14
- .cfi_restore %r14
- mov -32(%rsi),%r13
- .cfi_restore %r13
- mov -24(%rsi),%r12
- .cfi_restore %r12
- mov -16(%rsi),%rbp
- .cfi_restore %rbp
- mov -8(%rsi),%rbx
- .cfi_restore %rbx
- lea (%rsi),%rsp
- .cfi_def_cfa_register %rsp
- .Lepilogue_xop:
- ret
- .cfi_endproc
- .size ${func}_xop,.-${func}_xop
- ___
- ######################################################################
- # AVX+shrd code path
- #
- local *ror = sub { &shrd(@_[0],@_) };
- $code.=<<___;
- .type ${func}_avx,\@function,6
- .align 64
- ${func}_avx:
- .cfi_startproc
- .Lavx_shortcut:
- mov `($win64?56:8)`(%rsp),$in0 # load 7th parameter
- mov %rsp,%rax # copy %rsp
- .cfi_def_cfa_register %rax
- push %rbx
- .cfi_push %rbx
- push %rbp
- .cfi_push %rbp
- push %r12
- .cfi_push %r12
- push %r13
- .cfi_push %r13
- push %r14
- .cfi_push %r14
- push %r15
- .cfi_push %r15
- sub \$`$framesz+$win64*16*10`,%rsp
- and \$-64,%rsp # align stack frame
- shl \$6,$len
- sub $inp,$out # re-bias
- sub $inp,$in0
- add $inp,$len # end of input
- #mov $inp,$_inp # saved later
- mov $out,$_out
- mov $len,$_end
- #mov $key,$_key # remains resident in $inp register
- mov $ivp,$_ivp
- mov $ctx,$_ctx
- mov $in0,$_in0
- mov %rax,$_rsp
- .cfi_cfa_expression $_rsp,deref,+8
- ___
- $code.=<<___ if ($win64);
- movaps %xmm6,`$framesz+16*0`(%rsp)
- movaps %xmm7,`$framesz+16*1`(%rsp)
- movaps %xmm8,`$framesz+16*2`(%rsp)
- movaps %xmm9,`$framesz+16*3`(%rsp)
- movaps %xmm10,`$framesz+16*4`(%rsp)
- movaps %xmm11,`$framesz+16*5`(%rsp)
- movaps %xmm12,`$framesz+16*6`(%rsp)
- movaps %xmm13,`$framesz+16*7`(%rsp)
- movaps %xmm14,`$framesz+16*8`(%rsp)
- movaps %xmm15,`$framesz+16*9`(%rsp)
- ___
- $code.=<<___;
- .Lprologue_avx:
- vzeroall
- mov $inp,%r12 # borrow $a4
- lea 0x80($key),$inp # size optimization, reassign
- lea $TABLE+`$SZ*2*$rounds+32`(%rip),%r13 # borrow $a0
- mov 0xf0-0x80($inp),%r14d # rounds, borrow $a1
- mov $ctx,%r15 # borrow $a2
- mov $in0,%rsi # borrow $a3
- vmovdqu ($ivp),$iv # load IV
- sub \$9,%r14
- mov $SZ*0(%r15),$A
- mov $SZ*1(%r15),$B
- mov $SZ*2(%r15),$C
- mov $SZ*3(%r15),$D
- mov $SZ*4(%r15),$E
- mov $SZ*5(%r15),$F
- mov $SZ*6(%r15),$G
- mov $SZ*7(%r15),$H
- vmovdqa 0x00(%r13,%r14,8),$mask14
- vmovdqa 0x10(%r13,%r14,8),$mask12
- vmovdqa 0x20(%r13,%r14,8),$mask10
- vmovdqu 0x00-0x80($inp),$roundkey
- ___
- if ($SZ==4) { # SHA256
- my @X = map("%xmm$_",(0..3));
- my ($t0,$t1,$t2,$t3) = map("%xmm$_",(4..7));
- $code.=<<___;
- jmp .Lloop_avx
- .align 16
- .Lloop_avx:
- vmovdqa $TABLE+`$SZ*2*$rounds`(%rip),$t3
- vmovdqu 0x00(%rsi,%r12),@X[0]
- vmovdqu 0x10(%rsi,%r12),@X[1]
- vmovdqu 0x20(%rsi,%r12),@X[2]
- vmovdqu 0x30(%rsi,%r12),@X[3]
- vpshufb $t3,@X[0],@X[0]
- lea $TABLE(%rip),$Tbl
- vpshufb $t3,@X[1],@X[1]
- vpshufb $t3,@X[2],@X[2]
- vpaddd 0x00($Tbl),@X[0],$t0
- vpshufb $t3,@X[3],@X[3]
- vpaddd 0x20($Tbl),@X[1],$t1
- vpaddd 0x40($Tbl),@X[2],$t2
- vpaddd 0x60($Tbl),@X[3],$t3
- vmovdqa $t0,0x00(%rsp)
- mov $A,$a1
- vmovdqa $t1,0x10(%rsp)
- mov $B,$a3
- vmovdqa $t2,0x20(%rsp)
- xor $C,$a3 # magic
- vmovdqa $t3,0x30(%rsp)
- mov $E,$a0
- jmp .Lavx_00_47
- .align 16
- .Lavx_00_47:
- sub \$-16*2*$SZ,$Tbl # size optimization
- vmovdqu (%r12),$inout # $a4
- mov %r12,$_inp # $a4
- ___
- sub Xupdate_256_AVX () {
- (
- '&vpalignr ($t0,@X[1],@X[0],$SZ)', # X[1..4]
- '&vpalignr ($t3,@X[3],@X[2],$SZ)', # X[9..12]
- '&vpsrld ($t2,$t0,$sigma0[0]);',
- '&vpaddd (@X[0],@X[0],$t3)', # X[0..3] += X[9..12]
- '&vpsrld ($t3,$t0,$sigma0[2])',
- '&vpslld ($t1,$t0,8*$SZ-$sigma0[1]);',
- '&vpxor ($t0,$t3,$t2)',
- '&vpshufd ($t3,@X[3],0b11111010)',# X[14..15]
- '&vpsrld ($t2,$t2,$sigma0[1]-$sigma0[0]);',
- '&vpxor ($t0,$t0,$t1)',
- '&vpslld ($t1,$t1,$sigma0[1]-$sigma0[0]);',
- '&vpxor ($t0,$t0,$t2)',
- '&vpsrld ($t2,$t3,$sigma1[2]);',
- '&vpxor ($t0,$t0,$t1)', # sigma0(X[1..4])
- '&vpsrlq ($t3,$t3,$sigma1[0]);',
- '&vpaddd (@X[0],@X[0],$t0)', # X[0..3] += sigma0(X[1..4])
- '&vpxor ($t2,$t2,$t3);',
- '&vpsrlq ($t3,$t3,$sigma1[1]-$sigma1[0])',
- '&vpxor ($t2,$t2,$t3)', # sigma1(X[14..15])
- '&vpshufd ($t2,$t2,0b10000100)',
- '&vpsrldq ($t2,$t2,8)',
- '&vpaddd (@X[0],@X[0],$t2)', # X[0..1] += sigma1(X[14..15])
- '&vpshufd ($t3,@X[0],0b01010000)',# X[16..17]
- '&vpsrld ($t2,$t3,$sigma1[2])',
- '&vpsrlq ($t3,$t3,$sigma1[0])',
- '&vpxor ($t2,$t2,$t3);',
- '&vpsrlq ($t3,$t3,$sigma1[1]-$sigma1[0])',
- '&vpxor ($t2,$t2,$t3)',
- '&vpshufd ($t2,$t2,0b11101000)',
- '&vpslldq ($t2,$t2,8)',
- '&vpaddd (@X[0],@X[0],$t2)' # X[2..3] += sigma1(X[16..17])
- );
- }
- sub AVX_256_00_47 () {
- my $j = shift;
- my $body = shift;
- my @X = @_;
- my @insns = (&$body,&$body,&$body,&$body); # 104 instructions
- foreach (Xupdate_256_AVX()) { # 29 instructions
- eval;
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- }
- &vpaddd ($t2,@X[0],16*2*$j."($Tbl)");
- foreach (@insns) { eval; } # remaining instructions
- &vmovdqa (16*$j."(%rsp)",$t2);
- }
- $aesni_cbc_idx=0;
- for ($i=0,$j=0; $j<4; $j++) {
- &AVX_256_00_47($j,\&body_00_15,@X);
- push(@X,shift(@X)); # rotate(@X)
- }
- &mov ("%r12",$_inp); # borrow $a4
- &vpand ($temp,$temp,$mask14);
- &mov ("%r15",$_out); # borrow $a2
- &vpor ($iv,$iv,$temp);
- &vmovdqu ("(%r15,%r12)",$iv); # write output
- &lea ("%r12","16(%r12)"); # inp++
- &cmpb ($SZ-1+16*2*$SZ."($Tbl)",0);
- &jne (".Lavx_00_47");
- &vmovdqu ($inout,"(%r12)");
- &mov ($_inp,"%r12");
- $aesni_cbc_idx=0;
- for ($i=0; $i<16; ) {
- foreach(body_00_15()) { eval; }
- }
- }
- $code.=<<___;
- mov $_inp,%r12 # borrow $a4
- mov $_out,%r13 # borrow $a0
- mov $_ctx,%r15 # borrow $a2
- mov $_in0,%rsi # borrow $a3
- vpand $mask14,$temp,$temp
- mov $a1,$A
- vpor $temp,$iv,$iv
- vmovdqu $iv,(%r13,%r12) # write output
- lea 16(%r12),%r12 # inp++
- add $SZ*0(%r15),$A
- add $SZ*1(%r15),$B
- add $SZ*2(%r15),$C
- add $SZ*3(%r15),$D
- add $SZ*4(%r15),$E
- add $SZ*5(%r15),$F
- add $SZ*6(%r15),$G
- add $SZ*7(%r15),$H
- cmp $_end,%r12
- mov $A,$SZ*0(%r15)
- mov $B,$SZ*1(%r15)
- mov $C,$SZ*2(%r15)
- mov $D,$SZ*3(%r15)
- mov $E,$SZ*4(%r15)
- mov $F,$SZ*5(%r15)
- mov $G,$SZ*6(%r15)
- mov $H,$SZ*7(%r15)
- jb .Lloop_avx
- mov $_ivp,$ivp
- mov $_rsp,%rsi
- .cfi_def_cfa %rsi,8
- vmovdqu $iv,($ivp) # output IV
- vzeroall
- ___
- $code.=<<___ if ($win64);
- movaps `$framesz+16*0`(%rsp),%xmm6
- movaps `$framesz+16*1`(%rsp),%xmm7
- movaps `$framesz+16*2`(%rsp),%xmm8
- movaps `$framesz+16*3`(%rsp),%xmm9
- movaps `$framesz+16*4`(%rsp),%xmm10
- movaps `$framesz+16*5`(%rsp),%xmm11
- movaps `$framesz+16*6`(%rsp),%xmm12
- movaps `$framesz+16*7`(%rsp),%xmm13
- movaps `$framesz+16*8`(%rsp),%xmm14
- movaps `$framesz+16*9`(%rsp),%xmm15
- ___
- $code.=<<___;
- mov -48(%rsi),%r15
- .cfi_restore %r15
- mov -40(%rsi),%r14
- .cfi_restore %r14
- mov -32(%rsi),%r13
- .cfi_restore %r13
- mov -24(%rsi),%r12
- .cfi_restore %r12
- mov -16(%rsi),%rbp
- .cfi_restore %rbp
- mov -8(%rsi),%rbx
- .cfi_restore %rbx
- lea (%rsi),%rsp
- .cfi_def_cfa_register %rsp
- .Lepilogue_avx:
- ret
- .cfi_endproc
- .size ${func}_avx,.-${func}_avx
- ___
- if ($avx>1) {{
- ######################################################################
- # AVX2+BMI code path
- #
- my $a5=$SZ==4?"%esi":"%rsi"; # zap $inp
- my $PUSH8=8*2*$SZ;
- use integer;
- sub bodyx_00_15 () {
- # at start $a1 should be zero, $a3 - $b^$c and $a4 copy of $f
- (
- '($a,$b,$c,$d,$e,$f,$g,$h)=@ROT;'.
- '&add ($h,(32*($i/(16/$SZ))+$SZ*($i%(16/$SZ)))%$PUSH8.$base)', # h+=X[i]+K[i]
- '&and ($a4,$e)', # f&e
- '&rorx ($a0,$e,$Sigma1[2])',
- '&rorx ($a2,$e,$Sigma1[1])',
- '&lea ($a,"($a,$a1)")', # h+=Sigma0(a) from the past
- '&lea ($h,"($h,$a4)")',
- '&andn ($a4,$e,$g)', # ~e&g
- '&xor ($a0,$a2)',
- '&rorx ($a1,$e,$Sigma1[0])',
- '&lea ($h,"($h,$a4)")', # h+=Ch(e,f,g)=(e&f)+(~e&g)
- '&xor ($a0,$a1)', # Sigma1(e)
- '&mov ($a2,$a)',
- '&rorx ($a4,$a,$Sigma0[2])',
- '&lea ($h,"($h,$a0)")', # h+=Sigma1(e)
- '&xor ($a2,$b)', # a^b, b^c in next round
- '&rorx ($a1,$a,$Sigma0[1])',
- '&rorx ($a0,$a,$Sigma0[0])',
- '&lea ($d,"($d,$h)")', # d+=h
- '&and ($a3,$a2)', # (b^c)&(a^b)
- @aesni_cbc_block[$aesni_cbc_idx++].
- '&xor ($a1,$a4)',
- '&xor ($a3,$b)', # Maj(a,b,c)=Ch(a^b,c,b)
- '&xor ($a1,$a0)', # Sigma0(a)
- '&lea ($h,"($h,$a3)");'. # h+=Maj(a,b,c)
- '&mov ($a4,$e)', # copy of f in future
- '($a2,$a3) = ($a3,$a2); unshift(@ROT,pop(@ROT)); $i++;'
- );
- # and at the finish one has to $a+=$a1
- }
- $code.=<<___;
- .type ${func}_avx2,\@function,6
- .align 64
- ${func}_avx2:
- .cfi_startproc
- .Lavx2_shortcut:
- mov `($win64?56:8)`(%rsp),$in0 # load 7th parameter
- mov %rsp,%rax # copy %rsp
- .cfi_def_cfa_register %rax
- push %rbx
- .cfi_push %rbx
- push %rbp
- .cfi_push %rbp
- push %r12
- .cfi_push %r12
- push %r13
- .cfi_push %r13
- push %r14
- .cfi_push %r14
- push %r15
- .cfi_push %r15
- sub \$`2*$SZ*$rounds+8*8+$win64*16*10`,%rsp
- and \$-256*$SZ,%rsp # align stack frame
- add \$`2*$SZ*($rounds-8)`,%rsp
- shl \$6,$len
- sub $inp,$out # re-bias
- sub $inp,$in0
- add $inp,$len # end of input
- #mov $inp,$_inp # saved later
- #mov $out,$_out # kept in $offload
- mov $len,$_end
- #mov $key,$_key # remains resident in $inp register
- mov $ivp,$_ivp
- mov $ctx,$_ctx
- mov $in0,$_in0
- mov %rax,$_rsp
- .cfi_cfa_expression $_rsp,deref,+8
- ___
- $code.=<<___ if ($win64);
- movaps %xmm6,`$framesz+16*0`(%rsp)
- movaps %xmm7,`$framesz+16*1`(%rsp)
- movaps %xmm8,`$framesz+16*2`(%rsp)
- movaps %xmm9,`$framesz+16*3`(%rsp)
- movaps %xmm10,`$framesz+16*4`(%rsp)
- movaps %xmm11,`$framesz+16*5`(%rsp)
- movaps %xmm12,`$framesz+16*6`(%rsp)
- movaps %xmm13,`$framesz+16*7`(%rsp)
- movaps %xmm14,`$framesz+16*8`(%rsp)
- movaps %xmm15,`$framesz+16*9`(%rsp)
- ___
- $code.=<<___;
- .Lprologue_avx2:
- vzeroall
- mov $inp,%r13 # borrow $a0
- vpinsrq \$1,$out,$offload,$offload
- lea 0x80($key),$inp # size optimization, reassign
- lea $TABLE+`$SZ*2*$rounds+32`(%rip),%r12 # borrow $a4
- mov 0xf0-0x80($inp),%r14d # rounds, borrow $a1
- mov $ctx,%r15 # borrow $a2
- mov $in0,%rsi # borrow $a3
- vmovdqu ($ivp),$iv # load IV
- lea -9(%r14),%r14
- vmovdqa 0x00(%r12,%r14,8),$mask14
- vmovdqa 0x10(%r12,%r14,8),$mask12
- vmovdqa 0x20(%r12,%r14,8),$mask10
- sub \$-16*$SZ,%r13 # inp++, size optimization
- mov $SZ*0(%r15),$A
- lea (%rsi,%r13),%r12 # borrow $a0
- mov $SZ*1(%r15),$B
- cmp $len,%r13 # $_end
- mov $SZ*2(%r15),$C
- cmove %rsp,%r12 # next block or random data
- mov $SZ*3(%r15),$D
- mov $SZ*4(%r15),$E
- mov $SZ*5(%r15),$F
- mov $SZ*6(%r15),$G
- mov $SZ*7(%r15),$H
- vmovdqu 0x00-0x80($inp),$roundkey
- ___
- if ($SZ==4) { # SHA256
- my @X = map("%ymm$_",(0..3));
- my ($t0,$t1,$t2,$t3) = map("%ymm$_",(4..7));
- $code.=<<___;
- jmp .Loop_avx2
- .align 16
- .Loop_avx2:
- vmovdqa $TABLE+`$SZ*2*$rounds`(%rip),$t3
- vmovdqu -16*$SZ+0(%rsi,%r13),%xmm0
- vmovdqu -16*$SZ+16(%rsi,%r13),%xmm1
- vmovdqu -16*$SZ+32(%rsi,%r13),%xmm2
- vmovdqu -16*$SZ+48(%rsi,%r13),%xmm3
- vinserti128 \$1,(%r12),@X[0],@X[0]
- vinserti128 \$1,16(%r12),@X[1],@X[1]
- vpshufb $t3,@X[0],@X[0]
- vinserti128 \$1,32(%r12),@X[2],@X[2]
- vpshufb $t3,@X[1],@X[1]
- vinserti128 \$1,48(%r12),@X[3],@X[3]
- lea $TABLE(%rip),$Tbl
- vpshufb $t3,@X[2],@X[2]
- lea -16*$SZ(%r13),%r13
- vpaddd 0x00($Tbl),@X[0],$t0
- vpshufb $t3,@X[3],@X[3]
- vpaddd 0x20($Tbl),@X[1],$t1
- vpaddd 0x40($Tbl),@X[2],$t2
- vpaddd 0x60($Tbl),@X[3],$t3
- vmovdqa $t0,0x00(%rsp)
- xor $a1,$a1
- vmovdqa $t1,0x20(%rsp)
- ___
- $code.=<<___ if (!$win64);
- # temporarily use %rsi as frame pointer
- mov $_rsp,%rsi
- .cfi_def_cfa %rsi,8
- ___
- $code.=<<___;
- lea -$PUSH8(%rsp),%rsp
- ___
- $code.=<<___ if (!$win64);
- # the frame info is at $_rsp, but the stack is moving...
- # so a second frame pointer is saved at -8(%rsp)
- # that is in the red zone
- mov %rsi,-8(%rsp)
- .cfi_cfa_expression %rsp-8,deref,+8
- ___
- $code.=<<___;
- mov $B,$a3
- vmovdqa $t2,0x00(%rsp)
- xor $C,$a3 # magic
- vmovdqa $t3,0x20(%rsp)
- mov $F,$a4
- sub \$-16*2*$SZ,$Tbl # size optimization
- jmp .Lavx2_00_47
- .align 16
- .Lavx2_00_47:
- vmovdqu (%r13),$inout
- vpinsrq \$0,%r13,$offload,$offload
- ___
- sub AVX2_256_00_47 () {
- my $j = shift;
- my $body = shift;
- my @X = @_;
- my @insns = (&$body,&$body,&$body,&$body); # 96 instructions
- my $base = "+2*$PUSH8(%rsp)";
- if (($j%2)==0) {
- &lea ("%rsp","-$PUSH8(%rsp)");
- $code.=<<___ if (!$win64);
- .cfi_cfa_expression %rsp+`$PUSH8-8`,deref,+8
- # copy secondary frame pointer to new location again at -8(%rsp)
- pushq $PUSH8-8(%rsp)
- .cfi_cfa_expression %rsp,deref,+8
- lea 8(%rsp),%rsp
- .cfi_cfa_expression %rsp-8,deref,+8
- ___
- }
- foreach (Xupdate_256_AVX()) { # 29 instructions
- eval;
- eval(shift(@insns));
- eval(shift(@insns));
- eval(shift(@insns));
- }
- &vpaddd ($t2,@X[0],16*2*$j."($Tbl)");
- foreach (@insns) { eval; } # remaining instructions
- &vmovdqa ((32*$j)%$PUSH8."(%rsp)",$t2);
- }
- $aesni_cbc_idx=0;
- for ($i=0,$j=0; $j<4; $j++) {
- &AVX2_256_00_47($j,\&bodyx_00_15,@X);
- push(@X,shift(@X)); # rotate(@X)
- }
- &vmovq ("%r13",$offload); # borrow $a0
- &vpextrq ("%r15",$offload,1); # borrow $a2
- &vpand ($temp,$temp,$mask14);
- &vpor ($iv,$iv,$temp);
- &vmovdqu ("(%r15,%r13)",$iv); # write output
- &lea ("%r13","16(%r13)"); # inp++
- &lea ($Tbl,16*2*$SZ."($Tbl)");
- &cmpb (($SZ-1)."($Tbl)",0);
- &jne (".Lavx2_00_47");
- &vmovdqu ($inout,"(%r13)");
- &vpinsrq ($offload,$offload,"%r13",0);
- $aesni_cbc_idx=0;
- for ($i=0; $i<16; ) {
- my $base=$i<8?"+$PUSH8(%rsp)":"(%rsp)";
- foreach(bodyx_00_15()) { eval; }
- }
- }
- $code.=<<___;
- vpextrq \$1,$offload,%r12 # $_out, borrow $a4
- vmovq $offload,%r13 # $_inp, borrow $a0
- mov `2*$SZ*$rounds+5*8`(%rsp),%r15 # $_ctx, borrow $a2
- add $a1,$A
- lea `2*$SZ*($rounds-8)`(%rsp),$Tbl
- vpand $mask14,$temp,$temp
- vpor $temp,$iv,$iv
- vmovdqu $iv,(%r12,%r13) # write output
- lea 16(%r13),%r13
- add $SZ*0(%r15),$A
- add $SZ*1(%r15),$B
- add $SZ*2(%r15),$C
- add $SZ*3(%r15),$D
- add $SZ*4(%r15),$E
- add $SZ*5(%r15),$F
- add $SZ*6(%r15),$G
- add $SZ*7(%r15),$H
- mov $A,$SZ*0(%r15)
- mov $B,$SZ*1(%r15)
- mov $C,$SZ*2(%r15)
- mov $D,$SZ*3(%r15)
- mov $E,$SZ*4(%r15)
- mov $F,$SZ*5(%r15)
- mov $G,$SZ*6(%r15)
- mov $H,$SZ*7(%r15)
- cmp `$PUSH8+2*8`($Tbl),%r13 # $_end
- je .Ldone_avx2
- xor $a1,$a1
- mov $B,$a3
- mov $F,$a4
- xor $C,$a3 # magic
- jmp .Lower_avx2
- .align 16
- .Lower_avx2:
- vmovdqu (%r13),$inout
- vpinsrq \$0,%r13,$offload,$offload
- ___
- $aesni_cbc_idx=0;
- for ($i=0; $i<16; ) {
- my $base="+16($Tbl)";
- foreach(bodyx_00_15()) { eval; }
- &lea ($Tbl,"-$PUSH8($Tbl)") if ($i==8);
- }
- $code.=<<___;
- vmovq $offload,%r13 # borrow $a0
- vpextrq \$1,$offload,%r15 # borrow $a2
- vpand $mask14,$temp,$temp
- vpor $temp,$iv,$iv
- lea -$PUSH8($Tbl),$Tbl
- vmovdqu $iv,(%r15,%r13) # write output
- lea 16(%r13),%r13 # inp++
- cmp %rsp,$Tbl
- jae .Lower_avx2
- mov `2*$SZ*$rounds+5*8`(%rsp),%r15 # $_ctx, borrow $a2
- lea 16*$SZ(%r13),%r13
- mov `2*$SZ*$rounds+6*8`(%rsp),%rsi # $_in0, borrow $a3
- add $a1,$A
- lea `2*$SZ*($rounds-8)`(%rsp),%rsp
- add $SZ*0(%r15),$A
- add $SZ*1(%r15),$B
- add $SZ*2(%r15),$C
- add $SZ*3(%r15),$D
- add $SZ*4(%r15),$E
- add $SZ*5(%r15),$F
- add $SZ*6(%r15),$G
- lea (%rsi,%r13),%r12
- add $SZ*7(%r15),$H
- cmp $_end,%r13
- mov $A,$SZ*0(%r15)
- cmove %rsp,%r12 # next block or stale data
- mov $B,$SZ*1(%r15)
- mov $C,$SZ*2(%r15)
- mov $D,$SZ*3(%r15)
- mov $E,$SZ*4(%r15)
- mov $F,$SZ*5(%r15)
- mov $G,$SZ*6(%r15)
- mov $H,$SZ*7(%r15)
- jbe .Loop_avx2
- lea (%rsp),$Tbl
- # temporarily use $Tbl as index to $_rsp
- # this avoids the need to save a secondary frame pointer at -8(%rsp)
- .cfi_cfa_expression $Tbl+`16*$SZ+7*8`,deref,+8
- .Ldone_avx2:
- mov 16*$SZ+4*8($Tbl),$ivp
- mov 16*$SZ+7*8($Tbl),%rsi
- .cfi_def_cfa %rsi,8
- vmovdqu $iv,($ivp) # output IV
- vzeroall
- ___
- $code.=<<___ if ($win64);
- movaps `$framesz+16*0`($Tbl),%xmm6
- movaps `$framesz+16*1`($Tbl),%xmm7
- movaps `$framesz+16*2`($Tbl),%xmm8
- movaps `$framesz+16*3`($Tbl),%xmm9
- movaps `$framesz+16*4`($Tbl),%xmm10
- movaps `$framesz+16*5`($Tbl),%xmm11
- movaps `$framesz+16*6`($Tbl),%xmm12
- movaps `$framesz+16*7`($Tbl),%xmm13
- movaps `$framesz+16*8`($Tbl),%xmm14
- movaps `$framesz+16*9`($Tbl),%xmm15
- ___
- $code.=<<___;
- mov -48(%rsi),%r15
- .cfi_restore %r15
- mov -40(%rsi),%r14
- .cfi_restore %r14
- mov -32(%rsi),%r13
- .cfi_restore %r13
- mov -24(%rsi),%r12
- .cfi_restore %r12
- mov -16(%rsi),%rbp
- .cfi_restore %rbp
- mov -8(%rsi),%rbx
- .cfi_restore %rbx
- lea (%rsi),%rsp
- .cfi_def_cfa_register %rsp
- .Lepilogue_avx2:
- ret
- .cfi_endproc
- .size ${func}_avx2,.-${func}_avx2
- ___
- }}
- }}
- {{
- my ($in0,$out,$len,$key,$ivp,$ctx,$inp)=("%rdi","%rsi","%rdx","%rcx","%r8","%r9","%r10");
- my ($rounds,$Tbl)=("%r11d","%rbx");
- my ($iv,$in,$rndkey0)=map("%xmm$_",(6,14,15));
- my @rndkey=("%xmm4","%xmm5");
- my $r=0;
- my $sn=0;
- my ($Wi,$ABEF,$CDGH,$TMP,$BSWAP,$ABEF_SAVE,$CDGH_SAVE)=map("%xmm$_",(0..3,7..9));
- my @MSG=map("%xmm$_",(10..13));
- my $aesenc=sub {
- use integer;
- my ($n,$k)=($r/10,$r%10);
- if ($k==0) {
- $code.=<<___;
- movups `16*$n`($in0),$in # load input
- xorps $rndkey0,$in
- ___
- $code.=<<___ if ($n);
- movups $iv,`16*($n-1)`($out,$in0) # write output
- ___
- $code.=<<___;
- xorps $in,$iv
- movups `32+16*$k-112`($key),$rndkey[1]
- aesenc $rndkey[0],$iv
- ___
- } elsif ($k==9) {
- $sn++;
- $code.=<<___;
- cmp \$11,$rounds
- jb .Laesenclast$sn
- movups `32+16*($k+0)-112`($key),$rndkey[1]
- aesenc $rndkey[0],$iv
- movups `32+16*($k+1)-112`($key),$rndkey[0]
- aesenc $rndkey[1],$iv
- je .Laesenclast$sn
- movups `32+16*($k+2)-112`($key),$rndkey[1]
- aesenc $rndkey[0],$iv
- movups `32+16*($k+3)-112`($key),$rndkey[0]
- aesenc $rndkey[1],$iv
- .Laesenclast$sn:
- aesenclast $rndkey[0],$iv
- movups 16-112($key),$rndkey[1] # forward reference
- nop
- ___
- } else {
- $code.=<<___;
- movups `32+16*$k-112`($key),$rndkey[1]
- aesenc $rndkey[0],$iv
- ___
- }
- $r++; unshift(@rndkey,pop(@rndkey));
- };
- if ($shaext) {
- my $Tbl="%rax";
- $code.=<<___;
- .type ${func}_shaext,\@function,6
- .align 32
- ${func}_shaext:
- .cfi_startproc
- mov `($win64?56:8)`(%rsp),$inp # load 7th argument
- ___
- $code.=<<___ if ($win64);
- lea `-8-10*16`(%rsp),%rsp
- movaps %xmm6,-8-10*16(%rax)
- movaps %xmm7,-8-9*16(%rax)
- movaps %xmm8,-8-8*16(%rax)
- movaps %xmm9,-8-7*16(%rax)
- movaps %xmm10,-8-6*16(%rax)
- movaps %xmm11,-8-5*16(%rax)
- movaps %xmm12,-8-4*16(%rax)
- movaps %xmm13,-8-3*16(%rax)
- movaps %xmm14,-8-2*16(%rax)
- movaps %xmm15,-8-1*16(%rax)
- .Lprologue_shaext:
- ___
- $code.=<<___;
- lea K256+0x80(%rip),$Tbl
- movdqu ($ctx),$ABEF # DCBA
- movdqu 16($ctx),$CDGH # HGFE
- movdqa 0x200-0x80($Tbl),$TMP # byte swap mask
- mov 240($key),$rounds
- sub $in0,$out
- movups ($key),$rndkey0 # $key[0]
- movups ($ivp),$iv # load IV
- movups 16($key),$rndkey[0] # forward reference
- lea 112($key),$key # size optimization
- pshufd \$0x1b,$ABEF,$Wi # ABCD
- pshufd \$0xb1,$ABEF,$ABEF # CDAB
- pshufd \$0x1b,$CDGH,$CDGH # EFGH
- movdqa $TMP,$BSWAP # offload
- palignr \$8,$CDGH,$ABEF # ABEF
- punpcklqdq $Wi,$CDGH # CDGH
- jmp .Loop_shaext
- .align 16
- .Loop_shaext:
- movdqu ($inp),@MSG[0]
- movdqu 0x10($inp),@MSG[1]
- movdqu 0x20($inp),@MSG[2]
- pshufb $TMP,@MSG[0]
- movdqu 0x30($inp),@MSG[3]
- movdqa 0*32-0x80($Tbl),$Wi
- paddd @MSG[0],$Wi
- pshufb $TMP,@MSG[1]
- movdqa $CDGH,$CDGH_SAVE # offload
- movdqa $ABEF,$ABEF_SAVE # offload
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $ABEF,$CDGH # 0-3
- pshufd \$0x0e,$Wi,$Wi
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $CDGH,$ABEF
- movdqa 1*32-0x80($Tbl),$Wi
- paddd @MSG[1],$Wi
- pshufb $TMP,@MSG[2]
- lea 0x40($inp),$inp
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $ABEF,$CDGH # 4-7
- pshufd \$0x0e,$Wi,$Wi
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $CDGH,$ABEF
- movdqa 2*32-0x80($Tbl),$Wi
- paddd @MSG[2],$Wi
- pshufb $TMP,@MSG[3]
- sha256msg1 @MSG[1],@MSG[0]
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $ABEF,$CDGH # 8-11
- pshufd \$0x0e,$Wi,$Wi
- movdqa @MSG[3],$TMP
- palignr \$4,@MSG[2],$TMP
- paddd $TMP,@MSG[0]
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $CDGH,$ABEF
- movdqa 3*32-0x80($Tbl),$Wi
- paddd @MSG[3],$Wi
- sha256msg2 @MSG[3],@MSG[0]
- sha256msg1 @MSG[2],@MSG[1]
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $ABEF,$CDGH # 12-15
- pshufd \$0x0e,$Wi,$Wi
- ___
- &$aesenc();
- $code.=<<___;
- movdqa @MSG[0],$TMP
- palignr \$4,@MSG[3],$TMP
- paddd $TMP,@MSG[1]
- sha256rnds2 $CDGH,$ABEF
- ___
- for($i=4;$i<16-3;$i++) {
- &$aesenc() if (($r%10)==0);
- $code.=<<___;
- movdqa $i*32-0x80($Tbl),$Wi
- paddd @MSG[0],$Wi
- sha256msg2 @MSG[0],@MSG[1]
- sha256msg1 @MSG[3],@MSG[2]
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $ABEF,$CDGH # 16-19...
- pshufd \$0x0e,$Wi,$Wi
- movdqa @MSG[1],$TMP
- palignr \$4,@MSG[0],$TMP
- paddd $TMP,@MSG[2]
- ___
- &$aesenc();
- &$aesenc() if ($r==19);
- $code.=<<___;
- sha256rnds2 $CDGH,$ABEF
- ___
- push(@MSG,shift(@MSG));
- }
- $code.=<<___;
- movdqa 13*32-0x80($Tbl),$Wi
- paddd @MSG[0],$Wi
- sha256msg2 @MSG[0],@MSG[1]
- sha256msg1 @MSG[3],@MSG[2]
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $ABEF,$CDGH # 52-55
- pshufd \$0x0e,$Wi,$Wi
- movdqa @MSG[1],$TMP
- palignr \$4,@MSG[0],$TMP
- paddd $TMP,@MSG[2]
- ___
- &$aesenc();
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $CDGH,$ABEF
- movdqa 14*32-0x80($Tbl),$Wi
- paddd @MSG[1],$Wi
- sha256msg2 @MSG[1],@MSG[2]
- movdqa $BSWAP,$TMP
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $ABEF,$CDGH # 56-59
- pshufd \$0x0e,$Wi,$Wi
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $CDGH,$ABEF
- movdqa 15*32-0x80($Tbl),$Wi
- paddd @MSG[2],$Wi
- ___
- &$aesenc();
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $ABEF,$CDGH # 60-63
- pshufd \$0x0e,$Wi,$Wi
- ___
- &$aesenc();
- $code.=<<___;
- sha256rnds2 $CDGH,$ABEF
- #pxor $CDGH,$rndkey0 # black magic
- ___
- while ($r<40) { &$aesenc(); } # remaining aesenc's
- $code.=<<___;
- #xorps $CDGH,$rndkey0 # black magic
- paddd $CDGH_SAVE,$CDGH
- paddd $ABEF_SAVE,$ABEF
- dec $len
- movups $iv,48($out,$in0) # write output
- lea 64($in0),$in0
- jnz .Loop_shaext
- pshufd \$0xb1,$CDGH,$CDGH # DCHG
- pshufd \$0x1b,$ABEF,$TMP # FEBA
- pshufd \$0xb1,$ABEF,$ABEF # BAFE
- punpckhqdq $CDGH,$ABEF # DCBA
- palignr \$8,$TMP,$CDGH # HGFE
- movups $iv,($ivp) # write IV
- movdqu $ABEF,($ctx)
- movdqu $CDGH,16($ctx)
- ___
- $code.=<<___ if ($win64);
- movaps 0*16(%rsp),%xmm6
- movaps 1*16(%rsp),%xmm7
- movaps 2*16(%rsp),%xmm8
- movaps 3*16(%rsp),%xmm9
- movaps 4*16(%rsp),%xmm10
- movaps 5*16(%rsp),%xmm11
- movaps 6*16(%rsp),%xmm12
- movaps 7*16(%rsp),%xmm13
- movaps 8*16(%rsp),%xmm14
- movaps 9*16(%rsp),%xmm15
- lea 8+10*16(%rsp),%rsp
- .Lepilogue_shaext:
- ___
- $code.=<<___;
- ret
- .cfi_endproc
- .size ${func}_shaext,.-${func}_shaext
- ___
- }
- }}}}}
- # EXCEPTION_DISPOSITION handler (EXCEPTION_RECORD *rec,ULONG64 frame,
- # CONTEXT *context,DISPATCHER_CONTEXT *disp)
- if ($win64 && $avx) {
- $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
- mov 8($disp),%rsi # disp->ImageBase
- mov 56($disp),%r11 # disp->HanderlData
- mov 0(%r11),%r10d # HandlerData[0]
- lea (%rsi,%r10),%r10 # prologue label
- cmp %r10,%rbx # context->Rip<prologue label
- jb .Lin_prologue
- mov 152($context),%rax # pull context->Rsp
- mov 4(%r11),%r10d # HandlerData[1]
- lea (%rsi,%r10),%r10 # epilogue label
- cmp %r10,%rbx # context->Rip>=epilogue label
- jae .Lin_prologue
- ___
- $code.=<<___ if ($shaext);
- lea aesni_cbc_sha256_enc_shaext(%rip),%r10
- cmp %r10,%rbx
- jb .Lnot_in_shaext
- lea (%rax),%rsi
- lea 512($context),%rdi # &context.Xmm6
- mov \$20,%ecx
- .long 0xa548f3fc # cld; rep movsq
- lea 168(%rax),%rax # adjust stack pointer
- jmp .Lin_prologue
- .Lnot_in_shaext:
- ___
- $code.=<<___ if ($avx>1);
- lea .Lavx2_shortcut(%rip),%r10
- cmp %r10,%rbx # context->Rip<avx2_shortcut
- jb .Lnot_in_avx2
- and \$-256*$SZ,%rax
- add \$`2*$SZ*($rounds-8)`,%rax
- .Lnot_in_avx2:
- ___
- $code.=<<___;
- mov %rax,%rsi # put aside Rsp
- mov 16*$SZ+7*8(%rax),%rax # pull $_rsp
- 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
- lea 16*$SZ+8*8(%rsi),%rsi # Xmm6- save area
- lea 512($context),%rdi # &context.Xmm6
- mov \$20,%ecx
- .long 0xa548f3fc # cld; rep movsq
- .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
- .rva .LSEH_begin_${func}_xop
- .rva .LSEH_end_${func}_xop
- .rva .LSEH_info_${func}_xop
- .rva .LSEH_begin_${func}_avx
- .rva .LSEH_end_${func}_avx
- .rva .LSEH_info_${func}_avx
- ___
- $code.=<<___ if ($avx>1);
- .rva .LSEH_begin_${func}_avx2
- .rva .LSEH_end_${func}_avx2
- .rva .LSEH_info_${func}_avx2
- ___
- $code.=<<___ if ($shaext);
- .rva .LSEH_begin_${func}_shaext
- .rva .LSEH_end_${func}_shaext
- .rva .LSEH_info_${func}_shaext
- ___
- $code.=<<___;
- .section .xdata
- .align 8
- .LSEH_info_${func}_xop:
- .byte 9,0,0,0
- .rva se_handler
- .rva .Lprologue_xop,.Lepilogue_xop # HandlerData[]
- .LSEH_info_${func}_avx:
- .byte 9,0,0,0
- .rva se_handler
- .rva .Lprologue_avx,.Lepilogue_avx # HandlerData[]
- ___
- $code.=<<___ if ($avx>1);
- .LSEH_info_${func}_avx2:
- .byte 9,0,0,0
- .rva se_handler
- .rva .Lprologue_avx2,.Lepilogue_avx2 # HandlerData[]
- ___
- $code.=<<___ if ($shaext);
- .LSEH_info_${func}_shaext:
- .byte 9,0,0,0
- .rva se_handler
- .rva .Lprologue_shaext,.Lepilogue_shaext # HandlerData[]
- ___
- }
- ####################################################################
- sub rex {
- local *opcode=shift;
- my ($dst,$src)=@_;
- my $rex=0;
- $rex|=0x04 if($dst>=8);
- $rex|=0x01 if($src>=8);
- unshift @opcode,$rex|0x40 if($rex);
- }
- {
- my %opcodelet = (
- "sha256rnds2" => 0xcb,
- "sha256msg1" => 0xcc,
- "sha256msg2" => 0xcd );
- sub sha256op38 {
- my $instr = shift;
- if (defined($opcodelet{$instr}) && @_[0] =~ /%xmm([0-9]+),\s*%xmm([0-9]+)/) {
- my @opcode=(0x0f,0x38);
- rex(\@opcode,$2,$1);
- push @opcode,$opcodelet{$instr};
- push @opcode,0xc0|($1&7)|(($2&7)<<3); # ModR/M
- return ".byte\t".join(',',@opcode);
- } else {
- return $instr."\t".@_[0];
- }
- }
- }
- $code =~ s/\`([^\`]*)\`/eval $1/gem;
- $code =~ s/\b(sha256[^\s]*)\s+(.*)/sha256op38($1,$2)/gem;
- print $code;
- close STDOUT or die "error closing STDOUT: $!";
|