# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
# USA.
-import sys
+import sys, getopt
try:
import avahi, gobject, dbus
from twisted.internet import reactor
from twisted.web import server, resource
except ImportError:
- print "Sorry, to use this tool you need to install twisted."
+ print "Sorry, to use this tool you need to install twisted and twisted.web."
sys.exit(1)
+urlproto = { "_http._tcp" : "http", "_https._tcp" : "https", "_ftp._tcp" : "ftp" }
+
+port = 8080
+address = "127.0.0.1"
+use_host_names = False
+domain = "local"
+
class AvahiBookmarks(resource.Resource):
isLeaf = True
def __init__(self):
resource.Resource.__init__(self)
-
self.bus = dbus.SystemBus()
self.server = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER), avahi.DBUS_INTERFACE_SERVER)
self.version_string = self.server.GetVersionString()
- self.browser = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, self.server.ServiceBrowserNew(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, "_http._tcp", "local")), avahi.DBUS_INTERFACE_SERVICE_BROWSER)
+ self.browse_service_type("_http._tcp")
+
+ # Hurrah! if I enable one of the following lines, python segfaults.
+ #self.browse_service_type("_https._tcp")
+ #self.browse_service_type("_ftp._tcp")
+
+ def browse_service_type(self, stype):
- self.browser.connect_to_signal('ItemNew', self.new_service)
- self.browser.connect_to_signal('ItemRemove', self.remove_service)
+ global domain
+
+ browser = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, self.server.ServiceBrowserNew(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, stype, domain, 0)), avahi.DBUS_INTERFACE_SERVICE_BROWSER)
+ browser.connect_to_signal('ItemNew', self.new_service)
+ browser.connect_to_signal('ItemRemove', self.remove_service)
def find_path(self, txt):
- for k in txt:
+ l = avahi.txt_array_to_string_array(txt)
+
+ for k in l:
if k[:5] == "path=":
- return k[5:]
+ if k[5:].startswith("/"):
+ return k[5:]
+ else:
+ return "/" + k[5:]
return "/"
port = ':%i' % v[3]
path = self.find_path(v[4])
-
-
- t += '<li><a href="http://%s%s%s">%s</a></li>' % (v[2], port, path, k[2])
+ t += '<li><a href="%s://%s%s%s">%s</a></li>' % (urlproto[k[3]], v[2], port, path, k[2])
t += '</ul>'
return t
- def new_service(self, interface, protocol, name, type, domain):
+ def new_service(self, interface, protocol, name, type, domain, flags):
- interface, protocol, name, type, domain, host, aprotocol, address, port, txt = self.server.ResolveService(interface, protocol, name, type, domain, avahi.PROTO_UNSPEC)
+ interface, protocol, name, type, domain, host, aprotocol, address, port, txt, flags = self.server.ResolveService(interface, protocol, name, type, domain, avahi.PROTO_UNSPEC, 0)
- self.services[(interface, protocol, name, type, domain)] = (host, aprotocol, address, port, txt)
+ if use_host_names:
+ h = host
+ else:
+ if aprotocol == avahi.PROTO_INET6:
+ h = "[" + address + "]"
+ else:
+ h = address
+
+ self.services[(interface, protocol, name, type, domain)] = (host, aprotocol, h, port, txt)
def remove_service(self, interface, protocol, name, type, domain):
del self.services[(interface, protocol, name, type, domain)]
-port = 8080
+def usage(retval = 0):
+ print "%s [options]\n" % sys.argv[0]
+ print " -h --help Show this help"
+ print " -p --port PORT Specify the port to use (default %u)" % port
+ print " -a --address ADDRESS Specify the address to bind to (default %s)" % address
+ print " -H --host-names Show all services, regardless of the type"
+ print " -d --domain DOMAIN Specify the domain to browse"
+ sys.exit(retval)
+
+try:
+ opts, args = getopt.getopt(sys.argv[1:], "hp:a:Hd:", ["help", "port=", "address=", "host-names", "domain="])
+except getopt.GetoptError:
+ usage(2)
+
+for o, a in opts:
+ if o in ("-h", "--help"):
+ usage()
-if __name__ == '__main__':
- site = server.Site(AvahiBookmarks())
- reactor.listenTCP(port, site)
+ if o in ("-p", "--port"):
+ port = int(a)
- print "Now point your web browser to http://localhost:%u/!" % port
+ if o in ("-a", "--address"):
+ address = a
- try:
- reactor.run()
- except KeyboardInterrupt, k:
- pass
+ if o in ("-H", "--host-names"):
+ use_host_names = True
+
+ if o in ("-d", "--domain"):
+ domain = a
+
+site = server.Site(AvahiBookmarks())
+reactor.listenTCP(port, site, interface=address)
+
+print "Now point your web browser to http://%s:%u/!" % (address, port)
+
+try:
+ reactor.run()
+except KeyboardInterrupt, k:
+ pass