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