#!@PYTHON@
# -*-python-*-
-# $Id$
-
# This file is part of avahi.
#
# avahi is free software; you can redistribute it and/or modify it
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
# USA.
-import sys, getopt
+import sys, getopt, os
try:
import avahi, gobject, dbus
except ImportError:
- print "Sorry, to use this tool you need to install Avahi, pygtk and python-dbus."
+ print "Sorry, to use this tool you need to install Avahi and python-dbus."
sys.exit(1)
try:
except ImportError:
pass
-try:
- from twisted.internet import gtk2reactor
- gtk2reactor.install()
- from twisted.internet import reactor
- from twisted.web import server, resource
-except ImportError:
- 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
+use_host_names = None
+use_CGI = None
domain = "local"
+timeout = 3000
-class AvahiBookmarks(resource.Resource):
- isLeaf = True
-
+class AvahiBookmarks:
services = {}
- def __init__(self):
- resource.Resource.__init__(self)
+ def __init__(self, use_host_names):
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.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")
+ self.browse_service_type("_https._tcp")
+ self.browse_service_type("_ftp._tcp")
+
+ if use_host_names is None:
+ try:
+ self.use_host_names = self.server.IsNSSSupportAvailable()
+ except:
+ self.use_host_names = False
+ else:
+ self.use_host_names = use_host_names
def browse_service_type(self, stype):
- global domain
+ global domain, use_CGI
browser = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, self.server.ServiceBrowserNew(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, stype, domain, dbus.UInt32(0))), avahi.DBUS_INTERFACE_SERVICE_BROWSER)
browser.connect_to_signal('ItemNew', self.new_service)
browser.connect_to_signal('ItemRemove', self.remove_service)
+ if use_CGI:
+ browser.connect_to_signal('AllForNow', self.all_for_now)
def find_path(self, txt):
return "/"
- def render_GET(self, request):
+ def render_html(self):
+
+ global domain
- t = '<html><head><title>Zeroconf Bookmarks</title></head><body><h1>Zeroconf Bookmarks</h1>'
+ t = '<html><head><title>%s Zeroconf Bookmarks</title></head><body><h1>%s Zeroconf Bookmarks</h1>' % (domain, domain)
if len(self.services) == 0:
- t += '<p>Sorry, no web services have been registered on the local LAN.</p>'
+ t += '<p>Sorry, no Zeroconf web services have been registered on the %s domain.</p>' % domain
else:
t += '<ul style="padding: 0px; margin: 20px; list-style-type: none">'
t += '<hr noshade/><p style="font-size: 8; font-family: sans-serif">Served by %s</p></body></html>' % self.version_string
- return t
+ return str(t)
def new_service(self, interface, protocol, name, type, domain, flags):
interface, protocol, name, type, domain, host, aprotocol, address, port, txt, flags = self.server.ResolveService(interface, protocol, name, type, domain, avahi.PROTO_UNSPEC, dbus.UInt32(0))
- if use_host_names:
+ if self.use_host_names:
h = host
else:
if aprotocol == avahi.PROTO_INET6:
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)]
+ # Only reachable with use_CGI
+ def all_for_now(self):
+
+ mainloop.quit()
+
def usage(retval = 0):
+
print "%s [options]\n" % sys.argv[0]
print " -h --help Show this help"
+ print " -c --cgi Run as a CGI instead of as a server (default to server"
+ print " unless environment variable GATEWAY_INTERFACE is set)"
+ print " -t --timeout MS Specify the max time for CGI browsing (default %u)" % timeout
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 " -H --host-names Show links with real hostnames"
+ print " -A --addresses Show links with numeric IP addresses"
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="])
+ opts, args = getopt.getopt(sys.argv[1:], "hct:p:a:HAd:", ["help", "cgi", "port=", "timeout=", "address=", "host-names", "addresses", "domain="])
except getopt.GetoptError:
usage(2)
if o in ("-h", "--help"):
usage()
+ if o in ("-c", "--cgi"):
+ use_CGI = True
+
+ if o in ("-t", "--timeout"):
+ timeout = int(a)
+
if o in ("-p", "--port"):
port = int(a)
if o in ("-H", "--host-names"):
use_host_names = True
+ if o in ("-A", "--addresses"):
+ use_host_names = False
+
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)
+if use_CGI is None:
+ use_CGI = os.environ.has_key("GATEWAY_INTERFACE")
-try:
- reactor.run()
-except KeyboardInterrupt, k:
- pass
+if use_CGI:
+ cgi = AvahiBookmarks(use_host_names)
+
+ mainloop = gobject.MainLoop()
+ gobject.timeout_add(timeout, mainloop.quit)
+
+ try:
+ mainloop.run()
+ except KeyboardInterrupt:
+ pass
+
+ print 'Content-type: text/html\n\n' + cgi.render_html()
+
+else:
+ try:
+ from twisted.internet import glib2reactor
+ glib2reactor.install()
+ from twisted.internet import reactor
+ from twisted.web import server, resource
+ except ImportError:
+ print "Sorry, to use this tool as a server you need to install twisted and twisted.web.\n"
+ sys.exit(1)
+
+ class AvahiBookmarksServer(AvahiBookmarks, resource.Resource):
+ isLeaf = True
+
+ def __init__(self, use_host_names):
+ resource.Resource.__init__(self)
+ AvahiBookmarks.__init__(self, use_host_names)
+
+ def render_GET(self, request):
+ return self.render_html()
+
+ site = server.Site(AvahiBookmarksServer(use_host_names))
+ reactor.listenTCP(port, site, interface=address)
+
+ print "Now point your web browser to http://%s:%u/!" % (address, port)
+
+ try:
+ reactor.run()
+ except KeyboardInterrupt:
+ pass
projects / catta / blobdiff