dns/tests/dbiterator_test.c

	dbiterator_test.c revision d64e3b29bd1b66a4bc80b60ef87a5bafe741d7ca
/*
 * Copyright (C) 2011  Internet Systems Consortium, Inc. ("ISC")
 *
 * Permission to use, copy, modify, and/or 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 ISC DISCLAIMS ALL WARRANTIES WITH
 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS.  IN NO EVENT SHALL ISC 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.
 */

/* $Id: dbiterator_test.c,v 1.5 2011/08/29 23:44:07 marka Exp $ */

/*! \file */

#include <config.h>

#include <atf-c.h>

#include <unistd.h>
#include <stdlib.h>

#include <dns/db.h>
#include <dns/dbiterator.h>
#include <dns/name.h>

#include "dnstest.h"

/*
 * Helper functions
 */

#define BUFLEN      255
#define BIGBUFLEN   (64 * 1024)
#define TEST_ORIGIN "test"

static isc_result_t
setup_db(const char *testfile, dns_dbtype_t dbtype, dns_db_t **db) {
    isc_result_t        result;
    int         len;
    char            origin[sizeof(TEST_ORIGIN)];
    dns_name_t      dns_origin;
    isc_buffer_t        source;
    isc_buffer_t        target;
    unsigned char       name_buf[BUFLEN];

    strcpy(origin, TEST_ORIGIN);
    len = strlen(origin);
    isc_buffer_init(&source, origin, len);
    isc_buffer_add(&source, len);
    isc_buffer_setactive(&source, len);
    isc_buffer_init(&target, name_buf, BUFLEN);
    dns_name_init(&dns_origin, NULL);

    result = dns_name_fromtext(&dns_origin, &source, dns_rootname,
                   0, &target);
    if (result != ISC_R_SUCCESS)
        return(result);

    result = dns_db_create(mctx, "rbt", &dns_origin, dbtype,
                   dns_rdataclass_in, 0, NULL, db);
    if (result != ISC_R_SUCCESS)
        return (result);

    /*
     * atf-run changes us to a /tmp directory, so tests
     * that access test data files must first chdir to the proper
     * location.
     */
    if (chdir(TESTS) == -1)
        return (ISC_R_FAILURE);

    result = dns_db_load(*db, testfile);
    return (result);
}

static isc_result_t
make_name(const char *src, dns_name_t *name) {
    isc_buffer_t b;
    isc_buffer_init(&b, src, strlen(src));
    isc_buffer_add(&b, strlen(src));
    return (dns_name_fromtext(name, &b, dns_rootname, 0, NULL));
}

/*
 * Individual unit tests
 */

/* create: make sure we can create a dbiterator */
static void
test_create(const atf_tc_t *tc) {
    isc_result_t result;
    dns_db_t *db = NULL;
    dns_dbiterator_t *iter = NULL;

    result = dns_test_begin(NULL, ISC_FALSE);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = setup_db(atf_tc_get_md_var(tc, "X-filename"),
              dns_dbtype_cache, &db);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_db_createiterator(db, 0, &iter);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    dns_dbiterator_destroy(&iter);
    dns_db_detach(&db);
    dns_test_end();
}

ATF_TC(create);
ATF_TC_HEAD(create, tc) {
    atf_tc_set_md_var(tc, "descr", "create a database iterator");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone1.data");
}
ATF_TC_BODY(create, tc) {
    test_create(tc);
}

ATF_TC(create_nsec3);
ATF_TC_HEAD(create_nsec3, tc) {
    atf_tc_set_md_var(tc, "descr", "create a database iterator (NSEC3)");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone2.data");
}
ATF_TC_BODY(create_nsec3, tc) {
    test_create(tc);
}

/* walk: walk a database */
static void
test_walk(const atf_tc_t *tc) {
    isc_result_t result;
    dns_db_t *db = NULL;
    dns_dbiterator_t *iter = NULL;
    dns_dbnode_t *node = NULL;
    dns_name_t *name;
    dns_fixedname_t f;
    int i = 0;

    UNUSED(tc);

    dns_fixedname_init(&f);
    name = dns_fixedname_name(&f);

    result = dns_test_begin(NULL, ISC_FALSE);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = setup_db(atf_tc_get_md_var(tc, "X-filename"),
              dns_dbtype_cache, &db);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_db_createiterator(db, 0, &iter);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    for (result = dns_dbiterator_first(iter);
         result == ISC_R_SUCCESS;
         result = dns_dbiterator_next(iter)) {
        result = dns_dbiterator_current(iter, &node, name);
        if (result == DNS_R_NEWORIGIN)
            result = ISC_R_SUCCESS;
        ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
        dns_db_detachnode(db, &node);
        i++;
    }

    ATF_CHECK_EQ(i, atoi(atf_tc_get_md_var(tc, "X-nodes")));

    dns_dbiterator_destroy(&iter);
    dns_db_detach(&db);
    dns_test_end();
}

