author | Edouard Tisserant |
Mon, 31 Aug 2020 13:54:08 +0200 | |
changeset 2706 | 65f35ac97cd6 |
parent 2622 | e10a1095f577 |
child 3750 | f62625418bff |
child 3861 | 7e17f7e02a2b |
permissions | -rw-r--r-- |
#!/usr/bin/env python # -*- coding: utf-8 -*- # This file is part of Beremiz, a Integrated Development Environment for # programming IEC 61131-3 automates supporting plcopen standard and CanFestival. # # Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD # # See COPYING file for copyrights details. # # This program is free software; you can redistribute it and/or # modify it under the terms of the GNU General Public License # as published by the Free Software Foundation; either version 2 # of the License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. from __future__ import absolute_import from __future__ import print_function import traceback from time import sleep import copy import socket import os.path import Pyro import Pyro.core import Pyro.util from Pyro.errors import PyroError import PSKManagement as PSK from connectors.PYRO.PSK_Adapter import setupPSKAdapter from runtime import PlcStatus def switch_pyro_adapter(use_ssl): """ Reloads Pyro module with new settings. This is workaround for Pyro, because it doesn't work with SSL wrapper. """ # Pyro.config.PYRO_BROKEN_MSGWAITALL = use_ssl reload(Pyro.protocol) if use_ssl: setupPSKAdapter() def PYRO_connector_factory(uri, confnodesroot): """ This returns the connector to Pyro style PLCobject """ confnodesroot.logger.write(_("PYRO connecting to URI : %s\n") % uri) scheme, location = uri.split("://") use_ssl = scheme == "PYROS" switch_pyro_adapter(use_ssl) if use_ssl: schemename = "PYROLOCPSK" url, ID = location.split('#') # TODO fix exception when # not found # load PSK from project secpath = os.path.join(str(confnodesroot.ProjectPath), 'psk', ID+'.secret') if not os.path.exists(secpath): confnodesroot.logger.write_error( 'Error: Pre-Shared-Key Secret in %s is missing!\n' % secpath) return None secret = open(secpath).read().partition(':')[2].rstrip('\n\r') Pyro.config.PYROPSK = (secret, ID) # strip ID from URL, so that pyro can understand it. location = url else: schemename = "PYROLOC" # Try to get the proxy object try: RemotePLCObjectProxy = Pyro.core.getAttrProxyForURI(schemename + "://" + location + "/PLCObject") except Exception as e: confnodesroot.logger.write_error( _("Connection to {loc} failed with exception {ex}\n").format( loc=location, ex=str(e))) return None RemotePLCObjectProxy.adapter.setTimeout(60) def PyroCatcher(func, default=None): """ A function that catch a Pyro exceptions, write error to logger and return default value when it happen """ def catcher_func(*args, **kwargs): try: return func(*args, **kwargs) except Pyro.errors.ConnectionClosedError as e: confnodesroot._SetConnector(None) confnodesroot.logger.write_error(_("Connection lost!\n")) except Pyro.errors.ProtocolError as e: confnodesroot.logger.write_error(_("Pyro exception: %s\n") % e) except Exception as e: # confnodesroot.logger.write_error(traceback.format_exc()) errmess = ''.join(Pyro.util.getPyroTraceback(e)) confnodesroot.logger.write_error(errmess + "\n") print(errmess) confnodesroot._SetConnector(None) return default return catcher_func # Check connection is effective. # lambda is for getattr of GetPLCstatus to happen inside catcher IDPSK = PyroCatcher(RemotePLCObjectProxy.GetPLCID)() if IDPSK is None: confnodesroot.logger.write_warning(_("PLC did not provide identity and security infomation.\n")) else: ID, secret = IDPSK PSK.UpdateID(confnodesroot.ProjectPath, ID, secret, uri) _special_return_funcs = { "StartPLC": False, "GetTraceVariables": (PlcStatus.Broken, None), "GetPLCstatus": (PlcStatus.Broken, None), "RemoteExec": (-1, "RemoteExec script failed!") } class PyroProxyProxy(object): """ A proxy proxy class to handle Beremiz Pyro interface specific behavior. And to put Pyro exception catcher in between caller and Pyro proxy """ def __getattr__(self, attrName): member = self.__dict__.get(attrName, None) if member is None: def my_local_func(*args, **kwargs): return RemotePLCObjectProxy.__getattr__(attrName)(*args, **kwargs) member = PyroCatcher(my_local_func, _special_return_funcs.get(attrName, None)) self.__dict__[attrName] = member return member return PyroProxyProxy