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 _cleanup_dbus_message_unref_ 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,
72 static int print_inhibitors(DBusConnection *bus, DBusError *error) {
73 _cleanup_dbus_message_unref_ DBusMessage *reply = NULL;
75 DBusMessageIter iter, sub, sub2;
78 r = bus_method_call_with_reply(
80 "org.freedesktop.login1",
81 "/org/freedesktop/login1",
82 "org.freedesktop.login1.Manager",
90 if (!dbus_message_iter_init(reply, &iter))
93 if (dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_ARRAY)
96 dbus_message_iter_recurse(&iter, &sub);
97 while (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_INVALID) {
98 const char *what, *who, *why, *mode;
99 _cleanup_free_ char *comm = NULL, *u = NULL;
100 dbus_uint32_t uid, pid;
102 if (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_STRUCT)
105 dbus_message_iter_recurse(&sub, &sub2);
107 if (bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &what, true) < 0 ||
108 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &who, true) < 0 ||
109 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &why, true) < 0 ||
110 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &mode, true) < 0 ||
111 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_UINT32, &uid, true) < 0 ||
112 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_UINT32, &pid, false) < 0)
115 get_process_comm(pid, &comm);
116 u = uid_to_name(uid);
118 printf(" Who: %s (UID %lu/%s, PID %lu/%s)\n"
122 who, (unsigned long) uid, strna(u), (unsigned long) pid, strna(comm),
127 dbus_message_iter_next(&sub);
132 printf("%u inhibitors listed.\n", n);
136 static int help(void) {
138 printf("%s [OPTIONS...] {COMMAND} ...\n\n"
139 "Execute a process while inhibiting shutdown/sleep/idle.\n\n"
140 " -h --help Show this help\n"
141 " --version Show package version\n"
142 " --what=WHAT Operations to inhibit, colon separated list of:\n"
143 " shutdown, sleep, idle, handle-power-key,\n"
144 " handle-suspend-key, handle-hibernate-key,\n"
145 " handle-lid-switch\n"
146 " --who=STRING A descriptive string who is inhibiting\n"
147 " --why=STRING A descriptive string why is being inhibited\n"
148 " --mode=MODE One of block or delay\n"
149 " --list List active inhibitors\n",
150 program_invocation_short_name);
155 static int parse_argv(int argc, char *argv[]) {
166 static const struct option options[] = {
167 { "help", no_argument, NULL, 'h' },
168 { "version", no_argument, NULL, ARG_VERSION },
169 { "what", required_argument, NULL, ARG_WHAT },
170 { "who", required_argument, NULL, ARG_WHO },
171 { "why", required_argument, NULL, ARG_WHY },
172 { "mode", required_argument, NULL, ARG_MODE },
173 { "list", no_argument, NULL, ARG_LIST },
182 while ((c = getopt_long(argc, argv, "+h", options, NULL)) >= 0) {
191 puts(PACKAGE_STRING);
192 puts(SYSTEMD_FEATURES);
212 arg_action = ACTION_LIST;
216 log_error("Unknown option code %c", c);
221 if (arg_action == ACTION_INHIBIT && argc == 1)
222 arg_action = ACTION_LIST;
224 else if (arg_action == ACTION_INHIBIT && optind >= argc) {
225 log_error("Missing command line to execute.");
232 int main(int argc, char *argv[]) {
233 int r, exit_code = 0;
234 DBusConnection *bus = NULL;
236 _cleanup_close_ int fd = -1;
238 dbus_error_init(&error);
240 log_parse_environment();
243 r = parse_argv(argc, argv);
247 bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &error);
249 log_error("Failed to connect to bus: %s", bus_error_message(&error));
254 if (arg_action == ACTION_LIST) {
256 r = print_inhibitors(bus, &error);
258 log_error("Failed to list inhibitors: %s", bus_error(&error, r));
267 arg_who = w = strv_join(argv + optind, " ");
269 fd = inhibit(bus, &error);
273 log_error("Failed to inhibit: %s", bus_error(&error, r));
280 log_error("Failed to fork: %m");
288 close_nointr_nofail(fd);
289 close_all_fds(NULL, 0);
291 execvp(argv[optind], argv + optind);
292 log_error("Failed to execute %s: %m", argv[optind]);
296 r = wait_for_terminate_and_warn(argv[optind], pid);
303 dbus_connection_close(bus);
304 dbus_connection_unref(bus);
307 dbus_error_free(&error);
309 return r < 0 ? EXIT_FAILURE : exit_code;