-
Notifications
You must be signed in to change notification settings - Fork 0
/
aes_engine_tb.v
168 lines (145 loc) · 2.83 KB
/
aes_engine_tb.v
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
`timescale 1ns/1ns // define simulation timescale
// define testbench module for aes_engine
module aes_engine_tb();
// define tb registers and wires
reg clk_tb = 0;
reg rst_tb = 0;
reg[7:0] din_tb = 0;
reg[1:0] cmd_tb = 0;
wire ready_tb, dok_tb;
wire[7:0] dout_tb;
// define tb for dout
reg[127:0] cipher_out;
initial cipher_out = 128'h00000000000000000000000000000000;
// define tb->aes_engine connections
aes_engine i_uut
(
.clk(clk_tb),
.rst_(rst_tb),
.din(din_tb),
.cmd(cmd_tb),
.interface_ready(ready_tb),
.dout(dout_tb),
.data_ok(dok_tb)
);
// set periodic clock pulse every 1ns
always clk_tb = #1 ~clk_tb;
// state defines
localparam C_ID = 2'b00; // Idle state
localparam C_SP = 2'b01; // Set plaintext state
localparam C_SK = 2'b10; // Set key state
localparam C_ST = 2'b11; // Start encryption
// set callback for reading data
// read data
always @(negedge clk_tb) begin
if (dok_tb) begin
// append one byte to cipher_out reading register
cipher_out = { cipher_out[119:0], dout_tb };
end
end
// set testing sequence signals
initial begin
// reset everything
rst_tb = 0;
#2;
rst_tb = 1;
#2;
// enter plaintext
cmd_tb = C_SP;
din_tb = 8'h00;
#2;
din_tb = 8'h04;
#2;
din_tb = 8'h12;
#2;
din_tb = 8'h14;
#2;
din_tb = 8'h12;
#2;
din_tb = 8'h04;
#2;
din_tb = 8'h12;
#2;
din_tb = 8'h00;
#2;
din_tb = 8'h0C;
#2;
din_tb = 8'h00;
#2;
din_tb = 8'h13;
#2;
din_tb = 8'h11;
#2;
din_tb = 8'h08;
#2;
din_tb = 8'h23;
#2;
din_tb = 8'h19;
#2;
din_tb = 8'h19;
#2;
// // Idle
// cmd_tb = C_ID;
// #2;
// enter key
cmd_tb = C_SK;
din_tb = 8'h24;
#2;
din_tb = 8'h75;
#2;
din_tb = 8'hA2;
#2;
din_tb = 8'hB3;
#2;
din_tb = 8'h34;
#2;
din_tb = 8'h75;
#2;
din_tb = 8'h56;
#2;
din_tb = 8'h88;
#2;
din_tb = 8'h31;
#2;
din_tb = 8'hE2;
#2;
din_tb = 8'h12;
#2;
din_tb = 8'h00;
#2;
din_tb = 8'h13;
#2;
din_tb = 8'hAA;
#2;
din_tb = 8'h54;
#2;
din_tb = 8'h87;
#2;
// // Idle
// cmd_tb = C_ID;
// #2;
// Start command
cmd_tb = C_ST;
// wait for data to be ok to go high
@(posedge dok_tb)
begin
// Idle input interface
cmd_tb = C_ID;
// ? OR start inputting next plaintext
// Read output ciphertext
// in the routine above
end
// Signal read finished
// wait for data_ok to go low again
@(negedge dok_tb) begin
$display("----------------");
$display("Recieved ciphertext from output_interface: ");
$write("%02X %02X %02X %02X\n", cipher_out[127:120], cipher_out[95:88], cipher_out[63:56], cipher_out[31:24]);
$write("%02X %02X %02X %02X\n", cipher_out[119:112], cipher_out[87:80], cipher_out[55:48], cipher_out[23:16]);
$write("%02X %02X %02X %02X\n", cipher_out[111:104], cipher_out[79:72], cipher_out[47:40], cipher_out[15:8]);
$write("%02X %02X %02X %02X\n", cipher_out[103:96], cipher_out[71:64], cipher_out[39:32], cipher_out[7:0]);
#10;
$finish;
end
end
endmodule