2 * Copyright (C) 2004-2012, 2014, 2015 Internet Systems Consortium, Inc. ("ISC")
3 * Copyright (C) 1999-2003 Internet Software Consortium.
5 * Permission to use, copy, modify, and/or distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
9 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
10 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
11 * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
12 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
13 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
14 * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
15 * PERFORMANCE OF THIS SOFTWARE.
26 #include <isc/buffer.h>
28 #include <isc/random.h>
29 #include <isc/serial.h>
33 #include <dns/ncache.h>
34 #include <dns/rdata.h>
35 #include <dns/rdataset.h>
36 #include <dns/compress.h>
38 static const char *trustnames[] = {
48 "local" /* aka ultimate */
52 dns_trust_totext(dns_trust_t trust) {
53 if (trust >= sizeof(trustnames)/sizeof(*trustnames))
55 return (trustnames[trust]);
59 dns_rdataset_init(dns_rdataset_t *rdataset) {
62 * Make 'rdataset' a valid, disassociated rdataset.
65 REQUIRE(rdataset != NULL);
67 rdataset->magic = DNS_RDATASET_MAGIC;
68 rdataset->methods = NULL;
69 ISC_LINK_INIT(rdataset, link);
70 rdataset->rdclass = 0;
75 rdataset->attributes = 0;
76 rdataset->count = ISC_UINT32_MAX;
77 rdataset->private1 = NULL;
78 rdataset->private2 = NULL;
79 rdataset->private3 = NULL;
80 rdataset->privateuint4 = 0;
81 rdataset->private5 = NULL;
82 rdataset->private6 = NULL;
83 rdataset->private7 = NULL;
88 dns_rdataset_invalidate(dns_rdataset_t *rdataset) {
91 * Invalidate 'rdataset'.
94 REQUIRE(DNS_RDATASET_VALID(rdataset));
95 REQUIRE(rdataset->methods == NULL);
98 ISC_LINK_INIT(rdataset, link);
99 rdataset->rdclass = 0;
103 rdataset->covers = 0;
104 rdataset->attributes = 0;
105 rdataset->count = ISC_UINT32_MAX;
106 rdataset->private1 = NULL;
107 rdataset->private2 = NULL;
108 rdataset->private3 = NULL;
109 rdataset->privateuint4 = 0;
110 rdataset->private5 = NULL;
114 dns_rdataset_disassociate(dns_rdataset_t *rdataset) {
117 * Disassociate 'rdataset' from its rdata, allowing it to be reused.
120 REQUIRE(DNS_RDATASET_VALID(rdataset));
121 REQUIRE(rdataset->methods != NULL);
123 (rdataset->methods->disassociate)(rdataset);
124 rdataset->methods = NULL;
125 ISC_LINK_INIT(rdataset, link);
126 rdataset->rdclass = 0;
130 rdataset->covers = 0;
131 rdataset->attributes = 0;
132 rdataset->count = ISC_UINT32_MAX;
133 rdataset->private1 = NULL;
134 rdataset->private2 = NULL;
135 rdataset->private3 = NULL;
136 rdataset->privateuint4 = 0;
137 rdataset->private5 = NULL;
138 rdataset->private6 = NULL;
142 dns_rdataset_isassociated(dns_rdataset_t *rdataset) {
144 * Is 'rdataset' associated?
147 REQUIRE(DNS_RDATASET_VALID(rdataset));
149 if (rdataset->methods != NULL)
156 question_disassociate(dns_rdataset_t *rdataset) {
161 question_cursor(dns_rdataset_t *rdataset) {
164 return (ISC_R_NOMORE);
168 question_current(dns_rdataset_t *rdataset, dns_rdata_t *rdata) {
170 * This routine should never be called.
179 question_clone(dns_rdataset_t *source, dns_rdataset_t *target) {
184 question_count(dns_rdataset_t *rdataset) {
186 * This routine should never be called.
194 static dns_rdatasetmethods_t question_methods = {
195 question_disassociate,
213 dns_rdataset_makequestion(dns_rdataset_t *rdataset, dns_rdataclass_t rdclass,
214 dns_rdatatype_t type)
218 * Make 'rdataset' a valid, associated, question rdataset, with a
219 * question class of 'rdclass' and type 'type'.
222 REQUIRE(DNS_RDATASET_VALID(rdataset));
223 REQUIRE(rdataset->methods == NULL);
225 rdataset->methods = &question_methods;
226 rdataset->rdclass = rdclass;
227 rdataset->type = type;
228 rdataset->attributes |= DNS_RDATASETATTR_QUESTION;
232 dns_rdataset_count(dns_rdataset_t *rdataset) {
235 * Return the number of records in 'rdataset'.
238 REQUIRE(DNS_RDATASET_VALID(rdataset));
239 REQUIRE(rdataset->methods != NULL);
241 return ((rdataset->methods->count)(rdataset));
245 dns_rdataset_clone(dns_rdataset_t *source, dns_rdataset_t *target) {
248 * Make 'target' refer to the same rdataset as 'source'.
251 REQUIRE(DNS_RDATASET_VALID(source));
252 REQUIRE(source->methods != NULL);
253 REQUIRE(DNS_RDATASET_VALID(target));
254 REQUIRE(target->methods == NULL);
256 (source->methods->clone)(source, target);
260 dns_rdataset_first(dns_rdataset_t *rdataset) {
263 * Move the rdata cursor to the first rdata in the rdataset (if any).
266 REQUIRE(DNS_RDATASET_VALID(rdataset));
267 REQUIRE(rdataset->methods != NULL);
269 return ((rdataset->methods->first)(rdataset));
273 dns_rdataset_next(dns_rdataset_t *rdataset) {
276 * Move the rdata cursor to the next rdata in the rdataset (if any).
279 REQUIRE(DNS_RDATASET_VALID(rdataset));
280 REQUIRE(rdataset->methods != NULL);
282 return ((rdataset->methods->next)(rdataset));
286 dns_rdataset_current(dns_rdataset_t *rdataset, dns_rdata_t *rdata) {
289 * Make 'rdata' refer to the current rdata.
292 REQUIRE(DNS_RDATASET_VALID(rdataset));
293 REQUIRE(rdataset->methods != NULL);
295 (rdataset->methods->current)(rdataset, rdata);
298 #define MAX_SHUFFLE 32
299 #define WANT_FIXED(r) (((r)->attributes & DNS_RDATASETATTR_FIXEDORDER) != 0)
300 #define WANT_RANDOM(r) (((r)->attributes & DNS_RDATASETATTR_RANDOMIZE) != 0)
308 towire_compare(const void *av, const void *bv) {
309 const struct towire_sort *a = (const struct towire_sort *) av;
310 const struct towire_sort *b = (const struct towire_sort *) bv;
311 return (a->key - b->key);
315 towiresorted(dns_rdataset_t *rdataset, const dns_name_t *owner_name,
316 dns_compress_t *cctx, isc_buffer_t *target,
317 dns_rdatasetorderfunc_t order, const void *order_arg,
318 isc_boolean_t partial, unsigned int options,
319 unsigned int *countp, void **state)
321 dns_rdata_t rdata = DNS_RDATA_INIT;
324 unsigned int i, count = 0, added, choice;
325 isc_buffer_t savedbuffer, rdlen, rrbuffer;
326 unsigned int headlen;
327 isc_boolean_t question = ISC_FALSE;
328 isc_boolean_t shuffle = ISC_FALSE;
329 dns_rdata_t *shuffled = NULL, shuffled_fixed[MAX_SHUFFLE];
330 struct towire_sort *sorted = NULL, sorted_fixed[MAX_SHUFFLE];
335 * Convert 'rdataset' to wire format, compressing names as specified
336 * in cctx, and storing the result in 'target'.
339 REQUIRE(DNS_RDATASET_VALID(rdataset));
340 REQUIRE(countp != NULL);
341 REQUIRE((order == NULL) == (order_arg == NULL));
342 REQUIRE(cctx != NULL && cctx->mctx != NULL);
344 if ((rdataset->attributes & DNS_RDATASETATTR_QUESTION) != 0) {
347 result = dns_rdataset_first(rdataset);
348 INSIST(result == ISC_R_NOMORE);
349 } else if ((rdataset->attributes & DNS_RDATASETATTR_NEGATIVE) != 0) {
351 * This is a negative caching rdataset.
353 unsigned int ncache_opts = 0;
354 if ((options & DNS_RDATASETTOWIRE_OMITDNSSEC) != 0)
355 ncache_opts |= DNS_NCACHETOWIRE_OMITDNSSEC;
356 return (dns_ncache_towire(rdataset, cctx, target, ncache_opts,
359 count = (rdataset->methods->count)(rdataset);
360 result = dns_rdataset_first(rdataset);
361 if (result == ISC_R_NOMORE)
362 return (ISC_R_SUCCESS);
363 if (result != ISC_R_SUCCESS)
368 * Do we want to shuffle this answer?
370 if (!question && count > 1 &&
371 (!WANT_FIXED(rdataset) || order != NULL) &&
372 rdataset->type != dns_rdatatype_rrsig)
375 if (shuffle && count > MAX_SHUFFLE) {
376 shuffled = isc_mem_get(cctx->mctx, count * sizeof(*shuffled));
377 sorted = isc_mem_get(cctx->mctx, count * sizeof(*sorted));
378 if (shuffled == NULL || sorted == NULL)
381 shuffled = shuffled_fixed;
382 sorted = sorted_fixed;
387 * First we get handles to all of the rdata.
392 dns_rdata_init(&shuffled[i]);
393 dns_rdataset_current(rdataset, &shuffled[i]);
395 result = dns_rdataset_next(rdataset);
396 } while (result == ISC_R_SUCCESS);
397 if (result != ISC_R_NOMORE)
404 if (WANT_FIXED(rdataset)) {
408 INSIST(order != NULL);
409 for (i = 0; i < count; i++) {
410 sorted[i].key = (*order)(&shuffled[i],
412 sorted[i].rdata = &shuffled[i];
414 } else if (WANT_RANDOM(rdataset)) {
418 for (i = 0; i < count; i++) {
421 isc_random_get(&val);
422 choice = i + (val % (count - i));
424 shuffled[i] = shuffled[choice];
425 shuffled[choice] = rdata;
427 sorted[i].key = (*order)(&shuffled[i],
430 sorted[i].key = 0; /* Unused */
431 sorted[i].rdata = &shuffled[i];
440 val = rdataset->count;
441 if (val == ISC_UINT32_MAX)
442 isc_random_get(&val);
444 for (i = 0; i < count; i++) {
446 sorted[i].key = (*order)(&shuffled[j],
449 sorted[i].key = 0; /* Unused */
450 sorted[i].rdata = &shuffled[j];
453 j = 0; /* Wrap around. */
461 qsort(sorted, count, sizeof(sorted[0]),
465 savedbuffer = *target;
471 * Copy out the name, type, class, ttl.
475 dns_compress_setmethods(cctx, DNS_COMPRESS_GLOBAL14);
476 result = dns_name_towire(owner_name, cctx, target);
477 if (result != ISC_R_SUCCESS)
479 headlen = sizeof(dns_rdataclass_t) + sizeof(dns_rdatatype_t);
481 headlen += sizeof(dns_ttl_t)
482 + 2; /* XXX 2 for rdata len */
483 isc_buffer_availableregion(target, &r);
484 if (r.length < headlen) {
485 result = ISC_R_NOSPACE;
488 isc_buffer_putuint16(target, rdataset->type);
489 isc_buffer_putuint16(target, rdataset->rdclass);
491 isc_buffer_putuint32(target, rdataset->ttl);
494 * Save space for rdlen.
497 isc_buffer_add(target, 2);
503 rdata = *(sorted[i].rdata);
505 dns_rdata_reset(&rdata);
506 dns_rdataset_current(rdataset, &rdata);
508 result = dns_rdata_towire(&rdata, cctx, target);
509 if (result != ISC_R_SUCCESS)
511 INSIST((target->used >= rdlen.used + 2) &&
512 (target->used - rdlen.used - 2 < 65536));
513 isc_buffer_putuint16(&rdlen,
514 (isc_uint16_t)(target->used -
522 result = ISC_R_NOMORE;
524 result = ISC_R_SUCCESS;
526 result = dns_rdataset_next(rdataset);
528 } while (result == ISC_R_SUCCESS);
530 if (result != ISC_R_NOMORE)
535 result = ISC_R_SUCCESS;
539 if (partial && result == ISC_R_NOSPACE) {
540 INSIST(rrbuffer.used < 65536);
541 dns_compress_rollback(cctx, (isc_uint16_t)rrbuffer.used);
546 INSIST(savedbuffer.used < 65536);
547 dns_compress_rollback(cctx, (isc_uint16_t)savedbuffer.used);
549 *target = savedbuffer;
552 if (sorted != NULL && sorted != sorted_fixed)
553 isc_mem_put(cctx->mctx, sorted, count * sizeof(*sorted));
554 if (shuffled != NULL && shuffled != shuffled_fixed)
555 isc_mem_put(cctx->mctx, shuffled, count * sizeof(*shuffled));
560 dns_rdataset_towiresorted(dns_rdataset_t *rdataset,
561 const dns_name_t *owner_name,
562 dns_compress_t *cctx,
563 isc_buffer_t *target,
564 dns_rdatasetorderfunc_t order,
565 const void *order_arg,
566 unsigned int options,
567 unsigned int *countp)
569 return (towiresorted(rdataset, owner_name, cctx, target,
570 order, order_arg, ISC_FALSE, options,
575 dns_rdataset_towirepartial(dns_rdataset_t *rdataset,
576 const dns_name_t *owner_name,
577 dns_compress_t *cctx,
578 isc_buffer_t *target,
579 dns_rdatasetorderfunc_t order,
580 const void *order_arg,
581 unsigned int options,
582 unsigned int *countp,
585 REQUIRE(state == NULL); /* XXX remove when implemented */
586 return (towiresorted(rdataset, owner_name, cctx, target,
587 order, order_arg, ISC_TRUE, options,
592 dns_rdataset_towire(dns_rdataset_t *rdataset,
593 dns_name_t *owner_name,
594 dns_compress_t *cctx,
595 isc_buffer_t *target,
596 unsigned int options,
597 unsigned int *countp)
599 return (towiresorted(rdataset, owner_name, cctx, target,
600 NULL, NULL, ISC_FALSE, options, countp, NULL));
604 dns_rdataset_additionaldata(dns_rdataset_t *rdataset,
605 dns_additionaldatafunc_t add, void *arg)
607 dns_rdata_t rdata = DNS_RDATA_INIT;
611 * For each rdata in rdataset, call 'add' for each name and type in the
612 * rdata which is subject to additional section processing.
615 REQUIRE(DNS_RDATASET_VALID(rdataset));
616 REQUIRE((rdataset->attributes & DNS_RDATASETATTR_QUESTION) == 0);
618 result = dns_rdataset_first(rdataset);
619 if (result != ISC_R_SUCCESS)
623 dns_rdataset_current(rdataset, &rdata);
624 result = dns_rdata_additionaldata(&rdata, add, arg);
625 if (result == ISC_R_SUCCESS)
626 result = dns_rdataset_next(rdataset);
627 dns_rdata_reset(&rdata);
628 } while (result == ISC_R_SUCCESS);
630 if (result != ISC_R_NOMORE)
633 return (ISC_R_SUCCESS);
637 dns_rdataset_addnoqname(dns_rdataset_t *rdataset, dns_name_t *name) {
639 REQUIRE(DNS_RDATASET_VALID(rdataset));
640 REQUIRE(rdataset->methods != NULL);
641 if (rdataset->methods->addnoqname == NULL)
642 return (ISC_R_NOTIMPLEMENTED);
643 return((rdataset->methods->addnoqname)(rdataset, name));
647 dns_rdataset_getnoqname(dns_rdataset_t *rdataset, dns_name_t *name,
648 dns_rdataset_t *neg, dns_rdataset_t *negsig)
650 REQUIRE(DNS_RDATASET_VALID(rdataset));
651 REQUIRE(rdataset->methods != NULL);
653 if (rdataset->methods->getnoqname == NULL)
654 return (ISC_R_NOTIMPLEMENTED);
655 return((rdataset->methods->getnoqname)(rdataset, name, neg, negsig));
659 dns_rdataset_addclosest(dns_rdataset_t *rdataset, dns_name_t *name) {
661 REQUIRE(DNS_RDATASET_VALID(rdataset));
662 REQUIRE(rdataset->methods != NULL);
663 if (rdataset->methods->addclosest == NULL)
664 return (ISC_R_NOTIMPLEMENTED);
665 return((rdataset->methods->addclosest)(rdataset, name));
669 dns_rdataset_getclosest(dns_rdataset_t *rdataset, dns_name_t *name,
670 dns_rdataset_t *neg, dns_rdataset_t *negsig)
672 REQUIRE(DNS_RDATASET_VALID(rdataset));
673 REQUIRE(rdataset->methods != NULL);
675 if (rdataset->methods->getclosest == NULL)
676 return (ISC_R_NOTIMPLEMENTED);
677 return((rdataset->methods->getclosest)(rdataset, name, neg, negsig));
681 * Additional cache stuff
684 dns_rdataset_getadditional(dns_rdataset_t *rdataset,
685 dns_rdatasetadditional_t type,
686 dns_rdatatype_t qtype,
687 dns_acache_t *acache,
690 dns_dbversion_t **versionp,
691 dns_dbnode_t **nodep,
696 REQUIRE(DNS_RDATASET_VALID(rdataset));
697 REQUIRE(rdataset->methods != NULL);
698 REQUIRE(zonep == NULL || *zonep == NULL);
699 REQUIRE(dbp != NULL && *dbp == NULL);
700 REQUIRE(versionp != NULL && *versionp == NULL);
701 REQUIRE(nodep != NULL && *nodep == NULL);
702 REQUIRE(fname != NULL);
703 REQUIRE(msg != NULL);
705 if (acache != NULL && rdataset->methods->getadditional != NULL) {
706 return ((rdataset->methods->getadditional)(rdataset, type,
713 return (ISC_R_FAILURE);
717 dns_rdataset_setadditional(dns_rdataset_t *rdataset,
718 dns_rdatasetadditional_t type,
719 dns_rdatatype_t qtype,
720 dns_acache_t *acache,
723 dns_dbversion_t *version,
727 REQUIRE(DNS_RDATASET_VALID(rdataset));
728 REQUIRE(rdataset->methods != NULL);
730 if (acache != NULL && rdataset->methods->setadditional != NULL) {
731 return ((rdataset->methods->setadditional)(rdataset, type,
737 return (ISC_R_FAILURE);
741 dns_rdataset_putadditional(dns_acache_t *acache,
742 dns_rdataset_t *rdataset,
743 dns_rdatasetadditional_t type,
744 dns_rdatatype_t qtype)
746 REQUIRE(DNS_RDATASET_VALID(rdataset));
747 REQUIRE(rdataset->methods != NULL);
749 if (acache != NULL && rdataset->methods->putadditional != NULL) {
750 return ((rdataset->methods->putadditional)(acache, rdataset,
754 return (ISC_R_FAILURE);
758 dns_rdataset_settrust(dns_rdataset_t *rdataset, dns_trust_t trust) {
759 REQUIRE(DNS_RDATASET_VALID(rdataset));
760 REQUIRE(rdataset->methods != NULL);
762 if (rdataset->methods->settrust != NULL)
763 (rdataset->methods->settrust)(rdataset, trust);
765 rdataset->trust = trust;
769 dns_rdataset_expire(dns_rdataset_t *rdataset) {
770 REQUIRE(DNS_RDATASET_VALID(rdataset));
771 REQUIRE(rdataset->methods != NULL);
773 if (rdataset->methods->expire != NULL)
774 (rdataset->methods->expire)(rdataset);
778 dns_rdataset_trimttl(dns_rdataset_t *rdataset, dns_rdataset_t *sigrdataset,
779 dns_rdata_rrsig_t *rrsig, isc_stdtime_t now,
780 isc_boolean_t acceptexpired)
782 isc_uint32_t ttl = 0;
784 REQUIRE(DNS_RDATASET_VALID(rdataset));
785 REQUIRE(DNS_RDATASET_VALID(sigrdataset));
786 REQUIRE(rrsig != NULL);
789 * If we accept expired RRsets keep them for no more than 120 seconds.
792 (isc_serial_le(rrsig->timeexpire, ((now + 120) & 0xffffffff)) ||
793 isc_serial_le(rrsig->timeexpire, now)))
795 else if (isc_serial_ge(rrsig->timeexpire, now))
796 ttl = rrsig->timeexpire - now;
798 ttl = ISC_MIN(ISC_MIN(rdataset->ttl, sigrdataset->ttl),
799 ISC_MIN(rrsig->originalttl, ttl));
801 sigrdataset->ttl = ttl;