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