Skip to content

Commit b5ab5b0

Browse files
committed
Validate actual key length in ChaCha
1 parent a885a8d commit b5ab5b0

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

wolfcrypt/ciphers.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -498,10 +498,11 @@ def __init__(self, key="", size=32):
498498
self._dec = None
499499
self._key = None
500500
if len(key) > 0:
501-
if not size in self._key_sizes:
502-
raise ValueError("Invalid key size %d" % size)
503501
self._key = t2b(key)
504-
self.key_size = size
502+
if len(self._key) not in self._key_sizes:
503+
raise ValueError("key must be %s in length, not %d" %
504+
(self._key_sizes, len(self._key)))
505+
self.key_size = len(self._key)
505506
self._IV_nonce = []
506507
self._IV_counter = 0
507508

0 commit comments

Comments
 (0)