X-Git-Url: http://wien.tomnetworks.com/gitweb/?a=blobdiff_plain;f=frprog.py;h=ce102ee8af94f89f61fe3a9f429d525e9a0cd1d9;hb=e55c405c605504aade27b7b5060f313f6ddf6ead;hp=293ed4dd449c834a96db2b42ae0e84d3f2fe2ef7;hpb=0ccc9709cad2069ca6e4193e1924aed1691729d1;p=pyfrprog.git diff --git a/frprog.py b/frprog.py index 293ed4d..ce102ee 100755 --- a/frprog.py +++ b/frprog.py @@ -2,20 +2,23 @@ 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 after init -REAL_BAUDRATE=38400 +# 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 +class FlashSequence(object): + def __init__(self, address, data): + self.address = address + self.data = data + def sendByte(byte): - time.sleep(0.001) # just to get sure, wait 1ms tty.write(chr(byte)) - tty.flush() def sendWord(word): sendByte(word & 0xFF) @@ -35,7 +38,7 @@ def recvChecksum(): last_checksum = recvByte() last_checksum |= (recvByte() << 8) -def cmdREAD(address, size): +def bootromREAD(address, size): # send READ command sendByte(0x01) if (recvByte() != 0xF1): @@ -54,7 +57,7 @@ def cmdREAD(address, size): recvChecksum() return data -def cmdWRITE(address, size, data): +def bootromWRITE(address, size, data): # send WRITE command sendByte(0x01) if (recvByte() != 0xF1): @@ -72,7 +75,7 @@ def cmdWRITE(address, size, data): recvChecksum() # TODO: test this function! -def cmdCALL(address): +def bootromCALL(address): # send CALL command sendByte(0x01) if (recvByte() != 0xF1): @@ -86,7 +89,7 @@ def cmdCALL(address): #return recvByte() # TODO: test this function! -def cmdCHECKSUM(): +def bootromCHECKSUM(): # call CHECKSUM command sendByte(0x01) if (recvByte() != 0xF1): @@ -97,9 +100,7 @@ def cmdCHECKSUM(): # get checksum recvChecksum() -def cmdBAUDRATE(baudrate): - global last_checksum - +def bootromBAUDRATE(baudrate): # send BAUDRATE command sendByte(0x01) if (recvByte() != 0xF1): @@ -108,67 +109,183 @@ def cmdBAUDRATE(baudrate): if (recvByte() != 0x86): raise Exception # send desired baudrate - sendByte(baudrate & 0xFF) - sendByte((baudrate >> 8) & 0xFF) - sendByte((baudrate >> 16) & 0xFF) - sendByte((baudrate >> 24) & 0xFF) - -print "Initializing serial port..." -tty = SerialPort(DEVICE, 100, INIT_BAUDRATE) - -print "Please press RESET on your 1337 board..." - -while 1: - tty.write('V') - tty.flush() - try: - if tty.read() == 'F': - break - except SerialPortException: - # timeout happened, who cares ;-) - pass - -print "OK, trying to set baudrate..." - -# set baudrate -cmdBAUDRATE(REAL_BAUDRATE) -tty = SerialPort(DEVICE, 100, REAL_BAUDRATE) -""" -print -sendByte(0x01) -print recvByte() -sendByte(0x02) -print recvByte() -sys.exit(0) -""" - -#cmdCALL(0x00030000) -cmdCALL(0x00033ffc) -sys.exit(0) - -# read something from the IRAM -#print cmdREAD(0x00030000, 32) - -#data = [] -#for i in range(0, 32): -# data.append(i) -#cmdWRITE(0x00030000, 32, data) - - -# write something to the begin of the IRAM -data_wr = [] -checksum = 0 -for i in range(0, 0x400): - value = i%256 - data_wr.append(value) - checksum = (checksum + value) % (2**16) - -print "Calculated checksum:", checksum -print "Writing", data_wr, "to the IRAM..." -cmdWRITE(0x00030000, len(data_wr), data_wr) -print "Received Checksum:", last_checksum -print - -print "Reading from the IRAM again..." -data_re = cmdREAD(0x00030000, len(data_wr)) -print "Received data:", data_re, "Checksum:", last_checksum + 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 + +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 + +def usage(execfile): + print "Usage: " + execfile + " [-d DEVICE]" + +def main(argv=None): + # check command line arguments + if argv is None: + argv = sys.argv + + if len(argv) == 2 and (argv[1] == "-v" or argv[1] == "--version"): + print "Version: %VERSION%" + return 0 + + if len(argv) != 2 and len(argv) != 4: + usage(argv[0]) + return 1 + + # standard serial device to communicate with + DEVICE="/dev/ttyUSB0" + + # overrule standard device if provided with -d + if len(argv) == 4: + if argv[2] == "-d": + DEVICE = argv[3] + else: + usage(argv[0]) + return 1 + + # read in data from mhx-files before starting + try: + try: + bootloaderseqs = readMHXFile("pkernel/pkernel.mhx") + except IOError as error1: + bootloaderseqs = readMHXFile("%PREFIX%/share/frprog/pkernel.mhx") + pkernelseqs = readMHXFile(argv[1]) + except IOError as error: + print argv[0] + ": Error - couldn't open file " + error.filename + "!" + return 1 + + print "Initializing serial port..." + global tty + tty = SerialPort(DEVICE, 100, INIT_BAUDRATE) + + print "Please press RESET on your board..." + + while True: + tty.write('V') + tty.flush() + try: + if tty.read() == 'F': + break + except SerialPortException: + # timeout happened, who cares ;-) + pass + + starttime = time.time() # save time at this point for evaluating the duration at the end + + print "OK, trying to set baudrate..." + # set baudrate + try: + bootromBAUDRATE(BOOTLOADER_BAUDRATE) + except SerialPortException: + print "timeout exception: try again ->" + 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 + + # 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 + + duration = time.time() - starttime + print "Procedure complete, took", round(duration, 2), "seconds." + + sendByte(0x97) # exit and restart + print "Program was started. Have fun!" + + +if __name__ == '__main__': + sys.exit(main())