Skip to content

Commit c2d86fd

Browse files
author
Inbal Tako
committed
Fix encryption utils
1 parent 9973936 commit c2d86fd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

securenative/utils/encryption_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ def encrypt(cls, text, cipher_key):
1717
try:
1818
key = cipher_key[:cls.KEY_SIZE]
1919
iv = Random.new().read(AES.block_size)
20-
cipher = AES.new(key, AES.MODE_CBC, iv)
20+
cipher = AES.new(key.encode("utf8"), AES.MODE_CBC, iv)
2121
raw = str(cls._pad(text))
22-
return hexlify(iv + cipher.encrypt(raw))
22+
return hexlify(iv + cipher.encrypt(raw.encode("utf8")))
2323
except Exception as e:
2424
Logger.error("Could not encrypt text {}; {}".format(text, e))
2525
return None
@@ -31,7 +31,7 @@ def decrypt(cls, encrypted, cipher_key):
3131
content = unhexlify(encrypted)
3232
iv = content[:cls.BLOCK_SIZE]
3333
cipher_text = content[cls.BLOCK_SIZE:]
34-
aes = AES.new(key, AES.MODE_CBC, iv)
34+
aes = AES.new(key.encode("utf8"), AES.MODE_CBC, iv)
3535
rv = aes.decrypt(cipher_text).decode("utf-8").strip()
3636
secret = json.loads(rv)
3737
return ClientToken(secret.get("cid"), secret.get("vid"), secret.get("fp"))

0 commit comments

Comments
 (0)