git.meshlink.io Git - catta/blob - avahi-utils/avahi-browse.c

   1 /***
   2   This file is part of avahi.
   3
   4   avahi is free software; you can redistribute it and/or modify it
   5   under the terms of the GNU Lesser General Public License as
   6   published by the Free Software Foundation; either version 2.1 of the
   7   License, or (at your option) any later version.
   8
   9   avahi is distributed in the hope that it will be useful, but WITHOUT
  10   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  12   Public License for more details.
  13
  14   You should have received a copy of the GNU Lesser General Public
  15   License along with avahi; if not, write to the Free Software
  16   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  17   USA.
  18 ***/
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include <config.h>
  22 #endif
  23
  24 #include <stdlib.h>
  25 #include <stdio.h>
  26 #include <getopt.h>
  27 #include <assert.h>
  28 #include <string.h>
  29 #include <sys/types.h>
  30 #include <sys/socket.h>
  31 #include <net/if.h>
  32 #include <locale.h>
  33 #include <ctype.h>
  34
  35 #include <avahi-common/simple-watch.h>
  36 #include <avahi-common/error.h>
  37 #include <avahi-common/malloc.h>
  38 #include <avahi-common/domain.h>
  39 #include <avahi-common/llist.h>
  40 #include <avahi-common/i18n.h>
  41 #include <avahi-client/client.h>
  42 #include <avahi-client/lookup.h>
  43
  44 #include "sigint.h"
  45
  46 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
  47 #include "stdb.h"
  48 #endif
  49
  50 typedef enum {
  51     COMMAND_HELP,
  52     COMMAND_VERSION,
  53     COMMAND_BROWSE_SERVICES,
  54     COMMAND_BROWSE_ALL_SERVICES,
  55     COMMAND_BROWSE_DOMAINS
  56 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
  57     , COMMAND_DUMP_STDB
  58 #endif
  59 } Command;
  60
  61 typedef struct Config {
  62     int verbose;
  63     int terminate_on_all_for_now;
  64     int terminate_on_cache_exhausted;
  65     char *domain;
  66     char *stype;
  67     int ignore_local;
  68     Command command;
  69     int resolve;
  70     int no_fail;
  71     int parsable;
  72 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
  73     int no_db_lookup;
  74 #endif
  75 } Config;
  76
  77 typedef struct ServiceInfo ServiceInfo;
  78
  79 struct ServiceInfo {
  80     AvahiIfIndex interface;
  81     AvahiProtocol protocol;
  82     char *name, *type, *domain;
  83
  84     AvahiServiceResolver *resolver;
  85     Config *config;
  86
  87     AVAHI_LLIST_FIELDS(ServiceInfo, info);
  88 };
  89
  90 static AvahiSimplePoll *simple_poll = NULL;
  91 static AvahiClient *client = NULL;
  92 static int n_all_for_now = 0, n_cache_exhausted = 0, n_resolving = 0;
  93 static AvahiStringList *browsed_types = NULL;
  94 static ServiceInfo *services = NULL;
  95 static int n_columns = 80;
  96 static int browsing = 0;
  97
  98 static void check_terminate(Config *c) {
  99
 100     assert(n_all_for_now >= 0);
 101     assert(n_cache_exhausted >= 0);
 102     assert(n_resolving >= 0);
 103
 104     if (n_all_for_now <= 0 && n_resolving <= 0) {
 105
 106         if (c->verbose && !c->parsable) {
 107             printf(_(": All for now\n"));
 108             n_all_for_now++; /* Make sure that this event is not repeated */
 109         }
 110
 111         if (c->terminate_on_all_for_now)
 112             avahi_simple_poll_quit(simple_poll);
 113     }
 114
 115     if (n_cache_exhausted <= 0 && n_resolving <= 0) {
 116
 117         if (c->verbose && !c->parsable) {
 118             printf(_(": Cache exhausted\n"));
 119             n_cache_exhausted++; /* Make sure that this event is not repeated */
 120         }
 121
 122         if (c->terminate_on_cache_exhausted)
 123             avahi_simple_poll_quit(simple_poll);
 124     }
 125 }
 126
 127 static ServiceInfo *find_service(AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char *domain) {
 128     ServiceInfo *i;
 129
 130     for (i = services; i; i = i->info_next)
 131         if (i->interface == interface &&
 132             i->protocol == protocol &&
 133             strcasecmp(i->name, name) == 0 &&
 134             avahi_domain_equal(i->type, type) &&
 135             avahi_domain_equal(i->domain, domain))
 136
 137             return i;
 138
 139     return NULL;
 140 }
 141
 142 static char *make_printable(const char *from, char *to) {
 143     const char *f;
 144     char *t;
 145
 146     for (f = from, t = to; *f; f++, t++)
 147         *t = isprint(*f) ? *f : '_';
 148
 149     *t = 0;
 150
 151     return to;
 152 }
 153
 154 static void print_service_line(Config *config, char c, AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char *domain, int nl) {
 155     char ifname[IF_NAMESIZE];
 156
 157 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 158     if (!config->no_db_lookup)
 159         type = stdb_lookup(type);
 160 #endif
 161
 162     if (config->parsable) {
 163         char sn[AVAHI_DOMAIN_NAME_MAX], *e = sn;
 164         size_t l = sizeof(sn);
 165
 166         printf("%c;%s;%s;%s;%s;%s%s",
 167                c,
 168                interface != AVAHI_IF_UNSPEC ? if_indextoname(interface, ifname) : _("n/a"),
 169                protocol != AVAHI_PROTO_UNSPEC ? avahi_proto_to_string(protocol) : _("n/a"),
 170                avahi_escape_label(name, strlen(name), &e, &l), type, domain, nl ? "\n" : "");
 171
 172     } else {
 173         char label[AVAHI_LABEL_MAX];
 174         make_printable(name, label);
 175
 176         printf("%c %6s %4s %-*s %-20s %s\n",
 177                c,
 178                interface != AVAHI_IF_UNSPEC ? if_indextoname(interface, ifname) : _("n/a"),
 179                protocol != AVAHI_PROTO_UNSPEC ? avahi_proto_to_string(protocol) : _("n/a"),
 180                n_columns-35, label, type, domain);
 181     }
 182
 183     fflush(stdout);
 184 }
 185
 186 static void service_resolver_callback(
 187     AvahiServiceResolver *r,
 188     AvahiIfIndex interface,
 189     AvahiProtocol protocol,
 190     AvahiResolverEvent event,
 191     const char *name,
 192     const char *type,
 193     const char *domain,
 194     const char *host_name,
 195     const AvahiAddress *a,
 196     uint16_t port,
 197     AvahiStringList *txt,
 198     AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
 199     void *userdata) {
 200
 201     ServiceInfo *i = userdata;
 202
 203     assert(r);
 204     assert(i);
 205
 206     switch (event) {
 207         case AVAHI_RESOLVER_FOUND: {
 208             char address[AVAHI_ADDRESS_STR_MAX], *t;
 209
 210             avahi_address_snprint(address, sizeof(address), a);
 211
 212             t = avahi_string_list_to_string(txt);
 213
 214             print_service_line(i->config, '=', interface, protocol, name, type, domain, 0);
 215
 216             if (i->config->parsable)
 217                 printf(";%s;%s;%u;%s\n",
 218                        host_name,
 219                        address,
 220                        port,
 221                        t);
 222             else
 223                 printf("   hostname = [%s]\n"
 224                        "   address = [%s]\n"
 225                        "   port = [%u]\n"
 226                        "   txt = [%s]\n",
 227                        host_name,
 228                        address,
 229                        port,
 230                        t);
 231
 232             avahi_free(t);
 233
 234             break;
 235         }
 236
 237         case AVAHI_RESOLVER_FAILURE:
 238
 239             fprintf(stderr, _("Failed to resolve service '%s' of type '%s' in domain '%s': %s\n"), name, type, domain, avahi_strerror(avahi_client_errno(client)));
 240             break;
 241     }
 242
 243
 244     avahi_service_resolver_free(i->resolver);
 245     i->resolver = NULL;
 246
 247     assert(n_resolving > 0);
 248     n_resolving--;
 249     check_terminate(i->config);
 250     fflush(stdout);
 251 }
 252
 253 static ServiceInfo *add_service(Config *c, AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char *domain) {
 254     ServiceInfo *i;
 255
 256     i = avahi_new(ServiceInfo, 1);
 257
 258     if (c->resolve) {
 259         if (!(i->resolver = avahi_service_resolver_new(client, interface, protocol, name, type, domain, AVAHI_PROTO_UNSPEC, 0, service_resolver_callback, i))) {
 260             avahi_free(i);
 261             fprintf(stderr, _("Failed to resolve service '%s' of type '%s' in domain '%s': %s\n"), name, type, domain, avahi_strerror(avahi_client_errno(client)));
 262             return NULL;
 263         }
 264
 265         n_resolving++;
 266     } else
 267         i->resolver = NULL;
 268
 269     i->interface = interface;
 270     i->protocol = protocol;
 271     i->name = avahi_strdup(name);
 272     i->type = avahi_strdup(type);
 273     i->domain = avahi_strdup(domain);
 274     i->config = c;
 275
 276     AVAHI_LLIST_PREPEND(ServiceInfo, info, services, i);
 277
 278     return i;
 279 }
 280
 281 static void remove_service(Config *c, ServiceInfo *i) {
 282     assert(c);
 283     assert(i);
 284
 285     AVAHI_LLIST_REMOVE(ServiceInfo, info, services, i);
 286
 287     if (i->resolver)
 288         avahi_service_resolver_free(i->resolver);
 289
 290     avahi_free(i->name);
 291     avahi_free(i->type);
 292     avahi_free(i->domain);
 293     avahi_free(i);
 294 }
 295
 296 static void service_browser_callback(
 297     AvahiServiceBrowser *b,
 298     AvahiIfIndex interface,
 299     AvahiProtocol protocol,
 300     AvahiBrowserEvent event,
 301     const char *name,
 302     const char *type,
 303     const char *domain,
 304     AvahiLookupResultFlags flags,
 305     void *userdata) {
 306
 307     Config *c = userdata;
 308
 309     assert(b);
 310     assert(c);
 311
 312     switch (event) {
 313         case AVAHI_BROWSER_NEW: {
 314             if (c->ignore_local && (flags & AVAHI_LOOKUP_RESULT_LOCAL))
 315                 break;
 316
 317             if (find_service(interface, protocol, name, type, domain))
 318                 return;
 319
 320             add_service(c, interface, protocol, name, type, domain);
 321
 322             print_service_line(c, '+', interface, protocol, name, type, domain, 1);
 323             break;
 324
 325         }
 326
 327         case AVAHI_BROWSER_REMOVE: {
 328             ServiceInfo *info;
 329
 330             if (!(info = find_service(interface, protocol, name, type, domain)))
 331                 return;
 332
 333             remove_service(c, info);
 334
 335             print_service_line(c, '-', interface, protocol, name, type, domain, 1);
 336             break;
 337         }
 338
 339         case AVAHI_BROWSER_FAILURE:
 340             fprintf(stderr, _("service_browser failed: %s\n"), avahi_strerror(avahi_client_errno(client)));
 341             avahi_simple_poll_quit(simple_poll);
 342             break;
 343
 344         case AVAHI_BROWSER_CACHE_EXHAUSTED:
 345             n_cache_exhausted --;
 346             check_terminate(c);
 347             break;
 348
 349         case AVAHI_BROWSER_ALL_FOR_NOW:
 350             n_all_for_now --;
 351             check_terminate(c);
 352             break;
 353     }
 354 }
 355
 356 static void browse_service_type(Config *c, const char *stype, const char *domain) {
 357     AvahiServiceBrowser *b;
 358     AvahiStringList *i;
 359
 360     assert(c);
 361     assert(client);
 362     assert(stype);
 363
 364     for (i = browsed_types; i; i = i->next)
 365         if (avahi_domain_equal(stype, (char*) i->text))
 366             return;
 367
 368     if (!(b = avahi_service_browser_new(
 369               client,
 370               AVAHI_IF_UNSPEC,
 371               AVAHI_PROTO_UNSPEC,
 372               stype,
 373               domain,
 374               0,
 375               service_browser_callback,
 376               c))) {
 377
 378         fprintf(stderr, _("avahi_service_browser_new() failed: %s\n"), avahi_strerror(avahi_client_errno(client)));
 379         avahi_simple_poll_quit(simple_poll);
 380     }
 381
 382     browsed_types = avahi_string_list_add(browsed_types, stype);
 383
 384     n_all_for_now++;
 385     n_cache_exhausted++;
 386 }
 387
 388 static void service_type_browser_callback(
 389     AvahiServiceTypeBrowser *b,
 390     AVAHI_GCC_UNUSED AvahiIfIndex interface,
 391     AVAHI_GCC_UNUSED AvahiProtocol protocol,
 392     AvahiBrowserEvent event,
 393     const char *type,
 394     const char *domain,
 395     AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
 396     void *userdata) {
 397
 398     Config *c = userdata;
 399
 400     assert(b);
 401     assert(c);
 402
 403     switch (event) {
 404
 405         case AVAHI_BROWSER_NEW:
 406             browse_service_type(c, type, domain);
 407             break;
 408
 409         case AVAHI_BROWSER_REMOVE:
 410             /* We're dirty and never remove the browser again */
 411             break;
 412
 413         case AVAHI_BROWSER_FAILURE:
 414             fprintf(stderr, _("service_type_browser failed: %s\n"), avahi_strerror(avahi_client_errno(client)));
 415             avahi_simple_poll_quit(simple_poll);
 416             break;
 417
 418         case AVAHI_BROWSER_CACHE_EXHAUSTED:
 419             n_cache_exhausted --;
 420             check_terminate(c);
 421             break;
 422
 423         case AVAHI_BROWSER_ALL_FOR_NOW:
 424             n_all_for_now --;
 425             check_terminate(c);
 426             break;
 427     }
 428 }
 429
 430 static void browse_all(Config *c) {
 431     AvahiServiceTypeBrowser *b;
 432
 433     assert(c);
 434
 435     if (!(b = avahi_service_type_browser_new(
 436               client,
 437               AVAHI_IF_UNSPEC,
 438               AVAHI_PROTO_UNSPEC,
 439               c->domain,
 440               0,
 441               service_type_browser_callback,
 442               c))) {
 443
 444         fprintf(stderr, _("avahi_service_type_browser_new() failed: %s\n"), avahi_strerror(avahi_client_errno(client)));
 445         avahi_simple_poll_quit(simple_poll);
 446     }
 447
 448     n_cache_exhausted++;
 449     n_all_for_now++;
 450 }
 451
 452 static void domain_browser_callback(
 453     AvahiDomainBrowser *b,
 454     AVAHI_GCC_UNUSED AvahiIfIndex interface,
 455     AVAHI_GCC_UNUSED AvahiProtocol protocol,
 456     AvahiBrowserEvent event,
 457     const char *domain,
 458     AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
 459     void *userdata) {
 460
 461     Config *c = userdata;
 462
 463     assert(b);
 464     assert(c);
 465
 466     switch (event) {
 467
 468         case AVAHI_BROWSER_NEW:
 469         case AVAHI_BROWSER_REMOVE: {
 470             char ifname[IF_NAMESIZE];
 471
 472             if (c->parsable)
 473                 printf("%c;%s;%s;%s\n",
 474                        event == AVAHI_BROWSER_NEW ? '+' : '-',
 475                        interface != AVAHI_IF_UNSPEC ? if_indextoname(interface, ifname) : "",
 476                        protocol != AVAHI_PROTO_UNSPEC ? avahi_proto_to_string(protocol) : "",
 477                        domain);
 478             else
 479                 printf("%c %4s %4s %s\n",
 480                        event == AVAHI_BROWSER_NEW ? '+' : '-',
 481                        interface != AVAHI_IF_UNSPEC ? if_indextoname(interface, ifname) : "n/a",
 482                        protocol != AVAHI_PROTO_UNSPEC ? avahi_proto_to_string(protocol) : "n/a",
 483                        domain);
 484             break;
 485         }
 486
 487         case AVAHI_BROWSER_FAILURE:
 488             fprintf(stderr, ("domain_browser failed: %s\n"), avahi_strerror(avahi_client_errno(client)));
 489             avahi_simple_poll_quit(simple_poll);
 490             break;
 491
 492         case AVAHI_BROWSER_CACHE_EXHAUSTED:
 493             n_cache_exhausted --;
 494             check_terminate(c);
 495             break;
 496
 497         case AVAHI_BROWSER_ALL_FOR_NOW:
 498             n_all_for_now --;
 499             check_terminate(c);
 500             break;
 501     }
 502 }
 503
 504 static void browse_domains(Config *c) {
 505     AvahiDomainBrowser *b;
 506
 507     assert(c);
 508
 509     if (!(b = avahi_domain_browser_new(
 510               client,
 511               AVAHI_IF_UNSPEC,
 512               AVAHI_PROTO_UNSPEC,
 513               c->domain,
 514               AVAHI_DOMAIN_BROWSER_BROWSE,
 515               0,
 516               domain_browser_callback,
 517               c))) {
 518
 519         fprintf(stderr, _("avahi_domain_browser_new() failed: %s\n"), avahi_strerror(avahi_client_errno(client)));
 520         avahi_simple_poll_quit(simple_poll);
 521     }
 522
 523     n_cache_exhausted++;
 524     n_all_for_now++;
 525 }
 526
 527 static int start(Config *config) {
 528
 529     assert(!browsing);
 530
 531     if (config->verbose && !config->parsable) {
 532         const char *version, *hn;
 533
 534         if (!(version = avahi_client_get_version_string(client))) {
 535             fprintf(stderr, _("Failed to query version string: %s\n"), avahi_strerror(avahi_client_errno(client)));
 536             return -1;
 537         }
 538
 539         if (!(hn = avahi_client_get_host_name_fqdn(client))) {
 540             fprintf(stderr, _("Failed to query host name: %s\n"), avahi_strerror(avahi_client_errno(client)));
 541             return -1;
 542         }
 543
 544         fprintf(stderr, _("Server version: %s; Host name: %s\n"), version, hn);
 545
 546         if (config->command == COMMAND_BROWSE_DOMAINS) {
 547             /* Translators: This is a column heading with abbreviations for
 548              *   Event (+/-), Network Interface, Protocol (IPv4/v6), Domain */
 549             fprintf(stderr, _("E Ifce Prot Domain\n"));
 550         } else {
 551             /* Translators: This is a column heading with abbreviations for
 552              *   Event (+/-), Network Interface, Protocol (IPv4/v6), Domain */
 553             fprintf(stderr, _("E Ifce Prot %-*s %-20s Domain\n"), n_columns-35, _("Name"), _("Type"));
 554         }
 555     }
 556
 557     if (config->command == COMMAND_BROWSE_SERVICES)
 558         browse_service_type(config, config->stype, config->domain);
 559     else if (config->command == COMMAND_BROWSE_ALL_SERVICES)
 560         browse_all(config);
 561     else {
 562         assert(config->command == COMMAND_BROWSE_DOMAINS);
 563         browse_domains(config);
 564     }
 565
 566     browsing = 1;
 567     return 0;
 568 }
 569
 570 static void client_callback(AvahiClient *c, AvahiClientState state, AVAHI_GCC_UNUSED void * userdata) {
 571     Config *config = userdata;
 572
 573     /* This function might be called when avahi_client_new() has not
 574      * returned yet.*/
 575     client = c;
 576
 577     switch (state) {
 578         case AVAHI_CLIENT_FAILURE:
 579
 580             if (config->no_fail && avahi_client_errno(c) == AVAHI_ERR_DISCONNECTED) {
 581                 int error;
 582
 583                 /* We have been disconnected, so let reconnect */
 584
 585                 fprintf(stderr, _("Disconnected, reconnecting ...\n"));
 586
 587                 avahi_client_free(client);
 588                 client = NULL;
 589
 590                 avahi_string_list_free(browsed_types);
 591                 browsed_types = NULL;
 592
 593                 while (services)
 594                     remove_service(config, services);
 595
 596                 browsing = 0;
 597
 598                 if (!(client = avahi_client_new(avahi_simple_poll_get(simple_poll), AVAHI_CLIENT_NO_FAIL, client_callback, config, &error))) {
 599                     fprintf(stderr, _("Failed to create client object: %s\n"), avahi_strerror(error));
 600                     avahi_simple_poll_quit(simple_poll);
 601                 }
 602
 603             } else {
 604                 fprintf(stderr, _("Client failure, exiting: %s\n"), avahi_strerror(avahi_client_errno(c)));
 605                 avahi_simple_poll_quit(simple_poll);
 606             }
 607
 608             break;
 609
 610         case AVAHI_CLIENT_S_REGISTERING:
 611         case AVAHI_CLIENT_S_RUNNING:
 612         case AVAHI_CLIENT_S_COLLISION:
 613
 614             if (!browsing)
 615                 if (start(config) < 0)
 616                     avahi_simple_poll_quit(simple_poll);
 617
 618             break;
 619
 620         case AVAHI_CLIENT_CONNECTING:
 621
 622             if (config->verbose && !config->parsable)
 623                 fprintf(stderr, _("Waiting for daemon ...\n"));
 624
 625             break;
 626     }
 627 }
 628
 629 static void help(FILE *f, const char *argv0) {
 630     if (strstr(argv0, "domain"))
 631         fprintf(f, "%s [options] \n\n", argv0);
 632     else
 633         fprintf(f,
 634                 "%s [options] <service type>\n"
 635                 "%s [options] -a\n"
 636                 "%s [options] -D\n"
 637 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 638                 "%s [options] -b\n"
 639 #endif
 640                 "\n",
 641 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 642                 argv0,
 643 #endif
 644                 argv0, argv0, argv0);
 645
 646     fprintf(f, "%s%s",
 647             _("    -h --help            Show this help\n"
 648               "    -V --version         Show version\n"
 649               "    -D --browse-domains  Browse for browsing domains instead of services\n"
 650               "    -a --all             Show all services, regardless of the type\n"
 651               "    -d --domain=DOMAIN   The domain to browse in\n"
 652               "    -v --verbose         Enable verbose mode\n"
 653               "    -t --terminate       Terminate after dumping a more or less complete list\n"
 654               "    -c --cache           Terminate after dumping all entries from the cache\n"
 655               "    -l --ignore-local    Ignore local services\n"
 656               "    -r --resolve         Resolve services found\n"
 657               "    -f --no-fail         Don't fail if the daemon is not available\n"
 658               "    -p --parsable        Output in parsable format\n"),
 659 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 660             _("    -k --no-db-lookup    Don't lookup service types\n"
 661               "    -b --dump-db         Dump service type database\n")
 662 #else
 663             ""
 664 #endif
 665             );
 666 }
 667
 668 static int parse_command_line(Config *c, const char *argv0, int argc, char *argv[]) {
 669     int o;
 670
 671     static const struct option long_options[] = {
 672         { "help",           no_argument,       NULL, 'h' },
 673         { "version",        no_argument,       NULL, 'V' },
 674         { "browse-domains", no_argument,       NULL, 'D' },
 675         { "domain",         required_argument, NULL, 'd' },
 676         { "all",            no_argument,       NULL, 'a' },
 677         { "verbose",        no_argument,       NULL, 'v' },
 678         { "terminate",      no_argument,       NULL, 't' },
 679         { "cache",          no_argument,       NULL, 'c' },
 680         { "ignore-local",   no_argument,       NULL, 'l' },
 681         { "resolve",        no_argument,       NULL, 'r' },
 682         { "no-fail",        no_argument,       NULL, 'f' },
 683         { "parsable",      no_argument,       NULL, 'p' },
 684 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 685         { "no-db-lookup",   no_argument,       NULL, 'k' },
 686         { "dump-db",        no_argument,       NULL, 'b' },
 687 #endif
 688         { NULL, 0, NULL, 0 }
 689     };
 690
 691     assert(c);
 692
 693     c->command = strstr(argv0, "domain") ? COMMAND_BROWSE_DOMAINS : COMMAND_BROWSE_SERVICES;
 694     c->verbose =
 695         c->terminate_on_cache_exhausted =
 696         c->terminate_on_all_for_now =
 697         c->ignore_local =
 698         c->resolve =
 699         c->no_fail =
 700         c->parsable = 0;
 701     c->domain = c->stype = NULL;
 702
 703 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 704     c->no_db_lookup = 0;
 705 #endif
 706
 707     while ((o = getopt_long(argc, argv, "hVd:avtclrDfp"
 708 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 709                             "kb"
 710 #endif
 711                             , long_options, NULL)) >= 0) {
 712
 713         switch(o) {
 714             case 'h':
 715                 c->command = COMMAND_HELP;
 716                 break;
 717             case 'V':
 718                 c->command = COMMAND_VERSION;
 719                 break;
 720             case 'a':
 721                 c->command = COMMAND_BROWSE_ALL_SERVICES;
 722                 break;
 723             case 'D':
 724                 c->command = COMMAND_BROWSE_DOMAINS;
 725                 break;
 726             case 'd':
 727                 avahi_free(c->domain);
 728                 c->domain = avahi_strdup(optarg);
 729                 break;
 730             case 'v':
 731                 c->verbose = 1;
 732                 break;
 733             case 't':
 734                 c->terminate_on_all_for_now = 1;
 735                 break;
 736             case 'c':
 737                 c->terminate_on_cache_exhausted = 1;
 738                 break;
 739             case 'l':
 740                 c->ignore_local = 1;
 741                 break;
 742             case 'r':
 743                 c->resolve = 1;
 744                 break;
 745             case 'f':
 746                 c->no_fail = 1;
 747                 break;
 748             case 'p':
 749                 c->parsable = 1;
 750                 break;
 751 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 752             case 'k':
 753                 c->no_db_lookup = 1;
 754                 break;
 755             case 'b':
 756                 c->command = COMMAND_DUMP_STDB;
 757                 break;
 758 #endif
 759             default:
 760                 return -1;
 761         }
 762     }
 763
 764     if (c->command == COMMAND_BROWSE_SERVICES) {
 765         if (optind >= argc) {
 766             fprintf(stderr, _("Too few arguments\n"));
 767             return -1;
 768         }
 769
 770         c->stype = avahi_strdup(argv[optind]);
 771         optind++;
 772     }
 773
 774     if (optind < argc) {
 775         fprintf(stderr, _("Too many arguments\n"));
 776         return -1;
 777     }
 778
 779     return 0;
 780 }
 781
 782 int main(int argc, char *argv[]) {
 783     int ret = 1, error;
 784     Config config;
 785     const char *argv0;
 786     char *ec;
 787
 788     avahi_init_i18n();
 789     setlocale(LC_ALL, "");
 790
 791     if ((argv0 = strrchr(argv[0], '/')))
 792         argv0++;
 793     else
 794         argv0 = argv[0];
 795
 796     if ((ec = getenv("COLUMNS")))
 797         n_columns = atoi(ec);
 798
 799     if (n_columns < 40)
 800         n_columns = 40;
 801
 802     if (parse_command_line(&config, argv0, argc, argv) < 0)
 803         goto fail;
 804
 805     switch (config.command) {
 806         case COMMAND_HELP:
 807             help(stdout, argv0);
 808             ret = 0;
 809             break;
 810
 811         case COMMAND_VERSION:
 812             printf("%s "PACKAGE_VERSION"\n", argv0);
 813             ret = 0;
 814             break;
 815
 816         case COMMAND_BROWSE_SERVICES:
 817         case COMMAND_BROWSE_ALL_SERVICES:
 818         case COMMAND_BROWSE_DOMAINS:
 819
 820             if (!(simple_poll = avahi_simple_poll_new())) {
 821                 fprintf(stderr, _("Failed to create simple poll object.\n"));
 822                 goto fail;
 823             }
 824
 825             if (sigint_install(simple_poll) < 0)
 826                 goto fail;
 827
 828             if (!(client = avahi_client_new(avahi_simple_poll_get(simple_poll), config.no_fail ? AVAHI_CLIENT_NO_FAIL : 0, client_callback, &config, &error))) {
 829                 fprintf(stderr, _("Failed to create client object: %s\n"), avahi_strerror(error));
 830                 goto fail;
 831             }
 832
 833             avahi_simple_poll_loop(simple_poll);
 834             ret = 0;
 835             break;
 836
 837 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 838         case COMMAND_DUMP_STDB: {
 839             char *t;
 840             stdb_setent();
 841
 842             while ((t = stdb_getent())) {
 843                 if (config.no_db_lookup)
 844                     printf("%s\n", t);
 845                 else
 846                     printf("%s\n", stdb_lookup(t));
 847             }
 848
 849             ret = 0;
 850             break;
 851         }
 852 #endif
 853     }
 854
 855
 856 fail:
 857
 858     while (services)
 859         remove_service(&config, services);
 860
 861     if (client)
 862         avahi_client_free(client);
 863
 864     sigint_uninstall();
 865
 866     if (simple_poll)
 867         avahi_simple_poll_free(simple_poll);
 868
 869     avahi_free(config.domain);
 870     avahi_free(config.stype);
 871
 872     avahi_string_list_free(browsed_types);
 873
 874 #if defined(HAVE_GDBM) || defined(HAVE_DBM)
 875     stdb_shutdown();
 876 #endif
 877
 878     return ret;
 879 }