1238106Sdes/* 2238106Sdes * iterator/iter_priv.c - iterative resolver private address and domain store 3238106Sdes * 4238106Sdes * Copyright (c) 2008, NLnet Labs. All rights reserved. 5238106Sdes * 6238106Sdes * This software is open source. 7238106Sdes * 8238106Sdes * Redistribution and use in source and binary forms, with or without 9238106Sdes * modification, are permitted provided that the following conditions 10238106Sdes * are met: 11238106Sdes * 12238106Sdes * Redistributions of source code must retain the above copyright notice, 13238106Sdes * this list of conditions and the following disclaimer. 14238106Sdes * 15238106Sdes * Redistributions in binary form must reproduce the above copyright notice, 16238106Sdes * this list of conditions and the following disclaimer in the documentation 17238106Sdes * and/or other materials provided with the distribution. 18238106Sdes * 19238106Sdes * Neither the name of the NLNET LABS nor the names of its contributors may 20238106Sdes * be used to endorse or promote products derived from this software without 21238106Sdes * specific prior written permission. 22238106Sdes * 23238106Sdes * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 24238106Sdes * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 25238106Sdes * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 26238106Sdes * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE 27238106Sdes * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 28238106Sdes * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 29238106Sdes * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 30238106Sdes * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 31238106Sdes * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 32238106Sdes * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 33238106Sdes * POSSIBILITY OF SUCH DAMAGE. 34238106Sdes */ 35238106Sdes 36238106Sdes/** 37238106Sdes * \file 38238106Sdes * 39238106Sdes * This file contains functions to assist the iterator module. 40238106Sdes * Keep track of the private addresses and lookup fast. 41238106Sdes */ 42238106Sdes 43238106Sdes#include "config.h" 44238106Sdes#include <ldns/dname.h> 45238106Sdes#include "iterator/iter_priv.h" 46238106Sdes#include "util/regional.h" 47238106Sdes#include "util/log.h" 48238106Sdes#include "util/config_file.h" 49238106Sdes#include "util/data/dname.h" 50238106Sdes#include "util/data/msgparse.h" 51238106Sdes#include "util/net_help.h" 52238106Sdes#include "util/storage/dnstree.h" 53238106Sdes 54238106Sdesstruct iter_priv* priv_create(void) 55238106Sdes{ 56238106Sdes struct iter_priv* priv = (struct iter_priv*)calloc(1, sizeof(*priv)); 57238106Sdes if(!priv) 58238106Sdes return NULL; 59238106Sdes priv->region = regional_create(); 60238106Sdes if(!priv->region) { 61238106Sdes priv_delete(priv); 62238106Sdes return NULL; 63238106Sdes } 64238106Sdes addr_tree_init(&priv->a); 65238106Sdes name_tree_init(&priv->n); 66238106Sdes return priv; 67238106Sdes} 68238106Sdes 69238106Sdesvoid priv_delete(struct iter_priv* priv) 70238106Sdes{ 71238106Sdes if(!priv) return; 72238106Sdes regional_destroy(priv->region); 73238106Sdes free(priv); 74238106Sdes} 75238106Sdes 76238106Sdes/** Read private-addr declarations from config */ 77238106Sdesstatic int read_addrs(struct iter_priv* priv, struct config_file* cfg) 78238106Sdes{ 79238106Sdes /* parse addresses, report errors, insert into tree */ 80238106Sdes struct config_strlist* p; 81238106Sdes struct addr_tree_node* n; 82238106Sdes struct sockaddr_storage addr; 83238106Sdes int net; 84238106Sdes socklen_t addrlen; 85238106Sdes 86238106Sdes for(p = cfg->private_address; p; p = p->next) { 87238106Sdes log_assert(p->str); 88238106Sdes if(!netblockstrtoaddr(p->str, UNBOUND_DNS_PORT, &addr, 89238106Sdes &addrlen, &net)) { 90238106Sdes log_err("cannot parse private-address: %s", p->str); 91238106Sdes return 0; 92238106Sdes } 93238106Sdes n = (struct addr_tree_node*)regional_alloc(priv->region, 94238106Sdes sizeof(*n)); 95238106Sdes if(!n) { 96238106Sdes log_err("out of memory"); 97238106Sdes return 0; 98238106Sdes } 99238106Sdes if(!addr_tree_insert(&priv->a, n, &addr, addrlen, net)) { 100238106Sdes verbose(VERB_QUERY, "ignoring duplicate " 101238106Sdes "private-address: %s", p->str); 102238106Sdes } 103238106Sdes } 104238106Sdes return 1; 105238106Sdes} 106238106Sdes 107238106Sdes/** Read private-domain declarations from config */ 108238106Sdesstatic int read_names(struct iter_priv* priv, struct config_file* cfg) 109238106Sdes{ 110238106Sdes /* parse names, report errors, insert into tree */ 111238106Sdes struct config_strlist* p; 112238106Sdes struct name_tree_node* n; 113238106Sdes uint8_t* nm; 114238106Sdes size_t nm_len; 115238106Sdes int nm_labs; 116238106Sdes ldns_rdf* rdf; 117238106Sdes 118238106Sdes for(p = cfg->private_domain; p; p = p->next) { 119238106Sdes log_assert(p->str); 120238106Sdes rdf = ldns_dname_new_frm_str(p->str); 121238106Sdes if(!rdf) { 122238106Sdes log_err("cannot parse private-domain: %s", p->str); 123238106Sdes return 0; 124238106Sdes } 125238106Sdes nm = ldns_rdf_data(rdf); 126238106Sdes nm_labs = dname_count_size_labels(nm, &nm_len); 127238106Sdes nm = (uint8_t*)regional_alloc_init(priv->region, nm, nm_len); 128238106Sdes ldns_rdf_deep_free(rdf); 129238106Sdes if(!nm) { 130238106Sdes log_err("out of memory"); 131238106Sdes return 0; 132238106Sdes } 133238106Sdes n = (struct name_tree_node*)regional_alloc(priv->region, 134238106Sdes sizeof(*n)); 135238106Sdes if(!n) { 136238106Sdes log_err("out of memory"); 137238106Sdes return 0; 138238106Sdes } 139238106Sdes if(!name_tree_insert(&priv->n, n, nm, nm_len, nm_labs, 140238106Sdes LDNS_RR_CLASS_IN)) { 141238106Sdes verbose(VERB_QUERY, "ignoring duplicate " 142238106Sdes "private-domain: %s", p->str); 143238106Sdes } 144238106Sdes } 145238106Sdes return 1; 146238106Sdes} 147238106Sdes 148238106Sdesint priv_apply_cfg(struct iter_priv* priv, struct config_file* cfg) 149238106Sdes{ 150238106Sdes /* empty the current contents */ 151238106Sdes regional_free_all(priv->region); 152238106Sdes addr_tree_init(&priv->a); 153238106Sdes name_tree_init(&priv->n); 154238106Sdes 155238106Sdes /* read new contents */ 156238106Sdes if(!read_addrs(priv, cfg)) 157238106Sdes return 0; 158238106Sdes if(!read_names(priv, cfg)) 159238106Sdes return 0; 160238106Sdes 161238106Sdes /* prepare for lookups */ 162238106Sdes addr_tree_init_parents(&priv->a); 163238106Sdes name_tree_init_parents(&priv->n); 164238106Sdes return 1; 165238106Sdes} 166238106Sdes 167238106Sdes/** 168238106Sdes * See if an address is blocked. 169238106Sdes * @param priv: structure for address storage. 170238106Sdes * @param addr: address to check 171238106Sdes * @param addrlen: length of addr. 172238106Sdes * @return: true if the address must not be queried. false if unlisted. 173238106Sdes */ 174238106Sdesstatic int 175238106Sdespriv_lookup_addr(struct iter_priv* priv, struct sockaddr_storage* addr, 176238106Sdes socklen_t addrlen) 177238106Sdes{ 178238106Sdes return addr_tree_lookup(&priv->a, addr, addrlen) != NULL; 179238106Sdes} 180238106Sdes 181238106Sdes/** 182238106Sdes * See if a name is whitelisted. 183238106Sdes * @param priv: structure for address storage. 184238106Sdes * @param pkt: the packet (for compression ptrs). 185238106Sdes * @param name: name to check. 186238106Sdes * @param name_len: uncompressed length of the name to check. 187238106Sdes * @param dclass: class to check. 188238106Sdes * @return: true if the name is OK. false if unlisted. 189238106Sdes */ 190238106Sdesstatic int 191238106Sdespriv_lookup_name(struct iter_priv* priv, ldns_buffer* pkt, 192238106Sdes uint8_t* name, size_t name_len, uint16_t dclass) 193238106Sdes{ 194238106Sdes size_t len; 195238106Sdes uint8_t decomp[256]; 196238106Sdes int labs; 197238106Sdes if(name_len >= sizeof(decomp)) 198238106Sdes return 0; 199238106Sdes dname_pkt_copy(pkt, decomp, name); 200238106Sdes labs = dname_count_size_labels(decomp, &len); 201238106Sdes log_assert(name_len == len); 202238106Sdes return name_tree_lookup(&priv->n, decomp, len, labs, dclass) != NULL; 203238106Sdes} 204238106Sdes 205238106Sdessize_t priv_get_mem(struct iter_priv* priv) 206238106Sdes{ 207238106Sdes if(!priv) return 0; 208238106Sdes return sizeof(*priv) + regional_get_mem(priv->region); 209238106Sdes} 210238106Sdes 211238106Sdesint priv_rrset_bad(struct iter_priv* priv, ldns_buffer* pkt, 212238106Sdes struct rrset_parse* rrset) 213238106Sdes{ 214238106Sdes if(priv->a.count == 0) 215238106Sdes return 0; /* there are no blocked addresses */ 216238106Sdes 217238106Sdes /* see if it is a private name, that is allowed to have any */ 218238106Sdes if(priv_lookup_name(priv, pkt, rrset->dname, rrset->dname_len, 219238106Sdes ntohs(rrset->rrset_class))) { 220238106Sdes return 0; 221238106Sdes } else { 222238106Sdes /* so its a public name, check the address */ 223238106Sdes socklen_t len; 224238106Sdes struct rr_parse* rr; 225238106Sdes if(rrset->type == LDNS_RR_TYPE_A) { 226238106Sdes struct sockaddr_storage addr; 227238106Sdes struct sockaddr_in sa; 228238106Sdes 229238106Sdes len = (socklen_t)sizeof(sa); 230238106Sdes memset(&sa, 0, len); 231238106Sdes sa.sin_family = AF_INET; 232238106Sdes sa.sin_port = (in_port_t)htons(UNBOUND_DNS_PORT); 233238106Sdes for(rr = rrset->rr_first; rr; rr = rr->next) { 234238106Sdes if(ldns_read_uint16(rr->ttl_data+4) 235238106Sdes != INET_SIZE) 236238106Sdes continue; 237238106Sdes memmove(&sa.sin_addr, rr->ttl_data+4+2, 238238106Sdes INET_SIZE); 239238106Sdes memmove(&addr, &sa, len); 240238106Sdes if(priv_lookup_addr(priv, &addr, len)) 241238106Sdes return 1; 242238106Sdes } 243238106Sdes } else if(rrset->type == LDNS_RR_TYPE_AAAA) { 244238106Sdes struct sockaddr_storage addr; 245238106Sdes struct sockaddr_in6 sa; 246238106Sdes len = (socklen_t)sizeof(sa); 247238106Sdes memset(&sa, 0, len); 248238106Sdes sa.sin6_family = AF_INET6; 249238106Sdes sa.sin6_port = (in_port_t)htons(UNBOUND_DNS_PORT); 250238106Sdes for(rr = rrset->rr_first; rr; rr = rr->next) { 251238106Sdes if(ldns_read_uint16(rr->ttl_data+4) 252238106Sdes != INET6_SIZE) 253238106Sdes continue; 254238106Sdes memmove(&sa.sin6_addr, rr->ttl_data+4+2, 255238106Sdes INET6_SIZE); 256238106Sdes memmove(&addr, &sa, len); 257238106Sdes if(priv_lookup_addr(priv, &addr, len)) 258238106Sdes return 1; 259238106Sdes } 260238106Sdes } 261238106Sdes } 262238106Sdes return 0; 263238106Sdes} 264