Tooltron

#!/usr/bin/python

import sys

import serial

import sys

sys.path.append('../common')

import common

MAX_PROG_SIZE = 1024-32

DUMB_SERIAL = 0

MAX_DATA_LEN = common.MAX_PAYLOAD_LEN

if len(sys.argv) < 4:

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

    exit(1)

#for testing, tool number 3 hardcoded

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(common.TT_BOOT), (3 ^ 1 ^ ord(common.TT_BOOT))

                print '<',

                self.step += 1

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

            else:

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

                print '<',

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

        else:

            return 0

common.initBus(sys.argv[1], timeout=10)

def sendCommand(t, cmd):

    common.sendMessage(t,cmd,"")

def sendProgram(t, size):

    if size > 65535:

        print "ERROR: program too big!!"

    def lsb(s): return s & 0xFF

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

    print "sending P"

    common.sendMessage(t,common.TT_PROGRAM_MODE,chr(lsb(size))+chr(msb(size)))

dCount = 0

def sendData(t, data):

    #PROGD messages must be 32 bytes, so expand if neccesary

    n = len(data)

    while n < MAX_DATA_LEN:

        data = data + chr(0)

        n = n + 1

    common.sendMessage(t,common.TT_PROGD,data)

    print '.',

    sys.stdout.flush()

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)

if flen > MAX_PROG_SIZE:

    print "Program too Large!"

    print "prog size:",flen

    print "max size:",MAX_PROG_SIZE

    exit(-1)

print "prog size:",flen

print "sending RESET"

sendCommand(toolID, common.TT_RESET)

print "waiting for boot packet..."

m = common.readMessage()

if m == None:

    print "Didn't get a packet (timeout?), bailing out"

    exit(-4)

[src,dest,cmd,data] = m

if cmd != common.TT_BOOT:

    print "Didn't get a boot packet, bailing out"

    print "got", cmd

    exit(-4)

else:

    print "got boot packet from " + str(ord(src))

r = 0

while r != common.TT_ACK:

    sendProgram(toolID, flen)

    m = common.readMessage()

    if m == None:

        print "ERROR: probably a timeout"

        exit(-1)

    [src, dest, r, data] = m

    if r == common.TT_NACK:

        print "WARNING: got NACK, trying again"

    elif r == common.TT_BOOT:

        print "ERROR: got stray boot packet"

        #TODO: restart the whole state machine after the boot packet

        exit(-2)

    elif r != common.TT_ACK:

        print "ERROR: got another packet:",r

        exit(-1)

print 'sending data',

sys.stdout.flush()

while True:

    data = fin.read(MAX_DATA_LEN)

    #print data

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

    if len(data) == 0:

        break

    r = 0

    while r != common.TT_ACK:

        sendData(toolID, data)

        m = common.readMessage()

        if m == None:

            print "ERROR: probably a timeout"

            exit(-1)

        [src, dest, r, data] = m

        if r == common.TT_NACK:

            print "WARNING: got NACK, trying again"

        elif r == common.TT_BOOT:

            print "ERROR: got stray boot packet from "+str(ord(src))

            #TODO: restart the whole state machine after the boot packet

            exit(-2)

        elif r != common.TT_ACK:

            print "ERROR: got another packet:",r

            exit(-1)