Add tgcrypto & pyaesni for faster work

optional-requirements.txt

@@ -1,4 +1,6 @@
+tgcrypto
 cryptg
+pyaesni
 pysocks
 python-socks[asyncio]
 hachoir

readthedocs/basic/installation.rst

@@ -51,7 +51,7 @@ The version number of the library should show in the output.
 Optional Dependencies
 =====================
 
-If cryptg_ is installed, **the library will work a lot faster**, since
+If tgcrypto_ or cryptg_ or pyaesni(intel aes-ni) is installed, **the library will work a lot faster**, since
 encryption and decryption will be made in C instead of Python. If your
 code deals with a lot of updates or you are downloading/uploading a lot
 of files, you will notice a considerable speed-up (from a hundred kilobytes
@@ -83,11 +83,12 @@ manually.
         apt update
         apt install clang lib{jpeg-turbo,webp}-dev python{,-dev} zlib-dev
         pip install -U --user setuptools
-        pip install -U --user telethon cryptg pillow
+        pip install -U --user telethon tgcrypto pillow
 
     Thanks to `@bb010g`_ for writing down this nice list.
 
 
+.. _tgcrypto: https://github.com/pyrogram/tgcrypto
 .. _cryptg: https://github.com/cher-nov/cryptg
 .. _pyaes: https://github.com/ricmoo/pyaes
 .. _pillow: https://python-pillow.org

setup.py

@@ -234,7 +234,7 @@ def main(argv):
             ]),
             install_requires=['pyaes', 'rsa'],
             extras_require={
-                'cryptg': ['cryptg']
+                'tgcrypto': ['tgcrypto']
             }
         )
 

telethon/_client/telegramclient.py

@@ -1676,7 +1676,7 @@ async def download_media(
         Downloads the given media from a message object.
 
         Note that if the download is too slow, you should consider installing
-        ``cryptg`` (through ``pip install cryptg``) so that decrypting the
+        ``tgcrypto`` (through ``pip install tgcrypto``) so that decrypting the
         received data is done in C instead of Python (much faster).
 
         See also `Message.download_media() <telethon.tl._custom.message.Message.download_media>`.

telethon/_crypto/aes.py

@@ -2,6 +2,7 @@
 AES IGE implementation in Python.
 
 If available, cryptg will be used instead, otherwise
+If available, pyaesni(intel aes-ni for python) will be used instead, otherwise
 if available, libssl will be used instead, otherwise
 the Python implementation will be used.
 """
@@ -12,19 +13,35 @@
 
 
 __log__ = logging.getLogger(__name__)
-
+# tgcrypto
+try:
+    import tgcrypto
+    __log__.info('tgcrypto detected, it will be used for encryption')
+except ImportError:
+    tgcrypto = None
+    __log__.info('tgcrypto module not installed and libssl not found, '
+                 'falling back to (slower) Python encryption')
+# pyaesni (intel aes-ni)
+try:
+    if not tgcrypto:
+        import pyaesni
+        __log__.info('pyaesni detected, it will be used for encryption')
+except ImportError:
+    pyaesni = None
+    __log__.info('pyaesni module not installed and libssl not found, '
+                 'falling back to (slower) Python encryption')
 
 try:
-    import cryptg
-    __log__.info('cryptg detected, it will be used for encryption')
+    if not tgcrypto and not pyaesni:
+      import cryptg
+      __log__.info('cryptg detected, it will be used for encryption')
 except ImportError:
     cryptg = None
+    __log__.info('cryptg module not installed and libssl not found, '
+                 'falling back to (slower) Python encryption')
+if not tgcrypto and not pyaesni and not cryptg:
     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, '
-                     'falling back to (slower) Python encryption')
-
 
 class AES:
     """
@@ -37,8 +54,12 @@ def decrypt_ige(cipher_text, key, iv):
         Decrypts the given text in 16-bytes blocks by using the
         given key and 32-bytes initialization vector.
         """
-        if cryptg:
+        if tgcrypto:
+            return tgcrypto.ige256_decrypt(cipher_text, key, iv)
+        elif 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)
 
@@ -78,8 +99,12 @@ def encrypt_ige(plain_text, key, iv):
         if padding:
             plain_text += os.urandom(16 - padding)
 
-        if cryptg:
+        if tgcrypto:
+            return tgcrypto.ige256_encrypt(plain_text, key, iv)
+        elif 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)