Tooltron

#!/usr/bin/python

import sys

import serial

TT_BOOT = 'b'

TT_PROGRAM_MODE = 'p'

TT_ACK     = 'a'

TT_NACK    = 'n'

TT_BAD = 0

MAX_DATA_LEN = 32

if len(sys.argv) < 4:

    print "usage: netprog.py /path/to/bus/device toolnum binaryfile"

    exit(1)

#for testing, tool number 3

class dumbBus (serial.Serial):

    def __init__(self):

        self.step = 0

    def flushInput(self):

        pass

    def write(self, s):

        print '>',

    def read(self, n):

        if n == 5:

            if self.step == 0:

                #print ord('^'), 3,1,ord(TT_BOOT), (3 ^ 1 ^ ord(TT_BOOT))

                print '<',

                self.step += 1

                return '^' + chr(3) + chr(1) + TT_BOOT + chr(3 ^ 1 ^ ord(TT_BOOT))

            else:

                #print ord('^'), 3,1,ord(TT_ACK), (3 ^ 1 ^ ord(TT_ACK))

                print '<'

                return '^' + chr(3) + chr(1) + TT_ACK + chr(3 ^ 1 ^ ord(TT_ACK))

        else:

            return 0

bus = dumbBus() #serial.Serial(sys.argv[1], BAUD_RATE, timeout = 0) 

bus.flushInput()

#print bus

def sendCommand(t, cmd):

    msg = '^' + chr(1) + chr(t) + cmd + chr(1 ^ t ^ ord(cmd))

    print "sending..."

    bus.write(msg)

def sendProgram(t, size):

    if size > 2**16:

        print "ERROR: program too big!!"

    def lsb(s): return s & 0xFF

    def msb(s): return (s>>8) & 0xFF

    body = chr(1) + chr(t) + TT_PROGRAM_MODE + chr(lsb(size)) + chr(msb(size))

    xor = 0

    for c in body[1:]:

        xor = xor ^ ord(c)

    msg = '^' + body + chr(xor)

    print "sending P"

    bus.write(msg)

def sendData(t, data):

    body = chr(1) + chr(t) + data

    xor = 0

    for c in body[1:]:

        xor = xor ^ ord(c)

    msg = '^' + body + chr(xor)

    print "sending data"

    bus.write(msg)

def readPacket():

    msg = bus.read(5)

    if msg[0] != '^':

        print "did not get start delimeter!"

        return TT_BAD

    src = msg[1]

    #print "src: ", ord(src)

    if msg[2] != chr(1):

        print "packet dest not 1!"

        return TT_BAD

    cmd = msg[3]

    if msg[4] != chr(ord(src) ^ 1 ^ ord(msg[3])):

        print "checksum was wrong!"

        print ord(msg[0]), ord(msg[1]), ord(msg[2]), ord(msg[3]), ord(msg[4])

        print (ord(src) ^ 1 ^ ord(msg[3]))

        return TT_BAD

    return cmd

toolID = int(sys.argv[2])

print "sending file: ", sys.argv[3], " to tool ", toolID

fin = open(sys.argv[3], "r")

fin.seek(0,2)

flen = fin.tell()

fin.seek(0,0)

print "waiting for boot packet..."

while readPacket() != TT_BOOT:

    print "got some other byte ???"

    exit(-4)

sendProgram(toolID, flen)

if readPacket() != TT_ACK:

    print "ERROR: tool did not ACK!!!"

    exit(-1)

while True:

    data = fin.read(MAX_DATA_LEN)

    #print data

    #print '-------------------'

    sendData(toolID, data)

    if readPacket() != TT_ACK:

        print "last data packet not ACK'd!!"

        break

    if len(data) < MAX_DATA_LEN:

        break