forked from HorstBaerbel/ubootwrite
-
Notifications
You must be signed in to change notification settings - Fork 4
/
ubootwrite.py
173 lines (148 loc) · 7.61 KB
/
ubootwrite.py
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
# -*- coding: utf-8 -*-
from __future__ import division #1/2 = float, 1//2 = integer, python 3.0 behaviour in 2.6, to make future port to 3 easier.
from __future__ import print_function
from optparse import OptionParser
import os
import struct
import sys
import zlib
import time
debug = False
if not debug:
import serial
# The maximum size to transfer if we can determinate the size of the file (if input data comes from stdin).
MAX_SIZE = 2 ** 30
LINE_FEED = "\n"
# Wait for the prompt
def getprompt(ser, addr, verbose):
# Send a command who does not produce a result so when receiving the next line feed, only show the prompt will be returned
ser.write("mw {0:08x} 0".format(addr) + LINE_FEED)
# Flushing read buffer
while ser.read(256):
pass
if verbose:
print("Waiting for a prompt...", file = sys.stderr)
while True:
# Write carriage return and wait for a response
ser.write(LINE_FEED)
# Read the response
buf = ser.read(256);
if (buf.endswith(b"> ") or buf.endswith(b"# ")):
print("Prompt is '", buf[2:], "'", sep='', file = sys.stderr)
# The prompt returned starts with a line feed. This is the echo of the line feed we send to get the prompt.
# We keep this linefeed
return buf
else:
# Flush read buffer
while ser.read(256):
pass
# Wait for the prompt and return True if received or False otherwise
def writecommand(ser, command, prompt, verbose):
# Write the command and a line feed, so we must get back the command and the prompt
ser.write(command + LINE_FEED)
buf = ser.read(len(command))
if (buf != command):
if verbose:
print("Echo command not received. Instead received '", buf, "'", sep='', file = sys.stderr)
return False
if verbose:
print("Waiting for prompt...", file = sys.stderr)
buf = ser.read(len(prompt))
if (buf == prompt):
if verbose:
print("Ok, prompt received", file = sys.stderr)
return True
else:
if verbose:
print("Prompt not received. Instead received '", buf, "'", sep='', file = sys.stderr)
return False
def memwrite(ser, path, size, start_addr, verbose, debug):
if not debug:
prompt = getprompt(ser, start_addr, verbose)
if (path == "-"):
fd = sys.stdin
if (size <= 0):
size = MAX_SIZE
else:
fd = open(path,"rb")
if (size <= 0):
# Get the size of the file
fd.seek(0, os.SEEK_END);
size = fd.tell();
fd.seek(0, os.SEEK_SET);
addr = start_addr
bytes_read = 0
crc32_checksum = 0
startTime = time.time();
bytesLastSecond = 0
while (bytes_read < size):
if ((size - bytes_read) > 4):
read_bytes = fd.read(4);
else:
read_bytes = fd.read(size - bytes_read);
if (len(read_bytes) == 0):
if (path == "-"):
size = bytes_read
break
bytesLastSecond += len(read_bytes)
bytes_read += len(read_bytes)
crc32_checksum = zlib.crc32(read_bytes, crc32_checksum) & 0xFFFFFFFF
while (len(read_bytes) < 4):
read_bytes += b'\x00'
(val, ) = struct.unpack(">L", read_bytes)
read_bytes = "".format(val)
str_to_write = "mw {0:08x} {1:08x}".format(addr, val)
if verbose:
print("Writing:", str_to_write, "at:", "0x{0:08x}".format(addr), file = sys.stderr)
if debug:
str_to_write = struct.pack(">L", int("{0:08x}".format(val), 16))
else:
if not writecommand(ser, str_to_write, prompt, verbose):
print("Found an error, so aborting", file = sys.stderr)
fd.close()
return
# Print progress
currentTime = time.time();
if ((currentTime - startTime) > 1):
print("\rProgress {:2.1f}%".format((bytes_read * 100) / size), end="", file = sys.stderr)
print(", {:3.1f}kb/s".format(bytesLastSecond / (currentTime - startTime) / 1024), end="", file = sys.stderr)
print(", ETA {0}s ".format(round((size - bytes_read) / bytesLastSecond / (currentTime - startTime))), end="", file = sys.stderr)
bytesLastSecond = 0
startTime = time.time();
# Increment address
addr += 4
if (bytes_read != size):
print("Error while reading file '", fd.name, "' at offset ", bytes_read, sep='', file = sys.stderr)
else:
print("\rProgress 100% ", file = sys.stderr)
#not possible for zsun print("File successfully written. You should run 'crc32", " {0:08x}".format(start_addr), " {0:08x}".format(bytes_read), "' on the modem and the result must be", " {0:08x}".format(crc32_checksum), ".", sep='', file = sys.stderr)
#not needed for zsun print("To copy from RAM to flash, unprotect flash: 'protect off all'...")
print("Then erase flash: 'erase 9f020000 " +{0:08x}".format(bytes_read), "'.", sep='', file = sys.stderr)
print("Then copy from RAM to flash: 'cp.b 80060000 9f020000 " {0:08x}".format(bytes_read), "'.", sep='', file = sys.stderr)
fd.close()
return
def main():
optparser = OptionParser("usage: %prog [options]", version = "%prog 0.2")
optparser.add_option("--verbose", action = "store_true", dest = "verbose", help = "be verbose", default = False)
optparser.add_option("--serial", dest = "serial", help = "specify serial port", default = "/dev/ttyUSB0", metavar = "dev")
optparser.add_option("--write", dest = "write", help = "write mem from file", metavar = "path")
optparser.add_option("--addr", dest = "addr", help = "mem address", default = "0x80060000", metavar = "addr")
optparser.add_option("--size", dest = "size", help = "# bytes to write", default = "0", metavar = "size")
(options, args) = optparser.parse_args()
if (len(args) != 0):
optparser.error("incorrect number of arguments")
if not debug:
ser = serial.Serial(options.serial, 115200, timeout=0.1)
else:
ser = open(options.write + ".out", "wb")
if debug:
prompt = getprompt(ser, options.verbose)
writecommand(ser, "mw 80600000 01234567", prompt, options.verbose)
buf = ser.read(256)
print("buf = '", buf, "'", sep = "")
return
if options.write:
memwrite(ser, options.write, int(options.size, 0), int(options.addr, 0), options.verbose, debug)
return
if __name__ == '__main__':
main()