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) 2016 Mario de Sousa (msousa@fe.up.pt)

     8

#

     9

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

    10

# it under the terms of the GNU General Public License as published by

    11

# the Free Software Foundation, either version 2 of the License, or

    12

# (at your option) any later version.

    13

#

    14

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

    15

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

    16

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    17

# GNU General Public License for more details.

    18

#

    19

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

    20

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

    21

#

    22

# This code is made available on the understanding that it will not be

    23

# used in safety-critical situations without a full and competent review.

    24

    25

    26

from __future__ import absolute_import

    27

import os

    28

from six.moves import xrange

    29

    30

from modbus.mb_utils import *

    31

from ConfigTreeNode import ConfigTreeNode

    32

from PLCControler import LOCATION_CONFNODE, LOCATION_VAR_MEMORY

    33

import util.paths as paths

    34

    35

base_folder = os.path.split(os.path.dirname(os.path.realpath(__file__)))[0]

    36

base_folder = os.path.join(base_folder, "..")

    37

ModbusPath = os.path.join(base_folder, "Modbus")

    38

    39

    40

#

    41

#

    42

#

    43

# C L I E N T    R E Q U E S T            #

    44

#

    45

#

    46

#

    47

    48

    49

class _RequestPlug(object):

    50

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

    51

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

    52

      <xsd:element name="ModbusRequest">

    53

        <xsd:complexType>

    54

          <xsd:attribute name="Function" type="xsd:string" use="optional" default="01 - Read Coils"/>

    55

          <xsd:attribute name="SlaveID" use="optional" default="1">

    56

            <xsd:simpleType>

    57

                <xsd:restriction base="xsd:integer">

    58

                    <xsd:minInclusive value="0"/>

    59

                    <xsd:maxInclusive value="255"/>

    60

                </xsd:restriction>

    61

            </xsd:simpleType>

    62

          </xsd:attribute>

    63

          <xsd:attribute name="Nr_of_Channels" use="optional" default="1">

    64

            <xsd:simpleType>

    65

                <xsd:restriction base="xsd:integer">

    66

                    <xsd:minInclusive value="1"/>

    67

                    <xsd:maxInclusive value="2000"/>

    68

                </xsd:restriction>

    69

            </xsd:simpleType>

    70

          </xsd:attribute>

    71

          <xsd:attribute name="Start_Address" use="optional" default="0">

    72

            <xsd:simpleType>

    73

                <xsd:restriction base="xsd:integer">

    74

                    <xsd:minInclusive value="0"/>

    75

                    <xsd:maxInclusive value="65535"/>

    76

                </xsd:restriction>

    77

            </xsd:simpleType>

    78

          </xsd:attribute>

    79

          <xsd:attribute name="Timeout_in_ms" use="optional" default="10">

    80

            <xsd:simpleType>

    81

                <xsd:restriction base="xsd:integer">

    82

                    <xsd:minInclusive value="1"/>

    83

                    <xsd:maxInclusive value="100000"/>

    84

                </xsd:restriction>

    85

            </xsd:simpleType>

    86

          </xsd:attribute>

    87

          <xsd:attribute name="Write_on_change" type="xsd:boolean" use="optional" default="false"/>

    88

        </xsd:complexType>

    89

      </xsd:element>

    90

    </xsd:schema>

    91

    """

    92

    93

    def GetParamsAttributes(self, path=None):

    94

        infos = ConfigTreeNode.GetParamsAttributes(self, path=path)

    95

        for element in infos:

    96

            if element["name"] == "ModbusRequest":

    97

                for child in element["children"]:

    98

                    if child["name"] == "Function":

    99

                        list = modbus_function_dict.keys()

   100

                        list.sort()

   101

                        child["type"] = list

   102

        return infos

   103

   104

    def GetVariableLocationTree(self):

   105

        current_location = self.GetCurrentLocation()

   106

        name = self.BaseParams.getName()

   107

        address = self.GetParamsAttributes()[0]["children"][3]["value"]

   108

        count = self.GetParamsAttributes()[0]["children"][2]["value"]

   109

        function = self.GetParamsAttributes()[0]["children"][0]["value"]

   110

        # 'BOOL' or 'WORD'

   111

        datatype = modbus_function_dict[function][3]

   112

        # 1 or 16

   113

        datasize = modbus_function_dict[function][4]

   114

        # 'Q' for coils and holding registers, 'I' for input discretes and input registers

   115

        # datazone = modbus_function_dict[function][5]

   116

        # 'X' for bits, 'W' for words

   117

        datatacc = modbus_function_dict[function][6]

   118

        # 'Coil', 'Holding Register', 'Input Discrete' or 'Input Register'

   119

        dataname = modbus_function_dict[function][7]

   120

        # start off with a boolean entry

   121

        # This is a flag used to allow the user program to control when to 

   122

        # execute the Modbus request.

   123

        # NOTE: If the Modbus request has a 'current_location' of

   124

        #          %QX1.2.3

   125

        #       then the execution control flag will be

   126

        #          %QX1.2.3.0.0

   127

        #       and all the Modbus registers/coils will be

   128

        #          %QX1.2.3.0

   129

        #          %QX1.2.3.1

   130

        #          %QX1.2.3.2

   131

        #            ..

   132

        #          %QX1.2.3.n

   133

        entries = []

   134

        entries.append({

   135

            "name": "Exec. request flag",

   136

            "type": LOCATION_VAR_MEMORY,

   137

            "size": 1,           # BOOL flag

   138

            "IEC_type": "BOOL",  # BOOL flag

   139

            "var_name": "var_name",

   140

            "location": "X" + ".".join([str(i) for i in current_location]) + ".0.0",

   141

            "description": "MB request execution control flag",

   142

            "children": []})        

   143

        for offset in range(address, address + count):

   144

            entries.append({

   145

                "name": dataname + " " + str(offset),

   146

                "type": LOCATION_VAR_MEMORY,

   147

                "size": datasize,

   148

                "IEC_type": datatype,

   149

                "var_name": "MB_" + "".join([w[0] for w in dataname.split()]) + "_" + str(offset),

   150

                "location": datatacc + ".".join([str(i) for i in current_location]) + "." + str(offset),

   151

                "description": "description",

   152

                "children": []})

   153

        return {"name": name,

   154

                "type": LOCATION_CONFNODE,

   155

                "location": ".".join([str(i) for i in current_location]) + ".x",

   156

                "children": entries}

   157

   158

    def CTNGenerate_C(self, buildpath, locations):

   159

        """

   160

        Generate C code

   161

        @param current_location: Tupple containing plugin IEC location : %I0.0.4.5 => (0,0,4,5)

   162

        @param locations: List of complete variables locations \

   163

            [{"IEC_TYPE" : the IEC type (i.e. "INT", "STRING", ...)

   164

            "NAME" : name of the variable (generally "__IW0_1_2" style)

   165

            "DIR" : direction "Q","I" or "M"

   166

            "SIZE" : size "X", "B", "W", "D", "L"

   167

            "LOC" : tuple of interger for IEC location (0,1,2,...)

   168

            }, ...]

   169

        @return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND

   170

        """

   171

        return [], "", False

   172

   173

   174

#

   175

#

   176

#

   177

# S E R V E R    M E M O R Y    A R E A       #

   178

#

   179

#

   180

#

   181

   182

# dictionary implementing:

   183

# key - string with the description we want in the request plugin GUI

   184

# list - (modbus function number, request type, max count value)

   185

modbus_memtype_dict = {

   186

    "01 - Coils":            ('1', 'rw_bits',  65536, "BOOL", 1, "Q", "X", "Coil"),

   187

    "02 - Input Discretes":  ('2', 'ro_bits',  65536, "BOOL", 1, "I", "X", "Input Discrete"),

   188

    "03 - Holding Registers": ('3', 'rw_words', 65536, "WORD", 16, "Q", "W", "Holding Register"),

   189

    "04 - Input Registers":  ('4', 'ro_words', 65536, "WORD", 16, "I", "W", "Input Register"),

   190

}

   191

   192

   193

class _MemoryAreaPlug(object):

   194

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

   195

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

   196

      <xsd:element name="MemoryArea">

   197

        <xsd:complexType>

   198

          <xsd:attribute name="MemoryAreaType" type="xsd:string" use="optional" default="01 - Coils"/>

   199

          <xsd:attribute name="Nr_of_Channels" use="optional" default="1">

   200

            <xsd:simpleType>

   201

                <xsd:restriction base="xsd:integer">

   202

                    <xsd:minInclusive value="1"/>

   203

                    <xsd:maxInclusive value="65536"/>

   204

                </xsd:restriction>

   205

            </xsd:simpleType>

   206

          </xsd:attribute>

   207

          <xsd:attribute name="Start_Address" use="optional" default="0">

   208

            <xsd:simpleType>

   209

                <xsd:restriction base="xsd:integer">

   210

                    <xsd:minInclusive value="0"/>

   211

                    <xsd:maxInclusive value="65535"/>

   212

                </xsd:restriction>

   213

            </xsd:simpleType>

   214

          </xsd:attribute>

   215

        </xsd:complexType>

   216

      </xsd:element>

   217

    </xsd:schema>

   218

    """

   219

   220

    def GetParamsAttributes(self, path=None):

   221

        infos = ConfigTreeNode.GetParamsAttributes(self, path=path)

   222

        for element in infos:

   223

            if element["name"] == "MemoryArea":

   224

                for child in element["children"]:

   225

                    if child["name"] == "MemoryAreaType":

   226

                        list = modbus_memtype_dict.keys()

   227

                        list.sort()

   228

                        child["type"] = list

   229

        return infos

   230

   231

    def GetVariableLocationTree(self):

   232

        current_location = self.GetCurrentLocation()

   233

        name = self.BaseParams.getName()

   234

        address = self.GetParamsAttributes()[0]["children"][2]["value"]

   235

        count = self.GetParamsAttributes()[0]["children"][1]["value"]

   236

        function = self.GetParamsAttributes()[0]["children"][0]["value"]

   237

        # 'BOOL' or 'WORD'

   238

        datatype = modbus_memtype_dict[function][3]

   239

        # 1 or 16

   240

        datasize = modbus_memtype_dict[function][4]

   241

        # 'Q' for coils and holding registers, 'I' for input discretes and input registers

   242

        # datazone = modbus_memtype_dict[function][5]

   243

        # 'X' for bits, 'W' for words

   244

        datatacc = modbus_memtype_dict[function][6]

   245

        # 'Coil', 'Holding Register', 'Input Discrete' or 'Input Register'

   246

        dataname = modbus_memtype_dict[function][7]

   247

        entries = []

   248

        for offset in range(address, address + count):

   249

            entries.append({

   250

                "name": dataname + " " + str(offset),

   251

                "type": LOCATION_VAR_MEMORY,

   252

                "size": datasize,

   253

                "IEC_type": datatype,

   254

                "var_name": "MB_" + "".join([w[0] for w in dataname.split()]) + "_" + str(offset),

   255

                "location": datatacc + ".".join([str(i) for i in current_location]) + "." + str(offset),

   256

                "description": "description",

   257

                "children": []})

   258

        return {"name": name,

   259

                "type": LOCATION_CONFNODE,

   260

                "location": ".".join([str(i) for i in current_location]) + ".x",

   261

                "children": entries}

   262

   263

    def CTNGenerate_C(self, buildpath, locations):

   264

        """

   265

        Generate C code

   266

        @param current_location: Tupple containing plugin IEC location : %I0.0.4.5 => (0,0,4,5)

   267

        @param locations: List of complete variables locations \

   268

            [{"IEC_TYPE" : the IEC type (i.e. "INT", "STRING", ...)

   269

            "NAME" : name of the variable (generally "__IW0_1_2" style)

   270

            "DIR" : direction "Q","I" or "M"

   271

            "SIZE" : size "X", "B", "W", "D", "L"

   272

            "LOC" : tuple of interger for IEC location (0,1,2,...)

   273

            }, ...]

   274

        @return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND

   275

        """

   276

        return [], "", False

   277

   278

   279

