mirror of
https://gitlab.isc.org/isc-projects/bind9
synced 2025-08-31 14:35:26 +00:00
add nxtify
This commit is contained in:
Bob Halley
@@ -37,6 +37,7 @@ LIBS = @LIBS@
|
||||
|
||||
SUBDIRS = master names rbt tasks db timers mem dst
|
||||
TARGETS = sdig \
|
||||
nxtify \
|
||||
res_test \
|
||||
lex_test \
|
||||
name_test \
|
||||
@@ -57,6 +58,7 @@ TARGETS = sdig \
|
||||
dispatch_tcp_test
|
||||
|
||||
SRCS = sdig.c \
|
||||
nxtify.c \
|
||||
res_test.c \
|
||||
lex_test.c \
|
||||
name_test.c \
|
||||
@@ -83,6 +85,10 @@ sdig: sdig.o printmsg.o ${ISCDEPLIBS} ${DNSDEPLIBS}
|
||||
${LIBTOOL} ${CC} ${CFLAGS} -o $@ sdig.o printmsg.o \
|
||||
${DNSLIBS} ${ISCLIBS} ${LIBS}
|
||||
|
||||
nxtify: nxtify.o ${ISCDEPLIBS} ${DNSDEPLIBS}
|
||||
${LIBTOOL} ${CC} ${CFLAGS} -o $@ nxtify.o \
|
||||
${DNSLIBS} ${ISCLIBS} ${LIBS}
|
||||
|
||||
res_test: res_test.o ${ISCDEPLIBS} ${DNSDEPLIBS}
|
||||
${LIBTOOL} ${CC} ${CFLAGS} -o $@ res_test.o \
|
||||
${DNSLIBS} ${ISCLIBS} ${LIBS}
|
||||
|
||||
285
bin/tests/nxtify.c
Normal file
285
bin/tests/nxtify.c
Normal file
@@ -0,0 +1,285 @@
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include <isc/types.h>
|
||||
#include <isc/assertions.h>
|
||||
#include <isc/buffer.h>
|
||||
#include <isc/error.h>
|
||||
#include <isc/mem.h>
|
||||
|
||||
#include <dns/types.h>
|
||||
#include <dns/name.h>
|
||||
#include <dns/fixedname.h>
|
||||
#include <dns/db.h>
|
||||
#include <dns/dbiterator.h>
|
||||
#include <dns/rdata.h>
|
||||
#include <dns/rdatalist.h>
|
||||
#include <dns/rdataset.h>
|
||||
#include <dns/rdatasetiter.h>
|
||||
#include <dns/result.h>
|
||||
|
||||
static isc_mem_t *mctx = NULL;
|
||||
|
||||
static inline void
|
||||
fatal(char *message) {
|
||||
fprintf(stderr, "%s\n", message);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
static inline void
|
||||
check_result(isc_result_t result, char *message) {
|
||||
if (result != ISC_R_SUCCESS) {
|
||||
fprintf(stderr, "%s: %s\n", message,
|
||||
isc_result_totext(result));
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
set_bit(unsigned char *array, unsigned int index, unsigned int bit) {
|
||||
unsigned int byte, shift, mask;
|
||||
|
||||
byte = array[index / 8];
|
||||
shift = 7 - (index % 8);
|
||||
mask = 1 << shift;
|
||||
|
||||
if (bit)
|
||||
array[index / 8] |= mask;
|
||||
else
|
||||
array[index / 8] &= (~mask & 0xFF);
|
||||
}
|
||||
|
||||
/*
|
||||
* XXXRTH Something like this will become a library function.
|
||||
*/
|
||||
static void
|
||||
build_nxt(dns_db_t *db, dns_dbversion_t *version, dns_dbnode_t *node,
|
||||
dns_name_t *target)
|
||||
{
|
||||
isc_result_t result;
|
||||
dns_rdata_t rdata;
|
||||
dns_rdataset_t rdataset;
|
||||
dns_rdatalist_t rdatalist;
|
||||
isc_region_t r;
|
||||
unsigned char data[256 + 16];
|
||||
unsigned char *nxt_bits;
|
||||
unsigned int max_type;
|
||||
dns_rdatasetiter_t *rdsiter;
|
||||
|
||||
memset(data, 0, sizeof data);
|
||||
dns_name_toregion(target, &r);
|
||||
memcpy(data, r.base, r.length);
|
||||
r.base = data;
|
||||
nxt_bits = r.base + r.length;
|
||||
set_bit(nxt_bits, dns_rdatatype_nxt, 1);
|
||||
max_type = dns_rdatatype_nxt;
|
||||
dns_rdataset_init(&rdataset);
|
||||
rdsiter = NULL;
|
||||
result = dns_db_allrdatasets(db, node, version, 0, &rdsiter);
|
||||
check_result(result, "dns_db_allrdatasets()");
|
||||
result = dns_rdatasetiter_first(rdsiter);
|
||||
while (result == ISC_R_SUCCESS) {
|
||||
dns_rdatasetiter_current(rdsiter, &rdataset);
|
||||
if (rdataset.type > 127)
|
||||
fatal("rdataset type too large");
|
||||
if (rdataset.type != dns_rdatatype_nxt) {
|
||||
if (rdataset.type > max_type)
|
||||
max_type = rdataset.type;
|
||||
set_bit(nxt_bits, rdataset.type, 1);
|
||||
}
|
||||
dns_rdataset_disassociate(&rdataset);
|
||||
result = dns_rdatasetiter_next(rdsiter);
|
||||
}
|
||||
if (result != DNS_R_NOMORE)
|
||||
fatal("rdataset iteration failed");
|
||||
dns_rdatasetiter_destroy(&rdsiter);
|
||||
|
||||
dns_rdata_init(&rdata);
|
||||
r.length += (max_type / 8);
|
||||
if (max_type % 8 != 0)
|
||||
r.length++;
|
||||
dns_rdata_fromregion(&rdata, dns_rdataclass_in,
|
||||
dns_rdatatype_nxt,
|
||||
&r);
|
||||
rdatalist.rdclass = dns_rdataclass_in;
|
||||
rdatalist.type = dns_rdatatype_nxt;
|
||||
rdatalist.ttl = 3600; /* XXXRTH */
|
||||
ISC_LIST_INIT(rdatalist.rdata);
|
||||
ISC_LIST_APPEND(rdatalist.rdata, &rdata, link);
|
||||
result = dns_rdatalist_tordataset(&rdatalist, &rdataset);
|
||||
check_result(result, "dns_rdatalist_tordataset");
|
||||
result = dns_db_addrdataset(db, node, version, 0, &rdataset,
|
||||
ISC_FALSE, NULL);
|
||||
if (result == DNS_R_UNCHANGED)
|
||||
result = ISC_R_SUCCESS;
|
||||
check_result(result, "dns_db_addrdataset");
|
||||
dns_rdataset_disassociate(&rdataset);
|
||||
}
|
||||
|
||||
static inline isc_boolean_t
|
||||
active_node(dns_db_t *db, dns_dbversion_t *version, dns_dbnode_t *node) {
|
||||
dns_rdatasetiter_t *rdsiter;
|
||||
isc_boolean_t active = ISC_FALSE;
|
||||
isc_result_t result;
|
||||
dns_rdataset_t rdataset;
|
||||
|
||||
dns_rdataset_init(&rdataset);
|
||||
rdsiter = NULL;
|
||||
result = dns_db_allrdatasets(db, node, version, 0, &rdsiter);
|
||||
check_result(result, "dns_db_allrdatasets()");
|
||||
result = dns_rdatasetiter_first(rdsiter);
|
||||
while (result == ISC_R_SUCCESS) {
|
||||
dns_rdatasetiter_current(rdsiter, &rdataset);
|
||||
if (rdataset.type != dns_rdatatype_nxt)
|
||||
active = ISC_TRUE;
|
||||
dns_rdataset_disassociate(&rdataset);
|
||||
if (!active)
|
||||
result = dns_rdatasetiter_next(rdsiter);
|
||||
else
|
||||
result = DNS_R_NOMORE;
|
||||
}
|
||||
if (result != DNS_R_NOMORE)
|
||||
fatal("rdataset iteration failed");
|
||||
dns_rdatasetiter_destroy(&rdsiter);
|
||||
|
||||
if (!active) {
|
||||
/*
|
||||
* Make sure there is no NXT record for this node.
|
||||
*/
|
||||
result = dns_db_deleterdataset(db, node, version,
|
||||
dns_rdatatype_nxt);
|
||||
if (result == DNS_R_UNCHANGED)
|
||||
result = ISC_R_SUCCESS;
|
||||
check_result(result, "dns_db_deleterdataset");
|
||||
}
|
||||
|
||||
return (active);
|
||||
}
|
||||
|
||||
static inline isc_result_t
|
||||
next_active(dns_db_t *db, dns_dbversion_t *version, dns_dbiterator_t *dbiter,
|
||||
dns_name_t *name, dns_dbnode_t **nodep)
|
||||
{
|
||||
isc_result_t result;
|
||||
isc_boolean_t active;
|
||||
|
||||
do {
|
||||
active = ISC_FALSE;
|
||||
result = dns_dbiterator_current(dbiter, nodep, name);
|
||||
if (result == ISC_R_SUCCESS) {
|
||||
active = active_node(db, version, *nodep);
|
||||
if (!active) {
|
||||
dns_db_detachnode(db, nodep);
|
||||
result = dns_dbiterator_next(dbiter);
|
||||
}
|
||||
}
|
||||
} while (result == ISC_R_SUCCESS && !active);
|
||||
|
||||
return (result);
|
||||
}
|
||||
|
||||
static void
|
||||
nxtify(char *filename) {
|
||||
isc_result_t result;
|
||||
dns_db_t *db;
|
||||
dns_dbversion_t *cversion, *wversion;
|
||||
dns_dbnode_t *node, *nextnode;
|
||||
char *origintext;
|
||||
dns_fixedname_t fname, fnextname;
|
||||
dns_name_t *name, *nextname, *target;
|
||||
isc_buffer_t b;
|
||||
size_t len;
|
||||
dns_dbiterator_t *dbiter;
|
||||
char newfilename[1024];
|
||||
|
||||
dns_fixedname_init(&fname);
|
||||
name = dns_fixedname_name(&fname);
|
||||
dns_fixedname_init(&fnextname);
|
||||
nextname = dns_fixedname_name(&fnextname);
|
||||
|
||||
origintext = strrchr(filename, '/');
|
||||
if (origintext == NULL)
|
||||
origintext = filename;
|
||||
else
|
||||
origintext++; /* Skip '/'. */
|
||||
len = strlen(origintext);
|
||||
isc_buffer_init(&b, origintext, len, ISC_BUFFERTYPE_TEXT);
|
||||
isc_buffer_add(&b, len);
|
||||
result = dns_name_fromtext(name, &b, dns_rootname, ISC_FALSE, NULL);
|
||||
check_result(result, "dns_name_fromtext()");
|
||||
|
||||
db = NULL;
|
||||
result = dns_db_create(mctx, "rbt", name, ISC_FALSE,
|
||||
dns_rdataclass_in, 0, NULL, &db);
|
||||
check_result(result, "dns_db_create()");
|
||||
result = dns_db_load(db, filename);
|
||||
check_result(result, "dns_db_load()");
|
||||
cversion = NULL;
|
||||
dns_db_currentversion(db, &cversion);
|
||||
wversion = NULL;
|
||||
result = dns_db_newversion(db, &wversion);
|
||||
check_result(result, "dns_db_newversion()");
|
||||
dbiter = NULL;
|
||||
result = dns_db_createiterator(db, ISC_FALSE, &dbiter);
|
||||
check_result(result, "dns_db_createiterator()");
|
||||
result = dns_dbiterator_first(dbiter);
|
||||
node = NULL;
|
||||
result = next_active(db, wversion, dbiter, name, &node);
|
||||
while (result == ISC_R_SUCCESS) {
|
||||
nextnode = NULL;
|
||||
result = dns_dbiterator_next(dbiter);
|
||||
if (result == ISC_R_SUCCESS)
|
||||
result = next_active(db, wversion, dbiter, nextname,
|
||||
&nextnode);
|
||||
if (result == ISC_R_SUCCESS)
|
||||
target = nextname;
|
||||
else if (result == DNS_R_NOMORE)
|
||||
target = dns_db_origin(db);
|
||||
else {
|
||||
target = NULL; /* Make compiler happy. */
|
||||
fatal("db iteration failed");
|
||||
}
|
||||
build_nxt(db, wversion, node, target);
|
||||
dns_db_detachnode(db, &node);
|
||||
node = nextnode;
|
||||
}
|
||||
if (result != DNS_R_NOMORE)
|
||||
fatal("db iteration failed");
|
||||
dns_dbiterator_destroy(&dbiter);
|
||||
/*
|
||||
* XXXRTH For now, we don't increment the SOA serial.
|
||||
*/
|
||||
dns_db_closeversion(db, &cversion, ISC_FALSE);
|
||||
dns_db_closeversion(db, &wversion, ISC_TRUE);
|
||||
len = strlen(filename);
|
||||
if (len + 4 + 1 > sizeof newfilename)
|
||||
fatal("filename too long");
|
||||
sprintf(newfilename, "%s.new", filename);
|
||||
result = dns_db_dump(db, NULL, newfilename);
|
||||
check_result(result, "dns_db_dump");
|
||||
dns_db_detach(&db);
|
||||
}
|
||||
|
||||
int
|
||||
main(int argc, char *argv[]) {
|
||||
int i;
|
||||
isc_result_t result;
|
||||
|
||||
dns_result_register();
|
||||
|
||||
result = isc_mem_create(0, 0, &mctx);
|
||||
check_result(result, "isc_mem_create()");
|
||||
|
||||
argc--;
|
||||
argv++;
|
||||
|
||||
for (i = 0; i < argc; i++)
|
||||
nxtify(argv[i]);
|
||||
|
||||
/* isc_mem_stats(mctx, stdout); */
|
||||
isc_mem_destroy(&mctx);
|
||||
|
||||
return (0);
|
||||
}
|
||||
Reference in New Issue
Block a user