ATF_TC(walk);
ATF_TC_HEAD(walk, tc) {
    atf_tc_set_md_var(tc, "descr", "walk database");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone1.data");
    atf_tc_set_md_var(tc, "X-nodes", "12");
}
ATF_TC_BODY(walk, tc) {
    test_walk(tc);
}

ATF_TC(walk_nsec3);
ATF_TC_HEAD(walk_nsec3, tc) {
    atf_tc_set_md_var(tc, "descr", "walk database");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone2.data");
    atf_tc_set_md_var(tc, "X-nodes", "33");
}
ATF_TC_BODY(walk_nsec3, tc) {
    test_walk(tc);
}

/* reverse: walk database backwards */
static void test_reverse(const atf_tc_t *tc) {
    isc_result_t result;
    dns_db_t *db = NULL;
    dns_dbiterator_t *iter = NULL;
    dns_dbnode_t *node = NULL;
    dns_name_t *name;
    dns_fixedname_t f;
    int i = 0;

    UNUSED(tc);

    dns_fixedname_init(&f);
    name = dns_fixedname_name(&f);

    result = dns_test_begin(NULL, ISC_FALSE);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = setup_db(atf_tc_get_md_var(tc, "X-filename"),
              dns_dbtype_cache, &db);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_db_createiterator(db, 0, &iter);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    for (result = dns_dbiterator_last(iter);
         result == ISC_R_SUCCESS;
         result = dns_dbiterator_prev(iter)) {
        result = dns_dbiterator_current(iter, &node, name);
        if (result == DNS_R_NEWORIGIN)
            result = ISC_R_SUCCESS;
        ATF_CHECK_EQ(result, ISC_R_SUCCESS);
        dns_db_detachnode(db, &node);
        i++;
    }

    ATF_CHECK_EQ(i, 12);

    dns_dbiterator_destroy(&iter);
    dns_db_detach(&db);
    dns_test_end();
}

ATF_TC(reverse);
ATF_TC_HEAD(reverse, tc) {
    atf_tc_set_md_var(tc, "descr", "walk database backwards");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone1.data");
}
ATF_TC_BODY(reverse, tc) {
    test_reverse(tc);
}

ATF_TC(reverse_nsec3);
ATF_TC_HEAD(reverse_nsec3, tc) {
    atf_tc_set_md_var(tc, "descr", "walk database backwards");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone2.data");
}
ATF_TC_BODY(reverse_nsec3, tc) {
    test_reverse(tc);
}

/* seek: walk database starting at a particular node */
static void test_seek(const atf_tc_t *tc) {
    isc_result_t result;
    dns_db_t *db = NULL;
    dns_dbiterator_t *iter = NULL;
    dns_dbnode_t *node = NULL;
    dns_name_t *name, *seekname;
    dns_fixedname_t f1, f2;
    int i = 0;

    UNUSED(tc);

    dns_fixedname_init(&f1);
    name = dns_fixedname_name(&f1);
    dns_fixedname_init(&f2);
    seekname = dns_fixedname_name(&f2);

    result = dns_test_begin(NULL, ISC_FALSE);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = setup_db(atf_tc_get_md_var(tc, "X-filename"),
              dns_dbtype_cache, &db);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_db_createiterator(db, 0, &iter);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = make_name("c." TEST_ORIGIN, seekname);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_dbiterator_seek(iter, seekname);
    ATF_CHECK_EQ(result, ISC_R_SUCCESS);

    while (result == ISC_R_SUCCESS) {
        result = dns_dbiterator_current(iter, &node, name);
        if (result == DNS_R_NEWORIGIN)
            result = ISC_R_SUCCESS;
        ATF_CHECK_EQ(result, ISC_R_SUCCESS);
        dns_db_detachnode(db, &node);
        result = dns_dbiterator_next(iter);
        i++;
    }

    ATF_CHECK_EQ(i, atoi(atf_tc_get_md_var(tc, "X-nodes")));

    dns_dbiterator_destroy(&iter);
    dns_db_detach(&db);
    dns_test_end();
}

ATF_TC(seek);
ATF_TC_HEAD(seek, tc) {
    atf_tc_set_md_var(tc, "descr", "walk database starting at "
                       "a particular node");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone1.data");
    atf_tc_set_md_var(tc, "X-nodes", "9");
}
ATF_TC_BODY(seek, tc) {
    test_seek(tc);
}

