#!/usr/bin/env perl # ==================================================================== # Written by Andy Polyakov for the OpenSSL # project. The module is, however, dual licensed under OpenSSL and # CRYPTOGAMS licenses depending on where you obtain it. For further # details see http://www.openssl.org/~appro/cryptogams/. # ==================================================================== # September 2011 # # Assembler helpers for Padlock engine. Compared to original engine # version relying on inline assembler and compiled with gcc 3.4.6 it # was measured to provide ~100% improvement on misaligned data in ECB # mode and ~75% in CBC mode. For aligned data improvement can be # observed for short inputs only, e.g. 45% for 64-byte messages in # ECB mode, 20% in CBC. Difference in performance for aligned vs. # misaligned data depends on misalignment and is either ~1.8x or 2.9x. # These are approximately same factors as for hardware support, so # there is little reason to rely on the latter. On the contrary, it # might actually hurt performance in mixture of aligned and misaligned # buffers, because a) if you choose to flip 'align' flag in control # word on per-buffer basis, then you'd have to reload key context, # which incurs penalty; b) if you choose to set 'align' flag # permanently, it limits performance even for aligned data to ~1/2. # All above mentioned results were collected on 1.5GHz C7. Nano on the # other hand handles unaligned data more gracefully. Depending on # algorithm and how unaligned data is, hardware can be up to 70% more # efficient than below software alignment procedures, nor does 'align' # flag have affect on aligned performance [if has any meaning at all]. # Therefore suggestion is to unconditionally set 'align' flag on Nano # for optimal performance. $0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1; push(@INC,"${dir}","${dir}../../crypto/perlasm"); require "x86asm.pl"; $output=pop; open STDOUT,">$output"; &asm_init($ARGV[0],$0); %PADLOCK_PREFETCH=(ecb=>128, cbc=>64); # prefetch errata $PADLOCK_CHUNK=512; # Must be a power of 2 larger than 16 $ctx="edx"; $out="edi"; $inp="esi"; $len="ecx"; $chunk="ebx"; &function_begin_B("padlock_capability"); &push ("ebx"); &pushf (); &pop ("eax"); &mov ("ecx","eax"); &xor ("eax",1<<21); &push ("eax"); &popf (); &pushf (); &pop ("eax"); &xor ("ecx","eax"); &xor ("eax","eax"); &bt ("ecx",21); &jnc (&label("noluck")); &cpuid (); &xor ("eax","eax"); &cmp ("ebx","0x".unpack("H*",'tneC')); &jne (&label("noluck")); &cmp ("edx","0x".unpack("H*",'Hrua')); &jne (&label("noluck")); &cmp ("ecx","0x".unpack("H*",'slua')); &jne (&label("noluck")); &mov ("eax",0xC0000000); &cpuid (); &mov ("edx","eax"); &xor ("eax","eax"); &cmp ("edx",0xC0000001); &jb (&label("noluck")); &mov ("eax",1); &cpuid (); &or ("eax",0x0f); &xor ("ebx","ebx"); &and ("eax",0x0fff); &cmp ("eax",0x06ff); # check for Nano &sete ("bl"); &mov ("eax",0xC0000001); &push ("ebx"); &cpuid (); &pop ("ebx"); &mov ("eax","edx"); &shl ("ebx",4); # bit#4 denotes Nano &and ("eax",0xffffffef); &or ("eax","ebx") &set_label("noluck"); &pop ("ebx"); &ret (); &function_end_B("padlock_capability") &function_begin_B("padlock_key_bswap"); &mov ("edx",&wparam(0)); &mov ("ecx",&DWP(240,"edx")); &set_label("bswap_loop"); &mov ("eax",&DWP(0,"edx")); &bswap ("eax"); &mov (&DWP(0,"edx"),"eax"); &lea ("edx",&DWP(4,"edx")); &sub ("ecx",1); &jnz (&label("bswap_loop")); &ret (); &function_end_B("padlock_key_bswap"); # This is heuristic key context tracing. At first one # believes that one should use atomic swap instructions, # but it's not actually necessary. Point is that if # padlock_saved_context was changed by another thread # after we've read it and before we compare it with ctx, # our key *shall* be reloaded upon thread context switch # and we are therefore set in either case... &static_label("padlock_saved_context"); &function_begin_B("padlock_verify_context"); &mov ($ctx,&wparam(0)); &lea ("eax",($::win32 or $::coff) ? &DWP(&label("padlock_saved_context")) : &DWP(&label("padlock_saved_context")."-".&label("verify_pic_point"))); &pushf (); &call ("_padlock_verify_ctx"); &set_label("verify_pic_point"); &lea ("esp",&DWP(4,"esp")); &ret (); &function_end_B("padlock_verify_context"); &function_begin_B("_padlock_verify_ctx"); &add ("eax",&DWP(0,"esp")) if(!($::win32 or $::coff));# &padlock_saved_context &bt (&DWP(4,"esp"),30); # eflags &jnc (&label("verified")); &cmp ($ctx,&DWP(0,"eax")); &je (&label("verified")); &pushf (); &popf (); &set_label("verified"); &mov (&DWP(0,"eax"),$ctx); &ret (); &function_end_B("_padlock_verify_ctx"); &function_begin_B("padlock_reload_key"); &pushf (); &popf (); &ret (); &function_end_B("padlock_reload_key"); &function_begin_B("padlock_aes_block"); &push ("edi"); &push ("esi"); &push ("ebx"); &mov ($out,&wparam(0)); # must be 16-byte aligned &mov ($inp,&wparam(1)); # must be 16-byte aligned &mov ($ctx,&wparam(2)); &mov ($len,1); &lea ("ebx",&DWP(32,$ctx)); # key &lea ($ctx,&DWP(16,$ctx)); # control word &data_byte(0xf3,0x0f,0xa7,0xc8); # rep xcryptecb &pop ("ebx"); &pop ("esi"); &pop ("edi"); &ret (); &function_end_B("padlock_aes_block"); sub generate_mode { my ($mode,$opcode) = @_; # int padlock_$mode_encrypt(void *out, const void *inp, # struct padlock_cipher_data *ctx, size_t len); &function_begin("padlock_${mode}_encrypt"); &mov ($out,&wparam(0)); &mov ($inp,&wparam(1)); &mov ($ctx,&wparam(2)); &mov ($len,&wparam(3)); &test ($ctx,15); &jnz (&label("${mode}_abort")); &test ($len,15); &jnz (&label("${mode}_abort")); &lea ("eax",($::win32 or $::coff) ? &DWP(&label("padlock_saved_context")) : &DWP(&label("padlock_saved_context")."-".&label("${mode}_pic_point"))); &pushf (); &cld (); &call ("_padlock_verify_ctx"); &set_label("${mode}_pic_point"); &lea ($ctx,&DWP(16,$ctx)); # control word &xor ("eax","eax"); if ($mode eq "ctr32") { &movq ("mm0",&QWP(-16,$ctx)); # load [upper part of] counter } else { &xor ("ebx","ebx"); &test (&DWP(0,$ctx),1<<5); # align bit in control word &jnz (&label("${mode}_aligned")); &test ($out,0x0f); &setz ("al"); # !out_misaligned &test ($inp,0x0f); &setz ("bl"); # !inp_misaligned &test ("eax","ebx"); &jnz (&label("${mode}_aligned")); &neg ("eax"); } &mov ($chunk,$PADLOCK_CHUNK); ¬ ("eax"); # out_misaligned?-1:0 &lea ("ebp",&DWP(-24,"esp")); &cmp ($len,$chunk); &cmovc ($chunk,$len); # chunk=len>PADLOCK_CHUNK?PADLOCK_CHUNK:len &and ("eax",$chunk); # out_misaligned?chunk:0 &mov ($chunk,$len); &neg ("eax"); &and ($chunk,$PADLOCK_CHUNK-1); # chunk=len%PADLOCK_CHUNK &lea ("esp",&DWP(0,"eax","ebp")); # alloca &mov ("eax",$PADLOCK_CHUNK); &cmovz ($chunk,"eax"); # chunk=chunk?:PADLOCK_CHUNK &mov ("eax","ebp"); &and ("ebp",-16); &and ("esp",-16); &mov (&DWP(16,"ebp"),"eax"); if ($PADLOCK_PREFETCH{$mode}) { &cmp ($len,$chunk); &ja (&label("${mode}_loop")); &mov ("eax",$inp); # check if prefetch crosses page &cmp ("ebp","esp"); &cmove ("eax",$out); &add ("eax",$len); &neg ("eax"); &and ("eax",0xfff); # distance to page boundary &cmp ("eax",$PADLOCK_PREFETCH{$mode}); &mov ("eax",-$PADLOCK_PREFETCH{$mode}); &cmovae ("eax",$chunk); # mask=distanceExceptionCode == STATUS_ACCESS_VIOLATION &jne (&label("ret")); &add (&DWP(184,"ecx"),4); # skip over rep sha* &mov ("eax",0); # ExceptionContinueExecution &set_label("ret"); &ret (); &function_end_B("_win32_segv_handler"); &safeseh("_win32_segv_handler") if ($::win32); &function_begin_B("padlock_sha1_oneshot"); &push ("edi"); &push ("esi"); &xor ("eax","eax"); &mov ("edi",&wparam(0)); &mov ("esi",&wparam(1)); &mov ("ecx",&wparam(2)); if ($::win32 or $::coff) { &push (&::islabel("_win32_segv_handler")); &data_byte(0x64,0xff,0x30); # push %fs:(%eax) &data_byte(0x64,0x89,0x20); # mov %esp,%fs:(%eax) } &mov ("edx","esp"); # put aside %esp &add ("esp",-128); # 32 is enough but spec says 128 &movups ("xmm0",&QWP(0,"edi")); # copy-in context &and ("esp",-16); &mov ("eax",&DWP(16,"edi")); &movaps (&QWP(0,"esp"),"xmm0"); &mov ("edi","esp"); &mov (&DWP(16,"esp"),"eax"); &xor ("eax","eax"); &data_byte(0xf3,0x0f,0xa6,0xc8); # rep xsha1 &movaps ("xmm0",&QWP(0,"esp")); &mov ("eax",&DWP(16,"esp")); &mov ("esp","edx"); # restore %esp if ($::win32 or $::coff) { &data_byte(0x64,0x8f,0x05,0,0,0,0); # pop %fs:0 &lea ("esp",&DWP(4,"esp")); } &mov ("edi",&wparam(0)); &movups (&QWP(0,"edi"),"xmm0"); # copy-out context &mov (&DWP(16,"edi"),"eax"); &pop ("esi"); &pop ("edi"); &ret (); &function_end_B("padlock_sha1_oneshot"); &function_begin_B("padlock_sha1_blocks"); &push ("edi"); &push ("esi"); &mov ("edi",&wparam(0)); &mov ("esi",&wparam(1)); &mov ("edx","esp"); # put aside %esp &mov ("ecx",&wparam(2)); &add ("esp",-128); &movups ("xmm0",&QWP(0,"edi")); # copy-in context &and ("esp",-16); &mov ("eax",&DWP(16,"edi")); &movaps (&QWP(0,"esp"),"xmm0"); &mov ("edi","esp"); &mov (&DWP(16,"esp"),"eax"); &mov ("eax",-1); &data_byte(0xf3,0x0f,0xa6,0xc8); # rep xsha1 &movaps ("xmm0",&QWP(0,"esp")); &mov ("eax",&DWP(16,"esp")); &mov ("esp","edx"); # restore %esp &mov ("edi",&wparam(0)); &movups (&QWP(0,"edi"),"xmm0"); # copy-out context &mov (&DWP(16,"edi"),"eax"); &pop ("esi"); &pop ("edi"); &ret (); &function_end_B("padlock_sha1_blocks"); &function_begin_B("padlock_sha256_oneshot"); &push ("edi"); &push ("esi"); &xor ("eax","eax"); &mov ("edi",&wparam(0)); &mov ("esi",&wparam(1)); &mov ("ecx",&wparam(2)); if ($::win32 or $::coff) { &push (&::islabel("_win32_segv_handler")); &data_byte(0x64,0xff,0x30); # push %fs:(%eax) &data_byte(0x64,0x89,0x20); # mov %esp,%fs:(%eax) } &mov ("edx","esp"); # put aside %esp &add ("esp",-128); &movups ("xmm0",&QWP(0,"edi")); # copy-in context &and ("esp",-16); &movups ("xmm1",&QWP(16,"edi")); &movaps (&QWP(0,"esp"),"xmm0"); &mov ("edi","esp"); &movaps (&QWP(16,"esp"),"xmm1"); &xor ("eax","eax"); &data_byte(0xf3,0x0f,0xa6,0xd0); # rep xsha256 &movaps ("xmm0",&QWP(0,"esp")); &movaps ("xmm1",&QWP(16,"esp")); &mov ("esp","edx"); # restore %esp if ($::win32 or $::coff) { &data_byte(0x64,0x8f,0x05,0,0,0,0); # pop %fs:0 &lea ("esp",&DWP(4,"esp")); } &mov ("edi",&wparam(0)); &movups (&QWP(0,"edi"),"xmm0"); # copy-out context &movups (&QWP(16,"edi"),"xmm1"); &pop ("esi"); &pop ("edi"); &ret (); &function_end_B("padlock_sha256_oneshot"); &function_begin_B("padlock_sha256_blocks"); &push ("edi"); &push ("esi"); &mov ("edi",&wparam(0)); &mov ("esi",&wparam(1)); &mov ("ecx",&wparam(2)); &mov ("edx","esp"); # put aside %esp &add ("esp",-128); &movups ("xmm0",&QWP(0,"edi")); # copy-in context &and ("esp",-16); &movups ("xmm1",&QWP(16,"edi")); &movaps (&QWP(0,"esp"),"xmm0"); &mov ("edi","esp"); &movaps (&QWP(16,"esp"),"xmm1"); &mov ("eax",-1); &data_byte(0xf3,0x0f,0xa6,0xd0); # rep xsha256 &movaps ("xmm0",&QWP(0,"esp")); &movaps ("xmm1",&QWP(16,"esp")); &mov ("esp","edx"); # restore %esp &mov ("edi",&wparam(0)); &movups (&QWP(0,"edi"),"xmm0"); # copy-out context &movups (&QWP(16,"edi"),"xmm1"); &pop ("esi"); &pop ("edi"); &ret (); &function_end_B("padlock_sha256_blocks"); &function_begin_B("padlock_sha512_blocks"); &push ("edi"); &push ("esi"); &mov ("edi",&wparam(0)); &mov ("esi",&wparam(1)); &mov ("ecx",&wparam(2)); &mov ("edx","esp"); # put aside %esp &add ("esp",-128); &movups ("xmm0",&QWP(0,"edi")); # copy-in context &and ("esp",-16); &movups ("xmm1",&QWP(16,"edi")); &movups ("xmm2",&QWP(32,"edi")); &movups ("xmm3",&QWP(48,"edi")); &movaps (&QWP(0,"esp"),"xmm0"); &mov ("edi","esp"); &movaps (&QWP(16,"esp"),"xmm1"); &movaps (&QWP(32,"esp"),"xmm2"); &movaps (&QWP(48,"esp"),"xmm3"); &data_byte(0xf3,0x0f,0xa6,0xe0); # rep xsha512 &movaps ("xmm0",&QWP(0,"esp")); &movaps ("xmm1",&QWP(16,"esp")); &movaps ("xmm2",&QWP(32,"esp")); &movaps ("xmm3",&QWP(48,"esp")); &mov ("esp","edx"); # restore %esp &mov ("edi",&wparam(0)); &movups (&QWP(0,"edi"),"xmm0"); # copy-out context &movups (&QWP(16,"edi"),"xmm1"); &movups (&QWP(32,"edi"),"xmm2"); &movups (&QWP(48,"edi"),"xmm3"); &pop ("esi"); &pop ("edi"); &ret (); &function_end_B("padlock_sha512_blocks"); &asciz ("VIA Padlock x86 module, CRYPTOGAMS by "); &align (16); &dataseg(); # Essentially this variable belongs in thread local storage. # Having this variable global on the other hand can only cause # few bogus key reloads [if any at all on signle-CPU system], # so we accept the penalty... &set_label("padlock_saved_context",4); &data_word(0); &asm_finish(); close STDOUT;