]> git.ipfire.org Git - thirdparty/systemd.git/blame - src/socket.c
dbus: include NextElapse field in timer properties
[thirdparty/systemd.git] / src / socket.c
CommitLineData
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>
4f2d528d 29#include <arpa/inet.h>
83c60c9f 30
87f0e418 31#include "unit.h"
5cb5a6ff 32#include "socket.h"
83c60c9f 33#include "log.h"
23a177ef
LP
34#include "load-dropin.h"
35#include "load-fragment.h"
9e2f7c11 36#include "strv.h"
4f2d528d 37#include "unit-name.h"
4139c1b2 38#include "dbus-socket.h"
4fd5948e 39#include "missing.h"
a40eb732 40#include "special.h"
83c60c9f 41
acbb0225 42static const UnitActiveState state_translation_table[_SOCKET_STATE_MAX] = {
87f0e418
LP
43 [SOCKET_DEAD] = UNIT_INACTIVE,
44 [SOCKET_START_PRE] = UNIT_ACTIVATING,
45 [SOCKET_START_POST] = UNIT_ACTIVATING,
46 [SOCKET_LISTENING] = UNIT_ACTIVE,
47 [SOCKET_RUNNING] = UNIT_ACTIVE,
48 [SOCKET_STOP_PRE] = UNIT_DEACTIVATING,
49 [SOCKET_STOP_PRE_SIGTERM] = UNIT_DEACTIVATING,
50 [SOCKET_STOP_PRE_SIGKILL] = UNIT_DEACTIVATING,
51 [SOCKET_STOP_POST] = UNIT_DEACTIVATING,
80876c20
LP
52 [SOCKET_FINAL_SIGTERM] = UNIT_DEACTIVATING,
53 [SOCKET_FINAL_SIGKILL] = UNIT_DEACTIVATING,
032ff4af 54 [SOCKET_MAINTENANCE] = UNIT_MAINTENANCE
83c60c9f 55};
5cb5a6ff 56
a16e1123
LP
57static void socket_init(Unit *u) {
58 Socket *s = SOCKET(u);
59
60 assert(u);
61 assert(u->meta.load_state == UNIT_STUB);
62
a16e1123
LP
63 s->backlog = SOMAXCONN;
64 s->timeout_usec = DEFAULT_TIMEOUT_USEC;
65 s->directory_mode = 0755;
66 s->socket_mode = 0666;
67
6cf6bbc2
LP
68 s->max_connections = 64;
69
4fd5948e 70 s->priority = -1;
4fd5948e
LP
71 s->ip_tos = -1;
72 s->ip_ttl = -1;
4fd5948e 73 s->mark = -1;
4fd5948e 74
a16e1123
LP
75 exec_context_init(&s->exec_context);
76
77 s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
78}
acbb0225 79
5e94833f
LP
80static void socket_unwatch_control_pid(Socket *s) {
81 assert(s);
82
83 if (s->control_pid <= 0)
84 return;
85
86 unit_unwatch_pid(UNIT(s), s->control_pid);
87 s->control_pid = 0;
88}
89
87f0e418
LP
90static void socket_done(Unit *u) {
91 Socket *s = SOCKET(u);
034c6ed7 92 SocketPort *p;
6cf6bbc2 93 Meta *i;
034c6ed7
LP
94
95 assert(s);
96
97 while ((p = s->ports)) {
98 LIST_REMOVE(SocketPort, port, s->ports, p);
99
a16e1123
LP
100 if (p->fd >= 0) {
101 unit_unwatch_fd(UNIT(s), &p->fd_watch);
102 close_nointr_nofail(p->fd);
103 }
104
034c6ed7
LP
105 free(p->path);
106 free(p);
107 }
108
109 exec_context_done(&s->exec_context);
e537352b 110 exec_command_free_array(s->exec_command, _SOCKET_EXEC_COMMAND_MAX);
034c6ed7
LP
111 s->control_command = NULL;
112
5e94833f 113 socket_unwatch_control_pid(s);
034c6ed7
LP
114
115 s->service = NULL;
116
acbb0225 117 free(s->bind_to_device);
e537352b 118 s->bind_to_device = NULL;
acbb0225
LP
119
120 unit_unwatch_timer(u, &s->timer_watch);
6cf6bbc2
LP
121
122 /* Make sure no service instance refers to us anymore. */
123 LIST_FOREACH(units_per_type, i, u->meta.manager->units_per_type[UNIT_SERVICE]) {
124 Service *service = (Service *) i;
125
126 if (service->socket == s)
127 service->socket = NULL;
128 }
5cb5a6ff
LP
129}
130
4f2d528d
LP
131static bool have_non_accept_socket(Socket *s) {
132 SocketPort *p;
133
134 assert(s);
135
136 if (!s->accept)
137 return true;
138
dd5ad9d4
LP
139 LIST_FOREACH(port, p, s->ports) {
140
141 if (p->type != SOCKET_SOCKET)
142 return true;
143
4f2d528d
LP
144 if (!socket_address_can_accept(&p->address))
145 return true;
dd5ad9d4 146 }
4f2d528d
LP
147
148 return false;
149}
150
151static int socket_verify(Socket *s) {
152 assert(s);
153
4cd1fbcc 154 if (s->meta.load_state != UNIT_LOADED)
4f2d528d
LP
155 return 0;
156
157 if (!s->ports) {
4cd1fbcc 158 log_error("%s lacks Listen setting. Refusing.", s->meta.id);
4f2d528d
LP
159 return -EINVAL;
160 }
161
6cf6bbc2 162 if (s->accept && s->max_connections <= 0) {
4d0e5dbd
LP
163 log_error("%s's MaxConnection setting too small. Refusing.", s->meta.id);
164 return -EINVAL;
165 }
166
167 if (s->exec_context.pam_name && s->kill_mode != KILL_CONTROL_GROUP) {
168 log_error("%s has PAM enabled. Kill mode must be set to 'control-group'. Refusing.", s->meta.id);
6cf6bbc2
LP
169 return -EINVAL;
170 }
171
4f2d528d
LP
172 return 0;
173}
174
6e2ef85b
LP
175static bool socket_needs_mount(Socket *s, const char *prefix) {
176 SocketPort *p;
177
178 assert(s);
179
180 LIST_FOREACH(port, p, s->ports) {
181
182 if (p->type == SOCKET_SOCKET) {
183 if (socket_address_needs_mount(&p->address, prefix))
184 return true;
185 } else {
186 assert(p->type == SOCKET_FIFO);
187 if (path_startswith(p->path, prefix))
188 return true;
189 }
190 }
191
192 return false;
193}
194
195int socket_add_one_mount_link(Socket *s, Mount *m) {
196 int r;
197
198 assert(s);
199 assert(m);
200
201 if (s->meta.load_state != UNIT_LOADED ||
202 m->meta.load_state != UNIT_LOADED)
203 return 0;
204
205 if (!socket_needs_mount(s, m->where))
206 return 0;
207
2c966c03 208 if ((r = unit_add_two_dependencies(UNIT(s), UNIT_AFTER, UNIT_REQUIRES, UNIT(m), true)) < 0)
6e2ef85b
LP
209 return r;
210
211 return 0;
212}
213
214static int socket_add_mount_links(Socket *s) {
215 Meta *other;
216 int r;
217
218 assert(s);
219
220 LIST_FOREACH(units_per_type, other, s->meta.manager->units_per_type[UNIT_MOUNT])
221 if ((r = socket_add_one_mount_link(s, (Mount*) other)) < 0)
222 return r;
223
224 return 0;
225}
226
227static int socket_add_device_link(Socket *s) {
228 char *t;
229 int r;
230
231 assert(s);
232
233 if (!s->bind_to_device)
234 return 0;
235
236 if (asprintf(&t, "/sys/subsystem/net/devices/%s", s->bind_to_device) < 0)
237 return -ENOMEM;
238
239 r = unit_add_node_link(UNIT(s), t, false);
240 free(t);
241
242 return r;
243}
244
a40eb732
LP
245static int socket_add_default_dependencies(Socket *s) {
246 int r;
247 assert(s);
248
249 if (s->meta.manager->running_as == MANAGER_SYSTEM)
250 if ((r = unit_add_two_dependencies_by_name(UNIT(s), UNIT_AFTER, UNIT_REQUIRES, SPECIAL_SYSINIT_TARGET, NULL, true)) < 0)
251 return r;
252
253 return unit_add_two_dependencies_by_name(UNIT(s), UNIT_BEFORE, UNIT_CONFLICTS, SPECIAL_SHUTDOWN_TARGET, NULL, true);
254}
255
e537352b
LP
256static int socket_load(Unit *u) {
257 Socket *s = SOCKET(u);
258 int r;
44d8db9e 259
e537352b
LP
260 assert(u);
261 assert(u->meta.load_state == UNIT_STUB);
44d8db9e 262
e537352b 263 if ((r = unit_load_fragment_and_dropin(u)) < 0)
23a177ef 264 return r;
44d8db9e 265
23a177ef 266 /* This is a new unit? Then let's add in some extras */
e537352b 267 if (u->meta.load_state == UNIT_LOADED) {
44d8db9e 268
4f2d528d
LP
269 if (have_non_accept_socket(s)) {
270 if ((r = unit_load_related_unit(u, ".service", (Unit**) &s->service)))
271 return r;
23a177ef 272
701cc384 273 if ((r = unit_add_dependency(u, UNIT_BEFORE, UNIT(s->service), true)) < 0)
4f2d528d
LP
274 return r;
275 }
44d8db9e 276
6e2ef85b
LP
277 if ((r = socket_add_mount_links(s)) < 0)
278 return r;
279
280 if ((r = socket_add_device_link(s)) < 0)
281 return r;
282
23a177ef
LP
283 if ((r = unit_add_exec_dependencies(u, &s->exec_context)) < 0)
284 return r;
285
286 if ((r = unit_add_default_cgroup(u)) < 0)
287 return r;
a40eb732
LP
288
289 if (s->meta.default_dependencies)
290 if ((r = socket_add_default_dependencies(s)) < 0)
291 return r;
23a177ef
LP
292 }
293
4f2d528d 294 return socket_verify(s);
44d8db9e
LP
295}
296
542563ba
LP
297static const char* listen_lookup(int type) {
298
299 if (type == SOCK_STREAM)
300 return "ListenStream";
301 else if (type == SOCK_DGRAM)
302 return "ListenDatagram";
303 else if (type == SOCK_SEQPACKET)
304 return "ListenSequentialPacket";
305
034c6ed7 306 assert_not_reached("Unknown socket type");
542563ba
LP
307 return NULL;
308}
309
87f0e418 310static void socket_dump(Unit *u, FILE *f, const char *prefix) {
5cb5a6ff 311
5cb5a6ff 312 SocketExecCommand c;
87f0e418 313 Socket *s = SOCKET(u);
542563ba 314 SocketPort *p;
82ba9f08
LP
315 const char *prefix2;
316 char *p2;
5cb5a6ff
LP
317
318 assert(s);
fa068367 319 assert(f);
5cb5a6ff 320
82ba9f08
LP
321 p2 = strappend(prefix, "\t");
322 prefix2 = p2 ? p2 : prefix;
c43d20a0 323
5cb5a6ff
LP
324 fprintf(f,
325 "%sSocket State: %s\n"
542563ba 326 "%sBindIPv6Only: %s\n"
b5a0699f 327 "%sBacklog: %u\n"
50159e6a 328 "%sKillMode: %s\n"
b5a0699f 329 "%sSocketMode: %04o\n"
4fd5948e
LP
330 "%sDirectoryMode: %04o\n"
331 "%sKeepAlive: %s\n"
332 "%sFreeBind: %s\n",
a16e1123 333 prefix, socket_state_to_string(s->state),
c0120d99 334 prefix, socket_address_bind_ipv6_only_to_string(s->bind_ipv6_only),
b5a0699f 335 prefix, s->backlog,
50159e6a 336 prefix, kill_mode_to_string(s->kill_mode),
b5a0699f 337 prefix, s->socket_mode,
4fd5948e
LP
338 prefix, s->directory_mode,
339 prefix, yes_no(s->keep_alive),
340 prefix, yes_no(s->free_bind));
542563ba 341
70123e68
LP
342 if (s->control_pid > 0)
343 fprintf(f,
bb00e604
LP
344 "%sControl PID: %lu\n",
345 prefix, (unsigned long) s->control_pid);
70123e68 346
acbb0225
LP
347 if (s->bind_to_device)
348 fprintf(f,
349 "%sBindToDevice: %s\n",
350 prefix, s->bind_to_device);
351
4f2d528d
LP
352 if (s->accept)
353 fprintf(f,
6cf6bbc2
LP
354 "%sAccepted: %u\n"
355 "%sNConnections: %u\n"
356 "%sMaxConnections: %u\n",
357 prefix, s->n_accepted,
358 prefix, s->n_connections,
359 prefix, s->max_connections);
4f2d528d 360
4fd5948e
LP
361 if (s->priority >= 0)
362 fprintf(f,
363 "%sPriority: %i\n",
364 prefix, s->priority);
365
366 if (s->receive_buffer > 0)
367 fprintf(f,
368 "%sReceiveBuffer: %zu\n",
369 prefix, s->receive_buffer);
370
371 if (s->send_buffer > 0)
372 fprintf(f,
373 "%sSendBuffer: %zu\n",
374 prefix, s->send_buffer);
375
376 if (s->ip_tos >= 0)
377 fprintf(f,
378 "%sIPTOS: %i\n",
379 prefix, s->ip_tos);
380
381 if (s->ip_ttl >= 0)
382 fprintf(f,
383 "%sIPTTL: %i\n",
384 prefix, s->ip_ttl);
385
386 if (s->pipe_size > 0)
387 fprintf(f,
388 "%sPipeSize: %zu\n",
389 prefix, s->pipe_size);
390
391 if (s->mark >= 0)
392 fprintf(f,
393 "%sMark: %i\n",
394 prefix, s->mark);
395
034c6ed7 396 LIST_FOREACH(port, p, s->ports) {
5cb5a6ff 397
542563ba
LP
398 if (p->type == SOCKET_SOCKET) {
399 const char *t;
400 int r;
401 char *k;
402
403 if ((r = socket_address_print(&p->address, &k)) < 0)
404 t = strerror(-r);
405 else
406 t = k;
407
408 fprintf(f, "%s%s: %s\n", prefix, listen_lookup(p->address.type), k);
409 free(k);
410 } else
411 fprintf(f, "%sListenFIFO: %s\n", prefix, p->path);
412 }
5cb5a6ff
LP
413
414 exec_context_dump(&s->exec_context, f, prefix);
415
e537352b 416 for (c = 0; c < _SOCKET_EXEC_COMMAND_MAX; c++) {
c43d20a0
LP
417 if (!s->exec_command[c])
418 continue;
5cb5a6ff 419
40d50879 420 fprintf(f, "%s-> %s:\n",
a16e1123 421 prefix, socket_exec_command_to_string(c));
c43d20a0
LP
422
423 exec_command_dump_list(s->exec_command[c], f, prefix2);
5cb5a6ff 424 }
c43d20a0 425
82ba9f08 426 free(p2);
5cb5a6ff
LP
427}
428
4f2d528d
LP
429static int instance_from_socket(int fd, unsigned nr, char **instance) {
430 socklen_t l;
431 char *r;
432 union {
433 struct sockaddr sa;
434 struct sockaddr_un un;
435 struct sockaddr_in in;
436 struct sockaddr_in6 in6;
437 struct sockaddr_storage storage;
438 } local, remote;
439
440 assert(fd >= 0);
441 assert(instance);
442
443 l = sizeof(local);
444 if (getsockname(fd, &local.sa, &l) < 0)
445 return -errno;
446
447 l = sizeof(remote);
448 if (getpeername(fd, &remote.sa, &l) < 0)
449 return -errno;
450
451 switch (local.sa.sa_family) {
452
453 case AF_INET: {
454 uint32_t
455 a = ntohl(local.in.sin_addr.s_addr),
456 b = ntohl(remote.in.sin_addr.s_addr);
457
458 if (asprintf(&r,
a16e1123 459 "%u-%u.%u.%u.%u:%u-%u.%u.%u.%u:%u",
4f2d528d
LP
460 nr,
461 a >> 24, (a >> 16) & 0xFF, (a >> 8) & 0xFF, a & 0xFF,
462 ntohs(local.in.sin_port),
463 b >> 24, (b >> 16) & 0xFF, (b >> 8) & 0xFF, b & 0xFF,
464 ntohs(remote.in.sin_port)) < 0)
465 return -ENOMEM;
466
467 break;
468 }
469
470 case AF_INET6: {
2b061f5a
LP
471 static const char ipv4_prefix[] = {
472 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xFF, 0xFF
473 };
474
475 if (memcmp(&local.in6.sin6_addr, ipv4_prefix, sizeof(ipv4_prefix)) == 0 &&
476 memcmp(&remote.in6.sin6_addr, ipv4_prefix, sizeof(ipv4_prefix)) == 0) {
477 const uint8_t
478 *a = local.in6.sin6_addr.s6_addr+12,
479 *b = remote.in6.sin6_addr.s6_addr+12;
480
481 if (asprintf(&r,
482 "%u-%u.%u.%u.%u:%u-%u.%u.%u.%u:%u",
483 nr,
484 a[0], a[1], a[2], a[3],
485 ntohs(local.in6.sin6_port),
486 b[0], b[1], b[2], b[3],
487 ntohs(remote.in6.sin6_port)) < 0)
488 return -ENOMEM;
489 } else {
490 char a[INET6_ADDRSTRLEN], b[INET6_ADDRSTRLEN];
491
492 if (asprintf(&r,
493 "%u-%s:%u-%s:%u",
494 nr,
495 inet_ntop(AF_INET6, &local.in6.sin6_addr, a, sizeof(a)),
496 ntohs(local.in6.sin6_port),
497 inet_ntop(AF_INET6, &remote.in6.sin6_addr, b, sizeof(b)),
498 ntohs(remote.in6.sin6_port)) < 0)
499 return -ENOMEM;
500 }
4f2d528d
LP
501
502 break;
503 }
504
505 case AF_UNIX: {
506 struct ucred ucred;
507
508 l = sizeof(ucred);
509 if (getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &ucred, &l) < 0)
510 return -errno;
511
512 if (asprintf(&r,
bb00e604 513 "%u-%lu-%lu",
4f2d528d 514 nr,
bb00e604
LP
515 (unsigned long) ucred.pid,
516 (unsigned long) ucred.uid) < 0)
4f2d528d
LP
517 return -ENOMEM;
518
519 break;
520 }
521
522 default:
523 assert_not_reached("Unhandled socket type.");
524 }
525
526 *instance = r;
527 return 0;
528}
529
034c6ed7 530static void socket_close_fds(Socket *s) {
83c60c9f
LP
531 SocketPort *p;
532
533 assert(s);
534
034c6ed7 535 LIST_FOREACH(port, p, s->ports) {
83c60c9f
LP
536 if (p->fd < 0)
537 continue;
538
acbb0225 539 unit_unwatch_fd(UNIT(s), &p->fd_watch);
a16e1123
LP
540 close_nointr_nofail(p->fd);
541
542 /* One little note: we should never delete any sockets
543 * in the file system here! After all some other
544 * process we spawned might still have a reference of
545 * this fd and wants to continue to use it. Therefore
546 * we delete sockets in the file system before we
547 * create a new one, not after we stopped using
548 * one! */
9152c765 549
83c60c9f
LP
550 p->fd = -1;
551 }
552}
553
4fd5948e
LP
554static void socket_apply_socket_options(Socket *s, int fd) {
555 assert(s);
556 assert(fd >= 0);
557
558 if (s->keep_alive) {
559 int b = s->keep_alive;
560 if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &b, sizeof(b)) < 0)
561 log_warning("SO_KEEPALIVE failed: %m");
562 }
563
564 if (s->priority >= 0)
565 if (setsockopt(fd, SOL_SOCKET, SO_PRIORITY, &s->priority, sizeof(s->priority)) < 0)
566 log_warning("SO_PRIORITY failed: %m");
567
568 if (s->receive_buffer > 0) {
569 int value = (int) s->receive_buffer;
570 if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, sizeof(value)) < 0)
571 log_warning("SO_RCVBUF failed: %m");
572 }
573
574 if (s->send_buffer > 0) {
575 int value = (int) s->send_buffer;
576 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, sizeof(value)) < 0)
577 log_warning("SO_SNDBUF failed: %m");
578 }
579
580 if (s->mark >= 0)
581 if (setsockopt(fd, SOL_SOCKET, SO_MARK, &s->mark, sizeof(s->mark)) < 0)
582 log_warning("SO_MARK failed: %m");
583
584 if (s->ip_tos >= 0)
585 if (setsockopt(fd, IPPROTO_IP, IP_TOS, &s->ip_tos, sizeof(s->ip_tos)) < 0)
586 log_warning("IP_TOS failed: %m");
587
46925ac5
LP
588 if (s->ip_ttl >= 0) {
589 int r, x;
590
591 r = setsockopt(fd, IPPROTO_IP, IP_TTL, &s->ip_ttl, sizeof(s->ip_ttl));
592 x = setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &s->ip_ttl, sizeof(s->ip_ttl));
593
594 if (r < 0 && x < 0)
595 log_warning("IP_TTL/IPV6_UNICAST_HOPS failed: %m");
596 }
4fd5948e
LP
597}
598
599static void socket_apply_pipe_options(Socket *s, int fd) {
600 assert(s);
601 assert(fd >= 0);
602
603 if (s->pipe_size > 0)
604 if (fcntl(fd, F_SETPIPE_SZ, s->pipe_size) < 0)
605 log_warning("F_SETPIPE_SZ: %m");
606}
607
034c6ed7 608static int socket_open_fds(Socket *s) {
83c60c9f
LP
609 SocketPort *p;
610 int r;
611
612 assert(s);
613
034c6ed7 614 LIST_FOREACH(port, p, s->ports) {
83c60c9f 615
034c6ed7
LP
616 if (p->fd >= 0)
617 continue;
83c60c9f
LP
618
619 if (p->type == SOCKET_SOCKET) {
620
b5a0699f
LP
621 if ((r = socket_address_listen(
622 &p->address,
623 s->backlog,
624 s->bind_ipv6_only,
625 s->bind_to_device,
4fd5948e 626 s->free_bind,
b5a0699f
LP
627 s->directory_mode,
628 s->socket_mode,
629 &p->fd)) < 0)
83c60c9f
LP
630 goto rollback;
631
4fd5948e
LP
632 socket_apply_socket_options(s, p->fd);
633
83c60c9f
LP
634 } else {
635 struct stat st;
636 assert(p->type == SOCKET_FIFO);
637
8cb45bf8
LP
638 mkdir_parents(p->path, s->directory_mode);
639
640 if (mkfifo(p->path, s->socket_mode) < 0 && errno != EEXIST) {
83c60c9f
LP
641 r = -errno;
642 goto rollback;
643 }
644
645 if ((p->fd = open(p->path, O_RDWR|O_CLOEXEC|O_NOCTTY|O_NONBLOCK|O_NOFOLLOW)) < 0) {
646 r = -errno;
647 goto rollback;
648 }
649
650 if (fstat(p->fd, &st) < 0) {
651 r = -errno;
652 goto rollback;
653 }
654
655 /* FIXME verify user, access mode */
656
657 if (!S_ISFIFO(st.st_mode)) {
658 r = -EEXIST;
659 goto rollback;
660 }
4fd5948e
LP
661
662 socket_apply_pipe_options(s, p->fd);
83c60c9f 663 }
034c6ed7
LP
664 }
665
666 return 0;
667
668rollback:
669 socket_close_fds(s);
670 return r;
671}
672
673static void socket_unwatch_fds(Socket *s) {
674 SocketPort *p;
9152c765 675
034c6ed7
LP
676 assert(s);
677
678 LIST_FOREACH(port, p, s->ports) {
679 if (p->fd < 0)
680 continue;
681
acbb0225 682 unit_unwatch_fd(UNIT(s), &p->fd_watch);
83c60c9f 683 }
034c6ed7
LP
684}
685
686static int socket_watch_fds(Socket *s) {
687 SocketPort *p;
688 int r;
689
690 assert(s);
83c60c9f 691
034c6ed7
LP
692 LIST_FOREACH(port, p, s->ports) {
693 if (p->fd < 0)
694 continue;
695
cabab516 696 p->fd_watch.socket_accept =
4f2d528d 697 s->accept &&
dd5ad9d4 698 p->type == SOCKET_SOCKET &&
4f2d528d
LP
699 socket_address_can_accept(&p->address);
700
f94ea366 701 if ((r = unit_watch_fd(UNIT(s), p->fd, EPOLLIN, &p->fd_watch)) < 0)
034c6ed7
LP
702 goto fail;
703 }
83c60c9f 704
542563ba 705 return 0;
83c60c9f 706
034c6ed7
LP
707fail:
708 socket_unwatch_fds(s);
709 return r;
710}
711
712static void socket_set_state(Socket *s, SocketState state) {
713 SocketState old_state;
714 assert(s);
715
716 old_state = s->state;
717 s->state = state;
718
719 if (state != SOCKET_START_PRE &&
720 state != SOCKET_START_POST &&
721 state != SOCKET_STOP_PRE &&
722 state != SOCKET_STOP_PRE_SIGTERM &&
723 state != SOCKET_STOP_PRE_SIGKILL &&
724 state != SOCKET_STOP_POST &&
80876c20
LP
725 state != SOCKET_FINAL_SIGTERM &&
726 state != SOCKET_FINAL_SIGKILL) {
acbb0225 727 unit_unwatch_timer(UNIT(s), &s->timer_watch);
5e94833f 728 socket_unwatch_control_pid(s);
034c6ed7 729 s->control_command = NULL;
a16e1123 730 s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
e537352b 731 }
034c6ed7 732
a16e1123
LP
733 if (state != SOCKET_LISTENING)
734 socket_unwatch_fds(s);
735
034c6ed7
LP
736 if (state != SOCKET_START_POST &&
737 state != SOCKET_LISTENING &&
738 state != SOCKET_RUNNING &&
739 state != SOCKET_STOP_PRE &&
740 state != SOCKET_STOP_PRE_SIGTERM &&
741 state != SOCKET_STOP_PRE_SIGKILL)
742 socket_close_fds(s);
743
e537352b 744 if (state != old_state)
40d50879 745 log_debug("%s changed %s -> %s",
a16e1123
LP
746 s->meta.id,
747 socket_state_to_string(old_state),
748 socket_state_to_string(state));
acbb0225
LP
749
750 unit_notify(UNIT(s), state_translation_table[old_state], state_translation_table[state]);
034c6ed7
LP
751}
752
a16e1123
LP
753static int socket_coldplug(Unit *u) {
754 Socket *s = SOCKET(u);
755 int r;
756
757 assert(s);
758 assert(s->state == SOCKET_DEAD);
759
760 if (s->deserialized_state != s->state) {
761
762 if (s->deserialized_state == SOCKET_START_PRE ||
763 s->deserialized_state == SOCKET_START_POST ||
764 s->deserialized_state == SOCKET_STOP_PRE ||
765 s->deserialized_state == SOCKET_STOP_PRE_SIGTERM ||
766 s->deserialized_state == SOCKET_STOP_PRE_SIGKILL ||
767 s->deserialized_state == SOCKET_STOP_POST ||
768 s->deserialized_state == SOCKET_FINAL_SIGTERM ||
769 s->deserialized_state == SOCKET_FINAL_SIGKILL) {
770
771 if (s->control_pid <= 0)
772 return -EBADMSG;
773
774 if ((r = unit_watch_pid(UNIT(s), s->control_pid)) < 0)
775 return r;
776
777 if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
778 return r;
779 }
780
781 if (s->deserialized_state == SOCKET_START_POST ||
782 s->deserialized_state == SOCKET_LISTENING ||
783 s->deserialized_state == SOCKET_RUNNING ||
784 s->deserialized_state == SOCKET_STOP_PRE ||
785 s->deserialized_state == SOCKET_STOP_PRE_SIGTERM ||
786 s->deserialized_state == SOCKET_STOP_PRE_SIGKILL)
787 if ((r = socket_open_fds(s)) < 0)
788 return r;
789
790 if (s->deserialized_state == SOCKET_LISTENING)
791 if ((r = socket_watch_fds(s)) < 0)
792 return r;
793
794 socket_set_state(s, s->deserialized_state);
795 }
796
797 return 0;
798}
799
e537352b 800static int socket_spawn(Socket *s, ExecCommand *c, pid_t *_pid) {
034c6ed7
LP
801 pid_t pid;
802 int r;
9e2f7c11 803 char **argv;
034c6ed7
LP
804
805 assert(s);
806 assert(c);
807 assert(_pid);
808
e537352b
LP
809 if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
810 goto fail;
034c6ed7 811
9e2f7c11
LP
812 if (!(argv = unit_full_printf_strv(UNIT(s), c->argv))) {
813 r = -ENOMEM;
814 goto fail;
815 }
816
817 r = exec_spawn(c,
818 argv,
819 &s->exec_context,
820 NULL, 0,
1137a57c 821 s->meta.manager->environment,
9e2f7c11
LP
822 true,
823 true,
4cd1fbcc
LP
824 s->meta.manager->confirm_spawn,
825 s->meta.cgroup_bondings,
9e2f7c11
LP
826 &pid);
827
828 strv_free(argv);
829 if (r < 0)
034c6ed7
LP
830 goto fail;
831
87f0e418 832 if ((r = unit_watch_pid(UNIT(s), pid)) < 0)
034c6ed7
LP
833 /* FIXME: we need to do something here */
834 goto fail;
83c60c9f 835
034c6ed7
LP
836 *_pid = pid;
837
838 return 0;
839
840fail:
e537352b 841 unit_unwatch_timer(UNIT(s), &s->timer_watch);
83c60c9f
LP
842
843 return r;
542563ba
LP
844}
845
034c6ed7
LP
846static void socket_enter_dead(Socket *s, bool success) {
847 assert(s);
848
849 if (!success)
850 s->failure = true;
851
18c78fb1 852 socket_set_state(s, s->failure ? SOCKET_MAINTENANCE : SOCKET_DEAD);
034c6ed7
LP
853}
854
80876c20
LP
855static void socket_enter_signal(Socket *s, SocketState state, bool success);
856
034c6ed7
LP
857static void socket_enter_stop_post(Socket *s, bool success) {
858 int r;
859 assert(s);
860
861 if (!success)
862 s->failure = true;
863
5e94833f
LP
864 socket_unwatch_control_pid(s);
865
a16e1123
LP
866 s->control_command_id = SOCKET_EXEC_STOP_POST;
867
80876c20 868 if ((s->control_command = s->exec_command[SOCKET_EXEC_STOP_POST])) {
e537352b 869 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
034c6ed7
LP
870 goto fail;
871
80876c20
LP
872 socket_set_state(s, SOCKET_STOP_POST);
873 } else
874 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, true);
034c6ed7
LP
875
876 return;
877
878fail:
6cf6bbc2 879 log_warning("%s failed to run 'stop-post' task: %s", s->meta.id, strerror(-r));
80876c20 880 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
034c6ed7
LP
881}
882
883static void socket_enter_signal(Socket *s, SocketState state, bool success) {
884 int r;
80876c20 885 bool sent = false;
034c6ed7
LP
886
887 assert(s);
888
889 if (!success)
890 s->failure = true;
891
80876c20
LP
892 if (s->kill_mode != KILL_NONE) {
893 int sig = (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_FINAL_SIGTERM) ? SIGTERM : SIGKILL;
034c6ed7 894
50159e6a
LP
895 if (s->kill_mode == KILL_CONTROL_GROUP) {
896
4cd1fbcc 897 if ((r = cgroup_bonding_kill_list(s->meta.cgroup_bondings, sig)) < 0) {
50159e6a
LP
898 if (r != -EAGAIN && r != -ESRCH)
899 goto fail;
900 } else
901 sent = true;
034c6ed7 902 }
50159e6a 903
80876c20 904 if (!sent && s->control_pid > 0)
50159e6a
LP
905 if (kill(s->kill_mode == KILL_PROCESS ? s->control_pid : -s->control_pid, sig) < 0 && errno != ESRCH) {
906 r = -errno;
907 goto fail;
908 }
d6ea93e3 909 }
034c6ed7 910
e93bc5a6 911 if (sent && s->control_pid > 0) {
80876c20
LP
912 if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
913 goto fail;
d6ea93e3 914
80876c20
LP
915 socket_set_state(s, state);
916 } else if (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_STOP_PRE_SIGKILL)
917 socket_enter_stop_post(s, true);
918 else
034c6ed7
LP
919 socket_enter_dead(s, true);
920
921 return;
922
923fail:
9e2f7c11 924 log_warning("%s failed to kill processes: %s", s->meta.id, strerror(-r));
034c6ed7
LP
925
926 if (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_STOP_PRE_SIGKILL)
927 socket_enter_stop_post(s, false);
928 else
929 socket_enter_dead(s, false);
930}
931
932static void socket_enter_stop_pre(Socket *s, bool success) {
933 int r;
934 assert(s);
935
936 if (!success)
937 s->failure = true;
938
5e94833f
LP
939 socket_unwatch_control_pid(s);
940
a16e1123
LP
941 s->control_command_id = SOCKET_EXEC_STOP_PRE;
942
80876c20 943 if ((s->control_command = s->exec_command[SOCKET_EXEC_STOP_PRE])) {
e537352b 944 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
034c6ed7
LP
945 goto fail;
946
80876c20
LP
947 socket_set_state(s, SOCKET_STOP_PRE);
948 } else
034c6ed7
LP
949 socket_enter_stop_post(s, true);
950
951 return;
952
953fail:
6cf6bbc2 954 log_warning("%s failed to run 'stop-pre' task: %s", s->meta.id, strerror(-r));
034c6ed7
LP
955 socket_enter_stop_post(s, false);
956}
957
e9af15c3
LP
958static void socket_enter_listening(Socket *s) {
959 int r;
960 assert(s);
961
962 if ((r = socket_watch_fds(s)) < 0) {
9e2f7c11 963 log_warning("%s failed to watch sockets: %s", s->meta.id, strerror(-r));
e9af15c3
LP
964 goto fail;
965 }
966
967 socket_set_state(s, SOCKET_LISTENING);
968 return;
969
970fail:
971 socket_enter_stop_pre(s, false);
972}
973
034c6ed7
LP
974static void socket_enter_start_post(Socket *s) {
975 int r;
976 assert(s);
977
e9af15c3 978 if ((r = socket_open_fds(s)) < 0) {
9e2f7c11 979 log_warning("%s failed to listen on sockets: %s", s->meta.id, strerror(-r));
034c6ed7
LP
980 goto fail;
981 }
982
5e94833f
LP
983 socket_unwatch_control_pid(s);
984
a16e1123
LP
985 s->control_command_id = SOCKET_EXEC_START_POST;
986
80876c20 987 if ((s->control_command = s->exec_command[SOCKET_EXEC_START_POST])) {
e537352b 988 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0) {
6cf6bbc2 989 log_warning("%s failed to run 'start-post' task: %s", s->meta.id, strerror(-r));
034c6ed7
LP
990 goto fail;
991 }
992
80876c20
LP
993 socket_set_state(s, SOCKET_START_POST);
994 } else
e9af15c3 995 socket_enter_listening(s);
034c6ed7
LP
996
997 return;
998
999fail:
1000 socket_enter_stop_pre(s, false);
1001}
1002
1003static void socket_enter_start_pre(Socket *s) {
1004 int r;
1005 assert(s);
1006
5e94833f
LP
1007 socket_unwatch_control_pid(s);
1008
a16e1123
LP
1009 s->control_command_id = SOCKET_EXEC_START_PRE;
1010
80876c20 1011 if ((s->control_command = s->exec_command[SOCKET_EXEC_START_PRE])) {
e537352b 1012 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
034c6ed7
LP
1013 goto fail;
1014
80876c20
LP
1015 socket_set_state(s, SOCKET_START_PRE);
1016 } else
034c6ed7
LP
1017 socket_enter_start_post(s);
1018
1019 return;
1020
1021fail:
6cf6bbc2 1022 log_warning("%s failed to run 'start-pre' task: %s", s->meta.id, strerror(-r));
034c6ed7
LP
1023 socket_enter_dead(s, false);
1024}
1025
4f2d528d 1026static void socket_enter_running(Socket *s, int cfd) {
034c6ed7
LP
1027 int r;
1028
1029 assert(s);
1030
4f2d528d 1031 if (cfd < 0) {
4cd1fbcc 1032 if ((r = manager_add_job(s->meta.manager, JOB_START, UNIT(s->service), JOB_REPLACE, true, NULL)) < 0)
4f2d528d
LP
1033 goto fail;
1034
1035 socket_set_state(s, SOCKET_RUNNING);
1036 } else {
1037 Unit *u;
1038 char *prefix, *instance, *name;
1039
6cf6bbc2
LP
1040 if (s->n_connections >= s->max_connections) {
1041 log_warning("Too many incoming connections (%u)", s->n_connections);
1042 close_nointr_nofail(cfd);
1043 return;
1044 }
1045
b6dbbe1c 1046 if ((r = instance_from_socket(cfd, s->n_accepted++, &instance)) < 0)
4f2d528d
LP
1047 goto fail;
1048
4cd1fbcc 1049 if (!(prefix = unit_name_to_prefix(s->meta.id))) {
4f2d528d
LP
1050 free(instance);
1051 r = -ENOMEM;
1052 goto fail;
1053 }
1054
1055 name = unit_name_build(prefix, instance, ".service");
1056 free(prefix);
1057 free(instance);
1058
b6dbbe1c 1059 if (!name) {
4f2d528d 1060 r = -ENOMEM;
b6dbbe1c
LP
1061 goto fail;
1062 }
4f2d528d 1063
4cd1fbcc 1064 r = manager_load_unit(s->meta.manager, name, NULL, &u);
4f2d528d
LP
1065 free(name);
1066
1067 if (r < 0)
1068 goto fail;
1069
6cf6bbc2 1070 if ((r = service_set_socket_fd(SERVICE(u), cfd, s)) < 0)
4f2d528d
LP
1071 goto fail;
1072
1073 cfd = -1;
1074
6cf6bbc2
LP
1075 s->n_connections ++;
1076
4f2d528d
LP
1077 if ((r = manager_add_job(u->meta.manager, JOB_START, u, JOB_REPLACE, true, NULL)) < 0)
1078 goto fail;
1079 }
034c6ed7 1080
034c6ed7
LP
1081 return;
1082
1083fail:
9e2f7c11 1084 log_warning("%s failed to queue socket startup job: %s", s->meta.id, strerror(-r));
80876c20 1085 socket_enter_stop_pre(s, false);
4f2d528d
LP
1086
1087 if (cfd >= 0)
1088 close_nointr_nofail(cfd);
034c6ed7
LP
1089}
1090
1091static void socket_run_next(Socket *s, bool success) {
1092 int r;
1093
1094 assert(s);
1095 assert(s->control_command);
1096 assert(s->control_command->command_next);
1097
1098 if (!success)
1099 s->failure = true;
1100
5e94833f
LP
1101 socket_unwatch_control_pid(s);
1102
034c6ed7
LP
1103 s->control_command = s->control_command->command_next;
1104
e537352b 1105 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
034c6ed7
LP
1106 goto fail;
1107
1108 return;
1109
1110fail:
6cf6bbc2 1111 log_warning("%s failed to run next task: %s", s->meta.id, strerror(-r));
80876c20
LP
1112
1113 if (s->state == SOCKET_START_POST)
1114 socket_enter_stop_pre(s, false);
034c6ed7
LP
1115 else if (s->state == SOCKET_STOP_POST)
1116 socket_enter_dead(s, false);
1117 else
80876c20 1118 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
034c6ed7
LP
1119}
1120
87f0e418
LP
1121static int socket_start(Unit *u) {
1122 Socket *s = SOCKET(u);
83c60c9f
LP
1123
1124 assert(s);
1125
034c6ed7
LP
1126 /* We cannot fulfill this request right now, try again later
1127 * please! */
1128 if (s->state == SOCKET_STOP_PRE ||
1129 s->state == SOCKET_STOP_PRE_SIGKILL ||
1130 s->state == SOCKET_STOP_PRE_SIGTERM ||
1131 s->state == SOCKET_STOP_POST ||
80876c20
LP
1132 s->state == SOCKET_FINAL_SIGTERM ||
1133 s->state == SOCKET_FINAL_SIGKILL)
034c6ed7
LP
1134 return -EAGAIN;
1135
83c60c9f
LP
1136 if (s->state == SOCKET_START_PRE ||
1137 s->state == SOCKET_START_POST)
034c6ed7 1138 return 0;
83c60c9f 1139
034c6ed7 1140 /* Cannot run this without the service being around */
4f2d528d
LP
1141 if (s->service) {
1142 if (s->service->meta.load_state != UNIT_LOADED)
1143 return -ENOENT;
1144
1145 /* If the service is alredy actvie we cannot start the
1146 * socket */
1147 if (s->service->state != SERVICE_DEAD &&
18c78fb1 1148 s->service->state != SERVICE_MAINTENANCE &&
4f2d528d
LP
1149 s->service->state != SERVICE_AUTO_RESTART)
1150 return -EBUSY;
1151 }
e537352b 1152
18c78fb1 1153 assert(s->state == SOCKET_DEAD || s->state == SOCKET_MAINTENANCE);
83c60c9f 1154
034c6ed7
LP
1155 s->failure = false;
1156 socket_enter_start_pre(s);
1157 return 0;
1158}
83c60c9f 1159
87f0e418
LP
1160static int socket_stop(Unit *u) {
1161 Socket *s = SOCKET(u);
034c6ed7
LP
1162
1163 assert(s);
1164
1165 /* We cannot fulfill this request right now, try again later
1166 * please! */
1167 if (s->state == SOCKET_START_PRE ||
1168 s->state == SOCKET_START_POST)
1169 return -EAGAIN;
83c60c9f 1170
e537352b
LP
1171 /* Already on it */
1172 if (s->state == SOCKET_STOP_PRE ||
1173 s->state == SOCKET_STOP_PRE_SIGTERM ||
1174 s->state == SOCKET_STOP_PRE_SIGKILL ||
1175 s->state == SOCKET_STOP_POST ||
80876c20
LP
1176 s->state == SOCKET_FINAL_SIGTERM ||
1177 s->state == SOCKET_FINAL_SIGTERM)
e537352b
LP
1178 return 0;
1179
034c6ed7 1180 assert(s->state == SOCKET_LISTENING || s->state == SOCKET_RUNNING);
83c60c9f 1181
034c6ed7 1182 socket_enter_stop_pre(s, true);
542563ba
LP
1183 return 0;
1184}
1185
a16e1123
LP
1186static int socket_serialize(Unit *u, FILE *f, FDSet *fds) {
1187 Socket *s = SOCKET(u);
1188 SocketPort *p;
1189 int r;
1190
1191 assert(u);
1192 assert(f);
1193 assert(fds);
1194
1195 unit_serialize_item(u, f, "state", socket_state_to_string(s->state));
1196 unit_serialize_item(u, f, "failure", yes_no(s->failure));
1197 unit_serialize_item_format(u, f, "n-accepted", "%u", s->n_accepted);
1198
1199 if (s->control_pid > 0)
5925dd3c 1200 unit_serialize_item_format(u, f, "control-pid", "%lu", (unsigned long) s->control_pid);
a16e1123
LP
1201
1202 if (s->control_command_id >= 0)
1203 unit_serialize_item(u, f, "control-command", socket_exec_command_to_string(s->control_command_id));
1204
1205 LIST_FOREACH(port, p, s->ports) {
1206 int copy;
1207
1208 if (p->fd < 0)
1209 continue;
1210
1211 if ((copy = fdset_put_dup(fds, p->fd)) < 0)
1212 return copy;
1213
1214 if (p->type == SOCKET_SOCKET) {
1215 char *t;
1216
1217 if ((r = socket_address_print(&p->address, &t)) < 0)
1218 return r;
1219
27ca8d7a 1220 unit_serialize_item_format(u, f, "socket", "%i %i %s", copy, p->address.type, t);
a16e1123
LP
1221 free(t);
1222 } else {
1223 assert(p->type == SOCKET_FIFO);
1224 unit_serialize_item_format(u, f, "fifo", "%i %s", copy, p->path);
1225 }
1226 }
1227
1228 return 0;
1229}
1230
1231static int socket_deserialize_item(Unit *u, const char *key, const char *value, FDSet *fds) {
1232 Socket *s = SOCKET(u);
1233 int r;
1234
1235 assert(u);
1236 assert(key);
1237 assert(value);
1238 assert(fds);
1239
1240 if (streq(key, "state")) {
1241 SocketState state;
1242
1243 if ((state = socket_state_from_string(value)) < 0)
1244 log_debug("Failed to parse state value %s", value);
1245 else
1246 s->deserialized_state = state;
1247 } else if (streq(key, "failure")) {
1248 int b;
1249
1250 if ((b = parse_boolean(value)) < 0)
1251 log_debug("Failed to parse failure value %s", value);
1252 else
1253 s->failure = b || s->failure;
1254
1255 } else if (streq(key, "n-accepted")) {
1256 unsigned k;
1257
1258 if ((r = safe_atou(value, &k)) < 0)
1259 log_debug("Failed to parse n-accepted value %s", value);
1260 else
1261 s->n_accepted += k;
1262 } else if (streq(key, "control-pid")) {
5925dd3c 1263 pid_t pid;
a16e1123 1264
5925dd3c 1265 if ((r = parse_pid(value, &pid)) < 0)
a16e1123
LP
1266 log_debug("Failed to parse control-pid value %s", value);
1267 else
5925dd3c 1268 s->control_pid = pid;
a16e1123
LP
1269 } else if (streq(key, "control-command")) {
1270 SocketExecCommand id;
1271
1272 if ((id = socket_exec_command_from_string(value)) < 0)
1273 log_debug("Failed to parse exec-command value %s", value);
1274 else {
1275 s->control_command_id = id;
1276 s->control_command = s->exec_command[id];
1277 }
1278 } else if (streq(key, "fifo")) {
1279 int fd, skip = 0;
1280 SocketPort *p;
1281
1282 if (sscanf(value, "%i %n", &fd, &skip) < 1 || fd < 0 || !fdset_contains(fds, fd))
1283 log_debug("Failed to parse fifo value %s", value);
1284 else {
1285
1286 LIST_FOREACH(port, p, s->ports)
1287 if (streq(p->path, value+skip))
1288 break;
1289
1290 if (p) {
1291 if (p->fd >= 0)
1292 close_nointr_nofail(p->fd);
1293 p->fd = fdset_remove(fds, fd);
1294 }
1295 }
1296
1297 } else if (streq(key, "socket")) {
27ca8d7a 1298 int fd, type, skip = 0;
a16e1123
LP
1299 SocketPort *p;
1300
27ca8d7a 1301 if (sscanf(value, "%i %i %n", &fd, &type, &skip) < 2 || fd < 0 || type < 0 || !fdset_contains(fds, fd))
a16e1123
LP
1302 log_debug("Failed to parse socket value %s", value);
1303 else {
1304
1305 LIST_FOREACH(port, p, s->ports)
27ca8d7a 1306 if (socket_address_is(&p->address, value+skip, type))
a16e1123
LP
1307 break;
1308
1309 if (p) {
1310 if (p->fd >= 0)
1311 close_nointr_nofail(p->fd);
1312 p->fd = fdset_remove(fds, fd);
1313 }
1314 }
1315
1316 } else
1317 log_debug("Unknown serialization key '%s'", key);
1318
1319 return 0;
1320}
1321
87f0e418
LP
1322static UnitActiveState socket_active_state(Unit *u) {
1323 assert(u);
5cb5a6ff 1324
acbb0225 1325 return state_translation_table[SOCKET(u)->state];
5cb5a6ff
LP
1326}
1327
10a94420
LP
1328static const char *socket_sub_state_to_string(Unit *u) {
1329 assert(u);
1330
a16e1123 1331 return socket_state_to_string(SOCKET(u)->state);
10a94420
LP
1332}
1333
6cf6bbc2
LP
1334static bool socket_check_gc(Unit *u) {
1335 Socket *s = SOCKET(u);
1336
1337 assert(u);
1338
1339 return s->n_connections > 0;
1340}
1341
acbb0225 1342static void socket_fd_event(Unit *u, int fd, uint32_t events, Watch *w) {
87f0e418 1343 Socket *s = SOCKET(u);
4f2d528d 1344 int cfd = -1;
9152c765 1345
034c6ed7 1346 assert(s);
8d567588 1347 assert(fd >= 0);
9152c765 1348
871d7de4
LP
1349 if (s->state != SOCKET_LISTENING)
1350 return;
1351
9e2f7c11 1352 log_debug("Incoming traffic on %s", u->meta.id);
9152c765 1353
4f2d528d
LP
1354 if (events != EPOLLIN) {
1355 log_error("Got invalid poll event on socket.");
8d567588 1356 goto fail;
4f2d528d
LP
1357 }
1358
cabab516 1359 if (w->socket_accept) {
4f2d528d
LP
1360 for (;;) {
1361
1362 if ((cfd = accept4(fd, NULL, NULL, SOCK_NONBLOCK)) < 0) {
1363
1364 if (errno == EINTR)
1365 continue;
1366
1367 log_error("Failed to accept socket: %m");
8d567588 1368 goto fail;
4f2d528d
LP
1369 }
1370
1371 break;
1372 }
4fd5948e
LP
1373
1374 socket_apply_socket_options(s, cfd);
4f2d528d 1375 }
9152c765 1376
4f2d528d 1377 socket_enter_running(s, cfd);
8d567588
LP
1378 return;
1379
1380fail:
1381 socket_enter_stop_pre(s, false);
9152c765
LP
1382}
1383
87f0e418
LP
1384static void socket_sigchld_event(Unit *u, pid_t pid, int code, int status) {
1385 Socket *s = SOCKET(u);
034c6ed7 1386 bool success;
5cb5a6ff
LP
1387
1388 assert(s);
034c6ed7 1389 assert(pid >= 0);
5cb5a6ff 1390
8c47c732
LP
1391 if (pid != s->control_pid)
1392 return;
542563ba 1393
034c6ed7
LP
1394 s->control_pid = 0;
1395
8c47c732
LP
1396 success = is_clean_exit(code, status);
1397 s->failure = s->failure || !success;
1398
a16e1123
LP
1399 if (s->control_command)
1400 exec_status_fill(&s->control_command->exec_status, pid, code, status);
1401
9e2f7c11 1402 log_debug("%s control process exited, code=%s status=%i", u->meta.id, sigchld_code_to_string(code), status);
034c6ed7 1403
a16e1123
LP
1404 if (s->control_command && s->control_command->command_next && success) {
1405 log_debug("%s running next command for state %s", u->meta.id, socket_state_to_string(s->state));
034c6ed7 1406 socket_run_next(s, success);
acbb0225 1407 } else {
a16e1123
LP
1408 s->control_command = NULL;
1409 s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
1410
034c6ed7
LP
1411 /* No further commands for this step, so let's figure
1412 * out what to do next */
5cb5a6ff 1413
a16e1123 1414 log_debug("%s got final SIGCHLD for state %s", u->meta.id, socket_state_to_string(s->state));
acbb0225 1415
034c6ed7
LP
1416 switch (s->state) {
1417
1418 case SOCKET_START_PRE:
1419 if (success)
acbb0225 1420 socket_enter_start_post(s);
034c6ed7 1421 else
80876c20 1422 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
034c6ed7
LP
1423 break;
1424
1425 case SOCKET_START_POST:
1426 if (success)
e9af15c3 1427 socket_enter_listening(s);
034c6ed7
LP
1428 else
1429 socket_enter_stop_pre(s, false);
1430 break;
1431
1432 case SOCKET_STOP_PRE:
1433 case SOCKET_STOP_PRE_SIGTERM:
1434 case SOCKET_STOP_PRE_SIGKILL:
1435 socket_enter_stop_post(s, success);
1436 break;
1437
1438 case SOCKET_STOP_POST:
80876c20
LP
1439 case SOCKET_FINAL_SIGTERM:
1440 case SOCKET_FINAL_SIGKILL:
034c6ed7
LP
1441 socket_enter_dead(s, success);
1442 break;
1443
1444 default:
1445 assert_not_reached("Uh, control process died at wrong time.");
1446 }
1447 }
1448}
5cb5a6ff 1449
acbb0225 1450static void socket_timer_event(Unit *u, uint64_t elapsed, Watch *w) {
87f0e418 1451 Socket *s = SOCKET(u);
5cb5a6ff 1452
034c6ed7
LP
1453 assert(s);
1454 assert(elapsed == 1);
acbb0225 1455 assert(w == &s->timer_watch);
034c6ed7
LP
1456
1457 switch (s->state) {
1458
1459 case SOCKET_START_PRE:
9e2f7c11 1460 log_warning("%s starting timed out. Terminating.", u->meta.id);
80876c20
LP
1461 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
1462
034c6ed7 1463 case SOCKET_START_POST:
9e2f7c11 1464 log_warning("%s starting timed out. Stopping.", u->meta.id);
034c6ed7
LP
1465 socket_enter_stop_pre(s, false);
1466 break;
1467
1468 case SOCKET_STOP_PRE:
9e2f7c11 1469 log_warning("%s stopping timed out. Terminating.", u->meta.id);
034c6ed7
LP
1470 socket_enter_signal(s, SOCKET_STOP_PRE_SIGTERM, false);
1471 break;
1472
1473 case SOCKET_STOP_PRE_SIGTERM:
9e2f7c11 1474 log_warning("%s stopping timed out. Killing.", u->meta.id);
034c6ed7
LP
1475 socket_enter_signal(s, SOCKET_STOP_PRE_SIGKILL, false);
1476 break;
1477
1478 case SOCKET_STOP_PRE_SIGKILL:
9e2f7c11 1479 log_warning("%s still around after SIGKILL. Ignoring.", u->meta.id);
034c6ed7
LP
1480 socket_enter_stop_post(s, false);
1481 break;
1482
1483 case SOCKET_STOP_POST:
9e2f7c11 1484 log_warning("%s stopping timed out (2). Terminating.", u->meta.id);
80876c20 1485 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
034c6ed7
LP
1486 break;
1487
80876c20 1488 case SOCKET_FINAL_SIGTERM:
9e2f7c11 1489 log_warning("%s stopping timed out (2). Killing.", u->meta.id);
80876c20 1490 socket_enter_signal(s, SOCKET_FINAL_SIGKILL, false);
034c6ed7
LP
1491 break;
1492
80876c20 1493 case SOCKET_FINAL_SIGKILL:
18c78fb1 1494 log_warning("%s still around after SIGKILL (2). Entering maintenance mode.", u->meta.id);
034c6ed7
LP
1495 socket_enter_dead(s, false);
1496 break;
1497
1498 default:
1499 assert_not_reached("Timeout at wrong time.");
1500 }
5cb5a6ff
LP
1501}
1502
44d8db9e
LP
1503int socket_collect_fds(Socket *s, int **fds, unsigned *n_fds) {
1504 int *rfds;
1505 unsigned rn_fds, k;
1506 SocketPort *p;
1507
1508 assert(s);
1509 assert(fds);
1510 assert(n_fds);
1511
1512 /* Called from the service code for requesting our fds */
1513
1514 rn_fds = 0;
1515 LIST_FOREACH(port, p, s->ports)
1516 if (p->fd >= 0)
1517 rn_fds++;
1518
1519 if (!(rfds = new(int, rn_fds)) < 0)
1520 return -ENOMEM;
1521
1522 k = 0;
1523 LIST_FOREACH(port, p, s->ports)
1524 if (p->fd >= 0)
1525 rfds[k++] = p->fd;
1526
1527 assert(k == rn_fds);
1528
1529 *fds = rfds;
1530 *n_fds = rn_fds;
1531
1532 return 0;
1533}
1534
ceee3d82
LP
1535void socket_notify_service_dead(Socket *s) {
1536 assert(s);
1537
6cf6bbc2
LP
1538 /* The service is dead. Dang!
1539 *
1540 * This is strictly for one-instance-for-all-connections
1541 * services. */
ceee3d82
LP
1542
1543 if (s->state == SOCKET_RUNNING) {
9e2f7c11 1544 log_debug("%s got notified about service death.", s->meta.id);
ceee3d82
LP
1545 socket_enter_listening(s);
1546 }
1547}
1548
6cf6bbc2
LP
1549void socket_connection_unref(Socket *s) {
1550 assert(s);
1551
1552 /* The service is dead. Yay!
1553 *
1554 * This is strictly for one-onstance-per-connection
1555 * services. */
1556
1557 assert(s->n_connections > 0);
1558 s->n_connections--;
1559
1560 log_debug("%s: One connection closed, %u left.", s->meta.id, s->n_connections);
1561}
1562
a16e1123
LP
1563static const char* const socket_state_table[_SOCKET_STATE_MAX] = {
1564 [SOCKET_DEAD] = "dead",
1565 [SOCKET_START_PRE] = "start-pre",
1566 [SOCKET_START_POST] = "start-post",
1567 [SOCKET_LISTENING] = "listening",
1568 [SOCKET_RUNNING] = "running",
1569 [SOCKET_STOP_PRE] = "stop-pre",
1570 [SOCKET_STOP_PRE_SIGTERM] = "stop-pre-sigterm",
1571 [SOCKET_STOP_PRE_SIGKILL] = "stop-pre-sigkill",
1572 [SOCKET_STOP_POST] = "stop-post",
1573 [SOCKET_FINAL_SIGTERM] = "final-sigterm",
1574 [SOCKET_FINAL_SIGKILL] = "final-sigkill",
18c78fb1 1575 [SOCKET_MAINTENANCE] = "maintenance"
a16e1123
LP
1576};
1577
1578DEFINE_STRING_TABLE_LOOKUP(socket_state, SocketState);
1579
1580static const char* const socket_exec_command_table[_SOCKET_EXEC_COMMAND_MAX] = {
1581 [SOCKET_EXEC_START_PRE] = "StartPre",
1582 [SOCKET_EXEC_START_POST] = "StartPost",
1583 [SOCKET_EXEC_STOP_PRE] = "StopPre",
1584 [SOCKET_EXEC_STOP_POST] = "StopPost"
1585};
1586
1587DEFINE_STRING_TABLE_LOOKUP(socket_exec_command, SocketExecCommand);
1588
87f0e418 1589const UnitVTable socket_vtable = {
5cb5a6ff
LP
1590 .suffix = ".socket",
1591
034c6ed7
LP
1592 .init = socket_init,
1593 .done = socket_done,
a16e1123
LP
1594 .load = socket_load,
1595
1596 .coldplug = socket_coldplug,
034c6ed7 1597
5cb5a6ff
LP
1598 .dump = socket_dump,
1599
542563ba
LP
1600 .start = socket_start,
1601 .stop = socket_stop,
5cb5a6ff 1602
a16e1123
LP
1603 .serialize = socket_serialize,
1604 .deserialize_item = socket_deserialize_item,
1605
5cb5a6ff 1606 .active_state = socket_active_state,
10a94420 1607 .sub_state_to_string = socket_sub_state_to_string,
5cb5a6ff 1608
6cf6bbc2
LP
1609 .check_gc = socket_check_gc,
1610
9152c765 1611 .fd_event = socket_fd_event,
034c6ed7 1612 .sigchld_event = socket_sigchld_event,
4139c1b2
LP
1613 .timer_event = socket_timer_event,
1614
1615 .bus_message_handler = bus_socket_message_handler
5cb5a6ff 1616};