#

   280

#

   281

#

   282

# T C P    C L I E N T                 #

   283

#

   284

#

   285

#

   286

   287

class _ModbusTCPclientPlug(object):

   288

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

   289

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

   290

      <xsd:element name="ModbusTCPclient">

   291

        <xsd:complexType>

   292

          <xsd:attribute name="Configuration_Name" type="xsd:string" use="optional" default=""/>

   293

          <xsd:attribute name="Remote_IP_Address" type="xsd:string" use="optional" default="localhost"/>

   294

          <xsd:attribute name="Remote_Port_Number" type="xsd:string" use="optional" default="502"/>

   295

          <xsd:attribute name="Invocation_Rate_in_ms" use="optional" default="100">

   296

            <xsd:simpleType>

   297

                <xsd:restriction base="xsd:unsignedLong">

   298

                    <xsd:minInclusive value="0"/>

   299

                    <xsd:maxInclusive value="2147483647"/>

   300

                </xsd:restriction>

   301

            </xsd:simpleType>

   302

          </xsd:attribute>

   303

        </xsd:complexType>

   304

      </xsd:element>

   305

    </xsd:schema>

   306

    """

   307

    # NOTE: Max value of 2147483647 (i32_max) for Invocation_Rate_in_ms

   308

    # corresponds to aprox 25 days.

   309

    CTNChildrenTypes = [("ModbusRequest", _RequestPlug, "Request")]

   310

    # TODO: Replace with CTNType !!!

   311

    PlugType = "ModbusTCPclient"

   312

   313

   314

    def __init__(self):

   315

        # NOTE:

   316

        # The ModbusTCPclient attribute is added dynamically by ConfigTreeNode._AddParamsMembers()

   317

        # It will be an XML parser object created by

   318

        # GenerateParserFromXSDstring(self.XSD).CreateRoot()

   319

   320

        # Set the default value for the "Configuration_Name" parameter

   321

        # The default value will need to be different for each instance of the 

   322

        # _ModbusTCPclientPlug class, so we cannot hardcode the default value in the XSD above

   323

        # This value will be used by the web interface 

   324

        #   (i.e. the extension to the web server used to configure the Modbus parameters).

   325

        #   (The web server is run/activated/started by Beremiz_service.py)

   326

        #   (The web server code is found in runtime/NevowServer.py)

   327

        #   (The Modbus extension to the web server is found in runtime/Modbus_config.py)

   328

        loc_str = ".".join(map(str, self.GetCurrentLocation()))

   329

        self.ModbusTCPclient.setConfiguration_Name("Modbus TCP Client " + loc_str)

   330

   331

    # Return the number of (modbus library) nodes this specific TCP client will need

   332

    #   return type: (tcp nodes, rtu nodes, ascii nodes)

   333

    def GetNodeCount(self):

   334

        return (1, 0, 0)

   335

   336

    def GetConfigName(self):

   337

        """ Return the node's Configuration_Name """

   338

        return self.ModbusTCPclient.getConfiguration_Name()

   339

   340

    def CTNGenerate_C(self, buildpath, locations):

   341

        """

   342

        Generate C code

   343

        @param current_location: Tupple containing plugin IEC location : %I0.0.4.5 => (0,0,4,5)

   344

        @param locations: List of complete variables locations \

   345

            [{"IEC_TYPE" : the IEC type (i.e. "INT", "STRING", ...)

   346

            "NAME" : name of the variable (generally "__IW0_1_2" style)

   347

            "DIR" : direction "Q","I" or "M"

   348

            "SIZE" : size "X", "B", "W", "D", "L"

   349

            "LOC" : tuple of interger for IEC location (0,1,2,...)

   350

            }, ...]

   351

        @return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND

   352

        """

   353

        return [], "", False

   354

   355

   356

