4 #include <sys/socket.h>
11 flxKey *flx_key_new(const gchar *name, guint16 class, guint16 type) {
17 k->name = flx_normalize_name(name);
21 /* g_message("%p %% ref=1", k); */
26 flxKey *flx_key_ref(flxKey *k) {
28 g_assert(k->ref >= 1);
32 /* g_message("%p ++ ref=%i", k, k->ref); */
37 void flx_key_unref(flxKey *k) {
39 g_assert(k->ref >= 1);
41 /* g_message("%p -- ref=%i", k, k->ref-1); */
43 if ((--k->ref) <= 0) {
49 flxRecord *flx_record_new(flxKey *k) {
54 r = g_new(flxRecord, 1);
56 r->key = flx_key_ref(k);
58 memset(&r->data, 0, sizeof(r->data));
60 r->ttl = FLX_DEFAULT_TTL;
65 flxRecord *flx_record_new_full(const gchar *name, guint16 class, guint16 type) {
71 k = flx_key_new(name, class, type);
72 r = flx_record_new(k);
78 flxRecord *flx_record_ref(flxRecord *r) {
80 g_assert(r->ref >= 1);
86 void flx_record_unref(flxRecord *r) {
88 g_assert(r->ref >= 1);
90 if ((--r->ref) <= 0) {
91 switch (r->key->type) {
93 case FLX_DNS_TYPE_SRV:
94 g_free(r->data.srv.name);
97 case FLX_DNS_TYPE_PTR:
98 case FLX_DNS_TYPE_CNAME:
99 g_free(r->data.ptr.name);
102 case FLX_DNS_TYPE_HINFO:
103 g_free(r->data.hinfo.cpu);
104 g_free(r->data.hinfo.os);
107 case FLX_DNS_TYPE_TXT:
108 flx_string_list_free(r->data.txt.string_list);
112 case FLX_DNS_TYPE_AAAA:
116 g_free(r->data.generic.data);
119 flx_key_unref(r->key);
124 const gchar *flx_dns_class_to_string(guint16 class) {
125 if (class & FLX_DNS_CACHE_FLUSH)
128 if (class == FLX_DNS_CLASS_IN)
134 const gchar *flx_dns_type_to_string(guint16 type) {
136 case FLX_DNS_TYPE_CNAME:
140 case FLX_DNS_TYPE_AAAA:
142 case FLX_DNS_TYPE_PTR:
144 case FLX_DNS_TYPE_HINFO:
146 case FLX_DNS_TYPE_TXT:
148 case FLX_DNS_TYPE_SRV:
150 case FLX_DNS_TYPE_ANY:
158 gchar *flx_key_to_string(const flxKey *k) {
159 return g_strdup_printf("%s\t%s\t%s",
161 flx_dns_class_to_string(k->class),
162 flx_dns_type_to_string(k->type));
165 gchar *flx_record_to_string(const flxRecord *r) {
167 char buf[257], *t = NULL, *d = NULL;
169 switch (r->key->type) {
171 inet_ntop(AF_INET, &r->data.a.address.address, t = buf, sizeof(buf));
174 case FLX_DNS_TYPE_AAAA:
175 inet_ntop(AF_INET6, &r->data.aaaa.address.address, t = buf, sizeof(buf));
178 case FLX_DNS_TYPE_PTR:
179 case FLX_DNS_TYPE_CNAME :
181 t = r->data.ptr.name;
184 case FLX_DNS_TYPE_TXT:
185 t = d = flx_string_list_to_string(r->data.txt.string_list);
188 case FLX_DNS_TYPE_HINFO:
190 snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
193 case FLX_DNS_TYPE_SRV:
195 snprintf(t = buf, sizeof(buf), "%u %u %u %s",
196 r->data.srv.priority,
204 p = flx_key_to_string(r->key);
205 s = g_strdup_printf("%s %s ; ttl=%u", p, t ? t : "<unparsable>", r->ttl);
212 gboolean flx_key_equal(const flxKey *a, const flxKey *b) {
219 /* g_message("equal: %p %p", a, b); */
221 return flx_domain_equal(a->name, b->name) &&
222 a->type == b->type &&
223 a->class == b->class;
226 gboolean flx_key_pattern_match(const flxKey *pattern, const flxKey *k) {
230 /* g_message("equal: %p %p", a, b); */
232 g_assert(!flx_key_is_pattern(k));
237 return flx_domain_equal(pattern->name, k->name) &&
238 (pattern->type == k->type || pattern->type == FLX_DNS_TYPE_ANY) &&
239 pattern->class == k->class;
242 gboolean flx_key_is_pattern(const flxKey *k) {
245 return k->type == FLX_DNS_TYPE_ANY;
249 guint flx_key_hash(const flxKey *k) {
252 return flx_domain_hash(k->name) + k->type + k->class;
255 static gboolean rdata_equal(const flxRecord *a, const flxRecord *b) {
258 g_assert(a->key->type == b->key->type);
260 /* t = flx_record_to_string(a); */
261 /* g_message("comparing %s", t); */
264 /* t = flx_record_to_string(b); */
265 /* g_message("and %s", t); */
269 switch (a->key->type) {
270 case FLX_DNS_TYPE_SRV:
272 a->data.srv.priority == b->data.srv.priority &&
273 a->data.srv.weight == b->data.srv.weight &&
274 a->data.srv.port == b->data.srv.port &&
275 flx_domain_equal(a->data.srv.name, b->data.srv.name);
277 case FLX_DNS_TYPE_PTR:
278 case FLX_DNS_TYPE_CNAME:
279 return flx_domain_equal(a->data.ptr.name, b->data.ptr.name);
281 case FLX_DNS_TYPE_HINFO:
283 !strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
284 !strcmp(a->data.hinfo.os, b->data.hinfo.os);
286 case FLX_DNS_TYPE_TXT:
287 return flx_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
290 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(flxIPv4Address)) == 0;
292 case FLX_DNS_TYPE_AAAA:
293 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(flxIPv6Address)) == 0;
296 return a->data.generic.size == b->data.generic.size &&
297 (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
302 gboolean flx_record_equal_no_ttl(const flxRecord *a, const flxRecord *b) {
310 flx_key_equal(a->key, b->key) &&
315 flxRecord *flx_record_copy(flxRecord *r) {
318 copy = g_new(flxRecord, 1);
320 copy->key = flx_key_ref(r->key);
323 switch (r->key->type) {
324 case FLX_DNS_TYPE_PTR:
325 case FLX_DNS_TYPE_CNAME:
326 copy->data.ptr.name = g_strdup(r->data.ptr.name);
329 case FLX_DNS_TYPE_SRV:
330 copy->data.srv.priority = r->data.srv.priority;
331 copy->data.srv.weight = r->data.srv.weight;
332 copy->data.srv.port = r->data.srv.port;
333 copy->data.srv.name = g_strdup(r->data.srv.name);
336 case FLX_DNS_TYPE_HINFO:
337 copy->data.hinfo.os = g_strdup(r->data.hinfo.os);
338 copy->data.hinfo.cpu = g_strdup(r->data.hinfo.cpu);
341 case FLX_DNS_TYPE_TXT:
342 copy->data.txt.string_list = flx_string_list_copy(r->data.txt.string_list);
346 copy->data.a.address = r->data.a.address;
349 case FLX_DNS_TYPE_AAAA:
350 copy->data.aaaa.address = r->data.aaaa.address;
354 copy->data.generic.data = g_memdup(r->data.generic.data, r->data.generic.size);
355 copy->data.generic.size = r->data.generic.size;
364 guint flx_key_get_estimate_size(flxKey *k) {
367 return strlen(k->name)+1+4;
370 guint flx_record_get_estimate_size(flxRecord *r) {
374 n = flx_key_get_estimate_size(r->key) + 4 + 2;
376 switch (r->key->type) {
377 case FLX_DNS_TYPE_PTR:
378 case FLX_DNS_TYPE_CNAME:
379 n += strlen(r->data.ptr.name) + 1;
382 case FLX_DNS_TYPE_SRV:
383 n += 6 + strlen(r->data.srv.name) + 1;
386 case FLX_DNS_TYPE_HINFO:
387 n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
390 case FLX_DNS_TYPE_TXT:
391 n += flx_string_list_serialize(r->data.txt.string_list, NULL, 0);
395 n += sizeof(flxIPv4Address);
398 case FLX_DNS_TYPE_AAAA:
399 n += sizeof(flxIPv6Address);
403 n += r->data.generic.size;
409 static gint lexicographical_memcmp(gconstpointer a, size_t al, gconstpointer b, size_t bl) {
416 c = al < bl ? al : bl;
417 if ((ret = memcmp(a, b, c)) != 0)
423 return al == c ? 1 : -1;
426 static gint uint16_cmp(guint16 a, guint16 b) {
427 return a == b ? 0 : (a < b ? a : b);
430 static gint lexicographical_domain_cmp(const gchar *a, const gchar *b) {
450 flx_unescape_label(t1, sizeof(t1), &a);
451 flx_unescape_label(t2, sizeof(t2), &b);
457 return al < bl ? -1 : 1;
459 if ((r = strcmp(t1, t2)) != 0)
464 gint flx_record_lexicographical_compare(flxRecord *a, flxRecord *b) {
473 /* g_message("comparing [%s]", t = flx_record_to_string(a)); */
476 /* g_message("and [%s]", t = flx_record_to_string(b)); */
479 if (a->key->class < b->key->class)
481 else if (a->key->class > b->key->class)
484 if (a->key->type < b->key->type)
486 else if (a->key->type > b->key->type)
489 switch (a->key->type) {
491 case FLX_DNS_TYPE_PTR:
492 case FLX_DNS_TYPE_CNAME:
493 return lexicographical_domain_cmp(a->data.ptr.name, b->data.ptr.name);
495 case FLX_DNS_TYPE_SRV: {
497 if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) == 0 &&
498 (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) == 0 &&
499 (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) == 0)
500 r = lexicographical_domain_cmp(a->data.srv.name, b->data.srv.name);
505 case FLX_DNS_TYPE_HINFO: {
506 size_t al = strlen(a->data.hinfo.cpu), bl = strlen(b->data.hinfo.cpu);
510 return al < bl ? -1 : 1;
512 if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) != 0)
515 al = strlen(a->data.hinfo.os), bl = strlen(b->data.hinfo.os);
518 return al < bl ? -1 : 1;
520 if ((r = strcmp(a->data.hinfo.os, b->data.hinfo.os)) != 0)
527 case FLX_DNS_TYPE_TXT: {
533 ma = g_new(guint8, asize = flx_string_list_serialize(a->data.txt.string_list, NULL, 0));
534 mb = g_new(guint8, bsize = flx_string_list_serialize(b->data.txt.string_list, NULL, 0));
535 flx_string_list_serialize(a->data.txt.string_list, ma, asize);
536 flx_string_list_serialize(a->data.txt.string_list, mb, bsize);
538 r = lexicographical_memcmp(ma, asize, mb, bsize);
546 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(flxIPv4Address));
548 case FLX_DNS_TYPE_AAAA:
549 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(flxIPv6Address));
552 return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
553 b->data.generic.data, b->data.generic.size);