X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Fuser-sessions.c;h=4518d953ed7bebc80b3f110d2981ee3961391278;hp=dc4ee0fb4ac402a7c3332777b864f994812e4799;hb=4466ee6a7151b30b7738b09c4c21ffa3872a5047;hpb=e92787416c691c3f34f47349e5eae3fa68eae856 diff --git a/src/user-sessions.c b/src/user-sessions.c index dc4ee0fb4..4518d953e 100644 --- a/src/user-sessions.c +++ b/src/user-sessions.c @@ -40,26 +40,42 @@ int main(int argc, char*argv[]) { log_open(); if (streq(argv[1], "start")) { + int q = 0, r = 0; - if (unlink("/var/run/nologin") < 0 || - unlink("/etc/nologin") < 0) { + if (unlink("/run/nologin") < 0 && errno != ENOENT) { + log_error("Failed to remove /run/nologin file: %m"); + r = -errno; + } - if (errno != ENOENT) { - log_error("Failed to remove nologin files: %m"); - goto finish; - } + if (unlink("/etc/nologin") < 0 && errno != ENOENT) { + log_error("Failed to remove /etc/nologin file: %m"); + q = -errno; } + if (r < 0 || q < 0) + goto finish; + } else if (streq(argv[1], "stop")) { int r, q; + char *cgroup_user_tree = NULL; - if ((r = write_one_line_file("/var/run/nologin", "System is going down.")) < 0) - log_error("Failed to create /var/run/nologin: %s", strerror(-r)); + if ((r = write_one_line_file("/run/nologin", "System is going down.")) < 0) + log_error("Failed to create /run/nologin: %s", strerror(-r)); - if ((q = cg_kill_recursive_and_wait(SYSTEMD_CGROUP_CONTROLLER, "/user", true)) < 0) + if ((q = cg_get_user_path(&cgroup_user_tree)) < 0) { + log_error("Failed to determine use path: %s", strerror(-q)); + goto finish; + } + + q = cg_kill_recursive_and_wait(SYSTEMD_CGROUP_CONTROLLER, cgroup_user_tree, true); + free(cgroup_user_tree); + + if (q < 0) { log_error("Failed to kill sessions: %s", strerror(-q)); + goto finish; + } - if (r < 0 || q < 0) + if (r < 0) goto finish; } else {