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 /* 17 * Checks if two given values differ by less than err% of their sum. 18 */ 19 static inline int values_close(long a, long b, int err) 20 { 21 return labs(a - b) <= (a + b) / 100 * err; 22 } 23 24 extern ssize_t read_text(const char *path, char *buf, size_t max_len); 25 extern ssize_t write_text(const char *path, char *buf, ssize_t len); 26 27 extern int cg_find_controller_root(char *root, size_t len, const char *controller); 28 extern int cg_find_unified_root(char *root, size_t len, bool *nsdelegate); 29 extern char *cg_name(const char *root, const char *name); 30 extern char *cg_name_indexed(const char *root, const char *name, int index); 31 extern char *cg_control(const char *cgroup, const char *control); 32 extern int cg_create(const char *cgroup); 33 extern int cg_destroy(const char *cgroup); 34 extern int cg_read(const char *cgroup, const char *control, 35 char *buf, size_t len); 36 extern int cg_read_strcmp(const char *cgroup, const char *control, 37 const char *expected); 38 extern int cg_read_strstr(const char *cgroup, const char *control, 39 const char *needle); 40 extern long cg_read_long(const char *cgroup, const char *control); 41 extern long cg_read_long_fd(int fd); 42 long cg_read_key_long(const char *cgroup, const char *control, const char *key); 43 extern long cg_read_lc(const char *cgroup, const char *control); 44 extern int cg_write(const char *cgroup, const char *control, char *buf); 45 extern int cg_open(const char *cgroup, const char *control, int flags); 46 int cg_write_numeric(const char *cgroup, const char *control, long value); 47 extern int cg_run(const char *cgroup, 48 int (*fn)(const char *cgroup, void *arg), 49 void *arg); 50 extern int cg_enter(const char *cgroup, int pid); 51 extern int cg_enter_current(const char *cgroup); 52 extern int cg_enter_current_thread(const char *cgroup); 53 extern int cg_run_nowait(const char *cgroup, 54 int (*fn)(const char *cgroup, void *arg), 55 void *arg); 56 extern int cg_wait_for_proc_count(const char *cgroup, int count); 57 extern int cg_killall(const char *cgroup); 58 int proc_mount_contains(const char *option); 59 extern ssize_t proc_read_text(int pid, bool thread, const char *item, char *buf, size_t size); 60 extern int proc_read_strstr(int pid, bool thread, const char *item, const char *needle); 61 extern pid_t clone_into_cgroup(int cgroup_fd); 62 extern int clone_reap(pid_t pid, int options); 63 extern int clone_into_cgroup_run_wait(const char *cgroup); 64 extern int dirfd_open_opath(const char *dir); 65 extern int cg_prepare_for_wait(const char *cgroup); 66 extern int memcg_prepare_for_wait(const char *cgroup); 67 extern int cg_wait_for(int fd); 68