ATF_TC(seek_nsec3);
ATF_TC_HEAD(seek_nsec3, tc) {
    atf_tc_set_md_var(tc, "descr", "walk database starting at "
                       "a particular node");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone2.data");
    atf_tc_set_md_var(tc, "X-nodes", "30");
}
ATF_TC_BODY(seek_nsec3, tc) {
    test_seek(tc);
}

/*
 * seek_emty: walk database starting at an empty nonterminal node
 * (should fail)
 */
static void test_seek_empty(const atf_tc_t *tc) {
    isc_result_t result;
    dns_db_t *db = NULL;
    dns_dbiterator_t *iter = NULL;
    dns_name_t *seekname;
    dns_fixedname_t f1;

    UNUSED(tc);

    dns_fixedname_init(&f1);
    seekname = dns_fixedname_name(&f1);

    result = dns_test_begin(NULL, ISC_FALSE);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = setup_db(atf_tc_get_md_var(tc, "X-filename"),
              dns_dbtype_cache, &db);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_db_createiterator(db, 0, &iter);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = make_name("d." TEST_ORIGIN, seekname);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_dbiterator_seek(iter, seekname);
    ATF_CHECK_EQ(result, ISC_R_NOTFOUND);

    dns_dbiterator_destroy(&iter);
    dns_db_detach(&db);
    dns_test_end();
}

ATF_TC(seek_empty);
ATF_TC_HEAD(seek_empty, tc) {
    atf_tc_set_md_var(tc, "descr", "walk database starting at an "
                       "empty nonterminal node");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone1.data");
}
ATF_TC_BODY(seek_empty, tc) {
    test_seek_empty(tc);
}

ATF_TC(seek_empty_nsec3);
ATF_TC_HEAD(seek_empty_nsec3, tc) {
    atf_tc_set_md_var(tc, "descr", "walk database starting at an "
                       "empty nonterminal node");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone2.data");
}
ATF_TC_BODY(seek_empty_nsec3, tc) {
    test_seek_empty(tc);
}

/*
 * seek_emty: walk database starting at an empty nonterminal node
 * (should fail)
 */
static void test_seek_nx(const atf_tc_t *tc) {
    isc_result_t result;
    dns_db_t *db = NULL;
    dns_dbiterator_t *iter = NULL;
    dns_name_t *seekname;
    dns_fixedname_t f1;

    UNUSED(tc);

    dns_fixedname_init(&f1);
    seekname = dns_fixedname_name(&f1);

    result = dns_test_begin(NULL, ISC_FALSE);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = setup_db(atf_tc_get_md_var(tc, "X-filename"),
              dns_dbtype_cache, &db);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_db_createiterator(db, 0, &iter);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = make_name("nonexistent." TEST_ORIGIN, seekname);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    result = dns_dbiterator_seek(iter, seekname);
    ATF_CHECK_EQ(result, ISC_R_NOTFOUND);

    dns_dbiterator_destroy(&iter);
    dns_db_detach(&db);
    dns_test_end();
}

ATF_TC(seek_nx);
ATF_TC_HEAD(seek_nx, tc) {
    atf_tc_set_md_var(tc, "descr", "attempt to walk database starting "
                       "at a nonexistent node");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone1.data");
}
ATF_TC_BODY(seek_nx, tc) {
    test_seek_nx(tc);
}

ATF_TC(seek_nx_nsec3);
ATF_TC_HEAD(seek_nx_nsec3, tc) {
    atf_tc_set_md_var(tc, "descr", "attempt to walk database starting "
                       "at a nonexistent node");
    atf_tc_set_md_var(tc, "X-filename", "testdata/dbiterator/zone2.data");
}
ATF_TC_BODY(seek_nx_nsec3, tc) {
    test_seek_nx(tc);
}

/*
 * Main
 */
ATF_TP_ADD_TCS(tp) {
    ATF_TP_ADD_TC(tp, create);
    ATF_TP_ADD_TC(tp, create_nsec3);
    ATF_TP_ADD_TC(tp, walk);
    ATF_TP_ADD_TC(tp, walk_nsec3);
    ATF_TP_ADD_TC(tp, reverse);
    ATF_TP_ADD_TC(tp, reverse_nsec3);
    ATF_TP_ADD_TC(tp, seek);
    ATF_TP_ADD_TC(tp, seek_nsec3);
    ATF_TP_ADD_TC(tp, seek_empty);
    ATF_TP_ADD_TC(tp, seek_empty_nsec3);
    ATF_TP_ADD_TC(tp, seek_nx);
    ATF_TP_ADD_TC(tp, seek_nx_nsec3);
    return (atf_no_error());
}

/*
 * XXX:
 * dns_dbiterator API calls that are not yet part of this unit test:
 *
 * dns_dbiterator_pause
 * dns_dbiterator_origin
 * dns_dbiterator_setcleanmode
 */