LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-04 20:33:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add pyaesni support

Browse Source
This commit is contained in:
BlackCatDevel0per 2022-03-23 01:26:19 +05:00
parent 2357116f01
commit d4a387992f
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
telethon/_crypto/aes.py
View File

@ -2,6 +2,7 @@
AES IGE implementation in Python.
If available, cryptg will be used instead, otherwise
If available, pyaesni will be used instead, otherwise
if available, libssl will be used instead, otherwise
the Python implementation will be used.
"""
@ -13,8 +14,19 @@ from . import libssl
__log__ = logging.getLogger(__name__)
try:
import pyaesni
__log__.info('pyaesni detected, it will be used for encryption')
except ImportError:
pyaesni = None
if libssl.encrypt_ige and libssl.decrypt_ige:
__log__.info('libssl detected, it will be used for encryption')
else:
__log__.info('pyaesni module not installed and libssl not found, '
'falling back to (slower) Python encryption')
try:
if not pyaesni:
import cryptg
__log__.info('cryptg detected, it will be used for encryption')
except ImportError:
@ -39,6 +51,8 @@ class AES:
"""
if cryptg:
return cryptg.decrypt_ige(cipher_text, key, iv)
elif pyaesni:
return pyaesni.ige256_decrypt(cipher_text, key, iv)
if libssl.decrypt_ige:
return libssl.decrypt_ige(cipher_text, key, iv)
@ -80,6 +94,8 @@ class AES:
if cryptg:
return cryptg.encrypt_ige(plain_text, key, iv)
elif pyaesni:
return pyaesni.ige256_encrypt(plain_text, key, iv)
if libssl.encrypt_ige:
return libssl.encrypt_ige(plain_text, key, iv)