修改包生成方式,改为宏定义方式,自动生成填包的长度数据等信息

This commit is contained in:
2025-05-21 15:56:35 +08:00
parent 5b3fcd2e82
commit 2da3c461a5
2 changed files with 79 additions and 40 deletions

View File

@@ -50,7 +50,7 @@ class GenPkt(object):
i = 4
if self.pkt_type == 1:
pkt_len = 0
pkt_len = 4
self.pkt_buf[0] = 0
self.pkt_buf[1] = (self.pkt_type & 0x3)
self.pkt_buf[2] = 0x3f
@@ -102,8 +102,36 @@ class GenPkt(object):
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]))
if filetype == 1:
define_str = "`define DEF_"
pos1 = filename.rfind("/")
if (pos1 < -1):
pos1 = 0
else:
pos1 = pos1 + 1
pos2 = filename.rfind(".")
if (pos1 > pos2):
define_str = define_str + filename[pos1:]
else:
define_str = define_str + filename[pos1:pos2]
f.write(define_str + "_len \t%d\n"%(pkt_len))
f.write(define_str + "_sm(array) \\\n\tarray``[%d]=8'h%02X;\\\n\tarray``[%d]=8'h%02X;\n"%(
3, pkt_len & 0xff,
4, pkt_len >> 8
))
f.write(define_str + "(array) \\\n")
for _k in range(i):
if _k < (i - 1):
f.write("\tarray``[%d]=8'h%02X;\\\n"%(3 + _k, self.pkt_buf[_k]))
else:
f.write("\tarray``[%d]=8'h%02X;\n"%(3 + _k, self.pkt_buf[_k]))
else:
for _k in range(i):
f.write("txBuf[%d]=8'h%02X;\n"%(3 + _k, self.pkt_buf[_k]))
f.close()
if __name__ == '__main__':
@@ -112,7 +140,7 @@ if __name__ == '__main__':
#-------------------------------------------------------------------------
#>>>>>>嗅探数据包
genPkt.setPktType(1)
genPkt.gen("./generated/shin/detect_pkt.vh")
genPkt.gen("./generated/shin/detect_pkt.vh", 1)
#-------------------------------------------------------------------------
#>>>>>>普通数据包
@@ -129,7 +157,7 @@ if __name__ == '__main__':
56, None)
#genPkt.addSubPkt(10, 2, 0x300, 0x2, 4, None)
#genPkt.addSubPkt(11, 1, 0x300, 0x2, 4, None)
genPkt.gen("./generated/shin/seq_0_pkt.vh")
genPkt.gen("./generated/shin/seq_0_pkt.vh", 1)
#-------------------------------------------------------------------------
#>>>>>>普通数据包0,用来操作sync
@@ -156,7 +184,7 @@ if __name__ == '__main__':
genPkt.addSubPkt(0x03, 0, 0x2800, 0x1, 64, data)
genPkt.addSubPkt(0x04, 0, 0x3800, 0x1, 64, data)
genPkt.gen("./generated/shin/seq_1_pkt.vh")
genPkt.gen("./generated/shin/seq_1_pkt.vh", 1)
#-------------------------------------------------------------------------
#>>>>>>同步数据包
@@ -164,7 +192,7 @@ if __name__ == '__main__':
genPkt.setPktType(0)
#subPkt参数索引、索引类型、地址、命令、长度、数据
genPkt.addSubPkt(0x0, 0, random.randint(0, 65536), 0xe, 12, b'\0'*12)
genPkt.gen("./generated/shin/sync_pkt.vh")
genPkt.gen("./generated/shin/sync_pkt.vh", 1)
#-------------------------------------------------------------------------
#>>>>>>设置同步信号的数据包
@@ -193,4 +221,4 @@ if __name__ == '__main__':
genPkt.addSubPkt(0x4, 0, 0x120, 0x1, 0x28, data)
genPkt.gen("./generated/shin/sync_sig_pkt.vh")
genPkt.gen("./generated/shin/sync_sig_pkt.vh", 1)