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