🗓 Apr 6, 2024 6:26:41 PM

🔧 fixes #34
securisec · Apr 6, 2024 · 625be78 · 625be78
1 parent 1d1ea70
commit 625be78
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/chepy/modules/encryptionencoding.py b/chepy/modules/encryptionencoding.py
@@ -84,9 +84,9 @@ def _convert_key(
         if iv_format == "base64" or iv_format == "b64":
             iv = base64.b64decode(iv)
         if iv_format == "utf-8" or iv_format == "utf8":
-            iv = key.decode().encode("utf-8")
+            iv = iv.decode().encode("utf-8")
         if iv_format == "latin-1":
-            iv = key.decode().encode("latin-1")
+            iv = iv.decode().encode("latin-1")
         if iv_format == "raw":
             iv = iv
         else:

diff --git a/tests/test_encryptionencoding.py b/tests/test_encryptionencoding.py
@@ -412,6 +412,10 @@ def test_aes_encrypt():
         .o
         == b"hello"
     )
+    assert (
+        Chepy('hello')
+        .aes_encrypt(key='supersecret!!!!!', key_format='utf8', iv='abcdefghijklmnop', iv_format='utf-8')
+        .to_hex().o == b'd18c2cba21d2eef40c9ed4771a9b320e')
 
 
 def test_aes_decrypt():