gandc/mtprotoproxy
1
0
Fork 0
mirror of https://github.com/alexbers/mtprotoproxy.git synced 2026-03-20 09:35:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

do not create the copy of cryptography adapter classes

Browse Source
This commit is contained in:
Alexander Bersenev
2019-08-21 16:14:48 +05:00
parent 014e450e62
commit 25d76bee09
1 changed files with 30 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
mtprotoproxy.py
View File

@@ -207,40 +207,27 @@ def try_use_cryptography_module():
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
from cryptography.hazmat.backends import default_backend from cryptography.hazmat.backends import default_backend
class CryptographyEncryptorAdapter:
__slots__ = ('encryptor', 'decryptor')
def __init__(self, cipher):
self.encryptor = cipher.encryptor()
self.decryptor = cipher.decryptor()
def encrypt(self, data):
return self.encryptor.update(data)
def decrypt(self, data):
return self.decryptor.update(data)
def create_aes_ctr(key, iv): def create_aes_ctr(key, iv):
class EncryptorAdapter:
__slots__ = ('encryptor', 'decryptor')
def __init__(self, cipher):
self.encryptor = cipher.encryptor()
self.decryptor = cipher.decryptor()
def encrypt(self, data):
return self.encryptor.update(data)
def decrypt(self, data):
return self.decryptor.update(data)
iv_bytes = int.to_bytes(iv, 16, "big") iv_bytes = int.to_bytes(iv, 16, "big")
cipher = Cipher(algorithms.AES(key), modes.CTR(iv_bytes), default_backend()) cipher = Cipher(algorithms.AES(key), modes.CTR(iv_bytes), default_backend())
return EncryptorAdapter(cipher) return CryptographyEncryptorAdapter(cipher)
def create_aes_cbc(key, iv): def create_aes_cbc(key, iv):
class EncryptorAdapter:
__slots__ = ('encryptor', 'decryptor')
def __init__(self, cipher):
self.encryptor = cipher.encryptor()
self.decryptor = cipher.decryptor()
def encrypt(self, data):
return self.encryptor.update(data)
def decrypt(self, data):
return self.decryptor.update(data)
cipher = Cipher(algorithms.AES(key), modes.CBC(iv), default_backend()) cipher = Cipher(algorithms.AES(key), modes.CBC(iv), default_backend())
return EncryptorAdapter(cipher) return CryptographyEncryptorAdapter(cipher)
return create_aes_ctr, create_aes_cbc return create_aes_ctr, create_aes_cbc
@@ -266,27 +253,27 @@ def use_slow_bundled_cryptography_module():
msg += "pip install cryptography\n" msg += "pip install cryptography\n"
print(msg, flush=True, file=sys.stderr) print(msg, flush=True, file=sys.stderr)
class BundledEncryptorAdapter:
__slots__ = ('mode', )
def __init__(self, mode):
self.mode = mode
def encrypt(self, data):
encrypter = pyaes.Encrypter(self.mode, pyaes.PADDING_NONE)
return encrypter.feed(data) + encrypter.feed()
def decrypt(self, data):
decrypter = pyaes.Decrypter(self.mode, pyaes.PADDING_NONE)
return decrypter.feed(data) + decrypter.feed()
def create_aes_ctr(key, iv): def create_aes_ctr(key, iv):
ctr = pyaes.Counter(iv) ctr = pyaes.Counter(iv)
return pyaes.AESModeOfOperationCTR(key, ctr) return pyaes.AESModeOfOperationCTR(key, ctr)
def create_aes_cbc(key, iv): def create_aes_cbc(key, iv):
class EncryptorAdapter:
__slots__ = ('mode', )
def __init__(self, mode):
self.mode = mode
def encrypt(self, data):
encrypter = pyaes.Encrypter(self.mode, pyaes.PADDING_NONE)
return encrypter.feed(data) + encrypter.feed()
def decrypt(self, data):
decrypter = pyaes.Decrypter(self.mode, pyaes.PADDING_NONE)
return decrypter.feed(data) + decrypter.feed()
mode = pyaes.AESModeOfOperationCBC(key, iv) mode = pyaes.AESModeOfOperationCBC(key, iv)
return EncryptorAdapter(mode) return BundledEncryptorAdapter(mode)
return create_aes_ctr, create_aes_cbc return create_aes_ctr, create_aes_cbc