fix CBC IV

selfcustody · Nov 19, 2024 · afd5ec6 · afd5ec6
1 parent 9cd6e97
commit afd5ec6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/krux/encryption.py b/src/krux/encryption.py
@@ -64,12 +64,13 @@ def encrypt(self, raw, mode=ucryptolib.MODE_ECB, i_vector=None):
         data_bytes = raw.encode("latin-1") if isinstance(raw, str) else raw
         if i_vector:
             encryptor = ucryptolib.aes(self.key, mode, i_vector)
-            data_bytes = i_vector + data_bytes
         else:
             encryptor = ucryptolib.aes(self.key, mode)
         encrypted = encryptor.encrypt(
             data_bytes + b"\x00" * ((16 - (len(data_bytes) % 16)) % 16)
         )
+        if i_vector:
+            encrypted = i_vector + encrypted
         return base_encode(encrypted, 64)
 
     def decrypt(self, encrypted, mode, i_vector=None):