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