blacked
This commit is contained in:
parent
1c543e3a84
commit
27ce17cd74
1 changed files with 3 additions and 3 deletions
|
|
@ -73,10 +73,10 @@ class AESCipher(object):
|
||||||
final_key += key
|
final_key += key
|
||||||
return final_key[:output]
|
return final_key[:output]
|
||||||
|
|
||||||
def decrypt(self, encrypted: str) -> str: #type: ignore
|
def decrypt(self, encrypted: str) -> str: # type: ignore
|
||||||
"""Decrypts a string using AES-256-CBC."""
|
"""Decrypts a string using AES-256-CBC."""
|
||||||
passphrase = self.passphrase
|
passphrase = self.passphrase
|
||||||
encrypted = base64.b64decode(encrypted) #type: ignore
|
encrypted = base64.b64decode(encrypted) # type: ignore
|
||||||
assert encrypted[0:8] == b"Salted__"
|
assert encrypted[0:8] == b"Salted__"
|
||||||
salt = encrypted[8:16]
|
salt = encrypted[8:16]
|
||||||
key_iv = self.bytes_to_key(passphrase.encode(), salt, 32 + 16)
|
key_iv = self.bytes_to_key(passphrase.encode(), salt, 32 + 16)
|
||||||
|
|
@ -84,7 +84,7 @@ class AESCipher(object):
|
||||||
iv = key_iv[32:]
|
iv = key_iv[32:]
|
||||||
aes = AES.new(key, AES.MODE_CBC, iv)
|
aes = AES.new(key, AES.MODE_CBC, iv)
|
||||||
try:
|
try:
|
||||||
return self.unpad(aes.decrypt(encrypted[16:])).decode() #type: ignore
|
return self.unpad(aes.decrypt(encrypted[16:])).decode() # type: ignore
|
||||||
except UnicodeDecodeError:
|
except UnicodeDecodeError:
|
||||||
raise ValueError("Wrong passphrase")
|
raise ValueError("Wrong passphrase")
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue