fix: include all hashlib.new inputs

equinor · Sep 25, 2024 · 15f02ea · 15f02ea
1 parent eb644c6
commit 15f02ea
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tagreader/web_handlers.py b/tagreader/web_handlers.py
@@ -38,11 +38,15 @@ def hexdigest(self):
         return self._hash_obj.hexdigest()
 
 
-def patched_hashlib_new(name, data=b""):
+def patched_hashlib_new(name, data=b"", usedforsecurity=True):
     if name.lower() == "md4":
         return MD4(data)
     else:
-        return hashlib_new_method(name, data)
+        # Try / Catch easier than detecting python version
+        try:
+            return hashlib_new_method(name, data=data, usedforsecurity=usedforsecurity)
+        except:
+            return hashlib_new_method(name, data=data)
 
 
 # Monkey-patch md4 in hashlib.new due to missing support for md4 in later releases of Python: