]> git.meshlink.io Git - catta/blob - avahi-core/entry.c
split entry and entry-group management stuff off to its own file entry.c
[catta] / avahi-core / entry.c
1 /* $Id$ */
2
3 /***
4   This file is part of avahi.
5  
6   avahi is free software; you can redistribute it and/or modify it
7   under the terms of the GNU Lesser General Public License as
8   published by the Free Software Foundation; either version 2.1 of the
9   License, or (at your option) any later version.
10  
11   avahi is distributed in the hope that it will be useful, but WITHOUT
12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
13   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
14   Public License for more details.
15  
16   You should have received a copy of the GNU Lesser General Public
17   License along with avahi; if not, write to the Free Software
18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
19   USA.
20 ***/
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <sys/socket.h>
27 #include <arpa/inet.h>
28 #include <string.h>
29 #include <sys/utsname.h>
30 #include <unistd.h>
31 #include <errno.h>
32 #include <stdio.h>
33 #include <assert.h>
34 #include <stdlib.h>
35
36 #include <avahi-common/domain.h>
37 #include <avahi-common/timeval.h>
38 #include <avahi-common/malloc.h>
39 #include <avahi-common/error.h>
40
41 #include "server.h"
42 #include "iface.h"
43 #include "socket.h"
44 #include "browse.h"
45 #include "log.h"
46 #include "util.h"
47
48 void avahi_entry_free(AvahiServer*s, AvahiEntry *e) {
49     AvahiEntry *t;
50
51     assert(s);
52     assert(e);
53
54     avahi_goodbye_entry(s, e, 1, 1);
55
56     /* Remove from linked list */
57     AVAHI_LLIST_REMOVE(AvahiEntry, entries, s->entries, e);
58
59     /* Remove from hash table indexed by name */
60     t = avahi_hashmap_lookup(s->entries_by_key, e->record->key);
61     AVAHI_LLIST_REMOVE(AvahiEntry, by_key, t, e);
62     if (t)
63         avahi_hashmap_replace(s->entries_by_key, t->record->key, t);
64     else
65         avahi_hashmap_remove(s->entries_by_key, e->record->key);
66
67     /* Remove from associated group */
68     if (e->group)
69         AVAHI_LLIST_REMOVE(AvahiEntry, by_group, e->group->entries, e);
70
71     avahi_record_unref(e->record);
72     avahi_free(e);
73 }
74
75 void avahi_entry_group_free(AvahiServer *s, AvahiSEntryGroup *g) {
76     assert(s);
77     assert(g);
78
79     while (g->entries)
80         avahi_entry_free(s, g->entries);
81
82     if (g->register_time_event)
83         avahi_time_event_free(g->register_time_event);
84     
85     AVAHI_LLIST_REMOVE(AvahiSEntryGroup, groups, s->groups, g);
86     avahi_free(g);
87 }
88
89 void avahi_cleanup_dead_entries(AvahiServer *s) {
90     assert(s);
91
92     if (s->need_group_cleanup) {
93         AvahiSEntryGroup *g, *next;
94         
95         for (g = s->groups; g; g = next) {
96             next = g->groups_next;
97             
98             if (g->dead)
99                 avahi_entry_group_free(s, g);
100         }
101
102         s->need_group_cleanup = 0;
103     }
104
105     if (s->need_entry_cleanup) {
106         AvahiEntry *e, *next;
107         
108         for (e = s->entries; e; e = next) {
109             next = e->entries_next;
110             
111             if (e->dead)
112                 avahi_entry_free(s, e);
113         }
114
115         s->need_entry_cleanup = 0;
116     }
117
118     if (s->need_browser_cleanup)
119         avahi_browser_cleanup(s);
120 }
121
122 static int check_record_conflict(AvahiServer *s, AvahiIfIndex interface, AvahiProtocol protocol, AvahiRecord *r, AvahiPublishFlags flags) {
123     AvahiEntry *e;
124     
125     assert(s);
126     assert(r);
127
128     for (e = avahi_hashmap_lookup(s->entries_by_key, r->key); e; e = e->by_key_next) {
129         if (e->dead)
130             continue;
131
132         if (!(flags & AVAHI_PUBLISH_UNIQUE) && !(e->flags & AVAHI_PUBLISH_UNIQUE))
133             continue;
134         
135         if ((flags & AVAHI_PUBLISH_ALLOW_MULTIPLE) && (e->flags & AVAHI_PUBLISH_ALLOW_MULTIPLE) )
136             continue;
137
138         if ((interface <= 0 ||
139              e->interface <= 0 ||
140              e->interface == interface) &&
141             (protocol == AVAHI_PROTO_UNSPEC ||
142              e->protocol == AVAHI_PROTO_UNSPEC ||
143              e->protocol == protocol))
144
145             return -1;
146     }
147
148     return 0;
149 }
150
151 int avahi_server_add(
152     AvahiServer *s,
153     AvahiSEntryGroup *g,
154     AvahiIfIndex interface,
155     AvahiProtocol protocol,
156     AvahiPublishFlags flags,
157     AvahiRecord *r) {
158     
159     assert(s);
160     assert(r);
161
162     if (!AVAHI_IF_VALID(interface))
163         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
164
165     if (!AVAHI_PROTO_VALID(protocol))
166         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
167     
168     if (!AVAHI_FLAGS_VALID(
169             flags,
170             AVAHI_PUBLISH_NO_ANNOUNCE|
171             AVAHI_PUBLISH_NO_PROBE|
172             AVAHI_PUBLISH_UNIQUE|
173             AVAHI_PUBLISH_ALLOW_MULTIPLE|
174             AVAHI_PUBLISH_IS_PROXY|
175             AVAHI_PUBLISH_UPDATE))
176         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
177     
178     if (!avahi_is_valid_domain_name(r->key->name))
179         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
180     
181     if (r->ttl == 0)
182         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_TTL);
183
184     if (avahi_key_is_pattern(r->key))
185         return avahi_server_set_errno(s, AVAHI_ERR_IS_PATTERN);
186
187     if (!avahi_record_is_valid(r))
188         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_RECORD);
189
190     if (flags & AVAHI_PUBLISH_UPDATE) {
191         AvahiEntry *e;
192         AvahiRecord *old_record;
193         int is_first = 1;
194         
195         /* Update and existing record */
196
197         /* Find the first matching entry */
198         for (e = avahi_hashmap_lookup(s->entries_by_key, r->key); e; e = e->by_key_next) {
199             if (!e->dead && e->group == g && e->interface == interface && e->protocol == protocol)
200                 break;
201
202             is_first = 0;
203         }
204
205         /* Hmm, nothing found? */
206         if (!e)
207             return avahi_server_set_errno(s, AVAHI_ERR_NOT_FOUND);
208
209         /* Update the entry */
210         old_record = e->record;
211         e->record = avahi_record_ref(r);
212         e->flags = flags;
213
214         /* Announce our changes when needed */
215         if (!avahi_record_equal_no_ttl(old_record, r) && (!g || g->state != AVAHI_ENTRY_GROUP_UNCOMMITED)) {
216
217             /* Remove the old entry from all caches, if needed */
218             if (!(e->flags & AVAHI_PUBLISH_UNIQUE))
219                 avahi_goodbye_entry(s, e, 1, 0);
220
221             /* Reannounce our updated entry */
222             avahi_reannounce_entry(s, e);
223         }
224
225         /* If we were the first entry in the list, we need to update the key */
226         if (is_first)
227             avahi_hashmap_replace(s->entries_by_key, e->record->key, e);
228         
229         avahi_record_unref(old_record);
230
231     } else {
232         AvahiEntry *e, *t;
233
234         /* Add a new record */
235     
236         if (check_record_conflict(s, interface, protocol, r, flags) < 0)
237             return avahi_server_set_errno(s, AVAHI_ERR_LOCAL_COLLISION);
238
239         if (!(e = avahi_new(AvahiEntry, 1)))
240             return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
241         
242         e->server = s;
243         e->record = avahi_record_ref(r);
244         e->group = g;
245         e->interface = interface;
246         e->protocol = protocol;
247         e->flags = flags;
248         e->dead = 0;
249         
250         AVAHI_LLIST_HEAD_INIT(AvahiAnnouncer, e->announcers);
251         
252         AVAHI_LLIST_PREPEND(AvahiEntry, entries, s->entries, e);
253         
254         /* Insert into hash table indexed by name */
255         t = avahi_hashmap_lookup(s->entries_by_key, e->record->key);
256         AVAHI_LLIST_PREPEND(AvahiEntry, by_key, t, e);
257         avahi_hashmap_replace(s->entries_by_key, e->record->key, t);
258         
259         /* Insert into group list */
260         if (g)
261             AVAHI_LLIST_PREPEND(AvahiEntry, by_group, g->entries, e); 
262         
263         avahi_announce_entry(s, e);
264     }
265
266     return AVAHI_OK;
267 }
268
269 const AvahiRecord *avahi_server_iterate(AvahiServer *s, AvahiSEntryGroup *g, void **state) {
270     AvahiEntry **e = (AvahiEntry**) state;
271     assert(s);
272     assert(e);
273
274     if (!*e)
275         *e = g ? g->entries : s->entries;
276     
277     while (*e && (*e)->dead)
278         *e = g ? (*e)->by_group_next : (*e)->entries_next;
279         
280     if (!*e)
281         return NULL;
282
283     return avahi_record_ref((*e)->record);
284 }
285
286 int avahi_server_dump(AvahiServer *s, AvahiDumpCallback callback, void* userdata) {
287     AvahiEntry *e;
288     
289     assert(s);
290     assert(callback);
291
292     callback(";;; ZONE DUMP FOLLOWS ;;;", userdata);
293
294     for (e = s->entries; e; e = e->entries_next) {
295         char *t;
296         char ln[256];
297
298         if (e->dead)
299             continue;
300         
301         if (!(t = avahi_record_to_string(e->record)))
302             return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
303         
304         snprintf(ln, sizeof(ln), "%s ; iface=%i proto=%i", t, e->interface, e->protocol);
305         avahi_free(t);
306
307         callback(ln, userdata);
308     }
309
310     avahi_dump_caches(s->monitor, callback, userdata);
311
312     if (s->wide_area_lookup_engine)
313         avahi_wide_area_cache_dump(s->wide_area_lookup_engine, callback, userdata);
314     return AVAHI_OK;
315 }
316
317 int avahi_server_add_ptr(
318     AvahiServer *s,
319     AvahiSEntryGroup *g,
320     AvahiIfIndex interface,
321     AvahiProtocol protocol,
322     AvahiPublishFlags flags,
323     uint32_t ttl,
324     const char *name,
325     const char *dest) {
326
327     AvahiRecord *r;
328     int ret;
329
330     assert(s);
331     assert(dest);
332
333     if ((name && !avahi_is_valid_domain_name(name)) || !avahi_is_valid_domain_name(dest))
334         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
335
336     if (!name)
337         name = s->host_name_fqdn;
338     
339     if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_PTR, ttl)))
340         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
341         
342     r->data.ptr.name = avahi_normalize_name_strdup(dest);
343     ret = avahi_server_add(s, g, interface, protocol, flags, r);
344     avahi_record_unref(r);
345     return ret;
346 }
347
348 int avahi_server_add_address(
349     AvahiServer *s,
350     AvahiSEntryGroup *g,
351     AvahiIfIndex interface,
352     AvahiProtocol protocol,
353     AvahiPublishFlags flags,
354     const char *name,
355     AvahiAddress *a) {
356
357     char *n = NULL;
358     int ret = AVAHI_OK;
359     
360     assert(s);
361     assert(a);
362
363     if (!AVAHI_IF_VALID(interface))
364         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
365
366     if (!AVAHI_PROTO_VALID(protocol) || !AVAHI_PROTO_VALID(a->proto))
367         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
368     
369     if (!AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_REVERSE|AVAHI_PUBLISH_NO_ANNOUNCE|AVAHI_PUBLISH_NO_PROBE|AVAHI_PUBLISH_UPDATE))
370         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
371     
372     if (name && !avahi_is_valid_domain_name(name))
373         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
374
375     if (!name)
376         name = s->host_name_fqdn;
377     else {
378         if (!(n = avahi_normalize_name_strdup(name)))
379             return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
380
381         name = n;
382     }
383     
384     if (a->proto == AVAHI_PROTO_INET) {
385         AvahiRecord  *r;
386
387         if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_A, AVAHI_DEFAULT_TTL_HOST_NAME))) {
388             ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
389             goto fail;
390         }
391         
392         r->data.a.address = a->data.ipv4;
393         ret = avahi_server_add(s, g, interface, protocol, (flags & ~ AVAHI_PUBLISH_NO_REVERSE) | AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
394         avahi_record_unref(r);
395
396         if (ret < 0)
397             goto fail;
398
399         if (!(flags & AVAHI_PUBLISH_NO_REVERSE)) {
400             char *reverse;
401             
402             if (!(reverse = avahi_reverse_lookup_name_ipv4(&a->data.ipv4))) {
403                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
404                 goto fail;
405             }
406             
407             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
408             avahi_free(reverse);
409         }
410         
411     } else {
412         AvahiRecord *r;
413
414         assert(a->proto == AVAHI_PROTO_INET6);
415             
416         if (!(r = avahi_record_new_full(name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_AAAA, AVAHI_DEFAULT_TTL_HOST_NAME))) {
417             ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
418             goto fail;
419         }
420         
421         r->data.aaaa.address = a->data.ipv6;
422         ret = avahi_server_add(s, g, interface, protocol, (flags & ~ AVAHI_PUBLISH_NO_REVERSE) | AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
423         avahi_record_unref(r);
424
425         if (ret < 0)
426             goto fail;
427
428         if (!(flags & AVAHI_PUBLISH_NO_REVERSE)) {
429             char *reverse;
430
431             if (!(reverse = avahi_reverse_lookup_name_ipv6_arpa(&a->data.ipv6))) {
432                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
433                 goto fail;
434             }
435             
436             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
437             avahi_free(reverse);
438             
439             if (ret < 0)
440                 goto fail;
441             
442             if (!(reverse = avahi_reverse_lookup_name_ipv6_int(&a->data.ipv6))) {
443                 ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
444                 goto fail;
445             }
446             
447             ret = avahi_server_add_ptr(s, g, interface, protocol, flags | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL_HOST_NAME, reverse, name);
448             avahi_free(reverse);
449         }
450     }
451
452 fail:
453     
454     avahi_free(n);
455
456     return ret;
457 }
458
459 static int server_add_txt_strlst_nocopy(
460     AvahiServer *s,
461     AvahiSEntryGroup *g,
462     AvahiIfIndex interface,
463     AvahiProtocol protocol,
464     AvahiPublishFlags flags,
465     uint32_t ttl,
466     const char *name,
467     AvahiStringList *strlst) {
468
469     AvahiRecord *r;
470     int ret;
471     
472     assert(s);
473
474     if (!(r = avahi_record_new_full(name ? name : s->host_name_fqdn, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_TXT, ttl))) {
475         avahi_string_list_free(strlst);
476         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
477     }
478     
479     r->data.txt.string_list = strlst;
480     ret = avahi_server_add(s, g, interface, protocol, flags, r);
481     avahi_record_unref(r);
482
483     return ret;
484 }
485
486 int avahi_server_add_txt_strlst(
487     AvahiServer *s,
488     AvahiSEntryGroup *g,
489     AvahiIfIndex interface,
490     AvahiProtocol protocol,
491     AvahiPublishFlags flags,
492     uint32_t ttl,
493     const char *name,
494     AvahiStringList *strlst) {
495
496     assert(s);
497
498     return server_add_txt_strlst_nocopy(s, g, interface, protocol, flags, ttl, name, avahi_string_list_copy(strlst));
499 }
500
501 int avahi_server_add_txt_va(
502     AvahiServer *s,
503     AvahiSEntryGroup *g,
504     AvahiIfIndex interface,
505     AvahiProtocol protocol,
506     AvahiPublishFlags flags,
507     uint32_t ttl,
508     const char *name,
509     va_list va) {
510
511     assert(s);
512
513     return server_add_txt_strlst_nocopy(s, g, interface, protocol, flags, ttl, name, avahi_string_list_new_va(va));
514 }
515
516 int avahi_server_add_txt(
517     AvahiServer *s,
518     AvahiSEntryGroup *g,
519     AvahiIfIndex interface,
520     AvahiProtocol protocol,
521     AvahiPublishFlags flags,
522     uint32_t ttl,
523     const char *name,
524     ...) {
525
526     va_list va;
527     int ret;
528     
529     assert(s);
530
531     va_start(va, name);
532     ret = avahi_server_add_txt_va(s, g, interface, protocol, flags, ttl, name, va);
533     va_end(va);
534
535     return ret;
536 }
537
538 static AvahiStringList *add_magic_cookie(
539     AvahiServer *s,
540     AvahiStringList *strlst) {
541
542     assert(s);
543
544     if (!s->config.add_service_cookie)
545         return strlst;
546
547     if (avahi_string_list_find(strlst, AVAHI_SERVICE_COOKIE))
548         /* This string list already contains a magic cookie */
549         return strlst;
550
551     return avahi_string_list_add_printf(strlst, AVAHI_SERVICE_COOKIE"=%u", s->local_service_cookie);
552 }
553
554 static int server_add_service_strlst_nocopy(
555     AvahiServer *s,
556     AvahiSEntryGroup *g,
557     AvahiIfIndex interface,
558     AvahiProtocol protocol,
559     AvahiPublishFlags flags,
560     const char *name,
561     const char *type,
562     const char *domain,
563     const char *host,
564     uint16_t port,
565     AvahiStringList *strlst) {
566
567     char ptr_name[AVAHI_DOMAIN_NAME_MAX], svc_name[AVAHI_DOMAIN_NAME_MAX], enum_ptr[AVAHI_DOMAIN_NAME_MAX], *h = NULL;
568     AvahiRecord *r = NULL;
569     int ret = AVAHI_OK;
570     
571     assert(s);
572     assert(type);
573     assert(name);
574
575     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
576     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
577     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_NO_COOKIE|AVAHI_PUBLISH_IS_PROXY|AVAHI_PUBLISH_UPDATE), AVAHI_ERR_INVALID_FLAGS);
578     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
579     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
580     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
581     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !host || avahi_is_valid_domain_name(host), AVAHI_ERR_INVALID_HOST_NAME);
582
583     if (!domain)
584         domain = s->domain_name;
585
586     if (!host)
587         host = s->host_name_fqdn;
588
589     if (!(h = avahi_normalize_name_strdup(host))) {
590         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
591         goto fail;
592     }
593
594     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0 ||
595         (ret = avahi_service_name_join(ptr_name, sizeof(ptr_name), NULL, type, domain)) < 0 ||
596         (ret = avahi_service_name_join(enum_ptr, sizeof(enum_ptr), NULL, "_services._dns-sd._udp", domain)) < 0) {
597         avahi_server_set_errno(s, ret);
598         goto fail;
599     }
600
601     /* Add service enumeration PTR record */
602     
603     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, flags & AVAHI_PUBLISH_IS_PROXY, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
604         goto fail;
605
606     /* Add SRV record */
607     
608     if (!(r = avahi_record_new_full(svc_name, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
609         ret = avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
610         goto fail;
611     }
612     
613     r->data.srv.priority = 0;
614     r->data.srv.weight = 0;
615     r->data.srv.port = port;
616     r->data.srv.name = h;
617     h = NULL;
618     ret = avahi_server_add(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE, r);
619     avahi_record_unref(r);
620
621     if (ret < 0)
622         goto fail;
623
624     /* Add TXT record */
625
626     if (!(flags & AVAHI_PUBLISH_NO_COOKIE))
627         strlst = add_magic_cookie(s, strlst);
628     
629     ret = server_add_txt_strlst_nocopy(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE, AVAHI_DEFAULT_TTL, svc_name, strlst);
630     strlst = NULL;
631
632     if (ret < 0)
633         goto fail;
634
635     /* Add service type enumeration record */
636     
637     ret = avahi_server_add_ptr(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY), AVAHI_DEFAULT_TTL, enum_ptr, ptr_name);
638
639 fail:
640     
641     avahi_string_list_free(strlst);
642     avahi_free(h);
643     
644     return ret;
645 }
646
647 int avahi_server_add_service_strlst(
648     AvahiServer *s,
649     AvahiSEntryGroup *g,
650     AvahiIfIndex interface,
651     AvahiProtocol protocol,
652     AvahiPublishFlags flags,
653     const char *name,
654     const char *type,
655     const char *domain,
656     const char *host,
657     uint16_t port,
658     AvahiStringList *strlst) {
659
660     assert(s);
661     assert(type);
662     assert(name);
663
664     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_copy(strlst));
665 }
666
667 int avahi_server_add_service_va(
668     AvahiServer *s,
669     AvahiSEntryGroup *g,
670     AvahiIfIndex interface,
671     AvahiProtocol protocol,
672     AvahiPublishFlags flags,
673     const char *name,
674     const char *type,
675     const char *domain,
676     const char *host,
677     uint16_t port,
678     va_list va) {
679
680     assert(s);
681     assert(type);
682     assert(name);
683
684     return server_add_service_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, host, port, avahi_string_list_new_va(va));
685 }
686
687 int avahi_server_add_service(
688     AvahiServer *s,
689     AvahiSEntryGroup *g,
690     AvahiIfIndex interface,
691     AvahiProtocol protocol,
692     AvahiPublishFlags flags,
693     const char *name,
694     const char *type,
695     const char *domain,
696     const char *host,
697     uint16_t port,
698     ... ){
699
700     va_list va;
701     int ret;
702     
703     va_start(va, port);
704     ret = avahi_server_add_service_va(s, g, interface, protocol, flags, name, type, domain, host, port, va);
705     va_end(va);
706     
707     return ret;
708 }
709
710 static int server_update_service_txt_strlst_nocopy(
711     AvahiServer *s,
712     AvahiSEntryGroup *g,
713     AvahiIfIndex interface,
714     AvahiProtocol protocol,
715     AvahiPublishFlags flags,
716     const char *name,     
717     const char *type,     
718     const char *domain,   
719     AvahiStringList *strlst) {
720
721     char svc_name[AVAHI_DOMAIN_NAME_MAX];
722     int ret = AVAHI_OK;
723     
724     assert(s);
725     assert(type);
726     assert(name);
727
728     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
729     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
730     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, AVAHI_PUBLISH_IS_PROXY|AVAHI_PUBLISH_NO_COOKIE), AVAHI_ERR_INVALID_FLAGS);
731     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
732     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
733     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
734
735     if (!domain)
736         domain = s->domain_name;
737
738     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0) {
739         avahi_server_set_errno(s, ret);
740         goto fail;
741     }
742
743     /* Add TXT record */
744     if (!(flags & AVAHI_PUBLISH_NO_COOKIE))
745         strlst = add_magic_cookie(s, strlst);
746     
747     ret = server_add_txt_strlst_nocopy(s, g, interface, protocol, (flags & AVAHI_PUBLISH_IS_PROXY) | AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_UPDATE, AVAHI_DEFAULT_TTL, svc_name, strlst);
748     strlst = NULL;
749
750 fail:
751     
752     avahi_string_list_free(strlst);
753     
754     return ret;
755 }
756
757 int avahi_server_update_service_txt_strlst(
758     AvahiServer *s,
759     AvahiSEntryGroup *g,
760     AvahiIfIndex interface,
761     AvahiProtocol protocol,
762     AvahiPublishFlags flags,
763     const char *name,     
764     const char *type,     
765     const char *domain,   
766     AvahiStringList *strlst) {
767
768     return server_update_service_txt_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, avahi_string_list_copy(strlst));
769 }
770
771 /** Update the TXT record for a service with the NULL terminated list of strings of the va_list. */
772 int avahi_server_update_service_txt_va(
773     AvahiServer *s,
774     AvahiSEntryGroup *g,
775     AvahiIfIndex interface,
776     AvahiProtocol protocol,
777     AvahiPublishFlags flags,
778     const char *name,     
779     const char *type,     
780     const char *domain,   
781     va_list va) {
782
783     return server_update_service_txt_strlst_nocopy(s, g, interface, protocol, flags, name, type, domain, avahi_string_list_new_va(va));
784 }
785
786 /** Update the TXT record for a service with the NULL termonate list of strings */
787 int avahi_server_update_service_txt(
788     AvahiServer *s,
789     AvahiSEntryGroup *g,
790     AvahiIfIndex interface,
791     AvahiProtocol protocol,
792     AvahiPublishFlags flags,
793     const char *name,     
794     const char *type,     
795     const char *domain,   
796     ...) {
797
798     va_list va;
799     int ret;
800     
801     va_start(va, domain);
802     ret = avahi_server_update_service_txt_va(s, g, interface, protocol, flags, name, type, domain, va);
803     va_end(va);
804     
805     return ret;
806 }
807
808 int avahi_server_add_service_subtype(
809     AvahiServer *s,
810     AvahiSEntryGroup *g,
811     AvahiIfIndex interface,
812     AvahiProtocol protocol,
813     AvahiPublishFlags flags,
814     const char *name,        
815     const char *type,        
816     const char *domain,      
817     const char *subtype) {
818
819     int ret = AVAHI_OK;
820     char svc_name[AVAHI_DOMAIN_NAME_MAX], ptr_name[AVAHI_DOMAIN_NAME_MAX];
821     
822     assert(name);
823     assert(type);
824     assert(subtype);
825
826     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE);
827     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL);
828     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, AVAHI_FLAGS_VALID(flags, 0), AVAHI_ERR_INVALID_FLAGS);
829     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_name(name), AVAHI_ERR_INVALID_SERVICE_NAME);
830     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_type_strict(type), AVAHI_ERR_INVALID_SERVICE_TYPE);
831     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
832     AVAHI_CHECK_VALIDITY_SET_RET_GOTO_FAIL(s, avahi_is_valid_service_subtype(subtype), AVAHI_ERR_INVALID_SERVICE_SUBTYPE);
833
834     if (!domain)
835         domain = s->domain_name;
836
837     if ((ret = avahi_service_name_join(svc_name, sizeof(svc_name), name, type, domain)) < 0 ||
838         (ret = avahi_service_name_join(ptr_name, sizeof(ptr_name), NULL, subtype, domain)) < 0) {
839         avahi_server_set_errno(s, ret);
840         goto fail;
841     }
842
843     if ((ret = avahi_server_add_ptr(s, g, interface, protocol, flags & AVAHI_PUBLISH_IS_PROXY, AVAHI_DEFAULT_TTL, ptr_name, svc_name)) < 0)
844         goto fail;
845
846 fail:
847     
848     return ret;
849 }
850
851 static void hexstring(char *s, size_t sl, const void *p, size_t pl) {
852     static const char hex[] = "0123456789abcdef";
853     int b = 0;
854     const uint8_t *k = p;
855
856     while (sl > 1 && pl > 0) {
857         *(s++) = hex[(b ? *k : *k >> 4) & 0xF];
858
859         if (b) {
860             k++;
861             pl--;
862         }
863         
864         b = !b;
865
866         sl--;
867     }
868
869     if (sl > 0)
870         *s = 0;
871 }
872
873 int avahi_server_add_dns_server_address(
874     AvahiServer *s,
875     AvahiSEntryGroup *g,
876     AvahiIfIndex interface,
877     AvahiProtocol protocol,
878     AvahiPublishFlags flags,
879     const char *domain,
880     AvahiDNSServerType type,
881     const AvahiAddress *address,
882     uint16_t port /** should be 53 */) {
883
884     AvahiRecord *r;
885     int ret;
886     char n[64], h[64];
887
888     assert(s);
889     assert(address);
890
891     if (!AVAHI_IF_VALID(interface))
892         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
893
894     if (!AVAHI_PROTO_VALID(protocol) || !AVAHI_PROTO_VALID(address->proto))
895         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
896     
897     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
898         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
899     
900     if (port == 0)
901         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
902     
903     if (domain && !avahi_is_valid_domain_name(domain))
904         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
905
906     if (!domain)
907         domain = s->domain_name;
908
909     if (address->proto == AVAHI_PROTO_INET) {
910         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv4Address));
911         snprintf(n, sizeof(n), "ip-%s.%s", h, domain);
912         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_A, AVAHI_DEFAULT_TTL_HOST_NAME);
913         r->data.a.address = address->data.ipv4;
914     } else {
915         hexstring(h, sizeof(h), &address->data, sizeof(AvahiIPv6Address));
916         snprintf(n, sizeof(n), "ip6-%s.%s", h, domain);
917         r = avahi_record_new_full(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_AAAA, AVAHI_DEFAULT_TTL_HOST_NAME);
918         r->data.aaaa.address = address->data.ipv6;
919     }
920
921     if (!r)
922         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
923     
924     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_UNIQUE | AVAHI_PUBLISH_ALLOW_MULTIPLE, r);
925     avahi_record_unref(r);
926
927     if (ret < 0)
928         return ret;
929     
930     return avahi_server_add_dns_server_name(s, g, interface, protocol, flags, domain, type, n, port);
931 }
932
933 int avahi_server_add_dns_server_name(
934     AvahiServer *s,
935     AvahiSEntryGroup *g,
936     AvahiIfIndex interface,
937     AvahiProtocol protocol,
938     AvahiPublishFlags flags,
939     const char *domain,
940     AvahiDNSServerType type,
941     const char *name,
942     uint16_t port /** should be 53 */) {
943
944     int ret = -1;
945     char t[256], *d = NULL, *n = NULL;
946     AvahiRecord *r;
947     
948     assert(s);
949     assert(name);
950
951     if (!AVAHI_IF_VALID(interface))
952         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_INTERFACE);
953
954     if (!AVAHI_PROTO_VALID(protocol))
955         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PROTOCOL);
956
957     if (!AVAHI_FLAGS_VALID(flags, 0) || (type != AVAHI_DNS_SERVER_UPDATE && type != AVAHI_DNS_SERVER_RESOLVE))
958         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_FLAGS);
959     
960     if (port == 0)
961         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_PORT);
962
963     if (!avahi_is_valid_domain_name(name))
964         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_HOST_NAME);
965
966     if (domain && !avahi_is_valid_domain_name(domain))
967         return avahi_server_set_errno(s, AVAHI_ERR_INVALID_DOMAIN_NAME);
968     
969     if (!domain)
970         domain = s->domain_name;
971
972     if (!(n = avahi_normalize_name_strdup(name)) ||
973         !(d = avahi_normalize_name_strdup(domain))) {
974         avahi_free(n);
975         avahi_free(d);
976         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
977     }
978
979     snprintf(t, sizeof(t), "%s.%s", type == AVAHI_DNS_SERVER_RESOLVE ? "_domain._udp" : "_dns-update._udp", d);
980     avahi_free(d);
981     
982     if (!(r = avahi_record_new_full(t, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV, AVAHI_DEFAULT_TTL_HOST_NAME))) {
983         avahi_free(n);
984         return avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
985     }
986     
987     r->data.srv.priority = 0;
988     r->data.srv.weight = 0;
989     r->data.srv.port = port;
990     r->data.srv.name = n;
991     ret = avahi_server_add(s, g, interface, protocol, AVAHI_PUBLISH_NULL, r);
992     avahi_record_unref(r);
993
994     return ret;
995 }
996
997 void avahi_s_entry_group_change_state(AvahiSEntryGroup *g, AvahiEntryGroupState state) {
998     assert(g);
999
1000     if (g->state == state)
1001         return;
1002
1003     assert(state <= AVAHI_ENTRY_GROUP_COLLISION);
1004
1005     if (g->state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
1006
1007         /* If the entry group was established for a time longer then
1008          * 5s, reset the establishment trial counter */
1009         
1010         if (avahi_age(&g->established_at) > 5000000)
1011             g->n_register_try = 0;
1012     }
1013     
1014     if (state == AVAHI_ENTRY_GROUP_ESTABLISHED)
1015
1016         /* If the entry group is now established, remember the time
1017          * this happened */
1018         
1019         gettimeofday(&g->established_at, NULL);
1020     
1021     g->state = state;
1022     
1023     if (g->callback)
1024         g->callback(g->server, g, state, g->userdata);
1025 }
1026
1027 AvahiSEntryGroup *avahi_s_entry_group_new(AvahiServer *s, AvahiSEntryGroupCallback callback, void* userdata) {
1028     AvahiSEntryGroup *g;
1029     
1030     assert(s);
1031
1032     if (!(g = avahi_new(AvahiSEntryGroup, 1))) {
1033         avahi_server_set_errno(s, AVAHI_ERR_NO_MEMORY);
1034         return NULL;
1035     }
1036     
1037     g->server = s;
1038     g->callback = callback;
1039     g->userdata = userdata;
1040     g->dead = 0;
1041     g->state = AVAHI_ENTRY_GROUP_UNCOMMITED;
1042     g->n_probing = 0;
1043     g->n_register_try = 0;
1044     g->register_time_event = NULL;
1045     g->register_time.tv_sec = 0;
1046     g->register_time.tv_usec = 0;
1047     AVAHI_LLIST_HEAD_INIT(AvahiEntry, g->entries);
1048
1049     AVAHI_LLIST_PREPEND(AvahiSEntryGroup, groups, s->groups, g);
1050     return g;
1051 }
1052
1053 void avahi_s_entry_group_free(AvahiSEntryGroup *g) {
1054     AvahiEntry *e;
1055     
1056     assert(g);
1057     assert(g->server);
1058
1059     for (e = g->entries; e; e = e->by_group_next) {
1060         if (!e->dead) {
1061             avahi_goodbye_entry(g->server, e, 1, 1);
1062             e->dead = 1;
1063         }
1064     }
1065
1066     if (g->register_time_event) {
1067         avahi_time_event_free(g->register_time_event);
1068         g->register_time_event = NULL;
1069     }
1070
1071     g->dead = 1;
1072     
1073     g->server->need_group_cleanup = 1;
1074     g->server->need_entry_cleanup = 1;
1075 }
1076
1077 static void entry_group_commit_real(AvahiSEntryGroup *g) {
1078     assert(g);
1079
1080     gettimeofday(&g->register_time, NULL);
1081
1082     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
1083
1084     if (!g->dead) {
1085         avahi_announce_group(g->server, g);
1086         avahi_s_entry_group_check_probed(g, 0);
1087     }
1088 }
1089
1090 static void entry_group_register_time_event_callback(AvahiTimeEvent *e, void* userdata) {
1091     AvahiSEntryGroup *g = userdata;
1092     assert(g);
1093
1094 /*     avahi_log_debug("Holdoff passed, waking up and going on."); */
1095
1096     avahi_time_event_free(g->register_time_event);
1097     g->register_time_event = NULL;
1098     
1099     /* Holdoff time passed, so let's start probing */
1100     entry_group_commit_real(g);
1101 }
1102
1103 int avahi_s_entry_group_commit(AvahiSEntryGroup *g) {
1104     struct timeval now;
1105     
1106     assert(g);
1107     assert(!g->dead);
1108
1109     if (g->state != AVAHI_ENTRY_GROUP_UNCOMMITED && g->state != AVAHI_ENTRY_GROUP_COLLISION)
1110         return avahi_server_set_errno(g->server, AVAHI_ERR_BAD_STATE);
1111
1112     g->n_register_try++;
1113
1114     avahi_timeval_add(&g->register_time,
1115                       1000*(g->n_register_try >= AVAHI_RR_RATE_LIMIT_COUNT ?
1116                             AVAHI_RR_HOLDOFF_MSEC_RATE_LIMIT :
1117                             AVAHI_RR_HOLDOFF_MSEC));
1118
1119     gettimeofday(&now, NULL);
1120
1121     if (avahi_timeval_compare(&g->register_time, &now) <= 0) {
1122         /* Holdoff time passed, so let's start probing */
1123 /*         avahi_log_debug("Holdoff passed, directly going on.");  */
1124
1125         entry_group_commit_real(g);
1126     } else {
1127 /*          avahi_log_debug("Holdoff not passed, sleeping.");  */
1128
1129          /* Holdoff time has not yet passed, so let's wait */
1130         assert(!g->register_time_event);
1131         g->register_time_event = avahi_time_event_new(g->server->time_event_queue, &g->register_time, entry_group_register_time_event_callback, g);
1132         
1133         avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_REGISTERING);
1134     }
1135
1136     return AVAHI_OK;
1137 }
1138
1139 void avahi_s_entry_group_reset(AvahiSEntryGroup *g) {
1140     AvahiEntry *e;
1141     assert(g);
1142     
1143     for (e = g->entries; e; e = e->by_group_next) {
1144         if (!e->dead) {
1145             avahi_goodbye_entry(g->server, e, 1, 1);
1146             e->dead = 1;
1147         }
1148     }
1149     g->server->need_entry_cleanup = 1;
1150
1151     if (g->register_time_event) {
1152         avahi_time_event_free(g->register_time_event);
1153         g->register_time_event = NULL;
1154     }
1155     
1156     g->n_probing = 0;
1157
1158     gettimeofday(&g->register_time, NULL);
1159
1160     avahi_s_entry_group_change_state(g, AVAHI_ENTRY_GROUP_UNCOMMITED);
1161 }
1162
1163 int avahi_entry_is_commited(AvahiEntry *e) {
1164     assert(e);
1165     assert(!e->dead);
1166
1167     return !e->group ||
1168         e->group->state == AVAHI_ENTRY_GROUP_REGISTERING ||
1169         e->group->state == AVAHI_ENTRY_GROUP_ESTABLISHED;
1170 }
1171
1172 AvahiEntryGroupState avahi_s_entry_group_get_state(AvahiSEntryGroup *g) {
1173     assert(g);
1174     assert(!g->dead);
1175
1176     return g->state;
1177 }
1178
1179 void avahi_s_entry_group_set_data(AvahiSEntryGroup *g, void* userdata) {
1180     assert(g);
1181
1182     g->userdata = userdata;
1183 }
1184
1185 void* avahi_s_entry_group_get_data(AvahiSEntryGroup *g) {
1186     assert(g);
1187
1188     return g->userdata;
1189 }
1190
1191 int avahi_s_entry_group_is_empty(AvahiSEntryGroup *g) {
1192     AvahiEntry *e;
1193     assert(g);
1194
1195     /* Look for an entry that is not dead */
1196     for (e = g->entries; e; e = e->by_group_next)
1197         if (!e->dead)
1198             return 0;
1199
1200     return 1;
1201 }