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