3666 lines
88 KiB
Perl
3666 lines
88 KiB
Perl
|
#! /usr/bin/env perl
|
||
|
|
# Copyright 2016 The OpenSSL Project Authors. All Rights Reserved.
|
||
|
|
#
|
||
|
|
# Licensed under the OpenSSL license (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/.
|
||
|
|
# ====================================================================
|
||
|
|
#
|
||
|
|
# November 2014
|
||
|
|
#
|
||
|
|
# chacha20 for x86_64.
|
||
|
|
#
|
||
|
|
# December 2016
|
||
|
|
#
|
||
|
|
# Add AVX512F code path.
|
||
|
|
#
|
||
|
|
# December 2017
|
||
|
|
#
|
||
|
|
# Add AVX512VL code path.
|
||
|
|
#
|
||
|
|
# Performance in cycles per byte out of large buffer.
|
||
|
|
#
|
||
|
|
# IALU/gcc 4.8(i) 1xSSSE3/SSE2 4xSSSE3 NxAVX(v)
|
||
|
|
#
|
||
|
|
# P4 9.48/+99% -/22.7(ii) -
|
||
|
|
# Core2 7.83/+55% 7.90/8.08 4.35
|
||
|
|
# Westmere 7.19/+50% 5.60/6.70 3.00
|
||
|
|
# Sandy Bridge 8.31/+42% 5.45/6.76 2.72
|
||
|
|
# Ivy Bridge 6.71/+46% 5.40/6.49 2.41
|
||
|
|
# Haswell 5.92/+43% 5.20/6.45 2.42 1.23
|
||
|
|
# Skylake[-X] 5.87/+39% 4.70/- 2.31 1.19[0.80(vi)]
|
||
|
|
# Silvermont 12.0/+33% 7.75/7.40 7.03(iii)
|
||
|
|
# Knights L 11.7/- - 9.60(iii) 0.80
|
||
|
|
# Goldmont 10.6/+17% 5.10/- 3.28
|
||
|
|
# Sledgehammer 7.28/+52% -/14.2(ii) -
|
||
|
|
# Bulldozer 9.66/+28% 9.85/11.1 3.06(iv)
|
||
|
|
# Ryzen 5.96/+50% 5.19/- 2.40 2.09
|
||
|
|
# VIA Nano 10.5/+46% 6.72/8.60 6.05
|
||
|
|
#
|
||
|
|
# (i) compared to older gcc 3.x one can observe >2x improvement on
|
||
|
|
# most platforms;
|
||
|
|
# (ii) as it can be seen, SSE2 performance is too low on legacy
|
||
|
|
# processors; NxSSE2 results are naturally better, but not
|
||
|
|
# impressively better than IALU ones, which is why you won't
|
||
|
|
# find SSE2 code below;
|
||
|
|
# (iii) this is not optimal result for Atom because of MSROM
|
||
|
|
# limitations, SSE2 can do better, but gain is considered too
|
||
|
|
# low to justify the [maintenance] effort;
|
||
|
|
# (iv) Bulldozer actually executes 4xXOP code path that delivers 2.20;
|
||
|
|
# (v) 8xAVX2, 8xAVX512VL or 16xAVX512F, whichever best applicable;
|
||
|
|
# (vi) even though Skylake-X can execute AVX512F code and deliver 0.57
|
||
|
|
# cpb in single thread, the corresponding capability is suppressed;
|
||
|
|
|
||
|
|
$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";
|
||
|
|
|
||
|
|
$avx = 3;
|
||
|
|
$avx = 2 if ($flavour =~ /macosx/);
|
||
|
|
|
||
|
|
open OUT,"| \"$^X\" \"$xlate\" $flavour \"$output\"";
|
||
|
|
*STDOUT=*OUT;
|
||
|
|
|
||
|
|
# input parameter block
|
||
|
|
($out,$inp,$len,$key,$counter)=("%rdi","%rsi","%rdx","%rcx","%r8");
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.text
|
||
|
|
|
||
|
|
.align 64
|
||
|
|
.Lzero:
|
||
|
|
.long 0,0,0,0
|
||
|
|
.Lone:
|
||
|
|
.long 1,0,0,0
|
||
|
|
.Linc:
|
||
|
|
.long 0,1,2,3
|
||
|
|
.Lfour:
|
||
|
|
.long 4,4,4,4
|
||
|
|
.Lincy:
|
||
|
|
.long 0,2,4,6,1,3,5,7
|
||
|
|
.Leight:
|
||
|
|
.long 8,8,8,8,8,8,8,8
|
||
|
|
.Lrot16:
|
||
|
|
.byte 0x2,0x3,0x0,0x1, 0x6,0x7,0x4,0x5, 0xa,0xb,0x8,0x9, 0xe,0xf,0xc,0xd
|
||
|
|
.Lrot24:
|
||
|
|
.byte 0x3,0x0,0x1,0x2, 0x7,0x4,0x5,0x6, 0xb,0x8,0x9,0xa, 0xf,0xc,0xd,0xe
|
||
|
|
.Lsigma:
|
||
|
|
.asciz "expand 32-byte k"
|
||
|
|
.align 64
|
||
|
|
.Lzeroz:
|
||
|
|
.long 0,0,0,0, 1,0,0,0, 2,0,0,0, 3,0,0,0
|
||
|
|
.Lfourz:
|
||
|
|
.long 4,0,0,0, 4,0,0,0, 4,0,0,0, 4,0,0,0
|
||
|
|
.Lincz:
|
||
|
|
.long 0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15
|
||
|
|
.Lsixteen:
|
||
|
|
.long 16,16,16,16,16,16,16,16,16,16,16,16,16,16,16,16
|
||
|
|
.align 64
|
||
|
|
.Ltwoy:
|
||
|
|
.long 2,0,0,0, 2,0,0,0
|
||
|
|
___
|
||
|
|
|
||
|
|
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";
|
||
|
|
}
|
||
|
|
|
||
|
|
@x=("%eax","%ebx","%ecx","%edx",map("%r${_}d",(8..11)),
|
||
|
|
"%nox","%nox","%nox","%nox",map("%r${_}d",(12..15)));
|
||
|
|
@t=("%esi","%edi");
|
||
|
|
|
||
|
|
sub ROUND { # critical path is 24 cycles per round
|
||
|
|
my ($a0,$b0,$c0,$d0)=@_;
|
||
|
|
my ($a1,$b1,$c1,$d1)=map(($_&~3)+(($_+1)&3),($a0,$b0,$c0,$d0));
|
||
|
|
my ($a2,$b2,$c2,$d2)=map(($_&~3)+(($_+1)&3),($a1,$b1,$c1,$d1));
|
||
|
|
my ($a3,$b3,$c3,$d3)=map(($_&~3)+(($_+1)&3),($a2,$b2,$c2,$d2));
|
||
|
|
my ($xc,$xc_)=map("\"$_\"",@t);
|
||
|
|
my @x=map("\"$_\"",@x);
|
||
|
|
|
||
|
|
# Consider order in which variables are addressed by their
|
||
|
|
# index:
|
||
|
|
#
|
||
|
|
# a b c d
|
||
|
|
#
|
||
|
|
# 0 4 8 12 < even round
|
||
|
|
# 1 5 9 13
|
||
|
|
# 2 6 10 14
|
||
|
|
# 3 7 11 15
|
||
|
|
# 0 5 10 15 < odd round
|
||
|
|
# 1 6 11 12
|
||
|
|
# 2 7 8 13
|
||
|
|
# 3 4 9 14
|
||
|
|
#
|
||
|
|
# 'a', 'b' and 'd's are permanently allocated in registers,
|
||
|
|
# @x[0..7,12..15], while 'c's are maintained in memory. If
|
||
|
|
# you observe 'c' column, you'll notice that pair of 'c's is
|
||
|
|
# invariant between rounds. This means that we have to reload
|
||
|
|
# them once per round, in the middle. This is why you'll see
|
||
|
|
# bunch of 'c' stores and loads in the middle, but none in
|
||
|
|
# the beginning or end.
|
||
|
|
|
||
|
|
# Normally instructions would be interleaved to favour in-order
|
||
|
|
# execution. Generally out-of-order cores manage it gracefully,
|
||
|
|
# but not this time for some reason. As in-order execution
|
||
|
|
# cores are dying breed, old Atom is the only one around,
|
||
|
|
# instructions are left uninterleaved. Besides, Atom is better
|
||
|
|
# off executing 1xSSSE3 code anyway...
|
||
|
|
|
||
|
|
(
|
||
|
|
"&add (@x[$a0],@x[$b0])", # Q1
|
||
|
|
"&xor (@x[$d0],@x[$a0])",
|
||
|
|
"&rol (@x[$d0],16)",
|
||
|
|
"&add (@x[$a1],@x[$b1])", # Q2
|
||
|
|
"&xor (@x[$d1],@x[$a1])",
|
||
|
|
"&rol (@x[$d1],16)",
|
||
|
|
|
||
|
|
"&add ($xc,@x[$d0])",
|
||
|
|
"&xor (@x[$b0],$xc)",
|
||
|
|
"&rol (@x[$b0],12)",
|
||
|
|
"&add ($xc_,@x[$d1])",
|
||
|
|
"&xor (@x[$b1],$xc_)",
|
||
|
|
"&rol (@x[$b1],12)",
|
||
|
|
|
||
|
|
"&add (@x[$a0],@x[$b0])",
|
||
|
|
"&xor (@x[$d0],@x[$a0])",
|
||
|
|
"&rol (@x[$d0],8)",
|
||
|
|
"&add (@x[$a1],@x[$b1])",
|
||
|
|
"&xor (@x[$d1],@x[$a1])",
|
||
|
|
"&rol (@x[$d1],8)",
|
||
|
|
|
||
|
|
"&add ($xc,@x[$d0])",
|
||
|
|
"&xor (@x[$b0],$xc)",
|
||
|
|
"&rol (@x[$b0],7)",
|
||
|
|
"&add ($xc_,@x[$d1])",
|
||
|
|
"&xor (@x[$b1],$xc_)",
|
||
|
|
"&rol (@x[$b1],7)",
|
||
|
|
|
||
|
|
"&mov (\"4*$c0(%rsp)\",$xc)", # reload pair of 'c's
|
||
|
|
"&mov (\"4*$c1(%rsp)\",$xc_)",
|
||
|
|
"&mov ($xc,\"4*$c2(%rsp)\")",
|
||
|
|
"&mov ($xc_,\"4*$c3(%rsp)\")",
|
||
|
|
|
||
|
|
"&add (@x[$a2],@x[$b2])", # Q3
|
||
|
|
"&xor (@x[$d2],@x[$a2])",
|
||
|
|
"&rol (@x[$d2],16)",
|
||
|
|
"&add (@x[$a3],@x[$b3])", # Q4
|
||
|
|
"&xor (@x[$d3],@x[$a3])",
|
||
|
|
"&rol (@x[$d3],16)",
|
||
|
|
|
||
|
|
"&add ($xc,@x[$d2])",
|
||
|
|
"&xor (@x[$b2],$xc)",
|
||
|
|
"&rol (@x[$b2],12)",
|
||
|
|
"&add ($xc_,@x[$d3])",
|
||
|
|
"&xor (@x[$b3],$xc_)",
|
||
|
|
"&rol (@x[$b3],12)",
|
||
|
|
|
||
|
|
"&add (@x[$a2],@x[$b2])",
|
||
|
|
"&xor (@x[$d2],@x[$a2])",
|
||
|
|
"&rol (@x[$d2],8)",
|
||
|
|
"&add (@x[$a3],@x[$b3])",
|
||
|
|
"&xor (@x[$d3],@x[$a3])",
|
||
|
|
"&rol (@x[$d3],8)",
|
||
|
|
|
||
|
|
"&add ($xc,@x[$d2])",
|
||
|
|
"&xor (@x[$b2],$xc)",
|
||
|
|
"&rol (@x[$b2],7)",
|
||
|
|
"&add ($xc_,@x[$d3])",
|
||
|
|
"&xor (@x[$b3],$xc_)",
|
||
|
|
"&rol (@x[$b3],7)"
|
||
|
|
);
|
||
|
|
}
|
||
|
|
########################################################################
|
||
|
|
# HCHACHA20_SSSE3
|
||
|
|
$code.=<<___;
|
||
|
|
|
||
|
|
.global hchacha20_ssse3
|
||
|
|
.type hchacha20_ssse3,\@function,5
|
||
|
|
.align 32
|
||
|
|
hchacha20_ssse3:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lhchacha20_ssse3:
|
||
|
|
movdqa .Lsigma(%rip),%xmm0
|
||
|
|
movdqu (%rdx),%xmm1
|
||
|
|
movdqu 16(%rdx),%xmm2
|
||
|
|
movdqu (%rsi),%xmm3
|
||
|
|
movdqa .Lrot16(%rip),%xmm6
|
||
|
|
movdqa .Lrot24(%rip),%xmm7
|
||
|
|
movq 10,%r8
|
||
|
|
.align 32
|
||
|
|
.Loop_hssse3:
|
||
|
|
paddd %xmm1,%xmm0
|
||
|
|
pxor %xmm0,%xmm3
|
||
|
|
pshufb %xmm6,%xmm3
|
||
|
|
paddd %xmm3,%xmm2
|
||
|
|
pxor %xmm2,%xmm1
|
||
|
|
movdqa %xmm1,%xmm4
|
||
|
|
psrld 20,%xmm1
|
||
|
|
pslld 12,%xmm4
|
||
|
|
por %xmm4,%xmm1
|
||
|
|
paddd %xmm1,%xmm0
|
||
|
|
pxor %xmm0,%xmm3
|
||
|
|
pshufb %xmm7,%xmm3
|
||
|
|
paddd %xmm3,%xmm2
|
||
|
|
pxor %xmm2,%xmm1
|
||
|
|
movdqa %xmm1,%xmm4
|
||
|
|
psrld 25,%xmm1
|
||
|
|
pslld 7,%xmm4
|
||
|
|
por %xmm4,%xmm1
|
||
|
|
pshufd \$78,%xmm2,%xmm2
|
||
|
|
pshufd \$57,%xmm1,%xmm1
|
||
|
|
pshufd \$147,%xmm3,%xmm3
|
||
|
|
nop
|
||
|
|
paddd %xmm1,%xmm0
|
||
|
|
pxor %xmm0,%xmm3
|
||
|
|
pshufb %xmm6,%xmm3
|
||
|
|
paddd %xmm3,%xmm2
|
||
|
|
pxor %xmm2,%xmm1
|
||
|
|
movdqa %xmm1,%xmm4
|
||
|
|
psrld 20,%xmm1
|
||
|
|
pslld 12,%xmm4
|
||
|
|
por %xmm4,%xmm1
|
||
|
|
paddd %xmm1,%xmm0
|
||
|
|
pxor %xmm0,%xmm3
|
||
|
|
pshufb %xmm7,%xmm3
|
||
|
|
paddd %xmm3,%xmm2
|
||
|
|
pxor %xmm2,%xmm1
|
||
|
|
movdqa %xmm1,%xmm4
|
||
|
|
psrld 25,%xmm1
|
||
|
|
pslld 7,%xmm4
|
||
|
|
por %xmm4,%xmm1
|
||
|
|
pshufd \$78,%xmm2,%xmm2
|
||
|
|
pshufd \$147,%xmm1,%xmm1
|
||
|
|
pshufd \$57,%xmm3,%xmm3
|
||
|
|
decq %r8
|
||
|
|
jnz .Loop_hssse3
|
||
|
|
movdqu %xmm0,0(%rdi)
|
||
|
|
movdqu %xmm3,16(%rdi)
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size hchacha20_ssse3,.-hchacha20_ssse3
|
||
|
|
___
|
||
|
|
|
||
|
|
|
||
|
|
########################################################################
|
||
|
|
# SSSE3 code path that handles shorter lengths
|
||
|
|
{
|
||
|
|
my ($a,$b,$c,$d,$t,$t1,$rot16,$rot24)=map("%xmm$_",(0..7));
|
||
|
|
|
||
|
|
sub SSSE3ROUND { # critical path is 20 "SIMD ticks" per round
|
||
|
|
&paddd ($a,$b);
|
||
|
|
&pxor ($d,$a);
|
||
|
|
&pshufb ($d,$rot16);
|
||
|
|
|
||
|
|
&paddd ($c,$d);
|
||
|
|
&pxor ($b,$c);
|
||
|
|
&movdqa ($t,$b);
|
||
|
|
&psrld ($b,20);
|
||
|
|
&pslld ($t,12);
|
||
|
|
&por ($b,$t);
|
||
|
|
|
||
|
|
&paddd ($a,$b);
|
||
|
|
&pxor ($d,$a);
|
||
|
|
&pshufb ($d,$rot24);
|
||
|
|
|
||
|
|
&paddd ($c,$d);
|
||
|
|
&pxor ($b,$c);
|
||
|
|
&movdqa ($t,$b);
|
||
|
|
&psrld ($b,25);
|
||
|
|
&pslld ($t,7);
|
||
|
|
&por ($b,$t);
|
||
|
|
}
|
||
|
|
|
||
|
|
my $xframe = $win64 ? 32+8 : 8;
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.global chacha20_ssse3
|
||
|
|
.type chacha20_ssse3,\@function,5
|
||
|
|
.align 32
|
||
|
|
chacha20_ssse3:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lchacha20_ssse3:
|
||
|
|
mov %rsp,%r9 # frame pointer
|
||
|
|
.cfi_def_cfa_register %r9
|
||
|
|
___
|
||
|
|
#$code.=<<___ if ($avx);
|
||
|
|
# test \$`1<<(43-32)`,%r10d
|
||
|
|
# jnz .Lchacha20_4xop # XOP is fastest even if we use 1/4
|
||
|
|
#___
|
||
|
|
$code.=<<___;
|
||
|
|
cmp \$128,$len # we might throw away some data,
|
||
|
|
ja .Lchacha20_4x # but overall it won't be slower
|
||
|
|
|
||
|
|
.Ldo_sse3_after_all:
|
||
|
|
sub \$64+$xframe,%rsp
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps %xmm6,-0x28(%r9)
|
||
|
|
movaps %xmm7,-0x18(%r9)
|
||
|
|
.Lssse3_body:
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
movdqa .Lsigma(%rip),$a
|
||
|
|
movdqu ($key),$b
|
||
|
|
movdqu 16($key),$c
|
||
|
|
movdqu ($counter),$d
|
||
|
|
movdqa .Lrot16(%rip),$rot16
|
||
|
|
movdqa .Lrot24(%rip),$rot24
|
||
|
|
|
||
|
|
movdqa $a,0x00(%rsp)
|
||
|
|
movdqa $b,0x10(%rsp)
|
||
|
|
movdqa $c,0x20(%rsp)
|
||
|
|
movdqa $d,0x30(%rsp)
|
||
|
|
mov \$10,$counter # reuse $counter
|
||
|
|
jmp .Loop_ssse3
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_outer_ssse3:
|
||
|
|
movdqa .Lone(%rip),$d
|
||
|
|
movdqa 0x00(%rsp),$a
|
||
|
|
movdqa 0x10(%rsp),$b
|
||
|
|
movdqa 0x20(%rsp),$c
|
||
|
|
paddd 0x30(%rsp),$d
|
||
|
|
mov \$10,$counter
|
||
|
|
movdqa $d,0x30(%rsp)
|
||
|
|
jmp .Loop_ssse3
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_ssse3:
|
||
|
|
___
|
||
|
|
&SSSE3ROUND();
|
||
|
|
&pshufd ($c,$c,0b01001110);
|
||
|
|
&pshufd ($b,$b,0b00111001);
|
||
|
|
&pshufd ($d,$d,0b10010011);
|
||
|
|
&nop ();
|
||
|
|
|
||
|
|
&SSSE3ROUND();
|
||
|
|
&pshufd ($c,$c,0b01001110);
|
||
|
|
&pshufd ($b,$b,0b10010011);
|
||
|
|
&pshufd ($d,$d,0b00111001);
|
||
|
|
|
||
|
|
&dec ($counter);
|
||
|
|
&jnz (".Loop_ssse3");
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
paddd 0x00(%rsp),$a
|
||
|
|
paddd 0x10(%rsp),$b
|
||
|
|
paddd 0x20(%rsp),$c
|
||
|
|
paddd 0x30(%rsp),$d
|
||
|
|
|
||
|
|
cmp \$64,$len
|
||
|
|
jb .Ltail_ssse3
|
||
|
|
|
||
|
|
movdqu 0x00($inp),$t
|
||
|
|
movdqu 0x10($inp),$t1
|
||
|
|
pxor $t,$a # xor with input
|
||
|
|
movdqu 0x20($inp),$t
|
||
|
|
pxor $t1,$b
|
||
|
|
movdqu 0x30($inp),$t1
|
||
|
|
lea 0x40($inp),$inp # inp+=64
|
||
|
|
pxor $t,$c
|
||
|
|
pxor $t1,$d
|
||
|
|
|
||
|
|
movdqu $a,0x00($out) # write output
|
||
|
|
movdqu $b,0x10($out)
|
||
|
|
movdqu $c,0x20($out)
|
||
|
|
movdqu $d,0x30($out)
|
||
|
|
lea 0x40($out),$out # out+=64
|
||
|
|
|
||
|
|
sub \$64,$len
|
||
|
|
jnz .Loop_outer_ssse3
|
||
|
|
|
||
|
|
jmp .Ldone_ssse3
|
||
|
|
|
||
|
|
.align 16
|
||
|
|
.Ltail_ssse3:
|
||
|
|
movdqa $a,0x00(%rsp)
|
||
|
|
movdqa $b,0x10(%rsp)
|
||
|
|
movdqa $c,0x20(%rsp)
|
||
|
|
movdqa $d,0x30(%rsp)
|
||
|
|
xor $counter,$counter
|
||
|
|
|
||
|
|
.Loop_tail_ssse3:
|
||
|
|
movzb ($inp,$counter),%eax
|
||
|
|
movzb (%rsp,$counter),%ecx
|
||
|
|
lea 1($counter),$counter
|
||
|
|
xor %ecx,%eax
|
||
|
|
mov %al,-1($out,$counter)
|
||
|
|
dec $len
|
||
|
|
jnz .Loop_tail_ssse3
|
||
|
|
|
||
|
|
.Ldone_ssse3:
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps -0x28(%r9),%xmm6
|
||
|
|
movaps -0x18(%r9),%xmm7
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
lea (%r9),%rsp
|
||
|
|
.cfi_def_cfa_register %rsp
|
||
|
|
.Lssse3_epilogue:
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size chacha20_ssse3,.-chacha20_ssse3
|
||
|
|
___
|
||
|
|
}
|
||
|
|
|
||
|
|
########################################################################
|
||
|
|
# SSSE3 code path that handles longer messages.
|
||
|
|
{
|
||
|
|
# assign variables to favor Atom front-end
|
||
|
|
my ($xd0,$xd1,$xd2,$xd3, $xt0,$xt1,$xt2,$xt3,
|
||
|
|
$xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3)=map("%xmm$_",(0..15));
|
||
|
|
my @xx=($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,
|
||
|
|
"%nox","%nox","%nox","%nox", $xd0,$xd1,$xd2,$xd3);
|
||
|
|
|
||
|
|
sub SSSE3_lane_ROUND {
|
||
|
|
my ($a0,$b0,$c0,$d0)=@_;
|
||
|
|
my ($a1,$b1,$c1,$d1)=map(($_&~3)+(($_+1)&3),($a0,$b0,$c0,$d0));
|
||
|
|
my ($a2,$b2,$c2,$d2)=map(($_&~3)+(($_+1)&3),($a1,$b1,$c1,$d1));
|
||
|
|
my ($a3,$b3,$c3,$d3)=map(($_&~3)+(($_+1)&3),($a2,$b2,$c2,$d2));
|
||
|
|
my ($xc,$xc_,$t0,$t1)=map("\"$_\"",$xt0,$xt1,$xt2,$xt3);
|
||
|
|
my @x=map("\"$_\"",@xx);
|
||
|
|
|
||
|
|
# Consider order in which variables are addressed by their
|
||
|
|
# index:
|
||
|
|
#
|
||
|
|
# a b c d
|
||
|
|
#
|
||
|
|
# 0 4 8 12 < even round
|
||
|
|
# 1 5 9 13
|
||
|
|
# 2 6 10 14
|
||
|
|
# 3 7 11 15
|
||
|
|
# 0 5 10 15 < odd round
|
||
|
|
# 1 6 11 12
|
||
|
|
# 2 7 8 13
|
||
|
|
# 3 4 9 14
|
||
|
|
#
|
||
|
|
# 'a', 'b' and 'd's are permanently allocated in registers,
|
||
|
|
# @x[0..7,12..15], while 'c's are maintained in memory. If
|
||
|
|
# you observe 'c' column, you'll notice that pair of 'c's is
|
||
|
|
# invariant between rounds. This means that we have to reload
|
||
|
|
# them once per round, in the middle. This is why you'll see
|
||
|
|
# bunch of 'c' stores and loads in the middle, but none in
|
||
|
|
# the beginning or end.
|
||
|
|
|
||
|
|
(
|
||
|
|
"&paddd (@x[$a0],@x[$b0])", # Q1
|
||
|
|
"&paddd (@x[$a1],@x[$b1])", # Q2
|
||
|
|
"&pxor (@x[$d0],@x[$a0])",
|
||
|
|
"&pxor (@x[$d1],@x[$a1])",
|
||
|
|
"&pshufb (@x[$d0],$t1)",
|
||
|
|
"&pshufb (@x[$d1],$t1)",
|
||
|
|
|
||
|
|
"&paddd ($xc,@x[$d0])",
|
||
|
|
"&paddd ($xc_,@x[$d1])",
|
||
|
|
"&pxor (@x[$b0],$xc)",
|
||
|
|
"&pxor (@x[$b1],$xc_)",
|
||
|
|
"&movdqa ($t0,@x[$b0])",
|
||
|
|
"&pslld (@x[$b0],12)",
|
||
|
|
"&psrld ($t0,20)",
|
||
|
|
"&movdqa ($t1,@x[$b1])",
|
||
|
|
"&pslld (@x[$b1],12)",
|
||
|
|
"&por (@x[$b0],$t0)",
|
||
|
|
"&psrld ($t1,20)",
|
||
|
|
"&movdqa ($t0,'(%r11)')", # .Lrot24(%rip)
|
||
|
|
"&por (@x[$b1],$t1)",
|
||
|
|
|
||
|
|
"&paddd (@x[$a0],@x[$b0])",
|
||
|
|
"&paddd (@x[$a1],@x[$b1])",
|
||
|
|
"&pxor (@x[$d0],@x[$a0])",
|
||
|
|
"&pxor (@x[$d1],@x[$a1])",
|
||
|
|
"&pshufb (@x[$d0],$t0)",
|
||
|
|
"&pshufb (@x[$d1],$t0)",
|
||
|
|
|
||
|
|
"&paddd ($xc,@x[$d0])",
|
||
|
|
"&paddd ($xc_,@x[$d1])",
|
||
|
|
"&pxor (@x[$b0],$xc)",
|
||
|
|
"&pxor (@x[$b1],$xc_)",
|
||
|
|
"&movdqa ($t1,@x[$b0])",
|
||
|
|
"&pslld (@x[$b0],7)",
|
||
|
|
"&psrld ($t1,25)",
|
||
|
|
"&movdqa ($t0,@x[$b1])",
|
||
|
|
"&pslld (@x[$b1],7)",
|
||
|
|
"&por (@x[$b0],$t1)",
|
||
|
|
"&psrld ($t0,25)",
|
||
|
|
"&movdqa ($t1,'(%r10)')", # .Lrot16(%rip)
|
||
|
|
"&por (@x[$b1],$t0)",
|
||
|
|
|
||
|
|
"&movdqa (\"`16*($c0-8)`(%rsp)\",$xc)", # reload pair of 'c's
|
||
|
|
"&movdqa (\"`16*($c1-8)`(%rsp)\",$xc_)",
|
||
|
|
"&movdqa ($xc,\"`16*($c2-8)`(%rsp)\")",
|
||
|
|
"&movdqa ($xc_,\"`16*($c3-8)`(%rsp)\")",
|
||
|
|
|
||
|
|
"&paddd (@x[$a2],@x[$b2])", # Q3
|
||
|
|
"&paddd (@x[$a3],@x[$b3])", # Q4
|
||
|
|
"&pxor (@x[$d2],@x[$a2])",
|
||
|
|
"&pxor (@x[$d3],@x[$a3])",
|
||
|
|
"&pshufb (@x[$d2],$t1)",
|
||
|
|
"&pshufb (@x[$d3],$t1)",
|
||
|
|
|
||
|
|
"&paddd ($xc,@x[$d2])",
|
||
|
|
"&paddd ($xc_,@x[$d3])",
|
||
|
|
"&pxor (@x[$b2],$xc)",
|
||
|
|
"&pxor (@x[$b3],$xc_)",
|
||
|
|
"&movdqa ($t0,@x[$b2])",
|
||
|
|
"&pslld (@x[$b2],12)",
|
||
|
|
"&psrld ($t0,20)",
|
||
|
|
"&movdqa ($t1,@x[$b3])",
|
||
|
|
"&pslld (@x[$b3],12)",
|
||
|
|
"&por (@x[$b2],$t0)",
|
||
|
|
"&psrld ($t1,20)",
|
||
|
|
"&movdqa ($t0,'(%r11)')", # .Lrot24(%rip)
|
||
|
|
"&por (@x[$b3],$t1)",
|
||
|
|
|
||
|
|
"&paddd (@x[$a2],@x[$b2])",
|
||
|
|
"&paddd (@x[$a3],@x[$b3])",
|
||
|
|
"&pxor (@x[$d2],@x[$a2])",
|
||
|
|
"&pxor (@x[$d3],@x[$a3])",
|
||
|
|
"&pshufb (@x[$d2],$t0)",
|
||
|
|
"&pshufb (@x[$d3],$t0)",
|
||
|
|
|
||
|
|
"&paddd ($xc,@x[$d2])",
|
||
|
|
"&paddd ($xc_,@x[$d3])",
|
||
|
|
"&pxor (@x[$b2],$xc)",
|
||
|
|
"&pxor (@x[$b3],$xc_)",
|
||
|
|
"&movdqa ($t1,@x[$b2])",
|
||
|
|
"&pslld (@x[$b2],7)",
|
||
|
|
"&psrld ($t1,25)",
|
||
|
|
"&movdqa ($t0,@x[$b3])",
|
||
|
|
"&pslld (@x[$b3],7)",
|
||
|
|
"&por (@x[$b2],$t1)",
|
||
|
|
"&psrld ($t0,25)",
|
||
|
|
"&movdqa ($t1,'(%r10)')", # .Lrot16(%rip)
|
||
|
|
"&por (@x[$b3],$t0)"
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
my $xframe = $win64 ? 0xa8 : 8;
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.global chacha20_4x
|
||
|
|
.type chacha20_4x,\@function,5
|
||
|
|
.align 32
|
||
|
|
chacha20_4x:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lchacha20_4x:
|
||
|
|
mov %rsp,%r9 # frame pointer
|
||
|
|
.cfi_def_cfa_register %r9
|
||
|
|
# mov %r10,%r11
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($avx>1);
|
||
|
|
# shr \$32,%r10 # OPENSSL_ia32cap_P+8
|
||
|
|
# test \$`1<<5`,%r10 # test AVX2
|
||
|
|
# jnz .Lchacha20_avx2
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
# cmp \$192,$len
|
||
|
|
# ja .Lproceed4x
|
||
|
|
|
||
|
|
# and \$`1<<26|1<<22`,%r11 # isolate XSAVE+MOVBE
|
||
|
|
# cmp \$`1<<22`,%r11 # check for MOVBE without XSAVE
|
||
|
|
# je .Ldo_sse3_after_all # to detect Atom
|
||
|
|
|
||
|
|
.Lproceed4x:
|
||
|
|
sub \$0x140+$xframe,%rsp
|
||
|
|
___
|
||
|
|
################ stack layout
|
||
|
|
# +0x00 SIMD equivalent of @x[8-12]
|
||
|
|
# ...
|
||
|
|
# +0x40 constant copy of key[0-2] smashed by lanes
|
||
|
|
# ...
|
||
|
|
# +0x100 SIMD counters (with nonce smashed by lanes)
|
||
|
|
# ...
|
||
|
|
# +0x140
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps %xmm6,-0xa8(%r9)
|
||
|
|
movaps %xmm7,-0x98(%r9)
|
||
|
|
movaps %xmm8,-0x88(%r9)
|
||
|
|
movaps %xmm9,-0x78(%r9)
|
||
|
|
movaps %xmm10,-0x68(%r9)
|
||
|
|
movaps %xmm11,-0x58(%r9)
|
||
|
|
movaps %xmm12,-0x48(%r9)
|
||
|
|
movaps %xmm13,-0x38(%r9)
|
||
|
|
movaps %xmm14,-0x28(%r9)
|
||
|
|
movaps %xmm15,-0x18(%r9)
|
||
|
|
.L4x_body:
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
movdqa .Lsigma(%rip),$xa3 # key[0]
|
||
|
|
movdqu ($key),$xb3 # key[1]
|
||
|
|
movdqu 16($key),$xt3 # key[2]
|
||
|
|
movdqu ($counter),$xd3 # key[3]
|
||
|
|
lea 0x100(%rsp),%rcx # size optimization
|
||
|
|
lea .Lrot16(%rip),%r10
|
||
|
|
lea .Lrot24(%rip),%r11
|
||
|
|
|
||
|
|
pshufd \$0x00,$xa3,$xa0 # smash key by lanes...
|
||
|
|
pshufd \$0x55,$xa3,$xa1
|
||
|
|
movdqa $xa0,0x40(%rsp) # ... and offload
|
||
|
|
pshufd \$0xaa,$xa3,$xa2
|
||
|
|
movdqa $xa1,0x50(%rsp)
|
||
|
|
pshufd \$0xff,$xa3,$xa3
|
||
|
|
movdqa $xa2,0x60(%rsp)
|
||
|
|
movdqa $xa3,0x70(%rsp)
|
||
|
|
|
||
|
|
pshufd \$0x00,$xb3,$xb0
|
||
|
|
pshufd \$0x55,$xb3,$xb1
|
||
|
|
movdqa $xb0,0x80-0x100(%rcx)
|
||
|
|
pshufd \$0xaa,$xb3,$xb2
|
||
|
|
movdqa $xb1,0x90-0x100(%rcx)
|
||
|
|
pshufd \$0xff,$xb3,$xb3
|
||
|
|
movdqa $xb2,0xa0-0x100(%rcx)
|
||
|
|
movdqa $xb3,0xb0-0x100(%rcx)
|
||
|
|
|
||
|
|
pshufd \$0x00,$xt3,$xt0 # "$xc0"
|
||
|
|
pshufd \$0x55,$xt3,$xt1 # "$xc1"
|
||
|
|
movdqa $xt0,0xc0-0x100(%rcx)
|
||
|
|
pshufd \$0xaa,$xt3,$xt2 # "$xc2"
|
||
|
|
movdqa $xt1,0xd0-0x100(%rcx)
|
||
|
|
pshufd \$0xff,$xt3,$xt3 # "$xc3"
|
||
|
|
movdqa $xt2,0xe0-0x100(%rcx)
|
||
|
|
movdqa $xt3,0xf0-0x100(%rcx)
|
||
|
|
|
||
|
|
pshufd \$0x00,$xd3,$xd0
|
||
|
|
pshufd \$0x55,$xd3,$xd1
|
||
|
|
paddd .Linc(%rip),$xd0 # don't save counters yet
|
||
|
|
pshufd \$0xaa,$xd3,$xd2
|
||
|
|
movdqa $xd1,0x110-0x100(%rcx)
|
||
|
|
pshufd \$0xff,$xd3,$xd3
|
||
|
|
movdqa $xd2,0x120-0x100(%rcx)
|
||
|
|
movdqa $xd3,0x130-0x100(%rcx)
|
||
|
|
|
||
|
|
jmp .Loop_enter4x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_outer4x:
|
||
|
|
movdqa 0x40(%rsp),$xa0 # re-load smashed key
|
||
|
|
movdqa 0x50(%rsp),$xa1
|
||
|
|
movdqa 0x60(%rsp),$xa2
|
||
|
|
movdqa 0x70(%rsp),$xa3
|
||
|
|
movdqa 0x80-0x100(%rcx),$xb0
|
||
|
|
movdqa 0x90-0x100(%rcx),$xb1
|
||
|
|
movdqa 0xa0-0x100(%rcx),$xb2
|
||
|
|
movdqa 0xb0-0x100(%rcx),$xb3
|
||
|
|
movdqa 0xc0-0x100(%rcx),$xt0 # "$xc0"
|
||
|
|
movdqa 0xd0-0x100(%rcx),$xt1 # "$xc1"
|
||
|
|
movdqa 0xe0-0x100(%rcx),$xt2 # "$xc2"
|
||
|
|
movdqa 0xf0-0x100(%rcx),$xt3 # "$xc3"
|
||
|
|
movdqa 0x100-0x100(%rcx),$xd0
|
||
|
|
movdqa 0x110-0x100(%rcx),$xd1
|
||
|
|
movdqa 0x120-0x100(%rcx),$xd2
|
||
|
|
movdqa 0x130-0x100(%rcx),$xd3
|
||
|
|
paddd .Lfour(%rip),$xd0 # next SIMD counters
|
||
|
|
|
||
|
|
.Loop_enter4x:
|
||
|
|
movdqa $xt2,0x20(%rsp) # SIMD equivalent of "@x[10]"
|
||
|
|
movdqa $xt3,0x30(%rsp) # SIMD equivalent of "@x[11]"
|
||
|
|
movdqa (%r10),$xt3 # .Lrot16(%rip)
|
||
|
|
mov \$10,%eax
|
||
|
|
movdqa $xd0,0x100-0x100(%rcx) # save SIMD counters
|
||
|
|
jmp .Loop4x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop4x:
|
||
|
|
___
|
||
|
|
foreach (&SSSE3_lane_ROUND(0, 4, 8,12)) { eval; }
|
||
|
|
foreach (&SSSE3_lane_ROUND(0, 5,10,15)) { eval; }
|
||
|
|
$code.=<<___;
|
||
|
|
dec %eax
|
||
|
|
jnz .Loop4x
|
||
|
|
|
||
|
|
paddd 0x40(%rsp),$xa0 # accumulate key material
|
||
|
|
paddd 0x50(%rsp),$xa1
|
||
|
|
paddd 0x60(%rsp),$xa2
|
||
|
|
paddd 0x70(%rsp),$xa3
|
||
|
|
|
||
|
|
movdqa $xa0,$xt2 # "de-interlace" data
|
||
|
|
punpckldq $xa1,$xa0
|
||
|
|
movdqa $xa2,$xt3
|
||
|
|
punpckldq $xa3,$xa2
|
||
|
|
punpckhdq $xa1,$xt2
|
||
|
|
punpckhdq $xa3,$xt3
|
||
|
|
movdqa $xa0,$xa1
|
||
|
|
punpcklqdq $xa2,$xa0 # "a0"
|
||
|
|
movdqa $xt2,$xa3
|
||
|
|
punpcklqdq $xt3,$xt2 # "a2"
|
||
|
|
punpckhqdq $xa2,$xa1 # "a1"
|
||
|
|
punpckhqdq $xt3,$xa3 # "a3"
|
||
|
|
___
|
||
|
|
($xa2,$xt2)=($xt2,$xa2);
|
||
|
|
$code.=<<___;
|
||
|
|
paddd 0x80-0x100(%rcx),$xb0
|
||
|
|
paddd 0x90-0x100(%rcx),$xb1
|
||
|
|
paddd 0xa0-0x100(%rcx),$xb2
|
||
|
|
paddd 0xb0-0x100(%rcx),$xb3
|
||
|
|
|
||
|
|
movdqa $xa0,0x00(%rsp) # offload $xaN
|
||
|
|
movdqa $xa1,0x10(%rsp)
|
||
|
|
movdqa 0x20(%rsp),$xa0 # "xc2"
|
||
|
|
movdqa 0x30(%rsp),$xa1 # "xc3"
|
||
|
|
|
||
|
|
movdqa $xb0,$xt2
|
||
|
|
punpckldq $xb1,$xb0
|
||
|
|
movdqa $xb2,$xt3
|
||
|
|
punpckldq $xb3,$xb2
|
||
|
|
punpckhdq $xb1,$xt2
|
||
|
|
punpckhdq $xb3,$xt3
|
||
|
|
movdqa $xb0,$xb1
|
||
|
|
punpcklqdq $xb2,$xb0 # "b0"
|
||
|
|
movdqa $xt2,$xb3
|
||
|
|
punpcklqdq $xt3,$xt2 # "b2"
|
||
|
|
punpckhqdq $xb2,$xb1 # "b1"
|
||
|
|
punpckhqdq $xt3,$xb3 # "b3"
|
||
|
|
___
|
||
|
|
($xb2,$xt2)=($xt2,$xb2);
|
||
|
|
my ($xc0,$xc1,$xc2,$xc3)=($xt0,$xt1,$xa0,$xa1);
|
||
|
|
$code.=<<___;
|
||
|
|
paddd 0xc0-0x100(%rcx),$xc0
|
||
|
|
paddd 0xd0-0x100(%rcx),$xc1
|
||
|
|
paddd 0xe0-0x100(%rcx),$xc2
|
||
|
|
paddd 0xf0-0x100(%rcx),$xc3
|
||
|
|
|
||
|
|
movdqa $xa2,0x20(%rsp) # keep offloading $xaN
|
||
|
|
movdqa $xa3,0x30(%rsp)
|
||
|
|
|
||
|
|
movdqa $xc0,$xt2
|
||
|
|
punpckldq $xc1,$xc0
|
||
|
|
movdqa $xc2,$xt3
|
||
|
|
punpckldq $xc3,$xc2
|
||
|
|
punpckhdq $xc1,$xt2
|
||
|
|
punpckhdq $xc3,$xt3
|
||
|
|
movdqa $xc0,$xc1
|
||
|
|
punpcklqdq $xc2,$xc0 # "c0"
|
||
|
|
movdqa $xt2,$xc3
|
||
|
|
punpcklqdq $xt3,$xt2 # "c2"
|
||
|
|
punpckhqdq $xc2,$xc1 # "c1"
|
||
|
|
punpckhqdq $xt3,$xc3 # "c3"
|
||
|
|
___
|
||
|
|
($xc2,$xt2)=($xt2,$xc2);
|
||
|
|
($xt0,$xt1)=($xa2,$xa3); # use $xaN as temporary
|
||
|
|
$code.=<<___;
|
||
|
|
paddd 0x100-0x100(%rcx),$xd0
|
||
|
|
paddd 0x110-0x100(%rcx),$xd1
|
||
|
|
paddd 0x120-0x100(%rcx),$xd2
|
||
|
|
paddd 0x130-0x100(%rcx),$xd3
|
||
|
|
|
||
|
|
movdqa $xd0,$xt2
|
||
|
|
punpckldq $xd1,$xd0
|
||
|
|
movdqa $xd2,$xt3
|
||
|
|
punpckldq $xd3,$xd2
|
||
|
|
punpckhdq $xd1,$xt2
|
||
|
|
punpckhdq $xd3,$xt3
|
||
|
|
movdqa $xd0,$xd1
|
||
|
|
punpcklqdq $xd2,$xd0 # "d0"
|
||
|
|
movdqa $xt2,$xd3
|
||
|
|
punpcklqdq $xt3,$xt2 # "d2"
|
||
|
|
punpckhqdq $xd2,$xd1 # "d1"
|
||
|
|
punpckhqdq $xt3,$xd3 # "d3"
|
||
|
|
___
|
||
|
|
($xd2,$xt2)=($xt2,$xd2);
|
||
|
|
$code.=<<___;
|
||
|
|
cmp \$64*4,$len
|
||
|
|
jb .Ltail4x
|
||
|
|
|
||
|
|
movdqu 0x00($inp),$xt0 # xor with input
|
||
|
|
movdqu 0x10($inp),$xt1
|
||
|
|
movdqu 0x20($inp),$xt2
|
||
|
|
movdqu 0x30($inp),$xt3
|
||
|
|
pxor 0x00(%rsp),$xt0 # $xaN is offloaded, remember?
|
||
|
|
pxor $xb0,$xt1
|
||
|
|
pxor $xc0,$xt2
|
||
|
|
pxor $xd0,$xt3
|
||
|
|
|
||
|
|
movdqu $xt0,0x00($out)
|
||
|
|
movdqu 0x40($inp),$xt0
|
||
|
|
movdqu $xt1,0x10($out)
|
||
|
|
movdqu 0x50($inp),$xt1
|
||
|
|
movdqu $xt2,0x20($out)
|
||
|
|
movdqu 0x60($inp),$xt2
|
||
|
|
movdqu $xt3,0x30($out)
|
||
|
|
movdqu 0x70($inp),$xt3
|
||
|
|
lea 0x80($inp),$inp # size optimization
|
||
|
|
pxor 0x10(%rsp),$xt0
|
||
|
|
pxor $xb1,$xt1
|
||
|
|
pxor $xc1,$xt2
|
||
|
|
pxor $xd1,$xt3
|
||
|
|
|
||
|
|
movdqu $xt0,0x40($out)
|
||
|
|
movdqu 0x00($inp),$xt0
|
||
|
|
movdqu $xt1,0x50($out)
|
||
|
|
movdqu 0x10($inp),$xt1
|
||
|
|
movdqu $xt2,0x60($out)
|
||
|
|
movdqu 0x20($inp),$xt2
|
||
|
|
movdqu $xt3,0x70($out)
|
||
|
|
lea 0x80($out),$out # size optimization
|
||
|
|
movdqu 0x30($inp),$xt3
|
||
|
|
pxor 0x20(%rsp),$xt0
|
||
|
|
pxor $xb2,$xt1
|
||
|
|
pxor $xc2,$xt2
|
||
|
|
pxor $xd2,$xt3
|
||
|
|
|
||
|
|
movdqu $xt0,0x00($out)
|
||
|
|
movdqu 0x40($inp),$xt0
|
||
|
|
movdqu $xt1,0x10($out)
|
||
|
|
movdqu 0x50($inp),$xt1
|
||
|
|
movdqu $xt2,0x20($out)
|
||
|
|
movdqu 0x60($inp),$xt2
|
||
|
|
movdqu $xt3,0x30($out)
|
||
|
|
movdqu 0x70($inp),$xt3
|
||
|
|
lea 0x80($inp),$inp # inp+=64*4
|
||
|
|
pxor 0x30(%rsp),$xt0
|
||
|
|
pxor $xb3,$xt1
|
||
|
|
pxor $xc3,$xt2
|
||
|
|
pxor $xd3,$xt3
|
||
|
|
movdqu $xt0,0x40($out)
|
||
|
|
movdqu $xt1,0x50($out)
|
||
|
|
movdqu $xt2,0x60($out)
|
||
|
|
movdqu $xt3,0x70($out)
|
||
|
|
lea 0x80($out),$out # out+=64*4
|
||
|
|
|
||
|
|
sub \$64*4,$len
|
||
|
|
jnz .Loop_outer4x
|
||
|
|
|
||
|
|
jmp .Ldone4x
|
||
|
|
|
||
|
|
.Ltail4x:
|
||
|
|
cmp \$192,$len
|
||
|
|
jae .L192_or_more4x
|
||
|
|
cmp \$128,$len
|
||
|
|
jae .L128_or_more4x
|
||
|
|
cmp \$64,$len
|
||
|
|
jae .L64_or_more4x
|
||
|
|
|
||
|
|
#movdqa 0x00(%rsp),$xt0 # $xaN is offloaded, remember?
|
||
|
|
xor %r10,%r10
|
||
|
|
#movdqa $xt0,0x00(%rsp)
|
||
|
|
movdqa $xb0,0x10(%rsp)
|
||
|
|
movdqa $xc0,0x20(%rsp)
|
||
|
|
movdqa $xd0,0x30(%rsp)
|
||
|
|
jmp .Loop_tail4x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L64_or_more4x:
|
||
|
|
movdqu 0x00($inp),$xt0 # xor with input
|
||
|
|
movdqu 0x10($inp),$xt1
|
||
|
|
movdqu 0x20($inp),$xt2
|
||
|
|
movdqu 0x30($inp),$xt3
|
||
|
|
pxor 0x00(%rsp),$xt0 # $xaxN is offloaded, remember?
|
||
|
|
pxor $xb0,$xt1
|
||
|
|
pxor $xc0,$xt2
|
||
|
|
pxor $xd0,$xt3
|
||
|
|
movdqu $xt0,0x00($out)
|
||
|
|
movdqu $xt1,0x10($out)
|
||
|
|
movdqu $xt2,0x20($out)
|
||
|
|
movdqu $xt3,0x30($out)
|
||
|
|
je .Ldone4x
|
||
|
|
|
||
|
|
movdqa 0x10(%rsp),$xt0 # $xaN is offloaded, remember?
|
||
|
|
lea 0x40($inp),$inp # inp+=64*1
|
||
|
|
xor %r10,%r10
|
||
|
|
movdqa $xt0,0x00(%rsp)
|
||
|
|
movdqa $xb1,0x10(%rsp)
|
||
|
|
lea 0x40($out),$out # out+=64*1
|
||
|
|
movdqa $xc1,0x20(%rsp)
|
||
|
|
sub \$64,$len # len-=64*1
|
||
|
|
movdqa $xd1,0x30(%rsp)
|
||
|
|
jmp .Loop_tail4x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L128_or_more4x:
|
||
|
|
movdqu 0x00($inp),$xt0 # xor with input
|
||
|
|
movdqu 0x10($inp),$xt1
|
||
|
|
movdqu 0x20($inp),$xt2
|
||
|
|
movdqu 0x30($inp),$xt3
|
||
|
|
pxor 0x00(%rsp),$xt0 # $xaN is offloaded, remember?
|
||
|
|
pxor $xb0,$xt1
|
||
|
|
pxor $xc0,$xt2
|
||
|
|
pxor $xd0,$xt3
|
||
|
|
|
||
|
|
movdqu $xt0,0x00($out)
|
||
|
|
movdqu 0x40($inp),$xt0
|
||
|
|
movdqu $xt1,0x10($out)
|
||
|
|
movdqu 0x50($inp),$xt1
|
||
|
|
movdqu $xt2,0x20($out)
|
||
|
|
movdqu 0x60($inp),$xt2
|
||
|
|
movdqu $xt3,0x30($out)
|
||
|
|
movdqu 0x70($inp),$xt3
|
||
|
|
pxor 0x10(%rsp),$xt0
|
||
|
|
pxor $xb1,$xt1
|
||
|
|
pxor $xc1,$xt2
|
||
|
|
pxor $xd1,$xt3
|
||
|
|
movdqu $xt0,0x40($out)
|
||
|
|
movdqu $xt1,0x50($out)
|
||
|
|
movdqu $xt2,0x60($out)
|
||
|
|
movdqu $xt3,0x70($out)
|
||
|
|
je .Ldone4x
|
||
|
|
|
||
|
|
movdqa 0x20(%rsp),$xt0 # $xaN is offloaded, remember?
|
||
|
|
lea 0x80($inp),$inp # inp+=64*2
|
||
|
|
xor %r10,%r10
|
||
|
|
movdqa $xt0,0x00(%rsp)
|
||
|
|
movdqa $xb2,0x10(%rsp)
|
||
|
|
lea 0x80($out),$out # out+=64*2
|
||
|
|
movdqa $xc2,0x20(%rsp)
|
||
|
|
sub \$128,$len # len-=64*2
|
||
|
|
movdqa $xd2,0x30(%rsp)
|
||
|
|
jmp .Loop_tail4x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L192_or_more4x:
|
||
|
|
movdqu 0x00($inp),$xt0 # xor with input
|
||
|
|
movdqu 0x10($inp),$xt1
|
||
|
|
movdqu 0x20($inp),$xt2
|
||
|
|
movdqu 0x30($inp),$xt3
|
||
|
|
pxor 0x00(%rsp),$xt0 # $xaN is offloaded, remember?
|
||
|
|
pxor $xb0,$xt1
|
||
|
|
pxor $xc0,$xt2
|
||
|
|
pxor $xd0,$xt3
|
||
|
|
|
||
|
|
movdqu $xt0,0x00($out)
|
||
|
|
movdqu 0x40($inp),$xt0
|
||
|
|
movdqu $xt1,0x10($out)
|
||
|
|
movdqu 0x50($inp),$xt1
|
||
|
|
movdqu $xt2,0x20($out)
|
||
|
|
movdqu 0x60($inp),$xt2
|
||
|
|
movdqu $xt3,0x30($out)
|
||
|
|
movdqu 0x70($inp),$xt3
|
||
|
|
lea 0x80($inp),$inp # size optimization
|
||
|
|
pxor 0x10(%rsp),$xt0
|
||
|
|
pxor $xb1,$xt1
|
||
|
|
pxor $xc1,$xt2
|
||
|
|
pxor $xd1,$xt3
|
||
|
|
|
||
|
|
movdqu $xt0,0x40($out)
|
||
|
|
movdqu 0x00($inp),$xt0
|
||
|
|
movdqu $xt1,0x50($out)
|
||
|
|
movdqu 0x10($inp),$xt1
|
||
|
|
movdqu $xt2,0x60($out)
|
||
|
|
movdqu 0x20($inp),$xt2
|
||
|
|
movdqu $xt3,0x70($out)
|
||
|
|
lea 0x80($out),$out # size optimization
|
||
|
|
movdqu 0x30($inp),$xt3
|
||
|
|
pxor 0x20(%rsp),$xt0
|
||
|
|
pxor $xb2,$xt1
|
||
|
|
pxor $xc2,$xt2
|
||
|
|
pxor $xd2,$xt3
|
||
|
|
movdqu $xt0,0x00($out)
|
||
|
|
movdqu $xt1,0x10($out)
|
||
|
|
movdqu $xt2,0x20($out)
|
||
|
|
movdqu $xt3,0x30($out)
|
||
|
|
je .Ldone4x
|
||
|
|
|
||
|
|
movdqa 0x30(%rsp),$xt0 # $xaN is offloaded, remember?
|
||
|
|
lea 0x40($inp),$inp # inp+=64*3
|
||
|
|
xor %r10,%r10
|
||
|
|
movdqa $xt0,0x00(%rsp)
|
||
|
|
movdqa $xb3,0x10(%rsp)
|
||
|
|
lea 0x40($out),$out # out+=64*3
|
||
|
|
movdqa $xc3,0x20(%rsp)
|
||
|
|
sub \$192,$len # len-=64*3
|
||
|
|
movdqa $xd3,0x30(%rsp)
|
||
|
|
|
||
|
|
.Loop_tail4x:
|
||
|
|
movzb ($inp,%r10),%eax
|
||
|
|
movzb (%rsp,%r10),%ecx
|
||
|
|
lea 1(%r10),%r10
|
||
|
|
xor %ecx,%eax
|
||
|
|
mov %al,-1($out,%r10)
|
||
|
|
dec $len
|
||
|
|
jnz .Loop_tail4x
|
||
|
|
|
||
|
|
.Ldone4x:
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps -0xa8(%r9),%xmm6
|
||
|
|
movaps -0x98(%r9),%xmm7
|
||
|
|
movaps -0x88(%r9),%xmm8
|
||
|
|
movaps -0x78(%r9),%xmm9
|
||
|
|
movaps -0x68(%r9),%xmm10
|
||
|
|
movaps -0x58(%r9),%xmm11
|
||
|
|
movaps -0x48(%r9),%xmm12
|
||
|
|
movaps -0x38(%r9),%xmm13
|
||
|
|
movaps -0x28(%r9),%xmm14
|
||
|
|
movaps -0x18(%r9),%xmm15
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
lea (%r9),%rsp
|
||
|
|
.cfi_def_cfa_register %rsp
|
||
|
|
.L4x_epilogue:
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size chacha20_4x,.-chacha20_4x
|
||
|
|
___
|
||
|
|
}
|
||
|
|
|
||
|
|
########################################################################
|
||
|
|
# XOP code path that handles all lengths.
|
||
|
|
if ($avx && 0) {
|
||
|
|
# There is some "anomaly" observed depending on instructions' size or
|
||
|
|
# alignment. If you look closely at below code you'll notice that
|
||
|
|
# sometimes argument order varies. The order affects instruction
|
||
|
|
# encoding by making it larger, and such fiddling gives 5% performance
|
||
|
|
# improvement. This is on FX-4100...
|
||
|
|
|
||
|
|
my ($xb0,$xb1,$xb2,$xb3, $xd0,$xd1,$xd2,$xd3,
|
||
|
|
$xa0,$xa1,$xa2,$xa3, $xt0,$xt1,$xt2,$xt3)=map("%xmm$_",(0..15));
|
||
|
|
my @xx=($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,
|
||
|
|
$xt0,$xt1,$xt2,$xt3, $xd0,$xd1,$xd2,$xd3);
|
||
|
|
|
||
|
|
sub XOP_lane_ROUND {
|
||
|
|
my ($a0,$b0,$c0,$d0)=@_;
|
||
|
|
my ($a1,$b1,$c1,$d1)=map(($_&~3)+(($_+1)&3),($a0,$b0,$c0,$d0));
|
||
|
|
my ($a2,$b2,$c2,$d2)=map(($_&~3)+(($_+1)&3),($a1,$b1,$c1,$d1));
|
||
|
|
my ($a3,$b3,$c3,$d3)=map(($_&~3)+(($_+1)&3),($a2,$b2,$c2,$d2));
|
||
|
|
my @x=map("\"$_\"",@xx);
|
||
|
|
|
||
|
|
(
|
||
|
|
"&vpaddd (@x[$a0],@x[$a0],@x[$b0])", # Q1
|
||
|
|
"&vpaddd (@x[$a1],@x[$a1],@x[$b1])", # Q2
|
||
|
|
"&vpaddd (@x[$a2],@x[$a2],@x[$b2])", # Q3
|
||
|
|
"&vpaddd (@x[$a3],@x[$a3],@x[$b3])", # Q4
|
||
|
|
"&vpxor (@x[$d0],@x[$a0],@x[$d0])",
|
||
|
|
"&vpxor (@x[$d1],@x[$a1],@x[$d1])",
|
||
|
|
"&vpxor (@x[$d2],@x[$a2],@x[$d2])",
|
||
|
|
"&vpxor (@x[$d3],@x[$a3],@x[$d3])",
|
||
|
|
"&vprotd (@x[$d0],@x[$d0],16)",
|
||
|
|
"&vprotd (@x[$d1],@x[$d1],16)",
|
||
|
|
"&vprotd (@x[$d2],@x[$d2],16)",
|
||
|
|
"&vprotd (@x[$d3],@x[$d3],16)",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$c0],@x[$c0],@x[$d0])",
|
||
|
|
"&vpaddd (@x[$c1],@x[$c1],@x[$d1])",
|
||
|
|
"&vpaddd (@x[$c2],@x[$c2],@x[$d2])",
|
||
|
|
"&vpaddd (@x[$c3],@x[$c3],@x[$d3])",
|
||
|
|
"&vpxor (@x[$b0],@x[$c0],@x[$b0])",
|
||
|
|
"&vpxor (@x[$b1],@x[$c1],@x[$b1])",
|
||
|
|
"&vpxor (@x[$b2],@x[$b2],@x[$c2])", # flip
|
||
|
|
"&vpxor (@x[$b3],@x[$b3],@x[$c3])", # flip
|
||
|
|
"&vprotd (@x[$b0],@x[$b0],12)",
|
||
|
|
"&vprotd (@x[$b1],@x[$b1],12)",
|
||
|
|
"&vprotd (@x[$b2],@x[$b2],12)",
|
||
|
|
"&vprotd (@x[$b3],@x[$b3],12)",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$a0],@x[$b0],@x[$a0])", # flip
|
||
|
|
"&vpaddd (@x[$a1],@x[$b1],@x[$a1])", # flip
|
||
|
|
"&vpaddd (@x[$a2],@x[$a2],@x[$b2])",
|
||
|
|
"&vpaddd (@x[$a3],@x[$a3],@x[$b3])",
|
||
|
|
"&vpxor (@x[$d0],@x[$a0],@x[$d0])",
|
||
|
|
"&vpxor (@x[$d1],@x[$a1],@x[$d1])",
|
||
|
|
"&vpxor (@x[$d2],@x[$a2],@x[$d2])",
|
||
|
|
"&vpxor (@x[$d3],@x[$a3],@x[$d3])",
|
||
|
|
"&vprotd (@x[$d0],@x[$d0],8)",
|
||
|
|
"&vprotd (@x[$d1],@x[$d1],8)",
|
||
|
|
"&vprotd (@x[$d2],@x[$d2],8)",
|
||
|
|
"&vprotd (@x[$d3],@x[$d3],8)",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$c0],@x[$c0],@x[$d0])",
|
||
|
|
"&vpaddd (@x[$c1],@x[$c1],@x[$d1])",
|
||
|
|
"&vpaddd (@x[$c2],@x[$c2],@x[$d2])",
|
||
|
|
"&vpaddd (@x[$c3],@x[$c3],@x[$d3])",
|
||
|
|
"&vpxor (@x[$b0],@x[$c0],@x[$b0])",
|
||
|
|
"&vpxor (@x[$b1],@x[$c1],@x[$b1])",
|
||
|
|
"&vpxor (@x[$b2],@x[$b2],@x[$c2])", # flip
|
||
|
|
"&vpxor (@x[$b3],@x[$b3],@x[$c3])", # flip
|
||
|
|
"&vprotd (@x[$b0],@x[$b0],7)",
|
||
|
|
"&vprotd (@x[$b1],@x[$b1],7)",
|
||
|
|
"&vprotd (@x[$b2],@x[$b2],7)",
|
||
|
|
"&vprotd (@x[$b3],@x[$b3],7)"
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
my $xframe = $win64 ? 0xa8 : 8;
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.global chacha20_4xop
|
||
|
|
.type chacha20_4xop,\@function,5
|
||
|
|
.align 32
|
||
|
|
chacha20_4xop:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lchacha20_4xop:
|
||
|
|
mov %rsp,%r9 # frame pointer
|
||
|
|
.cfi_def_cfa_register %r9
|
||
|
|
sub \$0x140+$xframe,%rsp
|
||
|
|
___
|
||
|
|
################ stack layout
|
||
|
|
# +0x00 SIMD equivalent of @x[8-12]
|
||
|
|
# ...
|
||
|
|
# +0x40 constant copy of key[0-2] smashed by lanes
|
||
|
|
# ...
|
||
|
|
# +0x100 SIMD counters (with nonce smashed by lanes)
|
||
|
|
# ...
|
||
|
|
# +0x140
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps %xmm6,-0xa8(%r9)
|
||
|
|
movaps %xmm7,-0x98(%r9)
|
||
|
|
movaps %xmm8,-0x88(%r9)
|
||
|
|
movaps %xmm9,-0x78(%r9)
|
||
|
|
movaps %xmm10,-0x68(%r9)
|
||
|
|
movaps %xmm11,-0x58(%r9)
|
||
|
|
movaps %xmm12,-0x48(%r9)
|
||
|
|
movaps %xmm13,-0x38(%r9)
|
||
|
|
movaps %xmm14,-0x28(%r9)
|
||
|
|
movaps %xmm15,-0x18(%r9)
|
||
|
|
.L4xop_body:
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
vzeroupper
|
||
|
|
|
||
|
|
vmovdqa .Lsigma(%rip),$xa3 # key[0]
|
||
|
|
vmovdqu ($key),$xb3 # key[1]
|
||
|
|
vmovdqu 16($key),$xt3 # key[2]
|
||
|
|
vmovdqu ($counter),$xd3 # key[3]
|
||
|
|
lea 0x100(%rsp),%rcx # size optimization
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xa3,$xa0 # smash key by lanes...
|
||
|
|
vpshufd \$0x55,$xa3,$xa1
|
||
|
|
vmovdqa $xa0,0x40(%rsp) # ... and offload
|
||
|
|
vpshufd \$0xaa,$xa3,$xa2
|
||
|
|
vmovdqa $xa1,0x50(%rsp)
|
||
|
|
vpshufd \$0xff,$xa3,$xa3
|
||
|
|
vmovdqa $xa2,0x60(%rsp)
|
||
|
|
vmovdqa $xa3,0x70(%rsp)
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xb3,$xb0
|
||
|
|
vpshufd \$0x55,$xb3,$xb1
|
||
|
|
vmovdqa $xb0,0x80-0x100(%rcx)
|
||
|
|
vpshufd \$0xaa,$xb3,$xb2
|
||
|
|
vmovdqa $xb1,0x90-0x100(%rcx)
|
||
|
|
vpshufd \$0xff,$xb3,$xb3
|
||
|
|
vmovdqa $xb2,0xa0-0x100(%rcx)
|
||
|
|
vmovdqa $xb3,0xb0-0x100(%rcx)
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xt3,$xt0 # "$xc0"
|
||
|
|
vpshufd \$0x55,$xt3,$xt1 # "$xc1"
|
||
|
|
vmovdqa $xt0,0xc0-0x100(%rcx)
|
||
|
|
vpshufd \$0xaa,$xt3,$xt2 # "$xc2"
|
||
|
|
vmovdqa $xt1,0xd0-0x100(%rcx)
|
||
|
|
vpshufd \$0xff,$xt3,$xt3 # "$xc3"
|
||
|
|
vmovdqa $xt2,0xe0-0x100(%rcx)
|
||
|
|
vmovdqa $xt3,0xf0-0x100(%rcx)
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xd3,$xd0
|
||
|
|
vpshufd \$0x55,$xd3,$xd1
|
||
|
|
vpaddd .Linc(%rip),$xd0,$xd0 # don't save counters yet
|
||
|
|
vpshufd \$0xaa,$xd3,$xd2
|
||
|
|
vmovdqa $xd1,0x110-0x100(%rcx)
|
||
|
|
vpshufd \$0xff,$xd3,$xd3
|
||
|
|
vmovdqa $xd2,0x120-0x100(%rcx)
|
||
|
|
vmovdqa $xd3,0x130-0x100(%rcx)
|
||
|
|
|
||
|
|
jmp .Loop_enter4xop
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_outer4xop:
|
||
|
|
vmovdqa 0x40(%rsp),$xa0 # re-load smashed key
|
||
|
|
vmovdqa 0x50(%rsp),$xa1
|
||
|
|
vmovdqa 0x60(%rsp),$xa2
|
||
|
|
vmovdqa 0x70(%rsp),$xa3
|
||
|
|
vmovdqa 0x80-0x100(%rcx),$xb0
|
||
|
|
vmovdqa 0x90-0x100(%rcx),$xb1
|
||
|
|
vmovdqa 0xa0-0x100(%rcx),$xb2
|
||
|
|
vmovdqa 0xb0-0x100(%rcx),$xb3
|
||
|
|
vmovdqa 0xc0-0x100(%rcx),$xt0 # "$xc0"
|
||
|
|
vmovdqa 0xd0-0x100(%rcx),$xt1 # "$xc1"
|
||
|
|
vmovdqa 0xe0-0x100(%rcx),$xt2 # "$xc2"
|
||
|
|
vmovdqa 0xf0-0x100(%rcx),$xt3 # "$xc3"
|
||
|
|
vmovdqa 0x100-0x100(%rcx),$xd0
|
||
|
|
vmovdqa 0x110-0x100(%rcx),$xd1
|
||
|
|
vmovdqa 0x120-0x100(%rcx),$xd2
|
||
|
|
vmovdqa 0x130-0x100(%rcx),$xd3
|
||
|
|
vpaddd .Lfour(%rip),$xd0,$xd0 # next SIMD counters
|
||
|
|
|
||
|
|
.Loop_enter4xop:
|
||
|
|
mov \$10,%eax
|
||
|
|
vmovdqa $xd0,0x100-0x100(%rcx) # save SIMD counters
|
||
|
|
jmp .Loop4xop
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop4xop:
|
||
|
|
___
|
||
|
|
foreach (&XOP_lane_ROUND(0, 4, 8,12)) { eval; }
|
||
|
|
foreach (&XOP_lane_ROUND(0, 5,10,15)) { eval; }
|
||
|
|
$code.=<<___;
|
||
|
|
dec %eax
|
||
|
|
jnz .Loop4xop
|
||
|
|
|
||
|
|
vpaddd 0x40(%rsp),$xa0,$xa0 # accumulate key material
|
||
|
|
vpaddd 0x50(%rsp),$xa1,$xa1
|
||
|
|
vpaddd 0x60(%rsp),$xa2,$xa2
|
||
|
|
vpaddd 0x70(%rsp),$xa3,$xa3
|
||
|
|
|
||
|
|
vmovdqa $xt2,0x20(%rsp) # offload $xc2,3
|
||
|
|
vmovdqa $xt3,0x30(%rsp)
|
||
|
|
|
||
|
|
vpunpckldq $xa1,$xa0,$xt2 # "de-interlace" data
|
||
|
|
vpunpckldq $xa3,$xa2,$xt3
|
||
|
|
vpunpckhdq $xa1,$xa0,$xa0
|
||
|
|
vpunpckhdq $xa3,$xa2,$xa2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xa1 # "a0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "a1"
|
||
|
|
vpunpcklqdq $xa2,$xa0,$xa3 # "a2"
|
||
|
|
vpunpckhqdq $xa2,$xa0,$xa0 # "a3"
|
||
|
|
___
|
||
|
|
($xa0,$xa1,$xa2,$xa3,$xt2)=($xa1,$xt2,$xa3,$xa0,$xa2);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd 0x80-0x100(%rcx),$xb0,$xb0
|
||
|
|
vpaddd 0x90-0x100(%rcx),$xb1,$xb1
|
||
|
|
vpaddd 0xa0-0x100(%rcx),$xb2,$xb2
|
||
|
|
vpaddd 0xb0-0x100(%rcx),$xb3,$xb3
|
||
|
|
|
||
|
|
vmovdqa $xa0,0x00(%rsp) # offload $xa0,1
|
||
|
|
vmovdqa $xa1,0x10(%rsp)
|
||
|
|
vmovdqa 0x20(%rsp),$xa0 # "xc2"
|
||
|
|
vmovdqa 0x30(%rsp),$xa1 # "xc3"
|
||
|
|
|
||
|
|
vpunpckldq $xb1,$xb0,$xt2
|
||
|
|
vpunpckldq $xb3,$xb2,$xt3
|
||
|
|
vpunpckhdq $xb1,$xb0,$xb0
|
||
|
|
vpunpckhdq $xb3,$xb2,$xb2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xb1 # "b0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "b1"
|
||
|
|
vpunpcklqdq $xb2,$xb0,$xb3 # "b2"
|
||
|
|
vpunpckhqdq $xb2,$xb0,$xb0 # "b3"
|
||
|
|
___
|
||
|
|
($xb0,$xb1,$xb2,$xb3,$xt2)=($xb1,$xt2,$xb3,$xb0,$xb2);
|
||
|
|
my ($xc0,$xc1,$xc2,$xc3)=($xt0,$xt1,$xa0,$xa1);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd 0xc0-0x100(%rcx),$xc0,$xc0
|
||
|
|
vpaddd 0xd0-0x100(%rcx),$xc1,$xc1
|
||
|
|
vpaddd 0xe0-0x100(%rcx),$xc2,$xc2
|
||
|
|
vpaddd 0xf0-0x100(%rcx),$xc3,$xc3
|
||
|
|
|
||
|
|
vpunpckldq $xc1,$xc0,$xt2
|
||
|
|
vpunpckldq $xc3,$xc2,$xt3
|
||
|
|
vpunpckhdq $xc1,$xc0,$xc0
|
||
|
|
vpunpckhdq $xc3,$xc2,$xc2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xc1 # "c0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "c1"
|
||
|
|
vpunpcklqdq $xc2,$xc0,$xc3 # "c2"
|
||
|
|
vpunpckhqdq $xc2,$xc0,$xc0 # "c3"
|
||
|
|
___
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xt2)=($xc1,$xt2,$xc3,$xc0,$xc2);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd 0x100-0x100(%rcx),$xd0,$xd0
|
||
|
|
vpaddd 0x110-0x100(%rcx),$xd1,$xd1
|
||
|
|
vpaddd 0x120-0x100(%rcx),$xd2,$xd2
|
||
|
|
vpaddd 0x130-0x100(%rcx),$xd3,$xd3
|
||
|
|
|
||
|
|
vpunpckldq $xd1,$xd0,$xt2
|
||
|
|
vpunpckldq $xd3,$xd2,$xt3
|
||
|
|
vpunpckhdq $xd1,$xd0,$xd0
|
||
|
|
vpunpckhdq $xd3,$xd2,$xd2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xd1 # "d0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "d1"
|
||
|
|
vpunpcklqdq $xd2,$xd0,$xd3 # "d2"
|
||
|
|
vpunpckhqdq $xd2,$xd0,$xd0 # "d3"
|
||
|
|
___
|
||
|
|
($xd0,$xd1,$xd2,$xd3,$xt2)=($xd1,$xt2,$xd3,$xd0,$xd2);
|
||
|
|
($xa0,$xa1)=($xt2,$xt3);
|
||
|
|
$code.=<<___;
|
||
|
|
vmovdqa 0x00(%rsp),$xa0 # restore $xa0,1
|
||
|
|
vmovdqa 0x10(%rsp),$xa1
|
||
|
|
|
||
|
|
cmp \$64*4,$len
|
||
|
|
jb .Ltail4xop
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x10($inp),$xb0,$xb0
|
||
|
|
vpxor 0x20($inp),$xc0,$xc0
|
||
|
|
vpxor 0x30($inp),$xd0,$xd0
|
||
|
|
vpxor 0x40($inp),$xa1,$xa1
|
||
|
|
vpxor 0x50($inp),$xb1,$xb1
|
||
|
|
vpxor 0x60($inp),$xc1,$xc1
|
||
|
|
vpxor 0x70($inp),$xd1,$xd1
|
||
|
|
lea 0x80($inp),$inp # size optimization
|
||
|
|
vpxor 0x00($inp),$xa2,$xa2
|
||
|
|
vpxor 0x10($inp),$xb2,$xb2
|
||
|
|
vpxor 0x20($inp),$xc2,$xc2
|
||
|
|
vpxor 0x30($inp),$xd2,$xd2
|
||
|
|
vpxor 0x40($inp),$xa3,$xa3
|
||
|
|
vpxor 0x50($inp),$xb3,$xb3
|
||
|
|
vpxor 0x60($inp),$xc3,$xc3
|
||
|
|
vpxor 0x70($inp),$xd3,$xd3
|
||
|
|
lea 0x80($inp),$inp # inp+=64*4
|
||
|
|
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x10($out)
|
||
|
|
vmovdqu $xc0,0x20($out)
|
||
|
|
vmovdqu $xd0,0x30($out)
|
||
|
|
vmovdqu $xa1,0x40($out)
|
||
|
|
vmovdqu $xb1,0x50($out)
|
||
|
|
vmovdqu $xc1,0x60($out)
|
||
|
|
vmovdqu $xd1,0x70($out)
|
||
|
|
lea 0x80($out),$out # size optimization
|
||
|
|
vmovdqu $xa2,0x00($out)
|
||
|
|
vmovdqu $xb2,0x10($out)
|
||
|
|
vmovdqu $xc2,0x20($out)
|
||
|
|
vmovdqu $xd2,0x30($out)
|
||
|
|
vmovdqu $xa3,0x40($out)
|
||
|
|
vmovdqu $xb3,0x50($out)
|
||
|
|
vmovdqu $xc3,0x60($out)
|
||
|
|
vmovdqu $xd3,0x70($out)
|
||
|
|
lea 0x80($out),$out # out+=64*4
|
||
|
|
|
||
|
|
sub \$64*4,$len
|
||
|
|
jnz .Loop_outer4xop
|
||
|
|
|
||
|
|
jmp .Ldone4xop
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Ltail4xop:
|
||
|
|
cmp \$192,$len
|
||
|
|
jae .L192_or_more4xop
|
||
|
|
cmp \$128,$len
|
||
|
|
jae .L128_or_more4xop
|
||
|
|
cmp \$64,$len
|
||
|
|
jae .L64_or_more4xop
|
||
|
|
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xa0,0x00(%rsp)
|
||
|
|
vmovdqa $xb0,0x10(%rsp)
|
||
|
|
vmovdqa $xc0,0x20(%rsp)
|
||
|
|
vmovdqa $xd0,0x30(%rsp)
|
||
|
|
jmp .Loop_tail4xop
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L64_or_more4xop:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x10($inp),$xb0,$xb0
|
||
|
|
vpxor 0x20($inp),$xc0,$xc0
|
||
|
|
vpxor 0x30($inp),$xd0,$xd0
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x10($out)
|
||
|
|
vmovdqu $xc0,0x20($out)
|
||
|
|
vmovdqu $xd0,0x30($out)
|
||
|
|
je .Ldone4xop
|
||
|
|
|
||
|
|
lea 0x40($inp),$inp # inp+=64*1
|
||
|
|
vmovdqa $xa1,0x00(%rsp)
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xb1,0x10(%rsp)
|
||
|
|
lea 0x40($out),$out # out+=64*1
|
||
|
|
vmovdqa $xc1,0x20(%rsp)
|
||
|
|
sub \$64,$len # len-=64*1
|
||
|
|
vmovdqa $xd1,0x30(%rsp)
|
||
|
|
jmp .Loop_tail4xop
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L128_or_more4xop:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x10($inp),$xb0,$xb0
|
||
|
|
vpxor 0x20($inp),$xc0,$xc0
|
||
|
|
vpxor 0x30($inp),$xd0,$xd0
|
||
|
|
vpxor 0x40($inp),$xa1,$xa1
|
||
|
|
vpxor 0x50($inp),$xb1,$xb1
|
||
|
|
vpxor 0x60($inp),$xc1,$xc1
|
||
|
|
vpxor 0x70($inp),$xd1,$xd1
|
||
|
|
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x10($out)
|
||
|
|
vmovdqu $xc0,0x20($out)
|
||
|
|
vmovdqu $xd0,0x30($out)
|
||
|
|
vmovdqu $xa1,0x40($out)
|
||
|
|
vmovdqu $xb1,0x50($out)
|
||
|
|
vmovdqu $xc1,0x60($out)
|
||
|
|
vmovdqu $xd1,0x70($out)
|
||
|
|
je .Ldone4xop
|
||
|
|
|
||
|
|
lea 0x80($inp),$inp # inp+=64*2
|
||
|
|
vmovdqa $xa2,0x00(%rsp)
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xb2,0x10(%rsp)
|
||
|
|
lea 0x80($out),$out # out+=64*2
|
||
|
|
vmovdqa $xc2,0x20(%rsp)
|
||
|
|
sub \$128,$len # len-=64*2
|
||
|
|
vmovdqa $xd2,0x30(%rsp)
|
||
|
|
jmp .Loop_tail4xop
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L192_or_more4xop:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x10($inp),$xb0,$xb0
|
||
|
|
vpxor 0x20($inp),$xc0,$xc0
|
||
|
|
vpxor 0x30($inp),$xd0,$xd0
|
||
|
|
vpxor 0x40($inp),$xa1,$xa1
|
||
|
|
vpxor 0x50($inp),$xb1,$xb1
|
||
|
|
vpxor 0x60($inp),$xc1,$xc1
|
||
|
|
vpxor 0x70($inp),$xd1,$xd1
|
||
|
|
lea 0x80($inp),$inp # size optimization
|
||
|
|
vpxor 0x00($inp),$xa2,$xa2
|
||
|
|
vpxor 0x10($inp),$xb2,$xb2
|
||
|
|
vpxor 0x20($inp),$xc2,$xc2
|
||
|
|
vpxor 0x30($inp),$xd2,$xd2
|
||
|
|
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x10($out)
|
||
|
|
vmovdqu $xc0,0x20($out)
|
||
|
|
vmovdqu $xd0,0x30($out)
|
||
|
|
vmovdqu $xa1,0x40($out)
|
||
|
|
vmovdqu $xb1,0x50($out)
|
||
|
|
vmovdqu $xc1,0x60($out)
|
||
|
|
vmovdqu $xd1,0x70($out)
|
||
|
|
lea 0x80($out),$out # size optimization
|
||
|
|
vmovdqu $xa2,0x00($out)
|
||
|
|
vmovdqu $xb2,0x10($out)
|
||
|
|
vmovdqu $xc2,0x20($out)
|
||
|
|
vmovdqu $xd2,0x30($out)
|
||
|
|
je .Ldone4xop
|
||
|
|
|
||
|
|
lea 0x40($inp),$inp # inp+=64*3
|
||
|
|
vmovdqa $xa3,0x00(%rsp)
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xb3,0x10(%rsp)
|
||
|
|
lea 0x40($out),$out # out+=64*3
|
||
|
|
vmovdqa $xc3,0x20(%rsp)
|
||
|
|
sub \$192,$len # len-=64*3
|
||
|
|
vmovdqa $xd3,0x30(%rsp)
|
||
|
|
|
||
|
|
.Loop_tail4xop:
|
||
|
|
movzb ($inp,%r10),%eax
|
||
|
|
movzb (%rsp,%r10),%ecx
|
||
|
|
lea 1(%r10),%r10
|
||
|
|
xor %ecx,%eax
|
||
|
|
mov %al,-1($out,%r10)
|
||
|
|
dec $len
|
||
|
|
jnz .Loop_tail4xop
|
||
|
|
|
||
|
|
.Ldone4xop:
|
||
|
|
vzeroupper
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps -0xa8(%r9),%xmm6
|
||
|
|
movaps -0x98(%r9),%xmm7
|
||
|
|
movaps -0x88(%r9),%xmm8
|
||
|
|
movaps -0x78(%r9),%xmm9
|
||
|
|
movaps -0x68(%r9),%xmm10
|
||
|
|
movaps -0x58(%r9),%xmm11
|
||
|
|
movaps -0x48(%r9),%xmm12
|
||
|
|
movaps -0x38(%r9),%xmm13
|
||
|
|
movaps -0x28(%r9),%xmm14
|
||
|
|
movaps -0x18(%r9),%xmm15
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
lea (%r9),%rsp
|
||
|
|
.cfi_def_cfa_register %rsp
|
||
|
|
.L4xop_epilogue:
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size chacha20_4xop,.-chacha20_4xop
|
||
|
|
___
|
||
|
|
}
|
||
|
|
|
||
|
|
########################################################################
|
||
|
|
# AVX2 code path
|
||
|
|
if ($avx>1) {
|
||
|
|
my ($xb0,$xb1,$xb2,$xb3, $xd0,$xd1,$xd2,$xd3,
|
||
|
|
$xa0,$xa1,$xa2,$xa3, $xt0,$xt1,$xt2,$xt3)=map("%ymm$_",(0..15));
|
||
|
|
my @xx=($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,
|
||
|
|
"%nox","%nox","%nox","%nox", $xd0,$xd1,$xd2,$xd3);
|
||
|
|
|
||
|
|
sub AVX2_lane_ROUND {
|
||
|
|
my ($a0,$b0,$c0,$d0)=@_;
|
||
|
|
my ($a1,$b1,$c1,$d1)=map(($_&~3)+(($_+1)&3),($a0,$b0,$c0,$d0));
|
||
|
|
my ($a2,$b2,$c2,$d2)=map(($_&~3)+(($_+1)&3),($a1,$b1,$c1,$d1));
|
||
|
|
my ($a3,$b3,$c3,$d3)=map(($_&~3)+(($_+1)&3),($a2,$b2,$c2,$d2));
|
||
|
|
my ($xc,$xc_,$t0,$t1)=map("\"$_\"",$xt0,$xt1,$xt2,$xt3);
|
||
|
|
my @x=map("\"$_\"",@xx);
|
||
|
|
|
||
|
|
# Consider order in which variables are addressed by their
|
||
|
|
# index:
|
||
|
|
#
|
||
|
|
# a b c d
|
||
|
|
#
|
||
|
|
# 0 4 8 12 < even round
|
||
|
|
# 1 5 9 13
|
||
|
|
# 2 6 10 14
|
||
|
|
# 3 7 11 15
|
||
|
|
# 0 5 10 15 < odd round
|
||
|
|
# 1 6 11 12
|
||
|
|
# 2 7 8 13
|
||
|
|
# 3 4 9 14
|
||
|
|
#
|
||
|
|
# 'a', 'b' and 'd's are permanently allocated in registers,
|
||
|
|
# @x[0..7,12..15], while 'c's are maintained in memory. If
|
||
|
|
# you observe 'c' column, you'll notice that pair of 'c's is
|
||
|
|
# invariant between rounds. This means that we have to reload
|
||
|
|
# them once per round, in the middle. This is why you'll see
|
||
|
|
# bunch of 'c' stores and loads in the middle, but none in
|
||
|
|
# the beginning or end.
|
||
|
|
|
||
|
|
(
|
||
|
|
"&vpaddd (@x[$a0],@x[$a0],@x[$b0])", # Q1
|
||
|
|
"&vpxor (@x[$d0],@x[$a0],@x[$d0])",
|
||
|
|
"&vpshufb (@x[$d0],@x[$d0],$t1)",
|
||
|
|
"&vpaddd (@x[$a1],@x[$a1],@x[$b1])", # Q2
|
||
|
|
"&vpxor (@x[$d1],@x[$a1],@x[$d1])",
|
||
|
|
"&vpshufb (@x[$d1],@x[$d1],$t1)",
|
||
|
|
|
||
|
|
"&vpaddd ($xc,$xc,@x[$d0])",
|
||
|
|
"&vpxor (@x[$b0],$xc,@x[$b0])",
|
||
|
|
"&vpslld ($t0,@x[$b0],12)",
|
||
|
|
"&vpsrld (@x[$b0],@x[$b0],20)",
|
||
|
|
"&vpor (@x[$b0],$t0,@x[$b0])",
|
||
|
|
"&vbroadcasti128($t0,'(%r11)')", # .Lrot24(%rip)
|
||
|
|
"&vpaddd ($xc_,$xc_,@x[$d1])",
|
||
|
|
"&vpxor (@x[$b1],$xc_,@x[$b1])",
|
||
|
|
"&vpslld ($t1,@x[$b1],12)",
|
||
|
|
"&vpsrld (@x[$b1],@x[$b1],20)",
|
||
|
|
"&vpor (@x[$b1],$t1,@x[$b1])",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$a0],@x[$a0],@x[$b0])",
|
||
|
|
"&vpxor (@x[$d0],@x[$a0],@x[$d0])",
|
||
|
|
"&vpshufb (@x[$d0],@x[$d0],$t0)",
|
||
|
|
"&vpaddd (@x[$a1],@x[$a1],@x[$b1])",
|
||
|
|
"&vpxor (@x[$d1],@x[$a1],@x[$d1])",
|
||
|
|
"&vpshufb (@x[$d1],@x[$d1],$t0)",
|
||
|
|
|
||
|
|
"&vpaddd ($xc,$xc,@x[$d0])",
|
||
|
|
"&vpxor (@x[$b0],$xc,@x[$b0])",
|
||
|
|
"&vpslld ($t1,@x[$b0],7)",
|
||
|
|
"&vpsrld (@x[$b0],@x[$b0],25)",
|
||
|
|
"&vpor (@x[$b0],$t1,@x[$b0])",
|
||
|
|
"&vbroadcasti128($t1,'(%r10)')", # .Lrot16(%rip)
|
||
|
|
"&vpaddd ($xc_,$xc_,@x[$d1])",
|
||
|
|
"&vpxor (@x[$b1],$xc_,@x[$b1])",
|
||
|
|
"&vpslld ($t0,@x[$b1],7)",
|
||
|
|
"&vpsrld (@x[$b1],@x[$b1],25)",
|
||
|
|
"&vpor (@x[$b1],$t0,@x[$b1])",
|
||
|
|
|
||
|
|
"&vmovdqa (\"`32*($c0-8)`(%rsp)\",$xc)", # reload pair of 'c's
|
||
|
|
"&vmovdqa (\"`32*($c1-8)`(%rsp)\",$xc_)",
|
||
|
|
"&vmovdqa ($xc,\"`32*($c2-8)`(%rsp)\")",
|
||
|
|
"&vmovdqa ($xc_,\"`32*($c3-8)`(%rsp)\")",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$a2],@x[$a2],@x[$b2])", # Q3
|
||
|
|
"&vpxor (@x[$d2],@x[$a2],@x[$d2])",
|
||
|
|
"&vpshufb (@x[$d2],@x[$d2],$t1)",
|
||
|
|
"&vpaddd (@x[$a3],@x[$a3],@x[$b3])", # Q4
|
||
|
|
"&vpxor (@x[$d3],@x[$a3],@x[$d3])",
|
||
|
|
"&vpshufb (@x[$d3],@x[$d3],$t1)",
|
||
|
|
|
||
|
|
"&vpaddd ($xc,$xc,@x[$d2])",
|
||
|
|
"&vpxor (@x[$b2],$xc,@x[$b2])",
|
||
|
|
"&vpslld ($t0,@x[$b2],12)",
|
||
|
|
"&vpsrld (@x[$b2],@x[$b2],20)",
|
||
|
|
"&vpor (@x[$b2],$t0,@x[$b2])",
|
||
|
|
"&vbroadcasti128($t0,'(%r11)')", # .Lrot24(%rip)
|
||
|
|
"&vpaddd ($xc_,$xc_,@x[$d3])",
|
||
|
|
"&vpxor (@x[$b3],$xc_,@x[$b3])",
|
||
|
|
"&vpslld ($t1,@x[$b3],12)",
|
||
|
|
"&vpsrld (@x[$b3],@x[$b3],20)",
|
||
|
|
"&vpor (@x[$b3],$t1,@x[$b3])",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$a2],@x[$a2],@x[$b2])",
|
||
|
|
"&vpxor (@x[$d2],@x[$a2],@x[$d2])",
|
||
|
|
"&vpshufb (@x[$d2],@x[$d2],$t0)",
|
||
|
|
"&vpaddd (@x[$a3],@x[$a3],@x[$b3])",
|
||
|
|
"&vpxor (@x[$d3],@x[$a3],@x[$d3])",
|
||
|
|
"&vpshufb (@x[$d3],@x[$d3],$t0)",
|
||
|
|
|
||
|
|
"&vpaddd ($xc,$xc,@x[$d2])",
|
||
|
|
"&vpxor (@x[$b2],$xc,@x[$b2])",
|
||
|
|
"&vpslld ($t1,@x[$b2],7)",
|
||
|
|
"&vpsrld (@x[$b2],@x[$b2],25)",
|
||
|
|
"&vpor (@x[$b2],$t1,@x[$b2])",
|
||
|
|
"&vbroadcasti128($t1,'(%r10)')", # .Lrot16(%rip)
|
||
|
|
"&vpaddd ($xc_,$xc_,@x[$d3])",
|
||
|
|
"&vpxor (@x[$b3],$xc_,@x[$b3])",
|
||
|
|
"&vpslld ($t0,@x[$b3],7)",
|
||
|
|
"&vpsrld (@x[$b3],@x[$b3],25)",
|
||
|
|
"&vpor (@x[$b3],$t0,@x[$b3])"
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
my $xframe = $win64 ? 0xa8 : 8;
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.global chacha20_avx2
|
||
|
|
.type chacha20_avx2,\@function,5
|
||
|
|
.align 32
|
||
|
|
chacha20_avx2:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lchacha20_avx2:
|
||
|
|
mov %rsp,%r9 # frame register
|
||
|
|
.cfi_def_cfa_register %r9
|
||
|
|
sub \$0x280+$xframe,%rsp
|
||
|
|
and \$-32,%rsp
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps %xmm6,-0xa8(%r9)
|
||
|
|
movaps %xmm7,-0x98(%r9)
|
||
|
|
movaps %xmm8,-0x88(%r9)
|
||
|
|
movaps %xmm9,-0x78(%r9)
|
||
|
|
movaps %xmm10,-0x68(%r9)
|
||
|
|
movaps %xmm11,-0x58(%r9)
|
||
|
|
movaps %xmm12,-0x48(%r9)
|
||
|
|
movaps %xmm13,-0x38(%r9)
|
||
|
|
movaps %xmm14,-0x28(%r9)
|
||
|
|
movaps %xmm15,-0x18(%r9)
|
||
|
|
.L8x_body:
|
||
|
|
___
|
||
|
|
################ stack layout
|
||
|
|
# +0x00 SIMD equivalent of @x[8-12]
|
||
|
|
# ...
|
||
|
|
# +0x80 constant copy of key[0-2] smashed by lanes
|
||
|
|
# ...
|
||
|
|
# +0x200 SIMD counters (with nonce smashed by lanes)
|
||
|
|
# ...
|
||
|
|
# +0x280
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
vzeroupper
|
||
|
|
|
||
|
|
vbroadcasti128 .Lsigma(%rip),$xa3 # key[0]
|
||
|
|
vbroadcasti128 ($key),$xb3 # key[1]
|
||
|
|
vbroadcasti128 16($key),$xt3 # key[2]
|
||
|
|
vbroadcasti128 ($counter),$xd3 # key[3]
|
||
|
|
lea 0x100(%rsp),%rcx # size optimization
|
||
|
|
lea 0x200(%rsp),%rax # size optimization
|
||
|
|
lea .Lrot16(%rip),%r10
|
||
|
|
lea .Lrot24(%rip),%r11
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xa3,$xa0 # smash key by lanes...
|
||
|
|
vpshufd \$0x55,$xa3,$xa1
|
||
|
|
vmovdqa $xa0,0x80-0x100(%rcx) # ... and offload
|
||
|
|
vpshufd \$0xaa,$xa3,$xa2
|
||
|
|
vmovdqa $xa1,0xa0-0x100(%rcx)
|
||
|
|
vpshufd \$0xff,$xa3,$xa3
|
||
|
|
vmovdqa $xa2,0xc0-0x100(%rcx)
|
||
|
|
vmovdqa $xa3,0xe0-0x100(%rcx)
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xb3,$xb0
|
||
|
|
vpshufd \$0x55,$xb3,$xb1
|
||
|
|
vmovdqa $xb0,0x100-0x100(%rcx)
|
||
|
|
vpshufd \$0xaa,$xb3,$xb2
|
||
|
|
vmovdqa $xb1,0x120-0x100(%rcx)
|
||
|
|
vpshufd \$0xff,$xb3,$xb3
|
||
|
|
vmovdqa $xb2,0x140-0x100(%rcx)
|
||
|
|
vmovdqa $xb3,0x160-0x100(%rcx)
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xt3,$xt0 # "xc0"
|
||
|
|
vpshufd \$0x55,$xt3,$xt1 # "xc1"
|
||
|
|
vmovdqa $xt0,0x180-0x200(%rax)
|
||
|
|
vpshufd \$0xaa,$xt3,$xt2 # "xc2"
|
||
|
|
vmovdqa $xt1,0x1a0-0x200(%rax)
|
||
|
|
vpshufd \$0xff,$xt3,$xt3 # "xc3"
|
||
|
|
vmovdqa $xt2,0x1c0-0x200(%rax)
|
||
|
|
vmovdqa $xt3,0x1e0-0x200(%rax)
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xd3,$xd0
|
||
|
|
vpshufd \$0x55,$xd3,$xd1
|
||
|
|
vpaddd .Lincy(%rip),$xd0,$xd0 # don't save counters yet
|
||
|
|
vpshufd \$0xaa,$xd3,$xd2
|
||
|
|
vmovdqa $xd1,0x220-0x200(%rax)
|
||
|
|
vpshufd \$0xff,$xd3,$xd3
|
||
|
|
vmovdqa $xd2,0x240-0x200(%rax)
|
||
|
|
vmovdqa $xd3,0x260-0x200(%rax)
|
||
|
|
|
||
|
|
jmp .Loop_enter8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_outer8x:
|
||
|
|
vmovdqa 0x80-0x100(%rcx),$xa0 # re-load smashed key
|
||
|
|
vmovdqa 0xa0-0x100(%rcx),$xa1
|
||
|
|
vmovdqa 0xc0-0x100(%rcx),$xa2
|
||
|
|
vmovdqa 0xe0-0x100(%rcx),$xa3
|
||
|
|
vmovdqa 0x100-0x100(%rcx),$xb0
|
||
|
|
vmovdqa 0x120-0x100(%rcx),$xb1
|
||
|
|
vmovdqa 0x140-0x100(%rcx),$xb2
|
||
|
|
vmovdqa 0x160-0x100(%rcx),$xb3
|
||
|
|
vmovdqa 0x180-0x200(%rax),$xt0 # "xc0"
|
||
|
|
vmovdqa 0x1a0-0x200(%rax),$xt1 # "xc1"
|
||
|
|
vmovdqa 0x1c0-0x200(%rax),$xt2 # "xc2"
|
||
|
|
vmovdqa 0x1e0-0x200(%rax),$xt3 # "xc3"
|
||
|
|
vmovdqa 0x200-0x200(%rax),$xd0
|
||
|
|
vmovdqa 0x220-0x200(%rax),$xd1
|
||
|
|
vmovdqa 0x240-0x200(%rax),$xd2
|
||
|
|
vmovdqa 0x260-0x200(%rax),$xd3
|
||
|
|
vpaddd .Leight(%rip),$xd0,$xd0 # next SIMD counters
|
||
|
|
|
||
|
|
.Loop_enter8x:
|
||
|
|
vmovdqa $xt2,0x40(%rsp) # SIMD equivalent of "@x[10]"
|
||
|
|
vmovdqa $xt3,0x60(%rsp) # SIMD equivalent of "@x[11]"
|
||
|
|
vbroadcasti128 (%r10),$xt3
|
||
|
|
vmovdqa $xd0,0x200-0x200(%rax) # save SIMD counters
|
||
|
|
mov \$10,%eax
|
||
|
|
jmp .Loop8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop8x:
|
||
|
|
___
|
||
|
|
foreach (&AVX2_lane_ROUND(0, 4, 8,12)) { eval; }
|
||
|
|
foreach (&AVX2_lane_ROUND(0, 5,10,15)) { eval; }
|
||
|
|
$code.=<<___;
|
||
|
|
dec %eax
|
||
|
|
jnz .Loop8x
|
||
|
|
|
||
|
|
lea 0x200(%rsp),%rax # size optimization
|
||
|
|
vpaddd 0x80-0x100(%rcx),$xa0,$xa0 # accumulate key
|
||
|
|
vpaddd 0xa0-0x100(%rcx),$xa1,$xa1
|
||
|
|
vpaddd 0xc0-0x100(%rcx),$xa2,$xa2
|
||
|
|
vpaddd 0xe0-0x100(%rcx),$xa3,$xa3
|
||
|
|
|
||
|
|
vpunpckldq $xa1,$xa0,$xt2 # "de-interlace" data
|
||
|
|
vpunpckldq $xa3,$xa2,$xt3
|
||
|
|
vpunpckhdq $xa1,$xa0,$xa0
|
||
|
|
vpunpckhdq $xa3,$xa2,$xa2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xa1 # "a0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "a1"
|
||
|
|
vpunpcklqdq $xa2,$xa0,$xa3 # "a2"
|
||
|
|
vpunpckhqdq $xa2,$xa0,$xa0 # "a3"
|
||
|
|
___
|
||
|
|
($xa0,$xa1,$xa2,$xa3,$xt2)=($xa1,$xt2,$xa3,$xa0,$xa2);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd 0x100-0x100(%rcx),$xb0,$xb0
|
||
|
|
vpaddd 0x120-0x100(%rcx),$xb1,$xb1
|
||
|
|
vpaddd 0x140-0x100(%rcx),$xb2,$xb2
|
||
|
|
vpaddd 0x160-0x100(%rcx),$xb3,$xb3
|
||
|
|
|
||
|
|
vpunpckldq $xb1,$xb0,$xt2
|
||
|
|
vpunpckldq $xb3,$xb2,$xt3
|
||
|
|
vpunpckhdq $xb1,$xb0,$xb0
|
||
|
|
vpunpckhdq $xb3,$xb2,$xb2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xb1 # "b0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "b1"
|
||
|
|
vpunpcklqdq $xb2,$xb0,$xb3 # "b2"
|
||
|
|
vpunpckhqdq $xb2,$xb0,$xb0 # "b3"
|
||
|
|
___
|
||
|
|
($xb0,$xb1,$xb2,$xb3,$xt2)=($xb1,$xt2,$xb3,$xb0,$xb2);
|
||
|
|
$code.=<<___;
|
||
|
|
vperm2i128 \$0x20,$xb0,$xa0,$xt3 # "de-interlace" further
|
||
|
|
vperm2i128 \$0x31,$xb0,$xa0,$xb0
|
||
|
|
vperm2i128 \$0x20,$xb1,$xa1,$xa0
|
||
|
|
vperm2i128 \$0x31,$xb1,$xa1,$xb1
|
||
|
|
vperm2i128 \$0x20,$xb2,$xa2,$xa1
|
||
|
|
vperm2i128 \$0x31,$xb2,$xa2,$xb2
|
||
|
|
vperm2i128 \$0x20,$xb3,$xa3,$xa2
|
||
|
|
vperm2i128 \$0x31,$xb3,$xa3,$xb3
|
||
|
|
___
|
||
|
|
($xa0,$xa1,$xa2,$xa3,$xt3)=($xt3,$xa0,$xa1,$xa2,$xa3);
|
||
|
|
my ($xc0,$xc1,$xc2,$xc3)=($xt0,$xt1,$xa0,$xa1);
|
||
|
|
$code.=<<___;
|
||
|
|
vmovdqa $xa0,0x00(%rsp) # offload $xaN
|
||
|
|
vmovdqa $xa1,0x20(%rsp)
|
||
|
|
vmovdqa 0x40(%rsp),$xc2 # $xa0
|
||
|
|
vmovdqa 0x60(%rsp),$xc3 # $xa1
|
||
|
|
|
||
|
|
vpaddd 0x180-0x200(%rax),$xc0,$xc0
|
||
|
|
vpaddd 0x1a0-0x200(%rax),$xc1,$xc1
|
||
|
|
vpaddd 0x1c0-0x200(%rax),$xc2,$xc2
|
||
|
|
vpaddd 0x1e0-0x200(%rax),$xc3,$xc3
|
||
|
|
|
||
|
|
vpunpckldq $xc1,$xc0,$xt2
|
||
|
|
vpunpckldq $xc3,$xc2,$xt3
|
||
|
|
vpunpckhdq $xc1,$xc0,$xc0
|
||
|
|
vpunpckhdq $xc3,$xc2,$xc2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xc1 # "c0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "c1"
|
||
|
|
vpunpcklqdq $xc2,$xc0,$xc3 # "c2"
|
||
|
|
vpunpckhqdq $xc2,$xc0,$xc0 # "c3"
|
||
|
|
___
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xt2)=($xc1,$xt2,$xc3,$xc0,$xc2);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd 0x200-0x200(%rax),$xd0,$xd0
|
||
|
|
vpaddd 0x220-0x200(%rax),$xd1,$xd1
|
||
|
|
vpaddd 0x240-0x200(%rax),$xd2,$xd2
|
||
|
|
vpaddd 0x260-0x200(%rax),$xd3,$xd3
|
||
|
|
|
||
|
|
vpunpckldq $xd1,$xd0,$xt2
|
||
|
|
vpunpckldq $xd3,$xd2,$xt3
|
||
|
|
vpunpckhdq $xd1,$xd0,$xd0
|
||
|
|
vpunpckhdq $xd3,$xd2,$xd2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xd1 # "d0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "d1"
|
||
|
|
vpunpcklqdq $xd2,$xd0,$xd3 # "d2"
|
||
|
|
vpunpckhqdq $xd2,$xd0,$xd0 # "d3"
|
||
|
|
___
|
||
|
|
($xd0,$xd1,$xd2,$xd3,$xt2)=($xd1,$xt2,$xd3,$xd0,$xd2);
|
||
|
|
$code.=<<___;
|
||
|
|
vperm2i128 \$0x20,$xd0,$xc0,$xt3 # "de-interlace" further
|
||
|
|
vperm2i128 \$0x31,$xd0,$xc0,$xd0
|
||
|
|
vperm2i128 \$0x20,$xd1,$xc1,$xc0
|
||
|
|
vperm2i128 \$0x31,$xd1,$xc1,$xd1
|
||
|
|
vperm2i128 \$0x20,$xd2,$xc2,$xc1
|
||
|
|
vperm2i128 \$0x31,$xd2,$xc2,$xd2
|
||
|
|
vperm2i128 \$0x20,$xd3,$xc3,$xc2
|
||
|
|
vperm2i128 \$0x31,$xd3,$xc3,$xd3
|
||
|
|
___
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xt3)=($xt3,$xc0,$xc1,$xc2,$xc3);
|
||
|
|
($xb0,$xb1,$xb2,$xb3,$xc0,$xc1,$xc2,$xc3)=
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xb0,$xb1,$xb2,$xb3);
|
||
|
|
($xa0,$xa1)=($xt2,$xt3);
|
||
|
|
$code.=<<___;
|
||
|
|
vmovdqa 0x00(%rsp),$xa0 # $xaN was offloaded, remember?
|
||
|
|
vmovdqa 0x20(%rsp),$xa1
|
||
|
|
|
||
|
|
cmp \$64*8,$len
|
||
|
|
jb .Ltail8x
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vpxor 0x40($inp),$xc0,$xc0
|
||
|
|
vpxor 0x60($inp),$xd0,$xd0
|
||
|
|
lea 0x80($inp),$inp # size optimization
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
vmovdqu $xc0,0x40($out)
|
||
|
|
vmovdqu $xd0,0x60($out)
|
||
|
|
lea 0x80($out),$out # size optimization
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$xa1,$xa1
|
||
|
|
vpxor 0x20($inp),$xb1,$xb1
|
||
|
|
vpxor 0x40($inp),$xc1,$xc1
|
||
|
|
vpxor 0x60($inp),$xd1,$xd1
|
||
|
|
lea 0x80($inp),$inp # size optimization
|
||
|
|
vmovdqu $xa1,0x00($out)
|
||
|
|
vmovdqu $xb1,0x20($out)
|
||
|
|
vmovdqu $xc1,0x40($out)
|
||
|
|
vmovdqu $xd1,0x60($out)
|
||
|
|
lea 0x80($out),$out # size optimization
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$xa2,$xa2
|
||
|
|
vpxor 0x20($inp),$xb2,$xb2
|
||
|
|
vpxor 0x40($inp),$xc2,$xc2
|
||
|
|
vpxor 0x60($inp),$xd2,$xd2
|
||
|
|
lea 0x80($inp),$inp # size optimization
|
||
|
|
vmovdqu $xa2,0x00($out)
|
||
|
|
vmovdqu $xb2,0x20($out)
|
||
|
|
vmovdqu $xc2,0x40($out)
|
||
|
|
vmovdqu $xd2,0x60($out)
|
||
|
|
lea 0x80($out),$out # size optimization
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$xa3,$xa3
|
||
|
|
vpxor 0x20($inp),$xb3,$xb3
|
||
|
|
vpxor 0x40($inp),$xc3,$xc3
|
||
|
|
vpxor 0x60($inp),$xd3,$xd3
|
||
|
|
lea 0x80($inp),$inp # size optimization
|
||
|
|
vmovdqu $xa3,0x00($out)
|
||
|
|
vmovdqu $xb3,0x20($out)
|
||
|
|
vmovdqu $xc3,0x40($out)
|
||
|
|
vmovdqu $xd3,0x60($out)
|
||
|
|
lea 0x80($out),$out # size optimization
|
||
|
|
|
||
|
|
sub \$64*8,$len
|
||
|
|
jnz .Loop_outer8x
|
||
|
|
|
||
|
|
jmp .Ldone8x
|
||
|
|
|
||
|
|
.Ltail8x:
|
||
|
|
cmp \$448,$len
|
||
|
|
jae .L448_or_more8x
|
||
|
|
cmp \$384,$len
|
||
|
|
jae .L384_or_more8x
|
||
|
|
cmp \$320,$len
|
||
|
|
jae .L320_or_more8x
|
||
|
|
cmp \$256,$len
|
||
|
|
jae .L256_or_more8x
|
||
|
|
cmp \$192,$len
|
||
|
|
jae .L192_or_more8x
|
||
|
|
cmp \$128,$len
|
||
|
|
jae .L128_or_more8x
|
||
|
|
cmp \$64,$len
|
||
|
|
jae .L64_or_more8x
|
||
|
|
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xa0,0x00(%rsp)
|
||
|
|
vmovdqa $xb0,0x20(%rsp)
|
||
|
|
jmp .Loop_tail8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L64_or_more8x:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
je .Ldone8x
|
||
|
|
|
||
|
|
lea 0x40($inp),$inp # inp+=64*1
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xc0,0x00(%rsp)
|
||
|
|
lea 0x40($out),$out # out+=64*1
|
||
|
|
sub \$64,$len # len-=64*1
|
||
|
|
vmovdqa $xd0,0x20(%rsp)
|
||
|
|
jmp .Loop_tail8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L128_or_more8x:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vpxor 0x40($inp),$xc0,$xc0
|
||
|
|
vpxor 0x60($inp),$xd0,$xd0
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
vmovdqu $xc0,0x40($out)
|
||
|
|
vmovdqu $xd0,0x60($out)
|
||
|
|
je .Ldone8x
|
||
|
|
|
||
|
|
lea 0x80($inp),$inp # inp+=64*2
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xa1,0x00(%rsp)
|
||
|
|
lea 0x80($out),$out # out+=64*2
|
||
|
|
sub \$128,$len # len-=64*2
|
||
|
|
vmovdqa $xb1,0x20(%rsp)
|
||
|
|
jmp .Loop_tail8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L192_or_more8x:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vpxor 0x40($inp),$xc0,$xc0
|
||
|
|
vpxor 0x60($inp),$xd0,$xd0
|
||
|
|
vpxor 0x80($inp),$xa1,$xa1
|
||
|
|
vpxor 0xa0($inp),$xb1,$xb1
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
vmovdqu $xc0,0x40($out)
|
||
|
|
vmovdqu $xd0,0x60($out)
|
||
|
|
vmovdqu $xa1,0x80($out)
|
||
|
|
vmovdqu $xb1,0xa0($out)
|
||
|
|
je .Ldone8x
|
||
|
|
|
||
|
|
lea 0xc0($inp),$inp # inp+=64*3
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xc1,0x00(%rsp)
|
||
|
|
lea 0xc0($out),$out # out+=64*3
|
||
|
|
sub \$192,$len # len-=64*3
|
||
|
|
vmovdqa $xd1,0x20(%rsp)
|
||
|
|
jmp .Loop_tail8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L256_or_more8x:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vpxor 0x40($inp),$xc0,$xc0
|
||
|
|
vpxor 0x60($inp),$xd0,$xd0
|
||
|
|
vpxor 0x80($inp),$xa1,$xa1
|
||
|
|
vpxor 0xa0($inp),$xb1,$xb1
|
||
|
|
vpxor 0xc0($inp),$xc1,$xc1
|
||
|
|
vpxor 0xe0($inp),$xd1,$xd1
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
vmovdqu $xc0,0x40($out)
|
||
|
|
vmovdqu $xd0,0x60($out)
|
||
|
|
vmovdqu $xa1,0x80($out)
|
||
|
|
vmovdqu $xb1,0xa0($out)
|
||
|
|
vmovdqu $xc1,0xc0($out)
|
||
|
|
vmovdqu $xd1,0xe0($out)
|
||
|
|
je .Ldone8x
|
||
|
|
|
||
|
|
lea 0x100($inp),$inp # inp+=64*4
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xa2,0x00(%rsp)
|
||
|
|
lea 0x100($out),$out # out+=64*4
|
||
|
|
sub \$256,$len # len-=64*4
|
||
|
|
vmovdqa $xb2,0x20(%rsp)
|
||
|
|
jmp .Loop_tail8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L320_or_more8x:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vpxor 0x40($inp),$xc0,$xc0
|
||
|
|
vpxor 0x60($inp),$xd0,$xd0
|
||
|
|
vpxor 0x80($inp),$xa1,$xa1
|
||
|
|
vpxor 0xa0($inp),$xb1,$xb1
|
||
|
|
vpxor 0xc0($inp),$xc1,$xc1
|
||
|
|
vpxor 0xe0($inp),$xd1,$xd1
|
||
|
|
vpxor 0x100($inp),$xa2,$xa2
|
||
|
|
vpxor 0x120($inp),$xb2,$xb2
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
vmovdqu $xc0,0x40($out)
|
||
|
|
vmovdqu $xd0,0x60($out)
|
||
|
|
vmovdqu $xa1,0x80($out)
|
||
|
|
vmovdqu $xb1,0xa0($out)
|
||
|
|
vmovdqu $xc1,0xc0($out)
|
||
|
|
vmovdqu $xd1,0xe0($out)
|
||
|
|
vmovdqu $xa2,0x100($out)
|
||
|
|
vmovdqu $xb2,0x120($out)
|
||
|
|
je .Ldone8x
|
||
|
|
|
||
|
|
lea 0x140($inp),$inp # inp+=64*5
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xc2,0x00(%rsp)
|
||
|
|
lea 0x140($out),$out # out+=64*5
|
||
|
|
sub \$320,$len # len-=64*5
|
||
|
|
vmovdqa $xd2,0x20(%rsp)
|
||
|
|
jmp .Loop_tail8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L384_or_more8x:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vpxor 0x40($inp),$xc0,$xc0
|
||
|
|
vpxor 0x60($inp),$xd0,$xd0
|
||
|
|
vpxor 0x80($inp),$xa1,$xa1
|
||
|
|
vpxor 0xa0($inp),$xb1,$xb1
|
||
|
|
vpxor 0xc0($inp),$xc1,$xc1
|
||
|
|
vpxor 0xe0($inp),$xd1,$xd1
|
||
|
|
vpxor 0x100($inp),$xa2,$xa2
|
||
|
|
vpxor 0x120($inp),$xb2,$xb2
|
||
|
|
vpxor 0x140($inp),$xc2,$xc2
|
||
|
|
vpxor 0x160($inp),$xd2,$xd2
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
vmovdqu $xc0,0x40($out)
|
||
|
|
vmovdqu $xd0,0x60($out)
|
||
|
|
vmovdqu $xa1,0x80($out)
|
||
|
|
vmovdqu $xb1,0xa0($out)
|
||
|
|
vmovdqu $xc1,0xc0($out)
|
||
|
|
vmovdqu $xd1,0xe0($out)
|
||
|
|
vmovdqu $xa2,0x100($out)
|
||
|
|
vmovdqu $xb2,0x120($out)
|
||
|
|
vmovdqu $xc2,0x140($out)
|
||
|
|
vmovdqu $xd2,0x160($out)
|
||
|
|
je .Ldone8x
|
||
|
|
|
||
|
|
lea 0x180($inp),$inp # inp+=64*6
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xa3,0x00(%rsp)
|
||
|
|
lea 0x180($out),$out # out+=64*6
|
||
|
|
sub \$384,$len # len-=64*6
|
||
|
|
vmovdqa $xb3,0x20(%rsp)
|
||
|
|
jmp .Loop_tail8x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.L448_or_more8x:
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vpxor 0x40($inp),$xc0,$xc0
|
||
|
|
vpxor 0x60($inp),$xd0,$xd0
|
||
|
|
vpxor 0x80($inp),$xa1,$xa1
|
||
|
|
vpxor 0xa0($inp),$xb1,$xb1
|
||
|
|
vpxor 0xc0($inp),$xc1,$xc1
|
||
|
|
vpxor 0xe0($inp),$xd1,$xd1
|
||
|
|
vpxor 0x100($inp),$xa2,$xa2
|
||
|
|
vpxor 0x120($inp),$xb2,$xb2
|
||
|
|
vpxor 0x140($inp),$xc2,$xc2
|
||
|
|
vpxor 0x160($inp),$xd2,$xd2
|
||
|
|
vpxor 0x180($inp),$xa3,$xa3
|
||
|
|
vpxor 0x1a0($inp),$xb3,$xb3
|
||
|
|
vmovdqu $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
vmovdqu $xc0,0x40($out)
|
||
|
|
vmovdqu $xd0,0x60($out)
|
||
|
|
vmovdqu $xa1,0x80($out)
|
||
|
|
vmovdqu $xb1,0xa0($out)
|
||
|
|
vmovdqu $xc1,0xc0($out)
|
||
|
|
vmovdqu $xd1,0xe0($out)
|
||
|
|
vmovdqu $xa2,0x100($out)
|
||
|
|
vmovdqu $xb2,0x120($out)
|
||
|
|
vmovdqu $xc2,0x140($out)
|
||
|
|
vmovdqu $xd2,0x160($out)
|
||
|
|
vmovdqu $xa3,0x180($out)
|
||
|
|
vmovdqu $xb3,0x1a0($out)
|
||
|
|
je .Ldone8x
|
||
|
|
|
||
|
|
lea 0x1c0($inp),$inp # inp+=64*7
|
||
|
|
xor %r10,%r10
|
||
|
|
vmovdqa $xc3,0x00(%rsp)
|
||
|
|
lea 0x1c0($out),$out # out+=64*7
|
||
|
|
sub \$448,$len # len-=64*7
|
||
|
|
vmovdqa $xd3,0x20(%rsp)
|
||
|
|
|
||
|
|
.Loop_tail8x:
|
||
|
|
movzb ($inp,%r10),%eax
|
||
|
|
movzb (%rsp,%r10),%ecx
|
||
|
|
lea 1(%r10),%r10
|
||
|
|
xor %ecx,%eax
|
||
|
|
mov %al,-1($out,%r10)
|
||
|
|
dec $len
|
||
|
|
jnz .Loop_tail8x
|
||
|
|
|
||
|
|
.Ldone8x:
|
||
|
|
vzeroall
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps -0xa8(%r9),%xmm6
|
||
|
|
movaps -0x98(%r9),%xmm7
|
||
|
|
movaps -0x88(%r9),%xmm8
|
||
|
|
movaps -0x78(%r9),%xmm9
|
||
|
|
movaps -0x68(%r9),%xmm10
|
||
|
|
movaps -0x58(%r9),%xmm11
|
||
|
|
movaps -0x48(%r9),%xmm12
|
||
|
|
movaps -0x38(%r9),%xmm13
|
||
|
|
movaps -0x28(%r9),%xmm14
|
||
|
|
movaps -0x18(%r9),%xmm15
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
lea (%r9),%rsp
|
||
|
|
.cfi_def_cfa_register %rsp
|
||
|
|
.L8x_epilogue:
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size chacha20_avx2,.-chacha20_avx2
|
||
|
|
___
|
||
|
|
}
|
||
|
|
|
||
|
|
########################################################################
|
||
|
|
# AVX512 code paths
|
||
|
|
if ($avx>2) {
|
||
|
|
# This one handles shorter inputs...
|
||
|
|
|
||
|
|
my ($a,$b,$c,$d, $a_,$b_,$c_,$d_,$fourz) = map("%zmm$_",(0..3,16..20));
|
||
|
|
my ($t0,$t1,$t2,$t3) = map("%xmm$_",(4..7));
|
||
|
|
|
||
|
|
sub vpxord() # size optimization
|
||
|
|
{ my $opcode = "vpxor"; # adhere to vpxor when possible
|
||
|
|
|
||
|
|
foreach (@_) {
|
||
|
|
if (/%([zy])mm([0-9]+)/ && ($1 eq "z" || $2>=16)) {
|
||
|
|
$opcode = "vpxord";
|
||
|
|
last;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
$code .= "\t$opcode\t".join(',',reverse @_)."\n";
|
||
|
|
}
|
||
|
|
|
||
|
|
sub AVX512ROUND { # critical path is 14 "SIMD ticks" per round
|
||
|
|
&vpaddd ($a,$a,$b);
|
||
|
|
&vpxord ($d,$d,$a);
|
||
|
|
&vprold ($d,$d,16);
|
||
|
|
|
||
|
|
&vpaddd ($c,$c,$d);
|
||
|
|
&vpxord ($b,$b,$c);
|
||
|
|
&vprold ($b,$b,12);
|
||
|
|
|
||
|
|
&vpaddd ($a,$a,$b);
|
||
|
|
&vpxord ($d,$d,$a);
|
||
|
|
&vprold ($d,$d,8);
|
||
|
|
|
||
|
|
&vpaddd ($c,$c,$d);
|
||
|
|
&vpxord ($b,$b,$c);
|
||
|
|
&vprold ($b,$b,7);
|
||
|
|
}
|
||
|
|
|
||
|
|
my $xframe = $win64 ? 32+8 : 8;
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.global chacha20_avx512
|
||
|
|
.type chacha20_avx512,\@function,5
|
||
|
|
.align 32
|
||
|
|
chacha20_avx512:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lchacha20_avx512:
|
||
|
|
mov %rsp,%r9 # frame pointer
|
||
|
|
.cfi_def_cfa_register %r9
|
||
|
|
cmp \$512,$len
|
||
|
|
ja .Lchacha20_16x
|
||
|
|
|
||
|
|
sub \$64+$xframe,%rsp
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps %xmm6,-0x28(%r9)
|
||
|
|
movaps %xmm7,-0x18(%r9)
|
||
|
|
.Lavx512_body:
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
vbroadcasti32x4 .Lsigma(%rip),$a
|
||
|
|
vbroadcasti32x4 ($key),$b
|
||
|
|
vbroadcasti32x4 16($key),$c
|
||
|
|
vbroadcasti32x4 ($counter),$d
|
||
|
|
|
||
|
|
vmovdqa32 $a,$a_
|
||
|
|
vmovdqa32 $b,$b_
|
||
|
|
vmovdqa32 $c,$c_
|
||
|
|
vpaddd .Lzeroz(%rip),$d,$d
|
||
|
|
vmovdqa32 .Lfourz(%rip),$fourz
|
||
|
|
mov \$10,$counter # reuse $counter
|
||
|
|
vmovdqa32 $d,$d_
|
||
|
|
jmp .Loop_avx512
|
||
|
|
|
||
|
|
.align 16
|
||
|
|
.Loop_outer_avx512:
|
||
|
|
vmovdqa32 $a_,$a
|
||
|
|
vmovdqa32 $b_,$b
|
||
|
|
vmovdqa32 $c_,$c
|
||
|
|
vpaddd $fourz,$d_,$d
|
||
|
|
mov \$10,$counter
|
||
|
|
vmovdqa32 $d,$d_
|
||
|
|
jmp .Loop_avx512
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_avx512:
|
||
|
|
___
|
||
|
|
&AVX512ROUND();
|
||
|
|
&vpshufd ($c,$c,0b01001110);
|
||
|
|
&vpshufd ($b,$b,0b00111001);
|
||
|
|
&vpshufd ($d,$d,0b10010011);
|
||
|
|
|
||
|
|
&AVX512ROUND();
|
||
|
|
&vpshufd ($c,$c,0b01001110);
|
||
|
|
&vpshufd ($b,$b,0b10010011);
|
||
|
|
&vpshufd ($d,$d,0b00111001);
|
||
|
|
|
||
|
|
&dec ($counter);
|
||
|
|
&jnz (".Loop_avx512");
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd $a_,$a,$a
|
||
|
|
vpaddd $b_,$b,$b
|
||
|
|
vpaddd $c_,$c,$c
|
||
|
|
vpaddd $d_,$d,$d
|
||
|
|
|
||
|
|
sub \$64,$len
|
||
|
|
jb .Ltail64_avx512
|
||
|
|
|
||
|
|
vpxor 0x00($inp),%x#$a,$t0 # xor with input
|
||
|
|
vpxor 0x10($inp),%x#$b,$t1
|
||
|
|
vpxor 0x20($inp),%x#$c,$t2
|
||
|
|
vpxor 0x30($inp),%x#$d,$t3
|
||
|
|
lea 0x40($inp),$inp # inp+=64
|
||
|
|
|
||
|
|
vmovdqu $t0,0x00($out) # write output
|
||
|
|
vmovdqu $t1,0x10($out)
|
||
|
|
vmovdqu $t2,0x20($out)
|
||
|
|
vmovdqu $t3,0x30($out)
|
||
|
|
lea 0x40($out),$out # out+=64
|
||
|
|
|
||
|
|
jz .Ldone_avx512
|
||
|
|
|
||
|
|
vextracti32x4 \$1,$a,$t0
|
||
|
|
vextracti32x4 \$1,$b,$t1
|
||
|
|
vextracti32x4 \$1,$c,$t2
|
||
|
|
vextracti32x4 \$1,$d,$t3
|
||
|
|
|
||
|
|
sub \$64,$len
|
||
|
|
jb .Ltail_avx512
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$t0,$t0 # xor with input
|
||
|
|
vpxor 0x10($inp),$t1,$t1
|
||
|
|
vpxor 0x20($inp),$t2,$t2
|
||
|
|
vpxor 0x30($inp),$t3,$t3
|
||
|
|
lea 0x40($inp),$inp # inp+=64
|
||
|
|
|
||
|
|
vmovdqu $t0,0x00($out) # write output
|
||
|
|
vmovdqu $t1,0x10($out)
|
||
|
|
vmovdqu $t2,0x20($out)
|
||
|
|
vmovdqu $t3,0x30($out)
|
||
|
|
lea 0x40($out),$out # out+=64
|
||
|
|
|
||
|
|
jz .Ldone_avx512
|
||
|
|
|
||
|
|
vextracti32x4 \$2,$a,$t0
|
||
|
|
vextracti32x4 \$2,$b,$t1
|
||
|
|
vextracti32x4 \$2,$c,$t2
|
||
|
|
vextracti32x4 \$2,$d,$t3
|
||
|
|
|
||
|
|
sub \$64,$len
|
||
|
|
jb .Ltail_avx512
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$t0,$t0 # xor with input
|
||
|
|
vpxor 0x10($inp),$t1,$t1
|
||
|
|
vpxor 0x20($inp),$t2,$t2
|
||
|
|
vpxor 0x30($inp),$t3,$t3
|
||
|
|
lea 0x40($inp),$inp # inp+=64
|
||
|
|
|
||
|
|
vmovdqu $t0,0x00($out) # write output
|
||
|
|
vmovdqu $t1,0x10($out)
|
||
|
|
vmovdqu $t2,0x20($out)
|
||
|
|
vmovdqu $t3,0x30($out)
|
||
|
|
lea 0x40($out),$out # out+=64
|
||
|
|
|
||
|
|
jz .Ldone_avx512
|
||
|
|
|
||
|
|
vextracti32x4 \$3,$a,$t0
|
||
|
|
vextracti32x4 \$3,$b,$t1
|
||
|
|
vextracti32x4 \$3,$c,$t2
|
||
|
|
vextracti32x4 \$3,$d,$t3
|
||
|
|
|
||
|
|
sub \$64,$len
|
||
|
|
jb .Ltail_avx512
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$t0,$t0 # xor with input
|
||
|
|
vpxor 0x10($inp),$t1,$t1
|
||
|
|
vpxor 0x20($inp),$t2,$t2
|
||
|
|
vpxor 0x30($inp),$t3,$t3
|
||
|
|
lea 0x40($inp),$inp # inp+=64
|
||
|
|
|
||
|
|
vmovdqu $t0,0x00($out) # write output
|
||
|
|
vmovdqu $t1,0x10($out)
|
||
|
|
vmovdqu $t2,0x20($out)
|
||
|
|
vmovdqu $t3,0x30($out)
|
||
|
|
lea 0x40($out),$out # out+=64
|
||
|
|
|
||
|
|
jnz .Loop_outer_avx512
|
||
|
|
|
||
|
|
jmp .Ldone_avx512
|
||
|
|
|
||
|
|
.align 16
|
||
|
|
.Ltail64_avx512:
|
||
|
|
vmovdqa %x#$a,0x00(%rsp)
|
||
|
|
vmovdqa %x#$b,0x10(%rsp)
|
||
|
|
vmovdqa %x#$c,0x20(%rsp)
|
||
|
|
vmovdqa %x#$d,0x30(%rsp)
|
||
|
|
add \$64,$len
|
||
|
|
jmp .Loop_tail_avx512
|
||
|
|
|
||
|
|
.align 16
|
||
|
|
.Ltail_avx512:
|
||
|
|
vmovdqa $t0,0x00(%rsp)
|
||
|
|
vmovdqa $t1,0x10(%rsp)
|
||
|
|
vmovdqa $t2,0x20(%rsp)
|
||
|
|
vmovdqa $t3,0x30(%rsp)
|
||
|
|
add \$64,$len
|
||
|
|
|
||
|
|
.Loop_tail_avx512:
|
||
|
|
movzb ($inp,$counter),%eax
|
||
|
|
movzb (%rsp,$counter),%ecx
|
||
|
|
lea 1($counter),$counter
|
||
|
|
xor %ecx,%eax
|
||
|
|
mov %al,-1($out,$counter)
|
||
|
|
dec $len
|
||
|
|
jnz .Loop_tail_avx512
|
||
|
|
|
||
|
|
vmovdqu32 $a_,0x00(%rsp)
|
||
|
|
|
||
|
|
.Ldone_avx512:
|
||
|
|
vzeroall
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps -0x28(%r9),%xmm6
|
||
|
|
movaps -0x18(%r9),%xmm7
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
lea (%r9),%rsp
|
||
|
|
.cfi_def_cfa_register %rsp
|
||
|
|
.Lavx512_epilogue:
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size chacha20_avx512,.-chacha20_avx512
|
||
|
|
___
|
||
|
|
|
||
|
|
map(s/%z/%y/, $a,$b,$c,$d, $a_,$b_,$c_,$d_,$fourz);
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.global chacha20_avx512vl
|
||
|
|
.type chacha20_avx512vl,\@function,5
|
||
|
|
.align 32
|
||
|
|
chacha20_avx512vl:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lchacha20_avx512vl:
|
||
|
|
mov %rsp,%r9 # frame pointer
|
||
|
|
.cfi_def_cfa_register %r9
|
||
|
|
cmp \$128,$len
|
||
|
|
ja .Lchacha20_8xvl
|
||
|
|
|
||
|
|
sub \$64+$xframe,%rsp
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps %xmm6,-0x28(%r9)
|
||
|
|
movaps %xmm7,-0x18(%r9)
|
||
|
|
.Lavx512vl_body:
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
vbroadcasti128 .Lsigma(%rip),$a
|
||
|
|
vbroadcasti128 ($key),$b
|
||
|
|
vbroadcasti128 16($key),$c
|
||
|
|
vbroadcasti128 ($counter),$d
|
||
|
|
|
||
|
|
vmovdqa32 $a,$a_
|
||
|
|
vmovdqa32 $b,$b_
|
||
|
|
vmovdqa32 $c,$c_
|
||
|
|
vpaddd .Lzeroz(%rip),$d,$d
|
||
|
|
vmovdqa32 .Ltwoy(%rip),$fourz
|
||
|
|
mov \$10,$counter # reuse $counter
|
||
|
|
vmovdqa32 $d,$d_
|
||
|
|
jmp .Loop_avx512vl
|
||
|
|
|
||
|
|
.align 16
|
||
|
|
.Loop_outer_avx512vl:
|
||
|
|
vmovdqa32 $c_,$c
|
||
|
|
vpaddd $fourz,$d_,$d
|
||
|
|
mov \$10,$counter
|
||
|
|
vmovdqa32 $d,$d_
|
||
|
|
jmp .Loop_avx512vl
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_avx512vl:
|
||
|
|
___
|
||
|
|
&AVX512ROUND();
|
||
|
|
&vpshufd ($c,$c,0b01001110);
|
||
|
|
&vpshufd ($b,$b,0b00111001);
|
||
|
|
&vpshufd ($d,$d,0b10010011);
|
||
|
|
|
||
|
|
&AVX512ROUND();
|
||
|
|
&vpshufd ($c,$c,0b01001110);
|
||
|
|
&vpshufd ($b,$b,0b10010011);
|
||
|
|
&vpshufd ($d,$d,0b00111001);
|
||
|
|
|
||
|
|
&dec ($counter);
|
||
|
|
&jnz (".Loop_avx512vl");
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd $a_,$a,$a
|
||
|
|
vpaddd $b_,$b,$b
|
||
|
|
vpaddd $c_,$c,$c
|
||
|
|
vpaddd $d_,$d,$d
|
||
|
|
|
||
|
|
sub \$64,$len
|
||
|
|
jb .Ltail64_avx512vl
|
||
|
|
|
||
|
|
vpxor 0x00($inp),%x#$a,$t0 # xor with input
|
||
|
|
vpxor 0x10($inp),%x#$b,$t1
|
||
|
|
vpxor 0x20($inp),%x#$c,$t2
|
||
|
|
vpxor 0x30($inp),%x#$d,$t3
|
||
|
|
lea 0x40($inp),$inp # inp+=64
|
||
|
|
|
||
|
|
vmovdqu $t0,0x00($out) # write output
|
||
|
|
vmovdqu $t1,0x10($out)
|
||
|
|
vmovdqu $t2,0x20($out)
|
||
|
|
vmovdqu $t3,0x30($out)
|
||
|
|
lea 0x40($out),$out # out+=64
|
||
|
|
|
||
|
|
jz .Ldone_avx512vl
|
||
|
|
|
||
|
|
vextracti128 \$1,$a,$t0
|
||
|
|
vextracti128 \$1,$b,$t1
|
||
|
|
vextracti128 \$1,$c,$t2
|
||
|
|
vextracti128 \$1,$d,$t3
|
||
|
|
|
||
|
|
sub \$64,$len
|
||
|
|
jb .Ltail_avx512vl
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$t0,$t0 # xor with input
|
||
|
|
vpxor 0x10($inp),$t1,$t1
|
||
|
|
vpxor 0x20($inp),$t2,$t2
|
||
|
|
vpxor 0x30($inp),$t3,$t3
|
||
|
|
lea 0x40($inp),$inp # inp+=64
|
||
|
|
|
||
|
|
vmovdqu $t0,0x00($out) # write output
|
||
|
|
vmovdqu $t1,0x10($out)
|
||
|
|
vmovdqu $t2,0x20($out)
|
||
|
|
vmovdqu $t3,0x30($out)
|
||
|
|
lea 0x40($out),$out # out+=64
|
||
|
|
|
||
|
|
vmovdqa32 $a_,$a
|
||
|
|
vmovdqa32 $b_,$b
|
||
|
|
jnz .Loop_outer_avx512vl
|
||
|
|
|
||
|
|
jmp .Ldone_avx512vl
|
||
|
|
|
||
|
|
.align 16
|
||
|
|
.Ltail64_avx512vl:
|
||
|
|
vmovdqa %x#$a,0x00(%rsp)
|
||
|
|
vmovdqa %x#$b,0x10(%rsp)
|
||
|
|
vmovdqa %x#$c,0x20(%rsp)
|
||
|
|
vmovdqa %x#$d,0x30(%rsp)
|
||
|
|
add \$64,$len
|
||
|
|
jmp .Loop_tail_avx512vl
|
||
|
|
|
||
|
|
.align 16
|
||
|
|
.Ltail_avx512vl:
|
||
|
|
vmovdqa $t0,0x00(%rsp)
|
||
|
|
vmovdqa $t1,0x10(%rsp)
|
||
|
|
vmovdqa $t2,0x20(%rsp)
|
||
|
|
vmovdqa $t3,0x30(%rsp)
|
||
|
|
add \$64,$len
|
||
|
|
|
||
|
|
.Loop_tail_avx512vl:
|
||
|
|
movzb ($inp,$counter),%eax
|
||
|
|
movzb (%rsp,$counter),%ecx
|
||
|
|
lea 1($counter),$counter
|
||
|
|
xor %ecx,%eax
|
||
|
|
mov %al,-1($out,$counter)
|
||
|
|
dec $len
|
||
|
|
jnz .Loop_tail_avx512vl
|
||
|
|
|
||
|
|
vmovdqu32 $a_,0x00(%rsp)
|
||
|
|
vmovdqu32 $a_,0x20(%rsp)
|
||
|
|
|
||
|
|
.Ldone_avx512vl:
|
||
|
|
vzeroall
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps -0x28(%r9),%xmm6
|
||
|
|
movaps -0x18(%r9),%xmm7
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
lea (%r9),%rsp
|
||
|
|
.cfi_def_cfa_register %rsp
|
||
|
|
.Lavx512vl_epilogue:
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size chacha20_avx512vl,.-chacha20_avx512vl
|
||
|
|
___
|
||
|
|
}
|
||
|
|
if ($avx>2) {
|
||
|
|
# This one handles longer inputs...
|
||
|
|
|
||
|
|
my ($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,
|
||
|
|
$xc0,$xc1,$xc2,$xc3, $xd0,$xd1,$xd2,$xd3)=map("%zmm$_",(0..15));
|
||
|
|
my @xx=($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,
|
||
|
|
$xc0,$xc1,$xc2,$xc3, $xd0,$xd1,$xd2,$xd3);
|
||
|
|
my @key=map("%zmm$_",(16..31));
|
||
|
|
my ($xt0,$xt1,$xt2,$xt3)=@key[0..3];
|
||
|
|
|
||
|
|
sub AVX512_lane_ROUND {
|
||
|
|
my ($a0,$b0,$c0,$d0)=@_;
|
||
|
|
my ($a1,$b1,$c1,$d1)=map(($_&~3)+(($_+1)&3),($a0,$b0,$c0,$d0));
|
||
|
|
my ($a2,$b2,$c2,$d2)=map(($_&~3)+(($_+1)&3),($a1,$b1,$c1,$d1));
|
||
|
|
my ($a3,$b3,$c3,$d3)=map(($_&~3)+(($_+1)&3),($a2,$b2,$c2,$d2));
|
||
|
|
my @x=map("\"$_\"",@xx);
|
||
|
|
|
||
|
|
(
|
||
|
|
"&vpaddd (@x[$a0],@x[$a0],@x[$b0])", # Q1
|
||
|
|
"&vpaddd (@x[$a1],@x[$a1],@x[$b1])", # Q2
|
||
|
|
"&vpaddd (@x[$a2],@x[$a2],@x[$b2])", # Q3
|
||
|
|
"&vpaddd (@x[$a3],@x[$a3],@x[$b3])", # Q4
|
||
|
|
"&vpxord (@x[$d0],@x[$d0],@x[$a0])",
|
||
|
|
"&vpxord (@x[$d1],@x[$d1],@x[$a1])",
|
||
|
|
"&vpxord (@x[$d2],@x[$d2],@x[$a2])",
|
||
|
|
"&vpxord (@x[$d3],@x[$d3],@x[$a3])",
|
||
|
|
"&vprold (@x[$d0],@x[$d0],16)",
|
||
|
|
"&vprold (@x[$d1],@x[$d1],16)",
|
||
|
|
"&vprold (@x[$d2],@x[$d2],16)",
|
||
|
|
"&vprold (@x[$d3],@x[$d3],16)",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$c0],@x[$c0],@x[$d0])",
|
||
|
|
"&vpaddd (@x[$c1],@x[$c1],@x[$d1])",
|
||
|
|
"&vpaddd (@x[$c2],@x[$c2],@x[$d2])",
|
||
|
|
"&vpaddd (@x[$c3],@x[$c3],@x[$d3])",
|
||
|
|
"&vpxord (@x[$b0],@x[$b0],@x[$c0])",
|
||
|
|
"&vpxord (@x[$b1],@x[$b1],@x[$c1])",
|
||
|
|
"&vpxord (@x[$b2],@x[$b2],@x[$c2])",
|
||
|
|
"&vpxord (@x[$b3],@x[$b3],@x[$c3])",
|
||
|
|
"&vprold (@x[$b0],@x[$b0],12)",
|
||
|
|
"&vprold (@x[$b1],@x[$b1],12)",
|
||
|
|
"&vprold (@x[$b2],@x[$b2],12)",
|
||
|
|
"&vprold (@x[$b3],@x[$b3],12)",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$a0],@x[$a0],@x[$b0])",
|
||
|
|
"&vpaddd (@x[$a1],@x[$a1],@x[$b1])",
|
||
|
|
"&vpaddd (@x[$a2],@x[$a2],@x[$b2])",
|
||
|
|
"&vpaddd (@x[$a3],@x[$a3],@x[$b3])",
|
||
|
|
"&vpxord (@x[$d0],@x[$d0],@x[$a0])",
|
||
|
|
"&vpxord (@x[$d1],@x[$d1],@x[$a1])",
|
||
|
|
"&vpxord (@x[$d2],@x[$d2],@x[$a2])",
|
||
|
|
"&vpxord (@x[$d3],@x[$d3],@x[$a3])",
|
||
|
|
"&vprold (@x[$d0],@x[$d0],8)",
|
||
|
|
"&vprold (@x[$d1],@x[$d1],8)",
|
||
|
|
"&vprold (@x[$d2],@x[$d2],8)",
|
||
|
|
"&vprold (@x[$d3],@x[$d3],8)",
|
||
|
|
|
||
|
|
"&vpaddd (@x[$c0],@x[$c0],@x[$d0])",
|
||
|
|
"&vpaddd (@x[$c1],@x[$c1],@x[$d1])",
|
||
|
|
"&vpaddd (@x[$c2],@x[$c2],@x[$d2])",
|
||
|
|
"&vpaddd (@x[$c3],@x[$c3],@x[$d3])",
|
||
|
|
"&vpxord (@x[$b0],@x[$b0],@x[$c0])",
|
||
|
|
"&vpxord (@x[$b1],@x[$b1],@x[$c1])",
|
||
|
|
"&vpxord (@x[$b2],@x[$b2],@x[$c2])",
|
||
|
|
"&vpxord (@x[$b3],@x[$b3],@x[$c3])",
|
||
|
|
"&vprold (@x[$b0],@x[$b0],7)",
|
||
|
|
"&vprold (@x[$b1],@x[$b1],7)",
|
||
|
|
"&vprold (@x[$b2],@x[$b2],7)",
|
||
|
|
"&vprold (@x[$b3],@x[$b3],7)"
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
my $xframe = $win64 ? 0xa8 : 8;
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.global chacha20_16x
|
||
|
|
.type chacha20_16x,\@function,5
|
||
|
|
.align 32
|
||
|
|
chacha20_16x:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lchacha20_16x:
|
||
|
|
mov %rsp,%r9 # frame register
|
||
|
|
.cfi_def_cfa_register %r9
|
||
|
|
sub \$64+$xframe,%rsp
|
||
|
|
and \$-64,%rsp
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps %xmm6,-0xa8(%r9)
|
||
|
|
movaps %xmm7,-0x98(%r9)
|
||
|
|
movaps %xmm8,-0x88(%r9)
|
||
|
|
movaps %xmm9,-0x78(%r9)
|
||
|
|
movaps %xmm10,-0x68(%r9)
|
||
|
|
movaps %xmm11,-0x58(%r9)
|
||
|
|
movaps %xmm12,-0x48(%r9)
|
||
|
|
movaps %xmm13,-0x38(%r9)
|
||
|
|
movaps %xmm14,-0x28(%r9)
|
||
|
|
movaps %xmm15,-0x18(%r9)
|
||
|
|
.L16x_body:
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
vzeroupper
|
||
|
|
|
||
|
|
lea .Lsigma(%rip),%r10
|
||
|
|
vbroadcasti32x4 (%r10),$xa3 # key[0]
|
||
|
|
vbroadcasti32x4 ($key),$xb3 # key[1]
|
||
|
|
vbroadcasti32x4 16($key),$xc3 # key[2]
|
||
|
|
vbroadcasti32x4 ($counter),$xd3 # key[3]
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xa3,$xa0 # smash key by lanes...
|
||
|
|
vpshufd \$0x55,$xa3,$xa1
|
||
|
|
vpshufd \$0xaa,$xa3,$xa2
|
||
|
|
vpshufd \$0xff,$xa3,$xa3
|
||
|
|
vmovdqa64 $xa0,@key[0]
|
||
|
|
vmovdqa64 $xa1,@key[1]
|
||
|
|
vmovdqa64 $xa2,@key[2]
|
||
|
|
vmovdqa64 $xa3,@key[3]
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xb3,$xb0
|
||
|
|
vpshufd \$0x55,$xb3,$xb1
|
||
|
|
vpshufd \$0xaa,$xb3,$xb2
|
||
|
|
vpshufd \$0xff,$xb3,$xb3
|
||
|
|
vmovdqa64 $xb0,@key[4]
|
||
|
|
vmovdqa64 $xb1,@key[5]
|
||
|
|
vmovdqa64 $xb2,@key[6]
|
||
|
|
vmovdqa64 $xb3,@key[7]
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xc3,$xc0
|
||
|
|
vpshufd \$0x55,$xc3,$xc1
|
||
|
|
vpshufd \$0xaa,$xc3,$xc2
|
||
|
|
vpshufd \$0xff,$xc3,$xc3
|
||
|
|
vmovdqa64 $xc0,@key[8]
|
||
|
|
vmovdqa64 $xc1,@key[9]
|
||
|
|
vmovdqa64 $xc2,@key[10]
|
||
|
|
vmovdqa64 $xc3,@key[11]
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xd3,$xd0
|
||
|
|
vpshufd \$0x55,$xd3,$xd1
|
||
|
|
vpshufd \$0xaa,$xd3,$xd2
|
||
|
|
vpshufd \$0xff,$xd3,$xd3
|
||
|
|
vpaddd .Lincz(%rip),$xd0,$xd0 # don't save counters yet
|
||
|
|
vmovdqa64 $xd0,@key[12]
|
||
|
|
vmovdqa64 $xd1,@key[13]
|
||
|
|
vmovdqa64 $xd2,@key[14]
|
||
|
|
vmovdqa64 $xd3,@key[15]
|
||
|
|
|
||
|
|
mov \$10,%eax
|
||
|
|
jmp .Loop16x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_outer16x:
|
||
|
|
vpbroadcastd 0(%r10),$xa0 # reload key
|
||
|
|
vpbroadcastd 4(%r10),$xa1
|
||
|
|
vpbroadcastd 8(%r10),$xa2
|
||
|
|
vpbroadcastd 12(%r10),$xa3
|
||
|
|
vpaddd .Lsixteen(%rip),@key[12],@key[12] # next SIMD counters
|
||
|
|
vmovdqa64 @key[4],$xb0
|
||
|
|
vmovdqa64 @key[5],$xb1
|
||
|
|
vmovdqa64 @key[6],$xb2
|
||
|
|
vmovdqa64 @key[7],$xb3
|
||
|
|
vmovdqa64 @key[8],$xc0
|
||
|
|
vmovdqa64 @key[9],$xc1
|
||
|
|
vmovdqa64 @key[10],$xc2
|
||
|
|
vmovdqa64 @key[11],$xc3
|
||
|
|
vmovdqa64 @key[12],$xd0
|
||
|
|
vmovdqa64 @key[13],$xd1
|
||
|
|
vmovdqa64 @key[14],$xd2
|
||
|
|
vmovdqa64 @key[15],$xd3
|
||
|
|
|
||
|
|
vmovdqa64 $xa0,@key[0]
|
||
|
|
vmovdqa64 $xa1,@key[1]
|
||
|
|
vmovdqa64 $xa2,@key[2]
|
||
|
|
vmovdqa64 $xa3,@key[3]
|
||
|
|
|
||
|
|
mov \$10,%eax
|
||
|
|
jmp .Loop16x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop16x:
|
||
|
|
___
|
||
|
|
foreach (&AVX512_lane_ROUND(0, 4, 8,12)) { eval; }
|
||
|
|
foreach (&AVX512_lane_ROUND(0, 5,10,15)) { eval; }
|
||
|
|
$code.=<<___;
|
||
|
|
dec %eax
|
||
|
|
jnz .Loop16x
|
||
|
|
|
||
|
|
vpaddd @key[0],$xa0,$xa0 # accumulate key
|
||
|
|
vpaddd @key[1],$xa1,$xa1
|
||
|
|
vpaddd @key[2],$xa2,$xa2
|
||
|
|
vpaddd @key[3],$xa3,$xa3
|
||
|
|
|
||
|
|
vpunpckldq $xa1,$xa0,$xt2 # "de-interlace" data
|
||
|
|
vpunpckldq $xa3,$xa2,$xt3
|
||
|
|
vpunpckhdq $xa1,$xa0,$xa0
|
||
|
|
vpunpckhdq $xa3,$xa2,$xa2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xa1 # "a0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "a1"
|
||
|
|
vpunpcklqdq $xa2,$xa0,$xa3 # "a2"
|
||
|
|
vpunpckhqdq $xa2,$xa0,$xa0 # "a3"
|
||
|
|
___
|
||
|
|
($xa0,$xa1,$xa2,$xa3,$xt2)=($xa1,$xt2,$xa3,$xa0,$xa2);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd @key[4],$xb0,$xb0
|
||
|
|
vpaddd @key[5],$xb1,$xb1
|
||
|
|
vpaddd @key[6],$xb2,$xb2
|
||
|
|
vpaddd @key[7],$xb3,$xb3
|
||
|
|
|
||
|
|
vpunpckldq $xb1,$xb0,$xt2
|
||
|
|
vpunpckldq $xb3,$xb2,$xt3
|
||
|
|
vpunpckhdq $xb1,$xb0,$xb0
|
||
|
|
vpunpckhdq $xb3,$xb2,$xb2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xb1 # "b0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "b1"
|
||
|
|
vpunpcklqdq $xb2,$xb0,$xb3 # "b2"
|
||
|
|
vpunpckhqdq $xb2,$xb0,$xb0 # "b3"
|
||
|
|
___
|
||
|
|
($xb0,$xb1,$xb2,$xb3,$xt2)=($xb1,$xt2,$xb3,$xb0,$xb2);
|
||
|
|
$code.=<<___;
|
||
|
|
vshufi32x4 \$0x44,$xb0,$xa0,$xt3 # "de-interlace" further
|
||
|
|
vshufi32x4 \$0xee,$xb0,$xa0,$xb0
|
||
|
|
vshufi32x4 \$0x44,$xb1,$xa1,$xa0
|
||
|
|
vshufi32x4 \$0xee,$xb1,$xa1,$xb1
|
||
|
|
vshufi32x4 \$0x44,$xb2,$xa2,$xa1
|
||
|
|
vshufi32x4 \$0xee,$xb2,$xa2,$xb2
|
||
|
|
vshufi32x4 \$0x44,$xb3,$xa3,$xa2
|
||
|
|
vshufi32x4 \$0xee,$xb3,$xa3,$xb3
|
||
|
|
___
|
||
|
|
($xa0,$xa1,$xa2,$xa3,$xt3)=($xt3,$xa0,$xa1,$xa2,$xa3);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd @key[8],$xc0,$xc0
|
||
|
|
vpaddd @key[9],$xc1,$xc1
|
||
|
|
vpaddd @key[10],$xc2,$xc2
|
||
|
|
vpaddd @key[11],$xc3,$xc3
|
||
|
|
|
||
|
|
vpunpckldq $xc1,$xc0,$xt2
|
||
|
|
vpunpckldq $xc3,$xc2,$xt3
|
||
|
|
vpunpckhdq $xc1,$xc0,$xc0
|
||
|
|
vpunpckhdq $xc3,$xc2,$xc2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xc1 # "c0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "c1"
|
||
|
|
vpunpcklqdq $xc2,$xc0,$xc3 # "c2"
|
||
|
|
vpunpckhqdq $xc2,$xc0,$xc0 # "c3"
|
||
|
|
___
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xt2)=($xc1,$xt2,$xc3,$xc0,$xc2);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd @key[12],$xd0,$xd0
|
||
|
|
vpaddd @key[13],$xd1,$xd1
|
||
|
|
vpaddd @key[14],$xd2,$xd2
|
||
|
|
vpaddd @key[15],$xd3,$xd3
|
||
|
|
|
||
|
|
vpunpckldq $xd1,$xd0,$xt2
|
||
|
|
vpunpckldq $xd3,$xd2,$xt3
|
||
|
|
vpunpckhdq $xd1,$xd0,$xd0
|
||
|
|
vpunpckhdq $xd3,$xd2,$xd2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xd1 # "d0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "d1"
|
||
|
|
vpunpcklqdq $xd2,$xd0,$xd3 # "d2"
|
||
|
|
vpunpckhqdq $xd2,$xd0,$xd0 # "d3"
|
||
|
|
___
|
||
|
|
($xd0,$xd1,$xd2,$xd3,$xt2)=($xd1,$xt2,$xd3,$xd0,$xd2);
|
||
|
|
$code.=<<___;
|
||
|
|
vshufi32x4 \$0x44,$xd0,$xc0,$xt3 # "de-interlace" further
|
||
|
|
vshufi32x4 \$0xee,$xd0,$xc0,$xd0
|
||
|
|
vshufi32x4 \$0x44,$xd1,$xc1,$xc0
|
||
|
|
vshufi32x4 \$0xee,$xd1,$xc1,$xd1
|
||
|
|
vshufi32x4 \$0x44,$xd2,$xc2,$xc1
|
||
|
|
vshufi32x4 \$0xee,$xd2,$xc2,$xd2
|
||
|
|
vshufi32x4 \$0x44,$xd3,$xc3,$xc2
|
||
|
|
vshufi32x4 \$0xee,$xd3,$xc3,$xd3
|
||
|
|
___
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xt3)=($xt3,$xc0,$xc1,$xc2,$xc3);
|
||
|
|
$code.=<<___;
|
||
|
|
vshufi32x4 \$0x88,$xc0,$xa0,$xt0 # "de-interlace" further
|
||
|
|
vshufi32x4 \$0xdd,$xc0,$xa0,$xa0
|
||
|
|
vshufi32x4 \$0x88,$xd0,$xb0,$xc0
|
||
|
|
vshufi32x4 \$0xdd,$xd0,$xb0,$xd0
|
||
|
|
vshufi32x4 \$0x88,$xc1,$xa1,$xt1
|
||
|
|
vshufi32x4 \$0xdd,$xc1,$xa1,$xa1
|
||
|
|
vshufi32x4 \$0x88,$xd1,$xb1,$xc1
|
||
|
|
vshufi32x4 \$0xdd,$xd1,$xb1,$xd1
|
||
|
|
vshufi32x4 \$0x88,$xc2,$xa2,$xt2
|
||
|
|
vshufi32x4 \$0xdd,$xc2,$xa2,$xa2
|
||
|
|
vshufi32x4 \$0x88,$xd2,$xb2,$xc2
|
||
|
|
vshufi32x4 \$0xdd,$xd2,$xb2,$xd2
|
||
|
|
vshufi32x4 \$0x88,$xc3,$xa3,$xt3
|
||
|
|
vshufi32x4 \$0xdd,$xc3,$xa3,$xa3
|
||
|
|
vshufi32x4 \$0x88,$xd3,$xb3,$xc3
|
||
|
|
vshufi32x4 \$0xdd,$xd3,$xb3,$xd3
|
||
|
|
___
|
||
|
|
($xa0,$xa1,$xa2,$xa3,$xb0,$xb1,$xb2,$xb3)=
|
||
|
|
($xt0,$xt1,$xt2,$xt3,$xa0,$xa1,$xa2,$xa3);
|
||
|
|
|
||
|
|
($xa0,$xb0,$xc0,$xd0, $xa1,$xb1,$xc1,$xd1,
|
||
|
|
$xa2,$xb2,$xc2,$xd2, $xa3,$xb3,$xc3,$xd3) =
|
||
|
|
($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,
|
||
|
|
$xc0,$xc1,$xc2,$xc3, $xd0,$xd1,$xd2,$xd3);
|
||
|
|
$code.=<<___;
|
||
|
|
cmp \$64*16,$len
|
||
|
|
jb .Ltail16x
|
||
|
|
|
||
|
|
vpxord 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxord 0x40($inp),$xb0,$xb0
|
||
|
|
vpxord 0x80($inp),$xc0,$xc0
|
||
|
|
vpxord 0xc0($inp),$xd0,$xd0
|
||
|
|
vmovdqu32 $xa0,0x00($out)
|
||
|
|
vmovdqu32 $xb0,0x40($out)
|
||
|
|
vmovdqu32 $xc0,0x80($out)
|
||
|
|
vmovdqu32 $xd0,0xc0($out)
|
||
|
|
|
||
|
|
vpxord 0x100($inp),$xa1,$xa1
|
||
|
|
vpxord 0x140($inp),$xb1,$xb1
|
||
|
|
vpxord 0x180($inp),$xc1,$xc1
|
||
|
|
vpxord 0x1c0($inp),$xd1,$xd1
|
||
|
|
vmovdqu32 $xa1,0x100($out)
|
||
|
|
vmovdqu32 $xb1,0x140($out)
|
||
|
|
vmovdqu32 $xc1,0x180($out)
|
||
|
|
vmovdqu32 $xd1,0x1c0($out)
|
||
|
|
|
||
|
|
vpxord 0x200($inp),$xa2,$xa2
|
||
|
|
vpxord 0x240($inp),$xb2,$xb2
|
||
|
|
vpxord 0x280($inp),$xc2,$xc2
|
||
|
|
vpxord 0x2c0($inp),$xd2,$xd2
|
||
|
|
vmovdqu32 $xa2,0x200($out)
|
||
|
|
vmovdqu32 $xb2,0x240($out)
|
||
|
|
vmovdqu32 $xc2,0x280($out)
|
||
|
|
vmovdqu32 $xd2,0x2c0($out)
|
||
|
|
|
||
|
|
vpxord 0x300($inp),$xa3,$xa3
|
||
|
|
vpxord 0x340($inp),$xb3,$xb3
|
||
|
|
vpxord 0x380($inp),$xc3,$xc3
|
||
|
|
vpxord 0x3c0($inp),$xd3,$xd3
|
||
|
|
lea 0x400($inp),$inp
|
||
|
|
vmovdqu32 $xa3,0x300($out)
|
||
|
|
vmovdqu32 $xb3,0x340($out)
|
||
|
|
vmovdqu32 $xc3,0x380($out)
|
||
|
|
vmovdqu32 $xd3,0x3c0($out)
|
||
|
|
lea 0x400($out),$out
|
||
|
|
|
||
|
|
sub \$64*16,$len
|
||
|
|
jnz .Loop_outer16x
|
||
|
|
|
||
|
|
jmp .Ldone16x
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Ltail16x:
|
||
|
|
xor %r10,%r10
|
||
|
|
sub $inp,$out
|
||
|
|
cmp \$64*1,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xa0,$xa0 # xor with input
|
||
|
|
vmovdqu32 $xa0,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xb0,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*2,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xb0,$xb0
|
||
|
|
vmovdqu32 $xb0,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xc0,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*3,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xc0,$xc0
|
||
|
|
vmovdqu32 $xc0,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xd0,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*4,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xd0,$xd0
|
||
|
|
vmovdqu32 $xd0,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xa1,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*5,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xa1,$xa1
|
||
|
|
vmovdqu32 $xa1,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xb1,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*6,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xb1,$xb1
|
||
|
|
vmovdqu32 $xb1,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xc1,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*7,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xc1,$xc1
|
||
|
|
vmovdqu32 $xc1,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xd1,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*8,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xd1,$xd1
|
||
|
|
vmovdqu32 $xd1,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xa2,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*9,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xa2,$xa2
|
||
|
|
vmovdqu32 $xa2,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xb2,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*10,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xb2,$xb2
|
||
|
|
vmovdqu32 $xb2,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xc2,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*11,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xc2,$xc2
|
||
|
|
vmovdqu32 $xc2,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xd2,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*12,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xd2,$xd2
|
||
|
|
vmovdqu32 $xd2,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xa3,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*13,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xa3,$xa3
|
||
|
|
vmovdqu32 $xa3,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xb3,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*14,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xb3,$xb3
|
||
|
|
vmovdqu32 $xb3,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xc3,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*15,$len
|
||
|
|
jb .Less_than_64_16x
|
||
|
|
vpxord ($inp),$xc3,$xc3
|
||
|
|
vmovdqu32 $xc3,($out,$inp)
|
||
|
|
je .Ldone16x
|
||
|
|
vmovdqa32 $xd3,$xa0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
.Less_than_64_16x:
|
||
|
|
vmovdqa32 $xa0,0x00(%rsp)
|
||
|
|
lea ($out,$inp),$out
|
||
|
|
and \$63,$len
|
||
|
|
|
||
|
|
.Loop_tail16x:
|
||
|
|
movzb ($inp,%r10),%eax
|
||
|
|
movzb (%rsp,%r10),%ecx
|
||
|
|
lea 1(%r10),%r10
|
||
|
|
xor %ecx,%eax
|
||
|
|
mov %al,-1($out,%r10)
|
||
|
|
dec $len
|
||
|
|
jnz .Loop_tail16x
|
||
|
|
|
||
|
|
vpxord $xa0,$xa0,$xa0
|
||
|
|
vmovdqa32 $xa0,0(%rsp)
|
||
|
|
|
||
|
|
.Ldone16x:
|
||
|
|
vzeroall
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps -0xa8(%r9),%xmm6
|
||
|
|
movaps -0x98(%r9),%xmm7
|
||
|
|
movaps -0x88(%r9),%xmm8
|
||
|
|
movaps -0x78(%r9),%xmm9
|
||
|
|
movaps -0x68(%r9),%xmm10
|
||
|
|
movaps -0x58(%r9),%xmm11
|
||
|
|
movaps -0x48(%r9),%xmm12
|
||
|
|
movaps -0x38(%r9),%xmm13
|
||
|
|
movaps -0x28(%r9),%xmm14
|
||
|
|
movaps -0x18(%r9),%xmm15
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
lea (%r9),%rsp
|
||
|
|
.cfi_def_cfa_register %rsp
|
||
|
|
.L16x_epilogue:
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size chacha20_16x,.-chacha20_16x
|
||
|
|
___
|
||
|
|
|
||
|
|
# switch to %ymm domain
|
||
|
|
($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,
|
||
|
|
$xc0,$xc1,$xc2,$xc3, $xd0,$xd1,$xd2,$xd3)=map("%ymm$_",(0..15));
|
||
|
|
@xx=($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,
|
||
|
|
$xc0,$xc1,$xc2,$xc3, $xd0,$xd1,$xd2,$xd3);
|
||
|
|
@key=map("%ymm$_",(16..31));
|
||
|
|
($xt0,$xt1,$xt2,$xt3)=@key[0..3];
|
||
|
|
|
||
|
|
$code.=<<___;
|
||
|
|
.global chacha20_8xvl
|
||
|
|
.type chacha20_8xvl,\@function,5
|
||
|
|
.align 32
|
||
|
|
chacha20_8xvl:
|
||
|
|
.cfi_startproc
|
||
|
|
.Lchacha20_8xvl:
|
||
|
|
mov %rsp,%r9 # frame register
|
||
|
|
.cfi_def_cfa_register %r9
|
||
|
|
sub \$64+$xframe,%rsp
|
||
|
|
and \$-64,%rsp
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps %xmm6,-0xa8(%r9)
|
||
|
|
movaps %xmm7,-0x98(%r9)
|
||
|
|
movaps %xmm8,-0x88(%r9)
|
||
|
|
movaps %xmm9,-0x78(%r9)
|
||
|
|
movaps %xmm10,-0x68(%r9)
|
||
|
|
movaps %xmm11,-0x58(%r9)
|
||
|
|
movaps %xmm12,-0x48(%r9)
|
||
|
|
movaps %xmm13,-0x38(%r9)
|
||
|
|
movaps %xmm14,-0x28(%r9)
|
||
|
|
movaps %xmm15,-0x18(%r9)
|
||
|
|
.L8xvl_body:
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
vzeroupper
|
||
|
|
|
||
|
|
lea .Lsigma(%rip),%r10
|
||
|
|
vbroadcasti128 (%r10),$xa3 # key[0]
|
||
|
|
vbroadcasti128 ($key),$xb3 # key[1]
|
||
|
|
vbroadcasti128 16($key),$xc3 # key[2]
|
||
|
|
vbroadcasti128 ($counter),$xd3 # key[3]
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xa3,$xa0 # smash key by lanes...
|
||
|
|
vpshufd \$0x55,$xa3,$xa1
|
||
|
|
vpshufd \$0xaa,$xa3,$xa2
|
||
|
|
vpshufd \$0xff,$xa3,$xa3
|
||
|
|
vmovdqa64 $xa0,@key[0]
|
||
|
|
vmovdqa64 $xa1,@key[1]
|
||
|
|
vmovdqa64 $xa2,@key[2]
|
||
|
|
vmovdqa64 $xa3,@key[3]
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xb3,$xb0
|
||
|
|
vpshufd \$0x55,$xb3,$xb1
|
||
|
|
vpshufd \$0xaa,$xb3,$xb2
|
||
|
|
vpshufd \$0xff,$xb3,$xb3
|
||
|
|
vmovdqa64 $xb0,@key[4]
|
||
|
|
vmovdqa64 $xb1,@key[5]
|
||
|
|
vmovdqa64 $xb2,@key[6]
|
||
|
|
vmovdqa64 $xb3,@key[7]
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xc3,$xc0
|
||
|
|
vpshufd \$0x55,$xc3,$xc1
|
||
|
|
vpshufd \$0xaa,$xc3,$xc2
|
||
|
|
vpshufd \$0xff,$xc3,$xc3
|
||
|
|
vmovdqa64 $xc0,@key[8]
|
||
|
|
vmovdqa64 $xc1,@key[9]
|
||
|
|
vmovdqa64 $xc2,@key[10]
|
||
|
|
vmovdqa64 $xc3,@key[11]
|
||
|
|
|
||
|
|
vpshufd \$0x00,$xd3,$xd0
|
||
|
|
vpshufd \$0x55,$xd3,$xd1
|
||
|
|
vpshufd \$0xaa,$xd3,$xd2
|
||
|
|
vpshufd \$0xff,$xd3,$xd3
|
||
|
|
vpaddd .Lincy(%rip),$xd0,$xd0 # don't save counters yet
|
||
|
|
vmovdqa64 $xd0,@key[12]
|
||
|
|
vmovdqa64 $xd1,@key[13]
|
||
|
|
vmovdqa64 $xd2,@key[14]
|
||
|
|
vmovdqa64 $xd3,@key[15]
|
||
|
|
|
||
|
|
mov \$10,%eax
|
||
|
|
jmp .Loop8xvl
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop_outer8xvl:
|
||
|
|
#vpbroadcastd 0(%r10),$xa0 # reload key
|
||
|
|
#vpbroadcastd 4(%r10),$xa1
|
||
|
|
vpbroadcastd 8(%r10),$xa2
|
||
|
|
vpbroadcastd 12(%r10),$xa3
|
||
|
|
vpaddd .Leight(%rip),@key[12],@key[12] # next SIMD counters
|
||
|
|
vmovdqa64 @key[4],$xb0
|
||
|
|
vmovdqa64 @key[5],$xb1
|
||
|
|
vmovdqa64 @key[6],$xb2
|
||
|
|
vmovdqa64 @key[7],$xb3
|
||
|
|
vmovdqa64 @key[8],$xc0
|
||
|
|
vmovdqa64 @key[9],$xc1
|
||
|
|
vmovdqa64 @key[10],$xc2
|
||
|
|
vmovdqa64 @key[11],$xc3
|
||
|
|
vmovdqa64 @key[12],$xd0
|
||
|
|
vmovdqa64 @key[13],$xd1
|
||
|
|
vmovdqa64 @key[14],$xd2
|
||
|
|
vmovdqa64 @key[15],$xd3
|
||
|
|
|
||
|
|
vmovdqa64 $xa0,@key[0]
|
||
|
|
vmovdqa64 $xa1,@key[1]
|
||
|
|
vmovdqa64 $xa2,@key[2]
|
||
|
|
vmovdqa64 $xa3,@key[3]
|
||
|
|
|
||
|
|
mov \$10,%eax
|
||
|
|
jmp .Loop8xvl
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Loop8xvl:
|
||
|
|
___
|
||
|
|
foreach (&AVX512_lane_ROUND(0, 4, 8,12)) { eval; }
|
||
|
|
foreach (&AVX512_lane_ROUND(0, 5,10,15)) { eval; }
|
||
|
|
$code.=<<___;
|
||
|
|
dec %eax
|
||
|
|
jnz .Loop8xvl
|
||
|
|
|
||
|
|
vpaddd @key[0],$xa0,$xa0 # accumulate key
|
||
|
|
vpaddd @key[1],$xa1,$xa1
|
||
|
|
vpaddd @key[2],$xa2,$xa2
|
||
|
|
vpaddd @key[3],$xa3,$xa3
|
||
|
|
|
||
|
|
vpunpckldq $xa1,$xa0,$xt2 # "de-interlace" data
|
||
|
|
vpunpckldq $xa3,$xa2,$xt3
|
||
|
|
vpunpckhdq $xa1,$xa0,$xa0
|
||
|
|
vpunpckhdq $xa3,$xa2,$xa2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xa1 # "a0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "a1"
|
||
|
|
vpunpcklqdq $xa2,$xa0,$xa3 # "a2"
|
||
|
|
vpunpckhqdq $xa2,$xa0,$xa0 # "a3"
|
||
|
|
___
|
||
|
|
($xa0,$xa1,$xa2,$xa3,$xt2)=($xa1,$xt2,$xa3,$xa0,$xa2);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd @key[4],$xb0,$xb0
|
||
|
|
vpaddd @key[5],$xb1,$xb1
|
||
|
|
vpaddd @key[6],$xb2,$xb2
|
||
|
|
vpaddd @key[7],$xb3,$xb3
|
||
|
|
|
||
|
|
vpunpckldq $xb1,$xb0,$xt2
|
||
|
|
vpunpckldq $xb3,$xb2,$xt3
|
||
|
|
vpunpckhdq $xb1,$xb0,$xb0
|
||
|
|
vpunpckhdq $xb3,$xb2,$xb2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xb1 # "b0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "b1"
|
||
|
|
vpunpcklqdq $xb2,$xb0,$xb3 # "b2"
|
||
|
|
vpunpckhqdq $xb2,$xb0,$xb0 # "b3"
|
||
|
|
___
|
||
|
|
($xb0,$xb1,$xb2,$xb3,$xt2)=($xb1,$xt2,$xb3,$xb0,$xb2);
|
||
|
|
$code.=<<___;
|
||
|
|
vshufi32x4 \$0,$xb0,$xa0,$xt3 # "de-interlace" further
|
||
|
|
vshufi32x4 \$3,$xb0,$xa0,$xb0
|
||
|
|
vshufi32x4 \$0,$xb1,$xa1,$xa0
|
||
|
|
vshufi32x4 \$3,$xb1,$xa1,$xb1
|
||
|
|
vshufi32x4 \$0,$xb2,$xa2,$xa1
|
||
|
|
vshufi32x4 \$3,$xb2,$xa2,$xb2
|
||
|
|
vshufi32x4 \$0,$xb3,$xa3,$xa2
|
||
|
|
vshufi32x4 \$3,$xb3,$xa3,$xb3
|
||
|
|
___
|
||
|
|
($xa0,$xa1,$xa2,$xa3,$xt3)=($xt3,$xa0,$xa1,$xa2,$xa3);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd @key[8],$xc0,$xc0
|
||
|
|
vpaddd @key[9],$xc1,$xc1
|
||
|
|
vpaddd @key[10],$xc2,$xc2
|
||
|
|
vpaddd @key[11],$xc3,$xc3
|
||
|
|
|
||
|
|
vpunpckldq $xc1,$xc0,$xt2
|
||
|
|
vpunpckldq $xc3,$xc2,$xt3
|
||
|
|
vpunpckhdq $xc1,$xc0,$xc0
|
||
|
|
vpunpckhdq $xc3,$xc2,$xc2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xc1 # "c0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "c1"
|
||
|
|
vpunpcklqdq $xc2,$xc0,$xc3 # "c2"
|
||
|
|
vpunpckhqdq $xc2,$xc0,$xc0 # "c3"
|
||
|
|
___
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xt2)=($xc1,$xt2,$xc3,$xc0,$xc2);
|
||
|
|
$code.=<<___;
|
||
|
|
vpaddd @key[12],$xd0,$xd0
|
||
|
|
vpaddd @key[13],$xd1,$xd1
|
||
|
|
vpaddd @key[14],$xd2,$xd2
|
||
|
|
vpaddd @key[15],$xd3,$xd3
|
||
|
|
|
||
|
|
vpunpckldq $xd1,$xd0,$xt2
|
||
|
|
vpunpckldq $xd3,$xd2,$xt3
|
||
|
|
vpunpckhdq $xd1,$xd0,$xd0
|
||
|
|
vpunpckhdq $xd3,$xd2,$xd2
|
||
|
|
vpunpcklqdq $xt3,$xt2,$xd1 # "d0"
|
||
|
|
vpunpckhqdq $xt3,$xt2,$xt2 # "d1"
|
||
|
|
vpunpcklqdq $xd2,$xd0,$xd3 # "d2"
|
||
|
|
vpunpckhqdq $xd2,$xd0,$xd0 # "d3"
|
||
|
|
___
|
||
|
|
($xd0,$xd1,$xd2,$xd3,$xt2)=($xd1,$xt2,$xd3,$xd0,$xd2);
|
||
|
|
$code.=<<___;
|
||
|
|
vperm2i128 \$0x20,$xd0,$xc0,$xt3 # "de-interlace" further
|
||
|
|
vperm2i128 \$0x31,$xd0,$xc0,$xd0
|
||
|
|
vperm2i128 \$0x20,$xd1,$xc1,$xc0
|
||
|
|
vperm2i128 \$0x31,$xd1,$xc1,$xd1
|
||
|
|
vperm2i128 \$0x20,$xd2,$xc2,$xc1
|
||
|
|
vperm2i128 \$0x31,$xd2,$xc2,$xd2
|
||
|
|
vperm2i128 \$0x20,$xd3,$xc3,$xc2
|
||
|
|
vperm2i128 \$0x31,$xd3,$xc3,$xd3
|
||
|
|
___
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xt3)=($xt3,$xc0,$xc1,$xc2,$xc3);
|
||
|
|
($xb0,$xb1,$xb2,$xb3,$xc0,$xc1,$xc2,$xc3)=
|
||
|
|
($xc0,$xc1,$xc2,$xc3,$xb0,$xb1,$xb2,$xb3);
|
||
|
|
$code.=<<___;
|
||
|
|
cmp \$64*8,$len
|
||
|
|
jb .Ltail8xvl
|
||
|
|
|
||
|
|
mov \$0x80,%eax # size optimization
|
||
|
|
vpxord 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vpxor 0x40($inp),$xc0,$xc0
|
||
|
|
vpxor 0x60($inp),$xd0,$xd0
|
||
|
|
lea ($inp,%rax),$inp # size optimization
|
||
|
|
vmovdqu32 $xa0,0x00($out)
|
||
|
|
vmovdqu $xb0,0x20($out)
|
||
|
|
vmovdqu $xc0,0x40($out)
|
||
|
|
vmovdqu $xd0,0x60($out)
|
||
|
|
lea ($out,%rax),$out # size optimization
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$xa1,$xa1
|
||
|
|
vpxor 0x20($inp),$xb1,$xb1
|
||
|
|
vpxor 0x40($inp),$xc1,$xc1
|
||
|
|
vpxor 0x60($inp),$xd1,$xd1
|
||
|
|
lea ($inp,%rax),$inp # size optimization
|
||
|
|
vmovdqu $xa1,0x00($out)
|
||
|
|
vmovdqu $xb1,0x20($out)
|
||
|
|
vmovdqu $xc1,0x40($out)
|
||
|
|
vmovdqu $xd1,0x60($out)
|
||
|
|
lea ($out,%rax),$out # size optimization
|
||
|
|
|
||
|
|
vpxord 0x00($inp),$xa2,$xa2
|
||
|
|
vpxor 0x20($inp),$xb2,$xb2
|
||
|
|
vpxor 0x40($inp),$xc2,$xc2
|
||
|
|
vpxor 0x60($inp),$xd2,$xd2
|
||
|
|
lea ($inp,%rax),$inp # size optimization
|
||
|
|
vmovdqu32 $xa2,0x00($out)
|
||
|
|
vmovdqu $xb2,0x20($out)
|
||
|
|
vmovdqu $xc2,0x40($out)
|
||
|
|
vmovdqu $xd2,0x60($out)
|
||
|
|
lea ($out,%rax),$out # size optimization
|
||
|
|
|
||
|
|
vpxor 0x00($inp),$xa3,$xa3
|
||
|
|
vpxor 0x20($inp),$xb3,$xb3
|
||
|
|
vpxor 0x40($inp),$xc3,$xc3
|
||
|
|
vpxor 0x60($inp),$xd3,$xd3
|
||
|
|
lea ($inp,%rax),$inp # size optimization
|
||
|
|
vmovdqu $xa3,0x00($out)
|
||
|
|
vmovdqu $xb3,0x20($out)
|
||
|
|
vmovdqu $xc3,0x40($out)
|
||
|
|
vmovdqu $xd3,0x60($out)
|
||
|
|
lea ($out,%rax),$out # size optimization
|
||
|
|
|
||
|
|
vpbroadcastd 0(%r10),%ymm0 # reload key
|
||
|
|
vpbroadcastd 4(%r10),%ymm1
|
||
|
|
|
||
|
|
sub \$64*8,$len
|
||
|
|
jnz .Loop_outer8xvl
|
||
|
|
|
||
|
|
jmp .Ldone8xvl
|
||
|
|
|
||
|
|
.align 32
|
||
|
|
.Ltail8xvl:
|
||
|
|
vmovdqa64 $xa0,%ymm8 # size optimization
|
||
|
|
___
|
||
|
|
$xa0 = "%ymm8";
|
||
|
|
$code.=<<___;
|
||
|
|
xor %r10,%r10
|
||
|
|
sub $inp,$out
|
||
|
|
cmp \$64*1,$len
|
||
|
|
jb .Less_than_64_8xvl
|
||
|
|
vpxor 0x00($inp),$xa0,$xa0 # xor with input
|
||
|
|
vpxor 0x20($inp),$xb0,$xb0
|
||
|
|
vmovdqu $xa0,0x00($out,$inp)
|
||
|
|
vmovdqu $xb0,0x20($out,$inp)
|
||
|
|
je .Ldone8xvl
|
||
|
|
vmovdqa $xc0,$xa0
|
||
|
|
vmovdqa $xd0,$xb0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*2,$len
|
||
|
|
jb .Less_than_64_8xvl
|
||
|
|
vpxor 0x00($inp),$xc0,$xc0
|
||
|
|
vpxor 0x20($inp),$xd0,$xd0
|
||
|
|
vmovdqu $xc0,0x00($out,$inp)
|
||
|
|
vmovdqu $xd0,0x20($out,$inp)
|
||
|
|
je .Ldone8xvl
|
||
|
|
vmovdqa $xa1,$xa0
|
||
|
|
vmovdqa $xb1,$xb0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*3,$len
|
||
|
|
jb .Less_than_64_8xvl
|
||
|
|
vpxor 0x00($inp),$xa1,$xa1
|
||
|
|
vpxor 0x20($inp),$xb1,$xb1
|
||
|
|
vmovdqu $xa1,0x00($out,$inp)
|
||
|
|
vmovdqu $xb1,0x20($out,$inp)
|
||
|
|
je .Ldone8xvl
|
||
|
|
vmovdqa $xc1,$xa0
|
||
|
|
vmovdqa $xd1,$xb0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*4,$len
|
||
|
|
jb .Less_than_64_8xvl
|
||
|
|
vpxor 0x00($inp),$xc1,$xc1
|
||
|
|
vpxor 0x20($inp),$xd1,$xd1
|
||
|
|
vmovdqu $xc1,0x00($out,$inp)
|
||
|
|
vmovdqu $xd1,0x20($out,$inp)
|
||
|
|
je .Ldone8xvl
|
||
|
|
vmovdqa32 $xa2,$xa0
|
||
|
|
vmovdqa $xb2,$xb0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*5,$len
|
||
|
|
jb .Less_than_64_8xvl
|
||
|
|
vpxord 0x00($inp),$xa2,$xa2
|
||
|
|
vpxor 0x20($inp),$xb2,$xb2
|
||
|
|
vmovdqu32 $xa2,0x00($out,$inp)
|
||
|
|
vmovdqu $xb2,0x20($out,$inp)
|
||
|
|
je .Ldone8xvl
|
||
|
|
vmovdqa $xc2,$xa0
|
||
|
|
vmovdqa $xd2,$xb0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*6,$len
|
||
|
|
jb .Less_than_64_8xvl
|
||
|
|
vpxor 0x00($inp),$xc2,$xc2
|
||
|
|
vpxor 0x20($inp),$xd2,$xd2
|
||
|
|
vmovdqu $xc2,0x00($out,$inp)
|
||
|
|
vmovdqu $xd2,0x20($out,$inp)
|
||
|
|
je .Ldone8xvl
|
||
|
|
vmovdqa $xa3,$xa0
|
||
|
|
vmovdqa $xb3,$xb0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
cmp \$64*7,$len
|
||
|
|
jb .Less_than_64_8xvl
|
||
|
|
vpxor 0x00($inp),$xa3,$xa3
|
||
|
|
vpxor 0x20($inp),$xb3,$xb3
|
||
|
|
vmovdqu $xa3,0x00($out,$inp)
|
||
|
|
vmovdqu $xb3,0x20($out,$inp)
|
||
|
|
je .Ldone8xvl
|
||
|
|
vmovdqa $xc3,$xa0
|
||
|
|
vmovdqa $xd3,$xb0
|
||
|
|
lea 64($inp),$inp
|
||
|
|
|
||
|
|
.Less_than_64_8xvl:
|
||
|
|
vmovdqa $xa0,0x00(%rsp)
|
||
|
|
vmovdqa $xb0,0x20(%rsp)
|
||
|
|
lea ($out,$inp),$out
|
||
|
|
and \$63,$len
|
||
|
|
|
||
|
|
.Loop_tail8xvl:
|
||
|
|
movzb ($inp,%r10),%eax
|
||
|
|
movzb (%rsp,%r10),%ecx
|
||
|
|
lea 1(%r10),%r10
|
||
|
|
xor %ecx,%eax
|
||
|
|
mov %al,-1($out,%r10)
|
||
|
|
dec $len
|
||
|
|
jnz .Loop_tail8xvl
|
||
|
|
|
||
|
|
vpxor $xa0,$xa0,$xa0
|
||
|
|
vmovdqa $xa0,0x00(%rsp)
|
||
|
|
vmovdqa $xa0,0x20(%rsp)
|
||
|
|
|
||
|
|
.Ldone8xvl:
|
||
|
|
vzeroall
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($win64);
|
||
|
|
movaps -0xa8(%r9),%xmm6
|
||
|
|
movaps -0x98(%r9),%xmm7
|
||
|
|
movaps -0x88(%r9),%xmm8
|
||
|
|
movaps -0x78(%r9),%xmm9
|
||
|
|
movaps -0x68(%r9),%xmm10
|
||
|
|
movaps -0x58(%r9),%xmm11
|
||
|
|
movaps -0x48(%r9),%xmm12
|
||
|
|
movaps -0x38(%r9),%xmm13
|
||
|
|
movaps -0x28(%r9),%xmm14
|
||
|
|
movaps -0x18(%r9),%xmm15
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
lea (%r9),%rsp
|
||
|
|
.cfi_def_cfa_register %rsp
|
||
|
|
.L8xvl_epilogue:
|
||
|
|
ret
|
||
|
|
.cfi_endproc
|
||
|
|
.size chacha20_8xvl,.-chacha20_8xvl
|
||
|
|
___
|
||
|
|
}
|
||
|
|
|
||
|
|
# 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 ssse3_handler,\@abi-omnipotent
|
||
|
|
.align 16
|
||
|
|
ssse3_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->HandlerData
|
||
|
|
|
||
|
|
mov 0(%r11),%r10d # HandlerData[0]
|
||
|
|
lea (%rsi,%r10),%r10 # prologue label
|
||
|
|
cmp %r10,%rbx # context->Rip<prologue label
|
||
|
|
jb .Lcommon_seh_tail
|
||
|
|
|
||
|
|
mov 192($context),%rax # pull context->R9
|
||
|
|
|
||
|
|
mov 4(%r11),%r10d # HandlerData[1]
|
||
|
|
lea (%rsi,%r10),%r10 # epilogue label
|
||
|
|
cmp %r10,%rbx # context->Rip>=epilogue label
|
||
|
|
jae .Lcommon_seh_tail
|
||
|
|
|
||
|
|
lea -0x28(%rax),%rsi
|
||
|
|
lea 512($context),%rdi # &context.Xmm6
|
||
|
|
mov \$4,%ecx
|
||
|
|
.long 0xa548f3fc # cld; rep movsq
|
||
|
|
|
||
|
|
.Lcommon_seh_tail:
|
||
|
|
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 ssse3_handler,.-ssse3_handler
|
||
|
|
|
||
|
|
.type full_handler,\@abi-omnipotent
|
||
|
|
.align 16
|
||
|
|
full_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->HandlerData
|
||
|
|
|
||
|
|
mov 0(%r11),%r10d # HandlerData[0]
|
||
|
|
lea (%rsi,%r10),%r10 # prologue label
|
||
|
|
cmp %r10,%rbx # context->Rip<prologue label
|
||
|
|
jb .Lcommon_seh_tail
|
||
|
|
|
||
|
|
mov 192($context),%rax # pull context->R9
|
||
|
|
|
||
|
|
mov 4(%r11),%r10d # HandlerData[1]
|
||
|
|
lea (%rsi,%r10),%r10 # epilogue label
|
||
|
|
cmp %r10,%rbx # context->Rip>=epilogue label
|
||
|
|
jae .Lcommon_seh_tail
|
||
|
|
|
||
|
|
lea -0xa8(%rax),%rsi
|
||
|
|
lea 512($context),%rdi # &context.Xmm6
|
||
|
|
mov \$20,%ecx
|
||
|
|
.long 0xa548f3fc # cld; rep movsq
|
||
|
|
|
||
|
|
jmp .Lcommon_seh_tail
|
||
|
|
.size full_handler,.-full_handler
|
||
|
|
|
||
|
|
.section .pdata
|
||
|
|
.align 4
|
||
|
|
.rva .LSEH_begin_chacha20_ssse3
|
||
|
|
.rva .LSEH_end_chacha20_ssse3
|
||
|
|
.rva .LSEH_info_chacha20_ssse3
|
||
|
|
|
||
|
|
.rva .LSEH_begin_chacha20_4x
|
||
|
|
.rva .LSEH_end_chacha20_4x
|
||
|
|
.rva .LSEH_info_chacha20_4x
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($avx && 0);
|
||
|
|
.rva .LSEH_begin_chacha20_4xop
|
||
|
|
.rva .LSEH_end_chacha20_4xop
|
||
|
|
.rva .LSEH_info_chacha20_4xop
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($avx>1);
|
||
|
|
.rva .LSEH_begin_chacha20_avx2
|
||
|
|
.rva .LSEH_end_chacha20_avx2
|
||
|
|
.rva .LSEH_info_chacha20_avx2
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($avx>2);
|
||
|
|
.rva .LSEH_begin_chacha20_avx512
|
||
|
|
.rva .LSEH_end_chacha20_avx512
|
||
|
|
.rva .LSEH_info_chacha20_avx512
|
||
|
|
|
||
|
|
.rva .LSEH_begin_chacha20_avx512vl
|
||
|
|
.rva .LSEH_end_chacha20_avx512vl
|
||
|
|
.rva .LSEH_info_chacha20_avx512vl
|
||
|
|
|
||
|
|
.rva .LSEH_begin_chacha20_16x
|
||
|
|
.rva .LSEH_end_chacha20_16x
|
||
|
|
.rva .LSEH_info_chacha20_16x
|
||
|
|
|
||
|
|
.rva .LSEH_begin_chacha20_8xvl
|
||
|
|
.rva .LSEH_end_chacha20_8xvl
|
||
|
|
.rva .LSEH_info_chacha20_8xvl
|
||
|
|
___
|
||
|
|
$code.=<<___;
|
||
|
|
.section .xdata
|
||
|
|
.align 8
|
||
|
|
.LSEH_info_chacha20_ssse3:
|
||
|
|
.byte 9,0,0,0
|
||
|
|
.rva ssse3_handler
|
||
|
|
.rva .Lssse3_body,.Lssse3_epilogue
|
||
|
|
|
||
|
|
.LSEH_info_chacha20_4x:
|
||
|
|
.byte 9,0,0,0
|
||
|
|
.rva full_handler
|
||
|
|
.rva .L4x_body,.L4x_epilogue
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($avx&&0);
|
||
|
|
.LSEH_info_chacha20_4xop:
|
||
|
|
.byte 9,0,0,0
|
||
|
|
.rva full_handler
|
||
|
|
.rva .L4xop_body,.L4xop_epilogue # HandlerData[]
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($avx>1);
|
||
|
|
.LSEH_info_chacha20_avx2:
|
||
|
|
.byte 9,0,0,0
|
||
|
|
.rva full_handler
|
||
|
|
.rva .L8x_body,.L8x_epilogue # HandlerData[]
|
||
|
|
___
|
||
|
|
$code.=<<___ if ($avx>2);
|
||
|
|
.LSEH_info_chacha20_avx512:
|
||
|
|
.byte 9,0,0,0
|
||
|
|
.rva ssse3_handler
|
||
|
|
.rva .Lavx512_body,.Lavx512_epilogue # HandlerData[]
|
||
|
|
|
||
|
|
.LSEH_info_chacha20_avx512vl:
|
||
|
|
.byte 9,0,0,0
|
||
|
|
.rva ssse3_handler
|
||
|
|
.rva .Lavx512vl_body,.Lavx512vl_epilogue # HandlerData[]
|
||
|
|
|
||
|
|
.LSEH_info_chacha20_16x:
|
||
|
|
.byte 9,0,0,0
|
||
|
|
.rva full_handler
|
||
|
|
.rva .L16x_body,.L16x_epilogue # HandlerData[]
|
||
|
|
|
||
|
|
.LSEH_info_chacha20_8xvl:
|
||
|
|
.byte 9,0,0,0
|
||
|
|
.rva full_handler
|
||
|
|
.rva .L8xvl_body,.L8xvl_epilogue # HandlerData[]
|
||
|
|
___
|
||
|
|
}
|
||
|
|
|
||
|
|
foreach (split("\n",$code)) {
|
||
|
|
s/\`([^\`]*)\`/eval $1/ge;
|
||
|
|
|
||
|
|
s/%x#%[yz]/%x/g; # "down-shift"
|
||
|
|
|
||
|
|
print $_,"\n";
|
||
|
|
}
|
||
|
|
|
||
|
|
close STDOUT;
|