1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2013 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 "unit-name.h"
30 #include "path-util.h"
31 #include "bus-error.h"
33 static bool arg_scope = false;
34 static bool arg_remain_after_exit = false;
35 static const char *arg_unit = NULL;
36 static const char *arg_description = NULL;
37 static const char *arg_slice = NULL;
38 static bool arg_send_sighup = false;
39 static BusTransport arg_transport = BUS_TRANSPORT_LOCAL;
40 static char *arg_host = NULL;
41 static bool arg_user = false;
43 static int help(void) {
45 printf("%s [OPTIONS...] COMMAND [ARGS...]\n\n"
46 "Run the specified command in a transient scope or service unit.\n\n"
47 " -h --help Show this help\n"
48 " --version Show package version\n"
49 " --user Run as user unit\n"
50 " -H --host=[USER@]HOST Operate on remote host\n"
51 " -M --machine=CONTAINER Operate on local container\n"
52 " --scope Run this as scope rather than service\n"
53 " --unit=UNIT Run under the specified unit name\n"
54 " --description=TEXT Description for unit\n"
55 " --slice=SLICE Run in the specified slice\n"
56 " -r --remain-after-exit Leave service around until explicitly stopped\n"
57 " --send-sighup Send SIGHUP when terminating\n",
58 program_invocation_short_name);
63 static int parse_argv(int argc, char *argv[]) {
76 static const struct option options[] = {
77 { "help", no_argument, NULL, 'h' },
78 { "version", no_argument, NULL, ARG_VERSION },
79 { "user", no_argument, NULL, ARG_USER },
80 { "system", no_argument, NULL, ARG_SYSTEM },
81 { "scope", no_argument, NULL, ARG_SCOPE },
82 { "unit", required_argument, NULL, ARG_UNIT },
83 { "description", required_argument, NULL, ARG_DESCRIPTION },
84 { "slice", required_argument, NULL, ARG_SLICE },
85 { "remain-after-exit", no_argument, NULL, 'r' },
86 { "send-sighup", no_argument, NULL, ARG_SEND_SIGHUP },
87 { "host", required_argument, NULL, 'H' },
88 { "machine", required_argument, NULL, 'M' },
97 while ((c = getopt_long(argc, argv, "+hrH:M:", options, NULL)) >= 0) {
105 puts(PACKAGE_STRING);
106 puts(SYSTEMD_FEATURES);
125 case ARG_DESCRIPTION:
126 arg_description = optarg;
133 case ARG_SEND_SIGHUP:
134 arg_send_sighup = true;
138 arg_remain_after_exit = true;
142 arg_transport = BUS_TRANSPORT_REMOTE;
147 arg_transport = BUS_TRANSPORT_CONTAINER;
155 assert_not_reached("Unhandled option");
159 if (optind >= argc) {
160 log_error("Command line to execute required.");
164 if (arg_user && arg_transport != BUS_TRANSPORT_LOCAL) {
165 log_error("Execution in user context is not supported on non-local systems.");
169 if (arg_scope && arg_transport != BUS_TRANSPORT_LOCAL) {
170 log_error("Scope execution is not supported on non-local systems.");
177 static int message_start_transient_unit_new(sd_bus *bus, const char *name, sd_bus_message **ret) {
178 _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
185 log_info("Running as unit %s.", name);
187 r = sd_bus_message_new_method_call(
189 "org.freedesktop.systemd1",
190 "/org/freedesktop/systemd1",
191 "org.freedesktop.systemd1.Manager",
192 "StartTransientUnit", &m);
196 r = sd_bus_message_append(m, "ss", name, "fail");
200 r = sd_bus_message_open_container(m, 'a', "(sv)");
204 r = sd_bus_message_append(m, "(sv)", "Description", "s", arg_description);
208 if (!isempty(arg_slice)) {
209 _cleanup_free_ char *slice;
211 slice = unit_name_mangle_with_suffix(arg_slice, ".slice");
215 r = sd_bus_message_append(m, "(sv)", "Slice", "s", slice);
220 r = sd_bus_message_append(m, "(sv)", "SendSIGHUP", "b", arg_send_sighup);
230 static int message_start_transient_unit_send(sd_bus *bus, sd_bus_message *m, sd_bus_error *error, sd_bus_message **reply) {
236 r = sd_bus_message_close_container(m);
240 r = sd_bus_message_append(m, "a(sa(sv))", 0);
244 return sd_bus_call(bus, m, 0, error, reply);
247 static int start_transient_service(
250 sd_bus_error *error) {
252 _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
253 _cleanup_free_ char *name = NULL;
258 name = unit_name_mangle_with_suffix(arg_unit, ".service");
260 asprintf(&name, "run-%lu.service", (unsigned long) getpid());
264 r = message_start_transient_unit_new(bus, name, &m);
268 r = sd_bus_message_append(m, "(sv)", "RemainAfterExit", "b", arg_remain_after_exit);
272 r = sd_bus_message_open_container(m, 'r', "sv");
276 r = sd_bus_message_append(m, "s", "ExecStart");
280 r = sd_bus_message_open_container(m, 'v', "a(sasb)");
284 r = sd_bus_message_open_container(m, 'a', "(sasb)");
288 r = sd_bus_message_open_container(m, 'r', "sasb");
292 r = sd_bus_message_append(m, "s", argv[0]);
296 r = sd_bus_message_open_container(m, 'a', "s");
300 STRV_FOREACH(i, argv) {
301 r = sd_bus_message_append(m, "s", *i);
306 r = sd_bus_message_close_container(m);
310 r = sd_bus_message_append(m, "b", false);
314 r = sd_bus_message_close_container(m);
318 r = sd_bus_message_close_container(m);
322 r = sd_bus_message_close_container(m);
326 r = sd_bus_message_close_container(m);
330 return message_start_transient_unit_send(bus, m, error, NULL);
333 static int start_transient_scope(
336 sd_bus_error *error) {
338 _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
339 _cleanup_free_ char *name = NULL;
345 name = unit_name_mangle_with_suffix(arg_unit, ".scope");
347 asprintf(&name, "run-%lu.scope", (unsigned long) getpid());
351 r = message_start_transient_unit_new(bus, name, &m);
355 r = sd_bus_message_append(m, "(sv)", "PIDs", "au", 1, (uint32_t) getpid());
359 r = message_start_transient_unit_send(bus, m, error, NULL);
363 execvp(argv[0], argv);
364 log_error("Failed to execute: %m");
368 int main(int argc, char* argv[]) {
369 _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
370 _cleanup_bus_unref_ sd_bus *bus = NULL;
371 _cleanup_free_ char *description = NULL, *command = NULL;
374 log_parse_environment();
377 r = parse_argv(argc, argv);
381 r = find_binary(argv[optind], &command);
383 log_error("Failed to find executable %s: %s", argv[optind], strerror(-r));
386 argv[optind] = command;
388 if (!arg_description) {
389 description = strv_join(argv + optind, " ");
395 arg_description = description;
398 r = bus_open_transport(arg_transport, arg_host, arg_user, &bus);
400 log_error("Failed to create bus connection: %s", strerror(-r));
405 r = start_transient_scope(bus, argv + optind, &error);
407 r = start_transient_service(bus, argv + optind, &error);
409 log_error("Failed start transient unit: %s", bus_error_message(&error, r));
412 return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;