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) {
216 /* g_message("equal: %p %p", a, b); */
218 return flx_domain_equal(a->name, b->name) &&
219 a->type == b->type &&
220 a->class == b->class;
223 gboolean flx_key_pattern_match(const flxKey *pattern, const flxKey *k) {
227 /* g_message("equal: %p %p", a, b); */
229 g_assert(!flx_key_is_pattern(k));
231 return flx_domain_equal(pattern->name, k->name) &&
232 (pattern->type == k->type || pattern->type == FLX_DNS_TYPE_ANY) &&
233 pattern->class == k->class;
236 gboolean flx_key_is_pattern(const flxKey *k) {
239 return k->type == FLX_DNS_TYPE_ANY;
243 guint flx_key_hash(const flxKey *k) {
246 return flx_domain_hash(k->name) + k->type + k->class;
249 static gboolean rdata_equal(const flxRecord *a, const flxRecord *b) {
252 g_assert(a->key->type == b->key->type);
254 /* t = flx_record_to_string(a); */
255 /* g_message("comparing %s", t); */
258 /* t = flx_record_to_string(b); */
259 /* g_message("and %s", t); */
263 switch (a->key->type) {
264 case FLX_DNS_TYPE_SRV:
266 a->data.srv.priority == b->data.srv.priority &&
267 a->data.srv.weight == b->data.srv.weight &&
268 a->data.srv.port == b->data.srv.port &&
269 flx_domain_equal(a->data.srv.name, b->data.srv.name);
271 case FLX_DNS_TYPE_PTR:
272 case FLX_DNS_TYPE_CNAME:
273 return flx_domain_equal(a->data.ptr.name, b->data.ptr.name);
275 case FLX_DNS_TYPE_HINFO:
277 !strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
278 !strcmp(a->data.hinfo.os, b->data.hinfo.os);
280 case FLX_DNS_TYPE_TXT:
281 return flx_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
284 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(flxIPv4Address)) == 0;
286 case FLX_DNS_TYPE_AAAA:
287 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(flxIPv6Address)) == 0;
290 return a->data.generic.size == b->data.generic.size &&
291 (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
296 gboolean flx_record_equal_no_ttl(const flxRecord *a, const flxRecord *b) {
301 flx_key_equal(a->key, b->key) &&
306 flxRecord *flx_record_copy(flxRecord *r) {
309 copy = g_new(flxRecord, 1);
311 copy->key = flx_key_ref(r->key);
314 switch (r->key->type) {
315 case FLX_DNS_TYPE_PTR:
316 case FLX_DNS_TYPE_CNAME:
317 copy->data.ptr.name = g_strdup(r->data.ptr.name);
320 case FLX_DNS_TYPE_SRV:
321 copy->data.srv.priority = r->data.srv.priority;
322 copy->data.srv.weight = r->data.srv.weight;
323 copy->data.srv.port = r->data.srv.port;
324 copy->data.srv.name = g_strdup(r->data.srv.name);
327 case FLX_DNS_TYPE_HINFO:
328 copy->data.hinfo.os = g_strdup(r->data.hinfo.os);
329 copy->data.hinfo.cpu = g_strdup(r->data.hinfo.cpu);
332 case FLX_DNS_TYPE_TXT:
333 copy->data.txt.string_list = flx_string_list_copy(r->data.txt.string_list);
337 copy->data.a.address = r->data.a.address;
340 case FLX_DNS_TYPE_AAAA:
341 copy->data.aaaa.address = r->data.aaaa.address;
345 copy->data.generic.data = g_memdup(r->data.generic.data, r->data.generic.size);
346 copy->data.generic.size = r->data.generic.size;
355 guint flx_key_get_estimate_size(flxKey *k) {
358 return strlen(k->name)+1+4;
361 guint flx_record_get_estimate_size(flxRecord *r) {
365 n = flx_key_get_estimate_size(r->key) + 4 + 2;
367 switch (r->key->type) {
368 case FLX_DNS_TYPE_PTR:
369 case FLX_DNS_TYPE_CNAME:
370 n += strlen(r->data.ptr.name) + 1;
373 case FLX_DNS_TYPE_SRV:
374 n += 6 + strlen(r->data.srv.name) + 1;
377 case FLX_DNS_TYPE_HINFO:
378 n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
381 case FLX_DNS_TYPE_TXT:
382 n += flx_string_list_serialize(r->data.txt.string_list, NULL, 0);
386 n += sizeof(flxIPv4Address);
389 case FLX_DNS_TYPE_AAAA:
390 n += sizeof(flxIPv6Address);
394 n += r->data.generic.size;
400 static gint lexicographical_memcmp(gconstpointer a, size_t al, gconstpointer b, size_t bl) {
407 c = al < bl ? al : bl;
408 if ((ret = memcmp(a, b, c)) != 0)
414 return al == c ? 1 : -1;
417 static gint uint16_cmp(guint16 a, guint16 b) {
418 return a == b ? 0 : (a < b ? a : b);
421 static gint lexicographical_domain_cmp(const gchar *a, const gchar *b) {
441 flx_unescape_label(t1, sizeof(t1), &a);
442 flx_unescape_label(t2, sizeof(t2), &b);
448 return al < bl ? -1 : 1;
450 if ((r = strcmp(t1, t2)) != 0)
455 gint flx_record_lexicographical_compare(flxRecord *a, flxRecord *b) {
459 if (a->key->class < b->key->class)
461 else if (a->key->class > b->key->class)
464 if (a->key->type < b->key->type)
466 else if (a->key->type > b->key->type)
469 switch (a->key->type) {
471 case FLX_DNS_TYPE_PTR:
472 case FLX_DNS_TYPE_CNAME:
473 return lexicographical_domain_cmp(a->data.ptr.name, b->data.ptr.name);
475 case FLX_DNS_TYPE_SRV: {
477 if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) != 0 ||
478 (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) != 0 ||
479 (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) != 0)
480 r = lexicographical_domain_cmp(a->data.srv.name, b->data.srv.name);
485 case FLX_DNS_TYPE_HINFO: {
486 size_t al = strlen(a->data.hinfo.cpu), bl = strlen(b->data.hinfo.cpu);
490 return al < bl ? -1 : 1;
492 if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) != 0)
495 al = strlen(a->data.hinfo.os), bl = strlen(b->data.hinfo.os);
498 return al < bl ? -1 : 1;
500 if ((r = strcmp(a->data.hinfo.os, b->data.hinfo.os)) != 0)
507 case FLX_DNS_TYPE_TXT: {
513 ma = g_new(guint8, asize = flx_string_list_serialize(a->data.txt.string_list, NULL, 0));
514 mb = g_new(guint8, bsize = flx_string_list_serialize(b->data.txt.string_list, NULL, 0));
515 flx_string_list_serialize(a->data.txt.string_list, ma, asize);
516 flx_string_list_serialize(a->data.txt.string_list, mb, bsize);
518 r = lexicographical_memcmp(ma, asize, mb, bsize);
526 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(flxIPv4Address));
528 case FLX_DNS_TYPE_AAAA:
529 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(flxIPv6Address));
532 return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
533 b->data.generic.data, b->data.generic.size);