1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
#ifndef __INCUS_SYSCALL_WRAPPER_H
#define __INCUS_SYSCALL_WRAPPER_H
#ifndef _GNU_SOURCE
#define _GNU_SOURCE 1
#endif
#include <asm/unistd.h>
#include <errno.h>
#include <linux/kcmp.h>
#include <sys/prctl.h>
#include <sys/syscall.h>
#include <sys/types.h>
#include <unistd.h>
#include "syscall_numbers.h"
#ifndef CLOSE_RANGE_UNSHARE
#define CLOSE_RANGE_UNSHARE (1U << 1)
#endif
#ifndef CLOSE_RANGE_CLOEXEC
#define CLOSE_RANGE_CLOEXEC (1U << 2)
#endif
static inline int incus_close_range(unsigned int fd, unsigned int max_fd, unsigned int flags)
{
return syscall(__NR_close_range, fd, max_fd, flags);
}
/* arg1 of prctl() */
#ifndef PR_SCHED_CORE
#define PR_SCHED_CORE 62
#endif
/* arg2 of prctl() */
#ifndef PR_SCHED_CORE_GET
#define PR_SCHED_CORE_GET 0
#endif
#ifndef PR_SCHED_CORE_CREATE
#define PR_SCHED_CORE_CREATE 1 /* create unique core_sched cookie */
#endif
#ifndef PR_SCHED_CORE_SHARE_TO
#define PR_SCHED_CORE_SHARE_TO 2 /* push core_sched cookie to pid */
#endif
#ifndef PR_SCHED_CORE_SHARE_FROM
#define PR_SCHED_CORE_SHARE_FROM 3 /* pull core_sched cookie to pid */
#endif
#ifndef PR_SCHED_CORE_MAX
#define PR_SCHED_CORE_MAX 4
#endif
/* arg3 of prctl() */
#ifndef PR_SCHED_CORE_SCOPE_THREAD
#define PR_SCHED_CORE_SCOPE_THREAD 0
#endif
#ifndef PR_SCHED_CORE_SCOPE_THREAD_GROUP
#define PR_SCHED_CORE_SCOPE_THREAD_GROUP 1
#endif
#ifndef PR_SCHED_CORE_SCOPE_PROCESS_GROUP
#define PR_SCHED_CORE_SCOPE_PROCESS_GROUP 2
#endif
#define INVALID_SCHED_CORE_COOKIE ((__u64)-1)
static inline bool core_scheduling_cookie_valid(__u64 cookie)
{
return (cookie > 0) && (cookie != INVALID_SCHED_CORE_COOKIE);
}
static inline __u64 core_scheduling_cookie_get(pid_t pid)
{
__u64 cookie;
int ret;
ret = prctl(PR_SCHED_CORE, PR_SCHED_CORE_GET, pid,
PR_SCHED_CORE_SCOPE_THREAD, (unsigned long)&cookie);
if (ret)
return INVALID_SCHED_CORE_COOKIE;
return cookie;
}
static inline int core_scheduling_cookie_create_threadgroup(pid_t pid)
{
int ret;
ret = prctl(PR_SCHED_CORE, PR_SCHED_CORE_CREATE, pid,
PR_SCHED_CORE_SCOPE_THREAD_GROUP, 0);
if (ret)
return -errno;
return 0;
}
static inline int core_scheduling_cookie_create_thread(pid_t pid)
{
int ret;
ret = prctl(PR_SCHED_CORE, PR_SCHED_CORE_CREATE, pid,
PR_SCHED_CORE_SCOPE_THREAD, 0);
if (ret)
return -errno;
return 0;
}
static inline int core_scheduling_cookie_share_with(pid_t pid)
{
return prctl(PR_SCHED_CORE, PR_SCHED_CORE_SHARE_FROM, pid,
PR_SCHED_CORE_SCOPE_THREAD, 0);
}
static inline int core_scheduling_cookie_share_to(pid_t pid)
{
return prctl(PR_SCHED_CORE, PR_SCHED_CORE_SHARE_TO, pid,
PR_SCHED_CORE_SCOPE_THREAD, 0);
}
static int kcmp(pid_t pid1, pid_t pid2, int type, unsigned long idx1,
unsigned long idx2)
{
return syscall(__NR_kcmp, pid1, pid2, type, idx1, idx2);
}
static inline bool filetable_shared(pid_t pid1, pid_t pid2)
{
return kcmp(pid1, pid2, KCMP_FILES, -EBADF, -EBADF) == 0;
}
#endif /* __INCUS_SYSCALL_WRAPPER_H */
|
