git.meshlink.io Git - catta/blob - avahi-core/dns.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 <stdlib.h>
  27 #include <string.h>
  28 #include <stdio.h>
  29 #include <assert.h>
  30
  31 #include <netinet/in.h>
  32
  33 #include <avahi-common/defs.h>
  34 #include <avahi-common/domain.h>
  35 #include <avahi-common/malloc.h>
  36
  37 #include "dns.h"
  38 #include "log.h"
  39
  40 AvahiDnsPacket* avahi_dns_packet_new(unsigned mtu) {
  41     AvahiDnsPacket *p;
  42     size_t max_size;
  43
  44     if (mtu <= 0)
  45         max_size = AVAHI_DNS_PACKET_SIZE_MAX;
  46     else if (mtu >= AVAHI_DNS_PACKET_EXTRA_SIZE)
  47         max_size = mtu - AVAHI_DNS_PACKET_EXTRA_SIZE;
  48     else
  49         max_size = 0;
  50
  51     if (max_size < AVAHI_DNS_PACKET_HEADER_SIZE)
  52         max_size = AVAHI_DNS_PACKET_HEADER_SIZE;
  53
  54     if (!(p = avahi_malloc(sizeof(AvahiDnsPacket) + max_size)))
  55         return p;
  56
  57     p->size = p->rindex = AVAHI_DNS_PACKET_HEADER_SIZE;
  58     p->max_size = max_size;
  59     p->name_table = NULL;
  60     p->data = NULL;
  61
  62     memset(AVAHI_DNS_PACKET_DATA(p), 0, p->size);
  63     return p;
  64 }
  65
  66 AvahiDnsPacket* avahi_dns_packet_new_query(unsigned mtu) {
  67     AvahiDnsPacket *p;
  68
  69     if (!(p = avahi_dns_packet_new(mtu)))
  70         return NULL;
  71
  72     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_FLAGS, AVAHI_DNS_FLAGS(0, 0, 0, 0, 0, 0, 0, 0, 0, 0));
  73     return p;
  74 }
  75
  76 AvahiDnsPacket* avahi_dns_packet_new_response(unsigned mtu, int aa) {
  77     AvahiDnsPacket *p;
  78
  79     if (!(p = avahi_dns_packet_new(mtu)))
  80         return NULL;
  81
  82     avahi_dns_packet_set_field(p, AVAHI_DNS_FIELD_FLAGS, AVAHI_DNS_FLAGS(1, 0, aa, 0, 0, 0, 0, 0, 0, 0));
  83     return p;
  84 }
  85
  86 AvahiDnsPacket* avahi_dns_packet_new_reply(AvahiDnsPacket* p, unsigned mtu, int copy_queries, int aa) {
  87     AvahiDnsPacket *r;
  88     assert(p);
  89
  90     if (!(r = avahi_dns_packet_new_response(mtu, aa)))
  91         return NULL;
  92
  93     if (copy_queries) {
  94         unsigned saved_rindex;
  95         uint32_t n;
  96
  97         saved_rindex = p->rindex;
  98         p->rindex = AVAHI_DNS_PACKET_HEADER_SIZE;
  99
 100         for (n = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_QDCOUNT); n > 0; n--) {
 101             AvahiKey *k;
 102             int unicast_response;
 103
 104             if ((k = avahi_dns_packet_consume_key(p, &unicast_response))) {
 105                 avahi_dns_packet_append_key(r, k, unicast_response);
 106                 avahi_key_unref(k);
 107             }
 108         }
 109
 110         p->rindex = saved_rindex;
 111
 112         avahi_dns_packet_set_field(r, AVAHI_DNS_FIELD_QDCOUNT, avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_QDCOUNT));
 113     }
 114
 115     avahi_dns_packet_set_field(r, AVAHI_DNS_FIELD_ID, avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_ID));
 116
 117     avahi_dns_packet_set_field(r, AVAHI_DNS_FIELD_FLAGS,
 118                                (avahi_dns_packet_get_field(r, AVAHI_DNS_FIELD_FLAGS) & ~AVAHI_DNS_FLAG_OPCODE) |
 119                                (avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS) & AVAHI_DNS_FLAG_OPCODE));
 120
 121     return r;
 122 }
 123
 124
 125 void avahi_dns_packet_free(AvahiDnsPacket *p) {
 126     assert(p);
 127
 128     if (p->name_table)
 129         avahi_hashmap_free(p->name_table);
 130
 131     avahi_free(p);
 132 }
 133
 134 void avahi_dns_packet_set_field(AvahiDnsPacket *p, unsigned idx, uint16_t v) {
 135     assert(p);
 136     assert(idx < AVAHI_DNS_PACKET_HEADER_SIZE);
 137
 138     ((uint16_t*) AVAHI_DNS_PACKET_DATA(p))[idx] = htons(v);
 139 }
 140
 141 uint16_t avahi_dns_packet_get_field(AvahiDnsPacket *p, unsigned idx) {
 142     assert(p);
 143     assert(idx < AVAHI_DNS_PACKET_HEADER_SIZE);
 144
 145     return ntohs(((uint16_t*) AVAHI_DNS_PACKET_DATA(p))[idx]);
 146 }
 147
 148 void avahi_dns_packet_inc_field(AvahiDnsPacket *p, unsigned idx) {
 149     assert(p);
 150     assert(idx < AVAHI_DNS_PACKET_HEADER_SIZE);
 151
 152     avahi_dns_packet_set_field(p, idx, avahi_dns_packet_get_field(p, idx) + 1);
 153 }
 154
 155 uint8_t* avahi_dns_packet_append_name(AvahiDnsPacket *p, const char *name) {
 156     uint8_t *d, *saved_ptr = NULL;
 157     size_t saved_size;
 158
 159     assert(p);
 160     assert(name);
 161
 162     saved_size = p->size;
 163     saved_ptr = avahi_dns_packet_extend(p, 0);
 164
 165     while (*name) {
 166         uint8_t* prev;
 167         const char *pname;
 168         char label[64], *u;
 169
 170         /* Check whether we can compress this name. */
 171
 172         if (p->name_table && (prev = avahi_hashmap_lookup(p->name_table, name))) {
 173             unsigned idx;
 174
 175             assert(prev >= AVAHI_DNS_PACKET_DATA(p));
 176             idx = (unsigned) (prev - AVAHI_DNS_PACKET_DATA(p));
 177
 178             assert(idx < p->size);
 179
 180             if (idx < 0x4000) {
 181                 uint8_t *t;
 182                 if (!(t = (uint8_t*) avahi_dns_packet_extend(p, sizeof(uint16_t))))
 183                     return NULL;
 184
 185                 t[0] = (uint8_t) ((0xC000 | idx) >> 8);
 186                 t[1] = (uint8_t) idx;
 187                 return saved_ptr;
 188             }
 189         }
 190
 191         pname = name;
 192
 193         if (!(avahi_unescape_label(&name, label, sizeof(label))))
 194             goto fail;
 195
 196         if (!(d = avahi_dns_packet_append_string(p, label)))
 197             goto fail;
 198
 199         if (!p->name_table)
 200             /* This works only for normalized domain names */
 201             p->name_table = avahi_hashmap_new(avahi_string_hash, avahi_string_equal, avahi_free, NULL);
 202
 203         if (!(u = avahi_strdup(pname)))
 204             avahi_log_error("avahi_strdup() failed.");
 205         else
 206             avahi_hashmap_insert(p->name_table, u, d);
 207     }
 208
 209     if (!(d = avahi_dns_packet_extend(p, 1)))
 210         goto fail;
 211
 212     *d = 0;
 213
 214     return saved_ptr;
 215
 216 fail:
 217     p->size = saved_size;
 218     return NULL;
 219 }
 220
 221 uint8_t* avahi_dns_packet_append_uint16(AvahiDnsPacket *p, uint16_t v) {
 222     uint8_t *d;
 223     assert(p);
 224
 225     if (!(d = avahi_dns_packet_extend(p, sizeof(uint16_t))))
 226         return NULL;
 227
 228     d[0] = (uint8_t) (v >> 8);
 229     d[1] = (uint8_t) v;
 230     return d;
 231 }
 232
 233 uint8_t *avahi_dns_packet_append_uint32(AvahiDnsPacket *p, uint32_t v) {
 234     uint8_t *d;
 235     assert(p);
 236
 237     if (!(d = avahi_dns_packet_extend(p, sizeof(uint32_t))))
 238         return NULL;
 239
 240     d[0] = (uint8_t) (v >> 24);
 241     d[1] = (uint8_t) (v >> 16);
 242     d[2] = (uint8_t) (v >> 8);
 243     d[3] = (uint8_t) v;
 244
 245     return d;
 246 }
 247
 248 uint8_t *avahi_dns_packet_append_bytes(AvahiDnsPacket  *p, const void *b, size_t l) {
 249     uint8_t* d;
 250
 251     assert(p);
 252     assert(b);
 253     assert(l);
 254
 255     if (!(d = avahi_dns_packet_extend(p, l)))
 256         return NULL;
 257
 258     memcpy(d, b, l);
 259     return d;
 260 }
 261
 262 uint8_t* avahi_dns_packet_append_string(AvahiDnsPacket *p, const char *s) {
 263     uint8_t* d;
 264     size_t k;
 265
 266     assert(p);
 267     assert(s);
 268
 269     if ((k = strlen(s)) >= 255)
 270         k = 255;
 271
 272     if (!(d = avahi_dns_packet_extend(p, k+1)))
 273         return NULL;
 274
 275     *d = (uint8_t) k;
 276     memcpy(d+1, s, k);
 277
 278     return d;
 279 }
 280
 281 uint8_t *avahi_dns_packet_extend(AvahiDnsPacket *p, size_t l) {
 282     uint8_t *d;
 283
 284     assert(p);
 285
 286     if (p->size+l > p->max_size)
 287         return NULL;
 288
 289     d = AVAHI_DNS_PACKET_DATA(p) + p->size;
 290     p->size += l;
 291
 292     return d;
 293 }
 294
 295 int avahi_dns_packet_check_valid(AvahiDnsPacket *p) {
 296     uint16_t flags;
 297     assert(p);
 298
 299     if (p->size < AVAHI_DNS_PACKET_HEADER_SIZE)
 300         return -1;
 301
 302     flags = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS);
 303
 304     if (flags & AVAHI_DNS_FLAG_OPCODE)
 305         return -1;
 306
 307     return 0;
 308 }
 309
 310 int avahi_dns_packet_check_valid_multicast(AvahiDnsPacket *p) {
 311     uint16_t flags;
 312     assert(p);
 313
 314     if (avahi_dns_packet_check_valid(p) < 0)
 315         return -1;
 316
 317     flags = avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS);
 318
 319     if (flags & AVAHI_DNS_FLAG_RCODE)
 320         return -1;
 321
 322     return 0;
 323 }
 324
 325 int avahi_dns_packet_is_query(AvahiDnsPacket *p) {
 326     assert(p);
 327
 328     return !(avahi_dns_packet_get_field(p, AVAHI_DNS_FIELD_FLAGS) & AVAHI_DNS_FLAG_QR);
 329 }
 330
 331 static int consume_labels(AvahiDnsPacket *p, unsigned idx, char *ret_name, size_t l) {
 332     int ret = 0;
 333     int compressed = 0;
 334     int first_label = 1;
 335     unsigned label_ptr;
 336     int i;
 337     assert(p && ret_name && l);
 338
 339     for (i = 0; i < AVAHI_DNS_LABELS_MAX; i++) {
 340         uint8_t n;
 341
 342         if (idx+1 > p->size)
 343             return -1;
 344
 345         n = AVAHI_DNS_PACKET_DATA(p)[idx];
 346
 347         if (!n) {
 348             idx++;
 349             if (!compressed)
 350                 ret++;
 351
 352             if (l < 1)
 353                 return -1;
 354             *ret_name = 0;
 355
 356             return ret;
 357
 358         } else if (n <= 63) {
 359             /* Uncompressed label */
 360             idx++;
 361             if (!compressed)
 362                 ret++;
 363
 364             if (idx + n > p->size)
 365                 return -1;
 366
 367             if ((size_t) n + 1 > l)
 368                 return -1;
 369
 370             if (!first_label) {
 371                 *(ret_name++) = '.';
 372                 l--;
 373             } else
 374                 first_label = 0;
 375
 376             if (!(avahi_escape_label((char*) AVAHI_DNS_PACKET_DATA(p) + idx, n, &ret_name, &l)))
 377                 return -1;
 378
 379             idx += n;
 380
 381             if (!compressed)
 382                 ret += n;
 383         } else if ((n & 0xC0) == 0xC0) {
 384             /* Compressed label */
 385
 386             if (idx+2 > p->size)
 387                 return -1;
 388
 389             label_ptr = ((unsigned) (AVAHI_DNS_PACKET_DATA(p)[idx] & ~0xC0)) << 8 | AVAHI_DNS_PACKET_DATA(p)[idx+1];
 390
 391             if ((label_ptr < AVAHI_DNS_PACKET_HEADER_SIZE) || (label_ptr >= idx))
 392                 return -1;
 393
 394             idx = label_ptr;
 395
 396             if (!compressed)
 397                 ret += 2;
 398
 399             compressed = 1;
 400         } else
 401             return -1;
 402     }
 403
 404     return -1;
 405 }
 406
 407 int avahi_dns_packet_consume_name(AvahiDnsPacket *p, char *ret_name, size_t l) {
 408     int r;
 409
 410     if ((r = consume_labels(p, p->rindex, ret_name, l)) < 0)
 411         return -1;
 412
 413     p->rindex += r;
 414     return 0;
 415 }
 416
 417 int avahi_dns_packet_consume_uint16(AvahiDnsPacket *p, uint16_t *ret_v) {
 418     uint8_t *d;
 419
 420     assert(p);
 421     assert(ret_v);
 422
 423     if (p->rindex + sizeof(uint16_t) > p->size)
 424         return -1;
 425
 426     d = (uint8_t*) (AVAHI_DNS_PACKET_DATA(p) + p->rindex);
 427     *ret_v = (d[0] << 8) | d[1];
 428     p->rindex += sizeof(uint16_t);
 429
 430     return 0;
 431 }
 432
 433 int avahi_dns_packet_consume_uint32(AvahiDnsPacket *p, uint32_t *ret_v) {
 434     uint8_t* d;
 435
 436     assert(p);
 437     assert(ret_v);
 438
 439     if (p->rindex + sizeof(uint32_t) > p->size)
 440         return -1;
 441
 442     d = (uint8_t*) (AVAHI_DNS_PACKET_DATA(p) + p->rindex);
 443     *ret_v = (d[0] << 24) | (d[1] << 16) | (d[2] << 8) | d[3];
 444     p->rindex += sizeof(uint32_t);
 445
 446     return 0;
 447 }
 448
 449 int avahi_dns_packet_consume_bytes(AvahiDnsPacket *p, void * ret_data, size_t l) {
 450     assert(p);
 451     assert(ret_data);
 452     assert(l > 0);
 453
 454     if (p->rindex + l > p->size)
 455         return -1;
 456
 457     memcpy(ret_data, AVAHI_DNS_PACKET_DATA(p) + p->rindex, l);
 458     p->rindex += l;
 459
 460     return 0;
 461 }
 462
 463 int avahi_dns_packet_consume_string(AvahiDnsPacket *p, char *ret_string, size_t l) {
 464     size_t k;
 465
 466     assert(p);
 467     assert(ret_string);
 468     assert(l > 0);
 469
 470     if (p->rindex >= p->size)
 471         return -1;
 472
 473     k = AVAHI_DNS_PACKET_DATA(p)[p->rindex];
 474
 475     if (p->rindex+1+k > p->size)
 476         return -1;
 477
 478     if (l > k+1)
 479         l = k+1;
 480
 481     memcpy(ret_string, AVAHI_DNS_PACKET_DATA(p)+p->rindex+1, l-1);
 482     ret_string[l-1] = 0;
 483
 484     p->rindex += 1+k;
 485
 486     return 0;
 487 }
 488
 489 const void* avahi_dns_packet_get_rptr(AvahiDnsPacket *p) {
 490     assert(p);
 491
 492     if (p->rindex > p->size)
 493         return NULL;
 494
 495     return AVAHI_DNS_PACKET_DATA(p) + p->rindex;
 496 }
 497
 498 int avahi_dns_packet_skip(AvahiDnsPacket *p, size_t length) {
 499     assert(p);
 500
 501     if (p->rindex + length > p->size)
 502         return -1;
 503
 504     p->rindex += length;
 505     return 0;
 506 }
 507
 508 static int parse_rdata(AvahiDnsPacket *p, AvahiRecord *r, uint16_t rdlength) {
 509     char buf[AVAHI_DOMAIN_NAME_MAX];
 510     const void* start;
 511
 512     assert(p);
 513     assert(r);
 514
 515     start = avahi_dns_packet_get_rptr(p);
 516
 517     switch (r->key->type) {
 518         case AVAHI_DNS_TYPE_PTR:
 519         case AVAHI_DNS_TYPE_CNAME:
 520         case AVAHI_DNS_TYPE_NS:
 521
 522             if (avahi_dns_packet_consume_name(p, buf, sizeof(buf)) < 0)
 523                 return -1;
 524
 525             r->data.ptr.name = avahi_strdup(buf);
 526             break;
 527
 528
 529         case AVAHI_DNS_TYPE_SRV:
 530
 531             if (avahi_dns_packet_consume_uint16(p, &r->data.srv.priority) < 0 ||
 532                 avahi_dns_packet_consume_uint16(p, &r->data.srv.weight) < 0 ||
 533                 avahi_dns_packet_consume_uint16(p, &r->data.srv.port) < 0 ||
 534                 avahi_dns_packet_consume_name(p, buf, sizeof(buf)) < 0)
 535                 return -1;
 536
 537             r->data.srv.name = avahi_strdup(buf);
 538             break;
 539
 540         case AVAHI_DNS_TYPE_HINFO:
 541
 542             if (avahi_dns_packet_consume_string(p, buf, sizeof(buf)) < 0)
 543                 return -1;
 544
 545             r->data.hinfo.cpu = avahi_strdup(buf);
 546
 547             if (avahi_dns_packet_consume_string(p, buf, sizeof(buf)) < 0)
 548                 return -1;
 549
 550             r->data.hinfo.os = avahi_strdup(buf);
 551             break;
 552
 553         case AVAHI_DNS_TYPE_TXT:
 554
 555             if (rdlength > 0) {
 556                 if (avahi_string_list_parse(avahi_dns_packet_get_rptr(p), rdlength, &r->data.txt.string_list) < 0)
 557                     return -1;
 558
 559                 if (avahi_dns_packet_skip(p, rdlength) < 0)
 560                     return -1;
 561             } else
 562                 r->data.txt.string_list = NULL;
 563
 564             break;
 565
 566         case AVAHI_DNS_TYPE_A:
 567
 568 /*             avahi_log_debug("A"); */
 569
 570             if (avahi_dns_packet_consume_bytes(p, &r->data.a.address, sizeof(AvahiIPv4Address)) < 0)
 571                 return -1;
 572
 573             break;
 574
 575         case AVAHI_DNS_TYPE_AAAA:
 576
 577 /*             avahi_log_debug("aaaa"); */
 578
 579             if (avahi_dns_packet_consume_bytes(p, &r->data.aaaa.address, sizeof(AvahiIPv6Address)) < 0)
 580                 return -1;
 581
 582             break;
 583
 584         default:
 585
 586 /*             avahi_log_debug("generic"); */
 587
 588             if (rdlength > 0) {
 589
 590                 r->data.generic.data = avahi_memdup(avahi_dns_packet_get_rptr(p), rdlength);
 591
 592                 if (avahi_dns_packet_skip(p, rdlength) < 0)
 593                     return -1;
 594             }
 595
 596             break;
 597     }
 598
 599     /* Check if we read enough data */
 600     if ((const uint8_t*) avahi_dns_packet_get_rptr(p) - (const uint8_t*) start != rdlength)
 601         return -1;
 602
 603     return 0;
 604 }
 605
 606 AvahiRecord* avahi_dns_packet_consume_record(AvahiDnsPacket *p, int *ret_cache_flush) {
 607     char name[AVAHI_DOMAIN_NAME_MAX];
 608     uint16_t type, class;
 609     uint32_t ttl;
 610     uint16_t rdlength;
 611     AvahiRecord *r = NULL;
 612
 613     assert(p);
 614
 615     if (avahi_dns_packet_consume_name(p, name, sizeof(name)) < 0 ||
 616         avahi_dns_packet_consume_uint16(p, &type) < 0 ||
 617         avahi_dns_packet_consume_uint16(p, &class) < 0 ||
 618         avahi_dns_packet_consume_uint32(p, &ttl) < 0 ||
 619         avahi_dns_packet_consume_uint16(p, &rdlength) < 0 ||
 620         p->rindex + rdlength > p->size)
 621         goto fail;
 622
 623     if (ret_cache_flush)
 624         *ret_cache_flush = !!(class & AVAHI_DNS_CACHE_FLUSH);
 625     class &= ~AVAHI_DNS_CACHE_FLUSH;
 626
 627     if (!(r = avahi_record_new_full(name, class, type, ttl)))
 628         goto fail;
 629
 630     if (parse_rdata(p, r, rdlength) < 0)
 631         goto fail;
 632
 633     if (!avahi_record_is_valid(r))
 634         goto fail;
 635
 636     return r;
 637
 638 fail:
 639     if (r)
 640         avahi_record_unref(r);
 641
 642     return NULL;
 643 }
 644
 645 AvahiKey* avahi_dns_packet_consume_key(AvahiDnsPacket *p, int *ret_unicast_response) {
 646     char name[256];
 647     uint16_t type, class;
 648     AvahiKey *k;
 649
 650     assert(p);
 651
 652     if (avahi_dns_packet_consume_name(p, name, sizeof(name)) < 0 ||
 653         avahi_dns_packet_consume_uint16(p, &type) < 0 ||
 654         avahi_dns_packet_consume_uint16(p, &class) < 0)
 655         return NULL;
 656
 657     if (ret_unicast_response)
 658         *ret_unicast_response = !!(class & AVAHI_DNS_UNICAST_RESPONSE);
 659
 660     class &= ~AVAHI_DNS_UNICAST_RESPONSE;
 661
 662     if (!(k = avahi_key_new(name, class, type)))
 663         return NULL;
 664
 665     if (!avahi_key_is_valid(k)) {
 666         avahi_key_unref(k);
 667         return NULL;
 668     }
 669
 670     return k;
 671 }
 672
 673 uint8_t* avahi_dns_packet_append_key(AvahiDnsPacket *p, AvahiKey *k, int unicast_response) {
 674     uint8_t *t;
 675     size_t size;
 676
 677     assert(p);
 678     assert(k);
 679
 680     size = p->size;
 681
 682     if (!(t = avahi_dns_packet_append_name(p, k->name)) ||
 683         !avahi_dns_packet_append_uint16(p, k->type) ||
 684         !avahi_dns_packet_append_uint16(p, k->clazz | (unicast_response ? AVAHI_DNS_UNICAST_RESPONSE : 0))) {
 685         p->size = size;
 686         return NULL;
 687     }
 688
 689     return t;
 690 }
 691
 692 static int append_rdata(AvahiDnsPacket *p, AvahiRecord *r) {
 693     assert(p);
 694     assert(r);
 695
 696     switch (r->key->type) {
 697
 698         case AVAHI_DNS_TYPE_PTR:
 699         case AVAHI_DNS_TYPE_CNAME:
 700         case AVAHI_DNS_TYPE_NS:
 701
 702             if (!(avahi_dns_packet_append_name(p, r->data.ptr.name)))
 703                 return -1;
 704
 705             break;
 706
 707         case AVAHI_DNS_TYPE_SRV:
 708
 709             if (!avahi_dns_packet_append_uint16(p, r->data.srv.priority) ||
 710                 !avahi_dns_packet_append_uint16(p, r->data.srv.weight) ||
 711                 !avahi_dns_packet_append_uint16(p, r->data.srv.port) ||
 712                 !avahi_dns_packet_append_name(p, r->data.srv.name))
 713                 return -1;
 714
 715             break;
 716
 717         case AVAHI_DNS_TYPE_HINFO:
 718             if (!avahi_dns_packet_append_string(p, r->data.hinfo.cpu) ||
 719                 !avahi_dns_packet_append_string(p, r->data.hinfo.os))
 720                 return -1;
 721
 722             break;
 723
 724         case AVAHI_DNS_TYPE_TXT: {
 725
 726             uint8_t *data;
 727             size_t n;
 728
 729             n = avahi_string_list_serialize(r->data.txt.string_list, NULL, 0);
 730
 731             if (!(data = avahi_dns_packet_extend(p, n)))
 732                 return -1;
 733
 734             avahi_string_list_serialize(r->data.txt.string_list, data, n);
 735             break;
 736         }
 737
 738
 739         case AVAHI_DNS_TYPE_A:
 740
 741             if (!avahi_dns_packet_append_bytes(p, &r->data.a.address, sizeof(r->data.a.address)))
 742                 return -1;
 743
 744             break;
 745
 746         case AVAHI_DNS_TYPE_AAAA:
 747
 748             if (!avahi_dns_packet_append_bytes(p, &r->data.aaaa.address, sizeof(r->data.aaaa.address)))
 749                 return -1;
 750
 751             break;
 752
 753         default:
 754
 755             if (r->data.generic.size)
 756                 if (avahi_dns_packet_append_bytes(p, r->data.generic.data, r->data.generic.size))
 757                     return -1;
 758
 759             break;
 760     }
 761
 762     return 0;
 763 }
 764
 765
 766 uint8_t* avahi_dns_packet_append_record(AvahiDnsPacket *p, AvahiRecord *r, int cache_flush, unsigned max_ttl) {
 767     uint8_t *t, *l, *start;
 768     size_t size;
 769
 770     assert(p);
 771     assert(r);
 772
 773     size = p->size;
 774
 775     if (!(t = avahi_dns_packet_append_name(p, r->key->name)) ||
 776         !avahi_dns_packet_append_uint16(p, r->key->type) ||
 777         !avahi_dns_packet_append_uint16(p, cache_flush ? (r->key->clazz | AVAHI_DNS_CACHE_FLUSH) : (r->key->clazz &~ AVAHI_DNS_CACHE_FLUSH)) ||
 778         !avahi_dns_packet_append_uint32(p, (max_ttl && r->ttl > max_ttl) ? max_ttl : r->ttl) ||
 779         !(l = avahi_dns_packet_append_uint16(p, 0)))
 780         goto fail;
 781
 782     start = avahi_dns_packet_extend(p, 0);
 783
 784     if (append_rdata(p, r) < 0)
 785         goto fail;
 786
 787     size = avahi_dns_packet_extend(p, 0) - start;
 788     assert(size <= 0xFFFF);
 789
 790 /*     avahi_log_debug("appended %u", size); */
 791
 792     l[0] = (uint8_t) ((uint16_t) size >> 8);
 793     l[1] = (uint8_t) ((uint16_t) size);
 794
 795     return t;
 796
 797
 798 fail:
 799     p->size = size;
 800     return NULL;
 801 }
 802
 803 int avahi_dns_packet_is_empty(AvahiDnsPacket *p) {
 804     assert(p);
 805
 806     return p->size <= AVAHI_DNS_PACKET_HEADER_SIZE;
 807 }
 808
 809 size_t avahi_dns_packet_space(AvahiDnsPacket *p) {
 810     assert(p);
 811
 812     assert(p->size <= p->max_size);
 813
 814     return p->max_size - p->size;
 815 }
 816
 817 int avahi_rdata_parse(AvahiRecord *record, const void* rdata, size_t size) {
 818     int ret;
 819     AvahiDnsPacket p;
 820
 821     assert(record);
 822     assert(rdata);
 823
 824     p.data = (void*) rdata;
 825     p.max_size = p.size = size;
 826     p.rindex = 0;
 827     p.name_table = NULL;
 828
 829     ret = parse_rdata(&p, record, size);
 830
 831     assert(!p.name_table);
 832
 833     return ret;
 834 }
 835
 836 size_t avahi_rdata_serialize(AvahiRecord *record, void *rdata, size_t max_size) {
 837     int ret;
 838     AvahiDnsPacket p;
 839
 840     assert(record);
 841     assert(rdata);
 842     assert(max_size > 0);
 843
 844     p.data = (void*) rdata;
 845     p.max_size = max_size;
 846     p.size = p.rindex = 0;
 847     p.name_table = NULL;
 848
 849     ret = append_rdata(&p, record);
 850
 851     if (p.name_table)
 852          avahi_hashmap_free(p.name_table);
 853
 854     if (ret < 0)
 855         return (size_t) -1;
 856
 857     return p.size;
 858 }