2
0
mirror of https://gitlab.isc.org/isc-projects/bind9 synced 2025-08-22 10:10:06 +00:00
bind/lib/dns/tests/name_test.c

813 lines
20 KiB
C
Raw Normal View History

/*
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
*
* 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 http://mozilla.org/MPL/2.0/.
*
* See the COPYRIGHT file distributed with this work for additional
* information regarding copyright ownership.
*/
2018-11-14 20:19:50 +08:00
#if HAVE_CMOCKA
#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <inttypes.h>
Include <sched.h> where necessary for musl libc All unit tests define the UNIT_TESTING macro, which causes <cmocka.h> to replace malloc(), calloc(), realloc(), and free() with its own functions tracking memory allocations. In order for this not to break compilation, the system header declaring the prototypes for these standard functions must be included before <cmocka.h>. Normally, these prototypes are only present in <stdlib.h>, so we make sure it is included before <cmocka.h>. However, musl libc also defines the prototypes for calloc() and free() in <sched.h>, which is included by <pthread.h>, which is included e.g. by <isc/mutex.h>. Thus, unit tests including "dnstest.h" (which includes <isc/mem.h>, which includes <isc/mutex.h>) after <cmocka.h> will not compile with musl libc as for these programs, <sched.h> will be included after <cmocka.h>. Always including <cmocka.h> after all other header files is not a feasible solution as that causes the mock assertion macros defined in <isc/util.h> to mangle the contents of <cmocka.h>, thus breaking compilation. We cannot really use the __noreturn__ or analyzer_noreturn attributes with cmocka assertion functions because they do return if the tested condition is true. The problem is that what BIND unit tests do is incompatible with Clang Static Analyzer's assumptions: since we use cmocka, our custom assertion handlers are present in a shared library (i.e. it is the cmocka library that checks the assertion condition, not a macro in unit test code). Redefining cmocka's assertion macros in <isc/util.h> is an ugly hack to overcome that problem - unfortunately, this is the only way we can think of to make Clang Static Analyzer properly process unit test code. Giving up on Clang Static Analyzer being able to properly process unit test code is not a satisfactory solution. Undefining _GNU_SOURCE for unit test code could work around the problem (musl libc's <sched.h> only defines the prototypes for calloc() and free() when _GNU_SOURCE is defined), but doing that could introduce discrepancies for unit tests including entire *.c files, so it is also not a good solution. All in all, including <sched.h> before <cmocka.h> for all affected unit tests seems to be the most benign way of working around this musl libc quirk. While quite an ugly solution, it achieves our goals here, which are to keep the benefit of proper static analysis of unit test code and to fix compilation against musl libc.
2019-07-30 21:08:40 +02:00
#include <sched.h> /* IWYU pragma: keep */
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
2018-11-14 20:19:50 +08:00
#define UNIT_TESTING
#include <cmocka.h>
#include <isc/buffer.h>
#include <isc/commandline.h>
#include <isc/mem.h>
2016-03-04 13:41:54 +05:30
#include <isc/os.h>
#include <isc/print.h>
2016-03-04 13:58:35 +05:30
#include <isc/thread.h>
#include <isc/util.h>
#include <dns/compress.h>
#include <dns/name.h>
#include <dns/fixedname.h>
#include "dnstest.h"
/* Set to true (or use -v option) for verbose output */
static bool verbose = false;
2018-11-14 20:19:50 +08:00
static int
_setup(void **state) {
isc_result_t result;
UNUSED(state);
result = dns_test_begin(NULL, false);
assert_int_equal(result, ISC_R_SUCCESS);
2018-11-14 20:19:50 +08:00
return (0);
}
2018-11-14 20:19:50 +08:00
static int
_teardown(void **state) {
UNUSED(state);
dns_test_end();
return (0);
}
/* dns_name_fullcompare test */
static void
fullcompare_test(void **state) {
dns_fixedname_t fixed1;
dns_fixedname_t fixed2;
dns_name_t *name1;
dns_name_t *name2;
dns_namereln_t relation;
int i;
isc_result_t result;
struct {
const char *name1;
const char *name2;
dns_namereln_t relation;
int order;
unsigned int nlabels;
} data[] = {
/* relative */
{ "", "", dns_namereln_equal, 0, 0 },
{ "foo", "", dns_namereln_subdomain, 1, 0 },
{ "", "foo", dns_namereln_contains, -1, 0 },
{ "foo", "bar", dns_namereln_none, 4, 0 },
{ "bar", "foo", dns_namereln_none, -4, 0 },
{ "bar.foo", "foo", dns_namereln_subdomain, 1, 1 },
{ "foo", "bar.foo", dns_namereln_contains, -1, 1 },
{ "baz.bar.foo", "bar.foo", dns_namereln_subdomain, 1, 2 },
{ "bar.foo", "baz.bar.foo", dns_namereln_contains, -1, 2 },
{ "foo.example", "bar.example", dns_namereln_commonancestor,
4, 1 },
/* absolute */
{ ".", ".", dns_namereln_equal, 0, 1 },
{ "foo.", "bar.", dns_namereln_commonancestor, 4, 1 },
{ "bar.", "foo.", dns_namereln_commonancestor, -4, 1 },
{ "foo.example.", "bar.example.", dns_namereln_commonancestor,
4, 2 },
{ "bar.foo.", "foo.", dns_namereln_subdomain, 1, 2 },
{ "foo.", "bar.foo.", dns_namereln_contains, -1, 2 },
{ "baz.bar.foo.", "bar.foo.", dns_namereln_subdomain, 1, 3 },
{ "bar.foo.", "baz.bar.foo.", dns_namereln_contains, -1, 3 },
{ NULL, NULL, dns_namereln_none, 0, 0 }
};
2018-11-14 20:19:50 +08:00
UNUSED(state);
name1 = dns_fixedname_initname(&fixed1);
name2 = dns_fixedname_initname(&fixed2);
for (i = 0; data[i].name1 != NULL; i++) {
int order = 3000;
unsigned int nlabels = 3000;
if (data[i].name1[0] == 0) {
dns_fixedname_init(&fixed1);
} else {
result = dns_name_fromstring2(name1, data[i].name1,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
}
if (data[i].name2[0] == 0) {
dns_fixedname_init(&fixed2);
} else {
result = dns_name_fromstring2(name2, data[i].name2,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
}
relation = dns_name_fullcompare(name1, name1, &order, &nlabels);
2018-11-14 20:19:50 +08:00
assert_int_equal(relation, dns_namereln_equal);
assert_int_equal(order, 0);
assert_int_equal(nlabels, name1->labels);
/* Some random initializer */
order = 3001;
nlabels = 3001;
relation = dns_name_fullcompare(name1, name2, &order, &nlabels);
2018-11-14 20:19:50 +08:00
assert_int_equal(relation, data[i].relation);
assert_int_equal(order, data[i].order);
assert_int_equal(nlabels, data[i].nlabels);
}
}
static void
compress_test(dns_name_t *name1, dns_name_t *name2, dns_name_t *name3,
unsigned char *expected, unsigned int length,
dns_compress_t *cctx, dns_decompress_t *dctx)
{
isc_buffer_t source;
isc_buffer_t target;
dns_name_t name;
unsigned char buf1[1024];
unsigned char buf2[1024];
isc_buffer_init(&source, buf1, sizeof(buf1));
isc_buffer_init(&target, buf2, sizeof(buf2));
2018-11-14 20:19:50 +08:00
assert_int_equal(dns_name_towire(name1, cctx, &source), ISC_R_SUCCESS);
2018-11-14 20:19:50 +08:00
assert_int_equal(dns_name_towire(name2, cctx, &source), ISC_R_SUCCESS);
assert_int_equal(dns_name_towire(name2, cctx, &source), ISC_R_SUCCESS);
assert_int_equal(dns_name_towire(name3, cctx, &source), ISC_R_SUCCESS);
isc_buffer_setactive(&source, source.used);
dns_name_init(&name, NULL);
2019-04-18 13:02:30 +10:00
RUNTIME_CHECK(dns_name_fromwire(&name, &source, dctx, 0,
&target) == ISC_R_SUCCESS);
2019-04-18 13:02:30 +10:00
RUNTIME_CHECK(dns_name_fromwire(&name, &source, dctx, 0,
&target) == ISC_R_SUCCESS);
2019-04-18 13:02:30 +10:00
RUNTIME_CHECK(dns_name_fromwire(&name, &source, dctx, 0,
&target) == ISC_R_SUCCESS);
2019-04-18 13:02:30 +10:00
RUNTIME_CHECK(dns_name_fromwire(&name, &source, dctx, 0,
&target) == ISC_R_SUCCESS);
dns_decompress_invalidate(dctx);
2018-11-14 20:19:50 +08:00
assert_int_equal(target.used, length);
assert_true(memcmp(target.base, expected, target.used) == 0);
}
2018-11-14 20:19:50 +08:00
/* name compression test */
static void
compression_test(void **state) {
unsigned int allowed;
dns_compress_t cctx;
dns_decompress_t dctx;
dns_name_t name1;
dns_name_t name2;
dns_name_t name3;
isc_region_t r;
unsigned char plain1[] = "\003yyy\003foo";
unsigned char plain2[] = "\003bar\003yyy\003foo";
unsigned char plain3[] = "\003xxx\003bar\003foo";
unsigned char plain[] = "\003yyy\003foo\0\003bar\003yyy\003foo\0\003"
"bar\003yyy\003foo\0\003xxx\003bar\003foo";
2018-11-14 20:19:50 +08:00
UNUSED(state);
dns_name_init(&name1, NULL);
r.base = plain1;
r.length = sizeof(plain1);
dns_name_fromregion(&name1, &r);
dns_name_init(&name2, NULL);
r.base = plain2;
r.length = sizeof(plain2);
dns_name_fromregion(&name2, &r);
dns_name_init(&name3, NULL);
r.base = plain3;
r.length = sizeof(plain3);
dns_name_fromregion(&name3, &r);
/* Test 1: NONE */
allowed = DNS_COMPRESS_NONE;
assert_int_equal(dns_compress_init(&cctx, -1, dt_mctx), ISC_R_SUCCESS);
dns_compress_setmethods(&cctx, allowed);
dns_decompress_init(&dctx, -1, DNS_DECOMPRESS_STRICT);
dns_decompress_setmethods(&dctx, allowed);
compress_test(&name1, &name2, &name3, plain, sizeof(plain),
&cctx, &dctx);
dns_compress_rollback(&cctx, 0);
dns_compress_invalidate(&cctx);
/* Test2: GLOBAL14 */
allowed = DNS_COMPRESS_GLOBAL14;
assert_int_equal(dns_compress_init(&cctx, -1, dt_mctx), ISC_R_SUCCESS);
dns_compress_setmethods(&cctx, allowed);
dns_decompress_init(&dctx, -1, DNS_DECOMPRESS_STRICT);
dns_decompress_setmethods(&dctx, allowed);
compress_test(&name1, &name2, &name3, plain, sizeof(plain),
&cctx, &dctx);
dns_compress_rollback(&cctx, 0);
dns_compress_invalidate(&cctx);
/* Test3: ALL */
allowed = DNS_COMPRESS_ALL;
assert_int_equal(dns_compress_init(&cctx, -1, dt_mctx), ISC_R_SUCCESS);
dns_compress_setmethods(&cctx, allowed);
dns_decompress_init(&dctx, -1, DNS_DECOMPRESS_STRICT);
dns_decompress_setmethods(&dctx, allowed);
compress_test(&name1, &name2, &name3, plain, sizeof(plain),
&cctx, &dctx);
dns_compress_rollback(&cctx, 0);
dns_compress_invalidate(&cctx);
/* Test4: NONE disabled */
allowed = DNS_COMPRESS_NONE;
assert_int_equal(dns_compress_init(&cctx, -1, dt_mctx), ISC_R_SUCCESS);
dns_compress_setmethods(&cctx, allowed);
dns_compress_disable(&cctx);
dns_decompress_init(&dctx, -1, DNS_DECOMPRESS_STRICT);
dns_decompress_setmethods(&dctx, allowed);
compress_test(&name1, &name2, &name3, plain, sizeof(plain),
&cctx, &dctx);
dns_compress_rollback(&cctx, 0);
dns_compress_invalidate(&cctx);
/* Test5: GLOBAL14 disabled */
allowed = DNS_COMPRESS_GLOBAL14;
assert_int_equal(dns_compress_init(&cctx, -1, dt_mctx), ISC_R_SUCCESS);
dns_compress_setmethods(&cctx, allowed);
dns_compress_disable(&cctx);
dns_decompress_init(&dctx, -1, DNS_DECOMPRESS_STRICT);
dns_decompress_setmethods(&dctx, allowed);
compress_test(&name1, &name2, &name3, plain, sizeof(plain),
&cctx, &dctx);
dns_compress_rollback(&cctx, 0);
dns_compress_invalidate(&cctx);
/* Test6: ALL disabled */
allowed = DNS_COMPRESS_ALL;
assert_int_equal(dns_compress_init(&cctx, -1, dt_mctx), ISC_R_SUCCESS);
dns_compress_setmethods(&cctx, allowed);
dns_compress_disable(&cctx);
dns_decompress_init(&dctx, -1, DNS_DECOMPRESS_STRICT);
dns_decompress_setmethods(&dctx, allowed);
compress_test(&name1, &name2, &name3, plain, sizeof(plain),
&cctx, &dctx);
dns_compress_rollback(&cctx, 0);
dns_compress_invalidate(&cctx);
}
2018-11-14 20:19:50 +08:00
/* is trust-anchor-telemetry test */
static void
istat_test(void **state) {
dns_fixedname_t fixed;
dns_name_t *name;
isc_result_t result;
size_t i;
struct {
const char *name;
bool istat;
} data[] = {
{ ".", false },
{ "_ta-", false },
{ "_ta-1234", true },
{ "_TA-1234", true },
{ "+TA-1234", false },
{ "_fa-1234", false },
{ "_td-1234", false },
{ "_ta_1234", false },
{ "_ta-g234", false },
{ "_ta-1h34", false },
{ "_ta-12i4", false },
{ "_ta-123j", false },
{ "_ta-1234-abcf", true },
{ "_ta-1234-abcf-ED89", true },
{ "_ta-12345-abcf-ED89", false },
{ "_ta-.example", false },
{ "_ta-1234.example", true },
{ "_ta-1234-abcf.example", true },
{ "_ta-1234-abcf-ED89.example", true },
{ "_ta-12345-abcf-ED89.example", false },
{ "_ta-1234-abcfe-ED89.example", false },
{ "_ta-1234-abcf-EcD89.example", false }
};
2018-11-14 20:19:50 +08:00
UNUSED(state);
name = dns_fixedname_initname(&fixed);
2018-11-14 20:19:50 +08:00
for (i = 0; i < (sizeof(data) / sizeof(data[0])); i++) {
result = dns_name_fromstring(name, data[i].name, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
assert_int_equal(dns_name_istat(name), data[i].istat);
}
}
2018-11-14 20:19:50 +08:00
/* dns_nane_init */
static void
init_test(void **state) {
dns_name_t name;
unsigned char offsets[1];
2018-11-14 20:19:50 +08:00
UNUSED(state);
dns_name_init(&name, offsets);
2018-11-14 20:19:50 +08:00
assert_null(name.ndata);
assert_int_equal(name.length, 0);
assert_int_equal(name.labels, 0);
assert_int_equal(name.attributes, 0);
assert_int_equal(name.offsets, offsets);
assert_null(name.buffer);
}
2018-11-14 20:19:50 +08:00
/* dns_nane_invalidate */
static void
invalidate_test(void **state) {
dns_name_t name;
unsigned char offsets[1];
2018-11-14 20:19:50 +08:00
UNUSED(state);
dns_name_init(&name, offsets);
dns_name_invalidate(&name);
2018-11-14 20:19:50 +08:00
assert_null(name.ndata);
assert_int_equal(name.length, 0);
assert_int_equal(name.labels, 0);
assert_int_equal(name.attributes, 0);
assert_null(name.offsets);
assert_null(name.buffer);
}
2018-11-14 20:19:50 +08:00
/* dns_nane_setbuffer/hasbuffer */
static void
buffer_test(void **state) {
dns_name_t name;
unsigned char buf[BUFSIZ];
isc_buffer_t b;
2018-11-14 20:19:50 +08:00
UNUSED(state);
isc_buffer_init(&b, buf, BUFSIZ);
dns_name_init(&name, NULL);
dns_name_setbuffer(&name, &b);
2018-11-14 20:19:50 +08:00
assert_int_equal(name.buffer, &b);
assert_true(dns_name_hasbuffer(&name));
}
2018-11-14 20:19:50 +08:00
/* dns_nane_isabsolute */
static void
isabsolute_test(void **state) {
struct {
const char *namestr;
bool expect;
} testcases[] = {
{ "x", false },
{ "a.b.c.d.", true },
{ "x.z", false}
};
unsigned int i;
2018-11-14 20:19:50 +08:00
UNUSED(state);
for (i = 0; i < (sizeof(testcases) / sizeof(testcases[0])); i++) {
isc_result_t result;
dns_name_t name;
unsigned char data[BUFSIZ];
isc_buffer_t b, nb;
size_t len;
len = strlen(testcases[i].namestr);
isc_buffer_constinit(&b, testcases[i].namestr, len);
isc_buffer_add(&b, len);
dns_name_init(&name, NULL);
isc_buffer_init(&nb, data, BUFSIZ);
dns_name_setbuffer(&name, &nb);
result = dns_name_fromtext(&name, &b, NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
2018-11-14 20:19:50 +08:00
assert_int_equal(dns_name_isabsolute(&name),
testcases[i].expect);
}
}
2018-11-14 20:19:50 +08:00
/* dns_nane_hash */
static void
hash_test(void **state) {
struct {
const char *name1;
const char *name2;
bool expect;
bool expecti;
} testcases[] = {
{ "a.b.c.d", "A.B.C.D", true, false },
{ "a.b.c.d.", "A.B.C.D.", true, false },
{ "a.b.c.d", "a.b.c.d", true, true },
{ "A.B.C.D.", "A.B.C.D.", true, false },
{ "x.y.z.w", "a.b.c.d", false, false },
{ "x.y.z.w.", "a.b.c.d.", false, false },
};
unsigned int i;
2018-11-14 20:19:50 +08:00
UNUSED(state);
for (i = 0; i < (sizeof(testcases) / sizeof(testcases[0])); i++) {
isc_result_t result;
dns_fixedname_t f1, f2;
dns_name_t *n1, *n2;
unsigned int h1, h2;
n1 = dns_fixedname_initname(&f1);
n2 = dns_fixedname_initname(&f2);
result = dns_name_fromstring2(n1, testcases[i].name1,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
result = dns_name_fromstring2(n2, testcases[i].name2,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
/* Check case-insensitive hashing first */
h1 = dns_name_hash(n1, false);
h2 = dns_name_hash(n2, false);
if (verbose) {
print_message("# %s hashes to %u, "
"%s to %u, case insensitive\n",
testcases[i].name1, h1,
testcases[i].name2, h2);
}
2018-11-14 20:19:50 +08:00
assert_int_equal((h1 == h2), testcases[i].expect);
/* Now case-sensitive */
h1 = dns_name_hash(n1, false);
h2 = dns_name_hash(n2, false);
if (verbose) {
print_message("# %s hashes to %u, "
"%s to %u, case sensitive\n",
testcases[i].name1, h1,
testcases[i].name2, h2);
}
2018-11-14 20:19:50 +08:00
assert_int_equal((h1 == h2), testcases[i].expect);
}
}
2018-11-14 20:19:50 +08:00
/* dns_nane_issubdomain */
static void
issubdomain_test(void **state) {
struct {
const char *name1;
const char *name2;
bool expect;
} testcases[] = {
{ "c.d", "a.b.c.d", false },
{ "c.d.", "a.b.c.d.", false },
{ "b.c.d", "c.d", true },
{ "a.b.c.d.", "c.d.", true },
{ "a.b.c", "a.b.c", true },
{ "a.b.c.", "a.b.c.", true },
{ "x.y.z", "a.b.c", false}
};
unsigned int i;
2018-11-14 20:19:50 +08:00
UNUSED(state);
for (i = 0; i < (sizeof(testcases) / sizeof(testcases[0])); i++) {
isc_result_t result;
dns_fixedname_t f1, f2;
dns_name_t *n1, *n2;
n1 = dns_fixedname_initname(&f1);
n2 = dns_fixedname_initname(&f2);
result = dns_name_fromstring2(n1, testcases[i].name1,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
result = dns_name_fromstring2(n2, testcases[i].name2,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
if (verbose) {
print_message("# check: %s %s a subdomain of %s\n",
testcases[i].name1,
testcases[i].expect ? "is" : "is not",
testcases[i].name2);
}
2018-11-14 20:19:50 +08:00
assert_int_equal(dns_name_issubdomain(n1, n2),
testcases[i].expect);
}
}
2018-11-14 20:19:50 +08:00
/* dns_nane_countlabels */
static void
countlabels_test(void **state) {
struct {
const char *namestr;
unsigned int expect;
} testcases[] = {
{ "c.d", 2 },
{ "c.d.", 3 },
{ "a.b.c.d.", 5 },
{ "a.b.c.d", 4 },
{ "a.b.c", 3 },
{ ".", 1 },
};
unsigned int i;
2018-11-14 20:19:50 +08:00
UNUSED(state);
for (i = 0; i < (sizeof(testcases) / sizeof(testcases[0])); i++) {
isc_result_t result;
dns_fixedname_t fname;
dns_name_t *name;
name = dns_fixedname_initname(&fname);
result = dns_name_fromstring2(name, testcases[i].namestr,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
if (verbose) {
print_message("# %s: expect %u labels\n",
testcases[i].namestr,
testcases[i].expect);
}
2018-11-14 20:19:50 +08:00
assert_int_equal(dns_name_countlabels(name),
testcases[i].expect);
}
}
2018-11-14 20:19:50 +08:00
/* dns_nane_getlabel */
static void
getlabel_test(void **state) {
struct {
const char *name1;
unsigned int pos1;
const char *name2;
unsigned int pos2;
} testcases[] = {
{ "c.d", 1, "a.b.c.d", 3 },
{ "a.b.c.d", 3, "c.d", 1 },
{ "a.b.c.", 3, "A.B.C.", 3 },
};
unsigned int i;
2018-11-14 20:19:50 +08:00
UNUSED(state);
for (i = 0; i < (sizeof(testcases) / sizeof(testcases[0])); i++) {
isc_result_t result;
dns_fixedname_t f1, f2;
dns_name_t *n1, *n2;
dns_label_t l1, l2;
unsigned int j;
n1 = dns_fixedname_initname(&f1);
n2 = dns_fixedname_initname(&f2);
result = dns_name_fromstring2(n1, testcases[i].name1,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
result = dns_name_fromstring2(n2, testcases[i].name2,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
dns_name_getlabel(n1, testcases[i].pos1, &l1);
dns_name_getlabel(n2, testcases[i].pos2, &l2);
2018-11-14 20:19:50 +08:00
assert_int_equal(l1.length, l2.length);
for (j = 0; j < l1.length; j++) {
assert_int_equal(l1.base[j], l2.base[j]);
}
}
}
2018-11-14 20:19:50 +08:00
/* dns_nane_getlabelsequence */
static void
getlabelsequence_test(void **state) {
struct {
const char *name1;
unsigned int pos1;
const char *name2;
unsigned int pos2;
unsigned int range;
} testcases[] = {
{ "c.d", 1, "a.b.c.d", 3, 1 },
{ "a.b.c.d.e", 2, "c.d", 0, 2 },
{ "a.b.c", 0, "a.b.c", 0, 3 },
};
unsigned int i;
2018-11-14 20:19:50 +08:00
UNUSED(state);
for (i = 0; i < (sizeof(testcases) / sizeof(testcases[0])); i++) {
isc_result_t result;
dns_name_t t1, t2;
dns_fixedname_t f1, f2;
dns_name_t *n1, *n2;
/* target names */
dns_name_init(&t1, NULL);
dns_name_init(&t2, NULL);
/* source names */
n1 = dns_fixedname_initname(&f1);
n2 = dns_fixedname_initname(&f2);
result = dns_name_fromstring2(n1, testcases[i].name1,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
result = dns_name_fromstring2(n2, testcases[i].name2,
NULL, 0, NULL);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
dns_name_getlabelsequence(n1, testcases[i].pos1,
testcases[i].range, &t1);
dns_name_getlabelsequence(n2, testcases[i].pos2,
testcases[i].range, &t2);
2018-11-14 20:19:50 +08:00
assert_true(dns_name_equal(&t1, &t2));
}
}
2016-03-04 13:41:54 +05:30
#ifdef DNS_BENCHMARK_TESTS
/*
* XXXMUKS: Don't delete this code. It is useful in benchmarking the
* name parser, but we don't require it as part of the unit test runs.
*/
2018-11-14 20:19:50 +08:00
/* Benchmark dns_name_fromwire() implementation */
2016-03-04 13:41:54 +05:30
static void *
fromwire_thread(void *arg) {
unsigned int maxval = 32000000;
uint8_t data[] = {
2016-03-04 13:41:54 +05:30
3, 'w', 'w', 'w',
7, 'e', 'x', 'a', 'm', 'p', 'l', 'e',
7, 'i', 'n', 'v', 'a', 'l', 'i', 'd',
0
};
unsigned char output_data[DNS_NAME_MAXWIRE];
isc_buffer_t source, target;
unsigned int i;
dns_decompress_t dctx;
UNUSED(arg);
dns_decompress_init(&dctx, -1, DNS_DECOMPRESS_STRICT);
dns_decompress_setmethods(&dctx, DNS_COMPRESS_NONE);
isc_buffer_init(&source, data, sizeof(data));
isc_buffer_add(&source, sizeof(data));
isc_buffer_init(&target, output_data, sizeof(output_data));
/* Parse 32 million names in each thread */
for (i = 0; i < maxval; i++) {
dns_name_t name;
isc_buffer_clear(&source);
isc_buffer_clear(&target);
isc_buffer_add(&source, sizeof(data));
isc_buffer_setactive(&source, sizeof(data));
dns_name_init(&name, NULL);
(void) dns_name_fromwire(&name, &source, &dctx, 0, &target);
}
return (NULL);
}
2018-11-14 20:19:50 +08:00
static void
benchmark_test(void **state) {
2016-03-04 13:41:54 +05:30
isc_result_t result;
unsigned int i;
isc_time_t ts1, ts2;
double t;
unsigned int nthreads;
2016-03-04 13:58:35 +05:30
isc_thread_t threads[32];
2016-03-04 13:41:54 +05:30
2018-11-14 20:19:50 +08:00
UNUSED(state);
2016-03-04 13:41:54 +05:30
debug_mem_record = false;
2016-03-04 13:41:54 +05:30
result = isc_time_now(&ts1);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
2016-03-04 13:41:54 +05:30
nthreads = ISC_MIN(isc_os_ncpus(), 32);
nthreads = ISC_MAX(nthreads, 1);
for (i = 0; i < nthreads; i++) {
isc_thread_create(fromwire_thread, NULL, &threads[i]);
2016-03-04 13:41:54 +05:30
}
for (i = 0; i < nthreads; i++) {
isc_thread_join(threads[i], NULL);
2016-03-04 13:41:54 +05:30
}
result = isc_time_now(&ts2);
2018-11-14 20:19:50 +08:00
assert_int_equal(result, ISC_R_SUCCESS);
2016-03-04 13:41:54 +05:30
t = isc_time_microdiff(&ts2, &ts1);
printf("%u dns_name_fromwire() calls, %f seconds, %f calls/second\n",
nthreads * 32000000, t / 1000000.0,
(nthreads * 32000000) / (t / 1000000.0));
}
#endif /* DNS_BENCHMARK_TESTS */
2018-11-14 20:19:50 +08:00
int
main(int argc, char **argv) {
2018-11-14 20:19:50 +08:00
const struct CMUnitTest tests[] = {
cmocka_unit_test(fullcompare_test),
cmocka_unit_test_setup_teardown(compression_test,
_setup, _teardown),
cmocka_unit_test(istat_test),
cmocka_unit_test(init_test),
cmocka_unit_test(invalidate_test),
cmocka_unit_test(buffer_test),
cmocka_unit_test(isabsolute_test),
cmocka_unit_test(hash_test),
cmocka_unit_test(issubdomain_test),
cmocka_unit_test(countlabels_test),
cmocka_unit_test(getlabel_test),
cmocka_unit_test(getlabelsequence_test),
2016-03-04 13:41:54 +05:30
#ifdef DNS_BENCHMARK_TESTS
2018-11-14 20:19:50 +08:00
cmocka_unit_test_setup_teardown(benchmark_test,
_setup, _teardown),
2016-03-04 13:41:54 +05:30
#endif /* DNS_BENCHMARK_TESTS */
2018-11-14 20:19:50 +08:00
};
int c;
while ((c = isc_commandline_parse(argc, argv, "v")) != -1) {
switch (c) {
case 'v':
verbose = true;
break;
default:
break;
}
}
2018-11-14 20:19:50 +08:00
return (cmocka_run_group_tests(tests, NULL, NULL));
}
2018-11-14 20:19:50 +08:00
#else /* HAVE_CMOCKA */
#include <stdio.h>
int
main(void) {
printf("1..0 # Skipped: cmocka not available\n");
return (0);
}
#endif