Stowage/pycryptodome
2
0
Fork 0
mirror of https://github.com/Legrandin/pycryptodome.git synced 2025-10-19 16:03:45 +00:00
Code Issues Projects Releases Packages Wiki Activity

Better autodetect of AES support, in case of aggressive optimization

Browse source
This commit is contained in:
Helder Eijs 2023-12-29 14:36:19 +01:00
parent 51cd585835
commit b4083688fd
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
compiler_opt.py
View file

@ -140,7 +140,7 @@ def compiler_has_intrin_h():
{ {
int a, b[4]; int a, b[4];
__cpuid(b, a); __cpuid(b, a);
return 0; return a;
} }
""" """
return test_compilation(source, msg="intrin.h header") return test_compilation(source, msg="intrin.h header")
@ -154,7 +154,7 @@ def compiler_has_cpuid_h():
{ {
unsigned int eax, ebx, ecx, edx; unsigned int eax, ebx, ecx, edx;
__get_cpuid(1, &eax, &ebx, &ecx, &edx); __get_cpuid(1, &eax, &ebx, &ecx, &edx);
return 0; return eax;
} }
""" """
return test_compilation(source, msg="cpuid.h header") return test_compilation(source, msg="cpuid.h header")
@ -163,11 +163,16 @@ def compiler_has_cpuid_h():
def compiler_supports_aesni(): def compiler_supports_aesni():
source = """ source = """
#include <wmmintrin.h> #include <wmmintrin.h>
#include <string.h>
__m128i f(__m128i x, __m128i y) { __m128i f(__m128i x, __m128i y) {
return _mm_aesenc_si128(x, y); return _mm_aesenc_si128(x, y);
} }
int main(void) { int main(void) {
return 0; int ret;
__m128i x = _mm_setzero_si128();
x = f(x, x);
memcpy(&ret, &x, sizeof(ret));
return ret;
} }
""" """