op public repos

Blame

Date:: Tue Nov 22 13:11:10 2022 UTC
Message:: endian.h is not universally available FreeBSD and NetBSD have sys/endian.h, on MacOS we need to use the functions from libkern/OSByteOrder.h see github issue #1
Actions:: History | Blob | Raw File
  1 fb1a36c0 2022-01-09 op /*
  2 1e84c7b7 2022-01-10 op  * Copyright (c) 2021, 2022 Omar Polo <op@omarpolo.com>
  3 fb1a36c0 2022-01-09 op  * Copyright (c) 2018 Florian Obser <florian@openbsd.org>
  4 fb1a36c0 2022-01-09 op  * Copyright (c) 2004, 2005 Claudio Jeker <claudio@openbsd.org>
  5 fb1a36c0 2022-01-09 op  * Copyright (c) 2004 Esben Norby <norby@openbsd.org>
  6 fb1a36c0 2022-01-09 op  * Copyright (c) 2003, 2004 Henning Brauer <henning@openbsd.org>
  7 fb1a36c0 2022-01-09 op  *
  8 fb1a36c0 2022-01-09 op  * Permission to use, copy, modify, and distribute this software for any
  9 fb1a36c0 2022-01-09 op  * purpose with or without fee is hereby granted, provided that the above
 10 fb1a36c0 2022-01-09 op  * copyright notice and this permission notice appear in all copies.
 11 fb1a36c0 2022-01-09 op  *
 12 fb1a36c0 2022-01-09 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 13 fb1a36c0 2022-01-09 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 14 fb1a36c0 2022-01-09 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 15 fb1a36c0 2022-01-09 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 16 fb1a36c0 2022-01-09 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 17 fb1a36c0 2022-01-09 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 18 fb1a36c0 2022-01-09 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 19 fb1a36c0 2022-01-09 op  */
 20 fb1a36c0 2022-01-09 op 
 21 bbcba3ed 2022-01-10 op #include "compat.h"
 22 bbcba3ed 2022-01-10 op 
 23 fb1a36c0 2022-01-09 op #include <sys/socket.h>
 24 fb1a36c0 2022-01-09 op #include <sys/types.h>
 25 fb1a36c0 2022-01-09 op #include <sys/uio.h>
 26 fb1a36c0 2022-01-09 op 
 27 fb1a36c0 2022-01-09 op #include <errno.h>
 28 fb1a36c0 2022-01-09 op #include <inttypes.h>
 29 fb1a36c0 2022-01-09 op #include <pwd.h>
 30 fb1a36c0 2022-01-09 op #include <signal.h>
 31 fb1a36c0 2022-01-09 op #include <stdint.h>
 32 fb1a36c0 2022-01-09 op #include <stdio.h>
 33 fb1a36c0 2022-01-09 op #include <stdlib.h>
 34 fb1a36c0 2022-01-09 op #include <string.h>
 35 fb1a36c0 2022-01-09 op #include <syslog.h>
 36 fb1a36c0 2022-01-09 op #include <unistd.h>
 37 fb1a36c0 2022-01-09 op 
 38 fb1a36c0 2022-01-09 op #include "control.h"
 39 fb1a36c0 2022-01-09 op #include "kami.h"
 40 fb1a36c0 2022-01-09 op #include "kamid.h"
 41 fb1a36c0 2022-01-09 op #include "listener.h"
 42 fb1a36c0 2022-01-09 op #include "log.h"
 43 fb1a36c0 2022-01-09 op #include "sandbox.h"
 44 fb1a36c0 2022-01-09 op #include "utils.h"
 45 fb1a36c0 2022-01-09 op 
 46 8b0a5062 2022-05-22 op #define IMSG_MAXSIZE (MAX_IMSGSIZE - IMSG_HEADER_SIZE)
 47 8b0a5062 2022-05-22 op 
 48 fb1a36c0 2022-01-09 op static struct kd_conf	*listener_conf;
 49 fb1a36c0 2022-01-09 op static struct imsgev	*iev_main;
 50 fb1a36c0 2022-01-09 op 
 51 fb1a36c0 2022-01-09 op static void	listener_sig_handler(int, short, void *);
 52 fb1a36c0 2022-01-09 op __dead void	listener_shutdown(void);
 53 fb1a36c0 2022-01-09 op 
 54 fb1a36c0 2022-01-09 op SPLAY_HEAD(clients_tree_id, client) clients;
 55 fb1a36c0 2022-01-09 op 
 56 fb1a36c0 2022-01-09 op struct client {
 57 fb1a36c0 2022-01-09 op 	uint32_t		 id;
 58 fb1a36c0 2022-01-09 op 	uint32_t		 lid;
 59 38a6fd3d 2022-01-28 op 	uint32_t		 lflags;
 60 fb1a36c0 2022-01-09 op 	uint32_t		 msize;
 61 8b0a5062 2022-05-22 op 	uint32_t		 left;
 62 fb1a36c0 2022-01-09 op 	int			 fd;
 63 fb1a36c0 2022-01-09 op 	struct tls		*ctx;
 64 fb1a36c0 2022-01-09 op 	struct event		 event;
 65 fb1a36c0 2022-01-09 op 	struct imsgev		 iev;
 66 fb1a36c0 2022-01-09 op 	struct bufferevent	*bev;
 67 fb1a36c0 2022-01-09 op 	SPLAY_ENTRY(client)	 sp_entry;
 68 fb1a36c0 2022-01-09 op };
 69 fb1a36c0 2022-01-09 op 
 70 fb1a36c0 2022-01-09 op static void	listener_imsg_event_add(struct imsgev *, void *);
 71 fb1a36c0 2022-01-09 op static void	listener_dispatch_client(int, short, void *);
 72 fb1a36c0 2022-01-09 op static int	listener_imsg_compose_client(struct client *, int,
 73 fb1a36c0 2022-01-09 op     uint32_t, const void *, uint16_t);
 74 fb1a36c0 2022-01-09 op 
 75 fb1a36c0 2022-01-09 op static void	apply_config(struct kd_conf *);
 76 fb1a36c0 2022-01-09 op static void	handle_accept(int, short, void *);
 77 fb1a36c0 2022-01-09 op 
 78 fb1a36c0 2022-01-09 op static void	handle_handshake(int, short, void *);
 79 fb1a36c0 2022-01-09 op static void	client_read(struct bufferevent *, void *);
 80 fb1a36c0 2022-01-09 op static void	client_write(struct bufferevent *, void *);
 81 fb1a36c0 2022-01-09 op static void	client_error(struct bufferevent *, short, void *);
 82 fb1a36c0 2022-01-09 op static void	client_tls_readcb(int, short, void *);
 83 fb1a36c0 2022-01-09 op static void	client_tls_writecb(int, short, void *);
 84 fb1a36c0 2022-01-09 op static void	close_conn(struct client *);
 85 fb1a36c0 2022-01-09 op static void	handle_close(int, short, void *);
 86 fb1a36c0 2022-01-09 op 
 87 fb1a36c0 2022-01-09 op static inline int
 88 fb1a36c0 2022-01-09 op clients_tree_cmp(struct client *a, struct client *b)
 89 fb1a36c0 2022-01-09 op {
 90 fb1a36c0 2022-01-09 op 	if (a->id == b->id)
 91 fb1a36c0 2022-01-09 op 		return 0;
 92 fb1a36c0 2022-01-09 op 	else if (a->id < b->id)
 93 fb1a36c0 2022-01-09 op 		return -1;
 94 fb1a36c0 2022-01-09 op 	else
 95 fb1a36c0 2022-01-09 op 		return +1;
 96 fb1a36c0 2022-01-09 op }
 97 fb1a36c0 2022-01-09 op 
 98 fb1a36c0 2022-01-09 op SPLAY_PROTOTYPE(clients_tree_id, client, sp_entry, clients_tree_cmp);
 99 fb1a36c0 2022-01-09 op SPLAY_GENERATE(clients_tree_id, client, sp_entry, clients_tree_cmp)
