db_test.c revision b6b21d80450f81d873d7e8cd21e7b72fdf512507
13faa91230bde46da937bf33010b9accc5bdeb59sd/*
13faa91230bde46da937bf33010b9accc5bdeb59sd * Copyright (C) 2004 Internet Systems Consortium, Inc. ("ISC")
13faa91230bde46da937bf33010b9accc5bdeb59sd * Copyright (C) 1999-2001 Internet Software Consortium.
13faa91230bde46da937bf33010b9accc5bdeb59sd *
13faa91230bde46da937bf33010b9accc5bdeb59sd * Permission to use, copy, modify, and distribute this software for any
13faa91230bde46da937bf33010b9accc5bdeb59sd * purpose with or without fee is hereby granted, provided that the above
13faa91230bde46da937bf33010b9accc5bdeb59sd * copyright notice and this permission notice appear in all copies.
13faa91230bde46da937bf33010b9accc5bdeb59sd *
13faa91230bde46da937bf33010b9accc5bdeb59sd * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
13faa91230bde46da937bf33010b9accc5bdeb59sd * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
13faa91230bde46da937bf33010b9accc5bdeb59sd * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
13faa91230bde46da937bf33010b9accc5bdeb59sd * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
13faa91230bde46da937bf33010b9accc5bdeb59sd * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
13faa91230bde46da937bf33010b9accc5bdeb59sd * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13faa91230bde46da937bf33010b9accc5bdeb59sd * PERFORMANCE OF THIS SOFTWARE.
13faa91230bde46da937bf33010b9accc5bdeb59sd */
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd/* $Id: db_test.c,v 1.60 2005/03/16 22:22:27 marka Exp $ */
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd/*
13faa91230bde46da937bf33010b9accc5bdeb59sd * Principal Author: Bob Halley
13faa91230bde46da937bf33010b9accc5bdeb59sd */
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <config.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <stdlib.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <isc/commandline.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <isc/log.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <isc/mem.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <isc/time.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <isc/string.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <isc/util.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <dns/db.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <dns/dbiterator.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <dns/dbtable.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <dns/fixedname.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <dns/log.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <dns/rdataset.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <dns/rdatasetiter.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd#include <dns/result.h>
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd#define MAXHOLD 100
13faa91230bde46da937bf33010b9accc5bdeb59sd#define MAXVERSIONS 100
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdtypedef struct dbinfo {
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_t * db;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbversion_t * version;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbversion_t * wversion;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbversion_t * rversions[MAXVERSIONS];
13faa91230bde46da937bf33010b9accc5bdeb59sd int rcount;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbnode_t * hold_nodes[MAXHOLD];
13faa91230bde46da937bf33010b9accc5bdeb59sd int hold_count;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbiterator_t * dbiterator;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbversion_t * iversion;
13faa91230bde46da937bf33010b9accc5bdeb59sd int pause_every;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t ascending;
13faa91230bde46da937bf33010b9accc5bdeb59sd ISC_LINK(struct dbinfo) link;
13faa91230bde46da937bf33010b9accc5bdeb59sd} dbinfo;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic isc_mem_t * mctx = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic char dbtype[128];
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic dns_dbtable_t * dbtable;
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic ISC_LIST(dbinfo) dbs;
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic dbinfo * cache_dbi = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic int pause_every = 0;
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic isc_boolean_t ascending = ISC_TRUE;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic void
13faa91230bde46da937bf33010b9accc5bdeb59sdprint_result(const char *message, isc_result_t result) {
13faa91230bde46da937bf33010b9accc5bdeb59sd size_t len;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd if (message == NULL) {
13faa91230bde46da937bf33010b9accc5bdeb59sd len = 0;
13faa91230bde46da937bf33010b9accc5bdeb59sd message = "";
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd len = strlen(message);
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("%s%sresult %08x: %s\n", message, (len == 0U) ? "" : " ",
13faa91230bde46da937bf33010b9accc5bdeb59sd result, isc_result_totext(result));
13faa91230bde46da937bf33010b9accc5bdeb59sd}
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic void
13faa91230bde46da937bf33010b9accc5bdeb59sdprint_rdataset(dns_name_t *name, dns_rdataset_t *rdataset) {
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_t text;
13faa91230bde46da937bf33010b9accc5bdeb59sd char t[1000];
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_result_t result;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_region_t r;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_init(&text, t, sizeof(t));
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_rdataset_totext(rdataset, name, ISC_FALSE, ISC_FALSE,
13faa91230bde46da937bf33010b9accc5bdeb59sd &text);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_usedregion(&text, &r);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result == ISC_R_SUCCESS)
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("%.*s", (int)r.length, (char *)r.base);
13faa91230bde46da937bf33010b9accc5bdeb59sd else
13faa91230bde46da937bf33010b9accc5bdeb59sd print_result("", result);
13faa91230bde46da937bf33010b9accc5bdeb59sd}
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic void
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyenprint_rdatasets(dns_name_t *name, dns_rdatasetiter_t *rdsiter) {
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_result_t result;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdataset_t rdataset;
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdataset_init(&rdataset);
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_rdatasetiter_first(rdsiter);
13faa91230bde46da937bf33010b9accc5bdeb59sd while (result == ISC_R_SUCCESS) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdatasetiter_current(rdsiter, &rdataset);
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk print_rdataset(name, &rdataset);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdataset_disassociate(&rdataset);
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_rdatasetiter_next(rdsiter);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_NOMORE)
13faa91230bde46da937bf33010b9accc5bdeb59sd print_result("", result);
13faa91230bde46da937bf33010b9accc5bdeb59sd}
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic dbinfo *
13faa91230bde46da937bf33010b9accc5bdeb59sdselect_db(char *origintext) {
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen dns_fixedname_t forigin;
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen dns_name_t *origin;
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen isc_buffer_t source;
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen size_t len;
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen dbinfo *dbi;
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen isc_result_t result;
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen if (strcasecmp(origintext, "cache") == 0) {
13faa91230bde46da937bf33010b9accc5bdeb59sd if (cache_dbi == NULL)
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("the cache does not exist\n");
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk return (cache_dbi);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen len = strlen(origintext);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_init(&source, origintext, len);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_add(&source, len);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_init(&forigin);
13faa91230bde46da937bf33010b9accc5bdeb59sd origin = dns_fixedname_name(&forigin);
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_name_fromtext(origin, &source, dns_rootname, ISC_FALSE,
13faa91230bde46da937bf33010b9accc5bdeb59sd NULL);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_SUCCESS) {
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk print_result("bad name", result);
13faa91230bde46da937bf33010b9accc5bdeb59sd return (NULL);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd for (dbi = ISC_LIST_HEAD(dbs);
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi != NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi = ISC_LIST_NEXT(dbi, link)) {
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dns_name_compare(dns_db_origin(dbi->db), origin) == 0)
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd }
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk
13faa91230bde46da937bf33010b9accc5bdeb59sd return (dbi);
13faa91230bde46da937bf33010b9accc5bdeb59sd}
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic void
13faa91230bde46da937bf33010b9accc5bdeb59sdlist(dbinfo *dbi, char *seektext) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_t fname;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_name_t *name;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbnode_t *node;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdatasetiter_t *rdsiter;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_result_t result;
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk int i;
13faa91230bde46da937bf33010b9accc5bdeb59sd size_t len;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_t fseekname;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_name_t *seekname;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_t source;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_init(&fname);
13faa91230bde46da937bf33010b9accc5bdeb59sd name = dns_fixedname_name(&fname);
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->dbiterator == NULL) {
13faa91230bde46da937bf33010b9accc5bdeb59sd INSIST(dbi->iversion == NULL);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dns_db_iszone(dbi->db)) {
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->version != NULL)
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_attachversion(dbi->db, dbi->version,
13faa91230bde46da937bf33010b9accc5bdeb59sd &dbi->iversion);
13faa91230bde46da937bf33010b9accc5bdeb59sd else
186d582bd9dbcd38e0aeea49036d47d3426a3536Surya Prakki dns_db_currentversion(dbi->db, &dbi->iversion);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_db_createiterator(dbi->db, ISC_FALSE,
13faa91230bde46da937bf33010b9accc5bdeb59sd &dbi->dbiterator);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result == ISC_R_SUCCESS) {
13faa91230bde46da937bf33010b9accc5bdeb59sd if (seektext != NULL) {
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk len = strlen(seektext);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_init(&source, seektext, len);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_add(&source, len);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_init(&fseekname);
13faa91230bde46da937bf33010b9accc5bdeb59sd seekname = dns_fixedname_name(&fseekname);
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_name_fromtext(seekname, &source,
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_origin(
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->db),
13faa91230bde46da937bf33010b9accc5bdeb59sd ISC_FALSE,
13faa91230bde46da937bf33010b9accc5bdeb59sd NULL);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result == ISC_R_SUCCESS)
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_dbiterator_seek(
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->dbiterator,
13faa91230bde46da937bf33010b9accc5bdeb59sd seekname);
13faa91230bde46da937bf33010b9accc5bdeb59sd } else if (dbi->ascending)
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_dbiterator_first(dbi->dbiterator);
13faa91230bde46da937bf33010b9accc5bdeb59sd else
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_dbiterator_last(dbi->dbiterator);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd } else
13faa91230bde46da937bf33010b9accc5bdeb59sd result = ISC_R_SUCCESS;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd node = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd rdsiter = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd i = 0;
13faa91230bde46da937bf33010b9accc5bdeb59sd while (result == ISC_R_SUCCESS) {
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_dbiterator_current(dbi->dbiterator, &node, name);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_SUCCESS && result != DNS_R_NEWORIGIN)
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk break;
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_db_allrdatasets(dbi->db, node, dbi->iversion, 0,
13faa91230bde46da937bf33010b9accc5bdeb59sd &rdsiter);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_SUCCESS) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_detachnode(dbi->db, &node);
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd print_rdatasets(name, rdsiter);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdatasetiter_destroy(&rdsiter);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_detachnode(dbi->db, &node);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->ascending)
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_dbiterator_next(dbi->dbiterator);
13faa91230bde46da937bf33010b9accc5bdeb59sd else
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_dbiterator_prev(dbi->dbiterator);
13faa91230bde46da937bf33010b9accc5bdeb59sd i++;
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result == ISC_R_SUCCESS && i == dbi->pause_every) {
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("[more...]\n");
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_dbiterator_pause(dbi->dbiterator);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result == ISC_R_SUCCESS)
13faa91230bde46da937bf33010b9accc5bdeb59sd return;
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_NOMORE)
13faa91230bde46da937bf33010b9accc5bdeb59sd print_result("", result);
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbiterator_destroy(&dbi->dbiterator);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->iversion != NULL)
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_closeversion(dbi->db, &dbi->iversion, ISC_FALSE);
13faa91230bde46da937bf33010b9accc5bdeb59sd}
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic isc_result_t
13faa91230bde46da937bf33010b9accc5bdeb59sdload(const char *filename, const char *origintext, isc_boolean_t cache) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_t forigin;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_name_t *origin;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_result_t result;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_t source;
13faa91230bde46da937bf33010b9accc5bdeb59sd size_t len;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbinfo *dbi;
13faa91230bde46da937bf33010b9accc5bdeb59sd unsigned int i;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi = isc_mem_get(mctx, sizeof(*dbi));
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi == NULL)
13faa91230bde46da937bf33010b9accc5bdeb59sd return (ISC_R_NOMEMORY);
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->db = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->version = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->wversion = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd for (i = 0; i < MAXVERSIONS; i++)
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->rversions[i] = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->hold_count = 0;
13faa91230bde46da937bf33010b9accc5bdeb59sd for (i = 0; i < MAXHOLD; i++)
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->hold_nodes[i] = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->dbiterator = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->iversion = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->pause_every = pause_every;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->ascending = ascending;
13faa91230bde46da937bf33010b9accc5bdeb59sd ISC_LINK_INIT(dbi, link);
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd len = strlen(origintext);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_init(&source, origintext, len);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_add(&source, len);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_init(&forigin);
13faa91230bde46da937bf33010b9accc5bdeb59sd origin = dns_fixedname_name(&forigin);
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_name_fromtext(origin, &source, dns_rootname, ISC_FALSE,
13faa91230bde46da937bf33010b9accc5bdeb59sd NULL);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_SUCCESS)
13faa91230bde46da937bf33010b9accc5bdeb59sd return (result);
13faa91230bde46da937bf33010b9accc5bdeb59sd
9c94f155585ea35e938fea603bc227c685223abdCheng Sean Ye result = dns_db_create(mctx, dbtype, origin,
9c94f155585ea35e938fea603bc227c685223abdCheng Sean Ye cache ? dns_dbtype_cache : dns_dbtype_zone,
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdataclass_in,
13faa91230bde46da937bf33010b9accc5bdeb59sd 0, NULL, &dbi->db);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_SUCCESS) {
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_mem_put(mctx, dbi, sizeof(*dbi));
13faa91230bde46da937bf33010b9accc5bdeb59sd return (result);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("loading %s (%s)\n", filename, origintext);
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_db_load(dbi->db, filename);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_SUCCESS && result != DNS_R_SEENINCLUDE) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_detach(&dbi->db);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_mem_put(mctx, dbi, sizeof(*dbi));
13faa91230bde46da937bf33010b9accc5bdeb59sd return (result);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("loaded\n");
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd if (cache) {
13faa91230bde46da937bf33010b9accc5bdeb59sd INSIST(cache_dbi == NULL);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbtable_adddefault(dbtable, dbi->db);
13faa91230bde46da937bf33010b9accc5bdeb59sd cache_dbi = dbi;
13faa91230bde46da937bf33010b9accc5bdeb59sd } else {
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dns_dbtable_add(dbtable, dbi->db) != ISC_R_SUCCESS) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_detach(&dbi->db);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_mem_put(mctx, dbi, sizeof(*dbi));
13faa91230bde46da937bf33010b9accc5bdeb59sd return (result);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd ISC_LIST_APPEND(dbs, dbi, link);
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd return (ISC_R_SUCCESS);
13faa91230bde46da937bf33010b9accc5bdeb59sd}
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdstatic void
13faa91230bde46da937bf33010b9accc5bdeb59sdunload_all(void) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dbinfo *dbi, *dbi_next;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd for (dbi = ISC_LIST_HEAD(dbs); dbi != NULL; dbi = dbi_next) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi_next = ISC_LIST_NEXT(dbi, link);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dns_db_iszone(dbi->db))
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbtable_remove(dbtable, dbi->db);
13faa91230bde46da937bf33010b9accc5bdeb59sd else {
13faa91230bde46da937bf33010b9accc5bdeb59sd INSIST(dbi == cache_dbi);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbtable_removedefault(dbtable);
13faa91230bde46da937bf33010b9accc5bdeb59sd cache_dbi = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_detach(&dbi->db);
13faa91230bde46da937bf33010b9accc5bdeb59sd ISC_LIST_UNLINK(dbs, dbi, link);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_mem_put(mctx, dbi, sizeof(*dbi));
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd}
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd#define DBI_CHECK(dbi) \
13faa91230bde46da937bf33010b9accc5bdeb59sdif ((dbi) == NULL) { \
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("You must first select a database with !DB\n"); \
13faa91230bde46da937bf33010b9accc5bdeb59sd continue; \
13faa91230bde46da937bf33010b9accc5bdeb59sd}
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sdint
13faa91230bde46da937bf33010b9accc5bdeb59sdmain(int argc, char *argv[]) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_t *db;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbnode_t *node;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_result_t result;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_name_t name;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_offsets_t offsets;
13faa91230bde46da937bf33010b9accc5bdeb59sd size_t len;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_t source, target;
13faa91230bde46da937bf33010b9accc5bdeb59sd char s[1000];
13faa91230bde46da937bf33010b9accc5bdeb59sd char b[255];
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdataset_t rdataset, sigrdataset;
13faa91230bde46da937bf33010b9accc5bdeb59sd int ch;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdatatype_t type = 1;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t printnode = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t addmode = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t delmode = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t holdmode = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t verbose = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t done = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t quiet = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t time_lookups = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t found_as;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t find_zonecut = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_boolean_t noexact_zonecut = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd int i, v;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_rdatasetiter_t *rdsiter;
13faa91230bde46da937bf33010b9accc5bdeb59sd char t1[256];
13faa91230bde46da937bf33010b9accc5bdeb59sd char t2[256];
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_buffer_t tb1, tb2;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_region_t r1, r2;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_t foundname;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_name_t *fname;
13faa91230bde46da937bf33010b9accc5bdeb59sd unsigned int options = 0, zcoptions;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_time_t start, finish;
13faa91230bde46da937bf33010b9accc5bdeb59sd char *origintext;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbinfo *dbi;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_dbversion_t *version;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_name_t *origin;
13faa91230bde46da937bf33010b9accc5bdeb59sd size_t memory_quota = 0;
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_trust_t trust = 0;
13faa91230bde46da937bf33010b9accc5bdeb59sd unsigned int addopts;
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_log_t *lctx = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_result_register();
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd RUNTIME_CHECK(isc_mem_create(0, 0, &mctx) == ISC_R_SUCCESS);
13faa91230bde46da937bf33010b9accc5bdeb59sd RUNTIME_CHECK(dns_dbtable_create(mctx, dns_rdataclass_in, &dbtable) ==
13faa91230bde46da937bf33010b9accc5bdeb59sd ISC_R_SUCCESS);
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd strcpy(dbtype, "rbt");
13faa91230bde46da937bf33010b9accc5bdeb59sd while ((ch = isc_commandline_parse(argc, argv, "c:d:t:z:P:Q:glpqvT"))
13faa91230bde46da937bf33010b9accc5bdeb59sd != -1) {
13faa91230bde46da937bf33010b9accc5bdeb59sd switch (ch) {
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'c':
13faa91230bde46da937bf33010b9accc5bdeb59sd result = load(isc_commandline_argument, ".", ISC_TRUE);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (result != ISC_R_SUCCESS)
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("cache load(%s) %08x: %s\n",
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_commandline_argument, result,
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_result_totext(result));
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'd':
13faa91230bde46da937bf33010b9accc5bdeb59sd strcpy(dbtype, isc_commandline_argument);
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'g':
13faa91230bde46da937bf33010b9accc5bdeb59sd options |= (DNS_DBFIND_GLUEOK|DNS_DBFIND_VALIDATEGLUE);
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'l':
13faa91230bde46da937bf33010b9accc5bdeb59sd RUNTIME_CHECK(isc_log_create(mctx, &lctx,
13faa91230bde46da937bf33010b9accc5bdeb59sd NULL) == ISC_R_SUCCESS);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_log_setcontext(lctx);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_log_init(lctx);
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_log_setcontext(lctx);
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'q':
13faa91230bde46da937bf33010b9accc5bdeb59sd quiet = ISC_TRUE;
13faa91230bde46da937bf33010b9accc5bdeb59sd verbose = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'p':
13faa91230bde46da937bf33010b9accc5bdeb59sd printnode = ISC_TRUE;
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'P':
13faa91230bde46da937bf33010b9accc5bdeb59sd pause_every = atoi(isc_commandline_argument);
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'Q':
13faa91230bde46da937bf33010b9accc5bdeb59sd memory_quota = atoi(isc_commandline_argument);
13faa91230bde46da937bf33010b9accc5bdeb59sd isc_mem_setquota(mctx, memory_quota);
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 't':
13faa91230bde46da937bf33010b9accc5bdeb59sd type = atoi(isc_commandline_argument);
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'T':
13faa91230bde46da937bf33010b9accc5bdeb59sd time_lookups = ISC_TRUE;
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'v':
13faa91230bde46da937bf33010b9accc5bdeb59sd verbose = ISC_TRUE;
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
13faa91230bde46da937bf33010b9accc5bdeb59sd case 'z':
13faa91230bde46da937bf33010b9accc5bdeb59sd origintext = strrchr(isc_commandline_argument, '/');
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do if (origintext == NULL)
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do origintext = isc_commandline_argument;
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do else
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do origintext++; /* Skip '/'. */
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do result = load(isc_commandline_argument, origintext,
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do ISC_FALSE);
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do if (result != ISC_R_SUCCESS)
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do printf("zone load(%s) %08x: %s\n",
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do isc_commandline_argument, result,
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do isc_result_totext(result));
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do break;
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do }
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do }
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do argc -= isc_commandline_index;
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do argv += isc_commandline_index;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd if (argc != 0)
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("ignoring trailing arguments\n");
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd /*
13faa91230bde46da937bf33010b9accc5bdeb59sd * Some final initialization...
13faa91230bde46da937bf33010b9accc5bdeb59sd */
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_fixedname_init(&foundname);
13faa91230bde46da937bf33010b9accc5bdeb59sd fname = dns_fixedname_name(&foundname);
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd origin = dns_rootname;
13faa91230bde46da937bf33010b9accc5bdeb59sd version = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd if (time_lookups) {
13faa91230bde46da937bf33010b9accc5bdeb59sd TIME_NOW(&start);
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd
13faa91230bde46da937bf33010b9accc5bdeb59sd while (!done) {
13faa91230bde46da937bf33010b9accc5bdeb59sd if (!quiet)
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("\n");
13faa91230bde46da937bf33010b9accc5bdeb59sd if (fgets(s, sizeof(s), stdin) == NULL) {
13faa91230bde46da937bf33010b9accc5bdeb59sd done = ISC_TRUE;
13faa91230bde46da937bf33010b9accc5bdeb59sd continue;
13faa91230bde46da937bf33010b9accc5bdeb59sd }
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk len = strlen(s);
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk if (len > 0U && s[len - 1] == '\n') {
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk s[len - 1] = '\0';
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk len--;
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk }
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk if (verbose && dbi != NULL) {
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk if (dbi->wversion != NULL)
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk printf("future version (%p)\n", dbi->wversion);
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk for (i = 0; i < dbi->rcount; i++)
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk if (dbi->rversions[i] != NULL)
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk printf("open version %d (%p)\n", i,
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk dbi->rversions[i]);
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk }
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk dns_name_init(&name, offsets);
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk if (strcmp(s, "!R") == 0) {
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk DBI_CHECK(dbi);
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk if (dbi->rcount == MAXVERSIONS) {
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk printf("too many open versions\n");
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk continue;
67d4b2f88b8e27bb035d67a046d5aad7db3bfc71gk }
13faa91230bde46da937bf33010b9accc5bdeb59sd dns_db_currentversion(dbi->db,
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do &dbi->rversions[dbi->rcount]);
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do printf("opened version %d\n", dbi->rcount);
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->version = dbi->rversions[dbi->rcount];
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do version = dbi->version;
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->rcount++;
13faa91230bde46da937bf33010b9accc5bdeb59sd continue;
13faa91230bde46da937bf33010b9accc5bdeb59sd } else if (strcmp(s, "!W") == 0) {
13faa91230bde46da937bf33010b9accc5bdeb59sd DBI_CHECK(dbi);
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->wversion != NULL) {
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("using existing future version\n");
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->version = dbi->wversion;
13faa91230bde46da937bf33010b9accc5bdeb59sd version = dbi->version;
13faa91230bde46da937bf33010b9accc5bdeb59sd continue;
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd result = dns_db_newversion(dbi->db, &dbi->wversion);
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do if (result != ISC_R_SUCCESS)
13faa91230bde46da937bf33010b9accc5bdeb59sd print_result("", result);
13faa91230bde46da937bf33010b9accc5bdeb59sd else
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("newversion\n");
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->version = dbi->wversion;
13faa91230bde46da937bf33010b9accc5bdeb59sd version = dbi->version;
13faa91230bde46da937bf33010b9accc5bdeb59sd continue;
13faa91230bde46da937bf33010b9accc5bdeb59sd } else if (strcmp(s, "!C") == 0) {
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do DBI_CHECK(dbi);
13faa91230bde46da937bf33010b9accc5bdeb59sd addmode = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd delmode = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->version == NULL)
13faa91230bde46da937bf33010b9accc5bdeb59sd continue;
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->version == dbi->wversion) {
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("closing future version\n");
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do dbi->wversion = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd } else {
13faa91230bde46da937bf33010b9accc5bdeb59sd for (i = 0; i < dbi->rcount; i++) {
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->version ==
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->rversions[i]) {
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->rversions[i] = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd printf("closing open version %d\n",
13faa91230bde46da937bf33010b9accc5bdeb59sd i);
13faa91230bde46da937bf33010b9accc5bdeb59sd break;
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do }
13faa91230bde46da937bf33010b9accc5bdeb59sd }
13faa91230bde46da937bf33010b9accc5bdeb59sd }
25351652d920ae27c5a56c199da581033ce763f6Vuong Nguyen dns_db_closeversion(dbi->db, &dbi->version, ISC_TRUE);
13faa91230bde46da937bf33010b9accc5bdeb59sd version = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd continue;
13faa91230bde46da937bf33010b9accc5bdeb59sd } else if (strcmp(s, "!X") == 0) {
13faa91230bde46da937bf33010b9accc5bdeb59sd DBI_CHECK(dbi);
13faa91230bde46da937bf33010b9accc5bdeb59sd addmode = ISC_FALSE;
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do delmode = ISC_FALSE;
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->version == NULL)
13faa91230bde46da937bf33010b9accc5bdeb59sd continue;
13faa91230bde46da937bf33010b9accc5bdeb59sd if (dbi->version == dbi->wversion) {
b64bfe7dc77dc5c5561cdcd10c80b0b550701a24Trang Do printf("aborting future version\n");
13faa91230bde46da937bf33010b9accc5bdeb59sd dbi->wversion = NULL;
13faa91230bde46da937bf33010b9accc5bdeb59sd } else {
for (i = 0; i < dbi->rcount; i++) {
if (dbi->version ==
dbi->rversions[i]) {
dbi->rversions[i] = NULL;
printf("closing open version %d\n",
i);
break;
}
}
}
dns_db_closeversion(dbi->db, &dbi->version, ISC_FALSE);
version = NULL;
continue;
} else if (strcmp(s, "!A") == 0) {
DBI_CHECK(dbi);
delmode = ISC_FALSE;
if (addmode)
addmode = ISC_FALSE;
else
addmode = ISC_TRUE;
printf("addmode = %s\n", addmode ? "TRUE" : "FALSE");
continue;
} else if (strcmp(s, "!D") == 0) {
DBI_CHECK(dbi);
addmode = ISC_FALSE;
if (delmode)
delmode = ISC_FALSE;
else
delmode = ISC_TRUE;
printf("delmode = %s\n", delmode ? "TRUE" : "FALSE");
continue;
} else if (strcmp(s, "!H") == 0) {
DBI_CHECK(dbi);
if (holdmode)
holdmode = ISC_FALSE;
else
holdmode = ISC_TRUE;
printf("holdmode = %s\n", holdmode ? "TRUE" : "FALSE");
continue;
} else if (strcmp(s, "!HR") == 0) {
DBI_CHECK(dbi);
for (i = 0; i < dbi->hold_count; i++)
dns_db_detachnode(dbi->db,
&dbi->hold_nodes[i]);
dbi->hold_count = 0;
holdmode = ISC_FALSE;
printf("held nodes have been detached\n");
continue;
} else if (strcmp(s, "!VC") == 0) {
DBI_CHECK(dbi);
printf("switching to current version\n");
dbi->version = NULL;
version = NULL;
continue;
} else if (strstr(s, "!V") == s) {
DBI_CHECK(dbi);
v = atoi(&s[2]);
if (v >= dbi->rcount) {
printf("unknown open version %d\n", v);
continue;
} else if (dbi->rversions[v] == NULL) {
printf("version %d is not open\n", v);
continue;
}
printf("switching to open version %d\n", v);
dbi->version = dbi->rversions[v];
version = dbi->version;
continue;
} else if (strstr(s, "!TR") == s) {
trust = (unsigned int)atoi(&s[3]);
printf("trust level is now %u\n", (unsigned int)trust);
continue;
} else if (strstr(s, "!T") == s) {
type = (unsigned int)atoi(&s[2]);
printf("now searching for type %u\n", type);
continue;
} else if (strcmp(s, "!G") == 0) {
if ((options & DNS_DBFIND_GLUEOK) != 0)
options &= ~DNS_DBFIND_GLUEOK;
else
options |= DNS_DBFIND_GLUEOK;
printf("glue ok = %s\n",
((options & DNS_DBFIND_GLUEOK) != 0) ?
"TRUE" : "FALSE");
continue;
} else if (strcmp(s, "!GV") == 0) {
if ((options & DNS_DBFIND_VALIDATEGLUE) != 0)
options &= ~DNS_DBFIND_VALIDATEGLUE;
else
options |= DNS_DBFIND_VALIDATEGLUE;
printf("validate glue = %s\n",
((options & DNS_DBFIND_VALIDATEGLUE) != 0) ?
"TRUE" : "FALSE");
continue;
} else if (strcmp(s, "!WC") == 0) {
if ((options & DNS_DBFIND_NOWILD) != 0)
options &= ~DNS_DBFIND_NOWILD;
else
options |= DNS_DBFIND_NOWILD;
printf("wildcard matching = %s\n",
((options & DNS_DBFIND_NOWILD) == 0) ?
"TRUE" : "FALSE");
continue;
} else if (strstr(s, "!LS ") == s) {
DBI_CHECK(dbi);
list(dbi, &s[4]);
continue;
} else if (strcmp(s, "!LS") == 0) {
DBI_CHECK(dbi);
list(dbi, NULL);
continue;
} else if (strstr(s, "!DU ") == s) {
DBI_CHECK(dbi);
result = dns_db_dump(dbi->db, dbi->version, s+4);
if (result != ISC_R_SUCCESS) {
printf("\n");
print_result("", result);
}
continue;
} else if (strcmp(s, "!PN") == 0) {
if (printnode)
printnode = ISC_FALSE;
else
printnode = ISC_TRUE;
printf("printnode = %s\n",
printnode ? "TRUE" : "FALSE");
continue;
} else if (strstr(s, "!P") == s) {
DBI_CHECK(dbi);
v = atoi(&s[2]);
dbi->pause_every = v;
continue;
} else if (strcmp(s, "!+") == 0) {
DBI_CHECK(dbi);
dbi->ascending = ISC_TRUE;
continue;
} else if (strcmp(s, "!-") == 0) {
DBI_CHECK(dbi);
dbi->ascending = ISC_FALSE;
continue;
} else if (strcmp(s, "!DB") == 0) {
dbi = NULL;
origin = dns_rootname;
version = NULL;
printf("now searching all databases\n");
continue;
} else if (strncmp(s, "!DB ", 4) == 0) {
dbi = select_db(s+4);
if (dbi != NULL) {
db = dbi->db;
origin = dns_db_origin(dbi->db);
version = dbi->version;
addmode = ISC_FALSE;
delmode = ISC_FALSE;
holdmode = ISC_FALSE;
} else {
db = NULL;
version = NULL;
origin = dns_rootname;
printf("database not found; "
"now searching all databases\n");
}
continue;
} else if (strcmp(s, "!ZC") == 0) {
if (find_zonecut)
find_zonecut = ISC_FALSE;
else
find_zonecut = ISC_TRUE;
printf("find_zonecut = %s\n",
find_zonecut ? "TRUE" : "FALSE");
continue;
} else if (strcmp(s, "!NZ") == 0) {
if (noexact_zonecut)
noexact_zonecut = ISC_FALSE;
else
noexact_zonecut = ISC_TRUE;
printf("noexact_zonecut = %s\n",
noexact_zonecut ? "TRUE" : "FALSE");
continue;
}
isc_buffer_init(&source, s, len);
isc_buffer_add(&source, len);
isc_buffer_init(&target, b, sizeof(b));
result = dns_name_fromtext(&name, &source, origin,
ISC_FALSE, &target);
if (result != ISC_R_SUCCESS) {
print_result("bad name: ", result);
continue;
}
if (dbi == NULL) {
zcoptions = 0;
if (noexact_zonecut)
zcoptions |= DNS_DBTABLEFIND_NOEXACT;
db = NULL;
result = dns_dbtable_find(dbtable, &name, zcoptions,
&db);
if (result != ISC_R_SUCCESS &&
result != DNS_R_PARTIALMATCH) {
if (!quiet) {
printf("\n");
print_result("", result);
}
continue;
}
isc_buffer_init(&tb1, t1, sizeof(t1));
result = dns_name_totext(dns_db_origin(db), ISC_FALSE,
&tb1);
if (result != ISC_R_SUCCESS) {
printf("\n");
print_result("", result);
dns_db_detach(&db);
continue;
}
isc_buffer_usedregion(&tb1, &r1);
printf("\ndatabase = %.*s (%s)\n",
(int)r1.length, r1.base,
(dns_db_iszone(db)) ? "zone" : "cache");
}
node = NULL;
dns_rdataset_init(&rdataset);
dns_rdataset_init(&sigrdataset);
if (find_zonecut && dns_db_iscache(db)) {
zcoptions = options;
if (noexact_zonecut)
zcoptions |= DNS_DBFIND_NOEXACT;
result = dns_db_findzonecut(db, &name, zcoptions,
0, &node, fname,
&rdataset, &sigrdataset);
} else {
result = dns_db_find(db, &name, version, type,
options, 0, &node, fname,
&rdataset, &sigrdataset);
}
if (!quiet) {
if (dbi != NULL)
printf("\n");
print_result("", result);
}
found_as = ISC_FALSE;
switch (result) {
case ISC_R_SUCCESS:
case DNS_R_GLUE:
case DNS_R_CNAME:
case DNS_R_ZONECUT:
break;
case DNS_R_DNAME:
case DNS_R_DELEGATION:
found_as = ISC_TRUE;
break;
case DNS_R_NXRRSET:
if (dns_rdataset_isassociated(&rdataset))
break;
if (dbi != NULL) {
if (holdmode) {
RUNTIME_CHECK(dbi->hold_count <
MAXHOLD);
dbi->hold_nodes[dbi->hold_count++] =
node;
node = NULL;
} else
dns_db_detachnode(db, &node);
} else {
dns_db_detachnode(db, &node);
dns_db_detach(&db);
}
continue;
case DNS_R_NXDOMAIN:
if (dns_rdataset_isassociated(&rdataset))
break;
/* FALLTHROUGH */
default:
if (dbi == NULL)
dns_db_detach(&db);
if (quiet)
print_result("", result);
continue;
}
if (found_as && !quiet) {
isc_buffer_init(&tb1, t1, sizeof(t1));
isc_buffer_init(&tb2, t2, sizeof(t2));
result = dns_name_totext(&name, ISC_FALSE, &tb1);
if (result != ISC_R_SUCCESS) {
print_result("", result);
dns_db_detachnode(db, &node);
if (dbi == NULL)
dns_db_detach(&db);
continue;
}
result = dns_name_totext(fname, ISC_FALSE, &tb2);
if (result != ISC_R_SUCCESS) {
print_result("", result);
dns_db_detachnode(db, &node);
if (dbi == NULL)
dns_db_detach(&db);
continue;
}
isc_buffer_usedregion(&tb1, &r1);
isc_buffer_usedregion(&tb2, &r2);
printf("found %.*s as %.*s\n",
(int)r1.length, r1.base,
(int)r2.length, r2.base);
}
if (printnode)
dns_db_printnode(db, node, stdout);
if (!found_as && type == dns_rdatatype_any) {
rdsiter = NULL;
result = dns_db_allrdatasets(db, node, version, 0,
&rdsiter);
if (result == ISC_R_SUCCESS) {
if (!quiet)
print_rdatasets(fname, rdsiter);
dns_rdatasetiter_destroy(&rdsiter);
} else
print_result("", result);
} else {
if (!quiet)
print_rdataset(fname, &rdataset);
if (dns_rdataset_isassociated(&sigrdataset)) {
if (!quiet)
print_rdataset(fname, &sigrdataset);
dns_rdataset_disassociate(&sigrdataset);
}
if (dbi != NULL && addmode && !found_as) {
rdataset.ttl++;
rdataset.trust = trust;
if (dns_db_iszone(db))
addopts = DNS_DBADD_MERGE;
else
addopts = 0;
result = dns_db_addrdataset(db, node, version,
0, &rdataset,
addopts, NULL);
if (result != ISC_R_SUCCESS)
print_result("", result);
if (printnode)
dns_db_printnode(db, node, stdout);
} else if (dbi != NULL && delmode && !found_as) {
result = dns_db_deleterdataset(db, node,
version, type,
0);
if (result != ISC_R_SUCCESS)
print_result("", result);
if (printnode)
dns_db_printnode(db, node, stdout);
}
dns_rdataset_disassociate(&rdataset);
}
if (dbi != NULL) {
if (holdmode) {
RUNTIME_CHECK(dbi->hold_count < MAXHOLD);
dbi->hold_nodes[dbi->hold_count++] = node;
node = NULL;
} else
dns_db_detachnode(db, &node);
} else {
dns_db_detachnode(db, &node);
dns_db_detach(&db);
}
}
if (time_lookups) {
isc_uint64_t usec;
TIME_NOW(&finish);
usec = isc_time_microdiff(&finish, &start);
printf("elapsed time: %lu.%06lu seconds\n",
(unsigned long)(usec / 1000000),
(unsigned long)(usec % 1000000));
}
unload_all();
dns_dbtable_detach(&dbtable);
if (lctx != NULL)
isc_log_destroy(&lctx);
if (!quiet)
isc_mem_stats(mctx, stdout);
return (0);
}