1 /* 2 * services/localzone.c - local zones authority service. 3 * 4 * Copyright (c) 2007, NLnet Labs. All rights reserved. 5 * 6 * This software is open source. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 12 * Redistributions of source code must retain the above copyright notice, 13 * this list of conditions and the following disclaimer. 14 * 15 * Redistributions in binary form must reproduce the above copyright notice, 16 * this list of conditions and the following disclaimer in the documentation 17 * and/or other materials provided with the distribution. 18 * 19 * Neither the name of the NLNET LABS nor the names of its contributors may 20 * be used to endorse or promote products derived from this software without 21 * specific prior written permission. 22 * 23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 27 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 29 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 30 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 31 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 32 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 33 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 34 */ 35 36 /** 37 * \file 38 * 39 * This file contains functions to enable local zone authority service. 40 */ 41 #include "config.h" 42 #include "services/localzone.h" 43 #include "sldns/str2wire.h" 44 #include "util/regional.h" 45 #include "util/config_file.h" 46 #include "util/data/dname.h" 47 #include "util/data/packed_rrset.h" 48 #include "util/data/msgencode.h" 49 #include "util/net_help.h" 50 #include "util/netevent.h" 51 #include "util/data/msgreply.h" 52 #include "util/data/msgparse.h" 53 #include "util/as112.h" 54 55 /* maximum RRs in an RRset, to cap possible 'endless' list RRs. 56 * with 16 bytes for an A record, a 64K packet has about 4000 max */ 57 #define LOCALZONE_RRSET_COUNT_MAX 4096 58 59 static const char* default_zones_reverse_array[] = { 60 "127.in-addr.arpa.", /* reverse ip4 zone */ 61 "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.", /* reverse ip6 zone */ 62 0 63 }; 64 const char** local_zones_default_reverse = default_zones_reverse_array; 65 66 static const char* default_zones_special_array[] = { 67 "test.", /* RFC 6761 */ 68 "invalid.", /* RFC 6761 */ 69 "onion.", /* RFC 7686 */ 70 "home.arpa.", /* RFC 8375 */ 71 "resolver.arpa.", /* RFC 9462 */ 72 "service.arpa.", /* RFC 9665 */ 73 0 74 }; 75 const char** local_zones_default_special = default_zones_special_array; 76 77 /** print all RRsets in local zone */ 78 static void 79 local_zone_out(struct local_zone* z) 80 { 81 struct local_data* d; 82 struct local_rrset* p; 83 RBTREE_FOR(d, struct local_data*, &z->data) { 84 for(p = d->rrsets; p; p = p->next) { 85 log_nametypeclass(NO_VERBOSE, "rrset", d->name, 86 ntohs(p->rrset->rk.type), 87 ntohs(p->rrset->rk.rrset_class)); 88 } 89 } 90 } 91 92 static void 93 local_zone_print(struct local_zone* z) 94 { 95 char buf[64]; 96 lock_rw_rdlock(&z->lock); 97 snprintf(buf, sizeof(buf), "%s zone", 98 local_zone_type2str(z->type)); 99 log_nametypeclass(NO_VERBOSE, buf, z->name, 0, z->dclass); 100 local_zone_out(z); 101 lock_rw_unlock(&z->lock); 102 } 103 104 void local_zones_print(struct local_zones* zones) 105 { 106 struct local_zone* z; 107 lock_rw_rdlock(&zones->lock); 108 log_info("number of auth zones %u", (unsigned)zones->ztree.count); 109 RBTREE_FOR(z, struct local_zone*, &zones->ztree) { 110 local_zone_print(z); 111 } 112 lock_rw_unlock(&zones->lock); 113 } 114 115 struct local_zones* 116 local_zones_create(void) 117 { 118 struct local_zones* zones = (struct local_zones*)calloc(1, 119 sizeof(*zones)); 120 if(!zones) 121 return NULL; 122 rbtree_init(&zones->ztree, &local_zone_cmp); 123 lock_rw_init(&zones->lock); 124 lock_protect(&zones->lock, &zones->ztree, sizeof(zones->ztree)); 125 /* also lock protects the rbnode's in struct local_zone */ 126 return zones; 127 } 128 129 /** helper traverse to delete zones */ 130 static void 131 lzdel(rbnode_type* n, void* ATTR_UNUSED(arg)) 132 { 133 struct local_zone* z = (struct local_zone*)n->key; 134 local_zone_delete(z); 135 } 136 137 void 138 local_zones_delete(struct local_zones* zones) 139 { 140 if(!zones) 141 return; 142 lock_rw_destroy(&zones->lock); 143 /* walk through zones and delete them all */ 144 traverse_postorder(&zones->ztree, lzdel, NULL); 145 free(zones); 146 } 147 148 void 149 local_zone_delete(struct local_zone* z) 150 { 151 if(!z) 152 return; 153 lock_rw_destroy(&z->lock); 154 regional_destroy(z->region); 155 free(z->name); 156 free(z->taglist); 157 free(z); 158 } 159 160 int 161 local_zone_cmp(const void* z1, const void* z2) 162 { 163 /* first sort on class, so that hierarchy can be maintained within 164 * a class */ 165 struct local_zone* a = (struct local_zone*)z1; 166 struct local_zone* b = (struct local_zone*)z2; 167 int m; 168 if(a->dclass != b->dclass) { 169 if(a->dclass < b->dclass) 170 return -1; 171 return 1; 172 } 173 return dname_lab_cmp(a->name, a->namelabs, b->name, b->namelabs, &m); 174 } 175 176 int 177 local_data_cmp(const void* d1, const void* d2) 178 { 179 struct local_data* a = (struct local_data*)d1; 180 struct local_data* b = (struct local_data*)d2; 181 int m; 182 return dname_canon_lab_cmp(a->name, a->namelabs, b->name, 183 b->namelabs, &m); 184 } 185 186 /* form wireformat from text format domain name */ 187 int 188 parse_dname(const char* str, uint8_t** res, size_t* len, int* labs) 189 { 190 *res = sldns_str2wire_dname(str, len); 191 *labs = 0; 192 if(!*res) { 193 log_err("cannot parse name %s", str); 194 return 0; 195 } 196 *labs = dname_count_size_labels(*res, len); 197 return 1; 198 } 199 200 /** create a new localzone */ 201 static struct local_zone* 202 local_zone_create(uint8_t* nm, size_t len, int labs, 203 enum localzone_type t, uint16_t dclass) 204 { 205 struct local_zone* z = (struct local_zone*)calloc(1, sizeof(*z)); 206 if(!z) { 207 return NULL; 208 } 209 z->node.key = z; 210 z->dclass = dclass; 211 z->type = t; 212 z->name = nm; 213 z->namelen = len; 214 z->namelabs = labs; 215 lock_rw_init(&z->lock); 216 z->region = regional_create_nochunk(sizeof(struct regional)); 217 if(!z->region) { 218 free(z); 219 return NULL; 220 } 221 rbtree_init(&z->data, &local_data_cmp); 222 lock_protect(&z->lock, &z->parent, sizeof(*z)-sizeof(rbnode_type)); 223 /* also the zones->lock protects node, parent, name*, class */ 224 return z; 225 } 226 227 /** enter a new zone with allocated dname returns with WRlock */ 228 static struct local_zone* 229 lz_enter_zone_dname(struct local_zones* zones, uint8_t* nm, size_t len, 230 int labs, enum localzone_type t, uint16_t c) 231 { 232 struct local_zone* z = local_zone_create(nm, len, labs, t, c); 233 if(!z) { 234 free(nm); 235 log_err("out of memory"); 236 return NULL; 237 } 238 239 /* add to rbtree */ 240 lock_rw_wrlock(&zones->lock); 241 lock_rw_wrlock(&z->lock); 242 if(!rbtree_insert(&zones->ztree, &z->node)) { 243 struct local_zone* oldz; 244 char str[LDNS_MAX_DOMAINLEN]; 245 dname_str(nm, str); 246 log_warn("duplicate local-zone %s", str); 247 lock_rw_unlock(&z->lock); 248 /* save zone name locally before deallocation, 249 * otherwise, nm is gone if we zone_delete now. */ 250 oldz = z; 251 /* find the correct zone, so not an error for duplicate */ 252 z = local_zones_find(zones, nm, len, labs, c); 253 lock_rw_wrlock(&z->lock); 254 lock_rw_unlock(&zones->lock); 255 local_zone_delete(oldz); 256 return z; 257 } 258 lock_rw_unlock(&zones->lock); 259 return z; 260 } 261 262 /** enter a new zone */ 263 struct local_zone* 264 lz_enter_zone(struct local_zones* zones, const char* name, const char* type, 265 uint16_t dclass) 266 { 267 struct local_zone* z; 268 enum localzone_type t; 269 uint8_t* nm; 270 size_t len; 271 int labs; 272 if(!parse_dname(name, &nm, &len, &labs)) { 273 log_err("bad zone name %s %s", name, type); 274 return NULL; 275 } 276 if(!local_zone_str2type(type, &t)) { 277 log_err("bad lz_enter_zone type %s %s", name, type); 278 free(nm); 279 return NULL; 280 } 281 if(!(z=lz_enter_zone_dname(zones, nm, len, labs, t, dclass))) { 282 log_err("could not enter zone %s %s", name, type); 283 return NULL; 284 } 285 return z; 286 } 287 288 int 289 rrstr_get_rr_content(const char* str, uint8_t** nm, uint16_t* type, 290 uint16_t* dclass, time_t* ttl, uint8_t* rr, size_t len, 291 uint8_t** rdata, size_t* rdata_len) 292 { 293 size_t dname_len = 0; 294 int e = sldns_str2wire_rr_buf(str, rr, &len, &dname_len, 3600, 295 NULL, 0, NULL, 0); 296 if(e) { 297 log_err("error parsing local-data at %d: '%s': %s", 298 LDNS_WIREPARSE_OFFSET(e), str, 299 sldns_get_errorstr_parse(e)); 300 return 0; 301 } 302 *nm = memdup(rr, dname_len); 303 if(!*nm) { 304 log_err("out of memory"); 305 return 0; 306 } 307 *dclass = sldns_wirerr_get_class(rr, len, dname_len); 308 *type = sldns_wirerr_get_type(rr, len, dname_len); 309 *ttl = (time_t)sldns_wirerr_get_ttl(rr, len, dname_len); 310 *rdata = sldns_wirerr_get_rdatawl(rr, len, dname_len); 311 *rdata_len = sldns_wirerr_get_rdatalen(rr, len, dname_len)+2; 312 return 1; 313 } 314 315 /** return name and class of rr; parses string */ 316 static int 317 get_rr_nameclass(const char* str, uint8_t** nm, uint16_t* dclass, 318 uint16_t* dtype) 319 { 320 uint8_t rr[LDNS_RR_BUF_SIZE]; 321 size_t len = sizeof(rr), dname_len = 0; 322 int s = sldns_str2wire_rr_buf(str, rr, &len, &dname_len, 3600, 323 NULL, 0, NULL, 0); 324 if(s != 0) { 325 log_err("error parsing local-data at %d '%s': %s", 326 LDNS_WIREPARSE_OFFSET(s), str, 327 sldns_get_errorstr_parse(s)); 328 return 0; 329 } 330 *nm = memdup(rr, dname_len); 331 *dclass = sldns_wirerr_get_class(rr, len, dname_len); 332 *dtype = sldns_wirerr_get_type(rr, len, dname_len); 333 if(!*nm) { 334 log_err("out of memory"); 335 return 0; 336 } 337 return 1; 338 } 339 340 /** 341 * Find an rrset in local data structure. 342 * @param data: local data domain name structure. 343 * @param type: type to look for (host order). 344 * @param alias_ok: 1 if matching a non-exact, alias type such as CNAME is 345 * allowed. otherwise 0. 346 * @return rrset pointer or NULL if not found. 347 */ 348 static struct local_rrset* 349 local_data_find_type(struct local_data* data, uint16_t type, int alias_ok) 350 { 351 struct local_rrset* p, *cname = NULL; 352 type = htons(type); 353 for(p = data->rrsets; p; p = p->next) { 354 if(p->rrset->rk.type == type) 355 return p; 356 if(alias_ok && p->rrset->rk.type == htons(LDNS_RR_TYPE_CNAME)) 357 cname = p; 358 } 359 if(alias_ok) 360 return cname; 361 return NULL; 362 } 363 364 /** check for RR duplicates */ 365 static int 366 rr_is_duplicate(struct packed_rrset_data* pd, uint8_t* rdata, size_t rdata_len) 367 { 368 size_t i; 369 for(i=0; i<pd->count; i++) { 370 if(pd->rr_len[i] == rdata_len && 371 memcmp(pd->rr_data[i], rdata, rdata_len) == 0) 372 return 1; 373 } 374 return 0; 375 } 376 377 /** new local_rrset */ 378 static struct local_rrset* 379 new_local_rrset(struct regional* region, struct local_data* node, 380 uint16_t rrtype, uint16_t rrclass) 381 { 382 struct packed_rrset_data* pd; 383 struct local_rrset* rrset = (struct local_rrset*) 384 regional_alloc_zero(region, sizeof(*rrset)); 385 if(!rrset) { 386 log_err("out of memory"); 387 return NULL; 388 } 389 rrset->rrset = (struct ub_packed_rrset_key*) 390 regional_alloc_zero(region, sizeof(*rrset->rrset)); 391 if(!rrset->rrset) { 392 log_err("out of memory"); 393 return NULL; 394 } 395 rrset->rrset->entry.key = rrset->rrset; 396 pd = (struct packed_rrset_data*)regional_alloc_zero(region, 397 sizeof(*pd)); 398 if(!pd) { 399 log_err("out of memory"); 400 return NULL; 401 } 402 pd->trust = rrset_trust_prim_noglue; 403 pd->security = sec_status_insecure; 404 rrset->rrset->entry.data = pd; 405 rrset->rrset->rk.dname = node->name; 406 rrset->rrset->rk.dname_len = node->namelen; 407 rrset->rrset->rk.type = htons(rrtype); 408 rrset->rrset->rk.rrset_class = htons(rrclass); 409 rrset->next = node->rrsets; 410 node->rrsets = rrset; 411 return rrset; 412 } 413 414 /** insert RR into RRset data structure; Wastes a couple of bytes */ 415 int 416 rrset_insert_rr(struct regional* region, struct packed_rrset_data* pd, 417 uint8_t* rdata, size_t rdata_len, time_t ttl, const char* rrstr) 418 { 419 size_t* oldlen = pd->rr_len; 420 time_t* oldttl = pd->rr_ttl; 421 uint8_t** olddata = pd->rr_data; 422 423 /* add RR to rrset */ 424 if(pd->count > LOCALZONE_RRSET_COUNT_MAX) { 425 log_warn("RRset '%s' has more than %d records, record ignored", 426 rrstr, LOCALZONE_RRSET_COUNT_MAX); 427 return 1; 428 } 429 pd->count++; 430 pd->rr_len = regional_alloc(region, sizeof(*pd->rr_len)*pd->count); 431 pd->rr_ttl = regional_alloc(region, sizeof(*pd->rr_ttl)*pd->count); 432 pd->rr_data = regional_alloc(region, sizeof(*pd->rr_data)*pd->count); 433 if(!pd->rr_len || !pd->rr_ttl || !pd->rr_data) { 434 pd->count--; 435 pd->rr_len = oldlen; 436 pd->rr_ttl = oldttl; 437 pd->rr_data = olddata; 438 log_err("out of memory"); 439 return 0; 440 } 441 if(pd->count > 1) { 442 memcpy(pd->rr_len+1, oldlen, 443 sizeof(*pd->rr_len)*(pd->count-1)); 444 memcpy(pd->rr_ttl+1, oldttl, 445 sizeof(*pd->rr_ttl)*(pd->count-1)); 446 memcpy(pd->rr_data+1, olddata, 447 sizeof(*pd->rr_data)*(pd->count-1)); 448 } 449 pd->rr_len[0] = rdata_len; 450 pd->rr_ttl[0] = ttl; 451 pd->rr_data[0] = regional_alloc_init(region, rdata, rdata_len); 452 if(!pd->rr_data[0]) { 453 pd->count--; 454 pd->rr_len = oldlen; 455 pd->rr_ttl = oldttl; 456 pd->rr_data = olddata; 457 log_err("out of memory"); 458 return 0; 459 } 460 return 1; 461 } 462 463 /** Delete RR from local-zone RRset, wastes memory as the deleted RRs cannot be 464 * free'd (regionally alloc'd) */ 465 int 466 local_rrset_remove_rr(struct packed_rrset_data* pd, size_t index) 467 { 468 log_assert(pd->count > 0); 469 if(index >= pd->count) { 470 log_warn("Trying to remove RR with out of bound index"); 471 return 0; 472 } 473 if(index + 1 < pd->count) { 474 /* not removing last element */ 475 size_t nexti = index + 1; 476 size_t num = pd->count - nexti; 477 memmove(pd->rr_len+index, pd->rr_len+nexti, sizeof(*pd->rr_len)*num); 478 memmove(pd->rr_ttl+index, pd->rr_ttl+nexti, sizeof(*pd->rr_ttl)*num); 479 memmove(pd->rr_data+index, pd->rr_data+nexti, sizeof(*pd->rr_data)*num); 480 } 481 pd->count--; 482 return 1; 483 } 484 485 struct local_data* 486 local_zone_find_data(struct local_zone* z, uint8_t* nm, size_t nmlen, int nmlabs) 487 { 488 struct local_data key; 489 key.node.key = &key; 490 key.name = nm; 491 key.namelen = nmlen; 492 key.namelabs = nmlabs; 493 return (struct local_data*)rbtree_search(&z->data, &key.node); 494 } 495 496 /** find a node, create it if not and all its empty nonterminal parents */ 497 static int 498 lz_find_create_node(struct local_zone* z, uint8_t* nm, size_t nmlen, 499 int nmlabs, struct local_data** res) 500 { 501 struct local_data* ld = local_zone_find_data(z, nm, nmlen, nmlabs); 502 if(!ld) { 503 /* create a domain name to store rr. */ 504 ld = (struct local_data*)regional_alloc_zero(z->region, 505 sizeof(*ld)); 506 if(!ld) { 507 log_err("out of memory adding local data"); 508 return 0; 509 } 510 ld->node.key = ld; 511 ld->name = regional_alloc_init(z->region, nm, nmlen); 512 if(!ld->name) { 513 log_err("out of memory"); 514 return 0; 515 } 516 ld->namelen = nmlen; 517 ld->namelabs = nmlabs; 518 if(!rbtree_insert(&z->data, &ld->node)) { 519 log_assert(0); /* duplicate name */ 520 } 521 /* see if empty nonterminals need to be created */ 522 if(nmlabs > z->namelabs) { 523 dname_remove_label(&nm, &nmlen); 524 if(!lz_find_create_node(z, nm, nmlen, nmlabs-1, res)) 525 return 0; 526 } 527 } 528 *res = ld; 529 return 1; 530 } 531 532 /* Mark the SOA record for the zone. This only marks the SOA rrset; the data 533 * for the RR is entered later on local_zone_enter_rr() as with the other 534 * records. An artificial soa_negative record with a modified TTL (minimum of 535 * the TTL and the SOA.MINIMUM) is also created and marked for usage with 536 * negative answers and to avoid allocations during those answers. */ 537 static int 538 lz_mark_soa_for_zone(struct local_zone* z, struct ub_packed_rrset_key* soa_rrset, 539 uint8_t* rdata, size_t rdata_len, time_t ttl, const char* rrstr) 540 { 541 struct packed_rrset_data* pd = (struct packed_rrset_data*) 542 regional_alloc_zero(z->region, sizeof(*pd)); 543 struct ub_packed_rrset_key* rrset_negative = (struct ub_packed_rrset_key*) 544 regional_alloc_zero(z->region, sizeof(*rrset_negative)); 545 time_t minimum; 546 if(!rrset_negative||!pd) { 547 log_err("out of memory"); 548 return 0; 549 } 550 /* Mark the original SOA record and then continue with the negative one. */ 551 z->soa = soa_rrset; 552 rrset_negative->entry.key = rrset_negative; 553 pd->trust = rrset_trust_prim_noglue; 554 pd->security = sec_status_insecure; 555 rrset_negative->entry.data = pd; 556 rrset_negative->rk.dname = soa_rrset->rk.dname; 557 rrset_negative->rk.dname_len = soa_rrset->rk.dname_len; 558 rrset_negative->rk.type = soa_rrset->rk.type; 559 rrset_negative->rk.rrset_class = soa_rrset->rk.rrset_class; 560 if(!rrset_insert_rr(z->region, pd, rdata, rdata_len, ttl, rrstr)) 561 return 0; 562 /* last 4 bytes are minimum ttl in network format */ 563 if(pd->count == 0 || pd->rr_len[0] < 2+4) 564 return 0; 565 minimum = (time_t)sldns_read_uint32(pd->rr_data[0]+(pd->rr_len[0]-4)); 566 minimum = ttl<minimum?ttl:minimum; 567 pd->ttl = minimum; 568 pd->rr_ttl[0] = minimum; 569 570 z->soa_negative = rrset_negative; 571 return 1; 572 } 573 574 int 575 local_zone_enter_rr(struct local_zone* z, uint8_t* nm, size_t nmlen, 576 int nmlabs, uint16_t rrtype, uint16_t rrclass, time_t ttl, 577 uint8_t* rdata, size_t rdata_len, const char* rrstr) 578 { 579 struct local_data* node; 580 struct local_rrset* rrset; 581 struct packed_rrset_data* pd; 582 583 if(!lz_find_create_node(z, nm, nmlen, nmlabs, &node)) { 584 return 0; 585 } 586 log_assert(node); 587 588 /* Reject it if we would end up having CNAME and other data (including 589 * another CNAME) for a redirect zone. */ 590 if((z->type == local_zone_redirect || 591 z->type == local_zone_inform_redirect) && node->rrsets) { 592 const char* othertype = NULL; 593 if (rrtype == LDNS_RR_TYPE_CNAME) 594 othertype = "other"; 595 else if (node->rrsets->rrset->rk.type == 596 htons(LDNS_RR_TYPE_CNAME)) { 597 othertype = "CNAME"; 598 } 599 if(othertype) { 600 log_err("local-data '%s' in redirect zone must not " 601 "coexist with %s local-data", rrstr, othertype); 602 return 0; 603 } 604 } 605 rrset = local_data_find_type(node, rrtype, 0); 606 if(!rrset) { 607 rrset = new_local_rrset(z->region, node, rrtype, rrclass); 608 if(!rrset) 609 return 0; 610 if(query_dname_compare(node->name, z->name) == 0) { 611 if(rrtype == LDNS_RR_TYPE_NSEC) 612 rrset->rrset->rk.flags = PACKED_RRSET_NSEC_AT_APEX; 613 if(rrtype == LDNS_RR_TYPE_SOA && 614 !lz_mark_soa_for_zone(z, rrset->rrset, rdata, rdata_len, ttl, 615 rrstr)) 616 return 0; 617 } 618 } 619 pd = (struct packed_rrset_data*)rrset->rrset->entry.data; 620 log_assert(rrset && pd); 621 622 /* check for duplicate RR */ 623 if(rr_is_duplicate(pd, rdata, rdata_len)) { 624 verbose(VERB_ALGO, "ignoring duplicate RR: %s", rrstr); 625 return 1; 626 } 627 return rrset_insert_rr(z->region, pd, rdata, rdata_len, ttl, rrstr); 628 } 629 630 /** enter data RR into auth zone */ 631 static int 632 lz_enter_rr_into_zone(struct local_zone* z, const char* rrstr) 633 { 634 uint8_t* nm; 635 size_t nmlen; 636 int nmlabs, ret; 637 uint16_t rrtype = 0, rrclass = 0; 638 time_t ttl = 0; 639 uint8_t rr[LDNS_RR_BUF_SIZE]; 640 uint8_t* rdata; 641 size_t rdata_len; 642 if(!rrstr_get_rr_content(rrstr, &nm, &rrtype, &rrclass, &ttl, rr, 643 sizeof(rr), &rdata, &rdata_len)) { 644 log_err("bad local-data: %s", rrstr); 645 return 0; 646 } 647 log_assert(z->dclass == rrclass); 648 if((z->type == local_zone_redirect || 649 z->type == local_zone_inform_redirect) && 650 query_dname_compare(z->name, nm) != 0) { 651 log_err("local-data in redirect zone must reside at top of zone" 652 ", not at %s", rrstr); 653 free(nm); 654 return 0; 655 } 656 nmlabs = dname_count_size_labels(nm, &nmlen); 657 ret = local_zone_enter_rr(z, nm, nmlen, nmlabs, rrtype, rrclass, ttl, 658 rdata, rdata_len, rrstr); 659 free(nm); 660 return ret; 661 } 662 663 /** enter a data RR into auth data; a zone for it must exist */ 664 static int 665 lz_enter_rr_str(struct local_zones* zones, const char* rr) 666 { 667 uint8_t* rr_name; 668 uint16_t rr_class, rr_type; 669 size_t len; 670 int labs; 671 struct local_zone* z; 672 int r; 673 if(!get_rr_nameclass(rr, &rr_name, &rr_class, &rr_type)) { 674 log_err("bad rr %s", rr); 675 return 0; 676 } 677 labs = dname_count_size_labels(rr_name, &len); 678 lock_rw_rdlock(&zones->lock); 679 z = local_zones_lookup(zones, rr_name, len, labs, rr_class, rr_type, 1); 680 if(!z) { 681 lock_rw_unlock(&zones->lock); 682 log_err("internal error: no zone for rr %s", rr); 683 free(rr_name); 684 return 0; 685 } 686 lock_rw_wrlock(&z->lock); 687 lock_rw_unlock(&zones->lock); 688 free(rr_name); 689 r = lz_enter_rr_into_zone(z, rr); 690 lock_rw_unlock(&z->lock); 691 return r; 692 } 693 694 /** enter tagstring into zone */ 695 static int 696 lz_enter_zone_tag(struct local_zones* zones, char* zname, uint8_t* list, 697 size_t len, uint16_t rr_class) 698 { 699 uint8_t dname[LDNS_MAX_DOMAINLEN+1]; 700 size_t dname_len = sizeof(dname); 701 int dname_labs, r = 0; 702 struct local_zone* z; 703 704 if(sldns_str2wire_dname_buf(zname, dname, &dname_len) != 0) { 705 log_err("cannot parse zone name in local-zone-tag: %s", zname); 706 return 0; 707 } 708 dname_labs = dname_count_labels(dname); 709 710 lock_rw_rdlock(&zones->lock); 711 z = local_zones_find(zones, dname, dname_len, dname_labs, rr_class); 712 if(!z) { 713 lock_rw_unlock(&zones->lock); 714 log_err("no local-zone for tag %s", zname); 715 return 0; 716 } 717 lock_rw_wrlock(&z->lock); 718 lock_rw_unlock(&zones->lock); 719 free(z->taglist); 720 z->taglist = memdup(list, len); 721 z->taglen = len; 722 if(z->taglist) 723 r = 1; 724 lock_rw_unlock(&z->lock); 725 return r; 726 } 727 728 /** enter override into zone */ 729 static int 730 lz_enter_override(struct local_zones* zones, char* zname, char* netblock, 731 char* type, uint16_t rr_class) 732 { 733 uint8_t dname[LDNS_MAX_DOMAINLEN+1]; 734 size_t dname_len = sizeof(dname); 735 int dname_labs; 736 struct sockaddr_storage addr; 737 int net; 738 socklen_t addrlen; 739 struct local_zone* z; 740 enum localzone_type t; 741 742 /* parse zone name */ 743 if(sldns_str2wire_dname_buf(zname, dname, &dname_len) != 0) { 744 log_err("cannot parse zone name in local-zone-override: %s %s", 745 zname, netblock); 746 return 0; 747 } 748 dname_labs = dname_count_labels(dname); 749 750 /* parse netblock */ 751 if(!netblockstrtoaddr(netblock, UNBOUND_DNS_PORT, &addr, &addrlen, 752 &net)) { 753 log_err("cannot parse netblock in local-zone-override: %s %s", 754 zname, netblock); 755 return 0; 756 } 757 758 /* parse zone type */ 759 if(!local_zone_str2type(type, &t)) { 760 log_err("cannot parse type in local-zone-override: %s %s %s", 761 zname, netblock, type); 762 return 0; 763 } 764 765 /* find localzone entry */ 766 lock_rw_rdlock(&zones->lock); 767 z = local_zones_find(zones, dname, dname_len, dname_labs, rr_class); 768 if(!z) { 769 lock_rw_unlock(&zones->lock); 770 log_err("no local-zone for local-zone-override %s", zname); 771 return 0; 772 } 773 lock_rw_wrlock(&z->lock); 774 lock_rw_unlock(&zones->lock); 775 776 /* create netblock addr_tree if not present yet */ 777 if(!z->override_tree) { 778 z->override_tree = (struct rbtree_type*)regional_alloc_zero( 779 z->region, sizeof(*z->override_tree)); 780 if(!z->override_tree) { 781 lock_rw_unlock(&z->lock); 782 log_err("out of memory"); 783 return 0; 784 } 785 addr_tree_init(z->override_tree); 786 } 787 /* add new elem to tree */ 788 if(z->override_tree) { 789 struct local_zone_override* n; 790 n = (struct local_zone_override*)regional_alloc_zero( 791 z->region, sizeof(*n)); 792 if(!n) { 793 lock_rw_unlock(&z->lock); 794 log_err("out of memory"); 795 return 0; 796 } 797 n->type = t; 798 if(!addr_tree_insert(z->override_tree, 799 (struct addr_tree_node*)n, &addr, addrlen, net)) { 800 lock_rw_unlock(&z->lock); 801 log_err("duplicate local-zone-override %s %s", 802 zname, netblock); 803 return 1; 804 } 805 } 806 807 lock_rw_unlock(&z->lock); 808 return 1; 809 } 810 811 /** parse local-zone: statements */ 812 static int 813 lz_enter_zones(struct local_zones* zones, struct config_file* cfg) 814 { 815 struct config_str2list* p; 816 #ifndef THREADS_DISABLED 817 struct local_zone* z; 818 #endif 819 for(p = cfg->local_zones; p; p = p->next) { 820 if(!( 821 #ifndef THREADS_DISABLED 822 z= 823 #endif 824 lz_enter_zone(zones, p->str, p->str2, 825 LDNS_RR_CLASS_IN))) 826 return 0; 827 lock_rw_unlock(&z->lock); 828 } 829 return 1; 830 } 831 832 /** lookup a zone in rbtree; exact match only; SLOW due to parse */ 833 static int 834 lz_exists(struct local_zones* zones, const char* name) 835 { 836 struct local_zone z; 837 z.node.key = &z; 838 z.dclass = LDNS_RR_CLASS_IN; 839 if(!parse_dname(name, &z.name, &z.namelen, &z.namelabs)) { 840 log_err("bad name %s", name); 841 return 0; 842 } 843 lock_rw_rdlock(&zones->lock); 844 if(rbtree_search(&zones->ztree, &z.node)) { 845 lock_rw_unlock(&zones->lock); 846 free(z.name); 847 return 1; 848 } 849 lock_rw_unlock(&zones->lock); 850 free(z.name); 851 return 0; 852 } 853 854 /** lookup a zone in cfg->nodefault list */ 855 static int 856 lz_nodefault(struct config_file* cfg, const char* name) 857 { 858 struct config_strlist* p; 859 size_t len = strlen(name); 860 if(len == 0) return 0; 861 if(name[len-1] == '.') len--; 862 863 for(p = cfg->local_zones_nodefault; p; p = p->next) { 864 /* compare zone name, lowercase, compare without ending . */ 865 if(strncasecmp(p->str, name, len) == 0 && 866 (strlen(p->str) == len || (strlen(p->str)==len+1 && 867 p->str[len] == '.'))) 868 return 1; 869 } 870 return 0; 871 } 872 873 /** enter reverse default zone */ 874 static int 875 add_reverse_default(struct local_zones* zones, struct config_file* cfg, 876 const char* name) 877 { 878 struct local_zone* z; 879 char str[1024]; /* known long enough */ 880 if(lz_exists(zones, name) || lz_nodefault(cfg, name)) 881 return 1; /* do not enter default content */ 882 if(!(z=lz_enter_zone(zones, name, "static", LDNS_RR_CLASS_IN))) 883 return 0; 884 snprintf(str, sizeof(str), "%s 10800 IN SOA localhost. " 885 "nobody.invalid. 1 3600 1200 604800 10800", name); 886 if(!lz_enter_rr_into_zone(z, str)) { 887 lock_rw_unlock(&z->lock); 888 return 0; 889 } 890 snprintf(str, sizeof(str), "%s 10800 IN NS localhost. ", name); 891 if(!lz_enter_rr_into_zone(z, str)) { 892 lock_rw_unlock(&z->lock); 893 return 0; 894 } 895 if(strncasecmp("127.in-addr.arpa.", name, 17) == 0) { 896 if(!lz_enter_rr_into_zone(z, 897 "1.0.0.127.in-addr.arpa. 10800 IN PTR localhost.")) { 898 lock_rw_unlock(&z->lock); 899 return 0; 900 } 901 } else if(strncasecmp("1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.", name, 73) == 0) { 902 snprintf(str, sizeof(str), "%s 10800 IN PTR localhost.", name); 903 if(!lz_enter_rr_into_zone(z, str)) { 904 lock_rw_unlock(&z->lock); 905 return 0; 906 } 907 } 908 lock_rw_unlock(&z->lock); 909 return 1; 910 } 911 912 /** enter (AS112) empty default zone */ 913 static int 914 add_empty_default(struct local_zones* zones, struct config_file* cfg, 915 const char* name) 916 { 917 struct local_zone* z; 918 char str[1024]; /* known long enough */ 919 if(lz_exists(zones, name) || lz_nodefault(cfg, name)) 920 return 1; /* do not enter default content */ 921 if(!(z=lz_enter_zone(zones, name, "static", LDNS_RR_CLASS_IN))) 922 return 0; 923 snprintf(str, sizeof(str), "%s 10800 IN SOA localhost. " 924 "nobody.invalid. 1 3600 1200 604800 10800", name); 925 if(!lz_enter_rr_into_zone(z, str)) { 926 lock_rw_unlock(&z->lock); 927 return 0; 928 } 929 snprintf(str, sizeof(str), "%s 10800 IN NS localhost. ", name); 930 if(!lz_enter_rr_into_zone(z, str)) { 931 lock_rw_unlock(&z->lock); 932 return 0; 933 } 934 lock_rw_unlock(&z->lock); 935 return 1; 936 } 937 938 /** enter default zones */ 939 int local_zone_enter_defaults(struct local_zones* zones, struct config_file* cfg) 940 { 941 struct local_zone* z; 942 const char** zstr; 943 944 /* Do not add any default */ 945 if(cfg->local_zones_disable_default) 946 return 1; 947 948 /* this list of zones is from RFC 6303 and RFC 7686 */ 949 950 /* block localhost level zones first, then onion and later the LAN zones */ 951 952 /* localhost. zone */ 953 if(!lz_exists(zones, "localhost.") && 954 !lz_nodefault(cfg, "localhost.")) { 955 if(!(z=lz_enter_zone(zones, "localhost.", "redirect", 956 LDNS_RR_CLASS_IN)) || 957 !lz_enter_rr_into_zone(z, 958 "localhost. 10800 IN NS localhost.") || 959 !lz_enter_rr_into_zone(z, 960 "localhost. 10800 IN SOA localhost. nobody.invalid. " 961 "1 3600 1200 604800 10800") || 962 !lz_enter_rr_into_zone(z, 963 "localhost. 10800 IN A 127.0.0.1") || 964 !lz_enter_rr_into_zone(z, 965 "localhost. 10800 IN AAAA ::1")) { 966 log_err("out of memory adding default zone"); 967 if(z) { lock_rw_unlock(&z->lock); } 968 return 0; 969 } 970 lock_rw_unlock(&z->lock); 971 } 972 973 /* ip4 and ip6 reverse */ 974 for(zstr = local_zones_default_reverse; *zstr; zstr++) { 975 if(!add_reverse_default(zones, cfg, *zstr)) { 976 log_err("out of memory adding default zone"); 977 return 0; 978 } 979 } 980 981 /* special-use zones */ 982 for(zstr = local_zones_default_special; *zstr; zstr++) { 983 if(!add_empty_default(zones, cfg, *zstr)) { 984 log_err("out of memory adding default zone"); 985 return 0; 986 } 987 } 988 989 /* block AS112 zones, unless asked not to */ 990 if(!cfg->unblock_lan_zones) { 991 for(zstr = as112_zones; *zstr; zstr++) { 992 if(!add_empty_default(zones, cfg, *zstr)) { 993 log_err("out of memory adding default zone"); 994 return 0; 995 } 996 } 997 } 998 return 1; 999 } 1000 1001 /** parse local-zone-override: statements */ 1002 static int 1003 lz_enter_overrides(struct local_zones* zones, struct config_file* cfg) 1004 { 1005 struct config_str3list* p; 1006 for(p = cfg->local_zone_overrides; p; p = p->next) { 1007 if(!lz_enter_override(zones, p->str, p->str2, p->str3, 1008 LDNS_RR_CLASS_IN)) 1009 return 0; 1010 } 1011 return 1; 1012 } 1013 1014 /* return closest parent in the tree, NULL if none */ 1015 static struct local_zone* find_closest_parent(struct local_zone* curr, 1016 struct local_zone* prev) 1017 { 1018 struct local_zone* p; 1019 int m; 1020 if(!prev || prev->dclass != curr->dclass) return NULL; 1021 (void)dname_lab_cmp(prev->name, prev->namelabs, curr->name, 1022 curr->namelabs, &m); /* we know prev is smaller */ 1023 /* sort order like: . com. bla.com. zwb.com. net. */ 1024 /* find the previous, or parent-parent-parent */ 1025 for(p = prev; p; p = p->parent) { 1026 /* looking for name with few labels, a parent */ 1027 if(p->namelabs <= m) { 1028 /* ==: since prev matched m, this is closest*/ 1029 /* <: prev matches more, but is not a parent, 1030 * this one is a (grand)parent */ 1031 return p; 1032 } 1033 } 1034 return NULL; 1035 } 1036 1037 /** setup parent pointers, so that a lookup can be done for closest match */ 1038 void 1039 lz_init_parents(struct local_zones* zones) 1040 { 1041 struct local_zone* node, *prev = NULL; 1042 lock_rw_wrlock(&zones->lock); 1043 RBTREE_FOR(node, struct local_zone*, &zones->ztree) { 1044 lock_rw_wrlock(&node->lock); 1045 node->parent = find_closest_parent(node, prev); 1046 prev = node; 1047 if(node->override_tree) 1048 addr_tree_init_parents(node->override_tree); 1049 lock_rw_unlock(&node->lock); 1050 } 1051 lock_rw_unlock(&zones->lock); 1052 } 1053 1054 /** enter implicit transparent zone for local-data: without local-zone: */ 1055 static int 1056 lz_setup_implicit(struct local_zones* zones, struct config_file* cfg) 1057 { 1058 /* walk over all items that have no parent zone and find 1059 * the name that covers them all (could be the root) and 1060 * add that as a transparent zone */ 1061 struct config_strlist* p; 1062 int have_name = 0; 1063 int have_other_classes = 0; 1064 uint16_t dclass = 0; 1065 uint8_t* nm = 0; 1066 size_t nmlen = 0; 1067 int nmlabs = 0; 1068 int match = 0; /* number of labels match count */ 1069 1070 lz_init_parents(zones); /* to enable local_zones_lookup() */ 1071 for(p = cfg->local_data; p; p = p->next) { 1072 uint8_t* rr_name; 1073 uint16_t rr_class, rr_type; 1074 size_t len; 1075 int labs; 1076 if(!get_rr_nameclass(p->str, &rr_name, &rr_class, &rr_type)) { 1077 log_err("Bad local-data RR %s", p->str); 1078 return 0; 1079 } 1080 labs = dname_count_size_labels(rr_name, &len); 1081 lock_rw_rdlock(&zones->lock); 1082 if(!local_zones_lookup(zones, rr_name, len, labs, rr_class, 1083 rr_type, 1)) { 1084 /* Check if there is a zone that this could go 1085 * under but for different class; created zones are 1086 * always for LDNS_RR_CLASS_IN. Create the zone with 1087 * a different class but the same configured 1088 * local_zone_type. */ 1089 struct local_zone* z = local_zones_lookup(zones, 1090 rr_name, len, labs, LDNS_RR_CLASS_IN, rr_type, 1091 1); 1092 if(z) { 1093 uint8_t* name = memdup(z->name, z->namelen); 1094 size_t znamelen = z->namelen; 1095 int znamelabs = z->namelabs; 1096 enum localzone_type ztype = z->type; 1097 lock_rw_unlock(&zones->lock); 1098 if(!name) { 1099 log_err("out of memory"); 1100 free(rr_name); 1101 return 0; 1102 } 1103 if(!( 1104 #ifndef THREADS_DISABLED 1105 z = 1106 #endif 1107 lz_enter_zone_dname(zones, name, 1108 znamelen, znamelabs, 1109 ztype, rr_class))) { 1110 free(rr_name); 1111 return 0; 1112 } 1113 lock_rw_unlock(&z->lock); 1114 free(rr_name); 1115 continue; 1116 } 1117 if(!have_name) { 1118 dclass = rr_class; 1119 nm = rr_name; 1120 nmlen = len; 1121 nmlabs = labs; 1122 match = labs; 1123 have_name = 1; 1124 } else { 1125 int m; 1126 if(rr_class != dclass) { 1127 /* process other classes later */ 1128 free(rr_name); 1129 have_other_classes = 1; 1130 lock_rw_unlock(&zones->lock); 1131 continue; 1132 } 1133 /* find smallest shared topdomain */ 1134 (void)dname_lab_cmp(nm, nmlabs, 1135 rr_name, labs, &m); 1136 free(rr_name); 1137 if(m < match) 1138 match = m; 1139 } 1140 } else free(rr_name); 1141 lock_rw_unlock(&zones->lock); 1142 } 1143 if(have_name) { 1144 uint8_t* n2; 1145 #ifndef THREADS_DISABLED 1146 struct local_zone* z; 1147 #endif 1148 /* allocate zone of smallest shared topdomain to contain em */ 1149 n2 = nm; 1150 dname_remove_labels(&n2, &nmlen, nmlabs - match); 1151 n2 = memdup(n2, nmlen); 1152 free(nm); 1153 if(!n2) { 1154 log_err("out of memory"); 1155 return 0; 1156 } 1157 log_nametypeclass(VERB_ALGO, "implicit transparent local-zone", 1158 n2, 0, dclass); 1159 if(!( 1160 #ifndef THREADS_DISABLED 1161 z= 1162 #endif 1163 lz_enter_zone_dname(zones, n2, nmlen, match, 1164 local_zone_transparent, dclass))) { 1165 return 0; 1166 } 1167 lock_rw_unlock(&z->lock); 1168 } 1169 if(have_other_classes) { 1170 /* restart to setup other class */ 1171 return lz_setup_implicit(zones, cfg); 1172 } 1173 return 1; 1174 } 1175 1176 /** enter local-zone-tag info */ 1177 static int 1178 lz_enter_zone_tags(struct local_zones* zones, struct config_file* cfg) 1179 { 1180 struct config_strbytelist* p; 1181 int c = 0; 1182 for(p = cfg->local_zone_tags; p; p = p->next) { 1183 if(!lz_enter_zone_tag(zones, p->str, p->str2, p->str2len, 1184 LDNS_RR_CLASS_IN)) 1185 return 0; 1186 c++; 1187 } 1188 if(c) verbose(VERB_ALGO, "applied tags to %d local zones", c); 1189 return 1; 1190 } 1191 1192 /** enter auth data */ 1193 static int 1194 lz_enter_data(struct local_zones* zones, struct config_file* cfg) 1195 { 1196 struct config_strlist* p; 1197 for(p = cfg->local_data; p; p = p->next) { 1198 if(!lz_enter_rr_str(zones, p->str)) 1199 return 0; 1200 } 1201 return 1; 1202 } 1203 1204 /** free memory from config */ 1205 static void 1206 lz_freeup_cfg(struct config_file* cfg) 1207 { 1208 config_deldblstrlist(cfg->local_zones); 1209 cfg->local_zones = NULL; 1210 config_delstrlist(cfg->local_zones_nodefault); 1211 cfg->local_zones_nodefault = NULL; 1212 config_delstrlist(cfg->local_data); 1213 cfg->local_data = NULL; 1214 } 1215 1216 int 1217 local_zones_apply_cfg(struct local_zones* zones, struct config_file* cfg) 1218 { 1219 /* create zones from zone statements. */ 1220 if(!lz_enter_zones(zones, cfg)) { 1221 return 0; 1222 } 1223 /* apply default zones+content (unless disabled, or overridden) */ 1224 if(!local_zone_enter_defaults(zones, cfg)) { 1225 return 0; 1226 } 1227 /* enter local zone overrides */ 1228 if(!lz_enter_overrides(zones, cfg)) { 1229 return 0; 1230 } 1231 /* create implicit transparent zone from data. */ 1232 if(!lz_setup_implicit(zones, cfg)) { 1233 return 0; 1234 } 1235 1236 /* setup parent ptrs for lookup during data entry */ 1237 lz_init_parents(zones); 1238 /* insert local zone tags */ 1239 if(!lz_enter_zone_tags(zones, cfg)) { 1240 return 0; 1241 } 1242 /* insert local data */ 1243 if(!lz_enter_data(zones, cfg)) { 1244 return 0; 1245 } 1246 /* freeup memory from cfg struct. */ 1247 lz_freeup_cfg(cfg); 1248 return 1; 1249 } 1250 1251 struct local_zone* 1252 local_zones_lookup(struct local_zones* zones, 1253 uint8_t* name, size_t len, int labs, uint16_t dclass, uint16_t dtype, 1254 int foradd) 1255 { 1256 return local_zones_tags_lookup(zones, name, len, labs, 1257 dclass, dtype, NULL, 0, 1, foradd); 1258 } 1259 1260 struct local_zone* 1261 local_zones_tags_lookup(struct local_zones* zones, 1262 uint8_t* name, size_t len, int labs, uint16_t dclass, uint16_t dtype, 1263 uint8_t* taglist, size_t taglen, int ignoretags, int foradd) 1264 { 1265 rbnode_type* res = NULL; 1266 struct local_zone *result; 1267 struct local_zone key; 1268 int m; 1269 key.node.key = &key; 1270 key.dclass = dclass; 1271 /* for type DS use a zone higher when on a zonecut */ 1272 if(dtype == LDNS_RR_TYPE_DS && !dname_is_root(name)) { 1273 /* If this is at a zone cut, of a local-zone, and it is 1274 * of type always_refuse. Then also refuse the type DS 1275 * for it. That could make it DNSSEC bogus, but it is 1276 * REFUSED anyway. It stops CNAME type answers in the 1277 * type DS lookup. */ 1278 key.name = name; 1279 key.namelen = len; 1280 key.namelabs = labs; 1281 /* For additions and removals, use the ordinary rule, 1282 * to remove a label for type DS to locate the parent zone. 1283 * That is where the DS RR needs to be put. */ 1284 if(!foradd && 1285 (result=(struct local_zone*)rbtree_search( 1286 &zones->ztree, &key)) != NULL && 1287 result->type == local_zone_always_refuse) { 1288 /* The type DS does not go up one label. */ 1289 return result; 1290 } else { 1291 dname_remove_label(&name, &len); 1292 labs--; 1293 } 1294 } 1295 key.name = name; 1296 key.namelen = len; 1297 key.namelabs = labs; 1298 rbtree_find_less_equal(&zones->ztree, &key, &res); 1299 result = (struct local_zone*)res; 1300 /* exact or smaller element (or no element) */ 1301 if(!result || result->dclass != dclass) 1302 return NULL; 1303 /* count number of labels matched */ 1304 (void)dname_lab_cmp(result->name, result->namelabs, key.name, 1305 key.namelabs, &m); 1306 while(result) { /* go up until qname is zone or subdomain of zone */ 1307 if(result->namelabs <= m) 1308 if(ignoretags || !result->taglist || 1309 taglist_intersect(result->taglist, 1310 result->taglen, taglist, taglen)) 1311 break; 1312 result = result->parent; 1313 } 1314 return result; 1315 } 1316 1317 struct local_zone* 1318 local_zones_find(struct local_zones* zones, 1319 uint8_t* name, size_t len, int labs, uint16_t dclass) 1320 { 1321 struct local_zone key; 1322 key.node.key = &key; 1323 key.dclass = dclass; 1324 key.name = name; 1325 key.namelen = len; 1326 key.namelabs = labs; 1327 /* exact */ 1328 return (struct local_zone*)rbtree_search(&zones->ztree, &key); 1329 } 1330 1331 struct local_zone* 1332 local_zones_find_le(struct local_zones* zones, 1333 uint8_t* name, size_t len, int labs, uint16_t dclass, 1334 int* exact) 1335 { 1336 struct local_zone key; 1337 rbnode_type *node; 1338 key.node.key = &key; 1339 key.dclass = dclass; 1340 key.name = name; 1341 key.namelen = len; 1342 key.namelabs = labs; 1343 *exact = rbtree_find_less_equal(&zones->ztree, &key, &node); 1344 return (struct local_zone*)node; 1345 } 1346 1347 /** encode answer consisting of 1 rrset */ 1348 static int 1349 local_encode(struct query_info* qinfo, struct module_env* env, 1350 struct edns_data* edns, struct comm_reply* repinfo, sldns_buffer* buf, 1351 struct regional* temp, struct ub_packed_rrset_key* rrset, int ansec, 1352 int rcode) 1353 { 1354 struct reply_info rep; 1355 uint16_t udpsize; 1356 /* make answer with time=0 for fixed TTL values */ 1357 memset(&rep, 0, sizeof(rep)); 1358 rep.flags = (uint16_t)((BIT_QR | BIT_AA | BIT_RA) | rcode); 1359 rep.qdcount = 1; 1360 if(ansec) 1361 rep.an_numrrsets = 1; 1362 else rep.ns_numrrsets = 1; 1363 rep.rrset_count = 1; 1364 rep.rrsets = &rrset; 1365 rep.reason_bogus = LDNS_EDE_NONE; 1366 udpsize = edns->udp_size; 1367 edns->edns_version = EDNS_ADVERTISED_VERSION; 1368 edns->udp_size = EDNS_ADVERTISED_SIZE; 1369 edns->ext_rcode = 0; 1370 edns->bits &= EDNS_DO; 1371 if(!inplace_cb_reply_local_call(env, qinfo, NULL, &rep, rcode, edns, 1372 repinfo, temp, env->now_tv) || !reply_info_answer_encode(qinfo, &rep, 1373 *(uint16_t*)sldns_buffer_begin(buf), sldns_buffer_read_u16_at(buf, 2), 1374 buf, 0, 0, temp, udpsize, edns, (int)(edns->bits&EDNS_DO), 0)) { 1375 error_encode(buf, (LDNS_RCODE_SERVFAIL|BIT_AA), qinfo, 1376 *(uint16_t*)sldns_buffer_begin(buf), 1377 sldns_buffer_read_u16_at(buf, 2), edns); 1378 } 1379 return 1; 1380 } 1381 1382 /** encode local error answer */ 1383 static void 1384 local_error_encode(struct query_info* qinfo, struct module_env* env, 1385 struct edns_data* edns, struct comm_reply* repinfo, sldns_buffer* buf, 1386 struct regional* temp, int rcode, int r, int ede_code, 1387 const char* ede_txt) 1388 { 1389 edns->edns_version = EDNS_ADVERTISED_VERSION; 1390 edns->udp_size = EDNS_ADVERTISED_SIZE; 1391 edns->ext_rcode = 0; 1392 edns->bits &= EDNS_DO; 1393 1394 if(!inplace_cb_reply_local_call(env, qinfo, NULL, NULL, 1395 rcode, edns, repinfo, temp, env->now_tv)) 1396 edns->opt_list_inplace_cb_out = NULL; 1397 1398 if(ede_code != LDNS_EDE_NONE && env->cfg->ede) { 1399 edns_opt_list_append_ede(&edns->opt_list_out, temp, 1400 ede_code, ede_txt); 1401 } 1402 1403 error_encode(buf, r, qinfo, *(uint16_t*)sldns_buffer_begin(buf), 1404 sldns_buffer_read_u16_at(buf, 2), edns); 1405 } 1406 1407 /** find local data tag string match for the given type in the list */ 1408 int 1409 local_data_find_tag_datas(const struct query_info* qinfo, 1410 struct config_strlist* list, struct ub_packed_rrset_key* r, 1411 struct regional* temp) 1412 { 1413 struct config_strlist* p; 1414 char buf[65536]; 1415 uint8_t rr[LDNS_RR_BUF_SIZE]; 1416 size_t len; 1417 int res; 1418 struct packed_rrset_data* d; 1419 for(p=list; p; p=p->next) { 1420 uint16_t rdr_type; 1421 1422 len = sizeof(rr); 1423 /* does this element match the type? */ 1424 snprintf(buf, sizeof(buf), ". %s", p->str); 1425 res = sldns_str2wire_rr_buf(buf, rr, &len, NULL, 3600, 1426 NULL, 0, NULL, 0); 1427 if(res != 0) 1428 /* parse errors are already checked before, in 1429 * acllist check_data, skip this for robustness */ 1430 continue; 1431 if(len < 1 /* . */ + 8 /* typeclassttl*/ + 2 /*rdatalen*/) 1432 continue; 1433 rdr_type = sldns_wirerr_get_type(rr, len, 1); 1434 if(rdr_type != qinfo->qtype && rdr_type != LDNS_RR_TYPE_CNAME) 1435 continue; 1436 1437 /* do we have entries already? if not setup key */ 1438 if(r->rk.dname == NULL) { 1439 r->entry.key = r; 1440 r->rk.dname = qinfo->qname; 1441 r->rk.dname_len = qinfo->qname_len; 1442 r->rk.type = htons(rdr_type); 1443 r->rk.rrset_class = htons(qinfo->qclass); 1444 r->rk.flags = 0; 1445 d = (struct packed_rrset_data*)regional_alloc_zero( 1446 temp, sizeof(struct packed_rrset_data) 1447 + sizeof(size_t) + sizeof(uint8_t*) + 1448 sizeof(time_t)); 1449 if(!d) return 0; /* out of memory */ 1450 r->entry.data = d; 1451 d->ttl = sldns_wirerr_get_ttl(rr, len, 1); 1452 d->rr_len = (size_t*)((uint8_t*)d + 1453 sizeof(struct packed_rrset_data)); 1454 d->rr_data = (uint8_t**)&(d->rr_len[1]); 1455 d->rr_ttl = (time_t*)&(d->rr_data[1]); 1456 } 1457 d = (struct packed_rrset_data*)r->entry.data; 1458 /* add entry to the data */ 1459 if(d->count != 0) { 1460 size_t* oldlen = d->rr_len; 1461 uint8_t** olddata = d->rr_data; 1462 time_t* oldttl = d->rr_ttl; 1463 /* increase arrays for lookup */ 1464 /* this is of course slow for very many records, 1465 * but most redirects are expected with few records */ 1466 d->rr_len = (size_t*)regional_alloc_zero(temp, 1467 (d->count+1)*sizeof(size_t)); 1468 d->rr_data = (uint8_t**)regional_alloc_zero(temp, 1469 (d->count+1)*sizeof(uint8_t*)); 1470 d->rr_ttl = (time_t*)regional_alloc_zero(temp, 1471 (d->count+1)*sizeof(time_t)); 1472 if(!d->rr_len || !d->rr_data || !d->rr_ttl) 1473 return 0; /* out of memory */ 1474 /* first one was allocated after struct d, but new 1475 * ones get their own array increment alloc, so 1476 * copy old content */ 1477 memmove(d->rr_len, oldlen, d->count*sizeof(size_t)); 1478 memmove(d->rr_data, olddata, d->count*sizeof(uint8_t*)); 1479 memmove(d->rr_ttl, oldttl, d->count*sizeof(time_t)); 1480 } 1481 1482 d->rr_len[d->count] = sldns_wirerr_get_rdatalen(rr, len, 1)+2; 1483 d->rr_ttl[d->count] = sldns_wirerr_get_ttl(rr, len, 1); 1484 d->rr_data[d->count] = regional_alloc_init(temp, 1485 sldns_wirerr_get_rdatawl(rr, len, 1), 1486 d->rr_len[d->count]); 1487 if(!d->rr_data[d->count]) 1488 return 0; /* out of memory */ 1489 d->count++; 1490 } 1491 if(r->rk.dname) 1492 return 1; 1493 return 0; 1494 } 1495 1496 static int 1497 find_tag_datas(struct query_info* qinfo, struct config_strlist* list, 1498 struct ub_packed_rrset_key* r, struct regional* temp) 1499 { 1500 int result = local_data_find_tag_datas(qinfo, list, r, temp); 1501 1502 /* If we've found a non-exact alias type of local data, make a shallow 1503 * copy of the RRset and remember it in qinfo to complete the alias 1504 * chain later. */ 1505 if(result && qinfo->qtype != LDNS_RR_TYPE_CNAME && 1506 r->rk.type == htons(LDNS_RR_TYPE_CNAME)) { 1507 qinfo->local_alias = 1508 regional_alloc_zero(temp, sizeof(struct local_rrset)); 1509 if(!qinfo->local_alias) 1510 return 0; /* out of memory */ 1511 qinfo->local_alias->rrset = 1512 regional_alloc_init(temp, r, sizeof(*r)); 1513 if(!qinfo->local_alias->rrset) { 1514 qinfo->local_alias = NULL; 1515 return 0; /* out of memory */ 1516 } 1517 } 1518 return result; 1519 } 1520 1521 int 1522 local_data_answer(struct local_zone* z, struct module_env* env, 1523 struct query_info* qinfo, struct edns_data* edns, 1524 struct comm_reply* repinfo, sldns_buffer* buf, 1525 struct regional* temp, int labs, struct local_data** ldp, 1526 enum localzone_type lz_type, int tag, struct config_strlist** tag_datas, 1527 size_t tag_datas_size, char** tagname, int num_tags) 1528 { 1529 struct local_data key; 1530 struct local_data* ld; 1531 struct local_rrset* lr; 1532 key.node.key = &key; 1533 key.name = qinfo->qname; 1534 key.namelen = qinfo->qname_len; 1535 key.namelabs = labs; 1536 if(lz_type == local_zone_redirect || 1537 lz_type == local_zone_inform_redirect) { 1538 key.name = z->name; 1539 key.namelen = z->namelen; 1540 key.namelabs = z->namelabs; 1541 if(tag != -1 && (size_t)tag<tag_datas_size && tag_datas[tag]) { 1542 struct ub_packed_rrset_key r; 1543 memset(&r, 0, sizeof(r)); 1544 if(find_tag_datas(qinfo, tag_datas[tag], &r, temp)) { 1545 verbose(VERB_ALGO, "redirect with tag data [%d] %s", 1546 tag, (tag<num_tags?tagname[tag]:"null")); 1547 1548 /* If we found a matching alias, we should 1549 * use it as part of the answer, but we can't 1550 * encode it until we complete the alias 1551 * chain. */ 1552 if(qinfo->local_alias) 1553 return 1; 1554 return local_encode(qinfo, env, edns, repinfo, buf, temp, 1555 &r, 1, LDNS_RCODE_NOERROR); 1556 } 1557 } 1558 } 1559 ld = (struct local_data*)rbtree_search(&z->data, &key.node); 1560 *ldp = ld; 1561 if(!ld) { 1562 return 0; 1563 } 1564 lr = local_data_find_type(ld, qinfo->qtype, 1); 1565 if(!lr) 1566 return 0; 1567 1568 /* Special case for alias matching. See local_data_answer(). */ 1569 if((lz_type == local_zone_redirect || 1570 lz_type == local_zone_inform_redirect) && 1571 qinfo->qtype != LDNS_RR_TYPE_CNAME && 1572 lr->rrset->rk.type == htons(LDNS_RR_TYPE_CNAME)) { 1573 uint8_t* ctarget; 1574 size_t ctargetlen = 0; 1575 1576 qinfo->local_alias = 1577 regional_alloc_zero(temp, sizeof(struct local_rrset)); 1578 if(!qinfo->local_alias) 1579 return 0; /* out of memory */ 1580 qinfo->local_alias->rrset = regional_alloc_init( 1581 temp, lr->rrset, sizeof(*lr->rrset)); 1582 if(!qinfo->local_alias->rrset) { 1583 qinfo->local_alias = NULL; 1584 return 0; /* out of memory */ 1585 } 1586 qinfo->local_alias->rrset->rk.dname = qinfo->qname; 1587 qinfo->local_alias->rrset->rk.dname_len = qinfo->qname_len; 1588 get_cname_target(lr->rrset, &ctarget, &ctargetlen); 1589 if(!ctargetlen) { 1590 qinfo->local_alias = NULL; 1591 return 0; /* invalid cname */ 1592 } 1593 if(dname_is_wild(ctarget)) { 1594 /* synthesize cname target */ 1595 struct packed_rrset_data* d, *lr_d; 1596 /* -3 for wildcard label and root label from qname */ 1597 size_t newtargetlen = qinfo->qname_len + ctargetlen - 3; 1598 1599 log_assert(ctargetlen >= 3); 1600 log_assert(qinfo->qname_len >= 1); 1601 1602 if(newtargetlen > LDNS_MAX_DOMAINLEN) { 1603 qinfo->local_alias = NULL; 1604 local_error_encode(qinfo, env, edns, repinfo, 1605 buf, temp, LDNS_RCODE_YXDOMAIN, 1606 (LDNS_RCODE_YXDOMAIN|BIT_AA), 1607 LDNS_EDE_OTHER, 1608 "DNAME expansion became too large"); 1609 return 1; 1610 } 1611 memset(&qinfo->local_alias->rrset->entry, 0, 1612 sizeof(qinfo->local_alias->rrset->entry)); 1613 qinfo->local_alias->rrset->entry.key = 1614 qinfo->local_alias->rrset; 1615 qinfo->local_alias->rrset->entry.hash = 1616 rrset_key_hash(&qinfo->local_alias->rrset->rk); 1617 d = (struct packed_rrset_data*)regional_alloc_zero(temp, 1618 sizeof(struct packed_rrset_data) + sizeof(size_t) + 1619 sizeof(uint8_t*) + sizeof(time_t) + sizeof(uint16_t) 1620 + newtargetlen); 1621 if(!d) { 1622 qinfo->local_alias = NULL; 1623 return 0; /* out of memory */ 1624 } 1625 lr_d = (struct packed_rrset_data*)lr->rrset->entry.data; 1626 qinfo->local_alias->rrset->entry.data = d; 1627 d->ttl = lr_d->rr_ttl[0]; /* RFC6672-like behavior: 1628 synth CNAME TTL uses original TTL*/ 1629 d->count = 1; 1630 d->rrsig_count = 0; 1631 d->trust = rrset_trust_ans_noAA; 1632 d->rr_len = (size_t*)((uint8_t*)d + 1633 sizeof(struct packed_rrset_data)); 1634 d->rr_len[0] = newtargetlen + sizeof(uint16_t); 1635 packed_rrset_ptr_fixup(d); 1636 d->rr_ttl[0] = d->ttl; 1637 sldns_write_uint16(d->rr_data[0], newtargetlen); 1638 /* write qname */ 1639 memmove(d->rr_data[0] + sizeof(uint16_t), qinfo->qname, 1640 qinfo->qname_len - 1); 1641 /* write cname target wildcard label */ 1642 memmove(d->rr_data[0] + sizeof(uint16_t) + 1643 qinfo->qname_len - 1, ctarget + 2, 1644 ctargetlen - 2); 1645 } 1646 return 1; 1647 } 1648 if(lz_type == local_zone_redirect || 1649 lz_type == local_zone_inform_redirect) { 1650 /* convert rrset name to query name; like a wildcard */ 1651 struct ub_packed_rrset_key r = *lr->rrset; 1652 r.rk.dname = qinfo->qname; 1653 r.rk.dname_len = qinfo->qname_len; 1654 return local_encode(qinfo, env, edns, repinfo, buf, temp, &r, 1, 1655 LDNS_RCODE_NOERROR); 1656 } 1657 return local_encode(qinfo, env, edns, repinfo, buf, temp, lr->rrset, 1, 1658 LDNS_RCODE_NOERROR); 1659 } 1660 1661 /** 1662 * See if the local zone does not cover the name, eg. the name is not 1663 * in the zone and the zone is transparent */ 1664 static int 1665 local_zone_does_not_cover(struct local_zone* z, struct query_info* qinfo, 1666 int labs) 1667 { 1668 struct local_data key; 1669 struct local_data* ld = NULL; 1670 struct local_rrset* lr = NULL; 1671 if(z->type == local_zone_always_transparent || z->type == local_zone_block_a || z->type == local_zone_block_aaaa) 1672 return 1; 1673 if(z->type != local_zone_transparent 1674 && z->type != local_zone_typetransparent 1675 && z->type != local_zone_inform) 1676 return 0; 1677 key.node.key = &key; 1678 key.name = qinfo->qname; 1679 key.namelen = qinfo->qname_len; 1680 key.namelabs = labs; 1681 ld = (struct local_data*)rbtree_search(&z->data, &key.node); 1682 if(z->type == local_zone_transparent || z->type == local_zone_inform 1683 || z->type == local_zone_block_a_wdata 1684 || z->type == local_zone_block_aaaa_wdata) 1685 return (ld == NULL); 1686 if(ld) 1687 lr = local_data_find_type(ld, qinfo->qtype, 1); 1688 /* local_zone_typetransparent */ 1689 return (lr == NULL); 1690 } 1691 1692 static inline int 1693 local_zone_is_udp_query(struct comm_reply* repinfo) { 1694 return repinfo != NULL 1695 ? (repinfo->c != NULL 1696 ? repinfo->c->type == comm_udp 1697 : 0) 1698 : 0; 1699 } 1700 1701 int 1702 local_zones_zone_answer(struct local_zone* z, struct module_env* env, 1703 struct query_info* qinfo, struct edns_data* edns, 1704 struct comm_reply* repinfo, sldns_buffer* buf, struct regional* temp, 1705 struct local_data* ld, enum localzone_type lz_type) 1706 { 1707 if(lz_type == local_zone_deny || 1708 lz_type == local_zone_always_deny || 1709 lz_type == local_zone_inform_deny) { 1710 /** no reply at all, signal caller by clearing buffer. */ 1711 sldns_buffer_clear(buf); 1712 sldns_buffer_flip(buf); 1713 return 1; 1714 } else if(lz_type == local_zone_refuse 1715 || lz_type == local_zone_always_refuse) { 1716 local_error_encode(qinfo, env, edns, repinfo, buf, temp, 1717 LDNS_RCODE_REFUSED, (LDNS_RCODE_REFUSED|BIT_AA), 1718 LDNS_EDE_NONE, NULL); 1719 return 1; 1720 } else if(lz_type == local_zone_static || 1721 lz_type == local_zone_redirect || 1722 lz_type == local_zone_inform_redirect || 1723 lz_type == local_zone_always_nxdomain || 1724 lz_type == local_zone_always_nodata || 1725 (lz_type == local_zone_truncate 1726 && local_zone_is_udp_query(repinfo))) { 1727 /* for static, reply nodata or nxdomain 1728 * for redirect, reply nodata */ 1729 /* no additional section processing, 1730 * cname, dname or wildcard processing, 1731 * or using closest match for NSEC. 1732 * or using closest match for returning delegation downwards 1733 */ 1734 int rcode = (ld || lz_type == local_zone_redirect || 1735 lz_type == local_zone_inform_redirect || 1736 lz_type == local_zone_always_nodata || 1737 lz_type == local_zone_truncate)? 1738 LDNS_RCODE_NOERROR:LDNS_RCODE_NXDOMAIN; 1739 rcode = (lz_type == local_zone_truncate ? (rcode|BIT_TC) : rcode); 1740 if(z != NULL && z->soa && z->soa_negative) 1741 return local_encode(qinfo, env, edns, repinfo, buf, temp, 1742 z->soa_negative, 0, rcode); 1743 local_error_encode(qinfo, env, edns, repinfo, buf, temp, 1744 rcode, (rcode|BIT_AA), LDNS_EDE_NONE, NULL); 1745 return 1; 1746 } else if(lz_type == local_zone_typetransparent 1747 || lz_type == local_zone_always_transparent) { 1748 /* no NODATA or NXDOMAINS for this zone type */ 1749 return 0; 1750 } else if(lz_type == local_zone_block_a || 1751 lz_type == local_zone_block_a_wdata) { 1752 /* Return NODATA for all A queries */ 1753 if(qinfo->qtype == LDNS_RR_TYPE_A) { 1754 local_error_encode(qinfo, env, edns, repinfo, buf, temp, 1755 LDNS_RCODE_NOERROR, (LDNS_RCODE_NOERROR|BIT_AA), 1756 LDNS_EDE_NONE, NULL); 1757 return 1; 1758 } 1759 1760 return 0; 1761 } else if(lz_type == local_zone_block_aaaa || 1762 lz_type == local_zone_block_aaaa_wdata) { 1763 /* Return NODATA for all AAAA queries */ 1764 if(qinfo->qtype == LDNS_RR_TYPE_AAAA) { 1765 local_error_encode(qinfo, env, edns, repinfo, buf, temp, 1766 LDNS_RCODE_NOERROR, (LDNS_RCODE_NOERROR|BIT_AA), 1767 LDNS_EDE_NONE, NULL); 1768 return 1; 1769 } 1770 1771 return 0; 1772 } else if(lz_type == local_zone_always_null) { 1773 /* 0.0.0.0 or ::0 or noerror/nodata for this zone type, 1774 * used for blocklists. */ 1775 if(qinfo->qtype == LDNS_RR_TYPE_A || 1776 qinfo->qtype == LDNS_RR_TYPE_AAAA) { 1777 struct ub_packed_rrset_key lrr; 1778 struct packed_rrset_data d; 1779 time_t rr_ttl = 3600; 1780 size_t rr_len = 0; 1781 uint8_t rr_data[2+16] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0}; 1782 uint8_t* rr_datas = rr_data; 1783 memset(&lrr, 0, sizeof(lrr)); 1784 memset(&d, 0, sizeof(d)); 1785 lrr.entry.data = &d; 1786 lrr.rk.dname = qinfo->qname; 1787 lrr.rk.dname_len = qinfo->qname_len; 1788 lrr.rk.type = htons(qinfo->qtype); 1789 lrr.rk.rrset_class = htons(qinfo->qclass); 1790 if(qinfo->qtype == LDNS_RR_TYPE_A) { 1791 rr_len = 4; 1792 sldns_write_uint16(rr_data, rr_len); 1793 rr_len += 2; 1794 } else { 1795 rr_len = 16; 1796 sldns_write_uint16(rr_data, rr_len); 1797 rr_len += 2; 1798 } 1799 d.ttl = rr_ttl; 1800 d.count = 1; 1801 d.rr_len = &rr_len; 1802 d.rr_data = &rr_datas; 1803 d.rr_ttl = &rr_ttl; 1804 return local_encode(qinfo, env, edns, repinfo, buf, temp, 1805 &lrr, 1, LDNS_RCODE_NOERROR); 1806 } else { 1807 /* NODATA: No EDE needed */ 1808 local_error_encode(qinfo, env, edns, repinfo, buf, 1809 temp, LDNS_RCODE_NOERROR, 1810 (LDNS_RCODE_NOERROR|BIT_AA), -1, NULL); 1811 } 1812 return 1; 1813 } 1814 /* else lz_type == local_zone_transparent */ 1815 1816 /* if the zone is transparent and the name exists, but the type 1817 * does not, then we should make this noerror/nodata */ 1818 if(ld && ld->rrsets) { 1819 int rcode = LDNS_RCODE_NOERROR; 1820 if(z != NULL && z->soa && z->soa_negative) 1821 return local_encode(qinfo, env, edns, repinfo, buf, temp, 1822 z->soa_negative, 0, rcode); 1823 /* NODATA: No EDE needed */ 1824 local_error_encode(qinfo, env, edns, repinfo, buf, temp, rcode, 1825 (rcode|BIT_AA), LDNS_EDE_NONE, NULL); 1826 return 1; 1827 } 1828 1829 /* stop here, and resolve further on */ 1830 return 0; 1831 } 1832 1833 /** print log information for an inform zone query */ 1834 static void 1835 lz_inform_print(struct local_zone* z, struct query_info* qinfo, 1836 struct sockaddr_storage* addr, socklen_t addrlen) 1837 { 1838 char ip[128], txt[512]; 1839 char zname[LDNS_MAX_DOMAINLEN]; 1840 uint16_t port = ntohs(((struct sockaddr_in*)addr)->sin_port); 1841 dname_str(z->name, zname); 1842 addr_to_str(addr, addrlen, ip, sizeof(ip)); 1843 snprintf(txt, sizeof(txt), "%s %s %s@%u", zname, local_zone_type2str(z->type), ip, 1844 (unsigned)port); 1845 log_nametypeclass(NO_VERBOSE, txt, qinfo->qname, qinfo->qtype, qinfo->qclass); 1846 } 1847 1848 static enum localzone_type 1849 lz_type(uint8_t *taglist, size_t taglen, uint8_t *taglist2, size_t taglen2, 1850 uint8_t *tagactions, size_t tagactionssize, enum localzone_type lzt, 1851 struct comm_reply* repinfo, struct rbtree_type* override_tree, 1852 int* tag, char** tagname, int num_tags) 1853 { 1854 struct local_zone_override* lzo; 1855 if(repinfo && override_tree) { 1856 lzo = (struct local_zone_override*)addr_tree_lookup( 1857 override_tree, &repinfo->client_addr, 1858 repinfo->client_addrlen); 1859 if(lzo && lzo->type) { 1860 verbose(VERB_ALGO, "local zone override to type %s", 1861 local_zone_type2str(lzo->type)); 1862 return lzo->type; 1863 } 1864 } 1865 if(!taglist || !taglist2) 1866 return lzt; 1867 return local_data_find_tag_action(taglist, taglen, taglist2, taglen2, 1868 tagactions, tagactionssize, lzt, tag, tagname, num_tags); 1869 } 1870 1871 enum localzone_type 1872 local_data_find_tag_action(const uint8_t* taglist, size_t taglen, 1873 const uint8_t* taglist2, size_t taglen2, const uint8_t* tagactions, 1874 size_t tagactionssize, enum localzone_type lzt, int* tag, 1875 char* const* tagname, int num_tags) 1876 { 1877 size_t i, j; 1878 uint8_t tagmatch; 1879 1880 for(i=0; i<taglen && i<taglen2; i++) { 1881 tagmatch = (taglist[i] & taglist2[i]); 1882 for(j=0; j<8 && tagmatch>0; j++) { 1883 if((tagmatch & 0x1)) { 1884 *tag = (int)(i*8+j); 1885 verbose(VERB_ALGO, "matched tag [%d] %s", 1886 *tag, (*tag<num_tags?tagname[*tag]:"null")); 1887 /* does this tag have a tag action? */ 1888 if(i*8+j < tagactionssize && tagactions 1889 && tagactions[i*8+j] != 0) { 1890 verbose(VERB_ALGO, "tag action [%d] %s to type %s", 1891 *tag, (*tag<num_tags?tagname[*tag]:"null"), 1892 local_zone_type2str( 1893 (enum localzone_type) 1894 tagactions[i*8+j])); 1895 return (enum localzone_type)tagactions[i*8+j]; 1896 } 1897 return lzt; 1898 } 1899 tagmatch >>= 1; 1900 } 1901 } 1902 return lzt; 1903 } 1904 1905 int 1906 local_zones_answer(struct local_zones* zones, struct module_env* env, 1907 struct query_info* qinfo, struct edns_data* edns, sldns_buffer* buf, 1908 struct regional* temp, struct comm_reply* repinfo, uint8_t* taglist, 1909 size_t taglen, uint8_t* tagactions, size_t tagactionssize, 1910 struct config_strlist** tag_datas, size_t tag_datas_size, 1911 char** tagname, int num_tags, struct view* view) 1912 { 1913 /* see if query is covered by a zone, 1914 * if so: - try to match (exact) local data 1915 * - look at zone type for negative response. */ 1916 int labs = dname_count_labels(qinfo->qname); 1917 struct local_data* ld = NULL; 1918 struct local_zone* z = NULL; 1919 enum localzone_type lzt = local_zone_transparent; 1920 int r, tag = -1; 1921 1922 if(view) { 1923 lock_rw_rdlock(&view->lock); 1924 if(view->local_zones && 1925 (z = local_zones_lookup(view->local_zones, 1926 qinfo->qname, qinfo->qname_len, labs, 1927 qinfo->qclass, qinfo->qtype, 0))) { 1928 lock_rw_rdlock(&z->lock); 1929 lzt = z->type; 1930 } 1931 if(lzt == local_zone_noview) { 1932 lock_rw_unlock(&z->lock); 1933 z = NULL; 1934 } 1935 if(z && (lzt == local_zone_transparent || 1936 lzt == local_zone_typetransparent || 1937 lzt == local_zone_inform || 1938 lzt == local_zone_always_transparent || 1939 lzt == local_zone_block_a || 1940 lzt == local_zone_block_aaaa || 1941 lzt == local_zone_block_a_wdata || 1942 lzt == local_zone_block_aaaa_wdata) && 1943 local_zone_does_not_cover(z, qinfo, labs)) { 1944 lock_rw_unlock(&z->lock); 1945 z = NULL; 1946 } 1947 if(view->local_zones && !z && !view->isfirst){ 1948 lock_rw_unlock(&view->lock); 1949 return 0; 1950 } 1951 if(z && verbosity >= VERB_ALGO) { 1952 char zname[LDNS_MAX_DOMAINLEN]; 1953 dname_str(z->name, zname); 1954 verbose(VERB_ALGO, "using localzone %s %s from view %s", 1955 zname, local_zone_type2str(lzt), view->name); 1956 } 1957 lock_rw_unlock(&view->lock); 1958 } 1959 if(!z) { 1960 /* try global local_zones tree */ 1961 lock_rw_rdlock(&zones->lock); 1962 if(!(z = local_zones_tags_lookup(zones, qinfo->qname, 1963 qinfo->qname_len, labs, qinfo->qclass, qinfo->qtype, 1964 taglist, taglen, 0, 0))) { 1965 lock_rw_unlock(&zones->lock); 1966 return 0; 1967 } 1968 lock_rw_rdlock(&z->lock); 1969 lzt = lz_type(taglist, taglen, z->taglist, z->taglen, 1970 tagactions, tagactionssize, z->type, repinfo, 1971 z->override_tree, &tag, tagname, num_tags); 1972 lock_rw_unlock(&zones->lock); 1973 if(z && verbosity >= VERB_ALGO) { 1974 char zname[LDNS_MAX_DOMAINLEN]; 1975 dname_str(z->name, zname); 1976 verbose(VERB_ALGO, "using localzone %s %s", zname, 1977 local_zone_type2str(lzt)); 1978 } 1979 } 1980 if((env->cfg->log_local_actions || 1981 lzt == local_zone_inform || 1982 lzt == local_zone_inform_deny || 1983 lzt == local_zone_inform_redirect) 1984 && repinfo) 1985 lz_inform_print(z, qinfo, &repinfo->client_addr, 1986 repinfo->client_addrlen); 1987 1988 if(lzt != local_zone_always_refuse 1989 && lzt != local_zone_always_transparent 1990 && lzt != local_zone_block_a 1991 && lzt != local_zone_block_aaaa 1992 && lzt != local_zone_always_nxdomain 1993 && lzt != local_zone_always_nodata 1994 && lzt != local_zone_always_deny 1995 && local_data_answer(z, env, qinfo, edns, repinfo, buf, temp, labs, 1996 &ld, lzt, tag, tag_datas, tag_datas_size, tagname, num_tags)) { 1997 lock_rw_unlock(&z->lock); 1998 /* We should tell the caller that encode is deferred if we found 1999 * a local alias. */ 2000 return !qinfo->local_alias; 2001 } 2002 r = local_zones_zone_answer(z, env, qinfo, edns, repinfo, buf, temp, ld, lzt); 2003 lock_rw_unlock(&z->lock); 2004 return r && !qinfo->local_alias; /* see above */ 2005 } 2006 2007 const char* local_zone_type2str(enum localzone_type t) 2008 { 2009 switch(t) { 2010 case local_zone_unset: return "unset"; 2011 case local_zone_deny: return "deny"; 2012 case local_zone_refuse: return "refuse"; 2013 case local_zone_redirect: return "redirect"; 2014 case local_zone_transparent: return "transparent"; 2015 case local_zone_typetransparent: return "typetransparent"; 2016 case local_zone_static: return "static"; 2017 case local_zone_nodefault: return "nodefault"; 2018 case local_zone_inform: return "inform"; 2019 case local_zone_inform_deny: return "inform_deny"; 2020 case local_zone_inform_redirect: return "inform_redirect"; 2021 case local_zone_always_transparent: return "always_transparent"; 2022 case local_zone_block_a: return "block_a"; 2023 case local_zone_block_aaaa: return "block_aaaa"; 2024 case local_zone_block_a_wdata: return "block_a_wdata"; 2025 case local_zone_block_aaaa_wdata: return "block_aaaa_wdata"; 2026 case local_zone_always_refuse: return "always_refuse"; 2027 case local_zone_always_nxdomain: return "always_nxdomain"; 2028 case local_zone_always_nodata: return "always_nodata"; 2029 case local_zone_always_deny: return "always_deny"; 2030 case local_zone_always_null: return "always_null"; 2031 case local_zone_noview: return "noview"; 2032 case local_zone_truncate: return "truncate"; 2033 case local_zone_invalid: return "invalid"; 2034 } 2035 return "badtyped"; 2036 } 2037 2038 int local_zone_str2type(const char* type, enum localzone_type* t) 2039 { 2040 if(strcmp(type, "deny") == 0) 2041 *t = local_zone_deny; 2042 else if(strcmp(type, "refuse") == 0) 2043 *t = local_zone_refuse; 2044 else if(strcmp(type, "static") == 0) 2045 *t = local_zone_static; 2046 else if(strcmp(type, "transparent") == 0) 2047 *t = local_zone_transparent; 2048 else if(strcmp(type, "typetransparent") == 0) 2049 *t = local_zone_typetransparent; 2050 else if(strcmp(type, "redirect") == 0) 2051 *t = local_zone_redirect; 2052 else if(strcmp(type, "inform") == 0) 2053 *t = local_zone_inform; 2054 else if(strcmp(type, "inform_deny") == 0) 2055 *t = local_zone_inform_deny; 2056 else if(strcmp(type, "inform_redirect") == 0) 2057 *t = local_zone_inform_redirect; 2058 else if(strcmp(type, "always_transparent") == 0) 2059 *t = local_zone_always_transparent; 2060 else if(strcmp(type, "block_a") == 0) 2061 *t = local_zone_block_a; 2062 else if(strcmp(type, "block_aaaa") == 0) 2063 *t = local_zone_block_aaaa; 2064 else if(strcmp(type, "block_a_wdata") == 0) 2065 *t = local_zone_block_a_wdata; 2066 else if(strcmp(type, "block_aaaa_wdata") == 0) 2067 *t = local_zone_block_aaaa_wdata; 2068 else if(strcmp(type, "always_refuse") == 0) 2069 *t = local_zone_always_refuse; 2070 else if(strcmp(type, "always_nxdomain") == 0) 2071 *t = local_zone_always_nxdomain; 2072 else if(strcmp(type, "always_nodata") == 0) 2073 *t = local_zone_always_nodata; 2074 else if(strcmp(type, "always_deny") == 0) 2075 *t = local_zone_always_deny; 2076 else if(strcmp(type, "always_null") == 0) 2077 *t = local_zone_always_null; 2078 else if(strcmp(type, "noview") == 0) 2079 *t = local_zone_noview; 2080 else if(strcmp(type, "truncate") == 0) 2081 *t = local_zone_truncate; 2082 else if(strcmp(type, "nodefault") == 0) 2083 *t = local_zone_nodefault; 2084 else return 0; 2085 return 1; 2086 } 2087 2088 /** iterate over the kiddies of the given name and set their parent ptr */ 2089 static void 2090 set_kiddo_parents(struct local_zone* z, struct local_zone* match, 2091 struct local_zone* newp) 2092 { 2093 /* both zones and z are locked already */ 2094 /* in the sorted rbtree, the kiddies of z are located after z */ 2095 /* z must be present in the tree */ 2096 struct local_zone* p = z; 2097 p = (struct local_zone*)rbtree_next(&p->node); 2098 while(p!=(struct local_zone*)RBTREE_NULL && 2099 p->dclass == z->dclass && dname_strict_subdomain(p->name, 2100 p->namelabs, z->name, z->namelabs)) { 2101 /* update parent ptr */ 2102 /* only when matches with existing parent pointer, so that 2103 * deeper child structures are not touched, i.e. 2104 * update of x, and a.x, b.x, f.b.x, g.b.x, c.x, y 2105 * gets to update a.x, b.x and c.x */ 2106 lock_rw_wrlock(&p->lock); 2107 if(p->parent == match) 2108 p->parent = newp; 2109 lock_rw_unlock(&p->lock); 2110 p = (struct local_zone*)rbtree_next(&p->node); 2111 } 2112 } 2113 2114 struct local_zone* local_zones_add_zone(struct local_zones* zones, 2115 uint8_t* name, size_t len, int labs, uint16_t dclass, 2116 enum localzone_type tp) 2117 { 2118 int exact; 2119 /* create */ 2120 struct local_zone *prev; 2121 struct local_zone* z = local_zone_create(name, len, labs, tp, dclass); 2122 if(!z) { 2123 free(name); 2124 return NULL; 2125 } 2126 lock_rw_wrlock(&z->lock); 2127 2128 /* find the closest parent */ 2129 prev = local_zones_find_le(zones, name, len, labs, dclass, &exact); 2130 if(!exact) 2131 z->parent = find_closest_parent(z, prev); 2132 2133 /* insert into the tree */ 2134 if(exact||!rbtree_insert(&zones->ztree, &z->node)) { 2135 /* duplicate entry! */ 2136 lock_rw_unlock(&z->lock); 2137 local_zone_delete(z); 2138 log_err("internal: duplicate entry in local_zones_add_zone"); 2139 return NULL; 2140 } 2141 2142 /* set parent pointers right */ 2143 set_kiddo_parents(z, z->parent, z); 2144 2145 lock_rw_unlock(&z->lock); 2146 return z; 2147 } 2148 2149 void local_zones_del_zone(struct local_zones* zones, struct local_zone* z) 2150 { 2151 /* fix up parents in tree */ 2152 lock_rw_wrlock(&z->lock); 2153 set_kiddo_parents(z, z, z->parent); 2154 2155 /* remove from tree */ 2156 (void)rbtree_delete(&zones->ztree, z); 2157 2158 /* delete the zone */ 2159 lock_rw_unlock(&z->lock); 2160 local_zone_delete(z); 2161 } 2162 2163 int 2164 local_zones_add_RR(struct local_zones* zones, const char* rr) 2165 { 2166 uint8_t* rr_name; 2167 uint16_t rr_class, rr_type; 2168 size_t len; 2169 int labs; 2170 struct local_zone* z; 2171 int r; 2172 if(!get_rr_nameclass(rr, &rr_name, &rr_class, &rr_type)) { 2173 return 0; 2174 } 2175 labs = dname_count_size_labels(rr_name, &len); 2176 /* could first try readlock then get writelock if zone does not exist, 2177 * but we do not add enough RRs (from multiple threads) to optimize */ 2178 lock_rw_wrlock(&zones->lock); 2179 z = local_zones_lookup(zones, rr_name, len, labs, rr_class, rr_type, 2180 1); 2181 if(!z) { 2182 z = local_zones_add_zone(zones, rr_name, len, labs, rr_class, 2183 local_zone_transparent); 2184 if(!z) { 2185 lock_rw_unlock(&zones->lock); 2186 return 0; 2187 } 2188 } else { 2189 free(rr_name); 2190 } 2191 lock_rw_wrlock(&z->lock); 2192 lock_rw_unlock(&zones->lock); 2193 r = lz_enter_rr_into_zone(z, rr); 2194 lock_rw_unlock(&z->lock); 2195 return r; 2196 } 2197 2198 /** returns true if the node is terminal so no deeper domain names exist */ 2199 static int 2200 is_terminal(struct local_data* d) 2201 { 2202 /* for empty nonterminals, the deeper domain names are sorted 2203 * right after them, so simply check the next name in the tree 2204 */ 2205 struct local_data* n = (struct local_data*)rbtree_next(&d->node); 2206 if(n == (struct local_data*)RBTREE_NULL) 2207 return 1; /* last in tree, no deeper node */ 2208 if(dname_strict_subdomain(n->name, n->namelabs, d->name, d->namelabs)) 2209 return 0; /* there is a deeper node */ 2210 return 1; 2211 } 2212 2213 /** delete empty terminals from tree when final data is deleted */ 2214 static void 2215 del_empty_term(struct local_zone* z, struct local_data* d, 2216 uint8_t* name, size_t len, int labs) 2217 { 2218 while(d && d->rrsets == NULL && is_terminal(d)) { 2219 /* is this empty nonterminal? delete */ 2220 /* note, no memory recycling in zone region */ 2221 (void)rbtree_delete(&z->data, d); 2222 2223 /* go up and to the next label */ 2224 if(dname_is_root(name)) 2225 return; 2226 dname_remove_label(&name, &len); 2227 labs--; 2228 d = local_zone_find_data(z, name, len, labs); 2229 } 2230 } 2231 2232 /** find and remove type from list in domain struct */ 2233 static void 2234 del_local_rrset(struct local_data* d, uint16_t dtype) 2235 { 2236 struct local_rrset* prev=NULL, *p=d->rrsets; 2237 while(p && ntohs(p->rrset->rk.type) != dtype) { 2238 prev = p; 2239 p = p->next; 2240 } 2241 if(!p) 2242 return; /* rrset type not found */ 2243 /* unlink it */ 2244 if(prev) prev->next = p->next; 2245 else d->rrsets = p->next; 2246 /* no memory recycling for zone deletions ... */ 2247 } 2248 2249 void local_zones_del_data(struct local_zones* zones, 2250 uint8_t* name, size_t len, int labs, uint16_t dclass) 2251 { 2252 /* find zone */ 2253 struct local_zone* z; 2254 struct local_data* d; 2255 2256 /* remove DS */ 2257 lock_rw_rdlock(&zones->lock); 2258 z = local_zones_lookup(zones, name, len, labs, dclass, LDNS_RR_TYPE_DS, 2259 1); 2260 if(z) { 2261 lock_rw_wrlock(&z->lock); 2262 d = local_zone_find_data(z, name, len, labs); 2263 if(d) { 2264 del_local_rrset(d, LDNS_RR_TYPE_DS); 2265 del_empty_term(z, d, name, len, labs); 2266 } 2267 lock_rw_unlock(&z->lock); 2268 } 2269 lock_rw_unlock(&zones->lock); 2270 2271 /* remove other types */ 2272 lock_rw_rdlock(&zones->lock); 2273 z = local_zones_lookup(zones, name, len, labs, dclass, 0, 1); 2274 if(!z) { 2275 /* no such zone, we're done */ 2276 lock_rw_unlock(&zones->lock); 2277 return; 2278 } 2279 lock_rw_wrlock(&z->lock); 2280 lock_rw_unlock(&zones->lock); 2281 2282 /* find the domain */ 2283 d = local_zone_find_data(z, name, len, labs); 2284 if(d) { 2285 /* no memory recycling for zone deletions ... */ 2286 d->rrsets = NULL; 2287 /* did we delete the soa record ? */ 2288 if(query_dname_compare(d->name, z->name) == 0) { 2289 z->soa = NULL; 2290 z->soa_negative = NULL; 2291 } 2292 2293 /* cleanup the empty nonterminals for this name */ 2294 del_empty_term(z, d, name, len, labs); 2295 } 2296 2297 lock_rw_unlock(&z->lock); 2298 } 2299 2300 /** Get memory usage for local_zone */ 2301 static size_t 2302 local_zone_get_mem(struct local_zone* z) 2303 { 2304 size_t m = sizeof(*z); 2305 lock_rw_rdlock(&z->lock); 2306 m += z->namelen + z->taglen + regional_get_mem(z->region); 2307 lock_rw_unlock(&z->lock); 2308 return m; 2309 } 2310 2311 size_t local_zones_get_mem(struct local_zones* zones) 2312 { 2313 struct local_zone* z; 2314 size_t m; 2315 if(!zones) return 0; 2316 m = sizeof(*zones); 2317 lock_rw_rdlock(&zones->lock); 2318 RBTREE_FOR(z, struct local_zone*, &zones->ztree) { 2319 m += local_zone_get_mem(z); 2320 } 2321 lock_rw_unlock(&zones->lock); 2322 return m; 2323 } 2324 2325 void local_zones_swap_tree(struct local_zones* zones, struct local_zones* data) 2326 { 2327 rbtree_type oldtree = zones->ztree; 2328 zones->ztree = data->ztree; 2329 data->ztree = oldtree; 2330 } 2331