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

#

     9

# See COPYING file for copyrights details.

    10

#

    11

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

    12

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

    13

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

    14

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

    15

#

    16

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

    17

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

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

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

    22

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

    23

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

    24

    25

    26

from __future__ import absolute_import

    27

from __future__ import print_function

    28

import traceback

    29

from time import sleep

    30

import copy

    31

import socket

    32

import os.path

    33

    34

import Pyro

    35

import Pyro.core

    36

import Pyro.util

    37

from Pyro.errors import PyroError

    38

    39

    40

service_type = '_PYRO._tcp.local.'

    41

# this module attribute contains a list of DNS-SD (Zeroconf) service types

    42

# supported by this connector confnode.

    43

#

    44

# for connectors that do not support DNS-SD, this attribute can be omitted

    45

# or set to an empty list.

    46

    47

    48

def PYRO_connector_factory(uri, confnodesroot):

    49

    """

    50

    This returns the connector to Pyro style PLCobject

    51

    """

    52

    confnodesroot.logger.write(_("PYRO connecting to URI : %s\n") % uri)

    53

    54

    servicetype, location = uri.split("://")

    55

    if servicetype == "PYROS":

    56

        schemename = "PYROLOCSSL"

    57

        # Protect against name->IP substitution in Pyro3

    58

        Pyro.config.PYRO_DNS_URI = True

    59

        # Beware Pyro lib need str path, not unicode

    60

        # don't rely on PYRO_STORAGE ! see documentation

    61

        Pyro.config.PYROSSL_CERTDIR = os.path.abspath(str(confnodesroot.ProjectPath) + '/certs')

    62

        if not os.path.exists(Pyro.config.PYROSSL_CERTDIR):

    63

            confnodesroot.logger.write_error(

    64

                'Error : the directory %s is missing for SSL certificates (certs_dir).'

    65

                'Please fix it in your project.\n' % Pyro.config.PYROSSL_CERTDIR)

    66

            return None

    67

        else:

    68

            confnodesroot.logger.write(_("PYRO using certificates in '%s' \n")

    69

                                       % (Pyro.config.PYROSSL_CERTDIR))

    70

        Pyro.config.PYROSSL_CERT = "client.crt"

    71

        Pyro.config.PYROSSL_KEY = "client.key"

    72

    73

        # Ugly Monkey Patching

    74

        def _gettimeout(self):

    75

            return self.timeout

    76

    77

        def _settimeout(self, timeout):

    78

            self.timeout = timeout

    79

        from M2Crypto.SSL import Connection  # pylint: disable=import-error

    80

        Connection.timeout = None

    81

        Connection.gettimeout = _gettimeout

    82

        Connection.settimeout = _settimeout

    83

        # M2Crypto.SSL.Checker.WrongHost: Peer certificate commonName does not

    84

        # match host, expected 127.0.0.1, got server

    85

        Connection.clientPostConnectionCheck = None

    86

    else:

    87

        schemename = "PYROLOC"

    88

    if location.find(service_type) != -1:

    89

        try:

    90

            from zeroconf import Zeroconf

    91

            r = Zeroconf()

    92

            i = r.get_service_info(service_type, location)

    93

            if i is None:

    94

                raise Exception("'%s' not found" % location)

    95

            ip = str(socket.inet_ntoa(i.address))

    96

            port = str(i.port)

    97

            newlocation = ip + ':' + port

    98

            confnodesroot.logger.write(_("'{a1}' is located at {a2}\n").format(a1=location, a2=newlocation))

    99

            location = newlocation

   100

            r.close()

   101

        except Exception:

   102

            confnodesroot.logger.write_error(_("MDNS resolution failure for '%s'\n") % location)

   103

            confnodesroot.logger.write_error(traceback.format_exc())

   104

            return None

   105

   106

    # Try to get the proxy object

   107

    try:

   108

        RemotePLCObjectProxy = Pyro.core.getAttrProxyForURI(schemename + "://" + location + "/PLCObject")

   109

    except Exception:

   110

        confnodesroot.logger.write_error(_("Connection to '%s' failed.\n") % location)

   111

        confnodesroot.logger.write_error(traceback.format_exc())

   112

        return None

   113

   114

    def PyroCatcher(func, default=None):

   115

        """

   116

        A function that catch a Pyro exceptions, write error to logger

   117

        and return default value when it happen

   118

        """

   119

        def catcher_func(*args, **kwargs):

   120

            try:

   121

                return func(*args, **kwargs)

   122

            except Pyro.errors.ConnectionClosedError, e:

   123

                confnodesroot.logger.write_error(_("Connection lost!\n"))

   124

                confnodesroot._SetConnector(None)

   125

            except Pyro.errors.ProtocolError, e:

   126

                confnodesroot.logger.write_error(_("Pyro exception: %s\n") % e)

   127

            except Exception, e:

   128

                # confnodesroot.logger.write_error(traceback.format_exc())

   129

                errmess = ''.join(Pyro.util.getPyroTraceback(e))

   130

                confnodesroot.logger.write_error(errmess + "\n")

   131

                print(errmess)

   132

                confnodesroot._SetConnector(None)

   133

            return default

   134

        return catcher_func

   135

   136

    # Check connection is effective.

   137

    # lambda is for getattr of GetPLCstatus to happen inside catcher

   138

    if PyroCatcher(RemotePLCObjectProxy.GetPLCstatus)() is None:

   139

        confnodesroot.logger.write_error(_("Cannot get PLC status - connection failed.\n"))

   140

        return None

   141

   142

    _special_return_funcs = {

   143

        "StartPLC": False,

   144

        "GetTraceVariables": ("Broken", None),

   145

        "GetPLCstatus": ("Broken", None),

   146

        "RemoteExec": (-1, "RemoteExec script failed!")

   147

    }

   148

   149

    class PyroProxyProxy(object):

   150

        """

   151

        A proxy proxy class to handle Beremiz Pyro interface specific behavior.

   152

        And to put Pyro exception catcher in between caller and Pyro proxy

   153

        """

   154

        def __getattr__(self, attrName):

   155

            member = self.__dict__.get(attrName, None)

   156

            if member is None:

   157

                def my_local_func(*args, **kwargs):

   158

                    return RemotePLCObjectProxy.__getattr__(attrName)(*args, **kwargs)

   159

                member = PyroCatcher(my_local_func, _special_return_funcs.get(attrName, None))

   160

                self.__dict__[attrName] = member

   161

            return member

   162

   163

    return PyroProxyProxy()