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  *
   4 fb1a36c0 2022-01-09 op  * Permission to use, copy, modify, and distribute this software for any
   5 fb1a36c0 2022-01-09 op  * purpose with or without fee is hereby granted, provided that the above
   6 fb1a36c0 2022-01-09 op  * copyright notice and this permission notice appear in all copies.
   7 fb1a36c0 2022-01-09 op  *
   8 fb1a36c0 2022-01-09 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9 fb1a36c0 2022-01-09 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10 fb1a36c0 2022-01-09 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11 fb1a36c0 2022-01-09 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12 fb1a36c0 2022-01-09 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13 fb1a36c0 2022-01-09 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14 fb1a36c0 2022-01-09 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15 fb1a36c0 2022-01-09 op  */
  16 fb1a36c0 2022-01-09 op 
  17 bbcba3ed 2022-01-10 op #include "compat.h"
  18 bbcba3ed 2022-01-10 op 
  19 fb1a36c0 2022-01-09 op #include <sys/types.h>
  20 fb1a36c0 2022-01-09 op #include <sys/socket.h>
  21 fb1a36c0 2022-01-09 op 
  22 fb1a36c0 2022-01-09 op #include <netdb.h>
  23 fb1a36c0 2022-01-09 op 
  24 fb1a36c0 2022-01-09 op #include <assert.h>
  25 fb1a36c0 2022-01-09 op #include <endian.h>
  26 fb1a36c0 2022-01-09 op #include <errno.h>
  27 fb1a36c0 2022-01-09 op #include <fcntl.h>
  28 fb1a36c0 2022-01-09 op #include <inttypes.h>
  29 fb1a36c0 2022-01-09 op #include <signal.h>
  30 fb1a36c0 2022-01-09 op #include <stdio.h>
  31 fb1a36c0 2022-01-09 op #include <stdlib.h>
  32 fb1a36c0 2022-01-09 op #include <string.h>
  33 fb1a36c0 2022-01-09 op #include <syslog.h>
  34 fb1a36c0 2022-01-09 op #include <tls.h>
  35 fb1a36c0 2022-01-09 op #include <unistd.h>
  36 fb1a36c0 2022-01-09 op 
  37 fb1a36c0 2022-01-09 op #include "kami.h"
  38 fb1a36c0 2022-01-09 op #include "log.h"
  39 fb1a36c0 2022-01-09 op #include "utils.h"
  40 893d3be6 2022-01-19 op #include "9pclib.h"
  41 fb1a36c0 2022-01-09 op 
  42 fb1a36c0 2022-01-09 op #define DEBUG_PACKETS 0
  43 fb1a36c0 2022-01-09 op 
  44 fb1a36c0 2022-01-09 op #define PROMPT "=% "
  45 fb1a36c0 2022-01-09 op 
  46 fb1a36c0 2022-01-09 op /* flags */
  47 fb1a36c0 2022-01-09 op int			 verbose;
  48 fb1a36c0 2022-01-09 op int			 tls;
  49 fb1a36c0 2022-01-09 op const char		*keypath;
  50 fb1a36c0 2022-01-09 op const char		*crtpath;
  51 fb1a36c0 2022-01-09 op const char		*host;
  52 fb1a36c0 2022-01-09 op const char		*port;
  53 fb1a36c0 2022-01-09 op 
  54 fb1a36c0 2022-01-09 op /* state */
  55 fb1a36c0 2022-01-09 op struct tls_config	*tlsconf;
  56 fb1a36c0 2022-01-09 op struct tls		*ctx;
  57 fb1a36c0 2022-01-09 op struct bufferevent	*bev, *inbev;
  58 fb1a36c0 2022-01-09 op 
  59 fb1a36c0 2022-01-09 op static void __dead	 usage(int);
  60 fb1a36c0 2022-01-09 op 
  61 fb1a36c0 2022-01-09 op static void		 sig_handler(int, short, void *);
  62 fb1a36c0 2022-01-09 op 
  63 fb1a36c0 2022-01-09 op static int		 openconn(void);
  64 fb1a36c0 2022-01-09 op static void		 mark_nonblock(int);
  65 fb1a36c0 2022-01-09 op 
  66 fb1a36c0 2022-01-09 op static void		 tls_readcb(int, short, void *);
  67 fb1a36c0 2022-01-09 op static void		 tls_writecb(int, short, void *);
  68 fb1a36c0 2022-01-09 op 
  69 fb1a36c0 2022-01-09 op static void		 client_read(struct bufferevent *, void *);
  70 fb1a36c0 2022-01-09 op static void		 client_write(struct bufferevent *, void *);
  71 fb1a36c0 2022-01-09 op static void		 client_error(struct bufferevent *, short, void *);
  72 fb1a36c0 2022-01-09 op 
  73 fb1a36c0 2022-01-09 op static void		 repl_read(struct bufferevent *, void *);
  74 fb1a36c0 2022-01-09 op static void		 repl_error(struct bufferevent *, short, void *);
  75 fb1a36c0 2022-01-09 op 
  76 fb1a36c0 2022-01-09 op static void		 excmd_version(const char **, int);
  77 fb1a36c0 2022-01-09 op static void		 excmd_attach(const char **, int);
  78 fb1a36c0 2022-01-09 op static void		 excmd_clunk(const char **, int);
  79 fb1a36c0 2022-01-09 op static void		 excmd_flush(const char **, int);
  80 fb1a36c0 2022-01-09 op static void		 excmd_walk(const char ** , int);
  81 fb1a36c0 2022-01-09 op static void		 excmd_open(const char ** , int);
  82 fb1a36c0 2022-01-09 op static void		 excmd_create(const char ** , int);
  83 fb1a36c0 2022-01-09 op static void		 excmd_read(const char ** , int);
  84 fb1a36c0 2022-01-09 op static void		 excmd_write(const char **, int);
  85 fb1a36c0 2022-01-09 op static void		 excmd(const char **, int);
  86 fb1a36c0 2022-01-09 op 
  87 fb1a36c0 2022-01-09 op static void		 pp_qid(const uint8_t *, uint32_t);
  88 fb1a36c0 2022-01-09 op static void		 pp_msg(uint32_t, uint8_t, uint16_t, const uint8_t *);
  89 fb1a36c0 2022-01-09 op static void		 handle_9p(const uint8_t *, size_t);
  90 fb1a36c0 2022-01-09 op static void		 clr(void);
  91 fb1a36c0 2022-01-09 op static void		 prompt(void);
  92 fb1a36c0 2022-01-09 op 
  93 fb1a36c0 2022-01-09 op static void __dead
  94 fb1a36c0 2022-01-09 op usage(int ret)
  95 fb1a36c0 2022-01-09 op {
  96 fb1a36c0 2022-01-09 op 	fprintf(stderr,
  97 fb1a36c0 2022-01-09 op 	    "usage: %s [-chv] [-C crt] [-K key] [-H host] [-P port]\n",
  98 fb1a36c0 2022-01-09 op 	    getprogname());
  99 fb1a36c0 2022-01-09 op 	fprintf(stderr, "kamid suite version " KAMID_VERSION "\n");
 100 fb1a36c0 2022-01-09 op 	exit(ret);
 101 fb1a36c0 2022-01-09 op }
 102 fb1a36c0 2022-01-09 op 
 103 fb1a36c0 2022-01-09 op static void
 104 fb1a36c0 2022-01-09 op sig_handler(int sig, short event, void *d)
 105 fb1a36c0 2022-01-09 op {
 106 fb1a36c0 2022-01-09 op 	/*
 107 fb1a36c0 2022-01-09 op 	 * Normal signal handler rules don't apply because libevent
 108 fb1a36c0 2022-01-09 op 	 * decouples for us.
 109 fb1a36c0 2022-01-09 op 	 */
 110 fb1a36c0 2022-01-09 op 
 111 fb1a36c0 2022-01-09 op 	switch (sig) {
 112 fb1a36c0 2022-01-09 op 	case SIGINT:
 113 fb1a36c0 2022-01-09 op 	case SIGTERM:
 114 fb1a36c0 2022-01-09 op 		clr();
 115 fb1a36c0 2022-01-09 op 		log_warnx("Shutting down...");
 116 fb1a36c0 2022-01-09 op 		event_loopbreak();
 117 fb1a36c0 2022-01-09 op 		return;
 118 fb1a36c0 2022-01-09 op 	default:
 119 fb1a36c0 2022-01-09 op 		fatalx("unexpected signal %d", sig);
 120 fb1a36c0 2022-01-09 op 	}
 121 fb1a36c0 2022-01-09 op }
 122 fb1a36c0 2022-01-09 op 
 123 fb1a36c0 2022-01-09 op static int
 124 fb1a36c0 2022-01-09 op openconn(void)
 125 fb1a36c0 2022-01-09 op {
 126 fb1a36c0 2022-01-09 op 	struct addrinfo hints, *res, *res0;
 127 fb1a36c0 2022-01-09 op 	int error;
 128 fb1a36c0 2022-01-09 op 	int save_errno;
 129 fb1a36c0 2022-01-09 op 	int s;
 130 fb1a36c0 2022-01-09 op 	const char *cause = NULL;
 131 fb1a36c0 2022-01-09 op 
 132 fb1a36c0 2022-01-09 op 	memset(&hints, 0, sizeof(hints));
 133 fb1a36c0 2022-01-09 op 	hints.ai_family = AF_UNSPEC;
 134 fb1a36c0 2022-01-09 op 	hints.ai_socktype = SOCK_STREAM;
 135 fb1a36c0 2022-01-09 op 	if ((error = getaddrinfo(host, port, &hints, &res0))) {
 136 fb1a36c0 2022-01-09 op 		warnx("%s", gai_strerror(error));
 137 fb1a36c0 2022-01-09 op 		return -1;
 138 fb1a36c0 2022-01-09 op 	}
 139 fb1a36c0 2022-01-09 op 
 140 fb1a36c0 2022-01-09 op 	s = -1;
 141 fb1a36c0 2022-01-09 op 	for (res = res0; res; res = res->ai_next) {
 142 fb1a36c0 2022-01-09 op 		s = socket(res->ai_family, res->ai_socktype,
 143 fb1a36c0 2022-01-09 op 		    res->ai_protocol);
 144 fb1a36c0 2022-01-09 op 		if (s == -1) {
 145 fb1a36c0 2022-01-09 op 			cause = "socket";
 146 fb1a36c0 2022-01-09 op 			continue;
 147 fb1a36c0 2022-01-09 op 		}
 148 fb1a36c0 2022-01-09 op 
 149 fb1a36c0 2022-01-09 op 		if (connect(s, res->ai_addr, res->ai_addrlen) == -1) {
 150 fb1a36c0 2022-01-09 op 			cause = "connect";
 151 fb1a36c0 2022-01-09 op 			save_errno = errno;
 152 fb1a36c0 2022-01-09 op 			close(s);
 153 fb1a36c0 2022-01-09 op 			errno = save_errno;
 154 fb1a36c0 2022-01-09 op 			s = -1;
 155 fb1a36c0 2022-01-09 op 			continue;
 156 fb1a36c0 2022-01-09 op 		}
 157 fb1a36c0 2022-01-09 op 
 158 fb1a36c0 2022-01-09 op 		break;
 159 fb1a36c0 2022-01-09 op 	}
 160 fb1a36c0 2022-01-09 op 
 161 fb1a36c0 2022-01-09 op 	freeaddrinfo(res0);
 162 fb1a36c0 2022-01-09 op 
 163 fb1a36c0 2022-01-09 op 	if (s == -1)
 164 fb1a36c0 2022-01-09 op 		warn("%s", cause);
 165 fb1a36c0 2022-01-09 op 
 166 fb1a36c0 2022-01-09 op 	return s;
 167 fb1a36c0 2022-01-09 op }
 168 fb1a36c0 2022-01-09 op 
 169 fb1a36c0 2022-01-09 op static void
 170 fb1a36c0 2022-01-09 op mark_nonblock(int fd)
 171 fb1a36c0 2022-01-09 op {
 172 fb1a36c0 2022-01-09 op 	int flags;
 173 fb1a36c0 2022-01-09 op 
 174 fb1a36c0 2022-01-09 op 	if ((flags = fcntl(fd, F_GETFL)) == -1)
 175 fb1a36c0 2022-01-09 op 		fatal("fcntl(F_GETFL)");
 176 fb1a36c0 2022-01-09 op 	if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1)
 177 fb1a36c0 2022-01-09 op 		fatal("fcntl(F_SETFL)");
 178 fb1a36c0 2022-01-09 op }
 179 fb1a36c0 2022-01-09 op 
 180 fb1a36c0 2022-01-09 op static void
 181 fb1a36c0 2022-01-09 op tls_readcb(int fd, short event, void *d)
 182 fb1a36c0 2022-01-09 op {
 183 fb1a36c0 2022-01-09 op 	struct bufferevent	*bufev = d;
 184 fb1a36c0 2022-01-09 op 	char			 buf[IBUF_READ_SIZE];
 185 fb1a36c0 2022-01-09 op 	int			 what = EVBUFFER_READ;
 186 fb1a36c0 2022-01-09 op 	int			 howmuch = IBUF_READ_SIZE;
 187 fb1a36c0 2022-01-09 op 	ssize_t			 ret;
 188 fb1a36c0 2022-01-09 op 	size_t			 len;
 189 fb1a36c0 2022-01-09 op 
 190 fb1a36c0 2022-01-09 op 	if (event == EV_TIMEOUT) {
 191 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_TIMEOUT;
 192 fb1a36c0 2022-01-09 op 		goto err;
 193 fb1a36c0 2022-01-09 op 	}
 194 fb1a36c0 2022-01-09 op 
 195 fb1a36c0 2022-01-09 op 	if (bufev->wm_read.high != 0)
 196 fb1a36c0 2022-01-09 op 		howmuch = MIN(sizeof(buf), bufev->wm_read.high);
 197 fb1a36c0 2022-01-09 op 
 198 fb1a36c0 2022-01-09 op 	switch (ret = tls_read(ctx, buf, howmuch)) {
 199 fb1a36c0 2022-01-09 op 	case TLS_WANT_POLLIN:
 200 fb1a36c0 2022-01-09 op 	case TLS_WANT_POLLOUT:
 201 fb1a36c0 2022-01-09 op 		goto retry;
 202 fb1a36c0 2022-01-09 op 	case -1:
 203 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_ERROR;
 204 fb1a36c0 2022-01-09 op 		goto err;
 205 fb1a36c0 2022-01-09 op 	}
 206 fb1a36c0 2022-01-09 op 	len = ret;
 207 fb1a36c0 2022-01-09 op 
 208 fb1a36c0 2022-01-09 op 	if (len == 0) {
 209 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_EOF;
 210 fb1a36c0 2022-01-09 op 		goto err;
 211 fb1a36c0 2022-01-09 op 	}
 212 fb1a36c0 2022-01-09 op 
 213 fb1a36c0 2022-01-09 op 	if (evbuffer_add(bufev->input, buf, len) == -1) {
 214 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_ERROR;
 215 fb1a36c0 2022-01-09 op 		goto err;
 216 fb1a36c0 2022-01-09 op 	}
 217 fb1a36c0 2022-01-09 op 
 218 fb1a36c0 2022-01-09 op 	event_add(&bufev->ev_read, NULL);
 219 fb1a36c0 2022-01-09 op 
 220 fb1a36c0 2022-01-09 op 	len = EVBUFFER_LENGTH(bufev->input);
 221 fb1a36c0 2022-01-09 op 	if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
 222 fb1a36c0 2022-01-09 op 		return;
 223 fb1a36c0 2022-01-09 op 	if (bufev->readcb != NULL)
 224 fb1a36c0 2022-01-09 op 		(*bufev->readcb)(bufev, bufev->cbarg);
 225 fb1a36c0 2022-01-09 op 	return;
 226 fb1a36c0 2022-01-09 op 
 227 fb1a36c0 2022-01-09 op retry:
 228 fb1a36c0 2022-01-09 op 	event_add(&bufev->ev_read, NULL);
 229 fb1a36c0 2022-01-09 op 	return;
 230 fb1a36c0 2022-01-09 op 
 231 fb1a36c0 2022-01-09 op err:
 232 fb1a36c0 2022-01-09 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
 233 fb1a36c0 2022-01-09 op }
 234 fb1a36c0 2022-01-09 op 
 235 fb1a36c0 2022-01-09 op static void
 236 fb1a36c0 2022-01-09 op tls_writecb(int fd, short event, void *d)
 237 fb1a36c0 2022-01-09 op {
 238 fb1a36c0 2022-01-09 op 	struct bufferevent	*bufev = d;
 239 fb1a36c0 2022-01-09 op 	ssize_t			 ret;
 240 fb1a36c0 2022-01-09 op 	size_t			 len;
 241 fb1a36c0 2022-01-09 op 	short			 what = EVBUFFER_WRITE;
 242 fb1a36c0 2022-01-09 op 	void			*data;
 243 fb1a36c0 2022-01-09 op 
 244 fb1a36c0 2022-01-09 op 	if (event == EV_TIMEOUT) {
 245 fb1a36c0 2022-01-09 op 		what |= EVBUFFER_TIMEOUT;
 246 fb1a36c0 2022-01-09 op 		goto err;
 247 fb1a36c0 2022-01-09 op 	}
 248 fb1a36c0 2022-01-09 op 
 249 fb1a36c0 2022-01-09 op 	len = EVBUFFER_LENGTH(bufev->output);
 250 fb1a36c0 2022-01-09 op 	if (len != 0) {
 251 fb1a36c0 2022-01-09 op 		data = EVBUFFER_DATA(bufev->output);
 252 fb1a36c0 2022-01-09 op 
 253 fb1a36c0 2022-01-09 op #if DEBUG_PACKETS
 254 fb1a36c0 2022-01-09 op 		hexdump("outgoing msg", data, len);
 255 fb1a36c0 2022-01-09 op #endif
 256 fb1a36c0 2022-01-09 op 
 257 fb1a36c0 2022-01-09 op 		switch (ret = tls_write(ctx, data, len)) {
 258 fb1a36c0 2022-01-09 op 		case TLS_WANT_POLLIN:
 259 fb1a36c0 2022-01-09 op 		case TLS_WANT_POLLOUT:
 260 fb1a36c0 2022-01-09 op 			goto retry;
 261 fb1a36c0 2022-01-09 op 		case -1:
 262 fb1a36c0 2022-01-09 op 			what |= EVBUFFER_ERROR;
 263 fb1a36c0 2022-01-09 op 			goto err;
 264 fb1a36c0 2022-01-09 op 		}
 265 fb1a36c0 2022-01-09 op 		evbuffer_drain(bufev->output, ret);
 266 fb1a36c0 2022-01-09 op 	}
 267 fb1a36c0 2022-01-09 op 
 268 fb1a36c0 2022-01-09 op 	if (EVBUFFER_LENGTH(bufev->output) != 0)
 269 fb1a36c0 2022-01-09 op 		event_add(&bufev->ev_write, NULL);
 270 fb1a36c0 2022-01-09 op 
 271 fb1a36c0 2022-01-09 op 	if (bufev->writecb != NULL &&
 272 fb1a36c0 2022-01-09 op 	    EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
 273 fb1a36c0 2022-01-09 op 		(*bufev->writecb)(bufev, bufev->cbarg);
 274 fb1a36c0 2022-01-09 op 	return;
 275 fb1a36c0 2022-01-09 op 
 276 fb1a36c0 2022-01-09 op retry:
 277 fb1a36c0 2022-01-09 op 	event_add(&bufev->ev_write, NULL);
 278 fb1a36c0 2022-01-09 op 	return;
 279 fb1a36c0 2022-01-09 op err:
 280 fb1a36c0 2022-01-09 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
 281 fb1a36c0 2022-01-09 op }
 282 fb1a36c0 2022-01-09 op 
 283 fb1a36c0 2022-01-09 op static void
 284 fb1a36c0 2022-01-09 op client_read(struct bufferevent *bev, void *d)
 285 fb1a36c0 2022-01-09 op {
 286 fb1a36c0 2022-01-09 op 	struct evbuffer	*src = EVBUFFER_INPUT(bev);
 287 fb1a36c0 2022-01-09 op 	uint32_t	 len;
 288 fb1a36c0 2022-01-09 op 	uint8_t		*data;
 289 fb1a36c0 2022-01-09 op 
 290 fb1a36c0 2022-01-09 op 	for (;;) {
 291 fb1a36c0 2022-01-09 op 		if (EVBUFFER_LENGTH(src) < sizeof(len))
 292 fb1a36c0 2022-01-09 op 			return;
 293 fb1a36c0 2022-01-09 op 
 294 fb1a36c0 2022-01-09 op 		data = EVBUFFER_DATA(src);
 295 fb1a36c0 2022-01-09 op 
 296 fb1a36c0 2022-01-09 op 		memcpy(&len, data, sizeof(len));
 297 fb1a36c0 2022-01-09 op 		len = le32toh(len);
 298 fb1a36c0 2022-01-09 op 
 299 fb1a36c0 2022-01-09 op 		if (len < HEADERSIZE)
 300 fb1a36c0 2022-01-09 op 			fatal("incoming message is too small! (%d bytes)",
 301 fb1a36c0 2022-01-09 op 			    len);
 302 fb1a36c0 2022-01-09 op 
 303 fb1a36c0 2022-01-09 op 		if (len > EVBUFFER_LENGTH(src))
 304 fb1a36c0 2022-01-09 op 			return;
 305 fb1a36c0 2022-01-09 op 
 306 fb1a36c0 2022-01-09 op #if DEBUG_PACKETS
 307 fb1a36c0 2022-01-09 op 		hexdump("incoming msg", data, len);
 308 fb1a36c0 2022-01-09 op #endif
 309 fb1a36c0 2022-01-09 op 
 310 fb1a36c0 2022-01-09 op 		handle_9p(data, len);
 311 fb1a36c0 2022-01-09 op 		evbuffer_drain(src, len);
 312 fb1a36c0 2022-01-09 op 	}
 313 fb1a36c0 2022-01-09 op }
 314 fb1a36c0 2022-01-09 op 
 315 fb1a36c0 2022-01-09 op static void
 316 fb1a36c0 2022-01-09 op client_write(struct bufferevent *bev, void *data)
 317 fb1a36c0 2022-01-09 op {
 318 fb1a36c0 2022-01-09 op 	return; /* nothing to do */
 319 fb1a36c0 2022-01-09 op }
 320 fb1a36c0 2022-01-09 op 
 321 fb1a36c0 2022-01-09 op static void
 322 fb1a36c0 2022-01-09 op client_error(struct bufferevent *bev, short err, void *data)
 323 fb1a36c0 2022-01-09 op {
 324 fb1a36c0 2022-01-09 op 	if (err & EVBUFFER_ERROR)
 325 fb1a36c0 2022-01-09 op 		fatal("buffer event error");
 326 fb1a36c0 2022-01-09 op 
 327 fb1a36c0 2022-01-09 op 	if (err & EVBUFFER_EOF) {
 328 fb1a36c0 2022-01-09 op 		clr();
 329 fb1a36c0 2022-01-09 op 		log_info("EOF");
 330 fb1a36c0 2022-01-09 op 		event_loopbreak();
 331 fb1a36c0 2022-01-09 op 		return;
 332 fb1a36c0 2022-01-09 op 	}
 333 fb1a36c0 2022-01-09 op 
 334 fb1a36c0 2022-01-09 op 	clr();
 335 fb1a36c0 2022-01-09 op 	log_warnx("unknown event error");
 336 fb1a36c0 2022-01-09 op 	event_loopbreak();
 337 fb1a36c0 2022-01-09 op }
 338 fb1a36c0 2022-01-09 op 
 339 fb1a36c0 2022-01-09 op static void
 340 fb1a36c0 2022-01-09 op repl_read(struct bufferevent *bev, void *d)
 341 fb1a36c0 2022-01-09 op {
 342 fb1a36c0 2022-01-09 op 	size_t		 len;
 343 fb1a36c0 2022-01-09 op 	int		 argc;
 344 fb1a36c0 2022-01-09 op 	const char	*argv[10], **ap;
 345 fb1a36c0 2022-01-09 op 	char		*line;
 346 fb1a36c0 2022-01-09 op 
 347 fb1a36c0 2022-01-09 op 	line = evbuffer_readln(bev->input, &len, EVBUFFER_EOL_LF);
 348 fb1a36c0 2022-01-09 op 	if (line == NULL)
 349 fb1a36c0 2022-01-09 op 		return;
 350 fb1a36c0 2022-01-09 op 
 351 fb1a36c0 2022-01-09 op 	for (argc = 0, ap = argv; ap < &argv[9] &&
 352 fb1a36c0 2022-01-09 op 	    (*ap = strsep(&line, " \t")) != NULL;) {
 353 fb1a36c0 2022-01-09 op 		if (**ap != '\0')
 354 fb1a36c0 2022-01-09 op 			ap++, argc++;
 355 fb1a36c0 2022-01-09 op 	}
 356 fb1a36c0 2022-01-09 op 
 357 fb1a36c0 2022-01-09 op 	clr();
 358 fb1a36c0 2022-01-09 op 	excmd(argv, argc);
 359 fb1a36c0 2022-01-09 op 	prompt();
 360 fb1a36c0 2022-01-09 op 
 361 fb1a36c0 2022-01-09 op 	free(line);
 362 fb1a36c0 2022-01-09 op }
 363 fb1a36c0 2022-01-09 op 
 364 fb1a36c0 2022-01-09 op static void
 365 fb1a36c0 2022-01-09 op repl_error(struct bufferevent *bev, short error, void *d)
 366 fb1a36c0 2022-01-09 op {
 367 fb1a36c0 2022-01-09 op 	fatalx("an error occurred");
 368 fb1a36c0 2022-01-09 op }
 369 fb1a36c0 2022-01-09 op 
 370 fb1a36c0 2022-01-09 op static inline void
 371 fb1a36c0 2022-01-09 op do_send(void)
 372 fb1a36c0 2022-01-09 op {
 373 fb1a36c0 2022-01-09 op 	bufferevent_write_buffer(bev, evb);
 374 fb1a36c0 2022-01-09 op }
 375 fb1a36c0 2022-01-09 op 
 376 fb1a36c0 2022-01-09 op /* version [version-str] */
 377 fb1a36c0 2022-01-09 op static void
 378 fb1a36c0 2022-01-09 op excmd_version(const char **argv, int argc)
 379 fb1a36c0 2022-01-09 op {
 380 fb1a36c0 2022-01-09 op 	const char	*s;
 381 fb1a36c0 2022-01-09 op 
 382 fb1a36c0 2022-01-09 op 	s = VERSION9P;
 383 fb1a36c0 2022-01-09 op 	if (argc == 2)
 384 fb1a36c0 2022-01-09 op 		s = argv[1];
 385 fb1a36c0 2022-01-09 op 
 386 fb1a36c0 2022-01-09 op 	tversion(s, MSIZE9P);
 387 fb1a36c0 2022-01-09 op 	do_send();
 388 fb1a36c0 2022-01-09 op }
 389 fb1a36c0 2022-01-09 op 
 390 fb1a36c0 2022-01-09 op /* attach fid uname aname */
 391 fb1a36c0 2022-01-09 op static void
 392 fb1a36c0 2022-01-09 op excmd_attach(const char **argv, int argc)
 393 fb1a36c0 2022-01-09 op {
 394 fb1a36c0 2022-01-09 op 	uint32_t	 fid;
 395 fb1a36c0 2022-01-09 op 	const char	*errstr;
 396 fb1a36c0 2022-01-09 op 
 397 fb1a36c0 2022-01-09 op 	if (argc != 4)
 398 fb1a36c0 2022-01-09 op 		goto usage;
 399 fb1a36c0 2022-01-09 op 
 400 fb1a36c0 2022-01-09 op         fid = strtonum(argv[1], 0, UINT32_MAX, &errstr);
 401 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 402 fb1a36c0 2022-01-09 op 		log_warnx("fid is %s: %s", errstr, argv[1]);
 403 fb1a36c0 2022-01-09 op 		return;
 404 fb1a36c0 2022-01-09 op 	}
 405 fb1a36c0 2022-01-09 op 
 406 fb1a36c0 2022-01-09 op 	tattach(fid, NOFID, argv[2], argv[3]);
 407 fb1a36c0 2022-01-09 op 	do_send();
 408 fb1a36c0 2022-01-09 op 	return;
 409 fb1a36c0 2022-01-09 op 
 410 fb1a36c0 2022-01-09 op usage:
 411 fb1a36c0 2022-01-09 op 	log_warnx("usage: attach fid uname aname");
 412 fb1a36c0 2022-01-09 op }
 413 fb1a36c0 2022-01-09 op 
 414 fb1a36c0 2022-01-09 op /* clunk fid */
 415 fb1a36c0 2022-01-09 op static void
 416 fb1a36c0 2022-01-09 op excmd_clunk(const char **argv, int argc)
 417 fb1a36c0 2022-01-09 op {
 418 fb1a36c0 2022-01-09 op 	uint32_t	 fid;
 419 fb1a36c0 2022-01-09 op 	const char	*errstr;
 420 fb1a36c0 2022-01-09 op 
 421 fb1a36c0 2022-01-09 op 	if (argc != 2)
 422 fb1a36c0 2022-01-09 op 		goto usage;
 423 fb1a36c0 2022-01-09 op 
 424 fb1a36c0 2022-01-09 op 	fid = strtonum(argv[1], 0, UINT32_MAX, &errstr);
 425 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 426 fb1a36c0 2022-01-09 op 		log_warnx("fid is %s: %s", errstr, argv[1]);
 427 fb1a36c0 2022-01-09 op 		return;
 428 fb1a36c0 2022-01-09 op 	}
 429 fb1a36c0 2022-01-09 op 
 430 fb1a36c0 2022-01-09 op 	tclunk(fid);
 431 fb1a36c0 2022-01-09 op 	do_send();
 432 fb1a36c0 2022-01-09 op 	return;
 433 fb1a36c0 2022-01-09 op 
 434 fb1a36c0 2022-01-09 op usage:
 435 fb1a36c0 2022-01-09 op 	log_warnx("usage: clunk fid");
 436 fb1a36c0 2022-01-09 op }
 437 fb1a36c0 2022-01-09 op 
 438 fb1a36c0 2022-01-09 op /* flush oldtag */
 439 fb1a36c0 2022-01-09 op static void
 440 fb1a36c0 2022-01-09 op excmd_flush(const char **argv, int argc)
 441 fb1a36c0 2022-01-09 op {
 442 fb1a36c0 2022-01-09 op 	uint16_t	 oldtag;
 443 fb1a36c0 2022-01-09 op 	const char	*errstr;
 444 fb1a36c0 2022-01-09 op 
 445 fb1a36c0 2022-01-09 op 	if (argc != 2)
 446 fb1a36c0 2022-01-09 op 		goto usage;
 447 fb1a36c0 2022-01-09 op 
 448 fb1a36c0 2022-01-09 op 	oldtag = strtonum(argv[1], 0, UINT16_MAX, &errstr);
 449 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 450 fb1a36c0 2022-01-09 op 		log_warnx("oldtag is %s: %s", errstr, argv[1]);
 451 fb1a36c0 2022-01-09 op 		return;
 452 fb1a36c0 2022-01-09 op 	}
 453 fb1a36c0 2022-01-09 op 
 454 fb1a36c0 2022-01-09 op 	tflush(oldtag);
 455 fb1a36c0 2022-01-09 op 	do_send();
 456 fb1a36c0 2022-01-09 op 	return;
 457 fb1a36c0 2022-01-09 op 
 458 fb1a36c0 2022-01-09 op usage:
 459 fb1a36c0 2022-01-09 op 	log_warnx("usage: flush oldtag");
 460 fb1a36c0 2022-01-09 op }
 461 fb1a36c0 2022-01-09 op 
 462 fb1a36c0 2022-01-09 op /* walk fid newfid wnames... */
 463 fb1a36c0 2022-01-09 op static void
 464 fb1a36c0 2022-01-09 op excmd_walk(const char **argv, int argc)
 465 fb1a36c0 2022-01-09 op {
 466 fb1a36c0 2022-01-09 op 	uint32_t	 fid, newfid;
 467 fb1a36c0 2022-01-09 op 	const char	*errstr;
 468 fb1a36c0 2022-01-09 op 
 469 fb1a36c0 2022-01-09 op 	if (argc < 3)
 470 fb1a36c0 2022-01-09 op 		goto usage;
 471 fb1a36c0 2022-01-09 op 
 472 fb1a36c0 2022-01-09 op 	fid = strtonum(argv[1], 0, UINT32_MAX, &errstr);
 473 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 474 fb1a36c0 2022-01-09 op 		log_warnx("fid is %s: %s", errstr, argv[1]);
 475 fb1a36c0 2022-01-09 op 		return;
 476 fb1a36c0 2022-01-09 op 	}
 477 fb1a36c0 2022-01-09 op 
 478 fb1a36c0 2022-01-09 op 	newfid = strtonum(argv[2], 0, UINT32_MAX, &errstr);
 479 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 480 fb1a36c0 2022-01-09 op 		log_warnx("newfid is %s: %s", errstr, argv[1]);
 481 fb1a36c0 2022-01-09 op 		return;
 482 fb1a36c0 2022-01-09 op 	}
 483 fb1a36c0 2022-01-09 op 
 484 fb1a36c0 2022-01-09 op 	twalk(fid, newfid, argv + 3, argc - 3);
 485 fb1a36c0 2022-01-09 op 	do_send();
 486 fb1a36c0 2022-01-09 op 	return;
 487 fb1a36c0 2022-01-09 op 
 488 fb1a36c0 2022-01-09 op usage:
 489 fb1a36c0 2022-01-09 op 	log_warnx("usage: walk fid newfid wnames...");
 490 fb1a36c0 2022-01-09 op }
 491 fb1a36c0 2022-01-09 op 
 492 fb1a36c0 2022-01-09 op /* open fid mode [flag] */
 493 fb1a36c0 2022-01-09 op static void
 494 fb1a36c0 2022-01-09 op excmd_open(const char **argv, int argc)
 495 fb1a36c0 2022-01-09 op {
 496 fb1a36c0 2022-01-09 op 	const char	*errstr;
 497 fb1a36c0 2022-01-09 op 	uint32_t	 fid;
 498 fb1a36c0 2022-01-09 op 	uint8_t		 mode = 0;
 499 fb1a36c0 2022-01-09 op 
 500 fb1a36c0 2022-01-09 op 	if (argc != 3 && argc != 4)
 501 fb1a36c0 2022-01-09 op 		goto usage;
 502 fb1a36c0 2022-01-09 op 
 503 fb1a36c0 2022-01-09 op 	fid = strtonum(argv[1], 0, UINT32_MAX, &errstr);
 504 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 505 fb1a36c0 2022-01-09 op 		log_warnx("fid is %s: %s", errstr, argv[1]);
 506 fb1a36c0 2022-01-09 op 		return;
 507 fb1a36c0 2022-01-09 op 	}
 508 fb1a36c0 2022-01-09 op 
 509 fb1a36c0 2022-01-09 op 	/* parse mode */
 510 fb1a36c0 2022-01-09 op 	if (!strcmp("read", argv[2]) || !strcmp("r", argv[2]))
 511 fb1a36c0 2022-01-09 op 		mode = KOREAD;
 512 fb1a36c0 2022-01-09 op 	else if (!strcmp("write", argv[2]) || !strcmp("w", argv[2]))
 513 fb1a36c0 2022-01-09 op 		mode = KOWRITE;
 514 fb1a36c0 2022-01-09 op 	else if (!strcmp("readwrite", argv[2]) || !strcmp("rw", argv[2]))
 515 fb1a36c0 2022-01-09 op 		mode = KORDWR;
 516 fb1a36c0 2022-01-09 op 	else {
 517 fb1a36c0 2022-01-09 op 		log_warnx("invalid mode %s", argv[2]);
 518 fb1a36c0 2022-01-09 op 		return;
 519 fb1a36c0 2022-01-09 op 	}
 520 fb1a36c0 2022-01-09 op 
 521 fb1a36c0 2022-01-09 op 	/* parse flag */
 522 fb1a36c0 2022-01-09 op 	if (argv[3] != NULL) {
 523 fb1a36c0 2022-01-09 op 		if (!strcmp("trunc", argv[3]))
 524 fb1a36c0 2022-01-09 op 			mode |= KOTRUNC;
 525 fb1a36c0 2022-01-09 op 		else if (!strcmp("rclose", argv[3]))
 526 fb1a36c0 2022-01-09 op 			mode |= KORCLOSE;
 527 fb1a36c0 2022-01-09 op 		else {
 528 fb1a36c0 2022-01-09 op 			log_warnx("invalid flag %s", argv[3]);
 529 fb1a36c0 2022-01-09 op 			return;
 530 fb1a36c0 2022-01-09 op 		}
 531 fb1a36c0 2022-01-09 op 	}
 532 fb1a36c0 2022-01-09 op 
 533 fb1a36c0 2022-01-09 op 	topen(fid, mode);
 534 fb1a36c0 2022-01-09 op 	do_send();
 535 fb1a36c0 2022-01-09 op 	return;
 536 fb1a36c0 2022-01-09 op 
 537 fb1a36c0 2022-01-09 op usage:
 538 fb1a36c0 2022-01-09 op 	log_warnx("usage: open fid mode [flag]");
 539 fb1a36c0 2022-01-09 op }
 540 fb1a36c0 2022-01-09 op 
 541 fb1a36c0 2022-01-09 op /* create fid path perm mode */
 542 fb1a36c0 2022-01-09 op static void
 543 fb1a36c0 2022-01-09 op excmd_create(const char **argv, int argc)
 544 fb1a36c0 2022-01-09 op {
 545 fb1a36c0 2022-01-09 op 	const char	*errstr;
 546 fb1a36c0 2022-01-09 op 	uint32_t	 fid;
 547 fb1a36c0 2022-01-09 op 	uint8_t		 mode = 0;
 548 fb1a36c0 2022-01-09 op 
 549 fb1a36c0 2022-01-09 op 	if (argc != 5)
 550 fb1a36c0 2022-01-09 op 		goto usage;
 551 fb1a36c0 2022-01-09 op 
 552 fb1a36c0 2022-01-09 op 	fid = strtonum(argv[1], 0, UINT32_MAX, &errstr);
 553 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 554 fb1a36c0 2022-01-09 op 		log_warnx("fid is %s: %s", errstr, argv[1]);
 555 fb1a36c0 2022-01-09 op 		return;
 556 fb1a36c0 2022-01-09 op 	}
 557 fb1a36c0 2022-01-09 op 
 558 fb1a36c0 2022-01-09 op 	/* parse mode */
 559 fb1a36c0 2022-01-09 op 	if (!strcmp("write", argv[4]) || !strcmp("w", argv[4]))
 560 fb1a36c0 2022-01-09 op 		mode = KOWRITE;
 561 fb1a36c0 2022-01-09 op 	else if (!strcmp("readwrite", argv[4]) || !strcmp("rw", argv[4]))
 562 fb1a36c0 2022-01-09 op 		mode = KORDWR;
 563 fb1a36c0 2022-01-09 op 	else {	    
 564 fb1a36c0 2022-01-09 op 		log_warnx("invalid mode %s for create", argv[4]);
 565 fb1a36c0 2022-01-09 op 		return;
 566 fb1a36c0 2022-01-09 op 	}
 567 fb1a36c0 2022-01-09 op 
 568 fb1a36c0 2022-01-09 op 	tcreate(fid, argv[2], 0, mode);
 569 fb1a36c0 2022-01-09 op 	do_send();
 570 fb1a36c0 2022-01-09 op 	return;
 571 fb1a36c0 2022-01-09 op 
 572 fb1a36c0 2022-01-09 op usage:
 573 fb1a36c0 2022-01-09 op 	log_warnx("usage: create fid path perm mode ; perm is unused");
 574 fb1a36c0 2022-01-09 op }
 575 fb1a36c0 2022-01-09 op 
 576 fb1a36c0 2022-01-09 op 
 577 fb1a36c0 2022-01-09 op /* read fid offset count */
 578 fb1a36c0 2022-01-09 op static void
 579 fb1a36c0 2022-01-09 op excmd_read(const char **argv, int argc)
 580 fb1a36c0 2022-01-09 op {
 581 fb1a36c0 2022-01-09 op 	uint64_t	 off;
 582 fb1a36c0 2022-01-09 op 	uint32_t	 fid, count;
 583 fb1a36c0 2022-01-09 op 	const char	*errstr;
 584 fb1a36c0 2022-01-09 op 
 585 fb1a36c0 2022-01-09 op 	if (argc != 4)
 586 fb1a36c0 2022-01-09 op 		goto usage;
 587 fb1a36c0 2022-01-09 op 
 588 fb1a36c0 2022-01-09 op 	fid = strtonum(argv[1], 0, UINT32_MAX, &errstr);
 589 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 590 fb1a36c0 2022-01-09 op 		log_warnx("fid is %s: %s", errstr, argv[1]);
 591 fb1a36c0 2022-01-09 op 		return;
 592 fb1a36c0 2022-01-09 op 	}
 593 fb1a36c0 2022-01-09 op 
 594 fb1a36c0 2022-01-09 op 	/* should really be UNT64_MAX but it fails... */
 595 fb1a36c0 2022-01-09 op 	off = strtonum(argv[2], -1, UINT32_MAX, &errstr);
 596 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 597 fb1a36c0 2022-01-09 op 		log_warnx("offset is %s: %s", errstr, argv[2]);
 598 fb1a36c0 2022-01-09 op 		return;
 599 fb1a36c0 2022-01-09 op 	}
 600 fb1a36c0 2022-01-09 op 
 601 fb1a36c0 2022-01-09 op 	count = strtonum(argv[3], 0, UINT32_MAX, &errstr);
 602 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 603 fb1a36c0 2022-01-09 op 		log_warnx("count is %s: %s", errstr, argv[3]);
 604 fb1a36c0 2022-01-09 op 		return;
 605 fb1a36c0 2022-01-09 op 	}
 606 fb1a36c0 2022-01-09 op 
 607 fb1a36c0 2022-01-09 op 	tread(fid, off, count);
 608 fb1a36c0 2022-01-09 op 	do_send();
 609 fb1a36c0 2022-01-09 op 	return;
 610 fb1a36c0 2022-01-09 op 
 611 fb1a36c0 2022-01-09 op usage:
 612 fb1a36c0 2022-01-09 op 	log_warnx("usage: read fid offset count");
 613 fb1a36c0 2022-01-09 op }
 614 fb1a36c0 2022-01-09 op 
 615 fb1a36c0 2022-01-09 op /* write fid offset content */
 616 fb1a36c0 2022-01-09 op static void
 617 fb1a36c0 2022-01-09 op excmd_write(const char **argv, int argc)
 618 fb1a36c0 2022-01-09 op {
 619 fb1a36c0 2022-01-09 op 	uint64_t	 off;
 620 fb1a36c0 2022-01-09 op 	uint32_t	 fid, count;
 621 fb1a36c0 2022-01-09 op 	const char	*errstr;
 622 fb1a36c0 2022-01-09 op 
 623 fb1a36c0 2022-01-09 op 	if (argc != 4)
 624 fb1a36c0 2022-01-09 op 		goto usage;
 625 fb1a36c0 2022-01-09 op 
 626 fb1a36c0 2022-01-09 op 	fid = strtonum(argv[1], 0, UINT32_MAX, &errstr);
 627 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 628 fb1a36c0 2022-01-09 op 		log_warnx("fid is %s: %s", errstr, argv[1]);
 629 fb1a36c0 2022-01-09 op 		return;
 630 fb1a36c0 2022-01-09 op 	}
 631 fb1a36c0 2022-01-09 op 
 632 fb1a36c0 2022-01-09 op 	/* should really be UINT64_MAX but... */
 633 fb1a36c0 2022-01-09 op 	off = strtonum(argv[2], 0, UINT32_MAX, &errstr);
 634 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 635 fb1a36c0 2022-01-09 op 		log_warnx("offset is %s: %s", errstr, argv[2]);
 636 fb1a36c0 2022-01-09 op 		return;
 637 fb1a36c0 2022-01-09 op 	}
 638 fb1a36c0 2022-01-09 op 
 639 fb1a36c0 2022-01-09 op 	count = strlen(argv[3]);
 640 fb1a36c0 2022-01-09 op 	twrite(fid, off, argv[3], count);
 641 fb1a36c0 2022-01-09 op 	do_send();
 642 fb1a36c0 2022-01-09 op 	return;
 643 fb1a36c0 2022-01-09 op 
 644 fb1a36c0 2022-01-09 op usage:
 645 fb1a36c0 2022-01-09 op 	log_warnx("usage: write fid offset content");
 646 fb1a36c0 2022-01-09 op }
 647 fb1a36c0 2022-01-09 op 
 648 fb1a36c0 2022-01-09 op /* remove fid */
 649 fb1a36c0 2022-01-09 op static void
 650 fb1a36c0 2022-01-09 op excmd_remove(const char **argv, int argc)
 651 fb1a36c0 2022-01-09 op {
 652 fb1a36c0 2022-01-09 op 	const char	*errstr;
 653 fb1a36c0 2022-01-09 op 	uint32_t	 fid;
 654 fb1a36c0 2022-01-09 op 
 655 fb1a36c0 2022-01-09 op 	if (argc != 2)
 656 fb1a36c0 2022-01-09 op 		goto usage;
 657 fb1a36c0 2022-01-09 op 
 658 fb1a36c0 2022-01-09 op 	fid = strtonum(argv[1], 0, UINT32_MAX, &errstr);
 659 fb1a36c0 2022-01-09 op 	if (errstr != NULL) {
 660 fb1a36c0 2022-01-09 op 		log_warnx("fid is %s: %s", errstr, argv[1]);
 661 fb1a36c0 2022-01-09 op 		return;
 662 fb1a36c0 2022-01-09 op 	}
 663 fb1a36c0 2022-01-09 op 
 664 fb1a36c0 2022-01-09 op 	tremove(fid);
 665 fb1a36c0 2022-01-09 op 	do_send();
 666 fb1a36c0 2022-01-09 op 	return;
 667 fb1a36c0 2022-01-09 op 
 668 fb1a36c0 2022-01-09 op usage:
 669 fb1a36c0 2022-01-09 op 	log_warnx("usage: remove fid");
 670 fb1a36c0 2022-01-09 op }
 671 fb1a36c0 2022-01-09 op 
 672 fb1a36c0 2022-01-09 op static void
 673 fb1a36c0 2022-01-09 op excmd(const char **argv, int argc)
 674 fb1a36c0 2022-01-09 op {
 675 fb1a36c0 2022-01-09 op 	struct cmd {
 676 fb1a36c0 2022-01-09 op 		const char	*name;
 677 fb1a36c0 2022-01-09 op 		void		(*fn)(const char **, int);
 678 fb1a36c0 2022-01-09 op 	} cmds[] = {
 679 fb1a36c0 2022-01-09 op 		{"version",	excmd_version},
 680 fb1a36c0 2022-01-09 op 		{"attach",	excmd_attach},
 681 fb1a36c0 2022-01-09 op 		{"clunk",	excmd_clunk},
 682 fb1a36c0 2022-01-09 op 		{"flush",	excmd_flush},
 683 fb1a36c0 2022-01-09 op 		{"walk",	excmd_walk},
 684 fb1a36c0 2022-01-09 op 		{"open",	excmd_open},
 685 fb1a36c0 2022-01-09 op 		{"create",	excmd_create},
 686 fb1a36c0 2022-01-09 op 		{"read",	excmd_read},
 687 fb1a36c0 2022-01-09 op 		{"write",	excmd_write},
 688 fb1a36c0 2022-01-09 op 		/* TODO: stat */
 689 fb1a36c0 2022-01-09 op 		{"remove",	excmd_remove},
 690 fb1a36c0 2022-01-09 op 	};
 691 fb1a36c0 2022-01-09 op 	size_t i;
 692 fb1a36c0 2022-01-09 op 
 693 fb1a36c0 2022-01-09 op 	if (argc == 0)
 694 fb1a36c0 2022-01-09 op 		return;
 695 fb1a36c0 2022-01-09 op 
 696 fb1a36c0 2022-01-09 op 	for (i = 0; i < sizeof(cmds)/sizeof(cmds[0]); ++i) {
 697 fb1a36c0 2022-01-09 op 		if (!strcmp(cmds[i].name, argv[0])) {
 698 fb1a36c0 2022-01-09 op 			cmds[i].fn(argv, argc);
 699 fb1a36c0 2022-01-09 op 			return;
 700 fb1a36c0 2022-01-09 op 		}
 701 fb1a36c0 2022-01-09 op 	}
 702 fb1a36c0 2022-01-09 op 
 703 fb1a36c0 2022-01-09 op 	log_warnx("Unknown command %s", *argv);
 704 fb1a36c0 2022-01-09 op }
 705 fb1a36c0 2022-01-09 op 
 706 fb1a36c0 2022-01-09 op static void
 707 fb1a36c0 2022-01-09 op pp_qid(const uint8_t *d, uint32_t len)
 708 fb1a36c0 2022-01-09 op {
 709 fb1a36c0 2022-01-09 op 	uint64_t	path;
 710 fb1a36c0 2022-01-09 op 	uint32_t	vers;
 711 fb1a36c0 2022-01-09 op 	uint8_t		type;
 712 fb1a36c0 2022-01-09 op 
 713 fb1a36c0 2022-01-09 op 	if (len < 13) {
 714 fb1a36c0 2022-01-09 op 		printf("invalid");
 715 fb1a36c0 2022-01-09 op 		return;
 716 fb1a36c0 2022-01-09 op 	}
 717 fb1a36c0 2022-01-09 op 
 718 fb1a36c0 2022-01-09 op 	type = *d++;
 719 fb1a36c0 2022-01-09 op 
 720 fb1a36c0 2022-01-09 op 	memcpy(&vers, d, sizeof(vers));
 721 fb1a36c0 2022-01-09 op 	d += sizeof(vers);
 722 fb1a36c0 2022-01-09 op 	vers = le64toh(vers);
 723 fb1a36c0 2022-01-09 op 
 724 fb1a36c0 2022-01-09 op 	memcpy(&path, d, sizeof(path));
 725 fb1a36c0 2022-01-09 op 	d += sizeof(path);
 726 fb1a36c0 2022-01-09 op 	path = le64toh(path);
 727 fb1a36c0 2022-01-09 op 
 728 fb1a36c0 2022-01-09 op 	printf("qid{path=%"PRIu64" version=%"PRIu32" type=0x%x\"%s\"}",
 729 fb1a36c0 2022-01-09 op 	    path, vers, type, pp_qid_type(type));
 730 fb1a36c0 2022-01-09 op }
 731 fb1a36c0 2022-01-09 op 
 732 fb1a36c0 2022-01-09 op static void
 733 fb1a36c0 2022-01-09 op pp_msg(uint32_t len, uint8_t type, uint16_t tag, const uint8_t *d)
 734 fb1a36c0 2022-01-09 op {
 735 fb1a36c0 2022-01-09 op 	uint32_t	 msize, iounit, count;
 736 fb1a36c0 2022-01-09 op 	uint16_t	 slen;
 737 fb1a36c0 2022-01-09 op 	char		*v;
 738 fb1a36c0 2022-01-09 op 
 739 fb1a36c0 2022-01-09 op 	printf("len=%"PRIu32" type=%d[%s] tag=0x%x[%d] ", len,
 740 fb1a36c0 2022-01-09 op 	    type, pp_msg_type(type), tag, tag);
 741 fb1a36c0 2022-01-09 op 
 742 fb1a36c0 2022-01-09 op 	len -= HEADERSIZE;
 743 fb1a36c0 2022-01-09 op 
 744 fb1a36c0 2022-01-09 op 	switch (type) {
 745 fb1a36c0 2022-01-09 op 	case Rversion:
 746 fb1a36c0 2022-01-09 op 		if (len < 6) {
 747 fb1a36c0 2022-01-09 op 			printf("invalid: not enough space for msize "
 748 fb1a36c0 2022-01-09 op 			    "and version provided.");
 749 fb1a36c0 2022-01-09 op 			break;
 750 fb1a36c0 2022-01-09 op 		}
 751 fb1a36c0 2022-01-09 op 
 752 fb1a36c0 2022-01-09 op 		memcpy(&msize, d, sizeof(msize));
 753 fb1a36c0 2022-01-09 op 		d += sizeof(msize);
 754 fb1a36c0 2022-01-09 op 		len -= sizeof(msize);
 755 fb1a36c0 2022-01-09 op 		msize = le32toh(msize);
 756 fb1a36c0 2022-01-09 op 
 757 fb1a36c0 2022-01-09 op 		memcpy(&slen, d, sizeof(slen));
 758 fb1a36c0 2022-01-09 op 		d += sizeof(slen);
 759 fb1a36c0 2022-01-09 op 		len -= sizeof(slen);
 760 fb1a36c0 2022-01-09 op 		slen = le16toh(slen);
 761 fb1a36c0 2022-01-09 op 
 762 fb1a36c0 2022-01-09 op 		if (len != slen) {
 763 fb1a36c0 2022-01-09 op 			printf("invalid: version string length doesn't "
 764 fb1a36c0 2022-01-09 op 			    "match.  Got %d; want %d", slen, len);
 765 fb1a36c0 2022-01-09 op 			break;
 766 fb1a36c0 2022-01-09 op 		}
 767 fb1a36c0 2022-01-09 op 
 768 fb1a36c0 2022-01-09 op 		printf("msize=%"PRIu32" version[%"PRIu16"]=\"",
 769 fb1a36c0 2022-01-09 op 		    msize, slen);
 770 fb1a36c0 2022-01-09 op 		fwrite(d, 1, slen, stdout);
 771 fb1a36c0 2022-01-09 op 		printf("\"");
 772 fb1a36c0 2022-01-09 op 
 773 fb1a36c0 2022-01-09 op 		break;
 774 fb1a36c0 2022-01-09 op 
 775 fb1a36c0 2022-01-09 op 	case Rattach:
 776 fb1a36c0 2022-01-09 op 		pp_qid(d, len);
 777 fb1a36c0 2022-01-09 op 		break;
 778 fb1a36c0 2022-01-09 op 
 779 fb1a36c0 2022-01-09 op 	case Rclunk:
 780 fb1a36c0 2022-01-09 op 	case Rflush:
 781 fb1a36c0 2022-01-09 op 	case Rremove:
 782 fb1a36c0 2022-01-09 op 		if (len != 0)
 783 fb1a36c0 2022-01-09 op 			printf("invalid %s: %"PRIu32" extra bytes", 
 784 fb1a36c0 2022-01-09 op 			    pp_msg_type(type), len);
 785 fb1a36c0 2022-01-09 op 		break;
 786 fb1a36c0 2022-01-09 op 
 787 fb1a36c0 2022-01-09 op 	case Rwalk:
 788 fb1a36c0 2022-01-09 op 		if (len < 2) {
 789 fb1a36c0 2022-01-09 op 			printf("invaild Rwalk: less than two bytes (%d)",
 790 fb1a36c0 2022-01-09 op 			    (int)len);
 791 fb1a36c0 2022-01-09 op 			break;
 792 fb1a36c0 2022-01-09 op 		}
 793 fb1a36c0 2022-01-09 op 
 794 fb1a36c0 2022-01-09 op 		memcpy(&slen, d, sizeof(slen));
 795 fb1a36c0 2022-01-09 op 		d += sizeof(slen);
 796 fb1a36c0 2022-01-09 op 		len -= sizeof(slen);
 797 fb1a36c0 2022-01-09 op 		slen = le16toh(slen);
 798 fb1a36c0 2022-01-09 op 
 799 fb1a36c0 2022-01-09 op 		if (len != QIDSIZE * slen) {
 800 fb1a36c0 2022-01-09 op 			printf("invalid Rwalk: wanted %d bytes for %d qids "
 801 fb1a36c0 2022-01-09 op 			    "but got %"PRIu32" bytes instead",
 802 fb1a36c0 2022-01-09 op 			    QIDSIZE*slen, slen, len);
 803 fb1a36c0 2022-01-09 op 			break;
 804 fb1a36c0 2022-01-09 op 		}
 805 fb1a36c0 2022-01-09 op 
 806 fb1a36c0 2022-01-09 op 		printf("nwqid=%"PRIu16, slen);
 807 fb1a36c0 2022-01-09 op 
 808 fb1a36c0 2022-01-09 op 		for (; slen != 0; slen--) {
 809 fb1a36c0 2022-01-09 op 			printf(" ");
 810 fb1a36c0 2022-01-09 op 			pp_qid(d, len);
 811 fb1a36c0 2022-01-09 op 			d += QIDSIZE;
 812 fb1a36c0 2022-01-09 op 			len -= QIDSIZE;
 813 fb1a36c0 2022-01-09 op 		}
 814 fb1a36c0 2022-01-09 op 
 815 fb1a36c0 2022-01-09 op 		break;
 816 fb1a36c0 2022-01-09 op 
 817 fb1a36c0 2022-01-09 op 	case Ropen:
 818 fb1a36c0 2022-01-09 op 	case Rcreate:
 819 fb1a36c0 2022-01-09 op 		if (len != QIDSIZE + 4) {
 820 fb1a36c0 2022-01-09 op 			printf("invalid %s: expected %d bytes; "
 821 fb1a36c0 2022-01-09 op 			    "got %u\n", pp_msg_type(type), QIDSIZE + 4, len);
 822 fb1a36c0 2022-01-09 op 			break;
 823 fb1a36c0 2022-01-09 op 		}
 824 fb1a36c0 2022-01-09 op 
 825 fb1a36c0 2022-01-09 op 		pp_qid(d, len);
 826 fb1a36c0 2022-01-09 op 		d += QIDSIZE;
 827 fb1a36c0 2022-01-09 op 		len -= QIDSIZE;
 828 fb1a36c0 2022-01-09 op 
 829 fb1a36c0 2022-01-09 op 		memcpy(&iounit, d, sizeof(iounit));
 830 fb1a36c0 2022-01-09 op 		d += sizeof(iounit);
 831 fb1a36c0 2022-01-09 op 		len -= sizeof(iounit);
 832 fb1a36c0 2022-01-09 op 		iounit = le32toh(iounit);
 833 fb1a36c0 2022-01-09 op 		printf(" iounit=%"PRIu32, iounit);
 834 fb1a36c0 2022-01-09 op 		break;
 835 fb1a36c0 2022-01-09 op 
 836 fb1a36c0 2022-01-09 op 	case Rread:
 837 fb1a36c0 2022-01-09 op 		if (len < sizeof(count)) {
 838 fb1a36c0 2022-01-09 op 			printf("invalid Rread: expected %zu bytes at least; "
 839 fb1a36c0 2022-01-09 op 			    "got %u\n", sizeof(count), len);
 840 fb1a36c0 2022-01-09 op 			break;
 841 fb1a36c0 2022-01-09 op 		}
 842 fb1a36c0 2022-01-09 op 
 843 fb1a36c0 2022-01-09 op 		memcpy(&count, d, sizeof(count));
 844 fb1a36c0 2022-01-09 op 		d += sizeof(count);
 845 fb1a36c0 2022-01-09 op 		len -= sizeof(count);
 846 fb1a36c0 2022-01-09 op 		count = le32toh(count);
 847 fb1a36c0 2022-01-09 op 
 848 fb1a36c0 2022-01-09 op 		if (len != count) {
 849 fb1a36c0 2022-01-09 op 			printf("invalid Rread: expected %d data bytes; "
 850 fb1a36c0 2022-01-09 op 			    "got %u\n", count, len);
 851 fb1a36c0 2022-01-09 op 			break;
 852 fb1a36c0 2022-01-09 op 		}
 853 fb1a36c0 2022-01-09 op 
 854 fb1a36c0 2022-01-09 op 		/* allocates three extra bytes, oh well... */
 855 fb1a36c0 2022-01-09 op 		if ((v = calloc(count + 1, 4)) == NULL)
 856 fb1a36c0 2022-01-09 op 			fatal("calloc");
 857 fb1a36c0 2022-01-09 op 		strvisx(v, d, count, VIS_SAFE | VIS_TAB | VIS_NL | VIS_CSTYLE);
 858 fb1a36c0 2022-01-09 op 		printf("data=%s", v);
 859 fb1a36c0 2022-01-09 op 		free(v);
 860 fb1a36c0 2022-01-09 op 
 861 fb1a36c0 2022-01-09 op 		break;
 862 fb1a36c0 2022-01-09 op 
 863 fb1a36c0 2022-01-09 op 	case Rwrite:
 864 fb1a36c0 2022-01-09 op 		if (len != sizeof(count)) {
 865 fb1a36c0 2022-01-09 op 			printf("invalid Rwrite: expected %zu data bytes; "
 866 fb1a36c0 2022-01-09 op 			    "got %u\n", sizeof(count), len);
 867 fb1a36c0 2022-01-09 op 			break;
 868 fb1a36c0 2022-01-09 op 		}
 869 fb1a36c0 2022-01-09 op 
 870 fb1a36c0 2022-01-09 op 		memcpy(&count, d, sizeof(count));
 871 fb1a36c0 2022-01-09 op 		d += sizeof(count);
 872 fb1a36c0 2022-01-09 op 		len -= sizeof(count);
 873 fb1a36c0 2022-01-09 op 		count = le32toh(count);
 874 fb1a36c0 2022-01-09 op 
 875 fb1a36c0 2022-01-09 op 		printf("count=%d", count);
 876 fb1a36c0 2022-01-09 op 		break;
 877 fb1a36c0 2022-01-09 op 
 878 fb1a36c0 2022-01-09 op 	case Rerror:
 879 fb1a36c0 2022-01-09 op 		memcpy(&slen, d, sizeof(slen));
 880 fb1a36c0 2022-01-09 op 		d += sizeof(slen);
 881 fb1a36c0 2022-01-09 op 		len -= sizeof(slen);
 882 fb1a36c0 2022-01-09 op 		slen = le16toh(slen);
 883 fb1a36c0 2022-01-09 op 
 884 fb1a36c0 2022-01-09 op 		if (slen != len) {
 885 fb1a36c0 2022-01-09 op 			printf("invalid: error string length doesn't "
 886 fb1a36c0 2022-01-09 op 			    "match.  Got %d; want %d", slen, len);
 887 fb1a36c0 2022-01-09 op 			break;
 888 fb1a36c0 2022-01-09 op 		}
 889 fb1a36c0 2022-01-09 op 
 890 fb1a36c0 2022-01-09 op 		printf("error=\"");
 891 fb1a36c0 2022-01-09 op 		fwrite(d, 1, slen, stdout);
 892 fb1a36c0 2022-01-09 op 		printf("\"");
 893 fb1a36c0 2022-01-09 op 
 894 fb1a36c0 2022-01-09 op 		break;
 895 fb1a36c0 2022-01-09 op 
 896 fb1a36c0 2022-01-09 op 	default:
 897 fb1a36c0 2022-01-09 op 		if ((v = calloc(len + 1, 4)) == NULL)
 898 fb1a36c0 2022-01-09 op 			fatal("calloc");
 899 fb1a36c0 2022-01-09 op 		strvisx(v, d, len, VIS_SAFE | VIS_TAB | VIS_NL | VIS_CSTYLE);
 900 fb1a36c0 2022-01-09 op 		printf("body=%s", v);
 901 fb1a36c0 2022-01-09 op 		free(v);
 902 fb1a36c0 2022-01-09 op 	}
 903 fb1a36c0 2022-01-09 op 
 904 fb1a36c0 2022-01-09 op 	printf("\n");
 905 fb1a36c0 2022-01-09 op }
 906 fb1a36c0 2022-01-09 op 
 907 fb1a36c0 2022-01-09 op static void
 908 fb1a36c0 2022-01-09 op handle_9p(const uint8_t *data, size_t size)
 909 fb1a36c0 2022-01-09 op {
 910 fb1a36c0 2022-01-09 op         uint32_t len;
 911 fb1a36c0 2022-01-09 op 	uint16_t tag;
 912 fb1a36c0 2022-01-09 op 	uint8_t type;
 913 fb1a36c0 2022-01-09 op 
 914 fb1a36c0 2022-01-09 op 	assert(size >= HEADERSIZE);
 915 fb1a36c0 2022-01-09 op 
 916 fb1a36c0 2022-01-09 op 	memcpy(&len, data, sizeof(len));
 917 fb1a36c0 2022-01-09 op 	data += sizeof(len);
 918 fb1a36c0 2022-01-09 op 
 919 fb1a36c0 2022-01-09 op 	memcpy(&type, data, sizeof(type));
 920 fb1a36c0 2022-01-09 op 	data += sizeof(type);
 921 fb1a36c0 2022-01-09 op 
 922 fb1a36c0 2022-01-09 op 	memcpy(&tag, data, sizeof(tag));
 923 fb1a36c0 2022-01-09 op 	data += sizeof(tag);
 924 fb1a36c0 2022-01-09 op 
 925 fb1a36c0 2022-01-09 op 	len = le32toh(len);
 926 fb1a36c0 2022-01-09 op 	/* type is one byte long, no endianness issues */
 927 fb1a36c0 2022-01-09 op 	tag = le16toh(tag);
 928 fb1a36c0 2022-01-09 op 
 929 fb1a36c0 2022-01-09 op 	clr();
 930 fb1a36c0 2022-01-09 op 	pp_msg(len, type, tag, data);
 931 fb1a36c0 2022-01-09 op 	prompt();
 932 fb1a36c0 2022-01-09 op }
 933 fb1a36c0 2022-01-09 op 
 934 fb1a36c0 2022-01-09 op static void
 935 fb1a36c0 2022-01-09 op clr(void)
 936 fb1a36c0 2022-01-09 op {
 937 fb1a36c0 2022-01-09 op 	printf("\r");
 938 fb1a36c0 2022-01-09 op 	fflush(stdout);
 939 fb1a36c0 2022-01-09 op }
 940 fb1a36c0 2022-01-09 op 
 941 fb1a36c0 2022-01-09 op static void
 942 fb1a36c0 2022-01-09 op prompt(void)
 943 fb1a36c0 2022-01-09 op {
 944 fb1a36c0 2022-01-09 op 	printf("%s", PROMPT);
 945 fb1a36c0 2022-01-09 op 	fflush(stdout);
 946 fb1a36c0 2022-01-09 op }
 947 fb1a36c0 2022-01-09 op 
 948 fb1a36c0 2022-01-09 op int
 949 fb1a36c0 2022-01-09 op main(int argc, char **argv)
 950 fb1a36c0 2022-01-09 op {
 951 fb1a36c0 2022-01-09 op 	int		 ch, sock, handshake;
 952 fb1a36c0 2022-01-09 op 	struct event	 ev_sigint, ev_sigterm;
 953 fb1a36c0 2022-01-09 op 
 954 fb1a36c0 2022-01-09 op 	signal(SIGPIPE, SIG_IGN);
 955 fb1a36c0 2022-01-09 op 
 956 fb1a36c0 2022-01-09 op 	while ((ch = getopt(argc, argv, "C:cH:hK:P:v")) != -1) {
 957 fb1a36c0 2022-01-09 op 		switch (ch) {
 958 fb1a36c0 2022-01-09 op 		case 'C':
 959 fb1a36c0 2022-01-09 op 			crtpath = optarg;
 960 fb1a36c0 2022-01-09 op 			break;
 961 fb1a36c0 2022-01-09 op 		case 'c':
 962 fb1a36c0 2022-01-09 op 			tls = 1;
 963 fb1a36c0 2022-01-09 op 			break;
 964 fb1a36c0 2022-01-09 op 		case 'H':
 965 fb1a36c0 2022-01-09 op 			host = optarg;
 966 fb1a36c0 2022-01-09 op 			break;
 967 fb1a36c0 2022-01-09 op 		case 'h':
 968 fb1a36c0 2022-01-09 op 			usage(0);
 969 fb1a36c0 2022-01-09 op 			break;
 970 fb1a36c0 2022-01-09 op 		case 'K':
 971 fb1a36c0 2022-01-09 op 			keypath = optarg;
 972 fb1a36c0 2022-01-09 op 			break;
 973 fb1a36c0 2022-01-09 op 		case 'P':
 974 fb1a36c0 2022-01-09 op 			port = optarg;
 975 fb1a36c0 2022-01-09 op 			break;
 976 fb1a36c0 2022-01-09 op 		case 'v':
 977 fb1a36c0 2022-01-09 op 			verbose = 1;
 978 fb1a36c0 2022-01-09 op 			break;
 979 fb1a36c0 2022-01-09 op 		default:
 980 fb1a36c0 2022-01-09 op 			usage(1);
 981 fb1a36c0 2022-01-09 op 		}
 982 fb1a36c0 2022-01-09 op 	}
 983 fb1a36c0 2022-01-09 op 
 984 fb1a36c0 2022-01-09 op 	if (host == NULL)
 985 fb1a36c0 2022-01-09 op 		host = "localhost";
 986 fb1a36c0 2022-01-09 op 	if (port == NULL)
 987 fb1a36c0 2022-01-09 op 		port = "1337";
 988 fb1a36c0 2022-01-09 op 
 989 fb1a36c0 2022-01-09 op 	argc -= optind;
 990 fb1a36c0 2022-01-09 op 	argv += optind;
 991 fb1a36c0 2022-01-09 op 
 992 fb1a36c0 2022-01-09 op 	if (argc != 0)
 993 fb1a36c0 2022-01-09 op 		usage(1);
 994 fb1a36c0 2022-01-09 op 	/* if (!tls || (crtpath != NULL || keypath != NULL)) */
 995 fb1a36c0 2022-01-09 op 		/* usage(1); */
 996 fb1a36c0 2022-01-09 op 	if (!tls)
 997 fb1a36c0 2022-01-09 op                 errx(1, "must enable tls (for now)");
 998 fb1a36c0 2022-01-09 op 
 999 fb1a36c0 2022-01-09 op 	log_init(1, LOG_DAEMON);
1000 fb1a36c0 2022-01-09 op 	log_setverbose(verbose);
1001 fb1a36c0 2022-01-09 op 	log_procinit(getprogname());
1002 fb1a36c0 2022-01-09 op 
1003 fb1a36c0 2022-01-09 op 	if ((tlsconf = tls_config_new()) == NULL)
1004 fb1a36c0 2022-01-09 op 		fatalx("tls_config_new");
1005 fb1a36c0 2022-01-09 op 	tls_config_insecure_noverifycert(tlsconf);
1006 fb1a36c0 2022-01-09 op 	tls_config_insecure_noverifyname(tlsconf);
1007 fb1a36c0 2022-01-09 op 	if (tls_config_set_keypair_file(tlsconf, crtpath, keypath) == -1)
1008 fb1a36c0 2022-01-09 op 		fatalx("can't load certs (%s, %s)", crtpath, keypath);
1009 fb1a36c0 2022-01-09 op 
1010 fb1a36c0 2022-01-09 op 	if ((ctx = tls_client()) == NULL)
1011 fb1a36c0 2022-01-09 op 		fatal("tls_client");
1012 fb1a36c0 2022-01-09 op 	if (tls_configure(ctx, tlsconf) == -1)
1013 fb1a36c0 2022-01-09 op 		fatalx("tls_configure: %s", tls_error(ctx));
1014 fb1a36c0 2022-01-09 op 
1015 fb1a36c0 2022-01-09 op 	log_info("connecting to %s:%s...", host, port);
1016 fb1a36c0 2022-01-09 op 
1017 fb1a36c0 2022-01-09 op 	if ((sock = openconn()) == -1)
1018 fb1a36c0 2022-01-09 op 		fatalx("can't connect to %s:%s", host, port);
1019 fb1a36c0 2022-01-09 op 
1020 fb1a36c0 2022-01-09 op 	if (tls_connect_socket(ctx, sock, host) == -1)
1021 fb1a36c0 2022-01-09 op 		fatalx("tls_connect_socket: %s", tls_error(ctx));
1022 fb1a36c0 2022-01-09 op 
1023 fb1a36c0 2022-01-09 op 	for (handshake = 0; !handshake;) {
1024 fb1a36c0 2022-01-09 op 		switch (tls_handshake(ctx)) {
1025 fb1a36c0 2022-01-09 op 		case -1:
1026 fb1a36c0 2022-01-09 op 			fatalx("tls_handshake: %s", tls_error(ctx));
1027 fb1a36c0 2022-01-09 op 		case 0:
1028 fb1a36c0 2022-01-09 op 			handshake = 1;
1029 fb1a36c0 2022-01-09 op 			break;
1030 fb1a36c0 2022-01-09 op 		}
1031 fb1a36c0 2022-01-09 op 	}
1032 fb1a36c0 2022-01-09 op 
1033 fb1a36c0 2022-01-09 op 	log_info("connected!");
1034 fb1a36c0 2022-01-09 op 
1035 fb1a36c0 2022-01-09 op 	mark_nonblock(sock);
1036 fb1a36c0 2022-01-09 op 
1037 fb1a36c0 2022-01-09 op 	event_init();
1038 fb1a36c0 2022-01-09 op 
1039 fb1a36c0 2022-01-09 op 	/* initialize global evb */
1040 fb1a36c0 2022-01-09 op 	if ((evb = evbuffer_new()) == NULL)
1041 fb1a36c0 2022-01-09 op 		fatal("evbuffer_new");
1042 fb1a36c0 2022-01-09 op 
1043 fb1a36c0 2022-01-09 op 	signal_set(&ev_sigint, SIGINT, sig_handler, NULL);
1044 fb1a36c0 2022-01-09 op 	signal_set(&ev_sigterm, SIGINT, sig_handler, NULL);
1045 fb1a36c0 2022-01-09 op 
1046 fb1a36c0 2022-01-09 op 	signal_add(&ev_sigint, NULL);
1047 fb1a36c0 2022-01-09 op 	signal_add(&ev_sigterm, NULL);
1048 fb1a36c0 2022-01-09 op 
1049 fb1a36c0 2022-01-09 op 	bev = bufferevent_new(sock, client_read, client_write, client_error,
1050 fb1a36c0 2022-01-09 op 	    NULL);
1051 fb1a36c0 2022-01-09 op 	if (bev == NULL)
1052 fb1a36c0 2022-01-09 op 		fatal("bufferevent_new");
1053 fb1a36c0 2022-01-09 op 
1054 fb1a36c0 2022-01-09 op 	/* setup tls/io */
1055 fb1a36c0 2022-01-09 op 	event_set(&bev->ev_read, sock, EV_READ, tls_readcb, bev);
1056 fb1a36c0 2022-01-09 op 	event_set(&bev->ev_write, sock, EV_WRITE, tls_writecb, bev);
1057 fb1a36c0 2022-01-09 op 
1058 fb1a36c0 2022-01-09 op 	bufferevent_enable(bev, EV_READ|EV_WRITE);
1059 fb1a36c0 2022-01-09 op 
1060 fb1a36c0 2022-01-09 op 	mark_nonblock(0);
1061 fb1a36c0 2022-01-09 op 	inbev = bufferevent_new(0, repl_read, NULL, repl_error, NULL);
1062 fb1a36c0 2022-01-09 op 	bufferevent_enable(inbev, EV_READ);
1063 fb1a36c0 2022-01-09 op 
1064 fb1a36c0 2022-01-09 op 	prompt();
1065 fb1a36c0 2022-01-09 op 	event_dispatch();
1066 fb1a36c0 2022-01-09 op 
1067 fb1a36c0 2022-01-09 op 	bufferevent_free(bev);
1068 fb1a36c0 2022-01-09 op 	tls_free(ctx);
1069 fb1a36c0 2022-01-09 op 	tls_config_free(tlsconf);
1070 fb1a36c0 2022-01-09 op 	close(sock);
1071 fb1a36c0 2022-01-09 op 
1072 fb1a36c0 2022-01-09 op 	return 0;
1073 fb1a36c0 2022-01-09 op }