ruige/eb001
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加了生成包的工具，采用include包含发送包的初始化，尚未做严谨的测试

Browse Source
This commit is contained in:
bacon
2025-05-20 01:12:03 +08:00
parent e86acf4ac4
commit 95d45b4ed4
3 changed files with 127 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
Makefile
View File

@@ -41,6 +41,7 @@ shin:
sbt "test:runMain test.testShinModule"
testShin:
./tb/shinTest/gen_pkt.py
iverilog -Wall \
-o ./tb/build/shinTest.iverilog \
-y ./generated/shin \

12
tb/shinTest/ShinTopSpiTb.v
View File

@@ -438,8 +438,9 @@ initial begin
#1000
$fdisplay(spidata_file, "测试用例1-从IP中读取寄存器，上电后的寄存器期望值为0");
spi_trans( (16'h00), (`SPI_READ), (1024), (`IF_IQSPI(0)) );
spi_trans( (16'h400), (`SPI_READ), (1024), (`IF_IQSPI(0)) );
//spi_trans( (16'h00), (`SPI_READ), (1024), (`IF_IQSPI(0)) );
//spi_trans( (16'h400), (`SPI_READ), (1024), (`IF_IQSPI(0)) );
spi_trans( (16'h00), (`SPI_READ), (64), (`IF_IQSPI(0)) );
spi_trans( (16'h00), (`SPI_READ), (64), (`IF_IQSPI(1)) );
spi_trans( (16'h00), (`SPI_READ), (64), (`IF_IQSPI(2)) );
spi_trans( (16'h00), (`SPI_READ), (64), (`IF_IQSPI(3)) );
@@ -512,10 +513,13 @@ initial begin
spi_trans( (16'h0202), (`SPI_WRITE), (2), (`IF_IDX(0)) );
//无子报文
/*
txBuf[0+3] = 8'h00;
txBuf[1+3] = 8'h01;//header length 0 嗅探包1
txBuf[2+3] = 8'h3f;
txBuf[3+3] = 8'hff;
*/
`include "detect_pkt.vh"
spi_trans( (16'h1000), (`SPI_WRITE), (4), (`IF_IDX(0)) ); //写入SM1
//发布请求
@@ -625,7 +629,7 @@ initial begin
txBuf[1+3] = 8'h02;
spi_trans( (16'h0202), (`SPI_WRITE), (2), (`IF_IDX(0)) );
/*
txBuf[0+3] = 8'h20;
txBuf[1+3] = 8'h00;//heeader length 512
txBuf[2+3] = 8'h00;
@@ -698,6 +702,8 @@ initial begin
txBuf[196+3+i*64+8+j] = $random;
end
end
*/
`include "seq_pkt.vh"
spi_trans( (16'h1000), (`SPI_WRITE), (516), (`IF_IDX(0)) ); //写入SM1

153
tb/shinTest/gen_pkt.py
View File

@@ -1,49 +1,130 @@
#!/usr/bin/env python
#encoding=utf-8
import random
from ctypes import *
class GenPkt(object):
def __init__(self):
super(GenPkt, self).__init__()
self.pkt_buf = bytearray(b'\0' * 2048)
self.pkt_type = 0
self.pkt_sub_pkt = 0
self.sub_pkt = list()
class TPkt(BigEndianStructure):
pass
TPkt._pack_ = 1
TPkt._fields_=[
('len', c_uint, 12),
('rsv1', c_uint, 2),
('type', c_uint, 2),
('rsv2', c_uint16),
('data', c_uint8 * 2040),
]
def clean(self):
self.pkt_buf = bytearray(b'\0' * 2048)
self.pkt_type = 0
self.pkt_sub_pkt = 0
self.sub_pkt = list()
class TSubPkt(BigEndianStructure):
pass
def setPktType(self, type):
if (type != 0) and (type != 1):
print('Error...set type failed:', type)
else:
self.pkt_type = type
TSubPkt._pack_ = 1
TSubPkt._fields_=[
('dev', c_uint, 14),
('mode', c_uint, 2),
('addr', c_uint16),
('cmd', c_uint, 4),
('slen', c_uint, 12),
('wkr', c_uint16)
]
def addSubPkt(self, dev_idx, ptype, addr, cmd, len, data):
if (len > 2032):
print('Error, bad subpkt len:%d', len)
return
if not (ptype in [0, 2]):
print('Error, bad subpkt type:', ptype)
return
if not (cmd in [0, 1, 2, 0xb, 0xc, 0xd, 0xe, 0xf]):
print('Error, bad cmd:', cmd)
return
self.pkt_sub_pkt = self.pkt_sub_pkt + 1
self.sub_pkt.append([dev_idx, ptype, addr, cmd, len, data])
def gen(self, filename="pkt_verilog_test.v", filetype=0):
pkt_len = 4
for sp in self.sub_pkt:
pkt_len = pkt_len + sp[4] + 8
if pkt_len > 2048:
print('Error, bad pkt_len:', pkt_len)
return
print("pkt_len:%d, data_len:%d"%(pkt_len, pkt_len - 4))
i = 4
if self.pkt_type == 1:
pkt_len = 0
self.pkt_buf[0] = 0
self.pkt_buf[1] = (self.pkt_type & 0x3)
self.pkt_buf[2] = 0x3f
self.pkt_buf[3] = 0xff
else:
#for i in range(pkt_len):
self.pkt_buf[0] = ((pkt_len - 4) >> 4) & 0xff
self.pkt_buf[1] = (((pkt_len - 4) << 4) & 0xf0) | (self.pkt_type & 0x3)
self.pkt_buf[2] = 0
self.pkt_buf[3] = 0
for sp in self.sub_pkt:
#设备索引(14bit)+类型(2bit)
temp = ((sp[0] & 0x3fff) << 2) | (sp[1] & 0x3)
self.pkt_buf[i] = (temp >> 8) & 0xff
i = i + 1
self.pkt_buf[i] = temp & 0xff
i = i + 1
#地址
self.pkt_buf[i] = (sp[2] >> 8) & 0xff
i = i + 1
self.pkt_buf[i] = sp[2] & 0xff
i = i + 1
#命令类型(4bit)+长度(12bit)
temp = (sp[3] << 12) | (sp[4] & 0xfff)
self.pkt_buf[i] = (temp >> 8) & 0xff
i = i + 1
self.pkt_buf[i] = temp & 0xff
i = i + 1
#工作计数
self.pkt_buf[i] = 0
i = i + 1
self.pkt_buf[i] = 0
i = i + 1
#数据
if sp[5] == None:
for _k in range(sp[4]):
self.pkt_buf[_k + i] = random.randint(0, 255)
else:
for _k in range(sp[4]):
self.pkt_buf[_k + i] = sp[5][_k]
i = i + sp[4]
f = open(filename, "wt+")
for _k in range(i):
f.write("txBuf[%d]=8'h%02X;\n"%(3 + _k, self.pkt_buf[_k]))
f.close()
if __name__ == '__main__':
packet = b'\0' * 2048
pkt = cast (packet, POINTER(TPkt)).contents
pkt.len = 512
pkt.rsv1 = 0
pkt.type = 0
pkt.rsv2 = 0
pkt.data[0] = 1
pkt.data[1] = 2
pkt.data[2] = 3
pkt.data[3] = 4
for i in range(10):
print(hex(packet[i]), ' ')
genPkt = GenPkt()
genPkt.setPktType(1)
genPkt.gen("./generated/shin/detect_pkt.vh")
genPkt.clean()
genPkt.setPktType(0)
#subPkt参数：索引、索引类型、地址、命令、长度、数据
genPkt.addSubPkt(0x02, 0, 0x0830, 0x2, 56, None)
genPkt.addSubPkt(0x01, 0, 0x0840, 0x1, 56, None)
genPkt.addSubPkt(0x03, 0, 0x1040, 0x0, 56, None)
for i in range(5):
genPkt.addSubPkt(random.randint(100, 4000), 0, random.randint(0, 65536), 0x0, 56, None)
#genPkt.addSubPkt(10, 2, 0x300, 0x2, 4, None)
#genPkt.addSubPkt(11, 1, 0x300, 0x2, 4, None)
genPkt.gen("./generated/shin/seq_pkt.vh")