Polish mrbridge and add ftpserve

ftpserve.py

@@ -0,0 +1,79 @@
+import argparse
+import logging
+import os
+import serial
+
+
+kCrc16Polynomial = 0x1021
+kCrc16Preset = 0xFFFF
+
+kDefaultSerialPort = None
+kDefaultBaudrate = 115200
+kDefaultChunkSize = 0x20
+
+logger = logging.getLogger(__name__)
+
+
+def crc16(data):
+    crc = kCrc16Preset
+    for c in data:
+        crc ^= c << 8
+        for j in range(8):
+            mix = crc & 0x8000
+            crc <<= 1
+            if mix:
+                crc ^= kCrc16Polynomial
+            crc &= 0xFFFF
+    return crc
+
+
+def getserial(port=None, baudrate=None, **kwargs):
+    if port is not None:
+        return serial.Serial(port, baudrate)
+
+    # return first /dev/ttyACMx found
+    for i in range(10):
+        try:
+            return serial.Serial(f"/dev/ttyACM{i}", baudrate)
+        except serial.SerialException:
+            pass
+    raise RuntimeError("No serial device available.")
+
+
+def ftpserve(file, device, chunksize=0x20):
+    with open(file, "rb") as f:
+        data = f.read()
+    device.reset_input_buffer()
+    logger.info(f"Serving file {file}")
+    for i in range(0, len(data), chunksize):
+        device.write(data[i : i + chunksize])
+        r = device.read()
+        if r != b"\x42":
+            raise RuntimeError(f"wrong response {r}")
+        logger.info(f"ack for chunk @{i}")
+    logger.info(f"Done serving file {file}")
+    crc = crc16(data)
+    logger.info(f"File length: {len(data):04x}, crc16: {crc:04x}")
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(description="Serve a file to ftpget.")
+    parser.add_argument("file", help="file to serve")
+    parser.add_argument("--port", help="path to the serial port")
+    parser.add_argument("--baudrate", default=kDefaultBaudrate)
+    parser.add_argument("--chunksize", default=kDefaultChunkSize)
+    return parser.parse_args()
+
+
+def main():
+    logging.basicConfig(level=logging.INFO)
+    args = parse_args()
+    device = getserial(**vars(args))
+    ftpserve(args.file, device, args.chunksize)
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except KeyboardInterrupt:
+        pass

mrbridge.py

@@ -4,12 +4,29 @@ import threading
 import time
 import websocket
 
+kDefaultSerialPort = None
+kDefaultBaudrate = 115200
+
+
+def getserial(port=None, baudrate=None, **kwargs):
+    if port is not None:
+        return serial.Serial(port, baudrate)
+
+    # return first /dev/ttyACMx found
+    for i in range(10):
+        try:
+            return serial.Serial(f"/dev/ttyACM{i}", baudrate)
+        except serial.SerialException:
+            pass
+    raise RuntimeError("No serial device available.")
+
 
 def parse_args():
     parser = argparse.ArgumentParser(description="Bridge a serial port to a websocket")
-    parser.add_argument("port", help="path to the serial port")
-    parser.add_argument("--baudrate", default=115200)
-    parser.add_argument("--escape", action='store_true')
+    parser.add_argument(
+        "--port", help="path to the serial port", default=kDefaultSerialPort
+    )
+    parser.add_argument("--baudrate", default=kDefaultBaudrate)
     parser.add_argument("ws", help="URL of the websocket")
     return parser.parse_args()
 
@@ -28,28 +45,23 @@ def slowwrite(device, data):
 
 def ws_thread(device, ws):
     while True:
-        data = ws.recv();
-        slowwrite(device, data.replace(b'\n', b'\r\n'))
+        data = ws.recv()
+        slowwrite(device, data.replace(b"\n", b"\r\n"))
 
 
 def main():
     args = parse_args()
-    device = serial.Serial(args.port, baudrate=args.baudrate)
+    device = getserial(**vars(args))
     ws = websocket.create_connection(args.ws)
 
     wst = threading.Thread(target=ws_thread, args=(device, ws), daemon=True)
     wst.start()
 
-    if args.escape:
-        device.write([0x03])
-    try:
     device_thread(device, ws)
-    except KeyboardInterrupt:
-        pass
-    finally:
-        if args.escape:
-            device.write([0x02])
 
 
 if __name__ == "__main__":
+    try:
         main()
+    except KeyboardInterrupt:
+        pass