Adding API control bits to read and write key.

1 files changed, 7 insertions, 1 deletions

diff --git a/src/tb/tb_keywrap_core.v b/src/tb/tb_keywrap_core.v
index 358b512..4b94b26 100644
--- a/src/tb/tb_keywrap_core.v
+++ b/src/tb/tb_keywrap_core.v
@@ -64,6 +64,8 @@ module tb_keywrap_core();
   reg                           tb_reset_n;
   reg                           tb_init;
   reg                           tb_next;
+  reg                           tb_read;
+  reg                           tb_write;
   reg                           tb_encdec;
   wire                          tb_ready;
   wire                          tb_valid;
@@ -98,7 +100,9 @@ module tb_keywrap_core();
 
                    .init(tb_init),
                    .next(tb_next),
-                   .encdec(tb_encdec),
+                   .read(tb_read),
+                   .write(tb_write),
+                  .encdec(tb_encdec),
 
                    .ready(tb_ready),
                    .valid(tb_valid),
@@ -159,6 +163,8 @@ module tb_keywrap_core();
 
       tb_init        = 0;
       tb_next        = 0;
+      tb_read        = 0;
+      tb_write       = 0;
       tb_encdec      = 0;
       tb_rlen        = 13'h0;
       tb_key         = 256'h0;