|
| 1 | +/* |
| 2 | + * Copyright (c) mlibc & plct lab |
| 3 | + * |
| 4 | + * SPDX-License-Identifier: MIT |
| 5 | + * |
| 6 | + * Change Logs: |
| 7 | + * Date Author Notes |
| 8 | + * 2024/8/8 0Bitbiscuits the first version |
| 9 | + */ |
| 10 | +#ifndef MLIBC_SCHED_H__ |
| 11 | +#define MLIBC_SCHED_H__ |
| 12 | + |
| 13 | +#ifdef __cplusplus |
| 14 | +extern "C" { |
| 15 | +#endif |
| 16 | + |
| 17 | +#include <features.h> |
| 18 | +#include <alltypes.h> |
| 19 | + |
| 20 | +/* Defined a set of scheduling policies and a scheduling flag */ |
| 21 | +#define SCHED_OTHER 0 |
| 22 | +#define SCHED_FIFO 1 |
| 23 | +#define SCHED_RR 2 |
| 24 | +#define SCHED_BATCH 3 |
| 25 | +#define SCHED_IDLE 5 |
| 26 | +#define SCHED_DEADLINE 6 |
| 27 | +#define SCHED_RESET_ON_FORK 0x40000000 |
| 28 | + |
| 29 | +/* Defined a set of flags for the clone system call, |
| 30 | +** which control how the new process (or thread) |
| 31 | +** inherits and shares the parent process's resources. |
| 32 | +**/ |
| 33 | +#ifdef __GNUC__ |
| 34 | +#define CSIGNAL 0x000000ff |
| 35 | +#define CLONE_NEWTIME 0x00000080 |
| 36 | +#define CLONE_VM 0x00000100 |
| 37 | +#define CLONE_FS 0x00000200 |
| 38 | +#define CLONE_FILES 0x00000400 |
| 39 | +#define CLONE_SIGHAND 0x00000800 |
| 40 | +#define CLONE_PIDFD 0x00001000 |
| 41 | +#define CLONE_PTRACE 0x00002000 |
| 42 | +#define CLONE_VFORK 0x00004000 |
| 43 | +#define CLONE_PARENT 0x00008000 |
| 44 | +#define CLONE_THREAD 0x00010000 |
| 45 | +#define CLONE_NEWNS 0x00020000 |
| 46 | +#define CLONE_SYSVSEM 0x00040000 |
| 47 | +#define CLONE_SETTLS 0x00080000 |
| 48 | +#define CLONE_PARENT_SETTID 0x00100000 |
| 49 | +#define CLONE_CHILD_CLEARTID 0x00200000 |
| 50 | +#define CLONE_DETACHED 0x00400000 |
| 51 | +#define CLONE_UNTRACED 0x00800000 |
| 52 | +#define CLONE_CHILD_SETTID 0x01000000 |
| 53 | +#define CLONE_NEWCGROUP 0x02000000 |
| 54 | +#define CLONE_NEWUTS 0x04000000 |
| 55 | +#define CLONE_NEWIPC 0x08000000 |
| 56 | +#define CLONE_NEWUSER 0x10000000 |
| 57 | +#define CLONE_NEWPID 0x20000000 |
| 58 | +#define CLONE_NEWNET 0x40000000 |
| 59 | +#define CLONE_IO 0x80000000 |
| 60 | +#endif /* __GNUC__ */ |
| 61 | + |
| 62 | +struct sched_param { |
| 63 | + int sched_priority; |
| 64 | + int __reserved1; |
| 65 | + struct { |
| 66 | + time_t __reserved1; |
| 67 | + long __reserved2; |
| 68 | + } __reserved2[2]; |
| 69 | + int __reserved3; |
| 70 | +}; |
| 71 | + |
| 72 | +int sched_get_priority_max(int); |
| 73 | +int sched_get_priority_min(int); |
| 74 | +int sched_getparam(pid_t, struct sched_param *); |
| 75 | +int sched_getscheduler(pid_t); |
| 76 | +int sched_rr_get_interval(pid_t, struct timespec *); |
| 77 | +int sched_setparam(pid_t, const struct sched_param *); |
| 78 | +int sched_setscheduler(pid_t, int, const struct sched_param *); |
| 79 | +int sched_yield(void); |
| 80 | + |
| 81 | +int clone (int (*)(void *), void *, int, void *, ...); |
| 82 | +int unshare(int); |
| 83 | +int setns(int, int); |
| 84 | + |
| 85 | +typedef struct cpu_set_t { unsigned long __bits[128/sizeof(long)]; } cpu_set_t; |
| 86 | +int __sched_cpucount(size_t, const cpu_set_t *); |
| 87 | +int sched_getcpu(void); |
| 88 | +int sched_getaffinity(pid_t, size_t, cpu_set_t *); |
| 89 | +int sched_setaffinity(pid_t, size_t, const cpu_set_t *); |
| 90 | + |
| 91 | +#define __CPU_op_S(i, size, set, op) ( (i)/8U >= (size) ? 0 : \ |
| 92 | + (((unsigned long *)(set))[(i)/8/sizeof(long)] op (1UL<<((i)%(8*sizeof(long))))) ) |
| 93 | + |
| 94 | +#define CPU_SET_S(i, size, set) __CPU_op_S(i, size, set, |=) |
| 95 | +#define CPU_CLR_S(i, size, set) __CPU_op_S(i, size, set, &=~) |
| 96 | +#define CPU_ISSET_S(i, size, set) __CPU_op_S(i, size, set, &) |
| 97 | + |
| 98 | +#define __CPU_op_func_S(func, op) \ |
| 99 | +static __inline void __CPU_##func##_S(size_t __size, cpu_set_t *__dest, \ |
| 100 | + const cpu_set_t *__src1, const cpu_set_t *__src2) \ |
| 101 | +{ \ |
| 102 | + size_t __i; \ |
| 103 | + for (__i=0; __i<__size/sizeof(long); __i++) \ |
| 104 | + ((unsigned long *)__dest)[__i] = ((unsigned long *)__src1)[__i] \ |
| 105 | + op ((unsigned long *)__src2)[__i] ; \ |
| 106 | +} |
| 107 | + |
| 108 | +__CPU_op_func_S(AND, &) |
| 109 | +__CPU_op_func_S(OR, |) |
| 110 | +__CPU_op_func_S(XOR, ^) |
| 111 | + |
| 112 | +#define CPU_AND_S(a,b,c,d) __CPU_AND_S(a,b,c,d) |
| 113 | +#define CPU_OR_S(a,b,c,d) __CPU_OR_S(a,b,c,d) |
| 114 | +#define CPU_XOR_S(a,b,c,d) __CPU_XOR_S(a,b,c,d) |
| 115 | + |
| 116 | +#define CPU_COUNT_S(size,set) __sched_cpucount(size,set) |
| 117 | +#define CPU_ZERO_S(size,set) memset(set,0,size) |
| 118 | +#define CPU_EQUAL_S(size,set1,set2) (!memcmp(set1,set2,size)) |
| 119 | + |
| 120 | +#define CPU_ALLOC_SIZE(n) (sizeof(long) * ( (n)/(8*sizeof(long)) \ |
| 121 | + + ((n)%(8*sizeof(long)) + 8*sizeof(long)-1)/(8*sizeof(long)) ) ) |
| 122 | +#define CPU_ALLOC(n) ((cpu_set_t *)calloc(1,CPU_ALLOC_SIZE(n))) |
| 123 | +#define CPU_FREE(set) free(set) |
| 124 | + |
| 125 | +#define CPU_SETSIZE 1024 |
| 126 | + |
| 127 | +#define CPU_SET(i, set) CPU_SET_S(i,sizeof(cpu_set_t),set) |
| 128 | +#define CPU_CLR(i, set) CPU_CLR_S(i,sizeof(cpu_set_t),set) |
| 129 | +#define CPU_ISSET(i, set) CPU_ISSET_S(i,sizeof(cpu_set_t),set) |
| 130 | +#define CPU_AND(d,s1,s2) CPU_AND_S(sizeof(cpu_set_t),d,s1,s2) |
| 131 | +#define CPU_OR(d,s1,s2) CPU_OR_S(sizeof(cpu_set_t),d,s1,s2) |
| 132 | +#define CPU_XOR(d,s1,s2) CPU_XOR_S(sizeof(cpu_set_t),d,s1,s2) |
| 133 | +#define CPU_COUNT(set) CPU_COUNT_S(sizeof(cpu_set_t),set) |
| 134 | +#define CPU_ZERO(set) CPU_ZERO_S(sizeof(cpu_set_t),set) |
| 135 | +#define CPU_EQUAL(s1,s2) CPU_EQUAL_S(sizeof(cpu_set_t),s1,s2) |
| 136 | + |
| 137 | +#ifdef __cplusplus |
| 138 | +} |
| 139 | +#endif |
| 140 | + |
| 141 | +#endif |
0 commit comments