git.meshlink.io Git - catta/blob - avahi-core/entry.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 <sys/socket.h>
  27 #include <arpa/inet.h>
  28 #include <string.h>
  29 #include <sys/utsname.h>
  30 #include <unistd.h>
  31 #include <errno.h>
  32 #include <stdio.h>
  33 #include <assert.h>
  34 #include <stdlib.h>
  35
  36 #include <avahi-common/domain.h>
  37 #include <avahi-common/timeval.h>
  38 #include <avahi-common/malloc.h>
  39 #include <avahi-common/error.h>
  40
  41 #include "server.h"
  42 #include "iface.h"
  43 #include "socket.h"
  44 #include "browse.h"
  45 #include "log.h"
  46 #include "util.h"
  47
  48 void avahi_entry_free(AvahiServer*s, AvahiEntry *e) {
  49     AvahiEntry *t;
  50
  51     assert(s);
  52     assert(e);
  53
  54     avahi_goodbye_entry(s, e, 1, 1);
  55
  56     /* Remove from linked list */
  57     AVAHI_LLIST_REMOVE(AvahiEntry, entries, s->entries, e);
  58
  59     /* Remove from hash table indexed by name */
  60     t = avahi_hashmap_lookup(s->entries_by_key, e->record->key);
  61     AVAHI_LLIST_REMOVE(AvahiEntry, by_key, t, e);
  62     if (t)
  63         avahi_hashmap_replace(s->entries_by_key, t->record->key, t);
  64     else
  65         avahi_hashmap_remove(s->entries_by_key, e->record->key);
  66
  67     /* Remove from associated group */
  68     if (e->group)
  69         AVAHI_LLIST_REMOVE(AvahiEntry, by_group, e->group->entries, e);
  70
  71     avahi_record_unref(e->record);
  72     avahi_free(e);
  73 }
  74
  75 void avahi_entry_group_free(AvahiServer *s, AvahiSEntryGroup *g) {
  76     assert(s);
  77     assert(g);
  78
  79     while (g->entries)
  80         avahi_entry_free(s, g->entries);
  81
  82     if (g->register_time_event)
  83         avahi_time_event_free(g->register_time_event);
  84
  85     AVAHI_LLIST_REMOVE(AvahiSEntryGroup, groups, s->groups, g);
  86     avahi_free(g);
  87 }
  88
  89 void avahi_cleanup_dead_entries(AvahiServer *s) {
  90     assert(s);
  91
  92     if (s->need_group_cleanup) {
  93         AvahiSEntryGroup *g, *next;
  94
  95         for (g = s->groups; g; g = next) {
  96             next = g->groups_next;
  97
  98             if (g->dead)
  99                 avahi_entry_group_free(s, g);
 100         }
 101
 102         s->need_group_cleanup = 0;
 103     }
 104
 105     if (s->need_entry_cleanup) {
 106         AvahiEntry *e, *next;
 107
 108         for (e = s->entries; e; e = next) {
 109             next = e->entries_next;
 110
 111             if (e->dead)
 112                 avahi_entry_free(s, e);
 113         }
 114
 115         s->need_entry_cleanup = 0;
 116     }
 117
 118     if (s->need_browser_cleanup)
 119         avahi_browser_cleanup(s);
 120 }
 121
 122 static int check_record_conflict(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, AvahiRecord *r, AvahiPublishFlags flags) {
 123     AvahiEntry *e;
 124
 125     assert(s);
 126     assert(r);
 127
 128     for (e = avahi_hashmap_lookup(s->entries_by_key, r->key); e; e = e->by_key_next) {
 129         if (e->dead)
 130             continue;
 131
 132         if (!(flags & AVAHI_PUBLISH_UNIQUE) && !(e->flags & AVAHI_PUBLISH_UNIQUE))
 133             continue;
 134
 135         if ((flags & AVAHI_PUBLISH_ALLOW_MULTIPLE) && (e->flags & AVAHI_PUBLISH_ALLOW_MULTIPLE) )
 136             continue;
 137
 138         if ((interface <= 0 ||
 139              e->interface <= 0 ||
 140              e->interface == interface) &&
 141             (protocol == AVAHI_PROTO_UNSPEC ||
 142              e->protocol == AVAHI_PROTO_UNSPEC ||
 143              e->protocol == protocol))
 144
 145             return -1;
 146     }
 147
 148     return 0;
 149 }
 150
 151 int avahi_server_add(
 152     AvahiServer *s,
 153     AvahiSEntryGroup *g,
 154     AvahiIfIndex interface,
 155     AvahiProtocol protocol,
 156     AvahiPublishFlags flags,
 157     AvahiRecord *r) {
 158
 159     assert(s);
 160     assert(r);
 161
 162     if (!AVAHI_IF_VALID(interface))
 163         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
 164
 165     if (!AVAHI_PROTO_VALID(protocol))
 166         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
 167
 168     if (!AVAHI_FLAGS_VALID(
 169             flags,
 170             AVAHI_PUBLISH_NO_ANNOUNCE|
 171             AVAHI_PUBLISH_NO_PROBE|
 172             AVAHI_PUBLISH_UNIQUE|
 173             AVAHI_PUBLISH_ALLOW_MULTIPLE|
 174             AVAHI_PUBLISH_IS_PROXY|
 175             AVAHI_PUBLISH_UPDATE))
 176         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
 177
 178     if (!avahi_is_valid_domain_name(r->key->name))
 179         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
 180
 181     if (r->ttl == 0)
 182         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_TTL);
 183
 184     if (avahi_key_is_pattern(r->key))
 185         return avahi_server_set_errno(s, AVAHI_ERR_IS_PATTERN);
 186
 187     if (!avahi_record_is_valid(r))
 188         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_RECORD);
 189
 190     if (flags & AVAHI_PUBLISH_UPDATE) {
 191         AvahiEntry *e;
 192         AvahiRecord *old_record;
 193         int is_first = 1;
 194
 195         /* Update and existing record */
 196
 197         /* Find the first matching entry */
 198         for (e = avahi_hashmap_lookup(s->entries_by_key, r->key); e; e = e->by_key_next) {
 199             if (!e->dead && e->group == g && e->interface == interface && e->protocol == protocol)
 200                 break;
 201
 202             is_first = 0;
 203         }
 204
 205         /* Hmm, nothing found? */
 206         if (!e)
 207             return avahi_server_set_errno(s, AVAHI_ERR_NOT_FOUND);
 208
 209         /* Update the entry */
 210         old_record = e->record;
 211         e->record = avahi_record_ref(r);
 212         e->flags = flags;
 213
 214         /* Announce our changes when needed */
 215         if (!avahi_record_equal_no_ttl(old_record, r) && (!g || g->state != AVAHI_ENTRY_GROUP_UNCOMMITED)) {
 216
 217             /* Remove the old entry from all caches, if needed */
 218             if (!(e->flags & AVAHI_PUBLISH_UNIQUE))
 219                 avahi_goodbye_entry(s, e, 1, 0);
 220
 221             /* Reannounce our updated entry */
 222             avahi_reannounce_entry(s, e);
 223         }
 224
 225         /* If we were the first entry in the list, we need to update the key */
 226         if (is_first)
 227             avahi_hashmap_replace(s->entries_by_key, e->record->key, e);
 228
 229         avahi_record_unref(old_record);
 230
 231     } else {
 232         AvahiEntry *e, *t;
 233
 234         /* Add a new record */
 235
 236         if (check_record_conflict(s, interface, protocol, r, flags) < 0)
 237             return avahi_server_set_errno(s, AVAHI_ERR_LOCAL_COLLISION);
 238
 239         if (!(e = avahi_new(AvahiEntry, 1)))
 240             return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 241
 242         e->server = s;
 243         e->record = avahi_record_ref(r);
 244         e->group = g;
 245         e->interface = interface;
 246         e->protocol = protocol;
 247         e->flags = flags;
 248         e->dead = 0;
 249
 250         AVAHI_LLIST_HEAD_INIT(AvahiAnnouncer, e->announcers);
 251
 252         AVAHI_LLIST_PREPEND(AvahiEntry, entries, s->entries, e);
 253
 254         /* Insert into hash table indexed by name */
 255         t = avahi_hashmap_lookup(s->entries_by_key, e->record->key);
 256         AVAHI_LLIST_PREPEND(AvahiEntry, by_key, t, e);
 257         avahi_hashmap_replace(s->entries_by_key, e->record->key, t);
 258
 259         /* Insert into group list */
 260         if (g)
 261             AVAHI_LLIST_PREPEND(AvahiEntry, by_group, g->entries, e);
 262
 263         avahi_announce_entry(s, e);
 264     }
 265
 266     return AVAHI_OK;
 267 }
 268
 269 const AvahiRecord *avahi_server_iterate(AvahiServer *s, AvahiSEntryGroup *g, void **state) {
 270     AvahiEntry **e = (AvahiEntry**) state;
 271     assert(s);
 272     assert(e);
 273
 274     if (!*e)
 275         *e = g ? g->entries : s->entries;
 276
 277     while (*e && (*e)->dead)
 278         *e = g ? (*e)->by_group_next : (*e)->entries_next;
 279
 280     if (!*e)
 281         return NULL;
 282
 283     return avahi_record_ref((*e)->record);
 284 }
 285
 286 int avahi_server_dump(AvahiServer *s, AvahiDumpCallback callback, void* userdata) {
 287     AvahiEntry *e;
 288
 289     assert(s);
 290     assert(callback);
 291
 292     callback(";;; ZONE DUMP FOLLOWS ;;;", userdata);
 293
 294     for (e = s->entries; e; e = e->entries_next) {
 295         char *t;
 296         char ln[256];
 297
 298         if (e->dead)
 299             continue;
 300
 301         if (!(t = avahi_record_to_string(e->record)))
 302             return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 303
 304         snprintf(ln, sizeof(ln), "%s ; iface=%i proto=%i", t, e->interface, e->protocol);
 305         avahi_free(t);
 306
 307         callback(ln, userdata);
 308     }
 309
 310     avahi_dump_caches(s->monitor, callback, userdata);
 311
 312     if (s->wide_area_lookup_engine)
 313         avahi_wide_area_cache_dump(s->wide_area_lookup_engine, callback, userdata);
 314     return AVAHI_OK;
 315 }
 316
 317 int avahi_server_add_ptr(
 318     AvahiServer *s,
 319     AvahiSEntryGroup *g,
 320     AvahiIfIndex interface,
 321     AvahiProtocol protocol,
 322     AvahiPublishFlags flags,
 323     uint32_t ttl,
 324     const char *name,
 325     const char *dest) {
 326
 327     AvahiRecord *r;
 328     int ret;
 329
 330     assert(s);
 331     assert(dest);
 332
 333     if ((name && !avahi_is_valid_domain_name(name)) || !avahi_is_valid_domain_name(dest))
 334         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
 335
 336     if (!name)
 337         name = s->host_name_fqdn;
 338
 339     if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_PTR, ttl)))
 340         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 341
 342     r->data.ptr.name = avahi_normalize_name_strdup(dest);
 343     ret = avahi_server_add(s, g, interface, protocol, flags, r);
 344     avahi_record_unref(r);
 345     return ret;
 346 }
 347
 348 int avahi_server_add_address(
 349     AvahiServer *s,
 350     AvahiSEntryGroup *g,
 351     AvahiIfIndex interface,
 352     AvahiProtocol protocol,
 353     AvahiPublishFlags flags,
 354     const char *name,
 355     AvahiAddress *a) {
 356
 357     char *n = NULL;
 358     int ret = AVAHI_OK;
 359
 360     assert(s);
 361     assert(a);
 362
 363     if (!AVAHI_IF_VALID(interface))
 364         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
 365
 366     if (!AVAHI_PROTO_VALID(protocol) || !AVAHI_PROTO_VALID(a->proto))
 367         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
 368
 369     if (!AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_REVERSE|AVAHI_PUBLISH_NO_ANNOUNCE|AVAHI_PUBLISH_NO_PROBE|AVAHI_PUBLISH_UPDATE))
 370         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
 371
 372     if (name && !avahi_is_valid_domain_name(name))
 373         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
 374
 375     if (!name)
 376         name = s->host_name_fqdn;
 377     else {
 378         if (!(n = avahi_normalize_name_strdup(name)))
 379             return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 380
 381         name = n;
 382     }
 383
 384     if (a->proto == AVAHI_PROTO_INET) {
 385         AvahiRecord  *r;
 386
 387         if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_A, AVAHI_DEFAULT_TTL_HOST_NAME))) {
 388             ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 389             goto fail;
 390         }
 391
 392         r->data.a.address = a->data.ipv4;
 393         ret = avahi_server_add(s, g, interface, protocol, (flags & ~ AVAHI_PUBLISH_NO_REVERSE) | AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
 394         avahi_record_unref(r);
 395
 396         if (ret < 0)
 397             goto fail;
 398
 399         if (!(flags & AVAHI_PUBLISH_NO_REVERSE)) {
 400             char *reverse;
 401
 402             if (!(reverse = avahi_reverse_lookup_name_ipv4(&a->data.ipv4))) {
 403                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 404                 goto fail;
 405             }
 406
 407             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
 408             avahi_free(reverse);
 409         }
 410
 411     } else {
 412         AvahiRecord *r;
 413
 414         assert(a->proto == AVAHI_PROTO_INET6);
 415
 416         if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_AAAA, AVAHI_DEFAULT_TTL_HOST_NAME))) {
 417             ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 418             goto fail;
 419         }
 420
 421         r->data.aaaa.address = a->data.ipv6;
 422         ret = avahi_server_add(s, g, interface, protocol, (flags & ~ AVAHI_PUBLISH_NO_REVERSE) | AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
 423         avahi_record_unref(r);
 424
 425         if (ret < 0)
 426             goto fail;
 427
 428         if (!(flags & AVAHI_PUBLISH_NO_REVERSE)) {
 429             char *reverse;
 430
 431             if (!(reverse = avahi_reverse_lookup_name_ipv6_arpa(&a->data.ipv6))) {
 432                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 433                 goto fail;
 434             }
 435
 436             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
 437             avahi_free(reverse);
 438
 439             if (ret < 0)
 440                 goto fail;
 441
 442             if (!(reverse = avahi_reverse_lookup_name_ipv6_int(&a->data.ipv6))) {
 443                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 444                 goto fail;
 445             }
 446
 447             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
 448             avahi_free(reverse);
 449         }
 450     }
 451
 452 fail:
 453
 454     avahi_free(n);
 455
 456     return ret;
 457 }
 458
 459 static int server_add_txt_strlst_nocopy(
 460     AvahiServer *s,
 461     AvahiSEntryGroup *g,
 462     AvahiIfIndex interface,
 463     AvahiProtocol protocol,
 464     AvahiPublishFlags flags,
 465     uint32_t ttl,
 466     const char *name,
 467     AvahiStringList *strlst) {
 468
 469     AvahiRecord *r;
 470     int ret;
 471
 472     assert(s);
 473
 474     if (!(r = avahi_record_new_full(name ? name : s->host_name_fqdn, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_TXT, ttl))) {
 475         avahi_string_list_free(strlst);
 476         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 477     }
 478
 479     r->data.txt.string_list = strlst;
 480     ret = avahi_server_add(s, g, interface, protocol, flags, r);
 481     avahi_record_unref(r);
 482
 483     return ret;
 484 }
 485
 486 static AvahiStringList *add_magic_cookie(
 487     AvahiServer *s,
 488     AvahiStringList *strlst) {
 489
 490     assert(s);
 491
 492     if (!s->config.add_service_cookie)
 493         return strlst;
 494
 495     if (avahi_string_list_find(strlst, AVAHI_SERVICE_COOKIE))
 496         /* This string list already contains a magic cookie */
 497         return strlst;
 498
 499     return avahi_string_list_add_printf(strlst, AVAHI_SERVICE_COOKIE"=%u", s->local_service_cookie);
 500 }
 501
 502 static int server_add_service_strlst_nocopy(
 503     AvahiServer *s,
 504     AvahiSEntryGroup *g,
 505     AvahiIfIndex interface,
 506     AvahiProtocol protocol,
 507     AvahiPublishFlags flags,
 508     const char *name,
 509     const char *type,
 510     const char *domain,
 511     const char *host,
 512     uint16_t port,
 513     AvahiStringList *strlst) {
 514
 515     char ptr_name[AVAHI_DOMAIN_NAME_MAX], svc_name[AVAHI_DOMAIN_NAME_MAX], enum_ptr[AVAHI_DOMAIN_NAME_MAX], *h = NULL;
 516     AvahiRecord *r = NULL;
 517     int ret = AVAHI_OK;
 518
 519     assert(s);
 520     assert(type);
 521     assert(name);
 522
 523     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
 524     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
 525     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_COOKIE|AVAHI_PUBLISH_IS_PROXY|AVAHI_PUBLISH_UPDATE), AVAHI_ERR_INVALID_FLAGS);
 526     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
 527     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
 528     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
 529     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !host || avahi_is_valid_domain_name(host), AVAHI_ERR_INVALID_HOST_NAME);
 530
 531     if (!domain)
 532         domain = s->domain_name;
 533
 534     if (!host)
 535         host = s->host_name_fqdn;
 536
 537     if (!(h = avahi_normalize_name_strdup(host))) {
 538         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 539         goto fail;
 540     }
 541
 542     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0 ||
 543         (ret = avahi_service_name_join(ptr_name, sizeof(ptr_name), NULL, type, domain)) < 0 ||
 544         (ret = avahi_service_name_join(enum_ptr, sizeof(enum_ptr), NULL, "_services._dns-sd._udp", domain)) < 0) {
 545         avahi_server_set_errno(s, ret);
 546         goto fail;
 547     }
 548
 549     /* Add service enumeration PTR record */
 550
 551     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, flags & AVAHI_PUBLISH_IS_PROXY, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
 552         goto fail;
 553
 554     /* Add SRV record */
 555
 556     if (!(r = avahi_record_new_full(svc_name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
 557         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 558         goto fail;
 559     }
 560
 561     r->data.srv.priority = 0;
 562     r->data.srv.weight = 0;
 563     r->data.srv.port = port;
 564     r->data.srv.name = h;
 565     h = NULL;
 566     ret = avahi_server_add(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE, r);
 567     avahi_record_unref(r);
 568
 569     if (ret < 0)
 570         goto fail;
 571
 572     /* Add TXT record */
 573
 574     if (!(flags & AVAHI_PUBLISH_NO_COOKIE))
 575         strlst = add_magic_cookie(s, strlst);
 576
 577     ret = server_add_txt_strlst_nocopy(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL, svc_name, strlst);
 578     strlst = NULL;
 579
 580     if (ret < 0)
 581         goto fail;
 582
 583     /* Add service type enumeration record */
 584
 585     ret = avahi_server_add_ptr(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY), AVAHI_DEFAULT_TTL, enum_ptr, ptr_name);
 586
 587 fail:
 588
 589     avahi_string_list_free(strlst);
 590     avahi_free(h);
 591
 592     return ret;
 593 }
 594
 595 int avahi_server_add_service_strlst(
 596     AvahiServer *s,
 597     AvahiSEntryGroup *g,
 598     AvahiIfIndex interface,
 599     AvahiProtocol protocol,
 600     AvahiPublishFlags flags,
 601     const char *name,
 602     const char *type,
 603     const char *domain,
 604     const char *host,
 605     uint16_t port,
 606     AvahiStringList *strlst) {
 607
 608     assert(s);
 609     assert(type);
 610     assert(name);
 611
 612     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_copy(strlst));
 613 }
 614
 615 int avahi_server_add_service_va(
 616     AvahiServer *s,
 617     AvahiSEntryGroup *g,
 618     AvahiIfIndex interface,
 619     AvahiProtocol protocol,
 620     AvahiPublishFlags flags,
 621     const char *name,
 622     const char *type,
 623     const char *domain,
 624     const char *host,
 625     uint16_t port,
 626     va_list va) {
 627
 628     assert(s);
 629     assert(type);
 630     assert(name);
 631
 632     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_new_va(va));
 633 }
 634
 635 int avahi_server_add_service(
 636     AvahiServer *s,
 637     AvahiSEntryGroup *g,
 638     AvahiIfIndex interface,
 639     AvahiProtocol protocol,
 640     AvahiPublishFlags flags,
 641     const char *name,
 642     const char *type,
 643     const char *domain,
 644     const char *host,
 645     uint16_t port,
 646     ... ){
 647
 648     va_list va;
 649     int ret;
 650
 651     va_start(va, port);
 652     ret = avahi_server_add_service_va(s, g, interface, protocol, flags, name, type, domain, host, port, va);
 653     va_end(va);
 654
 655     return ret;
 656 }
 657
 658 static int server_update_service_txt_strlst_nocopy(
 659     AvahiServer *s,
 660     AvahiSEntryGroup *g,
 661     AvahiIfIndex interface,
 662     AvahiProtocol protocol,
 663     AvahiPublishFlags flags,
 664     const char *name,
 665     const char *type,
 666     const char *domain,
 667     AvahiStringList *strlst) {
 668
 669     char svc_name[AVAHI_DOMAIN_NAME_MAX];
 670     int ret = AVAHI_OK;
 671
 672     assert(s);
 673     assert(type);
 674     assert(name);
 675
 676     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
 677     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
 678     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_IS_PROXY|AVAHI_PUBLISH_NO_COOKIE), AVAHI_ERR_INVALID_FLAGS);
 679     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
 680     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
 681     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
 682
 683     if (!domain)
 684         domain = s->domain_name;
 685
 686     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0) {
 687         avahi_server_set_errno(s, ret);
 688         goto fail;
 689     }
 690
 691     /* Add TXT record */
 692     if (!(flags & AVAHI_PUBLISH_NO_COOKIE))
 693         strlst = add_magic_cookie(s, strlst);
 694
 695     ret = server_add_txt_strlst_nocopy(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_UPDATE, AVAHI_DEFAULT_TTL, svc_name, strlst);
 696     strlst = NULL;
 697
 698 fail:
 699
 700     avahi_string_list_free(strlst);
 701
 702     return ret;
 703 }
 704
 705 int avahi_server_update_service_txt_strlst(
 706     AvahiServer *s,
 707     AvahiSEntryGroup *g,
 708     AvahiIfIndex interface,
 709     AvahiProtocol protocol,
 710     AvahiPublishFlags flags,
 711     const char *name,
 712     const char *type,
 713     const char *domain,
 714     AvahiStringList *strlst) {
 715
 716     return server_update_service_txt_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, avahi_string_list_copy(strlst));
 717 }
 718
 719 /** Update the TXT record for a service with the NULL terminated list of strings of the va_list. */
 720 int avahi_server_update_service_txt_va(
 721     AvahiServer *s,
 722     AvahiSEntryGroup *g,
 723     AvahiIfIndex interface,
 724     AvahiProtocol protocol,
 725     AvahiPublishFlags flags,
 726     const char *name,
 727     const char *type,
 728     const char *domain,
 729     va_list va) {
 730
 731     return server_update_service_txt_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, avahi_string_list_new_va(va));
 732 }
 733
 734 /** Update the TXT record for a service with the NULL termonate list of strings */
 735 int avahi_server_update_service_txt(
 736     AvahiServer *s,
 737     AvahiSEntryGroup *g,
 738     AvahiIfIndex interface,
 739     AvahiProtocol protocol,
 740     AvahiPublishFlags flags,
 741     const char *name,
 742     const char *type,
 743     const char *domain,
 744     ...) {
 745
 746     va_list va;
 747     int ret;
 748
 749     va_start(va, domain);
 750     ret = avahi_server_update_service_txt_va(s, g, interface, protocol, flags, name, type, domain, va);
 751     va_end(va);
 752
 753     return ret;
 754 }
 755
 756 int avahi_server_add_service_subtype(
 757     AvahiServer *s,
 758     AvahiSEntryGroup *g,
 759     AvahiIfIndex interface,
 760     AvahiProtocol protocol,
 761     AvahiPublishFlags flags,
 762     const char *name,
 763     const char *type,
 764     const char *domain,
 765     const char *subtype) {
 766
 767     int ret = AVAHI_OK;
 768     char svc_name[AVAHI_DOMAIN_NAME_MAX], ptr_name[AVAHI_DOMAIN_NAME_MAX];
 769
 770     assert(name);
 771     assert(type);
 772     assert(subtype);
 773
 774     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
 775     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
 776     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, 0), AVAHI_ERR_INVALID_FLAGS);
 777     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
 778     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
 779     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
 780     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_subtype(subtype), AVAHI_ERR_INVALID_SERVICE_SUBTYPE);
 781
 782     if (!domain)
 783         domain = s->domain_name;
 784
 785     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0 ||
 786         (ret = avahi_service_name_join(ptr_name, sizeof(ptr_name), NULL, subtype, domain)) < 0) {
 787         avahi_server_set_errno(s, ret);
 788         goto fail;
 789     }
 790
 791     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, flags & AVAHI_PUBLISH_IS_PROXY, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
 792         goto fail;
 793
 794 fail:
 795
 796     return ret;
 797 }
 798
 799 static void hexstring(char *s, size_t sl, const void *p, size_t pl) {
 800     static const char hex[] = "0123456789abcdef";
 801     int b = 0;
 802     const uint8_t *k = p;
 803
 804     while (sl > 1 && pl > 0) {
 805         *(s++) = hex[(b ? *k : *k >> 4) & 0xF];
 806
 807         if (b) {
 808             k++;
 809             pl--;
 810         }
 811
 812         b = !b;
 813
 814         sl--;
 815     }
 816
 817     if (sl > 0)
 818         *s = 0;
 819 }
 820
 821 int avahi_server_add_dns_server_address(
 822     AvahiServer *s,
 823     AvahiSEntryGroup *g,
 824     AvahiIfIndex interface,
 825     AvahiProtocol protocol,
 826     AvahiPublishFlags flags,
 827     const char *domain,
 828     AvahiDNSServerType type,
 829     const AvahiAddress *address,
 830     uint16_t port /** should be 53 */) {
 831
 832     AvahiRecord *r;
 833     int ret;
 834     char n[64], h[64];
 835
 836     assert(s);
 837     assert(address);
 838
 839     if (!AVAHI_IF_VALID(interface))
 840         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
 841
 842     if (!AVAHI_PROTO_VALID(protocol) || !AVAHI_PROTO_VALID(address->proto))
 843         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
 844
 845     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
 846         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
 847
 848     if (port == 0)
 849         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
 850
 851     if (domain && !avahi_is_valid_domain_name(domain))
 852         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
 853
 854     if (!domain)
 855         domain = s->domain_name;
 856
 857     if (address->proto == AVAHI_PROTO_INET) {
 858         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv4Address));
 859         snprintf(n, sizeof(n), "ip-%s.%s", h, domain);
 860         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_A, AVAHI_DEFAULT_TTL_HOST_NAME);
 861         r->data.a.address = address->data.ipv4;
 862     } else {
 863         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv6Address));
 864         snprintf(n, sizeof(n), "ip6-%s.%s", h, domain);
 865         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_AAAA, AVAHI_DEFAULT_TTL_HOST_NAME);
 866         r->data.aaaa.address = address->data.ipv6;
 867     }
 868
 869     if (!r)
 870         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 871
 872     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
 873     avahi_record_unref(r);
 874
 875     if (ret < 0)
 876         return ret;
 877
 878     return avahi_server_add_dns_server_name(s, g, interface, protocol, flags, domain, type, n, port);
 879 }
 880
 881 int avahi_server_add_dns_server_name(
 882     AvahiServer *s,
 883     AvahiSEntryGroup *g,
 884     AvahiIfIndex interface,
 885     AvahiProtocol protocol,
 886     AvahiPublishFlags flags,
 887     const char *domain,
 888     AvahiDNSServerType type,
 889     const char *name,
 890     uint16_t port /** should be 53 */) {
 891
 892     int ret = -1;
 893     char t[256], *d = NULL, *n = NULL;
 894     AvahiRecord *r;
 895
 896     assert(s);
 897     assert(name);
 898
 899     if (!AVAHI_IF_VALID(interface))
 900         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
 901
 902     if (!AVAHI_PROTO_VALID(protocol))
 903         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
 904
 905     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
 906         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
 907
 908     if (port == 0)
 909         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
 910
 911     if (!avahi_is_valid_domain_name(name))
 912         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
 913
 914     if (domain && !avahi_is_valid_domain_name(domain))
 915         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
 916
 917     if (!domain)
 918         domain = s->domain_name;
 919
 920     if (!(n = avahi_normalize_name_strdup(name)) ||
 921         !(d = avahi_normalize_name_strdup(domain))) {
 922         avahi_free(n);
 923         avahi_free(d);
 924         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 925     }
 926
 927     snprintf(t, sizeof(t), "%s.%s", type == AVAHI_DNS_SERVER_RESOLVE ? "_domain._udp" : "_dns-update._udp", d);
 928     avahi_free(d);
 929
 930     if (!(r = avahi_record_new_full(t, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
 931         avahi_free(n);
 932         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 933     }
 934
 935     r->data.srv.priority = 0;
 936     r->data.srv.weight = 0;
 937     r->data.srv.port = port;
 938     r->data.srv.name = n;
 939     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_NULL, r);
 940     avahi_record_unref(r);
 941
 942     return ret;
 943 }
 944
 945 void avahi_s_entry_group_change_state(AvahiSEntryGroup *g, AvahiEntryGroupState state) {
 946     assert(g);
 947
 948     if (g->state == state)
 949         return;
 950
 951     assert(state <= AVAHI_ENTRY_GROUP_COLLISION);
 952
 953     if (g->state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
 954
 955         /* If the entry group was established for a time longer then
 956          * 5s, reset the establishment trial counter */
 957
 958         if (avahi_age(&g->established_at) > 5000000)
 959             g->n_register_try = 0;
 960     }
 961
 962     if (state == AVAHI_ENTRY_GROUP_ESTABLISHED)
 963
 964         /* If the entry group is now established, remember the time
 965          * this happened */
 966
 967         gettimeofday(&g->established_at, NULL);
 968
 969     g->state = state;
 970
 971     if (g->callback)
 972         g->callback(g->server, g, state, g->userdata);
 973 }
 974
 975 AvahiSEntryGroup *avahi_s_entry_group_new(AvahiServer *s, AvahiSEntryGroupCallback callback, void* userdata) {
 976     AvahiSEntryGroup *g;
 977
 978     assert(s);
 979
 980     if (!(g = avahi_new(AvahiSEntryGroup, 1))) {
 981         avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 982         return NULL;
 983     }
 984
 985     g->server = s;
 986     g->callback = callback;
 987     g->userdata = userdata;
 988     g->dead = 0;
 989     g->state = AVAHI_ENTRY_GROUP_UNCOMMITED;
 990     g->n_probing = 0;
 991     g->n_register_try = 0;
 992     g->register_time_event = NULL;
 993     g->register_time.tv_sec = 0;
 994     g->register_time.tv_usec = 0;
 995     AVAHI_LLIST_HEAD_INIT(AvahiEntry, g->entries);
 996
 997     AVAHI_LLIST_PREPEND(AvahiSEntryGroup, groups, s->groups, g);
 998     return g;
 999 }
1000
1001 void avahi_s_entry_group_free(AvahiSEntryGroup *g) {
1002     AvahiEntry *e;
1003
1004     assert(g);
1005     assert(g->server);
1006
1007     for (e = g->entries; e; e = e->by_group_next) {
1008         if (!e->dead) {
1009             avahi_goodbye_entry(g->server, e, 1, 1);
1010             e->dead = 1;
1011         }
1012     }
1013
1014     if (g->register_time_event) {
1015         avahi_time_event_free(g->register_time_event);
1016         g->register_time_event = NULL;
1017     }
1018
1019     g->dead = 1;
1020
1021     g->server->need_group_cleanup = 1;
1022     g->server->need_entry_cleanup = 1;
1023 }
1024
1025 static void entry_group_commit_real(AvahiSEntryGroup *g) {
1026     assert(g);
1027
1028     gettimeofday(&g->register_time, NULL);
1029
1030     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
1031
1032     if (!g->dead) {
1033         avahi_announce_group(g->server, g);
1034         avahi_s_entry_group_check_probed(g, 0);
1035     }
1036 }
1037
1038 static void entry_group_register_time_event_callback(AvahiTimeEvent *e, void* userdata) {
1039     AvahiSEntryGroup *g = userdata;
1040     assert(g);
1041
1042 /*     avahi_log_debug("Holdoff passed, waking up and going on."); */
1043
1044     avahi_time_event_free(g->register_time_event);
1045     g->register_time_event = NULL;
1046
1047     /* Holdoff time passed, so let's start probing */
1048     entry_group_commit_real(g);
1049 }
1050
1051 int avahi_s_entry_group_commit(AvahiSEntryGroup *g) {
1052     struct timeval now;
1053
1054     assert(g);
1055     assert(!g->dead);
1056
1057     if (g->state != AVAHI_ENTRY_GROUP_UNCOMMITED && g->state != AVAHI_ENTRY_GROUP_COLLISION)
1058         return avahi_server_set_errno(g->server, AVAHI_ERR_BAD_STATE);
1059
1060     g->n_register_try++;
1061
1062     avahi_timeval_add(&g->register_time,
1063                       1000*(g->n_register_try >= AVAHI_RR_RATE_LIMIT_COUNT ?
1064                             AVAHI_RR_HOLDOFF_MSEC_RATE_LIMIT :
1065                             AVAHI_RR_HOLDOFF_MSEC));
1066
1067     gettimeofday(&now, NULL);
1068
1069     if (avahi_timeval_compare(&g->register_time, &now) <= 0) {
1070         /* Holdoff time passed, so let's start probing */
1071 /*         avahi_log_debug("Holdoff passed, directly going on.");  */
1072
1073         entry_group_commit_real(g);
1074     } else {
1075 /*          avahi_log_debug("Holdoff not passed, sleeping.");  */
1076
1077          /* Holdoff time has not yet passed, so let's wait */
1078         assert(!g->register_time_event);
1079         g->register_time_event = avahi_time_event_new(g->server->time_event_queue, &g->register_time, entry_group_register_time_event_callback, g);
1080
1081         avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
1082     }
1083
1084     return AVAHI_OK;
1085 }
1086
1087 void avahi_s_entry_group_reset(AvahiSEntryGroup *g) {
1088     AvahiEntry *e;
1089     assert(g);
1090
1091     for (e = g->entries; e; e = e->by_group_next) {
1092         if (!e->dead) {
1093             avahi_goodbye_entry(g->server, e, 1, 1);
1094             e->dead = 1;
1095         }
1096     }
1097     g->server->need_entry_cleanup = 1;
1098
1099     if (g->register_time_event) {
1100         avahi_time_event_free(g->register_time_event);
1101         g->register_time_event = NULL;
1102     }
1103
1104     g->n_probing = 0;
1105
1106     gettimeofday(&g->register_time, NULL);
1107
1108     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_UNCOMMITED);
1109 }
1110
1111 int avahi_entry_is_commited(AvahiEntry *e) {
1112     assert(e);
1113     assert(!e->dead);
1114
1115     return !e->group ||
1116         e->group->state == AVAHI_ENTRY_GROUP_REGISTERING ||
1117         e->group->state == AVAHI_ENTRY_GROUP_ESTABLISHED;
1118 }
1119
1120 AvahiEntryGroupState avahi_s_entry_group_get_state(AvahiSEntryGroup *g) {
1121     assert(g);
1122     assert(!g->dead);
1123
1124     return g->state;
1125 }
1126
1127 void avahi_s_entry_group_set_data(AvahiSEntryGroup *g, void* userdata) {
1128     assert(g);
1129
1130     g->userdata = userdata;
1131 }
1132
1133 void* avahi_s_entry_group_get_data(AvahiSEntryGroup *g) {
1134     assert(g);
1135
1136     return g->userdata;
1137 }
1138
1139 int avahi_s_entry_group_is_empty(AvahiSEntryGroup *g) {
1140     AvahiEntry *e;
1141     assert(g);
1142
1143     /* Look for an entry that is not dead */
1144     for (e = g->entries; e; e = e->by_group_next)
1145         if (!e->dead)
1146             return 0;
1147
1148     return 1;
1149 }