xref: /freebsd/contrib/unbound/services/rpz.c (revision a39a5a6905612447def27b66ffe73b9d11efd80c)
1091e9e46SCy Schubert /*
2091e9e46SCy Schubert  * services/rpz.c - rpz service
3091e9e46SCy Schubert  *
4091e9e46SCy Schubert  * Copyright (c) 2019, NLnet Labs. All rights reserved.
5091e9e46SCy Schubert  *
6091e9e46SCy Schubert  * This software is open source.
7091e9e46SCy Schubert  *
8091e9e46SCy Schubert  * Redistribution and use in source and binary forms, with or without
9091e9e46SCy Schubert  * modification, are permitted provided that the following conditions
10091e9e46SCy Schubert  * are met:
11091e9e46SCy Schubert  *
12091e9e46SCy Schubert  * Redistributions of source code must retain the above copyright notice,
13091e9e46SCy Schubert  * this list of conditions and the following disclaimer.
14091e9e46SCy Schubert  *
15091e9e46SCy Schubert  * Redistributions in binary form must reproduce the above copyright notice,
16091e9e46SCy Schubert  * this list of conditions and the following disclaimer in the documentation
17091e9e46SCy Schubert  * and/or other materials provided with the distribution.
18091e9e46SCy Schubert  *
19091e9e46SCy Schubert  * Neither the name of the NLNET LABS nor the names of its contributors may
20091e9e46SCy Schubert  * be used to endorse or promote products derived from this software without
21091e9e46SCy Schubert  * specific prior written permission.
22091e9e46SCy Schubert  *
23091e9e46SCy Schubert  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24091e9e46SCy Schubert  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25091e9e46SCy Schubert  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26091e9e46SCy Schubert  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27091e9e46SCy Schubert  * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28091e9e46SCy Schubert  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
29091e9e46SCy Schubert  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
30091e9e46SCy Schubert  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
31091e9e46SCy Schubert  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
32091e9e46SCy Schubert  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
33091e9e46SCy Schubert  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34091e9e46SCy Schubert  */
35091e9e46SCy Schubert 
36091e9e46SCy Schubert /**
37091e9e46SCy Schubert  * \file
38091e9e46SCy Schubert  *
39091e9e46SCy Schubert  * This file contains functions to enable RPZ service.
40091e9e46SCy Schubert  */
41091e9e46SCy Schubert 
42091e9e46SCy Schubert #include "config.h"
43091e9e46SCy Schubert #include "services/rpz.h"
44091e9e46SCy Schubert #include "util/config_file.h"
45091e9e46SCy Schubert #include "sldns/wire2str.h"
46091e9e46SCy Schubert #include "sldns/str2wire.h"
47091e9e46SCy Schubert #include "util/data/dname.h"
48091e9e46SCy Schubert #include "util/net_help.h"
49091e9e46SCy Schubert #include "util/log.h"
50091e9e46SCy Schubert #include "util/data/dname.h"
51091e9e46SCy Schubert #include "util/locks.h"
52091e9e46SCy Schubert #include "util/regional.h"
5324e36522SCy Schubert #include "util/data/msgencode.h"
5424e36522SCy Schubert #include "services/cache/dns.h"
5524e36522SCy Schubert #include "iterator/iterator.h"
5624e36522SCy Schubert #include "iterator/iter_delegpt.h"
5724e36522SCy Schubert #include "daemon/worker.h"
5824e36522SCy Schubert 
5924e36522SCy Schubert typedef struct resp_addr rpz_aclnode_type;
6024e36522SCy Schubert 
6124e36522SCy Schubert struct matched_delegation_point {
6224e36522SCy Schubert 	uint8_t* dname;
6324e36522SCy Schubert 	size_t dname_len;
6424e36522SCy Schubert };
65091e9e46SCy Schubert 
66091e9e46SCy Schubert /** string for RPZ action enum */
67091e9e46SCy Schubert const char*
68091e9e46SCy Schubert rpz_action_to_string(enum rpz_action a)
69091e9e46SCy Schubert {
70091e9e46SCy Schubert 	switch(a) {
7124e36522SCy Schubert 	case RPZ_NXDOMAIN_ACTION: return "rpz-nxdomain";
7224e36522SCy Schubert 	case RPZ_NODATA_ACTION: return "rpz-nodata";
7324e36522SCy Schubert 	case RPZ_PASSTHRU_ACTION: return "rpz-passthru";
7424e36522SCy Schubert 	case RPZ_DROP_ACTION: return "rpz-drop";
7524e36522SCy Schubert 	case RPZ_TCP_ONLY_ACTION: return "rpz-tcp-only";
7624e36522SCy Schubert 	case RPZ_INVALID_ACTION: return "rpz-invalid";
7724e36522SCy Schubert 	case RPZ_LOCAL_DATA_ACTION: return "rpz-local-data";
7824e36522SCy Schubert 	case RPZ_DISABLED_ACTION: return "rpz-disabled";
7924e36522SCy Schubert 	case RPZ_CNAME_OVERRIDE_ACTION: return "rpz-cname-override";
8024e36522SCy Schubert 	case RPZ_NO_OVERRIDE_ACTION: return "rpz-no-override";
8124e36522SCy Schubert 	default: return "rpz-unknown-action";
82091e9e46SCy Schubert 	}
83091e9e46SCy Schubert }
84091e9e46SCy Schubert 
85091e9e46SCy Schubert /** RPZ action enum for config string */
86091e9e46SCy Schubert static enum rpz_action
87091e9e46SCy Schubert rpz_config_to_action(char* a)
88091e9e46SCy Schubert {
8924e36522SCy Schubert 	if(strcmp(a, "nxdomain") == 0) return RPZ_NXDOMAIN_ACTION;
9024e36522SCy Schubert 	else if(strcmp(a, "nodata") == 0) return RPZ_NODATA_ACTION;
9124e36522SCy Schubert 	else if(strcmp(a, "passthru") == 0) return RPZ_PASSTHRU_ACTION;
9224e36522SCy Schubert 	else if(strcmp(a, "drop") == 0) return RPZ_DROP_ACTION;
9324e36522SCy Schubert 	else if(strcmp(a, "tcp_only") == 0) return RPZ_TCP_ONLY_ACTION;
9424e36522SCy Schubert 	else if(strcmp(a, "cname") == 0) return RPZ_CNAME_OVERRIDE_ACTION;
9524e36522SCy Schubert 	else if(strcmp(a, "disabled") == 0) return RPZ_DISABLED_ACTION;
9624e36522SCy Schubert 	else return RPZ_INVALID_ACTION;
97091e9e46SCy Schubert }
98091e9e46SCy Schubert 
99091e9e46SCy Schubert /** string for RPZ trigger enum */
100091e9e46SCy Schubert static const char*
101091e9e46SCy Schubert rpz_trigger_to_string(enum rpz_trigger r)
102091e9e46SCy Schubert {
103091e9e46SCy Schubert 	switch(r) {
10424e36522SCy Schubert 	case RPZ_QNAME_TRIGGER: return "rpz-qname";
10524e36522SCy Schubert 	case RPZ_CLIENT_IP_TRIGGER: return "rpz-client-ip";
10624e36522SCy Schubert 	case RPZ_RESPONSE_IP_TRIGGER: return "rpz-response-ip";
10724e36522SCy Schubert 	case RPZ_NSDNAME_TRIGGER: return "rpz-nsdname";
10824e36522SCy Schubert 	case RPZ_NSIP_TRIGGER: return "rpz-nsip";
10924e36522SCy Schubert 	case RPZ_INVALID_TRIGGER: return "rpz-invalid";
11024e36522SCy Schubert 	default: return "rpz-unknown-trigger";
111091e9e46SCy Schubert 	}
112091e9e46SCy Schubert }
113091e9e46SCy Schubert 
114091e9e46SCy Schubert /**
115091e9e46SCy Schubert  * Get the label that is just before the root label.
116091e9e46SCy Schubert  * @param dname: dname to work on
117091e9e46SCy Schubert  * @param maxdnamelen: maximum length of the dname
118091e9e46SCy Schubert  * @return: pointer to TLD label, NULL if not found or invalid dname
119091e9e46SCy Schubert  */
120091e9e46SCy Schubert static uint8_t*
121091e9e46SCy Schubert get_tld_label(uint8_t* dname, size_t maxdnamelen)
122091e9e46SCy Schubert {
123091e9e46SCy Schubert 	uint8_t* prevlab = dname;
124091e9e46SCy Schubert 	size_t dnamelen = 0;
125091e9e46SCy Schubert 
126091e9e46SCy Schubert 	/* one byte needed for label length */
127091e9e46SCy Schubert 	if(dnamelen+1 > maxdnamelen)
128091e9e46SCy Schubert 		return NULL;
129091e9e46SCy Schubert 
130091e9e46SCy Schubert 	/* only root label */
131091e9e46SCy Schubert 	if(*dname == 0)
132091e9e46SCy Schubert 		return NULL;
133091e9e46SCy Schubert 
134091e9e46SCy Schubert 	while(*dname) {
135091e9e46SCy Schubert 		dnamelen += ((size_t)*dname)+1;
136091e9e46SCy Schubert 		if(dnamelen+1 > maxdnamelen)
137091e9e46SCy Schubert 			return NULL;
138091e9e46SCy Schubert 		dname = dname+((size_t)*dname)+1;
139091e9e46SCy Schubert 		if(*dname != 0)
140091e9e46SCy Schubert 			prevlab = dname;
141091e9e46SCy Schubert 	}
142091e9e46SCy Schubert 	return prevlab;
143091e9e46SCy Schubert }
144091e9e46SCy Schubert 
145091e9e46SCy Schubert /**
14624e36522SCy Schubert  * The RR types that are to be ignored.
14724e36522SCy Schubert  * DNSSEC RRs at the apex, and SOA and NS are ignored.
14824e36522SCy Schubert  */
14924e36522SCy Schubert static int
15024e36522SCy Schubert rpz_type_ignored(uint16_t rr_type)
15124e36522SCy Schubert {
15224e36522SCy Schubert 	switch(rr_type) {
15324e36522SCy Schubert 		case LDNS_RR_TYPE_SOA:
15424e36522SCy Schubert 		case LDNS_RR_TYPE_NS:
15524e36522SCy Schubert 		case LDNS_RR_TYPE_DNAME:
15624e36522SCy Schubert 		/* all DNSSEC-related RRs must be ignored */
15724e36522SCy Schubert 		case LDNS_RR_TYPE_DNSKEY:
15824e36522SCy Schubert 		case LDNS_RR_TYPE_DS:
15924e36522SCy Schubert 		case LDNS_RR_TYPE_RRSIG:
16024e36522SCy Schubert 		case LDNS_RR_TYPE_NSEC:
16124e36522SCy Schubert 		case LDNS_RR_TYPE_NSEC3:
16224e36522SCy Schubert 		case LDNS_RR_TYPE_NSEC3PARAM:
16324e36522SCy Schubert 			return 1;
16424e36522SCy Schubert 		default:
16524e36522SCy Schubert 			break;
16624e36522SCy Schubert 	}
16724e36522SCy Schubert 	return 0;
16824e36522SCy Schubert }
16924e36522SCy Schubert 
17024e36522SCy Schubert /**
171091e9e46SCy Schubert  * Classify RPZ action for RR type/rdata
172091e9e46SCy Schubert  * @param rr_type: the RR type
173091e9e46SCy Schubert  * @param rdatawl: RDATA with 2 bytes length
174091e9e46SCy Schubert  * @param rdatalen: the length of rdatawl (including its 2 bytes length)
175091e9e46SCy Schubert  * @return: the RPZ action
176091e9e46SCy Schubert  */
177091e9e46SCy Schubert static enum rpz_action
178091e9e46SCy Schubert rpz_rr_to_action(uint16_t rr_type, uint8_t* rdatawl, size_t rdatalen)
179091e9e46SCy Schubert {
180091e9e46SCy Schubert 	char* endptr;
181091e9e46SCy Schubert 	uint8_t* rdata;
182091e9e46SCy Schubert 	int rdatalabs;
183091e9e46SCy Schubert 	uint8_t* tldlab = NULL;
184091e9e46SCy Schubert 
185091e9e46SCy Schubert 	switch(rr_type) {
186091e9e46SCy Schubert 		case LDNS_RR_TYPE_SOA:
187091e9e46SCy Schubert 		case LDNS_RR_TYPE_NS:
188091e9e46SCy Schubert 		case LDNS_RR_TYPE_DNAME:
189091e9e46SCy Schubert 		/* all DNSSEC-related RRs must be ignored */
190091e9e46SCy Schubert 		case LDNS_RR_TYPE_DNSKEY:
191091e9e46SCy Schubert 		case LDNS_RR_TYPE_DS:
192091e9e46SCy Schubert 		case LDNS_RR_TYPE_RRSIG:
193091e9e46SCy Schubert 		case LDNS_RR_TYPE_NSEC:
194091e9e46SCy Schubert 		case LDNS_RR_TYPE_NSEC3:
1955469a995SCy Schubert 		case LDNS_RR_TYPE_NSEC3PARAM:
196091e9e46SCy Schubert 			return RPZ_INVALID_ACTION;
197091e9e46SCy Schubert 		case LDNS_RR_TYPE_CNAME:
198091e9e46SCy Schubert 			break;
199091e9e46SCy Schubert 		default:
200091e9e46SCy Schubert 			return RPZ_LOCAL_DATA_ACTION;
201091e9e46SCy Schubert 	}
202091e9e46SCy Schubert 
203091e9e46SCy Schubert 	/* use CNAME target to determine RPZ action */
204091e9e46SCy Schubert 	log_assert(rr_type == LDNS_RR_TYPE_CNAME);
205091e9e46SCy Schubert 	if(rdatalen < 3)
206091e9e46SCy Schubert 		return RPZ_INVALID_ACTION;
207091e9e46SCy Schubert 
208091e9e46SCy Schubert 	rdata = rdatawl + 2; /* 2 bytes of rdata length */
209091e9e46SCy Schubert 	if(dname_valid(rdata, rdatalen-2) != rdatalen-2)
210091e9e46SCy Schubert 		return RPZ_INVALID_ACTION;
211091e9e46SCy Schubert 
212091e9e46SCy Schubert 	rdatalabs = dname_count_labels(rdata);
213091e9e46SCy Schubert 	if(rdatalabs == 1)
214091e9e46SCy Schubert 		return RPZ_NXDOMAIN_ACTION;
215091e9e46SCy Schubert 	else if(rdatalabs == 2) {
216091e9e46SCy Schubert 		if(dname_subdomain_c(rdata, (uint8_t*)&"\001*\000"))
217091e9e46SCy Schubert 			return RPZ_NODATA_ACTION;
218091e9e46SCy Schubert 		else if(dname_subdomain_c(rdata,
219091e9e46SCy Schubert 			(uint8_t*)&"\014rpz-passthru\000"))
220091e9e46SCy Schubert 			return RPZ_PASSTHRU_ACTION;
221091e9e46SCy Schubert 		else if(dname_subdomain_c(rdata, (uint8_t*)&"\010rpz-drop\000"))
222091e9e46SCy Schubert 			return RPZ_DROP_ACTION;
223091e9e46SCy Schubert 		else if(dname_subdomain_c(rdata,
224091e9e46SCy Schubert 			(uint8_t*)&"\014rpz-tcp-only\000"))
225091e9e46SCy Schubert 			return RPZ_TCP_ONLY_ACTION;
226091e9e46SCy Schubert 	}
227091e9e46SCy Schubert 
228091e9e46SCy Schubert 	/* all other TLDs starting with "rpz-" are invalid */
229091e9e46SCy Schubert 	tldlab = get_tld_label(rdata, rdatalen-2);
230091e9e46SCy Schubert 	if(tldlab && dname_lab_startswith(tldlab, "rpz-", &endptr))
231091e9e46SCy Schubert 		return RPZ_INVALID_ACTION;
232091e9e46SCy Schubert 
233091e9e46SCy Schubert 	/* no special label found */
234091e9e46SCy Schubert 	return RPZ_LOCAL_DATA_ACTION;
235091e9e46SCy Schubert }
236091e9e46SCy Schubert 
237091e9e46SCy Schubert static enum localzone_type
238091e9e46SCy Schubert rpz_action_to_localzone_type(enum rpz_action a)
239091e9e46SCy Schubert {
240091e9e46SCy Schubert 	switch(a) {
241091e9e46SCy Schubert 	case RPZ_NXDOMAIN_ACTION: return local_zone_always_nxdomain;
242091e9e46SCy Schubert 	case RPZ_NODATA_ACTION: return local_zone_always_nodata;
243091e9e46SCy Schubert 	case RPZ_DROP_ACTION: return local_zone_always_deny;
244091e9e46SCy Schubert 	case RPZ_PASSTHRU_ACTION: return local_zone_always_transparent;
245091e9e46SCy Schubert 	case RPZ_LOCAL_DATA_ACTION:	/* fallthrough */
246091e9e46SCy Schubert 	case RPZ_CNAME_OVERRIDE_ACTION: return local_zone_redirect;
24724e36522SCy Schubert 	case RPZ_TCP_ONLY_ACTION: return local_zone_truncate;
248091e9e46SCy Schubert 	case RPZ_INVALID_ACTION: /* fallthrough */
249091e9e46SCy Schubert 	default: return local_zone_invalid;
250091e9e46SCy Schubert 	}
251091e9e46SCy Schubert }
252091e9e46SCy Schubert 
253091e9e46SCy Schubert enum respip_action
254091e9e46SCy Schubert rpz_action_to_respip_action(enum rpz_action a)
255091e9e46SCy Schubert {
256091e9e46SCy Schubert 	switch(a) {
257091e9e46SCy Schubert 	case RPZ_NXDOMAIN_ACTION: return respip_always_nxdomain;
258091e9e46SCy Schubert 	case RPZ_NODATA_ACTION: return respip_always_nodata;
259091e9e46SCy Schubert 	case RPZ_DROP_ACTION: return respip_always_deny;
260091e9e46SCy Schubert 	case RPZ_PASSTHRU_ACTION: return respip_always_transparent;
261091e9e46SCy Schubert 	case RPZ_LOCAL_DATA_ACTION: /* fallthrough */
262091e9e46SCy Schubert 	case RPZ_CNAME_OVERRIDE_ACTION: return respip_redirect;
26324e36522SCy Schubert 	case RPZ_TCP_ONLY_ACTION: return respip_truncate;
264091e9e46SCy Schubert 	case RPZ_INVALID_ACTION: /* fallthrough */
265091e9e46SCy Schubert 	default: return respip_invalid;
266091e9e46SCy Schubert 	}
267091e9e46SCy Schubert }
268091e9e46SCy Schubert 
269091e9e46SCy Schubert static enum rpz_action
270091e9e46SCy Schubert localzone_type_to_rpz_action(enum localzone_type lzt)
271091e9e46SCy Schubert {
272091e9e46SCy Schubert 	switch(lzt) {
273091e9e46SCy Schubert 	case local_zone_always_nxdomain: return RPZ_NXDOMAIN_ACTION;
274091e9e46SCy Schubert 	case local_zone_always_nodata: return RPZ_NODATA_ACTION;
275091e9e46SCy Schubert 	case local_zone_always_deny: return RPZ_DROP_ACTION;
276091e9e46SCy Schubert 	case local_zone_always_transparent: return RPZ_PASSTHRU_ACTION;
277091e9e46SCy Schubert 	case local_zone_redirect: return RPZ_LOCAL_DATA_ACTION;
27824e36522SCy Schubert 	case local_zone_truncate: return RPZ_TCP_ONLY_ACTION;
27924e36522SCy Schubert 	case local_zone_invalid: /* fallthrough */
28024e36522SCy Schubert 	default: return RPZ_INVALID_ACTION;
281091e9e46SCy Schubert 	}
282091e9e46SCy Schubert }
283091e9e46SCy Schubert 
284091e9e46SCy Schubert enum rpz_action
285091e9e46SCy Schubert respip_action_to_rpz_action(enum respip_action a)
286091e9e46SCy Schubert {
287091e9e46SCy Schubert 	switch(a) {
288091e9e46SCy Schubert 	case respip_always_nxdomain: return RPZ_NXDOMAIN_ACTION;
289091e9e46SCy Schubert 	case respip_always_nodata: return RPZ_NODATA_ACTION;
290091e9e46SCy Schubert 	case respip_always_deny: return RPZ_DROP_ACTION;
291091e9e46SCy Schubert 	case respip_always_transparent: return RPZ_PASSTHRU_ACTION;
292091e9e46SCy Schubert 	case respip_redirect: return RPZ_LOCAL_DATA_ACTION;
29324e36522SCy Schubert 	case respip_truncate: return RPZ_TCP_ONLY_ACTION;
29424e36522SCy Schubert 	case respip_invalid: /* fallthrough */
29524e36522SCy Schubert 	default: return RPZ_INVALID_ACTION;
296091e9e46SCy Schubert 	}
297091e9e46SCy Schubert }
298091e9e46SCy Schubert 
299091e9e46SCy Schubert /**
300091e9e46SCy Schubert  * Get RPZ trigger for dname
301091e9e46SCy Schubert  * @param dname: dname containing RPZ trigger
302091e9e46SCy Schubert  * @param dname_len: length of the dname
303091e9e46SCy Schubert  * @return: RPZ trigger enum
304091e9e46SCy Schubert  */
305091e9e46SCy Schubert static enum rpz_trigger
306091e9e46SCy Schubert rpz_dname_to_trigger(uint8_t* dname, size_t dname_len)
307091e9e46SCy Schubert {
308091e9e46SCy Schubert 	uint8_t* tldlab;
309091e9e46SCy Schubert 	char* endptr;
310091e9e46SCy Schubert 
311091e9e46SCy Schubert 	if(dname_valid(dname, dname_len) != dname_len)
312091e9e46SCy Schubert 		return RPZ_INVALID_TRIGGER;
313091e9e46SCy Schubert 
314091e9e46SCy Schubert 	tldlab = get_tld_label(dname, dname_len);
315091e9e46SCy Schubert 	if(!tldlab || !dname_lab_startswith(tldlab, "rpz-", &endptr))
316091e9e46SCy Schubert 		return RPZ_QNAME_TRIGGER;
317091e9e46SCy Schubert 
318091e9e46SCy Schubert 	if(dname_subdomain_c(tldlab,
319091e9e46SCy Schubert 		(uint8_t*)&"\015rpz-client-ip\000"))
320091e9e46SCy Schubert 		return RPZ_CLIENT_IP_TRIGGER;
321091e9e46SCy Schubert 	else if(dname_subdomain_c(tldlab, (uint8_t*)&"\006rpz-ip\000"))
322091e9e46SCy Schubert 		return RPZ_RESPONSE_IP_TRIGGER;
323091e9e46SCy Schubert 	else if(dname_subdomain_c(tldlab, (uint8_t*)&"\013rpz-nsdname\000"))
324091e9e46SCy Schubert 		return RPZ_NSDNAME_TRIGGER;
325091e9e46SCy Schubert 	else if(dname_subdomain_c(tldlab, (uint8_t*)&"\010rpz-nsip\000"))
326091e9e46SCy Schubert 		return RPZ_NSIP_TRIGGER;
327091e9e46SCy Schubert 
328091e9e46SCy Schubert 	return RPZ_QNAME_TRIGGER;
329091e9e46SCy Schubert }
330091e9e46SCy Schubert 
33124e36522SCy Schubert static inline struct clientip_synthesized_rrset*
33224e36522SCy Schubert rpz_clientip_synthesized_set_create(void)
33324e36522SCy Schubert {
33424e36522SCy Schubert 	struct clientip_synthesized_rrset* set = calloc(1, sizeof(*set));
33524e36522SCy Schubert 	if(set == NULL) {
33624e36522SCy Schubert 		return NULL;
33724e36522SCy Schubert 	}
33824e36522SCy Schubert 	set->region = regional_create();
33924e36522SCy Schubert 	if(set->region == NULL) {
34024e36522SCy Schubert 		free(set);
34124e36522SCy Schubert 		return NULL;
34224e36522SCy Schubert 	}
34324e36522SCy Schubert 	addr_tree_init(&set->entries);
34424e36522SCy Schubert 	lock_rw_init(&set->lock);
34524e36522SCy Schubert 	return set;
34624e36522SCy Schubert }
34724e36522SCy Schubert 
34824e36522SCy Schubert static void
34924e36522SCy Schubert rpz_clientip_synthesized_rr_delete(rbnode_type* n, void* ATTR_UNUSED(arg))
35024e36522SCy Schubert {
35124e36522SCy Schubert 	struct clientip_synthesized_rr* r = (struct clientip_synthesized_rr*)n->key;
35224e36522SCy Schubert 	lock_rw_destroy(&r->lock);
35324e36522SCy Schubert #ifdef THREADS_DISABLED
35424e36522SCy Schubert 	(void)r;
35524e36522SCy Schubert #endif
35624e36522SCy Schubert }
35724e36522SCy Schubert 
35824e36522SCy Schubert static inline void
35924e36522SCy Schubert rpz_clientip_synthesized_set_delete(struct clientip_synthesized_rrset* set)
36024e36522SCy Schubert {
36124e36522SCy Schubert 	if(set == NULL) {
36224e36522SCy Schubert 		return;
36324e36522SCy Schubert 	}
36424e36522SCy Schubert 	lock_rw_destroy(&set->lock);
36524e36522SCy Schubert 	traverse_postorder(&set->entries, rpz_clientip_synthesized_rr_delete, NULL);
36624e36522SCy Schubert 	regional_destroy(set->region);
36724e36522SCy Schubert 	free(set);
36824e36522SCy Schubert }
36924e36522SCy Schubert 
37024e36522SCy Schubert void
37124e36522SCy Schubert rpz_delete(struct rpz* r)
372091e9e46SCy Schubert {
373091e9e46SCy Schubert 	if(!r)
374091e9e46SCy Schubert 		return;
375091e9e46SCy Schubert 	local_zones_delete(r->local_zones);
37624e36522SCy Schubert 	local_zones_delete(r->nsdname_zones);
377091e9e46SCy Schubert 	respip_set_delete(r->respip_set);
37824e36522SCy Schubert 	rpz_clientip_synthesized_set_delete(r->client_set);
37924e36522SCy Schubert 	rpz_clientip_synthesized_set_delete(r->ns_set);
380091e9e46SCy Schubert 	regional_destroy(r->region);
381091e9e46SCy Schubert 	free(r->taglist);
382091e9e46SCy Schubert 	free(r->log_name);
383091e9e46SCy Schubert 	free(r);
384091e9e46SCy Schubert }
385091e9e46SCy Schubert 
386091e9e46SCy Schubert int
387091e9e46SCy Schubert rpz_clear(struct rpz* r)
388091e9e46SCy Schubert {
389091e9e46SCy Schubert 	/* must hold write lock on auth_zone */
390091e9e46SCy Schubert 	local_zones_delete(r->local_zones);
39124e36522SCy Schubert 	r->local_zones = NULL;
39224e36522SCy Schubert 	local_zones_delete(r->nsdname_zones);
39324e36522SCy Schubert 	r->nsdname_zones = NULL;
394091e9e46SCy Schubert 	respip_set_delete(r->respip_set);
39524e36522SCy Schubert 	r->respip_set = NULL;
39624e36522SCy Schubert 	rpz_clientip_synthesized_set_delete(r->client_set);
39724e36522SCy Schubert 	r->client_set = NULL;
39824e36522SCy Schubert 	rpz_clientip_synthesized_set_delete(r->ns_set);
39924e36522SCy Schubert 	r->ns_set = NULL;
400091e9e46SCy Schubert 	if(!(r->local_zones = local_zones_create())){
401091e9e46SCy Schubert 		return 0;
402091e9e46SCy Schubert 	}
40324e36522SCy Schubert 	r->nsdname_zones = local_zones_create();
40424e36522SCy Schubert 	if(r->nsdname_zones == NULL) {
40524e36522SCy Schubert 		return 0;
40624e36522SCy Schubert 	}
407091e9e46SCy Schubert 	if(!(r->respip_set = respip_set_create())) {
408091e9e46SCy Schubert 		return 0;
409091e9e46SCy Schubert 	}
41024e36522SCy Schubert 	if(!(r->client_set = rpz_clientip_synthesized_set_create())) {
41124e36522SCy Schubert 		return 0;
41224e36522SCy Schubert 	}
41324e36522SCy Schubert 	if(!(r->ns_set = rpz_clientip_synthesized_set_create())) {
41424e36522SCy Schubert 		return 0;
41524e36522SCy Schubert 	}
416091e9e46SCy Schubert 	return 1;
417091e9e46SCy Schubert }
418091e9e46SCy Schubert 
419091e9e46SCy Schubert void
420091e9e46SCy Schubert rpz_finish_config(struct rpz* r)
421091e9e46SCy Schubert {
422091e9e46SCy Schubert 	lock_rw_wrlock(&r->respip_set->lock);
423091e9e46SCy Schubert 	addr_tree_init_parents(&r->respip_set->ip_tree);
424091e9e46SCy Schubert 	lock_rw_unlock(&r->respip_set->lock);
42524e36522SCy Schubert 
42624e36522SCy Schubert 	lock_rw_wrlock(&r->client_set->lock);
42724e36522SCy Schubert 	addr_tree_init_parents(&r->client_set->entries);
42824e36522SCy Schubert 	lock_rw_unlock(&r->client_set->lock);
42924e36522SCy Schubert 
43024e36522SCy Schubert 	lock_rw_wrlock(&r->ns_set->lock);
43124e36522SCy Schubert 	addr_tree_init_parents(&r->ns_set->entries);
43224e36522SCy Schubert 	lock_rw_unlock(&r->ns_set->lock);
433091e9e46SCy Schubert }
434091e9e46SCy Schubert 
435091e9e46SCy Schubert /** new rrset containing CNAME override, does not yet contain a dname */
436091e9e46SCy Schubert static struct ub_packed_rrset_key*
437091e9e46SCy Schubert new_cname_override(struct regional* region, uint8_t* ct, size_t ctlen)
438091e9e46SCy Schubert {
439091e9e46SCy Schubert 	struct ub_packed_rrset_key* rrset;
440091e9e46SCy Schubert 	struct packed_rrset_data* pd;
441091e9e46SCy Schubert 	uint16_t rdlength = htons(ctlen);
442091e9e46SCy Schubert 	rrset = (struct ub_packed_rrset_key*)regional_alloc_zero(region,
443091e9e46SCy Schubert 		sizeof(*rrset));
444091e9e46SCy Schubert 	if(!rrset) {
445091e9e46SCy Schubert 		log_err("out of memory");
446091e9e46SCy Schubert 		return NULL;
447091e9e46SCy Schubert 	}
448091e9e46SCy Schubert 	rrset->entry.key = rrset;
449091e9e46SCy Schubert 	pd = (struct packed_rrset_data*)regional_alloc_zero(region, sizeof(*pd));
450091e9e46SCy Schubert 	if(!pd) {
451091e9e46SCy Schubert 		log_err("out of memory");
452091e9e46SCy Schubert 		return NULL;
453091e9e46SCy Schubert 	}
454091e9e46SCy Schubert 	pd->trust = rrset_trust_prim_noglue;
455091e9e46SCy Schubert 	pd->security = sec_status_insecure;
456091e9e46SCy Schubert 
457091e9e46SCy Schubert 	pd->count = 1;
458091e9e46SCy Schubert 	pd->rr_len = regional_alloc_zero(region, sizeof(*pd->rr_len));
459091e9e46SCy Schubert 	pd->rr_ttl = regional_alloc_zero(region, sizeof(*pd->rr_ttl));
460091e9e46SCy Schubert 	pd->rr_data = regional_alloc_zero(region, sizeof(*pd->rr_data));
461091e9e46SCy Schubert 	if(!pd->rr_len || !pd->rr_ttl || !pd->rr_data) {
462091e9e46SCy Schubert 		log_err("out of memory");
463091e9e46SCy Schubert 		return NULL;
464091e9e46SCy Schubert 	}
465091e9e46SCy Schubert 	pd->rr_len[0] = ctlen+2;
466091e9e46SCy Schubert 	pd->rr_ttl[0] = 3600;
467091e9e46SCy Schubert 	pd->rr_data[0] = regional_alloc_zero(region, 2 /* rdlength */ + ctlen);
468091e9e46SCy Schubert 	if(!pd->rr_data[0]) {
469091e9e46SCy Schubert 		log_err("out of memory");
470091e9e46SCy Schubert 		return NULL;
471091e9e46SCy Schubert 	}
472091e9e46SCy Schubert 	memmove(pd->rr_data[0], &rdlength, 2);
473091e9e46SCy Schubert 	memmove(pd->rr_data[0]+2, ct, ctlen);
474091e9e46SCy Schubert 
475091e9e46SCy Schubert 	rrset->entry.data = pd;
476091e9e46SCy Schubert 	rrset->rk.type = htons(LDNS_RR_TYPE_CNAME);
477091e9e46SCy Schubert 	rrset->rk.rrset_class = htons(LDNS_RR_CLASS_IN);
478091e9e46SCy Schubert 	return rrset;
479091e9e46SCy Schubert }
480091e9e46SCy Schubert 
481091e9e46SCy Schubert struct rpz*
482091e9e46SCy Schubert rpz_create(struct config_auth* p)
483091e9e46SCy Schubert {
484091e9e46SCy Schubert 	struct rpz* r = calloc(1, sizeof(*r));
485091e9e46SCy Schubert 	if(!r)
486091e9e46SCy Schubert 		goto err;
487091e9e46SCy Schubert 
488091e9e46SCy Schubert 	r->region = regional_create_custom(sizeof(struct regional));
489091e9e46SCy Schubert 	if(!r->region) {
490091e9e46SCy Schubert 		goto err;
491091e9e46SCy Schubert 	}
492091e9e46SCy Schubert 
493091e9e46SCy Schubert 	if(!(r->local_zones = local_zones_create())){
494091e9e46SCy Schubert 		goto err;
495091e9e46SCy Schubert 	}
49624e36522SCy Schubert 
49724e36522SCy Schubert 	r->nsdname_zones = local_zones_create();
49824e36522SCy Schubert 	if(r->local_zones == NULL){
49924e36522SCy Schubert 		goto err;
50024e36522SCy Schubert 	}
50124e36522SCy Schubert 
502091e9e46SCy Schubert 	if(!(r->respip_set = respip_set_create())) {
503091e9e46SCy Schubert 		goto err;
504091e9e46SCy Schubert 	}
50524e36522SCy Schubert 
50624e36522SCy Schubert 	r->client_set = rpz_clientip_synthesized_set_create();
50724e36522SCy Schubert 	if(r->client_set == NULL) {
50824e36522SCy Schubert 		goto err;
50924e36522SCy Schubert 	}
51024e36522SCy Schubert 
51124e36522SCy Schubert 	r->ns_set = rpz_clientip_synthesized_set_create();
51224e36522SCy Schubert 	if(r->ns_set == NULL) {
51324e36522SCy Schubert 		goto err;
51424e36522SCy Schubert 	}
51524e36522SCy Schubert 
516091e9e46SCy Schubert 	r->taglistlen = p->rpz_taglistlen;
517091e9e46SCy Schubert 	r->taglist = memdup(p->rpz_taglist, r->taglistlen);
518091e9e46SCy Schubert 	if(p->rpz_action_override) {
519091e9e46SCy Schubert 		r->action_override = rpz_config_to_action(p->rpz_action_override);
520091e9e46SCy Schubert 	}
521091e9e46SCy Schubert 	else
522091e9e46SCy Schubert 		r->action_override = RPZ_NO_OVERRIDE_ACTION;
523091e9e46SCy Schubert 
524091e9e46SCy Schubert 	if(r->action_override == RPZ_CNAME_OVERRIDE_ACTION) {
525091e9e46SCy Schubert 		uint8_t nm[LDNS_MAX_DOMAINLEN+1];
526091e9e46SCy Schubert 		size_t nmlen = sizeof(nm);
527091e9e46SCy Schubert 
528091e9e46SCy Schubert 		if(!p->rpz_cname) {
529*a39a5a69SCy Schubert 			log_err("rpz: override with cname action found, but no "
530091e9e46SCy Schubert 				"rpz-cname-override configured");
531091e9e46SCy Schubert 			goto err;
532091e9e46SCy Schubert 		}
533091e9e46SCy Schubert 
534091e9e46SCy Schubert 		if(sldns_str2wire_dname_buf(p->rpz_cname, nm, &nmlen) != 0) {
535*a39a5a69SCy Schubert 			log_err("rpz: cannot parse cname override: %s",
536091e9e46SCy Schubert 				p->rpz_cname);
537091e9e46SCy Schubert 			goto err;
538091e9e46SCy Schubert 		}
539091e9e46SCy Schubert 		r->cname_override = new_cname_override(r->region, nm, nmlen);
540091e9e46SCy Schubert 		if(!r->cname_override) {
541091e9e46SCy Schubert 			goto err;
542091e9e46SCy Schubert 		}
543091e9e46SCy Schubert 	}
544091e9e46SCy Schubert 	r->log = p->rpz_log;
5459cf5bc93SCy Schubert 	r->signal_nxdomain_ra = p->rpz_signal_nxdomain_ra;
546091e9e46SCy Schubert 	if(p->rpz_log_name) {
547091e9e46SCy Schubert 		if(!(r->log_name = strdup(p->rpz_log_name))) {
548091e9e46SCy Schubert 			log_err("malloc failure on RPZ log_name strdup");
549091e9e46SCy Schubert 			goto err;
550091e9e46SCy Schubert 		}
551091e9e46SCy Schubert 	}
552091e9e46SCy Schubert 	return r;
553091e9e46SCy Schubert err:
554091e9e46SCy Schubert 	if(r) {
555091e9e46SCy Schubert 		if(r->local_zones)
556091e9e46SCy Schubert 			local_zones_delete(r->local_zones);
55724e36522SCy Schubert 		if(r->nsdname_zones)
55824e36522SCy Schubert 			local_zones_delete(r->nsdname_zones);
559091e9e46SCy Schubert 		if(r->respip_set)
560091e9e46SCy Schubert 			respip_set_delete(r->respip_set);
56124e36522SCy Schubert 		if(r->client_set != NULL)
56224e36522SCy Schubert 			rpz_clientip_synthesized_set_delete(r->client_set);
56324e36522SCy Schubert 		if(r->ns_set != NULL)
56424e36522SCy Schubert 			rpz_clientip_synthesized_set_delete(r->ns_set);
565091e9e46SCy Schubert 		if(r->taglist)
566091e9e46SCy Schubert 			free(r->taglist);
567369c6923SCy Schubert 		if(r->region)
568369c6923SCy Schubert 			regional_destroy(r->region);
569091e9e46SCy Schubert 		free(r);
570091e9e46SCy Schubert 	}
571091e9e46SCy Schubert 	return NULL;
572091e9e46SCy Schubert }
573091e9e46SCy Schubert 
574091e9e46SCy Schubert /**
575091e9e46SCy Schubert  * Remove RPZ zone name from dname
576091e9e46SCy Schubert  * Copy dname to newdname, without the originlen number of trailing bytes
577091e9e46SCy Schubert  */
578091e9e46SCy Schubert static size_t
579091e9e46SCy Schubert strip_dname_origin(uint8_t* dname, size_t dnamelen, size_t originlen,
580091e9e46SCy Schubert 	uint8_t* newdname, size_t maxnewdnamelen)
581091e9e46SCy Schubert {
582091e9e46SCy Schubert 	size_t newdnamelen;
583091e9e46SCy Schubert 	if(dnamelen < originlen)
584091e9e46SCy Schubert 		return 0;
585091e9e46SCy Schubert 	newdnamelen = dnamelen - originlen;
586091e9e46SCy Schubert 	if(newdnamelen+1 > maxnewdnamelen)
587091e9e46SCy Schubert 		return 0;
588091e9e46SCy Schubert 	memmove(newdname, dname, newdnamelen);
589091e9e46SCy Schubert 	newdname[newdnamelen] = 0;
590091e9e46SCy Schubert 	return newdnamelen + 1;	/* + 1 for root label */
591091e9e46SCy Schubert }
592091e9e46SCy Schubert 
593091e9e46SCy Schubert static void
59424e36522SCy Schubert rpz_insert_local_zones_trigger(struct local_zones* lz, uint8_t* dname,
59524e36522SCy Schubert 	size_t dnamelen, enum rpz_action a, uint16_t rrtype, uint16_t rrclass,
59624e36522SCy Schubert 	uint32_t ttl, uint8_t* rdata, size_t rdata_len, uint8_t* rr, size_t rr_len)
597091e9e46SCy Schubert {
598091e9e46SCy Schubert 	struct local_zone* z;
599091e9e46SCy Schubert 	enum localzone_type tp = local_zone_always_transparent;
600091e9e46SCy Schubert 	int dnamelabs = dname_count_labels(dname);
601091e9e46SCy Schubert 	int newzone = 0;
602091e9e46SCy Schubert 
60324e36522SCy Schubert 	if(a == RPZ_INVALID_ACTION) {
6045469a995SCy Schubert 		char str[255+1];
6055469a995SCy Schubert 		if(rrtype == LDNS_RR_TYPE_SOA || rrtype == LDNS_RR_TYPE_NS ||
6065469a995SCy Schubert 			rrtype == LDNS_RR_TYPE_DNAME ||
6075469a995SCy Schubert 			rrtype == LDNS_RR_TYPE_DNSKEY ||
6085469a995SCy Schubert 			rrtype == LDNS_RR_TYPE_RRSIG ||
6095469a995SCy Schubert 			rrtype == LDNS_RR_TYPE_NSEC ||
6105469a995SCy Schubert 			rrtype == LDNS_RR_TYPE_NSEC3PARAM ||
6115469a995SCy Schubert 			rrtype == LDNS_RR_TYPE_NSEC3 ||
6125469a995SCy Schubert 			rrtype == LDNS_RR_TYPE_DS) {
6135469a995SCy Schubert 			free(dname);
6145469a995SCy Schubert 			return; /* no need to log these types as unsupported */
6155469a995SCy Schubert 		}
6165469a995SCy Schubert 		dname_str(dname, str);
617*a39a5a69SCy Schubert 		verbose(VERB_ALGO, "rpz: qname trigger, %s skipping unsupported action: %s",
6185469a995SCy Schubert 			str, rpz_action_to_string(a));
619091e9e46SCy Schubert 		free(dname);
620091e9e46SCy Schubert 		return;
621091e9e46SCy Schubert 	}
622091e9e46SCy Schubert 
62324e36522SCy Schubert 	lock_rw_wrlock(&lz->lock);
624091e9e46SCy Schubert 	/* exact match */
62524e36522SCy Schubert 	z = local_zones_find(lz, dname, dnamelen, dnamelabs, LDNS_RR_CLASS_IN);
62624e36522SCy Schubert 	if(z != NULL && a != RPZ_LOCAL_DATA_ACTION) {
62724e36522SCy Schubert 		char* rrstr = sldns_wire2str_rr(rr, rr_len);
62824e36522SCy Schubert 		if(rrstr == NULL) {
62924e36522SCy Schubert 			log_err("malloc error while inserting rpz nsdname trigger");
630091e9e46SCy Schubert 			free(dname);
63124e36522SCy Schubert 			lock_rw_unlock(&lz->lock);
632091e9e46SCy Schubert 			return;
633091e9e46SCy Schubert 		}
63424e36522SCy Schubert 		if(rrstr[0])
63524e36522SCy Schubert 			rrstr[strlen(rrstr)-1]=0; /* remove newline */
63624e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: skipping duplicate record: '%s'", rrstr);
637091e9e46SCy Schubert 		free(rrstr);
638091e9e46SCy Schubert 		free(dname);
63924e36522SCy Schubert 		lock_rw_unlock(&lz->lock);
640091e9e46SCy Schubert 		return;
641091e9e46SCy Schubert 	}
64224e36522SCy Schubert 	if(z == NULL) {
643091e9e46SCy Schubert 		tp = rpz_action_to_localzone_type(a);
64424e36522SCy Schubert 		z = local_zones_add_zone(lz, dname, dnamelen,
64524e36522SCy Schubert 					 dnamelabs, rrclass, tp);
64624e36522SCy Schubert 		if(z == NULL) {
64724e36522SCy Schubert 			log_warn("rpz: create failed");
64824e36522SCy Schubert 			lock_rw_unlock(&lz->lock);
649091e9e46SCy Schubert 			/* dname will be free'd in failed local_zone_create() */
650091e9e46SCy Schubert 			return;
651091e9e46SCy Schubert 		}
652091e9e46SCy Schubert 		newzone = 1;
653091e9e46SCy Schubert 	}
654091e9e46SCy Schubert 	if(a == RPZ_LOCAL_DATA_ACTION) {
65524e36522SCy Schubert 		char* rrstr = sldns_wire2str_rr(rr, rr_len);
65624e36522SCy Schubert 		if(rrstr == NULL) {
65724e36522SCy Schubert 			log_err("malloc error while inserting rpz nsdname trigger");
658091e9e46SCy Schubert 			free(dname);
65924e36522SCy Schubert 			lock_rw_unlock(&lz->lock);
660091e9e46SCy Schubert 			return;
661091e9e46SCy Schubert 		}
662091e9e46SCy Schubert 		lock_rw_wrlock(&z->lock);
66324e36522SCy Schubert 		local_zone_enter_rr(z, dname, dnamelen, dnamelabs, rrtype,
66424e36522SCy Schubert 				    rrclass, ttl, rdata, rdata_len, rrstr);
665091e9e46SCy Schubert 		lock_rw_unlock(&z->lock);
666091e9e46SCy Schubert 		free(rrstr);
667091e9e46SCy Schubert 	}
66824e36522SCy Schubert 	if(!newzone) {
669091e9e46SCy Schubert 		free(dname);
67024e36522SCy Schubert 	}
67124e36522SCy Schubert 	lock_rw_unlock(&lz->lock);
67224e36522SCy Schubert }
67324e36522SCy Schubert 
67424e36522SCy Schubert static void
67524e36522SCy Schubert rpz_log_dname(char const* msg, uint8_t* dname, size_t dname_len)
67624e36522SCy Schubert {
67724e36522SCy Schubert 	char buf[LDNS_MAX_DOMAINLEN+1];
67824e36522SCy Schubert 	(void)dname_len;
67924e36522SCy Schubert 	dname_str(dname, buf);
68024e36522SCy Schubert 	verbose(VERB_ALGO, "rpz: %s: <%s>", msg, buf);
68124e36522SCy Schubert }
68224e36522SCy Schubert 
68324e36522SCy Schubert static void
68424e36522SCy Schubert rpz_insert_qname_trigger(struct rpz* r, uint8_t* dname, size_t dnamelen,
68524e36522SCy Schubert 	enum rpz_action a, uint16_t rrtype, uint16_t rrclass, uint32_t ttl,
68624e36522SCy Schubert 	uint8_t* rdata, size_t rdata_len, uint8_t* rr, size_t rr_len)
68724e36522SCy Schubert {
68824e36522SCy Schubert 	if(a == RPZ_INVALID_ACTION) {
68924e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: skipping invalid action");
69024e36522SCy Schubert 		free(dname);
691091e9e46SCy Schubert 		return;
692091e9e46SCy Schubert 	}
693091e9e46SCy Schubert 
69424e36522SCy Schubert 	rpz_insert_local_zones_trigger(r->local_zones, dname, dnamelen, a, rrtype,
69524e36522SCy Schubert 				       rrclass, ttl, rdata, rdata_len, rr, rr_len);
69624e36522SCy Schubert }
69724e36522SCy Schubert 
69824e36522SCy Schubert static int
69924e36522SCy Schubert rpz_strip_nsdname_suffix(uint8_t* dname, size_t maxdnamelen,
70024e36522SCy Schubert 	uint8_t** stripdname, size_t* stripdnamelen)
70124e36522SCy Schubert {
70224e36522SCy Schubert 	uint8_t* tldstart = get_tld_label(dname, maxdnamelen);
70324e36522SCy Schubert 	uint8_t swap;
70424e36522SCy Schubert 	if(tldstart == NULL) {
70524e36522SCy Schubert 		if(dname == NULL) {
70624e36522SCy Schubert 			*stripdname = NULL;
70724e36522SCy Schubert 			*stripdnamelen = 0;
70824e36522SCy Schubert 			return 0;
70924e36522SCy Schubert 		}
71024e36522SCy Schubert 		*stripdname = memdup(dname, maxdnamelen);
71124e36522SCy Schubert 		if(!*stripdname) {
71224e36522SCy Schubert 			*stripdnamelen = 0;
71324e36522SCy Schubert 			log_err("malloc failure for rpz strip suffix");
71424e36522SCy Schubert 			return 0;
71524e36522SCy Schubert 		}
71624e36522SCy Schubert 		*stripdnamelen = maxdnamelen;
71724e36522SCy Schubert 		return 1;
71824e36522SCy Schubert 	}
71924e36522SCy Schubert 	/* shorten the domain name briefly,
72024e36522SCy Schubert 	 * then we allocate a new name with the correct length */
72124e36522SCy Schubert 	swap = *tldstart;
72224e36522SCy Schubert 	*tldstart = 0;
72324e36522SCy Schubert 	(void)dname_count_size_labels(dname, stripdnamelen);
72424e36522SCy Schubert 	*stripdname = memdup(dname, *stripdnamelen);
72524e36522SCy Schubert 	*tldstart = swap;
72624e36522SCy Schubert 	if(!*stripdname) {
72724e36522SCy Schubert 		*stripdnamelen = 0;
72824e36522SCy Schubert 		log_err("malloc failure for rpz strip suffix");
72924e36522SCy Schubert 		return 0;
73024e36522SCy Schubert 	}
73124e36522SCy Schubert 	return 1;
73224e36522SCy Schubert }
73324e36522SCy Schubert 
73424e36522SCy Schubert static void
73524e36522SCy Schubert rpz_insert_nsdname_trigger(struct rpz* r, uint8_t* dname, size_t dnamelen,
73624e36522SCy Schubert 	enum rpz_action a, uint16_t rrtype, uint16_t rrclass, uint32_t ttl,
73724e36522SCy Schubert 	uint8_t* rdata, size_t rdata_len, uint8_t* rr, size_t rr_len)
73824e36522SCy Schubert {
73924e36522SCy Schubert 	uint8_t* dname_stripped = NULL;
74024e36522SCy Schubert 	size_t dnamelen_stripped = 0;
74124e36522SCy Schubert 
74224e36522SCy Schubert 	rpz_strip_nsdname_suffix(dname, dnamelen, &dname_stripped,
74324e36522SCy Schubert 		&dnamelen_stripped);
74424e36522SCy Schubert 	if(a == RPZ_INVALID_ACTION) {
74524e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: skipping invalid action");
74624e36522SCy Schubert 		free(dname_stripped);
74724e36522SCy Schubert 		return;
74824e36522SCy Schubert 	}
74924e36522SCy Schubert 
75024e36522SCy Schubert 	/* dname_stripped is consumed or freed by the insert routine */
75124e36522SCy Schubert 	rpz_insert_local_zones_trigger(r->nsdname_zones, dname_stripped,
75224e36522SCy Schubert 		dnamelen_stripped, a, rrtype, rrclass, ttl, rdata, rdata_len,
75324e36522SCy Schubert 		rr, rr_len);
75424e36522SCy Schubert }
75524e36522SCy Schubert 
75624e36522SCy Schubert static int
75724e36522SCy Schubert rpz_insert_ipaddr_based_trigger(struct respip_set* set, struct sockaddr_storage* addr,
75824e36522SCy Schubert 	socklen_t addrlen, int net, enum rpz_action a, uint16_t rrtype,
75924e36522SCy Schubert 	uint16_t rrclass, uint32_t ttl, uint8_t* rdata, size_t rdata_len,
76024e36522SCy Schubert 	uint8_t* rr, size_t rr_len)
76124e36522SCy Schubert {
76224e36522SCy Schubert 	struct resp_addr* node;
76324e36522SCy Schubert 	char* rrstr;
76424e36522SCy Schubert 	enum respip_action respa = rpz_action_to_respip_action(a);
76524e36522SCy Schubert 
76624e36522SCy Schubert 	lock_rw_wrlock(&set->lock);
76724e36522SCy Schubert 	rrstr = sldns_wire2str_rr(rr, rr_len);
76824e36522SCy Schubert 	if(rrstr == NULL) {
76924e36522SCy Schubert 		log_err("malloc error while inserting rpz ipaddr based trigger");
77024e36522SCy Schubert 		lock_rw_unlock(&set->lock);
77124e36522SCy Schubert 		return 0;
77224e36522SCy Schubert 	}
77324e36522SCy Schubert 
77424e36522SCy Schubert 	node = respip_sockaddr_find_or_create(set, addr, addrlen, net, 1, rrstr);
77524e36522SCy Schubert 	if(node == NULL) {
77624e36522SCy Schubert 		lock_rw_unlock(&set->lock);
77724e36522SCy Schubert 		free(rrstr);
77824e36522SCy Schubert 		return 0;
77924e36522SCy Schubert 	}
78024e36522SCy Schubert 
78124e36522SCy Schubert 	lock_rw_wrlock(&node->lock);
78224e36522SCy Schubert 	lock_rw_unlock(&set->lock);
78324e36522SCy Schubert 
78424e36522SCy Schubert 	node->action = respa;
78524e36522SCy Schubert 
78624e36522SCy Schubert 	if(a == RPZ_LOCAL_DATA_ACTION) {
78724e36522SCy Schubert 		respip_enter_rr(set->region, node, rrtype,
78824e36522SCy Schubert 				rrclass, ttl, rdata, rdata_len, rrstr, "");
78924e36522SCy Schubert 	}
79024e36522SCy Schubert 
79124e36522SCy Schubert 	lock_rw_unlock(&node->lock);
79224e36522SCy Schubert 	free(rrstr);
79324e36522SCy Schubert 	return 1;
79424e36522SCy Schubert }
79524e36522SCy Schubert 
79624e36522SCy Schubert static inline struct clientip_synthesized_rr*
79724e36522SCy Schubert rpz_clientip_ensure_entry(struct clientip_synthesized_rrset* set,
79824e36522SCy Schubert 	struct sockaddr_storage* addr, socklen_t addrlen, int net)
79924e36522SCy Schubert {
80024e36522SCy Schubert 	int insert_ok;
80124e36522SCy Schubert 	struct clientip_synthesized_rr* node =
80224e36522SCy Schubert 		(struct clientip_synthesized_rr*)addr_tree_find(&set->entries,
80324e36522SCy Schubert 								addr, addrlen, net);
80424e36522SCy Schubert 
80524e36522SCy Schubert 	if(node != NULL) { return node; }
80624e36522SCy Schubert 
80724e36522SCy Schubert 	/* node does not yet exist => allocate one */
80824e36522SCy Schubert 	node = regional_alloc_zero(set->region, sizeof(*node));
80924e36522SCy Schubert 	if(node == NULL) {
81024e36522SCy Schubert 		log_err("out of memory");
81124e36522SCy Schubert 		return NULL;
81224e36522SCy Schubert 	}
81324e36522SCy Schubert 
81424e36522SCy Schubert 	lock_rw_init(&node->lock);
81524e36522SCy Schubert 	node->action = RPZ_INVALID_ACTION;
81624e36522SCy Schubert 	insert_ok = addr_tree_insert(&set->entries, &node->node,
81724e36522SCy Schubert 				     addr, addrlen, net);
81824e36522SCy Schubert 	if (!insert_ok) {
81924e36522SCy Schubert 		log_warn("rpz: unexpected: unable to insert clientip address node");
82024e36522SCy Schubert 		/* we can not free the just allocated node.
82124e36522SCy Schubert 		 * theoretically a memleak */
82224e36522SCy Schubert 		return NULL;
82324e36522SCy Schubert 	}
82424e36522SCy Schubert 
82524e36522SCy Schubert 	return node;
82624e36522SCy Schubert }
82724e36522SCy Schubert 
82824e36522SCy Schubert static void
82924e36522SCy Schubert rpz_report_rrset_error(const char* msg, uint8_t* rr, size_t rr_len) {
83024e36522SCy Schubert 	char* rrstr = sldns_wire2str_rr(rr, rr_len);
83124e36522SCy Schubert 	if(rrstr == NULL) {
83224e36522SCy Schubert 		log_err("malloc error while inserting rpz clientip based record");
83324e36522SCy Schubert 		return;
83424e36522SCy Schubert 	}
83524e36522SCy Schubert 	log_err("rpz: unexpected: unable to insert %s: %s", msg, rrstr);
83624e36522SCy Schubert 	free(rrstr);
83724e36522SCy Schubert }
83824e36522SCy Schubert 
83924e36522SCy Schubert /* from localzone.c; difference is we don't have a dname */
8409cf5bc93SCy Schubert static struct local_rrset*
84124e36522SCy Schubert rpz_clientip_new_rrset(struct regional* region,
84224e36522SCy Schubert 	struct clientip_synthesized_rr* raddr, uint16_t rrtype, uint16_t rrclass)
84324e36522SCy Schubert {
84424e36522SCy Schubert 	struct packed_rrset_data* pd;
84524e36522SCy Schubert 	struct local_rrset* rrset = (struct local_rrset*)
84624e36522SCy Schubert 		regional_alloc_zero(region, sizeof(*rrset));
84724e36522SCy Schubert 	if(rrset == NULL) {
84824e36522SCy Schubert 		log_err("out of memory");
84924e36522SCy Schubert 		return NULL;
85024e36522SCy Schubert 	}
85124e36522SCy Schubert 	rrset->next = raddr->data;
85224e36522SCy Schubert 	raddr->data = rrset;
85324e36522SCy Schubert 	rrset->rrset = (struct ub_packed_rrset_key*)
85424e36522SCy Schubert 		regional_alloc_zero(region, sizeof(*rrset->rrset));
85524e36522SCy Schubert 	if(rrset->rrset == NULL) {
85624e36522SCy Schubert 		log_err("out of memory");
85724e36522SCy Schubert 		return NULL;
85824e36522SCy Schubert 	}
85924e36522SCy Schubert 	rrset->rrset->entry.key = rrset->rrset;
86024e36522SCy Schubert 	pd = (struct packed_rrset_data*)regional_alloc_zero(region, sizeof(*pd));
86124e36522SCy Schubert 	if(pd == NULL) {
86224e36522SCy Schubert 		log_err("out of memory");
86324e36522SCy Schubert 		return NULL;
86424e36522SCy Schubert 	}
86524e36522SCy Schubert 	pd->trust = rrset_trust_prim_noglue;
86624e36522SCy Schubert 	pd->security = sec_status_insecure;
86724e36522SCy Schubert 	rrset->rrset->entry.data = pd;
86824e36522SCy Schubert 	rrset->rrset->rk.type = htons(rrtype);
86924e36522SCy Schubert 	rrset->rrset->rk.rrset_class = htons(rrclass);
87024e36522SCy Schubert 	rrset->rrset->rk.dname = regional_alloc_zero(region, 1);
87124e36522SCy Schubert 	if(rrset->rrset->rk.dname == NULL) {
87224e36522SCy Schubert 		log_err("out of memory");
87324e36522SCy Schubert 		return NULL;
87424e36522SCy Schubert 	}
87524e36522SCy Schubert 	rrset->rrset->rk.dname_len = 1;
87624e36522SCy Schubert 	return rrset;
87724e36522SCy Schubert }
87824e36522SCy Schubert 
87924e36522SCy Schubert static int
88024e36522SCy Schubert rpz_clientip_enter_rr(struct regional* region, struct clientip_synthesized_rr* raddr,
88124e36522SCy Schubert 	uint16_t rrtype, uint16_t rrclass, time_t ttl, uint8_t* rdata,
88224e36522SCy Schubert 	size_t rdata_len)
88324e36522SCy Schubert {
88424e36522SCy Schubert 	struct local_rrset* rrset;
88524e36522SCy Schubert 	if (rrtype == LDNS_RR_TYPE_CNAME && raddr->data != NULL) {
88624e36522SCy Schubert 		log_err("CNAME response-ip data can not co-exist with other "
88724e36522SCy Schubert 			"client-ip data");
88824e36522SCy Schubert 		return 0;
88924e36522SCy Schubert 	}
89024e36522SCy Schubert 
89124e36522SCy Schubert 	rrset = rpz_clientip_new_rrset(region, raddr, rrtype, rrclass);
89224e36522SCy Schubert 	if(raddr->data == NULL) {
89324e36522SCy Schubert 		return 0;
89424e36522SCy Schubert 	}
89524e36522SCy Schubert 
89624e36522SCy Schubert 	return rrset_insert_rr(region, rrset->rrset->entry.data, rdata, rdata_len, ttl, "");
89724e36522SCy Schubert }
89824e36522SCy Schubert 
89924e36522SCy Schubert static int
90024e36522SCy Schubert rpz_clientip_insert_trigger_rr(struct clientip_synthesized_rrset* set, struct sockaddr_storage* addr,
90124e36522SCy Schubert 	socklen_t addrlen, int net, enum rpz_action a, uint16_t rrtype,
90224e36522SCy Schubert 	uint16_t rrclass, uint32_t ttl, uint8_t* rdata, size_t rdata_len,
90324e36522SCy Schubert 	uint8_t* rr, size_t rr_len)
90424e36522SCy Schubert {
90524e36522SCy Schubert 	struct clientip_synthesized_rr* node;
90624e36522SCy Schubert 
90724e36522SCy Schubert 	lock_rw_wrlock(&set->lock);
90824e36522SCy Schubert 
90924e36522SCy Schubert 	node = rpz_clientip_ensure_entry(set, addr, addrlen, net);
91024e36522SCy Schubert 	if(node == NULL) {
91124e36522SCy Schubert 		lock_rw_unlock(&set->lock);
91224e36522SCy Schubert 		rpz_report_rrset_error("client ip address", rr, rr_len);
91324e36522SCy Schubert 		return 0;
91424e36522SCy Schubert 	}
91524e36522SCy Schubert 
91624e36522SCy Schubert 	lock_rw_wrlock(&node->lock);
91724e36522SCy Schubert 	lock_rw_unlock(&set->lock);
91824e36522SCy Schubert 
91924e36522SCy Schubert 	node->action = a;
92024e36522SCy Schubert 	if(a == RPZ_LOCAL_DATA_ACTION) {
92124e36522SCy Schubert 		if(!rpz_clientip_enter_rr(set->region, node, rrtype,
92224e36522SCy Schubert 			rrclass, ttl, rdata, rdata_len)) {
92324e36522SCy Schubert 			verbose(VERB_ALGO, "rpz: unable to insert clientip rr");
92424e36522SCy Schubert 			lock_rw_unlock(&node->lock);
92524e36522SCy Schubert 			return 0;
92624e36522SCy Schubert 		}
92724e36522SCy Schubert 
92824e36522SCy Schubert 	}
92924e36522SCy Schubert 
93024e36522SCy Schubert 	lock_rw_unlock(&node->lock);
93124e36522SCy Schubert 
93224e36522SCy Schubert 	return 1;
93324e36522SCy Schubert }
93424e36522SCy Schubert 
93524e36522SCy Schubert static int
93624e36522SCy Schubert rpz_insert_clientip_trigger(struct rpz* r, uint8_t* dname, size_t dnamelen,
93724e36522SCy Schubert 	enum rpz_action a, uint16_t rrtype, uint16_t rrclass, uint32_t ttl,
93824e36522SCy Schubert 	uint8_t* rdata, size_t rdata_len, uint8_t* rr, size_t rr_len)
93924e36522SCy Schubert {
94024e36522SCy Schubert 	struct sockaddr_storage addr;
94124e36522SCy Schubert 	socklen_t addrlen;
94224e36522SCy Schubert 	int net, af;
94324e36522SCy Schubert 
94424e36522SCy Schubert 	if(a == RPZ_INVALID_ACTION) {
94524e36522SCy Schubert 		return 0;
94624e36522SCy Schubert 	}
94724e36522SCy Schubert 
94824e36522SCy Schubert 	if(!netblockdnametoaddr(dname, dnamelen, &addr, &addrlen, &net, &af)) {
94924e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: unable to parse client ip");
95024e36522SCy Schubert 		return 0;
95124e36522SCy Schubert 	}
95224e36522SCy Schubert 
95324e36522SCy Schubert 	return rpz_clientip_insert_trigger_rr(r->client_set, &addr, addrlen, net,
95424e36522SCy Schubert 			a, rrtype, rrclass, ttl, rdata, rdata_len, rr, rr_len);
95524e36522SCy Schubert }
95624e36522SCy Schubert 
95724e36522SCy Schubert static int
95824e36522SCy Schubert rpz_insert_nsip_trigger(struct rpz* r, uint8_t* dname, size_t dnamelen,
95924e36522SCy Schubert 	enum rpz_action a, uint16_t rrtype, uint16_t rrclass, uint32_t ttl,
96024e36522SCy Schubert 	uint8_t* rdata, size_t rdata_len, uint8_t* rr, size_t rr_len)
96124e36522SCy Schubert {
96224e36522SCy Schubert 	struct sockaddr_storage addr;
96324e36522SCy Schubert 	socklen_t addrlen;
96424e36522SCy Schubert 	int net, af;
96524e36522SCy Schubert 
96624e36522SCy Schubert 	if(a == RPZ_INVALID_ACTION) {
96724e36522SCy Schubert 		return 0;
96824e36522SCy Schubert 	}
96924e36522SCy Schubert 
97024e36522SCy Schubert 	if(!netblockdnametoaddr(dname, dnamelen, &addr, &addrlen, &net, &af)) {
97124e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: unable to parse ns ip");
97224e36522SCy Schubert 		return 0;
97324e36522SCy Schubert 	}
97424e36522SCy Schubert 
97524e36522SCy Schubert 	return rpz_clientip_insert_trigger_rr(r->ns_set, &addr, addrlen, net,
97624e36522SCy Schubert 			a, rrtype, rrclass, ttl, rdata, rdata_len, rr, rr_len);
97724e36522SCy Schubert }
97824e36522SCy Schubert 
979091e9e46SCy Schubert /** Insert RR into RPZ's respip_set */
980091e9e46SCy Schubert static int
981091e9e46SCy Schubert rpz_insert_response_ip_trigger(struct rpz* r, uint8_t* dname, size_t dnamelen,
982091e9e46SCy Schubert 	enum rpz_action a, uint16_t rrtype, uint16_t rrclass, uint32_t ttl,
983091e9e46SCy Schubert 	uint8_t* rdata, size_t rdata_len, uint8_t* rr, size_t rr_len)
984091e9e46SCy Schubert {
985091e9e46SCy Schubert 	struct sockaddr_storage addr;
986091e9e46SCy Schubert 	socklen_t addrlen;
987091e9e46SCy Schubert 	int net, af;
988091e9e46SCy Schubert 
98924e36522SCy Schubert 	if(a == RPZ_INVALID_ACTION) {
99024e36522SCy Schubert 		return 0;
99124e36522SCy Schubert 	}
99224e36522SCy Schubert 
99324e36522SCy Schubert 	if(!netblockdnametoaddr(dname, dnamelen, &addr, &addrlen, &net, &af)) {
99424e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: unable to parse response ip");
99524e36522SCy Schubert 		return 0;
99624e36522SCy Schubert 	}
99724e36522SCy Schubert 
99824e36522SCy Schubert 	if(a == RPZ_INVALID_ACTION ||
99924e36522SCy Schubert 		rpz_action_to_respip_action(a) == respip_invalid) {
10005469a995SCy Schubert 		char str[255+1];
10015469a995SCy Schubert 		dname_str(dname, str);
1002*a39a5a69SCy Schubert 		verbose(VERB_ALGO, "rpz: respip trigger, %s skipping unsupported action: %s",
10035469a995SCy Schubert 			str, rpz_action_to_string(a));
1004091e9e46SCy Schubert 		return 0;
1005091e9e46SCy Schubert 	}
1006091e9e46SCy Schubert 
100724e36522SCy Schubert 	return rpz_insert_ipaddr_based_trigger(r->respip_set, &addr, addrlen, net,
100824e36522SCy Schubert 			a, rrtype, rrclass, ttl, rdata, rdata_len, rr, rr_len);
1009091e9e46SCy Schubert }
1010091e9e46SCy Schubert 
1011091e9e46SCy Schubert int
101225039b37SCy Schubert rpz_insert_rr(struct rpz* r, uint8_t* azname, size_t aznamelen, uint8_t* dname,
1013091e9e46SCy Schubert 	size_t dnamelen, uint16_t rr_type, uint16_t rr_class, uint32_t rr_ttl,
1014091e9e46SCy Schubert 	uint8_t* rdatawl, size_t rdatalen, uint8_t* rr, size_t rr_len)
1015091e9e46SCy Schubert {
1016091e9e46SCy Schubert 	size_t policydnamelen;
1017091e9e46SCy Schubert 	/* name is free'd in local_zone delete */
1018091e9e46SCy Schubert 	enum rpz_trigger t;
1019091e9e46SCy Schubert 	enum rpz_action a;
1020091e9e46SCy Schubert 	uint8_t* policydname;
1021091e9e46SCy Schubert 
102224e36522SCy Schubert 	if(rpz_type_ignored(rr_type)) {
102324e36522SCy Schubert 		/* this rpz action is not valid, eg. this is the SOA or NS RR */
102424e36522SCy Schubert 		return 1;
102524e36522SCy Schubert 	}
102625039b37SCy Schubert 	if(!dname_subdomain_c(dname, azname)) {
1027c0caa2e2SCy Schubert 		char* dname_str = sldns_wire2str_dname(dname, dnamelen);
1028c0caa2e2SCy Schubert 		char* azname_str = sldns_wire2str_dname(azname, aznamelen);
1029c0caa2e2SCy Schubert 		if(dname_str && azname_str) {
103024e36522SCy Schubert 			log_err("rpz: name of record (%s) to insert into RPZ is not a "
1031c0caa2e2SCy Schubert 				"subdomain of the configured name of the RPZ zone (%s)",
1032c0caa2e2SCy Schubert 				dname_str, azname_str);
1033c0caa2e2SCy Schubert 		} else {
103424e36522SCy Schubert 			log_err("rpz: name of record to insert into RPZ is not a "
103525039b37SCy Schubert 				"subdomain of the configured name of the RPZ zone");
1036c0caa2e2SCy Schubert 		}
1037c0caa2e2SCy Schubert 		free(dname_str);
1038c0caa2e2SCy Schubert 		free(azname_str);
1039091e9e46SCy Schubert 		return 0;
104025039b37SCy Schubert 	}
104125039b37SCy Schubert 
104225039b37SCy Schubert 	log_assert(dnamelen >= aznamelen);
104325039b37SCy Schubert 	if(!(policydname = calloc(1, (dnamelen-aznamelen)+1))) {
104425039b37SCy Schubert 		log_err("malloc error while inserting RPZ RR");
104525039b37SCy Schubert 		return 0;
104625039b37SCy Schubert 	}
1047091e9e46SCy Schubert 
1048091e9e46SCy Schubert 	a = rpz_rr_to_action(rr_type, rdatawl, rdatalen);
1049091e9e46SCy Schubert 	if(!(policydnamelen = strip_dname_origin(dname, dnamelen, aznamelen,
1050091e9e46SCy Schubert 		policydname, (dnamelen-aznamelen)+1))) {
1051091e9e46SCy Schubert 		free(policydname);
1052091e9e46SCy Schubert 		return 0;
1053091e9e46SCy Schubert 	}
1054091e9e46SCy Schubert 	t = rpz_dname_to_trigger(policydname, policydnamelen);
1055091e9e46SCy Schubert 	if(t == RPZ_INVALID_TRIGGER) {
1056091e9e46SCy Schubert 		free(policydname);
105724e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: skipping invalid trigger");
1058091e9e46SCy Schubert 		return 1;
1059091e9e46SCy Schubert 	}
1060091e9e46SCy Schubert 	if(t == RPZ_QNAME_TRIGGER) {
106124e36522SCy Schubert 		/* policydname will be consumed, no free */
1062091e9e46SCy Schubert 		rpz_insert_qname_trigger(r, policydname, policydnamelen,
1063091e9e46SCy Schubert 			a, rr_type, rr_class, rr_ttl, rdatawl, rdatalen, rr,
1064091e9e46SCy Schubert 			rr_len);
106524e36522SCy Schubert 	} else if(t == RPZ_RESPONSE_IP_TRIGGER) {
1066091e9e46SCy Schubert 		rpz_insert_response_ip_trigger(r, policydname, policydnamelen,
1067091e9e46SCy Schubert 			a, rr_type, rr_class, rr_ttl, rdatawl, rdatalen, rr,
1068091e9e46SCy Schubert 			rr_len);
1069091e9e46SCy Schubert 		free(policydname);
107024e36522SCy Schubert 	} else if(t == RPZ_CLIENT_IP_TRIGGER) {
107124e36522SCy Schubert 		rpz_insert_clientip_trigger(r, policydname, policydnamelen,
107224e36522SCy Schubert 			a, rr_type, rr_class, rr_ttl, rdatawl, rdatalen, rr,
107324e36522SCy Schubert 			rr_len);
1074091e9e46SCy Schubert 		free(policydname);
107524e36522SCy Schubert 	} else if(t == RPZ_NSIP_TRIGGER) {
107624e36522SCy Schubert 		rpz_insert_nsip_trigger(r, policydname, policydnamelen,
107724e36522SCy Schubert 			a, rr_type, rr_class, rr_ttl, rdatawl, rdatalen, rr,
107824e36522SCy Schubert 			rr_len);
107924e36522SCy Schubert 		free(policydname);
108024e36522SCy Schubert 	} else if(t == RPZ_NSDNAME_TRIGGER) {
108124e36522SCy Schubert 		rpz_insert_nsdname_trigger(r, policydname, policydnamelen,
108224e36522SCy Schubert 			a, rr_type, rr_class, rr_ttl, rdatawl, rdatalen, rr,
108324e36522SCy Schubert 			rr_len);
108424e36522SCy Schubert 		free(policydname);
108524e36522SCy Schubert 	} else {
108624e36522SCy Schubert 		free(policydname);
108724e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: skipping unsupported trigger: %s",
1088091e9e46SCy Schubert 			rpz_trigger_to_string(t));
1089091e9e46SCy Schubert 	}
1090091e9e46SCy Schubert 	return 1;
1091091e9e46SCy Schubert }
1092091e9e46SCy Schubert 
1093091e9e46SCy Schubert /**
1094091e9e46SCy Schubert  * Find RPZ local-zone by qname.
109524e36522SCy Schubert  * @param zones: local-zone tree
1096091e9e46SCy Schubert  * @param qname: qname
1097091e9e46SCy Schubert  * @param qname_len: length of qname
1098091e9e46SCy Schubert  * @param qclass: qclass
109924e36522SCy Schubert  * @param only_exact: if 1 only exact (non wildcard) matches are returned
1100091e9e46SCy Schubert  * @param wr: get write lock for local-zone if 1, read lock if 0
1101091e9e46SCy Schubert  * @param zones_keep_lock: if set do not release the r->local_zones lock, this
1102091e9e46SCy Schubert  * 	  makes the caller of this function responsible for releasing the lock.
1103091e9e46SCy Schubert  * @return: NULL or local-zone holding rd or wr lock
1104091e9e46SCy Schubert  */
1105091e9e46SCy Schubert static struct local_zone*
110624e36522SCy Schubert rpz_find_zone(struct local_zones* zones, uint8_t* qname, size_t qname_len, uint16_t qclass,
1107091e9e46SCy Schubert 	int only_exact, int wr, int zones_keep_lock)
1108091e9e46SCy Schubert {
1109091e9e46SCy Schubert 	uint8_t* ce;
1110f44e67d1SCy Schubert 	size_t ce_len;
1111f44e67d1SCy Schubert 	int ce_labs;
1112091e9e46SCy Schubert 	uint8_t wc[LDNS_MAX_DOMAINLEN+1];
1113091e9e46SCy Schubert 	int exact;
1114091e9e46SCy Schubert 	struct local_zone* z = NULL;
111524e36522SCy Schubert 
1116091e9e46SCy Schubert 	if(wr) {
111724e36522SCy Schubert 		lock_rw_wrlock(&zones->lock);
1118091e9e46SCy Schubert 	} else {
111924e36522SCy Schubert 		lock_rw_rdlock(&zones->lock);
1120091e9e46SCy Schubert 	}
112124e36522SCy Schubert 	z = local_zones_find_le(zones, qname, qname_len,
1122091e9e46SCy Schubert 		dname_count_labels(qname),
1123091e9e46SCy Schubert 		LDNS_RR_CLASS_IN, &exact);
1124091e9e46SCy Schubert 	if(!z || (only_exact && !exact)) {
112524e36522SCy Schubert 		if(!zones_keep_lock) {
112624e36522SCy Schubert 			lock_rw_unlock(&zones->lock);
112724e36522SCy Schubert 		}
1128091e9e46SCy Schubert 		return NULL;
1129091e9e46SCy Schubert 	}
1130091e9e46SCy Schubert 	if(wr) {
1131091e9e46SCy Schubert 		lock_rw_wrlock(&z->lock);
1132091e9e46SCy Schubert 	} else {
1133091e9e46SCy Schubert 		lock_rw_rdlock(&z->lock);
1134091e9e46SCy Schubert 	}
1135091e9e46SCy Schubert 	if(!zones_keep_lock) {
113624e36522SCy Schubert 		lock_rw_unlock(&zones->lock);
1137091e9e46SCy Schubert 	}
1138091e9e46SCy Schubert 
1139091e9e46SCy Schubert 	if(exact)
1140091e9e46SCy Schubert 		return z;
1141091e9e46SCy Schubert 
1142091e9e46SCy Schubert 	/* No exact match found, lookup wildcard. closest encloser must
1143091e9e46SCy Schubert 	 * be the shared parent between the qname and the best local
1144091e9e46SCy Schubert 	 * zone match, append '*' to that and do another lookup. */
1145091e9e46SCy Schubert 
1146091e9e46SCy Schubert 	ce = dname_get_shared_topdomain(z->name, qname);
11475469a995SCy Schubert 	if(!ce /* should not happen */) {
1148091e9e46SCy Schubert 		lock_rw_unlock(&z->lock);
1149091e9e46SCy Schubert 		if(zones_keep_lock) {
115024e36522SCy Schubert 			lock_rw_unlock(&zones->lock);
1151091e9e46SCy Schubert 		}
1152091e9e46SCy Schubert 		return NULL;
1153091e9e46SCy Schubert 	}
1154091e9e46SCy Schubert 	ce_labs = dname_count_size_labels(ce, &ce_len);
1155091e9e46SCy Schubert 	if(ce_len+2 > sizeof(wc)) {
1156091e9e46SCy Schubert 		lock_rw_unlock(&z->lock);
1157091e9e46SCy Schubert 		if(zones_keep_lock) {
115824e36522SCy Schubert 			lock_rw_unlock(&zones->lock);
1159091e9e46SCy Schubert 		}
1160091e9e46SCy Schubert 		return NULL;
1161091e9e46SCy Schubert 	}
1162091e9e46SCy Schubert 	wc[0] = 1; /* length of wildcard label */
1163091e9e46SCy Schubert 	wc[1] = (uint8_t)'*'; /* wildcard label */
1164091e9e46SCy Schubert 	memmove(wc+2, ce, ce_len);
1165091e9e46SCy Schubert 	lock_rw_unlock(&z->lock);
1166091e9e46SCy Schubert 
1167091e9e46SCy Schubert 	if(!zones_keep_lock) {
1168091e9e46SCy Schubert 		if(wr) {
116924e36522SCy Schubert 			lock_rw_wrlock(&zones->lock);
1170091e9e46SCy Schubert 		} else {
117124e36522SCy Schubert 			lock_rw_rdlock(&zones->lock);
1172091e9e46SCy Schubert 		}
1173091e9e46SCy Schubert 	}
117424e36522SCy Schubert 	z = local_zones_find_le(zones, wc,
1175091e9e46SCy Schubert 		ce_len+2, ce_labs+1, qclass, &exact);
1176091e9e46SCy Schubert 	if(!z || !exact) {
117724e36522SCy Schubert 		lock_rw_unlock(&zones->lock);
1178091e9e46SCy Schubert 		return NULL;
1179091e9e46SCy Schubert 	}
1180091e9e46SCy Schubert 	if(wr) {
1181091e9e46SCy Schubert 		lock_rw_wrlock(&z->lock);
1182091e9e46SCy Schubert 	} else {
1183091e9e46SCy Schubert 		lock_rw_rdlock(&z->lock);
1184091e9e46SCy Schubert 	}
1185091e9e46SCy Schubert 	if(!zones_keep_lock) {
118624e36522SCy Schubert 		lock_rw_unlock(&zones->lock);
1187091e9e46SCy Schubert 	}
1188091e9e46SCy Schubert 	return z;
1189091e9e46SCy Schubert }
1190091e9e46SCy Schubert 
1191091e9e46SCy Schubert /**
1192091e9e46SCy Schubert  * Remove RR from RPZ's local-data
1193091e9e46SCy Schubert  * @param z: local-zone for RPZ, holding write lock
1194091e9e46SCy Schubert  * @param policydname: dname of RR to remove
119524e36522SCy Schubert  * @param policydnamelen: length of policydname
1196091e9e46SCy Schubert  * @param rr_type: RR type of RR to remove
1197091e9e46SCy Schubert  * @param rdata: rdata of RR to remove
1198091e9e46SCy Schubert  * @param rdatalen: length of rdata
1199091e9e46SCy Schubert  * @return: 1 if zone must be removed after RR deletion
1200091e9e46SCy Schubert  */
1201091e9e46SCy Schubert static int
1202091e9e46SCy Schubert rpz_data_delete_rr(struct local_zone* z, uint8_t* policydname,
1203091e9e46SCy Schubert 	size_t policydnamelen, uint16_t rr_type, uint8_t* rdata,
1204091e9e46SCy Schubert 	size_t rdatalen)
1205091e9e46SCy Schubert {
1206091e9e46SCy Schubert 	struct local_data* ld;
1207091e9e46SCy Schubert 	struct packed_rrset_data* d;
1208091e9e46SCy Schubert 	size_t index;
1209091e9e46SCy Schubert 	ld = local_zone_find_data(z, policydname, policydnamelen,
1210091e9e46SCy Schubert 		dname_count_labels(policydname));
1211091e9e46SCy Schubert 	if(ld) {
1212091e9e46SCy Schubert 		struct local_rrset* prev=NULL, *p=ld->rrsets;
1213091e9e46SCy Schubert 		while(p && ntohs(p->rrset->rk.type) != rr_type) {
1214091e9e46SCy Schubert 			prev = p;
1215091e9e46SCy Schubert 			p = p->next;
1216091e9e46SCy Schubert 		}
1217091e9e46SCy Schubert 		if(!p)
1218091e9e46SCy Schubert 			return 0;
1219091e9e46SCy Schubert 		d = (struct packed_rrset_data*)p->rrset->entry.data;
1220091e9e46SCy Schubert 		if(packed_rrset_find_rr(d, rdata, rdatalen, &index)) {
1221091e9e46SCy Schubert 			if(d->count == 1) {
1222091e9e46SCy Schubert 				/* no memory recycling for zone deletions ... */
1223091e9e46SCy Schubert 				if(prev) prev->next = p->next;
1224091e9e46SCy Schubert 				else ld->rrsets = p->next;
1225091e9e46SCy Schubert 			}
1226091e9e46SCy Schubert 			if(d->count > 1) {
1227091e9e46SCy Schubert 				if(!local_rrset_remove_rr(d, index))
1228091e9e46SCy Schubert 					return 0;
1229091e9e46SCy Schubert 			}
1230091e9e46SCy Schubert 		}
1231091e9e46SCy Schubert 	}
1232091e9e46SCy Schubert 	if(ld && ld->rrsets)
1233091e9e46SCy Schubert 		return 0;
1234091e9e46SCy Schubert 	return 1;
1235091e9e46SCy Schubert }
1236091e9e46SCy Schubert 
1237091e9e46SCy Schubert /**
1238091e9e46SCy Schubert  * Remove RR from RPZ's respip set
1239091e9e46SCy Schubert  * @param raddr: respip node
1240091e9e46SCy Schubert  * @param rr_type: RR type of RR to remove
1241091e9e46SCy Schubert  * @param rdata: rdata of RR to remove
1242091e9e46SCy Schubert  * @param rdatalen: length of rdata
1243091e9e46SCy Schubert  * @return: 1 if zone must be removed after RR deletion
1244091e9e46SCy Schubert  */
1245091e9e46SCy Schubert static int
1246091e9e46SCy Schubert rpz_rrset_delete_rr(struct resp_addr* raddr, uint16_t rr_type, uint8_t* rdata,
1247091e9e46SCy Schubert 	size_t rdatalen)
1248091e9e46SCy Schubert {
1249091e9e46SCy Schubert 	size_t index;
1250091e9e46SCy Schubert 	struct packed_rrset_data* d;
1251091e9e46SCy Schubert 	if(!raddr->data)
1252091e9e46SCy Schubert 		return 1;
1253091e9e46SCy Schubert 	d = raddr->data->entry.data;
1254091e9e46SCy Schubert 	if(ntohs(raddr->data->rk.type) != rr_type) {
1255091e9e46SCy Schubert 		return 0;
1256091e9e46SCy Schubert 	}
1257091e9e46SCy Schubert 	if(packed_rrset_find_rr(d, rdata, rdatalen, &index)) {
1258091e9e46SCy Schubert 		if(d->count == 1) {
1259091e9e46SCy Schubert 			/* regional alloc'd */
1260091e9e46SCy Schubert 			raddr->data->entry.data = NULL;
1261091e9e46SCy Schubert 			raddr->data = NULL;
1262091e9e46SCy Schubert 			return 1;
1263091e9e46SCy Schubert 		}
1264091e9e46SCy Schubert 		if(d->count > 1) {
1265091e9e46SCy Schubert 			if(!local_rrset_remove_rr(d, index))
1266091e9e46SCy Schubert 				return 0;
1267091e9e46SCy Schubert 		}
1268091e9e46SCy Schubert 	}
1269091e9e46SCy Schubert 	return 0;
1270091e9e46SCy Schubert 
1271091e9e46SCy Schubert }
1272091e9e46SCy Schubert 
1273091e9e46SCy Schubert /** Remove RR from RPZ's local-zone */
1274091e9e46SCy Schubert static void
1275091e9e46SCy Schubert rpz_remove_qname_trigger(struct rpz* r, uint8_t* dname, size_t dnamelen,
1276091e9e46SCy Schubert 	enum rpz_action a, uint16_t rr_type, uint16_t rr_class,
1277091e9e46SCy Schubert 	uint8_t* rdatawl, size_t rdatalen)
1278091e9e46SCy Schubert {
1279091e9e46SCy Schubert 	struct local_zone* z;
1280091e9e46SCy Schubert 	int delete_zone = 1;
128124e36522SCy Schubert 	z = rpz_find_zone(r->local_zones, dname, dnamelen, rr_class,
1282091e9e46SCy Schubert 		1 /* only exact */, 1 /* wr lock */, 1 /* keep lock*/);
1283091e9e46SCy Schubert 	if(!z) {
128424e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: cannot remove RR from IXFR, "
1285091e9e46SCy Schubert 			"RPZ domain not found");
1286091e9e46SCy Schubert 		return;
1287091e9e46SCy Schubert 	}
1288091e9e46SCy Schubert 	if(a == RPZ_LOCAL_DATA_ACTION)
1289091e9e46SCy Schubert 		delete_zone = rpz_data_delete_rr(z, dname,
1290091e9e46SCy Schubert 			dnamelen, rr_type, rdatawl, rdatalen);
1291091e9e46SCy Schubert 	else if(a != localzone_type_to_rpz_action(z->type)) {
129225039b37SCy Schubert 		lock_rw_unlock(&z->lock);
129325039b37SCy Schubert 		lock_rw_unlock(&r->local_zones->lock);
1294091e9e46SCy Schubert 		return;
1295091e9e46SCy Schubert 	}
1296091e9e46SCy Schubert 	lock_rw_unlock(&z->lock);
1297091e9e46SCy Schubert 	if(delete_zone) {
1298091e9e46SCy Schubert 		local_zones_del_zone(r->local_zones, z);
1299091e9e46SCy Schubert 	}
1300091e9e46SCy Schubert 	lock_rw_unlock(&r->local_zones->lock);
1301091e9e46SCy Schubert 	return;
1302091e9e46SCy Schubert }
1303091e9e46SCy Schubert 
1304091e9e46SCy Schubert static void
1305091e9e46SCy Schubert rpz_remove_response_ip_trigger(struct rpz* r, uint8_t* dname, size_t dnamelen,
1306091e9e46SCy Schubert 	enum rpz_action a, uint16_t rr_type, uint8_t* rdatawl, size_t rdatalen)
1307091e9e46SCy Schubert {
1308091e9e46SCy Schubert 	struct resp_addr* node;
1309091e9e46SCy Schubert 	struct sockaddr_storage addr;
1310091e9e46SCy Schubert 	socklen_t addrlen;
1311091e9e46SCy Schubert 	int net, af;
1312091e9e46SCy Schubert 	int delete_respip = 1;
1313091e9e46SCy Schubert 
1314091e9e46SCy Schubert 	if(!netblockdnametoaddr(dname, dnamelen, &addr, &addrlen, &net, &af))
1315091e9e46SCy Schubert 		return;
1316091e9e46SCy Schubert 
1317091e9e46SCy Schubert 	lock_rw_wrlock(&r->respip_set->lock);
1318091e9e46SCy Schubert 	if(!(node = (struct resp_addr*)addr_tree_find(
1319091e9e46SCy Schubert 		&r->respip_set->ip_tree, &addr, addrlen, net))) {
132024e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: cannot remove RR from IXFR, "
1321091e9e46SCy Schubert 			"RPZ domain not found");
1322091e9e46SCy Schubert 		lock_rw_unlock(&r->respip_set->lock);
1323091e9e46SCy Schubert 		return;
1324091e9e46SCy Schubert 	}
1325091e9e46SCy Schubert 
1326091e9e46SCy Schubert 	lock_rw_wrlock(&node->lock);
1327091e9e46SCy Schubert 	if(a == RPZ_LOCAL_DATA_ACTION) {
1328091e9e46SCy Schubert 		/* remove RR, signal whether RR can be removed */
1329091e9e46SCy Schubert 		delete_respip = rpz_rrset_delete_rr(node, rr_type, rdatawl,
1330091e9e46SCy Schubert 			rdatalen);
1331091e9e46SCy Schubert 	}
1332091e9e46SCy Schubert 	lock_rw_unlock(&node->lock);
1333091e9e46SCy Schubert 	if(delete_respip)
1334091e9e46SCy Schubert 		respip_sockaddr_delete(r->respip_set, node);
1335091e9e46SCy Schubert 	lock_rw_unlock(&r->respip_set->lock);
1336091e9e46SCy Schubert }
1337091e9e46SCy Schubert 
1338091e9e46SCy Schubert void
1339091e9e46SCy Schubert rpz_remove_rr(struct rpz* r, size_t aznamelen, uint8_t* dname, size_t dnamelen,
1340091e9e46SCy Schubert 	uint16_t rr_type, uint16_t rr_class, uint8_t* rdatawl, size_t rdatalen)
1341091e9e46SCy Schubert {
1342091e9e46SCy Schubert 	size_t policydnamelen;
1343091e9e46SCy Schubert 	enum rpz_trigger t;
1344091e9e46SCy Schubert 	enum rpz_action a;
1345091e9e46SCy Schubert 	uint8_t* policydname;
1346091e9e46SCy Schubert 
1347091e9e46SCy Schubert 	if(!(policydname = calloc(1, LDNS_MAX_DOMAINLEN + 1)))
1348091e9e46SCy Schubert 		return;
1349091e9e46SCy Schubert 
1350091e9e46SCy Schubert 	a = rpz_rr_to_action(rr_type, rdatawl, rdatalen);
1351091e9e46SCy Schubert 	if(a == RPZ_INVALID_ACTION) {
1352091e9e46SCy Schubert 		free(policydname);
1353091e9e46SCy Schubert 		return;
1354091e9e46SCy Schubert 	}
1355091e9e46SCy Schubert 	if(!(policydnamelen = strip_dname_origin(dname, dnamelen, aznamelen,
1356091e9e46SCy Schubert 		policydname, LDNS_MAX_DOMAINLEN + 1))) {
1357091e9e46SCy Schubert 		free(policydname);
1358091e9e46SCy Schubert 		return;
1359091e9e46SCy Schubert 	}
1360091e9e46SCy Schubert 	t = rpz_dname_to_trigger(policydname, policydnamelen);
1361091e9e46SCy Schubert 	if(t == RPZ_QNAME_TRIGGER) {
1362091e9e46SCy Schubert 		rpz_remove_qname_trigger(r, policydname, policydnamelen, a,
1363091e9e46SCy Schubert 			rr_type, rr_class, rdatawl, rdatalen);
1364091e9e46SCy Schubert 	} else if(t == RPZ_RESPONSE_IP_TRIGGER) {
1365091e9e46SCy Schubert 		rpz_remove_response_ip_trigger(r, policydname, policydnamelen,
1366091e9e46SCy Schubert 			a, rr_type, rdatawl, rdatalen);
1367091e9e46SCy Schubert 	}
1368091e9e46SCy Schubert 	free(policydname);
1369091e9e46SCy Schubert }
1370091e9e46SCy Schubert 
1371091e9e46SCy Schubert /** print log information for an applied RPZ policy. Based on local-zone's
1372091e9e46SCy Schubert  * lz_inform_print().
137324e36522SCy Schubert  * The repinfo contains the reply address. If it is NULL, the module
137424e36522SCy Schubert  * state is used to report the first IP address (if any).
137524e36522SCy Schubert  * The dname is used, for the applied rpz, if NULL, addrnode is used.
1376091e9e46SCy Schubert  */
1377091e9e46SCy Schubert static void
137824e36522SCy Schubert log_rpz_apply(char* trigger, uint8_t* dname, struct addr_tree_node* addrnode,
137924e36522SCy Schubert 	enum rpz_action a, struct query_info* qinfo,
138024e36522SCy Schubert 	struct comm_reply* repinfo, struct module_qstate* ms, char* log_name)
1381091e9e46SCy Schubert {
138224e36522SCy Schubert 	char ip[128], txt[512], portstr[32];
1383091e9e46SCy Schubert 	char dnamestr[LDNS_MAX_DOMAINLEN+1];
138424e36522SCy Schubert 	uint16_t port = 0;
138524e36522SCy Schubert 	if(dname) {
1386091e9e46SCy Schubert 		dname_str(dname, dnamestr);
138724e36522SCy Schubert 	} else if(addrnode) {
13889cf5bc93SCy Schubert 		char addrbuf[128];
13899cf5bc93SCy Schubert 		addr_to_str(&addrnode->addr, addrnode->addrlen, addrbuf, sizeof(addrbuf));
13909cf5bc93SCy Schubert 		snprintf(dnamestr, sizeof(dnamestr), "%s/%d", addrbuf, addrnode->net);
139124e36522SCy Schubert 	} else {
139224e36522SCy Schubert 		dnamestr[0]=0;
139324e36522SCy Schubert 	}
139424e36522SCy Schubert 	if(repinfo) {
1395091e9e46SCy Schubert 		addr_to_str(&repinfo->addr, repinfo->addrlen, ip, sizeof(ip));
139624e36522SCy Schubert 		port = ntohs(((struct sockaddr_in*)&repinfo->addr)->sin_port);
139724e36522SCy Schubert 	} else if(ms && ms->mesh_info && ms->mesh_info->reply_list) {
139824e36522SCy Schubert 		addr_to_str(&ms->mesh_info->reply_list->query_reply.addr, ms->mesh_info->reply_list->query_reply.addrlen, ip, sizeof(ip));
139924e36522SCy Schubert 		port = ntohs(((struct sockaddr_in*)&ms->mesh_info->reply_list->query_reply.addr)->sin_port);
140024e36522SCy Schubert 	} else {
140124e36522SCy Schubert 		ip[0]=0;
140224e36522SCy Schubert 		port = 0;
140324e36522SCy Schubert 	}
140424e36522SCy Schubert 	snprintf(portstr, sizeof(portstr), "@%u", (unsigned)port);
140524e36522SCy Schubert 	snprintf(txt, sizeof(txt), "rpz: applied %s%s%s%s%s%s %s %s%s",
140624e36522SCy Schubert 		(log_name?"[":""), (log_name?log_name:""), (log_name?"] ":""),
140724e36522SCy Schubert 		(strcmp(trigger,"qname")==0?"":trigger),
140824e36522SCy Schubert 		(strcmp(trigger,"qname")==0?"":" "),
140924e36522SCy Schubert 		dnamestr, rpz_action_to_string(a),
141024e36522SCy Schubert 		(ip[0]?ip:""), (ip[0]?portstr:""));
1411091e9e46SCy Schubert 	log_nametypeclass(0, txt, qinfo->qname, qinfo->qtype, qinfo->qclass);
1412091e9e46SCy Schubert }
1413091e9e46SCy Schubert 
141424e36522SCy Schubert static struct clientip_synthesized_rr*
141524e36522SCy Schubert rpz_ipbased_trigger_lookup(struct clientip_synthesized_rrset* set,
141624e36522SCy Schubert 	struct sockaddr_storage* addr, socklen_t addrlen, char* triggername)
1417091e9e46SCy Schubert {
141824e36522SCy Schubert 	struct clientip_synthesized_rr* raddr = NULL;
141924e36522SCy Schubert 	enum rpz_action action = RPZ_INVALID_ACTION;
142024e36522SCy Schubert 
142124e36522SCy Schubert 	lock_rw_rdlock(&set->lock);
142224e36522SCy Schubert 
142324e36522SCy Schubert 	raddr = (struct clientip_synthesized_rr*)addr_tree_lookup(&set->entries,
142424e36522SCy Schubert 			addr, addrlen);
142524e36522SCy Schubert 	if(raddr != NULL) {
142624e36522SCy Schubert 		lock_rw_rdlock(&raddr->lock);
142724e36522SCy Schubert 		action = raddr->action;
142824e36522SCy Schubert 		if(verbosity >= VERB_ALGO) {
142924e36522SCy Schubert 			char ip[256], net[256];
143024e36522SCy Schubert 			addr_to_str(addr, addrlen, ip, sizeof(ip));
143124e36522SCy Schubert 			addr_to_str(&raddr->node.addr, raddr->node.addrlen,
143224e36522SCy Schubert 				net, sizeof(net));
143324e36522SCy Schubert 			verbose(VERB_ALGO, "rpz: trigger %s %s/%d on %s action=%s",
143424e36522SCy Schubert 				triggername, net, raddr->node.net, ip, rpz_action_to_string(action));
143524e36522SCy Schubert 		}
143624e36522SCy Schubert 	}
143724e36522SCy Schubert 	lock_rw_unlock(&set->lock);
143824e36522SCy Schubert 
143924e36522SCy Schubert 	return raddr;
144024e36522SCy Schubert }
144124e36522SCy Schubert 
144224e36522SCy Schubert static inline
144324e36522SCy Schubert struct clientip_synthesized_rr*
144424e36522SCy Schubert rpz_resolve_client_action_and_zone(struct auth_zones* az, struct query_info* qinfo,
144524e36522SCy Schubert 	struct comm_reply* repinfo, uint8_t* taglist, size_t taglen,
144624e36522SCy Schubert 	struct ub_server_stats* stats,
144724e36522SCy Schubert 	/* output parameters */
144824e36522SCy Schubert 	struct local_zone** z_out, struct auth_zone** a_out, struct rpz** r_out)
144924e36522SCy Schubert {
145024e36522SCy Schubert 	struct clientip_synthesized_rr* node = NULL;
145124e36522SCy Schubert 	struct auth_zone* a = NULL;
145225039b37SCy Schubert 	struct rpz* r = NULL;
1453091e9e46SCy Schubert 	struct local_zone* z = NULL;
145424e36522SCy Schubert 
1455091e9e46SCy Schubert 	lock_rw_rdlock(&az->rpz_lock);
145624e36522SCy Schubert 
145725039b37SCy Schubert 	for(a = az->rpz_first; a; a = a->rpz_az_next) {
145825039b37SCy Schubert 		lock_rw_rdlock(&a->lock);
145925039b37SCy Schubert 		r = a->rpz;
146024e36522SCy Schubert 		if(r->disabled) {
146124e36522SCy Schubert 			lock_rw_unlock(&a->lock);
146224e36522SCy Schubert 			continue;
146324e36522SCy Schubert 		}
146424e36522SCy Schubert 		if(r->taglist && !taglist_intersect(r->taglist,
146524e36522SCy Schubert 					r->taglistlen, taglist, taglen)) {
146624e36522SCy Schubert 			lock_rw_unlock(&a->lock);
146724e36522SCy Schubert 			continue;
146824e36522SCy Schubert 		}
146924e36522SCy Schubert 		z = rpz_find_zone(r->local_zones, qinfo->qname, qinfo->qname_len,
1470091e9e46SCy Schubert 			qinfo->qclass, 0, 0, 0);
147124e36522SCy Schubert 		node = rpz_ipbased_trigger_lookup(r->client_set, &repinfo->addr, repinfo->addrlen, "clientip");
147224e36522SCy Schubert 		if((z || node) && r->action_override == RPZ_DISABLED_ACTION) {
1473091e9e46SCy Schubert 			if(r->log)
147424e36522SCy Schubert 				log_rpz_apply((node?"clientip":"qname"),
147524e36522SCy Schubert 					(z?z->name:NULL),
147624e36522SCy Schubert 					(node?&node->node:NULL),
1477091e9e46SCy Schubert 					r->action_override,
147824e36522SCy Schubert 					qinfo, repinfo, NULL, r->log_name);
1479091e9e46SCy Schubert 			stats->rpz_action[r->action_override]++;
148024e36522SCy Schubert 			if(z != NULL) {
1481091e9e46SCy Schubert 				lock_rw_unlock(&z->lock);
1482091e9e46SCy Schubert 				z = NULL;
1483091e9e46SCy Schubert 			}
148424e36522SCy Schubert 			if(node != NULL) {
148524e36522SCy Schubert 				lock_rw_unlock(&node->lock);
148624e36522SCy Schubert 				node = NULL;
148724e36522SCy Schubert 			}
148824e36522SCy Schubert 		}
148924e36522SCy Schubert 		if(z || node) {
1490091e9e46SCy Schubert 			break;
1491091e9e46SCy Schubert 		}
149224e36522SCy Schubert 		/* not found in this auth_zone */
149325039b37SCy Schubert 		lock_rw_unlock(&a->lock);
149424e36522SCy Schubert 	}
149524e36522SCy Schubert 
149624e36522SCy Schubert 	lock_rw_unlock(&az->rpz_lock);
149724e36522SCy Schubert 
149824e36522SCy Schubert 	*r_out = r;
149924e36522SCy Schubert 	*a_out = a;
150024e36522SCy Schubert 	*z_out = z;
150124e36522SCy Schubert 
150224e36522SCy Schubert 	return node;
150324e36522SCy Schubert }
150424e36522SCy Schubert 
150524e36522SCy Schubert static inline int
150624e36522SCy Schubert rpz_is_udp_query(struct comm_reply* repinfo) {
150724e36522SCy Schubert 	return repinfo != NULL
150824e36522SCy Schubert 			? (repinfo->c != NULL
150924e36522SCy Schubert 				? repinfo->c->type == comm_udp
151024e36522SCy Schubert 				: 0)
151124e36522SCy Schubert 			: 0;
151224e36522SCy Schubert }
151324e36522SCy Schubert 
151424e36522SCy Schubert /** encode answer consisting of 1 rrset */
151524e36522SCy Schubert static int
151624e36522SCy Schubert rpz_local_encode(struct module_env* env, struct query_info* qinfo,
151724e36522SCy Schubert 	struct edns_data* edns, struct comm_reply* repinfo, sldns_buffer* buf,
151824e36522SCy Schubert 	struct regional* temp, struct ub_packed_rrset_key* rrset, int ansec,
151924e36522SCy Schubert 	int rcode, struct ub_packed_rrset_key* soa_rrset)
152024e36522SCy Schubert {
152124e36522SCy Schubert 	struct reply_info rep;
152224e36522SCy Schubert 	uint16_t udpsize;
152324e36522SCy Schubert 	struct ub_packed_rrset_key* rrsetlist[3];
152424e36522SCy Schubert 
152524e36522SCy Schubert 	memset(&rep, 0, sizeof(rep));
152624e36522SCy Schubert 	rep.flags = (uint16_t)((BIT_QR | BIT_AA | BIT_RA) | rcode);
152724e36522SCy Schubert 	rep.qdcount = 1;
152824e36522SCy Schubert 	rep.rrset_count = ansec;
152924e36522SCy Schubert 	rep.rrsets = rrsetlist;
153024e36522SCy Schubert 	if(ansec > 0) {
153124e36522SCy Schubert 		rep.an_numrrsets = 1;
153224e36522SCy Schubert 		rep.rrsets[0] = rrset;
153324e36522SCy Schubert 		rep.ttl = ((struct packed_rrset_data*)rrset->entry.data)->rr_ttl[0];
153424e36522SCy Schubert 	}
153524e36522SCy Schubert 	if(soa_rrset != NULL) {
153624e36522SCy Schubert 		rep.ar_numrrsets = 1;
153724e36522SCy Schubert 		rep.rrsets[rep.rrset_count] = soa_rrset;
153824e36522SCy Schubert 		rep.rrset_count ++;
153924e36522SCy Schubert 		if(rep.ttl < ((struct packed_rrset_data*)soa_rrset->entry.data)->rr_ttl[0]) {
154024e36522SCy Schubert 			rep.ttl = ((struct packed_rrset_data*)soa_rrset->entry.data)->rr_ttl[0];
154124e36522SCy Schubert 		}
154224e36522SCy Schubert 	}
154324e36522SCy Schubert 
154424e36522SCy Schubert 	udpsize = edns->udp_size;
154524e36522SCy Schubert 	edns->edns_version = EDNS_ADVERTISED_VERSION;
154624e36522SCy Schubert 	edns->udp_size = EDNS_ADVERTISED_SIZE;
154724e36522SCy Schubert 	edns->ext_rcode = 0;
154824e36522SCy Schubert 	edns->bits &= EDNS_DO;
154924e36522SCy Schubert 	if(!inplace_cb_reply_local_call(env, qinfo, NULL, &rep, rcode, edns,
155024e36522SCy Schubert 		repinfo, temp, env->now_tv) ||
155124e36522SCy Schubert 	  !reply_info_answer_encode(qinfo, &rep,
155224e36522SCy Schubert 		*(uint16_t*)sldns_buffer_begin(buf), sldns_buffer_read_u16_at(buf, 2),
155324e36522SCy Schubert 		buf, 0, 0, temp, udpsize, edns, (int)(edns->bits&EDNS_DO), 0)) {
155424e36522SCy Schubert 		error_encode(buf, (LDNS_RCODE_SERVFAIL|BIT_AA), qinfo,
155524e36522SCy Schubert 			*(uint16_t*)sldns_buffer_begin(buf),
155624e36522SCy Schubert 			sldns_buffer_read_u16_at(buf, 2), edns);
155724e36522SCy Schubert 	}
155824e36522SCy Schubert 
155924e36522SCy Schubert 	return 1;
156024e36522SCy Schubert }
156124e36522SCy Schubert 
156224e36522SCy Schubert static struct local_rrset*
1563*a39a5a69SCy Schubert rpz_find_synthesized_rrset(uint16_t qtype,
1564*a39a5a69SCy Schubert 	struct clientip_synthesized_rr* data)
1565*a39a5a69SCy Schubert {
156624e36522SCy Schubert 	struct local_rrset* cursor = data->data;
156724e36522SCy Schubert 	while( cursor != NULL) {
156824e36522SCy Schubert 		struct packed_rrset_key* packed_rrset = &cursor->rrset->rk;
156924e36522SCy Schubert 		if(htons(qtype) == packed_rrset->type) {
157024e36522SCy Schubert 			return cursor;
157124e36522SCy Schubert 		}
157224e36522SCy Schubert 		cursor = cursor->next;
157324e36522SCy Schubert 	}
157424e36522SCy Schubert 	return NULL;
157524e36522SCy Schubert }
157624e36522SCy Schubert 
157724e36522SCy Schubert /** allocate SOA record ubrrsetkey in region */
157824e36522SCy Schubert static struct ub_packed_rrset_key*
157924e36522SCy Schubert make_soa_ubrrset(struct auth_zone* auth_zone, struct auth_rrset* soa,
158024e36522SCy Schubert 	struct regional* temp)
158124e36522SCy Schubert {
158224e36522SCy Schubert 	struct ub_packed_rrset_key csoa;
158324e36522SCy Schubert 	if(!soa)
158424e36522SCy Schubert 		return NULL;
158524e36522SCy Schubert 	memset(&csoa, 0, sizeof(csoa));
158624e36522SCy Schubert 	csoa.entry.key = &csoa;
158724e36522SCy Schubert 	csoa.rk.rrset_class = htons(LDNS_RR_CLASS_IN);
158824e36522SCy Schubert 	csoa.rk.type = htons(LDNS_RR_TYPE_SOA);
158924e36522SCy Schubert 	csoa.rk.flags |= PACKED_RRSET_FIXEDTTL
159024e36522SCy Schubert 		| PACKED_RRSET_RPZ;
159124e36522SCy Schubert 	csoa.rk.dname = auth_zone->name;
159224e36522SCy Schubert 	csoa.rk.dname_len = auth_zone->namelen;
159324e36522SCy Schubert 	csoa.entry.hash = rrset_key_hash(&csoa.rk);
159424e36522SCy Schubert 	csoa.entry.data = soa->data;
159524e36522SCy Schubert 	return respip_copy_rrset(&csoa, temp);
159624e36522SCy Schubert }
159724e36522SCy Schubert 
159824e36522SCy Schubert static void
159924e36522SCy Schubert rpz_apply_clientip_localdata_action(struct clientip_synthesized_rr* raddr,
160024e36522SCy Schubert 	struct module_env* env, struct query_info* qinfo,
160124e36522SCy Schubert 	struct edns_data* edns, struct comm_reply* repinfo, sldns_buffer* buf,
160224e36522SCy Schubert 	struct regional* temp, struct auth_zone* auth_zone)
160324e36522SCy Schubert {
160424e36522SCy Schubert 	struct local_rrset* rrset;
160524e36522SCy Schubert 	enum rpz_action action = RPZ_INVALID_ACTION;
160624e36522SCy Schubert 	struct ub_packed_rrset_key* rp = NULL;
160724e36522SCy Schubert 	struct ub_packed_rrset_key* rsoa = NULL;
160824e36522SCy Schubert 	int rcode = LDNS_RCODE_NOERROR|BIT_AA;
160924e36522SCy Schubert 	int rrset_count = 1;
161024e36522SCy Schubert 
161124e36522SCy Schubert 	/* prepare synthesized answer for client */
161224e36522SCy Schubert 	action = raddr->action;
161324e36522SCy Schubert 	if(action == RPZ_LOCAL_DATA_ACTION && raddr->data == NULL ) {
161424e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: bug: local-data action but no local data");
161524e36522SCy Schubert 		return;
161624e36522SCy Schubert 	}
161724e36522SCy Schubert 
161824e36522SCy Schubert 	/* check query type / rr type */
161924e36522SCy Schubert 	rrset = rpz_find_synthesized_rrset(qinfo->qtype, raddr);
162024e36522SCy Schubert 	if(rrset == NULL) {
162124e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: unable to find local-data for query");
162224e36522SCy Schubert 		rrset_count = 0;
162324e36522SCy Schubert 		goto nodata;
162424e36522SCy Schubert 	}
162524e36522SCy Schubert 
162624e36522SCy Schubert 	rp = respip_copy_rrset(rrset->rrset, temp);
162724e36522SCy Schubert 	if(!rp) {
162824e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: local data action: out of memory");
162924e36522SCy Schubert 		return;
163024e36522SCy Schubert 	}
163124e36522SCy Schubert 
163224e36522SCy Schubert 	rp->rk.flags |= PACKED_RRSET_FIXEDTTL | PACKED_RRSET_RPZ;
163324e36522SCy Schubert 	rp->rk.dname = qinfo->qname;
163424e36522SCy Schubert 	rp->rk.dname_len = qinfo->qname_len;
163524e36522SCy Schubert 	rp->entry.hash = rrset_key_hash(&rp->rk);
163624e36522SCy Schubert nodata:
163724e36522SCy Schubert 	if(auth_zone) {
163824e36522SCy Schubert 		struct auth_rrset* soa = NULL;
163924e36522SCy Schubert 		soa = auth_zone_get_soa_rrset(auth_zone);
164024e36522SCy Schubert 		if(soa) {
164124e36522SCy Schubert 			rsoa = make_soa_ubrrset(auth_zone, soa, temp);
164224e36522SCy Schubert 			if(!rsoa) {
164324e36522SCy Schubert 				verbose(VERB_ALGO, "rpz: local data action soa: out of memory");
164424e36522SCy Schubert 				return;
164524e36522SCy Schubert 			}
164624e36522SCy Schubert 		}
164724e36522SCy Schubert 	}
164824e36522SCy Schubert 
164924e36522SCy Schubert 	rpz_local_encode(env, qinfo, edns, repinfo, buf, temp, rp,
165024e36522SCy Schubert 		rrset_count, rcode, rsoa);
165124e36522SCy Schubert }
165224e36522SCy Schubert 
165324e36522SCy Schubert /** add additional section SOA record to the reply.
165424e36522SCy Schubert  * Since this gets fed into the normal iterator answer creation, it
165524e36522SCy Schubert  * gets minimal-responses applied to it, that can remove the additional SOA
165624e36522SCy Schubert  * again. */
165724e36522SCy Schubert static int
165824e36522SCy Schubert rpz_add_soa(struct reply_info* rep, struct module_qstate* ms,
165924e36522SCy Schubert 	struct auth_zone* az)
166024e36522SCy Schubert {
166124e36522SCy Schubert 	struct auth_rrset* soa = NULL;
166224e36522SCy Schubert 	struct ub_packed_rrset_key* rsoa = NULL;
166324e36522SCy Schubert 	struct ub_packed_rrset_key** prevrrsets;
166424e36522SCy Schubert 	if(!az) return 1;
166524e36522SCy Schubert 	soa = auth_zone_get_soa_rrset(az);
166624e36522SCy Schubert 	if(!soa) return 1;
166724e36522SCy Schubert 	if(!rep) return 0;
166824e36522SCy Schubert 	rsoa = make_soa_ubrrset(az, soa, ms->region);
166924e36522SCy Schubert 	if(!rsoa) return 0;
167024e36522SCy Schubert 	prevrrsets = rep->rrsets;
167124e36522SCy Schubert 	rep->rrsets = regional_alloc_zero(ms->region,
167224e36522SCy Schubert 		sizeof(*rep->rrsets)*(rep->rrset_count+1));
167324e36522SCy Schubert 	if(!rep->rrsets)
167424e36522SCy Schubert 		return 0;
167524e36522SCy Schubert 	if(prevrrsets && rep->rrset_count > 0)
167624e36522SCy Schubert 		memcpy(rep->rrsets, prevrrsets, rep->rrset_count*sizeof(*rep->rrsets));
167724e36522SCy Schubert 	rep->rrset_count++;
167824e36522SCy Schubert 	rep->ar_numrrsets++;
167924e36522SCy Schubert 	rep->rrsets[rep->rrset_count-1] = rsoa;
168024e36522SCy Schubert 	return 1;
168124e36522SCy Schubert }
168224e36522SCy Schubert 
168324e36522SCy Schubert static inline struct dns_msg*
168424e36522SCy Schubert rpz_dns_msg_new(struct regional* region)
168524e36522SCy Schubert {
168624e36522SCy Schubert 	struct dns_msg* msg =
168724e36522SCy Schubert 			(struct dns_msg*)regional_alloc(region,
168824e36522SCy Schubert 							sizeof(struct dns_msg));
168924e36522SCy Schubert 	if(msg == NULL) { return NULL; }
169024e36522SCy Schubert 	memset(msg, 0, sizeof(struct dns_msg));
169124e36522SCy Schubert 
169224e36522SCy Schubert 	return msg;
169324e36522SCy Schubert }
169424e36522SCy Schubert 
169524e36522SCy Schubert static inline struct dns_msg*
169624e36522SCy Schubert rpz_synthesize_nodata(struct rpz* ATTR_UNUSED(r), struct module_qstate* ms,
169724e36522SCy Schubert 	struct query_info* qinfo, struct auth_zone* az)
169824e36522SCy Schubert {
169924e36522SCy Schubert 	struct dns_msg* msg = rpz_dns_msg_new(ms->region);
170024e36522SCy Schubert 	if(msg == NULL) { return msg; }
170124e36522SCy Schubert 	msg->qinfo = *qinfo;
170224e36522SCy Schubert 	msg->rep = construct_reply_info_base(ms->region,
17039cf5bc93SCy Schubert 					     LDNS_RCODE_NOERROR | BIT_QR | BIT_AA | BIT_RA,
170424e36522SCy Schubert 					     1, /* qd */
170524e36522SCy Schubert 					     0, /* ttl */
170624e36522SCy Schubert 					     0, /* prettl */
170724e36522SCy Schubert 					     0, /* expttl */
170824e36522SCy Schubert 					     0, /* an */
170924e36522SCy Schubert 					     0, /* ns */
171024e36522SCy Schubert 					     0, /* ar */
171124e36522SCy Schubert 					     0, /* total */
171224e36522SCy Schubert 					     sec_status_insecure);
171324e36522SCy Schubert 	if(msg->rep)
171424e36522SCy Schubert 		msg->rep->authoritative = 1;
171524e36522SCy Schubert 	if(!rpz_add_soa(msg->rep, ms, az))
171624e36522SCy Schubert 		return NULL;
171724e36522SCy Schubert 	return msg;
171824e36522SCy Schubert }
171924e36522SCy Schubert 
172024e36522SCy Schubert static inline struct dns_msg*
17219cf5bc93SCy Schubert rpz_synthesize_nxdomain(struct rpz* r, struct module_qstate* ms,
172224e36522SCy Schubert 	struct query_info* qinfo, struct auth_zone* az)
172324e36522SCy Schubert {
172424e36522SCy Schubert 	struct dns_msg* msg = rpz_dns_msg_new(ms->region);
17259cf5bc93SCy Schubert 	uint16_t flags;
172624e36522SCy Schubert 	if(msg == NULL) { return msg; }
172724e36522SCy Schubert 	msg->qinfo = *qinfo;
17289cf5bc93SCy Schubert 	flags = LDNS_RCODE_NXDOMAIN | BIT_QR | BIT_AA | BIT_RA;
17299cf5bc93SCy Schubert 	if(r->signal_nxdomain_ra)
17309cf5bc93SCy Schubert 		flags &= ~BIT_RA;
173124e36522SCy Schubert 	msg->rep = construct_reply_info_base(ms->region,
17329cf5bc93SCy Schubert 					     flags,
173324e36522SCy Schubert 					     1, /* qd */
173424e36522SCy Schubert 					     0, /* ttl */
173524e36522SCy Schubert 					     0, /* prettl */
173624e36522SCy Schubert 					     0, /* expttl */
173724e36522SCy Schubert 					     0, /* an */
173824e36522SCy Schubert 					     0, /* ns */
173924e36522SCy Schubert 					     0, /* ar */
174024e36522SCy Schubert 					     0, /* total */
174124e36522SCy Schubert 					     sec_status_insecure);
174224e36522SCy Schubert 	if(msg->rep)
174324e36522SCy Schubert 		msg->rep->authoritative = 1;
174424e36522SCy Schubert 	if(!rpz_add_soa(msg->rep, ms, az))
174524e36522SCy Schubert 		return NULL;
174624e36522SCy Schubert 	return msg;
174724e36522SCy Schubert }
174824e36522SCy Schubert 
174924e36522SCy Schubert static inline struct dns_msg*
175024e36522SCy Schubert rpz_synthesize_localdata_from_rrset(struct rpz* ATTR_UNUSED(r), struct module_qstate* ms,
175124e36522SCy Schubert 	struct query_info* qi, struct local_rrset* rrset, struct auth_zone* az)
175224e36522SCy Schubert {
175324e36522SCy Schubert 	struct dns_msg* msg = NULL;
175424e36522SCy Schubert 	struct reply_info* new_reply_info;
175524e36522SCy Schubert 	struct ub_packed_rrset_key* rp;
175624e36522SCy Schubert 
175724e36522SCy Schubert 
175824e36522SCy Schubert 	msg = rpz_dns_msg_new(ms->region);
175924e36522SCy Schubert 	if(msg == NULL) { return NULL; }
176024e36522SCy Schubert 
176124e36522SCy Schubert         new_reply_info = construct_reply_info_base(ms->region,
17629cf5bc93SCy Schubert                                                    LDNS_RCODE_NOERROR | BIT_QR | BIT_AA | BIT_RA,
176324e36522SCy Schubert                                                    1, /* qd */
176424e36522SCy Schubert                                                    0, /* ttl */
176524e36522SCy Schubert                                                    0, /* prettl */
176624e36522SCy Schubert                                                    0, /* expttl */
176724e36522SCy Schubert                                                    1, /* an */
176824e36522SCy Schubert                                                    0, /* ns */
176924e36522SCy Schubert                                                    0, /* ar */
177024e36522SCy Schubert                                                    1, /* total */
177124e36522SCy Schubert                                                    sec_status_insecure);
177224e36522SCy Schubert 	if(new_reply_info == NULL) {
177324e36522SCy Schubert 		log_err("out of memory");
177424e36522SCy Schubert 		return NULL;
177524e36522SCy Schubert 	}
177624e36522SCy Schubert 	new_reply_info->authoritative = 1;
177724e36522SCy Schubert 	rp = respip_copy_rrset(rrset->rrset, ms->region);
177824e36522SCy Schubert 	if(rp == NULL) {
177924e36522SCy Schubert 		log_err("out of memory");
178024e36522SCy Schubert 		return NULL;
178124e36522SCy Schubert 	}
178224e36522SCy Schubert 	rp->rk.dname = qi->qname;
178324e36522SCy Schubert 	rp->rk.dname_len = qi->qname_len;
178424e36522SCy Schubert 	/* this rrset is from the rpz data, or synthesized.
178524e36522SCy Schubert 	 * It is not actually from the network, so we flag it with this
178624e36522SCy Schubert 	 * flags as a fake RRset. If later the cache is used to look up
178724e36522SCy Schubert 	 * rrsets, then the fake ones are not returned (if you look without
178824e36522SCy Schubert 	 * the flag). For like CNAME lookups from the iterator or A, AAAA
178924e36522SCy Schubert 	 * lookups for nameserver targets, it would use the without flag
179024e36522SCy Schubert 	 * actual data. So that the actual network data and fake data
179124e36522SCy Schubert 	 * are kept track of separately. */
179224e36522SCy Schubert 	rp->rk.flags |= PACKED_RRSET_RPZ;
179324e36522SCy Schubert 	new_reply_info->rrsets[0] = rp;
179424e36522SCy Schubert 	msg->rep = new_reply_info;
179524e36522SCy Schubert 	if(!rpz_add_soa(msg->rep, ms, az))
179624e36522SCy Schubert 		return NULL;
179724e36522SCy Schubert 	return msg;
179824e36522SCy Schubert }
179924e36522SCy Schubert 
180024e36522SCy Schubert static inline struct dns_msg*
180124e36522SCy Schubert rpz_synthesize_nsip_localdata(struct rpz* r, struct module_qstate* ms,
180224e36522SCy Schubert 	struct clientip_synthesized_rr* data, struct auth_zone* az)
180324e36522SCy Schubert {
180424e36522SCy Schubert 	struct query_info* qi = &ms->qinfo;
180524e36522SCy Schubert 	struct local_rrset* rrset;
180624e36522SCy Schubert 
180724e36522SCy Schubert 	rrset = rpz_find_synthesized_rrset(qi->qtype, data);
180824e36522SCy Schubert 	if(rrset == NULL) {
180924e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: nsip: no matching local data found");
181024e36522SCy Schubert 		return NULL;
181124e36522SCy Schubert 	}
181224e36522SCy Schubert 
181324e36522SCy Schubert 	return rpz_synthesize_localdata_from_rrset(r, ms, &ms->qinfo, rrset, az);
181424e36522SCy Schubert }
181524e36522SCy Schubert 
181624e36522SCy Schubert /* copy'n'paste from localzone.c */
181724e36522SCy Schubert static struct local_rrset*
181824e36522SCy Schubert local_data_find_type(struct local_data* data, uint16_t type, int alias_ok)
181924e36522SCy Schubert {
182024e36522SCy Schubert 	struct local_rrset* p;
182124e36522SCy Schubert 	type = htons(type);
182224e36522SCy Schubert 	for(p = data->rrsets; p; p = p->next) {
182324e36522SCy Schubert 		if(p->rrset->rk.type == type)
182424e36522SCy Schubert 			return p;
182524e36522SCy Schubert 		if(alias_ok && p->rrset->rk.type == htons(LDNS_RR_TYPE_CNAME))
182624e36522SCy Schubert 			return p;
182724e36522SCy Schubert 	}
182824e36522SCy Schubert 	return NULL;
182924e36522SCy Schubert }
183024e36522SCy Schubert 
183124e36522SCy Schubert /* based on localzone.c:local_data_answer() */
183224e36522SCy Schubert static inline struct dns_msg*
183324e36522SCy Schubert rpz_synthesize_nsdname_localdata(struct rpz* r, struct module_qstate* ms,
183424e36522SCy Schubert 	struct local_zone* z, struct matched_delegation_point const* match,
183524e36522SCy Schubert 	struct auth_zone* az)
183624e36522SCy Schubert {
183724e36522SCy Schubert 	struct local_data key;
183824e36522SCy Schubert 	struct local_data* ld;
183924e36522SCy Schubert 	struct local_rrset* rrset;
184024e36522SCy Schubert 
184124e36522SCy Schubert 	if(match->dname == NULL) { return NULL; }
184224e36522SCy Schubert 
184324e36522SCy Schubert 	key.node.key = &key;
184424e36522SCy Schubert 	key.name = match->dname;
184524e36522SCy Schubert 	key.namelen = match->dname_len;
184624e36522SCy Schubert 	key.namelabs = dname_count_labels(match->dname);
184724e36522SCy Schubert 
184824e36522SCy Schubert 	rpz_log_dname("nsdname local data", key.name, key.namelen);
184924e36522SCy Schubert 
185024e36522SCy Schubert 	ld = (struct local_data*)rbtree_search(&z->data, &key.node);
185124e36522SCy Schubert 	if(ld == NULL) {
185224e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: nsdname: impossible: qname not found");
185324e36522SCy Schubert 		return NULL;
185424e36522SCy Schubert 	}
185524e36522SCy Schubert 
185624e36522SCy Schubert 	rrset = local_data_find_type(ld, ms->qinfo.qtype, 1);
185724e36522SCy Schubert 	if(rrset == NULL) {
185824e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: nsdname: no matching local data found");
185924e36522SCy Schubert 		return NULL;
186024e36522SCy Schubert 	}
186124e36522SCy Schubert 
186224e36522SCy Schubert 	return rpz_synthesize_localdata_from_rrset(r, ms, &ms->qinfo, rrset, az);
186324e36522SCy Schubert }
186424e36522SCy Schubert 
186524e36522SCy Schubert /* like local_data_answer for qname triggers after a cname */
186624e36522SCy Schubert static struct dns_msg*
186724e36522SCy Schubert rpz_synthesize_qname_localdata_msg(struct rpz* r, struct module_qstate* ms,
186824e36522SCy Schubert 	struct query_info* qinfo, struct local_zone* z, struct auth_zone* az)
186924e36522SCy Schubert {
187024e36522SCy Schubert 	struct local_data key;
187124e36522SCy Schubert 	struct local_data* ld;
187224e36522SCy Schubert 	struct local_rrset* rrset;
187324e36522SCy Schubert 	key.node.key = &key;
187424e36522SCy Schubert 	key.name = qinfo->qname;
187524e36522SCy Schubert 	key.namelen = qinfo->qname_len;
187624e36522SCy Schubert 	key.namelabs = dname_count_labels(qinfo->qname);
187724e36522SCy Schubert 	ld = (struct local_data*)rbtree_search(&z->data, &key.node);
187824e36522SCy Schubert 	if(ld == NULL) {
187924e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: qname after cname: name not found");
188024e36522SCy Schubert 		return NULL;
188124e36522SCy Schubert 	}
188224e36522SCy Schubert 	rrset = local_data_find_type(ld, qinfo->qtype, 1);
188324e36522SCy Schubert 	if(rrset == NULL) {
188424e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: qname after cname: type not found");
188524e36522SCy Schubert 		return NULL;
188624e36522SCy Schubert 	}
188724e36522SCy Schubert 	return rpz_synthesize_localdata_from_rrset(r, ms, qinfo, rrset, az);
188824e36522SCy Schubert }
188924e36522SCy Schubert 
189024e36522SCy Schubert static int
189124e36522SCy Schubert rpz_synthesize_qname_localdata(struct module_env* env, struct rpz* r,
189224e36522SCy Schubert 	struct local_zone* z, enum localzone_type lzt, struct query_info* qinfo,
189324e36522SCy Schubert 	struct edns_data* edns, sldns_buffer* buf, struct regional* temp,
189424e36522SCy Schubert 	struct comm_reply* repinfo, struct ub_server_stats* stats)
189524e36522SCy Schubert {
189624e36522SCy Schubert 	struct local_data* ld = NULL;
189724e36522SCy Schubert 	int ret = 0;
189824e36522SCy Schubert 	if(r->action_override == RPZ_CNAME_OVERRIDE_ACTION) {
189924e36522SCy Schubert 		qinfo->local_alias = regional_alloc_zero(temp, sizeof(struct local_rrset));
190024e36522SCy Schubert 		if(qinfo->local_alias == NULL) {
1901091e9e46SCy Schubert 			return 0; /* out of memory */
1902091e9e46SCy Schubert 		}
190324e36522SCy Schubert 		qinfo->local_alias->rrset = regional_alloc_init(temp, r->cname_override,
1904091e9e46SCy Schubert 								sizeof(*r->cname_override));
190524e36522SCy Schubert 		if(qinfo->local_alias->rrset == NULL) {
1906091e9e46SCy Schubert 			return 0; /* out of memory */
1907091e9e46SCy Schubert 		}
1908091e9e46SCy Schubert 		qinfo->local_alias->rrset->rk.dname = qinfo->qname;
1909091e9e46SCy Schubert 		qinfo->local_alias->rrset->rk.dname_len = qinfo->qname_len;
191024e36522SCy Schubert 		if(r->log) {
191124e36522SCy Schubert 			log_rpz_apply("qname", z->name, NULL, RPZ_CNAME_OVERRIDE_ACTION,
191224e36522SCy Schubert 				      qinfo, repinfo, NULL, r->log_name);
191324e36522SCy Schubert 		}
1914091e9e46SCy Schubert 		stats->rpz_action[RPZ_CNAME_OVERRIDE_ACTION]++;
1915091e9e46SCy Schubert 		return 0;
1916091e9e46SCy Schubert 	}
1917091e9e46SCy Schubert 
1918091e9e46SCy Schubert 	if(lzt == local_zone_redirect && local_data_answer(z, env, qinfo,
1919091e9e46SCy Schubert 		edns, repinfo, buf, temp, dname_count_labels(qinfo->qname),
1920091e9e46SCy Schubert 		&ld, lzt, -1, NULL, 0, NULL, 0)) {
192124e36522SCy Schubert 		if(r->log) {
192224e36522SCy Schubert 			log_rpz_apply("qname", z->name, NULL,
1923091e9e46SCy Schubert 				localzone_type_to_rpz_action(lzt), qinfo,
192424e36522SCy Schubert 				repinfo, NULL, r->log_name);
192524e36522SCy Schubert 		}
1926091e9e46SCy Schubert 		stats->rpz_action[localzone_type_to_rpz_action(lzt)]++;
1927091e9e46SCy Schubert 		return !qinfo->local_alias;
1928091e9e46SCy Schubert 	}
1929091e9e46SCy Schubert 
1930091e9e46SCy Schubert 	ret = local_zones_zone_answer(z, env, qinfo, edns, repinfo, buf, temp,
1931091e9e46SCy Schubert 		0 /* no local data used */, lzt);
19329cf5bc93SCy Schubert 	if(r->signal_nxdomain_ra && LDNS_RCODE_WIRE(sldns_buffer_begin(buf))
19339cf5bc93SCy Schubert 		== LDNS_RCODE_NXDOMAIN)
19349cf5bc93SCy Schubert 		LDNS_RA_CLR(sldns_buffer_begin(buf));
193524e36522SCy Schubert 	if(r->log) {
193624e36522SCy Schubert 		log_rpz_apply("qname", z->name, NULL, localzone_type_to_rpz_action(lzt),
193724e36522SCy Schubert 			      qinfo, repinfo, NULL, r->log_name);
193824e36522SCy Schubert 	}
1939091e9e46SCy Schubert 	stats->rpz_action[localzone_type_to_rpz_action(lzt)]++;
194024e36522SCy Schubert 	return ret;
194124e36522SCy Schubert }
194224e36522SCy Schubert 
19439cf5bc93SCy Schubert static struct clientip_synthesized_rr*
194424e36522SCy Schubert rpz_delegation_point_ipbased_trigger_lookup(struct rpz* rpz, struct iter_qstate* is)
194524e36522SCy Schubert {
194624e36522SCy Schubert 	struct delegpt_addr* cursor;
194724e36522SCy Schubert 	struct clientip_synthesized_rr* action = NULL;
194824e36522SCy Schubert 	if(is->dp == NULL) { return NULL; }
194924e36522SCy Schubert 	for(cursor = is->dp->target_list;
195024e36522SCy Schubert 	    cursor != NULL;
195124e36522SCy Schubert 	    cursor = cursor->next_target) {
195224e36522SCy Schubert 		if(cursor->bogus) { continue; }
195324e36522SCy Schubert 		action = rpz_ipbased_trigger_lookup(rpz->ns_set, &cursor->addr,
195424e36522SCy Schubert 						    cursor->addrlen, "nsip");
195524e36522SCy Schubert 		if(action != NULL) { return action; }
195624e36522SCy Schubert 	}
195724e36522SCy Schubert 	return NULL;
195824e36522SCy Schubert }
195924e36522SCy Schubert 
19609cf5bc93SCy Schubert static struct dns_msg*
196124e36522SCy Schubert rpz_apply_nsip_trigger(struct module_qstate* ms, struct rpz* r,
196224e36522SCy Schubert 	struct clientip_synthesized_rr* raddr, struct auth_zone* az)
196324e36522SCy Schubert {
196424e36522SCy Schubert 	enum rpz_action action = raddr->action;
196524e36522SCy Schubert 	struct dns_msg* ret = NULL;
196624e36522SCy Schubert 
196724e36522SCy Schubert 	if(r->action_override != RPZ_NO_OVERRIDE_ACTION) {
196824e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: using override action=%s (replaces=%s)",
196924e36522SCy Schubert 			rpz_action_to_string(r->action_override), rpz_action_to_string(action));
197024e36522SCy Schubert 		action = r->action_override;
197124e36522SCy Schubert 	}
197224e36522SCy Schubert 
197324e36522SCy Schubert 	if(action == RPZ_LOCAL_DATA_ACTION && raddr->data == NULL) {
197424e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: bug: nsip local data action but no local data");
197524e36522SCy Schubert 		ret = rpz_synthesize_nodata(r, ms, &ms->qinfo, az);
197624e36522SCy Schubert 		goto done;
197724e36522SCy Schubert 	}
197824e36522SCy Schubert 
197924e36522SCy Schubert 	switch(action) {
198024e36522SCy Schubert 	case RPZ_NXDOMAIN_ACTION:
198124e36522SCy Schubert 		ret = rpz_synthesize_nxdomain(r, ms, &ms->qinfo, az);
198224e36522SCy Schubert 		break;
198324e36522SCy Schubert 	case RPZ_NODATA_ACTION:
198424e36522SCy Schubert 		ret = rpz_synthesize_nodata(r, ms, &ms->qinfo, az);
198524e36522SCy Schubert 		break;
198624e36522SCy Schubert 	case RPZ_TCP_ONLY_ACTION:
198724e36522SCy Schubert 		/* basically a passthru here but the tcp-only will be
198824e36522SCy Schubert 		 * honored before the query gets sent. */
198924e36522SCy Schubert 		ms->respip_action_info->action = respip_truncate;
199024e36522SCy Schubert 		ret = NULL;
199124e36522SCy Schubert 		break;
199224e36522SCy Schubert 	case RPZ_DROP_ACTION:
199324e36522SCy Schubert 		ret = rpz_synthesize_nodata(r, ms, &ms->qinfo, az);
199424e36522SCy Schubert 		ms->is_drop = 1;
199524e36522SCy Schubert 		break;
199624e36522SCy Schubert 	case RPZ_LOCAL_DATA_ACTION:
199724e36522SCy Schubert 		ret = rpz_synthesize_nsip_localdata(r, ms, raddr, az);
199824e36522SCy Schubert 		if(ret == NULL) { ret = rpz_synthesize_nodata(r, ms, &ms->qinfo, az); }
199924e36522SCy Schubert 		break;
200024e36522SCy Schubert 	case RPZ_PASSTHRU_ACTION:
200124e36522SCy Schubert 		ret = NULL;
2002*a39a5a69SCy Schubert 		ms->rpz_passthru = 1;
200324e36522SCy Schubert 		break;
200424e36522SCy Schubert 	default:
200524e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: nsip: bug: unhandled or invalid action: '%s'",
200624e36522SCy Schubert 			rpz_action_to_string(action));
200724e36522SCy Schubert 		ret = NULL;
200824e36522SCy Schubert 	}
200924e36522SCy Schubert 
201024e36522SCy Schubert done:
201124e36522SCy Schubert 	if(r->log)
201224e36522SCy Schubert 		log_rpz_apply("nsip", NULL, &raddr->node,
201324e36522SCy Schubert 			action, &ms->qinfo, NULL, ms, r->log_name);
201424e36522SCy Schubert 	if(ms->env->worker)
201524e36522SCy Schubert 		ms->env->worker->stats.rpz_action[action]++;
201624e36522SCy Schubert 	lock_rw_unlock(&raddr->lock);
201724e36522SCy Schubert 	return ret;
201824e36522SCy Schubert }
201924e36522SCy Schubert 
20209cf5bc93SCy Schubert static struct dns_msg*
202124e36522SCy Schubert rpz_apply_nsdname_trigger(struct module_qstate* ms, struct rpz* r,
202224e36522SCy Schubert 	struct local_zone* z, struct matched_delegation_point const* match,
202324e36522SCy Schubert 	struct auth_zone* az)
202424e36522SCy Schubert {
202524e36522SCy Schubert 	struct dns_msg* ret = NULL;
202624e36522SCy Schubert 	enum rpz_action action = localzone_type_to_rpz_action(z->type);
202724e36522SCy Schubert 
202824e36522SCy Schubert 	if(r->action_override != RPZ_NO_OVERRIDE_ACTION) {
202924e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: using override action=%s (replaces=%s)",
203024e36522SCy Schubert 			rpz_action_to_string(r->action_override), rpz_action_to_string(action));
203124e36522SCy Schubert 		action = r->action_override;
203224e36522SCy Schubert 	}
203324e36522SCy Schubert 
203424e36522SCy Schubert 	switch(action) {
203524e36522SCy Schubert 	case RPZ_NXDOMAIN_ACTION:
203624e36522SCy Schubert 		ret = rpz_synthesize_nxdomain(r, ms, &ms->qinfo, az);
203724e36522SCy Schubert 		break;
203824e36522SCy Schubert 	case RPZ_NODATA_ACTION:
203924e36522SCy Schubert 		ret = rpz_synthesize_nodata(r, ms, &ms->qinfo, az);
204024e36522SCy Schubert 		break;
204124e36522SCy Schubert 	case RPZ_TCP_ONLY_ACTION:
204224e36522SCy Schubert 		/* basically a passthru here but the tcp-only will be
204324e36522SCy Schubert 		 * honored before the query gets sent. */
204424e36522SCy Schubert 		ms->respip_action_info->action = respip_truncate;
204524e36522SCy Schubert 		ret = NULL;
204624e36522SCy Schubert 		break;
204724e36522SCy Schubert 	case RPZ_DROP_ACTION:
204824e36522SCy Schubert 		ret = rpz_synthesize_nodata(r, ms, &ms->qinfo, az);
204924e36522SCy Schubert 		ms->is_drop = 1;
205024e36522SCy Schubert 		break;
205124e36522SCy Schubert 	case RPZ_LOCAL_DATA_ACTION:
205224e36522SCy Schubert 		ret = rpz_synthesize_nsdname_localdata(r, ms, z, match, az);
205324e36522SCy Schubert 		if(ret == NULL) { ret = rpz_synthesize_nodata(r, ms, &ms->qinfo, az); }
205424e36522SCy Schubert 		break;
205524e36522SCy Schubert 	case RPZ_PASSTHRU_ACTION:
205624e36522SCy Schubert 		ret = NULL;
2057*a39a5a69SCy Schubert 		ms->rpz_passthru = 1;
205824e36522SCy Schubert 		break;
205924e36522SCy Schubert 	default:
206024e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: nsip: bug: unhandled or invalid action: '%s'",
206124e36522SCy Schubert 			rpz_action_to_string(action));
206224e36522SCy Schubert 		ret = NULL;
206324e36522SCy Schubert 	}
206424e36522SCy Schubert 
206524e36522SCy Schubert 	if(r->log)
206624e36522SCy Schubert 		log_rpz_apply("nsdname", match->dname, NULL,
206724e36522SCy Schubert 			action, &ms->qinfo, NULL, ms, r->log_name);
206824e36522SCy Schubert 	if(ms->env->worker)
206924e36522SCy Schubert 		ms->env->worker->stats.rpz_action[action]++;
207024e36522SCy Schubert 	lock_rw_unlock(&z->lock);
207124e36522SCy Schubert 	return ret;
207224e36522SCy Schubert }
207324e36522SCy Schubert 
207424e36522SCy Schubert static struct local_zone*
207524e36522SCy Schubert rpz_delegation_point_zone_lookup(struct delegpt* dp, struct local_zones* zones,
207624e36522SCy Schubert 	uint16_t qclass,
207724e36522SCy Schubert 	/* output parameter */
207824e36522SCy Schubert 	struct matched_delegation_point* match)
207924e36522SCy Schubert {
208024e36522SCy Schubert 	struct delegpt_ns* nameserver;
208124e36522SCy Schubert 	struct local_zone* z = NULL;
208224e36522SCy Schubert 
208324e36522SCy Schubert 	/* the rpz specs match the nameserver names (NS records), not the
208424e36522SCy Schubert 	 * name of the delegation point itself, to the nsdname triggers */
208524e36522SCy Schubert 	for(nameserver = dp->nslist;
208624e36522SCy Schubert 	    nameserver != NULL;
208724e36522SCy Schubert 	    nameserver = nameserver->next) {
208824e36522SCy Schubert 		z = rpz_find_zone(zones, nameserver->name, nameserver->namelen,
208924e36522SCy Schubert 				  qclass, 0, 0, 0);
209024e36522SCy Schubert 		if(z != NULL) {
209124e36522SCy Schubert 			match->dname = nameserver->name;
209224e36522SCy Schubert 			match->dname_len = nameserver->namelen;
209324e36522SCy Schubert 			if(verbosity >= VERB_ALGO) {
209424e36522SCy Schubert 				char nm[255+1], zn[255+1];
209524e36522SCy Schubert 				dname_str(match->dname, nm);
209624e36522SCy Schubert 				dname_str(z->name, zn);
209724e36522SCy Schubert 				if(strcmp(nm, zn) != 0)
209824e36522SCy Schubert 					verbose(VERB_ALGO, "rpz: trigger nsdname %s on %s action=%s",
209924e36522SCy Schubert 						zn, nm, rpz_action_to_string(localzone_type_to_rpz_action(z->type)));
210024e36522SCy Schubert 				else
210124e36522SCy Schubert 					verbose(VERB_ALGO, "rpz: trigger nsdname %s action=%s",
210224e36522SCy Schubert 						nm, rpz_action_to_string(localzone_type_to_rpz_action(z->type)));
210324e36522SCy Schubert 			}
210424e36522SCy Schubert 			break;
210524e36522SCy Schubert 		}
210624e36522SCy Schubert 	}
210724e36522SCy Schubert 
210824e36522SCy Schubert 	return z;
210924e36522SCy Schubert }
211024e36522SCy Schubert 
211124e36522SCy Schubert struct dns_msg*
211224e36522SCy Schubert rpz_callback_from_iterator_module(struct module_qstate* ms, struct iter_qstate* is)
211324e36522SCy Schubert {
211424e36522SCy Schubert 	struct auth_zones* az;
211524e36522SCy Schubert 	struct auth_zone* a;
211624e36522SCy Schubert 	struct clientip_synthesized_rr* raddr = NULL;
211724e36522SCy Schubert 	struct rpz* r = NULL;
211824e36522SCy Schubert 	struct local_zone* z = NULL;
211924e36522SCy Schubert 	struct matched_delegation_point match = {0};
212024e36522SCy Schubert 
2121*a39a5a69SCy Schubert 	if(ms->rpz_passthru) {
2122*a39a5a69SCy Schubert 		verbose(VERB_ALGO, "query is rpz_passthru, no further processing");
2123*a39a5a69SCy Schubert 		return NULL;
2124*a39a5a69SCy Schubert 	}
2125*a39a5a69SCy Schubert 
212624e36522SCy Schubert 	if(ms->env == NULL || ms->env->auth_zones == NULL) { return 0; }
212724e36522SCy Schubert 
212824e36522SCy Schubert 	az = ms->env->auth_zones;
212924e36522SCy Schubert 
213024e36522SCy Schubert 	verbose(VERB_ALGO, "rpz: iterator module callback: have_rpz=%d", az->rpz_first != NULL);
213124e36522SCy Schubert 
213224e36522SCy Schubert 	lock_rw_rdlock(&az->rpz_lock);
213324e36522SCy Schubert 
213424e36522SCy Schubert 	/* precedence of RPZ works, loosely, like this:
213524e36522SCy Schubert 	 * CNAMEs in order of the CNAME chain. rpzs in the order they are
213624e36522SCy Schubert 	 * configured. In an RPZ: first client-IP addr, then QNAME, then
213724e36522SCy Schubert 	 * response IP, then NSDNAME, then NSIP. Longest match first. Smallest
213824e36522SCy Schubert 	 * one from a set. */
213924e36522SCy Schubert 	/* we use the precedence rules for the topics and triggers that
214024e36522SCy Schubert 	 * are pertinent at this stage of the resolve processing */
214124e36522SCy Schubert 	for(a = az->rpz_first; a != NULL; a = a->rpz_az_next) {
214224e36522SCy Schubert 		lock_rw_rdlock(&a->lock);
214324e36522SCy Schubert 		r = a->rpz;
214424e36522SCy Schubert 		if(r->disabled) {
214524e36522SCy Schubert 			lock_rw_unlock(&a->lock);
214624e36522SCy Schubert 			continue;
214724e36522SCy Schubert 		}
214824e36522SCy Schubert 
214924e36522SCy Schubert 		/* the nsdname has precedence over the nsip triggers */
215024e36522SCy Schubert 		z = rpz_delegation_point_zone_lookup(is->dp, r->nsdname_zones,
215124e36522SCy Schubert 						     ms->qinfo.qclass, &match);
215224e36522SCy Schubert 		if(z != NULL) {
215324e36522SCy Schubert 			lock_rw_unlock(&a->lock);
215424e36522SCy Schubert 			break;
215524e36522SCy Schubert 		}
215624e36522SCy Schubert 
215724e36522SCy Schubert 		raddr = rpz_delegation_point_ipbased_trigger_lookup(r, is);
215824e36522SCy Schubert 		if(raddr != NULL) {
215924e36522SCy Schubert 			lock_rw_unlock(&a->lock);
216024e36522SCy Schubert 			break;
216124e36522SCy Schubert 		}
216224e36522SCy Schubert 		lock_rw_unlock(&a->lock);
216324e36522SCy Schubert 	}
216424e36522SCy Schubert 
216524e36522SCy Schubert 	lock_rw_unlock(&az->rpz_lock);
216624e36522SCy Schubert 
216724e36522SCy Schubert 	if(raddr == NULL && z == NULL) { return NULL; }
216824e36522SCy Schubert 	else if(raddr != NULL) {
216924e36522SCy Schubert 		if(z) {
217024e36522SCy Schubert 			lock_rw_unlock(&z->lock);
217124e36522SCy Schubert 		}
217224e36522SCy Schubert 		return rpz_apply_nsip_trigger(ms, r, raddr, a);
217324e36522SCy Schubert 	} else if(z != NULL) {
217424e36522SCy Schubert 		if(raddr) {
217524e36522SCy Schubert 			lock_rw_unlock(&raddr->lock);
217624e36522SCy Schubert 		}
217724e36522SCy Schubert 		return rpz_apply_nsdname_trigger(ms, r, z, &match, a);
217824e36522SCy Schubert 	} else { return NULL; }
217924e36522SCy Schubert }
218024e36522SCy Schubert 
218124e36522SCy Schubert struct dns_msg* rpz_callback_from_iterator_cname(struct module_qstate* ms,
218224e36522SCy Schubert 	struct iter_qstate* is)
218324e36522SCy Schubert {
218424e36522SCy Schubert 	struct auth_zones* az;
218524e36522SCy Schubert 	struct auth_zone* a = NULL;
218624e36522SCy Schubert 	struct rpz* r = NULL;
218724e36522SCy Schubert 	struct local_zone* z = NULL;
218824e36522SCy Schubert 	enum localzone_type lzt;
218924e36522SCy Schubert 	struct dns_msg* ret = NULL;
219024e36522SCy Schubert 
2191*a39a5a69SCy Schubert 	if(ms->rpz_passthru) {
2192*a39a5a69SCy Schubert 		verbose(VERB_ALGO, "query is rpz_passthru, no further processing");
2193*a39a5a69SCy Schubert 		return NULL;
2194*a39a5a69SCy Schubert 	}
2195*a39a5a69SCy Schubert 
219624e36522SCy Schubert 	if(ms->env == NULL || ms->env->auth_zones == NULL) { return 0; }
219724e36522SCy Schubert 	az = ms->env->auth_zones;
219824e36522SCy Schubert 
219924e36522SCy Schubert 	lock_rw_rdlock(&az->rpz_lock);
220024e36522SCy Schubert 
220124e36522SCy Schubert 	for(a = az->rpz_first; a; a = a->rpz_az_next) {
220224e36522SCy Schubert 		lock_rw_rdlock(&a->lock);
220324e36522SCy Schubert 		r = a->rpz;
220424e36522SCy Schubert 		if(r->disabled) {
220524e36522SCy Schubert 			lock_rw_unlock(&a->lock);
220624e36522SCy Schubert 			continue;
220724e36522SCy Schubert 		}
220824e36522SCy Schubert 		z = rpz_find_zone(r->local_zones, is->qchase.qname,
220924e36522SCy Schubert 			is->qchase.qname_len, is->qchase.qclass, 0, 0, 0);
221024e36522SCy Schubert 		if(z && r->action_override == RPZ_DISABLED_ACTION) {
221124e36522SCy Schubert 			if(r->log)
221224e36522SCy Schubert 				log_rpz_apply("qname", z->name, NULL,
221324e36522SCy Schubert 					r->action_override,
221424e36522SCy Schubert 					&ms->qinfo, NULL, ms, r->log_name);
221524e36522SCy Schubert 			if(ms->env->worker)
221624e36522SCy Schubert 				ms->env->worker->stats.rpz_action[r->action_override]++;
221724e36522SCy Schubert 			lock_rw_unlock(&z->lock);
221824e36522SCy Schubert 			z = NULL;
221924e36522SCy Schubert 		}
222024e36522SCy Schubert 		if(z) {
222124e36522SCy Schubert 			break;
222224e36522SCy Schubert 		}
222324e36522SCy Schubert 		/* not found in this auth_zone */
222424e36522SCy Schubert 		lock_rw_unlock(&a->lock);
222524e36522SCy Schubert 	}
222624e36522SCy Schubert 	lock_rw_unlock(&az->rpz_lock);
222724e36522SCy Schubert 
222824e36522SCy Schubert 	if(z == NULL)
222924e36522SCy Schubert 		return NULL;
223024e36522SCy Schubert 	if(r->action_override == RPZ_NO_OVERRIDE_ACTION) {
223124e36522SCy Schubert 		lzt = z->type;
223224e36522SCy Schubert 	} else {
223324e36522SCy Schubert 		lzt = rpz_action_to_localzone_type(r->action_override);
223424e36522SCy Schubert 	}
223524e36522SCy Schubert 
223624e36522SCy Schubert 	if(verbosity >= VERB_ALGO) {
223724e36522SCy Schubert 		char nm[255+1], zn[255+1];
223824e36522SCy Schubert 		dname_str(is->qchase.qname, nm);
223924e36522SCy Schubert 		dname_str(z->name, zn);
224024e36522SCy Schubert 		if(strcmp(zn, nm) != 0)
224124e36522SCy Schubert 			verbose(VERB_ALGO, "rpz: qname trigger after cname %s on %s, with action=%s",
224224e36522SCy Schubert 				zn, nm, rpz_action_to_string(localzone_type_to_rpz_action(lzt)));
224324e36522SCy Schubert 		else
224424e36522SCy Schubert 			verbose(VERB_ALGO, "rpz: qname trigger after cname %s, with action=%s",
224524e36522SCy Schubert 				nm, rpz_action_to_string(localzone_type_to_rpz_action(lzt)));
224624e36522SCy Schubert 	}
224724e36522SCy Schubert 	switch(localzone_type_to_rpz_action(lzt)) {
224824e36522SCy Schubert 	case RPZ_NXDOMAIN_ACTION:
224924e36522SCy Schubert 		ret = rpz_synthesize_nxdomain(r, ms, &is->qchase, a);
225024e36522SCy Schubert 		break;
225124e36522SCy Schubert 	case RPZ_NODATA_ACTION:
225224e36522SCy Schubert 		ret = rpz_synthesize_nodata(r, ms, &is->qchase, a);
225324e36522SCy Schubert 		break;
225424e36522SCy Schubert 	case RPZ_TCP_ONLY_ACTION:
225524e36522SCy Schubert 		/* basically a passthru here but the tcp-only will be
225624e36522SCy Schubert 		 * honored before the query gets sent. */
225724e36522SCy Schubert 		ms->respip_action_info->action = respip_truncate;
225824e36522SCy Schubert 		ret = NULL;
225924e36522SCy Schubert 		break;
226024e36522SCy Schubert 	case RPZ_DROP_ACTION:
226124e36522SCy Schubert 		ret = rpz_synthesize_nodata(r, ms, &is->qchase, a);
226224e36522SCy Schubert 		ms->is_drop = 1;
226324e36522SCy Schubert 		break;
226424e36522SCy Schubert 	case RPZ_LOCAL_DATA_ACTION:
226524e36522SCy Schubert 		ret = rpz_synthesize_qname_localdata_msg(r, ms, &is->qchase, z, a);
226624e36522SCy Schubert 		if(ret == NULL) { ret = rpz_synthesize_nodata(r, ms, &is->qchase, a); }
226724e36522SCy Schubert 		break;
226824e36522SCy Schubert 	case RPZ_PASSTHRU_ACTION:
226924e36522SCy Schubert 		ret = NULL;
2270*a39a5a69SCy Schubert 		ms->rpz_passthru = 1;
227124e36522SCy Schubert 		break;
227224e36522SCy Schubert 	default:
227324e36522SCy Schubert 		verbose(VERB_ALGO, "rpz: qname trigger after cname: bug: unhandled or invalid action: '%s'",
227424e36522SCy Schubert 			rpz_action_to_string(localzone_type_to_rpz_action(lzt)));
227524e36522SCy Schubert 		ret = NULL;
227624e36522SCy Schubert 	}
227724e36522SCy Schubert 	lock_rw_unlock(&z->lock);
227824e36522SCy Schubert 	lock_rw_unlock(&a->lock);
227924e36522SCy Schubert 	return ret;
228024e36522SCy Schubert }
228124e36522SCy Schubert 
228224e36522SCy Schubert static int
228324e36522SCy Schubert rpz_apply_maybe_clientip_trigger(struct auth_zones* az, struct module_env* env,
228424e36522SCy Schubert 	struct query_info* qinfo, struct edns_data* edns, struct comm_reply* repinfo,
228524e36522SCy Schubert 	uint8_t* taglist, size_t taglen, struct ub_server_stats* stats,
228624e36522SCy Schubert 	sldns_buffer* buf, struct regional* temp,
228724e36522SCy Schubert 	/* output parameters */
2288*a39a5a69SCy Schubert 	struct local_zone** z_out, struct auth_zone** a_out, struct rpz** r_out,
2289*a39a5a69SCy Schubert 	int* passthru)
229024e36522SCy Schubert {
229124e36522SCy Schubert 	int ret = 0;
229224e36522SCy Schubert 	enum rpz_action client_action;
229324e36522SCy Schubert 	struct clientip_synthesized_rr* node = rpz_resolve_client_action_and_zone(
229424e36522SCy Schubert 		az, qinfo, repinfo, taglist, taglen, stats, z_out, a_out, r_out);
229524e36522SCy Schubert 
229624e36522SCy Schubert 	client_action = ((node == NULL) ? RPZ_INVALID_ACTION : node->action);
2297*a39a5a69SCy Schubert 	if(client_action == RPZ_PASSTHRU_ACTION) {
2298*a39a5a69SCy Schubert 		*passthru = 1;
2299*a39a5a69SCy Schubert 	}
230024e36522SCy Schubert 	if(*z_out == NULL || (client_action != RPZ_INVALID_ACTION &&
230124e36522SCy Schubert 			      client_action != RPZ_PASSTHRU_ACTION)) {
230224e36522SCy Schubert 		if(client_action == RPZ_PASSTHRU_ACTION
230324e36522SCy Schubert 			|| client_action == RPZ_INVALID_ACTION
230424e36522SCy Schubert 			|| (client_action == RPZ_TCP_ONLY_ACTION
230524e36522SCy Schubert 				&& !rpz_is_udp_query(repinfo))) {
230624e36522SCy Schubert 			ret = 0;
230724e36522SCy Schubert 			goto done;
230824e36522SCy Schubert 		}
230924e36522SCy Schubert 		stats->rpz_action[client_action]++;
231024e36522SCy Schubert 		if(client_action == RPZ_LOCAL_DATA_ACTION) {
231124e36522SCy Schubert 			rpz_apply_clientip_localdata_action(node, env, qinfo,
231224e36522SCy Schubert 				edns, repinfo, buf, temp, *a_out);
231324e36522SCy Schubert 		} else {
231424e36522SCy Schubert 			if(*r_out && (*r_out)->log)
231524e36522SCy Schubert 				log_rpz_apply(
231624e36522SCy Schubert 					(node?"clientip":"qname"),
231724e36522SCy Schubert 					((*z_out)?(*z_out)->name:NULL),
231824e36522SCy Schubert 					(node?&node->node:NULL),
231924e36522SCy Schubert 					client_action, qinfo, repinfo, NULL,
232024e36522SCy Schubert 					(*r_out)->log_name);
232124e36522SCy Schubert 			local_zones_zone_answer(*z_out /*likely NULL, no zone*/, env, qinfo, edns,
232224e36522SCy Schubert 				repinfo, buf, temp, 0 /* no local data used */,
232324e36522SCy Schubert 				rpz_action_to_localzone_type(client_action));
23249cf5bc93SCy Schubert 			if(*r_out && (*r_out)->signal_nxdomain_ra &&
23259cf5bc93SCy Schubert 				LDNS_RCODE_WIRE(sldns_buffer_begin(buf))
23269cf5bc93SCy Schubert 				== LDNS_RCODE_NXDOMAIN)
23279cf5bc93SCy Schubert 				LDNS_RA_CLR(sldns_buffer_begin(buf));
232824e36522SCy Schubert 		}
232924e36522SCy Schubert 		ret = 1;
233024e36522SCy Schubert 		goto done;
233124e36522SCy Schubert 	}
233224e36522SCy Schubert 	ret = -1;
233324e36522SCy Schubert done:
233424e36522SCy Schubert 	if(node != NULL) {
233524e36522SCy Schubert 		lock_rw_unlock(&node->lock);
233624e36522SCy Schubert 	}
233724e36522SCy Schubert 	return ret;
233824e36522SCy Schubert }
233924e36522SCy Schubert 
234024e36522SCy Schubert int
234124e36522SCy Schubert rpz_callback_from_worker_request(struct auth_zones* az, struct module_env* env,
234224e36522SCy Schubert 	struct query_info* qinfo, struct edns_data* edns, sldns_buffer* buf,
234324e36522SCy Schubert 	struct regional* temp, struct comm_reply* repinfo, uint8_t* taglist,
2344*a39a5a69SCy Schubert 	size_t taglen, struct ub_server_stats* stats, int* passthru)
234524e36522SCy Schubert {
234624e36522SCy Schubert 	struct rpz* r = NULL;
234724e36522SCy Schubert 	struct auth_zone* a = NULL;
234824e36522SCy Schubert 	struct local_zone* z = NULL;
234924e36522SCy Schubert 	int ret;
235024e36522SCy Schubert 	enum localzone_type lzt;
235124e36522SCy Schubert 
235224e36522SCy Schubert 	int clientip_trigger = rpz_apply_maybe_clientip_trigger(az, env, qinfo,
2353*a39a5a69SCy Schubert 		edns, repinfo, taglist, taglen, stats, buf, temp, &z, &a, &r,
2354*a39a5a69SCy Schubert 		passthru);
235524e36522SCy Schubert 	if(clientip_trigger >= 0) {
235624e36522SCy Schubert 		if(a) {
235724e36522SCy Schubert 			lock_rw_unlock(&a->lock);
235824e36522SCy Schubert 		}
235924e36522SCy Schubert 		if(z) {
236024e36522SCy Schubert 			lock_rw_unlock(&z->lock);
236124e36522SCy Schubert 		}
236224e36522SCy Schubert 		return clientip_trigger;
236324e36522SCy Schubert 	}
236424e36522SCy Schubert 
236524e36522SCy Schubert 	if(z == NULL) {
236624e36522SCy Schubert 		if(a) {
236724e36522SCy Schubert 			lock_rw_unlock(&a->lock);
236824e36522SCy Schubert 		}
236924e36522SCy Schubert 		return 0;
237024e36522SCy Schubert 	}
237124e36522SCy Schubert 
237224e36522SCy Schubert 	log_assert(r);
237324e36522SCy Schubert 
237424e36522SCy Schubert 	if(r->action_override == RPZ_NO_OVERRIDE_ACTION) {
237524e36522SCy Schubert 		lzt = z->type;
237624e36522SCy Schubert 	} else {
237724e36522SCy Schubert 		lzt = rpz_action_to_localzone_type(r->action_override);
237824e36522SCy Schubert 	}
2379*a39a5a69SCy Schubert 	if(r->action_override == RPZ_PASSTHRU_ACTION ||
2380*a39a5a69SCy Schubert 		lzt == local_zone_always_transparent /* RPZ_PASSTHRU_ACTION */) {
2381*a39a5a69SCy Schubert 		*passthru = 1;
2382*a39a5a69SCy Schubert 	}
238324e36522SCy Schubert 
238424e36522SCy Schubert 	if(verbosity >= VERB_ALGO) {
238524e36522SCy Schubert 		char nm[255+1], zn[255+1];
238624e36522SCy Schubert 		dname_str(qinfo->qname, nm);
238724e36522SCy Schubert 		dname_str(z->name, zn);
238824e36522SCy Schubert 		if(strcmp(zn, nm) != 0)
238924e36522SCy Schubert 			verbose(VERB_ALGO, "rpz: qname trigger %s on %s with action=%s",
239024e36522SCy Schubert 				zn, nm, rpz_action_to_string(localzone_type_to_rpz_action(lzt)));
239124e36522SCy Schubert 		else
239224e36522SCy Schubert 			verbose(VERB_ALGO, "rpz: qname trigger %s with action=%s",
239324e36522SCy Schubert 				nm, rpz_action_to_string(localzone_type_to_rpz_action(lzt)));
239424e36522SCy Schubert 	}
239524e36522SCy Schubert 
239624e36522SCy Schubert 	ret = rpz_synthesize_qname_localdata(env, r, z, lzt, qinfo, edns, buf, temp,
239724e36522SCy Schubert 					     repinfo, stats);
239824e36522SCy Schubert 
2399091e9e46SCy Schubert 	lock_rw_unlock(&z->lock);
240025039b37SCy Schubert 	lock_rw_unlock(&a->lock);
2401091e9e46SCy Schubert 
2402091e9e46SCy Schubert 	return ret;
2403091e9e46SCy Schubert }
2404f44e67d1SCy Schubert 
2405f44e67d1SCy Schubert void rpz_enable(struct rpz* r)
2406f44e67d1SCy Schubert {
2407f44e67d1SCy Schubert     if(!r)
2408f44e67d1SCy Schubert         return;
2409f44e67d1SCy Schubert     r->disabled = 0;
2410f44e67d1SCy Schubert }
2411f44e67d1SCy Schubert 
2412f44e67d1SCy Schubert void rpz_disable(struct rpz* r)
2413f44e67d1SCy Schubert {
2414f44e67d1SCy Schubert     if(!r)
2415f44e67d1SCy Schubert         return;
2416f44e67d1SCy Schubert     r->disabled = 1;
2417f44e67d1SCy Schubert }
2418