cgroup_util.h

 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 */
23static inline int values_close(long a, long b, int err)
24{
25	return labs(a - b) <= (a + b) / 100 * err;
26}
27
28/*
29 * Checks if two given values differ by less than err% of their sum and assert
30 * with detailed debug info if not.
31 */
32static inline int values_close_report(long a, long b, int err)
33{
34	long diff  = labs(a - b);
35	long limit = (a + b) / 100 * err;
36	double actual_err = (a + b) ? (100.0 * diff / (a + b)) : 0.0;
37	int close = diff <= limit;
38
39	if (!close)
40		fprintf(stderr,
41			"[FAIL] actual=%ld expected=%ld | diff=%ld | limit=%ld | "
42			"tolerance=%d%% | actual_error=%.2f%%\n",
43			a, b, diff, limit, err, actual_err);
44
45	return close;
46}
47
48extern ssize_t read_text(const char *path, char *buf, size_t max_len);
49extern ssize_t write_text(const char *path, char *buf, ssize_t len);
50
51extern int cg_find_controller_root(char *root, size_t len, const char *controller);
52extern int cg_find_unified_root(char *root, size_t len, bool *nsdelegate);
53extern char *cg_name(const char *root, const char *name);
54extern char *cg_name_indexed(const char *root, const char *name, int index);
55extern char *cg_control(const char *cgroup, const char *control);
56extern int cg_create(const char *cgroup);
57extern int cg_destroy(const char *cgroup);
58extern int cg_read(const char *cgroup, const char *control,
59		   char *buf, size_t len);
60extern int cg_read_strcmp(const char *cgroup, const char *control,
61			  const char *expected);
62extern int cg_read_strcmp_wait(const char *cgroup, const char *control,
63				   const char *expected);
64extern int cg_read_strstr(const char *cgroup, const char *control,
65			  const char *needle);
66extern long cg_read_long(const char *cgroup, const char *control);
67extern long cg_read_long_fd(int fd);
68long cg_read_key_long(const char *cgroup, const char *control, const char *key);
69extern long cg_read_lc(const char *cgroup, const char *control);
70extern int cg_write(const char *cgroup, const char *control, char *buf);
71extern int cg_open(const char *cgroup, const char *control, int flags);
72int cg_write_numeric(const char *cgroup, const char *control, long value);
73extern int cg_run(const char *cgroup,
74		  int (*fn)(const char *cgroup, void *arg),
75		  void *arg);
76extern int cg_enter(const char *cgroup, int pid);
77extern int cg_enter_current(const char *cgroup);
78extern int cg_enter_current_thread(const char *cgroup);
79extern int cg_run_nowait(const char *cgroup,
80			 int (*fn)(const char *cgroup, void *arg),
81			 void *arg);
82extern int cg_wait_for_proc_count(const char *cgroup, int count);
83extern int cg_killall(const char *cgroup);
84int proc_mount_contains(const char *option);
85int cgroup_feature(const char *feature);
86extern ssize_t proc_read_text(int pid, bool thread, const char *item, char *buf, size_t size);
87extern int proc_read_strstr(int pid, bool thread, const char *item, const char *needle);
88extern pid_t clone_into_cgroup(int cgroup_fd);
89extern int clone_reap(pid_t pid, int options);
90extern int clone_into_cgroup_run_wait(const char *cgroup);
91extern int dirfd_open_opath(const char *dir);
92extern int cg_prepare_for_wait(const char *cgroup);
93extern int memcg_prepare_for_wait(const char *cgroup);
94extern int cg_wait_for(int fd);
95extern bool cg_test_v1_named;