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

#

     8

# See COPYING.Runtime file for copyrights details.

     9

#

    10

# This library is free software; you can redistribute it and/or

    11

# modify it under the terms of the GNU Lesser General Public

    12

# License as published by the Free Software Foundation; either

    13

# version 2.1 of the License, or (at your option) any later version.

    14

    15

# This library is distributed in the hope that it will be useful,

    16

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    17

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    18

# Lesser General Public License for more details.

    19

    20

# You should have received a copy of the GNU Lesser General Public

    21

# License along with this library; if not, write to the Free Software

    22

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

    23

    24

    25

    26

    27

import socket

    28

import threading

    29

import zeroconf

    30

    31

    32

service_type = '_Beremiz._tcp.local.'

    33

    34

    35

class ServicePublisher(object):

    36

    def __init__(self, protocol):

    37

        self.serviceproperties = {

    38

            'description': 'Beremiz remote PLC',

    39

            'protocol': protocol

    40

        }

    41

    42

        self.name = None

    43

        self.ip_32b = None

    44

        self.port = None

    45

        self.server = None

    46

        self.service_name = None

    47

        self.retrytimer = None

    48

    49

    def RegisterService(self, name, ip, port):

    50

        try:

    51

            self._RegisterService(name, ip, port)

    52

        except Exception as e:

    53

            print(f"Failed to register service ({str(e)}), retrying in 2 seconds")

    54

            self.retrytimer = threading.Timer(2, self.RegisterService, [name, ip, port])

    55

            self.retrytimer.start()

    56

    57

    def _RegisterService(self, name, ip, port):

    58

        self.service_name = '%s.%s' % (name, service_type)

    59

        self.name = name

    60

        self.port = port

    61

    62

        if ip == "0.0.0.0":

    63

            print("MDNS brodcasted on all interfaces")

    64

            interfaces = zeroconf.InterfaceChoice.All

    65

            ip = self.gethostaddr()

    66

        else:

    67

            interfaces = [ip]

    68

    69

        self.server = zeroconf.Zeroconf(interfaces=interfaces)

    70

    71

        print("MDNS brodcasted service address :" + ip)

    72

        self.ip_32b = socket.inet_aton(ip)

    73

    74

        self.service_info = zeroconf.ServiceInfo(

    75

            service_type,

    76

            self.service_name,

    77

            self.port,

    78

            addresses=[self.ip_32b],

    79

            properties=self.serviceproperties)

    80

    81

        self.server.register_service(self.service_info)

    82

        self.retrytimer = None

    83

    84

    def UnRegisterService(self):

    85

        if self.retrytimer is not None:

    86

            self.retrytimer.cancel()

    87

    88

        if self.server is not None:

    89

            self.server.unregister_service(self.service_info)

    90

            self.server.close()

    91

            self.server = None

    92

    93

    def gethostaddr(self, dst='224.0.1.41'):

    94

        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    95

        try:

    96

            s.connect((dst, 7))

    97

            (host, _port) = s.getsockname()

    98

            s.close()

    99

            if host != '0.0.0.0':

   100

                return host

   101

        except Exception:

   102

            pass

   103

        return socket.gethostbyname(socket.gethostname())