| #!/usr/bin/perl -w |
| # |
| # MD5 optimized for AMD64. |
| # |
| # Author: Marc Bevand <bevand_m (at) epita.fr> |
| # Licence: I hereby disclaim the copyright on this code and place it |
| # in the public domain. |
| # |
| |
| use strict; |
| |
| my $code; |
| |
| # round1_step() does: |
| # dst = x + ((dst + F(x,y,z) + X[k] + T_i) <<< s) |
| # %r10d = X[k_next] |
| # %r11d = z' (copy of z for the next step) |
| # Each round1_step() takes about 5.3 clocks (9 instructions, 1.7 IPC) |
| sub round1_step |
| { |
| my ($pos, $dst, $x, $y, $z, $k_next, $T_i, $s) = @_; |
| $code .= " mov 0*4(%rsi), %r10d /* (NEXT STEP) X[0] */\n" if ($pos == -1); |
| $code .= " mov %edx, %r11d /* (NEXT STEP) z' = %edx */\n" if ($pos == -1); |
| $code .= <<EOF; |
| xor $y, %r11d /* y ^ ... */ |
| lea $T_i($dst,%r10d),$dst /* Const + dst + ... */ |
| and $x, %r11d /* x & ... */ |
| xor $z, %r11d /* z ^ ... */ |
| mov $k_next*4(%rsi),%r10d /* (NEXT STEP) X[$k_next] */ |
| add %r11d, $dst /* dst += ... */ |
| rol \$$s, $dst /* dst <<< s */ |
| mov $y, %r11d /* (NEXT STEP) z' = $y */ |
| add $x, $dst /* dst += x */ |
| EOF |
| } |
| |
| # round2_step() does: |
| # dst = x + ((dst + G(x,y,z) + X[k] + T_i) <<< s) |
| # %r10d = X[k_next] |
| # %r11d = z' (copy of z for the next step) |
| # %r12d = z' (copy of z for the next step) |
| # Each round2_step() takes about 5.4 clocks (11 instructions, 2.0 IPC) |
| sub round2_step |
| { |
| my ($pos, $dst, $x, $y, $z, $k_next, $T_i, $s) = @_; |
| $code .= " mov 1*4(%rsi), %r10d /* (NEXT STEP) X[1] */\n" if ($pos == -1); |
| $code .= " mov %edx, %r11d /* (NEXT STEP) z' = %edx */\n" if ($pos == -1); |
| $code .= " mov %edx, %r12d /* (NEXT STEP) z' = %edx */\n" if ($pos == -1); |
| $code .= <<EOF; |
| not %r11d /* not z */ |
| lea $T_i($dst,%r10d),$dst /* Const + dst + ... */ |
| and $x, %r12d /* x & z */ |
| and $y, %r11d /* y & (not z) */ |
| mov $k_next*4(%rsi),%r10d /* (NEXT STEP) X[$k_next] */ |
| or %r11d, %r12d /* (y & (not z)) | (x & z) */ |
| mov $y, %r11d /* (NEXT STEP) z' = $y */ |
| add %r12d, $dst /* dst += ... */ |
| mov $y, %r12d /* (NEXT STEP) z' = $y */ |
| rol \$$s, $dst /* dst <<< s */ |
| add $x, $dst /* dst += x */ |
| EOF |
| } |
| |
| # round3_step() does: |
| # dst = x + ((dst + H(x,y,z) + X[k] + T_i) <<< s) |
| # %r10d = X[k_next] |
| # %r11d = y' (copy of y for the next step) |
| # Each round3_step() takes about 4.2 clocks (8 instructions, 1.9 IPC) |
| sub round3_step |
| { |
| my ($pos, $dst, $x, $y, $z, $k_next, $T_i, $s) = @_; |
| $code .= " mov 5*4(%rsi), %r10d /* (NEXT STEP) X[5] */\n" if ($pos == -1); |
| $code .= " mov %ecx, %r11d /* (NEXT STEP) y' = %ecx */\n" if ($pos == -1); |
| $code .= <<EOF; |
| lea $T_i($dst,%r10d),$dst /* Const + dst + ... */ |
| mov $k_next*4(%rsi),%r10d /* (NEXT STEP) X[$k_next] */ |
| xor $z, %r11d /* z ^ ... */ |
| xor $x, %r11d /* x ^ ... */ |
| add %r11d, $dst /* dst += ... */ |
| rol \$$s, $dst /* dst <<< s */ |
| mov $x, %r11d /* (NEXT STEP) y' = $x */ |
| add $x, $dst /* dst += x */ |
| EOF |
| } |
| |
| # round4_step() does: |
| # dst = x + ((dst + I(x,y,z) + X[k] + T_i) <<< s) |
| # %r10d = X[k_next] |
| # %r11d = not z' (copy of not z for the next step) |
| # Each round4_step() takes about 5.2 clocks (9 instructions, 1.7 IPC) |
| sub round4_step |
| { |
| my ($pos, $dst, $x, $y, $z, $k_next, $T_i, $s) = @_; |
| $code .= " mov 0*4(%rsi), %r10d /* (NEXT STEP) X[0] */\n" if ($pos == -1); |
| $code .= " mov \$0xffffffff, %r11d\n" if ($pos == -1); |
| $code .= " xor %edx, %r11d /* (NEXT STEP) not z' = not %edx*/\n" |
| if ($pos == -1); |
| $code .= <<EOF; |
| lea $T_i($dst,%r10d),$dst /* Const + dst + ... */ |
| or $x, %r11d /* x | ... */ |
| xor $y, %r11d /* y ^ ... */ |
| add %r11d, $dst /* dst += ... */ |
| mov $k_next*4(%rsi),%r10d /* (NEXT STEP) X[$k_next] */ |
| mov \$0xffffffff, %r11d |
| rol \$$s, $dst /* dst <<< s */ |
| xor $y, %r11d /* (NEXT STEP) not z' = not $y */ |
| add $x, $dst /* dst += x */ |
| EOF |
| } |
| |
| my $flavour = shift; |
| my $output = shift; |
| if ($flavour =~ /\./) { $output = $flavour; undef $flavour; } |
| |
| my $win64=0; $win64=1 if ($flavour =~ /[nm]asm|mingw64/ || $output =~ /\.asm$/); |
| |
| $0 =~ m/(.*[\/\\])[^\/\\]+$/; my $dir=$1; my $xlate; |
| ( $xlate="${dir}x86_64-xlate.pl" and -f $xlate ) or |
| ( $xlate="${dir}../../perlasm/x86_64-xlate.pl" and -f $xlate) or |
| die "can't locate x86_64-xlate.pl"; |
| |
| no warnings qw(uninitialized); |
| open OUT,"| \"$^X\" $xlate $flavour $output"; |
| *STDOUT=*OUT; |
| |
| $code .= <<EOF; |
| .text |
| .align 16 |
| |
| .globl md5_block_asm_data_order |
| .type md5_block_asm_data_order,\@function,3 |
| md5_block_asm_data_order: |
| push %rbp |
| push %rbx |
| push %r12 |
| push %r14 |
| push %r15 |
| .Lprologue: |
| |
| # rdi = arg #1 (ctx, MD5_CTX pointer) |
| # rsi = arg #2 (ptr, data pointer) |
| # rdx = arg #3 (nbr, number of 16-word blocks to process) |
| mov %rdi, %rbp # rbp = ctx |
| shl \$6, %rdx # rdx = nbr in bytes |
| lea (%rsi,%rdx), %rdi # rdi = end |
| mov 0*4(%rbp), %eax # eax = ctx->A |
| mov 1*4(%rbp), %ebx # ebx = ctx->B |
| mov 2*4(%rbp), %ecx # ecx = ctx->C |
| mov 3*4(%rbp), %edx # edx = ctx->D |
| # end is 'rdi' |
| # ptr is 'rsi' |
| # A is 'eax' |
| # B is 'ebx' |
| # C is 'ecx' |
| # D is 'edx' |
| |
| cmp %rdi, %rsi # cmp end with ptr |
| je .Lend # jmp if ptr == end |
| |
| # BEGIN of loop over 16-word blocks |
| .Lloop: # save old values of A, B, C, D |
| mov %eax, %r8d |
| mov %ebx, %r9d |
| mov %ecx, %r14d |
| mov %edx, %r15d |
| EOF |
| round1_step(-1,'%eax','%ebx','%ecx','%edx', '1','0xd76aa478', '7'); |
| round1_step( 0,'%edx','%eax','%ebx','%ecx', '2','0xe8c7b756','12'); |
| round1_step( 0,'%ecx','%edx','%eax','%ebx', '3','0x242070db','17'); |
| round1_step( 0,'%ebx','%ecx','%edx','%eax', '4','0xc1bdceee','22'); |
| round1_step( 0,'%eax','%ebx','%ecx','%edx', '5','0xf57c0faf', '7'); |
| round1_step( 0,'%edx','%eax','%ebx','%ecx', '6','0x4787c62a','12'); |
| round1_step( 0,'%ecx','%edx','%eax','%ebx', '7','0xa8304613','17'); |
| round1_step( 0,'%ebx','%ecx','%edx','%eax', '8','0xfd469501','22'); |
| round1_step( 0,'%eax','%ebx','%ecx','%edx', '9','0x698098d8', '7'); |
| round1_step( 0,'%edx','%eax','%ebx','%ecx','10','0x8b44f7af','12'); |
| round1_step( 0,'%ecx','%edx','%eax','%ebx','11','0xffff5bb1','17'); |
| round1_step( 0,'%ebx','%ecx','%edx','%eax','12','0x895cd7be','22'); |
| round1_step( 0,'%eax','%ebx','%ecx','%edx','13','0x6b901122', '7'); |
| round1_step( 0,'%edx','%eax','%ebx','%ecx','14','0xfd987193','12'); |
| round1_step( 0,'%ecx','%edx','%eax','%ebx','15','0xa679438e','17'); |
| round1_step( 1,'%ebx','%ecx','%edx','%eax', '0','0x49b40821','22'); |
| |
| round2_step(-1,'%eax','%ebx','%ecx','%edx', '6','0xf61e2562', '5'); |
| round2_step( 0,'%edx','%eax','%ebx','%ecx','11','0xc040b340', '9'); |
| round2_step( 0,'%ecx','%edx','%eax','%ebx', '0','0x265e5a51','14'); |
| round2_step( 0,'%ebx','%ecx','%edx','%eax', '5','0xe9b6c7aa','20'); |
| round2_step( 0,'%eax','%ebx','%ecx','%edx','10','0xd62f105d', '5'); |
| round2_step( 0,'%edx','%eax','%ebx','%ecx','15', '0x2441453', '9'); |
| round2_step( 0,'%ecx','%edx','%eax','%ebx', '4','0xd8a1e681','14'); |
| round2_step( 0,'%ebx','%ecx','%edx','%eax', '9','0xe7d3fbc8','20'); |
| round2_step( 0,'%eax','%ebx','%ecx','%edx','14','0x21e1cde6', '5'); |
| round2_step( 0,'%edx','%eax','%ebx','%ecx', '3','0xc33707d6', '9'); |
| round2_step( 0,'%ecx','%edx','%eax','%ebx', '8','0xf4d50d87','14'); |
| round2_step( 0,'%ebx','%ecx','%edx','%eax','13','0x455a14ed','20'); |
| round2_step( 0,'%eax','%ebx','%ecx','%edx', '2','0xa9e3e905', '5'); |
| round2_step( 0,'%edx','%eax','%ebx','%ecx', '7','0xfcefa3f8', '9'); |
| round2_step( 0,'%ecx','%edx','%eax','%ebx','12','0x676f02d9','14'); |
| round2_step( 1,'%ebx','%ecx','%edx','%eax', '0','0x8d2a4c8a','20'); |
| |
| round3_step(-1,'%eax','%ebx','%ecx','%edx', '8','0xfffa3942', '4'); |
| round3_step( 0,'%edx','%eax','%ebx','%ecx','11','0x8771f681','11'); |
| round3_step( 0,'%ecx','%edx','%eax','%ebx','14','0x6d9d6122','16'); |
| round3_step( 0,'%ebx','%ecx','%edx','%eax', '1','0xfde5380c','23'); |
| round3_step( 0,'%eax','%ebx','%ecx','%edx', '4','0xa4beea44', '4'); |
| round3_step( 0,'%edx','%eax','%ebx','%ecx', '7','0x4bdecfa9','11'); |
| round3_step( 0,'%ecx','%edx','%eax','%ebx','10','0xf6bb4b60','16'); |
| round3_step( 0,'%ebx','%ecx','%edx','%eax','13','0xbebfbc70','23'); |
| round3_step( 0,'%eax','%ebx','%ecx','%edx', '0','0x289b7ec6', '4'); |
| round3_step( 0,'%edx','%eax','%ebx','%ecx', '3','0xeaa127fa','11'); |
| round3_step( 0,'%ecx','%edx','%eax','%ebx', '6','0xd4ef3085','16'); |
| round3_step( 0,'%ebx','%ecx','%edx','%eax', '9', '0x4881d05','23'); |
| round3_step( 0,'%eax','%ebx','%ecx','%edx','12','0xd9d4d039', '4'); |
| round3_step( 0,'%edx','%eax','%ebx','%ecx','15','0xe6db99e5','11'); |
| round3_step( 0,'%ecx','%edx','%eax','%ebx', '2','0x1fa27cf8','16'); |
| round3_step( 1,'%ebx','%ecx','%edx','%eax', '0','0xc4ac5665','23'); |
| |
| round4_step(-1,'%eax','%ebx','%ecx','%edx', '7','0xf4292244', '6'); |
| round4_step( 0,'%edx','%eax','%ebx','%ecx','14','0x432aff97','10'); |
| round4_step( 0,'%ecx','%edx','%eax','%ebx', '5','0xab9423a7','15'); |
| round4_step( 0,'%ebx','%ecx','%edx','%eax','12','0xfc93a039','21'); |
| round4_step( 0,'%eax','%ebx','%ecx','%edx', '3','0x655b59c3', '6'); |
| round4_step( 0,'%edx','%eax','%ebx','%ecx','10','0x8f0ccc92','10'); |
| round4_step( 0,'%ecx','%edx','%eax','%ebx', '1','0xffeff47d','15'); |
| round4_step( 0,'%ebx','%ecx','%edx','%eax', '8','0x85845dd1','21'); |
| round4_step( 0,'%eax','%ebx','%ecx','%edx','15','0x6fa87e4f', '6'); |
| round4_step( 0,'%edx','%eax','%ebx','%ecx', '6','0xfe2ce6e0','10'); |
| round4_step( 0,'%ecx','%edx','%eax','%ebx','13','0xa3014314','15'); |
| round4_step( 0,'%ebx','%ecx','%edx','%eax', '4','0x4e0811a1','21'); |
| round4_step( 0,'%eax','%ebx','%ecx','%edx','11','0xf7537e82', '6'); |
| round4_step( 0,'%edx','%eax','%ebx','%ecx', '2','0xbd3af235','10'); |
| round4_step( 0,'%ecx','%edx','%eax','%ebx', '9','0x2ad7d2bb','15'); |
| round4_step( 1,'%ebx','%ecx','%edx','%eax', '0','0xeb86d391','21'); |
| $code .= <<EOF; |
| # add old values of A, B, C, D |
| add %r8d, %eax |
| add %r9d, %ebx |
| add %r14d, %ecx |
| add %r15d, %edx |
| |
| # loop control |
| add \$64, %rsi # ptr += 64 |
| cmp %rdi, %rsi # cmp end with ptr |
| jb .Lloop # jmp if ptr < end |
| # END of loop over 16-word blocks |
| |
| .Lend: |
| mov %eax, 0*4(%rbp) # ctx->A = A |
| mov %ebx, 1*4(%rbp) # ctx->B = B |
| mov %ecx, 2*4(%rbp) # ctx->C = C |
| mov %edx, 3*4(%rbp) # ctx->D = D |
| |
| mov (%rsp),%r15 |
| mov 8(%rsp),%r14 |
| mov 16(%rsp),%r12 |
| mov 24(%rsp),%rbx |
| mov 32(%rsp),%rbp |
| add \$40,%rsp |
| .Lepilogue: |
| ret |
| .size md5_block_asm_data_order,.-md5_block_asm_data_order |
| EOF |
| |
| # EXCEPTION_DISPOSITION handler (EXCEPTION_RECORD *rec,ULONG64 frame, |
| # CONTEXT *context,DISPATCHER_CONTEXT *disp) |
| if ($win64) { |
| my $rec="%rcx"; |
| my $frame="%rdx"; |
| my $context="%r8"; |
| my $disp="%r9"; |
| |
| $code.=<<___; |
| .extern __imp_RtlVirtualUnwind |
| .type se_handler,\@abi-omnipotent |
| .align 16 |
| se_handler: |
| push %rsi |
| push %rdi |
| push %rbx |
| push %rbp |
| push %r12 |
| push %r13 |
| push %r14 |
| push %r15 |
| pushfq |
| sub \$64,%rsp |
| |
| mov 120($context),%rax # pull context->Rax |
| mov 248($context),%rbx # pull context->Rip |
| |
| lea .Lprologue(%rip),%r10 |
| cmp %r10,%rbx # context->Rip<.Lprologue |
| jb .Lin_prologue |
| |
| mov 152($context),%rax # pull context->Rsp |
| |
| lea .Lepilogue(%rip),%r10 |
| cmp %r10,%rbx # context->Rip>=.Lepilogue |
| jae .Lin_prologue |
| |
| lea 40(%rax),%rax |
| |
| mov -8(%rax),%rbp |
| mov -16(%rax),%rbx |
| mov -24(%rax),%r12 |
| mov -32(%rax),%r14 |
| mov -40(%rax),%r15 |
| mov %rbx,144($context) # restore context->Rbx |
| mov %rbp,160($context) # restore context->Rbp |
| mov %r12,216($context) # restore context->R12 |
| mov %r14,232($context) # restore context->R14 |
| mov %r15,240($context) # restore context->R15 |
| |
| .Lin_prologue: |
| mov 8(%rax),%rdi |
| mov 16(%rax),%rsi |
| mov %rax,152($context) # restore context->Rsp |
| mov %rsi,168($context) # restore context->Rsi |
| mov %rdi,176($context) # restore context->Rdi |
| |
| mov 40($disp),%rdi # disp->ContextRecord |
| mov $context,%rsi # context |
| mov \$154,%ecx # sizeof(CONTEXT) |
| .long 0xa548f3fc # cld; rep movsq |
| |
| mov $disp,%rsi |
| xor %rcx,%rcx # arg1, UNW_FLAG_NHANDLER |
| mov 8(%rsi),%rdx # arg2, disp->ImageBase |
| mov 0(%rsi),%r8 # arg3, disp->ControlPc |
| mov 16(%rsi),%r9 # arg4, disp->FunctionEntry |
| mov 40(%rsi),%r10 # disp->ContextRecord |
| lea 56(%rsi),%r11 # &disp->HandlerData |
| lea 24(%rsi),%r12 # &disp->EstablisherFrame |
| mov %r10,32(%rsp) # arg5 |
| mov %r11,40(%rsp) # arg6 |
| mov %r12,48(%rsp) # arg7 |
| mov %rcx,56(%rsp) # arg8, (NULL) |
| call *__imp_RtlVirtualUnwind(%rip) |
| |
| mov \$1,%eax # ExceptionContinueSearch |
| add \$64,%rsp |
| popfq |
| pop %r15 |
| pop %r14 |
| pop %r13 |
| pop %r12 |
| pop %rbp |
| pop %rbx |
| pop %rdi |
| pop %rsi |
| ret |
| .size se_handler,.-se_handler |
| |
| .section .pdata |
| .align 4 |
| .rva .LSEH_begin_md5_block_asm_data_order |
| .rva .LSEH_end_md5_block_asm_data_order |
| .rva .LSEH_info_md5_block_asm_data_order |
| |
| .section .xdata |
| .align 8 |
| .LSEH_info_md5_block_asm_data_order: |
| .byte 9,0,0,0 |
| .rva se_handler |
| ___ |
| } |
| |
| print $code; |
| |
| close STDOUT; |