Skip to content

Commit 805915b

Browse files
author
Hans Hörberg
committed
Correction so all aes encryption alg. can be used while decrypting.
1 parent d38e947 commit 805915b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/saml2/aes.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def encrypt(self, msg, iv=None, alg="aes_128_cbc", padding="PKCS#7",
8787
return cmsg
8888

8989

90-
def decrypt(self, msg, iv=None, padding="PKCS#7", b64dec=True):
90+
def decrypt(self, msg, iv=None, alg="aes_128_cbc", padding="PKCS#7", b64dec=True):
9191
"""
9292
:param key: The encryption key
9393
:param iv: init vector
@@ -102,7 +102,7 @@ def decrypt(self, msg, iv=None, padding="PKCS#7", b64dec=True):
102102
_iv = data[:AES.block_size]
103103
if iv:
104104
assert iv == _iv
105-
cipher, iv = self.build_cipher(iv)
105+
cipher, iv = self.build_cipher(iv, alg=alg)
106106
res = cipher.decrypt(data)[AES.block_size:]
107107
if padding in ["PKCS#5", "PKCS#7"]:
108108
res = res[:-ord(res[-1])]

0 commit comments

Comments
 (0)