1238106Sdes/* 2238106Sdes * iterator/iter_donotq.c - iterative resolver donotqueryaddresses storage. 3238106Sdes * 4238106Sdes * Copyright (c) 2007, 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 24269257Sdes * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 25269257Sdes * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 26269257Sdes * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 27269257Sdes * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 28269257Sdes * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 29269257Sdes * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 30269257Sdes * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 31269257Sdes * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 32269257Sdes * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 33269257Sdes * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 34238106Sdes */ 35238106Sdes 36238106Sdes/** 37238106Sdes * \file 38238106Sdes * 39238106Sdes * This file contains functions to assist the iterator module. 40238106Sdes * The donotqueryaddresses are stored and looked up. These addresses 41238106Sdes * (like 127.0.0.1) must not be used to send queries to, and can be 42238106Sdes * discarded immediately from the server selection. 43238106Sdes */ 44238106Sdes#include "config.h" 45238106Sdes#include "iterator/iter_donotq.h" 46238106Sdes#include "util/regional.h" 47238106Sdes#include "util/log.h" 48238106Sdes#include "util/config_file.h" 49238106Sdes#include "util/net_help.h" 50238106Sdes 51238106Sdesstruct iter_donotq* 52238106Sdesdonotq_create(void) 53238106Sdes{ 54238106Sdes struct iter_donotq* dq = (struct iter_donotq*)calloc(1, 55238106Sdes sizeof(struct iter_donotq)); 56238106Sdes if(!dq) 57238106Sdes return NULL; 58238106Sdes dq->region = regional_create(); 59238106Sdes if(!dq->region) { 60238106Sdes donotq_delete(dq); 61238106Sdes return NULL; 62238106Sdes } 63238106Sdes return dq; 64238106Sdes} 65238106Sdes 66238106Sdesvoid 67238106Sdesdonotq_delete(struct iter_donotq* dq) 68238106Sdes{ 69238106Sdes if(!dq) 70238106Sdes return; 71238106Sdes regional_destroy(dq->region); 72238106Sdes free(dq); 73238106Sdes} 74238106Sdes 75238106Sdes/** insert new address into donotq structure */ 76238106Sdesstatic int 77238106Sdesdonotq_insert(struct iter_donotq* dq, struct sockaddr_storage* addr, 78238106Sdes socklen_t addrlen, int net) 79238106Sdes{ 80238106Sdes struct addr_tree_node* node = (struct addr_tree_node*)regional_alloc( 81238106Sdes dq->region, sizeof(*node)); 82238106Sdes if(!node) 83238106Sdes return 0; 84238106Sdes if(!addr_tree_insert(&dq->tree, node, addr, addrlen, net)) { 85238106Sdes verbose(VERB_QUERY, "duplicate donotquery address ignored."); 86238106Sdes } 87238106Sdes return 1; 88238106Sdes} 89238106Sdes 90238106Sdes/** apply donotq string */ 91238106Sdesstatic int 92238106Sdesdonotq_str_cfg(struct iter_donotq* dq, const char* str) 93238106Sdes{ 94238106Sdes struct sockaddr_storage addr; 95238106Sdes int net; 96238106Sdes socklen_t addrlen; 97238106Sdes verbose(VERB_ALGO, "donotq: %s", str); 98238106Sdes if(!netblockstrtoaddr(str, UNBOUND_DNS_PORT, &addr, &addrlen, &net)) { 99238106Sdes log_err("cannot parse donotquery netblock: %s", str); 100238106Sdes return 0; 101238106Sdes } 102238106Sdes if(!donotq_insert(dq, &addr, addrlen, net)) { 103238106Sdes log_err("out of memory"); 104238106Sdes return 0; 105238106Sdes } 106238106Sdes return 1; 107238106Sdes} 108238106Sdes 109238106Sdes/** read donotq config */ 110238106Sdesstatic int 111238106Sdesread_donotq(struct iter_donotq* dq, struct config_file* cfg) 112238106Sdes{ 113238106Sdes struct config_strlist* p; 114238106Sdes for(p = cfg->donotqueryaddrs; p; p = p->next) { 115238106Sdes log_assert(p->str); 116238106Sdes if(!donotq_str_cfg(dq, p->str)) 117238106Sdes return 0; 118238106Sdes } 119238106Sdes return 1; 120238106Sdes} 121238106Sdes 122238106Sdesint 123238106Sdesdonotq_apply_cfg(struct iter_donotq* dq, struct config_file* cfg) 124238106Sdes{ 125238106Sdes regional_free_all(dq->region); 126238106Sdes addr_tree_init(&dq->tree); 127238106Sdes if(!read_donotq(dq, cfg)) 128238106Sdes return 0; 129238106Sdes if(cfg->donotquery_localhost) { 130238106Sdes if(!donotq_str_cfg(dq, "127.0.0.0/8")) 131238106Sdes return 0; 132238106Sdes if(cfg->do_ip6) { 133238106Sdes if(!donotq_str_cfg(dq, "::1")) 134238106Sdes return 0; 135238106Sdes } 136238106Sdes } 137238106Sdes addr_tree_init_parents(&dq->tree); 138238106Sdes return 1; 139238106Sdes} 140238106Sdes 141238106Sdesint 142238106Sdesdonotq_lookup(struct iter_donotq* donotq, struct sockaddr_storage* addr, 143238106Sdes socklen_t addrlen) 144238106Sdes{ 145238106Sdes return addr_tree_lookup(&donotq->tree, addr, addrlen) != NULL; 146238106Sdes} 147238106Sdes 148238106Sdessize_t 149238106Sdesdonotq_get_mem(struct iter_donotq* donotq) 150238106Sdes{ 151238106Sdes if(!donotq) return 0; 152238106Sdes return sizeof(*donotq) + regional_get_mem(donotq->region); 153238106Sdes} 154