1 | /* |
2 | * systemd socket activation support |
3 | * |
4 | * Copyright 2017 Red Hat, Inc. and/or its affiliates |
5 | * |
6 | * Authors: |
7 | * Richard W.M. Jones <rjones@redhat.com> |
8 | * |
9 | * This work is licensed under the terms of the GNU GPL, version 2 or later. |
10 | * See the COPYING file in the top-level directory. |
11 | */ |
12 | |
13 | #include "qemu/osdep.h" |
14 | #include "qemu/systemd.h" |
15 | #include "qemu/cutils.h" |
16 | #include "qemu/error-report.h" |
17 | |
18 | #ifndef _WIN32 |
19 | unsigned int check_socket_activation(void) |
20 | { |
21 | const char *s; |
22 | unsigned long pid; |
23 | unsigned long nr_fds; |
24 | unsigned int i; |
25 | int fd; |
26 | int err; |
27 | |
28 | s = getenv("LISTEN_PID" ); |
29 | if (s == NULL) { |
30 | return 0; |
31 | } |
32 | err = qemu_strtoul(s, NULL, 10, &pid); |
33 | if (err) { |
34 | return 0; |
35 | } |
36 | if (pid != getpid()) { |
37 | return 0; |
38 | } |
39 | |
40 | s = getenv("LISTEN_FDS" ); |
41 | if (s == NULL) { |
42 | return 0; |
43 | } |
44 | err = qemu_strtoul(s, NULL, 10, &nr_fds); |
45 | if (err) { |
46 | return 0; |
47 | } |
48 | assert(nr_fds <= UINT_MAX); |
49 | |
50 | /* So these are not passed to any child processes we might start. */ |
51 | unsetenv("LISTEN_FDS" ); |
52 | unsetenv("LISTEN_PID" ); |
53 | |
54 | /* So the file descriptors don't leak into child processes. */ |
55 | for (i = 0; i < nr_fds; ++i) { |
56 | fd = FIRST_SOCKET_ACTIVATION_FD + i; |
57 | if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1) { |
58 | /* If we cannot set FD_CLOEXEC then it probably means the file |
59 | * descriptor is invalid, so socket activation has gone wrong |
60 | * and we should exit. |
61 | */ |
62 | error_report("Socket activation failed: " |
63 | "invalid file descriptor fd = %d: %m" , |
64 | fd); |
65 | exit(EXIT_FAILURE); |
66 | } |
67 | } |
68 | |
69 | return (unsigned int) nr_fds; |
70 | } |
71 | |
72 | #else /* !_WIN32 */ |
73 | unsigned int check_socket_activation(void) |
74 | { |
75 | return 0; |
76 | } |
77 | #endif |
78 | |