iter_donotq.c revision 269257
1116742Ssam/* 2116904Ssam * iterator/iter_donotq.c - iterative resolver donotqueryaddresses storage. 3139530Ssam * 4116742Ssam * Copyright (c) 2007, NLnet Labs. All rights reserved. 5116742Ssam * 6116742Ssam * This software is open source. 7116742Ssam * 8116742Ssam * Redistribution and use in source and binary forms, with or without 9116742Ssam * modification, are permitted provided that the following conditions 10116742Ssam * are met: 11116742Ssam * 12116742Ssam * Redistributions of source code must retain the above copyright notice, 13116742Ssam * this list of conditions and the following disclaimer. 14116904Ssam * 15116904Ssam * Redistributions in binary form must reproduce the above copyright notice, 16116742Ssam * this list of conditions and the following disclaimer in the documentation 17116904Ssam * and/or other materials provided with the distribution. 18116904Ssam * 19116904Ssam * Neither the name of the NLNET LABS nor the names of its contributors may 20116742Ssam * be used to endorse or promote products derived from this software without 21116904Ssam * specific prior written permission. 22116904Ssam * 23116904Ssam * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 24116904Ssam * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 25116904Ssam * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 26116904Ssam * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 27116904Ssam * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 28116904Ssam * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 29116904Ssam * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 30116904Ssam * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 31116904Ssam * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 32116742Ssam * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 33116742Ssam * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 34116742Ssam */ 35116742Ssam 36116742Ssam/** 37116742Ssam * \file 38116742Ssam * 39116742Ssam * This file contains functions to assist the iterator module. 40116742Ssam * The donotqueryaddresses are stored and looked up. These addresses 41116742Ssam * (like 127.0.0.1) must not be used to send queries to, and can be 42117811Ssam * discarded immediately from the server selection. 43117811Ssam */ 44117811Ssam#include "config.h" 45117811Ssam#include "iterator/iter_donotq.h" 46117811Ssam#include "util/regional.h" 47116742Ssam#include "util/log.h" 48117811Ssam#include "util/config_file.h" 49116742Ssam#include "util/net_help.h" 50116742Ssam 51116742Ssamstruct iter_donotq* 52116742Ssamdonotq_create(void) 53116742Ssam{ 54139502Ssam struct iter_donotq* dq = (struct iter_donotq*)calloc(1, 55116742Ssam sizeof(struct iter_donotq)); 56138568Ssam if(!dq) 57138568Ssam return NULL; 58116742Ssam dq->region = regional_create(); 59119150Ssam if(!dq->region) { 60138568Ssam donotq_delete(dq); 61119150Ssam return NULL; 62119150Ssam } 63119150Ssam return dq; 64138568Ssam} 65138568Ssam 66116742Ssamvoid 67116742Ssamdonotq_delete(struct iter_donotq* dq) 68138568Ssam{ 69138568Ssam if(!dq) 70138568Ssam return; 71138568Ssam regional_destroy(dq->region); 72138568Ssam free(dq); 73138568Ssam} 74138568Ssam 75138568Ssam/** insert new address into donotq structure */ 76138568Ssamstatic int 77138568Ssamdonotq_insert(struct iter_donotq* dq, struct sockaddr_storage* addr, 78138568Ssam socklen_t addrlen, int net) 79138568Ssam{ 80138568Ssam struct addr_tree_node* node = (struct addr_tree_node*)regional_alloc( 81138568Ssam dq->region, sizeof(*node)); 82138568Ssam if(!node) 83138568Ssam return 0; 84138568Ssam if(!addr_tree_insert(&dq->tree, node, addr, addrlen, net)) { 85139525Ssam verbose(VERB_QUERY, "duplicate donotquery address ignored."); 86138568Ssam } 87138568Ssam return 1; 88138568Ssam} 89138568Ssam 90138568Ssam/** apply donotq string */ 91138568Ssamstatic int 92138568Ssamdonotq_str_cfg(struct iter_donotq* dq, const char* str) 93138568Ssam{ 94138568Ssam struct sockaddr_storage addr; 95138568Ssam int net; 96138568Ssam socklen_t addrlen; 97138568Ssam verbose(VERB_ALGO, "donotq: %s", str); 98138568Ssam if(!netblockstrtoaddr(str, UNBOUND_DNS_PORT, &addr, &addrlen, &net)) { 99138568Ssam log_err("cannot parse donotquery netblock: %s", str); 100138568Ssam return 0; 101138568Ssam } 102138568Ssam if(!donotq_insert(dq, &addr, addrlen, net)) { 103138568Ssam log_err("out of memory"); 104139525Ssam return 0; 105138568Ssam } 106138568Ssam return 1; 107138568Ssam} 108138568Ssam 109138568Ssam/** read donotq config */ 110138568Ssamstatic int 111138568Ssamread_donotq(struct iter_donotq* dq, struct config_file* cfg) 112138568Ssam{ 113138568Ssam struct config_strlist* p; 114138568Ssam for(p = cfg->donotqueryaddrs; p; p = p->next) { 115138568Ssam log_assert(p->str); 116138568Ssam if(!donotq_str_cfg(dq, p->str)) 117138568Ssam return 0; 118138568Ssam } 119138568Ssam return 1; 120138568Ssam} 121138568Ssam 122138568Ssamint 123138568Ssamdonotq_apply_cfg(struct iter_donotq* dq, struct config_file* cfg) 124138568Ssam{ 125138568Ssam regional_free_all(dq->region); 126138568Ssam addr_tree_init(&dq->tree); 127138568Ssam if(!read_donotq(dq, cfg)) 128138568Ssam return 0; 129138568Ssam if(cfg->donotquery_localhost) { 130138568Ssam if(!donotq_str_cfg(dq, "127.0.0.0/8")) 131138568Ssam return 0; 132138568Ssam if(cfg->do_ip6) { 133138568Ssam if(!donotq_str_cfg(dq, "::1")) 134138568Ssam return 0; 135138568Ssam } 136138568Ssam } 137138568Ssam addr_tree_init_parents(&dq->tree); 138138568Ssam return 1; 139138568Ssam} 140138568Ssam 141138568Ssamint 142138568Ssamdonotq_lookup(struct iter_donotq* donotq, struct sockaddr_storage* addr, 143138568Ssam socklen_t addrlen) 144138568Ssam{ 145138568Ssam return addr_tree_lookup(&donotq->tree, addr, addrlen) != NULL; 146138568Ssam} 147138568Ssam 148138568Ssamsize_t 149138568Ssamdonotq_get_mem(struct iter_donotq* donotq) 150138568Ssam{ 151138568Ssam if(!donotq) return 0; 152138568Ssam return sizeof(*donotq) + regional_get_mem(donotq->region); 153138568Ssam} 154138568Ssam