1 ce062d50 2019-08-21 martijn /*	$OpenBSD: ioev.c,v 1.42 2019/06/12 17:42:53 eric Exp $	*/
   2 ce062d50 2019-08-21 martijn /*
   3 ce062d50 2019-08-21 martijn  * Copyright (c) 2012 Eric Faurot <eric@openbsd.org>
   4 ce062d50 2019-08-21 martijn  *
   5 ce062d50 2019-08-21 martijn  * Permission to use, copy, modify, and distribute this software for any
   6 ce062d50 2019-08-21 martijn  * purpose with or without fee is hereby granted, provided that the above
   7 ce062d50 2019-08-21 martijn  * copyright notice and this permission notice appear in all copies.
   8 ce062d50 2019-08-21 martijn  *
   9 ce062d50 2019-08-21 martijn  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  10 ce062d50 2019-08-21 martijn  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11 ce062d50 2019-08-21 martijn  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  12 ce062d50 2019-08-21 martijn  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13 ce062d50 2019-08-21 martijn  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14 ce062d50 2019-08-21 martijn  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  15 ce062d50 2019-08-21 martijn  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16 ce062d50 2019-08-21 martijn  */
  17 ce062d50 2019-08-21 martijn 
  18 ce062d50 2019-08-21 martijn #include <sys/types.h>
  19 ce062d50 2019-08-21 martijn #include <sys/queue.h>
  20 ce062d50 2019-08-21 martijn #include <sys/socket.h>
  21 ce062d50 2019-08-21 martijn 
  22 ce062d50 2019-08-21 martijn #include <err.h>
  23 ce062d50 2019-08-21 martijn #include <errno.h>
  24 ce062d50 2019-08-21 martijn #include <event.h>
  25 ce062d50 2019-08-21 martijn #include <fcntl.h>
  26 ce062d50 2019-08-21 martijn #include <inttypes.h>
  27 ce062d50 2019-08-21 martijn #include <stdlib.h>
  28 ce062d50 2019-08-21 martijn #include <string.h>
  29 ce062d50 2019-08-21 martijn #include <stdio.h>
  30 ce062d50 2019-08-21 martijn #include <unistd.h>
  31 ce062d50 2019-08-21 martijn 
  32 ce062d50 2019-08-21 martijn #include "ioev.h"
  33 ce062d50 2019-08-21 martijn #include "iobuf.h"
  34 ce062d50 2019-08-21 martijn 
  35 ce062d50 2019-08-21 martijn #ifdef IO_TLS
  36 ce062d50 2019-08-21 martijn #include <openssl/err.h>
  37 ce062d50 2019-08-21 martijn #include <openssl/ssl.h>
  38 ce062d50 2019-08-21 martijn #endif
  39 ce062d50 2019-08-21 martijn 
  40 ce062d50 2019-08-21 martijn enum {
  41 ce062d50 2019-08-21 martijn 	IO_STATE_NONE,
  42 ce062d50 2019-08-21 martijn 	IO_STATE_CONNECT,
  43 ce062d50 2019-08-21 martijn 	IO_STATE_CONNECT_TLS,
  44 ce062d50 2019-08-21 martijn 	IO_STATE_ACCEPT_TLS,
  45 ce062d50 2019-08-21 martijn 	IO_STATE_UP,
  46 ce062d50 2019-08-21 martijn 
  47 ce062d50 2019-08-21 martijn 	IO_STATE_MAX,
  48 ce062d50 2019-08-21 martijn };
  49 ce062d50 2019-08-21 martijn 
  50 ce062d50 2019-08-21 martijn #define IO_PAUSE_IN 		IO_IN
  51 ce062d50 2019-08-21 martijn #define IO_PAUSE_OUT		IO_OUT
  52 ce062d50 2019-08-21 martijn #define IO_READ			0x04
  53 ce062d50 2019-08-21 martijn #define IO_WRITE		0x08
  54 ce062d50 2019-08-21 martijn #define IO_RW			(IO_READ | IO_WRITE)
  55 ce062d50 2019-08-21 martijn #define IO_RESET		0x10  /* internal */
  56 ce062d50 2019-08-21 martijn #define IO_HELD			0x20  /* internal */
  57 ce062d50 2019-08-21 martijn 
  58 ce062d50 2019-08-21 martijn struct io {
  59 ce062d50 2019-08-21 martijn 	int		 sock;
  60 ce062d50 2019-08-21 martijn 	void		*arg;
  61 ce062d50 2019-08-21 martijn 	void		(*cb)(struct io*, int, void *);
  62 ce062d50 2019-08-21 martijn 	struct iobuf	 iobuf;
  63 ce062d50 2019-08-21 martijn 	size_t		 lowat;
  64 ce062d50 2019-08-21 martijn 	int		 timeout;
  65 ce062d50 2019-08-21 martijn 	int		 flags;
  66 ce062d50 2019-08-21 martijn 	int		 state;
  67 ce062d50 2019-08-21 martijn 	struct event	 ev;
  68 ce062d50 2019-08-21 martijn 	void		*tls;
  69 ce062d50 2019-08-21 martijn 	const char	*error; /* only valid immediately on callback */
  70 ce062d50 2019-08-21 martijn };
  71 ce062d50 2019-08-21 martijn 
  72 ce062d50 2019-08-21 martijn const char* io_strflags(int);
  73 ce062d50 2019-08-21 martijn const char* io_evstr(short);
  74 ce062d50 2019-08-21 martijn 
  75 ce062d50 2019-08-21 martijn void	_io_init(void);
  76 ce062d50 2019-08-21 martijn void	io_hold(struct io *);
  77 ce062d50 2019-08-21 martijn void	io_release(struct io *);
  78 ce062d50 2019-08-21 martijn void	io_callback(struct io*, int);
  79 ce062d50 2019-08-21 martijn void	io_dispatch(int, short, void *);
  80 ce062d50 2019-08-21 martijn void	io_dispatch_connect(int, short, void *);
  81 ce062d50 2019-08-21 martijn size_t	io_pending(struct io *);
  82 ce062d50 2019-08-21 martijn size_t	io_queued(struct io*);
  83 ce062d50 2019-08-21 martijn void	io_reset(struct io *, short, void (*)(int, short, void*));
  84 ce062d50 2019-08-21 martijn void	io_frame_enter(const char *, struct io *, int);
  85 ce062d50 2019-08-21 martijn void	io_frame_leave(struct io *);
  86 ce062d50 2019-08-21 martijn 
  87 ce062d50 2019-08-21 martijn #ifdef IO_TLS
  88 ce062d50 2019-08-21 martijn void	ssl_error(const char *); /* XXX external */
  89 ce062d50 2019-08-21 martijn 
  90 ce062d50 2019-08-21 martijn static const char* io_tls_error(void);
  91 ce062d50 2019-08-21 martijn void	io_dispatch_accept_tls(int, short, void *);
  92 ce062d50 2019-08-21 martijn void	io_dispatch_connect_tls(int, short, void *);
  93 ce062d50 2019-08-21 martijn void	io_dispatch_read_tls(int, short, void *);
  94 ce062d50 2019-08-21 martijn void	io_dispatch_write_tls(int, short, void *);
  95 ce062d50 2019-08-21 martijn void	io_reload_tls(struct io *io);
  96 ce062d50 2019-08-21 martijn #endif
  97 ce062d50 2019-08-21 martijn 
  98 ce062d50 2019-08-21 martijn static struct io	*current = NULL;
  99 ce062d50 2019-08-21 martijn static uint64_t		 frame = 0;
 100 ce062d50 2019-08-21 martijn static int		_io_debug = 0;
 101 ce062d50 2019-08-21 martijn 
 102 ce062d50 2019-08-21 martijn #define io_debug(args...) do { if (_io_debug) printf(args); } while(0)
 103 ce062d50 2019-08-21 martijn 
 104 ce062d50 2019-08-21 martijn 
 105 ce062d50 2019-08-21 martijn const char*
 106 ce062d50 2019-08-21 martijn io_strio(struct io *io)
 107 ce062d50 2019-08-21 martijn {
 108 ce062d50 2019-08-21 martijn 	static char	buf[128];
 109 ce062d50 2019-08-21 martijn 	char		ssl[128];
 110 ce062d50 2019-08-21 martijn 
 111 ce062d50 2019-08-21 martijn 	ssl[0] = '\0';
 112 ce062d50 2019-08-21 martijn #ifdef IO_TLS
 113 ce062d50 2019-08-21 martijn 	if (io->tls) {
 114 ce062d50 2019-08-21 martijn 		(void)snprintf(ssl, sizeof ssl, " tls=%s:%s:%d",
 115 ce062d50 2019-08-21 martijn 		    SSL_get_version(io->tls),
 116 ce062d50 2019-08-21 martijn 		    SSL_get_cipher_name(io->tls),
 117 ce062d50 2019-08-21 martijn 		    SSL_get_cipher_bits(io->tls, NULL));
 118 ce062d50 2019-08-21 martijn 	}
 119 ce062d50 2019-08-21 martijn #endif
 120 ce062d50 2019-08-21 martijn 
 121 ce062d50 2019-08-21 martijn 	(void)snprintf(buf, sizeof buf,
 122 ce062d50 2019-08-21 martijn 	    "<io:%p fd=%d to=%d fl=%s%s ib=%zu ob=%zu>",
 123 ce062d50 2019-08-21 martijn 	    io, io->sock, io->timeout, io_strflags(io->flags), ssl,
 124 ce062d50 2019-08-21 martijn 	    io_pending(io), io_queued(io));
 125 ce062d50 2019-08-21 martijn 
 126 ce062d50 2019-08-21 martijn 	return (buf);
 127 ce062d50 2019-08-21 martijn }
 128 ce062d50 2019-08-21 martijn 
 129 ce062d50 2019-08-21 martijn #define CASE(x) case x : return #x
 130 ce062d50 2019-08-21 martijn 
 131 ce062d50 2019-08-21 martijn const char*
 132 ce062d50 2019-08-21 martijn io_strevent(int evt)
 133 ce062d50 2019-08-21 martijn {
 134 ce062d50 2019-08-21 martijn 	static char buf[32];
 135 ce062d50 2019-08-21 martijn 
 136 ce062d50 2019-08-21 martijn 	switch (evt) {
 137 ce062d50 2019-08-21 martijn 	CASE(IO_CONNECTED);
 138 ce062d50 2019-08-21 martijn 	CASE(IO_TLSREADY);
 139 ce062d50 2019-08-21 martijn 	CASE(IO_DATAIN);
 140 ce062d50 2019-08-21 martijn 	CASE(IO_LOWAT);
 141 ce062d50 2019-08-21 martijn 	CASE(IO_DISCONNECTED);
 142 ce062d50 2019-08-21 martijn 	CASE(IO_TIMEOUT);
 143 ce062d50 2019-08-21 martijn 	CASE(IO_ERROR);
 144 ce062d50 2019-08-21 martijn 	default:
 145 ce062d50 2019-08-21 martijn 		(void)snprintf(buf, sizeof(buf), "IO_? %d", evt);
 146 ce062d50 2019-08-21 martijn 		return buf;
 147 ce062d50 2019-08-21 martijn 	}
 148 ce062d50 2019-08-21 martijn }
 149 ce062d50 2019-08-21 martijn 
 150 ce062d50 2019-08-21 martijn void
 151 ce062d50 2019-08-21 martijn io_set_nonblocking(int fd)
 152 ce062d50 2019-08-21 martijn {
 153 ce062d50 2019-08-21 martijn 	int	flags;
 154 ce062d50 2019-08-21 martijn 
 155 ce062d50 2019-08-21 martijn 	if ((flags = fcntl(fd, F_GETFL)) == -1)
 156 ce062d50 2019-08-21 martijn 		err(1, "io_set_blocking:fcntl(F_GETFL)");
 157 ce062d50 2019-08-21 martijn 
 158 ce062d50 2019-08-21 martijn 	flags |= O_NONBLOCK;
 159 ce062d50 2019-08-21 martijn 
 160 ce062d50 2019-08-21 martijn 	if (fcntl(fd, F_SETFL, flags) == -1)
 161 ce062d50 2019-08-21 martijn 		err(1, "io_set_blocking:fcntl(F_SETFL)");
 162 ce062d50 2019-08-21 martijn }
 163 ce062d50 2019-08-21 martijn 
 164 ce062d50 2019-08-21 martijn void
 165 ce062d50 2019-08-21 martijn io_set_nolinger(int fd)
 166 ce062d50 2019-08-21 martijn {
 167 ce062d50 2019-08-21 martijn 	struct linger    l;
 168 ce062d50 2019-08-21 martijn 
 169 ce062d50 2019-08-21 martijn 	memset(&l, 0, sizeof(l));
 170 ce062d50 2019-08-21 martijn 	if (setsockopt(fd, SOL_SOCKET, SO_LINGER, &l, sizeof(l)) == -1)
 171 ce062d50 2019-08-21 martijn 		err(1, "io_set_linger:setsockopt()");
 172 ce062d50 2019-08-21 martijn }
 173 ce062d50 2019-08-21 martijn 
 174 ce062d50 2019-08-21 martijn /*
 175 ce062d50 2019-08-21 martijn  * Event framing must not rely on an io pointer to refer to the "same" io
 176 ce062d50 2019-08-21 martijn  * throughout the frame, because this is not always the case:
 177 ce062d50 2019-08-21 martijn  *
 178 ce062d50 2019-08-21 martijn  * 1) enter(addr0) -> free(addr0) -> leave(addr0) = SEGV
 179 ce062d50 2019-08-21 martijn  * 2) enter(addr0) -> free(addr0) -> malloc == addr0 -> leave(addr0) = BAD!
 180 ce062d50 2019-08-21 martijn  *
 181 ce062d50 2019-08-21 martijn  * In both case, the problem is that the io is freed in the callback, so
 182 ce062d50 2019-08-21 martijn  * the pointer becomes invalid. If that happens, the user is required to
 183 ce062d50 2019-08-21 martijn  * call io_clear, so we can adapt the frame state there.
 184 ce062d50 2019-08-21 martijn  */
 185 ce062d50 2019-08-21 martijn void
 186 ce062d50 2019-08-21 martijn io_frame_enter(const char *where, struct io *io, int ev)
 187 ce062d50 2019-08-21 martijn {
 188 ce062d50 2019-08-21 martijn 	io_debug("\n=== %" PRIu64 " ===\n"
 189 ce062d50 2019-08-21 martijn 	    "io_frame_enter(%s, %s, %s)\n",
 190 ce062d50 2019-08-21 martijn 	    frame, where, io_evstr(ev), io_strio(io));
 191 ce062d50 2019-08-21 martijn 
 192 ce062d50 2019-08-21 martijn 	if (current)
 193 ce062d50 2019-08-21 martijn 		errx(1, "io_frame_enter: interleaved frames");
 194 ce062d50 2019-08-21 martijn 
 195 ce062d50 2019-08-21 martijn 	current = io;
 196 ce062d50 2019-08-21 martijn 
 197 ce062d50 2019-08-21 martijn 	io_hold(io);
 198 ce062d50 2019-08-21 martijn }
 199 ce062d50 2019-08-21 martijn 
 200 ce062d50 2019-08-21 martijn void
 201 ce062d50 2019-08-21 martijn io_frame_leave(struct io *io)
 202 ce062d50 2019-08-21 martijn {
 203 ce062d50 2019-08-21 martijn 	io_debug("io_frame_leave(%" PRIu64 ")\n", frame);
 204 ce062d50 2019-08-21 martijn 
 205 ce062d50 2019-08-21 martijn 	if (current && current != io)
 206 ce062d50 2019-08-21 martijn 		errx(1, "io_frame_leave: io mismatch");
 207 ce062d50 2019-08-21 martijn 
 208 ce062d50 2019-08-21 martijn 	/* io has been cleared */
 209 ce062d50 2019-08-21 martijn 	if (current == NULL)
 210 ce062d50 2019-08-21 martijn 		goto done;
 211 ce062d50 2019-08-21 martijn 
 212 ce062d50 2019-08-21 martijn 	/* TODO: There is a possible optimization there:
 213 ce062d50 2019-08-21 martijn 	 * In a typical half-duplex request/response scenario,
 214 ce062d50 2019-08-21 martijn 	 * the io is waiting to read a request, and when done, it queues
 215 ce062d50 2019-08-21 martijn 	 * the response in the output buffer and goes to write mode.
 216 ce062d50 2019-08-21 martijn 	 * There, the write event is set and will be triggered in the next
 217 ce062d50 2019-08-21 martijn 	 * event frame.  In most case, the write call could be done
 218 ce062d50 2019-08-21 martijn 	 * immediately as part of the last read frame, thus avoiding to go
 219 ce062d50 2019-08-21 martijn 	 * through the event loop machinery. So, as an optimisation, we
 220 ce062d50 2019-08-21 martijn 	 * could detect that case here and force an event dispatching.
 221 ce062d50 2019-08-21 martijn 	 */
 222 ce062d50 2019-08-21 martijn 
 223 ce062d50 2019-08-21 martijn 	/* Reload the io if it has not been reset already. */
 224 ce062d50 2019-08-21 martijn 	io_release(io);
 225 ce062d50 2019-08-21 martijn 	current = NULL;
 226 ce062d50 2019-08-21 martijn     done:
 227 ce062d50 2019-08-21 martijn 	io_debug("=== /%" PRIu64 "\n", frame);
 228 ce062d50 2019-08-21 martijn 
 229 ce062d50 2019-08-21 martijn 	frame += 1;
 230 ce062d50 2019-08-21 martijn }
 231 ce062d50 2019-08-21 martijn 
 232 ce062d50 2019-08-21 martijn void
 233 ce062d50 2019-08-21 martijn _io_init()
 234 ce062d50 2019-08-21 martijn {
 235 ce062d50 2019-08-21 martijn 	static int init = 0;
 236 ce062d50 2019-08-21 martijn 
 237 ce062d50 2019-08-21 martijn 	if (init)
 238 ce062d50 2019-08-21 martijn 		return;
 239 ce062d50 2019-08-21 martijn 
 240 ce062d50 2019-08-21 martijn 	init = 1;
 241 ce062d50 2019-08-21 martijn 	_io_debug = getenv("IO_DEBUG") != NULL;
 242 ce062d50 2019-08-21 martijn }
 243 ce062d50 2019-08-21 martijn 
 244 ce062d50 2019-08-21 martijn struct io *
 245 ce062d50 2019-08-21 martijn io_new(void)
 246 ce062d50 2019-08-21 martijn {
 247 ce062d50 2019-08-21 martijn 	struct io *io;
 248 ce062d50 2019-08-21 martijn 
 249 ce062d50 2019-08-21 martijn 	_io_init();
 250 ce062d50 2019-08-21 martijn 
 251 ce062d50 2019-08-21 martijn 	if ((io = calloc(1, sizeof(*io))) == NULL)
 252 ce062d50 2019-08-21 martijn 		return NULL;
 253 ce062d50 2019-08-21 martijn 
 254 ce062d50 2019-08-21 martijn 	io->sock = -1;
 255 ce062d50 2019-08-21 martijn 	io->timeout = -1;
 256 ce062d50 2019-08-21 martijn 
 257 ce062d50 2019-08-21 martijn 	if (iobuf_init(&io->iobuf, 0, 0) == -1) {
 258 ce062d50 2019-08-21 martijn 		free(io);
 259 ce062d50 2019-08-21 martijn 		return NULL;
 260 ce062d50 2019-08-21 martijn 	}
 261 ce062d50 2019-08-21 martijn 
 262 ce062d50 2019-08-21 martijn 	return io;
 263 ce062d50 2019-08-21 martijn }
 264 ce062d50 2019-08-21 martijn 
 265 ce062d50 2019-08-21 martijn void
 266 ce062d50 2019-08-21 martijn io_free(struct io *io)
 267 ce062d50 2019-08-21 martijn {
 268 ce062d50 2019-08-21 martijn 	io_debug("io_clear(%p)\n", io);
 269 ce062d50 2019-08-21 martijn 
 270 ce062d50 2019-08-21 martijn 	/* the current io is virtually dead */
 271 ce062d50 2019-08-21 martijn 	if (io == current)
 272 ce062d50 2019-08-21 martijn 		current = NULL;
 273 ce062d50 2019-08-21 martijn 
 274 ce062d50 2019-08-21 martijn #ifdef IO_TLS
 275 ce062d50 2019-08-21 martijn 	SSL_free(io->tls);
 276 ce062d50 2019-08-21 martijn 	io->tls = NULL;
 277 ce062d50 2019-08-21 martijn #endif
 278 ce062d50 2019-08-21 martijn 
 279 ce062d50 2019-08-21 martijn 	if (event_initialized(&io->ev))
 280 ce062d50 2019-08-21 martijn 		event_del(&io->ev);
 281 ce062d50 2019-08-21 martijn 	if (io->sock != -1) {
 282 ce062d50 2019-08-21 martijn 		close(io->sock);
 283 ce062d50 2019-08-21 martijn 		io->sock = -1;
 284 ce062d50 2019-08-21 martijn 	}
 285 ce062d50 2019-08-21 martijn 
 286 ce062d50 2019-08-21 martijn 	iobuf_clear(&io->iobuf);
 287 ce062d50 2019-08-21 martijn 	free(io);
 288 ce062d50 2019-08-21 martijn }
 289 ce062d50 2019-08-21 martijn 
 290 ce062d50 2019-08-21 martijn void
 291 ce062d50 2019-08-21 martijn io_hold(struct io *io)
 292 ce062d50 2019-08-21 martijn {
 293 ce062d50 2019-08-21 martijn 	io_debug("io_enter(%p)\n", io);
 294 ce062d50 2019-08-21 martijn 
 295 ce062d50 2019-08-21 martijn 	if (io->flags & IO_HELD)
 296 ce062d50 2019-08-21 martijn 		errx(1, "io_hold: io is already held");
 297 ce062d50 2019-08-21 martijn 
 298 ce062d50 2019-08-21 martijn 	io->flags &= ~IO_RESET;
 299 ce062d50 2019-08-21 martijn 	io->flags |= IO_HELD;
 300 ce062d50 2019-08-21 martijn }
 301 ce062d50 2019-08-21 martijn 
 302 ce062d50 2019-08-21 martijn void
 303 ce062d50 2019-08-21 martijn io_release(struct io *io)
 304 ce062d50 2019-08-21 martijn {
 305 ce062d50 2019-08-21 martijn 	if (!(io->flags & IO_HELD))
 306 ce062d50 2019-08-21 martijn 		errx(1, "io_release: io is not held");
 307 ce062d50 2019-08-21 martijn 
 308 ce062d50 2019-08-21 martijn 	io->flags &= ~IO_HELD;
 309 ce062d50 2019-08-21 martijn 	if (!(io->flags & IO_RESET))
 310 ce062d50 2019-08-21 martijn 		io_reload(io);
 311 ce062d50 2019-08-21 martijn }
 312 ce062d50 2019-08-21 martijn 
 313 ce062d50 2019-08-21 martijn void
 314 ce062d50 2019-08-21 martijn io_set_fd(struct io *io, int fd)
 315 ce062d50 2019-08-21 martijn {
 316 ce062d50 2019-08-21 martijn 	io->sock = fd;
 317 ce062d50 2019-08-21 martijn 	if (fd != -1)
 318 ce062d50 2019-08-21 martijn 		io_reload(io);
 319 ce062d50 2019-08-21 martijn }
 320 ce062d50 2019-08-21 martijn 
 321 ce062d50 2019-08-21 martijn void
 322 ce062d50 2019-08-21 martijn io_set_callback(struct io *io, void(*cb)(struct io *, int, void *), void *arg)
 323 ce062d50 2019-08-21 martijn {
 324 ce062d50 2019-08-21 martijn 	io->cb = cb;
 325 ce062d50 2019-08-21 martijn 	io->arg = arg;
 326 ce062d50 2019-08-21 martijn }
 327 ce062d50 2019-08-21 martijn 
 328 ce062d50 2019-08-21 martijn void
 329 ce062d50 2019-08-21 martijn io_set_timeout(struct io *io, int msec)
 330 ce062d50 2019-08-21 martijn {
 331 ce062d50 2019-08-21 martijn 	io_debug("io_set_timeout(%p, %d)\n", io, msec);
 332 ce062d50 2019-08-21 martijn 
 333 ce062d50 2019-08-21 martijn 	io->timeout = msec;
 334 ce062d50 2019-08-21 martijn }
 335 ce062d50 2019-08-21 martijn 
 336 ce062d50 2019-08-21 martijn void
 337 ce062d50 2019-08-21 martijn io_set_lowat(struct io *io, size_t lowat)
 338 ce062d50 2019-08-21 martijn {
 339 ce062d50 2019-08-21 martijn 	io_debug("io_set_lowat(%p, %zu)\n", io, lowat);
 340 ce062d50 2019-08-21 martijn 
 341 ce062d50 2019-08-21 martijn 	io->lowat = lowat;
 342 ce062d50 2019-08-21 martijn }
 343 ce062d50 2019-08-21 martijn 
 344 ce062d50 2019-08-21 martijn void
 345 ce062d50 2019-08-21 martijn io_pause(struct io *io, int dir)
 346 ce062d50 2019-08-21 martijn {
 347 ce062d50 2019-08-21 martijn 	io_debug("io_pause(%p, %x)\n", io, dir);
 348 ce062d50 2019-08-21 martijn 
 349 ce062d50 2019-08-21 martijn 	io->flags |= dir & (IO_PAUSE_IN | IO_PAUSE_OUT);
 350 ce062d50 2019-08-21 martijn 	io_reload(io);
 351 ce062d50 2019-08-21 martijn }
 352 ce062d50 2019-08-21 martijn 
 353 ce062d50 2019-08-21 martijn void
 354 ce062d50 2019-08-21 martijn io_resume(struct io *io, int dir)
 355 ce062d50 2019-08-21 martijn {
 356 ce062d50 2019-08-21 martijn 	io_debug("io_resume(%p, %x)\n", io, dir);
 357 ce062d50 2019-08-21 martijn 
 358 ce062d50 2019-08-21 martijn 	io->flags &= ~(dir & (IO_PAUSE_IN | IO_PAUSE_OUT));
 359 ce062d50 2019-08-21 martijn 	io_reload(io);
 360 ce062d50 2019-08-21 martijn }
 361 ce062d50 2019-08-21 martijn 
 362 ce062d50 2019-08-21 martijn void
 363 ce062d50 2019-08-21 martijn io_set_read(struct io *io)
 364 ce062d50 2019-08-21 martijn {
 365 ce062d50 2019-08-21 martijn 	int	mode;
 366 ce062d50 2019-08-21 martijn 
 367 ce062d50 2019-08-21 martijn 	io_debug("io_set_read(%p)\n", io);
 368 ce062d50 2019-08-21 martijn 
 369 ce062d50 2019-08-21 martijn 	mode = io->flags & IO_RW;
 370 ce062d50 2019-08-21 martijn 	if (!(mode == 0 || mode == IO_WRITE))
 371 ce062d50 2019-08-21 martijn 		errx(1, "io_set_read(): full-duplex or reading");
 372 ce062d50 2019-08-21 martijn 
 373 ce062d50 2019-08-21 martijn 	io->flags &= ~IO_RW;
 374 ce062d50 2019-08-21 martijn 	io->flags |= IO_READ;
 375 ce062d50 2019-08-21 martijn 	io_reload(io);
 376 ce062d50 2019-08-21 martijn }
 377 ce062d50 2019-08-21 martijn 
 378 ce062d50 2019-08-21 martijn void
 379 ce062d50 2019-08-21 martijn io_set_write(struct io *io)
 380 ce062d50 2019-08-21 martijn {
 381 ce062d50 2019-08-21 martijn 	int	mode;
 382 ce062d50 2019-08-21 martijn 
 383 ce062d50 2019-08-21 martijn 	io_debug("io_set_write(%p)\n", io);
 384 ce062d50 2019-08-21 martijn 
 385 ce062d50 2019-08-21 martijn 	mode = io->flags & IO_RW;
 386 ce062d50 2019-08-21 martijn 	if (!(mode == 0 || mode == IO_READ))
 387 ce062d50 2019-08-21 martijn 		errx(1, "io_set_write(): full-duplex or writing");
 388 ce062d50 2019-08-21 martijn 
 389 ce062d50 2019-08-21 martijn 	io->flags &= ~IO_RW;
 390 ce062d50 2019-08-21 martijn 	io->flags |= IO_WRITE;
 391 ce062d50 2019-08-21 martijn 	io_reload(io);
 392 ce062d50 2019-08-21 martijn }
 393 ce062d50 2019-08-21 martijn 
 394 ce062d50 2019-08-21 martijn const char *
 395 ce062d50 2019-08-21 martijn io_error(struct io *io)
 396 ce062d50 2019-08-21 martijn {
 397 ce062d50 2019-08-21 martijn 	return io->error;
 398 ce062d50 2019-08-21 martijn }
 399 ce062d50 2019-08-21 martijn 
 400 ce062d50 2019-08-21 martijn void *
 401 ce062d50 2019-08-21 martijn io_tls(struct io *io)
 402 ce062d50 2019-08-21 martijn {
 403 ce062d50 2019-08-21 martijn 	return io->tls;
 404 ce062d50 2019-08-21 martijn }
 405 ce062d50 2019-08-21 martijn 
 406 ce062d50 2019-08-21 martijn int
 407 ce062d50 2019-08-21 martijn io_fileno(struct io *io)
 408 ce062d50 2019-08-21 martijn {
 409 ce062d50 2019-08-21 martijn 	return io->sock;
 410 ce062d50 2019-08-21 martijn }
 411 ce062d50 2019-08-21 martijn 
 412 ce062d50 2019-08-21 martijn int
 413 ce062d50 2019-08-21 martijn io_paused(struct io *io, int what)
 414 ce062d50 2019-08-21 martijn {
 415 ce062d50 2019-08-21 martijn 	return (io->flags & (IO_PAUSE_IN | IO_PAUSE_OUT)) == what;
 416 ce062d50 2019-08-21 martijn }
 417 ce062d50 2019-08-21 martijn 
 418 ce062d50 2019-08-21 martijn /*
 419 ce062d50 2019-08-21 martijn  * Buffered output functions
 420 ce062d50 2019-08-21 martijn  */
 421 ce062d50 2019-08-21 martijn 
 422 ce062d50 2019-08-21 martijn int
 423 ce062d50 2019-08-21 martijn io_write(struct io *io, const void *buf, size_t len)
 424 ce062d50 2019-08-21 martijn {
 425 ce062d50 2019-08-21 martijn 	int r;
 426 ce062d50 2019-08-21 martijn 
 427 ce062d50 2019-08-21 martijn 	r = iobuf_queue(&io->iobuf, buf, len);
 428 ce062d50 2019-08-21 martijn 
 429 ce062d50 2019-08-21 martijn 	io_reload(io);
 430 ce062d50 2019-08-21 martijn 
 431 ce062d50 2019-08-21 martijn 	return r;
 432 ce062d50 2019-08-21 martijn }
 433 ce062d50 2019-08-21 martijn 
 434 ce062d50 2019-08-21 martijn int
 435 ce062d50 2019-08-21 martijn io_writev(struct io *io, const struct iovec *iov, int iovcount)
 436 ce062d50 2019-08-21 martijn {
 437 ce062d50 2019-08-21 martijn 	int r;
 438 ce062d50 2019-08-21 martijn 
 439 ce062d50 2019-08-21 martijn 	r = iobuf_queuev(&io->iobuf, iov, iovcount);
 440 ce062d50 2019-08-21 martijn 
 441 ce062d50 2019-08-21 martijn 	io_reload(io);
 442 ce062d50 2019-08-21 martijn 
 443 ce062d50 2019-08-21 martijn 	return r;
 444 ce062d50 2019-08-21 martijn }
 445 ce062d50 2019-08-21 martijn 
 446 ce062d50 2019-08-21 martijn int
 447 ce062d50 2019-08-21 martijn io_print(struct io *io, const char *s)
 448 ce062d50 2019-08-21 martijn {
 449 ce062d50 2019-08-21 martijn 	return io_write(io, s, strlen(s));
 450 ce062d50 2019-08-21 martijn }
 451 ce062d50 2019-08-21 martijn 
 452 ce062d50 2019-08-21 martijn int
 453 ce062d50 2019-08-21 martijn io_printf(struct io *io, const char *fmt, ...)
 454 ce062d50 2019-08-21 martijn {
 455 ce062d50 2019-08-21 martijn 	va_list ap;
 456 ce062d50 2019-08-21 martijn 	int r;
 457 ce062d50 2019-08-21 martijn 
 458 ce062d50 2019-08-21 martijn 	va_start(ap, fmt);
 459 ce062d50 2019-08-21 martijn 	r = io_vprintf(io, fmt, ap);
 460 ce062d50 2019-08-21 martijn 	va_end(ap);
 461 ce062d50 2019-08-21 martijn 
 462 ce062d50 2019-08-21 martijn 	return r;
 463 ce062d50 2019-08-21 martijn }
 464 ce062d50 2019-08-21 martijn 
 465 ce062d50 2019-08-21 martijn int
 466 ce062d50 2019-08-21 martijn io_vprintf(struct io *io, const char *fmt, va_list ap)
 467 ce062d50 2019-08-21 martijn {
 468 ce062d50 2019-08-21 martijn 
 469 ce062d50 2019-08-21 martijn 	char *buf;
 470 ce062d50 2019-08-21 martijn 	int len;
 471 ce062d50 2019-08-21 martijn 
 472 ce062d50 2019-08-21 martijn 	len = vasprintf(&buf, fmt, ap);
 473 ce062d50 2019-08-21 martijn 	if (len == -1)
 474 ce062d50 2019-08-21 martijn 		return -1;
 475 ce062d50 2019-08-21 martijn 	len = io_write(io, buf, len);
 476 ce062d50 2019-08-21 martijn 	free(buf);
 477 ce062d50 2019-08-21 martijn 
 478 ce062d50 2019-08-21 martijn 	return len;
 479 ce062d50 2019-08-21 martijn }
 480 ce062d50 2019-08-21 martijn 
 481 ce062d50 2019-08-21 martijn size_t
 482 ce062d50 2019-08-21 martijn io_queued(struct io *io)
 483 ce062d50 2019-08-21 martijn {
 484 ce062d50 2019-08-21 martijn 	return iobuf_queued(&io->iobuf);
 485 ce062d50 2019-08-21 martijn }
 486 ce062d50 2019-08-21 martijn 
 487 ce062d50 2019-08-21 martijn /*
 488 ce062d50 2019-08-21 martijn  * Buffered input functions
 489 ce062d50 2019-08-21 martijn  */
 490 ce062d50 2019-08-21 martijn 
 491 ce062d50 2019-08-21 martijn void *
 492 ce062d50 2019-08-21 martijn io_data(struct io *io)
 493 ce062d50 2019-08-21 martijn {
 494 ce062d50 2019-08-21 martijn 	return iobuf_data(&io->iobuf);
 495 ce062d50 2019-08-21 martijn }
 496 ce062d50 2019-08-21 martijn 
 497 ce062d50 2019-08-21 martijn size_t
 498 ce062d50 2019-08-21 martijn io_datalen(struct io *io)
 499 ce062d50 2019-08-21 martijn {
 500 ce062d50 2019-08-21 martijn 	return iobuf_len(&io->iobuf);
 501 ce062d50 2019-08-21 martijn }
 502 ce062d50 2019-08-21 martijn 
 503 ce062d50 2019-08-21 martijn char *
 504 ce062d50 2019-08-21 martijn io_getline(struct io *io, size_t *sz)
 505 ce062d50 2019-08-21 martijn {
 506 ce062d50 2019-08-21 martijn 	return iobuf_getline(&io->iobuf, sz);
 507 ce062d50 2019-08-21 martijn }
 508 ce062d50 2019-08-21 martijn 
 509 ce062d50 2019-08-21 martijn void
 510 ce062d50 2019-08-21 martijn io_drop(struct io *io, size_t sz)
 511 ce062d50 2019-08-21 martijn {
 512 ce062d50 2019-08-21 martijn 	return iobuf_drop(&io->iobuf, sz);
 513 ce062d50 2019-08-21 martijn }
 514 ce062d50 2019-08-21 martijn 
 515 ce062d50 2019-08-21 martijn 
 516 ce062d50 2019-08-21 martijn #define IO_READING(io) (((io)->flags & IO_RW) != IO_WRITE)
 517 ce062d50 2019-08-21 martijn #define IO_WRITING(io) (((io)->flags & IO_RW) != IO_READ)
 518 ce062d50 2019-08-21 martijn 
 519 ce062d50 2019-08-21 martijn /*
 520 ce062d50 2019-08-21 martijn  * Setup the necessary events as required by the current io state,
 521 ce062d50 2019-08-21 martijn  * honouring duplex mode and i/o pauses.
 522 ce062d50 2019-08-21 martijn  */
 523 ce062d50 2019-08-21 martijn void
 524 ce062d50 2019-08-21 martijn io_reload(struct io *io)
 525 ce062d50 2019-08-21 martijn {
 526 ce062d50 2019-08-21 martijn 	short	events;
 527 ce062d50 2019-08-21 martijn 
 528 ce062d50 2019-08-21 martijn 	/* io will be reloaded at release time */
 529 ce062d50 2019-08-21 martijn 	if (io->flags & IO_HELD)
 530 ce062d50 2019-08-21 martijn 		return;
 531 ce062d50 2019-08-21 martijn 
 532 ce062d50 2019-08-21 martijn 	iobuf_normalize(&io->iobuf);
 533 ce062d50 2019-08-21 martijn 
 534 ce062d50 2019-08-21 martijn #ifdef IO_TLS
 535 ce062d50 2019-08-21 martijn 	if (io->tls) {
 536 ce062d50 2019-08-21 martijn 		io_reload_tls(io);
 537 ce062d50 2019-08-21 martijn 		return;
 538 ce062d50 2019-08-21 martijn 	}
 539 ce062d50 2019-08-21 martijn #endif
 540 ce062d50 2019-08-21 martijn 
 541 ce062d50 2019-08-21 martijn 	io_debug("io_reload(%p)\n", io);
 542 ce062d50 2019-08-21 martijn 
 543 ce062d50 2019-08-21 martijn 	events = 0;
 544 ce062d50 2019-08-21 martijn 	if (IO_READING(io) && !(io->flags & IO_PAUSE_IN))
 545 ce062d50 2019-08-21 martijn 		events = EV_READ;
 546 ce062d50 2019-08-21 martijn 	if (IO_WRITING(io) && !(io->flags & IO_PAUSE_OUT) && io_queued(io))
 547 ce062d50 2019-08-21 martijn 		events |= EV_WRITE;
 548 ce062d50 2019-08-21 martijn 
 549 ce062d50 2019-08-21 martijn 	io_reset(io, events, io_dispatch);
 550 ce062d50 2019-08-21 martijn }
 551 ce062d50 2019-08-21 martijn 
 552 ce062d50 2019-08-21 martijn /* Set the requested event. */
 553 ce062d50 2019-08-21 martijn void
 554 ce062d50 2019-08-21 martijn io_reset(struct io *io, short events, void (*dispatch)(int, short, void*))
 555 ce062d50 2019-08-21 martijn {
 556 ce062d50 2019-08-21 martijn 	struct timeval	tv, *ptv;
 557 ce062d50 2019-08-21 martijn 
 558 ce062d50 2019-08-21 martijn 	io_debug("io_reset(%p, %s, %p) -> %s\n",
 559 ce062d50 2019-08-21 martijn 	    io, io_evstr(events), dispatch, io_strio(io));
 560 ce062d50 2019-08-21 martijn 
 561 ce062d50 2019-08-21 martijn 	/*
 562 ce062d50 2019-08-21 martijn 	 * Indicate that the event has already been reset so that reload
 563 ce062d50 2019-08-21 martijn 	 * is not called on frame_leave.
 564 ce062d50 2019-08-21 martijn 	 */
 565 ce062d50 2019-08-21 martijn 	io->flags |= IO_RESET;
 566 ce062d50 2019-08-21 martijn 
 567 ce062d50 2019-08-21 martijn 	if (event_initialized(&io->ev))
 568 ce062d50 2019-08-21 martijn 		event_del(&io->ev);
 569 ce062d50 2019-08-21 martijn 
 570 ce062d50 2019-08-21 martijn 	/*
 571 ce062d50 2019-08-21 martijn 	 * The io is paused by the user, so we don't want the timeout to be
 572 ce062d50 2019-08-21 martijn 	 * effective.
 573 ce062d50 2019-08-21 martijn 	 */
 574 ce062d50 2019-08-21 martijn 	if (events == 0)
 575 ce062d50 2019-08-21 martijn 		return;
 576 ce062d50 2019-08-21 martijn 
 577 ce062d50 2019-08-21 martijn 	event_set(&io->ev, io->sock, events, dispatch, io);
 578 ce062d50 2019-08-21 martijn 	if (io->timeout >= 0) {
 579 ce062d50 2019-08-21 martijn 		tv.tv_sec = io->timeout / 1000;
 580 ce062d50 2019-08-21 martijn 		tv.tv_usec = (io->timeout % 1000) * 1000;
 581 ce062d50 2019-08-21 martijn 		ptv = &tv;
 582 ce062d50 2019-08-21 martijn 	} else
 583 ce062d50 2019-08-21 martijn 		ptv = NULL;
 584 ce062d50 2019-08-21 martijn 
 585 ce062d50 2019-08-21 martijn 	event_add(&io->ev, ptv);
 586 ce062d50 2019-08-21 martijn }
 587 ce062d50 2019-08-21 martijn 
 588 ce062d50 2019-08-21 martijn size_t
 589 ce062d50 2019-08-21 martijn io_pending(struct io *io)
 590 ce062d50 2019-08-21 martijn {
 591 ce062d50 2019-08-21 martijn 	return iobuf_len(&io->iobuf);
 592 ce062d50 2019-08-21 martijn }
 593 ce062d50 2019-08-21 martijn 
 594 ce062d50 2019-08-21 martijn const char*
 595 ce062d50 2019-08-21 martijn io_strflags(int flags)
 596 ce062d50 2019-08-21 martijn {
 597 ce062d50 2019-08-21 martijn 	static char	buf[64];
 598 ce062d50 2019-08-21 martijn 
 599 ce062d50 2019-08-21 martijn 	buf[0] = '\0';
 600 ce062d50 2019-08-21 martijn 
 601 ce062d50 2019-08-21 martijn 	switch (flags & IO_RW) {
 602 ce062d50 2019-08-21 martijn 	case 0:
 603 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "rw", sizeof buf);
 604 ce062d50 2019-08-21 martijn 		break;
 605 ce062d50 2019-08-21 martijn 	case IO_READ:
 606 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "R", sizeof buf);
 607 ce062d50 2019-08-21 martijn 		break;
 608 ce062d50 2019-08-21 martijn 	case IO_WRITE:
 609 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "W", sizeof buf);
 610 ce062d50 2019-08-21 martijn 		break;
 611 ce062d50 2019-08-21 martijn 	case IO_RW:
 612 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "RW", sizeof buf);
 613 ce062d50 2019-08-21 martijn 		break;
 614 ce062d50 2019-08-21 martijn 	}
 615 ce062d50 2019-08-21 martijn 
 616 ce062d50 2019-08-21 martijn 	if (flags & IO_PAUSE_IN)
 617 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, ",F_PI", sizeof buf);
 618 ce062d50 2019-08-21 martijn 	if (flags & IO_PAUSE_OUT)
 619 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, ",F_PO", sizeof buf);
 620 ce062d50 2019-08-21 martijn 
 621 ce062d50 2019-08-21 martijn 	return buf;
 622 ce062d50 2019-08-21 martijn }
 623 ce062d50 2019-08-21 martijn 
 624 ce062d50 2019-08-21 martijn const char*
 625 ce062d50 2019-08-21 martijn io_evstr(short ev)
 626 ce062d50 2019-08-21 martijn {
 627 ce062d50 2019-08-21 martijn 	static char	buf[64];
 628 ce062d50 2019-08-21 martijn 	char		buf2[16];
 629 ce062d50 2019-08-21 martijn 	int		n;
 630 ce062d50 2019-08-21 martijn 
 631 ce062d50 2019-08-21 martijn 	n = 0;
 632 ce062d50 2019-08-21 martijn 	buf[0] = '\0';
 633 ce062d50 2019-08-21 martijn 
 634 ce062d50 2019-08-21 martijn 	if (ev == 0) {
 635 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "<NONE>", sizeof(buf));
 636 ce062d50 2019-08-21 martijn 		return buf;
 637 ce062d50 2019-08-21 martijn 	}
 638 ce062d50 2019-08-21 martijn 
 639 ce062d50 2019-08-21 martijn 	if (ev & EV_TIMEOUT) {
 640 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "EV_TIMEOUT", sizeof(buf));
 641 ce062d50 2019-08-21 martijn 		ev &= ~EV_TIMEOUT;
 642 ce062d50 2019-08-21 martijn 		n++;
 643 ce062d50 2019-08-21 martijn 	}
 644 ce062d50 2019-08-21 martijn 
 645 ce062d50 2019-08-21 martijn 	if (ev & EV_READ) {
 646 ce062d50 2019-08-21 martijn 		if (n)
 647 ce062d50 2019-08-21 martijn 			(void)strlcat(buf, "|", sizeof(buf));
 648 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "EV_READ", sizeof(buf));
 649 ce062d50 2019-08-21 martijn 		ev &= ~EV_READ;
 650 ce062d50 2019-08-21 martijn 		n++;
 651 ce062d50 2019-08-21 martijn 	}
 652 ce062d50 2019-08-21 martijn 
 653 ce062d50 2019-08-21 martijn 	if (ev & EV_WRITE) {
 654 ce062d50 2019-08-21 martijn 		if (n)
 655 ce062d50 2019-08-21 martijn 			(void)strlcat(buf, "|", sizeof(buf));
 656 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "EV_WRITE", sizeof(buf));
 657 ce062d50 2019-08-21 martijn 		ev &= ~EV_WRITE;
 658 ce062d50 2019-08-21 martijn 		n++;
 659 ce062d50 2019-08-21 martijn 	}
 660 ce062d50 2019-08-21 martijn 
 661 ce062d50 2019-08-21 martijn 	if (ev & EV_SIGNAL) {
 662 ce062d50 2019-08-21 martijn 		if (n)
 663 ce062d50 2019-08-21 martijn 			(void)strlcat(buf, "|", sizeof(buf));
 664 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "EV_SIGNAL", sizeof(buf));
 665 ce062d50 2019-08-21 martijn 		ev &= ~EV_SIGNAL;
 666 ce062d50 2019-08-21 martijn 		n++;
 667 ce062d50 2019-08-21 martijn 	}
 668 ce062d50 2019-08-21 martijn 
 669 ce062d50 2019-08-21 martijn 	if (ev) {
 670 ce062d50 2019-08-21 martijn 		if (n)
 671 ce062d50 2019-08-21 martijn 			(void)strlcat(buf, "|", sizeof(buf));
 672 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, "EV_?=0x", sizeof(buf));
 673 ce062d50 2019-08-21 martijn 		(void)snprintf(buf2, sizeof(buf2), "%hx", ev);
 674 ce062d50 2019-08-21 martijn 		(void)strlcat(buf, buf2, sizeof(buf));
 675 ce062d50 2019-08-21 martijn 	}
 676 ce062d50 2019-08-21 martijn 
 677 ce062d50 2019-08-21 martijn 	return buf;
 678 ce062d50 2019-08-21 martijn }
 679 ce062d50 2019-08-21 martijn 
 680 ce062d50 2019-08-21 martijn void
 681 ce062d50 2019-08-21 martijn io_dispatch(int fd, short ev, void *humppa)
 682 ce062d50 2019-08-21 martijn {
 683 ce062d50 2019-08-21 martijn 	struct io	*io = humppa;
 684 ce062d50 2019-08-21 martijn 	size_t		 w;
 685 ce062d50 2019-08-21 martijn 	ssize_t		 n;
 686 ce062d50 2019-08-21 martijn 	int		 saved_errno;
 687 ce062d50 2019-08-21 martijn 
 688 ce062d50 2019-08-21 martijn 	io_frame_enter("io_dispatch", io, ev);
 689 ce062d50 2019-08-21 martijn 
 690 ce062d50 2019-08-21 martijn 	if (ev == EV_TIMEOUT) {
 691 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TIMEOUT);
 692 ce062d50 2019-08-21 martijn 		goto leave;
 693 ce062d50 2019-08-21 martijn 	}
 694 ce062d50 2019-08-21 martijn 
 695 ce062d50 2019-08-21 martijn 	if (ev & EV_WRITE && (w = io_queued(io))) {
 696 ce062d50 2019-08-21 martijn 		if ((n = iobuf_write(&io->iobuf, io->sock)) < 0) {
 697 ce062d50 2019-08-21 martijn 			if (n == IOBUF_WANT_WRITE) /* kqueue bug? */
 698 ce062d50 2019-08-21 martijn 				goto read;
 699 ce062d50 2019-08-21 martijn 			if (n == IOBUF_CLOSED)
 700 ce062d50 2019-08-21 martijn 				io_callback(io, IO_DISCONNECTED);
 701 ce062d50 2019-08-21 martijn 			else {
 702 ce062d50 2019-08-21 martijn 				saved_errno = errno;
 703 ce062d50 2019-08-21 martijn 				io->error = strerror(errno);
 704 ce062d50 2019-08-21 martijn 				errno = saved_errno;
 705 ce062d50 2019-08-21 martijn 				io_callback(io, IO_ERROR);
 706 ce062d50 2019-08-21 martijn 			}
 707 ce062d50 2019-08-21 martijn 			goto leave;
 708 ce062d50 2019-08-21 martijn 		}
 709 ce062d50 2019-08-21 martijn 		if (w > io->lowat && w - n <= io->lowat)
 710 ce062d50 2019-08-21 martijn 			io_callback(io, IO_LOWAT);
 711 ce062d50 2019-08-21 martijn 	}
 712 ce062d50 2019-08-21 martijn     read:
 713 ce062d50 2019-08-21 martijn 
 714 ce062d50 2019-08-21 martijn 	if (ev & EV_READ) {
 715 ce062d50 2019-08-21 martijn 		iobuf_normalize(&io->iobuf);
 716 ce062d50 2019-08-21 martijn 		if ((n = iobuf_read(&io->iobuf, io->sock)) < 0) {
 717 ce062d50 2019-08-21 martijn 			if (n == IOBUF_CLOSED)
 718 ce062d50 2019-08-21 martijn 				io_callback(io, IO_DISCONNECTED);
 719 ce062d50 2019-08-21 martijn 			else {
 720 ce062d50 2019-08-21 martijn 				saved_errno = errno;
 721 ce062d50 2019-08-21 martijn 				io->error = strerror(errno);
 722 ce062d50 2019-08-21 martijn 				errno = saved_errno;
 723 ce062d50 2019-08-21 martijn 				io_callback(io, IO_ERROR);
 724 ce062d50 2019-08-21 martijn 			}
 725 ce062d50 2019-08-21 martijn 			goto leave;
 726 ce062d50 2019-08-21 martijn 		}
 727 ce062d50 2019-08-21 martijn 		if (n)
 728 ce062d50 2019-08-21 martijn 			io_callback(io, IO_DATAIN);
 729 ce062d50 2019-08-21 martijn 	}
 730 ce062d50 2019-08-21 martijn 
 731 ce062d50 2019-08-21 martijn leave:
 732 ce062d50 2019-08-21 martijn 	io_frame_leave(io);
 733 ce062d50 2019-08-21 martijn }
 734 ce062d50 2019-08-21 martijn 
 735 ce062d50 2019-08-21 martijn void
 736 ce062d50 2019-08-21 martijn io_callback(struct io *io, int evt)
 737 ce062d50 2019-08-21 martijn {
 738 ce062d50 2019-08-21 martijn 	io->cb(io, evt, io->arg);
 739 ce062d50 2019-08-21 martijn }
 740 ce062d50 2019-08-21 martijn 
 741 ce062d50 2019-08-21 martijn int
 742 ce062d50 2019-08-21 martijn io_connect(struct io *io, const struct sockaddr *sa, const struct sockaddr *bsa)
 743 ce062d50 2019-08-21 martijn {
 744 ce062d50 2019-08-21 martijn 	int	sock, errno_save;
 745 ce062d50 2019-08-21 martijn 
 746 ce062d50 2019-08-21 martijn 	if ((sock = socket(sa->sa_family, SOCK_STREAM, 0)) == -1)
 747 ce062d50 2019-08-21 martijn 		goto fail;
 748 ce062d50 2019-08-21 martijn 
 749 ce062d50 2019-08-21 martijn 	io_set_nonblocking(sock);
 750 ce062d50 2019-08-21 martijn 	io_set_nolinger(sock);
 751 ce062d50 2019-08-21 martijn 
 752 ce062d50 2019-08-21 martijn 	if (bsa && bind(sock, bsa, bsa->sa_len) == -1)
 753 ce062d50 2019-08-21 martijn 		goto fail;
 754 ce062d50 2019-08-21 martijn 
 755 ce062d50 2019-08-21 martijn 	if (connect(sock, sa, sa->sa_len) == -1)
 756 ce062d50 2019-08-21 martijn 		if (errno != EINPROGRESS)
 757 ce062d50 2019-08-21 martijn 			goto fail;
 758 ce062d50 2019-08-21 martijn 
 759 ce062d50 2019-08-21 martijn 	io->sock = sock;
 760 ce062d50 2019-08-21 martijn 	io_reset(io, EV_WRITE, io_dispatch_connect);
 761 ce062d50 2019-08-21 martijn 
 762 ce062d50 2019-08-21 martijn 	return (sock);
 763 ce062d50 2019-08-21 martijn 
 764 ce062d50 2019-08-21 martijn     fail:
 765 ce062d50 2019-08-21 martijn 	if (sock != -1) {
 766 ce062d50 2019-08-21 martijn 		errno_save = errno;
 767 ce062d50 2019-08-21 martijn 		close(sock);
 768 ce062d50 2019-08-21 martijn 		errno = errno_save;
 769 ce062d50 2019-08-21 martijn 		io->error = strerror(errno);
 770 ce062d50 2019-08-21 martijn 	}
 771 ce062d50 2019-08-21 martijn 	return (-1);
 772 ce062d50 2019-08-21 martijn }
 773 ce062d50 2019-08-21 martijn 
 774 ce062d50 2019-08-21 martijn void
 775 ce062d50 2019-08-21 martijn io_dispatch_connect(int fd, short ev, void *humppa)
 776 ce062d50 2019-08-21 martijn {
 777 ce062d50 2019-08-21 martijn 	struct io	*io = humppa;
 778 ce062d50 2019-08-21 martijn 	int		 r, e;
 779 ce062d50 2019-08-21 martijn 	socklen_t	 sl;
 780 ce062d50 2019-08-21 martijn 
 781 ce062d50 2019-08-21 martijn 	io_frame_enter("io_dispatch_connect", io, ev);
 782 ce062d50 2019-08-21 martijn 
 783 ce062d50 2019-08-21 martijn 	if (ev == EV_TIMEOUT) {
 784 ce062d50 2019-08-21 martijn 		close(fd);
 785 ce062d50 2019-08-21 martijn 		io->sock = -1;
 786 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TIMEOUT);
 787 ce062d50 2019-08-21 martijn 	} else {
 788 ce062d50 2019-08-21 martijn 		sl = sizeof(e);
 789 ce062d50 2019-08-21 martijn 		r = getsockopt(fd, SOL_SOCKET, SO_ERROR, &e, &sl);
 790 ce062d50 2019-08-21 martijn 		if (r == -1)  {
 791 ce062d50 2019-08-21 martijn 			warn("io_dispatch_connect: getsockopt");
 792 ce062d50 2019-08-21 martijn 			e = errno;
 793 ce062d50 2019-08-21 martijn 		}
 794 ce062d50 2019-08-21 martijn 		if (e) {
 795 ce062d50 2019-08-21 martijn 			close(fd);
 796 ce062d50 2019-08-21 martijn 			io->sock = -1;
 797 ce062d50 2019-08-21 martijn 			io->error = strerror(e);
 798 ce062d50 2019-08-21 martijn 			io_callback(io, e == ETIMEDOUT ? IO_TIMEOUT : IO_ERROR);
 799 ce062d50 2019-08-21 martijn 		}
 800 ce062d50 2019-08-21 martijn 		else {
 801 ce062d50 2019-08-21 martijn 			io->state = IO_STATE_UP;
 802 ce062d50 2019-08-21 martijn 			io_callback(io, IO_CONNECTED);
 803 ce062d50 2019-08-21 martijn 		}
 804 ce062d50 2019-08-21 martijn 	}
 805 ce062d50 2019-08-21 martijn 
 806 ce062d50 2019-08-21 martijn 	io_frame_leave(io);
 807 ce062d50 2019-08-21 martijn }
 808 ce062d50 2019-08-21 martijn 
 809 ce062d50 2019-08-21 martijn #ifdef IO_TLS
 810 ce062d50 2019-08-21 martijn 
 811 ce062d50 2019-08-21 martijn static const char*
 812 ce062d50 2019-08-21 martijn io_tls_error(void)
 813 ce062d50 2019-08-21 martijn {
 814 ce062d50 2019-08-21 martijn 	static char	buf[128];
 815 ce062d50 2019-08-21 martijn 	unsigned long	e;
 816 ce062d50 2019-08-21 martijn 
 817 ce062d50 2019-08-21 martijn 	e = ERR_peek_last_error();
 818 ce062d50 2019-08-21 martijn 	if (e) {
 819 ce062d50 2019-08-21 martijn 		ERR_error_string(e, buf);
 820 ce062d50 2019-08-21 martijn 		return (buf);
 821 ce062d50 2019-08-21 martijn 	}
 822 ce062d50 2019-08-21 martijn 
 823 ce062d50 2019-08-21 martijn 	return ("No TLS error");
 824 ce062d50 2019-08-21 martijn }
 825 ce062d50 2019-08-21 martijn 
 826 ce062d50 2019-08-21 martijn int
 827 ce062d50 2019-08-21 martijn io_start_tls(struct io *io, void *tls)
 828 ce062d50 2019-08-21 martijn {
 829 ce062d50 2019-08-21 martijn 	int	mode;
 830 ce062d50 2019-08-21 martijn 
 831 ce062d50 2019-08-21 martijn 	mode = io->flags & IO_RW;
 832 ce062d50 2019-08-21 martijn 	if (mode == 0 || mode == IO_RW)
 833 ce062d50 2019-08-21 martijn 		errx(1, "io_start_tls(): full-duplex or unset");
 834 ce062d50 2019-08-21 martijn 
 835 ce062d50 2019-08-21 martijn 	if (io->tls)
 836 ce062d50 2019-08-21 martijn 		errx(1, "io_start_tls(): TLS already started");
 837 ce062d50 2019-08-21 martijn 	io->tls = tls;
 838 ce062d50 2019-08-21 martijn 
 839 ce062d50 2019-08-21 martijn 	if (SSL_set_fd(io->tls, io->sock) == 0) {
 840 ce062d50 2019-08-21 martijn 		ssl_error("io_start_tls:SSL_set_fd");
 841 ce062d50 2019-08-21 martijn 		return (-1);
 842 ce062d50 2019-08-21 martijn 	}
 843 ce062d50 2019-08-21 martijn 
 844 ce062d50 2019-08-21 martijn 	if (mode == IO_WRITE) {
 845 ce062d50 2019-08-21 martijn 		io->state = IO_STATE_CONNECT_TLS;
 846 ce062d50 2019-08-21 martijn 		SSL_set_connect_state(io->tls);
 847 ce062d50 2019-08-21 martijn 		io_reset(io, EV_WRITE, io_dispatch_connect_tls);
 848 ce062d50 2019-08-21 martijn 	} else {
 849 ce062d50 2019-08-21 martijn 		io->state = IO_STATE_ACCEPT_TLS;
 850 ce062d50 2019-08-21 martijn 		SSL_set_accept_state(io->tls);
 851 ce062d50 2019-08-21 martijn 		io_reset(io, EV_READ, io_dispatch_accept_tls);
 852 ce062d50 2019-08-21 martijn 	}
 853 ce062d50 2019-08-21 martijn 
 854 ce062d50 2019-08-21 martijn 	return (0);
 855 ce062d50 2019-08-21 martijn }
 856 ce062d50 2019-08-21 martijn 
 857 ce062d50 2019-08-21 martijn void
 858 ce062d50 2019-08-21 martijn io_dispatch_accept_tls(int fd, short event, void *humppa)
 859 ce062d50 2019-08-21 martijn {
 860 ce062d50 2019-08-21 martijn 	struct io	*io = humppa;
 861 ce062d50 2019-08-21 martijn 	int		 e, ret;
 862 ce062d50 2019-08-21 martijn 
 863 ce062d50 2019-08-21 martijn 	io_frame_enter("io_dispatch_accept_tls", io, event);
 864 ce062d50 2019-08-21 martijn 
 865 ce062d50 2019-08-21 martijn 	if (event == EV_TIMEOUT) {
 866 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TIMEOUT);
 867 ce062d50 2019-08-21 martijn 		goto leave;
 868 ce062d50 2019-08-21 martijn 	}
 869 ce062d50 2019-08-21 martijn 
 870 ce062d50 2019-08-21 martijn 	if ((ret = SSL_accept(io->tls)) > 0) {
 871 ce062d50 2019-08-21 martijn 		io->state = IO_STATE_UP;
 872 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TLSREADY);
 873 ce062d50 2019-08-21 martijn 		goto leave;
 874 ce062d50 2019-08-21 martijn 	}
 875 ce062d50 2019-08-21 martijn 
 876 ce062d50 2019-08-21 martijn 	switch ((e = SSL_get_error(io->tls, ret))) {
 877 ce062d50 2019-08-21 martijn 	case SSL_ERROR_WANT_READ:
 878 ce062d50 2019-08-21 martijn 		io_reset(io, EV_READ, io_dispatch_accept_tls);
 879 ce062d50 2019-08-21 martijn 		break;
 880 ce062d50 2019-08-21 martijn 	case SSL_ERROR_WANT_WRITE:
 881 ce062d50 2019-08-21 martijn 		io_reset(io, EV_WRITE, io_dispatch_accept_tls);
 882 ce062d50 2019-08-21 martijn 		break;
 883 ce062d50 2019-08-21 martijn 	default:
 884 ce062d50 2019-08-21 martijn 		io->error = io_tls_error();
 885 ce062d50 2019-08-21 martijn 		ssl_error("io_dispatch_accept_tls:SSL_accept");
 886 ce062d50 2019-08-21 martijn 		io_callback(io, IO_ERROR);
 887 ce062d50 2019-08-21 martijn 		break;
 888 ce062d50 2019-08-21 martijn 	}
 889 ce062d50 2019-08-21 martijn 
 890 ce062d50 2019-08-21 martijn     leave:
 891 ce062d50 2019-08-21 martijn 	io_frame_leave(io);
 892 ce062d50 2019-08-21 martijn }
 893 ce062d50 2019-08-21 martijn 
 894 ce062d50 2019-08-21 martijn void
 895 ce062d50 2019-08-21 martijn io_dispatch_connect_tls(int fd, short event, void *humppa)
 896 ce062d50 2019-08-21 martijn {
 897 ce062d50 2019-08-21 martijn 	struct io	*io = humppa;
 898 ce062d50 2019-08-21 martijn 	int		 e, ret;
 899 ce062d50 2019-08-21 martijn 
 900 ce062d50 2019-08-21 martijn 	io_frame_enter("io_dispatch_connect_tls", io, event);
 901 ce062d50 2019-08-21 martijn 
 902 ce062d50 2019-08-21 martijn 	if (event == EV_TIMEOUT) {
 903 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TIMEOUT);
 904 ce062d50 2019-08-21 martijn 		goto leave;
 905 ce062d50 2019-08-21 martijn 	}
 906 ce062d50 2019-08-21 martijn 
 907 ce062d50 2019-08-21 martijn 	if ((ret = SSL_connect(io->tls)) > 0) {
 908 ce062d50 2019-08-21 martijn 		io->state = IO_STATE_UP;
 909 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TLSREADY);
 910 ce062d50 2019-08-21 martijn 		goto leave;
 911 ce062d50 2019-08-21 martijn 	}
 912 ce062d50 2019-08-21 martijn 
 913 ce062d50 2019-08-21 martijn 	switch ((e = SSL_get_error(io->tls, ret))) {
 914 ce062d50 2019-08-21 martijn 	case SSL_ERROR_WANT_READ:
 915 ce062d50 2019-08-21 martijn 		io_reset(io, EV_READ, io_dispatch_connect_tls);
 916 ce062d50 2019-08-21 martijn 		break;
 917 ce062d50 2019-08-21 martijn 	case SSL_ERROR_WANT_WRITE:
 918 ce062d50 2019-08-21 martijn 		io_reset(io, EV_WRITE, io_dispatch_connect_tls);
 919 ce062d50 2019-08-21 martijn 		break;
 920 ce062d50 2019-08-21 martijn 	default:
 921 ce062d50 2019-08-21 martijn 		io->error = io_tls_error();
 922 ce062d50 2019-08-21 martijn 		ssl_error("io_dispatch_connect_ssl:SSL_connect");
 923 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TLSERROR);
 924 ce062d50 2019-08-21 martijn 		break;
 925 ce062d50 2019-08-21 martijn 	}
 926 ce062d50 2019-08-21 martijn 
 927 ce062d50 2019-08-21 martijn     leave:
 928 ce062d50 2019-08-21 martijn 	io_frame_leave(io);
 929 ce062d50 2019-08-21 martijn }
 930 ce062d50 2019-08-21 martijn 
 931 ce062d50 2019-08-21 martijn void
 932 ce062d50 2019-08-21 martijn io_dispatch_read_tls(int fd, short event, void *humppa)
 933 ce062d50 2019-08-21 martijn {
 934 ce062d50 2019-08-21 martijn 	struct io	*io = humppa;
 935 ce062d50 2019-08-21 martijn 	int		 n, saved_errno;
 936 ce062d50 2019-08-21 martijn 
 937 ce062d50 2019-08-21 martijn 	io_frame_enter("io_dispatch_read_tls", io, event);
 938 ce062d50 2019-08-21 martijn 
 939 ce062d50 2019-08-21 martijn 	if (event == EV_TIMEOUT) {
 940 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TIMEOUT);
 941 ce062d50 2019-08-21 martijn 		goto leave;
 942 ce062d50 2019-08-21 martijn 	}
 943 ce062d50 2019-08-21 martijn 
 944 ce062d50 2019-08-21 martijn again:
 945 ce062d50 2019-08-21 martijn 	iobuf_normalize(&io->iobuf);
 946 ce062d50 2019-08-21 martijn 	switch ((n = iobuf_read_tls(&io->iobuf, (SSL*)io->tls))) {
 947 ce062d50 2019-08-21 martijn 	case IOBUF_WANT_READ:
 948 ce062d50 2019-08-21 martijn 		io_reset(io, EV_READ, io_dispatch_read_tls);
 949 ce062d50 2019-08-21 martijn 		break;
 950 ce062d50 2019-08-21 martijn 	case IOBUF_WANT_WRITE:
 951 ce062d50 2019-08-21 martijn 		io_reset(io, EV_WRITE, io_dispatch_read_tls);
 952 ce062d50 2019-08-21 martijn 		break;
 953 ce062d50 2019-08-21 martijn 	case IOBUF_CLOSED:
 954 ce062d50 2019-08-21 martijn 		io_callback(io, IO_DISCONNECTED);
 955 ce062d50 2019-08-21 martijn 		break;
 956 ce062d50 2019-08-21 martijn 	case IOBUF_ERROR:
 957 ce062d50 2019-08-21 martijn 		saved_errno = errno;
 958 ce062d50 2019-08-21 martijn 		io->error = strerror(errno);
 959 ce062d50 2019-08-21 martijn 		errno = saved_errno;
 960 ce062d50 2019-08-21 martijn 		io_callback(io, IO_ERROR);
 961 ce062d50 2019-08-21 martijn 		break;
 962 ce062d50 2019-08-21 martijn 	case IOBUF_TLSERROR:
 963 ce062d50 2019-08-21 martijn 		io->error = io_tls_error();
 964 ce062d50 2019-08-21 martijn 		ssl_error("io_dispatch_read_tls:SSL_read");
 965 ce062d50 2019-08-21 martijn 		io_callback(io, IO_ERROR);
 966 ce062d50 2019-08-21 martijn 		break;
 967 ce062d50 2019-08-21 martijn 	default:
 968 ce062d50 2019-08-21 martijn 		io_debug("io_dispatch_read_tls(...) -> r=%d\n", n);
 969 ce062d50 2019-08-21 martijn 		io_callback(io, IO_DATAIN);
 970 ce062d50 2019-08-21 martijn 		if (current == io && IO_READING(io) && SSL_pending(io->tls))
 971 ce062d50 2019-08-21 martijn 			goto again;
 972 ce062d50 2019-08-21 martijn 	}
 973 ce062d50 2019-08-21 martijn 
 974 ce062d50 2019-08-21 martijn     leave:
 975 ce062d50 2019-08-21 martijn 	io_frame_leave(io);
 976 ce062d50 2019-08-21 martijn }
 977 ce062d50 2019-08-21 martijn 
 978 ce062d50 2019-08-21 martijn void
 979 ce062d50 2019-08-21 martijn io_dispatch_write_tls(int fd, short event, void *humppa)
 980 ce062d50 2019-08-21 martijn {
 981 ce062d50 2019-08-21 martijn 	struct io	*io = humppa;
 982 ce062d50 2019-08-21 martijn 	int		 n, saved_errno;
 983 ce062d50 2019-08-21 martijn 	size_t		 w2, w;
 984 ce062d50 2019-08-21 martijn 
 985 ce062d50 2019-08-21 martijn 	io_frame_enter("io_dispatch_write_tls", io, event);
 986 ce062d50 2019-08-21 martijn 
 987 ce062d50 2019-08-21 martijn 	if (event == EV_TIMEOUT) {
 988 ce062d50 2019-08-21 martijn 		io_callback(io, IO_TIMEOUT);
 989 ce062d50 2019-08-21 martijn 		goto leave;
 990 ce062d50 2019-08-21 martijn 	}
 991 ce062d50 2019-08-21 martijn 
 992 ce062d50 2019-08-21 martijn 	w = io_queued(io);
 993 ce062d50 2019-08-21 martijn 	switch ((n = iobuf_write_tls(&io->iobuf, (SSL*)io->tls))) {
 994 ce062d50 2019-08-21 martijn 	case IOBUF_WANT_READ:
 995 ce062d50 2019-08-21 martijn 		io_reset(io, EV_READ, io_dispatch_write_tls);
 996 ce062d50 2019-08-21 martijn 		break;
 997 ce062d50 2019-08-21 martijn 	case IOBUF_WANT_WRITE:
 998 ce062d50 2019-08-21 martijn 		io_reset(io, EV_WRITE, io_dispatch_write_tls);
 999 ce062d50 2019-08-21 martijn 		break;
1000 ce062d50 2019-08-21 martijn 	case IOBUF_CLOSED:
1001 ce062d50 2019-08-21 martijn 		io_callback(io, IO_DISCONNECTED);
1002 ce062d50 2019-08-21 martijn 		break;
1003 ce062d50 2019-08-21 martijn 	case IOBUF_ERROR:
1004 ce062d50 2019-08-21 martijn 		saved_errno = errno;
1005 ce062d50 2019-08-21 martijn 		io->error = strerror(errno);
1006 ce062d50 2019-08-21 martijn 		errno = saved_errno;
1007 ce062d50 2019-08-21 martijn 		io_callback(io, IO_ERROR);
1008 ce062d50 2019-08-21 martijn 		break;
1009 ce062d50 2019-08-21 martijn 	case IOBUF_TLSERROR:
1010 ce062d50 2019-08-21 martijn 		io->error = io_tls_error();
1011 ce062d50 2019-08-21 martijn 		ssl_error("io_dispatch_write_tls:SSL_write");
1012 ce062d50 2019-08-21 martijn 		io_callback(io, IO_ERROR);
1013 ce062d50 2019-08-21 martijn 		break;
1014 ce062d50 2019-08-21 martijn 	default:
1015 ce062d50 2019-08-21 martijn 		io_debug("io_dispatch_write_tls(...) -> w=%d\n", n);
1016 ce062d50 2019-08-21 martijn 		w2 = io_queued(io);
1017 ce062d50 2019-08-21 martijn 		if (w > io->lowat && w2 <= io->lowat)
1018 ce062d50 2019-08-21 martijn 			io_callback(io, IO_LOWAT);
1019 ce062d50 2019-08-21 martijn 		break;
1020 ce062d50 2019-08-21 martijn 	}
1021 ce062d50 2019-08-21 martijn 
1022 ce062d50 2019-08-21 martijn     leave:
1023 ce062d50 2019-08-21 martijn 	io_frame_leave(io);
1024 ce062d50 2019-08-21 martijn }
1025 ce062d50 2019-08-21 martijn 
1026 ce062d50 2019-08-21 martijn void
1027 ce062d50 2019-08-21 martijn io_reload_tls(struct io *io)
1028 ce062d50 2019-08-21 martijn {
1029 ce062d50 2019-08-21 martijn 	short	ev = 0;
1030 ce062d50 2019-08-21 martijn 	void	(*dispatch)(int, short, void*) = NULL;
1031 ce062d50 2019-08-21 martijn 
1032 ce062d50 2019-08-21 martijn 	switch (io->state) {
1033 ce062d50 2019-08-21 martijn 	case IO_STATE_CONNECT_TLS:
1034 ce062d50 2019-08-21 martijn 		ev = EV_WRITE;
1035 ce062d50 2019-08-21 martijn 		dispatch = io_dispatch_connect_tls;
1036 ce062d50 2019-08-21 martijn 		break;
1037 ce062d50 2019-08-21 martijn 	case IO_STATE_ACCEPT_TLS:
1038 ce062d50 2019-08-21 martijn 		ev = EV_READ;
1039 ce062d50 2019-08-21 martijn 		dispatch = io_dispatch_accept_tls;
1040 ce062d50 2019-08-21 martijn 		break;
1041 ce062d50 2019-08-21 martijn 	case IO_STATE_UP:
1042 ce062d50 2019-08-21 martijn 		ev = 0;
1043 ce062d50 2019-08-21 martijn 		if (IO_READING(io) && !(io->flags & IO_PAUSE_IN)) {
1044 ce062d50 2019-08-21 martijn 			ev = EV_READ;
1045 ce062d50 2019-08-21 martijn 			dispatch = io_dispatch_read_tls;
1046 ce062d50 2019-08-21 martijn 		}
1047 ce062d50 2019-08-21 martijn 		else if (IO_WRITING(io) && !(io->flags & IO_PAUSE_OUT) &&
1048 ce062d50 2019-08-21 martijn 		    io_queued(io)) {
1049 ce062d50 2019-08-21 martijn 			ev = EV_WRITE;
1050 ce062d50 2019-08-21 martijn 			dispatch = io_dispatch_write_tls;
1051 ce062d50 2019-08-21 martijn 		}
1052 ce062d50 2019-08-21 martijn 		if (!ev)
1053 ce062d50 2019-08-21 martijn 			return; /* paused */
1054 ce062d50 2019-08-21 martijn 		break;
1055 ce062d50 2019-08-21 martijn 	default:
1056 ce062d50 2019-08-21 martijn 		errx(1, "io_reload_tls(): bad state");
1057 ce062d50 2019-08-21 martijn 	}
1058 ce062d50 2019-08-21 martijn 
1059 ce062d50 2019-08-21 martijn 	io_reset(io, ev, dispatch);
1060 ce062d50 2019-08-21 martijn }
1061 ce062d50 2019-08-21 martijn 
1062 ce062d50 2019-08-21 martijn #endif /* IO_TLS */