SHA224/384/512: Py3k compatibility

3 files changed, 12 insertions, 9 deletions

diff --git a/lib/Crypto/Hash/SHA224.py b/lib/Crypto/Hash/SHA224.py
index e2a6876..7499a85 100644
--- a/lib/Crypto/Hash/SHA224.py
+++ b/lib/Crypto/Hash/SHA224.py
@@ -24,8 +24,9 @@ __revision__ = "$Id$"
 
 __all__ = ['new', 'digest_size']
 
-import hashlib
-def new(data=""):
-    return hashlib.sha224(data)
+from hashlib import sha224      # This will only work in versions of Python that support SHA224
+from Crypto.Util.py3compat import *
+def new(data=b("")):
+    return sha224(data)
 digest_size = new().digest_size
 block_size = 64
diff --git a/lib/Crypto/Hash/SHA384.py b/lib/Crypto/Hash/SHA384.py
index 8d0fa30..248955d 100644
--- a/lib/Crypto/Hash/SHA384.py
+++ b/lib/Crypto/Hash/SHA384.py
@@ -24,8 +24,9 @@ __revision__ = "$Id$"
 
 __all__ = ['new', 'digest_size']
 
-import hashlib
-def new(data=""):
-    return hashlib.sha384(data)
+from hashlib import sha384      # This will only work in versions of Python that support SHA384
+from Crypto.Util.py3compat import *
+def new(data=b("")):
+    return sha384(data)
 digest_size = new().digest_size
 block_size = 128
diff --git a/lib/Crypto/Hash/SHA512.py b/lib/Crypto/Hash/SHA512.py
index f6c4787..38bd957 100644
--- a/lib/Crypto/Hash/SHA512.py
+++ b/lib/Crypto/Hash/SHA512.py
@@ -24,8 +24,9 @@ __revision__ = "$Id$"
 
 __all__ = ['new', 'digest_size']
 
-import hashlib
-def new(data=""):
-    return hashlib.sha512(data)
+from hashlib import sha512      # This will only work in versions of Python that support SHA512
+from Crypto.Util.py3compat import *
+def new(data=b("")):
+    return sha512(data)
 digest_size = new().digest_size
 block_size = 128