1/* Definition of the cpu_set_t structure used by the POSIX 1003.1b-1993
2 scheduling interface.
3 Copyright (C) 1996-2024 Free Software Foundation, Inc.
4 This file is part of the GNU C Library.
5
6 The GNU C Library is free software; you can redistribute it and/or
7 modify it under the terms of the GNU Lesser General Public
8 License as published by the Free Software Foundation; either
9 version 2.1 of the License, or (at your option) any later version.
10
11 The GNU C Library is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 Lesser General Public License for more details.
15
16 You should have received a copy of the GNU Lesser General Public
17 License along with the GNU C Library; if not, see
18 <https://www.gnu.org/licenses/>. */
19
20#ifndef _BITS_CPU_SET_H
21#define _BITS_CPU_SET_H 1
22
23#ifndef _SCHED_H
24# error "Never include <bits/cpu-set.h> directly; use <sched.h> instead."
25#endif
26
27/* Size definition for CPU sets. */
28#define __CPU_SETSIZE 1024
29#define __NCPUBITS (8 * sizeof (__cpu_mask))
30
31/* Type for array elements in 'cpu_set_t'. */
32typedef __CPU_MASK_TYPE __cpu_mask;
33
34/* Basic access functions. */
35#define __CPUELT(cpu) ((cpu) / __NCPUBITS)
36#define __CPUMASK(cpu) ((__cpu_mask) 1 << ((cpu) % __NCPUBITS))
37
38/* Data structure to describe CPU mask. */
39typedef struct
40{
41 __cpu_mask __bits[__CPU_SETSIZE / __NCPUBITS];
42} cpu_set_t;
43
44/* Access functions for CPU masks. */
45#if __GNUC_PREREQ (2, 91)
46# define __CPU_ZERO_S(setsize, cpusetp) \
47 do __builtin_memset (cpusetp, '\0', setsize); while (0)
48#else
49# define __CPU_ZERO_S(setsize, cpusetp) \
50 do { \
51 size_t __i; \
52 size_t __imax = (setsize) / sizeof (__cpu_mask); \
53 __cpu_mask *__bits = (cpusetp)->__bits; \
54 for (__i = 0; __i < __imax; ++__i) \
55 __bits[__i] = 0; \
56 } while (0)
57#endif
58#define __CPU_SET_S(cpu, setsize, cpusetp) \
59 (__extension__ \
60 ({ size_t __cpu = (cpu); \
61 __cpu / 8 < (setsize) \
62 ? (((__cpu_mask *) ((cpusetp)->__bits))[__CPUELT (__cpu)] \
63 |= __CPUMASK (__cpu)) \
64 : 0; }))
65#define __CPU_CLR_S(cpu, setsize, cpusetp) \
66 (__extension__ \
67 ({ size_t __cpu = (cpu); \
68 __cpu / 8 < (setsize) \
69 ? (((__cpu_mask *) ((cpusetp)->__bits))[__CPUELT (__cpu)] \
70 &= ~__CPUMASK (__cpu)) \
71 : 0; }))
72#define __CPU_ISSET_S(cpu, setsize, cpusetp) \
73 (__extension__ \
74 ({ size_t __cpu = (cpu); \
75 __cpu / 8 < (setsize) \
76 ? ((((const __cpu_mask *) ((cpusetp)->__bits))[__CPUELT (__cpu)] \
77 & __CPUMASK (__cpu))) != 0 \
78 : 0; }))
79
80#define __CPU_COUNT_S(setsize, cpusetp) \
81 __sched_cpucount (setsize, cpusetp)
82
83#if __GNUC_PREREQ (2, 91)
84# define __CPU_EQUAL_S(setsize, cpusetp1, cpusetp2) \
85 (__builtin_memcmp (cpusetp1, cpusetp2, setsize) == 0)
86#else
87# define __CPU_EQUAL_S(setsize, cpusetp1, cpusetp2) \
88 (__extension__ \
89 ({ const __cpu_mask *__arr1 = (cpusetp1)->__bits; \
90 const __cpu_mask *__arr2 = (cpusetp2)->__bits; \
91 size_t __imax = (setsize) / sizeof (__cpu_mask); \
92 size_t __i; \
93 for (__i = 0; __i < __imax; ++__i) \
94 if (__arr1[__i] != __arr2[__i]) \
95 break; \
96 __i == __imax; }))
97#endif
98
99#define __CPU_OP_S(setsize, destset, srcset1, srcset2, op) \
100 (__extension__ \
101 ({ cpu_set_t *__dest = (destset); \
102 const __cpu_mask *__arr1 = (srcset1)->__bits; \
103 const __cpu_mask *__arr2 = (srcset2)->__bits; \
104 size_t __imax = (setsize) / sizeof (__cpu_mask); \
105 size_t __i; \
106 for (__i = 0; __i < __imax; ++__i) \
107 ((__cpu_mask *) __dest->__bits)[__i] = __arr1[__i] op __arr2[__i]; \
108 __dest; }))
109
110#define __CPU_ALLOC_SIZE(count) \
111 ((((count) + __NCPUBITS - 1) / __NCPUBITS) * sizeof (__cpu_mask))
112#define __CPU_ALLOC(count) __sched_cpualloc (count)
113#define __CPU_FREE(cpuset) __sched_cpufree (cpuset)
114
115__BEGIN_DECLS
116
117extern int __sched_cpucount (size_t __setsize, const cpu_set_t *__setp)
118 __THROW;
119extern cpu_set_t *__sched_cpualloc (size_t __count) __THROW __wur;
120extern void __sched_cpufree (cpu_set_t *__set) __THROW;
121
122__END_DECLS
123
124#endif /* bits/cpu-set.h */
125