src/user-sessions.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU General Public License as published by
  10   the Free Software Foundation; either version 2 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <string.h>
  23 #include <unistd.h>
  24 #include <errno.h>
  25
  26 #include "log.h"
  27 #include "util.h"
  28 #include "cgroup-util.h"
  29
  30 int main(int argc, char*argv[]) {
  31         int ret = EXIT_FAILURE;
  32
  33         if (argc != 2) {
  34                 log_error("This program requires one argument.");
  35                 return EXIT_FAILURE;
  36         }
  37
  38         log_set_target(LOG_TARGET_SYSLOG_OR_KMSG);
  39         log_parse_environment();
  40         log_open();
  41
  42         if (streq(argv[1], "start")) {
  43                 int q = 0, r = 0;
  44
  45                 if (unlink("/run/nologin") < 0 && errno != ENOENT) {
  46                         log_error("Failed to remove /run/nologin file: %m");
  47                         r = -errno;
  48                 }
  49
  50                 if (unlink("/etc/nologin") < 0 && errno != ENOENT) {
  51
  52                         /* If the file doesn't exist and /etc simply
  53                          * was read-only (in which case unlink()
  54                          * returns EROFS even if the file doesn't
  55                          * exist), don't complain */
  56
  57                         if (errno != EROFS || access("/etc/nologin", F_OK) >= 0) {
  58                                 log_error("Failed to remove /etc/nologin file: %m");
  59                                 q = -errno;
  60                         }
  61                 }
  62
  63                 if (r < 0 || q < 0)
  64                         goto finish;
  65
  66         } else if (streq(argv[1], "stop")) {
  67                 int r, q;
  68                 char *cgroup_user_tree = NULL;
  69
  70                 if ((r = write_one_line_file_atomic("/run/nologin", "System is going down.")) < 0)
  71                         log_error("Failed to create /run/nologin: %s", strerror(-r));
  72
  73                 if ((q = cg_get_user_path(&cgroup_user_tree)) < 0) {
  74                         log_error("Failed to determine use path: %s", strerror(-q));
  75                         goto finish;
  76                 }
  77
  78                 q = cg_kill_recursive_and_wait(SYSTEMD_CGROUP_CONTROLLER, cgroup_user_tree, true);
  79                 free(cgroup_user_tree);
  80
  81                 if (q < 0) {
  82                         log_error("Failed to kill sessions: %s", strerror(-q));
  83                         goto finish;
  84                 }
  85
  86                 if (r < 0)
  87                         goto finish;
  88
  89         } else {
  90                 log_error("Unknown verb %s.", argv[1]);
  91                 goto finish;
  92         }
  93
  94         ret = EXIT_SUCCESS;
  95
  96 finish:
  97         return ret;
  98 }