chiark / gitweb /
journal-file: protect against alloca(0)
[elogind.git] / 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
24 #include "sd-messages.h"
25 #include "conf-parser.h"
26 #include "special.h"
27 #include "sleep-config.h"
28 #include "bus-util.h"
29 #include "bus-error.h"
30 #include "logind-action.h"
31
32 int manager_handle_action(
33                 Manager *m,
34                 InhibitWhat inhibit_key,
35                 HandleAction handle,
36                 bool ignore_inhibited,
37                 bool is_edge) {
38
39         static const char * const message_table[_HANDLE_ACTION_MAX] = {
40                 [HANDLE_POWEROFF] = "Powering Off...",
41                 [HANDLE_REBOOT] = "Rebooting...",
42                 [HANDLE_HALT] = "Halting...",
43                 [HANDLE_KEXEC] = "Rebooting via kexec...",
44                 [HANDLE_SUSPEND] = "Suspending...",
45                 [HANDLE_HIBERNATE] = "Hibernating...",
46                 [HANDLE_HYBRID_SLEEP] = "Hibernating and suspending..."
47         };
48
49         static const char * const target_table[_HANDLE_ACTION_MAX] = {
50                 [HANDLE_POWEROFF] = SPECIAL_POWEROFF_TARGET,
51                 [HANDLE_REBOOT] = SPECIAL_REBOOT_TARGET,
52                 [HANDLE_HALT] = SPECIAL_HALT_TARGET,
53                 [HANDLE_KEXEC] = SPECIAL_KEXEC_TARGET,
54                 [HANDLE_SUSPEND] = SPECIAL_SUSPEND_TARGET,
55                 [HANDLE_HIBERNATE] = SPECIAL_HIBERNATE_TARGET,
56                 [HANDLE_HYBRID_SLEEP] = SPECIAL_HYBRID_SLEEP_TARGET
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 the key handling is inhibited, don't do anything */
74         if (inhibit_key > 0) {
75                 if (manager_is_inhibited(m, inhibit_key, INHIBIT_BLOCK, NULL, true, false, 0, NULL)) {
76                         log_debug("Refusing operation, %s is inhibited.", inhibit_what_to_string(inhibit_key));
77                         return 0;
78                 }
79         }
80
81         /* Locking is handled differently from the rest. */
82         if (handle == HANDLE_LOCK) {
83                 log_info("Locking sessions...");
84                 session_send_lock_all(m, true);
85                 return 1;
86         }
87
88         if (handle == HANDLE_SUSPEND)
89                 supported = can_sleep("suspend") > 0;
90         else if (handle == HANDLE_HIBERNATE)
91                 supported = can_sleep("hibernate") > 0;
92         else if (handle == HANDLE_HYBRID_SLEEP)
93                 supported = can_sleep("hybrid-sleep") > 0;
94         else if (handle == HANDLE_KEXEC)
95                 supported = access(KEXEC, X_OK) >= 0;
96         else
97                 supported = true;
98
99         if (!supported) {
100                 log_warning("Requested operation not supported, ignoring.");
101                 return -ENOTSUP;
102         }
103
104         if (m->action_what) {
105                 log_debug("Action already in progress, ignoring.");
106                 return -EALREADY;
107         }
108
109         inhibit_operation = handle == HANDLE_SUSPEND || handle == HANDLE_HIBERNATE || handle == HANDLE_HYBRID_SLEEP ? INHIBIT_SLEEP : INHIBIT_SHUTDOWN;
110
111         /* If the actual operation is inhibited, warn and fail */
112         if (!ignore_inhibited &&
113             manager_is_inhibited(m, inhibit_operation, INHIBIT_BLOCK, NULL, false, false, 0, &offending)) {
114                 _cleanup_free_ char *comm = NULL, *u = NULL;
115
116                 get_process_comm(offending->pid, &comm);
117                 u = uid_to_name(offending->uid);
118
119                 /* If this is just a recheck of the lid switch then don't warn about anything */
120                 if (!is_edge) {
121                         log_debug("Refusing operation, %s is inhibited by UID %lu/%s, PID %lu/%s.",
122                                   inhibit_what_to_string(inhibit_operation),
123                                   (unsigned long) offending->uid, strna(u),
124                                   (unsigned long) offending->pid, strna(comm));
125                         return 0;
126                 }
127
128                 log_error("Refusing operation, %s is inhibited by UID %lu/%s, PID %lu/%s.",
129                           inhibit_what_to_string(inhibit_operation),
130                           (unsigned long) offending->uid, strna(u),
131                           (unsigned long) offending->pid, strna(comm));
132
133                 warn_melody();
134                 return -EPERM;
135         }
136
137         log_info("%s", message_table[handle]);
138
139         r = bus_manager_shutdown_or_sleep_now_or_later(m, target_table[handle], inhibit_operation, &error);
140         if (r < 0) {
141                 log_error("Failed to execute operation: %s", bus_error_message(&error, r));
142                 return r;
143         }
144
145         return 1;
146 }
147
148 static const char* const handle_action_table[_HANDLE_ACTION_MAX] = {
149         [HANDLE_IGNORE] = "ignore",
150         [HANDLE_POWEROFF] = "poweroff",
151         [HANDLE_REBOOT] = "reboot",
152         [HANDLE_HALT] = "halt",
153         [HANDLE_KEXEC] = "kexec",
154         [HANDLE_SUSPEND] = "suspend",
155         [HANDLE_HIBERNATE] = "hibernate",
156         [HANDLE_HYBRID_SLEEP] = "hybrid-sleep",
157         [HANDLE_LOCK] = "lock"
158 };
159
160 DEFINE_STRING_TABLE_LOOKUP(handle_action, HandleAction);
161 DEFINE_CONFIG_PARSE_ENUM(config_parse_handle_action, handle_action, HandleAction, "Failed to parse handle action setting");