diff --git a/hmac.py b/hmac.py index 665c470..0af90b7 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(self.key).digest()) + i = len(self.key) + while i < self.blocksize: + self.key += b"\x00" + i += 1 def digest(self):