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) {
 230         _cleanup_strv_free_ char **modes = NULL, **states = NULL;
 231         char *arguments[] = {
 232                 NULL,
 233                 (char*) "pre",
 234                 (char*) arg_verb,
 235                 NULL
 236         };
 237         static const char* const dirs[] = { SYSTEM_SLEEP_PATH, NULL};
 238         int r;
 239         _cleanup_fclose_ FILE *f = NULL;
 240
 241         r = parse_sleep_config(arg_verb, &modes, &states);
 242         if (r < 0)
 243                 return r;
 244
 245         /* This file is opened first, so that if we hit an error,
 246          * we can abort before modifying any state. */
 247         f = fopen("/sys/power/state", "we");
 248         if (!f)
 249                 return log_error_errno(errno, "Failed to open /sys/power/state: %m");
 250
 251         /* Configure the hibernation mode */
 252         r = write_mode(modes);
 253         if (r < 0)
 254                 return r;
 255
 256         execute_directories(dirs, DEFAULT_TIMEOUT_USEC, arguments);
 257
 258         log_struct(LOG_INFO,
 259                    LOG_MESSAGE_ID(SD_MESSAGE_SLEEP_START),
 260                    LOG_MESSAGE("Suspending system..."),
 261                    "SLEEP=%s", arg_verb,
 262                    NULL);
 263
 264         r = write_state(&f, states);
 265         if (r < 0)
 266                 return r;
 267
 268         log_struct(LOG_INFO,
 269                    LOG_MESSAGE_ID(SD_MESSAGE_SLEEP_STOP),
 270                    LOG_MESSAGE("System resumed."),
 271                    "SLEEP=%s", arg_verb,
 272                    NULL);
 273
 274         arguments[1] = (char*) "post";
 275         execute_directories(dirs, DEFAULT_TIMEOUT_USEC, arguments);
 276
 277         return r;
 278 }
 279
 280 int shutdown_or_sleep(HandleAction action) {
 281         switch (action) {
 282         case HANDLE_POWEROFF:
 283                 return run_helper(HALT);
 284         case HANDLE_REBOOT:
 285                 return run_helper(REBOOT);
 286         case HANDLE_HALT:
 287                 return run_helper(HALT);
 288         case HANDLE_KEXEC:
 289                 return run_helper(KEXEC);
 290         case HANDLE_SUSPEND:
 291                 return do_sleep("suspend");
 292         case HANDLE_HIBERNATE:
 293                 return do_sleep("hibernate");
 294         case HANDLE_HYBRID_SLEEP:
 295                 return do_sleep("hybrid-sleep");
 296         default:
 297                 return -EINVAL;
 298         }
 299 }
 300
 301 static const char* const handle_action_table[_HANDLE_ACTION_MAX] = {
 302         [HANDLE_IGNORE] = "ignore",
 303         [HANDLE_POWEROFF] = "poweroff",
 304         [HANDLE_REBOOT] = "reboot",
 305         [HANDLE_HALT] = "halt",
 306         [HANDLE_KEXEC] = "kexec",
 307         [HANDLE_SUSPEND] = "suspend",
 308         [HANDLE_HIBERNATE] = "hibernate",
 309         [HANDLE_HYBRID_SLEEP] = "hybrid-sleep",
 310         [HANDLE_LOCK] = "lock"
 311 };
 312
 313 DEFINE_STRING_TABLE_LOOKUP(handle_action, HandleAction);
 314 DEFINE_CONFIG_PARSE_ENUM(config_parse_handle_action, handle_action, HandleAction, "Failed to parse handle action setting");