4 This file is part of avahi.
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.
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.
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
28 #include <sys/types.h>
29 #include <sys/socket.h>
30 #include <arpa/inet.h>
33 #include <avahi-common/domain.h>
34 #include <avahi-common/malloc.h>
41 AvahiKey *avahi_key_new(const char *name, uint16_t class, uint16_t type) {
45 if (!(k = avahi_new(AvahiKey, 1))) {
46 avahi_log_error("avahi_new() failed.");
50 if (!(k->name = avahi_normalize_name_strdup(name))) {
51 avahi_log_error("avahi_normalize_name() failed.");
63 AvahiKey *avahi_key_new_cname(AvahiKey *key) {
66 if (key->clazz != AVAHI_DNS_CLASS_IN)
69 if (key->type == AVAHI_DNS_TYPE_CNAME)
72 return avahi_key_new(key->name, key->clazz, AVAHI_DNS_TYPE_CNAME);
75 AvahiKey *avahi_key_ref(AvahiKey *k) {
84 void avahi_key_unref(AvahiKey *k) {
88 if ((--k->ref) <= 0) {
94 AvahiRecord *avahi_record_new(AvahiKey *k, uint32_t ttl) {
99 if (!(r = avahi_new(AvahiRecord, 1))) {
100 avahi_log_error("avahi_new() failed.");
105 r->key = avahi_key_ref(k);
107 memset(&r->data, 0, sizeof(r->data));
109 r->ttl = ttl != (uint32_t) -1 ? ttl : AVAHI_DEFAULT_TTL;
114 AvahiRecord *avahi_record_new_full(const char *name, uint16_t class, uint16_t type, uint32_t ttl) {
120 if (!(k = avahi_key_new(name, class, type))) {
121 avahi_log_error("avahi_key_new() failed.");
125 r = avahi_record_new(k, ttl);
129 avahi_log_error("avahi_record_new() failed.");
136 AvahiRecord *avahi_record_ref(AvahiRecord *r) {
144 void avahi_record_unref(AvahiRecord *r) {
148 if ((--r->ref) <= 0) {
149 switch (r->key->type) {
151 case AVAHI_DNS_TYPE_SRV:
152 avahi_free(r->data.srv.name);
155 case AVAHI_DNS_TYPE_PTR:
156 case AVAHI_DNS_TYPE_CNAME:
157 case AVAHI_DNS_TYPE_NS:
158 avahi_free(r->data.ptr.name);
161 case AVAHI_DNS_TYPE_HINFO:
162 avahi_free(r->data.hinfo.cpu);
163 avahi_free(r->data.hinfo.os);
166 case AVAHI_DNS_TYPE_TXT:
167 avahi_string_list_free(r->data.txt.string_list);
170 case AVAHI_DNS_TYPE_A:
171 case AVAHI_DNS_TYPE_AAAA:
175 avahi_free(r->data.generic.data);
178 avahi_key_unref(r->key);
183 const char *avahi_dns_class_to_string(uint16_t class) {
184 if (class & AVAHI_DNS_CACHE_FLUSH)
188 case AVAHI_DNS_CLASS_IN:
190 case AVAHI_DNS_CLASS_ANY:
197 const char *avahi_dns_type_to_string(uint16_t type) {
199 case AVAHI_DNS_TYPE_CNAME:
201 case AVAHI_DNS_TYPE_A:
203 case AVAHI_DNS_TYPE_AAAA:
205 case AVAHI_DNS_TYPE_PTR:
207 case AVAHI_DNS_TYPE_HINFO:
209 case AVAHI_DNS_TYPE_TXT:
211 case AVAHI_DNS_TYPE_SRV:
213 case AVAHI_DNS_TYPE_ANY:
215 case AVAHI_DNS_TYPE_SOA:
217 case AVAHI_DNS_TYPE_NS:
224 char *avahi_key_to_string(const AvahiKey *k) {
228 return avahi_strdup_printf("%s\t%s\t%s",
230 avahi_dns_class_to_string(k->clazz),
231 avahi_dns_type_to_string(k->type));
234 char *avahi_record_to_string(const AvahiRecord *r) {
236 char buf[257], *t = NULL, *d = NULL;
241 switch (r->key->type) {
242 case AVAHI_DNS_TYPE_A:
243 inet_ntop(AF_INET, &r->data.a.address.address, t = buf, sizeof(buf));
246 case AVAHI_DNS_TYPE_AAAA:
247 inet_ntop(AF_INET6, &r->data.aaaa.address.address, t = buf, sizeof(buf));
250 case AVAHI_DNS_TYPE_PTR:
251 case AVAHI_DNS_TYPE_CNAME:
252 case AVAHI_DNS_TYPE_NS:
254 t = r->data.ptr.name;
257 case AVAHI_DNS_TYPE_TXT:
258 t = d = avahi_string_list_to_string(r->data.txt.string_list);
261 case AVAHI_DNS_TYPE_HINFO:
263 snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
266 case AVAHI_DNS_TYPE_SRV:
268 snprintf(t = buf, sizeof(buf), "%u %u %u %s",
269 r->data.srv.priority,
277 p = avahi_key_to_string(r->key);
278 s = avahi_strdup_printf("%s %s ; ttl=%u", p, t ? t : "<unparsable>", r->ttl);
285 int avahi_key_equal(const AvahiKey *a, const AvahiKey *b) {
292 /* g_message("equal: %p %p", a, b); */
294 return avahi_domain_equal(a->name, b->name) &&
295 a->type == b->type &&
296 a->clazz == b->clazz;
299 int avahi_key_pattern_match(const AvahiKey *pattern, const AvahiKey *k) {
303 /* g_message("equal: %p %p", a, b); */
305 assert(!avahi_key_is_pattern(k));
310 return avahi_domain_equal(pattern->name, k->name) &&
311 (pattern->type == k->type || pattern->type == AVAHI_DNS_TYPE_ANY) &&
312 (pattern->clazz == k->clazz || pattern->clazz == AVAHI_DNS_CLASS_ANY);
315 int avahi_key_is_pattern(const AvahiKey *k) {
319 k->type == AVAHI_DNS_TYPE_ANY ||
320 k->clazz == AVAHI_DNS_CLASS_ANY;
323 unsigned avahi_key_hash(const AvahiKey *k) {
327 avahi_domain_hash(k->name) +
332 static int rdata_equal(const AvahiRecord *a, const AvahiRecord *b) {
335 assert(a->key->type == b->key->type);
337 /* t = avahi_record_to_string(a); */
338 /* g_message("comparing %s", t); */
341 /* t = avahi_record_to_string(b); */
342 /* g_message("and %s", t); */
346 switch (a->key->type) {
347 case AVAHI_DNS_TYPE_SRV:
349 a->data.srv.priority == b->data.srv.priority &&
350 a->data.srv.weight == b->data.srv.weight &&
351 a->data.srv.port == b->data.srv.port &&
352 avahi_domain_equal(a->data.srv.name, b->data.srv.name);
354 case AVAHI_DNS_TYPE_PTR:
355 case AVAHI_DNS_TYPE_CNAME:
356 case AVAHI_DNS_TYPE_NS:
357 return avahi_domain_equal(a->data.ptr.name, b->data.ptr.name);
359 case AVAHI_DNS_TYPE_HINFO:
361 !strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
362 !strcmp(a->data.hinfo.os, b->data.hinfo.os);
364 case AVAHI_DNS_TYPE_TXT:
365 return avahi_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
367 case AVAHI_DNS_TYPE_A:
368 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address)) == 0;
370 case AVAHI_DNS_TYPE_AAAA:
371 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address)) == 0;
374 return a->data.generic.size == b->data.generic.size &&
375 (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
380 int avahi_record_equal_no_ttl(const AvahiRecord *a, const AvahiRecord *b) {
388 avahi_key_equal(a->key, b->key) &&
393 AvahiRecord *avahi_record_copy(AvahiRecord *r) {
396 if (!(copy = avahi_new(AvahiRecord, 1))) {
397 avahi_log_error("avahi_new() failed.");
402 copy->key = avahi_key_ref(r->key);
405 switch (r->key->type) {
406 case AVAHI_DNS_TYPE_PTR:
407 case AVAHI_DNS_TYPE_CNAME:
408 case AVAHI_DNS_TYPE_NS:
409 if (!(copy->data.ptr.name = avahi_strdup(r->data.ptr.name)))
413 case AVAHI_DNS_TYPE_SRV:
414 copy->data.srv.priority = r->data.srv.priority;
415 copy->data.srv.weight = r->data.srv.weight;
416 copy->data.srv.port = r->data.srv.port;
417 if (!(copy->data.srv.name = avahi_strdup(r->data.srv.name)))
421 case AVAHI_DNS_TYPE_HINFO:
422 if (!(copy->data.hinfo.os = avahi_strdup(r->data.hinfo.os)))
425 if (!(copy->data.hinfo.cpu = avahi_strdup(r->data.hinfo.cpu))) {
426 avahi_free(r->data.hinfo.os);
431 case AVAHI_DNS_TYPE_TXT:
432 copy->data.txt.string_list = avahi_string_list_copy(r->data.txt.string_list);
435 case AVAHI_DNS_TYPE_A:
436 copy->data.a.address = r->data.a.address;
439 case AVAHI_DNS_TYPE_AAAA:
440 copy->data.aaaa.address = r->data.aaaa.address;
444 if (!(copy->data.generic.data = avahi_memdup(r->data.generic.data, r->data.generic.size)))
446 copy->data.generic.size = r->data.generic.size;
454 avahi_log_error("Failed to allocate memory");
456 avahi_key_unref(copy->key);
463 size_t avahi_key_get_estimate_size(AvahiKey *k) {
466 return strlen(k->name)+1+4;
469 size_t avahi_record_get_estimate_size(AvahiRecord *r) {
473 n = avahi_key_get_estimate_size(r->key) + 4 + 2;
475 switch (r->key->type) {
476 case AVAHI_DNS_TYPE_PTR:
477 case AVAHI_DNS_TYPE_CNAME:
478 case AVAHI_DNS_TYPE_NS:
479 n += strlen(r->data.ptr.name) + 1;
482 case AVAHI_DNS_TYPE_SRV:
483 n += 6 + strlen(r->data.srv.name) + 1;
486 case AVAHI_DNS_TYPE_HINFO:
487 n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
490 case AVAHI_DNS_TYPE_TXT:
491 n += avahi_string_list_serialize(r->data.txt.string_list, NULL, 0);
494 case AVAHI_DNS_TYPE_A:
495 n += sizeof(AvahiIPv4Address);
498 case AVAHI_DNS_TYPE_AAAA:
499 n += sizeof(AvahiIPv6Address);
503 n += r->data.generic.size;
509 static int lexicographical_memcmp(const void* a, size_t al, const void* b, size_t bl) {
516 c = al < bl ? al : bl;
517 if ((ret = memcmp(a, b, c)))
523 return al == c ? 1 : -1;
526 static int uint16_cmp(uint16_t a, uint16_t b) {
527 return a == b ? 0 : (a < b ? -1 : 1);
530 int avahi_record_lexicographical_compare(AvahiRecord *a, AvahiRecord *b) {
537 /* t1 = avahi_record_to_string(a); */
538 /* t2 = avahi_record_to_string(b); */
539 /* g_message("lexicocmp: %s %s", t1, t2); */
540 /* avahi_free(t1); */
541 /* avahi_free(t2); */
546 if ((r = uint16_cmp(a->key->clazz, b->key->clazz)) ||
547 (r = uint16_cmp(a->key->type, b->key->type)))
550 switch (a->key->type) {
552 case AVAHI_DNS_TYPE_PTR:
553 case AVAHI_DNS_TYPE_CNAME:
554 case AVAHI_DNS_TYPE_NS:
555 return avahi_binary_domain_cmp(a->data.ptr.name, b->data.ptr.name);
557 case AVAHI_DNS_TYPE_SRV: {
558 if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) == 0 &&
559 (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) == 0 &&
560 (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) == 0)
561 r = avahi_binary_domain_cmp(a->data.srv.name, b->data.srv.name);
566 case AVAHI_DNS_TYPE_HINFO: {
568 if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) ||
569 (r = strcmp(a->data.hinfo.os, b->data.hinfo.os)))
576 case AVAHI_DNS_TYPE_TXT: {
578 uint8_t *ma = NULL, *mb = NULL;
581 asize = avahi_string_list_serialize(a->data.txt.string_list, NULL, 0);
582 bsize = avahi_string_list_serialize(b->data.txt.string_list, NULL, 0);
584 if (asize > 0 && !(ma = avahi_new(uint8_t, asize)))
587 if (bsize > 0 && !(mb = avahi_new(uint8_t, bsize))) {
592 avahi_string_list_serialize(a->data.txt.string_list, ma, asize);
593 avahi_string_list_serialize(b->data.txt.string_list, mb, bsize);
596 r = lexicographical_memcmp(ma, asize, mb, bsize);
597 else if (asize && !bsize)
599 else if (!asize && bsize)
610 case AVAHI_DNS_TYPE_A:
611 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address));
613 case AVAHI_DNS_TYPE_AAAA:
614 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address));
617 return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
618 b->data.generic.data, b->data.generic.size);
623 avahi_log_error(__FILE__": Out of memory");
624 return -1; /* or whatever ... */
627 int avahi_record_is_goodbye(AvahiRecord *r) {
633 int avahi_key_is_valid(AvahiKey *k) {
636 if (!avahi_is_valid_domain_name(k->name))
642 int avahi_record_is_valid(AvahiRecord *r) {
645 if (!avahi_key_is_valid(r->key))
648 switch (r->key->type) {
650 case AVAHI_DNS_TYPE_PTR:
651 case AVAHI_DNS_TYPE_CNAME:
652 case AVAHI_DNS_TYPE_NS:
653 return avahi_is_valid_domain_name(r->data.ptr.name);
655 case AVAHI_DNS_TYPE_SRV:
656 return avahi_is_valid_domain_name(r->data.srv.name);
658 case AVAHI_DNS_TYPE_HINFO:
660 strlen(r->data.hinfo.os) <= 255 &&
661 strlen(r->data.hinfo.cpu) <= 255;
664 case AVAHI_DNS_TYPE_TXT: {
666 AvahiStringList *strlst;
668 for (strlst = r->data.txt.string_list; strlst; strlst = strlst->next)
669 if (strlst->size > 255)