mirror of
https://gitlab.isc.org/isc-projects/bind9
synced 2025-08-29 05:28:00 +00: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.
136 lines
2.3 KiB
C
136 lines
2.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.
|
|
*/
|
|
|
|
/*! \file */
|
|
|
|
#include <inttypes.h>
|
|
#include <signal.h>
|
|
#include <stdbool.h>
|
|
#include <stdlib.h>
|
|
#include <sys/resource.h>
|
|
#include <time.h>
|
|
|
|
#include <isc/buffer.h>
|
|
#include <isc/hash.h>
|
|
#include <isc/loop.h>
|
|
#include <isc/managers.h>
|
|
#include <isc/mem.h>
|
|
#include <isc/os.h>
|
|
#include <isc/string.h>
|
|
#include <isc/timer.h>
|
|
#include <isc/util.h>
|
|
|
|
#include <tests/isc.h>
|
|
|
|
isc_mem_t *mctx = NULL;
|
|
isc_nm_t *netmgr = NULL;
|
|
unsigned int workers = 0;
|
|
bool debug = false;
|
|
|
|
static void
|
|
adjustnofile(void) {
|
|
struct rlimit rl;
|
|
|
|
if (getrlimit(RLIMIT_NOFILE, &rl) == 0) {
|
|
if (rl.rlim_cur != rl.rlim_max) {
|
|
rl.rlim_cur = rl.rlim_max;
|
|
setrlimit(RLIMIT_NOFILE, &rl);
|
|
}
|
|
}
|
|
}
|
|
|
|
int
|
|
setup_workers(void **state ISC_ATTR_UNUSED) {
|
|
char *env_workers = getenv("ISC_TASK_WORKERS");
|
|
if (env_workers != NULL) {
|
|
workers = atoi(env_workers);
|
|
} else {
|
|
workers = isc_os_ncpus();
|
|
|
|
/* We always need at least two loops for some of the tests */
|
|
if (workers < 2) {
|
|
workers = 2;
|
|
}
|
|
}
|
|
INSIST(workers != 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
setup_mctx(void **state ISC_ATTR_UNUSED) {
|
|
isc_mem_debugging |= ISC_MEM_DEBUGRECORD;
|
|
isc_mem_create("test", &mctx);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
teardown_mctx(void **state ISC_ATTR_UNUSED) {
|
|
isc_mem_detach(&mctx);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
setup_loopmgr(void **state ISC_ATTR_UNUSED) {
|
|
REQUIRE(mctx != NULL);
|
|
|
|
setup_workers(state);
|
|
|
|
isc_loopmgr_create(mctx, workers);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
teardown_loopmgr(void **state ISC_ATTR_UNUSED) {
|
|
REQUIRE(netmgr == NULL);
|
|
|
|
isc_loopmgr_destroy();
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
setup_netmgr(void **state ISC_ATTR_UNUSED) {
|
|
adjustnofile();
|
|
|
|
isc_netmgr_create(mctx, &netmgr);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
teardown_netmgr(void **state ISC_ATTR_UNUSED) {
|
|
isc_nm_detach(&netmgr);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
setup_managers(void **state) {
|
|
setup_loopmgr(state);
|
|
setup_netmgr(state);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
teardown_managers(void **state) {
|
|
teardown_netmgr(state);
|
|
teardown_loopmgr(state);
|
|
|
|
return 0;
|
|
}
|