(1) Added NIST KWP test to core. (2) Moved all other tests in core tb to a separate task. (3) Minor cleanup in top tb.

1 files changed, 13 insertions, 14 deletions

diff --git a/src/tb/tb_aes.v b/src/tb/tb_aes.v
index 1e229d9..6561b4d 100644
--- a/src/tb/tb_aes.v
+++ b/src/tb/tb_aes.v
@@ -429,11 +429,11 @@ module tb_aes();
 
 
   //----------------------------------------------------------------
-  // aes_test()
+  // nist_fips_tests()
   //
-  // Main test task will perform complete NIST test of AES.
+  // Perform tests based on NIST FIPS-197 test vectors.
   //----------------------------------------------------------------
-  task aes_test;
+  task nist_fips_tests;
     reg [255 : 0] nist_aes128_key;
     reg [255 : 0] nist_aes256_key;
 
@@ -472,8 +472,8 @@ module tb_aes();
       nist_ecb_256_enc_expected3 = 128'h23304b7a39f9f3ff067d8d8f9e24ecc7;
 
 
-      $display("ECB 128 bit key tests");
-      $display("---------------------");
+      $display("NIST FIPS ECB 128 bit key tests");
+      $display("-------------------------------");
       ecb_mode_single_block_test(8'h01, AES_ENCIPHER, nist_aes128_key, AES_128_BIT_KEY,
                                  nist_plaintext0, nist_ecb_128_enc_expected0);
 
@@ -501,8 +501,8 @@ module tb_aes();
 
 
       $display("");
-      $display("ECB 256 bit key tests");
-      $display("---------------------");
+      $display("NIST FIPS ECB 256 bit key tests");
+      $display("-------------------------------");
       ecb_mode_single_block_test(8'h10, AES_ENCIPHER, nist_aes256_key, AES_256_BIT_KEY,
                                  nist_plaintext0, nist_ecb_256_enc_expected0);
 
@@ -531,14 +531,13 @@ module tb_aes();
   endtask // aes_test
 
 
-
   //----------------------------------------------------------------
-  // nist_kwp_ae_128_test()
+  // nist_kwp_test()
   //
   // Test that we can perform operations based on NIST KWP
   // AE 128 test vectors.
   //----------------------------------------------------------------
-  task nist_kwp_ae_128_test;
+  task nist_kwp_test;
     reg [255 : 0] kwp_key;
     reg [127 : 0] kwp_plaintext;
     reg [127 : 0] kwp_expected;
@@ -551,10 +550,10 @@ module tb_aes();
       $display("");
       $display("NIST KWP AE 128 bit test");
 
-      ecb_mode_single_block_test(8'h01, AES_ENCIPHER, kwp_key, AES_128_BIT_KEY,
+      ecb_mode_single_block_test(8'h18, AES_ENCIPHER, kwp_key, AES_128_BIT_KEY,
                                  kwp_plaintext, kwp_expected);
     end
-  endtask // nist_kwp_ae_128_test
+  endtask // nist_kwp_test
 
 
 
@@ -574,8 +573,8 @@ module tb_aes();
       reset_dut();
       dump_dut_state();
 
-      aes_test();
-      nist_kwp_ae_128_test();
+      nist_fips_tests();
+      nist_kwp_test();
 
       display_test_results();