]> git.meshlink.io Git - catta/blob - rr.c
8b7fab7d861493814faac1e5ed6e0f2d6d5e4967
[catta] / 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 #include <assert.h>
32
33 #include <avahi-common/domain.h>
34 #include <avahi-common/malloc.h>
35 #include <avahi-common/defs.h>
36
37 #include "rr.h"
38 #include "log.h"
39 #include "util.h"
40 #include "hashmap.h"
41 #include "domain-util.h"
42 #include "rr-util.h"
43
44 AvahiKey *avahi_key_new(const char *name, uint16_t class, uint16_t type) {
45     AvahiKey *k;
46     assert(name);
47
48     if (!(k = avahi_new(AvahiKey, 1))) {
49         avahi_log_error("avahi_new() failed.");
50         return NULL;
51     }
52     
53     if (!(k->name = avahi_normalize_name_strdup(name))) {
54         avahi_log_error("avahi_normalize_name() failed.");
55         avahi_free(k);
56         return NULL;
57     }
58     
59     k->ref = 1;
60     k->clazz = class;
61     k->type = type;
62
63     return k;
64 }
65
66 AvahiKey *avahi_key_new_cname(AvahiKey *key) {
67     assert(key);
68
69     if (key->clazz != AVAHI_DNS_CLASS_IN)
70         return NULL;
71
72     if (key->type == AVAHI_DNS_TYPE_CNAME)
73         return NULL;
74
75     return avahi_key_new(key->name, key->clazz, AVAHI_DNS_TYPE_CNAME);
76 }
77
78 AvahiKey *avahi_key_ref(AvahiKey *k) {
79     assert(k);
80     assert(k->ref >= 1);
81
82     k->ref++;
83
84     return k;
85 }
86
87 void avahi_key_unref(AvahiKey *k) {
88     assert(k);
89     assert(k->ref >= 1);
90     
91     if ((--k->ref) <= 0) {
92         avahi_free(k->name);
93         avahi_free(k);
94     }
95 }
96
97 AvahiRecord *avahi_record_new(AvahiKey *k, uint32_t ttl) {
98     AvahiRecord *r;
99     
100     assert(k);
101     
102     if (!(r = avahi_new(AvahiRecord, 1))) {
103         avahi_log_error("avahi_new() failed.");
104         return NULL;
105     }
106         
107     r->ref = 1;
108     r->key = avahi_key_ref(k);
109
110     memset(&r->data, 0, sizeof(r->data));
111
112     r->ttl = ttl != (uint32_t) -1 ? ttl : AVAHI_DEFAULT_TTL;
113
114     return r;
115 }
116
117 AvahiRecord *avahi_record_new_full(const char *name, uint16_t class, uint16_t type, uint32_t ttl) {
118     AvahiRecord *r;
119     AvahiKey *k;
120
121     assert(name);
122     
123     if (!(k = avahi_key_new(name, class, type))) {
124         avahi_log_error("avahi_key_new() failed.");
125         return NULL;
126     }
127
128     r = avahi_record_new(k, ttl);
129     avahi_key_unref(k);
130
131     if (!r) {
132         avahi_log_error("avahi_record_new() failed.");
133         return NULL;
134     }
135
136     return r;
137 }
138
139 AvahiRecord *avahi_record_ref(AvahiRecord *r) {
140     assert(r);
141     assert(r->ref >= 1);
142
143     r->ref++;
144     return r;
145 }
146
147 void avahi_record_unref(AvahiRecord *r) {
148     assert(r);
149     assert(r->ref >= 1);
150
151     if ((--r->ref) <= 0) {
152         switch (r->key->type) {
153
154             case AVAHI_DNS_TYPE_SRV:
155                 avahi_free(r->data.srv.name);
156                 break;
157
158             case AVAHI_DNS_TYPE_PTR:
159             case AVAHI_DNS_TYPE_CNAME:
160             case AVAHI_DNS_TYPE_NS:
161                 avahi_free(r->data.ptr.name);
162                 break;
163
164             case AVAHI_DNS_TYPE_HINFO:
165                 avahi_free(r->data.hinfo.cpu);
166                 avahi_free(r->data.hinfo.os);
167                 break;
168
169             case AVAHI_DNS_TYPE_TXT:
170                 avahi_string_list_free(r->data.txt.string_list);
171                 break;
172
173             case AVAHI_DNS_TYPE_A:
174             case AVAHI_DNS_TYPE_AAAA:
175                 break;
176             
177             default:
178                 avahi_free(r->data.generic.data);
179         }
180         
181         avahi_key_unref(r->key);
182         avahi_free(r);
183     }
184 }
185
186 const char *avahi_dns_class_to_string(uint16_t class) {
187     if (class & AVAHI_DNS_CACHE_FLUSH) 
188         return "FLUSH";
189
190     switch (class) {
191         case AVAHI_DNS_CLASS_IN:
192             return "IN";
193         case AVAHI_DNS_CLASS_ANY:
194             return "ANY";
195         default:
196             return NULL;
197     }
198 }
199
200 const char *avahi_dns_type_to_string(uint16_t type) {
201     switch (type) {
202         case AVAHI_DNS_TYPE_CNAME:
203             return "CNAME";
204         case AVAHI_DNS_TYPE_A:
205             return "A";
206         case AVAHI_DNS_TYPE_AAAA:
207             return "AAAA";
208         case AVAHI_DNS_TYPE_PTR:
209             return "PTR";
210         case AVAHI_DNS_TYPE_HINFO:
211             return "HINFO";
212         case AVAHI_DNS_TYPE_TXT:
213             return "TXT";
214         case AVAHI_DNS_TYPE_SRV:
215             return "SRV";
216         case AVAHI_DNS_TYPE_ANY:
217             return "ANY";
218         case AVAHI_DNS_TYPE_SOA:
219             return "SOA";
220         case AVAHI_DNS_TYPE_NS:
221             return "NS";
222         default:
223             return NULL;
224     }
225 }
226
227 char *avahi_key_to_string(const AvahiKey *k) {
228     char class[16], type[16];
229     const char *c, *t;
230     
231     assert(k);
232     assert(k->ref >= 1);
233
234     /* According to RFC3597 */
235     
236     if (!(c = avahi_dns_class_to_string(k->clazz))) {
237         snprintf(class, sizeof(class), "CLASS%u", k->clazz);
238         c = class;
239     }
240
241     if (!(t = avahi_dns_type_to_string(k->type))) {
242         snprintf(type, sizeof(type), "TYPE%u", k->type);
243         t = type;
244     }
245     
246     return avahi_strdup_printf("%s\t%s\t%s", k->name, c, t);
247 }
248
249 char *avahi_record_to_string(const AvahiRecord *r) {
250     char *p, *s;
251     char buf[1024], *t = NULL, *d = NULL;
252
253     assert(r);
254     assert(r->ref >= 1);
255     
256     switch (r->key->type) {
257         case AVAHI_DNS_TYPE_A:
258             inet_ntop(AF_INET, &r->data.a.address.address, t = buf, sizeof(buf));
259             break;
260             
261         case AVAHI_DNS_TYPE_AAAA:
262             inet_ntop(AF_INET6, &r->data.aaaa.address.address, t = buf, sizeof(buf));
263             break;
264             
265         case AVAHI_DNS_TYPE_PTR:
266         case AVAHI_DNS_TYPE_CNAME:
267         case AVAHI_DNS_TYPE_NS:
268
269             t = r->data.ptr.name;
270             break;
271
272         case AVAHI_DNS_TYPE_TXT:
273             t = d = avahi_string_list_to_string(r->data.txt.string_list);
274             break;
275
276         case AVAHI_DNS_TYPE_HINFO:
277
278             snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
279             break;
280
281         case AVAHI_DNS_TYPE_SRV:
282
283             snprintf(t = buf, sizeof(buf), "%u %u %u %s",
284                      r->data.srv.priority,
285                      r->data.srv.weight,
286                      r->data.srv.port,
287                      r->data.srv.name);
288
289             break;
290
291         default: {
292
293             uint8_t *c;
294             uint16_t n;
295             int i;
296             char *e;
297
298             /* According to RFC3597 */
299             
300             snprintf(t = buf, sizeof(buf), "\\# %u", r->data.generic.size);
301
302             e = strchr(t, 0);
303             
304             for (c = r->data.generic.data, n = r->data.generic.size, i = 0;
305                  n > 0 && i < 20;
306                  c ++, n --, i++) {
307
308                 sprintf(e, " %02X", *c);
309                 e = strchr(e, 0);
310             }
311
312             break;
313         }
314     }
315
316     p = avahi_key_to_string(r->key);
317     s = avahi_strdup_printf("%s %s ; ttl=%u", p, t, r->ttl);
318     avahi_free(p);
319     avahi_free(d);
320     
321     return s;
322 }
323
324 int avahi_key_equal(const AvahiKey *a, const AvahiKey *b) {
325     assert(a);
326     assert(b);
327
328     if (a == b)
329         return 1;
330     
331     return avahi_domain_equal(a->name, b->name) &&
332         a->type == b->type &&
333         a->clazz == b->clazz;
334 }
335
336 int avahi_key_pattern_match(const AvahiKey *pattern, const AvahiKey *k) {
337     assert(pattern);
338     assert(k);
339
340     assert(!avahi_key_is_pattern(k));
341
342     if (pattern == k)
343         return 1;
344     
345     return avahi_domain_equal(pattern->name, k->name) &&
346         (pattern->type == k->type || pattern->type == AVAHI_DNS_TYPE_ANY) &&
347         (pattern->clazz == k->clazz || pattern->clazz == AVAHI_DNS_CLASS_ANY);
348 }
349
350 int avahi_key_is_pattern(const AvahiKey *k) {
351     assert(k);
352
353     return
354         k->type == AVAHI_DNS_TYPE_ANY ||
355         k->clazz == AVAHI_DNS_CLASS_ANY;
356 }
357
358 unsigned avahi_key_hash(const AvahiKey *k) {
359     assert(k);
360
361     return
362         avahi_domain_hash(k->name) + 
363         k->type +
364         k->clazz;
365 }
366
367 static int rdata_equal(const AvahiRecord *a, const AvahiRecord *b) {
368     assert(a);
369     assert(b);
370     assert(a->key->type == b->key->type);
371
372     switch (a->key->type) {
373         case AVAHI_DNS_TYPE_SRV:
374             return
375                 a->data.srv.priority == b->data.srv.priority &&
376                 a->data.srv.weight == b->data.srv.weight &&
377                 a->data.srv.port == b->data.srv.port &&
378                 avahi_domain_equal(a->data.srv.name, b->data.srv.name);
379
380         case AVAHI_DNS_TYPE_PTR:
381         case AVAHI_DNS_TYPE_CNAME:
382         case AVAHI_DNS_TYPE_NS:
383             return avahi_domain_equal(a->data.ptr.name, b->data.ptr.name);
384
385         case AVAHI_DNS_TYPE_HINFO:
386             return
387                 !strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
388                 !strcmp(a->data.hinfo.os, b->data.hinfo.os);
389
390         case AVAHI_DNS_TYPE_TXT:
391             return avahi_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
392
393         case AVAHI_DNS_TYPE_A:
394             return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address)) == 0;
395
396         case AVAHI_DNS_TYPE_AAAA:
397             return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address)) == 0;
398
399         default:
400             return a->data.generic.size == b->data.generic.size &&
401                 (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
402     }
403     
404 }
405
406 int avahi_record_equal_no_ttl(const AvahiRecord *a, const AvahiRecord *b) {
407     assert(a);
408     assert(b);
409
410     if (a == b)
411         return 1;
412
413     return
414         avahi_key_equal(a->key, b->key) &&
415         rdata_equal(a, b);
416 }
417
418
419 AvahiRecord *avahi_record_copy(AvahiRecord *r) {
420     AvahiRecord *copy;
421
422     if (!(copy = avahi_new(AvahiRecord, 1))) {
423         avahi_log_error("avahi_new() failed.");
424         return NULL;
425     }
426     
427     copy->ref = 1;
428     copy->key = avahi_key_ref(r->key);
429     copy->ttl = r->ttl;
430
431     switch (r->key->type) {
432         case AVAHI_DNS_TYPE_PTR:
433         case AVAHI_DNS_TYPE_CNAME:
434         case AVAHI_DNS_TYPE_NS:
435             if (!(copy->data.ptr.name = avahi_strdup(r->data.ptr.name)))
436                 goto fail;
437             break;
438
439         case AVAHI_DNS_TYPE_SRV:
440             copy->data.srv.priority = r->data.srv.priority;
441             copy->data.srv.weight = r->data.srv.weight;
442             copy->data.srv.port = r->data.srv.port;
443             if (!(copy->data.srv.name = avahi_strdup(r->data.srv.name)))
444                 goto fail;
445             break;
446
447         case AVAHI_DNS_TYPE_HINFO:
448             if (!(copy->data.hinfo.os = avahi_strdup(r->data.hinfo.os)))
449                 goto fail;
450
451             if (!(copy->data.hinfo.cpu = avahi_strdup(r->data.hinfo.cpu))) {
452                 avahi_free(r->data.hinfo.os);
453                 goto fail;
454             }
455             break;
456
457         case AVAHI_DNS_TYPE_TXT:
458             copy->data.txt.string_list = avahi_string_list_copy(r->data.txt.string_list);
459             break;
460
461         case AVAHI_DNS_TYPE_A:
462             copy->data.a.address = r->data.a.address;
463             break;
464
465         case AVAHI_DNS_TYPE_AAAA:
466             copy->data.aaaa.address = r->data.aaaa.address;
467             break;
468
469         default:
470             if (!(copy->data.generic.data = avahi_memdup(r->data.generic.data, r->data.generic.size)))
471                 goto fail;
472             copy->data.generic.size = r->data.generic.size;
473             break;
474                 
475     }
476
477     return copy;
478
479 fail:
480     avahi_log_error("Failed to allocate memory");
481
482     avahi_key_unref(copy->key);
483     avahi_free(copy);
484     
485     return NULL;
486 }
487
488
489 size_t avahi_key_get_estimate_size(AvahiKey *k) {
490     assert(k);
491
492     return strlen(k->name)+1+4;
493 }
494
495 size_t avahi_record_get_estimate_size(AvahiRecord *r) {
496     size_t n;
497     assert(r);
498
499     n = avahi_key_get_estimate_size(r->key) + 4 + 2;
500
501     switch (r->key->type) {
502         case AVAHI_DNS_TYPE_PTR:
503         case AVAHI_DNS_TYPE_CNAME:
504         case AVAHI_DNS_TYPE_NS:
505             n += strlen(r->data.ptr.name) + 1;
506             break;
507
508         case AVAHI_DNS_TYPE_SRV:
509             n += 6 + strlen(r->data.srv.name) + 1;
510             break;
511
512         case AVAHI_DNS_TYPE_HINFO:
513             n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
514             break;
515
516         case AVAHI_DNS_TYPE_TXT:
517             n += avahi_string_list_serialize(r->data.txt.string_list, NULL, 0);
518             break;
519
520         case AVAHI_DNS_TYPE_A:
521             n += sizeof(AvahiIPv4Address);
522             break;
523
524         case AVAHI_DNS_TYPE_AAAA:
525             n += sizeof(AvahiIPv6Address);
526             break;
527
528         default:
529             n += r->data.generic.size;
530     }
531
532     return n;
533 }
534
535 static int lexicographical_memcmp(const void* a, size_t al, const void* b, size_t bl) {
536     size_t c;
537     int ret;
538     
539     assert(a);
540     assert(b);
541
542     c = al < bl ? al : bl;
543     if ((ret = memcmp(a, b, c)))
544         return ret;
545
546     if (al == bl)
547         return 0;
548     else
549         return al == c ? 1 : -1;
550 }
551
552 static int uint16_cmp(uint16_t a, uint16_t b) {
553     return a == b ? 0 : (a < b ? -1 : 1);
554 }
555
556 int avahi_record_lexicographical_compare(AvahiRecord *a, AvahiRecord *b) {
557     int r;
558 /*      char *t1, *t2; */
559
560     assert(a);
561     assert(b);
562
563 /*     t1 = avahi_record_to_string(a); */
564 /*     t2 = avahi_record_to_string(b); */
565 /*     g_message("lexicocmp: %s %s", t1, t2); */
566 /*     avahi_free(t1); */
567 /*     avahi_free(t2); */
568
569     if (a == b)
570         return 0;
571
572     if ((r = uint16_cmp(a->key->clazz, b->key->clazz)) ||
573         (r = uint16_cmp(a->key->type, b->key->type)))
574         return r;
575
576     switch (a->key->type) {
577
578         case AVAHI_DNS_TYPE_PTR:
579         case AVAHI_DNS_TYPE_CNAME:
580         case AVAHI_DNS_TYPE_NS:
581             return avahi_binary_domain_cmp(a->data.ptr.name, b->data.ptr.name);
582
583         case AVAHI_DNS_TYPE_SRV: {
584             if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) == 0 &&
585                 (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) == 0 &&
586                 (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) == 0)
587                 r = avahi_binary_domain_cmp(a->data.srv.name, b->data.srv.name);
588             
589             return r;
590         }
591
592         case AVAHI_DNS_TYPE_HINFO: {
593
594             if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) ||
595                 (r = strcmp(a->data.hinfo.os, b->data.hinfo.os)))
596                 return r;
597
598             return 0;
599
600         }
601
602         case AVAHI_DNS_TYPE_TXT: {
603
604             uint8_t *ma = NULL, *mb = NULL;
605             size_t asize, bsize;
606
607             asize = avahi_string_list_serialize(a->data.txt.string_list, NULL, 0);
608             bsize = avahi_string_list_serialize(b->data.txt.string_list, NULL, 0);
609             
610             if (asize > 0 && !(ma = avahi_new(uint8_t, asize)))
611                 goto fail;
612             
613             if (bsize > 0 && !(mb = avahi_new(uint8_t, bsize))) {
614                 avahi_free(ma);
615                 goto fail;
616             }
617             
618             avahi_string_list_serialize(a->data.txt.string_list, ma, asize);
619             avahi_string_list_serialize(b->data.txt.string_list, mb, bsize);
620
621             if (asize && bsize)
622                 r = lexicographical_memcmp(ma, asize, mb, bsize);
623             else if (asize && !bsize)
624                 r = 1;
625             else if (!asize && bsize)
626                 r = -1;
627             else
628                 r = 0;
629             
630             avahi_free(ma);
631             avahi_free(mb);
632
633             return r;
634         }
635         
636         case AVAHI_DNS_TYPE_A:
637             return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address));
638
639         case AVAHI_DNS_TYPE_AAAA:
640             return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address));
641
642         default:
643             return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
644                                           b->data.generic.data, b->data.generic.size);
645     }
646
647     
648 fail:
649     avahi_log_error(__FILE__": Out of memory");
650     return -1; /* or whatever ... */
651 }
652
653 int avahi_record_is_goodbye(AvahiRecord *r) {
654     assert(r);
655
656     return r->ttl == 0;
657 }
658
659 int avahi_key_is_valid(AvahiKey *k) {
660     assert(k);
661
662     if (!avahi_is_valid_domain_name(k->name))
663         return 0;
664     
665     return 1;
666 }
667
668 int avahi_record_is_valid(AvahiRecord *r) {
669     assert(r);
670
671     if (!avahi_key_is_valid(r->key))
672         return 0;
673
674     switch (r->key->type) {
675
676         case AVAHI_DNS_TYPE_PTR:
677         case AVAHI_DNS_TYPE_CNAME:
678         case AVAHI_DNS_TYPE_NS:
679             return avahi_is_valid_domain_name(r->data.ptr.name);
680
681         case AVAHI_DNS_TYPE_SRV:
682             return avahi_is_valid_domain_name(r->data.srv.name);
683
684         case AVAHI_DNS_TYPE_HINFO:
685             return
686                 strlen(r->data.hinfo.os) <= 255 &&
687                 strlen(r->data.hinfo.cpu) <= 255;
688             
689         case AVAHI_DNS_TYPE_TXT: {
690
691             AvahiStringList *strlst;
692
693             for (strlst = r->data.txt.string_list; strlst; strlst = strlst->next)
694                 if (strlst->size > 255 || strlst->size <= 0)
695                     return 0;
696
697             return 1;
698         }
699     }
700             
701
702     return 1;
703 }