1 /*
2 * Copyright (c) 2015 Cyril Hrubis <chrubis@suse.cz>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms in version 2 of the GNU General Public License as
6 * published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
11 * the GNU General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public License
14 * along with this program; if not, write to the Free Software Foundation,
15 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16 */
17
18 /*
19 * This code was extracted form the sched_setparam testcases.
20 */
21
22 #include <unistd.h>
23
24 #ifdef BSD
25 # include <sys/types.h>
26 # include <sys/param.h>
27 # include <sys/sysctl.h>
28 #endif
29
30 #ifdef HPUX
31 # include <sys/param.h>
32 # include <sys/pstat.h>
33 #endif
34
get_ncpu(void)35 static int get_ncpu(void)
36 {
37 int ncpu = -1;
38
39 /* This syscall is not POSIX but it should work on many system */
40 #ifdef _SC_NPROCESSORS_ONLN
41 ncpu = sysconf(_SC_NPROCESSORS_ONLN);
42 #else
43 # ifdef BSD
44 int mib[2];
45 size_t len = sizeof(ncpu);
46 mib[0] = CTL_HW;
47 mib[1] = HW_NCPU;
48 sysctl(mib, 2, &ncpu, &len, NULL, 0);
49 # endif
50 # ifdef HPUX
51 struct pst_dynamic psd;
52 pstat_getdynamic(&psd, sizeof(psd), 1, 0);
53 ncpu = (int)psd.psd_proc_cnt;
54 # endif
55 #endif /* _SC_NPROCESSORS_ONLN */
56
57 return ncpu;
58 }
59