selftests/cgroup/cgroup_util.h

0001 /* SPDX-License-Identifier: GPL-2.0 */
0002 #include <stdbool.h>
0003 #include <stdlib.h>
0004
0005 #include "../kselftest.h"
0006
0007 #define PAGE_SIZE 4096
0008
0009 #define MB(x) (x << 20)
0010
0011 #define USEC_PER_SEC    1000000L
0012 #define NSEC_PER_SEC    1000000000L
0013
0014 /*
0015  * Checks if two given values differ by less than err% of their sum.
0016  */
0017 static inline int values_close(long a, long b, int err)
0018 {
0019     return abs(a - b) <= (a + b) / 100 * err;
0020 }
0021
0022 extern int cg_find_unified_root(char *root, size_t len);
0023 extern char *cg_name(const char *root, const char *name);
0024 extern char *cg_name_indexed(const char *root, const char *name, int index);
0025 extern char *cg_control(const char *cgroup, const char *control);
0026 extern int cg_create(const char *cgroup);
0027 extern int cg_destroy(const char *cgroup);
0028 extern int cg_read(const char *cgroup, const char *control,
0029            char *buf, size_t len);
0030 extern int cg_read_strcmp(const char *cgroup, const char *control,
0031               const char *expected);
0032 extern int cg_read_strstr(const char *cgroup, const char *control,
0033               const char *needle);
0034 extern long cg_read_long(const char *cgroup, const char *control);
0035 long cg_read_key_long(const char *cgroup, const char *control, const char *key);
0036 extern long cg_read_lc(const char *cgroup, const char *control);
0037 extern int cg_write(const char *cgroup, const char *control, char *buf);
0038 int cg_write_numeric(const char *cgroup, const char *control, long value);
0039 extern int cg_run(const char *cgroup,
0040           int (*fn)(const char *cgroup, void *arg),
0041           void *arg);
0042 extern int cg_enter(const char *cgroup, int pid);
0043 extern int cg_enter_current(const char *cgroup);
0044 extern int cg_enter_current_thread(const char *cgroup);
0045 extern int cg_run_nowait(const char *cgroup,
0046              int (*fn)(const char *cgroup, void *arg),
0047              void *arg);
0048 extern int get_temp_fd(void);
0049 extern int alloc_pagecache(int fd, size_t size);
0050 extern int alloc_anon(const char *cgroup, void *arg);
0051 extern int is_swap_enabled(void);
0052 extern int set_oom_adj_score(int pid, int score);
0053 extern int cg_wait_for_proc_count(const char *cgroup, int count);
0054 extern int cg_killall(const char *cgroup);
0055 int proc_mount_contains(const char *option);
0056 extern ssize_t proc_read_text(int pid, bool thread, const char *item, char *buf, size_t size);
0057 extern int proc_read_strstr(int pid, bool thread, const char *item, const char *needle);
0058 extern pid_t clone_into_cgroup(int cgroup_fd);
0059 extern int clone_reap(pid_t pid, int options);
0060 extern int clone_into_cgroup_run_wait(const char *cgroup);
0061 extern int dirfd_open_opath(const char *dir);
0062 extern int cg_prepare_for_wait(const char *cgroup);
0063 extern int memcg_prepare_for_wait(const char *cgroup);
0064 extern int cg_wait_for(int fd);