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 int avahi_client_set_errno (AvahiClient *client, int error) {
  43     assert(client);
  44
  45     return client->error = error;
  46 }
  47
  48 int avahi_client_set_dbus_error(AvahiClient *client, DBusError *error) {
  49     assert(client);
  50     assert(error);
  51
  52     return avahi_client_set_errno(client, avahi_error_dbus_to_number(error->name));
  53 }
  54
  55 static void client_set_state (AvahiClient *client, AvahiServerState state) {
  56     assert(client);
  57
  58     if (client->state == state)
  59         return;
  60
  61     client->state = state;
  62
  63     switch (client->state) {
  64         case AVAHI_CLIENT_DISCONNECTED:
  65             if (client->bus) {
  66                 dbus_connection_disconnect(client->bus);
  67                 dbus_connection_unref(client->bus);
  68                 client->bus = NULL;
  69             }
  70
  71             /* Fall through */
  72
  73         case AVAHI_CLIENT_S_COLLISION:
  74         case AVAHI_CLIENT_S_REGISTERING:
  75
  76             /* Clear cached strings */
  77             avahi_free(client->host_name);
  78             avahi_free(client->host_name_fqdn);
  79             avahi_free(client->domain_name);
  80
  81             client->host_name =  NULL;
  82             client->host_name_fqdn = NULL;
  83             client->domain_name = NULL;
  84             break;
  85
  86         case AVAHI_CLIENT_S_INVALID:
  87         case AVAHI_CLIENT_S_RUNNING:
  88             break;
  89
  90     }
  91
  92     if (client->callback)
  93         client->callback (client, state, client->userdata);
  94 }
  95
  96 static DBusHandlerResult filter_func(DBusConnection *bus, DBusMessage *message, void *userdata) {
  97     AvahiClient *client = userdata;
  98     DBusError error;
  99
 100     assert(bus);
 101     assert(message);
 102
 103     dbus_error_init (&error);
 104
 105 /*     fprintf(stderr, "dbus: interface=%s, path=%s, member=%s\n", */
 106 /*             dbus_message_get_interface (message), */
 107 /*             dbus_message_get_path (message), */
 108 /*             dbus_message_get_member (message)); */
 109
 110     if (client->state == AVAHI_CLIENT_DISCONNECTED)
 111         goto fail;
 112
 113     if (dbus_message_is_signal(message, DBUS_INTERFACE_LOCAL, "Disconnected")) {
 114
 115         /* The DBUS server died or kicked us */
 116         client_set_state(client, AVAHI_CLIENT_DISCONNECTED);
 117
 118     } if (dbus_message_is_signal(message, DBUS_INTERFACE_DBUS, "NameOwnerChanged")) {
 119         char *name, *old, *new;
 120
 121         if (!(dbus_message_get_args(
 122                   message, &error,
 123                   DBUS_TYPE_STRING, &name,
 124                   DBUS_TYPE_STRING, &old,
 125                   DBUS_TYPE_STRING, &new,
 126                   DBUS_TYPE_INVALID) || dbus_error_is_set (&error))) {
 127
 128             fprintf(stderr, "WARNING: Failed to parse NameOwnerChanged signal: %s\n", error.message);
 129             goto fail;
 130         }
 131
 132         if (strcmp(name, AVAHI_DBUS_NAME) == 0)
 133
 134             /* Regardless if the server lost or acquired its name or
 135              * if the name was transfered: our services are no longer
 136              * available, so we disconnect ourselves */
 137
 138             client_set_state(client, AVAHI_CLIENT_DISCONNECTED);
 139
 140     } else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_SERVER, "StateChanged")) {
 141         int32_t state;
 142
 143         if (!(dbus_message_get_args(
 144                   message, &error,
 145                   DBUS_TYPE_INT32, &state,
 146                   DBUS_TYPE_INVALID) || dbus_error_is_set (&error))) {
 147             fprintf(stderr, "WARNING: Failed to parse Server.StateChanged signal: %s\n", error.message);
 148             goto fail;
 149         }
 150
 151         client_set_state(client, (AvahiClientState) state);
 152
 153     } else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_ENTRY_GROUP, "StateChanged")) {
 154         const char *path;
 155         AvahiEntryGroup *g;
 156         path = dbus_message_get_path(message);
 157
 158         for (g = client->groups; g; g = g->groups_next)
 159             if (strcmp(g->path, path) == 0)
 160                 break;
 161
 162         if (g) {
 163             int32_t state;
 164             if (!(dbus_message_get_args (message, &error, DBUS_TYPE_INT32, &state, DBUS_TYPE_INVALID)) ||
 165                 dbus_error_is_set(&error)) {
 166                 fprintf(stderr, "WARNING: Failed to parse EntryGroup.StateChanged signal: %s\n", error.message);
 167                 goto fail;
 168             }
 169
 170             avahi_entry_group_set_state(g, state);
 171         }
 172
 173     } else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_DOMAIN_BROWSER, "ItemNew"))
 174         return avahi_domain_browser_event(client, AVAHI_BROWSER_NEW, message);
 175     else if (dbus_message_is_signal (message, AVAHI_DBUS_INTERFACE_DOMAIN_BROWSER, "ItemRemove"))
 176         return avahi_domain_browser_event(client, AVAHI_BROWSER_REMOVE, message);
 177
 178     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_TYPE_BROWSER, "ItemNew"))
 179         return avahi_service_type_browser_event (client, AVAHI_BROWSER_NEW, message);
 180     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_TYPE_BROWSER, "ItemRemove"))
 181         return avahi_service_type_browser_event (client, AVAHI_BROWSER_REMOVE, message);
 182
 183     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_BROWSER, "ItemNew"))
 184         return avahi_service_browser_event (client, AVAHI_BROWSER_NEW, message);
 185     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_BROWSER, "ItemRemove"))
 186         return avahi_service_browser_event (client, AVAHI_BROWSER_REMOVE, message);
 187
 188     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_RESOLVER, "Found"))
 189         return avahi_service_resolver_event (client, AVAHI_RESOLVER_FOUND, message);
 190     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_SERVICE_RESOLVER, "Timeout"))
 191         return avahi_service_resolver_event (client, AVAHI_RESOLVER_TIMEOUT, message);
 192
 193     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_HOST_NAME_RESOLVER, "Found"))
 194         return avahi_host_name_resolver_event (client, AVAHI_RESOLVER_FOUND, message);
 195     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_HOST_NAME_RESOLVER, "Timeout"))
 196         return avahi_host_name_resolver_event (client, AVAHI_RESOLVER_TIMEOUT, message);
 197
 198     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_ADDRESS_RESOLVER, "Found"))
 199         return avahi_address_resolver_event (client, AVAHI_RESOLVER_FOUND, message);
 200     else if (dbus_message_is_signal(message, AVAHI_DBUS_INTERFACE_ADDRESS_RESOLVER, "Timeout"))
 201         return avahi_address_resolver_event (client, AVAHI_RESOLVER_TIMEOUT, message);
 202
 203
 204     return DBUS_HANDLER_RESULT_HANDLED;
 205
 206 fail:
 207
 208     dbus_error_free (&error);
 209     return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 210 }
 211
 212 static int get_server_state(AvahiClient *client, int *ret_error) {
 213     DBusMessage *message, *reply;
 214     DBusError error;
 215     int32_t state;
 216     int e;
 217
 218     assert(client);
 219
 220     dbus_error_init(&error);
 221
 222     if (!(message = dbus_message_new_method_call(AVAHI_DBUS_NAME, AVAHI_DBUS_PATH_SERVER, AVAHI_DBUS_INTERFACE_SERVER, "GetState")))
 223         goto fail;
 224
 225     reply = dbus_connection_send_with_reply_and_block (client->bus, message, -1, &error);
 226     dbus_message_unref(message);
 227
 228     if (!reply)
 229         goto fail;
 230
 231     if (!(dbus_message_get_args(reply, &error, DBUS_TYPE_INT32, &state, DBUS_TYPE_INVALID)))
 232         goto fail;
 233
 234     client_set_state(client, (AvahiServerState) state);
 235
 236     return AVAHI_OK;
 237
 238 fail:
 239     if (dbus_error_is_set(&error)) {
 240         e = avahi_error_dbus_to_number (error.name);
 241         dbus_error_free(&error);
 242     } else
 243         e = AVAHI_ERR_NO_MEMORY;
 244
 245     if (ret_error)
 246         *ret_error = e;
 247
 248     return e;
 249 }
 250
 251 /* This function acts like dbus_bus_get but creates a private
 252  * connection instead */
 253 static DBusConnection*
 254 avahi_dbus_bus_get (DBusError *error)
 255 {
 256     DBusConnection *conn;
 257     const char *env_addr;
 258
 259     env_addr = getenv ("DBUS_SYSTEM_BUS_ADDRESS");
 260
 261     if (env_addr == NULL || (*env_addr == 0))
 262     {
 263         env_addr = DBUS_SYSTEM_BUS_DEFAULT_ADDRESS;
 264     }
 265
 266     conn = dbus_connection_open_private (env_addr, error);
 267
 268     if (!conn)
 269         return NULL;
 270
 271     dbus_connection_set_exit_on_disconnect (conn, FALSE);
 272
 273     if (!dbus_bus_register (conn, error))
 274     {
 275         dbus_connection_close (conn);
 276         dbus_connection_unref (conn);
 277
 278         return NULL;
 279     }
 280
 281     return conn;
 282 }
 283
 284 AvahiClient *avahi_client_new(const AvahiPoll *poll_api, AvahiClientCallback callback, void *userdata, int *ret_error) {
 285     AvahiClient *client = NULL;
 286     DBusError error;
 287
 288     dbus_error_init (&error);
 289
 290     if (!(client = avahi_new(AvahiClient, 1))) {
 291         if (ret_error)
 292             *ret_error = AVAHI_ERR_NO_MEMORY;
 293         goto fail;
 294     }
 295
 296     client->poll_api = poll_api;
 297     client->error = AVAHI_OK;
 298     client->callback = callback;
 299     client->userdata = userdata;
 300     client->state = AVAHI_CLIENT_DISCONNECTED;
 301
 302     client->host_name = NULL;
 303     client->host_name_fqdn = NULL;
 304     client->domain_name = NULL;
 305     client->version_string = NULL;
 306
 307     AVAHI_LLIST_HEAD_INIT(AvahiEntryGroup, client->groups);
 308     AVAHI_LLIST_HEAD_INIT(AvahiDomainBrowser, client->domain_browsers);
 309     AVAHI_LLIST_HEAD_INIT(AvahiServiceBrowser, client->service_browsers);
 310     AVAHI_LLIST_HEAD_INIT(AvahiServiceTypeBrowser, client->service_type_browsers);
 311     AVAHI_LLIST_HEAD_INIT(AvahiServiceResolver, client->service_resolvers);
 312     AVAHI_LLIST_HEAD_INIT(AvahiHostNameResolver, client->host_name_resolvers);
 313     AVAHI_LLIST_HEAD_INIT(AvahiAddressResolver, client->address_resolvers);
 314
 315     if (!(client->bus = avahi_dbus_bus_get(&error)) ||
 316         dbus_error_is_set (&error))
 317         goto fail;
 318
 319     if (avahi_dbus_connection_glue(client->bus, poll_api) < 0) {
 320         if (ret_error)
 321             *ret_error = AVAHI_ERR_NO_MEMORY; /* Not optimal */
 322         goto fail;
 323     }
 324
 325     if (!dbus_connection_add_filter (client->bus, filter_func, client, NULL)) {
 326         if (ret_error)
 327             *ret_error = AVAHI_ERR_NO_MEMORY;
 328         goto fail;
 329     }
 330
 331     dbus_bus_add_match(
 332         client->bus,
 333         "type='signal', "
 334         "interface='" AVAHI_DBUS_INTERFACE_SERVER "', "
 335         "sender='" AVAHI_DBUS_NAME "', "
 336         "path='" AVAHI_DBUS_PATH_SERVER "'",
 337         &error);
 338
 339     if (dbus_error_is_set (&error))
 340         goto fail;
 341
 342     dbus_bus_add_match (
 343         client->bus,
 344         "type='signal', "
 345         "interface='" DBUS_INTERFACE_DBUS "', "
 346         "sender='" DBUS_SERVICE_DBUS "', "
 347         "path='" DBUS_PATH_DBUS "'",
 348         &error);
 349
 350     if (dbus_error_is_set (&error))
 351         goto fail;
 352
 353         dbus_bus_add_match (
 354         client->bus,
 355         "type='signal', "
 356         "interface='" DBUS_INTERFACE_LOCAL "'",
 357         &error);
 358
 359     if (dbus_error_is_set (&error))
 360         goto fail;
 361
 362     if (!(dbus_bus_name_has_owner(client->bus, AVAHI_DBUS_NAME, &error)) ||
 363         dbus_error_is_set(&error)) {
 364
 365         if (ret_error)
 366             *ret_error = AVAHI_ERR_NO_DAEMON;
 367
 368         goto fail;
 369     }
 370
 371     if (get_server_state(client, ret_error) < 0)
 372         goto fail;
 373
 374     return client;
 375
 376 fail:
 377
 378     if (client)
 379         avahi_client_free(client);
 380
 381     if (dbus_error_is_set(&error)) {
 382
 383         if (ret_error)
 384             *ret_error = avahi_error_dbus_to_number(error.name);
 385
 386         dbus_error_free(&error);
 387     }
 388
 389     return NULL;
 390 }
 391
 392 void avahi_client_free(AvahiClient *client) {
 393     assert(client);
 394
 395     while (client->groups)
 396         avahi_entry_group_free(client->groups);
 397
 398     while (client->domain_browsers)
 399         avahi_domain_browser_free(client->domain_browsers);
 400
 401     while (client->service_browsers)
 402         avahi_service_browser_free(client->service_browsers);
 403
 404     while (client->service_type_browsers)
 405         avahi_service_type_browser_free(client->service_type_browsers);
 406
 407     while (client->service_resolvers)
 408         avahi_service_resolver_free(client->service_resolvers);
 409
 410     if (client->bus) {
 411         dbus_connection_disconnect(client->bus);
 412         dbus_connection_unref(client->bus);
 413     }
 414
 415     avahi_free(client->version_string);
 416     avahi_free(client->host_name);
 417     avahi_free(client->host_name_fqdn);
 418     avahi_free(client->domain_name);
 419
 420     avahi_free(client);
 421 }
 422
 423 static char* avahi_client_get_string_reply_and_block (AvahiClient *client, const char *method, const char *param) {
 424     DBusMessage *message = NULL, *reply = NULL;
 425     DBusError error;
 426     char *ret, *n;
 427
 428     assert(client);
 429     assert(method);
 430
 431     dbus_error_init (&error);
 432
 433     if (!(message = dbus_message_new_method_call (AVAHI_DBUS_NAME, AVAHI_DBUS_PATH_SERVER, AVAHI_DBUS_INTERFACE_SERVER, method))) {
 434         avahi_client_set_errno(client, AVAHI_ERR_NO_MEMORY);
 435         goto fail;
 436     }
 437
 438     if (param) {
 439         if (!dbus_message_append_args (message, DBUS_TYPE_STRING, &param, DBUS_TYPE_INVALID)) {
 440             avahi_client_set_errno (client, AVAHI_ERR_NO_MEMORY);
 441             goto fail;
 442         }
 443     }
 444
 445     reply = dbus_connection_send_with_reply_and_block (client->bus, message, -1, &error);
 446
 447     if (!reply || dbus_error_is_set (&error))
 448         goto fail;
 449
 450     if (!dbus_message_get_args (reply, &error, DBUS_TYPE_STRING, &ret, DBUS_TYPE_INVALID) ||
 451         dbus_error_is_set (&error))
 452         goto fail;
 453
 454     if (!(n = avahi_strdup(ret))) {
 455         avahi_client_set_errno(client, AVAHI_ERR_NO_MEMORY);
 456         goto fail;
 457     }
 458
 459     dbus_message_unref(message);
 460     dbus_message_unref(reply);
 461
 462     return n;
 463
 464 fail:
 465
 466     if (message)
 467         dbus_message_unref(message);
 468     if (reply)
 469         dbus_message_unref(reply);
 470
 471     if (dbus_error_is_set(&error)) {
 472         avahi_client_set_dbus_error(client, &error);
 473         dbus_error_free(&error);
 474     }
 475
 476     return NULL;
 477 }
 478
 479 const char* avahi_client_get_version_string (AvahiClient *client) {
 480     assert(client);
 481
 482     if (client->state == AVAHI_CLIENT_DISCONNECTED) {
 483         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 484         return NULL;
 485     }
 486
 487     if (!client->version_string)
 488         client->version_string = avahi_client_get_string_reply_and_block(client, "GetVersionString", NULL);
 489
 490     return client->version_string;
 491 }
 492
 493 const char* avahi_client_get_domain_name (AvahiClient *client) {
 494     assert(client);
 495
 496     if (client->state == AVAHI_CLIENT_DISCONNECTED) {
 497         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 498         return NULL;
 499     }
 500
 501     if (!client->domain_name)
 502         client->domain_name = avahi_client_get_string_reply_and_block(client, "GetDomainName", NULL);
 503
 504     return client->domain_name;
 505 }
 506
 507 const char* avahi_client_get_host_name (AvahiClient *client) {
 508     assert(client);
 509
 510     if (client->state == AVAHI_CLIENT_DISCONNECTED) {
 511         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 512         return NULL;
 513     }
 514
 515     if (!client->host_name)
 516         client->host_name = avahi_client_get_string_reply_and_block(client, "GetHostName", NULL);
 517
 518     return client->host_name;
 519 }
 520
 521 const char* avahi_client_get_host_name_fqdn (AvahiClient *client) {
 522     assert(client);
 523
 524     if (client->state == AVAHI_CLIENT_DISCONNECTED) {
 525         avahi_client_set_errno(client, AVAHI_ERR_BAD_STATE);
 526         return NULL;
 527     }
 528
 529     if (!client->host_name_fqdn)
 530         client->host_name_fqdn = avahi_client_get_string_reply_and_block(client, "GetHostNameFqdn", NULL);
 531
 532     return client->host_name_fqdn;
 533 }
 534
 535 AvahiClientState avahi_client_get_state(AvahiClient *client) {
 536     assert(client);
 537
 538     return client->state;
 539 }
 540
 541 int avahi_client_errno(AvahiClient *client) {
 542     assert(client);
 543
 544     return client->error;
 545 }
 546
 547 /* Just for internal use */
 548 int avahi_client_simple_method_call(AvahiClient *client, const char *path, const char *interface, const char *method) {
 549     DBusMessage *message = NULL, *reply = NULL;
 550     DBusError error;
 551     int r = AVAHI_OK;
 552
 553     dbus_error_init(&error);
 554
 555     assert(client);
 556     assert(path);
 557     assert(interface);
 558     assert(method);
 559
 560     if (!(message = dbus_message_new_method_call(AVAHI_DBUS_NAME, path, interface, method))) {
 561         r = avahi_client_set_errno(client, AVAHI_ERR_NO_MEMORY);
 562         goto fail;
 563     }
 564
 565     if (!(reply = dbus_connection_send_with_reply_and_block(client->bus, message, -1, &error)) ||
 566         dbus_error_is_set (&error)) {
 567         r = avahi_client_set_errno(client, AVAHI_ERR_DBUS_ERROR);
 568         goto fail;
 569     }
 570
 571     if (!dbus_message_get_args(reply, &error, DBUS_TYPE_INVALID) ||
 572         dbus_error_is_set (&error)) {
 573         r = avahi_client_set_errno(client, AVAHI_ERR_DBUS_ERROR);
 574         goto fail;
 575     }
 576
 577     dbus_message_unref(message);
 578     dbus_message_unref(reply);
 579
 580     return AVAHI_OK;
 581
 582 fail:
 583     if (dbus_error_is_set(&error)) {
 584         r = avahi_client_set_dbus_error(client, &error);
 585         dbus_error_free(&error);
 586     }
 587
 588     if (message)
 589         dbus_message_unref(message);
 590
 591     if (reply)
 592         dbus_message_unref(reply);
 593
 594     return r;
 595 }
 596