#!/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 sys

import os

import traceback

from functools import partial

from threading import Thread, Event

from six import text_type as text

from twisted.internet import reactor, threads

from twisted.internet._sslverify import OpenSSLCertificateAuthorities

from twisted.internet.ssl import optionsForClientTLS, VerificationError

from autobahn.twisted import wamp

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

from autobahn.wamp import types, auth

from autobahn.wamp.exception import TransportLost

from autobahn.wamp.serializer import MsgPackSerializer

from OpenSSL import crypto

from connectors.ConnectorBase import ConnectorBase

import PSKManagement as PSK

import CertManagement as Cert

_WampSession = None

_WampConnection = None

_WampConnectEvent = Event()

_WampError = ""

class WampSession(wamp.ApplicationSession):

def onConnect(self):

user = self.config.extra["IDE_ID"]

self.join(unicode(self.config.realm), [u"wampcra"], unicode(user))

def onChallenge(self, challenge):

if challenge.method == "wampcra":

secret = self.config.extra["secret"]

if 'salt' in challenge.extra:

# salted secret

key = auth.derive_key(secret,

challenge.extra['salt'],

challenge.extra['iterations'],

challenge.extra['keylen'])

else:

# plain, unsalted secret

key = secret

signature = auth.compute_wcs(key, challenge.extra['challenge'])

return signature

else:

raise Exception("Invalid authmethod {}".format(challenge.method))

def onJoin(self, details):

global _WampSession, _WampConnectEvent

_WampSession = self

_WampConnectEvent.set()

print('WAMP session joined for: ', self.config.extra["IDE_ID"])

def onLeave(self, details):

global _WampSession, _WampError, _WampConnectEvent

_WampSession = None

if details.reason == "wamp.close.normal":

_WampError = "Closed normally"

elif details.reason == "wamp.error.not_authorized":

_WampError = "WAMP authentication failed. Check IDE identity in security manager."

else:

_WampError = "WAMP closed with error {}: {}".format(details.reason, details.message)

_WampConnectEvent.set()

class ComplainingWampWebSocketClientFactory(WampWebSocketClientFactory):

def clientConnectionLost(self, connector, reason):

global _WampError, _WampConnectEvent, _WampSession

if not reason.check(VerificationError):

# Verification failed

_WampError = "WAMP TLS certificate verification failed. "+\

"Provide valid certicate in identity manager."

else:

_WampError = "WAMP connection lost: "+reason.getErrorMessage()

_WampSession = None

_WampConnectEvent.set()

clientConnectionFailed = clientConnectionLost

def _WAMP_connector_factory(cls, uri, confnodesroot):

"""

WAMP://127.0.0.1:12345/path#realm#PLC_ID

WAMPS://127.0.0.1:12345/path#realm#PLC_ID

"""

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

urlpath, realm, PLC_ID = location.split('#')

urlprefix = {"WAMP": "ws",

"WAMPS": "wss"}[scheme]

url = urlprefix+"://"+urlpath

CN = urlpath.split("/")[0].split(":")[0]

try:

IDE_ID, secret = PSK.GetIDEIdentity()

trust_store = Cert.GetCertPath(CN)

except Exception as e:

confnodesroot.logger.write_error(

_("Connection to {loc} failed with exception {ex}\n").format(

loc=uri, ex=str(e)))

return None

def RegisterWampClient():

# start logging to console

# log.startLogging(sys.stdout)

# create a WAMP application session factory

component_config = types.ComponentConfig(

realm=text(realm),

extra={

"IDE_ID": IDE_ID,

"secret": secret

})

session_factory = wamp.ApplicationSessionFactory(

config=component_config)

session_factory.session = cls

# create a WAMP-over-WebSocket transport client factory

transport_factory = ComplainingWampWebSocketClientFactory(

session_factory,

url=url,

serializers=[MsgPackSerializer()])

contextFactory=None

if transport_factory.isSecure:

trustRoot=None

if trust_store:

if os.path.exists(trust_store):

cert = crypto.load_certificate(

crypto.FILETYPE_PEM,

open(trust_store, 'rb').read()

)

trustRoot=OpenSSLCertificateAuthorities([cert])

else:

confnodesroot.logger.write_warning("Wamp trust store not found")

contextFactory = optionsForClientTLS(unicode(transport_factory.host), trustRoot=trustRoot)

# start the client from a Twisted endpoint

conn = connectWS(transport_factory, contextFactory)

confnodesroot.logger.write(_("WAMP connecting to URL : %s\n") % url)

return conn

AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

def ThreadProc():

global _WampConnection

_WampConnection = RegisterWampClient()

AddToDoBeforeQuit(reactor.stop)

reactor.run(installSignalHandlers=False)

global _WampConnection, _WampSession, _WampConnectEvent, _WampError

_WampConnectEvent.clear()

if not reactor.running:

Thread(target=ThreadProc).start()

else:

_WampConnection = threads.blockingCallFromThread(

reactor, RegisterWampClient)

if not _WampConnectEvent.wait(4):

confnodesroot.logger.write_error("WAMP connection timeout\n")

try:

threads.blockingCallFromThread(

reactor, _WampConnection.stopConnecting)

except:

pass

return None

else:

if _WampSession is None:

confnodesroot.logger.write_error("WAMP connection failed: {}\n".format(_WampError))

return None

class WampPLCObjectProxy(ConnectorBase):

def __del__(self):

_WampConnection.disconnect()

#

# reactor.stop()

def WampSessionProcMapper(self, funcname):

wampfuncname = text('.'.join((PLC_ID, funcname)))

def catcher_func(*args, **kwargs):

if _WampSession is not None:

try:

return threads.blockingCallFromThread(

reactor, _WampSession.call, wampfuncname,

*args, **kwargs)

except TransportLost:

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

confnodesroot._SetConnector(None)

except Exception:

errmess = traceback.format_exc()

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

print(errmess)

# confnodesroot._SetConnector(None)

return self.PLCObjDefaults.get(funcname)

return catcher_func

def __getattr__(self, attrName):

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

if member is None:

member = self.WampSessionProcMapper(attrName)

self.__dict__[attrName] = member

return member

# TODO : GetPLCID()

# TODO : PSK.UpdateID()

return WampPLCObjectProxy()

WAMP_connector_factory = partial(_WAMP_connector_factory, WampSession)