Reuse the AESModeCTR class on CdnDecrypter

This commit is contained in:
Lonami Exo
2017-08-28 20:30:33 +02:00
parent 5404670469
commit bc72e52834
2 changed files with 7 additions and 19 deletions
+3 -11
View File
@@ -12,18 +12,10 @@ class AESModeCTR:
assert isinstance(iv, bytes)
assert len(iv) == 16
self.iv = iv
self._aes._counter._counter = list(self.iv)
def reset(self):
pass
self._aes._counter._counter = list(iv)
def encrypt(self, data):
result = self._aes.encrypt(data)
self.reset()
return result
return self._aes.encrypt(data)
def decrypt(self, data):
result = self._aes.decrypt(data)
self.reset()
return result
return self._aes.decrypt(data)