Path: blob/master/tools/testing/selftests/cgroup/lib/include/cgroup_util.h
49329 views
/* SPDX-License-Identifier: GPL-2.0 */1#include <stdbool.h>2#include <stdlib.h>34#ifndef PAGE_SIZE5#define PAGE_SIZE 40966#endif78#define MB(x) (x << 20)910#define USEC_PER_SEC 1000000L11#define NSEC_PER_SEC 1000000000L1213#define TEST_UID 65534 /* usually nobody, any !root is fine */1415#define CG_THREADS_FILE (!cg_test_v1_named ? "cgroup.threads" : "tasks")16#define CG_NAMED_NAME "selftest"17#define CG_PATH_FORMAT (!cg_test_v1_named ? "0::%s" : (":name=" CG_NAMED_NAME ":%s"))1819/*20* Checks if two given values differ by less than err% of their sum.21*/22static inline int values_close(long a, long b, int err)23{24return labs(a - b) <= (a + b) / 100 * err;25}2627/*28* Checks if two given values differ by less than err% of their sum and assert29* with detailed debug info if not.30*/31static inline int values_close_report(long a, long b, int err)32{33long diff = labs(a - b);34long limit = (a + b) / 100 * err;35double actual_err = (a + b) ? (100.0 * diff / (a + b)) : 0.0;36int close = diff <= limit;3738if (!close)39fprintf(stderr,40"[FAIL] actual=%ld expected=%ld | diff=%ld | limit=%ld | "41"tolerance=%d%% | actual_error=%.2f%%\n",42a, b, diff, limit, err, actual_err);4344return close;45}4647extern ssize_t read_text(const char *path, char *buf, size_t max_len);48extern ssize_t write_text(const char *path, char *buf, ssize_t len);4950extern int cg_find_controller_root(char *root, size_t len, const char *controller);51extern int cg_find_unified_root(char *root, size_t len, bool *nsdelegate);52extern char *cg_name(const char *root, const char *name);53extern char *cg_name_indexed(const char *root, const char *name, int index);54extern char *cg_control(const char *cgroup, const char *control);55extern int cg_create(const char *cgroup);56extern int cg_destroy(const char *cgroup);57extern int cg_read(const char *cgroup, const char *control,58char *buf, size_t len);59extern int cg_read_strcmp(const char *cgroup, const char *control,60const char *expected);61extern int cg_read_strstr(const char *cgroup, const char *control,62const char *needle);63extern long cg_read_long(const char *cgroup, const char *control);64extern long cg_read_long_fd(int fd);65long cg_read_key_long(const char *cgroup, const char *control, const char *key);66extern long cg_read_lc(const char *cgroup, const char *control);67extern int cg_write(const char *cgroup, const char *control, char *buf);68extern int cg_open(const char *cgroup, const char *control, int flags);69int cg_write_numeric(const char *cgroup, const char *control, long value);70extern int cg_run(const char *cgroup,71int (*fn)(const char *cgroup, void *arg),72void *arg);73extern int cg_enter(const char *cgroup, int pid);74extern int cg_enter_current(const char *cgroup);75extern int cg_enter_current_thread(const char *cgroup);76extern int cg_run_nowait(const char *cgroup,77int (*fn)(const char *cgroup, void *arg),78void *arg);79extern int cg_wait_for_proc_count(const char *cgroup, int count);80extern int cg_killall(const char *cgroup);81int proc_mount_contains(const char *option);82int cgroup_feature(const char *feature);83extern ssize_t proc_read_text(int pid, bool thread, const char *item, char *buf, size_t size);84extern int proc_read_strstr(int pid, bool thread, const char *item, const char *needle);85extern pid_t clone_into_cgroup(int cgroup_fd);86extern int clone_reap(pid_t pid, int options);87extern int clone_into_cgroup_run_wait(const char *cgroup);88extern int dirfd_open_opath(const char *dir);89extern int cg_prepare_for_wait(const char *cgroup);90extern int memcg_prepare_for_wait(const char *cgroup);91extern int cg_wait_for(int fd);92extern bool cg_test_v1_named;939495