mirror of
https://github.com/isc-projects/bind9.git
synced 2026-03-11 02:30:44 -04:00
f5c204ac3e
Instead of relying on unreliable order of execution of the library constructors and destructors, move them to individual binaries. The advantage is that the execution time and order will remain constant and will not depend on the dynamic load dependency solver. This requires more work, but that was mitigated by a simple requirement, any executable using libisc and libdns, must include <isc/lib.h> and <dns/lib.h> respectively (in this particular order). In turn, these two headers must not be included from within any library as they contain inlined functions marked with constructor/destructor attributes.
282 lines
6.8 KiB
C
282 lines
6.8 KiB
C
/*
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* SPDX-License-Identifier: MPL-2.0
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
*/
|
|
|
|
#include <inttypes.h>
|
|
#include <sched.h> /* IWYU pragma: keep */
|
|
#include <setjmp.h>
|
|
#include <stdarg.h>
|
|
#include <stdbool.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#define UNIT_TESTING
|
|
#include <cmocka.h>
|
|
|
|
#include <isc/atomic.h>
|
|
#include <isc/buffer.h>
|
|
#include <isc/lib.h>
|
|
#include <isc/loop.h>
|
|
#include <isc/timer.h>
|
|
#include <isc/urcu.h>
|
|
#include <isc/util.h>
|
|
|
|
#include <dns/db.h>
|
|
#include <dns/lib.h>
|
|
#include <dns/name.h>
|
|
#include <dns/view.h>
|
|
#include <dns/zone.h>
|
|
#include <dns/zt.h>
|
|
|
|
#include <tests/dns.h>
|
|
|
|
static dns_db_t *db = NULL;
|
|
static FILE *zonefile, *origfile;
|
|
static dns_view_t *view = NULL;
|
|
|
|
static isc_result_t
|
|
count_zone(dns_zone_t *zone, void *uap) {
|
|
int *nzones = (int *)uap;
|
|
|
|
UNUSED(zone);
|
|
|
|
*nzones += 1;
|
|
return ISC_R_SUCCESS;
|
|
}
|
|
|
|
/* apply a function to a zone table */
|
|
ISC_LOOP_TEST_IMPL(apply) {
|
|
isc_result_t result;
|
|
dns_zone_t *zone = NULL;
|
|
dns_zt_t *zt = NULL;
|
|
int nzones = 0;
|
|
|
|
result = dns_test_makezone("foo", &zone, NULL, true);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
view = dns_zone_getview(zone);
|
|
rcu_read_lock();
|
|
zt = rcu_dereference(view->zonetable);
|
|
rcu_read_unlock();
|
|
|
|
assert_non_null(zt);
|
|
|
|
assert_int_equal(nzones, 0);
|
|
result = dns_view_apply(view, false, NULL, count_zone, &nzones);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(nzones, 1);
|
|
|
|
/* These steps are necessary so the zone can be detached properly */
|
|
dns_test_setupzonemgr();
|
|
result = dns_test_managezone(zone);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
dns_test_releasezone(zone);
|
|
dns_test_closezonemgr();
|
|
|
|
/* The view was left attached in dns_test_makezone() */
|
|
dns_view_detach(&view);
|
|
dns_zone_detach(&zone);
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
static isc_result_t
|
|
load_done_last(void *uap) {
|
|
dns_zone_t *zone = uap;
|
|
isc_result_t result;
|
|
|
|
/* The zone should now be loaded; test it */
|
|
result = dns_zone_getdb(zone, &db);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
assert_non_null(db);
|
|
if (db != NULL) {
|
|
dns_db_detach(&db);
|
|
}
|
|
|
|
dns_test_releasezone(zone);
|
|
dns_test_closezonemgr();
|
|
|
|
dns_zone_detach(&zone);
|
|
dns_view_detach(&view);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
|
|
return ISC_R_SUCCESS;
|
|
}
|
|
|
|
static isc_result_t
|
|
load_done_new_only(void *uap) {
|
|
dns_zone_t *zone = uap;
|
|
isc_result_t result;
|
|
|
|
/* The zone should now be loaded; test it */
|
|
result = dns_zone_getdb(zone, &db);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
dns_db_detach(&db);
|
|
|
|
dns_zone_asyncload(zone, true, load_done_last, zone);
|
|
|
|
return ISC_R_SUCCESS;
|
|
}
|
|
|
|
static isc_result_t
|
|
load_done_first(void *uap) {
|
|
dns_zone_t *zone = uap;
|
|
isc_result_t result;
|
|
|
|
/* The zone should now be loaded; test it */
|
|
result = dns_zone_getdb(zone, &db);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
dns_db_detach(&db);
|
|
|
|
/*
|
|
* Add something to zone file, reload zone with newonly - it should
|
|
* not be reloaded.
|
|
*/
|
|
fprintf(zonefile, "\nb in b 1.2.3.4\n");
|
|
fflush(zonefile);
|
|
fclose(zonefile);
|
|
|
|
dns_zone_asyncload(zone, true, load_done_new_only, zone);
|
|
|
|
return ISC_R_SUCCESS;
|
|
}
|
|
|
|
/* asynchronous zone load */
|
|
ISC_LOOP_TEST_IMPL(asyncload_zone) {
|
|
isc_result_t result;
|
|
int n;
|
|
dns_zone_t *zone = NULL;
|
|
dns_zt_t *zt = NULL;
|
|
char buf[4096];
|
|
|
|
result = dns_test_makezone("foo", &zone, NULL, true);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_test_setupzonemgr();
|
|
result = dns_test_managezone(zone);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
view = dns_zone_getview(zone);
|
|
rcu_read_lock();
|
|
zt = rcu_dereference(view->zonetable);
|
|
rcu_read_unlock();
|
|
assert_non_null(zt);
|
|
|
|
assert_false(dns__zone_loadpending(zone));
|
|
zonefile = fopen("./zone.data", "wb");
|
|
assert_non_null(zonefile);
|
|
origfile = fopen(TESTS_DIR "/testdata/zt/zone1.db", "r+b");
|
|
assert_non_null(origfile);
|
|
n = fread(buf, 1, 4096, origfile);
|
|
fclose(origfile);
|
|
fwrite(buf, 1, n, zonefile);
|
|
fflush(zonefile);
|
|
|
|
dns_zone_setfile(zone, "./zone.data", dns_masterformat_text,
|
|
&dns_master_style_default);
|
|
|
|
dns_zone_asyncload(zone, false, load_done_first, zone);
|
|
}
|
|
|
|
dns_zone_t *zone1 = NULL, *zone2 = NULL, *zone3 = NULL;
|
|
|
|
static isc_result_t
|
|
all_done(void *arg ISC_ATTR_UNUSED) {
|
|
isc_result_t result;
|
|
|
|
/* Both zones should now be loaded; test them */
|
|
result = dns_zone_getdb(zone1, &db);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_non_null(db);
|
|
if (db != NULL) {
|
|
dns_db_detach(&db);
|
|
}
|
|
|
|
result = dns_zone_getdb(zone2, &db);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_non_null(db);
|
|
if (db != NULL) {
|
|
dns_db_detach(&db);
|
|
}
|
|
|
|
dns_test_releasezone(zone3);
|
|
dns_test_releasezone(zone2);
|
|
dns_test_releasezone(zone1);
|
|
dns_test_closezonemgr();
|
|
|
|
dns_zone_detach(&zone1);
|
|
dns_zone_detach(&zone2);
|
|
dns_zone_detach(&zone3);
|
|
dns_view_detach(&view);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
return ISC_R_SUCCESS;
|
|
}
|
|
|
|
/* asynchronous zone table load */
|
|
ISC_LOOP_TEST_IMPL(asyncload_zt) {
|
|
isc_result_t result;
|
|
dns_zt_t *zt = NULL;
|
|
atomic_bool done;
|
|
|
|
atomic_init(&done, false);
|
|
|
|
result = dns_test_makezone("foo", &zone1, NULL, true);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
dns_zone_setfile(zone1, TESTS_DIR "/testdata/zt/zone1.db",
|
|
dns_masterformat_text, &dns_master_style_default);
|
|
view = dns_zone_getview(zone1);
|
|
|
|
result = dns_test_makezone("bar", &zone2, view, false);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
dns_zone_setfile(zone2, TESTS_DIR "/testdata/zt/zone1.db",
|
|
dns_masterformat_text, &dns_master_style_default);
|
|
|
|
/* This one will fail to load */
|
|
result = dns_test_makezone("fake", &zone3, view, false);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
dns_zone_setfile(zone3, TESTS_DIR "/testdata/zt/nonexistent.db",
|
|
dns_masterformat_text, &dns_master_style_default);
|
|
|
|
rcu_read_lock();
|
|
zt = rcu_dereference(view->zonetable);
|
|
rcu_read_unlock();
|
|
assert_non_null(zt);
|
|
|
|
dns_test_setupzonemgr();
|
|
result = dns_test_managezone(zone1);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
result = dns_test_managezone(zone2);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
result = dns_test_managezone(zone3);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
assert_false(dns__zone_loadpending(zone1));
|
|
assert_false(dns__zone_loadpending(zone2));
|
|
assert_false(atomic_load(&done));
|
|
|
|
rcu_read_lock();
|
|
zt = rcu_dereference(view->zonetable);
|
|
dns_zt_asyncload(zt, false, all_done, NULL);
|
|
rcu_read_unlock();
|
|
}
|
|
|
|
ISC_TEST_LIST_START
|
|
ISC_TEST_ENTRY_CUSTOM(apply, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(asyncload_zone, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(asyncload_zt, setup_managers, teardown_managers)
|
|
ISC_TEST_LIST_END
|
|
|
|
ISC_TEST_MAIN
|