chiark / gitweb /
util: make malloc0 ask calloc for one block of size n
[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 "conf-parser.h"
25 #include "special.h"
26 #include "sleep-config.h"
27 #include "bus-util.h"
28 #include "bus-error.h"
29 #include "logind-action.h"
30 #include "formats-util.h"
31 #include "process-util.h"
32 #include "terminal-util.h"
33
34 int manager_handle_action(
35                 Manager *m,
36                 InhibitWhat inhibit_key,
37                 HandleAction handle,
38                 bool ignore_inhibited,
39                 bool is_edge) {
40
41         static const char * const message_table[_HANDLE_ACTION_MAX] = {
42                 [HANDLE_POWEROFF] = "Powering Off...",
43                 [HANDLE_REBOOT] = "Rebooting...",
44                 [HANDLE_HALT] = "Halting...",
45                 [HANDLE_KEXEC] = "Rebooting via kexec...",
46                 [HANDLE_SUSPEND] = "Suspending...",
47                 [HANDLE_HIBERNATE] = "Hibernating...",
48                 [HANDLE_HYBRID_SLEEP] = "Hibernating and suspending..."
49         };
50
51         static const char * const target_table[_HANDLE_ACTION_MAX] = {
52                 [HANDLE_POWEROFF] = SPECIAL_POWEROFF_TARGET,
53                 [HANDLE_REBOOT] = SPECIAL_REBOOT_TARGET,
54                 [HANDLE_HALT] = SPECIAL_HALT_TARGET,
55                 [HANDLE_KEXEC] = SPECIAL_KEXEC_TARGET,
56                 [HANDLE_SUSPEND] = SPECIAL_SUSPEND_TARGET,
57                 [HANDLE_HIBERNATE] = SPECIAL_HIBERNATE_TARGET,
58                 [HANDLE_HYBRID_SLEEP] = SPECIAL_HYBRID_SLEEP_TARGET
59         };
60
61         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
62         InhibitWhat inhibit_operation;
63         Inhibitor *offending = NULL;
64         bool supported;
65         int r;
66
67         assert(m);
68
69         /* If the key handling is turned off, don't do anything */
70         if (handle == HANDLE_IGNORE) {
71                 log_debug("Refusing operation, as it is turned off.");
72                 return 0;
73         }
74
75         if (inhibit_key == INHIBIT_HANDLE_LID_SWITCH) {
76                 /* If the last system suspend or startup is too close,
77                  * let's not suspend for now, to give USB docking
78                  * stations some time to settle so that we can
79                  * properly watch its displays. */
80                 if (m->lid_switch_ignore_event_source) {
81                         log_debug("Ignoring lid switch request, system startup or resume too close.");
82                         return 0;
83                 }
84         }
85
86         /* If the key handling is inhibited, don't do anything */
87         if (inhibit_key > 0) {
88                 if (manager_is_inhibited(m, inhibit_key, INHIBIT_BLOCK, NULL, true, false, 0, NULL)) {
89                         log_debug("Refusing operation, %s is inhibited.", inhibit_what_to_string(inhibit_key));
90                         return 0;
91                 }
92         }
93
94         /* Locking is handled differently from the rest. */
95         if (handle == HANDLE_LOCK) {
96
97                 if (!is_edge)
98                         return 0;
99
100                 log_info("Locking sessions...");
101                 session_send_lock_all(m, true);
102                 return 1;
103         }
104
105         if (handle == HANDLE_SUSPEND)
106                 supported = can_sleep("suspend") > 0;
107         else if (handle == HANDLE_HIBERNATE)
108                 supported = can_sleep("hibernate") > 0;
109         else if (handle == HANDLE_HYBRID_SLEEP)
110                 supported = can_sleep("hybrid-sleep") > 0;
111         else if (handle == HANDLE_KEXEC)
112                 supported = access(KEXEC, X_OK) >= 0;
113         else
114                 supported = true;
115
116         if (!supported) {
117                 log_warning("Requested operation not supported, ignoring.");
118                 return -EOPNOTSUPP;
119         }
120
121         if (m->action_what) {
122                 log_debug("Action already in progress, ignoring.");
123                 return -EALREADY;
124         }
125
126         inhibit_operation = handle == HANDLE_SUSPEND || handle == HANDLE_HIBERNATE || handle == HANDLE_HYBRID_SLEEP ? INHIBIT_SLEEP : INHIBIT_SHUTDOWN;
127
128         /* If the actual operation is inhibited, warn and fail */
129         if (!ignore_inhibited &&
130             manager_is_inhibited(m, inhibit_operation, INHIBIT_BLOCK, NULL, false, false, 0, &offending)) {
131                 _cleanup_free_ char *comm = NULL, *u = NULL;
132
133                 get_process_comm(offending->pid, &comm);
134                 u = uid_to_name(offending->uid);
135
136                 /* If this is just a recheck of the lid switch then don't warn about anything */
137                 if (!is_edge) {
138                         log_debug("Refusing operation, %s is inhibited by UID "UID_FMT"/%s, PID "PID_FMT"/%s.",
139                                   inhibit_what_to_string(inhibit_operation),
140                                   offending->uid, strna(u),
141                                   offending->pid, strna(comm));
142                         return 0;
143                 }
144
145                 log_error("Refusing operation, %s is inhibited by UID "UID_FMT"/%s, PID "PID_FMT"/%s.",
146                           inhibit_what_to_string(inhibit_operation),
147                           offending->uid, strna(u),
148                           offending->pid, strna(comm));
149
150                 warn_melody();
151                 return -EPERM;
152         }
153
154         log_info("%s", message_table[handle]);
155
156         r = bus_manager_shutdown_or_sleep_now_or_later(m, target_table[handle], inhibit_operation, &error);
157         if (r < 0) {
158                 log_error("Failed to execute operation: %s", bus_error_message(&error, r));
159                 return r;
160         }
161
162         return 1;
163 }
164
165 static const char* const handle_action_table[_HANDLE_ACTION_MAX] = {
166         [HANDLE_IGNORE] = "ignore",
167         [HANDLE_POWEROFF] = "poweroff",
168         [HANDLE_REBOOT] = "reboot",
169         [HANDLE_HALT] = "halt",
170         [HANDLE_KEXEC] = "kexec",
171         [HANDLE_SUSPEND] = "suspend",
172         [HANDLE_HIBERNATE] = "hibernate",
173         [HANDLE_HYBRID_SLEEP] = "hybrid-sleep",
174         [HANDLE_LOCK] = "lock"
175 };
176
177 DEFINE_STRING_TABLE_LOOKUP(handle_action, HandleAction);
178 DEFINE_CONFIG_PARSE_ENUM(config_parse_handle_action, handle_action, HandleAction, "Failed to parse handle action setting");