1 /*
2 * util/edns.c - handle base EDNS options.
3 *
4 * Copyright (c) 2018, 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 for base EDNS options.
40 */
41
42 #include "config.h"
43 #include "util/edns.h"
44 #include "util/config_file.h"
45 #include "util/netevent.h"
46 #include "util/net_help.h"
47 #include "util/regional.h"
48 #include "util/data/msgparse.h"
49 #include "util/data/msgreply.h"
50
51 #if 0
52 /* XXX: remove me */
53 #include "edns.h"
54 #endif
55
edns_strings_create(void)56 struct edns_strings* edns_strings_create(void)
57 {
58 struct edns_strings* edns_strings = calloc(1,
59 sizeof(struct edns_strings));
60 if(!edns_strings)
61 return NULL;
62 if(!(edns_strings->region = regional_create())) {
63 edns_strings_delete(edns_strings);
64 return NULL;
65 }
66 return edns_strings;
67 }
68
edns_strings_delete(struct edns_strings * edns_strings)69 void edns_strings_delete(struct edns_strings* edns_strings)
70 {
71 if(!edns_strings)
72 return;
73 regional_destroy(edns_strings->region);
74 free(edns_strings);
75 }
76
77 static int
edns_strings_client_insert(struct edns_strings * edns_strings,struct sockaddr_storage * addr,socklen_t addrlen,int net,const char * string)78 edns_strings_client_insert(struct edns_strings* edns_strings,
79 struct sockaddr_storage* addr, socklen_t addrlen, int net,
80 const char* string)
81 {
82 struct edns_string_addr* esa = regional_alloc_zero(edns_strings->region,
83 sizeof(struct edns_string_addr));
84 if(!esa)
85 return 0;
86 esa->string_len = strlen(string);
87 esa->string = regional_alloc_init(edns_strings->region, string,
88 esa->string_len);
89 if(!esa->string)
90 return 0;
91 if(!addr_tree_insert(&edns_strings->client_strings, &esa->node, addr,
92 addrlen, net)) {
93 verbose(VERB_QUERY, "duplicate EDNS client string ignored.");
94 }
95 return 1;
96 }
97
edns_strings_apply_cfg(struct edns_strings * edns_strings,struct config_file * config)98 int edns_strings_apply_cfg(struct edns_strings* edns_strings,
99 struct config_file* config)
100 {
101 struct config_str2list* c;
102 regional_free_all(edns_strings->region);
103 addr_tree_init(&edns_strings->client_strings);
104
105 for(c=config->edns_client_strings; c; c=c->next) {
106 struct sockaddr_storage addr;
107 socklen_t addrlen;
108 int net;
109 log_assert(c->str && c->str2);
110
111 if(!netblockstrtoaddr(c->str, UNBOUND_DNS_PORT, &addr, &addrlen,
112 &net)) {
113 log_err("cannot parse EDNS client string IP netblock: "
114 "%s", c->str);
115 return 0;
116 }
117 if(!edns_strings_client_insert(edns_strings, &addr, addrlen,
118 net, c->str2)) {
119 log_err("out of memory while adding EDNS strings");
120 return 0;
121 }
122 }
123 edns_strings->client_string_opcode = config->edns_client_string_opcode;
124
125 addr_tree_init_parents(&edns_strings->client_strings);
126 return 1;
127 }
128
129 struct edns_string_addr*
edns_string_addr_lookup(rbtree_type * tree,struct sockaddr_storage * addr,socklen_t addrlen)130 edns_string_addr_lookup(rbtree_type* tree, struct sockaddr_storage* addr,
131 socklen_t addrlen)
132 {
133 return (struct edns_string_addr*)addr_tree_lookup(tree, addr, addrlen);
134 }
135
136