Stowage/pycryptodome
2
0
Fork 0
mirror of https://github.com/Legrandin/pycryptodome.git synced 2025-12-08 05:19:46 +00:00
Code Issues Projects Releases Packages Wiki Activity

Speed up AES-GCM with PCLMULDQD

Browse source
This commit is contained in:
Helder Eijs 2018-05-31 22:41:26 +02:00
parent 3bb6b31c17
commit 2522b86c28
12 changed files with 701 additions and 106 deletions
Download patch file Download diff file Expand all files Collapse all files

13
lib/Crypto/Cipher/AES.py
View file

@ -42,9 +42,8 @@ from Crypto.Util._raw_api import (load_pycryptodome_raw_lib,
VoidPointer, SmartPointer,
c_size_t, c_uint8_ptr)
from Crypto.Util import _cpuid
_raw_cpuid_lib = load_pycryptodome_raw_lib("Crypto.Util._cpuid",
"int have_aes_ni(void);")
_cproto = """
int AES_start_operation(const uint8_t key[],
@ -62,15 +61,18 @@ _cproto = """
"""
# Load portable AES
_raw_aes_lib = load_pycryptodome_raw_lib("Crypto.Cipher._raw_aes",
_cproto)
_raw_aesni_lib = None
# Try to load AES with AES NI instructions
try:
if _raw_cpuid_lib.have_aes_ni() == 1:
_raw_aesni_lib = None
if _cpuid.have_aes_ni():
_raw_aesni_lib = load_pycryptodome_raw_lib("Crypto.Cipher._raw_aesni",
_cproto.replace("AES",
"AESNI"))
"AESNI"))
# _raw_aesni may not have been compiled in
except OSError:
pass
@ -199,6 +201,7 @@ def new(key, mode, *args, **kwargs):
kwargs["add_aes_modes"] = True
return _create_cipher(sys.modules[__name__], key, mode, *args, **kwargs)
MODE_ECB = 1
MODE_CBC = 2
MODE_CFB = 3