]> git.ipfire.org Git - thirdparty/systemd.git/blob - src/run/run.c
projects / thirdparty / systemd.git / blob
? search:
summary | shortlog | log | commit | commitdiff | tree
blame | history | raw | HEAD
tree-wide: make bus_map_all_properties return a proper sd_bus_error
[thirdparty/systemd.git] / src / run / run.c
1 /***
2 This file is part of systemd.
3
4 Copyright 2013 Lennart Poettering
5
6 systemd is free software; you can redistribute it and/or modify it
7 under the terms of the GNU Lesser General Public License as published by
8 the Free Software Foundation; either version 2.1 of the License, or
9 (at your option) any later version.
10
11 systemd is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 Lesser General Public License for more details.
15
16 You should have received a copy of the GNU Lesser General Public License
17 along with systemd; If not, see <http://www.gnu.org/licenses/>.
18 ***/
19
20 #include <getopt.h>
21 #include <stdio.h>
22
23 #include "sd-bus.h"
24 #include "sd-event.h"
25
26 #include "alloc-util.h"
27 #include "bus-error.h"
28 #include "bus-unit-util.h"
29 #include "bus-util.h"
30 #include "calendarspec.h"
31 #include "env-util.h"
32 #include "fd-util.h"
33 #include "format-util.h"
34 #include "parse-util.h"
35 #include "path-util.h"
36 #include "process-util.h"
37 #include "ptyfwd.h"
38 #include "signal-util.h"
39 #include "spawn-polkit-agent.h"
40 #include "strv.h"
41 #include "terminal-util.h"
42 #include "unit-name.h"
43 #include "user-util.h"
44
45 static bool arg_ask_password = true;
46 static bool arg_scope = false;
47 static bool arg_remain_after_exit = false;
48 static bool arg_no_block = false;
49 static bool arg_wait = false;
50 static const char *arg_unit = NULL;
51 static const char *arg_description = NULL;
52 static const char *arg_slice = NULL;
53 static bool arg_send_sighup = false;
54 static BusTransport arg_transport = BUS_TRANSPORT_LOCAL;
55 static const char *arg_host = NULL;
56 static bool arg_user = false;
57 static const char *arg_service_type = NULL;
58 static const char *arg_exec_user = NULL;
59 static const char *arg_exec_group = NULL;
60 static int arg_nice = 0;
61 static bool arg_nice_set = false;
62 static char **arg_environment = NULL;
63 static char **arg_property = NULL;
64 static bool arg_pty = false;
65 static usec_t arg_on_active = 0;
66 static usec_t arg_on_boot = 0;
67 static usec_t arg_on_startup = 0;
68 static usec_t arg_on_unit_active = 0;
69 static usec_t arg_on_unit_inactive = 0;
70 static const char *arg_on_calendar = NULL;
71 static char **arg_timer_property = NULL;
72 static bool arg_quiet = false;
73
74 static void polkit_agent_open_if_enabled(void) {
75
76 /* Open the polkit agent as a child process if necessary */
77 if (!arg_ask_password)
78 return;
79
80 if (arg_transport != BUS_TRANSPORT_LOCAL)
81 return;
82
83 polkit_agent_open();
84 }
85
86 static void help(void) {
87 printf("%s [OPTIONS...] {COMMAND} [ARGS...]\n\n"
88 "Run the specified command in a transient scope or service.\n\n"
89 " -h --help Show this help\n"
90 " --version Show package version\n"
91 " --no-ask-password Do not prompt for password\n"
92 " --user Run as user unit\n"
93 " -H --host=[USER@]HOST Operate on remote host\n"
94 " -M --machine=CONTAINER Operate on local container\n"
95 " --scope Run this as scope rather than service\n"
96 " --unit=UNIT Run under the specified unit name\n"
97 " -p --property=NAME=VALUE Set service or scope unit property\n"
98 " --description=TEXT Description for unit\n"
99 " --slice=SLICE Run in the specified slice\n"
100 " --no-block Do not wait until operation finished\n"
101 " -r --remain-after-exit Leave service around until explicitly stopped\n"
102 " --wait Wait until service stopped again\n"
103 " --send-sighup Send SIGHUP when terminating\n"
104 " --service-type=TYPE Service type\n"
105 " --uid=USER Run as system user\n"
106 " --gid=GROUP Run as system group\n"
107 " --nice=NICE Nice level\n"
108 " -E --setenv=NAME=VALUE Set environment\n"
109 " -t --pty Run service on pseudo tty\n"
110 " -q --quiet Suppress information messages during runtime\n\n"
111 "Timer options:\n"
112 " --on-active=SECONDS Run after SECONDS delay\n"
113 " --on-boot=SECONDS Run SECONDS after machine was booted up\n"
114 " --on-startup=SECONDS Run SECONDS after systemd activation\n"
115 " --on-unit-active=SECONDS Run SECONDS after the last activation\n"
116 " --on-unit-inactive=SECONDS Run SECONDS after the last deactivation\n"
117 " --on-calendar=SPEC Realtime timer\n"
118 " --timer-property=NAME=VALUE Set timer unit property\n"
119 , program_invocation_short_name);
120 }
121
122 static bool with_timer(void) {
123 return arg_on_active || arg_on_boot || arg_on_startup || arg_on_unit_active || arg_on_unit_inactive || arg_on_calendar;
124 }
125
126 static int parse_argv(int argc, char *argv[]) {
127
128 enum {
129 ARG_VERSION = 0x100,
130 ARG_USER,
131 ARG_SYSTEM,
132 ARG_SCOPE,
133 ARG_UNIT,
134 ARG_DESCRIPTION,
135 ARG_SLICE,
136 ARG_SEND_SIGHUP,
137 ARG_SERVICE_TYPE,
138 ARG_EXEC_USER,
139 ARG_EXEC_GROUP,
140 ARG_NICE,
141 ARG_ON_ACTIVE,
142 ARG_ON_BOOT,
143 ARG_ON_STARTUP,
144 ARG_ON_UNIT_ACTIVE,
145 ARG_ON_UNIT_INACTIVE,
146 ARG_ON_CALENDAR,
147 ARG_TIMER_PROPERTY,
148 ARG_NO_BLOCK,
149 ARG_NO_ASK_PASSWORD,
150 ARG_WAIT,
151 };
152
153 static const struct option options[] = {
154 { "help", no_argument, NULL, 'h' },
155 { "version", no_argument, NULL, ARG_VERSION },
156 { "user", no_argument, NULL, ARG_USER },
157 { "system", no_argument, NULL, ARG_SYSTEM },
158 { "scope", no_argument, NULL, ARG_SCOPE },
159 { "unit", required_argument, NULL, ARG_UNIT },
160 { "description", required_argument, NULL, ARG_DESCRIPTION },
161 { "slice", required_argument, NULL, ARG_SLICE },
162 { "remain-after-exit", no_argument, NULL, 'r' },
163 { "send-sighup", no_argument, NULL, ARG_SEND_SIGHUP },
164 { "host", required_argument, NULL, 'H' },
165 { "machine", required_argument, NULL, 'M' },
166 { "service-type", required_argument, NULL, ARG_SERVICE_TYPE },
167 { "wait", no_argument, NULL, ARG_WAIT },
168 { "uid", required_argument, NULL, ARG_EXEC_USER },
169 { "gid", required_argument, NULL, ARG_EXEC_GROUP },
170 { "nice", required_argument, NULL, ARG_NICE },
171 { "setenv", required_argument, NULL, 'E' },
172 { "property", required_argument, NULL, 'p' },
173 { "tty", no_argument, NULL, 't' }, /* deprecated */
174 { "pty", no_argument, NULL, 't' },
175 { "quiet", no_argument, NULL, 'q' },
176 { "on-active", required_argument, NULL, ARG_ON_ACTIVE },
177 { "on-boot", required_argument, NULL, ARG_ON_BOOT },
178 { "on-startup", required_argument, NULL, ARG_ON_STARTUP },
179 { "on-unit-active", required_argument, NULL, ARG_ON_UNIT_ACTIVE },
180 { "on-unit-inactive", required_argument, NULL, ARG_ON_UNIT_INACTIVE },
181 { "on-calendar", required_argument, NULL, ARG_ON_CALENDAR },
182 { "timer-property", required_argument, NULL, ARG_TIMER_PROPERTY },
183 { "no-block", no_argument, NULL, ARG_NO_BLOCK },
184 { "no-ask-password", no_argument, NULL, ARG_NO_ASK_PASSWORD },
185 {},
186 };
187
188 int r, c;
189
190 assert(argc >= 0);
191 assert(argv);
192
193 while ((c = getopt_long(argc, argv, "+hrH:M:E:p:tq", options, NULL)) >= 0)
194
195 switch (c) {
196
197 case 'h':
198 help();
199 return 0;
200
201 case ARG_VERSION:
202 return version();
203
204 case ARG_NO_ASK_PASSWORD:
205 arg_ask_password = false;
206 break;
207
208 case ARG_USER:
209 arg_user = true;
210 break;
211
212 case ARG_SYSTEM:
213 arg_user = false;
214 break;
215
216 case ARG_SCOPE:
217 arg_scope = true;
218 break;
219
220 case ARG_UNIT:
221 arg_unit = optarg;
222 break;
223
224 case ARG_DESCRIPTION:
225 arg_description = optarg;
226 break;
227
228 case ARG_SLICE:
229 arg_slice = optarg;
230 break;
231
232 case ARG_SEND_SIGHUP:
233 arg_send_sighup = true;
234 break;
235
236 case 'r':
237 arg_remain_after_exit = true;
238 break;
239
240 case 'H':
241 arg_transport = BUS_TRANSPORT_REMOTE;
242 arg_host = optarg;
243 break;
244
245 case 'M':
246 arg_transport = BUS_TRANSPORT_MACHINE;
247 arg_host = optarg;
248 break;
249
250 case ARG_SERVICE_TYPE:
251 arg_service_type = optarg;
252 break;
253
254 case ARG_EXEC_USER:
255 arg_exec_user = optarg;
256 break;
257
258 case ARG_EXEC_GROUP:
259 arg_exec_group = optarg;
260 break;
261
262 case ARG_NICE:
263 r = parse_nice(optarg, &arg_nice);
264 if (r < 0)
265 return log_error_errno(r, "Failed to parse nice value: %s", optarg);
266
267 arg_nice_set = true;
268 break;
269
270 case 'E':
271 if (strv_extend(&arg_environment, optarg) < 0)
272 return log_oom();
273
274 break;
275
276 case 'p':
277 if (strv_extend(&arg_property, optarg) < 0)
278 return log_oom();
279
280 break;
281
282 case 't':
283 arg_pty = true;
284 break;
285
286 case 'q':
287 arg_quiet = true;
288 break;
289
290 case ARG_ON_ACTIVE:
291
292 r = parse_sec(optarg, &arg_on_active);
293 if (r < 0) {
294 log_error("Failed to parse timer value: %s", optarg);
295 return r;
296 }
297
298 break;
299
300 case ARG_ON_BOOT:
301
302 r = parse_sec(optarg, &arg_on_boot);
303 if (r < 0) {
304 log_error("Failed to parse timer value: %s", optarg);
305 return r;
306 }
307
308 break;
309
310 case ARG_ON_STARTUP:
311
312 r = parse_sec(optarg, &arg_on_startup);
313 if (r < 0) {
314 log_error("Failed to parse timer value: %s", optarg);
315 return r;
316 }
317
318 break;
319
320 case ARG_ON_UNIT_ACTIVE:
321
322 r = parse_sec(optarg, &arg_on_unit_active);
323 if (r < 0) {
324 log_error("Failed to parse timer value: %s", optarg);
325 return r;
326 }
327
328 break;
329
330 case ARG_ON_UNIT_INACTIVE:
331
332 r = parse_sec(optarg, &arg_on_unit_inactive);
333 if (r < 0) {
334 log_error("Failed to parse timer value: %s", optarg);
335 return r;
336 }
337
338 break;
339
340 case ARG_ON_CALENDAR: {
341 CalendarSpec *spec = NULL;
342
343 r = calendar_spec_from_string(optarg, &spec);
344 if (r < 0) {
345 log_error("Invalid calendar spec: %s", optarg);
346 return r;
347 }
348
349 calendar_spec_free(spec);
350 arg_on_calendar = optarg;
351 break;
352 }
353
354 case ARG_TIMER_PROPERTY:
355
356 if (strv_extend(&arg_timer_property, optarg) < 0)
357 return log_oom();
358
359 break;
360
361 case ARG_NO_BLOCK:
362 arg_no_block = true;
363 break;
364
365 case ARG_WAIT:
366 arg_wait = true;
367 break;
368
369 case '?':
370 return -EINVAL;
371
372 default:
373 assert_not_reached("Unhandled option");
374 }
375
376 if ((optind >= argc) && (!arg_unit || !with_timer())) {
377 log_error("Command line to execute required.");
378 return -EINVAL;
379 }
380
381 if (arg_user && arg_transport != BUS_TRANSPORT_LOCAL) {
382 log_error("Execution in user context is not supported on non-local systems.");
383 return -EINVAL;
384 }
385
386 if (arg_scope && arg_transport != BUS_TRANSPORT_LOCAL) {
387 log_error("Scope execution is not supported on non-local systems.");
388 return -EINVAL;
389 }
390
391 if (arg_scope && (arg_remain_after_exit || arg_service_type)) {
392 log_error("--remain-after-exit and --service-type= are not supported in --scope mode.");
393 return -EINVAL;
394 }
395
396 if (arg_pty && (with_timer() || arg_scope)) {
397 log_error("--pty is not compatible in timer or --scope mode.");
398 return -EINVAL;
399 }
400
401 if (arg_pty && arg_transport == BUS_TRANSPORT_REMOTE) {
402 log_error("--pty is only supported when connecting to the local system or containers.");
403 return -EINVAL;
404 }
405
406 if (arg_pty && arg_no_block) {
407 log_error("--pty is not compatible with --no-block.");
408 return -EINVAL;
409 }
410
411 if (arg_scope && with_timer()) {
412 log_error("Timer options are not supported in --scope mode.");
413 return -EINVAL;
414 }
415
416 if (arg_timer_property && !with_timer()) {
417 log_error("--timer-property= has no effect without any other timer options.");
418 return -EINVAL;
419 }
420
421 if (arg_wait) {
422 if (arg_no_block) {
423 log_error("--wait may not be combined with --no-block.");
424 return -EINVAL;
425 }
426
427 if (with_timer()) {
428 log_error("--wait may not be combined with timer operations.");
429 return -EINVAL;
430 }
431
432 if (arg_scope) {
433 log_error("--wait may not be combined with --scope.");
434 return -EINVAL;
435 }
436 }
437
438 return 1;
439 }
440
441 static int transient_unit_set_properties(sd_bus_message *m, char **properties) {
442 int r;
443
444 r = sd_bus_message_append(m, "(sv)", "Description", "s", arg_description);
445 if (r < 0)
446 return r;
447
448 r = bus_append_unit_property_assignment_many(m, properties);
449 if (r < 0)
450 return r;
451
452 return 0;
453 }
454
455 static int transient_cgroup_set_properties(sd_bus_message *m) {
456 int r;
457 assert(m);
458
459 if (!isempty(arg_slice)) {
460 _cleanup_free_ char *slice;
461
462 r = unit_name_mangle_with_suffix(arg_slice, UNIT_NAME_NOGLOB, ".slice", &slice);
463 if (r < 0)
464 return r;
465
466 r = sd_bus_message_append(m, "(sv)", "Slice", "s", slice);
467 if (r < 0)
468 return r;
469 }
470
471 return 0;
472 }
473
474 static int transient_kill_set_properties(sd_bus_message *m) {
475 assert(m);
476
477 if (arg_send_sighup)
478 return sd_bus_message_append(m, "(sv)", "SendSIGHUP", "b", arg_send_sighup);
479 else
480 return 0;
481 }
482
483 static int transient_service_set_properties(sd_bus_message *m, char **argv, const char *pty_path) {
484 int r;
485
486 assert(m);
487
488 r = transient_unit_set_properties(m, arg_property);
489 if (r < 0)
490 return r;
491
492 r = transient_kill_set_properties(m);
493 if (r < 0)
494 return r;
495
496 r = transient_cgroup_set_properties(m);
497 if (r < 0)
498 return r;
499
500 if (arg_wait) {
501 r = sd_bus_message_append(m, "(sv)", "AddRef", "b", 1);
502 if (r < 0)
503 return r;
504 }
505
506 if (arg_remain_after_exit) {
507 r = sd_bus_message_append(m, "(sv)", "RemainAfterExit", "b", arg_remain_after_exit);
508 if (r < 0)
509 return r;
510 }
511
512 if (arg_service_type) {
513 r = sd_bus_message_append(m, "(sv)", "Type", "s", arg_service_type);
514 if (r < 0)
515 return r;
516 }
517
518 if (arg_exec_user) {
519 r = sd_bus_message_append(m, "(sv)", "User", "s", arg_exec_user);
520 if (r < 0)
521 return r;
522 }
523
524 if (arg_exec_group) {
525 r = sd_bus_message_append(m, "(sv)", "Group", "s", arg_exec_group);
526 if (r < 0)
527 return r;
528 }
529
530 if (arg_nice_set) {
531 r = sd_bus_message_append(m, "(sv)", "Nice", "i", arg_nice);
532 if (r < 0)
533 return r;
534 }
535
536 if (pty_path) {
537 const char *e;
538
539 r = sd_bus_message_append(m,
540 "(sv)(sv)(sv)(sv)",
541 "StandardInput", "s", "tty",
542 "StandardOutput", "s", "tty",
543 "StandardError", "s", "tty",
544 "TTYPath", "s", pty_path);
545 if (r < 0)
546 return r;
547
548 e = getenv("TERM");
549 if (e) {
550 char *n;
551
552 n = strjoina("TERM=", e);
553 r = sd_bus_message_append(m,
554 "(sv)",
555 "Environment", "as", 1, n);
556 if (r < 0)
557 return r;
558 }
559 }
560
561 if (!strv_isempty(arg_environment)) {
562 r = sd_bus_message_open_container(m, 'r', "sv");
563 if (r < 0)
564 return r;
565
566 r = sd_bus_message_append(m, "s", "Environment");
567 if (r < 0)
568 return r;
569
570 r = sd_bus_message_open_container(m, 'v', "as");
571 if (r < 0)
572 return r;
573
574 r = sd_bus_message_append_strv(m, arg_environment);
575 if (r < 0)
576 return r;
577
578 r = sd_bus_message_close_container(m);
579 if (r < 0)
580 return r;
581
582 r = sd_bus_message_close_container(m);
583 if (r < 0)
584 return r;
585 }
586
587 /* Exec container */
588 {
589 r = sd_bus_message_open_container(m, 'r', "sv");
590 if (r < 0)
591 return r;
592
593 r = sd_bus_message_append(m, "s", "ExecStart");
594 if (r < 0)
595 return r;
596
597 r = sd_bus_message_open_container(m, 'v', "a(sasb)");
598 if (r < 0)
599 return r;
600
601 r = sd_bus_message_open_container(m, 'a', "(sasb)");
602 if (r < 0)
603 return r;
604
605 r = sd_bus_message_open_container(m, 'r', "sasb");
606 if (r < 0)
607 return r;
608
609 r = sd_bus_message_append(m, "s", argv[0]);
610 if (r < 0)
611 return r;
612
613 r = sd_bus_message_append_strv(m, argv);
614 if (r < 0)
615 return r;
616
617 r = sd_bus_message_append(m, "b", false);
618 if (r < 0)
619 return r;
620
621 r = sd_bus_message_close_container(m);
622 if (r < 0)
623 return r;
624
625 r = sd_bus_message_close_container(m);
626 if (r < 0)
627 return r;
628
629 r = sd_bus_message_close_container(m);
630 if (r < 0)
631 return r;
632
633 r = sd_bus_message_close_container(m);
634 if (r < 0)
635 return r;
636 }
637
638 return 0;
639 }
640
641 static int transient_scope_set_properties(sd_bus_message *m) {
642 int r;
643
644 assert(m);
645
646 r = transient_unit_set_properties(m, arg_property);
647 if (r < 0)
648 return r;
649
650 r = transient_kill_set_properties(m);
651 if (r < 0)
652 return r;
653
654 r = transient_cgroup_set_properties(m);
655 if (r < 0)
656 return r;
657
658 r = sd_bus_message_append(m, "(sv)", "PIDs", "au", 1, (uint32_t) getpid());
659 if (r < 0)
660 return r;
661
662 return 0;
663 }
664
665 static int transient_timer_set_properties(sd_bus_message *m) {
666 int r;
667
668 assert(m);
669
670 r = transient_unit_set_properties(m, arg_timer_property);
671 if (r < 0)
672 return r;
673
674 /* Automatically clean up our transient timers */
675 r = sd_bus_message_append(m, "(sv)", "RemainAfterElapse", "b", false);
676 if (r < 0)
677 return r;
678
679 if (arg_on_active) {
680 r = sd_bus_message_append(m, "(sv)", "OnActiveSec", "t", arg_on_active);
681 if (r < 0)
682 return r;
683 }
684
685 if (arg_on_boot) {
686 r = sd_bus_message_append(m, "(sv)", "OnBootSec", "t", arg_on_boot);
687 if (r < 0)
688 return r;
689 }
690
691 if (arg_on_startup) {
692 r = sd_bus_message_append(m, "(sv)", "OnStartupSec", "t", arg_on_startup);
693 if (r < 0)
694 return r;
695 }
696
697 if (arg_on_unit_active) {
698 r = sd_bus_message_append(m, "(sv)", "OnUnitActiveSec", "t", arg_on_unit_active);
699 if (r < 0)
700 return r;
701 }
702
703 if (arg_on_unit_inactive) {
704 r = sd_bus_message_append(m, "(sv)", "OnUnitInactiveSec", "t", arg_on_unit_inactive);
705 if (r < 0)
706 return r;
707 }
708
709 if (arg_on_calendar) {
710 r = sd_bus_message_append(m, "(sv)", "OnCalendar", "s", arg_on_calendar);
711 if (r < 0)
712 return r;
713 }
714
715 return 0;
716 }
717
718 static int make_unit_name(sd_bus *bus, UnitType t, char **ret) {
719 const char *unique, *id;
720 char *p;
721 int r;
722
723 assert(bus);
724 assert(t >= 0);
725 assert(t < _UNIT_TYPE_MAX);
726
727 r = sd_bus_get_unique_name(bus, &unique);
728 if (r < 0) {
729 sd_id128_t rnd;
730
731 /* We couldn't get the unique name, which is a pretty
732 * common case if we are connected to systemd
733 * directly. In that case, just pick a random uuid as
734 * name */
735
736 r = sd_id128_randomize(&rnd);
737 if (r < 0)
738 return log_error_errno(r, "Failed to generate random run unit name: %m");
739
740 if (asprintf(ret, "run-r" SD_ID128_FORMAT_STR ".%s", SD_ID128_FORMAT_VAL(rnd), unit_type_to_string(t)) < 0)
741 return log_oom();
742
743 return 0;
744 }
745
746 /* We managed to get the unique name, then let's use that to
747 * name our transient units. */
748
749 id = startswith(unique, ":1.");
750 if (!id) {
751 log_error("Unique name %s has unexpected format.", unique);
752 return -EINVAL;
753 }
754
755 p = strjoin("run-u", id, ".", unit_type_to_string(t));
756 if (!p)
757 return log_oom();
758
759 *ret = p;
760 return 0;
761 }
762
763 typedef struct RunContext {
764 sd_bus *bus;
765 sd_event *event;
766 PTYForward *forward;
767 sd_bus_slot *match;
768
769 /* The exit data of the unit */
770 char *active_state;
771 uint64_t inactive_exit_usec;
772 uint64_t inactive_enter_usec;
773 char *result;
774 uint64_t cpu_usage_nsec;
775 uint32_t exit_code;
776 uint32_t exit_status;
777 } RunContext;
778
779 static void run_context_free(RunContext *c) {
780 assert(c);
781
782 c->forward = pty_forward_free(c->forward);
783 c->match = sd_bus_slot_unref(c->match);
784 c->bus = sd_bus_unref(c->bus);
785 c->event = sd_event_unref(c->event);
786
787 free(c->active_state);
788 free(c->result);
789 }
790
791 static void run_context_check_done(RunContext *c) {
792 bool done;
793
794 assert(c);
795
796 if (c->match)
797 done = STRPTR_IN_SET(c->active_state, "inactive", "failed");
798 else
799 done = true;
800
801 if (c->forward && done) /* If the service is gone, it's time to drain the output */
802 done = pty_forward_drain(c->forward);
803
804 if (done)
805 sd_event_exit(c->event, EXIT_SUCCESS);
806 }
807
808 static int run_context_update(RunContext *c, const char *path) {
809
810 static const struct bus_properties_map map[] = {
811 { "ActiveState", "s", NULL, offsetof(RunContext, active_state) },
812 { "InactiveExitTimestampMonotonic", "t", NULL, offsetof(RunContext, inactive_exit_usec) },
813 { "InactiveEnterTimestampMonotonic", "t", NULL, offsetof(RunContext, inactive_enter_usec) },
814 { "Result", "s", NULL, offsetof(RunContext, result) },
815 { "ExecMainCode", "i", NULL, offsetof(RunContext, exit_code) },
816 { "ExecMainStatus", "i", NULL, offsetof(RunContext, exit_status) },
817 { "CPUUsageNSec", "t", NULL, offsetof(RunContext, cpu_usage_nsec) },
818 {}
819 };
820
821 _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
822 int r;
823
824 r = bus_map_all_properties(c->bus,
825 "org.freedesktop.systemd1",
826 path,
827 map,
828 &error,
829 c);
830 if (r < 0) {
831 sd_event_exit(c->event, EXIT_FAILURE);
832 return log_error_errno(r, "Failed to query unit state: %s", bus_error_message(&error, r));
833 }
834
835 run_context_check_done(c);
836 return 0;
837 }
838
839 static int on_properties_changed(sd_bus_message *m, void *userdata, sd_bus_error *error) {
840 RunContext *c = userdata;
841
842 assert(m);
843 assert(c);
844
845 return run_context_update(c, sd_bus_message_get_path(m));
846 }
847
848 static int pty_forward_handler(PTYForward *f, int rcode, void *userdata) {
849 RunContext *c = userdata;
850
851 assert(f);
852
853 if (rcode < 0) {
854 sd_event_exit(c->event, EXIT_FAILURE);
855 return log_error_errno(rcode, "Error on PTY forwarding logic: %m");
856 }
857
858 run_context_check_done(c);
859 return 0;
860 }
861
862 static int start_transient_service(
863 sd_bus *bus,
864 char **argv,
865 int *retval) {
866
867 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL, *reply = NULL;
868 _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
869 _cleanup_(bus_wait_for_jobs_freep) BusWaitForJobs *w = NULL;
870 _cleanup_free_ char *service = NULL, *pty_path = NULL;
871 _cleanup_close_ int master = -1;
872 int r;
873
874 assert(bus);
875 assert(argv);
876 assert(retval);
877
878 if (arg_pty) {
879
880 if (arg_transport == BUS_TRANSPORT_LOCAL) {
881 master = posix_openpt(O_RDWR|O_NOCTTY|O_CLOEXEC|O_NDELAY);
882 if (master < 0)
883 return log_error_errno(errno, "Failed to acquire pseudo tty: %m");
884
885 r = ptsname_malloc(master, &pty_path);
886 if (r < 0)
887 return log_error_errno(r, "Failed to determine tty name: %m");
888
889 if (unlockpt(master) < 0)
890 return log_error_errno(errno, "Failed to unlock tty: %m");
891
892 } else if (arg_transport == BUS_TRANSPORT_MACHINE) {
893 _cleanup_(sd_bus_unrefp) sd_bus *system_bus = NULL;
894 _cleanup_(sd_bus_message_unrefp) sd_bus_message *pty_reply = NULL;
895 const char *s;
896
897 r = sd_bus_default_system(&system_bus);
898 if (r < 0)
899 return log_error_errno(r, "Failed to connect to system bus: %m");
900
901 r = sd_bus_call_method(system_bus,
902 "org.freedesktop.machine1",
903 "/org/freedesktop/machine1",
904 "org.freedesktop.machine1.Manager",
905 "OpenMachinePTY",
906 &error,
907 &pty_reply,
908 "s", arg_host);
909 if (r < 0) {
910 log_error("Failed to get machine PTY: %s", bus_error_message(&error, -r));
911 return r;
912 }
913
914 r = sd_bus_message_read(pty_reply, "hs", &master, &s);
915 if (r < 0)
916 return bus_log_parse_error(r);
917
918 master = fcntl(master, F_DUPFD_CLOEXEC, 3);
919 if (master < 0)
920 return log_error_errno(errno, "Failed to duplicate master fd: %m");
921
922 pty_path = strdup(s);
923 if (!pty_path)
924 return log_oom();
925 } else
926 assert_not_reached("Can't allocate tty via ssh");
927 }
928
929 if (!arg_no_block) {
930 r = bus_wait_for_jobs_new(bus, &w);
931 if (r < 0)
932 return log_error_errno(r, "Could not watch jobs: %m");
933 }
934
935 if (arg_unit) {
936 r = unit_name_mangle_with_suffix(arg_unit, UNIT_NAME_NOGLOB, ".service", &service);
937 if (r < 0)
938 return log_error_errno(r, "Failed to mangle unit name: %m");
939 } else {
940 r = make_unit_name(bus, UNIT_SERVICE, &service);
941 if (r < 0)
942 return r;
943 }
944
945 r = sd_bus_message_new_method_call(
946 bus,
947 &m,
948 "org.freedesktop.systemd1",
949 "/org/freedesktop/systemd1",
950 "org.freedesktop.systemd1.Manager",
951 "StartTransientUnit");
952 if (r < 0)
953 return bus_log_create_error(r);
954
955 r = sd_bus_message_set_allow_interactive_authorization(m, arg_ask_password);
956 if (r < 0)
957 return bus_log_create_error(r);
958
959 /* Name and mode */
960 r = sd_bus_message_append(m, "ss", service, "fail");
961 if (r < 0)
962 return bus_log_create_error(r);
963
964 /* Properties */
965 r = sd_bus_message_open_container(m, 'a', "(sv)");
966 if (r < 0)
967 return bus_log_create_error(r);
968
969 r = transient_service_set_properties(m, argv, pty_path);
970 if (r < 0)
971 return bus_log_create_error(r);
972
973 r = sd_bus_message_close_container(m);
974 if (r < 0)
975 return bus_log_create_error(r);
976
977 /* Auxiliary units */
978 r = sd_bus_message_append(m, "a(sa(sv))", 0);
979 if (r < 0)
980 return bus_log_create_error(r);
981
982 polkit_agent_open_if_enabled();
983
984 r = sd_bus_call(bus, m, 0, &error, &reply);
985 if (r < 0)
986 return log_error_errno(r, "Failed to start transient service unit: %s", bus_error_message(&error, r));
987
988 if (w) {
989 const char *object;
990
991 r = sd_bus_message_read(reply, "o", &object);
992 if (r < 0)
993 return bus_log_parse_error(r);
994
995 r = bus_wait_for_jobs_one(w, object, arg_quiet);
996 if (r < 0)
997 return r;
998 }
999
1000 if (!arg_quiet)
1001 log_info("Running as unit: %s", service);
1002
1003 if (arg_wait || master >= 0) {
1004 _cleanup_(run_context_free) RunContext c = {};
1005 _cleanup_free_ char *path = NULL;
1006 const char *mt;
1007
1008 c.bus = sd_bus_ref(bus);
1009
1010 r = sd_event_default(&c.event);
1011 if (r < 0)
1012 return log_error_errno(r, "Failed to get event loop: %m");
1013
1014 if (master >= 0) {
1015 assert_se(sigprocmask_many(SIG_BLOCK, NULL, SIGWINCH, SIGTERM, SIGINT, -1) >= 0);
1016 (void) sd_event_add_signal(c.event, NULL, SIGINT, NULL, NULL);
1017 (void) sd_event_add_signal(c.event, NULL, SIGTERM, NULL, NULL);
1018
1019 if (!arg_quiet)
1020 log_info("Press ^] three times within 1s to disconnect TTY.");
1021
1022 r = pty_forward_new(c.event, master, PTY_FORWARD_IGNORE_INITIAL_VHANGUP, &c.forward);
1023 if (r < 0)
1024 return log_error_errno(r, "Failed to create PTY forwarder: %m");
1025
1026 pty_forward_set_handler(c.forward, pty_forward_handler, &c);
1027 }
1028
1029
1030 path = unit_dbus_path_from_name(service);
1031 if (!path)
1032 return log_oom();
1033
1034 mt = strjoina("type='signal',"
1035 "sender='org.freedesktop.systemd1',"
1036 "path='", path, "',"
1037 "interface='org.freedesktop.DBus.Properties',"
1038 "member='PropertiesChanged'");
1039 r = sd_bus_add_match(bus, &c.match, mt, on_properties_changed, &c);
1040 if (r < 0)
1041 return log_error_errno(r, "Failed to add properties changed signal.");
1042
1043 r = sd_bus_attach_event(bus, c.event, 0);
1044 if (r < 0)
1045 return log_error_errno(r, "Failed to attach bus to event loop.");
1046
1047 r = run_context_update(&c, path);
1048 if (r < 0)
1049 return r;
1050
1051 r = sd_event_loop(c.event);
1052 if (r < 0)
1053 return log_error_errno(r, "Failed to run event loop: %m");
1054
1055 if (c.forward) {
1056 char last_char = 0;
1057
1058 r = pty_forward_get_last_char(c.forward, &last_char);
1059 if (r >= 0 && !arg_quiet && last_char != '\n')
1060 fputc('\n', stdout);
1061 }
1062
1063 if (arg_wait && !arg_quiet) {
1064
1065 /* Explicitly destroy the PTY forwarder, so that the PTY device is usable again, in its
1066 * original settings (i.e. proper line breaks), so that we can show the summary in a pretty
1067 * way. */
1068 c.forward = pty_forward_free(c.forward);
1069
1070 if (!isempty(c.result))
1071 log_info("Finished with result: %s", strna(c.result));
1072
1073 if (c.exit_code == CLD_EXITED)
1074 log_info("Main processes terminated with: code=%s/status=%i", sigchld_code_to_string(c.exit_code), c.exit_status);
1075 else if (c.exit_code > 0)
1076 log_info("Main processes terminated with: code=%s/status=%s", sigchld_code_to_string(c.exit_code), signal_to_string(c.exit_status));
1077
1078 if (c.inactive_enter_usec > 0 && c.inactive_enter_usec != USEC_INFINITY &&
1079 c.inactive_exit_usec > 0 && c.inactive_exit_usec != USEC_INFINITY &&
1080 c.inactive_enter_usec > c.inactive_exit_usec) {
1081 char ts[FORMAT_TIMESPAN_MAX];
1082 log_info("Service runtime: %s", format_timespan(ts, sizeof(ts), c.inactive_enter_usec - c.inactive_exit_usec, USEC_PER_MSEC));
1083 }
1084
1085 if (c.cpu_usage_nsec > 0 && c.cpu_usage_nsec != NSEC_INFINITY) {
1086 char ts[FORMAT_TIMESPAN_MAX];
1087 log_info("CPU time consumed: %s", format_timespan(ts, sizeof(ts), (c.cpu_usage_nsec + NSEC_PER_USEC - 1) / NSEC_PER_USEC, USEC_PER_MSEC));
1088 }
1089 }
1090
1091 /* Try to propagate the service's return value */
1092 if (c.result && STR_IN_SET(c.result, "success", "exit-code") && c.exit_code == CLD_EXITED)
1093 *retval = c.exit_status;
1094 else
1095 *retval = EXIT_FAILURE;
1096 }
1097
1098 return 0;
1099 }
1100
1101 static int start_transient_scope(
1102 sd_bus *bus,
1103 char **argv) {
1104
1105 _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
1106 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL, *reply = NULL;
1107 _cleanup_(bus_wait_for_jobs_freep) BusWaitForJobs *w = NULL;
1108 _cleanup_strv_free_ char **env = NULL, **user_env = NULL;
1109 _cleanup_free_ char *scope = NULL;
1110 const char *object = NULL;
1111 int r;
1112
1113 assert(bus);
1114 assert(argv);
1115
1116 r = bus_wait_for_jobs_new(bus, &w);
1117 if (r < 0)
1118 return log_oom();
1119
1120 if (arg_unit) {
1121 r = unit_name_mangle_with_suffix(arg_unit, UNIT_NAME_NOGLOB, ".scope", &scope);
1122 if (r < 0)
1123 return log_error_errno(r, "Failed to mangle scope name: %m");
1124 } else {
1125 r = make_unit_name(bus, UNIT_SCOPE, &scope);
1126 if (r < 0)
1127 return r;
1128 }
1129
1130 r = sd_bus_message_new_method_call(
1131 bus,
1132 &m,
1133 "org.freedesktop.systemd1",
1134 "/org/freedesktop/systemd1",
1135 "org.freedesktop.systemd1.Manager",
1136 "StartTransientUnit");
1137 if (r < 0)
1138 return bus_log_create_error(r);
1139
1140 r = sd_bus_message_set_allow_interactive_authorization(m, arg_ask_password);
1141 if (r < 0)
1142 return bus_log_create_error(r);
1143
1144 /* Name and Mode */
1145 r = sd_bus_message_append(m, "ss", scope, "fail");
1146 if (r < 0)
1147 return bus_log_create_error(r);
1148
1149 /* Properties */
1150 r = sd_bus_message_open_container(m, 'a', "(sv)");
1151 if (r < 0)
1152 return bus_log_create_error(r);
1153
1154 r = transient_scope_set_properties(m);
1155 if (r < 0)
1156 return bus_log_create_error(r);
1157
1158 r = sd_bus_message_close_container(m);
1159 if (r < 0)
1160 return bus_log_create_error(r);
1161
1162 /* Auxiliary units */
1163 r = sd_bus_message_append(m, "a(sa(sv))", 0);
1164 if (r < 0)
1165 return bus_log_create_error(r);
1166
1167 polkit_agent_open_if_enabled();
1168
1169 r = sd_bus_call(bus, m, 0, &error, &reply);
1170 if (r < 0) {
1171 log_error("Failed to start transient scope unit: %s", bus_error_message(&error, -r));
1172 return r;
1173 }
1174
1175 if (arg_nice_set) {
1176 if (setpriority(PRIO_PROCESS, 0, arg_nice) < 0)
1177 return log_error_errno(errno, "Failed to set nice level: %m");
1178 }
1179
1180 if (arg_exec_group) {
1181 gid_t gid;
1182
1183 r = get_group_creds(&arg_exec_group, &gid);
1184 if (r < 0)
1185 return log_error_errno(r, "Failed to resolve group %s: %m", arg_exec_group);
1186
1187 if (setresgid(gid, gid, gid) < 0)
1188 return log_error_errno(errno, "Failed to change GID to " GID_FMT ": %m", gid);
1189 }
1190
1191 if (arg_exec_user) {
1192 const char *home, *shell;
1193 uid_t uid;
1194 gid_t gid;
1195
1196 r = get_user_creds_clean(&arg_exec_user, &uid, &gid, &home, &shell);
1197 if (r < 0)
1198 return log_error_errno(r, "Failed to resolve user %s: %m", arg_exec_user);
1199
1200 if (home) {
1201 r = strv_extendf(&user_env, "HOME=%s", home);
1202 if (r < 0)
1203 return log_oom();
1204 }
1205
1206 if (shell) {
1207 r = strv_extendf(&user_env, "SHELL=%s", shell);
1208 if (r < 0)
1209 return log_oom();
1210 }
1211
1212 r = strv_extendf(&user_env, "USER=%s", arg_exec_user);
1213 if (r < 0)
1214 return log_oom();
1215
1216 r = strv_extendf(&user_env, "LOGNAME=%s", arg_exec_user);
1217 if (r < 0)
1218 return log_oom();
1219
1220 if (!arg_exec_group) {
1221 if (setresgid(gid, gid, gid) < 0)
1222 return log_error_errno(errno, "Failed to change GID to " GID_FMT ": %m", gid);
1223 }
1224
1225 if (setresuid(uid, uid, uid) < 0)
1226 return log_error_errno(errno, "Failed to change UID to " UID_FMT ": %m", uid);
1227 }
1228
1229 env = strv_env_merge(3, environ, user_env, arg_environment);
1230 if (!env)
1231 return log_oom();
1232
1233 r = sd_bus_message_read(reply, "o", &object);
1234 if (r < 0)
1235 return bus_log_parse_error(r);
1236
1237 r = bus_wait_for_jobs_one(w, object, arg_quiet);
1238 if (r < 0)
1239 return r;
1240
1241 if (!arg_quiet)
1242 log_info("Running scope as unit: %s", scope);
1243
1244 execvpe(argv[0], argv, env);
1245
1246 return log_error_errno(errno, "Failed to execute: %m");
1247 }
1248
1249 static int start_transient_timer(
1250 sd_bus *bus,
1251 char **argv) {
1252
1253 _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
1254 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL, *reply = NULL;
1255 _cleanup_(bus_wait_for_jobs_freep) BusWaitForJobs *w = NULL;
1256 _cleanup_free_ char *timer = NULL, *service = NULL;
1257 const char *object = NULL;
1258 int r;
1259
1260 assert(bus);
1261 assert(argv);
1262
1263 r = bus_wait_for_jobs_new(bus, &w);
1264 if (r < 0)
1265 return log_oom();
1266
1267 if (arg_unit) {
1268 switch (unit_name_to_type(arg_unit)) {
1269
1270 case UNIT_SERVICE:
1271 service = strdup(arg_unit);
1272 if (!service)
1273 return log_oom();
1274
1275 r = unit_name_change_suffix(service, ".timer", &timer);
1276 if (r < 0)
1277 return log_error_errno(r, "Failed to change unit suffix: %m");
1278 break;
1279
1280 case UNIT_TIMER:
1281 timer = strdup(arg_unit);
1282 if (!timer)
1283 return log_oom();
1284
1285 r = unit_name_change_suffix(timer, ".service", &service);
1286 if (r < 0)
1287 return log_error_errno(r, "Failed to change unit suffix: %m");
1288 break;
1289
1290 default:
1291 r = unit_name_mangle_with_suffix(arg_unit, UNIT_NAME_NOGLOB, ".service", &service);
1292 if (r < 0)
1293 return log_error_errno(r, "Failed to mangle unit name: %m");
1294
1295 r = unit_name_mangle_with_suffix(arg_unit, UNIT_NAME_NOGLOB, ".timer", &timer);
1296 if (r < 0)
1297 return log_error_errno(r, "Failed to mangle unit name: %m");
1298
1299 break;
1300 }
1301 } else {
1302 r = make_unit_name(bus, UNIT_SERVICE, &service);
1303 if (r < 0)
1304 return r;
1305
1306 r = unit_name_change_suffix(service, ".timer", &timer);
1307 if (r < 0)
1308 return log_error_errno(r, "Failed to change unit suffix: %m");
1309 }
1310
1311 r = sd_bus_message_new_method_call(
1312 bus,
1313 &m,
1314 "org.freedesktop.systemd1",
1315 "/org/freedesktop/systemd1",
1316 "org.freedesktop.systemd1.Manager",
1317 "StartTransientUnit");
1318 if (r < 0)
1319 return bus_log_create_error(r);
1320
1321 r = sd_bus_message_set_allow_interactive_authorization(m, arg_ask_password);
1322 if (r < 0)
1323 return bus_log_create_error(r);
1324
1325 /* Name and Mode */
1326 r = sd_bus_message_append(m, "ss", timer, "fail");
1327 if (r < 0)
1328 return bus_log_create_error(r);
1329
1330 /* Properties */
1331 r = sd_bus_message_open_container(m, 'a', "(sv)");
1332 if (r < 0)
1333 return bus_log_create_error(r);
1334
1335 r = transient_timer_set_properties(m);
1336 if (r < 0)
1337 return bus_log_create_error(r);
1338
1339 r = sd_bus_message_close_container(m);
1340 if (r < 0)
1341 return bus_log_create_error(r);
1342
1343 r = sd_bus_message_open_container(m, 'a', "(sa(sv))");
1344 if (r < 0)
1345 return bus_log_create_error(r);
1346
1347 if (!strv_isempty(argv)) {
1348 r = sd_bus_message_open_container(m, 'r', "sa(sv)");
1349 if (r < 0)
1350 return bus_log_create_error(r);
1351
1352 r = sd_bus_message_append(m, "s", service);
1353 if (r < 0)
1354 return bus_log_create_error(r);
1355
1356 r = sd_bus_message_open_container(m, 'a', "(sv)");
1357 if (r < 0)
1358 return bus_log_create_error(r);
1359
1360 r = transient_service_set_properties(m, argv, NULL);
1361 if (r < 0)
1362 return bus_log_create_error(r);
1363
1364 r = sd_bus_message_close_container(m);
1365 if (r < 0)
1366 return bus_log_create_error(r);
1367
1368 r = sd_bus_message_close_container(m);
1369 if (r < 0)
1370 return bus_log_create_error(r);
1371 }
1372
1373 r = sd_bus_message_close_container(m);
1374 if (r < 0)
1375 return bus_log_create_error(r);
1376
1377 polkit_agent_open_if_enabled();
1378
1379 r = sd_bus_call(bus, m, 0, &error, &reply);
1380 if (r < 0) {
1381 log_error("Failed to start transient timer unit: %s", bus_error_message(&error, -r));
1382 return r;
1383 }
1384
1385 r = sd_bus_message_read(reply, "o", &object);
1386 if (r < 0)
1387 return bus_log_parse_error(r);
1388
1389 r = bus_wait_for_jobs_one(w, object, arg_quiet);
1390 if (r < 0)
1391 return r;
1392
1393 if (!arg_quiet) {
1394 log_info("Running timer as unit: %s", timer);
1395 if (argv[0])
1396 log_info("Will run service as unit: %s", service);
1397 }
1398
1399 return 0;
1400 }
1401
1402 int main(int argc, char* argv[]) {
1403 _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
1404 _cleanup_free_ char *description = NULL, *command = NULL;
1405 int r, retval = EXIT_SUCCESS;
1406
1407 log_parse_environment();
1408 log_open();
1409
1410 r = parse_argv(argc, argv);
1411 if (r <= 0)
1412 goto finish;
1413
1414 if (argc > optind && arg_transport == BUS_TRANSPORT_LOCAL) {
1415 /* Patch in an absolute path */
1416
1417 r = find_binary(argv[optind], &command);
1418 if (r < 0) {
1419 log_error_errno(r, "Failed to find executable %s: %m", argv[optind]);
1420 goto finish;
1421 }
1422
1423 argv[optind] = command;
1424 }
1425
1426 if (!arg_description) {
1427 description = strv_join(argv + optind, " ");
1428 if (!description) {
1429 r = log_oom();
1430 goto finish;
1431 }
1432
1433 if (arg_unit && isempty(description)) {
1434 r = free_and_strdup(&description, arg_unit);
1435 if (r < 0)
1436 goto finish;
1437 }
1438
1439 arg_description = description;
1440 }
1441
1442 /* If --wait is used connect via the bus, unconditionally, as ref/unref is not supported via the limited direct
1443 * connection */
1444 if (arg_wait || arg_pty)
1445 r = bus_connect_transport(arg_transport, arg_host, arg_user, &bus);
1446 else
1447 r = bus_connect_transport_systemd(arg_transport, arg_host, arg_user, &bus);
1448 if (r < 0) {
1449 log_error_errno(r, "Failed to create bus connection: %m");
1450 goto finish;
1451 }
1452
1453 if (arg_scope)
1454 r = start_transient_scope(bus, argv + optind);
1455 else if (with_timer())
1456 r = start_transient_timer(bus, argv + optind);
1457 else
1458 r = start_transient_service(bus, argv + optind, &retval);
1459
1460 finish:
1461 strv_free(arg_environment);
1462 strv_free(arg_property);
1463 strv_free(arg_timer_property);
1464
1465 return r < 0 ? EXIT_FAILURE : retval;
1466 }
Unnamed repository; edit this file 'description' to name the repository.