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