mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-15 05:56:36 +03:00
4d3ff0e175
This reverts commit 42cc9e61fb.
tgcrypto was made for Pyrogram, and seeing it used elsewhere
without much credit "hurts" the author. I personally do not endorse
its use, hence the lack of attention or notes in the documentation.
People who still want to benefit from the speed boost should go
out of their way to discover, install and patch Telethon's aes.py
module instead, all while complying with the respective license
(another reason to avoid said code in Telethon, which is under the
much more permissive MIT license).
People using tgcrypto for anything other than Pyrogram will do so
knowing full-well that this was not the library's intended usage.
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""
|
|
This module holds the AESModeCTR wrapper class.
|
|
"""
|
|
import pyaes
|
|
|
|
|
|
class AESModeCTR:
|
|
"""Wrapper around pyaes.AESModeOfOperationCTR mode with custom IV"""
|
|
# TODO Maybe make a pull request to pyaes to support iv on CTR
|
|
|
|
def __init__(self, key, iv):
|
|
"""
|
|
Initializes the AES CTR mode with the given key/iv pair.
|
|
|
|
:param key: the key to be used as bytes.
|
|
:param iv: the bytes initialization vector. Must have a length of 16.
|
|
"""
|
|
# TODO Use libssl if available
|
|
assert isinstance(key, bytes)
|
|
self._aes = pyaes.AESModeOfOperationCTR(key)
|
|
|
|
assert isinstance(iv, bytes)
|
|
assert len(iv) == 16
|
|
self._aes._counter._counter = list(iv)
|
|
|
|
def encrypt(self, data):
|
|
"""
|
|
Encrypts the given plain text through AES CTR.
|
|
|
|
:param data: the plain text to be encrypted.
|
|
:return: the encrypted cipher text.
|
|
"""
|
|
return self._aes.encrypt(data)
|
|
|
|
def decrypt(self, data):
|
|
"""
|
|
Decrypts the given cipher text through AES CTR
|
|
|
|
:param data: the cipher text to be decrypted.
|
|
:return: the decrypted plain text.
|
|
"""
|
|
return self._aes.decrypt(data)
|