]> git.ipfire.org Git - thirdparty/systemd.git/blame - src/core/manager.c
util: rework rm_rf() logic
[thirdparty/systemd.git] / src / core / manager.c
CommitLineData
d6c9574f 1/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
60918275 2
a7334b09
LP
3/***
4 This file is part of systemd.
5
6 Copyright 2010 Lennart Poettering
7
8 systemd is free software; you can redistribute it and/or modify it
5430f7f2
LP
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
a7334b09
LP
11 (at your option) any later version.
12
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
5430f7f2 16 Lesser General Public License for more details.
a7334b09 17
5430f7f2 18 You should have received a copy of the GNU Lesser General Public License
a7334b09
LP
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
20***/
21
60918275 22#include <errno.h>
87d1515d 23#include <string.h>
9152c765 24#include <signal.h>
9152c765
LP
25#include <sys/wait.h>
26#include <unistd.h>
e46b13c8
ZJS
27#include <sys/inotify.h>
28#include <sys/epoll.h>
e1414003
LP
29#include <sys/reboot.h>
30#include <sys/ioctl.h>
31#include <linux/kd.h>
80876c20 32#include <fcntl.h>
fe51822e 33#include <dirent.h>
8742514c 34#include <sys/timerfd.h>
830f6caa
LP
35
36#ifdef HAVE_AUDIT
4927fcae 37#include <libaudit.h>
830f6caa 38#endif
60918275 39
718db961 40#include "sd-daemon.h"
718db961 41#include "sd-messages.h"
81527be1 42
60918275 43#include "manager.h"
75778e21 44#include "transaction.h"
60918275
LP
45#include "hashmap.h"
46#include "macro.h"
47#include "strv.h"
16354eff 48#include "log.h"
2a987ee8 49#include "util.h"
49e942b2 50#include "mkdir.h"
ea430986 51#include "ratelimit.h"
e21fea24 52#include "locale-setup.h"
9e2f7c11 53#include "unit-name.h"
1137a57c 54#include "missing.h"
c6878637 55#include "rm-rf.h"
84e3543e 56#include "path-lookup.h"
514f4ef5 57#include "special.h"
d06dacd0 58#include "exit-status.h"
5dc4c17f 59#include "virt.h"
e96d6be7 60#include "watchdog.h"
9eb977db 61#include "path-util.h"
c1165f82 62#include "audit-fd.h"
c51d84dc 63#include "boot-timestamps.h"
8b55b8c4 64#include "env-util.h"
96aad8d1 65#include "bus-common-errors.h"
718db961
LP
66#include "bus-error.h"
67#include "bus-util.h"
68#include "dbus.h"
69#include "dbus-unit.h"
70#include "dbus-job.h"
71#include "dbus-manager.h"
e3dd987c 72#include "bus-kernel.h"
e4746b57 73#include "time-util.h"
60918275 74
03b717a3 75/* Initial delay and the interval for printing status messages about running jobs */
fd08a840
ZJS
76#define JOBS_IN_PROGRESS_WAIT_USEC (5*USEC_PER_SEC)
77#define JOBS_IN_PROGRESS_PERIOD_USEC (USEC_PER_SEC / 3)
03b717a3
MS
78#define JOBS_IN_PROGRESS_PERIOD_DIVISOR 3
79
718db961
LP
80static int manager_dispatch_notify_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);
81static int manager_dispatch_signal_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);
82static int manager_dispatch_time_change_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);
83static int manager_dispatch_idle_pipe_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);
84static int manager_dispatch_jobs_in_progress(sd_event_source *source, usec_t usec, void *userdata);
752b5905 85static int manager_dispatch_run_queue(sd_event_source *source, void *userdata);
e801700e
ZJS
86static int manager_run_generators(Manager *m);
87static void manager_undo_generators(Manager *m);
718db961 88
2ae56591 89static void manager_watch_jobs_in_progress(Manager *m) {
e5723c89
ZJS
90 usec_t next;
91
718db961 92 assert(m);
03b717a3 93
718db961 94 if (m->jobs_in_progress_event_source)
2ae56591 95 return;
03b717a3 96
e5723c89 97 next = now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_WAIT_USEC;
2ae56591 98 (void) sd_event_add_time(
6a0f1f6d
LP
99 m->event,
100 &m->jobs_in_progress_event_source,
101 CLOCK_MONOTONIC,
102 next, 0,
103 manager_dispatch_jobs_in_progress, m);
03b717a3
MS
104}
105
718db961 106#define CYLON_BUFFER_EXTRA (2*(sizeof(ANSI_RED_ON)-1) + sizeof(ANSI_HIGHLIGHT_RED_ON)-1 + 2*(sizeof(ANSI_HIGHLIGHT_OFF)-1))
03b717a3 107
03b717a3
MS
108static void draw_cylon(char buffer[], size_t buflen, unsigned width, unsigned pos) {
109 char *p = buffer;
110
111 assert(buflen >= CYLON_BUFFER_EXTRA + width + 1);
112 assert(pos <= width+1); /* 0 or width+1 mean that the center light is behind the corner */
113
114 if (pos > 1) {
6282c859
MS
115 if (pos > 2)
116 p = mempset(p, ' ', pos-2);
5052495b 117 p = stpcpy(p, ANSI_RED_ON);
03b717a3
MS
118 *p++ = '*';
119 }
120
121 if (pos > 0 && pos <= width) {
5052495b 122 p = stpcpy(p, ANSI_HIGHLIGHT_RED_ON);
03b717a3
MS
123 *p++ = '*';
124 }
125
5052495b 126 p = stpcpy(p, ANSI_HIGHLIGHT_OFF);
03b717a3
MS
127
128 if (pos < width) {
5052495b 129 p = stpcpy(p, ANSI_RED_ON);
03b717a3 130 *p++ = '*';
6282c859
MS
131 if (pos < width-1)
132 p = mempset(p, ' ', width-1-pos);
51d122af 133 strcpy(p, ANSI_HIGHLIGHT_OFF);
03b717a3 134 }
03b717a3
MS
135}
136
cb8ccb22 137void manager_flip_auto_status(Manager *m, bool enable) {
f755e3b7
LP
138 assert(m);
139
cb8ccb22
ZJS
140 if (enable) {
141 if (m->show_status == SHOW_STATUS_AUTO)
142 manager_set_show_status(m, SHOW_STATUS_TEMPORARY);
143 } else {
144 if (m->show_status == SHOW_STATUS_TEMPORARY)
145 manager_set_show_status(m, SHOW_STATUS_AUTO);
146 }
147}
148
03b717a3 149static void manager_print_jobs_in_progress(Manager *m) {
718db961 150 _cleanup_free_ char *job_of_n = NULL;
03b717a3
MS
151 Iterator i;
152 Job *j;
03b717a3
MS
153 unsigned counter = 0, print_nr;
154 char cylon[6 + CYLON_BUFFER_EXTRA + 1];
155 unsigned cylon_pos;
8bb310c3
ZJS
156 char time[FORMAT_TIMESPAN_MAX], limit[FORMAT_TIMESPAN_MAX] = "no limit";
157 uint64_t x;
03b717a3 158
718db961 159 assert(m);
9c3349e2 160 assert(m->n_running_jobs > 0);
718db961 161
cb8ccb22 162 manager_flip_auto_status(m, true);
d450b6f2 163
03b717a3
MS
164 print_nr = (m->jobs_in_progress_iteration / JOBS_IN_PROGRESS_PERIOD_DIVISOR) % m->n_running_jobs;
165
166 HASHMAP_FOREACH(j, m->jobs, i)
167 if (j->state == JOB_RUNNING && counter++ == print_nr)
168 break;
169
e970a72e
MS
170 /* m->n_running_jobs must be consistent with the contents of m->jobs,
171 * so the above loop must have succeeded in finding j. */
172 assert(counter == print_nr + 1);
51d122af 173 assert(j);
5a82a91a 174
03b717a3
MS
175 cylon_pos = m->jobs_in_progress_iteration % 14;
176 if (cylon_pos >= 8)
177 cylon_pos = 14 - cylon_pos;
178 draw_cylon(cylon, sizeof(cylon), 6, cylon_pos);
179
8bb310c3
ZJS
180 m->jobs_in_progress_iteration++;
181
d6483ba7
ZJS
182 if (m->n_running_jobs > 1) {
183 if (asprintf(&job_of_n, "(%u of %u) ", counter, m->n_running_jobs) < 0)
184 job_of_n = NULL;
185 }
03b717a3 186
8bb310c3
ZJS
187 format_timespan(time, sizeof(time), now(CLOCK_MONOTONIC) - j->begin_usec, 1*USEC_PER_SEC);
188 if (job_get_timeout(j, &x) > 0)
189 format_timespan(limit, sizeof(limit), x - j->begin_usec, 1*USEC_PER_SEC);
190
127d5fd1 191 manager_status_printf(m, STATUS_TYPE_EPHEMERAL, cylon,
8bb310c3
ZJS
192 "%sA %s job is running for %s (%s / %s)",
193 strempty(job_of_n),
194 job_type_to_string(j->type),
195 unit_description(j->unit),
196 time, limit);
03b717a3
MS
197}
198
e46b13c8
ZJS
199static int have_ask_password(void) {
200 _cleanup_closedir_ DIR *dir;
201
202 dir = opendir("/run/systemd/ask-password");
203 if (!dir) {
204 if (errno == ENOENT)
205 return false;
206 else
207 return -errno;
208 }
209
210 for (;;) {
211 struct dirent *de;
212
213 errno = 0;
214 de = readdir(dir);
215 if (!de && errno != 0)
216 return -errno;
217 if (!de)
218 return false;
219
220 if (startswith(de->d_name, "ask."))
221 return true;
222 }
223}
224
225static int manager_dispatch_ask_password_fd(sd_event_source *source,
226 int fd, uint32_t revents, void *userdata) {
227 Manager *m = userdata;
228
229 assert(m);
230
231 flush_fd(fd);
232
233 m->have_ask_password = have_ask_password();
234 if (m->have_ask_password < 0)
235 /* Log error but continue. Negative have_ask_password
236 * is treated as unknown status. */
c33b3297 237 log_error_errno(m->have_ask_password, "Failed to list /run/systemd/ask-password: %m");
e46b13c8
ZJS
238
239 return 0;
240}
241
242static void manager_close_ask_password(Manager *m) {
243 assert(m);
244
245 m->ask_password_inotify_fd = safe_close(m->ask_password_inotify_fd);
246 m->ask_password_event_source = sd_event_source_unref(m->ask_password_event_source);
247 m->have_ask_password = -EINVAL;
248}
249
250static int manager_check_ask_password(Manager *m) {
251 int r;
252
253 assert(m);
254
255 if (!m->ask_password_event_source) {
256 assert(m->ask_password_inotify_fd < 0);
257
258 mkdir_p_label("/run/systemd/ask-password", 0755);
259
260 m->ask_password_inotify_fd = inotify_init1(IN_NONBLOCK|IN_CLOEXEC);
4a62c710
MS
261 if (m->ask_password_inotify_fd < 0)
262 return log_error_errno(errno, "inotify_init1() failed: %m");
e46b13c8
ZJS
263
264 if (inotify_add_watch(m->ask_password_inotify_fd, "/run/systemd/ask-password", IN_CREATE|IN_DELETE|IN_MOVE) < 0) {
56f64d95 265 log_error_errno(errno, "Failed to add watch on /run/systemd/ask-password: %m");
e46b13c8
ZJS
266 manager_close_ask_password(m);
267 return -errno;
268 }
269
270 r = sd_event_add_io(m->event, &m->ask_password_event_source,
271 m->ask_password_inotify_fd, EPOLLIN,
272 manager_dispatch_ask_password_fd, m);
273 if (r < 0) {
56f64d95 274 log_error_errno(errno, "Failed to add event source for /run/systemd/ask-password: %m");
e46b13c8
ZJS
275 manager_close_ask_password(m);
276 return -errno;
277 }
278
279 /* Queries might have been added meanwhile... */
280 manager_dispatch_ask_password_fd(m->ask_password_event_source,
281 m->ask_password_inotify_fd, EPOLLIN, m);
282 }
283
284 return m->have_ask_password;
285}
286
31a7eb86 287static int manager_watch_idle_pipe(Manager *m) {
31a7eb86
ZJS
288 int r;
289
718db961
LP
290 assert(m);
291
292 if (m->idle_pipe_event_source)
31a7eb86
ZJS
293 return 0;
294
295 if (m->idle_pipe[2] < 0)
296 return 0;
297
151b9b96 298 r = sd_event_add_io(m->event, &m->idle_pipe_event_source, m->idle_pipe[2], EPOLLIN, manager_dispatch_idle_pipe_fd, m);
23bbb0de
MS
299 if (r < 0)
300 return log_error_errno(r, "Failed to watch idle pipe: %m");
31a7eb86 301
31a7eb86 302 return 0;
31a7eb86
ZJS
303}
304
718db961
LP
305static void manager_close_idle_pipe(Manager *m) {
306 assert(m);
31a7eb86 307
3d94f76c
LP
308 safe_close_pair(m->idle_pipe);
309 safe_close_pair(m->idle_pipe + 2);
31a7eb86
ZJS
310}
311
8742514c 312static int manager_setup_time_change(Manager *m) {
718db961 313 int r;
b92bea5d
ZJS
314
315 /* We only care for the cancellation event, hence we set the
316 * timeout to the latest possible value. */
317 struct itimerspec its = {
318 .it_value.tv_sec = TIME_T_MAX,
319 };
8742514c 320
718db961
LP
321 assert(m);
322 assert_cc(sizeof(time_t) == sizeof(TIME_T_MAX));
8742514c 323
0d8c31ff
ZJS
324 if (m->test_run)
325 return 0;
326
8742514c
LP
327 /* Uses TFD_TIMER_CANCEL_ON_SET to get notifications whenever
328 * CLOCK_REALTIME makes a jump relative to CLOCK_MONOTONIC */
329
718db961 330 m->time_change_fd = timerfd_create(CLOCK_REALTIME, TFD_NONBLOCK|TFD_CLOEXEC);
4a62c710
MS
331 if (m->time_change_fd < 0)
332 return log_error_errno(errno, "Failed to create timerfd: %m");
8742514c 333
718db961 334 if (timerfd_settime(m->time_change_fd, TFD_TIMER_ABSTIME|TFD_TIMER_CANCEL_ON_SET, &its, NULL) < 0) {
56f64d95 335 log_debug_errno(errno, "Failed to set up TFD_TIMER_CANCEL_ON_SET, ignoring: %m");
03e334a1 336 m->time_change_fd = safe_close(m->time_change_fd);
8742514c
LP
337 return 0;
338 }
339
151b9b96 340 r = sd_event_add_io(m->event, &m->time_change_event_source, m->time_change_fd, EPOLLIN, manager_dispatch_time_change_fd, m);
23bbb0de
MS
341 if (r < 0)
342 return log_error_errno(r, "Failed to create time change event source: %m");
8742514c
LP
343
344 log_debug("Set up TFD_TIMER_CANCEL_ON_SET timerfd.");
345
346 return 0;
347}
348
80876c20 349static int enable_special_signals(Manager *m) {
718db961 350 _cleanup_close_ int fd = -1;
80876c20
LP
351
352 assert(m);
353
a41b539e 354 /* Enable that we get SIGINT on control-alt-del. In containers
c9999773
LP
355 * this will fail with EPERM (older) or EINVAL (newer), so
356 * ignore that. */
357 if (reboot(RB_DISABLE_CAD) < 0 && errno != EPERM && errno != EINVAL)
56f64d95 358 log_warning_errno(errno, "Failed to enable ctrl-alt-del handling: %m");
80876c20 359
a41b539e
LP
360 fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
361 if (fd < 0) {
362 /* Support systems without virtual console */
363 if (fd != -ENOENT)
56f64d95 364 log_warning_errno(errno, "Failed to open /dev/tty0: %m");
a41b539e 365 } else {
80876c20
LP
366 /* Enable that we get SIGWINCH on kbrequest */
367 if (ioctl(fd, KDSIGACCEPT, SIGWINCH) < 0)
56f64d95 368 log_warning_errno(errno, "Failed to enable kbrequest handling: %m");
80876c20
LP
369 }
370
371 return 0;
372}
373
ce578209 374static int manager_setup_signals(Manager *m) {
b92bea5d
ZJS
375 struct sigaction sa = {
376 .sa_handler = SIG_DFL,
377 .sa_flags = SA_NOCLDSTOP|SA_RESTART,
378 };
718db961
LP
379 sigset_t mask;
380 int r;
60918275 381
ce578209
LP
382 assert(m);
383
57c0c30e
LP
384 assert_se(sigaction(SIGCHLD, &sa, NULL) == 0);
385
4dffec14
LP
386 /* We make liberal use of realtime signals here. On
387 * Linux/glibc we have 30 of them (with the exception of Linux
388 * on hppa, see below), between SIGRTMIN+0 ... SIGRTMIN+30
389 * (aka SIGRTMAX). */
7d793605 390
4dffec14 391 assert_se(sigemptyset(&mask) == 0);
7d793605
LP
392 sigset_add_many(&mask,
393 SIGCHLD, /* Child died */
394 SIGTERM, /* Reexecute daemon */
395 SIGHUP, /* Reload configuration */
396 SIGUSR1, /* systemd/upstart: reconnect to D-Bus */
397 SIGUSR2, /* systemd: dump status */
398 SIGINT, /* Kernel sends us this on control-alt-del */
399 SIGWINCH, /* Kernel sends us this on kbrequest (alt-arrowup) */
400 SIGPWR, /* Some kernel drivers and upsd send us this on power failure */
4dffec14 401
7d793605 402 SIGRTMIN+0, /* systemd: start default.target */
0003d1ab 403 SIGRTMIN+1, /* systemd: isolate rescue.target */
7d793605
LP
404 SIGRTMIN+2, /* systemd: isolate emergency.target */
405 SIGRTMIN+3, /* systemd: start halt.target */
406 SIGRTMIN+4, /* systemd: start poweroff.target */
407 SIGRTMIN+5, /* systemd: start reboot.target */
0003d1ab 408 SIGRTMIN+6, /* systemd: start kexec.target */
4dffec14
LP
409
410 /* ... space for more special targets ... */
411
0003d1ab
LP
412 SIGRTMIN+13, /* systemd: Immediate halt */
413 SIGRTMIN+14, /* systemd: Immediate poweroff */
414 SIGRTMIN+15, /* systemd: Immediate reboot */
415 SIGRTMIN+16, /* systemd: Immediate kexec */
4dffec14
LP
416
417 /* ... space for more immediate system state changes ... */
418
0658666b
LP
419 SIGRTMIN+20, /* systemd: enable status messages */
420 SIGRTMIN+21, /* systemd: disable status messages */
253ee27a
LP
421 SIGRTMIN+22, /* systemd: set log level to LOG_DEBUG */
422 SIGRTMIN+23, /* systemd: set log level to LOG_INFO */
600b704e 423 SIGRTMIN+24, /* systemd: Immediate exit (--user only) */
4dffec14
LP
424
425 /* .. one free signal here ... */
426
427#if !defined(__hppa64__) && !defined(__hppa__)
428 /* Apparently Linux on hppa has fewer RT
429 * signals (SIGRTMAX is SIGRTMIN+25 there),
430 * hence let's not try to make use of them
431 * here. Since these commands are accessible
432 * by different means and only really a safety
433 * net, the missing functionality on hppa
434 * shouldn't matter. */
435
4cfa2c99 436 SIGRTMIN+26, /* systemd: set log target to journal-or-kmsg */
253ee27a
LP
437 SIGRTMIN+27, /* systemd: set log target to console */
438 SIGRTMIN+28, /* systemd: set log target to kmsg */
ee33e53a 439 SIGRTMIN+29, /* systemd: set log target to syslog-or-kmsg (obsolete) */
4dffec14
LP
440
441 /* ... one free signal here SIGRTMIN+30 ... */
442#endif
7d793605 443 -1);
ce578209
LP
444 assert_se(sigprocmask(SIG_SETMASK, &mask, NULL) == 0);
445
718db961
LP
446 m->signal_fd = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC);
447 if (m->signal_fd < 0)
ce578209
LP
448 return -errno;
449
151b9b96 450 r = sd_event_add_io(m->event, &m->signal_event_source, m->signal_fd, EPOLLIN, manager_dispatch_signal_fd, m);
718db961
LP
451 if (r < 0)
452 return r;
ce578209 453
fa28bc2d 454 /* Process signals a bit earlier than the rest of things, but
46849c3f 455 * later than notify_fd processing, so that the notify
fa28bc2d
LP
456 * processing can still figure out to which process/service a
457 * message belongs, before we reap the process. */
29083707
LP
458 r = sd_event_source_set_priority(m->signal_event_source, -5);
459 if (r < 0)
460 return r;
461
67445f4e 462 if (m->running_as == SYSTEMD_SYSTEM)
80876c20 463 return enable_special_signals(m);
e1414003 464
ce578209
LP
465 return 0;
466}
467
f069efb4
LP
468static void manager_clean_environment(Manager *m) {
469 assert(m);
470
471 /* Let's remove some environment variables that we
472 * need ourselves to communicate with our clients */
473 strv_env_unset_many(
474 m->environment,
475 "NOTIFY_SOCKET",
476 "MAINPID",
477 "MANAGERPID",
478 "LISTEN_PID",
479 "LISTEN_FDS",
480 "WATCHDOG_PID",
481 "WATCHDOG_USEC",
482 NULL);
483}
484
e21fea24 485static int manager_default_environment(Manager *m) {
71ecc858
LP
486 assert(m);
487
e21fea24
KS
488 if (m->running_as == SYSTEMD_SYSTEM) {
489 /* The system manager always starts with a clean
490 * environment for its children. It does not import
491 * the kernel or the parents exported variables.
492 *
493 * The initial passed environ is untouched to keep
494 * /proc/self/environ valid; it is used for tagging
495 * the init process inside containers. */
43638332
ZJS
496 m->environment = strv_new("PATH=" DEFAULT_PATH,
497 NULL);
e21fea24
KS
498
499 /* Import locale variables LC_*= from configuration */
500 locale_setup(&m->environment);
43d03a83 501 } else {
e21fea24
KS
502 /* The user manager passes its own environment
503 * along to its children. */
504 m->environment = strv_copy(environ);
43d03a83
LP
505 }
506
e21fea24
KS
507 if (!m->environment)
508 return -ENOMEM;
8b55b8c4 509
f069efb4 510 manager_clean_environment(m);
9d5a3757
LP
511 strv_sort(m->environment);
512
e21fea24 513 return 0;
71ecc858
LP
514}
515
0d8c31ff 516int manager_new(SystemdRunningAs running_as, bool test_run, Manager **_m) {
ce578209 517 Manager *m;
e3dd987c 518 int r;
8e274523
LP
519
520 assert(_m);
a5dab5ce 521 assert(running_as >= 0);
67445f4e 522 assert(running_as < _SYSTEMD_RUNNING_AS_MAX);
ce578209 523
915b3753
LP
524 m = new0(Manager, 1);
525 if (!m)
8e274523 526 return -ENOMEM;
60918275 527
4f8d551f 528#ifdef ENABLE_EFI
530a9662 529 if (running_as == SYSTEMD_SYSTEM && detect_container(NULL) <= 0)
c51d84dc 530 boot_timestamps(&m->userspace_timestamp, &m->firmware_timestamp, &m->loader_timestamp);
4f8d551f
ZC
531#endif
532
a5dab5ce 533 m->running_as = running_as;
a16e1123 534 m->exit_code = _MANAGER_EXIT_CODE_INVALID;
bd8f585b 535 m->default_timer_accuracy_usec = USEC_PER_MINUTE;
80876c20 536
718db961 537 m->idle_pipe[0] = m->idle_pipe[1] = m->idle_pipe[2] = m->idle_pipe[3] = -1;
8742514c 538
befb6d54 539 m->pin_cgroupfs_fd = m->notify_fd = m->signal_fd = m->time_change_fd = m->dev_autofs_fd = m->private_listen_fd = m->kdbus_fd = m->utab_inotify_fd = -1;
ea430986 540 m->current_job_id = 1; /* start as id #1, so that we can leave #0 around as "null-like" value */
9152c765 541
e46b13c8
ZJS
542 m->ask_password_inotify_fd = -1;
543 m->have_ask_password = -EINVAL; /* we don't know */
544
0d8c31ff
ZJS
545 m->test_run = test_run;
546
2e5c94b9
LP
547 /* Reboot immediately if the user hits C-A-D more often than 7x per 2s */
548 RATELIMIT_INIT(m->ctrl_alt_del_ratelimit, 2 * USEC_PER_SEC, 7);
549
e21fea24
KS
550 r = manager_default_environment(m);
551 if (r < 0)
1137a57c
LP
552 goto fail;
553
d5099efc 554 r = hashmap_ensure_allocated(&m->units, &string_hash_ops);
718db961 555 if (r < 0)
60918275
LP
556 goto fail;
557
d5099efc 558 r = hashmap_ensure_allocated(&m->jobs, NULL);
718db961 559 if (r < 0)
60918275
LP
560 goto fail;
561
d5099efc 562 r = hashmap_ensure_allocated(&m->cgroup_unit, &string_hash_ops);
718db961 563 if (r < 0)
9152c765
LP
564 goto fail;
565
d5099efc 566 r = hashmap_ensure_allocated(&m->watch_bus, &string_hash_ops);
718db961 567 if (r < 0)
05e343b7
LP
568 goto fail;
569
d5099efc 570 r = set_ensure_allocated(&m->startup_units, NULL);
95ae05c0
WC
571 if (r < 0)
572 goto fail;
573
d5099efc 574 r = set_ensure_allocated(&m->failed_units, NULL);
f755e3b7
LP
575 if (r < 0)
576 goto fail;
577
718db961
LP
578 r = sd_event_default(&m->event);
579 if (r < 0)
8742514c
LP
580 goto fail;
581
151b9b96 582 r = sd_event_add_defer(m->event, &m->run_queue_event_source, manager_dispatch_run_queue, m);
752b5905
LP
583 if (r < 0)
584 goto fail;
585
586 r = sd_event_source_set_priority(m->run_queue_event_source, SD_EVENT_PRIORITY_IDLE);
587 if (r < 0)
588 goto fail;
589
590 r = sd_event_source_set_enabled(m->run_queue_event_source, SD_EVENT_OFF);
591 if (r < 0)
592 goto fail;
593
8742514c
LP
594 r = manager_setup_signals(m);
595 if (r < 0)
9152c765
LP
596 goto fail;
597
8742514c
LP
598 r = manager_setup_cgroup(m);
599 if (r < 0)
8e274523
LP
600 goto fail;
601
8742514c
LP
602 r = manager_setup_time_change(m);
603 if (r < 0)
8c47c732
LP
604 goto fail;
605
9670d583
LP
606 m->udev = udev_new();
607 if (!m->udev) {
608 r = -ENOMEM;
609 goto fail;
610 }
611
d86f9d52
LP
612 /* Note that we set up neither kdbus, nor the notify fd
613 * here. We do that after deserialization, since they might
614 * have gotten serialized across the reexec. */
615
72bc8d00
LP
616 m->taint_usr = dir_is_empty("/usr") > 0;
617
8e274523
LP
618 *_m = m;
619 return 0;
60918275
LP
620
621fail:
622 manager_free(m);
8e274523 623 return r;
60918275
LP
624}
625
d86f9d52 626static int manager_setup_notify(Manager *m) {
7181dbdb 627 int r;
d86f9d52 628
0d8c31ff
ZJS
629 if (m->test_run)
630 return 0;
631
d86f9d52
LP
632 if (m->notify_fd < 0) {
633 _cleanup_close_ int fd = -1;
55836941 634 union sockaddr_union sa = {
7181dbdb
LP
635 .sa.sa_family = AF_UNIX,
636 };
55836941 637 static const int one = 1;
d86f9d52
LP
638
639 /* First free all secondary fields */
640 free(m->notify_socket);
641 m->notify_socket = NULL;
642 m->notify_event_source = sd_event_source_unref(m->notify_event_source);
643
644 fd = socket(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC|SOCK_NONBLOCK, 0);
4a62c710
MS
645 if (fd < 0)
646 return log_error_errno(errno, "Failed to allocate notification socket: %m");
d86f9d52 647
498e87d6 648 if (m->running_as == SYSTEMD_SYSTEM)
7181dbdb 649 m->notify_socket = strdup("/run/systemd/notify");
498e87d6 650 else {
7181dbdb 651 const char *e;
d86f9d52 652
7181dbdb
LP
653 e = getenv("XDG_RUNTIME_DIR");
654 if (!e) {
56f64d95 655 log_error_errno(errno, "XDG_RUNTIME_DIR is not set: %m");
7181dbdb
LP
656 return -EINVAL;
657 }
658
659 m->notify_socket = strappend(e, "/systemd/notify");
660 }
498e87d6
LP
661 if (!m->notify_socket)
662 return log_oom();
663
664 (void) mkdir_parents_label(m->notify_socket, 0755);
f0e62e89 665 (void) unlink(m->notify_socket);
7181dbdb
LP
666
667 strncpy(sa.un.sun_path, m->notify_socket, sizeof(sa.un.sun_path)-1);
668 r = bind(fd, &sa.sa, offsetof(struct sockaddr_un, sun_path) + strlen(sa.un.sun_path));
4a62c710
MS
669 if (r < 0)
670 return log_error_errno(errno, "bind(%s) failed: %m", sa.un.sun_path);
d86f9d52
LP
671
672 r = setsockopt(fd, SOL_SOCKET, SO_PASSCRED, &one, sizeof(one));
4a62c710
MS
673 if (r < 0)
674 return log_error_errno(errno, "SO_PASSCRED failed: %m");
d86f9d52 675
d86f9d52
LP
676 m->notify_fd = fd;
677 fd = -1;
678
679 log_debug("Using notification socket %s", m->notify_socket);
680 }
681
682 if (!m->notify_event_source) {
151b9b96 683 r = sd_event_add_io(m->event, &m->notify_event_source, m->notify_fd, EPOLLIN, manager_dispatch_notify_fd, m);
895b3a7b
MS
684 if (r < 0)
685 return log_error_errno(r, "Failed to allocate notify event source: %m");
d86f9d52
LP
686
687 /* Process signals a bit earlier than SIGCHLD, so that we can
688 * still identify to which service an exit message belongs */
689 r = sd_event_source_set_priority(m->notify_event_source, -7);
23bbb0de
MS
690 if (r < 0)
691 return log_error_errno(r, "Failed to set priority of notify event source: %m");
d86f9d52
LP
692 }
693
694 return 0;
695}
696
697static int manager_setup_kdbus(Manager *m) {
5bf348d7 698#ifdef ENABLE_KDBUS
d86f9d52
LP
699 _cleanup_free_ char *p = NULL;
700
701 assert(m);
702
0d8c31ff 703 if (m->test_run || m->kdbus_fd >= 0)
d86f9d52
LP
704 return 0;
705
90b99192 706 if (m->running_as == SYSTEMD_SYSTEM && detect_container(NULL) <= 0)
387eba00 707 bus_kernel_fix_attach_mask();
1a299299
LP
708
709 m->kdbus_fd = bus_kernel_create_bus(
710 m->running_as == SYSTEMD_SYSTEM ? "system" : "user",
711 m->running_as == SYSTEMD_SYSTEM, &p);
712
eb56eb9b
MS
713 if (m->kdbus_fd < 0)
714 return log_debug_errno(m->kdbus_fd, "Failed to set up kdbus: %m");
d86f9d52
LP
715
716 log_debug("Successfully set up kdbus on %s", p);
d86f9d52
LP
717#endif
718
719 return 0;
720}
721
722static int manager_connect_bus(Manager *m, bool reexecuting) {
723 bool try_bus_connect;
724
725 assert(m);
726
0d8c31ff
ZJS
727 if (m->test_run)
728 return 0;
729
d86f9d52
LP
730 try_bus_connect =
731 m->kdbus_fd >= 0 ||
732 reexecuting ||
733 (m->running_as == SYSTEMD_USER && getenv("DBUS_SESSION_BUS_ADDRESS"));
734
735 /* Try to connect to the busses, if possible. */
736 return bus_init(m, try_bus_connect);
737}
738
23a177ef 739static unsigned manager_dispatch_cleanup_queue(Manager *m) {
595ed347 740 Unit *u;
23a177ef
LP
741 unsigned n = 0;
742
743 assert(m);
744
595ed347
MS
745 while ((u = m->cleanup_queue)) {
746 assert(u->in_cleanup_queue);
23a177ef 747
595ed347 748 unit_free(u);
23a177ef
LP
749 n++;
750 }
751
752 return n;
753}
754
eced69b3 755enum {
35b8ca3a 756 GC_OFFSET_IN_PATH, /* This one is on the path we were traveling */
eced69b3
LP
757 GC_OFFSET_UNSURE, /* No clue */
758 GC_OFFSET_GOOD, /* We still need this unit */
759 GC_OFFSET_BAD, /* We don't need this unit anymore */
760 _GC_OFFSET_MAX
761};
762
763static void unit_gc_sweep(Unit *u, unsigned gc_marker) {
701cc384
LP
764 Iterator i;
765 Unit *other;
eced69b3 766 bool is_bad;
701cc384
LP
767
768 assert(u);
769
ac155bb8
MS
770 if (u->gc_marker == gc_marker + GC_OFFSET_GOOD ||
771 u->gc_marker == gc_marker + GC_OFFSET_BAD ||
772 u->gc_marker == gc_marker + GC_OFFSET_IN_PATH)
701cc384
LP
773 return;
774
ac155bb8 775 if (u->in_cleanup_queue)
701cc384
LP
776 goto bad;
777
778 if (unit_check_gc(u))
779 goto good;
780
ac155bb8 781 u->gc_marker = gc_marker + GC_OFFSET_IN_PATH;
eced69b3
LP
782
783 is_bad = true;
784
ac155bb8 785 SET_FOREACH(other, u->dependencies[UNIT_REFERENCED_BY], i) {
701cc384
LP
786 unit_gc_sweep(other, gc_marker);
787
ac155bb8 788 if (other->gc_marker == gc_marker + GC_OFFSET_GOOD)
701cc384 789 goto good;
eced69b3 790
ac155bb8 791 if (other->gc_marker != gc_marker + GC_OFFSET_BAD)
eced69b3 792 is_bad = false;
701cc384
LP
793 }
794
eced69b3
LP
795 if (is_bad)
796 goto bad;
797
798 /* We were unable to find anything out about this entry, so
799 * let's investigate it later */
ac155bb8 800 u->gc_marker = gc_marker + GC_OFFSET_UNSURE;
eced69b3
LP
801 unit_add_to_gc_queue(u);
802 return;
803
701cc384 804bad:
eced69b3
LP
805 /* We definitely know that this one is not useful anymore, so
806 * let's mark it for deletion */
ac155bb8 807 u->gc_marker = gc_marker + GC_OFFSET_BAD;
eced69b3 808 unit_add_to_cleanup_queue(u);
701cc384
LP
809 return;
810
811good:
ac155bb8 812 u->gc_marker = gc_marker + GC_OFFSET_GOOD;
701cc384
LP
813}
814
815static unsigned manager_dispatch_gc_queue(Manager *m) {
595ed347 816 Unit *u;
701cc384 817 unsigned n = 0;
eced69b3 818 unsigned gc_marker;
701cc384
LP
819
820 assert(m);
821
cf1265e1 822 /* log_debug("Running GC..."); */
701cc384 823
eced69b3
LP
824 m->gc_marker += _GC_OFFSET_MAX;
825 if (m->gc_marker + _GC_OFFSET_MAX <= _GC_OFFSET_MAX)
c9c0cadb 826 m->gc_marker = 1;
701cc384 827
eced69b3
LP
828 gc_marker = m->gc_marker;
829
595ed347
MS
830 while ((u = m->gc_queue)) {
831 assert(u->in_gc_queue);
701cc384 832
595ed347 833 unit_gc_sweep(u, gc_marker);
eced69b3 834
71fda00f 835 LIST_REMOVE(gc_queue, m->gc_queue, u);
595ed347 836 u->in_gc_queue = false;
701cc384
LP
837
838 n++;
839
595ed347
MS
840 if (u->gc_marker == gc_marker + GC_OFFSET_BAD ||
841 u->gc_marker == gc_marker + GC_OFFSET_UNSURE) {
cc3bc3e6
ZJS
842 if (u->id)
843 log_unit_debug(u->id, "Collecting %s", u->id);
595ed347
MS
844 u->gc_marker = gc_marker + GC_OFFSET_BAD;
845 unit_add_to_cleanup_queue(u);
701cc384
LP
846 }
847 }
848
849 m->n_in_gc_queue = 0;
701cc384
LP
850
851 return n;
852}
853
a16e1123 854static void manager_clear_jobs_and_units(Manager *m) {
a16e1123 855 Unit *u;
60918275
LP
856
857 assert(m);
858
87f0e418
LP
859 while ((u = hashmap_first(m->units)))
860 unit_free(u);
964e0949
LP
861
862 manager_dispatch_cleanup_queue(m);
863
864 assert(!m->load_queue);
865 assert(!m->run_queue);
866 assert(!m->dbus_unit_queue);
867 assert(!m->dbus_job_queue);
868 assert(!m->cleanup_queue);
869 assert(!m->gc_queue);
870
964e0949
LP
871 assert(hashmap_isempty(m->jobs));
872 assert(hashmap_isempty(m->units));
9e9e2b72
MS
873
874 m->n_on_console = 0;
875 m->n_running_jobs = 0;
a16e1123
LP
876}
877
06d8d842 878Manager* manager_free(Manager *m) {
a16e1123 879 UnitType c;
c93ff2e9 880 int i;
87f0e418 881
06d8d842
ZJS
882 if (!m)
883 return NULL;
a16e1123
LP
884
885 manager_clear_jobs_and_units(m);
23a177ef 886
7824bbeb
LP
887 for (c = 0; c < _UNIT_TYPE_MAX; c++)
888 if (unit_vtable[c]->shutdown)
889 unit_vtable[c]->shutdown(m);
890
a16e1123
LP
891 /* If we reexecute ourselves, we keep the root cgroup
892 * around */
c6c18be3 893 manager_shutdown_cgroup(m, m->exit_code != MANAGER_REEXECUTE);
8e274523 894
5a1e9937
LP
895 manager_undo_generators(m);
896
5e8d1c9a 897 bus_done(m);
ea430986 898
87f0e418 899 hashmap_free(m->units);
60918275 900 hashmap_free(m->jobs);
5ba6985b
LP
901 hashmap_free(m->watch_pids1);
902 hashmap_free(m->watch_pids2);
05e343b7 903 hashmap_free(m->watch_bus);
9152c765 904
95ae05c0 905 set_free(m->startup_units);
f755e3b7
LP
906 set_free(m->failed_units);
907
718db961
LP
908 sd_event_source_unref(m->signal_event_source);
909 sd_event_source_unref(m->notify_event_source);
910 sd_event_source_unref(m->time_change_event_source);
911 sd_event_source_unref(m->jobs_in_progress_event_source);
912 sd_event_source_unref(m->idle_pipe_event_source);
752b5905 913 sd_event_source_unref(m->run_queue_event_source);
718db961 914
03e334a1
LP
915 safe_close(m->signal_fd);
916 safe_close(m->notify_fd);
917 safe_close(m->time_change_fd);
918 safe_close(m->kdbus_fd);
718db961 919
e46b13c8
ZJS
920 manager_close_ask_password(m);
921
718db961
LP
922 manager_close_idle_pipe(m);
923
9670d583 924 udev_unref(m->udev);
718db961 925 sd_event_unref(m->event);
60918275 926
c952c6ec
LP
927 free(m->notify_socket);
928
84e3543e 929 lookup_paths_free(&m->lookup_paths);
1137a57c 930 strv_free(m->environment);
036643a2 931
4ad49000 932 hashmap_free(m->cgroup_unit);
c6c18be3 933 set_free_free(m->unit_path_cache);
33be102a 934
664f88a7
LP
935 free(m->switch_root);
936 free(m->switch_root_init);
937
517d56b1 938 for (i = 0; i < _RLIMIT_MAX; i++)
c93ff2e9
FC
939 free(m->rlimit[i]);
940
a57f7e2c
LP
941 assert(hashmap_isempty(m->units_requiring_mounts_for));
942 hashmap_free(m->units_requiring_mounts_for);
943
60918275 944 free(m);
06d8d842 945 return NULL;
60918275
LP
946}
947
a16e1123 948int manager_enumerate(Manager *m) {
0faacd47 949 int r = 0;
f50e0a01 950 UnitType c;
f50e0a01
LP
951
952 assert(m);
953
a16e1123
LP
954 /* Let's ask every type to load all units from disk/kernel
955 * that it might know */
0faacd47
LP
956 for (c = 0; c < _UNIT_TYPE_MAX; c++) {
957 int q;
958
959 if (unit_vtable[c]->supported && !unit_vtable[c]->supported(m)) {
03afec3c 960 log_debug("Unit type .%s is not supported on this system.", unit_type_to_string(c));
0faacd47 961 continue;
a57f7e2c 962 }
f50e0a01 963
0faacd47
LP
964 if (!unit_vtable[c]->enumerate)
965 continue;
966
967 q = unit_vtable[c]->enumerate(m);
968 if (q < 0)
969 r = q;
970 }
971
f50e0a01 972 manager_dispatch_load_queue(m);
a16e1123
LP
973 return r;
974}
975
9588bc32 976static int manager_coldplug(Manager *m) {
8f8f05a9 977 int r = 0;
a16e1123
LP
978 Iterator i;
979 Unit *u;
980 char *k;
981
6e392c9c
IS
982 /*
983 * Some unit types tend to spawn jobs or check other units' state
984 * during coldplug. This is wrong because it is undefined whether the
985 * units in question have been already coldplugged (i. e. their state
986 * restored). This way, we can easily re-start an already started unit
987 * or otherwise make a wrong decision based on the unit's state.
988 *
989 * Solve this by providing a way for coldplug functions to defer
990 * such actions until after all units have been coldplugged.
991 *
992 * We store Unit* -> int(*)(Unit*).
993 *
994 * https://bugs.freedesktop.org/show_bug.cgi?id=88401
995 */
996 _cleanup_hashmap_free_ Hashmap *deferred_work = NULL;
997 int(*proc)(Unit*);
998
a16e1123 999 assert(m);
f50e0a01 1000
6e392c9c
IS
1001 deferred_work = hashmap_new(&trivial_hash_ops);
1002 if (!deferred_work)
1003 return -ENOMEM;
1004
f50e0a01
LP
1005 /* Then, let's set up their initial state. */
1006 HASHMAP_FOREACH_KEY(u, k, m->units, i) {
8f8f05a9 1007 int q;
f50e0a01
LP
1008
1009 /* ignore aliases */
ac155bb8 1010 if (u->id != k)
f50e0a01
LP
1011 continue;
1012
6e392c9c
IS
1013 q = unit_coldplug(u, deferred_work);
1014 if (q < 0)
1015 r = q;
1016 }
1017
1018 /* After coldplugging and setting up initial state of the units,
1019 * let's perform operations which spawn jobs or query units' state. */
1020 HASHMAP_FOREACH_KEY(proc, u, deferred_work, i) {
1021 int q;
1022
1023 q = proc(u);
8f8f05a9 1024 if (q < 0)
cca098b0 1025 r = q;
f50e0a01
LP
1026 }
1027
a16e1123
LP
1028 return r;
1029}
1030
fe51822e
LP
1031static void manager_build_unit_path_cache(Manager *m) {
1032 char **i;
807d0cca 1033 _cleanup_closedir_ DIR *d = NULL;
fe51822e
LP
1034 int r;
1035
1036 assert(m);
1037
1038 set_free_free(m->unit_path_cache);
1039
d5099efc 1040 m->unit_path_cache = set_new(&string_hash_ops);
874310b7 1041 if (!m->unit_path_cache) {
fe51822e
LP
1042 log_error("Failed to allocate unit path cache.");
1043 return;
1044 }
1045
1046 /* This simply builds a list of files we know exist, so that
1047 * we don't always have to go to disk */
1048
1049 STRV_FOREACH(i, m->lookup_paths.unit_path) {
1050 struct dirent *de;
1051
bd0af849
ZJS
1052 d = opendir(*i);
1053 if (!d) {
874310b7 1054 if (errno != ENOENT)
56f64d95 1055 log_error_errno(errno, "Failed to open directory %s: %m", *i);
fe51822e
LP
1056 continue;
1057 }
1058
1059 while ((de = readdir(d))) {
1060 char *p;
1061
a34bf9db 1062 if (hidden_file(de->d_name))
fe51822e
LP
1063 continue;
1064
b7def684 1065 p = strjoin(streq(*i, "/") ? "" : *i, "/", de->d_name, NULL);
44d91056 1066 if (!p) {
fe51822e
LP
1067 r = -ENOMEM;
1068 goto fail;
1069 }
1070
ef42202a
ZJS
1071 r = set_consume(m->unit_path_cache, p);
1072 if (r < 0)
fe51822e 1073 goto fail;
fe51822e
LP
1074 }
1075
1076 closedir(d);
1077 d = NULL;
1078 }
1079
1080 return;
1081
1082fail:
da927ba9 1083 log_error_errno(r, "Failed to build unit path cache: %m");
fe51822e
LP
1084
1085 set_free_free(m->unit_path_cache);
1086 m->unit_path_cache = NULL;
fe51822e
LP
1087}
1088
9588bc32
LP
1089
1090static int manager_distribute_fds(Manager *m, FDSet *fds) {
1091 Unit *u;
1092 Iterator i;
1093 int r;
1094
1095 assert(m);
1096
1097 HASHMAP_FOREACH(u, m->units, i) {
1098
1099 if (fdset_size(fds) <= 0)
1100 break;
1101
1102 if (UNIT_VTABLE(u)->distribute_fds) {
1103 r = UNIT_VTABLE(u)->distribute_fds(u, fds);
1104 if (r < 0)
1105 return r;
1106 }
1107 }
1108
1109 return 0;
1110}
1111
a16e1123
LP
1112int manager_startup(Manager *m, FILE *serialization, FDSet *fds) {
1113 int r, q;
1114
1115 assert(m);
1116
518d10e9 1117 dual_timestamp_get(&m->generators_start_timestamp);
e801700e 1118 r = manager_run_generators(m);
518d10e9 1119 dual_timestamp_get(&m->generators_finish_timestamp);
e801700e
ZJS
1120 if (r < 0)
1121 return r;
5a1e9937 1122
07719a21
LP
1123 r = lookup_paths_init(
1124 &m->lookup_paths, m->running_as, true,
12ed81d9 1125 NULL,
07719a21
LP
1126 m->generator_unit_path,
1127 m->generator_unit_path_early,
1128 m->generator_unit_path_late);
1129 if (r < 0)
1130 return r;
1131
fe51822e
LP
1132 manager_build_unit_path_cache(m);
1133
9f611ad8
LP
1134 /* If we will deserialize make sure that during enumeration
1135 * this is already known, so we increase the counter here
1136 * already */
1137 if (serialization)
a7556052 1138 m->n_reloading ++;
9f611ad8 1139
a16e1123 1140 /* First, enumerate what we can from all config files */
718db961 1141 dual_timestamp_get(&m->units_load_start_timestamp);
a16e1123 1142 r = manager_enumerate(m);
718db961 1143 dual_timestamp_get(&m->units_load_finish_timestamp);
a16e1123
LP
1144
1145 /* Second, deserialize if there is something to deserialize */
1cd974ed
ZJS
1146 if (serialization)
1147 r = manager_deserialize(m, serialization, fds);
a16e1123 1148
01e10de3
LP
1149 /* Any fds left? Find some unit which wants them. This is
1150 * useful to allow container managers to pass some file
1151 * descriptors to us pre-initialized. This enables
1152 * socket-based activation of entire containers. */
1153 if (fdset_size(fds) > 0) {
1154 q = manager_distribute_fds(m, fds);
1cd974ed 1155 if (q < 0 && r == 0)
01e10de3
LP
1156 r = q;
1157 }
1158
d86f9d52
LP
1159 /* We might have deserialized the notify fd, but if we didn't
1160 * then let's create the bus now */
1cd974ed
ZJS
1161 q = manager_setup_notify(m);
1162 if (q < 0 && r == 0)
1163 r = q;
d86f9d52 1164
e3dd987c
LP
1165 /* We might have deserialized the kdbus control fd, but if we
1166 * didn't, then let's create the bus now. */
1167 manager_setup_kdbus(m);
1168 manager_connect_bus(m, !!serialization);
8f8f05a9 1169 bus_track_coldplug(m, &m->subscribed, &m->deserialized_subscribed);
e3dd987c 1170
a16e1123 1171 /* Third, fire things up! */
07719a21 1172 q = manager_coldplug(m);
1cd974ed 1173 if (q < 0 && r == 0)
a16e1123
LP
1174 r = q;
1175
9f611ad8 1176 if (serialization) {
a7556052
LP
1177 assert(m->n_reloading > 0);
1178 m->n_reloading --;
71445ae7
LP
1179
1180 /* Let's wait for the UnitNew/JobNew messages being
1181 * sent, before we notify that the reload is
1182 * finished */
1183 m->send_reloading_done = true;
9f611ad8
LP
1184 }
1185
a16e1123 1186 return r;
f50e0a01
LP
1187}
1188
718db961 1189int manager_add_job(Manager *m, JobType type, Unit *unit, JobMode mode, bool override, sd_bus_error *e, Job **_ret) {
e5b5ae50 1190 int r;
7527cb52 1191 Transaction *tr;
e5b5ae50
LP
1192
1193 assert(m);
1194 assert(type < _JOB_TYPE_MAX);
87f0e418 1195 assert(unit);
e5b5ae50 1196 assert(mode < _JOB_MODE_MAX);
60918275 1197
7358dc02
ZJS
1198 if (mode == JOB_ISOLATE && type != JOB_START)
1199 return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Isolate is only valid for start.");
c497c7a9 1200
7358dc02
ZJS
1201 if (mode == JOB_ISOLATE && !unit->allow_isolate)
1202 return sd_bus_error_setf(e, BUS_ERROR_NO_ISOLATION, "Operation refused, unit may not be isolated.");
2528a7a6 1203
79008bdd 1204 log_unit_debug(unit->id,
66870f90
ZJS
1205 "Trying to enqueue job %s/%s/%s", unit->id,
1206 job_type_to_string(type), job_mode_to_string(mode));
9f04bd52 1207
e0209d83
MS
1208 job_type_collapse(&type, unit);
1209
23ade460 1210 tr = transaction_new(mode == JOB_REPLACE_IRREVERSIBLY);
7527cb52
MS
1211 if (!tr)
1212 return -ENOMEM;
11dd41ce 1213
7527cb52
MS
1214 r = transaction_add_job_and_dependencies(tr, type, unit, NULL, true, override, false,
1215 mode == JOB_IGNORE_DEPENDENCIES || mode == JOB_IGNORE_REQUIREMENTS,
b94fbd30 1216 mode == JOB_IGNORE_DEPENDENCIES, e);
7527cb52
MS
1217 if (r < 0)
1218 goto tr_abort;
c497c7a9 1219
7527cb52
MS
1220 if (mode == JOB_ISOLATE) {
1221 r = transaction_add_isolate_jobs(tr, m);
1222 if (r < 0)
1223 goto tr_abort;
1224 }
1225
1226 r = transaction_activate(tr, m, mode, e);
1227 if (r < 0)
1228 goto tr_abort;
e5b5ae50 1229
79008bdd 1230 log_unit_debug(unit->id,
66870f90
ZJS
1231 "Enqueued job %s/%s as %u", unit->id,
1232 job_type_to_string(type), (unsigned) tr->anchor_job->id);
f50e0a01 1233
e5b5ae50 1234 if (_ret)
b94fbd30 1235 *_ret = tr->anchor_job;
60918275 1236
7527cb52 1237 transaction_free(tr);
e5b5ae50 1238 return 0;
7527cb52
MS
1239
1240tr_abort:
1241 transaction_abort(tr);
1242 transaction_free(tr);
1243 return r;
e5b5ae50 1244}
60918275 1245
718db961 1246int manager_add_job_by_name(Manager *m, JobType type, const char *name, JobMode mode, bool override, sd_bus_error *e, Job **_ret) {
28247076
LP
1247 Unit *unit;
1248 int r;
1249
1250 assert(m);
1251 assert(type < _JOB_TYPE_MAX);
1252 assert(name);
1253 assert(mode < _JOB_MODE_MAX);
1254
c3090674
LP
1255 r = manager_load_unit(m, name, NULL, NULL, &unit);
1256 if (r < 0)
28247076
LP
1257 return r;
1258
398ef8ba 1259 return manager_add_job(m, type, unit, mode, override, e, _ret);
28247076
LP
1260}
1261
60918275
LP
1262Job *manager_get_job(Manager *m, uint32_t id) {
1263 assert(m);
1264
1265 return hashmap_get(m->jobs, UINT32_TO_PTR(id));
1266}
1267
87f0e418 1268Unit *manager_get_unit(Manager *m, const char *name) {
60918275
LP
1269 assert(m);
1270 assert(name);
1271
87f0e418 1272 return hashmap_get(m->units, name);
60918275
LP
1273}
1274
c1e1601e 1275unsigned manager_dispatch_load_queue(Manager *m) {
595ed347 1276 Unit *u;
c1e1601e 1277 unsigned n = 0;
60918275
LP
1278
1279 assert(m);
1280
223dabab
LP
1281 /* Make sure we are not run recursively */
1282 if (m->dispatching_load_queue)
c1e1601e 1283 return 0;
223dabab
LP
1284
1285 m->dispatching_load_queue = true;
1286
87f0e418 1287 /* Dispatches the load queue. Takes a unit from the queue and
60918275
LP
1288 * tries to load its data until the queue is empty */
1289
595ed347
MS
1290 while ((u = m->load_queue)) {
1291 assert(u->in_load_queue);
034c6ed7 1292
595ed347 1293 unit_load(u);
c1e1601e 1294 n++;
60918275
LP
1295 }
1296
223dabab 1297 m->dispatching_load_queue = false;
c1e1601e 1298 return n;
60918275
LP
1299}
1300
c2756a68
LP
1301int manager_load_unit_prepare(
1302 Manager *m,
1303 const char *name,
1304 const char *path,
718db961 1305 sd_bus_error *e,
c2756a68
LP
1306 Unit **_ret) {
1307
87f0e418 1308 Unit *ret;
7d17cfbc 1309 UnitType t;
60918275
LP
1310 int r;
1311
1312 assert(m);
9e2f7c11 1313 assert(name || path);
60918275 1314
db06e3b6
LP
1315 /* This will prepare the unit for loading, but not actually
1316 * load anything from disk. */
0301abf4 1317
718db961
LP
1318 if (path && !is_path(path))
1319 return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Path %s is not absolute.", path);
9e2f7c11
LP
1320
1321 if (!name)
2b6bf07d 1322 name = basename(path);
9e2f7c11 1323
7d17cfbc
MS
1324 t = unit_name_to_type(name);
1325
f78e6385 1326 if (t == _UNIT_TYPE_INVALID || !unit_name_is_valid(name, TEMPLATE_INVALID))
718db961 1327 return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Unit name %s is not valid.", name);
60918275 1328
7d17cfbc
MS
1329 ret = manager_get_unit(m, name);
1330 if (ret) {
034c6ed7 1331 *_ret = ret;
413d6313 1332 return 1;
034c6ed7 1333 }
60918275 1334
7d17cfbc
MS
1335 ret = unit_new(m, unit_vtable[t]->object_size);
1336 if (!ret)
60918275
LP
1337 return -ENOMEM;
1338
7d17cfbc 1339 if (path) {
ac155bb8
MS
1340 ret->fragment_path = strdup(path);
1341 if (!ret->fragment_path) {
0301abf4
LP
1342 unit_free(ret);
1343 return -ENOMEM;
1344 }
7d17cfbc 1345 }
0301abf4 1346
1058cbf2
ZJS
1347 r = unit_add_name(ret, name);
1348 if (r < 0) {
87f0e418 1349 unit_free(ret);
1ffba6fe 1350 return r;
60918275
LP
1351 }
1352
87f0e418 1353 unit_add_to_load_queue(ret);
c1e1601e 1354 unit_add_to_dbus_queue(ret);
949061f0 1355 unit_add_to_gc_queue(ret);
c1e1601e 1356
db06e3b6
LP
1357 if (_ret)
1358 *_ret = ret;
1359
1360 return 0;
1361}
1362
c2756a68
LP
1363int manager_load_unit(
1364 Manager *m,
1365 const char *name,
1366 const char *path,
718db961 1367 sd_bus_error *e,
c2756a68
LP
1368 Unit **_ret) {
1369
db06e3b6
LP
1370 int r;
1371
1372 assert(m);
1373
1374 /* This will load the service information files, but not actually
1375 * start any services or anything. */
1376
c3090674
LP
1377 r = manager_load_unit_prepare(m, name, path, e, _ret);
1378 if (r != 0)
db06e3b6
LP
1379 return r;
1380
f50e0a01 1381 manager_dispatch_load_queue(m);
60918275 1382
9e2f7c11 1383 if (_ret)
413d6313 1384 *_ret = unit_follow_merge(*_ret);
9e2f7c11 1385
60918275
LP
1386 return 0;
1387}
a66d02c3 1388
cea8e32e 1389void manager_dump_jobs(Manager *s, FILE *f, const char *prefix) {
034c6ed7 1390 Iterator i;
a66d02c3
LP
1391 Job *j;
1392
1393 assert(s);
1394 assert(f);
1395
034c6ed7 1396 HASHMAP_FOREACH(j, s->jobs, i)
cea8e32e 1397 job_dump(j, f, prefix);
a66d02c3
LP
1398}
1399
87f0e418 1400void manager_dump_units(Manager *s, FILE *f, const char *prefix) {
034c6ed7 1401 Iterator i;
87f0e418 1402 Unit *u;
11dd41ce 1403 const char *t;
a66d02c3
LP
1404
1405 assert(s);
1406 assert(f);
1407
87f0e418 1408 HASHMAP_FOREACH_KEY(u, t, s->units, i)
ac155bb8 1409 if (u->id == t)
87f0e418 1410 unit_dump(u, f, prefix);
a66d02c3 1411}
7fad411c
LP
1412
1413void manager_clear_jobs(Manager *m) {
1414 Job *j;
1415
1416 assert(m);
1417
7fad411c 1418 while ((j = hashmap_first(m->jobs)))
5273510e
MS
1419 /* No need to recurse. We're cancelling all jobs. */
1420 job_finish_and_invalidate(j, JOB_CANCELED, false);
7fad411c 1421}
83c60c9f 1422
752b5905
LP
1423static int manager_dispatch_run_queue(sd_event_source *source, void *userdata) {
1424 Manager *m = userdata;
83c60c9f 1425 Job *j;
034c6ed7 1426
752b5905
LP
1427 assert(source);
1428 assert(m);
9152c765 1429
034c6ed7 1430 while ((j = m->run_queue)) {
ac1135be 1431 assert(j->installed);
034c6ed7
LP
1432 assert(j->in_run_queue);
1433
1434 job_run_and_invalidate(j);
9152c765 1435 }
034c6ed7 1436
a0b64226 1437 if (m->n_running_jobs > 0)
03b717a3
MS
1438 manager_watch_jobs_in_progress(m);
1439
31a7eb86
ZJS
1440 if (m->n_on_console > 0)
1441 manager_watch_idle_pipe(m);
1442
752b5905 1443 return 1;
c1e1601e
LP
1444}
1445
9588bc32 1446static unsigned manager_dispatch_dbus_queue(Manager *m) {
c1e1601e 1447 Job *j;
595ed347 1448 Unit *u;
c1e1601e
LP
1449 unsigned n = 0;
1450
1451 assert(m);
1452
1453 if (m->dispatching_dbus_queue)
1454 return 0;
1455
1456 m->dispatching_dbus_queue = true;
1457
595ed347
MS
1458 while ((u = m->dbus_unit_queue)) {
1459 assert(u->in_dbus_queue);
c1e1601e 1460
595ed347 1461 bus_unit_send_change_signal(u);
c1e1601e
LP
1462 n++;
1463 }
1464
1465 while ((j = m->dbus_job_queue)) {
1466 assert(j->in_dbus_queue);
1467
1468 bus_job_send_change_signal(j);
1469 n++;
1470 }
1471
1472 m->dispatching_dbus_queue = false;
71445ae7
LP
1473
1474 if (m->send_reloading_done) {
1475 m->send_reloading_done = false;
1476
718db961 1477 bus_manager_send_reloading(m, false);
71445ae7
LP
1478 }
1479
718db961
LP
1480 if (m->queued_message)
1481 bus_send_queued_message(m);
1482
c1e1601e 1483 return n;
9152c765
LP
1484}
1485
a354329f 1486static void manager_invoke_notify_message(Manager *m, Unit *u, pid_t pid, char *buf, size_t n, FDSet *fds) {
5ba6985b
LP
1487 _cleanup_strv_free_ char **tags = NULL;
1488
1489 assert(m);
1490 assert(u);
1491 assert(buf);
1492 assert(n > 0);
1493
1494 tags = strv_split(buf, "\n\r");
1495 if (!tags) {
1496 log_oom();
1497 return;
1498 }
1499
79008bdd 1500 log_unit_debug(u->id, "Got notification message for unit %s", u->id);
5ba6985b
LP
1501
1502 if (UNIT_VTABLE(u)->notify_message)
a354329f 1503 UNIT_VTABLE(u)->notify_message(u, pid, tags, fds);
5ba6985b
LP
1504}
1505
718db961
LP
1506static int manager_dispatch_notify_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) {
1507 Manager *m = userdata;
8c47c732 1508 ssize_t n;
a354329f 1509 int r;
8c47c732
LP
1510
1511 assert(m);
718db961
LP
1512 assert(m->notify_fd == fd);
1513
1514 if (revents != EPOLLIN) {
1515 log_warning("Got unexpected poll event for notify fd.");
1516 return 0;
1517 }
8c47c732
LP
1518
1519 for (;;) {
a354329f
LP
1520 _cleanup_fdset_free_ FDSet *fds = NULL;
1521 char buf[NOTIFY_BUFFER_MAX+1];
b92bea5d
ZJS
1522 struct iovec iovec = {
1523 .iov_base = buf,
1524 .iov_len = sizeof(buf)-1,
1525 };
8c47c732
LP
1526 union {
1527 struct cmsghdr cmsghdr;
a354329f
LP
1528 uint8_t buf[CMSG_SPACE(sizeof(struct ucred)) +
1529 CMSG_SPACE(sizeof(int) * NOTIFY_FD_MAX)];
b92bea5d 1530 } control = {};
b92bea5d
ZJS
1531 struct msghdr msghdr = {
1532 .msg_iov = &iovec,
1533 .msg_iovlen = 1,
1534 .msg_control = &control,
1535 .msg_controllen = sizeof(control),
1536 };
a354329f
LP
1537 struct cmsghdr *cmsg;
1538 struct ucred *ucred = NULL;
1539 bool found = false;
70af4d17 1540 Unit *u1, *u2, *u3;
a354329f
LP
1541 int *fd_array = NULL;
1542 unsigned n_fds = 0;
8c47c732 1543
a354329f
LP
1544 n = recvmsg(m->notify_fd, &msghdr, MSG_DONTWAIT|MSG_CMSG_CLOEXEC);
1545 if (n < 0) {
f6144808 1546 if (errno == EAGAIN || errno == EINTR)
8c47c732
LP
1547 break;
1548
1549 return -errno;
1550 }
a354329f
LP
1551
1552 for (cmsg = CMSG_FIRSTHDR(&msghdr); cmsg; cmsg = CMSG_NXTHDR(&msghdr, cmsg)) {
1553 if (cmsg->cmsg_level == SOL_SOCKET && cmsg->cmsg_type == SCM_RIGHTS) {
1554
1555 fd_array = (int*) CMSG_DATA(cmsg);
1556 n_fds = (cmsg->cmsg_len - CMSG_LEN(0)) / sizeof(int);
1557
1558 } else if (cmsg->cmsg_level == SOL_SOCKET &&
1559 cmsg->cmsg_type == SCM_CREDENTIALS &&
1560 cmsg->cmsg_len == CMSG_LEN(sizeof(struct ucred))) {
1561
1562 ucred = (struct ucred*) CMSG_DATA(cmsg);
1563 }
1564 }
1565
1566 if (n_fds > 0) {
1567 assert(fd_array);
1568
1569 r = fdset_new_array(&fds, fd_array, n_fds);
1570 if (r < 0) {
1571 close_many(fd_array, n_fds);
1572 return log_oom();
1573 }
1574 }
8c47c732 1575
a354329f
LP
1576 if (!ucred || ucred->pid <= 0) {
1577 log_warning("Received notify message without valid credentials. Ignoring.");
8c47c732
LP
1578 continue;
1579 }
1580
a354329f
LP
1581 if ((size_t) n >= sizeof(buf)) {
1582 log_warning("Received notify message exceeded maximum size. Ignoring.");
1583 continue;
1584 }
8c47c732 1585
8c40acf7 1586 buf[n] = 0;
8c47c732 1587
70af4d17
LP
1588 /* Notify every unit that might be interested, but try
1589 * to avoid notifying the same one multiple times. */
1590 u1 = manager_get_unit_by_pid(m, ucred->pid);
1591 if (u1) {
a354329f 1592 manager_invoke_notify_message(m, u1, ucred->pid, buf, n, fds);
5ba6985b
LP
1593 found = true;
1594 }
1595
70af4d17
LP
1596 u2 = hashmap_get(m->watch_pids1, LONG_TO_PTR(ucred->pid));
1597 if (u2 && u2 != u1) {
a354329f 1598 manager_invoke_notify_message(m, u2, ucred->pid, buf, n, fds);
5ba6985b
LP
1599 found = true;
1600 }
1601
70af4d17
LP
1602 u3 = hashmap_get(m->watch_pids2, LONG_TO_PTR(ucred->pid));
1603 if (u3 && u3 != u2 && u3 != u1) {
a354329f 1604 manager_invoke_notify_message(m, u3, ucred->pid, buf, n, fds);
5ba6985b
LP
1605 found = true;
1606 }
8c47c732 1607
5ba6985b
LP
1608 if (!found)
1609 log_warning("Cannot find unit for notify message of PID "PID_FMT".", ucred->pid);
a354329f
LP
1610
1611 if (fdset_size(fds) > 0)
1612 log_warning("Got auxiliary fds with notification message, closing all.");
8c47c732
LP
1613 }
1614
1615 return 0;
1616}
1617
5ba6985b
LP
1618static void invoke_sigchld_event(Manager *m, Unit *u, siginfo_t *si) {
1619 assert(m);
1620 assert(u);
1621 assert(si);
1622
79008bdd 1623 log_unit_debug(u->id, "Child "PID_FMT" belongs to %s", si->si_pid, u->id);
5ba6985b
LP
1624
1625 unit_unwatch_pid(u, si->si_pid);
1626 UNIT_VTABLE(u)->sigchld_event(u, si->si_pid, si->si_code, si->si_status);
1627}
1628
034c6ed7 1629static int manager_dispatch_sigchld(Manager *m) {
9152c765
LP
1630 assert(m);
1631
1632 for (;;) {
b92bea5d 1633 siginfo_t si = {};
9152c765 1634
4112df16
LP
1635 /* First we call waitd() for a PID and do not reap the
1636 * zombie. That way we can still access /proc/$PID for
1637 * it while it is a zombie. */
1638 if (waitid(P_ALL, 0, &si, WEXITED|WNOHANG|WNOWAIT) < 0) {
acbb0225
LP
1639
1640 if (errno == ECHILD)
1641 break;
1642
4112df16
LP
1643 if (errno == EINTR)
1644 continue;
1645
9152c765 1646 return -errno;
acbb0225 1647 }
9152c765 1648
4112df16 1649 if (si.si_pid <= 0)
9152c765
LP
1650 break;
1651
15d5d9d9 1652 if (si.si_code == CLD_EXITED || si.si_code == CLD_KILLED || si.si_code == CLD_DUMPED) {
7fd1b19b 1653 _cleanup_free_ char *name = NULL;
70af4d17 1654 Unit *u1, *u2, *u3;
4112df16 1655
87d2c1ff 1656 get_process_comm(si.si_pid, &name);
4112df16 1657
5ba6985b
LP
1658 log_debug("Child "PID_FMT" (%s) died (code=%s, status=%i/%s)",
1659 si.si_pid, strna(name),
1660 sigchld_code_to_string(si.si_code),
1661 si.si_status,
1662 strna(si.si_code == CLD_EXITED
1663 ? exit_status_to_string(si.si_status, EXIT_STATUS_FULL)
1664 : signal_to_string(si.si_status)));
1665
1666 /* And now figure out the unit this belongs
1667 * to, it might be multiple... */
70af4d17
LP
1668 u1 = manager_get_unit_by_pid(m, si.si_pid);
1669 if (u1)
1670 invoke_sigchld_event(m, u1, &si);
1671 u2 = hashmap_get(m->watch_pids1, LONG_TO_PTR(si.si_pid));
1672 if (u2 && u2 != u1)
1673 invoke_sigchld_event(m, u2, &si);
1674 u3 = hashmap_get(m->watch_pids2, LONG_TO_PTR(si.si_pid));
1675 if (u3 && u3 != u2 && u3 != u1)
1676 invoke_sigchld_event(m, u3, &si);
5ba6985b 1677 }
8c47c732 1678
4112df16
LP
1679 /* And now, we actually reap the zombie. */
1680 if (waitid(P_PID, si.si_pid, &si, WEXITED) < 0) {
1681 if (errno == EINTR)
1682 continue;
1683
1684 return -errno;
1685 }
9152c765
LP
1686 }
1687
1688 return 0;
1689}
1690
7d793605 1691static int manager_start_target(Manager *m, const char *name, JobMode mode) {
718db961 1692 _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
28247076 1693 int r;
398ef8ba 1694
79008bdd 1695 log_unit_debug(name, "Activating special unit %s", name);
1e001f52 1696
bd0af849
ZJS
1697 r = manager_add_job_by_name(m, JOB_START, name, mode, true, &error, NULL);
1698 if (r < 0)
79008bdd 1699 log_unit_error(name, "Failed to enqueue %s job: %s", name, bus_error_message(&error, r));
a1b256b0
LP
1700
1701 return r;
28247076
LP
1702}
1703
718db961
LP
1704static int manager_dispatch_signal_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) {
1705 Manager *m = userdata;
9152c765
LP
1706 ssize_t n;
1707 struct signalfd_siginfo sfsi;
1708 bool sigchld = false;
1709
1710 assert(m);
718db961
LP
1711 assert(m->signal_fd == fd);
1712
1713 if (revents != EPOLLIN) {
1714 log_warning("Got unexpected events from signal file descriptor.");
1715 return 0;
1716 }
9152c765
LP
1717
1718 for (;;) {
718db961 1719 n = read(m->signal_fd, &sfsi, sizeof(sfsi));
57cb4adf 1720 if (n != sizeof(sfsi)) {
9152c765
LP
1721
1722 if (n >= 0)
1723 return -EIO;
1724
63090775 1725 if (errno == EINTR || errno == EAGAIN)
acbb0225 1726 break;
9152c765
LP
1727
1728 return -errno;
1729 }
1730
4daf54a8
ZJS
1731 log_received_signal(sfsi.ssi_signo == SIGCHLD ||
1732 (sfsi.ssi_signo == SIGTERM && m->running_as == SYSTEMD_USER)
1733 ? LOG_DEBUG : LOG_INFO,
1734 &sfsi);
1e001f52 1735
b9cd2ec1
LP
1736 switch (sfsi.ssi_signo) {
1737
4112df16 1738 case SIGCHLD:
9152c765 1739 sigchld = true;
b9cd2ec1
LP
1740 break;
1741
6632c602 1742 case SIGTERM:
67445f4e 1743 if (m->running_as == SYSTEMD_SYSTEM) {
db06e3b6
LP
1744 /* This is for compatibility with the
1745 * original sysvinit */
e11dc4a2 1746 m->exit_code = MANAGER_REEXECUTE;
a1b256b0
LP
1747 break;
1748 }
84e9af1e 1749
a1b256b0 1750 /* Fall through */
e11dc4a2
LP
1751
1752 case SIGINT:
67445f4e 1753 if (m->running_as == SYSTEMD_SYSTEM) {
2e5c94b9 1754
d1f6b1b4
LP
1755 /* If the user presses C-A-D more than
1756 * 7 times within 2s, we reboot
2e5c94b9
LP
1757 * immediately. */
1758
1759 if (ratelimit_test(&m->ctrl_alt_del_ratelimit))
1760 manager_start_target(m, SPECIAL_CTRL_ALT_DEL_TARGET, JOB_REPLACE_IRREVERSIBLY);
1761 else {
1762 log_notice("Ctrl-Alt-Del was pressed more than 7 times within 2s, rebooting immediately.");
a626df3e 1763 status_printf(NULL, true, false, "Ctrl-Alt-Del was pressed more than 7 times within 2s, rebooting immediately.");
2e5c94b9
LP
1764 m->exit_code = MANAGER_REBOOT;
1765 }
1766
84e9af1e
LP
1767 break;
1768 }
1769
a1b256b0 1770 /* Run the exit target if there is one, if not, just exit. */
0003d1ab 1771 if (manager_start_target(m, SPECIAL_EXIT_TARGET, JOB_REPLACE) < 0) {
a1b256b0
LP
1772 m->exit_code = MANAGER_EXIT;
1773 return 0;
1774 }
1775
1776 break;
84e9af1e 1777
28247076 1778 case SIGWINCH:
67445f4e 1779 if (m->running_as == SYSTEMD_SYSTEM)
7d793605 1780 manager_start_target(m, SPECIAL_KBREQUEST_TARGET, JOB_REPLACE);
84e9af1e 1781
28247076
LP
1782 /* This is a nop on non-init */
1783 break;
84e9af1e 1784
28247076 1785 case SIGPWR:
67445f4e 1786 if (m->running_as == SYSTEMD_SYSTEM)
7d793605 1787 manager_start_target(m, SPECIAL_SIGPWR_TARGET, JOB_REPLACE);
84e9af1e 1788
28247076 1789 /* This is a nop on non-init */
84e9af1e 1790 break;
6632c602 1791
1005d14f 1792 case SIGUSR1: {
57ee42ce
LP
1793 Unit *u;
1794
1795 u = manager_get_unit(m, SPECIAL_DBUS_SERVICE);
1796
1797 if (!u || UNIT_IS_ACTIVE_OR_RELOADING(unit_active_state(u))) {
1798 log_info("Trying to reconnect to bus...");
3996fbe2 1799 bus_init(m, true);
57ee42ce
LP
1800 }
1801
1802 if (!u || !UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(u))) {
1803 log_info("Loading D-Bus service...");
7d793605 1804 manager_start_target(m, SPECIAL_DBUS_SERVICE, JOB_REPLACE);
57ee42ce
LP
1805 }
1806
1807 break;
1808 }
1809
2149e37c 1810 case SIGUSR2: {
718db961
LP
1811 _cleanup_free_ char *dump = NULL;
1812 _cleanup_fclose_ FILE *f = NULL;
2149e37c
LP
1813 size_t size;
1814
718db961
LP
1815 f = open_memstream(&dump, &size);
1816 if (!f) {
2149e37c
LP
1817 log_warning("Failed to allocate memory stream.");
1818 break;
1819 }
1820
1821 manager_dump_units(m, f, "\t");
1822 manager_dump_jobs(m, f, "\t");
1823
1824 if (ferror(f)) {
2149e37c
LP
1825 log_warning("Failed to write status stream");
1826 break;
1827 }
1828
b2cdc666
DM
1829 if (fflush(f)) {
1830 log_warning("Failed to flush status stream");
1831 break;
1832 }
1833
2149e37c 1834 log_dump(LOG_INFO, dump);
1005d14f 1835 break;
2149e37c 1836 }
1005d14f 1837
a16e1123
LP
1838 case SIGHUP:
1839 m->exit_code = MANAGER_RELOAD;
1840 break;
1841
7d793605 1842 default: {
253ee27a 1843
0003d1ab
LP
1844 /* Starting SIGRTMIN+0 */
1845 static const char * const target_table[] = {
7d793605
LP
1846 [0] = SPECIAL_DEFAULT_TARGET,
1847 [1] = SPECIAL_RESCUE_TARGET,
f057408c 1848 [2] = SPECIAL_EMERGENCY_TARGET,
7d793605
LP
1849 [3] = SPECIAL_HALT_TARGET,
1850 [4] = SPECIAL_POWEROFF_TARGET,
0003d1ab
LP
1851 [5] = SPECIAL_REBOOT_TARGET,
1852 [6] = SPECIAL_KEXEC_TARGET
1853 };
1854
1855 /* Starting SIGRTMIN+13, so that target halt and system halt are 10 apart */
1856 static const ManagerExitCode code_table[] = {
1857 [0] = MANAGER_HALT,
1858 [1] = MANAGER_POWEROFF,
1859 [2] = MANAGER_REBOOT,
1860 [3] = MANAGER_KEXEC
7d793605
LP
1861 };
1862
1863 if ((int) sfsi.ssi_signo >= SIGRTMIN+0 &&
0003d1ab 1864 (int) sfsi.ssi_signo < SIGRTMIN+(int) ELEMENTSOF(target_table)) {
764e9b5f
MS
1865 int idx = (int) sfsi.ssi_signo - SIGRTMIN;
1866 manager_start_target(m, target_table[idx],
1867 (idx == 1 || idx == 2) ? JOB_ISOLATE : JOB_REPLACE);
7d793605
LP
1868 break;
1869 }
1870
0003d1ab
LP
1871 if ((int) sfsi.ssi_signo >= SIGRTMIN+13 &&
1872 (int) sfsi.ssi_signo < SIGRTMIN+13+(int) ELEMENTSOF(code_table)) {
1873 m->exit_code = code_table[sfsi.ssi_signo - SIGRTMIN - 13];
1874 break;
1875 }
1876
0658666b
LP
1877 switch (sfsi.ssi_signo - SIGRTMIN) {
1878
1879 case 20:
1880 log_debug("Enabling showing of status.");
d450b6f2 1881 manager_set_show_status(m, SHOW_STATUS_YES);
0658666b
LP
1882 break;
1883
1884 case 21:
1885 log_debug("Disabling showing of status.");
d450b6f2 1886 manager_set_show_status(m, SHOW_STATUS_NO);
0658666b
LP
1887 break;
1888
253ee27a
LP
1889 case 22:
1890 log_set_max_level(LOG_DEBUG);
1891 log_notice("Setting log level to debug.");
1892 break;
1893
1894 case 23:
1895 log_set_max_level(LOG_INFO);
1896 log_notice("Setting log level to info.");
1897 break;
1898
600b704e
LP
1899 case 24:
1900 if (m->running_as == SYSTEMD_USER) {
1901 m->exit_code = MANAGER_EXIT;
1902 return 0;
1903 }
1904
1905 /* This is a nop on init */
1906 break;
1907
4cfa2c99 1908 case 26:
c1dc6153 1909 case 29: /* compatibility: used to be mapped to LOG_TARGET_SYSLOG_OR_KMSG */
4cfa2c99
LP
1910 log_set_target(LOG_TARGET_JOURNAL_OR_KMSG);
1911 log_notice("Setting log target to journal-or-kmsg.");
1912 break;
1913
253ee27a
LP
1914 case 27:
1915 log_set_target(LOG_TARGET_CONSOLE);
1916 log_notice("Setting log target to console.");
1917 break;
1918
1919 case 28:
1920 log_set_target(LOG_TARGET_KMSG);
1921 log_notice("Setting log target to kmsg.");
1922 break;
1923
0658666b 1924 default:
4e240ab0 1925 log_warning("Got unhandled signal <%s>.", signal_to_string(sfsi.ssi_signo));
0658666b 1926 }
b9cd2ec1 1927 }
7d793605 1928 }
9152c765
LP
1929 }
1930
1931 if (sigchld)
7b77ed8c 1932 manager_dispatch_sigchld(m);
034c6ed7
LP
1933
1934 return 0;
1935}
1936
718db961
LP
1937static int manager_dispatch_time_change_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) {
1938 Manager *m = userdata;
1939 Iterator i;
1940 Unit *u;
034c6ed7
LP
1941
1942 assert(m);
718db961 1943 assert(m->time_change_fd == fd);
034c6ed7 1944
718db961 1945 log_struct(LOG_INFO,
e2cc6eca
LP
1946 LOG_MESSAGE_ID(SD_MESSAGE_TIME_CHANGE),
1947 LOG_MESSAGE("Time has been changed"),
718db961 1948 NULL);
034c6ed7 1949
718db961
LP
1950 /* Restart the watch */
1951 m->time_change_event_source = sd_event_source_unref(m->time_change_event_source);
03e334a1 1952 m->time_change_fd = safe_close(m->time_change_fd);
ef734fd6 1953
718db961 1954 manager_setup_time_change(m);
4e434314 1955
718db961
LP
1956 HASHMAP_FOREACH(u, m->units, i)
1957 if (UNIT_VTABLE(u)->time_change)
1958 UNIT_VTABLE(u)->time_change(u);
ea430986 1959
718db961
LP
1960 return 0;
1961}
ea430986 1962
718db961
LP
1963static int manager_dispatch_idle_pipe_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) {
1964 Manager *m = userdata;
8742514c 1965
718db961
LP
1966 assert(m);
1967 assert(m->idle_pipe[2] == fd);
8742514c 1968
718db961 1969 m->no_console_output = m->n_on_console > 0;
03b717a3 1970
718db961
LP
1971 m->idle_pipe_event_source = sd_event_source_unref(m->idle_pipe_event_source);
1972 manager_close_idle_pipe(m);
03b717a3 1973
718db961
LP
1974 return 0;
1975}
31a7eb86 1976
718db961
LP
1977static int manager_dispatch_jobs_in_progress(sd_event_source *source, usec_t usec, void *userdata) {
1978 Manager *m = userdata;
fd08a840
ZJS
1979 int r;
1980 uint64_t next;
31a7eb86 1981
718db961 1982 assert(m);
fd08a840 1983 assert(source);
9152c765 1984
718db961 1985 manager_print_jobs_in_progress(m);
fd08a840
ZJS
1986
1987 next = now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_PERIOD_USEC;
1988 r = sd_event_source_set_time(source, next);
1989 if (r < 0)
1990 return r;
1991
1992 return sd_event_source_set_enabled(source, SD_EVENT_ONESHOT);
9152c765
LP
1993}
1994
1995int manager_loop(Manager *m) {
1996 int r;
9152c765 1997
fac9f8df 1998 RATELIMIT_DEFINE(rl, 1*USEC_PER_SEC, 50000);
ea430986 1999
9152c765 2000 assert(m);
f755e3b7 2001 m->exit_code = MANAGER_OK;
9152c765 2002
fe51822e
LP
2003 /* Release the path cache */
2004 set_free_free(m->unit_path_cache);
2005 m->unit_path_cache = NULL;
2006
b0c918b9
LP
2007 manager_check_finished(m);
2008
a4312405 2009 /* There might still be some zombies hanging around from
f3669545 2010 * before we were exec()'ed. Let's reap them. */
e96d6be7
LP
2011 r = manager_dispatch_sigchld(m);
2012 if (r < 0)
a4312405
LP
2013 return r;
2014
f755e3b7 2015 while (m->exit_code == MANAGER_OK) {
718db961 2016 usec_t wait_usec;
9152c765 2017
67445f4e 2018 if (m->runtime_watchdog > 0 && m->running_as == SYSTEMD_SYSTEM)
e96d6be7
LP
2019 watchdog_ping();
2020
ea430986
LP
2021 if (!ratelimit_test(&rl)) {
2022 /* Yay, something is going seriously wrong, pause a little */
2023 log_warning("Looping too fast. Throttling execution a little.");
2024 sleep(1);
e96d6be7 2025 continue;
ea430986
LP
2026 }
2027
37a8e683 2028 if (manager_dispatch_load_queue(m) > 0)
23a177ef
LP
2029 continue;
2030
cf1265e1 2031 if (manager_dispatch_gc_queue(m) > 0)
701cc384
LP
2032 continue;
2033
cf1265e1 2034 if (manager_dispatch_cleanup_queue(m) > 0)
c1e1601e 2035 continue;
034c6ed7 2036
cf1265e1 2037 if (manager_dispatch_cgroup_queue(m) > 0)
c1e1601e
LP
2038 continue;
2039
c1e1601e 2040 if (manager_dispatch_dbus_queue(m) > 0)
ea430986 2041 continue;
ea430986 2042
c757a65b 2043 /* Sleep for half the watchdog time */
67445f4e 2044 if (m->runtime_watchdog > 0 && m->running_as == SYSTEMD_SYSTEM) {
718db961
LP
2045 wait_usec = m->runtime_watchdog / 2;
2046 if (wait_usec <= 0)
2047 wait_usec = 1;
c757a65b 2048 } else
3a43da28 2049 wait_usec = USEC_INFINITY;
9152c765 2050
718db961 2051 r = sd_event_run(m->event, wait_usec);
23bbb0de
MS
2052 if (r < 0)
2053 return log_error_errno(r, "Failed to run event loop: %m");
a16e1123 2054 }
957ca890 2055
a16e1123 2056 return m->exit_code;
83c60c9f 2057}
ea430986 2058
718db961 2059int manager_load_unit_from_dbus_path(Manager *m, const char *s, sd_bus_error *e, Unit **_u) {
ede3a796 2060 _cleanup_free_ char *n = NULL;
ea430986 2061 Unit *u;
80fbf05e 2062 int r;
ea430986
LP
2063
2064 assert(m);
2065 assert(s);
2066 assert(_u);
2067
ede3a796
LP
2068 r = unit_name_from_dbus_path(s, &n);
2069 if (r < 0)
2070 return r;
ea430986 2071
80fbf05e 2072 r = manager_load_unit(m, n, NULL, e, &u);
80fbf05e
MS
2073 if (r < 0)
2074 return r;
ea430986
LP
2075
2076 *_u = u;
2077
2078 return 0;
2079}
86fbf370
LP
2080
2081int manager_get_job_from_dbus_path(Manager *m, const char *s, Job **_j) {
718db961 2082 const char *p;
86fbf370 2083 unsigned id;
718db961 2084 Job *j;
86fbf370
LP
2085 int r;
2086
2087 assert(m);
2088 assert(s);
2089 assert(_j);
2090
718db961
LP
2091 p = startswith(s, "/org/freedesktop/systemd1/job/");
2092 if (!p)
86fbf370
LP
2093 return -EINVAL;
2094
718db961 2095 r = safe_atou(p, &id);
8742514c 2096 if (r < 0)
86fbf370
LP
2097 return r;
2098
8742514c
LP
2099 j = manager_get_job(m, id);
2100 if (!j)
86fbf370
LP
2101 return -ENOENT;
2102
2103 *_j = j;
2104
2105 return 0;
2106}
dfcd764e 2107
4927fcae 2108void manager_send_unit_audit(Manager *m, Unit *u, int type, bool success) {
e537352b 2109
4927fcae 2110#ifdef HAVE_AUDIT
2ba11090 2111 _cleanup_free_ char *p = NULL;
0aa281df 2112 const char *msg;
c1165f82 2113 int audit_fd;
e537352b 2114
c1165f82
LP
2115 audit_fd = get_audit_fd();
2116 if (audit_fd < 0)
e537352b
LP
2117 return;
2118
bbd3a7ba
LP
2119 /* Don't generate audit events if the service was already
2120 * started and we're just deserializing */
a7556052 2121 if (m->n_reloading > 0)
bbd3a7ba
LP
2122 return;
2123
67445f4e 2124 if (m->running_as != SYSTEMD_SYSTEM)
f1dd0c3f
LP
2125 return;
2126
ac155bb8 2127 if (u->type != UNIT_SERVICE)
f1dd0c3f
LP
2128 return;
2129
bd0af849
ZJS
2130 p = unit_name_to_prefix_and_instance(u->id);
2131 if (!p) {
0aa281df 2132 log_oom();
e537352b
LP
2133 return;
2134 }
2135
63c372cb 2136 msg = strjoina("unit=", p);
0aa281df
LP
2137 if (audit_log_user_comm_message(audit_fd, type, msg, "systemd", NULL, NULL, NULL, success) < 0) {
2138 if (errno == EPERM)
391ade86 2139 /* We aren't allowed to send audit messages?
44785992 2140 * Then let's not retry again. */
c1165f82 2141 close_audit_fd();
0aa281df 2142 else
56f64d95 2143 log_warning_errno(errno, "Failed to send audit message: %m");
391ade86 2144 }
4927fcae 2145#endif
e537352b 2146
e537352b
LP
2147}
2148
e983b760 2149void manager_send_unit_plymouth(Manager *m, Unit *u) {
1d749d04 2150 union sockaddr_union sa = PLYMOUTH_SOCKET;
2ba11090 2151
e983b760 2152 int n = 0;
2ba11090
ZJS
2153 _cleanup_free_ char *message = NULL;
2154 _cleanup_close_ int fd = -1;
e983b760
LP
2155
2156 /* Don't generate plymouth events if the service was already
2157 * started and we're just deserializing */
a7556052 2158 if (m->n_reloading > 0)
e983b760
LP
2159 return;
2160
67445f4e 2161 if (m->running_as != SYSTEMD_SYSTEM)
e983b760
LP
2162 return;
2163
3772995a
LP
2164 if (detect_container(NULL) > 0)
2165 return;
2166
ac155bb8
MS
2167 if (u->type != UNIT_SERVICE &&
2168 u->type != UNIT_MOUNT &&
2169 u->type != UNIT_SWAP)
e983b760
LP
2170 return;
2171
2172 /* We set SOCK_NONBLOCK here so that we rather drop the
2173 * message then wait for plymouth */
e62d8c39
ZJS
2174 fd = socket(AF_UNIX, SOCK_STREAM|SOCK_CLOEXEC|SOCK_NONBLOCK, 0);
2175 if (fd < 0) {
56f64d95 2176 log_error_errno(errno, "socket() failed: %m");
e983b760
LP
2177 return;
2178 }
2179
96707269 2180 if (connect(fd, &sa.sa, offsetof(struct sockaddr_un, sun_path) + 1 + strlen(sa.un.sun_path+1)) < 0) {
e983b760 2181
2ba11090 2182 if (!IN_SET(errno, EPIPE, EAGAIN, ENOENT, ECONNREFUSED, ECONNRESET, ECONNABORTED))
56f64d95 2183 log_error_errno(errno, "connect() failed: %m");
2ba11090 2184 return;
e983b760
LP
2185 }
2186
ac155bb8 2187 if (asprintf(&message, "U\002%c%s%n", (int) (strlen(u->id) + 1), u->id, &n) < 0) {
0d0f0c50 2188 log_oom();
2ba11090 2189 return;
e983b760
LP
2190 }
2191
2192 errno = 0;
2ba11090
ZJS
2193 if (write(fd, message, n + 1) != n + 1)
2194 if (!IN_SET(errno, EPIPE, EAGAIN, ENOENT, ECONNREFUSED, ECONNRESET, ECONNABORTED))
56f64d95 2195 log_error_errno(errno, "Failed to write Plymouth message: %m");
e983b760
LP
2196}
2197
05e343b7
LP
2198void manager_dispatch_bus_name_owner_changed(
2199 Manager *m,
2200 const char *name,
2201 const char* old_owner,
2202 const char *new_owner) {
2203
2204 Unit *u;
2205
2206 assert(m);
2207 assert(name);
2208
718db961
LP
2209 u = hashmap_get(m->watch_bus, name);
2210 if (!u)
05e343b7
LP
2211 return;
2212
2213 UNIT_VTABLE(u)->bus_name_owner_change(u, name, old_owner, new_owner);
2214}
2215
d8d5ab98 2216int manager_open_serialization(Manager *m, FILE **_f) {
8e33886e 2217 const char *path;
df28bc08 2218 int fd = -1;
a16e1123
LP
2219 FILE *f;
2220
2221 assert(_f);
2222
8e33886e
ZJS
2223 path = m->running_as == SYSTEMD_SYSTEM ? "/run/systemd" : "/tmp";
2224 fd = open_tmpfile(path, O_RDWR|O_CLOEXEC);
d86f9d52 2225 if (fd < 0)
a16e1123 2226 return -errno;
a16e1123 2227
a16e1123 2228 log_debug("Serializing state to %s", path);
a16e1123 2229
01e10de3 2230 f = fdopen(fd, "w+");
d86f9d52 2231 if (!f) {
03e334a1 2232 safe_close(fd);
a16e1123 2233 return -errno;
d86f9d52 2234 }
a16e1123
LP
2235
2236 *_f = f;
2237
2238 return 0;
2239}
2240
b3680f49 2241int manager_serialize(Manager *m, FILE *f, FDSet *fds, bool switching_root) {
a16e1123
LP
2242 Iterator i;
2243 Unit *u;
2244 const char *t;
4a9fd066 2245 char **e;
a16e1123
LP
2246 int r;
2247
2248 assert(m);
2249 assert(f);
2250 assert(fds);
2251
a7556052 2252 m->n_reloading ++;
38c52d46 2253
1fa2f38f 2254 fprintf(f, "current-job-id=%"PRIu32"\n", m->current_job_id);
01d67b43 2255 fprintf(f, "taint-usr=%s\n", yes_no(m->taint_usr));
33c5fae9
LP
2256 fprintf(f, "n-installed-jobs=%u\n", m->n_installed_jobs);
2257 fprintf(f, "n-failed-jobs=%u\n", m->n_failed_jobs);
01d67b43 2258
915b3753 2259 dual_timestamp_serialize(f, "firmware-timestamp", &m->firmware_timestamp);
915b3753 2260 dual_timestamp_serialize(f, "loader-timestamp", &m->loader_timestamp);
718db961 2261 dual_timestamp_serialize(f, "kernel-timestamp", &m->kernel_timestamp);
e9ddabc2 2262 dual_timestamp_serialize(f, "initrd-timestamp", &m->initrd_timestamp);
f38ed060 2263
26a1efdf 2264 if (!in_initrd()) {
915b3753 2265 dual_timestamp_serialize(f, "userspace-timestamp", &m->userspace_timestamp);
f38ed060 2266 dual_timestamp_serialize(f, "finish-timestamp", &m->finish_timestamp);
718db961
LP
2267 dual_timestamp_serialize(f, "security-start-timestamp", &m->security_start_timestamp);
2268 dual_timestamp_serialize(f, "security-finish-timestamp", &m->security_finish_timestamp);
2269 dual_timestamp_serialize(f, "generators-start-timestamp", &m->generators_start_timestamp);
2270 dual_timestamp_serialize(f, "generators-finish-timestamp", &m->generators_finish_timestamp);
2271 dual_timestamp_serialize(f, "units-load-start-timestamp", &m->units_load_start_timestamp);
2272 dual_timestamp_serialize(f, "units-load-finish-timestamp", &m->units_load_finish_timestamp);
f38ed060 2273 }
47a483a1 2274
b3680f49
HH
2275 if (!switching_root) {
2276 STRV_FOREACH(e, m->environment) {
2277 _cleanup_free_ char *ce;
4a9fd066 2278
b3680f49 2279 ce = cescape(*e);
e3dd987c
LP
2280 if (!ce)
2281 return -ENOMEM;
2282
2283 fprintf(f, "env=%s\n", *e);
b3680f49 2284 }
4a9fd066
OS
2285 }
2286
d86f9d52
LP
2287 if (m->notify_fd >= 0) {
2288 int copy;
2289
2290 copy = fdset_put_dup(fds, m->notify_fd);
2291 if (copy < 0)
2292 return copy;
2293
2294 fprintf(f, "notify-fd=%i\n", copy);
2295 fprintf(f, "notify-socket=%s\n", m->notify_socket);
2296 }
2297
e3dd987c
LP
2298 if (m->kdbus_fd >= 0) {
2299 int copy;
2300
2301 copy = fdset_put_dup(fds, m->kdbus_fd);
2302 if (copy < 0)
2303 return copy;
2304
2305 fprintf(f, "kdbus-fd=%i\n", copy);
2306 }
2307
8f8f05a9 2308 bus_track_serialize(m->subscribed, f);
6fa48533 2309
f2382a94
LP
2310 fputc('\n', f);
2311
a16e1123 2312 HASHMAP_FOREACH_KEY(u, t, m->units, i) {
ac155bb8 2313 if (u->id != t)
a16e1123
LP
2314 continue;
2315
a16e1123 2316 /* Start marker */
ac155bb8 2317 fputs(u->id, f);
a16e1123
LP
2318 fputc('\n', f);
2319
6fa48533
LP
2320 r = unit_serialize(u, f, fds, !switching_root);
2321 if (r < 0) {
a7556052 2322 m->n_reloading --;
a16e1123 2323 return r;
38c52d46 2324 }
a16e1123
LP
2325 }
2326
a7556052
LP
2327 assert(m->n_reloading > 0);
2328 m->n_reloading --;
38c52d46 2329
a16e1123
LP
2330 if (ferror(f))
2331 return -EIO;
2332
b23de6af
LP
2333 r = bus_fdset_add_all(m, fds);
2334 if (r < 0)
2335 return r;
2336
a16e1123
LP
2337 return 0;
2338}
2339
2340int manager_deserialize(Manager *m, FILE *f, FDSet *fds) {
2341 int r = 0;
2342
2343 assert(m);
2344 assert(f);
2345
2346 log_debug("Deserializing state...");
2347
a7556052 2348 m->n_reloading ++;
82c64bf5 2349
10f8e83c 2350 for (;;) {
20c03b7b 2351 char line[LINE_MAX], *l;
10f8e83c
LP
2352
2353 if (!fgets(line, sizeof(line), f)) {
2354 if (feof(f))
2355 r = 0;
2356 else
2357 r = -errno;
2358
2359 goto finish;
2360 }
2361
2362 char_array_0(line);
2363 l = strstrip(line);
2364
2365 if (l[0] == 0)
2366 break;
2367
01d67b43
LP
2368 if (startswith(l, "current-job-id=")) {
2369 uint32_t id;
2370
2371 if (safe_atou32(l+15, &id) < 0)
e5035a27 2372 log_debug("Failed to parse current job id value %s", l+15);
01d67b43
LP
2373 else
2374 m->current_job_id = MAX(m->current_job_id, id);
718db961 2375
33c5fae9
LP
2376 } else if (startswith(l, "n-installed-jobs=")) {
2377 uint32_t n;
2378
2379 if (safe_atou32(l+17, &n) < 0)
e5035a27 2380 log_debug("Failed to parse installed jobs counter %s", l+17);
33c5fae9
LP
2381 else
2382 m->n_installed_jobs += n;
718db961 2383
33c5fae9
LP
2384 } else if (startswith(l, "n-failed-jobs=")) {
2385 uint32_t n;
2386
2387 if (safe_atou32(l+14, &n) < 0)
e5035a27 2388 log_debug("Failed to parse failed jobs counter %s", l+14);
33c5fae9
LP
2389 else
2390 m->n_failed_jobs += n;
718db961 2391
01d67b43
LP
2392 } else if (startswith(l, "taint-usr=")) {
2393 int b;
2394
e3dd987c
LP
2395 b = parse_boolean(l+10);
2396 if (b < 0)
e5035a27 2397 log_debug("Failed to parse taint /usr flag %s", l+10);
01d67b43
LP
2398 else
2399 m->taint_usr = m->taint_usr || b;
718db961 2400
915b3753
LP
2401 } else if (startswith(l, "firmware-timestamp="))
2402 dual_timestamp_deserialize(l+19, &m->firmware_timestamp);
2403 else if (startswith(l, "loader-timestamp="))
2404 dual_timestamp_deserialize(l+17, &m->loader_timestamp);
2405 else if (startswith(l, "kernel-timestamp="))
2406 dual_timestamp_deserialize(l+17, &m->kernel_timestamp);
2407 else if (startswith(l, "initrd-timestamp="))
e9ddabc2 2408 dual_timestamp_deserialize(l+17, &m->initrd_timestamp);
915b3753
LP
2409 else if (startswith(l, "userspace-timestamp="))
2410 dual_timestamp_deserialize(l+20, &m->userspace_timestamp);
10717a1a 2411 else if (startswith(l, "finish-timestamp="))
799fd0fd 2412 dual_timestamp_deserialize(l+17, &m->finish_timestamp);
718db961
LP
2413 else if (startswith(l, "security-start-timestamp="))
2414 dual_timestamp_deserialize(l+25, &m->security_start_timestamp);
2415 else if (startswith(l, "security-finish-timestamp="))
2416 dual_timestamp_deserialize(l+26, &m->security_finish_timestamp);
2417 else if (startswith(l, "generators-start-timestamp="))
2418 dual_timestamp_deserialize(l+27, &m->generators_start_timestamp);
2419 else if (startswith(l, "generators-finish-timestamp="))
2420 dual_timestamp_deserialize(l+28, &m->generators_finish_timestamp);
2421 else if (startswith(l, "units-load-start-timestamp="))
2422 dual_timestamp_deserialize(l+27, &m->units_load_start_timestamp);
2423 else if (startswith(l, "units-load-finish-timestamp="))
2424 dual_timestamp_deserialize(l+28, &m->units_load_finish_timestamp);
4a9fd066
OS
2425 else if (startswith(l, "env=")) {
2426 _cleanup_free_ char *uce = NULL;
2427 char **e;
2428
2429 uce = cunescape(l+4);
2430 if (!uce) {
2431 r = -ENOMEM;
2432 goto finish;
2433 }
2434
2435 e = strv_env_set(m->environment, uce);
2436 if (!e) {
2437 r = -ENOMEM;
2438 goto finish;
2439 }
2440
2441 strv_free(m->environment);
2442 m->environment = e;
e3dd987c 2443
d86f9d52
LP
2444 } else if (startswith(l, "notify-fd=")) {
2445 int fd;
2446
2447 if (safe_atoi(l + 10, &fd) < 0 || fd < 0 || !fdset_contains(fds, fd))
e5035a27 2448 log_debug("Failed to parse notify fd: %s", l + 10);
d86f9d52 2449 else {
03e334a1
LP
2450 m->notify_event_source = sd_event_source_unref(m->notify_event_source);
2451 safe_close(m->notify_fd);
d86f9d52
LP
2452 m->notify_fd = fdset_remove(fds, fd);
2453 }
2454
2455 } else if (startswith(l, "notify-socket=")) {
2456 char *n;
2457
2458 n = strdup(l+14);
2459 if (!n) {
2460 r = -ENOMEM;
2461 goto finish;
2462 }
2463
2464 free(m->notify_socket);
2465 m->notify_socket = n;
2466
e3dd987c
LP
2467 } else if (startswith(l, "kdbus-fd=")) {
2468 int fd;
2469
8bf9fcf4 2470 if (safe_atoi(l + 9, &fd) < 0 || fd < 0 || !fdset_contains(fds, fd))
e5035a27 2471 log_debug("Failed to parse kdbus fd: %s", l + 9);
e3dd987c 2472 else {
03e334a1 2473 safe_close(m->kdbus_fd);
e3dd987c
LP
2474 m->kdbus_fd = fdset_remove(fds, fd);
2475 }
2476
230314d7
LP
2477 } else {
2478 int k;
2479
2480 k = bus_track_deserialize_item(&m->deserialized_subscribed, l);
2481 if (k < 0)
2482 log_debug_errno(k, "Failed to deserialize bus tracker object: %m");
2483 else if (k == 0)
2484 log_debug("Unknown serialization item '%s'", l);
2485 }
10f8e83c
LP
2486 }
2487
a16e1123
LP
2488 for (;;) {
2489 Unit *u;
2490 char name[UNIT_NAME_MAX+2];
2491
2492 /* Start marker */
2493 if (!fgets(name, sizeof(name), f)) {
2494 if (feof(f))
10f8e83c
LP
2495 r = 0;
2496 else
2497 r = -errno;
a16e1123 2498
82c64bf5 2499 goto finish;
a16e1123
LP
2500 }
2501
2502 char_array_0(name);
2503
bd0af849
ZJS
2504 r = manager_load_unit(m, strstrip(name), NULL, NULL, &u);
2505 if (r < 0)
82c64bf5 2506 goto finish;
a16e1123 2507
01e10de3
LP
2508 r = unit_deserialize(u, f, fds);
2509 if (r < 0)
82c64bf5 2510 goto finish;
a16e1123
LP
2511 }
2512
10f8e83c 2513finish:
145b1f79 2514 if (ferror(f))
82c64bf5 2515 r = -EIO;
a16e1123 2516
a7556052
LP
2517 assert(m->n_reloading > 0);
2518 m->n_reloading --;
82c64bf5
LP
2519
2520 return r;
a16e1123
LP
2521}
2522
2523int manager_reload(Manager *m) {
2524 int r, q;
51d122af
ZJS
2525 _cleanup_fclose_ FILE *f = NULL;
2526 _cleanup_fdset_free_ FDSet *fds = NULL;
a16e1123
LP
2527
2528 assert(m);
2529
07719a21
LP
2530 r = manager_open_serialization(m, &f);
2531 if (r < 0)
a16e1123
LP
2532 return r;
2533
a7556052 2534 m->n_reloading ++;
718db961 2535 bus_manager_send_reloading(m, true);
38c52d46 2536
07719a21
LP
2537 fds = fdset_new();
2538 if (!fds) {
a7556052 2539 m->n_reloading --;
51d122af 2540 return -ENOMEM;
a16e1123
LP
2541 }
2542
b3680f49 2543 r = manager_serialize(m, f, fds, false);
07719a21 2544 if (r < 0) {
a7556052 2545 m->n_reloading --;
51d122af 2546 return r;
38c52d46 2547 }
a16e1123
LP
2548
2549 if (fseeko(f, 0, SEEK_SET) < 0) {
a7556052 2550 m->n_reloading --;
51d122af 2551 return -errno;
a16e1123
LP
2552 }
2553
2554 /* From here on there is no way back. */
2555 manager_clear_jobs_and_units(m);
5a1e9937 2556 manager_undo_generators(m);
84e3543e 2557 lookup_paths_free(&m->lookup_paths);
2ded0c04 2558
07719a21 2559 /* Find new unit paths */
e801700e
ZJS
2560 q = manager_run_generators(m);
2561 if (q < 0 && r >= 0)
2562 r = q;
5a1e9937 2563
07719a21
LP
2564 q = lookup_paths_init(
2565 &m->lookup_paths, m->running_as, true,
12ed81d9 2566 NULL,
07719a21
LP
2567 m->generator_unit_path,
2568 m->generator_unit_path_early,
2569 m->generator_unit_path_late);
e801700e 2570 if (q < 0 && r >= 0)
07719a21
LP
2571 r = q;
2572
5a1e9937
LP
2573 manager_build_unit_path_cache(m);
2574
a16e1123 2575 /* First, enumerate what we can from all config files */
07719a21 2576 q = manager_enumerate(m);
e801700e 2577 if (q < 0 && r >= 0)
a16e1123
LP
2578 r = q;
2579
2580 /* Second, deserialize our stored data */
07719a21 2581 q = manager_deserialize(m, f, fds);
e801700e 2582 if (q < 0 && r >= 0)
a16e1123
LP
2583 r = q;
2584
2585 fclose(f);
2586 f = NULL;
2587
a2cc4a6c
ZJS
2588 /* Re-register notify_fd as event source */
2589 q = manager_setup_notify(m);
e801700e 2590 if (q < 0 && r >= 0)
a2cc4a6c
ZJS
2591 r = q;
2592
a16e1123 2593 /* Third, fire things up! */
07719a21 2594 q = manager_coldplug(m);
e801700e 2595 if (q < 0 && r >= 0)
a16e1123
LP
2596 r = q;
2597
a7556052
LP
2598 assert(m->n_reloading > 0);
2599 m->n_reloading--;
9f611ad8 2600
71445ae7
LP
2601 m->send_reloading_done = true;
2602
a16e1123
LP
2603 return r;
2604}
2605
c17ec25e
MS
2606bool manager_is_reloading_or_reexecuting(Manager *m) {
2607 assert(m);
2608
2609 return m->n_reloading != 0;
2610}
2611
fdf20a31 2612void manager_reset_failed(Manager *m) {
5632e374
LP
2613 Unit *u;
2614 Iterator i;
2615
2616 assert(m);
2617
2618 HASHMAP_FOREACH(u, m->units, i)
fdf20a31 2619 unit_reset_failed(u);
5632e374
LP
2620}
2621
31afa0a4 2622bool manager_unit_inactive_or_pending(Manager *m, const char *name) {
8f6df3fa
LP
2623 Unit *u;
2624
2625 assert(m);
2626 assert(name);
2627
2628 /* Returns true if the unit is inactive or going down */
bd0af849
ZJS
2629 u = manager_get_unit(m, name);
2630 if (!u)
8f6df3fa
LP
2631 return true;
2632
31afa0a4 2633 return unit_inactive_or_pending(u);
8f6df3fa
LP
2634}
2635
56dacdbc 2636static void manager_notify_finished(Manager *m) {
7ceba241 2637 char userspace[FORMAT_TIMESPAN_MAX], initrd[FORMAT_TIMESPAN_MAX], kernel[FORMAT_TIMESPAN_MAX], sum[FORMAT_TIMESPAN_MAX];
915b3753 2638 usec_t firmware_usec, loader_usec, kernel_usec, initrd_usec, userspace_usec, total_usec;
b0c918b9 2639
56dacdbc 2640 if (m->test_run)
b0c918b9
LP
2641 return;
2642
67445f4e 2643 if (m->running_as == SYSTEMD_SYSTEM && detect_container(NULL) <= 0) {
e03ae661 2644
915b3753
LP
2645 /* Note that m->kernel_usec.monotonic is always at 0,
2646 * and m->firmware_usec.monotonic and
2647 * m->loader_usec.monotonic should be considered
2648 * negative values. */
2649
7ceba241
LP
2650 firmware_usec = m->firmware_timestamp.monotonic - m->loader_timestamp.monotonic;
2651 loader_usec = m->loader_timestamp.monotonic - m->kernel_timestamp.monotonic;
915b3753 2652 userspace_usec = m->finish_timestamp.monotonic - m->userspace_timestamp.monotonic;
7ceba241 2653 total_usec = m->firmware_timestamp.monotonic + m->finish_timestamp.monotonic;
18fa6b27 2654
e9ddabc2 2655 if (dual_timestamp_is_set(&m->initrd_timestamp)) {
18fa6b27 2656
915b3753
LP
2657 kernel_usec = m->initrd_timestamp.monotonic - m->kernel_timestamp.monotonic;
2658 initrd_usec = m->userspace_timestamp.monotonic - m->initrd_timestamp.monotonic;
18fa6b27 2659
e12919e8 2660 log_struct(LOG_INFO,
e2cc6eca 2661 LOG_MESSAGE_ID(SD_MESSAGE_STARTUP_FINISHED),
e12919e8
LP
2662 "KERNEL_USEC="USEC_FMT, kernel_usec,
2663 "INITRD_USEC="USEC_FMT, initrd_usec,
2664 "USERSPACE_USEC="USEC_FMT, userspace_usec,
e2cc6eca
LP
2665 LOG_MESSAGE("Startup finished in %s (kernel) + %s (initrd) + %s (userspace) = %s.",
2666 format_timespan(kernel, sizeof(kernel), kernel_usec, USEC_PER_MSEC),
2667 format_timespan(initrd, sizeof(initrd), initrd_usec, USEC_PER_MSEC),
2668 format_timespan(userspace, sizeof(userspace), userspace_usec, USEC_PER_MSEC),
2669 format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC)),
e12919e8 2670 NULL);
18fa6b27 2671 } else {
915b3753 2672 kernel_usec = m->userspace_timestamp.monotonic - m->kernel_timestamp.monotonic;
18fa6b27
LP
2673 initrd_usec = 0;
2674
81270860 2675 log_struct(LOG_INFO,
e2cc6eca 2676 LOG_MESSAGE_ID(SD_MESSAGE_STARTUP_FINISHED),
e12919e8 2677 "KERNEL_USEC="USEC_FMT, kernel_usec,
ccd06097 2678 "USERSPACE_USEC="USEC_FMT, userspace_usec,
e2cc6eca
LP
2679 LOG_MESSAGE("Startup finished in %s (kernel) + %s (userspace) = %s.",
2680 format_timespan(kernel, sizeof(kernel), kernel_usec, USEC_PER_MSEC),
2681 format_timespan(userspace, sizeof(userspace), userspace_usec, USEC_PER_MSEC),
2682 format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC)),
81270860 2683 NULL);
e12919e8
LP
2684 }
2685 } else {
2686 firmware_usec = loader_usec = initrd_usec = kernel_usec = 0;
2687 total_usec = userspace_usec = m->finish_timestamp.monotonic - m->userspace_timestamp.monotonic;
2688
2689 log_struct(LOG_INFO,
e2cc6eca 2690 LOG_MESSAGE_ID(SD_MESSAGE_STARTUP_FINISHED),
e12919e8 2691 "USERSPACE_USEC="USEC_FMT, userspace_usec,
e2cc6eca
LP
2692 LOG_MESSAGE("Startup finished in %s.",
2693 format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC)),
e12919e8 2694 NULL);
18fa6b27 2695 }
b0c918b9 2696
718db961 2697 bus_manager_send_finished(m, firmware_usec, loader_usec, kernel_usec, initrd_usec, userspace_usec, total_usec);
530345e7
LP
2698
2699 sd_notifyf(false,
af4ec430
LP
2700 "READY=1\n"
2701 "STATUS=Startup finished in %s.",
2fa4092c 2702 format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC));
b0c918b9
LP
2703}
2704
56dacdbc
ZJS
2705void manager_check_finished(Manager *m) {
2706 Unit *u = NULL;
2707 Iterator i;
2708
2709 assert(m);
2710
56dacdbc
ZJS
2711 if (hashmap_size(m->jobs) > 0) {
2712
2713 if (m->jobs_in_progress_event_source)
2ae56591
ZJS
2714 /* Ignore any failure, this is only for feedback */
2715 (void) sd_event_source_set_time(m->jobs_in_progress_event_source,
2716 now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_WAIT_USEC);
56dacdbc
ZJS
2717
2718 return;
2719 }
2720
2721 manager_flip_auto_status(m, false);
2722
2723 /* Notify Type=idle units that we are done now */
2724 m->idle_pipe_event_source = sd_event_source_unref(m->idle_pipe_event_source);
2725 manager_close_idle_pipe(m);
2726
2727 /* Turn off confirm spawn now */
2728 m->confirm_spawn = false;
2729
2730 /* No need to update ask password status when we're going non-interactive */
2731 manager_close_ask_password(m);
2732
2733 /* This is no longer the first boot */
2734 manager_set_first_boot(m, false);
2735
2736 if (dual_timestamp_is_set(&m->finish_timestamp))
2737 return;
2738
2739 dual_timestamp_get(&m->finish_timestamp);
2740
2741 manager_notify_finished(m);
2742
2743 SET_FOREACH(u, m->startup_units, i)
2744 if (u->cgroup_path)
2745 cgroup_context_apply(unit_get_cgroup_context(u), unit_get_cgroup_mask(u), u->cgroup_path, manager_state(m));
2746}
2747
07719a21
LP
2748static int create_generator_dir(Manager *m, char **generator, const char *name) {
2749 char *p;
2750 int r;
2751
2752 assert(m);
2753 assert(generator);
2754 assert(name);
2755
2756 if (*generator)
2757 return 0;
2758
67445f4e 2759 if (m->running_as == SYSTEMD_SYSTEM && getpid() == 1) {
fcc81ea3 2760 /* systemd --system, not running --test */
07719a21
LP
2761
2762 p = strappend("/run/systemd/", name);
0d0f0c50
SL
2763 if (!p)
2764 return log_oom();
07719a21 2765
fcc81ea3
KS
2766 r = mkdir_p_label(p, 0755);
2767 if (r < 0) {
c33b3297 2768 log_error_errno(r, "Failed to create generator directory %s: %m", p);
fcc81ea3
KS
2769 free(p);
2770 return r;
2771 }
2772 } else if (m->running_as == SYSTEMD_USER) {
2773 const char *s = NULL;
2774
2775 s = getenv("XDG_RUNTIME_DIR");
2776 if (!s)
2777 return -EINVAL;
2778 p = strjoin(s, "/systemd/", name, NULL);
2779 if (!p)
2780 return log_oom();
2781
d2e54fae 2782 r = mkdir_p_label(p, 0755);
07719a21 2783 if (r < 0) {
c33b3297 2784 log_error_errno(r, "Failed to create generator directory %s: %m", p);
07719a21
LP
2785 free(p);
2786 return r;
2787 }
2788 } else {
fcc81ea3
KS
2789 /* systemd --system --test */
2790
b7def684 2791 p = strjoin("/tmp/systemd-", name, ".XXXXXX", NULL);
0d0f0c50
SL
2792 if (!p)
2793 return log_oom();
07719a21
LP
2794
2795 if (!mkdtemp(p)) {
56f64d95 2796 log_error_errno(errno, "Failed to create generator directory %s: %m",
7ad94c71 2797 p);
34bf0281 2798 free(p);
07719a21
LP
2799 return -errno;
2800 }
2801 }
2802
2803 *generator = p;
2804 return 0;
2805}
2806
2807static void trim_generator_dir(Manager *m, char **generator) {
2808 assert(m);
2809 assert(generator);
2810
2811 if (!*generator)
2812 return;
2813
2814 if (rmdir(*generator) >= 0) {
2815 free(*generator);
2816 *generator = NULL;
2817 }
2818
2819 return;
2820}
2821
e801700e
ZJS
2822static int manager_run_generators(Manager *m) {
2823 _cleanup_free_ char **paths = NULL;
07719a21 2824 const char *argv[5];
e801700e 2825 char **path;
07719a21 2826 int r;
5a1e9937
LP
2827
2828 assert(m);
2829
0d8c31ff 2830 if (m->test_run)
e801700e 2831 return 0;
0d8c31ff 2832
e801700e
ZJS
2833 paths = generator_paths(m->running_as);
2834 if (!paths)
2835 return log_oom();
5a1e9937 2836
49681057
ZJS
2837 /* Optimize by skipping the whole process by not creating output directories
2838 * if no generators are found. */
e801700e
ZJS
2839 STRV_FOREACH(path, paths) {
2840 r = access(*path, F_OK);
2841 if (r == 0)
2842 goto found;
49681057 2843 if (errno != ENOENT)
e801700e 2844 log_warning_errno(errno, "Failed to open generator directory %s: %m", *path);
5a1e9937 2845 }
e801700e 2846 return 0;
5a1e9937 2847
e801700e 2848 found:
07719a21
LP
2849 r = create_generator_dir(m, &m->generator_unit_path, "generator");
2850 if (r < 0)
2851 goto finish;
f1d19aa4 2852
07719a21
LP
2853 r = create_generator_dir(m, &m->generator_unit_path_early, "generator.early");
2854 if (r < 0)
2855 goto finish;
5a1e9937 2856
07719a21
LP
2857 r = create_generator_dir(m, &m->generator_unit_path_late, "generator.late");
2858 if (r < 0)
2859 goto finish;
5a1e9937 2860
83cc030f
LP
2861 argv[0] = NULL; /* Leave this empty, execute_directory() will fill something in */
2862 argv[1] = m->generator_unit_path;
07719a21
LP
2863 argv[2] = m->generator_unit_path_early;
2864 argv[3] = m->generator_unit_path_late;
2865 argv[4] = NULL;
5a1e9937 2866
718db961 2867 RUN_WITH_UMASK(0022)
e801700e 2868 execute_directories((const char* const*) paths, DEFAULT_TIMEOUT_USEC, (char**) argv);
5a1e9937 2869
718db961 2870finish:
07719a21
LP
2871 trim_generator_dir(m, &m->generator_unit_path);
2872 trim_generator_dir(m, &m->generator_unit_path_early);
2873 trim_generator_dir(m, &m->generator_unit_path_late);
e801700e 2874 return r;
5a1e9937
LP
2875}
2876
07719a21 2877static void remove_generator_dir(Manager *m, char **generator) {
5a1e9937 2878 assert(m);
07719a21 2879 assert(generator);
5a1e9937 2880
07719a21 2881 if (!*generator)
5a1e9937
LP
2882 return;
2883
07719a21 2884 strv_remove(m->lookup_paths.unit_path, *generator);
c6878637 2885 (void) rm_rf(*generator, REMOVE_ROOT);
5a1e9937 2886
07719a21
LP
2887 free(*generator);
2888 *generator = NULL;
2889}
2890
e801700e 2891static void manager_undo_generators(Manager *m) {
07719a21
LP
2892 assert(m);
2893
2894 remove_generator_dir(m, &m->generator_unit_path);
2895 remove_generator_dir(m, &m->generator_unit_path_early);
2896 remove_generator_dir(m, &m->generator_unit_path_late);
5a1e9937
LP
2897}
2898
718db961
LP
2899int manager_environment_add(Manager *m, char **minus, char **plus) {
2900 char **a = NULL, **b = NULL, **l;
97d0e5f8 2901 assert(m);
bcd8e6d1 2902
718db961 2903 l = m->environment;
bcd8e6d1 2904
718db961
LP
2905 if (!strv_isempty(minus)) {
2906 a = strv_env_delete(l, 1, minus);
2907 if (!a)
2908 return -ENOMEM;
2909
2910 l = a;
2911 }
2912
2913 if (!strv_isempty(plus)) {
2914 b = strv_env_merge(2, l, plus);
aa9f8a30
AH
2915 if (!b) {
2916 strv_free(a);
718db961 2917 return -ENOMEM;
aa9f8a30 2918 }
bcd8e6d1 2919
718db961
LP
2920 l = b;
2921 }
2922
2923 if (m->environment != l)
2924 strv_free(m->environment);
2925 if (a != l)
2926 strv_free(a);
2927 if (b != l)
2928 strv_free(b);
2929
f069efb4
LP
2930 m->environment = l;
2931 manager_clean_environment(m);
2932 strv_sort(m->environment);
2933
97d0e5f8
UTL
2934 return 0;
2935}
2936
c93ff2e9
FC
2937int manager_set_default_rlimits(Manager *m, struct rlimit **default_rlimit) {
2938 int i;
2939
2940 assert(m);
2941
517d56b1 2942 for (i = 0; i < _RLIMIT_MAX; i++) {
07719a21
LP
2943 if (!default_rlimit[i])
2944 continue;
c93ff2e9 2945
07719a21
LP
2946 m->rlimit[i] = newdup(struct rlimit, default_rlimit[i], 1);
2947 if (!m->rlimit[i])
2948 return -ENOMEM;
c93ff2e9
FC
2949 }
2950
2951 return 0;
2952}
2953
4cfa2c99 2954void manager_recheck_journal(Manager *m) {
f1dd0c3f
LP
2955 Unit *u;
2956
2957 assert(m);
2958
67445f4e 2959 if (m->running_as != SYSTEMD_SYSTEM)
f1dd0c3f
LP
2960 return;
2961
731a676c
LP
2962 u = manager_get_unit(m, SPECIAL_JOURNALD_SOCKET);
2963 if (u && SOCKET(u)->state != SOCKET_RUNNING) {
4cfa2c99 2964 log_close_journal();
731a676c 2965 return;
f1dd0c3f
LP
2966 }
2967
731a676c
LP
2968 u = manager_get_unit(m, SPECIAL_JOURNALD_SERVICE);
2969 if (u && SERVICE(u)->state != SERVICE_RUNNING) {
4cfa2c99 2970 log_close_journal();
731a676c
LP
2971 return;
2972 }
f1dd0c3f 2973
731a676c
LP
2974 /* Hmm, OK, so the socket is fully up and the service is up
2975 * too, then let's make use of the thing. */
f1dd0c3f
LP
2976 log_open();
2977}
2978
d450b6f2 2979void manager_set_show_status(Manager *m, ShowStatus mode) {
27d340c7 2980 assert(m);
d450b6f2 2981 assert(IN_SET(mode, SHOW_STATUS_AUTO, SHOW_STATUS_NO, SHOW_STATUS_YES, SHOW_STATUS_TEMPORARY));
27d340c7 2982
67445f4e 2983 if (m->running_as != SYSTEMD_SYSTEM)
27d340c7
LP
2984 return;
2985
d450b6f2 2986 m->show_status = mode;
27d340c7 2987
d450b6f2 2988 if (mode > 0)
27d340c7
LP
2989 touch("/run/systemd/show-status");
2990 else
2991 unlink("/run/systemd/show-status");
2992}
2993
127d5fd1 2994static bool manager_get_show_status(Manager *m, StatusType type) {
27d340c7
LP
2995 assert(m);
2996
67445f4e 2997 if (m->running_as != SYSTEMD_SYSTEM)
27d340c7
LP
2998 return false;
2999
31a7eb86
ZJS
3000 if (m->no_console_output)
3001 return false;
3002
d81afec1 3003 if (!IN_SET(manager_state(m), MANAGER_INITIALIZING, MANAGER_STARTING, MANAGER_STOPPING))
08510627
LP
3004 return false;
3005
e46b13c8 3006 /* If we cannot find out the status properly, just proceed. */
ebc5788e 3007 if (type != STATUS_TYPE_EMERGENCY && manager_check_ask_password(m) > 0)
e46b13c8
ZJS
3008 return false;
3009
d450b6f2 3010 if (m->show_status > 0)
27d340c7
LP
3011 return true;
3012
031886ed 3013 return false;
27d340c7 3014}
68b29a9f 3015
e2680723
LP
3016void manager_set_first_boot(Manager *m, bool b) {
3017 assert(m);
3018
3019 if (m->running_as != SYSTEMD_SYSTEM)
3020 return;
3021
3022 m->first_boot = b;
3023
3024 if (m->first_boot)
3025 touch("/run/systemd/first-boot");
3026 else
3027 unlink("/run/systemd/first-boot");
3028}
3029
127d5fd1 3030void manager_status_printf(Manager *m, StatusType type, const char *status, const char *format, ...) {
25cee550
MS
3031 va_list ap;
3032
cb6531be
ZJS
3033 /* If m is NULL, assume we're after shutdown and let the messages through. */
3034
3035 if (m && !manager_get_show_status(m, type))
25cee550
MS
3036 return;
3037
03b717a3
MS
3038 /* XXX We should totally drop the check for ephemeral here
3039 * and thus effectively make 'Type=idle' pointless. */
cb6531be 3040 if (type == STATUS_TYPE_EPHEMERAL && m && m->n_on_console > 0)
03b717a3
MS
3041 return;
3042
25cee550 3043 va_start(ap, format);
127d5fd1 3044 status_vprintf(status, true, type == STATUS_TYPE_EPHEMERAL, format, ap);
25cee550
MS
3045 va_end(ap);
3046}
3047
a57f7e2c
LP
3048int manager_get_unit_by_path(Manager *m, const char *path, const char *suffix, Unit **_found) {
3049 _cleanup_free_ char *p = NULL;
3050 Unit *found;
3051
3052 assert(m);
3053 assert(path);
3054 assert(suffix);
3055 assert(_found);
3056
3057 p = unit_name_from_path(path, suffix);
3058 if (!p)
3059 return -ENOMEM;
3060
3061 found = manager_get_unit(m, p);
3062 if (!found) {
3063 *_found = NULL;
3064 return 0;
3065 }
3066
3067 *_found = found;
3068 return 1;
3069}
3070
3071Set *manager_get_units_requiring_mounts_for(Manager *m, const char *path) {
3072 char p[strlen(path)+1];
3073
3074 assert(m);
3075 assert(path);
3076
3077 strcpy(p, path);
3078 path_kill_slashes(p);
3079
3080 return hashmap_get(m->units_requiring_mounts_for, streq(p, "/") ? "" : p);
3081}
e66cf1a3
LP
3082
3083const char *manager_get_runtime_prefix(Manager *m) {
f755e3b7 3084 assert(m);
e66cf1a3
LP
3085
3086 return m->running_as == SYSTEMD_SYSTEM ?
3087 "/run" :
3088 getenv("XDG_RUNTIME_DIR");
3089}
f755e3b7 3090
03455c28
LDM
3091void manager_update_failed_units(Manager *m, Unit *u, bool failed) {
3092 unsigned size;
3093
3094 assert(m);
3095 assert(u->manager == m);
3096
3097 size = set_size(m->failed_units);
3098
9fff8981
ZJS
3099 if (failed) {
3100 if (set_put(m->failed_units, u) < 0)
3101 log_oom();
3102 } else
03455c28
LDM
3103 set_remove(m->failed_units, u);
3104
3105 if (set_size(m->failed_units) != size)
3106 bus_manager_send_change_signal(m);
3107}
3108
f755e3b7
LP
3109ManagerState manager_state(Manager *m) {
3110 Unit *u;
3111
3112 assert(m);
3113
3114 /* Did we ever finish booting? If not then we are still starting up */
d81afec1
LP
3115 if (!dual_timestamp_is_set(&m->finish_timestamp)) {
3116
3117 u = manager_get_unit(m, SPECIAL_BASIC_TARGET);
3118 if (!u || !UNIT_IS_ACTIVE_OR_RELOADING(unit_active_state(u)))
3119 return MANAGER_INITIALIZING;
3120
f755e3b7 3121 return MANAGER_STARTING;
d81afec1 3122 }
f755e3b7
LP
3123
3124 /* Is the special shutdown target queued? If so, we are in shutdown state */
3125 u = manager_get_unit(m, SPECIAL_SHUTDOWN_TARGET);
3126 if (u && u->job && IN_SET(u->job->type, JOB_START, JOB_RESTART, JOB_TRY_RESTART, JOB_RELOAD_OR_START))
3127 return MANAGER_STOPPING;
3128
3129 /* Are the rescue or emergency targets active or queued? If so we are in maintenance state */
3130 u = manager_get_unit(m, SPECIAL_RESCUE_TARGET);
3131 if (u && (UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(u)) ||
3132 (u->job && IN_SET(u->job->type, JOB_START, JOB_RESTART, JOB_TRY_RESTART, JOB_RELOAD_OR_START))))
3133 return MANAGER_MAINTENANCE;
3134
3135 u = manager_get_unit(m, SPECIAL_EMERGENCY_TARGET);
3136 if (u && (UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(u)) ||
3137 (u->job && IN_SET(u->job->type, JOB_START, JOB_RESTART, JOB_TRY_RESTART, JOB_RELOAD_OR_START))))
3138 return MANAGER_MAINTENANCE;
3139
3140 /* Are there any failed units? If so, we are in degraded mode */
3141 if (set_size(m->failed_units) > 0)
3142 return MANAGER_DEGRADED;
3143
3144 return MANAGER_RUNNING;
3145}
3146
3147static const char *const manager_state_table[_MANAGER_STATE_MAX] = {
d81afec1 3148 [MANAGER_INITIALIZING] = "initializing",
f755e3b7
LP
3149 [MANAGER_STARTING] = "starting",
3150 [MANAGER_RUNNING] = "running",
3151 [MANAGER_DEGRADED] = "degraded",
3152 [MANAGER_MAINTENANCE] = "maintenance",
3153 [MANAGER_STOPPING] = "stopping",
3154};
3155
3156DEFINE_STRING_TABLE_LOOKUP(manager_state, ManagerState);