2

from subprocess import Popen,PIPE

     3

from toolchain_makefile import toolchain_makefile

     4

import hashlib

     5

     6

class LPC_target(toolchain_makefile):

     7

    #extension = ".ld"

     8

    #DebugEnabled = False

     9

    def __init__(self, CTRInstance):

    10

        self.binmd5key = None

    11

        toolchain_makefile.__init__(self, CTRInstance)

    12

    13

    def _GetBinMD5FileName(self):

    14

        return os.path.join(self.buildpath, "lastbuildPLCbin.md5")

    15

    16

    def _get_md5_header(self):

    17

        """Returns signature header"""

    18

        buildpath = self.buildpath.encode(sys.getfilesystemencoding())

    19

        size = int(Popen(

    20

             ['arm-elf-size','-B',os.path.join(buildpath,"ArmPLC_rom.elf")],

    21

             stdout=PIPE).communicate()[0].splitlines()[1].split()[0])

    22

        res = "&" + hashlib.md5(open(os.path.join(buildpath, "ArmPLC_rom.bin"), "rb").read(size)).hexdigest() + '\n' +\

    23

              "$" + str(size) + '\n'

    24

        return res

    25

    26

    def GetBinaryCode(self):

    27

        """Returns ready to send signed + sized intel formated hex program"""

    28

        try:

    29

            res = self._get_md5_header() +\

    30

                   open(os.path.join(self.buildpath, "ArmPLC_rom.hex"), "r").read()

    31

            return res

    32

        except Exception, e:

    33

            return None

    34

    35

    def _get_cached_md5_header(self):

    36

        if self.binmd5key is not None:

    37

            return self.binmd5key

    38

        else:

    39

            try:

    40

                return open(self._GetBinMD5FileName(), "r").read()

    41

            except IOError, e:

    42

                return None

    43

    44

    def ResetBinaryCodeMD5(self, mode):

    45

        if mode == "BOOTLOADER":

    46

            self.binmd5key = None

    47

            try:

    48

                os.remove(self._GetBinMD5FileName())

    49

            except Exception, e:

    50

                pass

    51

        else:

    52

            return toolchain_makefile.ResetBinaryCodeMD5(self)

    53

    54

    def GetBinaryCodeMD5(self, mode):

    55

        if mode == "BOOTLOADER":

    56

            return self._get_cached_md5_header()

    57

        else:

    58

            return toolchain_makefile.GetBinaryCodeMD5(self)

    59

    60

    def build(self):

    61

        res = toolchain_makefile.build(self)

    62

        if res:

    63

            self.binmd5key = self._get_md5_header()

    64

            f = open(self._GetBinMD5FileName(), "w")

    65

            f.write(self.binmd5key)

    66

            f.close()

    67

            try:

    68

                self.CTRInstance.logger.write(

    69

                    _("Binary is %s bytes long\n")%

    70

                        str(os.path.getsize(

    71

                            os.path.join(self.buildpath, "ArmPLC_rom.bin"))))

    72

            except Exception, e:

    73

                pass

    74

        return res

    75