bind/bin/tests/db_test.c

/*
 * 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 <stddef.h>
#include <stdlib.h>
#include <string.h>

#include <isc/assertions.h>
#include <isc/error.h>
#include <isc/boolean.h>
#include <isc/region.h>

#include <dns/types.h>
#include <dns/result.h>
#include <dns/name.h>
#include <dns/rdata.h>
#include <dns/rdataclass.h>
#include <dns/rdatatype.h>
#include <dns/rdatalist.h>
#include <dns/rdataset.h>
#include <dns/compress.h>
#include <dns/db.h>

static void
makename(isc_mem_t *mctx, char *text, dns_name_t *name, dns_name_t *origin) {
	char b[255];
	isc_buffer_t source, target;
	size_t len;
	isc_region_t r1, r2;
	dns_result_t result;

	if (origin == NULL)
		origin = dns_rootname;
	dns_name_init(name, NULL);
	len = strlen(text);
	isc_buffer_init(&source, text, len, ISC_BUFFERTYPE_TEXT);
	isc_buffer_add(&source, len);
	isc_buffer_init(&target, b, sizeof b, ISC_BUFFERTYPE_BINARY);
	result = dns_name_fromtext(name, &source, origin, ISC_FALSE, &target);
	RUNTIME_CHECK(result == DNS_R_SUCCESS);
	dns_name_toregion(name, &r1);
	r2.base = isc_mem_get(mctx, r1.length);
	RUNTIME_CHECK(r2.base != NULL);
	r2.length = r1.length;
	memcpy(r2.base, r1.base, r1.length);
	dns_name_fromregion(name, &r2);
}

static void
freename(isc_mem_t *mctx, dns_name_t *name) {
	isc_region_t r;

	dns_name_toregion(name, &r);
	isc_mem_put(mctx, r.base, r.length);
	dns_name_invalidate(name);
}

int
main(int argc, char *argv[]) {
	isc_mem_t *mctx = NULL;
	dns_db_t *db;
	dns_dbnode_t *node;
	dns_result_t result;
	dns_name_t name, base, *origin;
	dns_offsets_t offsets;
	size_t len;
	isc_buffer_t source, target;
	char s[1000];
	char b[255];
#if 0
	FILE *f;
#endif

	if (argc < 2) {
		fprintf(stderr, "usage: db_test filename\n");
		exit(1);
	}
#if 0
	f = fopen(argv[1], "r");
	if (f == NULL) {
		fprintf(stderr, "fopen failed.\n");
		exit(2);
	}
#endif

	RUNTIME_CHECK(isc_mem_create(0, 0, &mctx) == ISC_R_SUCCESS);

	makename(mctx, "vix.com.", &base, NULL);

	db = NULL;
	result = dns_db_create(mctx, "rbt", &base, ISC_FALSE, 1, 0, NULL,
			       &db);
	RUNTIME_CHECK(result == DNS_R_SUCCESS);
	
	origin = dns_rootname;
#if 0
	while (fgets(s, sizeof s, f) != NULL) {
		dns_name_init(&name, offsets);
		len = strlen(s);
		if (len > 0 && s[len - 1] == '\n') {
			len--;
			s[len] = '\0';
		}
		isc_buffer_init(&source, s, len, ISC_BUFFERTYPE_TEXT);
		isc_buffer_add(&source, len);
		isc_buffer_init(&target, b, sizeof b, ISC_BUFFERTYPE_BINARY);
		result = dns_name_fromtext(&name, &source, origin, ISC_FALSE,
					   &target);
		if (result != DNS_R_SUCCESS) {
			printf("bad db name: %s\n", dns_result_totext(result));
			continue;
		}
		node = NULL;
		result = dns_db_findnode(db, &name, ISC_TRUE, &node);
		RUNTIME_CHECK(result == DNS_R_SUCCESS);
		dns_db_detachnode(db, &node);
	}
#endif
	result = dns_db_load(db, argv[1]);
	if (result != DNS_R_SUCCESS) {
		printf("couldn't load master file: %s\n",
		       dns_result_totext(result));
		exit(1);
	}

	while (gets(s) != NULL) {
		dns_name_init(&name, offsets);
		len = strlen(s);
		isc_buffer_init(&source, s, len, ISC_BUFFERTYPE_TEXT);
		isc_buffer_add(&source, len);
		isc_buffer_init(&target, b, sizeof b, ISC_BUFFERTYPE_BINARY);
		result = dns_name_fromtext(&name, &source, origin, ISC_FALSE,
					   &target);
		if (result != DNS_R_SUCCESS) {
			printf("bad name: %s\n", dns_result_totext(result));
			continue;
		}
		node = NULL;
		result = dns_db_findnode(db, &name, ISC_FALSE, &node);
		if (result == DNS_R_NOTFOUND)
			printf("not found\n");
		else if (result != DNS_R_SUCCESS)
			printf("%s\n", dns_result_totext(result));
		else {
			printf("success\n");
			dns_db_detachnode(db, &node);
		}
	}

	dns_db_detach(&db);
	freename(mctx, &base);

	isc_mem_stats(mctx, stdout);

	return (0);
}