重构代码.
This commit is contained in:
242
source/device/function/file_upgrade.py
Normal file
242
source/device/function/file_upgrade.py
Normal file
@@ -0,0 +1,242 @@
|
||||
import hashlib
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
from crc import Calculator, Crc16
|
||||
|
||||
from tools.IntelHex import file_IntelHex_to_Bin, file_Bin_to_IntelHex
|
||||
|
||||
|
||||
Header_Tag = {
|
||||
'file_type': [0x00, 2], # 0 - 文件类型; 2byte
|
||||
'file_version': [0x01, 2], # 1 - 文件版本; 2byte
|
||||
'file_length': [0x02, 4], # 2 - 文件长度; 4byte
|
||||
'md5': [0x03, 16], # 3 - 文件MD5; 16byte
|
||||
'encrypt': [0x04, 1], # 4 - 加密算法; 1byte
|
||||
'update_type': [0x05, 1], # 5 - 升级文件类别; 1byte
|
||||
'update_spec': [0x06, 4], # 6 - 升级特征字; 4byte
|
||||
'update_verison': [0x07, 4], # 7 - 升级版本号; 4byte
|
||||
'update_date': [0x08, 3], # 8 - 升级版本日期; 3byte
|
||||
'area_code': [0x09, 4], # 9 - 省份特征; 4byte
|
||||
'uptate_str': [0x0A, -1, 64], # 10 - 升级段描述; less than 64byte
|
||||
'device_str': [0x0D, -1, 64], # 13 - 设备特征描述; less than 64byte
|
||||
'hex_name': [0xFF, -1, 80], # 255 - Hex文件名; less than 80byte
|
||||
}
|
||||
|
||||
|
||||
def file_encryption(buffer):
|
||||
""" 文件加密算法 """
|
||||
pwd_idx = 0
|
||||
pwd = b'moc.mmocspot.www'
|
||||
pwd = list(map(lambda x: (x - 0x30 + 0x100) % 0x100, pwd))
|
||||
result = bytearray(len(buffer))
|
||||
for i in range(len(buffer)):
|
||||
k = i
|
||||
k |= i >> 8
|
||||
k |= i >> 16
|
||||
k |= i >> 24
|
||||
result[i] = buffer[i] ^ pwd[pwd_idx] ^ (k & 0xFF)
|
||||
pwd_idx = (pwd_idx + 1) % len(pwd)
|
||||
return result
|
||||
|
||||
|
||||
def build_header(config: dict, len_max=512):
|
||||
"""
|
||||
基于配置参数, 生成文件信息头;
|
||||
V1版本, 依据字典生成tag标签组;
|
||||
"""
|
||||
# 定义文件头
|
||||
m_file_header = bytearray(len_max)
|
||||
|
||||
header_len = 11
|
||||
tag_num = 0
|
||||
for tag, value in config.items():
|
||||
if tag in Header_Tag.keys():
|
||||
if tag == 'hex_name' and len_max < 256:
|
||||
""" 当文件头长度不足时, 跳过文件名标签 """
|
||||
continue
|
||||
elif Header_Tag[tag][1] == -1:
|
||||
tag_len = min(len(value), Header_Tag[tag][2])
|
||||
else:
|
||||
tag_len = Header_Tag[tag][1]
|
||||
tag_date = [Header_Tag[tag][0], tag_len] + value[:tag_len]
|
||||
m_file_header[header_len: header_len + tag_len + 2] = bytearray(tag_date)
|
||||
tag_num += 1
|
||||
header_len += 2 + tag_len
|
||||
m_file_header[0:8] = b"TOPSCOMM"
|
||||
m_file_header[8] = ((header_len - 10) % 0x100)
|
||||
m_file_header[9] = ((header_len - 10) // 0x100)
|
||||
m_file_header[10] = tag_num
|
||||
m_file_header[header_len] = sum(m_file_header[:header_len]) % 0x100
|
||||
m_file_header[header_len+1] = sum(m_file_header[:header_len]) // 0x100
|
||||
|
||||
if header_len+2 > len_max:
|
||||
return None
|
||||
else:
|
||||
return m_file_header
|
||||
|
||||
|
||||
def build_header_lite(config: dict):
|
||||
"""
|
||||
基于配置参数, 生成文件信息头;
|
||||
V2版本, 仅提供必要信息;
|
||||
"""
|
||||
# 定义文件头
|
||||
m_file_header = bytearray(64)
|
||||
m_file_header[0:4] = bytearray(config["update_verison"])
|
||||
m_file_header[4:8] = bytearray(config["file_length"])
|
||||
m_file_header[8:24] = bytearray(config["md5"])
|
||||
|
||||
return m_file_header
|
||||
|
||||
|
||||
def build_header_new(config: dict):
|
||||
"""
|
||||
基于配置参数, 生成新版文件信息头;
|
||||
V3版本, 依据新版格式填充数据;
|
||||
"""
|
||||
# 定义文件头
|
||||
m_file_header = [0xFF] * 184
|
||||
|
||||
m_file_header[0:8] = list(b"TOPSCOMM")
|
||||
m_file_header[8:10] = config['prod_type']
|
||||
m_file_header[10:22] = config['prog_id'] + [0] * (12 - len(config['prog_id']))
|
||||
if config['method_compress'] == True:
|
||||
m_file_header[23] = 0x01
|
||||
else:
|
||||
m_file_header[23] = 0x00
|
||||
|
||||
if 'crc32' in config.keys():
|
||||
m_file_header[22] = 0x00
|
||||
m_file_header[24: 40] = config['crc32'] + [0x00] * 12
|
||||
elif 'md5' in config.keys():
|
||||
m_file_header[22] = 0x01
|
||||
m_file_header[24: 40] = config['md5']
|
||||
else:
|
||||
raise Exception("Error, Unknown method verify.")
|
||||
|
||||
# 时间戳生成
|
||||
time_now = datetime.now()
|
||||
time_stamp = list(map(lambda x: int(x),
|
||||
time_now.strftime("%Y-%m-%d-%H-%M").split('-')))
|
||||
time_stamp.insert(1, time_stamp[0] // 0x100)
|
||||
time_stamp[0] = time_stamp[0] % 0x100
|
||||
m_file_header[40: 46] = time_stamp
|
||||
|
||||
m_file_header[46: 50] = config['file_length']
|
||||
# Cpu1
|
||||
m_file_header[50: 54] = [0x00] * 4
|
||||
m_file_header[54: 70] = [0x00] * 16
|
||||
# Cpu2
|
||||
m_file_header[70: 74] = [0x00] * 4
|
||||
m_file_header[74: 90] = [0x00] * 16
|
||||
|
||||
if config['prog_type'] == 'app':
|
||||
m_file_header[90: 92] = [0x00, 0x00]
|
||||
elif config['prog_type'] == 'boot':
|
||||
m_file_header[90: 92] = [0x01, 0x00]
|
||||
elif config['prog_type'] == 'diff':
|
||||
m_file_header[90: 92] = [0x02, 0x00]
|
||||
elif config['prog_type'] == 'font':
|
||||
m_file_header[90: 92] = [0x03, 0x00]
|
||||
elif config['prog_type'] == 'config':
|
||||
m_file_header[90: 92] = [0x04, 0x00]
|
||||
elif config['prog_type'] == 'data':
|
||||
m_file_header[90: 92] = [0x05, 0x00]
|
||||
elif config['prog_type'] == 'test':
|
||||
m_file_header[90: 92] = [0x06, 0x00]
|
||||
else:
|
||||
raise Exception("Error, Unknown Program Type.")
|
||||
|
||||
m_file_header[92: 94] = config['area_code']
|
||||
m_file_header[94: 158] = config['hex_name']
|
||||
if 'upgrade_type' in config.keys():
|
||||
m_file_header[158: 160] = config['upgrade_type']
|
||||
m_file_header[160: 182] = [0x00] * 22
|
||||
else:
|
||||
m_file_header[158: 182] = [0x00] * 24
|
||||
|
||||
m_file_header = bytearray(m_file_header)
|
||||
|
||||
calculator = Calculator(Crc16.MODBUS)
|
||||
code_crc16 = calculator.checksum(m_file_header[:-2])
|
||||
m_file_header[182: 184] = [code_crc16 // 0x100, code_crc16 % 0x100]
|
||||
|
||||
return m_file_header
|
||||
|
||||
|
||||
def parser_version_info(file_bin: bytearray, base_addr:int):
|
||||
""" 解析Bin文件内的版本信息结构体 """
|
||||
address_block = (file_bin[6] << 24) + (file_bin[7] << 16) + (file_bin[4] << 8) + file_bin[5]
|
||||
address_block -= base_addr
|
||||
address_block *= 2
|
||||
|
||||
offset = address_block
|
||||
block_version = {}
|
||||
block_version['name'] = file_bin[offset + 1 : offset + 64 : 2]
|
||||
offset += 64
|
||||
block_version['device'] = file_bin[offset + 1 : offset + 64 : 2]
|
||||
offset += 64
|
||||
block_version['factory'] = file_bin[offset + 1 : offset + 32 : 2]
|
||||
offset += 32
|
||||
block_version['hardware'] = file_bin[offset + 1 : offset + 64 : 2]
|
||||
offset += 64
|
||||
|
||||
return block_version
|
||||
|
||||
def test1(fp: Path):
|
||||
""" bin文件生成测试 """
|
||||
file1 = Path(fp)
|
||||
path1 = file1.parent / (file1.stem + ".bin")
|
||||
data1 = file1.read_text()
|
||||
data2 = file_IntelHex_to_Bin(data1, 0x3F0100)
|
||||
path1.write_bytes(data2)
|
||||
|
||||
# 测试Bin到IntelHex
|
||||
bin = Path(fp).read_bytes()
|
||||
result = file_Bin_to_IntelHex(bin, 0x80000, memory_width=2)
|
||||
(fp.parent / (fp.stem + ".hex")).write_text(result)
|
||||
|
||||
|
||||
def test2():
|
||||
""" 校验, 加密测试 """
|
||||
# Header - Crc16
|
||||
bin_offcial = Path(r"D:\WorkSpace\UserTool\SelfTool\FrameParser\test\p460\result\lamina_adapter_t1.dat")
|
||||
data_offcial = bin_offcial.read_bytes()
|
||||
byte_data = data_offcial[:182]
|
||||
crc = data_offcial[182:184]
|
||||
# data = "54 4F 50 53 43 4F 4D 4D 45 00 53 4C 43 50 30 30 31 00 00 00 00 00 01 00 B6 61 A8 73 BF 82 9E A7 4C 79 F6 BB 94 E2 A5 18 E8 07 05 18 0B 17 18 4C 01 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 6C 61 6D 69 6E 61 5F 61 64 61 70 74 65 72 5F 6D 61 69 6E 02 00 00 00 00 20 10 53 06 00 00 00 00 80 A5 8F 02 00 00 00 00 EC 1F 40 00 00 00 00 00 00 00 00 00 00 00 00 00 7A CA 61 0A FD 7F 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00"
|
||||
# byte_data = list(map(lambda x: int(x, 16), data.split(' ')))
|
||||
# crc = "99 CB"
|
||||
calculator = Calculator(Crc16.MODBUS)
|
||||
code_crc16 = calculator.checksum(bytearray(byte_data))
|
||||
print(f"Result = {hex(code_crc16)}, Offcial Result = {crc}")
|
||||
|
||||
# File - md5
|
||||
data_bin = '123456 123456 '.encode()
|
||||
md5_ctx = hashlib.md5()
|
||||
md5_ctx.update(data_bin)
|
||||
hash = md5_ctx.hexdigest()
|
||||
# File - encrypt
|
||||
buffer1 = data_bin[:]
|
||||
buffer1_en = file_encryption(buffer1)
|
||||
buffer2 = buffer1_en[:6] + buffer1[6:]
|
||||
buffer2_de = file_encryption(buffer2)
|
||||
pass
|
||||
|
||||
|
||||
def task5():
|
||||
""" 文件缓冲区对比测试 """
|
||||
file_dat = Path(r"test\p280039\result\lamina_controller_dsp_t1.dat")
|
||||
file_dat_buffer = Path(r"test\p280039\result\lamina_controller_dsp_buffer.bin")
|
||||
file_dat_buffer.exists(), file_dat.exists()
|
||||
data_dat = file_dat.read_bytes()
|
||||
data_dat_buffer = file_dat_buffer.read_bytes()
|
||||
for i in range(len(data_dat)):
|
||||
if data_dat[i] != data_dat_buffer[2*i]:
|
||||
print(f"Diff in {hex(i)}, Data: {data_dat[i]}!={data_dat_buffer[2*i]}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# path_bin = Path("F:\\Work\\FPGA\\Test\\Vivado\\test_update\\test_update.vitis\\upgrade_system\\Debug\\sd_card\\BOOT.BIN")
|
||||
# GeneratePackage_Demo_Xilinx(path_bin)
|
||||
pass
|
||||
395
source/device/function/frame.py
Normal file
395
source/device/function/frame.py
Normal file
@@ -0,0 +1,395 @@
|
||||
import struct
|
||||
from crc import Calculator, Crc16
|
||||
from tools.ByteConv import trans_list_to_str, trans_str_to_list, display_hex
|
||||
|
||||
modbus_map = {
|
||||
# 1 - Hex
|
||||
# 2 - Int16
|
||||
# 3 - lnt32
|
||||
# 4 - str
|
||||
# 5 - addr
|
||||
# 6 - float
|
||||
# 7 - numberList
|
||||
0x01: ["Hex示例", 1],
|
||||
0x02: ["Int示例", 2],
|
||||
0x03: ["Int32示例", 3],
|
||||
0x04: ["str示例", 4, 16],
|
||||
0x10: ["addr示例", 5, 6],
|
||||
0x20: ["Float示例", 6],
|
||||
}
|
||||
|
||||
frame_modbus = {
|
||||
# func: len_base, type, idx
|
||||
# type = 0, fixed length
|
||||
# type = 1, add uint8 length by idx
|
||||
# type = 2, add uint16 length by idx
|
||||
# type = 3, check sub func code by idx, choose fixed length
|
||||
0x03: [5, 1, 2],
|
||||
0x04: [5, 1, 2],
|
||||
0x06: [8, 0],
|
||||
0x07: [8, 3, 2, {0x01: 2, 0x02: 0, 0x03: 0}],
|
||||
0x10: [8, 0],
|
||||
0x11: [11, 2, 7]
|
||||
}
|
||||
|
||||
|
||||
def make_frame_modbus(block:dict) -> bytearray:
|
||||
""" modbus 生成函数"""
|
||||
frame = []
|
||||
calculator = Calculator(Crc16.MODBUS)
|
||||
|
||||
frame.append(block['addr_dev'])
|
||||
if block['type'] == 'update':
|
||||
""" 升级系列报文 """
|
||||
frame.append(0x07)
|
||||
if len(block['file']) <= block["header_offset"]:
|
||||
raise Exception("Modbus Update error, file too small.")
|
||||
|
||||
if block['step'] == 'start':
|
||||
frame.append(0x01)
|
||||
frame.append(0x00)
|
||||
frame.append(0x00)
|
||||
frame.append(0x00)
|
||||
frame.append(block['header_offset'] // 256)
|
||||
frame.append(block['header_offset'] % 256)
|
||||
frame += list(block['file'][:block['header_offset']])
|
||||
elif block['step'] == 'trans':
|
||||
file_offset = block["header_offset"] + block['index'] * block['file_block_size']
|
||||
file_block = block['file'][file_offset: file_offset + block['file_block_size']]
|
||||
frame.append(0x02)
|
||||
frame.append(0x00)
|
||||
frame.append(block['index'] // 256)
|
||||
frame.append(block['index'] % 256)
|
||||
frame.append(len(file_block) // 256)
|
||||
frame.append(len(file_block) % 256)
|
||||
frame += list(file_block)
|
||||
elif block['step'] == 'end':
|
||||
frame.append(0x03)
|
||||
frame.append(0x00)
|
||||
frame.append(block['index'] // 256)
|
||||
frame.append(block['index'] % 256)
|
||||
frame.append(0x00)
|
||||
frame.append(0x00)
|
||||
else:
|
||||
raise Exception("Modbus Update Frame Step Error.")
|
||||
elif block['type'][:6] == 'record':
|
||||
""" 录波系列报文 """
|
||||
frame.append(0x11)
|
||||
if block['type'][7:10] == 'cfg':
|
||||
frame.append(0x01)
|
||||
elif block['type'][7:11] == 'data':
|
||||
frame.append(0x02)
|
||||
else:
|
||||
raise Exception("Modbus Record Frame Type Error.")
|
||||
|
||||
if block['step'] == 'start':
|
||||
frame.append(0x00)
|
||||
elif block['step'] == 'next':
|
||||
frame.append(0x01)
|
||||
elif block['step'] == 'repeat':
|
||||
frame.append(0x02)
|
||||
else:
|
||||
raise Exception("Modbus Record Frame Step Error.")
|
||||
|
||||
frame.append(block['file_block_size'] // 256)
|
||||
frame.append(block['file_block_size'] % 256)
|
||||
else:
|
||||
""" 数据读取系列报文 """
|
||||
data_addr = block['data_addr']
|
||||
if block['type'] == "read":
|
||||
frame.append(0x03)
|
||||
data_len = block['data_len']
|
||||
frame.append(data_addr // 256 % 256)
|
||||
frame.append(data_addr % 256)
|
||||
frame.append(data_len // 256 % 256)
|
||||
frame.append(data_len % 256)
|
||||
elif block['type'] == "write_one":
|
||||
frame.append(0x06)
|
||||
data_val = block['data_val']
|
||||
if data_val < 0:
|
||||
data_val += 0x1_0000
|
||||
frame.append(data_addr // 256 % 256)
|
||||
frame.append(data_addr % 256)
|
||||
frame.append(data_val // 256 % 256)
|
||||
frame.append(data_val % 256)
|
||||
elif block['type'] == "write_dual":
|
||||
frame.append(0x10)
|
||||
data_val = block['data_val']
|
||||
if data_val < 0:
|
||||
data_val += 0x1_0000_0000
|
||||
frame.append(data_addr // 256 % 256)
|
||||
frame.append(data_addr % 256)
|
||||
frame.append(0x00)
|
||||
frame.append(0x02)
|
||||
frame.append(0x04)
|
||||
frame.append(data_val // 256 % 256)
|
||||
frame.append(data_val % 256)
|
||||
data_val //= 0x1_0000
|
||||
frame.append(data_val // 256 % 256)
|
||||
frame.append(data_val % 256)
|
||||
elif block['type'] == "write_str":
|
||||
frame.append(0x10)
|
||||
data_len = len(block['data_val'])
|
||||
item_len = 2 * block['data_define'][data_addr][2]
|
||||
data_val = block['data_val']
|
||||
if data_len > item_len:
|
||||
raise Exception("Modbus data len oversize.")
|
||||
elif data_len < item_len:
|
||||
data_val += b'\000' * (item_len - data_len)
|
||||
data_len = len(data_val)
|
||||
frame.append(data_addr // 256 % 256)
|
||||
frame.append(data_addr % 256)
|
||||
frame.append(0x00)
|
||||
frame.append(data_len // 2)
|
||||
frame.append(data_len)
|
||||
|
||||
if type(data_val[0]) == str:
|
||||
data_val = list(map(lambda x: x.encode()[0], data_val))
|
||||
for i in range(data_len//2):
|
||||
frame.append(data_val[2*i + 1])
|
||||
frame.append(data_val[2*i])
|
||||
else:
|
||||
raise Exception("Modbus Frame Type Error.")
|
||||
|
||||
crc = calculator.checksum(bytearray(frame))
|
||||
frame.append(crc % 256)
|
||||
frame.append(crc // 256)
|
||||
|
||||
return bytearray(frame)
|
||||
|
||||
|
||||
def make_frame_dlt645(block:dict) -> bytearray:
|
||||
""" dlt645 生成函数"""
|
||||
frame = []
|
||||
|
||||
if block['type'] == 'modbus':
|
||||
""" Modbus 透传帧 """
|
||||
ctrl_code = 0x1F # Modbus 透传帧功能码
|
||||
data_frame = make_frame_modbus(block["data"])
|
||||
len_data = len(data_frame)
|
||||
else:
|
||||
raise Exception("Unknown dlt645 frame type.")
|
||||
|
||||
frame.append(0x68)
|
||||
frame += block['addr'][:6]
|
||||
frame.append(0x68)
|
||||
frame.append(ctrl_code)
|
||||
frame.append(len_data)
|
||||
frame += data_frame
|
||||
frame.append(sum(frame) % 256)
|
||||
frame.append(0x16)
|
||||
|
||||
return bytearray(frame)
|
||||
|
||||
|
||||
def check_frame_modbus(frame:bytes, block:dict) -> dict:
|
||||
""" 校验modbus帧回复 """
|
||||
if len(frame:=find_frame_modbus(frame, block['addr_dev'])) == 0:
|
||||
raise Exception("No frame data")
|
||||
|
||||
output = {
|
||||
'result': False,
|
||||
'code_func': frame[1],
|
||||
'code_sub': frame[2],
|
||||
}
|
||||
if output['code_func'] == 0x07:
|
||||
""" 升级回复帧 """
|
||||
output['upgrade'] = {}
|
||||
output['upgrade']['index'] = frame[4] * 256 + frame[5]
|
||||
if output['code_sub'] == 0x01:
|
||||
""" 升级开始, 处理文件头 """
|
||||
output['upgrade']['length'] = frame[6] * 256 + frame[7]
|
||||
elif output['code_sub'] == 0x02:
|
||||
""" 升级传输, 解析帧序号及返回值, 不做序列号校验 """
|
||||
pass
|
||||
elif output['code_sub'] == 0x03:
|
||||
""" 升级结束 """
|
||||
pass
|
||||
else:
|
||||
raise Exception(" Upgrade SubFunc code error.")
|
||||
output['result'] = True if frame[3] == 0x00 else False
|
||||
output['code_error'] = frame[3]
|
||||
elif output['code_func'] == 0x03 or output['code_func'] == 0x04:
|
||||
""" 数据读取帧 """
|
||||
if frame[2] == len(frame[3:-2]):
|
||||
output['Regs'] = display_data(block['data_addr'], frame[3:-2], block['data_define'])
|
||||
else:
|
||||
raise Exception("Frame read data length error.")
|
||||
output['result'] = True
|
||||
elif output['code_func'] == 0x06:
|
||||
""" 单个数据写入帧 """
|
||||
output['result'] = True
|
||||
elif output['code_func'] == 0x10:
|
||||
""" 多个数据写入帧 """
|
||||
output['result'] = True
|
||||
elif output['code_func'] == 0x11:
|
||||
""" 录波功能帧 """
|
||||
data_record = {}
|
||||
if frame[2] == 0x01:
|
||||
data_record['type'] = 'config'
|
||||
elif frame[2] == 0x02:
|
||||
data_record["type"] = 'data'
|
||||
else:
|
||||
raise Exception("Unknow data type")
|
||||
data_record['seq'] = frame[5] * 0x100 + frame[6]
|
||||
data_record['total'] = frame[3] * 0x100 + frame[4]
|
||||
data_record['data'] = frame[9:-2]
|
||||
output['record'] = data_record
|
||||
output['result'] = True
|
||||
elif output['code_func'] & 0x80:
|
||||
""" 错误返回帧 """
|
||||
output['code_error'] = frame[2] if output['code_func'] == 0x87 else frame[3]
|
||||
else:
|
||||
raise Exception(f"Frame Date error. func={output['code_func']}, func_sub={output['code_sub']}, len={len(frame)}")
|
||||
|
||||
return output
|
||||
|
||||
|
||||
def check_frame_dlt645(frame:bytes, block:dict) -> dict:
|
||||
""" 校验dlt645帧回复 """
|
||||
if len(frame:=find_frame_dlt645(frame, block['addr'])) == 0:
|
||||
raise Exception("No frame data")
|
||||
|
||||
code_func = frame[8]
|
||||
if code_func == 0x9F:
|
||||
return check_frame_modbus(frame[10:-2], block['data'])
|
||||
else:
|
||||
raise Exception("DLT645 Frame type error.")
|
||||
|
||||
|
||||
def find_frame_modbus(buffer:bytes, address:int, frame_defines: dict=frame_modbus) -> bytes:
|
||||
""" 搜索合法modbus帧子串 """
|
||||
len_buffer = len(buffer)
|
||||
|
||||
pos_frame, len_frame = 0, 0
|
||||
calculator = Calculator(Crc16.MODBUS)
|
||||
for i in range(len_buffer):
|
||||
if buffer[i] != address:
|
||||
continue
|
||||
if (buffer[i+1] & 0x7F) not in frame_defines.keys():
|
||||
continue
|
||||
|
||||
frame_define = frame_defines[buffer[i+1] & 0x7F]
|
||||
j = frame_define[0]
|
||||
if buffer[i+1] & 0x80:
|
||||
j = 5
|
||||
elif frame_define[1] == 0:
|
||||
pass
|
||||
elif frame_define[1] == 1:
|
||||
j += buffer[i+frame_define[2]]
|
||||
elif frame_define[1] == 2:
|
||||
j += buffer[i+frame_define[2]] * 0x100 + buffer[i+frame_define[2]+1]
|
||||
elif frame_define[1] == 3:
|
||||
if buffer[i+frame_define[2]] not in frame_define[3].keys():
|
||||
continue
|
||||
j += frame_define[3][buffer[i+frame_define[2]]]
|
||||
else:
|
||||
raise Exception("Unknow Modbus Define type.")
|
||||
|
||||
if ((i+j) <= len_buffer) and calculator.checksum(buffer[i:i+j-2]) == (buffer[i+j-1] * 0x100 + buffer[i+j-2]):
|
||||
pos_frame, len_frame = i, j
|
||||
break
|
||||
|
||||
return buffer[pos_frame: pos_frame+len_frame]
|
||||
|
||||
|
||||
def find_frame_dlt645(buffer:bytes, address: list) -> bytes:
|
||||
""" 搜索合法645帧子串 """
|
||||
len_buffer = len(buffer)
|
||||
|
||||
pos_frame, len_frame = 0, 0
|
||||
for i in range(len_buffer):
|
||||
if buffer[i] != 0x68 or buffer[i+7] != 0x68:
|
||||
continue
|
||||
if address[0] != 0xAA and buffer[i+1:i+7] != bytes(address):
|
||||
continue
|
||||
|
||||
j = buffer[i+9] + 12
|
||||
if sum(buffer[i:i+j-2]) % 0x100 == buffer[i+j-2]:
|
||||
pos_frame, len_frame = i, j
|
||||
break
|
||||
|
||||
return buffer[pos_frame: pos_frame+len_frame]
|
||||
|
||||
|
||||
def display_data(address: int, data: bytes, modbus_map: dict=modbus_map) -> dict:
|
||||
""" 格式化表示数据, 得到显示数据字典 """
|
||||
def swapping_words(data:bytes, length:int) -> bytearray:
|
||||
item = bytearray(data[:2 * length])
|
||||
for i in range(length):
|
||||
item[2*i], item[2*i+1] = item[2*i+1], item[2*i]
|
||||
return item
|
||||
def convert_regs_to_int(data:bytes, length:int, signed: bool=True) -> int:
|
||||
result = 0
|
||||
for idx_data, byte_data in enumerate(swapping_words(data, length)):
|
||||
result += byte_data * 2 ** (8 * idx_data)
|
||||
result -= 2 ** (16 * length) if signed and result >= (2 ** (16 * length - 1)) else 0
|
||||
return result
|
||||
|
||||
output_data = {}
|
||||
idx = address
|
||||
while data:
|
||||
data_label, data_len = "未知数据", 1
|
||||
if idx not in modbus_map.keys():
|
||||
item = convert_regs_to_int(data, 1, signed=False)
|
||||
item = display_hex(item, 4)
|
||||
else:
|
||||
current_map = modbus_map[idx]
|
||||
data_label = current_map[0]
|
||||
if current_map[1] == 1:
|
||||
""" Hex字符表示 """
|
||||
item = convert_regs_to_int(data, 1, signed=False)
|
||||
item = display_hex(item, 4)
|
||||
elif current_map[1] == 2:
|
||||
""" 16位数值表示 """
|
||||
item = convert_regs_to_int(data, 1)
|
||||
if len(current_map) > 2:
|
||||
item /= current_map[2]
|
||||
elif current_map[1] == 3:
|
||||
""" 32位数值表示 """
|
||||
data_len = 2
|
||||
item = convert_regs_to_int(data, 2)
|
||||
if len(current_map) > 2:
|
||||
item /= current_map[2]
|
||||
elif current_map[1] == 4:
|
||||
""" 字符串表示 """
|
||||
data_len = current_map[2]
|
||||
item = swapping_words(data, data_len)
|
||||
try:
|
||||
item = item.decode()
|
||||
except Exception as ex:
|
||||
item_len = sum([any(item[i:]) for i in range(len(item))])
|
||||
item = trans_list_to_str(item[:item_len])
|
||||
elif current_map[1] == 5:
|
||||
""" 载波地址表示 """
|
||||
data_len = current_map[2]
|
||||
item = swapping_words(data, data_len)
|
||||
item = trans_list_to_str(item)
|
||||
elif current_map[1] == 6:
|
||||
""" 浮点数值表示 """
|
||||
data_len = 2
|
||||
item = struct.unpack('>f', bytes([data[2], data[3], data[0], data[1]]))[0]
|
||||
elif current_map[1] == 7:
|
||||
""" 正序数值表示 """
|
||||
data_len = current_map[2]
|
||||
item = list(map(lambda x: x, data[:2 * data_len]))
|
||||
|
||||
output_data[idx] = data_label, item
|
||||
idx += data_len
|
||||
data = data[2*data_len:]
|
||||
return output_data
|
||||
|
||||
|
||||
def print_display(output_data: dict):
|
||||
""" 格式化表示输出数据 """
|
||||
print("Parse Result:")
|
||||
label_len_max = max(map(lambda x: len(x[0]), output_data.values()))
|
||||
data_len_max = max(map(lambda x: len(str(x[1])), output_data.values()))
|
||||
for key, value in output_data.items():
|
||||
label = value[0]
|
||||
data = "-".join(map(str, value[1])) if type(value) == list else value[1]
|
||||
print(f"{display_hex(key, 4)}: {data:<{data_len_max}} {label:<{label_len_max}}")
|
||||
|
||||
|
||||
|
||||
|
||||
29
source/device/function/tools/ByteConv.py
Normal file
29
source/device/function/tools/ByteConv.py
Normal file
@@ -0,0 +1,29 @@
|
||||
def trans_list_to_str(data: list) -> str:
|
||||
""" 标准串口字符串表示 """
|
||||
func_trans = lambda x: ('00' + hex(x % 256)[2:])[-2:].upper()
|
||||
return " ".join(map(func_trans, data))
|
||||
|
||||
|
||||
def trans_str_to_list(data: str) -> list:
|
||||
""" 标准串口字符串转换列表 """
|
||||
func_trans = lambda x: int(x, 16)
|
||||
return list(map(func_trans, data.strip().split(" ")))
|
||||
|
||||
|
||||
def conv_int_to_array(num: int, big_end=False):
|
||||
""" 数值转字节数组 """
|
||||
result = [0, 0, 0, 0]
|
||||
if big_end:
|
||||
indexlist = reversed(range(len(result)))
|
||||
else:
|
||||
indexlist = range(len(result))
|
||||
|
||||
for i in indexlist:
|
||||
result[i] = int(num % 0x100)
|
||||
num //= 0x100
|
||||
return result
|
||||
|
||||
|
||||
def display_hex(data:int, length:int=2) -> str:
|
||||
""" Hex字符固定最小长度表示 """
|
||||
return f"0x{data:0{length}X}"
|
||||
142
source/device/function/tools/IntelHex.py
Normal file
142
source/device/function/tools/IntelHex.py
Normal file
@@ -0,0 +1,142 @@
|
||||
def calculate_checksum(data):
|
||||
""" 计算校验域 """
|
||||
checksum = 0
|
||||
for i in range(0, len(data), 2):
|
||||
checksum += int(data[i: i+2], 16)
|
||||
return (0x100 - checksum) & 0xFF
|
||||
|
||||
|
||||
def file_IntelHex_to_Bin(file_data, base_address=0, len_max=1, **kwargs):
|
||||
"""
|
||||
将Intel Hex格式文件转换为Bin格式;
|
||||
"""
|
||||
if base_address == 0:
|
||||
if file_data[8] == '2':
|
||||
base_address = int(file_data[9: 13], 16) * 0x100
|
||||
offset_begin = 16
|
||||
elif file_data[8] == '4':
|
||||
base_address = int(file_data[9: 13], 16) * 0x10000
|
||||
offset_begin = 16
|
||||
else:
|
||||
base_address = 0
|
||||
offset_begin = 0
|
||||
base_address += int(file_data[offset_begin + 3: offset_begin + 7], 16)
|
||||
base_address &= ~0x0FFF
|
||||
|
||||
len_space = len_max
|
||||
result = bytearray(len_space).replace(b'\x00', b'\xff')
|
||||
lines = file_data.split('\n')
|
||||
offset = 0
|
||||
max_address = 0
|
||||
for line in lines:
|
||||
if line[0] == ':':
|
||||
checksum = sum([int(x, 16) * (15 * (~i & 0x01) + 1) for i, x in enumerate(line[1:])])
|
||||
if (checksum & 0x00FF) != 0:
|
||||
raise Exception('Hex file checksum error!')
|
||||
|
||||
if line[7:9] == '00':
|
||||
len_line = int(line[1:3], 16)
|
||||
address = offset + int(line[3:7], 16) - base_address
|
||||
data = bytearray.fromhex(line[9:-2])
|
||||
if 'conv_end' in kwargs.keys():
|
||||
address *= 2
|
||||
if kwargs['conv_end']:
|
||||
for i in range(0, len_line, 2):
|
||||
t = data[i]
|
||||
data[i] = data[i+1]
|
||||
data[i+1] = t
|
||||
|
||||
if (address + len_line) > max_address:
|
||||
max_address = address + len_line
|
||||
if max_address >= len_space:
|
||||
if len_max == 1:
|
||||
result += bytearray(max_address - len_space).replace(b'\x00', b'\xff')
|
||||
len_space += max_address - len_space
|
||||
else:
|
||||
raise Exception("Bin file Oversize.")
|
||||
result[address:address+len_line] = data
|
||||
|
||||
elif line[7:9] == '01':
|
||||
break
|
||||
elif line[7:9] == '02':
|
||||
offset = int(line[9:-2], 16) * 16
|
||||
elif line[7:9] == '03':
|
||||
pass
|
||||
elif line[7:9] == '04': # 拓展地址偏移
|
||||
offset = int(line[9:-2], 16) * 2**16
|
||||
elif line[7:9] == '05':
|
||||
pass
|
||||
return result[:max_address]
|
||||
|
||||
|
||||
def file_Bin_to_IntelHex(file_data: bytearray, base_address=0, **kwargs):
|
||||
"""
|
||||
将Bin格式文件转换为Intel Hex格式;
|
||||
"""
|
||||
chunk_size = 32
|
||||
rom_width = 8
|
||||
memory_width = 8
|
||||
if 'chunk_size' in kwargs.keys():
|
||||
""" 数据分块大小 """
|
||||
chunk_size = kwargs['chunk_size']
|
||||
if 'memory_width' in kwargs.keys():
|
||||
""" 地址数据位宽 """
|
||||
memory_width = kwargs['memory_width'] * 8
|
||||
if 'rom_width' in kwargs.keys():
|
||||
""" rom数据位宽 """
|
||||
rom_width = kwargs['rom_width'] * 8
|
||||
|
||||
def record_extern_address(address, record_type):
|
||||
""" 添加地址记录 """
|
||||
if record_type == 4:
|
||||
record_data = (address & 0xFFFF0000) >> 16
|
||||
elif record_type == 2:
|
||||
record_data = (address & 0xFFFF0) >> 4
|
||||
else:
|
||||
raise Exception("Unknow record tye.")
|
||||
|
||||
hex_record = f':020000{record_type:02X}{record_data:04X}'
|
||||
hex_record += f'{calculate_checksum(hex_record[1:]):02X}\n'
|
||||
return hex_record
|
||||
|
||||
result = str()
|
||||
|
||||
# 添加开始记录
|
||||
extern_address = 0
|
||||
if base_address >= 2 ** 32:
|
||||
""" 超过最大可表示范围, 抛出异常 """
|
||||
raise Exception("base_address oversize")
|
||||
elif base_address >= 2 ** 16:
|
||||
result += record_extern_address(base_address, 4)
|
||||
extern_address = base_address & 0xFFFF0000
|
||||
elif base_address >= 2 ** 4:
|
||||
result += record_extern_address(base_address, 2)
|
||||
extern_address = base_address & 0xFFFF0
|
||||
|
||||
record_type = 0
|
||||
binary_offset = 0
|
||||
while binary_offset < len(file_data):
|
||||
""" 循环分块处理数据 """
|
||||
data_chunk = file_data[binary_offset: binary_offset + chunk_size]
|
||||
|
||||
record_len = len(data_chunk)
|
||||
record_data = data_chunk
|
||||
record_addr = (base_address - extern_address + (binary_offset // (memory_width // 8)))
|
||||
if record_addr >= 0x10000:
|
||||
result += record_extern_address((base_address + (binary_offset // (memory_width // 8))), 4)
|
||||
extern_address = (base_address + (binary_offset // (memory_width // 8))) & 0xFFFF0000
|
||||
continue
|
||||
|
||||
# 构造记录
|
||||
hex_record = f':{record_len:02X}{record_addr:04X}{record_type:02X}'
|
||||
hex_record += ''.join(f'{byte:02X}' for byte in record_data)
|
||||
hex_record += f'{calculate_checksum(hex_record[1:]):02X}\n'
|
||||
result += hex_record
|
||||
|
||||
binary_offset += chunk_size
|
||||
|
||||
# 添加结束记录
|
||||
hex_record = f':00000001FF\n'
|
||||
result += hex_record
|
||||
|
||||
return result
|
||||
Reference in New Issue
Block a user