only flush once for a sequence, detect 0xffs seqs

[pyfrprog.git] / frprog.py

diff --git a/frprog.py b/frprog.py
index b90465f8dcb2e15f5275e60e52aa221a9decc782..da4f399dcdbdc179a4d63b50f984e77bd0962768 100755 (executable)
--- a/frprog.py
+++ b/frprog.py
@@ -15,7 +15,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)
@@ -113,6 +112,59 @@ def cmdBAUDRATE(baudrate):
        sendByte((baudrate >> 16) & 0xFF)
        sendByte((baudrate >> 24) & 0xFF)
 
+class FlashSequence(object):
+       def __init__(self, address, data):
+               self.address = address
+               self.data = data
+
+# list of all our address/data pairs to flash
+flashseqs = []
+
+# 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:
+       fp = open(sys.argv[1], "r")
+except IOError:
+       print sys.argv[0] + ": Error - couldn't open file " + sys.argv[1] + "!"
+       sys.exit(1)
+
+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
+               flashseqs.append(FlashSequence(address, data))
+
+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)
 
@@ -133,14 +185,21 @@ print "OK, trying to set baudrate..."
 cmdBAUDRATE(REAL_BAUDRATE)
 tty = SerialPort(DEVICE, 100, REAL_BAUDRATE)
 
-"""
-print
-sendByte(0x01)
-print recvByte()
-sendByte(0x02)
-print recvByte()
+# 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)
@@ -218,7 +277,7 @@ print "Received Checksum:", last_checksum
 print
 """
 
-
+"""
 # write some data in the iram and try to execute it
 data_wr =[
                0x9B,0x00,
@@ -242,3 +301,4 @@ cmdWRITE(0x00030000, len(data_wr), data_wr)
 print "Received Checksum:", last_checksum
 print
 cmdCALL(0x00030000)
+"""