src/login/logind-action.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2012 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 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   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <unistd.h>
  23 #include <stdio.h>
  24 #include <errno.h>
  25
  26 #include "sd-messages.h"
  27 #include "log.h"
  28 #include "util.h"
  29 #include "strv.h"
  30 #include "fileio.h"
  31 #include "build.h"
  32 #include "def.h"
  33 #include "conf-parser.h"
  34 #include "sleep-config.h"
  35 #include "bus-error.h"
  36 #include "bus-util.h"
  37 #include "logind-action.h"
  38 #include "formats-util.h"
  39 #include "process-util.h"
  40 #include "terminal-util.h"
  41
  42 int manager_handle_action(
  43                 Manager *m,
  44                 InhibitWhat inhibit_key,
  45                 HandleAction handle,
  46                 bool ignore_inhibited,
  47                 bool is_edge) {
  48
  49         static const char * const message_table[_HANDLE_ACTION_MAX] = {
  50                 [HANDLE_POWEROFF] = "Powering Off...",
  51                 [HANDLE_REBOOT] = "Rebooting...",
  52                 [HANDLE_HALT] = "Halting...",
  53                 [HANDLE_KEXEC] = "Rebooting via kexec...",
  54                 [HANDLE_SUSPEND] = "Suspending...",
  55                 [HANDLE_HIBERNATE] = "Hibernating...",
  56                 [HANDLE_HYBRID_SLEEP] = "Hibernating and suspending..."
  57         };
  58
  59         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
  60         InhibitWhat inhibit_operation;
  61         Inhibitor *offending = NULL;
  62         bool supported;
  63         int r;
  64
  65         assert(m);
  66
  67         /* If the key handling is turned off, don't do anything */
  68         if (handle == HANDLE_IGNORE) {
  69                 log_debug("Refusing operation, as it is turned off.");
  70                 return 0;
  71         }
  72
  73         if (inhibit_key == INHIBIT_HANDLE_LID_SWITCH) {
  74                 /* If the last system suspend or startup is too close,
  75                  * let's not suspend for now, to give USB docking
  76                  * stations some time to settle so that we can
  77                  * properly watch its displays. */
  78                 if (m->lid_switch_ignore_event_source) {
  79                         log_debug("Ignoring lid switch request, system startup or resume too close.");
  80                         return 0;
  81                 }
  82         }
  83
  84         /* If the key handling is inhibited, don't do anything */
  85         if (inhibit_key > 0) {
  86                 if (manager_is_inhibited(m, inhibit_key, INHIBIT_BLOCK, NULL, true, false, 0, NULL)) {
  87                         log_debug("Refusing operation, %s is inhibited.", inhibit_what_to_string(inhibit_key));
  88                         return 0;
  89                 }
  90         }
  91
  92         /* Locking is handled differently from the rest. */
  93         if (handle == HANDLE_LOCK) {
  94
  95                 if (!is_edge)
  96                         return 0;
  97
  98                 log_info("Locking sessions...");
  99                 session_send_lock_all(m, true);
 100                 return 1;
 101         }
 102
 103         if (handle == HANDLE_SUSPEND)
 104                 supported = can_sleep("suspend") > 0;
 105         else if (handle == HANDLE_HIBERNATE)
 106                 supported = can_sleep("hibernate") > 0;
 107         else if (handle == HANDLE_HYBRID_SLEEP)
 108                 supported = can_sleep("hybrid-sleep") > 0;
 109         else if (handle == HANDLE_KEXEC)
 110                 supported = access(KEXEC, X_OK) >= 0;
 111         else
 112                 supported = true;
 113
 114         if (!supported) {
 115                 log_warning("Requested operation not supported, ignoring.");
 116                 return -EOPNOTSUPP;
 117         }
 118
 119         if (m->action_what) {
 120                 log_debug("Action already in progress, ignoring.");
 121                 return -EALREADY;
 122         }
 123
 124         inhibit_operation = handle == HANDLE_SUSPEND || handle == HANDLE_HIBERNATE || handle == HANDLE_HYBRID_SLEEP ? INHIBIT_SLEEP : INHIBIT_SHUTDOWN;
 125
 126         /* If the actual operation is inhibited, warn and fail */
 127         if (!ignore_inhibited &&
 128             manager_is_inhibited(m, inhibit_operation, INHIBIT_BLOCK, NULL, false, false, 0, &offending)) {
 129                 _cleanup_free_ char *comm = NULL, *u = NULL;
 130
 131                 get_process_comm(offending->pid, &comm);
 132                 u = uid_to_name(offending->uid);
 133
 134                 /* If this is just a recheck of the lid switch then don't warn about anything */
 135                 if (!is_edge) {
 136                         log_debug("Refusing operation, %s is inhibited by UID "UID_FMT"/%s, PID "PID_FMT"/%s.",
 137                                   inhibit_what_to_string(inhibit_operation),
 138                                   offending->uid, strna(u),
 139                                   offending->pid, strna(comm));
 140                         return 0;
 141                 }
 142
 143                 log_error("Refusing operation, %s is inhibited by UID "UID_FMT"/%s, PID "PID_FMT"/%s.",
 144                           inhibit_what_to_string(inhibit_operation),
 145                           offending->uid, strna(u),
 146                           offending->pid, strna(comm));
 147
 148                 warn_melody();
 149                 return -EPERM;
 150         }
 151
 152         log_info("%s", message_table[handle]);
 153
 154         r = bus_manager_shutdown_or_sleep_now_or_later(m, handle, inhibit_operation, &error);
 155         if (r < 0) {
 156                 log_error("Failed to execute operation: %s", bus_error_message(&error, r));
 157                 return r;
 158         }
 159
 160         return 1;
 161 }
 162
 163 static int run_helper(const char *helper) {
 164         int pid = fork();
 165         if (pid < 0) {
 166                 return log_error_errno(errno, "Failed to fork: %m");
 167         }
 168
 169         if (pid == 0) {
 170                 /* Child */
 171
 172                 close_all_fds(NULL, 0);
 173
 174                 execlp(helper, helper, NULL);
 175                 log_error_errno(errno, "Failed to execute %s: %m", helper);
 176                 _exit(EXIT_FAILURE);
 177         }
 178
 179         return wait_for_terminate_and_warn(helper, pid, true);
 180 }
 181
 182 static int write_mode(char **modes) {
 183         int r = 0;
 184         char **mode;
 185
 186         STRV_FOREACH(mode, modes) {
 187                 int k;
 188
 189                 k = write_string_file("/sys/power/disk", *mode);
 190                 if (k == 0)
 191                         return 0;
 192
 193                 log_debug_errno(k, "Failed to write '%s' to /sys/power/disk: %m",
 194                                 *mode);
 195                 if (r == 0)
 196                         r = k;
 197         }
 198
 199         if (r < 0)
 200                 log_error_errno(r, "Failed to write mode to /sys/power/disk: %m");
 201
 202         return r;
 203 }
 204
 205 static int write_state(FILE **f, char **states) {
 206         char **state;
 207         int r = 0;
 208
 209         STRV_FOREACH(state, states) {
 210                 int k;
 211
 212                 k = write_string_stream(*f, *state);
 213                 if (k == 0)
 214                         return 0;
 215                 log_debug_errno(k, "Failed to write '%s' to /sys/power/state: %m",
 216                                 *state);
 217                 if (r == 0)
 218                         r = k;
 219
 220                 fclose(*f);
 221                 *f = fopen("/sys/power/state", "we");
 222                 if (!*f)
 223                         return log_error_errno(errno, "Failed to open /sys/power/state: %m");
 224         }
 225
 226         return r;
 227 }
 228
 229 static int do_sleep(const char *arg_verb, const char **modes, const char **states) {
 230         char *arguments[] = {
 231                 NULL,
 232                 (char*) "pre",
 233                 (char*) arg_verb,
 234                 NULL
 235         };
 236         static const char* const dirs[] = { SYSTEM_SLEEP_PATH, NULL};
 237         int r;
 238         _cleanup_fclose_ FILE *f = NULL;
 239
 240         /* This file is opened first, so that if we hit an error,
 241          * we can abort before modifying any state. */
 242         f = fopen("/sys/power/state", "we");
 243         if (!f)
 244                 return log_error_errno(errno, "Failed to open /sys/power/state: %m");
 245
 246         /* Configure the hibernation mode */
 247         r = write_mode(modes);
 248         if (r < 0)
 249                 return r;
 250
 251         execute_directories(dirs, DEFAULT_TIMEOUT_USEC, arguments);
 252
 253         log_struct(LOG_INFO,
 254                    LOG_MESSAGE_ID(SD_MESSAGE_SLEEP_START),
 255                    LOG_MESSAGE("Suspending system..."),
 256                    "SLEEP=%s", arg_verb,
 257                    NULL);
 258
 259         r = write_state(&f, states);
 260         if (r < 0)
 261                 return r;
 262
 263         log_struct(LOG_INFO,
 264                    LOG_MESSAGE_ID(SD_MESSAGE_SLEEP_STOP),
 265                    LOG_MESSAGE("System resumed."),
 266                    "SLEEP=%s", arg_verb,
 267                    NULL);
 268
 269         arguments[1] = (char*) "post";
 270         execute_directories(dirs, DEFAULT_TIMEOUT_USEC, arguments);
 271
 272         return r;
 273 }
 274
 275 int shutdown_or_sleep(Manager *m, HandleAction action) {
 276         switch (action) {
 277         case HANDLE_POWEROFF:
 278                 return run_helper(HALT);
 279         case HANDLE_REBOOT:
 280                 return run_helper(REBOOT);
 281         case HANDLE_HALT:
 282                 return run_helper(HALT);
 283         case HANDLE_KEXEC:
 284                 return run_helper(KEXEC);
 285         case HANDLE_SUSPEND:
 286                 return do_sleep("suspend", m->suspend_mode, m->suspend_state);
 287         case HANDLE_HIBERNATE:
 288                 return do_sleep("hibernate", m->hibernate_mode, m->hibernate_state);
 289         case HANDLE_HYBRID_SLEEP:
 290                 return do_sleep("hybrid-sleep", m->hybrid_sleep_mode, m->hybrid_sleep_state);
 291         default:
 292                 return -EINVAL;
 293         }
 294 }
 295
 296 static const char* const handle_action_table[_HANDLE_ACTION_MAX] = {
 297         [HANDLE_IGNORE] = "ignore",
 298         [HANDLE_POWEROFF] = "poweroff",
 299         [HANDLE_REBOOT] = "reboot",
 300         [HANDLE_HALT] = "halt",
 301         [HANDLE_KEXEC] = "kexec",
 302         [HANDLE_SUSPEND] = "suspend",
 303         [HANDLE_HIBERNATE] = "hibernate",
 304         [HANDLE_HYBRID_SLEEP] = "hybrid-sleep",
 305         [HANDLE_LOCK] = "lock"
 306 };
 307
 308 DEFINE_STRING_TABLE_LOOKUP(handle_action, HandleAction);
 309 DEFINE_CONFIG_PARSE_ENUM(config_parse_handle_action, handle_action, HandleAction, "Failed to parse handle action setting");