add avahi-browse.c

git-svn-id: file:///home/lennart/svn/public/avahi/trunk@880 941a03a8-eaeb-0310-b9a0-b1bbd8fe43fe

1 files changed, 593 insertions, 0 deletions

diff --git a/avahi-utils/avahi-browse.c b/avahi-utils/avahi-browse.c
new file mode 100644
index 0000000..09792a4
--- /dev/null
+++ b/avahi-utils/avahi-browse.c
@@ -0,0 +1,593 @@
+/* $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.1 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 Lesser 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.
+***/
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <getopt.h>
+#include <assert.h>
+#include <string.h>
+#include <net/if.h>
+
+#include <avahi-common/simple-watch.h>
+#include <avahi-common/error.h>
+#include <avahi-common/malloc.h>
+#include <avahi-common/domain.h>
+#include <avahi-common/llist.h>
+#include <avahi-client/client.h>
+#include <avahi-client/lookup.h>
+
+#include "sigint.h"
+
+typedef enum {
+    COMMAND_HELP,
+    COMMAND_VERSION,
+    COMMAND_RUN
+} Command;
+
+typedef struct Config {
+    int verbose;
+    int terminate_on_all_for_now;
+    int terminate_on_cache_exhausted;
+    char *domain;
+    char *stype;
+    int ignore_local;
+    int show_all;
+    Command command;
+    int resolve;
+} Config;
+
+
+typedef struct ServiceInfo ServiceInfo;
+
+struct ServiceInfo {
+    AvahiIfIndex interface;
+    AvahiProtocol protocol;
+    char *name, *type, *domain;
+
+    AvahiServiceResolver *resolver;
+    Config *config;
+
+    AVAHI_LLIST_FIELDS(ServiceInfo, info);
+};
+
+static AvahiSimplePoll *simple_poll = NULL;
+static AvahiClient *client = NULL;
+static int n_all_for_now = 0, n_cache_exhausted = 0, n_resolving = 0;
+static AvahiStringList *browsed_types = NULL;
+static ServiceInfo *services = NULL;
+static int n_columns = 80;
+
+static void check_terminate(Config *c) {
+
+    assert(n_all_for_now >= 0);
+    assert(n_cache_exhausted >= 0);
+    assert(n_resolving >= 0);
+    
+    if (n_all_for_now <= 0 && n_resolving <= 0) {
+
+        if (c->verbose) {
+            printf(": All for now\n");
+            n_all_for_now++; /* Make sure that this event is not repeated */
+        }
+        
+        if (c->terminate_on_all_for_now)
+            avahi_simple_poll_quit(simple_poll);
+    }
+    
+    if (n_cache_exhausted <= 0 && n_resolving <= 0) {
+
+        if (c->verbose) {
+            printf(": Cache exhausted\n");
+            n_cache_exhausted++; /* Make sure that this event is not repeated */
+        }
+        
+        if (c->terminate_on_cache_exhausted)
+            avahi_simple_poll_quit(simple_poll);
+    }
+}
+
+static ServiceInfo *find_service(AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char *domain) {
+    ServiceInfo *i;
+
+    for (i = services; i; i = i->info_next)
+        if (i->interface == interface &&
+            i->protocol == protocol &&
+            strcasecmp(i->name, name) == 0 &&
+            avahi_domain_equal(i->type, type) == 0 &&
+            avahi_domain_equal(i->domain, domain) == 0)
+
+            return i;
+
+    return NULL;
+}
+
+static void service_resolver_callback(
+    AvahiServiceResolver *r,
+    AvahiIfIndex interface,
+    AvahiProtocol protocol,
+    AvahiResolverEvent event,
+    const char *name,
+    const char *type,
+    const char *domain,
+    const char *host_name,
+    const AvahiAddress *a,
+    uint16_t port,
+    AvahiStringList *txt,
+    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
+    void *userdata) {
+    
+    ServiceInfo *i = userdata;
+    
+    assert(r);
+    assert(i);
+
+    switch (event) {
+        case AVAHI_RESOLVER_FOUND: {
+            char address[AVAHI_ADDRESS_STR_MAX], *t;
+            char ifname[IF_NAMESIZE];
+
+            avahi_address_snprint(address, sizeof(address), a);
+
+            t = avahi_string_list_to_string(txt);
+
+            printf("= %4s %4s %-*s %-20s %s\n"
+                   "   hostname = [%s]\n"
+                   "   address = [%s]\n"
+                   "   port = [%i]\n"
+                   "   txt = [%s]\n",
+                   if_indextoname(interface, ifname), avahi_proto_to_string(protocol), 
+                   n_columns-35, name, type, domain,
+                   host_name,
+                   address,
+                   port,
+                   t);
+            avahi_free(t);
+
+            break;
+        }
+
+        case AVAHI_RESOLVER_FAILURE:
+            
+            fprintf(stderr, "Failed to resolve service '%s' of type '%s' in domain '%s': %s\n", name, type, domain, avahi_strerror(avahi_client_errno(client)));
+            break;
+    }
+
+    
+    avahi_service_resolver_free(i->resolver);
+    i->resolver = NULL;
+
+    assert(n_resolving > 0);
+    n_resolving--;
+    check_terminate(i->config);
+}
+
+static ServiceInfo *add_service(Config *c, AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char *domain) {
+    ServiceInfo *i;
+
+    i = avahi_new(ServiceInfo, 1);
+
+    if (c->resolve) {
+        if (!(i->resolver = avahi_service_resolver_new(client, interface, protocol, name, type, domain, AVAHI_PROTO_UNSPEC, 0, service_resolver_callback, i))) {
+            avahi_free(i);
+            fprintf(stderr, "Failed to resolve service '%s' of type '%s' in domain '%s': %s\n", name, type, domain, avahi_strerror(avahi_client_errno(client)));
+            return NULL;
+        }
+
+        n_resolving++;
+    } else
+        i->resolver = NULL;
+
+    i->interface = interface;
+    i->protocol = protocol;
+    i->name = avahi_strdup(name);
+    i->type = avahi_strdup(type);
+    i->domain = avahi_strdup(domain);
+    i->config = c;
+
+    AVAHI_LLIST_PREPEND(ServiceInfo, info, services, i);
+
+    return i;
+}
+
+static void remove_service(Config *c, ServiceInfo *i) {
+    assert(c);
+    assert(i);
+
+    AVAHI_LLIST_REMOVE(ServiceInfo, info, services, i);
+
+    if (i->resolver)
+        avahi_service_resolver_free(i->resolver);
+    
+    avahi_free(i->name);
+    avahi_free(i->type);
+    avahi_free(i->domain);
+    avahi_free(i);
+}
+
+static void service_browser_callback(
+    AvahiServiceBrowser *b,
+    AvahiIfIndex interface,
+    AvahiProtocol protocol,
+    AvahiBrowserEvent event,
+    const char *name,
+    const char *type,
+    const char *domain,
+    AvahiLookupResultFlags flags,
+    void *userdata) {
+
+    Config *c = userdata;
+    
+    assert(b);
+    assert(c);
+
+    switch (event) {
+        case AVAHI_BROWSER_NEW: {
+            char ifname[IF_NAMESIZE];
+            if (c->ignore_local && (flags & AVAHI_LOOKUP_RESULT_LOCAL))
+                break;
+
+            if (find_service(interface, protocol, name, type, domain))
+                return;
+
+            add_service(c, interface, protocol, name, type, domain);
+
+            printf("+ %4s %4s %-*s %-20s %s\n", if_indextoname(interface, ifname), avahi_proto_to_string(protocol), n_columns-35, name, type, domain);
+            break;
+
+        }
+
+        case AVAHI_BROWSER_REMOVE: {
+            char ifname[IF_NAMESIZE];
+            ServiceInfo *info;
+            
+            if (!(info = find_service(interface, protocol, name, type, domain)))
+                return;
+
+            remove_service(c, info);
+            
+            printf("- %4s %4s %-*s %-20s %s\n", if_indextoname(interface, ifname), avahi_proto_to_string(protocol), n_columns-35, name, type, domain);
+            break;
+        }
+            
+        case AVAHI_BROWSER_FAILURE:
+            fprintf(stderr, "service_browser failed: %s\n", avahi_strerror(avahi_client_errno(client)));
+            avahi_simple_poll_quit(simple_poll);
+            break;
+
+        case AVAHI_BROWSER_CACHE_EXHAUSTED:
+            n_cache_exhausted --;
+            check_terminate(c);
+            break;
+            
+        case AVAHI_BROWSER_ALL_FOR_NOW:
+            n_all_for_now --;
+            check_terminate(c);
+            break;
+    }
+}
+
+static void browse_service_type(Config *c, const char *stype, const char *domain) {
+    AvahiServiceBrowser *b;
+    AvahiStringList *i;
+    
+    assert(c);
+    assert(client);
+    assert(stype);
+
+    for (i = browsed_types; i; i = i->next)
+        if (avahi_domain_equal(stype, (char*) i->text))
+            return;
+
+    if (!(b = avahi_service_browser_new(
+              client,
+              AVAHI_IF_UNSPEC,
+              AVAHI_PROTO_UNSPEC,
+              stype,
+              domain,
+              0,
+              service_browser_callback,
+              c))) {
+
+        fprintf(stderr, "avahi_service_browser_new() failed: %s\n", avahi_strerror(avahi_client_errno(client)));
+        avahi_simple_poll_quit(simple_poll);
+    }
+
+    browsed_types = avahi_string_list_add(browsed_types, stype);
+
+    n_all_for_now++;
+    n_cache_exhausted++;
+}
+
+static void service_type_browser_callback(
+    AvahiServiceTypeBrowser *b,
+    AVAHI_GCC_UNUSED AvahiIfIndex interface,
+    AVAHI_GCC_UNUSED AvahiProtocol protocol,
+    AvahiBrowserEvent event,
+    const char *type,
+    const char *domain,
+    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
+    void *userdata) {
+
+    Config *c = userdata;
+
+    assert(b);
+    assert(c);
+    
+    switch (event) {
+        
+        case AVAHI_BROWSER_NEW:
+            browse_service_type(c, type, domain);
+            break;
+
+        case AVAHI_BROWSER_REMOVE:
+            /* We're dirty and never remove the browser again */
+            break;
+
+        case AVAHI_BROWSER_FAILURE:
+            fprintf(stderr, "service_type_browser failed: %s\n", avahi_strerror(avahi_client_errno(client)));
+            avahi_simple_poll_quit(simple_poll);
+            break;
+            
+        case AVAHI_BROWSER_CACHE_EXHAUSTED:
+            n_cache_exhausted --;
+            check_terminate(c);
+            break;
+            
+        case AVAHI_BROWSER_ALL_FOR_NOW:
+            n_all_for_now --;
+            check_terminate(c);
+            break;
+    }
+}
+
+static void browse_all(Config *c) {
+    AvahiServiceTypeBrowser *b;
+    
+    assert(c);
+
+    if (!(b = avahi_service_type_browser_new(
+              client,
+              AVAHI_IF_UNSPEC,
+              AVAHI_PROTO_UNSPEC,
+              c->domain,
+              0,
+              service_type_browser_callback,
+              c))) {
+        
+        fprintf(stderr, "avahi_service_type_browser_new() failed: %s\n", avahi_strerror(avahi_client_errno(client)));
+        avahi_simple_poll_quit(simple_poll);
+    }
+
+    n_cache_exhausted++;
+    n_all_for_now++;
+}
+
+static void client_callback(AVAHI_GCC_UNUSED AvahiClient *c, AvahiClientState state, AVAHI_GCC_UNUSED void * userdata) {
+    switch (state) {
+        case AVAHI_CLIENT_DISCONNECTED:
+            fprintf(stderr, "Client disconnected, exiting.\n");
+            avahi_simple_poll_quit(simple_poll);
+            break;
+
+        case AVAHI_CLIENT_S_REGISTERING:
+        case AVAHI_CLIENT_S_RUNNING:
+        case AVAHI_CLIENT_S_COLLISION:
+            ;
+    }
+}
+
+static void help(FILE *f, const char *argv0) {
+    fprintf(f,
+            "%s [options] <type>\n"
+            "%s [options] -a\n\n"
+            "    -h --help          Show this help\n"
+            "    -V --version       Show version\n"
+            "    -d --domain=DOMAIN The domain to browse\n"
+            "    -a --all           Show all services, regardless of the type\n"
+            "    -v --verbose       Enable verbose mode\n"
+            "    -t --terminate     Terminate after getting or more or less complete list\n"
+            "    -c --cache         Terminate after dumping all entries from the cache\n"
+            "    -l --ignore-local  Ignore local services\n"
+            "    -r --resolve       Resolve services found\n",
+            argv0, argv0);
+}
+
+
+static int parse_command_line(Config *c, int argc, char *argv[]) {
+    int o;
+
+    static const struct option long_options[] = {
+        { "help",         no_argument,       NULL, 'h' },
+        { "version",      no_argument,       NULL, 'V' },
+        { "domain",       required_argument, NULL, 'd' },
+        { "all",          no_argument,       NULL, 'a' },
+        { "verbose",      no_argument,       NULL, 'v' },
+        { "terminate",    no_argument,       NULL, 't' },
+        { "cache",        no_argument,       NULL, 'c' },
+        { "ignore-local", no_argument,       NULL, 'l' },
+        { "resolve",      no_argument,       NULL, 'r' },
+        { NULL, 0, NULL, 0 }
+    };
+
+    assert(c);
+
+    c->command = COMMAND_RUN;
+    c->verbose =
+        c->terminate_on_cache_exhausted =
+        c->terminate_on_all_for_now =
+        c->show_all =
+        c->ignore_local =
+        c->resolve = 0;
+    c->domain = c->stype = NULL;
+    
+    opterr = 0;
+    while ((o = getopt_long(argc, argv, "hVd:avtclr", long_options, NULL)) >= 0) {
+
+        switch(o) {
+            case 'h':
+                c->command = COMMAND_HELP;
+                break;
+            case 'V':
+                c->command = COMMAND_VERSION;
+                break;
+            case 'd':
+                c->domain = avahi_strdup(optarg);
+                break;
+            case 'a':
+                c->show_all = 1;
+                break;
+            case 'v':
+                c->verbose = 1;
+                break;
+            case 't':
+                c->terminate_on_all_for_now = 1;
+                break;
+            case 'c':
+                c->terminate_on_cache_exhausted = 1;
+                break;
+            case 'l':
+                c->ignore_local = 1;
+                break;
+            case 'r':
+                c->resolve = 1;
+                break;
+            default:
+                fprintf(stderr, "Invalid command line argument: %c\n", o);
+                return -1;
+        }
+    }
+
+    if (c->command == COMMAND_RUN && !c->show_all) {
+        if (optind >= argc) {
+            fprintf(stderr, "Too few arguments\n");
+            return -1;
+        }
+
+        c->stype = avahi_strdup(argv[optind]);
+        optind++;
+    }
+    
+    if (optind < argc) {
+        fprintf(stderr, "Too many arguments\n");
+        return -1;
+    }
+        
+    return 0;
+}
+
+int main(int argc, char *argv[]) {
+    int ret = 1, error;
+    Config config;
+    const char *argv0;
+    char *ec;
+
+    if ((argv0 = strrchr(argv[0], '/')))
+        argv0++;
+    else
+        argv0 = argv[0];
+
+    if ((ec = getenv("COLUMNS")))
+        n_columns = atoi(ec);
+
+    if (n_columns < 40)
+        n_columns = 40;
+    
+    if (parse_command_line(&config, argc, argv) < 0)
+        goto fail;
+
+    switch (config.command) {
+        case COMMAND_HELP:
+            help(stdout, argv0);
+            ret = 0;
+            break;
+            
+        case COMMAND_VERSION:
+            printf("%s "PACKAGE_VERSION"\n", argv0);
+            ret = 0;
+            break;
+
+        case COMMAND_RUN:
+            
+            if (!(simple_poll = avahi_simple_poll_new())) {
+                fprintf(stderr, "Failed to create simple poll object.\n");
+                goto fail;
+            }
+            
+            if (sigint_install(simple_poll) < 0)
+                goto fail;
+            
+            if (!(client = avahi_client_new(avahi_simple_poll_get(simple_poll), client_callback, NULL, &error))) {
+                fprintf(stderr, "Failed to create client object: %s\n", avahi_strerror(error));
+                goto fail;
+            }
+
+            if (config.verbose) {
+                const char *version, *hn;
+
+                if (!(version = avahi_client_get_version_string(client))) {
+                    fprintf(stderr, "Failed to query version string: %s\n", avahi_strerror(avahi_client_errno(client)));
+                    goto fail;
+                }
+
+                if (!(hn = avahi_client_get_host_name_fqdn(client))) {
+                    fprintf(stderr, "Failed to query host name: %s\n", avahi_strerror(avahi_client_errno(client)));
+                    goto fail;
+                }
+                
+                fprintf(stderr, "Server version: %s; Host name: %s\n\n", version, hn);
+                fprintf(stderr, "E Ifce Prot %-*s %-20s Domain\n", n_columns-35, "Name", "Type");
+            }
+            
+            if (config.show_all)
+                browse_all(&config);
+            else
+                browse_service_type(&config, config.stype, config.domain);
+            
+            avahi_simple_poll_loop(simple_poll);
+            ret = 0;
+            break;
+    }
+    
+    
+fail:
+
+    while (services)
+        remove_service(&config, services);
+
+    if (client)
+        avahi_client_free(client);
+
+    sigint_uninstall();
+    
+    if (simple_poll)
+        avahi_simple_poll_free(simple_poll);
+
+    avahi_free(config.domain);
+    avahi_free(config.stype);
+
+    avahi_string_list_free(browsed_types);
+
+    return ret;
+}