git.meshlink.io Git - catta/blob - avahi-core/server.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 #define AVAHI_RR_HOLDOFF_MSEC 1000
  49 #define AVAHI_RR_HOLDOFF_MSEC_RATE_LIMIT 60000
  50 #define AVAHI_RR_RATE_LIMIT_COUNT 15
  51
  52 static void free_entry(AvahiServer*s, AvahiEntry *e) {
  53     AvahiEntry *t;
  54
  55     assert(s);
  56     assert(e);
  57
  58     avahi_goodbye_entry(s, e, 1);
  59
  60     /* Remove from linked list */
  61     AVAHI_LLIST_REMOVE(AvahiEntry, entries, s->entries, e);
  62
  63     /* Remove from hash table indexed by name */
  64     t = avahi_hashmap_lookup(s->entries_by_key, e->record->key);
  65     AVAHI_LLIST_REMOVE(AvahiEntry, by_key, t, e);
  66     if (t)
  67         avahi_hashmap_replace(s->entries_by_key, t->record->key, t);
  68     else
  69         avahi_hashmap_remove(s->entries_by_key, e->record->key);
  70
  71     /* Remove from associated group */
  72     if (e->group)
  73         AVAHI_LLIST_REMOVE(AvahiEntry, by_group, e->group->entries, e);
  74
  75     avahi_record_unref(e->record);
  76     avahi_free(e);
  77 }
  78
  79 static void free_group(AvahiServer *s, AvahiSEntryGroup *g) {
  80     assert(s);
  81     assert(g);
  82
  83     while (g->entries)
  84         free_entry(s, g->entries);
  85
  86     if (g->register_time_event)
  87         avahi_time_event_free(g->register_time_event);
  88
  89     AVAHI_LLIST_REMOVE(AvahiSEntryGroup, groups, s->groups, g);
  90     avahi_free(g);
  91 }
  92
  93 static void cleanup_dead(AvahiServer *s) {
  94     assert(s);
  95
  96     if (s->need_group_cleanup) {
  97         AvahiSEntryGroup *g, *next;
  98
  99         for (g = s->groups; g; g = next) {
 100             next = g->groups_next;
 101
 102             if (g->dead)
 103                 free_group(s, g);
 104         }
 105
 106         s->need_group_cleanup = 0;
 107     }
 108
 109     if (s->need_entry_cleanup) {
 110         AvahiEntry *e, *next;
 111
 112         for (e = s->entries; e; e = next) {
 113             next = e->entries_next;
 114
 115             if (e->dead)
 116                 free_entry(s, e);
 117         }
 118
 119         s->need_entry_cleanup = 0;
 120     }
 121
 122     if (s->need_browser_cleanup)
 123         avahi_browser_cleanup(s);
 124 }
 125
 126 static void enum_aux_records(AvahiServer *s, AvahiInterface *i, const char *name, uint16_t type, void (*callback)(AvahiServer *s, AvahiRecord *r, int flush_cache, void* userdata), void* userdata) {
 127     AvahiKey *k;
 128     AvahiEntry *e;
 129
 130     assert(s);
 131     assert(i);
 132     assert(name);
 133     assert(callback);
 134
 135     assert(type != AVAHI_DNS_TYPE_ANY);
 136
 137     if (!(k = avahi_key_new(name, AVAHI_DNS_CLASS_IN, type)))
 138         return; /** OOM */
 139
 140     for (e = avahi_hashmap_lookup(s->entries_by_key, k); e; e = e->by_key_next)
 141         if (!e->dead && avahi_entry_is_registered(s, e, i))
 142             callback(s, e->record, e->flags & AVAHI_PUBLISH_UNIQUE, userdata);
 143
 144     avahi_key_unref(k);
 145 }
 146
 147 void avahi_server_enumerate_aux_records(AvahiServer *s, AvahiInterface *i, AvahiRecord *r, void (*callback)(AvahiServer *s, AvahiRecord *r, int flush_cache, void* userdata), void* userdata) {
 148     assert(s);
 149     assert(i);
 150     assert(r);
 151     assert(callback);
 152
 153     if (r->key->clazz == AVAHI_DNS_CLASS_IN) {
 154         if (r->key->type == AVAHI_DNS_TYPE_PTR) {
 155             enum_aux_records(s, i, r->data.ptr.name, AVAHI_DNS_TYPE_SRV, callback, userdata);
 156             enum_aux_records(s, i, r->data.ptr.name, AVAHI_DNS_TYPE_TXT, callback, userdata);
 157         } else if (r->key->type == AVAHI_DNS_TYPE_SRV) {
 158             enum_aux_records(s, i, r->data.srv.name, AVAHI_DNS_TYPE_A, callback, userdata);
 159             enum_aux_records(s, i, r->data.srv.name, AVAHI_DNS_TYPE_AAAA, callback, userdata);
 160         }
 161     }
 162 }
 163
 164 void avahi_server_prepare_response(AvahiServer *s, AvahiInterface *i, AvahiEntry *e, int unicast_response, int auxiliary) {
 165     assert(s);
 166     assert(i);
 167     assert(e);
 168
 169     avahi_record_list_push(s->record_list, e->record, e->flags & AVAHI_PUBLISH_UNIQUE, unicast_response, auxiliary);
 170 }
 171
 172 void avahi_server_prepare_matching_responses(AvahiServer *s, AvahiInterface *i, AvahiKey *k, int unicast_response) {
 173     AvahiEntry *e;
 174 /*     char *txt; */
 175
 176     assert(s);
 177     assert(i);
 178     assert(k);
 179
 180 /*     avahi_log_debug("Posting responses matching [%s]", txt = avahi_key_to_string(k)); */
 181 /*     avahi_free(txt); */
 182
 183     if (avahi_key_is_pattern(k)) {
 184
 185         /* Handle ANY query */
 186
 187         for (e = s->entries; e; e = e->entries_next)
 188             if (!e->dead && avahi_key_pattern_match(k, e->record->key) && avahi_entry_is_registered(s, e, i))
 189                 avahi_server_prepare_response(s, i, e, unicast_response, 0);
 190
 191     } else {
 192
 193         /* Handle all other queries */
 194
 195         for (e = avahi_hashmap_lookup(s->entries_by_key, k); e; e = e->by_key_next)
 196             if (!e->dead && avahi_entry_is_registered(s, e, i))
 197                 avahi_server_prepare_response(s, i, e, unicast_response, 0);
 198     }
 199 }
 200
 201 static void withdraw_entry(AvahiServer *s, AvahiEntry *e) {
 202     assert(s);
 203     assert(e);
 204
 205     if (e->group) {
 206         AvahiEntry *k;
 207
 208         for (k = e->group->entries; k; k = k->by_group_next) {
 209             if (!k->dead) {
 210                 avahi_goodbye_entry(s, k, 0);
 211                 k->dead = 1;
 212             }
 213         }
 214
 215         e->group->n_probing = 0;
 216
 217         avahi_s_entry_group_change_state(e->group, AVAHI_ENTRY_GROUP_COLLISION);
 218     } else {
 219         avahi_goodbye_entry(s, e, 0);
 220         e->dead = 1;
 221     }
 222
 223     s->need_entry_cleanup = 1;
 224 }
 225
 226 static void withdraw_rrset(AvahiServer *s, AvahiKey *key) {
 227     AvahiEntry *e;
 228
 229     assert(s);
 230     assert(key);
 231
 232    for (e = avahi_hashmap_lookup(s->entries_by_key, key); e; e = e->by_key_next)
 233        if (!e->dead)
 234            withdraw_entry(s, e);
 235 }
 236
 237 static void incoming_probe(AvahiServer *s, AvahiRecord *record, AvahiInterface *i) {
 238     AvahiEntry *e, *n;
 239     char *t;
 240     int ours = 0, won = 0, lost = 0;
 241
 242     assert(s);
 243     assert(record);
 244     assert(i);
 245
 246     t = avahi_record_to_string(record);
 247
 248 /*     avahi_log_debug("incoming_probe()");  */
 249
 250     for (e = avahi_hashmap_lookup(s->entries_by_key, record->key); e; e = n) {
 251         int cmp;
 252         n = e->by_key_next;
 253
 254         if (e->dead)
 255             continue;
 256
 257         if ((cmp = avahi_record_lexicographical_compare(e->record, record)) == 0) {
 258             ours = 1;
 259             break;
 260         } else {
 261
 262             if (avahi_entry_is_probing(s, e, i)) {
 263                 if (cmp > 0)
 264                     won = 1;
 265                 else /* cmp < 0 */
 266                     lost = 1;
 267             }
 268         }
 269     }
 270
 271     if (!ours) {
 272
 273         if (won)
 274             avahi_log_debug("Recieved conflicting probe [%s]. Local host won.", t);
 275         else if (lost) {
 276             avahi_log_debug("Recieved conflicting probe [%s]. Local host lost. Withdrawing.", t);
 277             withdraw_rrset(s, record->key);
 278         }/*  else */
 279 /*             avahi_log_debug("Not conflicting probe"); */
 280     }
 281
 282     avahi_free(t);
 283 }
 284
 285 static int handle_conflict(AvahiServer *s, AvahiInterface *i, AvahiRecord *record, int unique, const AvahiAddress *a) {
 286     int valid = 1, ours = 0, conflict = 0, withdraw_immediately = 0;
 287     AvahiEntry *e, *n, *conflicting_entry = NULL;
 288
 289     assert(s);
 290     assert(i);
 291     assert(record);
 292
 293
 294 /*     avahi_log_debug("CHECKING FOR CONFLICT: [%s]", t);   */
 295
 296     for (e = avahi_hashmap_lookup(s->entries_by_key, record->key); e; e = n) {
 297         n = e->by_key_next;
 298
 299         if (e->dead)
 300             continue;
 301
 302         /* Check if the incoming is a goodbye record */
 303         if (avahi_record_is_goodbye(record)) {
 304
 305             if (avahi_record_equal_no_ttl(e->record, record)) {
 306                 char *t;
 307
 308                 /* Refresh */
 309                 t = avahi_record_to_string(record);
 310                 avahi_log_debug("Recieved goodbye record for one of our records [%s]. Refreshing.", t);
 311                 avahi_server_prepare_matching_responses(s, i, e->record->key, 0);
 312
 313                 valid = 0;
 314                 avahi_free(t);
 315                 break;
 316             }
 317
 318             /* If the goodybe packet doesn't match one of our own RRs, we simply ignore it. */
 319             continue;
 320         }
 321
 322         if (!(e->flags & AVAHI_PUBLISH_UNIQUE) && !unique)
 323             continue;
 324
 325         /* Either our entry or the other is intended to be unique, so let's check */
 326
 327         if (avahi_record_equal_no_ttl(e->record, record)) {
 328             ours = 1; /* We have an identical record, so this is no conflict */
 329
 330             /* Check wheter there is a TTL conflict */
 331             if (record->ttl <= e->record->ttl/2 &&
 332                 avahi_entry_is_registered(s, e, i)) {
 333                 char *t;
 334                 /* Refresh */
 335                 t = avahi_record_to_string(record);
 336
 337                 avahi_log_debug("Recieved record with bad TTL [%s]. Refreshing.", t);
 338                 avahi_server_prepare_matching_responses(s, i, e->record->key, 0);
 339                 valid = 0;
 340
 341                 avahi_free(t);
 342             }
 343
 344             /* There's no need to check the other entries of this RRset */
 345             break;
 346
 347         } else {
 348
 349             if (avahi_entry_is_registered(s, e, i)) {
 350
 351                 /* A conflict => we have to return to probe mode */
 352                 conflict = 1;
 353                 conflicting_entry = e;
 354
 355             } else if (avahi_entry_is_probing(s, e, i)) {
 356
 357                 /* We are currently registering a matching record, but
 358                  * someone else already claimed it, so let's
 359                  * withdraw */
 360                 conflict = 1;
 361                 withdraw_immediately = 1;
 362             }
 363         }
 364     }
 365
 366 /*     avahi_log_debug("ours=%i conflict=%i", ours, conflict); */
 367
 368     if (!ours && conflict) {
 369         char *t;
 370
 371         valid = 0;
 372
 373         t = avahi_record_to_string(record);
 374
 375         if (withdraw_immediately) {
 376             avahi_log_debug("Recieved conflicting record [%s] with local record to be. Withdrawing.", t);
 377             withdraw_rrset(s, record->key);
 378         } else {
 379             assert(conflicting_entry);
 380             avahi_log_debug("Recieved conflicting record [%s]. Resetting our record.", t);
 381             avahi_entry_return_to_initial_state(s, conflicting_entry, i);
 382
 383             /* Local unique records are returned to probing
 384              * state. Local shared records are reannounced. */
 385         }
 386
 387         avahi_free(t);
 388     }
 389
 390     return valid;
 391 }
 392
 393 static void append_aux_callback(AvahiServer *s, AvahiRecord *r, int flush_cache, void* userdata) {
 394     int *unicast_response = userdata;
 395
 396     assert(s);
 397     assert(r);
 398     assert(unicast_response);
 399
 400     avahi_record_list_push(s->record_list, r, flush_cache, *unicast_response, 1);
 401 }
 402
 403 static void append_aux_records_to_list(AvahiServer *s, AvahiInterface *i, AvahiRecord *r, int unicast_response) {
 404     assert(s);
 405     assert(r);
 406
 407     avahi_server_enumerate_aux_records(s, i, r, append_aux_callback, &unicast_response);
 408 }
 409
 410 void avahi_server_generate_response(AvahiServer *s, AvahiInterface *i, AvahiDnsPacket *p, const AvahiAddress *a, uint16_t port, int legacy_unicast, int immediately) {
 411
 412     assert(s);
 413     assert(i);
 414     assert(!legacy_unicast || (a && port > 0 && p));
 415
 416     if (legacy_unicast) {
 417         AvahiDnsPacket *reply;
 418         AvahiRecord *r;
 419
 420         if (!(reply = avahi_dns_packet_new_reply(p, 512 /* unicast DNS maximum packet size is 512 */ , 1, 1)))
 421             return; /* OOM */
 422
 423         while ((r = avahi_record_list_next(s->record_list, NULL, NULL, NULL))) {
 424
 425             append_aux_records_to_list(s, i, r, 0);
 426
 427             if (avahi_dns_packet_append_record(reply, r, 0, 10))
 428                 avahi_dns_packet_inc_field(reply, AVAHI_DNS_FIELD_ANCOUNT);
 429             else {
 430                 char *t = avahi_record_to_string(r);
 431                 avahi_log_warn("Record [%s] not fitting in legacy unicast packet, dropping.", t);
 432                 avahi_free(t);
 433             }
 434
 435             avahi_record_unref(r);
 436         }
 437
 438         if (avahi_dns_packet_get_field(reply, AVAHI_DNS_FIELD_ANCOUNT) != 0)
 439             avahi_interface_send_packet_unicast(i, reply, a, port);
 440
 441         avahi_dns_packet_free(reply);
 442
 443     } else {
 444         int unicast_response, flush_cache, auxiliary;
 445         AvahiDnsPacket *reply = NULL;
 446         AvahiRecord *r;
 447
 448         /* In case the query packet was truncated never respond
 449         immediately, because known answer suppression records might be
 450         contained in later packets */
 451         int tc = p && !!(avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS) & AVAHI_DNS_FLAG_TC);
 452
 453         while ((r = avahi_record_list_next(s->record_list, &flush_cache, &unicast_response, &auxiliary))) {
 454
 455             if (!avahi_interface_post_response(i, r, flush_cache, a, immediately || (flush_cache && !tc && !auxiliary)) && unicast_response) {
 456
 457                 append_aux_records_to_list(s, i, r, unicast_response);
 458
 459                 /* Due to some reasons the record has not been scheduled.
 460                  * The client requested an unicast response in that
 461                  * case. Therefore we prepare such a response */
 462
 463                 for (;;) {
 464
 465                     if (!reply) {
 466                         assert(p);
 467
 468                         if (!(reply = avahi_dns_packet_new_reply(p, i->hardware->mtu, 0, 0)))
 469                             break; /* OOM */
 470                     }
 471
 472                     if (avahi_dns_packet_append_record(reply, r, flush_cache, 0)) {
 473
 474                         /* Appending this record succeeded, so incremeant
 475                          * the specific header field, and return to the caller */
 476
 477                         avahi_dns_packet_inc_field(reply, AVAHI_DNS_FIELD_ANCOUNT);
 478
 479                         break;
 480                     }
 481
 482                     if (avahi_dns_packet_get_field(reply, AVAHI_DNS_FIELD_ANCOUNT) == 0) {
 483                         size_t size;
 484
 485                         /* The record is too large for one packet, so create a larger packet */
 486
 487                         avahi_dns_packet_free(reply);
 488                         size = avahi_record_get_estimate_size(r) + AVAHI_DNS_PACKET_HEADER_SIZE;
 489                         if (size > AVAHI_DNS_PACKET_MAX_SIZE)
 490                             size = AVAHI_DNS_PACKET_MAX_SIZE;
 491
 492                         if (!(reply = avahi_dns_packet_new_reply(p, size, 0, 1)))
 493                             break; /* OOM */
 494
 495                         if (!avahi_dns_packet_append_record(reply, r, flush_cache, 0)) {
 496                             char *t;
 497                             avahi_dns_packet_free(reply);
 498                             t = avahi_record_to_string(r);
 499                             avahi_log_warn("Record [%s] too large, doesn't fit in any packet!", t);
 500                             avahi_free(t);
 501                             break;
 502                         } else
 503                             avahi_dns_packet_inc_field(reply, AVAHI_DNS_FIELD_ANCOUNT);
 504                     }
 505
 506                     /* Appending the record didn't succeeed, so let's send this packet, and create a new one */
 507                     avahi_interface_send_packet_unicast(i, reply, a, port);
 508                     avahi_dns_packet_free(reply);
 509                     reply = NULL;
 510                 }
 511             }
 512
 513             avahi_record_unref(r);
 514         }
 515
 516         if (reply) {
 517             if (avahi_dns_packet_get_field(reply, AVAHI_DNS_FIELD_ANCOUNT) != 0)
 518                 avahi_interface_send_packet_unicast(i, reply, a, port);
 519             avahi_dns_packet_free(reply);
 520         }
 521     }
 522
 523     avahi_record_list_flush(s->record_list);
 524 }
 525
 526
 527 static void reflect_response(AvahiServer *s, AvahiInterface *i, AvahiRecord *r, int flush_cache) {
 528     AvahiInterface *j;
 529
 530     assert(s);
 531     assert(i);
 532     assert(r);
 533
 534     if (!s->config.enable_reflector)
 535         return;
 536
 537     for (j = s->monitor->interfaces; j; j = j->interface_next)
 538         if (j != i && (s->config.reflect_ipv || j->protocol == i->protocol))
 539             avahi_interface_post_response(j, r, flush_cache, NULL, 1);
 540 }
 541
 542 static void* reflect_cache_walk_callback(AvahiCache *c, AvahiKey *pattern, AvahiCacheEntry *e, void* userdata) {
 543     AvahiServer *s = userdata;
 544
 545     assert(c);
 546     assert(pattern);
 547     assert(e);
 548     assert(s);
 549
 550     avahi_record_list_push(s->record_list, e->record, e->cache_flush, 0, 0);
 551     return NULL;
 552 }
 553
 554 static void reflect_query(AvahiServer *s, AvahiInterface *i, AvahiKey *k) {
 555     AvahiInterface *j;
 556
 557     assert(s);
 558     assert(i);
 559     assert(k);
 560
 561     if (!s->config.enable_reflector)
 562         return;
 563
 564     for (j = s->monitor->interfaces; j; j = j->interface_next)
 565         if (j != i && (s->config.reflect_ipv || j->protocol == i->protocol)) {
 566             /* Post the query to other networks */
 567             avahi_interface_post_query(j, k, 1);
 568
 569             /* Reply from caches of other network. This is needed to
 570              * "work around" known answer suppression. */
 571
 572             avahi_cache_walk(j->cache, k, reflect_cache_walk_callback, s);
 573         }
 574 }
 575
 576 static void reflect_probe(AvahiServer *s, AvahiInterface *i, AvahiRecord *r) {
 577     AvahiInterface *j;
 578
 579     assert(s);
 580     assert(i);
 581     assert(r);
 582
 583     if (!s->config.enable_reflector)
 584         return;
 585
 586     for (j = s->monitor->interfaces; j; j = j->interface_next)
 587         if (j != i && (s->config.reflect_ipv || j->protocol == i->protocol))
 588             avahi_interface_post_probe(j, r, 1);
 589 }
 590
 591 static void handle_query_packet(AvahiServer *s, AvahiDnsPacket *p, AvahiInterface *i, const AvahiAddress *a, uint16_t port, int legacy_unicast, int from_local_iface) {
 592     size_t n;
 593     int is_probe;
 594
 595     assert(s);
 596     assert(p);
 597     assert(i);
 598     assert(a);
 599
 600 /*     avahi_log_debug("query"); */
 601
 602     assert(avahi_record_list_is_empty(s->record_list));
 603
 604     is_probe = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT) > 0;
 605
 606     /* Handle the questions */
 607     for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_QDCOUNT); n > 0; n --) {
 608         AvahiKey *key;
 609         int unicast_response = 0;
 610
 611         if (!(key = avahi_dns_packet_consume_key(p, &unicast_response))) {
 612             avahi_log_warn("Packet too short (1)");
 613             goto fail;
 614         }
 615
 616         if (!legacy_unicast && !from_local_iface) {
 617             reflect_query(s, i, key);
 618             avahi_cache_start_poof(i->cache, key, a);
 619         }
 620
 621         if (avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) == 0 &&
 622             !(avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS) & AVAHI_DNS_FLAG_TC))
 623             /* Allow our own queries to be suppressed by incoming
 624              * queries only when they do not include known answers */
 625             avahi_query_scheduler_incoming(i->query_scheduler, key);
 626
 627         avahi_server_prepare_matching_responses(s, i, key, unicast_response);
 628         avahi_key_unref(key);
 629     }
 630
 631     if (!legacy_unicast) {
 632
 633         /* Known Answer Suppression */
 634         for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT); n > 0; n --) {
 635             AvahiRecord *record;
 636             int unique = 0;
 637
 638             if (!(record = avahi_dns_packet_consume_record(p, &unique))) {
 639                 avahi_log_warn("Packet too short (2)");
 640                 goto fail;
 641             }
 642
 643             if (handle_conflict(s, i, record, unique, a)) {
 644                 avahi_response_scheduler_suppress(i->response_scheduler, record, a);
 645                 avahi_record_list_drop(s->record_list, record);
 646                 avahi_cache_stop_poof(i->cache, record, a);
 647             }
 648
 649             avahi_record_unref(record);
 650         }
 651
 652         /* Probe record */
 653         for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT); n > 0; n --) {
 654             AvahiRecord *record;
 655             int unique = 0;
 656
 657             if (!(record = avahi_dns_packet_consume_record(p, &unique))) {
 658                 avahi_log_warn("Packet too short (3)");
 659                 goto fail;
 660             }
 661
 662             if (!avahi_key_is_pattern(record->key)) {
 663                 if (!from_local_iface)
 664                     reflect_probe(s, i, record);
 665                 incoming_probe(s, record, i);
 666             }
 667
 668             avahi_record_unref(record);
 669         }
 670     }
 671
 672     if (!avahi_record_list_is_empty(s->record_list))
 673         avahi_server_generate_response(s, i, p, a, port, legacy_unicast, is_probe);
 674
 675     return;
 676
 677 fail:
 678     avahi_record_list_flush(s->record_list);
 679 }
 680
 681 static void handle_response_packet(AvahiServer *s, AvahiDnsPacket *p, AvahiInterface *i, const AvahiAddress *a, int from_local_iface) {
 682     unsigned n;
 683
 684     assert(s);
 685     assert(p);
 686     assert(i);
 687     assert(a);
 688
 689 /*     avahi_log_debug("response"); */
 690
 691     for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) +
 692              avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ARCOUNT); n > 0; n--) {
 693         AvahiRecord *record;
 694         int cache_flush = 0;
 695 /*         char *txt; */
 696
 697         if (!(record = avahi_dns_packet_consume_record(p, &cache_flush))) {
 698             avahi_log_warn("Packet too short (4)");
 699             break;
 700         }
 701
 702         if (!avahi_key_is_pattern(record->key)) {
 703
 704 /*             avahi_log_debug("Handling response: %s", txt = avahi_record_to_string(record)); */
 705 /*             avahi_free(txt); */
 706
 707             if (handle_conflict(s, i, record, cache_flush, a)) {
 708                 if (!from_local_iface)
 709                     reflect_response(s, i, record, cache_flush);
 710                 avahi_cache_update(i->cache, record, cache_flush, a);
 711                 avahi_response_scheduler_incoming(i->response_scheduler, record, cache_flush);
 712             }
 713         }
 714
 715         avahi_record_unref(record);
 716     }
 717
 718     /* If the incoming response contained a conflicting record, some
 719        records have been scheduling for sending. We need to flush them
 720        here. */
 721     if (!avahi_record_list_is_empty(s->record_list))
 722         avahi_server_generate_response(s, i, NULL, NULL, 0, 0, 1);
 723 }
 724
 725 static AvahiLegacyUnicastReflectSlot* allocate_slot(AvahiServer *s) {
 726     unsigned n, idx = (unsigned) -1;
 727     AvahiLegacyUnicastReflectSlot *slot;
 728
 729     assert(s);
 730
 731     if (!s->legacy_unicast_reflect_slots)
 732         s->legacy_unicast_reflect_slots = avahi_new0(AvahiLegacyUnicastReflectSlot*, AVAHI_MAX_LEGACY_UNICAST_REFLECT_SLOTS);
 733
 734     for (n = 0; n < AVAHI_MAX_LEGACY_UNICAST_REFLECT_SLOTS; n++, s->legacy_unicast_reflect_id++) {
 735         idx = s->legacy_unicast_reflect_id % AVAHI_MAX_LEGACY_UNICAST_REFLECT_SLOTS;
 736
 737         if (!s->legacy_unicast_reflect_slots[idx])
 738             break;
 739     }
 740
 741     if (idx == (unsigned) -1 || s->legacy_unicast_reflect_slots[idx])
 742         return NULL;
 743
 744     if (!(slot = avahi_new(AvahiLegacyUnicastReflectSlot, 1)))
 745         return NULL; /* OOM */
 746
 747     s->legacy_unicast_reflect_slots[idx] = slot;
 748     slot->id = s->legacy_unicast_reflect_id++;
 749     slot->server = s;
 750
 751     return slot;
 752 }
 753
 754 static void deallocate_slot(AvahiServer *s, AvahiLegacyUnicastReflectSlot *slot) {
 755     unsigned idx;
 756
 757     assert(s);
 758     assert(slot);
 759
 760     idx = slot->id % AVAHI_MAX_LEGACY_UNICAST_REFLECT_SLOTS;
 761
 762     assert(s->legacy_unicast_reflect_slots[idx] == slot);
 763
 764     avahi_time_event_free(slot->time_event);
 765
 766     avahi_free(slot);
 767     s->legacy_unicast_reflect_slots[idx] = NULL;
 768 }
 769
 770 static void free_slots(AvahiServer *s) {
 771     unsigned idx;
 772     assert(s);
 773
 774     if (!s->legacy_unicast_reflect_slots)
 775         return;
 776
 777     for (idx = 0; idx < AVAHI_MAX_LEGACY_UNICAST_REFLECT_SLOTS; idx ++)
 778         if (s->legacy_unicast_reflect_slots[idx])
 779             deallocate_slot(s, s->legacy_unicast_reflect_slots[idx]);
 780
 781     avahi_free(s->legacy_unicast_reflect_slots);
 782     s->legacy_unicast_reflect_slots = NULL;
 783 }
 784
 785 static AvahiLegacyUnicastReflectSlot* find_slot(AvahiServer *s, uint16_t id) {
 786     unsigned idx;
 787
 788     assert(s);
 789
 790     if (!s->legacy_unicast_reflect_slots)
 791         return NULL;
 792
 793     idx = id % AVAHI_MAX_LEGACY_UNICAST_REFLECT_SLOTS;
 794
 795     if (!s->legacy_unicast_reflect_slots[idx] || s->legacy_unicast_reflect_slots[idx]->id != id)
 796         return NULL;
 797
 798     return s->legacy_unicast_reflect_slots[idx];
 799 }
 800
 801 static void legacy_unicast_reflect_slot_timeout(AvahiTimeEvent *e, void *userdata) {
 802     AvahiLegacyUnicastReflectSlot *slot = userdata;
 803
 804     assert(e);
 805     assert(slot);
 806     assert(slot->time_event == e);
 807
 808     deallocate_slot(slot->server, slot);
 809 }
 810
 811 static void reflect_legacy_unicast_query_packet(AvahiServer *s, AvahiDnsPacket *p, AvahiInterface *i, const AvahiAddress *a, uint16_t port) {
 812     AvahiLegacyUnicastReflectSlot *slot;
 813     AvahiInterface *j;
 814
 815     assert(s);
 816     assert(p);
 817     assert(i);
 818     assert(a);
 819     assert(port > 0);
 820     assert(i->protocol == a->proto);
 821
 822     if (!s->config.enable_reflector)
 823         return;
 824
 825 /*     avahi_log_debug("legacy unicast reflector"); */
 826
 827     /* Reflecting legacy unicast queries is a little more complicated
 828        than reflecting normal queries, since we must route the
 829        responses back to the right client. Therefore we must store
 830        some information for finding the right client contact data for
 831        response packets. In contrast to normal queries legacy
 832        unicast query and response packets are reflected untouched and
 833        are not reassembled into larger packets */
 834
 835     if (!(slot = allocate_slot(s))) {
 836         /* No slot available, we drop this legacy unicast query */
 837         avahi_log_warn("No slot available for legacy unicast reflection, dropping query packet.");
 838         return;
 839     }
 840
 841     slot->original_id = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ID);
 842     slot->address = *a;
 843     slot->port = port;
 844     slot->interface = i->hardware->index;
 845
 846     avahi_elapse_time(&slot->elapse_time, 2000, 0);
 847     slot->time_event = avahi_time_event_new(s->time_event_queue, &slot->elapse_time, legacy_unicast_reflect_slot_timeout, slot);
 848
 849     /* Patch the packet with our new locally generatedt id */
 850     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_ID, slot->id);
 851
 852     for (j = s->monitor->interfaces; j; j = j->interface_next)
 853         if (avahi_interface_relevant(j) &&
 854             j != i &&
 855             (s->config.reflect_ipv || j->protocol == i->protocol)) {
 856
 857             if (j->protocol == AVAHI_PROTO_INET && s->fd_legacy_unicast_ipv4 >= 0) {
 858                 avahi_send_dns_packet_ipv4(s->fd_legacy_unicast_ipv4, j->hardware->index, p, NULL, 0);
 859                 } else if (j->protocol == AVAHI_PROTO_INET6 && s->fd_legacy_unicast_ipv6 >= 0)
 860                 avahi_send_dns_packet_ipv6(s->fd_legacy_unicast_ipv6, j->hardware->index, p, NULL, 0);
 861         }
 862
 863     /* Reset the id */
 864     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_ID, slot->original_id);
 865 }
 866
 867 static int originates_from_local_legacy_unicast_socket(AvahiServer *s, const struct sockaddr *sa) {
 868     AvahiAddress a;
 869     assert(s);
 870     assert(sa);
 871
 872     if (!s->config.enable_reflector)
 873         return 0;
 874
 875     avahi_address_from_sockaddr(sa, &a);
 876
 877     if (!avahi_address_is_local(s->monitor, &a))
 878         return 0;
 879
 880     if (sa->sa_family == AF_INET && s->fd_legacy_unicast_ipv4 >= 0) {
 881         struct sockaddr_in lsa;
 882         socklen_t l = sizeof(lsa);
 883
 884         if (getsockname(s->fd_legacy_unicast_ipv4, &lsa, &l) != 0)
 885             avahi_log_warn("getsockname(): %s", strerror(errno));
 886         else
 887             return lsa.sin_port == ((const struct sockaddr_in*) sa)->sin_port;
 888
 889     }
 890
 891     if (sa->sa_family == AF_INET6 && s->fd_legacy_unicast_ipv6 >= 0) {
 892         struct sockaddr_in6 lsa;
 893         socklen_t l = sizeof(lsa);
 894
 895         if (getsockname(s->fd_legacy_unicast_ipv6, &lsa, &l) != 0)
 896             avahi_log_warn("getsockname(): %s", strerror(errno));
 897         else
 898             return lsa.sin6_port == ((const struct sockaddr_in6*) sa)->sin6_port;
 899     }
 900
 901     return 0;
 902 }
 903
 904 static int is_mdns_mcast_address(const AvahiAddress *a) {
 905     AvahiAddress b;
 906     assert(a);
 907
 908     avahi_address_parse(a->proto == AVAHI_PROTO_INET ? AVAHI_IPV4_MCAST_GROUP : AVAHI_IPV6_MCAST_GROUP, a->proto, &b);
 909     return avahi_address_cmp(a, &b) == 0;
 910 }
 911
 912 static int originates_from_local_iface(AvahiServer *s, AvahiIfIndex iface, const AvahiAddress *a, uint16_t port) {
 913     assert(s);
 914     assert(iface != AVAHI_IF_UNSPEC);
 915     assert(a);
 916
 917     /* If it isn't the MDNS port it can't be generated by us */
 918     if (port != AVAHI_MDNS_PORT)
 919         return 0;
 920
 921     return avahi_interface_has_address(s->monitor, iface, a);
 922 }
 923
 924 static void dispatch_packet(AvahiServer *s, AvahiDnsPacket *p, const struct sockaddr *sa, AvahiAddress *dest, AvahiIfIndex iface, int ttl) {
 925     AvahiInterface *i;
 926     AvahiAddress a;
 927     uint16_t port;
 928     int from_local_iface = 0;
 929
 930     assert(s);
 931     assert(p);
 932     assert(sa);
 933     assert(dest);
 934     assert(iface > 0);
 935
 936     if (!(i = avahi_interface_monitor_get_interface(s->monitor, iface, avahi_af_to_proto(sa->sa_family))) ||
 937         !avahi_interface_relevant(i)) {
 938         avahi_log_warn("Recieved packet from invalid interface.");
 939         return;
 940     }
 941
 942 /*     avahi_log_debug("new packet recieved on interface '%s.%i'.", i->hardware->name, i->protocol); */
 943
 944     port = avahi_port_from_sockaddr(sa);
 945     avahi_address_from_sockaddr(sa, &a);
 946
 947     if (avahi_address_is_ipv4_in_ipv6(&a))
 948         /* This is an IPv4 address encapsulated in IPv6, so let's ignore it. */
 949         return;
 950
 951     if (originates_from_local_legacy_unicast_socket(s, sa))
 952         /* This originates from our local reflector, so let's ignore it */
 953         return;
 954
 955     /* We don't want to reflect local traffic, so we check if this packet is generated locally. */
 956     if (s->config.enable_reflector)
 957         from_local_iface = originates_from_local_iface(s, iface, &a, port);
 958
 959     if (avahi_dns_packet_check_valid_multicast(p) < 0) {
 960         avahi_log_warn("Recieved invalid packet.");
 961         return;
 962     }
 963
 964     if (avahi_dns_packet_is_query(p)) {
 965         int legacy_unicast = 0;
 966
 967         if (avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ARCOUNT) != 0) {
 968             avahi_log_warn("Invalid query packet.");
 969             return;
 970         }
 971
 972         if (port != AVAHI_MDNS_PORT) {
 973             /* Legacy Unicast */
 974
 975             if ((avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) != 0 ||
 976                  avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT) != 0)) {
 977                 avahi_log_warn("Invalid legacy unicast query packet.");
 978                 return;
 979             }
 980
 981             legacy_unicast = 1;
 982         }
 983
 984         if (legacy_unicast)
 985             reflect_legacy_unicast_query_packet(s, p, i, &a, port);
 986
 987         handle_query_packet(s, p, i, &a, port, legacy_unicast, from_local_iface);
 988
 989 /*         avahi_log_debug("Handled query"); */
 990     } else {
 991         if (port != AVAHI_MDNS_PORT) {
 992             avahi_log_warn("Recieved repsonse with invalid source port %u on interface '%s.%i'", port, i->hardware->name, i->protocol);
 993             return;
 994         }
 995
 996         if (ttl != 255 && s->config.check_response_ttl) {
 997             avahi_log_warn("Recieved response with invalid TTL %u on interface '%s.%i'.", ttl, i->hardware->name, i->protocol);
 998             return;
 999         }
1000
1001         if (!is_mdns_mcast_address(dest) &&
1002             !avahi_interface_address_on_link(i, &a)) {
1003             avahi_log_warn("Recivied non-local response on interface '%s.%i'.", i->hardware->name, i->protocol);
1004             return;
1005         }
1006
1007         if (avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_QDCOUNT) != 0 ||
1008             avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) == 0 ||
1009             avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT) != 0) {
1010             avahi_log_warn("Invalid response packet.");
1011             return;
1012         }
1013
1014         handle_response_packet(s, p, i, &a, from_local_iface);
1015 /*         avahi_log_debug("Handled response"); */
1016     }
1017 }
1018
1019 static void dispatch_legacy_unicast_packet(AvahiServer *s, AvahiDnsPacket *p, const struct sockaddr *sa, AvahiIfIndex iface) {
1020     AvahiInterface *i, *j;
1021     AvahiAddress a;
1022     AvahiLegacyUnicastReflectSlot *slot;
1023
1024     assert(s);
1025     assert(p);
1026     assert(sa);
1027     assert(iface > 0);
1028
1029     if (!(i = avahi_interface_monitor_get_interface(s->monitor, iface, avahi_af_to_proto(sa->sa_family))) ||
1030         !avahi_interface_relevant(i)) {
1031         avahi_log_warn("Recieved packet from invalid interface.");
1032         return;
1033     }
1034
1035 /*     avahi_log_debug("new legacy unicast packet recieved on interface '%s.%i'.", i->hardware->name, i->protocol); */
1036
1037     avahi_address_from_sockaddr(sa, &a);
1038
1039     if (avahi_address_is_ipv4_in_ipv6(&a))
1040         /* This is an IPv4 address encapsulated in IPv6, so let's ignore it. */
1041         return;
1042
1043     if (avahi_dns_packet_check_valid(p) < 0 || avahi_dns_packet_is_query(p)) {
1044         avahi_log_warn("Recieved invalid packet.");
1045         return;
1046     }
1047
1048     if (!(slot = find_slot(s, avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ID)))) {
1049         avahi_log_warn("Recieved legacy unicast response with unknown id");
1050         return;
1051     }
1052
1053     if (!(j = avahi_interface_monitor_get_interface(s->monitor, slot->interface, slot->address.proto)) ||
1054         !avahi_interface_relevant(j))
1055         return;
1056
1057     /* Patch the original ID into this response */
1058     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_ID, slot->original_id);
1059
1060     /* Forward the response to the correct client */
1061     avahi_interface_send_packet_unicast(j, p, &slot->address, slot->port);
1062
1063     /* Undo changes to packet */
1064     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_ID, slot->id);
1065 }
1066
1067 static void socket_event(AvahiWatch *w, int fd, AvahiWatchEvent events, void *userdata) {
1068     AvahiServer *s = userdata;
1069     AvahiAddress dest;
1070     AvahiDnsPacket *p;
1071     AvahiIfIndex iface;
1072     uint8_t ttl;
1073     struct sockaddr_in sa;
1074     struct sockaddr_in6 sa6;
1075
1076     assert(w);
1077     assert(fd >= 0);
1078
1079     if (events & AVAHI_WATCH_IN) {
1080
1081         if (fd == s->fd_ipv4) {
1082             dest.proto = AVAHI_PROTO_INET;
1083             if ((p = avahi_recv_dns_packet_ipv4(s->fd_ipv4, &sa, &dest.data.ipv4, &iface, &ttl))) {
1084                 dispatch_packet(s, p, (struct sockaddr*) &sa, &dest, iface, ttl);
1085                 avahi_dns_packet_free(p);
1086             }
1087         } else if (fd == s->fd_ipv6) {
1088             dest.proto = AVAHI_PROTO_INET6;
1089
1090             if ((p = avahi_recv_dns_packet_ipv6(s->fd_ipv6, &sa6, &dest.data.ipv6, &iface, &ttl))) {
1091                 dispatch_packet(s, p, (struct sockaddr*) &sa6, &dest, iface, ttl);
1092                 avahi_dns_packet_free(p);
1093             }
1094         } else if (fd == s->fd_legacy_unicast_ipv4) {
1095             dest.proto = AVAHI_PROTO_INET;
1096
1097             if ((p = avahi_recv_dns_packet_ipv4(s->fd_legacy_unicast_ipv4, &sa, &dest.data.ipv4, &iface, &ttl))) {
1098                 dispatch_legacy_unicast_packet(s, p, (struct sockaddr*) &sa, iface);
1099                 avahi_dns_packet_free(p);
1100             }
1101         } else if (fd == s->fd_legacy_unicast_ipv6) {
1102             dest.proto = AVAHI_PROTO_INET6;
1103
1104             if ((p = avahi_recv_dns_packet_ipv6(s->fd_legacy_unicast_ipv6, &sa6, &dest.data.ipv6, &iface, &ttl))) {
1105                 dispatch_legacy_unicast_packet(s, p, (struct sockaddr*) &sa6, iface);
1106                 avahi_dns_packet_free(p);
1107             }
1108         }
1109
1110         cleanup_dead(s);
1111     } else
1112         abort();
1113 }
1114
1115 static void server_set_state(AvahiServer *s, AvahiServerState state) {
1116     assert(s);
1117
1118     if (s->state == state)
1119         return;
1120
1121     s->state = state;
1122
1123     if (s->callback)
1124         s->callback(s, state, s->userdata);
1125 }
1126
1127 static void withdraw_host_rrs(AvahiServer *s) {
1128     assert(s);
1129
1130     if (s->hinfo_entry_group)
1131         avahi_s_entry_group_reset(s->hinfo_entry_group);
1132
1133     if (s->browse_domain_entry_group)
1134         avahi_s_entry_group_reset(s->browse_domain_entry_group);
1135
1136     avahi_update_host_rrs(s->monitor, 1);
1137     s->n_host_rr_pending = 0;
1138 }
1139
1140 void avahi_server_decrease_host_rr_pending(AvahiServer *s) {
1141     assert(s);
1142
1143     assert(s->n_host_rr_pending > 0);
1144
1145     if (--s->n_host_rr_pending == 0)
1146         server_set_state(s, AVAHI_SERVER_RUNNING);
1147 }
1148
1149 void avahi_server_increase_host_rr_pending(AvahiServer *s) {
1150     assert(s);
1151
1152     s->n_host_rr_pending ++;
1153 }
1154
1155 void avahi_host_rr_entry_group_callback(AvahiServer *s, AvahiSEntryGroup *g, AvahiEntryGroupState state, void *userdata) {
1156     assert(s);
1157     assert(g);
1158
1159     if (state == AVAHI_ENTRY_GROUP_REGISTERING &&
1160         s->state == AVAHI_SERVER_REGISTERING)
1161         avahi_server_increase_host_rr_pending(s);
1162
1163     else if (state == AVAHI_ENTRY_GROUP_COLLISION &&
1164         (s->state == AVAHI_SERVER_REGISTERING || s->state == AVAHI_SERVER_RUNNING)) {
1165         withdraw_host_rrs(s);
1166         server_set_state(s, AVAHI_SERVER_COLLISION);
1167
1168     } else if (state == AVAHI_ENTRY_GROUP_ESTABLISHED &&
1169                s->state == AVAHI_SERVER_REGISTERING)
1170         avahi_server_decrease_host_rr_pending(s);
1171 }
1172
1173 static void register_hinfo(AvahiServer *s) {
1174     struct utsname utsname;
1175     AvahiRecord *r;
1176
1177     assert(s);
1178
1179     if (!s->config.publish_hinfo)
1180         return;
1181
1182     if (s->hinfo_entry_group)
1183         assert(avahi_s_entry_group_is_empty(s->hinfo_entry_group));
1184     else
1185         s->hinfo_entry_group = avahi_s_entry_group_new(s, avahi_host_rr_entry_group_callback, NULL);
1186
1187     if (!s->hinfo_entry_group) {
1188         avahi_log_warn("Failed to create HINFO entry group: %s", avahi_strerror(s->error));
1189         return;
1190     }
1191
1192     /* Fill in HINFO rr */
1193     if ((r = avahi_record_new_full(s->host_name_fqdn, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_HINFO, AVAHI_DEFAULT_TTL_HOST_NAME))) {
1194         uname(&utsname);
1195         r->data.hinfo.cpu = avahi_strdup(avahi_strup(utsname.machine));
1196         r->data.hinfo.os = avahi_strdup(avahi_strup(utsname.sysname));
1197
1198         if (avahi_server_add(s, s->hinfo_entry_group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, AVAHI_PUBLISH_UNIQUE, r) < 0) {
1199             avahi_log_warn("Failed to add HINFO RR: %s", avahi_strerror(s->error));
1200             return;
1201         }
1202
1203         avahi_record_unref(r);
1204     }
1205
1206     if (avahi_s_entry_group_commit(s->hinfo_entry_group) < 0)
1207         avahi_log_warn("Failed to commit HINFO entry group: %s", avahi_strerror(s->error));
1208
1209 }
1210
1211 static void register_localhost(AvahiServer *s) {
1212     AvahiAddress a;
1213     assert(s);
1214
1215     /* Add localhost entries */
1216     avahi_address_parse("127.0.0.1", AVAHI_PROTO_INET, &a);
1217     avahi_server_add_address(s, NULL, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, AVAHI_PUBLISH_NO_PROBE|AVAHI_PUBLISH_NO_ANNOUNCE, "localhost", &a);
1218
1219     avahi_address_parse("::1", AVAHI_PROTO_INET6, &a);
1220     avahi_server_add_address(s, NULL, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, AVAHI_PUBLISH_NO_PROBE|AVAHI_PUBLISH_NO_ANNOUNCE, "ip6-localhost", &a);
1221 }
1222
1223 static void register_browse_domain(AvahiServer *s) {
1224     assert(s);
1225
1226     if (!s->config.publish_domain)
1227         return;
1228
1229     if (s->browse_domain_entry_group)
1230         assert(avahi_s_entry_group_is_empty(s->browse_domain_entry_group));
1231     else
1232         s->browse_domain_entry_group = avahi_s_entry_group_new(s, NULL, NULL);
1233
1234     if (!s->browse_domain_entry_group) {
1235         avahi_log_warn("Failed to create browse domain entry group: %s", avahi_strerror(s->error));
1236         return;
1237     }
1238
1239     if (avahi_server_add_ptr(s, s->browse_domain_entry_group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, 0, AVAHI_DEFAULT_TTL, "b._dns-sd._udp.local", s->domain_name) < 0) {
1240         avahi_log_warn("Failed to add browse domain RR: %s", avahi_strerror(s->error));
1241         return;
1242     }
1243
1244     if (avahi_s_entry_group_commit(s->browse_domain_entry_group) < 0)
1245         avahi_log_warn("Failed to commit browse domain entry group: %s", avahi_strerror(s->error));
1246 }
1247
1248 static void register_stuff(AvahiServer *s) {
1249     assert(s);
1250
1251     server_set_state(s, AVAHI_SERVER_REGISTERING);
1252     s->n_host_rr_pending ++;  /** Make sure that the state isn't changed tp AVAHI_SERVER_RUNNING too early */
1253
1254     register_hinfo(s);
1255     register_browse_domain(s);
1256     avahi_update_host_rrs(s->monitor, 0);
1257
1258     s->n_host_rr_pending --;
1259
1260     if (s->n_host_rr_pending == 0)
1261         server_set_state(s, AVAHI_SERVER_RUNNING);
1262 }
1263
1264 static void update_fqdn(AvahiServer *s) {
1265     char *n;
1266
1267     assert(s);
1268     assert(s->host_name);
1269     assert(s->domain_name);
1270
1271     if (!(n = avahi_strdup_printf("%s.%s", s->host_name, s->domain_name)))
1272         return; /* OOM */
1273
1274     avahi_free(s->host_name_fqdn);
1275     s->host_name_fqdn = n;
1276 }
1277
1278 int avahi_server_set_host_name(AvahiServer *s, const char *host_name) {
1279     assert(s);
1280     assert(host_name);
1281
1282     if (host_name && !avahi_is_valid_host_name(host_name))
1283         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
1284
1285     withdraw_host_rrs(s);
1286
1287     avahi_free(s->host_name);
1288     s->host_name = host_name ? avahi_normalize_name_strdup(host_name) : avahi_get_host_name_strdup();
1289     s->host_name[strcspn(s->host_name, ".")] = 0;
1290     update_fqdn(s);
1291
1292     register_stuff(s);
1293     return AVAHI_OK;
1294 }
1295
1296 int avahi_server_set_domain_name(AvahiServer *s, const char *domain_name) {
1297     assert(s);
1298     assert(domain_name);
1299
1300     if (domain_name && !avahi_is_valid_domain_name(domain_name))
1301         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
1302
1303     withdraw_host_rrs(s);
1304
1305     avahi_free(s->domain_name);
1306     s->domain_name = domain_name ? avahi_normalize_name_strdup(domain_name) : avahi_strdup("local");
1307     update_fqdn(s);
1308
1309     register_stuff(s);
1310     return AVAHI_OK;
1311 }
1312
1313 static int valid_server_config(const AvahiServerConfig *sc) {
1314
1315     if (sc->host_name && !avahi_is_valid_host_name(sc->host_name))
1316         return AVAHI_ERR_INVALID_HOST_NAME;
1317
1318     if (sc->domain_name && !avahi_is_valid_domain_name(sc->domain_name))
1319         return AVAHI_ERR_INVALID_DOMAIN_NAME;
1320
1321     return AVAHI_OK;
1322 }
1323
1324 static int setup_sockets(AvahiServer *s) {
1325     assert(s);
1326
1327     s->fd_ipv4 = s->config.use_ipv4 ? avahi_open_socket_ipv4(s->config.disallow_other_stacks) : -1;
1328     s->fd_ipv6 = s->config.use_ipv6 ? avahi_open_socket_ipv6(s->config.disallow_other_stacks) : -1;
1329
1330     if (s->fd_ipv6 < 0 && s->fd_ipv4 < 0)
1331         return AVAHI_ERR_NO_NETWORK;
1332
1333     if (s->fd_ipv4 < 0 && s->config.use_ipv4)
1334         avahi_log_notice("Failed to create IPv4 socket, proceeding in IPv6 only mode");
1335     else if (s->fd_ipv6 < 0 && s->config.use_ipv6)
1336         avahi_log_notice("Failed to create IPv6 socket, proceeding in IPv4 only mode");
1337
1338     s->fd_legacy_unicast_ipv4 = s->fd_ipv4 >= 0 && s->config.enable_reflector ? avahi_open_unicast_socket_ipv4() : -1;
1339     s->fd_legacy_unicast_ipv6 = s->fd_ipv6 >= 0 && s->config.enable_reflector ? avahi_open_unicast_socket_ipv6() : -1;
1340
1341     s->watch_ipv4 =
1342         s->watch_ipv6 =
1343         s->watch_legacy_unicast_ipv4 =
1344         s->watch_legacy_unicast_ipv6 = NULL;
1345
1346     if (s->fd_ipv4 >= 0)
1347         s->watch_ipv4 = s->poll_api->watch_new(s->poll_api, s->fd_ipv4, AVAHI_WATCH_IN, socket_event, s);
1348     if (s->fd_ipv6 >= 0)
1349         s->watch_ipv6 = s->poll_api->watch_new(s->poll_api, s->fd_ipv6, AVAHI_WATCH_IN, socket_event, s);
1350
1351     if (s->fd_legacy_unicast_ipv4 >= 0)
1352         s->watch_legacy_unicast_ipv4 = s->poll_api->watch_new(s->poll_api, s->fd_legacy_unicast_ipv4, AVAHI_WATCH_IN, socket_event, s);
1353     if (s->fd_legacy_unicast_ipv6 >= 0)
1354         s->watch_legacy_unicast_ipv6 = s->poll_api->watch_new(s->poll_api, s->fd_legacy_unicast_ipv6, AVAHI_WATCH_IN, socket_event, s);
1355
1356     return 0;
1357 }
1358
1359 AvahiServer *avahi_server_new(const AvahiPoll *poll_api, const AvahiServerConfig *sc, AvahiServerCallback callback, void* userdata, int *error) {
1360     AvahiServer *s;
1361     int e;
1362
1363     if ((e = valid_server_config(sc)) < 0) {
1364         if (error)
1365             *error = e;
1366         return NULL;
1367     }
1368
1369     if (!(s = avahi_new(AvahiServer, 1))) {
1370         if (error)
1371             *error = AVAHI_ERR_NO_MEMORY;
1372
1373         return NULL;
1374     }
1375
1376     s->poll_api = poll_api;
1377
1378     if (sc)
1379         avahi_server_config_copy(&s->config, sc);
1380     else
1381         avahi_server_config_init(&s->config);
1382
1383     if ((e = setup_sockets(s)) < 0) {
1384         if (error)
1385             *error = e;
1386
1387         avahi_server_config_free(&s->config);
1388         avahi_free(s);
1389
1390         return NULL;
1391     }
1392
1393
1394     s->n_host_rr_pending = 0;
1395     s->need_entry_cleanup = 0;
1396     s->need_group_cleanup = 0;
1397     s->need_browser_cleanup = 0;
1398
1399     s->time_event_queue = avahi_time_event_queue_new(poll_api);
1400
1401     s->callback = callback;
1402     s->userdata = userdata;
1403
1404     s->entries_by_key = avahi_hashmap_new((AvahiHashFunc) avahi_key_hash, (AvahiEqualFunc) avahi_key_equal, NULL, NULL);
1405     AVAHI_LLIST_HEAD_INIT(AvahiEntry, s->entries);
1406     AVAHI_LLIST_HEAD_INIT(AvahiGroup, s->groups);
1407
1408     s->record_browser_hashmap = avahi_hashmap_new((AvahiHashFunc) avahi_key_hash, (AvahiEqualFunc) avahi_key_equal, NULL, NULL);
1409     AVAHI_LLIST_HEAD_INIT(AvahiSRecordBrowser, s->record_browsers);
1410     AVAHI_LLIST_HEAD_INIT(AvahiSHostNameResolver, s->host_name_resolvers);
1411     AVAHI_LLIST_HEAD_INIT(AvahiSAddressResolver, s->address_resolvers);
1412     AVAHI_LLIST_HEAD_INIT(AvahiSDomainBrowser, s->domain_browsers);
1413     AVAHI_LLIST_HEAD_INIT(AvahiSServiceTypeBrowser, s->service_type_browsers);
1414     AVAHI_LLIST_HEAD_INIT(AvahiSServiceBrowser, s->service_browsers);
1415     AVAHI_LLIST_HEAD_INIT(AvahiSServiceResolver, s->service_resolvers);
1416     AVAHI_LLIST_HEAD_INIT(AvahiSDNSServerBrowser, s->dns_server_browsers);
1417
1418     s->legacy_unicast_reflect_slots = NULL;
1419     s->legacy_unicast_reflect_id = 0;
1420
1421     if (s->config.enable_wide_area) {
1422         s->wide_area_lookup_engine = avahi_wide_area_engine_new(s);
1423         avahi_wide_area_set_servers(s->wide_area_lookup_engine, s->config.wide_area_servers, s->config.n_wide_area_servers);
1424     } else
1425         s->wide_area_lookup_engine = NULL;
1426
1427     s->multicast_lookup_engine = avahi_multicast_lookup_engine_new(s);
1428
1429     do {
1430         s->local_service_cookie = (uint32_t) rand() * (uint32_t) rand();
1431     } while (s->local_service_cookie == AVAHI_SERVICE_COOKIE_INVALID);
1432
1433     /* Get host name */
1434     s->host_name = s->config.host_name ? avahi_normalize_name_strdup(s->config.host_name) : avahi_get_host_name_strdup();
1435     s->host_name[strcspn(s->host_name, ".")] = 0;
1436     s->domain_name = s->config.domain_name ? avahi_normalize_name_strdup(s->config.domain_name) : avahi_strdup("local");
1437     s->host_name_fqdn = NULL;
1438     update_fqdn(s);
1439
1440     s->record_list = avahi_record_list_new();
1441
1442     s->state = AVAHI_SERVER_INVALID;
1443
1444     s->monitor = avahi_interface_monitor_new(s);
1445     avahi_interface_monitor_sync(s->monitor);
1446
1447     register_localhost(s);
1448
1449     s->hinfo_entry_group = NULL;
1450     s->browse_domain_entry_group = NULL;
1451     register_stuff(s);
1452
1453     s->error = AVAHI_OK;
1454
1455     return s;
1456 }
1457
1458 void avahi_server_free(AvahiServer* s) {
1459     assert(s);
1460
1461     /* Remove all browsers */
1462
1463     while (s->dns_server_browsers)
1464         avahi_s_dns_server_browser_free(s->dns_server_browsers);
1465     while (s->host_name_resolvers)
1466         avahi_s_host_name_resolver_free(s->host_name_resolvers);
1467     while (s->address_resolvers)
1468         avahi_s_address_resolver_free(s->address_resolvers);
1469     while (s->domain_browsers)
1470         avahi_s_domain_browser_free(s->domain_browsers);
1471     while (s->service_type_browsers)
1472         avahi_s_service_type_browser_free(s->service_type_browsers);
1473     while (s->service_browsers)
1474         avahi_s_service_browser_free(s->service_browsers);
1475     while (s->service_resolvers)
1476         avahi_s_service_resolver_free(s->service_resolvers);
1477     while (s->record_browsers)
1478         avahi_s_record_browser_destroy(s->record_browsers);
1479
1480     /* Remove all locally rgeistered stuff */
1481
1482     while(s->entries)
1483         free_entry(s, s->entries);
1484
1485     avahi_interface_monitor_free(s->monitor);
1486
1487     while (s->groups)
1488         free_group(s, s->groups);
1489
1490     free_slots(s);
1491
1492     avahi_hashmap_free(s->entries_by_key);
1493     avahi_record_list_free(s->record_list);
1494     avahi_hashmap_free(s->record_browser_hashmap);
1495
1496     if (s->wide_area_lookup_engine)
1497         avahi_wide_area_engine_free(s->wide_area_lookup_engine);
1498     avahi_multicast_lookup_engine_free(s->multicast_lookup_engine);
1499
1500     avahi_time_event_queue_free(s->time_event_queue);
1501
1502     /* Free watches */
1503
1504     if (s->watch_ipv4)
1505         s->poll_api->watch_free(s->watch_ipv4);
1506     if (s->watch_ipv6)
1507         s->poll_api->watch_free(s->watch_ipv6);
1508
1509     if (s->watch_legacy_unicast_ipv4)
1510         s->poll_api->watch_free(s->watch_legacy_unicast_ipv4);
1511     if (s->watch_legacy_unicast_ipv6)
1512         s->poll_api->watch_free(s->watch_legacy_unicast_ipv6);
1513
1514     /* Free sockets */
1515
1516     if (s->fd_ipv4 >= 0)
1517         close(s->fd_ipv4);
1518     if (s->fd_ipv6 >= 0)
1519         close(s->fd_ipv6);
1520
1521     if (s->fd_legacy_unicast_ipv4 >= 0)
1522         close(s->fd_legacy_unicast_ipv4);
1523     if (s->fd_legacy_unicast_ipv6 >= 0)
1524         close(s->fd_legacy_unicast_ipv6);
1525
1526     /* Free other stuff */
1527
1528     avahi_free(s->host_name);
1529     avahi_free(s->domain_name);
1530     avahi_free(s->host_name_fqdn);
1531
1532     avahi_server_config_free(&s->config);
1533
1534     avahi_free(s);
1535 }
1536
1537 static int check_record_conflict(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, AvahiRecord *r, AvahiPublishFlags flags) {
1538     AvahiEntry *e;
1539
1540     assert(s);
1541     assert(r);
1542
1543     for (e = avahi_hashmap_lookup(s->entries_by_key, r->key); e; e = e->by_key_next) {
1544         if (e->dead)
1545             continue;
1546
1547         if (!(flags & AVAHI_PUBLISH_UNIQUE) && !(e->flags & AVAHI_PUBLISH_UNIQUE))
1548             continue;
1549
1550         if ((flags & AVAHI_PUBLISH_ALLOW_MULTIPLE) && (e->flags & AVAHI_PUBLISH_ALLOW_MULTIPLE) )
1551             continue;
1552
1553         if ((interface <= 0 ||
1554              e->interface <= 0 ||
1555              e->interface == interface) &&
1556             (protocol == AVAHI_PROTO_UNSPEC ||
1557              e->protocol == AVAHI_PROTO_UNSPEC ||
1558              e->protocol == protocol))
1559
1560             return -1;
1561     }
1562
1563     return 0;
1564 }
1565
1566 int avahi_server_add(
1567     AvahiServer *s,
1568     AvahiSEntryGroup *g,
1569     AvahiIfIndex interface,
1570     AvahiProtocol protocol,
1571     AvahiPublishFlags flags,
1572     AvahiRecord *r) {
1573
1574     AvahiEntry *e, *t;
1575
1576     assert(s);
1577     assert(r);
1578
1579     if (!AVAHI_IF_VALID(interface))
1580         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
1581
1582     if (!AVAHI_PROTO_VALID(protocol))
1583         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
1584
1585     if (!AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_ANNOUNCE|AVAHI_PUBLISH_NO_PROBE|AVAHI_PUBLISH_UNIQUE|AVAHI_PUBLISH_ALLOW_MULTIPLE|AVAHI_PUBLISH_IS_PROXY))
1586         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
1587
1588     if (!avahi_is_valid_domain_name(r->key->name))
1589         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
1590
1591     if (r->ttl == 0)
1592         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_TTL);
1593
1594     if (avahi_key_is_pattern(r->key))
1595         return avahi_server_set_errno(s, AVAHI_ERR_IS_PATTERN);
1596
1597     if (!avahi_record_is_valid(r))
1598         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_RECORD);
1599
1600     if (check_record_conflict(s, interface, protocol, r, flags) < 0)
1601         return avahi_server_set_errno(s, AVAHI_ERR_LOCAL_COLLISION);
1602
1603     if (!(e = avahi_new(AvahiEntry, 1)))
1604         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1605
1606     e->server = s;
1607     e->record = avahi_record_ref(r);
1608     e->group = g;
1609     e->interface = interface;
1610     e->protocol = protocol;
1611     e->flags = flags;
1612     e->dead = 0;
1613
1614     AVAHI_LLIST_HEAD_INIT(AvahiAnnouncement, e->announcements);
1615
1616     AVAHI_LLIST_PREPEND(AvahiEntry, entries, s->entries, e);
1617
1618     /* Insert into hash table indexed by name */
1619     t = avahi_hashmap_lookup(s->entries_by_key, e->record->key);
1620     AVAHI_LLIST_PREPEND(AvahiEntry, by_key, t, e);
1621     avahi_hashmap_replace(s->entries_by_key, e->record->key, t);
1622
1623     /* Insert into group list */
1624     if (g)
1625         AVAHI_LLIST_PREPEND(AvahiEntry, by_group, g->entries, e);
1626
1627     avahi_announce_entry(s, e);
1628
1629     return 0;
1630 }
1631
1632 const AvahiRecord *avahi_server_iterate(AvahiServer *s, AvahiSEntryGroup *g, void **state) {
1633     AvahiEntry **e = (AvahiEntry**) state;
1634     assert(s);
1635     assert(e);
1636
1637     if (!*e)
1638         *e = g ? g->entries : s->entries;
1639
1640     while (*e && (*e)->dead)
1641         *e = g ? (*e)->by_group_next : (*e)->entries_next;
1642
1643     if (!*e)
1644         return NULL;
1645
1646     return avahi_record_ref((*e)->record);
1647 }
1648
1649 int avahi_server_dump(AvahiServer *s, AvahiDumpCallback callback, void* userdata) {
1650     AvahiEntry *e;
1651
1652     assert(s);
1653     assert(callback);
1654
1655     callback(";;; ZONE DUMP FOLLOWS ;;;", userdata);
1656
1657     for (e = s->entries; e; e = e->entries_next) {
1658         char *t;
1659         char ln[256];
1660
1661         if (e->dead)
1662             continue;
1663
1664         if (!(t = avahi_record_to_string(e->record)))
1665             return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1666
1667         snprintf(ln, sizeof(ln), "%s ; iface=%i proto=%i", t, e->interface, e->protocol);
1668         avahi_free(t);
1669
1670         callback(ln, userdata);
1671     }
1672
1673     avahi_dump_caches(s->monitor, callback, userdata);
1674
1675     if (s->wide_area_lookup_engine)
1676         avahi_wide_area_cache_dump(s->wide_area_lookup_engine, callback, userdata);
1677     return AVAHI_OK;
1678 }
1679
1680 int avahi_server_add_ptr(
1681     AvahiServer *s,
1682     AvahiSEntryGroup *g,
1683     AvahiIfIndex interface,
1684     AvahiProtocol protocol,
1685     AvahiPublishFlags flags,
1686     uint32_t ttl,
1687     const char *name,
1688     const char *dest) {
1689
1690     AvahiRecord *r;
1691     int ret;
1692
1693     assert(s);
1694     assert(dest);
1695
1696     if ((name && !avahi_is_valid_domain_name(name)) || !avahi_is_valid_domain_name(dest))
1697         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
1698
1699     if (!name)
1700         name = s->host_name_fqdn;
1701
1702     if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_PTR, ttl)))
1703         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1704
1705     r->data.ptr.name = avahi_normalize_name_strdup(dest);
1706     ret = avahi_server_add(s, g, interface, protocol, flags, r);
1707     avahi_record_unref(r);
1708     return ret;
1709 }
1710
1711 int avahi_server_add_address(
1712     AvahiServer *s,
1713     AvahiSEntryGroup *g,
1714     AvahiIfIndex interface,
1715     AvahiProtocol protocol,
1716     AvahiPublishFlags flags,
1717     const char *name,
1718     AvahiAddress *a) {
1719
1720     char *n = NULL;
1721     int ret = AVAHI_OK;
1722
1723     assert(s);
1724     assert(a);
1725
1726     if (!AVAHI_IF_VALID(interface))
1727         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
1728
1729     if (!AVAHI_PROTO_VALID(protocol) || !AVAHI_PROTO_VALID(a->proto))
1730         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
1731
1732     if (!AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_REVERSE|AVAHI_PUBLISH_NO_ANNOUNCE|AVAHI_PUBLISH_NO_PROBE))
1733         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
1734
1735     if (name && !avahi_is_valid_domain_name(name))
1736         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
1737
1738     if (!name)
1739         name = s->host_name_fqdn;
1740     else {
1741         if (!(n = avahi_normalize_name_strdup(name)))
1742             return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1743
1744         name = n;
1745     }
1746
1747     if (a->proto == AVAHI_PROTO_INET) {
1748         AvahiRecord  *r;
1749
1750         if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_A, AVAHI_DEFAULT_TTL_HOST_NAME))) {
1751             ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1752             goto fail;
1753         }
1754
1755         r->data.a.address = a->data.ipv4;
1756         ret = avahi_server_add(s, g, interface, protocol, (flags & ~ AVAHI_PUBLISH_NO_REVERSE) | AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
1757         avahi_record_unref(r);
1758
1759         if (ret < 0)
1760             goto fail;
1761
1762         if (!(flags & AVAHI_PUBLISH_NO_REVERSE)) {
1763             char *reverse;
1764
1765             if (!(reverse = avahi_reverse_lookup_name_ipv4(&a->data.ipv4))) {
1766                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1767                 goto fail;
1768             }
1769
1770             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
1771             avahi_free(reverse);
1772         }
1773
1774     } else {
1775         AvahiRecord *r;
1776
1777         assert(a->proto == AVAHI_PROTO_INET6);
1778
1779         if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_AAAA, AVAHI_DEFAULT_TTL_HOST_NAME))) {
1780             ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1781             goto fail;
1782         }
1783
1784         r->data.aaaa.address = a->data.ipv6;
1785         ret = avahi_server_add(s, g, interface, protocol, (flags & ~ AVAHI_PUBLISH_NO_REVERSE) | AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
1786         avahi_record_unref(r);
1787
1788         if (ret < 0)
1789             goto fail;
1790
1791         if (!(flags & AVAHI_PUBLISH_NO_REVERSE)) {
1792             char *reverse;
1793
1794             if (!(reverse = avahi_reverse_lookup_name_ipv6_arpa(&a->data.ipv6))) {
1795                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1796                 goto fail;
1797             }
1798
1799             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
1800             avahi_free(reverse);
1801
1802             if (ret < 0)
1803                 goto fail;
1804
1805             if (!(reverse = avahi_reverse_lookup_name_ipv6_int(&a->data.ipv6))) {
1806                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1807                 goto fail;
1808             }
1809
1810             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
1811             avahi_free(reverse);
1812         }
1813     }
1814
1815 fail:
1816
1817     avahi_free(n);
1818
1819     return ret;
1820 }
1821
1822 static int server_add_txt_strlst_nocopy(
1823     AvahiServer *s,
1824     AvahiSEntryGroup *g,
1825     AvahiIfIndex interface,
1826     AvahiProtocol protocol,
1827     AvahiPublishFlags flags,
1828     uint32_t ttl,
1829     const char *name,
1830     AvahiStringList *strlst) {
1831
1832     AvahiRecord *r;
1833     int ret;
1834
1835     assert(s);
1836
1837     if (!(r = avahi_record_new_full(name ? name : s->host_name_fqdn, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_TXT, ttl))) {
1838         avahi_string_list_free(strlst);
1839         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1840     }
1841
1842     r->data.txt.string_list = strlst;
1843     ret = avahi_server_add(s, g, interface, protocol, flags, r);
1844     avahi_record_unref(r);
1845
1846     return ret;
1847 }
1848
1849 int avahi_server_add_txt_strlst(
1850     AvahiServer *s,
1851     AvahiSEntryGroup *g,
1852     AvahiIfIndex interface,
1853     AvahiProtocol protocol,
1854     AvahiPublishFlags flags,
1855     uint32_t ttl,
1856     const char *name,
1857     AvahiStringList *strlst) {
1858
1859     assert(s);
1860
1861     return server_add_txt_strlst_nocopy(s, g, interface, protocol, flags, ttl, name, avahi_string_list_copy(strlst));
1862 }
1863
1864 int avahi_server_add_txt_va(
1865     AvahiServer *s,
1866     AvahiSEntryGroup *g,
1867     AvahiIfIndex interface,
1868     AvahiProtocol protocol,
1869     AvahiPublishFlags flags,
1870     uint32_t ttl,
1871     const char *name,
1872     va_list va) {
1873
1874     assert(s);
1875
1876     return server_add_txt_strlst_nocopy(s, g, interface, protocol, flags, ttl, name, avahi_string_list_new_va(va));
1877 }
1878
1879 int avahi_server_add_txt(
1880     AvahiServer *s,
1881     AvahiSEntryGroup *g,
1882     AvahiIfIndex interface,
1883     AvahiProtocol protocol,
1884     AvahiPublishFlags flags,
1885     uint32_t ttl,
1886     const char *name,
1887     ...) {
1888
1889     va_list va;
1890     int ret;
1891
1892     assert(s);
1893
1894     va_start(va, name);
1895     ret = avahi_server_add_txt_va(s, g, interface, protocol, flags, ttl, name, va);
1896     va_end(va);
1897
1898     return ret;
1899 }
1900
1901 static AvahiStringList *add_magic_cookie(
1902     AvahiServer *s,
1903     AvahiStringList *strlst) {
1904
1905     assert(s);
1906
1907     if (!s->config.add_service_cookie)
1908         return strlst;
1909
1910     if (avahi_string_list_find(strlst, AVAHI_SERVICE_COOKIE))
1911         /* This string list already contains a magic cookie */
1912         return strlst;
1913
1914     return avahi_string_list_add_printf(strlst, AVAHI_SERVICE_COOKIE"=%u", s->local_service_cookie);
1915 }
1916
1917 static int server_add_service_strlst_nocopy(
1918     AvahiServer *s,
1919     AvahiSEntryGroup *g,
1920     AvahiIfIndex interface,
1921     AvahiProtocol protocol,
1922     AvahiPublishFlags flags,
1923     const char *name,
1924     const char *type,
1925     const char *domain,
1926     const char *host,
1927     uint16_t port,
1928     AvahiStringList *strlst) {
1929
1930     char ptr_name[AVAHI_DOMAIN_NAME_MAX], svc_name[AVAHI_DOMAIN_NAME_MAX], enum_ptr[AVAHI_DOMAIN_NAME_MAX], *h = NULL;
1931     AvahiRecord *r = NULL;
1932     int ret = AVAHI_OK;
1933
1934     assert(s);
1935     assert(type);
1936     assert(name);
1937
1938     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
1939     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
1940     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_COOKIE|AVAHI_PUBLISH_IS_PROXY), AVAHI_ERR_INVALID_FLAGS);
1941     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
1942     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
1943     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
1944     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !host || avahi_is_valid_domain_name(host), AVAHI_ERR_INVALID_HOST_NAME);
1945
1946     if (!domain)
1947         domain = s->domain_name;
1948
1949     if (!host)
1950         host = s->host_name_fqdn;
1951
1952     if (!(h = avahi_normalize_name_strdup(host))) {
1953         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1954         goto fail;
1955     }
1956
1957     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0 ||
1958         (ret = avahi_service_name_join(ptr_name, sizeof(ptr_name), NULL, type, domain)) < 0 ||
1959         (ret = avahi_service_name_join(enum_ptr, sizeof(enum_ptr), NULL, "_services._dns-sd._udp", domain)) < 0) {
1960         avahi_server_set_errno(s, ret);
1961         goto fail;
1962     }
1963
1964     /* Add service enumeration PTR record */
1965
1966     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, flags & AVAHI_PUBLISH_IS_PROXY, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
1967         goto fail;
1968
1969     /* Add SRV record */
1970
1971     if (!(r = avahi_record_new_full(svc_name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
1972         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1973         goto fail;
1974     }
1975
1976     r->data.srv.priority = 0;
1977     r->data.srv.weight = 0;
1978     r->data.srv.port = port;
1979     r->data.srv.name = h;
1980     h = NULL;
1981     ret = avahi_server_add(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE, r);
1982     avahi_record_unref(r);
1983
1984     if (ret < 0)
1985         goto fail;
1986
1987     /* Add TXT record */
1988
1989     if (!(flags & AVAHI_PUBLISH_NO_COOKIE))
1990         strlst = add_magic_cookie(s, strlst);
1991
1992     ret = server_add_txt_strlst_nocopy(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL, svc_name, strlst);
1993     strlst = NULL;
1994
1995     if (ret < 0)
1996         goto fail;
1997
1998     /* Add service type enumeration record */
1999
2000     ret = avahi_server_add_ptr(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY), AVAHI_DEFAULT_TTL, enum_ptr, ptr_name);
2001
2002 fail:
2003
2004     avahi_string_list_free(strlst);
2005     avahi_free(h);
2006
2007     return ret;
2008 }
2009
2010 int avahi_server_add_service_strlst(
2011     AvahiServer *s,
2012     AvahiSEntryGroup *g,
2013     AvahiIfIndex interface,
2014     AvahiProtocol protocol,
2015     AvahiPublishFlags flags,
2016     const char *name,
2017     const char *type,
2018     const char *domain,
2019     const char *host,
2020     uint16_t port,
2021     AvahiStringList *strlst) {
2022
2023     assert(s);
2024     assert(type);
2025     assert(name);
2026
2027     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_copy(strlst));
2028 }
2029
2030 int avahi_server_add_service_va(
2031     AvahiServer *s,
2032     AvahiSEntryGroup *g,
2033     AvahiIfIndex interface,
2034     AvahiProtocol protocol,
2035     AvahiPublishFlags flags,
2036     const char *name,
2037     const char *type,
2038     const char *domain,
2039     const char *host,
2040     uint16_t port,
2041     va_list va) {
2042
2043     assert(s);
2044     assert(type);
2045     assert(name);
2046
2047     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_new_va(va));
2048 }
2049
2050 int avahi_server_add_service(
2051     AvahiServer *s,
2052     AvahiSEntryGroup *g,
2053     AvahiIfIndex interface,
2054     AvahiProtocol protocol,
2055     AvahiPublishFlags flags,
2056     const char *name,
2057     const char *type,
2058     const char *domain,
2059     const char *host,
2060     uint16_t port,
2061     ... ){
2062
2063     va_list va;
2064     int ret;
2065
2066     assert(s);
2067     assert(type);
2068     assert(name);
2069
2070     va_start(va, port);
2071     ret = avahi_server_add_service_va(s, g, interface, protocol, flags, name, type, domain, host, port, va);
2072     va_end(va);
2073
2074     return ret;
2075 }
2076
2077 int avahi_server_add_service_subtype(
2078     AvahiServer *s,
2079     AvahiSEntryGroup *g,
2080     AvahiIfIndex interface,
2081     AvahiProtocol protocol,
2082     AvahiPublishFlags flags,
2083     const char *name,
2084     const char *type,
2085     const char *domain,
2086     const char *subtype) {
2087
2088     int ret = AVAHI_OK;
2089     char svc_name[AVAHI_DOMAIN_NAME_MAX], ptr_name[AVAHI_DOMAIN_NAME_MAX];
2090
2091     assert(name);
2092     assert(type);
2093     assert(subtype);
2094
2095     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
2096     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
2097     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_COOKIE|AVAHI_PUBLISH_IS_PROXY), AVAHI_ERR_INVALID_FLAGS);
2098     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
2099     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
2100     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
2101     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_subtype(subtype), AVAHI_ERR_INVALID_SERVICE_SUBTYPE);
2102
2103     if (!domain)
2104         domain = s->domain_name;
2105
2106     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0 ||
2107         (ret = avahi_service_name_join(ptr_name, sizeof(ptr_name), NULL, subtype, domain)) < 0) {
2108         avahi_server_set_errno(s, ret);
2109         goto fail;
2110     }
2111
2112     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, flags & AVAHI_PUBLISH_IS_PROXY, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
2113         goto fail;
2114
2115 fail:
2116
2117     return ret;
2118 }
2119
2120 static void hexstring(char *s, size_t sl, const void *p, size_t pl) {
2121     static const char hex[] = "0123456789abcdef";
2122     int b = 0;
2123     const uint8_t *k = p;
2124
2125     while (sl > 1 && pl > 0) {
2126         *(s++) = hex[(b ? *k : *k >> 4) & 0xF];
2127
2128         if (b) {
2129             k++;
2130             pl--;
2131         }
2132
2133         b = !b;
2134
2135         sl--;
2136     }
2137
2138     if (sl > 0)
2139         *s = 0;
2140 }
2141
2142 int avahi_server_add_dns_server_address(
2143     AvahiServer *s,
2144     AvahiSEntryGroup *g,
2145     AvahiIfIndex interface,
2146     AvahiProtocol protocol,
2147     AvahiPublishFlags flags,
2148     const char *domain,
2149     AvahiDNSServerType type,
2150     const AvahiAddress *address,
2151     uint16_t port /** should be 53 */) {
2152
2153     AvahiRecord *r;
2154     int ret;
2155     char n[64], h[64];
2156
2157     assert(s);
2158     assert(address);
2159
2160     if (!AVAHI_IF_VALID(interface))
2161         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
2162
2163     if (!AVAHI_PROTO_VALID(protocol) || !AVAHI_PROTO_VALID(address->proto))
2164         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
2165
2166     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
2167         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
2168
2169     if (port == 0)
2170         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
2171
2172     if (domain && !avahi_is_valid_domain_name(domain))
2173         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
2174
2175     if (!domain)
2176         domain = s->domain_name;
2177
2178     if (address->proto == AVAHI_PROTO_INET) {
2179         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv4Address));
2180         snprintf(n, sizeof(n), "ip-%s.%s", h, domain);
2181         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_A, AVAHI_DEFAULT_TTL_HOST_NAME);
2182         r->data.a.address = address->data.ipv4;
2183     } else {
2184         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv6Address));
2185         snprintf(n, sizeof(n), "ip6-%s.%s", h, domain);
2186         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_AAAA, AVAHI_DEFAULT_TTL_HOST_NAME);
2187         r->data.aaaa.address = address->data.ipv6;
2188     }
2189
2190     if (!r)
2191         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2192
2193     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
2194     avahi_record_unref(r);
2195
2196     if (ret < 0)
2197         return ret;
2198
2199     return avahi_server_add_dns_server_name(s, g, interface, protocol, flags, domain, type, n, port);
2200 }
2201
2202 int avahi_server_add_dns_server_name(
2203     AvahiServer *s,
2204     AvahiSEntryGroup *g,
2205     AvahiIfIndex interface,
2206     AvahiProtocol protocol,
2207     AvahiPublishFlags flags,
2208     const char *domain,
2209     AvahiDNSServerType type,
2210     const char *name,
2211     uint16_t port /** should be 53 */) {
2212
2213     int ret = -1;
2214     char t[256], *d = NULL, *n = NULL;
2215     AvahiRecord *r;
2216
2217     assert(s);
2218     assert(name);
2219
2220     if (!AVAHI_IF_VALID(interface))
2221         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
2222
2223     if (!AVAHI_PROTO_VALID(protocol))
2224         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
2225
2226     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
2227         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
2228
2229     if (port == 0)
2230         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
2231
2232     if (!avahi_is_valid_domain_name(name))
2233         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
2234
2235     if (domain && !avahi_is_valid_domain_name(domain))
2236         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
2237
2238     if (!domain)
2239         domain = s->domain_name;
2240
2241     if (!(n = avahi_normalize_name_strdup(name)) ||
2242         !(d = avahi_normalize_name_strdup(domain))) {
2243         avahi_free(n);
2244         avahi_free(d);
2245         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2246     }
2247
2248     snprintf(t, sizeof(t), "%s.%s", type == AVAHI_DNS_SERVER_RESOLVE ? "_domain._udp" : "_dns-update._udp", d);
2249     avahi_free(d);
2250
2251     if (!(r = avahi_record_new_full(t, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
2252         avahi_free(n);
2253         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2254     }
2255
2256     r->data.srv.priority = 0;
2257     r->data.srv.weight = 0;
2258     r->data.srv.port = port;
2259     r->data.srv.name = n;
2260     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_NULL, r);
2261     avahi_record_unref(r);
2262
2263     return ret;
2264 }
2265
2266 static void post_query_callback(AvahiInterfaceMonitor *m, AvahiInterface *i, void* userdata) {
2267     AvahiKey *k = userdata;
2268
2269     assert(m);
2270     assert(i);
2271     assert(k);
2272
2273     avahi_interface_post_query(i, k, 0);
2274 }
2275
2276 void avahi_server_post_query(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, AvahiKey *key) {
2277     assert(s);
2278     assert(key);
2279
2280     avahi_interface_monitor_walk(s->monitor, interface, protocol, post_query_callback, key);
2281 }
2282
2283 void avahi_s_entry_group_change_state(AvahiSEntryGroup *g, AvahiEntryGroupState state) {
2284     assert(g);
2285
2286     if (g->state == state)
2287         return;
2288
2289     assert(state <= AVAHI_ENTRY_GROUP_COLLISION);
2290
2291     if (g->state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
2292
2293         /* If the entry group was established for a time longer then
2294          * 5s, reset the establishment trial counter */
2295
2296         if (avahi_age(&g->established_at) > 5000000)
2297             g->n_register_try = 0;
2298     }
2299
2300     if (state == AVAHI_ENTRY_GROUP_ESTABLISHED)
2301
2302         /* If the entry group is now established, remember the time
2303          * this happened */
2304
2305         gettimeofday(&g->established_at, NULL);
2306
2307     g->state = state;
2308
2309     if (g->callback)
2310         g->callback(g->server, g, state, g->userdata);
2311 }
2312
2313 AvahiSEntryGroup *avahi_s_entry_group_new(AvahiServer *s, AvahiSEntryGroupCallback callback, void* userdata) {
2314     AvahiSEntryGroup *g;
2315
2316     assert(s);
2317
2318     if (!(g = avahi_new(AvahiSEntryGroup, 1))) {
2319         avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2320         return NULL;
2321     }
2322
2323     g->server = s;
2324     g->callback = callback;
2325     g->userdata = userdata;
2326     g->dead = 0;
2327     g->state = AVAHI_ENTRY_GROUP_UNCOMMITED;
2328     g->n_probing = 0;
2329     g->n_register_try = 0;
2330     g->register_time_event = NULL;
2331     g->register_time.tv_sec = 0;
2332     g->register_time.tv_usec = 0;
2333     AVAHI_LLIST_HEAD_INIT(AvahiEntry, g->entries);
2334
2335     AVAHI_LLIST_PREPEND(AvahiSEntryGroup, groups, s->groups, g);
2336     return g;
2337 }
2338
2339 void avahi_s_entry_group_free(AvahiSEntryGroup *g) {
2340     AvahiEntry *e;
2341
2342     assert(g);
2343     assert(g->server);
2344
2345     for (e = g->entries; e; e = e->by_group_next) {
2346         if (!e->dead) {
2347             avahi_goodbye_entry(g->server, e, 1);
2348             e->dead = 1;
2349         }
2350     }
2351
2352     if (g->register_time_event) {
2353         avahi_time_event_free(g->register_time_event);
2354         g->register_time_event = NULL;
2355     }
2356
2357     g->dead = 1;
2358
2359     g->server->need_group_cleanup = 1;
2360     g->server->need_entry_cleanup = 1;
2361 }
2362
2363 static void entry_group_commit_real(AvahiSEntryGroup *g) {
2364     assert(g);
2365
2366     gettimeofday(&g->register_time, NULL);
2367
2368     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
2369
2370     if (!g->dead) {
2371         avahi_announce_group(g->server, g);
2372         avahi_s_entry_group_check_probed(g, 0);
2373     }
2374 }
2375
2376 static void entry_group_register_time_event_callback(AvahiTimeEvent *e, void* userdata) {
2377     AvahiSEntryGroup *g = userdata;
2378     assert(g);
2379
2380 /*     avahi_log_debug("Holdoff passed, waking up and going on."); */
2381
2382     avahi_time_event_free(g->register_time_event);
2383     g->register_time_event = NULL;
2384
2385     /* Holdoff time passed, so let's start probing */
2386     entry_group_commit_real(g);
2387 }
2388
2389 int avahi_s_entry_group_commit(AvahiSEntryGroup *g) {
2390     struct timeval now;
2391
2392     assert(g);
2393     assert(!g->dead);
2394
2395     if (g->state != AVAHI_ENTRY_GROUP_UNCOMMITED && g->state != AVAHI_ENTRY_GROUP_COLLISION)
2396         return avahi_server_set_errno(g->server, AVAHI_ERR_BAD_STATE);
2397
2398     g->n_register_try++;
2399
2400     avahi_timeval_add(&g->register_time,
2401                       1000*(g->n_register_try >= AVAHI_RR_RATE_LIMIT_COUNT ?
2402                             AVAHI_RR_HOLDOFF_MSEC_RATE_LIMIT :
2403                             AVAHI_RR_HOLDOFF_MSEC));
2404
2405     gettimeofday(&now, NULL);
2406
2407     if (avahi_timeval_compare(&g->register_time, &now) <= 0) {
2408         /* Holdoff time passed, so let's start probing */
2409 /*         avahi_log_debug("Holdoff passed, directly going on.");  */
2410
2411         entry_group_commit_real(g);
2412     } else {
2413 /*          avahi_log_debug("Holdoff not passed, sleeping.");  */
2414
2415          /* Holdoff time has not yet passed, so let's wait */
2416         assert(!g->register_time_event);
2417         g->register_time_event = avahi_time_event_new(g->server->time_event_queue, &g->register_time, entry_group_register_time_event_callback, g);
2418
2419         avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
2420     }
2421
2422     return AVAHI_OK;
2423 }
2424
2425 void avahi_s_entry_group_reset(AvahiSEntryGroup *g) {
2426     AvahiEntry *e;
2427     assert(g);
2428
2429     for (e = g->entries; e; e = e->by_group_next) {
2430         if (!e->dead) {
2431             avahi_goodbye_entry(g->server, e, 1);
2432             e->dead = 1;
2433         }
2434     }
2435     g->server->need_entry_cleanup = 1;
2436
2437     if (g->register_time_event) {
2438         avahi_time_event_free(g->register_time_event);
2439         g->register_time_event = NULL;
2440     }
2441
2442     g->n_probing = 0;
2443
2444     gettimeofday(&g->register_time, NULL);
2445
2446     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_UNCOMMITED);
2447 }
2448
2449 int avahi_entry_is_commited(AvahiEntry *e) {
2450     assert(e);
2451     assert(!e->dead);
2452
2453     return !e->group ||
2454         e->group->state == AVAHI_ENTRY_GROUP_REGISTERING ||
2455         e->group->state == AVAHI_ENTRY_GROUP_ESTABLISHED;
2456 }
2457
2458 AvahiEntryGroupState avahi_s_entry_group_get_state(AvahiSEntryGroup *g) {
2459     assert(g);
2460     assert(!g->dead);
2461
2462     return g->state;
2463 }
2464
2465 void avahi_s_entry_group_set_data(AvahiSEntryGroup *g, void* userdata) {
2466     assert(g);
2467
2468     g->userdata = userdata;
2469 }
2470
2471 void* avahi_s_entry_group_get_data(AvahiSEntryGroup *g) {
2472     assert(g);
2473
2474     return g->userdata;
2475 }
2476
2477 int avahi_s_entry_group_is_empty(AvahiSEntryGroup *g) {
2478     AvahiEntry *e;
2479     assert(g);
2480
2481     /* Look for an entry that is not dead */
2482     for (e = g->entries; e; e = e->by_group_next)
2483         if (!e->dead)
2484             return 0;
2485
2486     return 1;
2487 }
2488
2489 const char* avahi_server_get_domain_name(AvahiServer *s) {
2490     assert(s);
2491
2492     return s->domain_name;
2493 }
2494
2495 const char* avahi_server_get_host_name(AvahiServer *s) {
2496     assert(s);
2497
2498     return s->host_name;
2499 }
2500
2501 const char* avahi_server_get_host_name_fqdn(AvahiServer *s) {
2502     assert(s);
2503
2504     return s->host_name_fqdn;
2505 }
2506
2507 void* avahi_server_get_data(AvahiServer *s) {
2508     assert(s);
2509
2510     return s->userdata;
2511 }
2512
2513 void avahi_server_set_data(AvahiServer *s, void* userdata) {
2514     assert(s);
2515
2516     s->userdata = userdata;
2517 }
2518
2519 AvahiServerState avahi_server_get_state(AvahiServer *s) {
2520     assert(s);
2521
2522     return s->state;
2523 }
2524
2525 AvahiServerConfig* avahi_server_config_init(AvahiServerConfig *c) {
2526     assert(c);
2527
2528     memset(c, 0, sizeof(AvahiServerConfig));
2529     c->use_ipv6 = 1;
2530     c->use_ipv4 = 1;
2531     c->host_name = NULL;
2532     c->domain_name = NULL;
2533     c->check_response_ttl = 0;
2534     c->publish_hinfo = 1;
2535     c->publish_addresses = 1;
2536     c->publish_workstation = 1;
2537     c->publish_domain = 1;
2538     c->use_iff_running = 0;
2539     c->enable_reflector = 0;
2540     c->reflect_ipv = 0;
2541     c->add_service_cookie = 1;
2542     c->enable_wide_area = 0;
2543     c->n_wide_area_servers = 0;
2544     c->disallow_other_stacks = 0;
2545
2546     return c;
2547 }
2548
2549 void avahi_server_config_free(AvahiServerConfig *c) {
2550     assert(c);
2551
2552     avahi_free(c->host_name);
2553     avahi_free(c->domain_name);
2554 }
2555
2556 AvahiServerConfig* avahi_server_config_copy(AvahiServerConfig *ret, const AvahiServerConfig *c) {
2557     char *d = NULL, *h = NULL;
2558     assert(ret);
2559     assert(c);
2560
2561     if (c->host_name)
2562         if (!(h = avahi_strdup(c->host_name)))
2563             return NULL;
2564
2565     if (c->domain_name)
2566         if (!(d = avahi_strdup(c->domain_name))) {
2567             avahi_free(h);
2568             return NULL;
2569         }
2570
2571     *ret = *c;
2572     ret->host_name = h;
2573     ret->domain_name = d;
2574
2575     return ret;
2576 }
2577
2578 int avahi_server_errno(AvahiServer *s) {
2579     assert(s);
2580
2581     return s->error;
2582 }
2583
2584 /* Just for internal use */
2585 int avahi_server_set_errno(AvahiServer *s, int error) {
2586     assert(s);
2587
2588     return s->error = error;
2589 }
2590
2591 uint32_t avahi_server_get_local_service_cookie(AvahiServer *s) {
2592     assert(s);
2593
2594     return s->local_service_cookie;
2595 }
2596
2597 int avahi_server_is_service_local(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char*domain) {
2598     AvahiKey *key = NULL;
2599     char n[256];
2600     int ret;
2601     AvahiEntry *e;
2602
2603     assert(s);
2604     assert(name);
2605     assert(type);
2606     assert(domain);
2607
2608     if (!avahi_is_valid_service_name(name))
2609         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_SERVICE_NAME);
2610
2611     if (!avahi_is_valid_service_type_strict(type))
2612         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_SERVICE_TYPE);
2613
2614     if (domain && !avahi_is_valid_domain_name(domain))
2615         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
2616
2617     if ((ret = avahi_service_name_join(n, sizeof(n), name, type, domain) < 0))
2618         return avahi_server_set_errno(s, ret);
2619
2620     if (!(key = avahi_key_new(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV)))
2621         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2622
2623     ret = 0;
2624
2625     for (e = avahi_hashmap_lookup(s->entries_by_key, key); e; e = e->by_key_next) {
2626
2627         if ((e->interface == interface || e->interface <= 0 || interface <= 0) &&
2628             (e->protocol == protocol || e->protocol == AVAHI_PROTO_UNSPEC || protocol == AVAHI_PROTO_UNSPEC) &&
2629             !(e->flags & AVAHI_PUBLISH_IS_PROXY)) {
2630             ret = 1;
2631             break;
2632         }
2633     }
2634
2635     avahi_key_unref(key);
2636
2637     return ret;
2638 }
2639
2640 /** Set the wide area DNS servers */
2641 int avahi_server_set_wide_area_servers(AvahiServer *s, const AvahiAddress *a, unsigned n) {
2642     assert(s);
2643
2644     if (!s->wide_area_lookup_engine)
2645         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_CONFIG);
2646
2647     avahi_wide_area_set_servers(s->wide_area_lookup_engine, a, n);
2648     return AVAHI_OK;
2649 }