cgroup: fix alloca() misuse in cg_shorten_controllers()

This commit is contained in:
Lennart Poettering 2012-05-03 23:23:38 +02:00
parent 6e476bc9d1
commit 37099707e2
2 changed files with 24 additions and 11 deletions

2
TODO
View File

@ -60,8 +60,6 @@ Features:
* add RequiredBy to [Install] * add RequiredBy to [Install]
* cg_shorten_controllers() misuses alloca()
* udev: move to LGPL * udev: move to LGPL
* udev systemd unify: * udev systemd unify:

View File

@ -565,9 +565,23 @@ int cg_get_path(const char *controller, const char *path, const char *suffix, ch
return join_path(p, path, suffix, fs); return join_path(p, path, suffix, fs);
} }
static int check(const char *p) {
char *cc;
assert(p);
/* Check if this controller actually really exists */
cc = alloca(sizeof("/sys/fs/cgroup/") + strlen(p));
strcpy(stpcpy(cc, "/sys/fs/cgroup/"), p);
if (access(cc, F_OK) < 0)
return -errno;
return 0;
}
int cg_get_path_and_check(const char *controller, const char *path, const char *suffix, char **fs) { int cg_get_path_and_check(const char *controller, const char *path, const char *suffix, char **fs) {
const char *p; const char *p;
char *cc; int r;
assert(controller); assert(controller);
assert(fs); assert(fs);
@ -575,13 +589,13 @@ int cg_get_path_and_check(const char *controller, const char *path, const char *
if (isempty(controller)) if (isempty(controller))
return -EINVAL; return -EINVAL;
/* Normalize the controller syntax */
p = normalize_controller(controller); p = normalize_controller(controller);
/* Check if this controller actually really exists */ /* Check if this controller actually really exists */
cc = alloca(sizeof("/sys/fs/cgroup/") + strlen(p)); r = check(p);
strcpy(stpcpy(cc, "/sys/fs/cgroup/"), p); if (r < 0)
if (access(cc, F_OK) < 0) return r;
return -errno;
return join_path(p, path, suffix, fs); return join_path(p, path, suffix, fs);
} }
@ -1111,17 +1125,18 @@ char **cg_shorten_controllers(char **controllers) {
return controllers; return controllers;
for (f = controllers, t = controllers; *f; f++) { for (f = controllers, t = controllers; *f; f++) {
char *cc; int r;
const char *p;
if (streq(*f, "systemd") || streq(*f, SYSTEMD_CGROUP_CONTROLLER)) { if (streq(*f, "systemd") || streq(*f, SYSTEMD_CGROUP_CONTROLLER)) {
free(*f); free(*f);
continue; continue;
} }
cc = alloca(sizeof("/sys/fs/cgroup/") + strlen(*f)); p = normalize_controller(*f);
strcpy(stpcpy(cc, "/sys/fs/cgroup/"), *f);
if (access(cc, F_OK) < 0) { r = check(p);
if (r < 0) {
log_debug("Controller %s is not available, removing from controllers list.", *f); log_debug("Controller %s is not available, removing from controllers list.", *f);
free(*f); free(*f);
continue; continue;