op public repos

Blame

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