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