| 1 | /* Definition of the cpu_set_t structure used by the POSIX 1003.1b-1993 |
| 2 | scheduling interface. |
| 3 | Copyright (C) 1996-2022 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'. */ |
| 32 | typedef __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. */ |
| 39 | typedef 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 | |
| 117 | extern int __sched_cpucount (size_t __setsize, const cpu_set_t *__setp) |
| 118 | __THROW; |
| 119 | extern cpu_set_t *__sched_cpualloc (size_t __count) __THROW __wur; |
| 120 | extern void __sched_cpufree (cpu_set_t *__set) __THROW; |
| 121 | |
| 122 | __END_DECLS |
| 123 | |
| 124 | #endif /* bits/cpu-set.h */ |
| 125 | |