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