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