384 lines
12 KiB
Python
Executable File
384 lines
12 KiB
Python
Executable File
#!/usr/bin/env python
|
||
#encoding=utf-8
|
||
import random
|
||
|
||
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()
|
||
|
||
def clean(self):
|
||
self.pkt_buf = bytearray(b'\0' * 2048)
|
||
self.pkt_type = 0
|
||
self.pkt_sub_pkt = 0
|
||
self.sub_pkt = list()
|
||
|
||
def setPktType(self, type):
|
||
if (type != 0) and (type != 1):
|
||
print('Error...set type failed:', type)
|
||
else:
|
||
self.pkt_type = type
|
||
|
||
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 = 4
|
||
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+")
|
||
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__':
|
||
genPkt = GenPkt()
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>嗅探数据包
|
||
genPkt.setPktType(1)
|
||
genPkt.gen("./generated/shin/detect_pkt.vh", 1)
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>普通数据包 sm1
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
#genPkt.addSubPkt(0x02, 0, 0x0830, 0x2, 56, None)
|
||
genPkt.addSubPkt(0x01, 0, 0x00, 0x0, 64, None)
|
||
genPkt.addSubPkt(0x03, 0, 0xc0, 0x0, 64, None)
|
||
#genPkt.addSubPkt(10, 2, 0x300, 0x2, 4, None)
|
||
#genPkt.addSubPkt(11, 1, 0x300, 0x2, 4, None)
|
||
genPkt.gen("./generated/shin/sm1_pkt.vh", 1)
|
||
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>普通数据包 sm2
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
#genPkt.addSubPkt(0x02, 0, 0x0830, 0x2, 56, None)
|
||
genPkt.addSubPkt(0x01, 0, 0x00, 0xd, 128, None)
|
||
#genPkt.addSubPkt(10, 2, 0x300, 0x2, 4, None)
|
||
#genPkt.addSubPkt(11, 1, 0x300, 0x2, 4, None)
|
||
genPkt.gen("./generated/shin/sm2_pkt.vh", 1)
|
||
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>交换数据 SMx
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
data = bytearray(b"\0" * 10)
|
||
data[0] = 16
|
||
data[1] = 0
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0x0302, 0xe, 2, data) #0x030a, 0xe, 2, data)
|
||
|
||
data = bytearray(b"\0"*100)
|
||
for i in range(100):
|
||
data[i] = i + 1
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0x0800, 0xe, 16, data) #0x1800, 0xe, 16, data)
|
||
|
||
genPkt.gen("./generated/shin/sm3_pkt.vh", 1)
|
||
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>普通数据包
|
||
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_0_pkt.vh", 1)
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>普通数据包0,用来操作sync
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
data = bytearray(b"\0"*100)
|
||
data[0] = 0 #0x0300
|
||
data[1] = 0 #0x0301
|
||
data[2] = 64 #0x302 长度1
|
||
data[3] = 0 #0x303 长度2
|
||
data[4] = 0
|
||
data[5] = 0
|
||
data[6] = 64
|
||
data[7] = 0
|
||
data[8] = 0
|
||
data[9] = 0
|
||
data[10] = 64
|
||
data[11] = 0
|
||
data[12] = 0
|
||
data[13] = 0
|
||
data[14] = 64
|
||
data[15] = 0
|
||
|
||
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
genPkt.addSubPkt(0x02, 0, 0x0300, 0x1, 4, data) #sm0
|
||
genPkt.addSubPkt(0x01, 0, 0x0308, 0x1, 4, data) #sm2
|
||
genPkt.addSubPkt(0x03, 0, 0x0310, 0x1, 4, data) #sm4
|
||
genPkt.addSubPkt(0x04, 0, 0x0318, 0x1, 4, data) #sm6
|
||
|
||
data = bytearray(b"\0"*100)
|
||
for i in range(64):
|
||
data[i] = i
|
||
data[10] = ord('B')
|
||
data[11] = ord('A')
|
||
genPkt.addSubPkt(0x02, 0, 0x0800, 0x1, 64, data)
|
||
|
||
data = bytearray(b"\0"*100)
|
||
for i in range(64):
|
||
data[i] = i
|
||
data[11] = ord('B')
|
||
data[12] = ord('A')
|
||
genPkt.addSubPkt(0x01, 0, 0x1800, 0x1, 64, data)
|
||
|
||
data = bytearray(b"\0"*100)
|
||
for i in range(64):
|
||
data[i] = i
|
||
data[12] = ord('B')
|
||
data[13] = ord('A')
|
||
genPkt.addSubPkt(0x03, 0, 0x2800, 0x1, 64, data)
|
||
|
||
data = bytearray(b"\0"*100)
|
||
for i in range(64):
|
||
data[i] = i
|
||
data[13] = ord('B')
|
||
data[14] = ord('A')
|
||
genPkt.addSubPkt(0x04, 0, 0x3800, 0x1, 64, data)
|
||
|
||
genPkt.gen("./generated/shin/seq_1_pkt.vh", 1)
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>普通数据包2
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
genPkt.addSubPkt(3, 0, 0x3000, 0x0, 129, b"\0"*129)
|
||
genPkt.addSubPkt(4, 0, 0x4000, 0x0, 140, b"\0"*140)
|
||
genPkt.addSubPkt(1, 0, 0x1000, 0x0, 128, b"\0"*128)
|
||
genPkt.addSubPkt(2, 0, 0x2000, 0x0, 152, b"\0"*152)
|
||
genPkt.addSubPkt(1, 0, 0x0306, 0x0, 2, b"\0"*2)
|
||
genPkt.addSubPkt(2, 0, 0x030e, 0x0, 2, b"\0"*2)
|
||
genPkt.addSubPkt(3, 0, 0x0316, 0x0, 2, b"\0"*2)
|
||
genPkt.addSubPkt(4, 0, 0x031e, 0x0, 2, b"\0"*2)
|
||
|
||
genPkt.gen("./generated/shin/seq_2_pkt.vh", 1)
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>同步数据包
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
genPkt.addSubPkt(0x0, 0, random.randint(0, 65535), 0xc, 12, b'\0'*12)
|
||
genPkt.gen("./generated/shin/sync_pkt.vh", 1)
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>设置同步信号的数据包
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
data = bytearray(b"\0" * 100)
|
||
data[0] = 0x0f #地址0x120
|
||
data[2] = 0xa
|
||
data[4] = 0xa
|
||
data[6] = 0xa
|
||
data[8] = 0xa
|
||
|
||
data[16] = 0x00 #地址0x130, sync同步点
|
||
data[17] = 0x00 #地址0x131
|
||
#data[18] = 0x10
|
||
|
||
data[24] = 0x80 #地址0x138, sync周期 4B
|
||
data[25] = 0x00 #地址0x139
|
||
|
||
data[28] = 0x81 #sync1偏移
|
||
data[29] = 0x0
|
||
|
||
data[36] = 0x20#sync3偏移
|
||
data[37] = 0x00
|
||
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0x120, 0xe, 0x28, data)
|
||
|
||
|
||
genPkt.gen("./generated/shin/sync_sig_pkt.vh", 1)
|
||
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>设置顺序地址
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
data = bytearray(b"\0" * 2)
|
||
data[0] = 0x3f
|
||
data[1] = 0xff
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
genPkt.addSubPkt(0x0, 0, random.randint(0, 65535), 0xb, 2, data)
|
||
|
||
|
||
genPkt.gen("./generated/shin/set_addr_pkt.vh", 1)
|
||
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>广播读
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
genPkt.addSubPkt(0x0, 0, 0xc0, 0xd, 4, b"\0"*4)
|
||
genPkt.addSubPkt(0x1, 0, 0xc0, 0x0, 4, b"\0"*4)
|
||
genPkt.addSubPkt(0x2, 0, 0xc0, 0x0, 4, b"\0"*4)
|
||
genPkt.addSubPkt(0x3, 0, 0xc0, 0x0, 4, b"\0"*4)
|
||
genPkt.addSubPkt(0x4, 0, 0xc0, 0x0, 4, b"\0"*4)
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0xc0, 0xd, 4, b"\0"*4)
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0xc0, 0xd, 4, b"\0"*4)
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0xc0, 0xd, 4, b"\0"*4)
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0xc0, 0xd, 4, b"\0"*4)
|
||
|
||
genPkt.gen("./generated/shin/bc_r_pkt.vh", 1)
|
||
|
||
|
||
#-------------------------------------------------------------------------
|
||
#>>>>>>广播写
|
||
genPkt.clean()
|
||
genPkt.setPktType(0)
|
||
|
||
#subPkt参数：索引、索引类型、地址、命令、长度、数据
|
||
data = bytearray(b"\0" * 2)
|
||
data[0] = 32
|
||
data[1] = 0
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0x0302, 0xe, 2, data)
|
||
|
||
data = bytearray(b"\0"*100)
|
||
for i in range(100):
|
||
data[i] = i + 1
|
||
genPkt.addSubPkt(random.randint(0, 2048), 0, 0x0800, 0xe, 32, data)
|
||
|
||
genPkt.gen("./generated/shin/bc_w_pkt.vh", 1)
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|