Minor cleanup of states and register sizes.

1 files changed, 10 insertions, 12 deletions

diff --git a/src/rtl/aes_encipher_block.v b/src/rtl/aes_encipher_block.v
index 8cac95f..f98d755 100644
--- a/src/rtl/aes_encipher_block.v
+++ b/src/rtl/aes_encipher_block.v
@@ -64,16 +64,14 @@ module aes_encipher_block(
   localparam AES128_ROUNDS = 4'ha;
   localparam AES256_ROUNDS = 4'he;
 
-  localparam NO_UPDATE    = 3'h0;
-  localparam INIT_UPDATE  = 3'h1;
-  localparam MAIN_UPDATE  = 3'h2;
-  localparam FINAL_UPDATE = 3'h3;
+  localparam NO_UPDATE    = 2'h0;
+  localparam INIT_UPDATE  = 2'h1;
+  localparam MAIN_UPDATE  = 2'h2;
+  localparam FINAL_UPDATE = 2'h3;
 
-  localparam CTRL_IDLE  = 3'h0;
-  localparam CTRL_INIT  = 3'h1;
-  localparam CTRL_SBOX  = 3'h2;
-  localparam CTRL_MAIN  = 3'h3;
-  localparam CTRL_FINAL = 3'h4;
+  localparam CTRL_IDLE  = 2'h0;
+  localparam CTRL_INIT  = 2'h1;
+  localparam CTRL_MAIN  = 2'h2;
 
 
   //----------------------------------------------------------------
@@ -169,15 +167,15 @@ module aes_encipher_block(
   reg           ready_new;
   reg           ready_we;
 
-  reg [2 : 0]   enc_ctrl_reg;
-  reg [2 : 0]   enc_ctrl_new;
+  reg [1 : 0]   enc_ctrl_reg;
+  reg [1 : 0]   enc_ctrl_new;
   reg           enc_ctrl_we;
 
 
   //----------------------------------------------------------------
   // Wires.
   //----------------------------------------------------------------
-  reg [2 : 0]  update_type;
+  reg [1 : 0]  update_type;
 
   reg [31 : 0] sboxw0;
   reg [31 : 0] sboxw1;