view.c 17.5 KB
Newer Older
Bob Halley's avatar
add  
Bob Halley committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
/*
 * Copyright (C) 1999  Internet Software Consortium.
 * 
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
 * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
 * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
 * SOFTWARE.
 */

#include <config.h>

#include <string.h>

#include <isc/types.h>
#include <isc/result.h>
#include <isc/mem.h>
#include <isc/assertions.h>
#include <isc/error.h>
Michael Graff's avatar
Michael Graff committed
27
#include <isc/util.h>
Bob Halley's avatar
add  
Bob Halley committed
28 29

#include <dns/types.h>
Bob Halley's avatar
add adb  
Bob Halley committed
30
#include <dns/adb.h>
31
#include <dns/cache.h>
Bob Halley's avatar
add  
Bob Halley committed
32
#include <dns/dbtable.h>
Bob Halley's avatar
Bob Halley committed
33
#include <dns/db.h>
34
#include <dns/events.h>
Bob Halley's avatar
Bob Halley committed
35
#include <dns/fixedname.h>
Bob Halley's avatar
Bob Halley committed
36
#include <dns/rbt.h>
Bob Halley's avatar
Bob Halley committed
37
#include <dns/rdataset.h>
Bob Halley's avatar
add  
Bob Halley committed
38 39 40
#include <dns/resolver.h>
#include <dns/view.h>

41 42 43 44 45 46
#define RESSHUTDOWN(v)	(((v)->attributes & DNS_VIEWATTR_RESSHUTDOWN) != 0)
#define ADBSHUTDOWN(v)	(((v)->attributes & DNS_VIEWATTR_ADBSHUTDOWN) != 0)

static void resolver_shutdown(isc_task_t *task, isc_event_t *event);
static void adb_shutdown(isc_task_t *task, isc_event_t *event);

Bob Halley's avatar
add  
Bob Halley committed
47
isc_result_t
48 49
dns_view_create(isc_mem_t *mctx, dns_rdataclass_t rdclass,
		const char *name, dns_view_t **viewp)
Bob Halley's avatar
add  
Bob Halley committed
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
{
	dns_view_t *view;
	isc_result_t result;

	/*
	 * Create a view.
	 */

	REQUIRE(name != NULL);
	REQUIRE(viewp != NULL && *viewp == NULL);

	view = isc_mem_get(mctx, sizeof *view);
	if (view == NULL)
		return (ISC_R_NOMEMORY);
	view->name = isc_mem_strdup(mctx, name);
	if (view->name == NULL) {
		result = ISC_R_NOMEMORY;
		goto cleanup_view;
	}
	result = isc_mutex_init(&view->lock);
	if (result != ISC_R_SUCCESS) {
		UNEXPECTED_ERROR(__FILE__, __LINE__,
				 "isc_mutex_init() failed: %s",
				 isc_result_totext(result));
		result = ISC_R_UNEXPECTED;
		goto cleanup_name;
	}
77 78
	view->zonetable = NULL;
	result = dns_zt_create(mctx, rdclass, &view->zonetable);
Bob Halley's avatar
add  
Bob Halley committed
79 80
	if (result != ISC_R_SUCCESS) {
		UNEXPECTED_ERROR(__FILE__, __LINE__,
81
				 "dns_zt_create() failed: %s",
Bob Halley's avatar
add  
Bob Halley committed
82 83 84 85
				 isc_result_totext(result));
		result = ISC_R_UNEXPECTED;
		goto cleanup_mutex;
	}
Bob Halley's avatar
Bob Halley committed
86 87 88 89 90 91 92
	view->secroots = NULL;
	result = dns_rbt_create(mctx, NULL, NULL, &view->secroots);
	if (result != ISC_R_SUCCESS) {
		UNEXPECTED_ERROR(__FILE__, __LINE__,
				 "dns_rbt_create() failed: %s",
				 isc_result_totext(result));
		result = ISC_R_UNEXPECTED;
93
		goto cleanup_zt;
Bob Halley's avatar
Bob Halley committed
94
	}
95

96
	view->cache = NULL;
Bob Halley's avatar
Bob Halley committed
97
	view->cachedb = NULL;
Bob Halley's avatar
Bob Halley committed
98
	view->hints = NULL;
Bob Halley's avatar
add  
Bob Halley committed
99
	view->resolver = NULL;
Bob Halley's avatar
add adb  
Bob Halley committed
100
	view->adb = NULL;
Bob Halley's avatar
add  
Bob Halley committed
101 102
	view->mctx = mctx;
	view->rdclass = rdclass;
Bob Halley's avatar
Bob Halley committed
103
	view->frozen = ISC_FALSE;
104
	view->task = NULL;
Bob Halley's avatar
add  
Bob Halley committed
105
	view->references = 1;
106
	view->attributes = (DNS_VIEWATTR_RESSHUTDOWN|DNS_VIEWATTR_ADBSHUTDOWN);
Michael Graff's avatar
Michael Graff committed
107
	ISC_LINK_INIT(view, link);
108 109 110 111 112 113
	ISC_EVENT_INIT(&view->resevent, sizeof view->resevent, 0, NULL,
		       DNS_EVENT_VIEWRESSHUTDOWN, resolver_shutdown,
		       view, NULL, NULL, NULL);
	ISC_EVENT_INIT(&view->adbevent, sizeof view->adbevent, 0, NULL,
		       DNS_EVENT_VIEWADBSHUTDOWN, adb_shutdown,
		       view, NULL, NULL, NULL);
Bob Halley's avatar
add  
Bob Halley committed
114 115 116 117 118 119
	view->magic = DNS_VIEW_MAGIC;
	
	*viewp = view;

	return (ISC_R_SUCCESS);

120 121
 cleanup_zt:
	dns_zt_detach(&view->zonetable);
Bob Halley's avatar
Bob Halley committed
122

Bob Halley's avatar
add  
Bob Halley committed
123 124 125 126 127 128 129 130 131 132 133 134 135 136
 cleanup_mutex:
	isc_mutex_destroy(&view->lock);

 cleanup_name:
	isc_mem_free(mctx, view->name);

 cleanup_view:
	isc_mem_put(mctx, view, sizeof *view);

	return (result);
}

