crypto/md5/asm/md5-x86_64.pl

   1 #! /usr/bin/env perl
   2 # Author: Marc Bevand <bevand_m (at) epita.fr>
   3 # Copyright 2005-2016 The OpenSSL Project Authors. All Rights Reserved.
   4 #
   5 # Licensed under the Apache License 2.0 (the "License").  You may not use
   6 # this file except in compliance with the License.  You can obtain a copy
   7 # in the file LICENSE in the source distribution or at
   8 # https://www.openssl.org/source/license.html
   9
  10 # MD5 optimized for AMD64.
  11
  12 use strict;
  13
  14 my $code;
  15
  16 # round1_step() does:
  17 #   dst = x + ((dst + F(x,y,z) + X[k] + T_i) <<< s)
  18 #   %r10d = X[k_next]
  19 #   %r11d = z' (copy of z for the next step)
  20 # Each round1_step() takes about 5.3 clocks (9 instructions, 1.7 IPC)
  21 sub round1_step
  22 {
  23     my ($pos, $dst, $x, $y, $z, $k_next, $T_i, $s) = @_;
  24     $code .= " mov      0*4(%rsi),      %r10d           /* (NEXT STEP) X[0] */\n" if ($pos == -1);
  25     $code .= " mov      %edx,           %r11d           /* (NEXT STEP) z' = %edx */\n" if ($pos == -1);
  26     $code .= <<EOF;
  27         xor     $y,             %r11d           /* y ^ ... */
  28         lea     $T_i($dst,%r10d),$dst           /* Const + dst + ... */
  29         and     $x,             %r11d           /* x & ... */
  30         mov     $k_next*4(%rsi),%r10d           /* (NEXT STEP) X[$k_next] */
  31         xor     $z,             %r11d           /* z ^ ... */
  32         add     %r11d,          $dst            /* dst += ... */
  33         rol     \$$s,           $dst            /* dst <<< s */
  34         mov     $y,             %r11d           /* (NEXT STEP) z' = $y */
  35         add     $x,             $dst            /* dst += x */
  36 EOF
  37 }
  38
  39 # round2_step() does:
  40 #   dst = x + ((dst + G(x,y,z) + X[k] + T_i) <<< s)
  41 #   %r10d = X[k_next]
  42 #   %r11d = z' (copy of z for the next step)
  43 #   %r12d = z' (copy of z for the next step)
  44 # Each round2_step() takes about 5.4 clocks (11 instructions, 2.0 IPC)
  45 sub round2_step
  46 {
  47     my ($pos, $dst, $x, $y, $z, $k_next, $T_i, $s) = @_;
  48     $code .= " mov      %edx,           %r11d           /* (NEXT STEP) z' = %edx */\n" if ($pos == -1);
  49     $code .= " mov      %edx,           %r12d           /* (NEXT STEP) z' = %edx */\n" if ($pos == -1);
  50     $code .= <<EOF;
  51         not     %r11d                           /* not z */
  52         and     $x,             %r12d           /* x & z */
  53         lea     $T_i($dst,%r10d),$dst           /* Const + dst + ... */
  54         and     $y,             %r11d           /* y & (not z) */
  55         mov     $k_next*4(%rsi),%r10d           /* (NEXT STEP) X[$k_next] */
  56         or      %r11d,          %r12d           /* (y & (not z)) | (x & z) */
  57         mov     $y,             %r11d           /* (NEXT STEP) z' = $y */
  58         add     %r12d,          $dst            /* dst += ... */
  59         mov     $y,             %r12d           /* (NEXT STEP) z' = $y */
  60         rol     \$$s,           $dst            /* dst <<< s */
  61         add     $x,             $dst            /* dst += x */
  62 EOF
  63 }
  64
  65 # round3_step() does:
  66 #   dst = x + ((dst + H(x,y,z) + X[k] + T_i) <<< s)
  67 #   %r10d = X[k_next]
  68 #   %r11d = y' (copy of y for the next step)
  69 # Each round3_step() takes about 4.2 clocks (8 instructions, 1.9 IPC)
  70 { my $round3_alter=0;
  71 sub round3_step
  72 {
  73     my ($pos, $dst, $x, $y, $z, $k_next, $T_i, $s) = @_;
  74     $code .= " mov      %ecx,           %r11d           /* (NEXT STEP) y' = %ecx */\n" if ($pos == -1);
  75     $code .= <<EOF;
  76         lea     $T_i($dst,%r10d),$dst           /* Const + dst + ... */
  77         xor     $z,             %r11d           /* z ^ ... */
  78         mov     $k_next*4(%rsi),%r10d           /* (NEXT STEP) X[$k_next] */
  79         xor     $x,             %r11d           /* x ^ ... */
  80         add     %r11d,          $dst            /* dst += ... */
  81 EOF
  82     $code .= <<EOF if ($round3_alter);
  83         rol     \$$s,           $dst            /* dst <<< s */
  84         mov     $x,             %r11d           /* (NEXT STEP) y' = $x */
  85 EOF
  86     $code .= <<EOF if (!$round3_alter);
  87         mov     $x,             %r11d           /* (NEXT STEP) y' = $x */
  88         rol     \$$s,           $dst            /* dst <<< s */
  89 EOF
  90     $code .= <<EOF;
  91         add     $x,             $dst            /* dst += x */
  92 EOF
  93     $round3_alter^=1;
  94 }
  95 }
  96
  97 # round4_step() does:
  98 #   dst = x + ((dst + I(x,y,z) + X[k] + T_i) <<< s)
  99 #   %r10d = X[k_next]
 100 #   %r11d = not z' (copy of not z for the next step)
 101 # Each round4_step() takes about 5.2 clocks (9 instructions, 1.7 IPC)
 102 sub round4_step
 103 {
 104     my ($pos, $dst, $x, $y, $z, $k_next, $T_i, $s) = @_;
 105     $code .= " mov      \$0xffffffff,   %r11d\n" if ($pos == -1);
 106     $code .= " xor      %edx,           %r11d           /* (NEXT STEP) not z' = not %edx*/\n"
 107     if ($pos == -1);
 108     $code .= <<EOF;
 109         lea     $T_i($dst,%r10d),$dst           /* Const + dst + ... */
 110         or      $x,             %r11d           /* x | ... */
 111         mov     $k_next*4(%rsi),%r10d           /* (NEXT STEP) X[$k_next] */
 112         xor     $y,             %r11d           /* y ^ ... */
 113         add     %r11d,          $dst            /* dst += ... */
 114         mov     \$0xffffffff,   %r11d
 115         rol     \$$s,           $dst            /* dst <<< s */
 116         xor     $y,             %r11d           /* (NEXT STEP) not z' = not $y */
 117         add     $x,             $dst            /* dst += x */
 118 EOF
 119 }
 120
 121 no warnings qw(uninitialized);
 122 # $output is the last argument if it looks like a file (it has an extension)
 123 # $flavour is the first argument if it doesn't look like a file
 124 my $output = $#ARGV >= 0 && $ARGV[$#ARGV] =~ m|\.\w+$| ? pop : undef;
 125 my $flavour = $#ARGV >= 0 && $ARGV[0] !~ m|\.| ? shift : undef;
 126
 127 my $win64=0; $win64=1 if ($flavour =~ /[nm]asm|mingw64/ || $output =~ /\.asm$/);
 128
 129 $0 =~ m/(.*[\/\\])[^\/\\]+$/; my $dir=$1; my $xlate;
 130 ( $xlate="${dir}x86_64-xlate.pl" and -f $xlate ) or
 131 ( $xlate="${dir}../../perlasm/x86_64-xlate.pl" and -f $xlate) or
 132 die "can't locate x86_64-xlate.pl";
 133
 134 open OUT,"| \"$^X\" \"$xlate\" $flavour \"$output\""
 135     or die "can't call $xlate: $!";
 136 *STDOUT=*OUT;
 137
 138 $code .= <<EOF;
 139 .text
 140 .align 16
 141
 142 .globl md5_block_asm_data_order
 143 .type md5_block_asm_data_order,\@function,3
 144 md5_block_asm_data_order:
 145 .cfi_startproc
 146         push    %rbp
 147 .cfi_push       %rbp
 148         push    %rbx
 149 .cfi_push       %rbx
 150         push    %r12
 151 .cfi_push       %r12
 152         push    %r14
 153 .cfi_push       %r14
 154         push    %r15
 155 .cfi_push       %r15
 156 .Lprologue:
 157
 158         # rdi = arg #1 (ctx, MD5_CTX pointer)
 159         # rsi = arg #2 (ptr, data pointer)
 160         # rdx = arg #3 (nbr, number of 16-word blocks to process)
 161         mov     %rdi,           %rbp    # rbp = ctx
 162         shl     \$6,            %rdx    # rdx = nbr in bytes
 163         lea     (%rsi,%rdx),    %rdi    # rdi = end
 164         mov     0*4(%rbp),      %eax    # eax = ctx->A
 165         mov     1*4(%rbp),      %ebx    # ebx = ctx->B
 166         mov     2*4(%rbp),      %ecx    # ecx = ctx->C
 167         mov     3*4(%rbp),      %edx    # edx = ctx->D
 168         # end is 'rdi'
 169         # ptr is 'rsi'
 170         # A is 'eax'
 171         # B is 'ebx'
 172         # C is 'ecx'
 173         # D is 'edx'
 174
 175         cmp     %rdi,           %rsi            # cmp end with ptr
 176         je      .Lend                           # jmp if ptr == end
 177
 178         # BEGIN of loop over 16-word blocks
 179 .Lloop: # save old values of A, B, C, D
 180         mov     %eax,           %r8d
 181         mov     %ebx,           %r9d
 182         mov     %ecx,           %r14d
 183         mov     %edx,           %r15d
 184 EOF
 185 round1_step(-1,'%eax','%ebx','%ecx','%edx', '1','0xd76aa478', '7');
 186 round1_step( 0,'%edx','%eax','%ebx','%ecx', '2','0xe8c7b756','12');
 187 round1_step( 0,'%ecx','%edx','%eax','%ebx', '3','0x242070db','17');
 188 round1_step( 0,'%ebx','%ecx','%edx','%eax', '4','0xc1bdceee','22');
 189 round1_step( 0,'%eax','%ebx','%ecx','%edx', '5','0xf57c0faf', '7');
 190 round1_step( 0,'%edx','%eax','%ebx','%ecx', '6','0x4787c62a','12');
 191 round1_step( 0,'%ecx','%edx','%eax','%ebx', '7','0xa8304613','17');
 192 round1_step( 0,'%ebx','%ecx','%edx','%eax', '8','0xfd469501','22');
 193 round1_step( 0,'%eax','%ebx','%ecx','%edx', '9','0x698098d8', '7');
 194 round1_step( 0,'%edx','%eax','%ebx','%ecx','10','0x8b44f7af','12');
 195 round1_step( 0,'%ecx','%edx','%eax','%ebx','11','0xffff5bb1','17');
 196 round1_step( 0,'%ebx','%ecx','%edx','%eax','12','0x895cd7be','22');
 197 round1_step( 0,'%eax','%ebx','%ecx','%edx','13','0x6b901122', '7');
 198 round1_step( 0,'%edx','%eax','%ebx','%ecx','14','0xfd987193','12');
 199 round1_step( 0,'%ecx','%edx','%eax','%ebx','15','0xa679438e','17');
 200 round1_step( 1,'%ebx','%ecx','%edx','%eax', '1','0x49b40821','22');
 201
 202 round2_step(-1,'%eax','%ebx','%ecx','%edx', '6','0xf61e2562', '5');
 203 round2_step( 0,'%edx','%eax','%ebx','%ecx','11','0xc040b340', '9');
 204 round2_step( 0,'%ecx','%edx','%eax','%ebx', '0','0x265e5a51','14');
 205 round2_step( 0,'%ebx','%ecx','%edx','%eax', '5','0xe9b6c7aa','20');
 206 round2_step( 0,'%eax','%ebx','%ecx','%edx','10','0xd62f105d', '5');
 207 round2_step( 0,'%edx','%eax','%ebx','%ecx','15', '0x2441453', '9');
 208 round2_step( 0,'%ecx','%edx','%eax','%ebx', '4','0xd8a1e681','14');
 209 round2_step( 0,'%ebx','%ecx','%edx','%eax', '9','0xe7d3fbc8','20');
 210 round2_step( 0,'%eax','%ebx','%ecx','%edx','14','0x21e1cde6', '5');
 211 round2_step( 0,'%edx','%eax','%ebx','%ecx', '3','0xc33707d6', '9');
 212 round2_step( 0,'%ecx','%edx','%eax','%ebx', '8','0xf4d50d87','14');
 213 round2_step( 0,'%ebx','%ecx','%edx','%eax','13','0x455a14ed','20');
 214 round2_step( 0,'%eax','%ebx','%ecx','%edx', '2','0xa9e3e905', '5');
 215 round2_step( 0,'%edx','%eax','%ebx','%ecx', '7','0xfcefa3f8', '9');
 216 round2_step( 0,'%ecx','%edx','%eax','%ebx','12','0x676f02d9','14');
 217 round2_step( 1,'%ebx','%ecx','%edx','%eax', '5','0x8d2a4c8a','20');
 218
 219 round3_step(-1,'%eax','%ebx','%ecx','%edx', '8','0xfffa3942', '4');
 220 round3_step( 0,'%edx','%eax','%ebx','%ecx','11','0x8771f681','11');
 221 round3_step( 0,'%ecx','%edx','%eax','%ebx','14','0x6d9d6122','16');
 222 round3_step( 0,'%ebx','%ecx','%edx','%eax', '1','0xfde5380c','23');
 223 round3_step( 0,'%eax','%ebx','%ecx','%edx', '4','0xa4beea44', '4');
 224 round3_step( 0,'%edx','%eax','%ebx','%ecx', '7','0x4bdecfa9','11');
 225 round3_step( 0,'%ecx','%edx','%eax','%ebx','10','0xf6bb4b60','16');
 226 round3_step( 0,'%ebx','%ecx','%edx','%eax','13','0xbebfbc70','23');
 227 round3_step( 0,'%eax','%ebx','%ecx','%edx', '0','0x289b7ec6', '4');
 228 round3_step( 0,'%edx','%eax','%ebx','%ecx', '3','0xeaa127fa','11');
 229 round3_step( 0,'%ecx','%edx','%eax','%ebx', '6','0xd4ef3085','16');
 230 round3_step( 0,'%ebx','%ecx','%edx','%eax', '9', '0x4881d05','23');
 231 round3_step( 0,'%eax','%ebx','%ecx','%edx','12','0xd9d4d039', '4');
 232 round3_step( 0,'%edx','%eax','%ebx','%ecx','15','0xe6db99e5','11');
 233 round3_step( 0,'%ecx','%edx','%eax','%ebx', '2','0x1fa27cf8','16');
 234 round3_step( 1,'%ebx','%ecx','%edx','%eax', '0','0xc4ac5665','23');
 235
 236 round4_step(-1,'%eax','%ebx','%ecx','%edx', '7','0xf4292244', '6');
 237 round4_step( 0,'%edx','%eax','%ebx','%ecx','14','0x432aff97','10');
 238 round4_step( 0,'%ecx','%edx','%eax','%ebx', '5','0xab9423a7','15');
 239 round4_step( 0,'%ebx','%ecx','%edx','%eax','12','0xfc93a039','21');
 240 round4_step( 0,'%eax','%ebx','%ecx','%edx', '3','0x655b59c3', '6');
 241 round4_step( 0,'%edx','%eax','%ebx','%ecx','10','0x8f0ccc92','10');
 242 round4_step( 0,'%ecx','%edx','%eax','%ebx', '1','0xffeff47d','15');
 243 round4_step( 0,'%ebx','%ecx','%edx','%eax', '8','0x85845dd1','21');
 244 round4_step( 0,'%eax','%ebx','%ecx','%edx','15','0x6fa87e4f', '6');
 245 round4_step( 0,'%edx','%eax','%ebx','%ecx', '6','0xfe2ce6e0','10');
 246 round4_step( 0,'%ecx','%edx','%eax','%ebx','13','0xa3014314','15');
 247 round4_step( 0,'%ebx','%ecx','%edx','%eax', '4','0x4e0811a1','21');
 248 round4_step( 0,'%eax','%ebx','%ecx','%edx','11','0xf7537e82', '6');
 249 round4_step( 0,'%edx','%eax','%ebx','%ecx', '2','0xbd3af235','10');
 250 round4_step( 0,'%ecx','%edx','%eax','%ebx', '9','0x2ad7d2bb','15');
 251 round4_step( 1,'%ebx','%ecx','%edx','%eax', '0','0xeb86d391','21');
 252 $code .= <<EOF;
 253         # add old values of A, B, C, D
 254         add     %r8d,   %eax
 255         add     %r9d,   %ebx
 256         add     %r14d,  %ecx
 257         add     %r15d,  %edx
 258
 259         # loop control
 260         add     \$64,           %rsi            # ptr += 64
 261         cmp     %rdi,           %rsi            # cmp end with ptr
 262         jb      .Lloop                          # jmp if ptr < end
 263         # END of loop over 16-word blocks
 264
 265 .Lend:
 266         mov     %eax,           0*4(%rbp)       # ctx->A = A
 267         mov     %ebx,           1*4(%rbp)       # ctx->B = B
 268         mov     %ecx,           2*4(%rbp)       # ctx->C = C
 269         mov     %edx,           3*4(%rbp)       # ctx->D = D
 270
 271         mov     (%rsp),%r15
 272 .cfi_restore    %r15
 273         mov     8(%rsp),%r14
 274 .cfi_restore    %r14
 275         mov     16(%rsp),%r12
 276 .cfi_restore    %r12
 277         mov     24(%rsp),%rbx
 278 .cfi_restore    %rbx
 279         mov     32(%rsp),%rbp
 280 .cfi_restore    %rbp
 281         add     \$40,%rsp
 282 .cfi_adjust_cfa_offset  -40
 283 .Lepilogue:
 284         ret
 285 .cfi_endproc
 286 .size md5_block_asm_data_order,.-md5_block_asm_data_order
 287 EOF
 288
 289 # EXCEPTION_DISPOSITION handler (EXCEPTION_RECORD *rec,ULONG64 frame,
 290 #               CONTEXT *context,DISPATCHER_CONTEXT *disp)
 291 if ($win64) {
 292 my $rec="%rcx";
 293 my $frame="%rdx";
 294 my $context="%r8";
 295 my $disp="%r9";
 296
 297 $code.=<<___;
 298 .extern __imp_RtlVirtualUnwind
 299 .type   se_handler,\@abi-omnipotent
 300 .align  16
 301 se_handler:
 302         push    %rsi
 303         push    %rdi
 304         push    %rbx
 305         push    %rbp
 306         push    %r12
 307         push    %r13
 308         push    %r14
 309         push    %r15
 310         pushfq
 311         sub     \$64,%rsp
 312
 313         mov     120($context),%rax      # pull context->Rax
 314         mov     248($context),%rbx      # pull context->Rip
 315
 316         lea     .Lprologue(%rip),%r10
 317         cmp     %r10,%rbx               # context->Rip<.Lprologue
 318         jb      .Lin_prologue
 319
 320         mov     152($context),%rax      # pull context->Rsp
 321
 322         lea     .Lepilogue(%rip),%r10
 323         cmp     %r10,%rbx               # context->Rip>=.Lepilogue
 324         jae     .Lin_prologue
 325
 326         lea     40(%rax),%rax
 327
 328         mov     -8(%rax),%rbp
 329         mov     -16(%rax),%rbx
 330         mov     -24(%rax),%r12
 331         mov     -32(%rax),%r14
 332         mov     -40(%rax),%r15
 333         mov     %rbx,144($context)      # restore context->Rbx
 334         mov     %rbp,160($context)      # restore context->Rbp
 335         mov     %r12,216($context)      # restore context->R12
 336         mov     %r14,232($context)      # restore context->R14
 337         mov     %r15,240($context)      # restore context->R15
 338
 339 .Lin_prologue:
 340         mov     8(%rax),%rdi
 341         mov     16(%rax),%rsi
 342         mov     %rax,152($context)      # restore context->Rsp
 343         mov     %rsi,168($context)      # restore context->Rsi
 344         mov     %rdi,176($context)      # restore context->Rdi
 345
 346         mov     40($disp),%rdi          # disp->ContextRecord
 347         mov     $context,%rsi           # context
 348         mov     \$154,%ecx              # sizeof(CONTEXT)
 349         .long   0xa548f3fc              # cld; rep movsq
 350
 351         mov     $disp,%rsi
 352         xor     %rcx,%rcx               # arg1, UNW_FLAG_NHANDLER
 353         mov     8(%rsi),%rdx            # arg2, disp->ImageBase
 354         mov     0(%rsi),%r8             # arg3, disp->ControlPc
 355         mov     16(%rsi),%r9            # arg4, disp->FunctionEntry
 356         mov     40(%rsi),%r10           # disp->ContextRecord
 357         lea     56(%rsi),%r11           # &disp->HandlerData
 358         lea     24(%rsi),%r12           # &disp->EstablisherFrame
 359         mov     %r10,32(%rsp)           # arg5
 360         mov     %r11,40(%rsp)           # arg6
 361         mov     %r12,48(%rsp)           # arg7
 362         mov     %rcx,56(%rsp)           # arg8, (NULL)
 363         call    *__imp_RtlVirtualUnwind(%rip)
 364
 365         mov     \$1,%eax                # ExceptionContinueSearch
 366         add     \$64,%rsp
 367         popfq
 368         pop     %r15
 369         pop     %r14
 370         pop     %r13
 371         pop     %r12
 372         pop     %rbp
 373         pop     %rbx
 374         pop     %rdi
 375         pop     %rsi
 376         ret
 377 .size   se_handler,.-se_handler
 378
 379 .section        .pdata
 380 .align  4
 381         .rva    .LSEH_begin_md5_block_asm_data_order
 382         .rva    .LSEH_end_md5_block_asm_data_order
 383         .rva    .LSEH_info_md5_block_asm_data_order
 384
 385 .section        .xdata
 386 .align  8
 387 .LSEH_info_md5_block_asm_data_order:
 388         .byte   9,0,0,0
 389         .rva    se_handler
 390 ___
 391 }
 392
 393 print $code;
 394
 395 close STDOUT or die "error closing STDOUT: $!";