1 /*
2 * services/localzone.c - local zones authority service.
3 *
4 * Copyright (c) 2007, NLnet Labs. All rights reserved.
5 *
6 * This software is open source.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 *
12 * Redistributions of source code must retain the above copyright notice,
13 * this list of conditions and the following disclaimer.
14 *
15 * Redistributions in binary form must reproduce the above copyright notice,
16 * this list of conditions and the following disclaimer in the documentation
17 * and/or other materials provided with the distribution.
18 *
19 * Neither the name of the NLNET LABS nor the names of its contributors may
20 * be used to endorse or promote products derived from this software without
21 * specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
29 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
30 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
31 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
32 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
33 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 */
35
36 /**
37 * \file
38 *
39 * This file contains functions to enable local zone authority service.
40 */
41 #include "config.h"
42 #include "services/localzone.h"
43 #include "sldns/str2wire.h"
44 #include "util/regional.h"
45 #include "util/config_file.h"
46 #include "util/data/dname.h"
47 #include "util/data/packed_rrset.h"
48 #include "util/data/msgencode.h"
49 #include "util/net_help.h"
50 #include "util/netevent.h"
51 #include "util/data/msgreply.h"
52 #include "util/data/msgparse.h"
53 #include "util/as112.h"
54
55 /* maximum RRs in an RRset, to cap possible 'endless' list RRs.
56 * with 16 bytes for an A record, a 64K packet has about 4000 max */
57 #define LOCALZONE_RRSET_COUNT_MAX 4096
58
59 /** print all RRsets in local zone */
60 static void
local_zone_out(struct local_zone * z)61 local_zone_out(struct local_zone* z)
62 {
63 struct local_data* d;
64 struct local_rrset* p;
65 RBTREE_FOR(d, struct local_data*, &z->data) {
66 for(p = d->rrsets; p; p = p->next) {
67 log_nametypeclass(NO_VERBOSE, "rrset", d->name,
68 ntohs(p->rrset->rk.type),
69 ntohs(p->rrset->rk.rrset_class));
70 }
71 }
72 }
73
74 static void
local_zone_print(struct local_zone * z)75 local_zone_print(struct local_zone* z)
76 {
77 char buf[64];
78 lock_rw_rdlock(&z->lock);
79 snprintf(buf, sizeof(buf), "%s zone",
80 local_zone_type2str(z->type));
81 log_nametypeclass(NO_VERBOSE, buf, z->name, 0, z->dclass);
82 local_zone_out(z);
83 lock_rw_unlock(&z->lock);
84 }
85
local_zones_print(struct local_zones * zones)86 void local_zones_print(struct local_zones* zones)
87 {
88 struct local_zone* z;
89 lock_rw_rdlock(&zones->lock);
90 log_info("number of auth zones %u", (unsigned)zones->ztree.count);
91 RBTREE_FOR(z, struct local_zone*, &zones->ztree) {
92 local_zone_print(z);
93 }
94 lock_rw_unlock(&zones->lock);
95 }
96
97 struct local_zones*
local_zones_create(void)98 local_zones_create(void)
99 {
100 struct local_zones* zones = (struct local_zones*)calloc(1,
101 sizeof(*zones));
102 if(!zones)
103 return NULL;
104 rbtree_init(&zones->ztree, &local_zone_cmp);
105 lock_rw_init(&zones->lock);
106 lock_protect(&zones->lock, &zones->ztree, sizeof(zones->ztree));
107 /* also lock protects the rbnode's in struct local_zone */
108 return zones;
109 }
110
111 /** helper traverse to delete zones */
112 static void
lzdel(rbnode_type * n,void * ATTR_UNUSED (arg))113 lzdel(rbnode_type* n, void* ATTR_UNUSED(arg))
114 {
115 struct local_zone* z = (struct local_zone*)n->key;
116 local_zone_delete(z);
117 }
118
119 void
local_zones_delete(struct local_zones * zones)120 local_zones_delete(struct local_zones* zones)
121 {
122 if(!zones)
123 return;
124 lock_rw_destroy(&zones->lock);
125 /* walk through zones and delete them all */
126 traverse_postorder(&zones->ztree, lzdel, NULL);
127 free(zones);
128 }
129
130 void
local_zone_delete(struct local_zone * z)131 local_zone_delete(struct local_zone* z)
132 {
133 if(!z)
134 return;
135 lock_rw_destroy(&z->lock);
136 regional_destroy(z->region);
137 free(z->name);
138 free(z->taglist);
139 free(z);
140 }
141
142 int
local_zone_cmp(const void * z1,const void * z2)143 local_zone_cmp(const void* z1, const void* z2)
144 {
145 /* first sort on class, so that hierarchy can be maintained within
146 * a class */
147 struct local_zone* a = (struct local_zone*)z1;
148 struct local_zone* b = (struct local_zone*)z2;
149 int m;
150 if(a->dclass != b->dclass) {
151 if(a->dclass < b->dclass)
152 return -1;
153 return 1;
154 }
155 return dname_lab_cmp(a->name, a->namelabs, b->name, b->namelabs, &m);
156 }
157
158 int
local_data_cmp(const void * d1,const void * d2)159 local_data_cmp(const void* d1, const void* d2)
160 {
161 struct local_data* a = (struct local_data*)d1;
162 struct local_data* b = (struct local_data*)d2;
163 int m;
164 return dname_canon_lab_cmp(a->name, a->namelabs, b->name,
165 b->namelabs, &m);
166 }
167
168 /* form wireformat from text format domain name */
169 int
parse_dname(const char * str,uint8_t ** res,size_t * len,int * labs)170 parse_dname(const char* str, uint8_t** res, size_t* len, int* labs)
171 {
172 *res = sldns_str2wire_dname(str, len);
173 *labs = 0;
174 if(!*res) {
175 log_err("cannot parse name %s", str);
176 return 0;
177 }
178 *labs = dname_count_size_labels(*res, len);
179 return 1;
180 }
181
182 /** create a new localzone */
183 static struct local_zone*
local_zone_create(uint8_t * nm,size_t len,int labs,enum localzone_type t,uint16_t dclass)184 local_zone_create(uint8_t* nm, size_t len, int labs,
185 enum localzone_type t, uint16_t dclass)
186 {
187 struct local_zone* z = (struct local_zone*)calloc(1, sizeof(*z));
188 if(!z) {
189 return NULL;
190 }
191 z->node.key = z;
192 z->dclass = dclass;
193 z->type = t;
194 z->name = nm;
195 z->namelen = len;
196 z->namelabs = labs;
197 lock_rw_init(&z->lock);
198 z->region = regional_create_nochunk(sizeof(struct regional));
199 if(!z->region) {
200 free(z);
201 return NULL;
202 }
203 rbtree_init(&z->data, &local_data_cmp);
204 lock_protect(&z->lock, &z->parent, sizeof(*z)-sizeof(rbnode_type));
205 /* also the zones->lock protects node, parent, name*, class */
206 return z;
207 }
208
209 /** enter a new zone with allocated dname returns with WRlock */
210 static struct local_zone*
lz_enter_zone_dname(struct local_zones * zones,uint8_t * nm,size_t len,int labs,enum localzone_type t,uint16_t c)211 lz_enter_zone_dname(struct local_zones* zones, uint8_t* nm, size_t len,
212 int labs, enum localzone_type t, uint16_t c)
213 {
214 struct local_zone* z = local_zone_create(nm, len, labs, t, c);
215 if(!z) {
216 free(nm);
217 log_err("out of memory");
218 return NULL;
219 }
220
221 /* add to rbtree */
222 lock_rw_wrlock(&zones->lock);
223 lock_rw_wrlock(&z->lock);
224 if(!rbtree_insert(&zones->ztree, &z->node)) {
225 struct local_zone* oldz;
226 char str[LDNS_MAX_DOMAINLEN];
227 dname_str(nm, str);
228 log_warn("duplicate local-zone %s", str);
229 lock_rw_unlock(&z->lock);
230 /* save zone name locally before deallocation,
231 * otherwise, nm is gone if we zone_delete now. */
232 oldz = z;
233 /* find the correct zone, so not an error for duplicate */
234 z = local_zones_find(zones, nm, len, labs, c);
235 lock_rw_wrlock(&z->lock);
236 lock_rw_unlock(&zones->lock);
237 local_zone_delete(oldz);
238 return z;
239 }
240 lock_rw_unlock(&zones->lock);
241 return z;
242 }
243
244 /** enter a new zone */
245 struct local_zone*
lz_enter_zone(struct local_zones * zones,const char * name,const char * type,uint16_t dclass)246 lz_enter_zone(struct local_zones* zones, const char* name, const char* type,
247 uint16_t dclass)
248 {
249 struct local_zone* z;
250 enum localzone_type t;
251 uint8_t* nm;
252 size_t len;
253 int labs;
254 if(!parse_dname(name, &nm, &len, &labs)) {
255 log_err("bad zone name %s %s", name, type);
256 return NULL;
257 }
258 if(!local_zone_str2type(type, &t)) {
259 log_err("bad lz_enter_zone type %s %s", name, type);
260 free(nm);
261 return NULL;
262 }
263 if(!(z=lz_enter_zone_dname(zones, nm, len, labs, t, dclass))) {
264 log_err("could not enter zone %s %s", name, type);
265 return NULL;
266 }
267 return z;
268 }
269
270 int
rrstr_get_rr_content(const char * str,uint8_t ** nm,uint16_t * type,uint16_t * dclass,time_t * ttl,uint8_t * rr,size_t len,uint8_t ** rdata,size_t * rdata_len)271 rrstr_get_rr_content(const char* str, uint8_t** nm, uint16_t* type,
272 uint16_t* dclass, time_t* ttl, uint8_t* rr, size_t len,
273 uint8_t** rdata, size_t* rdata_len)
274 {
275 size_t dname_len = 0;
276 int e = sldns_str2wire_rr_buf(str, rr, &len, &dname_len, 3600,
277 NULL, 0, NULL, 0);
278 if(e) {
279 log_err("error parsing local-data at %d: '%s': %s",
280 LDNS_WIREPARSE_OFFSET(e), str,
281 sldns_get_errorstr_parse(e));
282 return 0;
283 }
284 *nm = memdup(rr, dname_len);
285 if(!*nm) {
286 log_err("out of memory");
287 return 0;
288 }
289 *dclass = sldns_wirerr_get_class(rr, len, dname_len);
290 *type = sldns_wirerr_get_type(rr, len, dname_len);
291 *ttl = (time_t)sldns_wirerr_get_ttl(rr, len, dname_len);
292 *rdata = sldns_wirerr_get_rdatawl(rr, len, dname_len);
293 *rdata_len = sldns_wirerr_get_rdatalen(rr, len, dname_len)+2;
294 return 1;
295 }
296
297 /** return name and class of rr; parses string */
298 static int
get_rr_nameclass(const char * str,uint8_t ** nm,uint16_t * dclass,uint16_t * dtype)299 get_rr_nameclass(const char* str, uint8_t** nm, uint16_t* dclass,
300 uint16_t* dtype)
301 {
302 uint8_t rr[LDNS_RR_BUF_SIZE];
303 size_t len = sizeof(rr), dname_len = 0;
304 int s = sldns_str2wire_rr_buf(str, rr, &len, &dname_len, 3600,
305 NULL, 0, NULL, 0);
306 if(s != 0) {
307 log_err("error parsing local-data at %d '%s': %s",
308 LDNS_WIREPARSE_OFFSET(s), str,
309 sldns_get_errorstr_parse(s));
310 return 0;
311 }
312 *nm = memdup(rr, dname_len);
313 *dclass = sldns_wirerr_get_class(rr, len, dname_len);
314 *dtype = sldns_wirerr_get_type(rr, len, dname_len);
315 if(!*nm) {
316 log_err("out of memory");
317 return 0;
318 }
319 return 1;
320 }
321
322 /**
323 * Find an rrset in local data structure.
324 * @param data: local data domain name structure.
325 * @param type: type to look for (host order).
326 * @param alias_ok: 1 if matching a non-exact, alias type such as CNAME is
327 * allowed. otherwise 0.
328 * @return rrset pointer or NULL if not found.
329 */
330 static struct local_rrset*
local_data_find_type(struct local_data * data,uint16_t type,int alias_ok)331 local_data_find_type(struct local_data* data, uint16_t type, int alias_ok)
332 {
333 struct local_rrset* p, *cname = NULL;
334 type = htons(type);
335 for(p = data->rrsets; p; p = p->next) {
336 if(p->rrset->rk.type == type)
337 return p;
338 if(alias_ok && p->rrset->rk.type == htons(LDNS_RR_TYPE_CNAME))
339 cname = p;
340 }
341 if(alias_ok)
342 return cname;
343 return NULL;
344 }
345
346 /** check for RR duplicates */
347 static int
rr_is_duplicate(struct packed_rrset_data * pd,uint8_t * rdata,size_t rdata_len)348 rr_is_duplicate(struct packed_rrset_data* pd, uint8_t* rdata, size_t rdata_len)
349 {
350 size_t i;
351 for(i=0; i<pd->count; i++) {
352 if(pd->rr_len[i] == rdata_len &&
353 memcmp(pd->rr_data[i], rdata, rdata_len) == 0)
354 return 1;
355 }
356 return 0;
357 }
358
359 /** new local_rrset */
360 static struct local_rrset*
new_local_rrset(struct regional * region,struct local_data * node,uint16_t rrtype,uint16_t rrclass)361 new_local_rrset(struct regional* region, struct local_data* node,
362 uint16_t rrtype, uint16_t rrclass)
363 {
364 struct packed_rrset_data* pd;
365 struct local_rrset* rrset = (struct local_rrset*)
366 regional_alloc_zero(region, sizeof(*rrset));
367 if(!rrset) {
368 log_err("out of memory");
369 return NULL;
370 }
371 rrset->next = node->rrsets;
372 node->rrsets = rrset;
373 rrset->rrset = (struct ub_packed_rrset_key*)
374 regional_alloc_zero(region, sizeof(*rrset->rrset));
375 if(!rrset->rrset) {
376 log_err("out of memory");
377 return NULL;
378 }
379 rrset->rrset->entry.key = rrset->rrset;
380 pd = (struct packed_rrset_data*)regional_alloc_zero(region,
381 sizeof(*pd));
382 if(!pd) {
383 log_err("out of memory");
384 return NULL;
385 }
386 pd->trust = rrset_trust_prim_noglue;
387 pd->security = sec_status_insecure;
388 rrset->rrset->entry.data = pd;
389 rrset->rrset->rk.dname = node->name;
390 rrset->rrset->rk.dname_len = node->namelen;
391 rrset->rrset->rk.type = htons(rrtype);
392 rrset->rrset->rk.rrset_class = htons(rrclass);
393 return rrset;
394 }
395
396 /** insert RR into RRset data structure; Wastes a couple of bytes */
397 int
rrset_insert_rr(struct regional * region,struct packed_rrset_data * pd,uint8_t * rdata,size_t rdata_len,time_t ttl,const char * rrstr)398 rrset_insert_rr(struct regional* region, struct packed_rrset_data* pd,
399 uint8_t* rdata, size_t rdata_len, time_t ttl, const char* rrstr)
400 {
401 size_t* oldlen = pd->rr_len;
402 time_t* oldttl = pd->rr_ttl;
403 uint8_t** olddata = pd->rr_data;
404
405 /* add RR to rrset */
406 if(pd->count > LOCALZONE_RRSET_COUNT_MAX) {
407 log_warn("RRset '%s' has more than %d records, record ignored",
408 rrstr, LOCALZONE_RRSET_COUNT_MAX);
409 return 1;
410 }
411 pd->count++;
412 pd->rr_len = regional_alloc(region, sizeof(*pd->rr_len)*pd->count);
413 pd->rr_ttl = regional_alloc(region, sizeof(*pd->rr_ttl)*pd->count);
414 pd->rr_data = regional_alloc(region, sizeof(*pd->rr_data)*pd->count);
415 if(!pd->rr_len || !pd->rr_ttl || !pd->rr_data) {
416 log_err("out of memory");
417 return 0;
418 }
419 if(pd->count > 1) {
420 memcpy(pd->rr_len+1, oldlen,
421 sizeof(*pd->rr_len)*(pd->count-1));
422 memcpy(pd->rr_ttl+1, oldttl,
423 sizeof(*pd->rr_ttl)*(pd->count-1));
424 memcpy(pd->rr_data+1, olddata,
425 sizeof(*pd->rr_data)*(pd->count-1));
426 }
427 pd->rr_len[0] = rdata_len;
428 pd->rr_ttl[0] = ttl;
429 pd->rr_data[0] = regional_alloc_init(region, rdata, rdata_len);
430 if(!pd->rr_data[0]) {
431 log_err("out of memory");
432 return 0;
433 }
434 return 1;
435 }
436
437 /** Delete RR from local-zone RRset, wastes memory as the deleted RRs cannot be
438 * free'd (regionally alloc'd) */
439 int
local_rrset_remove_rr(struct packed_rrset_data * pd,size_t index)440 local_rrset_remove_rr(struct packed_rrset_data* pd, size_t index)
441 {
442 log_assert(pd->count > 0);
443 if(index >= pd->count) {
444 log_warn("Trying to remove RR with out of bound index");
445 return 0;
446 }
447 if(index + 1 < pd->count) {
448 /* not removing last element */
449 size_t nexti = index + 1;
450 size_t num = pd->count - nexti;
451 memmove(pd->rr_len+index, pd->rr_len+nexti, sizeof(*pd->rr_len)*num);
452 memmove(pd->rr_ttl+index, pd->rr_ttl+nexti, sizeof(*pd->rr_ttl)*num);
453 memmove(pd->rr_data+index, pd->rr_data+nexti, sizeof(*pd->rr_data)*num);
454 }
455 pd->count--;
456 return 1;
457 }
458
459 struct local_data*
local_zone_find_data(struct local_zone * z,uint8_t * nm,size_t nmlen,int nmlabs)460 local_zone_find_data(struct local_zone* z, uint8_t* nm, size_t nmlen, int nmlabs)
461 {
462 struct local_data key;
463 key.node.key = &key;
464 key.name = nm;
465 key.namelen = nmlen;
466 key.namelabs = nmlabs;
467 return (struct local_data*)rbtree_search(&z->data, &key.node);
468 }
469
470 /** find a node, create it if not and all its empty nonterminal parents */
471 static int
lz_find_create_node(struct local_zone * z,uint8_t * nm,size_t nmlen,int nmlabs,struct local_data ** res)472 lz_find_create_node(struct local_zone* z, uint8_t* nm, size_t nmlen,
473 int nmlabs, struct local_data** res)
474 {
475 struct local_data* ld = local_zone_find_data(z, nm, nmlen, nmlabs);
476 if(!ld) {
477 /* create a domain name to store rr. */
478 ld = (struct local_data*)regional_alloc_zero(z->region,
479 sizeof(*ld));
480 if(!ld) {
481 log_err("out of memory adding local data");
482 return 0;
483 }
484 ld->node.key = ld;
485 ld->name = regional_alloc_init(z->region, nm, nmlen);
486 if(!ld->name) {
487 log_err("out of memory");
488 return 0;
489 }
490 ld->namelen = nmlen;
491 ld->namelabs = nmlabs;
492 if(!rbtree_insert(&z->data, &ld->node)) {
493 log_assert(0); /* duplicate name */
494 }
495 /* see if empty nonterminals need to be created */
496 if(nmlabs > z->namelabs) {
497 dname_remove_label(&nm, &nmlen);
498 if(!lz_find_create_node(z, nm, nmlen, nmlabs-1, res))
499 return 0;
500 }
501 }
502 *res = ld;
503 return 1;
504 }
505
506 /* Mark the SOA record for the zone. This only marks the SOA rrset; the data
507 * for the RR is entered later on local_zone_enter_rr() as with the other
508 * records. An artificial soa_negative record with a modified TTL (minimum of
509 * the TTL and the SOA.MINIMUM) is also created and marked for usage with
510 * negative answers and to avoid allocations during those answers. */
511 static int
lz_mark_soa_for_zone(struct local_zone * z,struct ub_packed_rrset_key * soa_rrset,uint8_t * rdata,size_t rdata_len,time_t ttl,const char * rrstr)512 lz_mark_soa_for_zone(struct local_zone* z, struct ub_packed_rrset_key* soa_rrset,
513 uint8_t* rdata, size_t rdata_len, time_t ttl, const char* rrstr)
514 {
515 struct packed_rrset_data* pd = (struct packed_rrset_data*)
516 regional_alloc_zero(z->region, sizeof(*pd));
517 struct ub_packed_rrset_key* rrset_negative = (struct ub_packed_rrset_key*)
518 regional_alloc_zero(z->region, sizeof(*rrset_negative));
519 time_t minimum;
520 if(!rrset_negative||!pd) {
521 log_err("out of memory");
522 return 0;
523 }
524 /* Mark the original SOA record and then continue with the negative one. */
525 z->soa = soa_rrset;
526 rrset_negative->entry.key = rrset_negative;
527 pd->trust = rrset_trust_prim_noglue;
528 pd->security = sec_status_insecure;
529 rrset_negative->entry.data = pd;
530 rrset_negative->rk.dname = soa_rrset->rk.dname;
531 rrset_negative->rk.dname_len = soa_rrset->rk.dname_len;
532 rrset_negative->rk.type = soa_rrset->rk.type;
533 rrset_negative->rk.rrset_class = soa_rrset->rk.rrset_class;
534 if(!rrset_insert_rr(z->region, pd, rdata, rdata_len, ttl, rrstr))
535 return 0;
536 /* last 4 bytes are minimum ttl in network format */
537 if(pd->count == 0 || pd->rr_len[0] < 2+4)
538 return 0;
539 minimum = (time_t)sldns_read_uint32(pd->rr_data[0]+(pd->rr_len[0]-4));
540 minimum = ttl<minimum?ttl:minimum;
541 pd->ttl = minimum;
542 pd->rr_ttl[0] = minimum;
543
544 z->soa_negative = rrset_negative;
545 return 1;
546 }
547
548 int
local_zone_enter_rr(struct local_zone * z,uint8_t * nm,size_t nmlen,int nmlabs,uint16_t rrtype,uint16_t rrclass,time_t ttl,uint8_t * rdata,size_t rdata_len,const char * rrstr)549 local_zone_enter_rr(struct local_zone* z, uint8_t* nm, size_t nmlen,
550 int nmlabs, uint16_t rrtype, uint16_t rrclass, time_t ttl,
551 uint8_t* rdata, size_t rdata_len, const char* rrstr)
552 {
553 struct local_data* node;
554 struct local_rrset* rrset;
555 struct packed_rrset_data* pd;
556
557 if(!lz_find_create_node(z, nm, nmlen, nmlabs, &node)) {
558 return 0;
559 }
560 log_assert(node);
561
562 /* Reject it if we would end up having CNAME and other data (including
563 * another CNAME) for a redirect zone. */
564 if((z->type == local_zone_redirect ||
565 z->type == local_zone_inform_redirect) && node->rrsets) {
566 const char* othertype = NULL;
567 if (rrtype == LDNS_RR_TYPE_CNAME)
568 othertype = "other";
569 else if (node->rrsets->rrset->rk.type ==
570 htons(LDNS_RR_TYPE_CNAME)) {
571 othertype = "CNAME";
572 }
573 if(othertype) {
574 log_err("local-data '%s' in redirect zone must not "
575 "coexist with %s local-data", rrstr, othertype);
576 return 0;
577 }
578 }
579 rrset = local_data_find_type(node, rrtype, 0);
580 if(!rrset) {
581 rrset = new_local_rrset(z->region, node, rrtype, rrclass);
582 if(!rrset)
583 return 0;
584 if(query_dname_compare(node->name, z->name) == 0) {
585 if(rrtype == LDNS_RR_TYPE_NSEC)
586 rrset->rrset->rk.flags = PACKED_RRSET_NSEC_AT_APEX;
587 if(rrtype == LDNS_RR_TYPE_SOA &&
588 !lz_mark_soa_for_zone(z, rrset->rrset, rdata, rdata_len, ttl,
589 rrstr))
590 return 0;
591 }
592 }
593 pd = (struct packed_rrset_data*)rrset->rrset->entry.data;
594 log_assert(rrset && pd);
595
596 /* check for duplicate RR */
597 if(rr_is_duplicate(pd, rdata, rdata_len)) {
598 verbose(VERB_ALGO, "ignoring duplicate RR: %s", rrstr);
599 return 1;
600 }
601 return rrset_insert_rr(z->region, pd, rdata, rdata_len, ttl, rrstr);
602 }
603
604 /** enter data RR into auth zone */
605 static int
lz_enter_rr_into_zone(struct local_zone * z,const char * rrstr)606 lz_enter_rr_into_zone(struct local_zone* z, const char* rrstr)
607 {
608 uint8_t* nm;
609 size_t nmlen;
610 int nmlabs, ret;
611 uint16_t rrtype = 0, rrclass = 0;
612 time_t ttl = 0;
613 uint8_t rr[LDNS_RR_BUF_SIZE];
614 uint8_t* rdata;
615 size_t rdata_len;
616 if(!rrstr_get_rr_content(rrstr, &nm, &rrtype, &rrclass, &ttl, rr,
617 sizeof(rr), &rdata, &rdata_len)) {
618 log_err("bad local-data: %s", rrstr);
619 return 0;
620 }
621 log_assert(z->dclass == rrclass);
622 if((z->type == local_zone_redirect ||
623 z->type == local_zone_inform_redirect) &&
624 query_dname_compare(z->name, nm) != 0) {
625 log_err("local-data in redirect zone must reside at top of zone"
626 ", not at %s", rrstr);
627 free(nm);
628 return 0;
629 }
630 nmlabs = dname_count_size_labels(nm, &nmlen);
631 ret = local_zone_enter_rr(z, nm, nmlen, nmlabs, rrtype, rrclass, ttl,
632 rdata, rdata_len, rrstr);
633 free(nm);
634 return ret;
635 }
636
637 /** enter a data RR into auth data; a zone for it must exist */
638 static int
lz_enter_rr_str(struct local_zones * zones,const char * rr)639 lz_enter_rr_str(struct local_zones* zones, const char* rr)
640 {
641 uint8_t* rr_name;
642 uint16_t rr_class, rr_type;
643 size_t len;
644 int labs;
645 struct local_zone* z;
646 int r;
647 if(!get_rr_nameclass(rr, &rr_name, &rr_class, &rr_type)) {
648 log_err("bad rr %s", rr);
649 return 0;
650 }
651 labs = dname_count_size_labels(rr_name, &len);
652 lock_rw_rdlock(&zones->lock);
653 z = local_zones_lookup(zones, rr_name, len, labs, rr_class, rr_type);
654 if(!z) {
655 lock_rw_unlock(&zones->lock);
656 fatal_exit("internal error: no zone for rr %s", rr);
657 }
658 lock_rw_wrlock(&z->lock);
659 lock_rw_unlock(&zones->lock);
660 free(rr_name);
661 r = lz_enter_rr_into_zone(z, rr);
662 lock_rw_unlock(&z->lock);
663 return r;
664 }
665
666 /** enter tagstring into zone */
667 static int
lz_enter_zone_tag(struct local_zones * zones,char * zname,uint8_t * list,size_t len,uint16_t rr_class)668 lz_enter_zone_tag(struct local_zones* zones, char* zname, uint8_t* list,
669 size_t len, uint16_t rr_class)
670 {
671 uint8_t dname[LDNS_MAX_DOMAINLEN+1];
672 size_t dname_len = sizeof(dname);
673 int dname_labs, r = 0;
674 struct local_zone* z;
675
676 if(sldns_str2wire_dname_buf(zname, dname, &dname_len) != 0) {
677 log_err("cannot parse zone name in local-zone-tag: %s", zname);
678 return 0;
679 }
680 dname_labs = dname_count_labels(dname);
681
682 lock_rw_rdlock(&zones->lock);
683 z = local_zones_find(zones, dname, dname_len, dname_labs, rr_class);
684 if(!z) {
685 lock_rw_unlock(&zones->lock);
686 log_err("no local-zone for tag %s", zname);
687 return 0;
688 }
689 lock_rw_wrlock(&z->lock);
690 lock_rw_unlock(&zones->lock);
691 free(z->taglist);
692 z->taglist = memdup(list, len);
693 z->taglen = len;
694 if(z->taglist)
695 r = 1;
696 lock_rw_unlock(&z->lock);
697 return r;
698 }
699
700 /** enter override into zone */
701 static int
lz_enter_override(struct local_zones * zones,char * zname,char * netblock,char * type,uint16_t rr_class)702 lz_enter_override(struct local_zones* zones, char* zname, char* netblock,
703 char* type, uint16_t rr_class)
704 {
705 uint8_t dname[LDNS_MAX_DOMAINLEN+1];
706 size_t dname_len = sizeof(dname);
707 int dname_labs;
708 struct sockaddr_storage addr;
709 int net;
710 socklen_t addrlen;
711 struct local_zone* z;
712 enum localzone_type t;
713
714 /* parse zone name */
715 if(sldns_str2wire_dname_buf(zname, dname, &dname_len) != 0) {
716 log_err("cannot parse zone name in local-zone-override: %s %s",
717 zname, netblock);
718 return 0;
719 }
720 dname_labs = dname_count_labels(dname);
721
722 /* parse netblock */
723 if(!netblockstrtoaddr(netblock, UNBOUND_DNS_PORT, &addr, &addrlen,
724 &net)) {
725 log_err("cannot parse netblock in local-zone-override: %s %s",
726 zname, netblock);
727 return 0;
728 }
729
730 /* parse zone type */
731 if(!local_zone_str2type(type, &t)) {
732 log_err("cannot parse type in local-zone-override: %s %s %s",
733 zname, netblock, type);
734 return 0;
735 }
736
737 /* find localzone entry */
738 lock_rw_rdlock(&zones->lock);
739 z = local_zones_find(zones, dname, dname_len, dname_labs, rr_class);
740 if(!z) {
741 lock_rw_unlock(&zones->lock);
742 log_err("no local-zone for local-zone-override %s", zname);
743 return 0;
744 }
745 lock_rw_wrlock(&z->lock);
746 lock_rw_unlock(&zones->lock);
747
748 /* create netblock addr_tree if not present yet */
749 if(!z->override_tree) {
750 z->override_tree = (struct rbtree_type*)regional_alloc_zero(
751 z->region, sizeof(*z->override_tree));
752 if(!z->override_tree) {
753 lock_rw_unlock(&z->lock);
754 log_err("out of memory");
755 return 0;
756 }
757 addr_tree_init(z->override_tree);
758 }
759 /* add new elem to tree */
760 if(z->override_tree) {
761 struct local_zone_override* n;
762 n = (struct local_zone_override*)regional_alloc_zero(
763 z->region, sizeof(*n));
764 if(!n) {
765 lock_rw_unlock(&z->lock);
766 log_err("out of memory");
767 return 0;
768 }
769 n->type = t;
770 if(!addr_tree_insert(z->override_tree,
771 (struct addr_tree_node*)n, &addr, addrlen, net)) {
772 lock_rw_unlock(&z->lock);
773 log_err("duplicate local-zone-override %s %s",
774 zname, netblock);
775 return 1;
776 }
777 }
778
779 lock_rw_unlock(&z->lock);
780 return 1;
781 }
782
783 /** parse local-zone: statements */
784 static int
lz_enter_zones(struct local_zones * zones,struct config_file * cfg)785 lz_enter_zones(struct local_zones* zones, struct config_file* cfg)
786 {
787 struct config_str2list* p;
788 #ifndef THREADS_DISABLED
789 struct local_zone* z;
790 #endif
791 for(p = cfg->local_zones; p; p = p->next) {
792 if(!(
793 #ifndef THREADS_DISABLED
794 z=
795 #endif
796 lz_enter_zone(zones, p->str, p->str2,
797 LDNS_RR_CLASS_IN)))
798 return 0;
799 lock_rw_unlock(&z->lock);
800 }
801 return 1;
802 }
803
804 /** lookup a zone in rbtree; exact match only; SLOW due to parse */
805 static int
lz_exists(struct local_zones * zones,const char * name)806 lz_exists(struct local_zones* zones, const char* name)
807 {
808 struct local_zone z;
809 z.node.key = &z;
810 z.dclass = LDNS_RR_CLASS_IN;
811 if(!parse_dname(name, &z.name, &z.namelen, &z.namelabs)) {
812 log_err("bad name %s", name);
813 return 0;
814 }
815 lock_rw_rdlock(&zones->lock);
816 if(rbtree_search(&zones->ztree, &z.node)) {
817 lock_rw_unlock(&zones->lock);
818 free(z.name);
819 return 1;
820 }
821 lock_rw_unlock(&zones->lock);
822 free(z.name);
823 return 0;
824 }
825
826 /** lookup a zone in cfg->nodefault list */
827 static int
lz_nodefault(struct config_file * cfg,const char * name)828 lz_nodefault(struct config_file* cfg, const char* name)
829 {
830 struct config_strlist* p;
831 size_t len = strlen(name);
832 if(len == 0) return 0;
833 if(name[len-1] == '.') len--;
834
835 for(p = cfg->local_zones_nodefault; p; p = p->next) {
836 /* compare zone name, lowercase, compare without ending . */
837 if(strncasecmp(p->str, name, len) == 0 &&
838 (strlen(p->str) == len || (strlen(p->str)==len+1 &&
839 p->str[len] == '.')))
840 return 1;
841 }
842 return 0;
843 }
844
845 /** enter (AS112) empty default zone */
846 static int
add_empty_default(struct local_zones * zones,struct config_file * cfg,const char * name)847 add_empty_default(struct local_zones* zones, struct config_file* cfg,
848 const char* name)
849 {
850 struct local_zone* z;
851 char str[1024]; /* known long enough */
852 if(lz_exists(zones, name) || lz_nodefault(cfg, name))
853 return 1; /* do not enter default content */
854 if(!(z=lz_enter_zone(zones, name, "static", LDNS_RR_CLASS_IN)))
855 return 0;
856 snprintf(str, sizeof(str), "%s 10800 IN SOA localhost. "
857 "nobody.invalid. 1 3600 1200 604800 10800", name);
858 if(!lz_enter_rr_into_zone(z, str)) {
859 lock_rw_unlock(&z->lock);
860 return 0;
861 }
862 snprintf(str, sizeof(str), "%s 10800 IN NS localhost. ", name);
863 if(!lz_enter_rr_into_zone(z, str)) {
864 lock_rw_unlock(&z->lock);
865 return 0;
866 }
867 lock_rw_unlock(&z->lock);
868 return 1;
869 }
870
871 /** enter default zones */
local_zone_enter_defaults(struct local_zones * zones,struct config_file * cfg)872 int local_zone_enter_defaults(struct local_zones* zones, struct config_file* cfg)
873 {
874 struct local_zone* z;
875 const char** zstr;
876
877 /* Do not add any default */
878 if(cfg->local_zones_disable_default)
879 return 1;
880
881 /* this list of zones is from RFC 6303 and RFC 7686 */
882
883 /* block localhost level zones first, then onion and later the LAN zones */
884
885 /* localhost. zone */
886 if(!lz_exists(zones, "localhost.") &&
887 !lz_nodefault(cfg, "localhost.")) {
888 if(!(z=lz_enter_zone(zones, "localhost.", "redirect",
889 LDNS_RR_CLASS_IN)) ||
890 !lz_enter_rr_into_zone(z,
891 "localhost. 10800 IN NS localhost.") ||
892 !lz_enter_rr_into_zone(z,
893 "localhost. 10800 IN SOA localhost. nobody.invalid. "
894 "1 3600 1200 604800 10800") ||
895 !lz_enter_rr_into_zone(z,
896 "localhost. 10800 IN A 127.0.0.1") ||
897 !lz_enter_rr_into_zone(z,
898 "localhost. 10800 IN AAAA ::1")) {
899 log_err("out of memory adding default zone");
900 if(z) { lock_rw_unlock(&z->lock); }
901 return 0;
902 }
903 lock_rw_unlock(&z->lock);
904 }
905 /* reverse ip4 zone */
906 if(!lz_exists(zones, "127.in-addr.arpa.") &&
907 !lz_nodefault(cfg, "127.in-addr.arpa.")) {
908 if(!(z=lz_enter_zone(zones, "127.in-addr.arpa.", "static",
909 LDNS_RR_CLASS_IN)) ||
910 !lz_enter_rr_into_zone(z,
911 "127.in-addr.arpa. 10800 IN NS localhost.") ||
912 !lz_enter_rr_into_zone(z,
913 "127.in-addr.arpa. 10800 IN SOA localhost. "
914 "nobody.invalid. 1 3600 1200 604800 10800") ||
915 !lz_enter_rr_into_zone(z,
916 "1.0.0.127.in-addr.arpa. 10800 IN PTR localhost.")) {
917 log_err("out of memory adding default zone");
918 if(z) { lock_rw_unlock(&z->lock); }
919 return 0;
920 }
921 lock_rw_unlock(&z->lock);
922 }
923 /* reverse ip6 zone */
924 if(!lz_exists(zones, "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.") &&
925 !lz_nodefault(cfg, "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.")) {
926 if(!(z=lz_enter_zone(zones, "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.", "static",
927 LDNS_RR_CLASS_IN)) ||
928 !lz_enter_rr_into_zone(z,
929 "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa. 10800 IN NS localhost.") ||
930 !lz_enter_rr_into_zone(z,
931 "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa. 10800 IN SOA localhost. "
932 "nobody.invalid. 1 3600 1200 604800 10800") ||
933 !lz_enter_rr_into_zone(z,
934 "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa. 10800 IN PTR localhost.")) {
935 log_err("out of memory adding default zone");
936 if(z) { lock_rw_unlock(&z->lock); }
937 return 0;
938 }
939 lock_rw_unlock(&z->lock);
940 }
941 /* home.arpa. zone (RFC 8375) */
942 if(!add_empty_default(zones, cfg, "home.arpa.")) {
943 log_err("out of memory adding default zone");
944 return 0;
945 }
946 /* resolver.arpa. zone (RFC 9462) */
947 if(!add_empty_default(zones, cfg, "resolver.arpa.")) {
948 log_err("out of memory adding default zone");
949 return 0;
950 }
951 /* service.arpa. zone (draft-ietf-dnssd-srp-25) */
952 if(!add_empty_default(zones, cfg, "service.arpa.")) {
953 log_err("out of memory adding default zone");
954 return 0;
955 }
956 /* onion. zone (RFC 7686) */
957 if(!add_empty_default(zones, cfg, "onion.")) {
958 log_err("out of memory adding default zone");
959 return 0;
960 }
961 /* test. zone (RFC 6761) */
962 if(!add_empty_default(zones, cfg, "test.")) {
963 log_err("out of memory adding default zone");
964 return 0;
965 }
966 /* invalid. zone (RFC 6761) */
967 if(!add_empty_default(zones, cfg, "invalid.")) {
968 log_err("out of memory adding default zone");
969 return 0;
970 }
971 /* block AS112 zones, unless asked not to */
972 if(!cfg->unblock_lan_zones) {
973 for(zstr = as112_zones; *zstr; zstr++) {
974 if(!add_empty_default(zones, cfg, *zstr)) {
975 log_err("out of memory adding default zone");
976 return 0;
977 }
978 }
979 }
980 return 1;
981 }
982
983 /** parse local-zone-override: statements */
984 static int
lz_enter_overrides(struct local_zones * zones,struct config_file * cfg)985 lz_enter_overrides(struct local_zones* zones, struct config_file* cfg)
986 {
987 struct config_str3list* p;
988 for(p = cfg->local_zone_overrides; p; p = p->next) {
989 if(!lz_enter_override(zones, p->str, p->str2, p->str3,
990 LDNS_RR_CLASS_IN))
991 return 0;
992 }
993 return 1;
994 }
995
996 /* return closest parent in the tree, NULL if none */
find_closest_parent(struct local_zone * curr,struct local_zone * prev)997 static struct local_zone* find_closest_parent(struct local_zone* curr,
998 struct local_zone* prev)
999 {
1000 struct local_zone* p;
1001 int m;
1002 if(!prev || prev->dclass != curr->dclass) return NULL;
1003 (void)dname_lab_cmp(prev->name, prev->namelabs, curr->name,
1004 curr->namelabs, &m); /* we know prev is smaller */
1005 /* sort order like: . com. bla.com. zwb.com. net. */
1006 /* find the previous, or parent-parent-parent */
1007 for(p = prev; p; p = p->parent) {
1008 /* looking for name with few labels, a parent */
1009 if(p->namelabs <= m) {
1010 /* ==: since prev matched m, this is closest*/
1011 /* <: prev matches more, but is not a parent,
1012 * this one is a (grand)parent */
1013 return p;
1014 }
1015 }
1016 return NULL;
1017 }
1018
1019 /** setup parent pointers, so that a lookup can be done for closest match */
1020 void
lz_init_parents(struct local_zones * zones)1021 lz_init_parents(struct local_zones* zones)
1022 {
1023 struct local_zone* node, *prev = NULL;
1024 lock_rw_wrlock(&zones->lock);
1025 RBTREE_FOR(node, struct local_zone*, &zones->ztree) {
1026 lock_rw_wrlock(&node->lock);
1027 node->parent = find_closest_parent(node, prev);
1028 prev = node;
1029 if(node->override_tree)
1030 addr_tree_init_parents(node->override_tree);
1031 lock_rw_unlock(&node->lock);
1032 }
1033 lock_rw_unlock(&zones->lock);
1034 }
1035
1036 /** enter implicit transparent zone for local-data: without local-zone: */
1037 static int
lz_setup_implicit(struct local_zones * zones,struct config_file * cfg)1038 lz_setup_implicit(struct local_zones* zones, struct config_file* cfg)
1039 {
1040 /* walk over all items that have no parent zone and find
1041 * the name that covers them all (could be the root) and
1042 * add that as a transparent zone */
1043 struct config_strlist* p;
1044 int have_name = 0;
1045 int have_other_classes = 0;
1046 uint16_t dclass = 0;
1047 uint8_t* nm = 0;
1048 size_t nmlen = 0;
1049 int nmlabs = 0;
1050 int match = 0; /* number of labels match count */
1051
1052 lz_init_parents(zones); /* to enable local_zones_lookup() */
1053 for(p = cfg->local_data; p; p = p->next) {
1054 uint8_t* rr_name;
1055 uint16_t rr_class, rr_type;
1056 size_t len;
1057 int labs;
1058 if(!get_rr_nameclass(p->str, &rr_name, &rr_class, &rr_type)) {
1059 log_err("Bad local-data RR %s", p->str);
1060 return 0;
1061 }
1062 labs = dname_count_size_labels(rr_name, &len);
1063 lock_rw_rdlock(&zones->lock);
1064 if(!local_zones_lookup(zones, rr_name, len, labs, rr_class,
1065 rr_type)) {
1066 /* Check if there is a zone that this could go
1067 * under but for different class; created zones are
1068 * always for LDNS_RR_CLASS_IN. Create the zone with
1069 * a different class but the same configured
1070 * local_zone_type. */
1071 struct local_zone* z = local_zones_lookup(zones,
1072 rr_name, len, labs, LDNS_RR_CLASS_IN, rr_type);
1073 if(z) {
1074 uint8_t* name = memdup(z->name, z->namelen);
1075 size_t znamelen = z->namelen;
1076 int znamelabs = z->namelabs;
1077 enum localzone_type ztype = z->type;
1078 lock_rw_unlock(&zones->lock);
1079 if(!name) {
1080 log_err("out of memory");
1081 free(rr_name);
1082 return 0;
1083 }
1084 if(!(
1085 #ifndef THREADS_DISABLED
1086 z =
1087 #endif
1088 lz_enter_zone_dname(zones, name,
1089 znamelen, znamelabs,
1090 ztype, rr_class))) {
1091 free(rr_name);
1092 return 0;
1093 }
1094 lock_rw_unlock(&z->lock);
1095 free(rr_name);
1096 continue;
1097 }
1098 if(!have_name) {
1099 dclass = rr_class;
1100 nm = rr_name;
1101 nmlen = len;
1102 nmlabs = labs;
1103 match = labs;
1104 have_name = 1;
1105 } else {
1106 int m;
1107 if(rr_class != dclass) {
1108 /* process other classes later */
1109 free(rr_name);
1110 have_other_classes = 1;
1111 lock_rw_unlock(&zones->lock);
1112 continue;
1113 }
1114 /* find smallest shared topdomain */
1115 (void)dname_lab_cmp(nm, nmlabs,
1116 rr_name, labs, &m);
1117 free(rr_name);
1118 if(m < match)
1119 match = m;
1120 }
1121 } else free(rr_name);
1122 lock_rw_unlock(&zones->lock);
1123 }
1124 if(have_name) {
1125 uint8_t* n2;
1126 #ifndef THREADS_DISABLED
1127 struct local_zone* z;
1128 #endif
1129 /* allocate zone of smallest shared topdomain to contain em */
1130 n2 = nm;
1131 dname_remove_labels(&n2, &nmlen, nmlabs - match);
1132 n2 = memdup(n2, nmlen);
1133 free(nm);
1134 if(!n2) {
1135 log_err("out of memory");
1136 return 0;
1137 }
1138 log_nametypeclass(VERB_ALGO, "implicit transparent local-zone",
1139 n2, 0, dclass);
1140 if(!(
1141 #ifndef THREADS_DISABLED
1142 z=
1143 #endif
1144 lz_enter_zone_dname(zones, n2, nmlen, match,
1145 local_zone_transparent, dclass))) {
1146 return 0;
1147 }
1148 lock_rw_unlock(&z->lock);
1149 }
1150 if(have_other_classes) {
1151 /* restart to setup other class */
1152 return lz_setup_implicit(zones, cfg);
1153 }
1154 return 1;
1155 }
1156
1157 /** enter local-zone-tag info */
1158 static int
lz_enter_zone_tags(struct local_zones * zones,struct config_file * cfg)1159 lz_enter_zone_tags(struct local_zones* zones, struct config_file* cfg)
1160 {
1161 struct config_strbytelist* p;
1162 int c = 0;
1163 for(p = cfg->local_zone_tags; p; p = p->next) {
1164 if(!lz_enter_zone_tag(zones, p->str, p->str2, p->str2len,
1165 LDNS_RR_CLASS_IN))
1166 return 0;
1167 c++;
1168 }
1169 if(c) verbose(VERB_ALGO, "applied tags to %d local zones", c);
1170 return 1;
1171 }
1172
1173 /** enter auth data */
1174 static int
lz_enter_data(struct local_zones * zones,struct config_file * cfg)1175 lz_enter_data(struct local_zones* zones, struct config_file* cfg)
1176 {
1177 struct config_strlist* p;
1178 for(p = cfg->local_data; p; p = p->next) {
1179 if(!lz_enter_rr_str(zones, p->str))
1180 return 0;
1181 }
1182 return 1;
1183 }
1184
1185 /** free memory from config */
1186 static void
lz_freeup_cfg(struct config_file * cfg)1187 lz_freeup_cfg(struct config_file* cfg)
1188 {
1189 config_deldblstrlist(cfg->local_zones);
1190 cfg->local_zones = NULL;
1191 config_delstrlist(cfg->local_zones_nodefault);
1192 cfg->local_zones_nodefault = NULL;
1193 config_delstrlist(cfg->local_data);
1194 cfg->local_data = NULL;
1195 }
1196
1197 int
local_zones_apply_cfg(struct local_zones * zones,struct config_file * cfg)1198 local_zones_apply_cfg(struct local_zones* zones, struct config_file* cfg)
1199 {
1200 /* create zones from zone statements. */
1201 if(!lz_enter_zones(zones, cfg)) {
1202 return 0;
1203 }
1204 /* apply default zones+content (unless disabled, or overridden) */
1205 if(!local_zone_enter_defaults(zones, cfg)) {
1206 return 0;
1207 }
1208 /* enter local zone overrides */
1209 if(!lz_enter_overrides(zones, cfg)) {
1210 return 0;
1211 }
1212 /* create implicit transparent zone from data. */
1213 if(!lz_setup_implicit(zones, cfg)) {
1214 return 0;
1215 }
1216
1217 /* setup parent ptrs for lookup during data entry */
1218 lz_init_parents(zones);
1219 /* insert local zone tags */
1220 if(!lz_enter_zone_tags(zones, cfg)) {
1221 return 0;
1222 }
1223 /* insert local data */
1224 if(!lz_enter_data(zones, cfg)) {
1225 return 0;
1226 }
1227 /* freeup memory from cfg struct. */
1228 lz_freeup_cfg(cfg);
1229 return 1;
1230 }
1231
1232 struct local_zone*
local_zones_lookup(struct local_zones * zones,uint8_t * name,size_t len,int labs,uint16_t dclass,uint16_t dtype)1233 local_zones_lookup(struct local_zones* zones,
1234 uint8_t* name, size_t len, int labs, uint16_t dclass, uint16_t dtype)
1235 {
1236 return local_zones_tags_lookup(zones, name, len, labs,
1237 dclass, dtype, NULL, 0, 1);
1238 }
1239
1240 struct local_zone*
local_zones_tags_lookup(struct local_zones * zones,uint8_t * name,size_t len,int labs,uint16_t dclass,uint16_t dtype,uint8_t * taglist,size_t taglen,int ignoretags)1241 local_zones_tags_lookup(struct local_zones* zones,
1242 uint8_t* name, size_t len, int labs, uint16_t dclass, uint16_t dtype,
1243 uint8_t* taglist, size_t taglen, int ignoretags)
1244 {
1245 rbnode_type* res = NULL;
1246 struct local_zone *result;
1247 struct local_zone key;
1248 int m;
1249 /* for type DS use a zone higher when on a zonecut */
1250 if(dtype == LDNS_RR_TYPE_DS && !dname_is_root(name)) {
1251 dname_remove_label(&name, &len);
1252 labs--;
1253 }
1254 key.node.key = &key;
1255 key.dclass = dclass;
1256 key.name = name;
1257 key.namelen = len;
1258 key.namelabs = labs;
1259 rbtree_find_less_equal(&zones->ztree, &key, &res);
1260 result = (struct local_zone*)res;
1261 /* exact or smaller element (or no element) */
1262 if(!result || result->dclass != dclass)
1263 return NULL;
1264 /* count number of labels matched */
1265 (void)dname_lab_cmp(result->name, result->namelabs, key.name,
1266 key.namelabs, &m);
1267 while(result) { /* go up until qname is zone or subdomain of zone */
1268 if(result->namelabs <= m)
1269 if(ignoretags || !result->taglist ||
1270 taglist_intersect(result->taglist,
1271 result->taglen, taglist, taglen))
1272 break;
1273 result = result->parent;
1274 }
1275 return result;
1276 }
1277
1278 struct local_zone*
local_zones_find(struct local_zones * zones,uint8_t * name,size_t len,int labs,uint16_t dclass)1279 local_zones_find(struct local_zones* zones,
1280 uint8_t* name, size_t len, int labs, uint16_t dclass)
1281 {
1282 struct local_zone key;
1283 key.node.key = &key;
1284 key.dclass = dclass;
1285 key.name = name;
1286 key.namelen = len;
1287 key.namelabs = labs;
1288 /* exact */
1289 return (struct local_zone*)rbtree_search(&zones->ztree, &key);
1290 }
1291
1292 struct local_zone*
local_zones_find_le(struct local_zones * zones,uint8_t * name,size_t len,int labs,uint16_t dclass,int * exact)1293 local_zones_find_le(struct local_zones* zones,
1294 uint8_t* name, size_t len, int labs, uint16_t dclass,
1295 int* exact)
1296 {
1297 struct local_zone key;
1298 rbnode_type *node;
1299 key.node.key = &key;
1300 key.dclass = dclass;
1301 key.name = name;
1302 key.namelen = len;
1303 key.namelabs = labs;
1304 *exact = rbtree_find_less_equal(&zones->ztree, &key, &node);
1305 return (struct local_zone*)node;
1306 }
1307
1308 /** encode answer consisting of 1 rrset */
1309 static int
local_encode(struct query_info * qinfo,struct module_env * env,struct edns_data * edns,struct comm_reply * repinfo,sldns_buffer * buf,struct regional * temp,struct ub_packed_rrset_key * rrset,int ansec,int rcode)1310 local_encode(struct query_info* qinfo, struct module_env* env,
1311 struct edns_data* edns, struct comm_reply* repinfo, sldns_buffer* buf,
1312 struct regional* temp, struct ub_packed_rrset_key* rrset, int ansec,
1313 int rcode)
1314 {
1315 struct reply_info rep;
1316 uint16_t udpsize;
1317 /* make answer with time=0 for fixed TTL values */
1318 memset(&rep, 0, sizeof(rep));
1319 rep.flags = (uint16_t)((BIT_QR | BIT_AA | BIT_RA) | rcode);
1320 rep.qdcount = 1;
1321 if(ansec)
1322 rep.an_numrrsets = 1;
1323 else rep.ns_numrrsets = 1;
1324 rep.rrset_count = 1;
1325 rep.rrsets = &rrset;
1326 rep.reason_bogus = LDNS_EDE_NONE;
1327 udpsize = edns->udp_size;
1328 edns->edns_version = EDNS_ADVERTISED_VERSION;
1329 edns->udp_size = EDNS_ADVERTISED_SIZE;
1330 edns->ext_rcode = 0;
1331 edns->bits &= EDNS_DO;
1332 if(!inplace_cb_reply_local_call(env, qinfo, NULL, &rep, rcode, edns,
1333 repinfo, temp, env->now_tv) || !reply_info_answer_encode(qinfo, &rep,
1334 *(uint16_t*)sldns_buffer_begin(buf), sldns_buffer_read_u16_at(buf, 2),
1335 buf, 0, 0, temp, udpsize, edns, (int)(edns->bits&EDNS_DO), 0)) {
1336 error_encode(buf, (LDNS_RCODE_SERVFAIL|BIT_AA), qinfo,
1337 *(uint16_t*)sldns_buffer_begin(buf),
1338 sldns_buffer_read_u16_at(buf, 2), edns);
1339 }
1340 return 1;
1341 }
1342
1343 /** encode local error answer */
1344 static void
local_error_encode(struct query_info * qinfo,struct module_env * env,struct edns_data * edns,struct comm_reply * repinfo,sldns_buffer * buf,struct regional * temp,int rcode,int r,int ede_code,const char * ede_txt)1345 local_error_encode(struct query_info* qinfo, struct module_env* env,
1346 struct edns_data* edns, struct comm_reply* repinfo, sldns_buffer* buf,
1347 struct regional* temp, int rcode, int r, int ede_code,
1348 const char* ede_txt)
1349 {
1350 edns->edns_version = EDNS_ADVERTISED_VERSION;
1351 edns->udp_size = EDNS_ADVERTISED_SIZE;
1352 edns->ext_rcode = 0;
1353 edns->bits &= EDNS_DO;
1354
1355 if(!inplace_cb_reply_local_call(env, qinfo, NULL, NULL,
1356 rcode, edns, repinfo, temp, env->now_tv))
1357 edns->opt_list_inplace_cb_out = NULL;
1358
1359 if(ede_code != LDNS_EDE_NONE && env->cfg->ede) {
1360 edns_opt_list_append_ede(&edns->opt_list_out, temp,
1361 ede_code, ede_txt);
1362 }
1363
1364 error_encode(buf, r, qinfo, *(uint16_t*)sldns_buffer_begin(buf),
1365 sldns_buffer_read_u16_at(buf, 2), edns);
1366 }
1367
1368 /** find local data tag string match for the given type in the list */
1369 int
local_data_find_tag_datas(const struct query_info * qinfo,struct config_strlist * list,struct ub_packed_rrset_key * r,struct regional * temp)1370 local_data_find_tag_datas(const struct query_info* qinfo,
1371 struct config_strlist* list, struct ub_packed_rrset_key* r,
1372 struct regional* temp)
1373 {
1374 struct config_strlist* p;
1375 char buf[65536];
1376 uint8_t rr[LDNS_RR_BUF_SIZE];
1377 size_t len;
1378 int res;
1379 struct packed_rrset_data* d;
1380 for(p=list; p; p=p->next) {
1381 uint16_t rdr_type;
1382
1383 len = sizeof(rr);
1384 /* does this element match the type? */
1385 snprintf(buf, sizeof(buf), ". %s", p->str);
1386 res = sldns_str2wire_rr_buf(buf, rr, &len, NULL, 3600,
1387 NULL, 0, NULL, 0);
1388 if(res != 0)
1389 /* parse errors are already checked before, in
1390 * acllist check_data, skip this for robustness */
1391 continue;
1392 if(len < 1 /* . */ + 8 /* typeclassttl*/ + 2 /*rdatalen*/)
1393 continue;
1394 rdr_type = sldns_wirerr_get_type(rr, len, 1);
1395 if(rdr_type != qinfo->qtype && rdr_type != LDNS_RR_TYPE_CNAME)
1396 continue;
1397
1398 /* do we have entries already? if not setup key */
1399 if(r->rk.dname == NULL) {
1400 r->entry.key = r;
1401 r->rk.dname = qinfo->qname;
1402 r->rk.dname_len = qinfo->qname_len;
1403 r->rk.type = htons(rdr_type);
1404 r->rk.rrset_class = htons(qinfo->qclass);
1405 r->rk.flags = 0;
1406 d = (struct packed_rrset_data*)regional_alloc_zero(
1407 temp, sizeof(struct packed_rrset_data)
1408 + sizeof(size_t) + sizeof(uint8_t*) +
1409 sizeof(time_t));
1410 if(!d) return 0; /* out of memory */
1411 r->entry.data = d;
1412 d->ttl = sldns_wirerr_get_ttl(rr, len, 1);
1413 d->rr_len = (size_t*)((uint8_t*)d +
1414 sizeof(struct packed_rrset_data));
1415 d->rr_data = (uint8_t**)&(d->rr_len[1]);
1416 d->rr_ttl = (time_t*)&(d->rr_data[1]);
1417 }
1418 d = (struct packed_rrset_data*)r->entry.data;
1419 /* add entry to the data */
1420 if(d->count != 0) {
1421 size_t* oldlen = d->rr_len;
1422 uint8_t** olddata = d->rr_data;
1423 time_t* oldttl = d->rr_ttl;
1424 /* increase arrays for lookup */
1425 /* this is of course slow for very many records,
1426 * but most redirects are expected with few records */
1427 d->rr_len = (size_t*)regional_alloc_zero(temp,
1428 (d->count+1)*sizeof(size_t));
1429 d->rr_data = (uint8_t**)regional_alloc_zero(temp,
1430 (d->count+1)*sizeof(uint8_t*));
1431 d->rr_ttl = (time_t*)regional_alloc_zero(temp,
1432 (d->count+1)*sizeof(time_t));
1433 if(!d->rr_len || !d->rr_data || !d->rr_ttl)
1434 return 0; /* out of memory */
1435 /* first one was allocated after struct d, but new
1436 * ones get their own array increment alloc, so
1437 * copy old content */
1438 memmove(d->rr_len, oldlen, d->count*sizeof(size_t));
1439 memmove(d->rr_data, olddata, d->count*sizeof(uint8_t*));
1440 memmove(d->rr_ttl, oldttl, d->count*sizeof(time_t));
1441 }
1442
1443 d->rr_len[d->count] = sldns_wirerr_get_rdatalen(rr, len, 1)+2;
1444 d->rr_ttl[d->count] = sldns_wirerr_get_ttl(rr, len, 1);
1445 d->rr_data[d->count] = regional_alloc_init(temp,
1446 sldns_wirerr_get_rdatawl(rr, len, 1),
1447 d->rr_len[d->count]);
1448 if(!d->rr_data[d->count])
1449 return 0; /* out of memory */
1450 d->count++;
1451 }
1452 if(r->rk.dname)
1453 return 1;
1454 return 0;
1455 }
1456
1457 static int
find_tag_datas(struct query_info * qinfo,struct config_strlist * list,struct ub_packed_rrset_key * r,struct regional * temp)1458 find_tag_datas(struct query_info* qinfo, struct config_strlist* list,
1459 struct ub_packed_rrset_key* r, struct regional* temp)
1460 {
1461 int result = local_data_find_tag_datas(qinfo, list, r, temp);
1462
1463 /* If we've found a non-exact alias type of local data, make a shallow
1464 * copy of the RRset and remember it in qinfo to complete the alias
1465 * chain later. */
1466 if(result && qinfo->qtype != LDNS_RR_TYPE_CNAME &&
1467 r->rk.type == htons(LDNS_RR_TYPE_CNAME)) {
1468 qinfo->local_alias =
1469 regional_alloc_zero(temp, sizeof(struct local_rrset));
1470 if(!qinfo->local_alias)
1471 return 0; /* out of memory */
1472 qinfo->local_alias->rrset =
1473 regional_alloc_init(temp, r, sizeof(*r));
1474 if(!qinfo->local_alias->rrset)
1475 return 0; /* out of memory */
1476 }
1477 return result;
1478 }
1479
1480 int
local_data_answer(struct local_zone * z,struct module_env * env,struct query_info * qinfo,struct edns_data * edns,struct comm_reply * repinfo,sldns_buffer * buf,struct regional * temp,int labs,struct local_data ** ldp,enum localzone_type lz_type,int tag,struct config_strlist ** tag_datas,size_t tag_datas_size,char ** tagname,int num_tags)1481 local_data_answer(struct local_zone* z, struct module_env* env,
1482 struct query_info* qinfo, struct edns_data* edns,
1483 struct comm_reply* repinfo, sldns_buffer* buf,
1484 struct regional* temp, int labs, struct local_data** ldp,
1485 enum localzone_type lz_type, int tag, struct config_strlist** tag_datas,
1486 size_t tag_datas_size, char** tagname, int num_tags)
1487 {
1488 struct local_data key;
1489 struct local_data* ld;
1490 struct local_rrset* lr;
1491 key.node.key = &key;
1492 key.name = qinfo->qname;
1493 key.namelen = qinfo->qname_len;
1494 key.namelabs = labs;
1495 if(lz_type == local_zone_redirect ||
1496 lz_type == local_zone_inform_redirect) {
1497 key.name = z->name;
1498 key.namelen = z->namelen;
1499 key.namelabs = z->namelabs;
1500 if(tag != -1 && (size_t)tag<tag_datas_size && tag_datas[tag]) {
1501 struct ub_packed_rrset_key r;
1502 memset(&r, 0, sizeof(r));
1503 if(find_tag_datas(qinfo, tag_datas[tag], &r, temp)) {
1504 verbose(VERB_ALGO, "redirect with tag data [%d] %s",
1505 tag, (tag<num_tags?tagname[tag]:"null"));
1506
1507 /* If we found a matching alias, we should
1508 * use it as part of the answer, but we can't
1509 * encode it until we complete the alias
1510 * chain. */
1511 if(qinfo->local_alias)
1512 return 1;
1513 return local_encode(qinfo, env, edns, repinfo, buf, temp,
1514 &r, 1, LDNS_RCODE_NOERROR);
1515 }
1516 }
1517 }
1518 ld = (struct local_data*)rbtree_search(&z->data, &key.node);
1519 *ldp = ld;
1520 if(!ld) {
1521 return 0;
1522 }
1523 lr = local_data_find_type(ld, qinfo->qtype, 1);
1524 if(!lr)
1525 return 0;
1526
1527 /* Special case for alias matching. See local_data_answer(). */
1528 if((lz_type == local_zone_redirect ||
1529 lz_type == local_zone_inform_redirect) &&
1530 qinfo->qtype != LDNS_RR_TYPE_CNAME &&
1531 lr->rrset->rk.type == htons(LDNS_RR_TYPE_CNAME)) {
1532 uint8_t* ctarget;
1533 size_t ctargetlen = 0;
1534
1535 qinfo->local_alias =
1536 regional_alloc_zero(temp, sizeof(struct local_rrset));
1537 if(!qinfo->local_alias)
1538 return 0; /* out of memory */
1539 qinfo->local_alias->rrset = regional_alloc_init(
1540 temp, lr->rrset, sizeof(*lr->rrset));
1541 if(!qinfo->local_alias->rrset)
1542 return 0; /* out of memory */
1543 qinfo->local_alias->rrset->rk.dname = qinfo->qname;
1544 qinfo->local_alias->rrset->rk.dname_len = qinfo->qname_len;
1545 get_cname_target(lr->rrset, &ctarget, &ctargetlen);
1546 if(!ctargetlen)
1547 return 0; /* invalid cname */
1548 if(dname_is_wild(ctarget)) {
1549 /* synthesize cname target */
1550 struct packed_rrset_data* d, *lr_d;
1551 /* -3 for wildcard label and root label from qname */
1552 size_t newtargetlen = qinfo->qname_len + ctargetlen - 3;
1553
1554 log_assert(ctargetlen >= 3);
1555 log_assert(qinfo->qname_len >= 1);
1556
1557 if(newtargetlen > LDNS_MAX_DOMAINLEN) {
1558 qinfo->local_alias = NULL;
1559 local_error_encode(qinfo, env, edns, repinfo,
1560 buf, temp, LDNS_RCODE_YXDOMAIN,
1561 (LDNS_RCODE_YXDOMAIN|BIT_AA),
1562 LDNS_EDE_OTHER,
1563 "DNAME expansion became too large");
1564 return 1;
1565 }
1566 memset(&qinfo->local_alias->rrset->entry, 0,
1567 sizeof(qinfo->local_alias->rrset->entry));
1568 qinfo->local_alias->rrset->entry.key =
1569 qinfo->local_alias->rrset;
1570 qinfo->local_alias->rrset->entry.hash =
1571 rrset_key_hash(&qinfo->local_alias->rrset->rk);
1572 d = (struct packed_rrset_data*)regional_alloc_zero(temp,
1573 sizeof(struct packed_rrset_data) + sizeof(size_t) +
1574 sizeof(uint8_t*) + sizeof(time_t) + sizeof(uint16_t)
1575 + newtargetlen);
1576 if(!d)
1577 return 0; /* out of memory */
1578 lr_d = (struct packed_rrset_data*)lr->rrset->entry.data;
1579 qinfo->local_alias->rrset->entry.data = d;
1580 d->ttl = lr_d->rr_ttl[0]; /* RFC6672-like behavior:
1581 synth CNAME TTL uses original TTL*/
1582 d->count = 1;
1583 d->rrsig_count = 0;
1584 d->trust = rrset_trust_ans_noAA;
1585 d->rr_len = (size_t*)((uint8_t*)d +
1586 sizeof(struct packed_rrset_data));
1587 d->rr_len[0] = newtargetlen + sizeof(uint16_t);
1588 packed_rrset_ptr_fixup(d);
1589 d->rr_ttl[0] = d->ttl;
1590 sldns_write_uint16(d->rr_data[0], newtargetlen);
1591 /* write qname */
1592 memmove(d->rr_data[0] + sizeof(uint16_t), qinfo->qname,
1593 qinfo->qname_len - 1);
1594 /* write cname target wildcard label */
1595 memmove(d->rr_data[0] + sizeof(uint16_t) +
1596 qinfo->qname_len - 1, ctarget + 2,
1597 ctargetlen - 2);
1598 }
1599 return 1;
1600 }
1601 if(lz_type == local_zone_redirect ||
1602 lz_type == local_zone_inform_redirect) {
1603 /* convert rrset name to query name; like a wildcard */
1604 struct ub_packed_rrset_key r = *lr->rrset;
1605 r.rk.dname = qinfo->qname;
1606 r.rk.dname_len = qinfo->qname_len;
1607 return local_encode(qinfo, env, edns, repinfo, buf, temp, &r, 1,
1608 LDNS_RCODE_NOERROR);
1609 }
1610 return local_encode(qinfo, env, edns, repinfo, buf, temp, lr->rrset, 1,
1611 LDNS_RCODE_NOERROR);
1612 }
1613
1614 /**
1615 * See if the local zone does not cover the name, eg. the name is not
1616 * in the zone and the zone is transparent */
1617 static int
local_zone_does_not_cover(struct local_zone * z,struct query_info * qinfo,int labs)1618 local_zone_does_not_cover(struct local_zone* z, struct query_info* qinfo,
1619 int labs)
1620 {
1621 struct local_data key;
1622 struct local_data* ld = NULL;
1623 struct local_rrset* lr = NULL;
1624 if(z->type == local_zone_always_transparent || z->type == local_zone_block_a)
1625 return 1;
1626 if(z->type != local_zone_transparent
1627 && z->type != local_zone_typetransparent
1628 && z->type != local_zone_inform)
1629 return 0;
1630 key.node.key = &key;
1631 key.name = qinfo->qname;
1632 key.namelen = qinfo->qname_len;
1633 key.namelabs = labs;
1634 ld = (struct local_data*)rbtree_search(&z->data, &key.node);
1635 if(z->type == local_zone_transparent || z->type == local_zone_inform)
1636 return (ld == NULL);
1637 if(ld)
1638 lr = local_data_find_type(ld, qinfo->qtype, 1);
1639 /* local_zone_typetransparent */
1640 return (lr == NULL);
1641 }
1642
1643 static inline int
local_zone_is_udp_query(struct comm_reply * repinfo)1644 local_zone_is_udp_query(struct comm_reply* repinfo) {
1645 return repinfo != NULL
1646 ? (repinfo->c != NULL
1647 ? repinfo->c->type == comm_udp
1648 : 0)
1649 : 0;
1650 }
1651
1652 int
local_zones_zone_answer(struct local_zone * z,struct module_env * env,struct query_info * qinfo,struct edns_data * edns,struct comm_reply * repinfo,sldns_buffer * buf,struct regional * temp,struct local_data * ld,enum localzone_type lz_type)1653 local_zones_zone_answer(struct local_zone* z, struct module_env* env,
1654 struct query_info* qinfo, struct edns_data* edns,
1655 struct comm_reply* repinfo, sldns_buffer* buf, struct regional* temp,
1656 struct local_data* ld, enum localzone_type lz_type)
1657 {
1658 if(lz_type == local_zone_deny ||
1659 lz_type == local_zone_always_deny ||
1660 lz_type == local_zone_inform_deny) {
1661 /** no reply at all, signal caller by clearing buffer. */
1662 sldns_buffer_clear(buf);
1663 sldns_buffer_flip(buf);
1664 return 1;
1665 } else if(lz_type == local_zone_refuse
1666 || lz_type == local_zone_always_refuse) {
1667 local_error_encode(qinfo, env, edns, repinfo, buf, temp,
1668 LDNS_RCODE_REFUSED, (LDNS_RCODE_REFUSED|BIT_AA),
1669 LDNS_EDE_NONE, NULL);
1670 return 1;
1671 } else if(lz_type == local_zone_static ||
1672 lz_type == local_zone_redirect ||
1673 lz_type == local_zone_inform_redirect ||
1674 lz_type == local_zone_always_nxdomain ||
1675 lz_type == local_zone_always_nodata ||
1676 (lz_type == local_zone_truncate
1677 && local_zone_is_udp_query(repinfo))) {
1678 /* for static, reply nodata or nxdomain
1679 * for redirect, reply nodata */
1680 /* no additional section processing,
1681 * cname, dname or wildcard processing,
1682 * or using closest match for NSEC.
1683 * or using closest match for returning delegation downwards
1684 */
1685 int rcode = (ld || lz_type == local_zone_redirect ||
1686 lz_type == local_zone_inform_redirect ||
1687 lz_type == local_zone_always_nodata ||
1688 lz_type == local_zone_truncate)?
1689 LDNS_RCODE_NOERROR:LDNS_RCODE_NXDOMAIN;
1690 rcode = (lz_type == local_zone_truncate ? (rcode|BIT_TC) : rcode);
1691 if(z != NULL && z->soa && z->soa_negative)
1692 return local_encode(qinfo, env, edns, repinfo, buf, temp,
1693 z->soa_negative, 0, rcode);
1694 local_error_encode(qinfo, env, edns, repinfo, buf, temp,
1695 rcode, (rcode|BIT_AA), LDNS_EDE_NONE, NULL);
1696 return 1;
1697 } else if(lz_type == local_zone_typetransparent
1698 || lz_type == local_zone_always_transparent) {
1699 /* no NODATA or NXDOMAINS for this zone type */
1700 return 0;
1701 } else if(lz_type == local_zone_block_a) {
1702 /* Return NODATA for all A queries */
1703 if(qinfo->qtype == LDNS_RR_TYPE_A) {
1704 local_error_encode(qinfo, env, edns, repinfo, buf, temp,
1705 LDNS_RCODE_NOERROR, (LDNS_RCODE_NOERROR|BIT_AA),
1706 LDNS_EDE_NONE, NULL);
1707 return 1;
1708 }
1709
1710 return 0;
1711 } else if(lz_type == local_zone_always_null) {
1712 /* 0.0.0.0 or ::0 or noerror/nodata for this zone type,
1713 * used for blocklists. */
1714 if(qinfo->qtype == LDNS_RR_TYPE_A ||
1715 qinfo->qtype == LDNS_RR_TYPE_AAAA) {
1716 struct ub_packed_rrset_key lrr;
1717 struct packed_rrset_data d;
1718 time_t rr_ttl = 3600;
1719 size_t rr_len = 0;
1720 uint8_t rr_data[2+16] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
1721 uint8_t* rr_datas = rr_data;
1722 memset(&lrr, 0, sizeof(lrr));
1723 memset(&d, 0, sizeof(d));
1724 lrr.entry.data = &d;
1725 lrr.rk.dname = qinfo->qname;
1726 lrr.rk.dname_len = qinfo->qname_len;
1727 lrr.rk.type = htons(qinfo->qtype);
1728 lrr.rk.rrset_class = htons(qinfo->qclass);
1729 if(qinfo->qtype == LDNS_RR_TYPE_A) {
1730 rr_len = 4;
1731 sldns_write_uint16(rr_data, rr_len);
1732 rr_len += 2;
1733 } else {
1734 rr_len = 16;
1735 sldns_write_uint16(rr_data, rr_len);
1736 rr_len += 2;
1737 }
1738 d.ttl = rr_ttl;
1739 d.count = 1;
1740 d.rr_len = &rr_len;
1741 d.rr_data = &rr_datas;
1742 d.rr_ttl = &rr_ttl;
1743 return local_encode(qinfo, env, edns, repinfo, buf, temp,
1744 &lrr, 1, LDNS_RCODE_NOERROR);
1745 } else {
1746 /* NODATA: No EDE needed */
1747 local_error_encode(qinfo, env, edns, repinfo, buf,
1748 temp, LDNS_RCODE_NOERROR,
1749 (LDNS_RCODE_NOERROR|BIT_AA), -1, NULL);
1750 }
1751 return 1;
1752 }
1753 /* else lz_type == local_zone_transparent */
1754
1755 /* if the zone is transparent and the name exists, but the type
1756 * does not, then we should make this noerror/nodata */
1757 if(ld && ld->rrsets) {
1758 int rcode = LDNS_RCODE_NOERROR;
1759 if(z != NULL && z->soa && z->soa_negative)
1760 return local_encode(qinfo, env, edns, repinfo, buf, temp,
1761 z->soa_negative, 0, rcode);
1762 /* NODATA: No EDE needed */
1763 local_error_encode(qinfo, env, edns, repinfo, buf, temp, rcode,
1764 (rcode|BIT_AA), LDNS_EDE_NONE, NULL);
1765 return 1;
1766 }
1767
1768 /* stop here, and resolve further on */
1769 return 0;
1770 }
1771
1772 /** print log information for an inform zone query */
1773 static void
lz_inform_print(struct local_zone * z,struct query_info * qinfo,struct sockaddr_storage * addr,socklen_t addrlen)1774 lz_inform_print(struct local_zone* z, struct query_info* qinfo,
1775 struct sockaddr_storage* addr, socklen_t addrlen)
1776 {
1777 char ip[128], txt[512];
1778 char zname[LDNS_MAX_DOMAINLEN];
1779 uint16_t port = ntohs(((struct sockaddr_in*)addr)->sin_port);
1780 dname_str(z->name, zname);
1781 addr_to_str(addr, addrlen, ip, sizeof(ip));
1782 snprintf(txt, sizeof(txt), "%s %s %s@%u", zname, local_zone_type2str(z->type), ip,
1783 (unsigned)port);
1784 log_nametypeclass(NO_VERBOSE, txt, qinfo->qname, qinfo->qtype, qinfo->qclass);
1785 }
1786
1787 static enum localzone_type
lz_type(uint8_t * taglist,size_t taglen,uint8_t * taglist2,size_t taglen2,uint8_t * tagactions,size_t tagactionssize,enum localzone_type lzt,struct comm_reply * repinfo,struct rbtree_type * override_tree,int * tag,char ** tagname,int num_tags)1788 lz_type(uint8_t *taglist, size_t taglen, uint8_t *taglist2, size_t taglen2,
1789 uint8_t *tagactions, size_t tagactionssize, enum localzone_type lzt,
1790 struct comm_reply* repinfo, struct rbtree_type* override_tree,
1791 int* tag, char** tagname, int num_tags)
1792 {
1793 struct local_zone_override* lzo;
1794 if(repinfo && override_tree) {
1795 lzo = (struct local_zone_override*)addr_tree_lookup(
1796 override_tree, &repinfo->client_addr,
1797 repinfo->client_addrlen);
1798 if(lzo && lzo->type) {
1799 verbose(VERB_ALGO, "local zone override to type %s",
1800 local_zone_type2str(lzo->type));
1801 return lzo->type;
1802 }
1803 }
1804 if(!taglist || !taglist2)
1805 return lzt;
1806 return local_data_find_tag_action(taglist, taglen, taglist2, taglen2,
1807 tagactions, tagactionssize, lzt, tag, tagname, num_tags);
1808 }
1809
1810 enum localzone_type
local_data_find_tag_action(const uint8_t * taglist,size_t taglen,const uint8_t * taglist2,size_t taglen2,const uint8_t * tagactions,size_t tagactionssize,enum localzone_type lzt,int * tag,char * const * tagname,int num_tags)1811 local_data_find_tag_action(const uint8_t* taglist, size_t taglen,
1812 const uint8_t* taglist2, size_t taglen2, const uint8_t* tagactions,
1813 size_t tagactionssize, enum localzone_type lzt, int* tag,
1814 char* const* tagname, int num_tags)
1815 {
1816 size_t i, j;
1817 uint8_t tagmatch;
1818
1819 for(i=0; i<taglen && i<taglen2; i++) {
1820 tagmatch = (taglist[i] & taglist2[i]);
1821 for(j=0; j<8 && tagmatch>0; j++) {
1822 if((tagmatch & 0x1)) {
1823 *tag = (int)(i*8+j);
1824 verbose(VERB_ALGO, "matched tag [%d] %s",
1825 *tag, (*tag<num_tags?tagname[*tag]:"null"));
1826 /* does this tag have a tag action? */
1827 if(i*8+j < tagactionssize && tagactions
1828 && tagactions[i*8+j] != 0) {
1829 verbose(VERB_ALGO, "tag action [%d] %s to type %s",
1830 *tag, (*tag<num_tags?tagname[*tag]:"null"),
1831 local_zone_type2str(
1832 (enum localzone_type)
1833 tagactions[i*8+j]));
1834 return (enum localzone_type)tagactions[i*8+j];
1835 }
1836 return lzt;
1837 }
1838 tagmatch >>= 1;
1839 }
1840 }
1841 return lzt;
1842 }
1843
1844 int
local_zones_answer(struct local_zones * zones,struct module_env * env,struct query_info * qinfo,struct edns_data * edns,sldns_buffer * buf,struct regional * temp,struct comm_reply * repinfo,uint8_t * taglist,size_t taglen,uint8_t * tagactions,size_t tagactionssize,struct config_strlist ** tag_datas,size_t tag_datas_size,char ** tagname,int num_tags,struct view * view)1845 local_zones_answer(struct local_zones* zones, struct module_env* env,
1846 struct query_info* qinfo, struct edns_data* edns, sldns_buffer* buf,
1847 struct regional* temp, struct comm_reply* repinfo, uint8_t* taglist,
1848 size_t taglen, uint8_t* tagactions, size_t tagactionssize,
1849 struct config_strlist** tag_datas, size_t tag_datas_size,
1850 char** tagname, int num_tags, struct view* view)
1851 {
1852 /* see if query is covered by a zone,
1853 * if so: - try to match (exact) local data
1854 * - look at zone type for negative response. */
1855 int labs = dname_count_labels(qinfo->qname);
1856 struct local_data* ld = NULL;
1857 struct local_zone* z = NULL;
1858 enum localzone_type lzt = local_zone_transparent;
1859 int r, tag = -1;
1860
1861 if(view) {
1862 lock_rw_rdlock(&view->lock);
1863 if(view->local_zones &&
1864 (z = local_zones_lookup(view->local_zones,
1865 qinfo->qname, qinfo->qname_len, labs,
1866 qinfo->qclass, qinfo->qtype))) {
1867 lock_rw_rdlock(&z->lock);
1868 lzt = z->type;
1869 }
1870 if(lzt == local_zone_noview) {
1871 lock_rw_unlock(&z->lock);
1872 z = NULL;
1873 }
1874 if(z && (lzt == local_zone_transparent ||
1875 lzt == local_zone_typetransparent ||
1876 lzt == local_zone_inform ||
1877 lzt == local_zone_always_transparent ||
1878 lzt == local_zone_block_a) &&
1879 local_zone_does_not_cover(z, qinfo, labs)) {
1880 lock_rw_unlock(&z->lock);
1881 z = NULL;
1882 }
1883 if(view->local_zones && !z && !view->isfirst){
1884 lock_rw_unlock(&view->lock);
1885 return 0;
1886 }
1887 if(z && verbosity >= VERB_ALGO) {
1888 char zname[LDNS_MAX_DOMAINLEN];
1889 dname_str(z->name, zname);
1890 verbose(VERB_ALGO, "using localzone %s %s from view %s",
1891 zname, local_zone_type2str(lzt), view->name);
1892 }
1893 lock_rw_unlock(&view->lock);
1894 }
1895 if(!z) {
1896 /* try global local_zones tree */
1897 lock_rw_rdlock(&zones->lock);
1898 if(!(z = local_zones_tags_lookup(zones, qinfo->qname,
1899 qinfo->qname_len, labs, qinfo->qclass, qinfo->qtype,
1900 taglist, taglen, 0))) {
1901 lock_rw_unlock(&zones->lock);
1902 return 0;
1903 }
1904 lock_rw_rdlock(&z->lock);
1905 lzt = lz_type(taglist, taglen, z->taglist, z->taglen,
1906 tagactions, tagactionssize, z->type, repinfo,
1907 z->override_tree, &tag, tagname, num_tags);
1908 lock_rw_unlock(&zones->lock);
1909 if(z && verbosity >= VERB_ALGO) {
1910 char zname[LDNS_MAX_DOMAINLEN];
1911 dname_str(z->name, zname);
1912 verbose(VERB_ALGO, "using localzone %s %s", zname,
1913 local_zone_type2str(lzt));
1914 }
1915 }
1916 if((env->cfg->log_local_actions ||
1917 lzt == local_zone_inform ||
1918 lzt == local_zone_inform_deny ||
1919 lzt == local_zone_inform_redirect)
1920 && repinfo)
1921 lz_inform_print(z, qinfo, &repinfo->client_addr,
1922 repinfo->client_addrlen);
1923
1924 if(lzt != local_zone_always_refuse
1925 && lzt != local_zone_always_transparent
1926 && lzt != local_zone_block_a
1927 && lzt != local_zone_always_nxdomain
1928 && lzt != local_zone_always_nodata
1929 && lzt != local_zone_always_deny
1930 && local_data_answer(z, env, qinfo, edns, repinfo, buf, temp, labs,
1931 &ld, lzt, tag, tag_datas, tag_datas_size, tagname, num_tags)) {
1932 lock_rw_unlock(&z->lock);
1933 /* We should tell the caller that encode is deferred if we found
1934 * a local alias. */
1935 return !qinfo->local_alias;
1936 }
1937 r = local_zones_zone_answer(z, env, qinfo, edns, repinfo, buf, temp, ld, lzt);
1938 lock_rw_unlock(&z->lock);
1939 return r && !qinfo->local_alias; /* see above */
1940 }
1941
local_zone_type2str(enum localzone_type t)1942 const char* local_zone_type2str(enum localzone_type t)
1943 {
1944 switch(t) {
1945 case local_zone_unset: return "unset";
1946 case local_zone_deny: return "deny";
1947 case local_zone_refuse: return "refuse";
1948 case local_zone_redirect: return "redirect";
1949 case local_zone_transparent: return "transparent";
1950 case local_zone_typetransparent: return "typetransparent";
1951 case local_zone_static: return "static";
1952 case local_zone_nodefault: return "nodefault";
1953 case local_zone_inform: return "inform";
1954 case local_zone_inform_deny: return "inform_deny";
1955 case local_zone_inform_redirect: return "inform_redirect";
1956 case local_zone_always_transparent: return "always_transparent";
1957 case local_zone_block_a: return "block_a";
1958 case local_zone_always_refuse: return "always_refuse";
1959 case local_zone_always_nxdomain: return "always_nxdomain";
1960 case local_zone_always_nodata: return "always_nodata";
1961 case local_zone_always_deny: return "always_deny";
1962 case local_zone_always_null: return "always_null";
1963 case local_zone_noview: return "noview";
1964 case local_zone_truncate: return "truncate";
1965 case local_zone_invalid: return "invalid";
1966 }
1967 return "badtyped";
1968 }
1969
local_zone_str2type(const char * type,enum localzone_type * t)1970 int local_zone_str2type(const char* type, enum localzone_type* t)
1971 {
1972 if(strcmp(type, "deny") == 0)
1973 *t = local_zone_deny;
1974 else if(strcmp(type, "refuse") == 0)
1975 *t = local_zone_refuse;
1976 else if(strcmp(type, "static") == 0)
1977 *t = local_zone_static;
1978 else if(strcmp(type, "transparent") == 0)
1979 *t = local_zone_transparent;
1980 else if(strcmp(type, "typetransparent") == 0)
1981 *t = local_zone_typetransparent;
1982 else if(strcmp(type, "redirect") == 0)
1983 *t = local_zone_redirect;
1984 else if(strcmp(type, "inform") == 0)
1985 *t = local_zone_inform;
1986 else if(strcmp(type, "inform_deny") == 0)
1987 *t = local_zone_inform_deny;
1988 else if(strcmp(type, "inform_redirect") == 0)
1989 *t = local_zone_inform_redirect;
1990 else if(strcmp(type, "always_transparent") == 0)
1991 *t = local_zone_always_transparent;
1992 else if(strcmp(type, "block_a") == 0)
1993 *t = local_zone_block_a;
1994 else if(strcmp(type, "always_refuse") == 0)
1995 *t = local_zone_always_refuse;
1996 else if(strcmp(type, "always_nxdomain") == 0)
1997 *t = local_zone_always_nxdomain;
1998 else if(strcmp(type, "always_nodata") == 0)
1999 *t = local_zone_always_nodata;
2000 else if(strcmp(type, "always_deny") == 0)
2001 *t = local_zone_always_deny;
2002 else if(strcmp(type, "always_null") == 0)
2003 *t = local_zone_always_null;
2004 else if(strcmp(type, "noview") == 0)
2005 *t = local_zone_noview;
2006 else if(strcmp(type, "truncate") == 0)
2007 *t = local_zone_truncate;
2008 else if(strcmp(type, "nodefault") == 0)
2009 *t = local_zone_nodefault;
2010 else return 0;
2011 return 1;
2012 }
2013
2014 /** iterate over the kiddies of the given name and set their parent ptr */
2015 static void
set_kiddo_parents(struct local_zone * z,struct local_zone * match,struct local_zone * newp)2016 set_kiddo_parents(struct local_zone* z, struct local_zone* match,
2017 struct local_zone* newp)
2018 {
2019 /* both zones and z are locked already */
2020 /* in the sorted rbtree, the kiddies of z are located after z */
2021 /* z must be present in the tree */
2022 struct local_zone* p = z;
2023 p = (struct local_zone*)rbtree_next(&p->node);
2024 while(p!=(struct local_zone*)RBTREE_NULL &&
2025 p->dclass == z->dclass && dname_strict_subdomain(p->name,
2026 p->namelabs, z->name, z->namelabs)) {
2027 /* update parent ptr */
2028 /* only when matches with existing parent pointer, so that
2029 * deeper child structures are not touched, i.e.
2030 * update of x, and a.x, b.x, f.b.x, g.b.x, c.x, y
2031 * gets to update a.x, b.x and c.x */
2032 lock_rw_wrlock(&p->lock);
2033 if(p->parent == match)
2034 p->parent = newp;
2035 lock_rw_unlock(&p->lock);
2036 p = (struct local_zone*)rbtree_next(&p->node);
2037 }
2038 }
2039
local_zones_add_zone(struct local_zones * zones,uint8_t * name,size_t len,int labs,uint16_t dclass,enum localzone_type tp)2040 struct local_zone* local_zones_add_zone(struct local_zones* zones,
2041 uint8_t* name, size_t len, int labs, uint16_t dclass,
2042 enum localzone_type tp)
2043 {
2044 int exact;
2045 /* create */
2046 struct local_zone *prev;
2047 struct local_zone* z = local_zone_create(name, len, labs, tp, dclass);
2048 if(!z) {
2049 free(name);
2050 return NULL;
2051 }
2052 lock_rw_wrlock(&z->lock);
2053
2054 /* find the closest parent */
2055 prev = local_zones_find_le(zones, name, len, labs, dclass, &exact);
2056 if(!exact)
2057 z->parent = find_closest_parent(z, prev);
2058
2059 /* insert into the tree */
2060 if(exact||!rbtree_insert(&zones->ztree, &z->node)) {
2061 /* duplicate entry! */
2062 lock_rw_unlock(&z->lock);
2063 local_zone_delete(z);
2064 log_err("internal: duplicate entry in local_zones_add_zone");
2065 return NULL;
2066 }
2067
2068 /* set parent pointers right */
2069 set_kiddo_parents(z, z->parent, z);
2070
2071 lock_rw_unlock(&z->lock);
2072 return z;
2073 }
2074
local_zones_del_zone(struct local_zones * zones,struct local_zone * z)2075 void local_zones_del_zone(struct local_zones* zones, struct local_zone* z)
2076 {
2077 /* fix up parents in tree */
2078 lock_rw_wrlock(&z->lock);
2079 set_kiddo_parents(z, z, z->parent);
2080
2081 /* remove from tree */
2082 (void)rbtree_delete(&zones->ztree, z);
2083
2084 /* delete the zone */
2085 lock_rw_unlock(&z->lock);
2086 local_zone_delete(z);
2087 }
2088
2089 int
local_zones_add_RR(struct local_zones * zones,const char * rr)2090 local_zones_add_RR(struct local_zones* zones, const char* rr)
2091 {
2092 uint8_t* rr_name;
2093 uint16_t rr_class, rr_type;
2094 size_t len;
2095 int labs;
2096 struct local_zone* z;
2097 int r;
2098 if(!get_rr_nameclass(rr, &rr_name, &rr_class, &rr_type)) {
2099 return 0;
2100 }
2101 labs = dname_count_size_labels(rr_name, &len);
2102 /* could first try readlock then get writelock if zone does not exist,
2103 * but we do not add enough RRs (from multiple threads) to optimize */
2104 lock_rw_wrlock(&zones->lock);
2105 z = local_zones_lookup(zones, rr_name, len, labs, rr_class, rr_type);
2106 if(!z) {
2107 z = local_zones_add_zone(zones, rr_name, len, labs, rr_class,
2108 local_zone_transparent);
2109 if(!z) {
2110 lock_rw_unlock(&zones->lock);
2111 return 0;
2112 }
2113 } else {
2114 free(rr_name);
2115 }
2116 lock_rw_wrlock(&z->lock);
2117 lock_rw_unlock(&zones->lock);
2118 r = lz_enter_rr_into_zone(z, rr);
2119 lock_rw_unlock(&z->lock);
2120 return r;
2121 }
2122
2123 /** returns true if the node is terminal so no deeper domain names exist */
2124 static int
is_terminal(struct local_data * d)2125 is_terminal(struct local_data* d)
2126 {
2127 /* for empty nonterminals, the deeper domain names are sorted
2128 * right after them, so simply check the next name in the tree
2129 */
2130 struct local_data* n = (struct local_data*)rbtree_next(&d->node);
2131 if(n == (struct local_data*)RBTREE_NULL)
2132 return 1; /* last in tree, no deeper node */
2133 if(dname_strict_subdomain(n->name, n->namelabs, d->name, d->namelabs))
2134 return 0; /* there is a deeper node */
2135 return 1;
2136 }
2137
2138 /** delete empty terminals from tree when final data is deleted */
2139 static void
del_empty_term(struct local_zone * z,struct local_data * d,uint8_t * name,size_t len,int labs)2140 del_empty_term(struct local_zone* z, struct local_data* d,
2141 uint8_t* name, size_t len, int labs)
2142 {
2143 while(d && d->rrsets == NULL && is_terminal(d)) {
2144 /* is this empty nonterminal? delete */
2145 /* note, no memory recycling in zone region */
2146 (void)rbtree_delete(&z->data, d);
2147
2148 /* go up and to the next label */
2149 if(dname_is_root(name))
2150 return;
2151 dname_remove_label(&name, &len);
2152 labs--;
2153 d = local_zone_find_data(z, name, len, labs);
2154 }
2155 }
2156
2157 /** find and remove type from list in domain struct */
2158 static void
del_local_rrset(struct local_data * d,uint16_t dtype)2159 del_local_rrset(struct local_data* d, uint16_t dtype)
2160 {
2161 struct local_rrset* prev=NULL, *p=d->rrsets;
2162 while(p && ntohs(p->rrset->rk.type) != dtype) {
2163 prev = p;
2164 p = p->next;
2165 }
2166 if(!p)
2167 return; /* rrset type not found */
2168 /* unlink it */
2169 if(prev) prev->next = p->next;
2170 else d->rrsets = p->next;
2171 /* no memory recycling for zone deletions ... */
2172 }
2173
local_zones_del_data(struct local_zones * zones,uint8_t * name,size_t len,int labs,uint16_t dclass)2174 void local_zones_del_data(struct local_zones* zones,
2175 uint8_t* name, size_t len, int labs, uint16_t dclass)
2176 {
2177 /* find zone */
2178 struct local_zone* z;
2179 struct local_data* d;
2180
2181 /* remove DS */
2182 lock_rw_rdlock(&zones->lock);
2183 z = local_zones_lookup(zones, name, len, labs, dclass, LDNS_RR_TYPE_DS);
2184 if(z) {
2185 lock_rw_wrlock(&z->lock);
2186 d = local_zone_find_data(z, name, len, labs);
2187 if(d) {
2188 del_local_rrset(d, LDNS_RR_TYPE_DS);
2189 del_empty_term(z, d, name, len, labs);
2190 }
2191 lock_rw_unlock(&z->lock);
2192 }
2193 lock_rw_unlock(&zones->lock);
2194
2195 /* remove other types */
2196 lock_rw_rdlock(&zones->lock);
2197 z = local_zones_lookup(zones, name, len, labs, dclass, 0);
2198 if(!z) {
2199 /* no such zone, we're done */
2200 lock_rw_unlock(&zones->lock);
2201 return;
2202 }
2203 lock_rw_wrlock(&z->lock);
2204 lock_rw_unlock(&zones->lock);
2205
2206 /* find the domain */
2207 d = local_zone_find_data(z, name, len, labs);
2208 if(d) {
2209 /* no memory recycling for zone deletions ... */
2210 d->rrsets = NULL;
2211 /* did we delete the soa record ? */
2212 if(query_dname_compare(d->name, z->name) == 0) {
2213 z->soa = NULL;
2214 z->soa_negative = NULL;
2215 }
2216
2217 /* cleanup the empty nonterminals for this name */
2218 del_empty_term(z, d, name, len, labs);
2219 }
2220
2221 lock_rw_unlock(&z->lock);
2222 }
2223
2224 /** Get memory usage for local_zone */
2225 static size_t
local_zone_get_mem(struct local_zone * z)2226 local_zone_get_mem(struct local_zone* z)
2227 {
2228 size_t m = sizeof(*z);
2229 lock_rw_rdlock(&z->lock);
2230 m += z->namelen + z->taglen + regional_get_mem(z->region);
2231 lock_rw_unlock(&z->lock);
2232 return m;
2233 }
2234
local_zones_get_mem(struct local_zones * zones)2235 size_t local_zones_get_mem(struct local_zones* zones)
2236 {
2237 struct local_zone* z;
2238 size_t m;
2239 if(!zones) return 0;
2240 m = sizeof(*zones);
2241 lock_rw_rdlock(&zones->lock);
2242 RBTREE_FOR(z, struct local_zone*, &zones->ztree) {
2243 m += local_zone_get_mem(z);
2244 }
2245 lock_rw_unlock(&zones->lock);
2246 return m;
2247 }
2248
local_zones_swap_tree(struct local_zones * zones,struct local_zones * data)2249 void local_zones_swap_tree(struct local_zones* zones, struct local_zones* data)
2250 {
2251 rbtree_type oldtree = zones->ztree;
2252 zones->ztree = data->ztree;
2253 data->ztree = oldtree;
2254 }
2255