cpu-affinity-wrapper.c

   1 /* Copyright (c) 2008 Nikodemus Siivola <nikodemus@random-state.net>
   2  *
   3  * Permission is hereby granted, free of charge, to any person
   4  * obtaining a copy of this software and associated documentation
   5  * files (the "Software"), to deal in the Software without
   6  * restriction, including without limitation the rights to use,
   7  * copy, modify, merge, publish, distribute, sublicense, and/or esell
   8  * copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following
  10  * conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be
  13  * included in all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  16  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
  17  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  18  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
  19  * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  20  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  22  * OTHER DEALINGS IN THE SOFTWARE. */
  23
  24 #define _GNU_SOURCE
  25 #include <sched.h>
  26
  27 int cpu_setsize = CPU_SETSIZE;
  28 int cpu_mask_size = sizeof(cpu_set_t);
  29
  30 int get_cpu_affinity_mask(cpu_set_t *mask)
  31 {
  32     return sched_getaffinity(0, CPU_SETSIZE, mask);
  33 }
  34
  35 int set_cpu_affinity_mask(cpu_set_t *mask)
  36 {
  37     return sched_setaffinity(0, CPU_SETSIZE, mask);
  38 }
  39
  40 void zero_cpu_affinity_mask(cpu_set_t *mask)
  41 {
  42     CPU_ZERO(mask);
  43 }
  44
  45 void set_cpu_affinity(int cpu, cpu_set_t *mask)
  46 {
  47     CPU_SET(cpu, mask);
  48 }
  49
  50 int cpu_affinity_p(int cpu, cpu_set_t *mask)
  51 {
  52     return CPU_ISSET(cpu, mask);
  53 }
  54
  55 void clear_cpu_affinity(int cpu, cpu_set_t *mask)
  56 {
  57     CPU_CLR(cpu, mask);
  58 }
  59
  60
  61