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
  44                 if (unlink("/var/run/nologin") < 0 ||
  45                     unlink("/etc/nologin") < 0) {
  46
  47                         if (errno != ENOENT) {
  48                                 log_error("Failed to remove nologin files: %m");
  49                                 goto finish;
  50                         }
  51                 }
  52
  53         } else if (streq(argv[1], "stop")) {
  54                 int r, q;
  55
  56                 if ((r = write_one_line_file("/var/run/nologin", "System is going down.")) < 0)
  57                         log_error("Failed to create /var/run/nologin: %s", strerror(-r));
  58
  59                 if ((q = cg_kill_recursive_and_wait(SYSTEMD_CGROUP_CONTROLLER, "/user", true)) < 0)
  60                         log_error("Failed to kill sessions: %s", strerror(-q));
  61
  62                 if (r < 0 || q < 0)
  63                         goto finish;
  64
  65         } else {
  66                 log_error("Unknown verb %s.", argv[1]);
  67                 goto finish;
  68         }
  69
  70         ret = EXIT_SUCCESS;
  71
  72 finish:
  73         return ret;
  74 }