diff --git a/lib/megaserver/cursor.py b/lib/megaserver/cursor.py index ec387b9c..32b84e6a 100644 --- a/lib/megaserver/cursor.py +++ b/lib/megaserver/cursor.py @@ -66,8 +66,14 @@ class Cursor(object): def prepare_decoder(self,offset): initial_value = self.initial_value + int(offset/16) - from lib import pyaes - self.decryptor = pyaes.AESModeOfOperationCTR(self._file._client.a32_to_str(self.k), counter=pyaes.Counter(initial_value=initial_value)) + try: + from Cryptodome.Cipher import AES + from Cryptodome.Util import Counter + self.decryptor = AES.new(self._file._client.a32_to_str(self.k), AES.MODE_CTR, counter = Counter.new(128, initial_value = initial_value)) + except: + from Crypto.Cipher import AES + from Crypto.Util import Counter + self.decryptor = AES.new(self._file._client.a32_to_str(self.k), AES.MODE_CTR, counter = Counter.new(128, initial_value = initial_value)) rest = offset - int(offset/16)*16 if rest: self.decode(str(0)*rest)