Exclude AES CTR from block size check

wolfSSL · Dec 27, 2023 · e65ce42 · e65ce42
1 parent 1c24265
commit e65ce42
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wolfcrypt/ciphers.py b/wolfcrypt/ciphers.py
@@ -181,7 +181,7 @@ def encrypt(self, string):
             raise ValueError(
                     "empty string not allowed")
 
-        if len(string) % self.block_size and not "ChaCha" in self._native_type:
+        if len(string) % self.block_size and not self.mode == MODE_CTR and not "ChaCha" in self._native_type:
             raise ValueError(
                 "string must be a multiple of %d in length" % self.block_size)
 
@@ -214,7 +214,7 @@ def decrypt(self, string):
             raise ValueError(
                     "empty string not allowed")
 
-        if len(string) % self.block_size and not "ChaCha" in self._native_type:
+        if len(string) % self.block_size and not self.mode == MODE_CTR and not "ChaCha" in self._native_type:
             raise ValueError(
                 "string must be a multiple of %d in length" % self.block_size)