nuttx-apps/system/ymodem/sbrb.py
anjiahao fd42d811c2 support ymodem transfer on a custom blocksize
Signed-off-by: anjiahao <anjiahao@xiaomi.com>
2023-04-29 21:23:55 +08:00

761 lines
19 KiB
Python
Executable File

#!/bin/python3
# apps/system/ymodem/sbrb.py
#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright ownership. The
# ASF licenses this file to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance with the
# License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#
import argparse
import binascii
import datetime
import io
import os
import signal
import sys
import termios
SOH = b"\x01" # Start of 128-byte data packet
STX = b"\x02" # Start of 1024-byte data packet
STC = b"\x03" # Start of a customsize data packet
EOT = b"\x04" # End of transmission
ACK = b"\x06" # Acknowledge
NAK = b"\x15" # Negative acknowledge
CAN = b"\x18" # Two of these in succession aborts transfer
CRC = b"\x43" # "C" == 0x43, request 16-bit CRC
PACKET_SIZE = 128
PACKET_1K_SIZE = 1024
EAGAIN = 1
EINVAL = 2
EEOT = 3
RETRIESMAX = 200
def format_time(seconds):
hours = seconds / 3600
seconds %= 3600
minutes = seconds / 60
seconds %= 60
time = "%02dh%02dm%02ds" % (int(hours), int(minutes), int(seconds))
return time
class Timeout(Exception):
pass
def timeout_handle(signum, frame):
sys.stderr.write("timeout!\n")
sys.stderr.flush()
raise Timeout("Timeout")
def ymodem_stdread(size):
fd = sys.stdin.fileno()
old_settings = termios.tcgetattr(fd)
signal.signal(signal.SIGALRM, timeout_handle)
signal.alarm(3)
try:
new_settings = termios.tcgetattr(fd)
new_settings[3] &= ~(termios.ICANON | termios.ECHO)
termios.tcsetattr(fd, termios.TCSADRAIN, new_settings)
sys.stdin.flush()
data = sys.stdin.buffer.read(size)
return data
except Timeout:
return
finally:
signal.alarm(0)
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
def ymodem_stdwrite(data):
fd = sys.stdout.fileno()
old_settings = termios.tcgetattr(fd)
try:
new_settings = termios.tcgetattr(fd)
new_settings[3] &= ~(termios.ICANON | termios.ECHO)
termios.tcsetattr(fd, termios.TCSADRAIN, new_settings)
data = sys.stdout.buffer.write(data)
sys.stdout.flush()
return data
finally:
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
def ymodem_stdprogress(data):
sys.stderr.write(data)
sys.stderr.flush()
def calc_crc16(data, crc=0):
crctable = [
0x0000,
0x1021,
0x2042,
0x3063,
0x4084,
0x50A5,
0x60C6,
0x70E7,
0x8108,
0x9129,
0xA14A,
0xB16B,
0xC18C,
0xD1AD,
0xE1CE,
0xF1EF,
0x1231,
0x0210,
0x3273,
0x2252,
0x52B5,
0x4294,
0x72F7,
0x62D6,
0x9339,
0x8318,
0xB37B,
0xA35A,
0xD3BD,
0xC39C,
0xF3FF,
0xE3DE,
0x2462,
0x3443,
0x0420,
0x1401,
0x64E6,
0x74C7,
0x44A4,
0x5485,
0xA56A,
0xB54B,
0x8528,
0x9509,
0xE5EE,
0xF5CF,
0xC5AC,
0xD58D,
0x3653,
0x2672,
0x1611,
0x0630,
0x76D7,
0x66F6,
0x5695,
0x46B4,
0xB75B,
0xA77A,
0x9719,
0x8738,
0xF7DF,
0xE7FE,
0xD79D,
0xC7BC,
0x48C4,
0x58E5,
0x6886,
0x78A7,
0x0840,
0x1861,
0x2802,
0x3823,
0xC9CC,
0xD9ED,
0xE98E,
0xF9AF,
0x8948,
0x9969,
0xA90A,
0xB92B,
0x5AF5,
0x4AD4,
0x7AB7,
0x6A96,
0x1A71,
0x0A50,
0x3A33,
0x2A12,
0xDBFD,
0xCBDC,
0xFBBF,
0xEB9E,
0x9B79,
0x8B58,
0xBB3B,
0xAB1A,
0x6CA6,
0x7C87,
0x4CE4,
0x5CC5,
0x2C22,
0x3C03,
0x0C60,
0x1C41,
0xEDAE,
0xFD8F,
0xCDEC,
0xDDCD,
0xAD2A,
0xBD0B,
0x8D68,
0x9D49,
0x7E97,
0x6EB6,
0x5ED5,
0x4EF4,
0x3E13,
0x2E32,
0x1E51,
0x0E70,
0xFF9F,
0xEFBE,
0xDFDD,
0xCFFC,
0xBF1B,
0xAF3A,
0x9F59,
0x8F78,
0x9188,
0x81A9,
0xB1CA,
0xA1EB,
0xD10C,
0xC12D,
0xF14E,
0xE16F,
0x1080,
0x00A1,
0x30C2,
0x20E3,
0x5004,
0x4025,
0x7046,
0x6067,
0x83B9,
0x9398,
0xA3FB,
0xB3DA,
0xC33D,
0xD31C,
0xE37F,
0xF35E,
0x02B1,
0x1290,
0x22F3,
0x32D2,
0x4235,
0x5214,
0x6277,
0x7256,
0xB5EA,
0xA5CB,
0x95A8,
0x8589,
0xF56E,
0xE54F,
0xD52C,
0xC50D,
0x34E2,
0x24C3,
0x14A0,
0x0481,
0x7466,
0x6447,
0x5424,
0x4405,
0xA7DB,
0xB7FA,
0x8799,
0x97B8,
0xE75F,
0xF77E,
0xC71D,
0xD73C,
0x26D3,
0x36F2,
0x0691,
0x16B0,
0x6657,
0x7676,
0x4615,
0x5634,
0xD94C,
0xC96D,
0xF90E,
0xE92F,
0x99C8,
0x89E9,
0xB98A,
0xA9AB,
0x5844,
0x4865,
0x7806,
0x6827,
0x18C0,
0x08E1,
0x3882,
0x28A3,
0xCB7D,
0xDB5C,
0xEB3F,
0xFB1E,
0x8BF9,
0x9BD8,
0xABBB,
0xBB9A,
0x4A75,
0x5A54,
0x6A37,
0x7A16,
0x0AF1,
0x1AD0,
0x2AB3,
0x3A92,
0xFD2E,
0xED0F,
0xDD6C,
0xCD4D,
0xBDAA,
0xAD8B,
0x9DE8,
0x8DC9,
0x7C26,
0x6C07,
0x5C64,
0x4C45,
0x3CA2,
0x2C83,
0x1CE0,
0x0CC1,
0xEF1F,
0xFF3E,
0xCF5D,
0xDF7C,
0xAF9B,
0xBFBA,
0x8FD9,
0x9FF8,
0x6E17,
0x7E36,
0x4E55,
0x5E74,
0x2E93,
0x3EB2,
0x0ED1,
0x1EF0,
]
crc = 0x0
for char in bytearray(data):
crctbl_idx = ((crc >> 8) ^ char) & 0xFF
crc = ((crc << 8) ^ crctable[crctbl_idx]) & 0xFFFF
return crc & 0xFFFF
class ymodem:
def __init__(
self,
read=ymodem_stdread,
write=ymodem_stdwrite,
progress=ymodem_stdprogress,
timeout=100,
debug="",
customsize=0,
):
self.read = read
self.write = write
self.timeout = timeout
self.progress = progress
self.customsize = customsize
if debug != "":
self.debugfd = open(debug, "w+")
else:
self.debugfd = 0
def debug(self, data):
if isinstance(self.debugfd, io.RawIOBase):
self.debugfd.write(data)
self.debugfd.flush()
def init_pkt(self):
self.head = SOH
self.seq0 = b"\x00"
self.seq1 = b"\xff"
self.data = b""
self.crch = b""
self.crcl = b""
def send_pkt(self):
crc16 = calc_crc16(self.data)
self.crch = bytes([(crc16 >> 8) & 0xFF])
self.crcl = bytes([crc16 & 0xFF])
pkt = self.head + self.seq0 + self.seq1 + self.data + self.crch + self.crcl
self.write(pkt)
def add_seq(self):
seq0 = bytearray(self.seq0)
if seq0[0] == 0xFF:
seq0[0] = 0x00
else:
seq0[0] += 1
seq1 = bytearray(self.seq1)
if seq1[0] == 0x00:
seq1[0] = 0xFF
else:
seq1[0] -= 1
self.seq0 = bytes(seq0)
self.seq1 = bytes(seq1)
def get_pkt_size(self):
if self.head == SOH:
return PACKET_SIZE
elif self.head == STX:
return PACKET_1K_SIZE
elif self.head == STC:
return self.customsize
return PACKET_SIZE
def recv_cmd(self, cmd):
chunk = self.read(1)
if chunk == NAK:
return -EAGAIN
if chunk != cmd:
self.debug("should be " + binascii.hexlify(cmd).decode("utf-8"))
self.debug("but receive " + binascii.hexlify(chunk).decode("utf-8") + "\n")
return -EINVAL
return 0
def send(self, filelist):
retries = 0
need_sendfile_num = len(filelist)
cnt = 0
now = datetime.datetime.now()
base = float(int(now.timestamp() * 1000)) / 1000
totolbytes = 0
while need_sendfile_num != 0:
now = datetime.datetime.now()
start = float(int(now.timestamp() * 1000)) / 1000
while retries < 10:
self.write(CRC)
chunk = self.read(1)
if chunk == CRC:
break
else:
retries += 1
if retries == 10:
return False
self.init_pkt()
self.head = SOH
filename = os.path.basename(filelist[cnt])
self.progress("name:" + filename)
self.data = filename.encode("utf-8")
self.data = self.data + bytes([0x00] * 1)
filesize = os.path.getsize(filelist[cnt])
sendfilesize = 0
self.progress(" filesize:%d\n" % (filesize))
self.data = self.data + str(filesize).encode("utf-8")
self.data = self.data.ljust(self.get_pkt_size(), b"\x00")
self.send_pkt()
ret = self.recv_cmd(ACK)
if ret == -EAGAIN:
continue
elif ret == -EINVAL:
return ret
ret = self.recv_cmd(CRC)
if ret == -EAGAIN:
continue
elif ret == -EINVAL:
return ret
self.add_seq()
readfd = open(filelist[cnt], "rb")
self.progress(" ")
while sendfilesize < filesize:
if sendfilesize + 128 >= filesize:
self.head = SOH
elif self.customsize != 0:
self.head = STC
else:
self.head = STX
self.data = readfd.read(self.get_pkt_size())
sendbytes = len(self.data)
self.data = self.data.ljust(self.get_pkt_size(), b"\x00")
self.send_pkt()
ret = self.recv_cmd(ACK)
if ret == -EAGAIN:
continue
elif ret == -EINVAL:
return ret
self.add_seq()
sendfilesize += sendbytes
totolbytes += sendbytes
self.progress("\r")
self.progress("%2.1f%%" % (float(sendfilesize) / filesize * 100))
self.progress(" %d:%d" % (sendfilesize, filesize))
now = datetime.datetime.now()
usedtime = float(int(now.timestamp() * 1000) / 1000) - start
realspeed = sendfilesize / 1024 / usedtime
left = (filesize - sendfilesize) / 1024 / (realspeed)
self.progress(" left:" + format_time(left))
retries = 0
while True:
if retries == 2:
return -EINVAL
retries += 1
self.write(EOT)
ret = self.recv_cmd(ACK)
if ret == -EAGAIN:
continue
elif ret < 0:
return ret
ret = self.recv_cmd(CRC)
if ret == -EAGAIN:
continue
elif ret < 0:
return ret
break
need_sendfile_num = need_sendfile_num - 1
readfd.close()
now = datetime.datetime.now()
time = float(int(now.timestamp() * 1000) / 1000)
time = time - start
self.progress("\ntime used:%.1fs" % time)
self.progress(" speed %.1fkB/s\n" % (float(sendfilesize) / 1024 / time))
if need_sendfile_num != 0:
cnt += 1
continue
retries = 0
while True:
if retries == 2:
return -EINVAL
retries += 1
self.head = SOH
self.seq0 = b"\x00"
self.seq1 = b"\xff"
self.data = bytes([0x00] * self.get_pkt_size())
self.send_pkt()
ret = self.recv_cmd(ACK)
if ret == -EAGAIN:
continue
elif ret < 0:
return ret
break
now = datetime.datetime.now()
time = float(int(now.timestamp() * 1000) / 1000)
totaltime = time - base
arvgspeed = float(totolbytes) / 1024 / totaltime
self.progress(
"\n all time:%.2fs average speed:%.2fkB/s" % (totaltime, arvgspeed)
)
def recv_packet(self):
chunk = self.read(1)
if chunk == SOH:
self.packetsize = PACKET_SIZE
elif chunk == STX:
self.packetsize = PACKET_1K_SIZE
elif chunk == STC:
self.packetsize = self.customsize
elif chunk == NAK:
return -EAGAIN
elif chunk == EOT:
return -EEOT
else:
self.debug("recv EBADMSG" + str(chunk) + "\n")
return -EINVAL
seq0 = self.read(1)
seq1 = self.read(1)
self.data = self.read(self.packetsize)
crch = self.read(1)
crcl = self.read(1)
if seq0 != self.seq0:
self.debug("recv bad seq0" + binascii.hexlify(seq0).decode("utf-8") + "\n")
return -EINVAL
if seq1 != self.seq1:
self.debug("recv bad seq1" + binascii.hexlify(seq1).decode("utf-8") + "\n")
return -EINVAL
crc16 = calc_crc16(self.data)
crch_b = (crc16 >> 8).to_bytes(1, byteorder="little")
if crch != crch_b:
self.debug(
"recv bad crch_b" + binascii.hexlify(crch_b).decode("utf-8") + "\n"
)
self.debug("recv bad crch" + binascii.hexlify(crch).decode("utf-8") + "\n")
return -EINVAL
crcl_b = (crc16 & 0xFF).to_bytes(1, byteorder="little")
if crcl != crcl_b:
self.debug(
"recv bad crcl_b" + binascii.hexlify(crcl_b).decode("utf-8") + "\n"
)
self.debug("recv bad crcl" + binascii.hexlify(crcl).decode("utf-8") + "\n")
return -EINVAL
self.add_seq()
return 0
def recv(self):
retries = 0
start_recv = False
now = datetime.datetime.now()
base = float(int(now.timestamp() * 1000)) / 1000
totolbytes = 0
while True:
self.write(CRC)
now = datetime.datetime.now()
start = float(int(now.timestamp() * 1000)) / 1000
self.init_pkt()
ret = self.recv_packet()
if ret < 0:
if retries > RETRIESMAX:
return -1
retries += 1
continue
self.debug("recv frist packet\n")
filename = bytes.decode(self.data.split(b"\x00")[0], "utf-8")
if not filename:
if start_recv:
self.debug("recv last packet\n")
break
self.debug("recv a none file\n")
retries += 1
continue
start_recv = True
self.progress("name:" + filename + " ")
size_str = bytes.decode(self.data.split(b"\x00")[1], "utf-8")
filesize = int(size_str)
self.progress("size:%d" % (filesize) + "\n")
self.write(ACK)
self.write(CRC)
fd = open(filename, "wb+")
writensize = 0
while writensize < filesize:
ret = self.recv_packet()
if ret < 0:
self.debug("recv a bad data packet\n")
if retries > RETRIESMAX:
return -1
retries += 1
continue
size = 0
if self.packetsize > filesize - writensize:
self.debug("last data packet\n")
size = self.packetsize - (filesize - writensize)
else:
size = self.packetsize
data = self.data[0:size]
fd.write(data)
writensize += size
self.progress("\r%.2f%%" % (float(writensize) / filesize * 100))
self.progress(" %d:%d" % (writensize, filesize))
now = datetime.datetime.now()
usedtime = float(int(now.timestamp() * 1000) / 1000) - start
realspeed = writensize / 1024 / usedtime
left = (filesize - writensize) / 1024 / (realspeed)
self.progress(" left:" + format_time(left))
self.write(ACK)
ret = self.recv_packet()
if ret == -EEOT:
self.debug("recv EOT cmd\n")
elif ret < 0:
self.debug("recv error packet")
return -EINVAL
self.write(ACK)
self.write(CRC)
now = datetime.datetime.now()
time = float(now.timestamp() * 1000) / 1000
time = time - start
self.progress("\ntime used:%.1fs" % time)
self.progress(" speed %.1fkB/s\n" % (float(filesize) / 1024 / time))
totolbytes += filesize
fd.close()
now = datetime.datetime.now()
time = float(int(now.timestamp() * 1000) / 1000)
totaltime = time - base
arvgspeed = float(totolbytes) / 1024 / totaltime
self.progress(
"\n all time:%.2fs average speed:%.2fkB/s" % (totaltime, arvgspeed)
)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument(
"filelist", help="if filelist is valid, that is sb, else is rb", nargs="*"
)
parser.add_argument(
"-k",
"--kblocksize",
help="This opthin can set a customsize block size to transfer",
type=int,
default=0,
)
parser.add_argument(
"--debug", help="This opthin is save debug log on host", default=""
)
args = parser.parse_args()
sbrb = ymodem(debug=args.debug, customsize=args.kblocksize * 1024)
if len(args.filelist) == 0:
sbrb.progress("receiving\n")
sbrb.recv()
else:
sbrb.progress("sending\n")
sbrb.send(args.filelist)