void
dns_view_attach(dns_view_t *source, dns_view_t **targetp) {
Bob Halley's avatar
Bob Halley committed
137 138 139 140 141

	/*
	 * Attach '*targetp' to 'source'.
	 */

Bob Halley's avatar
add  
Bob Halley committed
142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
	REQUIRE(DNS_VIEW_VALID(source));
	REQUIRE(targetp != NULL && *targetp == NULL);

	LOCK(&source->lock);

	INSIST(source->references > 0);
	source->references++;
	INSIST(source->references != 0);

	UNLOCK(&source->lock);

	*targetp = source;
}

static inline void
destroy(dns_view_t *view) {
	REQUIRE(!ISC_LINK_LINKED(view, link));
159 160 161
	REQUIRE(view->references == 0);
	REQUIRE(RESSHUTDOWN(view));
	REQUIRE(ADBSHUTDOWN(view));
Bob Halley's avatar
add  
Bob Halley committed
162

Bob Halley's avatar
add adb  
Bob Halley committed
163 164
	if (view->adb != NULL)
		dns_adb_detach(&view->adb);
Bob Halley's avatar
add  
Bob Halley committed
165 166
	if (view->resolver != NULL)
		dns_resolver_detach(&view->resolver);
167 168
	if (view->task != NULL)
		isc_task_detach(&view->task);
Bob Halley's avatar
Bob Halley committed
169 170
	if (view->hints != NULL)
		dns_db_detach(&view->hints);
Bob Halley's avatar
Bob Halley committed
171 172
	if (view->cachedb != NULL)
		dns_db_detach(&view->cachedb);
173 174
	if (view->cache != NULL)
		dns_cache_detach(&view->cache);
175
	dns_zt_detach(&view->zonetable);
Bob Halley's avatar
Bob Halley committed
176
	dns_rbt_destroy(&view->secroots);
Bob Halley's avatar
add  
Bob Halley committed
177 178 179 180 181
	isc_mutex_destroy(&view->lock);
	isc_mem_free(view->mctx, view->name);
	isc_mem_put(view->mctx, view, sizeof *view);
}

182 183 184 185 186 187 188 189 190 191 192 193
static isc_boolean_t
all_done(dns_view_t *view) {
	/*
	 * Caller must be holding the view lock.
	 */

	if (view->references == 0 && RESSHUTDOWN(view) && ADBSHUTDOWN(view))
		return (ISC_TRUE);

	return (ISC_FALSE);
}

