1 /***
2 This file is part of avahi.
3
4 avahi is free software; you can redistribute it and/or modify it
5 under the terms of the GNU Lesser General Public License as
6 published by the Free Software Foundation; either version 2.1 of the
7 License, or (at your option) any later version.
8
9 avahi is distributed in the hope that it will be useful, but WITHOUT
10 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
12 Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with avahi; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
17 USA.
18 ***/
19
20 #ifdef HAVE_CONFIG_H
21 #include <config.h>
22 #endif
23
24 #include <errno.h>
25 #include <string.h>
26 #include <unistd.h>
27 #include <stdlib.h>
28 #include <netinet/in.h>
29
30 #include <avahi-common/malloc.h>
31 #include <avahi-common/error.h>
32 #include <avahi-common/timeval.h>
33
34 #include "internal.h"
35 #include "browse.h"
36 #include "socket.h"
37 #include "log.h"
38 #include "hashmap.h"
39 #include "wide-area.h"
40 #include "addr-util.h"
41 #include "rr-util.h"
42
43 #define CACHE_ENTRIES_MAX 500
44
45 typedef struct AvahiWideAreaCacheEntry AvahiWideAreaCacheEntry;
46
47 struct AvahiWideAreaCacheEntry {
48 AvahiWideAreaLookupEngine *engine;
49
50 AvahiRecord *record;
51 struct timeval timestamp;
52 struct timeval expiry;
53
54 AvahiTimeEvent *time_event;
55
56 AVAHI_LLIST_FIELDS(AvahiWideAreaCacheEntry, by_key);
57 AVAHI_LLIST_FIELDS(AvahiWideAreaCacheEntry, cache);
58 };
59
60 struct AvahiWideAreaLookup {
61 AvahiWideAreaLookupEngine *engine;
62 int dead;
63
64 uint32_t id; /* effectively just an uint16_t, but we need it as an index for a hash table */
65 AvahiTimeEvent *time_event;
66
67 AvahiKey *key, *cname_key;
68
69 int n_send;
70 AvahiDnsPacket *packet;
71
72 AvahiWideAreaLookupCallback callback;
73 void *userdata;
74
75 AvahiAddress dns_server_used;
76
77 AVAHI_LLIST_FIELDS(AvahiWideAreaLookup, lookups);
78 AVAHI_LLIST_FIELDS(AvahiWideAreaLookup, by_key);
79 };
80
81 struct AvahiWideAreaLookupEngine {
82 AvahiServer *server;
83
84 int fd_ipv4, fd_ipv6;
85 AvahiWatch *watch_ipv4, *watch_ipv6;
86
87 uint16_t next_id;
88
89 /* Cache */
90 AVAHI_LLIST_HEAD(AvahiWideAreaCacheEntry, cache);
91 AvahiHashmap *cache_by_key;
92 unsigned cache_n_entries;
93
94 /* Lookups */
95 AVAHI_LLIST_HEAD(AvahiWideAreaLookup, lookups);
96 AvahiHashmap *lookups_by_id;
97 AvahiHashmap *lookups_by_key;
98
99 int cleanup_dead;
100
101 AvahiAddress dns_servers[AVAHI_WIDE_AREA_SERVERS_MAX];
102 unsigned n_dns_servers;
103 unsigned current_dns_server;
104 };
105
find_lookup(AvahiWideAreaLookupEngine * e,uint16_t id)106 static AvahiWideAreaLookup* find_lookup(AvahiWideAreaLookupEngine *e, uint16_t id) {
107 AvahiWideAreaLookup *l;
108 int i = (int) id;
109
110 assert(e);
111
112 if (!(l = avahi_hashmap_lookup(e->lookups_by_id, &i)))
113 return NULL;
114
115 assert(l->id == id);
116
117 if (l->dead)
118 return NULL;
119
120 return l;
121 }
122
send_to_dns_server(AvahiWideAreaLookup * l,AvahiDnsPacket * p)123 static int send_to_dns_server(AvahiWideAreaLookup *l, AvahiDnsPacket *p) {
124 AvahiAddress *a;
125
126 assert(l);
127 assert(p);
128
129 if (l->engine->n_dns_servers <= 0)
130 return -1;
131
132 assert(l->engine->current_dns_server < l->engine->n_dns_servers);
133
134 a = &l->engine->dns_servers[l->engine->current_dns_server];
135 l->dns_server_used = *a;
136
137 if (a->proto == AVAHI_PROTO_INET) {
138
139 if (l->engine->fd_ipv4 < 0)
140 return -1;
141
142 return avahi_send_dns_packet_ipv4(l->engine->fd_ipv4, AVAHI_IF_UNSPEC, p, NULL, &a->data.ipv4, AVAHI_DNS_PORT);
143
144 } else {
145 assert(a->proto == AVAHI_PROTO_INET6);
146
147 if (l->engine->fd_ipv6 < 0)
148 return -1;
149
150 return avahi_send_dns_packet_ipv6(l->engine->fd_ipv6, AVAHI_IF_UNSPEC, p, NULL, &a->data.ipv6, AVAHI_DNS_PORT);
151 }
152 }
153
next_dns_server(AvahiWideAreaLookupEngine * e)154 static void next_dns_server(AvahiWideAreaLookupEngine *e) {
155 assert(e);
156
157 e->current_dns_server++;
158
159 if (e->current_dns_server >= e->n_dns_servers)
160 e->current_dns_server = 0;
161 }
162
lookup_stop(AvahiWideAreaLookup * l)163 static void lookup_stop(AvahiWideAreaLookup *l) {
164 assert(l);
165
166 l->callback = NULL;
167
168 if (l->time_event) {
169 avahi_time_event_free(l->time_event);
170 l->time_event = NULL;
171 }
172 }
173
sender_timeout_callback(AvahiTimeEvent * e,void * userdata)174 static void sender_timeout_callback(AvahiTimeEvent *e, void *userdata) {
175 AvahiWideAreaLookup *l = userdata;
176 struct timeval tv;
177
178 assert(l);
179
180 /* Try another DNS server after three retries */
181 if (l->n_send >= 3 && avahi_address_cmp(&l->engine->dns_servers[l->engine->current_dns_server], &l->dns_server_used) == 0) {
182 next_dns_server(l->engine);
183
184 if (avahi_address_cmp(&l->engine->dns_servers[l->engine->current_dns_server], &l->dns_server_used) == 0)
185 /* There is no other DNS server, fail */
186 l->n_send = 1000;
187 }
188
189 if (l->n_send >= 6) {
190 avahi_log_warn(__FILE__": Query timed out.");
191 avahi_server_set_errno(l->engine->server, AVAHI_ERR_TIMEOUT);
192 l->callback(l->engine, AVAHI_BROWSER_FAILURE, AVAHI_LOOKUP_RESULT_WIDE_AREA, NULL, l->userdata);
193 lookup_stop(l);
194 return;
195 }
196
197 assert(l->packet);
198 send_to_dns_server(l, l->packet);
199 l->n_send++;
200
201 avahi_time_event_update(e, avahi_elapse_time(&tv, 1000, 0));
202 }
203
avahi_wide_area_lookup_new(AvahiWideAreaLookupEngine * e,AvahiKey * key,AvahiWideAreaLookupCallback callback,void * userdata)204 AvahiWideAreaLookup *avahi_wide_area_lookup_new(
205 AvahiWideAreaLookupEngine *e,
206 AvahiKey *key,
207 AvahiWideAreaLookupCallback callback,
208 void *userdata) {
209
210 struct timeval tv;
211 AvahiWideAreaLookup *l, *t;
212 uint8_t *p;
213
214 assert(e);
215 assert(key);
216 assert(callback);
217 assert(userdata);
218
219 l = avahi_new(AvahiWideAreaLookup, 1);
220 l->engine = e;
221 l->dead = 0;
222 l->key = avahi_key_ref(key);
223 l->cname_key = avahi_key_new_cname(l->key);
224 l->callback = callback;
225 l->userdata = userdata;
226
227 /* If more than 65K wide area quries are issued simultaneously,
228 * this will break. This should be limited by some higher level */
229
230 for (;; e->next_id++)
231 if (!find_lookup(e, e->next_id))
232 break; /* This ID is not yet used. */
233
234 l->id = e->next_id++;
235
236 /* We keep the packet around in case we need to repeat our query */
237 l->packet = avahi_dns_packet_new(0);
238
239 avahi_dns_packet_set_field(l->packet, AVAHI_DNS_FIELD_ID, (uint16_t) l->id);
240 avahi_dns_packet_set_field(l->packet, AVAHI_DNS_FIELD_FLAGS, AVAHI_DNS_FLAGS(0, 0, 0, 0, 1, 0, 0, 0, 0, 0));
241
242 p = avahi_dns_packet_append_key(l->packet, key, 0);
243 assert(p);
244
245 avahi_dns_packet_set_field(l->packet, AVAHI_DNS_FIELD_QDCOUNT, 1);
246
247 if (send_to_dns_server(l, l->packet) < 0) {
248 avahi_log_error(__FILE__": Failed to send packet.");
249 avahi_dns_packet_free(l->packet);
250 avahi_key_unref(l->key);
251 if (l->cname_key)
252 avahi_key_unref(l->cname_key);
253 avahi_free(l);
254 return NULL;
255 }
256
257 l->n_send = 1;
258
259 l->time_event = avahi_time_event_new(e->server->time_event_queue, avahi_elapse_time(&tv, 500, 0), sender_timeout_callback, l);
260
261 avahi_hashmap_insert(e->lookups_by_id, &l->id, l);
262
263 t = avahi_hashmap_lookup(e->lookups_by_key, l->key);
264 AVAHI_LLIST_PREPEND(AvahiWideAreaLookup, by_key, t, l);
265 avahi_hashmap_replace(e->lookups_by_key, avahi_key_ref(l->key), t);
266
267 AVAHI_LLIST_PREPEND(AvahiWideAreaLookup, lookups, e->lookups, l);
268
269 return l;
270 }
271
lookup_destroy(AvahiWideAreaLookup * l)272 static void lookup_destroy(AvahiWideAreaLookup *l) {
273 AvahiWideAreaLookup *t;
274 assert(l);
275
276 lookup_stop(l);
277
278 t = avahi_hashmap_lookup(l->engine->lookups_by_key, l->key);
279 AVAHI_LLIST_REMOVE(AvahiWideAreaLookup, by_key, t, l);
280 if (t)
281 avahi_hashmap_replace(l->engine->lookups_by_key, avahi_key_ref(l->key), t);
282 else
283 avahi_hashmap_remove(l->engine->lookups_by_key, l->key);
284
285 AVAHI_LLIST_REMOVE(AvahiWideAreaLookup, lookups, l->engine->lookups, l);
286
287 avahi_hashmap_remove(l->engine->lookups_by_id, &l->id);
288 avahi_dns_packet_free(l->packet);
289
290 if (l->key)
291 avahi_key_unref(l->key);
292
293 if (l->cname_key)
294 avahi_key_unref(l->cname_key);
295
296 avahi_free(l);
297 }
298
avahi_wide_area_lookup_free(AvahiWideAreaLookup * l)299 void avahi_wide_area_lookup_free(AvahiWideAreaLookup *l) {
300 assert(l);
301
302 if (l->dead)
303 return;
304
305 l->dead = 1;
306 l->engine->cleanup_dead = 1;
307 lookup_stop(l);
308 }
309
avahi_wide_area_cleanup(AvahiWideAreaLookupEngine * e)310 void avahi_wide_area_cleanup(AvahiWideAreaLookupEngine *e) {
311 AvahiWideAreaLookup *l, *n;
312 assert(e);
313
314 while (e->cleanup_dead) {
315 e->cleanup_dead = 0;
316
317 for (l = e->lookups; l; l = n) {
318 n = l->lookups_next;
319
320 if (l->dead)
321 lookup_destroy(l);
322 }
323 }
324 }
325
cache_entry_free(AvahiWideAreaCacheEntry * c)326 static void cache_entry_free(AvahiWideAreaCacheEntry *c) {
327 AvahiWideAreaCacheEntry *t;
328 assert(c);
329
330 if (c->time_event)
331 avahi_time_event_free(c->time_event);
332
333 AVAHI_LLIST_REMOVE(AvahiWideAreaCacheEntry, cache, c->engine->cache, c);
334
335 t = avahi_hashmap_lookup(c->engine->cache_by_key, c->record->key);
336 AVAHI_LLIST_REMOVE(AvahiWideAreaCacheEntry, by_key, t, c);
337 if (t)
338 avahi_hashmap_replace(c->engine->cache_by_key, avahi_key_ref(c->record->key), t);
339 else
340 avahi_hashmap_remove(c->engine->cache_by_key, c->record->key);
341
342 c->engine->cache_n_entries --;
343
344 avahi_record_unref(c->record);
345 avahi_free(c);
346 }
347
expiry_event(AvahiTimeEvent * te,void * userdata)348 static void expiry_event(AvahiTimeEvent *te, void *userdata) {
349 AvahiWideAreaCacheEntry *e = userdata;
350
351 assert(te);
352 assert(e);
353
354 cache_entry_free(e);
355 }
356
find_record_in_cache(AvahiWideAreaLookupEngine * e,AvahiRecord * r)357 static AvahiWideAreaCacheEntry* find_record_in_cache(AvahiWideAreaLookupEngine *e, AvahiRecord *r) {
358 AvahiWideAreaCacheEntry *c;
359
360 assert(e);
361 assert(r);
362
363 for (c = avahi_hashmap_lookup(e->cache_by_key, r->key); c; c = c->by_key_next)
364 if (avahi_record_equal_no_ttl(r, c->record))
365 return c;
366
367 return NULL;
368 }
369
run_callbacks(AvahiWideAreaLookupEngine * e,AvahiRecord * r)370 static void run_callbacks(AvahiWideAreaLookupEngine *e, AvahiRecord *r) {
371 AvahiWideAreaLookup *l;
372
373 assert(e);
374 assert(r);
375
376 for (l = avahi_hashmap_lookup(e->lookups_by_key, r->key); l; l = l->by_key_next) {
377 if (l->dead || !l->callback)
378 continue;
379
380 l->callback(e, AVAHI_BROWSER_NEW, AVAHI_LOOKUP_RESULT_WIDE_AREA, r, l->userdata);
381 }
382
383 if (r->key->clazz == AVAHI_DNS_CLASS_IN && r->key->type == AVAHI_DNS_TYPE_CNAME) {
384 /* It's a CNAME record, so we have to scan the all lookups to see if one matches */
385
386 for (l = e->lookups; l; l = l->lookups_next) {
387 AvahiKey *key;
388
389 if (l->dead || !l->callback)
390 continue;
391
392 if ((key = avahi_key_new_cname(l->key))) {
393 if (avahi_key_equal(r->key, key))
394 l->callback(e, AVAHI_BROWSER_NEW, AVAHI_LOOKUP_RESULT_WIDE_AREA, r, l->userdata);
395
396 avahi_key_unref(key);
397 }
398 }
399 }
400 }
401
add_to_cache(AvahiWideAreaLookupEngine * e,AvahiRecord * r)402 static void add_to_cache(AvahiWideAreaLookupEngine *e, AvahiRecord *r) {
403 AvahiWideAreaCacheEntry *c;
404 int is_new;
405
406 assert(e);
407 assert(r);
408
409 if ((c = find_record_in_cache(e, r))) {
410 is_new = 0;
411
412 /* Update the existing entry */
413 avahi_record_unref(c->record);
414 } else {
415 AvahiWideAreaCacheEntry *t;
416
417 is_new = 1;
418
419 /* Enforce cache size */
420 if (e->cache_n_entries >= CACHE_ENTRIES_MAX)
421 /* Eventually we should improve the caching algorithm here */
422 goto finish;
423
424 c = avahi_new(AvahiWideAreaCacheEntry, 1);
425 c->engine = e;
426 c->time_event = NULL;
427
428 AVAHI_LLIST_PREPEND(AvahiWideAreaCacheEntry, cache, e->cache, c);
429
430 /* Add the new entry to the cache entry hash table */
431 t = avahi_hashmap_lookup(e->cache_by_key, r->key);
432 AVAHI_LLIST_PREPEND(AvahiWideAreaCacheEntry, by_key, t, c);
433 avahi_hashmap_replace(e->cache_by_key, avahi_key_ref(r->key), t);
434
435 e->cache_n_entries ++;
436 }
437
438 c->record = avahi_record_ref(r);
439
440 gettimeofday(&c->timestamp, NULL);
441 c->expiry = c->timestamp;
442 avahi_timeval_add(&c->expiry, r->ttl * 1000000);
443
444 if (c->time_event)
445 avahi_time_event_update(c->time_event, &c->expiry);
446 else
447 c->time_event = avahi_time_event_new(e->server->time_event_queue, &c->expiry, expiry_event, c);
448
449 finish:
450
451 if (is_new)
452 run_callbacks(e, r);
453 }
454
map_dns_error(uint16_t error)455 static int map_dns_error(uint16_t error) {
456 static const int table[16] = {
457 AVAHI_OK,
458 AVAHI_ERR_DNS_FORMERR,
459 AVAHI_ERR_DNS_SERVFAIL,
460 AVAHI_ERR_DNS_NXDOMAIN,
461 AVAHI_ERR_DNS_NOTIMP,
462 AVAHI_ERR_DNS_REFUSED,
463 AVAHI_ERR_DNS_YXDOMAIN,
464 AVAHI_ERR_DNS_YXRRSET,
465 AVAHI_ERR_DNS_NXRRSET,
466 AVAHI_ERR_DNS_NOTAUTH,
467 AVAHI_ERR_DNS_NOTZONE,
468 AVAHI_ERR_INVALID_DNS_ERROR,
469 AVAHI_ERR_INVALID_DNS_ERROR,
470 AVAHI_ERR_INVALID_DNS_ERROR,
471 AVAHI_ERR_INVALID_DNS_ERROR,
472 AVAHI_ERR_INVALID_DNS_ERROR
473 };
474
475 assert(error <= 15);
476
477 return table[error];
478 }
479
handle_packet(AvahiWideAreaLookupEngine * e,AvahiDnsPacket * p)480 static void handle_packet(AvahiWideAreaLookupEngine *e, AvahiDnsPacket *p) {
481 AvahiWideAreaLookup *l = NULL;
482 int i, r;
483
484 AvahiBrowserEvent final_event = AVAHI_BROWSER_ALL_FOR_NOW;
485
486 assert(e);
487 assert(p);
488
489 /* Some superficial validity tests */
490 if (avahi_dns_packet_check_valid(p) < 0 || avahi_dns_packet_is_query(p)) {
491 avahi_log_warn(__FILE__": Ignoring invalid response for wide area datagram.");
492 goto finish;
493 }
494
495 /* Look for the lookup that issued this query */
496 if (!(l = find_lookup(e, avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ID))) || l->dead)
497 goto finish;
498
499 /* Check whether this a packet indicating a failure */
500 if ((r = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS) & 15) != 0 ||
501 avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) == 0) {
502
503 avahi_server_set_errno(e->server, r == 0 ? AVAHI_ERR_NOT_FOUND : map_dns_error(r));
504 /* Tell the user about the failure */
505 final_event = AVAHI_BROWSER_FAILURE;
506
507 /* We go on here, since some of the records contained in the
508 reply might be interesting in some way */
509 }
510
511 /* Skip over the question */
512 for (i = (int) avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_QDCOUNT); i > 0; i--) {
513 AvahiKey *k;
514
515 if (!(k = avahi_dns_packet_consume_key(p, NULL))) {
516 avahi_log_warn(__FILE__": Wide area response packet too short or invalid while reading question key. (Maybe a UTF-8 problem?)");
517 avahi_server_set_errno(e->server, AVAHI_ERR_INVALID_PACKET);
518 final_event = AVAHI_BROWSER_FAILURE;
519 goto finish;
520 }
521
522 avahi_key_unref(k);
523 }
524
525 /* Process responses */
526 for (i = (int) avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ANCOUNT) +
527 (int) avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_NSCOUNT) +
528 (int) avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ARCOUNT); i > 0; i--) {
529
530 AvahiRecord *rr;
531
532 if (!(rr = avahi_dns_packet_consume_record(p, NULL))) {
533 avahi_log_warn(__FILE__": Wide area response packet too short or invalid while reading response record. (Maybe a UTF-8 problem?)");
534 avahi_server_set_errno(e->server, AVAHI_ERR_INVALID_PACKET);
535 final_event = AVAHI_BROWSER_FAILURE;
536 goto finish;
537 }
538
539 add_to_cache(e, rr);
540 avahi_record_unref(rr);
541 }
542
543 finish:
544
545 if (l && !l->dead) {
546 if (l->callback)
547 l->callback(e, final_event, AVAHI_LOOKUP_RESULT_WIDE_AREA, NULL, l->userdata);
548
549 lookup_stop(l);
550 }
551 }
552
socket_event(AVAHI_GCC_UNUSED AvahiWatch * w,int fd,AVAHI_GCC_UNUSED AvahiWatchEvent events,void * userdata)553 static void socket_event(AVAHI_GCC_UNUSED AvahiWatch *w, int fd, AVAHI_GCC_UNUSED AvahiWatchEvent events, void *userdata) {
554 AvahiWideAreaLookupEngine *e = userdata;
555 AvahiDnsPacket *p = NULL;
556
557 if (fd == e->fd_ipv4)
558 p = avahi_recv_dns_packet_ipv4(e->fd_ipv4, NULL, NULL, NULL, NULL, NULL);
559 else {
560 assert(fd == e->fd_ipv6);
561 p = avahi_recv_dns_packet_ipv6(e->fd_ipv6, NULL, NULL, NULL, NULL, NULL);
562 }
563
564 if (p) {
565 handle_packet(e, p);
566 avahi_dns_packet_free(p);
567 }
568 }
569
avahi_wide_area_engine_new(AvahiServer * s)570 AvahiWideAreaLookupEngine *avahi_wide_area_engine_new(AvahiServer *s) {
571 AvahiWideAreaLookupEngine *e;
572
573 assert(s);
574
575 e = avahi_new(AvahiWideAreaLookupEngine, 1);
576 e->server = s;
577 e->cleanup_dead = 0;
578
579 /* Create sockets */
580 e->fd_ipv4 = s->config.use_ipv4 ? avahi_open_unicast_socket_ipv4() : -1;
581 e->fd_ipv6 = s->config.use_ipv6 ? avahi_open_unicast_socket_ipv6() : -1;
582
583 if (e->fd_ipv4 < 0 && e->fd_ipv6 < 0) {
584 avahi_log_error(__FILE__": Failed to create wide area sockets: %s", strerror(errno));
585
586 if (e->fd_ipv6 >= 0)
587 close(e->fd_ipv6);
588
589 if (e->fd_ipv4 >= 0)
590 close(e->fd_ipv4);
591
592 avahi_free(e);
593 return NULL;
594 }
595
596 /* Create watches */
597
598 e->watch_ipv4 = e->watch_ipv6 = NULL;
599
600 if (e->fd_ipv4 >= 0)
601 e->watch_ipv4 = s->poll_api->watch_new(e->server->poll_api, e->fd_ipv4, AVAHI_WATCH_IN, socket_event, e);
602 if (e->fd_ipv6 >= 0)
603 e->watch_ipv6 = s->poll_api->watch_new(e->server->poll_api, e->fd_ipv6, AVAHI_WATCH_IN, socket_event, e);
604
605 e->n_dns_servers = e->current_dns_server = 0;
606 e->next_id = (uint16_t) rand();
607
608 /* Initialize cache */
609 AVAHI_LLIST_HEAD_INIT(AvahiWideAreaCacheEntry, e->cache);
610 e->cache_by_key = avahi_hashmap_new((AvahiHashFunc) avahi_key_hash, (AvahiEqualFunc) avahi_key_equal, (AvahiFreeFunc) avahi_key_unref, NULL);
611 e->cache_n_entries = 0;
612
613 /* Initialize lookup list */
614 e->lookups_by_id = avahi_hashmap_new((AvahiHashFunc) avahi_int_hash, (AvahiEqualFunc) avahi_int_equal, NULL, NULL);
615 e->lookups_by_key = avahi_hashmap_new((AvahiHashFunc) avahi_key_hash, (AvahiEqualFunc) avahi_key_equal, (AvahiFreeFunc) avahi_key_unref, NULL);
616 AVAHI_LLIST_HEAD_INIT(AvahiWideAreaLookup, e->lookups);
617
618 return e;
619 }
620
avahi_wide_area_engine_free(AvahiWideAreaLookupEngine * e)621 void avahi_wide_area_engine_free(AvahiWideAreaLookupEngine *e) {
622 assert(e);
623
624 avahi_wide_area_clear_cache(e);
625
626 while (e->lookups)
627 lookup_destroy(e->lookups);
628
629 avahi_hashmap_free(e->cache_by_key);
630 avahi_hashmap_free(e->lookups_by_id);
631 avahi_hashmap_free(e->lookups_by_key);
632
633 if (e->watch_ipv4)
634 e->server->poll_api->watch_free(e->watch_ipv4);
635
636 if (e->watch_ipv6)
637 e->server->poll_api->watch_free(e->watch_ipv6);
638
639 if (e->fd_ipv6 >= 0)
640 close(e->fd_ipv6);
641
642 if (e->fd_ipv4 >= 0)
643 close(e->fd_ipv4);
644
645 avahi_free(e);
646 }
647
avahi_wide_area_clear_cache(AvahiWideAreaLookupEngine * e)648 void avahi_wide_area_clear_cache(AvahiWideAreaLookupEngine *e) {
649 assert(e);
650
651 while (e->cache)
652 cache_entry_free(e->cache);
653
654 assert(e->cache_n_entries == 0);
655 }
656
avahi_wide_area_set_servers(AvahiWideAreaLookupEngine * e,const AvahiAddress * a,unsigned n)657 void avahi_wide_area_set_servers(AvahiWideAreaLookupEngine *e, const AvahiAddress *a, unsigned n) {
658 assert(e);
659
660 if (a) {
661 for (e->n_dns_servers = 0; n > 0 && e->n_dns_servers < AVAHI_WIDE_AREA_SERVERS_MAX; a++, n--)
662 if ((a->proto == AVAHI_PROTO_INET && e->fd_ipv4 >= 0) || (a->proto == AVAHI_PROTO_INET6 && e->fd_ipv6 >= 0))
663 e->dns_servers[e->n_dns_servers++] = *a;
664 } else {
665 assert(n == 0);
666 e->n_dns_servers = 0;
667 }
668
669 e->current_dns_server = 0;
670
671 avahi_wide_area_clear_cache(e);
672 }
673
avahi_wide_area_cache_dump(AvahiWideAreaLookupEngine * e,AvahiDumpCallback callback,void * userdata)674 void avahi_wide_area_cache_dump(AvahiWideAreaLookupEngine *e, AvahiDumpCallback callback, void* userdata) {
675 AvahiWideAreaCacheEntry *c;
676
677 assert(e);
678 assert(callback);
679
680 callback(";; WIDE AREA CACHE ;;; ", userdata);
681
682 for (c = e->cache; c; c = c->cache_next) {
683 char *t = avahi_record_to_string(c->record);
684 callback(t, userdata);
685 avahi_free(t);
686 }
687 }
688
avahi_wide_area_scan_cache(AvahiWideAreaLookupEngine * e,AvahiKey * key,AvahiWideAreaLookupCallback callback,void * userdata)689 unsigned avahi_wide_area_scan_cache(AvahiWideAreaLookupEngine *e, AvahiKey *key, AvahiWideAreaLookupCallback callback, void *userdata) {
690 AvahiWideAreaCacheEntry *c;
691 AvahiKey *cname_key;
692 unsigned n = 0;
693
694 assert(e);
695 assert(key);
696 assert(callback);
697
698 for (c = avahi_hashmap_lookup(e->cache_by_key, key); c; c = c->by_key_next) {
699 callback(e, AVAHI_BROWSER_NEW, AVAHI_LOOKUP_RESULT_WIDE_AREA|AVAHI_LOOKUP_RESULT_CACHED, c->record, userdata);
700 n++;
701 }
702
703 if ((cname_key = avahi_key_new_cname(key))) {
704
705 for (c = avahi_hashmap_lookup(e->cache_by_key, cname_key); c; c = c->by_key_next) {
706 callback(e, AVAHI_BROWSER_NEW, AVAHI_LOOKUP_RESULT_WIDE_AREA|AVAHI_LOOKUP_RESULT_CACHED, c->record, userdata);
707 n++;
708 }
709
710 avahi_key_unref(cname_key);
711 }
712
713 return n;
714 }
715
avahi_wide_area_has_servers(AvahiWideAreaLookupEngine * e)716 int avahi_wide_area_has_servers(AvahiWideAreaLookupEngine *e) {
717 assert(e);
718
719 return e->n_dns_servers > 0;
720 }
721
722
723
724