crypto/rc4/asm/rc4-md5-x86_64.pl

   1 #!/usr/bin/env perl
   2 #
   3 # ====================================================================
   4 # Written by Andy Polyakov <appro@openssl.org> for the OpenSSL
   5 # project. The module is, however, dual licensed under OpenSSL and
   6 # CRYPTOGAMS licenses depending on where you obtain it. For further
   7 # details see http://www.openssl.org/~appro/cryptogams/.
   8 # ====================================================================
   9
  10 # June 2011
  11 #
  12 # This is RC4+MD5 "stitch" implementation. The idea, as spelled in
  13 # http://download.intel.com/design/intarch/papers/323686.pdf, is that
  14 # since both algorithms exhibit instruction-level parallelism, ILP,
  15 # below theoretical maximum, interleaving them would allow to utilize
  16 # processor resources better and achieve better performance. RC4
  17 # instruction sequence is virtually identical to rc4-x86_64.pl, which
  18 # is heavily based on submission by Maxim Perminov, Maxim Locktyukhin
  19 # and Jim Guilford of Intel. MD5 is fresh implementation aiming to
  20 # minimize register usage, which was used as "main thread" with RC4
  21 # weaved into it, one RC4 round per one MD5 round. In addition to the
  22 # stiched subroutine the script can generate standalone replacement
  23 # md5_block_asm_data_order and RC4. Below are performance numbers in
  24 # cycles per processed byte, less is better, for these the standalone
  25 # subroutines, sum of them, and stitched one:
  26 #
  27 #               RC4     MD5     RC4+MD5 stitch  gain
  28 # Opteron       6.5(*)  5.4     11.9    7.0     +70%(*)
  29 # Core2         6.5     5.8     12.3    7.7     +60%
  30 # Westmere      4.3     5.2     9.5     7.0     +36%
  31 # Sandy Bridge  4.2     5.5     9.7     6.8     +43%
  32 # Atom          9.3     6.5     15.8    11.1    +42%
  33 # VIA Nano      6.3     5.4     11.7    8.6     +37%
  34 # Ivy Bridge    4.1     5.2     9.3     6.0     +54%
  35 # Bulldozer     4.5     5.4     9.9     7.7     +29%
  36 #
  37 # (*)   rc4-x86_64.pl delivers 5.3 on Opteron, so real improvement
  38 #       is +53%...
  39
  40 my ($rc4,$md5)=(1,1);   # what to generate?
  41 my $D="#" if (!$md5);   # if set to "#", MD5 is stitched into RC4(),
  42                         # but its result is discarded. Idea here is
  43                         # to be able to use 'openssl speed rc4' for
  44                         # benchmarking the stitched subroutine...
  45
  46 my $flavour = shift;
  47 my $output  = shift;
  48 if ($flavour =~ /\./) { $output = $flavour; undef $flavour; }
  49
  50 my $win64=0; $win64=1 if ($flavour =~ /[nm]asm|mingw64/ || $output =~ /\.asm$/);
  51
  52 $0 =~ m/(.*[\/\\])[^\/\\]+$/; my $dir=$1; my $xlate;
  53 ( $xlate="${dir}x86_64-xlate.pl" and -f $xlate ) or
  54 ( $xlate="${dir}../../perlasm/x86_64-xlate.pl" and -f $xlate) or
  55 die "can't locate x86_64-xlate.pl";
  56
  57 open STDOUT,"| $^X $xlate $flavour $output";
  58
  59 my ($dat,$in0,$out,$ctx,$inp,$len, $func,$nargs);
  60
  61 if ($rc4 && !$md5) {
  62   ($dat,$len,$in0,$out) = ("%rdi","%rsi","%rdx","%rcx");
  63   $func="RC4";                          $nargs=4;
  64 } elsif ($md5 && !$rc4) {
  65   ($ctx,$inp,$len) = ("%rdi","%rsi","%rdx");
  66   $func="md5_block_asm_data_order";     $nargs=3;
  67 } else {
  68   ($dat,$in0,$out,$ctx,$inp,$len) = ("%rdi","%rsi","%rdx","%rcx","%r8","%r9");
  69   $func="rc4_md5_enc";                  $nargs=6;
  70   # void rc4_md5_enc(
  71   #             RC4_KEY *key,           #
  72   #             const void *in0,        # RC4 input
  73   #             void *out,              # RC4 output
  74   #             MD5_CTX *ctx,           #
  75   #             const void *inp,        # MD5 input
  76   #             size_t len);            # number of 64-byte blocks
  77 }
  78
  79 my @K=( 0xd76aa478,0xe8c7b756,0x242070db,0xc1bdceee,
  80         0xf57c0faf,0x4787c62a,0xa8304613,0xfd469501,
  81         0x698098d8,0x8b44f7af,0xffff5bb1,0x895cd7be,
  82         0x6b901122,0xfd987193,0xa679438e,0x49b40821,
  83
  84         0xf61e2562,0xc040b340,0x265e5a51,0xe9b6c7aa,
  85         0xd62f105d,0x02441453,0xd8a1e681,0xe7d3fbc8,
  86         0x21e1cde6,0xc33707d6,0xf4d50d87,0x455a14ed,
  87         0xa9e3e905,0xfcefa3f8,0x676f02d9,0x8d2a4c8a,
  88
  89         0xfffa3942,0x8771f681,0x6d9d6122,0xfde5380c,
  90         0xa4beea44,0x4bdecfa9,0xf6bb4b60,0xbebfbc70,
  91         0x289b7ec6,0xeaa127fa,0xd4ef3085,0x04881d05,
  92         0xd9d4d039,0xe6db99e5,0x1fa27cf8,0xc4ac5665,
  93
  94         0xf4292244,0x432aff97,0xab9423a7,0xfc93a039,
  95         0x655b59c3,0x8f0ccc92,0xffeff47d,0x85845dd1,
  96         0x6fa87e4f,0xfe2ce6e0,0xa3014314,0x4e0811a1,
  97         0xf7537e82,0xbd3af235,0x2ad7d2bb,0xeb86d391     );
  98
  99 my @V=("%r8d","%r9d","%r10d","%r11d");  # MD5 registers
 100 my $tmp="%r12d";
 101
 102 my @XX=("%rbp","%rsi");                 # RC4 registers
 103 my @TX=("%rax","%rbx");
 104 my $YY="%rcx";
 105 my $TY="%rdx";
 106
 107 my $MOD=32;                             # 16, 32 or 64
 108
 109 $code.=<<___;
 110 .text
 111 .align 16
 112
 113 .globl  $func
 114 .type   $func,\@function,$nargs
 115 $func:
 116         cmp     \$0,$len
 117         je      .Labort
 118         push    %rbx
 119         push    %rbp
 120         push    %r12
 121         push    %r13
 122         push    %r14
 123         push    %r15
 124         sub     \$40,%rsp
 125 .Lbody:
 126 ___
 127 if ($rc4) {
 128 $code.=<<___;
 129 $D#md5# mov     $ctx,%r11               # reassign arguments
 130         mov     $len,%r12
 131         mov     $in0,%r13
 132         mov     $out,%r14
 133 $D#md5# mov     $inp,%r15
 134 ___
 135     $ctx="%r11" if ($md5);              # reassign arguments
 136     $len="%r12";
 137     $in0="%r13";
 138     $out="%r14";
 139     $inp="%r15" if ($md5);
 140     $inp=$in0   if (!$md5);
 141 $code.=<<___;
 142         xor     $XX[0],$XX[0]
 143         xor     $YY,$YY
 144
 145         lea     8($dat),$dat
 146         mov     -8($dat),$XX[0]#b
 147         mov     -4($dat),$YY#b
 148
 149         inc     $XX[0]#b
 150         sub     $in0,$out
 151         movl    ($dat,$XX[0],4),$TX[0]#d
 152 ___
 153 $code.=<<___ if (!$md5);
 154         xor     $TX[1],$TX[1]
 155         test    \$-128,$len
 156         jz      .Loop1
 157         sub     $XX[0],$TX[1]
 158         and     \$`$MOD-1`,$TX[1]
 159         jz      .Loop${MOD}_is_hot
 160         sub     $TX[1],$len
 161 .Loop${MOD}_warmup:
 162         add     $TX[0]#b,$YY#b
 163         movl    ($dat,$YY,4),$TY#d
 164         movl    $TX[0]#d,($dat,$YY,4)
 165         movl    $TY#d,($dat,$XX[0],4)
 166         add     $TY#b,$TX[0]#b
 167         inc     $XX[0]#b
 168         movl    ($dat,$TX[0],4),$TY#d
 169         movl    ($dat,$XX[0],4),$TX[0]#d
 170         xorb    ($in0),$TY#b
 171         movb    $TY#b,($out,$in0)
 172         lea     1($in0),$in0
 173         dec     $TX[1]
 174         jnz     .Loop${MOD}_warmup
 175
 176         mov     $YY,$TX[1]
 177         xor     $YY,$YY
 178         mov     $TX[1]#b,$YY#b
 179
 180 .Loop${MOD}_is_hot:
 181         mov     $len,32(%rsp)           # save original $len
 182         shr     \$6,$len                # number of 64-byte blocks
 183 ___
 184   if ($D && !$md5) {                    # stitch in dummy MD5
 185     $md5=1;
 186     $ctx="%r11";
 187     $inp="%r15";
 188     $code.=<<___;
 189         mov     %rsp,$ctx
 190         mov     $in0,$inp
 191 ___
 192   }
 193 }
 194 $code.=<<___;
 195 #rc4#   add     $TX[0]#b,$YY#b
 196 #rc4#   lea     ($dat,$XX[0],4),$XX[1]
 197         shl     \$6,$len
 198         add     $inp,$len               # pointer to the end of input
 199         mov     $len,16(%rsp)
 200
 201 #md5#   mov     $ctx,24(%rsp)           # save pointer to MD5_CTX
 202 #md5#   mov     0*4($ctx),$V[0]         # load current hash value from MD5_CTX
 203 #md5#   mov     1*4($ctx),$V[1]
 204 #md5#   mov     2*4($ctx),$V[2]
 205 #md5#   mov     3*4($ctx),$V[3]
 206         jmp     .Loop
 207
 208 .align  16
 209 .Loop:
 210 #md5#   mov     $V[0],0*4(%rsp)         # put aside current hash value
 211 #md5#   mov     $V[1],1*4(%rsp)
 212 #md5#   mov     $V[2],2*4(%rsp)
 213 #md5#   mov     $V[3],$tmp              # forward reference
 214 #md5#   mov     $V[3],3*4(%rsp)
 215 ___
 216
 217 sub R0 {
 218   my ($i,$a,$b,$c,$d)=@_;
 219   my @rot0=(7,12,17,22);
 220   my $j=$i%16;
 221   my $k=$i%$MOD;
 222   my $xmm="%xmm".($j&1);
 223     $code.="    movdqu  ($in0),%xmm2\n"         if ($rc4 && $j==15);
 224     $code.="    add     \$$MOD,$XX[0]#b\n"      if ($rc4 && $j==15 && $k==$MOD-1);
 225     $code.="    pxor    $xmm,$xmm\n"            if ($rc4 && $j<=1);
 226     $code.=<<___;
 227 #rc4#   movl    ($dat,$YY,4),$TY#d
 228 #md5#   xor     $c,$tmp
 229 #rc4#   movl    $TX[0]#d,($dat,$YY,4)
 230 #md5#   and     $b,$tmp
 231 #md5#   add     4*`$j`($inp),$a
 232 #rc4#   add     $TY#b,$TX[0]#b
 233 #rc4#   movl    `4*(($k+1)%$MOD)`(`$k==$MOD-1?"$dat,$XX[0],4":"$XX[1]"`),$TX[1]#d
 234 #md5#   add     \$$K[$i],$a
 235 #md5#   xor     $d,$tmp
 236 #rc4#   movz    $TX[0]#b,$TX[0]#d
 237 #rc4#   movl    $TY#d,4*$k($XX[1])
 238 #md5#   add     $tmp,$a
 239 #rc4#   add     $TX[1]#b,$YY#b
 240 #md5#   rol     \$$rot0[$j%4],$a
 241 #md5#   mov     `$j==15?"$b":"$c"`,$tmp         # forward reference
 242 #rc4#   pinsrw  \$`($j>>1)&7`,($dat,$TX[0],4),$xmm\n
 243 #md5#   add     $b,$a
 244 ___
 245     $code.=<<___ if ($rc4 && $j==15 && $k==$MOD-1);
 246         mov     $YY,$XX[1]
 247         xor     $YY,$YY                         # keyword to partial register
 248         mov     $XX[1]#b,$YY#b
 249         lea     ($dat,$XX[0],4),$XX[1]
 250 ___
 251     $code.=<<___ if ($rc4 && $j==15);
 252         psllq   \$8,%xmm1
 253         pxor    %xmm0,%xmm2
 254         pxor    %xmm1,%xmm2
 255 ___
 256 }
 257 sub R1 {
 258   my ($i,$a,$b,$c,$d)=@_;
 259   my @rot1=(5,9,14,20);
 260   my $j=$i%16;
 261   my $k=$i%$MOD;
 262   my $xmm="%xmm".($j&1);
 263     $code.="    movdqu  16($in0),%xmm3\n"       if ($rc4 && $j==15);
 264     $code.="    add     \$$MOD,$XX[0]#b\n"      if ($rc4 && $j==15 && $k==$MOD-1);
 265     $code.="    pxor    $xmm,$xmm\n"            if ($rc4 && $j<=1);
 266     $code.=<<___;
 267 #rc4#   movl    ($dat,$YY,4),$TY#d
 268 #md5#   xor     $b,$tmp
 269 #rc4#   movl    $TX[0]#d,($dat,$YY,4)
 270 #md5#   and     $d,$tmp
 271 #md5#   add     4*`((1+5*$j)%16)`($inp),$a
 272 #rc4#   add     $TY#b,$TX[0]#b
 273 #rc4#   movl    `4*(($k+1)%$MOD)`(`$k==$MOD-1?"$dat,$XX[0],4":"$XX[1]"`),$TX[1]#d
 274 #md5#   add     \$$K[$i],$a
 275 #md5#   xor     $c,$tmp
 276 #rc4#   movz    $TX[0]#b,$TX[0]#d
 277 #rc4#   movl    $TY#d,4*$k($XX[1])
 278 #md5#   add     $tmp,$a
 279 #rc4#   add     $TX[1]#b,$YY#b
 280 #md5#   rol     \$$rot1[$j%4],$a
 281 #md5#   mov     `$j==15?"$c":"$b"`,$tmp         # forward reference
 282 #rc4#   pinsrw  \$`($j>>1)&7`,($dat,$TX[0],4),$xmm\n
 283 #md5#   add     $b,$a
 284 ___
 285     $code.=<<___ if ($rc4 && $j==15 && $k==$MOD-1);
 286         mov     $YY,$XX[1]
 287         xor     $YY,$YY                         # keyword to partial register
 288         mov     $XX[1]#b,$YY#b
 289         lea     ($dat,$XX[0],4),$XX[1]
 290 ___
 291     $code.=<<___ if ($rc4 && $j==15);
 292         psllq   \$8,%xmm1
 293         pxor    %xmm0,%xmm3
 294         pxor    %xmm1,%xmm3
 295 ___
 296 }
 297 sub R2 {
 298   my ($i,$a,$b,$c,$d)=@_;
 299   my @rot2=(4,11,16,23);
 300   my $j=$i%16;
 301   my $k=$i%$MOD;
 302   my $xmm="%xmm".($j&1);
 303     $code.="    movdqu  32($in0),%xmm4\n"       if ($rc4 && $j==15);
 304     $code.="    add     \$$MOD,$XX[0]#b\n"      if ($rc4 && $j==15 && $k==$MOD-1);
 305     $code.="    pxor    $xmm,$xmm\n"            if ($rc4 && $j<=1);
 306     $code.=<<___;
 307 #rc4#   movl    ($dat,$YY,4),$TY#d
 308 #md5#   xor     $c,$tmp
 309 #rc4#   movl    $TX[0]#d,($dat,$YY,4)
 310 #md5#   xor     $b,$tmp
 311 #md5#   add     4*`((5+3*$j)%16)`($inp),$a
 312 #rc4#   add     $TY#b,$TX[0]#b
 313 #rc4#   movl    `4*(($k+1)%$MOD)`(`$k==$MOD-1?"$dat,$XX[0],4":"$XX[1]"`),$TX[1]#d
 314 #md5#   add     \$$K[$i],$a
 315 #rc4#   movz    $TX[0]#b,$TX[0]#d
 316 #md5#   add     $tmp,$a
 317 #rc4#   movl    $TY#d,4*$k($XX[1])
 318 #rc4#   add     $TX[1]#b,$YY#b
 319 #md5#   rol     \$$rot2[$j%4],$a
 320 #md5#   mov     `$j==15?"\\\$-1":"$c"`,$tmp     # forward reference
 321 #rc4#   pinsrw  \$`($j>>1)&7`,($dat,$TX[0],4),$xmm\n
 322 #md5#   add     $b,$a
 323 ___
 324     $code.=<<___ if ($rc4 && $j==15 && $k==$MOD-1);
 325         mov     $YY,$XX[1]
 326         xor     $YY,$YY                         # keyword to partial register
 327         mov     $XX[1]#b,$YY#b
 328         lea     ($dat,$XX[0],4),$XX[1]
 329 ___
 330     $code.=<<___ if ($rc4 && $j==15);
 331         psllq   \$8,%xmm1
 332         pxor    %xmm0,%xmm4
 333         pxor    %xmm1,%xmm4
 334 ___
 335 }
 336 sub R3 {
 337   my ($i,$a,$b,$c,$d)=@_;
 338   my @rot3=(6,10,15,21);
 339   my $j=$i%16;
 340   my $k=$i%$MOD;
 341   my $xmm="%xmm".($j&1);
 342     $code.="    movdqu  48($in0),%xmm5\n"       if ($rc4 && $j==15);
 343     $code.="    add     \$$MOD,$XX[0]#b\n"      if ($rc4 && $j==15 && $k==$MOD-1);
 344     $code.="    pxor    $xmm,$xmm\n"            if ($rc4 && $j<=1);
 345     $code.=<<___;
 346 #rc4#   movl    ($dat,$YY,4),$TY#d
 347 #md5#   xor     $d,$tmp
 348 #rc4#   movl    $TX[0]#d,($dat,$YY,4)
 349 #md5#   or      $b,$tmp
 350 #md5#   add     4*`((7*$j)%16)`($inp),$a
 351 #rc4#   add     $TY#b,$TX[0]#b
 352 #rc4#   movl    `4*(($k+1)%$MOD)`(`$k==$MOD-1?"$dat,$XX[0],4":"$XX[1]"`),$TX[1]#d
 353 #md5#   add     \$$K[$i],$a
 354 #rc4#   movz    $TX[0]#b,$TX[0]#d
 355 #md5#   xor     $c,$tmp
 356 #rc4#   movl    $TY#d,4*$k($XX[1])
 357 #md5#   add     $tmp,$a
 358 #rc4#   add     $TX[1]#b,$YY#b
 359 #md5#   rol     \$$rot3[$j%4],$a
 360 #md5#   mov     \$-1,$tmp                       # forward reference
 361 #rc4#   pinsrw  \$`($j>>1)&7`,($dat,$TX[0],4),$xmm\n
 362 #md5#   add     $b,$a
 363 ___
 364     $code.=<<___ if ($rc4 && $j==15);
 365         mov     $XX[0],$XX[1]
 366         xor     $XX[0],$XX[0]                   # keyword to partial register
 367         mov     $XX[1]#b,$XX[0]#b
 368         mov     $YY,$XX[1]
 369         xor     $YY,$YY                         # keyword to partial register
 370         mov     $XX[1]#b,$YY#b
 371         lea     ($dat,$XX[0],4),$XX[1]
 372         psllq   \$8,%xmm1
 373         pxor    %xmm0,%xmm5
 374         pxor    %xmm1,%xmm5
 375 ___
 376 }
 377
 378 my $i=0;
 379 for(;$i<16;$i++) { R0($i,@V); unshift(@V,pop(@V)); push(@TX,shift(@TX)); }
 380 for(;$i<32;$i++) { R1($i,@V); unshift(@V,pop(@V)); push(@TX,shift(@TX)); }
 381 for(;$i<48;$i++) { R2($i,@V); unshift(@V,pop(@V)); push(@TX,shift(@TX)); }
 382 for(;$i<64;$i++) { R3($i,@V); unshift(@V,pop(@V)); push(@TX,shift(@TX)); }
 383
 384 $code.=<<___;
 385 #md5#   add     0*4(%rsp),$V[0]         # accumulate hash value
 386 #md5#   add     1*4(%rsp),$V[1]
 387 #md5#   add     2*4(%rsp),$V[2]
 388 #md5#   add     3*4(%rsp),$V[3]
 389
 390 #rc4#   movdqu  %xmm2,($out,$in0)       # write RC4 output
 391 #rc4#   movdqu  %xmm3,16($out,$in0)
 392 #rc4#   movdqu  %xmm4,32($out,$in0)
 393 #rc4#   movdqu  %xmm5,48($out,$in0)
 394 #md5#   lea     64($inp),$inp
 395 #rc4#   lea     64($in0),$in0
 396         cmp     16(%rsp),$inp           # are we done?
 397         jb      .Loop
 398
 399 #md5#   mov     24(%rsp),$len           # restore pointer to MD5_CTX
 400 #rc4#   sub     $TX[0]#b,$YY#b          # correct $YY
 401 #md5#   mov     $V[0],0*4($len)         # write MD5_CTX
 402 #md5#   mov     $V[1],1*4($len)
 403 #md5#   mov     $V[2],2*4($len)
 404 #md5#   mov     $V[3],3*4($len)
 405 ___
 406 $code.=<<___ if ($rc4 && (!$md5 || $D));
 407         mov     32(%rsp),$len           # restore original $len
 408         and     \$63,$len               # remaining bytes
 409         jnz     .Loop1
 410         jmp     .Ldone
 411
 412 .align  16
 413 .Loop1:
 414         add     $TX[0]#b,$YY#b
 415         movl    ($dat,$YY,4),$TY#d
 416         movl    $TX[0]#d,($dat,$YY,4)
 417         movl    $TY#d,($dat,$XX[0],4)
 418         add     $TY#b,$TX[0]#b
 419         inc     $XX[0]#b
 420         movl    ($dat,$TX[0],4),$TY#d
 421         movl    ($dat,$XX[0],4),$TX[0]#d
 422         xorb    ($in0),$TY#b
 423         movb    $TY#b,($out,$in0)
 424         lea     1($in0),$in0
 425         dec     $len
 426         jnz     .Loop1
 427
 428 .Ldone:
 429 ___
 430 $code.=<<___;
 431 #rc4#   sub     \$1,$XX[0]#b
 432 #rc4#   movl    $XX[0]#d,-8($dat)
 433 #rc4#   movl    $YY#d,-4($dat)
 434
 435         mov     40(%rsp),%r15
 436         mov     48(%rsp),%r14
 437         mov     56(%rsp),%r13
 438         mov     64(%rsp),%r12
 439         mov     72(%rsp),%rbp
 440         mov     80(%rsp),%rbx
 441         lea     88(%rsp),%rsp
 442 .Lepilogue:
 443 .Labort:
 444         ret
 445 .size $func,.-$func
 446 ___
 447
 448 if ($rc4 && $D) {       # sole purpose of this section is to provide
 449                         # option to use the generated module as drop-in
 450                         # replacement for rc4-x86_64.pl for debugging
 451                         # and testing purposes...
 452 my ($idx,$ido)=("%r8","%r9");
 453 my ($dat,$len,$inp)=("%rdi","%rsi","%rdx");
 454
 455 $code.=<<___;
 456 .globl  RC4_set_key
 457 .type   RC4_set_key,\@function,3
 458 .align  16
 459 RC4_set_key:
 460         lea     8($dat),$dat
 461         lea     ($inp,$len),$inp
 462         neg     $len
 463         mov     $len,%rcx
 464         xor     %eax,%eax
 465         xor     $ido,$ido
 466         xor     %r10,%r10
 467         xor     %r11,%r11
 468         jmp     .Lw1stloop
 469
 470 .align  16
 471 .Lw1stloop:
 472         mov     %eax,($dat,%rax,4)
 473         add     \$1,%al
 474         jnc     .Lw1stloop
 475
 476         xor     $ido,$ido
 477         xor     $idx,$idx
 478 .align  16
 479 .Lw2ndloop:
 480         mov     ($dat,$ido,4),%r10d
 481         add     ($inp,$len,1),$idx#b
 482         add     %r10b,$idx#b
 483         add     \$1,$len
 484         mov     ($dat,$idx,4),%r11d
 485         cmovz   %rcx,$len
 486         mov     %r10d,($dat,$idx,4)
 487         mov     %r11d,($dat,$ido,4)
 488         add     \$1,$ido#b
 489         jnc     .Lw2ndloop
 490
 491         xor     %eax,%eax
 492         mov     %eax,-8($dat)
 493         mov     %eax,-4($dat)
 494         ret
 495 .size   RC4_set_key,.-RC4_set_key
 496
 497 .globl  RC4_options
 498 .type   RC4_options,\@abi-omnipotent
 499 .align  16
 500 RC4_options:
 501         lea     .Lopts(%rip),%rax
 502         ret
 503 .align  64
 504 .Lopts:
 505 .asciz  "rc4(64x,int)"
 506 .align  64
 507 .size   RC4_options,.-RC4_options
 508 ___
 509 }
 510 # EXCEPTION_DISPOSITION handler (EXCEPTION_RECORD *rec,ULONG64 frame,
 511 #               CONTEXT *context,DISPATCHER_CONTEXT *disp)
 512 if ($win64) {
 513 my $rec="%rcx";
 514 my $frame="%rdx";
 515 my $context="%r8";
 516 my $disp="%r9";
 517
 518 $code.=<<___;
 519 .extern __imp_RtlVirtualUnwind
 520 .type   se_handler,\@abi-omnipotent
 521 .align  16
 522 se_handler:
 523         push    %rsi
 524         push    %rdi
 525         push    %rbx
 526         push    %rbp
 527         push    %r12
 528         push    %r13
 529         push    %r14
 530         push    %r15
 531         pushfq
 532         sub     \$64,%rsp
 533
 534         mov     120($context),%rax      # pull context->Rax
 535         mov     248($context),%rbx      # pull context->Rip
 536
 537         lea     .Lbody(%rip),%r10
 538         cmp     %r10,%rbx               # context->Rip<.Lbody
 539         jb      .Lin_prologue
 540
 541         mov     152($context),%rax      # pull context->Rsp
 542
 543         lea     .Lepilogue(%rip),%r10
 544         cmp     %r10,%rbx               # context->Rip>=.Lepilogue
 545         jae     .Lin_prologue
 546
 547         mov     40(%rax),%r15
 548         mov     48(%rax),%r14
 549         mov     56(%rax),%r13
 550         mov     64(%rax),%r12
 551         mov     72(%rax),%rbp
 552         mov     80(%rax),%rbx
 553         lea     88(%rax),%rax
 554
 555         mov     %rbx,144($context)      # restore context->Rbx
 556         mov     %rbp,160($context)      # restore context->Rbp
 557         mov     %r12,216($context)      # restore context->R12
 558         mov     %r13,224($context)      # restore context->R12
 559         mov     %r14,232($context)      # restore context->R14
 560         mov     %r15,240($context)      # restore context->R15
 561
 562 .Lin_prologue:
 563         mov     8(%rax),%rdi
 564         mov     16(%rax),%rsi
 565         mov     %rax,152($context)      # restore context->Rsp
 566         mov     %rsi,168($context)      # restore context->Rsi
 567         mov     %rdi,176($context)      # restore context->Rdi
 568
 569         mov     40($disp),%rdi          # disp->ContextRecord
 570         mov     $context,%rsi           # context
 571         mov     \$154,%ecx              # sizeof(CONTEXT)
 572         .long   0xa548f3fc              # cld; rep movsq
 573
 574         mov     $disp,%rsi
 575         xor     %rcx,%rcx               # arg1, UNW_FLAG_NHANDLER
 576         mov     8(%rsi),%rdx            # arg2, disp->ImageBase
 577         mov     0(%rsi),%r8             # arg3, disp->ControlPc
 578         mov     16(%rsi),%r9            # arg4, disp->FunctionEntry
 579         mov     40(%rsi),%r10           # disp->ContextRecord
 580         lea     56(%rsi),%r11           # &disp->HandlerData
 581         lea     24(%rsi),%r12           # &disp->EstablisherFrame
 582         mov     %r10,32(%rsp)           # arg5
 583         mov     %r11,40(%rsp)           # arg6
 584         mov     %r12,48(%rsp)           # arg7
 585         mov     %rcx,56(%rsp)           # arg8, (NULL)
 586         call    *__imp_RtlVirtualUnwind(%rip)
 587
 588         mov     \$1,%eax                # ExceptionContinueSearch
 589         add     \$64,%rsp
 590         popfq
 591         pop     %r15
 592         pop     %r14
 593         pop     %r13
 594         pop     %r12
 595         pop     %rbp
 596         pop     %rbx
 597         pop     %rdi
 598         pop     %rsi
 599         ret
 600 .size   se_handler,.-se_handler
 601
 602 .section        .pdata
 603 .align  4
 604         .rva    .LSEH_begin_$func
 605         .rva    .LSEH_end_$func
 606         .rva    .LSEH_info_$func
 607
 608 .section        .xdata
 609 .align  8
 610 .LSEH_info_$func:
 611         .byte   9,0,0,0
 612         .rva    se_handler
 613 ___
 614 }
 615
 616 sub reg_part {
 617 my ($reg,$conv)=@_;
 618     if ($reg =~ /%r[0-9]+/)     { $reg .= $conv; }
 619     elsif ($conv eq "b")        { $reg =~ s/%[er]([^x]+)x?/%$1l/;       }
 620     elsif ($conv eq "w")        { $reg =~ s/%[er](.+)/%$1/;             }
 621     elsif ($conv eq "d")        { $reg =~ s/%[er](.+)/%e$1/;            }
 622     return $reg;
 623 }
 624
 625 $code =~ s/(%[a-z0-9]+)#([bwd])/reg_part($1,$2)/gem;
 626 $code =~ s/\`([^\`]*)\`/eval $1/gem;
 627 $code =~ s/pinsrw\s+\$0,/movd   /gm;
 628
 629 $code =~ s/#md5#//gm    if ($md5);
 630 $code =~ s/#rc4#//gm    if ($rc4);
 631
 632 print $code;
 633
 634 close STDOUT;