mirror of
https://git.tukaani.org/xz.git
synced 2024-04-04 12:36:23 +02:00
53 lines
1.2 KiB
C
53 lines
1.2 KiB
C
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
/// \file cpucores.h
|
||
|
|
/// \brief Get the number of online CPU cores
|
||
|
|
//
|
||
|
|
// This code has been put into the public domain.
|
||
|
|
//
|
||
|
|
// This library is distributed in the hope that it will be useful,
|
||
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
||
|
|
//
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
#ifndef CPUCORES_H
|
||
|
|
#define CPUCORES_H
|
||
|
|
|
||
|
|
#if defined(HAVE_NCPU_SYSCONF)
|
||
|
|
# include <unistd.h>
|
||
|
|
|
||
|
|
#elif defined(HAVE_NCPU_SYSCTL)
|
||
|
|
# ifdef HAVE_SYS_PARAM_H
|
||
|
|
# include <sys/param.h>
|
||
|
|
# endif
|
||
|
|
# ifdef HAVE_SYS_SYSCTL_H
|
||
|
|
# include <sys/sysctl.h>
|
||
|
|
# endif
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
static inline uint32_t
|
||
|
|
cpucores(void)
|
||
|
|
{
|
||
|
|
uint32_t ret = 0;
|
||
|
|
|
||
|
|
#if defined(HAVE_CPUCORES_SYSCONF)
|
||
|
|
const long cpus = sysconf(_SC_NPROCESSORS_ONLN);
|
||
|
|
if (cpus > 0)
|
||
|
|
ret = (uint32_t)(cpus);
|
||
|
|
|
||
|
|
#elif defined(HAVE_CPUCORES_SYSCTL)
|
||
|
|
int name[2] = { CTL_HW, HW_NCPU };
|
||
|
|
int cpus;
|
||
|
|
size_t cpus_size = sizeof(cpus);
|
||
|
|
if (!sysctl(name, &cpus, &cpus_size, NULL, NULL)
|
||
|
|
&& cpus_size == sizeof(cpus) && cpus > 0)
|
||
|
|
ret = (uint32_t)(cpus);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|