Alexandre Savard | 1b09e31 | 2012-08-07 20:33:29 -0400 | [diff] [blame] | 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 | # ¬($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 | } |
| 307 | |