aboutsummaryrefslogtreecommitdiff
path: root/src/common/tuklib_cpucores.c
diff options
context:
space:
mode:
authorXin LI <delphij@FreeBSD.org>2015-04-08 21:06:02 +0000
committerXin LI <delphij@FreeBSD.org>2015-04-08 21:06:02 +0000
commitf4bb66fbc5dabc60ac06846b1913e0fff1d7a463 (patch)
tree5b2ce8c3a15b3201ab328b39c67502a6f8e8bd49 /src/common/tuklib_cpucores.c
parentd6a9376028870754d7d3de3f626c1deb6df81092 (diff)
downloadsrc-f4bb66fbc5dabc60ac06846b1913e0fff1d7a463.tar.gz
src-f4bb66fbc5dabc60ac06846b1913e0fff1d7a463.zip
Vendor import of xz-5.2.1 (trimmed).vendor/xz/5.2.1
Notes
Notes: svn path=/vendor/xz/dist/; revision=281277 svn path=/vendor/xz/5.2.1/; revision=281278; tag=vendor/xz/5.2.1
Diffstat (limited to 'src/common/tuklib_cpucores.c')
-rw-r--r--src/common/tuklib_cpucores.c18
1 files changed, 18 insertions, 0 deletions
diff --git a/src/common/tuklib_cpucores.c b/src/common/tuklib_cpucores.c
index 7574bc9c0645..e235fd1cedc3 100644
--- a/src/common/tuklib_cpucores.c
+++ b/src/common/tuklib_cpucores.c
@@ -18,6 +18,11 @@
# endif
# include <windows.h>
+// FreeBSD
+#elif defined(TUKLIB_CPUCORES_CPUSET)
+# include <sys/param.h>
+# include <sys/cpuset.h>
+
#elif defined(TUKLIB_CPUCORES_SYSCTL)
# ifdef HAVE_SYS_PARAM_H
# include <sys/param.h>
@@ -44,6 +49,19 @@ tuklib_cpucores(void)
GetSystemInfo(&sysinfo);
ret = sysinfo.dwNumberOfProcessors;
+#elif defined(TUKLIB_CPUCORES_CPUSET)
+ cpuset_t set;
+ if (cpuset_getaffinity(CPU_LEVEL_WHICH, CPU_WHICH_PID, -1,
+ sizeof(set), &set) == 0) {
+# ifdef CPU_COUNT
+ ret = CPU_COUNT(&set);
+# else
+ for (unsigned i = 0; i < CPU_SETSIZE; ++i)
+ if (CPU_ISSET(i, &set))
+ ++ret;
+# endif
+ }
+
#elif defined(TUKLIB_CPUCORES_SYSCTL)
int name[2] = { CTL_HW, HW_NCPU };
int cpus;