LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-03-22 02:44:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Avoid defining Python's AES class if libssl's is available

Browse Source
This commit is contained in:
Lonami Exo 2017-08-20 01:23:57 +02:00
parent c296459d0a
commit 274e16ac66
1 changed files with 72 additions and 70 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
telethon/crypto/aes.py
View File

@ -2,6 +2,12 @@ import os
import pyaes import pyaes
from . import libssl from . import libssl
if libssl.libssl:
# Use libssl if available, since it will be faster
AES = libssl.AES
else:
# Fallback to a pure Python implementation
class AES: class AES:
@staticmethod @staticmethod
def decrypt_ige(cipher_text, key, iv): def decrypt_ige(cipher_text, key, iv):
@ -71,7 +77,3 @@ class AES:
cipher_text.extend(cipher_text_block) cipher_text.extend(cipher_text_block)
return bytes(cipher_text) return bytes(cipher_text)
# use libssl if available
if libssl.libssl:
AES = libssl.AES