X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Fbasic%2Fcgroup-util.c;h=3506ca8bffaf0c1b1085c3121c2b10d76e4deffb;hp=08259fcfc5dfcdfe305e808860a2b9aa39c76241;hb=27e59415d0da0580569d6c48434ebc3c18c0d73c;hpb=d1325d57413c007e17847074c70005c40e32bc9d diff --git a/src/basic/cgroup-util.c b/src/basic/cgroup-util.c index 08259fcfc..3506ca8bf 100644 --- a/src/basic/cgroup-util.c +++ b/src/basic/cgroup-util.c @@ -1,5 +1,3 @@ -/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ - /*** This file is part of systemd. @@ -19,28 +17,45 @@ along with systemd; If not, see . ***/ +#include #include -#include +#include +//#include #include -#include +//#include #include -#include +#include #include +//#include #include -#include +#include -#include "set.h" -#include "macro.h" -#include "util.h" +#include "alloc-util.h" +#include "cgroup-util.h" +//#include "def.h" +#include "dirent-util.h" +#include "extract-word.h" +#include "fd-util.h" +#include "fileio.h" #include "formats-util.h" -#include "process-util.h" +#include "fs-util.h" +//#include "log.h" +#include "login-util.h" +#include "macro.h" +//#include "missing.h" +#include "mkdir.h" +#include "parse-util.h" #include "path-util.h" +#include "proc-cmdline.h" +#include "process-util.h" +#include "set.h" +//#include "special.h" +#include "stat-util.h" +#include "stdio-util.h" +#include "string-table.h" +#include "string-util.h" #include "unit-name.h" -#include "fileio.h" -#include "special.h" -#include "mkdir.h" -#include "login-util.h" -#include "cgroup-util.h" +#include "user-util.h" int cg_enumerate_processes(const char *controller, const char *path, FILE **_f) { _cleanup_free_ char *fs = NULL; @@ -76,7 +91,7 @@ int cg_read_pid(FILE *f, pid_t *_pid) { if (feof(f)) return 0; - return errno ? -errno : -EIO; + return errno > 0 ? -errno : -EIO; } if (ul <= 0) @@ -297,6 +312,10 @@ int cg_migrate(const char *cfrom, const char *pfrom, const char *cto, const char my_pid = getpid(); + log_debug_elogind("Migrating \"%s\"/\"%s\" to \"%s\"/\"%s\" (%s)", + cfrom, pfrom, cto, pto, + ignore_self ? "ignoring self" : "watching self"); + do { _cleanup_fclose_ FILE *f = NULL; pid_t pid = 0; @@ -457,7 +476,7 @@ static const char *controller_to_dirname(const char *controller) { if (e) return e; - return controller; + return controller; } static int join_path_legacy(const char *controller, const char *path, const char *suffix, char **fs) { @@ -516,7 +535,7 @@ int cg_get_path(const char *controller, const char *path, const char *suffix, ch * *below* the controllers, without any prefix. */ if (!path && !suffix) - return -EINVAL; + return -EINVAL; if (!suffix) t = strdup(path); @@ -542,12 +561,12 @@ int cg_get_path(const char *controller, const char *path, const char *suffix, ch r = join_path_unified(path, suffix, fs); else r = join_path_legacy(controller, path, suffix, fs); - if (r < 0) - return r; + if (r < 0) + return r; path_kill_slashes(*fs); return 0; - } +} static int controller_is_accessible(const char *controller) { int unified; @@ -569,8 +588,8 @@ static int controller_is_accessible(const char *controller) { /* We don't support named hierarchies if we are using * the unified hierarchy. */ - if (streq(controller, ELOGIND_CGROUP_CONTROLLER)) - return 0; + if (streq(controller, SYSTEMD_CGROUP_CONTROLLER)) + return 0; if (startswith(controller, "name=")) return -EOPNOTSUPP; @@ -581,8 +600,8 @@ static int controller_is_accessible(const char *controller) { dn = controller_to_dirname(controller); cc = strjoina("/sys/fs/cgroup/", dn); - if (laccess(cc, F_OK) < 0) - return -errno; + if (laccess(cc, F_OK) < 0) + return -errno; } return 0; @@ -631,7 +650,7 @@ int cg_trim(const char *controller, const char *path, bool delete_root) { if (nftw(fs, trim_cb, 64, FTW_DEPTH|FTW_MOUNT|FTW_PHYS) != 0) { if (errno == ENOENT) r = 0; - else if (errno != 0) + else if (errno > 0) r = -errno; else r = -EIO; @@ -668,8 +687,6 @@ int cg_create(const char *controller, const char *path) { return 1; } -/// UNNEEDED by elogind -#if 0 int cg_create_and_attach(const char *controller, const char *path, pid_t pid) { int r, q; @@ -686,7 +703,6 @@ int cg_create_and_attach(const char *controller, const char *path, pid_t pid) { /* This does not remove the cgroup on failure */ return r; } -#endif // 0 int cg_attach(const char *controller, const char *path, pid_t pid) { _cleanup_free_ char *fs = NULL; @@ -703,9 +719,9 @@ int cg_attach(const char *controller, const char *path, pid_t pid) { if (pid == 0) pid = getpid(); - snprintf(c, sizeof(c), PID_FMT"\n", pid); + xsprintf(c, PID_FMT "\n", pid); - return write_string_file_no_create(fs, c); + return write_string_file(fs, c, 0); } int cg_attach_fallback(const char *controller, const char *path, pid_t pid) { @@ -734,8 +750,7 @@ int cg_attach_fallback(const char *controller, const char *path, pid_t pid) { return r; } -/// UNNEEDED by elogind -#if 0 +#if 0 /// UNNEEDED by elogind int cg_set_group_access( const char *controller, const char *path, @@ -814,16 +829,18 @@ int cg_pid_get_path(const char *controller, pid_t pid, char **path) { if (unified < 0) return unified; if (unified == 0) { - if (controller) { - if (!cg_controller_is_valid(controller)) - return -EINVAL; - } else - controller = ELOGIND_CGROUP_CONTROLLER; + if (controller) { + if (!cg_controller_is_valid(controller)) + return -EINVAL; + } else + controller = SYSTEMD_CGROUP_CONTROLLER; cs = strlen(controller); } fs = procfs_file_alloca(pid, "cgroup"); + log_debug_elogind("Searching for PID %u in \"%s\" (controller \"%s\")", + pid, fs, controller); f = fopen(fs, "re"); if (!f) return errno == ENOENT ? -ESRCH : -errno; @@ -843,31 +860,32 @@ int cg_pid_get_path(const char *controller, pid_t pid, char **path) { continue; } else { char *l; - size_t k; - const char *word, *state; - bool found = false; + size_t k; + const char *word, *state; + bool found = false; - l = strchr(line, ':'); - if (!l) - continue; + l = strchr(line, ':'); + if (!l) + continue; - l++; - e = strchr(l, ':'); - if (!e) - continue; + l++; + e = strchr(l, ':'); + if (!e) + continue; - *e = 0; - FOREACH_WORD_SEPARATOR(word, k, l, ",", state) { - if (k == cs && memcmp(word, controller, cs) == 0) { - found = true; - break; + *e = 0; + FOREACH_WORD_SEPARATOR(word, k, l, ",", state) { + if (k == cs && memcmp(word, controller, cs) == 0) { + found = true; + break; + } } - } - if (!found) - continue; + if (!found) + continue; } + log_debug_elogind("Found %s:%s", line, e+1); p = strdup(e + 1); if (!p) return -ENOMEM; @@ -879,8 +897,6 @@ int cg_pid_get_path(const char *controller, pid_t pid, char **path) { return -ENODATA; } -/// UNNEEDED by elogind -#if 0 int cg_install_release_agent(const char *controller, const char *agent) { _cleanup_free_ char *fs = NULL, *contents = NULL; const char *sc; @@ -904,7 +920,7 @@ int cg_install_release_agent(const char *controller, const char *agent) { sc = strstrip(contents); if (isempty(sc)) { - r = write_string_file_no_create(fs, agent); + r = write_string_file(fs, agent, 0); if (r < 0) return r; } else if (!path_equal(sc, agent)) @@ -922,7 +938,7 @@ int cg_install_release_agent(const char *controller, const char *agent) { sc = strstrip(contents); if (streq(sc, "0")) { - r = write_string_file_no_create(fs, "1"); + r = write_string_file(fs, "1", 0); if (r < 0) return r; @@ -949,7 +965,7 @@ int cg_uninstall_release_agent(const char *controller) { if (r < 0) return r; - r = write_string_file_no_create(fs, "0"); + r = write_string_file(fs, "0", 0); if (r < 0) return r; @@ -959,13 +975,12 @@ int cg_uninstall_release_agent(const char *controller) { if (r < 0) return r; - r = write_string_file_no_create(fs, ""); + r = write_string_file(fs, "", 0); if (r < 0) return r; return 0; } -#endif // 0 int cg_is_empty(const char *controller, const char *path) { _cleanup_fclose_ FILE *f = NULL; @@ -1007,8 +1022,8 @@ int cg_is_empty_recursive(const char *controller, const char *path) { * via the "cgroup.populated" attribute. */ r = cg_get_path(controller, path, "cgroup.populated", &populated); - if (r < 0) - return r; + if (r < 0) + return r; r = read_one_line_file(populated, &t); if (r == -ENOENT) @@ -1018,33 +1033,33 @@ int cg_is_empty_recursive(const char *controller, const char *path) { return streq(t, "0"); } else { - _cleanup_closedir_ DIR *d = NULL; - char *fn; + _cleanup_closedir_ DIR *d = NULL; + char *fn; r = cg_is_empty(controller, path); - if (r <= 0) - return r; + if (r <= 0) + return r; - r = cg_enumerate_subgroups(controller, path, &d); - if (r == -ENOENT) - return 1; - if (r < 0) + r = cg_enumerate_subgroups(controller, path, &d); + if (r == -ENOENT) + return 1; + if (r < 0) return r; - while ((r = cg_read_subgroup(d, &fn)) > 0) { - _cleanup_free_ char *p = NULL; + while ((r = cg_read_subgroup(d, &fn)) > 0) { + _cleanup_free_ char *p = NULL; - p = strjoin(path, "/", fn, NULL); - free(fn); - if (!p) - return -ENOMEM; + p = strjoin(path, "/", fn, NULL); + free(fn); + if (!p) + return -ENOMEM; r = cg_is_empty_recursive(controller, p); - if (r <= 0) + if (r <= 0) + return r; + } + if (r < 0) return r; - } - if (r < 0) - return r; return true; } @@ -1157,16 +1172,17 @@ int cg_mangle_path(const char *path, char **result) { if (r < 0) return r; - return cg_get_path(c ? c : ELOGIND_CGROUP_CONTROLLER, p ? p : "/", NULL, result); + return cg_get_path(c ?: SYSTEMD_CGROUP_CONTROLLER, p ?: "/", NULL, result); } int cg_get_root_path(char **path) { +#if 0 /// elogind does not support systemd scopes and slices char *p, *e; int r; assert(path); - r = cg_pid_get_path(ELOGIND_CGROUP_CONTROLLER, 1, &p); + r = cg_pid_get_path(SYSTEMD_CGROUP_CONTROLLER, 1, &p); if (r < 0) return r; @@ -1180,6 +1196,10 @@ int cg_get_root_path(char **path) { *path = p; return 0; +#else + assert(path); + return cg_pid_get_path(SYSTEMD_CGROUP_CONTROLLER, 1, path); +#endif // 0 } int cg_shift_path(const char *cgroup, const char *root, const char **shifted) { @@ -1199,10 +1219,11 @@ int cg_shift_path(const char *cgroup, const char *root, const char **shifted) { return r; root = rt; + log_debug_elogind("Determined root path: \"%s\"", root); } p = path_startswith(cgroup, root); - if (p && p > cgroup) + if (p && p[0] && (p > cgroup)) *shifted = p - 1; else *shifted = cgroup; @@ -1218,10 +1239,12 @@ int cg_pid_get_path_shifted(pid_t pid, const char *root, char **cgroup) { assert(pid >= 0); assert(cgroup); - r = cg_pid_get_path(ELOGIND_CGROUP_CONTROLLER, pid, &raw); + r = cg_pid_get_path(SYSTEMD_CGROUP_CONTROLLER, pid, &raw); if (r < 0) return r; + log_debug_elogind("Shifting path: \"%s\" (PID %u, root: \"%s\")", + raw, pid, root ? root : "NULL"); r = cg_shift_path(raw, root, &c); if (r < 0) return r; @@ -1238,10 +1261,12 @@ int cg_pid_get_path_shifted(pid_t pid, const char *root, char **cgroup) { *cgroup = n; } + log_debug_elogind("Resulting cgroup:\"%s\"", *cgroup); return 0; } +#if 0 /// UNNEEDED by elogind int cg_path_decode_unit(const char *cgroup, char **unit){ char *c, *s; size_t n; @@ -1490,8 +1515,12 @@ int cg_pid_get_machine_name(pid_t pid, char **machine) { return cg_path_get_machine_name(cgroup, machine); } +#endif // 0 int cg_path_get_session(const char *path, char **session) { + /* Elogind uses a flat hierarchy, just "/SESSION". The only + wrinkle is that SESSION might be escaped. */ +#if 0 _cleanup_free_ char *unit = NULL; char *start, *end; int r; @@ -1512,10 +1541,29 @@ int cg_path_get_session(const char *path, char **session) { *end = 0; if (!session_id_valid(start)) return -ENXIO; +#else + const char *e, *n, *start; + + assert(path); + log_debug_elogind("path is \"%s\"", path); + assert(path[0] == '/'); + + e = path + 1; + n = strchrnul(e, '/'); + if (e == n) + return -ENOENT; + + start = strndupa(e, n - e); + start = cg_unescape(start); + + if (!start[0]) + return -ENOENT; +#endif // 0 if (session) { char *rr; + log_debug_elogind("found session: \"%s\"", start); rr = strdup(start); if (!rr) return -ENOMEM; @@ -1537,6 +1585,7 @@ int cg_pid_get_session(pid_t pid, char **session) { return cg_path_get_session(cgroup, session); } +#if 0 /// UNNEEDED by elogind int cg_path_get_owner_uid(const char *path, uid_t *uid) { _cleanup_free_ char *slice = NULL; char *start, *end; @@ -1648,6 +1697,7 @@ int cg_pid_get_user_slice(pid_t pid, char **slice) { return cg_path_get_user_slice(cgroup, slice); } +#endif // 0 char *cg_escape(const char *p) { bool need_prefix = false; @@ -1689,7 +1739,7 @@ char *cg_escape(const char *p) { if (memcmp(p, n, l) != 0) continue; - need_prefix = true; + need_prefix = true; break; } } @@ -1698,7 +1748,7 @@ char *cg_escape(const char *p) { if (need_prefix) return strappend("_", p); - return strdup(p); + return strdup(p); } char *cg_unescape(const char *p) { @@ -1740,8 +1790,7 @@ bool cg_controller_is_valid(const char *p) { return true; } -/// UNNEEDED by elogind -#if 0 +#if 0 /// UNNEEDED by elogind int cg_slice_to_path(const char *unit, char **ret) { _cleanup_free_ char *p = NULL, *s = NULL, *e = NULL; const char *dash; @@ -1810,6 +1859,7 @@ int cg_slice_to_path(const char *unit, char **ret) { return 0; } +#endif // 0 int cg_set_attribute(const char *controller, const char *path, const char *attribute, const char *value) { _cleanup_free_ char *p = NULL; @@ -1819,9 +1869,10 @@ int cg_set_attribute(const char *controller, const char *path, const char *attri if (r < 0) return r; - return write_string_file_no_create(p, value); + return write_string_file(p, value, 0); } +#if 0 /// UNNEEDED by elogind int cg_get_attribute(const char *controller, const char *path, const char *attribute, char **ret) { _cleanup_free_ char *p = NULL; int r; @@ -1832,10 +1883,7 @@ int cg_get_attribute(const char *controller, const char *path, const char *attri return read_one_line_file(p, ret); } -#endif // 0 -/// UNNEEDED by elogind -#if 0 int cg_create_everywhere(CGroupMask supported, CGroupMask mask, const char *path) { CGroupController c; int r, unified; @@ -1845,7 +1893,7 @@ int cg_create_everywhere(CGroupMask supported, CGroupMask mask, const char *path * in all others */ /* First create the cgroup in our own hierarchy. */ - r = cg_create(ELOGIND_CGROUP_CONTROLLER, path); + r = cg_create(SYSTEMD_CGROUP_CONTROLLER, path); if (r < 0) return r; @@ -1871,13 +1919,12 @@ int cg_create_everywhere(CGroupMask supported, CGroupMask mask, const char *path return 0; } -#endif // 0 int cg_attach_everywhere(CGroupMask supported, const char *path, pid_t pid, cg_migrate_callback_t path_callback, void *userdata) { CGroupController c; int r, unified; - r = cg_attach(ELOGIND_CGROUP_CONTROLLER, path, pid); + r = cg_attach(SYSTEMD_CGROUP_CONTROLLER, path, pid); if (r < 0) return r; @@ -1906,8 +1953,6 @@ int cg_attach_everywhere(CGroupMask supported, const char *path, pid_t pid, cg_m return 0; } -/// UNNEEDED by elogind -#if 0 int cg_attach_many_everywhere(CGroupMask supported, const char *path, Set* pids, cg_migrate_callback_t path_callback, void *userdata) { Iterator i; void *pidp; @@ -1930,7 +1975,7 @@ int cg_migrate_everywhere(CGroupMask supported, const char *from, const char *to int r = 0, unified; if (!path_equal(from, to)) { - r = cg_migrate_recursive(ELOGIND_CGROUP_CONTROLLER, from, ELOGIND_CGROUP_CONTROLLER, to, false, true); + r = cg_migrate_recursive(SYSTEMD_CGROUP_CONTROLLER, from, SYSTEMD_CGROUP_CONTROLLER, to, false, true); if (r < 0) return r; } @@ -1954,7 +1999,7 @@ int cg_migrate_everywhere(CGroupMask supported, const char *from, const char *to if (!p) p = to; - (void) cg_migrate_recursive_fallback(ELOGIND_CGROUP_CONTROLLER, to, cgroup_controller_to_string(c), p, false, false); + (void) cg_migrate_recursive_fallback(SYSTEMD_CGROUP_CONTROLLER, to, cgroup_controller_to_string(c), p, false, false); } return 0; @@ -1964,7 +2009,7 @@ int cg_trim_everywhere(CGroupMask supported, const char *path, bool delete_root) CGroupController c; int r, unified; - r = cg_trim(ELOGIND_CGROUP_CONTROLLER, path, delete_root); + r = cg_trim(SYSTEMD_CGROUP_CONTROLLER, path, delete_root); if (r < 0) return r; @@ -1985,6 +2030,7 @@ int cg_trim_everywhere(CGroupMask supported, const char *path, bool delete_root) return 0; } +#endif // 0 int cg_mask_supported(CGroupMask *ret) { CGroupMask mask = 0; @@ -2009,7 +2055,7 @@ int cg_mask_supported(CGroupMask *ret) { if (r < 0) return r; - r = cg_get_path(ELOGIND_CGROUP_CONTROLLER, root, "cgroup.controllers", &path); + r = cg_get_path(SYSTEMD_CGROUP_CONTROLLER, root, "cgroup.controllers", &path); if (r < 0) return r; @@ -2035,9 +2081,10 @@ int cg_mask_supported(CGroupMask *ret) { mask |= CGROUP_CONTROLLER_TO_MASK(v); } - /* Currently, we only support the memory controller in - * the unified hierarchy, mask everything else off. */ - mask &= CGROUP_MASK_MEMORY; + /* Currently, we only support the memory and pids + * controller in the unified hierarchy, mask + * everything else off. */ + mask &= CGROUP_MASK_MEMORY | CGROUP_MASK_PIDS; } else { CGroupController c; @@ -2058,6 +2105,7 @@ int cg_mask_supported(CGroupMask *ret) { return 0; } +#if 0 /// UNNEEDED by elogind int cg_kernel_controllers(Set *controllers) { _cleanup_fclose_ FILE *f = NULL; char buf[LINE_MAX]; @@ -2090,7 +2138,7 @@ int cg_kernel_controllers(Set *controllers) { if (feof(f)) break; - if (ferror(f) && errno != 0) + if (ferror(f) && errno > 0) return -errno; return -EBADMSG; @@ -2131,9 +2179,18 @@ int cg_unified(void) { if (statfs("/sys/fs/cgroup/", &fs) < 0) return -errno; +/// elogind can not support the unified hierarchy as a controller, +/// so always assume a classical hierarchy. +/// If, ond only *if*, someone really wants to substitute systemd-login +/// in an environment managed by systemd with elogin, we might have to +/// add such a support. +#if 0 if (F_TYPE_EQUAL(fs.f_type, CGROUP_SUPER_MAGIC)) unified_cache = true; else if (F_TYPE_EQUAL(fs.f_type, TMPFS_MAGIC)) +#else + if (F_TYPE_EQUAL(fs.f_type, TMPFS_MAGIC)) +#endif // 0 unified_cache = false; else return -ENOEXEC; @@ -2141,8 +2198,7 @@ int cg_unified(void) { return unified_cache; } -/// UNNEEDED by elogind -#if 0 +#if 0 /// UNNEEDED by elogind void cg_unified_flush(void) { unified_cache = -1; } @@ -2163,7 +2219,7 @@ int cg_enable_everywhere(CGroupMask supported, CGroupMask mask, const char *p) { if (!unified) /* on the legacy hiearchy there's no joining of controllers defined */ return 0; - r = cg_get_path(ELOGIND_CGROUP_CONTROLLER, p, "cgroup.subtree_control", &fs); + r = cg_get_path(SYSTEMD_CGROUP_CONTROLLER, p, "cgroup.subtree_control", &fs); if (r < 0) return r; @@ -2189,7 +2245,6 @@ int cg_enable_everywhere(CGroupMask supported, CGroupMask mask, const char *p) { return 0; } -#endif // 0 bool cg_is_unified_wanted(void) { static thread_local int wanted = -1; @@ -2223,11 +2278,55 @@ bool cg_is_unified_wanted(void) { } } -/// UNNEEDED by elogind -#if 0 bool cg_is_legacy_wanted(void) { return !cg_is_unified_wanted(); } +#else +bool cg_is_legacy_wanted(void) { + return true; +} +#endif // 0 + +#if 0 /// UNNEEDED by elogind +int cg_cpu_shares_parse(const char *s, uint64_t *ret) { + uint64_t u; + int r; + + if (isempty(s)) { + *ret = CGROUP_CPU_SHARES_INVALID; + return 0; + } + + r = safe_atou64(s, &u); + if (r < 0) + return r; + + if (u < CGROUP_CPU_SHARES_MIN || u > CGROUP_CPU_SHARES_MAX) + return -ERANGE; + + *ret = u; + return 0; +} + +int cg_blkio_weight_parse(const char *s, uint64_t *ret) { + uint64_t u; + int r; + + if (isempty(s)) { + *ret = CGROUP_BLKIO_WEIGHT_INVALID; + return 0; + } + + r = safe_atou64(s, &u); + if (r < 0) + return r; + + if (u < CGROUP_BLKIO_WEIGHT_MIN || u > CGROUP_BLKIO_WEIGHT_MAX) + return -ERANGE; + + *ret = u; + return 0; +} #endif // 0 static const char *cgroup_controller_table[_CGROUP_CONTROLLER_MAX] = { @@ -2235,7 +2334,8 @@ static const char *cgroup_controller_table[_CGROUP_CONTROLLER_MAX] = { [CGROUP_CONTROLLER_CPUACCT] = "cpuacct", [CGROUP_CONTROLLER_BLKIO] = "blkio", [CGROUP_CONTROLLER_MEMORY] = "memory", - [CGROUP_CONTROLLER_DEVICE] = "devices", + [CGROUP_CONTROLLER_DEVICES] = "devices", + [CGROUP_CONTROLLER_PIDS] = "pids", }; DEFINE_STRING_TABLE_LOOKUP(cgroup_controller, CGroupController);