Home | History | Annotate | Line # | Download | only in iterator 
      1 /*
      2  * iterator/iter_donotq.c - iterative resolver donotqueryaddresses storage.
      3  *
      4  * Copyright (c) 2007, NLnet Labs. All rights reserved.
      5  *
      6  * This software is open source.
      7  *
      8  * Redistribution and use in source and binary forms, with or without
      9  * modification, are permitted provided that the following conditions
     10  * are met:
     11  *
     12  * Redistributions of source code must retain the above copyright notice,
     13  * this list of conditions and the following disclaimer.
     14  *
     15  * Redistributions in binary form must reproduce the above copyright notice,
     16  * this list of conditions and the following disclaimer in the documentation
     17  * and/or other materials provided with the distribution.
     18  *
     19  * Neither the name of the NLNET LABS nor the names of its contributors may
     20  * be used to endorse or promote products derived from this software without
     21  * specific prior written permission.
     22  *
     23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     24  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     25  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     26  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     27  * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     28  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
     29  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
     30  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
     31  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
     32  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
     33  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     34  */
     35 
     36 /**
     37  * \file
     38  *
     39  * This file contains functions to assist the iterator module.
     40  * The donotqueryaddresses are stored and looked up. These addresses
     41  * (like 127.0.0.1) must not be used to send queries to, and can be
     42  * discarded immediately from the server selection.
     43  */
     44 #include "config.h"
     45 #include "iterator/iter_donotq.h"
     46 #include "util/regional.h"
     47 #include "util/log.h"
     48 #include "util/config_file.h"
     49 #include "util/net_help.h"
     50 
     51 struct iter_donotq*
     52 donotq_create(void)
     53 {
     54 	struct iter_donotq* dq = (struct iter_donotq*)calloc(1,
     55 		sizeof(struct iter_donotq));
     56 	if(!dq)
     57 		return NULL;
     58 	dq->region = regional_create();
     59 	if(!dq->region) {
     60 		donotq_delete(dq);
     61 		return NULL;
     62 	}
     63 	return dq;
     64 }
     65 
     66 void
     67 donotq_delete(struct iter_donotq* dq)
     68 {
     69 	if(!dq)
     70 		return;
     71 	regional_destroy(dq->region);
     72 	free(dq);
     73 }
     74 
     75 /** insert new address into donotq structure */
     76 static int
     77 donotq_insert(struct iter_donotq* dq, struct sockaddr_storage* addr,
     78 	socklen_t addrlen, int net)
     79 {
     80 	struct addr_tree_node* node = (struct addr_tree_node*)regional_alloc(
     81 		dq->region, sizeof(*node));
     82 	if(!node)
     83 		return 0;
     84 	if(!addr_tree_insert(&dq->tree, node, addr, addrlen, net)) {
     85 		verbose(VERB_QUERY, "duplicate donotquery address ignored.");
     86 	}
     87 	return 1;
     88 }
     89 
     90 /** apply donotq string */
     91 static int
     92 donotq_str_cfg(struct iter_donotq* dq, const char* str)
     93 {
     94 	struct sockaddr_storage addr;
     95 	int net;
     96 	socklen_t addrlen;
     97 	verbose(VERB_ALGO, "donotq: %s", str);
     98 	if(!netblockstrtoaddr(str, UNBOUND_DNS_PORT, &addr, &addrlen, &net)) {
     99 		log_err("cannot parse donotquery netblock: %s", str);
    100 		return 0;
    101 	}
    102 	if(!donotq_insert(dq, &addr, addrlen, net)) {
    103 		log_err("out of memory");
    104 		return 0;
    105 	}
    106 	return 1;
    107 }
    108 
    109 /** read donotq config */
    110 static int
    111 read_donotq(struct iter_donotq* dq, struct config_file* cfg)
    112 {
    113 	struct config_strlist* p;
    114 	for(p = cfg->donotqueryaddrs; p; p = p->next) {
    115 		log_assert(p->str);
    116 		if(!donotq_str_cfg(dq, p->str))
    117 			return 0;
    118 	}
    119 	return 1;
    120 }
    121 
    122 int
    123 donotq_apply_cfg(struct iter_donotq* dq, struct config_file* cfg)
    124 {
    125 	regional_free_all(dq->region);
    126 	addr_tree_init(&dq->tree);
    127 	if(!read_donotq(dq, cfg))
    128 		return 0;
    129 	if(cfg->donotquery_localhost) {
    130 		if(!donotq_str_cfg(dq, "127.0.0.0/8"))
    131 			return 0;
    132 		if(cfg->do_ip6) {
    133 			if(!donotq_str_cfg(dq, "::1"))
    134 				return 0;
    135 		}
    136 	}
    137 	addr_tree_init_parents(&dq->tree);
    138 	return 1;
    139 }
    140 
    141 int
    142 donotq_lookup(struct iter_donotq* donotq, struct sockaddr_storage* addr,
    143         socklen_t addrlen)
    144 {
    145 	return addr_tree_lookup(&donotq->tree, addr, addrlen) != NULL;
    146 }
    147 
    148 size_t
    149 donotq_get_mem(struct iter_donotq* donotq)
    150 {
    151 	if(!donotq) return 0;
    152 	return sizeof(*donotq) + regional_get_mem(donotq->region);
    153 }
    154