mirror of
https://github.com/isc-projects/bind9.git
synced 2026-05-28 04:34:54 -04:00
b8d00e2e18
All the applications built on top of the loop manager were required to create just a single instance of the loop manager. Refactor the loop manager to not expose this instance to the callers and keep the loop manager object internal to the isc_loop compilation unit. This significantly simplifies a number of data structures and calls to the isc_loop API.
134 lines
3 KiB
C
134 lines
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 <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "isc/attributes.h"
|
|
|
|
#define UNIT_TESTING
|
|
#include <cmocka.h>
|
|
|
|
#include <isc/atomic.h>
|
|
#include <isc/lib.h>
|
|
#include <isc/loop.h>
|
|
#include <isc/os.h>
|
|
#include <isc/result.h>
|
|
#include <isc/util.h>
|
|
|
|
#include "loop.c"
|
|
|
|
#include <tests/isc.h>
|
|
|
|
static atomic_uint scheduled = 0;
|
|
|
|
static void
|
|
count(void *arg ISC_ATTR_UNUSED) {
|
|
atomic_fetch_add(&scheduled, 1);
|
|
}
|
|
|
|
static void
|
|
shutdown_loopmgr(void *arg ISC_ATTR_UNUSED) {
|
|
while (atomic_load(&scheduled) != isc_loopmgr_nloops()) {
|
|
isc_thread_yield();
|
|
}
|
|
|
|
isc_loopmgr_shutdown();
|
|
}
|
|
|
|
ISC_RUN_TEST_IMPL(isc_loopmgr) {
|
|
atomic_store(&scheduled, 0);
|
|
|
|
isc_loopmgr_setup(count, NULL);
|
|
isc_loop_setup(isc_loop_main(), shutdown_loopmgr, NULL);
|
|
|
|
isc_loopmgr_run();
|
|
|
|
assert_int_equal(atomic_load(&scheduled), isc_loopmgr_nloops());
|
|
}
|
|
|
|
static void
|
|
runjob(void *arg ISC_ATTR_UNUSED) {
|
|
isc_async_current(count, NULL);
|
|
if (isc_tid() == 0) {
|
|
isc_async_current(shutdown_loopmgr, NULL);
|
|
}
|
|
}
|
|
|
|
ISC_RUN_TEST_IMPL(isc_loopmgr_runjob) {
|
|
atomic_store(&scheduled, 0);
|
|
|
|
isc_loopmgr_setup(runjob, NULL);
|
|
isc_loopmgr_run();
|
|
assert_int_equal(atomic_load(&scheduled), isc_loopmgr_nloops());
|
|
}
|
|
|
|
static void
|
|
pause_loopmgr(void *arg ISC_ATTR_UNUSED) {
|
|
isc_loopmgr_pause();
|
|
|
|
assert_true(isc_loopmgr_paused());
|
|
|
|
for (size_t i = 0; i < isc_loopmgr_nloops(); i++) {
|
|
isc_loop_t *loop = isc_loop_get(i);
|
|
|
|
assert_true(loop->paused);
|
|
}
|
|
|
|
atomic_init(&scheduled, isc_loopmgr_nloops());
|
|
|
|
isc_loopmgr_resume();
|
|
}
|
|
|
|
ISC_RUN_TEST_IMPL(isc_loopmgr_pause) {
|
|
isc_loop_setup(isc_loop_main(), pause_loopmgr, NULL);
|
|
isc_loop_setup(isc_loop_main(), shutdown_loopmgr, NULL);
|
|
isc_loopmgr_run();
|
|
}
|
|
|
|
static void
|
|
send_sigint(void *arg ISC_ATTR_UNUSED) {
|
|
kill(getpid(), SIGINT);
|
|
}
|
|
|
|
ISC_RUN_TEST_IMPL(isc_loopmgr_sigint) {
|
|
isc_loop_setup(isc_loop_main(), send_sigint, NULL);
|
|
isc_loopmgr_run();
|
|
}
|
|
|
|
static void
|
|
send_sigterm(void *arg ISC_ATTR_UNUSED) {
|
|
kill(getpid(), SIGINT);
|
|
}
|
|
|
|
ISC_RUN_TEST_IMPL(isc_loopmgr_sigterm) {
|
|
isc_loop_setup(isc_loop_main(), send_sigterm, NULL);
|
|
isc_loopmgr_run();
|
|
}
|
|
|
|
ISC_TEST_LIST_START
|
|
ISC_TEST_ENTRY_CUSTOM(isc_loopmgr, setup_loopmgr, teardown_loopmgr)
|
|
ISC_TEST_ENTRY_CUSTOM(isc_loopmgr_pause, setup_loopmgr, teardown_loopmgr)
|
|
ISC_TEST_ENTRY_CUSTOM(isc_loopmgr_runjob, setup_loopmgr, teardown_loopmgr)
|
|
ISC_TEST_ENTRY_CUSTOM(isc_loopmgr_sigint, setup_loopmgr, teardown_loopmgr)
|
|
ISC_TEST_ENTRY_CUSTOM(isc_loopmgr_sigterm, setup_loopmgr, teardown_loopmgr)
|
|
ISC_TEST_LIST_END
|
|
|
|
ISC_TEST_MAIN
|