jcs's openbsd hax
openbsd
1/* $OpenBSD: server.c,v 1.130 2026/01/02 08:45:16 rsadowski Exp $ */
2
3/*
4 * Copyright (c) 2006 - 2015 Reyk Floeter <reyk@openbsd.org>
5 *
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 */
18
19#include <sys/types.h>
20#include <sys/queue.h>
21#include <sys/time.h>
22#include <sys/stat.h>
23#include <sys/socket.h>
24#include <sys/uio.h>
25#include <sys/tree.h>
26
27#include <netinet/in.h>
28#include <netinet/tcp.h>
29#include <arpa/inet.h>
30
31#include <stdio.h>
32#include <stdlib.h>
33#include <stdarg.h>
34#include <limits.h>
35#include <errno.h>
36#include <fcntl.h>
37#include <string.h>
38#include <syslog.h>
39#include <unistd.h>
40#include <event.h>
41#include <imsg.h>
42#include <tls.h>
43#include <vis.h>
44
45#include "httpd.h"
46
47#define MINIMUM(a, b) (((a) < (b)) ? (a) : (b))
48
49int server_dispatch_parent(int, struct privsep_proc *,
50 struct imsg *);
51int server_dispatch_logger(int, struct privsep_proc *,
52 struct imsg *);
53void server_shutdown(void);
54
55void server_init(struct privsep *, struct privsep_proc *p, void *);
56void server_launch(void);
57int server_socket(struct sockaddr_storage *, in_port_t,
58 struct server_config *, int, int);
59int server_socket_listen(struct sockaddr_storage *, in_port_t,
60 struct server_config *);
61struct server *server_byid(uint32_t);
62
63int server_tls_init(struct server *);
64void server_tls_readcb(int, short, void *);
65void server_tls_writecb(int, short, void *);
66void server_tls_handshake(int, short, void *);
67
68void server_accept(int, short, void *);
69void server_input(struct client *);
70void server_inflight_dec(struct client *, const char *);
71
72extern void bufferevent_read_pressure_cb(struct evbuffer *, size_t,
73 size_t, void *);
74
75volatile int server_clients;
76volatile int server_inflight = 0;
77uint32_t server_cltid;
78
79static struct privsep_proc procs[] = {
80 { "parent", PROC_PARENT, server_dispatch_parent },
81 { "logger", PROC_LOGGER, server_dispatch_logger }
82};
83
84void
85server(struct privsep *ps, struct privsep_proc *p)
86{
87 proc_run(ps, p, procs, nitems(procs), server_init, NULL);
88 server_http();
89}
90
91void
92server_shutdown(void)
93{
94 config_purge(httpd_env, CONFIG_ALL);
95 usleep(200); /* XXX server needs to shutdown last */
96}
97
98int
99server_privinit(struct server *srv)
100{
101 struct server *s;
102
103 if (srv->srv_conf.flags & SRVFLAG_LOCATION)
104 return (0);
105
106 log_debug("%s: adding server %s", __func__, srv->srv_conf.name);
107
108 /*
109 * There's no need to open a new socket if a server with the
110 * same address already exists.
111 */
112 TAILQ_FOREACH(s, httpd_env->sc_servers, srv_entry) {
113 if (s != srv && s->srv_s != -1 &&
114 s->srv_conf.port == srv->srv_conf.port &&
115 sockaddr_cmp((struct sockaddr *)&s->srv_conf.ss,
116 (struct sockaddr *)&srv->srv_conf.ss,
117 s->srv_conf.prefixlen) == 0)
118 return (0);
119 }
120
121 /* Open listening socket in the privileged process */
122 if ((srv->srv_s = server_socket_listen(&srv->srv_conf.ss,
123 srv->srv_conf.port, &srv->srv_conf)) == -1)
124 return (-1);
125
126 return (0);
127}
128
129int
130server_tls_cmp(struct server *s1, struct server *s2)
131{
132 struct server_config *sc1, *sc2;
133
134 sc1 = &s1->srv_conf;
135 sc2 = &s2->srv_conf;
136
137 if (sc1->tls_flags != sc2->tls_flags)
138 return (-1);
139 if (sc1->tls_protocols != sc2->tls_protocols)
140 return (-1);
141 if (sc1->tls_ticket_lifetime != sc2->tls_ticket_lifetime)
142 return (-1);
143 if (strcmp(sc1->tls_ciphers, sc2->tls_ciphers) != 0)
144 return (-1);
145 if (strcmp(sc1->tls_dhe_params, sc2->tls_dhe_params) != 0)
146 return (-1);
147 if (strcmp(sc1->tls_ecdhe_curves, sc2->tls_ecdhe_curves) != 0)
148 return (-1);
149
150 return (0);
151}
152
153int
154server_tls_load_keypair(struct server *srv)
155{
156 if ((srv->srv_conf.flags & SRVFLAG_TLS) == 0)
157 return (0);
158
159 if ((srv->srv_conf.tls_cert = tls_load_file(srv->srv_conf.tls_cert_file,
160 &srv->srv_conf.tls_cert_len, NULL)) == NULL)
161 return (-1);
162 log_debug("%s: using certificate %s", __func__,
163 srv->srv_conf.tls_cert_file);
164
165 /* XXX allow to specify password for encrypted key */
166 if ((srv->srv_conf.tls_key = tls_load_file(srv->srv_conf.tls_key_file,
167 &srv->srv_conf.tls_key_len, NULL)) == NULL)
168 return (-1);
169 log_debug("%s: using private key %s", __func__,
170 srv->srv_conf.tls_key_file);
171
172 return (0);
173}
174
175int
176server_tls_load_ocsp(struct server *srv)
177{
178 if ((srv->srv_conf.flags & SRVFLAG_TLS) == 0)
179 return (0);
180
181 if (srv->srv_conf.tls_ocsp_staple_file == NULL)
182 return (0);
183
184 if ((srv->srv_conf.tls_ocsp_staple = tls_load_file(
185 srv->srv_conf.tls_ocsp_staple_file,
186 &srv->srv_conf.tls_ocsp_staple_len, NULL)) == NULL) {
187 log_warnx("%s: Failed to load ocsp staple from %s", __func__,
188 srv->srv_conf.tls_ocsp_staple_file);
189 return (-1);
190 }
191
192 if (srv->srv_conf.tls_ocsp_staple_len == 0) {
193 log_warnx("%s: ignoring 0 length ocsp staple from %s", __func__,
194 srv->srv_conf.tls_ocsp_staple_file);
195 return (0);
196 }
197
198 log_debug("%s: using ocsp staple from %s", __func__,
199 srv->srv_conf.tls_ocsp_staple_file);
200
201 return (0);
202}
203
204int
205server_tls_load_ca(struct server *srv)
206{
207 if ((srv->srv_conf.tls_flags & TLSFLAG_CA) == 0 ||
208 srv->srv_conf.tls_ca_file == NULL)
209 return (0);
210
211 if ((srv->srv_conf.tls_ca = tls_load_file(
212 srv->srv_conf.tls_ca_file,
213 &srv->srv_conf.tls_ca_len, NULL)) == NULL)
214 return (-1);
215 log_debug("%s: using ca cert(s) from %s", __func__,
216 srv->srv_conf.tls_ca_file);
217
218 return (0);
219}
220
221int
222server_tls_load_crl(struct server *srv)
223{
224 if ((srv->srv_conf.tls_flags & TLSFLAG_CA) == 0 ||
225 srv->srv_conf.tls_crl_file == NULL)
226 return (0);
227
228 if ((srv->srv_conf.tls_crl = tls_load_file(
229 srv->srv_conf.tls_crl_file,
230 &srv->srv_conf.tls_crl_len, NULL)) == NULL)
231 return (-1);
232 log_debug("%s: using crl(s) from %s", __func__,
233 srv->srv_conf.tls_crl_file);
234
235 return (0);
236}
237
238int
239server_tls_init(struct server *srv)
240{
241 struct server_config *srv_conf;
242
243 if ((srv->srv_conf.flags & SRVFLAG_TLS) == 0)
244 return (0);
245
246 log_debug("%s: setting up tls for %s", __func__, srv->srv_conf.name);
247
248 if ((srv->srv_tls_config = tls_config_new()) == NULL) {
249 log_warnx("%s: failed to get tls config", __func__);
250 return (-1);
251 }
252 if ((srv->srv_tls_ctx = tls_server()) == NULL) {
253 log_warnx("%s: failed to get tls server", __func__);
254 return (-1);
255 }
256
257 if (tls_config_set_protocols(srv->srv_tls_config,
258 srv->srv_conf.tls_protocols) != 0) {
259 log_warnx("%s: failed to set tls protocols: %s",
260 __func__, tls_config_error(srv->srv_tls_config));
261 return (-1);
262 }
263 if (tls_config_set_ciphers(srv->srv_tls_config,
264 srv->srv_conf.tls_ciphers) != 0) {
265 log_warnx("%s: failed to set tls ciphers: %s",
266 __func__, tls_config_error(srv->srv_tls_config));
267 return (-1);
268 }
269 if (tls_config_set_dheparams(srv->srv_tls_config,
270 srv->srv_conf.tls_dhe_params) != 0) {
271 log_warnx("%s: failed to set tls dhe params: %s",
272 __func__, tls_config_error(srv->srv_tls_config));
273 return (-1);
274 }
275 if (tls_config_set_ecdhecurves(srv->srv_tls_config,
276 srv->srv_conf.tls_ecdhe_curves) != 0) {
277 log_warnx("%s: failed to set tls ecdhe curves: %s",
278 __func__, tls_config_error(srv->srv_tls_config));
279 return (-1);
280 }
281
282 if (tls_config_set_keypair_ocsp_mem(srv->srv_tls_config,
283 srv->srv_conf.tls_cert, srv->srv_conf.tls_cert_len,
284 srv->srv_conf.tls_key, srv->srv_conf.tls_key_len,
285 srv->srv_conf.tls_ocsp_staple,
286 srv->srv_conf.tls_ocsp_staple_len) != 0) {
287 log_warnx("%s: failed to set tls certificate/key: %s",
288 __func__, tls_config_error(srv->srv_tls_config));
289 return (-1);
290 }
291
292 if (srv->srv_conf.tls_ca != NULL) {
293 if (tls_config_set_ca_mem(srv->srv_tls_config,
294 srv->srv_conf.tls_ca, srv->srv_conf.tls_ca_len) != 0) {
295 log_warnx("%s: failed to add ca cert(s)", __func__);
296 return (-1);
297 }
298 if (tls_config_set_crl_mem(srv->srv_tls_config,
299 srv->srv_conf.tls_crl, srv->srv_conf.tls_crl_len) != 0) {
300 log_warnx("%s: failed to add crl(s)", __func__);
301 return (-1);
302 }
303 if (srv->srv_conf.tls_flags & TLSFLAG_OPTIONAL)
304 tls_config_verify_client_optional(srv->srv_tls_config);
305 else
306 tls_config_verify_client(srv->srv_tls_config);
307 }
308
309 TAILQ_FOREACH(srv_conf, &srv->srv_hosts, entry) {
310 if (srv_conf->tls_cert == NULL || srv_conf->tls_key == NULL)
311 continue;
312 log_debug("%s: adding keypair for server %s", __func__,
313 srv->srv_conf.name);
314 if (tls_config_add_keypair_ocsp_mem(srv->srv_tls_config,
315 srv_conf->tls_cert, srv_conf->tls_cert_len,
316 srv_conf->tls_key, srv_conf->tls_key_len,
317 srv_conf->tls_ocsp_staple,
318 srv_conf->tls_ocsp_staple_len) != 0) {
319 log_warnx("%s: failed to add tls keypair", __func__);
320 return (-1);
321 }
322 }
323
324 /* set common session ID among all processes */
325 if (tls_config_set_session_id(srv->srv_tls_config,
326 httpd_env->sc_tls_sid, sizeof(httpd_env->sc_tls_sid)) == -1) {
327 log_warnx("%s: could not set the TLS session ID: %s",
328 __func__, tls_config_error(srv->srv_tls_config));
329 return (-1);
330 }
331
332 /* ticket support */
333 if (srv->srv_conf.tls_ticket_lifetime) {
334 if (tls_config_set_session_lifetime(srv->srv_tls_config,
335 srv->srv_conf.tls_ticket_lifetime) == -1) {
336 log_warnx("%s: could not set the TLS session lifetime: "
337 "%s", __func__,
338 tls_config_error(srv->srv_tls_config));
339 return (-1);
340 }
341 tls_config_add_ticket_key(srv->srv_tls_config,
342 srv->srv_conf.tls_ticket_key.tt_keyrev,
343 srv->srv_conf.tls_ticket_key.tt_key,
344 sizeof(srv->srv_conf.tls_ticket_key.tt_key));
345 explicit_bzero(&srv->srv_conf.tls_ticket_key,
346 sizeof(srv->srv_conf.tls_ticket_key));
347 }
348
349 if (tls_configure(srv->srv_tls_ctx, srv->srv_tls_config) != 0) {
350 log_warnx("%s: failed to configure tls - %s", __func__,
351 tls_error(srv->srv_tls_ctx));
352 return (-1);
353 }
354
355 /* We're now done with the public/private key & ca/crl... */
356 tls_config_clear_keys(srv->srv_tls_config);
357 freezero(srv->srv_conf.tls_cert, srv->srv_conf.tls_cert_len);
358 freezero(srv->srv_conf.tls_key, srv->srv_conf.tls_key_len);
359 free(srv->srv_conf.tls_ca);
360 free(srv->srv_conf.tls_crl);
361 srv->srv_conf.tls_ca = NULL;
362 srv->srv_conf.tls_cert = NULL;
363 srv->srv_conf.tls_crl = NULL;
364 srv->srv_conf.tls_key = NULL;
365 srv->srv_conf.tls_ca_len = 0;
366 srv->srv_conf.tls_cert_len = 0;
367 srv->srv_conf.tls_crl_len = 0;
368 srv->srv_conf.tls_key_len = 0;
369
370 return (0);
371}
372
373void
374server_generate_ticket_key(struct server_config *srv_conf)
375{
376 struct server_tls_ticket *key = &srv_conf->tls_ticket_key;
377
378 key->tt_id = srv_conf->id;
379 key->tt_keyrev = arc4random();
380 arc4random_buf(key->tt_key, sizeof(key->tt_key));
381}
382
383void
384server_init(struct privsep *ps, struct privsep_proc *p, void *arg)
385{
386 server_http();
387
388 if (config_init(ps->ps_env) == -1)
389 fatal("failed to initialize configuration");
390
391 /* We use a custom shutdown callback */
392 p->p_shutdown = server_shutdown;
393
394 /* Unlimited file descriptors (use system limits) */
395 socket_rlimit(-1);
396
397 if (pledge("stdio rpath inet unix recvfd", NULL) == -1)
398 fatal("pledge");
399
400#if 0
401 /* Schedule statistics timer */
402 evtimer_set(&ps->ps_env->sc_statev, server_statistics, NULL);
403 memcpy(&tv, &ps->ps_env->sc_statinterval, sizeof(tv));
404 evtimer_add(&ps->ps_env->sc_statev, &tv);
405#endif
406}
407
408void
409server_launch(void)
410{
411 struct server *srv;
412
413 TAILQ_FOREACH(srv, httpd_env->sc_servers, srv_entry) {
414 log_debug("%s: configuring server %s", __func__,
415 srv->srv_conf.name);
416
417 server_tls_init(srv);
418 server_http_init(srv);
419
420 log_debug("%s: running server %s", __func__,
421 srv->srv_conf.name);
422
423 event_set(&srv->srv_ev, srv->srv_s, EV_READ,
424 server_accept, srv);
425 event_add(&srv->srv_ev, NULL);
426 evtimer_set(&srv->srv_evt, server_accept, srv);
427 }
428}
429
430void
431server_purge(struct server *srv)
432{
433 struct client *clt;
434 struct server_config *srv_conf;
435
436 /* shutdown and remove server */
437 if (event_initialized(&srv->srv_ev))
438 event_del(&srv->srv_ev);
439 if (evtimer_initialized(&srv->srv_evt))
440 evtimer_del(&srv->srv_evt);
441
442 if (srv->srv_s != -1)
443 close(srv->srv_s);
444 TAILQ_REMOVE(httpd_env->sc_servers, srv, srv_entry);
445
446 /* cleanup sessions */
447 while ((clt =
448 SPLAY_ROOT(&srv->srv_clients)) != NULL)
449 server_close(clt, NULL);
450
451 /* cleanup hosts */
452 while ((srv_conf =
453 TAILQ_FIRST(&srv->srv_hosts)) != NULL) {
454 TAILQ_REMOVE(&srv->srv_hosts, srv_conf, entry);
455
456 /* It might point to our own "default" entry */
457 if (srv_conf != &srv->srv_conf) {
458 serverconfig_free(srv_conf);
459 free(srv_conf);
460 }
461 }
462
463 tls_config_free(srv->srv_tls_config);
464 tls_free(srv->srv_tls_ctx);
465
466 free(srv);
467}
468
469void
470serverconfig_free(struct server_config *srv_conf)
471{
472 struct fastcgi_param *param, *tparam;
473
474 free(srv_conf->return_uri);
475 free(srv_conf->tls_ca_file);
476 free(srv_conf->tls_ca);
477 free(srv_conf->tls_cert_file);
478 free(srv_conf->tls_crl_file);
479 free(srv_conf->tls_crl);
480 free(srv_conf->tls_key_file);
481 free(srv_conf->tls_ocsp_staple_file);
482 free(srv_conf->tls_ocsp_staple);
483 freezero(srv_conf->tls_cert, srv_conf->tls_cert_len);
484 freezero(srv_conf->tls_key, srv_conf->tls_key_len);
485
486 TAILQ_FOREACH_SAFE(param, &srv_conf->fcgiparams, entry, tparam)
487 free(param);
488}
489
490void
491serverconfig_reset(struct server_config *srv_conf)
492{
493 srv_conf->auth = NULL;
494 srv_conf->return_uri = NULL;
495 srv_conf->tls_ca = NULL;
496 srv_conf->tls_ca_file = NULL;
497 srv_conf->tls_cert = NULL;
498 srv_conf->tls_cert_file = NULL;
499 srv_conf->tls_crl = NULL;
500 srv_conf->tls_crl_file = NULL;
501 srv_conf->tls_key = NULL;
502 srv_conf->tls_key_file = NULL;
503 srv_conf->tls_ocsp_staple = NULL;
504 srv_conf->tls_ocsp_staple_file = NULL;
505 TAILQ_INIT(&srv_conf->fcgiparams);
506}
507
508struct server *
509server_byaddr(struct sockaddr *addr, in_port_t port)
510{
511 struct server *srv;
512
513 TAILQ_FOREACH(srv, httpd_env->sc_servers, srv_entry) {
514 if (port == srv->srv_conf.port &&
515 sockaddr_cmp((struct sockaddr *)&srv->srv_conf.ss,
516 addr, srv->srv_conf.prefixlen) == 0)
517 return (srv);
518 }
519
520 return (NULL);
521}
522
523struct server_config *
524serverconfig_byid(uint32_t id)
525{
526 struct server *srv;
527 struct server_config *srv_conf;
528
529 TAILQ_FOREACH(srv, httpd_env->sc_servers, srv_entry) {
530 if (srv->srv_conf.id == id)
531 return (&srv->srv_conf);
532 TAILQ_FOREACH(srv_conf, &srv->srv_hosts, entry) {
533 if (srv_conf->id == id)
534 return (srv_conf);
535 }
536 }
537
538 return (NULL);
539}
540
541struct server *
542server_byid(uint32_t id)
543{
544 struct server *srv;
545
546 TAILQ_FOREACH(srv, httpd_env->sc_servers, srv_entry) {
547 if (srv->srv_conf.id == id)
548 return (srv);
549 }
550 return (NULL);
551}
552
553int
554server_foreach(int (*srv_cb)(struct server *,
555 struct server_config *, void *), void *arg)
556{
557 struct server *srv;
558 struct server_config *srv_conf;
559
560 TAILQ_FOREACH(srv, httpd_env->sc_servers, srv_entry) {
561 if ((srv_cb)(srv, &srv->srv_conf, arg) == -1)
562 return (-1);
563 TAILQ_FOREACH(srv_conf, &srv->srv_hosts, entry) {
564 if ((srv_cb)(srv, srv_conf, arg) == -1)
565 return (-1);
566 }
567 }
568
569 return (0);
570}
571
572struct server *
573server_match(struct server *s2, int match_name)
574{
575 struct server *s1;
576
577 /* Attempt to find matching server. */
578 TAILQ_FOREACH(s1, httpd_env->sc_servers, srv_entry) {
579 if ((s1->srv_conf.flags & SRVFLAG_LOCATION) != 0)
580 continue;
581 if (match_name) {
582 if (strcmp(s1->srv_conf.name, s2->srv_conf.name) != 0)
583 continue;
584 }
585 if (s1->srv_conf.port != s2->srv_conf.port)
586 continue;
587 if (sockaddr_cmp(
588 (struct sockaddr *)&s1->srv_conf.ss,
589 (struct sockaddr *)&s2->srv_conf.ss,
590 s1->srv_conf.prefixlen) != 0)
591 continue;
592
593 return (s1);
594 }
595
596 return (NULL);
597}
598
599int
600server_socket_af(struct sockaddr_storage *ss, in_port_t port)
601{
602 switch (ss->ss_family) {
603 case AF_INET:
604 ((struct sockaddr_in *)ss)->sin_port = port;
605 ((struct sockaddr_in *)ss)->sin_len =
606 sizeof(struct sockaddr_in);
607 break;
608 case AF_INET6:
609 ((struct sockaddr_in6 *)ss)->sin6_port = port;
610 ((struct sockaddr_in6 *)ss)->sin6_len =
611 sizeof(struct sockaddr_in6);
612 break;
613 default:
614 return (-1);
615 }
616
617 return (0);
618}
619
620in_port_t
621server_socket_getport(struct sockaddr_storage *ss)
622{
623 switch (ss->ss_family) {
624 case AF_INET:
625 return (((struct sockaddr_in *)ss)->sin_port);
626 case AF_INET6:
627 return (((struct sockaddr_in6 *)ss)->sin6_port);
628 default:
629 return (0);
630 }
631
632 /* NOTREACHED */
633 return (0);
634}
635
636int
637server_socket(struct sockaddr_storage *ss, in_port_t port,
638 struct server_config *srv_conf, int fd, int reuseport)
639{
640 struct linger lng;
641 int s = -1, val;
642
643 if (server_socket_af(ss, port) == -1)
644 goto bad;
645
646 s = fd == -1 ? socket(ss->ss_family, SOCK_STREAM | SOCK_NONBLOCK,
647 IPPROTO_TCP) : fd;
648 if (s == -1)
649 goto bad;
650
651 /*
652 * Socket options
653 */
654 memset(&lng, 0, sizeof(lng));
655 if (setsockopt(s, SOL_SOCKET, SO_LINGER, &lng, sizeof(lng)) == -1)
656 goto bad;
657 if (reuseport) {
658 val = 1;
659 if (setsockopt(s, SOL_SOCKET, SO_REUSEPORT, &val,
660 sizeof(int)) == -1)
661 goto bad;
662 }
663 if (srv_conf->tcpflags & TCPFLAG_BUFSIZ) {
664 val = srv_conf->tcpbufsiz;
665 if (setsockopt(s, SOL_SOCKET, SO_RCVBUF,
666 &val, sizeof(val)) == -1)
667 goto bad;
668 val = srv_conf->tcpbufsiz;
669 if (setsockopt(s, SOL_SOCKET, SO_SNDBUF,
670 &val, sizeof(val)) == -1)
671 goto bad;
672 }
673
674 /*
675 * IP options
676 */
677 if (srv_conf->tcpflags & TCPFLAG_IPTTL) {
678 val = (int)srv_conf->tcpipttl;
679 switch (ss->ss_family) {
680 case AF_INET:
681 if (setsockopt(s, IPPROTO_IP, IP_TTL,
682 &val, sizeof(val)) == -1)
683 goto bad;
684 break;
685 case AF_INET6:
686 if (setsockopt(s, IPPROTO_IPV6, IPV6_UNICAST_HOPS,
687 &val, sizeof(val)) == -1)
688 goto bad;
689 break;
690 }
691 }
692 if (srv_conf->tcpflags & TCPFLAG_IPMINTTL) {
693 val = (int)srv_conf->tcpipminttl;
694 switch (ss->ss_family) {
695 case AF_INET:
696 if (setsockopt(s, IPPROTO_IP, IP_MINTTL,
697 &val, sizeof(val)) == -1)
698 goto bad;
699 break;
700 case AF_INET6:
701 if (setsockopt(s, IPPROTO_IPV6, IPV6_MINHOPCOUNT,
702 &val, sizeof(val)) == -1)
703 goto bad;
704 break;
705 }
706 }
707
708 /*
709 * TCP options
710 */
711 if (srv_conf->tcpflags & (TCPFLAG_NODELAY|TCPFLAG_NNODELAY)) {
712 if (srv_conf->tcpflags & TCPFLAG_NNODELAY)
713 val = 0;
714 else
715 val = 1;
716 if (setsockopt(s, IPPROTO_TCP, TCP_NODELAY,
717 &val, sizeof(val)) == -1)
718 goto bad;
719 }
720 if (srv_conf->tcpflags & (TCPFLAG_SACK|TCPFLAG_NSACK)) {
721 if (srv_conf->tcpflags & TCPFLAG_NSACK)
722 val = 0;
723 else
724 val = 1;
725 if (setsockopt(s, IPPROTO_TCP, TCP_SACK_ENABLE,
726 &val, sizeof(val)) == -1)
727 goto bad;
728 }
729
730 return (s);
731
732 bad:
733 if (s != -1)
734 close(s);
735 return (-1);
736}
737
738int
739server_socket_listen(struct sockaddr_storage *ss, in_port_t port,
740 struct server_config *srv_conf)
741{
742 int s;
743
744 if ((s = server_socket(ss, port, srv_conf, -1, 1)) == -1)
745 return (-1);
746
747 if (bind(s, (struct sockaddr *)ss, ss->ss_len) == -1)
748 goto bad;
749 if (listen(s, srv_conf->tcpbacklog) == -1)
750 goto bad;
751
752 return (s);
753
754 bad:
755 close(s);
756 return (-1);
757}
758
759int
760server_socket_connect(struct sockaddr_storage *ss, in_port_t port,
761 struct server_config *srv_conf)
762{
763 int s;
764
765 if ((s = server_socket(ss, port, srv_conf, -1, 0)) == -1)
766 return (-1);
767
768 if (connect(s, (struct sockaddr *)ss, ss->ss_len) == -1) {
769 if (errno != EINPROGRESS)
770 goto bad;
771 }
772
773 return (s);
774
775 bad:
776 close(s);
777 return (-1);
778}
779
780void
781server_tls_readcb(int fd, short event, void *arg)
782{
783 struct bufferevent *bufev = arg;
784 struct client *clt = bufev->cbarg;
785 char rbuf[IBUF_READ_SIZE];
786 int what = EVBUFFER_READ;
787 int howmuch = IBUF_READ_SIZE;
788 ssize_t ret;
789 size_t len;
790
791 if (event == EV_TIMEOUT) {
792 what |= EVBUFFER_TIMEOUT;
793 goto err;
794 }
795
796 if (bufev->wm_read.high != 0)
797 howmuch = MINIMUM(sizeof(rbuf), bufev->wm_read.high);
798
799 ret = tls_read(clt->clt_tls_ctx, rbuf, howmuch);
800 if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
801 goto retry;
802 } else if (ret == -1) {
803 what |= EVBUFFER_ERROR;
804 goto err;
805 }
806 len = ret;
807
808 if (len == 0) {
809 what |= EVBUFFER_EOF;
810 goto err;
811 }
812
813 if (evbuffer_add(bufev->input, rbuf, len) == -1) {
814 what |= EVBUFFER_ERROR;
815 goto err;
816 }
817
818 server_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
819
820 len = EVBUFFER_LENGTH(bufev->input);
821 if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
822 return;
823 if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
824 struct evbuffer *buf = bufev->input;
825 event_del(&bufev->ev_read);
826 evbuffer_setcb(buf, bufferevent_read_pressure_cb, bufev);
827 return;
828 }
829
830 if (bufev->readcb != NULL)
831 (*bufev->readcb)(bufev, bufev->cbarg);
832 return;
833
834 retry:
835 server_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
836 return;
837
838 err:
839 (*bufev->errorcb)(bufev, what, bufev->cbarg);
840}
841
842void
843server_tls_writecb(int fd, short event, void *arg)
844{
845 struct bufferevent *bufev = arg;
846 struct client *clt = bufev->cbarg;
847 ssize_t ret;
848 short what = EVBUFFER_WRITE;
849 size_t len;
850
851 if (event == EV_TIMEOUT) {
852 what |= EVBUFFER_TIMEOUT;
853 goto err;
854 }
855
856 if (EVBUFFER_LENGTH(bufev->output)) {
857 ret = tls_write(clt->clt_tls_ctx,
858 EVBUFFER_DATA(bufev->output),
859 EVBUFFER_LENGTH(bufev->output));
860 if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
861 goto retry;
862 } else if (ret == -1) {
863 what |= EVBUFFER_ERROR;
864 goto err;
865 }
866 len = ret;
867 evbuffer_drain(bufev->output, len);
868 }
869
870 if (EVBUFFER_LENGTH(bufev->output) != 0)
871 server_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
872
873 if (bufev->writecb != NULL &&
874 EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
875 (*bufev->writecb)(bufev, bufev->cbarg);
876 return;
877
878 retry:
879 server_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
880 return;
881
882 err:
883 (*bufev->errorcb)(bufev, what, bufev->cbarg);
884}
885
886void
887server_input(struct client *clt)
888{
889 struct server_config *srv_conf = clt->clt_srv_conf;
890 socklen_t slen;
891
892 if (server_httpdesc_init(clt) == -1) {
893 server_close(clt, "failed to allocate http descriptor");
894 return;
895 }
896
897 clt->clt_toread = TOREAD_HTTP_HEADER;
898
899 slen = sizeof(clt->clt_sndbufsiz);
900 if (getsockopt(clt->clt_s, SOL_SOCKET, SO_SNDBUF,
901 &clt->clt_sndbufsiz, &slen) == -1) {
902 server_close(clt, "failed to get send buffer size");
903 return;
904 }
905
906 /*
907 * Client <-> Server
908 */
909 clt->clt_bev = bufferevent_new(clt->clt_s, server_read_http,
910 server_write, server_error, clt);
911 if (clt->clt_bev == NULL) {
912 server_close(clt, "failed to allocate input buffer event");
913 return;
914 }
915
916 if (srv_conf->flags & SRVFLAG_TLS) {
917 event_set(&clt->clt_bev->ev_read, clt->clt_s, EV_READ,
918 server_tls_readcb, clt->clt_bev);
919 event_set(&clt->clt_bev->ev_write, clt->clt_s, EV_WRITE,
920 server_tls_writecb, clt->clt_bev);
921 }
922
923 /* Adjust write watermark to the socket buffer output size */
924 bufferevent_setwatermark(clt->clt_bev, EV_WRITE,
925 SERVER_MIN_PREFETCHED * clt->clt_sndbufsiz, 0);
926 /* Read at most amount of data that fits in one fcgi record. */
927 bufferevent_setwatermark(clt->clt_bev, EV_READ, 0, FCGI_CONTENT_SIZE);
928
929 bufferevent_settimeout(clt->clt_bev,
930 srv_conf->requesttimeout.tv_sec, srv_conf->requesttimeout.tv_sec);
931 bufferevent_enable(clt->clt_bev, EV_READ|EV_WRITE);
932}
933
934void
935server_write(struct bufferevent *bev, void *arg)
936{
937 struct client *clt = arg;
938 struct evbuffer *dst = EVBUFFER_OUTPUT(bev);
939
940 if (EVBUFFER_LENGTH(dst) == 0 &&
941 clt->clt_toread == TOREAD_HTTP_NONE)
942 goto done;
943
944 getmonotime(&clt->clt_tv_last);
945
946 if (clt->clt_done)
947 goto done;
948
949 if (clt->clt_srvbev && clt->clt_srvbev_throttled) {
950 bufferevent_enable(clt->clt_srvbev, EV_READ);
951 clt->clt_srvbev_throttled = 0;
952 }
953
954 return;
955 done:
956 (*bev->errorcb)(bev, EVBUFFER_WRITE, bev->cbarg);
957 return;
958}
959
960void
961server_dump(struct client *clt, const void *buf, size_t len)
962{
963 if (!len)
964 return;
965
966 /*
967 * This function will dump the specified message directly
968 * to the underlying client, without waiting for success
969 * of non-blocking events etc. This is useful to print an
970 * error message before gracefully closing the client.
971 */
972 if (clt->clt_tls_ctx != NULL)
973 (void)tls_write(clt->clt_tls_ctx, buf, len);
974 else
975 (void)write(clt->clt_s, buf, len);
976}
977
978void
979server_read(struct bufferevent *bev, void *arg)
980{
981 struct client *clt = arg;
982 struct evbuffer *src = EVBUFFER_INPUT(bev);
983
984 getmonotime(&clt->clt_tv_last);
985
986 if (!EVBUFFER_LENGTH(src))
987 return;
988 if (server_bufferevent_write_buffer(clt, src) == -1)
989 goto fail;
990 if (clt->clt_done)
991 goto done;
992
993 if (EVBUFFER_LENGTH(EVBUFFER_OUTPUT(clt->clt_bev)) > (size_t)
994 SERVER_MAX_PREFETCH * clt->clt_sndbufsiz) {
995 bufferevent_disable(clt->clt_srvbev, EV_READ);
996 clt->clt_srvbev_throttled = 1;
997 }
998
999 return;
1000 done:
1001 (*bev->errorcb)(bev, EVBUFFER_READ, bev->cbarg);
1002 return;
1003 fail:
1004 server_close(clt, strerror(errno));
1005}
1006
1007void
1008server_error(struct bufferevent *bev, short error, void *arg)
1009{
1010 struct client *clt = arg;
1011 struct evbuffer *dst;
1012
1013 if (error & EVBUFFER_TIMEOUT) {
1014 if (!clt->clt_headersdone && clt->clt_line > 0)
1015 server_abort_http(clt, 408, "timeout");
1016 else
1017 server_close(clt, "timeout");
1018 return;
1019 }
1020 if (error & EVBUFFER_ERROR) {
1021 if (errno == EFBIG) {
1022 bufferevent_enable(bev, EV_READ);
1023 return;
1024 }
1025 server_close(clt, "buffer event error");
1026 return;
1027 }
1028 if (error & EVBUFFER_EOF) {
1029 server_close(clt, "closed");
1030 return;
1031 }
1032 if (error & (EVBUFFER_READ|EVBUFFER_WRITE)) {
1033 bufferevent_disable(bev, EV_READ|EV_WRITE);
1034
1035 clt->clt_done = 1;
1036
1037 dst = EVBUFFER_OUTPUT(clt->clt_bev);
1038 if (EVBUFFER_LENGTH(dst)) {
1039 /* Finish writing all data first */
1040 bufferevent_enable(clt->clt_bev, EV_WRITE);
1041 return;
1042 }
1043
1044 server_close(clt, "done");
1045 return;
1046 }
1047 server_close(clt, "unknown event error");
1048 return;
1049}
1050
1051void
1052server_accept(int fd, short event, void *arg)
1053{
1054 struct server *srv = arg;
1055 struct client *clt = NULL;
1056 socklen_t slen;
1057 struct sockaddr_storage ss;
1058 int s = -1;
1059
1060 event_add(&srv->srv_ev, NULL);
1061 if ((event & EV_TIMEOUT))
1062 return;
1063
1064 slen = sizeof(ss);
1065 if ((s = accept_reserve(fd, (struct sockaddr *)&ss,
1066 &slen, FD_RESERVE, &server_inflight)) == -1) {
1067 /*
1068 * Pause accept if we are out of file descriptors, or
1069 * libevent will haunt us here too.
1070 */
1071 if (errno == ENFILE || errno == EMFILE) {
1072 struct timeval evtpause = { 1, 0 };
1073
1074 event_del(&srv->srv_ev);
1075 evtimer_add(&srv->srv_evt, &evtpause);
1076 log_debug("%s: deferring connections", __func__);
1077 }
1078 return;
1079 }
1080 if (server_clients >= SERVER_MAX_CLIENTS)
1081 goto err;
1082
1083 if ((clt = calloc(1, sizeof(*clt))) == NULL)
1084 goto err;
1085
1086 /* Pre-allocate log buffer */
1087 clt->clt_log = evbuffer_new();
1088 if (clt->clt_log == NULL)
1089 goto err;
1090
1091 clt->clt_s = s;
1092 clt->clt_fd = -1;
1093 clt->clt_toread = TOREAD_UNLIMITED;
1094 clt->clt_srv = srv;
1095 clt->clt_srv_conf = &srv->srv_conf;
1096 clt->clt_id = ++server_cltid;
1097 clt->clt_srv_id = srv->srv_conf.id;
1098 clt->clt_pid = getpid();
1099 clt->clt_inflight = 1;
1100
1101 /* get local address */
1102 slen = sizeof(clt->clt_srv_ss);
1103 if (getsockname(s, (struct sockaddr *)&clt->clt_srv_ss,
1104 &slen) == -1) {
1105 server_close(clt, "listen address lookup failed");
1106 return;
1107 }
1108
1109 /* get client address */
1110 memcpy(&clt->clt_ss, &ss, sizeof(clt->clt_ss));
1111
1112 /* get ports */
1113 switch (ss.ss_family) {
1114 case AF_INET:
1115 clt->clt_port = ((struct sockaddr_in *)&ss)->sin_port;
1116 break;
1117 case AF_INET6:
1118 clt->clt_port = ((struct sockaddr_in6 *)&ss)->sin6_port;
1119 break;
1120 }
1121
1122 getmonotime(&clt->clt_tv_start);
1123 memcpy(&clt->clt_tv_last, &clt->clt_tv_start, sizeof(clt->clt_tv_last));
1124
1125 server_clients++;
1126 SPLAY_INSERT(client_tree, &srv->srv_clients, clt);
1127
1128 /* Pre-allocate output buffer */
1129 clt->clt_output = evbuffer_new();
1130 if (clt->clt_output == NULL) {
1131 server_close(clt, "failed to allocate output buffer");
1132 return;
1133 }
1134
1135 if (srv->srv_conf.flags & SRVFLAG_TLS) {
1136 if (tls_accept_socket(srv->srv_tls_ctx, &clt->clt_tls_ctx,
1137 clt->clt_s) != 0) {
1138 server_close(clt, "failed to accept tls socket");
1139 return;
1140 }
1141 event_again(&clt->clt_ev, clt->clt_s, EV_TIMEOUT|EV_READ,
1142 server_tls_handshake, &clt->clt_tv_start,
1143 &srv->srv_conf.timeout, clt);
1144 return;
1145 }
1146
1147 server_input(clt);
1148 return;
1149
1150 err:
1151 if (s != -1) {
1152 close(s);
1153 free(clt);
1154 /*
1155 * the client struct was not completely set up, but still
1156 * counted as an inflight client. account for this.
1157 */
1158 server_inflight_dec(NULL, __func__);
1159 }
1160}
1161
1162void
1163server_tls_handshake(int fd, short event, void *arg)
1164{
1165 struct client *clt = (struct client *)arg;
1166 struct server *srv = (struct server *)clt->clt_srv;
1167 int ret;
1168
1169 if (event == EV_TIMEOUT) {
1170 server_close(clt, "tls handshake timeout");
1171 return;
1172 }
1173
1174 if (srv->srv_tls_ctx == NULL || clt->clt_tls_ctx == NULL)
1175 fatalx("NULL tls context");
1176
1177 ret = tls_handshake(clt->clt_tls_ctx);
1178 if (ret == 0) {
1179 server_input(clt);
1180 } else if (ret == TLS_WANT_POLLIN) {
1181 event_again(&clt->clt_ev, clt->clt_s, EV_TIMEOUT|EV_READ,
1182 server_tls_handshake, &clt->clt_tv_start,
1183 &srv->srv_conf.timeout, clt);
1184 } else if (ret == TLS_WANT_POLLOUT) {
1185 event_again(&clt->clt_ev, clt->clt_s, EV_TIMEOUT|EV_WRITE,
1186 server_tls_handshake, &clt->clt_tv_start,
1187 &srv->srv_conf.timeout, clt);
1188 } else {
1189 log_debug("%s: tls handshake failed - %s", __func__,
1190 tls_error(clt->clt_tls_ctx));
1191 server_close(clt, "tls handshake failed");
1192 }
1193}
1194
1195void
1196server_inflight_dec(struct client *clt, const char *why)
1197{
1198 if (clt != NULL) {
1199 /* the flight already left inflight mode. */
1200 if (clt->clt_inflight == 0)
1201 return;
1202 clt->clt_inflight = 0;
1203 }
1204
1205 /* the file was never opened, thus this was an inflight client. */
1206 server_inflight--;
1207 DPRINTF("%s: inflight decremented, now %d, %s",
1208 __func__, server_inflight, why);
1209}
1210
1211void
1212server_sendlog(struct server_config *srv_conf, int cmd, const char *emsg, ...)
1213{
1214 va_list ap;
1215 char *msg;
1216 int ret;
1217 struct iovec iov[2];
1218
1219 if (srv_conf->flags & SRVFLAG_SYSLOG) {
1220 va_start(ap, emsg);
1221 if (cmd == IMSG_LOG_ACCESS)
1222 vlog(LOG_INFO, emsg, ap);
1223 else
1224 vlog(LOG_DEBUG, emsg, ap);
1225 va_end(ap);
1226 return;
1227 }
1228
1229 va_start(ap, emsg);
1230 ret = vasprintf(&msg, emsg, ap);
1231 va_end(ap);
1232 if (ret == -1) {
1233 log_warn("%s: vasprintf", __func__);
1234 return;
1235 }
1236
1237 iov[0].iov_base = &srv_conf->id;
1238 iov[0].iov_len = sizeof(srv_conf->id);
1239 iov[1].iov_base = msg;
1240 iov[1].iov_len = ret + 1;
1241
1242 if (proc_composev(httpd_env->sc_ps, PROC_LOGGER, cmd, iov, 2) != 0) {
1243 log_warn("%s: failed to compose imsg", __func__);
1244 free(msg);
1245 return;
1246 }
1247 free(msg);
1248}
1249
1250void
1251server_log(struct client *clt, const char *msg)
1252{
1253 char ibuf[HOST_NAME_MAX+1], obuf[HOST_NAME_MAX+1];
1254 struct server_config *srv_conf = clt->clt_srv_conf;
1255 char *ptr = NULL, *vmsg = NULL;
1256 int debug_cmd = -1;
1257
1258 switch (srv_conf->logformat) {
1259 case LOG_FORMAT_CONNECTION:
1260 debug_cmd = IMSG_LOG_ACCESS;
1261 break;
1262 default:
1263 if (log_getverbose() > 1)
1264 debug_cmd = IMSG_LOG_ERROR;
1265 if (EVBUFFER_LENGTH(clt->clt_log)) {
1266 while ((ptr =
1267 evbuffer_readline(clt->clt_log)) != NULL) {
1268 server_sendlog(srv_conf,
1269 IMSG_LOG_ACCESS, "%s", ptr);
1270 free(ptr);
1271 }
1272 }
1273 break;
1274 }
1275
1276 if (debug_cmd != -1 && msg != NULL) {
1277 memset(ibuf, 0, sizeof(ibuf));
1278 memset(obuf, 0, sizeof(obuf));
1279 (void)print_host(&clt->clt_ss, ibuf, sizeof(ibuf));
1280 (void)server_http_host(&clt->clt_srv_ss, obuf, sizeof(obuf));
1281 if (EVBUFFER_LENGTH(clt->clt_log) &&
1282 evbuffer_add_printf(clt->clt_log, "\n") != -1)
1283 ptr = evbuffer_readline(clt->clt_log);
1284 (void)stravis(&vmsg, msg, HTTPD_LOGVIS);
1285 server_sendlog(srv_conf, debug_cmd, "server %s, "
1286 "client %d (%d active), %s:%u -> %s, "
1287 "%s%s%s", srv_conf->name, clt->clt_id, server_clients,
1288 ibuf, ntohs(clt->clt_port), obuf, vmsg == NULL ? "" : vmsg,
1289 ptr == NULL ? "" : ",", ptr == NULL ? "" : ptr);
1290 free(vmsg);
1291 free(ptr);
1292 }
1293}
1294
1295void
1296server_close(struct client *clt, const char *msg)
1297{
1298 struct server *srv = clt->clt_srv;
1299
1300 if (clt->clt_fcgi_count-- > 0) {
1301 clt->clt_fcgi_error = msg;
1302 return;
1303 }
1304
1305 SPLAY_REMOVE(client_tree, &srv->srv_clients, clt);
1306
1307 /* free the HTTP descriptors incl. headers */
1308 server_close_http(clt);
1309
1310 /* tls_close must be called before the underlying socket is closed. */
1311 if (clt->clt_tls_ctx != NULL)
1312 tls_close(clt->clt_tls_ctx); /* XXX - error handling */
1313 tls_free(clt->clt_tls_ctx);
1314
1315 event_del(&clt->clt_ev);
1316 if (clt->clt_bev != NULL)
1317 bufferevent_disable(clt->clt_bev, EV_READ|EV_WRITE);
1318 if (clt->clt_srvbev != NULL)
1319 bufferevent_disable(clt->clt_srvbev, EV_READ|EV_WRITE);
1320
1321 server_log(clt, msg);
1322
1323 if (clt->clt_bev != NULL)
1324 bufferevent_free(clt->clt_bev);
1325 if (clt->clt_output != NULL)
1326 evbuffer_free(clt->clt_output);
1327 if (clt->clt_srvevb != NULL)
1328 evbuffer_free(clt->clt_srvevb);
1329
1330 if (clt->clt_srvbev != NULL)
1331 bufferevent_free(clt->clt_srvbev);
1332
1333 if (clt->clt_fd != -1)
1334 close(clt->clt_fd);
1335 if (clt->clt_s != -1)
1336 close(clt->clt_s);
1337
1338 server_inflight_dec(clt, __func__);
1339
1340 if (clt->clt_log != NULL)
1341 evbuffer_free(clt->clt_log);
1342
1343 free(clt);
1344 server_clients--;
1345}
1346
1347int
1348server_dispatch_parent(int fd, struct privsep_proc *p, struct imsg *imsg)
1349{
1350 struct server *srv;
1351 struct server_tls_ticket key;
1352
1353 switch (imsg->hdr.type) {
1354 case IMSG_CFG_MEDIA:
1355 config_getmedia(httpd_env, imsg);
1356 break;
1357 case IMSG_CFG_AUTH:
1358 config_getauth(httpd_env, imsg);
1359 break;
1360 case IMSG_CFG_SERVER:
1361 config_getserver(httpd_env, imsg);
1362 break;
1363 case IMSG_CFG_TLS:
1364 config_getserver_tls(httpd_env, imsg);
1365 break;
1366 case IMSG_CFG_FCGI:
1367 config_getserver_fcgiparams(httpd_env, imsg);
1368 break;
1369 case IMSG_CFG_DONE:
1370 config_getcfg(httpd_env, imsg);
1371 break;
1372 case IMSG_CTL_START:
1373 server_launch();
1374 break;
1375 case IMSG_CTL_RESET:
1376 config_getreset(httpd_env, imsg);
1377 break;
1378 case IMSG_TLSTICKET_REKEY:
1379 IMSG_SIZE_CHECK(imsg, (&key));
1380 memcpy(&key, imsg->data, sizeof(key));
1381 /* apply to the right server */
1382 srv = server_byid(key.tt_id);
1383 if (srv) {
1384 tls_config_add_ticket_key(srv->srv_tls_config,
1385 key.tt_keyrev, key.tt_key, sizeof(key.tt_key));
1386 }
1387 break;
1388 default:
1389 return (-1);
1390 }
1391
1392 return (0);
1393}
1394
1395int
1396server_dispatch_logger(int fd, struct privsep_proc *p, struct imsg *imsg)
1397{
1398 switch (imsg->hdr.type) {
1399 default:
1400 return (-1);
1401 }
1402
1403 return (0);
1404}
1405
1406int
1407server_bufferevent_add(struct event *ev, int timeout)
1408{
1409 struct timeval tv, *ptv = NULL;
1410
1411 if (timeout) {
1412 timerclear(&tv);
1413 tv.tv_sec = timeout;
1414 ptv = &tv;
1415 }
1416
1417 return (event_add(ev, ptv));
1418}
1419
1420int
1421server_bufferevent_printf(struct client *clt, const char *fmt, ...)
1422{
1423 int ret;
1424 va_list ap;
1425 char *str;
1426
1427 va_start(ap, fmt);
1428 ret = vasprintf(&str, fmt, ap);
1429 va_end(ap);
1430
1431 if (ret == -1)
1432 return (ret);
1433
1434 ret = server_bufferevent_print(clt, str);
1435 free(str);
1436
1437 return (ret);
1438}
1439
1440int
1441server_bufferevent_print(struct client *clt, const char *str)
1442{
1443 if (clt->clt_bev == NULL)
1444 return (evbuffer_add(clt->clt_output, str, strlen(str)));
1445 return (bufferevent_write(clt->clt_bev, str, strlen(str)));
1446}
1447
1448int
1449server_bufferevent_write_buffer(struct client *clt, struct evbuffer *buf)
1450{
1451 if (clt->clt_bev == NULL)
1452 return (evbuffer_add_buffer(clt->clt_output, buf));
1453 return (bufferevent_write_buffer(clt->clt_bev, buf));
1454}
1455
1456int
1457server_bufferevent_write_chunk(struct client *clt,
1458 struct evbuffer *buf, size_t size)
1459{
1460 int ret;
1461 ret = server_bufferevent_write(clt, EVBUFFER_DATA(buf), size);
1462 if (ret != -1)
1463 evbuffer_drain(buf, size);
1464 return (ret);
1465}
1466
1467int
1468server_bufferevent_write(struct client *clt, void *data, size_t size)
1469{
1470 if (clt->clt_bev == NULL)
1471 return (evbuffer_add(clt->clt_output, data, size));
1472 return (bufferevent_write(clt->clt_bev, data, size));
1473}
1474
1475int
1476server_client_cmp(struct client *a, struct client *b)
1477{
1478 return ((int)a->clt_id - b->clt_id);
1479}
1480
1481SPLAY_GENERATE(client_tree, client, clt_nodes, server_client_cmp);