From 9414f3be0c3f191fdea212ecc3157af5e70a45de Mon Sep 17 00:00:00 2001 From: Dhaval Giani Date: Wed, 22 Apr 2009 15:30:19 +0530 Subject: libcgroup: Add new API to walk tasks Add a new API to iterate through the tasks file to get the list of all the tasks in a cgroup. Signed-off-by: Dhaval Giani Acked-by: Balbir Singh Acked-by: Bharata B Rao --- src/api.c | 68 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 68 insertions(+) (limited to 'src/api.c') diff --git a/src/api.c b/src/api.c index 4d5b524..baeb856 100644 --- a/src/api.c +++ b/src/api.c @@ -2407,3 +2407,71 @@ int cgroup_read_stats_begin(char *controller, char *path, void **handle, *handle = fp; return ret; } + +int cgroup_get_task_end(void **handle) +{ + if (!cgroup_initialized) + return ECGROUPNOTINITIALIZED; + + if (!*handle) + return ECGINVAL; + + fclose((FILE *) *handle); + *handle = NULL; + + return 0; +} + +int cgroup_get_task_next(void *handle, pid_t *pid) +{ + int ret; + + if (!cgroup_initialized) + return ECGROUPNOTINITIALIZED; + + if (!handle) + return ECGINVAL; + + ret = fscanf((FILE *) handle, "%u", pid); + + if (ret != 1) { + if (ret == EOF) + return ECGEOF; + last_errno = errno; + return ECGOTHER; + } + + return 0; +} + +int cgroup_get_task_begin(char *cgroup, char *controller, void **handle, + pid_t *pid) +{ + int ret = 0; + char path[FILENAME_MAX]; + char *fullpath = NULL; + + if (!cgroup_initialized) + return ECGROUPNOTINITIALIZED; + + if (!cg_build_path(cgroup, path, controller)) + return ECGOTHER; + + ret = asprintf(&fullpath, "%s/tasks", path); + + if (ret < 0) { + last_errno = errno; + return ECGOTHER; + } + + *handle = (void *) fopen(fullpath, "r"); + free(fullpath); + + if (!*handle) { + last_errno = errno; + return ECGOTHER; + } + ret = cgroup_get_task_next(*handle, pid); + + return ret; +} -- cgit