Bob Halley's avatar
add  
Bob Halley committed
194 195 196
void
dns_view_detach(dns_view_t **viewp) {
	dns_view_t *view;
197
	isc_boolean_t done = ISC_FALSE;
Bob Halley's avatar
add  
Bob Halley committed
198

Bob Halley's avatar
Bob Halley committed
199 200 201 202
	/*
	 * Detach '*viewp' from its view.
	 */

Bob Halley's avatar
add  
Bob Halley committed
203 204 205 206 207 208 209 210
	REQUIRE(viewp != NULL);
	view = *viewp;
	REQUIRE(DNS_VIEW_VALID(view));

	LOCK(&view->lock);

	INSIST(view->references > 0);
	view->references--;
211 212 213 214 215 216 217
	if (view->references == 0) {
		if (!RESSHUTDOWN(view))
			dns_resolver_shutdown(view->resolver);
		if (!ADBSHUTDOWN(view))
			dns_adb_shutdown(view->adb);
		done = all_done(view);
	}
Bob Halley's avatar
add  
Bob Halley committed
218 219 220 221
	UNLOCK(&view->lock);

	*viewp = NULL;

222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266
	if (done)
		destroy(view);
}

static void
resolver_shutdown(isc_task_t *task, isc_event_t *event) {
	dns_view_t *view = event->arg;
	isc_boolean_t done;
	
	REQUIRE(event->type == DNS_EVENT_VIEWRESSHUTDOWN);
	REQUIRE(DNS_VIEW_VALID(view));
	REQUIRE(view->task == task);

	LOCK(&view->lock);

	view->attributes |= DNS_VIEWATTR_RESSHUTDOWN;
	done = all_done(view);

	UNLOCK(&view->lock);

	isc_event_free(&event);

	if (done)
		destroy(view);
}

static void
adb_shutdown(isc_task_t *task, isc_event_t *event) {
	dns_view_t *view = event->arg;
	isc_boolean_t done;
	
	REQUIRE(event->type == DNS_EVENT_VIEWADBSHUTDOWN);
	REQUIRE(DNS_VIEW_VALID(view));
	REQUIRE(view->task == task);

	LOCK(&view->lock);

	view->attributes |= DNS_VIEWATTR_ADBSHUTDOWN;
	done = all_done(view);

	UNLOCK(&view->lock);

	isc_event_free(&event);

	if (done)
Bob Halley's avatar
add  
Bob Halley committed
267 268
		destroy(view);
}
269

Bob Halley's avatar
Bob Halley committed
270
isc_result_t
Bob Halley's avatar
Bob Halley committed
271 272 273 274
dns_view_createresolver(dns_view_t *view,
			isc_taskmgr_t *taskmgr, unsigned int ntasks,
			isc_socketmgr_t *socketmgr,
			isc_timermgr_t *timermgr,
Bob Halley's avatar
Bob Halley committed
275 276
			dns_dispatch_t *dispatch)
{
Bob Halley's avatar
add adb  
Bob Halley committed
277
	isc_result_t result;
278
	isc_event_t *event;
Bob Halley's avatar
add adb  
Bob Halley committed
279

Bob Halley's avatar
Bob Halley committed
280
	/*
Bob Halley's avatar
add adb  
Bob Halley committed
281
	 * Create a resolver and address database for the view.
Bob Halley's avatar
Bob Halley committed
282 283
	 */

284
	REQUIRE(DNS_VIEW_VALID(view));
Bob Halley's avatar
Bob Halley committed
285
	REQUIRE(!view->frozen);
286
	REQUIRE(view->resolver == NULL);
287 288 289 290 291

	result = isc_task_create(taskmgr, view->mctx, 0, &view->task);
	if (result != ISC_R_SUCCESS)
		return (result);

Bob Halley's avatar
add adb  
Bob Halley committed
292 293
	result = dns_resolver_create(view, taskmgr, ntasks, socketmgr,
				     timermgr, dispatch, &view->resolver);
294 295
	if (result != ISC_R_SUCCESS) {
		isc_task_detach(&view->task);
Bob Halley's avatar
add adb  
Bob Halley committed
296
		return (result);
297 298 299 300 301
	}
	event = &view->resevent;
	dns_resolver_whenshutdown(view->resolver, view->task, &event);
	view->attributes &= ~DNS_VIEWATTR_RESSHUTDOWN;

Bob Halley's avatar
add adb  
Bob Halley committed
302 303
	result = dns_adb_create(view->mctx, view, timermgr, taskmgr,
				&view->adb);
304 305 306 307 308 309 310
	if (result != ISC_R_SUCCESS) {
		dns_resolver_shutdown(view->resolver);
		return (result);
	}
	event = &view->adbevent;
	dns_adb_whenshutdown(view->adb, view->task, &event);
	view->attributes &= ~DNS_VIEWATTR_ADBSHUTDOWN;
Bob Halley's avatar
add adb  
Bob Halley committed
311

312
	return (ISC_R_SUCCESS);
313 314 315
}

