From: "S.Çağlar Onur" <cag...@10ur.org> Trying to stop multiple containers concurrently ends up with "cgroup is not mounted" errors as multiple threads corrupts the shared variables. Fix that stack corruption and start to use getmntent_r to support stopping multiple containers concurrently. --- src/lxc/cgroup.c | 152 +++++++++++++++++++++++++++++++++++------------------ src/lxc/freezer.c | 18 +++++-- src/lxc/state.c | 15 ++++-- 3 files changed, 126 insertions(+), 59 deletions(-)
diff --git a/src/lxc/cgroup.c b/src/lxc/cgroup.c index 812bfb8..7212f01 100644 --- a/src/lxc/cgroup.c +++ b/src/lxc/cgroup.c @@ -54,6 +54,11 @@ lxc_log_define(lxc_cgroup, lxc); #define MTAB "/proc/mounts" +/* In the case of a bind mount, there could be two long pathnames in the + * mntent plus options so use large enough buffer size + */ +#define LARGE_MAXPATHLEN 4 * MAXPATHLEN + /* Check if a mount is a cgroup hierarchy for any subsystem. * Return the first subsystem found (or NULL if none). */ @@ -98,29 +103,31 @@ static char *mount_has_subsystem(const struct mntent *mntent) */ static int get_cgroup_mount(const char *subsystem, char *mnt) { - struct mntent *mntent; + struct mntent *mntent, mntent_r; FILE *file = NULL; int ret, err = -1; + char buf[LARGE_MAXPATHLEN] = {0}; + file = setmntent(MTAB, "r"); if (!file) { SYSERROR("failed to open %s", MTAB); return -1; } - while ((mntent = getmntent(file))) { - if (strcmp(mntent->mnt_type, "cgroup")) + while ((mntent = getmntent_r(file, &mntent_r, buf, sizeof(buf)))) { + if (strcmp(mntent_r.mnt_type, "cgroup") != 0) continue; - + if (subsystem) { - if (!hasmntopt(mntent, subsystem)) + if (!hasmntopt(&mntent_r, subsystem)) continue; } else { - if (!mount_has_subsystem(mntent)) + if (!mount_has_subsystem(&mntent_r)) continue; } - ret = snprintf(mnt, MAXPATHLEN, "%s", mntent->mnt_dir); + ret = snprintf(mnt, MAXPATHLEN, "%s", mntent_r.mnt_dir); if (ret < 0 || ret >= MAXPATHLEN) goto fail; @@ -146,22 +153,33 @@ out: * * Returns 0 on success, -1 on error. * - * The answer is written in a static char[MAXPATHLEN] in this function and - * should not be freed. */ extern int cgroup_path_get(char **path, const char *subsystem, const char *cgpath) { - static char buf[MAXPATHLEN]; - static char retbuf[MAXPATHLEN]; int rc; + char *buf = NULL; + char *retbuf = NULL; + + buf = malloc(MAXPATHLEN * sizeof(char)); + if (!buf) { + ERROR("malloc failed"); + goto fail; + } + + retbuf = malloc(MAXPATHLEN * sizeof(char)); + if (!retbuf) { + ERROR("malloc failed"); + goto fail; + } + /* lxc_cgroup_set passes a state object for the subsystem, * so trim it to just the subsystem part */ if (subsystem) { rc = snprintf(retbuf, MAXPATHLEN, "%s", subsystem); if (rc < 0 || rc >= MAXPATHLEN) { ERROR("subsystem name too long"); - return -1; + goto fail; } char *s = index(retbuf, '.'); if (s) @@ -170,19 +188,28 @@ extern int cgroup_path_get(char **path, const char *subsystem, const char *cgpat } if (get_cgroup_mount(subsystem ? retbuf : NULL, buf)) { ERROR("cgroup is not mounted"); - return -1; + goto fail; } rc = snprintf(retbuf, MAXPATHLEN, "%s/%s", buf, cgpath); if (rc < 0 || rc >= MAXPATHLEN) { ERROR("name too long"); - return -1; + goto fail; } DEBUG("%s: returning %s for subsystem %s", __func__, retbuf, subsystem); + if(buf) + free(buf); + *path = retbuf; return 0; +fail: + if (buf) + free(buf); + if (retbuf) + free(retbuf); + return -1; } /* @@ -290,20 +317,25 @@ static int do_cgroup_set(const char *path, const char *value) int lxc_cgroup_set_bypath(const char *cgpath, const char *filename, const char *value) { int ret; - char *dirpath; + char *dirpath = NULL; char path[MAXPATHLEN]; ret = cgroup_path_get(&dirpath, filename, cgpath); if (ret) - return -1; + goto fail; ret = snprintf(path, MAXPATHLEN, "%s/%s", dirpath, filename); if (ret < 0 || ret >= MAXPATHLEN) { ERROR("pathname too long"); - return -1; + goto fail; } return do_cgroup_set(path, value); + +fail: + if(dirpath) + free(dirpath); + return -1; } /* @@ -321,20 +353,25 @@ int lxc_cgroup_set(const char *name, const char *filename, const char *value, const char *lxcpath) { int ret; - char *dirpath; + char *dirpath = NULL; char path[MAXPATHLEN]; ret = lxc_cgroup_path_get(&dirpath, filename, name, lxcpath); if (ret) - return -1; + goto fail; ret = snprintf(path, MAXPATHLEN, "%s/%s", dirpath, filename); if (ret < 0 || ret >= MAXPATHLEN) { ERROR("pathname too long"); - return -1; + goto fail; } return do_cgroup_set(path, value); + +fail: + if(dirpath) + free(dirpath); + return -1; } /* @@ -361,24 +398,24 @@ int lxc_cgroup_get(const char *name, const char *filename, char *value, size_t len, const char *lxcpath) { int fd, ret = -1; - char *dirpath; + char *dirpath = NULL; char path[MAXPATHLEN]; int rc; ret = lxc_cgroup_path_get(&dirpath, filename, name, lxcpath); if (ret) - return -1; + goto fail; rc = snprintf(path, MAXPATHLEN, "%s/%s", dirpath, filename); if (rc < 0 || rc >= MAXPATHLEN) { ERROR("pathname too long"); - return -1; + goto fail; } fd = open(path, O_RDONLY); if (fd < 0) { ERROR("open %s : %s", path, strerror(errno)); - return -1; + goto fail; } if (!len || !value) { @@ -398,24 +435,28 @@ int lxc_cgroup_get(const char *name, const char *filename, char *value, close(fd); return ret; +fail: + if(dirpath) + free(dirpath); + return -1; } int lxc_cgroup_nrtasks(const char *cgpath) { - char *dpath; + char *dirpath = NULL; char path[MAXPATHLEN]; int pid, ret, count = 0; FILE *file; int rc; - ret = cgroup_path_get(&dpath, NULL, cgpath); + ret = cgroup_path_get(&dirpath, NULL, cgpath); if (ret) - return -1; + goto fail; - rc = snprintf(path, MAXPATHLEN, "%s/tasks", dpath); + rc = snprintf(path, MAXPATHLEN, "%s/tasks", dirpath); if (rc < 0 || rc >= MAXPATHLEN) { ERROR("pathname too long"); - return -1; + goto fail; } file = fopen(path, "r"); @@ -430,6 +471,10 @@ int lxc_cgroup_nrtasks(const char *cgpath) fclose(file); return count; +fail: + if(dirpath) + free(dirpath); + return -1; } /* @@ -470,21 +515,23 @@ static void set_clone_children(const char *mntdir) static int create_lxcgroups(const char *lxcgroup) { FILE *file = NULL; - struct mntent *mntent; + struct mntent *mntent, mntent_r; int ret, retv = -1; char path[MAXPATHLEN]; + char buf[LARGE_MAXPATHLEN] = {0}; + file = setmntent(MTAB, "r"); if (!file) { SYSERROR("failed to open %s", MTAB); return -1; } - while ((mntent = getmntent(file))) { + while ((mntent = getmntent_r(file, &mntent_r, buf, sizeof(buf)))) { - if (strcmp(mntent->mnt_type, "cgroup")) + if (strcmp(mntent_r.mnt_type, "cgroup")) continue; - if (!mount_has_subsystem(mntent)) + if (!mount_has_subsystem(&mntent_r)) continue; /* @@ -492,11 +539,11 @@ static int create_lxcgroups(const char *lxcgroup) * We probably only want to support that for /users/joe */ ret = snprintf(path, MAXPATHLEN, "%s/%s", - mntent->mnt_dir, lxcgroup ? lxcgroup : "lxc"); + mntent_r.mnt_dir, lxcgroup ? lxcgroup : "lxc"); if (ret < 0 || ret >= MAXPATHLEN) goto fail; if (access(path, F_OK)) { - set_clone_children(mntent->mnt_dir); + set_clone_children(mntent_r.mnt_dir); ret = mkdir(path, 0755); if (ret == -1 && errno != EEXIST) { SYSERROR("failed to create '%s' directory", path); @@ -542,7 +589,9 @@ char *lxc_cgroup_path_create(const char *lxcgroup, const char *name) char *retpath, path[MAXPATHLEN]; char tail[12]; FILE *file = NULL; - struct mntent *mntent; + struct mntent *mntent, mntent_r; + + char buf[LARGE_MAXPATHLEN] = {0}; if (create_lxcgroups(lxcgroup) < 0) return NULL; @@ -559,15 +608,15 @@ again: else *tail = '\0'; - while ((mntent = getmntent(file))) { + while ((mntent = getmntent_r(file, &mntent_r, buf, sizeof(buf)))) { - if (strcmp(mntent->mnt_type, "cgroup")) + if (strcmp(mntent_r.mnt_type, "cgroup")) continue; - if (!mount_has_subsystem(mntent)) + if (!mount_has_subsystem(&mntent_r)) continue; /* find unused mnt_dir + lxcgroup + name + -$i */ - ret = snprintf(path, MAXPATHLEN, "%s/%s/%s%s", mntent->mnt_dir, + ret = snprintf(path, MAXPATHLEN, "%s/%s/%s%s", mntent_r.mnt_dir, lxcgroup ? lxcgroup : "lxc", name, tail); if (ret < 0 || ret >= MAXPATHLEN) goto fail; @@ -607,8 +656,9 @@ int lxc_cgroup_enter(const char *cgpath, pid_t pid) { char path[MAXPATHLEN]; FILE *file = NULL, *fout; - struct mntent *mntent; + struct mntent *mntent, mntent_r; int ret, retv = -1; + char buf[LARGE_MAXPATHLEN] = {0}; file = setmntent(MTAB, "r"); if (!file) { @@ -616,13 +666,13 @@ int lxc_cgroup_enter(const char *cgpath, pid_t pid) return -1; } - while ((mntent = getmntent(file))) { - if (strcmp(mntent->mnt_type, "cgroup")) + while ((mntent = getmntent_r(file, &mntent_r, buf, sizeof(buf)))) { + if (strcmp(mntent_r.mnt_type, "cgroup")) continue; - if (!mount_has_subsystem(mntent)) + if (!mount_has_subsystem(&mntent_r)) continue; ret = snprintf(path, MAXPATHLEN, "%s/%s/tasks", - mntent->mnt_dir, cgpath); + mntent_r.mnt_dir, cgpath); if (ret < 0 || ret >= MAXPATHLEN) { ERROR("entering cgroup"); goto out; @@ -714,23 +764,25 @@ static int lxc_one_cgroup_destroy(struct mntent *mntent, const char *cgpath) */ int lxc_cgroup_destroy(const char *cgpath) { - struct mntent *mntent; + struct mntent *mntent, mntent_r; FILE *file = NULL; int err, retv = 0; + char buf[LARGE_MAXPATHLEN] = {0}; + file = setmntent(MTAB, "r"); if (!file) { SYSERROR("failed to open %s", MTAB); return -1; } - while ((mntent = getmntent(file))) { - if (strcmp(mntent->mnt_type, "cgroup")) + while ((mntent = getmntent_r(file, &mntent_r, buf, sizeof(buf)))) { + if (strcmp(mntent_r.mnt_type, "cgroup")) continue; - if (!mount_has_subsystem(mntent)) + if (!mount_has_subsystem(&mntent_r)) continue; - err = lxc_one_cgroup_destroy(mntent, cgpath); + err = lxc_one_cgroup_destroy(&mntent_r, cgpath); if (err) // keep trying to clean up the others retv = -1; } diff --git a/src/lxc/freezer.c b/src/lxc/freezer.c index 111bc35..35bf3a7 100644 --- a/src/lxc/freezer.c +++ b/src/lxc/freezer.c @@ -120,14 +120,19 @@ out: static int freeze_unfreeze(const char *name, int freeze, const char *lxcpath) { - char *nsgroup; + char *nsgroup = NULL; int ret; ret = lxc_cgroup_path_get(&nsgroup, "freezer", name, lxcpath); if (ret) - return -1; + goto fail; return do_unfreeze(nsgroup, freeze, name, lxcpath); + +fail: + if (nsgroup) + free(nsgroup); + return -1; } int lxc_freeze(const char *name, const char *lxcpath) @@ -143,12 +148,17 @@ int lxc_unfreeze(const char *name, const char *lxcpath) int lxc_unfreeze_bypath(const char *cgpath) { - char *nsgroup; + char *nsgroup = NULL; int ret; ret = cgroup_path_get(&nsgroup, "freezer", cgpath); if (ret) - return -1; + goto fail; return do_unfreeze(nsgroup, 0, NULL, NULL); + +fail: + if (nsgroup) + free(nsgroup); + return -1; } diff --git a/src/lxc/state.c b/src/lxc/state.c index c50ef00..89e438c 100644 --- a/src/lxc/state.c +++ b/src/lxc/state.c @@ -68,7 +68,7 @@ lxc_state_t lxc_str2state(const char *state) static int freezer_state(const char *name, const char *lxcpath) { - char *nsgroup; + char *nsgroup = NULL; char freezer[MAXPATHLEN]; char status[MAXPATHLEN]; FILE *file; @@ -76,25 +76,30 @@ static int freezer_state(const char *name, const char *lxcpath) err = lxc_cgroup_path_get(&nsgroup, "freezer", name, lxcpath); if (err) - return -1; + goto fail; err = snprintf(freezer, MAXPATHLEN, "%s/freezer.state", nsgroup); if (err < 0 || err >= MAXPATHLEN) - return -1; + goto fail; file = fopen(freezer, "r"); if (!file) - return -1; + goto fail; err = fscanf(file, "%s", status); fclose(file); if (err == EOF) { SYSERROR("failed to read %s", freezer); - return -1; + goto fail; } return lxc_str2state(status); + +fail: + if (nsgroup) + free(nsgroup); + return -1; } static lxc_state_t __lxc_getstate(const char *name, const char *lxcpath) -- 1.7.10.4 ------------------------------------------------------------------------------ Precog is a next-generation analytics platform capable of advanced analytics on semi-structured data. The platform includes APIs for building apps and a phenomenal toolset for data science. Developers can use our toolset for easy data analysis & visualization. Get a free account! http://www2.precog.com/precogplatform/slashdotnewsletter _______________________________________________ Lxc-devel mailing list Lxc-devel@lists.sourceforge.net https://lists.sourceforge.net/lists/listinfo/lxc-devel