include/cpuset.h

   1 /*
   2  * This file may be redistributed under the terms of the
   3  * GNU Lesser General Public License.
   4  */
   5 #ifndef UTIL_LINUX_CPUSET_H
   6 #define UTIL_LINUX_CPUSET_H
   7
   8 #include <sched.h>
   9
  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
  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)                                      \
  51         if (__arr1[__i] != __arr2[__i])                                       \
  52           break;                                                              \
  53       __i == __imax; })
  54
  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
  65
  66 #define cpuset_nbits(setsize)   (8 * (setsize))
  67
  68 /*
  69  * The @idx parameter returns an index of the first mask from @ary array where
  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
  88 extern int get_max_number_of_cpus(void);
  89
  90 extern cpu_set_t *cpuset_alloc(int ncpus, size_t *setsize, size_t *nbits);
  91 extern void cpuset_free(cpu_set_t *set);
  92
  93 extern char *cpulist_create(char *str, size_t len, cpu_set_t *set, size_t setsize);
  94 extern int cpulist_parse(const char *str, cpu_set_t *set, size_t setsize, int fail);
  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);
  98
  99 #endif /* UTIL_LINUX_CPUSET_H */