msgreply.c revision 1.1.1.7.2.1 1 1.1 christos /*
2 1.1 christos * util/data/msgreply.c - store message and reply data.
3 1.1 christos *
4 1.1 christos * Copyright (c) 2007, NLnet Labs. All rights reserved.
5 1.1 christos *
6 1.1 christos * This software is open source.
7 1.1 christos *
8 1.1 christos * Redistribution and use in source and binary forms, with or without
9 1.1 christos * modification, are permitted provided that the following conditions
10 1.1 christos * are met:
11 1.1 christos *
12 1.1 christos * Redistributions of source code must retain the above copyright notice,
13 1.1 christos * this list of conditions and the following disclaimer.
14 1.1 christos *
15 1.1 christos * Redistributions in binary form must reproduce the above copyright notice,
16 1.1 christos * this list of conditions and the following disclaimer in the documentation
17 1.1 christos * and/or other materials provided with the distribution.
18 1.1 christos *
19 1.1 christos * Neither the name of the NLNET LABS nor the names of its contributors may
20 1.1 christos * be used to endorse or promote products derived from this software without
21 1.1 christos * specific prior written permission.
22 1.1 christos *
23 1.1 christos * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 1.1 christos * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 1.1 christos * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 1.1 christos * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 1.1 christos * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 1.1 christos * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
29 1.1 christos * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
30 1.1 christos * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
31 1.1 christos * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
32 1.1 christos * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
33 1.1 christos * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 1.1 christos */
35 1.1 christos
36 1.1 christos /**
37 1.1 christos * \file
38 1.1 christos *
39 1.1 christos * This file contains a data structure to store a message and its reply.
40 1.1 christos */
41 1.1 christos
42 1.1 christos #include "config.h"
43 1.1 christos #include "util/data/msgreply.h"
44 1.1 christos #include "util/storage/lookup3.h"
45 1.1 christos #include "util/log.h"
46 1.1 christos #include "util/alloc.h"
47 1.1 christos #include "util/netevent.h"
48 1.1 christos #include "util/net_help.h"
49 1.1 christos #include "util/data/dname.h"
50 1.1 christos #include "util/regional.h"
51 1.1 christos #include "util/data/msgparse.h"
52 1.1 christos #include "util/data/msgencode.h"
53 1.1 christos #include "sldns/sbuffer.h"
54 1.1 christos #include "sldns/wire2str.h"
55 1.1.1.2 christos #include "util/module.h"
56 1.1.1.2 christos #include "util/fptr_wlist.h"
57 1.1 christos
58 1.1 christos /** MAX TTL default for messages and rrsets */
59 1.1 christos time_t MAX_TTL = 3600 * 24 * 10; /* ten days */
60 1.1 christos /** MIN TTL default for messages and rrsets */
61 1.1 christos time_t MIN_TTL = 0;
62 1.1 christos /** MAX Negative TTL, for SOA records in authority section */
63 1.1 christos time_t MAX_NEG_TTL = 3600; /* one hour */
64 1.1.1.6 christos /** If we serve expired entries and prefetch them */
65 1.1.1.6 christos int SERVE_EXPIRED = 0;
66 1.1.1.4 christos /** Time to serve records after expiration */
67 1.1.1.4 christos time_t SERVE_EXPIRED_TTL = 0;
68 1.1.1.6 christos /** TTL to use for expired records */
69 1.1.1.6 christos time_t SERVE_EXPIRED_REPLY_TTL = 30;
70 1.1.1.6 christos /** If we serve the original TTL or decrementing TTLs */
71 1.1.1.6 christos int SERVE_ORIGINAL_TTL = 0;
72 1.1 christos
73 1.1 christos /** allocate qinfo, return 0 on error */
74 1.1 christos static int
75 1.1 christos parse_create_qinfo(sldns_buffer* pkt, struct msg_parse* msg,
76 1.1 christos struct query_info* qinf, struct regional* region)
77 1.1 christos {
78 1.1 christos if(msg->qname) {
79 1.1 christos if(region)
80 1.1 christos qinf->qname = (uint8_t*)regional_alloc(region,
81 1.1 christos msg->qname_len);
82 1.1 christos else qinf->qname = (uint8_t*)malloc(msg->qname_len);
83 1.1 christos if(!qinf->qname) return 0;
84 1.1 christos dname_pkt_copy(pkt, qinf->qname, msg->qname);
85 1.1 christos } else qinf->qname = 0;
86 1.1 christos qinf->qname_len = msg->qname_len;
87 1.1 christos qinf->qtype = msg->qtype;
88 1.1 christos qinf->qclass = msg->qclass;
89 1.1.1.2 christos qinf->local_alias = NULL;
90 1.1 christos return 1;
91 1.1 christos }
92 1.1 christos
93 1.1 christos /** constructor for replyinfo */
94 1.1 christos struct reply_info*
95 1.1 christos construct_reply_info_base(struct regional* region, uint16_t flags, size_t qd,
96 1.1.1.4 christos time_t ttl, time_t prettl, time_t expttl, size_t an, size_t ns,
97 1.1.1.7.2.1 martin size_t ar, size_t total, enum sec_status sec, sldns_ede_code reason_bogus)
98 1.1 christos {
99 1.1 christos struct reply_info* rep;
100 1.1 christos /* rrset_count-1 because the first ref is part of the struct. */
101 1.1 christos size_t s = sizeof(struct reply_info) - sizeof(struct rrset_ref) +
102 1.1 christos sizeof(struct ub_packed_rrset_key*) * total;
103 1.1 christos if(total >= RR_COUNT_MAX) return NULL; /* sanity check on numRRS*/
104 1.1 christos if(region)
105 1.1 christos rep = (struct reply_info*)regional_alloc(region, s);
106 1.1 christos else rep = (struct reply_info*)malloc(s +
107 1.1 christos sizeof(struct rrset_ref) * (total));
108 1.1 christos if(!rep)
109 1.1 christos return NULL;
110 1.1 christos rep->flags = flags;
111 1.1 christos rep->qdcount = qd;
112 1.1 christos rep->ttl = ttl;
113 1.1 christos rep->prefetch_ttl = prettl;
114 1.1.1.4 christos rep->serve_expired_ttl = expttl;
115 1.1 christos rep->an_numrrsets = an;
116 1.1 christos rep->ns_numrrsets = ns;
117 1.1 christos rep->ar_numrrsets = ar;
118 1.1 christos rep->rrset_count = total;
119 1.1 christos rep->security = sec;
120 1.1.1.7.2.1 martin rep->reason_bogus = reason_bogus;
121 1.1.1.7.2.1 martin /* this is only allocated and used for caching on copy */
122 1.1.1.7.2.1 martin rep->reason_bogus_str = NULL;
123 1.1 christos rep->authoritative = 0;
124 1.1 christos /* array starts after the refs */
125 1.1 christos if(region)
126 1.1 christos rep->rrsets = (struct ub_packed_rrset_key**)&(rep->ref[0]);
127 1.1 christos else rep->rrsets = (struct ub_packed_rrset_key**)&(rep->ref[total]);
128 1.1 christos /* zero the arrays to assist cleanup in case of malloc failure */
129 1.1 christos memset( rep->rrsets, 0, sizeof(struct ub_packed_rrset_key*) * total);
130 1.1 christos if(!region)
131 1.1 christos memset( &rep->ref[0], 0, sizeof(struct rrset_ref) * total);
132 1.1 christos return rep;
133 1.1 christos }
134 1.1 christos
135 1.1 christos /** allocate replyinfo, return 0 on error */
136 1.1 christos static int
137 1.1 christos parse_create_repinfo(struct msg_parse* msg, struct reply_info** rep,
138 1.1 christos struct regional* region)
139 1.1 christos {
140 1.1 christos *rep = construct_reply_info_base(region, msg->flags, msg->qdcount, 0,
141 1.1.1.4 christos 0, 0, msg->an_rrsets, msg->ns_rrsets, msg->ar_rrsets,
142 1.1.1.7.2.1 martin msg->rrset_count, sec_status_unchecked, LDNS_EDE_NONE);
143 1.1 christos if(!*rep)
144 1.1 christos return 0;
145 1.1 christos return 1;
146 1.1 christos }
147 1.1 christos
148 1.1.1.2 christos int
149 1.1.1.2 christos reply_info_alloc_rrset_keys(struct reply_info* rep, struct alloc_cache* alloc,
150 1.1 christos struct regional* region)
151 1.1 christos {
152 1.1 christos size_t i;
153 1.1 christos for(i=0; i<rep->rrset_count; i++) {
154 1.1 christos if(region) {
155 1.1 christos rep->rrsets[i] = (struct ub_packed_rrset_key*)
156 1.1 christos regional_alloc(region,
157 1.1 christos sizeof(struct ub_packed_rrset_key));
158 1.1 christos if(rep->rrsets[i]) {
159 1.1 christos memset(rep->rrsets[i], 0,
160 1.1 christos sizeof(struct ub_packed_rrset_key));
161 1.1 christos rep->rrsets[i]->entry.key = rep->rrsets[i];
162 1.1 christos }
163 1.1 christos }
164 1.1 christos else rep->rrsets[i] = alloc_special_obtain(alloc);
165 1.1 christos if(!rep->rrsets[i])
166 1.1 christos return 0;
167 1.1 christos rep->rrsets[i]->entry.data = NULL;
168 1.1 christos }
169 1.1 christos return 1;
170 1.1 christos }
171 1.1 christos
172 1.1.1.7 christos struct reply_info *
173 1.1.1.7 christos make_new_reply_info(const struct reply_info* rep, struct regional* region,
174 1.1.1.7 christos size_t an_numrrsets, size_t copy_rrsets)
175 1.1.1.7 christos {
176 1.1.1.7 christos struct reply_info* new_rep;
177 1.1.1.7 christos size_t i;
178 1.1.1.7 christos
179 1.1.1.7 christos /* create a base struct. we specify 'insecure' security status as
180 1.1.1.7 christos * the modified response won't be DNSSEC-valid. In our faked response
181 1.1.1.7 christos * the authority and additional sections will be empty (except possible
182 1.1.1.7 christos * EDNS0 OPT RR in the additional section appended on sending it out),
183 1.1.1.7 christos * so the total number of RRsets is an_numrrsets. */
184 1.1.1.7 christos new_rep = construct_reply_info_base(region, rep->flags,
185 1.1.1.7 christos rep->qdcount, rep->ttl, rep->prefetch_ttl,
186 1.1.1.7 christos rep->serve_expired_ttl, an_numrrsets, 0, 0, an_numrrsets,
187 1.1.1.7.2.1 martin sec_status_insecure, LDNS_EDE_NONE);
188 1.1.1.7 christos if(!new_rep)
189 1.1.1.7 christos return NULL;
190 1.1.1.7 christos if(!reply_info_alloc_rrset_keys(new_rep, NULL, region))
191 1.1.1.7 christos return NULL;
192 1.1.1.7 christos for(i=0; i<copy_rrsets; i++)
193 1.1.1.7 christos new_rep->rrsets[i] = rep->rrsets[i];
194 1.1.1.7 christos
195 1.1.1.7 christos return new_rep;
196 1.1.1.7 christos }
197 1.1.1.7 christos
198 1.1 christos /** find the minimumttl in the rdata of SOA record */
199 1.1 christos static time_t
200 1.1 christos soa_find_minttl(struct rr_parse* rr)
201 1.1 christos {
202 1.1 christos uint16_t rlen = sldns_read_uint16(rr->ttl_data+4);
203 1.1 christos if(rlen < 20)
204 1.1 christos return 0; /* rdata too small for SOA (dname, dname, 5*32bit) */
205 1.1 christos /* minimum TTL is the last 32bit value in the rdata of the record */
206 1.1 christos /* at position ttl_data + 4(ttl) + 2(rdatalen) + rdatalen - 4(timeval)*/
207 1.1 christos return (time_t)sldns_read_uint32(rr->ttl_data+6+rlen-4);
208 1.1 christos }
209 1.1 christos
210 1.1 christos /** do the rdata copy */
211 1.1 christos static int
212 1.1 christos rdata_copy(sldns_buffer* pkt, struct packed_rrset_data* data, uint8_t* to,
213 1.1 christos struct rr_parse* rr, time_t* rr_ttl, uint16_t type,
214 1.1 christos sldns_pkt_section section)
215 1.1 christos {
216 1.1 christos uint16_t pkt_len;
217 1.1 christos const sldns_rr_descriptor* desc;
218 1.1 christos
219 1.1 christos *rr_ttl = sldns_read_uint32(rr->ttl_data);
220 1.1 christos /* RFC 2181 Section 8. if msb of ttl is set treat as if zero. */
221 1.1 christos if(*rr_ttl & 0x80000000U)
222 1.1 christos *rr_ttl = 0;
223 1.1 christos if(type == LDNS_RR_TYPE_SOA && section == LDNS_SECTION_AUTHORITY) {
224 1.1 christos /* negative response. see if TTL of SOA record larger than the
225 1.1 christos * minimum-ttl in the rdata of the SOA record */
226 1.1 christos if(*rr_ttl > soa_find_minttl(rr))
227 1.1 christos *rr_ttl = soa_find_minttl(rr);
228 1.1 christos }
229 1.1.1.6 christos if(!SERVE_ORIGINAL_TTL && (*rr_ttl < MIN_TTL))
230 1.1 christos *rr_ttl = MIN_TTL;
231 1.1.1.6 christos if(!SERVE_ORIGINAL_TTL && (*rr_ttl > MAX_TTL))
232 1.1.1.4 christos *rr_ttl = MAX_TTL;
233 1.1.1.7 christos if(type == LDNS_RR_TYPE_SOA && section == LDNS_SECTION_AUTHORITY) {
234 1.1.1.7 christos /* max neg ttl overrides the min and max ttl of everything
235 1.1.1.7 christos * else, it is for a more specific record */
236 1.1.1.7 christos if(*rr_ttl > MAX_NEG_TTL)
237 1.1.1.7 christos *rr_ttl = MAX_NEG_TTL;
238 1.1.1.7 christos }
239 1.1 christos if(*rr_ttl < data->ttl)
240 1.1 christos data->ttl = *rr_ttl;
241 1.1 christos
242 1.1 christos if(rr->outside_packet) {
243 1.1 christos /* uncompressed already, only needs copy */
244 1.1 christos memmove(to, rr->ttl_data+sizeof(uint32_t), rr->size);
245 1.1 christos return 1;
246 1.1 christos }
247 1.1 christos
248 1.1 christos sldns_buffer_set_position(pkt, (size_t)
249 1.1 christos (rr->ttl_data - sldns_buffer_begin(pkt) + sizeof(uint32_t)));
250 1.1 christos /* insert decompressed size into rdata len stored in memory */
251 1.1 christos /* -2 because rdatalen bytes are not included. */
252 1.1 christos pkt_len = htons(rr->size - 2);
253 1.1 christos memmove(to, &pkt_len, sizeof(uint16_t));
254 1.1 christos to += 2;
255 1.1 christos /* read packet rdata len */
256 1.1 christos pkt_len = sldns_buffer_read_u16(pkt);
257 1.1 christos if(sldns_buffer_remaining(pkt) < pkt_len)
258 1.1 christos return 0;
259 1.1 christos desc = sldns_rr_descript(type);
260 1.1 christos if(pkt_len > 0 && desc && desc->_dname_count > 0) {
261 1.1 christos int count = (int)desc->_dname_count;
262 1.1 christos int rdf = 0;
263 1.1 christos size_t len;
264 1.1 christos size_t oldpos;
265 1.1 christos /* decompress dnames. */
266 1.1 christos while(pkt_len > 0 && count) {
267 1.1 christos switch(desc->_wireformat[rdf]) {
268 1.1 christos case LDNS_RDF_TYPE_DNAME:
269 1.1 christos oldpos = sldns_buffer_position(pkt);
270 1.1 christos dname_pkt_copy(pkt, to,
271 1.1 christos sldns_buffer_current(pkt));
272 1.1 christos to += pkt_dname_len(pkt);
273 1.1 christos pkt_len -= sldns_buffer_position(pkt)-oldpos;
274 1.1 christos count--;
275 1.1 christos len = 0;
276 1.1 christos break;
277 1.1 christos case LDNS_RDF_TYPE_STR:
278 1.1 christos len = sldns_buffer_current(pkt)[0] + 1;
279 1.1 christos break;
280 1.1 christos default:
281 1.1 christos len = get_rdf_size(desc->_wireformat[rdf]);
282 1.1 christos break;
283 1.1 christos }
284 1.1 christos if(len) {
285 1.1.1.5 christos log_assert(len <= pkt_len);
286 1.1 christos memmove(to, sldns_buffer_current(pkt), len);
287 1.1 christos to += len;
288 1.1 christos sldns_buffer_skip(pkt, (ssize_t)len);
289 1.1 christos pkt_len -= len;
290 1.1 christos }
291 1.1 christos rdf++;
292 1.1 christos }
293 1.1 christos }
294 1.1 christos /* copy remaining rdata */
295 1.1 christos if(pkt_len > 0)
296 1.1 christos memmove(to, sldns_buffer_current(pkt), pkt_len);
297 1.1 christos
298 1.1 christos return 1;
299 1.1 christos }
300 1.1 christos
301 1.1 christos /** copy over the data into packed rrset */
302 1.1 christos static int
303 1.1 christos parse_rr_copy(sldns_buffer* pkt, struct rrset_parse* pset,
304 1.1 christos struct packed_rrset_data* data)
305 1.1 christos {
306 1.1 christos size_t i;
307 1.1 christos struct rr_parse* rr = pset->rr_first;
308 1.1 christos uint8_t* nextrdata;
309 1.1 christos size_t total = pset->rr_count + pset->rrsig_count;
310 1.1 christos data->ttl = MAX_TTL;
311 1.1 christos data->count = pset->rr_count;
312 1.1 christos data->rrsig_count = pset->rrsig_count;
313 1.1 christos data->trust = rrset_trust_none;
314 1.1 christos data->security = sec_status_unchecked;
315 1.1 christos /* layout: struct - rr_len - rr_data - rr_ttl - rdata - rrsig */
316 1.1 christos data->rr_len = (size_t*)((uint8_t*)data +
317 1.1 christos sizeof(struct packed_rrset_data));
318 1.1 christos data->rr_data = (uint8_t**)&(data->rr_len[total]);
319 1.1 christos data->rr_ttl = (time_t*)&(data->rr_data[total]);
320 1.1 christos nextrdata = (uint8_t*)&(data->rr_ttl[total]);
321 1.1 christos for(i=0; i<data->count; i++) {
322 1.1 christos data->rr_len[i] = rr->size;
323 1.1 christos data->rr_data[i] = nextrdata;
324 1.1 christos nextrdata += rr->size;
325 1.1 christos if(!rdata_copy(pkt, data, data->rr_data[i], rr,
326 1.1 christos &data->rr_ttl[i], pset->type, pset->section))
327 1.1 christos return 0;
328 1.1 christos rr = rr->next;
329 1.1 christos }
330 1.1 christos /* if rrsig, its rdata is at nextrdata */
331 1.1 christos rr = pset->rrsig_first;
332 1.1 christos for(i=data->count; i<total; i++) {
333 1.1 christos data->rr_len[i] = rr->size;
334 1.1 christos data->rr_data[i] = nextrdata;
335 1.1 christos nextrdata += rr->size;
336 1.1 christos if(!rdata_copy(pkt, data, data->rr_data[i], rr,
337 1.1 christos &data->rr_ttl[i], LDNS_RR_TYPE_RRSIG, pset->section))
338 1.1 christos return 0;
339 1.1 christos rr = rr->next;
340 1.1 christos }
341 1.1 christos return 1;
342 1.1 christos }
343 1.1 christos
344 1.1 christos /** create rrset return 0 on failure */
345 1.1 christos static int
346 1.1 christos parse_create_rrset(sldns_buffer* pkt, struct rrset_parse* pset,
347 1.1 christos struct packed_rrset_data** data, struct regional* region)
348 1.1 christos {
349 1.1 christos /* allocate */
350 1.1 christos size_t s;
351 1.1 christos if(pset->rr_count > RR_COUNT_MAX || pset->rrsig_count > RR_COUNT_MAX ||
352 1.1 christos pset->size > RR_COUNT_MAX)
353 1.1 christos return 0; /* protect against integer overflow */
354 1.1 christos s = sizeof(struct packed_rrset_data) +
355 1.1 christos (pset->rr_count + pset->rrsig_count) *
356 1.1 christos (sizeof(size_t)+sizeof(uint8_t*)+sizeof(time_t)) +
357 1.1 christos pset->size;
358 1.1 christos if(region)
359 1.1.1.6 christos *data = regional_alloc_zero(region, s);
360 1.1.1.6 christos else *data = calloc(1, s);
361 1.1 christos if(!*data)
362 1.1 christos return 0;
363 1.1 christos /* copy & decompress */
364 1.1 christos if(!parse_rr_copy(pkt, pset, *data)) {
365 1.1.1.7 christos if(!region) {
366 1.1.1.7 christos free(*data);
367 1.1.1.7 christos *data = NULL;
368 1.1.1.7 christos }
369 1.1 christos return 0;
370 1.1 christos }
371 1.1 christos return 1;
372 1.1 christos }
373 1.1 christos
374 1.1 christos /** get trust value for rrset */
375 1.1 christos static enum rrset_trust
376 1.1 christos get_rrset_trust(struct msg_parse* msg, struct rrset_parse* rrset)
377 1.1 christos {
378 1.1 christos uint16_t AA = msg->flags & BIT_AA;
379 1.1 christos if(rrset->section == LDNS_SECTION_ANSWER) {
380 1.1 christos if(AA) {
381 1.1 christos /* RFC2181 says remainder of CNAME chain is nonauth*/
382 1.1 christos if(msg->rrset_first &&
383 1.1 christos msg->rrset_first->section==LDNS_SECTION_ANSWER
384 1.1 christos && msg->rrset_first->type==LDNS_RR_TYPE_CNAME){
385 1.1 christos if(rrset == msg->rrset_first)
386 1.1 christos return rrset_trust_ans_AA;
387 1.1 christos else return rrset_trust_ans_noAA;
388 1.1 christos }
389 1.1 christos if(msg->rrset_first &&
390 1.1 christos msg->rrset_first->section==LDNS_SECTION_ANSWER
391 1.1 christos && msg->rrset_first->type==LDNS_RR_TYPE_DNAME){
392 1.1 christos if(rrset == msg->rrset_first ||
393 1.1 christos rrset == msg->rrset_first->rrset_all_next)
394 1.1 christos return rrset_trust_ans_AA;
395 1.1 christos else return rrset_trust_ans_noAA;
396 1.1 christos }
397 1.1 christos return rrset_trust_ans_AA;
398 1.1 christos }
399 1.1 christos else return rrset_trust_ans_noAA;
400 1.1 christos } else if(rrset->section == LDNS_SECTION_AUTHORITY) {
401 1.1 christos if(AA) return rrset_trust_auth_AA;
402 1.1 christos else return rrset_trust_auth_noAA;
403 1.1 christos } else {
404 1.1 christos /* addit section */
405 1.1 christos if(AA) return rrset_trust_add_AA;
406 1.1 christos else return rrset_trust_add_noAA;
407 1.1 christos }
408 1.1 christos /* NOTREACHED */
409 1.1 christos return rrset_trust_none;
410 1.1 christos }
411 1.1 christos
412 1.1 christos int
413 1.1 christos parse_copy_decompress_rrset(sldns_buffer* pkt, struct msg_parse* msg,
414 1.1 christos struct rrset_parse *pset, struct regional* region,
415 1.1 christos struct ub_packed_rrset_key* pk)
416 1.1 christos {
417 1.1 christos struct packed_rrset_data* data;
418 1.1 christos pk->rk.flags = pset->flags;
419 1.1 christos pk->rk.dname_len = pset->dname_len;
420 1.1 christos if(region)
421 1.1 christos pk->rk.dname = (uint8_t*)regional_alloc(
422 1.1 christos region, pset->dname_len);
423 1.1 christos else pk->rk.dname =
424 1.1 christos (uint8_t*)malloc(pset->dname_len);
425 1.1 christos if(!pk->rk.dname)
426 1.1 christos return 0;
427 1.1 christos /** copy & decompress dname */
428 1.1 christos dname_pkt_copy(pkt, pk->rk.dname, pset->dname);
429 1.1 christos /** copy over type and class */
430 1.1 christos pk->rk.type = htons(pset->type);
431 1.1 christos pk->rk.rrset_class = pset->rrset_class;
432 1.1 christos /** read data part. */
433 1.1.1.7 christos if(!parse_create_rrset(pkt, pset, &data, region)) {
434 1.1.1.7 christos if(!region) {
435 1.1.1.7 christos free(pk->rk.dname);
436 1.1.1.7 christos pk->rk.dname = NULL;
437 1.1.1.7 christos }
438 1.1 christos return 0;
439 1.1.1.7 christos }
440 1.1 christos pk->entry.data = (void*)data;
441 1.1 christos pk->entry.key = (void*)pk;
442 1.1 christos pk->entry.hash = pset->hash;
443 1.1 christos data->trust = get_rrset_trust(msg, pset);
444 1.1 christos return 1;
445 1.1 christos }
446 1.1 christos
447 1.1 christos /**
448 1.1 christos * Copy and decompress rrs
449 1.1 christos * @param pkt: the packet for compression pointer resolution.
450 1.1 christos * @param msg: the parsed message
451 1.1 christos * @param rep: reply info to put rrs into.
452 1.1 christos * @param region: if not NULL, used for allocation.
453 1.1 christos * @return 0 on failure.
454 1.1 christos */
455 1.1 christos static int
456 1.1 christos parse_copy_decompress(sldns_buffer* pkt, struct msg_parse* msg,
457 1.1 christos struct reply_info* rep, struct regional* region)
458 1.1 christos {
459 1.1 christos size_t i;
460 1.1 christos struct rrset_parse *pset = msg->rrset_first;
461 1.1 christos struct packed_rrset_data* data;
462 1.1 christos log_assert(rep);
463 1.1 christos rep->ttl = MAX_TTL;
464 1.1 christos rep->security = sec_status_unchecked;
465 1.1 christos if(rep->rrset_count == 0)
466 1.1 christos rep->ttl = NORR_TTL;
467 1.1 christos
468 1.1 christos for(i=0; i<rep->rrset_count; i++) {
469 1.1 christos if(!parse_copy_decompress_rrset(pkt, msg, pset, region,
470 1.1 christos rep->rrsets[i]))
471 1.1 christos return 0;
472 1.1 christos data = (struct packed_rrset_data*)rep->rrsets[i]->entry.data;
473 1.1 christos if(data->ttl < rep->ttl)
474 1.1 christos rep->ttl = data->ttl;
475 1.1 christos
476 1.1 christos pset = pset->rrset_all_next;
477 1.1 christos }
478 1.1 christos rep->prefetch_ttl = PREFETCH_TTL_CALC(rep->ttl);
479 1.1.1.4 christos rep->serve_expired_ttl = rep->ttl + SERVE_EXPIRED_TTL;
480 1.1 christos return 1;
481 1.1 christos }
482 1.1 christos
483 1.1 christos int
484 1.1 christos parse_create_msg(sldns_buffer* pkt, struct msg_parse* msg,
485 1.1 christos struct alloc_cache* alloc, struct query_info* qinf,
486 1.1 christos struct reply_info** rep, struct regional* region)
487 1.1 christos {
488 1.1 christos log_assert(pkt && msg);
489 1.1 christos if(!parse_create_qinfo(pkt, msg, qinf, region))
490 1.1 christos return 0;
491 1.1 christos if(!parse_create_repinfo(msg, rep, region))
492 1.1 christos return 0;
493 1.1.1.4 christos if(!reply_info_alloc_rrset_keys(*rep, alloc, region)) {
494 1.1.1.4 christos if(!region) reply_info_parsedelete(*rep, alloc);
495 1.1 christos return 0;
496 1.1.1.4 christos }
497 1.1.1.4 christos if(!parse_copy_decompress(pkt, msg, *rep, region)) {
498 1.1.1.4 christos if(!region) reply_info_parsedelete(*rep, alloc);
499 1.1 christos return 0;
500 1.1.1.4 christos }
501 1.1 christos return 1;
502 1.1 christos }
503 1.1 christos
504 1.1 christos int reply_info_parse(sldns_buffer* pkt, struct alloc_cache* alloc,
505 1.1 christos struct query_info* qinf, struct reply_info** rep,
506 1.1 christos struct regional* region, struct edns_data* edns)
507 1.1 christos {
508 1.1 christos /* use scratch pad region-allocator during parsing. */
509 1.1 christos struct msg_parse* msg;
510 1.1 christos int ret;
511 1.1 christos
512 1.1 christos qinf->qname = NULL;
513 1.1.1.2 christos qinf->local_alias = NULL;
514 1.1 christos *rep = NULL;
515 1.1 christos if(!(msg = regional_alloc(region, sizeof(*msg)))) {
516 1.1 christos return LDNS_RCODE_SERVFAIL;
517 1.1 christos }
518 1.1 christos memset(msg, 0, sizeof(*msg));
519 1.1 christos
520 1.1 christos sldns_buffer_set_position(pkt, 0);
521 1.1 christos if((ret = parse_packet(pkt, msg, region)) != 0) {
522 1.1 christos return ret;
523 1.1 christos }
524 1.1.1.7 christos if((ret = parse_extract_edns_from_response_msg(msg, edns, region)) != 0)
525 1.1 christos return ret;
526 1.1 christos
527 1.1 christos /* parse OK, allocate return structures */
528 1.1 christos /* this also performs dname decompression */
529 1.1 christos if(!parse_create_msg(pkt, msg, alloc, qinf, rep, NULL)) {
530 1.1 christos query_info_clear(qinf);
531 1.1 christos *rep = NULL;
532 1.1 christos return LDNS_RCODE_SERVFAIL;
533 1.1 christos }
534 1.1 christos return 0;
535 1.1 christos }
536 1.1 christos
537 1.1 christos /** helper compare function to sort in lock order */
538 1.1 christos static int
539 1.1 christos reply_info_sortref_cmp(const void* a, const void* b)
540 1.1 christos {
541 1.1 christos struct rrset_ref* x = (struct rrset_ref*)a;
542 1.1 christos struct rrset_ref* y = (struct rrset_ref*)b;
543 1.1 christos if(x->key < y->key) return -1;
544 1.1 christos if(x->key > y->key) return 1;
545 1.1 christos return 0;
546 1.1 christos }
547 1.1 christos
548 1.1 christos void
549 1.1 christos reply_info_sortref(struct reply_info* rep)
550 1.1 christos {
551 1.1 christos qsort(&rep->ref[0], rep->rrset_count, sizeof(struct rrset_ref),
552 1.1 christos reply_info_sortref_cmp);
553 1.1 christos }
554 1.1 christos
555 1.1 christos void
556 1.1 christos reply_info_set_ttls(struct reply_info* rep, time_t timenow)
557 1.1 christos {
558 1.1 christos size_t i, j;
559 1.1 christos rep->ttl += timenow;
560 1.1 christos rep->prefetch_ttl += timenow;
561 1.1.1.4 christos rep->serve_expired_ttl += timenow;
562 1.1 christos for(i=0; i<rep->rrset_count; i++) {
563 1.1 christos struct packed_rrset_data* data = (struct packed_rrset_data*)
564 1.1 christos rep->ref[i].key->entry.data;
565 1.1 christos if(i>0 && rep->ref[i].key == rep->ref[i-1].key)
566 1.1 christos continue;
567 1.1 christos data->ttl += timenow;
568 1.1 christos for(j=0; j<data->count + data->rrsig_count; j++) {
569 1.1 christos data->rr_ttl[j] += timenow;
570 1.1 christos }
571 1.1.1.6 christos data->ttl_add = timenow;
572 1.1 christos }
573 1.1 christos }
574 1.1 christos
575 1.1 christos void
576 1.1 christos reply_info_parsedelete(struct reply_info* rep, struct alloc_cache* alloc)
577 1.1 christos {
578 1.1 christos size_t i;
579 1.1 christos if(!rep)
580 1.1 christos return;
581 1.1 christos /* no need to lock, since not shared in hashtables. */
582 1.1 christos for(i=0; i<rep->rrset_count; i++) {
583 1.1 christos ub_packed_rrset_parsedelete(rep->rrsets[i], alloc);
584 1.1 christos }
585 1.1.1.7.2.1 martin if(rep->reason_bogus_str) {
586 1.1.1.7.2.1 martin free(rep->reason_bogus_str);
587 1.1.1.7.2.1 martin rep->reason_bogus_str = NULL;
588 1.1.1.7.2.1 martin }
589 1.1 christos free(rep);
590 1.1 christos }
591 1.1 christos
592 1.1 christos int
593 1.1 christos query_info_parse(struct query_info* m, sldns_buffer* query)
594 1.1 christos {
595 1.1 christos uint8_t* q = sldns_buffer_begin(query);
596 1.1 christos /* minimum size: header + \0 + qtype + qclass */
597 1.1 christos if(sldns_buffer_limit(query) < LDNS_HEADER_SIZE + 5)
598 1.1 christos return 0;
599 1.1.1.3 christos if((LDNS_OPCODE_WIRE(q) != LDNS_PACKET_QUERY && LDNS_OPCODE_WIRE(q) !=
600 1.1.1.3 christos LDNS_PACKET_NOTIFY) || LDNS_QDCOUNT(q) != 1 ||
601 1.1.1.3 christos sldns_buffer_position(query) != 0)
602 1.1 christos return 0;
603 1.1 christos sldns_buffer_skip(query, LDNS_HEADER_SIZE);
604 1.1 christos m->qname = sldns_buffer_current(query);
605 1.1 christos if((m->qname_len = query_dname_len(query)) == 0)
606 1.1 christos return 0; /* parse error */
607 1.1 christos if(sldns_buffer_remaining(query) < 4)
608 1.1 christos return 0; /* need qtype, qclass */
609 1.1 christos m->qtype = sldns_buffer_read_u16(query);
610 1.1 christos m->qclass = sldns_buffer_read_u16(query);
611 1.1.1.2 christos m->local_alias = NULL;
612 1.1 christos return 1;
613 1.1 christos }
614 1.1 christos
615 1.1 christos /** tiny subroutine for msgreply_compare */
616 1.1 christos #define COMPARE_IT(x, y) \
617 1.1 christos if( (x) < (y) ) return -1; \
618 1.1 christos else if( (x) > (y) ) return +1; \
619 1.1 christos log_assert( (x) == (y) );
620 1.1 christos
621 1.1 christos int
622 1.1 christos query_info_compare(void* m1, void* m2)
623 1.1 christos {
624 1.1 christos struct query_info* msg1 = (struct query_info*)m1;
625 1.1 christos struct query_info* msg2 = (struct query_info*)m2;
626 1.1 christos int mc;
627 1.1 christos /* from most different to least different for speed */
628 1.1 christos COMPARE_IT(msg1->qtype, msg2->qtype);
629 1.1 christos if((mc = query_dname_compare(msg1->qname, msg2->qname)) != 0)
630 1.1 christos return mc;
631 1.1 christos log_assert(msg1->qname_len == msg2->qname_len);
632 1.1 christos COMPARE_IT(msg1->qclass, msg2->qclass);
633 1.1 christos return 0;
634 1.1 christos #undef COMPARE_IT
635 1.1 christos }
636 1.1 christos
637 1.1 christos void
638 1.1 christos query_info_clear(struct query_info* m)
639 1.1 christos {
640 1.1 christos free(m->qname);
641 1.1 christos m->qname = NULL;
642 1.1 christos }
643 1.1 christos
644 1.1 christos size_t
645 1.1 christos msgreply_sizefunc(void* k, void* d)
646 1.1 christos {
647 1.1 christos struct msgreply_entry* q = (struct msgreply_entry*)k;
648 1.1 christos struct reply_info* r = (struct reply_info*)d;
649 1.1 christos size_t s = sizeof(struct msgreply_entry) + sizeof(struct reply_info)
650 1.1 christos + q->key.qname_len + lock_get_mem(&q->entry.lock)
651 1.1 christos - sizeof(struct rrset_ref);
652 1.1 christos s += r->rrset_count * sizeof(struct rrset_ref);
653 1.1 christos s += r->rrset_count * sizeof(struct ub_packed_rrset_key*);
654 1.1 christos return s;
655 1.1 christos }
656 1.1 christos
657 1.1 christos void
658 1.1 christos query_entry_delete(void *k, void* ATTR_UNUSED(arg))
659 1.1 christos {
660 1.1 christos struct msgreply_entry* q = (struct msgreply_entry*)k;
661 1.1 christos lock_rw_destroy(&q->entry.lock);
662 1.1 christos query_info_clear(&q->key);
663 1.1 christos free(q);
664 1.1 christos }
665 1.1 christos
666 1.1 christos void
667 1.1 christos reply_info_delete(void* d, void* ATTR_UNUSED(arg))
668 1.1 christos {
669 1.1 christos struct reply_info* r = (struct reply_info*)d;
670 1.1.1.7.2.1 martin if(r->reason_bogus_str) {
671 1.1.1.7.2.1 martin free(r->reason_bogus_str);
672 1.1.1.7.2.1 martin r->reason_bogus_str = NULL;
673 1.1.1.7.2.1 martin }
674 1.1 christos free(r);
675 1.1 christos }
676 1.1 christos
677 1.1.1.2 christos hashvalue_type
678 1.1 christos query_info_hash(struct query_info *q, uint16_t flags)
679 1.1 christos {
680 1.1.1.2 christos hashvalue_type h = 0xab;
681 1.1 christos h = hashlittle(&q->qtype, sizeof(q->qtype), h);
682 1.1 christos if(q->qtype == LDNS_RR_TYPE_AAAA && (flags&BIT_CD))
683 1.1 christos h++;
684 1.1 christos h = hashlittle(&q->qclass, sizeof(q->qclass), h);
685 1.1 christos h = dname_query_hash(q->qname, h);
686 1.1 christos return h;
687 1.1 christos }
688 1.1 christos
689 1.1 christos struct msgreply_entry*
690 1.1 christos query_info_entrysetup(struct query_info* q, struct reply_info* r,
691 1.1.1.2 christos hashvalue_type h)
692 1.1 christos {
693 1.1 christos struct msgreply_entry* e = (struct msgreply_entry*)malloc(
694 1.1 christos sizeof(struct msgreply_entry));
695 1.1 christos if(!e) return NULL;
696 1.1 christos memcpy(&e->key, q, sizeof(*q));
697 1.1 christos e->entry.hash = h;
698 1.1 christos e->entry.key = e;
699 1.1 christos e->entry.data = r;
700 1.1 christos lock_rw_init(&e->entry.lock);
701 1.1.1.3 christos lock_protect(&e->entry.lock, &e->key.qname, sizeof(e->key.qname));
702 1.1.1.3 christos lock_protect(&e->entry.lock, &e->key.qname_len, sizeof(e->key.qname_len));
703 1.1.1.3 christos lock_protect(&e->entry.lock, &e->key.qtype, sizeof(e->key.qtype));
704 1.1.1.3 christos lock_protect(&e->entry.lock, &e->key.qclass, sizeof(e->key.qclass));
705 1.1.1.3 christos lock_protect(&e->entry.lock, &e->key.local_alias, sizeof(e->key.local_alias));
706 1.1.1.3 christos lock_protect(&e->entry.lock, &e->entry.hash, sizeof(e->entry.hash));
707 1.1.1.3 christos lock_protect(&e->entry.lock, &e->entry.key, sizeof(e->entry.key));
708 1.1.1.3 christos lock_protect(&e->entry.lock, &e->entry.data, sizeof(e->entry.data));
709 1.1 christos lock_protect(&e->entry.lock, e->key.qname, e->key.qname_len);
710 1.1 christos q->qname = NULL;
711 1.1 christos return e;
712 1.1 christos }
713 1.1 christos
714 1.1 christos /** copy rrsets from replyinfo to dest replyinfo */
715 1.1 christos static int
716 1.1 christos repinfo_copy_rrsets(struct reply_info* dest, struct reply_info* from,
717 1.1 christos struct regional* region)
718 1.1 christos {
719 1.1 christos size_t i, s;
720 1.1 christos struct packed_rrset_data* fd, *dd;
721 1.1 christos struct ub_packed_rrset_key* fk, *dk;
722 1.1 christos for(i=0; i<dest->rrset_count; i++) {
723 1.1 christos fk = from->rrsets[i];
724 1.1 christos dk = dest->rrsets[i];
725 1.1 christos fd = (struct packed_rrset_data*)fk->entry.data;
726 1.1 christos dk->entry.hash = fk->entry.hash;
727 1.1 christos dk->rk = fk->rk;
728 1.1 christos if(region) {
729 1.1 christos dk->id = fk->id;
730 1.1 christos dk->rk.dname = (uint8_t*)regional_alloc_init(region,
731 1.1 christos fk->rk.dname, fk->rk.dname_len);
732 1.1 christos } else
733 1.1 christos dk->rk.dname = (uint8_t*)memdup(fk->rk.dname,
734 1.1 christos fk->rk.dname_len);
735 1.1 christos if(!dk->rk.dname)
736 1.1 christos return 0;
737 1.1 christos s = packed_rrset_sizeof(fd);
738 1.1 christos if(region)
739 1.1 christos dd = (struct packed_rrset_data*)regional_alloc_init(
740 1.1 christos region, fd, s);
741 1.1 christos else dd = (struct packed_rrset_data*)memdup(fd, s);
742 1.1 christos if(!dd)
743 1.1 christos return 0;
744 1.1 christos packed_rrset_ptr_fixup(dd);
745 1.1 christos dk->entry.data = (void*)dd;
746 1.1 christos }
747 1.1 christos return 1;
748 1.1 christos }
749 1.1 christos
750 1.1.1.7.2.1 martin struct reply_info*
751 1.1.1.7.2.1 martin reply_info_copy(struct reply_info* rep, struct alloc_cache* alloc,
752 1.1 christos struct regional* region)
753 1.1 christos {
754 1.1 christos struct reply_info* cp;
755 1.1.1.7.2.1 martin cp = construct_reply_info_base(region, rep->flags, rep->qdcount,
756 1.1.1.7.2.1 martin rep->ttl, rep->prefetch_ttl, rep->serve_expired_ttl,
757 1.1.1.4 christos rep->an_numrrsets, rep->ns_numrrsets, rep->ar_numrrsets,
758 1.1.1.7.2.1 martin rep->rrset_count, rep->security, rep->reason_bogus);
759 1.1 christos if(!cp)
760 1.1 christos return NULL;
761 1.1.1.7.2.1 martin
762 1.1.1.7.2.1 martin if(rep->reason_bogus_str && *rep->reason_bogus_str != 0) {
763 1.1.1.7.2.1 martin if(region) {
764 1.1.1.7.2.1 martin cp->reason_bogus_str = (char*)regional_alloc(region,
765 1.1.1.7.2.1 martin sizeof(char)
766 1.1.1.7.2.1 martin * (strlen(rep->reason_bogus_str)+1));
767 1.1.1.7.2.1 martin } else {
768 1.1.1.7.2.1 martin cp->reason_bogus_str = malloc(sizeof(char)
769 1.1.1.7.2.1 martin * (strlen(rep->reason_bogus_str)+1));
770 1.1.1.7.2.1 martin }
771 1.1.1.7.2.1 martin if(!cp->reason_bogus_str) {
772 1.1.1.7.2.1 martin if(!region)
773 1.1.1.7.2.1 martin reply_info_parsedelete(cp, alloc);
774 1.1.1.7.2.1 martin return NULL;
775 1.1.1.7.2.1 martin }
776 1.1.1.7.2.1 martin memcpy(cp->reason_bogus_str, rep->reason_bogus_str,
777 1.1.1.7.2.1 martin strlen(rep->reason_bogus_str)+1);
778 1.1.1.7.2.1 martin }
779 1.1.1.7.2.1 martin
780 1.1 christos /* allocate ub_key structures special or not */
781 1.1.1.2 christos if(!reply_info_alloc_rrset_keys(cp, alloc, region)) {
782 1.1 christos if(!region)
783 1.1 christos reply_info_parsedelete(cp, alloc);
784 1.1 christos return NULL;
785 1.1 christos }
786 1.1 christos if(!repinfo_copy_rrsets(cp, rep, region)) {
787 1.1 christos if(!region)
788 1.1 christos reply_info_parsedelete(cp, alloc);
789 1.1 christos return NULL;
790 1.1 christos }
791 1.1 christos return cp;
792 1.1 christos }
793 1.1 christos
794 1.1 christos uint8_t*
795 1.1 christos reply_find_final_cname_target(struct query_info* qinfo, struct reply_info* rep)
796 1.1 christos {
797 1.1 christos uint8_t* sname = qinfo->qname;
798 1.1 christos size_t snamelen = qinfo->qname_len;
799 1.1 christos size_t i;
800 1.1 christos for(i=0; i<rep->an_numrrsets; i++) {
801 1.1 christos struct ub_packed_rrset_key* s = rep->rrsets[i];
802 1.1 christos /* follow CNAME chain (if any) */
803 1.1 christos if(ntohs(s->rk.type) == LDNS_RR_TYPE_CNAME &&
804 1.1 christos ntohs(s->rk.rrset_class) == qinfo->qclass &&
805 1.1 christos snamelen == s->rk.dname_len &&
806 1.1 christos query_dname_compare(sname, s->rk.dname) == 0) {
807 1.1 christos get_cname_target(s, &sname, &snamelen);
808 1.1 christos }
809 1.1 christos }
810 1.1 christos if(sname != qinfo->qname)
811 1.1 christos return sname;
812 1.1 christos return NULL;
813 1.1 christos }
814 1.1 christos
815 1.1 christos struct ub_packed_rrset_key*
816 1.1 christos reply_find_answer_rrset(struct query_info* qinfo, struct reply_info* rep)
817 1.1 christos {
818 1.1 christos uint8_t* sname = qinfo->qname;
819 1.1 christos size_t snamelen = qinfo->qname_len;
820 1.1 christos size_t i;
821 1.1 christos for(i=0; i<rep->an_numrrsets; i++) {
822 1.1 christos struct ub_packed_rrset_key* s = rep->rrsets[i];
823 1.1 christos /* first match type, for query of qtype cname */
824 1.1 christos if(ntohs(s->rk.type) == qinfo->qtype &&
825 1.1 christos ntohs(s->rk.rrset_class) == qinfo->qclass &&
826 1.1 christos snamelen == s->rk.dname_len &&
827 1.1 christos query_dname_compare(sname, s->rk.dname) == 0) {
828 1.1 christos return s;
829 1.1 christos }
830 1.1 christos /* follow CNAME chain (if any) */
831 1.1 christos if(ntohs(s->rk.type) == LDNS_RR_TYPE_CNAME &&
832 1.1 christos ntohs(s->rk.rrset_class) == qinfo->qclass &&
833 1.1 christos snamelen == s->rk.dname_len &&
834 1.1 christos query_dname_compare(sname, s->rk.dname) == 0) {
835 1.1 christos get_cname_target(s, &sname, &snamelen);
836 1.1 christos }
837 1.1 christos }
838 1.1 christos return NULL;
839 1.1 christos }
840 1.1 christos
841 1.1 christos struct ub_packed_rrset_key* reply_find_rrset_section_an(struct reply_info* rep,
842 1.1 christos uint8_t* name, size_t namelen, uint16_t type, uint16_t dclass)
843 1.1 christos {
844 1.1 christos size_t i;
845 1.1 christos for(i=0; i<rep->an_numrrsets; i++) {
846 1.1 christos struct ub_packed_rrset_key* s = rep->rrsets[i];
847 1.1 christos if(ntohs(s->rk.type) == type &&
848 1.1 christos ntohs(s->rk.rrset_class) == dclass &&
849 1.1 christos namelen == s->rk.dname_len &&
850 1.1 christos query_dname_compare(name, s->rk.dname) == 0) {
851 1.1 christos return s;
852 1.1 christos }
853 1.1 christos }
854 1.1 christos return NULL;
855 1.1 christos }
856 1.1 christos
857 1.1 christos struct ub_packed_rrset_key* reply_find_rrset_section_ns(struct reply_info* rep,
858 1.1 christos uint8_t* name, size_t namelen, uint16_t type, uint16_t dclass)
859 1.1 christos {
860 1.1 christos size_t i;
861 1.1 christos for(i=rep->an_numrrsets; i<rep->an_numrrsets+rep->ns_numrrsets; i++) {
862 1.1 christos struct ub_packed_rrset_key* s = rep->rrsets[i];
863 1.1 christos if(ntohs(s->rk.type) == type &&
864 1.1 christos ntohs(s->rk.rrset_class) == dclass &&
865 1.1 christos namelen == s->rk.dname_len &&
866 1.1 christos query_dname_compare(name, s->rk.dname) == 0) {
867 1.1 christos return s;
868 1.1 christos }
869 1.1 christos }
870 1.1 christos return NULL;
871 1.1 christos }
872 1.1 christos
873 1.1 christos struct ub_packed_rrset_key* reply_find_rrset(struct reply_info* rep,
874 1.1 christos uint8_t* name, size_t namelen, uint16_t type, uint16_t dclass)
875 1.1 christos {
876 1.1 christos size_t i;
877 1.1 christos for(i=0; i<rep->rrset_count; i++) {
878 1.1 christos struct ub_packed_rrset_key* s = rep->rrsets[i];
879 1.1 christos if(ntohs(s->rk.type) == type &&
880 1.1 christos ntohs(s->rk.rrset_class) == dclass &&
881 1.1 christos namelen == s->rk.dname_len &&
882 1.1 christos query_dname_compare(name, s->rk.dname) == 0) {
883 1.1 christos return s;
884 1.1 christos }
885 1.1 christos }
886 1.1 christos return NULL;
887 1.1 christos }
888 1.1 christos
889 1.1 christos void
890 1.1 christos log_dns_msg(const char* str, struct query_info* qinfo, struct reply_info* rep)
891 1.1 christos {
892 1.1 christos /* not particularly fast but flexible, make wireformat and print */
893 1.1 christos sldns_buffer* buf = sldns_buffer_new(65535);
894 1.1 christos struct regional* region = regional_create();
895 1.1.1.7 christos if(!(buf && region)) {
896 1.1.1.7 christos log_err("%s: log_dns_msg: out of memory", str);
897 1.1.1.7 christos sldns_buffer_free(buf);
898 1.1.1.7 christos regional_destroy(region);
899 1.1.1.7 christos return;
900 1.1.1.7 christos }
901 1.1.1.7 christos if(!reply_info_encode(qinfo, rep, 0, rep->flags, buf, 0,
902 1.1.1.5 christos region, 65535, 1, 0)) {
903 1.1.1.7 christos log_err("%s: log_dns_msg: out of memory", str);
904 1.1 christos } else {
905 1.1 christos char* s = sldns_wire2str_pkt(sldns_buffer_begin(buf),
906 1.1 christos sldns_buffer_limit(buf));
907 1.1 christos if(!s) {
908 1.1 christos log_info("%s: log_dns_msg: ldns tostr failed", str);
909 1.1 christos } else {
910 1.1 christos log_info("%s %s", str, s);
911 1.1 christos }
912 1.1 christos free(s);
913 1.1 christos }
914 1.1 christos sldns_buffer_free(buf);
915 1.1 christos regional_destroy(region);
916 1.1 christos }
917 1.1 christos
918 1.1.1.2 christos void
919 1.1.1.2 christos log_reply_info(enum verbosity_value v, struct query_info *qinf,
920 1.1.1.2 christos struct sockaddr_storage *addr, socklen_t addrlen, struct timeval dur,
921 1.1.1.2 christos int cached, struct sldns_buffer *rmsg)
922 1.1.1.2 christos {
923 1.1.1.2 christos char qname_buf[LDNS_MAX_DOMAINLEN+1];
924 1.1.1.2 christos char clientip_buf[128];
925 1.1.1.2 christos char rcode_buf[16];
926 1.1.1.2 christos char type_buf[16];
927 1.1.1.2 christos char class_buf[16];
928 1.1.1.2 christos size_t pktlen;
929 1.1.1.2 christos uint16_t rcode = FLAGS_GET_RCODE(sldns_buffer_read_u16_at(rmsg, 2));
930 1.1.1.2 christos
931 1.1.1.2 christos if(verbosity < v)
932 1.1.1.2 christos return;
933 1.1.1.2 christos
934 1.1.1.2 christos sldns_wire2str_rcode_buf((int)rcode, rcode_buf, sizeof(rcode_buf));
935 1.1.1.2 christos addr_to_str(addr, addrlen, clientip_buf, sizeof(clientip_buf));
936 1.1.1.2 christos if(rcode == LDNS_RCODE_FORMERR)
937 1.1.1.2 christos {
938 1.1.1.4 christos if(LOG_TAG_QUERYREPLY)
939 1.1.1.4 christos log_reply("%s - - - %s - - - ", clientip_buf, rcode_buf);
940 1.1.1.4 christos else log_info("%s - - - %s - - - ", clientip_buf, rcode_buf);
941 1.1.1.2 christos } else {
942 1.1.1.2 christos if(qinf->qname)
943 1.1.1.2 christos dname_str(qinf->qname, qname_buf);
944 1.1.1.2 christos else snprintf(qname_buf, sizeof(qname_buf), "null");
945 1.1.1.2 christos pktlen = sldns_buffer_limit(rmsg);
946 1.1.1.2 christos sldns_wire2str_type_buf(qinf->qtype, type_buf, sizeof(type_buf));
947 1.1.1.2 christos sldns_wire2str_class_buf(qinf->qclass, class_buf, sizeof(class_buf));
948 1.1.1.4 christos if(LOG_TAG_QUERYREPLY)
949 1.1.1.4 christos log_reply("%s %s %s %s %s " ARG_LL "d.%6.6d %d %d",
950 1.1.1.4 christos clientip_buf, qname_buf, type_buf, class_buf,
951 1.1.1.4 christos rcode_buf, (long long)dur.tv_sec, (int)dur.tv_usec, cached, (int)pktlen);
952 1.1.1.4 christos else log_info("%s %s %s %s %s " ARG_LL "d.%6.6d %d %d",
953 1.1.1.2 christos clientip_buf, qname_buf, type_buf, class_buf,
954 1.1.1.2 christos rcode_buf, (long long)dur.tv_sec, (int)dur.tv_usec, cached, (int)pktlen);
955 1.1.1.2 christos }
956 1.1.1.2 christos }
957 1.1.1.2 christos
958 1.1.1.2 christos void
959 1.1 christos log_query_info(enum verbosity_value v, const char* str,
960 1.1 christos struct query_info* qinf)
961 1.1 christos {
962 1.1 christos log_nametypeclass(v, str, qinf->qname, qinf->qtype, qinf->qclass);
963 1.1 christos }
964 1.1 christos
965 1.1 christos int
966 1.1 christos reply_check_cname_chain(struct query_info* qinfo, struct reply_info* rep)
967 1.1 christos {
968 1.1 christos /* check only answer section rrs for matching cname chain.
969 1.1 christos * the cache may return changed rdata, but owner names are untouched.*/
970 1.1 christos size_t i;
971 1.1 christos uint8_t* sname = qinfo->qname;
972 1.1 christos size_t snamelen = qinfo->qname_len;
973 1.1 christos for(i=0; i<rep->an_numrrsets; i++) {
974 1.1 christos uint16_t t = ntohs(rep->rrsets[i]->rk.type);
975 1.1 christos if(t == LDNS_RR_TYPE_DNAME)
976 1.1 christos continue; /* skip dnames; note TTL 0 not cached */
977 1.1 christos /* verify that owner matches current sname */
978 1.1 christos if(query_dname_compare(sname, rep->rrsets[i]->rk.dname) != 0){
979 1.1 christos /* cname chain broken */
980 1.1 christos return 0;
981 1.1 christos }
982 1.1 christos /* if this is a cname; move on */
983 1.1 christos if(t == LDNS_RR_TYPE_CNAME) {
984 1.1 christos get_cname_target(rep->rrsets[i], &sname, &snamelen);
985 1.1 christos }
986 1.1 christos }
987 1.1 christos return 1;
988 1.1 christos }
989 1.1 christos
990 1.1 christos int
991 1.1 christos reply_all_rrsets_secure(struct reply_info* rep)
992 1.1 christos {
993 1.1 christos size_t i;
994 1.1 christos for(i=0; i<rep->rrset_count; i++) {
995 1.1 christos if( ((struct packed_rrset_data*)rep->rrsets[i]->entry.data)
996 1.1 christos ->security != sec_status_secure )
997 1.1 christos return 0;
998 1.1 christos }
999 1.1 christos return 1;
1000 1.1 christos }
1001 1.1 christos
1002 1.1.1.3 christos struct reply_info*
1003 1.1.1.3 christos parse_reply_in_temp_region(sldns_buffer* pkt, struct regional* region,
1004 1.1.1.3 christos struct query_info* qi)
1005 1.1.1.3 christos {
1006 1.1.1.3 christos struct reply_info* rep;
1007 1.1.1.3 christos struct msg_parse* msg;
1008 1.1.1.3 christos if(!(msg = regional_alloc(region, sizeof(*msg)))) {
1009 1.1.1.3 christos return NULL;
1010 1.1.1.3 christos }
1011 1.1.1.3 christos memset(msg, 0, sizeof(*msg));
1012 1.1.1.3 christos sldns_buffer_set_position(pkt, 0);
1013 1.1.1.4 christos if(parse_packet(pkt, msg, region) != 0){
1014 1.1.1.3 christos return 0;
1015 1.1.1.4 christos }
1016 1.1.1.3 christos if(!parse_create_msg(pkt, msg, NULL, qi, &rep, region)) {
1017 1.1.1.3 christos return 0;
1018 1.1.1.3 christos }
1019 1.1.1.3 christos return rep;
1020 1.1.1.3 christos }
1021 1.1.1.3 christos
1022 1.1.1.7 christos int edns_opt_list_append_ede(struct edns_option** list, struct regional* region,
1023 1.1.1.7 christos sldns_ede_code code, const char *txt)
1024 1.1 christos {
1025 1.1 christos struct edns_option** prevp;
1026 1.1 christos struct edns_option* opt;
1027 1.1.1.7 christos size_t txt_len = txt ? strlen(txt) : 0;
1028 1.1 christos
1029 1.1 christos /* allocate new element */
1030 1.1 christos opt = (struct edns_option*)regional_alloc(region, sizeof(*opt));
1031 1.1 christos if(!opt)
1032 1.1 christos return 0;
1033 1.1 christos opt->next = NULL;
1034 1.1.1.7 christos opt->opt_code = LDNS_EDNS_EDE;
1035 1.1.1.7 christos opt->opt_len = txt_len + sizeof(uint16_t);
1036 1.1.1.7 christos opt->opt_data = regional_alloc(region, txt_len + sizeof(uint16_t));
1037 1.1.1.7 christos if(!opt->opt_data)
1038 1.1.1.7 christos return 0;
1039 1.1.1.7 christos sldns_write_uint16(opt->opt_data, (uint16_t)code);
1040 1.1.1.7 christos if (txt_len)
1041 1.1.1.7 christos memmove(opt->opt_data + 2, txt, txt_len);
1042 1.1.1.7 christos
1043 1.1 christos /* append at end of list */
1044 1.1.1.7 christos prevp = list;
1045 1.1 christos while(*prevp != NULL)
1046 1.1 christos prevp = &((*prevp)->next);
1047 1.1.1.7 christos verbose(VERB_ALGO, "attached EDE code: %d with message: %s", code, (txt?txt:"\"\""));
1048 1.1 christos *prevp = opt;
1049 1.1 christos return 1;
1050 1.1 christos }
1051 1.1 christos
1052 1.1.1.7.2.1 martin int edns_opt_list_append_keepalive(struct edns_option** list, int msec,
1053 1.1.1.7.2.1 martin struct regional* region)
1054 1.1.1.7.2.1 martin {
1055 1.1.1.7.2.1 martin uint8_t data[2]; /* For keepalive value */
1056 1.1.1.7.2.1 martin data[0] = (uint8_t)((msec >> 8) & 0xff);
1057 1.1.1.7.2.1 martin data[1] = (uint8_t)(msec & 0xff);
1058 1.1.1.7.2.1 martin return edns_opt_list_append(list, LDNS_EDNS_KEEPALIVE, sizeof(data),
1059 1.1.1.7.2.1 martin data, region);
1060 1.1.1.7.2.1 martin }
1061 1.1.1.7.2.1 martin
1062 1.1.1.2 christos int edns_opt_list_append(struct edns_option** list, uint16_t code, size_t len,
1063 1.1.1.2 christos uint8_t* data, struct regional* region)
1064 1.1 christos {
1065 1.1.1.2 christos struct edns_option** prevp;
1066 1.1.1.2 christos struct edns_option* opt;
1067 1.1.1.2 christos
1068 1.1.1.2 christos /* allocate new element */
1069 1.1.1.2 christos opt = (struct edns_option*)regional_alloc(region, sizeof(*opt));
1070 1.1.1.2 christos if(!opt)
1071 1.1.1.2 christos return 0;
1072 1.1.1.2 christos opt->next = NULL;
1073 1.1.1.2 christos opt->opt_code = code;
1074 1.1.1.2 christos opt->opt_len = len;
1075 1.1.1.2 christos opt->opt_data = NULL;
1076 1.1.1.2 christos if(len > 0) {
1077 1.1.1.2 christos opt->opt_data = regional_alloc_init(region, data, len);
1078 1.1.1.2 christos if(!opt->opt_data)
1079 1.1.1.2 christos return 0;
1080 1.1.1.2 christos }
1081 1.1.1.2 christos
1082 1.1.1.2 christos /* append at end of list */
1083 1.1.1.2 christos prevp = list;
1084 1.1.1.2 christos while(*prevp != NULL) {
1085 1.1.1.2 christos prevp = &((*prevp)->next);
1086 1.1.1.2 christos }
1087 1.1.1.2 christos *prevp = opt;
1088 1.1.1.2 christos return 1;
1089 1.1.1.2 christos }
1090 1.1.1.2 christos
1091 1.1.1.2 christos int edns_opt_list_remove(struct edns_option** list, uint16_t code)
1092 1.1.1.2 christos {
1093 1.1.1.2 christos /* The list should already be allocated in a region. Freeing the
1094 1.1.1.2 christos * allocated space in a region is not possible. We just unlink the
1095 1.1.1.2 christos * required elements and they will be freed together with the region. */
1096 1.1.1.2 christos
1097 1.1.1.2 christos struct edns_option* prev;
1098 1.1.1.2 christos struct edns_option* curr;
1099 1.1.1.2 christos if(!list || !(*list)) return 0;
1100 1.1.1.2 christos
1101 1.1.1.2 christos /* Unlink and repoint if the element(s) are first in list */
1102 1.1.1.2 christos while(list && *list && (*list)->opt_code == code) {
1103 1.1.1.2 christos *list = (*list)->next;
1104 1.1.1.2 christos }
1105 1.1.1.2 christos
1106 1.1.1.2 christos if(!list || !(*list)) return 1;
1107 1.1.1.2 christos /* Unlink elements and reattach the list */
1108 1.1.1.2 christos prev = *list;
1109 1.1.1.2 christos curr = (*list)->next;
1110 1.1.1.2 christos while(curr != NULL) {
1111 1.1.1.2 christos if(curr->opt_code == code) {
1112 1.1.1.2 christos prev->next = curr->next;
1113 1.1.1.2 christos curr = curr->next;
1114 1.1.1.2 christos } else {
1115 1.1.1.2 christos prev = curr;
1116 1.1.1.2 christos curr = curr->next;
1117 1.1.1.2 christos }
1118 1.1.1.2 christos }
1119 1.1.1.2 christos return 1;
1120 1.1.1.2 christos }
1121 1.1.1.2 christos
1122 1.1.1.2 christos static int inplace_cb_reply_call_generic(
1123 1.1.1.2 christos struct inplace_cb* callback_list, enum inplace_cb_list_type type,
1124 1.1.1.2 christos struct query_info* qinfo, struct module_qstate* qstate,
1125 1.1.1.2 christos struct reply_info* rep, int rcode, struct edns_data* edns,
1126 1.1.1.6 christos struct comm_reply* repinfo, struct regional* region,
1127 1.1.1.6 christos struct timeval* start_time)
1128 1.1.1.2 christos {
1129 1.1.1.2 christos struct inplace_cb* cb;
1130 1.1.1.2 christos struct edns_option* opt_list_out = NULL;
1131 1.1.1.2 christos #if defined(EXPORT_ALL_SYMBOLS)
1132 1.1.1.2 christos (void)type; /* param not used when fptr_ok disabled */
1133 1.1.1.2 christos #endif
1134 1.1.1.2 christos if(qstate)
1135 1.1.1.2 christos opt_list_out = qstate->edns_opts_front_out;
1136 1.1.1.2 christos for(cb=callback_list; cb; cb=cb->next) {
1137 1.1.1.2 christos fptr_ok(fptr_whitelist_inplace_cb_reply_generic(
1138 1.1.1.2 christos (inplace_cb_reply_func_type*)cb->cb, type));
1139 1.1.1.2 christos (void)(*(inplace_cb_reply_func_type*)cb->cb)(qinfo, qstate, rep,
1140 1.1.1.6 christos rcode, edns, &opt_list_out, repinfo, region, start_time, cb->id, cb->cb_arg);
1141 1.1.1.2 christos }
1142 1.1.1.7 christos edns->opt_list_inplace_cb_out = opt_list_out;
1143 1.1.1.2 christos return 1;
1144 1.1.1.2 christos }
1145 1.1.1.2 christos
1146 1.1.1.2 christos int inplace_cb_reply_call(struct module_env* env, struct query_info* qinfo,
1147 1.1.1.2 christos struct module_qstate* qstate, struct reply_info* rep, int rcode,
1148 1.1.1.6 christos struct edns_data* edns, struct comm_reply* repinfo, struct regional* region,
1149 1.1.1.6 christos struct timeval* start_time)
1150 1.1.1.2 christos {
1151 1.1.1.2 christos return inplace_cb_reply_call_generic(
1152 1.1.1.2 christos env->inplace_cb_lists[inplace_cb_reply], inplace_cb_reply, qinfo,
1153 1.1.1.6 christos qstate, rep, rcode, edns, repinfo, region, start_time);
1154 1.1.1.2 christos }
1155 1.1.1.2 christos
1156 1.1.1.2 christos int inplace_cb_reply_cache_call(struct module_env* env,
1157 1.1.1.2 christos struct query_info* qinfo, struct module_qstate* qstate,
1158 1.1.1.2 christos struct reply_info* rep, int rcode, struct edns_data* edns,
1159 1.1.1.6 christos struct comm_reply* repinfo, struct regional* region,
1160 1.1.1.6 christos struct timeval* start_time)
1161 1.1.1.2 christos {
1162 1.1.1.2 christos return inplace_cb_reply_call_generic(
1163 1.1.1.2 christos env->inplace_cb_lists[inplace_cb_reply_cache], inplace_cb_reply_cache,
1164 1.1.1.6 christos qinfo, qstate, rep, rcode, edns, repinfo, region, start_time);
1165 1.1.1.2 christos }
1166 1.1.1.2 christos
1167 1.1.1.2 christos int inplace_cb_reply_local_call(struct module_env* env,
1168 1.1.1.2 christos struct query_info* qinfo, struct module_qstate* qstate,
1169 1.1.1.2 christos struct reply_info* rep, int rcode, struct edns_data* edns,
1170 1.1.1.6 christos struct comm_reply* repinfo, struct regional* region,
1171 1.1.1.6 christos struct timeval* start_time)
1172 1.1.1.2 christos {
1173 1.1.1.2 christos return inplace_cb_reply_call_generic(
1174 1.1.1.2 christos env->inplace_cb_lists[inplace_cb_reply_local], inplace_cb_reply_local,
1175 1.1.1.6 christos qinfo, qstate, rep, rcode, edns, repinfo, region, start_time);
1176 1.1.1.2 christos }
1177 1.1.1.2 christos
1178 1.1.1.2 christos int inplace_cb_reply_servfail_call(struct module_env* env,
1179 1.1.1.2 christos struct query_info* qinfo, struct module_qstate* qstate,
1180 1.1.1.2 christos struct reply_info* rep, int rcode, struct edns_data* edns,
1181 1.1.1.6 christos struct comm_reply* repinfo, struct regional* region,
1182 1.1.1.6 christos struct timeval* start_time)
1183 1.1.1.2 christos {
1184 1.1.1.2 christos /* We are going to servfail. Remove any potential edns options. */
1185 1.1.1.2 christos if(qstate)
1186 1.1.1.2 christos qstate->edns_opts_front_out = NULL;
1187 1.1.1.2 christos return inplace_cb_reply_call_generic(
1188 1.1.1.2 christos env->inplace_cb_lists[inplace_cb_reply_servfail],
1189 1.1.1.4 christos inplace_cb_reply_servfail, qinfo, qstate, rep, rcode, edns, repinfo,
1190 1.1.1.6 christos region, start_time);
1191 1.1.1.2 christos }
1192 1.1.1.2 christos
1193 1.1.1.2 christos int inplace_cb_query_call(struct module_env* env, struct query_info* qinfo,
1194 1.1.1.2 christos uint16_t flags, struct sockaddr_storage* addr, socklen_t addrlen,
1195 1.1.1.2 christos uint8_t* zone, size_t zonelen, struct module_qstate* qstate,
1196 1.1.1.2 christos struct regional* region)
1197 1.1.1.2 christos {
1198 1.1.1.2 christos struct inplace_cb* cb = env->inplace_cb_lists[inplace_cb_query];
1199 1.1.1.2 christos for(; cb; cb=cb->next) {
1200 1.1.1.2 christos fptr_ok(fptr_whitelist_inplace_cb_query(
1201 1.1.1.2 christos (inplace_cb_query_func_type*)cb->cb));
1202 1.1.1.2 christos (void)(*(inplace_cb_query_func_type*)cb->cb)(qinfo, flags,
1203 1.1.1.2 christos qstate, addr, addrlen, zone, zonelen, region,
1204 1.1.1.2 christos cb->id, cb->cb_arg);
1205 1.1.1.2 christos }
1206 1.1.1.2 christos return 1;
1207 1.1.1.2 christos }
1208 1.1.1.2 christos
1209 1.1.1.2 christos int inplace_cb_edns_back_parsed_call(struct module_env* env,
1210 1.1.1.2 christos struct module_qstate* qstate)
1211 1.1.1.2 christos {
1212 1.1.1.2 christos struct inplace_cb* cb =
1213 1.1.1.2 christos env->inplace_cb_lists[inplace_cb_edns_back_parsed];
1214 1.1.1.2 christos for(; cb; cb=cb->next) {
1215 1.1.1.2 christos fptr_ok(fptr_whitelist_inplace_cb_edns_back_parsed(
1216 1.1.1.2 christos (inplace_cb_edns_back_parsed_func_type*)cb->cb));
1217 1.1.1.2 christos (void)(*(inplace_cb_edns_back_parsed_func_type*)cb->cb)(qstate,
1218 1.1.1.2 christos cb->id, cb->cb_arg);
1219 1.1.1.2 christos }
1220 1.1.1.2 christos return 1;
1221 1.1.1.2 christos }
1222 1.1.1.2 christos
1223 1.1.1.2 christos int inplace_cb_query_response_call(struct module_env* env,
1224 1.1.1.2 christos struct module_qstate* qstate, struct dns_msg* response) {
1225 1.1.1.2 christos struct inplace_cb* cb =
1226 1.1.1.2 christos env->inplace_cb_lists[inplace_cb_query_response];
1227 1.1.1.2 christos for(; cb; cb=cb->next) {
1228 1.1.1.2 christos fptr_ok(fptr_whitelist_inplace_cb_query_response(
1229 1.1.1.2 christos (inplace_cb_query_response_func_type*)cb->cb));
1230 1.1.1.2 christos (void)(*(inplace_cb_query_response_func_type*)cb->cb)(qstate,
1231 1.1.1.2 christos response, cb->id, cb->cb_arg);
1232 1.1.1.2 christos }
1233 1.1 christos return 1;
1234 1.1 christos }
1235 1.1 christos
1236 1.1 christos struct edns_option* edns_opt_copy_region(struct edns_option* list,
1237 1.1.1.7.2.1 martin struct regional* region)
1238 1.1 christos {
1239 1.1 christos struct edns_option* result = NULL, *cur = NULL, *s;
1240 1.1 christos while(list) {
1241 1.1 christos /* copy edns option structure */
1242 1.1 christos s = regional_alloc_init(region, list, sizeof(*list));
1243 1.1 christos if(!s) return NULL;
1244 1.1 christos s->next = NULL;
1245 1.1 christos
1246 1.1 christos /* copy option data */
1247 1.1 christos if(s->opt_data) {
1248 1.1 christos s->opt_data = regional_alloc_init(region, s->opt_data,
1249 1.1 christos s->opt_len);
1250 1.1 christos if(!s->opt_data)
1251 1.1 christos return NULL;
1252 1.1 christos }
1253 1.1 christos
1254 1.1 christos /* link into list */
1255 1.1 christos if(cur)
1256 1.1 christos cur->next = s;
1257 1.1 christos else result = s;
1258 1.1 christos cur = s;
1259 1.1 christos
1260 1.1 christos /* examine next element */
1261 1.1 christos list = list->next;
1262 1.1 christos }
1263 1.1 christos return result;
1264 1.1 christos }
1265 1.1 christos
1266 1.1.1.7.2.1 martin struct edns_option* edns_opt_copy_filter_region(struct edns_option* list,
1267 1.1.1.7.2.1 martin uint16_t* filter_list, size_t filter_list_len, struct regional* region)
1268 1.1.1.7.2.1 martin {
1269 1.1.1.7.2.1 martin struct edns_option* result = NULL, *cur = NULL, *s;
1270 1.1.1.7.2.1 martin size_t i;
1271 1.1.1.7.2.1 martin while(list) {
1272 1.1.1.7.2.1 martin for(i=0; i<filter_list_len; i++)
1273 1.1.1.7.2.1 martin if(filter_list[i] == list->opt_code) goto found;
1274 1.1.1.7.2.1 martin if(i == filter_list_len) goto next;
1275 1.1.1.7.2.1 martin found:
1276 1.1.1.7.2.1 martin /* copy edns option structure */
1277 1.1.1.7.2.1 martin s = regional_alloc_init(region, list, sizeof(*list));
1278 1.1.1.7.2.1 martin if(!s) return NULL;
1279 1.1.1.7.2.1 martin s->next = NULL;
1280 1.1.1.7.2.1 martin
1281 1.1.1.7.2.1 martin /* copy option data */
1282 1.1.1.7.2.1 martin if(s->opt_data) {
1283 1.1.1.7.2.1 martin s->opt_data = regional_alloc_init(region, s->opt_data,
1284 1.1.1.7.2.1 martin s->opt_len);
1285 1.1.1.7.2.1 martin if(!s->opt_data)
1286 1.1.1.7.2.1 martin return NULL;
1287 1.1.1.7.2.1 martin }
1288 1.1.1.7.2.1 martin
1289 1.1.1.7.2.1 martin /* link into list */
1290 1.1.1.7.2.1 martin if(cur)
1291 1.1.1.7.2.1 martin cur->next = s;
1292 1.1.1.7.2.1 martin else result = s;
1293 1.1.1.7.2.1 martin cur = s;
1294 1.1.1.7.2.1 martin
1295 1.1.1.7.2.1 martin next:
1296 1.1.1.7.2.1 martin /* examine next element */
1297 1.1.1.7.2.1 martin list = list->next;
1298 1.1.1.7.2.1 martin }
1299 1.1.1.7.2.1 martin return result;
1300 1.1.1.7.2.1 martin }
1301 1.1.1.7.2.1 martin
1302 1.1 christos int edns_opt_compare(struct edns_option* p, struct edns_option* q)
1303 1.1 christos {
1304 1.1 christos if(!p && !q) return 0;
1305 1.1 christos if(!p) return -1;
1306 1.1 christos if(!q) return 1;
1307 1.1 christos log_assert(p && q);
1308 1.1 christos if(p->opt_code != q->opt_code)
1309 1.1 christos return (int)q->opt_code - (int)p->opt_code;
1310 1.1 christos if(p->opt_len != q->opt_len)
1311 1.1 christos return (int)q->opt_len - (int)p->opt_len;
1312 1.1 christos if(p->opt_len != 0)
1313 1.1 christos return memcmp(p->opt_data, q->opt_data, p->opt_len);
1314 1.1 christos return 0;
1315 1.1 christos }
1316 1.1 christos
1317 1.1 christos int edns_opt_list_compare(struct edns_option* p, struct edns_option* q)
1318 1.1 christos {
1319 1.1 christos int r;
1320 1.1 christos while(p && q) {
1321 1.1 christos r = edns_opt_compare(p, q);
1322 1.1 christos if(r != 0)
1323 1.1 christos return r;
1324 1.1 christos p = p->next;
1325 1.1 christos q = q->next;
1326 1.1 christos }
1327 1.1 christos if(p || q) {
1328 1.1 christos /* uneven length lists */
1329 1.1 christos if(p) return 1;
1330 1.1 christos if(q) return -1;
1331 1.1 christos }
1332 1.1 christos return 0;
1333 1.1 christos }
1334 1.1 christos
1335 1.1 christos void edns_opt_list_free(struct edns_option* list)
1336 1.1 christos {
1337 1.1 christos struct edns_option* n;
1338 1.1 christos while(list) {
1339 1.1 christos free(list->opt_data);
1340 1.1 christos n = list->next;
1341 1.1 christos free(list);
1342 1.1 christos list = n;
1343 1.1 christos }
1344 1.1 christos }
1345 1.1 christos
1346 1.1 christos struct edns_option* edns_opt_copy_alloc(struct edns_option* list)
1347 1.1 christos {
1348 1.1 christos struct edns_option* result = NULL, *cur = NULL, *s;
1349 1.1 christos while(list) {
1350 1.1 christos /* copy edns option structure */
1351 1.1 christos s = memdup(list, sizeof(*list));
1352 1.1 christos if(!s) {
1353 1.1 christos edns_opt_list_free(result);
1354 1.1 christos return NULL;
1355 1.1 christos }
1356 1.1 christos s->next = NULL;
1357 1.1 christos
1358 1.1 christos /* copy option data */
1359 1.1 christos if(s->opt_data) {
1360 1.1 christos s->opt_data = memdup(s->opt_data, s->opt_len);
1361 1.1 christos if(!s->opt_data) {
1362 1.1.1.2 christos free(s);
1363 1.1 christos edns_opt_list_free(result);
1364 1.1 christos return NULL;
1365 1.1 christos }
1366 1.1 christos }
1367 1.1 christos
1368 1.1 christos /* link into list */
1369 1.1 christos if(cur)
1370 1.1 christos cur->next = s;
1371 1.1 christos else result = s;
1372 1.1 christos cur = s;
1373 1.1 christos
1374 1.1 christos /* examine next element */
1375 1.1 christos list = list->next;
1376 1.1 christos }
1377 1.1 christos return result;
1378 1.1 christos }
1379 1.1 christos
1380 1.1.1.2 christos struct edns_option* edns_opt_list_find(struct edns_option* list, uint16_t code)
1381 1.1 christos {
1382 1.1 christos struct edns_option* p;
1383 1.1 christos for(p=list; p; p=p->next) {
1384 1.1 christos if(p->opt_code == code)
1385 1.1 christos return p;
1386 1.1 christos }
1387 1.1 christos return NULL;
1388 1.1 christos }
1389