2
0
mirror of https://gitlab.isc.org/isc-projects/bind9 synced 2025-08-22 01:59:26 +00:00
Ondřej Surý f7e5c1db38
Change the 'isc_g_mctx' to be always available
This required couple of internal changes to the isc_mem_debugging.

The isc_mem_debugging is now internal to isc_mem unit and there are
three new functions:

1. isc_mem_setdebugging() can change the debugging setting for an
   individual memory context.  This is need for the memory contexts used
   for OpenSSL, libxml and libuv accounting as recording and tracing
   memory is broken there.

2. isc_mem_debugon() / isc_mem_debugoff() can be used to change default
   memory debugging flags as well as debugging flags for isc_g_mctx.

Additionally, the memory debugging is inconsistent across the code-base.
For now, we are keeping the existing flags, but three new environment
variables have been added 'ISC_MEM_DEBUGRECORD', 'ISC_MEM_DEBUGTRACE'
and 'ISC_MEM_DEBUGUSAGE' to set the global debugging flags at any
program using the memory contexts.
2025-08-04 11:29:50 +02:00

129 lines
2.2 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>
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_debugon(ISC_MEM_DEBUGRECORD);
return 0;
}
int
teardown_mctx(void **state ISC_ATTR_UNUSED) {
return 0;
}
int
setup_loopmgr(void **state ISC_ATTR_UNUSED) {
REQUIRE(isc_g_mctx != NULL);
setup_workers(state);
isc_loopmgr_create(isc_g_mctx, workers);
return 0;
}
int
teardown_loopmgr(void **state ISC_ATTR_UNUSED) {
isc_loopmgr_destroy();
return 0;
}
int
setup_netmgr(void **state ISC_ATTR_UNUSED) {
adjustnofile();
isc_netmgr_create(isc_g_mctx);
return 0;
}
int
teardown_netmgr(void **state ISC_ATTR_UNUSED) {
isc_netmgr_destroy();
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;
}