]> git.ipfire.org Git - thirdparty/systemd.git/blame - src/core/manager.c
projects / thirdparty / systemd.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
util: split out signal-util.[ch] from util.[ch]
[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
LP		 43#include "hashmap.h"
44#include "macro.h"
45#include "strv.h"
16354eff 46#include "log.h"
2a987ee8 47#include "util.h"
49e942b2 48#include "mkdir.h"
ea430986 49#include "ratelimit.h"
e21fea24 50#include "locale-setup.h"
9e2f7c11 51#include "unit-name.h"
1137a57c 52#include "missing.h"
c6878637 53#include "rm-rf.h"
84e3543e 54#include "path-lookup.h"
514f4ef5 55#include "special.h"
d06dacd0 56#include "exit-status.h"
5dc4c17f 57#include "virt.h"
e96d6be7 58#include "watchdog.h"
9eb977db 59#include "path-util.h"
c1165f82 60#include "audit-fd.h"
c51d84dc 61#include "boot-timestamps.h"
8b55b8c4 62#include "env-util.h"
96aad8d1 63#include "bus-common-errors.h"
718db961
LP		 64#include "bus-error.h"
65#include "bus-util.h"
e3dd987c 66#include "bus-kernel.h"
e4746b57 67#include "time-util.h"
0b452006 68#include "process-util.h"
288a74cc 69#include "terminal-util.h"
24882e06
LP		 70#include "signal-util.h"
71#include "dbus.h"
72#include "dbus-unit.h"
73#include "dbus-job.h"
74#include "dbus-manager.h"
75#include "manager.h"
76#include "transaction.h"
60918275 77
03b717a3 78/* Initial delay and the interval for printing status messages about running jobs */
fd08a840
ZJS		 79#define JOBS_IN_PROGRESS_WAIT_USEC (5*USEC_PER_SEC)
80#define JOBS_IN_PROGRESS_PERIOD_USEC (USEC_PER_SEC / 3)
03b717a3
MS		 81#define JOBS_IN_PROGRESS_PERIOD_DIVISOR 3
82
718db961
LP		 83static int manager_dispatch_notify_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);
84static int manager_dispatch_signal_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);
85static int manager_dispatch_time_change_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);
86static int manager_dispatch_idle_pipe_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata);
87static int manager_dispatch_jobs_in_progress(sd_event_source *source, usec_t usec, void *userdata);
752b5905 88static int manager_dispatch_run_queue(sd_event_source *source, void *userdata);
e801700e
ZJS		 89static int manager_run_generators(Manager *m);
90static void manager_undo_generators(Manager *m);
718db961 91
2ae56591 92static void manager_watch_jobs_in_progress(Manager *m) {
e5723c89 93 usec_t next;
cfa9677b 94 int r;
e5723c89 95
718db961 96 assert(m);
03b717a3 97
718db961 98 if (m->jobs_in_progress_event_source)
2ae56591 99 return;
03b717a3 100
e5723c89 101 next = now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_WAIT_USEC;
cfa9677b 102 r = sd_event_add_time(
6a0f1f6d
LP		 103 m->event,
104 &m->jobs_in_progress_event_source,
105 CLOCK_MONOTONIC,
106 next, 0,
107 manager_dispatch_jobs_in_progress, m);
cfa9677b
MM		 108 if (r < 0)
109 return;
7dfbe2e3
TG		 110
111 (void) sd_event_source_set_description(m->jobs_in_progress_event_source, "manager-jobs-in-progress");
03b717a3
MS		 112}
113
718db961 114#define CYLON_BUFFER_EXTRA (2*(sizeof(ANSI_RED_ON)-1) + sizeof(ANSI_HIGHLIGHT_RED_ON)-1 + 2*(sizeof(ANSI_HIGHLIGHT_OFF)-1))
03b717a3 115
03b717a3
MS		 116static void draw_cylon(char buffer[], size_t buflen, unsigned width, unsigned pos) {
117 char *p = buffer;
118
119 assert(buflen >= CYLON_BUFFER_EXTRA + width + 1);
120 assert(pos <= width+1); /* 0 or width+1 mean that the center light is behind the corner */
121
122 if (pos > 1) {
6282c859
MS		 123 if (pos > 2)
124 p = mempset(p, ' ', pos-2);
5052495b 125 p = stpcpy(p, ANSI_RED_ON);
03b717a3
MS		 126 *p++ = '*';
127 }
128
129 if (pos > 0 && pos <= width) {
5052495b 130 p = stpcpy(p, ANSI_HIGHLIGHT_RED_ON);
03b717a3
MS		 131 *p++ = '*';
132 }
133
5052495b 134 p = stpcpy(p, ANSI_HIGHLIGHT_OFF);
03b717a3
MS		 135
136 if (pos < width) {
5052495b 137 p = stpcpy(p, ANSI_RED_ON);
03b717a3 138 *p++ = '*';
6282c859
MS		 139 if (pos < width-1)
140 p = mempset(p, ' ', width-1-pos);
51d122af 141 strcpy(p, ANSI_HIGHLIGHT_OFF);
03b717a3 142 }
03b717a3
MS		 143}
144
cb8ccb22 145void manager_flip_auto_status(Manager *m, bool enable) {
f755e3b7
LP		 146 assert(m);
147
cb8ccb22
ZJS		 148 if (enable) {
149 if (m->show_status == SHOW_STATUS_AUTO)
150 manager_set_show_status(m, SHOW_STATUS_TEMPORARY);
151 } else {
152 if (m->show_status == SHOW_STATUS_TEMPORARY)
153 manager_set_show_status(m, SHOW_STATUS_AUTO);
154 }
155}
156
03b717a3 157static void manager_print_jobs_in_progress(Manager *m) {
718db961 158 _cleanup_free_ char *job_of_n = NULL;
03b717a3
MS		 159 Iterator i;
160 Job *j;
03b717a3
MS		 161 unsigned counter = 0, print_nr;
162 char cylon[6 + CYLON_BUFFER_EXTRA + 1];
163 unsigned cylon_pos;
8bb310c3
ZJS		 164 char time[FORMAT_TIMESPAN_MAX], limit[FORMAT_TIMESPAN_MAX] = "no limit";
165 uint64_t x;
03b717a3 166
718db961 167 assert(m);
9c3349e2 168 assert(m->n_running_jobs > 0);
718db961 169
cb8ccb22 170 manager_flip_auto_status(m, true);
d450b6f2 171
03b717a3
MS		 172 print_nr = (m->jobs_in_progress_iteration / JOBS_IN_PROGRESS_PERIOD_DIVISOR) % m->n_running_jobs;
173
174 HASHMAP_FOREACH(j, m->jobs, i)
175 if (j->state == JOB_RUNNING && counter++ == print_nr)
176 break;
177
e970a72e
MS		 178 /* m->n_running_jobs must be consistent with the contents of m->jobs,
179 * so the above loop must have succeeded in finding j. */
180 assert(counter == print_nr + 1);
51d122af 181 assert(j);
5a82a91a 182
03b717a3
MS		 183 cylon_pos = m->jobs_in_progress_iteration % 14;
184 if (cylon_pos >= 8)
185 cylon_pos = 14 - cylon_pos;
186 draw_cylon(cylon, sizeof(cylon), 6, cylon_pos);
187
8bb310c3
ZJS		 188 m->jobs_in_progress_iteration++;
189
d6483ba7
ZJS		 190 if (m->n_running_jobs > 1) {
191 if (asprintf(&job_of_n, "(%u of %u) ", counter, m->n_running_jobs) < 0)
192 job_of_n = NULL;
193 }
03b717a3 194
8bb310c3
ZJS		 195 format_timespan(time, sizeof(time), now(CLOCK_MONOTONIC) - j->begin_usec, 1*USEC_PER_SEC);
196 if (job_get_timeout(j, &x) > 0)
197 format_timespan(limit, sizeof(limit), x - j->begin_usec, 1*USEC_PER_SEC);
198
127d5fd1 199 manager_status_printf(m, STATUS_TYPE_EPHEMERAL, cylon,
8bb310c3
ZJS		 200 "%sA %s job is running for %s (%s / %s)",
201 strempty(job_of_n),
202 job_type_to_string(j->type),
203 unit_description(j->unit),
204 time, limit);
03b717a3
MS		 205}
206
e46b13c8
ZJS		 207static int have_ask_password(void) {
208 _cleanup_closedir_ DIR *dir;
209
210 dir = opendir("/run/systemd/ask-password");
211 if (!dir) {
212 if (errno == ENOENT)
213 return false;
214 else
215 return -errno;
216 }
217
218 for (;;) {
219 struct dirent *de;
220
221 errno = 0;
222 de = readdir(dir);
223 if (!de && errno != 0)
224 return -errno;
225 if (!de)
226 return false;
227
228 if (startswith(de->d_name, "ask."))
229 return true;
230 }
231}
232
233static int manager_dispatch_ask_password_fd(sd_event_source *source,
234 int fd, uint32_t revents, void *userdata) {
235 Manager *m = userdata;
236
237 assert(m);
238
239 flush_fd(fd);
240
241 m->have_ask_password = have_ask_password();
242 if (m->have_ask_password < 0)
243 /* Log error but continue. Negative have_ask_password
244 * is treated as unknown status. */
c33b3297 245 log_error_errno(m->have_ask_password, "Failed to list /run/systemd/ask-password: %m");
e46b13c8
ZJS		 246
247 return 0;
248}
249
250static void manager_close_ask_password(Manager *m) {
251 assert(m);
252
253 m->ask_password_inotify_fd = safe_close(m->ask_password_inotify_fd);
254 m->ask_password_event_source = sd_event_source_unref(m->ask_password_event_source);
255 m->have_ask_password = -EINVAL;
256}
257
258static int manager_check_ask_password(Manager *m) {
259 int r;
260
261 assert(m);
262
263 if (!m->ask_password_event_source) {
264 assert(m->ask_password_inotify_fd < 0);
265
266 mkdir_p_label("/run/systemd/ask-password", 0755);
267
268 m->ask_password_inotify_fd = inotify_init1(IN_NONBLOCK|IN_CLOEXEC);
4a62c710
MS		 269 if (m->ask_password_inotify_fd < 0)
270 return log_error_errno(errno, "inotify_init1() failed: %m");
e46b13c8
ZJS		 271
272 if (inotify_add_watch(m->ask_password_inotify_fd, "/run/systemd/ask-password", IN_CREATE|IN_DELETE|IN_MOVE) < 0) {
56f64d95 273 log_error_errno(errno, "Failed to add watch on /run/systemd/ask-password: %m");
e46b13c8
ZJS		 274 manager_close_ask_password(m);
275 return -errno;
276 }
277
278 r = sd_event_add_io(m->event, &m->ask_password_event_source,
279 m->ask_password_inotify_fd, EPOLLIN,
280 manager_dispatch_ask_password_fd, m);
281 if (r < 0) {
56f64d95 282 log_error_errno(errno, "Failed to add event source for /run/systemd/ask-password: %m");
e46b13c8
ZJS		 283 manager_close_ask_password(m);
284 return -errno;
285 }
286
7dfbe2e3
TG		 287 (void) sd_event_source_set_description(m->ask_password_event_source, "manager-ask-password");
288
e46b13c8
ZJS		 289 /* Queries might have been added meanwhile... */
290 manager_dispatch_ask_password_fd(m->ask_password_event_source,
291 m->ask_password_inotify_fd, EPOLLIN, m);
292 }
293
294 return m->have_ask_password;
295}
296
31a7eb86 297static int manager_watch_idle_pipe(Manager *m) {
31a7eb86
ZJS		 298 int r;
299
718db961
LP		 300 assert(m);
301
302 if (m->idle_pipe_event_source)
31a7eb86
ZJS		 303 return 0;
304
305 if (m->idle_pipe[2] < 0)
306 return 0;
307
151b9b96 308 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		 309 if (r < 0)
310 return log_error_errno(r, "Failed to watch idle pipe: %m");
31a7eb86 311
7dfbe2e3
TG		 312 (void) sd_event_source_set_description(m->idle_pipe_event_source, "manager-idle-pipe");
313
31a7eb86 314 return 0;
31a7eb86
ZJS		 315}
316
718db961
LP		 317static void manager_close_idle_pipe(Manager *m) {
318 assert(m);
31a7eb86 319
3d94f76c
LP		 320 safe_close_pair(m->idle_pipe);
321 safe_close_pair(m->idle_pipe + 2);
31a7eb86
ZJS		 322}
323
8742514c 324static int manager_setup_time_change(Manager *m) {
718db961 325 int r;
b92bea5d
ZJS		 326
327 /* We only care for the cancellation event, hence we set the
328 * timeout to the latest possible value. */
329 struct itimerspec its = {
330 .it_value.tv_sec = TIME_T_MAX,
331 };
8742514c 332
718db961
LP		 333 assert(m);
334 assert_cc(sizeof(time_t) == sizeof(TIME_T_MAX));
8742514c 335
0d8c31ff
ZJS		 336 if (m->test_run)
337 return 0;
338
8742514c
LP		 339 /* Uses TFD_TIMER_CANCEL_ON_SET to get notifications whenever
340 * CLOCK_REALTIME makes a jump relative to CLOCK_MONOTONIC */
341
718db961 342 m->time_change_fd = timerfd_create(CLOCK_REALTIME, TFD_NONBLOCK|TFD_CLOEXEC);
4a62c710
MS		 343 if (m->time_change_fd < 0)
344 return log_error_errno(errno, "Failed to create timerfd: %m");
8742514c 345
718db961 346 if (timerfd_settime(m->time_change_fd, TFD_TIMER_ABSTIME|TFD_TIMER_CANCEL_ON_SET, &its, NULL) < 0) {
56f64d95 347 log_debug_errno(errno, "Failed to set up TFD_TIMER_CANCEL_ON_SET, ignoring: %m");
03e334a1 348 m->time_change_fd = safe_close(m->time_change_fd);
8742514c
LP		 349 return 0;
350 }
351
151b9b96 352 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		 353 if (r < 0)
354 return log_error_errno(r, "Failed to create time change event source: %m");
8742514c 355
7dfbe2e3
TG		 356 (void) sd_event_source_set_description(m->time_change_event_source, "manager-time-change");
357
8742514c
LP		 358 log_debug("Set up TFD_TIMER_CANCEL_ON_SET timerfd.");
359
360 return 0;
361}
362
80876c20 363static int enable_special_signals(Manager *m) {
718db961 364 _cleanup_close_ int fd = -1;
80876c20
LP		 365
366 assert(m);
367
a41b539e 368 /* Enable that we get SIGINT on control-alt-del. In containers
c9999773
LP		 369 * this will fail with EPERM (older) or EINVAL (newer), so
370 * ignore that. */
371 if (reboot(RB_DISABLE_CAD) < 0 && errno != EPERM && errno != EINVAL)
56f64d95 372 log_warning_errno(errno, "Failed to enable ctrl-alt-del handling: %m");
80876c20 373
a41b539e
LP		 374 fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
375 if (fd < 0) {
376 /* Support systems without virtual console */
377 if (fd != -ENOENT)
56f64d95 378 log_warning_errno(errno, "Failed to open /dev/tty0: %m");
a41b539e 379 } else {
80876c20
LP		 380 /* Enable that we get SIGWINCH on kbrequest */
381 if (ioctl(fd, KDSIGACCEPT, SIGWINCH) < 0)
56f64d95 382 log_warning_errno(errno, "Failed to enable kbrequest handling: %m");
80876c20
LP		 383 }
384
385 return 0;
386}
387
ce578209 388static int manager_setup_signals(Manager *m) {
b92bea5d
ZJS		 389 struct sigaction sa = {
390 .sa_handler = SIG_DFL,
391 .sa_flags = SA_NOCLDSTOP|SA_RESTART,
392 };
718db961
LP		 393 sigset_t mask;
394 int r;
60918275 395
ce578209
LP		 396 assert(m);
397
57c0c30e
LP		 398 assert_se(sigaction(SIGCHLD, &sa, NULL) == 0);
399
4dffec14
LP		 400 /* We make liberal use of realtime signals here. On
401 * Linux/glibc we have 30 of them (with the exception of Linux
402 * on hppa, see below), between SIGRTMIN+0 ... SIGRTMIN+30
403 * (aka SIGRTMAX). */
7d793605 404
4dffec14 405 assert_se(sigemptyset(&mask) == 0);
7d793605
LP		 406 sigset_add_many(&mask,
407 SIGCHLD, /* Child died */
408 SIGTERM, /* Reexecute daemon */
409 SIGHUP, /* Reload configuration */
410 SIGUSR1, /* systemd/upstart: reconnect to D-Bus */
411 SIGUSR2, /* systemd: dump status */
412 SIGINT, /* Kernel sends us this on control-alt-del */
413 SIGWINCH, /* Kernel sends us this on kbrequest (alt-arrowup) */
414 SIGPWR, /* Some kernel drivers and upsd send us this on power failure */
4dffec14 415
7d793605 416 SIGRTMIN+0, /* systemd: start default.target */
0003d1ab 417 SIGRTMIN+1, /* systemd: isolate rescue.target */
7d793605
LP		 418 SIGRTMIN+2, /* systemd: isolate emergency.target */
419 SIGRTMIN+3, /* systemd: start halt.target */
420 SIGRTMIN+4, /* systemd: start poweroff.target */
421 SIGRTMIN+5, /* systemd: start reboot.target */
0003d1ab 422 SIGRTMIN+6, /* systemd: start kexec.target */
4dffec14
LP		 423
424 /* ... space for more special targets ... */
425
0003d1ab
LP		 426 SIGRTMIN+13, /* systemd: Immediate halt */
427 SIGRTMIN+14, /* systemd: Immediate poweroff */
428 SIGRTMIN+15, /* systemd: Immediate reboot */
429 SIGRTMIN+16, /* systemd: Immediate kexec */
4dffec14
LP		 430
431 /* ... space for more immediate system state changes ... */
432
0658666b
LP		 433 SIGRTMIN+20, /* systemd: enable status messages */
434 SIGRTMIN+21, /* systemd: disable status messages */
253ee27a
LP		 435 SIGRTMIN+22, /* systemd: set log level to LOG_DEBUG */
436 SIGRTMIN+23, /* systemd: set log level to LOG_INFO */
600b704e 437 SIGRTMIN+24, /* systemd: Immediate exit (--user only) */
4dffec14
LP		 438
439 /* .. one free signal here ... */
440
441#if !defined(__hppa64__) && !defined(__hppa__)
442 /* Apparently Linux on hppa has fewer RT
443 * signals (SIGRTMAX is SIGRTMIN+25 there),
444 * hence let's not try to make use of them
445 * here. Since these commands are accessible
446 * by different means and only really a safety
447 * net, the missing functionality on hppa
448 * shouldn't matter. */
449
4cfa2c99 450 SIGRTMIN+26, /* systemd: set log target to journal-or-kmsg */
253ee27a
LP		 451 SIGRTMIN+27, /* systemd: set log target to console */
452 SIGRTMIN+28, /* systemd: set log target to kmsg */
ee33e53a 453 SIGRTMIN+29, /* systemd: set log target to syslog-or-kmsg (obsolete) */
4dffec14
LP		 454
455 /* ... one free signal here SIGRTMIN+30 ... */
456#endif
7d793605 457 -1);
ce578209
LP		 458 assert_se(sigprocmask(SIG_SETMASK, &mask, NULL) == 0);
459
718db961
LP		 460 m->signal_fd = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC);
461 if (m->signal_fd < 0)
ce578209
LP		 462 return -errno;
463
151b9b96 464 r = sd_event_add_io(m->event, &m->signal_event_source, m->signal_fd, EPOLLIN, manager_dispatch_signal_fd, m);
718db961
LP		 465 if (r < 0)
466 return r;
ce578209 467
7dfbe2e3
TG		 468 (void) sd_event_source_set_description(m->signal_event_source, "manager-signal");
469
fa28bc2d 470 /* Process signals a bit earlier than the rest of things, but
46849c3f 471 * later than notify_fd processing, so that the notify
fa28bc2d
LP		 472 * processing can still figure out to which process/service a
473 * message belongs, before we reap the process. */
29083707
LP		 474 r = sd_event_source_set_priority(m->signal_event_source, -5);
475 if (r < 0)
476 return r;
477
b2c23da8 478 if (m->running_as == MANAGER_SYSTEM)
80876c20 479 return enable_special_signals(m);
e1414003 480
ce578209
LP		 481 return 0;
482}
483
f069efb4
LP		 484static void manager_clean_environment(Manager *m) {
485 assert(m);
486
487 /* Let's remove some environment variables that we
488 * need ourselves to communicate with our clients */
489 strv_env_unset_many(
490 m->environment,
491 "NOTIFY_SOCKET",
492 "MAINPID",
493 "MANAGERPID",
494 "LISTEN_PID",
495 "LISTEN_FDS",
496 "WATCHDOG_PID",
497 "WATCHDOG_USEC",
498 NULL);
499}
500
e21fea24 501static int manager_default_environment(Manager *m) {
71ecc858
LP		 502 assert(m);
503
b2c23da8 504 if (m->running_as == MANAGER_SYSTEM) {
e21fea24
KS		 505 /* The system manager always starts with a clean
506 * environment for its children. It does not import
507 * the kernel or the parents exported variables.
508 *
509 * The initial passed environ is untouched to keep
510 * /proc/self/environ valid; it is used for tagging
511 * the init process inside containers. */
43638332
ZJS		 512 m->environment = strv_new("PATH=" DEFAULT_PATH,
513 NULL);
e21fea24
KS		 514
515 /* Import locale variables LC_*= from configuration */
516 locale_setup(&m->environment);
43d03a83 517 } else {
e21fea24
KS		 518 /* The user manager passes its own environment
519 * along to its children. */
520 m->environment = strv_copy(environ);
43d03a83
LP		 521 }
522
e21fea24
KS		 523 if (!m->environment)
524 return -ENOMEM;
8b55b8c4 525
f069efb4 526 manager_clean_environment(m);
9d5a3757
LP		 527 strv_sort(m->environment);
528
e21fea24 529 return 0;
71ecc858
LP		 530}
531
f2341e0a 532
b2c23da8 533int manager_new(ManagerRunningAs running_as, bool test_run, Manager **_m) {
f2341e0a 534
b2c23da8
LP		 535 static const char * const unit_log_fields[_MANAGER_RUNNING_AS_MAX] = {
536 [MANAGER_SYSTEM] = "UNIT=",
537 [MANAGER_USER] = "USER_UNIT=",
f2341e0a
LP		 538 };
539
b2c23da8
LP		 540 static const char * const unit_log_format_strings[_MANAGER_RUNNING_AS_MAX] = {
541 [MANAGER_SYSTEM] = "UNIT=%s",
542 [MANAGER_USER] = "USER_UNIT=%s",
f2341e0a
LP		 543 };
544
ce578209 545 Manager *m;
e3dd987c 546 int r;
8e274523
LP		 547
548 assert(_m);
a5dab5ce 549 assert(running_as >= 0);
b2c23da8 550 assert(running_as < _MANAGER_RUNNING_AS_MAX);
ce578209 551
915b3753
LP		 552 m = new0(Manager, 1);
553 if (!m)
8e274523 554 return -ENOMEM;
60918275 555
4f8d551f 556#ifdef ENABLE_EFI
b2c23da8 557 if (running_as == MANAGER_SYSTEM && detect_container(NULL) <= 0)
c51d84dc 558 boot_timestamps(&m->userspace_timestamp, &m->firmware_timestamp, &m->loader_timestamp);
4f8d551f
ZC		 559#endif
560
a5dab5ce 561 m->running_as = running_as;
a16e1123 562 m->exit_code = _MANAGER_EXIT_CODE_INVALID;
bd8f585b 563 m->default_timer_accuracy_usec = USEC_PER_MINUTE;
80876c20 564
f2341e0a
LP		 565 /* Prepare log fields we can use for structured logging */
566 m->unit_log_field = unit_log_fields[running_as];
567 m->unit_log_format_string = unit_log_format_strings[running_as];
568
718db961 569 m->idle_pipe[0] = m->idle_pipe[1] = m->idle_pipe[2] = m->idle_pipe[3] = -1;
8742514c 570
befb6d54 571 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 572 m->current_job_id = 1; /* start as id #1, so that we can leave #0 around as "null-like" value */
9152c765 573
e46b13c8
ZJS		 574 m->ask_password_inotify_fd = -1;
575 m->have_ask_password = -EINVAL; /* we don't know */
576
0d8c31ff
ZJS		 577 m->test_run = test_run;
578
2e5c94b9
LP		 579 /* Reboot immediately if the user hits C-A-D more often than 7x per 2s */
580 RATELIMIT_INIT(m->ctrl_alt_del_ratelimit, 2 * USEC_PER_SEC, 7);
581
e21fea24
KS		 582 r = manager_default_environment(m);
583 if (r < 0)
1137a57c
LP		 584 goto fail;
585
d5099efc 586 r = hashmap_ensure_allocated(&m->units, &string_hash_ops);
718db961 587 if (r < 0)
60918275
LP		 588 goto fail;
589
d5099efc 590 r = hashmap_ensure_allocated(&m->jobs, NULL);
718db961 591 if (r < 0)
60918275
LP		 592 goto fail;
593
d5099efc 594 r = hashmap_ensure_allocated(&m->cgroup_unit, &string_hash_ops);
718db961 595 if (r < 0)
9152c765
LP		 596 goto fail;
597
d5099efc 598 r = hashmap_ensure_allocated(&m->watch_bus, &string_hash_ops);
718db961 599 if (r < 0)
05e343b7
LP		 600 goto fail;
601
d5099efc 602 r = set_ensure_allocated(&m->startup_units, NULL);
95ae05c0
WC		 603 if (r < 0)
604 goto fail;
605
d5099efc 606 r = set_ensure_allocated(&m->failed_units, NULL);
f755e3b7
LP		 607 if (r < 0)
608 goto fail;
609
718db961
LP		 610 r = sd_event_default(&m->event);
611 if (r < 0)
8742514c
LP		 612 goto fail;
613
151b9b96 614 r = sd_event_add_defer(m->event, &m->run_queue_event_source, manager_dispatch_run_queue, m);
752b5905
LP		 615 if (r < 0)
616 goto fail;
617
618 r = sd_event_source_set_priority(m->run_queue_event_source, SD_EVENT_PRIORITY_IDLE);
619 if (r < 0)
620 goto fail;
621
622 r = sd_event_source_set_enabled(m->run_queue_event_source, SD_EVENT_OFF);
623 if (r < 0)
624 goto fail;
625
7dfbe2e3
TG		 626 (void) sd_event_source_set_description(m->run_queue_event_source, "manager-run-queue");
627
8742514c
LP		 628 r = manager_setup_signals(m);
629 if (r < 0)
9152c765
LP		 630 goto fail;
631
8742514c
LP		 632 r = manager_setup_cgroup(m);
633 if (r < 0)
8e274523
LP		 634 goto fail;
635
8742514c
LP		 636 r = manager_setup_time_change(m);
637 if (r < 0)
8c47c732
LP		 638 goto fail;
639
9670d583
LP		 640 m->udev = udev_new();
641 if (!m->udev) {
642 r = -ENOMEM;
643 goto fail;
644 }
645
d86f9d52
LP		 646 /* Note that we set up neither kdbus, nor the notify fd
647 * here. We do that after deserialization, since they might
648 * have gotten serialized across the reexec. */
649
72bc8d00
LP		 650 m->taint_usr = dir_is_empty("/usr") > 0;
651
8e274523
LP		 652 *_m = m;
653 return 0;
60918275
LP		 654
655fail:
656 manager_free(m);
8e274523 657 return r;
60918275
LP		 658}
659
d86f9d52 660static int manager_setup_notify(Manager *m) {
7181dbdb 661 int r;
d86f9d52 662
0d8c31ff
ZJS		 663 if (m->test_run)
664 return 0;
665
d86f9d52
LP		 666 if (m->notify_fd < 0) {
667 _cleanup_close_ int fd = -1;
55836941 668 union sockaddr_union sa = {
7181dbdb
LP		 669 .sa.sa_family = AF_UNIX,
670 };
55836941 671 static const int one = 1;
d86f9d52
LP		 672
673 /* First free all secondary fields */
674 free(m->notify_socket);
675 m->notify_socket = NULL;
676 m->notify_event_source = sd_event_source_unref(m->notify_event_source);
677
678 fd = socket(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC|SOCK_NONBLOCK, 0);
4a62c710
MS		 679 if (fd < 0)
680 return log_error_errno(errno, "Failed to allocate notification socket: %m");
d86f9d52 681
b2c23da8 682 if (m->running_as == MANAGER_SYSTEM)
7181dbdb 683 m->notify_socket = strdup("/run/systemd/notify");
498e87d6 684 else {
7181dbdb 685 const char *e;
d86f9d52 686
7181dbdb
LP		 687 e = getenv("XDG_RUNTIME_DIR");
688 if (!e) {
56f64d95 689 log_error_errno(errno, "XDG_RUNTIME_DIR is not set: %m");
7181dbdb
LP		 690 return -EINVAL;
691 }
692
693 m->notify_socket = strappend(e, "/systemd/notify");
694 }
498e87d6
LP		 695 if (!m->notify_socket)
696 return log_oom();
697
698 (void) mkdir_parents_label(m->notify_socket, 0755);
f0e62e89 699 (void) unlink(m->notify_socket);
7181dbdb
LP		 700
701 strncpy(sa.un.sun_path, m->notify_socket, sizeof(sa.un.sun_path)-1);
702 r = bind(fd, &sa.sa, offsetof(struct sockaddr_un, sun_path) + strlen(sa.un.sun_path));
4a62c710
MS		 703 if (r < 0)
704 return log_error_errno(errno, "bind(%s) failed: %m", sa.un.sun_path);
d86f9d52
LP		 705
706 r = setsockopt(fd, SOL_SOCKET, SO_PASSCRED, &one, sizeof(one));
4a62c710
MS		 707 if (r < 0)
708 return log_error_errno(errno, "SO_PASSCRED failed: %m");
d86f9d52 709
d86f9d52
LP		 710 m->notify_fd = fd;
711 fd = -1;
712
713 log_debug("Using notification socket %s", m->notify_socket);
714 }
715
716 if (!m->notify_event_source) {
151b9b96 717 r = sd_event_add_io(m->event, &m->notify_event_source, m->notify_fd, EPOLLIN, manager_dispatch_notify_fd, m);
895b3a7b
MS		 718 if (r < 0)
719 return log_error_errno(r, "Failed to allocate notify event source: %m");
d86f9d52
LP		 720
721 /* Process signals a bit earlier than SIGCHLD, so that we can
722 * still identify to which service an exit message belongs */
723 r = sd_event_source_set_priority(m->notify_event_source, -7);
23bbb0de
MS		 724 if (r < 0)
725 return log_error_errno(r, "Failed to set priority of notify event source: %m");
7dfbe2e3
TG		 726
727 (void) sd_event_source_set_description(m->notify_event_source, "manager-notify");
d86f9d52
LP		 728 }
729
730 return 0;
731}
732
733static int manager_setup_kdbus(Manager *m) {
5bf348d7 734#ifdef ENABLE_KDBUS
d86f9d52
LP		 735 _cleanup_free_ char *p = NULL;
736
737 assert(m);
738
0d8c31ff 739 if (m->test_run || m->kdbus_fd >= 0)
d86f9d52 740 return 0;
d79acc30
DH		 741 if (!is_kdbus_available())
742 return -ESOCKTNOSUPPORT;
d86f9d52 743
b2c23da8 744 if (m->running_as == MANAGER_SYSTEM && detect_container(NULL) <= 0)
387eba00 745 bus_kernel_fix_attach_mask();
1a299299
LP		 746
747 m->kdbus_fd = bus_kernel_create_bus(
b2c23da8
LP		 748 m->running_as == MANAGER_SYSTEM ? "system" : "user",
749 m->running_as == MANAGER_SYSTEM, &p);
1a299299 750
eb56eb9b
MS		 751 if (m->kdbus_fd < 0)
752 return log_debug_errno(m->kdbus_fd, "Failed to set up kdbus: %m");
d86f9d52
LP		 753
754 log_debug("Successfully set up kdbus on %s", p);
d86f9d52
LP		 755#endif
756
757 return 0;
758}
759
760static int manager_connect_bus(Manager *m, bool reexecuting) {
761 bool try_bus_connect;
762
763 assert(m);
764
0d8c31ff
ZJS		 765 if (m->test_run)
766 return 0;
767
d86f9d52
LP		 768 try_bus_connect =
769 m->kdbus_fd >= 0 ||
770 reexecuting ||
b2c23da8 771 (m->running_as == MANAGER_USER && getenv("DBUS_SESSION_BUS_ADDRESS"));
d86f9d52 772
ff9b60f3 773 /* Try to connect to the buses, if possible. */
d86f9d52
LP		 774 return bus_init(m, try_bus_connect);
775}
776
23a177ef 777static unsigned manager_dispatch_cleanup_queue(Manager *m) {
595ed347 778 Unit *u;
23a177ef
LP		 779 unsigned n = 0;
780
781 assert(m);
782
595ed347
MS		 783 while ((u = m->cleanup_queue)) {
784 assert(u->in_cleanup_queue);
23a177ef 785
595ed347 786 unit_free(u);
23a177ef
LP		 787 n++;
788 }
789
790 return n;
791}
792
eced69b3 793enum {
35b8ca3a 794 GC_OFFSET_IN_PATH, /* This one is on the path we were traveling */
eced69b3
LP		 795 GC_OFFSET_UNSURE, /* No clue */
796 GC_OFFSET_GOOD, /* We still need this unit */
797 GC_OFFSET_BAD, /* We don't need this unit anymore */
798 _GC_OFFSET_MAX
799};
800
801static void unit_gc_sweep(Unit *u, unsigned gc_marker) {
701cc384
LP		 802 Iterator i;
803 Unit *other;
eced69b3 804 bool is_bad;
701cc384
LP		 805
806 assert(u);
807
ac155bb8
MS		 808 if (u->gc_marker == gc_marker + GC_OFFSET_GOOD ||
809 u->gc_marker == gc_marker + GC_OFFSET_BAD ||
810 u->gc_marker == gc_marker + GC_OFFSET_IN_PATH)
701cc384
LP		 811 return;
812
ac155bb8 813 if (u->in_cleanup_queue)
701cc384
LP		 814 goto bad;
815
816 if (unit_check_gc(u))
817 goto good;
818
ac155bb8 819 u->gc_marker = gc_marker + GC_OFFSET_IN_PATH;
eced69b3
LP		 820
821 is_bad = true;
822
ac155bb8 823 SET_FOREACH(other, u->dependencies[UNIT_REFERENCED_BY], i) {
701cc384
LP		 824 unit_gc_sweep(other, gc_marker);
825
ac155bb8 826 if (other->gc_marker == gc_marker + GC_OFFSET_GOOD)
701cc384 827 goto good;
eced69b3 828
ac155bb8 829 if (other->gc_marker != gc_marker + GC_OFFSET_BAD)
eced69b3 830 is_bad = false;
701cc384
LP		 831 }
832
eced69b3
LP		 833 if (is_bad)
834 goto bad;
835
836 /* We were unable to find anything out about this entry, so
837 * let's investigate it later */
ac155bb8 838 u->gc_marker = gc_marker + GC_OFFSET_UNSURE;
eced69b3
LP		 839 unit_add_to_gc_queue(u);
840 return;
841
701cc384 842bad:
eced69b3
LP		 843 /* We definitely know that this one is not useful anymore, so
844 * let's mark it for deletion */
ac155bb8 845 u->gc_marker = gc_marker + GC_OFFSET_BAD;
eced69b3 846 unit_add_to_cleanup_queue(u);
701cc384
LP		 847 return;
848
849good:
ac155bb8 850 u->gc_marker = gc_marker + GC_OFFSET_GOOD;
701cc384
LP		 851}
852
853static unsigned manager_dispatch_gc_queue(Manager *m) {
595ed347 854 Unit *u;
701cc384 855 unsigned n = 0;
eced69b3 856 unsigned gc_marker;
701cc384
LP		 857
858 assert(m);
859
cf1265e1 860 /* log_debug("Running GC..."); */
701cc384 861
eced69b3
LP		 862 m->gc_marker += _GC_OFFSET_MAX;
863 if (m->gc_marker + _GC_OFFSET_MAX <= _GC_OFFSET_MAX)
c9c0cadb 864 m->gc_marker = 1;
701cc384 865
eced69b3
LP		 866 gc_marker = m->gc_marker;
867
595ed347
MS		 868 while ((u = m->gc_queue)) {
869 assert(u->in_gc_queue);
701cc384 870
595ed347 871 unit_gc_sweep(u, gc_marker);
eced69b3 872
71fda00f 873 LIST_REMOVE(gc_queue, m->gc_queue, u);
595ed347 874 u->in_gc_queue = false;
701cc384
LP		 875
876 n++;
877
595ed347
MS		 878 if (u->gc_marker == gc_marker + GC_OFFSET_BAD ||
879 u->gc_marker == gc_marker + GC_OFFSET_UNSURE) {
cc3bc3e6 880 if (u->id)
f2341e0a 881 log_unit_debug(u, "Collecting.");
595ed347
MS		 882 u->gc_marker = gc_marker + GC_OFFSET_BAD;
883 unit_add_to_cleanup_queue(u);
701cc384
LP		 884 }
885 }
886
887 m->n_in_gc_queue = 0;
701cc384
LP		 888
889 return n;
890}
891
a16e1123 892static void manager_clear_jobs_and_units(Manager *m) {
a16e1123 893 Unit *u;
60918275
LP		 894
895 assert(m);
896
87f0e418
LP		 897 while ((u = hashmap_first(m->units)))
898 unit_free(u);
964e0949
LP		 899
900 manager_dispatch_cleanup_queue(m);
901
902 assert(!m->load_queue);
903 assert(!m->run_queue);
904 assert(!m->dbus_unit_queue);
905 assert(!m->dbus_job_queue);
906 assert(!m->cleanup_queue);
907 assert(!m->gc_queue);
908
964e0949
LP		 909 assert(hashmap_isempty(m->jobs));
910 assert(hashmap_isempty(m->units));
9e9e2b72
MS		 911
912 m->n_on_console = 0;
913 m->n_running_jobs = 0;
a16e1123
LP		 914}
915
06d8d842 916Manager* manager_free(Manager *m) {
a16e1123 917 UnitType c;
c93ff2e9 918 int i;
87f0e418 919
06d8d842
ZJS		 920 if (!m)
921 return NULL;
a16e1123
LP		 922
923 manager_clear_jobs_and_units(m);
23a177ef 924
7824bbeb
LP		 925 for (c = 0; c < _UNIT_TYPE_MAX; c++)
926 if (unit_vtable[c]->shutdown)
927 unit_vtable[c]->shutdown(m);
928
a16e1123
LP		 929 /* If we reexecute ourselves, we keep the root cgroup
930 * around */
c6c18be3 931 manager_shutdown_cgroup(m, m->exit_code != MANAGER_REEXECUTE);
8e274523 932
5a1e9937
LP		 933 manager_undo_generators(m);
934
5e8d1c9a 935 bus_done(m);
ea430986 936
87f0e418 937 hashmap_free(m->units);
60918275 938 hashmap_free(m->jobs);
5ba6985b
LP		 939 hashmap_free(m->watch_pids1);
940 hashmap_free(m->watch_pids2);
05e343b7 941 hashmap_free(m->watch_bus);
9152c765 942
95ae05c0 943 set_free(m->startup_units);
f755e3b7
LP		 944 set_free(m->failed_units);
945
718db961
LP		 946 sd_event_source_unref(m->signal_event_source);
947 sd_event_source_unref(m->notify_event_source);
948 sd_event_source_unref(m->time_change_event_source);
949 sd_event_source_unref(m->jobs_in_progress_event_source);
950 sd_event_source_unref(m->idle_pipe_event_source);
752b5905 951 sd_event_source_unref(m->run_queue_event_source);
718db961 952
03e334a1
LP		 953 safe_close(m->signal_fd);
954 safe_close(m->notify_fd);
955 safe_close(m->time_change_fd);
956 safe_close(m->kdbus_fd);
718db961 957
e46b13c8
ZJS		 958 manager_close_ask_password(m);
959
718db961
LP		 960 manager_close_idle_pipe(m);
961
9670d583 962 udev_unref(m->udev);
718db961 963 sd_event_unref(m->event);
60918275 964
c952c6ec
LP		 965 free(m->notify_socket);
966
84e3543e 967 lookup_paths_free(&m->lookup_paths);
1137a57c 968 strv_free(m->environment);
036643a2 969
4ad49000 970 hashmap_free(m->cgroup_unit);
c6c18be3 971 set_free_free(m->unit_path_cache);
33be102a 972
664f88a7
LP		 973 free(m->switch_root);
974 free(m->switch_root_init);
975
517d56b1 976 for (i = 0; i < _RLIMIT_MAX; i++)
c93ff2e9
FC		 977 free(m->rlimit[i]);
978
a57f7e2c
LP		 979 assert(hashmap_isempty(m->units_requiring_mounts_for));
980 hashmap_free(m->units_requiring_mounts_for);
981
60918275 982 free(m);
06d8d842 983 return NULL;
60918275
LP		 984}
985
a16e1123 986int manager_enumerate(Manager *m) {
0faacd47 987 int r = 0;
f50e0a01 988 UnitType c;
f50e0a01
LP		 989
990 assert(m);
991
a16e1123
LP		 992 /* Let's ask every type to load all units from disk/kernel
993 * that it might know */
0faacd47
LP		 994 for (c = 0; c < _UNIT_TYPE_MAX; c++) {
995 int q;
996
1c2e9646 997 if (!unit_type_supported(c)) {
03afec3c 998 log_debug("Unit type .%s is not supported on this system.", unit_type_to_string(c));
0faacd47 999 continue;
a57f7e2c 1000 }
f50e0a01 1001
0faacd47
LP		 1002 if (!unit_vtable[c]->enumerate)
1003 continue;
1004
1005 q = unit_vtable[c]->enumerate(m);
1006 if (q < 0)
1007 r = q;
1008 }
1009
f50e0a01 1010 manager_dispatch_load_queue(m);
a16e1123
LP		 1011 return r;
1012}
1013
007c6337 1014static void manager_coldplug(Manager *m) {
a16e1123
LP		 1015 Iterator i;
1016 Unit *u;
1017 char *k;
007c6337 1018 int r;
a16e1123
LP		 1019
1020 assert(m);
f50e0a01
LP		 1021
1022 /* Then, let's set up their initial state. */
1023 HASHMAP_FOREACH_KEY(u, k, m->units, i) {
1024
1025 /* ignore aliases */
ac155bb8 1026 if (u->id != k)
f50e0a01
LP		 1027 continue;
1028
007c6337
LP		 1029 r = unit_coldplug(u);
1030 if (r < 0)
1031 log_warning_errno(r, "We couldn't coldplug %s, proceeding anyway: %m", u->id);
f50e0a01 1032 }
a16e1123
LP		 1033}
1034
fe51822e
LP		 1035static void manager_build_unit_path_cache(Manager *m) {
1036 char **i;
807d0cca 1037 _cleanup_closedir_ DIR *d = NULL;
fe51822e
LP		 1038 int r;
1039
1040 assert(m);
1041
1042 set_free_free(m->unit_path_cache);
1043
d5099efc 1044 m->unit_path_cache = set_new(&string_hash_ops);
874310b7 1045 if (!m->unit_path_cache) {
fe51822e
LP		 1046 log_error("Failed to allocate unit path cache.");
1047 return;
1048 }
1049
1050 /* This simply builds a list of files we know exist, so that
1051 * we don't always have to go to disk */
1052
1053 STRV_FOREACH(i, m->lookup_paths.unit_path) {
1054 struct dirent *de;
1055
bd0af849
ZJS		 1056 d = opendir(*i);
1057 if (!d) {
874310b7 1058 if (errno != ENOENT)
56f64d95 1059 log_error_errno(errno, "Failed to open directory %s: %m", *i);
fe51822e
LP		 1060 continue;
1061 }
1062
1063 while ((de = readdir(d))) {
1064 char *p;
1065
a34bf9db 1066 if (hidden_file(de->d_name))
fe51822e
LP		 1067 continue;
1068
b7def684 1069 p = strjoin(streq(*i, "/") ? "" : *i, "/", de->d_name, NULL);
44d91056 1070 if (!p) {
fe51822e
LP		 1071 r = -ENOMEM;
1072 goto fail;
1073 }
1074
ef42202a
ZJS		 1075 r = set_consume(m->unit_path_cache, p);
1076 if (r < 0)
fe51822e 1077 goto fail;
fe51822e
LP		 1078 }
1079
1080 closedir(d);
1081 d = NULL;
1082 }
1083
1084 return;
1085
1086fail:
da927ba9 1087 log_error_errno(r, "Failed to build unit path cache: %m");
fe51822e
LP		 1088
1089 set_free_free(m->unit_path_cache);
1090 m->unit_path_cache = NULL;
fe51822e
LP		 1091}
1092
9588bc32
LP		 1093
1094static int manager_distribute_fds(Manager *m, FDSet *fds) {
1095 Unit *u;
1096 Iterator i;
1097 int r;
1098
1099 assert(m);
1100
1101 HASHMAP_FOREACH(u, m->units, i) {
1102
1103 if (fdset_size(fds) <= 0)
1104 break;
1105
1106 if (UNIT_VTABLE(u)->distribute_fds) {
1107 r = UNIT_VTABLE(u)->distribute_fds(u, fds);
1108 if (r < 0)
1109 return r;
1110 }
1111 }
1112
1113 return 0;
1114}
1115
a16e1123
LP		 1116int manager_startup(Manager *m, FILE *serialization, FDSet *fds) {
1117 int r, q;
1118
1119 assert(m);
1120
518d10e9 1121 dual_timestamp_get(&m->generators_start_timestamp);
e801700e 1122 r = manager_run_generators(m);
518d10e9 1123 dual_timestamp_get(&m->generators_finish_timestamp);
e801700e
ZJS		 1124 if (r < 0)
1125 return r;
5a1e9937 1126
07719a21
LP		 1127 r = lookup_paths_init(
1128 &m->lookup_paths, m->running_as, true,
12ed81d9 1129 NULL,
07719a21
LP		 1130 m->generator_unit_path,
1131 m->generator_unit_path_early,
1132 m->generator_unit_path_late);
1133 if (r < 0)
1134 return r;
1135
fe51822e
LP		 1136 manager_build_unit_path_cache(m);
1137
9f611ad8
LP		 1138 /* If we will deserialize make sure that during enumeration
1139 * this is already known, so we increase the counter here
1140 * already */
1141 if (serialization)
a7556052 1142 m->n_reloading ++;
9f611ad8 1143
a16e1123 1144 /* First, enumerate what we can from all config files */
718db961 1145 dual_timestamp_get(&m->units_load_start_timestamp);
a16e1123 1146 r = manager_enumerate(m);
718db961 1147 dual_timestamp_get(&m->units_load_finish_timestamp);
a16e1123
LP		 1148
1149 /* Second, deserialize if there is something to deserialize */
1cd974ed
ZJS		 1150 if (serialization)
1151 r = manager_deserialize(m, serialization, fds);
a16e1123 1152
01e10de3
LP		 1153 /* Any fds left? Find some unit which wants them. This is
1154 * useful to allow container managers to pass some file
1155 * descriptors to us pre-initialized. This enables
1156 * socket-based activation of entire containers. */
1157 if (fdset_size(fds) > 0) {
1158 q = manager_distribute_fds(m, fds);
1cd974ed 1159 if (q < 0 && r == 0)
01e10de3
LP		 1160 r = q;
1161 }
1162
d86f9d52
LP		 1163 /* We might have deserialized the notify fd, but if we didn't
1164 * then let's create the bus now */
1cd974ed
ZJS		 1165 q = manager_setup_notify(m);
1166 if (q < 0 && r == 0)
1167 r = q;
d86f9d52 1168
e3dd987c
LP		 1169 /* We might have deserialized the kdbus control fd, but if we
1170 * didn't, then let's create the bus now. */
1171 manager_setup_kdbus(m);
1172 manager_connect_bus(m, !!serialization);
8f8f05a9 1173 bus_track_coldplug(m, &m->subscribed, &m->deserialized_subscribed);
e3dd987c 1174
a16e1123 1175 /* Third, fire things up! */
007c6337 1176 manager_coldplug(m);
a16e1123 1177
9f611ad8 1178 if (serialization) {
a7556052
LP		 1179 assert(m->n_reloading > 0);
1180 m->n_reloading --;
71445ae7
LP		 1181
1182 /* Let's wait for the UnitNew/JobNew messages being
1183 * sent, before we notify that the reload is
1184 * finished */
1185 m->send_reloading_done = true;
9f611ad8
LP		 1186 }
1187
a16e1123 1188 return r;
f50e0a01
LP		 1189}
1190
718db961 1191int manager_add_job(Manager *m, JobType type, Unit *unit, JobMode mode, bool override, sd_bus_error *e, Job **_ret) {
e5b5ae50 1192 int r;
7527cb52 1193 Transaction *tr;
e5b5ae50
LP		 1194
1195 assert(m);
1196 assert(type < _JOB_TYPE_MAX);
87f0e418 1197 assert(unit);
e5b5ae50 1198 assert(mode < _JOB_MODE_MAX);
60918275 1199
7358dc02
ZJS		 1200 if (mode == JOB_ISOLATE && type != JOB_START)
1201 return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Isolate is only valid for start.");
c497c7a9 1202
7358dc02
ZJS		 1203 if (mode == JOB_ISOLATE && !unit->allow_isolate)
1204 return sd_bus_error_setf(e, BUS_ERROR_NO_ISOLATION, "Operation refused, unit may not be isolated.");
2528a7a6 1205
f2341e0a 1206 log_unit_debug(unit, "Trying to enqueue job %s/%s/%s", unit->id, job_type_to_string(type), job_mode_to_string(mode));
9f04bd52 1207
c6497ccb 1208 type = job_type_collapse(type, unit);
e0209d83 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
f2341e0a 1230 log_unit_debug(unit,
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
7410616c 1326 if (t == _UNIT_TYPE_INVALID || !unit_name_is_valid(name, UNIT_NAME_PLAIN|UNIT_NAME_INSTANCE))
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
5ba6985b 1500 if (UNIT_VTABLE(u)->notify_message)
a354329f 1501 UNIT_VTABLE(u)->notify_message(u, pid, tags, fds);
34959677
TG		 1502 else
1503 log_unit_debug(u, "Got notification message for unit. Ignoring.");
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
f2341e0a 1623 log_unit_debug(u, "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
f2341e0a 1695 log_debug("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)
f2341e0a 1699 log_error("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 1731 log_received_signal(sfsi.ssi_signo == SIGCHLD ||
b2c23da8 1732 (sfsi.ssi_signo == SIGTERM && m->running_as == MANAGER_USER)
4daf54a8
ZJS		 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:
b2c23da8 1743 if (m->running_as == MANAGER_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:
b2c23da8 1753 if (m->running_as == MANAGER_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:
b2c23da8 1779 if (m->running_as == MANAGER_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:
b2c23da8 1786 if (m->running_as == MANAGER_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 1899 case 24:
b2c23da8 1900 if (m->running_as == MANAGER_USER) {
600b704e
LP		 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
b2c23da8 2018 if (m->runtime_watchdog > 0 && m->running_as == MANAGER_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 */
b2c23da8 2044 if (m->runtime_watchdog > 0 && m->running_as == MANAGER_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;
7410616c 2113 int audit_fd, r;
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
b2c23da8 2124 if (m->running_as != MANAGER_SYSTEM)
f1dd0c3f
LP		 2125 return;
2126
ac155bb8 2127 if (u->type != UNIT_SERVICE)
f1dd0c3f
LP		 2128 return;
2129
7410616c
LP		 2130 r = unit_name_to_prefix_and_instance(u->id, &p);
2131 if (r < 0) {
2132 log_error_errno(r, "Failed to extract prefix and instance of unit name: %m");
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
b2c23da8 2161 if (m->running_as != MANAGER_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
b2c23da8 2223 path = m->running_as == MANAGER_SYSTEM ? "/run/systemd" : "/tmp";
8e33886e 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
527b7a42
LP		 2429 r = cunescape(l + 4, UNESCAPE_RELAX, &uce);
2430 if (r < 0)
4a9fd066 2431 goto finish;
4a9fd066
OS		 2432
2433 e = strv_env_set(m->environment, uce);
2434 if (!e) {
2435 r = -ENOMEM;
2436 goto finish;
2437 }
2438
2439 strv_free(m->environment);
2440 m->environment = e;
e3dd987c 2441
d86f9d52
LP		 2442 } else if (startswith(l, "notify-fd=")) {
2443 int fd;
2444
2445 if (safe_atoi(l + 10, &fd) < 0 || fd < 0 || !fdset_contains(fds, fd))
e5035a27 2446 log_debug("Failed to parse notify fd: %s", l + 10);
d86f9d52 2447 else {
03e334a1
LP		 2448 m->notify_event_source = sd_event_source_unref(m->notify_event_source);
2449 safe_close(m->notify_fd);
d86f9d52
LP		 2450 m->notify_fd = fdset_remove(fds, fd);
2451 }
2452
2453 } else if (startswith(l, "notify-socket=")) {
2454 char *n;
2455
2456 n = strdup(l+14);
2457 if (!n) {
2458 r = -ENOMEM;
2459 goto finish;
2460 }
2461
2462 free(m->notify_socket);
2463 m->notify_socket = n;
2464
e3dd987c
LP		 2465 } else if (startswith(l, "kdbus-fd=")) {
2466 int fd;
2467
8bf9fcf4 2468 if (safe_atoi(l + 9, &fd) < 0 || fd < 0 || !fdset_contains(fds, fd))
e5035a27 2469 log_debug("Failed to parse kdbus fd: %s", l + 9);
e3dd987c 2470 else {
03e334a1 2471 safe_close(m->kdbus_fd);
e3dd987c
LP		 2472 m->kdbus_fd = fdset_remove(fds, fd);
2473 }
2474
230314d7
LP		 2475 } else {
2476 int k;
2477
2478 k = bus_track_deserialize_item(&m->deserialized_subscribed, l);
2479 if (k < 0)
2480 log_debug_errno(k, "Failed to deserialize bus tracker object: %m");
2481 else if (k == 0)
2482 log_debug("Unknown serialization item '%s'", l);
2483 }
10f8e83c
LP		 2484 }
2485
a16e1123
LP		 2486 for (;;) {
2487 Unit *u;
2488 char name[UNIT_NAME_MAX+2];
2489
2490 /* Start marker */
2491 if (!fgets(name, sizeof(name), f)) {
2492 if (feof(f))
10f8e83c
LP		 2493 r = 0;
2494 else
2495 r = -errno;
a16e1123 2496
82c64bf5 2497 goto finish;
a16e1123
LP		 2498 }
2499
2500 char_array_0(name);
2501
bd0af849
ZJS		 2502 r = manager_load_unit(m, strstrip(name), NULL, NULL, &u);
2503 if (r < 0)
82c64bf5 2504 goto finish;
a16e1123 2505
01e10de3
LP		 2506 r = unit_deserialize(u, f, fds);
2507 if (r < 0)
82c64bf5 2508 goto finish;
a16e1123
LP		 2509 }
2510
10f8e83c 2511finish:
145b1f79 2512 if (ferror(f))
82c64bf5 2513 r = -EIO;
a16e1123 2514
a7556052
LP		 2515 assert(m->n_reloading > 0);
2516 m->n_reloading --;
82c64bf5
LP		 2517
2518 return r;
a16e1123
LP		 2519}
2520
2521int manager_reload(Manager *m) {
2522 int r, q;
51d122af
ZJS		 2523 _cleanup_fclose_ FILE *f = NULL;
2524 _cleanup_fdset_free_ FDSet *fds = NULL;
a16e1123
LP		 2525
2526 assert(m);
2527
07719a21
LP		 2528 r = manager_open_serialization(m, &f);
2529 if (r < 0)
a16e1123
LP		 2530 return r;
2531
a7556052 2532 m->n_reloading ++;
718db961 2533 bus_manager_send_reloading(m, true);
38c52d46 2534
07719a21
LP		 2535 fds = fdset_new();
2536 if (!fds) {
a7556052 2537 m->n_reloading --;
51d122af 2538 return -ENOMEM;
a16e1123
LP		 2539 }
2540
b3680f49 2541 r = manager_serialize(m, f, fds, false);
07719a21 2542 if (r < 0) {
a7556052 2543 m->n_reloading --;
51d122af 2544 return r;
38c52d46 2545 }
a16e1123
LP		 2546
2547 if (fseeko(f, 0, SEEK_SET) < 0) {
a7556052 2548 m->n_reloading --;
51d122af 2549 return -errno;
a16e1123
LP		 2550 }
2551
2552 /* From here on there is no way back. */
2553 manager_clear_jobs_and_units(m);
5a1e9937 2554 manager_undo_generators(m);
84e3543e 2555 lookup_paths_free(&m->lookup_paths);
2ded0c04 2556
07719a21 2557 /* Find new unit paths */
e801700e
ZJS		 2558 q = manager_run_generators(m);
2559 if (q < 0 && r >= 0)
2560 r = q;
5a1e9937 2561
07719a21
LP		 2562 q = lookup_paths_init(
2563 &m->lookup_paths, m->running_as, true,
12ed81d9 2564 NULL,
07719a21
LP		 2565 m->generator_unit_path,
2566 m->generator_unit_path_early,
2567 m->generator_unit_path_late);
e801700e 2568 if (q < 0 && r >= 0)
07719a21
LP		 2569 r = q;
2570
5a1e9937
LP		 2571 manager_build_unit_path_cache(m);
2572
a16e1123 2573 /* First, enumerate what we can from all config files */
07719a21 2574 q = manager_enumerate(m);
e801700e 2575 if (q < 0 && r >= 0)
a16e1123
LP		 2576 r = q;
2577
2578 /* Second, deserialize our stored data */
07719a21 2579 q = manager_deserialize(m, f, fds);
e801700e 2580 if (q < 0 && r >= 0)
a16e1123
LP		 2581 r = q;
2582
2583 fclose(f);
2584 f = NULL;
2585
a2cc4a6c
ZJS		 2586 /* Re-register notify_fd as event source */
2587 q = manager_setup_notify(m);
e801700e 2588 if (q < 0 && r >= 0)
a2cc4a6c
ZJS		 2589 r = q;
2590
a16e1123 2591 /* Third, fire things up! */
007c6337 2592 manager_coldplug(m);
a16e1123 2593
a7556052
LP		 2594 assert(m->n_reloading > 0);
2595 m->n_reloading--;
9f611ad8 2596
71445ae7
LP		 2597 m->send_reloading_done = true;
2598
a16e1123
LP		 2599 return r;
2600}
2601
c17ec25e
MS		 2602bool manager_is_reloading_or_reexecuting(Manager *m) {
2603 assert(m);
2604
2605 return m->n_reloading != 0;
2606}
2607
fdf20a31 2608void manager_reset_failed(Manager *m) {
5632e374
LP		 2609 Unit *u;
2610 Iterator i;
2611
2612 assert(m);
2613
2614 HASHMAP_FOREACH(u, m->units, i)
fdf20a31 2615 unit_reset_failed(u);
5632e374
LP		 2616}
2617
31afa0a4 2618bool manager_unit_inactive_or_pending(Manager *m, const char *name) {
8f6df3fa
LP		 2619 Unit *u;
2620
2621 assert(m);
2622 assert(name);
2623
2624 /* Returns true if the unit is inactive or going down */
bd0af849
ZJS		 2625 u = manager_get_unit(m, name);
2626 if (!u)
8f6df3fa
LP		 2627 return true;
2628
31afa0a4 2629 return unit_inactive_or_pending(u);
8f6df3fa
LP		 2630}
2631
56dacdbc 2632static void manager_notify_finished(Manager *m) {
7ceba241 2633 char userspace[FORMAT_TIMESPAN_MAX], initrd[FORMAT_TIMESPAN_MAX], kernel[FORMAT_TIMESPAN_MAX], sum[FORMAT_TIMESPAN_MAX];
915b3753 2634 usec_t firmware_usec, loader_usec, kernel_usec, initrd_usec, userspace_usec, total_usec;
b0c918b9 2635
56dacdbc 2636 if (m->test_run)
b0c918b9
LP		 2637 return;
2638
b2c23da8 2639 if (m->running_as == MANAGER_SYSTEM && detect_container(NULL) <= 0) {
e03ae661 2640
915b3753
LP		 2641 /* Note that m->kernel_usec.monotonic is always at 0,
2642 * and m->firmware_usec.monotonic and
2643 * m->loader_usec.monotonic should be considered
2644 * negative values. */
2645
7ceba241
LP		 2646 firmware_usec = m->firmware_timestamp.monotonic - m->loader_timestamp.monotonic;
2647 loader_usec = m->loader_timestamp.monotonic - m->kernel_timestamp.monotonic;
915b3753 2648 userspace_usec = m->finish_timestamp.monotonic - m->userspace_timestamp.monotonic;
7ceba241 2649 total_usec = m->firmware_timestamp.monotonic + m->finish_timestamp.monotonic;
18fa6b27 2650
e9ddabc2 2651 if (dual_timestamp_is_set(&m->initrd_timestamp)) {
18fa6b27 2652
915b3753
LP		 2653 kernel_usec = m->initrd_timestamp.monotonic - m->kernel_timestamp.monotonic;
2654 initrd_usec = m->userspace_timestamp.monotonic - m->initrd_timestamp.monotonic;
18fa6b27 2655
e12919e8 2656 log_struct(LOG_INFO,
e2cc6eca 2657 LOG_MESSAGE_ID(SD_MESSAGE_STARTUP_FINISHED),
e12919e8
LP		 2658 "KERNEL_USEC="USEC_FMT, kernel_usec,
2659 "INITRD_USEC="USEC_FMT, initrd_usec,
2660 "USERSPACE_USEC="USEC_FMT, userspace_usec,
e2cc6eca
LP		 2661 LOG_MESSAGE("Startup finished in %s (kernel) + %s (initrd) + %s (userspace) = %s.",
2662 format_timespan(kernel, sizeof(kernel), kernel_usec, USEC_PER_MSEC),
2663 format_timespan(initrd, sizeof(initrd), initrd_usec, USEC_PER_MSEC),
2664 format_timespan(userspace, sizeof(userspace), userspace_usec, USEC_PER_MSEC),
2665 format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC)),
e12919e8 2666 NULL);
18fa6b27 2667 } else {
915b3753 2668 kernel_usec = m->userspace_timestamp.monotonic - m->kernel_timestamp.monotonic;
18fa6b27
LP		 2669 initrd_usec = 0;
2670
81270860 2671 log_struct(LOG_INFO,
e2cc6eca 2672 LOG_MESSAGE_ID(SD_MESSAGE_STARTUP_FINISHED),
e12919e8 2673 "KERNEL_USEC="USEC_FMT, kernel_usec,
ccd06097 2674 "USERSPACE_USEC="USEC_FMT, userspace_usec,
e2cc6eca
LP		 2675 LOG_MESSAGE("Startup finished in %s (kernel) + %s (userspace) = %s.",
2676 format_timespan(kernel, sizeof(kernel), kernel_usec, USEC_PER_MSEC),
2677 format_timespan(userspace, sizeof(userspace), userspace_usec, USEC_PER_MSEC),
2678 format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC)),
81270860 2679 NULL);
e12919e8
LP		 2680 }
2681 } else {
2682 firmware_usec = loader_usec = initrd_usec = kernel_usec = 0;
2683 total_usec = userspace_usec = m->finish_timestamp.monotonic - m->userspace_timestamp.monotonic;
2684
2685 log_struct(LOG_INFO,
e2cc6eca 2686 LOG_MESSAGE_ID(SD_MESSAGE_STARTUP_FINISHED),
e12919e8 2687 "USERSPACE_USEC="USEC_FMT, userspace_usec,
e2cc6eca
LP		 2688 LOG_MESSAGE("Startup finished in %s.",
2689 format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC)),
e12919e8 2690 NULL);
18fa6b27 2691 }
b0c918b9 2692
718db961 2693 bus_manager_send_finished(m, firmware_usec, loader_usec, kernel_usec, initrd_usec, userspace_usec, total_usec);
530345e7
LP		 2694
2695 sd_notifyf(false,
af4ec430
LP		 2696 "READY=1\n"
2697 "STATUS=Startup finished in %s.",
2fa4092c 2698 format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC));
b0c918b9
LP		 2699}
2700
56dacdbc
ZJS		 2701void manager_check_finished(Manager *m) {
2702 Unit *u = NULL;
2703 Iterator i;
2704
2705 assert(m);
2706
aad1976f
LP		 2707 if (m->n_reloading > 0)
2708 return;
2709
9771b62d
LP		 2710 /* Verify that we are actually running currently. Initially
2711 * the exit code is set to invalid, and during operation it is
2712 * then set to MANAGER_OK */
2713 if (m->exit_code != MANAGER_OK)
2714 return;
2715
56dacdbc
ZJS		 2716 if (hashmap_size(m->jobs) > 0) {
2717
2718 if (m->jobs_in_progress_event_source)
2ae56591
ZJS		 2719 /* Ignore any failure, this is only for feedback */
2720 (void) sd_event_source_set_time(m->jobs_in_progress_event_source,
2721 now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_WAIT_USEC);
56dacdbc
ZJS		 2722
2723 return;
2724 }
2725
2726 manager_flip_auto_status(m, false);
2727
2728 /* Notify Type=idle units that we are done now */
2729 m->idle_pipe_event_source = sd_event_source_unref(m->idle_pipe_event_source);
2730 manager_close_idle_pipe(m);
2731
2732 /* Turn off confirm spawn now */
2733 m->confirm_spawn = false;
2734
2735 /* No need to update ask password status when we're going non-interactive */
2736 manager_close_ask_password(m);
2737
2738 /* This is no longer the first boot */
2739 manager_set_first_boot(m, false);
2740
2741 if (dual_timestamp_is_set(&m->finish_timestamp))
2742 return;
2743
2744 dual_timestamp_get(&m->finish_timestamp);
2745
2746 manager_notify_finished(m);
2747
2748 SET_FOREACH(u, m->startup_units, i)
2749 if (u->cgroup_path)
2750 cgroup_context_apply(unit_get_cgroup_context(u), unit_get_cgroup_mask(u), u->cgroup_path, manager_state(m));
2751}
2752
07719a21
LP		 2753static int create_generator_dir(Manager *m, char **generator, const char *name) {
2754 char *p;
2755 int r;
2756
2757 assert(m);
2758 assert(generator);
2759 assert(name);
2760
2761 if (*generator)
2762 return 0;
2763
b2c23da8 2764 if (m->running_as == MANAGER_SYSTEM && getpid() == 1) {
fcc81ea3 2765 /* systemd --system, not running --test */
07719a21
LP		 2766
2767 p = strappend("/run/systemd/", name);
0d0f0c50
SL		 2768 if (!p)
2769 return log_oom();
07719a21 2770
fcc81ea3
KS		 2771 r = mkdir_p_label(p, 0755);
2772 if (r < 0) {
c33b3297 2773 log_error_errno(r, "Failed to create generator directory %s: %m", p);
fcc81ea3
KS		 2774 free(p);
2775 return r;
2776 }
b2c23da8 2777 } else if (m->running_as == MANAGER_USER) {
fcc81ea3
KS		 2778 const char *s = NULL;
2779
2780 s = getenv("XDG_RUNTIME_DIR");
2781 if (!s)
2782 return -EINVAL;
2783 p = strjoin(s, "/systemd/", name, NULL);
2784 if (!p)
2785 return log_oom();
2786
d2e54fae 2787 r = mkdir_p_label(p, 0755);
07719a21 2788 if (r < 0) {
c33b3297 2789 log_error_errno(r, "Failed to create generator directory %s: %m", p);
07719a21
LP		 2790 free(p);
2791 return r;
2792 }
2793 } else {
fcc81ea3
KS		 2794 /* systemd --system --test */
2795
b7def684 2796 p = strjoin("/tmp/systemd-", name, ".XXXXXX", NULL);
0d0f0c50
SL		 2797 if (!p)
2798 return log_oom();
07719a21
LP		 2799
2800 if (!mkdtemp(p)) {
56f64d95 2801 log_error_errno(errno, "Failed to create generator directory %s: %m",
7ad94c71 2802 p);
34bf0281 2803 free(p);
07719a21
LP		 2804 return -errno;
2805 }
2806 }
2807
2808 *generator = p;
2809 return 0;
2810}
2811
2812static void trim_generator_dir(Manager *m, char **generator) {
2813 assert(m);
2814 assert(generator);
2815
2816 if (!*generator)
2817 return;
2818
2819 if (rmdir(*generator) >= 0) {
2820 free(*generator);
2821 *generator = NULL;
2822 }
2823
2824 return;
2825}
2826
e801700e 2827static int manager_run_generators(Manager *m) {
f42348ac 2828 _cleanup_strv_free_ char **paths = NULL;
07719a21 2829 const char *argv[5];
e801700e 2830 char **path;
07719a21 2831 int r;
5a1e9937
LP		 2832
2833 assert(m);
2834
0d8c31ff 2835 if (m->test_run)
e801700e 2836 return 0;
0d8c31ff 2837
e801700e
ZJS		 2838 paths = generator_paths(m->running_as);
2839 if (!paths)
2840 return log_oom();
5a1e9937 2841
49681057
ZJS		 2842 /* Optimize by skipping the whole process by not creating output directories
2843 * if no generators are found. */
e801700e
ZJS		 2844 STRV_FOREACH(path, paths) {
2845 r = access(*path, F_OK);
2846 if (r == 0)
2847 goto found;
49681057 2848 if (errno != ENOENT)
e801700e 2849 log_warning_errno(errno, "Failed to open generator directory %s: %m", *path);
5a1e9937 2850 }
e801700e 2851 return 0;
5a1e9937 2852
e801700e 2853 found:
07719a21
LP		 2854 r = create_generator_dir(m, &m->generator_unit_path, "generator");
2855 if (r < 0)
2856 goto finish;
f1d19aa4 2857
07719a21
LP		 2858 r = create_generator_dir(m, &m->generator_unit_path_early, "generator.early");
2859 if (r < 0)
2860 goto finish;
5a1e9937 2861
07719a21
LP		 2862 r = create_generator_dir(m, &m->generator_unit_path_late, "generator.late");
2863 if (r < 0)
2864 goto finish;
5a1e9937 2865
83cc030f
LP		 2866 argv[0] = NULL; /* Leave this empty, execute_directory() will fill something in */
2867 argv[1] = m->generator_unit_path;
07719a21
LP		 2868 argv[2] = m->generator_unit_path_early;
2869 argv[3] = m->generator_unit_path_late;
2870 argv[4] = NULL;
5a1e9937 2871
718db961 2872 RUN_WITH_UMASK(0022)
e801700e 2873 execute_directories((const char* const*) paths, DEFAULT_TIMEOUT_USEC, (char**) argv);
5a1e9937 2874
718db961 2875finish:
07719a21
LP		 2876 trim_generator_dir(m, &m->generator_unit_path);
2877 trim_generator_dir(m, &m->generator_unit_path_early);
2878 trim_generator_dir(m, &m->generator_unit_path_late);
e801700e 2879 return r;
5a1e9937
LP		 2880}
2881
07719a21 2882static void remove_generator_dir(Manager *m, char **generator) {
5a1e9937 2883 assert(m);
07719a21 2884 assert(generator);
5a1e9937 2885
07719a21 2886 if (!*generator)
5a1e9937
LP		 2887 return;
2888
07719a21 2889 strv_remove(m->lookup_paths.unit_path, *generator);
c6878637 2890 (void) rm_rf(*generator, REMOVE_ROOT);
5a1e9937 2891
07719a21
LP		 2892 free(*generator);
2893 *generator = NULL;
2894}
2895
e801700e 2896static void manager_undo_generators(Manager *m) {
07719a21
LP		 2897 assert(m);
2898
2899 remove_generator_dir(m, &m->generator_unit_path);
2900 remove_generator_dir(m, &m->generator_unit_path_early);
2901 remove_generator_dir(m, &m->generator_unit_path_late);
5a1e9937
LP		 2902}
2903
718db961
LP		 2904int manager_environment_add(Manager *m, char **minus, char **plus) {
2905 char **a = NULL, **b = NULL, **l;
97d0e5f8 2906 assert(m);
bcd8e6d1 2907
718db961 2908 l = m->environment;
bcd8e6d1 2909
718db961
LP		 2910 if (!strv_isempty(minus)) {
2911 a = strv_env_delete(l, 1, minus);
2912 if (!a)
2913 return -ENOMEM;
2914
2915 l = a;
2916 }
2917
2918 if (!strv_isempty(plus)) {
2919 b = strv_env_merge(2, l, plus);
aa9f8a30
AH		 2920 if (!b) {
2921 strv_free(a);
718db961 2922 return -ENOMEM;
aa9f8a30 2923 }
bcd8e6d1 2924
718db961
LP		 2925 l = b;
2926 }
2927
2928 if (m->environment != l)
2929 strv_free(m->environment);
2930 if (a != l)
2931 strv_free(a);
2932 if (b != l)
2933 strv_free(b);
2934
f069efb4
LP		 2935 m->environment = l;
2936 manager_clean_environment(m);
2937 strv_sort(m->environment);
2938
97d0e5f8
UTL		 2939 return 0;
2940}
2941
c93ff2e9
FC		 2942int manager_set_default_rlimits(Manager *m, struct rlimit **default_rlimit) {
2943 int i;
2944
2945 assert(m);
2946
517d56b1 2947 for (i = 0; i < _RLIMIT_MAX; i++) {
07719a21
LP		 2948 if (!default_rlimit[i])
2949 continue;
c93ff2e9 2950
07719a21
LP		 2951 m->rlimit[i] = newdup(struct rlimit, default_rlimit[i], 1);
2952 if (!m->rlimit[i])
2953 return -ENOMEM;
c93ff2e9
FC		 2954 }
2955
2956 return 0;
2957}
2958
4cfa2c99 2959void manager_recheck_journal(Manager *m) {
f1dd0c3f
LP		 2960 Unit *u;
2961
2962 assert(m);
2963
b2c23da8 2964 if (m->running_as != MANAGER_SYSTEM)
f1dd0c3f
LP		 2965 return;
2966
731a676c
LP		 2967 u = manager_get_unit(m, SPECIAL_JOURNALD_SOCKET);
2968 if (u && SOCKET(u)->state != SOCKET_RUNNING) {
4cfa2c99 2969 log_close_journal();
731a676c 2970 return;
f1dd0c3f
LP		 2971 }
2972
731a676c
LP		 2973 u = manager_get_unit(m, SPECIAL_JOURNALD_SERVICE);
2974 if (u && SERVICE(u)->state != SERVICE_RUNNING) {
4cfa2c99 2975 log_close_journal();
731a676c
LP		 2976 return;
2977 }
f1dd0c3f 2978
731a676c
LP		 2979 /* Hmm, OK, so the socket is fully up and the service is up
2980 * too, then let's make use of the thing. */
f1dd0c3f
LP		 2981 log_open();
2982}
2983
d450b6f2 2984void manager_set_show_status(Manager *m, ShowStatus mode) {
27d340c7 2985 assert(m);
d450b6f2 2986 assert(IN_SET(mode, SHOW_STATUS_AUTO, SHOW_STATUS_NO, SHOW_STATUS_YES, SHOW_STATUS_TEMPORARY));
27d340c7 2987
b2c23da8 2988 if (m->running_as != MANAGER_SYSTEM)
27d340c7
LP		 2989 return;
2990
d450b6f2 2991 m->show_status = mode;
27d340c7 2992
d450b6f2 2993 if (mode > 0)
27d340c7
LP		 2994 touch("/run/systemd/show-status");
2995 else
2996 unlink("/run/systemd/show-status");
2997}
2998
127d5fd1 2999static bool manager_get_show_status(Manager *m, StatusType type) {
27d340c7
LP		 3000 assert(m);
3001
b2c23da8 3002 if (m->running_as != MANAGER_SYSTEM)
27d340c7
LP		 3003 return false;
3004
31a7eb86
ZJS		 3005 if (m->no_console_output)
3006 return false;
3007
d81afec1 3008 if (!IN_SET(manager_state(m), MANAGER_INITIALIZING, MANAGER_STARTING, MANAGER_STOPPING))
08510627
LP		 3009 return false;
3010
e46b13c8 3011 /* If we cannot find out the status properly, just proceed. */
ebc5788e 3012 if (type != STATUS_TYPE_EMERGENCY && manager_check_ask_password(m) > 0)
e46b13c8
ZJS		 3013 return false;
3014
d450b6f2 3015 if (m->show_status > 0)
27d340c7
LP		 3016 return true;
3017
031886ed 3018 return false;
27d340c7 3019}
68b29a9f 3020
e2680723
LP		 3021void manager_set_first_boot(Manager *m, bool b) {
3022 assert(m);
3023
b2c23da8 3024 if (m->running_as != MANAGER_SYSTEM)
e2680723
LP		 3025 return;
3026
3027 m->first_boot = b;
3028
3029 if (m->first_boot)
3030 touch("/run/systemd/first-boot");
3031 else
3032 unlink("/run/systemd/first-boot");
3033}
3034
127d5fd1 3035void manager_status_printf(Manager *m, StatusType type, const char *status, const char *format, ...) {
25cee550
MS		 3036 va_list ap;
3037
cb6531be
ZJS		 3038 /* If m is NULL, assume we're after shutdown and let the messages through. */
3039
3040 if (m && !manager_get_show_status(m, type))
25cee550
MS		 3041 return;
3042
03b717a3
MS		 3043 /* XXX We should totally drop the check for ephemeral here
3044 * and thus effectively make 'Type=idle' pointless. */
cb6531be 3045 if (type == STATUS_TYPE_EPHEMERAL && m && m->n_on_console > 0)
03b717a3
MS		 3046 return;
3047
25cee550 3048 va_start(ap, format);
127d5fd1 3049 status_vprintf(status, true, type == STATUS_TYPE_EPHEMERAL, format, ap);
25cee550
MS		 3050 va_end(ap);
3051}
3052
a57f7e2c
LP		 3053int manager_get_unit_by_path(Manager *m, const char *path, const char *suffix, Unit **_found) {
3054 _cleanup_free_ char *p = NULL;
3055 Unit *found;
7410616c 3056 int r;
a57f7e2c
LP		 3057
3058 assert(m);
3059 assert(path);
3060 assert(suffix);
3061 assert(_found);
3062
7410616c
LP		 3063 r = unit_name_from_path(path, suffix, &p);
3064 if (r < 0)
3065 return r;
a57f7e2c
LP		 3066
3067 found = manager_get_unit(m, p);
3068 if (!found) {
3069 *_found = NULL;
3070 return 0;
3071 }
3072
3073 *_found = found;
3074 return 1;
3075}
3076
3077Set *manager_get_units_requiring_mounts_for(Manager *m, const char *path) {
3078 char p[strlen(path)+1];
3079
3080 assert(m);
3081 assert(path);
3082
3083 strcpy(p, path);
3084 path_kill_slashes(p);
3085
3086 return hashmap_get(m->units_requiring_mounts_for, streq(p, "/") ? "" : p);
3087}
e66cf1a3
LP		 3088
3089const char *manager_get_runtime_prefix(Manager *m) {
f755e3b7 3090 assert(m);
e66cf1a3 3091
b2c23da8 3092 return m->running_as == MANAGER_SYSTEM ?
e66cf1a3
LP		 3093 "/run" :
3094 getenv("XDG_RUNTIME_DIR");
3095}
f755e3b7 3096
03455c28
LDM		 3097void manager_update_failed_units(Manager *m, Unit *u, bool failed) {
3098 unsigned size;
3099
3100 assert(m);
3101 assert(u->manager == m);
3102
3103 size = set_size(m->failed_units);
3104
9fff8981
ZJS		 3105 if (failed) {
3106 if (set_put(m->failed_units, u) < 0)
3107 log_oom();
3108 } else
03455c28
LDM		 3109 set_remove(m->failed_units, u);
3110
3111 if (set_size(m->failed_units) != size)
3112 bus_manager_send_change_signal(m);
3113}
3114
f755e3b7
LP		 3115ManagerState manager_state(Manager *m) {
3116 Unit *u;
3117
3118 assert(m);
3119
3120 /* Did we ever finish booting? If not then we are still starting up */
d81afec1
LP		 3121 if (!dual_timestamp_is_set(&m->finish_timestamp)) {
3122
3123 u = manager_get_unit(m, SPECIAL_BASIC_TARGET);
3124 if (!u || !UNIT_IS_ACTIVE_OR_RELOADING(unit_active_state(u)))
3125 return MANAGER_INITIALIZING;
3126
f755e3b7 3127 return MANAGER_STARTING;
d81afec1 3128 }
f755e3b7
LP		 3129
3130 /* Is the special shutdown target queued? If so, we are in shutdown state */
3131 u = manager_get_unit(m, SPECIAL_SHUTDOWN_TARGET);
3132 if (u && u->job && IN_SET(u->job->type, JOB_START, JOB_RESTART, JOB_TRY_RESTART, JOB_RELOAD_OR_START))
3133 return MANAGER_STOPPING;
3134
3135 /* Are the rescue or emergency targets active or queued? If so we are in maintenance state */
3136 u = manager_get_unit(m, SPECIAL_RESCUE_TARGET);
3137 if (u && (UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(u)) ||
3138 (u->job && IN_SET(u->job->type, JOB_START, JOB_RESTART, JOB_TRY_RESTART, JOB_RELOAD_OR_START))))
3139 return MANAGER_MAINTENANCE;
3140
3141 u = manager_get_unit(m, SPECIAL_EMERGENCY_TARGET);
3142 if (u && (UNIT_IS_ACTIVE_OR_ACTIVATING(unit_active_state(u)) ||
3143 (u->job && IN_SET(u->job->type, JOB_START, JOB_RESTART, JOB_TRY_RESTART, JOB_RELOAD_OR_START))))
3144 return MANAGER_MAINTENANCE;
3145
3146 /* Are there any failed units? If so, we are in degraded mode */
3147 if (set_size(m->failed_units) > 0)
3148 return MANAGER_DEGRADED;
3149
3150 return MANAGER_RUNNING;
3151}
3152
3153static const char *const manager_state_table[_MANAGER_STATE_MAX] = {
d81afec1 3154 [MANAGER_INITIALIZING] = "initializing",
f755e3b7
LP		 3155 [MANAGER_STARTING] = "starting",
3156 [MANAGER_RUNNING] = "running",
3157 [MANAGER_DEGRADED] = "degraded",
3158 [MANAGER_MAINTENANCE] = "maintenance",
3159 [MANAGER_STOPPING] = "stopping",
3160};
3161
3162DEFINE_STRING_TABLE_LOOKUP(manager_state, ManagerState);
Unnamed repository; edit this file 'description' to name the repository.