1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8

# Copyright (C) 2017: Paul Beltyukov

     9

#

    10

# See COPYING file for copyrights details.

    11

#

    12

# This program is free software; you can redistribute it and/or

    13

# modify it under the terms of the GNU General Public License

    14

# as published by the Free Software Foundation; either version 2

    15

# of the License, or (at your option) any later version.

    16

#

    17

# This program is distributed in the hope that it will be useful,

    18

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    19

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    20

# GNU General Public License for more details.

    21

#

    22

# You should have received a copy of the GNU General Public License

    23

# along with this program; if not, write to the Free Software

    24

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    25

    26

import os, re, operator

    27

from util.ProcessLogger import ProcessLogger

    28

import hashlib

    29

    30

includes_re =  re.compile('\s*#include\s*["<]([^">]*)[">].*')

    31

    32

class toolchain_gcc():

    33

    """

    34

    This abstract class contains GCC specific code.

    35

    It cannot be used as this and should be inherited in a target specific

    36

    class such as target_linux or target_win32

    37

    """

    38

    def __init__(self, CTRInstance):

    39

        self.CTRInstance = CTRInstance

    40

        self.buildpath = None

    41

        self.SetBuildPath(self.CTRInstance._getBuildPath())

    42

    43

    def getBuilderCFLAGS(self):

    44

        """

    45

        Returns list of builder specific CFLAGS

    46

        """

    47

        return [self.CTRInstance.GetTarget().getcontent().getCFLAGS()]

    48

    49

    def getBuilderLDFLAGS(self):

    50

        """

    51

        Returns list of builder specific LDFLAGS

    52

        """

    53

        return self.CTRInstance.LDFLAGS + \

    54

               [self.CTRInstance.GetTarget().getcontent().getLDFLAGS()]

    55

    56

    def getCompiler(self):

    57

        """

    58

        Returns compiler

    59

        """

    60

        return self.CTRInstance.GetTarget().getcontent().getCompiler()

    61

    62

    def getLinker(self):

    63

        """

    64

        Returns linker

    65

        """

    66

        return self.CTRInstance.GetTarget().getcontent().getLinker()

    67

    68

    def GetBinaryCode(self):

    69

        try:

    70

            return open(self.exe_path, "rb").read()

    71

        except Exception, e:

    72

            return None

    73

    74

    def _GetMD5FileName(self):

    75

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

    76

    77

    def ResetBinaryCodeMD5(self):

    78

        self.md5key = None

    79

        try:

    80

            os.remove(self._GetMD5FileName())

    81

        except Exception, e:

    82

            pass

    83

    84

    def GetBinaryCodeMD5(self):

    85

        if self.md5key is not None:

    86

            return self.md5key

    87

        else:

    88

            try:

    89

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

    90

            except Exception, e:

    91

                return None

    92

    93

    def SetBuildPath(self, buildpath):

    94

        if self.buildpath != buildpath:

    95

            self.buildpath = buildpath

    96

            self.exe = self.CTRInstance.GetProjectName() + self.extension

    97

            self.exe_path = os.path.join(self.buildpath, self.exe)

    98

            self.md5key = None

    99

            self.srcmd5 = {}

   100

   101

    def append_cfile_deps(self, src, deps):

   102

        for l in src.splitlines():

   103

            res = includes_re.match(l)

   104

            if res is not None:

   105

                depfn = res.groups()[0]

   106

                if os.path.exists(os.path.join(self.buildpath, depfn)):

   107

                    deps.append(depfn)

   108

   109

    def concat_deps(self, bn):

   110

        # read source

   111

        src = open(os.path.join(self.buildpath, bn),"r").read()

   112

        # update direct dependencies

   113

        deps = []

   114

        self.append_cfile_deps(src, deps)

   115

        # recurse through deps

   116

        # TODO detect cicular deps.

   117

        return reduce(operator.concat, map(self.concat_deps, deps), src)

   118

   119

    def check_and_update_hash_and_deps(self, bn):

   120

        # Get latest computed hash and deps

   121

        oldhash, deps = self.srcmd5.get(bn,(None,[]))

   122

        # read source

   123

        src = open(os.path.join(self.buildpath, bn)).read()

   124

        # compute new hash

   125

        newhash = hashlib.md5(src).hexdigest()

   126

        # compare

   127

        match = (oldhash == newhash)

   128

        if not match:

   129

            # file have changed

   130

            # update direct dependencies

   131

            deps = []

   132

            self.append_cfile_deps(src, deps)

   133

            # store that hashand deps

   134

            self.srcmd5[bn] = (newhash, deps)

   135

        # recurse through deps

   136

        # TODO detect cicular deps.

   137

        return reduce(operator.and_, map(self.check_and_update_hash_and_deps, deps), match)

   138

   139

    def calc_source_md5(self):

   140

        wholesrcdata = ""

   141

        for Location, CFilesAndCFLAGS, DoCalls in self.CTRInstance.LocationCFilesAndCFLAGS:

   142

            # Get CFiles list to give it to makefile

   143

            for CFile, CFLAGS in CFilesAndCFLAGS:

   144

                CFileName = os.path.basename(CFile)

   145

                wholesrcdata += self.concat_deps(CFileName)

   146

        return hashlib.md5(wholesrcdata).hexdigest()

   147

   148

    def calc_md5(self):

   149

        return hashlib.md5(self.GetBinaryCode()).hexdigest()

   150

   151

    def build(self):

   152

        # Retrieve compiler and linker

   153

        self.compiler = self.getCompiler()

   154

        self.linker = self.getLinker()

   155

   156

        Builder_CFLAGS = ' '.join(self.getBuilderCFLAGS())

   157

   158

        ######### GENERATE OBJECT FILES ########################################

   159

        obns = []

   160

        objs = []

   161

        relink = self.GetBinaryCode() is None

   162

        for Location, CFilesAndCFLAGS, DoCalls in self.CTRInstance.LocationCFilesAndCFLAGS:

   163

            if CFilesAndCFLAGS:

   164

                if Location :

   165

                    self.CTRInstance.logger.write(".".join(map(str,Location))+" :\n")

   166

                else:

   167

                    self.CTRInstance.logger.write(_("PLC :\n"))

   168

   169

            for CFile, CFLAGS in CFilesAndCFLAGS:

   170

                if CFile.endswith(".c"):

   171

                    bn = os.path.basename(CFile)

   172

                    obn = os.path.splitext(bn)[0]+".o"

   173

                    objectfilename = os.path.splitext(CFile)[0]+".o"

   174

   175

                    match = self.check_and_update_hash_and_deps(bn)

   176

   177

                    if match:

   178

                        self.CTRInstance.logger.write("   [pass]  "+bn+" -> "+obn+"\n")

   179

                    else:

   180

                        relink = True

   181

   182

                        self.CTRInstance.logger.write("   [CC]  "+bn+" -> "+obn+"\n")

   183

   184

                        status, result, err_result = ProcessLogger(

   185

                               self.CTRInstance.logger,

   186

                               "\"%s\" -c \"%s\" -o \"%s\" %s %s"%

   187

                                   (self.compiler, CFile, objectfilename, Builder_CFLAGS, CFLAGS)

   188

                               ).spin()

   189

   190

                        if status :

   191

                            self.srcmd5.pop(bn)

   192

                            self.CTRInstance.logger.write_error(_("C compilation of %s failed.\n")%bn)

   193

                            return False

   194

                    obns.append(obn)

   195

                    objs.append(objectfilename)

   196

                elif CFile.endswith(".o"):

   197

                    obns.append(os.path.basename(CFile))

   198

                    objs.append(CFile)

   199

   200

        ######### GENERATE OUTPUT FILE ########################################

   201

        # Link all the object files into one binary file

   202

        self.CTRInstance.logger.write(_("Linking :\n"))

   203

        if relink:

   204

            objstring = []

   205

   206

            # Generate list .o files

   207

            listobjstring = '"' + '"  "'.join(objs) + '"'

   208

   209

            ALLldflags = ' '.join(self.getBuilderLDFLAGS())

   210

   211

            self.CTRInstance.logger.write("   [CC]  " + ' '.join(obns)+" -> " + self.exe + "\n")

   212

   213

            status, result, err_result = ProcessLogger(

   214

                   self.CTRInstance.logger,

   215

                   "\"%s\" %s -o \"%s\" %s"%

   216

                       (self.linker,

   217

                        listobjstring,

   218

                        self.exe_path,

   219

                        ALLldflags)

   220

                   ).spin()

   221

   222

            if status :

   223

                return False

   224

   225

        else:

   226

            self.CTRInstance.logger.write("   [pass]  " + ' '.join(obns)+" -> " + self.exe + "\n")

   227

   228

        # Calculate md5 key and get data for the new created PLC

   229

        self.md5key = self.calc_md5()

   230

   231

        # Store new PLC filename based on md5 key

   232

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

   233

        f.write(self.md5key)

   234

        f.close()

   235

   236

        return True

   237