Added a simple test case to verify that the aes model is usable.

1 files changed, 26 insertions, 0 deletions

diff --git a/src/model/keywrap.py b/src/model/keywrap.py
index e0bd5a6..08aac9a 100755
--- a/src/model/keywrap.py
+++ b/src/model/keywrap.py
@@ -43,6 +43,7 @@
 # Python module imports.
 #-------------------------------------------------------------------
 import sys
+import aes
 
 
 #-------------------------------------------------------------------
@@ -52,11 +53,36 @@ VERBOSE = True
 
 
 #-------------------------------------------------------------------
+# aes_test
+#
+# Check that the AES functionality is available and works
+# as expected.
+#-------------------------------------------------------------------
+def keywrap_aes_test():
+    nist_aes128_key = (0x2b7e1516, 0x28aed2a6, 0xabf71588, 0x09cf4f3c)
+    nist_plaintext0 = (0x6bc1bee2, 0x2e409f96, 0xe93d7e11, 0x7393172a)
+    nist_exp128_0 = (0x3ad77bb4, 0x0d7a3660, 0xa89ecaf3, 0x2466ef97)
+    enc_result128_0 = aes.aes_encipher_block(nist_aes128_key, nist_plaintext0)
+
+    print("Test 0 for AES-128.")
+    print("Key:")
+    aes.print_key(nist_aes128_key)
+    print("Block in:")
+    aes.print_block(nist_plaintext0)
+    print("Expected block out:")
+    aes.print_block(nist_exp128_0)
+    print("Got block out:")
+    aes.print_block(enc_result128_0)
+    print("")
+
+
+#-------------------------------------------------------------------
 #-------------------------------------------------------------------
 def main():
     print("Testing the KEY WRAP model")
     print("===========================")
     print
+    keywrap_aes_test()
 
 
 #-------------------------------------------------------------------