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         umask(0022);
  43
  44         if (streq(argv[1], "start")) {
  45                 int q = 0, r = 0;
  46
  47                 if (unlink("/run/nologin") < 0 && errno != ENOENT) {
  48                         log_error("Failed to remove /run/nologin file: %m");
  49                         r = -errno;
  50                 }
  51
  52                 if (unlink("/etc/nologin") < 0 && errno != ENOENT) {
  53
  54                         /* If the file doesn't exist and /etc simply
  55                          * was read-only (in which case unlink()
  56                          * returns EROFS even if the file doesn't
  57                          * exist), don't complain */
  58
  59                         if (errno != EROFS || access("/etc/nologin", F_OK) >= 0) {
  60                                 log_error("Failed to remove /etc/nologin file: %m");
  61                                 q = -errno;
  62                         }
  63                 }
  64
  65                 if (r < 0 || q < 0)
  66                         goto finish;
  67
  68         } else if (streq(argv[1], "stop")) {
  69                 int r, q;
  70                 char *cgroup_user_tree = NULL;
  71
  72                 if ((r = write_one_line_file_atomic("/run/nologin", "System is going down.")) < 0)
  73                         log_error("Failed to create /run/nologin: %s", strerror(-r));
  74
  75                 if ((q = cg_get_user_path(&cgroup_user_tree)) < 0) {
  76                         log_error("Failed to determine use path: %s", strerror(-q));
  77                         goto finish;
  78                 }
  79
  80                 q = cg_kill_recursive_and_wait(SYSTEMD_CGROUP_CONTROLLER, cgroup_user_tree, true);
  81                 free(cgroup_user_tree);
  82
  83                 if (q < 0) {
  84                         log_error("Failed to kill sessions: %s", strerror(-q));
  85                         goto finish;
  86                 }
  87
  88                 if (r < 0)
  89                         goto finish;
  90
  91         } else {
  92                 log_error("Unknown verb %s.", argv[1]);
  93                 goto finish;
  94         }
  95
  96         ret = EXIT_SUCCESS;
  97
  98 finish:
  99         return ret;
 100 }