tuklib_cpucores.c revision 292588
1/////////////////////////////////////////////////////////////////////////////// 2// 3/// \file tuklib_cpucores.c 4/// \brief Get the number of CPU cores online 5// 6// Author: Lasse Collin 7// 8// This file has been put into the public domain. 9// You can do whatever you want with this file. 10// 11/////////////////////////////////////////////////////////////////////////////// 12 13#include "tuklib_cpucores.h" 14 15#if defined(_WIN32) || defined(__CYGWIN__) 16# ifndef _WIN32_WINNT 17# define _WIN32_WINNT 0x0500 18# endif 19# include <windows.h> 20 21// FreeBSD 22#elif defined(TUKLIB_CPUCORES_CPUSET) 23# include <sys/param.h> 24# include <sys/cpuset.h> 25 26#elif defined(TUKLIB_CPUCORES_SYSCTL) 27# ifdef HAVE_SYS_PARAM_H 28# include <sys/param.h> 29# endif 30# include <sys/sysctl.h> 31 32#elif defined(TUKLIB_CPUCORES_SYSCONF) 33# include <unistd.h> 34 35// HP-UX 36#elif defined(TUKLIB_CPUCORES_PSTAT_GETDYNAMIC) 37# include <sys/param.h> 38# include <sys/pstat.h> 39#endif 40 41 42extern uint32_t 43tuklib_cpucores(void) 44{ 45 uint32_t ret = 0; 46 47#if defined(_WIN32) || defined(__CYGWIN__) 48 SYSTEM_INFO sysinfo; 49 GetSystemInfo(&sysinfo); 50 ret = sysinfo.dwNumberOfProcessors; 51 52#elif defined(TUKLIB_CPUCORES_CPUSET) 53 cpuset_t set; 54 if (cpuset_getaffinity(CPU_LEVEL_WHICH, CPU_WHICH_PID, -1, 55 sizeof(set), &set) == 0) { 56# ifdef CPU_COUNT 57 ret = CPU_COUNT(&set); 58# else 59 for (unsigned i = 0; i < CPU_SETSIZE; ++i) 60 if (CPU_ISSET(i, &set)) 61 ++ret; 62# endif 63 } 64 65#elif defined(TUKLIB_CPUCORES_SYSCTL) 66 int name[2] = { CTL_HW, HW_NCPU }; 67 int cpus; 68 size_t cpus_size = sizeof(cpus); 69 if (sysctl(name, 2, &cpus, &cpus_size, NULL, 0) != -1 70 && cpus_size == sizeof(cpus) && cpus > 0) 71 ret = cpus; 72 73#elif defined(TUKLIB_CPUCORES_SYSCONF) 74# ifdef _SC_NPROCESSORS_ONLN 75 // Most systems 76 const long cpus = sysconf(_SC_NPROCESSORS_ONLN); 77# else 78 // IRIX 79 const long cpus = sysconf(_SC_NPROC_ONLN); 80# endif 81 if (cpus > 0) 82 ret = cpus; 83 84#elif defined(TUKLIB_CPUCORES_PSTAT_GETDYNAMIC) 85 struct pst_dynamic pst; 86 if (pstat_getdynamic(&pst, sizeof(pst), 1, 0) != -1) 87 ret = pst.psd_proc_cnt; 88#endif 89 90 return ret; 91} 92