LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-30 17:59:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add pyaesni support when found

Browse Source
This commit is contained in:
painor 2021-04-01 14:32:08 +01:00
parent 4b16183d2b
commit b4737e52f3
2 changed files with 22 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
optional-requirements.txt
View File

@ -1,3 +1,4 @@
pyaesni
cryptg
pysocks
python-socks[asyncio]

16
telethon/crypto/aes.py
View File

@ -1,7 +1,8 @@
"""
AES IGE implementation in Python.
If available, cryptg will be used instead, otherwise
If available, pyaesni will be used instead, otherwise
if available, cryptg will be used instead, otherwise
if available, libssl will be used instead, otherwise
the Python implementation will be used.
"""
@ -10,19 +11,24 @@ import pyaes
import logging
from . import libssl
__log__ = logging.getLogger(__name__)
try:
import pyaesni
__log__.info('pyaesni detected, it will be used for encryption')
except ImportError:
pyaesni = None
try:
import cryptg
__log__.info('cryptg detected, it will be used for encryption')
except ImportError:
cryptg = None
if libssl.encrypt_ige and libssl.decrypt_ige:
__log__.info('libssl detected, it will be used for encryption')
else:
__log__.info('cryptg module not installed and libssl not found, '
__log__.info('pyaesni or cryptg modules not installed and libssl not found, '
'falling back to (slower) Python encryption')
@ -37,6 +43,8 @@ class AES:
Decrypts the given text in 16-bytes blocks by using the
given key and 32-bytes initialization vector.
"""
if pyaesni:
return pyaesni.ige256_decrypt(cipher_text, key, iv)
if cryptg:
return cryptg.decrypt_ige(cipher_text, key, iv)
if libssl.decrypt_ige:
@ -78,6 +86,8 @@ class AES:
if padding:
plain_text += os.urandom(16 - padding)
if pyaesni:
return pyaesni.ige256_encrypt(plain_text, key, iv)
if cryptg:
return cryptg.encrypt_ige(plain_text, key, iv)
if libssl.encrypt_ige: