1b7579f77SDag-Erling Smørgrav /* 2b7579f77SDag-Erling Smørgrav * util/data/msgreply.c - store message and reply data. 3b7579f77SDag-Erling Smørgrav * 4b7579f77SDag-Erling Smørgrav * Copyright (c) 2007, NLnet Labs. All rights reserved. 5b7579f77SDag-Erling Smørgrav * 6b7579f77SDag-Erling Smørgrav * This software is open source. 7b7579f77SDag-Erling Smørgrav * 8b7579f77SDag-Erling Smørgrav * Redistribution and use in source and binary forms, with or without 9b7579f77SDag-Erling Smørgrav * modification, are permitted provided that the following conditions 10b7579f77SDag-Erling Smørgrav * are met: 11b7579f77SDag-Erling Smørgrav * 12b7579f77SDag-Erling Smørgrav * Redistributions of source code must retain the above copyright notice, 13b7579f77SDag-Erling Smørgrav * this list of conditions and the following disclaimer. 14b7579f77SDag-Erling Smørgrav * 15b7579f77SDag-Erling Smørgrav * Redistributions in binary form must reproduce the above copyright notice, 16b7579f77SDag-Erling Smørgrav * this list of conditions and the following disclaimer in the documentation 17b7579f77SDag-Erling Smørgrav * and/or other materials provided with the distribution. 18b7579f77SDag-Erling Smørgrav * 19b7579f77SDag-Erling Smørgrav * Neither the name of the NLNET LABS nor the names of its contributors may 20b7579f77SDag-Erling Smørgrav * be used to endorse or promote products derived from this software without 21b7579f77SDag-Erling Smørgrav * specific prior written permission. 22b7579f77SDag-Erling Smørgrav * 23b7579f77SDag-Erling Smørgrav * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2417d15b25SDag-Erling Smørgrav * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 2517d15b25SDag-Erling Smørgrav * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 2617d15b25SDag-Erling Smørgrav * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 2717d15b25SDag-Erling Smørgrav * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 2817d15b25SDag-Erling Smørgrav * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 2917d15b25SDag-Erling Smørgrav * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 3017d15b25SDag-Erling Smørgrav * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 3117d15b25SDag-Erling Smørgrav * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 3217d15b25SDag-Erling Smørgrav * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 3317d15b25SDag-Erling Smørgrav * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 34b7579f77SDag-Erling Smørgrav */ 35b7579f77SDag-Erling Smørgrav 36b7579f77SDag-Erling Smørgrav /** 37b7579f77SDag-Erling Smørgrav * \file 38b7579f77SDag-Erling Smørgrav * 39b7579f77SDag-Erling Smørgrav * This file contains a data structure to store a message and its reply. 40b7579f77SDag-Erling Smørgrav */ 41b7579f77SDag-Erling Smørgrav 42b7579f77SDag-Erling Smørgrav #include "config.h" 43b7579f77SDag-Erling Smørgrav #include "util/data/msgreply.h" 44b7579f77SDag-Erling Smørgrav #include "util/storage/lookup3.h" 45b7579f77SDag-Erling Smørgrav #include "util/log.h" 46b7579f77SDag-Erling Smørgrav #include "util/alloc.h" 47b7579f77SDag-Erling Smørgrav #include "util/netevent.h" 48b7579f77SDag-Erling Smørgrav #include "util/net_help.h" 49b7579f77SDag-Erling Smørgrav #include "util/data/dname.h" 50b7579f77SDag-Erling Smørgrav #include "util/regional.h" 51b7579f77SDag-Erling Smørgrav #include "util/data/msgparse.h" 52b7579f77SDag-Erling Smørgrav #include "util/data/msgencode.h" 5309a3aaf3SDag-Erling Smørgrav #include "sldns/sbuffer.h" 5409a3aaf3SDag-Erling Smørgrav #include "sldns/wire2str.h" 55bc892140SDag-Erling Smørgrav #include "util/module.h" 56bc892140SDag-Erling Smørgrav #include "util/fptr_wlist.h" 57b7579f77SDag-Erling Smørgrav 58b7579f77SDag-Erling Smørgrav /** MAX TTL default for messages and rrsets */ 5917d15b25SDag-Erling Smørgrav time_t MAX_TTL = 3600 * 24 * 10; /* ten days */ 60b7579f77SDag-Erling Smørgrav /** MIN TTL default for messages and rrsets */ 6117d15b25SDag-Erling Smørgrav time_t MIN_TTL = 0; 6209a3aaf3SDag-Erling Smørgrav /** MAX Negative TTL, for SOA records in authority section */ 6309a3aaf3SDag-Erling Smørgrav time_t MAX_NEG_TTL = 3600; /* one hour */ 64335c7cdaSCy Schubert /** MIN Negative TTL, for SOA records in authority section */ 65335c7cdaSCy Schubert time_t MIN_NEG_TTL = 0; 66091e9e46SCy Schubert /** If we serve expired entries and prefetch them */ 67091e9e46SCy Schubert int SERVE_EXPIRED = 0; 684c75e3aaSDag-Erling Smørgrav /** Time to serve records after expiration */ 69*be771a7bSCy Schubert time_t SERVE_EXPIRED_TTL = 86400; 7046d2f618SCy Schubert /** Reset serve expired TTL after failed update attempt */ 7146d2f618SCy Schubert time_t SERVE_EXPIRED_TTL_RESET = 0; 72091e9e46SCy Schubert /** TTL to use for expired records */ 73091e9e46SCy Schubert time_t SERVE_EXPIRED_REPLY_TTL = 30; 74f44e67d1SCy Schubert /** If we serve the original TTL or decrementing TTLs */ 75f44e67d1SCy Schubert int SERVE_ORIGINAL_TTL = 0; 76b7579f77SDag-Erling Smørgrav 77b7579f77SDag-Erling Smørgrav /** allocate qinfo, return 0 on error */ 78b7579f77SDag-Erling Smørgrav static int 7917d15b25SDag-Erling Smørgrav parse_create_qinfo(sldns_buffer* pkt, struct msg_parse* msg, 80b7579f77SDag-Erling Smørgrav struct query_info* qinf, struct regional* region) 81b7579f77SDag-Erling Smørgrav { 82b7579f77SDag-Erling Smørgrav if(msg->qname) { 83b7579f77SDag-Erling Smørgrav if(region) 84b7579f77SDag-Erling Smørgrav qinf->qname = (uint8_t*)regional_alloc(region, 85b7579f77SDag-Erling Smørgrav msg->qname_len); 86b7579f77SDag-Erling Smørgrav else qinf->qname = (uint8_t*)malloc(msg->qname_len); 87b7579f77SDag-Erling Smørgrav if(!qinf->qname) return 0; 88b7579f77SDag-Erling Smørgrav dname_pkt_copy(pkt, qinf->qname, msg->qname); 89b7579f77SDag-Erling Smørgrav } else qinf->qname = 0; 90b7579f77SDag-Erling Smørgrav qinf->qname_len = msg->qname_len; 91b7579f77SDag-Erling Smørgrav qinf->qtype = msg->qtype; 92b7579f77SDag-Erling Smørgrav qinf->qclass = msg->qclass; 93bc892140SDag-Erling Smørgrav qinf->local_alias = NULL; 94b7579f77SDag-Erling Smørgrav return 1; 95b7579f77SDag-Erling Smørgrav } 96b7579f77SDag-Erling Smørgrav 97b7579f77SDag-Erling Smørgrav /** constructor for replyinfo */ 98ff825849SDag-Erling Smørgrav struct reply_info* 99b7579f77SDag-Erling Smørgrav construct_reply_info_base(struct regional* region, uint16_t flags, size_t qd, 10046d2f618SCy Schubert time_t ttl, time_t prettl, time_t expttl, time_t norecttl, size_t an, 10146d2f618SCy Schubert size_t ns, size_t ar, size_t total, enum sec_status sec, 10246d2f618SCy Schubert sldns_ede_code reason_bogus) 103b7579f77SDag-Erling Smørgrav { 104b7579f77SDag-Erling Smørgrav struct reply_info* rep; 105b7579f77SDag-Erling Smørgrav /* rrset_count-1 because the first ref is part of the struct. */ 106b7579f77SDag-Erling Smørgrav size_t s = sizeof(struct reply_info) - sizeof(struct rrset_ref) + 107b7579f77SDag-Erling Smørgrav sizeof(struct ub_packed_rrset_key*) * total; 10809a3aaf3SDag-Erling Smørgrav if(total >= RR_COUNT_MAX) return NULL; /* sanity check on numRRS*/ 109b7579f77SDag-Erling Smørgrav if(region) 110b7579f77SDag-Erling Smørgrav rep = (struct reply_info*)regional_alloc(region, s); 111b7579f77SDag-Erling Smørgrav else rep = (struct reply_info*)malloc(s + 112b7579f77SDag-Erling Smørgrav sizeof(struct rrset_ref) * (total)); 113b7579f77SDag-Erling Smørgrav if(!rep) 114b7579f77SDag-Erling Smørgrav return NULL; 115b7579f77SDag-Erling Smørgrav rep->flags = flags; 116b7579f77SDag-Erling Smørgrav rep->qdcount = qd; 117b7579f77SDag-Erling Smørgrav rep->ttl = ttl; 118b7579f77SDag-Erling Smørgrav rep->prefetch_ttl = prettl; 1194c75e3aaSDag-Erling Smørgrav rep->serve_expired_ttl = expttl; 12046d2f618SCy Schubert rep->serve_expired_norec_ttl = norecttl; 121b7579f77SDag-Erling Smørgrav rep->an_numrrsets = an; 122b7579f77SDag-Erling Smørgrav rep->ns_numrrsets = ns; 123b7579f77SDag-Erling Smørgrav rep->ar_numrrsets = ar; 124b7579f77SDag-Erling Smørgrav rep->rrset_count = total; 125b7579f77SDag-Erling Smørgrav rep->security = sec; 1268f76bb7dSCy Schubert rep->reason_bogus = reason_bogus; 1278f76bb7dSCy Schubert /* this is only allocated and used for caching on copy */ 1288f76bb7dSCy Schubert rep->reason_bogus_str = NULL; 129b7579f77SDag-Erling Smørgrav rep->authoritative = 0; 130b7579f77SDag-Erling Smørgrav /* array starts after the refs */ 131b7579f77SDag-Erling Smørgrav if(region) 132b7579f77SDag-Erling Smørgrav rep->rrsets = (struct ub_packed_rrset_key**)&(rep->ref[0]); 133b7579f77SDag-Erling Smørgrav else rep->rrsets = (struct ub_packed_rrset_key**)&(rep->ref[total]); 134b7579f77SDag-Erling Smørgrav /* zero the arrays to assist cleanup in case of malloc failure */ 135b7579f77SDag-Erling Smørgrav memset( rep->rrsets, 0, sizeof(struct ub_packed_rrset_key*) * total); 136b7579f77SDag-Erling Smørgrav if(!region) 137b7579f77SDag-Erling Smørgrav memset( &rep->ref[0], 0, sizeof(struct rrset_ref) * total); 138b7579f77SDag-Erling Smørgrav return rep; 139b7579f77SDag-Erling Smørgrav } 140b7579f77SDag-Erling Smørgrav 141b7579f77SDag-Erling Smørgrav /** allocate replyinfo, return 0 on error */ 142b7579f77SDag-Erling Smørgrav static int 143b7579f77SDag-Erling Smørgrav parse_create_repinfo(struct msg_parse* msg, struct reply_info** rep, 144b7579f77SDag-Erling Smørgrav struct regional* region) 145b7579f77SDag-Erling Smørgrav { 146b7579f77SDag-Erling Smørgrav *rep = construct_reply_info_base(region, msg->flags, msg->qdcount, 0, 14746d2f618SCy Schubert 0, 0, 0, msg->an_rrsets, msg->ns_rrsets, msg->ar_rrsets, 1488f76bb7dSCy Schubert msg->rrset_count, sec_status_unchecked, LDNS_EDE_NONE); 149b7579f77SDag-Erling Smørgrav if(!*rep) 150b7579f77SDag-Erling Smørgrav return 0; 151b7579f77SDag-Erling Smørgrav return 1; 152b7579f77SDag-Erling Smørgrav } 153b7579f77SDag-Erling Smørgrav 15465b390aaSDag-Erling Smørgrav int 15565b390aaSDag-Erling Smørgrav reply_info_alloc_rrset_keys(struct reply_info* rep, struct alloc_cache* alloc, 156b7579f77SDag-Erling Smørgrav struct regional* region) 157b7579f77SDag-Erling Smørgrav { 158b7579f77SDag-Erling Smørgrav size_t i; 159b7579f77SDag-Erling Smørgrav for(i=0; i<rep->rrset_count; i++) { 160b7579f77SDag-Erling Smørgrav if(region) { 161b7579f77SDag-Erling Smørgrav rep->rrsets[i] = (struct ub_packed_rrset_key*) 162b7579f77SDag-Erling Smørgrav regional_alloc(region, 163b7579f77SDag-Erling Smørgrav sizeof(struct ub_packed_rrset_key)); 164b7579f77SDag-Erling Smørgrav if(rep->rrsets[i]) { 165b7579f77SDag-Erling Smørgrav memset(rep->rrsets[i], 0, 166b7579f77SDag-Erling Smørgrav sizeof(struct ub_packed_rrset_key)); 167b7579f77SDag-Erling Smørgrav rep->rrsets[i]->entry.key = rep->rrsets[i]; 168b7579f77SDag-Erling Smørgrav } 169b7579f77SDag-Erling Smørgrav } 170b7579f77SDag-Erling Smørgrav else rep->rrsets[i] = alloc_special_obtain(alloc); 171b7579f77SDag-Erling Smørgrav if(!rep->rrsets[i]) 172b7579f77SDag-Erling Smørgrav return 0; 173b7579f77SDag-Erling Smørgrav rep->rrsets[i]->entry.data = NULL; 174b7579f77SDag-Erling Smørgrav } 175b7579f77SDag-Erling Smørgrav return 1; 176b7579f77SDag-Erling Smørgrav } 177b7579f77SDag-Erling Smørgrav 17846d2f618SCy Schubert int 17946d2f618SCy Schubert reply_info_can_answer_expired(struct reply_info* rep, time_t timenow) 18046d2f618SCy Schubert { 18146d2f618SCy Schubert log_assert(rep->ttl < timenow); 18246d2f618SCy Schubert /* Really expired */ 18346d2f618SCy Schubert if(SERVE_EXPIRED_TTL && rep->serve_expired_ttl < timenow) return 0; 18446d2f618SCy Schubert /* Ignore expired failure answers */ 18546d2f618SCy Schubert if(FLAGS_GET_RCODE(rep->flags) != LDNS_RCODE_NOERROR && 18646d2f618SCy Schubert FLAGS_GET_RCODE(rep->flags) != LDNS_RCODE_NXDOMAIN && 18746d2f618SCy Schubert FLAGS_GET_RCODE(rep->flags) != LDNS_RCODE_YXDOMAIN) return 0; 18846d2f618SCy Schubert return 1; 18946d2f618SCy Schubert } 19046d2f618SCy Schubert 19146d2f618SCy Schubert int reply_info_could_use_expired(struct reply_info* rep, time_t timenow) 19246d2f618SCy Schubert { 19346d2f618SCy Schubert log_assert(rep->ttl < timenow); 19446d2f618SCy Schubert /* Really expired */ 19546d2f618SCy Schubert if(SERVE_EXPIRED_TTL && rep->serve_expired_ttl < timenow && 19646d2f618SCy Schubert !SERVE_EXPIRED_TTL_RESET) return 0; 19746d2f618SCy Schubert /* Ignore expired failure answers */ 19846d2f618SCy Schubert if(FLAGS_GET_RCODE(rep->flags) != LDNS_RCODE_NOERROR && 19946d2f618SCy Schubert FLAGS_GET_RCODE(rep->flags) != LDNS_RCODE_NXDOMAIN && 20046d2f618SCy Schubert FLAGS_GET_RCODE(rep->flags) != LDNS_RCODE_YXDOMAIN) return 0; 20146d2f618SCy Schubert return 1; 20246d2f618SCy Schubert } 20346d2f618SCy Schubert 20424e36522SCy Schubert struct reply_info * 20524e36522SCy Schubert make_new_reply_info(const struct reply_info* rep, struct regional* region, 20624e36522SCy Schubert size_t an_numrrsets, size_t copy_rrsets) 20724e36522SCy Schubert { 20824e36522SCy Schubert struct reply_info* new_rep; 20924e36522SCy Schubert size_t i; 21024e36522SCy Schubert 21124e36522SCy Schubert /* create a base struct. we specify 'insecure' security status as 21224e36522SCy Schubert * the modified response won't be DNSSEC-valid. In our faked response 21324e36522SCy Schubert * the authority and additional sections will be empty (except possible 21424e36522SCy Schubert * EDNS0 OPT RR in the additional section appended on sending it out), 21524e36522SCy Schubert * so the total number of RRsets is an_numrrsets. */ 21624e36522SCy Schubert new_rep = construct_reply_info_base(region, rep->flags, 21724e36522SCy Schubert rep->qdcount, rep->ttl, rep->prefetch_ttl, 21846d2f618SCy Schubert rep->serve_expired_ttl, rep->serve_expired_norec_ttl, 21946d2f618SCy Schubert an_numrrsets, 0, 0, an_numrrsets, 2208f76bb7dSCy Schubert sec_status_insecure, LDNS_EDE_NONE); 22124e36522SCy Schubert if(!new_rep) 22224e36522SCy Schubert return NULL; 22324e36522SCy Schubert if(!reply_info_alloc_rrset_keys(new_rep, NULL, region)) 22424e36522SCy Schubert return NULL; 22524e36522SCy Schubert for(i=0; i<copy_rrsets; i++) 22624e36522SCy Schubert new_rep->rrsets[i] = rep->rrsets[i]; 22724e36522SCy Schubert 22824e36522SCy Schubert return new_rep; 22924e36522SCy Schubert } 23024e36522SCy Schubert 23109a3aaf3SDag-Erling Smørgrav /** find the minimumttl in the rdata of SOA record */ 23209a3aaf3SDag-Erling Smørgrav static time_t 23309a3aaf3SDag-Erling Smørgrav soa_find_minttl(struct rr_parse* rr) 23409a3aaf3SDag-Erling Smørgrav { 23509a3aaf3SDag-Erling Smørgrav uint16_t rlen = sldns_read_uint16(rr->ttl_data+4); 23609a3aaf3SDag-Erling Smørgrav if(rlen < 20) 23709a3aaf3SDag-Erling Smørgrav return 0; /* rdata too small for SOA (dname, dname, 5*32bit) */ 23809a3aaf3SDag-Erling Smørgrav /* minimum TTL is the last 32bit value in the rdata of the record */ 23909a3aaf3SDag-Erling Smørgrav /* at position ttl_data + 4(ttl) + 2(rdatalen) + rdatalen - 4(timeval)*/ 24009a3aaf3SDag-Erling Smørgrav return (time_t)sldns_read_uint32(rr->ttl_data+6+rlen-4); 24109a3aaf3SDag-Erling Smørgrav } 24209a3aaf3SDag-Erling Smørgrav 243b7579f77SDag-Erling Smørgrav /** do the rdata copy */ 244b7579f77SDag-Erling Smørgrav static int 24517d15b25SDag-Erling Smørgrav rdata_copy(sldns_buffer* pkt, struct packed_rrset_data* data, uint8_t* to, 24609a3aaf3SDag-Erling Smørgrav struct rr_parse* rr, time_t* rr_ttl, uint16_t type, 24709a3aaf3SDag-Erling Smørgrav sldns_pkt_section section) 248b7579f77SDag-Erling Smørgrav { 249b7579f77SDag-Erling Smørgrav uint16_t pkt_len; 25017d15b25SDag-Erling Smørgrav const sldns_rr_descriptor* desc; 251b7579f77SDag-Erling Smørgrav 25217d15b25SDag-Erling Smørgrav *rr_ttl = sldns_read_uint32(rr->ttl_data); 253b7579f77SDag-Erling Smørgrav /* RFC 2181 Section 8. if msb of ttl is set treat as if zero. */ 254b7579f77SDag-Erling Smørgrav if(*rr_ttl & 0x80000000U) 255b7579f77SDag-Erling Smørgrav *rr_ttl = 0; 25609a3aaf3SDag-Erling Smørgrav if(type == LDNS_RR_TYPE_SOA && section == LDNS_SECTION_AUTHORITY) { 25709a3aaf3SDag-Erling Smørgrav /* negative response. see if TTL of SOA record larger than the 25809a3aaf3SDag-Erling Smørgrav * minimum-ttl in the rdata of the SOA record */ 259335c7cdaSCy Schubert if(*rr_ttl > soa_find_minttl(rr)) *rr_ttl = soa_find_minttl(rr); 260335c7cdaSCy Schubert if(!SERVE_ORIGINAL_TTL) { 261335c7cdaSCy Schubert /* If MIN_NEG_TTL is configured skip setting MIN_TTL */ 262335c7cdaSCy Schubert if(MIN_NEG_TTL <= 0 && *rr_ttl < MIN_TTL) { 263b7579f77SDag-Erling Smørgrav *rr_ttl = MIN_TTL; 264335c7cdaSCy Schubert } 265335c7cdaSCy Schubert if(*rr_ttl > MAX_TTL) *rr_ttl = MAX_TTL; 266335c7cdaSCy Schubert } 267335c7cdaSCy Schubert /* MAX_NEG_TTL overrides the min and max ttl of everything 268335c7cdaSCy Schubert * else; it is for a more specific record */ 269335c7cdaSCy Schubert if(*rr_ttl > MAX_NEG_TTL) *rr_ttl = MAX_NEG_TTL; 270335c7cdaSCy Schubert /* MIN_NEG_TTL overrides the min and max ttl of everything 271335c7cdaSCy Schubert * else if configured; it is for a more specific record */ 272335c7cdaSCy Schubert if(MIN_NEG_TTL > 0 && *rr_ttl < MIN_NEG_TTL) { 273335c7cdaSCy Schubert *rr_ttl = MIN_NEG_TTL; 274335c7cdaSCy Schubert } 275335c7cdaSCy Schubert } else if(!SERVE_ORIGINAL_TTL) { 276335c7cdaSCy Schubert if(*rr_ttl < MIN_TTL) *rr_ttl = MIN_TTL; 277335c7cdaSCy Schubert if(*rr_ttl > MAX_TTL) *rr_ttl = MAX_TTL; 27824e36522SCy Schubert } 279b7579f77SDag-Erling Smørgrav if(*rr_ttl < data->ttl) 280b7579f77SDag-Erling Smørgrav data->ttl = *rr_ttl; 281b7579f77SDag-Erling Smørgrav 282b7579f77SDag-Erling Smørgrav if(rr->outside_packet) { 283b7579f77SDag-Erling Smørgrav /* uncompressed already, only needs copy */ 284b7579f77SDag-Erling Smørgrav memmove(to, rr->ttl_data+sizeof(uint32_t), rr->size); 285b7579f77SDag-Erling Smørgrav return 1; 286b7579f77SDag-Erling Smørgrav } 287b7579f77SDag-Erling Smørgrav 28817d15b25SDag-Erling Smørgrav sldns_buffer_set_position(pkt, (size_t) 28917d15b25SDag-Erling Smørgrav (rr->ttl_data - sldns_buffer_begin(pkt) + sizeof(uint32_t))); 290b7579f77SDag-Erling Smørgrav /* insert decompressed size into rdata len stored in memory */ 291b7579f77SDag-Erling Smørgrav /* -2 because rdatalen bytes are not included. */ 292b7579f77SDag-Erling Smørgrav pkt_len = htons(rr->size - 2); 293b7579f77SDag-Erling Smørgrav memmove(to, &pkt_len, sizeof(uint16_t)); 294b7579f77SDag-Erling Smørgrav to += 2; 295b7579f77SDag-Erling Smørgrav /* read packet rdata len */ 29617d15b25SDag-Erling Smørgrav pkt_len = sldns_buffer_read_u16(pkt); 29717d15b25SDag-Erling Smørgrav if(sldns_buffer_remaining(pkt) < pkt_len) 298b7579f77SDag-Erling Smørgrav return 0; 29917d15b25SDag-Erling Smørgrav desc = sldns_rr_descript(type); 300b7579f77SDag-Erling Smørgrav if(pkt_len > 0 && desc && desc->_dname_count > 0) { 301b7579f77SDag-Erling Smørgrav int count = (int)desc->_dname_count; 302b7579f77SDag-Erling Smørgrav int rdf = 0; 303b7579f77SDag-Erling Smørgrav size_t len; 304b7579f77SDag-Erling Smørgrav size_t oldpos; 305b7579f77SDag-Erling Smørgrav /* decompress dnames. */ 306b7579f77SDag-Erling Smørgrav while(pkt_len > 0 && count) { 307b7579f77SDag-Erling Smørgrav switch(desc->_wireformat[rdf]) { 308b7579f77SDag-Erling Smørgrav case LDNS_RDF_TYPE_DNAME: 30917d15b25SDag-Erling Smørgrav oldpos = sldns_buffer_position(pkt); 310b7579f77SDag-Erling Smørgrav dname_pkt_copy(pkt, to, 31117d15b25SDag-Erling Smørgrav sldns_buffer_current(pkt)); 312b7579f77SDag-Erling Smørgrav to += pkt_dname_len(pkt); 31317d15b25SDag-Erling Smørgrav pkt_len -= sldns_buffer_position(pkt)-oldpos; 314b7579f77SDag-Erling Smørgrav count--; 315b7579f77SDag-Erling Smørgrav len = 0; 316b7579f77SDag-Erling Smørgrav break; 317b7579f77SDag-Erling Smørgrav case LDNS_RDF_TYPE_STR: 31817d15b25SDag-Erling Smørgrav len = sldns_buffer_current(pkt)[0] + 1; 319b7579f77SDag-Erling Smørgrav break; 320b7579f77SDag-Erling Smørgrav default: 321b7579f77SDag-Erling Smørgrav len = get_rdf_size(desc->_wireformat[rdf]); 322b7579f77SDag-Erling Smørgrav break; 323b7579f77SDag-Erling Smørgrav } 324b7579f77SDag-Erling Smørgrav if(len) { 3250eefd307SCy Schubert log_assert(len <= pkt_len); 32617d15b25SDag-Erling Smørgrav memmove(to, sldns_buffer_current(pkt), len); 327b7579f77SDag-Erling Smørgrav to += len; 32817d15b25SDag-Erling Smørgrav sldns_buffer_skip(pkt, (ssize_t)len); 329b7579f77SDag-Erling Smørgrav pkt_len -= len; 330b7579f77SDag-Erling Smørgrav } 331b7579f77SDag-Erling Smørgrav rdf++; 332b7579f77SDag-Erling Smørgrav } 333b7579f77SDag-Erling Smørgrav } 334b7579f77SDag-Erling Smørgrav /* copy remaining rdata */ 335b7579f77SDag-Erling Smørgrav if(pkt_len > 0) 33617d15b25SDag-Erling Smørgrav memmove(to, sldns_buffer_current(pkt), pkt_len); 337b7579f77SDag-Erling Smørgrav 338b7579f77SDag-Erling Smørgrav return 1; 339b7579f77SDag-Erling Smørgrav } 340b7579f77SDag-Erling Smørgrav 341b7579f77SDag-Erling Smørgrav /** copy over the data into packed rrset */ 342b7579f77SDag-Erling Smørgrav static int 34317d15b25SDag-Erling Smørgrav parse_rr_copy(sldns_buffer* pkt, struct rrset_parse* pset, 344b7579f77SDag-Erling Smørgrav struct packed_rrset_data* data) 345b7579f77SDag-Erling Smørgrav { 346b7579f77SDag-Erling Smørgrav size_t i; 347b7579f77SDag-Erling Smørgrav struct rr_parse* rr = pset->rr_first; 348b7579f77SDag-Erling Smørgrav uint8_t* nextrdata; 349b7579f77SDag-Erling Smørgrav size_t total = pset->rr_count + pset->rrsig_count; 350b7579f77SDag-Erling Smørgrav data->ttl = MAX_TTL; 351b7579f77SDag-Erling Smørgrav data->count = pset->rr_count; 352b7579f77SDag-Erling Smørgrav data->rrsig_count = pset->rrsig_count; 353b7579f77SDag-Erling Smørgrav data->trust = rrset_trust_none; 354b7579f77SDag-Erling Smørgrav data->security = sec_status_unchecked; 355b7579f77SDag-Erling Smørgrav /* layout: struct - rr_len - rr_data - rr_ttl - rdata - rrsig */ 356b7579f77SDag-Erling Smørgrav data->rr_len = (size_t*)((uint8_t*)data + 357b7579f77SDag-Erling Smørgrav sizeof(struct packed_rrset_data)); 358b7579f77SDag-Erling Smørgrav data->rr_data = (uint8_t**)&(data->rr_len[total]); 35917d15b25SDag-Erling Smørgrav data->rr_ttl = (time_t*)&(data->rr_data[total]); 360b7579f77SDag-Erling Smørgrav nextrdata = (uint8_t*)&(data->rr_ttl[total]); 361b7579f77SDag-Erling Smørgrav for(i=0; i<data->count; i++) { 362b7579f77SDag-Erling Smørgrav data->rr_len[i] = rr->size; 363b7579f77SDag-Erling Smørgrav data->rr_data[i] = nextrdata; 364b7579f77SDag-Erling Smørgrav nextrdata += rr->size; 365b7579f77SDag-Erling Smørgrav if(!rdata_copy(pkt, data, data->rr_data[i], rr, 36609a3aaf3SDag-Erling Smørgrav &data->rr_ttl[i], pset->type, pset->section)) 367b7579f77SDag-Erling Smørgrav return 0; 368b7579f77SDag-Erling Smørgrav rr = rr->next; 369b7579f77SDag-Erling Smørgrav } 370b7579f77SDag-Erling Smørgrav /* if rrsig, its rdata is at nextrdata */ 371b7579f77SDag-Erling Smørgrav rr = pset->rrsig_first; 372b7579f77SDag-Erling Smørgrav for(i=data->count; i<total; i++) { 373b7579f77SDag-Erling Smørgrav data->rr_len[i] = rr->size; 374b7579f77SDag-Erling Smørgrav data->rr_data[i] = nextrdata; 375b7579f77SDag-Erling Smørgrav nextrdata += rr->size; 376b7579f77SDag-Erling Smørgrav if(!rdata_copy(pkt, data, data->rr_data[i], rr, 37709a3aaf3SDag-Erling Smørgrav &data->rr_ttl[i], LDNS_RR_TYPE_RRSIG, pset->section)) 378b7579f77SDag-Erling Smørgrav return 0; 379b7579f77SDag-Erling Smørgrav rr = rr->next; 380b7579f77SDag-Erling Smørgrav } 381b7579f77SDag-Erling Smørgrav return 1; 382b7579f77SDag-Erling Smørgrav } 383b7579f77SDag-Erling Smørgrav 384b7579f77SDag-Erling Smørgrav /** create rrset return 0 on failure */ 385b7579f77SDag-Erling Smørgrav static int 38617d15b25SDag-Erling Smørgrav parse_create_rrset(sldns_buffer* pkt, struct rrset_parse* pset, 387b7579f77SDag-Erling Smørgrav struct packed_rrset_data** data, struct regional* region) 388b7579f77SDag-Erling Smørgrav { 389b7579f77SDag-Erling Smørgrav /* allocate */ 39009a3aaf3SDag-Erling Smørgrav size_t s; 39109a3aaf3SDag-Erling Smørgrav if(pset->rr_count > RR_COUNT_MAX || pset->rrsig_count > RR_COUNT_MAX || 39209a3aaf3SDag-Erling Smørgrav pset->size > RR_COUNT_MAX) 39309a3aaf3SDag-Erling Smørgrav return 0; /* protect against integer overflow */ 39409a3aaf3SDag-Erling Smørgrav s = sizeof(struct packed_rrset_data) + 395b7579f77SDag-Erling Smørgrav (pset->rr_count + pset->rrsig_count) * 39617d15b25SDag-Erling Smørgrav (sizeof(size_t)+sizeof(uint8_t*)+sizeof(time_t)) + 397b7579f77SDag-Erling Smørgrav pset->size; 398b7579f77SDag-Erling Smørgrav if(region) 399f44e67d1SCy Schubert *data = regional_alloc_zero(region, s); 400f44e67d1SCy Schubert else *data = calloc(1, s); 401b7579f77SDag-Erling Smørgrav if(!*data) 402b7579f77SDag-Erling Smørgrav return 0; 403b7579f77SDag-Erling Smørgrav /* copy & decompress */ 404b7579f77SDag-Erling Smørgrav if(!parse_rr_copy(pkt, pset, *data)) { 4055469a995SCy Schubert if(!region) { 4065469a995SCy Schubert free(*data); 4075469a995SCy Schubert *data = NULL; 4085469a995SCy Schubert } 409b7579f77SDag-Erling Smørgrav return 0; 410b7579f77SDag-Erling Smørgrav } 411b7579f77SDag-Erling Smørgrav return 1; 412b7579f77SDag-Erling Smørgrav } 413b7579f77SDag-Erling Smørgrav 414b7579f77SDag-Erling Smørgrav /** get trust value for rrset */ 415b7579f77SDag-Erling Smørgrav static enum rrset_trust 416b7579f77SDag-Erling Smørgrav get_rrset_trust(struct msg_parse* msg, struct rrset_parse* rrset) 417b7579f77SDag-Erling Smørgrav { 418b7579f77SDag-Erling Smørgrav uint16_t AA = msg->flags & BIT_AA; 419b7579f77SDag-Erling Smørgrav if(rrset->section == LDNS_SECTION_ANSWER) { 420b7579f77SDag-Erling Smørgrav if(AA) { 421b7579f77SDag-Erling Smørgrav /* RFC2181 says remainder of CNAME chain is nonauth*/ 422b7579f77SDag-Erling Smørgrav if(msg->rrset_first && 423b7579f77SDag-Erling Smørgrav msg->rrset_first->section==LDNS_SECTION_ANSWER 424b7579f77SDag-Erling Smørgrav && msg->rrset_first->type==LDNS_RR_TYPE_CNAME){ 425b7579f77SDag-Erling Smørgrav if(rrset == msg->rrset_first) 426b7579f77SDag-Erling Smørgrav return rrset_trust_ans_AA; 427b7579f77SDag-Erling Smørgrav else return rrset_trust_ans_noAA; 428b7579f77SDag-Erling Smørgrav } 429b7579f77SDag-Erling Smørgrav if(msg->rrset_first && 430b7579f77SDag-Erling Smørgrav msg->rrset_first->section==LDNS_SECTION_ANSWER 431b7579f77SDag-Erling Smørgrav && msg->rrset_first->type==LDNS_RR_TYPE_DNAME){ 432b7579f77SDag-Erling Smørgrav if(rrset == msg->rrset_first || 433b7579f77SDag-Erling Smørgrav rrset == msg->rrset_first->rrset_all_next) 434b7579f77SDag-Erling Smørgrav return rrset_trust_ans_AA; 435b7579f77SDag-Erling Smørgrav else return rrset_trust_ans_noAA; 436b7579f77SDag-Erling Smørgrav } 437b7579f77SDag-Erling Smørgrav return rrset_trust_ans_AA; 438b7579f77SDag-Erling Smørgrav } 439b7579f77SDag-Erling Smørgrav else return rrset_trust_ans_noAA; 440b7579f77SDag-Erling Smørgrav } else if(rrset->section == LDNS_SECTION_AUTHORITY) { 441b7579f77SDag-Erling Smørgrav if(AA) return rrset_trust_auth_AA; 442b7579f77SDag-Erling Smørgrav else return rrset_trust_auth_noAA; 443b7579f77SDag-Erling Smørgrav } else { 444b7579f77SDag-Erling Smørgrav /* addit section */ 445b7579f77SDag-Erling Smørgrav if(AA) return rrset_trust_add_AA; 446b7579f77SDag-Erling Smørgrav else return rrset_trust_add_noAA; 447b7579f77SDag-Erling Smørgrav } 448b7579f77SDag-Erling Smørgrav /* NOTREACHED */ 449b7579f77SDag-Erling Smørgrav return rrset_trust_none; 450b7579f77SDag-Erling Smørgrav } 451b7579f77SDag-Erling Smørgrav 452b7579f77SDag-Erling Smørgrav int 45317d15b25SDag-Erling Smørgrav parse_copy_decompress_rrset(sldns_buffer* pkt, struct msg_parse* msg, 454b7579f77SDag-Erling Smørgrav struct rrset_parse *pset, struct regional* region, 455b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* pk) 456b7579f77SDag-Erling Smørgrav { 457b7579f77SDag-Erling Smørgrav struct packed_rrset_data* data; 458b7579f77SDag-Erling Smørgrav pk->rk.flags = pset->flags; 459b7579f77SDag-Erling Smørgrav pk->rk.dname_len = pset->dname_len; 460b7579f77SDag-Erling Smørgrav if(region) 461b7579f77SDag-Erling Smørgrav pk->rk.dname = (uint8_t*)regional_alloc( 462b7579f77SDag-Erling Smørgrav region, pset->dname_len); 463b7579f77SDag-Erling Smørgrav else pk->rk.dname = 464b7579f77SDag-Erling Smørgrav (uint8_t*)malloc(pset->dname_len); 465b7579f77SDag-Erling Smørgrav if(!pk->rk.dname) 466b7579f77SDag-Erling Smørgrav return 0; 467b7579f77SDag-Erling Smørgrav /** copy & decompress dname */ 468b7579f77SDag-Erling Smørgrav dname_pkt_copy(pkt, pk->rk.dname, pset->dname); 469b7579f77SDag-Erling Smørgrav /** copy over type and class */ 470b7579f77SDag-Erling Smørgrav pk->rk.type = htons(pset->type); 471b7579f77SDag-Erling Smørgrav pk->rk.rrset_class = pset->rrset_class; 472b7579f77SDag-Erling Smørgrav /** read data part. */ 4735469a995SCy Schubert if(!parse_create_rrset(pkt, pset, &data, region)) { 4745469a995SCy Schubert if(!region) { 4755469a995SCy Schubert free(pk->rk.dname); 4765469a995SCy Schubert pk->rk.dname = NULL; 4775469a995SCy Schubert } 478b7579f77SDag-Erling Smørgrav return 0; 4795469a995SCy Schubert } 480b7579f77SDag-Erling Smørgrav pk->entry.data = (void*)data; 481b7579f77SDag-Erling Smørgrav pk->entry.key = (void*)pk; 482b7579f77SDag-Erling Smørgrav pk->entry.hash = pset->hash; 483b7579f77SDag-Erling Smørgrav data->trust = get_rrset_trust(msg, pset); 484b7579f77SDag-Erling Smørgrav return 1; 485b7579f77SDag-Erling Smørgrav } 486b7579f77SDag-Erling Smørgrav 487b7579f77SDag-Erling Smørgrav /** 488b7579f77SDag-Erling Smørgrav * Copy and decompress rrs 489b7579f77SDag-Erling Smørgrav * @param pkt: the packet for compression pointer resolution. 490b7579f77SDag-Erling Smørgrav * @param msg: the parsed message 491b7579f77SDag-Erling Smørgrav * @param rep: reply info to put rrs into. 492b7579f77SDag-Erling Smørgrav * @param region: if not NULL, used for allocation. 493b7579f77SDag-Erling Smørgrav * @return 0 on failure. 494b7579f77SDag-Erling Smørgrav */ 495b7579f77SDag-Erling Smørgrav static int 49617d15b25SDag-Erling Smørgrav parse_copy_decompress(sldns_buffer* pkt, struct msg_parse* msg, 497b7579f77SDag-Erling Smørgrav struct reply_info* rep, struct regional* region) 498b7579f77SDag-Erling Smørgrav { 499b7579f77SDag-Erling Smørgrav size_t i; 500b7579f77SDag-Erling Smørgrav struct rrset_parse *pset = msg->rrset_first; 501b7579f77SDag-Erling Smørgrav struct packed_rrset_data* data; 502b7579f77SDag-Erling Smørgrav log_assert(rep); 503b7579f77SDag-Erling Smørgrav rep->ttl = MAX_TTL; 504b7579f77SDag-Erling Smørgrav rep->security = sec_status_unchecked; 505b7579f77SDag-Erling Smørgrav if(rep->rrset_count == 0) 506b7579f77SDag-Erling Smørgrav rep->ttl = NORR_TTL; 507b7579f77SDag-Erling Smørgrav 508b7579f77SDag-Erling Smørgrav for(i=0; i<rep->rrset_count; i++) { 509b7579f77SDag-Erling Smørgrav if(!parse_copy_decompress_rrset(pkt, msg, pset, region, 510b7579f77SDag-Erling Smørgrav rep->rrsets[i])) 511b7579f77SDag-Erling Smørgrav return 0; 512b7579f77SDag-Erling Smørgrav data = (struct packed_rrset_data*)rep->rrsets[i]->entry.data; 513b7579f77SDag-Erling Smørgrav if(data->ttl < rep->ttl) 514b7579f77SDag-Erling Smørgrav rep->ttl = data->ttl; 515b7579f77SDag-Erling Smørgrav 516b7579f77SDag-Erling Smørgrav pset = pset->rrset_all_next; 517b7579f77SDag-Erling Smørgrav } 518b7579f77SDag-Erling Smørgrav rep->prefetch_ttl = PREFETCH_TTL_CALC(rep->ttl); 5194c75e3aaSDag-Erling Smørgrav rep->serve_expired_ttl = rep->ttl + SERVE_EXPIRED_TTL; 52046d2f618SCy Schubert /* rep->serve_expired_norec_ttl should stay at 0 */ 52146d2f618SCy Schubert log_assert(rep->serve_expired_norec_ttl == 0); 522b7579f77SDag-Erling Smørgrav return 1; 523b7579f77SDag-Erling Smørgrav } 524b7579f77SDag-Erling Smørgrav 525b7579f77SDag-Erling Smørgrav int 52617d15b25SDag-Erling Smørgrav parse_create_msg(sldns_buffer* pkt, struct msg_parse* msg, 527b7579f77SDag-Erling Smørgrav struct alloc_cache* alloc, struct query_info* qinf, 528b7579f77SDag-Erling Smørgrav struct reply_info** rep, struct regional* region) 529b7579f77SDag-Erling Smørgrav { 530b7579f77SDag-Erling Smørgrav log_assert(pkt && msg); 531b7579f77SDag-Erling Smørgrav if(!parse_create_qinfo(pkt, msg, qinf, region)) 532b7579f77SDag-Erling Smørgrav return 0; 533b7579f77SDag-Erling Smørgrav if(!parse_create_repinfo(msg, rep, region)) 534b7579f77SDag-Erling Smørgrav return 0; 535a755b6f6SDag-Erling Smørgrav if(!reply_info_alloc_rrset_keys(*rep, alloc, region)) { 536a755b6f6SDag-Erling Smørgrav if(!region) reply_info_parsedelete(*rep, alloc); 537b7579f77SDag-Erling Smørgrav return 0; 538a755b6f6SDag-Erling Smørgrav } 539a755b6f6SDag-Erling Smørgrav if(!parse_copy_decompress(pkt, msg, *rep, region)) { 540a755b6f6SDag-Erling Smørgrav if(!region) reply_info_parsedelete(*rep, alloc); 541b7579f77SDag-Erling Smørgrav return 0; 542a755b6f6SDag-Erling Smørgrav } 543b7579f77SDag-Erling Smørgrav return 1; 544b7579f77SDag-Erling Smørgrav } 545b7579f77SDag-Erling Smørgrav 54617d15b25SDag-Erling Smørgrav int reply_info_parse(sldns_buffer* pkt, struct alloc_cache* alloc, 547b7579f77SDag-Erling Smørgrav struct query_info* qinf, struct reply_info** rep, 548b7579f77SDag-Erling Smørgrav struct regional* region, struct edns_data* edns) 549b7579f77SDag-Erling Smørgrav { 550b7579f77SDag-Erling Smørgrav /* use scratch pad region-allocator during parsing. */ 551b7579f77SDag-Erling Smørgrav struct msg_parse* msg; 552b7579f77SDag-Erling Smørgrav int ret; 553b7579f77SDag-Erling Smørgrav 554b7579f77SDag-Erling Smørgrav qinf->qname = NULL; 555bc892140SDag-Erling Smørgrav qinf->local_alias = NULL; 556b7579f77SDag-Erling Smørgrav *rep = NULL; 557b7579f77SDag-Erling Smørgrav if(!(msg = regional_alloc(region, sizeof(*msg)))) { 558b7579f77SDag-Erling Smørgrav return LDNS_RCODE_SERVFAIL; 559b7579f77SDag-Erling Smørgrav } 560b7579f77SDag-Erling Smørgrav memset(msg, 0, sizeof(*msg)); 561b7579f77SDag-Erling Smørgrav 56217d15b25SDag-Erling Smørgrav sldns_buffer_set_position(pkt, 0); 563b7579f77SDag-Erling Smørgrav if((ret = parse_packet(pkt, msg, region)) != 0) { 564b7579f77SDag-Erling Smørgrav return ret; 565b7579f77SDag-Erling Smørgrav } 56624e36522SCy Schubert if((ret = parse_extract_edns_from_response_msg(msg, edns, region)) != 0) 567b7579f77SDag-Erling Smørgrav return ret; 568b7579f77SDag-Erling Smørgrav 569b7579f77SDag-Erling Smørgrav /* parse OK, allocate return structures */ 570b7579f77SDag-Erling Smørgrav /* this also performs dname decompression */ 571b7579f77SDag-Erling Smørgrav if(!parse_create_msg(pkt, msg, alloc, qinf, rep, NULL)) { 572b7579f77SDag-Erling Smørgrav query_info_clear(qinf); 573b7579f77SDag-Erling Smørgrav *rep = NULL; 574b7579f77SDag-Erling Smørgrav return LDNS_RCODE_SERVFAIL; 575b7579f77SDag-Erling Smørgrav } 576b7579f77SDag-Erling Smørgrav return 0; 577b7579f77SDag-Erling Smørgrav } 578b7579f77SDag-Erling Smørgrav 579b7579f77SDag-Erling Smørgrav /** helper compare function to sort in lock order */ 580b7579f77SDag-Erling Smørgrav static int 581b7579f77SDag-Erling Smørgrav reply_info_sortref_cmp(const void* a, const void* b) 582b7579f77SDag-Erling Smørgrav { 583b7579f77SDag-Erling Smørgrav struct rrset_ref* x = (struct rrset_ref*)a; 584b7579f77SDag-Erling Smørgrav struct rrset_ref* y = (struct rrset_ref*)b; 585b7579f77SDag-Erling Smørgrav if(x->key < y->key) return -1; 586b7579f77SDag-Erling Smørgrav if(x->key > y->key) return 1; 587b7579f77SDag-Erling Smørgrav return 0; 588b7579f77SDag-Erling Smørgrav } 589b7579f77SDag-Erling Smørgrav 590b7579f77SDag-Erling Smørgrav void 591b7579f77SDag-Erling Smørgrav reply_info_sortref(struct reply_info* rep) 592b7579f77SDag-Erling Smørgrav { 593b7579f77SDag-Erling Smørgrav qsort(&rep->ref[0], rep->rrset_count, sizeof(struct rrset_ref), 594b7579f77SDag-Erling Smørgrav reply_info_sortref_cmp); 595b7579f77SDag-Erling Smørgrav } 596b7579f77SDag-Erling Smørgrav 597b7579f77SDag-Erling Smørgrav void 59817d15b25SDag-Erling Smørgrav reply_info_set_ttls(struct reply_info* rep, time_t timenow) 599b7579f77SDag-Erling Smørgrav { 600b7579f77SDag-Erling Smørgrav size_t i, j; 601b7579f77SDag-Erling Smørgrav rep->ttl += timenow; 602b7579f77SDag-Erling Smørgrav rep->prefetch_ttl += timenow; 6034c75e3aaSDag-Erling Smørgrav rep->serve_expired_ttl += timenow; 60446d2f618SCy Schubert /* Don't set rep->serve_expired_norec_ttl; this should only be set 60546d2f618SCy Schubert * on cached records when encountering an error */ 60646d2f618SCy Schubert log_assert(rep->serve_expired_norec_ttl == 0); 607b7579f77SDag-Erling Smørgrav for(i=0; i<rep->rrset_count; i++) { 608b7579f77SDag-Erling Smørgrav struct packed_rrset_data* data = (struct packed_rrset_data*) 609b7579f77SDag-Erling Smørgrav rep->ref[i].key->entry.data; 610b7579f77SDag-Erling Smørgrav if(i>0 && rep->ref[i].key == rep->ref[i-1].key) 611b7579f77SDag-Erling Smørgrav continue; 612b7579f77SDag-Erling Smørgrav data->ttl += timenow; 613b7579f77SDag-Erling Smørgrav for(j=0; j<data->count + data->rrsig_count; j++) { 614b7579f77SDag-Erling Smørgrav data->rr_ttl[j] += timenow; 615b7579f77SDag-Erling Smørgrav } 616f44e67d1SCy Schubert data->ttl_add = timenow; 617b7579f77SDag-Erling Smørgrav } 618b7579f77SDag-Erling Smørgrav } 619b7579f77SDag-Erling Smørgrav 620b7579f77SDag-Erling Smørgrav void 621b7579f77SDag-Erling Smørgrav reply_info_parsedelete(struct reply_info* rep, struct alloc_cache* alloc) 622b7579f77SDag-Erling Smørgrav { 623b7579f77SDag-Erling Smørgrav size_t i; 624b7579f77SDag-Erling Smørgrav if(!rep) 625b7579f77SDag-Erling Smørgrav return; 626b7579f77SDag-Erling Smørgrav /* no need to lock, since not shared in hashtables. */ 627b7579f77SDag-Erling Smørgrav for(i=0; i<rep->rrset_count; i++) { 628b7579f77SDag-Erling Smørgrav ub_packed_rrset_parsedelete(rep->rrsets[i], alloc); 629b7579f77SDag-Erling Smørgrav } 6308f76bb7dSCy Schubert if(rep->reason_bogus_str) { 6318f76bb7dSCy Schubert free(rep->reason_bogus_str); 6328f76bb7dSCy Schubert rep->reason_bogus_str = NULL; 6338f76bb7dSCy Schubert } 634b7579f77SDag-Erling Smørgrav free(rep); 635b7579f77SDag-Erling Smørgrav } 636b7579f77SDag-Erling Smørgrav 637b7579f77SDag-Erling Smørgrav int 63817d15b25SDag-Erling Smørgrav query_info_parse(struct query_info* m, sldns_buffer* query) 639b7579f77SDag-Erling Smørgrav { 64017d15b25SDag-Erling Smørgrav uint8_t* q = sldns_buffer_begin(query); 641b7579f77SDag-Erling Smørgrav /* minimum size: header + \0 + qtype + qclass */ 64217d15b25SDag-Erling Smørgrav if(sldns_buffer_limit(query) < LDNS_HEADER_SIZE + 5) 643b7579f77SDag-Erling Smørgrav return 0; 6440fb34990SDag-Erling Smørgrav if((LDNS_OPCODE_WIRE(q) != LDNS_PACKET_QUERY && LDNS_OPCODE_WIRE(q) != 6450fb34990SDag-Erling Smørgrav LDNS_PACKET_NOTIFY) || LDNS_QDCOUNT(q) != 1 || 6460fb34990SDag-Erling Smørgrav sldns_buffer_position(query) != 0) 647b7579f77SDag-Erling Smørgrav return 0; 64817d15b25SDag-Erling Smørgrav sldns_buffer_skip(query, LDNS_HEADER_SIZE); 64917d15b25SDag-Erling Smørgrav m->qname = sldns_buffer_current(query); 650b7579f77SDag-Erling Smørgrav if((m->qname_len = query_dname_len(query)) == 0) 651b7579f77SDag-Erling Smørgrav return 0; /* parse error */ 65217d15b25SDag-Erling Smørgrav if(sldns_buffer_remaining(query) < 4) 653b7579f77SDag-Erling Smørgrav return 0; /* need qtype, qclass */ 65417d15b25SDag-Erling Smørgrav m->qtype = sldns_buffer_read_u16(query); 65517d15b25SDag-Erling Smørgrav m->qclass = sldns_buffer_read_u16(query); 656bc892140SDag-Erling Smørgrav m->local_alias = NULL; 657b7579f77SDag-Erling Smørgrav return 1; 658b7579f77SDag-Erling Smørgrav } 659b7579f77SDag-Erling Smørgrav 660b7579f77SDag-Erling Smørgrav /** tiny subroutine for msgreply_compare */ 661b7579f77SDag-Erling Smørgrav #define COMPARE_IT(x, y) \ 662b7579f77SDag-Erling Smørgrav if( (x) < (y) ) return -1; \ 663b7579f77SDag-Erling Smørgrav else if( (x) > (y) ) return +1; \ 664b7579f77SDag-Erling Smørgrav log_assert( (x) == (y) ); 665b7579f77SDag-Erling Smørgrav 666b7579f77SDag-Erling Smørgrav int 667b7579f77SDag-Erling Smørgrav query_info_compare(void* m1, void* m2) 668b7579f77SDag-Erling Smørgrav { 669b7579f77SDag-Erling Smørgrav struct query_info* msg1 = (struct query_info*)m1; 670b7579f77SDag-Erling Smørgrav struct query_info* msg2 = (struct query_info*)m2; 671b7579f77SDag-Erling Smørgrav int mc; 672b7579f77SDag-Erling Smørgrav /* from most different to least different for speed */ 673b7579f77SDag-Erling Smørgrav COMPARE_IT(msg1->qtype, msg2->qtype); 674b7579f77SDag-Erling Smørgrav if((mc = query_dname_compare(msg1->qname, msg2->qname)) != 0) 675b7579f77SDag-Erling Smørgrav return mc; 676b7579f77SDag-Erling Smørgrav log_assert(msg1->qname_len == msg2->qname_len); 677b7579f77SDag-Erling Smørgrav COMPARE_IT(msg1->qclass, msg2->qclass); 678b7579f77SDag-Erling Smørgrav return 0; 679b7579f77SDag-Erling Smørgrav #undef COMPARE_IT 680b7579f77SDag-Erling Smørgrav } 681b7579f77SDag-Erling Smørgrav 682b7579f77SDag-Erling Smørgrav void 683b7579f77SDag-Erling Smørgrav query_info_clear(struct query_info* m) 684b7579f77SDag-Erling Smørgrav { 685b7579f77SDag-Erling Smørgrav free(m->qname); 686b7579f77SDag-Erling Smørgrav m->qname = NULL; 687b7579f77SDag-Erling Smørgrav } 688b7579f77SDag-Erling Smørgrav 689b7579f77SDag-Erling Smørgrav size_t 690b7579f77SDag-Erling Smørgrav msgreply_sizefunc(void* k, void* d) 691b7579f77SDag-Erling Smørgrav { 692b7579f77SDag-Erling Smørgrav struct msgreply_entry* q = (struct msgreply_entry*)k; 693b7579f77SDag-Erling Smørgrav struct reply_info* r = (struct reply_info*)d; 694b7579f77SDag-Erling Smørgrav size_t s = sizeof(struct msgreply_entry) + sizeof(struct reply_info) 695b7579f77SDag-Erling Smørgrav + q->key.qname_len + lock_get_mem(&q->entry.lock) 696b7579f77SDag-Erling Smørgrav - sizeof(struct rrset_ref); 697b7579f77SDag-Erling Smørgrav s += r->rrset_count * sizeof(struct rrset_ref); 698b7579f77SDag-Erling Smørgrav s += r->rrset_count * sizeof(struct ub_packed_rrset_key*); 699b7579f77SDag-Erling Smørgrav return s; 700b7579f77SDag-Erling Smørgrav } 701b7579f77SDag-Erling Smørgrav 702b7579f77SDag-Erling Smørgrav void 703b7579f77SDag-Erling Smørgrav query_entry_delete(void *k, void* ATTR_UNUSED(arg)) 704b7579f77SDag-Erling Smørgrav { 705b7579f77SDag-Erling Smørgrav struct msgreply_entry* q = (struct msgreply_entry*)k; 706b7579f77SDag-Erling Smørgrav lock_rw_destroy(&q->entry.lock); 707b7579f77SDag-Erling Smørgrav query_info_clear(&q->key); 708b7579f77SDag-Erling Smørgrav free(q); 709b7579f77SDag-Erling Smørgrav } 710b7579f77SDag-Erling Smørgrav 711b7579f77SDag-Erling Smørgrav void 712b7579f77SDag-Erling Smørgrav reply_info_delete(void* d, void* ATTR_UNUSED(arg)) 713b7579f77SDag-Erling Smørgrav { 714b7579f77SDag-Erling Smørgrav struct reply_info* r = (struct reply_info*)d; 7158f76bb7dSCy Schubert if(r->reason_bogus_str) { 7168f76bb7dSCy Schubert free(r->reason_bogus_str); 7178f76bb7dSCy Schubert r->reason_bogus_str = NULL; 7188f76bb7dSCy Schubert } 719b7579f77SDag-Erling Smørgrav free(r); 720b7579f77SDag-Erling Smørgrav } 721b7579f77SDag-Erling Smørgrav 7223005e0a3SDag-Erling Smørgrav hashvalue_type 723ff825849SDag-Erling Smørgrav query_info_hash(struct query_info *q, uint16_t flags) 724b7579f77SDag-Erling Smørgrav { 7253005e0a3SDag-Erling Smørgrav hashvalue_type h = 0xab; 726b7579f77SDag-Erling Smørgrav h = hashlittle(&q->qtype, sizeof(q->qtype), h); 727ff825849SDag-Erling Smørgrav if(q->qtype == LDNS_RR_TYPE_AAAA && (flags&BIT_CD)) 728ff825849SDag-Erling Smørgrav h++; 729b7579f77SDag-Erling Smørgrav h = hashlittle(&q->qclass, sizeof(q->qclass), h); 730b7579f77SDag-Erling Smørgrav h = dname_query_hash(q->qname, h); 731b7579f77SDag-Erling Smørgrav return h; 732b7579f77SDag-Erling Smørgrav } 733b7579f77SDag-Erling Smørgrav 734b7579f77SDag-Erling Smørgrav struct msgreply_entry* 735b7579f77SDag-Erling Smørgrav query_info_entrysetup(struct query_info* q, struct reply_info* r, 7363005e0a3SDag-Erling Smørgrav hashvalue_type h) 737b7579f77SDag-Erling Smørgrav { 738b7579f77SDag-Erling Smørgrav struct msgreply_entry* e = (struct msgreply_entry*)malloc( 739b7579f77SDag-Erling Smørgrav sizeof(struct msgreply_entry)); 740b7579f77SDag-Erling Smørgrav if(!e) return NULL; 741b7579f77SDag-Erling Smørgrav memcpy(&e->key, q, sizeof(*q)); 742b7579f77SDag-Erling Smørgrav e->entry.hash = h; 743b7579f77SDag-Erling Smørgrav e->entry.key = e; 744b7579f77SDag-Erling Smørgrav e->entry.data = r; 745b7579f77SDag-Erling Smørgrav lock_rw_init(&e->entry.lock); 74657bddd21SDag-Erling Smørgrav lock_protect(&e->entry.lock, &e->key.qname, sizeof(e->key.qname)); 74757bddd21SDag-Erling Smørgrav lock_protect(&e->entry.lock, &e->key.qname_len, sizeof(e->key.qname_len)); 74857bddd21SDag-Erling Smørgrav lock_protect(&e->entry.lock, &e->key.qtype, sizeof(e->key.qtype)); 74957bddd21SDag-Erling Smørgrav lock_protect(&e->entry.lock, &e->key.qclass, sizeof(e->key.qclass)); 75057bddd21SDag-Erling Smørgrav lock_protect(&e->entry.lock, &e->key.local_alias, sizeof(e->key.local_alias)); 75157bddd21SDag-Erling Smørgrav lock_protect(&e->entry.lock, &e->entry.hash, sizeof(e->entry.hash)); 75257bddd21SDag-Erling Smørgrav lock_protect(&e->entry.lock, &e->entry.key, sizeof(e->entry.key)); 75357bddd21SDag-Erling Smørgrav lock_protect(&e->entry.lock, &e->entry.data, sizeof(e->entry.data)); 754b7579f77SDag-Erling Smørgrav lock_protect(&e->entry.lock, e->key.qname, e->key.qname_len); 755b7579f77SDag-Erling Smørgrav q->qname = NULL; 756b7579f77SDag-Erling Smørgrav return e; 757b7579f77SDag-Erling Smørgrav } 758b7579f77SDag-Erling Smørgrav 759b7579f77SDag-Erling Smørgrav /** copy rrsets from replyinfo to dest replyinfo */ 760b7579f77SDag-Erling Smørgrav static int 761b7579f77SDag-Erling Smørgrav repinfo_copy_rrsets(struct reply_info* dest, struct reply_info* from, 762b7579f77SDag-Erling Smørgrav struct regional* region) 763b7579f77SDag-Erling Smørgrav { 764b7579f77SDag-Erling Smørgrav size_t i, s; 765b7579f77SDag-Erling Smørgrav struct packed_rrset_data* fd, *dd; 766b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* fk, *dk; 767b7579f77SDag-Erling Smørgrav for(i=0; i<dest->rrset_count; i++) { 768b7579f77SDag-Erling Smørgrav fk = from->rrsets[i]; 769b7579f77SDag-Erling Smørgrav dk = dest->rrsets[i]; 770b7579f77SDag-Erling Smørgrav fd = (struct packed_rrset_data*)fk->entry.data; 771b7579f77SDag-Erling Smørgrav dk->entry.hash = fk->entry.hash; 772b7579f77SDag-Erling Smørgrav dk->rk = fk->rk; 773b7579f77SDag-Erling Smørgrav if(region) { 774b7579f77SDag-Erling Smørgrav dk->id = fk->id; 775b7579f77SDag-Erling Smørgrav dk->rk.dname = (uint8_t*)regional_alloc_init(region, 776b7579f77SDag-Erling Smørgrav fk->rk.dname, fk->rk.dname_len); 777b7579f77SDag-Erling Smørgrav } else 778b7579f77SDag-Erling Smørgrav dk->rk.dname = (uint8_t*)memdup(fk->rk.dname, 779b7579f77SDag-Erling Smørgrav fk->rk.dname_len); 780b7579f77SDag-Erling Smørgrav if(!dk->rk.dname) 781b7579f77SDag-Erling Smørgrav return 0; 782b7579f77SDag-Erling Smørgrav s = packed_rrset_sizeof(fd); 783b7579f77SDag-Erling Smørgrav if(region) 784b7579f77SDag-Erling Smørgrav dd = (struct packed_rrset_data*)regional_alloc_init( 785b7579f77SDag-Erling Smørgrav region, fd, s); 786b7579f77SDag-Erling Smørgrav else dd = (struct packed_rrset_data*)memdup(fd, s); 787b7579f77SDag-Erling Smørgrav if(!dd) 788b7579f77SDag-Erling Smørgrav return 0; 789b7579f77SDag-Erling Smørgrav packed_rrset_ptr_fixup(dd); 790b7579f77SDag-Erling Smørgrav dk->entry.data = (void*)dd; 791b7579f77SDag-Erling Smørgrav } 792b7579f77SDag-Erling Smørgrav return 1; 793b7579f77SDag-Erling Smørgrav } 794b7579f77SDag-Erling Smørgrav 795b7579f77SDag-Erling Smørgrav struct reply_info* 796b7579f77SDag-Erling Smørgrav reply_info_copy(struct reply_info* rep, struct alloc_cache* alloc, 797b7579f77SDag-Erling Smørgrav struct regional* region) 798b7579f77SDag-Erling Smørgrav { 799b7579f77SDag-Erling Smørgrav struct reply_info* cp; 800b7579f77SDag-Erling Smørgrav cp = construct_reply_info_base(region, rep->flags, rep->qdcount, 8014c75e3aaSDag-Erling Smørgrav rep->ttl, rep->prefetch_ttl, rep->serve_expired_ttl, 80246d2f618SCy Schubert rep->serve_expired_norec_ttl, 8034c75e3aaSDag-Erling Smørgrav rep->an_numrrsets, rep->ns_numrrsets, rep->ar_numrrsets, 8048f76bb7dSCy Schubert rep->rrset_count, rep->security, rep->reason_bogus); 805b7579f77SDag-Erling Smørgrav if(!cp) 806b7579f77SDag-Erling Smørgrav return NULL; 8078f76bb7dSCy Schubert 8088f76bb7dSCy Schubert if(rep->reason_bogus_str && *rep->reason_bogus_str != 0) { 8098f76bb7dSCy Schubert if(region) { 8108f76bb7dSCy Schubert cp->reason_bogus_str = (char*)regional_alloc(region, 8118f76bb7dSCy Schubert sizeof(char) 8128f76bb7dSCy Schubert * (strlen(rep->reason_bogus_str)+1)); 8138f76bb7dSCy Schubert } else { 8148f76bb7dSCy Schubert cp->reason_bogus_str = malloc(sizeof(char) 8158f76bb7dSCy Schubert * (strlen(rep->reason_bogus_str)+1)); 8168f76bb7dSCy Schubert } 8178f76bb7dSCy Schubert if(!cp->reason_bogus_str) { 8188f76bb7dSCy Schubert if(!region) 8198f76bb7dSCy Schubert reply_info_parsedelete(cp, alloc); 8208f76bb7dSCy Schubert return NULL; 8218f76bb7dSCy Schubert } 8228f76bb7dSCy Schubert memcpy(cp->reason_bogus_str, rep->reason_bogus_str, 8238f76bb7dSCy Schubert strlen(rep->reason_bogus_str)+1); 8248f76bb7dSCy Schubert } 8258f76bb7dSCy Schubert 826b7579f77SDag-Erling Smørgrav /* allocate ub_key structures special or not */ 82765b390aaSDag-Erling Smørgrav if(!reply_info_alloc_rrset_keys(cp, alloc, region)) { 828b7579f77SDag-Erling Smørgrav if(!region) 829b7579f77SDag-Erling Smørgrav reply_info_parsedelete(cp, alloc); 830b7579f77SDag-Erling Smørgrav return NULL; 831b7579f77SDag-Erling Smørgrav } 832b7579f77SDag-Erling Smørgrav if(!repinfo_copy_rrsets(cp, rep, region)) { 833b7579f77SDag-Erling Smørgrav if(!region) 834b7579f77SDag-Erling Smørgrav reply_info_parsedelete(cp, alloc); 835b7579f77SDag-Erling Smørgrav return NULL; 836b7579f77SDag-Erling Smørgrav } 837b7579f77SDag-Erling Smørgrav return cp; 838b7579f77SDag-Erling Smørgrav } 839b7579f77SDag-Erling Smørgrav 840b7579f77SDag-Erling Smørgrav uint8_t* 841b7579f77SDag-Erling Smørgrav reply_find_final_cname_target(struct query_info* qinfo, struct reply_info* rep) 842b7579f77SDag-Erling Smørgrav { 843b7579f77SDag-Erling Smørgrav uint8_t* sname = qinfo->qname; 844b7579f77SDag-Erling Smørgrav size_t snamelen = qinfo->qname_len; 845b7579f77SDag-Erling Smørgrav size_t i; 846b7579f77SDag-Erling Smørgrav for(i=0; i<rep->an_numrrsets; i++) { 847b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* s = rep->rrsets[i]; 848b7579f77SDag-Erling Smørgrav /* follow CNAME chain (if any) */ 849b7579f77SDag-Erling Smørgrav if(ntohs(s->rk.type) == LDNS_RR_TYPE_CNAME && 850b7579f77SDag-Erling Smørgrav ntohs(s->rk.rrset_class) == qinfo->qclass && 851b7579f77SDag-Erling Smørgrav snamelen == s->rk.dname_len && 852b7579f77SDag-Erling Smørgrav query_dname_compare(sname, s->rk.dname) == 0) { 853b7579f77SDag-Erling Smørgrav get_cname_target(s, &sname, &snamelen); 854b7579f77SDag-Erling Smørgrav } 855b7579f77SDag-Erling Smørgrav } 856b7579f77SDag-Erling Smørgrav if(sname != qinfo->qname) 857b7579f77SDag-Erling Smørgrav return sname; 858b7579f77SDag-Erling Smørgrav return NULL; 859b7579f77SDag-Erling Smørgrav } 860b7579f77SDag-Erling Smørgrav 861b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* 862b7579f77SDag-Erling Smørgrav reply_find_answer_rrset(struct query_info* qinfo, struct reply_info* rep) 863b7579f77SDag-Erling Smørgrav { 864b7579f77SDag-Erling Smørgrav uint8_t* sname = qinfo->qname; 865b7579f77SDag-Erling Smørgrav size_t snamelen = qinfo->qname_len; 866b7579f77SDag-Erling Smørgrav size_t i; 867b7579f77SDag-Erling Smørgrav for(i=0; i<rep->an_numrrsets; i++) { 868b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* s = rep->rrsets[i]; 869b7579f77SDag-Erling Smørgrav /* first match type, for query of qtype cname */ 870b7579f77SDag-Erling Smørgrav if(ntohs(s->rk.type) == qinfo->qtype && 871b7579f77SDag-Erling Smørgrav ntohs(s->rk.rrset_class) == qinfo->qclass && 872b7579f77SDag-Erling Smørgrav snamelen == s->rk.dname_len && 873b7579f77SDag-Erling Smørgrav query_dname_compare(sname, s->rk.dname) == 0) { 874b7579f77SDag-Erling Smørgrav return s; 875b7579f77SDag-Erling Smørgrav } 876b7579f77SDag-Erling Smørgrav /* follow CNAME chain (if any) */ 877b7579f77SDag-Erling Smørgrav if(ntohs(s->rk.type) == LDNS_RR_TYPE_CNAME && 878b7579f77SDag-Erling Smørgrav ntohs(s->rk.rrset_class) == qinfo->qclass && 879b7579f77SDag-Erling Smørgrav snamelen == s->rk.dname_len && 880b7579f77SDag-Erling Smørgrav query_dname_compare(sname, s->rk.dname) == 0) { 881b7579f77SDag-Erling Smørgrav get_cname_target(s, &sname, &snamelen); 882b7579f77SDag-Erling Smørgrav } 883b7579f77SDag-Erling Smørgrav } 884b7579f77SDag-Erling Smørgrav return NULL; 885b7579f77SDag-Erling Smørgrav } 886b7579f77SDag-Erling Smørgrav 887b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* reply_find_rrset_section_an(struct reply_info* rep, 888b7579f77SDag-Erling Smørgrav uint8_t* name, size_t namelen, uint16_t type, uint16_t dclass) 889b7579f77SDag-Erling Smørgrav { 890b7579f77SDag-Erling Smørgrav size_t i; 891b7579f77SDag-Erling Smørgrav for(i=0; i<rep->an_numrrsets; i++) { 892b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* s = rep->rrsets[i]; 893b7579f77SDag-Erling Smørgrav if(ntohs(s->rk.type) == type && 894b7579f77SDag-Erling Smørgrav ntohs(s->rk.rrset_class) == dclass && 895b7579f77SDag-Erling Smørgrav namelen == s->rk.dname_len && 896b7579f77SDag-Erling Smørgrav query_dname_compare(name, s->rk.dname) == 0) { 897b7579f77SDag-Erling Smørgrav return s; 898b7579f77SDag-Erling Smørgrav } 899b7579f77SDag-Erling Smørgrav } 900b7579f77SDag-Erling Smørgrav return NULL; 901b7579f77SDag-Erling Smørgrav } 902b7579f77SDag-Erling Smørgrav 903b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* reply_find_rrset_section_ns(struct reply_info* rep, 904b7579f77SDag-Erling Smørgrav uint8_t* name, size_t namelen, uint16_t type, uint16_t dclass) 905b7579f77SDag-Erling Smørgrav { 906b7579f77SDag-Erling Smørgrav size_t i; 907b7579f77SDag-Erling Smørgrav for(i=rep->an_numrrsets; i<rep->an_numrrsets+rep->ns_numrrsets; i++) { 908b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* s = rep->rrsets[i]; 909b7579f77SDag-Erling Smørgrav if(ntohs(s->rk.type) == type && 910b7579f77SDag-Erling Smørgrav ntohs(s->rk.rrset_class) == dclass && 911b7579f77SDag-Erling Smørgrav namelen == s->rk.dname_len && 912b7579f77SDag-Erling Smørgrav query_dname_compare(name, s->rk.dname) == 0) { 913b7579f77SDag-Erling Smørgrav return s; 914b7579f77SDag-Erling Smørgrav } 915b7579f77SDag-Erling Smørgrav } 916b7579f77SDag-Erling Smørgrav return NULL; 917b7579f77SDag-Erling Smørgrav } 918b7579f77SDag-Erling Smørgrav 919b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* reply_find_rrset(struct reply_info* rep, 920b7579f77SDag-Erling Smørgrav uint8_t* name, size_t namelen, uint16_t type, uint16_t dclass) 921b7579f77SDag-Erling Smørgrav { 922b7579f77SDag-Erling Smørgrav size_t i; 923b7579f77SDag-Erling Smørgrav for(i=0; i<rep->rrset_count; i++) { 924b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* s = rep->rrsets[i]; 925b7579f77SDag-Erling Smørgrav if(ntohs(s->rk.type) == type && 926b7579f77SDag-Erling Smørgrav ntohs(s->rk.rrset_class) == dclass && 927b7579f77SDag-Erling Smørgrav namelen == s->rk.dname_len && 928b7579f77SDag-Erling Smørgrav query_dname_compare(name, s->rk.dname) == 0) { 929b7579f77SDag-Erling Smørgrav return s; 930b7579f77SDag-Erling Smørgrav } 931b7579f77SDag-Erling Smørgrav } 932b7579f77SDag-Erling Smørgrav return NULL; 933b7579f77SDag-Erling Smørgrav } 934b7579f77SDag-Erling Smørgrav 935b7579f77SDag-Erling Smørgrav void 936b7579f77SDag-Erling Smørgrav log_dns_msg(const char* str, struct query_info* qinfo, struct reply_info* rep) 937b7579f77SDag-Erling Smørgrav { 938b7579f77SDag-Erling Smørgrav /* not particularly fast but flexible, make wireformat and print */ 93917d15b25SDag-Erling Smørgrav sldns_buffer* buf = sldns_buffer_new(65535); 940b7579f77SDag-Erling Smørgrav struct regional* region = regional_create(); 9415469a995SCy Schubert if(!(buf && region)) { 9425469a995SCy Schubert log_err("%s: log_dns_msg: out of memory", str); 9435469a995SCy Schubert sldns_buffer_free(buf); 9445469a995SCy Schubert regional_destroy(region); 9455469a995SCy Schubert return; 9465469a995SCy Schubert } 947b7579f77SDag-Erling Smørgrav if(!reply_info_encode(qinfo, rep, 0, rep->flags, buf, 0, 9480eefd307SCy Schubert region, 65535, 1, 0)) { 9495469a995SCy Schubert log_err("%s: log_dns_msg: out of memory", str); 950b7579f77SDag-Erling Smørgrav } else { 951ff825849SDag-Erling Smørgrav char* s = sldns_wire2str_pkt(sldns_buffer_begin(buf), 95217d15b25SDag-Erling Smørgrav sldns_buffer_limit(buf)); 953ff825849SDag-Erling Smørgrav if(!s) { 95417d15b25SDag-Erling Smørgrav log_info("%s: log_dns_msg: ldns tostr failed", str); 955b7579f77SDag-Erling Smørgrav } else { 956ff825849SDag-Erling Smørgrav log_info("%s %s", str, s); 957b7579f77SDag-Erling Smørgrav } 958ff825849SDag-Erling Smørgrav free(s); 959b7579f77SDag-Erling Smørgrav } 96017d15b25SDag-Erling Smørgrav sldns_buffer_free(buf); 961b7579f77SDag-Erling Smørgrav regional_destroy(region); 962b7579f77SDag-Erling Smørgrav } 963b7579f77SDag-Erling Smørgrav 964b7579f77SDag-Erling Smørgrav void 9653005e0a3SDag-Erling Smørgrav log_reply_info(enum verbosity_value v, struct query_info *qinf, 9663005e0a3SDag-Erling Smørgrav struct sockaddr_storage *addr, socklen_t addrlen, struct timeval dur, 967b7c0c8c1SCy Schubert int cached, struct sldns_buffer *rmsg, struct sockaddr_storage* daddr, 968*be771a7bSCy Schubert enum comm_point_type tp, void* ssl) 9693005e0a3SDag-Erling Smørgrav { 9703005e0a3SDag-Erling Smørgrav char clientip_buf[128]; 9713005e0a3SDag-Erling Smørgrav char rcode_buf[16]; 972b7c0c8c1SCy Schubert char dest_buf[160]; 9733005e0a3SDag-Erling Smørgrav uint16_t rcode = FLAGS_GET_RCODE(sldns_buffer_read_u16_at(rmsg, 2)); 9743005e0a3SDag-Erling Smørgrav 9753005e0a3SDag-Erling Smørgrav if(verbosity < v) 9763005e0a3SDag-Erling Smørgrav return; 9773005e0a3SDag-Erling Smørgrav 9783005e0a3SDag-Erling Smørgrav sldns_wire2str_rcode_buf((int)rcode, rcode_buf, sizeof(rcode_buf)); 9793005e0a3SDag-Erling Smørgrav addr_to_str(addr, addrlen, clientip_buf, sizeof(clientip_buf)); 980b7c0c8c1SCy Schubert if(daddr) { 981b7c0c8c1SCy Schubert char da[128]; 982b7c0c8c1SCy Schubert int port = 0; 983b7c0c8c1SCy Schubert char* comm; 984b7c0c8c1SCy Schubert if(daddr->ss_family == AF_INET6) { 985b7c0c8c1SCy Schubert struct sockaddr_in6 *d = (struct sockaddr_in6 *)daddr; 986b7c0c8c1SCy Schubert if(inet_ntop(d->sin6_family, &d->sin6_addr, da, 987b7c0c8c1SCy Schubert sizeof(*d)) == 0) 988b7c0c8c1SCy Schubert snprintf(dest_buf, sizeof(dest_buf), 989b7c0c8c1SCy Schubert "(inet_ntop_error)"); 990b7c0c8c1SCy Schubert port = ntohs(d->sin6_port); 991b7c0c8c1SCy Schubert } else if(daddr->ss_family == AF_INET) { 992b7c0c8c1SCy Schubert struct sockaddr_in *d = (struct sockaddr_in *)daddr; 993b7c0c8c1SCy Schubert if(inet_ntop(d->sin_family, &d->sin_addr, da, 994b7c0c8c1SCy Schubert sizeof(*d)) == 0) 995b7c0c8c1SCy Schubert snprintf(dest_buf, sizeof(dest_buf), 996b7c0c8c1SCy Schubert "(inet_ntop_error)"); 997b7c0c8c1SCy Schubert port = ntohs(d->sin_port); 998b7c0c8c1SCy Schubert } else { 999b7c0c8c1SCy Schubert snprintf(da, sizeof(da), "socket%d", 1000b7c0c8c1SCy Schubert (int)daddr->ss_family); 1001b7c0c8c1SCy Schubert } 1002b7c0c8c1SCy Schubert comm = "udp"; 1003*be771a7bSCy Schubert if(tp == comm_tcp) comm = (ssl?"dot":"tcp"); 1004*be771a7bSCy Schubert else if(tp == comm_tcp_accept) comm = (ssl?"dot":"tcp"); 1005*be771a7bSCy Schubert else if(tp == comm_http) comm = "doh"; 1006b7c0c8c1SCy Schubert else if(tp == comm_local) comm = "unix"; 1007b7c0c8c1SCy Schubert else if(tp == comm_raw) comm = "raw"; 1008b7c0c8c1SCy Schubert snprintf(dest_buf, sizeof(dest_buf), " on %s %s %d", 1009b7c0c8c1SCy Schubert comm, da, port); 1010b7c0c8c1SCy Schubert } else { 1011b7c0c8c1SCy Schubert dest_buf[0]=0; 1012b7c0c8c1SCy Schubert } 10133005e0a3SDag-Erling Smørgrav if(rcode == LDNS_RCODE_FORMERR) 10143005e0a3SDag-Erling Smørgrav { 1015e86b9096SDag-Erling Smørgrav if(LOG_TAG_QUERYREPLY) 1016b7c0c8c1SCy Schubert log_reply("%s - - - %s - - -%s", clientip_buf, 1017b7c0c8c1SCy Schubert rcode_buf, dest_buf); 1018b7c0c8c1SCy Schubert else log_info("%s - - - %s - - -%s", clientip_buf, 1019b7c0c8c1SCy Schubert rcode_buf, dest_buf); 10203005e0a3SDag-Erling Smørgrav } else { 1021*be771a7bSCy Schubert char qname_buf[LDNS_MAX_DOMAINLEN]; 1022*be771a7bSCy Schubert char type_buf[16]; 1023*be771a7bSCy Schubert char class_buf[16]; 1024*be771a7bSCy Schubert size_t pktlen; 1025971980c3SDag-Erling Smørgrav if(qinf->qname) 10263005e0a3SDag-Erling Smørgrav dname_str(qinf->qname, qname_buf); 1027971980c3SDag-Erling Smørgrav else snprintf(qname_buf, sizeof(qname_buf), "null"); 10283005e0a3SDag-Erling Smørgrav pktlen = sldns_buffer_limit(rmsg); 10293005e0a3SDag-Erling Smørgrav sldns_wire2str_type_buf(qinf->qtype, type_buf, sizeof(type_buf)); 10303005e0a3SDag-Erling Smørgrav sldns_wire2str_class_buf(qinf->qclass, class_buf, sizeof(class_buf)); 1031e86b9096SDag-Erling Smørgrav if(LOG_TAG_QUERYREPLY) 1032b7c0c8c1SCy Schubert log_reply("%s %s %s %s %s " ARG_LL "d.%6.6d %d %d%s", 1033e86b9096SDag-Erling Smørgrav clientip_buf, qname_buf, type_buf, class_buf, 1034b7c0c8c1SCy Schubert rcode_buf, (long long)dur.tv_sec, (int)dur.tv_usec, 1035b7c0c8c1SCy Schubert cached, (int)pktlen, dest_buf); 1036b7c0c8c1SCy Schubert else log_info("%s %s %s %s %s " ARG_LL "d.%6.6d %d %d%s", 10373005e0a3SDag-Erling Smørgrav clientip_buf, qname_buf, type_buf, class_buf, 1038b7c0c8c1SCy Schubert rcode_buf, (long long)dur.tv_sec, (int)dur.tv_usec, 1039b7c0c8c1SCy Schubert cached, (int)pktlen, dest_buf); 10403005e0a3SDag-Erling Smørgrav } 10413005e0a3SDag-Erling Smørgrav } 10423005e0a3SDag-Erling Smørgrav 10433005e0a3SDag-Erling Smørgrav void 1044b7579f77SDag-Erling Smørgrav log_query_info(enum verbosity_value v, const char* str, 1045b7579f77SDag-Erling Smørgrav struct query_info* qinf) 1046b7579f77SDag-Erling Smørgrav { 1047b7579f77SDag-Erling Smørgrav log_nametypeclass(v, str, qinf->qname, qinf->qtype, qinf->qclass); 1048b7579f77SDag-Erling Smørgrav } 1049b7579f77SDag-Erling Smørgrav 1050b7579f77SDag-Erling Smørgrav int 105109a3aaf3SDag-Erling Smørgrav reply_check_cname_chain(struct query_info* qinfo, struct reply_info* rep) 1052b7579f77SDag-Erling Smørgrav { 1053b7579f77SDag-Erling Smørgrav /* check only answer section rrs for matching cname chain. 1054b7579f77SDag-Erling Smørgrav * the cache may return changed rdata, but owner names are untouched.*/ 1055b7579f77SDag-Erling Smørgrav size_t i; 105609a3aaf3SDag-Erling Smørgrav uint8_t* sname = qinfo->qname; 105709a3aaf3SDag-Erling Smørgrav size_t snamelen = qinfo->qname_len; 1058b7579f77SDag-Erling Smørgrav for(i=0; i<rep->an_numrrsets; i++) { 1059b7579f77SDag-Erling Smørgrav uint16_t t = ntohs(rep->rrsets[i]->rk.type); 1060b7579f77SDag-Erling Smørgrav if(t == LDNS_RR_TYPE_DNAME) 1061b7579f77SDag-Erling Smørgrav continue; /* skip dnames; note TTL 0 not cached */ 1062b7579f77SDag-Erling Smørgrav /* verify that owner matches current sname */ 1063b7579f77SDag-Erling Smørgrav if(query_dname_compare(sname, rep->rrsets[i]->rk.dname) != 0){ 1064b7579f77SDag-Erling Smørgrav /* cname chain broken */ 1065b7579f77SDag-Erling Smørgrav return 0; 1066b7579f77SDag-Erling Smørgrav } 1067b7579f77SDag-Erling Smørgrav /* if this is a cname; move on */ 1068b7579f77SDag-Erling Smørgrav if(t == LDNS_RR_TYPE_CNAME) { 1069b7579f77SDag-Erling Smørgrav get_cname_target(rep->rrsets[i], &sname, &snamelen); 1070b7579f77SDag-Erling Smørgrav } 1071b7579f77SDag-Erling Smørgrav } 1072b7579f77SDag-Erling Smørgrav return 1; 1073b7579f77SDag-Erling Smørgrav } 1074b7579f77SDag-Erling Smørgrav 1075b7579f77SDag-Erling Smørgrav int 1076b7579f77SDag-Erling Smørgrav reply_all_rrsets_secure(struct reply_info* rep) 1077b7579f77SDag-Erling Smørgrav { 1078b7579f77SDag-Erling Smørgrav size_t i; 1079b7579f77SDag-Erling Smørgrav for(i=0; i<rep->rrset_count; i++) { 1080b7579f77SDag-Erling Smørgrav if( ((struct packed_rrset_data*)rep->rrsets[i]->entry.data) 1081b7579f77SDag-Erling Smørgrav ->security != sec_status_secure ) 1082b7579f77SDag-Erling Smørgrav return 0; 1083b7579f77SDag-Erling Smørgrav } 1084b7579f77SDag-Erling Smørgrav return 1; 1085b7579f77SDag-Erling Smørgrav } 1086e2d15004SDag-Erling Smørgrav 108757bddd21SDag-Erling Smørgrav struct reply_info* 108857bddd21SDag-Erling Smørgrav parse_reply_in_temp_region(sldns_buffer* pkt, struct regional* region, 108957bddd21SDag-Erling Smørgrav struct query_info* qi) 109057bddd21SDag-Erling Smørgrav { 109157bddd21SDag-Erling Smørgrav struct reply_info* rep; 109257bddd21SDag-Erling Smørgrav struct msg_parse* msg; 109357bddd21SDag-Erling Smørgrav if(!(msg = regional_alloc(region, sizeof(*msg)))) { 109457bddd21SDag-Erling Smørgrav return NULL; 109557bddd21SDag-Erling Smørgrav } 109657bddd21SDag-Erling Smørgrav memset(msg, 0, sizeof(*msg)); 109757bddd21SDag-Erling Smørgrav sldns_buffer_set_position(pkt, 0); 10984c75e3aaSDag-Erling Smørgrav if(parse_packet(pkt, msg, region) != 0){ 109957bddd21SDag-Erling Smørgrav return 0; 11004c75e3aaSDag-Erling Smørgrav } 110157bddd21SDag-Erling Smørgrav if(!parse_create_msg(pkt, msg, NULL, qi, &rep, region)) { 110257bddd21SDag-Erling Smørgrav return 0; 110357bddd21SDag-Erling Smørgrav } 110457bddd21SDag-Erling Smørgrav return rep; 110557bddd21SDag-Erling Smørgrav } 110657bddd21SDag-Erling Smørgrav 1107a39a5a69SCy Schubert int edns_opt_list_append_ede(struct edns_option** list, struct regional* region, 1108a39a5a69SCy Schubert sldns_ede_code code, const char *txt) 1109a39a5a69SCy Schubert { 1110a39a5a69SCy Schubert struct edns_option** prevp; 1111a39a5a69SCy Schubert struct edns_option* opt; 1112a39a5a69SCy Schubert size_t txt_len = txt ? strlen(txt) : 0; 1113a39a5a69SCy Schubert 1114a39a5a69SCy Schubert /* allocate new element */ 1115a39a5a69SCy Schubert opt = (struct edns_option*)regional_alloc(region, sizeof(*opt)); 1116a39a5a69SCy Schubert if(!opt) 1117a39a5a69SCy Schubert return 0; 1118a39a5a69SCy Schubert opt->next = NULL; 1119a39a5a69SCy Schubert opt->opt_code = LDNS_EDNS_EDE; 1120a39a5a69SCy Schubert opt->opt_len = txt_len + sizeof(uint16_t); 1121a39a5a69SCy Schubert opt->opt_data = regional_alloc(region, txt_len + sizeof(uint16_t)); 1122a39a5a69SCy Schubert if(!opt->opt_data) 1123a39a5a69SCy Schubert return 0; 1124a39a5a69SCy Schubert sldns_write_uint16(opt->opt_data, (uint16_t)code); 1125a39a5a69SCy Schubert if (txt_len) 1126a39a5a69SCy Schubert memmove(opt->opt_data + 2, txt, txt_len); 1127a39a5a69SCy Schubert 1128a39a5a69SCy Schubert /* append at end of list */ 1129a39a5a69SCy Schubert prevp = list; 1130a39a5a69SCy Schubert while(*prevp != NULL) 1131a39a5a69SCy Schubert prevp = &((*prevp)->next); 1132790c6b24SCy Schubert verbose(VERB_ALGO, "attached EDE code: %d with message: %s", code, (txt?txt:"\"\"")); 1133a39a5a69SCy Schubert *prevp = opt; 1134a39a5a69SCy Schubert return 1; 1135a39a5a69SCy Schubert } 1136a39a5a69SCy Schubert 11378f76bb7dSCy Schubert int edns_opt_list_append_keepalive(struct edns_option** list, int msec, 11388f76bb7dSCy Schubert struct regional* region) 11398f76bb7dSCy Schubert { 11408f76bb7dSCy Schubert uint8_t data[2]; /* For keepalive value */ 11418f76bb7dSCy Schubert data[0] = (uint8_t)((msec >> 8) & 0xff); 11428f76bb7dSCy Schubert data[1] = (uint8_t)(msec & 0xff); 11438f76bb7dSCy Schubert return edns_opt_list_append(list, LDNS_EDNS_KEEPALIVE, sizeof(data), 11448f76bb7dSCy Schubert data, region); 11458f76bb7dSCy Schubert } 11468f76bb7dSCy Schubert 1147bc892140SDag-Erling Smørgrav int edns_opt_list_append(struct edns_option** list, uint16_t code, size_t len, 1148bc892140SDag-Erling Smørgrav uint8_t* data, struct regional* region) 1149e2d15004SDag-Erling Smørgrav { 1150bc892140SDag-Erling Smørgrav struct edns_option** prevp; 1151bc892140SDag-Erling Smørgrav struct edns_option* opt; 1152bc892140SDag-Erling Smørgrav 1153bc892140SDag-Erling Smørgrav /* allocate new element */ 1154bc892140SDag-Erling Smørgrav opt = (struct edns_option*)regional_alloc(region, sizeof(*opt)); 1155bc892140SDag-Erling Smørgrav if(!opt) 1156bc892140SDag-Erling Smørgrav return 0; 1157bc892140SDag-Erling Smørgrav opt->next = NULL; 1158bc892140SDag-Erling Smørgrav opt->opt_code = code; 1159bc892140SDag-Erling Smørgrav opt->opt_len = len; 1160bc892140SDag-Erling Smørgrav opt->opt_data = NULL; 1161bc892140SDag-Erling Smørgrav if(len > 0) { 1162bc892140SDag-Erling Smørgrav opt->opt_data = regional_alloc_init(region, data, len); 1163bc892140SDag-Erling Smørgrav if(!opt->opt_data) 1164bc892140SDag-Erling Smørgrav return 0; 1165bc892140SDag-Erling Smørgrav } 1166bc892140SDag-Erling Smørgrav 1167bc892140SDag-Erling Smørgrav /* append at end of list */ 1168bc892140SDag-Erling Smørgrav prevp = list; 1169bc892140SDag-Erling Smørgrav while(*prevp != NULL) { 1170bc892140SDag-Erling Smørgrav prevp = &((*prevp)->next); 1171bc892140SDag-Erling Smørgrav } 1172bc892140SDag-Erling Smørgrav *prevp = opt; 1173bc892140SDag-Erling Smørgrav return 1; 1174bc892140SDag-Erling Smørgrav } 1175bc892140SDag-Erling Smørgrav 1176bc892140SDag-Erling Smørgrav int edns_opt_list_remove(struct edns_option** list, uint16_t code) 1177bc892140SDag-Erling Smørgrav { 1178bc892140SDag-Erling Smørgrav /* The list should already be allocated in a region. Freeing the 1179bc892140SDag-Erling Smørgrav * allocated space in a region is not possible. We just unlink the 1180bc892140SDag-Erling Smørgrav * required elements and they will be freed together with the region. */ 1181bc892140SDag-Erling Smørgrav 1182bc892140SDag-Erling Smørgrav struct edns_option* prev; 1183bc892140SDag-Erling Smørgrav struct edns_option* curr; 1184bc892140SDag-Erling Smørgrav if(!list || !(*list)) return 0; 1185bc892140SDag-Erling Smørgrav 1186bc892140SDag-Erling Smørgrav /* Unlink and repoint if the element(s) are first in list */ 1187bc892140SDag-Erling Smørgrav while(list && *list && (*list)->opt_code == code) { 1188bc892140SDag-Erling Smørgrav *list = (*list)->next; 1189bc892140SDag-Erling Smørgrav } 1190bc892140SDag-Erling Smørgrav 1191bc892140SDag-Erling Smørgrav if(!list || !(*list)) return 1; 1192bc892140SDag-Erling Smørgrav /* Unlink elements and reattach the list */ 1193bc892140SDag-Erling Smørgrav prev = *list; 1194bc892140SDag-Erling Smørgrav curr = (*list)->next; 1195bc892140SDag-Erling Smørgrav while(curr != NULL) { 1196bc892140SDag-Erling Smørgrav if(curr->opt_code == code) { 1197bc892140SDag-Erling Smørgrav prev->next = curr->next; 1198bc892140SDag-Erling Smørgrav curr = curr->next; 1199bc892140SDag-Erling Smørgrav } else { 1200bc892140SDag-Erling Smørgrav prev = curr; 1201bc892140SDag-Erling Smørgrav curr = curr->next; 1202bc892140SDag-Erling Smørgrav } 1203bc892140SDag-Erling Smørgrav } 1204bc892140SDag-Erling Smørgrav return 1; 1205bc892140SDag-Erling Smørgrav } 1206bc892140SDag-Erling Smørgrav 1207bc892140SDag-Erling Smørgrav static int inplace_cb_reply_call_generic( 120865b390aaSDag-Erling Smørgrav struct inplace_cb* callback_list, enum inplace_cb_list_type type, 1209bc892140SDag-Erling Smørgrav struct query_info* qinfo, struct module_qstate* qstate, 1210bc892140SDag-Erling Smørgrav struct reply_info* rep, int rcode, struct edns_data* edns, 1211f44e67d1SCy Schubert struct comm_reply* repinfo, struct regional* region, 1212f44e67d1SCy Schubert struct timeval* start_time) 1213bc892140SDag-Erling Smørgrav { 121465b390aaSDag-Erling Smørgrav struct inplace_cb* cb; 1215bc892140SDag-Erling Smørgrav struct edns_option* opt_list_out = NULL; 12168a384985SDag-Erling Smørgrav #if defined(EXPORT_ALL_SYMBOLS) 12178a384985SDag-Erling Smørgrav (void)type; /* param not used when fptr_ok disabled */ 12188a384985SDag-Erling Smørgrav #endif 1219bc892140SDag-Erling Smørgrav if(qstate) 1220bc892140SDag-Erling Smørgrav opt_list_out = qstate->edns_opts_front_out; 1221bc892140SDag-Erling Smørgrav for(cb=callback_list; cb; cb=cb->next) { 122265b390aaSDag-Erling Smørgrav fptr_ok(fptr_whitelist_inplace_cb_reply_generic( 122365b390aaSDag-Erling Smørgrav (inplace_cb_reply_func_type*)cb->cb, type)); 122465b390aaSDag-Erling Smørgrav (void)(*(inplace_cb_reply_func_type*)cb->cb)(qinfo, qstate, rep, 1225f44e67d1SCy Schubert rcode, edns, &opt_list_out, repinfo, region, start_time, cb->id, cb->cb_arg); 1226bc892140SDag-Erling Smørgrav } 122724e36522SCy Schubert edns->opt_list_inplace_cb_out = opt_list_out; 1228bc892140SDag-Erling Smørgrav return 1; 1229bc892140SDag-Erling Smørgrav } 1230bc892140SDag-Erling Smørgrav 1231bc892140SDag-Erling Smørgrav int inplace_cb_reply_call(struct module_env* env, struct query_info* qinfo, 1232bc892140SDag-Erling Smørgrav struct module_qstate* qstate, struct reply_info* rep, int rcode, 1233f44e67d1SCy Schubert struct edns_data* edns, struct comm_reply* repinfo, struct regional* region, 1234f44e67d1SCy Schubert struct timeval* start_time) 1235bc892140SDag-Erling Smørgrav { 1236bc892140SDag-Erling Smørgrav return inplace_cb_reply_call_generic( 1237bc892140SDag-Erling Smørgrav env->inplace_cb_lists[inplace_cb_reply], inplace_cb_reply, qinfo, 1238f44e67d1SCy Schubert qstate, rep, rcode, edns, repinfo, region, start_time); 1239bc892140SDag-Erling Smørgrav } 1240bc892140SDag-Erling Smørgrav 1241bc892140SDag-Erling Smørgrav int inplace_cb_reply_cache_call(struct module_env* env, 1242bc892140SDag-Erling Smørgrav struct query_info* qinfo, struct module_qstate* qstate, 1243bc892140SDag-Erling Smørgrav struct reply_info* rep, int rcode, struct edns_data* edns, 1244f44e67d1SCy Schubert struct comm_reply* repinfo, struct regional* region, 1245f44e67d1SCy Schubert struct timeval* start_time) 1246bc892140SDag-Erling Smørgrav { 1247bc892140SDag-Erling Smørgrav return inplace_cb_reply_call_generic( 1248bc892140SDag-Erling Smørgrav env->inplace_cb_lists[inplace_cb_reply_cache], inplace_cb_reply_cache, 1249f44e67d1SCy Schubert qinfo, qstate, rep, rcode, edns, repinfo, region, start_time); 1250bc892140SDag-Erling Smørgrav } 1251bc892140SDag-Erling Smørgrav 1252bc892140SDag-Erling Smørgrav int inplace_cb_reply_local_call(struct module_env* env, 1253bc892140SDag-Erling Smørgrav struct query_info* qinfo, struct module_qstate* qstate, 1254bc892140SDag-Erling Smørgrav struct reply_info* rep, int rcode, struct edns_data* edns, 1255f44e67d1SCy Schubert struct comm_reply* repinfo, struct regional* region, 1256f44e67d1SCy Schubert struct timeval* start_time) 1257bc892140SDag-Erling Smørgrav { 1258bc892140SDag-Erling Smørgrav return inplace_cb_reply_call_generic( 1259bc892140SDag-Erling Smørgrav env->inplace_cb_lists[inplace_cb_reply_local], inplace_cb_reply_local, 1260f44e67d1SCy Schubert qinfo, qstate, rep, rcode, edns, repinfo, region, start_time); 1261bc892140SDag-Erling Smørgrav } 1262bc892140SDag-Erling Smørgrav 1263bc892140SDag-Erling Smørgrav int inplace_cb_reply_servfail_call(struct module_env* env, 1264bc892140SDag-Erling Smørgrav struct query_info* qinfo, struct module_qstate* qstate, 1265bc892140SDag-Erling Smørgrav struct reply_info* rep, int rcode, struct edns_data* edns, 1266f44e67d1SCy Schubert struct comm_reply* repinfo, struct regional* region, 1267f44e67d1SCy Schubert struct timeval* start_time) 1268bc892140SDag-Erling Smørgrav { 1269bc892140SDag-Erling Smørgrav /* We are going to servfail. Remove any potential edns options. */ 1270bc892140SDag-Erling Smørgrav if(qstate) 1271bc892140SDag-Erling Smørgrav qstate->edns_opts_front_out = NULL; 1272bc892140SDag-Erling Smørgrav return inplace_cb_reply_call_generic( 1273bc892140SDag-Erling Smørgrav env->inplace_cb_lists[inplace_cb_reply_servfail], 12744c75e3aaSDag-Erling Smørgrav inplace_cb_reply_servfail, qinfo, qstate, rep, rcode, edns, repinfo, 1275f44e67d1SCy Schubert region, start_time); 1276bc892140SDag-Erling Smørgrav } 1277bc892140SDag-Erling Smørgrav 1278bc892140SDag-Erling Smørgrav int inplace_cb_query_call(struct module_env* env, struct query_info* qinfo, 1279bc892140SDag-Erling Smørgrav uint16_t flags, struct sockaddr_storage* addr, socklen_t addrlen, 1280bc892140SDag-Erling Smørgrav uint8_t* zone, size_t zonelen, struct module_qstate* qstate, 1281bc892140SDag-Erling Smørgrav struct regional* region) 1282bc892140SDag-Erling Smørgrav { 128365b390aaSDag-Erling Smørgrav struct inplace_cb* cb = env->inplace_cb_lists[inplace_cb_query]; 1284bc892140SDag-Erling Smørgrav for(; cb; cb=cb->next) { 128565b390aaSDag-Erling Smørgrav fptr_ok(fptr_whitelist_inplace_cb_query( 128665b390aaSDag-Erling Smørgrav (inplace_cb_query_func_type*)cb->cb)); 128765b390aaSDag-Erling Smørgrav (void)(*(inplace_cb_query_func_type*)cb->cb)(qinfo, flags, 128865b390aaSDag-Erling Smørgrav qstate, addr, addrlen, zone, zonelen, region, 128965b390aaSDag-Erling Smørgrav cb->id, cb->cb_arg); 129065b390aaSDag-Erling Smørgrav } 129165b390aaSDag-Erling Smørgrav return 1; 129265b390aaSDag-Erling Smørgrav } 129365b390aaSDag-Erling Smørgrav 129465b390aaSDag-Erling Smørgrav int inplace_cb_edns_back_parsed_call(struct module_env* env, 129565b390aaSDag-Erling Smørgrav struct module_qstate* qstate) 129665b390aaSDag-Erling Smørgrav { 129765b390aaSDag-Erling Smørgrav struct inplace_cb* cb = 129865b390aaSDag-Erling Smørgrav env->inplace_cb_lists[inplace_cb_edns_back_parsed]; 129965b390aaSDag-Erling Smørgrav for(; cb; cb=cb->next) { 130065b390aaSDag-Erling Smørgrav fptr_ok(fptr_whitelist_inplace_cb_edns_back_parsed( 130165b390aaSDag-Erling Smørgrav (inplace_cb_edns_back_parsed_func_type*)cb->cb)); 130265b390aaSDag-Erling Smørgrav (void)(*(inplace_cb_edns_back_parsed_func_type*)cb->cb)(qstate, 130365b390aaSDag-Erling Smørgrav cb->id, cb->cb_arg); 130465b390aaSDag-Erling Smørgrav } 130565b390aaSDag-Erling Smørgrav return 1; 130665b390aaSDag-Erling Smørgrav } 130765b390aaSDag-Erling Smørgrav 130865b390aaSDag-Erling Smørgrav int inplace_cb_query_response_call(struct module_env* env, 130965b390aaSDag-Erling Smørgrav struct module_qstate* qstate, struct dns_msg* response) { 131065b390aaSDag-Erling Smørgrav struct inplace_cb* cb = 131165b390aaSDag-Erling Smørgrav env->inplace_cb_lists[inplace_cb_query_response]; 131265b390aaSDag-Erling Smørgrav for(; cb; cb=cb->next) { 131365b390aaSDag-Erling Smørgrav fptr_ok(fptr_whitelist_inplace_cb_query_response( 131465b390aaSDag-Erling Smørgrav (inplace_cb_query_response_func_type*)cb->cb)); 131565b390aaSDag-Erling Smørgrav (void)(*(inplace_cb_query_response_func_type*)cb->cb)(qstate, 131665b390aaSDag-Erling Smørgrav response, cb->id, cb->cb_arg); 1317bc892140SDag-Erling Smørgrav } 1318e2d15004SDag-Erling Smørgrav return 1; 1319e2d15004SDag-Erling Smørgrav } 1320e2d15004SDag-Erling Smørgrav 1321e2d15004SDag-Erling Smørgrav struct edns_option* edns_opt_copy_region(struct edns_option* list, 1322e2d15004SDag-Erling Smørgrav struct regional* region) 1323e2d15004SDag-Erling Smørgrav { 1324e2d15004SDag-Erling Smørgrav struct edns_option* result = NULL, *cur = NULL, *s; 1325e2d15004SDag-Erling Smørgrav while(list) { 1326e2d15004SDag-Erling Smørgrav /* copy edns option structure */ 1327e2d15004SDag-Erling Smørgrav s = regional_alloc_init(region, list, sizeof(*list)); 1328e2d15004SDag-Erling Smørgrav if(!s) return NULL; 1329e2d15004SDag-Erling Smørgrav s->next = NULL; 1330e2d15004SDag-Erling Smørgrav 1331e2d15004SDag-Erling Smørgrav /* copy option data */ 1332e2d15004SDag-Erling Smørgrav if(s->opt_data) { 1333e2d15004SDag-Erling Smørgrav s->opt_data = regional_alloc_init(region, s->opt_data, 1334e2d15004SDag-Erling Smørgrav s->opt_len); 1335e2d15004SDag-Erling Smørgrav if(!s->opt_data) 1336e2d15004SDag-Erling Smørgrav return NULL; 1337e2d15004SDag-Erling Smørgrav } 1338e2d15004SDag-Erling Smørgrav 1339e2d15004SDag-Erling Smørgrav /* link into list */ 1340e2d15004SDag-Erling Smørgrav if(cur) 1341e2d15004SDag-Erling Smørgrav cur->next = s; 1342e2d15004SDag-Erling Smørgrav else result = s; 1343e2d15004SDag-Erling Smørgrav cur = s; 1344e2d15004SDag-Erling Smørgrav 1345e2d15004SDag-Erling Smørgrav /* examine next element */ 1346e2d15004SDag-Erling Smørgrav list = list->next; 1347e2d15004SDag-Erling Smørgrav } 1348e2d15004SDag-Erling Smørgrav return result; 1349e2d15004SDag-Erling Smørgrav } 1350e2d15004SDag-Erling Smørgrav 13518f76bb7dSCy Schubert struct edns_option* edns_opt_copy_filter_region(struct edns_option* list, 13528f76bb7dSCy Schubert uint16_t* filter_list, size_t filter_list_len, struct regional* region) 13538f76bb7dSCy Schubert { 13548f76bb7dSCy Schubert struct edns_option* result = NULL, *cur = NULL, *s; 13558f76bb7dSCy Schubert size_t i; 13568f76bb7dSCy Schubert while(list) { 13578f76bb7dSCy Schubert for(i=0; i<filter_list_len; i++) 13588f76bb7dSCy Schubert if(filter_list[i] == list->opt_code) goto found; 13598f76bb7dSCy Schubert if(i == filter_list_len) goto next; 13608f76bb7dSCy Schubert found: 13618f76bb7dSCy Schubert /* copy edns option structure */ 13628f76bb7dSCy Schubert s = regional_alloc_init(region, list, sizeof(*list)); 13638f76bb7dSCy Schubert if(!s) return NULL; 13648f76bb7dSCy Schubert s->next = NULL; 13658f76bb7dSCy Schubert 13668f76bb7dSCy Schubert /* copy option data */ 13678f76bb7dSCy Schubert if(s->opt_data) { 13688f76bb7dSCy Schubert s->opt_data = regional_alloc_init(region, s->opt_data, 13698f76bb7dSCy Schubert s->opt_len); 13708f76bb7dSCy Schubert if(!s->opt_data) 13718f76bb7dSCy Schubert return NULL; 13728f76bb7dSCy Schubert } 13738f76bb7dSCy Schubert 13748f76bb7dSCy Schubert /* link into list */ 13758f76bb7dSCy Schubert if(cur) 13768f76bb7dSCy Schubert cur->next = s; 13778f76bb7dSCy Schubert else result = s; 13788f76bb7dSCy Schubert cur = s; 13798f76bb7dSCy Schubert 13808f76bb7dSCy Schubert next: 13818f76bb7dSCy Schubert /* examine next element */ 13828f76bb7dSCy Schubert list = list->next; 13838f76bb7dSCy Schubert } 13848f76bb7dSCy Schubert return result; 13858f76bb7dSCy Schubert } 13868f76bb7dSCy Schubert 1387e2d15004SDag-Erling Smørgrav int edns_opt_compare(struct edns_option* p, struct edns_option* q) 1388e2d15004SDag-Erling Smørgrav { 1389e2d15004SDag-Erling Smørgrav if(!p && !q) return 0; 1390e2d15004SDag-Erling Smørgrav if(!p) return -1; 1391e2d15004SDag-Erling Smørgrav if(!q) return 1; 1392e2d15004SDag-Erling Smørgrav log_assert(p && q); 1393e2d15004SDag-Erling Smørgrav if(p->opt_code != q->opt_code) 1394e2d15004SDag-Erling Smørgrav return (int)q->opt_code - (int)p->opt_code; 1395e2d15004SDag-Erling Smørgrav if(p->opt_len != q->opt_len) 1396e2d15004SDag-Erling Smørgrav return (int)q->opt_len - (int)p->opt_len; 1397e2d15004SDag-Erling Smørgrav if(p->opt_len != 0) 1398e2d15004SDag-Erling Smørgrav return memcmp(p->opt_data, q->opt_data, p->opt_len); 1399e2d15004SDag-Erling Smørgrav return 0; 1400e2d15004SDag-Erling Smørgrav } 1401e2d15004SDag-Erling Smørgrav 1402e2d15004SDag-Erling Smørgrav int edns_opt_list_compare(struct edns_option* p, struct edns_option* q) 1403e2d15004SDag-Erling Smørgrav { 1404e2d15004SDag-Erling Smørgrav int r; 1405e2d15004SDag-Erling Smørgrav while(p && q) { 1406e2d15004SDag-Erling Smørgrav r = edns_opt_compare(p, q); 1407e2d15004SDag-Erling Smørgrav if(r != 0) 1408e2d15004SDag-Erling Smørgrav return r; 1409e2d15004SDag-Erling Smørgrav p = p->next; 1410e2d15004SDag-Erling Smørgrav q = q->next; 1411e2d15004SDag-Erling Smørgrav } 1412e2d15004SDag-Erling Smørgrav if(p || q) { 1413e2d15004SDag-Erling Smørgrav /* uneven length lists */ 1414e2d15004SDag-Erling Smørgrav if(p) return 1; 1415e2d15004SDag-Erling Smørgrav if(q) return -1; 1416e2d15004SDag-Erling Smørgrav } 1417e2d15004SDag-Erling Smørgrav return 0; 1418e2d15004SDag-Erling Smørgrav } 1419e2d15004SDag-Erling Smørgrav 1420e2d15004SDag-Erling Smørgrav void edns_opt_list_free(struct edns_option* list) 1421e2d15004SDag-Erling Smørgrav { 1422e2d15004SDag-Erling Smørgrav struct edns_option* n; 1423e2d15004SDag-Erling Smørgrav while(list) { 1424e2d15004SDag-Erling Smørgrav free(list->opt_data); 1425e2d15004SDag-Erling Smørgrav n = list->next; 1426e2d15004SDag-Erling Smørgrav free(list); 1427e2d15004SDag-Erling Smørgrav list = n; 1428e2d15004SDag-Erling Smørgrav } 1429e2d15004SDag-Erling Smørgrav } 1430e2d15004SDag-Erling Smørgrav 1431e2d15004SDag-Erling Smørgrav struct edns_option* edns_opt_copy_alloc(struct edns_option* list) 1432e2d15004SDag-Erling Smørgrav { 1433e2d15004SDag-Erling Smørgrav struct edns_option* result = NULL, *cur = NULL, *s; 1434e2d15004SDag-Erling Smørgrav while(list) { 1435e2d15004SDag-Erling Smørgrav /* copy edns option structure */ 1436e2d15004SDag-Erling Smørgrav s = memdup(list, sizeof(*list)); 1437e2d15004SDag-Erling Smørgrav if(!s) { 1438e2d15004SDag-Erling Smørgrav edns_opt_list_free(result); 1439e2d15004SDag-Erling Smørgrav return NULL; 1440e2d15004SDag-Erling Smørgrav } 1441e2d15004SDag-Erling Smørgrav s->next = NULL; 1442e2d15004SDag-Erling Smørgrav 1443e2d15004SDag-Erling Smørgrav /* copy option data */ 1444e2d15004SDag-Erling Smørgrav if(s->opt_data) { 1445e2d15004SDag-Erling Smørgrav s->opt_data = memdup(s->opt_data, s->opt_len); 1446e2d15004SDag-Erling Smørgrav if(!s->opt_data) { 144765b390aaSDag-Erling Smørgrav free(s); 1448e2d15004SDag-Erling Smørgrav edns_opt_list_free(result); 1449e2d15004SDag-Erling Smørgrav return NULL; 1450e2d15004SDag-Erling Smørgrav } 1451e2d15004SDag-Erling Smørgrav } 1452e2d15004SDag-Erling Smørgrav 1453e2d15004SDag-Erling Smørgrav /* link into list */ 1454e2d15004SDag-Erling Smørgrav if(cur) 1455e2d15004SDag-Erling Smørgrav cur->next = s; 1456e2d15004SDag-Erling Smørgrav else result = s; 1457e2d15004SDag-Erling Smørgrav cur = s; 1458e2d15004SDag-Erling Smørgrav 1459e2d15004SDag-Erling Smørgrav /* examine next element */ 1460e2d15004SDag-Erling Smørgrav list = list->next; 1461e2d15004SDag-Erling Smørgrav } 1462e2d15004SDag-Erling Smørgrav return result; 1463e2d15004SDag-Erling Smørgrav } 1464e2d15004SDag-Erling Smørgrav 1465bc892140SDag-Erling Smørgrav struct edns_option* edns_opt_list_find(struct edns_option* list, uint16_t code) 1466e2d15004SDag-Erling Smørgrav { 1467e2d15004SDag-Erling Smørgrav struct edns_option* p; 1468e2d15004SDag-Erling Smørgrav for(p=list; p; p=p->next) { 1469e2d15004SDag-Erling Smørgrav if(p->opt_code == code) 1470e2d15004SDag-Erling Smørgrav return p; 1471e2d15004SDag-Erling Smørgrav } 1472e2d15004SDag-Erling Smørgrav return NULL; 1473e2d15004SDag-Erling Smørgrav } 1474