1 /*
2  * Copyright (C) 2008 The Android Open Source Project
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *  * Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  *  * Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in
12  *    the documentation and/or other materials provided with the
13  *    distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  */
28 
29 #ifndef _SCHED_H_
30 #define _SCHED_H_
31 
32 #include <bits/timespec.h>
33 #include <linux/sched.h>
34 #include <sys/cdefs.h>
35 
36 __BEGIN_DECLS
37 
38 /* This name is used by glibc, but not by the kernel. */
39 #define SCHED_OTHER SCHED_NORMAL
40 
41 struct sched_param {
42   int sched_priority;
43 };
44 
45 int sched_setscheduler(pid_t __pid, int __policy, const struct sched_param* __param);
46 int sched_getscheduler(pid_t __pid);
47 int sched_yield(void);
48 int sched_get_priority_max(int __policy);
49 int sched_get_priority_min(int __policy);
50 int sched_setparam(pid_t __pid, const struct sched_param* __param);
51 int sched_getparam(pid_t __pid, struct sched_param* __param);
52 int sched_rr_get_interval(pid_t __pid, struct timespec* __quantum);
53 
54 #if defined(__USE_GNU)
55 
56 int clone(int (*__fn)(void*), void* __child_stack, int __flags, void* __arg, ...) __INTRODUCED_IN_ARM(9) __INTRODUCED_IN_X86(17);
57 int unshare(int __flags) __INTRODUCED_IN(17);
58 int sched_getcpu(void);
59 int setns(int __fd, int __ns_type) __INTRODUCED_IN(21);
60 
61 #ifdef __LP64__
62 #define CPU_SETSIZE 1024
63 #else
64 #define CPU_SETSIZE 32
65 #endif
66 
67 #define __CPU_BITTYPE  unsigned long int  /* mandated by the kernel  */
68 #define __CPU_BITS     (8 * sizeof(__CPU_BITTYPE))
69 #define __CPU_ELT(x)   ((x) / __CPU_BITS)
70 #define __CPU_MASK(x)  ((__CPU_BITTYPE)1 << ((x) & (__CPU_BITS - 1)))
71 
72 typedef struct {
73   __CPU_BITTYPE  __bits[ CPU_SETSIZE / __CPU_BITS ];
74 } cpu_set_t;
75 
76 int sched_setaffinity(pid_t __pid, size_t __set_size, const cpu_set_t* __set);
77 int sched_getaffinity(pid_t __pid, size_t __set_size, cpu_set_t* __set);
78 
79 #define CPU_ZERO(set)          CPU_ZERO_S(sizeof(cpu_set_t), set)
80 #define CPU_SET(cpu, set)      CPU_SET_S(cpu, sizeof(cpu_set_t), set)
81 #define CPU_CLR(cpu, set)      CPU_CLR_S(cpu, sizeof(cpu_set_t), set)
82 #define CPU_ISSET(cpu, set)    CPU_ISSET_S(cpu, sizeof(cpu_set_t), set)
83 #define CPU_COUNT(set)         CPU_COUNT_S(sizeof(cpu_set_t), set)
84 #define CPU_EQUAL(set1, set2)  CPU_EQUAL_S(sizeof(cpu_set_t), set1, set2)
85 
86 #define CPU_AND(dst, set1, set2)  __CPU_OP(dst, set1, set2, &)
87 #define CPU_OR(dst, set1, set2)   __CPU_OP(dst, set1, set2, |)
88 #define CPU_XOR(dst, set1, set2)  __CPU_OP(dst, set1, set2, ^)
89 
90 #define __CPU_OP(dst, set1, set2, op)  __CPU_OP_S(sizeof(cpu_set_t), dst, set1, set2, op)
91 
92 /* Support for dynamically-allocated cpu_set_t */
93 
94 #define CPU_ALLOC_SIZE(count) \
95   __CPU_ELT((count) + (__CPU_BITS - 1)) * sizeof(__CPU_BITTYPE)
96 
97 #define CPU_ALLOC(count)  __sched_cpualloc((count))
98 #define CPU_FREE(set)     __sched_cpufree((set))
99 
100 cpu_set_t* __sched_cpualloc(size_t __count);
101 void __sched_cpufree(cpu_set_t* __set);
102 
103 #define CPU_ZERO_S(setsize, set)  __builtin_memset(set, 0, setsize)
104 
105 #define CPU_SET_S(cpu, setsize, set) \
106   do { \
107     size_t __cpu = (cpu); \
108     if (__cpu < 8 * (setsize)) \
109       (set)->__bits[__CPU_ELT(__cpu)] |= __CPU_MASK(__cpu); \
110   } while (0)
111 
112 #define CPU_CLR_S(cpu, setsize, set) \
113   do { \
114     size_t __cpu = (cpu); \
115     if (__cpu < 8 * (setsize)) \
116       (set)->__bits[__CPU_ELT(__cpu)] &= ~__CPU_MASK(__cpu); \
117   } while (0)
118 
119 #define CPU_ISSET_S(cpu, setsize, set) \
120   (__extension__ ({ \
121     size_t __cpu = (cpu); \
122     (__cpu < 8 * (setsize)) \
123       ? ((set)->__bits[__CPU_ELT(__cpu)] & __CPU_MASK(__cpu)) != 0 \
124       : 0; \
125   }))
126 
127 #define CPU_EQUAL_S(setsize, set1, set2)  (__builtin_memcmp(set1, set2, setsize) == 0)
128 
129 #define CPU_AND_S(setsize, dst, set1, set2)  __CPU_OP_S(setsize, dst, set1, set2, &)
130 #define CPU_OR_S(setsize, dst, set1, set2)   __CPU_OP_S(setsize, dst, set1, set2, |)
131 #define CPU_XOR_S(setsize, dst, set1, set2)  __CPU_OP_S(setsize, dst, set1, set2, ^)
132 
133 #define __CPU_OP_S(setsize, dstset, srcset1, srcset2, op) \
134   do { \
135     cpu_set_t* __dst = (dstset); \
136     const __CPU_BITTYPE* __src1 = (srcset1)->__bits; \
137     const __CPU_BITTYPE* __src2 = (srcset2)->__bits; \
138     size_t __nn = 0, __nn_max = (setsize)/sizeof(__CPU_BITTYPE); \
139     for (; __nn < __nn_max; __nn++) \
140       (__dst)->__bits[__nn] = __src1[__nn] op __src2[__nn]; \
141   } while (0)
142 
143 #define CPU_COUNT_S(setsize, set)  __sched_cpucount((setsize), (set))
144 
145 int __sched_cpucount(size_t __set_size, const cpu_set_t* __set);
146 
147 #endif /* __USE_GNU */
148 
149 __END_DECLS
150 
151 #endif /* _SCHED_H_ */
152