1

from LPCProto import *

     2

     3

class LPCBootProto(LPCProto):

     4

    def HandleTransaction(self, transaction):

     5

        res = None

     6

        try:

     7

            transaction.SetPseudoFile(self.serialPort)

     8

            res = transaction.ExchangeData()

     9

        except:

    10

            pass

    11

        return "Stopped", res

    12

    13

class LPCBootTransaction:

    14

    def __init__(self, optdata = ""):

    15

        self.OptData = optdata

    16

        self.pseudofile = None

    17

    18

    def SetPseudoFile(self, pseudofile):

    19

        self.pseudofile = pseudofile

    20

    21

    def ExchangeData(self): 

    22

        self.pseudofile.write(self.OptData)

    23

        return map(lambda x:self.pseudofile.readline(), xrange(self.expectedlines))

    24

    25

class KEEPBOOTINGTransaction(LPCBootTransaction):

    26

    def __init__(self):

    27

        self.expectedlines = 2

    28

        LPCBootTransaction.__init__(self, "md5\n")

    29

    30

class STARTTransaction(LPCBootTransaction):

    31

    def __init__(self):

    32

        self.expectedlines = 0

    33

        LPCBootTransaction.__init__(self, "go\n")

    34

    35

class CHECKMD5Transaction(LPCBootTransaction):

    36

    def __init__(self, md5ref):

    37

        self.expectedlines = 5 

    38

        LPCBootTransaction.__init__(self, md5ref+"md5\n")

    39

    40

class LOADTransaction(LPCBootTransaction):

    41

    def __init__(self, data, PLCprint):

    42

        self.PLCprint = PLCprint

    43

        LPCBootTransaction.__init__(self, data)

    44

    45

    def ExchangeData(self):

    46

        #file("fw.bin","w").write(self.OptData)

    47

        data = self.OptData

    48

        loptdata = len(self.OptData)

    49

        count=0

    50

        #self.PLCprint("%dkB:" % (loptdata/1024))

    51

        while len(data)>0:

    52

            res = self.pseudofile.write(data[:loptdata/100])

    53

            data = data[res:]

    54

            count += 1

    55

            if count % 10 == 0 :

    56

                self.PLCprint("%d%%" % count)

    57

            else :

    58

                self.PLCprint(".")

    59

        self.PLCprint("\n")

    60

        return True

    61

    62

if __name__ == "__main__":

    63

    __builtins__.BMZ_DBG = True

    64

    TestConnection = LPCBootProto(2,115200,1200)

    65

    mystr=file("fw.bin").read()

    66

    def mylog(blah):

    67

        print blah,

    68

    69

    TestConnection.HandleTransaction(LOADTransaction(mystr, mylog))