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