mirror of
https://github.com/isc-projects/bind9.git
synced 2026-02-18 01:58:49 -05:00
78afc0bf04
Development versions of cmocka require the intmax_t and uintmax_t types
to be defined by the time the test code includes the <cmocka.h> header.
These types are defined in the <stdint.h> header, which is included by
the <inttypes.h> header, which in turn is already explicitly included by
some of the programs in the tests/ directory. Ensure all programs in
that directory that include the <cmocka.h> header also include the
<inttypes.h> header to future-proof the code while keeping the change
set minimal and the resulting code consistent. Also prevent explicitly
including the <stdint.h> header in those programs as it is included by
the <inttypes.h> header.
(cherry picked from commit c2dcd055fe)
164 lines
4.3 KiB
C
164 lines
4.3 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 <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#define UNIT_TESTING
|
|
#include <cmocka.h>
|
|
|
|
#include <isc/task.h>
|
|
#include <isc/taskpool.h>
|
|
#include <isc/util.h>
|
|
|
|
#include <tests/isc.h>
|
|
|
|
#define TASK_MAGIC ISC_MAGIC('T', 'A', 'S', 'K')
|
|
#define VALID_TASK(t) ISC_MAGIC_VALID(t, TASK_MAGIC)
|
|
|
|
/* Create a taskpool */
|
|
ISC_RUN_TEST_IMPL(create_pool) {
|
|
isc_result_t result;
|
|
isc_taskpool_t *pool = NULL;
|
|
|
|
UNUSED(state);
|
|
|
|
result = isc_taskpool_create(taskmgr, mctx, 8, 2, false, &pool);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(isc_taskpool_size(pool), 8);
|
|
|
|
isc_taskpool_destroy(&pool);
|
|
assert_null(pool);
|
|
}
|
|
|
|
/* Resize a taskpool */
|
|
ISC_RUN_TEST_IMPL(expand_pool) {
|
|
isc_result_t result;
|
|
isc_taskpool_t *pool1 = NULL, *pool2 = NULL, *hold = NULL;
|
|
|
|
UNUSED(state);
|
|
|
|
result = isc_taskpool_create(taskmgr, mctx, 10, 2, false, &pool1);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(isc_taskpool_size(pool1), 10);
|
|
|
|
/* resizing to a smaller size should have no effect */
|
|
hold = pool1;
|
|
result = isc_taskpool_expand(&pool1, 5, false, &pool2);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(isc_taskpool_size(pool2), 10);
|
|
assert_ptr_equal(pool2, hold);
|
|
assert_null(pool1);
|
|
pool1 = pool2;
|
|
pool2 = NULL;
|
|
|
|
/* resizing to the same size should have no effect */
|
|
hold = pool1;
|
|
result = isc_taskpool_expand(&pool1, 10, false, &pool2);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(isc_taskpool_size(pool2), 10);
|
|
assert_ptr_equal(pool2, hold);
|
|
assert_null(pool1);
|
|
pool1 = pool2;
|
|
pool2 = NULL;
|
|
|
|
/* resizing to larger size should make a new pool */
|
|
hold = pool1;
|
|
result = isc_taskpool_expand(&pool1, 20, false, &pool2);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(isc_taskpool_size(pool2), 20);
|
|
assert_ptr_not_equal(pool2, hold);
|
|
assert_null(pool1);
|
|
|
|
isc_taskpool_destroy(&pool2);
|
|
assert_null(pool2);
|
|
}
|
|
|
|
/* Get tasks */
|
|
ISC_RUN_TEST_IMPL(get_tasks) {
|
|
isc_result_t result;
|
|
isc_taskpool_t *pool = NULL;
|
|
isc_task_t *task1 = NULL, *task2 = NULL, *task3 = NULL;
|
|
|
|
UNUSED(state);
|
|
|
|
result = isc_taskpool_create(taskmgr, mctx, 2, 2, false, &pool);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(isc_taskpool_size(pool), 2);
|
|
|
|
/* two tasks in pool; make sure we can access them more than twice */
|
|
isc_taskpool_gettask(pool, &task1);
|
|
assert_true(VALID_TASK(task1));
|
|
|
|
isc_taskpool_gettask(pool, &task2);
|
|
assert_true(VALID_TASK(task2));
|
|
|
|
isc_taskpool_gettask(pool, &task3);
|
|
assert_true(VALID_TASK(task3));
|
|
|
|
isc_task_destroy(&task1);
|
|
isc_task_destroy(&task2);
|
|
isc_task_destroy(&task3);
|
|
|
|
isc_taskpool_destroy(&pool);
|
|
assert_null(pool);
|
|
}
|
|
|
|
/* Set privileges */
|
|
ISC_RUN_TEST_IMPL(set_privilege) {
|
|
isc_result_t result;
|
|
isc_taskpool_t *pool = NULL;
|
|
isc_task_t *task1 = NULL, *task2 = NULL, *task3 = NULL;
|
|
|
|
UNUSED(state);
|
|
|
|
result = isc_taskpool_create(taskmgr, mctx, 2, 2, true, &pool);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(isc_taskpool_size(pool), 2);
|
|
|
|
isc_taskpool_gettask(pool, &task1);
|
|
isc_taskpool_gettask(pool, &task2);
|
|
isc_taskpool_gettask(pool, &task3);
|
|
|
|
assert_true(VALID_TASK(task1));
|
|
assert_true(VALID_TASK(task2));
|
|
assert_true(VALID_TASK(task3));
|
|
|
|
assert_true(isc_task_getprivilege(task1));
|
|
assert_true(isc_task_getprivilege(task2));
|
|
assert_true(isc_task_getprivilege(task3));
|
|
|
|
isc_task_destroy(&task1);
|
|
isc_task_destroy(&task2);
|
|
isc_task_destroy(&task3);
|
|
|
|
isc_taskpool_destroy(&pool);
|
|
assert_null(pool);
|
|
}
|
|
|
|
ISC_TEST_LIST_START
|
|
|
|
ISC_TEST_ENTRY_CUSTOM(create_pool, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(expand_pool, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(get_tasks, setup_managers, teardown_managers)
|
|
ISC_TEST_ENTRY_CUSTOM(set_privilege, setup_managers, teardown_managers)
|
|
|
|
ISC_TEST_LIST_END
|
|
|
|
ISC_TEST_MAIN
|