1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2012 Lennart Poettering
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.
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.
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/>.
30 #include "alloc-util.h"
31 #include "bus-error.h"
34 #include "formats-util.h"
35 #include "process-util.h"
36 #include "signal-util.h"
38 #include "user-util.h"
41 /// Additional includes needed by elogind
42 #include "musl_missing.h"
44 static const char* arg_what = "idle:sleep:shutdown";
45 static const char* arg_who = NULL;
46 static const char* arg_why = "Unknown reason";
47 static const char* arg_mode = NULL;
52 } arg_action = ACTION_INHIBIT;
54 static int inhibit(sd_bus *bus, sd_bus_error *error) {
55 _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
59 r = sd_bus_call_method(
61 "org.freedesktop.login1",
62 "/org/freedesktop/login1",
63 "org.freedesktop.login1.Manager",
67 "ssss", arg_what, arg_who, arg_why, arg_mode);
71 r = sd_bus_message_read_basic(reply, SD_BUS_TYPE_UNIX_FD, &fd);
75 r = fcntl(fd, F_DUPFD_CLOEXEC, 3);
82 static int print_inhibitors(sd_bus *bus, sd_bus_error *error) {
83 _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
84 const char *what, *who, *why, *mode;
85 unsigned int uid, pid;
89 r = sd_bus_call_method(
91 "org.freedesktop.login1",
92 "/org/freedesktop/login1",
93 "org.freedesktop.login1.Manager",
101 r = sd_bus_message_enter_container(reply, SD_BUS_TYPE_ARRAY, "(ssssuu)");
103 return bus_log_parse_error(r);
105 while ((r = sd_bus_message_read(reply, "(ssssuu)", &what, &who, &why, &mode, &uid, &pid)) > 0) {
106 _cleanup_free_ char *comm = NULL, *u = NULL;
108 if (arg_mode && !streq(mode, arg_mode))
111 get_process_comm(pid, &comm);
112 u = uid_to_name(uid);
114 printf(" Who: %s (UID "UID_FMT"/%s, PID "PID_FMT"/%s)\n"
118 who, uid, strna(u), pid, strna(comm),
126 return bus_log_parse_error(r);
128 r = sd_bus_message_exit_container(reply);
130 return bus_log_parse_error(r);
132 printf("%u inhibitors listed.\n", n);
136 static void help(void) {
137 printf("%s [OPTIONS...] {COMMAND} ...\n\n"
138 "Execute a process while inhibiting shutdown/sleep/idle.\n\n"
139 " -h --help Show this help\n"
140 " --version Show package version\n"
141 " --what=WHAT Operations to inhibit, colon separated list of:\n"
142 " shutdown, sleep, idle, handle-power-key,\n"
143 " handle-suspend-key, handle-hibernate-key,\n"
144 " handle-lid-switch\n"
145 " --who=STRING A descriptive string who is inhibiting\n"
146 " --why=STRING A descriptive string why is being inhibited\n"
147 " --mode=MODE One of block or delay\n"
148 " --list List active inhibitors\n"
149 , program_invocation_short_name);
152 static int parse_argv(int argc, char *argv[]) {
163 static const struct option options[] = {
164 { "help", no_argument, NULL, 'h' },
165 { "version", no_argument, NULL, ARG_VERSION },
166 { "what", required_argument, NULL, ARG_WHAT },
167 { "who", required_argument, NULL, ARG_WHO },
168 { "why", required_argument, NULL, ARG_WHY },
169 { "mode", required_argument, NULL, ARG_MODE },
170 { "list", no_argument, NULL, ARG_LIST },
179 while ((c = getopt_long(argc, argv, "+h", options, NULL)) >= 0)
207 arg_action = ACTION_LIST;
214 assert_not_reached("Unhandled option");
217 if (arg_action == ACTION_INHIBIT && optind == argc)
218 arg_action = ACTION_LIST;
220 else if (arg_action == ACTION_INHIBIT && optind >= argc) {
221 log_error("Missing command line to execute.");
228 int main(int argc, char *argv[]) {
229 _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
230 _cleanup_bus_flush_close_unref_ sd_bus *bus = NULL;
233 elogind_set_program_name(argv[0]);
234 log_parse_environment();
237 r = parse_argv(argc, argv);
243 r = sd_bus_default_system(&bus);
245 log_error_errno(r, "Failed to connect to bus: %m");
249 if (arg_action == ACTION_LIST) {
251 r = print_inhibitors(bus, &error);
253 log_error("Failed to list inhibitors: %s", bus_error_message(&error, -r));
258 _cleanup_close_ int fd = -1;
259 _cleanup_free_ char *w = NULL;
263 arg_who = w = strv_join(argv + optind, " ");
268 fd = inhibit(bus, &error);
270 log_error("Failed to inhibit: %s", bus_error_message(&error, fd));
276 log_error_errno(errno, "Failed to fork: %m");
283 (void) reset_all_signal_handlers();
284 (void) reset_signal_mask();
286 close_all_fds(NULL, 0);
288 execvp(argv[optind], argv + optind);
289 log_error_errno(errno, "Failed to execute %s: %m", argv[optind]);
293 r = wait_for_terminate_and_warn(argv[optind], pid, true);
294 return r < 0 ? EXIT_FAILURE : r;