1 /* SPDX-License-Identifier: GPL-2.0 */
2 #include <stdbool.h>
3 #include <stdlib.h>
4
5 #ifndef PAGE_SIZE
6 #define PAGE_SIZE 4096
7 #endif
8
9 #define MB(x) (x << 20)
10
11 #define USEC_PER_SEC 1000000L
12 #define NSEC_PER_SEC 1000000000L
13
14 #define TEST_UID 65534 /* usually nobody, any !root is fine */
15
16 #define CG_THREADS_FILE (!cg_test_v1_named ? "cgroup.threads" : "tasks")
17 #define CG_NAMED_NAME "selftest"
18 #define CG_PATH_FORMAT (!cg_test_v1_named ? "0::%s" : (":name=" CG_NAMED_NAME ":%s"))
19
20 /*
21 * Checks if two given values differ by less than err% of their sum.
22 */
values_close(long a,long b,int err)23 static inline int values_close(long a, long b, int err)
24 {
25 return labs(a - b) <= (a + b) / 100 * err;
26 }
27
28 extern ssize_t read_text(const char *path, char *buf, size_t max_len);
29 extern ssize_t write_text(const char *path, char *buf, ssize_t len);
30
31 extern int cg_find_controller_root(char *root, size_t len, const char *controller);
32 extern int cg_find_unified_root(char *root, size_t len, bool *nsdelegate);
33 extern char *cg_name(const char *root, const char *name);
34 extern char *cg_name_indexed(const char *root, const char *name, int index);
35 extern char *cg_control(const char *cgroup, const char *control);
36 extern int cg_create(const char *cgroup);
37 extern int cg_destroy(const char *cgroup);
38 extern int cg_read(const char *cgroup, const char *control,
39 char *buf, size_t len);
40 extern int cg_read_strcmp(const char *cgroup, const char *control,
41 const char *expected);
42 extern int cg_read_strstr(const char *cgroup, const char *control,
43 const char *needle);
44 extern long cg_read_long(const char *cgroup, const char *control);
45 extern long cg_read_long_fd(int fd);
46 long cg_read_key_long(const char *cgroup, const char *control, const char *key);
47 extern long cg_read_lc(const char *cgroup, const char *control);
48 extern int cg_write(const char *cgroup, const char *control, char *buf);
49 extern int cg_open(const char *cgroup, const char *control, int flags);
50 int cg_write_numeric(const char *cgroup, const char *control, long value);
51 extern int cg_run(const char *cgroup,
52 int (*fn)(const char *cgroup, void *arg),
53 void *arg);
54 extern int cg_enter(const char *cgroup, int pid);
55 extern int cg_enter_current(const char *cgroup);
56 extern int cg_enter_current_thread(const char *cgroup);
57 extern int cg_run_nowait(const char *cgroup,
58 int (*fn)(const char *cgroup, void *arg),
59 void *arg);
60 extern int cg_wait_for_proc_count(const char *cgroup, int count);
61 extern int cg_killall(const char *cgroup);
62 int proc_mount_contains(const char *option);
63 extern ssize_t proc_read_text(int pid, bool thread, const char *item, char *buf, size_t size);
64 extern int proc_read_strstr(int pid, bool thread, const char *item, const char *needle);
65 extern pid_t clone_into_cgroup(int cgroup_fd);
66 extern int clone_reap(pid_t pid, int options);
67 extern int clone_into_cgroup_run_wait(const char *cgroup);
68 extern int dirfd_open_opath(const char *dir);
69 extern int cg_prepare_for_wait(const char *cgroup);
70 extern int memcg_prepare_for_wait(const char *cgroup);
71 extern int cg_wait_for(int fd);
72 extern bool cg_test_v1_named;
73