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 <avahi-common/domain.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->clazz = 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, guint32 ttl) {
  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 = ttl != (guint32) -1 ? ttl : AVAHI_DEFAULT_TTL;
  85
  86     return r;
  87 }
  88
  89 AvahiRecord *avahi_record_new_full(const gchar *name, guint16 class, guint16 type, guint32 ttl) {
  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, ttl);
  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     switch (class) {
 153         case AVAHI_DNS_CLASS_IN:
 154             return "IN";
 155         case AVAHI_DNS_CLASS_ANY:
 156             return "ANY";
 157         default:
 158             return NULL;
 159     }
 160 }
 161
 162 const gchar *avahi_dns_type_to_string(guint16 type) {
 163     switch (type) {
 164         case AVAHI_DNS_TYPE_CNAME:
 165             return "CNAME";
 166         case AVAHI_DNS_TYPE_A:
 167             return "A";
 168         case AVAHI_DNS_TYPE_AAAA:
 169             return "AAAA";
 170         case AVAHI_DNS_TYPE_PTR:
 171             return "PTR";
 172         case AVAHI_DNS_TYPE_HINFO:
 173             return "HINFO";
 174         case AVAHI_DNS_TYPE_TXT:
 175             return "TXT";
 176         case AVAHI_DNS_TYPE_SRV:
 177             return "SRV";
 178         case AVAHI_DNS_TYPE_ANY:
 179             return "ANY";
 180         default:
 181             return NULL;
 182     }
 183 }
 184
 185 gchar *avahi_key_to_string(const AvahiKey *k) {
 186     g_assert(k);
 187     g_assert(k->ref >= 1);
 188
 189     return g_strdup_printf("%s\t%s\t%s",
 190                            k->name,
 191                            avahi_dns_class_to_string(k->clazz),
 192                            avahi_dns_type_to_string(k->type));
 193 }
 194
 195 gchar *avahi_record_to_string(const AvahiRecord *r) {
 196     gchar *p, *s;
 197     char buf[257], *t = NULL, *d = NULL;
 198
 199     g_assert(r);
 200     g_assert(r->ref >= 1);
 201
 202     switch (r->key->type) {
 203         case AVAHI_DNS_TYPE_A:
 204             inet_ntop(AF_INET, &r->data.a.address.address, t = buf, sizeof(buf));
 205             break;
 206
 207         case AVAHI_DNS_TYPE_AAAA:
 208             inet_ntop(AF_INET6, &r->data.aaaa.address.address, t = buf, sizeof(buf));
 209             break;
 210
 211         case AVAHI_DNS_TYPE_PTR:
 212         case AVAHI_DNS_TYPE_CNAME :
 213
 214             t = r->data.ptr.name;
 215             break;
 216
 217         case AVAHI_DNS_TYPE_TXT:
 218             t = d = avahi_string_list_to_string(r->data.txt.string_list);
 219             break;
 220
 221         case AVAHI_DNS_TYPE_HINFO:
 222
 223             snprintf(t = buf, sizeof(buf), "\"%s\" \"%s\"", r->data.hinfo.cpu, r->data.hinfo.os);
 224             break;
 225
 226         case AVAHI_DNS_TYPE_SRV:
 227
 228             snprintf(t = buf, sizeof(buf), "%u %u %u %s",
 229                      r->data.srv.priority,
 230                      r->data.srv.weight,
 231                      r->data.srv.port,
 232                      r->data.srv.name);
 233
 234             break;
 235     }
 236
 237     p = avahi_key_to_string(r->key);
 238     s = g_strdup_printf("%s %s ; ttl=%u", p, t ? t : "<unparsable>", r->ttl);
 239     g_free(p);
 240     g_free(d);
 241
 242     return s;
 243 }
 244
 245 gboolean avahi_key_equal(const AvahiKey *a, const AvahiKey *b) {
 246     g_assert(a);
 247     g_assert(b);
 248
 249     if (a == b)
 250         return TRUE;
 251
 252 /*     g_message("equal: %p %p", a, b); */
 253
 254     return avahi_domain_equal(a->name, b->name) &&
 255         a->type == b->type &&
 256         a->clazz == b->clazz;
 257 }
 258
 259 gboolean avahi_key_pattern_match(const AvahiKey *pattern, const AvahiKey *k) {
 260     g_assert(pattern);
 261     g_assert(k);
 262
 263 /*     g_message("equal: %p %p", a, b); */
 264
 265     g_assert(!avahi_key_is_pattern(k));
 266
 267     if (pattern == k)
 268         return TRUE;
 269
 270     return avahi_domain_equal(pattern->name, k->name) &&
 271         (pattern->type == k->type || pattern->type == AVAHI_DNS_TYPE_ANY) &&
 272         (pattern->clazz == k->clazz || pattern->clazz == AVAHI_DNS_CLASS_ANY);
 273 }
 274
 275 gboolean avahi_key_is_pattern(const AvahiKey *k) {
 276     g_assert(k);
 277
 278     return
 279         k->type == AVAHI_DNS_TYPE_ANY ||
 280         k->clazz == AVAHI_DNS_CLASS_ANY;
 281 }
 282
 283 guint avahi_key_hash(const AvahiKey *k) {
 284     g_assert(k);
 285
 286     return
 287         avahi_domain_hash(k->name) +
 288         k->type +
 289         k->clazz;
 290 }
 291
 292 static gboolean rdata_equal(const AvahiRecord *a, const AvahiRecord *b) {
 293     g_assert(a);
 294     g_assert(b);
 295     g_assert(a->key->type == b->key->type);
 296
 297 /*     t = avahi_record_to_string(a); */
 298 /*     g_message("comparing %s", t); */
 299 /*     g_free(t); */
 300
 301 /*     t = avahi_record_to_string(b); */
 302 /*     g_message("and %s", t); */
 303 /*     g_free(t); */
 304
 305
 306     switch (a->key->type) {
 307         case AVAHI_DNS_TYPE_SRV:
 308             return
 309                 a->data.srv.priority == b->data.srv.priority &&
 310                 a->data.srv.weight == b->data.srv.weight &&
 311                 a->data.srv.port == b->data.srv.port &&
 312                 avahi_domain_equal(a->data.srv.name, b->data.srv.name);
 313
 314         case AVAHI_DNS_TYPE_PTR:
 315         case AVAHI_DNS_TYPE_CNAME:
 316             return avahi_domain_equal(a->data.ptr.name, b->data.ptr.name);
 317
 318         case AVAHI_DNS_TYPE_HINFO:
 319             return
 320                 !strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu) &&
 321                 !strcmp(a->data.hinfo.os, b->data.hinfo.os);
 322
 323         case AVAHI_DNS_TYPE_TXT:
 324             return avahi_string_list_equal(a->data.txt.string_list, b->data.txt.string_list);
 325
 326         case AVAHI_DNS_TYPE_A:
 327             return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address)) == 0;
 328
 329         case AVAHI_DNS_TYPE_AAAA:
 330             return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address)) == 0;
 331
 332         default:
 333             return a->data.generic.size == b->data.generic.size &&
 334                 (a->data.generic.size == 0 || memcmp(a->data.generic.data, b->data.generic.data, a->data.generic.size) == 0);
 335     }
 336
 337 }
 338
 339 gboolean avahi_record_equal_no_ttl(const AvahiRecord *a, const AvahiRecord *b) {
 340     g_assert(a);
 341     g_assert(b);
 342
 343     if (a == b)
 344         return TRUE;
 345
 346     return
 347         avahi_key_equal(a->key, b->key) &&
 348         rdata_equal(a, b);
 349 }
 350
 351
 352 AvahiRecord *avahi_record_copy(AvahiRecord *r) {
 353     AvahiRecord *copy;
 354
 355     copy = g_new(AvahiRecord, 1);
 356     copy->ref = 1;
 357     copy->key = avahi_key_ref(r->key);
 358     copy->ttl = r->ttl;
 359
 360     switch (r->key->type) {
 361         case AVAHI_DNS_TYPE_PTR:
 362         case AVAHI_DNS_TYPE_CNAME:
 363             copy->data.ptr.name = g_strdup(r->data.ptr.name);
 364             break;
 365
 366         case AVAHI_DNS_TYPE_SRV:
 367             copy->data.srv.priority = r->data.srv.priority;
 368             copy->data.srv.weight = r->data.srv.weight;
 369             copy->data.srv.port = r->data.srv.port;
 370             copy->data.srv.name = g_strdup(r->data.srv.name);
 371             break;
 372
 373         case AVAHI_DNS_TYPE_HINFO:
 374             copy->data.hinfo.os = g_strdup(r->data.hinfo.os);
 375             copy->data.hinfo.cpu = g_strdup(r->data.hinfo.cpu);
 376             break;
 377
 378         case AVAHI_DNS_TYPE_TXT:
 379             copy->data.txt.string_list = avahi_string_list_copy(r->data.txt.string_list);
 380             break;
 381
 382         case AVAHI_DNS_TYPE_A:
 383             copy->data.a.address = r->data.a.address;
 384             break;
 385
 386         case AVAHI_DNS_TYPE_AAAA:
 387             copy->data.aaaa.address = r->data.aaaa.address;
 388             break;
 389
 390         default:
 391             copy->data.generic.data = g_memdup(r->data.generic.data, r->data.generic.size);
 392             copy->data.generic.size = r->data.generic.size;
 393             break;
 394
 395     }
 396
 397     return copy;
 398 }
 399
 400
 401 guint avahi_key_get_estimate_size(AvahiKey *k) {
 402     g_assert(k);
 403
 404     return strlen(k->name)+1+4;
 405 }
 406
 407 guint avahi_record_get_estimate_size(AvahiRecord *r) {
 408     guint n;
 409     g_assert(r);
 410
 411     n = avahi_key_get_estimate_size(r->key) + 4 + 2;
 412
 413     switch (r->key->type) {
 414         case AVAHI_DNS_TYPE_PTR:
 415         case AVAHI_DNS_TYPE_CNAME:
 416             n += strlen(r->data.ptr.name) + 1;
 417             break;
 418
 419         case AVAHI_DNS_TYPE_SRV:
 420             n += 6 + strlen(r->data.srv.name) + 1;
 421             break;
 422
 423         case AVAHI_DNS_TYPE_HINFO:
 424             n += strlen(r->data.hinfo.os) + 1 + strlen(r->data.hinfo.cpu) + 1;
 425             break;
 426
 427         case AVAHI_DNS_TYPE_TXT:
 428             n += avahi_string_list_serialize(r->data.txt.string_list, NULL, 0);
 429             break;
 430
 431         case AVAHI_DNS_TYPE_A:
 432             n += sizeof(AvahiIPv4Address);
 433             break;
 434
 435         case AVAHI_DNS_TYPE_AAAA:
 436             n += sizeof(AvahiIPv6Address);
 437             break;
 438
 439         default:
 440             n += r->data.generic.size;
 441     }
 442
 443     return n;
 444 }
 445
 446 static gint lexicographical_memcmp(gconstpointer a, size_t al, gconstpointer b, size_t bl) {
 447     size_t c;
 448     gint ret;
 449
 450     g_assert(a);
 451     g_assert(b);
 452
 453     c = al < bl ? al : bl;
 454     if ((ret = memcmp(a, b, c)))
 455         return ret;
 456
 457     if (al == bl)
 458         return 0;
 459     else
 460         return al == c ? 1 : -1;
 461 }
 462
 463 static gint uint16_cmp(guint16 a, guint16 b) {
 464     return a == b ? 0 : (a < b ? -1 : 1);
 465 }
 466
 467 gint avahi_record_lexicographical_compare(AvahiRecord *a, AvahiRecord *b) {
 468     gint r;
 469 /*      gchar *t1, *t2; */
 470
 471     g_assert(a);
 472     g_assert(b);
 473
 474 /*     t1 = avahi_record_to_string(a); */
 475 /*     t2 = avahi_record_to_string(b); */
 476 /*     g_message("lexicocmp: %s %s", t1, t2); */
 477 /*     g_free(t1); */
 478 /*     g_free(t2); */
 479
 480     if (a == b)
 481         return 0;
 482
 483     if ((r = uint16_cmp(a->key->clazz, b->key->clazz)) ||
 484         (r = uint16_cmp(a->key->type, b->key->type)))
 485         return r;
 486
 487     switch (a->key->type) {
 488
 489         case AVAHI_DNS_TYPE_PTR:
 490         case AVAHI_DNS_TYPE_CNAME:
 491             return avahi_binary_domain_cmp(a->data.ptr.name, b->data.ptr.name);
 492
 493         case AVAHI_DNS_TYPE_SRV: {
 494             if ((r = uint16_cmp(a->data.srv.priority, b->data.srv.priority)) == 0 &&
 495                 (r = uint16_cmp(a->data.srv.weight, b->data.srv.weight)) == 0 &&
 496                 (r = uint16_cmp(a->data.srv.port, b->data.srv.port)) == 0)
 497                 r = avahi_binary_domain_cmp(a->data.srv.name, b->data.srv.name);
 498
 499             return r;
 500         }
 501
 502         case AVAHI_DNS_TYPE_HINFO: {
 503
 504             if ((r = strcmp(a->data.hinfo.cpu, b->data.hinfo.cpu)) ||
 505                 (r = strcmp(a->data.hinfo.os, b->data.hinfo.os)))
 506                 return r;
 507
 508             return 0;
 509
 510         }
 511
 512         case AVAHI_DNS_TYPE_TXT: {
 513
 514             guint8 *ma, *mb;
 515             guint asize, bsize;
 516
 517             ma = g_new(guint8, asize = avahi_string_list_serialize(a->data.txt.string_list, NULL, 0));
 518             mb = g_new(guint8, bsize = avahi_string_list_serialize(b->data.txt.string_list, NULL, 0));
 519             avahi_string_list_serialize(a->data.txt.string_list, ma, asize);
 520             avahi_string_list_serialize(b->data.txt.string_list, mb, bsize);
 521
 522             if (asize && bsize)
 523                 r = lexicographical_memcmp(ma, asize, mb, bsize);
 524             else if (asize && !bsize)
 525                 r = 1;
 526             else if (!asize && bsize)
 527                 r = -1;
 528             else
 529                 r = 0;
 530
 531             g_free(ma);
 532             g_free(mb);
 533
 534             return r;
 535         }
 536
 537         case AVAHI_DNS_TYPE_A:
 538             return memcmp(&a->data.a.address, &b->data.a.address, sizeof(AvahiIPv4Address));
 539
 540         case AVAHI_DNS_TYPE_AAAA:
 541             return memcmp(&a->data.aaaa.address, &b->data.aaaa.address, sizeof(AvahiIPv6Address));
 542
 543         default:
 544             return lexicographical_memcmp(a->data.generic.data, a->data.generic.size,
 545                                           b->data.generic.data, b->data.generic.size);
 546     }
 547
 548 }
 549
 550 gboolean avahi_record_is_goodbye(AvahiRecord *r) {
 551     g_assert(r);
 552
 553     return r->ttl == 0;
 554 }
 555
 556 gboolean avahi_key_valid(AvahiKey *k) {
 557     g_assert(k);
 558
 559     if (!avahi_valid_domain_name(k->name))
 560         return FALSE;
 561
 562     return TRUE;
 563 }
 564
 565 gboolean avahi_record_valid(AvahiRecord *r) {
 566     g_assert(r);
 567
 568     if (!avahi_key_valid(r->key))
 569         return FALSE;
 570
 571     switch (r->key->type) {
 572
 573         case AVAHI_DNS_TYPE_PTR:
 574         case AVAHI_DNS_TYPE_CNAME:
 575             return avahi_valid_domain_name(r->data.ptr.name);
 576
 577         case AVAHI_DNS_TYPE_SRV:
 578             return avahi_valid_domain_name(r->data.srv.name);
 579
 580         case AVAHI_DNS_TYPE_HINFO:
 581             return
 582                 strlen(r->data.hinfo.os) <= 255 &&
 583                 strlen(r->data.hinfo.cpu) <= 255;
 584
 585
 586         case AVAHI_DNS_TYPE_TXT: {
 587
 588             AvahiStringList *strlst;
 589
 590             for (strlst = r->data.txt.string_list; strlst; strlst = strlst->next)
 591                 if (strlst->size > 255)
 592                     return FALSE;
 593
 594             return TRUE;
 595         }
 596     }
 597
 598
 599     return TRUE;
 600 }