1 #!/usr/bin/env perl 2 3 $0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1; 4 push(@INC, "${dir}perlasm", "perlasm"); 5 require "x86asm.pl"; 6 7 &asm_init($ARGV[0],"x86cpuid"); 8 9 for (@ARGV) { $sse2=1 if (/-DOPENSSL_IA32_SSE2/); } 10 11 &function_begin("OPENSSL_ia32_cpuid"); 12 &xor ("edx","edx"); 13 &pushf (); 14 &pop ("eax"); 15 &mov ("ecx","eax"); 16 &xor ("eax",1<<21); 17 &push ("eax"); 18 &popf (); 19 &pushf (); 20 &pop ("eax"); 21 &xor ("ecx","eax"); 22 &xor ("eax","eax"); 23 &bt ("ecx",21); 24 &jnc (&label("nocpuid")); 25 &cpuid (); 26 &mov ("edi","eax"); # max value for standard query level 27 28 &xor ("eax","eax"); 29 &cmp ("ebx",0x756e6547); # "Genu" 30 &setne (&LB("eax")); 31 &mov ("ebp","eax"); 32 &cmp ("edx",0x49656e69); # "ineI" 33 &setne (&LB("eax")); 34 &or ("ebp","eax"); 35 &cmp ("ecx",0x6c65746e); # "ntel" 36 &setne (&LB("eax")); 37 &or ("ebp","eax"); # 0 indicates Intel CPU 38 &jz (&label("intel")); 39 40 &cmp ("ebx",0x68747541); # "Auth" 41 &setne (&LB("eax")); 42 &mov ("esi","eax"); 43 &cmp ("edx",0x69746E65); # "enti" 44 &setne (&LB("eax")); 45 &or ("esi","eax"); 46 &cmp ("ecx",0x444D4163); # "cAMD" 47 &setne (&LB("eax")); 48 &or ("esi","eax"); # 0 indicates AMD CPU 49 &jnz (&label("intel")); 50 51 # AMD specific 52 &mov ("eax",0x80000000); 53 &cpuid (); 54 &cmp ("eax",0x80000001); 55 &jb (&label("intel")); 56 &mov ("esi","eax"); 57 &mov ("eax",0x80000001); 58 &cpuid (); 59 &or ("ebp","ecx"); 60 &and ("ebp",1<<11|1); # isolate XOP bit 61 &cmp ("esi",0x80000008); 62 &jb (&label("intel")); 63 64 &mov ("eax",0x80000008); 65 &cpuid (); 66 &movz ("esi",&LB("ecx")); # number of cores - 1 67 &inc ("esi"); # number of cores 68 69 &mov ("eax",1); 70 &xor ("ecx","ecx"); 71 &cpuid (); 72 &bt ("edx",28); 73 &jnc (&label("generic")); 74 &shr ("ebx",16); 75 &and ("ebx",0xff); 76 &cmp ("ebx","esi"); 77 &ja (&label("generic")); 78 &and ("edx",0xefffffff); # clear hyper-threading bit 79 &jmp (&label("generic")); 80 81 &set_label("intel"); 82 &cmp ("edi",4); 83 &mov ("edi",-1); 84 &jb (&label("nocacheinfo")); 85 86 &mov ("eax",4); 87 &mov ("ecx",0); # query L1D 88 &cpuid (); 89 &mov ("edi","eax"); 90 &shr ("edi",14); 91 &and ("edi",0xfff); # number of cores -1 per L1D 92 93 &set_label("nocacheinfo"); 94 &mov ("eax",1); 95 &xor ("ecx","ecx"); 96 &cpuid (); 97 &and ("edx",0xbfefffff); # force reserved bits #20, #30 to 0 98 &cmp ("ebp",0); 99 &jne (&label("notintel")); 100 &or ("edx",1<<30); # set reserved bit#30 on Intel CPUs 101 &and (&HB("eax"),15); # familiy ID 102 &cmp (&HB("eax"),15); # P4? 103 &jne (&label("notintel")); 104 &or ("edx",1<<20); # set reserved bit#20 to engage RC4_CHAR 105 &set_label("notintel"); 106 &bt ("edx",28); # test hyper-threading bit 107 &jnc (&label("generic")); 108 &and ("edx",0xefffffff); 109 &cmp ("edi",0); 110 &je (&label("generic")); 111 112 &or ("edx",0x10000000); 113 &shr ("ebx",16); 114 &cmp (&LB("ebx"),1); 115 &ja (&label("generic")); 116 &and ("edx",0xefffffff); # clear hyper-threading bit if not 117 118 &set_label("generic"); 119 &and ("ebp",1<<11); # isolate AMD XOP flag 120 &and ("ecx",0xfffff7ff); # force 11th bit to 0 121 &mov ("esi","edx"); 122 &or ("ebp","ecx"); # merge AMD XOP flag 123 124 &bt ("ecx",27); # check OSXSAVE bit 125 &jnc (&label("clear_avx")); 126 &xor ("ecx","ecx"); 127 &data_byte(0x0f,0x01,0xd0); # xgetbv 128 &and ("eax",6); 129 &cmp ("eax",6); 130 &je (&label("done")); 131 &cmp ("eax",2); 132 &je (&label("clear_avx")); 133 &set_label("clear_xmm"); 134 &and ("ebp",0xfdfffffd); # clear AESNI and PCLMULQDQ bits 135 &and ("esi",0xfeffffff); # clear FXSR 136 &set_label("clear_avx"); 137 &and ("ebp",0xefffe7ff); # clear AVX, FMA and AMD XOP bits 138 &set_label("done"); 139 &mov ("eax","esi"); 140 &mov ("edx","ebp"); 141 &set_label("nocpuid"); 142 &function_end("OPENSSL_ia32_cpuid"); 143 144 &external_label("OPENSSL_ia32cap_P"); 145 146 &function_begin_B("OPENSSL_rdtsc","EXTRN\t_OPENSSL_ia32cap_P:DWORD"); 147 &xor ("eax","eax"); 148 &xor ("edx","edx"); 149 &picmeup("ecx","OPENSSL_ia32cap_P"); 150 &bt (&DWP(0,"ecx"),4); 151 &jnc (&label("notsc")); 152 &rdtsc (); 153 &set_label("notsc"); 154 &ret (); 155 &function_end_B("OPENSSL_rdtsc"); 156 157 # This works in Ring 0 only [read DJGPP+MS-DOS+privileged DPMI host], 158 # but it's safe to call it on any [supported] 32-bit platform... 159 # Just check for [non-]zero return value... 160 &function_begin_B("OPENSSL_instrument_halt","EXTRN\t_OPENSSL_ia32cap_P:DWORD"); 161 &picmeup("ecx","OPENSSL_ia32cap_P"); 162 &bt (&DWP(0,"ecx"),4); 163 &jnc (&label("nohalt")); # no TSC 164 165 &data_word(0x9058900e); # push %cs; pop %eax 166 &and ("eax",3); 167 &jnz (&label("nohalt")); # not enough privileges 168 169 &pushf (); 170 &pop ("eax"); 171 &bt ("eax",9); 172 &jnc (&label("nohalt")); # interrupts are disabled 173 174 &rdtsc (); 175 &push ("edx"); 176 &push ("eax"); 177 &halt (); 178 &rdtsc (); 179 180 &sub ("eax",&DWP(0,"esp")); 181 &sbb ("edx",&DWP(4,"esp")); 182 &add ("esp",8); 183 &ret (); 184 185 &set_label("nohalt"); 186 &xor ("eax","eax"); 187 &xor ("edx","edx"); 188 &ret (); 189 &function_end_B("OPENSSL_instrument_halt"); 190 191 # Essentially there is only one use for this function. Under DJGPP: 192 # 193 # #include <go32.h> 194 # ... 195 # i=OPENSSL_far_spin(_dos_ds,0x46c); 196 # ... 197 # to obtain the number of spins till closest timer interrupt. 198 199 &function_begin_B("OPENSSL_far_spin"); 200 &pushf (); 201 &pop ("eax") 202 &bt ("eax",9); 203 &jnc (&label("nospin")); # interrupts are disabled 204 205 &mov ("eax",&DWP(4,"esp")); 206 &mov ("ecx",&DWP(8,"esp")); 207 &data_word (0x90d88e1e); # push %ds, mov %eax,%ds 208 &xor ("eax","eax"); 209 &mov ("edx",&DWP(0,"ecx")); 210 &jmp (&label("spin")); 211 212 &align (16); 213 &set_label("spin"); 214 &inc ("eax"); 215 &cmp ("edx",&DWP(0,"ecx")); 216 &je (&label("spin")); 217 218 &data_word (0x1f909090); # pop %ds 219 &ret (); 220 221 &set_label("nospin"); 222 &xor ("eax","eax"); 223 &xor ("edx","edx"); 224 &ret (); 225 &function_end_B("OPENSSL_far_spin"); 226 227 &function_begin_B("OPENSSL_wipe_cpu","EXTRN\t_OPENSSL_ia32cap_P:DWORD"); 228 &xor ("eax","eax"); 229 &xor ("edx","edx"); 230 &picmeup("ecx","OPENSSL_ia32cap_P"); 231 &mov ("ecx",&DWP(0,"ecx")); 232 &bt (&DWP(0,"ecx"),1); 233 &jnc (&label("no_x87")); 234 if ($sse2) { 235 &and ("ecx",1<<26|1<<24); # check SSE2 and FXSR bits 236 &cmp ("ecx",1<<26|1<<24); 237 &jne (&label("no_sse2")); 238 &pxor ("xmm0","xmm0"); 239 &pxor ("xmm1","xmm1"); 240 &pxor ("xmm2","xmm2"); 241 &pxor ("xmm3","xmm3"); 242 &pxor ("xmm4","xmm4"); 243 &pxor ("xmm5","xmm5"); 244 &pxor ("xmm6","xmm6"); 245 &pxor ("xmm7","xmm7"); 246 &set_label("no_sse2"); 247 } 248 # just a bunch of fldz to zap the fp/mm bank followed by finit... 249 &data_word(0xeed9eed9,0xeed9eed9,0xeed9eed9,0xeed9eed9,0x90e3db9b); 250 &set_label("no_x87"); 251 &lea ("eax",&DWP(4,"esp")); 252 &ret (); 253 &function_end_B("OPENSSL_wipe_cpu"); 254 255 &function_begin_B("OPENSSL_atomic_add"); 256 &mov ("edx",&DWP(4,"esp")); # fetch the pointer, 1st arg 257 &mov ("ecx",&DWP(8,"esp")); # fetch the increment, 2nd arg 258 &push ("ebx"); 259 &nop (); 260 &mov ("eax",&DWP(0,"edx")); 261 &set_label("spin"); 262 &lea ("ebx",&DWP(0,"eax","ecx")); 263 &nop (); 264 &data_word(0x1ab10ff0); # lock; cmpxchg %ebx,(%edx) # %eax is envolved and is always reloaded 265 &jne (&label("spin")); 266 &mov ("eax","ebx"); # OpenSSL expects the new value 267 &pop ("ebx"); 268 &ret (); 269 &function_end_B("OPENSSL_atomic_add"); 270 271 # This function can become handy under Win32 in situations when 272 # we don't know which calling convention, __stdcall or __cdecl(*), 273 # indirect callee is using. In C it can be deployed as 274 # 275 #ifdef OPENSSL_CPUID_OBJ 276 # type OPENSSL_indirect_call(void *f,...); 277 # ... 278 # OPENSSL_indirect_call(func,[up to $max arguments]); 279 #endif 280 # 281 # (*) it's designed to work even for __fastcall if number of 282 # arguments is 1 or 2! 283 &function_begin_B("OPENSSL_indirect_call"); 284 { 285 my ($max,$i)=(7,); # $max has to be chosen as 4*n-1 286 # in order to preserve eventual 287 # stack alignment 288 &push ("ebp"); 289 &mov ("ebp","esp"); 290 &sub ("esp",$max*4); 291 &mov ("ecx",&DWP(12,"ebp")); 292 &mov (&DWP(0,"esp"),"ecx"); 293 &mov ("edx",&DWP(16,"ebp")); 294 &mov (&DWP(4,"esp"),"edx"); 295 for($i=2;$i<$max;$i++) 296 { 297 # Some copies will be redundant/bogus... 298 &mov ("eax",&DWP(12+$i*4,"ebp")); 299 &mov (&DWP(0+$i*4,"esp"),"eax"); 300 } 301 &call_ptr (&DWP(8,"ebp"));# make the call... 302 &mov ("esp","ebp"); # ... and just restore the stack pointer 303 # without paying attention to what we called, 304 # (__cdecl *func) or (__stdcall *one). 305 &pop ("ebp"); 306 &ret (); 307 } 308 &function_end_B("OPENSSL_indirect_call"); 309 310 &function_begin_B("OPENSSL_cleanse"); 311 &mov ("edx",&wparam(0)); 312 &mov ("ecx",&wparam(1)); 313 &xor ("eax","eax"); 314 &cmp ("ecx",7); 315 &jae (&label("lot")); 316 &cmp ("ecx",0); 317 &je (&label("ret")); 318 &set_label("little"); 319 &mov (&BP(0,"edx"),"al"); 320 &sub ("ecx",1); 321 &lea ("edx",&DWP(1,"edx")); 322 &jnz (&label("little")); 323 &set_label("ret"); 324 &ret (); 325 326 &set_label("lot",16); 327 &test ("edx",3); 328 &jz (&label("aligned")); 329 &mov (&BP(0,"edx"),"al"); 330 &lea ("ecx",&DWP(-1,"ecx")); 331 &lea ("edx",&DWP(1,"edx")); 332 &jmp (&label("lot")); 333 &set_label("aligned"); 334 &mov (&DWP(0,"edx"),"eax"); 335 &lea ("ecx",&DWP(-4,"ecx")); 336 &test ("ecx",-4); 337 &lea ("edx",&DWP(4,"edx")); 338 &jnz (&label("aligned")); 339 &cmp ("ecx",0); 340 &jne (&label("little")); 341 &ret (); 342 &function_end_B("OPENSSL_cleanse"); 343 344 &function_begin_B("OPENSSL_ia32_rdrand"); 345 &mov ("ecx",8); 346 &set_label("loop"); 347 &rdrand ("eax"); 348 &jc (&label("break")); 349 &loop (&label("loop")); 350 &set_label("break"); 351 &cmp ("eax",0); 352 &cmove ("eax","ecx"); 353 &ret (); 354 &function_end_B("OPENSSL_ia32_rdrand"); 355 356 &initseg("OPENSSL_cpuid_setup"); 357 358 &hidden("OPENSSL_cpuid_setup"); 359 &hidden("OPENSSL_ia32cap_P"); 360 361 &asm_finish(); 362