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