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