sample_ntt was non-conformant: both RTL and the test reference re-ran
keccak_p after every 3-byte squeeze instead of consuming the full
1344-bit SHAKE-128 rate. Only coeff[0] matched a standard sampler, so
the generated A matrix would not interoperate with any compliant ML-KEM.
- sample_ntt_sync{,_shared}.v: walk all 56 groups of the rate block via
grp_ptr_r; re-permute only when the block is exhausted. Verified
256/256 against ml-kem-r Rust sample_ntt on two seeds, and 1536/1536
in the Verilator framework (runtime ~128x faster per poly).
- gen_vectors.py: use a self-contained hashlib.shake_128 oracle.
sha3 testbench fixes (all now self-check hash_o against verified vectors,
cross-checked with hashlib and ml-kem-r mlkem_G):
- tb_sha3_xsim_simple.v: test G/H/J modes, not just G.
- tb_keccak_core_xsim.v: correct the wrong EXPECTED_STATE constant
(RTL was correct; lane0 = 0xf1258f7940e1dde7 per FIPS 202).
- tb_sha3_xsim.v: read expected file and self-check per vector; add
vectors/g_basic_{input,expected}.hex (3 G / 2 H / 2 J).
Remove stale sha3_chain test (its RTL was deleted in 1cace51) and its
README references. Extend .gitignore for XSIM artifacts and result dumps.
262 lines
9.7 KiB
Verilog
262 lines
9.7 KiB
Verilog
// tb_sha3_xsim.v - Standard Verilog testbench for sha3_top targeting Vivado xsim
|
|
//
|
|
// Reads test vectors from a hex file using $readmemh.
|
|
// Each line is a single hex number encoding both mode and data:
|
|
// - Upper 8 bits [519:512]: mode[1:0] in bits [513:512]
|
|
// - Lower 512 bits [511:0]: data_i
|
|
// - Total: 130 hex chars per line, NO spaces
|
|
//
|
|
// Drives sha3_top, waits for valid_o, and writes "RESULT: MODE HASH_HEX"
|
|
// to the output file using $fwrite.
|
|
//
|
|
// Parameters:
|
|
// VECTOR_FILE - path to input hex file (default: "vectors/g_basic_input.hex")
|
|
// RESULT_FILE - path to output file (default: "vectors/g_basic_result.hex")
|
|
//
|
|
// Usage:
|
|
// xvlog -sv sha3_top.v tb_sha3_xsim.v
|
|
// xelab tb_sha3_xsim -s tb_sha3_xsim
|
|
// xsim tb_sha3_xsim -R
|
|
|
|
`timescale 1ns / 1ps
|
|
|
|
module tb_sha3_xsim;
|
|
|
|
// ================================================================
|
|
// Parameters
|
|
// ================================================================
|
|
parameter VECTOR_FILE = "sync_rtl/sha3/TB/vectors/g_basic_input.hex";
|
|
parameter EXPECTED_FILE = "sync_rtl/sha3/TB/vectors/g_basic_expected.hex";
|
|
parameter RESULT_FILE = "sync_rtl/sha3/TB/vectors/g_basic_result.hex";
|
|
parameter MAX_VECTORS = 256;
|
|
parameter TIMEOUT_CYCLES = 1000;
|
|
|
|
// ================================================================
|
|
// DUT signals
|
|
// ================================================================
|
|
reg clk;
|
|
reg rst_n;
|
|
reg [1:0] mode;
|
|
reg [511:0] data_i;
|
|
reg valid_i;
|
|
wire ready_o;
|
|
wire [511:0] hash_o;
|
|
wire valid_o;
|
|
reg ready_i;
|
|
|
|
// ================================================================
|
|
// DUT instantiation
|
|
// ================================================================
|
|
sha3_top u_dut (
|
|
.clk (clk),
|
|
.rst_n (rst_n),
|
|
.mode (mode),
|
|
.data_i (data_i),
|
|
.valid_i (valid_i),
|
|
.ready_o (ready_o),
|
|
.hash_o (hash_o),
|
|
.valid_o (valid_o),
|
|
.ready_i (ready_i)
|
|
);
|
|
|
|
// ================================================================
|
|
// Clock generation: 100 MHz (10 ns period)
|
|
// ================================================================
|
|
initial clk = 1'b0;
|
|
always #5 clk = ~clk;
|
|
|
|
// ================================================================
|
|
// Vector memory (loaded by $readmemh)
|
|
// 520 bits per word: bits[519:512]=padding+mode, bits[511:0]=data_i
|
|
// ================================================================
|
|
reg [519:0] vector_mem [0:MAX_VECTORS-1];
|
|
reg [511:0] expected_mem [0:MAX_VECTORS-1]; // expected hash per vector
|
|
integer vec_count;
|
|
integer idx;
|
|
integer cycle_count;
|
|
integer result_fd;
|
|
|
|
// Test result tracking
|
|
integer pass_count;
|
|
integer fail_count;
|
|
|
|
// ================================================================
|
|
// Hex-to-ASCII conversion helper
|
|
// ================================================================
|
|
function [7:0] nibble_to_ascii;
|
|
input [3:0] nibble;
|
|
begin
|
|
if (nibble < 4'd10)
|
|
nibble_to_ascii = 8'h30 + {4'd0, nibble}; // '0'-'9'
|
|
else
|
|
nibble_to_ascii = 8'h41 + ({4'd0, nibble} - 4'd10); // 'A'-'F'
|
|
end
|
|
endfunction
|
|
|
|
// ================================================================
|
|
// Main test sequence
|
|
// ================================================================
|
|
initial begin
|
|
// Count loaded vectors
|
|
vec_count = 0;
|
|
|
|
// Load vectors from hex file
|
|
$readmemh(VECTOR_FILE, vector_mem);
|
|
// Load expected hashes (one 512-bit hex per line, MSB-first)
|
|
$readmemh(EXPECTED_FILE, expected_mem);
|
|
|
|
// Count non-zero entries to determine actual vector count
|
|
// (XSim leaves unloaded entries as 520'hX)
|
|
begin
|
|
integer found_end;
|
|
found_end = 0;
|
|
for (idx = 0; idx < MAX_VECTORS; idx = idx + 1) begin
|
|
if (!found_end && (vector_mem[idx] === 520'hx || vector_mem[idx] === 520'hz))
|
|
found_end = 1;
|
|
else if (!found_end)
|
|
vec_count = vec_count + 1;
|
|
end
|
|
end
|
|
|
|
if (vec_count == 0) begin
|
|
$display("ERROR: No vectors loaded from %s", VECTOR_FILE);
|
|
$display(" Check that the file exists and is in the correct format.");
|
|
$display(" Each line: <130 hex chars> = {8-bit mode_header, 512-bit data}");
|
|
$finish;
|
|
end
|
|
|
|
$display("INFO: Loaded %0d test vectors from %s", vec_count, VECTOR_FILE);
|
|
|
|
// Open result file
|
|
result_fd = $fopen(RESULT_FILE, "w");
|
|
if (result_fd == 0) begin
|
|
$display("ERROR: Cannot open result file: %s", RESULT_FILE);
|
|
$finish;
|
|
end
|
|
|
|
// Initialize DUT inputs
|
|
mode <= 2'd0;
|
|
data_i <= 512'd0;
|
|
valid_i <= 1'b0;
|
|
ready_i <= 1'b1; // always ready to accept output
|
|
|
|
// Reset sequence: rst_n low for 3 cycles, then high
|
|
rst_n <= 1'b0;
|
|
repeat (3) @(posedge clk);
|
|
rst_n <= 1'b1;
|
|
@(posedge clk);
|
|
|
|
pass_count = 0;
|
|
fail_count = 0;
|
|
|
|
// ============================================================
|
|
// Process each vector
|
|
// ============================================================
|
|
for (idx = 0; idx < vec_count; idx = idx + 1) begin
|
|
// Extract mode and data from memory word
|
|
// mode in bits [513:512], data in bits [511:0]
|
|
begin
|
|
reg [1:0] vec_mode;
|
|
reg [511:0] vec_data;
|
|
reg [511:0] captured_hash;
|
|
|
|
vec_mode = vector_mem[idx][513:512];
|
|
vec_data = vector_mem[idx][511:0];
|
|
|
|
$display("INFO: Vector %0d - mode=%0d", idx, vec_mode);
|
|
|
|
// Drive DUT
|
|
mode <= vec_mode;
|
|
data_i <= vec_data;
|
|
valid_i <= 1'b1;
|
|
@(posedge clk);
|
|
valid_i <= 1'b0;
|
|
|
|
// Wait for ready_o (DUT enters PERMUTE state on this cycle)
|
|
// Then wait for valid_o asserted
|
|
cycle_count = 0;
|
|
while (!valid_o && cycle_count < TIMEOUT_CYCLES) begin
|
|
@(posedge clk);
|
|
cycle_count = cycle_count + 1;
|
|
end
|
|
|
|
if (cycle_count >= TIMEOUT_CYCLES) begin
|
|
$display("ERROR: Timeout waiting for valid_o on vector %0d", idx);
|
|
fail_count = fail_count + 1;
|
|
end else begin
|
|
// Capture hash output and self-check against expected.
|
|
// G (mode 0) uses all 512 bits; H/J use the low 256 bits.
|
|
captured_hash = hash_o;
|
|
|
|
begin
|
|
reg [511:0] exp_hash;
|
|
reg match;
|
|
exp_hash = expected_mem[idx];
|
|
if (vec_mode == 2'd0)
|
|
match = (captured_hash === exp_hash);
|
|
else
|
|
match = (captured_hash[255:0] === exp_hash[255:0]);
|
|
|
|
if (match) begin
|
|
pass_count = pass_count + 1;
|
|
$display("PASS: Vector %0d (mode=%0d)", idx, vec_mode);
|
|
end else begin
|
|
fail_count = fail_count + 1;
|
|
$display("FAIL: Vector %0d (mode=%0d) hash mismatch", idx, vec_mode);
|
|
$display(" got = %0h", (vec_mode==2'd0) ? captured_hash : {256'd0, captured_hash[255:0]});
|
|
$display(" exp = %0h", (vec_mode==2'd0) ? exp_hash : {256'd0, exp_hash[255:0]});
|
|
end
|
|
end
|
|
|
|
// Write result to output file
|
|
// Format: "RESULT: MODE HASH_HEX"
|
|
$fwrite(result_fd, "RESULT: %0d ", vec_mode);
|
|
// Write hash as hex (128 chars for 512 bits)
|
|
begin
|
|
integer bit_idx;
|
|
reg [3:0] nib;
|
|
for (bit_idx = 127; bit_idx >= 0; bit_idx = bit_idx - 1) begin
|
|
nib = captured_hash[(bit_idx*4)+:4];
|
|
$fwrite(result_fd, "%c", nibble_to_ascii(nib));
|
|
end
|
|
end
|
|
$fwrite(result_fd, "\n");
|
|
end
|
|
|
|
// One extra cycle for valid_o handshake
|
|
@(posedge clk);
|
|
end // inner begin block for variable scope
|
|
end
|
|
|
|
// ============================================================
|
|
// Summary
|
|
// ============================================================
|
|
$fclose(result_fd);
|
|
|
|
$display("========================================");
|
|
$display("TEST COMPLETE");
|
|
$display(" Total vectors: %0d", vec_count);
|
|
$display(" Passed: %0d", pass_count);
|
|
$display(" Failed: %0d", fail_count);
|
|
$display(" Results written to: %s", RESULT_FILE);
|
|
$display("========================================");
|
|
if (fail_count == 0)
|
|
$display("ALL TESTS PASSED (%0d/%0d)", pass_count, vec_count);
|
|
else
|
|
$display("TESTS FAILED: %0d of %0d", fail_count, vec_count);
|
|
$display("========================================");
|
|
|
|
$finish;
|
|
end
|
|
|
|
// ================================================================
|
|
// Timeout watchdog
|
|
// ================================================================
|
|
initial begin
|
|
#(TIMEOUT_CYCLES * 10 * 100); // TIMEOUT_CYCLES * 10ns per cycle * extra margin
|
|
$display("FATAL: Global simulation timeout reached");
|
|
$finish;
|
|
end
|
|
|
|
endmodule
|