]> git.meshlink.io Git - catta/blob - avahi-common/rr.c
move dns.[ch] back to avahi-core/
[catta] / avahi-common / rr.c
1 /* $Id$ */
2
3 /***
4   This file is part of avahi.
5  
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.
10  
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.
15  
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
19   USA.
20 ***/
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <string.h>
27 #include <stdio.h>
28 #include <sys/types.h>
29 #include <sys/socket.h>
30 #include <arpa/inet.h>
31
32 #include "util.h"
33 #include "rr.h"
34
35 AvahiKey *avahi_key_new(const gchar *name, guint16 class, guint16 type) {
36     AvahiKey *k;
37     g_assert(name);
38
39     k = g_new(AvahiKey, 1);
40     k->ref = 1;
41     k->name = avahi_normalize_name(name);    
42     k->class = class;
43     k->type = type;
44
45 /*     g_message("%p %% ref=1", k); */
46     
47     return k;
48 }
49
50 AvahiKey *avahi_key_ref(AvahiKey *k) {
51     g_assert(k);
52     g_assert(k->ref >= 1);
53
54     k->ref++;
55
56 /*     g_message("%p ++ ref=%i", k, k->ref); */
57
58     return k;
59 }
60
61 void avahi_key_unref(AvahiKey *k) {
62     g_assert(k);
63     g_assert(k->ref >= 1);
64
65 /*     g_message("%p -- ref=%i", k, k->ref-1); */
66     
67     if ((--k->ref) <= 0) {
68         g_free(k->name);
69         g_free(k);
70     }
71 }
72
73 AvahiRecord *avahi_record_new(AvahiKey *k) {
74     AvahiRecord *r;
75     
76     g_assert(k);
77     
78     r = g_new(AvahiRecord, 1);
79     r->ref = 1;
80     r->key = avahi_key_ref(k);
81
82     memset(&r->data, 0, sizeof(r->data));
83
84     r->ttl = AVAHI_DEFAULT_TTL;
85
86     return r;
87 }
88
89 AvahiRecord *avahi_record_new_full(const gchar *name, guint16 class, guint16 type) {
90     AvahiRecord *r;
91     AvahiKey *k;
92
93     g_assert(name);
94     
95     k = avahi_key_new(name, class, type);
96     r = avahi_record_new(k);
97     avahi_key_unref(k);
98
99     return r;
100 }
101
102 AvahiRecord *avahi_record_ref(AvahiRecord *r) {
103     g_assert(r);
104     g_assert(r->ref >= 1);
105
106     r->ref++;
107     return r;
108 }
109
110 void avahi_record_unref(AvahiRecord *r) {
111     g_assert(r);
112     g_assert(r->ref >= 1);
113
114     if ((--r->ref) <= 0) {
115         switch (r->key->type) {
116
117             case AVAHI_DNS_TYPE_SRV:
118                 g_free(r->data.srv.name);
119                 break;
120
121             case AVAHI_DNS_TYPE_PTR:
122             case AVAHI_DNS_TYPE_CNAME:
123                 g_free(r->data.ptr.name);
124                 break;
125
126             case AVAHI_DNS_TYPE_HINFO:
127                 g_free(r->data.hinfo.cpu);
128                 g_free(r->data.hinfo.os);
129                 break;
130
131             case AVAHI_DNS_TYPE_TXT:
132                 avahi_string_list_free(r->data.txt.string_list);
133                 break;
134
135             case AVAHI_DNS_TYPE_A:
136             case AVAHI_DNS_TYPE_AAAA:
137                 break;
138             
139             default:
140                 g_free(r->data.generic.data);
141         }
142         
143         avahi_key_unref(r->key);
144         g_free(r);
145     }
146 }
147
148 const gchar *avahi_dns_class_to_string(guint16 class) {
149     if (class & AVAHI_DNS_CACHE_FLUSH) 
150         return "FLUSH";
151     
152     if (class == AVAHI_DNS_CLASS_IN)
153         return "IN";
154
155     return NULL;
156 }
157
158 const gchar *avahi_dns_type_to_string(guint16 type) {
159     switch (type) {
160         case AVAHI_DNS_TYPE_CNAME:
161             return "CNAME";
162         case AVAHI_DNS_TYPE_A:
163             return "A";
164         case AVAHI_DNS_TYPE_AAAA:
165             return "AAAA";
166         case AVAHI_DNS_TYPE_PTR:
167             return "PTR";
168         case AVAHI_DNS_TYPE_HINFO:
169             return "HINFO";
170         case AVAHI_DNS_TYPE_TXT:
171             return "TXT";
172         case AVAHI_DNS_TYPE_SRV:
173             return "SRV";
174         case AVAHI_DNS_TYPE_ANY:
175             return "ANY";
176         default:
177             return NULL;
178     }
179 }
180
181
182 gchar *avahi_key_to_string(const AvahiKey *k) {
183     g_assert(k);
184     g_assert(k->ref >= 1);
185     
186     return g_strdup_printf("%s\t%s\t%s",
187                            k->name,
188                            avahi_dns_class_to_string(k->class),
189                            avahi_dns_type_to_string(k->type));
190 }
191
192 gchar *avahi_record_to_string(const AvahiRecord *r) {
193     gchar *p, *s;
194     char buf[257], *t = NULL, *d = NULL;
195
196     g_assert(r);
197     g_assert(r->ref >= 1);
198     
199     switch (r->key->type) {
200         case AVAHI_DNS_TYPE_A:
201             inet_ntop(AF_INET, &r->data.a.address.address, t = buf, sizeof(buf));
202             break;
203             
204         case AVAHI_DNS_TYPE_AAAA:
205             inet_ntop(AF_INET6, &r->data.aaaa.address.address, t = buf, sizeof(buf));
206             break;
207             
208         case AVAHI_DNS_TYPE_PTR:
209         case AVAHI_DNS_TYPE_CNAME :
210
211             t = r->data.ptr.name;
212             break;
213
214         case AVAHI_DNS_TYPE_TXT:
215             t = d = avahi_string_list_to_string(r->data.txt.string_list);
216             break;
217
218         case AVAHI_DNS_TYPE_HINFO:
219
220             snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
221             break;
222
223         case AVAHI_DNS_TYPE_SRV:
224
225             snprintf(t = buf, sizeof(buf), "%u %u %u %s",
226                      r->data.srv.priority,
227                      r->data.srv.weight,
228                      r->data.srv.port,
229                      r->data.srv.name);
230
231             break;
232     }
233
234     p = avahi_key_to_string(r->key);
235     s = g_strdup_printf("%s %s ; ttl=%u", p, t ? t : "<unparsable>", r->ttl);
236     g_free(p);
237     g_free(d);
238     
239     return s;
240 }
241
242 gboolean avahi_key_equal(const AvahiKey *a, const AvahiKey *b) {
243     g_assert(a);
244     g_assert(b);
245
246     if (a == b)
247         return TRUE;
248     
249 /*     g_message("equal: %p %p", a, b); */
250     
251     return avahi_domain_equal(a->name, b->name) &&
252         a->type == b->type &&
253         a->class == b->class;
254 }
255
256 gboolean avahi_key_pattern_match(const AvahiKey *pattern, const AvahiKey *k) {
257     g_assert(pattern);
258     g_assert(k);
259
260 /*     g_message("equal: %p %p", a, b); */
261
262     g_assert(!avahi_key_is_pattern(k));
263
264     if (pattern == k)
265         return TRUE;
266     
267     return avahi_domain_equal(pattern->name, k->name) &&
268         (pattern->type == k->type || pattern->type == AVAHI_DNS_TYPE_ANY) &&
269         pattern->class == k->class;
270 }
271
272 gboolean avahi_key_is_pattern(const AvahiKey *k) {
273     g_assert(k);
274
275     return k->type == AVAHI_DNS_TYPE_ANY;
276 }
277
278
279 guint avahi_key_hash(const AvahiKey *k) {
280     g_assert(k);
281
282     return
283         avahi_domain_hash(k->name) + 
284         k->type +
285         k->class;
286 }
287
288 static gboolean rdata_equal(const AvahiRecord *a, const AvahiRecord *b) {
289     g_assert(a);
290     g_assert(b);
291     g_assert(a->key->type == b->key->type);
292
293 /*     t = avahi_record_to_string(a); */
294 /*     g_message("comparing %s", t); */
295 /*     g_free(t); */
296
297 /*     t = avahi_record_to_string(b); */
298 /*     g_message("and %s", t); */
299 /*     g_free(t); */
300
301     
302     switch (a->key->type) {
303         case AVAHI_DNS_TYPE_SRV:
304             return
305                 a->data.srv.priority == b->data.srv.priority &&
306                 a->data.srv.weight == b->data.srv.weight &&
307                 a->data.srv.port == b->data.srv.port &&
308                 avahi_domain_equal(a->data.srv.name, b->data.srv.name);
309
310         case AVAHI_DNS_TYPE_PTR:
311         case AVAHI_DNS_TYPE_CNAME:
312             return avahi_domain_equal(a->data.ptr.name, b->data.ptr.name);
313
314         case AVAHI_DNS_TYPE_HINFO:
315             return
316                 !g_utf8_collate(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
317                 !g_utf8_collate(a->data.hinfo.os, b->data.hinfo.os);
318
319         case AVAHI_DNS_TYPE_TXT:
320             return avahi_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
321
322         case AVAHI_DNS_TYPE_A:
323             return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address)) == 0;
324
325         case AVAHI_DNS_TYPE_AAAA:
326             return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address)) == 0;
327
328         default:
329             return a->data.generic.size == b->data.generic.size &&
330                 (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
331     }
332     
333 }
334
335 gboolean avahi_record_equal_no_ttl(const AvahiRecord *a, const AvahiRecord *b) {
336     g_assert(a);
337     g_assert(b);
338
339     if (a == b)
340         return TRUE;
341
342     return
343         avahi_key_equal(a->key, b->key) &&
344         rdata_equal(a, b);
345 }
346
347
348 AvahiRecord *avahi_record_copy(AvahiRecord *r) {
349     AvahiRecord *copy;
350
351     copy = g_new(AvahiRecord, 1);
352     copy->ref = 1;
353     copy->key = avahi_key_ref(r->key);
354     copy->ttl = r->ttl;
355
356     switch (r->key->type) {
357         case AVAHI_DNS_TYPE_PTR:
358         case AVAHI_DNS_TYPE_CNAME:
359             copy->data.ptr.name = g_strdup(r->data.ptr.name);
360             break;
361
362         case AVAHI_DNS_TYPE_SRV:
363             copy->data.srv.priority = r->data.srv.priority;
364             copy->data.srv.weight = r->data.srv.weight;
365             copy->data.srv.port = r->data.srv.port;
366             copy->data.srv.name = g_strdup(r->data.srv.name);
367             break;
368
369         case AVAHI_DNS_TYPE_HINFO:
370             copy->data.hinfo.os = g_strdup(r->data.hinfo.os);
371             copy->data.hinfo.cpu = g_strdup(r->data.hinfo.cpu);
372             break;
373
374         case AVAHI_DNS_TYPE_TXT:
375             copy->data.txt.string_list = avahi_string_list_copy(r->data.txt.string_list);
376             break;
377
378         case AVAHI_DNS_TYPE_A:
379             copy->data.a.address = r->data.a.address;
380             break;
381
382         case AVAHI_DNS_TYPE_AAAA:
383             copy->data.aaaa.address = r->data.aaaa.address;
384             break;
385
386         default:
387             copy->data.generic.data = g_memdup(r->data.generic.data, r->data.generic.size);
388             copy->data.generic.size = r->data.generic.size;
389             break;
390                 
391     }
392
393     return copy;
394 }
395
396
397 guint avahi_key_get_estimate_size(AvahiKey *k) {
398     g_assert(k);
399
400     return strlen(k->name)+1+4;
401 }
402
403 guint avahi_record_get_estimate_size(AvahiRecord *r) {
404     guint n;
405     g_assert(r);
406
407     n = avahi_key_get_estimate_size(r->key) + 4 + 2;
408
409     switch (r->key->type) {
410         case AVAHI_DNS_TYPE_PTR:
411         case AVAHI_DNS_TYPE_CNAME:
412             n += strlen(r->data.ptr.name) + 1;
413             break;
414
415         case AVAHI_DNS_TYPE_SRV:
416             n += 6 + strlen(r->data.srv.name) + 1;
417             break;
418
419         case AVAHI_DNS_TYPE_HINFO:
420             n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
421             break;
422
423         case AVAHI_DNS_TYPE_TXT:
424             n += avahi_string_list_serialize(r->data.txt.string_list, NULL, 0);
425             break;
426
427         case AVAHI_DNS_TYPE_A:
428             n += sizeof(AvahiIPv4Address);
429             break;
430
431         case AVAHI_DNS_TYPE_AAAA:
432             n += sizeof(AvahiIPv6Address);
433             break;
434
435         default:
436             n += r->data.generic.size;
437     }
438
439     return n;
440 }
441
442 static gint lexicographical_memcmp(gconstpointer a, size_t al, gconstpointer b, size_t bl) {
443     size_t c;
444     gint ret;
445     
446     g_assert(a);
447     g_assert(b);
448
449     c = al < bl ? al : bl;
450     if ((ret = memcmp(a, b, c)))
451         return ret;
452
453     if (al == bl)
454         return 0;
455     else
456         return al == c ? 1 : -1;
457 }
458
459 static gint uint16_cmp(guint16 a, guint16 b) {
460     return a == b ? 0 : (a < b ? -1 : 1);
461 }
462
463 gint avahi_record_lexicographical_compare(AvahiRecord *a, AvahiRecord *b) {
464     gint r;
465 /*      gchar *t1, *t2; */
466
467     g_assert(a);
468     g_assert(b);
469
470 /*     t1 = avahi_record_to_string(a); */
471 /*     t2 = avahi_record_to_string(b); */
472 /*     g_message("lexicocmp: %s %s", t1, t2); */
473 /*     g_free(t1); */
474 /*     g_free(t2); */
475
476     if (a == b)
477         return 0;
478
479     if ((r = uint16_cmp(a->key->class, b->key->class)) ||
480         (r = uint16_cmp(a->key->type, b->key->type)))
481         return r;
482
483     switch (a->key->type) {
484
485         case AVAHI_DNS_TYPE_PTR:
486         case AVAHI_DNS_TYPE_CNAME:
487             return avahi_binary_domain_cmp(a->data.ptr.name, b->data.ptr.name);
488
489         case AVAHI_DNS_TYPE_SRV: {
490             if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) == 0 &&
491                 (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) == 0 &&
492                 (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) == 0)
493                 r = avahi_binary_domain_cmp(a->data.srv.name, b->data.srv.name);
494             
495             return r;
496         }
497
498         case AVAHI_DNS_TYPE_HINFO: {
499
500             if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) ||
501                 (r = strcmp(a->data.hinfo.os, b->data.hinfo.os)))
502                 return r;
503
504             return 0;
505
506         }
507
508         case AVAHI_DNS_TYPE_TXT: {
509
510             guint8 *ma, *mb;
511             guint asize, bsize;
512
513             ma = g_new(guint8, asize = avahi_string_list_serialize(a->data.txt.string_list, NULL, 0));
514             mb = g_new(guint8, bsize = avahi_string_list_serialize(b->data.txt.string_list, NULL, 0));
515             avahi_string_list_serialize(a->data.txt.string_list, ma, asize);
516             avahi_string_list_serialize(b->data.txt.string_list, mb, bsize);
517
518             if (asize && bsize)
519                 r = lexicographical_memcmp(ma, asize, mb, bsize);
520             else if (asize && !bsize)
521                 r = 1;
522             else if (!asize && bsize)
523                 r = -1;
524             else
525                 r = 0;
526             
527             g_free(ma);
528             g_free(mb);
529
530             return r;
531         }
532         
533         case AVAHI_DNS_TYPE_A:
534             return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address));
535
536         case AVAHI_DNS_TYPE_AAAA:
537             return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address));
538
539         default:
540             return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
541                                           b->data.generic.data, b->data.generic.size);
542     }
543     
544 }
545
546 gboolean avahi_record_is_goodbye(AvahiRecord *r) {
547     g_assert(r);
548
549     return r->ttl == 0;
550 }