import sys, time
from SerialPort_linux import *
+# serial device to communicate with
+DEVICE="/dev/ttyUSB0"
+# baudrate used for initialization
+INIT_BAUDRATE=9600
+# baudrate used for communication with the internal bootloader after init
+BOOTLOADER_BAUDRATE=38400
+# baudrate used for communication with the pkernel program that does the flashing eventually
+KERNEL_BAUDRATE=115200
+
# contains the last received checksum from a READ, WRITE or CHECKSUM command
last_checksum = 0
def sendByte(byte):
- time.sleep(0.01) # just to get sure, wait 10ms
tty.write(chr(byte))
- tty.flush()
def sendWord(word):
sendByte(word & 0xFF)
def recvChecksum():
global last_checksum
- # get checksum
last_checksum = recvByte()
last_checksum |= (recvByte() << 8)
-def cmdREAD(address, size):
+def bootromREAD(address, size):
# send READ command
sendByte(0x01)
if (recvByte() != 0xF1):
recvChecksum()
return data
-def cmdWRITE(address, size, data):
+def bootromWRITE(address, size, data):
# send WRITE command
sendByte(0x01)
if (recvByte() != 0xF1):
# get checksum
recvChecksum()
-def cmdBAUDRATE(baudrate):
- global last_checksum
+# TODO: test this function!
+def bootromCALL(address):
+ # send CALL command
+ sendByte(0x01)
+ if (recvByte() != 0xF1):
+ raise Exception
+ sendByte(0x04)
+ if (recvByte() != 0x84):
+ raise Exception
+ # tell desired address
+ sendDWord(address)
+ # wait for return parameter - not needed here!
+ #return recvByte()
+# TODO: test this function!
+def bootromCHECKSUM():
+ # call CHECKSUM command
+ sendByte(0x01)
+ if (recvByte() != 0xF1):
+ raise Exception
+ sendByte(0x05)
+ if (recvByte() != 0x84):
+ raise Exception
+ # get checksum
+ recvChecksum()
+
+def bootromBAUDRATE(baudrate):
# send BAUDRATE command
sendByte(0x01)
if (recvByte() != 0xF1):
if (recvByte() != 0x86):
raise Exception
# send desired baudrate
- sendByte(baudrate & 0xFF)
- sendByte((baudrate >> 8) & 0xFF)
- sendByte((baudrate >> 16) & 0xFF)
- sendByte((baudrate >> 24) & 0xFF)
+ sendDWord(baudrate)
+
+def pkernCHIPERASE():
+ sendByte(0x15)
+ if (recvByte() != 0x45):
+ raise Exception
+ # wait till completion...
+ if (recvByte() != 0x23):
+ raise Exception
+
+def pkernERASE(address, size):
+ sendByte(0x12)
+ if (recvByte() != 0x11):
+ raise Exception
+ sendDWord(address)
+ sendWord(size)
+ if (recvByte() != 0x18):
+ raise Exception
+
+
+def pkernWRITE(address, size, data):
+ # send WRITE command
+ sendByte(0x13)
+ if (recvByte() != 0x37):
+ raise Exception
+ # tell desired address and size
+ sendDWord(address)
+ sendWord(size)
+
+ # write binary stream of data
+ for i in range(0, size):
+ sendByte(data[i])
+
+ if (recvByte() != 0x28):
+ raise Exception
+
+
+class FlashSequence(object):
+ def __init__(self, address, data):
+ self.address = address
+ self.data = data
+
+def readMHXFile(filename): # desired mhx filename
+ fp = open(filename, "r")
+ retval = [] # returns a list of FlashSequence objects
+ linecount = 0
+ for line in fp:
+ linecount += 1
+ # get rid of newline characters
+ line = line.strip()
+
+ # we're only interested in S2 (data sequence with 3 address bytes) records by now
+ if line[0:2] == "S2":
+ byte_count = int(line[2:4], 16)
+ # just to get sure, check if byte count field is valid
+ if (len(line)-4) != (byte_count*2):
+ print sys.argv[0] + ": Warning - inavlid byte count field in " + \
+ sys.argv[1] + ":" + str(linecount) + ", skipping line!"
+ continue
+
+ # address and checksum bytes are not needed
+ byte_count -= 4
+ address = int(line[4:10], 16)
+ datastr = line[10:10+byte_count*2]
+
+ # convert data hex-byte-string to real byte data list
+ data = []
+ for i in range(0, len(datastr)/2):
+ data.append(int(datastr[2*i:2*i+2], 16))
+
+ # add flash sequence to our list
+ retval.append(FlashSequence(address, data))
+ fp.close()
+ return retval
+
+
+# check command line arguments
+if len(sys.argv) != 3:
+ print "Usage: " + sys.argv[0] + " [pkernel mhx-file] [target mhx-file]"
+ sys.exit(1)
+
+# read in data from mhx-files before starting
+try:
+ bootloaderseqs = readMHXFile(sys.argv[1])
+ pkernelseqs = readMHXFile(sys.argv[2])
+except IOError as error:
+ print sys.argv[0] + ": Error - couldn't open file " + error.filename + "!"
+ sys.exit(1)
print "Initializing serial port..."
-tty = SerialPort("/dev/ttyUSB0", 100, 9600)
+tty = SerialPort(DEVICE, 100, INIT_BAUDRATE)
-print "Please press RESET on your 1337 board..."
+print "Please press RESET on your board..."
-while 1:
+while True:
tty.write('V')
tty.flush()
try:
# timeout happened, who cares ;-)
pass
-print "OK, trying to set baudrate..."
+starttime = time.time() # save time at this point for evaluating the duration at the end
+print "OK, trying to set baudrate..."
# set baudrate
-cmdBAUDRATE(19200)
-tty = SerialPort("/dev/ttyUSB0", 100, 19200)
+bootromBAUDRATE(BOOTLOADER_BAUDRATE)
+time.sleep(0.1) # just to get sure that the bootloader is really running in new baudrate mode!
+del tty
+tty = SerialPort(DEVICE, 100, BOOTLOADER_BAUDRATE)
+print "Transfering pkernel program to IRAM",
+# let the fun begin!
+for seq in bootloaderseqs:
+ if(seq.address <= 0x40000):
+ addr = seq.address
+ else:
+ continue
+ #print "RAMing", len(seq.data), "bytes at address", hex(addr)
+ bootromWRITE(addr, len(seq.data), seq.data)
+ tty.flush()
+ sys.stdout.write(".")
+ sys.stdout.flush()
print
-# write something to the begin of the IRAM
-data = [255, 2, 3, 4]
-print "Writing", data, "to the IRAM..."
-cmdWRITE(0x00030000, 4, data)
-print "Received Checksum:", last_checksum
+# execute our pkernel finally and set pkernel conform baudrate
+bootromCALL(0x30000)
+time.sleep(0.1) # just to get sure that the pkernel is really running!
+del tty
+tty = SerialPort(DEVICE, None, KERNEL_BAUDRATE)
+
+print "Performing ChipErase..."
+pkernCHIPERASE()
+
+print "Flashing",
+for seq in pkernelseqs:
+ # skip seqs only consisting of 0xffs
+ seqset = list(set(seq.data))
+ if len(seqset) == 1 and seqset[0] == 0xff:
+ continue
+ #print "Flashing", len(seq.data), "bytes at address", hex(seq.address)
+ pkernWRITE(seq.address, len(seq.data), seq.data)
+ tty.flush()
+ sys.stdout.write(".")
+ sys.stdout.flush()
print
-print "Reading from the IRAM again..."
-data2 = cmdREAD(0x00030000, 4)
-print "Received data:", data2, "Checksum:", last_checksum
+duration = time.time() - starttime
+print "Procedure complete, took", round(duration, 2), "seconds."
+
+sendByte(0x97) # exit and restart
+print "Program was started. Have fun!"