void
316
dns_view_setcache(dns_view_t *view, dns_cache_t *cache) {
Bob Halley's avatar
Bob Halley committed
317 318

	/*
319
	 * Set the view's cache.
Bob Halley's avatar
Bob Halley committed
320 321
	 */

322
	REQUIRE(DNS_VIEW_VALID(view));
Bob Halley's avatar
Bob Halley committed
323
	REQUIRE(!view->frozen);
324

325
	if (view->cache != NULL) {
Bob Halley's avatar
add adb  
Bob Halley committed
326
		dns_db_detach(&view->cachedb);
327 328 329 330 331
		dns_cache_detach(&view->cache);
	}
	dns_cache_attach(cache, &view->cache);
	dns_cache_attachdb(cache, &view->cachedb);
	INSIST(DNS_DB_VALID(view->cachedb));
332 333
}

Bob Halley's avatar
Bob Halley committed
334 335 336 337 338 339 340 341 342 343 344 345 346 347 348
void
dns_view_sethints(dns_view_t *view, dns_db_t *hints) {

	/*
	 * Set the view's hints database.
	 */

	REQUIRE(DNS_VIEW_VALID(view));
	REQUIRE(!view->frozen);
	REQUIRE(view->hints == NULL);
	REQUIRE(dns_db_iszone(hints));

	dns_db_attach(hints, &view->hints);
}

349
isc_result_t
350
dns_view_addzone(dns_view_t *view, dns_zone_t *zone) {
Bob Halley's avatar
Bob Halley committed
351 352 353
	isc_result_t result;

	/*
354
	 * Add zone 'zone' to 'view'.
Bob Halley's avatar
Bob Halley committed
355 356
	 */

357
	REQUIRE(DNS_VIEW_VALID(view));
Bob Halley's avatar
Bob Halley committed
358 359
	REQUIRE(!view->frozen);

360
	result = dns_zt_mount(view->zonetable, zone);
361

Bob Halley's avatar
Bob Halley committed
362
	return (result);
363 364 365 366
}

void
dns_view_freeze(dns_view_t *view) {
Bob Halley's avatar
Bob Halley committed
367 368 369 370 371 372 373 374
	
	/*
	 * Freeze view.
	 */

	REQUIRE(DNS_VIEW_VALID(view));
	REQUIRE(!view->frozen);

Bob Halley's avatar
Bob Halley committed
375 376
	if (view->resolver != NULL)
		dns_resolver_freeze(view->resolver);
Bob Halley's avatar
Bob Halley committed
377 378 379
	view->frozen = ISC_TRUE;
}

380
isc_result_t
381
dns_view_findzone(dns_view_t *view, dns_name_t *name, dns_zone_t **zonep) {
382 383 384 385
	isc_result_t result;

	REQUIRE(DNS_VIEW_VALID(view));

386
	result = dns_zt_find(view->zonetable, name, NULL, zonep);
387
	if (result == DNS_R_PARTIALMATCH) {
388
		dns_zone_detach(zonep);
389 390
		result = DNS_R_NOTFOUND;
	}
391

392 393 394
	return (result);
}

