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