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