aboutsummaryrefslogtreecommitdiff
path: root/projects/cli-test/filetransfer
blob: 92b117f8d6a49f804d60e278651183b0e70e1d8a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
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)