git.meshlink.io Git - catta/blob - avahi-client/client.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of avahi.
   5
   6   avahi is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as
   8   published by the Free Software Foundation; either version 2.1 of the
   9   License, or (at your option) any later version.
  10
  11   avahi is distributed in the hope that it will be useful, but WITHOUT
  12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  13   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  14   Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public
  17   License along with avahi; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <stdlib.h>
  27 #include <stdio.h>
  28 #include <string.h>
  29
  30 #include <dbus/dbus.h>
  31
  32 #include <avahi-common/dbus.h>
  33 #include <avahi-common/llist.h>
  34 #include <avahi-common/error.h>
  35 #include <avahi-common/dbus.h>
  36 #include <avahi-common/malloc.h>
  37 #include <avahi-common/dbus-watch-glue.h>
  38
  39 #include "client.h"
  40 #include "internal.h"
  41
  42 #define AVAHI_CLIENT_DBUS_API_SUPPORTED ((uint32_t) 0x0201)
  43
  44 static int init_server(AvahiClient *client, int *ret_error);
  45
  46 int avahi_client_set_errno (AvahiClient *client, int error) {
  47     assert(client);
  48
  49     return client->error = error;
  50 }
  51
  52 int avahi_client_set_dbus_error(AvahiClient *client, DBusError *error) {
  53     assert(client);
  54     assert(error);
  55
  56     return avahi_client_set_errno(client, avahi_error_dbus_to_number(error->name));
  57 }
  58
  59 static void client_set_state (AvahiClient *client, AvahiServerState state) {
  60     assert(client);
  61
  62     if (client->state == state)
  63         return;
  64
  65     client->state = state;
  66
  67     switch (client->state) {
  68         case AVAHI_CLIENT_FAILURE:
  69             if (client->bus) {
  70                 dbus_connection_disconnect(client->bus);
  71                 dbus_connection_unref(client->bus);
  72                 client->bus = NULL;
  73             }
  74
  75             /* Fall through */
  76
  77         case AVAHI_CLIENT_S_COLLISION:
  78         case AVAHI_CLIENT_S_REGISTERING:
  79
  80             /* Clear cached strings */
  81             avahi_free(client->host_name);
  82             avahi_free(client->host_name_fqdn);
  83             avahi_free(client->domain_name);
  84
  85             client->host_name =  NULL;
  86             client->host_name_fqdn = NULL;
  87             client->domain_name = NULL;
  88             break;
  89
  90         case AVAHI_CLIENT_S_RUNNING:
  91         case AVAHI_CLIENT_CONNECTING:
  92             break;
  93
  94     }
  95
  96     if (client->callback)
  97         client->callback (client, state, client->userdata);
  98 }
  99
 100 static DBusHandlerResult filter_func(DBusConnection *bus, DBusMessage *message, void *userdata) {
 101     AvahiClient *client = userdata;
 102     DBusError error;
 103
 104     assert(bus);
 105     assert(message);
 106
 107     dbus_error_init(&error);
 108
 109 /*     fprintf(stderr, "dbus: interface=%s, path=%s, member=%s\n", */
 110 /*             dbus_message_get_interface (message), */
 111 /*             dbus_message_get_path (message), */
 112 /*             dbus_message_get_member (message)); */
 113
 114     if (dbus_message_is_signal(message, DBUS_INTERFACE_LOCAL, "Disconnected")) {
 115
 116         /* The DBUS server died or kicked us */
 117         avahi_client_set_errno(client, AVAHI_ERR_DISCONNECTED);
 118         goto fail;
 119
 120     } else if (dbus_message_is_signal(message, DBUS_INTERFACE_DBUS, "NameAcquired")) {
 121
 122         /* Ignore this message */
 123
 124     } else if (dbus_message_is_signal(message, DBUS_INTERFACE_DBUS, "NameOwnerChanged")) {
 125         char *name, *old, *new;
 126
 127         if (!dbus_message_get_args(
 128                   message, &error,
 129                   DBUS_TYPE_STRING, &name,
 130                   DBUS_TYPE_STRING, &old,
 131                   DBUS_TYPE_STRING, &new,
 132                   DBUS_TYPE_INVALID) || dbus_error_is_set(&error)) {
 133
 134             fprintf(stderr, "WARNING: Failed to parse NameOwnerChanged signal: %s\n", error.message);
 135             avahi_client_set_errno(client, AVAHI_ERR_DBUS_ERROR);
 136             goto fail;
 137         }
 138
 139         if (strcmp(name, AVAHI_DBUS_NAME) == 0) {
 140
 141             if (avahi_client_is_connected(client)) {
 142
 143                 /* Regardless if the server lost or acquired its name or
 144                  * if the name was transfered: our services are no longer
 145                  * available, so we disconnect ourselves */
 146                 avahi_client_set_errno(client, AVAHI_ERR_DISCONNECTED);
 147                 goto fail;
 148
 149             } else if (client->state == AVAHI_CLIENT_CONNECTING && (!old || *old == 0)) {
 150                 int ret;
 151
 152                 /* Server appeared */
 153
 154                 if ((ret = init_server(client, NULL)) < 0) {
 155                     avahi_client_set_errno(client, ret);
 156                     goto fail;
 157                 }
 158             }
 159         }
 160
 161     } else if (!avahi_client_is_connected(client)) {
 162
 163         /* Ignore messages we get in unconnected state */
 164
 165     } else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_SERVER, "StateChanged")) {
 166         int32_t state;
 167         char *e = NULL;
 168         int c;
 169
 170         if (!dbus_message_get_args(
 171                   message, &error,
 172                   DBUS_TYPE_INT32, &state,
 173                   DBUS_TYPE_STRING, &e,
 174                   DBUS_TYPE_INVALID) || dbus_error_is_set (&error)) {
 175
 176             fprintf(stderr, "WARNING: Failed to parse Server.StateChanged signal: %s\n", error.message);
 177             avahi_client_set_errno(client, AVAHI_ERR_DBUS_ERROR);
 178             goto fail;
 179         }
 180
 181         if ((c = avahi_error_dbus_to_number(e)) != AVAHI_OK)
 182             avahi_client_set_errno(client, c);
 183
 184         client_set_state(client, (AvahiClientState) state);
 185
 186     } else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_ENTRY_GROUP, "StateChanged")) {
 187         const char *path;
 188         AvahiEntryGroup *g;
 189         path = dbus_message_get_path(message);
 190
 191         for (g = client->groups; g; g = g->groups_next)
 192             if (strcmp(g->path, path) == 0)
 193                 break;
 194
 195         if (g) {
 196             int32_t state;
 197             char *e;
 198             int c;
 199
 200             if (!dbus_message_get_args(
 201                       message, &error,
 202                       DBUS_TYPE_INT32, &state,
 203                       DBUS_TYPE_STRING, &e,
 204                       DBUS_TYPE_INVALID) ||
 205                 dbus_error_is_set(&error)) {
 206
 207                 fprintf(stderr, "WARNING: Failed to parse EntryGroup.StateChanged signal: %s\n", error.message);
 208                 avahi_client_set_errno(client, AVAHI_ERR_DBUS_ERROR);
 209                 goto fail;
 210             }
 211
 212             if ((c = avahi_error_dbus_to_number(e)) != AVAHI_OK)
 213                 avahi_client_set_errno(client, c);
 214
 215             avahi_entry_group_set_state(g, state);
 216         }
 217
 218     } else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_DOMAIN_BROWSER, "ItemNew"))
 219         return avahi_domain_browser_event(client, AVAHI_BROWSER_NEW, message);
 220     else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_DOMAIN_BROWSER, "ItemRemove"))
 221         return avahi_domain_browser_event(client, AVAHI_BROWSER_REMOVE, message);
 222     else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_DOMAIN_BROWSER, "CacheExhausted"))
 223         return avahi_domain_browser_event(client, AVAHI_BROWSER_CACHE_EXHAUSTED, message);
 224     else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_DOMAIN_BROWSER, "AllForNow"))
 225         return avahi_domain_browser_event(client, AVAHI_BROWSER_ALL_FOR_NOW, message);
 226     else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_DOMAIN_BROWSER, "Failure"))
 227         return avahi_domain_browser_event(client, AVAHI_BROWSER_FAILURE, message);
 228
 229     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_TYPE_BROWSER, "ItemNew"))
 230         return avahi_service_type_browser_event (client, AVAHI_BROWSER_NEW, message);
 231     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_TYPE_BROWSER, "ItemRemove"))
 232         return avahi_service_type_browser_event (client, AVAHI_BROWSER_REMOVE, message);
 233     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_TYPE_BROWSER, "CacheExhausted"))
 234         return avahi_service_type_browser_event (client, AVAHI_BROWSER_CACHE_EXHAUSTED, message);
 235     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_TYPE_BROWSER, "AllForNow"))
 236         return avahi_service_type_browser_event (client, AVAHI_BROWSER_ALL_FOR_NOW, message);
 237     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_TYPE_BROWSER, "Failure"))
 238         return avahi_service_type_browser_event (client, AVAHI_BROWSER_FAILURE, message);
 239
 240     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_BROWSER, "ItemNew"))
 241         return avahi_service_browser_event (client, AVAHI_BROWSER_NEW, message);
 242     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_BROWSER, "ItemRemove"))
 243         return avahi_service_browser_event (client, AVAHI_BROWSER_REMOVE, message);
 244     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_BROWSER, "CacheExhausted"))
 245         return avahi_service_browser_event (client, AVAHI_BROWSER_CACHE_EXHAUSTED, message);
 246     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_BROWSER, "AllForNow"))
 247         return avahi_service_browser_event (client, AVAHI_BROWSER_ALL_FOR_NOW, message);
 248     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_BROWSER, "Failure"))
 249         return avahi_service_browser_event (client, AVAHI_BROWSER_FAILURE, message);
 250
 251     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_RESOLVER, "Found"))
 252         return avahi_service_resolver_event (client, AVAHI_RESOLVER_FOUND, message);
 253     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_RESOLVER, "Failure"))
 254         return avahi_service_resolver_event (client, AVAHI_RESOLVER_FAILURE, message);
 255
 256     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_HOST_NAME_RESOLVER, "Found"))
 257         return avahi_host_name_resolver_event (client, AVAHI_RESOLVER_FOUND, message);
 258     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_HOST_NAME_RESOLVER, "Failure"))
 259         return avahi_host_name_resolver_event (client, AVAHI_RESOLVER_FAILURE, message);
 260
 261     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_ADDRESS_RESOLVER, "Found"))
 262         return avahi_address_resolver_event (client, AVAHI_RESOLVER_FOUND, message);
 263     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_ADDRESS_RESOLVER, "Failure"))
 264         return avahi_address_resolver_event (client, AVAHI_RESOLVER_FAILURE, message);
 265
 266     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_RECORD_BROWSER, "ItemNew"))
 267         return avahi_record_browser_event (client, AVAHI_BROWSER_NEW, message);
 268     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_RECORD_BROWSER, "ItemRemove"))
 269         return avahi_record_browser_event (client, AVAHI_BROWSER_REMOVE, message);
 270     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_RECORD_BROWSER, "CacheExhausted"))
 271         return avahi_record_browser_event (client, AVAHI_BROWSER_CACHE_EXHAUSTED, message);
 272     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_RECORD_BROWSER, "AllForNow"))
 273         return avahi_record_browser_event (client, AVAHI_BROWSER_ALL_FOR_NOW, message);
 274     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_RECORD_BROWSER, "Failure"))
 275         return avahi_record_browser_event (client, AVAHI_BROWSER_FAILURE, message);
 276
 277     else {
 278
 279         fprintf(stderr, "WARNING: Unhandled message: interface=%s, path=%s, member=%s\n",
 280                dbus_message_get_interface(message),
 281                dbus_message_get_path(message),
 282                dbus_message_get_member(message));
 283
 284         return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 285     }
 286
 287     return DBUS_HANDLER_RESULT_HANDLED;
 288
 289 fail:
 290
 291     if (dbus_error_is_set(&error)) {
 292         avahi_client_set_errno(client, avahi_error_dbus_to_number(error.name));
 293         dbus_error_free(&error);
 294     }
 295
 296     client_set_state(client, AVAHI_CLIENT_FAILURE);
 297
 298     return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 299 }
 300
 301 static int get_server_state(AvahiClient *client, int *ret_error) {
 302     DBusMessage *message = NULL, *reply = NULL;
 303     DBusError error;
 304     int32_t state;
 305     int e = AVAHI_ERR_NO_MEMORY;
 306
 307     assert(client);
 308
 309     dbus_error_init(&error);
 310
 311     if (!(message = dbus_message_new_method_call(AVAHI_DBUS_NAME, AVAHI_DBUS_PATH_SERVER, AVAHI_DBUS_INTERFACE_SERVER, "GetState")))
 312         goto fail;
 313
 314     reply = dbus_connection_send_with_reply_and_block (client->bus, message, -1, &error);
 315
 316     if (!reply || dbus_error_is_set (&error))
 317         goto fail;
 318
 319     if (!dbus_message_get_args(reply, &error, DBUS_TYPE_INT32, &state, DBUS_TYPE_INVALID) ||
 320         dbus_error_is_set (&error))
 321         goto fail;
 322
 323     client_set_state(client, (AvahiServerState) state);
 324
 325     dbus_message_unref(message);
 326     dbus_message_unref(reply);
 327
 328     return AVAHI_OK;
 329
 330 fail:
 331     if (dbus_error_is_set(&error)) {
 332         e = avahi_error_dbus_to_number (error.name);
 333         dbus_error_free(&error);
 334     }
 335
 336     if (ret_error)
 337         *ret_error = e;
 338
 339     if (message)
 340         dbus_message_unref(message);
 341     if (reply)
 342         dbus_message_unref(reply);
 343
 344     return e;
 345 }
 346
 347 static int check_version(AvahiClient *client, int *ret_error) {
 348     DBusMessage *message = NULL, *reply  = NULL;
 349     DBusError error;
 350     uint32_t version;
 351     int e = AVAHI_ERR_NO_MEMORY;
 352
 353     assert(client);
 354
 355     dbus_error_init(&error);
 356
 357     if (!(message = dbus_message_new_method_call(AVAHI_DBUS_NAME, AVAHI_DBUS_PATH_SERVER, AVAHI_DBUS_INTERFACE_SERVER, "GetAPIVersion")))
 358         goto fail;
 359
 360     reply = dbus_connection_send_with_reply_and_block (client->bus, message, -1, &error);
 361
 362     if (!reply || dbus_error_is_set (&error)) {
 363         char *version_str;
 364
 365         if (!dbus_error_is_set(&error) || strcmp(error.name, DBUS_ERROR_UNKNOWN_METHOD))
 366             goto fail;
 367
 368         /* If the method GetAPIVersion is not known, we look if
 369          * GetVersionString matches "avahi 0.6" which is the only
 370          * version we support which doesn't have GetAPIVersion() .*/
 371
 372         dbus_message_unref(message);
 373         if (reply) dbus_message_unref(reply);
 374         dbus_error_free(&error);
 375
 376         if (!(message = dbus_message_new_method_call(AVAHI_DBUS_NAME, AVAHI_DBUS_PATH_SERVER, AVAHI_DBUS_INTERFACE_SERVER, "GetVersionString")))
 377             goto fail;
 378
 379         reply = dbus_connection_send_with_reply_and_block (client->bus, message, -1, &error);
 380
 381         if (!reply || dbus_error_is_set (&error))
 382             goto fail;
 383
 384         if (!dbus_message_get_args (reply, &error, DBUS_TYPE_STRING, &version_str, DBUS_TYPE_INVALID) ||
 385             dbus_error_is_set (&error))
 386             goto fail;
 387
 388         version = strcmp(version_str, "avahi 0.6") == 0 ? 0x0201 : 0x0000;
 389
 390     } else {
 391
 392         if (!dbus_message_get_args (reply, &error, DBUS_TYPE_UINT32, &version, DBUS_TYPE_INVALID) ||
 393             dbus_error_is_set(&error))
 394             goto fail;
 395     }
 396
 397     /*fprintf(stderr, "API Version 0x%04x\n", version);*/
 398
 399     if ((version & 0xFF00) != (AVAHI_CLIENT_DBUS_API_SUPPORTED & 0xFF00) ||
 400         (version & 0x00FF) < (AVAHI_CLIENT_DBUS_API_SUPPORTED & 0x00FF)) {
 401         e = AVAHI_ERR_VERSION_MISMATCH;
 402         goto fail;
 403     }
 404
 405     dbus_message_unref(message);
 406     dbus_message_unref(reply);
 407
 408     return AVAHI_OK;
 409
 410 fail:
 411     if (dbus_error_is_set(&error)) {
 412         e = avahi_error_dbus_to_number (error.name);
 413         dbus_error_free(&error);
 414     }
 415
 416     if (ret_error)
 417         *ret_error = e;
 418
 419     if (message)
 420         dbus_message_unref(message);
 421     if (reply)
 422         dbus_message_unref(reply);
 423
 424     return e;
 425 }
 426
 427 static int init_server(AvahiClient *client, int *ret_error) {
 428     int r;
 429
 430     if ((r = check_version(client, ret_error)) < 0)
 431         return r;
 432
 433     if ((r = get_server_state(client, ret_error)) < 0)
 434         return r;
 435
 436     return AVAHI_OK;
 437 }
 438
 439 /* This function acts like dbus_bus_get but creates a private
 440  * connection instead. Eventually this should be replaced by a DBUS
 441  * provided version. */
 442 static DBusConnection* avahi_dbus_bus_get(DBusError *error) {
 443     DBusConnection *c;
 444     const char *a;
 445
 446     if (!(a = getenv("DBUS_SYSTEM_BUS_ADDRESS")) || !*a)
 447         a = DBUS_SYSTEM_BUS_DEFAULT_ADDRESS;
 448
 449     if (!(c = dbus_connection_open_private(a, error)))
 450         return NULL;
 451
 452     dbus_connection_set_exit_on_disconnect(c, FALSE);
 453
 454     if (!dbus_bus_register(c, error)) {
 455         dbus_connection_close(c);
 456         dbus_connection_unref(c);
 457         return NULL;
 458     }
 459
 460     return c;
 461 }
 462
 463 AvahiClient *avahi_client_new(const AvahiPoll *poll_api, AvahiClientFlags flags, AvahiClientCallback callback, void *userdata, int *ret_error) {
 464     AvahiClient *client = NULL;
 465     DBusError error;
 466
 467     dbus_error_init(&error);
 468
 469     if (!(client = avahi_new(AvahiClient, 1))) {
 470         if (ret_error)
 471             *ret_error = AVAHI_ERR_NO_MEMORY;
 472         goto fail;
 473     }
 474
 475     client->poll_api = poll_api;
 476     client->error = AVAHI_OK;
 477     client->callback = callback;
 478     client->userdata = userdata;
 479     client->state = (AvahiClientState) -1;
 480     client->flags = flags;
 481
 482     client->host_name = NULL;
 483     client->host_name_fqdn = NULL;
 484     client->domain_name = NULL;
 485     client->version_string = NULL;
 486     client->local_service_cookie_valid = 0;
 487
 488     AVAHI_LLIST_HEAD_INIT(AvahiEntryGroup, client->groups);
 489     AVAHI_LLIST_HEAD_INIT(AvahiDomainBrowser, client->domain_browsers);
 490     AVAHI_LLIST_HEAD_INIT(AvahiServiceBrowser, client->service_browsers);
 491     AVAHI_LLIST_HEAD_INIT(AvahiServiceTypeBrowser, client->service_type_browsers);
 492     AVAHI_LLIST_HEAD_INIT(AvahiServiceResolver, client->service_resolvers);
 493     AVAHI_LLIST_HEAD_INIT(AvahiHostNameResolver, client->host_name_resolvers);
 494     AVAHI_LLIST_HEAD_INIT(AvahiAddressResolver, client->address_resolvers);
 495     AVAHI_LLIST_HEAD_INIT(AvahiRecordBrowser, client->record_browsers);
 496
 497     if (!(client->bus = avahi_dbus_bus_get(&error)) || dbus_error_is_set(&error)) {
 498         if (ret_error)
 499             *ret_error = AVAHI_ERR_DBUS_ERROR;
 500         goto fail;
 501     }
 502
 503     if (avahi_dbus_connection_glue(client->bus, poll_api) < 0) {
 504         if (ret_error)
 505             *ret_error = AVAHI_ERR_NO_MEMORY; /* Not optimal */
 506         goto fail;
 507     }
 508
 509     if (!dbus_connection_add_filter (client->bus, filter_func, client, NULL)) {
 510         if (ret_error)
 511             *ret_error = AVAHI_ERR_NO_MEMORY;
 512         goto fail;
 513     }
 514
 515     dbus_bus_add_match(
 516         client->bus,
 517         "type='signal', "
 518         "interface='" AVAHI_DBUS_INTERFACE_SERVER "', "
 519         "sender='" AVAHI_DBUS_NAME "', "
 520         "path='" AVAHI_DBUS_PATH_SERVER "'",
 521         &error);
 522
 523     if (dbus_error_is_set(&error))
 524         goto fail;
 525
 526     dbus_bus_add_match (
 527         client->bus,
 528         "type='signal', "
 529         "interface='" DBUS_INTERFACE_DBUS "', "
 530         "sender='" DBUS_SERVICE_DBUS "', "
 531         "path='" DBUS_PATH_DBUS "'",
 532         &error);
 533
 534     if (dbus_error_is_set(&error))
 535         goto fail;
 536
 537     dbus_bus_add_match (
 538         client->bus,
 539         "type='signal', "
 540         "interface='" DBUS_INTERFACE_LOCAL "'",
 541         &error);
 542
 543     if (dbus_error_is_set(&error))
 544         goto fail;
 545
 546
 547     if (!(dbus_bus_name_has_owner(client->bus, AVAHI_DBUS_NAME, &error)) ||
 548         dbus_error_is_set(&error)) {
 549
 550         /* We free the error so its not set, that way the fail target
 551          * will return the NO_DAEMON error rather than a DBUS error */
 552         dbus_error_free(&error);
 553
 554         if (!(flags & AVAHI_CLIENT_NO_FAIL)) {
 555
 556             if (ret_error)
 557                 *ret_error = AVAHI_ERR_NO_DAEMON;
 558
 559             goto fail;
 560         }
 561
 562         /* The user doesn't want this call to fail if the daemon is not
 563          * available, so let's return succesfully */
 564         client_set_state(client, AVAHI_CLIENT_CONNECTING);
 565
 566     } else {
 567
 568         if (init_server(client, ret_error) < 0)
 569             goto fail;
 570     }
 571
 572     return client;
 573
 574 fail:
 575
 576     if (client)
 577         avahi_client_free(client);
 578
 579     if (dbus_error_is_set(&error)) {
 580
 581         if (ret_error) {
 582             if (strcmp(error.name, DBUS_ERROR_FILE_NOT_FOUND) == 0)
 583                 /* DBUS returns this error when the DBUS daemon is not running */
 584                 *ret_error = AVAHI_ERR_NO_DAEMON;
 585             else
 586                 *ret_error = avahi_error_dbus_to_number(error.name);
 587         }
 588
 589         dbus_error_free(&error);
 590     }
 591
 592     return NULL;
 593 }
 594
 595 void avahi_client_free(AvahiClient *client) {
 596     assert(client);
 597
 598     while (client->groups)
 599         avahi_entry_group_free(client->groups);
 600
 601     while (client->domain_browsers)
 602         avahi_domain_browser_free(client->domain_browsers);
 603
 604     while (client->service_browsers)
 605         avahi_service_browser_free(client->service_browsers);
 606
 607     while (client->service_type_browsers)
 608         avahi_service_type_browser_free(client->service_type_browsers);
 609
 610     while (client->service_resolvers)
 611         avahi_service_resolver_free(client->service_resolvers);
 612
 613     while (client->host_name_resolvers)
 614         avahi_host_name_resolver_free(client->host_name_resolvers);
 615
 616     while (client->address_resolvers)
 617         avahi_address_resolver_free(client->address_resolvers);
 618
 619     while (client->record_browsers)
 620         avahi_record_browser_free(client->record_browsers);
 621
 622     if (client->bus) {
 623         dbus_connection_disconnect(client->bus);
 624         dbus_connection_unref(client->bus);
 625     }
 626
 627     avahi_free(client->version_string);
 628     avahi_free(client->host_name);
 629     avahi_free(client->host_name_fqdn);
 630     avahi_free(client->domain_name);
 631
 632     avahi_free(client);
 633 }
 634
 635 static char* avahi_client_get_string_reply_and_block (AvahiClient *client, const char *method, const char *param) {
 636     DBusMessage *message = NULL, *reply = NULL;
 637     DBusError error;
 638     char *ret, *n;
 639
 640     assert(client);
 641     assert(method);
 642
 643     dbus_error_init (&error);
 644
 645     if (!(message = dbus_message_new_method_call (AVAHI_DBUS_NAME, AVAHI_DBUS_PATH_SERVER, AVAHI_DBUS_INTERFACE_SERVER, method))) {
 646         avahi_client_set_errno(client, AVAHI_ERR_NO_MEMORY);
 647         goto fail;
 648     }
 649
 650     if (param) {
 651         if (!dbus_message_append_args (message, DBUS_TYPE_STRING, &param, DBUS_TYPE_INVALID)) {
 652             avahi_client_set_errno (client, AVAHI_ERR_NO_MEMORY);
 653             goto fail;
 654         }
 655     }
 656
 657     reply = dbus_connection_send_with_reply_and_block (client->bus, message, -1, &error);
 658
 659     if (!reply || dbus_error_is_set (&error))
 660         goto fail;
 661
 662     if (!dbus_message_get_args (reply, &error, DBUS_TYPE_STRING, &ret, DBUS_TYPE_INVALID) ||
 663         dbus_error_is_set (&error))
 664         goto fail;
 665
 666     if (!(n = avahi_strdup(ret))) {
 667         avahi_client_set_errno(client, AVAHI_ERR_NO_MEMORY);
 668         goto fail;
 669     }
 670
 671     dbus_message_unref(message);
 672     dbus_message_unref(reply);
 673
 674     return n;
 675
 676 fail:
 677
 678     if (message)
 679         dbus_message_unref(message);
 680     if (reply)
 681         dbus_message_unref(reply);
 682
 683     if (dbus_error_is_set(&error)) {
 684         avahi_client_set_dbus_error(client, &error);
 685         dbus_error_free(&error);
 686     }
 687
 688     return NULL;
 689 }
 690
 691 const char* avahi_client_get_version_string(AvahiClient *client) {
 692     assert(client);
 693
 694     if (!avahi_client_is_connected(client)) {
 695         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 696         return NULL;
 697     }
 698
 699     if (!client->version_string)
 700         client->version_string = avahi_client_get_string_reply_and_block(client, "GetVersionString", NULL);
 701
 702     return client->version_string;
 703 }
 704
 705 const char* avahi_client_get_domain_name(AvahiClient *client) {
 706     assert(client);
 707
 708     if (!avahi_client_is_connected(client)) {
 709         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 710         return NULL;
 711     }
 712
 713     if (!client->domain_name)
 714         client->domain_name = avahi_client_get_string_reply_and_block(client, "GetDomainName", NULL);
 715
 716     return client->domain_name;
 717 }
 718
 719 const char* avahi_client_get_host_name(AvahiClient *client) {
 720     assert(client);
 721
 722     if (!avahi_client_is_connected(client)) {
 723         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 724         return NULL;
 725     }
 726
 727     if (!client->host_name)
 728         client->host_name = avahi_client_get_string_reply_and_block(client, "GetHostName", NULL);
 729
 730     return client->host_name;
 731 }
 732
 733 const char* avahi_client_get_host_name_fqdn (AvahiClient *client) {
 734     assert(client);
 735
 736     if (!avahi_client_is_connected(client)) {
 737         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 738         return NULL;
 739     }
 740
 741     if (!client->host_name_fqdn)
 742         client->host_name_fqdn = avahi_client_get_string_reply_and_block(client, "GetHostNameFqdn", NULL);
 743
 744     return client->host_name_fqdn;
 745 }
 746
 747 AvahiClientState avahi_client_get_state(AvahiClient *client) {
 748     assert(client);
 749
 750     return client->state;
 751 }
 752
 753 int avahi_client_errno(AvahiClient *client) {
 754     assert(client);
 755
 756     return client->error;
 757 }
 758
 759 /* Just for internal use */
 760 int avahi_client_simple_method_call(AvahiClient *client, const char *path, const char *interface, const char *method) {
 761     DBusMessage *message = NULL, *reply = NULL;
 762     DBusError error;
 763     int r = AVAHI_OK;
 764
 765     dbus_error_init(&error);
 766
 767     assert(client);
 768     assert(path);
 769     assert(interface);
 770     assert(method);
 771
 772     if (!(message = dbus_message_new_method_call(AVAHI_DBUS_NAME, path, interface, method))) {
 773         r = avahi_client_set_errno(client, AVAHI_ERR_NO_MEMORY);
 774         goto fail;
 775     }
 776
 777     if (!(reply = dbus_connection_send_with_reply_and_block(client->bus, message, -1, &error)) ||
 778         dbus_error_is_set (&error)) {
 779         r = avahi_client_set_errno(client, AVAHI_ERR_DBUS_ERROR);
 780         goto fail;
 781     }
 782
 783     if (!dbus_message_get_args(reply, &error, DBUS_TYPE_INVALID) ||
 784         dbus_error_is_set (&error)) {
 785         r = avahi_client_set_errno(client, AVAHI_ERR_DBUS_ERROR);
 786         goto fail;
 787     }
 788
 789     dbus_message_unref(message);
 790     dbus_message_unref(reply);
 791
 792     return AVAHI_OK;
 793
 794 fail:
 795     if (dbus_error_is_set(&error)) {
 796         r = avahi_client_set_dbus_error(client, &error);
 797         dbus_error_free(&error);
 798     }
 799
 800     if (message)
 801         dbus_message_unref(message);
 802
 803     if (reply)
 804         dbus_message_unref(reply);
 805
 806     return r;
 807 }
 808
 809 uint32_t avahi_client_get_local_service_cookie(AvahiClient *client) {
 810     DBusMessage *message = NULL, *reply = NULL;
 811     DBusError error;
 812     assert(client);
 813
 814     if (!avahi_client_is_connected(client)) {
 815         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 816         return AVAHI_SERVICE_COOKIE_INVALID;
 817     }
 818
 819     if (client->local_service_cookie_valid)
 820         return client->local_service_cookie;
 821
 822     dbus_error_init (&error);
 823
 824     if (!(message = dbus_message_new_method_call(AVAHI_DBUS_NAME, AVAHI_DBUS_PATH_SERVER, AVAHI_DBUS_INTERFACE_SERVER, "GetLocalServiceCookie"))) {
 825         avahi_client_set_errno(client, AVAHI_ERR_NO_MEMORY);
 826         goto fail;
 827     }
 828
 829     reply = dbus_connection_send_with_reply_and_block (client->bus, message, -1, &error);
 830
 831     if (!reply || dbus_error_is_set (&error))
 832         goto fail;
 833
 834     if (!dbus_message_get_args (reply, &error, DBUS_TYPE_UINT32, &client->local_service_cookie, DBUS_TYPE_INVALID) ||
 835         dbus_error_is_set (&error))
 836         goto fail;
 837
 838     dbus_message_unref(message);
 839     dbus_message_unref(reply);
 840
 841     client->local_service_cookie_valid = 1;
 842     return client->local_service_cookie;
 843
 844 fail:
 845
 846     if (message)
 847         dbus_message_unref(message);
 848     if (reply)
 849         dbus_message_unref(reply);
 850
 851     if (dbus_error_is_set(&error)) {
 852         avahi_client_set_dbus_error(client, &error);
 853         dbus_error_free(&error);
 854     }
 855
 856     return AVAHI_SERVICE_COOKIE_INVALID;
 857 }
 858
 859 int avahi_client_is_connected(AvahiClient *client) {
 860     assert(client);
 861
 862     return client->state == AVAHI_CLIENT_S_RUNNING || client->state == AVAHI_CLIENT_S_REGISTERING || client->state == AVAHI_CLIENT_S_COLLISION;
 863 }