]>
Commit | Line | Data |
---|---|---|
5cb5a6ff LP |
1 | /*-*- Mode: C; c-basic-offset: 8 -*-*/ |
2 | ||
a7334b09 LP |
3 | /*** |
4 | This file is part of systemd. | |
5 | ||
6 | Copyright 2010 Lennart Poettering | |
7 | ||
8 | systemd is free software; you can redistribute it and/or modify it | |
9 | under the terms of the GNU General Public License as published by | |
10 | the Free Software Foundation; either version 2 of the License, or | |
11 | (at your option) any later version. | |
12 | ||
13 | systemd is distributed in the hope that it will be useful, but | |
14 | WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 | General Public License for more details. | |
17 | ||
18 | You should have received a copy of the GNU General Public License | |
19 | along with systemd; If not, see <http://www.gnu.org/licenses/>. | |
20 | ***/ | |
21 | ||
83c60c9f LP |
22 | #include <sys/types.h> |
23 | #include <sys/stat.h> | |
24 | #include <unistd.h> | |
25 | #include <errno.h> | |
26 | #include <fcntl.h> | |
f94ea366 | 27 | #include <sys/epoll.h> |
034c6ed7 | 28 | #include <signal.h> |
83c60c9f | 29 | |
87f0e418 | 30 | #include "unit.h" |
5cb5a6ff | 31 | #include "socket.h" |
83c60c9f | 32 | #include "log.h" |
23a177ef LP |
33 | #include "load-dropin.h" |
34 | #include "load-fragment.h" | |
83c60c9f | 35 | |
acbb0225 | 36 | static const UnitActiveState state_translation_table[_SOCKET_STATE_MAX] = { |
87f0e418 LP |
37 | [SOCKET_DEAD] = UNIT_INACTIVE, |
38 | [SOCKET_START_PRE] = UNIT_ACTIVATING, | |
39 | [SOCKET_START_POST] = UNIT_ACTIVATING, | |
40 | [SOCKET_LISTENING] = UNIT_ACTIVE, | |
41 | [SOCKET_RUNNING] = UNIT_ACTIVE, | |
42 | [SOCKET_STOP_PRE] = UNIT_DEACTIVATING, | |
43 | [SOCKET_STOP_PRE_SIGTERM] = UNIT_DEACTIVATING, | |
44 | [SOCKET_STOP_PRE_SIGKILL] = UNIT_DEACTIVATING, | |
45 | [SOCKET_STOP_POST] = UNIT_DEACTIVATING, | |
80876c20 LP |
46 | [SOCKET_FINAL_SIGTERM] = UNIT_DEACTIVATING, |
47 | [SOCKET_FINAL_SIGKILL] = UNIT_DEACTIVATING, | |
87f0e418 | 48 | [SOCKET_MAINTAINANCE] = UNIT_INACTIVE, |
83c60c9f | 49 | }; |
5cb5a6ff | 50 | |
acbb0225 LP |
51 | static const char* const state_string_table[_SOCKET_STATE_MAX] = { |
52 | [SOCKET_DEAD] = "dead", | |
53 | [SOCKET_START_PRE] = "start-pre", | |
54 | [SOCKET_START_POST] = "start-post", | |
55 | [SOCKET_LISTENING] = "listening", | |
56 | [SOCKET_RUNNING] = "running", | |
57 | [SOCKET_STOP_PRE] = "stop-pre", | |
58 | [SOCKET_STOP_PRE_SIGTERM] = "stop-pre-sigterm", | |
59 | [SOCKET_STOP_PRE_SIGKILL] = "stop-pre-sigkill", | |
60 | [SOCKET_STOP_POST] = "stop-post", | |
80876c20 LP |
61 | [SOCKET_FINAL_SIGTERM] = "final-sigterm", |
62 | [SOCKET_FINAL_SIGKILL] = "final-sigkill", | |
acbb0225 LP |
63 | [SOCKET_MAINTAINANCE] = "maintainance" |
64 | }; | |
65 | ||
5e94833f LP |
66 | static void socket_unwatch_control_pid(Socket *s) { |
67 | assert(s); | |
68 | ||
69 | if (s->control_pid <= 0) | |
70 | return; | |
71 | ||
72 | unit_unwatch_pid(UNIT(s), s->control_pid); | |
73 | s->control_pid = 0; | |
74 | } | |
75 | ||
87f0e418 LP |
76 | static void socket_done(Unit *u) { |
77 | Socket *s = SOCKET(u); | |
034c6ed7 LP |
78 | SocketPort *p; |
79 | ||
80 | assert(s); | |
81 | ||
82 | while ((p = s->ports)) { | |
83 | LIST_REMOVE(SocketPort, port, s->ports, p); | |
84 | ||
85 | if (p->fd >= 0) | |
86 | close_nointr(p->fd); | |
87 | free(p->path); | |
88 | free(p); | |
89 | } | |
90 | ||
91 | exec_context_done(&s->exec_context); | |
e537352b | 92 | exec_command_free_array(s->exec_command, _SOCKET_EXEC_COMMAND_MAX); |
034c6ed7 LP |
93 | s->control_command = NULL; |
94 | ||
5e94833f | 95 | socket_unwatch_control_pid(s); |
034c6ed7 LP |
96 | |
97 | s->service = NULL; | |
98 | ||
acbb0225 | 99 | free(s->bind_to_device); |
e537352b | 100 | s->bind_to_device = NULL; |
acbb0225 LP |
101 | |
102 | unit_unwatch_timer(u, &s->timer_watch); | |
5cb5a6ff LP |
103 | } |
104 | ||
e537352b | 105 | static void socket_init(Unit *u) { |
87f0e418 | 106 | Socket *s = SOCKET(u); |
44d8db9e | 107 | |
e537352b LP |
108 | assert(u); |
109 | assert(u->meta.load_state == UNIT_STUB); | |
44d8db9e LP |
110 | |
111 | s->state = 0; | |
acbb0225 | 112 | s->timer_watch.type = WATCH_INVALID; |
44d8db9e LP |
113 | s->bind_ipv6_only = false; |
114 | s->backlog = SOMAXCONN; | |
115 | s->timeout_usec = DEFAULT_TIMEOUT_USEC; | |
b5a0699f LP |
116 | s->directory_mode = 0755; |
117 | s->socket_mode = 0666; | |
50159e6a LP |
118 | s->kill_mode = 0; |
119 | s->failure = false; | |
120 | s->control_pid = 0; | |
e537352b | 121 | s->service = NULL; |
44d8db9e | 122 | exec_context_init(&s->exec_context); |
e537352b | 123 | } |
44d8db9e | 124 | |
e537352b LP |
125 | static int socket_load(Unit *u) { |
126 | Socket *s = SOCKET(u); | |
127 | int r; | |
44d8db9e | 128 | |
e537352b LP |
129 | assert(u); |
130 | assert(u->meta.load_state == UNIT_STUB); | |
44d8db9e | 131 | |
e537352b | 132 | if ((r = unit_load_fragment_and_dropin(u)) < 0) |
23a177ef | 133 | return r; |
44d8db9e | 134 | |
23a177ef | 135 | /* This is a new unit? Then let's add in some extras */ |
e537352b | 136 | if (u->meta.load_state == UNIT_LOADED) { |
44d8db9e | 137 | |
e537352b | 138 | if ((r = unit_load_related_unit(u, ".service", (Unit**) &s->service))) |
23a177ef LP |
139 | return r; |
140 | ||
141 | if ((r = unit_add_dependency(u, UNIT_BEFORE, UNIT(s->service))) < 0) | |
142 | return r; | |
44d8db9e | 143 | |
23a177ef LP |
144 | if ((r = unit_add_exec_dependencies(u, &s->exec_context)) < 0) |
145 | return r; | |
146 | ||
147 | if ((r = unit_add_default_cgroup(u)) < 0) | |
148 | return r; | |
149 | } | |
150 | ||
151 | return 0; | |
44d8db9e LP |
152 | } |
153 | ||
542563ba LP |
154 | static const char* listen_lookup(int type) { |
155 | ||
156 | if (type == SOCK_STREAM) | |
157 | return "ListenStream"; | |
158 | else if (type == SOCK_DGRAM) | |
159 | return "ListenDatagram"; | |
160 | else if (type == SOCK_SEQPACKET) | |
161 | return "ListenSequentialPacket"; | |
162 | ||
034c6ed7 | 163 | assert_not_reached("Unknown socket type"); |
542563ba LP |
164 | return NULL; |
165 | } | |
166 | ||
87f0e418 | 167 | static void socket_dump(Unit *u, FILE *f, const char *prefix) { |
5cb5a6ff | 168 | |
e537352b | 169 | static const char* const command_table[_SOCKET_EXEC_COMMAND_MAX] = { |
5cb5a6ff LP |
170 | [SOCKET_EXEC_START_PRE] = "StartPre", |
171 | [SOCKET_EXEC_START_POST] = "StartPost", | |
172 | [SOCKET_EXEC_STOP_PRE] = "StopPre", | |
173 | [SOCKET_EXEC_STOP_POST] = "StopPost" | |
174 | }; | |
175 | ||
176 | SocketExecCommand c; | |
87f0e418 | 177 | Socket *s = SOCKET(u); |
542563ba | 178 | SocketPort *p; |
82ba9f08 LP |
179 | const char *prefix2; |
180 | char *p2; | |
5cb5a6ff LP |
181 | |
182 | assert(s); | |
fa068367 | 183 | assert(f); |
5cb5a6ff | 184 | |
82ba9f08 LP |
185 | p2 = strappend(prefix, "\t"); |
186 | prefix2 = p2 ? p2 : prefix; | |
c43d20a0 | 187 | |
5cb5a6ff LP |
188 | fprintf(f, |
189 | "%sSocket State: %s\n" | |
542563ba | 190 | "%sBindIPv6Only: %s\n" |
b5a0699f | 191 | "%sBacklog: %u\n" |
50159e6a | 192 | "%sKillMode: %s\n" |
b5a0699f LP |
193 | "%sSocketMode: %04o\n" |
194 | "%sDirectoryMode: %04o\n", | |
acbb0225 | 195 | prefix, state_string_table[s->state], |
542563ba | 196 | prefix, yes_no(s->bind_ipv6_only), |
b5a0699f | 197 | prefix, s->backlog, |
50159e6a | 198 | prefix, kill_mode_to_string(s->kill_mode), |
b5a0699f LP |
199 | prefix, s->socket_mode, |
200 | prefix, s->directory_mode); | |
542563ba | 201 | |
70123e68 LP |
202 | if (s->control_pid > 0) |
203 | fprintf(f, | |
204 | "%sControl PID: %llu\n", | |
205 | prefix, (unsigned long long) s->control_pid); | |
206 | ||
acbb0225 LP |
207 | if (s->bind_to_device) |
208 | fprintf(f, | |
209 | "%sBindToDevice: %s\n", | |
210 | prefix, s->bind_to_device); | |
211 | ||
034c6ed7 | 212 | LIST_FOREACH(port, p, s->ports) { |
5cb5a6ff | 213 | |
542563ba LP |
214 | if (p->type == SOCKET_SOCKET) { |
215 | const char *t; | |
216 | int r; | |
217 | char *k; | |
218 | ||
219 | if ((r = socket_address_print(&p->address, &k)) < 0) | |
220 | t = strerror(-r); | |
221 | else | |
222 | t = k; | |
223 | ||
224 | fprintf(f, "%s%s: %s\n", prefix, listen_lookup(p->address.type), k); | |
225 | free(k); | |
226 | } else | |
227 | fprintf(f, "%sListenFIFO: %s\n", prefix, p->path); | |
228 | } | |
5cb5a6ff LP |
229 | |
230 | exec_context_dump(&s->exec_context, f, prefix); | |
231 | ||
e537352b | 232 | for (c = 0; c < _SOCKET_EXEC_COMMAND_MAX; c++) { |
c43d20a0 LP |
233 | if (!s->exec_command[c]) |
234 | continue; | |
5cb5a6ff | 235 | |
c43d20a0 LP |
236 | fprintf(f, "%s→ %s:\n", |
237 | prefix, command_table[c]); | |
238 | ||
239 | exec_command_dump_list(s->exec_command[c], f, prefix2); | |
5cb5a6ff | 240 | } |
c43d20a0 | 241 | |
82ba9f08 | 242 | free(p2); |
5cb5a6ff LP |
243 | } |
244 | ||
034c6ed7 | 245 | static void socket_close_fds(Socket *s) { |
83c60c9f LP |
246 | SocketPort *p; |
247 | ||
248 | assert(s); | |
249 | ||
034c6ed7 | 250 | LIST_FOREACH(port, p, s->ports) { |
83c60c9f LP |
251 | if (p->fd < 0) |
252 | continue; | |
253 | ||
acbb0225 | 254 | unit_unwatch_fd(UNIT(s), &p->fd_watch); |
9152c765 LP |
255 | assert_se(close_nointr(p->fd) >= 0); |
256 | ||
83c60c9f LP |
257 | p->fd = -1; |
258 | } | |
259 | } | |
260 | ||
034c6ed7 | 261 | static int socket_open_fds(Socket *s) { |
83c60c9f LP |
262 | SocketPort *p; |
263 | int r; | |
264 | ||
265 | assert(s); | |
266 | ||
034c6ed7 | 267 | LIST_FOREACH(port, p, s->ports) { |
83c60c9f | 268 | |
034c6ed7 LP |
269 | if (p->fd >= 0) |
270 | continue; | |
83c60c9f LP |
271 | |
272 | if (p->type == SOCKET_SOCKET) { | |
273 | ||
b5a0699f LP |
274 | if ((r = socket_address_listen( |
275 | &p->address, | |
276 | s->backlog, | |
277 | s->bind_ipv6_only, | |
278 | s->bind_to_device, | |
279 | s->directory_mode, | |
280 | s->socket_mode, | |
281 | &p->fd)) < 0) | |
83c60c9f LP |
282 | goto rollback; |
283 | ||
284 | } else { | |
285 | struct stat st; | |
286 | assert(p->type == SOCKET_FIFO); | |
287 | ||
8cb45bf8 LP |
288 | mkdir_parents(p->path, s->directory_mode); |
289 | ||
290 | if (mkfifo(p->path, s->socket_mode) < 0 && errno != EEXIST) { | |
83c60c9f LP |
291 | r = -errno; |
292 | goto rollback; | |
293 | } | |
294 | ||
295 | if ((p->fd = open(p->path, O_RDWR|O_CLOEXEC|O_NOCTTY|O_NONBLOCK|O_NOFOLLOW)) < 0) { | |
296 | r = -errno; | |
297 | goto rollback; | |
298 | } | |
299 | ||
300 | if (fstat(p->fd, &st) < 0) { | |
301 | r = -errno; | |
302 | goto rollback; | |
303 | } | |
304 | ||
305 | /* FIXME verify user, access mode */ | |
306 | ||
307 | if (!S_ISFIFO(st.st_mode)) { | |
308 | r = -EEXIST; | |
309 | goto rollback; | |
310 | } | |
311 | } | |
034c6ed7 LP |
312 | } |
313 | ||
314 | return 0; | |
315 | ||
316 | rollback: | |
317 | socket_close_fds(s); | |
318 | return r; | |
319 | } | |
320 | ||
321 | static void socket_unwatch_fds(Socket *s) { | |
322 | SocketPort *p; | |
9152c765 | 323 | |
034c6ed7 LP |
324 | assert(s); |
325 | ||
326 | LIST_FOREACH(port, p, s->ports) { | |
327 | if (p->fd < 0) | |
328 | continue; | |
329 | ||
acbb0225 | 330 | unit_unwatch_fd(UNIT(s), &p->fd_watch); |
83c60c9f | 331 | } |
034c6ed7 LP |
332 | } |
333 | ||
334 | static int socket_watch_fds(Socket *s) { | |
335 | SocketPort *p; | |
336 | int r; | |
337 | ||
338 | assert(s); | |
83c60c9f | 339 | |
034c6ed7 LP |
340 | LIST_FOREACH(port, p, s->ports) { |
341 | if (p->fd < 0) | |
342 | continue; | |
343 | ||
f94ea366 | 344 | if ((r = unit_watch_fd(UNIT(s), p->fd, EPOLLIN, &p->fd_watch)) < 0) |
034c6ed7 LP |
345 | goto fail; |
346 | } | |
83c60c9f | 347 | |
542563ba | 348 | return 0; |
83c60c9f | 349 | |
034c6ed7 LP |
350 | fail: |
351 | socket_unwatch_fds(s); | |
352 | return r; | |
353 | } | |
354 | ||
355 | static void socket_set_state(Socket *s, SocketState state) { | |
356 | SocketState old_state; | |
357 | assert(s); | |
358 | ||
359 | old_state = s->state; | |
360 | s->state = state; | |
361 | ||
362 | if (state != SOCKET_START_PRE && | |
363 | state != SOCKET_START_POST && | |
364 | state != SOCKET_STOP_PRE && | |
365 | state != SOCKET_STOP_PRE_SIGTERM && | |
366 | state != SOCKET_STOP_PRE_SIGKILL && | |
367 | state != SOCKET_STOP_POST && | |
80876c20 LP |
368 | state != SOCKET_FINAL_SIGTERM && |
369 | state != SOCKET_FINAL_SIGKILL) { | |
acbb0225 | 370 | unit_unwatch_timer(UNIT(s), &s->timer_watch); |
5e94833f | 371 | socket_unwatch_control_pid(s); |
034c6ed7 | 372 | s->control_command = NULL; |
e537352b | 373 | } |
034c6ed7 LP |
374 | |
375 | if (state != SOCKET_START_POST && | |
376 | state != SOCKET_LISTENING && | |
377 | state != SOCKET_RUNNING && | |
378 | state != SOCKET_STOP_PRE && | |
379 | state != SOCKET_STOP_PRE_SIGTERM && | |
380 | state != SOCKET_STOP_PRE_SIGKILL) | |
381 | socket_close_fds(s); | |
382 | ||
383 | if (state != SOCKET_LISTENING) | |
384 | socket_unwatch_fds(s); | |
385 | ||
e537352b LP |
386 | if (state != old_state) |
387 | log_debug("%s changed %s → %s", unit_id(UNIT(s)), state_string_table[old_state], state_string_table[state]); | |
acbb0225 LP |
388 | |
389 | unit_notify(UNIT(s), state_translation_table[old_state], state_translation_table[state]); | |
034c6ed7 LP |
390 | } |
391 | ||
e537352b | 392 | static int socket_spawn(Socket *s, ExecCommand *c, pid_t *_pid) { |
034c6ed7 LP |
393 | pid_t pid; |
394 | int r; | |
395 | ||
396 | assert(s); | |
397 | assert(c); | |
398 | assert(_pid); | |
399 | ||
e537352b LP |
400 | if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0) |
401 | goto fail; | |
034c6ed7 | 402 | |
8e274523 LP |
403 | if ((r = exec_spawn(c, |
404 | &s->exec_context, | |
405 | NULL, 0, | |
406 | true, | |
407 | true, | |
80876c20 | 408 | UNIT(s)->meta.manager->confirm_spawn, |
8e274523 LP |
409 | UNIT(s)->meta.cgroup_bondings, |
410 | &pid)) < 0) | |
034c6ed7 LP |
411 | goto fail; |
412 | ||
87f0e418 | 413 | if ((r = unit_watch_pid(UNIT(s), pid)) < 0) |
034c6ed7 LP |
414 | /* FIXME: we need to do something here */ |
415 | goto fail; | |
83c60c9f | 416 | |
034c6ed7 LP |
417 | *_pid = pid; |
418 | ||
419 | return 0; | |
420 | ||
421 | fail: | |
e537352b | 422 | unit_unwatch_timer(UNIT(s), &s->timer_watch); |
83c60c9f LP |
423 | |
424 | return r; | |
542563ba LP |
425 | } |
426 | ||
034c6ed7 LP |
427 | static void socket_enter_dead(Socket *s, bool success) { |
428 | assert(s); | |
429 | ||
430 | if (!success) | |
431 | s->failure = true; | |
432 | ||
433 | socket_set_state(s, s->failure ? SOCKET_MAINTAINANCE : SOCKET_DEAD); | |
434 | } | |
435 | ||
80876c20 LP |
436 | static void socket_enter_signal(Socket *s, SocketState state, bool success); |
437 | ||
034c6ed7 LP |
438 | static void socket_enter_stop_post(Socket *s, bool success) { |
439 | int r; | |
440 | assert(s); | |
441 | ||
442 | if (!success) | |
443 | s->failure = true; | |
444 | ||
5e94833f LP |
445 | socket_unwatch_control_pid(s); |
446 | ||
80876c20 | 447 | if ((s->control_command = s->exec_command[SOCKET_EXEC_STOP_POST])) { |
e537352b | 448 | if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0) |
034c6ed7 LP |
449 | goto fail; |
450 | ||
80876c20 LP |
451 | socket_set_state(s, SOCKET_STOP_POST); |
452 | } else | |
453 | socket_enter_signal(s, SOCKET_FINAL_SIGTERM, true); | |
034c6ed7 LP |
454 | |
455 | return; | |
456 | ||
457 | fail: | |
87f0e418 | 458 | log_warning("%s failed to run stop-post executable: %s", unit_id(UNIT(s)), strerror(-r)); |
80876c20 | 459 | socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false); |
034c6ed7 LP |
460 | } |
461 | ||
462 | static void socket_enter_signal(Socket *s, SocketState state, bool success) { | |
463 | int r; | |
80876c20 | 464 | bool sent = false; |
034c6ed7 LP |
465 | |
466 | assert(s); | |
467 | ||
468 | if (!success) | |
469 | s->failure = true; | |
470 | ||
80876c20 LP |
471 | if (s->kill_mode != KILL_NONE) { |
472 | int sig = (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_FINAL_SIGTERM) ? SIGTERM : SIGKILL; | |
034c6ed7 | 473 | |
50159e6a LP |
474 | if (s->kill_mode == KILL_CONTROL_GROUP) { |
475 | ||
476 | if ((r = cgroup_bonding_kill_list(UNIT(s)->meta.cgroup_bondings, sig)) < 0) { | |
477 | if (r != -EAGAIN && r != -ESRCH) | |
478 | goto fail; | |
479 | } else | |
480 | sent = true; | |
034c6ed7 | 481 | } |
50159e6a | 482 | |
80876c20 | 483 | if (!sent && s->control_pid > 0) |
50159e6a LP |
484 | if (kill(s->kill_mode == KILL_PROCESS ? s->control_pid : -s->control_pid, sig) < 0 && errno != ESRCH) { |
485 | r = -errno; | |
486 | goto fail; | |
487 | } | |
d6ea93e3 | 488 | } |
034c6ed7 | 489 | |
80876c20 LP |
490 | if (sent) { |
491 | if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0) | |
492 | goto fail; | |
d6ea93e3 | 493 | |
80876c20 LP |
494 | socket_set_state(s, state); |
495 | } else if (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_STOP_PRE_SIGKILL) | |
496 | socket_enter_stop_post(s, true); | |
497 | else | |
034c6ed7 LP |
498 | socket_enter_dead(s, true); |
499 | ||
500 | return; | |
501 | ||
502 | fail: | |
87f0e418 | 503 | log_warning("%s failed to kill processes: %s", unit_id(UNIT(s)), strerror(-r)); |
034c6ed7 LP |
504 | |
505 | if (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_STOP_PRE_SIGKILL) | |
506 | socket_enter_stop_post(s, false); | |
507 | else | |
508 | socket_enter_dead(s, false); | |
509 | } | |
510 | ||
511 | static void socket_enter_stop_pre(Socket *s, bool success) { | |
512 | int r; | |
513 | assert(s); | |
514 | ||
515 | if (!success) | |
516 | s->failure = true; | |
517 | ||
5e94833f LP |
518 | socket_unwatch_control_pid(s); |
519 | ||
80876c20 | 520 | if ((s->control_command = s->exec_command[SOCKET_EXEC_STOP_PRE])) { |
e537352b | 521 | if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0) |
034c6ed7 LP |
522 | goto fail; |
523 | ||
80876c20 LP |
524 | socket_set_state(s, SOCKET_STOP_PRE); |
525 | } else | |
034c6ed7 LP |
526 | socket_enter_stop_post(s, true); |
527 | ||
528 | return; | |
529 | ||
530 | fail: | |
87f0e418 | 531 | log_warning("%s failed to run stop-pre executable: %s", unit_id(UNIT(s)), strerror(-r)); |
034c6ed7 LP |
532 | socket_enter_stop_post(s, false); |
533 | } | |
534 | ||
e9af15c3 LP |
535 | static void socket_enter_listening(Socket *s) { |
536 | int r; | |
537 | assert(s); | |
538 | ||
539 | if ((r = socket_watch_fds(s)) < 0) { | |
540 | log_warning("%s failed to watch sockets: %s", unit_id(UNIT(s)), strerror(-r)); | |
541 | goto fail; | |
542 | } | |
543 | ||
544 | socket_set_state(s, SOCKET_LISTENING); | |
545 | return; | |
546 | ||
547 | fail: | |
548 | socket_enter_stop_pre(s, false); | |
549 | } | |
550 | ||
034c6ed7 LP |
551 | static void socket_enter_start_post(Socket *s) { |
552 | int r; | |
553 | assert(s); | |
554 | ||
e9af15c3 | 555 | if ((r = socket_open_fds(s)) < 0) { |
87f0e418 | 556 | log_warning("%s failed to listen on sockets: %s", unit_id(UNIT(s)), strerror(-r)); |
034c6ed7 LP |
557 | goto fail; |
558 | } | |
559 | ||
5e94833f LP |
560 | socket_unwatch_control_pid(s); |
561 | ||
80876c20 | 562 | if ((s->control_command = s->exec_command[SOCKET_EXEC_START_POST])) { |
e537352b | 563 | if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0) { |
87f0e418 | 564 | log_warning("%s failed to run start-post executable: %s", unit_id(UNIT(s)), strerror(-r)); |
034c6ed7 LP |
565 | goto fail; |
566 | } | |
567 | ||
80876c20 LP |
568 | socket_set_state(s, SOCKET_START_POST); |
569 | } else | |
e9af15c3 | 570 | socket_enter_listening(s); |
034c6ed7 LP |
571 | |
572 | return; | |
573 | ||
574 | fail: | |
575 | socket_enter_stop_pre(s, false); | |
576 | } | |
577 | ||
578 | static void socket_enter_start_pre(Socket *s) { | |
579 | int r; | |
580 | assert(s); | |
581 | ||
5e94833f LP |
582 | socket_unwatch_control_pid(s); |
583 | ||
80876c20 | 584 | if ((s->control_command = s->exec_command[SOCKET_EXEC_START_PRE])) { |
e537352b | 585 | if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0) |
034c6ed7 LP |
586 | goto fail; |
587 | ||
80876c20 LP |
588 | socket_set_state(s, SOCKET_START_PRE); |
589 | } else | |
034c6ed7 LP |
590 | socket_enter_start_post(s); |
591 | ||
592 | return; | |
593 | ||
594 | fail: | |
87f0e418 | 595 | log_warning("%s failed to run start-pre exectuable: %s", unit_id(UNIT(s)), strerror(-r)); |
034c6ed7 LP |
596 | socket_enter_dead(s, false); |
597 | } | |
598 | ||
599 | static void socket_enter_running(Socket *s) { | |
600 | int r; | |
601 | ||
602 | assert(s); | |
603 | ||
87f0e418 | 604 | if ((r = manager_add_job(UNIT(s)->meta.manager, JOB_START, UNIT(s->service), JOB_REPLACE, true, NULL)) < 0) |
034c6ed7 LP |
605 | goto fail; |
606 | ||
607 | socket_set_state(s, SOCKET_RUNNING); | |
608 | return; | |
609 | ||
610 | fail: | |
87f0e418 | 611 | log_warning("%s failed to queue socket startup job: %s", unit_id(UNIT(s)), strerror(-r)); |
80876c20 | 612 | socket_enter_stop_pre(s, false); |
034c6ed7 LP |
613 | } |
614 | ||
615 | static void socket_run_next(Socket *s, bool success) { | |
616 | int r; | |
617 | ||
618 | assert(s); | |
619 | assert(s->control_command); | |
620 | assert(s->control_command->command_next); | |
621 | ||
622 | if (!success) | |
623 | s->failure = true; | |
624 | ||
5e94833f LP |
625 | socket_unwatch_control_pid(s); |
626 | ||
034c6ed7 LP |
627 | s->control_command = s->control_command->command_next; |
628 | ||
e537352b | 629 | if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0) |
034c6ed7 LP |
630 | goto fail; |
631 | ||
632 | return; | |
633 | ||
634 | fail: | |
80876c20 LP |
635 | log_warning("%s failed to run spawn next executable: %s", unit_id(UNIT(s)), strerror(-r)); |
636 | ||
637 | if (s->state == SOCKET_START_POST) | |
638 | socket_enter_stop_pre(s, false); | |
034c6ed7 LP |
639 | else if (s->state == SOCKET_STOP_POST) |
640 | socket_enter_dead(s, false); | |
641 | else | |
80876c20 | 642 | socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false); |
034c6ed7 LP |
643 | } |
644 | ||
87f0e418 LP |
645 | static int socket_start(Unit *u) { |
646 | Socket *s = SOCKET(u); | |
83c60c9f LP |
647 | |
648 | assert(s); | |
649 | ||
034c6ed7 LP |
650 | /* We cannot fulfill this request right now, try again later |
651 | * please! */ | |
652 | if (s->state == SOCKET_STOP_PRE || | |
653 | s->state == SOCKET_STOP_PRE_SIGKILL || | |
654 | s->state == SOCKET_STOP_PRE_SIGTERM || | |
655 | s->state == SOCKET_STOP_POST || | |
80876c20 LP |
656 | s->state == SOCKET_FINAL_SIGTERM || |
657 | s->state == SOCKET_FINAL_SIGKILL) | |
034c6ed7 LP |
658 | return -EAGAIN; |
659 | ||
83c60c9f LP |
660 | if (s->state == SOCKET_START_PRE || |
661 | s->state == SOCKET_START_POST) | |
034c6ed7 | 662 | return 0; |
83c60c9f | 663 | |
034c6ed7 | 664 | /* Cannot run this without the service being around */ |
87f0e418 | 665 | if (s->service->meta.load_state != UNIT_LOADED) |
034c6ed7 | 666 | return -ENOENT; |
83c60c9f | 667 | |
e537352b LP |
668 | /* If the service is alredy actvie we cannot start the |
669 | * socket */ | |
670 | if (s->service->state != SERVICE_DEAD && | |
671 | s->service->state != SERVICE_MAINTAINANCE && | |
672 | s->service->state != SERVICE_AUTO_RESTART) | |
673 | return -EBUSY; | |
674 | ||
034c6ed7 | 675 | assert(s->state == SOCKET_DEAD || s->state == SOCKET_MAINTAINANCE); |
83c60c9f | 676 | |
034c6ed7 LP |
677 | s->failure = false; |
678 | socket_enter_start_pre(s); | |
679 | return 0; | |
680 | } | |
83c60c9f | 681 | |
87f0e418 LP |
682 | static int socket_stop(Unit *u) { |
683 | Socket *s = SOCKET(u); | |
034c6ed7 LP |
684 | |
685 | assert(s); | |
686 | ||
687 | /* We cannot fulfill this request right now, try again later | |
688 | * please! */ | |
689 | if (s->state == SOCKET_START_PRE || | |
690 | s->state == SOCKET_START_POST) | |
691 | return -EAGAIN; | |
83c60c9f | 692 | |
e537352b LP |
693 | /* Already on it */ |
694 | if (s->state == SOCKET_STOP_PRE || | |
695 | s->state == SOCKET_STOP_PRE_SIGTERM || | |
696 | s->state == SOCKET_STOP_PRE_SIGKILL || | |
697 | s->state == SOCKET_STOP_POST || | |
80876c20 LP |
698 | s->state == SOCKET_FINAL_SIGTERM || |
699 | s->state == SOCKET_FINAL_SIGTERM) | |
e537352b LP |
700 | return 0; |
701 | ||
034c6ed7 | 702 | assert(s->state == SOCKET_LISTENING || s->state == SOCKET_RUNNING); |
83c60c9f | 703 | |
034c6ed7 | 704 | socket_enter_stop_pre(s, true); |
542563ba LP |
705 | return 0; |
706 | } | |
707 | ||
87f0e418 LP |
708 | static UnitActiveState socket_active_state(Unit *u) { |
709 | assert(u); | |
5cb5a6ff | 710 | |
acbb0225 | 711 | return state_translation_table[SOCKET(u)->state]; |
5cb5a6ff LP |
712 | } |
713 | ||
10a94420 LP |
714 | static const char *socket_sub_state_to_string(Unit *u) { |
715 | assert(u); | |
716 | ||
717 | return state_string_table[SOCKET(u)->state]; | |
718 | } | |
719 | ||
acbb0225 | 720 | static void socket_fd_event(Unit *u, int fd, uint32_t events, Watch *w) { |
87f0e418 | 721 | Socket *s = SOCKET(u); |
9152c765 | 722 | |
034c6ed7 | 723 | assert(s); |
9152c765 | 724 | |
e9af15c3 | 725 | log_debug("Incoming traffic on %s", unit_id(u)); |
9152c765 | 726 | |
f94ea366 | 727 | if (events != EPOLLIN) |
034c6ed7 | 728 | socket_enter_stop_pre(s, false); |
9152c765 | 729 | |
034c6ed7 | 730 | socket_enter_running(s); |
9152c765 LP |
731 | } |
732 | ||
87f0e418 LP |
733 | static void socket_sigchld_event(Unit *u, pid_t pid, int code, int status) { |
734 | Socket *s = SOCKET(u); | |
034c6ed7 | 735 | bool success; |
5cb5a6ff LP |
736 | |
737 | assert(s); | |
034c6ed7 | 738 | assert(pid >= 0); |
5cb5a6ff | 739 | |
bd982a8b | 740 | success = code == CLD_EXITED && status == 0; |
034c6ed7 | 741 | s->failure = s->failure || !success; |
542563ba | 742 | |
034c6ed7 LP |
743 | assert(s->control_pid == pid); |
744 | assert(s->control_command); | |
745 | ||
746 | exec_status_fill(&s->control_command->exec_status, pid, code, status); | |
747 | s->control_pid = 0; | |
748 | ||
94f04347 | 749 | log_debug("%s control process exited, code=%s status=%i", unit_id(u), sigchld_code_to_string(code), status); |
034c6ed7 | 750 | |
80876c20 LP |
751 | if (s->control_command->command_next && success) { |
752 | log_debug("%s running next command for state %s", unit_id(u), state_string_table[s->state]); | |
034c6ed7 | 753 | socket_run_next(s, success); |
acbb0225 | 754 | } else { |
034c6ed7 LP |
755 | /* No further commands for this step, so let's figure |
756 | * out what to do next */ | |
5cb5a6ff | 757 | |
bd982a8b | 758 | log_debug("%s got final SIGCHLD for state %s", unit_id(u), state_string_table[s->state]); |
acbb0225 | 759 | |
034c6ed7 LP |
760 | switch (s->state) { |
761 | ||
762 | case SOCKET_START_PRE: | |
763 | if (success) | |
acbb0225 | 764 | socket_enter_start_post(s); |
034c6ed7 | 765 | else |
80876c20 | 766 | socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false); |
034c6ed7 LP |
767 | break; |
768 | ||
769 | case SOCKET_START_POST: | |
770 | if (success) | |
e9af15c3 | 771 | socket_enter_listening(s); |
034c6ed7 LP |
772 | else |
773 | socket_enter_stop_pre(s, false); | |
774 | break; | |
775 | ||
776 | case SOCKET_STOP_PRE: | |
777 | case SOCKET_STOP_PRE_SIGTERM: | |
778 | case SOCKET_STOP_PRE_SIGKILL: | |
779 | socket_enter_stop_post(s, success); | |
780 | break; | |
781 | ||
782 | case SOCKET_STOP_POST: | |
80876c20 LP |
783 | case SOCKET_FINAL_SIGTERM: |
784 | case SOCKET_FINAL_SIGKILL: | |
034c6ed7 LP |
785 | socket_enter_dead(s, success); |
786 | break; | |
787 | ||
788 | default: | |
789 | assert_not_reached("Uh, control process died at wrong time."); | |
790 | } | |
791 | } | |
792 | } | |
5cb5a6ff | 793 | |
acbb0225 | 794 | static void socket_timer_event(Unit *u, uint64_t elapsed, Watch *w) { |
87f0e418 | 795 | Socket *s = SOCKET(u); |
5cb5a6ff | 796 | |
034c6ed7 LP |
797 | assert(s); |
798 | assert(elapsed == 1); | |
acbb0225 | 799 | assert(w == &s->timer_watch); |
034c6ed7 LP |
800 | |
801 | switch (s->state) { | |
802 | ||
803 | case SOCKET_START_PRE: | |
80876c20 LP |
804 | log_warning("%s starting timed out. Terminating.", unit_id(u)); |
805 | socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false); | |
806 | ||
034c6ed7 | 807 | case SOCKET_START_POST: |
e537352b | 808 | log_warning("%s starting timed out. Stopping.", unit_id(u)); |
034c6ed7 LP |
809 | socket_enter_stop_pre(s, false); |
810 | break; | |
811 | ||
812 | case SOCKET_STOP_PRE: | |
87f0e418 | 813 | log_warning("%s stopping timed out. Terminating.", unit_id(u)); |
034c6ed7 LP |
814 | socket_enter_signal(s, SOCKET_STOP_PRE_SIGTERM, false); |
815 | break; | |
816 | ||
817 | case SOCKET_STOP_PRE_SIGTERM: | |
87f0e418 | 818 | log_warning("%s stopping timed out. Killing.", unit_id(u)); |
034c6ed7 LP |
819 | socket_enter_signal(s, SOCKET_STOP_PRE_SIGKILL, false); |
820 | break; | |
821 | ||
822 | case SOCKET_STOP_PRE_SIGKILL: | |
87f0e418 | 823 | log_warning("%s still around after SIGKILL. Ignoring.", unit_id(u)); |
034c6ed7 LP |
824 | socket_enter_stop_post(s, false); |
825 | break; | |
826 | ||
827 | case SOCKET_STOP_POST: | |
87f0e418 | 828 | log_warning("%s stopping timed out (2). Terminating.", unit_id(u)); |
80876c20 | 829 | socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false); |
034c6ed7 LP |
830 | break; |
831 | ||
80876c20 | 832 | case SOCKET_FINAL_SIGTERM: |
87f0e418 | 833 | log_warning("%s stopping timed out (2). Killing.", unit_id(u)); |
80876c20 | 834 | socket_enter_signal(s, SOCKET_FINAL_SIGKILL, false); |
034c6ed7 LP |
835 | break; |
836 | ||
80876c20 | 837 | case SOCKET_FINAL_SIGKILL: |
87f0e418 | 838 | log_warning("%s still around after SIGKILL (2). Entering maintainance mode.", unit_id(u)); |
034c6ed7 LP |
839 | socket_enter_dead(s, false); |
840 | break; | |
841 | ||
842 | default: | |
843 | assert_not_reached("Timeout at wrong time."); | |
844 | } | |
5cb5a6ff LP |
845 | } |
846 | ||
44d8db9e LP |
847 | int socket_collect_fds(Socket *s, int **fds, unsigned *n_fds) { |
848 | int *rfds; | |
849 | unsigned rn_fds, k; | |
850 | SocketPort *p; | |
851 | ||
852 | assert(s); | |
853 | assert(fds); | |
854 | assert(n_fds); | |
855 | ||
856 | /* Called from the service code for requesting our fds */ | |
857 | ||
858 | rn_fds = 0; | |
859 | LIST_FOREACH(port, p, s->ports) | |
860 | if (p->fd >= 0) | |
861 | rn_fds++; | |
862 | ||
863 | if (!(rfds = new(int, rn_fds)) < 0) | |
864 | return -ENOMEM; | |
865 | ||
866 | k = 0; | |
867 | LIST_FOREACH(port, p, s->ports) | |
868 | if (p->fd >= 0) | |
869 | rfds[k++] = p->fd; | |
870 | ||
871 | assert(k == rn_fds); | |
872 | ||
873 | *fds = rfds; | |
874 | *n_fds = rn_fds; | |
875 | ||
876 | return 0; | |
877 | } | |
878 | ||
ceee3d82 LP |
879 | void socket_notify_service_dead(Socket *s) { |
880 | assert(s); | |
881 | ||
882 | /* The service is dead. Dang. */ | |
883 | ||
884 | if (s->state == SOCKET_RUNNING) { | |
885 | log_debug("%s got notified about service death.", unit_id(UNIT(s))); | |
886 | socket_enter_listening(s); | |
887 | } | |
888 | } | |
889 | ||
87f0e418 | 890 | const UnitVTable socket_vtable = { |
5cb5a6ff LP |
891 | .suffix = ".socket", |
892 | ||
034c6ed7 | 893 | .init = socket_init, |
e537352b | 894 | .load = socket_load, |
034c6ed7 LP |
895 | .done = socket_done, |
896 | ||
5cb5a6ff LP |
897 | .dump = socket_dump, |
898 | ||
542563ba LP |
899 | .start = socket_start, |
900 | .stop = socket_stop, | |
5cb5a6ff LP |
901 | |
902 | .active_state = socket_active_state, | |
10a94420 | 903 | .sub_state_to_string = socket_sub_state_to_string, |
5cb5a6ff | 904 | |
9152c765 | 905 | .fd_event = socket_fd_event, |
034c6ed7 LP |
906 | .sigchld_event = socket_sigchld_event, |
907 | .timer_event = socket_timer_event | |
5cb5a6ff | 908 | }; |