X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Flogin%2Fuser-sessions.c;h=6edb823e8c5775f8871c94335d8e4808ecc22a71;hb=3e0a2048978fcc7aff00797bdb876b2592c2fbc7;hp=98b36c60e8c5d6032c2430ca53803809ad4687f2;hpb=5430f7f2bc7330f3088b894166bf3524a067e3d8;p=elogind.git diff --git a/src/login/user-sessions.c b/src/login/user-sessions.c index 98b36c60e..6edb823e8 100644 --- a/src/login/user-sessions.c +++ b/src/login/user-sessions.c @@ -25,10 +25,9 @@ #include "log.h" #include "util.h" -#include "cgroup-util.h" +#include "fileio.h" int main(int argc, char*argv[]) { - int ret = EXIT_FAILURE; if (argc != 2) { log_error("This program requires one argument."); @@ -42,59 +41,41 @@ int main(int argc, char*argv[]) { umask(0022); if (streq(argv[1], "start")) { - int q = 0, r = 0; + int r = 0; if (unlink("/run/nologin") < 0 && errno != ENOENT) { - log_error("Failed to remove /run/nologin file: %m"); + log_error_errno(errno, "Failed to remove /run/nologin file: %m"); r = -errno; } if (unlink("/etc/nologin") < 0 && errno != ENOENT) { - /* If the file doesn't exist and /etc simply * was read-only (in which case unlink() * returns EROFS even if the file doesn't * exist), don't complain */ if (errno != EROFS || access("/etc/nologin", F_OK) >= 0) { - log_error("Failed to remove /etc/nologin file: %m"); - q = -errno; + log_error_errno(errno, "Failed to remove /etc/nologin file: %m"); + return EXIT_FAILURE; } } - if (r < 0 || q < 0) - goto finish; + if (r < 0) + return EXIT_FAILURE; } else if (streq(argv[1], "stop")) { - int r, q; - char *cgroup_user_tree = NULL; - - if ((r = write_one_line_file_atomic("/run/nologin", "System is going down.")) < 0) - log_error("Failed to create /run/nologin: %s", strerror(-r)); - - if ((q = cg_get_user_path(&cgroup_user_tree)) < 0) { - log_error("Failed to determine use path: %s", strerror(-q)); - goto finish; - } + int r; - 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; + r = write_string_file_atomic("/run/nologin", "System is going down."); + if (r < 0) { + log_error_errno(r, "Failed to create /run/nologin: %m"); + return EXIT_FAILURE; } - if (r < 0) - goto finish; - } else { log_error("Unknown verb %s.", argv[1]); - goto finish; + return EXIT_FAILURE; } - ret = EXIT_SUCCESS; - -finish: - return ret; + return EXIT_SUCCESS; }