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