crypto/chacha/asm/chacha-x86.pl

   1 #! /usr/bin/env perl
   2 # Copyright 2016-2018 The OpenSSL Project Authors. All Rights Reserved.
   3 #
   4 # Licensed under the OpenSSL license (the "License").  You may not use
   5 # this file except in compliance with the License.  You can obtain a copy
   6 # in the file LICENSE in the source distribution or at
   7 # https://www.openssl.org/source/license.html
   8
   9 #
  10 # ====================================================================
  11 # Written by Andy Polyakov <appro@openssl.org> for the OpenSSL
  12 # project. The module is, however, dual licensed under OpenSSL and
  13 # CRYPTOGAMS licenses depending on where you obtain it. For further
  14 # details see http://www.openssl.org/~appro/cryptogams/.
  15 # ====================================================================
  16 #
  17 # January 2015
  18 #
  19 # ChaCha20 for x86.
  20 #
  21 # Performance in cycles per byte out of large buffer.
  22 #
  23 #               1xIALU/gcc      4xSSSE3
  24 # Pentium       17.5/+80%
  25 # PIII          14.2/+60%
  26 # P4            18.6/+84%
  27 # Core2         9.56/+89%       4.83
  28 # Westmere      9.50/+45%       3.35
  29 # Sandy Bridge  10.5/+47%       3.20
  30 # Haswell       8.15/+50%       2.83
  31 # Skylake       7.53/+22%       2.75
  32 # Silvermont    17.4/+36%       8.35
  33 # Goldmont      13.4/+40%       4.36
  34 # Sledgehammer  10.2/+54%
  35 # Bulldozer     13.4/+50%       4.38(*)
  36 #
  37 # (*)   Bulldozer actually executes 4xXOP code path that delivers 3.55;
  38
  39 $0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1;
  40 push(@INC,"${dir}","${dir}../../perlasm");
  41 require "x86asm.pl";
  42
  43 $output=pop;
  44 open STDOUT,">$output";
  45
  46 &asm_init($ARGV[0],$ARGV[$#ARGV] eq "386");
  47
  48 $xmm=$ymm=0;
  49 for (@ARGV) { $xmm=1 if (/-DOPENSSL_IA32_SSE2/); }
  50
  51 $ymm=1 if ($xmm &&
  52                 `$ENV{CC} -Wa,-v -c -o /dev/null -x assembler /dev/null 2>&1`
  53                         =~ /GNU assembler version ([2-9]\.[0-9]+)/ &&
  54                 ($gasver=$1)>=2.19);    # first version supporting AVX
  55
  56 $ymm=1 if ($xmm && !$ymm && $ARGV[0] eq "win32n" &&
  57                 `nasm -v 2>&1` =~ /NASM version ([2-9]\.[0-9]+)/ &&
  58                 $1>=2.03);      # first version supporting AVX
  59
  60 $ymm=1 if ($xmm && !$ymm && $ARGV[0] eq "win32" &&
  61                 `ml 2>&1` =~ /Version ([0-9]+)\./ &&
  62                 $1>=10);        # first version supporting AVX
  63
  64 $ymm=1 if ($xmm && !$ymm &&
  65                 `$ENV{CC} -v 2>&1` =~ /((?:^clang|LLVM) version|based on LLVM) ([3-9]\.[0-9]+)/ &&
  66                 $2>=3.0);       # first version supporting AVX
  67
  68 $a="eax";
  69 ($b,$b_)=("ebx","ebp");
  70 ($c,$c_)=("ecx","esi");
  71 ($d,$d_)=("edx","edi");
  72
  73 sub QUARTERROUND {
  74 my ($ai,$bi,$ci,$di,$i)=@_;
  75 my ($an,$bn,$cn,$dn)=map(($_&~3)+(($_+1)&3),($ai,$bi,$ci,$di)); # next
  76 my ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_-1)&3),($ai,$bi,$ci,$di)); # previous
  77
  78         #       a   b   c   d
  79         #
  80         #       0   4   8  12 < even round
  81         #       1   5   9  13
  82         #       2   6  10  14
  83         #       3   7  11  15
  84         #       0   5  10  15 < odd round
  85         #       1   6  11  12
  86         #       2   7   8  13
  87         #       3   4   9  14
  88
  89         if ($i==0) {
  90             my $j=4;
  91             ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_-$j--)&3),($ap,$bp,$cp,$dp));
  92         } elsif ($i==3) {
  93             my $j=0;
  94             ($an,$bn,$cn,$dn)=map(($_&~3)+(($_+$j++)&3),($an,$bn,$cn,$dn));
  95         } elsif ($i==4) {
  96             my $j=4;
  97             ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_+$j--)&3),($ap,$bp,$cp,$dp));
  98         } elsif ($i==7) {
  99             my $j=0;
 100             ($an,$bn,$cn,$dn)=map(($_&~3)+(($_-$j++)&3),($an,$bn,$cn,$dn));
 101         }
 102
 103         #&add   ($a,$b);                        # see elsewhere
 104         &xor    ($d,$a);
 105          &mov   (&DWP(4*$cp,"esp"),$c_)         if ($ai>0 && $ai<3);
 106         &rol    ($d,16);
 107          &mov   (&DWP(4*$bp,"esp"),$b_)         if ($i!=0);
 108         &add    ($c,$d);
 109          &mov   ($c_,&DWP(4*$cn,"esp"))         if ($ai>0 && $ai<3);
 110         &xor    ($b,$c);
 111          &mov   ($d_,&DWP(4*$dn,"esp"))         if ($di!=$dn);
 112         &rol    ($b,12);
 113          &mov   ($b_,&DWP(4*$bn,"esp"))         if ($i<7);
 114          &mov   ($b_,&DWP(128,"esp"))           if ($i==7);     # loop counter
 115         &add    ($a,$b);
 116         &xor    ($d,$a);
 117         &mov    (&DWP(4*$ai,"esp"),$a);
 118         &rol    ($d,8);
 119         &mov    ($a,&DWP(4*$an,"esp"));
 120         &add    ($c,$d);
 121         &mov    (&DWP(4*$di,"esp"),$d)          if ($di!=$dn);
 122         &mov    ($d_,$d)                        if ($di==$dn);
 123         &xor    ($b,$c);
 124          &add   ($a,$b_)                        if ($i<7);      # elsewhere
 125         &rol    ($b,7);
 126
 127         ($b,$b_)=($b_,$b);
 128         ($c,$c_)=($c_,$c);
 129         ($d,$d_)=($d_,$d);
 130 }
 131
 132 &static_label("ssse3_shortcut");
 133 &static_label("xop_shortcut");
 134 &static_label("ssse3_data");
 135 &static_label("pic_point");
 136
 137 &function_begin("ChaCha20_ctr32");
 138         &xor    ("eax","eax");
 139         &cmp    ("eax",&wparam(2));             # len==0?
 140         &je     (&label("no_data"));
 141 if ($xmm) {
 142         &call   (&label("pic_point"));
 143 &set_label("pic_point");
 144         &blindpop("eax");
 145         &picmeup("ebp","OPENSSL_ia32cap_P","eax",&label("pic_point"));
 146         &test   (&DWP(0,"ebp"),1<<24);          # test FXSR bit
 147         &jz     (&label("x86"));
 148         &test   (&DWP(4,"ebp"),1<<9);           # test SSSE3 bit
 149         &jz     (&label("x86"));
 150         &jmp    (&label("ssse3_shortcut"));
 151 &set_label("x86");
 152 }
 153         &mov    ("esi",&wparam(3));             # key
 154         &mov    ("edi",&wparam(4));             # counter and nonce
 155
 156         &stack_push(33);
 157
 158         &mov    ("eax",&DWP(4*0,"esi"));        # copy key
 159         &mov    ("ebx",&DWP(4*1,"esi"));
 160         &mov    ("ecx",&DWP(4*2,"esi"));
 161         &mov    ("edx",&DWP(4*3,"esi"));
 162         &mov    (&DWP(64+4*4,"esp"),"eax");
 163         &mov    (&DWP(64+4*5,"esp"),"ebx");
 164         &mov    (&DWP(64+4*6,"esp"),"ecx");
 165         &mov    (&DWP(64+4*7,"esp"),"edx");
 166         &mov    ("eax",&DWP(4*4,"esi"));
 167         &mov    ("ebx",&DWP(4*5,"esi"));
 168         &mov    ("ecx",&DWP(4*6,"esi"));
 169         &mov    ("edx",&DWP(4*7,"esi"));
 170         &mov    (&DWP(64+4*8,"esp"),"eax");
 171         &mov    (&DWP(64+4*9,"esp"),"ebx");
 172         &mov    (&DWP(64+4*10,"esp"),"ecx");
 173         &mov    (&DWP(64+4*11,"esp"),"edx");
 174         &mov    ("eax",&DWP(4*0,"edi"));        # copy counter and nonce
 175         &mov    ("ebx",&DWP(4*1,"edi"));
 176         &mov    ("ecx",&DWP(4*2,"edi"));
 177         &mov    ("edx",&DWP(4*3,"edi"));
 178         &sub    ("eax",1);
 179         &mov    (&DWP(64+4*12,"esp"),"eax");
 180         &mov    (&DWP(64+4*13,"esp"),"ebx");
 181         &mov    (&DWP(64+4*14,"esp"),"ecx");
 182         &mov    (&DWP(64+4*15,"esp"),"edx");
 183         &jmp    (&label("entry"));
 184
 185 &set_label("outer_loop",16);
 186         &mov    (&wparam(1),$b);                # save input
 187         &mov    (&wparam(0),$a);                # save output
 188         &mov    (&wparam(2),$c);                # save len
 189 &set_label("entry");
 190         &mov    ($a,0x61707865);
 191         &mov    (&DWP(4*1,"esp"),0x3320646e);
 192         &mov    (&DWP(4*2,"esp"),0x79622d32);
 193         &mov    (&DWP(4*3,"esp"),0x6b206574);
 194
 195         &mov    ($b, &DWP(64+4*5,"esp"));       # copy key material
 196         &mov    ($b_,&DWP(64+4*6,"esp"));
 197         &mov    ($c, &DWP(64+4*10,"esp"));
 198         &mov    ($c_,&DWP(64+4*11,"esp"));
 199         &mov    ($d, &DWP(64+4*13,"esp"));
 200         &mov    ($d_,&DWP(64+4*14,"esp"));
 201         &mov    (&DWP(4*5,"esp"),$b);
 202         &mov    (&DWP(4*6,"esp"),$b_);
 203         &mov    (&DWP(4*10,"esp"),$c);
 204         &mov    (&DWP(4*11,"esp"),$c_);
 205         &mov    (&DWP(4*13,"esp"),$d);
 206         &mov    (&DWP(4*14,"esp"),$d_);
 207
 208         &mov    ($b, &DWP(64+4*7,"esp"));
 209         &mov    ($d_,&DWP(64+4*15,"esp"));
 210         &mov    ($d, &DWP(64+4*12,"esp"));
 211         &mov    ($b_,&DWP(64+4*4,"esp"));
 212         &mov    ($c, &DWP(64+4*8,"esp"));
 213         &mov    ($c_,&DWP(64+4*9,"esp"));
 214         &add    ($d,1);                         # counter value
 215         &mov    (&DWP(4*7,"esp"),$b);
 216         &mov    (&DWP(4*15,"esp"),$d_);
 217         &mov    (&DWP(64+4*12,"esp"),$d);       # save counter value
 218
 219         &mov    ($b,10);                        # loop counter
 220         &jmp    (&label("loop"));
 221
 222 &set_label("loop",16);
 223         &add    ($a,$b_);                       # elsewhere
 224         &mov    (&DWP(128,"esp"),$b);           # save loop counter
 225         &mov    ($b,$b_);
 226         &QUARTERROUND(0, 4, 8, 12, 0);
 227         &QUARTERROUND(1, 5, 9, 13, 1);
 228         &QUARTERROUND(2, 6,10, 14, 2);
 229         &QUARTERROUND(3, 7,11, 15, 3);
 230         &QUARTERROUND(0, 5,10, 15, 4);
 231         &QUARTERROUND(1, 6,11, 12, 5);
 232         &QUARTERROUND(2, 7, 8, 13, 6);
 233         &QUARTERROUND(3, 4, 9, 14, 7);
 234         &dec    ($b);
 235         &jnz    (&label("loop"));
 236
 237         &mov    ($b,&wparam(2));                # load len
 238
 239         &add    ($a,0x61707865);                # accumulate key material
 240         &add    ($b_,&DWP(64+4*4,"esp"));
 241         &add    ($c, &DWP(64+4*8,"esp"));
 242         &add    ($c_,&DWP(64+4*9,"esp"));
 243
 244         &cmp    ($b,64);
 245         &jb     (&label("tail"));
 246
 247         &mov    ($b,&wparam(1));                # load input pointer
 248         &add    ($d, &DWP(64+4*12,"esp"));
 249         &add    ($d_,&DWP(64+4*14,"esp"));
 250
 251         &xor    ($a, &DWP(4*0,$b));             # xor with input
 252         &xor    ($b_,&DWP(4*4,$b));
 253         &mov    (&DWP(4*0,"esp"),$a);
 254         &mov    ($a,&wparam(0));                # load output pointer
 255         &xor    ($c, &DWP(4*8,$b));
 256         &xor    ($c_,&DWP(4*9,$b));
 257         &xor    ($d, &DWP(4*12,$b));
 258         &xor    ($d_,&DWP(4*14,$b));
 259         &mov    (&DWP(4*4,$a),$b_);             # write output
 260         &mov    (&DWP(4*8,$a),$c);
 261         &mov    (&DWP(4*9,$a),$c_);
 262         &mov    (&DWP(4*12,$a),$d);
 263         &mov    (&DWP(4*14,$a),$d_);
 264
 265         &mov    ($b_,&DWP(4*1,"esp"));
 266         &mov    ($c, &DWP(4*2,"esp"));
 267         &mov    ($c_,&DWP(4*3,"esp"));
 268         &mov    ($d, &DWP(4*5,"esp"));
 269         &mov    ($d_,&DWP(4*6,"esp"));
 270         &add    ($b_,0x3320646e);               # accumulate key material
 271         &add    ($c, 0x79622d32);
 272         &add    ($c_,0x6b206574);
 273         &add    ($d, &DWP(64+4*5,"esp"));
 274         &add    ($d_,&DWP(64+4*6,"esp"));
 275         &xor    ($b_,&DWP(4*1,$b));
 276         &xor    ($c, &DWP(4*2,$b));
 277         &xor    ($c_,&DWP(4*3,$b));
 278         &xor    ($d, &DWP(4*5,$b));
 279         &xor    ($d_,&DWP(4*6,$b));
 280         &mov    (&DWP(4*1,$a),$b_);
 281         &mov    (&DWP(4*2,$a),$c);
 282         &mov    (&DWP(4*3,$a),$c_);
 283         &mov    (&DWP(4*5,$a),$d);
 284         &mov    (&DWP(4*6,$a),$d_);
 285
 286         &mov    ($b_,&DWP(4*7,"esp"));
 287         &mov    ($c, &DWP(4*10,"esp"));
 288         &mov    ($c_,&DWP(4*11,"esp"));
 289         &mov    ($d, &DWP(4*13,"esp"));
 290         &mov    ($d_,&DWP(4*15,"esp"));
 291         &add    ($b_,&DWP(64+4*7,"esp"));
 292         &add    ($c, &DWP(64+4*10,"esp"));
 293         &add    ($c_,&DWP(64+4*11,"esp"));
 294         &add    ($d, &DWP(64+4*13,"esp"));
 295         &add    ($d_,&DWP(64+4*15,"esp"));
 296         &xor    ($b_,&DWP(4*7,$b));
 297         &xor    ($c, &DWP(4*10,$b));
 298         &xor    ($c_,&DWP(4*11,$b));
 299         &xor    ($d, &DWP(4*13,$b));
 300         &xor    ($d_,&DWP(4*15,$b));
 301         &lea    ($b,&DWP(4*16,$b));
 302         &mov    (&DWP(4*7,$a),$b_);
 303         &mov    ($b_,&DWP(4*0,"esp"));
 304         &mov    (&DWP(4*10,$a),$c);
 305         &mov    ($c,&wparam(2));                # len
 306         &mov    (&DWP(4*11,$a),$c_);
 307         &mov    (&DWP(4*13,$a),$d);
 308         &mov    (&DWP(4*15,$a),$d_);
 309         &mov    (&DWP(4*0,$a),$b_);
 310         &lea    ($a,&DWP(4*16,$a));
 311         &sub    ($c,64);
 312         &jnz    (&label("outer_loop"));
 313
 314         &jmp    (&label("done"));
 315
 316 &set_label("tail");
 317         &add    ($d, &DWP(64+4*12,"esp"));
 318         &add    ($d_,&DWP(64+4*14,"esp"));
 319         &mov    (&DWP(4*0,"esp"),$a);
 320         &mov    (&DWP(4*4,"esp"),$b_);
 321         &mov    (&DWP(4*8,"esp"),$c);
 322         &mov    (&DWP(4*9,"esp"),$c_);
 323         &mov    (&DWP(4*12,"esp"),$d);
 324         &mov    (&DWP(4*14,"esp"),$d_);
 325
 326         &mov    ($b_,&DWP(4*1,"esp"));
 327         &mov    ($c, &DWP(4*2,"esp"));
 328         &mov    ($c_,&DWP(4*3,"esp"));
 329         &mov    ($d, &DWP(4*5,"esp"));
 330         &mov    ($d_,&DWP(4*6,"esp"));
 331         &add    ($b_,0x3320646e);               # accumulate key material
 332         &add    ($c, 0x79622d32);
 333         &add    ($c_,0x6b206574);
 334         &add    ($d, &DWP(64+4*5,"esp"));
 335         &add    ($d_,&DWP(64+4*6,"esp"));
 336         &mov    (&DWP(4*1,"esp"),$b_);
 337         &mov    (&DWP(4*2,"esp"),$c);
 338         &mov    (&DWP(4*3,"esp"),$c_);
 339         &mov    (&DWP(4*5,"esp"),$d);
 340         &mov    (&DWP(4*6,"esp"),$d_);
 341
 342         &mov    ($b_,&DWP(4*7,"esp"));
 343         &mov    ($c, &DWP(4*10,"esp"));
 344         &mov    ($c_,&DWP(4*11,"esp"));
 345         &mov    ($d, &DWP(4*13,"esp"));
 346         &mov    ($d_,&DWP(4*15,"esp"));
 347         &add    ($b_,&DWP(64+4*7,"esp"));
 348         &add    ($c, &DWP(64+4*10,"esp"));
 349         &add    ($c_,&DWP(64+4*11,"esp"));
 350         &add    ($d, &DWP(64+4*13,"esp"));
 351         &add    ($d_,&DWP(64+4*15,"esp"));
 352         &mov    (&DWP(4*7,"esp"),$b_);
 353         &mov    ($b_,&wparam(1));               # load input
 354         &mov    (&DWP(4*10,"esp"),$c);
 355         &mov    ($c,&wparam(0));                # load output
 356         &mov    (&DWP(4*11,"esp"),$c_);
 357         &xor    ($c_,$c_);
 358         &mov    (&DWP(4*13,"esp"),$d);
 359         &mov    (&DWP(4*15,"esp"),$d_);
 360
 361         &xor    ("eax","eax");
 362         &xor    ("edx","edx");
 363 &set_label("tail_loop");
 364         &movb   ("al",&BP(0,$c_,$b_));
 365         &movb   ("dl",&BP(0,"esp",$c_));
 366         &lea    ($c_,&DWP(1,$c_));
 367         &xor    ("al","dl");
 368         &mov    (&BP(-1,$c,$c_),"al");
 369         &dec    ($b);
 370         &jnz    (&label("tail_loop"));
 371
 372 &set_label("done");
 373         &stack_pop(33);
 374 &set_label("no_data");
 375 &function_end("ChaCha20_ctr32");
 376
 377 if ($xmm) {
 378 my ($xa,$xa_,$xb,$xb_,$xc,$xc_,$xd,$xd_)=map("xmm$_",(0..7));
 379 my ($out,$inp,$len)=("edi","esi","ecx");
 380
 381 sub QUARTERROUND_SSSE3 {
 382 my ($ai,$bi,$ci,$di,$i)=@_;
 383 my ($an,$bn,$cn,$dn)=map(($_&~3)+(($_+1)&3),($ai,$bi,$ci,$di)); # next
 384 my ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_-1)&3),($ai,$bi,$ci,$di)); # previous
 385
 386         #       a   b   c   d
 387         #
 388         #       0   4   8  12 < even round
 389         #       1   5   9  13
 390         #       2   6  10  14
 391         #       3   7  11  15
 392         #       0   5  10  15 < odd round
 393         #       1   6  11  12
 394         #       2   7   8  13
 395         #       3   4   9  14
 396
 397         if ($i==0) {
 398             my $j=4;
 399             ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_-$j--)&3),($ap,$bp,$cp,$dp));
 400         } elsif ($i==3) {
 401             my $j=0;
 402             ($an,$bn,$cn,$dn)=map(($_&~3)+(($_+$j++)&3),($an,$bn,$cn,$dn));
 403         } elsif ($i==4) {
 404             my $j=4;
 405             ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_+$j--)&3),($ap,$bp,$cp,$dp));
 406         } elsif ($i==7) {
 407             my $j=0;
 408             ($an,$bn,$cn,$dn)=map(($_&~3)+(($_-$j++)&3),($an,$bn,$cn,$dn));
 409         }
 410
 411         #&paddd ($xa,$xb);                      # see elsewhere
 412         #&pxor  ($xd,$xa);                      # see elsewhere
 413          &movdqa(&QWP(16*$cp-128,"ebx"),$xc_)   if ($ai>0 && $ai<3);
 414         &pshufb ($xd,&QWP(0,"eax"));            # rot16
 415          &movdqa(&QWP(16*$bp-128,"ebx"),$xb_)   if ($i!=0);
 416         &paddd  ($xc,$xd);
 417          &movdqa($xc_,&QWP(16*$cn-128,"ebx"))   if ($ai>0 && $ai<3);
 418         &pxor   ($xb,$xc);
 419          &movdqa($xb_,&QWP(16*$bn-128,"ebx"))   if ($i<7);
 420         &movdqa ($xa_,$xb);                     # borrow as temporary
 421         &pslld  ($xb,12);
 422         &psrld  ($xa_,20);
 423         &por    ($xb,$xa_);
 424          &movdqa($xa_,&QWP(16*$an-128,"ebx"));
 425         &paddd  ($xa,$xb);
 426          &movdqa($xd_,&QWP(16*$dn-128,"ebx"))   if ($di!=$dn);
 427         &pxor   ($xd,$xa);
 428         &movdqa (&QWP(16*$ai-128,"ebx"),$xa);
 429         &pshufb ($xd,&QWP(16,"eax"));           # rot8
 430         &paddd  ($xc,$xd);
 431         &movdqa (&QWP(16*$di-128,"ebx"),$xd)    if ($di!=$dn);
 432         &movdqa ($xd_,$xd)                      if ($di==$dn);
 433         &pxor   ($xb,$xc);
 434          &paddd ($xa_,$xb_)                     if ($i<7);      # elsewhere
 435         &movdqa ($xa,$xb);                      # borrow as temporary
 436         &pslld  ($xb,7);
 437         &psrld  ($xa,25);
 438          &pxor  ($xd_,$xa_)                     if ($i<7);      # elsewhere
 439         &por    ($xb,$xa);
 440
 441         ($xa,$xa_)=($xa_,$xa);
 442         ($xb,$xb_)=($xb_,$xb);
 443         ($xc,$xc_)=($xc_,$xc);
 444         ($xd,$xd_)=($xd_,$xd);
 445 }
 446
 447 &function_begin("ChaCha20_ssse3");
 448 &set_label("ssse3_shortcut");
 449 if ($ymm) {
 450         &test           (&DWP(4,"ebp"),1<<11);          # test XOP bit
 451         &jnz            (&label("xop_shortcut"));
 452 }
 453
 454         &mov            ($out,&wparam(0));
 455         &mov            ($inp,&wparam(1));
 456         &mov            ($len,&wparam(2));
 457         &mov            ("edx",&wparam(3));             # key
 458         &mov            ("ebx",&wparam(4));             # counter and nonce
 459
 460         &mov            ("ebp","esp");
 461         &stack_push     (131);
 462         &and            ("esp",-64);
 463         &mov            (&DWP(512,"esp"),"ebp");
 464
 465         &lea            ("eax",&DWP(&label("ssse3_data")."-".
 466                                     &label("pic_point"),"eax"));
 467         &movdqu         ("xmm3",&QWP(0,"ebx"));         # counter and nonce
 468
 469 if (defined($gasver) && $gasver>=2.17) {                # even though we encode
 470                                                         # pshufb manually, we
 471                                                         # handle only register
 472                                                         # operands, while this
 473                                                         # segment uses memory
 474                                                         # operand...
 475         &cmp            ($len,64*4);
 476         &jb             (&label("1x"));
 477
 478         &mov            (&DWP(512+4,"esp"),"edx");      # offload pointers
 479         &mov            (&DWP(512+8,"esp"),"ebx");
 480         &sub            ($len,64*4);                    # bias len
 481         &lea            ("ebp",&DWP(256+128,"esp"));    # size optimization
 482
 483         &movdqu         ("xmm7",&QWP(0,"edx"));         # key
 484         &pshufd         ("xmm0","xmm3",0x00);
 485         &pshufd         ("xmm1","xmm3",0x55);
 486         &pshufd         ("xmm2","xmm3",0xaa);
 487         &pshufd         ("xmm3","xmm3",0xff);
 488          &paddd         ("xmm0",&QWP(16*3,"eax"));      # fix counters
 489         &pshufd         ("xmm4","xmm7",0x00);
 490         &pshufd         ("xmm5","xmm7",0x55);
 491          &psubd         ("xmm0",&QWP(16*4,"eax"));
 492         &pshufd         ("xmm6","xmm7",0xaa);
 493         &pshufd         ("xmm7","xmm7",0xff);
 494         &movdqa         (&QWP(16*12-128,"ebp"),"xmm0");
 495         &movdqa         (&QWP(16*13-128,"ebp"),"xmm1");
 496         &movdqa         (&QWP(16*14-128,"ebp"),"xmm2");
 497         &movdqa         (&QWP(16*15-128,"ebp"),"xmm3");
 498          &movdqu        ("xmm3",&QWP(16,"edx"));        # key
 499         &movdqa         (&QWP(16*4-128,"ebp"),"xmm4");
 500         &movdqa         (&QWP(16*5-128,"ebp"),"xmm5");
 501         &movdqa         (&QWP(16*6-128,"ebp"),"xmm6");
 502         &movdqa         (&QWP(16*7-128,"ebp"),"xmm7");
 503          &movdqa        ("xmm7",&QWP(16*2,"eax"));      # sigma
 504          &lea           ("ebx",&DWP(128,"esp"));        # size optimization
 505
 506         &pshufd         ("xmm0","xmm3",0x00);
 507         &pshufd         ("xmm1","xmm3",0x55);
 508         &pshufd         ("xmm2","xmm3",0xaa);
 509         &pshufd         ("xmm3","xmm3",0xff);
 510         &pshufd         ("xmm4","xmm7",0x00);
 511         &pshufd         ("xmm5","xmm7",0x55);
 512         &pshufd         ("xmm6","xmm7",0xaa);
 513         &pshufd         ("xmm7","xmm7",0xff);
 514         &movdqa         (&QWP(16*8-128,"ebp"),"xmm0");
 515         &movdqa         (&QWP(16*9-128,"ebp"),"xmm1");
 516         &movdqa         (&QWP(16*10-128,"ebp"),"xmm2");
 517         &movdqa         (&QWP(16*11-128,"ebp"),"xmm3");
 518         &movdqa         (&QWP(16*0-128,"ebp"),"xmm4");
 519         &movdqa         (&QWP(16*1-128,"ebp"),"xmm5");
 520         &movdqa         (&QWP(16*2-128,"ebp"),"xmm6");
 521         &movdqa         (&QWP(16*3-128,"ebp"),"xmm7");
 522
 523         &lea            ($inp,&DWP(128,$inp));          # size optimization
 524         &lea            ($out,&DWP(128,$out));          # size optimization
 525         &jmp            (&label("outer_loop"));
 526
 527 &set_label("outer_loop",16);
 528         #&movdqa        ("xmm0",&QWP(16*0-128,"ebp"));  # copy key material
 529         &movdqa         ("xmm1",&QWP(16*1-128,"ebp"));
 530         &movdqa         ("xmm2",&QWP(16*2-128,"ebp"));
 531         &movdqa         ("xmm3",&QWP(16*3-128,"ebp"));
 532         #&movdqa        ("xmm4",&QWP(16*4-128,"ebp"));
 533         &movdqa         ("xmm5",&QWP(16*5-128,"ebp"));
 534         &movdqa         ("xmm6",&QWP(16*6-128,"ebp"));
 535         &movdqa         ("xmm7",&QWP(16*7-128,"ebp"));
 536         #&movdqa        (&QWP(16*0-128,"ebx"),"xmm0");
 537         &movdqa         (&QWP(16*1-128,"ebx"),"xmm1");
 538         &movdqa         (&QWP(16*2-128,"ebx"),"xmm2");
 539         &movdqa         (&QWP(16*3-128,"ebx"),"xmm3");
 540         #&movdqa        (&QWP(16*4-128,"ebx"),"xmm4");
 541         &movdqa         (&QWP(16*5-128,"ebx"),"xmm5");
 542         &movdqa         (&QWP(16*6-128,"ebx"),"xmm6");
 543         &movdqa         (&QWP(16*7-128,"ebx"),"xmm7");
 544         #&movdqa        ("xmm0",&QWP(16*8-128,"ebp"));
 545         #&movdqa        ("xmm1",&QWP(16*9-128,"ebp"));
 546         &movdqa         ("xmm2",&QWP(16*10-128,"ebp"));
 547         &movdqa         ("xmm3",&QWP(16*11-128,"ebp"));
 548         &movdqa         ("xmm4",&QWP(16*12-128,"ebp"));
 549         &movdqa         ("xmm5",&QWP(16*13-128,"ebp"));
 550         &movdqa         ("xmm6",&QWP(16*14-128,"ebp"));
 551         &movdqa         ("xmm7",&QWP(16*15-128,"ebp"));
 552         &paddd          ("xmm4",&QWP(16*4,"eax"));      # counter value
 553         #&movdqa        (&QWP(16*8-128,"ebx"),"xmm0");
 554         #&movdqa        (&QWP(16*9-128,"ebx"),"xmm1");
 555         &movdqa         (&QWP(16*10-128,"ebx"),"xmm2");
 556         &movdqa         (&QWP(16*11-128,"ebx"),"xmm3");
 557         &movdqa         (&QWP(16*12-128,"ebx"),"xmm4");
 558         &movdqa         (&QWP(16*13-128,"ebx"),"xmm5");
 559         &movdqa         (&QWP(16*14-128,"ebx"),"xmm6");
 560         &movdqa         (&QWP(16*15-128,"ebx"),"xmm7");
 561         &movdqa         (&QWP(16*12-128,"ebp"),"xmm4"); # save counter value
 562
 563         &movdqa         ($xa, &QWP(16*0-128,"ebp"));
 564         &movdqa         ($xd, "xmm4");
 565         &movdqa         ($xb_,&QWP(16*4-128,"ebp"));
 566         &movdqa         ($xc, &QWP(16*8-128,"ebp"));
 567         &movdqa         ($xc_,&QWP(16*9-128,"ebp"));
 568
 569         &mov            ("edx",10);                     # loop counter
 570         &nop            ();
 571
 572 &set_label("loop",16);
 573         &paddd          ($xa,$xb_);                     # elsewhere
 574         &movdqa         ($xb,$xb_);
 575         &pxor           ($xd,$xa);                      # elsewhere
 576         &QUARTERROUND_SSSE3(0, 4, 8, 12, 0);
 577         &QUARTERROUND_SSSE3(1, 5, 9, 13, 1);
 578         &QUARTERROUND_SSSE3(2, 6,10, 14, 2);
 579         &QUARTERROUND_SSSE3(3, 7,11, 15, 3);
 580         &QUARTERROUND_SSSE3(0, 5,10, 15, 4);
 581         &QUARTERROUND_SSSE3(1, 6,11, 12, 5);
 582         &QUARTERROUND_SSSE3(2, 7, 8, 13, 6);
 583         &QUARTERROUND_SSSE3(3, 4, 9, 14, 7);
 584         &dec            ("edx");
 585         &jnz            (&label("loop"));
 586
 587         &movdqa         (&QWP(16*4-128,"ebx"),$xb_);
 588         &movdqa         (&QWP(16*8-128,"ebx"),$xc);
 589         &movdqa         (&QWP(16*9-128,"ebx"),$xc_);
 590         &movdqa         (&QWP(16*12-128,"ebx"),$xd);
 591         &movdqa         (&QWP(16*14-128,"ebx"),$xd_);
 592
 593     my ($xa0,$xa1,$xa2,$xa3,$xt0,$xt1,$xt2,$xt3)=map("xmm$_",(0..7));
 594
 595         #&movdqa        ($xa0,&QWP(16*0-128,"ebx"));    # it's there
 596         &movdqa         ($xa1,&QWP(16*1-128,"ebx"));
 597         &movdqa         ($xa2,&QWP(16*2-128,"ebx"));
 598         &movdqa         ($xa3,&QWP(16*3-128,"ebx"));
 599
 600     for($i=0;$i<256;$i+=64) {
 601         &paddd          ($xa0,&QWP($i+16*0-128,"ebp")); # accumulate key material
 602         &paddd          ($xa1,&QWP($i+16*1-128,"ebp"));
 603         &paddd          ($xa2,&QWP($i+16*2-128,"ebp"));
 604         &paddd          ($xa3,&QWP($i+16*3-128,"ebp"));
 605
 606         &movdqa         ($xt2,$xa0);            # "de-interlace" data
 607         &punpckldq      ($xa0,$xa1);
 608         &movdqa         ($xt3,$xa2);
 609         &punpckldq      ($xa2,$xa3);
 610         &punpckhdq      ($xt2,$xa1);
 611         &punpckhdq      ($xt3,$xa3);
 612         &movdqa         ($xa1,$xa0);
 613         &punpcklqdq     ($xa0,$xa2);            # "a0"
 614         &movdqa         ($xa3,$xt2);
 615         &punpcklqdq     ($xt2,$xt3);            # "a2"
 616         &punpckhqdq     ($xa1,$xa2);            # "a1"
 617         &punpckhqdq     ($xa3,$xt3);            # "a3"
 618
 619         #($xa2,$xt2)=($xt2,$xa2);
 620
 621         &movdqu         ($xt0,&QWP(64*0-128,$inp));     # load input
 622         &movdqu         ($xt1,&QWP(64*1-128,$inp));
 623         &movdqu         ($xa2,&QWP(64*2-128,$inp));
 624         &movdqu         ($xt3,&QWP(64*3-128,$inp));
 625         &lea            ($inp,&QWP($i<192?16:(64*4-16*3),$inp));
 626         &pxor           ($xt0,$xa0);
 627         &movdqa         ($xa0,&QWP($i+16*4-128,"ebx"))  if ($i<192);
 628         &pxor           ($xt1,$xa1);
 629         &movdqa         ($xa1,&QWP($i+16*5-128,"ebx"))  if ($i<192);
 630         &pxor           ($xt2,$xa2);
 631         &movdqa         ($xa2,&QWP($i+16*6-128,"ebx"))  if ($i<192);
 632         &pxor           ($xt3,$xa3);
 633         &movdqa         ($xa3,&QWP($i+16*7-128,"ebx"))  if ($i<192);
 634         &movdqu         (&QWP(64*0-128,$out),$xt0);     # store output
 635         &movdqu         (&QWP(64*1-128,$out),$xt1);
 636         &movdqu         (&QWP(64*2-128,$out),$xt2);
 637         &movdqu         (&QWP(64*3-128,$out),$xt3);
 638         &lea            ($out,&QWP($i<192?16:(64*4-16*3),$out));
 639     }
 640         &sub            ($len,64*4);
 641         &jnc            (&label("outer_loop"));
 642
 643         &add            ($len,64*4);
 644         &jz             (&label("done"));
 645
 646         &mov            ("ebx",&DWP(512+8,"esp"));      # restore pointers
 647         &lea            ($inp,&DWP(-128,$inp));
 648         &mov            ("edx",&DWP(512+4,"esp"));
 649         &lea            ($out,&DWP(-128,$out));
 650
 651         &movd           ("xmm2",&DWP(16*12-128,"ebp")); # counter value
 652         &movdqu         ("xmm3",&QWP(0,"ebx"));
 653         &paddd          ("xmm2",&QWP(16*6,"eax"));      # +four
 654         &pand           ("xmm3",&QWP(16*7,"eax"));
 655         &por            ("xmm3","xmm2");                # counter value
 656 }
 657 {
 658 my ($a,$b,$c,$d,$t,$t1,$rot16,$rot24)=map("xmm$_",(0..7));
 659
 660 sub SSSE3ROUND {        # critical path is 20 "SIMD ticks" per round
 661         &paddd          ($a,$b);
 662         &pxor           ($d,$a);
 663         &pshufb         ($d,$rot16);
 664
 665         &paddd          ($c,$d);
 666         &pxor           ($b,$c);
 667         &movdqa         ($t,$b);
 668         &psrld          ($b,20);
 669         &pslld          ($t,12);
 670         &por            ($b,$t);
 671
 672         &paddd          ($a,$b);
 673         &pxor           ($d,$a);
 674         &pshufb         ($d,$rot24);
 675
 676         &paddd          ($c,$d);
 677         &pxor           ($b,$c);
 678         &movdqa         ($t,$b);
 679         &psrld          ($b,25);
 680         &pslld          ($t,7);
 681         &por            ($b,$t);
 682 }
 683
 684 &set_label("1x");
 685         &movdqa         ($a,&QWP(16*2,"eax"));          # sigma
 686         &movdqu         ($b,&QWP(0,"edx"));
 687         &movdqu         ($c,&QWP(16,"edx"));
 688         #&movdqu        ($d,&QWP(0,"ebx"));             # already loaded
 689         &movdqa         ($rot16,&QWP(0,"eax"));
 690         &movdqa         ($rot24,&QWP(16,"eax"));
 691         &mov            (&DWP(16*3,"esp"),"ebp");
 692
 693         &movdqa         (&QWP(16*0,"esp"),$a);
 694         &movdqa         (&QWP(16*1,"esp"),$b);
 695         &movdqa         (&QWP(16*2,"esp"),$c);
 696         &movdqa         (&QWP(16*3,"esp"),$d);
 697         &mov            ("edx",10);
 698         &jmp            (&label("loop1x"));
 699
 700 &set_label("outer1x",16);
 701         &movdqa         ($d,&QWP(16*5,"eax"));          # one
 702         &movdqa         ($a,&QWP(16*0,"esp"));
 703         &movdqa         ($b,&QWP(16*1,"esp"));
 704         &movdqa         ($c,&QWP(16*2,"esp"));
 705         &paddd          ($d,&QWP(16*3,"esp"));
 706         &mov            ("edx",10);
 707         &movdqa         (&QWP(16*3,"esp"),$d);
 708         &jmp            (&label("loop1x"));
 709
 710 &set_label("loop1x",16);
 711         &SSSE3ROUND();
 712         &pshufd ($c,$c,0b01001110);
 713         &pshufd ($b,$b,0b00111001);
 714         &pshufd ($d,$d,0b10010011);
 715         &nop    ();
 716
 717         &SSSE3ROUND();
 718         &pshufd ($c,$c,0b01001110);
 719         &pshufd ($b,$b,0b10010011);
 720         &pshufd ($d,$d,0b00111001);
 721
 722         &dec            ("edx");
 723         &jnz            (&label("loop1x"));
 724
 725         &paddd          ($a,&QWP(16*0,"esp"));
 726         &paddd          ($b,&QWP(16*1,"esp"));
 727         &paddd          ($c,&QWP(16*2,"esp"));
 728         &paddd          ($d,&QWP(16*3,"esp"));
 729
 730         &cmp            ($len,64);
 731         &jb             (&label("tail"));
 732
 733         &movdqu         ($t,&QWP(16*0,$inp));
 734         &movdqu         ($t1,&QWP(16*1,$inp));
 735         &pxor           ($a,$t);                # xor with input
 736         &movdqu         ($t,&QWP(16*2,$inp));
 737         &pxor           ($b,$t1);
 738         &movdqu         ($t1,&QWP(16*3,$inp));
 739         &pxor           ($c,$t);
 740         &pxor           ($d,$t1);
 741         &lea            ($inp,&DWP(16*4,$inp)); # inp+=64
 742
 743         &movdqu         (&QWP(16*0,$out),$a);   # write output
 744         &movdqu         (&QWP(16*1,$out),$b);
 745         &movdqu         (&QWP(16*2,$out),$c);
 746         &movdqu         (&QWP(16*3,$out),$d);
 747         &lea            ($out,&DWP(16*4,$out)); # inp+=64
 748
 749         &sub            ($len,64);
 750         &jnz            (&label("outer1x"));
 751
 752         &jmp            (&label("done"));
 753
 754 &set_label("tail");
 755         &movdqa         (&QWP(16*0,"esp"),$a);
 756         &movdqa         (&QWP(16*1,"esp"),$b);
 757         &movdqa         (&QWP(16*2,"esp"),$c);
 758         &movdqa         (&QWP(16*3,"esp"),$d);
 759
 760         &xor            ("eax","eax");
 761         &xor            ("edx","edx");
 762         &xor            ("ebp","ebp");
 763
 764 &set_label("tail_loop");
 765         &movb           ("al",&BP(0,"esp","ebp"));
 766         &movb           ("dl",&BP(0,$inp,"ebp"));
 767         &lea            ("ebp",&DWP(1,"ebp"));
 768         &xor            ("al","dl");
 769         &movb           (&BP(-1,$out,"ebp"),"al");
 770         &dec            ($len);
 771         &jnz            (&label("tail_loop"));
 772 }
 773 &set_label("done");
 774         &mov            ("esp",&DWP(512,"esp"));
 775 &function_end("ChaCha20_ssse3");
 776
 777 &align  (64);
 778 &set_label("ssse3_data");
 779 &data_byte(0x2,0x3,0x0,0x1, 0x6,0x7,0x4,0x5, 0xa,0xb,0x8,0x9, 0xe,0xf,0xc,0xd);
 780 &data_byte(0x3,0x0,0x1,0x2, 0x7,0x4,0x5,0x6, 0xb,0x8,0x9,0xa, 0xf,0xc,0xd,0xe);
 781 &data_word(0x61707865,0x3320646e,0x79622d32,0x6b206574);
 782 &data_word(0,1,2,3);
 783 &data_word(4,4,4,4);
 784 &data_word(1,0,0,0);
 785 &data_word(4,0,0,0);
 786 &data_word(0,-1,-1,-1);
 787 &align  (64);
 788 }
 789 &asciz  ("ChaCha20 for x86, CRYPTOGAMS by <appro\@openssl.org>");
 790
 791 if ($ymm) {
 792 my ($xa,$xa_,$xb,$xb_,$xc,$xc_,$xd,$xd_)=map("xmm$_",(0..7));
 793 my ($out,$inp,$len)=("edi","esi","ecx");
 794
 795 sub QUARTERROUND_XOP {
 796 my ($ai,$bi,$ci,$di,$i)=@_;
 797 my ($an,$bn,$cn,$dn)=map(($_&~3)+(($_+1)&3),($ai,$bi,$ci,$di)); # next
 798 my ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_-1)&3),($ai,$bi,$ci,$di)); # previous
 799
 800         #       a   b   c   d
 801         #
 802         #       0   4   8  12 < even round
 803         #       1   5   9  13
 804         #       2   6  10  14
 805         #       3   7  11  15
 806         #       0   5  10  15 < odd round
 807         #       1   6  11  12
 808         #       2   7   8  13
 809         #       3   4   9  14
 810
 811         if ($i==0) {
 812             my $j=4;
 813             ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_-$j--)&3),($ap,$bp,$cp,$dp));
 814         } elsif ($i==3) {
 815             my $j=0;
 816             ($an,$bn,$cn,$dn)=map(($_&~3)+(($_+$j++)&3),($an,$bn,$cn,$dn));
 817         } elsif ($i==4) {
 818             my $j=4;
 819             ($ap,$bp,$cp,$dp)=map(($_&~3)+(($_+$j--)&3),($ap,$bp,$cp,$dp));
 820         } elsif ($i==7) {
 821             my $j=0;
 822             ($an,$bn,$cn,$dn)=map(($_&~3)+(($_-$j++)&3),($an,$bn,$cn,$dn));
 823         }
 824
 825         #&vpaddd        ($xa,$xa,$xb);                  # see elsewhere
 826         #&vpxor         ($xd,$xd,$xa);                  # see elsewhere
 827          &vmovdqa       (&QWP(16*$cp-128,"ebx"),$xc_)   if ($ai>0 && $ai<3);
 828         &vprotd         ($xd,$xd,16);
 829          &vmovdqa       (&QWP(16*$bp-128,"ebx"),$xb_)   if ($i!=0);
 830         &vpaddd         ($xc,$xc,$xd);
 831          &vmovdqa       ($xc_,&QWP(16*$cn-128,"ebx"))   if ($ai>0 && $ai<3);
 832         &vpxor          ($xb,$i!=0?$xb:$xb_,$xc);
 833          &vmovdqa       ($xa_,&QWP(16*$an-128,"ebx"));
 834         &vprotd         ($xb,$xb,12);
 835          &vmovdqa       ($xb_,&QWP(16*$bn-128,"ebx"))   if ($i<7);
 836         &vpaddd         ($xa,$xa,$xb);
 837          &vmovdqa       ($xd_,&QWP(16*$dn-128,"ebx"))   if ($di!=$dn);
 838         &vpxor          ($xd,$xd,$xa);
 839          &vpaddd        ($xa_,$xa_,$xb_)                if ($i<7);      # elsewhere
 840         &vprotd         ($xd,$xd,8);
 841         &vmovdqa        (&QWP(16*$ai-128,"ebx"),$xa);
 842         &vpaddd         ($xc,$xc,$xd);
 843         &vmovdqa        (&QWP(16*$di-128,"ebx"),$xd)    if ($di!=$dn);
 844         &vpxor          ($xb,$xb,$xc);
 845          &vpxor         ($xd_,$di==$dn?$xd:$xd_,$xa_)   if ($i<7);      # elsewhere
 846         &vprotd         ($xb,$xb,7);
 847
 848         ($xa,$xa_)=($xa_,$xa);
 849         ($xb,$xb_)=($xb_,$xb);
 850         ($xc,$xc_)=($xc_,$xc);
 851         ($xd,$xd_)=($xd_,$xd);
 852 }
 853
 854 &function_begin("ChaCha20_xop");
 855 &set_label("xop_shortcut");
 856         &mov            ($out,&wparam(0));
 857         &mov            ($inp,&wparam(1));
 858         &mov            ($len,&wparam(2));
 859         &mov            ("edx",&wparam(3));             # key
 860         &mov            ("ebx",&wparam(4));             # counter and nonce
 861         &vzeroupper     ();
 862
 863         &mov            ("ebp","esp");
 864         &stack_push     (131);
 865         &and            ("esp",-64);
 866         &mov            (&DWP(512,"esp"),"ebp");
 867
 868         &lea            ("eax",&DWP(&label("ssse3_data")."-".
 869                                     &label("pic_point"),"eax"));
 870         &vmovdqu        ("xmm3",&QWP(0,"ebx"));         # counter and nonce
 871
 872         &cmp            ($len,64*4);
 873         &jb             (&label("1x"));
 874
 875         &mov            (&DWP(512+4,"esp"),"edx");      # offload pointers
 876         &mov            (&DWP(512+8,"esp"),"ebx");
 877         &sub            ($len,64*4);                    # bias len
 878         &lea            ("ebp",&DWP(256+128,"esp"));    # size optimization
 879
 880         &vmovdqu        ("xmm7",&QWP(0,"edx"));         # key
 881         &vpshufd        ("xmm0","xmm3",0x00);
 882         &vpshufd        ("xmm1","xmm3",0x55);
 883         &vpshufd        ("xmm2","xmm3",0xaa);
 884         &vpshufd        ("xmm3","xmm3",0xff);
 885          &vpaddd        ("xmm0","xmm0",&QWP(16*3,"eax"));       # fix counters
 886         &vpshufd        ("xmm4","xmm7",0x00);
 887         &vpshufd        ("xmm5","xmm7",0x55);
 888          &vpsubd        ("xmm0","xmm0",&QWP(16*4,"eax"));
 889         &vpshufd        ("xmm6","xmm7",0xaa);
 890         &vpshufd        ("xmm7","xmm7",0xff);
 891         &vmovdqa        (&QWP(16*12-128,"ebp"),"xmm0");
 892         &vmovdqa        (&QWP(16*13-128,"ebp"),"xmm1");
 893         &vmovdqa        (&QWP(16*14-128,"ebp"),"xmm2");
 894         &vmovdqa        (&QWP(16*15-128,"ebp"),"xmm3");
 895          &vmovdqu       ("xmm3",&QWP(16,"edx"));        # key
 896         &vmovdqa        (&QWP(16*4-128,"ebp"),"xmm4");
 897         &vmovdqa        (&QWP(16*5-128,"ebp"),"xmm5");
 898         &vmovdqa        (&QWP(16*6-128,"ebp"),"xmm6");
 899         &vmovdqa        (&QWP(16*7-128,"ebp"),"xmm7");
 900          &vmovdqa       ("xmm7",&QWP(16*2,"eax"));      # sigma
 901          &lea           ("ebx",&DWP(128,"esp"));        # size optimization
 902
 903         &vpshufd        ("xmm0","xmm3",0x00);
 904         &vpshufd        ("xmm1","xmm3",0x55);
 905         &vpshufd        ("xmm2","xmm3",0xaa);
 906         &vpshufd        ("xmm3","xmm3",0xff);
 907         &vpshufd        ("xmm4","xmm7",0x00);
 908         &vpshufd        ("xmm5","xmm7",0x55);
 909         &vpshufd        ("xmm6","xmm7",0xaa);
 910         &vpshufd        ("xmm7","xmm7",0xff);
 911         &vmovdqa        (&QWP(16*8-128,"ebp"),"xmm0");
 912         &vmovdqa        (&QWP(16*9-128,"ebp"),"xmm1");
 913         &vmovdqa        (&QWP(16*10-128,"ebp"),"xmm2");
 914         &vmovdqa        (&QWP(16*11-128,"ebp"),"xmm3");
 915         &vmovdqa        (&QWP(16*0-128,"ebp"),"xmm4");
 916         &vmovdqa        (&QWP(16*1-128,"ebp"),"xmm5");
 917         &vmovdqa        (&QWP(16*2-128,"ebp"),"xmm6");
 918         &vmovdqa        (&QWP(16*3-128,"ebp"),"xmm7");
 919
 920         &lea            ($inp,&DWP(128,$inp));          # size optimization
 921         &lea            ($out,&DWP(128,$out));          # size optimization
 922         &jmp            (&label("outer_loop"));
 923
 924 &set_label("outer_loop",32);
 925         #&vmovdqa       ("xmm0",&QWP(16*0-128,"ebp"));  # copy key material
 926         &vmovdqa        ("xmm1",&QWP(16*1-128,"ebp"));
 927         &vmovdqa        ("xmm2",&QWP(16*2-128,"ebp"));
 928         &vmovdqa        ("xmm3",&QWP(16*3-128,"ebp"));
 929         #&vmovdqa       ("xmm4",&QWP(16*4-128,"ebp"));
 930         &vmovdqa        ("xmm5",&QWP(16*5-128,"ebp"));
 931         &vmovdqa        ("xmm6",&QWP(16*6-128,"ebp"));
 932         &vmovdqa        ("xmm7",&QWP(16*7-128,"ebp"));
 933         #&vmovdqa       (&QWP(16*0-128,"ebx"),"xmm0");
 934         &vmovdqa        (&QWP(16*1-128,"ebx"),"xmm1");
 935         &vmovdqa        (&QWP(16*2-128,"ebx"),"xmm2");
 936         &vmovdqa        (&QWP(16*3-128,"ebx"),"xmm3");
 937         #&vmovdqa       (&QWP(16*4-128,"ebx"),"xmm4");
 938         &vmovdqa        (&QWP(16*5-128,"ebx"),"xmm5");
 939         &vmovdqa        (&QWP(16*6-128,"ebx"),"xmm6");
 940         &vmovdqa        (&QWP(16*7-128,"ebx"),"xmm7");
 941         #&vmovdqa       ("xmm0",&QWP(16*8-128,"ebp"));
 942         #&vmovdqa       ("xmm1",&QWP(16*9-128,"ebp"));
 943         &vmovdqa        ("xmm2",&QWP(16*10-128,"ebp"));
 944         &vmovdqa        ("xmm3",&QWP(16*11-128,"ebp"));
 945         &vmovdqa        ("xmm4",&QWP(16*12-128,"ebp"));
 946         &vmovdqa        ("xmm5",&QWP(16*13-128,"ebp"));
 947         &vmovdqa        ("xmm6",&QWP(16*14-128,"ebp"));
 948         &vmovdqa        ("xmm7",&QWP(16*15-128,"ebp"));
 949         &vpaddd         ("xmm4","xmm4",&QWP(16*4,"eax"));       # counter value
 950         #&vmovdqa       (&QWP(16*8-128,"ebx"),"xmm0");
 951         #&vmovdqa       (&QWP(16*9-128,"ebx"),"xmm1");
 952         &vmovdqa        (&QWP(16*10-128,"ebx"),"xmm2");
 953         &vmovdqa        (&QWP(16*11-128,"ebx"),"xmm3");
 954         &vmovdqa        (&QWP(16*12-128,"ebx"),"xmm4");
 955         &vmovdqa        (&QWP(16*13-128,"ebx"),"xmm5");
 956         &vmovdqa        (&QWP(16*14-128,"ebx"),"xmm6");
 957         &vmovdqa        (&QWP(16*15-128,"ebx"),"xmm7");
 958         &vmovdqa        (&QWP(16*12-128,"ebp"),"xmm4"); # save counter value
 959
 960         &vmovdqa        ($xa, &QWP(16*0-128,"ebp"));
 961         &vmovdqa        ($xd, "xmm4");
 962         &vmovdqa        ($xb_,&QWP(16*4-128,"ebp"));
 963         &vmovdqa        ($xc, &QWP(16*8-128,"ebp"));
 964         &vmovdqa        ($xc_,&QWP(16*9-128,"ebp"));
 965
 966         &mov            ("edx",10);                     # loop counter
 967         &nop            ();
 968
 969 &set_label("loop",32);
 970         &vpaddd         ($xa,$xa,$xb_);                 # elsewhere
 971         &vpxor          ($xd,$xd,$xa);                  # elsewhere
 972         &QUARTERROUND_XOP(0, 4, 8, 12, 0);
 973         &QUARTERROUND_XOP(1, 5, 9, 13, 1);
 974         &QUARTERROUND_XOP(2, 6,10, 14, 2);
 975         &QUARTERROUND_XOP(3, 7,11, 15, 3);
 976         &QUARTERROUND_XOP(0, 5,10, 15, 4);
 977         &QUARTERROUND_XOP(1, 6,11, 12, 5);
 978         &QUARTERROUND_XOP(2, 7, 8, 13, 6);
 979         &QUARTERROUND_XOP(3, 4, 9, 14, 7);
 980         &dec            ("edx");
 981         &jnz            (&label("loop"));
 982
 983         &vmovdqa        (&QWP(16*4-128,"ebx"),$xb_);
 984         &vmovdqa        (&QWP(16*8-128,"ebx"),$xc);
 985         &vmovdqa        (&QWP(16*9-128,"ebx"),$xc_);
 986         &vmovdqa        (&QWP(16*12-128,"ebx"),$xd);
 987         &vmovdqa        (&QWP(16*14-128,"ebx"),$xd_);
 988
 989     my ($xa0,$xa1,$xa2,$xa3,$xt0,$xt1,$xt2,$xt3)=map("xmm$_",(0..7));
 990
 991         #&vmovdqa       ($xa0,&QWP(16*0-128,"ebx"));    # it's there
 992         &vmovdqa        ($xa1,&QWP(16*1-128,"ebx"));
 993         &vmovdqa        ($xa2,&QWP(16*2-128,"ebx"));
 994         &vmovdqa        ($xa3,&QWP(16*3-128,"ebx"));
 995
 996     for($i=0;$i<256;$i+=64) {
 997         &vpaddd         ($xa0,$xa0,&QWP($i+16*0-128,"ebp"));    # accumulate key material
 998         &vpaddd         ($xa1,$xa1,&QWP($i+16*1-128,"ebp"));
 999         &vpaddd         ($xa2,$xa2,&QWP($i+16*2-128,"ebp"));
1000         &vpaddd         ($xa3,$xa3,&QWP($i+16*3-128,"ebp"));
1001
1002         &vpunpckldq     ($xt2,$xa0,$xa1);       # "de-interlace" data
1003         &vpunpckldq     ($xt3,$xa2,$xa3);
1004         &vpunpckhdq     ($xa0,$xa0,$xa1);
1005         &vpunpckhdq     ($xa2,$xa2,$xa3);
1006         &vpunpcklqdq    ($xa1,$xt2,$xt3);       # "a0"
1007         &vpunpckhqdq    ($xt2,$xt2,$xt3);       # "a1"
1008         &vpunpcklqdq    ($xt3,$xa0,$xa2);       # "a2"
1009         &vpunpckhqdq    ($xa3,$xa0,$xa2);       # "a3"
1010
1011         &vpxor          ($xt0,$xa1,&QWP(64*0-128,$inp));
1012         &vpxor          ($xt1,$xt2,&QWP(64*1-128,$inp));
1013         &vpxor          ($xt2,$xt3,&QWP(64*2-128,$inp));
1014         &vpxor          ($xt3,$xa3,&QWP(64*3-128,$inp));
1015         &lea            ($inp,&QWP($i<192?16:(64*4-16*3),$inp));
1016         &vmovdqa        ($xa0,&QWP($i+16*4-128,"ebx"))  if ($i<192);
1017         &vmovdqa        ($xa1,&QWP($i+16*5-128,"ebx"))  if ($i<192);
1018         &vmovdqa        ($xa2,&QWP($i+16*6-128,"ebx"))  if ($i<192);
1019         &vmovdqa        ($xa3,&QWP($i+16*7-128,"ebx"))  if ($i<192);
1020         &vmovdqu        (&QWP(64*0-128,$out),$xt0);     # store output
1021         &vmovdqu        (&QWP(64*1-128,$out),$xt1);
1022         &vmovdqu        (&QWP(64*2-128,$out),$xt2);
1023         &vmovdqu        (&QWP(64*3-128,$out),$xt3);
1024         &lea            ($out,&QWP($i<192?16:(64*4-16*3),$out));
1025     }
1026         &sub            ($len,64*4);
1027         &jnc            (&label("outer_loop"));
1028
1029         &add            ($len,64*4);
1030         &jz             (&label("done"));
1031
1032         &mov            ("ebx",&DWP(512+8,"esp"));      # restore pointers
1033         &lea            ($inp,&DWP(-128,$inp));
1034         &mov            ("edx",&DWP(512+4,"esp"));
1035         &lea            ($out,&DWP(-128,$out));
1036
1037         &vmovd          ("xmm2",&DWP(16*12-128,"ebp")); # counter value
1038         &vmovdqu        ("xmm3",&QWP(0,"ebx"));
1039         &vpaddd         ("xmm2","xmm2",&QWP(16*6,"eax"));# +four
1040         &vpand          ("xmm3","xmm3",&QWP(16*7,"eax"));
1041         &vpor           ("xmm3","xmm3","xmm2");         # counter value
1042 {
1043 my ($a,$b,$c,$d,$t,$t1,$rot16,$rot24)=map("xmm$_",(0..7));
1044
1045 sub XOPROUND {
1046         &vpaddd         ($a,$a,$b);
1047         &vpxor          ($d,$d,$a);
1048         &vprotd         ($d,$d,16);
1049
1050         &vpaddd         ($c,$c,$d);
1051         &vpxor          ($b,$b,$c);
1052         &vprotd         ($b,$b,12);
1053
1054         &vpaddd         ($a,$a,$b);
1055         &vpxor          ($d,$d,$a);
1056         &vprotd         ($d,$d,8);
1057
1058         &vpaddd         ($c,$c,$d);
1059         &vpxor          ($b,$b,$c);
1060         &vprotd         ($b,$b,7);
1061 }
1062
1063 &set_label("1x");
1064         &vmovdqa        ($a,&QWP(16*2,"eax"));          # sigma
1065         &vmovdqu        ($b,&QWP(0,"edx"));
1066         &vmovdqu        ($c,&QWP(16,"edx"));
1067         #&vmovdqu       ($d,&QWP(0,"ebx"));             # already loaded
1068         &vmovdqa        ($rot16,&QWP(0,"eax"));
1069         &vmovdqa        ($rot24,&QWP(16,"eax"));
1070         &mov            (&DWP(16*3,"esp"),"ebp");
1071
1072         &vmovdqa        (&QWP(16*0,"esp"),$a);
1073         &vmovdqa        (&QWP(16*1,"esp"),$b);
1074         &vmovdqa        (&QWP(16*2,"esp"),$c);
1075         &vmovdqa        (&QWP(16*3,"esp"),$d);
1076         &mov            ("edx",10);
1077         &jmp            (&label("loop1x"));
1078
1079 &set_label("outer1x",16);
1080         &vmovdqa        ($d,&QWP(16*5,"eax"));          # one
1081         &vmovdqa        ($a,&QWP(16*0,"esp"));
1082         &vmovdqa        ($b,&QWP(16*1,"esp"));
1083         &vmovdqa        ($c,&QWP(16*2,"esp"));
1084         &vpaddd         ($d,$d,&QWP(16*3,"esp"));
1085         &mov            ("edx",10);
1086         &vmovdqa        (&QWP(16*3,"esp"),$d);
1087         &jmp            (&label("loop1x"));
1088
1089 &set_label("loop1x",16);
1090         &XOPROUND();
1091         &vpshufd        ($c,$c,0b01001110);
1092         &vpshufd        ($b,$b,0b00111001);
1093         &vpshufd        ($d,$d,0b10010011);
1094
1095         &XOPROUND();
1096         &vpshufd        ($c,$c,0b01001110);
1097         &vpshufd        ($b,$b,0b10010011);
1098         &vpshufd        ($d,$d,0b00111001);
1099
1100         &dec            ("edx");
1101         &jnz            (&label("loop1x"));
1102
1103         &vpaddd         ($a,$a,&QWP(16*0,"esp"));
1104         &vpaddd         ($b,$b,&QWP(16*1,"esp"));
1105         &vpaddd         ($c,$c,&QWP(16*2,"esp"));
1106         &vpaddd         ($d,$d,&QWP(16*3,"esp"));
1107
1108         &cmp            ($len,64);
1109         &jb             (&label("tail"));
1110
1111         &vpxor          ($a,$a,&QWP(16*0,$inp));        # xor with input
1112         &vpxor          ($b,$b,&QWP(16*1,$inp));
1113         &vpxor          ($c,$c,&QWP(16*2,$inp));
1114         &vpxor          ($d,$d,&QWP(16*3,$inp));
1115         &lea            ($inp,&DWP(16*4,$inp));         # inp+=64
1116
1117         &vmovdqu        (&QWP(16*0,$out),$a);           # write output
1118         &vmovdqu        (&QWP(16*1,$out),$b);
1119         &vmovdqu        (&QWP(16*2,$out),$c);
1120         &vmovdqu        (&QWP(16*3,$out),$d);
1121         &lea            ($out,&DWP(16*4,$out));         # inp+=64
1122
1123         &sub            ($len,64);
1124         &jnz            (&label("outer1x"));
1125
1126         &jmp            (&label("done"));
1127
1128 &set_label("tail");
1129         &vmovdqa        (&QWP(16*0,"esp"),$a);
1130         &vmovdqa        (&QWP(16*1,"esp"),$b);
1131         &vmovdqa        (&QWP(16*2,"esp"),$c);
1132         &vmovdqa        (&QWP(16*3,"esp"),$d);
1133
1134         &xor            ("eax","eax");
1135         &xor            ("edx","edx");
1136         &xor            ("ebp","ebp");
1137
1138 &set_label("tail_loop");
1139         &movb           ("al",&BP(0,"esp","ebp"));
1140         &movb           ("dl",&BP(0,$inp,"ebp"));
1141         &lea            ("ebp",&DWP(1,"ebp"));
1142         &xor            ("al","dl");
1143         &movb           (&BP(-1,$out,"ebp"),"al");
1144         &dec            ($len);
1145         &jnz            (&label("tail_loop"));
1146 }
1147 &set_label("done");
1148         &vzeroupper     ();
1149         &mov            ("esp",&DWP(512,"esp"));
1150 &function_end("ChaCha20_xop");
1151 }
1152
1153 &asm_finish();
1154
1155 close STDOUT;