aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_hashes.py104
1 files changed, 104 insertions, 0 deletions
diff --git a/tests/test_hashes.py b/tests/test_hashes.py
new file mode 100644
index 0000000..2e38ca3
--- /dev/null
+++ b/tests/test_hashes.py
@@ -0,0 +1,104 @@
+# Trivial test of cryptech hash cores via cryptlib python interface.
+# Might upgrade to Python's unittest framework eventually.
+
+import atexit, os.path
+from cryptlib_py import *
+
+cryptInit()
+atexit.register(cryptEnd)
+
+hwdev = cryptDeviceOpen(CRYPT_UNUSED, CRYPT_DEVICE_HARDWARE, None)
+atexit.register(cryptDeviceClose, hwdev)
+
+# Usual NIST sample messages.
+
+def hextext(s):
+ return "".join(s.split()).lower()
+
+NIST_512_SINGLE = "abc"
+SHA1_SINGLE_DIGEST = hextext("A9993E36 4706816A BA3E2571 7850C26C 9CD0D89D")
+SHA256_SINGLE_DIGEST = hextext("BA7816BF 8F01CFEA 414140DE 5DAE2223 B00361A3 96177A9C B410FF61 F20015AD")
+
+NIST_512_DOUBLE = "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq"
+SHA1_DOUBLE_DIGEST = hextext("84983E44 1C3BD26E BAAE4AA1 F95129E5 E54670F1")
+SHA256_DOUBLE_DIGEST = hextext("248D6A61 D20638B8 E5C02693 0C3E6039 A33CE459 64FF2167 F6ECEDD4 19DB06C1")
+
+NIST_1024_SINGLE = "abc"
+SHA384_SINGLE_DIGEST = hextext("CB00753F 45A35E8B B5A03D69 9AC65007 272C32AB 0EDED163"
+ "1A8B605A 43FF5BED 8086072B A1E7CC23 58BAECA1 34C825A7")
+SHA512_SINGLE_DIGEST = hextext("DDAF35A1 93617ABA CC417349 AE204131 12E6FA4E 89A97EA2 0A9EEEE6 4B55D39A"
+ "2192992A 274FC1A8 36BA3C23 A3FEEBBD 454D4423 643CE80E 2A9AC94F A54CA49F")
+
+NIST_1024_DOUBLE = "abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmn" \
+ "hijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu"
+SHA384_DOUBLE_DIGEST = hextext("09330C33 F71147E8 3D192FC7 82CD1B47 53111B17 3B3B05D2"
+ "2FA08086 E3B0F712 FCC7C71A 557E2DB9 66C3E9FA 91746039")
+SHA512_DOUBLE_DIGEST = hextext("8E959B75 DAE313DA 8CF4F728 14FC143F 8F7779C6 EB9F7FA1 7299AEAD B6889018"
+ "501D289E 4900F7E4 331B99DE C4B5433A C7D329EE B6DD2654 5E96E55B 874BE909")
+
+def do_hash(ctx, s):
+ try:
+ cryptEncrypt(ctx, array("c", s))
+ cryptEncrypt(ctx, array("c", ""))
+ result = ctx.CRYPT_CTXINFO_HASHVALUE
+ return result.encode("hex")
+ finally:
+ cryptDestroyContext(ctx)
+
+def sha1(d, s):
+ if d is None:
+ ctx = cryptCreateContext(CRYPT_UNUSED, CRYPT_ALGO_SHA1)
+ else:
+ ctx = cryptDeviceCreateContext(d, CRYPT_ALGO_SHA1)
+ return do_hash(ctx, s)
+
+def sha256(d, s):
+ if d is None:
+ ctx = cryptCreateContext(CRYPT_UNUSED, CRYPT_ALGO_SHA2)
+ else:
+ ctx = cryptDeviceCreateContext(d, CRYPT_ALGO_SHA2)
+ return do_hash(ctx, s)
+
+def sha384(d, s):
+ if d is None:
+ ctx = cryptCreateContext(CRYPT_UNUSED, CRYPT_ALGO_SHA2)
+ else:
+ ctx = cryptDeviceCreateContext(d, CRYPT_ALGO_SHA2)
+ ctx.CTXINFO_BLOCKSIZE = 48
+ return do_hash(ctx, s)
+
+def sha512(d, s):
+ if d is None:
+ ctx = cryptCreateContext(CRYPT_UNUSED, CRYPT_ALGO_SHA2)
+ else:
+ ctx = cryptDeviceCreateContext(d, CRYPT_ALGO_SHA2)
+ ctx.CTXINFO_BLOCKSIZE = 64
+ return do_hash(ctx, s)
+
+have_i2c = os.path.exists("/dev/i2c-2")
+
+if not have_i2c:
+ print
+ print "I2C device not found, so testing software only, no hardware cores tested"
+
+def test(digest, text, expect):
+ print
+ print "Testing %s(%r)" % (digest.__name__, text)
+ hashes = [digest(None, text)]
+ if have_i2c:
+ hashes.append(digest(hwdev, text))
+ for hash in hashes:
+ if hash == expect:
+ print "+", hash
+ else:
+ print "-", hash
+ print "!", expect
+
+test(sha1, NIST_512_SINGLE, SHA1_SINGLE_DIGEST)
+test(sha1, NIST_512_DOUBLE, SHA1_DOUBLE_DIGEST)
+test(sha256, NIST_512_SINGLE, SHA256_SINGLE_DIGEST)
+test(sha256, NIST_512_DOUBLE, SHA256_DOUBLE_DIGEST)
+test(sha384, NIST_1024_SINGLE, SHA384_SINGLE_DIGEST)
+test(sha384, NIST_1024_DOUBLE, SHA384_DOUBLE_DIGEST)
+test(sha512, NIST_1024_SINGLE, SHA512_SINGLE_DIGEST)
+test(sha512, NIST_1024_DOUBLE, SHA512_DOUBLE_DIGEST)