Add tools/upload.py

tools/upload.py

@@ -0,0 +1,59 @@
+#!/usr/bin/python
+
+# Push specified file (max 0xff bytes) to specified device and verify that
+# the contents is correct by sending a "peek" command afterwards and check
+# the output. Errors out if the contents isn't the same. The parameter
+# passed to the "peek" command is the length of the uploaded file.
+
+import argparse
+import os
+import sys
+
+def sendcmd(fd, cmd):
+    # The serial link echoes back all typed characters and expects us to read
+    # them. We have to send each char one at a time.
+    print("Executing {}".format(cmd.decode()))
+    for c in cmd:
+        os.write(fd, bytes([c]))
+        os.read(fd, 1)
+    os.write(fd, b'\n')
+    os.read(fd, 2)  # sends back \r\n
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('device')
+    parser.add_argument('filename')
+    args = parser.parse_args()
+
+    st = os.stat(args.filename)
+    if st.st_size > 0xff:
+        print("File too big. 0xff bytes max")
+        return 1
+    fd = os.open(args.device, os.O_RDWR)
+    sendcmd(fd, 'load {:x}'.format(st.st_size).encode())
+    print("Loading...")
+    with open(args.filename, 'rb') as fp:
+        fcontents = fp.read()
+        os.write(fd, fcontents)
+    print("Loaded")
+    os.read(fd, 5)
+    print("Peeking back...")
+    sendcmd(fd, 'peek {:x}'.format(st.st_size).encode())
+    peek = b''
+    while len(peek) < st.st_size * 2:
+        peek += os.read(fd, 1)
+    os.close(fd)
+    print("Got {}".format(peek.decode()))
+    print("Comparing...")
+    for i, c in enumerate(fcontents):
+        hexfmt = '{:02X}'.format(c).encode()
+        if hexfmt != peek[:2]:
+            print("Mismatch at byte {}! {} != {}".format(i, peek[:2], hexfmt))
+            return 1
+        peek = peek[2:]
+    print("All good!")
+    return 0
+
+if __name__ == '__main__':
+    sys.exit(main())