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/>.
29 #include "dbus-common.h"
34 static const char* arg_what = "idle:sleep:shutdown";
35 static const char* arg_who = NULL;
36 static const char* arg_why = "Unknown reason";
37 static const char* arg_mode = "block";
42 } arg_action = ACTION_INHIBIT;
44 static int inhibit(DBusConnection *bus, DBusError *error) {
45 DBusMessage *reply = NULL;
48 r = bus_method_call_with_reply(
50 "org.freedesktop.login1",
51 "/org/freedesktop/login1",
52 "org.freedesktop.login1.Manager",
56 DBUS_TYPE_STRING, &arg_what,
57 DBUS_TYPE_STRING, &arg_who,
58 DBUS_TYPE_STRING, &arg_why,
59 DBUS_TYPE_STRING, &arg_mode,
64 if (!dbus_message_get_args(reply, error,
65 DBUS_TYPE_UNIX_FD, &r,
69 dbus_message_unref(reply);
74 static int print_inhibitors(DBusConnection *bus, DBusError *error) {
75 DBusMessage *reply = NULL;
77 DBusMessageIter iter, sub, sub2;
80 r = bus_method_call_with_reply(
82 "org.freedesktop.login1",
83 "/org/freedesktop/login1",
84 "org.freedesktop.login1.Manager",
92 if (!dbus_message_iter_init(reply, &iter)) {
97 if (dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_ARRAY) {
102 dbus_message_iter_recurse(&iter, &sub);
103 while (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_INVALID) {
104 const char *what, *who, *why, *mode;
105 dbus_uint32_t uid, pid;
107 if (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_STRUCT) {
112 dbus_message_iter_recurse(&sub, &sub2);
114 if (bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &what, true) < 0 ||
115 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &who, true) < 0 ||
116 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &why, true) < 0 ||
117 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &mode, true) < 0 ||
118 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_UINT32, &uid, true) < 0 ||
119 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_UINT32, &pid, false) < 0) {
124 printf(" Who: %s (UID %lu, PID %lu)\n"
128 who, (unsigned long) uid, (unsigned long) pid,
133 dbus_message_iter_next(&sub);
138 printf("%u inhibitors listed.\n", n);
143 dbus_message_unref(reply);
148 static int help(void) {
150 printf("%s [OPTIONS...] {COMMAND} ...\n\n"
151 "Execute a process while inhibiting shutdown/sleep/idle.\n\n"
152 " -h --help Show this help\n"
153 " --version Show package version\n"
154 " --what=WHAT Operations to inhibit, colon separated list of:\n"
155 " shutdown, sleep, idle, handle-power-key,\n"
156 " handle-suspend-key, handle-hibernate-key,\n"
157 " handle-lid-switch\n"
158 " --who=STRING A descriptive string who is inhibiting\n"
159 " --why=STRING A descriptive string why is being inhibited\n"
160 " --mode=MODE One of block or delay\n"
161 " --list List active inhibitors\n",
162 program_invocation_short_name);
167 static int parse_argv(int argc, char *argv[]) {
178 static const struct option options[] = {
179 { "help", no_argument, NULL, 'h' },
180 { "version", no_argument, NULL, ARG_VERSION },
181 { "what", required_argument, NULL, ARG_WHAT },
182 { "who", required_argument, NULL, ARG_WHO },
183 { "why", required_argument, NULL, ARG_WHY },
184 { "mode", required_argument, NULL, ARG_MODE },
185 { "list", no_argument, NULL, ARG_LIST },
194 while ((c = getopt_long(argc, argv, "+h", options, NULL)) >= 0) {
203 puts(PACKAGE_STRING);
204 puts(SYSTEMD_FEATURES);
224 arg_action = ACTION_LIST;
228 log_error("Unknown option code %c", c);
233 if (arg_action == ACTION_INHIBIT && argc == 1)
234 arg_action = ACTION_LIST;
236 else if (arg_action == ACTION_INHIBIT && optind >= argc) {
237 log_error("Missing command line to execute.");
244 int main(int argc, char *argv[]) {
245 int r, exit_code = 0;
246 DBusConnection *bus = NULL;
250 dbus_error_init(&error);
252 log_parse_environment();
255 r = parse_argv(argc, argv);
259 bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &error);
261 log_error("Failed to connect to bus: %s", bus_error_message(&error));
266 if (arg_action == ACTION_LIST) {
268 r = print_inhibitors(bus, &error);
270 log_error("Failed to list inhibitors: %s", bus_error_message_or_strerror(&error, -r));
279 arg_who = w = strv_join(argv + optind, " ");
281 fd = inhibit(bus, &error);
285 log_error("Failed to inhibit: %s", bus_error_message_or_strerror(&error, -r));
292 log_error("Failed to fork: %m");
300 close_nointr_nofail(fd);
301 close_all_fds(NULL, 0);
303 execvp(argv[optind], argv + optind);
304 log_error("Failed to execute %s: %m", argv[optind]);
308 r = wait_for_terminate_and_warn(argv[optind], pid);
315 dbus_connection_close(bus);
316 dbus_connection_unref(bus);
319 dbus_error_free(&error);
322 close_nointr_nofail(fd);
324 return r < 0 ? EXIT_FAILURE : exit_code;