2 This file is part of catta.
4 catta 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.
9 catta 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.
14 You should have received a copy of the GNU Lesser General Public
15 License along with catta; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
26 #include <sys/types.h>
27 #include <sys/socket.h>
28 #include <arpa/inet.h>
31 #include <catta/domain.h>
32 #include <catta/malloc.h>
33 #include <catta/defs.h>
36 #include <catta/log.h>
39 #include "domain-util.h"
41 #include "addr-util.h"
43 CattaKey *catta_key_new(const char *name, uint16_t class, uint16_t type) {
47 if (!(k = catta_new(CattaKey, 1))) {
48 catta_log_error("catta_new() failed.");
52 if (!(k->name = catta_normalize_name_strdup(name))) {
53 catta_log_error("catta_normalize_name() failed.");
65 CattaKey *catta_key_new_cname(CattaKey *key) {
68 if (key->clazz != CATTA_DNS_CLASS_IN)
71 if (key->type == CATTA_DNS_TYPE_CNAME)
74 return catta_key_new(key->name, key->clazz, CATTA_DNS_TYPE_CNAME);
77 CattaKey *catta_key_ref(CattaKey *k) {
86 void catta_key_unref(CattaKey *k) {
90 if ((--k->ref) <= 0) {
96 CattaRecord *catta_record_new(CattaKey *k, uint32_t ttl) {
101 if (!(r = catta_new(CattaRecord, 1))) {
102 catta_log_error("catta_new() failed.");
107 r->key = catta_key_ref(k);
109 memset(&r->data, 0, sizeof(r->data));
111 r->ttl = ttl != (uint32_t) -1 ? ttl : CATTA_DEFAULT_TTL;
116 CattaRecord *catta_record_new_full(const char *name, uint16_t class, uint16_t type, uint32_t ttl) {
122 if (!(k = catta_key_new(name, class, type))) {
123 catta_log_error("catta_key_new() failed.");
127 r = catta_record_new(k, ttl);
131 catta_log_error("catta_record_new() failed.");
138 CattaRecord *catta_record_ref(CattaRecord *r) {
146 void catta_record_unref(CattaRecord *r) {
150 if ((--r->ref) <= 0) {
151 switch (r->key->type) {
153 case CATTA_DNS_TYPE_SRV:
154 catta_free(r->data.srv.name);
157 case CATTA_DNS_TYPE_PTR:
158 case CATTA_DNS_TYPE_CNAME:
159 case CATTA_DNS_TYPE_NS:
160 catta_free(r->data.ptr.name);
163 case CATTA_DNS_TYPE_HINFO:
164 catta_free(r->data.hinfo.cpu);
165 catta_free(r->data.hinfo.os);
168 case CATTA_DNS_TYPE_TXT:
169 catta_string_list_free(r->data.txt.string_list);
172 case CATTA_DNS_TYPE_A:
173 case CATTA_DNS_TYPE_AAAA:
177 catta_free(r->data.generic.data);
180 catta_key_unref(r->key);
185 const char *catta_dns_class_to_string(uint16_t class) {
186 if (class & CATTA_DNS_CACHE_FLUSH)
190 case CATTA_DNS_CLASS_IN:
192 case CATTA_DNS_CLASS_ANY:
199 const char *catta_dns_type_to_string(uint16_t type) {
201 case CATTA_DNS_TYPE_CNAME:
203 case CATTA_DNS_TYPE_A:
205 case CATTA_DNS_TYPE_AAAA:
207 case CATTA_DNS_TYPE_PTR:
209 case CATTA_DNS_TYPE_HINFO:
211 case CATTA_DNS_TYPE_TXT:
213 case CATTA_DNS_TYPE_SRV:
215 case CATTA_DNS_TYPE_ANY:
217 case CATTA_DNS_TYPE_SOA:
219 case CATTA_DNS_TYPE_NS:
226 char *catta_key_to_string(const CattaKey *k) {
227 char class[16], type[16];
233 /* According to RFC3597 */
235 if (!(c = catta_dns_class_to_string(k->clazz))) {
236 snprintf(class, sizeof(class), "CLASS%u", k->clazz);
240 if (!(t = catta_dns_type_to_string(k->type))) {
241 snprintf(type, sizeof(type), "TYPE%u", k->type);
245 return catta_strdup_printf("%s\t%s\t%s", k->name, c, t);
248 char *catta_record_to_string(const CattaRecord *r) {
250 char buf[1024], *t = NULL, *d = NULL;
255 switch (r->key->type) {
256 case CATTA_DNS_TYPE_A:
257 inet_ntop(AF_INET, (void *)&r->data.a.address.address, t = buf, sizeof(buf));
260 case CATTA_DNS_TYPE_AAAA:
261 inet_ntop(AF_INET6, (void *)&r->data.aaaa.address.address, t = buf, sizeof(buf));
264 case CATTA_DNS_TYPE_PTR:
265 case CATTA_DNS_TYPE_CNAME:
266 case CATTA_DNS_TYPE_NS:
268 t = r->data.ptr.name;
271 case CATTA_DNS_TYPE_TXT:
272 t = d = catta_string_list_to_string(r->data.txt.string_list);
275 case CATTA_DNS_TYPE_HINFO:
277 snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
280 case CATTA_DNS_TYPE_SRV:
282 snprintf(t = buf, sizeof(buf), "%u %u %u %s",
283 r->data.srv.priority,
297 /* According to RFC3597 */
299 snprintf(t = buf, sizeof(buf), "\\# %u", r->data.generic.size);
303 for (c = r->data.generic.data, n = r->data.generic.size, i = 0;
307 sprintf(e, " %02X", *c);
315 p = catta_key_to_string(r->key);
316 s = catta_strdup_printf("%s %s ; ttl=%u", p, t, r->ttl);
323 int catta_key_equal(const CattaKey *a, const CattaKey *b) {
330 return catta_domain_equal(a->name, b->name) &&
331 a->type == b->type &&
332 a->clazz == b->clazz;
335 int catta_key_pattern_match(const CattaKey *pattern, const CattaKey *k) {
339 assert(!catta_key_is_pattern(k));
344 return catta_domain_equal(pattern->name, k->name) &&
345 (pattern->type == k->type || pattern->type == CATTA_DNS_TYPE_ANY) &&
346 (pattern->clazz == k->clazz || pattern->clazz == CATTA_DNS_CLASS_ANY);
349 int catta_key_is_pattern(const CattaKey *k) {
353 k->type == CATTA_DNS_TYPE_ANY ||
354 k->clazz == CATTA_DNS_CLASS_ANY;
357 unsigned catta_key_hash(const CattaKey *k) {
361 catta_domain_hash(k->name) +
366 static int rdata_equal(const CattaRecord *a, const CattaRecord *b) {
369 assert(a->key->type == b->key->type);
371 switch (a->key->type) {
372 case CATTA_DNS_TYPE_SRV:
374 a->data.srv.priority == b->data.srv.priority &&
375 a->data.srv.weight == b->data.srv.weight &&
376 a->data.srv.port == b->data.srv.port &&
377 catta_domain_equal(a->data.srv.name, b->data.srv.name);
379 case CATTA_DNS_TYPE_PTR:
380 case CATTA_DNS_TYPE_CNAME:
381 case CATTA_DNS_TYPE_NS:
382 return catta_domain_equal(a->data.ptr.name, b->data.ptr.name);
384 case CATTA_DNS_TYPE_HINFO:
386 !strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
387 !strcmp(a->data.hinfo.os, b->data.hinfo.os);
389 case CATTA_DNS_TYPE_TXT:
390 return catta_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
392 case CATTA_DNS_TYPE_A:
393 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(CattaIPv4Address)) == 0;
395 case CATTA_DNS_TYPE_AAAA:
396 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(CattaIPv6Address)) == 0;
399 return a->data.generic.size == b->data.generic.size &&
400 (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
405 int catta_record_equal_no_ttl(const CattaRecord *a, const CattaRecord *b) {
413 catta_key_equal(a->key, b->key) &&
418 CattaRecord *catta_record_copy(CattaRecord *r) {
421 if (!(copy = catta_new(CattaRecord, 1))) {
422 catta_log_error("catta_new() failed.");
427 copy->key = catta_key_ref(r->key);
430 switch (r->key->type) {
431 case CATTA_DNS_TYPE_PTR:
432 case CATTA_DNS_TYPE_CNAME:
433 case CATTA_DNS_TYPE_NS:
434 if (!(copy->data.ptr.name = catta_strdup(r->data.ptr.name)))
438 case CATTA_DNS_TYPE_SRV:
439 copy->data.srv.priority = r->data.srv.priority;
440 copy->data.srv.weight = r->data.srv.weight;
441 copy->data.srv.port = r->data.srv.port;
442 if (!(copy->data.srv.name = catta_strdup(r->data.srv.name)))
446 case CATTA_DNS_TYPE_HINFO:
447 if (!(copy->data.hinfo.os = catta_strdup(r->data.hinfo.os)))
450 if (!(copy->data.hinfo.cpu = catta_strdup(r->data.hinfo.cpu))) {
451 catta_free(r->data.hinfo.os);
456 case CATTA_DNS_TYPE_TXT:
457 copy->data.txt.string_list = catta_string_list_copy(r->data.txt.string_list);
460 case CATTA_DNS_TYPE_A:
461 copy->data.a.address = r->data.a.address;
464 case CATTA_DNS_TYPE_AAAA:
465 copy->data.aaaa.address = r->data.aaaa.address;
469 if (!(copy->data.generic.data = catta_memdup(r->data.generic.data, r->data.generic.size)))
471 copy->data.generic.size = r->data.generic.size;
479 catta_log_error("Failed to allocate memory");
481 catta_key_unref(copy->key);
488 size_t catta_key_get_estimate_size(CattaKey *k) {
491 return strlen(k->name)+1+4;
494 size_t catta_record_get_estimate_size(CattaRecord *r) {
498 n = catta_key_get_estimate_size(r->key) + 4 + 2;
500 switch (r->key->type) {
501 case CATTA_DNS_TYPE_PTR:
502 case CATTA_DNS_TYPE_CNAME:
503 case CATTA_DNS_TYPE_NS:
504 n += strlen(r->data.ptr.name) + 1;
507 case CATTA_DNS_TYPE_SRV:
508 n += 6 + strlen(r->data.srv.name) + 1;
511 case CATTA_DNS_TYPE_HINFO:
512 n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
515 case CATTA_DNS_TYPE_TXT:
516 n += catta_string_list_serialize(r->data.txt.string_list, NULL, 0);
519 case CATTA_DNS_TYPE_A:
520 n += sizeof(CattaIPv4Address);
523 case CATTA_DNS_TYPE_AAAA:
524 n += sizeof(CattaIPv6Address);
528 n += r->data.generic.size;
534 static int lexicographical_memcmp(const void* a, size_t al, const void* b, size_t bl) {
541 c = al < bl ? al : bl;
542 if ((ret = memcmp(a, b, c)))
548 return al == c ? 1 : -1;
551 static int uint16_cmp(uint16_t a, uint16_t b) {
552 return a == b ? 0 : (a < b ? -1 : 1);
555 int catta_record_lexicographical_compare(CattaRecord *a, CattaRecord *b) {
562 /* t1 = catta_record_to_string(a); */
563 /* t2 = catta_record_to_string(b); */
564 /* g_message("lexicocmp: %s %s", t1, t2); */
565 /* catta_free(t1); */
566 /* catta_free(t2); */
571 if ((r = uint16_cmp(a->key->clazz, b->key->clazz)) ||
572 (r = uint16_cmp(a->key->type, b->key->type)))
575 switch (a->key->type) {
577 case CATTA_DNS_TYPE_PTR:
578 case CATTA_DNS_TYPE_CNAME:
579 case CATTA_DNS_TYPE_NS:
580 return catta_binary_domain_cmp(a->data.ptr.name, b->data.ptr.name);
582 case CATTA_DNS_TYPE_SRV: {
583 if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) == 0 &&
584 (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) == 0 &&
585 (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) == 0)
586 r = catta_binary_domain_cmp(a->data.srv.name, b->data.srv.name);
591 case CATTA_DNS_TYPE_HINFO: {
593 if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) ||
594 (r = strcmp(a->data.hinfo.os, b->data.hinfo.os)))
601 case CATTA_DNS_TYPE_TXT: {
603 uint8_t *ma = NULL, *mb = NULL;
606 asize = catta_string_list_serialize(a->data.txt.string_list, NULL, 0);
607 bsize = catta_string_list_serialize(b->data.txt.string_list, NULL, 0);
609 if (asize > 0 && !(ma = catta_new(uint8_t, asize)))
612 if (bsize > 0 && !(mb = catta_new(uint8_t, bsize))) {
617 catta_string_list_serialize(a->data.txt.string_list, ma, asize);
618 catta_string_list_serialize(b->data.txt.string_list, mb, bsize);
621 r = lexicographical_memcmp(ma, asize, mb, bsize);
622 else if (asize && !bsize)
624 else if (!asize && bsize)
635 case CATTA_DNS_TYPE_A:
636 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(CattaIPv4Address));
638 case CATTA_DNS_TYPE_AAAA:
639 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(CattaIPv6Address));
642 return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
643 b->data.generic.data, b->data.generic.size);
648 catta_log_error(__FILE__": Out of memory");
649 return -1; /* or whatever ... */
652 int catta_record_is_goodbye(CattaRecord *r) {
658 int catta_key_is_valid(CattaKey *k) {
661 if (!catta_is_valid_domain_name(k->name))
667 int catta_record_is_valid(CattaRecord *r) {
670 if (!catta_key_is_valid(r->key))
673 switch (r->key->type) {
675 case CATTA_DNS_TYPE_PTR:
676 case CATTA_DNS_TYPE_CNAME:
677 case CATTA_DNS_TYPE_NS:
678 return catta_is_valid_domain_name(r->data.ptr.name);
680 case CATTA_DNS_TYPE_SRV:
681 return catta_is_valid_domain_name(r->data.srv.name);
683 case CATTA_DNS_TYPE_HINFO:
685 strlen(r->data.hinfo.os) <= 255 &&
686 strlen(r->data.hinfo.cpu) <= 255;
688 case CATTA_DNS_TYPE_TXT: {
690 CattaStringList *strlst;
692 for (strlst = r->data.txt.string_list; strlst; strlst = strlst->next)
693 if (strlst->size > 255 || strlst->size <= 0)
703 static CattaAddress *get_address(const CattaRecord *r, CattaAddress *a) {
706 switch (r->key->type) {
707 case CATTA_DNS_TYPE_A:
708 a->proto = CATTA_PROTO_INET;
709 a->data.ipv4 = r->data.a.address;
712 case CATTA_DNS_TYPE_AAAA:
713 a->proto = CATTA_PROTO_INET6;
714 a->data.ipv6 = r->data.aaaa.address;
724 int catta_record_is_link_local_address(const CattaRecord *r) {
729 if (!get_address(r, &a))
732 return catta_address_is_link_local(&a);