#

   357

#

   358

#

   359

# T C P    S E R V E R                 #

   360

#

   361

#

   362

#

   363

   364

class _ModbusTCPserverPlug(object):

   365

    # NOTE: the Port number is a 'string' and not an 'integer'!

   366

    # This is because the underlying modbus library accepts strings

   367

    # (e.g.: well known port names!)

   368

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

   369

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

   370

      <xsd:element name="ModbusServerNode">

   371

        <xsd:complexType>

   372

          <xsd:attribute name="Configuration_Name" type="xsd:string" use="optional" default=""/>

   373

          <xsd:attribute name="Local_IP_Address" type="xsd:string" use="optional"  default="#ANY#"/>

   374

          <xsd:attribute name="Local_Port_Number" type="xsd:string" use="optional" default="502"/>

   375

          <xsd:attribute name="SlaveID" use="optional" default="0">

   376

            <xsd:simpleType>

   377

                <xsd:restriction base="xsd:integer">

   378

                    <xsd:minInclusive value="0"/>

   379

                    <xsd:maxInclusive value="255"/>

   380

                </xsd:restriction>

   381

            </xsd:simpleType>

   382

          </xsd:attribute>

   383

        </xsd:complexType>

   384

      </xsd:element>

   385

    </xsd:schema>

   386

    """

   387

    CTNChildrenTypes = [("MemoryArea", _MemoryAreaPlug, "Memory Area")]

   388

    # TODO: Replace with CTNType !!!

   389

    PlugType = "ModbusTCPserver"

   390

   391

    def __init__(self):

   392

        # NOTE:

   393

        # The ModbusServerNode attribute is added dynamically by ConfigTreeNode._AddParamsMembers()

   394

        # It will be an XML parser object created by

   395

        # GenerateParserFromXSDstring(self.XSD).CreateRoot()

   396

   397

        # Set the default value for the "Configuration_Name" parameter

   398

        # The default value will need to be different for each instance of the 

   399

        # _ModbusTCPclientPlug class, so we cannot hardcode the default value in the XSD above

   400

        # This value will be used by the web interface 

   401

        #   (i.e. the extension to the web server used to configure the Modbus parameters).

   402

        #   (The web server is run/activated/started by Beremiz_service.py)

   403

        #   (The web server code is found in runtime/NevowServer.py)

   404

        #   (The Modbus extension to the web server is found in runtime/Modbus_config.py)

   405

        loc_str = ".".join(map(str, self.GetCurrentLocation()))

   406

        self.ModbusServerNode.setConfiguration_Name("Modbus TCP Server " + loc_str)

   407

   408

    # Return the number of (modbus library) nodes this specific TCP server will need

   409

    #   return type: (tcp nodes, rtu nodes, ascii nodes)

   410

    def GetNodeCount(self):

   411

        return (1, 0, 0)

   412

   413

    # Return a list with a single tuple conatining the (location, IP address, port number)

   414

    #     location   : location of this node in the configuration tree

   415

    #     port number: IP port used by this Modbus/IP server

   416

    #     IP address : IP address of the network interface on which the server will be listening

   417

    #                  ("", "*", or "#ANY#" => listening on all interfaces!)

   418

    def GetIPServerPortNumbers(self):

   419

        port = self.ModbusServerNode.getLocal_Port_Number()

   420

        addr = self.ModbusServerNode.getLocal_IP_Address()

   421

        return [(self.GetCurrentLocation(), addr, port)]

   422

   423

    def GetConfigName(self):

   424

        """ Return the node's Configuration_Name """

   425

        return self.ModbusServerNode.getConfiguration_Name()

   426

   427

    def CTNGenerate_C(self, buildpath, locations):

   428

        """

   429

        Generate C code

   430

        @param current_location: Tupple containing plugin IEC location : %I0.0.4.5 => (0,0,4,5)

   431

        @param locations: List of complete variables locations \

   432

            [{"IEC_TYPE" : the IEC type (i.e. "INT", "STRING", ...)

   433

            "NAME" : name of the variable (generally "__IW0_1_2" style)

   434

            "DIR" : direction "Q","I" or "M"

   435

            "SIZE" : size "X", "B", "W", "D", "L"

   436

            "LOC" : tuple of interger for IEC location (0,1,2,...)

   437

            }, ...]

   438

        @return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND

   439

        """

   440

        return [], "", False

   441

   442

   443

