1 #!/usr/bin/env python

     2 # -*- coding: utf-8 -*-

     3 

     4 # This file is part of Beremiz runtime.

     5 

     6 # Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     7 # Copyright (C) 2017: Andrey Skvortsov

     8 # Copyright (C) 2018: Edouard TISSERANT

     9 

    10 # See COPYING file for copyrights details.

    11 

    12 

    13 

    14 import sys

    15 import os

    16 

    17 import Pyro5

    18 import Pyro5.server

    19 

    20 import runtime

    21 from runtime.ServicePublisher import ServicePublisher

    22 

    23 Pyro5.config.SERIALIZER = "msgpack"

    24 

    25 def make_pyro_exposed_stub(method_name):

    26     stub = lambda self, *args, **kwargs: \

    27         getattr(self.plc_object_instance, method_name)(*args, **kwargs)

    28     stub.__name__ = method_name

    29     Pyro5.server.expose(stub)

    30     return stub

    31     

    32 

    33 class PLCObjectPyroAdapter(type("PLCObjectPyroStubs", (), {

    34     name: make_pyro_exposed_stub(name) for name in [

    35         "AppendChunkToBlob",

    36         "GetLogMessage",

    37         "GetPLCID",

    38         "GetPLCstatus",

    39         "GetTraceVariables",

    40         "MatchMD5", 

    41         "NewPLC",

    42         "PurgeBlobs",

    43         "RemoteExec",

    44         "RepairPLC",

    45         "ResetLogCount",

    46         "SeedBlob",

    47         "SetTraceVariablesList",

    48         "StartPLC",

    49         "StopPLC"

    50     ]

    51 })):

    52     def __init__(self, plc_object_instance):

    53         self.plc_object_instance = plc_object_instance

    54     

    55 

    56 class PyroServer(object):

    57     def __init__(self, servicename, ip_addr, port):

    58         self.continueloop = True

    59         self.daemon = None

    60         self.servicename = servicename

    61         self.ip_addr = ip_addr

    62         self.port = port

    63         self.servicepublisher = None

    64         self.piper, self.pipew = None, None

    65 

    66     def _to_be_published(self):

    67         return self.servicename is not None and \

    68                self.ip_addr not in ["", "localhost", "127.0.0.1"]

    69 

    70     def PrintServerInfo(self):

    71         print(_("Pyro port :"), self.port)

    72 

    73         if self._to_be_published():

    74             print(_("Publishing service on local network"))

    75 

    76         if sys.stdout:

    77             sys.stdout.flush()

    78 

    79     def PyroLoop(self, when_ready):

    80         if self._to_be_published():

    81             self.Publish()

    82 

    83         while self.continueloop:

    84             self.daemon = Pyro5.server.Daemon(host=self.ip_addr, port=self.port)

    85 

    86             self.daemon.register(PLCObjectPyroAdapter(runtime.GetPLCObjectSingleton()), "PLCObject")

    87 

    88             when_ready()

    89 

    90             self.daemon.requestLoop()

    91 

    92         self.Unpublish()

    93 

    94     def Restart(self):

    95         self.daemon.shutdown(True)

    96 

    97     def Quit(self):

    98         self.continueloop = False

    99         self.daemon.shutdown()

   100         if not sys.platform.startswith('win'):

   101             if self.pipew is not None:

   102                 os.write(self.pipew, "goodbye")

   103 

   104     def Publish(self):

   105         self.servicepublisher = ServicePublisher("PYRO")

   106         self.servicepublisher.RegisterService(self.servicename,

   107                                               self.ip_addr, self.port)

   108 

   109     def Unpublish(self):

   110         if self.servicepublisher is not None:

   111             self.servicepublisher.UnRegisterService()

   112             self.servicepublisher = None