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