69 lines · c
1/* SPDX-License-Identifier: GPL-2.0 */2#include <stdbool.h>3#include <stdlib.h>4 5#include "../kselftest.h"6 7#define PAGE_SIZE 40968 9#define MB(x) (x << 20)10 11#define USEC_PER_SEC 1000000L12#define NSEC_PER_SEC 1000000000L13 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 */19static inline int values_close(long a, long b, int err)20{21 return labs(a - b) <= (a + b) / 100 * err;22}23 24extern int cg_find_unified_root(char *root, size_t len, bool *nsdelegate);25extern char *cg_name(const char *root, const char *name);26extern char *cg_name_indexed(const char *root, const char *name, int index);27extern char *cg_control(const char *cgroup, const char *control);28extern int cg_create(const char *cgroup);29extern int cg_destroy(const char *cgroup);30extern int cg_read(const char *cgroup, const char *control,31 char *buf, size_t len);32extern int cg_read_strcmp(const char *cgroup, const char *control,33 const char *expected);34extern int cg_read_strstr(const char *cgroup, const char *control,35 const char *needle);36extern long cg_read_long(const char *cgroup, const char *control);37extern long cg_read_long_fd(int fd);38long cg_read_key_long(const char *cgroup, const char *control, const char *key);39extern long cg_read_lc(const char *cgroup, const char *control);40extern int cg_write(const char *cgroup, const char *control, char *buf);41extern int cg_open(const char *cgroup, const char *control, int flags);42int cg_write_numeric(const char *cgroup, const char *control, long value);43extern int cg_run(const char *cgroup,44 int (*fn)(const char *cgroup, void *arg),45 void *arg);46extern int cg_enter(const char *cgroup, int pid);47extern int cg_enter_current(const char *cgroup);48extern int cg_enter_current_thread(const char *cgroup);49extern int cg_run_nowait(const char *cgroup,50 int (*fn)(const char *cgroup, void *arg),51 void *arg);52extern int get_temp_fd(void);53extern int alloc_pagecache(int fd, size_t size);54extern int alloc_anon(const char *cgroup, void *arg);55extern int is_swap_enabled(void);56extern int set_oom_adj_score(int pid, int score);57extern int cg_wait_for_proc_count(const char *cgroup, int count);58extern int cg_killall(const char *cgroup);59int proc_mount_contains(const char *option);60extern ssize_t proc_read_text(int pid, bool thread, const char *item, char *buf, size_t size);61extern int proc_read_strstr(int pid, bool thread, const char *item, const char *needle);62extern pid_t clone_into_cgroup(int cgroup_fd);63extern int clone_reap(pid_t pid, int options);64extern int clone_into_cgroup_run_wait(const char *cgroup);65extern int dirfd_open_opath(const char *dir);66extern int cg_prepare_for_wait(const char *cgroup);67extern int memcg_prepare_for_wait(const char *cgroup);68extern int cg_wait_for(int fd);69