1b7579f77SDag-Erling Smørgrav /* 2b7579f77SDag-Erling Smørgrav * services/mesh.c - deal with mesh of query states and handle events for that. 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 functions to assist in dealing with a mesh of 40b7579f77SDag-Erling Smørgrav * query states. This mesh is supposed to be thread-specific. 41b7579f77SDag-Erling Smørgrav * It consists of query states (per qname, qtype, qclass) and connections 42b7579f77SDag-Erling Smørgrav * between query states and the super and subquery states, and replies to 43b7579f77SDag-Erling Smørgrav * send back to clients. 44b7579f77SDag-Erling Smørgrav */ 45b7579f77SDag-Erling Smørgrav #include "config.h" 46b7579f77SDag-Erling Smørgrav #include "services/mesh.h" 47b7579f77SDag-Erling Smørgrav #include "services/outbound_list.h" 48b7579f77SDag-Erling Smørgrav #include "services/cache/dns.h" 49091e9e46SCy Schubert #include "services/cache/rrset.h" 50b7579f77SDag-Erling Smørgrav #include "util/log.h" 51b7579f77SDag-Erling Smørgrav #include "util/net_help.h" 52b7579f77SDag-Erling Smørgrav #include "util/module.h" 53b7579f77SDag-Erling Smørgrav #include "util/regional.h" 54b7579f77SDag-Erling Smørgrav #include "util/data/msgencode.h" 55b7579f77SDag-Erling Smørgrav #include "util/timehist.h" 56b7579f77SDag-Erling Smørgrav #include "util/fptr_wlist.h" 57b7579f77SDag-Erling Smørgrav #include "util/alloc.h" 58b7579f77SDag-Erling Smørgrav #include "util/config_file.h" 594c75e3aaSDag-Erling Smørgrav #include "util/edns.h" 6009a3aaf3SDag-Erling Smørgrav #include "sldns/sbuffer.h" 61bc892140SDag-Erling Smørgrav #include "sldns/wire2str.h" 62bc892140SDag-Erling Smørgrav #include "services/localzone.h" 63bc892140SDag-Erling Smørgrav #include "util/data/dname.h" 6465b390aaSDag-Erling Smørgrav #include "respip/respip.h" 65e86b9096SDag-Erling Smørgrav #include "services/listen_dnsport.h" 668f76bb7dSCy Schubert #include "util/timeval_func.h" 67b7579f77SDag-Erling Smørgrav 68a39a5a69SCy Schubert #ifdef CLIENT_SUBNET 69a39a5a69SCy Schubert #include "edns-subnet/subnetmod.h" 70a39a5a69SCy Schubert #include "edns-subnet/edns-subnet.h" 71a39a5a69SCy Schubert #endif 72a39a5a69SCy Schubert 73091e9e46SCy Schubert /** 7465b390aaSDag-Erling Smørgrav * Compare two response-ip client info entries for the purpose of mesh state 7565b390aaSDag-Erling Smørgrav * compare. It returns 0 if ci_a and ci_b are considered equal; otherwise 7665b390aaSDag-Erling Smørgrav * 1 or -1 (they mean 'ci_a is larger/smaller than ci_b', respectively, but 7765b390aaSDag-Erling Smørgrav * in practice it should be only used to mean they are different). 7865b390aaSDag-Erling Smørgrav * We cannot share the mesh state for two queries if different response-ip 7965b390aaSDag-Erling Smørgrav * actions can apply in the end, even if those queries are otherwise identical. 8065b390aaSDag-Erling Smørgrav * For this purpose we compare tag lists and tag action lists; they should be 8165b390aaSDag-Erling Smørgrav * identical to share the same state. 8265b390aaSDag-Erling Smørgrav * For tag data, we don't look into the data content, as it can be 8365b390aaSDag-Erling Smørgrav * expensive; unless tag data are not defined for both or they point to the 8465b390aaSDag-Erling Smørgrav * exact same data in memory (i.e., they come from the same ACL entry), we 8565b390aaSDag-Erling Smørgrav * consider these data different. 8665b390aaSDag-Erling Smørgrav * Likewise, if the client info is associated with views, we don't look into 8765b390aaSDag-Erling Smørgrav * the views. They are considered different unless they are exactly the same 8865b390aaSDag-Erling Smørgrav * even if the views only differ in the names. 8965b390aaSDag-Erling Smørgrav */ 9065b390aaSDag-Erling Smørgrav static int 9165b390aaSDag-Erling Smørgrav client_info_compare(const struct respip_client_info* ci_a, 9265b390aaSDag-Erling Smørgrav const struct respip_client_info* ci_b) 9365b390aaSDag-Erling Smørgrav { 9465b390aaSDag-Erling Smørgrav int cmp; 9565b390aaSDag-Erling Smørgrav 9665b390aaSDag-Erling Smørgrav if(!ci_a && !ci_b) 9765b390aaSDag-Erling Smørgrav return 0; 9865b390aaSDag-Erling Smørgrav if(ci_a && !ci_b) 9965b390aaSDag-Erling Smørgrav return -1; 10065b390aaSDag-Erling Smørgrav if(!ci_a && ci_b) 10165b390aaSDag-Erling Smørgrav return 1; 10265b390aaSDag-Erling Smørgrav if(ci_a->taglen != ci_b->taglen) 10365b390aaSDag-Erling Smørgrav return (ci_a->taglen < ci_b->taglen) ? -1 : 1; 10425039b37SCy Schubert if(ci_a->taglist && !ci_b->taglist) 10525039b37SCy Schubert return -1; 10625039b37SCy Schubert if(!ci_a->taglist && ci_b->taglist) 10725039b37SCy Schubert return 1; 10825039b37SCy Schubert if(ci_a->taglist && ci_b->taglist) { 10965b390aaSDag-Erling Smørgrav cmp = memcmp(ci_a->taglist, ci_b->taglist, ci_a->taglen); 11065b390aaSDag-Erling Smørgrav if(cmp != 0) 11165b390aaSDag-Erling Smørgrav return cmp; 11225039b37SCy Schubert } 11365b390aaSDag-Erling Smørgrav if(ci_a->tag_actions_size != ci_b->tag_actions_size) 11465b390aaSDag-Erling Smørgrav return (ci_a->tag_actions_size < ci_b->tag_actions_size) ? 11565b390aaSDag-Erling Smørgrav -1 : 1; 11625039b37SCy Schubert if(ci_a->tag_actions && !ci_b->tag_actions) 11725039b37SCy Schubert return -1; 11825039b37SCy Schubert if(!ci_a->tag_actions && ci_b->tag_actions) 11925039b37SCy Schubert return 1; 12025039b37SCy Schubert if(ci_a->tag_actions && ci_b->tag_actions) { 12165b390aaSDag-Erling Smørgrav cmp = memcmp(ci_a->tag_actions, ci_b->tag_actions, 12265b390aaSDag-Erling Smørgrav ci_a->tag_actions_size); 12365b390aaSDag-Erling Smørgrav if(cmp != 0) 12465b390aaSDag-Erling Smørgrav return cmp; 12525039b37SCy Schubert } 12665b390aaSDag-Erling Smørgrav if(ci_a->tag_datas != ci_b->tag_datas) 12765b390aaSDag-Erling Smørgrav return ci_a->tag_datas < ci_b->tag_datas ? -1 : 1; 12865b390aaSDag-Erling Smørgrav if(ci_a->view != ci_b->view) 12965b390aaSDag-Erling Smørgrav return ci_a->view < ci_b->view ? -1 : 1; 13065b390aaSDag-Erling Smørgrav /* For the unbound daemon these should be non-NULL and identical, 13165b390aaSDag-Erling Smørgrav * but we check that just in case. */ 13265b390aaSDag-Erling Smørgrav if(ci_a->respip_set != ci_b->respip_set) 13365b390aaSDag-Erling Smørgrav return ci_a->respip_set < ci_b->respip_set ? -1 : 1; 13465b390aaSDag-Erling Smørgrav return 0; 13565b390aaSDag-Erling Smørgrav } 13665b390aaSDag-Erling Smørgrav 137b7579f77SDag-Erling Smørgrav int 138b7579f77SDag-Erling Smørgrav mesh_state_compare(const void* ap, const void* bp) 139b7579f77SDag-Erling Smørgrav { 140b7579f77SDag-Erling Smørgrav struct mesh_state* a = (struct mesh_state*)ap; 141b7579f77SDag-Erling Smørgrav struct mesh_state* b = (struct mesh_state*)bp; 14265b390aaSDag-Erling Smørgrav int cmp; 143b7579f77SDag-Erling Smørgrav 144bc892140SDag-Erling Smørgrav if(a->unique < b->unique) 145bc892140SDag-Erling Smørgrav return -1; 146bc892140SDag-Erling Smørgrav if(a->unique > b->unique) 147bc892140SDag-Erling Smørgrav return 1; 148bc892140SDag-Erling Smørgrav 149b7579f77SDag-Erling Smørgrav if(a->s.is_priming && !b->s.is_priming) 150b7579f77SDag-Erling Smørgrav return -1; 151b7579f77SDag-Erling Smørgrav if(!a->s.is_priming && b->s.is_priming) 152b7579f77SDag-Erling Smørgrav return 1; 153b7579f77SDag-Erling Smørgrav 154ff825849SDag-Erling Smørgrav if(a->s.is_valrec && !b->s.is_valrec) 155ff825849SDag-Erling Smørgrav return -1; 156ff825849SDag-Erling Smørgrav if(!a->s.is_valrec && b->s.is_valrec) 157ff825849SDag-Erling Smørgrav return 1; 158ff825849SDag-Erling Smørgrav 159b7579f77SDag-Erling Smørgrav if((a->s.query_flags&BIT_RD) && !(b->s.query_flags&BIT_RD)) 160b7579f77SDag-Erling Smørgrav return -1; 161b7579f77SDag-Erling Smørgrav if(!(a->s.query_flags&BIT_RD) && (b->s.query_flags&BIT_RD)) 162b7579f77SDag-Erling Smørgrav return 1; 163b7579f77SDag-Erling Smørgrav 164b7579f77SDag-Erling Smørgrav if((a->s.query_flags&BIT_CD) && !(b->s.query_flags&BIT_CD)) 165b7579f77SDag-Erling Smørgrav return -1; 166b7579f77SDag-Erling Smørgrav if(!(a->s.query_flags&BIT_CD) && (b->s.query_flags&BIT_CD)) 167b7579f77SDag-Erling Smørgrav return 1; 168b7579f77SDag-Erling Smørgrav 16965b390aaSDag-Erling Smørgrav cmp = query_info_compare(&a->s.qinfo, &b->s.qinfo); 17065b390aaSDag-Erling Smørgrav if(cmp != 0) 17165b390aaSDag-Erling Smørgrav return cmp; 17265b390aaSDag-Erling Smørgrav return client_info_compare(a->s.client_info, b->s.client_info); 173b7579f77SDag-Erling Smørgrav } 174b7579f77SDag-Erling Smørgrav 175b7579f77SDag-Erling Smørgrav int 176b7579f77SDag-Erling Smørgrav mesh_state_ref_compare(const void* ap, const void* bp) 177b7579f77SDag-Erling Smørgrav { 178b7579f77SDag-Erling Smørgrav struct mesh_state_ref* a = (struct mesh_state_ref*)ap; 179b7579f77SDag-Erling Smørgrav struct mesh_state_ref* b = (struct mesh_state_ref*)bp; 180b7579f77SDag-Erling Smørgrav return mesh_state_compare(a->s, b->s); 181b7579f77SDag-Erling Smørgrav } 182b7579f77SDag-Erling Smørgrav 183b7579f77SDag-Erling Smørgrav struct mesh_area* 184b7579f77SDag-Erling Smørgrav mesh_create(struct module_stack* stack, struct module_env* env) 185b7579f77SDag-Erling Smørgrav { 186b7579f77SDag-Erling Smørgrav struct mesh_area* mesh = calloc(1, sizeof(struct mesh_area)); 187b7579f77SDag-Erling Smørgrav if(!mesh) { 188b7579f77SDag-Erling Smørgrav log_err("mesh area alloc: out of memory"); 189b7579f77SDag-Erling Smørgrav return NULL; 190b7579f77SDag-Erling Smørgrav } 191b7579f77SDag-Erling Smørgrav mesh->histogram = timehist_setup(); 19217d15b25SDag-Erling Smørgrav mesh->qbuf_bak = sldns_buffer_new(env->cfg->msg_buffer_size); 193b7579f77SDag-Erling Smørgrav if(!mesh->histogram || !mesh->qbuf_bak) { 194b7579f77SDag-Erling Smørgrav free(mesh); 195b7579f77SDag-Erling Smørgrav log_err("mesh area alloc: out of memory"); 196b7579f77SDag-Erling Smørgrav return NULL; 197b7579f77SDag-Erling Smørgrav } 198b7579f77SDag-Erling Smørgrav mesh->mods = *stack; 199b7579f77SDag-Erling Smørgrav mesh->env = env; 200b7579f77SDag-Erling Smørgrav rbtree_init(&mesh->run, &mesh_state_compare); 201b7579f77SDag-Erling Smørgrav rbtree_init(&mesh->all, &mesh_state_compare); 202b7579f77SDag-Erling Smørgrav mesh->num_reply_addrs = 0; 203b7579f77SDag-Erling Smørgrav mesh->num_reply_states = 0; 204b7579f77SDag-Erling Smørgrav mesh->num_detached_states = 0; 205b7579f77SDag-Erling Smørgrav mesh->num_forever_states = 0; 206b7579f77SDag-Erling Smørgrav mesh->stats_jostled = 0; 207b7579f77SDag-Erling Smørgrav mesh->stats_dropped = 0; 208091e9e46SCy Schubert mesh->ans_expired = 0; 2098f76bb7dSCy Schubert mesh->ans_cachedb = 0; 210b7579f77SDag-Erling Smørgrav mesh->max_reply_states = env->cfg->num_queries_per_thread; 211b7579f77SDag-Erling Smørgrav mesh->max_forever_states = (mesh->max_reply_states+1)/2; 212b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S 213b7579f77SDag-Erling Smørgrav mesh->jostle_max.tv_sec = (time_t)(env->cfg->jostle_time / 1000); 214b7579f77SDag-Erling Smørgrav mesh->jostle_max.tv_usec = (time_t)((env->cfg->jostle_time % 1000) 215b7579f77SDag-Erling Smørgrav *1000); 216b7579f77SDag-Erling Smørgrav #endif 217b7579f77SDag-Erling Smørgrav return mesh; 218b7579f77SDag-Erling Smørgrav } 219b7579f77SDag-Erling Smørgrav 220b7579f77SDag-Erling Smørgrav /** help mesh delete delete mesh states */ 221b7579f77SDag-Erling Smørgrav static void 2223005e0a3SDag-Erling Smørgrav mesh_delete_helper(rbnode_type* n) 223b7579f77SDag-Erling Smørgrav { 224b7579f77SDag-Erling Smørgrav struct mesh_state* mstate = (struct mesh_state*)n->key; 225b7579f77SDag-Erling Smørgrav /* perform a full delete, not only 'cleanup' routine, 226b7579f77SDag-Erling Smørgrav * because other callbacks expect a clean state in the mesh. 227b7579f77SDag-Erling Smørgrav * For 're-entrant' calls */ 228b7579f77SDag-Erling Smørgrav mesh_state_delete(&mstate->s); 229b7579f77SDag-Erling Smørgrav /* but because these delete the items from the tree, postorder 230b7579f77SDag-Erling Smørgrav * traversal and rbtree rebalancing do not work together */ 231b7579f77SDag-Erling Smørgrav } 232b7579f77SDag-Erling Smørgrav 233b7579f77SDag-Erling Smørgrav void 234b7579f77SDag-Erling Smørgrav mesh_delete(struct mesh_area* mesh) 235b7579f77SDag-Erling Smørgrav { 236b7579f77SDag-Erling Smørgrav if(!mesh) 237b7579f77SDag-Erling Smørgrav return; 238b7579f77SDag-Erling Smørgrav /* free all query states */ 239b7579f77SDag-Erling Smørgrav while(mesh->all.count) 240b7579f77SDag-Erling Smørgrav mesh_delete_helper(mesh->all.root); 241b7579f77SDag-Erling Smørgrav timehist_delete(mesh->histogram); 24217d15b25SDag-Erling Smørgrav sldns_buffer_free(mesh->qbuf_bak); 243b7579f77SDag-Erling Smørgrav free(mesh); 244b7579f77SDag-Erling Smørgrav } 245b7579f77SDag-Erling Smørgrav 246b7579f77SDag-Erling Smørgrav void 247b7579f77SDag-Erling Smørgrav mesh_delete_all(struct mesh_area* mesh) 248b7579f77SDag-Erling Smørgrav { 249b7579f77SDag-Erling Smørgrav /* free all query states */ 250b7579f77SDag-Erling Smørgrav while(mesh->all.count) 251b7579f77SDag-Erling Smørgrav mesh_delete_helper(mesh->all.root); 252b7579f77SDag-Erling Smørgrav mesh->stats_dropped += mesh->num_reply_addrs; 253b7579f77SDag-Erling Smørgrav /* clear mesh area references */ 254b7579f77SDag-Erling Smørgrav rbtree_init(&mesh->run, &mesh_state_compare); 255b7579f77SDag-Erling Smørgrav rbtree_init(&mesh->all, &mesh_state_compare); 256b7579f77SDag-Erling Smørgrav mesh->num_reply_addrs = 0; 257b7579f77SDag-Erling Smørgrav mesh->num_reply_states = 0; 258b7579f77SDag-Erling Smørgrav mesh->num_detached_states = 0; 259b7579f77SDag-Erling Smørgrav mesh->num_forever_states = 0; 260b7579f77SDag-Erling Smørgrav mesh->forever_first = NULL; 261b7579f77SDag-Erling Smørgrav mesh->forever_last = NULL; 262b7579f77SDag-Erling Smørgrav mesh->jostle_first = NULL; 263b7579f77SDag-Erling Smørgrav mesh->jostle_last = NULL; 264b7579f77SDag-Erling Smørgrav } 265b7579f77SDag-Erling Smørgrav 26617d15b25SDag-Erling Smørgrav int mesh_make_new_space(struct mesh_area* mesh, sldns_buffer* qbuf) 267b7579f77SDag-Erling Smørgrav { 268b7579f77SDag-Erling Smørgrav struct mesh_state* m = mesh->jostle_first; 269b7579f77SDag-Erling Smørgrav /* free space is available */ 270b7579f77SDag-Erling Smørgrav if(mesh->num_reply_states < mesh->max_reply_states) 271b7579f77SDag-Erling Smørgrav return 1; 272b7579f77SDag-Erling Smørgrav /* try to kick out a jostle-list item */ 273b7579f77SDag-Erling Smørgrav if(m && m->reply_list && m->list_select == mesh_jostle_list) { 274b7579f77SDag-Erling Smørgrav /* how old is it? */ 275b7579f77SDag-Erling Smørgrav struct timeval age; 276b7579f77SDag-Erling Smørgrav timeval_subtract(&age, mesh->env->now_tv, 277b7579f77SDag-Erling Smørgrav &m->reply_list->start_time); 278b7579f77SDag-Erling Smørgrav if(timeval_smaller(&mesh->jostle_max, &age)) { 279b7579f77SDag-Erling Smørgrav /* its a goner */ 280b7579f77SDag-Erling Smørgrav log_nametypeclass(VERB_ALGO, "query jostled out to " 281b7579f77SDag-Erling Smørgrav "make space for a new one", 282b7579f77SDag-Erling Smørgrav m->s.qinfo.qname, m->s.qinfo.qtype, 283b7579f77SDag-Erling Smørgrav m->s.qinfo.qclass); 284b7579f77SDag-Erling Smørgrav /* backup the query */ 28517d15b25SDag-Erling Smørgrav if(qbuf) sldns_buffer_copy(mesh->qbuf_bak, qbuf); 286b7579f77SDag-Erling Smørgrav /* notify supers */ 287b7579f77SDag-Erling Smørgrav if(m->super_set.count > 0) { 288b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "notify supers of failure"); 289b7579f77SDag-Erling Smørgrav m->s.return_msg = NULL; 290b7579f77SDag-Erling Smørgrav m->s.return_rcode = LDNS_RCODE_SERVFAIL; 291b7579f77SDag-Erling Smørgrav mesh_walk_supers(mesh, m); 292b7579f77SDag-Erling Smørgrav } 293b7579f77SDag-Erling Smørgrav mesh->stats_jostled ++; 294b7579f77SDag-Erling Smørgrav mesh_state_delete(&m->s); 295b7579f77SDag-Erling Smørgrav /* restore the query - note that the qinfo ptr to 296b7579f77SDag-Erling Smørgrav * the querybuffer is then correct again. */ 29717d15b25SDag-Erling Smørgrav if(qbuf) sldns_buffer_copy(qbuf, mesh->qbuf_bak); 298b7579f77SDag-Erling Smørgrav return 1; 299b7579f77SDag-Erling Smørgrav } 300b7579f77SDag-Erling Smørgrav } 301b7579f77SDag-Erling Smørgrav /* no space for new item */ 302b7579f77SDag-Erling Smørgrav return 0; 303b7579f77SDag-Erling Smørgrav } 304b7579f77SDag-Erling Smørgrav 305091e9e46SCy Schubert struct dns_msg* 306091e9e46SCy Schubert mesh_serve_expired_lookup(struct module_qstate* qstate, 307091e9e46SCy Schubert struct query_info* lookup_qinfo) 308091e9e46SCy Schubert { 309091e9e46SCy Schubert hashvalue_type h; 310091e9e46SCy Schubert struct lruhash_entry* e; 311091e9e46SCy Schubert struct dns_msg* msg; 312091e9e46SCy Schubert struct reply_info* data; 313091e9e46SCy Schubert struct msgreply_entry* key; 314091e9e46SCy Schubert time_t timenow = *qstate->env->now; 315091e9e46SCy Schubert int must_validate = (!(qstate->query_flags&BIT_CD) 316091e9e46SCy Schubert || qstate->env->cfg->ignore_cd) && qstate->env->need_to_validate; 317091e9e46SCy Schubert /* Lookup cache */ 318091e9e46SCy Schubert h = query_info_hash(lookup_qinfo, qstate->query_flags); 319091e9e46SCy Schubert e = slabhash_lookup(qstate->env->msg_cache, h, lookup_qinfo, 0); 320091e9e46SCy Schubert if(!e) return NULL; 321091e9e46SCy Schubert 322091e9e46SCy Schubert key = (struct msgreply_entry*)e->key; 323091e9e46SCy Schubert data = (struct reply_info*)e->data; 324091e9e46SCy Schubert msg = tomsg(qstate->env, &key->key, data, qstate->region, timenow, 325091e9e46SCy Schubert qstate->env->cfg->serve_expired, qstate->env->scratch); 326091e9e46SCy Schubert if(!msg) 327091e9e46SCy Schubert goto bail_out; 328091e9e46SCy Schubert 329091e9e46SCy Schubert /* Check CNAME chain (if any) 330091e9e46SCy Schubert * This is part of tomsg above; no need to check now. */ 331091e9e46SCy Schubert 332091e9e46SCy Schubert /* Check security status of the cached answer. 333091e9e46SCy Schubert * tomsg above has a subset of these checks, so we are leaving 334091e9e46SCy Schubert * these as is. 335091e9e46SCy Schubert * In case of bogus or revalidation we don't care to reply here. */ 336091e9e46SCy Schubert if(must_validate && (msg->rep->security == sec_status_bogus || 337091e9e46SCy Schubert msg->rep->security == sec_status_secure_sentinel_fail)) { 338091e9e46SCy Schubert verbose(VERB_ALGO, "Serve expired: bogus answer found in cache"); 339091e9e46SCy Schubert goto bail_out; 340091e9e46SCy Schubert } else if(msg->rep->security == sec_status_unchecked && must_validate) { 341091e9e46SCy Schubert verbose(VERB_ALGO, "Serve expired: unchecked entry needs " 342091e9e46SCy Schubert "validation"); 343091e9e46SCy Schubert goto bail_out; /* need to validate cache entry first */ 344091e9e46SCy Schubert } else if(msg->rep->security == sec_status_secure && 345091e9e46SCy Schubert !reply_all_rrsets_secure(msg->rep) && must_validate) { 346091e9e46SCy Schubert verbose(VERB_ALGO, "Serve expired: secure entry" 347091e9e46SCy Schubert " changed status"); 348091e9e46SCy Schubert goto bail_out; /* rrset changed, re-verify */ 349091e9e46SCy Schubert } 350091e9e46SCy Schubert 351091e9e46SCy Schubert lock_rw_unlock(&e->lock); 352091e9e46SCy Schubert return msg; 353091e9e46SCy Schubert 354091e9e46SCy Schubert bail_out: 355091e9e46SCy Schubert lock_rw_unlock(&e->lock); 356091e9e46SCy Schubert return NULL; 357091e9e46SCy Schubert } 358091e9e46SCy Schubert 359091e9e46SCy Schubert 360091e9e46SCy Schubert /** Init the serve expired data structure */ 361091e9e46SCy Schubert static int 362091e9e46SCy Schubert mesh_serve_expired_init(struct mesh_state* mstate, int timeout) 363091e9e46SCy Schubert { 364091e9e46SCy Schubert struct timeval t; 365091e9e46SCy Schubert 366091e9e46SCy Schubert /* Create serve_expired_data if not there yet */ 367091e9e46SCy Schubert if(!mstate->s.serve_expired_data) { 368091e9e46SCy Schubert mstate->s.serve_expired_data = (struct serve_expired_data*) 369091e9e46SCy Schubert regional_alloc_zero( 370091e9e46SCy Schubert mstate->s.region, sizeof(struct serve_expired_data)); 371091e9e46SCy Schubert if(!mstate->s.serve_expired_data) 372091e9e46SCy Schubert return 0; 373091e9e46SCy Schubert } 374091e9e46SCy Schubert 375091e9e46SCy Schubert /* Don't overwrite the function if already set */ 376091e9e46SCy Schubert mstate->s.serve_expired_data->get_cached_answer = 377091e9e46SCy Schubert mstate->s.serve_expired_data->get_cached_answer? 378091e9e46SCy Schubert mstate->s.serve_expired_data->get_cached_answer: 3795469a995SCy Schubert &mesh_serve_expired_lookup; 380091e9e46SCy Schubert 381091e9e46SCy Schubert /* In case this timer already popped, start it again */ 382091e9e46SCy Schubert if(!mstate->s.serve_expired_data->timer) { 383091e9e46SCy Schubert mstate->s.serve_expired_data->timer = comm_timer_create( 384091e9e46SCy Schubert mstate->s.env->worker_base, mesh_serve_expired_callback, mstate); 385091e9e46SCy Schubert if(!mstate->s.serve_expired_data->timer) 386091e9e46SCy Schubert return 0; 387091e9e46SCy Schubert #ifndef S_SPLINT_S 388091e9e46SCy Schubert t.tv_sec = timeout/1000; 389091e9e46SCy Schubert t.tv_usec = (timeout%1000)*1000; 390091e9e46SCy Schubert #endif 391091e9e46SCy Schubert comm_timer_set(mstate->s.serve_expired_data->timer, &t); 392091e9e46SCy Schubert } 393091e9e46SCy Schubert return 1; 394091e9e46SCy Schubert } 395091e9e46SCy Schubert 396b7579f77SDag-Erling Smørgrav void mesh_new_client(struct mesh_area* mesh, struct query_info* qinfo, 39765b390aaSDag-Erling Smørgrav struct respip_client_info* cinfo, uint16_t qflags, 398a39a5a69SCy Schubert struct edns_data* edns, struct comm_reply* rep, uint16_t qid, 399a39a5a69SCy Schubert int rpz_passthru) 400b7579f77SDag-Erling Smørgrav { 401bc892140SDag-Erling Smørgrav struct mesh_state* s = NULL; 40224e36522SCy Schubert int unique = unique_mesh_state(edns->opt_list_in, mesh->env); 403b7579f77SDag-Erling Smørgrav int was_detached = 0; 404b7579f77SDag-Erling Smørgrav int was_noreply = 0; 405b7579f77SDag-Erling Smørgrav int added = 0; 406091e9e46SCy Schubert int timeout = mesh->env->cfg->serve_expired? 407091e9e46SCy Schubert mesh->env->cfg->serve_expired_client_timeout:0; 408e86b9096SDag-Erling Smørgrav struct sldns_buffer* r_buffer = rep->c->buffer; 409e86b9096SDag-Erling Smørgrav if(rep->c->tcp_req_info) { 410e86b9096SDag-Erling Smørgrav r_buffer = rep->c->tcp_req_info->spool_buffer; 411e86b9096SDag-Erling Smørgrav } 412bc892140SDag-Erling Smørgrav if(!unique) 41365b390aaSDag-Erling Smørgrav s = mesh_area_find(mesh, cinfo, qinfo, qflags&(BIT_RD|BIT_CD), 0, 0); 414b7579f77SDag-Erling Smørgrav /* does this create a new reply state? */ 415b7579f77SDag-Erling Smørgrav if(!s || s->list_select == mesh_no_list) { 416b7579f77SDag-Erling Smørgrav if(!mesh_make_new_space(mesh, rep->c->buffer)) { 417b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "Too many queries. dropping " 418b7579f77SDag-Erling Smørgrav "incoming query."); 419b7579f77SDag-Erling Smørgrav comm_point_drop_reply(rep); 420b7579f77SDag-Erling Smørgrav mesh->stats_dropped++; 421b7579f77SDag-Erling Smørgrav return; 422b7579f77SDag-Erling Smørgrav } 423b7579f77SDag-Erling Smørgrav /* for this new reply state, the reply address is free, 424b7579f77SDag-Erling Smørgrav * so the limit of reply addresses does not stop reply states*/ 425b7579f77SDag-Erling Smørgrav } else { 426b7579f77SDag-Erling Smørgrav /* protect our memory usage from storing reply addresses */ 427b7579f77SDag-Erling Smørgrav if(mesh->num_reply_addrs > mesh->max_reply_states*16) { 428b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "Too many requests queued. " 429b7579f77SDag-Erling Smørgrav "dropping incoming query."); 430b7579f77SDag-Erling Smørgrav comm_point_drop_reply(rep); 431091e9e46SCy Schubert mesh->stats_dropped++; 432b7579f77SDag-Erling Smørgrav return; 433b7579f77SDag-Erling Smørgrav } 434b7579f77SDag-Erling Smørgrav } 435b7579f77SDag-Erling Smørgrav /* see if it already exists, if not, create one */ 436b7579f77SDag-Erling Smørgrav if(!s) { 437b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 4383005e0a3SDag-Erling Smørgrav struct rbnode_type* n; 439b7579f77SDag-Erling Smørgrav #endif 44065b390aaSDag-Erling Smørgrav s = mesh_state_create(mesh->env, qinfo, cinfo, 44165b390aaSDag-Erling Smørgrav qflags&(BIT_RD|BIT_CD), 0, 0); 442b7579f77SDag-Erling Smørgrav if(!s) { 443b7579f77SDag-Erling Smørgrav log_err("mesh_state_create: out of memory; SERVFAIL"); 444bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_servfail_call(mesh->env, qinfo, NULL, NULL, 445f44e67d1SCy Schubert LDNS_RCODE_SERVFAIL, edns, rep, mesh->env->scratch, mesh->env->now_tv)) 44624e36522SCy Schubert edns->opt_list_inplace_cb_out = NULL; 447e86b9096SDag-Erling Smørgrav error_encode(r_buffer, LDNS_RCODE_SERVFAIL, 448b7579f77SDag-Erling Smørgrav qinfo, qid, qflags, edns); 449b7579f77SDag-Erling Smørgrav comm_point_send_reply(rep); 450b7579f77SDag-Erling Smørgrav return; 451b7579f77SDag-Erling Smørgrav } 4528f76bb7dSCy Schubert /* set detached (it is now) */ 4538f76bb7dSCy Schubert mesh->num_detached_states++; 454bc892140SDag-Erling Smørgrav if(unique) 455bc892140SDag-Erling Smørgrav mesh_state_make_unique(s); 456a39a5a69SCy Schubert s->s.rpz_passthru = rpz_passthru; 457bc892140SDag-Erling Smørgrav /* copy the edns options we got from the front */ 45824e36522SCy Schubert if(edns->opt_list_in) { 45924e36522SCy Schubert s->s.edns_opts_front_in = edns_opt_copy_region(edns->opt_list_in, 460bc892140SDag-Erling Smørgrav s->s.region); 461bc892140SDag-Erling Smørgrav if(!s->s.edns_opts_front_in) { 4628f76bb7dSCy Schubert log_err("edns_opt_copy_region: out of memory; SERVFAIL"); 463bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_servfail_call(mesh->env, qinfo, NULL, 464f44e67d1SCy Schubert NULL, LDNS_RCODE_SERVFAIL, edns, rep, mesh->env->scratch, mesh->env->now_tv)) 46524e36522SCy Schubert edns->opt_list_inplace_cb_out = NULL; 466e86b9096SDag-Erling Smørgrav error_encode(r_buffer, LDNS_RCODE_SERVFAIL, 467bc892140SDag-Erling Smørgrav qinfo, qid, qflags, edns); 468bc892140SDag-Erling Smørgrav comm_point_send_reply(rep); 4698f76bb7dSCy Schubert mesh_state_delete(&s->s); 470bc892140SDag-Erling Smørgrav return; 471bc892140SDag-Erling Smørgrav } 472bc892140SDag-Erling Smørgrav } 473bc892140SDag-Erling Smørgrav 474b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 475b7579f77SDag-Erling Smørgrav n = 47617d15b25SDag-Erling Smørgrav #else 47717d15b25SDag-Erling Smørgrav (void) 478b7579f77SDag-Erling Smørgrav #endif 479b7579f77SDag-Erling Smørgrav rbtree_insert(&mesh->all, &s->node); 480b7579f77SDag-Erling Smørgrav log_assert(n != NULL); 481b7579f77SDag-Erling Smørgrav added = 1; 482b7579f77SDag-Erling Smørgrav } 483091e9e46SCy Schubert if(!s->reply_list && !s->cb_list) { 484b7579f77SDag-Erling Smørgrav was_noreply = 1; 485091e9e46SCy Schubert if(s->super_set.count == 0) { 486091e9e46SCy Schubert was_detached = 1; 487091e9e46SCy Schubert } 488091e9e46SCy Schubert } 489b7579f77SDag-Erling Smørgrav /* add reply to s */ 490bc892140SDag-Erling Smørgrav if(!mesh_state_add_reply(s, edns, rep, qid, qflags, qinfo)) { 491b7579f77SDag-Erling Smørgrav log_err("mesh_new_client: out of memory; SERVFAIL"); 492091e9e46SCy Schubert goto servfail_mem; 493b7579f77SDag-Erling Smørgrav } 494e86b9096SDag-Erling Smørgrav if(rep->c->tcp_req_info) { 495e86b9096SDag-Erling Smørgrav if(!tcp_req_info_add_meshstate(rep->c->tcp_req_info, mesh, s)) { 496e86b9096SDag-Erling Smørgrav log_err("mesh_new_client: out of memory add tcpreqinfo"); 497e86b9096SDag-Erling Smørgrav goto servfail_mem; 498e86b9096SDag-Erling Smørgrav } 499e86b9096SDag-Erling Smørgrav } 500c0caa2e2SCy Schubert if(rep->c->use_h2) { 501c0caa2e2SCy Schubert http2_stream_add_meshstate(rep->c->h2_stream, mesh, s); 502c0caa2e2SCy Schubert } 503091e9e46SCy Schubert /* add serve expired timer if required and not already there */ 504091e9e46SCy Schubert if(timeout && !mesh_serve_expired_init(s, timeout)) { 505091e9e46SCy Schubert log_err("mesh_new_client: out of memory initializing serve expired"); 506091e9e46SCy Schubert goto servfail_mem; 507091e9e46SCy Schubert } 508b7579f77SDag-Erling Smørgrav /* update statistics */ 509b7579f77SDag-Erling Smørgrav if(was_detached) { 510b7579f77SDag-Erling Smørgrav log_assert(mesh->num_detached_states > 0); 511b7579f77SDag-Erling Smørgrav mesh->num_detached_states--; 512b7579f77SDag-Erling Smørgrav } 513b7579f77SDag-Erling Smørgrav if(was_noreply) { 514b7579f77SDag-Erling Smørgrav mesh->num_reply_states ++; 515b7579f77SDag-Erling Smørgrav } 516b7579f77SDag-Erling Smørgrav mesh->num_reply_addrs++; 517b7579f77SDag-Erling Smørgrav if(s->list_select == mesh_no_list) { 518b7579f77SDag-Erling Smørgrav /* move to either the forever or the jostle_list */ 519b7579f77SDag-Erling Smørgrav if(mesh->num_forever_states < mesh->max_forever_states) { 520b7579f77SDag-Erling Smørgrav mesh->num_forever_states ++; 521b7579f77SDag-Erling Smørgrav mesh_list_insert(s, &mesh->forever_first, 522b7579f77SDag-Erling Smørgrav &mesh->forever_last); 523b7579f77SDag-Erling Smørgrav s->list_select = mesh_forever_list; 524b7579f77SDag-Erling Smørgrav } else { 525b7579f77SDag-Erling Smørgrav mesh_list_insert(s, &mesh->jostle_first, 526b7579f77SDag-Erling Smørgrav &mesh->jostle_last); 527b7579f77SDag-Erling Smørgrav s->list_select = mesh_jostle_list; 528b7579f77SDag-Erling Smørgrav } 529b7579f77SDag-Erling Smørgrav } 530b7579f77SDag-Erling Smørgrav if(added) 531b7579f77SDag-Erling Smørgrav mesh_run(mesh, s, module_event_new, NULL); 532091e9e46SCy Schubert return; 533091e9e46SCy Schubert 534091e9e46SCy Schubert servfail_mem: 535091e9e46SCy Schubert if(!inplace_cb_reply_servfail_call(mesh->env, qinfo, &s->s, 536f44e67d1SCy Schubert NULL, LDNS_RCODE_SERVFAIL, edns, rep, mesh->env->scratch, mesh->env->now_tv)) 53724e36522SCy Schubert edns->opt_list_inplace_cb_out = NULL; 538091e9e46SCy Schubert error_encode(r_buffer, LDNS_RCODE_SERVFAIL, 539091e9e46SCy Schubert qinfo, qid, qflags, edns); 540091e9e46SCy Schubert comm_point_send_reply(rep); 541091e9e46SCy Schubert if(added) 542091e9e46SCy Schubert mesh_state_delete(&s->s); 543091e9e46SCy Schubert return; 544b7579f77SDag-Erling Smørgrav } 545b7579f77SDag-Erling Smørgrav 546b7579f77SDag-Erling Smørgrav int 547b7579f77SDag-Erling Smørgrav mesh_new_callback(struct mesh_area* mesh, struct query_info* qinfo, 54817d15b25SDag-Erling Smørgrav uint16_t qflags, struct edns_data* edns, sldns_buffer* buf, 549a39a5a69SCy Schubert uint16_t qid, mesh_cb_func_type cb, void* cb_arg, int rpz_passthru) 550b7579f77SDag-Erling Smørgrav { 551bc892140SDag-Erling Smørgrav struct mesh_state* s = NULL; 55224e36522SCy Schubert int unique = unique_mesh_state(edns->opt_list_in, mesh->env); 553091e9e46SCy Schubert int timeout = mesh->env->cfg->serve_expired? 554091e9e46SCy Schubert mesh->env->cfg->serve_expired_client_timeout:0; 555b7579f77SDag-Erling Smørgrav int was_detached = 0; 556b7579f77SDag-Erling Smørgrav int was_noreply = 0; 557b7579f77SDag-Erling Smørgrav int added = 0; 558bc892140SDag-Erling Smørgrav if(!unique) 55965b390aaSDag-Erling Smørgrav s = mesh_area_find(mesh, NULL, qinfo, qflags&(BIT_RD|BIT_CD), 0, 0); 56065b390aaSDag-Erling Smørgrav 561b7579f77SDag-Erling Smørgrav /* there are no limits on the number of callbacks */ 562b7579f77SDag-Erling Smørgrav 563b7579f77SDag-Erling Smørgrav /* see if it already exists, if not, create one */ 564b7579f77SDag-Erling Smørgrav if(!s) { 565b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 5663005e0a3SDag-Erling Smørgrav struct rbnode_type* n; 567b7579f77SDag-Erling Smørgrav #endif 56865b390aaSDag-Erling Smørgrav s = mesh_state_create(mesh->env, qinfo, NULL, 56965b390aaSDag-Erling Smørgrav qflags&(BIT_RD|BIT_CD), 0, 0); 570b7579f77SDag-Erling Smørgrav if(!s) { 571b7579f77SDag-Erling Smørgrav return 0; 572b7579f77SDag-Erling Smørgrav } 5738f76bb7dSCy Schubert /* set detached (it is now) */ 5748f76bb7dSCy Schubert mesh->num_detached_states++; 575bc892140SDag-Erling Smørgrav if(unique) 576bc892140SDag-Erling Smørgrav mesh_state_make_unique(s); 577a39a5a69SCy Schubert s->s.rpz_passthru = rpz_passthru; 57824e36522SCy Schubert if(edns->opt_list_in) { 57924e36522SCy Schubert s->s.edns_opts_front_in = edns_opt_copy_region(edns->opt_list_in, 580bc892140SDag-Erling Smørgrav s->s.region); 581bc892140SDag-Erling Smørgrav if(!s->s.edns_opts_front_in) { 5828f76bb7dSCy Schubert mesh_state_delete(&s->s); 583bc892140SDag-Erling Smørgrav return 0; 584bc892140SDag-Erling Smørgrav } 585bc892140SDag-Erling Smørgrav } 586b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 587b7579f77SDag-Erling Smørgrav n = 58817d15b25SDag-Erling Smørgrav #else 58917d15b25SDag-Erling Smørgrav (void) 590b7579f77SDag-Erling Smørgrav #endif 591b7579f77SDag-Erling Smørgrav rbtree_insert(&mesh->all, &s->node); 592b7579f77SDag-Erling Smørgrav log_assert(n != NULL); 593b7579f77SDag-Erling Smørgrav added = 1; 594b7579f77SDag-Erling Smørgrav } 595091e9e46SCy Schubert if(!s->reply_list && !s->cb_list) { 596b7579f77SDag-Erling Smørgrav was_noreply = 1; 597091e9e46SCy Schubert if(s->super_set.count == 0) { 598091e9e46SCy Schubert was_detached = 1; 599091e9e46SCy Schubert } 600091e9e46SCy Schubert } 601b7579f77SDag-Erling Smørgrav /* add reply to s */ 602b7579f77SDag-Erling Smørgrav if(!mesh_state_add_cb(s, edns, buf, cb, cb_arg, qid, qflags)) { 603b7579f77SDag-Erling Smørgrav if(added) 604b7579f77SDag-Erling Smørgrav mesh_state_delete(&s->s); 605b7579f77SDag-Erling Smørgrav return 0; 606b7579f77SDag-Erling Smørgrav } 607091e9e46SCy Schubert /* add serve expired timer if not already there */ 608091e9e46SCy Schubert if(timeout && !mesh_serve_expired_init(s, timeout)) { 6098f76bb7dSCy Schubert if(added) 6108f76bb7dSCy Schubert mesh_state_delete(&s->s); 611091e9e46SCy Schubert return 0; 612091e9e46SCy Schubert } 613b7579f77SDag-Erling Smørgrav /* update statistics */ 614b7579f77SDag-Erling Smørgrav if(was_detached) { 615b7579f77SDag-Erling Smørgrav log_assert(mesh->num_detached_states > 0); 616b7579f77SDag-Erling Smørgrav mesh->num_detached_states--; 617b7579f77SDag-Erling Smørgrav } 618b7579f77SDag-Erling Smørgrav if(was_noreply) { 619b7579f77SDag-Erling Smørgrav mesh->num_reply_states ++; 620b7579f77SDag-Erling Smørgrav } 621b7579f77SDag-Erling Smørgrav mesh->num_reply_addrs++; 622b7579f77SDag-Erling Smørgrav if(added) 623b7579f77SDag-Erling Smørgrav mesh_run(mesh, s, module_event_new, NULL); 624b7579f77SDag-Erling Smørgrav return 1; 625b7579f77SDag-Erling Smørgrav } 626b7579f77SDag-Erling Smørgrav 6278a384985SDag-Erling Smørgrav /* Internal backend routine of mesh_new_prefetch(). It takes one additional 6288a384985SDag-Erling Smørgrav * parameter, 'run', which controls whether to run the prefetch state 6298a384985SDag-Erling Smørgrav * immediately. When this function is called internally 'run' could be 6308a384985SDag-Erling Smørgrav * 0 (false), in which case the new state is only made runnable so it 6318a384985SDag-Erling Smørgrav * will not be run recursively on top of the current state. */ 6328a384985SDag-Erling Smørgrav static void mesh_schedule_prefetch(struct mesh_area* mesh, 633a39a5a69SCy Schubert struct query_info* qinfo, uint16_t qflags, time_t leeway, int run, 634a39a5a69SCy Schubert int rpz_passthru) 6358a384985SDag-Erling Smørgrav { 63665b390aaSDag-Erling Smørgrav struct mesh_state* s = mesh_area_find(mesh, NULL, qinfo, 63765b390aaSDag-Erling Smørgrav qflags&(BIT_RD|BIT_CD), 0, 0); 638b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 6393005e0a3SDag-Erling Smørgrav struct rbnode_type* n; 640b7579f77SDag-Erling Smørgrav #endif 641b7579f77SDag-Erling Smørgrav /* already exists, and for a different purpose perhaps. 642b7579f77SDag-Erling Smørgrav * if mesh_no_list, keep it that way. */ 643b7579f77SDag-Erling Smørgrav if(s) { 644b7579f77SDag-Erling Smørgrav /* make it ignore the cache from now on */ 645b7579f77SDag-Erling Smørgrav if(!s->s.blacklist) 646b7579f77SDag-Erling Smørgrav sock_list_insert(&s->s.blacklist, NULL, 0, s->s.region); 647b7579f77SDag-Erling Smørgrav if(s->s.prefetch_leeway < leeway) 648b7579f77SDag-Erling Smørgrav s->s.prefetch_leeway = leeway; 649b7579f77SDag-Erling Smørgrav return; 650b7579f77SDag-Erling Smørgrav } 651b7579f77SDag-Erling Smørgrav if(!mesh_make_new_space(mesh, NULL)) { 652b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "Too many queries. dropped prefetch."); 653b7579f77SDag-Erling Smørgrav mesh->stats_dropped ++; 654b7579f77SDag-Erling Smørgrav return; 655b7579f77SDag-Erling Smørgrav } 656bc892140SDag-Erling Smørgrav 65765b390aaSDag-Erling Smørgrav s = mesh_state_create(mesh->env, qinfo, NULL, 65865b390aaSDag-Erling Smørgrav qflags&(BIT_RD|BIT_CD), 0, 0); 659b7579f77SDag-Erling Smørgrav if(!s) { 660b7579f77SDag-Erling Smørgrav log_err("prefetch mesh_state_create: out of memory"); 661b7579f77SDag-Erling Smørgrav return; 662b7579f77SDag-Erling Smørgrav } 663b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 664b7579f77SDag-Erling Smørgrav n = 66517d15b25SDag-Erling Smørgrav #else 66617d15b25SDag-Erling Smørgrav (void) 667b7579f77SDag-Erling Smørgrav #endif 668b7579f77SDag-Erling Smørgrav rbtree_insert(&mesh->all, &s->node); 669b7579f77SDag-Erling Smørgrav log_assert(n != NULL); 670b7579f77SDag-Erling Smørgrav /* set detached (it is now) */ 671b7579f77SDag-Erling Smørgrav mesh->num_detached_states++; 672b7579f77SDag-Erling Smørgrav /* make it ignore the cache */ 673b7579f77SDag-Erling Smørgrav sock_list_insert(&s->s.blacklist, NULL, 0, s->s.region); 674b7579f77SDag-Erling Smørgrav s->s.prefetch_leeway = leeway; 675b7579f77SDag-Erling Smørgrav 676b7579f77SDag-Erling Smørgrav if(s->list_select == mesh_no_list) { 677b7579f77SDag-Erling Smørgrav /* move to either the forever or the jostle_list */ 678b7579f77SDag-Erling Smørgrav if(mesh->num_forever_states < mesh->max_forever_states) { 679b7579f77SDag-Erling Smørgrav mesh->num_forever_states ++; 680b7579f77SDag-Erling Smørgrav mesh_list_insert(s, &mesh->forever_first, 681b7579f77SDag-Erling Smørgrav &mesh->forever_last); 682b7579f77SDag-Erling Smørgrav s->list_select = mesh_forever_list; 683b7579f77SDag-Erling Smørgrav } else { 684b7579f77SDag-Erling Smørgrav mesh_list_insert(s, &mesh->jostle_first, 685b7579f77SDag-Erling Smørgrav &mesh->jostle_last); 686b7579f77SDag-Erling Smørgrav s->list_select = mesh_jostle_list; 687b7579f77SDag-Erling Smørgrav } 688b7579f77SDag-Erling Smørgrav } 689a39a5a69SCy Schubert s->s.rpz_passthru = rpz_passthru; 6908a384985SDag-Erling Smørgrav 6918a384985SDag-Erling Smørgrav if(!run) { 6928a384985SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 6938a384985SDag-Erling Smørgrav n = 6948a384985SDag-Erling Smørgrav #else 6958a384985SDag-Erling Smørgrav (void) 6968a384985SDag-Erling Smørgrav #endif 6978a384985SDag-Erling Smørgrav rbtree_insert(&mesh->run, &s->run_node); 6988a384985SDag-Erling Smørgrav log_assert(n != NULL); 6998a384985SDag-Erling Smørgrav return; 7008a384985SDag-Erling Smørgrav } 7018a384985SDag-Erling Smørgrav 702b7579f77SDag-Erling Smørgrav mesh_run(mesh, s, module_event_new, NULL); 703b7579f77SDag-Erling Smørgrav } 704b7579f77SDag-Erling Smørgrav 705a39a5a69SCy Schubert #ifdef CLIENT_SUBNET 706a39a5a69SCy Schubert /* Same logic as mesh_schedule_prefetch but tailored to the subnet module logic 707a39a5a69SCy Schubert * like passing along the comm_reply info. This will be faked into an EDNS 708a39a5a69SCy Schubert * option for processing by the subnet module if the client has not already 709a39a5a69SCy Schubert * attached its own ECS data. */ 710a39a5a69SCy Schubert static void mesh_schedule_prefetch_subnet(struct mesh_area* mesh, 711a39a5a69SCy Schubert struct query_info* qinfo, uint16_t qflags, time_t leeway, int run, 7128f76bb7dSCy Schubert int rpz_passthru, struct sockaddr_storage* addr, struct edns_option* edns_list) 713091e9e46SCy Schubert { 714a39a5a69SCy Schubert struct mesh_state* s = NULL; 715a39a5a69SCy Schubert struct edns_option* opt = NULL; 716a39a5a69SCy Schubert #ifdef UNBOUND_DEBUG 717a39a5a69SCy Schubert struct rbnode_type* n; 718a39a5a69SCy Schubert #endif 719a39a5a69SCy Schubert if(!mesh_make_new_space(mesh, NULL)) { 720a39a5a69SCy Schubert verbose(VERB_ALGO, "Too many queries. dropped prefetch."); 721a39a5a69SCy Schubert mesh->stats_dropped ++; 722a39a5a69SCy Schubert return; 723a39a5a69SCy Schubert } 724a39a5a69SCy Schubert 725a39a5a69SCy Schubert s = mesh_state_create(mesh->env, qinfo, NULL, 726a39a5a69SCy Schubert qflags&(BIT_RD|BIT_CD), 0, 0); 727a39a5a69SCy Schubert if(!s) { 728a39a5a69SCy Schubert log_err("prefetch_subnet mesh_state_create: out of memory"); 729a39a5a69SCy Schubert return; 730a39a5a69SCy Schubert } 731a39a5a69SCy Schubert mesh_state_make_unique(s); 732a39a5a69SCy Schubert 733a39a5a69SCy Schubert opt = edns_opt_list_find(edns_list, mesh->env->cfg->client_subnet_opcode); 734a39a5a69SCy Schubert if(opt) { 735a39a5a69SCy Schubert /* Use the client's ECS data */ 736a39a5a69SCy Schubert if(!edns_opt_list_append(&s->s.edns_opts_front_in, opt->opt_code, 737a39a5a69SCy Schubert opt->opt_len, opt->opt_data, s->s.region)) { 738a39a5a69SCy Schubert log_err("prefetch_subnet edns_opt_list_append: out of memory"); 739a39a5a69SCy Schubert return; 740a39a5a69SCy Schubert } 741a39a5a69SCy Schubert } else { 7428f76bb7dSCy Schubert /* Store the client's address. Later in the subnet module, 7438f76bb7dSCy Schubert * it is decided whether to include an ECS option or not. 7448f76bb7dSCy Schubert */ 7458f76bb7dSCy Schubert s->s.client_addr = *addr; 746a39a5a69SCy Schubert } 747a39a5a69SCy Schubert #ifdef UNBOUND_DEBUG 748a39a5a69SCy Schubert n = 749a39a5a69SCy Schubert #else 750a39a5a69SCy Schubert (void) 751a39a5a69SCy Schubert #endif 752a39a5a69SCy Schubert rbtree_insert(&mesh->all, &s->node); 753a39a5a69SCy Schubert log_assert(n != NULL); 754a39a5a69SCy Schubert /* set detached (it is now) */ 755a39a5a69SCy Schubert mesh->num_detached_states++; 756a39a5a69SCy Schubert /* make it ignore the cache */ 757a39a5a69SCy Schubert sock_list_insert(&s->s.blacklist, NULL, 0, s->s.region); 758a39a5a69SCy Schubert s->s.prefetch_leeway = leeway; 759a39a5a69SCy Schubert 760a39a5a69SCy Schubert if(s->list_select == mesh_no_list) { 761a39a5a69SCy Schubert /* move to either the forever or the jostle_list */ 762a39a5a69SCy Schubert if(mesh->num_forever_states < mesh->max_forever_states) { 763a39a5a69SCy Schubert mesh->num_forever_states ++; 764a39a5a69SCy Schubert mesh_list_insert(s, &mesh->forever_first, 765a39a5a69SCy Schubert &mesh->forever_last); 766a39a5a69SCy Schubert s->list_select = mesh_forever_list; 767a39a5a69SCy Schubert } else { 768a39a5a69SCy Schubert mesh_list_insert(s, &mesh->jostle_first, 769a39a5a69SCy Schubert &mesh->jostle_last); 770a39a5a69SCy Schubert s->list_select = mesh_jostle_list; 771a39a5a69SCy Schubert } 772a39a5a69SCy Schubert } 773a39a5a69SCy Schubert s->s.rpz_passthru = rpz_passthru; 774a39a5a69SCy Schubert 775a39a5a69SCy Schubert if(!run) { 776a39a5a69SCy Schubert #ifdef UNBOUND_DEBUG 777a39a5a69SCy Schubert n = 778a39a5a69SCy Schubert #else 779a39a5a69SCy Schubert (void) 780a39a5a69SCy Schubert #endif 781a39a5a69SCy Schubert rbtree_insert(&mesh->run, &s->run_node); 782a39a5a69SCy Schubert log_assert(n != NULL); 783a39a5a69SCy Schubert return; 784a39a5a69SCy Schubert } 785a39a5a69SCy Schubert 786a39a5a69SCy Schubert mesh_run(mesh, s, module_event_new, NULL); 787a39a5a69SCy Schubert } 788a39a5a69SCy Schubert #endif /* CLIENT_SUBNET */ 789a39a5a69SCy Schubert 790a39a5a69SCy Schubert void mesh_new_prefetch(struct mesh_area* mesh, struct query_info* qinfo, 791a39a5a69SCy Schubert uint16_t qflags, time_t leeway, int rpz_passthru, 7928f76bb7dSCy Schubert struct sockaddr_storage* addr, struct edns_option* opt_list) 793a39a5a69SCy Schubert { 7948f76bb7dSCy Schubert (void)addr; 795a39a5a69SCy Schubert (void)opt_list; 796a39a5a69SCy Schubert #ifdef CLIENT_SUBNET 7978f76bb7dSCy Schubert if(addr) 798a39a5a69SCy Schubert mesh_schedule_prefetch_subnet(mesh, qinfo, qflags, leeway, 1, 7998f76bb7dSCy Schubert rpz_passthru, addr, opt_list); 800a39a5a69SCy Schubert else 801a39a5a69SCy Schubert #endif 802a39a5a69SCy Schubert mesh_schedule_prefetch(mesh, qinfo, qflags, leeway, 1, 803a39a5a69SCy Schubert rpz_passthru); 804091e9e46SCy Schubert } 805091e9e46SCy Schubert 806b7579f77SDag-Erling Smørgrav void mesh_report_reply(struct mesh_area* mesh, struct outbound_entry* e, 807b7579f77SDag-Erling Smørgrav struct comm_reply* reply, int what) 808b7579f77SDag-Erling Smørgrav { 809b7579f77SDag-Erling Smørgrav enum module_ev event = module_event_reply; 810b7579f77SDag-Erling Smørgrav e->qstate->reply = reply; 811b7579f77SDag-Erling Smørgrav if(what != NETEVENT_NOERROR) { 812b7579f77SDag-Erling Smørgrav event = module_event_noreply; 813b7579f77SDag-Erling Smørgrav if(what == NETEVENT_CAPSFAIL) 814b7579f77SDag-Erling Smørgrav event = module_event_capsfail; 815b7579f77SDag-Erling Smørgrav } 816b7579f77SDag-Erling Smørgrav mesh_run(mesh, e->qstate->mesh_info, event, e); 817b7579f77SDag-Erling Smørgrav } 818b7579f77SDag-Erling Smørgrav 819b7579f77SDag-Erling Smørgrav struct mesh_state* 820b7579f77SDag-Erling Smørgrav mesh_state_create(struct module_env* env, struct query_info* qinfo, 82165b390aaSDag-Erling Smørgrav struct respip_client_info* cinfo, uint16_t qflags, int prime, 82265b390aaSDag-Erling Smørgrav int valrec) 823b7579f77SDag-Erling Smørgrav { 824b7579f77SDag-Erling Smørgrav struct regional* region = alloc_reg_obtain(env->alloc); 825b7579f77SDag-Erling Smørgrav struct mesh_state* mstate; 826b7579f77SDag-Erling Smørgrav int i; 827b7579f77SDag-Erling Smørgrav if(!region) 828b7579f77SDag-Erling Smørgrav return NULL; 829b7579f77SDag-Erling Smørgrav mstate = (struct mesh_state*)regional_alloc(region, 830b7579f77SDag-Erling Smørgrav sizeof(struct mesh_state)); 831b7579f77SDag-Erling Smørgrav if(!mstate) { 832b7579f77SDag-Erling Smørgrav alloc_reg_release(env->alloc, region); 833b7579f77SDag-Erling Smørgrav return NULL; 834b7579f77SDag-Erling Smørgrav } 835b7579f77SDag-Erling Smørgrav memset(mstate, 0, sizeof(*mstate)); 836b7579f77SDag-Erling Smørgrav mstate->node = *RBTREE_NULL; 837b7579f77SDag-Erling Smørgrav mstate->run_node = *RBTREE_NULL; 838b7579f77SDag-Erling Smørgrav mstate->node.key = mstate; 839b7579f77SDag-Erling Smørgrav mstate->run_node.key = mstate; 840b7579f77SDag-Erling Smørgrav mstate->reply_list = NULL; 841b7579f77SDag-Erling Smørgrav mstate->list_select = mesh_no_list; 842b7579f77SDag-Erling Smørgrav mstate->replies_sent = 0; 843b7579f77SDag-Erling Smørgrav rbtree_init(&mstate->super_set, &mesh_state_ref_compare); 844b7579f77SDag-Erling Smørgrav rbtree_init(&mstate->sub_set, &mesh_state_ref_compare); 845b7579f77SDag-Erling Smørgrav mstate->num_activated = 0; 846bc892140SDag-Erling Smørgrav mstate->unique = NULL; 847b7579f77SDag-Erling Smørgrav /* init module qstate */ 848b7579f77SDag-Erling Smørgrav mstate->s.qinfo.qtype = qinfo->qtype; 849b7579f77SDag-Erling Smørgrav mstate->s.qinfo.qclass = qinfo->qclass; 850bc892140SDag-Erling Smørgrav mstate->s.qinfo.local_alias = NULL; 851b7579f77SDag-Erling Smørgrav mstate->s.qinfo.qname_len = qinfo->qname_len; 852b7579f77SDag-Erling Smørgrav mstate->s.qinfo.qname = regional_alloc_init(region, qinfo->qname, 853b7579f77SDag-Erling Smørgrav qinfo->qname_len); 854b7579f77SDag-Erling Smørgrav if(!mstate->s.qinfo.qname) { 855b7579f77SDag-Erling Smørgrav alloc_reg_release(env->alloc, region); 856b7579f77SDag-Erling Smørgrav return NULL; 857b7579f77SDag-Erling Smørgrav } 85865b390aaSDag-Erling Smørgrav if(cinfo) { 85965b390aaSDag-Erling Smørgrav mstate->s.client_info = regional_alloc_init(region, cinfo, 86065b390aaSDag-Erling Smørgrav sizeof(*cinfo)); 86165b390aaSDag-Erling Smørgrav if(!mstate->s.client_info) { 86265b390aaSDag-Erling Smørgrav alloc_reg_release(env->alloc, region); 86365b390aaSDag-Erling Smørgrav return NULL; 86465b390aaSDag-Erling Smørgrav } 86565b390aaSDag-Erling Smørgrav } 866b7579f77SDag-Erling Smørgrav /* remove all weird bits from qflags */ 867b7579f77SDag-Erling Smørgrav mstate->s.query_flags = (qflags & (BIT_RD|BIT_CD)); 868b7579f77SDag-Erling Smørgrav mstate->s.is_priming = prime; 869ff825849SDag-Erling Smørgrav mstate->s.is_valrec = valrec; 870b7579f77SDag-Erling Smørgrav mstate->s.reply = NULL; 871b7579f77SDag-Erling Smørgrav mstate->s.region = region; 872b7579f77SDag-Erling Smørgrav mstate->s.curmod = 0; 873b7579f77SDag-Erling Smørgrav mstate->s.return_msg = 0; 874b7579f77SDag-Erling Smørgrav mstate->s.return_rcode = LDNS_RCODE_NOERROR; 875b7579f77SDag-Erling Smørgrav mstate->s.env = env; 876b7579f77SDag-Erling Smørgrav mstate->s.mesh_info = mstate; 877b7579f77SDag-Erling Smørgrav mstate->s.prefetch_leeway = 0; 878091e9e46SCy Schubert mstate->s.serve_expired_data = NULL; 879bc892140SDag-Erling Smørgrav mstate->s.no_cache_lookup = 0; 880bc892140SDag-Erling Smørgrav mstate->s.no_cache_store = 0; 8818a384985SDag-Erling Smørgrav mstate->s.need_refetch = 0; 8824c75e3aaSDag-Erling Smørgrav mstate->s.was_ratelimited = 0; 883790c6b24SCy Schubert mstate->s.qstarttime = *env->now; 8848a384985SDag-Erling Smørgrav 885b7579f77SDag-Erling Smørgrav /* init modules */ 886b7579f77SDag-Erling Smørgrav for(i=0; i<env->mesh->mods.num; i++) { 887b7579f77SDag-Erling Smørgrav mstate->s.minfo[i] = NULL; 888b7579f77SDag-Erling Smørgrav mstate->s.ext_state[i] = module_state_initial; 889b7579f77SDag-Erling Smørgrav } 890bc892140SDag-Erling Smørgrav /* init edns option lists */ 891bc892140SDag-Erling Smørgrav mstate->s.edns_opts_front_in = NULL; 892bc892140SDag-Erling Smørgrav mstate->s.edns_opts_back_out = NULL; 893bc892140SDag-Erling Smørgrav mstate->s.edns_opts_back_in = NULL; 894bc892140SDag-Erling Smørgrav mstate->s.edns_opts_front_out = NULL; 895bc892140SDag-Erling Smørgrav 896b7579f77SDag-Erling Smørgrav return mstate; 897b7579f77SDag-Erling Smørgrav } 898b7579f77SDag-Erling Smørgrav 899bc892140SDag-Erling Smørgrav void 900bc892140SDag-Erling Smørgrav mesh_state_make_unique(struct mesh_state* mstate) 901bc892140SDag-Erling Smørgrav { 902bc892140SDag-Erling Smørgrav mstate->unique = mstate; 903bc892140SDag-Erling Smørgrav } 904bc892140SDag-Erling Smørgrav 905b7579f77SDag-Erling Smørgrav void 906b7579f77SDag-Erling Smørgrav mesh_state_cleanup(struct mesh_state* mstate) 907b7579f77SDag-Erling Smørgrav { 908b7579f77SDag-Erling Smørgrav struct mesh_area* mesh; 909b7579f77SDag-Erling Smørgrav int i; 910b7579f77SDag-Erling Smørgrav if(!mstate) 911b7579f77SDag-Erling Smørgrav return; 912b7579f77SDag-Erling Smørgrav mesh = mstate->s.env->mesh; 913091e9e46SCy Schubert /* Stop and delete the serve expired timer */ 914091e9e46SCy Schubert if(mstate->s.serve_expired_data && mstate->s.serve_expired_data->timer) { 915091e9e46SCy Schubert comm_timer_delete(mstate->s.serve_expired_data->timer); 916091e9e46SCy Schubert mstate->s.serve_expired_data->timer = NULL; 917091e9e46SCy Schubert } 918b7579f77SDag-Erling Smørgrav /* drop unsent replies */ 919b7579f77SDag-Erling Smørgrav if(!mstate->replies_sent) { 920e86b9096SDag-Erling Smørgrav struct mesh_reply* rep = mstate->reply_list; 921b7579f77SDag-Erling Smørgrav struct mesh_cb* cb; 922e86b9096SDag-Erling Smørgrav /* in tcp_req_info, the mstates linked are removed, but 923e86b9096SDag-Erling Smørgrav * the reply_list is now NULL, so the remove-from-empty-list 924e86b9096SDag-Erling Smørgrav * takes no time and also it does not do the mesh accounting */ 925e86b9096SDag-Erling Smørgrav mstate->reply_list = NULL; 926e86b9096SDag-Erling Smørgrav for(; rep; rep=rep->next) { 927b7579f77SDag-Erling Smørgrav comm_point_drop_reply(&rep->query_reply); 928091e9e46SCy Schubert log_assert(mesh->num_reply_addrs > 0); 929b7579f77SDag-Erling Smørgrav mesh->num_reply_addrs--; 930b7579f77SDag-Erling Smørgrav } 9310fb34990SDag-Erling Smørgrav while((cb = mstate->cb_list)!=NULL) { 9320fb34990SDag-Erling Smørgrav mstate->cb_list = cb->next; 933b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mesh_cb(cb->cb)); 934b7579f77SDag-Erling Smørgrav (*cb->cb)(cb->cb_arg, LDNS_RCODE_SERVFAIL, NULL, 9354c75e3aaSDag-Erling Smørgrav sec_status_unchecked, NULL, 0); 936091e9e46SCy Schubert log_assert(mesh->num_reply_addrs > 0); 937b7579f77SDag-Erling Smørgrav mesh->num_reply_addrs--; 938b7579f77SDag-Erling Smørgrav } 939b7579f77SDag-Erling Smørgrav } 940b7579f77SDag-Erling Smørgrav 941b7579f77SDag-Erling Smørgrav /* de-init modules */ 942b7579f77SDag-Erling Smørgrav for(i=0; i<mesh->mods.num; i++) { 943b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mod_clear(mesh->mods.mod[i]->clear)); 944b7579f77SDag-Erling Smørgrav (*mesh->mods.mod[i]->clear)(&mstate->s, i); 945b7579f77SDag-Erling Smørgrav mstate->s.minfo[i] = NULL; 946b7579f77SDag-Erling Smørgrav mstate->s.ext_state[i] = module_finished; 947b7579f77SDag-Erling Smørgrav } 948b7579f77SDag-Erling Smørgrav alloc_reg_release(mstate->s.env->alloc, mstate->s.region); 949b7579f77SDag-Erling Smørgrav } 950b7579f77SDag-Erling Smørgrav 951b7579f77SDag-Erling Smørgrav void 952b7579f77SDag-Erling Smørgrav mesh_state_delete(struct module_qstate* qstate) 953b7579f77SDag-Erling Smørgrav { 954b7579f77SDag-Erling Smørgrav struct mesh_area* mesh; 955b7579f77SDag-Erling Smørgrav struct mesh_state_ref* super, ref; 956b7579f77SDag-Erling Smørgrav struct mesh_state* mstate; 957b7579f77SDag-Erling Smørgrav if(!qstate) 958b7579f77SDag-Erling Smørgrav return; 959b7579f77SDag-Erling Smørgrav mstate = qstate->mesh_info; 960b7579f77SDag-Erling Smørgrav mesh = mstate->s.env->mesh; 961b7579f77SDag-Erling Smørgrav mesh_detach_subs(&mstate->s); 962b7579f77SDag-Erling Smørgrav if(mstate->list_select == mesh_forever_list) { 963b7579f77SDag-Erling Smørgrav mesh->num_forever_states --; 964b7579f77SDag-Erling Smørgrav mesh_list_remove(mstate, &mesh->forever_first, 965b7579f77SDag-Erling Smørgrav &mesh->forever_last); 966b7579f77SDag-Erling Smørgrav } else if(mstate->list_select == mesh_jostle_list) { 967b7579f77SDag-Erling Smørgrav mesh_list_remove(mstate, &mesh->jostle_first, 968b7579f77SDag-Erling Smørgrav &mesh->jostle_last); 969b7579f77SDag-Erling Smørgrav } 970b7579f77SDag-Erling Smørgrav if(!mstate->reply_list && !mstate->cb_list 971b7579f77SDag-Erling Smørgrav && mstate->super_set.count == 0) { 972b7579f77SDag-Erling Smørgrav log_assert(mesh->num_detached_states > 0); 973b7579f77SDag-Erling Smørgrav mesh->num_detached_states--; 974b7579f77SDag-Erling Smørgrav } 975b7579f77SDag-Erling Smørgrav if(mstate->reply_list || mstate->cb_list) { 976b7579f77SDag-Erling Smørgrav log_assert(mesh->num_reply_states > 0); 977b7579f77SDag-Erling Smørgrav mesh->num_reply_states--; 978b7579f77SDag-Erling Smørgrav } 979b7579f77SDag-Erling Smørgrav ref.node.key = &ref; 980b7579f77SDag-Erling Smørgrav ref.s = mstate; 981b7579f77SDag-Erling Smørgrav RBTREE_FOR(super, struct mesh_state_ref*, &mstate->super_set) { 982b7579f77SDag-Erling Smørgrav (void)rbtree_delete(&super->s->sub_set, &ref); 983b7579f77SDag-Erling Smørgrav } 984b7579f77SDag-Erling Smørgrav (void)rbtree_delete(&mesh->run, mstate); 985b7579f77SDag-Erling Smørgrav (void)rbtree_delete(&mesh->all, mstate); 986b7579f77SDag-Erling Smørgrav mesh_state_cleanup(mstate); 987b7579f77SDag-Erling Smørgrav } 988b7579f77SDag-Erling Smørgrav 989b7579f77SDag-Erling Smørgrav /** helper recursive rbtree find routine */ 990b7579f77SDag-Erling Smørgrav static int 991b7579f77SDag-Erling Smørgrav find_in_subsub(struct mesh_state* m, struct mesh_state* tofind, size_t *c) 992b7579f77SDag-Erling Smørgrav { 993b7579f77SDag-Erling Smørgrav struct mesh_state_ref* r; 994b7579f77SDag-Erling Smørgrav if((*c)++ > MESH_MAX_SUBSUB) 995b7579f77SDag-Erling Smørgrav return 1; 996b7579f77SDag-Erling Smørgrav RBTREE_FOR(r, struct mesh_state_ref*, &m->sub_set) { 997b7579f77SDag-Erling Smørgrav if(r->s == tofind || find_in_subsub(r->s, tofind, c)) 998b7579f77SDag-Erling Smørgrav return 1; 999b7579f77SDag-Erling Smørgrav } 1000b7579f77SDag-Erling Smørgrav return 0; 1001b7579f77SDag-Erling Smørgrav } 1002b7579f77SDag-Erling Smørgrav 1003b7579f77SDag-Erling Smørgrav /** find cycle for already looked up mesh_state */ 1004b7579f77SDag-Erling Smørgrav static int 1005b7579f77SDag-Erling Smørgrav mesh_detect_cycle_found(struct module_qstate* qstate, struct mesh_state* dep_m) 1006b7579f77SDag-Erling Smørgrav { 1007b7579f77SDag-Erling Smørgrav struct mesh_state* cyc_m = qstate->mesh_info; 1008b7579f77SDag-Erling Smørgrav size_t counter = 0; 1009b7579f77SDag-Erling Smørgrav if(!dep_m) 1010b7579f77SDag-Erling Smørgrav return 0; 1011b7579f77SDag-Erling Smørgrav if(dep_m == cyc_m || find_in_subsub(dep_m, cyc_m, &counter)) { 1012b7579f77SDag-Erling Smørgrav if(counter > MESH_MAX_SUBSUB) 1013b7579f77SDag-Erling Smørgrav return 2; 1014b7579f77SDag-Erling Smørgrav return 1; 1015b7579f77SDag-Erling Smørgrav } 1016b7579f77SDag-Erling Smørgrav return 0; 1017b7579f77SDag-Erling Smørgrav } 1018b7579f77SDag-Erling Smørgrav 1019b7579f77SDag-Erling Smørgrav void mesh_detach_subs(struct module_qstate* qstate) 1020b7579f77SDag-Erling Smørgrav { 1021b7579f77SDag-Erling Smørgrav struct mesh_area* mesh = qstate->env->mesh; 1022b7579f77SDag-Erling Smørgrav struct mesh_state_ref* ref, lookup; 1023b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 10243005e0a3SDag-Erling Smørgrav struct rbnode_type* n; 1025b7579f77SDag-Erling Smørgrav #endif 1026b7579f77SDag-Erling Smørgrav lookup.node.key = &lookup; 1027b7579f77SDag-Erling Smørgrav lookup.s = qstate->mesh_info; 1028b7579f77SDag-Erling Smørgrav RBTREE_FOR(ref, struct mesh_state_ref*, &qstate->mesh_info->sub_set) { 1029b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 1030b7579f77SDag-Erling Smørgrav n = 103117d15b25SDag-Erling Smørgrav #else 103217d15b25SDag-Erling Smørgrav (void) 1033b7579f77SDag-Erling Smørgrav #endif 1034b7579f77SDag-Erling Smørgrav rbtree_delete(&ref->s->super_set, &lookup); 1035b7579f77SDag-Erling Smørgrav log_assert(n != NULL); /* must have been present */ 1036b7579f77SDag-Erling Smørgrav if(!ref->s->reply_list && !ref->s->cb_list 1037b7579f77SDag-Erling Smørgrav && ref->s->super_set.count == 0) { 1038b7579f77SDag-Erling Smørgrav mesh->num_detached_states++; 1039b7579f77SDag-Erling Smørgrav log_assert(mesh->num_detached_states + 1040b7579f77SDag-Erling Smørgrav mesh->num_reply_states <= mesh->all.count); 1041b7579f77SDag-Erling Smørgrav } 1042b7579f77SDag-Erling Smørgrav } 1043b7579f77SDag-Erling Smørgrav rbtree_init(&qstate->mesh_info->sub_set, &mesh_state_ref_compare); 1044b7579f77SDag-Erling Smørgrav } 1045b7579f77SDag-Erling Smørgrav 1046c7f4d7adSDag-Erling Smørgrav int mesh_add_sub(struct module_qstate* qstate, struct query_info* qinfo, 1047c7f4d7adSDag-Erling Smørgrav uint16_t qflags, int prime, int valrec, struct module_qstate** newq, 1048c7f4d7adSDag-Erling Smørgrav struct mesh_state** sub) 1049b7579f77SDag-Erling Smørgrav { 1050b7579f77SDag-Erling Smørgrav /* find it, if not, create it */ 1051b7579f77SDag-Erling Smørgrav struct mesh_area* mesh = qstate->env->mesh; 1052c7f4d7adSDag-Erling Smørgrav *sub = mesh_area_find(mesh, NULL, qinfo, qflags, 105365b390aaSDag-Erling Smørgrav prime, valrec); 1054c7f4d7adSDag-Erling Smørgrav if(mesh_detect_cycle_found(qstate, *sub)) { 1055b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "attach failed, cycle detected"); 1056b7579f77SDag-Erling Smørgrav return 0; 1057b7579f77SDag-Erling Smørgrav } 1058c7f4d7adSDag-Erling Smørgrav if(!*sub) { 1059b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 10603005e0a3SDag-Erling Smørgrav struct rbnode_type* n; 1061b7579f77SDag-Erling Smørgrav #endif 1062b7579f77SDag-Erling Smørgrav /* create a new one */ 1063c7f4d7adSDag-Erling Smørgrav *sub = mesh_state_create(qstate->env, qinfo, NULL, qflags, prime, 106465b390aaSDag-Erling Smørgrav valrec); 1065c7f4d7adSDag-Erling Smørgrav if(!*sub) { 1066b7579f77SDag-Erling Smørgrav log_err("mesh_attach_sub: out of memory"); 1067b7579f77SDag-Erling Smørgrav return 0; 1068b7579f77SDag-Erling Smørgrav } 1069b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 1070b7579f77SDag-Erling Smørgrav n = 107117d15b25SDag-Erling Smørgrav #else 107217d15b25SDag-Erling Smørgrav (void) 1073b7579f77SDag-Erling Smørgrav #endif 1074c7f4d7adSDag-Erling Smørgrav rbtree_insert(&mesh->all, &(*sub)->node); 1075b7579f77SDag-Erling Smørgrav log_assert(n != NULL); 1076b7579f77SDag-Erling Smørgrav /* set detached (it is now) */ 1077b7579f77SDag-Erling Smørgrav mesh->num_detached_states++; 1078b7579f77SDag-Erling Smørgrav /* set new query state to run */ 1079b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 1080b7579f77SDag-Erling Smørgrav n = 108117d15b25SDag-Erling Smørgrav #else 108217d15b25SDag-Erling Smørgrav (void) 1083b7579f77SDag-Erling Smørgrav #endif 1084c7f4d7adSDag-Erling Smørgrav rbtree_insert(&mesh->run, &(*sub)->run_node); 1085b7579f77SDag-Erling Smørgrav log_assert(n != NULL); 1086c7f4d7adSDag-Erling Smørgrav *newq = &(*sub)->s; 1087b7579f77SDag-Erling Smørgrav } else 1088b7579f77SDag-Erling Smørgrav *newq = NULL; 1089c7f4d7adSDag-Erling Smørgrav return 1; 1090c7f4d7adSDag-Erling Smørgrav } 1091c7f4d7adSDag-Erling Smørgrav 1092c7f4d7adSDag-Erling Smørgrav int mesh_attach_sub(struct module_qstate* qstate, struct query_info* qinfo, 1093c7f4d7adSDag-Erling Smørgrav uint16_t qflags, int prime, int valrec, struct module_qstate** newq) 1094c7f4d7adSDag-Erling Smørgrav { 1095c7f4d7adSDag-Erling Smørgrav struct mesh_area* mesh = qstate->env->mesh; 1096c7f4d7adSDag-Erling Smørgrav struct mesh_state* sub = NULL; 1097c7f4d7adSDag-Erling Smørgrav int was_detached; 1098c7f4d7adSDag-Erling Smørgrav if(!mesh_add_sub(qstate, qinfo, qflags, prime, valrec, newq, &sub)) 1099c7f4d7adSDag-Erling Smørgrav return 0; 11008ed2b524SDag-Erling Smørgrav was_detached = (sub->super_set.count == 0); 1101b7579f77SDag-Erling Smørgrav if(!mesh_state_attachment(qstate->mesh_info, sub)) 1102b7579f77SDag-Erling Smørgrav return 0; 11038ed2b524SDag-Erling Smørgrav /* if it was a duplicate attachment, the count was not zero before */ 11048ed2b524SDag-Erling Smørgrav if(!sub->reply_list && !sub->cb_list && was_detached && 11058ed2b524SDag-Erling Smørgrav sub->super_set.count == 1) { 1106b7579f77SDag-Erling Smørgrav /* it used to be detached, before this one got added */ 1107b7579f77SDag-Erling Smørgrav log_assert(mesh->num_detached_states > 0); 1108b7579f77SDag-Erling Smørgrav mesh->num_detached_states--; 1109b7579f77SDag-Erling Smørgrav } 1110b7579f77SDag-Erling Smørgrav /* *newq will be run when inited after the current module stops */ 1111b7579f77SDag-Erling Smørgrav return 1; 1112b7579f77SDag-Erling Smørgrav } 1113b7579f77SDag-Erling Smørgrav 1114b7579f77SDag-Erling Smørgrav int mesh_state_attachment(struct mesh_state* super, struct mesh_state* sub) 1115b7579f77SDag-Erling Smørgrav { 1116b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 11173005e0a3SDag-Erling Smørgrav struct rbnode_type* n; 1118b7579f77SDag-Erling Smørgrav #endif 1119b7579f77SDag-Erling Smørgrav struct mesh_state_ref* subref; /* points to sub, inserted in super */ 1120b7579f77SDag-Erling Smørgrav struct mesh_state_ref* superref; /* points to super, inserted in sub */ 1121b7579f77SDag-Erling Smørgrav if( !(subref = regional_alloc(super->s.region, 1122b7579f77SDag-Erling Smørgrav sizeof(struct mesh_state_ref))) || 1123b7579f77SDag-Erling Smørgrav !(superref = regional_alloc(sub->s.region, 1124b7579f77SDag-Erling Smørgrav sizeof(struct mesh_state_ref))) ) { 1125b7579f77SDag-Erling Smørgrav log_err("mesh_state_attachment: out of memory"); 1126b7579f77SDag-Erling Smørgrav return 0; 1127b7579f77SDag-Erling Smørgrav } 1128b7579f77SDag-Erling Smørgrav superref->node.key = superref; 1129b7579f77SDag-Erling Smørgrav superref->s = super; 1130b7579f77SDag-Erling Smørgrav subref->node.key = subref; 1131b7579f77SDag-Erling Smørgrav subref->s = sub; 11328ed2b524SDag-Erling Smørgrav if(!rbtree_insert(&sub->super_set, &superref->node)) { 11338ed2b524SDag-Erling Smørgrav /* this should not happen, iterator and validator do not 11348ed2b524SDag-Erling Smørgrav * attach subqueries that are identical. */ 11358ed2b524SDag-Erling Smørgrav /* already attached, we are done, nothing todo. 11368ed2b524SDag-Erling Smørgrav * since superref and subref already allocated in region, 11378ed2b524SDag-Erling Smørgrav * we cannot free them */ 11388ed2b524SDag-Erling Smørgrav return 1; 11398ed2b524SDag-Erling Smørgrav } 1140b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG 1141b7579f77SDag-Erling Smørgrav n = 114217d15b25SDag-Erling Smørgrav #else 114317d15b25SDag-Erling Smørgrav (void) 1144b7579f77SDag-Erling Smørgrav #endif 1145b7579f77SDag-Erling Smørgrav rbtree_insert(&super->sub_set, &subref->node); 11468ed2b524SDag-Erling Smørgrav log_assert(n != NULL); /* we checked above if statement, the reverse 11478ed2b524SDag-Erling Smørgrav administration should not fail now, unless they are out of sync */ 1148b7579f77SDag-Erling Smørgrav return 1; 1149b7579f77SDag-Erling Smørgrav } 1150b7579f77SDag-Erling Smørgrav 1151b7579f77SDag-Erling Smørgrav /** 1152b7579f77SDag-Erling Smørgrav * callback results to mesh cb entry 1153b7579f77SDag-Erling Smørgrav * @param m: mesh state to send it for. 1154b7579f77SDag-Erling Smørgrav * @param rcode: if not 0, error code. 1155b7579f77SDag-Erling Smørgrav * @param rep: reply to send (or NULL if rcode is set). 1156b7579f77SDag-Erling Smørgrav * @param r: callback entry 1157f44e67d1SCy Schubert * @param start_time: the time to pass to callback functions, it is 0 or 1158f44e67d1SCy Schubert * a value from one of the packets if the mesh state had packets. 1159b7579f77SDag-Erling Smørgrav */ 1160b7579f77SDag-Erling Smørgrav static void 1161b7579f77SDag-Erling Smørgrav mesh_do_callback(struct mesh_state* m, int rcode, struct reply_info* rep, 1162f44e67d1SCy Schubert struct mesh_cb* r, struct timeval* start_time) 1163b7579f77SDag-Erling Smørgrav { 1164b7579f77SDag-Erling Smørgrav int secure; 1165b7579f77SDag-Erling Smørgrav char* reason = NULL; 11664c75e3aaSDag-Erling Smørgrav int was_ratelimited = m->s.was_ratelimited; 1167b7579f77SDag-Erling Smørgrav /* bogus messages are not made into servfail, sec_status passed 1168b7579f77SDag-Erling Smørgrav * to the callback function */ 1169b7579f77SDag-Erling Smørgrav if(rep && rep->security == sec_status_secure) 1170b7579f77SDag-Erling Smørgrav secure = 1; 1171b7579f77SDag-Erling Smørgrav else secure = 0; 1172b7579f77SDag-Erling Smørgrav if(!rep && rcode == LDNS_RCODE_NOERROR) 1173b7579f77SDag-Erling Smørgrav rcode = LDNS_RCODE_SERVFAIL; 11748f76bb7dSCy Schubert if(!rcode && rep && (rep->security == sec_status_bogus || 11750fb34990SDag-Erling Smørgrav rep->security == sec_status_secure_sentinel_fail)) { 11764c75e3aaSDag-Erling Smørgrav if(!(reason = errinf_to_str_bogus(&m->s))) 1177b7579f77SDag-Erling Smørgrav rcode = LDNS_RCODE_SERVFAIL; 1178b7579f77SDag-Erling Smørgrav } 1179b7579f77SDag-Erling Smørgrav /* send the reply */ 1180b7579f77SDag-Erling Smørgrav if(rcode) { 1181bc892140SDag-Erling Smørgrav if(rcode == LDNS_RCODE_SERVFAIL) { 1182bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_servfail_call(m->s.env, &m->s.qinfo, &m->s, 1183f44e67d1SCy Schubert rep, rcode, &r->edns, NULL, m->s.region, start_time)) 118424e36522SCy Schubert r->edns.opt_list_inplace_cb_out = NULL; 1185bc892140SDag-Erling Smørgrav } else { 1186bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_call(m->s.env, &m->s.qinfo, &m->s, rep, rcode, 1187f44e67d1SCy Schubert &r->edns, NULL, m->s.region, start_time)) 118824e36522SCy Schubert r->edns.opt_list_inplace_cb_out = NULL; 1189bc892140SDag-Erling Smørgrav } 1190b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mesh_cb(r->cb)); 11914c75e3aaSDag-Erling Smørgrav (*r->cb)(r->cb_arg, rcode, r->buf, sec_status_unchecked, NULL, 11924c75e3aaSDag-Erling Smørgrav was_ratelimited); 1193b7579f77SDag-Erling Smørgrav } else { 1194b7579f77SDag-Erling Smørgrav size_t udp_size = r->edns.udp_size; 119517d15b25SDag-Erling Smørgrav sldns_buffer_clear(r->buf); 1196b7579f77SDag-Erling Smørgrav r->edns.edns_version = EDNS_ADVERTISED_VERSION; 1197b7579f77SDag-Erling Smørgrav r->edns.udp_size = EDNS_ADVERTISED_SIZE; 1198b7579f77SDag-Erling Smørgrav r->edns.ext_rcode = 0; 1199b7579f77SDag-Erling Smørgrav r->edns.bits &= EDNS_DO; 1200*103ba509SCy Schubert if(m->s.env->cfg->disable_edns_do && (r->edns.bits&EDNS_DO)) 1201*103ba509SCy Schubert r->edns.edns_present = 0; 1202bc892140SDag-Erling Smørgrav 1203bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_call(m->s.env, &m->s.qinfo, &m->s, rep, 1204f44e67d1SCy Schubert LDNS_RCODE_NOERROR, &r->edns, NULL, m->s.region, start_time) || 1205e2d15004SDag-Erling Smørgrav !reply_info_answer_encode(&m->s.qinfo, rep, r->qid, 1206b7579f77SDag-Erling Smørgrav r->qflags, r->buf, 0, 1, 1207b7579f77SDag-Erling Smørgrav m->s.env->scratch, udp_size, &r->edns, 1208b7579f77SDag-Erling Smørgrav (int)(r->edns.bits & EDNS_DO), secure)) 1209b7579f77SDag-Erling Smørgrav { 1210b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mesh_cb(r->cb)); 1211b7579f77SDag-Erling Smørgrav (*r->cb)(r->cb_arg, LDNS_RCODE_SERVFAIL, r->buf, 12124c75e3aaSDag-Erling Smørgrav sec_status_unchecked, NULL, 0); 1213b7579f77SDag-Erling Smørgrav } else { 1214b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mesh_cb(r->cb)); 1215b7579f77SDag-Erling Smørgrav (*r->cb)(r->cb_arg, LDNS_RCODE_NOERROR, r->buf, 12168f76bb7dSCy Schubert (rep?rep->security:sec_status_unchecked), 12178f76bb7dSCy Schubert reason, was_ratelimited); 1218b7579f77SDag-Erling Smørgrav } 1219b7579f77SDag-Erling Smørgrav } 1220b7579f77SDag-Erling Smørgrav free(reason); 1221091e9e46SCy Schubert log_assert(m->s.env->mesh->num_reply_addrs > 0); 1222b7579f77SDag-Erling Smørgrav m->s.env->mesh->num_reply_addrs--; 1223b7579f77SDag-Erling Smørgrav } 1224b7579f77SDag-Erling Smørgrav 122524e36522SCy Schubert static inline int 122624e36522SCy Schubert mesh_is_rpz_respip_tcponly_action(struct mesh_state const* m) 122724e36522SCy Schubert { 122824e36522SCy Schubert struct respip_action_info const* respip_info = m->s.respip_action_info; 1229*103ba509SCy Schubert return (respip_info == NULL 123024e36522SCy Schubert ? 0 123124e36522SCy Schubert : (respip_info->rpz_used 123224e36522SCy Schubert && !respip_info->rpz_disabled 1233*103ba509SCy Schubert && respip_info->action == respip_truncate)) 1234*103ba509SCy Schubert || m->s.tcp_required; 123524e36522SCy Schubert } 123624e36522SCy Schubert 123724e36522SCy Schubert static inline int 12388f76bb7dSCy Schubert mesh_is_udp(struct mesh_reply const* r) 12398f76bb7dSCy Schubert { 124024e36522SCy Schubert return r->query_reply.c->type == comm_udp; 124124e36522SCy Schubert } 124224e36522SCy Schubert 12438f76bb7dSCy Schubert static inline void 12448f76bb7dSCy Schubert mesh_find_and_attach_ede_and_reason(struct mesh_state* m, 12458f76bb7dSCy Schubert struct reply_info* rep, struct mesh_reply* r) 12468f76bb7dSCy Schubert { 12478f76bb7dSCy Schubert /* OLD note: 12488f76bb7dSCy Schubert * During validation the EDE code can be received via two 12498f76bb7dSCy Schubert * code paths. One code path fills the reply_info EDE, and 12508f76bb7dSCy Schubert * the other fills it in the errinf_strlist. These paths 12518f76bb7dSCy Schubert * intersect at some points, but where is opaque due to 12528f76bb7dSCy Schubert * the complexity of the validator. At the time of writing 12538f76bb7dSCy Schubert * we make the choice to prefer the EDE from errinf_strlist 12548f76bb7dSCy Schubert * but a compelling reason to do otherwise is just as valid 12558f76bb7dSCy Schubert * NEW note: 12568f76bb7dSCy Schubert * The compelling reason is that with caching support, the value 12578f76bb7dSCy Schubert * in the reply_info is cached. 12588f76bb7dSCy Schubert * The reason members of the reply_info struct should be 12598f76bb7dSCy Schubert * updated as they are already cached. No reason to 12608f76bb7dSCy Schubert * try and find the EDE information in errinf anymore. 12618f76bb7dSCy Schubert */ 12628f76bb7dSCy Schubert if(rep->reason_bogus != LDNS_EDE_NONE) { 12638f76bb7dSCy Schubert edns_opt_list_append_ede(&r->edns.opt_list_out, 12648f76bb7dSCy Schubert m->s.region, rep->reason_bogus, rep->reason_bogus_str); 12658f76bb7dSCy Schubert } 12668f76bb7dSCy Schubert } 12678f76bb7dSCy Schubert 1268b7579f77SDag-Erling Smørgrav /** 1269b7579f77SDag-Erling Smørgrav * Send reply to mesh reply entry 1270b7579f77SDag-Erling Smørgrav * @param m: mesh state to send it for. 1271b7579f77SDag-Erling Smørgrav * @param rcode: if not 0, error code. 1272b7579f77SDag-Erling Smørgrav * @param rep: reply to send (or NULL if rcode is set). 1273b7579f77SDag-Erling Smørgrav * @param r: reply entry 1274e86b9096SDag-Erling Smørgrav * @param r_buffer: buffer to use for reply entry. 1275b7579f77SDag-Erling Smørgrav * @param prev: previous reply, already has its answer encoded in buffer. 1276e86b9096SDag-Erling Smørgrav * @param prev_buffer: buffer for previous reply. 1277b7579f77SDag-Erling Smørgrav */ 1278b7579f77SDag-Erling Smørgrav static void 1279b7579f77SDag-Erling Smørgrav mesh_send_reply(struct mesh_state* m, int rcode, struct reply_info* rep, 1280e86b9096SDag-Erling Smørgrav struct mesh_reply* r, struct sldns_buffer* r_buffer, 1281e86b9096SDag-Erling Smørgrav struct mesh_reply* prev, struct sldns_buffer* prev_buffer) 1282b7579f77SDag-Erling Smørgrav { 1283b7579f77SDag-Erling Smørgrav struct timeval end_time; 1284b7579f77SDag-Erling Smørgrav struct timeval duration; 1285b7579f77SDag-Erling Smørgrav int secure; 1286369c6923SCy Schubert /* briefly set the replylist to null in case the 1287369c6923SCy Schubert * meshsendreply calls tcpreqinfo sendreply that 1288369c6923SCy Schubert * comm_point_drops because of size, and then the 1289369c6923SCy Schubert * null stops the mesh state remove and thus 1290369c6923SCy Schubert * reply_list modification and accounting */ 1291369c6923SCy Schubert struct mesh_reply* rlist = m->reply_list; 129224e36522SCy Schubert 129324e36522SCy Schubert /* rpz: apply actions */ 129424e36522SCy Schubert rcode = mesh_is_udp(r) && mesh_is_rpz_respip_tcponly_action(m) 129524e36522SCy Schubert ? (rcode|BIT_TC) : rcode; 129624e36522SCy Schubert 1297b7579f77SDag-Erling Smørgrav /* examine security status */ 1298b7579f77SDag-Erling Smørgrav if(m->s.env->need_to_validate && (!(r->qflags&BIT_CD) || 1299b7579f77SDag-Erling Smørgrav m->s.env->cfg->ignore_cd) && rep && 13000fb34990SDag-Erling Smørgrav (rep->security <= sec_status_bogus || 13010fb34990SDag-Erling Smørgrav rep->security == sec_status_secure_sentinel_fail)) { 1302b7579f77SDag-Erling Smørgrav rcode = LDNS_RCODE_SERVFAIL; 1303b7579f77SDag-Erling Smørgrav if(m->s.env->cfg->stat_extended) 1304b7579f77SDag-Erling Smørgrav m->s.env->mesh->ans_bogus++; 1305b7579f77SDag-Erling Smørgrav } 1306b7579f77SDag-Erling Smørgrav if(rep && rep->security == sec_status_secure) 1307b7579f77SDag-Erling Smørgrav secure = 1; 1308b7579f77SDag-Erling Smørgrav else secure = 0; 1309b7579f77SDag-Erling Smørgrav if(!rep && rcode == LDNS_RCODE_NOERROR) 1310b7579f77SDag-Erling Smørgrav rcode = LDNS_RCODE_SERVFAIL; 1311c0caa2e2SCy Schubert if(r->query_reply.c->use_h2) { 1312c0caa2e2SCy Schubert r->query_reply.c->h2_stream = r->h2_stream; 1313c0caa2e2SCy Schubert /* Mesh reply won't exist for long anymore. Make it impossible 1314c0caa2e2SCy Schubert * for HTTP/2 stream to refer to mesh state, in case 1315c0caa2e2SCy Schubert * connection gets cleanup before HTTP/2 stream close. */ 1316c0caa2e2SCy Schubert r->h2_stream->mesh_state = NULL; 1317c0caa2e2SCy Schubert } 1318b7579f77SDag-Erling Smørgrav /* send the reply */ 1319369c6923SCy Schubert /* We don't reuse the encoded answer if: 1320369c6923SCy Schubert * - either the previous or current response has a local alias. We could 1321369c6923SCy Schubert * compare the alias records and still reuse the previous answer if they 1322369c6923SCy Schubert * are the same, but that would be complicated and error prone for the 1323369c6923SCy Schubert * relatively minor case. So we err on the side of safety. 1324369c6923SCy Schubert * - there are registered callback functions for the given rcode, as these 1325369c6923SCy Schubert * need to be called for each reply. */ 1326369c6923SCy Schubert if(((rcode != LDNS_RCODE_SERVFAIL && 1327369c6923SCy Schubert !m->s.env->inplace_cb_lists[inplace_cb_reply]) || 1328369c6923SCy Schubert (rcode == LDNS_RCODE_SERVFAIL && 1329369c6923SCy Schubert !m->s.env->inplace_cb_lists[inplace_cb_reply_servfail])) && 1330369c6923SCy Schubert prev && prev_buffer && prev->qflags == r->qflags && 1331bc892140SDag-Erling Smørgrav !prev->local_alias && !r->local_alias && 1332b7579f77SDag-Erling Smørgrav prev->edns.edns_present == r->edns.edns_present && 1333b7579f77SDag-Erling Smørgrav prev->edns.bits == r->edns.bits && 1334e2d15004SDag-Erling Smørgrav prev->edns.udp_size == r->edns.udp_size && 133524e36522SCy Schubert edns_opt_list_compare(prev->edns.opt_list_out, r->edns.opt_list_out) == 0 && 133624e36522SCy Schubert edns_opt_list_compare(prev->edns.opt_list_inplace_cb_out, r->edns.opt_list_inplace_cb_out) == 0 133724e36522SCy Schubert ) { 1338b7579f77SDag-Erling Smørgrav /* if the previous reply is identical to this one, fix ID */ 1339e86b9096SDag-Erling Smørgrav if(prev_buffer != r_buffer) 1340e86b9096SDag-Erling Smørgrav sldns_buffer_copy(r_buffer, prev_buffer); 1341e86b9096SDag-Erling Smørgrav sldns_buffer_write_at(r_buffer, 0, &r->qid, sizeof(uint16_t)); 1342e86b9096SDag-Erling Smørgrav sldns_buffer_write_at(r_buffer, 12, r->qname, 1343e86b9096SDag-Erling Smørgrav m->s.qinfo.qname_len); 1344369c6923SCy Schubert m->reply_list = NULL; 1345b7579f77SDag-Erling Smørgrav comm_point_send_reply(&r->query_reply); 1346369c6923SCy Schubert m->reply_list = rlist; 1347b7579f77SDag-Erling Smørgrav } else if(rcode) { 1348b7579f77SDag-Erling Smørgrav m->s.qinfo.qname = r->qname; 1349bc892140SDag-Erling Smørgrav m->s.qinfo.local_alias = r->local_alias; 1350bc892140SDag-Erling Smørgrav if(rcode == LDNS_RCODE_SERVFAIL) { 1351bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_servfail_call(m->s.env, &m->s.qinfo, &m->s, 1352f44e67d1SCy Schubert rep, rcode, &r->edns, &r->query_reply, m->s.region, &r->start_time)) 135324e36522SCy Schubert r->edns.opt_list_inplace_cb_out = NULL; 1354bc892140SDag-Erling Smørgrav } else { 1355bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_call(m->s.env, &m->s.qinfo, &m->s, rep, rcode, 1356f44e67d1SCy Schubert &r->edns, &r->query_reply, m->s.region, &r->start_time)) 135724e36522SCy Schubert r->edns.opt_list_inplace_cb_out = NULL; 1358bc892140SDag-Erling Smørgrav } 13598f76bb7dSCy Schubert /* Send along EDE EDNS0 option when SERVFAILing; usually 13608f76bb7dSCy Schubert * DNSSEC validation failures */ 13618f76bb7dSCy Schubert /* Since we are SERVFAILing here, CD bit and rep->security 13628f76bb7dSCy Schubert * is already handled. */ 13638f76bb7dSCy Schubert if(m->s.env->cfg->ede && rep) { 13648f76bb7dSCy Schubert mesh_find_and_attach_ede_and_reason(m, rep, r); 1365a39a5a69SCy Schubert } 1366e86b9096SDag-Erling Smørgrav error_encode(r_buffer, rcode, &m->s.qinfo, r->qid, 1367e86b9096SDag-Erling Smørgrav r->qflags, &r->edns); 1368369c6923SCy Schubert m->reply_list = NULL; 1369b7579f77SDag-Erling Smørgrav comm_point_send_reply(&r->query_reply); 1370369c6923SCy Schubert m->reply_list = rlist; 1371b7579f77SDag-Erling Smørgrav } else { 1372b7579f77SDag-Erling Smørgrav size_t udp_size = r->edns.udp_size; 1373b7579f77SDag-Erling Smørgrav r->edns.edns_version = EDNS_ADVERTISED_VERSION; 1374b7579f77SDag-Erling Smørgrav r->edns.udp_size = EDNS_ADVERTISED_SIZE; 1375b7579f77SDag-Erling Smørgrav r->edns.ext_rcode = 0; 1376b7579f77SDag-Erling Smørgrav r->edns.bits &= EDNS_DO; 1377*103ba509SCy Schubert if(m->s.env->cfg->disable_edns_do && (r->edns.bits&EDNS_DO)) 1378*103ba509SCy Schubert r->edns.edns_present = 0; 1379b7579f77SDag-Erling Smørgrav m->s.qinfo.qname = r->qname; 1380bc892140SDag-Erling Smørgrav m->s.qinfo.local_alias = r->local_alias; 13818f76bb7dSCy Schubert 13828f76bb7dSCy Schubert /* Attach EDE without SERVFAIL if the validation failed. 13838f76bb7dSCy Schubert * Need to explicitly check for rep->security otherwise failed 13848f76bb7dSCy Schubert * validation paths may attach to a secure answer. */ 13858f76bb7dSCy Schubert if(m->s.env->cfg->ede && rep && 13868f76bb7dSCy Schubert (rep->security <= sec_status_bogus || 13878f76bb7dSCy Schubert rep->security == sec_status_secure_sentinel_fail)) { 13888f76bb7dSCy Schubert mesh_find_and_attach_ede_and_reason(m, rep, r); 13898f76bb7dSCy Schubert } 13908f76bb7dSCy Schubert 1391bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_call(m->s.env, &m->s.qinfo, &m->s, rep, 1392f44e67d1SCy Schubert LDNS_RCODE_NOERROR, &r->edns, &r->query_reply, m->s.region, &r->start_time) || 1393e2d15004SDag-Erling Smørgrav !reply_info_answer_encode(&m->s.qinfo, rep, r->qid, 1394e86b9096SDag-Erling Smørgrav r->qflags, r_buffer, 0, 1, m->s.env->scratch, 1395e86b9096SDag-Erling Smørgrav udp_size, &r->edns, (int)(r->edns.bits & EDNS_DO), 1396e86b9096SDag-Erling Smørgrav secure)) 1397b7579f77SDag-Erling Smørgrav { 1398bc892140SDag-Erling Smørgrav if(!inplace_cb_reply_servfail_call(m->s.env, &m->s.qinfo, &m->s, 1399f44e67d1SCy Schubert rep, LDNS_RCODE_SERVFAIL, &r->edns, &r->query_reply, m->s.region, &r->start_time)) 140024e36522SCy Schubert r->edns.opt_list_inplace_cb_out = NULL; 1401a39a5a69SCy Schubert /* internal server error (probably malloc failure) so no 1402a39a5a69SCy Schubert * EDE (RFC8914) needed */ 1403e86b9096SDag-Erling Smørgrav error_encode(r_buffer, LDNS_RCODE_SERVFAIL, 1404e86b9096SDag-Erling Smørgrav &m->s.qinfo, r->qid, r->qflags, &r->edns); 1405b7579f77SDag-Erling Smørgrav } 1406369c6923SCy Schubert m->reply_list = NULL; 1407b7579f77SDag-Erling Smørgrav comm_point_send_reply(&r->query_reply); 1408369c6923SCy Schubert m->reply_list = rlist; 1409b7579f77SDag-Erling Smørgrav } 1410b7579f77SDag-Erling Smørgrav /* account */ 1411091e9e46SCy Schubert log_assert(m->s.env->mesh->num_reply_addrs > 0); 1412b7579f77SDag-Erling Smørgrav m->s.env->mesh->num_reply_addrs--; 1413b7579f77SDag-Erling Smørgrav end_time = *m->s.env->now_tv; 1414b7579f77SDag-Erling Smørgrav timeval_subtract(&duration, &end_time, &r->start_time); 141517d15b25SDag-Erling Smørgrav verbose(VERB_ALGO, "query took " ARG_LL "d.%6.6d sec", 141617d15b25SDag-Erling Smørgrav (long long)duration.tv_sec, (int)duration.tv_usec); 1417b7579f77SDag-Erling Smørgrav m->s.env->mesh->replies_sent++; 1418b7579f77SDag-Erling Smørgrav timeval_add(&m->s.env->mesh->replies_sum_wait, &duration); 1419b7579f77SDag-Erling Smørgrav timehist_insert(m->s.env->mesh->histogram, &duration); 1420b7579f77SDag-Erling Smørgrav if(m->s.env->cfg->stat_extended) { 1421e86b9096SDag-Erling Smørgrav uint16_t rc = FLAGS_GET_RCODE(sldns_buffer_read_u16_at( 1422e86b9096SDag-Erling Smørgrav r_buffer, 2)); 1423b7579f77SDag-Erling Smørgrav if(secure) m->s.env->mesh->ans_secure++; 1424b7579f77SDag-Erling Smørgrav m->s.env->mesh->ans_rcode[ rc ] ++; 1425e86b9096SDag-Erling Smørgrav if(rc == 0 && LDNS_ANCOUNT(sldns_buffer_begin(r_buffer)) == 0) 1426b7579f77SDag-Erling Smørgrav m->s.env->mesh->ans_nodata++; 1427b7579f77SDag-Erling Smørgrav } 14283005e0a3SDag-Erling Smørgrav /* Log reply sent */ 14293005e0a3SDag-Erling Smørgrav if(m->s.env->cfg->log_replies) { 1430865f46b2SCy Schubert log_reply_info(NO_VERBOSE, &m->s.qinfo, 1431865f46b2SCy Schubert &r->query_reply.client_addr, 1432865f46b2SCy Schubert r->query_reply.client_addrlen, duration, 0, r_buffer); 14333005e0a3SDag-Erling Smørgrav } 1434b7579f77SDag-Erling Smørgrav } 1435b7579f77SDag-Erling Smørgrav 1436b7579f77SDag-Erling Smørgrav void mesh_query_done(struct mesh_state* mstate) 1437b7579f77SDag-Erling Smørgrav { 143825039b37SCy Schubert struct mesh_reply* r; 1439b7579f77SDag-Erling Smørgrav struct mesh_reply* prev = NULL; 1440e86b9096SDag-Erling Smørgrav struct sldns_buffer* prev_buffer = NULL; 1441b7579f77SDag-Erling Smørgrav struct mesh_cb* c; 1442b7579f77SDag-Erling Smørgrav struct reply_info* rep = (mstate->s.return_msg? 1443b7579f77SDag-Erling Smørgrav mstate->s.return_msg->rep:NULL); 1444f44e67d1SCy Schubert struct timeval tv = {0, 0}; 14458f76bb7dSCy Schubert int i = 0; 1446091e9e46SCy Schubert /* No need for the serve expired timer anymore; we are going to reply. */ 1447091e9e46SCy Schubert if(mstate->s.serve_expired_data) { 1448091e9e46SCy Schubert comm_timer_delete(mstate->s.serve_expired_data->timer); 1449091e9e46SCy Schubert mstate->s.serve_expired_data->timer = NULL; 1450091e9e46SCy Schubert } 1451091e9e46SCy Schubert if(mstate->s.return_rcode == LDNS_RCODE_SERVFAIL || 1452091e9e46SCy Schubert (rep && FLAGS_GET_RCODE(rep->flags) == LDNS_RCODE_SERVFAIL)) { 145324e36522SCy Schubert /* we are SERVFAILing; check for expired answer here */ 1454091e9e46SCy Schubert mesh_serve_expired_callback(mstate); 1455091e9e46SCy Schubert if((mstate->reply_list || mstate->cb_list) 14564c75e3aaSDag-Erling Smørgrav && mstate->s.env->cfg->log_servfail 14574c75e3aaSDag-Erling Smørgrav && !mstate->s.env->cfg->val_log_squelch) { 14584c75e3aaSDag-Erling Smørgrav char* err = errinf_to_str_servfail(&mstate->s); 14594c75e3aaSDag-Erling Smørgrav if(err) 14604c75e3aaSDag-Erling Smørgrav log_err("%s", err); 14614c75e3aaSDag-Erling Smørgrav free(err); 14624c75e3aaSDag-Erling Smørgrav } 1463091e9e46SCy Schubert } 146425039b37SCy Schubert for(r = mstate->reply_list; r; r = r->next) { 14658f76bb7dSCy Schubert i++; 1466f44e67d1SCy Schubert tv = r->start_time; 1467f44e67d1SCy Schubert 146865b390aaSDag-Erling Smørgrav /* if a response-ip address block has been stored the 146965b390aaSDag-Erling Smørgrav * information should be logged for each client. */ 147065b390aaSDag-Erling Smørgrav if(mstate->s.respip_action_info && 147165b390aaSDag-Erling Smørgrav mstate->s.respip_action_info->addrinfo) { 1472091e9e46SCy Schubert respip_inform_print(mstate->s.respip_action_info, 147365b390aaSDag-Erling Smørgrav r->qname, mstate->s.qinfo.qtype, 147465b390aaSDag-Erling Smørgrav mstate->s.qinfo.qclass, r->local_alias, 1475865f46b2SCy Schubert &r->query_reply.client_addr, 1476865f46b2SCy Schubert r->query_reply.client_addrlen); 147765b390aaSDag-Erling Smørgrav } 147865b390aaSDag-Erling Smørgrav 147965b390aaSDag-Erling Smørgrav /* if this query is determined to be dropped during the 148065b390aaSDag-Erling Smørgrav * mesh processing, this is the point to take that action. */ 1481091e9e46SCy Schubert if(mstate->s.is_drop) { 148225039b37SCy Schubert /* briefly set the reply_list to NULL, so that the 148325039b37SCy Schubert * tcp req info cleanup routine that calls the mesh 148425039b37SCy Schubert * to deregister the meshstate for it is not done 148525039b37SCy Schubert * because the list is NULL and also accounting is not 148625039b37SCy Schubert * done there, but instead we do that here. */ 148725039b37SCy Schubert struct mesh_reply* reply_list = mstate->reply_list; 148825039b37SCy Schubert mstate->reply_list = NULL; 148965b390aaSDag-Erling Smørgrav comm_point_drop_reply(&r->query_reply); 149025039b37SCy Schubert mstate->reply_list = reply_list; 1491091e9e46SCy Schubert } else { 1492e86b9096SDag-Erling Smørgrav struct sldns_buffer* r_buffer = r->query_reply.c->buffer; 1493e86b9096SDag-Erling Smørgrav if(r->query_reply.c->tcp_req_info) { 1494e86b9096SDag-Erling Smørgrav r_buffer = r->query_reply.c->tcp_req_info->spool_buffer; 1495e86b9096SDag-Erling Smørgrav prev_buffer = NULL; 1496e86b9096SDag-Erling Smørgrav } 149765b390aaSDag-Erling Smørgrav mesh_send_reply(mstate, mstate->s.return_rcode, rep, 1498e86b9096SDag-Erling Smørgrav r, r_buffer, prev, prev_buffer); 1499e86b9096SDag-Erling Smørgrav if(r->query_reply.c->tcp_req_info) { 1500e86b9096SDag-Erling Smørgrav tcp_req_info_remove_mesh_state(r->query_reply.c->tcp_req_info, mstate); 1501e86b9096SDag-Erling Smørgrav r_buffer = NULL; 1502e86b9096SDag-Erling Smørgrav } 1503b7579f77SDag-Erling Smørgrav prev = r; 1504e86b9096SDag-Erling Smørgrav prev_buffer = r_buffer; 1505b7579f77SDag-Erling Smørgrav } 150665b390aaSDag-Erling Smørgrav } 15078f76bb7dSCy Schubert /* Account for each reply sent. */ 15088f76bb7dSCy Schubert if(i > 0 && mstate->s.respip_action_info && 15098f76bb7dSCy Schubert mstate->s.respip_action_info->addrinfo && 15108f76bb7dSCy Schubert mstate->s.env->cfg->stat_extended && 15118f76bb7dSCy Schubert mstate->s.respip_action_info->rpz_used) { 15128f76bb7dSCy Schubert if(mstate->s.respip_action_info->rpz_disabled) 15138f76bb7dSCy Schubert mstate->s.env->mesh->rpz_action[RPZ_DISABLED_ACTION] += i; 15148f76bb7dSCy Schubert if(mstate->s.respip_action_info->rpz_cname_override) 15158f76bb7dSCy Schubert mstate->s.env->mesh->rpz_action[RPZ_CNAME_OVERRIDE_ACTION] += i; 15168f76bb7dSCy Schubert else 15178f76bb7dSCy Schubert mstate->s.env->mesh->rpz_action[respip_action_to_rpz_action( 15188f76bb7dSCy Schubert mstate->s.respip_action_info->action)] += i; 15198f76bb7dSCy Schubert } 15208f76bb7dSCy Schubert if(!mstate->s.is_drop && i > 0) { 15218f76bb7dSCy Schubert if(mstate->s.env->cfg->stat_extended 15228f76bb7dSCy Schubert && mstate->s.is_cachedb_answer) { 15238f76bb7dSCy Schubert mstate->s.env->mesh->ans_cachedb += i; 15248f76bb7dSCy Schubert } 15258f76bb7dSCy Schubert } 15268f76bb7dSCy Schubert 15278f76bb7dSCy Schubert /* Mesh area accounting */ 152825039b37SCy Schubert if(mstate->reply_list) { 152925039b37SCy Schubert mstate->reply_list = NULL; 153025039b37SCy Schubert if(!mstate->reply_list && !mstate->cb_list) { 153125039b37SCy Schubert /* was a reply state, not anymore */ 153225039b37SCy Schubert log_assert(mstate->s.env->mesh->num_reply_states > 0); 153325039b37SCy Schubert mstate->s.env->mesh->num_reply_states--; 153425039b37SCy Schubert } 153525039b37SCy Schubert if(!mstate->reply_list && !mstate->cb_list && 153625039b37SCy Schubert mstate->super_set.count == 0) 153725039b37SCy Schubert mstate->s.env->mesh->num_detached_states++; 153825039b37SCy Schubert } 1539b7579f77SDag-Erling Smørgrav mstate->replies_sent = 1; 15408f76bb7dSCy Schubert 15410fb34990SDag-Erling Smørgrav while((c = mstate->cb_list) != NULL) { 15420fb34990SDag-Erling Smørgrav /* take this cb off the list; so that the list can be 15430fb34990SDag-Erling Smørgrav * changed, eg. by adds from the callback routine */ 15443bd4df0aSDag-Erling Smørgrav if(!mstate->reply_list && mstate->cb_list && !c->next) { 15453bd4df0aSDag-Erling Smørgrav /* was a reply state, not anymore */ 1546091e9e46SCy Schubert log_assert(mstate->s.env->mesh->num_reply_states > 0); 15473bd4df0aSDag-Erling Smørgrav mstate->s.env->mesh->num_reply_states--; 15483bd4df0aSDag-Erling Smørgrav } 15490fb34990SDag-Erling Smørgrav mstate->cb_list = c->next; 15500fb34990SDag-Erling Smørgrav if(!mstate->reply_list && !mstate->cb_list && 15510fb34990SDag-Erling Smørgrav mstate->super_set.count == 0) 15520fb34990SDag-Erling Smørgrav mstate->s.env->mesh->num_detached_states++; 1553f44e67d1SCy Schubert mesh_do_callback(mstate, mstate->s.return_rcode, rep, c, &tv); 1554b7579f77SDag-Erling Smørgrav } 1555b7579f77SDag-Erling Smørgrav } 1556b7579f77SDag-Erling Smørgrav 1557b7579f77SDag-Erling Smørgrav void mesh_walk_supers(struct mesh_area* mesh, struct mesh_state* mstate) 1558b7579f77SDag-Erling Smørgrav { 1559b7579f77SDag-Erling Smørgrav struct mesh_state_ref* ref; 1560b7579f77SDag-Erling Smørgrav RBTREE_FOR(ref, struct mesh_state_ref*, &mstate->super_set) 1561b7579f77SDag-Erling Smørgrav { 1562b7579f77SDag-Erling Smørgrav /* make super runnable */ 1563b7579f77SDag-Erling Smørgrav (void)rbtree_insert(&mesh->run, &ref->s->run_node); 1564b7579f77SDag-Erling Smørgrav /* callback the function to inform super of result */ 1565b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mod_inform_super( 1566b7579f77SDag-Erling Smørgrav mesh->mods.mod[ref->s->s.curmod]->inform_super)); 1567b7579f77SDag-Erling Smørgrav (*mesh->mods.mod[ref->s->s.curmod]->inform_super)(&mstate->s, 1568b7579f77SDag-Erling Smørgrav ref->s->s.curmod, &ref->s->s); 15694c75e3aaSDag-Erling Smørgrav /* copy state that is always relevant to super */ 15704c75e3aaSDag-Erling Smørgrav copy_state_to_super(&mstate->s, ref->s->s.curmod, &ref->s->s); 1571b7579f77SDag-Erling Smørgrav } 1572b7579f77SDag-Erling Smørgrav } 1573b7579f77SDag-Erling Smørgrav 1574b7579f77SDag-Erling Smørgrav struct mesh_state* mesh_area_find(struct mesh_area* mesh, 157565b390aaSDag-Erling Smørgrav struct respip_client_info* cinfo, struct query_info* qinfo, 157665b390aaSDag-Erling Smørgrav uint16_t qflags, int prime, int valrec) 1577b7579f77SDag-Erling Smørgrav { 1578b7579f77SDag-Erling Smørgrav struct mesh_state key; 1579b7579f77SDag-Erling Smørgrav struct mesh_state* result; 1580b7579f77SDag-Erling Smørgrav 1581b7579f77SDag-Erling Smørgrav key.node.key = &key; 1582b7579f77SDag-Erling Smørgrav key.s.is_priming = prime; 1583ff825849SDag-Erling Smørgrav key.s.is_valrec = valrec; 1584b7579f77SDag-Erling Smørgrav key.s.qinfo = *qinfo; 1585b7579f77SDag-Erling Smørgrav key.s.query_flags = qflags; 1586bc892140SDag-Erling Smørgrav /* We are searching for a similar mesh state when we DO want to 1587bc892140SDag-Erling Smørgrav * aggregate the state. Thus unique is set to NULL. (default when we 1588bc892140SDag-Erling Smørgrav * desire aggregation).*/ 1589bc892140SDag-Erling Smørgrav key.unique = NULL; 159065b390aaSDag-Erling Smørgrav key.s.client_info = cinfo; 1591b7579f77SDag-Erling Smørgrav 1592b7579f77SDag-Erling Smørgrav result = (struct mesh_state*)rbtree_search(&mesh->all, &key); 1593b7579f77SDag-Erling Smørgrav return result; 1594b7579f77SDag-Erling Smørgrav } 1595b7579f77SDag-Erling Smørgrav 1596b7579f77SDag-Erling Smørgrav int mesh_state_add_cb(struct mesh_state* s, struct edns_data* edns, 15973005e0a3SDag-Erling Smørgrav sldns_buffer* buf, mesh_cb_func_type cb, void* cb_arg, 1598b7579f77SDag-Erling Smørgrav uint16_t qid, uint16_t qflags) 1599b7579f77SDag-Erling Smørgrav { 1600b7579f77SDag-Erling Smørgrav struct mesh_cb* r = regional_alloc(s->s.region, 1601b7579f77SDag-Erling Smørgrav sizeof(struct mesh_cb)); 1602b7579f77SDag-Erling Smørgrav if(!r) 1603b7579f77SDag-Erling Smørgrav return 0; 1604b7579f77SDag-Erling Smørgrav r->buf = buf; 1605b7579f77SDag-Erling Smørgrav log_assert(fptr_whitelist_mesh_cb(cb)); /* early failure ifmissing*/ 1606b7579f77SDag-Erling Smørgrav r->cb = cb; 1607b7579f77SDag-Erling Smørgrav r->cb_arg = cb_arg; 1608b7579f77SDag-Erling Smørgrav r->edns = *edns; 160924e36522SCy Schubert if(edns->opt_list_in && !(r->edns.opt_list_in = 161024e36522SCy Schubert edns_opt_copy_region(edns->opt_list_in, s->s.region))) 1611e2d15004SDag-Erling Smørgrav return 0; 161224e36522SCy Schubert if(edns->opt_list_out && !(r->edns.opt_list_out = 161324e36522SCy Schubert edns_opt_copy_region(edns->opt_list_out, s->s.region))) 161424e36522SCy Schubert return 0; 161524e36522SCy Schubert if(edns->opt_list_inplace_cb_out && !(r->edns.opt_list_inplace_cb_out = 161624e36522SCy Schubert edns_opt_copy_region(edns->opt_list_inplace_cb_out, s->s.region))) 161724e36522SCy Schubert return 0; 1618b7579f77SDag-Erling Smørgrav r->qid = qid; 1619b7579f77SDag-Erling Smørgrav r->qflags = qflags; 1620b7579f77SDag-Erling Smørgrav r->next = s->cb_list; 1621b7579f77SDag-Erling Smørgrav s->cb_list = r; 1622b7579f77SDag-Erling Smørgrav return 1; 1623b7579f77SDag-Erling Smørgrav 1624b7579f77SDag-Erling Smørgrav } 1625b7579f77SDag-Erling Smørgrav 1626b7579f77SDag-Erling Smørgrav int mesh_state_add_reply(struct mesh_state* s, struct edns_data* edns, 1627bc892140SDag-Erling Smørgrav struct comm_reply* rep, uint16_t qid, uint16_t qflags, 1628bc892140SDag-Erling Smørgrav const struct query_info* qinfo) 1629b7579f77SDag-Erling Smørgrav { 1630b7579f77SDag-Erling Smørgrav struct mesh_reply* r = regional_alloc(s->s.region, 1631b7579f77SDag-Erling Smørgrav sizeof(struct mesh_reply)); 1632b7579f77SDag-Erling Smørgrav if(!r) 1633b7579f77SDag-Erling Smørgrav return 0; 1634b7579f77SDag-Erling Smørgrav r->query_reply = *rep; 1635b7579f77SDag-Erling Smørgrav r->edns = *edns; 163624e36522SCy Schubert if(edns->opt_list_in && !(r->edns.opt_list_in = 163724e36522SCy Schubert edns_opt_copy_region(edns->opt_list_in, s->s.region))) 1638e2d15004SDag-Erling Smørgrav return 0; 163924e36522SCy Schubert if(edns->opt_list_out && !(r->edns.opt_list_out = 164024e36522SCy Schubert edns_opt_copy_region(edns->opt_list_out, s->s.region))) 164124e36522SCy Schubert return 0; 164224e36522SCy Schubert if(edns->opt_list_inplace_cb_out && !(r->edns.opt_list_inplace_cb_out = 164324e36522SCy Schubert edns_opt_copy_region(edns->opt_list_inplace_cb_out, s->s.region))) 164424e36522SCy Schubert return 0; 1645b7579f77SDag-Erling Smørgrav r->qid = qid; 1646b7579f77SDag-Erling Smørgrav r->qflags = qflags; 1647b7579f77SDag-Erling Smørgrav r->start_time = *s->s.env->now_tv; 1648b7579f77SDag-Erling Smørgrav r->next = s->reply_list; 1649bc892140SDag-Erling Smørgrav r->qname = regional_alloc_init(s->s.region, qinfo->qname, 1650b7579f77SDag-Erling Smørgrav s->s.qinfo.qname_len); 1651b7579f77SDag-Erling Smørgrav if(!r->qname) 1652b7579f77SDag-Erling Smørgrav return 0; 1653c0caa2e2SCy Schubert if(rep->c->use_h2) 1654c0caa2e2SCy Schubert r->h2_stream = rep->c->h2_stream; 1655bc892140SDag-Erling Smørgrav 1656bc892140SDag-Erling Smørgrav /* Data related to local alias stored in 'qinfo' (if any) is ephemeral 1657bc892140SDag-Erling Smørgrav * and can be different for different original queries (even if the 1658bc892140SDag-Erling Smørgrav * replaced query name is the same). So we need to make a deep copy 1659bc892140SDag-Erling Smørgrav * and store the copy for each reply info. */ 1660bc892140SDag-Erling Smørgrav if(qinfo->local_alias) { 1661bc892140SDag-Erling Smørgrav struct packed_rrset_data* d; 1662bc892140SDag-Erling Smørgrav struct packed_rrset_data* dsrc; 1663bc892140SDag-Erling Smørgrav r->local_alias = regional_alloc_zero(s->s.region, 1664bc892140SDag-Erling Smørgrav sizeof(*qinfo->local_alias)); 1665bc892140SDag-Erling Smørgrav if(!r->local_alias) 1666bc892140SDag-Erling Smørgrav return 0; 1667bc892140SDag-Erling Smørgrav r->local_alias->rrset = regional_alloc_init(s->s.region, 1668bc892140SDag-Erling Smørgrav qinfo->local_alias->rrset, 1669bc892140SDag-Erling Smørgrav sizeof(*qinfo->local_alias->rrset)); 1670bc892140SDag-Erling Smørgrav if(!r->local_alias->rrset) 1671bc892140SDag-Erling Smørgrav return 0; 1672bc892140SDag-Erling Smørgrav dsrc = qinfo->local_alias->rrset->entry.data; 1673bc892140SDag-Erling Smørgrav 1674bc892140SDag-Erling Smørgrav /* In the current implementation, a local alias must be 1675bc892140SDag-Erling Smørgrav * a single CNAME RR (see worker_handle_request()). */ 1676bc892140SDag-Erling Smørgrav log_assert(!qinfo->local_alias->next && dsrc->count == 1 && 1677bc892140SDag-Erling Smørgrav qinfo->local_alias->rrset->rk.type == 1678bc892140SDag-Erling Smørgrav htons(LDNS_RR_TYPE_CNAME)); 16790eefd307SCy Schubert /* we should make a local copy for the owner name of 16800eefd307SCy Schubert * the RRset */ 16810eefd307SCy Schubert r->local_alias->rrset->rk.dname_len = 16820eefd307SCy Schubert qinfo->local_alias->rrset->rk.dname_len; 16830eefd307SCy Schubert r->local_alias->rrset->rk.dname = regional_alloc_init( 16840eefd307SCy Schubert s->s.region, qinfo->local_alias->rrset->rk.dname, 16850eefd307SCy Schubert qinfo->local_alias->rrset->rk.dname_len); 16860eefd307SCy Schubert if(!r->local_alias->rrset->rk.dname) 16870eefd307SCy Schubert return 0; 1688bc892140SDag-Erling Smørgrav 1689e86b9096SDag-Erling Smørgrav /* the rrset is not packed, like in the cache, but it is 169024e36522SCy Schubert * individually allocated with an allocator from localzone. */ 1691e86b9096SDag-Erling Smørgrav d = regional_alloc_zero(s->s.region, sizeof(*d)); 1692bc892140SDag-Erling Smørgrav if(!d) 1693bc892140SDag-Erling Smørgrav return 0; 1694bc892140SDag-Erling Smørgrav r->local_alias->rrset->entry.data = d; 1695e86b9096SDag-Erling Smørgrav if(!rrset_insert_rr(s->s.region, d, dsrc->rr_data[0], 1696e86b9096SDag-Erling Smørgrav dsrc->rr_len[0], dsrc->rr_ttl[0], "CNAME local alias")) 1697bc892140SDag-Erling Smørgrav return 0; 1698bc892140SDag-Erling Smørgrav } else 1699bc892140SDag-Erling Smørgrav r->local_alias = NULL; 1700bc892140SDag-Erling Smørgrav 1701b7579f77SDag-Erling Smørgrav s->reply_list = r; 1702b7579f77SDag-Erling Smørgrav return 1; 1703b7579f77SDag-Erling Smørgrav } 1704b7579f77SDag-Erling Smørgrav 17058a384985SDag-Erling Smørgrav /* Extract the query info and flags from 'mstate' into '*qinfop' and '*qflags'. 17068a384985SDag-Erling Smørgrav * Since this is only used for internal refetch of otherwise-expired answer, 17078a384985SDag-Erling Smørgrav * we simply ignore the rare failure mode when memory allocation fails. */ 17088a384985SDag-Erling Smørgrav static void 17098a384985SDag-Erling Smørgrav mesh_copy_qinfo(struct mesh_state* mstate, struct query_info** qinfop, 17108a384985SDag-Erling Smørgrav uint16_t* qflags) 17118a384985SDag-Erling Smørgrav { 17128a384985SDag-Erling Smørgrav struct regional* region = mstate->s.env->scratch; 17138a384985SDag-Erling Smørgrav struct query_info* qinfo; 17148a384985SDag-Erling Smørgrav 17158a384985SDag-Erling Smørgrav qinfo = regional_alloc_init(region, &mstate->s.qinfo, sizeof(*qinfo)); 17168a384985SDag-Erling Smørgrav if(!qinfo) 17178a384985SDag-Erling Smørgrav return; 17188a384985SDag-Erling Smørgrav qinfo->qname = regional_alloc_init(region, qinfo->qname, 17198a384985SDag-Erling Smørgrav qinfo->qname_len); 17208a384985SDag-Erling Smørgrav if(!qinfo->qname) 17218a384985SDag-Erling Smørgrav return; 17228a384985SDag-Erling Smørgrav *qinfop = qinfo; 17238a384985SDag-Erling Smørgrav *qflags = mstate->s.query_flags; 17248a384985SDag-Erling Smørgrav } 17258a384985SDag-Erling Smørgrav 1726b7579f77SDag-Erling Smørgrav /** 1727b7579f77SDag-Erling Smørgrav * Continue processing the mesh state at another module. 17288a384985SDag-Erling Smørgrav * Handles module to modules transfer of control. 1729b7579f77SDag-Erling Smørgrav * Handles module finished. 1730b7579f77SDag-Erling Smørgrav * @param mesh: the mesh area. 1731b7579f77SDag-Erling Smørgrav * @param mstate: currently active mesh state. 1732b7579f77SDag-Erling Smørgrav * Deleted if finished, calls _done and _supers to 1733b7579f77SDag-Erling Smørgrav * send replies to clients and inform other mesh states. 1734b7579f77SDag-Erling Smørgrav * This in turn may create additional runnable mesh states. 1735b7579f77SDag-Erling Smørgrav * @param s: state at which the current module exited. 1736b7579f77SDag-Erling Smørgrav * @param ev: the event sent to the module. 1737b7579f77SDag-Erling Smørgrav * returned is the event to send to the next module. 1738b7579f77SDag-Erling Smørgrav * @return true if continue processing at the new module. 1739b7579f77SDag-Erling Smørgrav * false if not continued processing is needed. 1740b7579f77SDag-Erling Smørgrav */ 1741b7579f77SDag-Erling Smørgrav static int 1742b7579f77SDag-Erling Smørgrav mesh_continue(struct mesh_area* mesh, struct mesh_state* mstate, 1743b7579f77SDag-Erling Smørgrav enum module_ext_state s, enum module_ev* ev) 1744b7579f77SDag-Erling Smørgrav { 1745b7579f77SDag-Erling Smørgrav mstate->num_activated++; 1746b7579f77SDag-Erling Smørgrav if(mstate->num_activated > MESH_MAX_ACTIVATION) { 1747b7579f77SDag-Erling Smørgrav /* module is looping. Stop it. */ 17488a384985SDag-Erling Smørgrav log_err("internal error: looping module (%s) stopped", 17498a384985SDag-Erling Smørgrav mesh->mods.mod[mstate->s.curmod]->name); 17500eefd307SCy Schubert log_query_info(NO_VERBOSE, "pass error for qstate", 1751b7579f77SDag-Erling Smørgrav &mstate->s.qinfo); 1752b7579f77SDag-Erling Smørgrav s = module_error; 1753b7579f77SDag-Erling Smørgrav } 1754b7579f77SDag-Erling Smørgrav if(s == module_wait_module || s == module_restart_next) { 1755b7579f77SDag-Erling Smørgrav /* start next module */ 1756b7579f77SDag-Erling Smørgrav mstate->s.curmod++; 1757b7579f77SDag-Erling Smørgrav if(mesh->mods.num == mstate->s.curmod) { 1758b7579f77SDag-Erling Smørgrav log_err("Cannot pass to next module; at last module"); 1759b7579f77SDag-Erling Smørgrav log_query_info(VERB_QUERY, "pass error for qstate", 1760b7579f77SDag-Erling Smørgrav &mstate->s.qinfo); 1761b7579f77SDag-Erling Smørgrav mstate->s.curmod--; 1762b7579f77SDag-Erling Smørgrav return mesh_continue(mesh, mstate, module_error, ev); 1763b7579f77SDag-Erling Smørgrav } 1764b7579f77SDag-Erling Smørgrav if(s == module_restart_next) { 176565b390aaSDag-Erling Smørgrav int curmod = mstate->s.curmod; 176665b390aaSDag-Erling Smørgrav for(; mstate->s.curmod < mesh->mods.num; 176765b390aaSDag-Erling Smørgrav mstate->s.curmod++) { 1768b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mod_clear( 1769b7579f77SDag-Erling Smørgrav mesh->mods.mod[mstate->s.curmod]->clear)); 1770b7579f77SDag-Erling Smørgrav (*mesh->mods.mod[mstate->s.curmod]->clear) 1771b7579f77SDag-Erling Smørgrav (&mstate->s, mstate->s.curmod); 1772b7579f77SDag-Erling Smørgrav mstate->s.minfo[mstate->s.curmod] = NULL; 1773b7579f77SDag-Erling Smørgrav } 177465b390aaSDag-Erling Smørgrav mstate->s.curmod = curmod; 177565b390aaSDag-Erling Smørgrav } 1776b7579f77SDag-Erling Smørgrav *ev = module_event_pass; 1777b7579f77SDag-Erling Smørgrav return 1; 1778b7579f77SDag-Erling Smørgrav } 1779b5663de9SDag-Erling Smørgrav if(s == module_wait_subquery && mstate->sub_set.count == 0) { 1780b5663de9SDag-Erling Smørgrav log_err("module cannot wait for subquery, subquery list empty"); 1781b5663de9SDag-Erling Smørgrav log_query_info(VERB_QUERY, "pass error for qstate", 1782b5663de9SDag-Erling Smørgrav &mstate->s.qinfo); 1783b5663de9SDag-Erling Smørgrav s = module_error; 1784b5663de9SDag-Erling Smørgrav } 1785b7579f77SDag-Erling Smørgrav if(s == module_error && mstate->s.return_rcode == LDNS_RCODE_NOERROR) { 1786b7579f77SDag-Erling Smørgrav /* error is bad, handle pass back up below */ 1787b7579f77SDag-Erling Smørgrav mstate->s.return_rcode = LDNS_RCODE_SERVFAIL; 1788b7579f77SDag-Erling Smørgrav } 17898a384985SDag-Erling Smørgrav if(s == module_error) { 1790b7579f77SDag-Erling Smørgrav mesh_query_done(mstate); 1791b7579f77SDag-Erling Smørgrav mesh_walk_supers(mesh, mstate); 1792b7579f77SDag-Erling Smørgrav mesh_state_delete(&mstate->s); 1793b7579f77SDag-Erling Smørgrav return 0; 1794b7579f77SDag-Erling Smørgrav } 17958a384985SDag-Erling Smørgrav if(s == module_finished) { 17968a384985SDag-Erling Smørgrav if(mstate->s.curmod == 0) { 17978a384985SDag-Erling Smørgrav struct query_info* qinfo = NULL; 17988f76bb7dSCy Schubert struct edns_option* opt_list = NULL; 17998f76bb7dSCy Schubert struct sockaddr_storage addr; 18008a384985SDag-Erling Smørgrav uint16_t qflags; 1801a39a5a69SCy Schubert int rpz_p = 0; 18028a384985SDag-Erling Smørgrav 18038f76bb7dSCy Schubert #ifdef CLIENT_SUBNET 18048f76bb7dSCy Schubert struct edns_option* ecs; 18058f76bb7dSCy Schubert if(mstate->s.need_refetch && mstate->reply_list && 18068f76bb7dSCy Schubert modstack_find(&mesh->mods, "subnetcache") != -1 && 18078f76bb7dSCy Schubert mstate->s.env->unique_mesh) { 18088f76bb7dSCy Schubert addr = mstate->reply_list->query_reply.client_addr; 18098f76bb7dSCy Schubert } else 18108f76bb7dSCy Schubert #endif 18118f76bb7dSCy Schubert memset(&addr, 0, sizeof(addr)); 18128f76bb7dSCy Schubert 18138a384985SDag-Erling Smørgrav mesh_query_done(mstate); 18148a384985SDag-Erling Smørgrav mesh_walk_supers(mesh, mstate); 18158a384985SDag-Erling Smørgrav 18168a384985SDag-Erling Smørgrav /* If the answer to the query needs to be refetched 18178a384985SDag-Erling Smørgrav * from an external DNS server, we'll need to schedule 18188a384985SDag-Erling Smørgrav * a prefetch after removing the current state, so 18198a384985SDag-Erling Smørgrav * we need to make a copy of the query info here. */ 1820a39a5a69SCy Schubert if(mstate->s.need_refetch) { 18218a384985SDag-Erling Smørgrav mesh_copy_qinfo(mstate, &qinfo, &qflags); 18228f76bb7dSCy Schubert #ifdef CLIENT_SUBNET 18238f76bb7dSCy Schubert /* Make also a copy of the ecs option if any */ 18248f76bb7dSCy Schubert if((ecs = edns_opt_list_find( 18258f76bb7dSCy Schubert mstate->s.edns_opts_front_in, 18268f76bb7dSCy Schubert mstate->s.env->cfg->client_subnet_opcode)) != NULL) { 18278f76bb7dSCy Schubert (void)edns_opt_list_append(&opt_list, 18288f76bb7dSCy Schubert ecs->opt_code, ecs->opt_len, 18298f76bb7dSCy Schubert ecs->opt_data, 18308f76bb7dSCy Schubert mstate->s.env->scratch); 18318f76bb7dSCy Schubert } 18328f76bb7dSCy Schubert #endif 1833a39a5a69SCy Schubert rpz_p = mstate->s.rpz_passthru; 1834a39a5a69SCy Schubert } 18358a384985SDag-Erling Smørgrav 18368a384985SDag-Erling Smørgrav if(qinfo) { 18378f76bb7dSCy Schubert mesh_state_delete(&mstate->s); 18388f76bb7dSCy Schubert mesh_new_prefetch(mesh, qinfo, qflags, 0, 18398f76bb7dSCy Schubert rpz_p, 18408f76bb7dSCy Schubert addr.ss_family!=AF_UNSPEC?&addr:NULL, 18418f76bb7dSCy Schubert opt_list); 18428f76bb7dSCy Schubert } else { 18438f76bb7dSCy Schubert mesh_state_delete(&mstate->s); 18448a384985SDag-Erling Smørgrav } 18458a384985SDag-Erling Smørgrav return 0; 18468a384985SDag-Erling Smørgrav } 1847b7579f77SDag-Erling Smørgrav /* pass along the locus of control */ 1848b7579f77SDag-Erling Smørgrav mstate->s.curmod --; 1849b7579f77SDag-Erling Smørgrav *ev = module_event_moddone; 1850b7579f77SDag-Erling Smørgrav return 1; 1851b7579f77SDag-Erling Smørgrav } 1852b7579f77SDag-Erling Smørgrav return 0; 1853b7579f77SDag-Erling Smørgrav } 1854b7579f77SDag-Erling Smørgrav 1855b7579f77SDag-Erling Smørgrav void mesh_run(struct mesh_area* mesh, struct mesh_state* mstate, 1856b7579f77SDag-Erling Smørgrav enum module_ev ev, struct outbound_entry* e) 1857b7579f77SDag-Erling Smørgrav { 1858b7579f77SDag-Erling Smørgrav enum module_ext_state s; 1859b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "mesh_run: start"); 1860b7579f77SDag-Erling Smørgrav while(mstate) { 1861b7579f77SDag-Erling Smørgrav /* run the module */ 1862b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mod_operate( 1863b7579f77SDag-Erling Smørgrav mesh->mods.mod[mstate->s.curmod]->operate)); 1864b7579f77SDag-Erling Smørgrav (*mesh->mods.mod[mstate->s.curmod]->operate) 1865b7579f77SDag-Erling Smørgrav (&mstate->s, ev, mstate->s.curmod, e); 1866b7579f77SDag-Erling Smørgrav 1867b7579f77SDag-Erling Smørgrav /* examine results */ 1868b7579f77SDag-Erling Smørgrav mstate->s.reply = NULL; 1869b7579f77SDag-Erling Smørgrav regional_free_all(mstate->s.env->scratch); 1870b7579f77SDag-Erling Smørgrav s = mstate->s.ext_state[mstate->s.curmod]; 1871b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "mesh_run: %s module exit state is %s", 1872b7579f77SDag-Erling Smørgrav mesh->mods.mod[mstate->s.curmod]->name, strextstate(s)); 1873b7579f77SDag-Erling Smørgrav e = NULL; 1874b7579f77SDag-Erling Smørgrav if(mesh_continue(mesh, mstate, s, &ev)) 1875b7579f77SDag-Erling Smørgrav continue; 1876b7579f77SDag-Erling Smørgrav 1877b7579f77SDag-Erling Smørgrav /* run more modules */ 1878b7579f77SDag-Erling Smørgrav ev = module_event_pass; 1879b7579f77SDag-Erling Smørgrav if(mesh->run.count > 0) { 1880b7579f77SDag-Erling Smørgrav /* pop random element off the runnable tree */ 1881b7579f77SDag-Erling Smørgrav mstate = (struct mesh_state*)mesh->run.root->key; 1882b7579f77SDag-Erling Smørgrav (void)rbtree_delete(&mesh->run, mstate); 1883b7579f77SDag-Erling Smørgrav } else mstate = NULL; 1884b7579f77SDag-Erling Smørgrav } 1885b7579f77SDag-Erling Smørgrav if(verbosity >= VERB_ALGO) { 1886b7579f77SDag-Erling Smørgrav mesh_stats(mesh, "mesh_run: end"); 1887b7579f77SDag-Erling Smørgrav mesh_log_list(mesh); 1888b7579f77SDag-Erling Smørgrav } 1889b7579f77SDag-Erling Smørgrav } 1890b7579f77SDag-Erling Smørgrav 1891b7579f77SDag-Erling Smørgrav void 1892b7579f77SDag-Erling Smørgrav mesh_log_list(struct mesh_area* mesh) 1893b7579f77SDag-Erling Smørgrav { 1894b7579f77SDag-Erling Smørgrav char buf[30]; 1895b7579f77SDag-Erling Smørgrav struct mesh_state* m; 1896b7579f77SDag-Erling Smørgrav int num = 0; 1897b7579f77SDag-Erling Smørgrav RBTREE_FOR(m, struct mesh_state*, &mesh->all) { 1898ff825849SDag-Erling Smørgrav snprintf(buf, sizeof(buf), "%d%s%s%s%s%s%s mod%d %s%s", 1899b7579f77SDag-Erling Smørgrav num++, (m->s.is_priming)?"p":"", /* prime */ 1900ff825849SDag-Erling Smørgrav (m->s.is_valrec)?"v":"", /* prime */ 1901b7579f77SDag-Erling Smørgrav (m->s.query_flags&BIT_RD)?"RD":"", 1902b7579f77SDag-Erling Smørgrav (m->s.query_flags&BIT_CD)?"CD":"", 1903b7579f77SDag-Erling Smørgrav (m->super_set.count==0)?"d":"", /* detached */ 1904b7579f77SDag-Erling Smørgrav (m->sub_set.count!=0)?"c":"", /* children */ 1905b7579f77SDag-Erling Smørgrav m->s.curmod, (m->reply_list)?"rep":"", /*hasreply*/ 1906b7579f77SDag-Erling Smørgrav (m->cb_list)?"cb":"" /* callbacks */ 1907b7579f77SDag-Erling Smørgrav ); 1908b7579f77SDag-Erling Smørgrav log_query_info(VERB_ALGO, buf, &m->s.qinfo); 1909b7579f77SDag-Erling Smørgrav } 1910b7579f77SDag-Erling Smørgrav } 1911b7579f77SDag-Erling Smørgrav 1912b7579f77SDag-Erling Smørgrav void 1913b7579f77SDag-Erling Smørgrav mesh_stats(struct mesh_area* mesh, const char* str) 1914b7579f77SDag-Erling Smørgrav { 1915b7579f77SDag-Erling Smørgrav verbose(VERB_DETAIL, "%s %u recursion states (%u with reply, " 1916b7579f77SDag-Erling Smørgrav "%u detached), %u waiting replies, %u recursion replies " 1917b7579f77SDag-Erling Smørgrav "sent, %d replies dropped, %d states jostled out", 1918b7579f77SDag-Erling Smørgrav str, (unsigned)mesh->all.count, 1919b7579f77SDag-Erling Smørgrav (unsigned)mesh->num_reply_states, 1920b7579f77SDag-Erling Smørgrav (unsigned)mesh->num_detached_states, 1921b7579f77SDag-Erling Smørgrav (unsigned)mesh->num_reply_addrs, 1922b7579f77SDag-Erling Smørgrav (unsigned)mesh->replies_sent, 1923b7579f77SDag-Erling Smørgrav (unsigned)mesh->stats_dropped, 1924b7579f77SDag-Erling Smørgrav (unsigned)mesh->stats_jostled); 1925b7579f77SDag-Erling Smørgrav if(mesh->replies_sent > 0) { 1926b7579f77SDag-Erling Smørgrav struct timeval avg; 1927b7579f77SDag-Erling Smørgrav timeval_divide(&avg, &mesh->replies_sum_wait, 1928b7579f77SDag-Erling Smørgrav mesh->replies_sent); 1929b7579f77SDag-Erling Smørgrav log_info("average recursion processing time " 193017d15b25SDag-Erling Smørgrav ARG_LL "d.%6.6d sec", 193117d15b25SDag-Erling Smørgrav (long long)avg.tv_sec, (int)avg.tv_usec); 1932b7579f77SDag-Erling Smørgrav log_info("histogram of recursion processing times"); 1933b7579f77SDag-Erling Smørgrav timehist_log(mesh->histogram, "recursions"); 1934b7579f77SDag-Erling Smørgrav } 1935b7579f77SDag-Erling Smørgrav } 1936b7579f77SDag-Erling Smørgrav 1937b7579f77SDag-Erling Smørgrav void 1938b7579f77SDag-Erling Smørgrav mesh_stats_clear(struct mesh_area* mesh) 1939b7579f77SDag-Erling Smørgrav { 1940b7579f77SDag-Erling Smørgrav if(!mesh) 1941b7579f77SDag-Erling Smørgrav return; 1942b7579f77SDag-Erling Smørgrav mesh->replies_sent = 0; 1943b7579f77SDag-Erling Smørgrav mesh->replies_sum_wait.tv_sec = 0; 1944b7579f77SDag-Erling Smørgrav mesh->replies_sum_wait.tv_usec = 0; 1945b7579f77SDag-Erling Smørgrav mesh->stats_jostled = 0; 1946b7579f77SDag-Erling Smørgrav mesh->stats_dropped = 0; 1947b7579f77SDag-Erling Smørgrav timehist_clear(mesh->histogram); 1948b7579f77SDag-Erling Smørgrav mesh->ans_secure = 0; 1949b7579f77SDag-Erling Smørgrav mesh->ans_bogus = 0; 1950091e9e46SCy Schubert mesh->ans_expired = 0; 19518f76bb7dSCy Schubert mesh->ans_cachedb = 0; 1952091e9e46SCy Schubert memset(&mesh->ans_rcode[0], 0, sizeof(size_t)*UB_STATS_RCODE_NUM); 1953091e9e46SCy Schubert memset(&mesh->rpz_action[0], 0, sizeof(size_t)*UB_STATS_RPZ_ACTION_NUM); 1954b7579f77SDag-Erling Smørgrav mesh->ans_nodata = 0; 1955b7579f77SDag-Erling Smørgrav } 1956b7579f77SDag-Erling Smørgrav 1957b7579f77SDag-Erling Smørgrav size_t 1958b7579f77SDag-Erling Smørgrav mesh_get_mem(struct mesh_area* mesh) 1959b7579f77SDag-Erling Smørgrav { 1960b7579f77SDag-Erling Smørgrav struct mesh_state* m; 1961b7579f77SDag-Erling Smørgrav size_t s = sizeof(*mesh) + sizeof(struct timehist) + 1962b7579f77SDag-Erling Smørgrav sizeof(struct th_buck)*mesh->histogram->num + 196317d15b25SDag-Erling Smørgrav sizeof(sldns_buffer) + sldns_buffer_capacity(mesh->qbuf_bak); 1964b7579f77SDag-Erling Smørgrav RBTREE_FOR(m, struct mesh_state*, &mesh->all) { 1965b7579f77SDag-Erling Smørgrav /* all, including m itself allocated in qstate region */ 1966b7579f77SDag-Erling Smørgrav s += regional_get_mem(m->s.region); 1967b7579f77SDag-Erling Smørgrav } 1968b7579f77SDag-Erling Smørgrav return s; 1969b7579f77SDag-Erling Smørgrav } 1970b7579f77SDag-Erling Smørgrav 1971b7579f77SDag-Erling Smørgrav int 1972b7579f77SDag-Erling Smørgrav mesh_detect_cycle(struct module_qstate* qstate, struct query_info* qinfo, 1973ff825849SDag-Erling Smørgrav uint16_t flags, int prime, int valrec) 1974b7579f77SDag-Erling Smørgrav { 1975b7579f77SDag-Erling Smørgrav struct mesh_area* mesh = qstate->env->mesh; 1976bc892140SDag-Erling Smørgrav struct mesh_state* dep_m = NULL; 197765b390aaSDag-Erling Smørgrav dep_m = mesh_area_find(mesh, NULL, qinfo, flags, prime, valrec); 1978b7579f77SDag-Erling Smørgrav return mesh_detect_cycle_found(qstate, dep_m); 1979b7579f77SDag-Erling Smørgrav } 1980b7579f77SDag-Erling Smørgrav 1981b7579f77SDag-Erling Smørgrav void mesh_list_insert(struct mesh_state* m, struct mesh_state** fp, 1982b7579f77SDag-Erling Smørgrav struct mesh_state** lp) 1983b7579f77SDag-Erling Smørgrav { 1984b7579f77SDag-Erling Smørgrav /* insert as last element */ 1985b7579f77SDag-Erling Smørgrav m->prev = *lp; 1986b7579f77SDag-Erling Smørgrav m->next = NULL; 1987b7579f77SDag-Erling Smørgrav if(*lp) 1988b7579f77SDag-Erling Smørgrav (*lp)->next = m; 1989b7579f77SDag-Erling Smørgrav else *fp = m; 1990b7579f77SDag-Erling Smørgrav *lp = m; 1991b7579f77SDag-Erling Smørgrav } 1992b7579f77SDag-Erling Smørgrav 1993b7579f77SDag-Erling Smørgrav void mesh_list_remove(struct mesh_state* m, struct mesh_state** fp, 1994b7579f77SDag-Erling Smørgrav struct mesh_state** lp) 1995b7579f77SDag-Erling Smørgrav { 1996b7579f77SDag-Erling Smørgrav if(m->next) 1997b7579f77SDag-Erling Smørgrav m->next->prev = m->prev; 1998b7579f77SDag-Erling Smørgrav else *lp = m->prev; 1999b7579f77SDag-Erling Smørgrav if(m->prev) 2000b7579f77SDag-Erling Smørgrav m->prev->next = m->next; 2001b7579f77SDag-Erling Smørgrav else *fp = m->next; 2002b7579f77SDag-Erling Smørgrav } 2003e86b9096SDag-Erling Smørgrav 2004e86b9096SDag-Erling Smørgrav void mesh_state_remove_reply(struct mesh_area* mesh, struct mesh_state* m, 2005e86b9096SDag-Erling Smørgrav struct comm_point* cp) 2006e86b9096SDag-Erling Smørgrav { 2007e86b9096SDag-Erling Smørgrav struct mesh_reply* n, *prev = NULL; 2008e86b9096SDag-Erling Smørgrav n = m->reply_list; 2009e86b9096SDag-Erling Smørgrav /* when in mesh_cleanup, it sets the reply_list to NULL, so that 2010e86b9096SDag-Erling Smørgrav * there is no accounting twice */ 2011e86b9096SDag-Erling Smørgrav if(!n) return; /* nothing to remove, also no accounting needed */ 2012e86b9096SDag-Erling Smørgrav while(n) { 2013e86b9096SDag-Erling Smørgrav if(n->query_reply.c == cp) { 2014e86b9096SDag-Erling Smørgrav /* unlink it */ 2015e86b9096SDag-Erling Smørgrav if(prev) prev->next = n->next; 2016e86b9096SDag-Erling Smørgrav else m->reply_list = n->next; 2017e86b9096SDag-Erling Smørgrav /* delete it, but allocated in m region */ 2018091e9e46SCy Schubert log_assert(mesh->num_reply_addrs > 0); 2019e86b9096SDag-Erling Smørgrav mesh->num_reply_addrs--; 2020e86b9096SDag-Erling Smørgrav 2021e86b9096SDag-Erling Smørgrav /* prev = prev; */ 2022e86b9096SDag-Erling Smørgrav n = n->next; 2023e86b9096SDag-Erling Smørgrav continue; 2024e86b9096SDag-Erling Smørgrav } 2025e86b9096SDag-Erling Smørgrav prev = n; 2026e86b9096SDag-Erling Smørgrav n = n->next; 2027e86b9096SDag-Erling Smørgrav } 2028e86b9096SDag-Erling Smørgrav /* it was not detached (because it had a reply list), could be now */ 2029e86b9096SDag-Erling Smørgrav if(!m->reply_list && !m->cb_list 2030e86b9096SDag-Erling Smørgrav && m->super_set.count == 0) { 2031e86b9096SDag-Erling Smørgrav mesh->num_detached_states++; 2032e86b9096SDag-Erling Smørgrav } 2033e86b9096SDag-Erling Smørgrav /* if not replies any more in mstate, it is no longer a reply_state */ 2034e86b9096SDag-Erling Smørgrav if(!m->reply_list && !m->cb_list) { 2035e86b9096SDag-Erling Smørgrav log_assert(mesh->num_reply_states > 0); 2036e86b9096SDag-Erling Smørgrav mesh->num_reply_states--; 2037e86b9096SDag-Erling Smørgrav } 2038e86b9096SDag-Erling Smørgrav } 2039091e9e46SCy Schubert 2040091e9e46SCy Schubert 2041091e9e46SCy Schubert static int 2042091e9e46SCy Schubert apply_respip_action(struct module_qstate* qstate, 2043091e9e46SCy Schubert const struct query_info* qinfo, struct respip_client_info* cinfo, 2044091e9e46SCy Schubert struct respip_action_info* actinfo, struct reply_info* rep, 2045091e9e46SCy Schubert struct ub_packed_rrset_key** alias_rrset, 2046091e9e46SCy Schubert struct reply_info** encode_repp, struct auth_zones* az) 2047091e9e46SCy Schubert { 2048091e9e46SCy Schubert if(qinfo->qtype != LDNS_RR_TYPE_A && 2049091e9e46SCy Schubert qinfo->qtype != LDNS_RR_TYPE_AAAA && 2050091e9e46SCy Schubert qinfo->qtype != LDNS_RR_TYPE_ANY) 2051091e9e46SCy Schubert return 1; 2052091e9e46SCy Schubert 2053091e9e46SCy Schubert if(!respip_rewrite_reply(qinfo, cinfo, rep, encode_repp, actinfo, 2054a39a5a69SCy Schubert alias_rrset, 0, qstate->region, az, NULL)) 2055091e9e46SCy Schubert return 0; 2056091e9e46SCy Schubert 2057091e9e46SCy Schubert /* xxx_deny actions mean dropping the reply, unless the original reply 2058091e9e46SCy Schubert * was redirected to response-ip data. */ 2059091e9e46SCy Schubert if((actinfo->action == respip_deny || 2060091e9e46SCy Schubert actinfo->action == respip_inform_deny) && 2061091e9e46SCy Schubert *encode_repp == rep) 2062091e9e46SCy Schubert *encode_repp = NULL; 2063091e9e46SCy Schubert 2064091e9e46SCy Schubert return 1; 2065091e9e46SCy Schubert } 2066091e9e46SCy Schubert 2067091e9e46SCy Schubert void 2068091e9e46SCy Schubert mesh_serve_expired_callback(void* arg) 2069091e9e46SCy Schubert { 2070091e9e46SCy Schubert struct mesh_state* mstate = (struct mesh_state*) arg; 2071091e9e46SCy Schubert struct module_qstate* qstate = &mstate->s; 2072369c6923SCy Schubert struct mesh_reply* r; 2073091e9e46SCy Schubert struct mesh_area* mesh = qstate->env->mesh; 2074091e9e46SCy Schubert struct dns_msg* msg; 2075091e9e46SCy Schubert struct mesh_cb* c; 2076091e9e46SCy Schubert struct mesh_reply* prev = NULL; 2077091e9e46SCy Schubert struct sldns_buffer* prev_buffer = NULL; 2078091e9e46SCy Schubert struct sldns_buffer* r_buffer = NULL; 2079091e9e46SCy Schubert struct reply_info* partial_rep = NULL; 2080091e9e46SCy Schubert struct ub_packed_rrset_key* alias_rrset = NULL; 2081091e9e46SCy Schubert struct reply_info* encode_rep = NULL; 2082091e9e46SCy Schubert struct respip_action_info actinfo; 2083091e9e46SCy Schubert struct query_info* lookup_qinfo = &qstate->qinfo; 2084091e9e46SCy Schubert struct query_info qinfo_tmp; 2085f44e67d1SCy Schubert struct timeval tv = {0, 0}; 2086091e9e46SCy Schubert int must_validate = (!(qstate->query_flags&BIT_CD) 2087091e9e46SCy Schubert || qstate->env->cfg->ignore_cd) && qstate->env->need_to_validate; 20888f76bb7dSCy Schubert int i = 0; 2089091e9e46SCy Schubert if(!qstate->serve_expired_data) return; 2090091e9e46SCy Schubert verbose(VERB_ALGO, "Serve expired: Trying to reply with expired data"); 2091091e9e46SCy Schubert comm_timer_delete(qstate->serve_expired_data->timer); 2092091e9e46SCy Schubert qstate->serve_expired_data->timer = NULL; 2093f44e67d1SCy Schubert /* If is_drop or no_cache_lookup (modules that handle their own cache e.g., 2094f44e67d1SCy Schubert * subnetmod) ignore stale data from the main cache. */ 2095f44e67d1SCy Schubert if(qstate->no_cache_lookup || qstate->is_drop) { 2096091e9e46SCy Schubert verbose(VERB_ALGO, 2097091e9e46SCy Schubert "Serve expired: Not allowed to look into cache for stale"); 2098091e9e46SCy Schubert return; 2099091e9e46SCy Schubert } 2100091e9e46SCy Schubert /* The following while is used instead of the `goto lookup_cache` 2101091e9e46SCy Schubert * like in the worker. */ 2102091e9e46SCy Schubert while(1) { 2103091e9e46SCy Schubert fptr_ok(fptr_whitelist_serve_expired_lookup( 2104091e9e46SCy Schubert qstate->serve_expired_data->get_cached_answer)); 21055469a995SCy Schubert msg = (*qstate->serve_expired_data->get_cached_answer)(qstate, 2106091e9e46SCy Schubert lookup_qinfo); 2107091e9e46SCy Schubert if(!msg) 2108091e9e46SCy Schubert return; 2109091e9e46SCy Schubert /* Reset these in case we pass a second time from here. */ 2110091e9e46SCy Schubert encode_rep = msg->rep; 2111091e9e46SCy Schubert memset(&actinfo, 0, sizeof(actinfo)); 2112091e9e46SCy Schubert actinfo.action = respip_none; 2113091e9e46SCy Schubert alias_rrset = NULL; 2114091e9e46SCy Schubert if((mesh->use_response_ip || mesh->use_rpz) && 2115091e9e46SCy Schubert !partial_rep && !apply_respip_action(qstate, &qstate->qinfo, 2116091e9e46SCy Schubert qstate->client_info, &actinfo, msg->rep, &alias_rrset, &encode_rep, 2117091e9e46SCy Schubert qstate->env->auth_zones)) { 2118091e9e46SCy Schubert return; 2119091e9e46SCy Schubert } else if(partial_rep && 2120091e9e46SCy Schubert !respip_merge_cname(partial_rep, &qstate->qinfo, msg->rep, 2121091e9e46SCy Schubert qstate->client_info, must_validate, &encode_rep, qstate->region, 2122091e9e46SCy Schubert qstate->env->auth_zones)) { 2123091e9e46SCy Schubert return; 2124091e9e46SCy Schubert } 2125091e9e46SCy Schubert if(!encode_rep || alias_rrset) { 2126091e9e46SCy Schubert if(!encode_rep) { 2127091e9e46SCy Schubert /* Needs drop */ 2128091e9e46SCy Schubert return; 2129091e9e46SCy Schubert } else { 2130091e9e46SCy Schubert /* A partial CNAME chain is found. */ 2131091e9e46SCy Schubert partial_rep = encode_rep; 2132091e9e46SCy Schubert } 2133091e9e46SCy Schubert } 2134091e9e46SCy Schubert /* We've found a partial reply ending with an 2135091e9e46SCy Schubert * alias. Replace the lookup qinfo for the 2136091e9e46SCy Schubert * alias target and lookup the cache again to 2137091e9e46SCy Schubert * (possibly) complete the reply. As we're 2138091e9e46SCy Schubert * passing the "base" reply, there will be no 2139091e9e46SCy Schubert * more alias chasing. */ 2140091e9e46SCy Schubert if(partial_rep) { 2141091e9e46SCy Schubert memset(&qinfo_tmp, 0, sizeof(qinfo_tmp)); 2142091e9e46SCy Schubert get_cname_target(alias_rrset, &qinfo_tmp.qname, 2143091e9e46SCy Schubert &qinfo_tmp.qname_len); 2144091e9e46SCy Schubert if(!qinfo_tmp.qname) { 2145091e9e46SCy Schubert log_err("Serve expired: unexpected: invalid answer alias"); 2146091e9e46SCy Schubert return; 2147091e9e46SCy Schubert } 2148091e9e46SCy Schubert qinfo_tmp.qtype = qstate->qinfo.qtype; 2149091e9e46SCy Schubert qinfo_tmp.qclass = qstate->qinfo.qclass; 2150091e9e46SCy Schubert lookup_qinfo = &qinfo_tmp; 2151091e9e46SCy Schubert continue; 2152091e9e46SCy Schubert } 2153091e9e46SCy Schubert break; 2154091e9e46SCy Schubert } 2155091e9e46SCy Schubert 2156091e9e46SCy Schubert if(verbosity >= VERB_ALGO) 2157091e9e46SCy Schubert log_dns_msg("Serve expired lookup", &qstate->qinfo, msg->rep); 2158091e9e46SCy Schubert 215925039b37SCy Schubert for(r = mstate->reply_list; r; r = r->next) { 21608f76bb7dSCy Schubert i++; 2161f44e67d1SCy Schubert tv = r->start_time; 2162f44e67d1SCy Schubert 2163091e9e46SCy Schubert /* If address info is returned, it means the action should be an 2164091e9e46SCy Schubert * 'inform' variant and the information should be logged. */ 2165091e9e46SCy Schubert if(actinfo.addrinfo) { 2166091e9e46SCy Schubert respip_inform_print(&actinfo, r->qname, 2167091e9e46SCy Schubert qstate->qinfo.qtype, qstate->qinfo.qclass, 2168865f46b2SCy Schubert r->local_alias, &r->query_reply.client_addr, 2169865f46b2SCy Schubert r->query_reply.client_addrlen); 2170091e9e46SCy Schubert } 2171091e9e46SCy Schubert 2172a39a5a69SCy Schubert /* Add EDE Stale Answer (RCF8914). Ignore global ede as this is 2173a39a5a69SCy Schubert * warning instead of an error */ 2174a39a5a69SCy Schubert if (r->edns.edns_present && qstate->env->cfg->ede_serve_expired && 2175a39a5a69SCy Schubert qstate->env->cfg->ede) { 2176a39a5a69SCy Schubert edns_opt_list_append_ede(&r->edns.opt_list_out, 2177a39a5a69SCy Schubert mstate->s.region, LDNS_EDE_STALE_ANSWER, NULL); 2178a39a5a69SCy Schubert } 2179a39a5a69SCy Schubert 2180091e9e46SCy Schubert r_buffer = r->query_reply.c->buffer; 2181091e9e46SCy Schubert if(r->query_reply.c->tcp_req_info) 2182091e9e46SCy Schubert r_buffer = r->query_reply.c->tcp_req_info->spool_buffer; 2183091e9e46SCy Schubert mesh_send_reply(mstate, LDNS_RCODE_NOERROR, msg->rep, 2184091e9e46SCy Schubert r, r_buffer, prev, prev_buffer); 2185091e9e46SCy Schubert if(r->query_reply.c->tcp_req_info) 2186091e9e46SCy Schubert tcp_req_info_remove_mesh_state(r->query_reply.c->tcp_req_info, mstate); 2187091e9e46SCy Schubert prev = r; 2188091e9e46SCy Schubert prev_buffer = r_buffer; 2189091e9e46SCy Schubert } 21908f76bb7dSCy Schubert /* Account for each reply sent. */ 21918f76bb7dSCy Schubert if(i > 0) { 21928f76bb7dSCy Schubert mesh->ans_expired += i; 21938f76bb7dSCy Schubert if(actinfo.addrinfo && qstate->env->cfg->stat_extended && 21948f76bb7dSCy Schubert actinfo.rpz_used) { 21958f76bb7dSCy Schubert if(actinfo.rpz_disabled) 21968f76bb7dSCy Schubert qstate->env->mesh->rpz_action[RPZ_DISABLED_ACTION] += i; 21978f76bb7dSCy Schubert if(actinfo.rpz_cname_override) 21988f76bb7dSCy Schubert qstate->env->mesh->rpz_action[RPZ_CNAME_OVERRIDE_ACTION] += i; 21998f76bb7dSCy Schubert else 22008f76bb7dSCy Schubert qstate->env->mesh->rpz_action[ 22018f76bb7dSCy Schubert respip_action_to_rpz_action(actinfo.action)] += i; 22028f76bb7dSCy Schubert } 22038f76bb7dSCy Schubert } 22048f76bb7dSCy Schubert 22058f76bb7dSCy Schubert /* Mesh area accounting */ 220625039b37SCy Schubert if(mstate->reply_list) { 220725039b37SCy Schubert mstate->reply_list = NULL; 220825039b37SCy Schubert if(!mstate->reply_list && !mstate->cb_list) { 220925039b37SCy Schubert log_assert(mesh->num_reply_states > 0); 221025039b37SCy Schubert mesh->num_reply_states--; 221125039b37SCy Schubert if(mstate->super_set.count == 0) { 221225039b37SCy Schubert mesh->num_detached_states++; 221325039b37SCy Schubert } 221425039b37SCy Schubert } 221525039b37SCy Schubert } 22168f76bb7dSCy Schubert 2217091e9e46SCy Schubert while((c = mstate->cb_list) != NULL) { 2218091e9e46SCy Schubert /* take this cb off the list; so that the list can be 2219091e9e46SCy Schubert * changed, eg. by adds from the callback routine */ 2220091e9e46SCy Schubert if(!mstate->reply_list && mstate->cb_list && !c->next) { 2221091e9e46SCy Schubert /* was a reply state, not anymore */ 2222091e9e46SCy Schubert log_assert(qstate->env->mesh->num_reply_states > 0); 2223091e9e46SCy Schubert qstate->env->mesh->num_reply_states--; 2224091e9e46SCy Schubert } 2225091e9e46SCy Schubert mstate->cb_list = c->next; 2226091e9e46SCy Schubert if(!mstate->reply_list && !mstate->cb_list && 2227091e9e46SCy Schubert mstate->super_set.count == 0) 2228091e9e46SCy Schubert qstate->env->mesh->num_detached_states++; 2229f44e67d1SCy Schubert mesh_do_callback(mstate, LDNS_RCODE_NOERROR, msg->rep, c, &tv); 2230091e9e46SCy Schubert } 2231091e9e46SCy Schubert } 22324f5c8956SCy Schubert 22334f5c8956SCy Schubert int mesh_jostle_exceeded(struct mesh_area* mesh) 22344f5c8956SCy Schubert { 22354f5c8956SCy Schubert if(mesh->all.count < mesh->max_reply_states) 22364f5c8956SCy Schubert return 0; 22374f5c8956SCy Schubert return 1; 22384f5c8956SCy Schubert } 2239