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