bind9/bin/tests/zone2_test.c
David Lawrence 1b6d529cb5 Megacommit of many files.
Mostly, several functions that take pointers as arguments, almost
always char * pointers, had those pointers qualified with "const".
Those that returned pointers to previously const-qualified arguments
had their return values qualified as const.  Some structure members
were qualified as const to retain that attribute from the variables
from which they were assigned.

Several unused functions removed from dispatch_tcp_test.c (left over
from when it was copied from dispatch_test.c).

Minor other ISC style cleanups.
2000-06-01 19:11:07 +00:00

426 lines
10 KiB
C

/*
* Copyright (C) 1999, 2000 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 <stdlib.h>
#include <unistd.h>
#include <isc/app.h>
#include <isc/mem.h>
#include <isc/string.h>
#include <isc/util.h>
#include <dns/confparser.h>
#include <dns/journal.h>
#include <dns/fixedname.h>
#include <dns/rdataset.h>
#include <dns/result.h>
#include <dns/view.h>
#include <dns/zone.h>
#include <dns/zoneconf.h>
#include <dns/zt.h>
#define ERRRET(result, function) \
do { \
if (result != ISC_R_SUCCESS) { \
fprintf(stdout, "%s() returned %s\n", \
function, dns_result_totext(result)); \
return; \
} \
} while (0)
#define ERRCONT(result, function) \
if (result != ISC_R_SUCCESS) { \
fprintf(stdout, "%s() returned %s\n", \
function, dns_result_totext(result)); \
continue; \
} else \
(void)NULL
typedef struct dns_zone_callbackarg dns_zone_callbackarg_t;
struct dns_zone_callbackarg {
isc_mem_t *mctx;
dns_viewlist_t oldviews;
dns_viewlist_t newviews;
};
static isc_result_t
dns_zone_callback(dns_c_ctx_t *cfg, dns_c_zone_t *czone, dns_c_view_t *cview,
void *uap) {
dns_zone_callbackarg_t *cba = uap;
dns_name_t *name = NULL;
dns_view_t *oldview = NULL;
dns_zone_t *oldzone = NULL;
dns_view_t *newview = NULL;
dns_zone_t *newzone = NULL;
dns_zone_t *tmpzone = NULL;
isc_result_t result;
isc_boolean_t boolean;
const char *viewname;
REQUIRE(czone != NULL);
REQUIRE(cba != NULL);
/*
* Find views by name.
*/
if (cview != NULL)
dns_c_view_getname(cview, &viewname);
else
viewname = "default";
printf("view %s\n", viewname);
result = dns_viewlist_find(&cba->oldviews, viewname, czone->zclass,
&oldview);
result = dns_viewlist_find(&cba->newviews, viewname, czone->zclass,
&newview);
if (newview == NULL) {
result = dns_view_create(cba->mctx, czone->zclass, viewname,
&newview);
if (result != ISC_R_SUCCESS)
goto cleanup;
ISC_LIST_APPEND(cba->newviews, newview, link);
}
/*
* Create and populate a new zone structure.
*/
result = dns_zone_create(&newzone, cba->mctx);
if (result != ISC_R_SUCCESS)
goto cleanup;
result = dns_zone_configure(cfg, NULL, czone, NULL, newzone);
if (result != ISC_R_SUCCESS)
goto cleanup;
#if 0
/* XXX hints should be a zone */
if (dns_zone_gettype(newzone) == dns_zone_hint) {
dns_view_sethints(newview, newzone);
goto cleanup;
}
#endif
/*
* Find zone in mount table.
*/
name = dns_zone_getorigin(newzone);
dns_zone_print(newzone);
result = dns_zt_find(newview->zonetable, name, 0, NULL, &tmpzone);
if (result == ISC_R_SUCCESS) {
printf("zone already exists=\n");
result = ISC_R_EXISTS;
goto cleanup;
} else if (result != DNS_R_PARTIALMATCH && result != ISC_R_NOTFOUND)
goto cleanup;
if (oldview != NULL)
result = dns_zt_find(oldview->zonetable, name, 0, NULL,
&oldzone);
else
result = ISC_R_NOTFOUND;
printf("dns_zt_find() returned %s\n", dns_result_totext(result));
if (result == ISC_R_NOTFOUND || result == DNS_R_PARTIALMATCH) {
if (result == DNS_R_PARTIALMATCH) {
dns_zone_print(oldzone);
}
result = dns_view_addzone(newview, newzone);
if (result != ISC_R_SUCCESS)
goto cleanup;
} else if (result == ISC_R_SUCCESS) {
dns_zone_print(oldzone);
/* Does the new configuration match the existing one? */
boolean = dns_zone_equal(newzone, oldzone);
printf("dns_zone_equal() returned %s\n", boolean ? "TRUE" : "FALSE");
if (boolean)
result = dns_view_addzone(newview, oldzone);
else
result = dns_view_addzone(newview, newzone);
}
cleanup:
if (tmpzone != NULL)
dns_zone_detach(&tmpzone);
if (newzone != NULL)
dns_zone_detach(&newzone);
if (oldzone != NULL)
dns_zone_detach(&oldzone);
return (result);
}
static void
print_rdataset(dns_name_t *name, dns_rdataset_t *rdataset) {
isc_buffer_t text;
char t[1000];
isc_result_t result;
isc_region_t r;
isc_buffer_init(&text, t, sizeof(t));
result = dns_rdataset_totext(rdataset, name, ISC_FALSE, ISC_FALSE,
&text);
isc_buffer_usedregion(&text, &r);
if (result == ISC_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;
isc_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",
stdout);
reload = 0;
continue;
}
dns_fixedname_init(&name);
isc_buffer_init(&buffer, buf, strlen(buf));
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), 0,
NULL, &zone);
if (result != ISC_R_SUCCESS) {
if (result == DNS_R_PARTIALMATCH)
dns_zone_detach(&zone);
reload = 0;
continue;
}
result = dns_zone_load(zone);
fprintf(stdout, "dns_zone_reload() returned %s\n",
dns_result_totext(result));
reload = 0;
dns_zone_detach(&zone);
} else {
result = dns_view_simplefind(view,
dns_fixedname_name(&name),
dns_rdatatype_a, 0, 0,
ISC_FALSE, &rdataset, &sigset);
fprintf(stdout, "%s() returned %s\n",
"dns_view_simplefind",
dns_result_totext(result));
switch (result) {
case ISC_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) {
const 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;
isc_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) == ISC_R_SUCCESS);
RUNTIME_CHECK(dns_view_create(mctx, dns_rdataclass_in,
"default/IN", &view2) == ISC_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(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(&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(conf, mctx, &configctx, &cbks);
fprintf(stdout, "%s() returned %s\n", "dns_c_parse_namedconf",
dns_result_totext(result));
if (result == ISC_R_SUCCESS) {
result = dns_c_ctx_getdirectory(configctx, &dir);
if (result == ISC_R_SUCCESS)
chdir(dir);
view = ISC_LIST_HEAD(cba.newviews);
while (view != NULL) {
dns_zt_print(view->zonetable);
dns_zt_load(view->zonetable, ISC_FALSE);
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(&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);
return (0);
}