Blame


1 a596b957 2022-07-14 tracey /*
2 a596b957 2022-07-14 tracey * Copyright (c) 2016, 2019, 2020-2021 Tracey Emery <tracey@traceyemery.net>
3 a596b957 2022-07-14 tracey * Copyright (c) 2015 Mike Larkin <mlarkin@openbsd.org>
4 a596b957 2022-07-14 tracey * Copyright (c) 2013 David Gwynne <dlg@openbsd.org>
5 a596b957 2022-07-14 tracey * Copyright (c) 2013 Florian Obser <florian@openbsd.org>
6 a596b957 2022-07-14 tracey *
7 a596b957 2022-07-14 tracey * Permission to use, copy, modify, and distribute this software for any
8 a596b957 2022-07-14 tracey * purpose with or without fee is hereby granted, provided that the above
9 a596b957 2022-07-14 tracey * copyright notice and this permission notice appear in all copies.
10 a596b957 2022-07-14 tracey *
11 a596b957 2022-07-14 tracey * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12 a596b957 2022-07-14 tracey * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13 a596b957 2022-07-14 tracey * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14 a596b957 2022-07-14 tracey * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15 a596b957 2022-07-14 tracey * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16 a596b957 2022-07-14 tracey * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17 a596b957 2022-07-14 tracey * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 a596b957 2022-07-14 tracey */
19 a596b957 2022-07-14 tracey
20 a596b957 2022-07-14 tracey #include <sys/param.h>
21 a596b957 2022-07-14 tracey #include <sys/ioctl.h>
22 a596b957 2022-07-14 tracey #include <sys/queue.h>
23 a596b957 2022-07-14 tracey #include <sys/wait.h>
24 a596b957 2022-07-14 tracey #include <sys/uio.h>
25 a596b957 2022-07-14 tracey #include <sys/resource.h>
26 a596b957 2022-07-14 tracey #include <sys/socket.h>
27 a596b957 2022-07-14 tracey #include <sys/stat.h>
28 a596b957 2022-07-14 tracey #include <sys/time.h>
29 a596b957 2022-07-14 tracey #include <sys/types.h>
30 a596b957 2022-07-14 tracey #include <sys/mman.h>
31 a596b957 2022-07-14 tracey #include <sys/un.h>
32 a596b957 2022-07-14 tracey
33 a596b957 2022-07-14 tracey #include <net/if.h>
34 a596b957 2022-07-14 tracey #include <netinet/in.h>
35 a596b957 2022-07-14 tracey
36 a596b957 2022-07-14 tracey #include <errno.h>
37 a596b957 2022-07-14 tracey #include <event.h>
38 a596b957 2022-07-14 tracey #include <fcntl.h>
39 a596b957 2022-07-14 tracey #include <ifaddrs.h>
40 a596b957 2022-07-14 tracey #include <imsg.h>
41 a596b957 2022-07-14 tracey #include <limits.h>
42 a596b957 2022-07-14 tracey #include <netdb.h>
43 a596b957 2022-07-14 tracey #include <poll.h>
44 a596b957 2022-07-14 tracey #include <pwd.h>
45 a596b957 2022-07-14 tracey #include <stddef.h>
46 a596b957 2022-07-14 tracey #include <stdio.h>
47 a596b957 2022-07-14 tracey #include <stdlib.h>
48 a596b957 2022-07-14 tracey #include <string.h>
49 a596b957 2022-07-14 tracey #include <unistd.h>
50 a596b957 2022-07-14 tracey #include <util.h>
51 a596b957 2022-07-14 tracey
52 a596b957 2022-07-14 tracey #include "got_error.h"
53 a596b957 2022-07-14 tracey #include "got_opentemp.h"
54 df2d3cd2 2023-03-11 op #include "got_reference.h"
55 b5c757f5 2022-09-01 stsp #include "got_repository.h"
56 6fe3b58a 2023-11-14 stsp #include "got_privsep.h"
57 a596b957 2022-07-14 tracey
58 a596b957 2022-07-14 tracey #include "gotwebd.h"
59 ed619ca0 2022-12-14 op #include "tmpl.h"
60 a596b957 2022-07-14 tracey
61 a596b957 2022-07-14 tracey #define SOCKS_BACKLOG 5
62 a596b957 2022-07-14 tracey #define MAXIMUM(a, b) (((a) > (b)) ? (a) : (b))
63 a596b957 2022-07-14 tracey
64 a596b957 2022-07-14 tracey volatile int client_cnt;
65 a596b957 2022-07-14 tracey
66 026ac2c4 2022-08-20 stsp static struct timeval timeout = { TIMEOUT_DEFAULT, 0 };
67 a596b957 2022-07-14 tracey
68 026ac2c4 2022-08-20 stsp static void sockets_sighdlr(int, short, void *);
69 26678add 2023-11-16 op static void sockets_shutdown(void);
70 026ac2c4 2022-08-20 stsp static void sockets_launch(void);
71 026ac2c4 2022-08-20 stsp static void sockets_purge(struct gotwebd *);
72 026ac2c4 2022-08-20 stsp static void sockets_accept_paused(int, short, void *);
73 026ac2c4 2022-08-20 stsp static void sockets_rlimit(int);
74 a596b957 2022-07-14 tracey
75 26678add 2023-11-16 op static void sockets_dispatch_main(int, short, void *);
76 26678add 2023-11-16 op static int sockets_unix_socket_listen(struct gotwebd *, struct socket *);
77 0c64c2f8 2023-11-15 op static int sockets_create_socket(struct address *);
78 026ac2c4 2022-08-20 stsp static int sockets_accept_reserve(int, struct sockaddr *, socklen_t *,
79 026ac2c4 2022-08-20 stsp int, volatile int *);
80 a596b957 2022-07-14 tracey
81 026ac2c4 2022-08-20 stsp static struct socket *sockets_conf_new_socket_unix(struct gotwebd *,
82 026ac2c4 2022-08-20 stsp struct server *, int);
83 026ac2c4 2022-08-20 stsp static struct socket *sockets_conf_new_socket_fcgi(struct gotwebd *,
84 026ac2c4 2022-08-20 stsp struct server *, int, struct address *);
85 a596b957 2022-07-14 tracey
86 a596b957 2022-07-14 tracey int cgi_inflight = 0;
87 a596b957 2022-07-14 tracey
88 a596b957 2022-07-14 tracey void
89 26678add 2023-11-16 op sockets(struct gotwebd *env, int fd)
90 a596b957 2022-07-14 tracey {
91 736fac90 2024-02-10 op struct event sighup, sigint, sigusr1, sigchld, sigterm;
92 a596b957 2022-07-14 tracey
93 26678add 2023-11-16 op event_init();
94 a596b957 2022-07-14 tracey
95 a596b957 2022-07-14 tracey sockets_rlimit(-1);
96 a596b957 2022-07-14 tracey
97 26678add 2023-11-16 op if (config_init(env) == -1)
98 26678add 2023-11-16 op fatal("failed to initialize configuration");
99 a596b957 2022-07-14 tracey
100 26678add 2023-11-16 op if ((env->iev_parent = malloc(sizeof(*env->iev_parent))) == NULL)
101 26678add 2023-11-16 op fatal("malloc");
102 26678add 2023-11-16 op imsg_init(&env->iev_parent->ibuf, fd);
103 26678add 2023-11-16 op env->iev_parent->handler = sockets_dispatch_main;
104 26678add 2023-11-16 op env->iev_parent->data = env->iev_parent;
105 26678add 2023-11-16 op event_set(&env->iev_parent->ev, fd, EV_READ, sockets_dispatch_main,
106 26678add 2023-11-16 op env->iev_parent);
107 26678add 2023-11-16 op event_add(&env->iev_parent->ev, NULL);
108 8def9425 2023-11-16 op
109 8def9425 2023-11-16 op signal(SIGPIPE, SIG_IGN);
110 26678add 2023-11-16 op
111 56a9d315 2023-11-18 op signal_set(&sighup, SIGHUP, sockets_sighdlr, env);
112 26678add 2023-11-16 op signal_add(&sighup, NULL);
113 736fac90 2024-02-10 op signal_set(&sigint, SIGINT, sockets_sighdlr, env);
114 736fac90 2024-02-10 op signal_add(&sigint, NULL);
115 56a9d315 2023-11-18 op signal_set(&sigusr1, SIGUSR1, sockets_sighdlr, env);
116 26678add 2023-11-16 op signal_add(&sigusr1, NULL);
117 26678add 2023-11-16 op signal_set(&sigchld, SIGCHLD, sockets_sighdlr, env);
118 26678add 2023-11-16 op signal_add(&sigchld, NULL);
119 736fac90 2024-02-10 op signal_set(&sigterm, SIGTERM, sockets_sighdlr, env);
120 736fac90 2024-02-10 op signal_add(&sigterm, NULL);
121 26678add 2023-11-16 op
122 a596b957 2022-07-14 tracey #ifndef PROFILE
123 63c6b10f 2023-11-14 stsp if (pledge("stdio rpath inet recvfd proc exec sendfd unveil",
124 63c6b10f 2023-11-14 stsp NULL) == -1)
125 a596b957 2022-07-14 tracey fatal("pledge");
126 a596b957 2022-07-14 tracey #endif
127 26678add 2023-11-16 op
128 26678add 2023-11-16 op event_dispatch();
129 26678add 2023-11-16 op sockets_shutdown();
130 a596b957 2022-07-14 tracey }
131 a596b957 2022-07-14 tracey
132 a596b957 2022-07-14 tracey void
133 a596b957 2022-07-14 tracey sockets_parse_sockets(struct gotwebd *env)
134 a596b957 2022-07-14 tracey {
135 a596b957 2022-07-14 tracey struct server *srv;
136 610dd8c9 2022-08-19 stsp struct address *a;
137 4fcc9f74 2022-08-16 stsp struct socket *new_sock = NULL;
138 610dd8c9 2022-08-19 stsp int sock_id = 1;
139 a596b957 2022-07-14 tracey
140 2ad48e9a 2022-08-16 stsp TAILQ_FOREACH(srv, &env->servers, entry) {
141 a596b957 2022-07-14 tracey if (srv->unix_socket) {
142 610dd8c9 2022-08-19 stsp new_sock = sockets_conf_new_socket_unix(env, srv,
143 610dd8c9 2022-08-19 stsp sock_id);
144 4fcc9f74 2022-08-16 stsp if (new_sock) {
145 610dd8c9 2022-08-19 stsp sock_id++;
146 4fcc9f74 2022-08-16 stsp TAILQ_INSERT_TAIL(&env->sockets, new_sock,
147 4fcc9f74 2022-08-16 stsp entry);
148 a596b957 2022-07-14 tracey }
149 610dd8c9 2022-08-19 stsp }
150 a596b957 2022-07-14 tracey
151 610dd8c9 2022-08-19 stsp if (srv->fcgi_socket) {
152 610dd8c9 2022-08-19 stsp if (TAILQ_EMPTY(&srv->al)) {
153 610dd8c9 2022-08-19 stsp fatalx("%s: server %s has no IP addresses to "
154 610dd8c9 2022-08-19 stsp "listen for FCGI connections", __func__,
155 610dd8c9 2022-08-19 stsp srv->name);
156 4fcc9f74 2022-08-16 stsp }
157 610dd8c9 2022-08-19 stsp TAILQ_FOREACH(a, &srv->al, entry) {
158 610dd8c9 2022-08-19 stsp if (a->ss.ss_family != AF_INET &&
159 610dd8c9 2022-08-19 stsp a->ss.ss_family != AF_INET6)
160 610dd8c9 2022-08-19 stsp continue;
161 610dd8c9 2022-08-19 stsp new_sock = sockets_conf_new_socket_fcgi(env,
162 610dd8c9 2022-08-19 stsp srv, sock_id, a);
163 610dd8c9 2022-08-19 stsp if (new_sock) {
164 610dd8c9 2022-08-19 stsp sock_id++;
165 610dd8c9 2022-08-19 stsp TAILQ_INSERT_TAIL(&env->sockets,
166 610dd8c9 2022-08-19 stsp new_sock, entry);
167 610dd8c9 2022-08-19 stsp }
168 a596b957 2022-07-14 tracey }
169 a596b957 2022-07-14 tracey }
170 a596b957 2022-07-14 tracey }
171 a596b957 2022-07-14 tracey }
172 a596b957 2022-07-14 tracey
173 026ac2c4 2022-08-20 stsp static struct socket *
174 610dd8c9 2022-08-19 stsp sockets_conf_new_socket_unix(struct gotwebd *env, struct server *srv, int id)
175 a596b957 2022-07-14 tracey {
176 a596b957 2022-07-14 tracey struct socket *sock;
177 a596b957 2022-07-14 tracey int n;
178 a596b957 2022-07-14 tracey
179 a596b957 2022-07-14 tracey if ((sock = calloc(1, sizeof(*sock))) == NULL)
180 a596b957 2022-07-14 tracey fatalx("%s: calloc", __func__);
181 a596b957 2022-07-14 tracey
182 a596b957 2022-07-14 tracey sock->conf.id = id;
183 a596b957 2022-07-14 tracey sock->fd = -1;
184 610dd8c9 2022-08-19 stsp sock->conf.af_type = AF_UNIX;
185 a596b957 2022-07-14 tracey
186 610dd8c9 2022-08-19 stsp if (strlcpy(sock->conf.unix_socket_name,
187 610dd8c9 2022-08-19 stsp srv->unix_socket_name,
188 610dd8c9 2022-08-19 stsp sizeof(sock->conf.unix_socket_name)) >=
189 610dd8c9 2022-08-19 stsp sizeof(sock->conf.unix_socket_name)) {
190 610dd8c9 2022-08-19 stsp free(sock);
191 610dd8c9 2022-08-19 stsp fatalx("%s: strlcpy", __func__);
192 4fcc9f74 2022-08-16 stsp }
193 a596b957 2022-07-14 tracey
194 a596b957 2022-07-14 tracey n = snprintf(sock->conf.name, GOTWEBD_MAXTEXT, "%s_parent",
195 a596b957 2022-07-14 tracey srv->name);
196 438d0cc3 2022-08-16 op if (n < 0 || (size_t)n >= GOTWEBD_MAXTEXT) {
197 a596b957 2022-07-14 tracey free(sock);
198 a596b957 2022-07-14 tracey fatalx("%s: snprintf", __func__);
199 a596b957 2022-07-14 tracey }
200 a596b957 2022-07-14 tracey
201 a596b957 2022-07-14 tracey if (strlcpy(sock->conf.srv_name, srv->name,
202 a596b957 2022-07-14 tracey sizeof(sock->conf.srv_name)) >= sizeof(sock->conf.srv_name)) {
203 a596b957 2022-07-14 tracey free(sock);
204 a596b957 2022-07-14 tracey fatalx("%s: strlcpy", __func__);
205 a596b957 2022-07-14 tracey }
206 a596b957 2022-07-14 tracey
207 610dd8c9 2022-08-19 stsp return sock;
208 610dd8c9 2022-08-19 stsp }
209 4fcc9f74 2022-08-16 stsp
210 026ac2c4 2022-08-20 stsp static struct socket *
211 610dd8c9 2022-08-19 stsp sockets_conf_new_socket_fcgi(struct gotwebd *env, struct server *srv, int id,
212 610dd8c9 2022-08-19 stsp struct address *a)
213 610dd8c9 2022-08-19 stsp {
214 610dd8c9 2022-08-19 stsp struct socket *sock;
215 610dd8c9 2022-08-19 stsp struct address *acp;
216 610dd8c9 2022-08-19 stsp int n;
217 a596b957 2022-07-14 tracey
218 610dd8c9 2022-08-19 stsp if ((sock = calloc(1, sizeof(*sock))) == NULL)
219 610dd8c9 2022-08-19 stsp fatalx("%s: calloc", __func__);
220 4fcc9f74 2022-08-16 stsp
221 610dd8c9 2022-08-19 stsp sock->conf.id = id;
222 610dd8c9 2022-08-19 stsp sock->fd = -1;
223 610dd8c9 2022-08-19 stsp sock->conf.af_type = a->ss.ss_family;
224 610dd8c9 2022-08-19 stsp
225 67d8de2a 2022-08-29 stsp sock->conf.fcgi_socket_port = a->port;
226 610dd8c9 2022-08-19 stsp
227 610dd8c9 2022-08-19 stsp n = snprintf(sock->conf.name, GOTWEBD_MAXTEXT, "%s_parent",
228 610dd8c9 2022-08-19 stsp srv->name);
229 610dd8c9 2022-08-19 stsp if (n < 0 || (size_t)n >= GOTWEBD_MAXTEXT) {
230 4fcc9f74 2022-08-16 stsp free(sock);
231 610dd8c9 2022-08-19 stsp fatalx("%s: snprintf", __func__);
232 4fcc9f74 2022-08-16 stsp }
233 4fcc9f74 2022-08-16 stsp
234 610dd8c9 2022-08-19 stsp if (strlcpy(sock->conf.srv_name, srv->name,
235 610dd8c9 2022-08-19 stsp sizeof(sock->conf.srv_name)) >= sizeof(sock->conf.srv_name)) {
236 610dd8c9 2022-08-19 stsp free(sock);
237 610dd8c9 2022-08-19 stsp fatalx("%s: strlcpy", __func__);
238 610dd8c9 2022-08-19 stsp }
239 610dd8c9 2022-08-19 stsp
240 610dd8c9 2022-08-19 stsp acp = &sock->conf.addr;
241 610dd8c9 2022-08-19 stsp
242 610dd8c9 2022-08-19 stsp memcpy(&acp->ss, &a->ss, sizeof(acp->ss));
243 cdfd248a 2023-11-15 op acp->slen = a->slen;
244 0c64c2f8 2023-11-15 op acp->ai_family = a->ai_family;
245 0c64c2f8 2023-11-15 op acp->ai_socktype = a->ai_socktype;
246 0c64c2f8 2023-11-15 op acp->ai_protocol = a->ai_protocol;
247 610dd8c9 2022-08-19 stsp acp->port = a->port;
248 4448825a 2023-06-16 op if (*a->ifname != '\0') {
249 610dd8c9 2022-08-19 stsp if (strlcpy(acp->ifname, a->ifname,
250 610dd8c9 2022-08-19 stsp sizeof(acp->ifname)) >= sizeof(acp->ifname)) {
251 610dd8c9 2022-08-19 stsp fatalx("%s: interface name truncated",
252 610dd8c9 2022-08-19 stsp __func__);
253 610dd8c9 2022-08-19 stsp }
254 610dd8c9 2022-08-19 stsp }
255 610dd8c9 2022-08-19 stsp
256 a596b957 2022-07-14 tracey return (sock);
257 a596b957 2022-07-14 tracey }
258 a596b957 2022-07-14 tracey
259 026ac2c4 2022-08-20 stsp static void
260 a596b957 2022-07-14 tracey sockets_launch(void)
261 a596b957 2022-07-14 tracey {
262 a596b957 2022-07-14 tracey struct socket *sock;
263 6fe3b58a 2023-11-14 stsp struct server *srv;
264 6fe3b58a 2023-11-14 stsp const struct got_error *error;
265 a596b957 2022-07-14 tracey
266 2ad48e9a 2022-08-16 stsp TAILQ_FOREACH(sock, &gotwebd_env->sockets, entry) {
267 a596b957 2022-07-14 tracey log_debug("%s: configuring socket %d (%d)", __func__,
268 a596b957 2022-07-14 tracey sock->conf.id, sock->fd);
269 a596b957 2022-07-14 tracey
270 a596b957 2022-07-14 tracey event_set(&sock->ev, sock->fd, EV_READ | EV_PERSIST,
271 a596b957 2022-07-14 tracey sockets_socket_accept, sock);
272 a596b957 2022-07-14 tracey
273 a596b957 2022-07-14 tracey if (event_add(&sock->ev, NULL))
274 a596b957 2022-07-14 tracey fatalx("event add sock");
275 a596b957 2022-07-14 tracey
276 a596b957 2022-07-14 tracey evtimer_set(&sock->pause, sockets_accept_paused, sock);
277 a596b957 2022-07-14 tracey
278 a596b957 2022-07-14 tracey log_debug("%s: running socket listener %d", __func__,
279 a596b957 2022-07-14 tracey sock->conf.id);
280 a596b957 2022-07-14 tracey }
281 6fe3b58a 2023-11-14 stsp
282 6fe3b58a 2023-11-14 stsp TAILQ_FOREACH(srv, &gotwebd_env->servers, entry) {
283 6fe3b58a 2023-11-14 stsp if (unveil(srv->repos_path, "r") == -1)
284 6fe3b58a 2023-11-14 stsp fatal("unveil %s", srv->repos_path);
285 6fe3b58a 2023-11-14 stsp }
286 6fe3b58a 2023-11-14 stsp
287 6fe3b58a 2023-11-14 stsp error = got_privsep_unveil_exec_helpers();
288 6fe3b58a 2023-11-14 stsp if (error)
289 6fe3b58a 2023-11-14 stsp fatal("%s", error->msg);
290 6fe3b58a 2023-11-14 stsp
291 6fe3b58a 2023-11-14 stsp if (unveil(NULL, NULL) == -1)
292 6fe3b58a 2023-11-14 stsp fatal("unveil");
293 a596b957 2022-07-14 tracey }
294 a596b957 2022-07-14 tracey
295 026ac2c4 2022-08-20 stsp static void
296 a596b957 2022-07-14 tracey sockets_purge(struct gotwebd *env)
297 a596b957 2022-07-14 tracey {
298 a596b957 2022-07-14 tracey struct socket *sock, *tsock;
299 a596b957 2022-07-14 tracey
300 a596b957 2022-07-14 tracey /* shutdown and remove sockets */
301 2ad48e9a 2022-08-16 stsp TAILQ_FOREACH_SAFE(sock, &env->sockets, entry, tsock) {
302 a596b957 2022-07-14 tracey if (event_initialized(&sock->ev))
303 a596b957 2022-07-14 tracey event_del(&sock->ev);
304 a596b957 2022-07-14 tracey if (evtimer_initialized(&sock->evt))
305 a596b957 2022-07-14 tracey evtimer_del(&sock->evt);
306 a596b957 2022-07-14 tracey if (evtimer_initialized(&sock->pause))
307 a596b957 2022-07-14 tracey evtimer_del(&sock->pause);
308 a596b957 2022-07-14 tracey if (sock->fd != -1)
309 a596b957 2022-07-14 tracey close(sock->fd);
310 2ad48e9a 2022-08-16 stsp TAILQ_REMOVE(&env->sockets, sock, entry);
311 a596b957 2022-07-14 tracey }
312 a596b957 2022-07-14 tracey }
313 a596b957 2022-07-14 tracey
314 26678add 2023-11-16 op static void
315 26678add 2023-11-16 op sockets_dispatch_main(int fd, short event, void *arg)
316 a596b957 2022-07-14 tracey {
317 26678add 2023-11-16 op struct imsgev *iev = arg;
318 26678add 2023-11-16 op struct imsgbuf *ibuf;
319 26678add 2023-11-16 op struct imsg imsg;
320 26678add 2023-11-16 op struct gotwebd *env = gotwebd_env;
321 26678add 2023-11-16 op ssize_t n;
322 26678add 2023-11-16 op int shut = 0;
323 a596b957 2022-07-14 tracey
324 26678add 2023-11-16 op ibuf = &iev->ibuf;
325 26678add 2023-11-16 op
326 26678add 2023-11-16 op if (event & EV_READ) {
327 26678add 2023-11-16 op if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
328 26678add 2023-11-16 op fatal("imsg_read error");
329 26678add 2023-11-16 op if (n == 0) /* Connection closed */
330 26678add 2023-11-16 op shut = 1;
331 a596b957 2022-07-14 tracey }
332 26678add 2023-11-16 op if (event & EV_WRITE) {
333 26678add 2023-11-16 op if ((n = msgbuf_write(&ibuf->w)) == -1 && errno != EAGAIN)
334 26678add 2023-11-16 op fatal("msgbuf_write");
335 26678add 2023-11-16 op if (n == 0) /* Connection closed */
336 26678add 2023-11-16 op shut = 1;
337 26678add 2023-11-16 op }
338 a596b957 2022-07-14 tracey
339 26678add 2023-11-16 op for (;;) {
340 26678add 2023-11-16 op if ((n = imsg_get(ibuf, &imsg)) == -1)
341 26678add 2023-11-16 op fatal("imsg_get");
342 26678add 2023-11-16 op if (n == 0) /* No more messages. */
343 26678add 2023-11-16 op break;
344 26678add 2023-11-16 op
345 26678add 2023-11-16 op switch (imsg.hdr.type) {
346 26678add 2023-11-16 op case IMSG_CFG_SRV:
347 26678add 2023-11-16 op config_getserver(env, &imsg);
348 26678add 2023-11-16 op break;
349 26678add 2023-11-16 op case IMSG_CFG_SOCK:
350 26678add 2023-11-16 op config_getsock(env, &imsg);
351 26678add 2023-11-16 op break;
352 26678add 2023-11-16 op case IMSG_CFG_FD:
353 26678add 2023-11-16 op config_getfd(env, &imsg);
354 26678add 2023-11-16 op break;
355 26678add 2023-11-16 op case IMSG_CFG_DONE:
356 26678add 2023-11-16 op config_getcfg(env, &imsg);
357 26678add 2023-11-16 op break;
358 26678add 2023-11-16 op case IMSG_CTL_START:
359 26678add 2023-11-16 op sockets_launch();
360 26678add 2023-11-16 op break;
361 26678add 2023-11-16 op default:
362 26678add 2023-11-16 op fatalx("%s: unknown imsg type %d", __func__,
363 26678add 2023-11-16 op imsg.hdr.type);
364 26678add 2023-11-16 op }
365 26678add 2023-11-16 op
366 26678add 2023-11-16 op imsg_free(&imsg);
367 a596b957 2022-07-14 tracey }
368 a596b957 2022-07-14 tracey
369 26678add 2023-11-16 op if (!shut)
370 26678add 2023-11-16 op imsg_event_add(iev);
371 26678add 2023-11-16 op else {
372 26678add 2023-11-16 op /* This pipe is dead. Remove its event handler */
373 26678add 2023-11-16 op event_del(&iev->ev);
374 26678add 2023-11-16 op event_loopexit(NULL);
375 26678add 2023-11-16 op }
376 a596b957 2022-07-14 tracey }
377 a596b957 2022-07-14 tracey
378 026ac2c4 2022-08-20 stsp static void
379 a596b957 2022-07-14 tracey sockets_sighdlr(int sig, short event, void *arg)
380 a596b957 2022-07-14 tracey {
381 a596b957 2022-07-14 tracey switch (sig) {
382 a596b957 2022-07-14 tracey case SIGHUP:
383 a596b957 2022-07-14 tracey log_info("%s: ignoring SIGHUP", __func__);
384 a596b957 2022-07-14 tracey break;
385 a596b957 2022-07-14 tracey case SIGPIPE:
386 a596b957 2022-07-14 tracey log_info("%s: ignoring SIGPIPE", __func__);
387 a596b957 2022-07-14 tracey break;
388 a596b957 2022-07-14 tracey case SIGUSR1:
389 a596b957 2022-07-14 tracey log_info("%s: ignoring SIGUSR1", __func__);
390 a596b957 2022-07-14 tracey break;
391 a596b957 2022-07-14 tracey case SIGCHLD:
392 736fac90 2024-02-10 op break;
393 736fac90 2024-02-10 op case SIGINT:
394 736fac90 2024-02-10 op case SIGTERM:
395 736fac90 2024-02-10 op sockets_shutdown();
396 a596b957 2022-07-14 tracey break;
397 a596b957 2022-07-14 tracey default:
398 a596b957 2022-07-14 tracey log_info("SIGNAL: %d", sig);
399 a596b957 2022-07-14 tracey fatalx("unexpected signal");
400 a596b957 2022-07-14 tracey }
401 a596b957 2022-07-14 tracey }
402 a596b957 2022-07-14 tracey
403 26678add 2023-11-16 op static void
404 a596b957 2022-07-14 tracey sockets_shutdown(void)
405 a596b957 2022-07-14 tracey {
406 a596b957 2022-07-14 tracey struct server *srv, *tsrv;
407 a596b957 2022-07-14 tracey struct socket *sock, *tsock;
408 a596b957 2022-07-14 tracey
409 a596b957 2022-07-14 tracey sockets_purge(gotwebd_env);
410 a596b957 2022-07-14 tracey
411 a596b957 2022-07-14 tracey /* clean sockets */
412 2ad48e9a 2022-08-16 stsp TAILQ_FOREACH_SAFE(sock, &gotwebd_env->sockets, entry, tsock) {
413 2ad48e9a 2022-08-16 stsp TAILQ_REMOVE(&gotwebd_env->sockets, sock, entry);
414 a596b957 2022-07-14 tracey close(sock->fd);
415 a596b957 2022-07-14 tracey free(sock);
416 a596b957 2022-07-14 tracey }
417 a596b957 2022-07-14 tracey
418 a596b957 2022-07-14 tracey /* clean servers */
419 1632f50a 2023-11-17 stsp TAILQ_FOREACH_SAFE(srv, &gotwebd_env->servers, entry, tsrv)
420 a596b957 2022-07-14 tracey free(srv);
421 a596b957 2022-07-14 tracey
422 a596b957 2022-07-14 tracey free(gotwebd_env);
423 736fac90 2024-02-10 op
424 736fac90 2024-02-10 op exit(0);
425 a596b957 2022-07-14 tracey }
426 a596b957 2022-07-14 tracey
427 a596b957 2022-07-14 tracey int
428 a596b957 2022-07-14 tracey sockets_privinit(struct gotwebd *env, struct socket *sock)
429 a596b957 2022-07-14 tracey {
430 4fcc9f74 2022-08-16 stsp if (sock->conf.af_type == AF_UNIX) {
431 a596b957 2022-07-14 tracey log_debug("%s: initializing unix socket %s", __func__,
432 a596b957 2022-07-14 tracey sock->conf.unix_socket_name);
433 26678add 2023-11-16 op sock->fd = sockets_unix_socket_listen(env, sock);
434 a596b957 2022-07-14 tracey if (sock->fd == -1) {
435 a596b957 2022-07-14 tracey log_warnx("%s: create unix socket failed", __func__);
436 a596b957 2022-07-14 tracey return -1;
437 a596b957 2022-07-14 tracey }
438 a596b957 2022-07-14 tracey }
439 a596b957 2022-07-14 tracey
440 4fcc9f74 2022-08-16 stsp if (sock->conf.af_type == AF_INET || sock->conf.af_type == AF_INET6) {
441 4fcc9f74 2022-08-16 stsp log_debug("%s: initializing %s FCGI socket on port %d for %s",
442 4fcc9f74 2022-08-16 stsp __func__, sock->conf.af_type == AF_INET ? "inet" : "inet6",
443 4fcc9f74 2022-08-16 stsp sock->conf.fcgi_socket_port, sock->conf.name);
444 0c64c2f8 2023-11-15 op sock->fd = sockets_create_socket(&sock->conf.addr);
445 a596b957 2022-07-14 tracey if (sock->fd == -1) {
446 610dd8c9 2022-08-19 stsp log_warnx("%s: create FCGI socket failed", __func__);
447 a596b957 2022-07-14 tracey return -1;
448 a596b957 2022-07-14 tracey }
449 a596b957 2022-07-14 tracey }
450 a596b957 2022-07-14 tracey
451 a596b957 2022-07-14 tracey return 0;
452 a596b957 2022-07-14 tracey }
453 a596b957 2022-07-14 tracey
454 026ac2c4 2022-08-20 stsp static int
455 26678add 2023-11-16 op sockets_unix_socket_listen(struct gotwebd *env, struct socket *sock)
456 a596b957 2022-07-14 tracey {
457 a596b957 2022-07-14 tracey struct sockaddr_un sun;
458 a596b957 2022-07-14 tracey struct socket *tsock;
459 a596b957 2022-07-14 tracey int u_fd = -1;
460 a596b957 2022-07-14 tracey mode_t old_umask, mode;
461 a596b957 2022-07-14 tracey
462 2ad48e9a 2022-08-16 stsp TAILQ_FOREACH(tsock, &env->sockets, entry) {
463 a596b957 2022-07-14 tracey if (strcmp(tsock->conf.unix_socket_name,
464 a596b957 2022-07-14 tracey sock->conf.unix_socket_name) == 0 &&
465 a596b957 2022-07-14 tracey tsock->fd != -1)
466 a596b957 2022-07-14 tracey return (tsock->fd);
467 a596b957 2022-07-14 tracey }
468 a596b957 2022-07-14 tracey
469 a596b957 2022-07-14 tracey u_fd = socket(AF_UNIX, SOCK_STREAM | SOCK_NONBLOCK| SOCK_CLOEXEC, 0);
470 a596b957 2022-07-14 tracey if (u_fd == -1) {
471 a596b957 2022-07-14 tracey log_warn("%s: socket", __func__);
472 a596b957 2022-07-14 tracey return -1;
473 a596b957 2022-07-14 tracey }
474 a596b957 2022-07-14 tracey
475 a596b957 2022-07-14 tracey sun.sun_family = AF_UNIX;
476 a596b957 2022-07-14 tracey if (strlcpy(sun.sun_path, sock->conf.unix_socket_name,
477 a596b957 2022-07-14 tracey sizeof(sun.sun_path)) >= sizeof(sun.sun_path)) {
478 a596b957 2022-07-14 tracey log_warn("%s: %s name too long", __func__,
479 a596b957 2022-07-14 tracey sock->conf.unix_socket_name);
480 a596b957 2022-07-14 tracey close(u_fd);
481 a596b957 2022-07-14 tracey return -1;
482 a596b957 2022-07-14 tracey }
483 a596b957 2022-07-14 tracey
484 a596b957 2022-07-14 tracey if (unlink(sock->conf.unix_socket_name) == -1) {
485 a596b957 2022-07-14 tracey if (errno != ENOENT) {
486 a596b957 2022-07-14 tracey log_warn("%s: unlink %s", __func__,
487 a596b957 2022-07-14 tracey sock->conf.unix_socket_name);
488 a596b957 2022-07-14 tracey close(u_fd);
489 a596b957 2022-07-14 tracey return -1;
490 a596b957 2022-07-14 tracey }
491 a596b957 2022-07-14 tracey }
492 a596b957 2022-07-14 tracey
493 a596b957 2022-07-14 tracey old_umask = umask(S_IXUSR|S_IXGRP|S_IWOTH|S_IROTH|S_IXOTH);
494 a596b957 2022-07-14 tracey mode = S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP;
495 a596b957 2022-07-14 tracey
496 a596b957 2022-07-14 tracey if (bind(u_fd, (struct sockaddr *)&sun, sizeof(sun)) == -1) {
497 a596b957 2022-07-14 tracey log_warn("%s: bind: %s", __func__, sock->conf.unix_socket_name);
498 a596b957 2022-07-14 tracey close(u_fd);
499 a596b957 2022-07-14 tracey (void)umask(old_umask);
500 a596b957 2022-07-14 tracey return -1;
501 a596b957 2022-07-14 tracey }
502 a596b957 2022-07-14 tracey
503 a596b957 2022-07-14 tracey (void)umask(old_umask);
504 a596b957 2022-07-14 tracey
505 a596b957 2022-07-14 tracey if (chmod(sock->conf.unix_socket_name, mode) == -1) {
506 a596b957 2022-07-14 tracey log_warn("%s: chmod", __func__);
507 a596b957 2022-07-14 tracey close(u_fd);
508 a596b957 2022-07-14 tracey (void)unlink(sock->conf.unix_socket_name);
509 a596b957 2022-07-14 tracey return -1;
510 a596b957 2022-07-14 tracey }
511 a596b957 2022-07-14 tracey
512 26678add 2023-11-16 op if (chown(sock->conf.unix_socket_name, env->pw->pw_uid,
513 26678add 2023-11-16 op env->pw->pw_gid) == -1) {
514 a596b957 2022-07-14 tracey log_warn("%s: chown", __func__);
515 a596b957 2022-07-14 tracey close(u_fd);
516 a596b957 2022-07-14 tracey (void)unlink(sock->conf.unix_socket_name);
517 a596b957 2022-07-14 tracey return -1;
518 a596b957 2022-07-14 tracey }
519 a596b957 2022-07-14 tracey
520 a596b957 2022-07-14 tracey if (listen(u_fd, SOCKS_BACKLOG) == -1) {
521 a596b957 2022-07-14 tracey log_warn("%s: listen", __func__);
522 a596b957 2022-07-14 tracey return -1;
523 a596b957 2022-07-14 tracey }
524 a596b957 2022-07-14 tracey
525 a596b957 2022-07-14 tracey return u_fd;
526 a596b957 2022-07-14 tracey }
527 a596b957 2022-07-14 tracey
528 026ac2c4 2022-08-20 stsp static int
529 0c64c2f8 2023-11-15 op sockets_create_socket(struct address *a)
530 a596b957 2022-07-14 tracey {
531 a596b957 2022-07-14 tracey int fd = -1, o_val = 1, flags;
532 a596b957 2022-07-14 tracey
533 0c64c2f8 2023-11-15 op fd = socket(a->ai_family, a->ai_socktype, a->ai_protocol);
534 610dd8c9 2022-08-19 stsp if (fd == -1)
535 610dd8c9 2022-08-19 stsp return -1;
536 a596b957 2022-07-14 tracey
537 610dd8c9 2022-08-19 stsp log_debug("%s: opened socket (%d) for %s", __func__,
538 610dd8c9 2022-08-19 stsp fd, a->ifname);
539 a596b957 2022-07-14 tracey
540 610dd8c9 2022-08-19 stsp if (setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &o_val,
541 610dd8c9 2022-08-19 stsp sizeof(int)) == -1) {
542 610dd8c9 2022-08-19 stsp log_warn("%s: setsockopt error", __func__);
543 610dd8c9 2022-08-19 stsp close(fd);
544 610dd8c9 2022-08-19 stsp return -1;
545 610dd8c9 2022-08-19 stsp }
546 a596b957 2022-07-14 tracey
547 610dd8c9 2022-08-19 stsp /* non-blocking */
548 610dd8c9 2022-08-19 stsp flags = fcntl(fd, F_GETFL);
549 610dd8c9 2022-08-19 stsp flags |= O_NONBLOCK;
550 610dd8c9 2022-08-19 stsp if (fcntl(fd, F_SETFL, flags) == -1) {
551 610dd8c9 2022-08-19 stsp log_info("%s: could not enable non-blocking I/O", __func__);
552 610dd8c9 2022-08-19 stsp close(fd);
553 610dd8c9 2022-08-19 stsp return -1;
554 610dd8c9 2022-08-19 stsp }
555 a596b957 2022-07-14 tracey
556 cdfd248a 2023-11-15 op if (bind(fd, (struct sockaddr *)&a->ss, a->slen) == -1) {
557 610dd8c9 2022-08-19 stsp close(fd);
558 0c64c2f8 2023-11-15 op log_info("%s: can't bind to port %d", __func__, a->port);
559 610dd8c9 2022-08-19 stsp return -1;
560 a596b957 2022-07-14 tracey }
561 a596b957 2022-07-14 tracey
562 610dd8c9 2022-08-19 stsp if (listen(fd, SOMAXCONN) == -1) {
563 610dd8c9 2022-08-19 stsp log_warn("%s, unable to listen on socket", __func__);
564 610dd8c9 2022-08-19 stsp close(fd);
565 610dd8c9 2022-08-19 stsp return -1;
566 610dd8c9 2022-08-19 stsp }
567 610dd8c9 2022-08-19 stsp
568 a596b957 2022-07-14 tracey return (fd);
569 a596b957 2022-07-14 tracey }
570 a596b957 2022-07-14 tracey
571 026ac2c4 2022-08-20 stsp static int
572 a596b957 2022-07-14 tracey sockets_accept_reserve(int sockfd, struct sockaddr *addr, socklen_t *addrlen,
573 a596b957 2022-07-14 tracey int reserve, volatile int *counter)
574 a596b957 2022-07-14 tracey {
575 a596b957 2022-07-14 tracey int ret;
576 a596b957 2022-07-14 tracey
577 a596b957 2022-07-14 tracey if (getdtablecount() + reserve +
578 a596b957 2022-07-14 tracey ((*counter + 1) * FD_NEEDED) >= getdtablesize()) {
579 a596b957 2022-07-14 tracey log_debug("inflight fds exceeded");
580 a596b957 2022-07-14 tracey errno = EMFILE;
581 a596b957 2022-07-14 tracey return -1;
582 a596b957 2022-07-14 tracey }
583 a596b957 2022-07-14 tracey
584 a596b957 2022-07-14 tracey if ((ret = accept4(sockfd, addr, addrlen,
585 a596b957 2022-07-14 tracey SOCK_NONBLOCK | SOCK_CLOEXEC)) > -1) {
586 a596b957 2022-07-14 tracey (*counter)++;
587 a596b957 2022-07-14 tracey log_debug("inflight incremented, now %d", *counter);
588 a596b957 2022-07-14 tracey }
589 a596b957 2022-07-14 tracey
590 a596b957 2022-07-14 tracey return ret;
591 a596b957 2022-07-14 tracey }
592 a596b957 2022-07-14 tracey
593 026ac2c4 2022-08-20 stsp static void
594 a596b957 2022-07-14 tracey sockets_accept_paused(int fd, short events, void *arg)
595 a596b957 2022-07-14 tracey {
596 a596b957 2022-07-14 tracey struct socket *sock = (struct socket *)arg;
597 a596b957 2022-07-14 tracey
598 a596b957 2022-07-14 tracey event_add(&sock->ev, NULL);
599 a596b957 2022-07-14 tracey }
600 a596b957 2022-07-14 tracey
601 a596b957 2022-07-14 tracey void
602 a596b957 2022-07-14 tracey sockets_socket_accept(int fd, short event, void *arg)
603 a596b957 2022-07-14 tracey {
604 a596b957 2022-07-14 tracey struct socket *sock = (struct socket *)arg;
605 a596b957 2022-07-14 tracey struct sockaddr_storage ss;
606 a596b957 2022-07-14 tracey struct timeval backoff;
607 a596b957 2022-07-14 tracey struct request *c = NULL;
608 a596b957 2022-07-14 tracey socklen_t len;
609 a596b957 2022-07-14 tracey int s;
610 a596b957 2022-07-14 tracey
611 a596b957 2022-07-14 tracey backoff.tv_sec = 1;
612 a596b957 2022-07-14 tracey backoff.tv_usec = 0;
613 a596b957 2022-07-14 tracey
614 a596b957 2022-07-14 tracey event_add(&sock->ev, NULL);
615 a596b957 2022-07-14 tracey if (event & EV_TIMEOUT)
616 a596b957 2022-07-14 tracey return;
617 a596b957 2022-07-14 tracey
618 a596b957 2022-07-14 tracey len = sizeof(ss);
619 a596b957 2022-07-14 tracey
620 a596b957 2022-07-14 tracey s = sockets_accept_reserve(fd, (struct sockaddr *)&ss, &len,
621 a596b957 2022-07-14 tracey FD_RESERVE, &cgi_inflight);
622 a596b957 2022-07-14 tracey
623 a596b957 2022-07-14 tracey if (s == -1) {
624 a596b957 2022-07-14 tracey switch (errno) {
625 a596b957 2022-07-14 tracey case EINTR:
626 a596b957 2022-07-14 tracey case EWOULDBLOCK:
627 a596b957 2022-07-14 tracey case ECONNABORTED:
628 a596b957 2022-07-14 tracey return;
629 a596b957 2022-07-14 tracey case EMFILE:
630 a596b957 2022-07-14 tracey case ENFILE:
631 a596b957 2022-07-14 tracey event_del(&sock->ev);
632 a596b957 2022-07-14 tracey evtimer_add(&sock->pause, &backoff);
633 a596b957 2022-07-14 tracey return;
634 a596b957 2022-07-14 tracey default:
635 a596b957 2022-07-14 tracey log_warn("%s: accept", __func__);
636 a596b957 2022-07-14 tracey }
637 a596b957 2022-07-14 tracey }
638 a596b957 2022-07-14 tracey
639 a596b957 2022-07-14 tracey if (client_cnt > GOTWEBD_MAXCLIENTS)
640 a596b957 2022-07-14 tracey goto err;
641 a596b957 2022-07-14 tracey
642 a596b957 2022-07-14 tracey c = calloc(1, sizeof(struct request));
643 a596b957 2022-07-14 tracey if (c == NULL) {
644 ed619ca0 2022-12-14 op log_warn("%s", __func__);
645 ed619ca0 2022-12-14 op close(s);
646 ed619ca0 2022-12-14 op cgi_inflight--;
647 ed619ca0 2022-12-14 op return;
648 ed619ca0 2022-12-14 op }
649 ed619ca0 2022-12-14 op
650 62eab86e 2023-09-13 op c->tp = template(c, &fcgi_write, c->outbuf, sizeof(c->outbuf));
651 ed619ca0 2022-12-14 op if (c->tp == NULL) {
652 a596b957 2022-07-14 tracey log_warn("%s", __func__);
653 a596b957 2022-07-14 tracey close(s);
654 a596b957 2022-07-14 tracey cgi_inflight--;
655 ed619ca0 2022-12-14 op free(c);
656 a596b957 2022-07-14 tracey return;
657 a596b957 2022-07-14 tracey }
658 a596b957 2022-07-14 tracey
659 a596b957 2022-07-14 tracey c->fd = s;
660 a596b957 2022-07-14 tracey c->sock = sock;
661 a596b957 2022-07-14 tracey memcpy(c->priv_fd, sock->priv_fd, sizeof(c->priv_fd));
662 a596b957 2022-07-14 tracey c->buf_pos = 0;
663 a596b957 2022-07-14 tracey c->buf_len = 0;
664 a596b957 2022-07-14 tracey c->request_started = 0;
665 a596b957 2022-07-14 tracey c->sock->client_status = CLIENT_CONNECT;
666 a596b957 2022-07-14 tracey
667 5add7f42 2023-03-10 op event_set(&c->ev, s, EV_READ|EV_PERSIST, fcgi_request, c);
668 a596b957 2022-07-14 tracey event_add(&c->ev, NULL);
669 a596b957 2022-07-14 tracey
670 a596b957 2022-07-14 tracey evtimer_set(&c->tmo, fcgi_timeout, c);
671 a596b957 2022-07-14 tracey evtimer_add(&c->tmo, &timeout);
672 a596b957 2022-07-14 tracey
673 a596b957 2022-07-14 tracey client_cnt++;
674 a596b957 2022-07-14 tracey
675 a596b957 2022-07-14 tracey return;
676 a596b957 2022-07-14 tracey err:
677 a596b957 2022-07-14 tracey cgi_inflight--;
678 a596b957 2022-07-14 tracey close(s);
679 a596b957 2022-07-14 tracey if (c != NULL)
680 a596b957 2022-07-14 tracey free(c);
681 a596b957 2022-07-14 tracey }
682 a596b957 2022-07-14 tracey
683 026ac2c4 2022-08-20 stsp static void
684 a596b957 2022-07-14 tracey sockets_rlimit(int maxfd)
685 a596b957 2022-07-14 tracey {
686 a596b957 2022-07-14 tracey struct rlimit rl;
687 a596b957 2022-07-14 tracey
688 a596b957 2022-07-14 tracey if (getrlimit(RLIMIT_NOFILE, &rl) == -1)
689 a596b957 2022-07-14 tracey fatal("%s: failed to get resource limit", __func__);
690 d6267a62 2022-07-17 naddy log_debug("%s: max open files %llu", __func__,
691 d6267a62 2022-07-17 naddy (unsigned long long)rl.rlim_max);
692 a596b957 2022-07-14 tracey
693 a596b957 2022-07-14 tracey /*
694 a596b957 2022-07-14 tracey * Allow the maximum number of open file descriptors for this
695 a596b957 2022-07-14 tracey * login class (which should be the class "daemon" by default).
696 a596b957 2022-07-14 tracey */
697 a596b957 2022-07-14 tracey if (maxfd == -1)
698 a596b957 2022-07-14 tracey rl.rlim_cur = rl.rlim_max;
699 a596b957 2022-07-14 tracey else
700 a596b957 2022-07-14 tracey rl.rlim_cur = MAXIMUM(rl.rlim_max, (rlim_t)maxfd);
701 a596b957 2022-07-14 tracey if (setrlimit(RLIMIT_NOFILE, &rl) == -1)
702 a596b957 2022-07-14 tracey fatal("%s: failed to set resource limit", __func__);
703 d6267a62 2022-07-17 naddy }