Bob Halley's avatar
Bob Halley committed
395 396
isc_result_t
dns_view_find(dns_view_t *view, dns_name_t *name, dns_rdatatype_t type,
Bob Halley's avatar
Bob Halley committed
397
	      isc_stdtime_t now, unsigned int options, isc_boolean_t use_hints,
Bob Halley's avatar
Bob Halley committed
398 399 400 401 402 403 404 405
	      dns_rdataset_t *rdataset, dns_rdataset_t *sigrdataset)
{
	isc_result_t result;
	dns_fixedname_t foundname;
	dns_db_t *db;
	dns_dbversion_t *version;
	isc_boolean_t is_zone;
	dns_rdataset_t zrdataset, zsigrdataset;
406
	dns_zone_t *zone;
Bob Halley's avatar
Bob Halley committed
407 408 409 410 411 412

	/*
	 * Find an rdataset whose owner name is 'name', and whose type is
	 * 'type'.
	 */

413
	REQUIRE(DNS_VIEW_VALID(view));
Bob Halley's avatar
Bob Halley committed
414 415 416 417 418 419 420 421 422 423 424 425
	REQUIRE(view->frozen);
	REQUIRE(type != dns_rdatatype_any && type != dns_rdatatype_sig);

	/*
	 * Initialize.
	 */
	dns_rdataset_init(&zrdataset);
	dns_rdataset_init(&zsigrdataset);

	/*
	 * Find a database to answer the query.
	 */
426
	zone = NULL;
Bob Halley's avatar
Bob Halley committed
427
	db = NULL;
428 429 430 431 432 433 434 435
	result = dns_zt_find(view->zonetable, name, NULL, &zone);
	if (result == ISC_R_SUCCESS || result == DNS_R_PARTIALMATCH) {
		result = dns_zone_getdb(zone, &db);
		if (result != DNS_R_SUCCESS && view->cachedb != NULL)
			dns_db_attach(view->cachedb, &db);
		else if (result != DNS_R_SUCCESS)
			goto cleanup;
	} else if (result == ISC_R_NOTFOUND && view->cachedb != NULL)
Bob Halley's avatar
Bob Halley committed
436
		dns_db_attach(view->cachedb, &db);
437
	else
Bob Halley's avatar
Bob Halley committed
438 439 440
		goto cleanup;

	is_zone = dns_db_iszone(db);
441

Bob Halley's avatar
Bob Halley committed
442 443 444 445 446 447 448 449 450 451
 db_find:
	/*
	 * Now look for an answer in the database.
	 */
	dns_fixedname_init(&foundname);
	result = dns_db_find(db, name, NULL, type, options,
			     now, NULL, dns_fixedname_name(&foundname),
			     rdataset, sigrdataset);

	if (result == DNS_R_DELEGATION ||
Bob Halley's avatar
Bob Halley committed
452
	    result == DNS_R_NOTFOUND) {
Bob Halley's avatar
Bob Halley committed
453 454
		if (rdataset->methods != NULL)
			dns_rdataset_disassociate(rdataset);
Bob Halley's avatar
Bob Halley committed
455
		if (sigrdataset != NULL && sigrdataset->methods != NULL)
Bob Halley's avatar
Bob Halley committed
456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475
			dns_rdataset_disassociate(sigrdataset);
		if (is_zone) {
			if (view->cachedb != NULL) {
				/*
				 * Either the answer is in the cache, or we
				 * don't know it.
				 */
				is_zone = ISC_FALSE;
				version = NULL;
				dns_db_detach(&db);
				dns_db_attach(view->cachedb, &db);
				goto db_find;
			}
		} else {
			/*
			 * We don't have the data in the cache.  If we've got
			 * glue from the zone, use it.
			 */
			if (zrdataset.methods != NULL) {
				dns_rdataset_clone(&zrdataset, rdataset);
Bob Halley's avatar
Bob Halley committed
476 477
				if (sigrdataset != NULL &&
				    zsigrdataset.methods != NULL)
Bob Halley's avatar
Bob Halley committed
478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497
					dns_rdataset_clone(&zsigrdataset,
							   sigrdataset);
				result = DNS_R_GLUE;
				goto cleanup;
			}
		}
		/*
		 * We don't know the answer.
		 */
		result = DNS_R_NOTFOUND;
	} else if (result == DNS_R_GLUE) {
		if (view->cachedb != NULL) {
			/*
			 * We found an answer, but the cache may be better.
			 * Remember what we've got and go look in the cache.
			 */
			is_zone = ISC_FALSE;
			version = NULL;
			dns_rdataset_clone(rdataset, &zrdataset);
			dns_rdataset_disassociate(rdataset);
Bob Halley's avatar
Bob Halley committed
498 499
			if (sigrdataset != NULL &&
			    sigrdataset->methods != NULL) {
Bob Halley's avatar
Bob Halley committed
500 501 502 503 504 505 506 507 508 509 510
				dns_rdataset_clone(sigrdataset, &zsigrdataset);
				dns_rdataset_disassociate(sigrdataset);
			}
			dns_db_detach(&db);
			dns_db_attach(view->cachedb, &db);
			goto db_find;
		}
		/*
		 * Otherwise, the glue is the best answer.
		 */
		result = ISC_R_SUCCESS;
Bob Halley's avatar
Bob Halley committed
511 512 513 514 515
	}

	if (result == DNS_R_NOTFOUND && use_hints && view->hints != NULL) {
		if (rdataset->methods != NULL)
			dns_rdataset_disassociate(rdataset);
Bob Halley's avatar
Bob Halley committed
516
		if (sigrdataset != NULL && sigrdataset->methods != NULL)
Bob Halley's avatar
Bob Halley committed
517 518 519 520 521 522 523 524
			dns_rdataset_disassociate(sigrdataset);
		dns_fixedname_init(&foundname);
		result = dns_db_find(view->hints, name, NULL, type, options,
				     now, NULL, dns_fixedname_name(&foundname),
				     rdataset, sigrdataset);
		if (result == ISC_R_SUCCESS || result == DNS_R_GLUE)
			result = DNS_R_HINT;
	}
Bob Halley's avatar
Bob Halley committed
525 526

 cleanup:
Bob Halley's avatar
Bob Halley committed
527 528
	if (result != ISC_R_SUCCESS &&
	    result != DNS_R_GLUE &&
529 530 531
	    result != DNS_R_HINT &&
	    result != DNS_R_NCACHENXDOMAIN &&
	    result != DNS_R_NCACHENXRRSET)
Bob Halley's avatar
Bob Halley committed
532 533
		result = DNS_R_NOTFOUND;

Bob Halley's avatar
Bob Halley committed
534 535 536 537 538 539 540
	if (zrdataset.methods != NULL) {
		dns_rdataset_disassociate(&zrdataset);
		if (zsigrdataset.methods != NULL)
			dns_rdataset_disassociate(&zsigrdataset);
	}
	if (db != NULL)
		dns_db_detach(&db);
541 542
	if (zone != NULL)
		dns_zone_detach(&zone);
Bob Halley's avatar
Bob Halley committed
543 544

	return (result);
545
}
546

