mirror of
https://github.com/isc-projects/bind9.git
synced 2026-02-23 18:04:10 -05:00
a290ed5bd0
we don't need two versions of dns_name_fromstring() any longer; we can just specify an origin value of dns_rootname for absolute names.
347 lines
11 KiB
C
347 lines
11 KiB
C
/*
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* SPDX-License-Identifier: MPL-2.0
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
*/
|
|
|
|
#include <inttypes.h>
|
|
#include <sched.h> /* IWYU pragma: keep */
|
|
#include <setjmp.h>
|
|
#include <stdarg.h>
|
|
#include <stdbool.h>
|
|
#include <stddef.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#define UNIT_TESTING
|
|
#include <cmocka.h>
|
|
|
|
#include <isc/buffer.h>
|
|
#include <isc/commandline.h>
|
|
#include <isc/md.h>
|
|
#include <isc/mem.h>
|
|
#include <isc/os.h>
|
|
#include <isc/thread.h>
|
|
#include <isc/urcu.h>
|
|
#include <isc/util.h>
|
|
#include <isc/uv.h>
|
|
|
|
#include <dns/badcache.h>
|
|
#include <dns/compress.h>
|
|
#include <dns/fixedname.h>
|
|
#include <dns/name.h>
|
|
#include <dns/rdatatype.h>
|
|
|
|
#include <tests/dns.h>
|
|
|
|
static uint32_t
|
|
crc32(const uint8_t *buf, size_t size) {
|
|
uint32_t crc;
|
|
crc = 0xFFFFFFFF;
|
|
while (size--) {
|
|
crc = crc ^ *buf++;
|
|
for (size_t j = 0; j < 8; j++) {
|
|
uint32_t mask = -(crc & 1);
|
|
crc = (crc >> 1) ^ (0xedb88320 & mask);
|
|
}
|
|
}
|
|
return (~crc);
|
|
}
|
|
|
|
#define BADCACHE_TEST_FLAG 1 << 3
|
|
|
|
ISC_LOOP_TEST_IMPL(basic) {
|
|
dns_badcache_t *bc = NULL;
|
|
dns_fixedname_t fname = { 0 };
|
|
dns_name_t *name = dns_fixedname_initname(&fname);
|
|
isc_stdtime_t now = isc_stdtime_now();
|
|
isc_result_t result;
|
|
uint32_t flags = BADCACHE_TEST_FLAG;
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
|
|
bc = dns_badcache_new(mctx);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
|
|
flags = 0;
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(flags, BADCACHE_TEST_FLAG);
|
|
|
|
flags = 0;
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_a, &flags, now);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
assert_int_equal(flags, 0);
|
|
|
|
dns_badcache_destroy(&bc);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(expire) {
|
|
dns_badcache_t *bc = NULL;
|
|
dns_fixedname_t fname = { 0 };
|
|
dns_name_t *name = dns_fixedname_initname(&fname);
|
|
isc_stdtime_t now = isc_stdtime_now();
|
|
isc_result_t result;
|
|
uint32_t flags = BADCACHE_TEST_FLAG;
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
|
|
bc = dns_badcache_new(mctx);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
dns_badcache_add(bc, name, dns_rdatatype_a, false, flags, now + 60);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(flags, BADCACHE_TEST_FLAG);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags,
|
|
now + 61);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_a, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(flags, BADCACHE_TEST_FLAG);
|
|
|
|
dns_badcache_add(bc, name, dns_rdatatype_a, true, flags, now + 120);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_a, &flags, now + 61);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(flags, BADCACHE_TEST_FLAG);
|
|
|
|
dns_badcache_destroy(&bc);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(print) {
|
|
dns_badcache_t *bc = NULL;
|
|
dns_fixedname_t fname = { 0 };
|
|
dns_name_t *name = dns_fixedname_initname(&fname);
|
|
isc_stdtime_t now = isc_stdtime_now();
|
|
isc_stdtime_t expire = now + 60;
|
|
uint32_t flags = BADCACHE_TEST_FLAG;
|
|
FILE *file = NULL;
|
|
uint8_t buf[4096];
|
|
size_t len;
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
|
|
bc = dns_badcache_new(mctx);
|
|
dns_badcache_add(bc, name, dns_rdatatype_a, false, flags, expire);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, expire);
|
|
|
|
file = fopen("./badcache.out", "w");
|
|
dns_badcache_print(bc, "badcache", file);
|
|
fclose(file);
|
|
|
|
file = fopen("./badcache.out", "r");
|
|
len = fread(buf, sizeof(buf[0]), ARRAY_SIZE(buf), file);
|
|
assert_int_equal(len, 68);
|
|
fclose(file);
|
|
|
|
/* Calculated as crc32 ./badcache.out */
|
|
assert_int_equal(crc32(buf, len), 0x7c96678f);
|
|
|
|
dns_badcache_destroy(&bc);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(flush) {
|
|
dns_badcache_t *bc = NULL;
|
|
dns_fixedname_t fname = { 0 };
|
|
dns_name_t *name = dns_fixedname_initname(&fname);
|
|
isc_stdtime_t now = isc_stdtime_now();
|
|
isc_result_t result;
|
|
uint32_t flags = BADCACHE_TEST_FLAG;
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
|
|
bc = dns_badcache_new(mctx);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_badcache_flush(bc);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
dns_badcache_destroy(&bc);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(flushname) {
|
|
dns_badcache_t *bc = NULL;
|
|
dns_fixedname_t fname = { 0 };
|
|
dns_name_t *name = dns_fixedname_initname(&fname);
|
|
isc_stdtime_t now = isc_stdtime_now();
|
|
isc_result_t result;
|
|
uint32_t flags = BADCACHE_TEST_FLAG;
|
|
|
|
bc = dns_badcache_new(mctx);
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_name_fromstring(name, "sub.example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_name_fromstring(name, "sub.sub.example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_name_fromstring(name, "sub.example.com.", NULL, 0, NULL);
|
|
dns_badcache_flushname(bc, name);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_a, &flags, now);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
dns_name_fromstring(name, "sub.sub.example.com.", NULL, 0, NULL);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_badcache_destroy(&bc);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(flushtree) {
|
|
dns_badcache_t *bc = NULL;
|
|
dns_fixedname_t fname = { 0 };
|
|
dns_name_t *name = dns_fixedname_initname(&fname);
|
|
isc_stdtime_t now = isc_stdtime_now();
|
|
isc_result_t result;
|
|
uint32_t flags = BADCACHE_TEST_FLAG;
|
|
|
|
bc = dns_badcache_new(mctx);
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(flags, BADCACHE_TEST_FLAG);
|
|
|
|
dns_name_fromstring(name, "sub.example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(flags, BADCACHE_TEST_FLAG);
|
|
|
|
dns_name_fromstring(name, "sub.sub.example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now + 60);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(flags, BADCACHE_TEST_FLAG);
|
|
|
|
dns_name_fromstring(name, "sub.example.com.", NULL, 0, NULL);
|
|
dns_badcache_flushtree(bc, name);
|
|
|
|
dns_name_fromstring(name, "sub.sub.example.com.", NULL, 0, NULL);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
dns_name_fromstring(name, "sub.example.com.", NULL, 0, NULL);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags, now);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(flags, BADCACHE_TEST_FLAG);
|
|
|
|
dns_badcache_destroy(&bc);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(purge) {
|
|
dns_badcache_t *bc = NULL;
|
|
dns_fixedname_t fname = { 0 };
|
|
dns_name_t *name = dns_fixedname_initname(&fname);
|
|
isc_stdtime_t now = isc_stdtime_now();
|
|
isc_result_t result;
|
|
uint32_t flags = BADCACHE_TEST_FLAG;
|
|
|
|
bc = dns_badcache_new(mctx);
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags,
|
|
now - 60);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_name_fromstring(name, "sub.example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags,
|
|
now - 60);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
dns_name_fromstring(name, "sub.sub.example.com.", NULL, 0, NULL);
|
|
dns_badcache_add(bc, name, dns_rdatatype_aaaa, false, flags, now);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags,
|
|
now - 60);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags,
|
|
now + 30);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
dns_name_fromstring(name, "sub.sub.example.com.", NULL, 0, NULL);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags,
|
|
now + 30);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
dns_name_fromstring(name, "sub.example.com.", NULL, 0, NULL);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags,
|
|
now + 30);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
dns_name_fromstring(name, "example.com.", NULL, 0, NULL);
|
|
result = dns_badcache_find(bc, name, dns_rdatatype_aaaa, &flags,
|
|
now + 30);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
|
|
dns_badcache_destroy(&bc);
|
|
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
ISC_TEST_LIST_START
|
|
ISC_TEST_ENTRY_CUSTOM(basic, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(expire, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(print, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(flush, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(flushname, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(flushtree, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(purge, setup_managers, teardown_managers)
|
|
ISC_TEST_LIST_END
|
|
|
|
ISC_TEST_MAIN
|