#!@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 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." sys.exit(1) try: import dbus.glib except ImportError: pass urlproto = { "_http._tcp" : "http", "_https._tcp" : "https", "_ftp._tcp" : "ftp" } port = 8080 address = "127.0.0.1" use_host_names = None use_CGI = None domain = "local" timeout = 3000 class AvahiBookmarks: services = {} 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") 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, 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): l = avahi.txt_array_to_string_array(txt) for k in l: if k[:5] == "path=": if k[5:].startswith("/"): return k[5:] else: return "/" + k[5:] return "/" def render_html(self): global domain t = '%s Zeroconf Bookmarks

%s Zeroconf Bookmarks

' % (domain, domain) if len(self.services) == 0: t += '

Sorry, no Zeroconf web services have been registered on the %s domain.

' % domain else: t += '' t += '

Served by %s

' % self.version_string 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 self.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)] # 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 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:], "hct:p:a:HAd:", ["help", "cgi", "port=", "timeout=", "address=", "host-names", "addresses", "domain="]) except getopt.GetoptError: usage(2) for o, a in opts: 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 ("-a", "--address"): address = 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 if use_CGI is None: use_CGI = os.environ.has_key("GATEWAY_INTERFACE") 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