Bob Halley's avatar
Bob Halley committed
547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721
isc_result_t
dns_view_findzonecut(dns_view_t *view, dns_name_t *name, dns_name_t *fname,
		     isc_stdtime_t now, unsigned int options,
		     isc_boolean_t use_hints,
		     dns_rdataset_t *rdataset, dns_rdataset_t *sigrdataset)
{
	isc_result_t result;
	dns_db_t *db;
	isc_boolean_t is_zone, use_zone, try_hints;
	dns_zone_t *zone;
	dns_name_t *zfname;
	dns_rdataset_t zrdataset, zsigrdataset;
	dns_fixedname_t zfixedname;

	/*
	 * Find the best known zonecut containing 'name'.
	 */

	REQUIRE(DNS_VIEW_VALID(view));
	REQUIRE(view->frozen);

	db = NULL;
	zone = NULL;
	use_zone = ISC_FALSE;
	try_hints = ISC_FALSE;
	zfname = NULL;

	/*
	 * Initialize.
	 */
	dns_fixedname_init(&zfixedname);
	dns_rdataset_init(&zrdataset);
	dns_rdataset_init(&zsigrdataset);

	/*
	 * Find the right database.
	 */
	result = dns_zt_find(view->zonetable, name, NULL, &zone);
	if (result == DNS_R_SUCCESS || result == DNS_R_PARTIALMATCH)
		result = dns_zone_getdb(zone, &db);
	if (result == ISC_R_NOTFOUND) {
		/*
		 * We're not directly authoritative for this query name, nor
		 * is it a subdomain of any zone for which we're
		 * authoritative.
		 */
		if (view->cachedb != NULL) {
			/*
			 * We have a cache; try it.
			 */
			dns_db_attach(view->cachedb, &db);
		} else {
			/*
			 * Maybe we have hints...
			 */
			try_hints = ISC_TRUE;
			goto finish;
		}
	} else if (result != ISC_R_SUCCESS) {
		/*
		 * Something is broken.
		 */
		goto cleanup;
	}
	is_zone = dns_db_iszone(db);

 db_find:
	/*
	 * Look for the zonecut.
	 */
	if (is_zone) {
		result = dns_db_find(db, name, NULL, dns_rdatatype_ns, options,
				     now, NULL, fname, rdataset, sigrdataset);
		if (result == DNS_R_DELEGATION)
			result = ISC_R_SUCCESS;
		else if (result != ISC_R_SUCCESS)
			goto cleanup;
		if (view->cachedb != NULL && db != view->hints) {
			/*
			 * We found an answer, but the cache may be better.
			 */
			zfname = dns_fixedname_name(&zfixedname);
			result = dns_name_concatenate(fname, NULL, zfname,
						      NULL);
			if (result != ISC_R_SUCCESS)
				goto cleanup;
			dns_rdataset_clone(rdataset, &zrdataset);
			dns_rdataset_disassociate(rdataset);
			if (sigrdataset != NULL &&
			    sigrdataset->methods != NULL) {
				dns_rdataset_clone(sigrdataset, &zsigrdataset);
				dns_rdataset_disassociate(sigrdataset);
			}
			dns_db_detach(&db);
			dns_db_attach(view->cachedb, &db);
			is_zone = ISC_FALSE;
			goto db_find;
		}
	} else {
		result = dns_db_findzonecut(db, name, options, now, NULL,
					    fname, rdataset, sigrdataset);
		if (result == ISC_R_SUCCESS) {
			if (zfname != NULL &&
			    !dns_name_issubdomain(fname, zfname)) {
				/*
				 * We found a zonecut in the cache, but our
				 * zone delegation is better.
				 */
				use_zone = ISC_TRUE;
			}
		} else if (result == ISC_R_NOTFOUND) {
			if (zfname != NULL) {
				/*
				 * We didn't find anything in the cache, but we
				 * have a zone delegation, so use it.
				 */
				use_zone = ISC_TRUE;
			} else {
				/*
				 * Maybe we have hints...
				 */
				try_hints = ISC_TRUE;
			}
		} else {
			/*
			 * Something bad happened.
			 */
			goto cleanup;
		}
	}

 finish:
	if (use_zone) {
		if (rdataset->methods != NULL) {
			dns_rdataset_disassociate(rdataset);
			if (sigrdataset != NULL &&
			    sigrdataset->methods != NULL)
				dns_rdataset_disassociate(sigrdataset);
		}
		result = dns_name_concatenate(zfname, NULL, fname, NULL);
		if (result != ISC_R_SUCCESS)
			goto cleanup;
		dns_rdataset_clone(&zrdataset, rdataset);
		if (sigrdataset != NULL && zrdataset.methods != NULL)
			dns_rdataset_clone(&zsigrdataset, sigrdataset);
	} else if (try_hints && use_hints && view->hints != NULL) {
		/*
		 * We've found nothing so far, but we have hints.
		 */
		result = dns_db_find(view->hints, dns_rootname, NULL,
				     dns_rdatatype_ns, 0, now, NULL, fname,
				     rdataset, NULL);
		if (result != ISC_R_SUCCESS) {
			/*
			 * We can't even find the hints for the root
			 * nameservers!
			 */
			result = ISC_R_NOTFOUND;
		}
	}

 cleanup:
	if (zrdataset.methods != NULL) {
		dns_rdataset_disassociate(&zrdataset);
		if (zsigrdataset.methods != NULL)
			dns_rdataset_disassociate(&zsigrdataset);
	}
	if (db != NULL)
		dns_db_detach(&db);
	if (zone != NULL)
		dns_zone_detach(&zone);

	return (result);
}

722 723 724 725
isc_result_t
dns_viewlist_find(dns_viewlist_t *list, const char *name,
		  dns_rdataclass_t rdclass, dns_view_t **viewp)
{
726 727 728 729
	dns_view_t *view;

	REQUIRE(list != NULL);

730 731 732
	for (view = ISC_LIST_HEAD(*list);
	     view != NULL;
	     view = ISC_LIST_NEXT(view, link)) {
733 734 735
		if (strcmp(view->name, name) == 0 && view->rdclass == rdclass)
			break;
	}
736 737 738 739 740 741
	if (view == NULL)
		return (ISC_R_NOTFOUND);

	dns_view_attach(view, viewp);

	return (ISC_R_SUCCESS);
742
}
Mark Andrews's avatar
Mark Andrews committed
743 744 745 746 747 748 749 750

void
dns_view_load(dns_view_t *view) {

	REQUIRE(DNS_VIEW_VALID(view));

	dns_zt_load(view->zonetable);
}