projects/linux·blob master · refs

102 lines · 3.9 KiB

1/* SPDX-License-Identifier: GPL-2.0 */
2#include <stdbool.h>
3#include <stdlib.h>
4
5#ifndef BUF_SIZE
6#define BUF_SIZE 4096
7#endif
8
9#define MB(x) (x << 20)
10#define GB(x) ((unsigned long long)(x) << 30)
11
12#define NSEC_PER_USEC 1000L
13#define USEC_PER_SEC 1000000L
14#define NSEC_PER_SEC 1000000000L
15
16#define TEST_UID 65534 /* usually nobody, any !root is fine */
17
18#define CG_THREADS_FILE (!cg_test_v1_named ? "cgroup.threads" : "tasks")
19#define CG_NAMED_NAME "selftest"
20#define CG_PATH_FORMAT (!cg_test_v1_named ? "0::%s" : (":name=" CG_NAMED_NAME ":%s"))
21
22#define DEFAULT_WAIT_INTERVAL_US (100 * 1000) /* 100 ms */
23
24/*
25 * Checks if two given values differ by less than err% of their sum.
26 */
27static inline int values_close(long a, long b, int err)
28{
29 return labs(a - b) <= (a + b) / 100 * err;
30}
31
32/*
33 * Checks if two given values differ by less than err% of their sum and assert
34 * with detailed debug info if not.
35 */
36static inline int values_close_report(long a, long b, int err)
37{
38 long diff = labs(a - b);
39 long limit = (a + b) / 100 * err;
40 double actual_err = (a + b) ? (100.0 * diff / (a + b)) : 0.0;
41 int close = diff <= limit;
42
43 if (!close)
44 fprintf(stderr,
45 "[FAIL] actual=%ld expected=%ld | diff=%ld | limit=%ld | "
46 "tolerance=%d%% | actual_error=%.2f%%\n",
47 a, b, diff, limit, err, actual_err);
48
49 return close;
50}
51
52extern ssize_t read_text(const char *path, char *buf, size_t max_len);
53extern ssize_t write_text(const char *path, char *buf, ssize_t len);
54
55extern int cg_find_controller_root(char *root, size_t len, const char *controller);
56extern int cg_find_unified_root(char *root, size_t len, bool *nsdelegate);
57extern char *cg_name(const char *root, const char *name);
58extern char *cg_name_indexed(const char *root, const char *name, int index);
59extern char *cg_control(const char *cgroup, const char *control);
60extern int cg_create(const char *cgroup);
61extern int cg_destroy(const char *cgroup);
62extern int cg_read(const char *cgroup, const char *control,
63 char *buf, size_t len);
64extern int cg_read_strcmp(const char *cgroup, const char *control,
65 const char *expected);
66extern int cg_read_strcmp_wait(const char *cgroup, const char *control,
67 const char *expected);
68extern int cg_read_strstr(const char *cgroup, const char *control,
69 const char *needle);
70extern long cg_read_long(const char *cgroup, const char *control);
71extern long cg_read_long_fd(int fd);
72long cg_read_key_long(const char *cgroup, const char *control, const char *key);
73long cg_read_key_long_poll(const char *cgroup, const char *control,
74 const char *key, long expected, int retries,
75 useconds_t wait_interval_us);
76extern long cg_read_lc(const char *cgroup, const char *control);
77extern int cg_write(const char *cgroup, const char *control, char *buf);
78extern int cg_open(const char *cgroup, const char *control, int flags);
79int cg_write_numeric(const char *cgroup, const char *control, long value);
80extern int cg_run(const char *cgroup,
81 int (*fn)(const char *cgroup, void *arg),
82 void *arg);
83extern int cg_enter(const char *cgroup, int pid);
84extern int cg_enter_current(const char *cgroup);
85extern int cg_enter_current_thread(const char *cgroup);
86extern int cg_run_nowait(const char *cgroup,
87 int (*fn)(const char *cgroup, void *arg),
88 void *arg);
89extern int cg_wait_for_proc_count(const char *cgroup, int count);
90extern int cg_killall(const char *cgroup);
91int proc_mount_contains(const char *option);
92int cgroup_feature(const char *feature);
93extern ssize_t proc_read_text(int pid, bool thread, const char *item, char *buf, size_t size);
94extern int proc_read_strstr(int pid, bool thread, const char *item, const char *needle);
95extern pid_t clone_into_cgroup(int cgroup_fd);
96extern int clone_reap(pid_t pid, int options);
97extern int clone_into_cgroup_run_wait(const char *cgroup);
98extern int dirfd_open_opath(const char *dir);
99extern int cg_prepare_for_wait(const char *cgroup);
100extern int memcg_prepare_for_wait(const char *cgroup);
101extern int cg_wait_for(int fd);
102extern bool cg_test_v1_named;