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

import socket

    25

import threading

    26

from util import Zeroconf

    27

    28

service_type = '_PYRO._tcp.local.'

    29

    30

    31

class ServicePublisher():

    32

    def __init__(self):

    33

        # type: fully qualified service type name

    34

        self.serviceproperties = {'description': 'Beremiz remote PLC'}

    35

    36

        self.name = None

    37

        self.ip_32b = None

    38

        self.port = None

    39

        self.server = None

    40

        self.service_name = None

    41

        self.retrytimer = None

    42

    43

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

    44

        try:

    45

            self._RegisterService(name, ip, port)

    46

        except Exception, e:

    47

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

    48

            self.retrytimer.start()

    49

    50

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

    51

        # name: fully qualified service name

    52

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

    53

        self.name = name

    54

        self.port = port

    55

    56

        self.server = Zeroconf.Zeroconf(ip)

    57

        print "MDNS brodcasting on :"+ip

    58

    59

        if ip == "0.0.0.0":

    60

            ip = self.gethostaddr()

    61

        print "MDNS brodcasted service address :"+ip

    62

        self.ip_32b = socket.inet_aton(ip)

    63

    64

        self.server.registerService(

    65

             Zeroconf.ServiceInfo(service_type,

    66

                                  self.service_name,

    67

                                  self.ip_32b,

    68

                                  self.port,

    69

                                  properties=self.serviceproperties))

    70

        self.retrytimer = None

    71

    72

    def UnRegisterService(self):

    73

        if self.retrytimer is not None:

    74

            self.retrytimer.cancel()

    75

    76

        self.server.unregisterService(

    77

                                      Zeroconf.ServiceInfo(service_type,

    78

                                                           self.service_name,

    79

                                                           self.ip_32b,

    80

                                                           self.port,

    81

                                                           properties=self.serviceproperties))

    82

        self.server.close()

    83

        self.server = None

    84

    85

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

    86

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

    87

        try:

    88

            s.connect((dst, 7))

    89

            (host, port) = s.getsockname()

    90

            s.close()

    91

            if host != '0.0.0.0':

    92

                return host

    93

        except Exception, e:

    94

            pass

    95

        return socket.gethostbyname(socket.gethostname())