]>
Commit | Line | Data |
---|---|---|
53e1b683 | 1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
a7334b09 | 2 | |
60918275 | 3 | #include <errno.h> |
400f1a33 LP |
4 | #include <fcntl.h> |
5 | #include <linux/kd.h> | |
9152c765 | 6 | #include <signal.h> |
713f6f90 | 7 | #include <stdio_ext.h> |
400f1a33 | 8 | #include <string.h> |
e46b13c8 | 9 | #include <sys/epoll.h> |
400f1a33 | 10 | #include <sys/inotify.h> |
e1414003 | 11 | #include <sys/ioctl.h> |
400f1a33 | 12 | #include <sys/reboot.h> |
8742514c | 13 | #include <sys/timerfd.h> |
400f1a33 LP |
14 | #include <sys/wait.h> |
15 | #include <unistd.h> | |
830f6caa | 16 | |
349cc4a5 | 17 | #if HAVE_AUDIT |
4927fcae | 18 | #include <libaudit.h> |
830f6caa | 19 | #endif |
60918275 | 20 | |
718db961 | 21 | #include "sd-daemon.h" |
718db961 | 22 | #include "sd-messages.h" |
3536f49e | 23 | #include "sd-path.h" |
81527be1 | 24 | |
b5efdb8a | 25 | #include "alloc-util.h" |
57b7a260 | 26 | #include "all-units.h" |
400f1a33 LP |
27 | #include "audit-fd.h" |
28 | #include "boot-timestamps.h" | |
29 | #include "bus-common-errors.h" | |
30 | #include "bus-error.h" | |
31 | #include "bus-kernel.h" | |
32 | #include "bus-util.h" | |
00d9ef85 | 33 | #include "clean-ipc.h" |
af6b0ecc | 34 | #include "clock-util.h" |
400f1a33 LP |
35 | #include "dbus-job.h" |
36 | #include "dbus-manager.h" | |
37 | #include "dbus-unit.h" | |
38 | #include "dbus.h" | |
d063a527 | 39 | #include "dirent-util.h" |
400f1a33 | 40 | #include "env-util.h" |
4f5dd394 | 41 | #include "escape.h" |
89711996 | 42 | #include "exec-util.h" |
d3070fbd | 43 | #include "execute.h" |
400f1a33 | 44 | #include "exit-status.h" |
3ffd4af2 | 45 | #include "fd-util.h" |
0d39fa9c | 46 | #include "fileio.h" |
f4f15635 | 47 | #include "fs-util.h" |
60918275 | 48 | #include "hashmap.h" |
c004493c | 49 | #include "io-util.h" |
d3070fbd | 50 | #include "label.h" |
400f1a33 | 51 | #include "locale-setup.h" |
16354eff | 52 | #include "log.h" |
400f1a33 | 53 | #include "macro.h" |
3ffd4af2 | 54 | #include "manager.h" |
400f1a33 | 55 | #include "missing.h" |
49e942b2 | 56 | #include "mkdir.h" |
6bedfcbb | 57 | #include "parse-util.h" |
400f1a33 LP |
58 | #include "path-lookup.h" |
59 | #include "path-util.h" | |
60 | #include "process-util.h" | |
ea430986 | 61 | #include "ratelimit.h" |
31ce987c | 62 | #include "rlimit-util.h" |
c6878637 | 63 | #include "rm-rf.h" |
400f1a33 | 64 | #include "signal-util.h" |
57b7a260 | 65 | #include "socket-util.h" |
514f4ef5 | 66 | #include "special.h" |
8fcde012 | 67 | #include "stat-util.h" |
8b43440b | 68 | #include "string-table.h" |
07630cea | 69 | #include "string-util.h" |
400f1a33 | 70 | #include "strv.h" |
dd1db3c2 | 71 | #include "strxcpyx.h" |
a6ecbf83 | 72 | #include "syslog-util.h" |
400f1a33 LP |
73 | #include "terminal-util.h" |
74 | #include "time-util.h" | |
75 | #include "transaction.h" | |
affb60b1 | 76 | #include "umask-util.h" |
400f1a33 | 77 | #include "unit-name.h" |
00d9ef85 | 78 | #include "user-util.h" |
400f1a33 | 79 | #include "util.h" |
5dc4c17f | 80 | #include "virt.h" |
e96d6be7 | 81 | #include "watchdog.h" |
60918275 | 82 | |
a47806fa | 83 | #define NOTIFY_RCVBUF_SIZE (8*1024*1024) |
d8fdc620 | 84 | #define CGROUPS_AGENT_RCVBUF_SIZE (8*1024*1024) |
a47806fa | 85 | |
03b717a3 | 86 | /* Initial delay and the interval for printing status messages about running jobs */ |
fd08a840 ZJS |
87 | #define JOBS_IN_PROGRESS_WAIT_USEC (5*USEC_PER_SEC) |
88 | #define JOBS_IN_PROGRESS_PERIOD_USEC (USEC_PER_SEC / 3) | |
03b717a3 MS |
89 | #define JOBS_IN_PROGRESS_PERIOD_DIVISOR 3 |
90 | ||
e0a08581 LP |
91 | /* If there are more than 1K bus messages queue across our API and direct busses, then let's not add more on top until |
92 | * the queue gets more empty. */ | |
93 | #define MANAGER_BUS_BUSY_THRESHOLD 1024LU | |
94 | ||
95 | /* How many units and jobs to process of the bus queue before returning to the event loop. */ | |
96 | #define MANAGER_BUS_MESSAGE_BUDGET 100U | |
97 | ||
718db961 | 98 | static int manager_dispatch_notify_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata); |
d8fdc620 | 99 | static int manager_dispatch_cgroups_agent_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata); |
718db961 LP |
100 | static int manager_dispatch_signal_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata); |
101 | static int manager_dispatch_time_change_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata); | |
102 | static int manager_dispatch_idle_pipe_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata); | |
00d9ef85 | 103 | static int manager_dispatch_user_lookup_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata); |
718db961 | 104 | static int manager_dispatch_jobs_in_progress(sd_event_source *source, usec_t usec, void *userdata); |
752b5905 | 105 | static int manager_dispatch_run_queue(sd_event_source *source, void *userdata); |
575b300b | 106 | static int manager_dispatch_sigchld(sd_event_source *source, void *userdata); |
bbf5fd8e | 107 | static int manager_dispatch_timezone_change(sd_event_source *source, const struct inotify_event *event, void *userdata); |
64691d20 | 108 | static int manager_run_environment_generators(Manager *m); |
e801700e | 109 | static int manager_run_generators(Manager *m); |
718db961 | 110 | |
2ae56591 | 111 | static void manager_watch_jobs_in_progress(Manager *m) { |
e5723c89 | 112 | usec_t next; |
cfa9677b | 113 | int r; |
e5723c89 | 114 | |
718db961 | 115 | assert(m); |
03b717a3 | 116 | |
42bf1ae1 FB |
117 | /* We do not want to show the cylon animation if the user |
118 | * needs to confirm service executions otherwise confirmation | |
119 | * messages will be screwed by the cylon animation. */ | |
b0eb2944 | 120 | if (!manager_is_confirm_spawn_disabled(m)) |
42bf1ae1 FB |
121 | return; |
122 | ||
718db961 | 123 | if (m->jobs_in_progress_event_source) |
2ae56591 | 124 | return; |
03b717a3 | 125 | |
e5723c89 | 126 | next = now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_WAIT_USEC; |
cfa9677b | 127 | r = sd_event_add_time( |
6a0f1f6d LP |
128 | m->event, |
129 | &m->jobs_in_progress_event_source, | |
130 | CLOCK_MONOTONIC, | |
131 | next, 0, | |
132 | manager_dispatch_jobs_in_progress, m); | |
cfa9677b MM |
133 | if (r < 0) |
134 | return; | |
7dfbe2e3 TG |
135 | |
136 | (void) sd_event_source_set_description(m->jobs_in_progress_event_source, "manager-jobs-in-progress"); | |
03b717a3 MS |
137 | } |
138 | ||
fbd0b64f | 139 | #define CYLON_BUFFER_EXTRA (2*STRLEN(ANSI_RED) + STRLEN(ANSI_HIGHLIGHT_RED) + 2*STRLEN(ANSI_NORMAL)) |
03b717a3 | 140 | |
03b717a3 MS |
141 | static void draw_cylon(char buffer[], size_t buflen, unsigned width, unsigned pos) { |
142 | char *p = buffer; | |
143 | ||
144 | assert(buflen >= CYLON_BUFFER_EXTRA + width + 1); | |
145 | assert(pos <= width+1); /* 0 or width+1 mean that the center light is behind the corner */ | |
146 | ||
147 | if (pos > 1) { | |
6282c859 MS |
148 | if (pos > 2) |
149 | p = mempset(p, ' ', pos-2); | |
64c3610b FB |
150 | if (log_get_show_color()) |
151 | p = stpcpy(p, ANSI_RED); | |
03b717a3 MS |
152 | *p++ = '*'; |
153 | } | |
154 | ||
155 | if (pos > 0 && pos <= width) { | |
64c3610b FB |
156 | if (log_get_show_color()) |
157 | p = stpcpy(p, ANSI_HIGHLIGHT_RED); | |
03b717a3 MS |
158 | *p++ = '*'; |
159 | } | |
160 | ||
64c3610b FB |
161 | if (log_get_show_color()) |
162 | p = stpcpy(p, ANSI_NORMAL); | |
03b717a3 MS |
163 | |
164 | if (pos < width) { | |
64c3610b FB |
165 | if (log_get_show_color()) |
166 | p = stpcpy(p, ANSI_RED); | |
03b717a3 | 167 | *p++ = '*'; |
6282c859 MS |
168 | if (pos < width-1) |
169 | p = mempset(p, ' ', width-1-pos); | |
64c3610b FB |
170 | if (log_get_show_color()) |
171 | strcpy(p, ANSI_NORMAL); | |
03b717a3 | 172 | } |
03b717a3 MS |
173 | } |
174 | ||
cb8ccb22 | 175 | void manager_flip_auto_status(Manager *m, bool enable) { |
f755e3b7 LP |
176 | assert(m); |
177 | ||
cb8ccb22 ZJS |
178 | if (enable) { |
179 | if (m->show_status == SHOW_STATUS_AUTO) | |
180 | manager_set_show_status(m, SHOW_STATUS_TEMPORARY); | |
181 | } else { | |
182 | if (m->show_status == SHOW_STATUS_TEMPORARY) | |
183 | manager_set_show_status(m, SHOW_STATUS_AUTO); | |
184 | } | |
185 | } | |
186 | ||
03b717a3 | 187 | static void manager_print_jobs_in_progress(Manager *m) { |
718db961 | 188 | _cleanup_free_ char *job_of_n = NULL; |
03b717a3 MS |
189 | Iterator i; |
190 | Job *j; | |
03b717a3 MS |
191 | unsigned counter = 0, print_nr; |
192 | char cylon[6 + CYLON_BUFFER_EXTRA + 1]; | |
193 | unsigned cylon_pos; | |
8bb310c3 ZJS |
194 | char time[FORMAT_TIMESPAN_MAX], limit[FORMAT_TIMESPAN_MAX] = "no limit"; |
195 | uint64_t x; | |
03b717a3 | 196 | |
718db961 | 197 | assert(m); |
9c3349e2 | 198 | assert(m->n_running_jobs > 0); |
718db961 | 199 | |
cb8ccb22 | 200 | manager_flip_auto_status(m, true); |
d450b6f2 | 201 | |
03b717a3 MS |
202 | print_nr = (m->jobs_in_progress_iteration / JOBS_IN_PROGRESS_PERIOD_DIVISOR) % m->n_running_jobs; |
203 | ||
204 | HASHMAP_FOREACH(j, m->jobs, i) | |
205 | if (j->state == JOB_RUNNING && counter++ == print_nr) | |
206 | break; | |
207 | ||
e970a72e MS |
208 | /* m->n_running_jobs must be consistent with the contents of m->jobs, |
209 | * so the above loop must have succeeded in finding j. */ | |
210 | assert(counter == print_nr + 1); | |
51d122af | 211 | assert(j); |
5a82a91a | 212 | |
03b717a3 MS |
213 | cylon_pos = m->jobs_in_progress_iteration % 14; |
214 | if (cylon_pos >= 8) | |
215 | cylon_pos = 14 - cylon_pos; | |
216 | draw_cylon(cylon, sizeof(cylon), 6, cylon_pos); | |
217 | ||
8bb310c3 ZJS |
218 | m->jobs_in_progress_iteration++; |
219 | ||
d6483ba7 ZJS |
220 | if (m->n_running_jobs > 1) { |
221 | if (asprintf(&job_of_n, "(%u of %u) ", counter, m->n_running_jobs) < 0) | |
222 | job_of_n = NULL; | |
223 | } | |
03b717a3 | 224 | |
8bb310c3 ZJS |
225 | format_timespan(time, sizeof(time), now(CLOCK_MONOTONIC) - j->begin_usec, 1*USEC_PER_SEC); |
226 | if (job_get_timeout(j, &x) > 0) | |
227 | format_timespan(limit, sizeof(limit), x - j->begin_usec, 1*USEC_PER_SEC); | |
228 | ||
127d5fd1 | 229 | manager_status_printf(m, STATUS_TYPE_EPHEMERAL, cylon, |
8bb310c3 ZJS |
230 | "%sA %s job is running for %s (%s / %s)", |
231 | strempty(job_of_n), | |
232 | job_type_to_string(j->type), | |
233 | unit_description(j->unit), | |
234 | time, limit); | |
03b717a3 MS |
235 | } |
236 | ||
e46b13c8 ZJS |
237 | static int have_ask_password(void) { |
238 | _cleanup_closedir_ DIR *dir; | |
8fb3f009 | 239 | struct dirent *de; |
e46b13c8 ZJS |
240 | |
241 | dir = opendir("/run/systemd/ask-password"); | |
242 | if (!dir) { | |
243 | if (errno == ENOENT) | |
244 | return false; | |
245 | else | |
246 | return -errno; | |
247 | } | |
248 | ||
8fb3f009 | 249 | FOREACH_DIRENT_ALL(de, dir, return -errno) { |
e46b13c8 ZJS |
250 | if (startswith(de->d_name, "ask.")) |
251 | return true; | |
252 | } | |
8fb3f009 | 253 | return false; |
e46b13c8 ZJS |
254 | } |
255 | ||
256 | static int manager_dispatch_ask_password_fd(sd_event_source *source, | |
257 | int fd, uint32_t revents, void *userdata) { | |
258 | Manager *m = userdata; | |
259 | ||
260 | assert(m); | |
261 | ||
665dfe93 | 262 | (void) flush_fd(fd); |
e46b13c8 ZJS |
263 | |
264 | m->have_ask_password = have_ask_password(); | |
265 | if (m->have_ask_password < 0) | |
266 | /* Log error but continue. Negative have_ask_password | |
267 | * is treated as unknown status. */ | |
c33b3297 | 268 | log_error_errno(m->have_ask_password, "Failed to list /run/systemd/ask-password: %m"); |
e46b13c8 ZJS |
269 | |
270 | return 0; | |
271 | } | |
272 | ||
273 | static void manager_close_ask_password(Manager *m) { | |
274 | assert(m); | |
275 | ||
e46b13c8 | 276 | m->ask_password_event_source = sd_event_source_unref(m->ask_password_event_source); |
90990e28 | 277 | m->ask_password_inotify_fd = safe_close(m->ask_password_inotify_fd); |
e46b13c8 ZJS |
278 | m->have_ask_password = -EINVAL; |
279 | } | |
280 | ||
281 | static int manager_check_ask_password(Manager *m) { | |
282 | int r; | |
283 | ||
284 | assert(m); | |
285 | ||
286 | if (!m->ask_password_event_source) { | |
287 | assert(m->ask_password_inotify_fd < 0); | |
288 | ||
289 | mkdir_p_label("/run/systemd/ask-password", 0755); | |
290 | ||
291 | m->ask_password_inotify_fd = inotify_init1(IN_NONBLOCK|IN_CLOEXEC); | |
4a62c710 MS |
292 | if (m->ask_password_inotify_fd < 0) |
293 | return log_error_errno(errno, "inotify_init1() failed: %m"); | |
e46b13c8 ZJS |
294 | |
295 | if (inotify_add_watch(m->ask_password_inotify_fd, "/run/systemd/ask-password", IN_CREATE|IN_DELETE|IN_MOVE) < 0) { | |
56f64d95 | 296 | log_error_errno(errno, "Failed to add watch on /run/systemd/ask-password: %m"); |
e46b13c8 ZJS |
297 | manager_close_ask_password(m); |
298 | return -errno; | |
299 | } | |
300 | ||
301 | r = sd_event_add_io(m->event, &m->ask_password_event_source, | |
302 | m->ask_password_inotify_fd, EPOLLIN, | |
303 | manager_dispatch_ask_password_fd, m); | |
304 | if (r < 0) { | |
56f64d95 | 305 | log_error_errno(errno, "Failed to add event source for /run/systemd/ask-password: %m"); |
e46b13c8 ZJS |
306 | manager_close_ask_password(m); |
307 | return -errno; | |
308 | } | |
309 | ||
7dfbe2e3 TG |
310 | (void) sd_event_source_set_description(m->ask_password_event_source, "manager-ask-password"); |
311 | ||
e46b13c8 ZJS |
312 | /* Queries might have been added meanwhile... */ |
313 | manager_dispatch_ask_password_fd(m->ask_password_event_source, | |
314 | m->ask_password_inotify_fd, EPOLLIN, m); | |
315 | } | |
316 | ||
317 | return m->have_ask_password; | |
318 | } | |
319 | ||
31a7eb86 | 320 | static int manager_watch_idle_pipe(Manager *m) { |
31a7eb86 ZJS |
321 | int r; |
322 | ||
718db961 LP |
323 | assert(m); |
324 | ||
325 | if (m->idle_pipe_event_source) | |
31a7eb86 ZJS |
326 | return 0; |
327 | ||
328 | if (m->idle_pipe[2] < 0) | |
329 | return 0; | |
330 | ||
151b9b96 | 331 | r = sd_event_add_io(m->event, &m->idle_pipe_event_source, m->idle_pipe[2], EPOLLIN, manager_dispatch_idle_pipe_fd, m); |
23bbb0de MS |
332 | if (r < 0) |
333 | return log_error_errno(r, "Failed to watch idle pipe: %m"); | |
31a7eb86 | 334 | |
7dfbe2e3 TG |
335 | (void) sd_event_source_set_description(m->idle_pipe_event_source, "manager-idle-pipe"); |
336 | ||
31a7eb86 | 337 | return 0; |
31a7eb86 ZJS |
338 | } |
339 | ||
718db961 LP |
340 | static void manager_close_idle_pipe(Manager *m) { |
341 | assert(m); | |
31a7eb86 | 342 | |
cd72bd8a LP |
343 | m->idle_pipe_event_source = sd_event_source_unref(m->idle_pipe_event_source); |
344 | ||
3d94f76c LP |
345 | safe_close_pair(m->idle_pipe); |
346 | safe_close_pair(m->idle_pipe + 2); | |
31a7eb86 ZJS |
347 | } |
348 | ||
8742514c | 349 | static int manager_setup_time_change(Manager *m) { |
718db961 | 350 | int r; |
b92bea5d | 351 | |
718db961 | 352 | assert(m); |
8742514c | 353 | |
e0a3da1f | 354 | if (m->test_run_flags) |
0d8c31ff ZJS |
355 | return 0; |
356 | ||
7feedd18 LP |
357 | m->time_change_event_source = sd_event_source_unref(m->time_change_event_source); |
358 | m->time_change_fd = safe_close(m->time_change_fd); | |
359 | ||
4f811d27 | 360 | m->time_change_fd = time_change_fd(); |
4a62c710 | 361 | if (m->time_change_fd < 0) |
4f811d27 | 362 | return log_error_errno(m->time_change_fd, "Failed to create timer change timer fd: %m"); |
8742514c | 363 | |
151b9b96 | 364 | r = sd_event_add_io(m->event, &m->time_change_event_source, m->time_change_fd, EPOLLIN, manager_dispatch_time_change_fd, m); |
23bbb0de MS |
365 | if (r < 0) |
366 | return log_error_errno(r, "Failed to create time change event source: %m"); | |
8742514c | 367 | |
a5cc7e5a LP |
368 | /* Schedule this slightly earlier than the .timer event sources */ |
369 | r = sd_event_source_set_priority(m->time_change_event_source, SD_EVENT_PRIORITY_NORMAL-1); | |
370 | if (r < 0) | |
371 | return log_error_errno(r, "Failed to set priority of time change event sources: %m"); | |
372 | ||
7dfbe2e3 TG |
373 | (void) sd_event_source_set_description(m->time_change_event_source, "manager-time-change"); |
374 | ||
8742514c LP |
375 | log_debug("Set up TFD_TIMER_CANCEL_ON_SET timerfd."); |
376 | ||
377 | return 0; | |
378 | } | |
379 | ||
bbf5fd8e LP |
380 | static int manager_read_timezone_stat(Manager *m) { |
381 | struct stat st; | |
382 | bool changed; | |
383 | ||
384 | assert(m); | |
385 | ||
386 | /* Read the current stat() data of /etc/localtime so that we detect changes */ | |
387 | if (lstat("/etc/localtime", &st) < 0) { | |
388 | log_debug_errno(errno, "Failed to stat /etc/localtime, ignoring: %m"); | |
389 | changed = m->etc_localtime_accessible; | |
390 | m->etc_localtime_accessible = false; | |
391 | } else { | |
392 | usec_t k; | |
393 | ||
394 | k = timespec_load(&st.st_mtim); | |
395 | changed = !m->etc_localtime_accessible || k != m->etc_localtime_mtime; | |
396 | ||
397 | m->etc_localtime_mtime = k; | |
398 | m->etc_localtime_accessible = true; | |
399 | } | |
400 | ||
401 | return changed; | |
402 | } | |
403 | ||
404 | static int manager_setup_timezone_change(Manager *m) { | |
a5cc7e5a | 405 | _cleanup_(sd_event_source_unrefp) sd_event_source *new_event = NULL; |
bbf5fd8e LP |
406 | int r; |
407 | ||
408 | assert(m); | |
409 | ||
410 | if (m->test_run_flags != 0) | |
411 | return 0; | |
412 | ||
413 | /* We watch /etc/localtime for three events: change of the link count (which might mean removal from /etc even | |
414 | * though another link might be kept), renames, and file close operations after writing. Note we don't bother | |
415 | * with IN_DELETE_SELF, as that would just report when the inode is removed entirely, i.e. after the link count | |
416 | * went to zero and all fds to it are closed. | |
417 | * | |
418 | * Note that we never follow symlinks here. This is a simplification, but should cover almost all cases | |
419 | * correctly. | |
420 | * | |
421 | * Note that we create the new event source first here, before releasing the old one. This should optimize | |
422 | * behaviour as this way sd-event can reuse the old watch in case the inode didn't change. */ | |
423 | ||
424 | r = sd_event_add_inotify(m->event, &new_event, "/etc/localtime", | |
425 | IN_ATTRIB|IN_MOVE_SELF|IN_CLOSE_WRITE|IN_DONT_FOLLOW, manager_dispatch_timezone_change, m); | |
426 | if (r == -ENOENT) /* If the file doesn't exist yet, subscribe to /etc instead, and wait until it is created | |
427 | * either by O_CREATE or by rename() */ | |
428 | r = sd_event_add_inotify(m->event, &new_event, "/etc", | |
429 | IN_CREATE|IN_MOVED_TO|IN_ONLYDIR, manager_dispatch_timezone_change, m); | |
430 | if (r < 0) | |
431 | return log_error_errno(r, "Failed to create timezone change event source: %m"); | |
432 | ||
a5cc7e5a LP |
433 | /* Schedule this slightly earlier than the .timer event sources */ |
434 | r = sd_event_source_set_priority(new_event, SD_EVENT_PRIORITY_NORMAL-1); | |
435 | if (r < 0) | |
436 | return log_error_errno(r, "Failed to set priority of timezone change event sources: %m"); | |
437 | ||
bbf5fd8e | 438 | sd_event_source_unref(m->timezone_change_event_source); |
a5cc7e5a | 439 | m->timezone_change_event_source = TAKE_PTR(new_event); |
bbf5fd8e LP |
440 | |
441 | return 0; | |
442 | } | |
443 | ||
80876c20 | 444 | static int enable_special_signals(Manager *m) { |
718db961 | 445 | _cleanup_close_ int fd = -1; |
80876c20 LP |
446 | |
447 | assert(m); | |
448 | ||
e0a3da1f | 449 | if (m->test_run_flags) |
37453b3a EV |
450 | return 0; |
451 | ||
a41b539e | 452 | /* Enable that we get SIGINT on control-alt-del. In containers |
c9999773 LP |
453 | * this will fail with EPERM (older) or EINVAL (newer), so |
454 | * ignore that. */ | |
4c701096 | 455 | if (reboot(RB_DISABLE_CAD) < 0 && !IN_SET(errno, EPERM, EINVAL)) |
56f64d95 | 456 | log_warning_errno(errno, "Failed to enable ctrl-alt-del handling: %m"); |
80876c20 | 457 | |
a41b539e LP |
458 | fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC); |
459 | if (fd < 0) { | |
460 | /* Support systems without virtual console */ | |
461 | if (fd != -ENOENT) | |
56f64d95 | 462 | log_warning_errno(errno, "Failed to open /dev/tty0: %m"); |
a41b539e | 463 | } else { |
80876c20 LP |
464 | /* Enable that we get SIGWINCH on kbrequest */ |
465 | if (ioctl(fd, KDSIGACCEPT, SIGWINCH) < 0) | |
56f64d95 | 466 | log_warning_errno(errno, "Failed to enable kbrequest handling: %m"); |
80876c20 LP |
467 | } |
468 | ||
469 | return 0; | |
470 | } | |
471 | ||
8750ac02 ZJS |
472 | #define RTSIG_IF_AVAILABLE(signum) (signum <= SIGRTMAX ? signum : -1) |
473 | ||
ce578209 | 474 | static int manager_setup_signals(Manager *m) { |
b92bea5d ZJS |
475 | struct sigaction sa = { |
476 | .sa_handler = SIG_DFL, | |
477 | .sa_flags = SA_NOCLDSTOP|SA_RESTART, | |
478 | }; | |
718db961 LP |
479 | sigset_t mask; |
480 | int r; | |
60918275 | 481 | |
ce578209 LP |
482 | assert(m); |
483 | ||
57c0c30e LP |
484 | assert_se(sigaction(SIGCHLD, &sa, NULL) == 0); |
485 | ||
4dffec14 LP |
486 | /* We make liberal use of realtime signals here. On |
487 | * Linux/glibc we have 30 of them (with the exception of Linux | |
488 | * on hppa, see below), between SIGRTMIN+0 ... SIGRTMIN+30 | |
489 | * (aka SIGRTMAX). */ | |
7d793605 | 490 | |
4dffec14 | 491 | assert_se(sigemptyset(&mask) == 0); |
7d793605 LP |
492 | sigset_add_many(&mask, |
493 | SIGCHLD, /* Child died */ | |
494 | SIGTERM, /* Reexecute daemon */ | |
495 | SIGHUP, /* Reload configuration */ | |
496 | SIGUSR1, /* systemd/upstart: reconnect to D-Bus */ | |
497 | SIGUSR2, /* systemd: dump status */ | |
498 | SIGINT, /* Kernel sends us this on control-alt-del */ | |
499 | SIGWINCH, /* Kernel sends us this on kbrequest (alt-arrowup) */ | |
500 | SIGPWR, /* Some kernel drivers and upsd send us this on power failure */ | |
4dffec14 | 501 | |
7d793605 | 502 | SIGRTMIN+0, /* systemd: start default.target */ |
0003d1ab | 503 | SIGRTMIN+1, /* systemd: isolate rescue.target */ |
7d793605 LP |
504 | SIGRTMIN+2, /* systemd: isolate emergency.target */ |
505 | SIGRTMIN+3, /* systemd: start halt.target */ | |
506 | SIGRTMIN+4, /* systemd: start poweroff.target */ | |
507 | SIGRTMIN+5, /* systemd: start reboot.target */ | |
0003d1ab | 508 | SIGRTMIN+6, /* systemd: start kexec.target */ |
4dffec14 LP |
509 | |
510 | /* ... space for more special targets ... */ | |
511 | ||
0003d1ab LP |
512 | SIGRTMIN+13, /* systemd: Immediate halt */ |
513 | SIGRTMIN+14, /* systemd: Immediate poweroff */ | |
514 | SIGRTMIN+15, /* systemd: Immediate reboot */ | |
515 | SIGRTMIN+16, /* systemd: Immediate kexec */ | |
4dffec14 LP |
516 | |
517 | /* ... space for more immediate system state changes ... */ | |
518 | ||
0658666b LP |
519 | SIGRTMIN+20, /* systemd: enable status messages */ |
520 | SIGRTMIN+21, /* systemd: disable status messages */ | |
253ee27a LP |
521 | SIGRTMIN+22, /* systemd: set log level to LOG_DEBUG */ |
522 | SIGRTMIN+23, /* systemd: set log level to LOG_INFO */ | |
600b704e | 523 | SIGRTMIN+24, /* systemd: Immediate exit (--user only) */ |
4dffec14 LP |
524 | |
525 | /* .. one free signal here ... */ | |
526 | ||
8750ac02 ZJS |
527 | /* Apparently Linux on hppa had fewer RT signals until v3.18, |
528 | * SIGRTMAX was SIGRTMIN+25, and then SIGRTMIN was lowered, | |
529 | * see commit v3.17-7614-g1f25df2eff. | |
530 | * | |
531 | * We cannot unconditionally make use of those signals here, | |
532 | * so let's use a runtime check. Since these commands are | |
533 | * accessible by different means and only really a safety | |
534 | * net, the missing functionality on hppa shouldn't matter. | |
535 | */ | |
536 | ||
537 | RTSIG_IF_AVAILABLE(SIGRTMIN+26), /* systemd: set log target to journal-or-kmsg */ | |
538 | RTSIG_IF_AVAILABLE(SIGRTMIN+27), /* systemd: set log target to console */ | |
539 | RTSIG_IF_AVAILABLE(SIGRTMIN+28), /* systemd: set log target to kmsg */ | |
540 | RTSIG_IF_AVAILABLE(SIGRTMIN+29), /* systemd: set log target to syslog-or-kmsg (obsolete) */ | |
4dffec14 LP |
541 | |
542 | /* ... one free signal here SIGRTMIN+30 ... */ | |
7d793605 | 543 | -1); |
ce578209 LP |
544 | assert_se(sigprocmask(SIG_SETMASK, &mask, NULL) == 0); |
545 | ||
718db961 LP |
546 | m->signal_fd = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC); |
547 | if (m->signal_fd < 0) | |
ce578209 LP |
548 | return -errno; |
549 | ||
151b9b96 | 550 | r = sd_event_add_io(m->event, &m->signal_event_source, m->signal_fd, EPOLLIN, manager_dispatch_signal_fd, m); |
718db961 LP |
551 | if (r < 0) |
552 | return r; | |
ce578209 | 553 | |
7dfbe2e3 TG |
554 | (void) sd_event_source_set_description(m->signal_event_source, "manager-signal"); |
555 | ||
d8fdc620 LP |
556 | /* Process signals a bit earlier than the rest of things, but later than notify_fd processing, so that the |
557 | * notify processing can still figure out to which process/service a message belongs, before we reap the | |
558 | * process. Also, process this before handling cgroup notifications, so that we always collect child exit | |
559 | * status information before detecting that there's no process in a cgroup. */ | |
560 | r = sd_event_source_set_priority(m->signal_event_source, SD_EVENT_PRIORITY_NORMAL-6); | |
29083707 LP |
561 | if (r < 0) |
562 | return r; | |
563 | ||
463d0d15 | 564 | if (MANAGER_IS_SYSTEM(m)) |
80876c20 | 565 | return enable_special_signals(m); |
e1414003 | 566 | |
ce578209 LP |
567 | return 0; |
568 | } | |
569 | ||
47cf8ff2 | 570 | static void manager_sanitize_environment(Manager *m) { |
f069efb4 LP |
571 | assert(m); |
572 | ||
47cf8ff2 | 573 | /* Let's remove some environment variables that we need ourselves to communicate with our clients */ |
f069efb4 LP |
574 | strv_env_unset_many( |
575 | m->environment, | |
47cf8ff2 LP |
576 | "EXIT_CODE", |
577 | "EXIT_STATUS", | |
578 | "INVOCATION_ID", | |
579 | "JOURNAL_STREAM", | |
580 | "LISTEN_FDNAMES", | |
581 | "LISTEN_FDS", | |
582 | "LISTEN_PID", | |
f069efb4 LP |
583 | "MAINPID", |
584 | "MANAGERPID", | |
47cf8ff2 LP |
585 | "NOTIFY_SOCKET", |
586 | "REMOTE_ADDR", | |
587 | "REMOTE_PORT", | |
588 | "SERVICE_RESULT", | |
f069efb4 LP |
589 | "WATCHDOG_PID", |
590 | "WATCHDOG_USEC", | |
591 | NULL); | |
47cf8ff2 LP |
592 | |
593 | /* Let's order the environment alphabetically, just to make it pretty */ | |
594 | strv_sort(m->environment); | |
f069efb4 LP |
595 | } |
596 | ||
e21fea24 | 597 | static int manager_default_environment(Manager *m) { |
71ecc858 LP |
598 | assert(m); |
599 | ||
463d0d15 | 600 | if (MANAGER_IS_SYSTEM(m)) { |
e21fea24 KS |
601 | /* The system manager always starts with a clean |
602 | * environment for its children. It does not import | |
71cb7d30 | 603 | * the kernel's or the parents' exported variables. |
e21fea24 | 604 | * |
71cb7d30 | 605 | * The initial passed environment is untouched to keep |
e21fea24 KS |
606 | * /proc/self/environ valid; it is used for tagging |
607 | * the init process inside containers. */ | |
43638332 ZJS |
608 | m->environment = strv_new("PATH=" DEFAULT_PATH, |
609 | NULL); | |
e21fea24 KS |
610 | |
611 | /* Import locale variables LC_*= from configuration */ | |
612 | locale_setup(&m->environment); | |
71cb7d30 | 613 | } else |
e21fea24 KS |
614 | /* The user manager passes its own environment |
615 | * along to its children. */ | |
616 | m->environment = strv_copy(environ); | |
43d03a83 | 617 | |
e21fea24 KS |
618 | if (!m->environment) |
619 | return -ENOMEM; | |
8b55b8c4 | 620 | |
47cf8ff2 | 621 | manager_sanitize_environment(m); |
9d5a3757 | 622 | |
e21fea24 | 623 | return 0; |
71ecc858 LP |
624 | } |
625 | ||
3536f49e YW |
626 | static int manager_setup_prefix(Manager *m) { |
627 | struct table_entry { | |
628 | uint64_t type; | |
629 | const char *suffix; | |
630 | }; | |
631 | ||
72fd1768 | 632 | static const struct table_entry paths_system[_EXEC_DIRECTORY_TYPE_MAX] = { |
3536f49e YW |
633 | [EXEC_DIRECTORY_RUNTIME] = { SD_PATH_SYSTEM_RUNTIME, NULL }, |
634 | [EXEC_DIRECTORY_STATE] = { SD_PATH_SYSTEM_STATE_PRIVATE, NULL }, | |
635 | [EXEC_DIRECTORY_CACHE] = { SD_PATH_SYSTEM_STATE_CACHE, NULL }, | |
636 | [EXEC_DIRECTORY_LOGS] = { SD_PATH_SYSTEM_STATE_LOGS, NULL }, | |
637 | [EXEC_DIRECTORY_CONFIGURATION] = { SD_PATH_SYSTEM_CONFIGURATION, NULL }, | |
638 | }; | |
639 | ||
72fd1768 | 640 | static const struct table_entry paths_user[_EXEC_DIRECTORY_TYPE_MAX] = { |
3536f49e YW |
641 | [EXEC_DIRECTORY_RUNTIME] = { SD_PATH_USER_RUNTIME, NULL }, |
642 | [EXEC_DIRECTORY_STATE] = { SD_PATH_USER_CONFIGURATION, NULL }, | |
c6218495 LP |
643 | [EXEC_DIRECTORY_CACHE] = { SD_PATH_USER_STATE_CACHE, NULL }, |
644 | [EXEC_DIRECTORY_LOGS] = { SD_PATH_USER_CONFIGURATION, "log" }, | |
645 | [EXEC_DIRECTORY_CONFIGURATION] = { SD_PATH_USER_CONFIGURATION, NULL }, | |
3536f49e YW |
646 | }; |
647 | ||
648 | const struct table_entry *p; | |
649 | ExecDirectoryType i; | |
650 | int r; | |
651 | ||
652 | assert(m); | |
653 | ||
654 | if (MANAGER_IS_SYSTEM(m)) | |
655 | p = paths_system; | |
656 | else | |
657 | p = paths_user; | |
658 | ||
72fd1768 | 659 | for (i = 0; i < _EXEC_DIRECTORY_TYPE_MAX; i++) { |
3536f49e YW |
660 | r = sd_path_home(p[i].type, p[i].suffix, &m->prefix[i]); |
661 | if (r < 0) | |
662 | return r; | |
663 | } | |
664 | ||
665 | return 0; | |
666 | } | |
667 | ||
279d81dd LP |
668 | static int manager_setup_run_queue(Manager *m) { |
669 | int r; | |
670 | ||
671 | assert(m); | |
672 | assert(!m->run_queue_event_source); | |
673 | ||
674 | r = sd_event_add_defer(m->event, &m->run_queue_event_source, manager_dispatch_run_queue, m); | |
675 | if (r < 0) | |
676 | return r; | |
677 | ||
678 | r = sd_event_source_set_priority(m->run_queue_event_source, SD_EVENT_PRIORITY_IDLE); | |
679 | if (r < 0) | |
680 | return r; | |
681 | ||
682 | r = sd_event_source_set_enabled(m->run_queue_event_source, SD_EVENT_OFF); | |
683 | if (r < 0) | |
684 | return r; | |
685 | ||
686 | (void) sd_event_source_set_description(m->run_queue_event_source, "manager-run-queue"); | |
687 | ||
688 | return 0; | |
689 | } | |
690 | ||
575b300b LP |
691 | static int manager_setup_sigchld_event_source(Manager *m) { |
692 | int r; | |
693 | ||
694 | assert(m); | |
695 | assert(!m->sigchld_event_source); | |
696 | ||
697 | r = sd_event_add_defer(m->event, &m->sigchld_event_source, manager_dispatch_sigchld, m); | |
698 | if (r < 0) | |
699 | return r; | |
700 | ||
701 | r = sd_event_source_set_priority(m->sigchld_event_source, SD_EVENT_PRIORITY_NORMAL-7); | |
702 | if (r < 0) | |
703 | return r; | |
704 | ||
705 | r = sd_event_source_set_enabled(m->sigchld_event_source, SD_EVENT_OFF); | |
706 | if (r < 0) | |
707 | return r; | |
708 | ||
709 | (void) sd_event_source_set_description(m->sigchld_event_source, "manager-sigchld"); | |
710 | ||
711 | return 0; | |
712 | } | |
713 | ||
e0a3da1f | 714 | int manager_new(UnitFileScope scope, unsigned test_run_flags, Manager **_m) { |
c70cac54 | 715 | _cleanup_(manager_freep) Manager *m = NULL; |
e3dd987c | 716 | int r; |
8e274523 LP |
717 | |
718 | assert(_m); | |
463d0d15 | 719 | assert(IN_SET(scope, UNIT_FILE_SYSTEM, UNIT_FILE_USER)); |
ce578209 | 720 | |
915b3753 LP |
721 | m = new0(Manager, 1); |
722 | if (!m) | |
8e274523 | 723 | return -ENOMEM; |
60918275 | 724 | |
463d0d15 | 725 | m->unit_file_scope = scope; |
a16e1123 | 726 | m->exit_code = _MANAGER_EXIT_CODE_INVALID; |
bd8f585b | 727 | m->default_timer_accuracy_usec = USEC_PER_MINUTE; |
444d5863 | 728 | m->default_memory_accounting = MEMORY_ACCOUNTING_DEFAULT; |
9ded9cd1 | 729 | m->default_tasks_accounting = true; |
79baeeb9 | 730 | m->default_tasks_max = UINT64_MAX; |
bd389aa7 LP |
731 | m->default_timeout_start_usec = DEFAULT_TIMEOUT_USEC; |
732 | m->default_timeout_stop_usec = DEFAULT_TIMEOUT_USEC; | |
733 | m->default_restart_usec = DEFAULT_RESTART_USEC; | |
a6ecbf83 | 734 | m->original_log_level = -1; |
bda7d78b | 735 | m->original_log_target = _LOG_TARGET_INVALID; |
80876c20 | 736 | |
349cc4a5 | 737 | #if ENABLE_EFI |
463d0d15 | 738 | if (MANAGER_IS_SYSTEM(m) && detect_container() <= 0) |
9f9f0342 LP |
739 | boot_timestamps(m->timestamps + MANAGER_TIMESTAMP_USERSPACE, |
740 | m->timestamps + MANAGER_TIMESTAMP_FIRMWARE, | |
741 | m->timestamps + MANAGER_TIMESTAMP_LOADER); | |
463d0d15 LP |
742 | #endif |
743 | ||
f2341e0a | 744 | /* Prepare log fields we can use for structured logging */ |
463d0d15 LP |
745 | if (MANAGER_IS_SYSTEM(m)) { |
746 | m->unit_log_field = "UNIT="; | |
747 | m->unit_log_format_string = "UNIT=%s"; | |
4b58153d LP |
748 | |
749 | m->invocation_log_field = "INVOCATION_ID="; | |
f1c50bec | 750 | m->invocation_log_format_string = "INVOCATION_ID=%s"; |
463d0d15 LP |
751 | } else { |
752 | m->unit_log_field = "USER_UNIT="; | |
753 | m->unit_log_format_string = "USER_UNIT=%s"; | |
4b58153d LP |
754 | |
755 | m->invocation_log_field = "USER_INVOCATION_ID="; | |
f1c50bec | 756 | m->invocation_log_format_string = "USER_INVOCATION_ID=%s"; |
463d0d15 | 757 | } |
f2341e0a | 758 | |
718db961 | 759 | m->idle_pipe[0] = m->idle_pipe[1] = m->idle_pipe[2] = m->idle_pipe[3] = -1; |
8742514c | 760 | |
d8fdc620 | 761 | m->pin_cgroupfs_fd = m->notify_fd = m->cgroups_agent_fd = m->signal_fd = m->time_change_fd = |
232f6754 | 762 | m->dev_autofs_fd = m->private_listen_fd = m->cgroup_inotify_fd = |
d8fdc620 | 763 | m->ask_password_inotify_fd = -1; |
d379d442 | 764 | |
00d9ef85 LP |
765 | m->user_lookup_fds[0] = m->user_lookup_fds[1] = -1; |
766 | ||
ea430986 | 767 | m->current_job_id = 1; /* start as id #1, so that we can leave #0 around as "null-like" value */ |
9152c765 | 768 | |
e46b13c8 | 769 | m->have_ask_password = -EINVAL; /* we don't know */ |
ae2a2c53 | 770 | m->first_boot = -1; |
e46b13c8 | 771 | |
e0a3da1f | 772 | m->test_run_flags = test_run_flags; |
0d8c31ff | 773 | |
2e5c94b9 LP |
774 | /* Reboot immediately if the user hits C-A-D more often than 7x per 2s */ |
775 | RATELIMIT_INIT(m->ctrl_alt_del_ratelimit, 2 * USEC_PER_SEC, 7); | |
776 | ||
e21fea24 KS |
777 | r = manager_default_environment(m); |
778 | if (r < 0) | |
c70cac54 | 779 | return r; |
1137a57c | 780 | |
d5099efc | 781 | r = hashmap_ensure_allocated(&m->units, &string_hash_ops); |
718db961 | 782 | if (r < 0) |
c70cac54 | 783 | return r; |
60918275 | 784 | |
d5099efc | 785 | r = hashmap_ensure_allocated(&m->jobs, NULL); |
718db961 | 786 | if (r < 0) |
c70cac54 | 787 | return r; |
60918275 | 788 | |
548f6937 | 789 | r = hashmap_ensure_allocated(&m->cgroup_unit, &path_hash_ops); |
718db961 | 790 | if (r < 0) |
c70cac54 | 791 | return r; |
9152c765 | 792 | |
d5099efc | 793 | r = hashmap_ensure_allocated(&m->watch_bus, &string_hash_ops); |
718db961 | 794 | if (r < 0) |
c70cac54 | 795 | return r; |
05e343b7 | 796 | |
e8112e67 | 797 | r = manager_setup_prefix(m); |
718db961 | 798 | if (r < 0) |
c70cac54 | 799 | return r; |
8742514c | 800 | |
e8112e67 ZJS |
801 | m->udev = udev_new(); |
802 | if (!m->udev) | |
803 | return -ENOMEM; | |
752b5905 | 804 | |
e8112e67 | 805 | r = sd_event_default(&m->event); |
8742514c | 806 | if (r < 0) |
c70cac54 | 807 | return r; |
9152c765 | 808 | |
e8112e67 | 809 | r = manager_setup_run_queue(m); |
a1d32bac | 810 | if (r < 0) |
c70cac54 | 811 | return r; |
8e274523 | 812 | |
e8112e67 ZJS |
813 | if (test_run_flags == MANAGER_TEST_RUN_MINIMAL) { |
814 | m->cgroup_root = strdup(""); | |
815 | if (!m->cgroup_root) | |
816 | return -ENOMEM; | |
817 | } else { | |
818 | r = manager_setup_signals(m); | |
819 | if (r < 0) | |
820 | return r; | |
8c47c732 | 821 | |
e8112e67 ZJS |
822 | r = manager_setup_cgroup(m); |
823 | if (r < 0) | |
824 | return r; | |
575b300b | 825 | |
e8112e67 ZJS |
826 | r = manager_setup_time_change(m); |
827 | if (r < 0) | |
828 | return r; | |
9670d583 | 829 | |
bbf5fd8e LP |
830 | r = manager_read_timezone_stat(m); |
831 | if (r < 0) | |
832 | return r; | |
833 | ||
834 | r = manager_setup_timezone_change(m); | |
835 | if (r < 0) | |
836 | return r; | |
837 | ||
e8112e67 ZJS |
838 | r = manager_setup_sigchld_event_source(m); |
839 | if (r < 0) | |
840 | return r; | |
841 | } | |
e27fe688 | 842 | |
5eb397cf | 843 | if (MANAGER_IS_SYSTEM(m) && test_run_flags == 0) { |
d3070fbd LP |
844 | r = mkdir_label("/run/systemd/units", 0755); |
845 | if (r < 0 && r != -EEXIST) | |
c70cac54 | 846 | return r; |
d3070fbd LP |
847 | } |
848 | ||
e27fe688 LP |
849 | m->taint_usr = |
850 | !in_initrd() && | |
851 | dir_is_empty("/usr") > 0; | |
852 | ||
232f6754 ZJS |
853 | /* Note that we do not set up the notify fd here. We do that after deserialization, |
854 | * since they might have gotten serialized across the reexec. */ | |
3536f49e | 855 | |
1cc6c93a YW |
856 | *_m = TAKE_PTR(m); |
857 | ||
8e274523 | 858 | return 0; |
60918275 LP |
859 | } |
860 | ||
d86f9d52 | 861 | static int manager_setup_notify(Manager *m) { |
7181dbdb | 862 | int r; |
d86f9d52 | 863 | |
e0a3da1f | 864 | if (m->test_run_flags) |
0d8c31ff ZJS |
865 | return 0; |
866 | ||
d86f9d52 LP |
867 | if (m->notify_fd < 0) { |
868 | _cleanup_close_ int fd = -1; | |
920b52e4 | 869 | union sockaddr_union sa = { |
7181dbdb LP |
870 | .sa.sa_family = AF_UNIX, |
871 | }; | |
55836941 | 872 | static const int one = 1; |
d86f9d52 LP |
873 | |
874 | /* First free all secondary fields */ | |
a1e58e8e | 875 | m->notify_socket = mfree(m->notify_socket); |
d86f9d52 LP |
876 | m->notify_event_source = sd_event_source_unref(m->notify_event_source); |
877 | ||
878 | fd = socket(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC|SOCK_NONBLOCK, 0); | |
4a62c710 MS |
879 | if (fd < 0) |
880 | return log_error_errno(errno, "Failed to allocate notification socket: %m"); | |
d86f9d52 | 881 | |
a47806fa LP |
882 | fd_inc_rcvbuf(fd, NOTIFY_RCVBUF_SIZE); |
883 | ||
3536f49e | 884 | m->notify_socket = strappend(m->prefix[EXEC_DIRECTORY_RUNTIME], "/systemd/notify"); |
498e87d6 LP |
885 | if (!m->notify_socket) |
886 | return log_oom(); | |
887 | ||
888 | (void) mkdir_parents_label(m->notify_socket, 0755); | |
f0e62e89 | 889 | (void) unlink(m->notify_socket); |
7181dbdb LP |
890 | |
891 | strncpy(sa.un.sun_path, m->notify_socket, sizeof(sa.un.sun_path)-1); | |
fc2fffe7 | 892 | r = bind(fd, &sa.sa, SOCKADDR_UN_LEN(sa.un)); |
4a62c710 MS |
893 | if (r < 0) |
894 | return log_error_errno(errno, "bind(%s) failed: %m", sa.un.sun_path); | |
d86f9d52 LP |
895 | |
896 | r = setsockopt(fd, SOL_SOCKET, SO_PASSCRED, &one, sizeof(one)); | |
4a62c710 MS |
897 | if (r < 0) |
898 | return log_error_errno(errno, "SO_PASSCRED failed: %m"); | |
d86f9d52 | 899 | |
c10d6bdb | 900 | m->notify_fd = TAKE_FD(fd); |
d86f9d52 LP |
901 | |
902 | log_debug("Using notification socket %s", m->notify_socket); | |
903 | } | |
904 | ||
905 | if (!m->notify_event_source) { | |
151b9b96 | 906 | r = sd_event_add_io(m->event, &m->notify_event_source, m->notify_fd, EPOLLIN, manager_dispatch_notify_fd, m); |
895b3a7b MS |
907 | if (r < 0) |
908 | return log_error_errno(r, "Failed to allocate notify event source: %m"); | |
d86f9d52 | 909 | |
d8fdc620 LP |
910 | /* Process notification messages a bit earlier than SIGCHLD, so that we can still identify to which |
911 | * service an exit message belongs. */ | |
575b300b | 912 | r = sd_event_source_set_priority(m->notify_event_source, SD_EVENT_PRIORITY_NORMAL-8); |
23bbb0de MS |
913 | if (r < 0) |
914 | return log_error_errno(r, "Failed to set priority of notify event source: %m"); | |
7dfbe2e3 TG |
915 | |
916 | (void) sd_event_source_set_description(m->notify_event_source, "manager-notify"); | |
d86f9d52 LP |
917 | } |
918 | ||
919 | return 0; | |
920 | } | |
921 | ||
d8fdc620 LP |
922 | static int manager_setup_cgroups_agent(Manager *m) { |
923 | ||
924 | static const union sockaddr_union sa = { | |
925 | .un.sun_family = AF_UNIX, | |
926 | .un.sun_path = "/run/systemd/cgroups-agent", | |
927 | }; | |
928 | int r; | |
929 | ||
930 | /* This creates a listening socket we receive cgroups agent messages on. We do not use D-Bus for delivering | |
931 | * these messages from the cgroups agent binary to PID 1, as the cgroups agent binary is very short-living, and | |
932 | * each instance of it needs a new D-Bus connection. Since D-Bus connections are SOCK_STREAM/AF_UNIX, on | |
933 | * overloaded systems the backlog of the D-Bus socket becomes relevant, as not more than the configured number | |
934 | * of D-Bus connections may be queued until the kernel will start dropping further incoming connections, | |
935 | * possibly resulting in lost cgroups agent messages. To avoid this, we'll use a private SOCK_DGRAM/AF_UNIX | |
936 | * socket, where no backlog is relevant as communication may take place without an actual connect() cycle, and | |
937 | * we thus won't lose messages. | |
938 | * | |
939 | * Note that PID 1 will forward the agent message to system bus, so that the user systemd instance may listen | |
940 | * to it. The system instance hence listens on this special socket, but the user instances listen on the system | |
941 | * bus for these messages. */ | |
942 | ||
e0a3da1f | 943 | if (m->test_run_flags) |
d8fdc620 LP |
944 | return 0; |
945 | ||
946 | if (!MANAGER_IS_SYSTEM(m)) | |
947 | return 0; | |
948 | ||
c22800e4 | 949 | r = cg_unified_controller(SYSTEMD_CGROUP_CONTROLLER); |
b4cccbc1 LP |
950 | if (r < 0) |
951 | return log_error_errno(r, "Failed to determine whether unified cgroups hierarchy is used: %m"); | |
952 | if (r > 0) /* We don't need this anymore on the unified hierarchy */ | |
d8fdc620 LP |
953 | return 0; |
954 | ||
955 | if (m->cgroups_agent_fd < 0) { | |
956 | _cleanup_close_ int fd = -1; | |
957 | ||
958 | /* First free all secondary fields */ | |
959 | m->cgroups_agent_event_source = sd_event_source_unref(m->cgroups_agent_event_source); | |
960 | ||
961 | fd = socket(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC|SOCK_NONBLOCK, 0); | |
962 | if (fd < 0) | |
963 | return log_error_errno(errno, "Failed to allocate cgroups agent socket: %m"); | |
964 | ||
965 | fd_inc_rcvbuf(fd, CGROUPS_AGENT_RCVBUF_SIZE); | |
966 | ||
967 | (void) unlink(sa.un.sun_path); | |
968 | ||
969 | /* Only allow root to connect to this socket */ | |
970 | RUN_WITH_UMASK(0077) | |
fc2fffe7 | 971 | r = bind(fd, &sa.sa, SOCKADDR_UN_LEN(sa.un)); |
d8fdc620 LP |
972 | if (r < 0) |
973 | return log_error_errno(errno, "bind(%s) failed: %m", sa.un.sun_path); | |
974 | ||
975 | m->cgroups_agent_fd = fd; | |
976 | fd = -1; | |
977 | } | |
978 | ||
979 | if (!m->cgroups_agent_event_source) { | |
980 | r = sd_event_add_io(m->event, &m->cgroups_agent_event_source, m->cgroups_agent_fd, EPOLLIN, manager_dispatch_cgroups_agent_fd, m); | |
981 | if (r < 0) | |
982 | return log_error_errno(r, "Failed to allocate cgroups agent event source: %m"); | |
983 | ||
984 | /* Process cgroups notifications early, but after having processed service notification messages or | |
985 | * SIGCHLD signals, so that a cgroup running empty is always just the last safety net of notification, | |
986 | * and we collected the metadata the notification and SIGCHLD stuff offers first. Also see handling of | |
987 | * cgroup inotify for the unified cgroup stuff. */ | |
09e24654 | 988 | r = sd_event_source_set_priority(m->cgroups_agent_event_source, SD_EVENT_PRIORITY_NORMAL-4); |
d8fdc620 LP |
989 | if (r < 0) |
990 | return log_error_errno(r, "Failed to set priority of cgroups agent event source: %m"); | |
991 | ||
992 | (void) sd_event_source_set_description(m->cgroups_agent_event_source, "manager-cgroups-agent"); | |
993 | } | |
994 | ||
995 | return 0; | |
996 | } | |
997 | ||
00d9ef85 LP |
998 | static int manager_setup_user_lookup_fd(Manager *m) { |
999 | int r; | |
1000 | ||
1001 | assert(m); | |
1002 | ||
1003 | /* Set up the socket pair used for passing UID/GID resolution results from forked off processes to PID | |
1004 | * 1. Background: we can't do name lookups (NSS) from PID 1, since it might involve IPC and thus activation, | |
1005 | * and we might hence deadlock on ourselves. Hence we do all user/group lookups asynchronously from the forked | |
1006 | * off processes right before executing the binaries to start. In order to be able to clean up any IPC objects | |
1007 | * created by a unit (see RemoveIPC=) we need to know in PID 1 the used UID/GID of the executed processes, | |
1008 | * hence we establish this communication channel so that forked off processes can pass their UID/GID | |
1009 | * information back to PID 1. The forked off processes send their resolved UID/GID to PID 1 in a simple | |
1010 | * datagram, along with their unit name, so that we can share one communication socket pair among all units for | |
1011 | * this purpose. | |
1012 | * | |
1013 | * You might wonder why we need a communication channel for this that is independent of the usual notification | |
1014 | * socket scheme (i.e. $NOTIFY_SOCKET). The primary difference is about trust: data sent via the $NOTIFY_SOCKET | |
1015 | * channel is only accepted if it originates from the right unit and if reception was enabled for it. The user | |
1016 | * lookup socket OTOH is only accessible by PID 1 and its children until they exec(), and always available. | |
1017 | * | |
1018 | * Note that this function is called under two circumstances: when we first initialize (in which case we | |
1019 | * allocate both the socket pair and the event source to listen on it), and when we deserialize after a reload | |
1020 | * (in which case the socket pair already exists but we still need to allocate the event source for it). */ | |
1021 | ||
1022 | if (m->user_lookup_fds[0] < 0) { | |
1023 | ||
1024 | /* Free all secondary fields */ | |
1025 | safe_close_pair(m->user_lookup_fds); | |
1026 | m->user_lookup_event_source = sd_event_source_unref(m->user_lookup_event_source); | |
1027 | ||
1028 | if (socketpair(AF_UNIX, SOCK_DGRAM|SOCK_CLOEXEC, 0, m->user_lookup_fds) < 0) | |
1029 | return log_error_errno(errno, "Failed to allocate user lookup socket: %m"); | |
1030 | ||
1031 | (void) fd_inc_rcvbuf(m->user_lookup_fds[0], NOTIFY_RCVBUF_SIZE); | |
1032 | } | |
1033 | ||
1034 | if (!m->user_lookup_event_source) { | |
1035 | r = sd_event_add_io(m->event, &m->user_lookup_event_source, m->user_lookup_fds[0], EPOLLIN, manager_dispatch_user_lookup_fd, m); | |
1036 | if (r < 0) | |
1037 | return log_error_errno(errno, "Failed to allocate user lookup event source: %m"); | |
1038 | ||
1039 | /* Process even earlier than the notify event source, so that we always know first about valid UID/GID | |
1040 | * resolutions */ | |
67ae4e8d | 1041 | r = sd_event_source_set_priority(m->user_lookup_event_source, SD_EVENT_PRIORITY_NORMAL-11); |
00d9ef85 LP |
1042 | if (r < 0) |
1043 | return log_error_errno(errno, "Failed to set priority ot user lookup event source: %m"); | |
1044 | ||
1045 | (void) sd_event_source_set_description(m->user_lookup_event_source, "user-lookup"); | |
1046 | } | |
1047 | ||
1048 | return 0; | |
1049 | } | |
1050 | ||
23a177ef | 1051 | static unsigned manager_dispatch_cleanup_queue(Manager *m) { |
595ed347 | 1052 | Unit *u; |
23a177ef LP |
1053 | unsigned n = 0; |
1054 | ||
1055 | assert(m); | |
1056 | ||
595ed347 MS |
1057 | while ((u = m->cleanup_queue)) { |
1058 | assert(u->in_cleanup_queue); | |
23a177ef | 1059 | |
595ed347 | 1060 | unit_free(u); |
23a177ef LP |
1061 | n++; |
1062 | } | |
1063 | ||
1064 | return n; | |
1065 | } | |
1066 | ||
eced69b3 | 1067 | enum { |
35b8ca3a | 1068 | GC_OFFSET_IN_PATH, /* This one is on the path we were traveling */ |
eced69b3 LP |
1069 | GC_OFFSET_UNSURE, /* No clue */ |
1070 | GC_OFFSET_GOOD, /* We still need this unit */ | |
1071 | GC_OFFSET_BAD, /* We don't need this unit anymore */ | |
1072 | _GC_OFFSET_MAX | |
1073 | }; | |
1074 | ||
00d9ef85 | 1075 | static void unit_gc_mark_good(Unit *u, unsigned gc_marker) { |
4892084f | 1076 | Unit *other; |
eef85c4a LP |
1077 | Iterator i; |
1078 | void *v; | |
4892084f LN |
1079 | |
1080 | u->gc_marker = gc_marker + GC_OFFSET_GOOD; | |
1081 | ||
1082 | /* Recursively mark referenced units as GOOD as well */ | |
eef85c4a | 1083 | HASHMAP_FOREACH_KEY(v, other, u->dependencies[UNIT_REFERENCES], i) |
4892084f LN |
1084 | if (other->gc_marker == gc_marker + GC_OFFSET_UNSURE) |
1085 | unit_gc_mark_good(other, gc_marker); | |
1086 | } | |
1087 | ||
eced69b3 | 1088 | static void unit_gc_sweep(Unit *u, unsigned gc_marker) { |
701cc384 | 1089 | Unit *other; |
eced69b3 | 1090 | bool is_bad; |
eef85c4a LP |
1091 | Iterator i; |
1092 | void *v; | |
701cc384 LP |
1093 | |
1094 | assert(u); | |
1095 | ||
4c701096 YW |
1096 | if (IN_SET(u->gc_marker - gc_marker, |
1097 | GC_OFFSET_GOOD, GC_OFFSET_BAD, GC_OFFSET_UNSURE, GC_OFFSET_IN_PATH)) | |
701cc384 LP |
1098 | return; |
1099 | ||
ac155bb8 | 1100 | if (u->in_cleanup_queue) |
701cc384 LP |
1101 | goto bad; |
1102 | ||
f2f725e5 | 1103 | if (!unit_may_gc(u)) |
701cc384 LP |
1104 | goto good; |
1105 | ||
ac155bb8 | 1106 | u->gc_marker = gc_marker + GC_OFFSET_IN_PATH; |
eced69b3 LP |
1107 | |
1108 | is_bad = true; | |
1109 | ||
eef85c4a | 1110 | HASHMAP_FOREACH_KEY(v, other, u->dependencies[UNIT_REFERENCED_BY], i) { |
701cc384 LP |
1111 | unit_gc_sweep(other, gc_marker); |
1112 | ||
ac155bb8 | 1113 | if (other->gc_marker == gc_marker + GC_OFFSET_GOOD) |
701cc384 | 1114 | goto good; |
eced69b3 | 1115 | |
ac155bb8 | 1116 | if (other->gc_marker != gc_marker + GC_OFFSET_BAD) |
eced69b3 | 1117 | is_bad = false; |
701cc384 LP |
1118 | } |
1119 | ||
2641f02e ZJS |
1120 | if (u->refs_by_target) { |
1121 | const UnitRef *ref; | |
1122 | ||
1123 | LIST_FOREACH(refs_by_target, ref, u->refs_by_target) { | |
1124 | unit_gc_sweep(ref->source, gc_marker); | |
1125 | ||
1126 | if (ref->source->gc_marker == gc_marker + GC_OFFSET_GOOD) | |
1127 | goto good; | |
1128 | ||
1129 | if (ref->source->gc_marker != gc_marker + GC_OFFSET_BAD) | |
1130 | is_bad = false; | |
1131 | } | |
1132 | } | |
701cc384 | 1133 | |
eced69b3 LP |
1134 | if (is_bad) |
1135 | goto bad; | |
1136 | ||
1137 | /* We were unable to find anything out about this entry, so | |
1138 | * let's investigate it later */ | |
ac155bb8 | 1139 | u->gc_marker = gc_marker + GC_OFFSET_UNSURE; |
eced69b3 LP |
1140 | unit_add_to_gc_queue(u); |
1141 | return; | |
1142 | ||
701cc384 | 1143 | bad: |
eced69b3 LP |
1144 | /* We definitely know that this one is not useful anymore, so |
1145 | * let's mark it for deletion */ | |
ac155bb8 | 1146 | u->gc_marker = gc_marker + GC_OFFSET_BAD; |
eced69b3 | 1147 | unit_add_to_cleanup_queue(u); |
701cc384 LP |
1148 | return; |
1149 | ||
1150 | good: | |
4892084f | 1151 | unit_gc_mark_good(u, gc_marker); |
701cc384 LP |
1152 | } |
1153 | ||
c5a97ed1 LP |
1154 | static unsigned manager_dispatch_gc_unit_queue(Manager *m) { |
1155 | unsigned n = 0, gc_marker; | |
595ed347 | 1156 | Unit *u; |
701cc384 LP |
1157 | |
1158 | assert(m); | |
1159 | ||
cf1265e1 | 1160 | /* log_debug("Running GC..."); */ |
701cc384 | 1161 | |
eced69b3 LP |
1162 | m->gc_marker += _GC_OFFSET_MAX; |
1163 | if (m->gc_marker + _GC_OFFSET_MAX <= _GC_OFFSET_MAX) | |
c9c0cadb | 1164 | m->gc_marker = 1; |
701cc384 | 1165 | |
eced69b3 LP |
1166 | gc_marker = m->gc_marker; |
1167 | ||
c5a97ed1 | 1168 | while ((u = m->gc_unit_queue)) { |
595ed347 | 1169 | assert(u->in_gc_queue); |
701cc384 | 1170 | |
595ed347 | 1171 | unit_gc_sweep(u, gc_marker); |
eced69b3 | 1172 | |
c5a97ed1 | 1173 | LIST_REMOVE(gc_queue, m->gc_unit_queue, u); |
595ed347 | 1174 | u->in_gc_queue = false; |
701cc384 LP |
1175 | |
1176 | n++; | |
1177 | ||
4c701096 YW |
1178 | if (IN_SET(u->gc_marker - gc_marker, |
1179 | GC_OFFSET_BAD, GC_OFFSET_UNSURE)) { | |
cc3bc3e6 | 1180 | if (u->id) |
f2341e0a | 1181 | log_unit_debug(u, "Collecting."); |
595ed347 MS |
1182 | u->gc_marker = gc_marker + GC_OFFSET_BAD; |
1183 | unit_add_to_cleanup_queue(u); | |
701cc384 LP |
1184 | } |
1185 | } | |
1186 | ||
701cc384 LP |
1187 | return n; |
1188 | } | |
1189 | ||
c5a97ed1 LP |
1190 | static unsigned manager_dispatch_gc_job_queue(Manager *m) { |
1191 | unsigned n = 0; | |
1192 | Job *j; | |
1193 | ||
1194 | assert(m); | |
1195 | ||
1196 | while ((j = m->gc_job_queue)) { | |
1197 | assert(j->in_gc_queue); | |
1198 | ||
1199 | LIST_REMOVE(gc_queue, m->gc_job_queue, j); | |
1200 | j->in_gc_queue = false; | |
1201 | ||
1202 | n++; | |
1203 | ||
2ab3050f | 1204 | if (!job_may_gc(j)) |
c5a97ed1 LP |
1205 | continue; |
1206 | ||
1207 | log_unit_debug(j->unit, "Collecting job."); | |
1208 | (void) job_finish_and_invalidate(j, JOB_COLLECTED, false, false); | |
1209 | } | |
1210 | ||
1211 | return n; | |
1212 | } | |
1213 | ||
a16e1123 | 1214 | static void manager_clear_jobs_and_units(Manager *m) { |
a16e1123 | 1215 | Unit *u; |
60918275 LP |
1216 | |
1217 | assert(m); | |
1218 | ||
87f0e418 LP |
1219 | while ((u = hashmap_first(m->units))) |
1220 | unit_free(u); | |
964e0949 LP |
1221 | |
1222 | manager_dispatch_cleanup_queue(m); | |
1223 | ||
1224 | assert(!m->load_queue); | |
1225 | assert(!m->run_queue); | |
1226 | assert(!m->dbus_unit_queue); | |
1227 | assert(!m->dbus_job_queue); | |
1228 | assert(!m->cleanup_queue); | |
c5a97ed1 LP |
1229 | assert(!m->gc_unit_queue); |
1230 | assert(!m->gc_job_queue); | |
964e0949 | 1231 | |
964e0949 LP |
1232 | assert(hashmap_isempty(m->jobs)); |
1233 | assert(hashmap_isempty(m->units)); | |
9e9e2b72 MS |
1234 | |
1235 | m->n_on_console = 0; | |
1236 | m->n_running_jobs = 0; | |
1e75824c MK |
1237 | m->n_installed_jobs = 0; |
1238 | m->n_failed_jobs = 0; | |
a16e1123 LP |
1239 | } |
1240 | ||
06d8d842 | 1241 | Manager* manager_free(Manager *m) { |
a16e1123 | 1242 | UnitType c; |
35aba85a | 1243 | ExecDirectoryType dt; |
87f0e418 | 1244 | |
06d8d842 ZJS |
1245 | if (!m) |
1246 | return NULL; | |
a16e1123 LP |
1247 | |
1248 | manager_clear_jobs_and_units(m); | |
23a177ef | 1249 | |
7824bbeb LP |
1250 | for (c = 0; c < _UNIT_TYPE_MAX; c++) |
1251 | if (unit_vtable[c]->shutdown) | |
1252 | unit_vtable[c]->shutdown(m); | |
1253 | ||
a1f31f47 | 1254 | /* If we reexecute ourselves, we keep the root cgroup around */ |
c6c18be3 | 1255 | manager_shutdown_cgroup(m, m->exit_code != MANAGER_REEXECUTE); |
8e274523 | 1256 | |
07a78643 | 1257 | lookup_paths_flush_generator(&m->lookup_paths); |
5a1e9937 | 1258 | |
5e8d1c9a | 1259 | bus_done(m); |
ea430986 | 1260 | |
e8a565cb YW |
1261 | exec_runtime_vacuum(m); |
1262 | hashmap_free(m->exec_runtime_by_id); | |
1263 | ||
29206d46 LP |
1264 | dynamic_user_vacuum(m, false); |
1265 | hashmap_free(m->dynamic_users); | |
1266 | ||
87f0e418 | 1267 | hashmap_free(m->units); |
4b58153d | 1268 | hashmap_free(m->units_by_invocation_id); |
60918275 | 1269 | hashmap_free(m->jobs); |
62a76913 | 1270 | hashmap_free(m->watch_pids); |
05e343b7 | 1271 | hashmap_free(m->watch_bus); |
9152c765 | 1272 | |
95ae05c0 | 1273 | set_free(m->startup_units); |
f755e3b7 LP |
1274 | set_free(m->failed_units); |
1275 | ||
718db961 | 1276 | sd_event_source_unref(m->signal_event_source); |
575b300b | 1277 | sd_event_source_unref(m->sigchld_event_source); |
718db961 | 1278 | sd_event_source_unref(m->notify_event_source); |
d8fdc620 | 1279 | sd_event_source_unref(m->cgroups_agent_event_source); |
718db961 | 1280 | sd_event_source_unref(m->time_change_event_source); |
bbf5fd8e | 1281 | sd_event_source_unref(m->timezone_change_event_source); |
718db961 | 1282 | sd_event_source_unref(m->jobs_in_progress_event_source); |
752b5905 | 1283 | sd_event_source_unref(m->run_queue_event_source); |
00d9ef85 | 1284 | sd_event_source_unref(m->user_lookup_event_source); |
5f109056 | 1285 | sd_event_source_unref(m->sync_bus_names_event_source); |
718db961 | 1286 | |
03e334a1 LP |
1287 | safe_close(m->signal_fd); |
1288 | safe_close(m->notify_fd); | |
d8fdc620 | 1289 | safe_close(m->cgroups_agent_fd); |
03e334a1 | 1290 | safe_close(m->time_change_fd); |
00d9ef85 | 1291 | safe_close_pair(m->user_lookup_fds); |
718db961 | 1292 | |
e46b13c8 ZJS |
1293 | manager_close_ask_password(m); |
1294 | ||
718db961 LP |
1295 | manager_close_idle_pipe(m); |
1296 | ||
9670d583 | 1297 | udev_unref(m->udev); |
718db961 | 1298 | sd_event_unref(m->event); |
60918275 | 1299 | |
c952c6ec LP |
1300 | free(m->notify_socket); |
1301 | ||
84e3543e | 1302 | lookup_paths_free(&m->lookup_paths); |
1137a57c | 1303 | strv_free(m->environment); |
036643a2 | 1304 | |
4ad49000 | 1305 | hashmap_free(m->cgroup_unit); |
c6c18be3 | 1306 | set_free_free(m->unit_path_cache); |
33be102a | 1307 | |
664f88a7 LP |
1308 | free(m->switch_root); |
1309 | free(m->switch_root_init); | |
1310 | ||
31ce987c | 1311 | rlimit_free_all(m->rlimit); |
c93ff2e9 | 1312 | |
a57f7e2c LP |
1313 | assert(hashmap_isempty(m->units_requiring_mounts_for)); |
1314 | hashmap_free(m->units_requiring_mounts_for); | |
1315 | ||
00d9ef85 LP |
1316 | hashmap_free(m->uid_refs); |
1317 | hashmap_free(m->gid_refs); | |
1318 | ||
72fd1768 | 1319 | for (dt = 0; dt < _EXEC_DIRECTORY_TYPE_MAX; dt++) |
35aba85a YW |
1320 | m->prefix[dt] = mfree(m->prefix[dt]); |
1321 | ||
6b430fdb | 1322 | return mfree(m); |
60918275 LP |
1323 | } |
1324 | ||
04eb582a LP |
1325 | static void manager_enumerate_perpetual(Manager *m) { |
1326 | UnitType c; | |
1327 | ||
1328 | assert(m); | |
1329 | ||
1330 | /* Let's ask every type to load all units from disk/kernel that it might know */ | |
1331 | for (c = 0; c < _UNIT_TYPE_MAX; c++) { | |
1332 | if (!unit_type_supported(c)) { | |
1333 | log_debug("Unit type .%s is not supported on this system.", unit_type_to_string(c)); | |
1334 | continue; | |
1335 | } | |
1336 | ||
1337 | if (unit_vtable[c]->enumerate_perpetual) | |
1338 | unit_vtable[c]->enumerate_perpetual(m); | |
1339 | } | |
1340 | } | |
1341 | ||
a1113e08 | 1342 | static void manager_enumerate(Manager *m) { |
f50e0a01 | 1343 | UnitType c; |
f50e0a01 LP |
1344 | |
1345 | assert(m); | |
1346 | ||
04eb582a | 1347 | /* Let's ask every type to load all units from disk/kernel that it might know */ |
0faacd47 | 1348 | for (c = 0; c < _UNIT_TYPE_MAX; c++) { |
1c2e9646 | 1349 | if (!unit_type_supported(c)) { |
03afec3c | 1350 | log_debug("Unit type .%s is not supported on this system.", unit_type_to_string(c)); |
0faacd47 | 1351 | continue; |
a57f7e2c | 1352 | } |
f50e0a01 | 1353 | |
94b01dae ZJS |
1354 | if (unit_vtable[c]->enumerate) |
1355 | unit_vtable[c]->enumerate(m); | |
0faacd47 LP |
1356 | } |
1357 | ||
f50e0a01 | 1358 | manager_dispatch_load_queue(m); |
a16e1123 LP |
1359 | } |
1360 | ||
007c6337 | 1361 | static void manager_coldplug(Manager *m) { |
a16e1123 LP |
1362 | Iterator i; |
1363 | Unit *u; | |
1364 | char *k; | |
007c6337 | 1365 | int r; |
a16e1123 LP |
1366 | |
1367 | assert(m); | |
f50e0a01 | 1368 | |
f0831ed2 LP |
1369 | log_debug("Invoking unit coldplug() handlers…"); |
1370 | ||
1371 | /* Let's place the units back into their deserialized state */ | |
f50e0a01 LP |
1372 | HASHMAP_FOREACH_KEY(u, k, m->units, i) { |
1373 | ||
1374 | /* ignore aliases */ | |
ac155bb8 | 1375 | if (u->id != k) |
f50e0a01 LP |
1376 | continue; |
1377 | ||
007c6337 LP |
1378 | r = unit_coldplug(u); |
1379 | if (r < 0) | |
1380 | log_warning_errno(r, "We couldn't coldplug %s, proceeding anyway: %m", u->id); | |
f50e0a01 | 1381 | } |
a16e1123 LP |
1382 | } |
1383 | ||
f0831ed2 LP |
1384 | static void manager_catchup(Manager *m) { |
1385 | Iterator i; | |
1386 | Unit *u; | |
1387 | char *k; | |
1388 | ||
1389 | assert(m); | |
1390 | ||
1391 | log_debug("Invoking unit catchup() handlers…"); | |
1392 | ||
1393 | /* Let's catch up on any state changes that happened while we were reloading/reexecing */ | |
1394 | HASHMAP_FOREACH_KEY(u, k, m->units, i) { | |
1395 | ||
1396 | /* ignore aliases */ | |
1397 | if (u->id != k) | |
1398 | continue; | |
1399 | ||
1400 | unit_catchup(u); | |
1401 | } | |
1402 | } | |
1403 | ||
fe51822e LP |
1404 | static void manager_build_unit_path_cache(Manager *m) { |
1405 | char **i; | |
fe51822e LP |
1406 | int r; |
1407 | ||
1408 | assert(m); | |
1409 | ||
1410 | set_free_free(m->unit_path_cache); | |
1411 | ||
548f6937 | 1412 | m->unit_path_cache = set_new(&path_hash_ops); |
874310b7 | 1413 | if (!m->unit_path_cache) { |
d063a527 LP |
1414 | r = -ENOMEM; |
1415 | goto fail; | |
fe51822e LP |
1416 | } |
1417 | ||
1418 | /* This simply builds a list of files we know exist, so that | |
1419 | * we don't always have to go to disk */ | |
1420 | ||
a3c4eb07 | 1421 | STRV_FOREACH(i, m->lookup_paths.search_path) { |
d063a527 | 1422 | _cleanup_closedir_ DIR *d = NULL; |
fe51822e LP |
1423 | struct dirent *de; |
1424 | ||
bd0af849 ZJS |
1425 | d = opendir(*i); |
1426 | if (!d) { | |
874310b7 | 1427 | if (errno != ENOENT) |
d063a527 | 1428 | log_warning_errno(errno, "Failed to open directory %s, ignoring: %m", *i); |
fe51822e LP |
1429 | continue; |
1430 | } | |
1431 | ||
d063a527 | 1432 | FOREACH_DIRENT(de, d, r = -errno; goto fail) { |
fe51822e LP |
1433 | char *p; |
1434 | ||
605405c6 | 1435 | p = strjoin(streq(*i, "/") ? "" : *i, "/", de->d_name); |
44d91056 | 1436 | if (!p) { |
fe51822e LP |
1437 | r = -ENOMEM; |
1438 | goto fail; | |
1439 | } | |
1440 | ||
ef42202a ZJS |
1441 | r = set_consume(m->unit_path_cache, p); |
1442 | if (r < 0) | |
fe51822e | 1443 | goto fail; |
fe51822e | 1444 | } |
fe51822e LP |
1445 | } |
1446 | ||
1447 | return; | |
1448 | ||
1449 | fail: | |
d063a527 LP |
1450 | log_warning_errno(r, "Failed to build unit path cache, proceeding without: %m"); |
1451 | m->unit_path_cache = set_free_free(m->unit_path_cache); | |
fe51822e LP |
1452 | } |
1453 | ||
9ff1a6f1 | 1454 | static void manager_distribute_fds(Manager *m, FDSet *fds) { |
9588bc32 | 1455 | Iterator i; |
9ff1a6f1 | 1456 | Unit *u; |
9588bc32 LP |
1457 | |
1458 | assert(m); | |
1459 | ||
1460 | HASHMAP_FOREACH(u, m->units, i) { | |
1461 | ||
1462 | if (fdset_size(fds) <= 0) | |
1463 | break; | |
1464 | ||
9ff1a6f1 LP |
1465 | if (!UNIT_VTABLE(u)->distribute_fds) |
1466 | continue; | |
9588bc32 | 1467 | |
9ff1a6f1 LP |
1468 | UNIT_VTABLE(u)->distribute_fds(u, fds); |
1469 | } | |
9588bc32 LP |
1470 | } |
1471 | ||
8559b3b7 LP |
1472 | static bool manager_dbus_is_running(Manager *m, bool deserialized) { |
1473 | Unit *u; | |
1474 | ||
1475 | assert(m); | |
1476 | ||
1477 | /* This checks whether the dbus instance we are supposed to expose our APIs on is up. We check both the socket | |
1478 | * and the service unit. If the 'deserialized' parameter is true we'll check the deserialized state of the unit | |
1479 | * rather than the current one. */ | |
1480 | ||
1481 | if (m->test_run_flags != 0) | |
1482 | return false; | |
1483 | ||
8559b3b7 LP |
1484 | u = manager_get_unit(m, SPECIAL_DBUS_SOCKET); |
1485 | if (!u) | |
1486 | return false; | |
1487 | if ((deserialized ? SOCKET(u)->deserialized_state : SOCKET(u)->state) != SOCKET_RUNNING) | |
1488 | return false; | |
1489 | ||
1490 | u = manager_get_unit(m, SPECIAL_DBUS_SERVICE); | |
1491 | if (!u) | |
1492 | return false; | |
1493 | if (!IN_SET((deserialized ? SERVICE(u)->deserialized_state : SERVICE(u)->state), SERVICE_RUNNING, SERVICE_RELOAD)) | |
1494 | return false; | |
1495 | ||
1496 | return true; | |
1497 | } | |
1498 | ||
9d4c195c LP |
1499 | static void manager_setup_bus(Manager *m) { |
1500 | assert(m); | |
1501 | ||
1502 | /* Let's set up our private bus connection now, unconditionally */ | |
1503 | (void) bus_init_private(m); | |
1504 | ||
1505 | /* If we are in --user mode also connect to the system bus now */ | |
1506 | if (MANAGER_IS_USER(m)) | |
1507 | (void) bus_init_system(m); | |
1508 | ||
1509 | /* Let's connect to the bus now, but only if the unit is supposed to be up */ | |
1510 | if (manager_dbus_is_running(m, MANAGER_IS_RELOADING(m))) { | |
1511 | (void) bus_init_api(m); | |
1512 | ||
1513 | if (MANAGER_IS_SYSTEM(m)) | |
1514 | (void) bus_init_system(m); | |
1515 | } | |
1516 | } | |
1517 | ||
159f1e76 LP |
1518 | static void manager_preset_all(Manager *m) { |
1519 | int r; | |
1520 | ||
1521 | assert(m); | |
1522 | ||
1523 | if (m->first_boot <= 0) | |
1524 | return; | |
1525 | ||
1526 | if (!MANAGER_IS_SYSTEM(m)) | |
1527 | return; | |
1528 | ||
1529 | if (m->test_run_flags != 0) | |
1530 | return; | |
1531 | ||
1532 | /* If this is the first boot, and we are in the host system, then preset everything */ | |
1533 | r = unit_file_preset_all(UNIT_FILE_SYSTEM, 0, NULL, UNIT_FILE_PRESET_ENABLE_ONLY, NULL, 0); | |
1534 | if (r < 0) | |
1535 | log_full_errno(r == -EEXIST ? LOG_NOTICE : LOG_WARNING, r, | |
1536 | "Failed to populate /etc with preset unit settings, ignoring: %m"); | |
1537 | else | |
1538 | log_info("Populated /etc with preset unit settings."); | |
1539 | } | |
1540 | ||
a16e1123 | 1541 | int manager_startup(Manager *m, FILE *serialization, FDSet *fds) { |
17f01ace | 1542 | int r; |
a16e1123 LP |
1543 | |
1544 | assert(m); | |
1545 | ||
a1f31f47 ZJS |
1546 | /* If we are running in test mode, we still want to run the generators, |
1547 | * but we should not touch the real generator directories. */ | |
1548 | r = lookup_paths_init(&m->lookup_paths, m->unit_file_scope, | |
e0a3da1f | 1549 | m->test_run_flags ? LOOKUP_PATHS_TEMPORARY_GENERATED : 0, |
a1f31f47 | 1550 | NULL); |
e801700e ZJS |
1551 | if (r < 0) |
1552 | return r; | |
5a1e9937 | 1553 | |
64691d20 ZJS |
1554 | r = manager_run_environment_generators(m); |
1555 | if (r < 0) | |
1556 | return r; | |
1557 | ||
d4ee7bd8 | 1558 | dual_timestamp_get(m->timestamps + manager_timestamp_initrd_mangle(MANAGER_TIMESTAMP_GENERATORS_START)); |
a3c4eb07 | 1559 | r = manager_run_generators(m); |
d4ee7bd8 | 1560 | dual_timestamp_get(m->timestamps + manager_timestamp_initrd_mangle(MANAGER_TIMESTAMP_GENERATORS_FINISH)); |
07719a21 LP |
1561 | if (r < 0) |
1562 | return r; | |
1563 | ||
159f1e76 | 1564 | manager_preset_all(m); |
a1453343 | 1565 | lookup_paths_reduce(&m->lookup_paths); |
fe51822e LP |
1566 | manager_build_unit_path_cache(m); |
1567 | ||
9f611ad8 LP |
1568 | /* If we will deserialize make sure that during enumeration |
1569 | * this is already known, so we increase the counter here | |
1570 | * already */ | |
1571 | if (serialization) | |
313cefa1 | 1572 | m->n_reloading++; |
9f611ad8 | 1573 | |
a16e1123 | 1574 | /* First, enumerate what we can from all config files */ |
d4ee7bd8 | 1575 | dual_timestamp_get(m->timestamps + manager_timestamp_initrd_mangle(MANAGER_TIMESTAMP_UNITS_LOAD_START)); |
04eb582a | 1576 | manager_enumerate_perpetual(m); |
ba64af90 | 1577 | manager_enumerate(m); |
d4ee7bd8 | 1578 | dual_timestamp_get(m->timestamps + manager_timestamp_initrd_mangle(MANAGER_TIMESTAMP_UNITS_LOAD_FINISH)); |
a16e1123 LP |
1579 | |
1580 | /* Second, deserialize if there is something to deserialize */ | |
07429866 | 1581 | if (serialization) { |
1cd974ed | 1582 | r = manager_deserialize(m, serialization, fds); |
07429866 | 1583 | if (r < 0) |
17f01ace | 1584 | return log_error_errno(r, "Deserialization failed: %m"); |
07429866 | 1585 | } |
a16e1123 | 1586 | |
01e10de3 LP |
1587 | /* Any fds left? Find some unit which wants them. This is |
1588 | * useful to allow container managers to pass some file | |
1589 | * descriptors to us pre-initialized. This enables | |
1590 | * socket-based activation of entire containers. */ | |
9ff1a6f1 | 1591 | manager_distribute_fds(m, fds); |
01e10de3 | 1592 | |
d86f9d52 LP |
1593 | /* We might have deserialized the notify fd, but if we didn't |
1594 | * then let's create the bus now */ | |
17f01ace ZJS |
1595 | r = manager_setup_notify(m); |
1596 | if (r < 0) | |
1597 | /* No sense to continue without notifications, our children would fail anyway. */ | |
1598 | return r; | |
d86f9d52 | 1599 | |
17f01ace ZJS |
1600 | r = manager_setup_cgroups_agent(m); |
1601 | if (r < 0) | |
1602 | /* Likewise, no sense to continue without empty cgroup notifications. */ | |
1603 | return r; | |
d8fdc620 | 1604 | |
17f01ace ZJS |
1605 | r = manager_setup_user_lookup_fd(m); |
1606 | if (r < 0) | |
1607 | /* This shouldn't fail, except if things are really broken. */ | |
1608 | return r; | |
00d9ef85 | 1609 | |
9d4c195c LP |
1610 | /* Connect to the bus if we are good for it */ |
1611 | manager_setup_bus(m); | |
8559b3b7 LP |
1612 | |
1613 | /* Now that we are connected to all possible busses, let's deserialize who is tracking us. */ | |
05a98afd LP |
1614 | (void) bus_track_coldplug(m, &m->subscribed, false, m->deserialized_subscribed); |
1615 | m->deserialized_subscribed = strv_free(m->deserialized_subscribed); | |
e3dd987c | 1616 | |
a16e1123 | 1617 | /* Third, fire things up! */ |
007c6337 | 1618 | manager_coldplug(m); |
a16e1123 | 1619 | |
29206d46 LP |
1620 | /* Release any dynamic users no longer referenced */ |
1621 | dynamic_user_vacuum(m, true); | |
1622 | ||
e8a565cb YW |
1623 | exec_runtime_vacuum(m); |
1624 | ||
00d9ef85 LP |
1625 | /* Release any references to UIDs/GIDs no longer referenced, and destroy any IPC owned by them */ |
1626 | manager_vacuum_uid_refs(m); | |
1627 | manager_vacuum_gid_refs(m); | |
1628 | ||
9f611ad8 | 1629 | if (serialization) { |
a7556052 | 1630 | assert(m->n_reloading > 0); |
313cefa1 | 1631 | m->n_reloading--; |
71445ae7 LP |
1632 | |
1633 | /* Let's wait for the UnitNew/JobNew messages being | |
1634 | * sent, before we notify that the reload is | |
1635 | * finished */ | |
1636 | m->send_reloading_done = true; | |
9f611ad8 LP |
1637 | } |
1638 | ||
f0831ed2 LP |
1639 | /* Let's finally catch up with any changes that took place while we were reloading/reexecing */ |
1640 | manager_catchup(m); | |
1641 | ||
17f01ace | 1642 | return 0; |
f50e0a01 LP |
1643 | } |
1644 | ||
4bd29fe5 | 1645 | int manager_add_job(Manager *m, JobType type, Unit *unit, JobMode mode, sd_bus_error *e, Job **_ret) { |
e5b5ae50 | 1646 | int r; |
7527cb52 | 1647 | Transaction *tr; |
e5b5ae50 LP |
1648 | |
1649 | assert(m); | |
1650 | assert(type < _JOB_TYPE_MAX); | |
87f0e418 | 1651 | assert(unit); |
e5b5ae50 | 1652 | assert(mode < _JOB_MODE_MAX); |
60918275 | 1653 | |
7358dc02 ZJS |
1654 | if (mode == JOB_ISOLATE && type != JOB_START) |
1655 | return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Isolate is only valid for start."); | |
c497c7a9 | 1656 | |
7358dc02 ZJS |
1657 | if (mode == JOB_ISOLATE && !unit->allow_isolate) |
1658 | return sd_bus_error_setf(e, BUS_ERROR_NO_ISOLATION, "Operation refused, unit may not be isolated."); | |
2528a7a6 | 1659 | |
f2341e0a | 1660 | log_unit_debug(unit, "Trying to enqueue job %s/%s/%s", unit->id, job_type_to_string(type), job_mode_to_string(mode)); |
9f04bd52 | 1661 | |
c6497ccb | 1662 | type = job_type_collapse(type, unit); |
e0209d83 | 1663 | |
23ade460 | 1664 | tr = transaction_new(mode == JOB_REPLACE_IRREVERSIBLY); |
7527cb52 MS |
1665 | if (!tr) |
1666 | return -ENOMEM; | |
11dd41ce | 1667 | |
4bd29fe5 | 1668 | r = transaction_add_job_and_dependencies(tr, type, unit, NULL, true, false, |
3742095b | 1669 | IN_SET(mode, JOB_IGNORE_DEPENDENCIES, JOB_IGNORE_REQUIREMENTS), |
b94fbd30 | 1670 | mode == JOB_IGNORE_DEPENDENCIES, e); |
7527cb52 MS |
1671 | if (r < 0) |
1672 | goto tr_abort; | |
c497c7a9 | 1673 | |
7527cb52 MS |
1674 | if (mode == JOB_ISOLATE) { |
1675 | r = transaction_add_isolate_jobs(tr, m); | |
1676 | if (r < 0) | |
1677 | goto tr_abort; | |
1678 | } | |
1679 | ||
1680 | r = transaction_activate(tr, m, mode, e); | |
1681 | if (r < 0) | |
1682 | goto tr_abort; | |
e5b5ae50 | 1683 | |
f2341e0a | 1684 | log_unit_debug(unit, |
66870f90 ZJS |
1685 | "Enqueued job %s/%s as %u", unit->id, |
1686 | job_type_to_string(type), (unsigned) tr->anchor_job->id); | |
f50e0a01 | 1687 | |
e5b5ae50 | 1688 | if (_ret) |
b94fbd30 | 1689 | *_ret = tr->anchor_job; |
60918275 | 1690 | |
7527cb52 | 1691 | transaction_free(tr); |
e5b5ae50 | 1692 | return 0; |
7527cb52 MS |
1693 | |
1694 | tr_abort: | |
1695 | transaction_abort(tr); | |
1696 | transaction_free(tr); | |
1697 | return r; | |
e5b5ae50 | 1698 | } |
60918275 | 1699 | |
53f18416 | 1700 | int manager_add_job_by_name(Manager *m, JobType type, const char *name, JobMode mode, sd_bus_error *e, Job **ret) { |
4440b27d | 1701 | Unit *unit = NULL; /* just to appease gcc, initialization is not really necessary */ |
28247076 LP |
1702 | int r; |
1703 | ||
1704 | assert(m); | |
1705 | assert(type < _JOB_TYPE_MAX); | |
1706 | assert(name); | |
1707 | assert(mode < _JOB_MODE_MAX); | |
1708 | ||
c3090674 LP |
1709 | r = manager_load_unit(m, name, NULL, NULL, &unit); |
1710 | if (r < 0) | |
28247076 | 1711 | return r; |
4440b27d | 1712 | assert(unit); |
28247076 | 1713 | |
53f18416 LP |
1714 | return manager_add_job(m, type, unit, mode, e, ret); |
1715 | } | |
1716 | ||
1717 | int manager_add_job_by_name_and_warn(Manager *m, JobType type, const char *name, JobMode mode, Job **ret) { | |
4afd3348 | 1718 | _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; |
53f18416 LP |
1719 | int r; |
1720 | ||
1721 | assert(m); | |
1722 | assert(type < _JOB_TYPE_MAX); | |
1723 | assert(name); | |
1724 | assert(mode < _JOB_MODE_MAX); | |
1725 | ||
1726 | r = manager_add_job_by_name(m, type, name, mode, &error, ret); | |
1727 | if (r < 0) | |
1728 | return log_warning_errno(r, "Failed to enqueue %s job for %s: %s", job_mode_to_string(mode), name, bus_error_message(&error, r)); | |
1729 | ||
1730 | return r; | |
28247076 LP |
1731 | } |
1732 | ||
15d167f8 JW |
1733 | int manager_propagate_reload(Manager *m, Unit *unit, JobMode mode, sd_bus_error *e) { |
1734 | int r; | |
1735 | Transaction *tr; | |
1736 | ||
1737 | assert(m); | |
1738 | assert(unit); | |
1739 | assert(mode < _JOB_MODE_MAX); | |
1740 | assert(mode != JOB_ISOLATE); /* Isolate is only valid for start */ | |
1741 | ||
1742 | tr = transaction_new(mode == JOB_REPLACE_IRREVERSIBLY); | |
1743 | if (!tr) | |
1744 | return -ENOMEM; | |
1745 | ||
1746 | /* We need an anchor job */ | |
1747 | r = transaction_add_job_and_dependencies(tr, JOB_NOP, unit, NULL, false, false, true, true, e); | |
1748 | if (r < 0) | |
1749 | goto tr_abort; | |
1750 | ||
1751 | /* Failure in adding individual dependencies is ignored, so this always succeeds. */ | |
1752 | transaction_add_propagate_reload_jobs(tr, unit, tr->anchor_job, mode == JOB_IGNORE_DEPENDENCIES, e); | |
1753 | ||
1754 | r = transaction_activate(tr, m, mode, e); | |
1755 | if (r < 0) | |
1756 | goto tr_abort; | |
1757 | ||
1758 | transaction_free(tr); | |
1759 | return 0; | |
1760 | ||
1761 | tr_abort: | |
1762 | transaction_abort(tr); | |
1763 | transaction_free(tr); | |
1764 | return r; | |
1765 | } | |
1766 | ||
60918275 LP |
1767 | Job *manager_get_job(Manager *m, uint32_t id) { |
1768 | assert(m); | |
1769 | ||
1770 | return hashmap_get(m->jobs, UINT32_TO_PTR(id)); | |
1771 | } | |
1772 | ||
87f0e418 | 1773 | Unit *manager_get_unit(Manager *m, const char *name) { |
60918275 LP |
1774 | assert(m); |
1775 | assert(name); | |
1776 | ||
87f0e418 | 1777 | return hashmap_get(m->units, name); |
60918275 LP |
1778 | } |
1779 | ||
19496554 MS |
1780 | static int manager_dispatch_target_deps_queue(Manager *m) { |
1781 | Unit *u; | |
1782 | unsigned k; | |
1783 | int r = 0; | |
1784 | ||
1785 | static const UnitDependency deps[] = { | |
1786 | UNIT_REQUIRED_BY, | |
1787 | UNIT_REQUISITE_OF, | |
1788 | UNIT_WANTED_BY, | |
1789 | UNIT_BOUND_BY | |
1790 | }; | |
1791 | ||
1792 | assert(m); | |
1793 | ||
1794 | while ((u = m->target_deps_queue)) { | |
1795 | assert(u->in_target_deps_queue); | |
1796 | ||
1797 | LIST_REMOVE(target_deps_queue, u->manager->target_deps_queue, u); | |
1798 | u->in_target_deps_queue = false; | |
1799 | ||
1800 | for (k = 0; k < ELEMENTSOF(deps); k++) { | |
1801 | Unit *target; | |
1802 | Iterator i; | |
1803 | void *v; | |
1804 | ||
1805 | HASHMAP_FOREACH_KEY(v, target, u->dependencies[deps[k]], i) { | |
1806 | r = unit_add_default_target_dependency(u, target); | |
1807 | if (r < 0) | |
1808 | return r; | |
1809 | } | |
1810 | } | |
1811 | } | |
1812 | ||
1813 | return r; | |
1814 | } | |
1815 | ||
c1e1601e | 1816 | unsigned manager_dispatch_load_queue(Manager *m) { |
595ed347 | 1817 | Unit *u; |
c1e1601e | 1818 | unsigned n = 0; |
60918275 LP |
1819 | |
1820 | assert(m); | |
1821 | ||
223dabab LP |
1822 | /* Make sure we are not run recursively */ |
1823 | if (m->dispatching_load_queue) | |
c1e1601e | 1824 | return 0; |
223dabab LP |
1825 | |
1826 | m->dispatching_load_queue = true; | |
1827 | ||
87f0e418 | 1828 | /* Dispatches the load queue. Takes a unit from the queue and |
60918275 LP |
1829 | * tries to load its data until the queue is empty */ |
1830 | ||
595ed347 MS |
1831 | while ((u = m->load_queue)) { |
1832 | assert(u->in_load_queue); | |
034c6ed7 | 1833 | |
595ed347 | 1834 | unit_load(u); |
c1e1601e | 1835 | n++; |
60918275 LP |
1836 | } |
1837 | ||
223dabab | 1838 | m->dispatching_load_queue = false; |
19496554 MS |
1839 | |
1840 | /* Dispatch the units waiting for their target dependencies to be added now, as all targets that we know about | |
1841 | * should be loaded and have aliases resolved */ | |
1842 | (void) manager_dispatch_target_deps_queue(m); | |
1843 | ||
c1e1601e | 1844 | return n; |
60918275 LP |
1845 | } |
1846 | ||
c2756a68 LP |
1847 | int manager_load_unit_prepare( |
1848 | Manager *m, | |
1849 | const char *name, | |
1850 | const char *path, | |
718db961 | 1851 | sd_bus_error *e, |
c2756a68 LP |
1852 | Unit **_ret) { |
1853 | ||
dc409696 | 1854 | _cleanup_(unit_freep) Unit *cleanup_ret = NULL; |
87f0e418 | 1855 | Unit *ret; |
7d17cfbc | 1856 | UnitType t; |
60918275 LP |
1857 | int r; |
1858 | ||
1859 | assert(m); | |
9e2f7c11 | 1860 | assert(name || path); |
7a6a095a | 1861 | assert(_ret); |
60918275 | 1862 | |
db06e3b6 LP |
1863 | /* This will prepare the unit for loading, but not actually |
1864 | * load anything from disk. */ | |
0301abf4 | 1865 | |
718db961 LP |
1866 | if (path && !is_path(path)) |
1867 | return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Path %s is not absolute.", path); | |
9e2f7c11 LP |
1868 | |
1869 | if (!name) | |
2b6bf07d | 1870 | name = basename(path); |
9e2f7c11 | 1871 | |
7d17cfbc MS |
1872 | t = unit_name_to_type(name); |
1873 | ||
5d512d54 LN |
1874 | if (t == _UNIT_TYPE_INVALID || !unit_name_is_valid(name, UNIT_NAME_PLAIN|UNIT_NAME_INSTANCE)) { |
1875 | if (unit_name_is_valid(name, UNIT_NAME_TEMPLATE)) | |
1876 | return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Unit name %s is missing the instance name.", name); | |
1877 | ||
718db961 | 1878 | return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, "Unit name %s is not valid.", name); |
5d512d54 | 1879 | } |
60918275 | 1880 | |
7d17cfbc MS |
1881 | ret = manager_get_unit(m, name); |
1882 | if (ret) { | |
034c6ed7 | 1883 | *_ret = ret; |
413d6313 | 1884 | return 1; |
034c6ed7 | 1885 | } |
60918275 | 1886 | |
dc409696 | 1887 | ret = cleanup_ret = unit_new(m, unit_vtable[t]->object_size); |
7d17cfbc | 1888 | if (!ret) |
60918275 LP |
1889 | return -ENOMEM; |
1890 | ||
7d17cfbc | 1891 | if (path) { |
ac155bb8 | 1892 | ret->fragment_path = strdup(path); |
dc409696 | 1893 | if (!ret->fragment_path) |
0301abf4 | 1894 | return -ENOMEM; |
7d17cfbc | 1895 | } |
0301abf4 | 1896 | |
1058cbf2 | 1897 | r = unit_add_name(ret, name); |
dc409696 | 1898 | if (r < 0) |
1ffba6fe | 1899 | return r; |
60918275 | 1900 | |
87f0e418 | 1901 | unit_add_to_load_queue(ret); |
c1e1601e | 1902 | unit_add_to_dbus_queue(ret); |
949061f0 | 1903 | unit_add_to_gc_queue(ret); |
c1e1601e | 1904 | |
7a6a095a | 1905 | *_ret = ret; |
dc409696 | 1906 | cleanup_ret = NULL; |
db06e3b6 LP |
1907 | |
1908 | return 0; | |
1909 | } | |
1910 | ||
c2756a68 LP |
1911 | int manager_load_unit( |
1912 | Manager *m, | |
1913 | const char *name, | |
1914 | const char *path, | |
718db961 | 1915 | sd_bus_error *e, |
c2756a68 LP |
1916 | Unit **_ret) { |
1917 | ||
db06e3b6 LP |
1918 | int r; |
1919 | ||
1920 | assert(m); | |
7a6a095a | 1921 | assert(_ret); |
db06e3b6 LP |
1922 | |
1923 | /* This will load the service information files, but not actually | |
1924 | * start any services or anything. */ | |
1925 | ||
c3090674 LP |
1926 | r = manager_load_unit_prepare(m, name, path, e, _ret); |
1927 | if (r != 0) | |
db06e3b6 LP |
1928 | return r; |
1929 | ||
f50e0a01 | 1930 | manager_dispatch_load_queue(m); |
60918275 | 1931 | |
7a6a095a | 1932 | *_ret = unit_follow_merge(*_ret); |
4109ede7 ZJS |
1933 | return 0; |
1934 | } | |
1935 | ||
1936 | int manager_load_startable_unit_or_warn( | |
1937 | Manager *m, | |
1938 | const char *name, | |
1939 | const char *path, | |
1940 | Unit **ret) { | |
1941 | ||
1942 | /* Load a unit, make sure it loaded fully and is not masked. */ | |
1943 | ||
1944 | _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; | |
1945 | Unit *unit; | |
1946 | int r; | |
1947 | ||
1948 | r = manager_load_unit(m, name, path, &error, &unit); | |
1949 | if (r < 0) | |
1950 | return log_error_errno(r, "Failed to load %s %s: %s", | |
8ace1db7 | 1951 | name ? "unit" : "unit file", name ?: path, |
4109ede7 | 1952 | bus_error_message(&error, r)); |
8ace1db7 LP |
1953 | |
1954 | r = bus_unit_validate_load_state(unit, &error); | |
1955 | if (r < 0) | |
1956 | return log_error_errno(r, "%s", bus_error_message(&error, r)); | |
9e2f7c11 | 1957 | |
4109ede7 | 1958 | *ret = unit; |
60918275 LP |
1959 | return 0; |
1960 | } | |
a66d02c3 | 1961 | |
cea8e32e | 1962 | void manager_dump_jobs(Manager *s, FILE *f, const char *prefix) { |
034c6ed7 | 1963 | Iterator i; |
a66d02c3 LP |
1964 | Job *j; |
1965 | ||
1966 | assert(s); | |
1967 | assert(f); | |
1968 | ||
034c6ed7 | 1969 | HASHMAP_FOREACH(j, s->jobs, i) |
cea8e32e | 1970 | job_dump(j, f, prefix); |
a66d02c3 LP |
1971 | } |
1972 | ||
87f0e418 | 1973 | void manager_dump_units(Manager *s, FILE *f, const char *prefix) { |
034c6ed7 | 1974 | Iterator i; |
87f0e418 | 1975 | Unit *u; |
11dd41ce | 1976 | const char *t; |
a66d02c3 LP |
1977 | |
1978 | assert(s); | |
1979 | assert(f); | |
1980 | ||
87f0e418 | 1981 | HASHMAP_FOREACH_KEY(u, t, s->units, i) |
ac155bb8 | 1982 | if (u->id == t) |
87f0e418 | 1983 | unit_dump(u, f, prefix); |
a66d02c3 | 1984 | } |
7fad411c | 1985 | |
ad75b9e7 LP |
1986 | void manager_dump(Manager *m, FILE *f, const char *prefix) { |
1987 | ManagerTimestamp q; | |
1988 | ||
1989 | assert(m); | |
1990 | assert(f); | |
1991 | ||
1992 | for (q = 0; q < _MANAGER_TIMESTAMP_MAX; q++) { | |
1993 | char buf[FORMAT_TIMESTAMP_MAX]; | |
1994 | ||
1995 | if (dual_timestamp_is_set(m->timestamps + q)) | |
1996 | fprintf(f, "%sTimestamp %s: %s\n", | |
1997 | strempty(prefix), | |
1998 | manager_timestamp_to_string(q), | |
1999 | format_timestamp(buf, sizeof(buf), m->timestamps[q].realtime)); | |
2000 | } | |
2001 | ||
2002 | manager_dump_units(m, f, prefix); | |
2003 | manager_dump_jobs(m, f, prefix); | |
2004 | } | |
2005 | ||
713f6f90 LP |
2006 | int manager_get_dump_string(Manager *m, char **ret) { |
2007 | _cleanup_free_ char *dump = NULL; | |
2008 | _cleanup_fclose_ FILE *f = NULL; | |
2009 | size_t size; | |
2010 | int r; | |
2011 | ||
2012 | assert(m); | |
2013 | assert(ret); | |
2014 | ||
2015 | f = open_memstream(&dump, &size); | |
2016 | if (!f) | |
2017 | return -errno; | |
2018 | ||
0d536673 | 2019 | (void) __fsetlocking(f, FSETLOCKING_BYCALLER); |
713f6f90 LP |
2020 | |
2021 | manager_dump(m, f, NULL); | |
2022 | ||
2023 | r = fflush_and_check(f); | |
2024 | if (r < 0) | |
2025 | return r; | |
2026 | ||
2027 | f = safe_fclose(f); | |
2028 | ||
1cc6c93a | 2029 | *ret = TAKE_PTR(dump); |
713f6f90 LP |
2030 | |
2031 | return 0; | |
2032 | } | |
2033 | ||
7fad411c LP |
2034 | void manager_clear_jobs(Manager *m) { |
2035 | Job *j; | |
2036 | ||
2037 | assert(m); | |
2038 | ||
7fad411c | 2039 | while ((j = hashmap_first(m->jobs))) |
5273510e | 2040 | /* No need to recurse. We're cancelling all jobs. */ |
833f92ad | 2041 | job_finish_and_invalidate(j, JOB_CANCELED, false, false); |
7fad411c | 2042 | } |
83c60c9f | 2043 | |
752b5905 LP |
2044 | static int manager_dispatch_run_queue(sd_event_source *source, void *userdata) { |
2045 | Manager *m = userdata; | |
83c60c9f | 2046 | Job *j; |
034c6ed7 | 2047 | |
752b5905 LP |
2048 | assert(source); |
2049 | assert(m); | |
9152c765 | 2050 | |
034c6ed7 | 2051 | while ((j = m->run_queue)) { |
ac1135be | 2052 | assert(j->installed); |
034c6ed7 LP |
2053 | assert(j->in_run_queue); |
2054 | ||
2055 | job_run_and_invalidate(j); | |
9152c765 | 2056 | } |
034c6ed7 | 2057 | |
a0b64226 | 2058 | if (m->n_running_jobs > 0) |
03b717a3 MS |
2059 | manager_watch_jobs_in_progress(m); |
2060 | ||
31a7eb86 ZJS |
2061 | if (m->n_on_console > 0) |
2062 | manager_watch_idle_pipe(m); | |
2063 | ||
752b5905 | 2064 | return 1; |
c1e1601e LP |
2065 | } |
2066 | ||
9588bc32 | 2067 | static unsigned manager_dispatch_dbus_queue(Manager *m) { |
e0a08581 | 2068 | unsigned n = 0, budget; |
595ed347 | 2069 | Unit *u; |
e0a08581 | 2070 | Job *j; |
c1e1601e LP |
2071 | |
2072 | assert(m); | |
2073 | ||
2074 | if (m->dispatching_dbus_queue) | |
2075 | return 0; | |
2076 | ||
e0a08581 LP |
2077 | /* Anything to do at all? */ |
2078 | if (!m->dbus_unit_queue && !m->dbus_job_queue && !m->send_reloading_done && !m->queued_message) | |
2079 | return 0; | |
2080 | ||
2081 | /* Do we have overly many messages queued at the moment? If so, let's not enqueue more on top, let's sit this | |
2082 | * cycle out, and process things in a later cycle when the queues got a bit emptier. */ | |
2083 | if (manager_bus_n_queued_write(m) > MANAGER_BUS_BUSY_THRESHOLD) | |
2084 | return 0; | |
2085 | ||
2086 | /* Only process a certain number of units/jobs per event loop iteration. Even if the bus queue wasn't overly | |
2087 | * full before this call we shouldn't increase it in size too wildly in one step, and we shouldn't monopolize | |
2088 | * CPU time with generating these messages. Note the difference in counting of this "budget" and the | |
2089 | * "threshold" above: the "budget" is decreased only once per generated message, regardless how many | |
2090 | * busses/direct connections it is enqueued on, while the "threshold" is applied to each queued instance of bus | |
2091 | * message, i.e. if the same message is enqueued to five busses/direct connections it will be counted five | |
2092 | * times. This difference in counting ("references" vs. "instances") is primarily a result of the fact that | |
2093 | * it's easier to implement it this way, however it also reflects the thinking that the "threshold" should put | |
2094 | * a limit on used queue memory, i.e. space, while the "budget" should put a limit on time. Also note that | |
2095 | * the "threshold" is currently chosen much higher than the "budget". */ | |
2096 | budget = MANAGER_BUS_MESSAGE_BUDGET; | |
2097 | ||
c1e1601e LP |
2098 | m->dispatching_dbus_queue = true; |
2099 | ||
e0a08581 LP |
2100 | while (budget > 0 && (u = m->dbus_unit_queue)) { |
2101 | ||
595ed347 | 2102 | assert(u->in_dbus_queue); |
c1e1601e | 2103 | |
595ed347 | 2104 | bus_unit_send_change_signal(u); |
e0a08581 | 2105 | n++, budget--; |
c1e1601e LP |
2106 | } |
2107 | ||
e0a08581 | 2108 | while (budget > 0 && (j = m->dbus_job_queue)) { |
c1e1601e LP |
2109 | assert(j->in_dbus_queue); |
2110 | ||
2111 | bus_job_send_change_signal(j); | |
e0a08581 | 2112 | n++, budget--; |
c1e1601e LP |
2113 | } |
2114 | ||
2115 | m->dispatching_dbus_queue = false; | |
71445ae7 | 2116 | |
e0a08581 | 2117 | if (budget > 0 && m->send_reloading_done) { |
71445ae7 | 2118 | m->send_reloading_done = false; |
718db961 | 2119 | bus_manager_send_reloading(m, false); |
e0a08581 | 2120 | n++, budget--; |
71445ae7 LP |
2121 | } |
2122 | ||
e0a08581 | 2123 | if (budget > 0 && m->queued_message) { |
718db961 | 2124 | bus_send_queued_message(m); |
e0a08581 LP |
2125 | n++; |
2126 | } | |
718db961 | 2127 | |
c1e1601e | 2128 | return n; |
9152c765 LP |
2129 | } |
2130 | ||
d8fdc620 LP |
2131 | static int manager_dispatch_cgroups_agent_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) { |
2132 | Manager *m = userdata; | |
2133 | char buf[PATH_MAX+1]; | |
2134 | ssize_t n; | |
2135 | ||
2136 | n = recv(fd, buf, sizeof(buf), 0); | |
2137 | if (n < 0) | |
2138 | return log_error_errno(errno, "Failed to read cgroups agent message: %m"); | |
2139 | if (n == 0) { | |
2140 | log_error("Got zero-length cgroups agent message, ignoring."); | |
2141 | return 0; | |
2142 | } | |
2143 | if ((size_t) n >= sizeof(buf)) { | |
2144 | log_error("Got overly long cgroups agent message, ignoring."); | |
2145 | return 0; | |
2146 | } | |
2147 | ||
2148 | if (memchr(buf, 0, n)) { | |
2149 | log_error("Got cgroups agent message with embedded NUL byte, ignoring."); | |
2150 | return 0; | |
2151 | } | |
2152 | buf[n] = 0; | |
2153 | ||
2154 | manager_notify_cgroup_empty(m, buf); | |
d5f15326 | 2155 | (void) bus_forward_agent_released(m, buf); |
d8fdc620 LP |
2156 | |
2157 | return 0; | |
2158 | } | |
2159 | ||
db256aab LP |
2160 | static void manager_invoke_notify_message( |
2161 | Manager *m, | |
2162 | Unit *u, | |
2163 | const struct ucred *ucred, | |
2164 | const char *buf, | |
2165 | FDSet *fds) { | |
2166 | ||
5ba6985b LP |
2167 | assert(m); |
2168 | assert(u); | |
db256aab | 2169 | assert(ucred); |
5ba6985b | 2170 | assert(buf); |
5ba6985b | 2171 | |
62a76913 | 2172 | if (u->notifygen == m->notifygen) /* Already invoked on this same unit in this same iteration? */ |
5ba6985b | 2173 | return; |
62a76913 LP |
2174 | u->notifygen = m->notifygen; |
2175 | ||
2176 | if (UNIT_VTABLE(u)->notify_message) { | |
2177 | _cleanup_strv_free_ char **tags = NULL; | |
2178 | ||
2179 | tags = strv_split(buf, NEWLINE); | |
2180 | if (!tags) { | |
2181 | log_oom(); | |
2182 | return; | |
2183 | } | |
5ba6985b | 2184 | |
db256aab | 2185 | UNIT_VTABLE(u)->notify_message(u, ucred, tags, fds); |
62a76913 LP |
2186 | |
2187 | } else if (DEBUG_LOGGING) { | |
a86b7675 ZJS |
2188 | _cleanup_free_ char *x = NULL, *y = NULL; |
2189 | ||
da5fb861 | 2190 | x = ellipsize(buf, 20, 90); |
a86b7675 | 2191 | if (x) |
da5fb861 LP |
2192 | y = cescape(x); |
2193 | ||
a86b7675 ZJS |
2194 | log_unit_debug(u, "Got notification message \"%s\", ignoring.", strnull(y)); |
2195 | } | |
5ba6985b LP |
2196 | } |
2197 | ||
718db961 | 2198 | static int manager_dispatch_notify_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) { |
3d0b8a55 | 2199 | |
b215b0ed | 2200 | _cleanup_fdset_free_ FDSet *fds = NULL; |
718db961 | 2201 | Manager *m = userdata; |
b215b0ed DH |
2202 | char buf[NOTIFY_BUFFER_MAX+1]; |
2203 | struct iovec iovec = { | |
2204 | .iov_base = buf, | |
2205 | .iov_len = sizeof(buf)-1, | |
2206 | }; | |
2207 | union { | |
2208 | struct cmsghdr cmsghdr; | |
2209 | uint8_t buf[CMSG_SPACE(sizeof(struct ucred)) + | |
2210 | CMSG_SPACE(sizeof(int) * NOTIFY_FD_MAX)]; | |
2211 | } control = {}; | |
2212 | struct msghdr msghdr = { | |
2213 | .msg_iov = &iovec, | |
2214 | .msg_iovlen = 1, | |
2215 | .msg_control = &control, | |
2216 | .msg_controllen = sizeof(control), | |
2217 | }; | |
2218 | ||
2219 | struct cmsghdr *cmsg; | |
2220 | struct ucred *ucred = NULL; | |
62a76913 LP |
2221 | _cleanup_free_ Unit **array_copy = NULL; |
2222 | Unit *u1, *u2, **array; | |
b215b0ed | 2223 | int r, *fd_array = NULL; |
da6053d0 | 2224 | size_t n_fds = 0; |
62a76913 | 2225 | bool found = false; |
8c47c732 LP |
2226 | ssize_t n; |
2227 | ||
2228 | assert(m); | |
718db961 LP |
2229 | assert(m->notify_fd == fd); |
2230 | ||
2231 | if (revents != EPOLLIN) { | |
2232 | log_warning("Got unexpected poll event for notify fd."); | |
2233 | return 0; | |
2234 | } | |
8c47c732 | 2235 | |
045a3d59 | 2236 | n = recvmsg(m->notify_fd, &msghdr, MSG_DONTWAIT|MSG_CMSG_CLOEXEC|MSG_TRUNC); |
b215b0ed | 2237 | if (n < 0) { |
c55ae51e LP |
2238 | if (IN_SET(errno, EAGAIN, EINTR)) |
2239 | return 0; /* Spurious wakeup, try again */ | |
8c47c732 | 2240 | |
c55ae51e LP |
2241 | /* If this is any other, real error, then let's stop processing this socket. This of course means we |
2242 | * won't take notification messages anymore, but that's still better than busy looping around this: | |
2243 | * being woken up over and over again but being unable to actually read the message off the socket. */ | |
2244 | return log_error_errno(errno, "Failed to receive notification message: %m"); | |
b215b0ed | 2245 | } |
a354329f | 2246 | |
b215b0ed DH |
2247 | CMSG_FOREACH(cmsg, &msghdr) { |
2248 | if (cmsg->cmsg_level == SOL_SOCKET && cmsg->cmsg_type == SCM_RIGHTS) { | |
a354329f | 2249 | |
b215b0ed DH |
2250 | fd_array = (int*) CMSG_DATA(cmsg); |
2251 | n_fds = (cmsg->cmsg_len - CMSG_LEN(0)) / sizeof(int); | |
a354329f | 2252 | |
b215b0ed DH |
2253 | } else if (cmsg->cmsg_level == SOL_SOCKET && |
2254 | cmsg->cmsg_type == SCM_CREDENTIALS && | |
2255 | cmsg->cmsg_len == CMSG_LEN(sizeof(struct ucred))) { | |
a354329f | 2256 | |
b215b0ed | 2257 | ucred = (struct ucred*) CMSG_DATA(cmsg); |
a354329f | 2258 | } |
b215b0ed | 2259 | } |
a354329f | 2260 | |
b215b0ed DH |
2261 | if (n_fds > 0) { |
2262 | assert(fd_array); | |
a354329f | 2263 | |
b215b0ed DH |
2264 | r = fdset_new_array(&fds, fd_array, n_fds); |
2265 | if (r < 0) { | |
2266 | close_many(fd_array, n_fds); | |
9987750e FB |
2267 | log_oom(); |
2268 | return 0; | |
a354329f | 2269 | } |
b215b0ed | 2270 | } |
8c47c732 | 2271 | |
15e23e8c | 2272 | if (!ucred || !pid_is_valid(ucred->pid)) { |
b215b0ed DH |
2273 | log_warning("Received notify message without valid credentials. Ignoring."); |
2274 | return 0; | |
2275 | } | |
8c47c732 | 2276 | |
045a3d59 | 2277 | if ((size_t) n >= sizeof(buf) || (msghdr.msg_flags & MSG_TRUNC)) { |
b215b0ed DH |
2278 | log_warning("Received notify message exceeded maximum size. Ignoring."); |
2279 | return 0; | |
2280 | } | |
8c47c732 | 2281 | |
875ca88d LP |
2282 | /* As extra safety check, let's make sure the string we get doesn't contain embedded NUL bytes. We permit one |
2283 | * trailing NUL byte in the message, but don't expect it. */ | |
2284 | if (n > 1 && memchr(buf, 0, n-1)) { | |
2285 | log_warning("Received notify message with embedded NUL bytes. Ignoring."); | |
2286 | return 0; | |
2287 | } | |
2288 | ||
2289 | /* Make sure it's NUL-terminated. */ | |
b215b0ed | 2290 | buf[n] = 0; |
8c47c732 | 2291 | |
62a76913 LP |
2292 | /* Increase the generation counter used for filtering out duplicate unit invocations. */ |
2293 | m->notifygen++; | |
2294 | ||
2295 | /* Notify every unit that might be interested, which might be multiple. */ | |
b215b0ed | 2296 | u1 = manager_get_unit_by_pid_cgroup(m, ucred->pid); |
62a76913 LP |
2297 | u2 = hashmap_get(m->watch_pids, PID_TO_PTR(ucred->pid)); |
2298 | array = hashmap_get(m->watch_pids, PID_TO_PTR(-ucred->pid)); | |
2299 | if (array) { | |
2300 | size_t k = 0; | |
5ba6985b | 2301 | |
62a76913 LP |
2302 | while (array[k]) |
2303 | k++; | |
5ba6985b | 2304 | |
62a76913 LP |
2305 | array_copy = newdup(Unit*, array, k+1); |
2306 | if (!array_copy) | |
2307 | log_oom(); | |
2308 | } | |
2309 | /* And now invoke the per-unit callbacks. Note that manager_invoke_notify_message() will handle duplicate units | |
2310 | * make sure we only invoke each unit's handler once. */ | |
2311 | if (u1) { | |
2312 | manager_invoke_notify_message(m, u1, ucred, buf, fds); | |
2313 | found = true; | |
2314 | } | |
2315 | if (u2) { | |
2316 | manager_invoke_notify_message(m, u2, ucred, buf, fds); | |
2317 | found = true; | |
2318 | } | |
2319 | if (array_copy) | |
2320 | for (size_t i = 0; array_copy[i]; i++) { | |
2321 | manager_invoke_notify_message(m, array_copy[i], ucred, buf, fds); | |
2322 | found = true; | |
2323 | } | |
8c47c732 | 2324 | |
62a76913 LP |
2325 | if (!found) |
2326 | log_warning("Cannot find unit for notify message of PID "PID_FMT", ignoring.", ucred->pid); | |
a354329f | 2327 | |
b215b0ed | 2328 | if (fdset_size(fds) > 0) |
5fd2c135 | 2329 | log_warning("Got extra auxiliary fds with notification message, closing them."); |
8c47c732 LP |
2330 | |
2331 | return 0; | |
2332 | } | |
2333 | ||
62a76913 LP |
2334 | static void manager_invoke_sigchld_event( |
2335 | Manager *m, | |
2336 | Unit *u, | |
2337 | const siginfo_t *si) { | |
36f20ae3 | 2338 | |
5ba6985b LP |
2339 | assert(m); |
2340 | assert(u); | |
2341 | assert(si); | |
2342 | ||
62a76913 LP |
2343 | /* Already invoked the handler of this unit in this iteration? Then don't process this again */ |
2344 | if (u->sigchldgen == m->sigchldgen) | |
2345 | return; | |
2346 | u->sigchldgen = m->sigchldgen; | |
5ba6985b | 2347 | |
62a76913 | 2348 | log_unit_debug(u, "Child "PID_FMT" belongs to %s.", si->si_pid, u->id); |
5ba6985b | 2349 | unit_unwatch_pid(u, si->si_pid); |
e57051f5 | 2350 | |
62a76913 LP |
2351 | if (UNIT_VTABLE(u)->sigchld_event) |
2352 | UNIT_VTABLE(u)->sigchld_event(u, si->si_pid, si->si_code, si->si_status); | |
5ba6985b LP |
2353 | } |
2354 | ||
575b300b LP |
2355 | static int manager_dispatch_sigchld(sd_event_source *source, void *userdata) { |
2356 | Manager *m = userdata; | |
2357 | siginfo_t si = {}; | |
2358 | int r; | |
2359 | ||
2360 | assert(source); | |
9152c765 LP |
2361 | assert(m); |
2362 | ||
575b300b LP |
2363 | /* First we call waitd() for a PID and do not reap the zombie. That way we can still access /proc/$PID for it |
2364 | * while it is a zombie. */ | |
9152c765 | 2365 | |
575b300b | 2366 | if (waitid(P_ALL, 0, &si, WEXITED|WNOHANG|WNOWAIT) < 0) { |
acbb0225 | 2367 | |
8afabc50 AJ |
2368 | if (errno != ECHILD) |
2369 | log_error_errno(errno, "Failed to peek for child with waitid(), ignoring: %m"); | |
acbb0225 | 2370 | |
8afabc50 | 2371 | goto turn_off; |
575b300b | 2372 | } |
4112df16 | 2373 | |
575b300b LP |
2374 | if (si.si_pid <= 0) |
2375 | goto turn_off; | |
2376 | ||
2377 | if (IN_SET(si.si_code, CLD_EXITED, CLD_KILLED, CLD_DUMPED)) { | |
62a76913 | 2378 | _cleanup_free_ Unit **array_copy = NULL; |
575b300b | 2379 | _cleanup_free_ char *name = NULL; |
62a76913 | 2380 | Unit *u1, *u2, **array; |
575b300b LP |
2381 | |
2382 | (void) get_process_comm(si.si_pid, &name); | |
2383 | ||
2384 | log_debug("Child "PID_FMT" (%s) died (code=%s, status=%i/%s)", | |
2385 | si.si_pid, strna(name), | |
2386 | sigchld_code_to_string(si.si_code), | |
2387 | si.si_status, | |
2388 | strna(si.si_code == CLD_EXITED | |
2389 | ? exit_status_to_string(si.si_status, EXIT_STATUS_FULL) | |
2390 | : signal_to_string(si.si_status))); | |
2391 | ||
62a76913 LP |
2392 | /* Increase the generation counter used for filtering out duplicate unit invocations */ |
2393 | m->sigchldgen++; | |
2394 | ||
2395 | /* And now figure out the unit this belongs to, it might be multiple... */ | |
575b300b | 2396 | u1 = manager_get_unit_by_pid_cgroup(m, si.si_pid); |
62a76913 LP |
2397 | u2 = hashmap_get(m->watch_pids, PID_TO_PTR(si.si_pid)); |
2398 | array = hashmap_get(m->watch_pids, PID_TO_PTR(-si.si_pid)); | |
2399 | if (array) { | |
2400 | size_t n = 0; | |
2401 | ||
2402 | /* Cound how many entries the array has */ | |
2403 | while (array[n]) | |
2404 | n++; | |
2405 | ||
2406 | /* Make a copy of the array so that we don't trip up on the array changing beneath us */ | |
2407 | array_copy = newdup(Unit*, array, n+1); | |
2408 | if (!array_copy) | |
2409 | log_oom(); | |
2410 | } | |
2411 | ||
2412 | /* Finally, execute them all. Note that u1, u2 and the array might contain duplicates, but | |
2413 | * that's fine, manager_invoke_sigchld_event() will ensure we only invoke the handlers once for | |
2414 | * each iteration. */ | |
575b300b | 2415 | if (u1) |
62a76913 LP |
2416 | manager_invoke_sigchld_event(m, u1, &si); |
2417 | if (u2) | |
2418 | manager_invoke_sigchld_event(m, u2, &si); | |
2419 | if (array_copy) | |
2420 | for (size_t i = 0; array_copy[i]; i++) | |
2421 | manager_invoke_sigchld_event(m, array_copy[i], &si); | |
575b300b | 2422 | } |
9152c765 | 2423 | |
575b300b LP |
2424 | /* And now, we actually reap the zombie. */ |
2425 | if (waitid(P_PID, si.si_pid, &si, WEXITED) < 0) { | |
2426 | log_error_errno(errno, "Failed to dequeue child, ignoring: %m"); | |
2427 | return 0; | |
2428 | } | |
9152c765 | 2429 | |
575b300b | 2430 | return 0; |
8c47c732 | 2431 | |
575b300b LP |
2432 | turn_off: |
2433 | /* All children processed for now, turn off event source */ | |
4112df16 | 2434 | |
575b300b LP |
2435 | r = sd_event_source_set_enabled(m->sigchld_event_source, SD_EVENT_OFF); |
2436 | if (r < 0) | |
2437 | return log_error_errno(r, "Failed to disable SIGCHLD event source: %m"); | |
9152c765 LP |
2438 | |
2439 | return 0; | |
2440 | } | |
2441 | ||
c75fbada | 2442 | static void manager_start_target(Manager *m, const char *name, JobMode mode) { |
4afd3348 | 2443 | _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; |
28247076 | 2444 | int r; |
398ef8ba | 2445 | |
f2341e0a | 2446 | log_debug("Activating special unit %s", name); |
1e001f52 | 2447 | |
4bd29fe5 | 2448 | r = manager_add_job_by_name(m, JOB_START, name, mode, &error, NULL); |
bd0af849 | 2449 | if (r < 0) |
f2341e0a | 2450 | log_error("Failed to enqueue %s job: %s", name, bus_error_message(&error, r)); |
28247076 LP |
2451 | } |
2452 | ||
24dd31c1 LN |
2453 | static void manager_handle_ctrl_alt_del(Manager *m) { |
2454 | /* If the user presses C-A-D more than | |
2455 | * 7 times within 2s, we reboot/shutdown immediately, | |
2456 | * unless it was disabled in system.conf */ | |
2457 | ||
7994ac1d | 2458 | if (ratelimit_below(&m->ctrl_alt_del_ratelimit) || m->cad_burst_action == EMERGENCY_ACTION_NONE) |
24dd31c1 | 2459 | manager_start_target(m, SPECIAL_CTRL_ALT_DEL_TARGET, JOB_REPLACE_IRREVERSIBLY); |
ae8c7939 LN |
2460 | else |
2461 | emergency_action(m, m->cad_burst_action, NULL, | |
2462 | "Ctrl-Alt-Del was pressed more than 7 times within 2s"); | |
24dd31c1 LN |
2463 | } |
2464 | ||
718db961 LP |
2465 | static int manager_dispatch_signal_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) { |
2466 | Manager *m = userdata; | |
9152c765 LP |
2467 | ssize_t n; |
2468 | struct signalfd_siginfo sfsi; | |
dacd6cee | 2469 | int r; |
9152c765 LP |
2470 | |
2471 | assert(m); | |
718db961 LP |
2472 | assert(m->signal_fd == fd); |
2473 | ||
2474 | if (revents != EPOLLIN) { | |
2475 | log_warning("Got unexpected events from signal file descriptor."); | |
2476 | return 0; | |
2477 | } | |
9152c765 | 2478 | |
575b300b LP |
2479 | n = read(m->signal_fd, &sfsi, sizeof(sfsi)); |
2480 | if (n != sizeof(sfsi)) { | |
2481 | if (n >= 0) { | |
2482 | log_warning("Truncated read from signal fd (%zu bytes), ignoring!", n); | |
2483 | return 0; | |
2484 | } | |
9152c765 | 2485 | |
575b300b LP |
2486 | if (IN_SET(errno, EINTR, EAGAIN)) |
2487 | return 0; | |
9152c765 | 2488 | |
575b300b LP |
2489 | /* We return an error here, which will kill this handler, |
2490 | * to avoid a busy loop on read error. */ | |
2491 | return log_error_errno(errno, "Reading from signal fd failed: %m"); | |
2492 | } | |
9152c765 | 2493 | |
575b300b LP |
2494 | log_received_signal(sfsi.ssi_signo == SIGCHLD || |
2495 | (sfsi.ssi_signo == SIGTERM && MANAGER_IS_USER(m)) | |
2496 | ? LOG_DEBUG : LOG_INFO, | |
2497 | &sfsi); | |
1e001f52 | 2498 | |
575b300b | 2499 | switch (sfsi.ssi_signo) { |
b9cd2ec1 | 2500 | |
575b300b LP |
2501 | case SIGCHLD: |
2502 | r = sd_event_source_set_enabled(m->sigchld_event_source, SD_EVENT_ON); | |
2503 | if (r < 0) | |
8afabc50 | 2504 | log_warning_errno(r, "Failed to enable SIGCHLD event source, ignoring: %m"); |
b9cd2ec1 | 2505 | |
575b300b | 2506 | break; |
84e9af1e | 2507 | |
575b300b LP |
2508 | case SIGTERM: |
2509 | if (MANAGER_IS_SYSTEM(m)) { | |
ba0c7754 | 2510 | /* This is for compatibility with the original sysvinit */ |
575b300b LP |
2511 | r = verify_run_space_and_log("Refusing to reexecute"); |
2512 | if (r >= 0) | |
2513 | m->exit_code = MANAGER_REEXECUTE; | |
a1b256b0 | 2514 | break; |
575b300b | 2515 | } |
84e9af1e | 2516 | |
575b300b LP |
2517 | _fallthrough_; |
2518 | case SIGINT: | |
2519 | if (MANAGER_IS_SYSTEM(m)) | |
2520 | manager_handle_ctrl_alt_del(m); | |
2521 | else | |
2522 | manager_start_target(m, SPECIAL_EXIT_TARGET, | |
2523 | JOB_REPLACE_IRREVERSIBLY); | |
2524 | break; | |
84e9af1e | 2525 | |
575b300b | 2526 | case SIGWINCH: |
ba0c7754 | 2527 | /* This is a nop on non-init */ |
575b300b LP |
2528 | if (MANAGER_IS_SYSTEM(m)) |
2529 | manager_start_target(m, SPECIAL_KBREQUEST_TARGET, JOB_REPLACE); | |
84e9af1e | 2530 | |
575b300b | 2531 | break; |
84e9af1e | 2532 | |
575b300b | 2533 | case SIGPWR: |
ba0c7754 | 2534 | /* This is a nop on non-init */ |
575b300b LP |
2535 | if (MANAGER_IS_SYSTEM(m)) |
2536 | manager_start_target(m, SPECIAL_SIGPWR_TARGET, JOB_REPLACE); | |
6632c602 | 2537 | |
575b300b | 2538 | break; |
57ee42ce | 2539 | |
8559b3b7 | 2540 | case SIGUSR1: |
8559b3b7 | 2541 | if (manager_dbus_is_running(m, false)) { |
575b300b | 2542 | log_info("Trying to reconnect to bus..."); |
575b300b | 2543 | |
8559b3b7 LP |
2544 | (void) bus_init_api(m); |
2545 | ||
2546 | if (MANAGER_IS_SYSTEM(m)) | |
2547 | (void) bus_init_system(m); | |
2548 | } else { | |
2549 | log_info("Starting D-Bus service..."); | |
575b300b LP |
2550 | manager_start_target(m, SPECIAL_DBUS_SERVICE, JOB_REPLACE); |
2551 | } | |
57ee42ce | 2552 | |
575b300b | 2553 | break; |
575b300b LP |
2554 | |
2555 | case SIGUSR2: { | |
2556 | _cleanup_free_ char *dump = NULL; | |
2557 | ||
2558 | r = manager_get_dump_string(m, &dump); | |
2559 | if (r < 0) { | |
2560 | log_warning_errno(errno, "Failed to acquire manager dump: %m"); | |
57ee42ce LP |
2561 | break; |
2562 | } | |
2563 | ||
575b300b LP |
2564 | log_dump(LOG_INFO, dump); |
2565 | break; | |
2566 | } | |
2149e37c | 2567 | |
575b300b LP |
2568 | case SIGHUP: |
2569 | r = verify_run_space_and_log("Refusing to reload"); | |
2570 | if (r >= 0) | |
2571 | m->exit_code = MANAGER_RELOAD; | |
2572 | break; | |
2573 | ||
2574 | default: { | |
2575 | ||
2576 | /* Starting SIGRTMIN+0 */ | |
2577 | static const struct { | |
2578 | const char *target; | |
2579 | JobMode mode; | |
2580 | } target_table[] = { | |
2581 | [0] = { SPECIAL_DEFAULT_TARGET, JOB_ISOLATE }, | |
2582 | [1] = { SPECIAL_RESCUE_TARGET, JOB_ISOLATE }, | |
2583 | [2] = { SPECIAL_EMERGENCY_TARGET, JOB_ISOLATE }, | |
2584 | [3] = { SPECIAL_HALT_TARGET, JOB_REPLACE_IRREVERSIBLY }, | |
2585 | [4] = { SPECIAL_POWEROFF_TARGET, JOB_REPLACE_IRREVERSIBLY }, | |
2586 | [5] = { SPECIAL_REBOOT_TARGET, JOB_REPLACE_IRREVERSIBLY }, | |
2587 | [6] = { SPECIAL_KEXEC_TARGET, JOB_REPLACE_IRREVERSIBLY }, | |
2588 | }; | |
2589 | ||
2590 | /* Starting SIGRTMIN+13, so that target halt and system halt are 10 apart */ | |
2591 | static const ManagerExitCode code_table[] = { | |
2592 | [0] = MANAGER_HALT, | |
2593 | [1] = MANAGER_POWEROFF, | |
2594 | [2] = MANAGER_REBOOT, | |
2595 | [3] = MANAGER_KEXEC, | |
2596 | }; | |
b2cdc666 | 2597 | |
575b300b LP |
2598 | if ((int) sfsi.ssi_signo >= SIGRTMIN+0 && |
2599 | (int) sfsi.ssi_signo < SIGRTMIN+(int) ELEMENTSOF(target_table)) { | |
2600 | int idx = (int) sfsi.ssi_signo - SIGRTMIN; | |
2601 | manager_start_target(m, target_table[idx].target, | |
2602 | target_table[idx].mode); | |
1005d14f | 2603 | break; |
2149e37c | 2604 | } |
1005d14f | 2605 | |
575b300b LP |
2606 | if ((int) sfsi.ssi_signo >= SIGRTMIN+13 && |
2607 | (int) sfsi.ssi_signo < SIGRTMIN+13+(int) ELEMENTSOF(code_table)) { | |
2608 | m->exit_code = code_table[sfsi.ssi_signo - SIGRTMIN - 13]; | |
2609 | break; | |
2610 | } | |
2611 | ||
2612 | switch (sfsi.ssi_signo - SIGRTMIN) { | |
2613 | ||
2614 | case 20: | |
2615 | manager_set_show_status(m, SHOW_STATUS_YES); | |
a16e1123 LP |
2616 | break; |
2617 | ||
575b300b LP |
2618 | case 21: |
2619 | manager_set_show_status(m, SHOW_STATUS_NO); | |
2620 | break; | |
7d793605 | 2621 | |
575b300b | 2622 | case 22: |
a6ecbf83 | 2623 | manager_override_log_level(m, LOG_DEBUG); |
575b300b LP |
2624 | break; |
2625 | ||
2626 | case 23: | |
a6ecbf83 | 2627 | manager_restore_original_log_level(m); |
575b300b | 2628 | break; |
0003d1ab | 2629 | |
575b300b LP |
2630 | case 24: |
2631 | if (MANAGER_IS_USER(m)) { | |
2632 | m->exit_code = MANAGER_EXIT; | |
2633 | return 0; | |
0658666b | 2634 | } |
9152c765 | 2635 | |
575b300b LP |
2636 | /* This is a nop on init */ |
2637 | break; | |
2638 | ||
2639 | case 26: | |
2640 | case 29: /* compatibility: used to be mapped to LOG_TARGET_SYSLOG_OR_KMSG */ | |
bda7d78b | 2641 | manager_restore_original_log_target(m); |
575b300b LP |
2642 | break; |
2643 | ||
2644 | case 27: | |
bda7d78b | 2645 | manager_override_log_target(m, LOG_TARGET_CONSOLE); |
575b300b LP |
2646 | break; |
2647 | ||
2648 | case 28: | |
bda7d78b | 2649 | manager_override_log_target(m, LOG_TARGET_KMSG); |
575b300b LP |
2650 | break; |
2651 | ||
2652 | default: | |
2653 | log_warning("Got unhandled signal <%s>.", signal_to_string(sfsi.ssi_signo)); | |
2654 | } | |
2655 | }} | |
034c6ed7 LP |
2656 | |
2657 | return 0; | |
2658 | } | |
2659 | ||
718db961 LP |
2660 | static int manager_dispatch_time_change_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) { |
2661 | Manager *m = userdata; | |
2662 | Iterator i; | |
2663 | Unit *u; | |
034c6ed7 LP |
2664 | |
2665 | assert(m); | |
718db961 | 2666 | assert(m->time_change_fd == fd); |
034c6ed7 | 2667 | |
a80c1575 | 2668 | log_struct(LOG_DEBUG, |
2b044526 | 2669 | "MESSAGE_ID=" SD_MESSAGE_TIME_CHANGE_STR, |
a1230ff9 | 2670 | LOG_MESSAGE("Time has been changed")); |
034c6ed7 | 2671 | |
718db961 | 2672 | /* Restart the watch */ |
7feedd18 | 2673 | (void) manager_setup_time_change(m); |
4e434314 | 2674 | |
718db961 LP |
2675 | HASHMAP_FOREACH(u, m->units, i) |
2676 | if (UNIT_VTABLE(u)->time_change) | |
2677 | UNIT_VTABLE(u)->time_change(u); | |
ea430986 | 2678 | |
718db961 LP |
2679 | return 0; |
2680 | } | |
ea430986 | 2681 | |
bbf5fd8e LP |
2682 | static int manager_dispatch_timezone_change( |
2683 | sd_event_source *source, | |
2684 | const struct inotify_event *e, | |
2685 | void *userdata) { | |
2686 | ||
2687 | Manager *m = userdata; | |
2688 | int changed; | |
2689 | Iterator i; | |
2690 | Unit *u; | |
2691 | ||
2692 | assert(m); | |
2693 | ||
2694 | log_debug("inotify event for /etc/localtime"); | |
2695 | ||
2696 | changed = manager_read_timezone_stat(m); | |
2697 | if (changed < 0) | |
2698 | return changed; | |
2699 | if (!changed) | |
2700 | return 0; | |
2701 | ||
2702 | /* Something changed, restart the watch, to ensure we watch the new /etc/localtime if it changed */ | |
2703 | (void) manager_setup_timezone_change(m); | |
2704 | ||
2705 | /* Read the new timezone */ | |
2706 | tzset(); | |
2707 | ||
2708 | log_debug("Timezone has been changed (now: %s).", tzname[daylight]); | |
2709 | ||
2710 | HASHMAP_FOREACH(u, m->units, i) | |
2711 | if (UNIT_VTABLE(u)->timezone_change) | |
2712 | UNIT_VTABLE(u)->timezone_change(u); | |
2713 | ||
2714 | return 0; | |
2715 | } | |
2716 | ||
718db961 LP |
2717 | static int manager_dispatch_idle_pipe_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) { |
2718 | Manager *m = userdata; | |
8742514c | 2719 | |
718db961 LP |
2720 | assert(m); |
2721 | assert(m->idle_pipe[2] == fd); | |
8742514c | 2722 | |
5a69973f LP |
2723 | /* There's at least one Type=idle child that just gave up on us waiting for the boot process to complete. Let's |
2724 | * now turn off any further console output if there's at least one service that needs console access, so that | |
2725 | * from now on our own output should not spill into that service's output anymore. After all, we support | |
2726 | * Type=idle only to beautify console output and it generally is set on services that want to own the console | |
2727 | * exclusively without our interference. */ | |
718db961 | 2728 | m->no_console_output = m->n_on_console > 0; |
03b717a3 | 2729 | |
5a69973f LP |
2730 | /* Acknowledge the child's request, and let all all other children know too that they shouldn't wait any longer |
2731 | * by closing the pipes towards them, which is what they are waiting for. */ | |
718db961 | 2732 | manager_close_idle_pipe(m); |
03b717a3 | 2733 | |
718db961 LP |
2734 | return 0; |
2735 | } | |
31a7eb86 | 2736 | |
718db961 LP |
2737 | static int manager_dispatch_jobs_in_progress(sd_event_source *source, usec_t usec, void *userdata) { |
2738 | Manager *m = userdata; | |
fd08a840 ZJS |
2739 | int r; |
2740 | uint64_t next; | |
31a7eb86 | 2741 | |
718db961 | 2742 | assert(m); |
fd08a840 | 2743 | assert(source); |
9152c765 | 2744 | |
718db961 | 2745 | manager_print_jobs_in_progress(m); |
fd08a840 ZJS |
2746 | |
2747 | next = now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_PERIOD_USEC; | |
2748 | r = sd_event_source_set_time(source, next); | |
2749 | if (r < 0) | |
2750 | return r; | |
2751 | ||
2752 | return sd_event_source_set_enabled(source, SD_EVENT_ONESHOT); | |
9152c765 LP |
2753 | } |
2754 | ||
2755 | int manager_loop(Manager *m) { | |
2756 | int r; | |
9152c765 | 2757 | |
fac9f8df | 2758 | RATELIMIT_DEFINE(rl, 1*USEC_PER_SEC, 50000); |
ea430986 | 2759 | |
9152c765 | 2760 | assert(m); |
f755e3b7 | 2761 | m->exit_code = MANAGER_OK; |
9152c765 | 2762 | |
fe51822e | 2763 | /* Release the path cache */ |
97044145 | 2764 | m->unit_path_cache = set_free_free(m->unit_path_cache); |
fe51822e | 2765 | |
b0c918b9 LP |
2766 | manager_check_finished(m); |
2767 | ||
575b300b LP |
2768 | /* There might still be some zombies hanging around from before we were exec()'ed. Let's reap them. */ |
2769 | r = sd_event_source_set_enabled(m->sigchld_event_source, SD_EVENT_ON); | |
e96d6be7 | 2770 | if (r < 0) |
575b300b | 2771 | return log_error_errno(r, "Failed to enable SIGCHLD event source: %m"); |
a4312405 | 2772 | |
f755e3b7 | 2773 | while (m->exit_code == MANAGER_OK) { |
718db961 | 2774 | usec_t wait_usec; |
9152c765 | 2775 | |
463d0d15 | 2776 | if (m->runtime_watchdog > 0 && m->runtime_watchdog != USEC_INFINITY && MANAGER_IS_SYSTEM(m)) |
e96d6be7 LP |
2777 | watchdog_ping(); |
2778 | ||
7994ac1d | 2779 | if (!ratelimit_below(&rl)) { |
ea430986 LP |
2780 | /* Yay, something is going seriously wrong, pause a little */ |
2781 | log_warning("Looping too fast. Throttling execution a little."); | |
2782 | sleep(1); | |
2783 | } | |
2784 | ||
37a8e683 | 2785 | if (manager_dispatch_load_queue(m) > 0) |
23a177ef LP |
2786 | continue; |
2787 | ||
c5a97ed1 LP |
2788 | if (manager_dispatch_gc_job_queue(m) > 0) |
2789 | continue; | |
2790 | ||
2791 | if (manager_dispatch_gc_unit_queue(m) > 0) | |
701cc384 LP |
2792 | continue; |
2793 | ||
cf1265e1 | 2794 | if (manager_dispatch_cleanup_queue(m) > 0) |
c1e1601e | 2795 | continue; |
034c6ed7 | 2796 | |
91a6073e | 2797 | if (manager_dispatch_cgroup_realize_queue(m) > 0) |
c1e1601e LP |
2798 | continue; |
2799 | ||
c1e1601e | 2800 | if (manager_dispatch_dbus_queue(m) > 0) |
ea430986 | 2801 | continue; |
ea430986 | 2802 | |
c757a65b | 2803 | /* Sleep for half the watchdog time */ |
463d0d15 | 2804 | if (m->runtime_watchdog > 0 && m->runtime_watchdog != USEC_INFINITY && MANAGER_IS_SYSTEM(m)) { |
718db961 LP |
2805 | wait_usec = m->runtime_watchdog / 2; |
2806 | if (wait_usec <= 0) | |
2807 | wait_usec = 1; | |
c757a65b | 2808 | } else |
3a43da28 | 2809 | wait_usec = USEC_INFINITY; |
9152c765 | 2810 | |
718db961 | 2811 | r = sd_event_run(m->event, wait_usec); |
23bbb0de MS |
2812 | if (r < 0) |
2813 | return log_error_errno(r, "Failed to run event loop: %m"); | |
a16e1123 | 2814 | } |
957ca890 | 2815 | |
a16e1123 | 2816 | return m->exit_code; |
83c60c9f | 2817 | } |
ea430986 | 2818 | |
718db961 | 2819 | int manager_load_unit_from_dbus_path(Manager *m, const char *s, sd_bus_error *e, Unit **_u) { |
ede3a796 | 2820 | _cleanup_free_ char *n = NULL; |
4b58153d | 2821 | sd_id128_t invocation_id; |
ea430986 | 2822 | Unit *u; |
80fbf05e | 2823 | int r; |
ea430986 LP |
2824 | |
2825 | assert(m); | |
2826 | assert(s); | |
2827 | assert(_u); | |
2828 | ||
ede3a796 LP |
2829 | r = unit_name_from_dbus_path(s, &n); |
2830 | if (r < 0) | |
2831 | return r; | |
ea430986 | 2832 | |
4b58153d LP |
2833 | /* Permit addressing units by invocation ID: if the passed bus path is suffixed by a 128bit ID then we use it |
2834 | * as invocation ID. */ | |
2835 | r = sd_id128_from_string(n, &invocation_id); | |
2836 | if (r >= 0) { | |
2837 | u = hashmap_get(m->units_by_invocation_id, &invocation_id); | |
2838 | if (u) { | |
2839 | *_u = u; | |
2840 | return 0; | |
2841 | } | |
2842 | ||
930c124c ZJS |
2843 | return sd_bus_error_setf(e, BUS_ERROR_NO_UNIT_FOR_INVOCATION_ID, |
2844 | "No unit with the specified invocation ID " SD_ID128_FORMAT_STR " known.", | |
2845 | SD_ID128_FORMAT_VAL(invocation_id)); | |
4b58153d LP |
2846 | } |
2847 | ||
00c83b43 | 2848 | /* If this didn't work, we check if this is a unit name */ |
930c124c ZJS |
2849 | if (!unit_name_is_valid(n, UNIT_NAME_PLAIN|UNIT_NAME_INSTANCE)) { |
2850 | _cleanup_free_ char *nn = NULL; | |
2851 | ||
2852 | nn = cescape(n); | |
2853 | return sd_bus_error_setf(e, SD_BUS_ERROR_INVALID_ARGS, | |
2854 | "Unit name %s is neither a valid invocation ID nor unit name.", strnull(nn)); | |
2855 | } | |
00c83b43 | 2856 | |
80fbf05e | 2857 | r = manager_load_unit(m, n, NULL, e, &u); |
80fbf05e MS |
2858 | if (r < 0) |
2859 | return r; | |
ea430986 LP |
2860 | |
2861 | *_u = u; | |
ea430986 LP |
2862 | return 0; |
2863 | } | |
86fbf370 LP |
2864 | |
2865 | int manager_get_job_from_dbus_path(Manager *m, const char *s, Job **_j) { | |
718db961 | 2866 | const char *p; |
86fbf370 | 2867 | unsigned id; |
718db961 | 2868 | Job *j; |
86fbf370 LP |
2869 | int r; |
2870 | ||
2871 | assert(m); | |
2872 | assert(s); | |
2873 | assert(_j); | |
2874 | ||
718db961 LP |
2875 | p = startswith(s, "/org/freedesktop/systemd1/job/"); |
2876 | if (!p) | |
86fbf370 LP |
2877 | return -EINVAL; |
2878 | ||
718db961 | 2879 | r = safe_atou(p, &id); |
8742514c | 2880 | if (r < 0) |
86fbf370 LP |
2881 | return r; |
2882 | ||
8742514c LP |
2883 | j = manager_get_job(m, id); |
2884 | if (!j) | |
86fbf370 LP |
2885 | return -ENOENT; |
2886 | ||
2887 | *_j = j; | |
2888 | ||
2889 | return 0; | |
2890 | } | |
dfcd764e | 2891 | |
4927fcae | 2892 | void manager_send_unit_audit(Manager *m, Unit *u, int type, bool success) { |
e537352b | 2893 | |
349cc4a5 | 2894 | #if HAVE_AUDIT |
2ba11090 | 2895 | _cleanup_free_ char *p = NULL; |
0aa281df | 2896 | const char *msg; |
7410616c | 2897 | int audit_fd, r; |
e537352b | 2898 | |
463d0d15 | 2899 | if (!MANAGER_IS_SYSTEM(m)) |
a1a078ee LP |
2900 | return; |
2901 | ||
c1165f82 LP |
2902 | audit_fd = get_audit_fd(); |
2903 | if (audit_fd < 0) | |
e537352b LP |
2904 | return; |
2905 | ||
bbd3a7ba LP |
2906 | /* Don't generate audit events if the service was already |
2907 | * started and we're just deserializing */ | |
2c289ea8 | 2908 | if (MANAGER_IS_RELOADING(m)) |
bbd3a7ba LP |
2909 | return; |
2910 | ||
ac155bb8 | 2911 | if (u->type != UNIT_SERVICE) |
f1dd0c3f LP |
2912 | return; |
2913 | ||
7410616c LP |
2914 | r = unit_name_to_prefix_and_instance(u->id, &p); |
2915 | if (r < 0) { | |
2916 | log_error_errno(r, "Failed to extract prefix and instance of unit name: %m"); | |
e537352b LP |
2917 | return; |
2918 | } | |
2919 | ||
63c372cb | 2920 | msg = strjoina("unit=", p); |
0aa281df LP |
2921 | if (audit_log_user_comm_message(audit_fd, type, msg, "systemd", NULL, NULL, NULL, success) < 0) { |
2922 | if (errno == EPERM) | |
391ade86 | 2923 | /* We aren't allowed to send audit messages? |
44785992 | 2924 | * Then let's not retry again. */ |
c1165f82 | 2925 | close_audit_fd(); |
0aa281df | 2926 | else |
56f64d95 | 2927 | log_warning_errno(errno, "Failed to send audit message: %m"); |
391ade86 | 2928 | } |
4927fcae | 2929 | #endif |
e537352b | 2930 | |
e537352b LP |
2931 | } |
2932 | ||
e983b760 | 2933 | void manager_send_unit_plymouth(Manager *m, Unit *u) { |
fc2fffe7 | 2934 | static const union sockaddr_union sa = PLYMOUTH_SOCKET; |
2ba11090 ZJS |
2935 | _cleanup_free_ char *message = NULL; |
2936 | _cleanup_close_ int fd = -1; | |
fc2fffe7 | 2937 | int n = 0; |
e983b760 LP |
2938 | |
2939 | /* Don't generate plymouth events if the service was already | |
2940 | * started and we're just deserializing */ | |
2c289ea8 | 2941 | if (MANAGER_IS_RELOADING(m)) |
e983b760 LP |
2942 | return; |
2943 | ||
463d0d15 | 2944 | if (!MANAGER_IS_SYSTEM(m)) |
e983b760 LP |
2945 | return; |
2946 | ||
75f86906 | 2947 | if (detect_container() > 0) |
3772995a LP |
2948 | return; |
2949 | ||
ec2ce0c5 | 2950 | if (!IN_SET(u->type, UNIT_SERVICE, UNIT_MOUNT, UNIT_SWAP)) |
e983b760 LP |
2951 | return; |
2952 | ||
2953 | /* We set SOCK_NONBLOCK here so that we rather drop the | |
2954 | * message then wait for plymouth */ | |
e62d8c39 ZJS |
2955 | fd = socket(AF_UNIX, SOCK_STREAM|SOCK_CLOEXEC|SOCK_NONBLOCK, 0); |
2956 | if (fd < 0) { | |
56f64d95 | 2957 | log_error_errno(errno, "socket() failed: %m"); |
e983b760 LP |
2958 | return; |
2959 | } | |
2960 | ||
fc2fffe7 | 2961 | if (connect(fd, &sa.sa, SOCKADDR_UN_LEN(sa.un)) < 0) { |
2ba11090 | 2962 | if (!IN_SET(errno, EPIPE, EAGAIN, ENOENT, ECONNREFUSED, ECONNRESET, ECONNABORTED)) |
56f64d95 | 2963 | log_error_errno(errno, "connect() failed: %m"); |
2ba11090 | 2964 | return; |
e983b760 LP |
2965 | } |
2966 | ||
ac155bb8 | 2967 | if (asprintf(&message, "U\002%c%s%n", (int) (strlen(u->id) + 1), u->id, &n) < 0) { |
0d0f0c50 | 2968 | log_oom(); |
2ba11090 | 2969 | return; |
e983b760 LP |
2970 | } |
2971 | ||
2972 | errno = 0; | |
2ba11090 ZJS |
2973 | if (write(fd, message, n + 1) != n + 1) |
2974 | if (!IN_SET(errno, EPIPE, EAGAIN, ENOENT, ECONNREFUSED, ECONNRESET, ECONNABORTED)) | |
56f64d95 | 2975 | log_error_errno(errno, "Failed to write Plymouth message: %m"); |
e983b760 LP |
2976 | } |
2977 | ||
d8d5ab98 | 2978 | int manager_open_serialization(Manager *m, FILE **_f) { |
504afd7c | 2979 | int fd; |
a16e1123 LP |
2980 | FILE *f; |
2981 | ||
2982 | assert(_f); | |
2983 | ||
504afd7c ZJS |
2984 | fd = open_serialization_fd("systemd-state"); |
2985 | if (fd < 0) | |
2986 | return fd; | |
a16e1123 | 2987 | |
01e10de3 | 2988 | f = fdopen(fd, "w+"); |
d86f9d52 | 2989 | if (!f) { |
03e334a1 | 2990 | safe_close(fd); |
a16e1123 | 2991 | return -errno; |
d86f9d52 | 2992 | } |
a16e1123 LP |
2993 | |
2994 | *_f = f; | |
a16e1123 LP |
2995 | return 0; |
2996 | } | |
2997 | ||
b3680f49 | 2998 | int manager_serialize(Manager *m, FILE *f, FDSet *fds, bool switching_root) { |
9f9f0342 LP |
2999 | ManagerTimestamp q; |
3000 | const char *t; | |
a16e1123 LP |
3001 | Iterator i; |
3002 | Unit *u; | |
a16e1123 LP |
3003 | int r; |
3004 | ||
3005 | assert(m); | |
3006 | assert(f); | |
3007 | assert(fds); | |
3008 | ||
313cefa1 | 3009 | m->n_reloading++; |
38c52d46 | 3010 | |
1fa2f38f | 3011 | fprintf(f, "current-job-id=%"PRIu32"\n", m->current_job_id); |
33c5fae9 LP |
3012 | fprintf(f, "n-installed-jobs=%u\n", m->n_installed_jobs); |
3013 | fprintf(f, "n-failed-jobs=%u\n", m->n_failed_jobs); | |
0c2826c6 ZJS |
3014 | fprintf(f, "taint-usr=%s\n", yes_no(m->taint_usr)); |
3015 | fprintf(f, "ready-sent=%s\n", yes_no(m->ready_sent)); | |
d8eb10d6 | 3016 | fprintf(f, "taint-logged=%s\n", yes_no(m->taint_logged)); |
2a12e32e | 3017 | fprintf(f, "service-watchdogs=%s\n", yes_no(m->service_watchdogs)); |
01d67b43 | 3018 | |
bee38b5c YW |
3019 | t = show_status_to_string(m->show_status); |
3020 | if (t) | |
3021 | fprintf(f, "show-status=%s\n", t); | |
3022 | ||
a6ecbf83 FB |
3023 | if (m->log_level_overridden) |
3024 | fprintf(f, "log-level-override=%i\n", log_get_max_level()); | |
bda7d78b FB |
3025 | if (m->log_target_overridden) |
3026 | fprintf(f, "log-target-override=%s\n", log_target_to_string(log_get_target())); | |
a6ecbf83 | 3027 | |
9f9f0342 | 3028 | for (q = 0; q < _MANAGER_TIMESTAMP_MAX; q++) { |
d4ee7bd8 YW |
3029 | /* The following timestamps only apply to the host system, hence only serialize them there */ |
3030 | if (in_initrd() && | |
3031 | IN_SET(q, MANAGER_TIMESTAMP_USERSPACE, MANAGER_TIMESTAMP_FINISH, | |
3032 | MANAGER_TIMESTAMP_SECURITY_START, MANAGER_TIMESTAMP_SECURITY_FINISH, | |
3033 | MANAGER_TIMESTAMP_GENERATORS_START, MANAGER_TIMESTAMP_GENERATORS_FINISH, | |
3034 | MANAGER_TIMESTAMP_UNITS_LOAD_START, MANAGER_TIMESTAMP_UNITS_LOAD_FINISH)) | |
9f9f0342 | 3035 | continue; |
f38ed060 | 3036 | |
9f9f0342 LP |
3037 | t = manager_timestamp_to_string(q); |
3038 | { | |
fbd0b64f | 3039 | char field[strlen(t) + STRLEN("-timestamp") + 1]; |
9f9f0342 LP |
3040 | strcpy(stpcpy(field, t), "-timestamp"); |
3041 | dual_timestamp_serialize(f, field, m->timestamps + q); | |
3042 | } | |
f38ed060 | 3043 | } |
47a483a1 | 3044 | |
fe902fa4 ZJS |
3045 | if (!switching_root) |
3046 | (void) serialize_environment(f, m->environment); | |
4a9fd066 | 3047 | |
d86f9d52 LP |
3048 | if (m->notify_fd >= 0) { |
3049 | int copy; | |
3050 | ||
3051 | copy = fdset_put_dup(fds, m->notify_fd); | |
3052 | if (copy < 0) | |
3053 | return copy; | |
3054 | ||
3055 | fprintf(f, "notify-fd=%i\n", copy); | |
3056 | fprintf(f, "notify-socket=%s\n", m->notify_socket); | |
3057 | } | |
3058 | ||
d8fdc620 LP |
3059 | if (m->cgroups_agent_fd >= 0) { |
3060 | int copy; | |
3061 | ||
3062 | copy = fdset_put_dup(fds, m->cgroups_agent_fd); | |
3063 | if (copy < 0) | |
3064 | return copy; | |
3065 | ||
3066 | fprintf(f, "cgroups-agent-fd=%i\n", copy); | |
3067 | } | |
3068 | ||
00d9ef85 LP |
3069 | if (m->user_lookup_fds[0] >= 0) { |
3070 | int copy0, copy1; | |
3071 | ||
3072 | copy0 = fdset_put_dup(fds, m->user_lookup_fds[0]); | |
3073 | if (copy0 < 0) | |
3074 | return copy0; | |
3075 | ||
3076 | copy1 = fdset_put_dup(fds, m->user_lookup_fds[1]); | |
3077 | if (copy1 < 0) | |
3078 | return copy1; | |
3079 | ||
3080 | fprintf(f, "user-lookup=%i %i\n", copy0, copy1); | |
3081 | } | |
3082 | ||
05a98afd | 3083 | bus_track_serialize(m->subscribed, f, "subscribed"); |
6fa48533 | 3084 | |
29206d46 LP |
3085 | r = dynamic_user_serialize(m, f, fds); |
3086 | if (r < 0) | |
3087 | return r; | |
3088 | ||
00d9ef85 LP |
3089 | manager_serialize_uid_refs(m, f); |
3090 | manager_serialize_gid_refs(m, f); | |
3091 | ||
e8a565cb YW |
3092 | r = exec_runtime_serialize(m, f, fds); |
3093 | if (r < 0) | |
3094 | return r; | |
3095 | ||
0d536673 | 3096 | (void) fputc('\n', f); |
f2382a94 | 3097 | |
a16e1123 | 3098 | HASHMAP_FOREACH_KEY(u, t, m->units, i) { |
ac155bb8 | 3099 | if (u->id != t) |
a16e1123 LP |
3100 | continue; |
3101 | ||
a16e1123 | 3102 | /* Start marker */ |
0d536673 LP |
3103 | fputs(u->id, f); |
3104 | fputc('\n', f); | |
a16e1123 | 3105 | |
6fa48533 LP |
3106 | r = unit_serialize(u, f, fds, !switching_root); |
3107 | if (r < 0) { | |
313cefa1 | 3108 | m->n_reloading--; |
a16e1123 | 3109 | return r; |
38c52d46 | 3110 | } |
a16e1123 LP |
3111 | } |
3112 | ||
a7556052 | 3113 | assert(m->n_reloading > 0); |
313cefa1 | 3114 | m->n_reloading--; |
38c52d46 | 3115 | |
01d32c00 LP |
3116 | r = fflush_and_check(f); |
3117 | if (r < 0) | |
3118 | return r; | |
a16e1123 | 3119 | |
b23de6af LP |
3120 | r = bus_fdset_add_all(m, fds); |
3121 | if (r < 0) | |
3122 | return r; | |
3123 | ||
a16e1123 LP |
3124 | return 0; |
3125 | } | |
3126 | ||
3127 | int manager_deserialize(Manager *m, FILE *f, FDSet *fds) { | |
3128 | int r = 0; | |
3129 | ||
3130 | assert(m); | |
3131 | assert(f); | |
3132 | ||
3133 | log_debug("Deserializing state..."); | |
3134 | ||
313cefa1 | 3135 | m->n_reloading++; |
82c64bf5 | 3136 | |
10f8e83c | 3137 | for (;;) { |
d233c99a ZJS |
3138 | char line[LINE_MAX]; |
3139 | const char *val, *l; | |
10f8e83c LP |
3140 | |
3141 | if (!fgets(line, sizeof(line), f)) { | |
3142 | if (feof(f)) | |
3143 | r = 0; | |
3144 | else | |
3145 | r = -errno; | |
3146 | ||
3147 | goto finish; | |
3148 | } | |
3149 | ||
3150 | char_array_0(line); | |
3151 | l = strstrip(line); | |
3152 | ||
3153 | if (l[0] == 0) | |
3154 | break; | |
3155 | ||
fb4650aa | 3156 | if ((val = startswith(l, "current-job-id="))) { |
01d67b43 LP |
3157 | uint32_t id; |
3158 | ||
fb4650aa | 3159 | if (safe_atou32(val, &id) < 0) |
62c460c6 | 3160 | log_notice("Failed to parse current job id value %s", val); |
01d67b43 LP |
3161 | else |
3162 | m->current_job_id = MAX(m->current_job_id, id); | |
718db961 | 3163 | |
fb4650aa | 3164 | } else if ((val = startswith(l, "n-installed-jobs="))) { |
33c5fae9 LP |
3165 | uint32_t n; |
3166 | ||
fb4650aa | 3167 | if (safe_atou32(val, &n) < 0) |
62c460c6 | 3168 | log_notice("Failed to parse installed jobs counter %s", val); |
33c5fae9 LP |
3169 | else |
3170 | m->n_installed_jobs += n; | |
718db961 | 3171 | |
fb4650aa | 3172 | } else if ((val = startswith(l, "n-failed-jobs="))) { |
33c5fae9 LP |
3173 | uint32_t n; |
3174 | ||
fb4650aa | 3175 | if (safe_atou32(val, &n) < 0) |
62c460c6 | 3176 | log_notice("Failed to parse failed jobs counter %s", val); |
33c5fae9 LP |
3177 | else |
3178 | m->n_failed_jobs += n; | |
718db961 | 3179 | |
fb4650aa | 3180 | } else if ((val = startswith(l, "taint-usr="))) { |
01d67b43 LP |
3181 | int b; |
3182 | ||
fb4650aa | 3183 | b = parse_boolean(val); |
e3dd987c | 3184 | if (b < 0) |
62c460c6 | 3185 | log_notice("Failed to parse taint /usr flag %s", val); |
01d67b43 LP |
3186 | else |
3187 | m->taint_usr = m->taint_usr || b; | |
718db961 | 3188 | |
0c2826c6 ZJS |
3189 | } else if ((val = startswith(l, "ready-sent="))) { |
3190 | int b; | |
3191 | ||
3192 | b = parse_boolean(val); | |
3193 | if (b < 0) | |
3194 | log_notice("Failed to parse ready-sent flag %s", val); | |
3195 | else | |
3196 | m->ready_sent = m->ready_sent || b; | |
3197 | ||
d8eb10d6 ZJS |
3198 | } else if ((val = startswith(l, "taint-logged="))) { |
3199 | int b; | |
3200 | ||
3201 | b = parse_boolean(val); | |
3202 | if (b < 0) | |
3203 | log_notice("Failed to parse taint-logged flag %s", val); | |
3204 | else | |
3205 | m->taint_logged = m->taint_logged || b; | |
3206 | ||
2a12e32e JK |
3207 | } else if ((val = startswith(l, "service-watchdogs="))) { |
3208 | int b; | |
3209 | ||
3210 | b = parse_boolean(val); | |
3211 | if (b < 0) | |
3212 | log_notice("Failed to parse service-watchdogs flag %s", val); | |
3213 | else | |
3214 | m->service_watchdogs = b; | |
3215 | ||
bee38b5c YW |
3216 | } else if ((val = startswith(l, "show-status="))) { |
3217 | ShowStatus s; | |
3218 | ||
3219 | s = show_status_from_string(val); | |
3220 | if (s < 0) | |
3221 | log_notice("Failed to parse show-status flag %s", val); | |
3222 | else | |
3223 | manager_set_show_status(m, s); | |
3224 | ||
a6ecbf83 FB |
3225 | } else if ((val = startswith(l, "log-level-override="))) { |
3226 | int level; | |
3227 | ||
3228 | level = log_level_from_string(val); | |
3229 | if (level < 0) | |
3230 | log_notice("Failed to parse log-level-override value '%s', ignoring.", val); | |
3231 | else | |
3232 | manager_override_log_level(m, level); | |
3233 | ||
bda7d78b FB |
3234 | } else if ((val = startswith(l, "log-target-override="))) { |
3235 | LogTarget target; | |
3236 | ||
3237 | target = log_target_from_string(val); | |
3238 | if (target < 0) | |
3239 | log_notice("Failed to parse log-target-override value '%s', ignoring.", val); | |
3240 | else | |
3241 | manager_override_log_target(m, target); | |
3242 | ||
9f9f0342 | 3243 | } else if (startswith(l, "env=")) { |
fe902fa4 | 3244 | r = deserialize_environment(&m->environment, l); |
d233c99a ZJS |
3245 | if (r == -ENOMEM) |
3246 | goto finish; | |
527b7a42 | 3247 | if (r < 0) |
d233c99a | 3248 | log_notice_errno(r, "Failed to parse environment entry: \"%s\": %m", l); |
e3dd987c | 3249 | |
fb4650aa | 3250 | } else if ((val = startswith(l, "notify-fd="))) { |
d86f9d52 LP |
3251 | int fd; |
3252 | ||
fb4650aa | 3253 | if (safe_atoi(val, &fd) < 0 || fd < 0 || !fdset_contains(fds, fd)) |
62c460c6 | 3254 | log_notice("Failed to parse notify fd: \"%s\"", val); |
d86f9d52 | 3255 | else { |
03e334a1 LP |
3256 | m->notify_event_source = sd_event_source_unref(m->notify_event_source); |
3257 | safe_close(m->notify_fd); | |
d86f9d52 LP |
3258 | m->notify_fd = fdset_remove(fds, fd); |
3259 | } | |
3260 | ||
fb4650aa | 3261 | } else if ((val = startswith(l, "notify-socket="))) { |
d86f9d52 LP |
3262 | char *n; |
3263 | ||
fb4650aa | 3264 | n = strdup(val); |
d86f9d52 LP |
3265 | if (!n) { |
3266 | r = -ENOMEM; | |
3267 | goto finish; | |
3268 | } | |
3269 | ||
3270 | free(m->notify_socket); | |
3271 | m->notify_socket = n; | |
3272 | ||
fb4650aa | 3273 | } else if ((val = startswith(l, "cgroups-agent-fd="))) { |
d8fdc620 LP |
3274 | int fd; |
3275 | ||
fb4650aa | 3276 | if (safe_atoi(val, &fd) < 0 || fd < 0 || !fdset_contains(fds, fd)) |
62c460c6 | 3277 | log_notice("Failed to parse cgroups agent fd: %s", val); |
d8fdc620 LP |
3278 | else { |
3279 | m->cgroups_agent_event_source = sd_event_source_unref(m->cgroups_agent_event_source); | |
3280 | safe_close(m->cgroups_agent_fd); | |
3281 | m->cgroups_agent_fd = fdset_remove(fds, fd); | |
3282 | } | |
3283 | ||
fb4650aa | 3284 | } else if ((val = startswith(l, "user-lookup="))) { |
00d9ef85 LP |
3285 | int fd0, fd1; |
3286 | ||
fb4650aa | 3287 | if (sscanf(val, "%i %i", &fd0, &fd1) != 2 || fd0 < 0 || fd1 < 0 || fd0 == fd1 || !fdset_contains(fds, fd0) || !fdset_contains(fds, fd1)) |
62c460c6 | 3288 | log_notice("Failed to parse user lookup fd: %s", val); |
00d9ef85 LP |
3289 | else { |
3290 | m->user_lookup_event_source = sd_event_source_unref(m->user_lookup_event_source); | |
3291 | safe_close_pair(m->user_lookup_fds); | |
3292 | m->user_lookup_fds[0] = fdset_remove(fds, fd0); | |
3293 | m->user_lookup_fds[1] = fdset_remove(fds, fd1); | |
3294 | } | |
3295 | ||
fb4650aa ZJS |
3296 | } else if ((val = startswith(l, "dynamic-user="))) |
3297 | dynamic_user_deserialize_one(m, val, fds); | |
3298 | else if ((val = startswith(l, "destroy-ipc-uid="))) | |
3299 | manager_deserialize_uid_refs_one(m, val); | |
3300 | else if ((val = startswith(l, "destroy-ipc-gid="))) | |
3301 | manager_deserialize_gid_refs_one(m, val); | |
e8a565cb YW |
3302 | else if ((val = startswith(l, "exec-runtime="))) |
3303 | exec_runtime_deserialize_one(m, val, fds); | |
fb4650aa | 3304 | else if ((val = startswith(l, "subscribed="))) { |
05a98afd | 3305 | |
fb4650aa | 3306 | if (strv_extend(&m->deserialized_subscribed, val) < 0) |
05a98afd | 3307 | log_oom(); |
9f9f0342 LP |
3308 | } else { |
3309 | ManagerTimestamp q; | |
3310 | ||
3311 | for (q = 0; q < _MANAGER_TIMESTAMP_MAX; q++) { | |
3312 | val = startswith(l, manager_timestamp_to_string(q)); | |
3313 | if (!val) | |
3314 | continue; | |
05a98afd | 3315 | |
9f9f0342 LP |
3316 | val = startswith(val, "-timestamp="); |
3317 | if (val) | |
3318 | break; | |
3319 | } | |
05a98afd | 3320 | |
9f9f0342 LP |
3321 | if (q < _MANAGER_TIMESTAMP_MAX) /* found it */ |
3322 | dual_timestamp_deserialize(val, m->timestamps + q); | |
3323 | else if (!startswith(l, "kdbus-fd=")) /* ignore kdbus */ | |
3324 | log_notice("Unknown serialization item '%s'", l); | |
3325 | } | |
10f8e83c LP |
3326 | } |
3327 | ||
a16e1123 LP |
3328 | for (;;) { |
3329 | Unit *u; | |
3330 | char name[UNIT_NAME_MAX+2]; | |
07429866 | 3331 | const char* unit_name; |
a16e1123 LP |
3332 | |
3333 | /* Start marker */ | |
3334 | if (!fgets(name, sizeof(name), f)) { | |
3335 | if (feof(f)) | |
10f8e83c LP |
3336 | r = 0; |
3337 | else | |
3338 | r = -errno; | |
a16e1123 | 3339 | |
82c64bf5 | 3340 | goto finish; |
a16e1123 LP |
3341 | } |
3342 | ||
3343 | char_array_0(name); | |
07429866 | 3344 | unit_name = strstrip(name); |
a16e1123 | 3345 | |
07429866 ZJS |
3346 | r = manager_load_unit(m, unit_name, NULL, NULL, &u); |
3347 | if (r < 0) { | |
3348 | log_notice_errno(r, "Failed to load unit \"%s\", skipping deserialization: %m", unit_name); | |
3349 | if (r == -ENOMEM) | |
3350 | goto finish; | |
3351 | unit_deserialize_skip(f); | |
3352 | continue; | |
3353 | } | |
a16e1123 | 3354 | |
01e10de3 | 3355 | r = unit_deserialize(u, f, fds); |
07429866 ZJS |
3356 | if (r < 0) { |
3357 | log_notice_errno(r, "Failed to deserialize unit \"%s\": %m", unit_name); | |
3358 | if (r == -ENOMEM) | |
3359 | goto finish; | |
3360 | } | |
a16e1123 LP |
3361 | } |
3362 | ||
10f8e83c | 3363 | finish: |
145b1f79 | 3364 | if (ferror(f)) |
82c64bf5 | 3365 | r = -EIO; |
a16e1123 | 3366 | |
a7556052 | 3367 | assert(m->n_reloading > 0); |
313cefa1 | 3368 | m->n_reloading--; |
82c64bf5 LP |
3369 | |
3370 | return r; | |
a16e1123 LP |
3371 | } |
3372 | ||
a7a7163d DT |
3373 | static void manager_flush_finished_jobs(Manager *m) { |
3374 | Job *j; | |
3375 | ||
3376 | while ((j = set_steal_first(m->pending_finished_jobs))) { | |
3377 | bus_job_send_removed_signal(j); | |
3378 | job_free(j); | |
3379 | } | |
3380 | ||
3381 | m->pending_finished_jobs = set_free(m->pending_finished_jobs); | |
3382 | } | |
3383 | ||
a16e1123 LP |
3384 | int manager_reload(Manager *m) { |
3385 | int r, q; | |
51d122af ZJS |
3386 | _cleanup_fclose_ FILE *f = NULL; |
3387 | _cleanup_fdset_free_ FDSet *fds = NULL; | |
a16e1123 LP |
3388 | |
3389 | assert(m); | |
3390 | ||
07719a21 LP |
3391 | r = manager_open_serialization(m, &f); |
3392 | if (r < 0) | |
a16e1123 LP |
3393 | return r; |
3394 | ||
313cefa1 | 3395 | m->n_reloading++; |
718db961 | 3396 | bus_manager_send_reloading(m, true); |
38c52d46 | 3397 | |
07719a21 LP |
3398 | fds = fdset_new(); |
3399 | if (!fds) { | |
313cefa1 | 3400 | m->n_reloading--; |
51d122af | 3401 | return -ENOMEM; |
a16e1123 LP |
3402 | } |
3403 | ||
b3680f49 | 3404 | r = manager_serialize(m, f, fds, false); |
07719a21 | 3405 | if (r < 0) { |
313cefa1 | 3406 | m->n_reloading--; |
51d122af | 3407 | return r; |
38c52d46 | 3408 | } |
a16e1123 LP |
3409 | |
3410 | if (fseeko(f, 0, SEEK_SET) < 0) { | |
313cefa1 | 3411 | m->n_reloading--; |
51d122af | 3412 | return -errno; |
a16e1123 LP |
3413 | } |
3414 | ||
3415 | /* From here on there is no way back. */ | |
3416 | manager_clear_jobs_and_units(m); | |
07a78643 | 3417 | lookup_paths_flush_generator(&m->lookup_paths); |
84e3543e | 3418 | lookup_paths_free(&m->lookup_paths); |
e8a565cb | 3419 | exec_runtime_vacuum(m); |
29206d46 | 3420 | dynamic_user_vacuum(m, false); |
00d9ef85 LP |
3421 | m->uid_refs = hashmap_free(m->uid_refs); |
3422 | m->gid_refs = hashmap_free(m->gid_refs); | |
2ded0c04 | 3423 | |
4943d143 | 3424 | q = lookup_paths_init(&m->lookup_paths, m->unit_file_scope, 0, NULL); |
e801700e ZJS |
3425 | if (q < 0 && r >= 0) |
3426 | r = q; | |
5a1e9937 | 3427 | |
64691d20 ZJS |
3428 | q = manager_run_environment_generators(m); |
3429 | if (q < 0 && r >= 0) | |
3430 | r = q; | |
3431 | ||
a3c4eb07 LP |
3432 | /* Find new unit paths */ |
3433 | q = manager_run_generators(m); | |
e801700e | 3434 | if (q < 0 && r >= 0) |
07719a21 LP |
3435 | r = q; |
3436 | ||
a1453343 | 3437 | lookup_paths_reduce(&m->lookup_paths); |
5a1e9937 LP |
3438 | manager_build_unit_path_cache(m); |
3439 | ||
a16e1123 | 3440 | /* First, enumerate what we can from all config files */ |
ba64af90 | 3441 | manager_enumerate(m); |
a16e1123 LP |
3442 | |
3443 | /* Second, deserialize our stored data */ | |
07719a21 | 3444 | q = manager_deserialize(m, f, fds); |
07429866 ZJS |
3445 | if (q < 0) { |
3446 | log_error_errno(q, "Deserialization failed: %m"); | |
3447 | ||
3448 | if (r >= 0) | |
3449 | r = q; | |
3450 | } | |
a16e1123 | 3451 | |
62b0cbb3 | 3452 | f = safe_fclose(f); |
a16e1123 | 3453 | |
a2cc4a6c ZJS |
3454 | /* Re-register notify_fd as event source */ |
3455 | q = manager_setup_notify(m); | |
e801700e | 3456 | if (q < 0 && r >= 0) |
a2cc4a6c ZJS |
3457 | r = q; |
3458 | ||
d8fdc620 LP |
3459 | q = manager_setup_cgroups_agent(m); |
3460 | if (q < 0 && r >= 0) | |
3461 | r = q; | |
3462 | ||
00d9ef85 LP |
3463 | q = manager_setup_user_lookup_fd(m); |
3464 | if (q < 0 && r >= 0) | |
3465 | r = q; | |
3466 | ||
a16e1123 | 3467 | /* Third, fire things up! */ |
007c6337 | 3468 | manager_coldplug(m); |
a16e1123 | 3469 | |
29206d46 LP |
3470 | /* Release any dynamic users no longer referenced */ |
3471 | dynamic_user_vacuum(m, true); | |
3472 | ||
00d9ef85 LP |
3473 | /* Release any references to UIDs/GIDs no longer referenced, and destroy any IPC owned by them */ |
3474 | manager_vacuum_uid_refs(m); | |
3475 | manager_vacuum_gid_refs(m); | |
3476 | ||
e8a565cb YW |
3477 | exec_runtime_vacuum(m); |
3478 | ||
31dc1ca3 LP |
3479 | assert(m->n_reloading > 0); |
3480 | m->n_reloading--; | |
3481 | ||
8559b3b7 | 3482 | /* It might be safe to log to the journal now and connect to dbus */ |
d075092f | 3483 | manager_recheck_journal(m); |
8559b3b7 | 3484 | manager_recheck_dbus(m); |
d075092f | 3485 | |
f0831ed2 LP |
3486 | /* Let's finally catch up with any changes that took place while we were reloading/reexecing */ |
3487 | manager_catchup(m); | |
3488 | ||
8936a5e3 | 3489 | /* Sync current state of bus names with our set of listening units */ |
5f109056 LP |
3490 | q = manager_enqueue_sync_bus_names(m); |
3491 | if (q < 0 && r >= 0) | |
3492 | r = q; | |
8936a5e3 | 3493 | |
a7a7163d DT |
3494 | if (!MANAGER_IS_RELOADING(m)) |
3495 | manager_flush_finished_jobs(m); | |
3496 | ||
71445ae7 LP |
3497 | m->send_reloading_done = true; |
3498 | ||
a16e1123 LP |
3499 | return r; |
3500 | } | |
3501 | ||
fdf20a31 | 3502 | void manager_reset_failed(Manager *m) { |
5632e374 LP |
3503 | Unit *u; |
3504 | Iterator i; | |
3505 | ||
3506 | assert(m); | |
3507 | ||
3508 | HASHMAP_FOREACH(u, m->units, i) | |
fdf20a31 | 3509 | unit_reset_failed(u); |
5632e374 LP |
3510 | } |
3511 | ||
31afa0a4 | 3512 | bool manager_unit_inactive_or_pending(Manager *m, const char *name) { |
8f6df3fa LP |
3513 | Unit *u; |
3514 | ||
3515 | assert(m); | |
3516 | assert(name); | |
3517 | ||
3518 | /* Returns true if the unit is inactive or going down */ | |
bd0af849 ZJS |
3519 | u = manager_get_unit(m, name); |
3520 | if (!u) | |
8f6df3fa LP |
3521 | return true; |
3522 | ||
31afa0a4 | 3523 | return unit_inactive_or_pending(u); |
8f6df3fa LP |
3524 | } |
3525 | ||
d8eb10d6 ZJS |
3526 | static void log_taint_string(Manager *m) { |
3527 | _cleanup_free_ char *taint = NULL; | |
3528 | ||
3529 | assert(m); | |
3530 | ||
3531 | if (MANAGER_IS_USER(m) || m->taint_logged) | |
3532 | return; | |
3533 | ||
3534 | m->taint_logged = true; /* only check for taint once */ | |
3535 | ||
3536 | taint = manager_taint_string(m); | |
3537 | if (isempty(taint)) | |
3538 | return; | |
3539 | ||
3540 | log_struct(LOG_NOTICE, | |
3541 | LOG_MESSAGE("System is tainted: %s", taint), | |
3542 | "TAINT=%s", taint, | |
a1230ff9 | 3543 | "MESSAGE_ID=" SD_MESSAGE_TAINTED_STR); |
d8eb10d6 ZJS |
3544 | } |
3545 | ||
56dacdbc | 3546 | static void manager_notify_finished(Manager *m) { |
7ceba241 | 3547 | char userspace[FORMAT_TIMESPAN_MAX], initrd[FORMAT_TIMESPAN_MAX], kernel[FORMAT_TIMESPAN_MAX], sum[FORMAT_TIMESPAN_MAX]; |
915b3753 | 3548 | usec_t firmware_usec, loader_usec, kernel_usec, initrd_usec, userspace_usec, total_usec; |
b0c918b9 | 3549 | |
e0a3da1f | 3550 | if (m->test_run_flags) |
b0c918b9 LP |
3551 | return; |
3552 | ||
463d0d15 | 3553 | if (MANAGER_IS_SYSTEM(m) && detect_container() <= 0) { |
dd1db3c2 | 3554 | char ts[FORMAT_TIMESPAN_MAX]; |
dc3c9f5e ZJS |
3555 | char buf[FORMAT_TIMESPAN_MAX + STRLEN(" (firmware) + ") + FORMAT_TIMESPAN_MAX + STRLEN(" (loader) + ")] |
3556 | = {}; | |
3557 | char *p = buf; | |
3558 | size_t size = sizeof buf; | |
e03ae661 | 3559 | |
9f9f0342 LP |
3560 | /* Note that MANAGER_TIMESTAMP_KERNEL's monotonic value is always at 0, and |
3561 | * MANAGER_TIMESTAMP_FIRMWARE's and MANAGER_TIMESTAMP_LOADER's monotonic value should be considered | |
915b3753 LP |
3562 | * negative values. */ |
3563 | ||
9f9f0342 LP |
3564 | firmware_usec = m->timestamps[MANAGER_TIMESTAMP_FIRMWARE].monotonic - m->timestamps[MANAGER_TIMESTAMP_LOADER].monotonic; |
3565 | loader_usec = m->timestamps[MANAGER_TIMESTAMP_LOADER].monotonic - m->timestamps[MANAGER_TIMESTAMP_KERNEL].monotonic; | |
3566 | userspace_usec = m->timestamps[MANAGER_TIMESTAMP_FINISH].monotonic - m->timestamps[MANAGER_TIMESTAMP_USERSPACE].monotonic; | |
3567 | total_usec = m->timestamps[MANAGER_TIMESTAMP_FIRMWARE].monotonic + m->timestamps[MANAGER_TIMESTAMP_FINISH].monotonic; | |
18fa6b27 | 3568 | |
dd1db3c2 YW |
3569 | if (firmware_usec > 0) |
3570 | size = strpcpyf(&p, size, "%s (firmware) + ", format_timespan(ts, sizeof(ts), firmware_usec, USEC_PER_MSEC)); | |
3571 | if (loader_usec > 0) | |
3572 | size = strpcpyf(&p, size, "%s (loader) + ", format_timespan(ts, sizeof(ts), loader_usec, USEC_PER_MSEC)); | |
3573 | ||
9f9f0342 | 3574 | if (dual_timestamp_is_set(&m->timestamps[MANAGER_TIMESTAMP_INITRD])) { |
18fa6b27 | 3575 | |
9f9f0342 LP |
3576 | /* The initrd case on bare-metal*/ |
3577 | kernel_usec = m->timestamps[MANAGER_TIMESTAMP_INITRD].monotonic - m->timestamps[MANAGER_TIMESTAMP_KERNEL].monotonic; | |
3578 | initrd_usec = m->timestamps[MANAGER_TIMESTAMP_USERSPACE].monotonic - m->timestamps[MANAGER_TIMESTAMP_INITRD].monotonic; | |
18fa6b27 | 3579 | |
e12919e8 | 3580 | log_struct(LOG_INFO, |
2b044526 | 3581 | "MESSAGE_ID=" SD_MESSAGE_STARTUP_FINISHED_STR, |
e12919e8 LP |
3582 | "KERNEL_USEC="USEC_FMT, kernel_usec, |
3583 | "INITRD_USEC="USEC_FMT, initrd_usec, | |
3584 | "USERSPACE_USEC="USEC_FMT, userspace_usec, | |
dd1db3c2 YW |
3585 | LOG_MESSAGE("Startup finished in %s%s (kernel) + %s (initrd) + %s (userspace) = %s.", |
3586 | buf, | |
e2cc6eca LP |
3587 | format_timespan(kernel, sizeof(kernel), kernel_usec, USEC_PER_MSEC), |
3588 | format_timespan(initrd, sizeof(initrd), initrd_usec, USEC_PER_MSEC), | |
3589 | format_timespan(userspace, sizeof(userspace), userspace_usec, USEC_PER_MSEC), | |
a1230ff9 | 3590 | format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC))); |
18fa6b27 | 3591 | } else { |
9f9f0342 LP |
3592 | /* The initrd-less case on bare-metal*/ |
3593 | ||
3594 | kernel_usec = m->timestamps[MANAGER_TIMESTAMP_USERSPACE].monotonic - m->timestamps[MANAGER_TIMESTAMP_KERNEL].monotonic; | |
18fa6b27 LP |
3595 | initrd_usec = 0; |
3596 | ||
81270860 | 3597 | log_struct(LOG_INFO, |
2b044526 | 3598 | "MESSAGE_ID=" SD_MESSAGE_STARTUP_FINISHED_STR, |
e12919e8 | 3599 | "KERNEL_USEC="USEC_FMT, kernel_usec, |
ccd06097 | 3600 | "USERSPACE_USEC="USEC_FMT, userspace_usec, |
dd1db3c2 YW |
3601 | LOG_MESSAGE("Startup finished in %s%s (kernel) + %s (userspace) = %s.", |
3602 | buf, | |
e2cc6eca LP |
3603 | format_timespan(kernel, sizeof(kernel), kernel_usec, USEC_PER_MSEC), |
3604 | format_timespan(userspace, sizeof(userspace), userspace_usec, USEC_PER_MSEC), | |
a1230ff9 | 3605 | format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC))); |
e12919e8 LP |
3606 | } |
3607 | } else { | |
4adf314b | 3608 | /* The container and --user case */ |
e12919e8 | 3609 | firmware_usec = loader_usec = initrd_usec = kernel_usec = 0; |
9f9f0342 | 3610 | total_usec = userspace_usec = m->timestamps[MANAGER_TIMESTAMP_FINISH].monotonic - m->timestamps[MANAGER_TIMESTAMP_USERSPACE].monotonic; |
e12919e8 LP |
3611 | |
3612 | log_struct(LOG_INFO, | |
2b044526 | 3613 | "MESSAGE_ID=" SD_MESSAGE_USER_STARTUP_FINISHED_STR, |
e12919e8 | 3614 | "USERSPACE_USEC="USEC_FMT, userspace_usec, |
e2cc6eca | 3615 | LOG_MESSAGE("Startup finished in %s.", |
a1230ff9 | 3616 | format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC))); |
18fa6b27 | 3617 | } |
b0c918b9 | 3618 | |
718db961 | 3619 | bus_manager_send_finished(m, firmware_usec, loader_usec, kernel_usec, initrd_usec, userspace_usec, total_usec); |
530345e7 LP |
3620 | |
3621 | sd_notifyf(false, | |
0c2826c6 ZJS |
3622 | m->ready_sent ? "STATUS=Startup finished in %s." |
3623 | : "READY=1\n" | |
3624 | "STATUS=Startup finished in %s.", | |
2fa4092c | 3625 | format_timespan(sum, sizeof(sum), total_usec, USEC_PER_MSEC)); |
0c2826c6 | 3626 | m->ready_sent = true; |
d8eb10d6 ZJS |
3627 | |
3628 | log_taint_string(m); | |
b0c918b9 LP |
3629 | } |
3630 | ||
4adf314b LP |
3631 | static void manager_send_ready(Manager *m) { |
3632 | assert(m); | |
3633 | ||
3634 | /* We send READY=1 on reaching basic.target only when running in --user mode. */ | |
3635 | if (!MANAGER_IS_USER(m) || m->ready_sent) | |
3636 | return; | |
3637 | ||
3638 | m->ready_sent = true; | |
3639 | ||
3640 | sd_notifyf(false, | |
3641 | "READY=1\n" | |
3642 | "STATUS=Reached " SPECIAL_BASIC_TARGET "."); | |
3643 | } | |
3644 | ||
3645 | static void manager_check_basic_target(Manager *m) { | |
3646 | Unit *u; | |
3647 | ||
3648 | assert(m); | |
3649 | ||
3650 | /* Small shortcut */ | |
3651 | if (m->ready_sent && m->taint_logged) | |
3652 | return; | |
3653 | ||
3654 | u = manager_get_unit(m, SPECIAL_BASIC_TARGET); | |
3655 | if (!u || !UNIT_IS_ACTIVE_OR_RELOADING(unit_active_state(u))) | |
3656 | return; | |
3657 | ||
3658 | /* For user managers, send out READY=1 as soon as we reach basic.target */ | |
3659 | manager_send_ready(m); | |
3660 | ||
3661 | /* Log the taint string as soon as we reach basic.target */ | |
3662 | log_taint_string(m); | |
3663 | } | |
3664 | ||
56dacdbc | 3665 | void manager_check_finished(Manager *m) { |
56dacdbc ZJS |
3666 | assert(m); |
3667 | ||
2c289ea8 | 3668 | if (MANAGER_IS_RELOADING(m)) |
aad1976f LP |
3669 | return; |
3670 | ||
4259d202 LP |
3671 | /* Verify that we have entered the event loop already, and not left it again. */ |
3672 | if (!MANAGER_IS_RUNNING(m)) | |
9771b62d LP |
3673 | return; |
3674 | ||
4adf314b | 3675 | manager_check_basic_target(m); |
0c2826c6 | 3676 | |
56dacdbc | 3677 | if (hashmap_size(m->jobs) > 0) { |
56dacdbc | 3678 | if (m->jobs_in_progress_event_source) |
2ae56591 | 3679 | /* Ignore any failure, this is only for feedback */ |
e7ab4d1a | 3680 | (void) sd_event_source_set_time(m->jobs_in_progress_event_source, now(CLOCK_MONOTONIC) + JOBS_IN_PROGRESS_WAIT_USEC); |
56dacdbc ZJS |
3681 | |
3682 | return; | |
3683 | } | |
3684 | ||
3685 | manager_flip_auto_status(m, false); | |
3686 | ||
3687 | /* Notify Type=idle units that we are done now */ | |
56dacdbc ZJS |
3688 | manager_close_idle_pipe(m); |
3689 | ||
3690 | /* Turn off confirm spawn now */ | |
7d5ceb64 | 3691 | m->confirm_spawn = NULL; |
56dacdbc ZJS |
3692 | |
3693 | /* No need to update ask password status when we're going non-interactive */ | |
3694 | manager_close_ask_password(m); | |
3695 | ||
3696 | /* This is no longer the first boot */ | |
3697 | manager_set_first_boot(m, false); | |
3698 | ||
49d5666c | 3699 | if (MANAGER_IS_FINISHED(m)) |
56dacdbc ZJS |
3700 | return; |
3701 | ||
9f9f0342 | 3702 | dual_timestamp_get(m->timestamps + MANAGER_TIMESTAMP_FINISH); |
56dacdbc ZJS |
3703 | |
3704 | manager_notify_finished(m); | |
3705 | ||
e7ab4d1a | 3706 | manager_invalidate_startup_units(m); |
56dacdbc ZJS |
3707 | } |
3708 | ||
64691d20 ZJS |
3709 | static bool generator_path_any(const char* const* paths) { |
3710 | char **path; | |
3711 | bool found = false; | |
3712 | ||
3713 | /* Optimize by skipping the whole process by not creating output directories | |
3714 | * if no generators are found. */ | |
3715 | STRV_FOREACH(path, (char**) paths) | |
3716 | if (access(*path, F_OK) == 0) | |
3717 | found = true; | |
3718 | else if (errno != ENOENT) | |
3719 | log_warning_errno(errno, "Failed to open generator directory %s: %m", *path); | |
3720 | ||
3721 | return found; | |
3722 | } | |
3723 | ||
3724 | static const char* system_env_generator_binary_paths[] = { | |
3725 | "/run/systemd/system-environment-generators", | |
3726 | "/etc/systemd/system-environment-generators", | |
3727 | "/usr/local/lib/systemd/system-environment-generators", | |
3728 | SYSTEM_ENV_GENERATOR_PATH, | |
3729 | NULL | |
3730 | }; | |
3731 | ||
3732 | static const char* user_env_generator_binary_paths[] = { | |
3733 | "/run/systemd/user-environment-generators", | |
3734 | "/etc/systemd/user-environment-generators", | |
3735 | "/usr/local/lib/systemd/user-environment-generators", | |
3736 | USER_ENV_GENERATOR_PATH, | |
3737 | NULL | |
3738 | }; | |
3739 | ||
3740 | static int manager_run_environment_generators(Manager *m) { | |
3741 | char **tmp = NULL; /* this is only used in the forked process, no cleanup here */ | |
3742 | const char **paths; | |
3743 | void* args[] = {&tmp, &tmp, &m->environment}; | |
3744 | ||
e0a3da1f ZJS |
3745 | if (m->test_run_flags && !(m->test_run_flags & MANAGER_TEST_RUN_ENV_GENERATORS)) |
3746 | return 0; | |
3747 | ||
64691d20 ZJS |
3748 | paths = MANAGER_IS_SYSTEM(m) ? system_env_generator_binary_paths : user_env_generator_binary_paths; |
3749 | ||
3750 | if (!generator_path_any(paths)) | |
3751 | return 0; | |
3752 | ||
3753 | return execute_directories(paths, DEFAULT_TIMEOUT_USEC, gather_environment, args, NULL); | |
3754 | } | |
3755 | ||
e801700e | 3756 | static int manager_run_generators(Manager *m) { |
f42348ac | 3757 | _cleanup_strv_free_ char **paths = NULL; |
07719a21 | 3758 | const char *argv[5]; |
07719a21 | 3759 | int r; |
5a1e9937 LP |
3760 | |
3761 | assert(m); | |
3762 | ||
e0a3da1f ZJS |
3763 | if (m->test_run_flags && !(m->test_run_flags & MANAGER_TEST_RUN_GENERATORS)) |
3764 | return 0; | |
3765 | ||
9183df70 | 3766 | paths = generator_binary_paths(m->unit_file_scope); |
e801700e ZJS |
3767 | if (!paths) |
3768 | return log_oom(); | |
5a1e9937 | 3769 | |
64691d20 ZJS |
3770 | if (!generator_path_any((const char* const*) paths)) |
3771 | return 0; | |
5a1e9937 | 3772 | |
cd64fd56 | 3773 | r = lookup_paths_mkdir_generator(&m->lookup_paths); |
07719a21 LP |
3774 | if (r < 0) |
3775 | goto finish; | |
5a1e9937 | 3776 | |
83cc030f | 3777 | argv[0] = NULL; /* Leave this empty, execute_directory() will fill something in */ |
a3c4eb07 LP |
3778 | argv[1] = m->lookup_paths.generator; |
3779 | argv[2] = m->lookup_paths.generator_early; | |
3780 | argv[3] = m->lookup_paths.generator_late; | |
07719a21 | 3781 | argv[4] = NULL; |
5a1e9937 | 3782 | |
718db961 | 3783 | RUN_WITH_UMASK(0022) |
c6e47247 ZJS |
3784 | execute_directories((const char* const*) paths, DEFAULT_TIMEOUT_USEC, |
3785 | NULL, NULL, (char**) argv); | |
5a1e9937 | 3786 | |
718db961 | 3787 | finish: |
cd64fd56 | 3788 | lookup_paths_trim_generator(&m->lookup_paths); |
e801700e | 3789 | return r; |
5a1e9937 LP |
3790 | } |
3791 | ||
718db961 LP |
3792 | int manager_environment_add(Manager *m, char **minus, char **plus) { |
3793 | char **a = NULL, **b = NULL, **l; | |
97d0e5f8 | 3794 | assert(m); |
bcd8e6d1 | 3795 | |
718db961 | 3796 | l = m->environment; |
bcd8e6d1 | 3797 | |
718db961 LP |
3798 | if (!strv_isempty(minus)) { |
3799 | a = strv_env_delete(l, 1, minus); | |
3800 | if (!a) | |
3801 | return -ENOMEM; | |
3802 | ||
3803 | l = a; | |
3804 | } | |
3805 | ||
3806 | if (!strv_isempty(plus)) { | |
3807 | b = strv_env_merge(2, l, plus); | |
aa9f8a30 AH |
3808 | if (!b) { |
3809 | strv_free(a); | |
718db961 | 3810 | return -ENOMEM; |
aa9f8a30 | 3811 | } |
bcd8e6d1 | 3812 | |
718db961 LP |
3813 | l = b; |
3814 | } | |
3815 | ||
3816 | if (m->environment != l) | |
3817 | strv_free(m->environment); | |
3818 | if (a != l) | |
3819 | strv_free(a); | |
3820 | if (b != l) | |
3821 | strv_free(b); | |
3822 | ||
f069efb4 | 3823 | m->environment = l; |
47cf8ff2 | 3824 | manager_sanitize_environment(m); |
f069efb4 | 3825 | |
97d0e5f8 UTL |
3826 | return 0; |
3827 | } | |
3828 | ||
c93ff2e9 FC |
3829 | int manager_set_default_rlimits(Manager *m, struct rlimit **default_rlimit) { |
3830 | int i; | |
3831 | ||
3832 | assert(m); | |
3833 | ||
517d56b1 | 3834 | for (i = 0; i < _RLIMIT_MAX; i++) { |
d9814c76 EV |
3835 | m->rlimit[i] = mfree(m->rlimit[i]); |
3836 | ||
07719a21 LP |
3837 | if (!default_rlimit[i]) |
3838 | continue; | |
c93ff2e9 | 3839 | |
07719a21 LP |
3840 | m->rlimit[i] = newdup(struct rlimit, default_rlimit[i], 1); |
3841 | if (!m->rlimit[i]) | |
3ce40911 | 3842 | return log_oom(); |
c93ff2e9 FC |
3843 | } |
3844 | ||
3845 | return 0; | |
3846 | } | |
3847 | ||
8559b3b7 LP |
3848 | void manager_recheck_dbus(Manager *m) { |
3849 | assert(m); | |
3850 | ||
3851 | /* Connects to the bus if the dbus service and socket are running. If we are running in user mode this is all | |
3852 | * it does. In system mode we'll also connect to the system bus (which will most likely just reuse the | |
3853 | * connection of the API bus). That's because the system bus after all runs as service of the system instance, | |
3854 | * while in the user instance we can assume it's already there. */ | |
3855 | ||
31dc1ca3 LP |
3856 | if (MANAGER_IS_RELOADING(m)) |
3857 | return; /* don't check while we are reloading… */ | |
3858 | ||
8559b3b7 LP |
3859 | if (manager_dbus_is_running(m, false)) { |
3860 | (void) bus_init_api(m); | |
3861 | ||
3862 | if (MANAGER_IS_SYSTEM(m)) | |
3863 | (void) bus_init_system(m); | |
3864 | } else { | |
3865 | (void) bus_done_api(m); | |
3866 | ||
3867 | if (MANAGER_IS_SYSTEM(m)) | |
3868 | (void) bus_done_system(m); | |
3869 | } | |
3870 | } | |
3871 | ||
d075092f | 3872 | static bool manager_journal_is_running(Manager *m) { |
f1dd0c3f LP |
3873 | Unit *u; |
3874 | ||
3875 | assert(m); | |
3876 | ||
7d814a19 LP |
3877 | if (m->test_run_flags != 0) |
3878 | return false; | |
3879 | ||
d075092f | 3880 | /* If we are the user manager we can safely assume that the journal is up */ |
463d0d15 | 3881 | if (!MANAGER_IS_SYSTEM(m)) |
d075092f | 3882 | return true; |
f1dd0c3f | 3883 | |
d075092f | 3884 | /* Check that the socket is not only up, but in RUNNING state */ |
731a676c | 3885 | u = manager_get_unit(m, SPECIAL_JOURNALD_SOCKET); |
d075092f LP |
3886 | if (!u) |
3887 | return false; | |
3888 | if (SOCKET(u)->state != SOCKET_RUNNING) | |
3889 | return false; | |
f1dd0c3f | 3890 | |
d075092f | 3891 | /* Similar, check if the daemon itself is fully up, too */ |
731a676c | 3892 | u = manager_get_unit(m, SPECIAL_JOURNALD_SERVICE); |
d075092f LP |
3893 | if (!u) |
3894 | return false; | |
217677ab | 3895 | if (!IN_SET(SERVICE(u)->state, SERVICE_RELOAD, SERVICE_RUNNING)) |
d075092f LP |
3896 | return false; |
3897 | ||
3898 | return true; | |
3899 | } | |
3900 | ||
3901 | void manager_recheck_journal(Manager *m) { | |
3902 | ||
3903 | assert(m); | |
3904 | ||
3905 | /* Don't bother with this unless we are in the special situation of being PID 1 */ | |
3906 | if (getpid_cached() != 1) | |
731a676c | 3907 | return; |
f1dd0c3f | 3908 | |
31dc1ca3 LP |
3909 | /* Don't check this while we are reloading, things might still change */ |
3910 | if (MANAGER_IS_RELOADING(m)) | |
3911 | return; | |
3912 | ||
cedf5088 LP |
3913 | /* The journal is fully and entirely up? If so, let's permit logging to it, if that's configured. If the |
3914 | * journal is down, don't ever log to it, otherwise we might end up deadlocking ourselves as we might trigger | |
3915 | * an activation ourselves we can't fulfill. */ | |
3916 | log_set_prohibit_ipc(!manager_journal_is_running(m)); | |
cc2b9e6b | 3917 | log_open(); |
f1dd0c3f LP |
3918 | } |
3919 | ||
d450b6f2 | 3920 | void manager_set_show_status(Manager *m, ShowStatus mode) { |
27d340c7 | 3921 | assert(m); |
d450b6f2 | 3922 | assert(IN_SET(mode, SHOW_STATUS_AUTO, SHOW_STATUS_NO, SHOW_STATUS_YES, SHOW_STATUS_TEMPORARY)); |
27d340c7 | 3923 | |
463d0d15 | 3924 | if (!MANAGER_IS_SYSTEM(m)) |
27d340c7 LP |
3925 | return; |
3926 | ||
76b6f3f6 ZJS |
3927 | if (m->show_status != mode) |
3928 | log_debug("%s showing of status.", | |
3929 | mode == SHOW_STATUS_NO ? "Disabling" : "Enabling"); | |
d450b6f2 | 3930 | m->show_status = mode; |
27d340c7 | 3931 | |
7a293242 | 3932 | if (IN_SET(mode, SHOW_STATUS_TEMPORARY, SHOW_STATUS_YES)) |
ac5b0c13 | 3933 | (void) touch("/run/systemd/show-status"); |
27d340c7 | 3934 | else |
ac5b0c13 | 3935 | (void) unlink("/run/systemd/show-status"); |
27d340c7 LP |
3936 | } |
3937 | ||
127d5fd1 | 3938 | static bool manager_get_show_status(Manager *m, StatusType type) { |
27d340c7 LP |
3939 | assert(m); |
3940 | ||
463d0d15 | 3941 | if (!MANAGER_IS_SYSTEM(m)) |
27d340c7 LP |
3942 | return false; |
3943 | ||
31a7eb86 ZJS |
3944 | if (m->no_console_output) |
3945 | return false; | |
3946 | ||
d81afec1 | 3947 | if (!IN_SET(manager_state(m), MANAGER_INITIALIZING, MANAGER_STARTING, MANAGER_STOPPING)) |
08510627 LP |
3948 | return false; |
3949 | ||
e46b13c8 | 3950 | /* If we cannot find out the status properly, just proceed. */ |
ebc5788e | 3951 | if (type != STATUS_TYPE_EMERGENCY && manager_check_ask_password(m) > 0) |
e46b13c8 ZJS |
3952 | return false; |
3953 | ||
7a293242 | 3954 | return IN_SET(m->show_status, SHOW_STATUS_TEMPORARY, SHOW_STATUS_YES); |
27d340c7 | 3955 | } |
68b29a9f | 3956 | |
7d5ceb64 FB |
3957 | const char *manager_get_confirm_spawn(Manager *m) { |
3958 | static int last_errno = 0; | |
3959 | const char *vc = m->confirm_spawn; | |
3960 | struct stat st; | |
3961 | int r; | |
3962 | ||
3963 | /* Here's the deal: we want to test the validity of the console but don't want | |
3964 | * PID1 to go through the whole console process which might block. But we also | |
3965 | * want to warn the user only once if something is wrong with the console so we | |
3966 | * cannot do the sanity checks after spawning our children. So here we simply do | |
3967 | * really basic tests to hopefully trap common errors. | |
3968 | * | |
3969 | * If the console suddenly disappear at the time our children will really it | |
3970 | * then they will simply fail to acquire it and a positive answer will be | |
3971 | * assumed. New children will fallback to /dev/console though. | |
3972 | * | |
3973 | * Note: TTYs are devices that can come and go any time, and frequently aren't | |
3974 | * available yet during early boot (consider a USB rs232 dongle...). If for any | |
3975 | * reason the configured console is not ready, we fallback to the default | |
3976 | * console. */ | |
3977 | ||
3978 | if (!vc || path_equal(vc, "/dev/console")) | |
3979 | return vc; | |
3980 | ||
3981 | r = stat(vc, &st); | |
3982 | if (r < 0) | |
3983 | goto fail; | |
3984 | ||
3985 | if (!S_ISCHR(st.st_mode)) { | |
3986 | errno = ENOTTY; | |
3987 | goto fail; | |
3988 | } | |
3989 | ||
3990 | last_errno = 0; | |
3991 | return vc; | |
3992 | fail: | |
3993 | if (last_errno != errno) { | |
3994 | last_errno = errno; | |
3995 | log_warning_errno(errno, "Failed to open %s: %m, using default console", vc); | |
3996 | } | |
3997 | return "/dev/console"; | |
3998 | } | |
3999 | ||
e2680723 LP |
4000 | void manager_set_first_boot(Manager *m, bool b) { |
4001 | assert(m); | |
4002 | ||
463d0d15 | 4003 | if (!MANAGER_IS_SYSTEM(m)) |
e2680723 LP |
4004 | return; |
4005 | ||
ae2a2c53 LP |
4006 | if (m->first_boot != (int) b) { |
4007 | if (b) | |
4008 | (void) touch("/run/systemd/first-boot"); | |
4009 | else | |
4010 | (void) unlink("/run/systemd/first-boot"); | |
4011 | } | |
e2680723 | 4012 | |
ae2a2c53 | 4013 | m->first_boot = b; |
e2680723 LP |
4014 | } |
4015 | ||
b0eb2944 FB |
4016 | void manager_disable_confirm_spawn(void) { |
4017 | (void) touch("/run/systemd/confirm_spawn_disabled"); | |
4018 | } | |
4019 | ||
4020 | bool manager_is_confirm_spawn_disabled(Manager *m) { | |
4021 | if (!m->confirm_spawn) | |
4022 | return true; | |
4023 | ||
4024 | return access("/run/systemd/confirm_spawn_disabled", F_OK) >= 0; | |
4025 | } | |
4026 | ||
127d5fd1 | 4027 | void manager_status_printf(Manager *m, StatusType type, const char *status, const char *format, ...) { |
25cee550 MS |
4028 | va_list ap; |
4029 | ||
cb6531be ZJS |
4030 | /* If m is NULL, assume we're after shutdown and let the messages through. */ |
4031 | ||
4032 | if (m && !manager_get_show_status(m, type)) | |
25cee550 MS |
4033 | return; |
4034 | ||
03b717a3 MS |
4035 | /* XXX We should totally drop the check for ephemeral here |
4036 | * and thus effectively make 'Type=idle' pointless. */ | |
cb6531be | 4037 | if (type == STATUS_TYPE_EPHEMERAL && m && m->n_on_console > 0) |
03b717a3 MS |
4038 | return; |
4039 | ||
25cee550 | 4040 | va_start(ap, format); |
127d5fd1 | 4041 | status_vprintf(status, true, type == STATUS_TYPE_EPHEMERAL, format, ap); |
25cee550 MS |
4042 | va_end(ap); |
4043 | } | |
4044 | ||
a57f7e2c LP |
4045 | Set *manager_get_units_requiring_mounts_for(Manager *m, const char *path) { |
4046 | char p[strlen(path)+1]; | |
4047 | ||
4048 | assert(m); | |
4049 | assert(path); | |
4050 | ||
4051 | strcpy(p, path); | |
858d36c1 | 4052 | path_simplify(p, false); |
a57f7e2c LP |
4053 | |
4054 | return hashmap_get(m->units_requiring_mounts_for, streq(p, "/") ? "" : p); | |
4055 | } | |
e66cf1a3 | 4056 | |
5269eb6b | 4057 | int manager_update_failed_units(Manager *m, Unit *u, bool failed) { |
03455c28 | 4058 | unsigned size; |
5269eb6b | 4059 | int r; |
03455c28 LDM |
4060 | |
4061 | assert(m); | |
4062 | assert(u->manager == m); | |
4063 | ||
4064 | size = set_size(m->failed_units); | |
4065 | ||
9fff8981 | 4066 | if (failed) { |
5269eb6b LP |
4067 | r = set_ensure_allocated(&m->failed_units, NULL); |
4068 | if (r < 0) | |
4069 | return log_oom(); | |
4070 | ||
9fff8981 | 4071 | if (set_put(m->failed_units, u) < 0) |
5269eb6b | 4072 | return log_oom(); |
9fff8981 | 4073 | } else |
5269eb6b | 4074 | (void) set_remove(m->failed_units, u); |
03455c28 LDM |
4075 | |
4076 | if (set_size(m->failed_units) != size) | |
4077 | bus_manager_send_change_signal(m); | |
5269eb6b LP |
4078 | |
4079 | return 0; | |
03455c28 LDM |
4080 | } |
4081 | ||
f755e3b7 LP |
4082 | ManagerState manager_state(Manager *m) { |
4083 | Unit *u; | |
4084 | ||
4085 | assert(m); | |
4086 | ||
4087 | /* Did we ever finish booting? If not then we are still starting up */ | |
49d5666c | 4088 | if (!MANAGER_IS_FINISHED(m)) { |
d81afec1 LP |
4089 | |
4090 | u = manager_get_unit(m, SPECIAL_BASIC_TARGET); | |
4091 | if (!u || !UNIT_IS_ACTIVE_OR_RELOADING(unit_active_state(u))) | |
4092 | return MANAGER_INITIALIZING; | |
4093 | ||
f755e3b7 | 4094 | return MANAGER_STARTING; |
d81afec1 | 4095 | } |
f755e3b7 | 4096 | |
e68537f0 | 4097 | /* Is the special shutdown target active or queued? If so, we are in shutdown state */ |
f755e3b7 | 4098 | u = manager_get_unit(m, SPECIAL_SHUTDOWN_TARGET); |
706424c2 | 4099 | if (u && unit_active_or_pending(u)) |
f755e3b7 LP |
4100 | return MANAGER_STOPPING; |
4101 | ||
45a7b16b LP |
4102 | if (MANAGER_IS_SYSTEM(m)) { |
4103 | /* Are the rescue or emergency targets active or queued? If so we are in maintenance state */ | |
4104 | u = manager_get_unit(m, SPECIAL_RESCUE_TARGET); | |
4105 | if (u && unit_active_or_pending(u)) | |
4106 | return MANAGER_MAINTENANCE; | |
f755e3b7 | 4107 | |
45a7b16b LP |
4108 | u = manager_get_unit(m, SPECIAL_EMERGENCY_TARGET); |
4109 | if (u && unit_active_or_pending(u)) | |
4110 | return MANAGER_MAINTENANCE; | |
4111 | } | |
f755e3b7 LP |
4112 | |
4113 | /* Are there any failed units? If so, we are in degraded mode */ | |
4114 | if (set_size(m->failed_units) > 0) | |
4115 | return MANAGER_DEGRADED; | |
4116 | ||
4117 | return MANAGER_RUNNING; | |
4118 | } | |
4119 | ||
00d9ef85 LP |
4120 | #define DESTROY_IPC_FLAG (UINT32_C(1) << 31) |
4121 | ||
4122 | static void manager_unref_uid_internal( | |
4123 | Manager *m, | |
4124 | Hashmap **uid_refs, | |
4125 | uid_t uid, | |
4126 | bool destroy_now, | |
4127 | int (*_clean_ipc)(uid_t uid)) { | |
4128 | ||
4129 | uint32_t c, n; | |
4130 | ||
4131 | assert(m); | |
4132 | assert(uid_refs); | |
4133 | assert(uid_is_valid(uid)); | |
4134 | assert(_clean_ipc); | |
4135 | ||
4136 | /* A generic implementation, covering both manager_unref_uid() and manager_unref_gid(), under the assumption | |
4137 | * that uid_t and gid_t are actually defined the same way, with the same validity rules. | |
4138 | * | |
4139 | * We store a hashmap where the UID/GID is they key and the value is a 32bit reference counter, whose highest | |
4140 | * bit is used as flag for marking UIDs/GIDs whose IPC objects to remove when the last reference to the UID/GID | |
4141 | * is dropped. The flag is set to on, once at least one reference from a unit where RemoveIPC= is set is added | |
4142 | * on a UID/GID. It is reset when the UID's/GID's reference counter drops to 0 again. */ | |
4143 | ||
4144 | assert_cc(sizeof(uid_t) == sizeof(gid_t)); | |
4145 | assert_cc(UID_INVALID == (uid_t) GID_INVALID); | |
4146 | ||
4147 | if (uid == 0) /* We don't keep track of root, and will never destroy it */ | |
4148 | return; | |
4149 | ||
4150 | c = PTR_TO_UINT32(hashmap_get(*uid_refs, UID_TO_PTR(uid))); | |
4151 | ||
4152 | n = c & ~DESTROY_IPC_FLAG; | |
4153 | assert(n > 0); | |
4154 | n--; | |
4155 | ||
4156 | if (destroy_now && n == 0) { | |
4157 | hashmap_remove(*uid_refs, UID_TO_PTR(uid)); | |
4158 | ||
4159 | if (c & DESTROY_IPC_FLAG) { | |
4160 | log_debug("%s " UID_FMT " is no longer referenced, cleaning up its IPC.", | |
4161 | _clean_ipc == clean_ipc_by_uid ? "UID" : "GID", | |
4162 | uid); | |
4163 | (void) _clean_ipc(uid); | |
4164 | } | |
4165 | } else { | |
4166 | c = n | (c & DESTROY_IPC_FLAG); | |
4167 | assert_se(hashmap_update(*uid_refs, UID_TO_PTR(uid), UINT32_TO_PTR(c)) >= 0); | |
4168 | } | |
4169 | } | |
4170 | ||
4171 | void manager_unref_uid(Manager *m, uid_t uid, bool destroy_now) { | |
4172 | manager_unref_uid_internal(m, &m->uid_refs, uid, destroy_now, clean_ipc_by_uid); | |
4173 | } | |
4174 | ||
4175 | void manager_unref_gid(Manager *m, gid_t gid, bool destroy_now) { | |
4176 | manager_unref_uid_internal(m, &m->gid_refs, (uid_t) gid, destroy_now, clean_ipc_by_gid); | |
4177 | } | |
4178 | ||
4179 | static int manager_ref_uid_internal( | |
4180 | Manager *m, | |
4181 | Hashmap **uid_refs, | |
4182 | uid_t uid, | |
4183 | bool clean_ipc) { | |
4184 | ||
4185 | uint32_t c, n; | |
4186 | int r; | |
4187 | ||
4188 | assert(m); | |
4189 | assert(uid_refs); | |
4190 | assert(uid_is_valid(uid)); | |
4191 | ||
4192 | /* A generic implementation, covering both manager_ref_uid() and manager_ref_gid(), under the assumption | |
4193 | * that uid_t and gid_t are actually defined the same way, with the same validity rules. */ | |
4194 | ||
4195 | assert_cc(sizeof(uid_t) == sizeof(gid_t)); | |
4196 | assert_cc(UID_INVALID == (uid_t) GID_INVALID); | |
4197 | ||
4198 | if (uid == 0) /* We don't keep track of root, and will never destroy it */ | |
4199 | return 0; | |
4200 | ||
4201 | r = hashmap_ensure_allocated(uid_refs, &trivial_hash_ops); | |
4202 | if (r < 0) | |
4203 | return r; | |
4204 | ||
4205 | c = PTR_TO_UINT32(hashmap_get(*uid_refs, UID_TO_PTR(uid))); | |
4206 | ||
4207 | n = c & ~DESTROY_IPC_FLAG; | |
4208 | n++; | |
4209 | ||
4210 | if (n & DESTROY_IPC_FLAG) /* check for overflow */ | |
4211 | return -EOVERFLOW; | |
4212 | ||
4213 | c = n | (c & DESTROY_IPC_FLAG) | (clean_ipc ? DESTROY_IPC_FLAG : 0); | |
4214 | ||
4215 | return hashmap_replace(*uid_refs, UID_TO_PTR(uid), UINT32_TO_PTR(c)); | |
4216 | } | |
4217 | ||
4218 | int manager_ref_uid(Manager *m, uid_t uid, bool clean_ipc) { | |
4219 | return manager_ref_uid_internal(m, &m->uid_refs, uid, clean_ipc); | |
4220 | } | |
4221 | ||
4222 | int manager_ref_gid(Manager *m, gid_t gid, bool clean_ipc) { | |
4223 | return manager_ref_uid_internal(m, &m->gid_refs, (uid_t) gid, clean_ipc); | |
4224 | } | |
4225 | ||
4226 | static void manager_vacuum_uid_refs_internal( | |
4227 | Manager *m, | |
4228 | Hashmap **uid_refs, | |
4229 | int (*_clean_ipc)(uid_t uid)) { | |
4230 | ||
4231 | Iterator i; | |
4232 | void *p, *k; | |
4233 | ||
4234 | assert(m); | |
4235 | assert(uid_refs); | |
4236 | assert(_clean_ipc); | |
4237 | ||
4238 | HASHMAP_FOREACH_KEY(p, k, *uid_refs, i) { | |
4239 | uint32_t c, n; | |
4240 | uid_t uid; | |
4241 | ||
4242 | uid = PTR_TO_UID(k); | |
4243 | c = PTR_TO_UINT32(p); | |
4244 | ||
4245 | n = c & ~DESTROY_IPC_FLAG; | |
4246 | if (n > 0) | |
4247 | continue; | |
4248 | ||
4249 | if (c & DESTROY_IPC_FLAG) { | |
4250 | log_debug("Found unreferenced %s " UID_FMT " after reload/reexec. Cleaning up.", | |
4251 | _clean_ipc == clean_ipc_by_uid ? "UID" : "GID", | |
4252 | uid); | |
4253 | (void) _clean_ipc(uid); | |
4254 | } | |
4255 | ||
4256 | assert_se(hashmap_remove(*uid_refs, k) == p); | |
4257 | } | |
4258 | } | |
4259 | ||
4260 | void manager_vacuum_uid_refs(Manager *m) { | |
4261 | manager_vacuum_uid_refs_internal(m, &m->uid_refs, clean_ipc_by_uid); | |
4262 | } | |
4263 | ||
4264 | void manager_vacuum_gid_refs(Manager *m) { | |
4265 | manager_vacuum_uid_refs_internal(m, &m->gid_refs, clean_ipc_by_gid); | |
4266 | } | |
4267 | ||
4268 | static void manager_serialize_uid_refs_internal( | |
4269 | Manager *m, | |
4270 | FILE *f, | |
4271 | Hashmap **uid_refs, | |
4272 | const char *field_name) { | |
4273 | ||
4274 | Iterator i; | |
4275 | void *p, *k; | |
4276 | ||
4277 | assert(m); | |
4278 | assert(f); | |
4279 | assert(uid_refs); | |
4280 | assert(field_name); | |
4281 | ||
4282 | /* Serialize the UID reference table. Or actually, just the IPC destruction flag of it, as the actual counter | |
4283 | * of it is better rebuild after a reload/reexec. */ | |
4284 | ||
4285 | HASHMAP_FOREACH_KEY(p, k, *uid_refs, i) { | |
4286 | uint32_t c; | |
4287 | uid_t uid; | |
4288 | ||
4289 | uid = PTR_TO_UID(k); | |
4290 | c = PTR_TO_UINT32(p); | |
4291 | ||
4292 | if (!(c & DESTROY_IPC_FLAG)) | |
4293 | continue; | |
4294 | ||
4295 | fprintf(f, "%s=" UID_FMT "\n", field_name, uid); | |
4296 | } | |
4297 | } | |
4298 | ||
4299 | void manager_serialize_uid_refs(Manager *m, FILE *f) { | |
4300 | manager_serialize_uid_refs_internal(m, f, &m->uid_refs, "destroy-ipc-uid"); | |
4301 | } | |
4302 | ||
4303 | void manager_serialize_gid_refs(Manager *m, FILE *f) { | |
4304 | manager_serialize_uid_refs_internal(m, f, &m->gid_refs, "destroy-ipc-gid"); | |
4305 | } | |
4306 | ||
4307 | static void manager_deserialize_uid_refs_one_internal( | |
4308 | Manager *m, | |
4309 | Hashmap** uid_refs, | |
4310 | const char *value) { | |
4311 | ||
4312 | uid_t uid; | |
4313 | uint32_t c; | |
4314 | int r; | |
4315 | ||
4316 | assert(m); | |
4317 | assert(uid_refs); | |
4318 | assert(value); | |
4319 | ||
4320 | r = parse_uid(value, &uid); | |
4321 | if (r < 0 || uid == 0) { | |
4322 | log_debug("Unable to parse UID reference serialization"); | |
4323 | return; | |
4324 | } | |
4325 | ||
4326 | r = hashmap_ensure_allocated(uid_refs, &trivial_hash_ops); | |
4327 | if (r < 0) { | |
4328 | log_oom(); | |
4329 | return; | |
4330 | } | |
4331 | ||
4332 | c = PTR_TO_UINT32(hashmap_get(*uid_refs, UID_TO_PTR(uid))); | |
4333 | if (c & DESTROY_IPC_FLAG) | |
4334 | return; | |
4335 | ||
4336 | c |= DESTROY_IPC_FLAG; | |
4337 | ||
4338 | r = hashmap_replace(*uid_refs, UID_TO_PTR(uid), UINT32_TO_PTR(c)); | |
4339 | if (r < 0) { | |
4340 | log_debug("Failed to add UID reference entry"); | |
4341 | return; | |
4342 | } | |
4343 | } | |
4344 | ||
4345 | void manager_deserialize_uid_refs_one(Manager *m, const char *value) { | |
4346 | manager_deserialize_uid_refs_one_internal(m, &m->uid_refs, value); | |
4347 | } | |
4348 | ||
4349 | void manager_deserialize_gid_refs_one(Manager *m, const char *value) { | |
4350 | manager_deserialize_uid_refs_one_internal(m, &m->gid_refs, value); | |
4351 | } | |
4352 | ||
4353 | int manager_dispatch_user_lookup_fd(sd_event_source *source, int fd, uint32_t revents, void *userdata) { | |
4354 | struct buffer { | |
4355 | uid_t uid; | |
4356 | gid_t gid; | |
4357 | char unit_name[UNIT_NAME_MAX+1]; | |
4358 | } _packed_ buffer; | |
4359 | ||
4360 | Manager *m = userdata; | |
4361 | ssize_t l; | |
4362 | size_t n; | |
4363 | Unit *u; | |
4364 | ||
4365 | assert_se(source); | |
4366 | assert_se(m); | |
4367 | ||
4368 | /* Invoked whenever a child process succeeded resolving its user/group to use and sent us the resulting UID/GID | |
4369 | * in a datagram. We parse the datagram here and pass it off to the unit, so that it can add a reference to the | |
4370 | * UID/GID so that it can destroy the UID/GID's IPC objects when the reference counter drops to 0. */ | |
4371 | ||
4372 | l = recv(fd, &buffer, sizeof(buffer), MSG_DONTWAIT); | |
4373 | if (l < 0) { | |
4c701096 | 4374 | if (IN_SET(errno, EINTR, EAGAIN)) |
00d9ef85 LP |
4375 | return 0; |
4376 | ||
4377 | return log_error_errno(errno, "Failed to read from user lookup fd: %m"); | |
4378 | } | |
4379 | ||
4380 | if ((size_t) l <= offsetof(struct buffer, unit_name)) { | |
4381 | log_warning("Received too short user lookup message, ignoring."); | |
4382 | return 0; | |
4383 | } | |
4384 | ||
4385 | if ((size_t) l > offsetof(struct buffer, unit_name) + UNIT_NAME_MAX) { | |
4386 | log_warning("Received too long user lookup message, ignoring."); | |
4387 | return 0; | |
4388 | } | |
4389 | ||
4390 | if (!uid_is_valid(buffer.uid) && !gid_is_valid(buffer.gid)) { | |
4391 | log_warning("Got user lookup message with invalid UID/GID pair, ignoring."); | |
4392 | return 0; | |
4393 | } | |
4394 | ||
4395 | n = (size_t) l - offsetof(struct buffer, unit_name); | |
4396 | if (memchr(buffer.unit_name, 0, n)) { | |
4397 | log_warning("Received lookup message with embedded NUL character, ignoring."); | |
4398 | return 0; | |
4399 | } | |
4400 | ||
4401 | buffer.unit_name[n] = 0; | |
4402 | u = manager_get_unit(m, buffer.unit_name); | |
4403 | if (!u) { | |
4404 | log_debug("Got user lookup message but unit doesn't exist, ignoring."); | |
4405 | return 0; | |
4406 | } | |
4407 | ||
4408 | log_unit_debug(u, "User lookup succeeded: uid=" UID_FMT " gid=" GID_FMT, buffer.uid, buffer.gid); | |
4409 | ||
4410 | unit_notify_user_lookup(u, buffer.uid, buffer.gid); | |
4411 | return 0; | |
4412 | } | |
4413 | ||
af6b0ecc | 4414 | char *manager_taint_string(Manager *m) { |
90d7464d | 4415 | _cleanup_free_ char *destination = NULL, *overflowuid = NULL, *overflowgid = NULL; |
af6b0ecc LP |
4416 | char *buf, *e; |
4417 | int r; | |
4418 | ||
198ce932 ZJS |
4419 | /* Returns a "taint string", e.g. "local-hwclock:var-run-bad". |
4420 | * Only things that are detected at runtime should be tagged | |
4421 | * here. For stuff that is set during compilation, emit a warning | |
4422 | * in the configuration phase. */ | |
4423 | ||
af6b0ecc LP |
4424 | assert(m); |
4425 | ||
4426 | buf = new(char, sizeof("split-usr:" | |
4427 | "cgroups-missing:" | |
4428 | "local-hwclock:" | |
4429 | "var-run-bad:" | |
90d7464d LP |
4430 | "overflowuid-not-65534:" |
4431 | "overflowgid-not-65534:")); | |
af6b0ecc LP |
4432 | if (!buf) |
4433 | return NULL; | |
4434 | ||
4435 | e = buf; | |
0fd402b0 | 4436 | buf[0] = 0; |
af6b0ecc LP |
4437 | |
4438 | if (m->taint_usr) | |
4439 | e = stpcpy(e, "split-usr:"); | |
4440 | ||
4441 | if (access("/proc/cgroups", F_OK) < 0) | |
4442 | e = stpcpy(e, "cgroups-missing:"); | |
4443 | ||
4444 | if (clock_is_localtime(NULL) > 0) | |
4445 | e = stpcpy(e, "local-hwclock:"); | |
4446 | ||
4447 | r = readlink_malloc("/var/run", &destination); | |
4448 | if (r < 0 || !PATH_IN_SET(destination, "../run", "/run")) | |
4449 | e = stpcpy(e, "var-run-bad:"); | |
4450 | ||
90d7464d LP |
4451 | r = read_one_line_file("/proc/sys/kernel/overflowuid", &overflowuid); |
4452 | if (r >= 0 && !streq(overflowuid, "65534")) | |
4453 | e = stpcpy(e, "overflowuid-not-65534:"); | |
4454 | ||
4455 | r = read_one_line_file("/proc/sys/kernel/overflowgid", &overflowgid); | |
4456 | if (r >= 0 && !streq(overflowgid, "65534")) | |
4457 | e = stpcpy(e, "overflowgid-not-65534:"); | |
4458 | ||
af6b0ecc LP |
4459 | /* remove the last ':' */ |
4460 | if (e != buf) | |
4461 | e[-1] = 0; | |
4462 | ||
4463 | return buf; | |
4464 | } | |
4465 | ||
adefcf28 LP |
4466 | void manager_ref_console(Manager *m) { |
4467 | assert(m); | |
4468 | ||
4469 | m->n_on_console++; | |
4470 | } | |
4471 | ||
4472 | void manager_unref_console(Manager *m) { | |
4473 | ||
4474 | assert(m->n_on_console > 0); | |
4475 | m->n_on_console--; | |
4476 | ||
4477 | if (m->n_on_console == 0) | |
4478 | m->no_console_output = false; /* unset no_console_output flag, since the console is definitely free now */ | |
4479 | } | |
4480 | ||
a6ecbf83 FB |
4481 | void manager_override_log_level(Manager *m, int level) { |
4482 | _cleanup_free_ char *s = NULL; | |
4483 | assert(m); | |
4484 | ||
4485 | if (!m->log_level_overridden) { | |
4486 | m->original_log_level = log_get_max_level(); | |
4487 | m->log_level_overridden = true; | |
4488 | } | |
4489 | ||
4490 | (void) log_level_to_string_alloc(level, &s); | |
4491 | log_info("Setting log level to %s.", strna(s)); | |
4492 | ||
4493 | log_set_max_level(level); | |
4494 | } | |
4495 | ||
4496 | void manager_restore_original_log_level(Manager *m) { | |
4497 | _cleanup_free_ char *s = NULL; | |
4498 | assert(m); | |
4499 | ||
4500 | if (!m->log_level_overridden) | |
4501 | return; | |
4502 | ||
4503 | (void) log_level_to_string_alloc(m->original_log_level, &s); | |
4504 | log_info("Restoring log level to original (%s).", strna(s)); | |
4505 | ||
4506 | log_set_max_level(m->original_log_level); | |
4507 | m->log_level_overridden = false; | |
4508 | } | |
4509 | ||
bda7d78b FB |
4510 | void manager_override_log_target(Manager *m, LogTarget target) { |
4511 | assert(m); | |
4512 | ||
4513 | if (!m->log_target_overridden) { | |
4514 | m->original_log_target = log_get_target(); | |
4515 | m->log_target_overridden = true; | |
4516 | } | |
4517 | ||
4518 | log_info("Setting log target to %s.", log_target_to_string(target)); | |
4519 | log_set_target(target); | |
4520 | } | |
4521 | ||
4522 | void manager_restore_original_log_target(Manager *m) { | |
4523 | assert(m); | |
4524 | ||
4525 | if (!m->log_target_overridden) | |
4526 | return; | |
4527 | ||
4528 | log_info("Restoring log target to original %s.", log_target_to_string(m->original_log_target)); | |
4529 | ||
4530 | log_set_target(m->original_log_target); | |
4531 | m->log_target_overridden = false; | |
4532 | } | |
4533 | ||
d4ee7bd8 YW |
4534 | ManagerTimestamp manager_timestamp_initrd_mangle(ManagerTimestamp s) { |
4535 | if (in_initrd() && | |
4536 | s >= MANAGER_TIMESTAMP_SECURITY_START && | |
4537 | s <= MANAGER_TIMESTAMP_UNITS_LOAD_FINISH) | |
4538 | return s - MANAGER_TIMESTAMP_SECURITY_START + MANAGER_TIMESTAMP_INITRD_SECURITY_START; | |
4539 | return s; | |
4540 | } | |
4541 | ||
f755e3b7 | 4542 | static const char *const manager_state_table[_MANAGER_STATE_MAX] = { |
d81afec1 | 4543 | [MANAGER_INITIALIZING] = "initializing", |
f755e3b7 LP |
4544 | [MANAGER_STARTING] = "starting", |
4545 | [MANAGER_RUNNING] = "running", | |
4546 | [MANAGER_DEGRADED] = "degraded", | |
4547 | [MANAGER_MAINTENANCE] = "maintenance", | |
4548 | [MANAGER_STOPPING] = "stopping", | |
4549 | }; | |
4550 | ||
4551 | DEFINE_STRING_TABLE_LOOKUP(manager_state, ManagerState); | |
9f9f0342 LP |
4552 | |
4553 | static const char *const manager_timestamp_table[_MANAGER_TIMESTAMP_MAX] = { | |
4554 | [MANAGER_TIMESTAMP_FIRMWARE] = "firmware", | |
4555 | [MANAGER_TIMESTAMP_LOADER] = "loader", | |
4556 | [MANAGER_TIMESTAMP_KERNEL] = "kernel", | |
4557 | [MANAGER_TIMESTAMP_INITRD] = "initrd", | |
4558 | [MANAGER_TIMESTAMP_USERSPACE] = "userspace", | |
4559 | [MANAGER_TIMESTAMP_FINISH] = "finish", | |
4560 | [MANAGER_TIMESTAMP_SECURITY_START] = "security-start", | |
4561 | [MANAGER_TIMESTAMP_SECURITY_FINISH] = "security-finish", | |
4562 | [MANAGER_TIMESTAMP_GENERATORS_START] = "generators-start", | |
4563 | [MANAGER_TIMESTAMP_GENERATORS_FINISH] = "generators-finish", | |
4564 | [MANAGER_TIMESTAMP_UNITS_LOAD_START] = "units-load-start", | |
4565 | [MANAGER_TIMESTAMP_UNITS_LOAD_FINISH] = "units-load-finish", | |
d4ee7bd8 YW |
4566 | [MANAGER_TIMESTAMP_INITRD_SECURITY_START] = "initrd-security-start", |
4567 | [MANAGER_TIMESTAMP_INITRD_SECURITY_FINISH] = "initrd-security-finish", | |
4568 | [MANAGER_TIMESTAMP_INITRD_GENERATORS_START] = "initrd-generators-start", | |
4569 | [MANAGER_TIMESTAMP_INITRD_GENERATORS_FINISH] = "initrd-generators-finish", | |
4570 | [MANAGER_TIMESTAMP_INITRD_UNITS_LOAD_START] = "initrd-units-load-start", | |
4571 | [MANAGER_TIMESTAMP_INITRD_UNITS_LOAD_FINISH] = "initrd-units-load-finish", | |
9f9f0342 LP |
4572 | }; |
4573 | ||
4574 | DEFINE_STRING_TABLE_LOOKUP(manager_timestamp, ManagerTimestamp); |