]> git.meshlink.io Git - catta/blob - avahi-core/server.c
do not alllow setting the host name to the one that is currently used anyway. return...
[catta] / 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/types.h>
27 #include <sys/socket.h>
28 #include <netinet/in.h>
29 #include <arpa/inet.h>
30 #include <string.h>
31 #include <sys/utsname.h>
32 #include <unistd.h>
33 #include <errno.h>
34 #include <stdio.h>
35 #include <assert.h>
36 #include <stdlib.h>
37
38 #include <avahi-common/domain.h>
39 #include <avahi-common/timeval.h>
40 #include <avahi-common/malloc.h>
41 #include <avahi-common/error.h>
42
43 #include "internal.h"
44 #include "iface.h"
45 #include "socket.h"
46 #include "browse.h"
47 #include "log.h"
48 #include "util.h"
49 #include "dns-srv-rr.h"
50 #include "addr-util.h"
51 #include "domain-util.h"
52 #include "rr-util.h"
53
54 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) {
55     assert(s);
56     assert(i);
57     assert(name);
58     assert(callback);
59
60     if (type == AVAHI_DNS_TYPE_ANY) {
61         AvahiEntry *e;
62         
63         for (e = s->entries; e; e = e->entries_next)
64             if (!e->dead &&
65                 avahi_entry_is_registered(s, e, i) &&
66                 e->record->key->clazz == AVAHI_DNS_CLASS_IN &&
67                 avahi_domain_equal(name, e->record->key->name))
68                 callback(s, e->record, e->flags & AVAHI_PUBLISH_UNIQUE, userdata);
69
70     } else {
71         AvahiEntry *e;
72         AvahiKey *k;
73         
74         if (!(k = avahi_key_new(name, AVAHI_DNS_CLASS_IN, type)))
75             return; /** OOM */
76         
77         for (e = avahi_hashmap_lookup(s->entries_by_key, k); e; e = e->by_key_next)
78             if (!e->dead && avahi_entry_is_registered(s, e, i)) 
79                 callback(s, e->record, e->flags & AVAHI_PUBLISH_UNIQUE, userdata);
80         
81         avahi_key_unref(k);
82     }
83 }
84
85 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) {
86     assert(s);
87     assert(i);
88     assert(r);
89     assert(callback);
90
91     /* Call the specified callback far all records referenced by the one specified in *r */
92     
93     if (r->key->clazz == AVAHI_DNS_CLASS_IN) {
94         if (r->key->type == AVAHI_DNS_TYPE_PTR) {
95             enum_aux_records(s, i, r->data.ptr.name, AVAHI_DNS_TYPE_SRV, callback, userdata);
96             enum_aux_records(s, i, r->data.ptr.name, AVAHI_DNS_TYPE_TXT, callback, userdata);
97         } else if (r->key->type == AVAHI_DNS_TYPE_SRV) {
98             enum_aux_records(s, i, r->data.srv.name, AVAHI_DNS_TYPE_A, callback, userdata);
99             enum_aux_records(s, i, r->data.srv.name, AVAHI_DNS_TYPE_AAAA, callback, userdata);
100         } else if (r->key->type == AVAHI_DNS_TYPE_CNAME)
101             enum_aux_records(s, i, r->data.cname.name, AVAHI_DNS_TYPE_ANY, callback, userdata);
102     }
103 }
104
105 void avahi_server_prepare_response(AvahiServer *s, AvahiInterface *i, AvahiEntry *e, int unicast_response, int auxiliary) {
106     assert(s);
107     assert(i);
108     assert(e);
109
110     avahi_record_list_push(s->record_list, e->record, e->flags & AVAHI_PUBLISH_UNIQUE, unicast_response, auxiliary);
111 }
112
113 void avahi_server_prepare_matching_responses(AvahiServer *s, AvahiInterface *i, AvahiKey *k, int unicast_response) {
114     assert(s);
115     assert(i);
116     assert(k);
117
118     /* Push all records that match the specified key to the record list */
119
120     if (avahi_key_is_pattern(k)) {
121         AvahiEntry *e;
122
123         /* Handle ANY query */
124         
125         for (e = s->entries; e; e = e->entries_next)
126             if (!e->dead && avahi_key_pattern_match(k, e->record->key) && avahi_entry_is_registered(s, e, i))
127                 avahi_server_prepare_response(s, i, e, unicast_response, 0);
128
129     } else {
130         AvahiEntry *e;
131
132         /* Handle all other queries */
133         
134         for (e = avahi_hashmap_lookup(s->entries_by_key, k); e; e = e->by_key_next)
135             if (!e->dead && avahi_entry_is_registered(s, e, i))
136                 avahi_server_prepare_response(s, i, e, unicast_response, 0);
137     }
138
139     /* Look for CNAME records */
140
141     if ((k->clazz == AVAHI_DNS_CLASS_IN || k->clazz == AVAHI_DNS_CLASS_ANY)
142         && k->type != AVAHI_DNS_TYPE_CNAME && k->type != AVAHI_DNS_TYPE_ANY) {
143
144         AvahiKey *cname_key;
145
146         if (!(cname_key = avahi_key_new(k->name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_CNAME)))
147             return;
148         
149         avahi_server_prepare_matching_responses(s, i, cname_key, unicast_response);
150         avahi_key_unref(cname_key);
151     }
152 }
153
154 static void withdraw_entry(AvahiServer *s, AvahiEntry *e) {
155     assert(s);
156     assert(e);
157
158     /* Withdraw the specified entry, and if is part of an entry group,
159      * put that into COLLISION state */
160
161     if (e->dead)
162         return;
163     
164     if (e->group) {
165         AvahiEntry *k;
166         
167         for (k = e->group->entries; k; k = k->by_group_next) 
168             if (!k->dead) {
169                 avahi_goodbye_entry(s, k, 0, 1);
170                 k->dead = 1;
171             }
172
173         e->group->n_probing = 0;
174
175         avahi_s_entry_group_change_state(e->group, AVAHI_ENTRY_GROUP_COLLISION);
176     } else {
177         avahi_goodbye_entry(s, e, 0, 1);
178         e->dead = 1;
179     }
180
181     s->need_entry_cleanup = 1;
182 }
183
184 static void withdraw_rrset(AvahiServer *s, AvahiKey *key) {
185     AvahiEntry *e;
186     
187     assert(s);
188     assert(key);
189
190     /* Withdraw an entry RRSset */
191     
192     for (e = avahi_hashmap_lookup(s->entries_by_key, key); e; e = e->by_key_next)
193         withdraw_entry(s, e);
194 }
195
196 static void incoming_probe(AvahiServer *s, AvahiRecord *record, AvahiInterface *i) {
197     AvahiEntry *e, *n;
198     int ours = 0, won = 0, lost = 0;
199     
200     assert(s);
201     assert(record);
202     assert(i);
203
204     /* Handle incoming probes and check if they conflict our own probes */
205     
206     for (e = avahi_hashmap_lookup(s->entries_by_key, record->key); e; e = n) {
207         int cmp;
208         n = e->by_key_next;
209
210         if (e->dead)
211             continue;
212         
213         if ((cmp = avahi_record_lexicographical_compare(e->record, record)) == 0) {
214             ours = 1;
215             break;
216         } else {
217             
218             if (avahi_entry_is_probing(s, e, i)) {
219                 if (cmp > 0)
220                     won = 1;
221                 else /* cmp < 0 */
222                     lost = 1;
223             }
224         }
225     }
226
227     if (!ours) {
228         char *t = avahi_record_to_string(record);
229         
230         if (won)
231             avahi_log_debug("Recieved conflicting probe [%s]. Local host won.", t);
232         else if (lost) {
233             avahi_log_debug("Recieved conflicting probe [%s]. Local host lost. Withdrawing.", t);
234             withdraw_rrset(s, record->key);
235         }
236         
237         avahi_free(t);
238     }
239 }
240
241 static int handle_conflict(AvahiServer *s, AvahiInterface *i, AvahiRecord *record, int unique) {
242     int valid = 1, ours = 0, conflict = 0, withdraw_immediately = 0;
243     AvahiEntry *e, *n, *conflicting_entry = NULL;
244     
245     assert(s);
246     assert(i);
247     assert(record);
248
249     /* Check whether an incoming record conflicts with one of our own */
250     
251     for (e = avahi_hashmap_lookup(s->entries_by_key, record->key); e; e = n) {
252         n = e->by_key_next;
253
254         if (e->dead)
255             continue;
256
257         /* Check if the incoming is a goodbye record */
258         if (avahi_record_is_goodbye(record)) {
259
260             if (avahi_record_equal_no_ttl(e->record, record)) {
261                 char *t;
262
263                 /* Refresh */
264                 t = avahi_record_to_string(record); 
265                 avahi_log_debug("Recieved goodbye record for one of our records [%s]. Refreshing.", t);
266                 avahi_server_prepare_matching_responses(s, i, e->record->key, 0);
267
268                 valid = 0;
269                 avahi_free(t);
270                 break;
271             }
272
273             /* If the goodybe packet doesn't match one of our own RRs, we simply ignore it. */
274             continue;
275         }
276         
277         if (!(e->flags & AVAHI_PUBLISH_UNIQUE) && !unique)
278             continue;
279
280         /* Either our entry or the other is intended to be unique, so let's check */
281         
282         if (avahi_record_equal_no_ttl(e->record, record)) {
283             ours = 1; /* We have an identical record, so this is no conflict */
284             
285             /* Check wheter there is a TTL conflict */
286             if (record->ttl <= e->record->ttl/2 &&
287                 avahi_entry_is_registered(s, e, i)) {
288                 char *t;
289                 /* Refresh */
290                 t = avahi_record_to_string(record); 
291                 
292                 avahi_log_debug("Recieved record with bad TTL [%s]. Refreshing.", t);
293                 avahi_server_prepare_matching_responses(s, i, e->record->key, 0);
294                 valid = 0;
295                 
296                 avahi_free(t);
297             }
298                 
299             /* There's no need to check the other entries of this RRset */
300             break;
301
302         } else {
303             
304             if (avahi_entry_is_registered(s, e, i)) {
305                 
306                 /* A conflict => we have to return to probe mode */
307                 conflict = 1;
308                 conflicting_entry = e;
309
310             } else if (avahi_entry_is_probing(s, e, i)) {
311
312                 /* We are currently registering a matching record, but
313                  * someone else already claimed it, so let's
314                  * withdraw */
315                 conflict = 1;
316                 withdraw_immediately = 1;
317             }
318         }
319     }
320
321     if (!ours && conflict) {
322         char *t;
323  
324         valid = 0;
325
326         t = avahi_record_to_string(record); 
327  
328         if (withdraw_immediately) {
329             avahi_log_debug("Recieved conflicting record [%s] with local record to be. Withdrawing.", t);
330             withdraw_rrset(s, record->key);
331         } else {
332             assert(conflicting_entry);
333             avahi_log_debug("Recieved conflicting record [%s]. Resetting our record.", t);
334             avahi_entry_return_to_initial_state(s, conflicting_entry, i);
335
336             /* Local unique records are returned to probing
337              * state. Local shared records are reannounced. */
338         }
339
340         avahi_free(t);
341     }
342
343     return valid;
344 }
345
346 static void append_aux_callback(AvahiServer *s, AvahiRecord *r, int flush_cache, void* userdata) {
347     int *unicast_response = userdata;
348
349     assert(s);
350     assert(r);
351     assert(unicast_response);
352     
353     avahi_record_list_push(s->record_list, r, flush_cache, *unicast_response, 1);
354 }
355
356 static void append_aux_records_to_list(AvahiServer *s, AvahiInterface *i, AvahiRecord *r, int unicast_response) {
357     assert(s);
358     assert(r);
359
360     avahi_server_enumerate_aux_records(s, i, r, append_aux_callback, &unicast_response);
361 }
362
363 void avahi_server_generate_response(AvahiServer *s, AvahiInterface *i, AvahiDnsPacket *p, const AvahiAddress *a, uint16_t port, int legacy_unicast, int immediately) {
364
365     assert(s);
366     assert(i);
367     assert(!legacy_unicast || (a && port > 0 && p));
368
369     if (legacy_unicast) {
370         AvahiDnsPacket *reply;
371         AvahiRecord *r;
372
373         if (!(reply = avahi_dns_packet_new_reply(p, 512 /* unicast DNS maximum packet size is 512 */ , 1, 1)))
374             return; /* OOM */
375         
376         while ((r = avahi_record_list_next(s->record_list, NULL, NULL, NULL))) {
377
378             append_aux_records_to_list(s, i, r, 0);
379             
380             if (avahi_dns_packet_append_record(reply, r, 0, 10))
381                 avahi_dns_packet_inc_field(reply, AVAHI_DNS_FIELD_ANCOUNT);
382             else {
383                 char *t = avahi_record_to_string(r);
384                 avahi_log_warn("Record [%s] not fitting in legacy unicast packet, dropping.", t);
385                 avahi_free(t);
386             }
387
388             avahi_record_unref(r);
389         }
390
391         if (avahi_dns_packet_get_field(reply, AVAHI_DNS_FIELD_ANCOUNT) != 0)
392             avahi_interface_send_packet_unicast(i, reply, a, port);
393
394         avahi_dns_packet_free(reply);
395
396     } else {
397         int unicast_response, flush_cache, auxiliary;
398         AvahiDnsPacket *reply = NULL;
399         AvahiRecord *r;
400
401         /* In case the query packet was truncated never respond
402         immediately, because known answer suppression records might be
403         contained in later packets */
404         int tc = p && !!(avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS) & AVAHI_DNS_FLAG_TC);
405         
406         while ((r = avahi_record_list_next(s->record_list, &flush_cache, &unicast_response, &auxiliary))) {
407
408             int im = immediately;
409
410             /* Only send the response immediately if it contains a
411              * unique entry AND it is not in reply to a truncated
412              * packet AND it is not an auxiliary record AND all other
413              * responses for this record are unique too. */
414             
415             if (flush_cache && !tc && !auxiliary && avahi_record_list_all_flush_cache(s->record_list))
416                 im = 1;
417
418             if (!avahi_interface_post_response(i, r, flush_cache, a, im) && unicast_response) {
419
420                 /* Due to some reasons the record has not been scheduled.
421                  * The client requested an unicast response in that
422                  * case. Therefore we prepare such a response */
423
424                 append_aux_records_to_list(s, i, r, unicast_response);
425                 
426                 for (;;) {
427                 
428                     if (!reply) {
429                         assert(p);
430
431                         if (!(reply = avahi_dns_packet_new_reply(p, i->hardware->mtu, 0, 0)))
432                             break; /* OOM */
433                     }
434                 
435                     if (avahi_dns_packet_append_record(reply, r, flush_cache, 0)) {
436
437                         /* Appending this record succeeded, so incremeant
438                          * the specific header field, and return to the caller */
439                         
440                         avahi_dns_packet_inc_field(reply, AVAHI_DNS_FIELD_ANCOUNT);
441
442                         break;
443                     }
444
445                     if (avahi_dns_packet_get_field(reply, AVAHI_DNS_FIELD_ANCOUNT) == 0) {
446                         size_t size;
447
448                         /* The record is too large for one packet, so create a larger packet */
449
450                         avahi_dns_packet_free(reply);
451                         size = avahi_record_get_estimate_size(r) + AVAHI_DNS_PACKET_HEADER_SIZE;
452                         if (size > AVAHI_DNS_PACKET_SIZE_MAX)
453                             size = AVAHI_DNS_PACKET_SIZE_MAX;
454
455                         if (!(reply = avahi_dns_packet_new_reply(p, size, 0, 1)))
456                             break; /* OOM */
457
458                         if (!avahi_dns_packet_append_record(reply, r, flush_cache, 0)) {
459                             char *t;
460                             avahi_dns_packet_free(reply);
461                             t = avahi_record_to_string(r);
462                             avahi_log_warn("Record [%s] too large, doesn't fit in any packet!", t);
463                             avahi_free(t);
464                             break;
465                         } else
466                             avahi_dns_packet_inc_field(reply, AVAHI_DNS_FIELD_ANCOUNT);
467                     }
468
469                     /* Appending the record didn't succeeed, so let's send this packet, and create a new one */
470                     avahi_interface_send_packet_unicast(i, reply, a, port);
471                     avahi_dns_packet_free(reply);
472                     reply = NULL;
473                 }
474             }
475
476             avahi_record_unref(r);
477         }
478
479         if (reply) {
480             if (avahi_dns_packet_get_field(reply, AVAHI_DNS_FIELD_ANCOUNT) != 0) 
481                 avahi_interface_send_packet_unicast(i, reply, a, port);
482             avahi_dns_packet_free(reply);
483         }
484     }
485
486     avahi_record_list_flush(s->record_list);
487 }
488
489 static void reflect_response(AvahiServer *s, AvahiInterface *i, AvahiRecord *r, int flush_cache) {
490     AvahiInterface *j;
491     
492     assert(s);
493     assert(i);
494     assert(r);
495
496     if (!s->config.enable_reflector)
497         return;
498
499     for (j = s->monitor->interfaces; j; j = j->interface_next)
500         if (j != i && (s->config.reflect_ipv || j->protocol == i->protocol))
501             avahi_interface_post_response(j, r, flush_cache, NULL, 1);
502 }
503
504 static void* reflect_cache_walk_callback(AvahiCache *c, AvahiKey *pattern, AvahiCacheEntry *e, void* userdata) {
505     AvahiServer *s = userdata;
506
507     assert(c);
508     assert(pattern);
509     assert(e);
510     assert(s);
511
512     avahi_record_list_push(s->record_list, e->record, e->cache_flush, 0, 0);
513     return NULL;
514 }
515
516 static void reflect_query(AvahiServer *s, AvahiInterface *i, AvahiKey *k) {
517     AvahiInterface *j;
518     
519     assert(s);
520     assert(i);
521     assert(k);
522
523     if (!s->config.enable_reflector)
524         return;
525
526     for (j = s->monitor->interfaces; j; j = j->interface_next)
527         if (j != i && (s->config.reflect_ipv || j->protocol == i->protocol)) {
528             /* Post the query to other networks */
529             avahi_interface_post_query(j, k, 1, NULL);
530
531             /* Reply from caches of other network. This is needed to
532              * "work around" known answer suppression. */
533
534             avahi_cache_walk(j->cache, k, reflect_cache_walk_callback, s);
535         }
536 }
537
538 static void reflect_probe(AvahiServer *s, AvahiInterface *i, AvahiRecord *r) {
539     AvahiInterface *j;
540     
541     assert(s);
542     assert(i);
543     assert(r);
544
545     if (!s->config.enable_reflector)
546         return;
547
548     for (j = s->monitor->interfaces; j; j = j->interface_next)
549         if (j != i && (s->config.reflect_ipv || j->protocol == i->protocol))
550             avahi_interface_post_probe(j, r, 1);
551 }
552
553 static void handle_query_packet(AvahiServer *s, AvahiDnsPacket *p, AvahiInterface *i, const AvahiAddress *a, uint16_t port, int legacy_unicast, int from_local_iface) {
554     size_t n;
555     int is_probe;
556     
557     assert(s);
558     assert(p);
559     assert(i);
560     assert(a);
561
562     assert(avahi_record_list_is_empty(s->record_list));
563
564     is_probe = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT) > 0;
565     
566     /* Handle the questions */
567     for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_QDCOUNT); n > 0; n --) {
568         AvahiKey *key;
569         int unicast_response = 0;
570
571         if (!(key = avahi_dns_packet_consume_key(p, &unicast_response))) {
572             avahi_log_warn(__FILE__": Packet too short or invalid while reading question key. (Maybe an UTF8 problem?)");
573             goto fail;
574         }
575
576         if (!legacy_unicast && !from_local_iface) {
577             reflect_query(s, i, key);
578             avahi_cache_start_poof(i->cache, key, a);
579         }
580
581         if (avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) == 0 &&
582             !(avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS) & AVAHI_DNS_FLAG_TC))
583             /* Allow our own queries to be suppressed by incoming
584              * queries only when they do not include known answers */
585             avahi_query_scheduler_incoming(i->query_scheduler, key);
586         
587         avahi_server_prepare_matching_responses(s, i, key, unicast_response);
588         avahi_key_unref(key);
589     }
590
591     if (!legacy_unicast) {
592
593         /* Known Answer Suppression */
594         for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT); n > 0; n --) {
595             AvahiRecord *record;
596             int unique = 0;
597             
598             if (!(record = avahi_dns_packet_consume_record(p, &unique))) {
599                 avahi_log_warn(__FILE__": Packet too short or invalid while reading known answer record. (Maybe an UTF8 problem?)");
600                 goto fail;
601             }
602             
603             if (handle_conflict(s, i, record, unique)) {
604                 avahi_response_scheduler_suppress(i->response_scheduler, record, a);
605                 avahi_record_list_drop(s->record_list, record);
606                 avahi_cache_stop_poof(i->cache, record, a);
607             }
608             
609             avahi_record_unref(record);
610         }
611         
612         /* Probe record */
613         for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT); n > 0; n --) {
614             AvahiRecord *record;
615             int unique = 0;
616             
617             if (!(record = avahi_dns_packet_consume_record(p, &unique))) {
618                 avahi_log_warn(__FILE__": Packet too short or invalid while reading probe record. (Maybe an UTF8 problem?)");
619                 goto fail;
620             }
621             
622             if (!avahi_key_is_pattern(record->key)) {
623                 if (!from_local_iface)
624                     reflect_probe(s, i, record);
625                 incoming_probe(s, record, i);
626             }
627             
628             avahi_record_unref(record);
629         }
630     }
631
632     if (!avahi_record_list_is_empty(s->record_list))
633         avahi_server_generate_response(s, i, p, a, port, legacy_unicast, is_probe);
634
635     return;
636     
637 fail:
638     avahi_record_list_flush(s->record_list);
639 }
640
641 static void handle_response_packet(AvahiServer *s, AvahiDnsPacket *p, AvahiInterface *i, const AvahiAddress *a, int from_local_iface) {
642     unsigned n;
643     
644     assert(s);
645     assert(p);
646     assert(i);
647     assert(a);
648
649     for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) +
650              avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ARCOUNT); n > 0; n--) {
651         AvahiRecord *record;
652         int cache_flush = 0;
653 /*         char *txt; */
654         
655         if (!(record = avahi_dns_packet_consume_record(p, &cache_flush))) {
656             avahi_log_warn(__FILE__": Packet too short or invalid while reading response record. (Maybe an UTF8 problem?)");
657             break;
658         }
659
660         if (!avahi_key_is_pattern(record->key)) {
661
662             if (handle_conflict(s, i, record, cache_flush)) {
663                 if (!from_local_iface)
664                     reflect_response(s, i, record, cache_flush);
665                 avahi_cache_update(i->cache, record, cache_flush, a);
666                 avahi_response_scheduler_incoming(i->response_scheduler, record, cache_flush);
667             }
668         }
669             
670         avahi_record_unref(record);
671     }
672
673     /* If the incoming response contained a conflicting record, some
674        records have been scheduling for sending. We need to flush them
675        here. */
676     if (!avahi_record_list_is_empty(s->record_list))
677         avahi_server_generate_response(s, i, NULL, NULL, 0, 0, 1);
678 }
679
680 static AvahiLegacyUnicastReflectSlot* allocate_slot(AvahiServer *s) {
681     unsigned n, idx = (unsigned) -1;
682     AvahiLegacyUnicastReflectSlot *slot;
683     
684     assert(s);
685
686     if (!s->legacy_unicast_reflect_slots)
687         s->legacy_unicast_reflect_slots = avahi_new0(AvahiLegacyUnicastReflectSlot*, AVAHI_LEGACY_UNICAST_REFLECT_SLOTS_MAX);
688
689     for (n = 0; n < AVAHI_LEGACY_UNICAST_REFLECT_SLOTS_MAX; n++, s->legacy_unicast_reflect_id++) {
690         idx = s->legacy_unicast_reflect_id % AVAHI_LEGACY_UNICAST_REFLECT_SLOTS_MAX;
691         
692         if (!s->legacy_unicast_reflect_slots[idx])
693             break;
694     }
695
696     if (idx == (unsigned) -1 || s->legacy_unicast_reflect_slots[idx])
697         return NULL;
698
699     if (!(slot = avahi_new(AvahiLegacyUnicastReflectSlot, 1)))
700         return NULL; /* OOM */
701
702     s->legacy_unicast_reflect_slots[idx] = slot;
703     slot->id = s->legacy_unicast_reflect_id++;
704     slot->server = s;
705     
706     return slot;
707 }
708
709 static void deallocate_slot(AvahiServer *s, AvahiLegacyUnicastReflectSlot *slot) {
710     unsigned idx;
711
712     assert(s);
713     assert(slot);
714
715     idx = slot->id % AVAHI_LEGACY_UNICAST_REFLECT_SLOTS_MAX;
716
717     assert(s->legacy_unicast_reflect_slots[idx] == slot);
718
719     avahi_time_event_free(slot->time_event);
720     
721     avahi_free(slot);
722     s->legacy_unicast_reflect_slots[idx] = NULL;
723 }
724
725 static void free_slots(AvahiServer *s) {
726     unsigned idx;
727     assert(s);
728
729     if (!s->legacy_unicast_reflect_slots)
730         return;
731
732     for (idx = 0; idx < AVAHI_LEGACY_UNICAST_REFLECT_SLOTS_MAX; idx ++)
733         if (s->legacy_unicast_reflect_slots[idx])
734             deallocate_slot(s, s->legacy_unicast_reflect_slots[idx]);
735
736     avahi_free(s->legacy_unicast_reflect_slots);
737     s->legacy_unicast_reflect_slots = NULL;
738 }
739
740 static AvahiLegacyUnicastReflectSlot* find_slot(AvahiServer *s, uint16_t id) {
741     unsigned idx;
742     
743     assert(s);
744
745     if (!s->legacy_unicast_reflect_slots)
746         return NULL;
747     
748     idx = id % AVAHI_LEGACY_UNICAST_REFLECT_SLOTS_MAX;
749
750     if (!s->legacy_unicast_reflect_slots[idx] || s->legacy_unicast_reflect_slots[idx]->id != id)
751         return NULL;
752
753     return s->legacy_unicast_reflect_slots[idx];
754 }
755
756 static void legacy_unicast_reflect_slot_timeout(AvahiTimeEvent *e, void *userdata) {
757     AvahiLegacyUnicastReflectSlot *slot = userdata;
758
759     assert(e);
760     assert(slot);
761     assert(slot->time_event == e);
762
763     deallocate_slot(slot->server, slot);
764 }
765
766 static void reflect_legacy_unicast_query_packet(AvahiServer *s, AvahiDnsPacket *p, AvahiInterface *i, const AvahiAddress *a, uint16_t port) {
767     AvahiLegacyUnicastReflectSlot *slot;
768     AvahiInterface *j;
769
770     assert(s);
771     assert(p);
772     assert(i);
773     assert(a);
774     assert(port > 0);
775     assert(i->protocol == a->proto);
776     
777     if (!s->config.enable_reflector)
778         return;
779
780     /* Reflecting legacy unicast queries is a little more complicated
781        than reflecting normal queries, since we must route the
782        responses back to the right client. Therefore we must store
783        some information for finding the right client contact data for
784        response packets. In contrast to normal queries legacy
785        unicast query and response packets are reflected untouched and
786        are not reassembled into larger packets */
787
788     if (!(slot = allocate_slot(s))) {
789         /* No slot available, we drop this legacy unicast query */
790         avahi_log_warn("No slot available for legacy unicast reflection, dropping query packet.");
791         return;
792     }
793
794     slot->original_id = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ID);
795     slot->address = *a;
796     slot->port = port;
797     slot->interface = i->hardware->index;
798
799     avahi_elapse_time(&slot->elapse_time, 2000, 0);
800     slot->time_event = avahi_time_event_new(s->time_event_queue, &slot->elapse_time, legacy_unicast_reflect_slot_timeout, slot);
801
802     /* Patch the packet with our new locally generatet id */
803     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_ID, slot->id);
804     
805     for (j = s->monitor->interfaces; j; j = j->interface_next)
806         if (avahi_interface_is_relevant(j) &&
807             j != i &&
808             (s->config.reflect_ipv || j->protocol == i->protocol)) {
809
810             if (j->protocol == AVAHI_PROTO_INET && s->fd_legacy_unicast_ipv4 >= 0) {
811                 avahi_send_dns_packet_ipv4(s->fd_legacy_unicast_ipv4, j->hardware->index, p, NULL, NULL, 0);
812             } else if (j->protocol == AVAHI_PROTO_INET6 && s->fd_legacy_unicast_ipv6 >= 0)
813                 avahi_send_dns_packet_ipv6(s->fd_legacy_unicast_ipv6, j->hardware->index, p, NULL, NULL, 0);
814         }
815
816     /* Reset the id */
817     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_ID, slot->original_id);
818 }
819
820 static int originates_from_local_legacy_unicast_socket(AvahiServer *s, const AvahiAddress *address, uint16_t port) {
821     assert(s);
822     assert(address);
823     assert(port > 0);
824
825     if (!s->config.enable_reflector)
826         return 0;
827     
828     if (!avahi_address_is_local(s->monitor, address))
829         return 0;
830     
831     if (address->proto == AVAHI_PROTO_INET && s->fd_legacy_unicast_ipv4 >= 0) {
832         struct sockaddr_in lsa;
833         socklen_t l = sizeof(lsa);
834         
835         if (getsockname(s->fd_legacy_unicast_ipv4, (struct sockaddr*) &lsa, &l) != 0)
836             avahi_log_warn("getsockname(): %s", strerror(errno));
837         else
838             return lsa.sin_port == port;
839
840     }
841
842     if (address->proto == AVAHI_PROTO_INET6 && s->fd_legacy_unicast_ipv6 >= 0) {
843         struct sockaddr_in6 lsa;
844         socklen_t l = sizeof(lsa);
845
846         if (getsockname(s->fd_legacy_unicast_ipv6, (struct sockaddr*) &lsa, &l) != 0)
847             avahi_log_warn("getsockname(): %s", strerror(errno));
848         else
849             return lsa.sin6_port == port;
850     }
851
852     return 0;
853 }
854
855 static int is_mdns_mcast_address(const AvahiAddress *a) {
856     AvahiAddress b;
857     assert(a);
858
859     avahi_address_parse(a->proto == AVAHI_PROTO_INET ? AVAHI_IPV4_MCAST_GROUP : AVAHI_IPV6_MCAST_GROUP, a->proto, &b);
860     return avahi_address_cmp(a, &b) == 0;
861 }
862
863 static int originates_from_local_iface(AvahiServer *s, AvahiIfIndex iface, const AvahiAddress *a, uint16_t port) {
864     assert(s);
865     assert(iface != AVAHI_IF_UNSPEC);
866     assert(a);
867
868     /* If it isn't the MDNS port it can't be generated by us */
869     if (port != AVAHI_MDNS_PORT)
870         return 0;
871
872     return avahi_interface_has_address(s->monitor, iface, a);
873 }
874
875 static void dispatch_packet(AvahiServer *s, AvahiDnsPacket *p, const AvahiAddress *src_address, uint16_t port, const AvahiAddress *dst_address, AvahiIfIndex iface, int ttl) {
876     AvahiInterface *i;
877     int from_local_iface = 0;
878     
879     assert(s);
880     assert(p);
881     assert(src_address);
882     assert(dst_address);
883     assert(iface > 0);
884     assert(src_address->proto == dst_address->proto);
885
886     if (!(i = avahi_interface_monitor_get_interface(s->monitor, iface, src_address->proto)) ||
887         !avahi_interface_is_relevant(i)) {
888         avahi_log_warn("Recieved packet from invalid interface.");
889         return;
890     }
891
892     if (avahi_address_is_ipv4_in_ipv6(src_address))
893         /* This is an IPv4 address encapsulated in IPv6, so let's ignore it. */
894         return;
895
896     if (originates_from_local_legacy_unicast_socket(s, src_address, port))
897         /* This originates from our local reflector, so let's ignore it */
898         return;
899
900     /* We don't want to reflect local traffic, so we check if this packet is generated locally. */
901     if (s->config.enable_reflector)
902         from_local_iface = originates_from_local_iface(s, iface, src_address, port);
903
904     if (avahi_dns_packet_check_valid_multicast(p) < 0) {
905         avahi_log_warn("Recieved invalid packet.");
906         return;
907     }
908
909     if (avahi_dns_packet_is_query(p)) {
910         int legacy_unicast = 0;
911
912         if (avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ARCOUNT) != 0) {
913             avahi_log_warn("Invalid query packet.");
914             return;
915         }
916
917         if (port != AVAHI_MDNS_PORT) {
918             /* Legacy Unicast */
919
920             if ((avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) != 0 ||
921                  avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT) != 0)) {
922                 avahi_log_warn("Invalid legacy unicast query packet.");
923                 return;
924             }
925         
926             legacy_unicast = 1;
927         }
928
929         if (legacy_unicast)
930             reflect_legacy_unicast_query_packet(s, p, i, src_address, port);
931         
932         handle_query_packet(s, p, i, src_address, port, legacy_unicast, from_local_iface);
933         
934     } else {
935         if (port != AVAHI_MDNS_PORT) {
936             avahi_log_warn("Recieved repsonse with invalid source port %u on interface '%s.%i'", port, i->hardware->name, i->protocol);
937             return;
938         }
939
940         if (ttl != 255 && s->config.check_response_ttl) {
941             avahi_log_warn("Recieved response with invalid TTL %u on interface '%s.%i'.", ttl, i->hardware->name, i->protocol);
942             return;
943         }
944
945         if (!is_mdns_mcast_address(dst_address) &&
946             !avahi_interface_address_on_link(i, src_address)) {
947             avahi_log_warn("Received non-local response on interface '%s.%i'.", i->hardware->name, i->protocol);
948             return;
949         }
950         
951         if (avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_QDCOUNT) != 0 ||
952             avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) == 0 ||
953             avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT) != 0) {
954             avahi_log_warn("Invalid response packet.");
955             return;
956         }
957
958         handle_response_packet(s, p, i, src_address, from_local_iface);
959     }
960 }
961
962 static void dispatch_legacy_unicast_packet(AvahiServer *s, AvahiDnsPacket *p) {
963     AvahiInterface *j;
964     AvahiLegacyUnicastReflectSlot *slot;
965     
966     assert(s);
967     assert(p);
968
969     if (avahi_dns_packet_check_valid(p) < 0 || avahi_dns_packet_is_query(p)) {
970         avahi_log_warn("Recieved invalid packet.");
971         return;
972     }
973
974     if (!(slot = find_slot(s, avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ID)))) {
975         avahi_log_warn("Recieved legacy unicast response with unknown id");
976         return;
977     }
978
979     if (!(j = avahi_interface_monitor_get_interface(s->monitor, slot->interface, slot->address.proto)) ||
980         !avahi_interface_is_relevant(j))
981         return;
982
983     /* Patch the original ID into this response */
984     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_ID, slot->original_id);
985
986     /* Forward the response to the correct client */
987     avahi_interface_send_packet_unicast(j, p, &slot->address, slot->port);
988
989     /* Undo changes to packet */
990     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_ID, slot->id);
991 }
992
993 static void cleanup_dead(AvahiServer *s) {
994     assert(s);
995     
996     avahi_cleanup_dead_entries(s);
997     avahi_browser_cleanup(s);
998 }
999
1000 static void mcast_socket_event(AvahiWatch *w, int fd, AvahiWatchEvent events, void *userdata) {
1001     AvahiServer *s = userdata;
1002     AvahiAddress dest, src;
1003     AvahiDnsPacket *p = NULL;
1004     AvahiIfIndex iface;
1005     uint16_t port;
1006     uint8_t ttl;
1007
1008     assert(w);
1009     assert(fd >= 0);
1010     assert(events & AVAHI_WATCH_IN);
1011     
1012     if (fd == s->fd_ipv4) {
1013         dest.proto = src.proto = AVAHI_PROTO_INET;
1014         p = avahi_recv_dns_packet_ipv4(s->fd_ipv4, &src.data.ipv4, &port, &dest.data.ipv4, &iface, &ttl);
1015     } else {
1016         assert(fd == s->fd_ipv6);
1017         dest.proto = src.proto = AVAHI_PROTO_INET6;
1018         p = avahi_recv_dns_packet_ipv6(s->fd_ipv6, &src.data.ipv6, &port, &dest.data.ipv6, &iface, &ttl);
1019     }
1020         
1021     if (p) {
1022         if (iface == AVAHI_IF_UNSPEC) 
1023             iface = avahi_find_interface_for_address(s->monitor, &dest);
1024         
1025         if (iface != AVAHI_IF_UNSPEC)
1026             dispatch_packet(s, p, &src, port, &dest, iface, ttl);
1027         else
1028             avahi_log_error("Incoming packet recieved on address that isn't local.");
1029
1030         avahi_dns_packet_free(p);
1031
1032         cleanup_dead(s);
1033     }
1034 }
1035
1036 static void legacy_unicast_socket_event(AvahiWatch *w, int fd, AvahiWatchEvent events, void *userdata) {
1037     AvahiServer *s = userdata;
1038     AvahiDnsPacket *p = NULL;
1039
1040     assert(w);
1041     assert(fd >= 0);
1042     assert(events & AVAHI_WATCH_IN);
1043     
1044     if (fd == s->fd_legacy_unicast_ipv4)
1045         p = avahi_recv_dns_packet_ipv4(s->fd_legacy_unicast_ipv4, NULL, NULL, NULL, NULL, NULL);
1046     else {
1047         assert(fd == s->fd_legacy_unicast_ipv6);
1048         p = avahi_recv_dns_packet_ipv6(s->fd_legacy_unicast_ipv6, NULL, NULL, NULL, NULL, NULL);
1049     }
1050
1051     if (p) {
1052         dispatch_legacy_unicast_packet(s, p);
1053         avahi_dns_packet_free(p);
1054
1055         cleanup_dead(s);
1056     }
1057 }
1058
1059 static void server_set_state(AvahiServer *s, AvahiServerState state) {
1060     assert(s);
1061
1062     if (s->state == state)
1063         return;
1064     
1065     s->state = state;
1066
1067     avahi_interface_monitor_update_rrs(s->monitor, 0);
1068     
1069     if (s->callback)
1070         s->callback(s, state, s->userdata);
1071 }
1072
1073 static void withdraw_host_rrs(AvahiServer *s) {
1074     assert(s);
1075
1076     if (s->hinfo_entry_group)
1077         avahi_s_entry_group_reset(s->hinfo_entry_group);
1078
1079     if (s->browse_domain_entry_group)
1080         avahi_s_entry_group_reset(s->browse_domain_entry_group);
1081
1082     avahi_interface_monitor_update_rrs(s->monitor, 1);
1083     s->n_host_rr_pending = 0;
1084 }
1085
1086 void avahi_server_decrease_host_rr_pending(AvahiServer *s) {
1087     assert(s);
1088     
1089     assert(s->n_host_rr_pending > 0);
1090
1091     if (--s->n_host_rr_pending == 0)
1092         server_set_state(s, AVAHI_SERVER_RUNNING);
1093 }
1094
1095 void avahi_host_rr_entry_group_callback(AvahiServer *s, AvahiSEntryGroup *g, AvahiEntryGroupState state, AVAHI_GCC_UNUSED void *userdata) {
1096     assert(s);
1097     assert(g);
1098
1099     if (state == AVAHI_ENTRY_GROUP_REGISTERING &&
1100         s->state == AVAHI_SERVER_REGISTERING)
1101         s->n_host_rr_pending ++;
1102     
1103     else if (state == AVAHI_ENTRY_GROUP_COLLISION &&
1104         (s->state == AVAHI_SERVER_REGISTERING || s->state == AVAHI_SERVER_RUNNING)) {
1105         withdraw_host_rrs(s);
1106         server_set_state(s, AVAHI_SERVER_COLLISION);
1107         
1108     } else if (state == AVAHI_ENTRY_GROUP_ESTABLISHED &&
1109                s->state == AVAHI_SERVER_REGISTERING)
1110         avahi_server_decrease_host_rr_pending(s);
1111 }
1112
1113 static void register_hinfo(AvahiServer *s) {
1114     struct utsname utsname;
1115     AvahiRecord *r;
1116     
1117     assert(s);
1118     
1119     if (!s->config.publish_hinfo)
1120         return;
1121
1122     if (s->hinfo_entry_group)
1123         assert(avahi_s_entry_group_is_empty(s->hinfo_entry_group));
1124     else
1125         s->hinfo_entry_group = avahi_s_entry_group_new(s, avahi_host_rr_entry_group_callback, NULL);
1126
1127     if (!s->hinfo_entry_group) {
1128         avahi_log_warn("Failed to create HINFO entry group: %s", avahi_strerror(s->error));
1129         return;
1130     }
1131     
1132     /* Fill in HINFO rr */
1133     if ((r = avahi_record_new_full(s->host_name_fqdn, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_HINFO, AVAHI_DEFAULT_TTL_HOST_NAME))) {
1134
1135         if (uname(&utsname) < 0)
1136             avahi_log_warn("uname() failed: %s\n", avahi_strerror(errno));
1137         else {
1138             
1139             r->data.hinfo.cpu = avahi_strdup(avahi_strup(utsname.machine));
1140             r->data.hinfo.os = avahi_strdup(avahi_strup(utsname.sysname));
1141             
1142             avahi_log_info("Registering HINFO record with values '%s'/'%s'.", r->data.hinfo.cpu, r->data.hinfo.os);
1143             
1144             if (avahi_server_add(s, s->hinfo_entry_group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, AVAHI_PUBLISH_UNIQUE, r) < 0) {
1145                 avahi_log_warn("Failed to add HINFO RR: %s", avahi_strerror(s->error));
1146                 return;
1147             }
1148         }
1149
1150         avahi_record_unref(r);
1151     }
1152
1153     if (avahi_s_entry_group_commit(s->hinfo_entry_group) < 0)
1154         avahi_log_warn("Failed to commit HINFO entry group: %s", avahi_strerror(s->error));
1155
1156 }
1157
1158 static void register_localhost(AvahiServer *s) {
1159     AvahiAddress a;
1160     assert(s);
1161     
1162     /* Add localhost entries */
1163     avahi_address_parse("127.0.0.1", AVAHI_PROTO_INET, &a);
1164     avahi_server_add_address(s, NULL, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, AVAHI_PUBLISH_NO_PROBE|AVAHI_PUBLISH_NO_ANNOUNCE, "localhost", &a);
1165
1166     avahi_address_parse("::1", AVAHI_PROTO_INET6, &a);
1167     avahi_server_add_address(s, NULL, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, AVAHI_PUBLISH_NO_PROBE|AVAHI_PUBLISH_NO_ANNOUNCE, "ip6-localhost", &a);
1168 }
1169
1170 static void register_browse_domain(AvahiServer *s) {
1171     assert(s);
1172
1173     if (!s->config.publish_domain)
1174         return;
1175
1176     if (avahi_domain_equal(s->domain_name, "local"))
1177         return;
1178
1179     if (s->browse_domain_entry_group)
1180         assert(avahi_s_entry_group_is_empty(s->browse_domain_entry_group));
1181     else
1182         s->browse_domain_entry_group = avahi_s_entry_group_new(s, NULL, NULL);
1183
1184     if (!s->browse_domain_entry_group) {
1185         avahi_log_warn("Failed to create browse domain entry group: %s", avahi_strerror(s->error));
1186         return;
1187     }
1188     
1189     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) {
1190         avahi_log_warn("Failed to add browse domain RR: %s", avahi_strerror(s->error));
1191         return;
1192     }
1193
1194     if (avahi_s_entry_group_commit(s->browse_domain_entry_group) < 0)
1195         avahi_log_warn("Failed to commit browse domain entry group: %s", avahi_strerror(s->error));
1196 }
1197
1198 static void register_stuff(AvahiServer *s) {
1199     assert(s);
1200
1201     server_set_state(s, AVAHI_SERVER_REGISTERING);
1202     s->n_host_rr_pending ++; /** Make sure that the state isn't changed tp AVAHI_SERVER_RUNNING too early */
1203
1204     register_hinfo(s);
1205     register_browse_domain(s);
1206     avahi_interface_monitor_update_rrs(s->monitor, 0);
1207
1208     s->n_host_rr_pending --;
1209     
1210     if (s->n_host_rr_pending == 0)
1211         server_set_state(s, AVAHI_SERVER_RUNNING);
1212 }
1213
1214 static void update_fqdn(AvahiServer *s) {
1215     char *n;
1216     
1217     assert(s);
1218     assert(s->host_name);
1219     assert(s->domain_name);
1220
1221     if (!(n = avahi_strdup_printf("%s.%s", s->host_name, s->domain_name)))
1222         return; /* OOM */
1223
1224     avahi_free(s->host_name_fqdn);
1225     s->host_name_fqdn = n;
1226 }
1227
1228 int avahi_server_set_host_name(AvahiServer *s, const char *host_name) {
1229     char *hn = NULL;
1230     assert(s);
1231     
1232     AVAHI_CHECK_VALIDITY(s, !host_name || avahi_is_valid_host_name(host_name), AVAHI_ERR_INVALID_HOST_NAME);
1233
1234     if (!host_name) {
1235         hn = avahi_get_host_name_strdup();
1236         hn[strcspn(hn, ".")] = 0;
1237         host_name = hn;
1238     }
1239
1240     if (avahi_domain_equal(s->host_name, host_name) && s->state != AVAHI_SERVER_COLLISION) {
1241         avahi_free(hn);
1242         return avahi_server_set_errno(s, AVAHI_ERR_NO_CHANGE);
1243     }
1244     
1245     withdraw_host_rrs(s);
1246
1247     avahi_free(s->host_name);
1248     s->host_name = hn ? hn : avahi_strdup(host_name);
1249     
1250     update_fqdn(s);
1251
1252     register_stuff(s);
1253     return AVAHI_OK;
1254 }
1255
1256 int avahi_server_set_domain_name(AvahiServer *s, const char *domain_name) {
1257     char *dn = NULL;
1258     assert(s);
1259
1260     AVAHI_CHECK_VALIDITY(s, !domain_name || avahi_is_valid_domain_name(domain_name), AVAHI_ERR_INVALID_DOMAIN_NAME);
1261
1262     if (!domain_name) {
1263         dn = avahi_strdup("local");
1264         domain_name = dn;
1265     }
1266     
1267     if (avahi_domain_equal(s->domain_name, domain_name)) {
1268         avahi_free(dn);
1269         return avahi_server_set_errno(s, AVAHI_ERR_NO_CHANGE);
1270     }
1271
1272     withdraw_host_rrs(s);
1273
1274     avahi_free(s->domain_name);
1275     s->domain_name = avahi_normalize_name_strdup(domain_name);
1276     update_fqdn(s);
1277
1278     register_stuff(s);
1279
1280     avahi_free(dn);
1281     return AVAHI_OK;
1282 }
1283
1284 static int valid_server_config(const AvahiServerConfig *sc) {
1285
1286     if (sc->host_name && !avahi_is_valid_host_name(sc->host_name))
1287         return AVAHI_ERR_INVALID_HOST_NAME;
1288     
1289     if (sc->domain_name && !avahi_is_valid_domain_name(sc->domain_name))
1290         return AVAHI_ERR_INVALID_DOMAIN_NAME;
1291
1292     return AVAHI_OK;
1293 }
1294
1295 static int setup_sockets(AvahiServer *s) {
1296     assert(s);
1297     
1298     s->fd_ipv4 = s->config.use_ipv4 ? avahi_open_socket_ipv4(s->config.disallow_other_stacks) : -1;
1299     s->fd_ipv6 = s->config.use_ipv6 ? avahi_open_socket_ipv6(s->config.disallow_other_stacks) : -1;
1300     
1301     if (s->fd_ipv6 < 0 && s->fd_ipv4 < 0)
1302         return AVAHI_ERR_NO_NETWORK;
1303
1304     if (s->fd_ipv4 < 0 && s->config.use_ipv4)
1305         avahi_log_notice("Failed to create IPv4 socket, proceeding in IPv6 only mode");
1306     else if (s->fd_ipv6 < 0 && s->config.use_ipv6)
1307         avahi_log_notice("Failed to create IPv6 socket, proceeding in IPv4 only mode");
1308
1309     s->fd_legacy_unicast_ipv4 = s->fd_ipv4 >= 0 && s->config.enable_reflector ? avahi_open_unicast_socket_ipv4() : -1;
1310     s->fd_legacy_unicast_ipv6 = s->fd_ipv6 >= 0 && s->config.enable_reflector ? avahi_open_unicast_socket_ipv6() : -1;
1311     
1312     s->watch_ipv4 =
1313         s->watch_ipv6 =
1314         s->watch_legacy_unicast_ipv4 =
1315         s->watch_legacy_unicast_ipv6 = NULL;
1316     
1317     if (s->fd_ipv4 >= 0)
1318         s->watch_ipv4 = s->poll_api->watch_new(s->poll_api, s->fd_ipv4, AVAHI_WATCH_IN, mcast_socket_event, s);
1319     if (s->fd_ipv6 >= 0)
1320         s->watch_ipv6 = s->poll_api->watch_new(s->poll_api, s->fd_ipv6, AVAHI_WATCH_IN, mcast_socket_event, s);
1321     
1322     if (s->fd_legacy_unicast_ipv4 >= 0)
1323         s->watch_legacy_unicast_ipv4 = s->poll_api->watch_new(s->poll_api, s->fd_legacy_unicast_ipv4, AVAHI_WATCH_IN, legacy_unicast_socket_event, s);
1324     if (s->fd_legacy_unicast_ipv6 >= 0)
1325         s->watch_legacy_unicast_ipv6 = s->poll_api->watch_new(s->poll_api, s->fd_legacy_unicast_ipv6, AVAHI_WATCH_IN, legacy_unicast_socket_event, s);
1326             
1327     return 0;
1328 }
1329
1330 AvahiServer *avahi_server_new(const AvahiPoll *poll_api, const AvahiServerConfig *sc, AvahiServerCallback callback, void* userdata, int *error) {
1331     AvahiServer *s;
1332     int e;
1333     
1334     if (sc && (e = valid_server_config(sc)) < 0) {
1335         if (error)
1336             *error = e;
1337         return NULL;
1338     }
1339     
1340     if (!(s = avahi_new(AvahiServer, 1))) {
1341         if (error)
1342             *error = AVAHI_ERR_NO_MEMORY;
1343
1344         return NULL;
1345     }
1346
1347     s->poll_api = poll_api;
1348
1349     if (sc)
1350         avahi_server_config_copy(&s->config, sc);
1351     else
1352         avahi_server_config_init(&s->config);
1353
1354     if ((e = setup_sockets(s)) < 0) {
1355         if (error)
1356             *error = e;
1357
1358         avahi_server_config_free(&s->config);
1359         avahi_free(s);
1360         
1361         return NULL;
1362     }
1363     
1364     s->n_host_rr_pending = 0;
1365     s->need_entry_cleanup = 0;
1366     s->need_group_cleanup = 0;
1367     s->need_browser_cleanup = 0;
1368     s->hinfo_entry_group = NULL;
1369     s->browse_domain_entry_group = NULL;
1370     s->error = AVAHI_OK;
1371     s->state = AVAHI_SERVER_INVALID;
1372
1373     s->callback = callback;
1374     s->userdata = userdata;
1375
1376     s->time_event_queue = avahi_time_event_queue_new(poll_api);
1377
1378     s->entries_by_key = avahi_hashmap_new((AvahiHashFunc) avahi_key_hash, (AvahiEqualFunc) avahi_key_equal, NULL, NULL);
1379     AVAHI_LLIST_HEAD_INIT(AvahiEntry, s->entries);
1380     AVAHI_LLIST_HEAD_INIT(AvahiGroup, s->groups);
1381
1382     s->record_browser_hashmap = avahi_hashmap_new((AvahiHashFunc) avahi_key_hash, (AvahiEqualFunc) avahi_key_equal, NULL, NULL);
1383     AVAHI_LLIST_HEAD_INIT(AvahiSRecordBrowser, s->record_browsers);
1384     AVAHI_LLIST_HEAD_INIT(AvahiSHostNameResolver, s->host_name_resolvers);
1385     AVAHI_LLIST_HEAD_INIT(AvahiSAddressResolver, s->address_resolvers);
1386     AVAHI_LLIST_HEAD_INIT(AvahiSDomainBrowser, s->domain_browsers);
1387     AVAHI_LLIST_HEAD_INIT(AvahiSServiceTypeBrowser, s->service_type_browsers);
1388     AVAHI_LLIST_HEAD_INIT(AvahiSServiceBrowser, s->service_browsers);
1389     AVAHI_LLIST_HEAD_INIT(AvahiSServiceResolver, s->service_resolvers);
1390     AVAHI_LLIST_HEAD_INIT(AvahiSDNSServerBrowser, s->dns_server_browsers);
1391
1392     s->legacy_unicast_reflect_slots = NULL;
1393     s->legacy_unicast_reflect_id = 0;
1394
1395     s->record_list = avahi_record_list_new();
1396     
1397     /* Get host name */
1398     s->host_name = s->config.host_name ? avahi_normalize_name_strdup(s->config.host_name) : avahi_get_host_name_strdup();
1399     s->host_name[strcspn(s->host_name, ".")] = 0;
1400     s->domain_name = s->config.domain_name ? avahi_normalize_name_strdup(s->config.domain_name) : avahi_strdup("local");
1401     s->host_name_fqdn = NULL;
1402     update_fqdn(s);
1403
1404     do {
1405         s->local_service_cookie = (uint32_t) rand() * (uint32_t) rand();
1406     } while (s->local_service_cookie == AVAHI_SERVICE_COOKIE_INVALID);
1407
1408     if (s->config.enable_wide_area) {
1409         s->wide_area_lookup_engine = avahi_wide_area_engine_new(s);
1410         avahi_wide_area_set_servers(s->wide_area_lookup_engine, s->config.wide_area_servers, s->config.n_wide_area_servers);
1411     } else
1412         s->wide_area_lookup_engine = NULL;
1413
1414     s->multicast_lookup_engine = avahi_multicast_lookup_engine_new(s);
1415     
1416     s->monitor = avahi_interface_monitor_new(s);
1417     avahi_interface_monitor_sync(s->monitor);
1418
1419     register_localhost(s);
1420     register_stuff(s);
1421     
1422     return s;
1423 }
1424
1425 void avahi_server_free(AvahiServer* s) {
1426     assert(s);
1427
1428     /* Remove all browsers */
1429
1430     while (s->dns_server_browsers)
1431         avahi_s_dns_server_browser_free(s->dns_server_browsers);
1432     while (s->host_name_resolvers)
1433         avahi_s_host_name_resolver_free(s->host_name_resolvers);
1434     while (s->address_resolvers)
1435         avahi_s_address_resolver_free(s->address_resolvers);
1436     while (s->domain_browsers)
1437         avahi_s_domain_browser_free(s->domain_browsers);
1438     while (s->service_type_browsers)
1439         avahi_s_service_type_browser_free(s->service_type_browsers);
1440     while (s->service_browsers)
1441         avahi_s_service_browser_free(s->service_browsers);
1442     while (s->service_resolvers)
1443         avahi_s_service_resolver_free(s->service_resolvers);
1444     while (s->record_browsers)
1445         avahi_s_record_browser_destroy(s->record_browsers);
1446     
1447     /* Remove all locally rgeistered stuff */
1448
1449     while(s->entries)
1450         avahi_entry_free(s, s->entries);
1451
1452     avahi_interface_monitor_free(s->monitor);
1453
1454     while (s->groups)
1455         avahi_entry_group_free(s, s->groups);
1456
1457     free_slots(s);
1458
1459     avahi_hashmap_free(s->entries_by_key);
1460     avahi_record_list_free(s->record_list);
1461     avahi_hashmap_free(s->record_browser_hashmap);
1462
1463     if (s->wide_area_lookup_engine)
1464         avahi_wide_area_engine_free(s->wide_area_lookup_engine);
1465     avahi_multicast_lookup_engine_free(s->multicast_lookup_engine);
1466
1467     avahi_time_event_queue_free(s->time_event_queue);
1468     
1469     /* Free watches */
1470     
1471     if (s->watch_ipv4)
1472         s->poll_api->watch_free(s->watch_ipv4);
1473     if (s->watch_ipv6)
1474         s->poll_api->watch_free(s->watch_ipv6);
1475     
1476     if (s->watch_legacy_unicast_ipv4)
1477         s->poll_api->watch_free(s->watch_legacy_unicast_ipv4);
1478     if (s->watch_legacy_unicast_ipv6)
1479         s->poll_api->watch_free(s->watch_legacy_unicast_ipv6);
1480
1481     /* Free sockets */
1482     
1483     if (s->fd_ipv4 >= 0)
1484         close(s->fd_ipv4);
1485     if (s->fd_ipv6 >= 0)
1486         close(s->fd_ipv6);
1487     
1488     if (s->fd_legacy_unicast_ipv4 >= 0)
1489         close(s->fd_legacy_unicast_ipv4);
1490     if (s->fd_legacy_unicast_ipv6 >= 0)
1491         close(s->fd_legacy_unicast_ipv6);
1492     
1493     /* Free other stuff */
1494     
1495     avahi_free(s->host_name);
1496     avahi_free(s->domain_name);
1497     avahi_free(s->host_name_fqdn);
1498
1499     avahi_server_config_free(&s->config);
1500
1501     avahi_free(s);
1502 }
1503
1504 const char* avahi_server_get_domain_name(AvahiServer *s) {
1505     assert(s);
1506
1507     return s->domain_name;
1508 }
1509
1510 const char* avahi_server_get_host_name(AvahiServer *s) {
1511     assert(s);
1512
1513     return s->host_name;
1514 }
1515
1516 const char* avahi_server_get_host_name_fqdn(AvahiServer *s) {
1517     assert(s);
1518
1519     return s->host_name_fqdn;
1520 }
1521
1522 void* avahi_server_get_data(AvahiServer *s) {
1523     assert(s);
1524
1525     return s->userdata;
1526 }
1527
1528 void avahi_server_set_data(AvahiServer *s, void* userdata) {
1529     assert(s);
1530
1531     s->userdata = userdata;
1532 }
1533
1534 AvahiServerState avahi_server_get_state(AvahiServer *s) {
1535     assert(s);
1536
1537     return s->state;
1538 }
1539
1540 AvahiServerConfig* avahi_server_config_init(AvahiServerConfig *c) {
1541     assert(c);
1542
1543     memset(c, 0, sizeof(AvahiServerConfig));
1544     c->use_ipv6 = 1;
1545     c->use_ipv4 = 1;
1546     c->host_name = NULL;
1547     c->domain_name = NULL;
1548     c->check_response_ttl = 0;
1549     c->publish_hinfo = 1;
1550     c->publish_addresses = 1;
1551     c->publish_workstation = 1;
1552     c->publish_domain = 1;
1553     c->use_iff_running = 0;
1554     c->enable_reflector = 0;
1555     c->reflect_ipv = 0;
1556     c->add_service_cookie = 1;
1557     c->enable_wide_area = 0;
1558     c->n_wide_area_servers = 0;
1559     c->disallow_other_stacks = 0;
1560     c->browse_domains = NULL;
1561     c->disable_publishing = 0;
1562     c->allow_point_to_point = 0;
1563     
1564     return c;
1565 }
1566
1567 void avahi_server_config_free(AvahiServerConfig *c) {
1568     assert(c);
1569
1570     avahi_free(c->host_name);
1571     avahi_free(c->domain_name);
1572     avahi_string_list_free(c->browse_domains);
1573 }
1574
1575 AvahiServerConfig* avahi_server_config_copy(AvahiServerConfig *ret, const AvahiServerConfig *c) {
1576     char *d = NULL, *h = NULL;
1577     AvahiStringList *l = NULL;
1578     assert(ret);
1579     assert(c);
1580
1581     if (c->host_name)
1582         if (!(h = avahi_strdup(c->host_name)))
1583             return NULL;
1584
1585     if (c->domain_name)
1586         if (!(d = avahi_strdup(c->domain_name))) {
1587             avahi_free(h);
1588             return NULL;
1589         }
1590
1591     if (!(l = avahi_string_list_copy(c->browse_domains)) && c->browse_domains) {
1592         avahi_free(h);
1593         avahi_free(d);
1594         return NULL;
1595     }
1596     
1597     *ret = *c;
1598     ret->host_name = h;
1599     ret->domain_name = d;
1600     ret->browse_domains = l;
1601
1602     return ret;
1603 }
1604
1605 int avahi_server_errno(AvahiServer *s) {
1606     assert(s);
1607     
1608     return s->error;
1609 }
1610
1611 /* Just for internal use */
1612 int avahi_server_set_errno(AvahiServer *s, int error) {
1613     assert(s);
1614
1615     return s->error = error;
1616 }
1617
1618 uint32_t avahi_server_get_local_service_cookie(AvahiServer *s) {
1619     assert(s);
1620
1621     return s->local_service_cookie;
1622 }
1623
1624 static AvahiEntry *find_entry(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, AvahiKey *key) {
1625     AvahiEntry *e;
1626     
1627     assert(s);
1628     assert(key);
1629
1630     for (e = avahi_hashmap_lookup(s->entries_by_key, key); e; e = e->by_key_next)
1631
1632         if ((e->interface == interface || e->interface <= 0 || interface <= 0) &&
1633             (e->protocol == protocol || e->protocol == AVAHI_PROTO_UNSPEC || protocol == AVAHI_PROTO_UNSPEC) &&
1634             (!e->group || e->group->state == AVAHI_ENTRY_GROUP_ESTABLISHED || e->group->state == AVAHI_ENTRY_GROUP_REGISTERING))
1635
1636             return e;
1637
1638     return NULL;
1639 }
1640
1641 int avahi_server_get_group_of_service(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, const char *name, const char *type, const char *domain, AvahiSEntryGroup** ret_group) {
1642     AvahiKey *key = NULL;
1643     AvahiEntry *e;
1644     int ret;
1645     char n[AVAHI_DOMAIN_NAME_MAX];
1646     
1647     assert(s);
1648     assert(name);
1649     assert(type);
1650     assert(ret_group);
1651
1652     AVAHI_CHECK_VALIDITY(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
1653     AVAHI_CHECK_VALIDITY(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
1654     AVAHI_CHECK_VALIDITY(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
1655     AVAHI_CHECK_VALIDITY(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
1656     AVAHI_CHECK_VALIDITY(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
1657
1658     if ((ret = avahi_service_name_join(n, sizeof(n), name, type, domain) < 0))
1659         return avahi_server_set_errno(s, ret);
1660         
1661     if (!(key = avahi_key_new(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV)))
1662         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1663
1664     e = find_entry(s, interface, protocol, key);
1665     avahi_key_unref(key);
1666
1667     if (e) {
1668         *ret_group = e->group;
1669         return AVAHI_OK;
1670     }
1671
1672     return avahi_server_set_errno(s, AVAHI_ERR_NOT_FOUND);
1673 }
1674
1675 int avahi_server_is_service_local(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, const char *name) {
1676     AvahiKey *key = NULL;
1677     AvahiEntry *e;
1678
1679     assert(s);
1680     assert(name);
1681
1682     if (!s->host_name_fqdn)
1683         return 0;
1684     
1685     if (!(key = avahi_key_new(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV)))
1686         return 0;
1687
1688     e = find_entry(s, interface, protocol, key);
1689     avahi_key_unref(key);
1690
1691     if (!e)
1692         return 0;
1693     
1694     return avahi_domain_equal(s->host_name_fqdn, e->record->data.srv.name);
1695 }
1696
1697 int avahi_server_is_record_local(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, AvahiRecord *record) {
1698     AvahiEntry *e;
1699
1700     assert(s);
1701     assert(record);
1702     
1703     for (e = avahi_hashmap_lookup(s->entries_by_key, record->key); e; e = e->by_key_next)
1704
1705         if ((e->interface == interface || e->interface <= 0 || interface <= 0) &&
1706             (e->protocol == protocol || e->protocol == AVAHI_PROTO_UNSPEC || protocol == AVAHI_PROTO_UNSPEC) &&
1707             (!e->group || e->group->state == AVAHI_ENTRY_GROUP_ESTABLISHED || e->group->state == AVAHI_ENTRY_GROUP_REGISTERING) &&
1708             avahi_record_equal_no_ttl(record, e->record))
1709             return 1;
1710
1711     return 0;
1712 }
1713
1714 /** Set the wide area DNS servers */
1715 int avahi_server_set_wide_area_servers(AvahiServer *s, const AvahiAddress *a, unsigned n) {
1716     assert(s);
1717
1718     if (!s->wide_area_lookup_engine)
1719         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_CONFIG);
1720
1721     avahi_wide_area_set_servers(s->wide_area_lookup_engine, a, n);
1722     return AVAHI_OK;
1723 }