diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_hashes.py | 104 |
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) |