aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorJoachim Strömbergson <joachim@secworks.se>2014-04-01 09:51:38 +0200
committerJoachim Strömbergson <joachim@secworks.se>2014-04-01 09:51:38 +0200
commitf69f454a5e3f56a96f6d28bb48c661630e31e300 (patch)
tree0edce1720d8e5297ef3edccf1dc2485fde1ee2dd /src
parent772560966804c97c7d27c2ac49b9a5e5e0b5453d (diff)
Updating sha256 python model with NIST dual block test case and test case with huge message. Disabling verbose mode.
Diffstat (limited to 'src')
-rwxr-xr-xsrc/model/sha256.py58
1 files changed, 51 insertions, 7 deletions
diff --git a/src/model/sha256.py b/src/model/sha256.py
index 2a56f41..ca0a77c 100755
--- a/src/model/sha256.py
+++ b/src/model/sha256.py
@@ -10,7 +10,7 @@
#
#
# Author: Joachim Strömbergson
-# Copyright (c) 2014 SUNET
+# Copyright (c) 2014, SUNET
#
# Redistribution and use in source and binary forms, with or
# without modification, are permitted provided that the following
@@ -46,11 +46,6 @@ import sys
#-------------------------------------------------------------------
-# Constants.
-#-------------------------------------------------------------------
-
-
-#-------------------------------------------------------------------
# ChaCha()
#-------------------------------------------------------------------
class SHA256():
@@ -240,9 +235,10 @@ def main():
print("---------------------------------")
print
- my_sha256 = SHA256(verbose=1);
+ my_sha256 = SHA256(verbose=0);
# TC1: NIST testcase with message "abc"
+ print("TC1: Single block message test specified by NIST.")
TC1_block = [0x61626380, 0x00000000, 0x00000000, 0x00000000,
0x00000000, 0x00000000, 0x00000000, 0x00000000,
0x00000000, 0x00000000, 0x00000000, 0x00000000,
@@ -255,7 +251,55 @@ def main():
my_sha256.next(TC1_block)
my_digest = my_sha256.get_digest()
compare_digests(my_digest, TC1_expected)
+ print("")
+
+
+ # TC2: NIST testcase with double block message."
+ print("TC2: Double block message test specified by NIST.")
+ TC2_1_block = [0x61626364, 0x62636465, 0x63646566, 0x64656667,
+ 0x65666768, 0x66676869, 0x6768696A, 0x68696A6B,
+ 0x696A6B6C, 0x6A6B6C6D, 0x6B6C6D6E, 0x6C6D6E6F,
+ 0x6D6E6F70, 0x6E6F7071, 0x80000000, 0x00000000]
+
+
+ TC2_2_block = [0x00000000, 0x00000000, 0x00000000, 0x00000000,
+ 0x00000000, 0x00000000, 0x00000000, 0x00000000,
+ 0x00000000, 0x00000000, 0x00000000, 0x00000000,
+ 0x00000000, 0x00000000, 0x00000000, 0x000001C0]
+
+ TC2_1_expected = [0x85E655D6, 0x417A1795, 0x3363376A, 0x624CDE5C,
+ 0x76E09589, 0xCAC5F811, 0xCC4B32C1, 0xF20E533A]
+
+ TC2_2_expected = [0x248D6A61, 0xD20638B8, 0xE5C02693, 0x0C3E6039,
+ 0xA33CE459, 0x64FF2167, 0xF6ECEDD4, 0x19DB06C1]
+ my_sha256.init()
+ my_sha256.next(TC2_1_block)
+ my_digest = my_sha256.get_digest()
+ compare_digests(my_digest, TC2_1_expected)
+
+ my_sha256.next(TC2_2_block)
+ my_digest = my_sha256.get_digest()
+ compare_digests(my_digest, TC2_2_expected)
+ print("")
+
+
+ # TC3: Huge message with n blocks
+ n = 10000
+ print("TC3: Huge message with %d blocks test case." % n)
+ TC3_block = [0xaa55aa55, 0xdeadbeef, 0x55aa55aa, 0xf00ff00f,
+ 0xaa55aa55, 0xdeadbeef, 0x55aa55aa, 0xf00ff00f,
+ 0xaa55aa55, 0xdeadbeef, 0x55aa55aa, 0xf00ff00f,
+ 0xaa55aa55, 0xdeadbeef, 0x55aa55aa, 0xf00ff00f]
+
+ TC3_expected = [0xf407ff0d, 0xb9dce2f6, 0x9b9759a9, 0xd3cdc805,
+ 0xf250086d, 0x73bbefd5, 0xa972e0f7, 0x61a9c13e]
+
+ my_sha256.init()
+ for i in range(n):
+ my_sha256.next(TC3_block)
+ my_digest = my_sha256.get_digest()
+ compare_digests(my_digest, TC3_expected)
#-------------------------------------------------------------------