two become one ;) merged kernel.py into frprog.py
[pyfrprog.git] / frprog.py
index 293ed4dd449c834a96db2b42ae0e84d3f2fe2ef7..a92b0fb9f2a5af2a776485151c5bebb997ee806e 100755 (executable)
--- a/frprog.py
+++ b/frprog.py
@@ -6,8 +6,10 @@ from SerialPort_linux import *
 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
@@ -15,7 +17,6 @@ last_checksum = 0
 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)
@@ -98,8 +99,6 @@ def cmdCHECKSUM():
        recvChecksum()
 
 def cmdBAUDRATE(baudrate):
-       global last_checksum
-
        # send BAUDRATE command
        sendByte(0x01)
        if (recvByte() != 0xF1):
@@ -108,17 +107,101 @@ 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)
+       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(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: 
@@ -129,22 +212,57 @@ while 1:
                pass
 
 print "OK, trying to set baudrate..."
-
 # set baudrate
-cmdBAUDRATE(REAL_BAUDRATE)
-tty = SerialPort(DEVICE, 100, REAL_BAUDRATE)
-"""
+cmdBAUDRATE(BOOTLOADER_BAUDRATE)
+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)
+       cmdWRITE(addr, len(seq.data), seq.data)
+       tty.flush()
+       sys.stdout.write(".")
+       sys.stdout.flush()
 print
-sendByte(0x01)
-print recvByte()
-sendByte(0x02)
-print recvByte()
-sys.exit(0)
-"""
 
-#cmdCALL(0x00030000)
+# execute our pkernel finally and set pkernel conform baudrate
+cmdCALL(0x30000)
+time.sleep(0.5) # just to get sure that the pkernel is really running!
+del tty
+tty = SerialPort(DEVICE, None, KERNEL_BAUDRATE)
+
+print "Performing ChipErase..."
+pkernCHIPERASE()
+print "Chip erasing done."
+
+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 "Flashing done."
+
+sendByte(0x97) # exit and restart
+print "Program was started. Have fun!"
+
+# some tests here.......
+"""
+# execute (existing) program in ram
 cmdCALL(0x00033ffc)
 sys.exit(0)
+"""
 
 # read something from the IRAM
 #print cmdREAD(0x00030000, 32)
@@ -155,6 +273,7 @@ sys.exit(0)
 #cmdWRITE(0x00030000, 32, data)
 
 
+"""
 # write something to the begin of the IRAM
 data_wr = []
 checksum = 0
@@ -172,3 +291,72 @@ 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)
+"""