修改包生成方式,改为宏定义方式,自动生成填包的长度数据等信息
This commit is contained in:
@@ -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)
|
||||
Reference in New Issue
Block a user