1 #!/usr/local/bin/perl
   2 
   3 # Normal is the
   4 # md5_block_x86(MD5_CTX *c, ULONG *X);
   5 # version, non-normal is the
   6 # md5_block_x86(MD5_CTX *c, ULONG *X,int blocks);
   7 
   8 $normal=0;
   9 
  10 $0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1;
  11 push(@INC,"${dir}","${dir}../../perlasm");
  12 require "x86asm.pl";
  13 
  14 &asm_init($ARGV[0],$0);
  15 
  16 $A="eax";
  17 $B="ebx";
  18 $C="ecx";
  19 $D="edx";
  20 $tmp1="edi";
  21 $tmp2="ebp";
  22 $X="esi";
  23 
  24 # What we need to load into $tmp for the next round
  25 %Ltmp1=("R0",&Np($C), "R1",&Np($C), "R2",&Np($C), "R3",&Np($D));
  26 @xo=(
  27  0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,  # R0
  28  1, 6, 11, 0, 5, 10, 15, 4, 9, 14, 3, 8, 13, 2, 7, 12,  # R1
  29  5, 8, 11, 14, 1, 4, 7, 10, 13, 0, 3, 6, 9, 12, 15, 2,  # R2
  30  0, 7, 14, 5, 12, 3, 10, 1, 8, 15, 6, 13, 4, 11, 2, 9,  # R3
  31  );
  32 
  33 &md5_block("md5_block_asm_data_order");
  34 &asm_finish();
  35 
  36 sub Np
  37         {
  38         local($p)=@_;
  39         local(%n)=($A,$D,$B,$A,$C,$B,$D,$C);
  40         return($n{$p});
  41         }
  42 
  43 sub R0
  44         {
  45         local($pos,$a,$b,$c,$d,$K,$ki,$s,$t)=@_;
  46 
  47         &mov($tmp1,$C)  if $pos < 0;
  48         &mov($tmp2,&DWP($xo[$ki]*4,$K,"",0)) if $pos < 0; # very first one
  49 
  50         # body proper
  51 
  52         &comment("R0 $ki");
  53         &xor($tmp1,$d); # F function - part 2
  54 
  55         &and($tmp1,$b); # F function - part 3
  56         &lea($a,&DWP($t,$a,$tmp2,1));
  57 
  58         &xor($tmp1,$d); # F function - part 4
  59 
  60         &add($a,$tmp1);
  61         &mov($tmp1,&Np($c)) if $pos < 1;     # next tmp1 for R0
  62         &mov($tmp1,&Np($c)) if $pos == 1;       # next tmp1 for R1
  63 
  64         &rotl($a,$s);
  65 
  66         &mov($tmp2,&DWP($xo[$ki+1]*4,$K,"",0)) if ($pos != 2);
  67 
  68         &add($a,$b);
  69         }
  70 
  71 sub R1
  72         {
  73         local($pos,$a,$b,$c,$d,$K,$ki,$s,$t)=@_;
  74 
  75         &comment("R1 $ki");
  76 
  77         &lea($a,&DWP($t,$a,$tmp2,1));
  78 
  79         &xor($tmp1,$b); # G function - part 2
  80         &and($tmp1,$d); # G function - part 3
  81 
  82         &mov($tmp2,&DWP($xo[$ki+1]*4,$K,"",0)) if ($pos != 2);
  83         &xor($tmp1,$c);                     # G function - part 4
  84 
  85         &add($a,$tmp1);
  86         &mov($tmp1,&Np($c)) if $pos < 1;     # G function - part 1
  87         &mov($tmp1,&Np($c)) if $pos == 1;       # G function - part 1
  88 
  89         &rotl($a,$s);
  90 
  91         &add($a,$b);
  92         }
  93 
  94 sub R2
  95         {
  96         local($n,$pos,$a,$b,$c,$d,$K,$ki,$s,$t)=@_;
  97         # This one is different, only 3 logical operations
  98 
  99 if (($n & 1) == 0)
 100         {
 101         &comment("R2 $ki");
 102         # make sure to do 'D' first, not 'B', else we clash with
 103         # the last add from the previous round.
 104 
 105         &xor($tmp1,$d); # H function - part 2
 106 
 107         &xor($tmp1,$b); # H function - part 3
 108         &lea($a,&DWP($t,$a,$tmp2,1));
 109 
 110         &add($a,$tmp1);
 111 
 112         &rotl($a,$s);
 113 
 114         &mov($tmp2,&DWP($xo[$ki+1]*4,$K,"",0));
 115         &mov($tmp1,&Np($c));
 116         }
 117 else
 118         {
 119         &comment("R2 $ki");
 120         # make sure to do 'D' first, not 'B', else we clash with
 121         # the last add from the previous round.
 122 
 123         &lea($a,&DWP($t,$a,$tmp2,1));
 124 
 125         &add($b,$c);                        # MOVED FORWARD
 126         &xor($tmp1,$d); # H function - part 2
 127 
 128         &xor($tmp1,$b); # H function - part 3
 129         &mov($tmp2,&DWP($xo[$ki+1]*4,$K,"",0)) if ($pos != 2);
 130 
 131         &add($a,$tmp1);
 132         &mov($tmp1,&Np($c)) if $pos < 1;     # H function - part 1
 133         &mov($tmp1,-1) if $pos == 1;                # I function - part 1
 134 
 135         &rotl($a,$s);
 136 
 137         &add($a,$b);
 138         }
 139         }
 140 
 141 sub R3
 142         {
 143         local($pos,$a,$b,$c,$d,$K,$ki,$s,$t)=@_;
 144 
 145         &comment("R3 $ki");
 146 
 147         # &not($tmp1)
 148         &xor($tmp1,$d) if $pos < 0;      # I function - part 2
 149 
 150         &or($tmp1,$b);                              # I function - part 3
 151         &lea($a,&DWP($t,$a,$tmp2,1));
 152 
 153         &xor($tmp1,$c);                     # I function - part 4
 154         &mov($tmp2,&DWP($xo[$ki+1]*4,$K,"",0))  if $pos != 2; # load X/k value
 155         &mov($tmp2,&wparam(0)) if $pos == 2;
 156 
 157         &add($a,$tmp1);
 158         &mov($tmp1,-1) if $pos < 1;      # H function - part 1
 159         &add($K,64) if $pos >=1 && !$normal;
 160 
 161         &rotl($a,$s);
 162 
 163         &xor($tmp1,&Np($d)) if $pos <= 0;    # I function - part = first time
 164         &mov($tmp1,&DWP( 0,$tmp2,"",0)) if $pos > 0;
 165         &add($a,$b);
 166         }
 167 
 168 
 169 sub md5_block
 170         {
 171         local($name)=@_;
 172 
 173         &function_begin_B($name,"",3);
 174 
 175         # parameter 1 is the MD5_CTX structure.
 176         # A     0
 177         # B     4
 178         # C     8
 179         # D     12
 180 
 181         &push("esi");
 182          &push("edi");
 183         &mov($tmp1, &wparam(0)); # edi
 184          &mov($X,   &wparam(1)); # esi
 185         &mov($C,    &wparam(2));
 186          &push("ebp");
 187         &shl($C,    6);
 188         &push("ebx");
 189          &add($C,   $X); # offset we end at
 190         &sub($C,    64);
 191          &mov($A,   &DWP( 0,$tmp1,"",0));
 192         &push($C);  # Put on the TOS
 193          &mov($B,   &DWP( 4,$tmp1,"",0));
 194         &mov($C,    &DWP( 8,$tmp1,"",0));
 195          &mov($D,   &DWP(12,$tmp1,"",0));
 196 
 197         &set_label("start") unless $normal;
 198         &comment("");
 199         &comment("R0 section");
 200 
 201         &R0(-2,$A,$B,$C,$D,$X, 0, 7,0xd76aa478);
 202         &R0( 0,$D,$A,$B,$C,$X, 1,12,0xe8c7b756);
 203         &R0( 0,$C,$D,$A,$B,$X, 2,17,0x242070db);
 204         &R0( 0,$B,$C,$D,$A,$X, 3,22,0xc1bdceee);
 205         &R0( 0,$A,$B,$C,$D,$X, 4, 7,0xf57c0faf);
 206         &R0( 0,$D,$A,$B,$C,$X, 5,12,0x4787c62a);
 207         &R0( 0,$C,$D,$A,$B,$X, 6,17,0xa8304613);
 208         &R0( 0,$B,$C,$D,$A,$X, 7,22,0xfd469501);
 209         &R0( 0,$A,$B,$C,$D,$X, 8, 7,0x698098d8);
 210         &R0( 0,$D,$A,$B,$C,$X, 9,12,0x8b44f7af);
 211         &R0( 0,$C,$D,$A,$B,$X,10,17,0xffff5bb1);
 212         &R0( 0,$B,$C,$D,$A,$X,11,22,0x895cd7be);
 213         &R0( 0,$A,$B,$C,$D,$X,12, 7,0x6b901122);
 214         &R0( 0,$D,$A,$B,$C,$X,13,12,0xfd987193);
 215         &R0( 0,$C,$D,$A,$B,$X,14,17,0xa679438e);
 216         &R0( 1,$B,$C,$D,$A,$X,15,22,0x49b40821);
 217 
 218         &comment("");
 219         &comment("R1 section");
 220         &R1(-1,$A,$B,$C,$D,$X,16, 5,0xf61e2562);
 221         &R1( 0,$D,$A,$B,$C,$X,17, 9,0xc040b340);
 222         &R1( 0,$C,$D,$A,$B,$X,18,14,0x265e5a51);
 223         &R1( 0,$B,$C,$D,$A,$X,19,20,0xe9b6c7aa);
 224         &R1( 0,$A,$B,$C,$D,$X,20, 5,0xd62f105d);
 225         &R1( 0,$D,$A,$B,$C,$X,21, 9,0x02441453);
 226         &R1( 0,$C,$D,$A,$B,$X,22,14,0xd8a1e681);
 227         &R1( 0,$B,$C,$D,$A,$X,23,20,0xe7d3fbc8);
 228         &R1( 0,$A,$B,$C,$D,$X,24, 5,0x21e1cde6);
 229         &R1( 0,$D,$A,$B,$C,$X,25, 9,0xc33707d6);
 230         &R1( 0,$C,$D,$A,$B,$X,26,14,0xf4d50d87);
 231         &R1( 0,$B,$C,$D,$A,$X,27,20,0x455a14ed);
 232         &R1( 0,$A,$B,$C,$D,$X,28, 5,0xa9e3e905);
 233         &R1( 0,$D,$A,$B,$C,$X,29, 9,0xfcefa3f8);
 234         &R1( 0,$C,$D,$A,$B,$X,30,14,0x676f02d9);
 235         &R1( 1,$B,$C,$D,$A,$X,31,20,0x8d2a4c8a);
 236 
 237         &comment("");
 238         &comment("R2 section");
 239         &R2( 0,-1,$A,$B,$C,$D,$X,32, 4,0xfffa3942);
 240         &R2( 1, 0,$D,$A,$B,$C,$X,33,11,0x8771f681);
 241         &R2( 2, 0,$C,$D,$A,$B,$X,34,16,0x6d9d6122);
 242         &R2( 3, 0,$B,$C,$D,$A,$X,35,23,0xfde5380c);
 243         &R2( 4, 0,$A,$B,$C,$D,$X,36, 4,0xa4beea44);
 244         &R2( 5, 0,$D,$A,$B,$C,$X,37,11,0x4bdecfa9);
 245         &R2( 6, 0,$C,$D,$A,$B,$X,38,16,0xf6bb4b60);
 246         &R2( 7, 0,$B,$C,$D,$A,$X,39,23,0xbebfbc70);
 247         &R2( 8, 0,$A,$B,$C,$D,$X,40, 4,0x289b7ec6);
 248         &R2( 9, 0,$D,$A,$B,$C,$X,41,11,0xeaa127fa);
 249         &R2(10, 0,$C,$D,$A,$B,$X,42,16,0xd4ef3085);
 250         &R2(11, 0,$B,$C,$D,$A,$X,43,23,0x04881d05);
 251         &R2(12, 0,$A,$B,$C,$D,$X,44, 4,0xd9d4d039);
 252         &R2(13, 0,$D,$A,$B,$C,$X,45,11,0xe6db99e5);
 253         &R2(14, 0,$C,$D,$A,$B,$X,46,16,0x1fa27cf8);
 254         &R2(15, 1,$B,$C,$D,$A,$X,47,23,0xc4ac5665);
 255 
 256         &comment("");
 257         &comment("R3 section");
 258         &R3(-1,$A,$B,$C,$D,$X,48, 6,0xf4292244);
 259         &R3( 0,$D,$A,$B,$C,$X,49,10,0x432aff97);
 260         &R3( 0,$C,$D,$A,$B,$X,50,15,0xab9423a7);
 261         &R3( 0,$B,$C,$D,$A,$X,51,21,0xfc93a039);
 262         &R3( 0,$A,$B,$C,$D,$X,52, 6,0x655b59c3);
 263         &R3( 0,$D,$A,$B,$C,$X,53,10,0x8f0ccc92);
 264         &R3( 0,$C,$D,$A,$B,$X,54,15,0xffeff47d);
 265         &R3( 0,$B,$C,$D,$A,$X,55,21,0x85845dd1);
 266         &R3( 0,$A,$B,$C,$D,$X,56, 6,0x6fa87e4f);
 267         &R3( 0,$D,$A,$B,$C,$X,57,10,0xfe2ce6e0);
 268         &R3( 0,$C,$D,$A,$B,$X,58,15,0xa3014314);
 269         &R3( 0,$B,$C,$D,$A,$X,59,21,0x4e0811a1);
 270         &R3( 0,$A,$B,$C,$D,$X,60, 6,0xf7537e82);
 271         &R3( 0,$D,$A,$B,$C,$X,61,10,0xbd3af235);
 272         &R3( 0,$C,$D,$A,$B,$X,62,15,0x2ad7d2bb);
 273         &R3( 2,$B,$C,$D,$A,$X,63,21,0xeb86d391);
 274 
 275         # &mov($tmp2,&wparam(0));       # done in the last R3
 276         # &mov($tmp1,       &DWP( 0,$tmp2,"",0)); # done is the last R3
 277 
 278         &add($A,$tmp1);
 279          &mov($tmp1,        &DWP( 4,$tmp2,"",0));
 280 
 281         &add($B,$tmp1);
 282         &mov($tmp1, &DWP( 8,$tmp2,"",0));
 283 
 284         &add($C,$tmp1);
 285         &mov($tmp1, &DWP(12,$tmp2,"",0));
 286 
 287         &add($D,$tmp1);
 288         &mov(&DWP( 0,$tmp2,"",0),$A);
 289 
 290         &mov(&DWP( 4,$tmp2,"",0),$B);
 291         &mov($tmp1,&swtmp(0)) unless $normal;
 292 
 293         &mov(&DWP( 8,$tmp2,"",0),$C);
 294          &mov(&DWP(12,$tmp2,"",0),$D);
 295 
 296         &cmp($tmp1,$X) unless $normal;                      # check count
 297          &jae(&label("start")) unless $normal;
 298 
 299         &pop("eax"); # pop the temp variable off the stack
 300          &pop("ebx");
 301         &pop("ebp");
 302          &pop("edi");
 303         &pop("esi");
 304          &ret();
 305         &function_end_B($name);
 306         }