4 #include <sys/socket.h>
11 AvahiKey *avahi_key_new(const gchar *name, guint16 class, guint16 type) {
15 k = g_new(AvahiKey, 1);
17 k->name = avahi_normalize_name(name);
21 /* g_message("%p %% ref=1", k); */
26 AvahiKey *avahi_key_ref(AvahiKey *k) {
28 g_assert(k->ref >= 1);
32 /* g_message("%p ++ ref=%i", k, k->ref); */
37 void avahi_key_unref(AvahiKey *k) {
39 g_assert(k->ref >= 1);
41 /* g_message("%p -- ref=%i", k, k->ref-1); */
43 if ((--k->ref) <= 0) {
49 AvahiRecord *avahi_record_new(AvahiKey *k) {
54 r = g_new(AvahiRecord, 1);
56 r->key = avahi_key_ref(k);
58 memset(&r->data, 0, sizeof(r->data));
60 r->ttl = AVAHI_DEFAULT_TTL;
65 AvahiRecord *avahi_record_new_full(const gchar *name, guint16 class, guint16 type) {
71 k = avahi_key_new(name, class, type);
72 r = avahi_record_new(k);
78 AvahiRecord *avahi_record_ref(AvahiRecord *r) {
80 g_assert(r->ref >= 1);
86 void avahi_record_unref(AvahiRecord *r) {
88 g_assert(r->ref >= 1);
90 if ((--r->ref) <= 0) {
91 switch (r->key->type) {
93 case AVAHI_DNS_TYPE_SRV:
94 g_free(r->data.srv.name);
97 case AVAHI_DNS_TYPE_PTR:
98 case AVAHI_DNS_TYPE_CNAME:
99 g_free(r->data.ptr.name);
102 case AVAHI_DNS_TYPE_HINFO:
103 g_free(r->data.hinfo.cpu);
104 g_free(r->data.hinfo.os);
107 case AVAHI_DNS_TYPE_TXT:
108 avahi_string_list_free(r->data.txt.string_list);
111 case AVAHI_DNS_TYPE_A:
112 case AVAHI_DNS_TYPE_AAAA:
116 g_free(r->data.generic.data);
119 avahi_key_unref(r->key);
124 const gchar *avahi_dns_class_to_string(guint16 class) {
125 if (class & AVAHI_DNS_CACHE_FLUSH)
128 if (class == AVAHI_DNS_CLASS_IN)
134 const gchar *avahi_dns_type_to_string(guint16 type) {
136 case AVAHI_DNS_TYPE_CNAME:
138 case AVAHI_DNS_TYPE_A:
140 case AVAHI_DNS_TYPE_AAAA:
142 case AVAHI_DNS_TYPE_PTR:
144 case AVAHI_DNS_TYPE_HINFO:
146 case AVAHI_DNS_TYPE_TXT:
148 case AVAHI_DNS_TYPE_SRV:
150 case AVAHI_DNS_TYPE_ANY:
158 gchar *avahi_key_to_string(const AvahiKey *k) {
159 return g_strdup_printf("%s\t%s\t%s",
161 avahi_dns_class_to_string(k->class),
162 avahi_dns_type_to_string(k->type));
165 gchar *avahi_record_to_string(const AvahiRecord *r) {
167 char buf[257], *t = NULL, *d = NULL;
169 switch (r->key->type) {
170 case AVAHI_DNS_TYPE_A:
171 inet_ntop(AF_INET, &r->data.a.address.address, t = buf, sizeof(buf));
174 case AVAHI_DNS_TYPE_AAAA:
175 inet_ntop(AF_INET6, &r->data.aaaa.address.address, t = buf, sizeof(buf));
178 case AVAHI_DNS_TYPE_PTR:
179 case AVAHI_DNS_TYPE_CNAME :
181 t = r->data.ptr.name;
184 case AVAHI_DNS_TYPE_TXT:
185 t = d = avahi_string_list_to_string(r->data.txt.string_list);
188 case AVAHI_DNS_TYPE_HINFO:
190 snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
193 case AVAHI_DNS_TYPE_SRV:
195 snprintf(t = buf, sizeof(buf), "%u %u %u %s",
196 r->data.srv.priority,
204 p = avahi_key_to_string(r->key);
205 s = g_strdup_printf("%s %s ; ttl=%u", p, t ? t : "<unparsable>", r->ttl);
212 gboolean avahi_key_equal(const AvahiKey *a, const AvahiKey *b) {
219 /* g_message("equal: %p %p", a, b); */
221 return avahi_domain_equal(a->name, b->name) &&
222 a->type == b->type &&
223 a->class == b->class;
226 gboolean avahi_key_pattern_match(const AvahiKey *pattern, const AvahiKey *k) {
230 /* g_message("equal: %p %p", a, b); */
232 g_assert(!avahi_key_is_pattern(k));
237 return avahi_domain_equal(pattern->name, k->name) &&
238 (pattern->type == k->type || pattern->type == AVAHI_DNS_TYPE_ANY) &&
239 pattern->class == k->class;
242 gboolean avahi_key_is_pattern(const AvahiKey *k) {
245 return k->type == AVAHI_DNS_TYPE_ANY;
249 guint avahi_key_hash(const AvahiKey *k) {
252 return avahi_domain_hash(k->name) + k->type + k->class;
255 static gboolean rdata_equal(const AvahiRecord *a, const AvahiRecord *b) {
258 g_assert(a->key->type == b->key->type);
260 /* t = avahi_record_to_string(a); */
261 /* g_message("comparing %s", t); */
264 /* t = avahi_record_to_string(b); */
265 /* g_message("and %s", t); */
269 switch (a->key->type) {
270 case AVAHI_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 avahi_domain_equal(a->data.srv.name, b->data.srv.name);
277 case AVAHI_DNS_TYPE_PTR:
278 case AVAHI_DNS_TYPE_CNAME:
279 return avahi_domain_equal(a->data.ptr.name, b->data.ptr.name);
281 case AVAHI_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 AVAHI_DNS_TYPE_TXT:
287 return avahi_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
289 case AVAHI_DNS_TYPE_A:
290 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address)) == 0;
292 case AVAHI_DNS_TYPE_AAAA:
293 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address)) == 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 avahi_record_equal_no_ttl(const AvahiRecord *a, const AvahiRecord *b) {
310 avahi_key_equal(a->key, b->key) &&
315 AvahiRecord *avahi_record_copy(AvahiRecord *r) {
318 copy = g_new(AvahiRecord, 1);
320 copy->key = avahi_key_ref(r->key);
323 switch (r->key->type) {
324 case AVAHI_DNS_TYPE_PTR:
325 case AVAHI_DNS_TYPE_CNAME:
326 copy->data.ptr.name = g_strdup(r->data.ptr.name);
329 case AVAHI_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 AVAHI_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 AVAHI_DNS_TYPE_TXT:
342 copy->data.txt.string_list = avahi_string_list_copy(r->data.txt.string_list);
345 case AVAHI_DNS_TYPE_A:
346 copy->data.a.address = r->data.a.address;
349 case AVAHI_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 avahi_key_get_estimate_size(AvahiKey *k) {
367 return strlen(k->name)+1+4;
370 guint avahi_record_get_estimate_size(AvahiRecord *r) {
374 n = avahi_key_get_estimate_size(r->key) + 4 + 2;
376 switch (r->key->type) {
377 case AVAHI_DNS_TYPE_PTR:
378 case AVAHI_DNS_TYPE_CNAME:
379 n += strlen(r->data.ptr.name) + 1;
382 case AVAHI_DNS_TYPE_SRV:
383 n += 6 + strlen(r->data.srv.name) + 1;
386 case AVAHI_DNS_TYPE_HINFO:
387 n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
390 case AVAHI_DNS_TYPE_TXT:
391 n += avahi_string_list_serialize(r->data.txt.string_list, NULL, 0);
394 case AVAHI_DNS_TYPE_A:
395 n += sizeof(AvahiIPv4Address);
398 case AVAHI_DNS_TYPE_AAAA:
399 n += sizeof(AvahiIPv6Address);
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 avahi_unescape_label(t1, sizeof(t1), &a);
451 avahi_unescape_label(t2, sizeof(t2), &b);
457 return al < bl ? -1 : 1;
459 if ((r = strcmp(t1, t2)) != 0)
464 gint avahi_record_lexicographical_compare(AvahiRecord *a, AvahiRecord *b) {
473 /* g_message("comparing [%s]", t = avahi_record_to_string(a)); */
476 /* g_message("and [%s]", t = avahi_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 AVAHI_DNS_TYPE_PTR:
492 case AVAHI_DNS_TYPE_CNAME:
493 return lexicographical_domain_cmp(a->data.ptr.name, b->data.ptr.name);
495 case AVAHI_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 AVAHI_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 AVAHI_DNS_TYPE_TXT: {
533 ma = g_new(guint8, asize = avahi_string_list_serialize(a->data.txt.string_list, NULL, 0));
534 mb = g_new(guint8, bsize = avahi_string_list_serialize(b->data.txt.string_list, NULL, 0));
535 avahi_string_list_serialize(a->data.txt.string_list, ma, asize);
536 avahi_string_list_serialize(a->data.txt.string_list, mb, bsize);
538 r = lexicographical_memcmp(ma, asize, mb, bsize);
545 case AVAHI_DNS_TYPE_A:
546 return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address));
548 case AVAHI_DNS_TYPE_AAAA:
549 return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address));
552 return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
553 b->data.generic.data, b->data.generic.size);