diff options
Diffstat (limited to 'tests/ns/listenlist_test.c')
-rw-r--r-- | tests/ns/listenlist_test.c | 119 |
1 files changed, 119 insertions, 0 deletions
diff --git a/tests/ns/listenlist_test.c b/tests/ns/listenlist_test.c new file mode 100644 index 0000000..53d5783 --- /dev/null +++ b/tests/ns/listenlist_test.c @@ -0,0 +1,119 @@ +/* + * 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 <stddef.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> + +#define UNIT_TESTING +#include <cmocka.h> + +#include <isc/list.h> +#include <isc/print.h> +#include <isc/random.h> +#include <isc/util.h> + +#include <dns/acl.h> + +#include <ns/listenlist.h> + +#include <tests/ns.h> + +static int +_setup(void **state) { + isc__nm_force_tid(0); + + setup_managers(state); + + return (0); +} + +static int +_teardown(void **state) { + isc__nm_force_tid(-1); + + teardown_managers(state); + + return (0); +} + +/* test that ns_listenlist_default() works */ +ISC_RUN_TEST_IMPL(ns_listenlist_default) { + isc_result_t result; + in_port_t port = 5300 + isc_random8(); + ns_listenlist_t *list = NULL; + ns_listenelt_t *elt; + int count; + + UNUSED(state); + + result = ns_listenlist_default(mctx, port, false, AF_INET, &list); + assert_int_equal(result, ISC_R_SUCCESS); + assert_non_null(list); + + assert_false(ISC_LIST_EMPTY(list->elts)); + + count = 0; + elt = ISC_LIST_HEAD(list->elts); + while (elt != NULL) { + ns_listenelt_t *next = ISC_LIST_NEXT(elt, link); + dns_acl_t *acl = NULL; + + dns_acl_attach(elt->acl, &acl); + ISC_LIST_UNLINK(list->elts, elt, link); + ns_listenelt_destroy(elt); + elt = next; + + assert_true(dns_acl_isnone(acl)); + dns_acl_detach(&acl); + count++; + } + + assert_true(ISC_LIST_EMPTY(list->elts)); + assert_int_equal(count, 1); + + ns_listenlist_detach(&list); + + result = ns_listenlist_default(mctx, port, true, AF_INET, &list); + assert_int_equal(result, ISC_R_SUCCESS); + + assert_false(ISC_LIST_EMPTY(list->elts)); + + /* This time just use ns_listenlist_detach() to destroy elements */ + count = 0; + elt = ISC_LIST_HEAD(list->elts); + while (elt != NULL) { + ns_listenelt_t *next = ISC_LIST_NEXT(elt, link); + assert_true(dns_acl_isany(elt->acl)); + elt = next; + count++; + } + + assert_int_equal(count, 1); + + ns_listenlist_detach(&list); +} + +ISC_TEST_LIST_START + +ISC_TEST_ENTRY_CUSTOM(ns_listenlist_default, _setup, _teardown) + +ISC_TEST_LIST_END + +ISC_TEST_MAIN |