2023-05-17 19:43:24 +00:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
|
|
from binascii import crc_hqx
|
2023-06-10 16:49:58 +00:00
|
|
|
from itertools import cycle
|
|
|
|
from sys import stderr, argv
|
|
|
|
from pathlib import Path
|
2023-05-17 19:43:24 +00:00
|
|
|
from time import time
|
2023-06-10 16:49:58 +00:00
|
|
|
from io import StringIO
|
2023-05-17 19:43:24 +00:00
|
|
|
|
|
|
|
from serial import Serial
|
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
DATA_DIR = Path(__file__).parent / 'data'
|
2023-06-10 05:59:55 +00:00
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
KEY_FW = (DATA_DIR / 'key-fw.bin').read_bytes()
|
|
|
|
KEY_COMM = (DATA_DIR / 'key-comm.bin').read_bytes()
|
2023-05-18 18:11:36 +00:00
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
V_START = 8192
|
|
|
|
V_END = V_START + 16
|
|
|
|
CRC_LEN = 2
|
2023-05-17 19:43:24 +00:00
|
|
|
|
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
def chunk(data, n):
|
|
|
|
for i in range(0,len(data), n):
|
|
|
|
yield data[i:i+n]
|
|
|
|
|
|
|
|
|
|
|
|
def eprint(*args, **kwargs):
|
|
|
|
print(*args, **kwargs, file=stderr)
|
|
|
|
|
|
|
|
|
2023-06-10 17:07:35 +00:00
|
|
|
def xor_fw(var):
|
2023-06-10 16:49:58 +00:00
|
|
|
return bytes(a ^ b for a, b in zip(var, cycle(KEY_FW)))
|
2023-05-17 19:43:24 +00:00
|
|
|
|
2023-05-18 18:11:36 +00:00
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
def xor_comm(var):
|
|
|
|
return bytes(a ^ b for a, b in zip(var, cycle(KEY_COMM)))
|
|
|
|
|
|
|
|
|
|
|
|
def make_16byte_version(version):
|
|
|
|
return bytes([ord(c) for c in version] + [0] * (16 - len(version)))
|
|
|
|
|
|
|
|
|
|
|
|
def i2b16(val):
|
|
|
|
return int(val).to_bytes(2,'little')
|
2023-05-17 21:17:24 +00:00
|
|
|
|
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
def i2b32(val):
|
|
|
|
return int(val).to_bytes(4,'little')
|
2023-05-17 21:17:24 +00:00
|
|
|
|
|
|
|
|
|
|
|
def b2i(data):
|
|
|
|
return int.from_bytes(data, 'little')
|
|
|
|
|
|
|
|
|
|
|
|
def len16(data):
|
|
|
|
return i2b16(len(data))
|
|
|
|
|
|
|
|
|
|
|
|
def crc16(data):
|
|
|
|
return i2b16(crc_hqx(data, 0))
|
|
|
|
|
2023-05-18 19:11:19 +00:00
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
def decrypt(data):
|
2023-06-10 17:07:35 +00:00
|
|
|
decrypted = xor_fw(data)
|
2023-06-10 16:49:58 +00:00
|
|
|
version = decrypted[V_START:V_END].decode().rstrip('\x00')
|
|
|
|
return (decrypted[:V_START] + decrypted[V_END:-CRC_LEN], version)
|
2023-05-17 21:17:24 +00:00
|
|
|
|
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
def encrypt(data, version='2.01.26'):
|
|
|
|
v = make_16byte_version(version)
|
2023-06-10 17:07:35 +00:00
|
|
|
encrypted = xor_fw(data[:V_START] + v + data[V_START:])
|
2023-06-10 16:49:58 +00:00
|
|
|
checksum = crc16(encrypted)
|
|
|
|
return encrypted + checksum
|
2023-05-17 21:17:24 +00:00
|
|
|
|
|
|
|
|
2023-05-17 20:26:21 +00:00
|
|
|
class UVK5(Serial):
|
2023-06-10 16:49:58 +00:00
|
|
|
BLOCK_SIZE = 0x80
|
|
|
|
|
|
|
|
PREAMBLE = b'\xab\xcd'
|
|
|
|
POSTAMBLE = b'\xdc\xba'
|
|
|
|
|
|
|
|
CMD_VERSION_REQ = 0x0514
|
|
|
|
CMD_VERSION_RES = 0x0515
|
|
|
|
CMD_SETTINGS_REQ = 0x051B
|
|
|
|
CMD_SETTINGS_RES = 0x051C
|
|
|
|
|
|
|
|
CMD_SETTINGS_WRITE_REQ = 0x051D # then addr (0x0E70) then size (0x0160) then data
|
|
|
|
|
2023-05-17 20:26:21 +00:00
|
|
|
def __init__(self, port: str | None = None) -> None:
|
2023-06-10 16:49:58 +00:00
|
|
|
self.timestamp = i2b32(time())
|
2023-05-17 20:26:21 +00:00
|
|
|
super().__init__(port, 38400, timeout=5)
|
2023-05-17 19:43:24 +00:00
|
|
|
|
2023-06-10 05:59:55 +00:00
|
|
|
def get_version(self):
|
2023-06-10 16:49:58 +00:00
|
|
|
return self.cmd(UVK5.CMD_VERSION_REQ)[1][:10].decode().rstrip('\x00')
|
2023-06-10 05:59:55 +00:00
|
|
|
|
2023-05-18 18:11:36 +00:00
|
|
|
def read_mem(self, offset, size):
|
2023-06-10 16:49:58 +00:00
|
|
|
return self.cmd(UVK5.CMD_SETTINGS_REQ, i2b16(offset) + i2b16(size))
|
2023-05-17 19:43:24 +00:00
|
|
|
|
2023-05-17 20:26:21 +00:00
|
|
|
def cmd(self, id, body = b''):
|
2023-06-10 16:49:58 +00:00
|
|
|
self.write(self._cmd_make_req(id, body))
|
2023-05-17 20:26:21 +00:00
|
|
|
preamble = self.read(2)
|
2023-05-17 19:43:24 +00:00
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
if preamble != UVK5.PREAMBLE:
|
2023-06-10 16:12:50 +00:00
|
|
|
raise ValueError('Bad response (PRE)', preamble)
|
2023-05-17 19:43:24 +00:00
|
|
|
|
2023-05-18 18:11:36 +00:00
|
|
|
payload_len = b2i(self.read(2)) + 2 # CRC len
|
2023-06-10 06:05:27 +00:00
|
|
|
payload = xor_comm(self.read(payload_len))
|
2023-05-17 19:43:24 +00:00
|
|
|
|
2023-05-18 18:11:36 +00:00
|
|
|
# crc = payload[-2:]
|
2023-05-17 20:26:21 +00:00
|
|
|
postamble = self.read(2)
|
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
if postamble != UVK5.POSTAMBLE:
|
2023-05-18 18:11:36 +00:00
|
|
|
raise ValueError('Bad response (POST)')
|
2023-05-17 20:26:21 +00:00
|
|
|
|
|
|
|
# print(data.hex())
|
2023-05-18 18:11:36 +00:00
|
|
|
cmd_id = b2i(payload[:2])
|
|
|
|
data_len = b2i(payload[2:4])
|
|
|
|
data = payload[4:4+data_len]
|
2023-05-17 19:43:24 +00:00
|
|
|
|
2023-05-18 18:11:36 +00:00
|
|
|
return (cmd_id, data)
|
2023-05-17 19:43:24 +00:00
|
|
|
|
2023-06-10 06:05:27 +00:00
|
|
|
def channels(self):
|
2023-06-10 05:59:55 +00:00
|
|
|
names = []
|
|
|
|
settings = []
|
2023-05-18 19:11:19 +00:00
|
|
|
|
2023-06-10 05:59:55 +00:00
|
|
|
data_size = 16 * 200
|
|
|
|
names_offset = 0x0F50
|
|
|
|
settings_offset = 0x0000
|
2023-05-18 19:11:19 +00:00
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
passes = data_size//UVK5.BLOCK_SIZE
|
|
|
|
|
|
|
|
out = StringIO()
|
2023-05-18 19:11:19 +00:00
|
|
|
|
2023-06-10 05:59:55 +00:00
|
|
|
for block in range(passes):
|
2023-06-10 16:49:58 +00:00
|
|
|
offset = names_offset + block*UVK5.BLOCK_SIZE
|
|
|
|
names_set = self.read_mem(offset, UVK5.BLOCK_SIZE)[1][4:]
|
2023-06-10 05:59:55 +00:00
|
|
|
names += [name.decode(errors='ignore').rstrip('\x00') for name in chunk(names_set, 16)]
|
2023-05-18 19:11:19 +00:00
|
|
|
|
2023-06-10 05:59:55 +00:00
|
|
|
for block in range(passes):
|
2023-06-10 16:49:58 +00:00
|
|
|
offset = settings_offset + block*UVK5.BLOCK_SIZE
|
|
|
|
settings_set = self.read_mem(offset, UVK5.BLOCK_SIZE)[1][4:]
|
2023-06-10 05:59:55 +00:00
|
|
|
settings += [(b2i(setting[:4])/100000.0, ) for setting in chunk(settings_set, 16)]
|
|
|
|
|
|
|
|
for i, name in enumerate(names):
|
|
|
|
if name:
|
2023-06-10 16:49:58 +00:00
|
|
|
print(f'{i+1:0>3}. {name: <16} {settings[i][0]:0<8} M', file=out)
|
2023-06-10 05:59:55 +00:00
|
|
|
else:
|
2023-06-10 16:49:58 +00:00
|
|
|
print(f'{i+1:0>3}. -', file=out)
|
2023-05-18 19:11:19 +00:00
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
return out.getvalue()
|
|
|
|
|
|
|
|
|
|
|
|
def _cmd_make_req(self, cmd_id, body=b''):
|
|
|
|
data = body + self.timestamp
|
|
|
|
payload = i2b16(cmd_id) + len16(data) + data
|
|
|
|
encoded_payload = xor_comm(payload + crc16(payload))
|
|
|
|
|
|
|
|
return UVK5.PREAMBLE + len16(payload) + encoded_payload + UVK5.POSTAMBLE
|
2023-06-10 16:12:50 +00:00
|
|
|
|
2023-05-17 19:43:24 +00:00
|
|
|
|
|
|
|
if __name__ == '__main__':
|
2023-06-10 05:59:55 +00:00
|
|
|
if len(argv) < 3:
|
2023-06-10 16:49:58 +00:00
|
|
|
eprint(f'Usage: {argv[0]} <port> <command:(channels|version)> [args]')
|
2023-06-10 05:59:55 +00:00
|
|
|
exit(255)
|
2023-06-10 16:49:58 +00:00
|
|
|
|
|
|
|
port = argv[1]
|
|
|
|
cmd = argv[2]
|
|
|
|
args = argv[3:]
|
|
|
|
|
|
|
|
with UVK5(port) as s:
|
2023-06-10 17:03:51 +00:00
|
|
|
version = s.get_version()
|
|
|
|
if cmd == 'version':
|
|
|
|
print('FW Version:', version)
|
|
|
|
exit(0)
|
|
|
|
|
2023-06-10 16:49:58 +00:00
|
|
|
print(getattr(s, cmd)(*args))
|
|
|
|
|