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 static const char* arg_what
= "idle:sleep:shutdown";
42 static const char* arg_who
= NULL
;
43 static const char* arg_why
= "Unknown reason";
44 static const char* arg_mode
= NULL
;
49 } arg_action
= ACTION_INHIBIT
;
51 static int inhibit(sd_bus
*bus
, sd_bus_error
*error
) {
52 _cleanup_(sd_bus_message_unrefp
) sd_bus_message
*reply
= NULL
;
56 r
= sd_bus_call_method(
58 "org.freedesktop.login1",
59 "/org/freedesktop/login1",
60 "org.freedesktop.login1.Manager",
64 "ssss", arg_what
, arg_who
, arg_why
, arg_mode
);
68 r
= sd_bus_message_read_basic(reply
, SD_BUS_TYPE_UNIX_FD
, &fd
);
72 r
= fcntl(fd
, F_DUPFD_CLOEXEC
, 3);
79 static int print_inhibitors(sd_bus
*bus
, sd_bus_error
*error
) {
80 _cleanup_(sd_bus_message_unrefp
) sd_bus_message
*reply
= NULL
;
81 const char *what
, *who
, *why
, *mode
;
82 unsigned int uid
, pid
;
86 r
= sd_bus_call_method(
88 "org.freedesktop.login1",
89 "/org/freedesktop/login1",
90 "org.freedesktop.login1.Manager",
98 r
= sd_bus_message_enter_container(reply
, SD_BUS_TYPE_ARRAY
, "(ssssuu)");
100 return bus_log_parse_error(r
);
102 while ((r
= sd_bus_message_read(reply
, "(ssssuu)", &what
, &who
, &why
, &mode
, &uid
, &pid
)) > 0) {
103 _cleanup_free_
char *comm
= NULL
, *u
= NULL
;
105 if (arg_mode
&& !streq(mode
, arg_mode
))
108 get_process_comm(pid
, &comm
);
109 u
= uid_to_name(uid
);
111 printf(" Who: %s (UID "UID_FMT
"/%s, PID "PID_FMT
"/%s)\n"
115 who
, uid
, strna(u
), pid
, strna(comm
),
123 return bus_log_parse_error(r
);
125 r
= sd_bus_message_exit_container(reply
);
127 return bus_log_parse_error(r
);
129 printf("%u inhibitors listed.\n", n
);
133 static void help(void) {
134 printf("%s [OPTIONS...] {COMMAND} ...\n\n"
135 "Execute a process while inhibiting shutdown/sleep/idle.\n\n"
136 " -h --help Show this help\n"
137 " --version Show package version\n"
138 " --what=WHAT Operations to inhibit, colon separated list of:\n"
139 " shutdown, sleep, idle, handle-power-key,\n"
140 " handle-suspend-key, handle-hibernate-key,\n"
141 " handle-lid-switch\n"
142 " --who=STRING A descriptive string who is inhibiting\n"
143 " --why=STRING A descriptive string why is being inhibited\n"
144 " --mode=MODE One of block or delay\n"
145 " --list List active inhibitors\n"
146 , program_invocation_short_name
);
149 static int parse_argv(int argc
, char *argv
[]) {
160 static const struct option options
[] = {
161 { "help", no_argument
, NULL
, 'h' },
162 { "version", no_argument
, NULL
, ARG_VERSION
},
163 { "what", required_argument
, NULL
, ARG_WHAT
},
164 { "who", required_argument
, NULL
, ARG_WHO
},
165 { "why", required_argument
, NULL
, ARG_WHY
},
166 { "mode", required_argument
, NULL
, ARG_MODE
},
167 { "list", no_argument
, NULL
, ARG_LIST
},
176 while ((c
= getopt_long(argc
, argv
, "+h", options
, NULL
)) >= 0)
204 arg_action
= ACTION_LIST
;
211 assert_not_reached("Unhandled option");
214 if (arg_action
== ACTION_INHIBIT
&& optind
== argc
)
215 arg_action
= ACTION_LIST
;
217 else if (arg_action
== ACTION_INHIBIT
&& optind
>= argc
) {
218 log_error("Missing command line to execute.");
225 int main(int argc
, char *argv
[]) {
226 _cleanup_(sd_bus_error_free
) sd_bus_error error
= SD_BUS_ERROR_NULL
;
227 _cleanup_(sd_bus_flush_close_unrefp
) sd_bus
*bus
= NULL
;
230 log_parse_environment();
233 r
= parse_argv(argc
, argv
);
239 r
= sd_bus_default_system(&bus
);
241 log_error_errno(r
, "Failed to connect to bus: %m");
245 if (arg_action
== ACTION_LIST
) {
247 r
= print_inhibitors(bus
, &error
);
249 log_error("Failed to list inhibitors: %s", bus_error_message(&error
, -r
));
254 _cleanup_close_
int fd
= -1;
255 _cleanup_free_
char *w
= NULL
;
259 arg_who
= w
= strv_join(argv
+ optind
, " ");
264 fd
= inhibit(bus
, &error
);
266 log_error("Failed to inhibit: %s", bus_error_message(&error
, fd
));
272 log_error_errno(errno
, "Failed to fork: %m");
279 (void) reset_all_signal_handlers();
280 (void) reset_signal_mask();
282 close_all_fds(NULL
, 0);
284 execvp(argv
[optind
], argv
+ optind
);
285 log_error_errno(errno
, "Failed to execute %s: %m", argv
[optind
]);
289 r
= wait_for_terminate_and_warn(argv
[optind
], pid
, true);
290 return r
< 0 ? EXIT_FAILURE
: r
;