Home | History | Annotate | Line # | Download | only in dns
ncache.c revision 1.1.1.6
      1      1.1  christos /*	$NetBSD: ncache.c,v 1.1.1.6 2022/09/23 12:09:17 christos Exp $	*/
      2      1.1  christos 
      3      1.1  christos /*
      4      1.1  christos  * Copyright (C) Internet Systems Consortium, Inc. ("ISC")
      5      1.1  christos  *
      6  1.1.1.6  christos  * SPDX-License-Identifier: MPL-2.0
      7  1.1.1.6  christos  *
      8      1.1  christos  * This Source Code Form is subject to the terms of the Mozilla Public
      9      1.1  christos  * License, v. 2.0. If a copy of the MPL was not distributed with this
     10  1.1.1.5  christos  * file, you can obtain one at https://mozilla.org/MPL/2.0/.
     11      1.1  christos  *
     12      1.1  christos  * See the COPYRIGHT file distributed with this work for additional
     13      1.1  christos  * information regarding copyright ownership.
     14      1.1  christos  */
     15      1.1  christos 
     16      1.1  christos /*! \file */
     17      1.1  christos 
     18  1.1.1.2  christos #include <inttypes.h>
     19  1.1.1.2  christos #include <stdbool.h>
     20  1.1.1.2  christos 
     21      1.1  christos #include <isc/buffer.h>
     22      1.1  christos #include <isc/util.h>
     23      1.1  christos 
     24      1.1  christos #include <dns/db.h>
     25      1.1  christos #include <dns/message.h>
     26      1.1  christos #include <dns/ncache.h>
     27      1.1  christos #include <dns/rdata.h>
     28      1.1  christos #include <dns/rdatalist.h>
     29      1.1  christos #include <dns/rdataset.h>
     30      1.1  christos #include <dns/rdatastruct.h>
     31      1.1  christos 
     32  1.1.1.3  christos #define DNS_NCACHE_RDATA 100U
     33      1.1  christos 
     34      1.1  christos /*
     35      1.1  christos  * The format of an ncache rdata is a sequence of zero or more records of
     36      1.1  christos  * the following format:
     37      1.1  christos  *
     38      1.1  christos  *	owner name
     39      1.1  christos  *	type
     40      1.1  christos  *	trust
     41      1.1  christos  *	rdata count
     42      1.1  christos  *		rdata length			These two occur 'rdata count'
     43      1.1  christos  *		rdata				times.
     44      1.1  christos  *
     45      1.1  christos  */
     46      1.1  christos 
     47      1.1  christos static isc_result_t
     48      1.1  christos addoptout(dns_message_t *message, dns_db_t *cache, dns_dbnode_t *node,
     49  1.1.1.4  christos 	  dns_rdatatype_t covers, isc_stdtime_t now, dns_ttl_t minttl,
     50  1.1.1.4  christos 	  dns_ttl_t maxttl, bool optout, bool secure,
     51      1.1  christos 	  dns_rdataset_t *addedrdataset);
     52      1.1  christos 
     53  1.1.1.6  christos static isc_result_t
     54      1.1  christos copy_rdataset(dns_rdataset_t *rdataset, isc_buffer_t *buffer) {
     55      1.1  christos 	isc_result_t result;
     56      1.1  christos 	unsigned int count;
     57      1.1  christos 	isc_region_t ar, r;
     58      1.1  christos 	dns_rdata_t rdata = DNS_RDATA_INIT;
     59      1.1  christos 
     60      1.1  christos 	/*
     61      1.1  christos 	 * Copy the rdataset count to the buffer.
     62      1.1  christos 	 */
     63      1.1  christos 	isc_buffer_availableregion(buffer, &ar);
     64  1.1.1.4  christos 	if (ar.length < 2) {
     65      1.1  christos 		return (ISC_R_NOSPACE);
     66  1.1.1.4  christos 	}
     67      1.1  christos 	count = dns_rdataset_count(rdataset);
     68      1.1  christos 	INSIST(count <= 65535);
     69  1.1.1.2  christos 	isc_buffer_putuint16(buffer, (uint16_t)count);
     70      1.1  christos 
     71      1.1  christos 	result = dns_rdataset_first(rdataset);
     72      1.1  christos 	while (result == ISC_R_SUCCESS) {
     73      1.1  christos 		dns_rdataset_current(rdataset, &rdata);
     74      1.1  christos 		dns_rdata_toregion(&rdata, &r);
     75      1.1  christos 		INSIST(r.length <= 65535);
     76      1.1  christos 		isc_buffer_availableregion(buffer, &ar);
     77  1.1.1.4  christos 		if (ar.length < 2) {
     78      1.1  christos 			return (ISC_R_NOSPACE);
     79  1.1.1.4  christos 		}
     80      1.1  christos 		/*
     81      1.1  christos 		 * Copy the rdata length to the buffer.
     82      1.1  christos 		 */
     83  1.1.1.2  christos 		isc_buffer_putuint16(buffer, (uint16_t)r.length);
     84      1.1  christos 		/*
     85      1.1  christos 		 * Copy the rdata to the buffer.
     86      1.1  christos 		 */
     87      1.1  christos 		result = isc_buffer_copyregion(buffer, &r);
     88  1.1.1.4  christos 		if (result != ISC_R_SUCCESS) {
     89      1.1  christos 			return (result);
     90  1.1.1.4  christos 		}
     91      1.1  christos 		dns_rdata_reset(&rdata);
     92      1.1  christos 		result = dns_rdataset_next(rdataset);
     93      1.1  christos 	}
     94  1.1.1.4  christos 	if (result != ISC_R_NOMORE) {
     95      1.1  christos 		return (result);
     96  1.1.1.4  christos 	}
     97      1.1  christos 
     98      1.1  christos 	return (ISC_R_SUCCESS);
     99      1.1  christos }
    100      1.1  christos 
    101      1.1  christos isc_result_t
    102      1.1  christos dns_ncache_add(dns_message_t *message, dns_db_t *cache, dns_dbnode_t *node,
    103  1.1.1.4  christos 	       dns_rdatatype_t covers, isc_stdtime_t now, dns_ttl_t minttl,
    104  1.1.1.4  christos 	       dns_ttl_t maxttl, dns_rdataset_t *addedrdataset) {
    105  1.1.1.2  christos 	return (addoptout(message, cache, node, covers, now, minttl, maxttl,
    106  1.1.1.2  christos 			  false, false, addedrdataset));
    107      1.1  christos }
    108      1.1  christos 
    109      1.1  christos isc_result_t
    110      1.1  christos dns_ncache_addoptout(dns_message_t *message, dns_db_t *cache,
    111      1.1  christos 		     dns_dbnode_t *node, dns_rdatatype_t covers,
    112  1.1.1.4  christos 		     isc_stdtime_t now, dns_ttl_t minttl, dns_ttl_t maxttl,
    113  1.1.1.4  christos 		     bool optout, dns_rdataset_t *addedrdataset) {
    114  1.1.1.2  christos 	return (addoptout(message, cache, node, covers, now, minttl, maxttl,
    115  1.1.1.2  christos 			  optout, true, addedrdataset));
    116      1.1  christos }
    117      1.1  christos 
    118      1.1  christos static isc_result_t
    119      1.1  christos addoptout(dns_message_t *message, dns_db_t *cache, dns_dbnode_t *node,
    120  1.1.1.4  christos 	  dns_rdatatype_t covers, isc_stdtime_t now, dns_ttl_t minttl,
    121  1.1.1.4  christos 	  dns_ttl_t maxttl, bool optout, bool secure,
    122  1.1.1.4  christos 	  dns_rdataset_t *addedrdataset) {
    123      1.1  christos 	isc_result_t result;
    124      1.1  christos 	isc_buffer_t buffer;
    125      1.1  christos 	isc_region_t r;
    126      1.1  christos 	dns_rdataset_t *rdataset;
    127      1.1  christos 	dns_rdatatype_t type;
    128      1.1  christos 	dns_name_t *name;
    129      1.1  christos 	dns_ttl_t ttl;
    130      1.1  christos 	dns_trust_t trust;
    131      1.1  christos 	dns_rdata_t rdata[DNS_NCACHE_RDATA];
    132      1.1  christos 	dns_rdataset_t ncrdataset;
    133      1.1  christos 	dns_rdatalist_t ncrdatalist;
    134  1.1.1.3  christos 	unsigned char data[65536];
    135      1.1  christos 	unsigned int next = 0;
    136      1.1  christos 
    137      1.1  christos 	/*
    138      1.1  christos 	 * Convert the authority data from 'message' into a negative cache
    139      1.1  christos 	 * rdataset, and store it in 'cache' at 'node'.
    140      1.1  christos 	 */
    141      1.1  christos 
    142      1.1  christos 	REQUIRE(message != NULL);
    143      1.1  christos 
    144      1.1  christos 	/*
    145      1.1  christos 	 * We assume that all data in the authority section has been
    146      1.1  christos 	 * validated by the caller.
    147      1.1  christos 	 */
    148      1.1  christos 
    149      1.1  christos 	/*
    150      1.1  christos 	 * Initialize the list.
    151      1.1  christos 	 */
    152      1.1  christos 	dns_rdatalist_init(&ncrdatalist);
    153      1.1  christos 	ncrdatalist.rdclass = dns_db_class(cache);
    154      1.1  christos 	ncrdatalist.covers = covers;
    155      1.1  christos 	ncrdatalist.ttl = maxttl;
    156      1.1  christos 
    157      1.1  christos 	/*
    158      1.1  christos 	 * Build an ncache rdatas into buffer.
    159      1.1  christos 	 */
    160      1.1  christos 	ttl = maxttl;
    161      1.1  christos 	trust = 0xffff;
    162      1.1  christos 	isc_buffer_init(&buffer, data, sizeof(data));
    163  1.1.1.4  christos 	if (message->counts[DNS_SECTION_AUTHORITY]) {
    164      1.1  christos 		result = dns_message_firstname(message, DNS_SECTION_AUTHORITY);
    165  1.1.1.4  christos 	} else {
    166      1.1  christos 		result = ISC_R_NOMORE;
    167  1.1.1.4  christos 	}
    168      1.1  christos 	while (result == ISC_R_SUCCESS) {
    169      1.1  christos 		name = NULL;
    170  1.1.1.4  christos 		dns_message_currentname(message, DNS_SECTION_AUTHORITY, &name);
    171      1.1  christos 		if ((name->attributes & DNS_NAMEATTR_NCACHE) != 0) {
    172      1.1  christos 			for (rdataset = ISC_LIST_HEAD(name->list);
    173      1.1  christos 			     rdataset != NULL;
    174  1.1.1.4  christos 			     rdataset = ISC_LIST_NEXT(rdataset, link))
    175  1.1.1.4  christos 			{
    176      1.1  christos 				if ((rdataset->attributes &
    177  1.1.1.4  christos 				     DNS_RDATASETATTR_NCACHE) == 0) {
    178      1.1  christos 					continue;
    179  1.1.1.4  christos 				}
    180      1.1  christos 				type = rdataset->type;
    181  1.1.1.4  christos 				if (type == dns_rdatatype_rrsig) {
    182      1.1  christos 					type = rdataset->covers;
    183  1.1.1.4  christos 				}
    184      1.1  christos 				if (type == dns_rdatatype_soa ||
    185      1.1  christos 				    type == dns_rdatatype_nsec ||
    186  1.1.1.4  christos 				    type == dns_rdatatype_nsec3)
    187  1.1.1.4  christos 				{
    188  1.1.1.2  christos 					if (ttl > rdataset->ttl) {
    189      1.1  christos 						ttl = rdataset->ttl;
    190  1.1.1.2  christos 					}
    191  1.1.1.2  christos 					if (ttl < minttl) {
    192  1.1.1.2  christos 						ttl = minttl;
    193  1.1.1.2  christos 					}
    194  1.1.1.2  christos 					if (trust > rdataset->trust) {
    195      1.1  christos 						trust = rdataset->trust;
    196  1.1.1.2  christos 					}
    197      1.1  christos 					/*
    198      1.1  christos 					 * Copy the owner name to the buffer.
    199      1.1  christos 					 */
    200      1.1  christos 					dns_name_toregion(name, &r);
    201      1.1  christos 					result = isc_buffer_copyregion(&buffer,
    202      1.1  christos 								       &r);
    203  1.1.1.4  christos 					if (result != ISC_R_SUCCESS) {
    204      1.1  christos 						return (result);
    205  1.1.1.4  christos 					}
    206      1.1  christos 					/*
    207      1.1  christos 					 * Copy the type to the buffer.
    208      1.1  christos 					 */
    209  1.1.1.4  christos 					isc_buffer_availableregion(&buffer, &r);
    210  1.1.1.4  christos 					if (r.length < 3) {
    211      1.1  christos 						return (ISC_R_NOSPACE);
    212  1.1.1.4  christos 					}
    213      1.1  christos 					isc_buffer_putuint16(&buffer,
    214      1.1  christos 							     rdataset->type);
    215  1.1.1.4  christos 					isc_buffer_putuint8(
    216  1.1.1.4  christos 						&buffer,
    217  1.1.1.4  christos 						(unsigned char)rdataset->trust);
    218      1.1  christos 					/*
    219      1.1  christos 					 * Copy the rdataset into the buffer.
    220      1.1  christos 					 */
    221      1.1  christos 					result = copy_rdataset(rdataset,
    222      1.1  christos 							       &buffer);
    223  1.1.1.4  christos 					if (result != ISC_R_SUCCESS) {
    224      1.1  christos 						return (result);
    225  1.1.1.4  christos 					}
    226      1.1  christos 
    227  1.1.1.4  christos 					if (next >= DNS_NCACHE_RDATA) {
    228      1.1  christos 						return (ISC_R_NOSPACE);
    229  1.1.1.4  christos 					}
    230      1.1  christos 					dns_rdata_init(&rdata[next]);
    231      1.1  christos 					isc_buffer_remainingregion(&buffer, &r);
    232      1.1  christos 					rdata[next].data = r.base;
    233      1.1  christos 					rdata[next].length = r.length;
    234      1.1  christos 					rdata[next].rdclass =
    235      1.1  christos 						ncrdatalist.rdclass;
    236      1.1  christos 					rdata[next].type = 0;
    237      1.1  christos 					rdata[next].flags = 0;
    238      1.1  christos 					ISC_LIST_APPEND(ncrdatalist.rdata,
    239      1.1  christos 							&rdata[next], link);
    240      1.1  christos 					isc_buffer_forward(&buffer, r.length);
    241      1.1  christos 					next++;
    242      1.1  christos 				}
    243      1.1  christos 			}
    244      1.1  christos 		}
    245      1.1  christos 		result = dns_message_nextname(message, DNS_SECTION_AUTHORITY);
    246      1.1  christos 	}
    247  1.1.1.4  christos 	if (result != ISC_R_NOMORE) {
    248      1.1  christos 		return (result);
    249  1.1.1.4  christos 	}
    250      1.1  christos 
    251      1.1  christos 	if (trust == 0xffff) {
    252      1.1  christos 		if ((message->flags & DNS_MESSAGEFLAG_AA) != 0 &&
    253  1.1.1.4  christos 		    message->counts[DNS_SECTION_ANSWER] == 0)
    254  1.1.1.4  christos 		{
    255      1.1  christos 			/*
    256      1.1  christos 			 * The response has aa set and we haven't followed
    257      1.1  christos 			 * any CNAME or DNAME chains.
    258      1.1  christos 			 */
    259      1.1  christos 			trust = dns_trust_authauthority;
    260  1.1.1.4  christos 		} else {
    261      1.1  christos 			trust = dns_trust_additional;
    262  1.1.1.4  christos 		}
    263      1.1  christos 		ttl = 0;
    264      1.1  christos 	}
    265      1.1  christos 
    266      1.1  christos 	INSIST(trust != 0xffff);
    267      1.1  christos 
    268      1.1  christos 	ncrdatalist.ttl = ttl;
    269      1.1  christos 
    270      1.1  christos 	dns_rdataset_init(&ncrdataset);
    271  1.1.1.4  christos 	RUNTIME_CHECK(dns_rdatalist_tordataset(&ncrdatalist, &ncrdataset) ==
    272  1.1.1.4  christos 		      ISC_R_SUCCESS);
    273  1.1.1.4  christos 	if (!secure && trust > dns_trust_answer) {
    274      1.1  christos 		trust = dns_trust_answer;
    275  1.1.1.4  christos 	}
    276      1.1  christos 	ncrdataset.trust = trust;
    277      1.1  christos 	ncrdataset.attributes |= DNS_RDATASETATTR_NEGATIVE;
    278  1.1.1.4  christos 	if (message->rcode == dns_rcode_nxdomain) {
    279      1.1  christos 		ncrdataset.attributes |= DNS_RDATASETATTR_NXDOMAIN;
    280  1.1.1.4  christos 	}
    281  1.1.1.4  christos 	if (optout) {
    282      1.1  christos 		ncrdataset.attributes |= DNS_RDATASETATTR_OPTOUT;
    283  1.1.1.4  christos 	}
    284      1.1  christos 
    285  1.1.1.4  christos 	return (dns_db_addrdataset(cache, node, NULL, now, &ncrdataset, 0,
    286  1.1.1.4  christos 				   addedrdataset));
    287      1.1  christos }
    288      1.1  christos 
    289      1.1  christos isc_result_t
    290      1.1  christos dns_ncache_towire(dns_rdataset_t *rdataset, dns_compress_t *cctx,
    291      1.1  christos 		  isc_buffer_t *target, unsigned int options,
    292  1.1.1.4  christos 		  unsigned int *countp) {
    293      1.1  christos 	dns_rdata_t rdata = DNS_RDATA_INIT;
    294      1.1  christos 	isc_result_t result;
    295      1.1  christos 	isc_region_t remaining, tavailable;
    296      1.1  christos 	isc_buffer_t source, savedbuffer, rdlen;
    297      1.1  christos 	dns_name_t name;
    298      1.1  christos 	dns_rdatatype_t type;
    299      1.1  christos 	unsigned int i, rcount, count;
    300      1.1  christos 
    301      1.1  christos 	/*
    302      1.1  christos 	 * Convert the negative caching rdataset 'rdataset' to wire format,
    303      1.1  christos 	 * compressing names as specified in 'cctx', and storing the result in
    304      1.1  christos 	 * 'target'.
    305      1.1  christos 	 */
    306      1.1  christos 
    307      1.1  christos 	REQUIRE(rdataset != NULL);
    308      1.1  christos 	REQUIRE(rdataset->type == 0);
    309      1.1  christos 	REQUIRE((rdataset->attributes & DNS_RDATASETATTR_NEGATIVE) != 0);
    310      1.1  christos 
    311      1.1  christos 	savedbuffer = *target;
    312      1.1  christos 	count = 0;
    313      1.1  christos 
    314      1.1  christos 	result = dns_rdataset_first(rdataset);
    315      1.1  christos 	while (result == ISC_R_SUCCESS) {
    316      1.1  christos 		dns_rdataset_current(rdataset, &rdata);
    317      1.1  christos 		isc_buffer_init(&source, rdata.data, rdata.length);
    318      1.1  christos 		isc_buffer_add(&source, rdata.length);
    319      1.1  christos 		dns_name_init(&name, NULL);
    320      1.1  christos 		isc_buffer_remainingregion(&source, &remaining);
    321      1.1  christos 		dns_name_fromregion(&name, &remaining);
    322      1.1  christos 		INSIST(remaining.length >= name.length);
    323      1.1  christos 		isc_buffer_forward(&source, name.length);
    324      1.1  christos 		remaining.length -= name.length;
    325      1.1  christos 
    326      1.1  christos 		INSIST(remaining.length >= 5);
    327      1.1  christos 		type = isc_buffer_getuint16(&source);
    328      1.1  christos 		isc_buffer_forward(&source, 1);
    329      1.1  christos 		rcount = isc_buffer_getuint16(&source);
    330      1.1  christos 
    331      1.1  christos 		for (i = 0; i < rcount; i++) {
    332      1.1  christos 			/*
    333      1.1  christos 			 * Get the length of this rdata and set up an
    334      1.1  christos 			 * rdata structure for it.
    335      1.1  christos 			 */
    336      1.1  christos 			isc_buffer_remainingregion(&source, &remaining);
    337      1.1  christos 			INSIST(remaining.length >= 2);
    338      1.1  christos 			dns_rdata_reset(&rdata);
    339      1.1  christos 			rdata.length = isc_buffer_getuint16(&source);
    340      1.1  christos 			isc_buffer_remainingregion(&source, &remaining);
    341      1.1  christos 			rdata.data = remaining.base;
    342      1.1  christos 			rdata.type = type;
    343      1.1  christos 			rdata.rdclass = rdataset->rdclass;
    344      1.1  christos 			INSIST(remaining.length >= rdata.length);
    345      1.1  christos 			isc_buffer_forward(&source, rdata.length);
    346      1.1  christos 
    347      1.1  christos 			if ((options & DNS_NCACHETOWIRE_OMITDNSSEC) != 0 &&
    348      1.1  christos 			    dns_rdatatype_isdnssec(type))
    349  1.1.1.4  christos 			{
    350      1.1  christos 				continue;
    351  1.1.1.4  christos 			}
    352      1.1  christos 
    353      1.1  christos 			/*
    354      1.1  christos 			 * Write the name.
    355      1.1  christos 			 */
    356      1.1  christos 			dns_compress_setmethods(cctx, DNS_COMPRESS_GLOBAL14);
    357      1.1  christos 			result = dns_name_towire(&name, cctx, target);
    358  1.1.1.4  christos 			if (result != ISC_R_SUCCESS) {
    359      1.1  christos 				goto rollback;
    360  1.1.1.4  christos 			}
    361      1.1  christos 
    362      1.1  christos 			/*
    363      1.1  christos 			 * See if we have space for type, class, ttl, and
    364      1.1  christos 			 * rdata length.  Write the type, class, and ttl.
    365      1.1  christos 			 */
    366      1.1  christos 			isc_buffer_availableregion(target, &tavailable);
    367      1.1  christos 			if (tavailable.length < 10) {
    368      1.1  christos 				result = ISC_R_NOSPACE;
    369      1.1  christos 				goto rollback;
    370      1.1  christos 			}
    371      1.1  christos 			isc_buffer_putuint16(target, type);
    372      1.1  christos 			isc_buffer_putuint16(target, rdataset->rdclass);
    373      1.1  christos 			isc_buffer_putuint32(target, rdataset->ttl);
    374      1.1  christos 
    375      1.1  christos 			/*
    376      1.1  christos 			 * Save space for rdata length.
    377      1.1  christos 			 */
    378      1.1  christos 			rdlen = *target;
    379      1.1  christos 			isc_buffer_add(target, 2);
    380      1.1  christos 
    381      1.1  christos 			/*
    382      1.1  christos 			 * Write the rdata.
    383      1.1  christos 			 */
    384      1.1  christos 			result = dns_rdata_towire(&rdata, cctx, target);
    385  1.1.1.4  christos 			if (result != ISC_R_SUCCESS) {
    386      1.1  christos 				goto rollback;
    387  1.1.1.4  christos 			}
    388      1.1  christos 
    389      1.1  christos 			/*
    390      1.1  christos 			 * Set the rdata length field to the compressed
    391      1.1  christos 			 * length.
    392      1.1  christos 			 */
    393      1.1  christos 			INSIST((target->used >= rdlen.used + 2) &&
    394      1.1  christos 			       (target->used - rdlen.used - 2 < 65536));
    395  1.1.1.4  christos 			isc_buffer_putuint16(
    396  1.1.1.4  christos 				&rdlen,
    397  1.1.1.4  christos 				(uint16_t)(target->used - rdlen.used - 2));
    398      1.1  christos 
    399      1.1  christos 			count++;
    400      1.1  christos 		}
    401      1.1  christos 		INSIST(isc_buffer_remaininglength(&source) == 0);
    402      1.1  christos 		result = dns_rdataset_next(rdataset);
    403      1.1  christos 		dns_rdata_reset(&rdata);
    404      1.1  christos 	}
    405  1.1.1.4  christos 	if (result != ISC_R_NOMORE) {
    406      1.1  christos 		goto rollback;
    407  1.1.1.4  christos 	}
    408      1.1  christos 
    409      1.1  christos 	*countp = count;
    410      1.1  christos 
    411      1.1  christos 	return (ISC_R_SUCCESS);
    412      1.1  christos 
    413  1.1.1.4  christos rollback:
    414      1.1  christos 	INSIST(savedbuffer.used < 65536);
    415  1.1.1.2  christos 	dns_compress_rollback(cctx, (uint16_t)savedbuffer.used);
    416      1.1  christos 	*countp = 0;
    417      1.1  christos 	*target = savedbuffer;
    418      1.1  christos 
    419      1.1  christos 	return (result);
    420      1.1  christos }
    421      1.1  christos 
    422      1.1  christos static void
    423      1.1  christos rdataset_disassociate(dns_rdataset_t *rdataset) {
    424      1.1  christos 	UNUSED(rdataset);
    425      1.1  christos }
    426      1.1  christos 
    427      1.1  christos static isc_result_t
    428      1.1  christos rdataset_first(dns_rdataset_t *rdataset) {
    429      1.1  christos 	unsigned char *raw = rdataset->private3;
    430      1.1  christos 	unsigned int count;
    431      1.1  christos 
    432      1.1  christos 	count = raw[0] * 256 + raw[1];
    433      1.1  christos 	if (count == 0) {
    434      1.1  christos 		rdataset->private5 = NULL;
    435      1.1  christos 		return (ISC_R_NOMORE);
    436      1.1  christos 	}
    437      1.1  christos 	raw += 2;
    438      1.1  christos 	/*
    439      1.1  christos 	 * The privateuint4 field is the number of rdata beyond the cursor
    440      1.1  christos 	 * position, so we decrement the total count by one before storing
    441      1.1  christos 	 * it.
    442      1.1  christos 	 */
    443      1.1  christos 	count--;
    444      1.1  christos 	rdataset->privateuint4 = count;
    445      1.1  christos 	rdataset->private5 = raw;
    446      1.1  christos 
    447      1.1  christos 	return (ISC_R_SUCCESS);
    448      1.1  christos }
    449      1.1  christos 
    450      1.1  christos static isc_result_t
    451      1.1  christos rdataset_next(dns_rdataset_t *rdataset) {
    452      1.1  christos 	unsigned int count;
    453      1.1  christos 	unsigned int length;
    454      1.1  christos 	unsigned char *raw;
    455      1.1  christos 
    456      1.1  christos 	count = rdataset->privateuint4;
    457  1.1.1.4  christos 	if (count == 0) {
    458      1.1  christos 		return (ISC_R_NOMORE);
    459  1.1.1.4  christos 	}
    460      1.1  christos 	count--;
    461      1.1  christos 	rdataset->privateuint4 = count;
    462      1.1  christos 	raw = rdataset->private5;
    463      1.1  christos 	length = raw[0] * 256 + raw[1];
    464      1.1  christos 	raw += length + 2;
    465      1.1  christos 	rdataset->private5 = raw;
    466      1.1  christos 
    467      1.1  christos 	return (ISC_R_SUCCESS);
    468      1.1  christos }
    469      1.1  christos 
    470      1.1  christos static void
    471      1.1  christos rdataset_current(dns_rdataset_t *rdataset, dns_rdata_t *rdata) {
    472      1.1  christos 	unsigned char *raw = rdataset->private5;
    473      1.1  christos 	isc_region_t r;
    474      1.1  christos 
    475      1.1  christos 	REQUIRE(raw != NULL);
    476      1.1  christos 
    477      1.1  christos 	r.length = raw[0] * 256 + raw[1];
    478      1.1  christos 	raw += 2;
    479      1.1  christos 	r.base = raw;
    480      1.1  christos 	dns_rdata_fromregion(rdata, rdataset->rdclass, rdataset->type, &r);
    481      1.1  christos }
    482      1.1  christos 
    483      1.1  christos static void
    484      1.1  christos rdataset_clone(dns_rdataset_t *source, dns_rdataset_t *target) {
    485      1.1  christos 	*target = *source;
    486      1.1  christos 
    487      1.1  christos 	/*
    488      1.1  christos 	 * Reset iterator state.
    489      1.1  christos 	 */
    490      1.1  christos 	target->privateuint4 = 0;
    491      1.1  christos 	target->private5 = NULL;
    492      1.1  christos }
    493      1.1  christos 
    494      1.1  christos static unsigned int
    495      1.1  christos rdataset_count(dns_rdataset_t *rdataset) {
    496      1.1  christos 	unsigned char *raw = rdataset->private3;
    497      1.1  christos 	unsigned int count;
    498      1.1  christos 
    499      1.1  christos 	count = raw[0] * 256 + raw[1];
    500      1.1  christos 
    501      1.1  christos 	return (count);
    502      1.1  christos }
    503      1.1  christos 
    504      1.1  christos static void
    505      1.1  christos rdataset_settrust(dns_rdataset_t *rdataset, dns_trust_t trust) {
    506      1.1  christos 	unsigned char *raw = rdataset->private3;
    507      1.1  christos 
    508      1.1  christos 	raw[-1] = (unsigned char)trust;
    509  1.1.1.6  christos 	rdataset->trust = trust;
    510      1.1  christos }
    511      1.1  christos 
    512      1.1  christos static dns_rdatasetmethods_t rdataset_methods = {
    513      1.1  christos 	rdataset_disassociate,
    514      1.1  christos 	rdataset_first,
    515      1.1  christos 	rdataset_next,
    516      1.1  christos 	rdataset_current,
    517      1.1  christos 	rdataset_clone,
    518      1.1  christos 	rdataset_count,
    519  1.1.1.4  christos 	NULL,		   /* addnoqname */
    520  1.1.1.4  christos 	NULL,		   /* getnoqname */
    521  1.1.1.4  christos 	NULL,		   /* addclosest */
    522  1.1.1.4  christos 	NULL,		   /* getclosest */
    523  1.1.1.4  christos 	rdataset_settrust, /* settrust */
    524  1.1.1.4  christos 	NULL,		   /* expire */
    525  1.1.1.4  christos 	NULL,		   /* clearprefetch */
    526  1.1.1.4  christos 	NULL,		   /* setownercase */
    527  1.1.1.4  christos 	NULL,		   /* getownercase */
    528  1.1.1.4  christos 	NULL		   /* addglue */
    529      1.1  christos };
    530      1.1  christos 
    531      1.1  christos isc_result_t
    532      1.1  christos dns_ncache_getrdataset(dns_rdataset_t *ncacherdataset, dns_name_t *name,
    533  1.1.1.4  christos 		       dns_rdatatype_t type, dns_rdataset_t *rdataset) {
    534      1.1  christos 	isc_result_t result;
    535      1.1  christos 	dns_rdata_t rdata = DNS_RDATA_INIT;
    536      1.1  christos 	isc_region_t remaining;
    537      1.1  christos 	isc_buffer_t source;
    538      1.1  christos 	dns_name_t tname;
    539      1.1  christos 	dns_rdatatype_t ttype;
    540      1.1  christos 	dns_trust_t trust = dns_trust_none;
    541      1.1  christos 	dns_rdataset_t rclone;
    542      1.1  christos 
    543      1.1  christos 	REQUIRE(ncacherdataset != NULL);
    544      1.1  christos 	REQUIRE(ncacherdataset->type == 0);
    545      1.1  christos 	REQUIRE((ncacherdataset->attributes & DNS_RDATASETATTR_NEGATIVE) != 0);
    546      1.1  christos 	REQUIRE(name != NULL);
    547      1.1  christos 	REQUIRE(!dns_rdataset_isassociated(rdataset));
    548      1.1  christos 	REQUIRE(type != dns_rdatatype_rrsig);
    549      1.1  christos 
    550      1.1  christos 	dns_rdataset_init(&rclone);
    551      1.1  christos 	dns_rdataset_clone(ncacherdataset, &rclone);
    552      1.1  christos 	result = dns_rdataset_first(&rclone);
    553      1.1  christos 	while (result == ISC_R_SUCCESS) {
    554      1.1  christos 		dns_rdataset_current(&rclone, &rdata);
    555      1.1  christos 		isc_buffer_init(&source, rdata.data, rdata.length);
    556      1.1  christos 		isc_buffer_add(&source, rdata.length);
    557      1.1  christos 		dns_name_init(&tname, NULL);
    558      1.1  christos 		isc_buffer_remainingregion(&source, &remaining);
    559      1.1  christos 		dns_name_fromregion(&tname, &remaining);
    560      1.1  christos 		INSIST(remaining.length >= tname.length);
    561      1.1  christos 		isc_buffer_forward(&source, tname.length);
    562      1.1  christos 		remaining.length -= tname.length;
    563      1.1  christos 
    564      1.1  christos 		INSIST(remaining.length >= 3);
    565      1.1  christos 		ttype = isc_buffer_getuint16(&source);
    566      1.1  christos 
    567      1.1  christos 		if (ttype == type && dns_name_equal(&tname, name)) {
    568      1.1  christos 			trust = isc_buffer_getuint8(&source);
    569      1.1  christos 			INSIST(trust <= dns_trust_ultimate);
    570      1.1  christos 			isc_buffer_remainingregion(&source, &remaining);
    571      1.1  christos 			break;
    572      1.1  christos 		}
    573      1.1  christos 		result = dns_rdataset_next(&rclone);
    574      1.1  christos 		dns_rdata_reset(&rdata);
    575      1.1  christos 	}
    576      1.1  christos 	dns_rdataset_disassociate(&rclone);
    577  1.1.1.4  christos 	if (result == ISC_R_NOMORE) {
    578      1.1  christos 		return (ISC_R_NOTFOUND);
    579  1.1.1.4  christos 	}
    580  1.1.1.4  christos 	if (result != ISC_R_SUCCESS) {
    581      1.1  christos 		return (result);
    582  1.1.1.4  christos 	}
    583      1.1  christos 
    584      1.1  christos 	INSIST(remaining.length != 0);
    585      1.1  christos 
    586      1.1  christos 	rdataset->methods = &rdataset_methods;
    587      1.1  christos 	rdataset->rdclass = ncacherdataset->rdclass;
    588      1.1  christos 	rdataset->type = type;
    589      1.1  christos 	rdataset->covers = 0;
    590      1.1  christos 	rdataset->ttl = ncacherdataset->ttl;
    591      1.1  christos 	rdataset->trust = trust;
    592      1.1  christos 	rdataset->private1 = NULL;
    593      1.1  christos 	rdataset->private2 = NULL;
    594      1.1  christos 
    595      1.1  christos 	rdataset->private3 = remaining.base;
    596      1.1  christos 
    597      1.1  christos 	/*
    598      1.1  christos 	 * Reset iterator state.
    599      1.1  christos 	 */
    600      1.1  christos 	rdataset->privateuint4 = 0;
    601      1.1  christos 	rdataset->private5 = NULL;
    602      1.1  christos 	rdataset->private6 = NULL;
    603      1.1  christos 	return (ISC_R_SUCCESS);
    604      1.1  christos }
    605      1.1  christos 
    606      1.1  christos isc_result_t
    607      1.1  christos dns_ncache_getsigrdataset(dns_rdataset_t *ncacherdataset, dns_name_t *name,
    608  1.1.1.4  christos 			  dns_rdatatype_t covers, dns_rdataset_t *rdataset) {
    609      1.1  christos 	dns_name_t tname;
    610      1.1  christos 	dns_rdata_rrsig_t rrsig;
    611      1.1  christos 	dns_rdata_t rdata = DNS_RDATA_INIT;
    612      1.1  christos 	dns_rdataset_t rclone;
    613      1.1  christos 	dns_rdatatype_t type;
    614      1.1  christos 	dns_trust_t trust = dns_trust_none;
    615      1.1  christos 	isc_buffer_t source;
    616      1.1  christos 	isc_region_t remaining, sigregion;
    617      1.1  christos 	isc_result_t result;
    618      1.1  christos 	unsigned char *raw;
    619      1.1  christos 	unsigned int count;
    620      1.1  christos 
    621      1.1  christos 	REQUIRE(ncacherdataset != NULL);
    622      1.1  christos 	REQUIRE(ncacherdataset->type == 0);
    623      1.1  christos 	REQUIRE((ncacherdataset->attributes & DNS_RDATASETATTR_NEGATIVE) != 0);
    624      1.1  christos 	REQUIRE(name != NULL);
    625      1.1  christos 	REQUIRE(!dns_rdataset_isassociated(rdataset));
    626      1.1  christos 
    627      1.1  christos 	dns_rdataset_init(&rclone);
    628      1.1  christos 	dns_rdataset_clone(ncacherdataset, &rclone);
    629      1.1  christos 	result = dns_rdataset_first(&rclone);
    630      1.1  christos 	while (result == ISC_R_SUCCESS) {
    631      1.1  christos 		dns_rdataset_current(&rclone, &rdata);
    632      1.1  christos 		isc_buffer_init(&source, rdata.data, rdata.length);
    633      1.1  christos 		isc_buffer_add(&source, rdata.length);
    634      1.1  christos 		dns_name_init(&tname, NULL);
    635      1.1  christos 		isc_buffer_remainingregion(&source, &remaining);
    636      1.1  christos 		dns_name_fromregion(&tname, &remaining);
    637      1.1  christos 		INSIST(remaining.length >= tname.length);
    638      1.1  christos 		isc_buffer_forward(&source, tname.length);
    639      1.1  christos 		isc_region_consume(&remaining, tname.length);
    640      1.1  christos 
    641      1.1  christos 		INSIST(remaining.length >= 2);
    642      1.1  christos 		type = isc_buffer_getuint16(&source);
    643      1.1  christos 		isc_region_consume(&remaining, 2);
    644      1.1  christos 
    645      1.1  christos 		if (type != dns_rdatatype_rrsig ||
    646      1.1  christos 		    !dns_name_equal(&tname, name)) {
    647      1.1  christos 			result = dns_rdataset_next(&rclone);
    648      1.1  christos 			dns_rdata_reset(&rdata);
    649      1.1  christos 			continue;
    650      1.1  christos 		}
    651      1.1  christos 
    652      1.1  christos 		INSIST(remaining.length >= 1);
    653      1.1  christos 		trust = isc_buffer_getuint8(&source);
    654      1.1  christos 		INSIST(trust <= dns_trust_ultimate);
    655      1.1  christos 		isc_region_consume(&remaining, 1);
    656      1.1  christos 
    657      1.1  christos 		raw = remaining.base;
    658      1.1  christos 		count = raw[0] * 256 + raw[1];
    659      1.1  christos 		INSIST(count > 0);
    660      1.1  christos 		raw += 2;
    661      1.1  christos 		sigregion.length = raw[0] * 256 + raw[1];
    662      1.1  christos 		raw += 2;
    663      1.1  christos 		sigregion.base = raw;
    664      1.1  christos 		dns_rdata_reset(&rdata);
    665      1.1  christos 		dns_rdata_fromregion(&rdata, rdataset->rdclass,
    666      1.1  christos 				     dns_rdatatype_rrsig, &sigregion);
    667      1.1  christos 		(void)dns_rdata_tostruct(&rdata, &rrsig, NULL);
    668      1.1  christos 		if (rrsig.covered == covers) {
    669      1.1  christos 			isc_buffer_remainingregion(&source, &remaining);
    670      1.1  christos 			break;
    671      1.1  christos 		}
    672      1.1  christos 
    673      1.1  christos 		result = dns_rdataset_next(&rclone);
    674      1.1  christos 		dns_rdata_reset(&rdata);
    675      1.1  christos 	}
    676      1.1  christos 	dns_rdataset_disassociate(&rclone);
    677  1.1.1.4  christos 	if (result == ISC_R_NOMORE) {
    678      1.1  christos 		return (ISC_R_NOTFOUND);
    679  1.1.1.4  christos 	}
    680  1.1.1.4  christos 	if (result != ISC_R_SUCCESS) {
    681      1.1  christos 		return (result);
    682  1.1.1.4  christos 	}
    683      1.1  christos 
    684      1.1  christos 	INSIST(remaining.length != 0);
    685      1.1  christos 
    686      1.1  christos 	rdataset->methods = &rdataset_methods;
    687      1.1  christos 	rdataset->rdclass = ncacherdataset->rdclass;
    688      1.1  christos 	rdataset->type = dns_rdatatype_rrsig;
    689      1.1  christos 	rdataset->covers = covers;
    690      1.1  christos 	rdataset->ttl = ncacherdataset->ttl;
    691      1.1  christos 	rdataset->trust = trust;
    692      1.1  christos 	rdataset->private1 = NULL;
    693      1.1  christos 	rdataset->private2 = NULL;
    694      1.1  christos 
    695      1.1  christos 	rdataset->private3 = remaining.base;
    696      1.1  christos 
    697      1.1  christos 	/*
    698      1.1  christos 	 * Reset iterator state.
    699      1.1  christos 	 */
    700      1.1  christos 	rdataset->privateuint4 = 0;
    701      1.1  christos 	rdataset->private5 = NULL;
    702      1.1  christos 	rdataset->private6 = NULL;
    703      1.1  christos 	return (ISC_R_SUCCESS);
    704      1.1  christos }
    705      1.1  christos 
    706      1.1  christos void
    707      1.1  christos dns_ncache_current(dns_rdataset_t *ncacherdataset, dns_name_t *found,
    708  1.1.1.4  christos 		   dns_rdataset_t *rdataset) {
    709      1.1  christos 	dns_rdata_t rdata = DNS_RDATA_INIT;
    710      1.1  christos 	dns_trust_t trust;
    711      1.1  christos 	isc_region_t remaining, sigregion;
    712      1.1  christos 	isc_buffer_t source;
    713      1.1  christos 	dns_name_t tname;
    714      1.1  christos 	dns_rdatatype_t type;
    715      1.1  christos 	unsigned int count;
    716      1.1  christos 	dns_rdata_rrsig_t rrsig;
    717      1.1  christos 	unsigned char *raw;
    718      1.1  christos 
    719      1.1  christos 	REQUIRE(ncacherdataset != NULL);
    720      1.1  christos 	REQUIRE(ncacherdataset->type == 0);
    721      1.1  christos 	REQUIRE((ncacherdataset->attributes & DNS_RDATASETATTR_NEGATIVE) != 0);
    722      1.1  christos 	REQUIRE(found != NULL);
    723      1.1  christos 	REQUIRE(!dns_rdataset_isassociated(rdataset));
    724      1.1  christos 
    725      1.1  christos 	dns_rdataset_current(ncacherdataset, &rdata);
    726      1.1  christos 	isc_buffer_init(&source, rdata.data, rdata.length);
    727      1.1  christos 	isc_buffer_add(&source, rdata.length);
    728      1.1  christos 
    729      1.1  christos 	dns_name_init(&tname, NULL);
    730      1.1  christos 	isc_buffer_remainingregion(&source, &remaining);
    731      1.1  christos 	dns_name_fromregion(found, &remaining);
    732      1.1  christos 	INSIST(remaining.length >= found->length);
    733      1.1  christos 	isc_buffer_forward(&source, found->length);
    734      1.1  christos 	remaining.length -= found->length;
    735      1.1  christos 
    736      1.1  christos 	INSIST(remaining.length >= 5);
    737      1.1  christos 	type = isc_buffer_getuint16(&source);
    738      1.1  christos 	trust = isc_buffer_getuint8(&source);
    739      1.1  christos 	INSIST(trust <= dns_trust_ultimate);
    740      1.1  christos 	isc_buffer_remainingregion(&source, &remaining);
    741      1.1  christos 
    742      1.1  christos 	rdataset->methods = &rdataset_methods;
    743      1.1  christos 	rdataset->rdclass = ncacherdataset->rdclass;
    744      1.1  christos 	rdataset->type = type;
    745      1.1  christos 	if (type == dns_rdatatype_rrsig) {
    746      1.1  christos 		/*
    747      1.1  christos 		 * Extract covers from RRSIG.
    748      1.1  christos 		 */
    749      1.1  christos 		raw = remaining.base;
    750      1.1  christos 		count = raw[0] * 256 + raw[1];
    751      1.1  christos 		INSIST(count > 0);
    752      1.1  christos 		raw += 2;
    753      1.1  christos 		sigregion.length = raw[0] * 256 + raw[1];
    754      1.1  christos 		raw += 2;
    755      1.1  christos 		sigregion.base = raw;
    756      1.1  christos 		dns_rdata_reset(&rdata);
    757  1.1.1.4  christos 		dns_rdata_fromregion(&rdata, rdataset->rdclass, rdataset->type,
    758  1.1.1.4  christos 				     &sigregion);
    759      1.1  christos 		(void)dns_rdata_tostruct(&rdata, &rrsig, NULL);
    760      1.1  christos 		rdataset->covers = rrsig.covered;
    761  1.1.1.4  christos 	} else {
    762      1.1  christos 		rdataset->covers = 0;
    763  1.1.1.4  christos 	}
    764      1.1  christos 	rdataset->ttl = ncacherdataset->ttl;
    765      1.1  christos 	rdataset->trust = trust;
    766      1.1  christos 	rdataset->private1 = NULL;
    767      1.1  christos 	rdataset->private2 = NULL;
    768      1.1  christos 
    769      1.1  christos 	rdataset->private3 = remaining.base;
    770      1.1  christos 
    771      1.1  christos 	/*
    772      1.1  christos 	 * Reset iterator state.
    773      1.1  christos 	 */
    774      1.1  christos 	rdataset->privateuint4 = 0;
    775      1.1  christos 	rdataset->private5 = NULL;
    776      1.1  christos 	rdataset->private6 = NULL;
    777      1.1  christos }
    778