From bb02aa9fe45cc97a3d1ce8b2f3611f42a0944b17 Mon Sep 17 00:00:00 2001 From: Lff5 Date: Sat, 7 Dec 2024 00:01:38 +0200 Subject: [PATCH 1/2] Fix init_key() Fix for longer keys and other hashes than md5 --- hmac.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/hmac.py b/hmac.py index 665c470..72f7d98 100644 --- a/hmac.py +++ b/hmac.py @@ -49,12 +49,11 @@ def init_key(self): """ key regeneration """ if len(self.key) > self.blocksize: - self.key = bytearray(md5(key).digest()) - elif len(self.key) < self.blocksize: - i = len(self.key) - while i < self.blocksize: - self.key += b"\x00" - i += 1 + self.key = bytearray(self.hash_h(key).digest()) + i = len(self.key) + while i < self.blocksize: + self.key += b"\x00" + i += 1 def digest(self): From 253dcffc5fae34f81fe4e4202fdd5353f91dcf07 Mon Sep 17 00:00:00 2001 From: Lff5 Date: Sat, 7 Dec 2024 00:44:46 +0200 Subject: [PATCH 2/2] Update hmac.py --- hmac.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hmac.py b/hmac.py index 72f7d98..0af90b7 100644 --- a/hmac.py +++ b/hmac.py @@ -49,7 +49,7 @@ def init_key(self): """ key regeneration """ if len(self.key) > self.blocksize: - self.key = bytearray(self.hash_h(key).digest()) + self.key = bytearray(self.hash_h(self.key).digest()) i = len(self.key) while i < self.blocksize: self.key += b"\x00"