#

   444

#

   445

#

   446

# R T U    C L I E N T                 #

   447

#

   448

#

   449

#

   450

   451

class _ModbusRTUclientPlug(object):

   452

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

   453

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

   454

      <xsd:element name="ModbusRTUclient">

   455

        <xsd:complexType>

   456

          <xsd:attribute name="Configuration_Name" type="xsd:string" use="optional" default=""/>

   457

          <xsd:attribute name="Serial_Port" type="xsd:string"  use="optional" default="/dev/ttyS0"/>

   458

          <xsd:attribute name="Baud_Rate"   type="xsd:string"  use="optional" default="9600"/>

   459

          <xsd:attribute name="Parity"      type="xsd:string"  use="optional" default="even"/>

   460

          <xsd:attribute name="Stop_Bits"   type="xsd:string"  use="optional" default="1"/>

   461

          <xsd:attribute name="Invocation_Rate_in_ms" use="optional" default="100">

   462

            <xsd:simpleType>

   463

                <xsd:restriction base="xsd:integer">

   464

                    <xsd:minInclusive value="0"/>

   465

                    <xsd:maxInclusive value="2147483647"/>

   466

                </xsd:restriction>

   467

            </xsd:simpleType>

   468

          </xsd:attribute>

   469

        </xsd:complexType>

   470

      </xsd:element>

   471

    </xsd:schema>

   472

    """

   473

    # NOTE: Max value of 2147483647 (i32_max) for Invocation_Rate_in_ms

   474

    # corresponds to aprox 25 days.

   475

    CTNChildrenTypes = [("ModbusRequest", _RequestPlug, "Request")]

   476

    # TODO: Replace with CTNType !!!

   477

    PlugType = "ModbusRTUclient"

   478

   479

    def __init__(self):

   480

        # NOTE:

   481

        # The ModbusRTUclient attribute is added dynamically by ConfigTreeNode._AddParamsMembers()

   482

        # It will be an XML parser object created by

   483

        # GenerateParserFromXSDstring(self.XSD).CreateRoot()

   484

   485

        # Set the default value for the "Configuration_Name" parameter

   486

        # The default value will need to be different for each instance of the 

   487

        # _ModbusTCPclientPlug class, so we cannot hardcode the default value in the XSD above

   488

        # This value will be used by the web interface 

   489

        #   (i.e. the extension to the web server used to configure the Modbus parameters).

   490

        #   (The web server is run/activated/started by Beremiz_service.py)

   491

        #   (The web server code is found in runtime/NevowServer.py)

   492

        #   (The Modbus extension to the web server is found in runtime/Modbus_config.py)

   493

        loc_str = ".".join(map(str, self.GetCurrentLocation()))

   494

        self.ModbusRTUclient.setConfiguration_Name("Modbus RTU Client " + loc_str)

   495

   496

    def GetParamsAttributes(self, path=None):

   497

        infos = ConfigTreeNode.GetParamsAttributes(self, path=path)

   498

        for element in infos:

   499

            if element["name"] == "ModbusRTUclient":

   500

                for child in element["children"]:

   501

                    if child["name"] == "Baud_Rate":

   502

                        child["type"] = modbus_serial_baudrate_list

   503

                    if child["name"] == "Stop_Bits":

   504

                        child["type"] = modbus_serial_stopbits_list

   505

                    if child["name"] == "Parity":

   506

                        child["type"] = modbus_serial_parity_dict.keys()

   507

        return infos

   508