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 "internal.h"
  42 #include "iface.h"
  43 #include "socket.h"
  44 #include "browse.h"
  45 #include "log.h"
  46 #include "util.h"
  47 #include "dns-srv-rr.h"
  48
  49 void avahi_entry_free(AvahiServer*s, AvahiEntry *e) {
  50     AvahiEntry *t;
  51
  52     assert(s);
  53     assert(e);
  54
  55     avahi_goodbye_entry(s, e, 1, 1);
  56
  57     /* Remove from linked list */
  58     AVAHI_LLIST_REMOVE(AvahiEntry, entries, s->entries, e);
  59
  60     /* Remove from hash table indexed by name */
  61     t = avahi_hashmap_lookup(s->entries_by_key, e->record->key);
  62     AVAHI_LLIST_REMOVE(AvahiEntry, by_key, t, e);
  63     if (t)
  64         avahi_hashmap_replace(s->entries_by_key, t->record->key, t);
  65     else
  66         avahi_hashmap_remove(s->entries_by_key, e->record->key);
  67
  68     /* Remove from associated group */
  69     if (e->group)
  70         AVAHI_LLIST_REMOVE(AvahiEntry, by_group, e->group->entries, e);
  71
  72     avahi_record_unref(e->record);
  73     avahi_free(e);
  74 }
  75
  76 void avahi_entry_group_free(AvahiServer *s, AvahiSEntryGroup *g) {
  77     assert(s);
  78     assert(g);
  79
  80     while (g->entries)
  81         avahi_entry_free(s, g->entries);
  82
  83     if (g->register_time_event)
  84         avahi_time_event_free(g->register_time_event);
  85
  86     AVAHI_LLIST_REMOVE(AvahiSEntryGroup, groups, s->groups, g);
  87     avahi_free(g);
  88 }
  89
  90 void avahi_cleanup_dead_entries(AvahiServer *s) {
  91     assert(s);
  92
  93     if (s->need_group_cleanup) {
  94         AvahiSEntryGroup *g, *next;
  95
  96         for (g = s->groups; g; g = next) {
  97             next = g->groups_next;
  98
  99             if (g->dead)
 100                 avahi_entry_group_free(s, g);
 101         }
 102
 103         s->need_group_cleanup = 0;
 104     }
 105
 106     if (s->need_entry_cleanup) {
 107         AvahiEntry *e, *next;
 108
 109         for (e = s->entries; e; e = next) {
 110             next = e->entries_next;
 111
 112             if (e->dead)
 113                 avahi_entry_free(s, e);
 114         }
 115
 116         s->need_entry_cleanup = 0;
 117     }
 118
 119     if (s->need_browser_cleanup)
 120         avahi_browser_cleanup(s);
 121 }
 122
 123 static int check_record_conflict(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, AvahiRecord *r, AvahiPublishFlags flags) {
 124     AvahiEntry *e;
 125
 126     assert(s);
 127     assert(r);
 128
 129     for (e = avahi_hashmap_lookup(s->entries_by_key, r->key); e; e = e->by_key_next) {
 130         if (e->dead)
 131             continue;
 132
 133         if (!(flags & AVAHI_PUBLISH_UNIQUE) && !(e->flags & AVAHI_PUBLISH_UNIQUE))
 134             continue;
 135
 136         if ((flags & AVAHI_PUBLISH_ALLOW_MULTIPLE) && (e->flags & AVAHI_PUBLISH_ALLOW_MULTIPLE) )
 137             continue;
 138
 139         if ((interface <= 0 ||
 140              e->interface <= 0 ||
 141              e->interface == interface) &&
 142             (protocol == AVAHI_PROTO_UNSPEC ||
 143              e->protocol == AVAHI_PROTO_UNSPEC ||
 144              e->protocol == protocol))
 145
 146             return -1;
 147     }
 148
 149     return 0;
 150 }
 151
 152 int avahi_server_add(
 153     AvahiServer *s,
 154     AvahiSEntryGroup *g,
 155     AvahiIfIndex interface,
 156     AvahiProtocol protocol,
 157     AvahiPublishFlags flags,
 158     AvahiRecord *r) {
 159
 160     assert(s);
 161     assert(r);
 162
 163     if (!AVAHI_IF_VALID(interface))
 164         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
 165
 166     if (!AVAHI_PROTO_VALID(protocol))
 167         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
 168
 169     if (!AVAHI_FLAGS_VALID(
 170             flags,
 171             AVAHI_PUBLISH_NO_ANNOUNCE|
 172             AVAHI_PUBLISH_NO_PROBE|
 173             AVAHI_PUBLISH_UNIQUE|
 174             AVAHI_PUBLISH_ALLOW_MULTIPLE|
 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(&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     }
 440
 441 fail:
 442
 443     avahi_free(n);
 444
 445     return ret;
 446 }
 447
 448 static int server_add_txt_strlst_nocopy(
 449     AvahiServer *s,
 450     AvahiSEntryGroup *g,
 451     AvahiIfIndex interface,
 452     AvahiProtocol protocol,
 453     AvahiPublishFlags flags,
 454     uint32_t ttl,
 455     const char *name,
 456     AvahiStringList *strlst) {
 457
 458     AvahiRecord *r;
 459     int ret;
 460
 461     assert(s);
 462
 463     if (!(r = avahi_record_new_full(name ? name : s->host_name_fqdn, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_TXT, ttl))) {
 464         avahi_string_list_free(strlst);
 465         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 466     }
 467
 468     r->data.txt.string_list = strlst;
 469     ret = avahi_server_add(s, g, interface, protocol, flags, r);
 470     avahi_record_unref(r);
 471
 472     return ret;
 473 }
 474
 475 static AvahiStringList *add_magic_cookie(
 476     AvahiServer *s,
 477     AvahiStringList *strlst) {
 478
 479     assert(s);
 480
 481     if (!s->config.add_service_cookie)
 482         return strlst;
 483
 484     if (avahi_string_list_find(strlst, AVAHI_SERVICE_COOKIE))
 485         /* This string list already contains a magic cookie */
 486         return strlst;
 487
 488     return avahi_string_list_add_printf(strlst, AVAHI_SERVICE_COOKIE"=%u", s->local_service_cookie);
 489 }
 490
 491 static int server_add_service_strlst_nocopy(
 492     AvahiServer *s,
 493     AvahiSEntryGroup *g,
 494     AvahiIfIndex interface,
 495     AvahiProtocol protocol,
 496     AvahiPublishFlags flags,
 497     const char *name,
 498     const char *type,
 499     const char *domain,
 500     const char *host,
 501     uint16_t port,
 502     AvahiStringList *strlst) {
 503
 504     char ptr_name[AVAHI_DOMAIN_NAME_MAX], svc_name[AVAHI_DOMAIN_NAME_MAX], enum_ptr[AVAHI_DOMAIN_NAME_MAX], *h = NULL;
 505     AvahiRecord *r = NULL;
 506     int ret = AVAHI_OK;
 507
 508     assert(s);
 509     assert(type);
 510     assert(name);
 511
 512     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
 513     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
 514     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_COOKIE|AVAHI_PUBLISH_UPDATE), AVAHI_ERR_INVALID_FLAGS);
 515     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
 516     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
 517     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
 518     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !host || avahi_is_valid_domain_name(host), AVAHI_ERR_INVALID_HOST_NAME);
 519
 520     if (!domain)
 521         domain = s->domain_name;
 522
 523     if (!host)
 524         host = s->host_name_fqdn;
 525
 526     if (!(h = avahi_normalize_name_strdup(host))) {
 527         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 528         goto fail;
 529     }
 530
 531     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0 ||
 532         (ret = avahi_service_name_join(ptr_name, sizeof(ptr_name), NULL, type, domain)) < 0 ||
 533         (ret = avahi_service_name_join(enum_ptr, sizeof(enum_ptr), NULL, "_services._dns-sd._udp", domain)) < 0) {
 534         avahi_server_set_errno(s, ret);
 535         goto fail;
 536     }
 537
 538     /* Add service enumeration PTR record */
 539
 540     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, 0, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
 541         goto fail;
 542
 543     /* Add SRV record */
 544
 545     if (!(r = avahi_record_new_full(svc_name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
 546         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 547         goto fail;
 548     }
 549
 550     r->data.srv.priority = 0;
 551     r->data.srv.weight = 0;
 552     r->data.srv.port = port;
 553     r->data.srv.name = h;
 554     h = NULL;
 555     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_UNIQUE, r);
 556     avahi_record_unref(r);
 557
 558     if (ret < 0)
 559         goto fail;
 560
 561     /* Add TXT record */
 562
 563     if (!(flags & AVAHI_PUBLISH_NO_COOKIE))
 564         strlst = add_magic_cookie(s, strlst);
 565
 566     ret = server_add_txt_strlst_nocopy(s, g, interface, protocol, AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL, svc_name, strlst);
 567     strlst = NULL;
 568
 569     if (ret < 0)
 570         goto fail;
 571
 572     /* Add service type enumeration record */
 573
 574     ret = avahi_server_add_ptr(s, g, interface, protocol, 0, AVAHI_DEFAULT_TTL, enum_ptr, ptr_name);
 575
 576 fail:
 577
 578     avahi_string_list_free(strlst);
 579     avahi_free(h);
 580
 581     return ret;
 582 }
 583
 584 int avahi_server_add_service_strlst(
 585     AvahiServer *s,
 586     AvahiSEntryGroup *g,
 587     AvahiIfIndex interface,
 588     AvahiProtocol protocol,
 589     AvahiPublishFlags flags,
 590     const char *name,
 591     const char *type,
 592     const char *domain,
 593     const char *host,
 594     uint16_t port,
 595     AvahiStringList *strlst) {
 596
 597     assert(s);
 598     assert(type);
 599     assert(name);
 600
 601     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_copy(strlst));
 602 }
 603
 604 int avahi_server_add_service(
 605     AvahiServer *s,
 606     AvahiSEntryGroup *g,
 607     AvahiIfIndex interface,
 608     AvahiProtocol protocol,
 609     AvahiPublishFlags flags,
 610     const char *name,
 611     const char *type,
 612     const char *domain,
 613     const char *host,
 614     uint16_t port,
 615     ... ){
 616
 617     va_list va;
 618     int ret;
 619
 620     va_start(va, port);
 621     ret = server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_new_va(va));
 622     va_end(va);
 623
 624     return ret;
 625 }
 626
 627 static int server_update_service_txt_strlst_nocopy(
 628     AvahiServer *s,
 629     AvahiSEntryGroup *g,
 630     AvahiIfIndex interface,
 631     AvahiProtocol protocol,
 632     AvahiPublishFlags flags,
 633     const char *name,
 634     const char *type,
 635     const char *domain,
 636     AvahiStringList *strlst) {
 637
 638     char svc_name[AVAHI_DOMAIN_NAME_MAX];
 639     int ret = AVAHI_OK;
 640
 641     assert(s);
 642     assert(type);
 643     assert(name);
 644
 645     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
 646     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
 647     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_COOKIE), AVAHI_ERR_INVALID_FLAGS);
 648     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
 649     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
 650     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
 651
 652     if (!domain)
 653         domain = s->domain_name;
 654
 655     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0) {
 656         avahi_server_set_errno(s, ret);
 657         goto fail;
 658     }
 659
 660     /* Add TXT record */
 661     if (!(flags & AVAHI_PUBLISH_NO_COOKIE))
 662         strlst = add_magic_cookie(s, strlst);
 663
 664     ret = server_add_txt_strlst_nocopy(s, g, interface, protocol, AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_UPDATE, AVAHI_DEFAULT_TTL, svc_name, strlst);
 665     strlst = NULL;
 666
 667 fail:
 668
 669     avahi_string_list_free(strlst);
 670
 671     return ret;
 672 }
 673
 674 int avahi_server_update_service_txt_strlst(
 675     AvahiServer *s,
 676     AvahiSEntryGroup *g,
 677     AvahiIfIndex interface,
 678     AvahiProtocol protocol,
 679     AvahiPublishFlags flags,
 680     const char *name,
 681     const char *type,
 682     const char *domain,
 683     AvahiStringList *strlst) {
 684
 685     return server_update_service_txt_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, avahi_string_list_copy(strlst));
 686 }
 687
 688 /** Update the TXT record for a service with the NULL termonate list of strings */
 689 int avahi_server_update_service_txt(
 690     AvahiServer *s,
 691     AvahiSEntryGroup *g,
 692     AvahiIfIndex interface,
 693     AvahiProtocol protocol,
 694     AvahiPublishFlags flags,
 695     const char *name,
 696     const char *type,
 697     const char *domain,
 698     ...) {
 699
 700     va_list va;
 701     int ret;
 702
 703     va_start(va, domain);
 704     ret = server_update_service_txt_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, avahi_string_list_new_va(va));
 705     va_end(va);
 706
 707     return ret;
 708 }
 709
 710 int avahi_server_add_service_subtype(
 711     AvahiServer *s,
 712     AvahiSEntryGroup *g,
 713     AvahiIfIndex interface,
 714     AvahiProtocol protocol,
 715     AvahiPublishFlags flags,
 716     const char *name,
 717     const char *type,
 718     const char *domain,
 719     const char *subtype) {
 720
 721     int ret = AVAHI_OK;
 722     char svc_name[AVAHI_DOMAIN_NAME_MAX], ptr_name[AVAHI_DOMAIN_NAME_MAX];
 723
 724     assert(name);
 725     assert(type);
 726     assert(subtype);
 727
 728     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
 729     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
 730     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, 0), AVAHI_ERR_INVALID_FLAGS);
 731     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
 732     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
 733     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
 734     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_subtype(subtype), AVAHI_ERR_INVALID_SERVICE_SUBTYPE);
 735
 736     if (!domain)
 737         domain = s->domain_name;
 738
 739     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0 ||
 740         (ret = avahi_service_name_join(ptr_name, sizeof(ptr_name), NULL, subtype, domain)) < 0) {
 741         avahi_server_set_errno(s, ret);
 742         goto fail;
 743     }
 744
 745     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, 0, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
 746         goto fail;
 747
 748 fail:
 749
 750     return ret;
 751 }
 752
 753 static void hexstring(char *s, size_t sl, const void *p, size_t pl) {
 754     static const char hex[] = "0123456789abcdef";
 755     int b = 0;
 756     const uint8_t *k = p;
 757
 758     while (sl > 1 && pl > 0) {
 759         *(s++) = hex[(b ? *k : *k >> 4) & 0xF];
 760
 761         if (b) {
 762             k++;
 763             pl--;
 764         }
 765
 766         b = !b;
 767
 768         sl--;
 769     }
 770
 771     if (sl > 0)
 772         *s = 0;
 773 }
 774
 775 int avahi_server_add_dns_server_address(
 776     AvahiServer *s,
 777     AvahiSEntryGroup *g,
 778     AvahiIfIndex interface,
 779     AvahiProtocol protocol,
 780     AvahiPublishFlags flags,
 781     const char *domain,
 782     AvahiDNSServerType type,
 783     const AvahiAddress *address,
 784     uint16_t port /** should be 53 */) {
 785
 786     AvahiRecord *r;
 787     int ret;
 788     char n[64], h[64];
 789
 790     assert(s);
 791     assert(address);
 792
 793     if (!AVAHI_IF_VALID(interface))
 794         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
 795
 796     if (!AVAHI_PROTO_VALID(protocol) || !AVAHI_PROTO_VALID(address->proto))
 797         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
 798
 799     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
 800         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
 801
 802     if (port == 0)
 803         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
 804
 805     if (domain && !avahi_is_valid_domain_name(domain))
 806         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
 807
 808     if (!domain)
 809         domain = s->domain_name;
 810
 811     if (address->proto == AVAHI_PROTO_INET) {
 812         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv4Address));
 813         snprintf(n, sizeof(n), "ip-%s.%s", h, domain);
 814         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_A, AVAHI_DEFAULT_TTL_HOST_NAME);
 815         r->data.a.address = address->data.ipv4;
 816     } else {
 817         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv6Address));
 818         snprintf(n, sizeof(n), "ip6-%s.%s", h, domain);
 819         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_AAAA, AVAHI_DEFAULT_TTL_HOST_NAME);
 820         r->data.aaaa.address = address->data.ipv6;
 821     }
 822
 823     if (!r)
 824         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 825
 826     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
 827     avahi_record_unref(r);
 828
 829     if (ret < 0)
 830         return ret;
 831
 832     return avahi_server_add_dns_server_name(s, g, interface, protocol, flags, domain, type, n, port);
 833 }
 834
 835 int avahi_server_add_dns_server_name(
 836     AvahiServer *s,
 837     AvahiSEntryGroup *g,
 838     AvahiIfIndex interface,
 839     AvahiProtocol protocol,
 840     AvahiPublishFlags flags,
 841     const char *domain,
 842     AvahiDNSServerType type,
 843     const char *name,
 844     uint16_t port /** should be 53 */) {
 845
 846     int ret = -1;
 847     char t[256], *d = NULL, *n = NULL;
 848     AvahiRecord *r;
 849
 850     assert(s);
 851     assert(name);
 852
 853     if (!AVAHI_IF_VALID(interface))
 854         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
 855
 856     if (!AVAHI_PROTO_VALID(protocol))
 857         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
 858
 859     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
 860         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
 861
 862     if (port == 0)
 863         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
 864
 865     if (!avahi_is_valid_domain_name(name))
 866         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
 867
 868     if (domain && !avahi_is_valid_domain_name(domain))
 869         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
 870
 871     if (!domain)
 872         domain = s->domain_name;
 873
 874     if (!(n = avahi_normalize_name_strdup(name)) ||
 875         !(d = avahi_normalize_name_strdup(domain))) {
 876         avahi_free(n);
 877         avahi_free(d);
 878         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 879     }
 880
 881     snprintf(t, sizeof(t), "%s.%s", type == AVAHI_DNS_SERVER_RESOLVE ? "_domain._udp" : "_dns-update._udp", d);
 882     avahi_free(d);
 883
 884     if (!(r = avahi_record_new_full(t, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
 885         avahi_free(n);
 886         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 887     }
 888
 889     r->data.srv.priority = 0;
 890     r->data.srv.weight = 0;
 891     r->data.srv.port = port;
 892     r->data.srv.name = n;
 893     ret = avahi_server_add(s, g, interface, protocol, 0, r);
 894     avahi_record_unref(r);
 895
 896     return ret;
 897 }
 898
 899 void avahi_s_entry_group_change_state(AvahiSEntryGroup *g, AvahiEntryGroupState state) {
 900     assert(g);
 901
 902     if (g->state == state)
 903         return;
 904
 905     assert(state <= AVAHI_ENTRY_GROUP_COLLISION);
 906
 907     if (g->state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
 908
 909         /* If the entry group was established for a time longer then
 910          * 5s, reset the establishment trial counter */
 911
 912         if (avahi_age(&g->established_at) > 5000000)
 913             g->n_register_try = 0;
 914     }
 915
 916     if (state == AVAHI_ENTRY_GROUP_ESTABLISHED)
 917
 918         /* If the entry group is now established, remember the time
 919          * this happened */
 920
 921         gettimeofday(&g->established_at, NULL);
 922
 923     g->state = state;
 924
 925     if (g->callback)
 926         g->callback(g->server, g, state, g->userdata);
 927 }
 928
 929 AvahiSEntryGroup *avahi_s_entry_group_new(AvahiServer *s, AvahiSEntryGroupCallback callback, void* userdata) {
 930     AvahiSEntryGroup *g;
 931
 932     assert(s);
 933
 934     if (!(g = avahi_new(AvahiSEntryGroup, 1))) {
 935         avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
 936         return NULL;
 937     }
 938
 939     g->server = s;
 940     g->callback = callback;
 941     g->userdata = userdata;
 942     g->dead = 0;
 943     g->state = AVAHI_ENTRY_GROUP_UNCOMMITED;
 944     g->n_probing = 0;
 945     g->n_register_try = 0;
 946     g->register_time_event = NULL;
 947     g->register_time.tv_sec = 0;
 948     g->register_time.tv_usec = 0;
 949     AVAHI_LLIST_HEAD_INIT(AvahiEntry, g->entries);
 950
 951     AVAHI_LLIST_PREPEND(AvahiSEntryGroup, groups, s->groups, g);
 952     return g;
 953 }
 954
 955 void avahi_s_entry_group_free(AvahiSEntryGroup *g) {
 956     AvahiEntry *e;
 957
 958     assert(g);
 959     assert(g->server);
 960
 961     for (e = g->entries; e; e = e->by_group_next) {
 962         if (!e->dead) {
 963             avahi_goodbye_entry(g->server, e, 1, 1);
 964             e->dead = 1;
 965         }
 966     }
 967
 968     if (g->register_time_event) {
 969         avahi_time_event_free(g->register_time_event);
 970         g->register_time_event = NULL;
 971     }
 972
 973     g->dead = 1;
 974
 975     g->server->need_group_cleanup = 1;
 976     g->server->need_entry_cleanup = 1;
 977 }
 978
 979 static void entry_group_commit_real(AvahiSEntryGroup *g) {
 980     assert(g);
 981
 982     gettimeofday(&g->register_time, NULL);
 983
 984     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
 985
 986     if (!g->dead) {
 987         avahi_announce_group(g->server, g);
 988         avahi_s_entry_group_check_probed(g, 0);
 989     }
 990 }
 991
 992 static void entry_group_register_time_event_callback(AvahiTimeEvent *e, void* userdata) {
 993     AvahiSEntryGroup *g = userdata;
 994     assert(g);
 995
 996 /*     avahi_log_debug("Holdoff passed, waking up and going on."); */
 997
 998     avahi_time_event_free(g->register_time_event);
 999     g->register_time_event = NULL;
1000
1001     /* Holdoff time passed, so let's start probing */
1002     entry_group_commit_real(g);
1003 }
1004
1005 int avahi_s_entry_group_commit(AvahiSEntryGroup *g) {
1006     struct timeval now;
1007
1008     assert(g);
1009     assert(!g->dead);
1010
1011     if (g->state != AVAHI_ENTRY_GROUP_UNCOMMITED && g->state != AVAHI_ENTRY_GROUP_COLLISION)
1012         return avahi_server_set_errno(g->server, AVAHI_ERR_BAD_STATE);
1013
1014     g->n_register_try++;
1015
1016     avahi_timeval_add(&g->register_time,
1017                       1000*(g->n_register_try >= AVAHI_RR_RATE_LIMIT_COUNT ?
1018                             AVAHI_RR_HOLDOFF_MSEC_RATE_LIMIT :
1019                             AVAHI_RR_HOLDOFF_MSEC));
1020
1021     gettimeofday(&now, NULL);
1022
1023     if (avahi_timeval_compare(&g->register_time, &now) <= 0) {
1024         /* Holdoff time passed, so let's start probing */
1025 /*         avahi_log_debug("Holdoff passed, directly going on.");  */
1026
1027         entry_group_commit_real(g);
1028     } else {
1029 /*          avahi_log_debug("Holdoff not passed, sleeping.");  */
1030
1031          /* Holdoff time has not yet passed, so let's wait */
1032         assert(!g->register_time_event);
1033         g->register_time_event = avahi_time_event_new(g->server->time_event_queue, &g->register_time, entry_group_register_time_event_callback, g);
1034
1035         avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
1036     }
1037
1038     return AVAHI_OK;
1039 }
1040
1041 void avahi_s_entry_group_reset(AvahiSEntryGroup *g) {
1042     AvahiEntry *e;
1043     assert(g);
1044
1045     for (e = g->entries; e; e = e->by_group_next) {
1046         if (!e->dead) {
1047             avahi_goodbye_entry(g->server, e, 1, 1);
1048             e->dead = 1;
1049         }
1050     }
1051     g->server->need_entry_cleanup = 1;
1052
1053     if (g->register_time_event) {
1054         avahi_time_event_free(g->register_time_event);
1055         g->register_time_event = NULL;
1056     }
1057
1058     g->n_probing = 0;
1059
1060     gettimeofday(&g->register_time, NULL);
1061
1062     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_UNCOMMITED);
1063 }
1064
1065 int avahi_entry_is_commited(AvahiEntry *e) {
1066     assert(e);
1067     assert(!e->dead);
1068
1069     return !e->group ||
1070         e->group->state == AVAHI_ENTRY_GROUP_REGISTERING ||
1071         e->group->state == AVAHI_ENTRY_GROUP_ESTABLISHED;
1072 }
1073
1074 AvahiEntryGroupState avahi_s_entry_group_get_state(AvahiSEntryGroup *g) {
1075     assert(g);
1076     assert(!g->dead);
1077
1078     return g->state;
1079 }
1080
1081 void avahi_s_entry_group_set_data(AvahiSEntryGroup *g, void* userdata) {
1082     assert(g);
1083
1084     g->userdata = userdata;
1085 }
1086
1087 void* avahi_s_entry_group_get_data(AvahiSEntryGroup *g) {
1088     assert(g);
1089
1090     return g->userdata;
1091 }
1092
1093 int avahi_s_entry_group_is_empty(AvahiSEntryGroup *g) {
1094     AvahiEntry *e;
1095     assert(g);
1096
1097     /* Look for an entry that is not dead */
1098     for (e = g->entries; e; e = e->by_group_next)
1099         if (!e->dead)
1100             return 0;
1101
1102     return 1;
1103 }