[apple/network_cmds.git] / unbound / iterator / iter_donotq.c

/*
 * iterator/iter_donotq.c - iterative resolver donotqueryaddresses storage.
 *
 * Copyright (c) 2007, NLnet Labs. All rights reserved.
 *
 * This software is open source.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * 
 * Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 * 
 * Neither the name of the NLNET LABS nor the names of its contributors may
 * be used to endorse or promote products derived from this software without
 * specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/**
 * \file
 *
 * This file contains functions to assist the iterator module.
 * The donotqueryaddresses are stored and looked up. These addresses
 * (like 127.0.0.1) must not be used to send queries to, and can be
 * discarded immediately from the server selection.
 */
#include "config.h"
#include "iterator/iter_donotq.h"
#include "util/regional.h"
#include "util/log.h"
#include "util/config_file.h"
#include "util/net_help.h"

struct iter_donotq* 
donotq_create(void)
{
	struct iter_donotq* dq = (struct iter_donotq*)calloc(1,
		sizeof(struct iter_donotq));
	if(!dq)
		return NULL;
	dq->region = regional_create();
	if(!dq->region) {
		donotq_delete(dq);
		return NULL;
	}
	return dq;
}

void 
donotq_delete(struct iter_donotq* dq)
{
	if(!dq) 
		return;
	regional_destroy(dq->region);
	free(dq);
}

/** insert new address into donotq structure */
static int
donotq_insert(struct iter_donotq* dq, struct sockaddr_storage* addr, 
	socklen_t addrlen, int net)
{
	struct addr_tree_node* node = (struct addr_tree_node*)regional_alloc(
		dq->region, sizeof(*node));
	if(!node)
		return 0;
	if(!addr_tree_insert(&dq->tree, node, addr, addrlen, net)) {
		verbose(VERB_QUERY, "duplicate donotquery address ignored.");
	}
	return 1;
}

/** apply donotq string */
static int
donotq_str_cfg(struct iter_donotq* dq, const char* str)
{
	struct sockaddr_storage addr;
	int net;
	socklen_t addrlen;
	verbose(VERB_ALGO, "donotq: %s", str);
	if(!netblockstrtoaddr(str, UNBOUND_DNS_PORT, &addr, &addrlen, &net)) {
		log_err("cannot parse donotquery netblock: %s", str);
		return 0;
	}
	if(!donotq_insert(dq, &addr, addrlen, net)) {
		log_err("out of memory");
		return 0;
	}
	return 1;
}

/** read donotq config */
static int 
read_donotq(struct iter_donotq* dq, struct config_file* cfg)
{
	struct config_strlist* p;
	for(p = cfg->donotqueryaddrs; p; p = p->next) {
		log_assert(p->str);
		if(!donotq_str_cfg(dq, p->str))
			return 0;
	}
	return 1;
}

int 
donotq_apply_cfg(struct iter_donotq* dq, struct config_file* cfg)
{
	regional_free_all(dq->region);
	addr_tree_init(&dq->tree);
	if(!read_donotq(dq, cfg))
		return 0;
	if(cfg->donotquery_localhost) {
		if(!donotq_str_cfg(dq, "127.0.0.0/8"))
			return 0;
		if(cfg->do_ip6) {
			if(!donotq_str_cfg(dq, "::1"))
				return 0;
		}
	}
	addr_tree_init_parents(&dq->tree);
	return 1;
}

int 
donotq_lookup(struct iter_donotq* donotq, struct sockaddr_storage* addr,
        socklen_t addrlen)
{
	return addr_tree_lookup(&donotq->tree, addr, addrlen) != NULL;
}

size_t 
donotq_get_mem(struct iter_donotq* donotq)
{
	if(!donotq) return 0;
	return sizeof(*donotq) + regional_get_mem(donotq->region);
}
Commit	Line	Data
89c4ed63 A	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	}