X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=cgroup.c;h=a297e2ac82e82f5145edfcb9cc509b23134a360b;hp=b0c7ea2003ac1a78937c45d562a57809fdd4e739;hb=ae626d4ead6416adf464cf209cdf3e8b85d58190;hpb=39f22027248f658ade599e89c2fe6afae02ac9d7 diff --git a/cgroup.c b/cgroup.c index b0c7ea20..a297e2ac 100644 --- a/cgroup.c +++ b/cgroup.c @@ -3,187 +3,204 @@ */ #include #include +#include +#include +#include #include "fio.h" #include "flist.h" #include "cgroup.h" #include "smalloc.h" -static struct flist_head *cgroup_list; static struct fio_mutex *lock; struct cgroup_member { struct flist_head list; char *root; + unsigned int cgroup_nodelete; }; -static void add_cgroup(const char *name) +static char *find_cgroup_mnt(struct thread_data *td) +{ + char *mntpoint = NULL; + struct mntent *mnt, dummy; + char buf[256] = {0}; + FILE *f; + + f = setmntent("/proc/mounts", "r"); + if (!f) { + td_verror(td, errno, "setmntent /proc/mounts"); + return NULL; + } + + while ((mnt = getmntent_r(f, &dummy, buf, sizeof(buf))) != NULL) { + if (!strcmp(mnt->mnt_type, "cgroup") && + strstr(mnt->mnt_opts, "blkio")) + break; + } + + if (mnt) + mntpoint = smalloc_strdup(mnt->mnt_dir); + else + log_err("fio: cgroup blkio does not appear to be mounted\n"); + + endmntent(f); + return mntpoint; +} + +static void add_cgroup(struct thread_data *td, const char *name, + struct flist_head *clist) { struct cgroup_member *cm; + if (!lock) + return; + cm = smalloc(sizeof(*cm)); + if (!cm) { +err: + log_err("fio: failed to allocate cgroup member\n"); + return; + } + INIT_FLIST_HEAD(&cm->list); cm->root = smalloc_strdup(name); - - fio_mutex_down(lock); - - if (!cgroup_list) { - cgroup_list = smalloc(sizeof(struct flist_head)); - INIT_FLIST_HEAD(cgroup_list); + if (!cm->root) { + sfree(cm); + goto err; } - - flist_add_tail(&cm->list, cgroup_list); + if (td->o.cgroup_nodelete) + cm->cgroup_nodelete = 1; + fio_mutex_down(lock); + flist_add_tail(&cm->list, clist); fio_mutex_up(lock); } -void cgroup_kill(void) +void cgroup_kill(struct flist_head *clist) { struct flist_head *n, *tmp; struct cgroup_member *cm; + if (!lock) + return; + fio_mutex_down(lock); - if (!cgroup_list) - goto out; - flist_for_each_safe(n, tmp, cgroup_list) { + flist_for_each_safe(n, tmp, clist) { cm = flist_entry(n, struct cgroup_member, list); - rmdir(cm->root); + if (!cm->cgroup_nodelete) + rmdir(cm->root); flist_del(&cm->list); sfree(cm->root); sfree(cm); } - sfree(cgroup_list); - cgroup_list = NULL; -out: fio_mutex_up(lock); } -/* - * Check if the given root appears valid - */ -static int cgroup_check_fs(struct thread_data *td) -{ - struct stat sb; - char tmp[256]; - - sprintf(tmp, "%s/tasks", td->o.cgroup_root); - return stat(tmp, &sb); -} - -static char *get_cgroup_root(struct thread_data *td) +static char *get_cgroup_root(struct thread_data *td, char *mnt) { char *str = malloc(64); if (td->o.cgroup) - sprintf(str, "%s/%s", td->o.cgroup_root, td->o.cgroup); + sprintf(str, "%s/%s", mnt, td->o.cgroup); else - sprintf(str, "%s/%s", td->o.cgroup_root, td->o.name); + sprintf(str, "%s/%s", mnt, td->o.name); return str; } -/* - * Add pid to given class - */ -static int cgroup_add_pid(struct thread_data *td) +static int write_int_to_file(struct thread_data *td, const char *path, + const char *filename, unsigned int val, + const char *onerr) { - char *root, tmp[256]; + char tmp[256]; FILE *f; - root = get_cgroup_root(td); - sprintf(tmp, "%s/tasks", root); + sprintf(tmp, "%s/%s", path, filename); f = fopen(tmp, "w"); if (!f) { - td_verror(td, errno, "cgroup open tasks"); + td_verror(td, errno, onerr); return 1; } - fprintf(f, "%d", td->pid); + fprintf(f, "%u", val); fclose(f); - free(root); return 0; + +} + +static int cgroup_write_pid(struct thread_data *td, const char *root) +{ + unsigned int val = td->pid; + + return write_int_to_file(td, root, "tasks", val, "cgroup write pid"); } /* * Move pid to root class */ -static int cgroup_del_pid(struct thread_data *td) +static int cgroup_del_pid(struct thread_data *td, char *mnt) { - char tmp[256]; - FILE *f; - - sprintf(tmp, "%s/tasks", td->o.cgroup_root); - f = fopen(tmp, "w"); - if (!f) { - td_verror(td, errno, "cgroup open tasks"); - return 1; - } - - fprintf(f, "%d", td->pid); - fclose(f); - return 0; + return cgroup_write_pid(td, mnt); } - -int cgroup_setup(struct thread_data *td) +int cgroup_setup(struct thread_data *td, struct flist_head *clist, char **mnt) { - char *root, tmp[256]; - FILE *f; + char *root; - if (cgroup_check_fs(td)) { - log_err("fio: blkio cgroup mount point %s not valid\n", - td->o.cgroup_root); - return 1; + if (!*mnt) { + *mnt = find_cgroup_mnt(td); + if (!*mnt) + return 1; } /* * Create container, if it doesn't exist */ - root = get_cgroup_root(td); + root = get_cgroup_root(td, *mnt); if (mkdir(root, 0755) < 0) { int __e = errno; if (__e != EEXIST) { td_verror(td, __e, "cgroup mkdir"); - return 1; + log_err("fio: path %s\n", root); + goto err; } } else - add_cgroup(root); + add_cgroup(td, root, clist); if (td->o.cgroup_weight) { - sprintf(tmp, "%s/blkio.weight", root); - f = fopen(tmp, "w"); - if (!f) { - td_verror(td, errno, "cgroup open weight"); - return 1; - } + if (write_int_to_file(td, root, "blkio.weight", + td->o.cgroup_weight, + "cgroup open weight")) + goto err; + } - fprintf(f, "%d", td->o.cgroup_weight); - fclose(f); + if (!cgroup_write_pid(td, root)) { + free(root); + return 0; } +err: free(root); - - if (cgroup_add_pid(td)) - return 1; - - return 0; + return 1; } -void cgroup_shutdown(struct thread_data *td) +void cgroup_shutdown(struct thread_data *td, char **mnt) { - if (cgroup_check_fs(td)) + if (*mnt == NULL) return; - if (!td->o.cgroup_weight && td->o.cgroup) + if (!td->o.cgroup_weight && !td->o.cgroup) return; - cgroup_del_pid(td); + cgroup_del_pid(td, *mnt); } - static void fio_init cgroup_init(void) { - lock = fio_mutex_init(1); + lock = fio_mutex_init(FIO_MUTEX_UNLOCKED); + if (!lock) + log_err("fio: failed to allocate cgroup lock\n"); } static void fio_exit cgroup_exit(void)