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