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(name))) {
51 avahi_log_error("avahi_normalize_name() failed.");
63 AvahiKey *avahi_key_ref(AvahiKey *k) {
72 void avahi_key_unref(AvahiKey *k) {
76 if ((--k->ref) <= 0) {
82 AvahiRecord *avahi_record_new(AvahiKey *k, uint32_t ttl) {
87 if (!(r = avahi_new(AvahiRecord, 1))) {
88 avahi_log_error("avahi_new() failed.");
93 r->key = avahi_key_ref(k);
95 memset(&r->data, 0, sizeof(r->data));
97 r->ttl = ttl != (uint32_t) -1 ? ttl : AVAHI_DEFAULT_TTL;
102 AvahiRecord *avahi_record_new_full(const char *name, uint16_t class, uint16_t type, uint32_t ttl) {
108 if (!(k = avahi_key_new(name, class, type))) {
109 avahi_log_error("avahi_key_new() failed.");
113 r = avahi_record_new(k, ttl);
117 avahi_log_error("avahi_record_new() failed.");
124 AvahiRecord *avahi_record_ref(AvahiRecord *r) {
132 void avahi_record_unref(AvahiRecord *r) {
136 if ((--r->ref) <= 0) {
137 switch (r->key->type) {
139 case AVAHI_DNS_TYPE_SRV:
140 avahi_free(r->data.srv.name);
143 case AVAHI_DNS_TYPE_PTR:
144 case AVAHI_DNS_TYPE_CNAME:
145 avahi_free(r->data.ptr.name);
148 case AVAHI_DNS_TYPE_HINFO:
149 avahi_free(r->data.hinfo.cpu);
150 avahi_free(r->data.hinfo.os);
153 case AVAHI_DNS_TYPE_TXT:
154 avahi_string_list_free(r->data.txt.string_list);
157 case AVAHI_DNS_TYPE_A:
158 case AVAHI_DNS_TYPE_AAAA:
162 avahi_free(r->data.generic.data);
165 avahi_key_unref(r->key);
170 const char *avahi_dns_class_to_string(uint16_t class) {
171 if (class & AVAHI_DNS_CACHE_FLUSH)
175 case AVAHI_DNS_CLASS_IN:
177 case AVAHI_DNS_CLASS_ANY:
184 const char *avahi_dns_type_to_string(uint16_t type) {
186 case AVAHI_DNS_TYPE_CNAME:
188 case AVAHI_DNS_TYPE_A:
190 case AVAHI_DNS_TYPE_AAAA:
192 case AVAHI_DNS_TYPE_PTR:
194 case AVAHI_DNS_TYPE_HINFO:
196 case AVAHI_DNS_TYPE_TXT:
198 case AVAHI_DNS_TYPE_SRV:
200 case AVAHI_DNS_TYPE_ANY:
207 char *avahi_key_to_string(const AvahiKey *k) {
211 return avahi_strdup_printf("%s\t%s\t%s",
213 avahi_dns_class_to_string(k->clazz),
214 avahi_dns_type_to_string(k->type));
217 char *avahi_record_to_string(const AvahiRecord *r) {
219 char buf[257], *t = NULL, *d = NULL;
224 switch (r->key->type) {
225 case AVAHI_DNS_TYPE_A:
226 inet_ntop(AF_INET, &r->data.a.address.address, t = buf, sizeof(buf));
229 case AVAHI_DNS_TYPE_AAAA:
230 inet_ntop(AF_INET6, &r->data.aaaa.address.address, t = buf, sizeof(buf));
233 case AVAHI_DNS_TYPE_PTR:
234 case AVAHI_DNS_TYPE_CNAME :
236 t = r->data.ptr.name;
239 case AVAHI_DNS_TYPE_TXT:
240 t = d = avahi_string_list_to_string(r->data.txt.string_list);
243 case AVAHI_DNS_TYPE_HINFO:
245 snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
248 case AVAHI_DNS_TYPE_SRV:
250 snprintf(t = buf, sizeof(buf), "%u %u %u %s",
251 r->data.srv.priority,
259 p = avahi_key_to_string(r->key);
260 s = avahi_strdup_printf("%s %s ; ttl=%u", p, t ? t : "<unparsable>", r->ttl);
267 int avahi_key_equal(const AvahiKey *a, const AvahiKey *b) {
274 /* g_message("equal: %p %p", a, b); */
276 return avahi_domain_equal(a->name, b->name) &&
277 a->type == b->type &&
278 a->clazz == b->clazz;
281 int avahi_key_pattern_match(const AvahiKey *pattern, const AvahiKey *k) {
285 /* g_message("equal: %p %p", a, b); */
287 assert(!avahi_key_is_pattern(k));
292 return avahi_domain_equal(pattern->name, k->name) &&
293 (pattern->type == k->type || pattern->type == AVAHI_DNS_TYPE_ANY) &&
294 (pattern->clazz == k->clazz || pattern->clazz == AVAHI_DNS_CLASS_ANY);
297 int avahi_key_is_pattern(const AvahiKey *k) {
301 k->type == AVAHI_DNS_TYPE_ANY ||
302 k->clazz == AVAHI_DNS_CLASS_ANY;
305 unsigned avahi_key_hash(const AvahiKey *k) {
309 avahi_domain_hash(k->name) +
314 static int rdata_equal(const AvahiRecord *a, const AvahiRecord *b) {
317 assert(a->key->type == b->key->type);
319 /* t = avahi_record_to_string(a); */
320 /* g_message("comparing %s", t); */
323 /* t = avahi_record_to_string(b); */
324 /* g_message("and %s", t); */
328 switch (a->key->type) {
329 case AVAHI_DNS_TYPE_SRV:
331 a->data.srv.priority == b->data.srv.priority &&
332 a->data.srv.weight == b->data.srv.weight &&
333 a->data.srv.port == b->data.srv.port &&
334 avahi_domain_equal(a->data.srv.name, b->data.srv.name);
336 case AVAHI_DNS_TYPE_PTR:
337 case AVAHI_DNS_TYPE_CNAME:
338 return avahi_domain_equal(a->data.ptr.name, b->data.ptr.name);
340 case AVAHI_DNS_TYPE_HINFO:
342 !strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
343 !strcmp(a->data.hinfo.os, b->data.hinfo.os);
345 case AVAHI_DNS_TYPE_TXT:
346 return avahi_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
348 case AVAHI_DNS_TYPE_A:
349 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address)) == 0;
351 case AVAHI_DNS_TYPE_AAAA:
352 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address)) == 0;
355 return a->data.generic.size == b->data.generic.size &&
356 (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
361 int avahi_record_equal_no_ttl(const AvahiRecord *a, const AvahiRecord *b) {
369 avahi_key_equal(a->key, b->key) &&
374 AvahiRecord *avahi_record_copy(AvahiRecord *r) {
377 if (!(copy = avahi_new(AvahiRecord, 1))) {
378 avahi_log_error("avahi_new() failed.");
383 copy->key = avahi_key_ref(r->key);
386 switch (r->key->type) {
387 case AVAHI_DNS_TYPE_PTR:
388 case AVAHI_DNS_TYPE_CNAME:
389 if (!(copy->data.ptr.name = avahi_strdup(r->data.ptr.name)))
393 case AVAHI_DNS_TYPE_SRV:
394 copy->data.srv.priority = r->data.srv.priority;
395 copy->data.srv.weight = r->data.srv.weight;
396 copy->data.srv.port = r->data.srv.port;
397 if (!(copy->data.srv.name = avahi_strdup(r->data.srv.name)))
401 case AVAHI_DNS_TYPE_HINFO:
402 if (!(copy->data.hinfo.os = avahi_strdup(r->data.hinfo.os)))
405 if (!(copy->data.hinfo.cpu = avahi_strdup(r->data.hinfo.cpu))) {
406 avahi_free(r->data.hinfo.os);
411 case AVAHI_DNS_TYPE_TXT:
412 copy->data.txt.string_list = avahi_string_list_copy(r->data.txt.string_list);
415 case AVAHI_DNS_TYPE_A:
416 copy->data.a.address = r->data.a.address;
419 case AVAHI_DNS_TYPE_AAAA:
420 copy->data.aaaa.address = r->data.aaaa.address;
424 if (!(copy->data.generic.data = avahi_memdup(r->data.generic.data, r->data.generic.size)))
426 copy->data.generic.size = r->data.generic.size;
434 avahi_log_error("Failed to allocate memory");
436 avahi_key_unref(copy->key);
443 size_t avahi_key_get_estimate_size(AvahiKey *k) {
446 return strlen(k->name)+1+4;
449 size_t avahi_record_get_estimate_size(AvahiRecord *r) {
453 n = avahi_key_get_estimate_size(r->key) + 4 + 2;
455 switch (r->key->type) {
456 case AVAHI_DNS_TYPE_PTR:
457 case AVAHI_DNS_TYPE_CNAME:
458 n += strlen(r->data.ptr.name) + 1;
461 case AVAHI_DNS_TYPE_SRV:
462 n += 6 + strlen(r->data.srv.name) + 1;
465 case AVAHI_DNS_TYPE_HINFO:
466 n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
469 case AVAHI_DNS_TYPE_TXT:
470 n += avahi_string_list_serialize(r->data.txt.string_list, NULL, 0);
473 case AVAHI_DNS_TYPE_A:
474 n += sizeof(AvahiIPv4Address);
477 case AVAHI_DNS_TYPE_AAAA:
478 n += sizeof(AvahiIPv6Address);
482 n += r->data.generic.size;
488 static int lexicographical_memcmp(const void* a, size_t al, const void* b, size_t bl) {
495 c = al < bl ? al : bl;
496 if ((ret = memcmp(a, b, c)))
502 return al == c ? 1 : -1;
505 static int uint16_cmp(uint16_t a, uint16_t b) {
506 return a == b ? 0 : (a < b ? -1 : 1);
509 int avahi_record_lexicographical_compare(AvahiRecord *a, AvahiRecord *b) {
516 /* t1 = avahi_record_to_string(a); */
517 /* t2 = avahi_record_to_string(b); */
518 /* g_message("lexicocmp: %s %s", t1, t2); */
519 /* avahi_free(t1); */
520 /* avahi_free(t2); */
525 if ((r = uint16_cmp(a->key->clazz, b->key->clazz)) ||
526 (r = uint16_cmp(a->key->type, b->key->type)))
529 switch (a->key->type) {
531 case AVAHI_DNS_TYPE_PTR:
532 case AVAHI_DNS_TYPE_CNAME:
533 return avahi_binary_domain_cmp(a->data.ptr.name, b->data.ptr.name);
535 case AVAHI_DNS_TYPE_SRV: {
536 if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) == 0 &&
537 (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) == 0 &&
538 (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) == 0)
539 r = avahi_binary_domain_cmp(a->data.srv.name, b->data.srv.name);
544 case AVAHI_DNS_TYPE_HINFO: {
546 if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) ||
547 (r = strcmp(a->data.hinfo.os, b->data.hinfo.os)))
554 case AVAHI_DNS_TYPE_TXT: {
556 uint8_t *ma = NULL, *mb = NULL;
559 asize = avahi_string_list_serialize(a->data.txt.string_list, NULL, 0);
560 bsize = avahi_string_list_serialize(b->data.txt.string_list, NULL, 0);
562 if (asize > 0 && !(ma = avahi_new(uint8_t, asize)))
565 if (bsize > 0 && !(mb = avahi_new(uint8_t, bsize))) {
570 avahi_string_list_serialize(a->data.txt.string_list, ma, asize);
571 avahi_string_list_serialize(b->data.txt.string_list, mb, bsize);
574 r = lexicographical_memcmp(ma, asize, mb, bsize);
575 else if (asize && !bsize)
577 else if (!asize && bsize)
588 case AVAHI_DNS_TYPE_A:
589 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address));
591 case AVAHI_DNS_TYPE_AAAA:
592 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address));
595 return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
596 b->data.generic.data, b->data.generic.size);
601 avahi_log_error(__FILE__": Out of memory");
602 return -1; /* or whatever ... */
605 int avahi_record_is_goodbye(AvahiRecord *r) {
611 int avahi_key_is_valid(AvahiKey *k) {
614 if (!avahi_is_valid_domain_name(k->name))
620 int avahi_record_is_valid(AvahiRecord *r) {
623 if (!avahi_key_is_valid(r->key))
626 switch (r->key->type) {
628 case AVAHI_DNS_TYPE_PTR:
629 case AVAHI_DNS_TYPE_CNAME:
630 return avahi_is_valid_domain_name(r->data.ptr.name);
632 case AVAHI_DNS_TYPE_SRV:
633 return avahi_is_valid_domain_name(r->data.srv.name);
635 case AVAHI_DNS_TYPE_HINFO:
637 strlen(r->data.hinfo.os) <= 255 &&
638 strlen(r->data.hinfo.cpu) <= 255;
641 case AVAHI_DNS_TYPE_TXT: {
643 AvahiStringList *strlst;
645 for (strlst = r->data.txt.string_list; strlst; strlst = strlst->next)
646 if (strlst->size > 255)