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(host_name) : avahi_get_host_name();
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(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(s->config.host_name) : avahi_get_host_name();
1435     s->host_name[strcspn(s->host_name, ".")] = 0;
1436     s->domain_name = s->config.domain_name ? avahi_normalize_name(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(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(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         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1839
1840     r->data.txt.string_list = strlst;
1841     ret = avahi_server_add(s, g, interface, protocol, flags, r);
1842     avahi_record_unref(r);
1843
1844     return ret;
1845 }
1846
1847 int avahi_server_add_txt_strlst(
1848     AvahiServer *s,
1849     AvahiSEntryGroup *g,
1850     AvahiIfIndex interface,
1851     AvahiProtocol protocol,
1852     AvahiPublishFlags flags,
1853     uint32_t ttl,
1854     const char *name,
1855     AvahiStringList *strlst) {
1856
1857     assert(s);
1858
1859     return server_add_txt_strlst_nocopy(s, g, interface, protocol, flags, ttl, name, avahi_string_list_copy(strlst));
1860 }
1861
1862 int avahi_server_add_txt_va(
1863     AvahiServer *s,
1864     AvahiSEntryGroup *g,
1865     AvahiIfIndex interface,
1866     AvahiProtocol protocol,
1867     AvahiPublishFlags flags,
1868     uint32_t ttl,
1869     const char *name,
1870     va_list va) {
1871
1872     assert(s);
1873
1874     return server_add_txt_strlst_nocopy(s, g, interface, protocol, flags, ttl, name, avahi_string_list_new_va(va));
1875 }
1876
1877 int avahi_server_add_txt(
1878     AvahiServer *s,
1879     AvahiSEntryGroup *g,
1880     AvahiIfIndex interface,
1881     AvahiProtocol protocol,
1882     AvahiPublishFlags flags,
1883     uint32_t ttl,
1884     const char *name,
1885     ...) {
1886
1887     va_list va;
1888     int ret;
1889
1890     assert(s);
1891
1892     va_start(va, name);
1893     ret = avahi_server_add_txt_va(s, g, interface, protocol, flags, ttl, name, va);
1894     va_end(va);
1895
1896     return ret;
1897 }
1898
1899 static void escape_service_name(char *d, size_t size, const char *s) {
1900     assert(d);
1901     assert(size);
1902     assert(s);
1903
1904     while (*s && size >= 2) {
1905         if (*s == '.' || *s == '\\') {
1906             if (size < 3)
1907                 break;
1908
1909             *(d++) = '\\';
1910             size--;
1911         }
1912
1913         *(d++) = *(s++);
1914         size--;
1915     }
1916
1917     assert(size > 0);
1918     *(d++) = 0;
1919 }
1920
1921 static AvahiStringList *add_magic_cookie(
1922     AvahiServer *s,
1923     AvahiStringList *strlst) {
1924
1925     assert(s);
1926
1927     if (!s->config.add_service_cookie)
1928         return strlst;
1929
1930     if (avahi_string_list_find(strlst, AVAHI_SERVICE_COOKIE))
1931         /* This string list already contains a magic cookie */
1932         return strlst;
1933
1934     return avahi_string_list_add_printf(strlst, AVAHI_SERVICE_COOKIE"=%u", s->local_service_cookie);
1935 }
1936
1937 static int server_add_service_strlst_nocopy(
1938     AvahiServer *s,
1939     AvahiSEntryGroup *g,
1940     AvahiIfIndex interface,
1941     AvahiProtocol protocol,
1942     AvahiPublishFlags flags,
1943     const char *name,
1944     const char *type,
1945     const char *domain,
1946     const char *host,
1947     uint16_t port,
1948     AvahiStringList *strlst) {
1949
1950     char ptr_name[256], svc_name[256], ename[64], enum_ptr[256];
1951     char *t = NULL, *d = NULL, *h = NULL;
1952     AvahiRecord *r = NULL;
1953     int ret = AVAHI_OK;
1954
1955     assert(s);
1956     assert(type);
1957     assert(name);
1958
1959     if (!AVAHI_IF_VALID(interface))
1960         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
1961
1962     if (!AVAHI_PROTO_VALID(protocol))
1963         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
1964
1965     if (!AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_COOKIE|AVAHI_PUBLISH_IS_PROXY))
1966         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
1967
1968     if (!avahi_is_valid_service_name(name))
1969         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_SERVICE_NAME);
1970
1971     if (!avahi_is_valid_service_type(type))
1972         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_SERVICE_TYPE);
1973
1974     if (domain && !avahi_is_valid_domain_name(domain))
1975         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
1976
1977     if (host && !avahi_is_valid_domain_name(host))
1978         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
1979
1980     escape_service_name(ename, sizeof(ename), name);
1981
1982     if (!domain)
1983         domain = s->domain_name;
1984
1985     if (!host)
1986         host = s->host_name_fqdn;
1987
1988     if (!(d = avahi_normalize_name(domain)) ||
1989         !(t = avahi_normalize_name(type)) ||
1990         !(h = avahi_normalize_name(host))) {
1991         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1992         goto fail;
1993     }
1994
1995     snprintf(ptr_name, sizeof(ptr_name), "%s.%s", t, d);
1996     snprintf(svc_name, sizeof(svc_name), "%s.%s.%s", ename, t, d);
1997
1998     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, flags & AVAHI_PUBLISH_IS_PROXY, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
1999         goto fail;
2000
2001     if (!(r = avahi_record_new_full(svc_name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
2002         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2003         goto fail;
2004     }
2005
2006     r->data.srv.priority = 0;
2007     r->data.srv.weight = 0;
2008     r->data.srv.port = port;
2009     r->data.srv.name = h;
2010     h = NULL;
2011     ret = avahi_server_add(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE, r);
2012     avahi_record_unref(r);
2013
2014     if (ret < 0)
2015         goto fail;
2016
2017     if (!(flags & AVAHI_PUBLISH_NO_COOKIE))
2018         strlst = add_magic_cookie(s, strlst);
2019
2020     ret = server_add_txt_strlst_nocopy(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL, svc_name, strlst);
2021     strlst = NULL;
2022
2023     if (ret < 0)
2024         goto fail;
2025
2026     snprintf(enum_ptr, sizeof(enum_ptr), "_services._dns-sd._udp.%s", d);
2027     ret = avahi_server_add_ptr(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY), AVAHI_DEFAULT_TTL, enum_ptr, ptr_name);
2028
2029 fail:
2030
2031     avahi_free(d);
2032     avahi_free(t);
2033     avahi_free(h);
2034
2035     avahi_string_list_free(strlst);
2036
2037     return ret;
2038 }
2039
2040 int avahi_server_add_service_strlst(
2041     AvahiServer *s,
2042     AvahiSEntryGroup *g,
2043     AvahiIfIndex interface,
2044     AvahiProtocol protocol,
2045     AvahiPublishFlags flags,
2046     const char *name,
2047     const char *type,
2048     const char *domain,
2049     const char *host,
2050     uint16_t port,
2051     AvahiStringList *strlst) {
2052
2053     assert(s);
2054     assert(type);
2055     assert(name);
2056
2057     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_copy(strlst));
2058 }
2059
2060 int avahi_server_add_service_va(
2061     AvahiServer *s,
2062     AvahiSEntryGroup *g,
2063     AvahiIfIndex interface,
2064     AvahiProtocol protocol,
2065     AvahiPublishFlags flags,
2066     const char *name,
2067     const char *type,
2068     const char *domain,
2069     const char *host,
2070     uint16_t port,
2071     va_list va){
2072
2073     assert(s);
2074     assert(type);
2075     assert(name);
2076
2077     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_new_va(va));
2078 }
2079
2080 int avahi_server_add_service(
2081     AvahiServer *s,
2082     AvahiSEntryGroup *g,
2083     AvahiIfIndex interface,
2084     AvahiProtocol protocol,
2085     AvahiPublishFlags flags,
2086     const char *name,
2087     const char *type,
2088     const char *domain,
2089     const char *host,
2090     uint16_t port,
2091     ... ){
2092
2093     va_list va;
2094     int ret;
2095
2096     assert(s);
2097     assert(type);
2098     assert(name);
2099
2100     va_start(va, port);
2101     ret = avahi_server_add_service_va(s, g, interface, protocol, flags, name, type, domain, host, port, va);
2102     va_end(va);
2103     return ret;
2104 }
2105
2106 static void hexstring(char *s, size_t sl, const void *p, size_t pl) {
2107     static const char hex[] = "0123456789abcdef";
2108     int b = 0;
2109     const uint8_t *k = p;
2110
2111     while (sl > 1 && pl > 0) {
2112         *(s++) = hex[(b ? *k : *k >> 4) & 0xF];
2113
2114         if (b) {
2115             k++;
2116             pl--;
2117         }
2118
2119         b = !b;
2120
2121         sl--;
2122     }
2123
2124     if (sl > 0)
2125         *s = 0;
2126 }
2127
2128 int avahi_server_add_dns_server_address(
2129     AvahiServer *s,
2130     AvahiSEntryGroup *g,
2131     AvahiIfIndex interface,
2132     AvahiProtocol protocol,
2133     AvahiPublishFlags flags,
2134     const char *domain,
2135     AvahiDNSServerType type,
2136     const AvahiAddress *address,
2137     uint16_t port /** should be 53 */) {
2138
2139     AvahiRecord *r;
2140     int ret;
2141     char n[64], h[64];
2142
2143     assert(s);
2144     assert(address);
2145
2146     if (!AVAHI_IF_VALID(interface))
2147         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
2148
2149     if (!AVAHI_PROTO_VALID(protocol) || !AVAHI_PROTO_VALID(address->proto))
2150         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
2151
2152     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
2153         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
2154
2155     if (port == 0)
2156         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
2157
2158     if (domain && !avahi_is_valid_domain_name(domain))
2159         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
2160
2161     if (!domain)
2162         domain = s->domain_name;
2163
2164     if (address->proto == AVAHI_PROTO_INET) {
2165         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv4Address));
2166         snprintf(n, sizeof(n), "ip-%s.%s", h, domain);
2167         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_A, AVAHI_DEFAULT_TTL_HOST_NAME);
2168         r->data.a.address = address->data.ipv4;
2169     } else {
2170         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv6Address));
2171         snprintf(n, sizeof(n), "ip6-%s.%s", h, domain);
2172         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_AAAA, AVAHI_DEFAULT_TTL_HOST_NAME);
2173         r->data.aaaa.address = address->data.ipv6;
2174     }
2175
2176     if (!r)
2177         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2178
2179     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
2180     avahi_record_unref(r);
2181
2182     if (ret < 0)
2183         return ret;
2184
2185     return avahi_server_add_dns_server_name(s, g, interface, protocol, flags, domain, type, n, port);
2186 }
2187
2188 int avahi_server_add_dns_server_name(
2189     AvahiServer *s,
2190     AvahiSEntryGroup *g,
2191     AvahiIfIndex interface,
2192     AvahiProtocol protocol,
2193     AvahiPublishFlags flags,
2194     const char *domain,
2195     AvahiDNSServerType type,
2196     const char *name,
2197     uint16_t port /** should be 53 */) {
2198
2199     int ret = -1;
2200     char t[256], *d = NULL, *n = NULL;
2201     AvahiRecord *r;
2202
2203     assert(s);
2204     assert(name);
2205
2206     if (!AVAHI_IF_VALID(interface))
2207         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
2208
2209     if (!AVAHI_PROTO_VALID(protocol))
2210         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
2211
2212     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
2213         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
2214
2215     if (port == 0)
2216         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
2217
2218     if (!avahi_is_valid_domain_name(name))
2219         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
2220
2221     if (domain && !avahi_is_valid_domain_name(domain))
2222         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
2223
2224     if (!domain)
2225         domain = s->domain_name;
2226
2227     if (!(n = avahi_normalize_name(name)) ||
2228         !(d = avahi_normalize_name(domain))) {
2229         avahi_free(n);
2230         avahi_free(d);
2231         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2232     }
2233
2234     snprintf(t, sizeof(t), "%s.%s", type == AVAHI_DNS_SERVER_RESOLVE ? "_domain._udp" : "_dns-update._udp", d);
2235     avahi_free(d);
2236
2237     if (!(r = avahi_record_new_full(t, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
2238         avahi_free(n);
2239         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2240     }
2241
2242     r->data.srv.priority = 0;
2243     r->data.srv.weight = 0;
2244     r->data.srv.port = port;
2245     r->data.srv.name = n;
2246     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_NULL, r);
2247     avahi_record_unref(r);
2248
2249     return ret;
2250 }
2251
2252 static void post_query_callback(AvahiInterfaceMonitor *m, AvahiInterface *i, void* userdata) {
2253     AvahiKey *k = userdata;
2254
2255     assert(m);
2256     assert(i);
2257     assert(k);
2258
2259     avahi_interface_post_query(i, k, 0);
2260 }
2261
2262 void avahi_server_post_query(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, AvahiKey *key) {
2263     assert(s);
2264     assert(key);
2265
2266     avahi_interface_monitor_walk(s->monitor, interface, protocol, post_query_callback, key);
2267 }
2268
2269 void avahi_s_entry_group_change_state(AvahiSEntryGroup *g, AvahiEntryGroupState state) {
2270     assert(g);
2271
2272     if (g->state == state)
2273         return;
2274
2275     assert(state <= AVAHI_ENTRY_GROUP_COLLISION);
2276
2277     if (g->state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
2278
2279         /* If the entry group was established for a time longer then
2280          * 5s, reset the establishment trial counter */
2281
2282         if (avahi_age(&g->established_at) > 5000000)
2283             g->n_register_try = 0;
2284     }
2285
2286     if (state == AVAHI_ENTRY_GROUP_ESTABLISHED)
2287
2288         /* If the entry group is now established, remember the time
2289          * this happened */
2290
2291         gettimeofday(&g->established_at, NULL);
2292
2293     g->state = state;
2294
2295     if (g->callback)
2296         g->callback(g->server, g, state, g->userdata);
2297 }
2298
2299 AvahiSEntryGroup *avahi_s_entry_group_new(AvahiServer *s, AvahiSEntryGroupCallback callback, void* userdata) {
2300     AvahiSEntryGroup *g;
2301
2302     assert(s);
2303
2304     if (!(g = avahi_new(AvahiSEntryGroup, 1))) {
2305         avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2306         return NULL;
2307     }
2308
2309     g->server = s;
2310     g->callback = callback;
2311     g->userdata = userdata;
2312     g->dead = 0;
2313     g->state = AVAHI_ENTRY_GROUP_UNCOMMITED;
2314     g->n_probing = 0;
2315     g->n_register_try = 0;
2316     g->register_time_event = NULL;
2317     g->register_time.tv_sec = 0;
2318     g->register_time.tv_usec = 0;
2319     AVAHI_LLIST_HEAD_INIT(AvahiEntry, g->entries);
2320
2321     AVAHI_LLIST_PREPEND(AvahiSEntryGroup, groups, s->groups, g);
2322     return g;
2323 }
2324
2325 void avahi_s_entry_group_free(AvahiSEntryGroup *g) {
2326     AvahiEntry *e;
2327
2328     assert(g);
2329     assert(g->server);
2330
2331     for (e = g->entries; e; e = e->by_group_next) {
2332         if (!e->dead) {
2333             avahi_goodbye_entry(g->server, e, 1);
2334             e->dead = 1;
2335         }
2336     }
2337
2338     if (g->register_time_event) {
2339         avahi_time_event_free(g->register_time_event);
2340         g->register_time_event = NULL;
2341     }
2342
2343     g->dead = 1;
2344
2345     g->server->need_group_cleanup = 1;
2346     g->server->need_entry_cleanup = 1;
2347 }
2348
2349 static void entry_group_commit_real(AvahiSEntryGroup *g) {
2350     assert(g);
2351
2352     gettimeofday(&g->register_time, NULL);
2353
2354     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
2355
2356     if (!g->dead) {
2357         avahi_announce_group(g->server, g);
2358         avahi_s_entry_group_check_probed(g, 0);
2359     }
2360 }
2361
2362 static void entry_group_register_time_event_callback(AvahiTimeEvent *e, void* userdata) {
2363     AvahiSEntryGroup *g = userdata;
2364     assert(g);
2365
2366 /*     avahi_log_debug("Holdoff passed, waking up and going on."); */
2367
2368     avahi_time_event_free(g->register_time_event);
2369     g->register_time_event = NULL;
2370
2371     /* Holdoff time passed, so let's start probing */
2372     entry_group_commit_real(g);
2373 }
2374
2375 int avahi_s_entry_group_commit(AvahiSEntryGroup *g) {
2376     struct timeval now;
2377
2378     assert(g);
2379     assert(!g->dead);
2380
2381     if (g->state != AVAHI_ENTRY_GROUP_UNCOMMITED && g->state != AVAHI_ENTRY_GROUP_COLLISION)
2382         return avahi_server_set_errno(g->server, AVAHI_ERR_BAD_STATE);
2383
2384     g->n_register_try++;
2385
2386     avahi_timeval_add(&g->register_time,
2387                       1000*(g->n_register_try >= AVAHI_RR_RATE_LIMIT_COUNT ?
2388                             AVAHI_RR_HOLDOFF_MSEC_RATE_LIMIT :
2389                             AVAHI_RR_HOLDOFF_MSEC));
2390
2391     gettimeofday(&now, NULL);
2392
2393     if (avahi_timeval_compare(&g->register_time, &now) <= 0) {
2394         /* Holdoff time passed, so let's start probing */
2395 /*         avahi_log_debug("Holdoff passed, directly going on.");  */
2396
2397         entry_group_commit_real(g);
2398     } else {
2399 /*          avahi_log_debug("Holdoff not passed, sleeping.");  */
2400
2401          /* Holdoff time has not yet passed, so let's wait */
2402         assert(!g->register_time_event);
2403         g->register_time_event = avahi_time_event_new(g->server->time_event_queue, &g->register_time, entry_group_register_time_event_callback, g);
2404
2405         avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
2406     }
2407
2408     return AVAHI_OK;
2409 }
2410
2411 void avahi_s_entry_group_reset(AvahiSEntryGroup *g) {
2412     AvahiEntry *e;
2413     assert(g);
2414
2415     for (e = g->entries; e; e = e->by_group_next) {
2416         if (!e->dead) {
2417             avahi_goodbye_entry(g->server, e, 1);
2418             e->dead = 1;
2419         }
2420     }
2421     g->server->need_entry_cleanup = 1;
2422
2423     if (g->register_time_event) {
2424         avahi_time_event_free(g->register_time_event);
2425         g->register_time_event = NULL;
2426     }
2427
2428     g->n_probing = 0;
2429
2430     gettimeofday(&g->register_time, NULL);
2431
2432     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_UNCOMMITED);
2433 }
2434
2435 int avahi_entry_is_commited(AvahiEntry *e) {
2436     assert(e);
2437     assert(!e->dead);
2438
2439     return !e->group ||
2440         e->group->state == AVAHI_ENTRY_GROUP_REGISTERING ||
2441         e->group->state == AVAHI_ENTRY_GROUP_ESTABLISHED;
2442 }
2443
2444 AvahiEntryGroupState avahi_s_entry_group_get_state(AvahiSEntryGroup *g) {
2445     assert(g);
2446     assert(!g->dead);
2447
2448     return g->state;
2449 }
2450
2451 void avahi_s_entry_group_set_data(AvahiSEntryGroup *g, void* userdata) {
2452     assert(g);
2453
2454     g->userdata = userdata;
2455 }
2456
2457 void* avahi_s_entry_group_get_data(AvahiSEntryGroup *g) {
2458     assert(g);
2459
2460     return g->userdata;
2461 }
2462
2463 int avahi_s_entry_group_is_empty(AvahiSEntryGroup *g) {
2464     AvahiEntry *e;
2465     assert(g);
2466
2467     /* Look for an entry that is not dead */
2468     for (e = g->entries; e; e = e->by_group_next)
2469         if (!e->dead)
2470             return 0;
2471
2472     return 1;
2473 }
2474
2475 const char* avahi_server_get_domain_name(AvahiServer *s) {
2476     assert(s);
2477
2478     return s->domain_name;
2479 }
2480
2481 const char* avahi_server_get_host_name(AvahiServer *s) {
2482     assert(s);
2483
2484     return s->host_name;
2485 }
2486
2487 const char* avahi_server_get_host_name_fqdn(AvahiServer *s) {
2488     assert(s);
2489
2490     return s->host_name_fqdn;
2491 }
2492
2493 void* avahi_server_get_data(AvahiServer *s) {
2494     assert(s);
2495
2496     return s->userdata;
2497 }
2498
2499 void avahi_server_set_data(AvahiServer *s, void* userdata) {
2500     assert(s);
2501
2502     s->userdata = userdata;
2503 }
2504
2505 AvahiServerState avahi_server_get_state(AvahiServer *s) {
2506     assert(s);
2507
2508     return s->state;
2509 }
2510
2511 AvahiServerConfig* avahi_server_config_init(AvahiServerConfig *c) {
2512     assert(c);
2513
2514     memset(c, 0, sizeof(AvahiServerConfig));
2515     c->use_ipv6 = 1;
2516     c->use_ipv4 = 1;
2517     c->host_name = NULL;
2518     c->domain_name = NULL;
2519     c->check_response_ttl = 0;
2520     c->publish_hinfo = 1;
2521     c->publish_addresses = 1;
2522     c->publish_workstation = 1;
2523     c->publish_domain = 1;
2524     c->use_iff_running = 0;
2525     c->enable_reflector = 0;
2526     c->reflect_ipv = 0;
2527     c->add_service_cookie = 1;
2528     c->enable_wide_area = 0;
2529     c->n_wide_area_servers = 0;
2530     c->disallow_other_stacks = 0;
2531
2532     return c;
2533 }
2534
2535 void avahi_server_config_free(AvahiServerConfig *c) {
2536     assert(c);
2537
2538     avahi_free(c->host_name);
2539     avahi_free(c->domain_name);
2540 }
2541
2542 AvahiServerConfig* avahi_server_config_copy(AvahiServerConfig *ret, const AvahiServerConfig *c) {
2543     char *d = NULL, *h = NULL;
2544     assert(ret);
2545     assert(c);
2546
2547     if (c->host_name)
2548         if (!(h = avahi_strdup(c->host_name)))
2549             return NULL;
2550
2551     if (c->domain_name)
2552         if (!(d = avahi_strdup(c->domain_name))) {
2553             avahi_free(h);
2554             return NULL;
2555         }
2556
2557     *ret = *c;
2558     ret->host_name = h;
2559     ret->domain_name = d;
2560
2561     return ret;
2562 }
2563
2564 int avahi_server_errno(AvahiServer *s) {
2565     assert(s);
2566
2567     return s->error;
2568 }
2569
2570 /* Just for internal use */
2571 int avahi_server_set_errno(AvahiServer *s, int error) {
2572     assert(s);
2573
2574     return s->error = error;
2575 }
2576
2577 uint32_t avahi_server_get_local_service_cookie(AvahiServer *s) {
2578     assert(s);
2579
2580     return s->local_service_cookie;
2581 }
2582
2583 int avahi_server_is_service_local(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char*domain) {
2584     AvahiKey *key = NULL;
2585     char *d, *t;
2586     char ename[64], n[256];
2587     int ret;
2588     AvahiEntry *e;
2589
2590     assert(s);
2591     assert(name);
2592     assert(type);
2593     assert(domain);
2594
2595     if (!avahi_is_valid_service_name(name))
2596         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_SERVICE_NAME);
2597
2598     if (!avahi_is_valid_service_type(type))
2599         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_SERVICE_TYPE);
2600
2601     if (domain && !avahi_is_valid_domain_name(domain))
2602         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
2603
2604     escape_service_name(ename, sizeof(ename), name);
2605
2606     if (!(d = avahi_normalize_name(domain)) ||
2607         !(t = avahi_normalize_name(type))) {
2608         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2609         goto fail;
2610     }
2611     snprintf(n, sizeof(n), "%s.%s.%s", ename, t, d);
2612
2613     if (!(key = avahi_key_new(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV))) {
2614         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
2615         goto fail;
2616     }
2617
2618     ret = 0;
2619
2620     for (e = avahi_hashmap_lookup(s->entries_by_key, key); e; e = e->by_key_next) {
2621
2622         if ((e->interface == interface || e->interface <= 0 || interface <= 0) &&
2623             (e->protocol == protocol || e->protocol == AVAHI_PROTO_UNSPEC || protocol == AVAHI_PROTO_UNSPEC) &&
2624             !(e->flags & AVAHI_PUBLISH_IS_PROXY)) {
2625             ret = 1;
2626             break;
2627         }
2628     }
2629
2630     avahi_key_unref(key);
2631
2632     avahi_free(d);
2633     avahi_free(t);
2634
2635     return ret;
2636
2637 fail:
2638     if (d)
2639         avahi_free(d);
2640
2641     if (t)
2642         avahi_free(t);
2643
2644     if (key)
2645         avahi_key_unref(key);
2646
2647     return ret;
2648 }
2649
2650 /** Set the wide area DNS servers */
2651 int avahi_server_set_wide_area_servers(AvahiServer *s, const AvahiAddress *a, unsigned n) {
2652     assert(s);
2653
2654     if (!s->wide_area_lookup_engine)
2655         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_CONFIG);
2656
2657     avahi_wide_area_set_servers(s->wide_area_lookup_engine, a, n);
2658     return AVAHI_OK;
2659 }