X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=avahi-utils%2Favahi-browse.c;h=169f62bd7c8d778c939d90263a6eb5d9413559eb;hb=e2c156ee15f40b13a451d6223b407de8fc7f0517;hp=09792a47ec4714dc53f84c18226b66630ffd7472;hpb=05f17e7620daacbb50c39c64823d914c33c7820a;p=catta diff --git a/avahi-utils/avahi-browse.c b/avahi-utils/avahi-browse.c index 09792a4..169f62b 100644 --- a/avahi-utils/avahi-browse.c +++ b/avahi-utils/avahi-browse.c @@ -28,7 +28,10 @@ #include #include #include +#include +#include #include +#include #include #include @@ -40,10 +43,16 @@ #include "sigint.h" +#ifdef HAVE_GDBM +#include "stdb.h" +#endif + typedef enum { COMMAND_HELP, COMMAND_VERSION, - COMMAND_RUN + COMMAND_BROWSE_SERVICES, + COMMAND_BROWSE_ALL_SERVICES, + COMMAND_BROWSE_DOMAINS } Command; typedef struct Config { @@ -53,12 +62,13 @@ typedef struct Config { char *domain; char *stype; int ignore_local; - int show_all; Command command; int resolve; +#ifdef HAVE_GDBM + int no_db_lookup; +#endif } Config; - typedef struct ServiceInfo ServiceInfo; struct ServiceInfo { @@ -123,6 +133,21 @@ static ServiceInfo *find_service(AvahiIfIndex interface, AvahiProtocol protocol, return NULL; } +static void print_service_line(Config *config, char c, AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char *domain) { + char ifname[IF_NAMESIZE]; + +#ifdef HAVE_GDBM + if (!config->no_db_lookup) + type = stdb_lookup(type); +#endif + + printf("%c %4s %4s %-*s %-20s %s\n", + c, + interface != AVAHI_IF_UNSPEC ? if_indextoname(interface, ifname) : "n/a", + protocol != AVAHI_PROTO_UNSPEC ? avahi_proto_to_string(protocol) : "n/a", + n_columns-35, name, type, domain); +} + static void service_resolver_callback( AvahiServiceResolver *r, AvahiIfIndex interface, @@ -146,19 +171,17 @@ static void service_resolver_callback( 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" + print_service_line(i->config, '=', interface, protocol, name, type, domain); + + printf(" 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, @@ -244,7 +267,6 @@ static void service_browser_callback( switch (event) { case AVAHI_BROWSER_NEW: { - char ifname[IF_NAMESIZE]; if (c->ignore_local && (flags & AVAHI_LOOKUP_RESULT_LOCAL)) break; @@ -253,13 +275,12 @@ static void service_browser_callback( 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); + print_service_line(c, '+', interface, protocol, name, type, domain); break; } case AVAHI_BROWSER_REMOVE: { - char ifname[IF_NAMESIZE]; ServiceInfo *info; if (!(info = find_service(interface, protocol, name, type, domain))) @@ -267,7 +288,7 @@ static void service_browser_callback( 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); + print_service_line(c, '-', interface, protocol, name, type, domain); break; } @@ -384,6 +405,74 @@ static void browse_all(Config *c) { n_all_for_now++; } +static void domain_browser_callback( + AvahiDomainBrowser *b, + AVAHI_GCC_UNUSED AvahiIfIndex interface, + AVAHI_GCC_UNUSED AvahiProtocol protocol, + AvahiBrowserEvent event, + const char *domain, + AVAHI_GCC_UNUSED AvahiLookupResultFlags flags, + void *userdata) { + + Config *c = userdata; + + assert(b); + assert(c); + + switch (event) { + + case AVAHI_BROWSER_NEW: + case AVAHI_BROWSER_REMOVE: { + char ifname[IF_NAMESIZE]; + + printf("%c %4s %4s %s\n", + event == AVAHI_BROWSER_NEW ? '+' : '-', + interface != AVAHI_IF_UNSPEC ? if_indextoname(interface, ifname) : "n/a", + protocol != AVAHI_PROTO_UNSPEC ? avahi_proto_to_string(protocol) : "n/a", + domain); + break; + } + + case AVAHI_BROWSER_FAILURE: + fprintf(stderr, "domain_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_domains(Config *c) { + AvahiDomainBrowser *b; + + assert(c); + + if (!(b = avahi_domain_browser_new( + client, + AVAHI_IF_UNSPEC, + AVAHI_PROTO_UNSPEC, + c->domain, + AVAHI_DOMAIN_BROWSER_BROWSE, + 0, + domain_browser_callback, + c))) { + + fprintf(stderr, "avahi_domain_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: @@ -400,50 +489,65 @@ static void client_callback(AVAHI_GCC_UNUSED AvahiClient *c, AvahiClientState st static void help(FILE *f, const char *argv0) { fprintf(f, - "%s [options] \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); + "%s [options] \n" + "%s [options] -a\n" + "%s [options] -D\n\n" + " -h --help Show this help\n" + " -V --version Show version\n" + " -D --browse-domains Browse for browsing domains instead of services\n" + " -a --all Show all services, regardless of the type\n" + " -d --domain=DOMAIN The domain to browse in\n" + " -v --verbose Enable verbose mode\n" + " -t --terminate Terminate after dumping a 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" +#ifdef HAVE_GDBM + " -k --no-db-lookup Don't lookup service types\n" +#endif + , argv0, 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' }, + { "help", no_argument, NULL, 'h' }, + { "version", no_argument, NULL, 'V' }, + { "browse-domains", no_argument, NULL, 'D' }, + { "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' }, +#ifdef HAVE_GDBM + { "no-db-lookup", no_argument, NULL, 'k' }, +#endif { NULL, 0, NULL, 0 } }; assert(c); - c->command = COMMAND_RUN; + c->command = COMMAND_BROWSE_SERVICES; 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; + +#ifdef HAVE_GDBM + c->no_db_lookup = 0; +#endif opterr = 0; - while ((o = getopt_long(argc, argv, "hVd:avtclr", long_options, NULL)) >= 0) { + while ((o = getopt_long(argc, argv, "hVd:avtclrD" +#ifdef HAVE_GDBM + "k" +#endif + , long_options, NULL)) >= 0) { switch(o) { case 'h': @@ -452,12 +556,15 @@ static int parse_command_line(Config *c, int argc, char *argv[]) { case 'V': c->command = COMMAND_VERSION; break; + case 'a': + c->command = COMMAND_BROWSE_ALL_SERVICES; + break; + case 'D': + c->command = COMMAND_BROWSE_DOMAINS; + break; case 'd': c->domain = avahi_strdup(optarg); break; - case 'a': - c->show_all = 1; - break; case 'v': c->verbose = 1; break; @@ -473,13 +580,18 @@ static int parse_command_line(Config *c, int argc, char *argv[]) { case 'r': c->resolve = 1; break; +#ifdef HAVE_GDBM + case 'k': + c->no_db_lookup = 1; + break; +#endif default: fprintf(stderr, "Invalid command line argument: %c\n", o); return -1; } } - if (c->command == COMMAND_RUN && !c->show_all) { + if (c->command == COMMAND_BROWSE_SERVICES) { if (optind >= argc) { fprintf(stderr, "Too few arguments\n"); return -1; @@ -503,6 +615,8 @@ int main(int argc, char *argv[]) { const char *argv0; char *ec; + setlocale(LC_ALL, ""); + if ((argv0 = strrchr(argv[0], '/'))) argv0++; else @@ -528,7 +642,9 @@ int main(int argc, char *argv[]) { ret = 0; break; - case COMMAND_RUN: + case COMMAND_BROWSE_SERVICES: + case COMMAND_BROWSE_ALL_SERVICES: + case COMMAND_BROWSE_DOMAINS: if (!(simple_poll = avahi_simple_poll_new())) { fprintf(stderr, "Failed to create simple poll object.\n"); @@ -556,14 +672,22 @@ int main(int argc, char *argv[]) { 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"); + fprintf(stderr, "Server version: %s; Host name: %s\n", version, hn); + + if (config.command == COMMAND_BROWSE_DOMAINS) + fprintf(stderr, "E Ifce Prot Domain\n"); + else + fprintf(stderr, "E Ifce Prot %-*s %-20s Domain\n", n_columns-35, "Name", "Type"); } - - if (config.show_all) - browse_all(&config); - else + + if (config.command == COMMAND_BROWSE_SERVICES) browse_service_type(&config, config.stype, config.domain); + else if (config.command == COMMAND_BROWSE_ALL_SERVICES) + browse_all(&config); + else { + assert(config.command == COMMAND_BROWSE_DOMAINS); + browse_domains(&config); + } avahi_simple_poll_loop(simple_poll); ret = 0; @@ -589,5 +713,9 @@ fail: avahi_string_list_free(browsed_types); +#ifdef HAVE_GDBM + stdb_shutdown(); +#endif + return ret; }