100 fb1a36c0 2022-01-09 op 
101 fb1a36c0 2022-01-09 op static void
102 fb1a36c0 2022-01-09 op listener_sig_handler(int sig, short event, void *d)
103 fb1a36c0 2022-01-09 op {
104 fb1a36c0 2022-01-09 op 	/*
105 fb1a36c0 2022-01-09 op 	 * Normal signal handler rules don't apply because libevent
106 fb1a36c0 2022-01-09 op 	 * decouples for us.
107 fb1a36c0 2022-01-09 op 	 */
108 fb1a36c0 2022-01-09 op 
109 fb1a36c0 2022-01-09 op 	switch (sig) {
110 fb1a36c0 2022-01-09 op 	case SIGINT:
111 fb1a36c0 2022-01-09 op 	case SIGTERM:
112 fb1a36c0 2022-01-09 op 		listener_shutdown();
113 fb1a36c0 2022-01-09 op 	default:
114 fb1a36c0 2022-01-09 op 		fatalx("unexpected signal %d", sig);
115 fb1a36c0 2022-01-09 op 	}
116 fb1a36c0 2022-01-09 op }
117 fb1a36c0 2022-01-09 op 
118 fb1a36c0 2022-01-09 op void
119 fb1a36c0 2022-01-09 op listener(int debug, int verbose)
120 fb1a36c0 2022-01-09 op {
121 fb1a36c0 2022-01-09 op 	struct event		 ev_sigint, ev_sigterm;
122 fb1a36c0 2022-01-09 op 	struct passwd		*pw;
123 fb1a36c0 2022-01-09 op 
124 fb1a36c0 2022-01-09 op 	/* listener_conf = config_new_empty(); */
125 fb1a36c0 2022-01-09 op 
126 fb1a36c0 2022-01-09 op 	log_init(debug, LOG_DAEMON);
127 fb1a36c0 2022-01-09 op 	log_setverbose(verbose);
128 fb1a36c0 2022-01-09 op 
129 fb1a36c0 2022-01-09 op 	if ((pw = getpwnam(KD_USER)) == NULL)
130 fb1a36c0 2022-01-09 op 		fatal("getpwnam");
131 fb1a36c0 2022-01-09 op 
132 fb1a36c0 2022-01-09 op 	if (chroot(pw->pw_dir) == -1)
133 fb1a36c0 2022-01-09 op 		fatal("chroot");
134 fb1a36c0 2022-01-09 op 	if (chdir("/") == -1)
135 fb1a36c0 2022-01-09 op 		fatal("chdir(\"/\")");
136 fb1a36c0 2022-01-09 op 
137 fb1a36c0 2022-01-09 op 	setproctitle("listener");
138 fb1a36c0 2022-01-09 op 	log_procinit("listener");
139 fb1a36c0 2022-01-09 op 
140 fb1a36c0 2022-01-09 op 	if (setgroups(1, &pw->pw_gid) ||
141 fb1a36c0 2022-01-09 op 	    setresgid(pw->pw_gid, pw->pw_gid, pw->pw_gid) ||
142 fb1a36c0 2022-01-09 op 	    setresuid(pw->pw_uid, pw->pw_uid, pw->pw_uid))
143 fb1a36c0 2022-01-09 op 		fatal("can't drop privileges");
144 fb1a36c0 2022-01-09 op 
145 fb1a36c0 2022-01-09 op 	event_init();
146 fb1a36c0 2022-01-09 op 
147 fb1a36c0 2022-01-09 op 	/* Setup signal handlers(s). */
148 fb1a36c0 2022-01-09 op 	signal_set(&ev_sigint, SIGINT, listener_sig_handler, NULL);
149 fb1a36c0 2022-01-09 op 	signal_set(&ev_sigterm, SIGTERM, listener_sig_handler, NULL);
150 fb1a36c0 2022-01-09 op 
151 fb1a36c0 2022-01-09 op 	signal_add(&ev_sigint, NULL);
152 fb1a36c0 2022-01-09 op 	signal_add(&ev_sigterm, NULL);
153 fb1a36c0 2022-01-09 op 
154 fb1a36c0 2022-01-09 op 	signal(SIGPIPE, SIG_IGN);
155 fb1a36c0 2022-01-09 op 	signal(SIGHUP, SIG_IGN);
156 fb1a36c0 2022-01-09 op 
157 fb1a36c0 2022-01-09 op 	/* Setup pipe and event handler to the main process. */
158 fb1a36c0 2022-01-09 op 	if ((iev_main = malloc(sizeof(*iev_main))) == NULL)
159 fb1a36c0 2022-01-09 op 		fatal(NULL);
160 fb1a36c0 2022-01-09 op 
161 fb1a36c0 2022-01-09 op 	imsg_init(&iev_main->ibuf, 3);
162 fb1a36c0 2022-01-09 op 	iev_main->handler = listener_dispatch_main;
163 fb1a36c0 2022-01-09 op 
164 fb1a36c0 2022-01-09 op 	/* Setup event handlers. */
165 fb1a36c0 2022-01-09 op 	iev_main->events = EV_READ;
166 fb1a36c0 2022-01-09 op 	event_set(&iev_main->ev, iev_main->ibuf.fd, iev_main->events,
167 fb1a36c0 2022-01-09 op 	    iev_main->handler, iev_main);
168 fb1a36c0 2022-01-09 op 	event_add(&iev_main->ev, NULL);
169 fb1a36c0 2022-01-09 op 
170 fb1a36c0 2022-01-09 op 	sandbox_listener();
171 fb1a36c0 2022-01-09 op 	event_dispatch();
172 fb1a36c0 2022-01-09 op 	listener_shutdown();
173 fb1a36c0 2022-01-09 op }
174 fb1a36c0 2022-01-09 op 
175 fb1a36c0 2022-01-09 op __dead void
176 fb1a36c0 2022-01-09 op listener_shutdown(void)
177 fb1a36c0 2022-01-09 op {
178 fb1a36c0 2022-01-09 op 	msgbuf_clear(&iev_main->ibuf.w);
179 fb1a36c0 2022-01-09 op 	close(iev_main->ibuf.fd);
180 fb1a36c0 2022-01-09 op 
181 ef9db8f6 2022-01-28 op 	clear_config(listener_conf);
182 fb1a36c0 2022-01-09 op 
183 fb1a36c0 2022-01-09 op 	free(iev_main);
184 fb1a36c0 2022-01-09 op 
185 fb1a36c0 2022-01-09 op 	log_info("listener exiting");
186 fb1a36c0 2022-01-09 op 	exit(0);
187 fb1a36c0 2022-01-09 op }
188 fb1a36c0 2022-01-09 op 
189 fb1a36c0 2022-01-09 op static void
190 fb1a36c0 2022-01-09 op listener_receive_config(struct imsg *imsg, struct kd_conf **nconf,
191 fb1a36c0 2022-01-09 op     struct kd_pki_conf **pki)
192 fb1a36c0 2022-01-09 op {
193 fb1a36c0 2022-01-09 op 	struct kd_listen_conf *listen;
194 fb1a36c0 2022-01-09 op 	char *t;
195 fb1a36c0 2022-01-09 op 
196 fb1a36c0 2022-01-09 op 	switch (imsg->hdr.type) {
197 fb1a36c0 2022-01-09 op 	case IMSG_RECONF_CONF:
198 fb1a36c0 2022-01-09 op 		if (*nconf != NULL)
199 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_CONF already in "
200 fb1a36c0 2022-01-09 op 			    "progress", __func__);
201 fb1a36c0 2022-01-09 op 
202 fb1a36c0 2022-01-09 op 		if (IMSG_DATA_SIZE(*imsg) != sizeof(struct kd_conf))
203 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_CONF wrong length: %lu",
204 fb1a36c0 2022-01-09 op 			    __func__, IMSG_DATA_SIZE(*imsg));
205 fb1a36c0 2022-01-09 op 		if ((*nconf = malloc(sizeof(**nconf))) == NULL)
206 fb1a36c0 2022-01-09 op 			fatal(NULL);
207 fb1a36c0 2022-01-09 op 		memcpy(*nconf, imsg->data, sizeof(**nconf));
208 d70ef554 2022-01-28 op 		STAILQ_INIT(&(*nconf)->pki_head);
209 d70ef554 2022-01-28 op 		STAILQ_INIT(&(*nconf)->table_head);
210 d70ef554 2022-01-28 op 		STAILQ_INIT(&(*nconf)->listen_head);
211 fb1a36c0 2022-01-09 op 		break;
212 fb1a36c0 2022-01-09 op 	case IMSG_RECONF_PKI:
213 fb1a36c0 2022-01-09 op 		if (*nconf == NULL)
214 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_PKI without "
215 fb1a36c0 2022-01-09 op 			    "IMSG_RECONF_CONF", __func__);
216 fb1a36c0 2022-01-09 op 		*pki = xcalloc(1, sizeof(**pki));
217 fb1a36c0 2022-01-09 op 		t = imsg->data;
218 fb1a36c0 2022-01-09 op 		t[IMSG_DATA_SIZE(*imsg)-1] = '\0';
219 fb1a36c0 2022-01-09 op 		strlcpy((*pki)->name, t, sizeof((*pki)->name));
220 fb1a36c0 2022-01-09 op 		break;
221 fb1a36c0 2022-01-09 op 	case IMSG_RECONF_PKI_CERT:
222 fb1a36c0 2022-01-09 op 		if (*pki == NULL)
223 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_PKI_CERT without "
224 fb1a36c0 2022-01-09 op 			    "IMSG_RECONF_PKI", __func__);
225 fb1a36c0 2022-01-09 op 		(*pki)->certlen = IMSG_DATA_SIZE(*imsg);
226 fb1a36c0 2022-01-09 op 		(*pki)->cert = xmemdup(imsg->data, (*pki)->certlen);
227 fb1a36c0 2022-01-09 op 		break;
228 fb1a36c0 2022-01-09 op 	case IMSG_RECONF_PKI_KEY:
229 fb1a36c0 2022-01-09 op 		if (*pki == NULL)
230 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_PKI_KEY without "
231 fb1a36c0 2022-01-09 op 			    "IMSG_RECONF_PKI", __func__);
232 fb1a36c0 2022-01-09 op 		(*pki)->keylen = IMSG_DATA_SIZE(*imsg);
233 fb1a36c0 2022-01-09 op 		(*pki)->key = xmemdup(imsg->data, (*pki)->keylen);
234 fb1a36c0 2022-01-09 op 		STAILQ_INSERT_HEAD(&(*nconf)->pki_head, *pki, entry);
235 fb1a36c0 2022-01-09 op 		pki = NULL;
236 fb1a36c0 2022-01-09 op 		break;
237 fb1a36c0 2022-01-09 op 	case IMSG_RECONF_LISTEN:
238 fb1a36c0 2022-01-09 op 		if (*nconf == NULL)
239 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_LISTEN without "
240 fb1a36c0 2022-01-09 op 			    "IMSG_RECONF_CONF", __func__);
241 fb1a36c0 2022-01-09 op 		if (IMSG_DATA_SIZE(*imsg) != sizeof(*listen))
242 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_LISTEN wrong length: %lu",
243 fb1a36c0 2022-01-09 op 			    __func__, IMSG_DATA_SIZE(*imsg));
244 fb1a36c0 2022-01-09 op 		listen = xcalloc(1, sizeof(*listen));
245 fb1a36c0 2022-01-09 op 		memcpy(listen, imsg->data, sizeof(*listen));
246 fb1a36c0 2022-01-09 op 		memset(&listen->entry, 0, sizeof(listen->entry));
247 fb1a36c0 2022-01-09 op 		if ((listen->fd = imsg->fd) == -1)
248 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_LISTEN no fd",
249 fb1a36c0 2022-01-09 op 			    __func__);
250 fb1a36c0 2022-01-09 op 		listen->auth_table = NULL;
251 fb1a36c0 2022-01-09 op 		memset(&listen->ev, 0, sizeof(listen->ev));
252 fb1a36c0 2022-01-09 op 		STAILQ_INSERT_HEAD(&(*nconf)->listen_head, listen, entry);
253 fb1a36c0 2022-01-09 op 		break;
254 fb1a36c0 2022-01-09 op 	case IMSG_RECONF_END:
255 fb1a36c0 2022-01-09 op 		if (*nconf == NULL)
256 fb1a36c0 2022-01-09 op 			fatalx("%s: IMSG_RECONF_END without "
257 fb1a36c0 2022-01-09 op 			    "IMSG_RECONF_CONF", __func__);
258 fb1a36c0 2022-01-09 op 		apply_config(*nconf);
259 fb1a36c0 2022-01-09 op 		*nconf = NULL;
260 fb1a36c0 2022-01-09 op 		break;
261 fb1a36c0 2022-01-09 op 	}
262 fb1a36c0 2022-01-09 op }
263 fb1a36c0 2022-01-09 op 
264 fb1a36c0 2022-01-09 op void
265 fb1a36c0 2022-01-09 op listener_dispatch_main(int fd, short event, void *d)
266 fb1a36c0 2022-01-09 op {
267 fb1a36c0 2022-01-09 op 	static struct kd_conf		*nconf;
268 fb1a36c0 2022-01-09 op 	static struct kd_pki_conf	*pki;
269 fb1a36c0 2022-01-09 op 	struct client			*client, find;
270 fb1a36c0 2022-01-09 op 	struct imsg			 imsg;
271 fb1a36c0 2022-01-09 op 	struct imsgev			*iev = d;
272 fb1a36c0 2022-01-09 op 	struct imsgbuf			*ibuf;
273 fb1a36c0 2022-01-09 op 	ssize_t				 n;
274 fb1a36c0 2022-01-09 op 	int				 shut = 0;
275 fb1a36c0 2022-01-09 op 
276 fb1a36c0 2022-01-09 op 	ibuf = &iev->ibuf;
277 fb1a36c0 2022-01-09 op 
278 fb1a36c0 2022-01-09 op 	if (event & EV_READ) {
279 fb1a36c0 2022-01-09 op 		if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
280 fb1a36c0 2022-01-09 op 			fatal("imsg_read error");
281 fb1a36c0 2022-01-09 op 		if (n == 0)	/* Connection closed. */
282 fb1a36c0 2022-01-09 op 			shut = 1;
283 fb1a36c0 2022-01-09 op 	}
284 fb1a36c0 2022-01-09 op 	if (event & EV_WRITE) {
285 fb1a36c0 2022-01-09 op 		if ((n = msgbuf_write(&ibuf->w)) == -1 && errno != EAGAIN)
286 fb1a36c0 2022-01-09 op 			fatal("msgbuf_write");
287 fb1a36c0 2022-01-09 op 		if (n == 0)	/* Connection closed. */
288 fb1a36c0 2022-01-09 op 			shut = 1;
289 fb1a36c0 2022-01-09 op 	}
290 fb1a36c0 2022-01-09 op 
291 fb1a36c0 2022-01-09 op 	for (;;) {
292 fb1a36c0 2022-01-09 op 		if ((n = imsg_get(ibuf, &imsg)) == -1)
293 fb1a36c0 2022-01-09 op 			fatal("%s: imsg_get error", __func__);
294 fb1a36c0 2022-01-09 op 		if (n == 0)	/* No more messages. */
295 fb1a36c0 2022-01-09 op 			break;
296 fb1a36c0 2022-01-09 op 
297 fb1a36c0 2022-01-09 op 		switch (imsg.hdr.type) {
298 982f38dc 2022-01-23 op 		case IMSG_CTL_LOG_VERBOSE:
299 982f38dc 2022-01-23 op 			if (IMSG_DATA_SIZE(imsg) != sizeof(verbose))
300 982f38dc 2022-01-23 op 				fatalx("wrong size for IMSG_CTL_LOG_VERBOSE");
301 982f38dc 2022-01-23 op 			memcpy(&verbose, imsg.data, sizeof(verbose));
302 982f38dc 2022-01-23 op 			log_setverbose(verbose);
303 982f38dc 2022-01-23 op 			SPLAY_FOREACH(client, clients_tree_id, &clients)
304 982f38dc 2022-01-23 op 				listener_imsg_compose_client(client,
305 982f38dc 2022-01-23 op 				    imsg.hdr.type, 0,
306 982f38dc 2022-01-23 op 				    &verbose, sizeof(verbose));
307 5c69b561 2022-01-29 op 			break;
308 5c69b561 2022-01-29 op 		case IMSG_CTL_DEBUG:
309 5c69b561 2022-01-29 op 			if (SPLAY_EMPTY(&clients))
310 5c69b561 2022-01-29 op 				listener_imsg_compose_main(IMSG_CTL_DEBUG_END,
311 5c69b561 2022-01-29 op 				    imsg.hdr.peerid, NULL, 0);
312 5c69b561 2022-01-29 op 			SPLAY_FOREACH(client, clients_tree_id, &clients)
313 5c69b561 2022-01-29 op 				listener_imsg_compose_client(client,
314 5c69b561 2022-01-29 op 				    imsg.hdr.type, imsg.hdr.peerid,
315 5c69b561 2022-01-29 op 				    imsg.data, IMSG_DATA_SIZE(imsg));
316 982f38dc 2022-01-23 op 			break;
317 fb1a36c0 2022-01-09 op 		case IMSG_RECONF_CONF:
318 fb1a36c0 2022-01-09 op 		case IMSG_RECONF_PKI:
319 fb1a36c0 2022-01-09 op 		case IMSG_RECONF_PKI_CERT:
320 fb1a36c0 2022-01-09 op 		case IMSG_RECONF_PKI_KEY:
321 fb1a36c0 2022-01-09 op 		case IMSG_RECONF_LISTEN:
322 fb1a36c0 2022-01-09 op 		case IMSG_RECONF_END:
323 fb1a36c0 2022-01-09 op 			listener_receive_config(&imsg, &nconf, &pki);
324 fb1a36c0 2022-01-09 op 			break;
325 fb1a36c0 2022-01-09 op 		case IMSG_AUTH:
326 1e84c7b7 2022-01-10 op 			if (IMSG_DATA_SIZE(imsg) != sizeof(struct kd_auth_proc))
327 1e84c7b7 2022-01-10 op 				fatalx("mismatching size for IMSG_AUTH");
328 1e84c7b7 2022-01-10 op 
329 fb1a36c0 2022-01-09 op 			find.id = imsg.hdr.peerid;
330 fb1a36c0 2022-01-09 op 			client = SPLAY_FIND(clients_tree_id, &clients, &find);
331 fb1a36c0 2022-01-09 op 			if (client == NULL) {
332 fb1a36c0 2022-01-09 op 				if (imsg.fd != -1)
333 fb1a36c0 2022-01-09 op 					close(imsg.fd);
334 fb1a36c0 2022-01-09 op 				break;
335 fb1a36c0 2022-01-09 op 			}
336 fb1a36c0 2022-01-09 op 			if (imsg.fd == -1) {
337 fb1a36c0 2022-01-09 op 				log_info("got fd = -1, auth failed?");
338 fb1a36c0 2022-01-09 op 				close_conn(client);
339 fb1a36c0 2022-01-09 op 				break;
340 fb1a36c0 2022-01-09 op 			}
341 fb1a36c0 2022-01-09 op 			imsg_init(&client->iev.ibuf, imsg.fd);
342 fb1a36c0 2022-01-09 op 			client->iev.events = EV_READ;
343 fb1a36c0 2022-01-09 op 			client->iev.handler = listener_dispatch_client;
344 fb1a36c0 2022-01-09 op 			event_set(&client->iev.ev, client->iev.ibuf.fd,
345 fb1a36c0 2022-01-09 op 			    client->iev.events, client->iev.handler, client);
346 fb1a36c0 2022-01-09 op 			listener_imsg_compose_client(client, IMSG_AUTH,
347 fb1a36c0 2022-01-09 op 			    client->id, imsg.data, IMSG_DATA_SIZE(imsg));
348 fb1a36c0 2022-01-09 op 
349 fb1a36c0 2022-01-09 op 			client->bev = bufferevent_new(client->fd,
350 fb1a36c0 2022-01-09 op 			    client_read, client_write, client_error,
351 fb1a36c0 2022-01-09 op 			    client);
352 fb1a36c0 2022-01-09 op 			if (client->bev == NULL) {
353 fb1a36c0 2022-01-09 op 				log_info("failed to allocate client buffer");
354 fb1a36c0 2022-01-09 op 				close_conn(client);
355 fb1a36c0 2022-01-09 op 				return;
356 fb1a36c0 2022-01-09 op 			}
357 fb1a36c0 2022-01-09 op 
358 fb1a36c0 2022-01-09 op #if HAVE_EVENT2
359 fb1a36c0 2022-01-09 op 			evbuffer_unfreeze(client->bev->input, 0);
360 fb1a36c0 2022-01-09 op 			evbuffer_unfreeze(client->bev->output, 1);
361 fb1a36c0 2022-01-09 op #endif
362 fb1a36c0 2022-01-09 op 
363 38a6fd3d 2022-01-28 op 			if (client->lflags & L_TLS) {
364 fb1a36c0 2022-01-09 op 				event_set(&client->bev->ev_read, client->fd,
365 fb1a36c0 2022-01-09 op 				    EV_READ, client_tls_readcb, client->bev);
366 fb1a36c0 2022-01-09 op 				event_set(&client->bev->ev_write, client->fd,
367 fb1a36c0 2022-01-09 op 				    EV_WRITE, client_tls_writecb, client->bev);
368 fb1a36c0 2022-01-09 op 			}
369 fb1a36c0 2022-01-09 op 
370 fb1a36c0 2022-01-09 op 			/*
371 fb1a36c0 2022-01-09 op 			 * Read or write at least a header before
372 fb1a36c0 2022-01-09 op 			 * firing the callbacks.  High watermark of 0
373 fb1a36c0 2022-01-09 op 			 * to never stop reading/writing; probably to
374 fb1a36c0 2022-01-09 op 			 * be revisited.
375 fb1a36c0 2022-01-09 op 			 */
376 fb1a36c0 2022-01-09 op 			/* bufferevent_setwatermark(client->bev, EV_READ|EV_WRITE, */
377 fb1a36c0 2022-01-09 op 			    /* sizeof(struct np_msg_header), 0); */
378 fb1a36c0 2022-01-09 op 			bufferevent_enable(client->bev, EV_READ|EV_WRITE);
379 fb1a36c0 2022-01-09 op 			break;
380 fb1a36c0 2022-01-09 op 
381 fb1a36c0 2022-01-09 op 		default:
382 fb1a36c0 2022-01-09 op 			log_debug("%s: unexpected imsg %d", __func__,
383 fb1a36c0 2022-01-09 op 			    imsg.hdr.type);
384 fb1a36c0 2022-01-09 op 			break;
385 fb1a36c0 2022-01-09 op 		}
386 fb1a36c0 2022-01-09 op 		imsg_free(&imsg);
387 fb1a36c0 2022-01-09 op 	}
388 fb1a36c0 2022-01-09 op 
389 fb1a36c0 2022-01-09 op 	if (!shut)
390 fb1a36c0 2022-01-09 op 		listener_imsg_event_add(iev, d);
391 fb1a36c0 2022-01-09 op 	else {
392 fb1a36c0 2022-01-09 op 		/* This pipe is dead.  Remove its event handler. */
393 fb1a36c0 2022-01-09 op 		event_del(&iev->ev);
394 fb1a36c0 2022-01-09 op 		log_warnx("pipe closed, shutting down...");
395 fb1a36c0 2022-01-09 op 		event_loopexit(NULL);
396 fb1a36c0 2022-01-09 op 	}
397 fb1a36c0 2022-01-09 op }
398 fb1a36c0 2022-01-09 op 
399 fb1a36c0 2022-01-09 op int
400 fb1a36c0 2022-01-09 op listener_imsg_compose_main(int type, uint32_t peerid, const void *data,
401 fb1a36c0 2022-01-09 op     uint16_t datalen)
402 fb1a36c0 2022-01-09 op {
403 fb1a36c0 2022-01-09 op 	return imsg_compose_event(iev_main, type, peerid, 0, -1, data,
404 fb1a36c0 2022-01-09 op 	    datalen);
405 fb1a36c0 2022-01-09 op }
406 fb1a36c0 2022-01-09 op 
407 fb1a36c0 2022-01-09 op static void
408 fb1a36c0 2022-01-09 op listener_imsg_event_add(struct imsgev *iev, void *d)
409 fb1a36c0 2022-01-09 op {
410 fb1a36c0 2022-01-09 op 	iev->events = EV_READ;
411 fb1a36c0 2022-01-09 op 	if (iev->ibuf.w.queued)
412 fb1a36c0 2022-01-09 op 		iev->events |= EV_WRITE;
413 fb1a36c0 2022-01-09 op 
414 fb1a36c0 2022-01-09 op 	event_del(&iev->ev);
415 fb1a36c0 2022-01-09 op 	event_set(&iev->ev, iev->ibuf.fd, iev->events, iev->handler, d);
416 fb1a36c0 2022-01-09 op 	event_add(&iev->ev, NULL);
417 fb1a36c0 2022-01-09 op }
418 fb1a36c0 2022-01-09 op 
419 fb1a36c0 2022-01-09 op static void
420 fb1a36c0 2022-01-09 op listener_dispatch_client(int fd, short event, void *d)
421 fb1a36c0 2022-01-09 op {
422 fb1a36c0 2022-01-09 op         struct client	 find, *client = d;
423 fb1a36c0 2022-01-09 op 	struct imsg	 imsg;
424 fb1a36c0 2022-01-09 op 	struct imsgev	*iev;
425 fb1a36c0 2022-01-09 op 	struct imsgbuf	*ibuf;
426 fb1a36c0 2022-01-09 op 	ssize_t		 n;
427 fb1a36c0 2022-01-09 op 	int		 r, shut = 0;
428 fb1a36c0 2022-01-09 op 
429 fb1a36c0 2022-01-09 op 	iev = &client->iev;
430 fb1a36c0 2022-01-09 op 	ibuf = &iev->ibuf;
431 fb1a36c0 2022-01-09 op 
432 fb1a36c0 2022-01-09 op 	if (event & EV_READ) {
433 fb1a36c0 2022-01-09 op 		if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
434 fb1a36c0 2022-01-09 op 			fatal("imsg_read error");
435 fb1a36c0 2022-01-09 op 		if (n == 0)	/* Connection closed */
436 fb1a36c0 2022-01-09 op 			shut = 1;
437 fb1a36c0 2022-01-09 op 	}
438 fb1a36c0 2022-01-09 op 
439 fb1a36c0 2022-01-09 op 	if (event & EV_WRITE) {
440 fb1a36c0 2022-01-09 op 		if ((n = msgbuf_write(&ibuf->w)) == -1 && errno != EAGAIN)
441 fb1a36c0 2022-01-09 op 			fatal("msgbuf_write");
442 fb1a36c0 2022-01-09 op 		if (n == 0)	/* Connection closed. */
443 fb1a36c0 2022-01-09 op 			shut = 1;
444 fb1a36c0 2022-01-09 op 	}
445 fb1a36c0 2022-01-09 op 
446 fb1a36c0 2022-01-09 op 	for (;;) {
447 fb1a36c0 2022-01-09 op 		if ((n = imsg_get(ibuf, &imsg)) == -1)
448 fb1a36c0 2022-01-09 op 			fatal("%s: imsg_get error", __func__);
449 fb1a36c0 2022-01-09 op 		if (n == 0)	/* No more messages. */
450 fb1a36c0 2022-01-09 op 			break;
451 fb1a36c0 2022-01-09 op 
452 fb1a36c0 2022-01-09 op 		switch (imsg.hdr.type) {
453 5c69b561 2022-01-29 op 		case IMSG_CTL_DEBUG_BACK:
454 5c69b561 2022-01-29 op 		case IMSG_CTL_DEBUG_END:
455 5c69b561 2022-01-29 op 			listener_imsg_compose_main(imsg.hdr.type,
456 5c69b561 2022-01-29 op 			    imsg.hdr.peerid, imsg.data, IMSG_DATA_SIZE(imsg));
457 5c69b561 2022-01-29 op 			break;
458 5c69b561 2022-01-29 op 
459 fb1a36c0 2022-01-09 op 		case IMSG_BUF:
460 fb1a36c0 2022-01-09 op 			find.id = imsg.hdr.peerid;
461 fb1a36c0 2022-01-09 op 			client = SPLAY_FIND(clients_tree_id, &clients, &find);
462 fb1a36c0 2022-01-09 op 			if (client == NULL) {
463 f4bc229f 2022-01-29 op 				log_info("got IMSG_BUF but client %d gone",
464 fb1a36c0 2022-01-09 op 				    imsg.hdr.peerid);
465 fb1a36c0 2022-01-09 op 				break;
466 fb1a36c0 2022-01-09 op 			}
467 fb1a36c0 2022-01-09 op 			r = bufferevent_write(client->bev, imsg.data,
468 fb1a36c0 2022-01-09 op 			    IMSG_DATA_SIZE(imsg));
469 fb1a36c0 2022-01-09 op 			if (r == -1) {
470 fb1a36c0 2022-01-09 op 				log_warn("%s: bufferevent_write failed",
471 fb1a36c0 2022-01-09 op 				    __func__);
472 fb1a36c0 2022-01-09 op 				close_conn(client);
473 fb1a36c0 2022-01-09 op 				break;
474 fb1a36c0 2022-01-09 op 			}
475 fb1a36c0 2022-01-09 op 			break;
476 fb1a36c0 2022-01-09 op 
477 fb1a36c0 2022-01-09 op 		case IMSG_MSIZE:
478 fb1a36c0 2022-01-09 op 			if (IMSG_DATA_SIZE(imsg) != sizeof(client->msize))
479 fb1a36c0 2022-01-09 op 				fatal("IMSG_MSIZE size mismatch: "
480 fb1a36c0 2022-01-09 op 				    "got %zu want %zu", IMSG_DATA_SIZE(imsg),
481 fb1a36c0 2022-01-09 op 				    sizeof(client->msize));
482 fb1a36c0 2022-01-09 op 
483 fb1a36c0 2022-01-09 op 			memcpy(&client->msize, imsg.data,
484 fb1a36c0 2022-01-09 op 			    sizeof(client->msize));
485 fb1a36c0 2022-01-09 op 
486 fb1a36c0 2022-01-09 op 			if (client->msize == 0)
487 fb1a36c0 2022-01-09 op 				fatal("IMSG_MSIZE got msize = 0");
488 8b0a5062 2022-05-22 op 			log_debug("set msize to %d", client->msize);
489 fb1a36c0 2022-01-09 op 			break;
490 fb1a36c0 2022-01-09 op 
491 fb1a36c0 2022-01-09 op 		case IMSG_CLOSE:
492 fb1a36c0 2022-01-09 op 			/*
493 fb1a36c0 2022-01-09 op 			 * Both EVBUFFER_READ or EVBUFFER_WRITE should
494 fb1a36c0 2022-01-09 op 			 * be fine.
495 fb1a36c0 2022-01-09 op 			 */
496 fb1a36c0 2022-01-09 op 			client_error(client->bev, EVBUFFER_READ, client);
497 fb1a36c0 2022-01-09 op 			break;
498 fb1a36c0 2022-01-09 op 
499 fb1a36c0 2022-01-09 op 		default:
500 fb1a36c0 2022-01-09 op 			log_debug("%s: unexpected imsg %d", __func__,
501 fb1a36c0 2022-01-09 op 			    imsg.hdr.type);
502 fb1a36c0 2022-01-09 op 			break;
503 fb1a36c0 2022-01-09 op 		}
504 fb1a36c0 2022-01-09 op 		imsg_free(&imsg);
505 fb1a36c0 2022-01-09 op 	}
506 fb1a36c0 2022-01-09 op 
507 fb1a36c0 2022-01-09 op 	if (!shut)
508 fb1a36c0 2022-01-09 op 		listener_imsg_event_add(iev, d);
509 fb1a36c0 2022-01-09 op 	else {
510 fb1a36c0 2022-01-09 op 		/* This pipe is dead.  Remove its handler */
511 fb1a36c0 2022-01-09 op 		log_debug("client proc vanished");
512 fb1a36c0 2022-01-09 op                 close_conn(client);
513 fb1a36c0 2022-01-09 op 	}
514 fb1a36c0 2022-01-09 op }
515 fb1a36c0 2022-01-09 op 
516 fb1a36c0 2022-01-09 op static int
517 fb1a36c0 2022-01-09 op listener_imsg_compose_client(struct client *client, int type,
518 fb1a36c0 2022-01-09 op     uint32_t peerid, const void *data, uint16_t len)
519 fb1a36c0 2022-01-09 op {
520 fb1a36c0 2022-01-09 op 	int ret;
521 fb1a36c0 2022-01-09 op 
522 fb1a36c0 2022-01-09 op 	if ((ret = imsg_compose(&client->iev.ibuf, type, peerid, 0, -1,
523 fb1a36c0 2022-01-09 op 	    data, len)) != -1)
524 fb1a36c0 2022-01-09 op 		listener_imsg_event_add(&client->iev, client);
525 fb1a36c0 2022-01-09 op 
526 fb1a36c0 2022-01-09 op 	return ret;
527 fb1a36c0 2022-01-09 op }
528 fb1a36c0 2022-01-09 op 
529 fb1a36c0 2022-01-09 op static inline struct kd_pki_conf *
530 fb1a36c0 2022-01-09 op pki_by_name(const char *name)
531 fb1a36c0 2022-01-09 op {
532 fb1a36c0 2022-01-09 op         struct kd_pki_conf *pki;
533 fb1a36c0 2022-01-09 op 
534 fb1a36c0 2022-01-09 op 	STAILQ_FOREACH(pki, &listener_conf->pki_head, entry) {
535 fb1a36c0 2022-01-09 op 		if (!strcmp(name, pki->name))
536 fb1a36c0 2022-01-09 op 			return pki;
537 fb1a36c0 2022-01-09 op 	}
538 fb1a36c0 2022-01-09 op 
539 fb1a36c0 2022-01-09 op 	return NULL;
540 fb1a36c0 2022-01-09 op }
541 fb1a36c0 2022-01-09 op 
542 fb1a36c0 2022-01-09 op static void
543 fb1a36c0 2022-01-09 op apply_config(struct kd_conf *conf)
544 fb1a36c0 2022-01-09 op {
545 fb1a36c0 2022-01-09 op 	struct kd_pki_conf *pki;
546 fb1a36c0 2022-01-09 op 	struct kd_listen_conf *listen;
547 ac796d32 2022-01-28 op 	struct client *c;
548 ac796d32 2022-01-28 op 
549 ac796d32 2022-01-28 op 	/* drop any pre-auth inflight connections */
550 ac796d32 2022-01-28 op 	SPLAY_FOREACH(c, clients_tree_id, &clients) {
551 ac796d32 2022-01-28 op 		/*
552 ac796d32 2022-01-28 op 		 * c->event is set only during the handshake and the teardown
553 ac796d32 2022-01-28 op 		 * of the connection; c->bev is set only after auth.  Checking
554 ac796d32 2022-01-28 op 		 * for both ensures we drop only incoming connection in the
555 ac796d32 2022-01-28 op 		 * pre-auth state.
556 ac796d32 2022-01-28 op 		 */
557 ac796d32 2022-01-28 op 		if (event_pending(&c->event, EV_READ|EV_WRITE, NULL) &&
558 ac796d32 2022-01-28 op 		    c->bev == NULL) {
559 ac796d32 2022-01-28 op 			log_warn("closing in-flight connection due to reload");
560 ac796d32 2022-01-28 op 			close_conn(c);
561 ac796d32 2022-01-28 op 		}
562 ac796d32 2022-01-28 op 	}
563 fb1a36c0 2022-01-09 op 
564 4c0ae0c3 2022-01-28 op 	/* swap the now config with the current one */
565 4c0ae0c3 2022-01-28 op 	clear_config(listener_conf);
566 fb1a36c0 2022-01-09 op 	listener_conf = conf;
567 fb1a36c0 2022-01-09 op 
568 fb1a36c0 2022-01-09 op 	/* prepare the various tls_config */
569 fb1a36c0 2022-01-09 op 	STAILQ_FOREACH(pki, &listener_conf->pki_head, entry) {
570 fb1a36c0 2022-01-09 op 		if ((pki->tlsconf = tls_config_new()) == NULL)
571 fb1a36c0 2022-01-09 op 			fatal("tls_config_new");
572 fb1a36c0 2022-01-09 op 		tls_config_verify_client_optional(pki->tlsconf);
573 fb1a36c0 2022-01-09 op 		tls_config_insecure_noverifycert(pki->tlsconf);
574 fb1a36c0 2022-01-09 op 		if (tls_config_set_keypair_mem(pki->tlsconf,
575 fb1a36c0 2022-01-09 op 		    pki->cert, pki->certlen,
576 fb1a36c0 2022-01-09 op 		    pki->key, pki->keylen) == -1)
577 fb1a36c0 2022-01-09 op 			fatalx("tls_config_set_keypair_mem: %s",
578 fb1a36c0 2022-01-09 op 			    tls_config_error(pki->tlsconf));
579 fb1a36c0 2022-01-09 op 	}
580 fb1a36c0 2022-01-09 op 
581 fb1a36c0 2022-01-09 op 	/* prepare and kickoff the listeners */
582 fb1a36c0 2022-01-09 op 	STAILQ_FOREACH(listen, &listener_conf->listen_head, entry) {
583 fb1a36c0 2022-01-09 op 		if ((listen->ctx = tls_server()) == NULL)
584 fb1a36c0 2022-01-09 op 			fatal("tls_server");
585 fb1a36c0 2022-01-09 op 
586 fb1a36c0 2022-01-09 op 		pki = pki_by_name(listen->pki);
587 fb1a36c0 2022-01-09 op 		if (tls_configure(listen->ctx, pki->tlsconf) == -1)
588 fb1a36c0 2022-01-09 op 			fatalx("tls_configure: %s",
589 fb1a36c0 2022-01-09 op 			    tls_config_error(pki->tlsconf));
590 fb1a36c0 2022-01-09 op 
591 fb1a36c0 2022-01-09 op 		event_set(&listen->ev, listen->fd, EV_READ|EV_PERSIST,
592 fb1a36c0 2022-01-09 op 		    handle_accept, listen);
593 fb1a36c0 2022-01-09 op 		event_add(&listen->ev, NULL);
594 fb1a36c0 2022-01-09 op 	}
595 fb1a36c0 2022-01-09 op }
596 fb1a36c0 2022-01-09 op 
597 fb1a36c0 2022-01-09 op static inline void
598 fb1a36c0 2022-01-09 op yield_r(struct client *c, void (*fn)(int, short, void *))
599 fb1a36c0 2022-01-09 op {
600 fb1a36c0 2022-01-09 op 	if (event_pending(&c->event, EV_WRITE|EV_READ, NULL))
601 fb1a36c0 2022-01-09 op 		event_del(&c->event);
602 fb1a36c0 2022-01-09 op 	event_set(&c->event, c->fd, EV_READ, fn, c);
603 fb1a36c0 2022-01-09 op 	event_add(&c->event, NULL);
604 fb1a36c0 2022-01-09 op }
605 fb1a36c0 2022-01-09 op 
606 fb1a36c0 2022-01-09 op static inline void
607 fb1a36c0 2022-01-09 op yield_w(struct client *c, void (*fn)(int, short, void *))
608 fb1a36c0 2022-01-09 op {
609 fb1a36c0 2022-01-09 op 	if (event_pending(&c->event, EV_WRITE|EV_READ, NULL))
610 fb1a36c0 2022-01-09 op 		event_del(&c->event);
611 fb1a36c0 2022-01-09 op 	event_set(&c->event, c->fd, EV_WRITE, fn, c);
612 fb1a36c0 2022-01-09 op 	event_add(&c->event, NULL);
613 fb1a36c0 2022-01-09 op }
614 fb1a36c0 2022-01-09 op 
615 fb1a36c0 2022-01-09 op static void
616 fb1a36c0 2022-01-09 op handle_accept(int fd, short ev, void *data)
617 fb1a36c0 2022-01-09 op {
618 057b5ab7 2022-01-28 op 	static uint32_t counter;
619 fb1a36c0 2022-01-09 op 	struct kd_listen_conf *listen = data;
620 fb1a36c0 2022-01-09 op 	struct client *c;
621 fb1a36c0 2022-01-09 op 	int s;
622 fb1a36c0 2022-01-09 op 
623 fb1a36c0 2022-01-09 op 	if ((s = accept(fd, NULL, NULL)) == -1) {
624 fb1a36c0 2022-01-09 op 		log_warn("accept");
625 fb1a36c0 2022-01-09 op 		return;
626 fb1a36c0 2022-01-09 op 	}
627 fb1a36c0 2022-01-09 op 
628 fb1a36c0 2022-01-09 op 	c = xcalloc(1, sizeof(*c));
629 fb1a36c0 2022-01-09 op 	c->msize = MSIZE9P;
630 fb1a36c0 2022-01-09 op 	c->lid = listen->id;
631 38a6fd3d 2022-01-28 op 	c->lflags = listen->flags;
632 fb1a36c0 2022-01-09 op 	c->iev.ibuf.fd = -1;
633 fb1a36c0 2022-01-09 op 
634 fb1a36c0 2022-01-09 op 	if (tls_accept_socket(listen->ctx, &c->ctx, s) == -1) {
635 fb1a36c0 2022-01-09 op 		log_warnx("tls_accept_socket: %s",
636 fb1a36c0 2022-01-09 op 		    tls_error(listen->ctx));
637 fb1a36c0 2022-01-09 op 		free(c);
638 fb1a36c0 2022-01-09 op 		close(s);
639 fb1a36c0 2022-01-09 op 		return;
640 fb1a36c0 2022-01-09 op 	}
641 fb1a36c0 2022-01-09 op 
642 fb1a36c0 2022-01-09 op 	c->fd = s;
643 057b5ab7 2022-01-28 op 	c->id = counter++;
644 fb1a36c0 2022-01-09 op 
645 fb1a36c0 2022-01-09 op 	SPLAY_INSERT(clients_tree_id, &clients, c);
646 fb1a36c0 2022-01-09 op 
647 fb1a36c0 2022-01-09 op 	/* initialize the event */
648 fb1a36c0 2022-01-09 op 	event_set(&c->event, c->fd, EV_READ, NULL, NULL);
649 fb1a36c0 2022-01-09 op 
650 fb1a36c0 2022-01-09 op 	yield_r(c, handle_handshake);
651 fb1a36c0 2022-01-09 op }
652 fb1a36c0 2022-01-09 op 
653 fb1a36c0 2022-01-09 op static void
654 fb1a36c0 2022-01-09 op handle_handshake(int fd, short ev, void *data)
655 fb1a36c0 2022-01-09 op {
656 fb1a36c0 2022-01-09 op 	struct client *c = data;
657 fb1a36c0 2022-01-09 op 	struct kd_auth_req auth;
658 fb1a36c0 2022-01-09 op 	ssize_t r;
659 fb1a36c0 2022-01-09 op 	const char *hash;
660 fb1a36c0 2022-01-09 op 
661 fb1a36c0 2022-01-09 op 	switch (r = tls_handshake(c->ctx)) {
662 fb1a36c0 2022-01-09 op 	case TLS_WANT_POLLIN:
663 fb1a36c0 2022-01-09 op 		yield_r(c, handle_handshake);
664 fb1a36c0 2022-01-09 op 		return;
665 fb1a36c0 2022-01-09 op 	case TLS_WANT_POLLOUT:
666 fb1a36c0 2022-01-09 op 		yield_w(c, handle_handshake);
667 fb1a36c0 2022-01-09 op 		return;
668 fb1a36c0 2022-01-09 op 	case -1:
669 fb1a36c0 2022-01-09 op 		log_debug("handhsake failed: %s", tls_error(c->ctx));
670 fb1a36c0 2022-01-09 op 		close_conn(c);
671 fb1a36c0 2022-01-09 op 		return;
672 fb1a36c0 2022-01-09 op 	}
673 fb1a36c0 2022-01-09 op 
674 fb1a36c0 2022-01-09 op 	if ((hash = tls_peer_cert_hash(c->ctx)) == NULL) {
675 fb1a36c0 2022-01-09 op 		log_warnx("client didn't provide certificate");
676 fb1a36c0 2022-01-09 op 		close_conn(c);
677 fb1a36c0 2022-01-09 op 		return;
678 fb1a36c0 2022-01-09 op 	}
679 fb1a36c0 2022-01-09 op 
680 fb1a36c0 2022-01-09 op 	memset(&auth, 0, sizeof(auth));
681 fb1a36c0 2022-01-09 op 	auth.listen_id = c->lid;
682 fb1a36c0 2022-01-09 op 	strlcpy(auth.hash, hash, sizeof(auth.hash));
683 fb1a36c0 2022-01-09 op 	log_debug("sending hash %s", auth.hash);
684 fb1a36c0 2022-01-09 op 
685 fb1a36c0 2022-01-09 op 	listener_imsg_compose_main(IMSG_AUTH_TLS, c->id,
686 fb1a36c0 2022-01-09 op 	    &auth, sizeof(auth));
687 fb1a36c0 2022-01-09 op }
688 fb1a36c0 2022-01-09 op 
689 fb1a36c0 2022-01-09 op static void
690 fb1a36c0 2022-01-09 op client_read(struct bufferevent *bev, void *d)
691 fb1a36c0 2022-01-09 op {
692 fb1a36c0 2022-01-09 op 	struct client	*client = d;
693 fb1a36c0 2022-01-09 op 	struct evbuffer	*src = EVBUFFER_INPUT(bev);
694 8b0a5062 2022-05-22 op 	size_t evlen;
695 8b0a5062 2022-05-22 op 	uint32_t len;
696 fb1a36c0 2022-01-09 op 
697 fb1a36c0 2022-01-09 op 	for (;;) {
698 8b0a5062 2022-05-22 op 		evlen = EVBUFFER_LENGTH(src);
699 8b0a5062 2022-05-22 op 
700 8b0a5062 2022-05-22 op 		if (client->left != 0) {
701 8b0a5062 2022-05-22 op 			/* wait to fill a whole imsg if possible */
702 8b0a5062 2022-05-22 op 			if (client->left >= IMSG_MAXSIZE &&
703 8b0a5062 2022-05-22 op 			    evlen < IMSG_MAXSIZE)
704 8b0a5062 2022-05-22 op 				return;
705 8b0a5062 2022-05-22 op 
706 8b0a5062 2022-05-22 op 			len = MIN(client->left, evlen);
707 c234caae 2022-05-23 op 			len = MIN(len, IMSG_MAXSIZE);
708 c234caae 2022-05-23 op 
709 8b0a5062 2022-05-22 op 			listener_imsg_compose_client(client, IMSG_BUF_CONT,
710 8b0a5062 2022-05-22 op 			    client->id, EVBUFFER_DATA(src), len);
711 8b0a5062 2022-05-22 op 			evbuffer_drain(src, len);
712 8b0a5062 2022-05-22 op 			client->left -= len;
713 8b0a5062 2022-05-22 op 			continue;
714 8b0a5062 2022-05-22 op 		}
715 8b0a5062 2022-05-22 op 
716 8b0a5062 2022-05-22 op 		if (evlen < 4)
717 fb1a36c0 2022-01-09 op 			return;
718 fb1a36c0 2022-01-09 op 
719 fb1a36c0 2022-01-09 op 		memcpy(&len, EVBUFFER_DATA(src), sizeof(len));
720 fb1a36c0 2022-01-09 op 		len = le32toh(len);
721 fb1a36c0 2022-01-09 op 		log_debug("expecting a message %"PRIu32" bytes long "
722 8b0a5062 2022-05-22 op 		    "(of wich %zu already read)", len, evlen);
723 fb1a36c0 2022-01-09 op 
724 fb1a36c0 2022-01-09 op 		if (len < HEADERSIZE) {
725 fb1a36c0 2022-01-09 op 			log_warnx("invalid message size %d (too low)", len);
726 fb1a36c0 2022-01-09 op 			client_error(bev, EVBUFFER_READ, client);
727 fb1a36c0 2022-01-09 op 			return;
728 fb1a36c0 2022-01-09 op 		}
729 fb1a36c0 2022-01-09 op 
730 fb1a36c0 2022-01-09 op 		if (len > client->msize) {
731 fb1a36c0 2022-01-09 op 			log_warnx("incoming message bigger than msize "
732 fb1a36c0 2022-01-09 op 			    "(%"PRIu32" vs %"PRIu32")", len, client->msize);
733 fb1a36c0 2022-01-09 op 			client_error(bev, EVBUFFER_READ, client);
734 fb1a36c0 2022-01-09 op 			return;
735 fb1a36c0 2022-01-09 op 		}
736 fb1a36c0 2022-01-09 op 
737 8b0a5062 2022-05-22 op 		if (len > IMSG_MAXSIZE && evlen >= len) {
738 8b0a5062 2022-05-22 op 			listener_imsg_compose_client(client, IMSG_BUF,
739 8b0a5062 2022-05-22 op 			    client->id, EVBUFFER_DATA(src), IMSG_MAXSIZE);
740 8b0a5062 2022-05-22 op 			evbuffer_drain(src, IMSG_MAXSIZE);
741 8b0a5062 2022-05-22 op 			client->left = len - IMSG_MAXSIZE;
742 8b0a5062 2022-05-22 op 			continue;
743 8b0a5062 2022-05-22 op 		}
744 8b0a5062 2022-05-22 op 
745 8b0a5062 2022-05-22 op 		if (len > evlen)
746 fb1a36c0 2022-01-09 op 			return;
747 fb1a36c0 2022-01-09 op 
748 fb1a36c0 2022-01-09 op 		listener_imsg_compose_client(client, IMSG_BUF, client->id,
749 fb1a36c0 2022-01-09 op 		    EVBUFFER_DATA(src), len);
750 fb1a36c0 2022-01-09 op 		evbuffer_drain(src, len);
751 fb1a36c0 2022-01-09 op 	}
752 fb1a36c0 2022-01-09 op }
753 fb1a36c0 2022-01-09 op 
754 fb1a36c0 2022-01-09 op static void
755 fb1a36c0 2022-01-09 op client_write(struct bufferevent *bev, void *d)
756 fb1a36c0 2022-01-09 op {
757 fb1a36c0 2022-01-09 op 	/*
758 fb1a36c0 2022-01-09 op 	 * here we can do some fancy logic like deciding when to call
759 fb1a36c0 2022-01-09 op 	 *
760 fb1a36c0 2022-01-09 op 	 *	(*bev->errorcb)(bev, EVBUFFER_WRITE, bev->cbarg)
761 fb1a36c0 2022-01-09 op 	 *
762 fb1a36c0 2022-01-09 op 	 * to signal the end of the transaction.
763 fb1a36c0 2022-01-09 op 	 */
764 fb1a36c0 2022-01-09 op 
765 fb1a36c0 2022-01-09 op 	return;
766 fb1a36c0 2022-01-09 op }
767 fb1a36c0 2022-01-09 op 
768 fb1a36c0 2022-01-09 op static void
769 fb1a36c0 2022-01-09 op client_error(struct bufferevent *bev, short err, void *d)
770 fb1a36c0 2022-01-09 op {
771 fb1a36c0 2022-01-09 op 	struct client	*client = d;
772 fb1a36c0 2022-01-09 op 	struct evbuffer	*buf;
773 fb1a36c0 2022-01-09 op 
774 fb1a36c0 2022-01-09 op         if (err & EVBUFFER_ERROR) {
775 fb1a36c0 2022-01-09 op 		if (errno == EFBIG) {
776 fb1a36c0 2022-01-09 op 			bufferevent_enable(bev, EV_READ);
777 fb1a36c0 2022-01-09 op 			return;
778 fb1a36c0 2022-01-09 op 		}
779 fb1a36c0 2022-01-09 op 		log_debug("buffer event error");
780 fb1a36c0 2022-01-09 op                 close_conn(client);
781 fb1a36c0 2022-01-09 op 		return;
782 fb1a36c0 2022-01-09 op 	}
783 fb1a36c0 2022-01-09 op 
784 fb1a36c0 2022-01-09 op 	if (err & EVBUFFER_EOF) {
785 fb1a36c0 2022-01-09 op                 close_conn(client);
786 fb1a36c0 2022-01-09 op 		return;
787 fb1a36c0 2022-01-09 op 	}
788 fb1a36c0 2022-01-09 op 
789 fb1a36c0 2022-01-09 op 	if (err & (EVBUFFER_READ|EVBUFFER_WRITE)) {
790 fb1a36c0 2022-01-09 op 		bufferevent_disable(bev, EV_READ|EV_WRITE);
791 fb1a36c0 2022-01-09 op 
792 fb1a36c0 2022-01-09 op 		buf = EVBUFFER_OUTPUT(client->bev);
793 fb1a36c0 2022-01-09 op 		if (EVBUFFER_LENGTH(buf) != 0) {
794 fb1a36c0 2022-01-09 op 			/* finish writing all the data first */
795 fb1a36c0 2022-01-09 op 			bufferevent_enable(client->bev, EV_WRITE);
796 fb1a36c0 2022-01-09 op 			return;
797 fb1a36c0 2022-01-09 op 		}
798 fb1a36c0 2022-01-09 op 
799 fb1a36c0 2022-01-09 op 		close_conn(client);
800 fb1a36c0 2022-01-09 op 		return;
801 fb1a36c0 2022-01-09 op 	}
802 fb1a36c0 2022-01-09 op 
803 fb1a36c0 2022-01-09 op 	log_warnx("unknown event error, closing client connection");
804 fb1a36c0 2022-01-09 op 	close_conn(client);
805 fb1a36c0 2022-01-09 op }
806 fb1a36c0 2022-01-09 op 
807 fb1a36c0 2022-01-09 op static void
808 fb1a36c0 2022-01-09 op client_tls_readcb(int fd, short event, void *d)
809 fb1a36c0 2022-01-09 op {
810 fb1a36c0 2022-01-09 op 	struct bufferevent	*bufev = d;
811 fb1a36c0 2022-01-09 op 	struct client		*client = bufev->cbarg;
812 fb1a36c0 2022-01-09 op 	char			 buf[IBUF_READ_SIZE];
813 fb1a36c0 2022-01-09 op 	int			 what = EVBUFFER_READ;
814 fb1a36c0 2022-01-09 op 	int			 howmuch = IBUF_READ_SIZE;
815 fb1a36c0 2022-01-09 op 	ssize_t			 ret;
816 fb1a36c0 2022-01-09 op 	size_t			 len;
817 fb1a36c0 2022-01-09 op 
818 fb1a36c0 2022-01-09 op 	if (event == EV_TIMEOUT) {
819 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_TIMEOUT;
820 fb1a36c0 2022-01-09 op 		goto err;
821 fb1a36c0 2022-01-09 op 	}
822 fb1a36c0 2022-01-09 op 
823 fb1a36c0 2022-01-09 op 	if (bufev->wm_read.high != 0)
824 fb1a36c0 2022-01-09 op 		howmuch = MIN(sizeof(buf), bufev->wm_read.high);
825 fb1a36c0 2022-01-09 op 
826 fb1a36c0 2022-01-09 op         switch (ret = tls_read(client->ctx, buf, howmuch)) {
827 fb1a36c0 2022-01-09 op 	case TLS_WANT_POLLIN:
828 fb1a36c0 2022-01-09 op 	case TLS_WANT_POLLOUT:
829 fb1a36c0 2022-01-09 op 		goto retry;
830 fb1a36c0 2022-01-09 op 	case -1:
831 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_ERROR;
832 fb1a36c0 2022-01-09 op 		goto err;
833 fb1a36c0 2022-01-09 op 	}
834 fb1a36c0 2022-01-09 op 	len = ret;
835 fb1a36c0 2022-01-09 op 
836 fb1a36c0 2022-01-09 op 	if (len == 0) {
837 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_EOF;
838 fb1a36c0 2022-01-09 op 		goto err;
839 fb1a36c0 2022-01-09 op 	}
840 fb1a36c0 2022-01-09 op 
841 fb1a36c0 2022-01-09 op 	if (evbuffer_add(bufev->input, buf, len) == -1) {
842 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_ERROR;
843 fb1a36c0 2022-01-09 op 		goto err;
844 fb1a36c0 2022-01-09 op 	}
845 fb1a36c0 2022-01-09 op 
846 fb1a36c0 2022-01-09 op 	event_add(&bufev->ev_read, NULL);
847 fb1a36c0 2022-01-09 op 
848 fb1a36c0 2022-01-09 op 	len = EVBUFFER_LENGTH(bufev->input);
849 fb1a36c0 2022-01-09 op 	if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
850 fb1a36c0 2022-01-09 op 		return;
851 fb1a36c0 2022-01-09 op 	if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
852 fb1a36c0 2022-01-09 op 		/*
853 fb1a36c0 2022-01-09 op 		 * here we could implement some read pressure
854 fb1a36c0 2022-01-09 op 		 * mechanism.
855 fb1a36c0 2022-01-09 op 		 */
856 fb1a36c0 2022-01-09 op 	}
857 fb1a36c0 2022-01-09 op 
858 fb1a36c0 2022-01-09 op 	if (bufev->readcb != NULL)
859 fb1a36c0 2022-01-09 op 		(*bufev->readcb)(bufev, bufev->cbarg);
860 fb1a36c0 2022-01-09 op 
861 fb1a36c0 2022-01-09 op 	return;
862 fb1a36c0 2022-01-09 op 
863 fb1a36c0 2022-01-09 op retry:
864 fb1a36c0 2022-01-09 op 	event_add(&bufev->ev_read, NULL);
865 fb1a36c0 2022-01-09 op 	return;
866 fb1a36c0 2022-01-09 op 
867 fb1a36c0 2022-01-09 op err:
868 fb1a36c0 2022-01-09 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
869 fb1a36c0 2022-01-09 op }
870 fb1a36c0 2022-01-09 op 
871 fb1a36c0 2022-01-09 op static void
872 fb1a36c0 2022-01-09 op client_tls_writecb(int fd, short event, void *d)
873 fb1a36c0 2022-01-09 op {
874 fb1a36c0 2022-01-09 op 	struct bufferevent	*bufev = d;
875 fb1a36c0 2022-01-09 op 	struct client		*client = bufev->cbarg;
876 fb1a36c0 2022-01-09 op 	ssize_t			 ret;
877 fb1a36c0 2022-01-09 op 	size_t			 len;
878 fb1a36c0 2022-01-09 op 	short			 what = EVBUFFER_WRITE;
879 fb1a36c0 2022-01-09 op 
880 fb1a36c0 2022-01-09 op 	if (event == EV_TIMEOUT) {
881 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_TIMEOUT;
882 fb1a36c0 2022-01-09 op 		goto err;
883 fb1a36c0 2022-01-09 op 	}
884 fb1a36c0 2022-01-09 op 
885 fb1a36c0 2022-01-09 op 	if (EVBUFFER_LENGTH(bufev->output) != 0) {
886 fb1a36c0 2022-01-09 op 		ret = tls_write(client->ctx,
887 fb1a36c0 2022-01-09 op 		    EVBUFFER_DATA(bufev->output),
888 fb1a36c0 2022-01-09 op 		    EVBUFFER_LENGTH(bufev->output));
889 fb1a36c0 2022-01-09 op 		switch (ret) {
890 fb1a36c0 2022-01-09 op 		case TLS_WANT_POLLIN:
891 fb1a36c0 2022-01-09 op 		case TLS_WANT_POLLOUT:
892 fb1a36c0 2022-01-09 op 			goto retry;
893 fb1a36c0 2022-01-09 op 		case -1:
894 fb1a36c0 2022-01-09 op 			what |= EVBUFFER_ERROR;
895 fb1a36c0 2022-01-09 op 			goto err;
896 fb1a36c0 2022-01-09 op 		}
897 fb1a36c0 2022-01-09 op 		len = ret;
898 fb1a36c0 2022-01-09 op 		evbuffer_drain(bufev->output, len);
899 fb1a36c0 2022-01-09 op 	}
900 fb1a36c0 2022-01-09 op 
901 fb1a36c0 2022-01-09 op 	if (EVBUFFER_LENGTH(bufev->output) != 0)
902 fb1a36c0 2022-01-09 op 		event_add(&bufev->ev_write, NULL);
903 fb1a36c0 2022-01-09 op 
904 fb1a36c0 2022-01-09 op 	if (bufev->writecb != NULL &&
905 fb1a36c0 2022-01-09 op 	    EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
906 fb1a36c0 2022-01-09 op 		(*bufev->writecb)(bufev, bufev->cbarg);
907 fb1a36c0 2022-01-09 op 	return;
908 fb1a36c0 2022-01-09 op 
909 fb1a36c0 2022-01-09 op retry:
910 fb1a36c0 2022-01-09 op 	event_add(&bufev->ev_write, NULL);
911 fb1a36c0 2022-01-09 op 	return;
912 fb1a36c0 2022-01-09 op 
913 fb1a36c0 2022-01-09 op err:
914 fb1a36c0 2022-01-09 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
915 fb1a36c0 2022-01-09 op }
916 fb1a36c0 2022-01-09 op 
917 fb1a36c0 2022-01-09 op static void
918 fb1a36c0 2022-01-09 op close_conn(struct client *c)
919 fb1a36c0 2022-01-09 op {
920 fb1a36c0 2022-01-09 op 	log_debug("closing connection");
921 2ab2c437 2022-01-28 op 
922 2ab2c437 2022-01-28 op 	SPLAY_REMOVE(clients_tree_id, &clients, c);
923 fb1a36c0 2022-01-09 op 
924 fb1a36c0 2022-01-09 op 	if (c->iev.ibuf.fd != -1) {
925 fb1a36c0 2022-01-09 op 		listener_imsg_compose_client(c, IMSG_CONN_GONE, 0, NULL, 0);
926 fb1a36c0 2022-01-09 op 		imsg_flush(&c->iev.ibuf);
927 fb1a36c0 2022-01-09 op 		msgbuf_clear(&c->iev.ibuf.w);
928 fb1a36c0 2022-01-09 op 		event_del(&c->iev.ev);
929 fb1a36c0 2022-01-09 op 		close(c->iev.ibuf.fd);
930 fb1a36c0 2022-01-09 op 	}
931 fb1a36c0 2022-01-09 op 
932 fb1a36c0 2022-01-09 op 	handle_close(c->fd, 0, c);
933 fb1a36c0 2022-01-09 op }
934 fb1a36c0 2022-01-09 op 
935 fb1a36c0 2022-01-09 op static void
936 fb1a36c0 2022-01-09 op handle_close(int fd, short ev, void *d)
937 fb1a36c0 2022-01-09 op {
938 fb1a36c0 2022-01-09 op 	struct client *c = d;
939 fb1a36c0 2022-01-09 op 
940 fb1a36c0 2022-01-09 op 	switch (tls_close(c->ctx)) {
941 fb1a36c0 2022-01-09 op 	case TLS_WANT_POLLIN:
942 fb1a36c0 2022-01-09 op 		yield_r(c, handle_close);
943 fb1a36c0 2022-01-09 op 		return;
944 fb1a36c0 2022-01-09 op 	case TLS_WANT_POLLOUT:
945 fb1a36c0 2022-01-09 op 		yield_w(c, handle_close);
946 fb1a36c0 2022-01-09 op 		return;
947 fb1a36c0 2022-01-09 op 	}
948 fb1a36c0 2022-01-09 op 
949 fb1a36c0 2022-01-09 op 	event_del(&c->event);
950 fb1a36c0 2022-01-09 op 	tls_free(c->ctx);
951 fb1a36c0 2022-01-09 op 	close(c->fd);
952 fb1a36c0 2022-01-09 op 	free(c);
953 fb1a36c0 2022-01-09 op }