1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#!/usr/bin/env @PYTHON@
# -*-python-*-
# $Id$
# This file is part of avahi.
#
# avahi is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation; either version 2 of the
# License, or (at your option) any later version.
#
# avahi 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 Lesser General Public
# License along with avahi; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
# USA.
import avahi, dbus, gobject, sys, getopt
try:
import dbus.glib
except ImportError, e:
pass
def usage(retval = 0):
print "%s [options] <name> <type> <port> [<txt> ...]\n" % sys.argv[0]
print " -h --help Show this help"
print " -d --domain Domain where to register this service"
print " -H --host Host where this service resides"
sys.exit(retval)
try:
opts, args = getopt.getopt(sys.argv[1:], "d:H:", ["help", "domain=", "host="])
except getopt.GetoptError:
usage(2)
domain = ""
host = ""
for o, a in opts:
if o in ("-h", "--help"):
usage()
if o in ("-d", "--domain"):
domain = a
if o in ("-H", "--host"):
host = a
if len(args) < 3:
sys.stderr.write("Invalid number of arguments\n")
sys.exit(1)
name = args[0]
stype = args[1]
port = int(args[2])
txt = args[3:]
# python-dbus doesn't allow transmission of empty arrays, therefore we "fix" it with a bogus entry
if len(txt) == 0:
txt.append("python-dbus=brain-damage")
group = None
n_rename = 0
def remove_service():
global group
if not group is None:
group.Reset()
def add_service():
global server, group, name, stype, domain, host, port, txt
if group is None:
group = dbus.Interface(bus.get_object(avahi.DBUS_NAME, server.EntryGroupNew()), avahi.DBUS_INTERFACE_ENTRY_GROUP)
assert group.IsEmpty()
print "Adding service '%s' of type '%s' ..." % (name, stype)
group.connect_to_signal('StateChanged', entry_group_state_changed)
group.AddService(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, name, stype, domain, host, dbus.UInt16(port), txt)
group.Commit()
def entry_group_state_changed(state):
global name, server, n_rename
if state == avahi.ENTRY_GROUP_ESTABLISHED:
print "Service established."
elif state == avahi.ENTRY_GROUP_COLLISION:
n_rename = n_rename + 1
if n_rename >= 12:
print "ERROR: No suitable service name found after %i retries, exiting." % n_rename
main_loop.quit()
else:
name = server.GetAlternativeServiceName(name)
print "WARNING: Service name collision, changing name to '%s' ..." % name
remove_service()
add_service()
def server_state_changed(state):
if state == avahi.SERVER_COLLISION:
print "WARNING: Server name collision"
remove_service()
elif state == avahi.SERVER_RUNNING:
add_service()
main_loop = gobject.MainLoop()
bus = dbus.SystemBus()
server = dbus.Interface(bus.get_object(avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER), avahi.DBUS_INTERFACE_SERVER)
server.connect_to_signal("StateChanged", server_state_changed)
server_state_changed(server.GetState())
try:
main_loop.run()
except KeyboardInterrupt, k:
pass
if not group is None:
group.Free()
|