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