mirror of
https://gitlab.isc.org/isc-projects/bind9
synced 2025-08-30 14:07:59 +00:00
Tests loading of zones from a conf file via dns_zone_callback().
This commit is contained in:
Mark Andrews
parent
2edd9e49c7
commit
db0a8db17b
284
bin/tests/zone2_test.c
Normal file
284
bin/tests/zone2_test.c
Normal file
@ -0,0 +1,284 @@
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include <config.h>
|
||||
|
||||
|
||||
#include <isc/app.h>
|
||||
#include <isc/error.h>
|
||||
#include <isc/mem.h>
|
||||
|
||||
#include <dns/confparser.h>
|
||||
#include <dns/view.h>
|
||||
#include <dns/zone.h>
|
||||
#include <dns/journal.h>
|
||||
|
||||
#define ERRRET(result, function) \
|
||||
do { \
|
||||
if (result != DNS_R_SUCCESS) { \
|
||||
fprintf(stdout, "%s() returned %s\n", \
|
||||
function, dns_result_totext(result)); \
|
||||
return; \
|
||||
} \
|
||||
} while (0)
|
||||
|
||||
#define ERRCONT(result, function) \
|
||||
if (result != DNS_R_SUCCESS) { \
|
||||
fprintf(stdout, "%s() returned %s\n", \
|
||||
function, dns_result_totext(result)); \
|
||||
continue; \
|
||||
} else \
|
||||
(void)NULL
|
||||
static void
|
||||
print_rdataset(dns_name_t *name, dns_rdataset_t *rdataset) {
|
||||
isc_buffer_t text;
|
||||
char t[1000];
|
||||
dns_result_t result;
|
||||
isc_region_t r;
|
||||
|
||||
isc_buffer_init(&text, t, sizeof t, ISC_BUFFERTYPE_TEXT);
|
||||
result = dns_rdataset_totext(rdataset, name, ISC_FALSE, ISC_FALSE,
|
||||
&text);
|
||||
isc_buffer_used(&text, &r);
|
||||
if (result == DNS_R_SUCCESS)
|
||||
printf("%.*s", (int)r.length, (char *)r.base);
|
||||
else
|
||||
printf("%s\n", dns_result_totext(result));
|
||||
}
|
||||
|
||||
static void
|
||||
query(dns_view_t *view) {
|
||||
char buf[1024];
|
||||
dns_fixedname_t name;
|
||||
char *s;
|
||||
isc_buffer_t buffer;
|
||||
dns_result_t result;
|
||||
dns_rdataset_t rdataset;
|
||||
dns_rdataset_t sigset;
|
||||
fd_set rfdset;
|
||||
int reload = 0;
|
||||
|
||||
|
||||
dns_rdataset_init(&rdataset);
|
||||
dns_rdataset_init(&sigset);
|
||||
|
||||
do {
|
||||
|
||||
fprintf(stdout, "zone_test ");
|
||||
fflush(stdout);
|
||||
FD_ZERO(&rfdset);
|
||||
FD_SET(0, &rfdset);
|
||||
select(1, &rfdset, NULL, NULL, NULL);
|
||||
if (fgets(buf, sizeof buf, stdin) == NULL) {
|
||||
fprintf(stdout, "\n");
|
||||
break;
|
||||
}
|
||||
buf[sizeof(buf) - 1] = '\0';
|
||||
|
||||
s = strchr(buf, '\n');
|
||||
if (s != NULL)
|
||||
*s = '\0';
|
||||
s = strchr(buf, '\r');
|
||||
if (s != NULL)
|
||||
*s = '\0';
|
||||
if (strlen(buf) == 0) {
|
||||
reload = 0;
|
||||
continue;
|
||||
}
|
||||
if (strcasecmp(buf, "exit") == 0 ||
|
||||
strcasecmp(buf, "quit") == 0)
|
||||
break;
|
||||
if (strcasecmp(buf, "reload") == 0) {
|
||||
reload = 1;
|
||||
continue;
|
||||
}
|
||||
if (strcasecmp(buf, "journal") == 0) {
|
||||
dns_journal_print(view->mctx, "dv.isc.org.ixfr");
|
||||
reload = 0;
|
||||
continue;
|
||||
}
|
||||
|
||||
dns_fixedname_init(&name);
|
||||
isc_buffer_init(&buffer, buf, strlen(buf), ISC_BUFFERTYPE_TEXT);
|
||||
isc_buffer_add(&buffer, strlen(buf));
|
||||
result = dns_name_fromtext(dns_fixedname_name(&name),
|
||||
&buffer, dns_rootname, ISC_FALSE, NULL);
|
||||
ERRCONT(result, "dns_name_fromtext");
|
||||
|
||||
if (reload) {
|
||||
dns_zone_t *zone = NULL;
|
||||
result = dns_zt_find(view->zonetable,
|
||||
dns_fixedname_name(&name), NULL,
|
||||
&zone);
|
||||
if (result != DNS_R_SUCCESS) {
|
||||
if (result == DNS_R_PARTIALMATCH)
|
||||
dns_zone_detach(&zone);
|
||||
reload = 0;
|
||||
continue;
|
||||
}
|
||||
result = dns_zone_load(zone);
|
||||
fprintf(stdout, "%s() returned %s\n", "dns_zone_reload",
|
||||
dns_result_totext(result));
|
||||
reload = 0;
|
||||
dns_zone_detach(&zone);
|
||||
} else {
|
||||
result = dns_view_find(view, dns_fixedname_name(&name),
|
||||
dns_rdatatype_a, 0, 0,
|
||||
ISC_FALSE, &rdataset, &sigset);
|
||||
fprintf(stdout, "%s() returned %s\n", "dns_view_find",
|
||||
dns_result_totext(result));
|
||||
switch (result) {
|
||||
case DNS_R_SUCCESS:
|
||||
print_rdataset(dns_fixedname_name(&name),
|
||||
&rdataset);
|
||||
break;
|
||||
default:
|
||||
continue;
|
||||
}
|
||||
dns_rdataset_disassociate(&rdataset);
|
||||
}
|
||||
} while (1);
|
||||
dns_rdataset_invalidate(&rdataset);
|
||||
}
|
||||
|
||||
int
|
||||
main(int argc, char **argv) {
|
||||
char *conf = "named.conf";
|
||||
isc_mem_t *mctx = NULL;
|
||||
dns_c_ctx_t *configctx = NULL;
|
||||
dns_view_t *view1 = NULL;
|
||||
dns_view_t *view2 = NULL;
|
||||
dns_view_t *view = NULL;
|
||||
dns_c_cbks_t cbks;
|
||||
int quiet = 0, stats = 0;
|
||||
dns_result_t result;
|
||||
int c;
|
||||
char *dir;
|
||||
dns_zone_callbackarg_t cba;
|
||||
|
||||
while ((c = getopt(argc, argv, "c:qs")) != EOF) {
|
||||
switch (c) {
|
||||
case 'c':
|
||||
conf = optarg;
|
||||
break;
|
||||
case 'q':
|
||||
quiet = 1;
|
||||
break;
|
||||
case 's':
|
||||
stats = 1;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
RUNTIME_CHECK(isc_app_start() == ISC_R_SUCCESS);
|
||||
RUNTIME_CHECK(isc_mem_create(0, 0, &mctx) == ISC_R_SUCCESS);
|
||||
#if 0
|
||||
RUNTIME_CHECK(isc_taskmgr_create(mctx, 2, 0, &manager) ==
|
||||
ISC_R_SUCCESS);
|
||||
#endif
|
||||
|
||||
/*
|
||||
RUNTIME_CHECK(dns_view_create(mctx, dns_rdataclass_in,
|
||||
"default/IN", &view1) == DNS_R_SUCCESS);
|
||||
RUNTIME_CHECK(dns_view_create(mctx, dns_rdataclass_in,
|
||||
"default/IN", &view2) == DNS_R_SUCCESS);
|
||||
*/
|
||||
|
||||
cba.mctx = mctx;
|
||||
ISC_LIST_INIT(cba.oldviews);
|
||||
ISC_LIST_INIT(cba.newviews);
|
||||
|
||||
cbks.zonecbk = dns_zone_callback;
|
||||
cbks.zonecbkuap = &cba;
|
||||
cbks.optscbk = NULL;
|
||||
cbks.optscbkuap = NULL;
|
||||
|
||||
result = dns_c_parse_namedconf(NULL, conf, mctx, &configctx, &cbks);
|
||||
fprintf(stdout, "%s() returned %s\n", "dns_c_parse_namedconf",
|
||||
dns_result_totext(result));
|
||||
if (configctx != NULL)
|
||||
dns_c_ctx_delete(NULL, &configctx);
|
||||
|
||||
view = ISC_LIST_HEAD(cba.newviews);
|
||||
|
||||
while (view != NULL) {
|
||||
dns_zt_print(view->zonetable);
|
||||
view = ISC_LIST_NEXT(view, link);
|
||||
}
|
||||
|
||||
/* mv new ->old */
|
||||
cba.oldviews = cba.newviews;
|
||||
ISC_LIST_INIT(cba.newviews);
|
||||
|
||||
/*
|
||||
view = ISC_LIST_HEAD(cba.newviews);
|
||||
while (view != NULL) {
|
||||
dns_view_t *next;
|
||||
next = ISC_LIST_NEXT(view, link);
|
||||
ISC_LIST_UNLINK(cba.newviews, view, link);
|
||||
ISC_LIST_APPEND(cba.oldviews, view, link);
|
||||
view = next;
|
||||
}
|
||||
*/
|
||||
|
||||
result = dns_c_parse_namedconf(NULL, conf, mctx, &configctx, &cbks);
|
||||
fprintf(stdout, "%s() returned %s\n", "dns_c_parse_namedconf",
|
||||
dns_result_totext(result));
|
||||
if (result == DNS_R_SUCCESS) {
|
||||
result = dns_c_ctx_getdirectory(NULL, configctx, &dir);
|
||||
if (result == DNS_R_SUCCESS)
|
||||
chdir(dir);
|
||||
view = ISC_LIST_HEAD(cba.newviews);
|
||||
while (view != NULL) {
|
||||
dns_zt_print(view->zonetable);
|
||||
dns_zt_load(view->zonetable);
|
||||
dns_view_freeze(view);
|
||||
view = ISC_LIST_NEXT(view, link);
|
||||
}
|
||||
view = ISC_LIST_HEAD(cba.newviews);
|
||||
query(view);
|
||||
} else {
|
||||
view = ISC_LIST_HEAD(cba.oldviews);
|
||||
while (view != NULL) {
|
||||
dns_view_t *next;
|
||||
next = ISC_LIST_NEXT(view, link);
|
||||
ISC_LIST_UNLINK(cba.oldviews, view, link);
|
||||
dns_view_detach(&view);
|
||||
}
|
||||
}
|
||||
|
||||
/* cleanup */
|
||||
view = ISC_LIST_HEAD(cba.oldviews);
|
||||
while (view != NULL) {
|
||||
dns_view_t *next;
|
||||
next = ISC_LIST_NEXT(view, link);
|
||||
ISC_LIST_UNLINK(cba.oldviews, view, link);
|
||||
dns_view_detach(&view);
|
||||
view = next;
|
||||
}
|
||||
|
||||
view = ISC_LIST_HEAD(cba.newviews);
|
||||
while (view != NULL) {
|
||||
dns_view_t *next;
|
||||
next = ISC_LIST_NEXT(view, link);
|
||||
ISC_LIST_UNLINK(cba.newviews, view, link);
|
||||
dns_view_detach(&view);
|
||||
view = next;
|
||||
}
|
||||
|
||||
if (configctx != NULL)
|
||||
dns_c_ctx_delete(NULL, &configctx);
|
||||
if (view2 != NULL)
|
||||
dns_view_detach(&view2);
|
||||
if (view1 != NULL)
|
||||
dns_view_detach(&view1);
|
||||
|
||||
if (!quiet && stats)
|
||||
isc_mem_stats(mctx, stdout);
|
||||
isc_mem_destroy(&mctx);
|
||||
|
||||
exit(0);
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user