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

def make_pyro_exposed_stub(method_name):

    24

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

    25

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

    26

    stub.__name__ = method_name

    27

    Pyro5.server.expose(stub)

    28

    return stub

    29

    30

    31

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

    32

    name: make_pyro_exposed_stub(name) for name in [

    33

        "AppendChunkToBlob",

    34

        "GetLogMessage",

    35

        "GetPLCID",

    36

        "GetPLCstatus",

    37

        "GetTraceVariables",

    38

        "MatchMD5", 

    39

        "NewPLC",

    40

        "PurgeBlobs",

    41

        "RemoteExec",

    42

        "RepairPLC",

    43

        "ResetLogCount",

    44

        "SeedBlob",

    45

        "SetTraceVariablesList",

    46

        "StartPLC",

    47

        "StopPLC"

    48

    ]

    49

})):

    50

    def __init__(self, plc_object_instance):

    51

        self.plc_object_instance = plc_object_instance

    52

    53

    54

class PyroServer(object):

    55

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

    56

        self.continueloop = True

    57

        self.daemon = None

    58

        self.servicename = servicename

    59

        self.ip_addr = ip_addr

    60

        self.port = port

    61

        self.servicepublisher = None

    62

        self.piper, self.pipew = None, None

    63

    64

    def _to_be_published(self):

    65

        return self.servicename is not None and \

    66

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

    67

    68

    def PrintServerInfo(self):

    69

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

    70

    71

        if self._to_be_published():

    72

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

    73

    74

        sys.stdout.flush()

    75

    76

    def PyroLoop(self, when_ready):

    77

        if self._to_be_published():

    78

            self.Publish()

    79

    80

        while self.continueloop:

    81

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

    82

    83

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

    84

    85

            when_ready()

    86

    87

            self.daemon.requestLoop()

    88

    89

        self.Unpublish()

    90

    91

    def Restart(self):

    92

        self.daemon.shutdown(True)

    93

    94

    def Quit(self):

    95

        self.continueloop = False

    96

        self.daemon.shutdown()

    97

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

    98

            if self.pipew is not None:

    99

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

   100

   101

    def Publish(self):

   102

        self.servicepublisher = ServicePublisher("PYRO")

   103

        self.servicepublisher.RegisterService(self.servicename,

   104

                                              self.ip_addr, self.port)

   105

   106

    def Unpublish(self):

   107

        if self.servicepublisher is not None:

   108

            self.servicepublisher.UnRegisterService()

   109

            self.servicepublisher = None