mirror of
https://github.com/isc-projects/bind9.git
synced 2026-03-24 19:35:00 -04:00
23751fe252
It was discovered that on some platforms (f.e. Alpine Linux with MUSL) the result of isc_os_ncpus() call differ when called before and after we drop privileges. This commit changes the isc_os_ncpus() call to cache the result from the first call and thus always return the same value during the runtime of the named. The first call to isc_os_ncpus() is made as soon as possible on the library initalization.
76 lines
1.8 KiB
C
76 lines
1.8 KiB
C
/*
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* 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 <isc/once.h>
|
|
#include <isc/os.h>
|
|
#include <isc/util.h>
|
|
|
|
static isc_once_t ncpus_once = ISC_ONCE_INIT;
|
|
static unsigned int ncpus = 0;
|
|
|
|
#ifdef HAVE_SYSCONF
|
|
|
|
#include <unistd.h>
|
|
|
|
static inline long
|
|
sysconf_ncpus(void) {
|
|
#if defined(_SC_NPROCESSORS_ONLN)
|
|
return (sysconf((_SC_NPROCESSORS_ONLN)));
|
|
#elif defined(_SC_NPROC_ONLN)
|
|
return (sysconf((_SC_NPROC_ONLN)));
|
|
#else /* if defined(_SC_NPROCESSORS_ONLN) */
|
|
return (0);
|
|
#endif /* if defined(_SC_NPROCESSORS_ONLN) */
|
|
}
|
|
#endif /* HAVE_SYSCONF */
|
|
|
|
#if defined(HAVE_SYS_SYSCTL_H) && defined(HAVE_SYSCTLBYNAME)
|
|
#include <sys/param.h> /* for NetBSD */
|
|
#include <sys/sysctl.h>
|
|
#include <sys/types.h> /* for FreeBSD */
|
|
|
|
static int
|
|
sysctl_ncpus(void) {
|
|
int ncpu, result;
|
|
size_t len;
|
|
|
|
len = sizeof(ncpu);
|
|
result = sysctlbyname("hw.ncpu", &ncpu, &len, 0, 0);
|
|
if (result != -1) {
|
|
return (ncpu);
|
|
}
|
|
return (0);
|
|
}
|
|
#endif /* if defined(HAVE_SYS_SYSCTL_H) && defined(HAVE_SYSCTLBYNAME) */
|
|
|
|
static void
|
|
ncpus_initialize(void) {
|
|
#if defined(HAVE_SYSCONF)
|
|
ncpus = sysconf_ncpus();
|
|
#endif /* if defined(HAVE_SYSCONF) */
|
|
#if defined(HAVE_SYS_SYSCTL_H) && defined(HAVE_SYSCTLBYNAME)
|
|
if (ncpus <= 0) {
|
|
ncpus = sysctl_ncpus();
|
|
}
|
|
#endif /* if defined(HAVE_SYS_SYSCTL_H) && defined(HAVE_SYSCTLBYNAME) */
|
|
if (ncpus <= 0) {
|
|
ncpus = 1;
|
|
}
|
|
}
|
|
|
|
unsigned int
|
|
isc_os_ncpus(void) {
|
|
isc_result_t result = isc_once_do(&ncpus_once, ncpus_initialize);
|
|
RUNTIME_CHECK(result == ISC_R_SUCCESS);
|
|
INSIST(ncpus > 0);
|
|
|
|
return (ncpus);
|
|
}
|