zig

fork of https://codeberg.org/ziglang/zig
Log | Files | Refs | README | LICENSE

cpu-set.h (4516B) - Raw


      1 /* Definition of the cpu_set_t structure used by the POSIX 1003.1b-1993
      2    scheduling interface.
      3    Copyright (C) 1996-2025 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 */