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

    class PyroProxyProxy(object):

   143

        """

   144

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

   145

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

   146

        """

   147

        def __init__(self):

   148

            # for safe use in from debug thread, must create a copy

   149

            self.RemotePLCObjectProxyCopy = None

   150

   151

        def _PyroStartPLC(self, *args, **kwargs):

   152

            return RemotePLCObjectProxy.StartPLC(*args, **kwargs)

   153

        StartPLC = PyroCatcher(_PyroStartPLC, False)

   154

   155

        def _PyroGetTraceVariables(self):

   156

            """

   157

            for use from debug thread, use a copy

   158

            pyro creates a new thread on server end proxy object is copied

   159

            """

   160

            if self.RemotePLCObjectProxyCopy is None:

   161

                self.RemotePLCObjectProxyCopy = copy.copy(RemotePLCObjectProxy)

   162

            return self.RemotePLCObjectProxyCopy.GetTraceVariables()

   163

        GetTraceVariables = PyroCatcher(_PyroGetTraceVariables, ("Broken", None))

   164

   165

        def _PyroGetPLCstatus(self):

   166

            return RemotePLCObjectProxy.GetPLCstatus()

   167

        GetPLCstatus = PyroCatcher(_PyroGetPLCstatus, ("Broken", None))

   168

   169

        def _PyroRemoteExec(self, script, **kwargs):

   170

            return RemotePLCObjectProxy.RemoteExec(script, **kwargs)

   171

        RemoteExec = PyroCatcher(_PyroRemoteExec, (-1, "RemoteExec script failed!"))

   172

   173

        def __getattr__(self, attrName):

   174

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

   175

            if member is None:

   176

                def my_local_func(*args, **kwargs):

   177

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

   178

                member = PyroCatcher(my_local_func, None)

   179

                self.__dict__[attrName] = member

   180

            return member

   181

   182

    return PyroProxyProxy()