aboutsummaryrefslogtreecommitdiff
path: root/projects/cli-test/filetransfer
diff options
context:
space:
mode:
authorFredrik Thulin <fredrik@thulin.net>2016-05-15 10:42:23 +0200
committerFredrik Thulin <fredrik@thulin.net>2016-05-15 10:43:03 +0200
commit9b73356f2831800d2328827998e1e5b2a1994b68 (patch)
tree194357028a78b8ef92e4914074c2b3c51df5b17b /projects/cli-test/filetransfer
parentcc07ee6bfcf4721254145ef388196be82532fd4a (diff)
add simple filetransfer poc
Diffstat (limited to 'projects/cli-test/filetransfer')
-rwxr-xr-xprojects/cli-test/filetransfer89
1 files changed, 89 insertions, 0 deletions
diff --git a/projects/cli-test/filetransfer b/projects/cli-test/filetransfer
new file mode 100755
index 0000000..92b117f
--- /dev/null
+++ b/projects/cli-test/filetransfer
@@ -0,0 +1,89 @@
+#!/usr/bin/python
+
+import os
+import sys
+import time
+import struct
+import serial
+
+from binascii import crc32
+
+
+def _write(dst, data):
+ dst.write(data)
+ if len(data) == 4:
+ print("Wrote 0x{:02x}{:02x}{:02x}{:02x}".format(ord(data[0]), ord(data[1]), ord(data[2]), ord(data[3])))
+ else:
+ print("Wrote {!r}".format(data))
+
+
+def _read(dst):
+ res = ''
+ while True:
+ x = dst.read(1)
+ if not x:
+ break
+ res += x
+ print ("Read {!r}".format(res))
+ return res
+
+
+def _execute(dst, cmd):
+ _write(dst, '\r')
+ prompt = _read(dst)
+ if prompt.endswith('Username: '):
+ _write(dst, 'ct\r')
+ prompt = _read(dst)
+ if prompt.endswith('Password: '):
+ _write(dst, 'ct\r')
+ prompt = _read(dst)
+ if not prompt.endswith('> '):
+ sys.stderr.write('Device does not seem to be ready for a file transfer (got {!r})\n'.format(prompt))
+ return False
+ _write(dst, cmd + '\r')
+ response = _read(dst)
+ return response
+
+def send_file(filename, device='/dev/ttyUSB0', initiate=True):
+ s = os.stat(filename)
+ size = s.st_size
+ src = open(filename, 'rb')
+
+ dst = serial.Serial(device, 115200, timeout=0.5)
+
+ if initiate:
+ response = _execute(dst, 'filetransfer')
+ if 'OK' not in response:
+ sys.stderr.write('Device did not accept the filetransfer command (got {!r})\n'.format(response))
+ return False
+
+ # 1. Write size of file (4 bytes)
+ _write(dst, struct.pack('<I', size))
+ _read(dst)
+ # 2. Write file contents while calculating CRC-32
+ crc = 0
+ while True:
+ data = src.read(1024)
+ if not data:
+ break
+ dst.write(data)
+ print("Wrote {!s} bytes".format(len(data)))
+ crc = crc32(data, crc) & 0xffffffff
+ # 3. Write CRC-32 (4 bytes)
+ _write(dst, struct.pack('<I', crc))
+ _read(dst)
+
+ src.close()
+ dst.close()
+ return True
+
+
+if len(sys.argv) != 2:
+ sys.stderr.write('Syntax: {!s} filename\n'.format(sys.argv[0]))
+ sys.exit(1)
+
+if send_file(sys.argv[1]):
+ sys.exit(0)
+
+sys.exit(1)
+