]> git.ipfire.org Git - thirdparty/systemd.git/blob - src/core/manager.c
projects / thirdparty / systemd.git / blob
? search:


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