| #!@PYTHON@ |
| # -*-python-*- |
| # 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 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 = '<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 Zeroconf web services have been registered on the %s domain.</p>' % domain |
| else: |
| t += '<ul style="padding: 0px; margin: 20px; list-style-type: none">' |
| |
| for k, v in self.services.iteritems(): |
| |
| if v[3] == 80: |
| port = '' |
| else: |
| port = ':%i' % v[3] |
| |
| path = self.find_path(v[4]) |
| t += '<li><a href="%s://%s%s%s">%s</a></li>' % (urlproto[k[3]], v[2], port, path, k[2]) |
| |
| t += '</ul>' |
| |
| t += '<hr noshade/><p style="font-size: 8; font-family: sans-serif">Served by %s</p></body></html>' % 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 |
