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

import Pyro

    26

import Pyro.core

    27

import Pyro.util

    28

from Pyro.errors import PyroError

    29

import traceback

    30

from time import sleep

    31

import copy

    32

import socket

    33

service_type = '_PYRO._tcp.local.'

    34

import os.path

    35

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

    36

# supported by this connector confnode.

    37

#

    38

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

    39

# or set to an empty list.

    40

    41

    42

def PYRO_connector_factory(uri, confnodesroot):

    43

    """

    44

    This returns the connector to Pyro style PLCobject

    45

    """

    46

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

    47

    48

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

    49

    if servicetype == "PYROS":

    50

        schemename = "PYROLOCSSL"

    51

        # Protect against name->IP substitution in Pyro3

    52

        Pyro.config.PYRO_DNS_URI = True

    53

        # Beware Pyro lib need str path, not unicode

    54

        # don't rely on PYRO_STORAGE ! see documentation

    55

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

    56

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

    57

            confnodesroot.logger.write_error(

    58

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

    59

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

    60

            return None

    61

        else:

    62

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

    63

                                       % (Pyro.config.PYROSSL_CERTDIR))

    64

        Pyro.config.PYROSSL_CERT = "client.crt"

    65

        Pyro.config.PYROSSL_KEY = "client.key"

    66

    67

        # Ugly Monkey Patching

    68

        def _gettimeout(self):

    69

            return self.timeout

    70

    71

        def _settimeout(self, timeout):

    72

            self.timeout = timeout

    73

        from M2Crypto.SSL import Connection

    74

        Connection.timeout = None

    75

        Connection.gettimeout = _gettimeout

    76

        Connection.settimeout = _settimeout

    77

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

    78

        # match host, expected 127.0.0.1, got server

    79

        Connection.clientPostConnectionCheck = None

    80

    else:

    81

        schemename = "PYROLOC"

    82

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

    83

        try:

    84

            from util.Zeroconf import Zeroconf

    85

            r = Zeroconf()

    86

            i = r.getServiceInfo(service_type, location)

    87

            if i is None:

    88

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

    89

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

    90

            port = str(i.getPort())

    91

            newlocation = ip + ':' + port

    92

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

    93

            location = newlocation

    94

            r.close()

    95

        except Exception, msg:

    96

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

    97

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

    98

            return None

    99

   100

    # Try to get the proxy object

   101

    try:

   102

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

   103

    except Exception, msg:

   104

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

   105

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

   106

        return None

   107

   108

    def PyroCatcher(func, default=None):

   109

        """

   110

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

   111

        and return default value when it happen

   112

        """

   113

        def catcher_func(*args, **kwargs):

   114

            try:

   115

                return func(*args, **kwargs)

   116

            except Pyro.errors.ConnectionClosedError, e:

   117

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

   118

                confnodesroot._SetConnector(None)

   119

            except Pyro.errors.ProtocolError, e:

   120

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

   121

            except Exception, e:

   122

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

   123

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

   124

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

   125

                print errmess

   126

                confnodesroot._SetConnector(None)

   127

            return default

   128

        return catcher_func

   129

   130

    # Check connection is effective.

   131

    # lambda is for getattr of GetPLCstatus to happen inside catcher

   132

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

   133

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

   134

        return None

   135

   136

    class PyroProxyProxy(object):

   137

        """

   138

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

   139

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

   140

        """

   141

        def __init__(self):

   142

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

   143

            self.RemotePLCObjectProxyCopy = None

   144

   145

        def GetPyroProxy(self):

   146

            """

   147

            This func returns the real Pyro Proxy.

   148

            Use this if you musn't keep reference to it.

   149

            """

   150

            return RemotePLCObjectProxy

   151

   152

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

   153

            """

   154

            confnodesroot._connector.GetPyroProxy() is used

   155

            rather than RemotePLCObjectProxy because

   156

            object is recreated meanwhile,

   157

            so we must not keep ref to it here

   158

            """

   159

            current_status, log_count = confnodesroot._connector.GetPyroProxy().GetPLCstatus()

   160

            if current_status == "Dirty":

   161

                """

   162

                Some bad libs with static symbols may polute PLC

   163

                ask runtime to suicide and come back again

   164

                """

   165

                confnodesroot.logger.write(_("Force runtime reload\n"))

   166

                confnodesroot._connector.GetPyroProxy().ForceReload()

   167

                confnodesroot._Disconnect()

   168

                # let remote PLC time to resurect.(freeze app)

   169

                sleep(0.5)

   170

                confnodesroot._Connect()

   171

            self.RemotePLCObjectProxyCopy = copy.copy(confnodesroot._connector.GetPyroProxy())

   172

            return confnodesroot._connector.GetPyroProxy().StartPLC(*args, **kwargs)

   173

        StartPLC = PyroCatcher(_PyroStartPLC, False)

   174

   175

        def _PyroGetTraceVariables(self):

   176

            """

   177

            for safe use in from debug thread, must use the copy

   178

            """

   179

            if self.RemotePLCObjectProxyCopy is None:

   180

                self.RemotePLCObjectProxyCopy = copy.copy(confnodesroot._connector.GetPyroProxy())

   181

            return self.RemotePLCObjectProxyCopy.GetTraceVariables()

   182

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

   183

   184

        def _PyroGetPLCstatus(self):

   185

            return RemotePLCObjectProxy.GetPLCstatus()

   186

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

   187

   188

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

   189

            return RemotePLCObjectProxy.RemoteExec(script, **kwargs)

   190

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

   191

   192

        def __getattr__(self, attrName):

   193

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

   194

            if member is None:

   195

                def my_local_func(*args, **kwargs):

   196

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

   197

                member = PyroCatcher(my_local_func, None)

   198

                self.__dict__[attrName] = member

   199

            return member

   200

   201

    return PyroProxyProxy()