X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;ds=inline;f=avahi-python%2Favahi-bookmarks.in;h=a1e8d0ce6ab3ca5b47baf6e4ad9e7e734ac36d13;hb=8b72620bc14e34d2a07ba0fc0159db62d7b50be0;hp=2de9186f535d271024e614fba8c2dfb6bc261944;hpb=ced0d86acf70f435ffdadb794db891be63a5b82a;p=catta diff --git a/avahi-python/avahi-bookmarks.in b/avahi-python/avahi-bookmarks.in index 2de9186..a1e8d0c 100755 --- a/avahi-python/avahi-bookmarks.in +++ b/avahi-python/avahi-bookmarks.in @@ -19,7 +19,7 @@ # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA. -import sys, getopt +import sys, getopt, os try: import avahi, gobject, dbus @@ -32,29 +32,19 @@ 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) @@ -62,18 +52,26 @@ class AvahiBookmarks(resource.Resource): 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): @@ -88,12 +86,14 @@ class AvahiBookmarks(resource.Resource): return "/" - def render_GET(self, request): + def render_html(self): - t = '
Sorry, no web services have been registered on the local LAN.
' + t += 'Sorry, no Zeroconf web services have been registered on the %s domain.
' % domain else: t += 'Served by %s
' % 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: @@ -129,20 +129,31 @@ class AvahiBookmarks(resource.Resource): 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) @@ -150,6 +161,12 @@ 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) @@ -159,15 +176,54 @@ for o, a in opts: 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