]> git.meshlink.io Git - catta/blobdiff - rr.c
rename flx.h to avahi.h
[catta] / rr.c
diff --git a/rr.c b/rr.c
index 600e7219bb7b58116f378f8b300b06ca05e4261a..f4102dd45cb01a38069f44c7566a58f5346813e3 100644 (file)
--- a/rr.c
+++ b/rr.c
@@ -6,14 +6,15 @@
 
 #include "util.h"
 #include "rr.h"
+#include "dns.h"
 
-flxKey *flx_key_new(const gchar *name, guint16 class, guint16 type) {
-    flxKey *k;
+AvahiKey *avahi_key_new(const gchar *name, guint16 class, guint16 type) {
+    AvahiKey *k;
     g_assert(name);
 
-    k = g_new(flxKey, 1);
+    k = g_new(AvahiKey, 1);
     k->ref = 1;
-    k->name = flx_normalize_name(name);    
+    k->name = avahi_normalize_name(name);    
     k->class = class;
     k->type = type;
 
@@ -22,7 +23,7 @@ flxKey *flx_key_new(const gchar *name, guint16 class, guint16 type) {
     return k;
 }
 
-flxKey *flx_key_ref(flxKey *k) {
+AvahiKey *avahi_key_ref(AvahiKey *k) {
     g_assert(k);
     g_assert(k->ref >= 1);
 
@@ -33,7 +34,7 @@ flxKey *flx_key_ref(flxKey *k) {
     return k;
 }
 
-void flx_key_unref(flxKey *k) {
+void avahi_key_unref(AvahiKey *k) {
     g_assert(k);
     g_assert(k->ref >= 1);
 
@@ -45,37 +46,36 @@ void flx_key_unref(flxKey *k) {
     }
 }
 
-flxRecord *flx_record_new(flxKey *k, gconstpointer data, guint16 size, guint32 ttl) {
-    flxRecord *r;
+AvahiRecord *avahi_record_new(AvahiKey *k) {
+    AvahiRecord *r;
     
     g_assert(k);
-    g_assert(size == 0 || data);
     
-    r = g_new(flxRecord, 1);
+    r = g_new(AvahiRecord, 1);
     r->ref = 1;
-    r->key = flx_key_ref(k);
-    r->data = size > 0 ? g_memdup(data, size) : NULL;
-    r->size = size;
-    r->ttl = ttl;
+    r->key = avahi_key_ref(k);
+
+    memset(&r->data, 0, sizeof(r->data));
+
+    r->ttl = AVAHI_DEFAULT_TTL;
 
     return r;
 }
 
-flxRecord *flx_record_new_full(const gchar *name, guint16 class, guint16 type, gconstpointer data, guint16 size, guint32 ttl) {
-    flxRecord *r;
-    flxKey *k;
+AvahiRecord *avahi_record_new_full(const gchar *name, guint16 class, guint16 type) {
+    AvahiRecord *r;
+    AvahiKey *k;
 
     g_assert(name);
-    g_assert(size == 0 || data);
     
-    k = flx_key_new(name, class, type);
-    r = flx_record_new(k, data, size, ttl);
-    flx_key_unref(k);
+    k = avahi_key_new(name, class, type);
+    r = avahi_record_new(k);
+    avahi_key_unref(k);
 
     return r;
 }
 
-flxRecord *flx_record_ref(flxRecord *r) {
+AvahiRecord *avahi_record_ref(AvahiRecord *r) {
     g_assert(r);
     g_assert(r->ref >= 1);
 
@@ -83,184 +83,474 @@ flxRecord *flx_record_ref(flxRecord *r) {
     return r;
 }
 
-void flx_record_unref(flxRecord *r) {
+void avahi_record_unref(AvahiRecord *r) {
     g_assert(r);
     g_assert(r->ref >= 1);
 
     if ((--r->ref) <= 0) {
-        flx_key_unref(r->key);
-        g_free(r->data);
+        switch (r->key->type) {
+
+            case AVAHI_DNS_TYPE_SRV:
+                g_free(r->data.srv.name);
+                break;
+
+            case AVAHI_DNS_TYPE_PTR:
+            case AVAHI_DNS_TYPE_CNAME:
+                g_free(r->data.ptr.name);
+                break;
+
+            case AVAHI_DNS_TYPE_HINFO:
+                g_free(r->data.hinfo.cpu);
+                g_free(r->data.hinfo.os);
+                break;
+
+            case AVAHI_DNS_TYPE_TXT:
+                avahi_string_list_free(r->data.txt.string_list);
+                break;
+
+            case AVAHI_DNS_TYPE_A:
+            case AVAHI_DNS_TYPE_AAAA:
+                break;
+            
+            default:
+                g_free(r->data.generic.data);
+        }
+        
+        avahi_key_unref(r->key);
         g_free(r);
     }
 }
 
-const gchar *flx_dns_class_to_string(guint16 class) {
-    if (class == FLX_DNS_CLASS_IN)
+const gchar *avahi_dns_class_to_string(guint16 class) {
+    if (class & AVAHI_DNS_CACHE_FLUSH) 
+        return "FLUSH";
+    
+    if (class == AVAHI_DNS_CLASS_IN)
         return "IN";
 
     return NULL;
 }
 
-const gchar *flx_dns_type_to_string(guint16 type) {
+const gchar *avahi_dns_type_to_string(guint16 type) {
     switch (type) {
-        case FLX_DNS_TYPE_CNAME:
+        case AVAHI_DNS_TYPE_CNAME:
             return "CNAME";
-        case FLX_DNS_TYPE_A:
+        case AVAHI_DNS_TYPE_A:
             return "A";
-        case FLX_DNS_TYPE_AAAA:
+        case AVAHI_DNS_TYPE_AAAA:
             return "AAAA";
-        case FLX_DNS_TYPE_PTR:
+        case AVAHI_DNS_TYPE_PTR:
             return "PTR";
-        case FLX_DNS_TYPE_HINFO:
+        case AVAHI_DNS_TYPE_HINFO:
             return "HINFO";
-        case FLX_DNS_TYPE_TXT:
+        case AVAHI_DNS_TYPE_TXT:
             return "TXT";
-        case FLX_DNS_TYPE_SRV:
+        case AVAHI_DNS_TYPE_SRV:
             return "SRV";
+        case AVAHI_DNS_TYPE_ANY:
+            return "ANY";
         default:
             return NULL;
     }
 }
 
 
-gchar *flx_key_to_string(flxKey *k) {
+gchar *avahi_key_to_string(const AvahiKey *k) {
     return g_strdup_printf("%s\t%s\t%s",
                            k->name,
-                           flx_dns_class_to_string(k->class),
-                           flx_dns_type_to_string(k->type));
+                           avahi_dns_class_to_string(k->class),
+                           avahi_dns_type_to_string(k->type));
 }
 
-gchar *flx_record_to_string(flxRecord *r) {
+gchar *avahi_record_to_string(const AvahiRecord *r) {
     gchar *p, *s;
-    char t[257] = "<unparsable>";
+    char buf[257], *t = NULL, *d = NULL;
 
     switch (r->key->type) {
-        case FLX_DNS_TYPE_A:
-            inet_ntop(AF_INET, r->data, t, sizeof(t));
+        case AVAHI_DNS_TYPE_A:
+            inet_ntop(AF_INET, &r->data.a.address.address, t = buf, sizeof(buf));
             break;
             
-        case FLX_DNS_TYPE_AAAA:
-            inet_ntop(AF_INET6, r->data, t, sizeof(t));
+        case AVAHI_DNS_TYPE_AAAA:
+            inet_ntop(AF_INET6, &r->data.aaaa.address.address, t = buf, sizeof(buf));
             break;
             
-        case FLX_DNS_TYPE_PTR: {
-            size_t l;
-        
-            l = r->size;
-            if (l > sizeof(t)-1)
-                l = sizeof(t)-1;
-            
-            memcpy(t, r->data, l);
-            t[l] = 0;
-            break;
-        }
+        case AVAHI_DNS_TYPE_PTR:
+        case AVAHI_DNS_TYPE_CNAME :
 
-        case FLX_DNS_TYPE_TXT: {
-
-            if (r->size == 0)
-                t[0] = 0;
-            else {
-                guchar l = ((guchar*) r->data)[0];
+            t = r->data.ptr.name;
+            break;
 
-                if ((size_t) l+1 <= r->size) {
-                    memcpy(t, r->data+1, ((guchar*) r->data)[0]);
-                    t[((guchar*) r->data)[0]] = 0;
-                }
-            }
+        case AVAHI_DNS_TYPE_TXT:
+            t = d = avahi_string_list_to_string(r->data.txt.string_list);
             break;
-        }
 
-        case FLX_DNS_TYPE_HINFO: {
-            gchar *s2;
-            gchar hi1[256], hi2[256];
-            guchar len;
-
-            if ((size_t) (len = ((guchar*) r->data)[0]) + 2 <= r->size) {
-                guchar len2;
-                memcpy(hi1, (gchar*) r->data +1, len);
-                hi1[len] = 0;
-
-                if ((size_t) (len2 = ((guchar*) r->data)[len+1]) + len + 2 <= r->size) {
-                    memcpy(hi2, (gchar*) r->data+len+2, len2);
-                    hi2[len2] = 0;
-                    snprintf(t, sizeof(t), "'%s' '%s'", hi1, hi2);
-                }
-                
-            }
+        case AVAHI_DNS_TYPE_HINFO:
 
+            snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
             break;
-        }
 
-        case FLX_DNS_TYPE_SRV: {
-            char k[257];
-            size_t l;
+        case AVAHI_DNS_TYPE_SRV:
+
+            snprintf(t = buf, sizeof(buf), "%u %u %u %s",
+                     r->data.srv.priority,
+                     r->data.srv.weight,
+                     r->data.srv.port,
+                     r->data.srv.name);
 
-            l = r->size-6;
-            if (l > sizeof(k)-1)
-                l = sizeof(k)-1;
-            
-            memcpy(k, r->data+6, l);
-            k[l] = 0;
-            
-            snprintf(t, sizeof(t), "%u %u %u %s",
-                     ntohs(((guint16*) r->data)[0]),
-                     ntohs(((guint16*) r->data)[1]),
-                     ntohs(((guint16*) r->data)[2]),
-                     k);
             break;
-        }
     }
 
-    p = flx_key_to_string(r->key);
-    s = g_strdup_printf("%s %s ; ttl=%u", p, t, r->ttl);
+    p = avahi_key_to_string(r->key);
+    s = g_strdup_printf("%s %s ; ttl=%u", p, t ? t : "<unparsable>", r->ttl);
     g_free(p);
+    g_free(d);
     
     return s;
 }
 
-gboolean flx_key_equal(const flxKey *a, const flxKey *b) {
+gboolean avahi_key_equal(const AvahiKey *a, const AvahiKey *b) {
     g_assert(a);
     g_assert(b);
 
+    if (a == b)
+        return TRUE;
+    
 /*     g_message("equal: %p %p", a, b); */
     
-    return strcmp(a->name, b->name) == 0 &&
+    return avahi_domain_equal(a->name, b->name) &&
         a->type == b->type &&
         a->class == b->class;
 }
 
-gboolean flx_key_pattern_match(const flxKey *pattern, const flxKey *k) {
+gboolean avahi_key_pattern_match(const AvahiKey *pattern, const AvahiKey *k) {
     g_assert(pattern);
     g_assert(k);
 
 /*     g_message("equal: %p %p", a, b); */
 
-    g_assert(!flx_key_is_pattern(k));
+    g_assert(!avahi_key_is_pattern(k));
+
+    if (pattern == k)
+        return TRUE;
     
-    return strcmp(pattern->name, k->name) == 0 &&
-        (pattern->type == k->type || pattern->type == FLX_DNS_TYPE_ANY) &&
+    return avahi_domain_equal(pattern->name, k->name) &&
+        (pattern->type == k->type || pattern->type == AVAHI_DNS_TYPE_ANY) &&
         pattern->class == k->class;
 }
 
-gboolean flx_key_is_pattern(const flxKey *k) {
+gboolean avahi_key_is_pattern(const AvahiKey *k) {
     g_assert(k);
 
-    return k->type == FLX_DNS_TYPE_ANY;
+    return k->type == AVAHI_DNS_TYPE_ANY;
+}
+
+
+guint avahi_key_hash(const AvahiKey *k) {
+    g_assert(k);
+
+    return avahi_domain_hash(k->name) + k->type + k->class;
+}
+
+static gboolean rdata_equal(const AvahiRecord *a, const AvahiRecord *b) {
+    g_assert(a);
+    g_assert(b);
+    g_assert(a->key->type == b->key->type);
+
+/*     t = avahi_record_to_string(a); */
+/*     g_message("comparing %s", t); */
+/*     g_free(t); */
+
+/*     t = avahi_record_to_string(b); */
+/*     g_message("and %s", t); */
+/*     g_free(t); */
+
+    
+    switch (a->key->type) {
+        case AVAHI_DNS_TYPE_SRV:
+            return
+                a->data.srv.priority == b->data.srv.priority &&
+                a->data.srv.weight == b->data.srv.weight &&
+                a->data.srv.port == b->data.srv.port &&
+                avahi_domain_equal(a->data.srv.name, b->data.srv.name);
+
+        case AVAHI_DNS_TYPE_PTR:
+        case AVAHI_DNS_TYPE_CNAME:
+            return avahi_domain_equal(a->data.ptr.name, b->data.ptr.name);
+
+        case AVAHI_DNS_TYPE_HINFO:
+            return
+                !strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
+                !strcmp(a->data.hinfo.os, b->data.hinfo.os);
+
+        case AVAHI_DNS_TYPE_TXT:
+            return avahi_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
+
+        case AVAHI_DNS_TYPE_A:
+            return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address)) == 0;
+
+        case AVAHI_DNS_TYPE_AAAA:
+            return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address)) == 0;
+
+        default:
+            return a->data.generic.size == b->data.generic.size &&
+                (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
+    }
+    
+}
+
+gboolean avahi_record_equal_no_ttl(const AvahiRecord *a, const AvahiRecord *b) {
+    g_assert(a);
+    g_assert(b);
+
+    if (a == b)
+        return TRUE;
+
+    return
+        avahi_key_equal(a->key, b->key) &&
+        rdata_equal(a, b);
+}
+
+
+AvahiRecord *avahi_record_copy(AvahiRecord *r) {
+    AvahiRecord *copy;
+
+    copy = g_new(AvahiRecord, 1);
+    copy->ref = 1;
+    copy->key = avahi_key_ref(r->key);
+    copy->ttl = r->ttl;
+
+    switch (r->key->type) {
+        case AVAHI_DNS_TYPE_PTR:
+        case AVAHI_DNS_TYPE_CNAME:
+            copy->data.ptr.name = g_strdup(r->data.ptr.name);
+            break;
+
+        case AVAHI_DNS_TYPE_SRV:
+            copy->data.srv.priority = r->data.srv.priority;
+            copy->data.srv.weight = r->data.srv.weight;
+            copy->data.srv.port = r->data.srv.port;
+            copy->data.srv.name = g_strdup(r->data.srv.name);
+            break;
+
+        case AVAHI_DNS_TYPE_HINFO:
+            copy->data.hinfo.os = g_strdup(r->data.hinfo.os);
+            copy->data.hinfo.cpu = g_strdup(r->data.hinfo.cpu);
+            break;
+
+        case AVAHI_DNS_TYPE_TXT:
+            copy->data.txt.string_list = avahi_string_list_copy(r->data.txt.string_list);
+            break;
+
+        case AVAHI_DNS_TYPE_A:
+            copy->data.a.address = r->data.a.address;
+            break;
+
+        case AVAHI_DNS_TYPE_AAAA:
+            copy->data.aaaa.address = r->data.aaaa.address;
+            break;
+
+        default:
+            copy->data.generic.data = g_memdup(r->data.generic.data, r->data.generic.size);
+            copy->data.generic.size = r->data.generic.size;
+            break;
+                
+    }
+
+    return copy;
 }
 
 
-guint flx_key_hash(const flxKey *k) {
+guint avahi_key_get_estimate_size(AvahiKey *k) {
     g_assert(k);
 
-    return g_str_hash(k->name) + k->type + k->class;
+    return strlen(k->name)+1+4;
+}
+
+guint avahi_record_get_estimate_size(AvahiRecord *r) {
+    guint n;
+    g_assert(r);
+
+    n = avahi_key_get_estimate_size(r->key) + 4 + 2;
+
+    switch (r->key->type) {
+        case AVAHI_DNS_TYPE_PTR:
+        case AVAHI_DNS_TYPE_CNAME:
+            n += strlen(r->data.ptr.name) + 1;
+            break;
+
+        case AVAHI_DNS_TYPE_SRV:
+            n += 6 + strlen(r->data.srv.name) + 1;
+            break;
+
+        case AVAHI_DNS_TYPE_HINFO:
+            n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
+            break;
+
+        case AVAHI_DNS_TYPE_TXT:
+            n += avahi_string_list_serialize(r->data.txt.string_list, NULL, 0);
+            break;
+
+        case AVAHI_DNS_TYPE_A:
+            n += sizeof(AvahiIPv4Address);
+            break;
+
+        case AVAHI_DNS_TYPE_AAAA:
+            n += sizeof(AvahiIPv6Address);
+            break;
+
+        default:
+            n += r->data.generic.size;
+    }
+
+    return n;
+}
+
+static gint lexicographical_memcmp(gconstpointer a, size_t al, gconstpointer b, size_t bl) {
+    size_t c;
+    gint ret;
+    
+    g_assert(a);
+    g_assert(b);
+
+    c = al < bl ? al : bl;
+    if ((ret = memcmp(a, b, c)) != 0)
+        return ret;
+
+    if (al == bl)
+        return 0;
+    else
+        return al == c ? 1 : -1;
+}
+
+static gint uint16_cmp(guint16 a, guint16 b) {
+    return a == b ? 0 : (a < b ? a : b);
+}
+
+static gint lexicographical_domain_cmp(const gchar *a, const gchar *b) {
+    g_assert(a);
+    g_assert(b);
+    
+
+    for (;;) {
+        gchar t1[64];
+        gchar t2[64];
+        size_t al, bl;
+        gint r;
+
+        if (!a && !b)
+            return 0;
+
+        if (a && !b)
+            return 1;
+
+        if (b && !a)
+            return -1;
+        
+        avahi_unescape_label(t1, sizeof(t1), &a);
+        avahi_unescape_label(t2, sizeof(t2), &b);
+
+        al = strlen(t1);
+        bl = strlen(t2);
+        
+        if (al != bl) 
+            return al < bl ? -1 : 1;
+
+        if ((r =  strcmp(t1, t2)) != 0)
+            return r;
+    }
 }
 
-gboolean flx_record_equal_no_ttl(const flxRecord *a, const flxRecord *b) {
+gint avahi_record_lexicographical_compare(AvahiRecord *a, AvahiRecord *b) {
     g_assert(a);
     g_assert(b);
 
-    return flx_key_equal(a->key, b->key) &&
-/*        a->ttl == b->ttl && */
-        a->size == b->size &&
-        (a->size == 0 || memcmp(a->data, b->data, a->size) == 0);
+    if (a == b)
+        return 0;
+    
+/*     gchar *t; */
+
+/*     g_message("comparing [%s]", t = avahi_record_to_string(a)); */
+/*     g_free(t); */
+
+/*     g_message("and [%s]", t = avahi_record_to_string(b)); */
+/*     g_free(t); */
+
+    if (a->key->class < b->key->class)
+        return -1;
+    else if (a->key->class > b->key->class)
+        return 1;
+
+    if (a->key->type < b->key->type)
+        return -1;
+    else if (a->key->type > b->key->type)
+        return 1;
+
+    switch (a->key->type) {
+
+        case AVAHI_DNS_TYPE_PTR:
+        case AVAHI_DNS_TYPE_CNAME:
+            return lexicographical_domain_cmp(a->data.ptr.name, b->data.ptr.name);
+
+        case AVAHI_DNS_TYPE_SRV: {
+            gint r;
+            if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) == 0 &&
+                (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) == 0 &&
+                (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) == 0)
+                r = lexicographical_domain_cmp(a->data.srv.name, b->data.srv.name);
+            
+            return r;
+        }
+
+        case AVAHI_DNS_TYPE_HINFO: {
+            size_t al = strlen(a->data.hinfo.cpu), bl = strlen(b->data.hinfo.cpu);
+            gint r;
+
+            if (al != bl)
+                return al < bl ? -1 : 1;
+
+            if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) != 0)
+                return r;
+
+            al = strlen(a->data.hinfo.os), bl = strlen(b->data.hinfo.os);
+
+            if (al != bl)
+                return al < bl ? -1 : 1;
+
+            if ((r = strcmp(a->data.hinfo.os, b->data.hinfo.os)) != 0)
+                return r;
+
+            return 0;
+
+        }
+
+        case AVAHI_DNS_TYPE_TXT: {
+
+            guint8 *ma, *mb;
+            guint asize, bsize;
+            gint r;
+
+            ma = g_new(guint8, asize = avahi_string_list_serialize(a->data.txt.string_list, NULL, 0));
+            mb = g_new(guint8, bsize = avahi_string_list_serialize(b->data.txt.string_list, NULL, 0));
+            avahi_string_list_serialize(a->data.txt.string_list, ma, asize);
+            avahi_string_list_serialize(a->data.txt.string_list, mb, bsize);
+
+            r = lexicographical_memcmp(ma, asize, mb, bsize);
+            g_free(ma);
+            g_free(mb);
+
+            return r;
+        }
+        
+        case AVAHI_DNS_TYPE_A:
+            return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address));
+
+        case AVAHI_DNS_TYPE_AAAA:
+            return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address));
+
+        default:
+            return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
+                                          b->data.generic.data, b->data.generic.size);
+    }
+    
 }