timeout fix
[pyfrprog.git] / frprog.py
index 2522bfd5449e5e317b29c012536ebbe3303db264..ce102ee8af94f89f61fe3a9f429d525e9a0cd1d9 100755 (executable)
--- a/frprog.py
+++ b/frprog.py
@@ -2,18 +2,22 @@
 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))
 
 def sendWord(word):
@@ -34,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):
@@ -53,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):
@@ -71,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):
@@ -85,7 +89,7 @@ def cmdCALL(address):
        #return recvByte()
 
 # TODO: test this function!
-def cmdCHECKSUM():
+def bootromCHECKSUM():
        # call CHECKSUM command
        sendByte(0x01)
        if (recvByte() != 0xF1):
@@ -96,7 +100,7 @@ def cmdCHECKSUM():
        # get checksum
        recvChecksum()
 
-def cmdBAUDRATE(baudrate):
+def bootromBAUDRATE(baudrate):
        # send BAUDRATE command
        sendByte(0x01)
        if (recvByte() != 0xF1):
@@ -107,11 +111,38 @@ def cmdBAUDRATE(baudrate):
        # send desired baudrate
        sendDWord(baudrate)
 
+def pkernCHIPERASE():
+       sendByte(0x15)
+       if (recvByte() != 0x45):
+               raise Exception
+       # wait till completion...
+       if (recvByte() != 0x23):
+               raise Exception
 
-class FlashSequence(object):
-       def __init__(self, address, data):
-               self.address = address
-               self.data = data
+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")
@@ -146,157 +177,115 @@ def readMHXFile(filename): # desired mhx filename
        fp.close()
        return retval
 
-
-# check command line arguments
-if len(sys.argv) != 2:
-       print "Usage: " + sys.argv[0] + " [mhx-file]"
-       sys.exit(1)
-
-# read in data from mhx-file before starting
-try:
-       flashseqs = readMHXFile(sys.argv[1])
-except IOError as error:
-       print sys.argv[0] + ": Error - couldn't open file " + error.filename + "!"
-       sys.exit(1)
-
-print "The following flash sequences have been read in:"
-for seq in flashseqs:
-       print hex(seq.address) + ":", seq.data
-
-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)
-
-# let the fun begin!
-for seq in flashseqs:
-       if(seq.address <= 0x40000):
-               addr = seq.address
-       else:
-               continue
-       print "RAMing", len(seq.data), "bytes at address", hex(addr)
-       cmdWRITE(addr, len(seq.data), seq.data)
-       tty.flush()
-
-cmdCALL(0x30000);
-sys.exit(0)
-
-
-# some tests here.......
-"""
-# execute (existing) program in ram
-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
-"""
-
-"""
-# see whats in the iram at the moment
-data_wr = []
-print "Reading from the IRAM..."
-data_re = cmdREAD(0x00030000, 0x10000-4)
-print "Received data:", data_re, "Checksum:", last_checksum
-"""
-
-"""
-# see whats in the dram at the moment
-data_wr = []
-print "Reading from the DRAM..."
-data_re = cmdREAD(0x0002C000, 0x10000-0xC000-4)
-print "Received data:", data_re, "Checksum:", last_checksum
-"""
-
-"""
-# blank the iram
-data_wr = []
-for i in range(0, 0x10000-4):
-       value = 0
-       data_wr.append(value)
-
-print "Writing", data_wr, "to the IRAM..."
-cmdWRITE(0x00030000, len(data_wr), data_wr)
-print "Received Checksum:", last_checksum
-print
-"""
-
-"""
-# blank the dram
-data_wr = []
-for i in range(0, 0x10000-0xC000-4):
-       value = 0
-       data_wr.append(value)
-
-print "Writing", data_wr, "to the DRAM..."
-cmdWRITE(0x0002C000, len(data_wr), data_wr)
-print "Received Checksum:", last_checksum
-print
-"""
-
-"""
-# write some data in the iram and try to execute it
-data_wr =[
-               0x9B,0x00,
-               0x0D,0x4e,
-               0xcf,0xf1,
-               0x16,0x01,
-               0x9b,0x05,
-               0x04,0xc7,
-               0xc1,0x06,
-               0x16,0x56,
-               0xe0,0xfb, #branch
-               0x9f,0xa0,0x9f,0xa0,0x9f,0xa0, #nop
-               0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
-               0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
-               0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
-               0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
-               0x9f,0xa0,0x9f,0xa0,0x9f,0xa0,
-               0x9f,0xa0,0x9f,0xa0,0x9f,0xa0]
-print "Writing", data_wr, "to the IRAM..."
-cmdWRITE(0x00030000, len(data_wr), data_wr)
-print "Received Checksum:", last_checksum
-print
-cmdCALL(0x00030000)
-"""
+def usage(execfile):
+       print "Usage: " + execfile + " <target mhx-file> [-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())