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 printf("%-21s %-20s %-20s %-5s %6s %6s\n",
110 dbus_message_iter_recurse(&iter, &sub);
111 while (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_INVALID) {
112 const char *what, *who, *why, *mode;
114 dbus_uint32_t uid, pid;
116 if (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_STRUCT) {
121 dbus_message_iter_recurse(&sub, &sub2);
123 if (bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &what, true) < 0 ||
124 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &who, true) < 0 ||
125 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &why, true) < 0 ||
126 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &mode, true) < 0 ||
127 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_UINT32, &uid, true) < 0 ||
128 bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_UINT32, &pid, false) < 0) {
133 ewho = ellipsize(who, 20, 66);
134 ewhy = ellipsize(why, 20, 66);
136 printf("%-21s %-20s %-20s %-5s %6lu %6lu\n",
137 what, ewho ? ewho : who, ewhy ? ewhy : why, mode, (unsigned long) uid, (unsigned long) pid);
142 dbus_message_iter_next(&sub);
147 printf("\n%u inhibitors listed.\n", n);
152 dbus_message_unref(reply);
157 static int help(void) {
159 printf("%s [OPTIONS...] {COMMAND} ...\n\n"
160 "Execute a process while inhibiting shutdown/sleep/idle.\n\n"
161 " -h --help Show this help\n"
162 " --version Show package version\n"
163 " --what=WHAT Operations to inhibit, colon separated list of:\n"
164 " shutdown, sleep, idle, handle-power-key,\n"
165 " handle-suspend-key, handle-hibernate-key,\n"
166 " handle-lid-switch\n"
167 " --who=STRING A descriptive string who is inhibiting\n"
168 " --why=STRING A descriptive string why is being inhibited\n"
169 " --mode=MODE One of block or delay\n"
170 " --list List active inhibitors\n",
171 program_invocation_short_name);
176 static int parse_argv(int argc, char *argv[]) {
187 static const struct option options[] = {
188 { "help", no_argument, NULL, 'h' },
189 { "version", no_argument, NULL, ARG_VERSION },
190 { "what", required_argument, NULL, ARG_WHAT },
191 { "who", required_argument, NULL, ARG_WHO },
192 { "why", required_argument, NULL, ARG_WHY },
193 { "mode", required_argument, NULL, ARG_MODE },
194 { "list", no_argument, NULL, ARG_LIST },
203 while ((c = getopt_long(argc, argv, "+h", options, NULL)) >= 0) {
212 puts(PACKAGE_STRING);
214 puts(SYSTEMD_FEATURES);
234 arg_action = ACTION_LIST;
238 log_error("Unknown option code %c", c);
243 if (arg_action == ACTION_INHIBIT && optind >= argc) {
244 log_error("Missing command line to execute.");
251 int main(int argc, char *argv[]) {
252 int r, exit_code = 0;
253 DBusConnection *bus = NULL;
257 dbus_error_init(&error);
259 log_parse_environment();
262 r = parse_argv(argc, argv);
266 bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &error);
268 log_error("Failed to connect to bus: %s", bus_error_message(&error));
273 if (arg_action == ACTION_LIST) {
275 r = print_inhibitors(bus, &error);
277 log_error("Failed to list inhibitors: %s", bus_error_message_or_strerror(&error, -r));
286 arg_who = w = strv_join(argv + optind, " ");
288 fd = inhibit(bus, &error);
292 log_error("Failed to inhibit: %s", bus_error_message_or_strerror(&error, -r));
299 log_error("Failed to fork: %m");
307 close_nointr_nofail(fd);
308 close_all_fds(NULL, 0);
310 execvp(argv[optind], argv + optind);
311 log_error("Failed to execute %s: %m", argv[optind]);
315 r = wait_for_terminate_and_warn(argv[optind], pid);
322 dbus_connection_close(bus);
323 dbus_connection_unref(bus);
326 dbus_error_free(&error);
329 close_nointr_nofail(fd);
331 return r < 0 ? EXIT_FAILURE : exit_code;