]>
Commit | Line | Data |
---|---|---|
0f23ee0c KZ |
1 | /* |
2 | * This file may be redistributed under the terms of the | |
3 | * GNU Lesser General Public License. | |
4 | */ | |
efcb71f8 KZ |
5 | #ifndef UTIL_LINUX_CPUSET_H |
6 | #define UTIL_LINUX_CPUSET_H | |
7 | ||
ff5a6d20 | 8 | #include <sched.h> |
efcb71f8 | 9 | |
ee32c514 KZ |
10 | /* |
11 | * Fallback for old or obscure libcs without dynamically allocated cpusets | |
12 | * | |
13 | * The following macros are based on code from glibc. | |
14 | * | |
15 | * The GNU C Library is free software; you can redistribute it and/or | |
16 | * modify it under the terms of the GNU Lesser General Public | |
17 | * License as published by the Free Software Foundation; either | |
18 | * version 2.1 of the License, or (at your option) any later version. | |
19 | */ | |
20 | #if !HAVE_DECL_CPU_ALLOC | |
21 | ||
22 | # define CPU_ZERO_S(setsize, cpusetp) \ | |
23 | do { \ | |
24 | size_t __i; \ | |
25 | size_t __imax = (setsize) / sizeof (__cpu_mask); \ | |
26 | __cpu_mask *__bits = (cpusetp)->__bits; \ | |
27 | for (__i = 0; __i < __imax; ++__i) \ | |
28 | __bits[__i] = 0; \ | |
29 | } while (0) | |
30 | ||
31 | # define CPU_SET_S(cpu, setsize, cpusetp) \ | |
32 | ({ size_t __cpu = (cpu); \ | |
33 | __cpu < 8 * (setsize) \ | |
34 | ? (((__cpu_mask *) ((cpusetp)->__bits))[__CPUELT (__cpu)] \ | |
35 | |= __CPUMASK (__cpu)) \ | |
36 | : 0; }) | |
37 | ||
38 | # define CPU_ISSET_S(cpu, setsize, cpusetp) \ | |
39 | ({ size_t __cpu = (cpu); \ | |
40 | __cpu < 8 * (setsize) \ | |
41 | ? ((((__cpu_mask *) ((cpusetp)->__bits))[__CPUELT (__cpu)] \ | |
42 | & __CPUMASK (__cpu))) != 0 \ | |
43 | : 0; }) | |
44 | ||
e67690f3 KZ |
45 | # define CPU_EQUAL_S(setsize, cpusetp1, cpusetp2) \ |
46 | ({ __cpu_mask *__arr1 = (cpusetp1)->__bits; \ | |
47 | __cpu_mask *__arr2 = (cpusetp2)->__bits; \ | |
48 | size_t __imax = (setsize) / sizeof (__cpu_mask); \ | |
49 | size_t __i; \ | |
50 | for (__i = 0; __i < __imax; ++__i) \ | |
7d2a8dd0 | 51 | if (__arr1[__i] != __arr2[__i]) \ |
e67690f3 KZ |
52 | break; \ |
53 | __i == __imax; }) | |
54 | ||
ee32c514 KZ |
55 | extern int __cpuset_count_s(size_t setsize, const cpu_set_t *set); |
56 | # define CPU_COUNT_S(setsize, cpusetp) __cpuset_count_s(setsize, cpusetp) | |
57 | ||
58 | # define CPU_ALLOC_SIZE(count) \ | |
59 | ((((count) + __NCPUBITS - 1) / __NCPUBITS) * sizeof (__cpu_mask)) | |
60 | # define CPU_ALLOC(count) (malloc(CPU_ALLOC_SIZE(count))) | |
61 | # define CPU_FREE(cpuset) (free(cpuset)) | |
62 | ||
63 | #endif /* !HAVE_DECL_CPU_ALLOC */ | |
64 | ||
efcb71f8 | 65 | |
ff5a6d20 | 66 | #define cpuset_nbits(setsize) (8 * (setsize)) |
efcb71f8 | 67 | |
e9d659ea | 68 | /* |
9e930041 | 69 | * The @idx parameter returns an index of the first mask from @ary array where |
e9d659ea KZ |
70 | * the @cpu is set. |
71 | * | |
72 | * Returns: 0 if found, otherwise 1. | |
73 | */ | |
74 | static inline int cpuset_ary_isset(size_t cpu, cpu_set_t **ary, size_t nmemb, | |
75 | size_t setsize, size_t *idx) | |
76 | { | |
77 | size_t i; | |
78 | ||
79 | for (i = 0; i < nmemb; i++) { | |
80 | if (CPU_ISSET_S(cpu, setsize, ary[i])) { | |
81 | *idx = i; | |
82 | return 0; | |
83 | } | |
84 | } | |
85 | return 1; | |
86 | } | |
87 | ||
bae91ecf KZ |
88 | extern int get_max_number_of_cpus(void); |
89 | ||
ff5a6d20 KZ |
90 | extern cpu_set_t *cpuset_alloc(int ncpus, size_t *setsize, size_t *nbits); |
91 | extern void cpuset_free(cpu_set_t *set); | |
efcb71f8 | 92 | |
ff5a6d20 | 93 | extern char *cpulist_create(char *str, size_t len, cpu_set_t *set, size_t setsize); |
b16f615a | 94 | extern int cpulist_parse(const char *str, cpu_set_t *set, size_t setsize, int fail); |
ff5a6d20 KZ |
95 | |
96 | extern char *cpumask_create(char *str, size_t len, cpu_set_t *set, size_t setsize); | |
97 | extern int cpumask_parse(const char *str, cpu_set_t *set, size_t setsize); | |
efcb71f8 KZ |
98 | |
99 | #endif /* UTIL_LINUX_CPUSET_H */ |