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 fb1a36c0 2022-01-09 op #include <sys/ioctl.h>
  18 fb1a36c0 2022-01-09 op #include <sys/types.h>
  19 fb1a36c0 2022-01-09 op #include <sys/socket.h>
  20 fb1a36c0 2022-01-09 op 
  21 fb1a36c0 2022-01-09 op #include <assert.h>
  22 fb1a36c0 2022-01-09 op #include <err.h>
  23 fb1a36c0 2022-01-09 op #include <errno.h>
  24 fb1a36c0 2022-01-09 op #include <event.h>
  25 fb1a36c0 2022-01-09 op #include <fcntl.h>
  26 fb1a36c0 2022-01-09 op #include <inttypes.h>
  27 fb1a36c0 2022-01-09 op #include <netdb.h>
  28 fb1a36c0 2022-01-09 op #include <limits.h>
  29 fb1a36c0 2022-01-09 op #include <signal.h>
  30 fb1a36c0 2022-01-09 op #include <stdio.h>
  31 fb1a36c0 2022-01-09 op #include <stdlib.h>
  32 fb1a36c0 2022-01-09 op #include <string.h>
  33 fb1a36c0 2022-01-09 op #include <syslog.h>
  34 fb1a36c0 2022-01-09 op #include <tls.h>
  35 fb1a36c0 2022-01-09 op #include <unistd.h>
  36 fb1a36c0 2022-01-09 op #include <util.h>
  37 fb1a36c0 2022-01-09 op 
  38 fb1a36c0 2022-01-09 op #if HAVE_LIBREADLINE
  39 fb1a36c0 2022-01-09 op #include <readline/readline.h>
  40 fb1a36c0 2022-01-09 op #include <readline/history.h>
  41 fb1a36c0 2022-01-09 op #endif
  42 fb1a36c0 2022-01-09 op 
  43 fb1a36c0 2022-01-09 op #ifndef nitems
  44 fb1a36c0 2022-01-09 op #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
  45 fb1a36c0 2022-01-09 op #endif
  46 fb1a36c0 2022-01-09 op 
  47 fb1a36c0 2022-01-09 op #include "9pclib.h"
  48 fb1a36c0 2022-01-09 op #include "kami.h"
  49 fb1a36c0 2022-01-09 op #include "utils.h"
  50 fb1a36c0 2022-01-09 op #include "log.h"
  51 fb1a36c0 2022-01-09 op 
  52 fb1a36c0 2022-01-09 op /* flags */
  53 fb1a36c0 2022-01-09 op int		 tls;
  54 fb1a36c0 2022-01-09 op const char	*crtpath;
  55 fb1a36c0 2022-01-09 op const char	*keypath;
  56 fb1a36c0 2022-01-09 op 
  57 fb1a36c0 2022-01-09 op /* state */
  58 fb1a36c0 2022-01-09 op struct tls_config	*tlsconf;
  59 fb1a36c0 2022-01-09 op struct tls		*ctx;
  60 fb1a36c0 2022-01-09 op int			 sock;
  61 fb1a36c0 2022-01-09 op struct evbuffer		*buf;
  62 fb1a36c0 2022-01-09 op struct evbuffer		*dirbuf;
  63 fb1a36c0 2022-01-09 op uint32_t		 msize;
  64 fb1a36c0 2022-01-09 op int			 bell;
  65 fb1a36c0 2022-01-09 op 
  66 fb1a36c0 2022-01-09 op volatile sig_atomic_t	 resized;
  67 fb1a36c0 2022-01-09 op int			 tty_p;
  68 fb1a36c0 2022-01-09 op int			 tty_width;
  69 fb1a36c0 2022-01-09 op 
  70 fb1a36c0 2022-01-09 op struct np_stat {
  71 fb1a36c0 2022-01-09 op 	uint16_t	 type;
  72 fb1a36c0 2022-01-09 op 	uint32_t	 dev;
  73 fb1a36c0 2022-01-09 op 	struct qid	 qid;
  74 fb1a36c0 2022-01-09 op 	uint32_t	 mode;
  75 fb1a36c0 2022-01-09 op 	uint32_t	 atime;
  76 fb1a36c0 2022-01-09 op 	uint32_t	 mtime;
  77 fb1a36c0 2022-01-09 op 	uint64_t	 length;
  78 fb1a36c0 2022-01-09 op 	char		*name;
  79 fb1a36c0 2022-01-09 op 	char		*uid;
  80 fb1a36c0 2022-01-09 op 	char		*gid;
  81 fb1a36c0 2022-01-09 op 	char		*muid;
  82 fb1a36c0 2022-01-09 op };
  83 fb1a36c0 2022-01-09 op 
  84 fb1a36c0 2022-01-09 op struct progress {
  85 fb1a36c0 2022-01-09 op 	uint64_t	max;
  86 fb1a36c0 2022-01-09 op 	uint64_t	done;
  87 fb1a36c0 2022-01-09 op };
  88 fb1a36c0 2022-01-09 op 
  89 fb1a36c0 2022-01-09 op int pwdfid;
  90 fb1a36c0 2022-01-09 op 
  91 fb1a36c0 2022-01-09 op #define ASSERT_EMPTYBUF() assert(EVBUFFER_LENGTH(buf) == 0)
  92 fb1a36c0 2022-01-09 op 
  93 fb1a36c0 2022-01-09 op #if HAVE_LIBREADLINE
  94 fb1a36c0 2022-01-09 op static char *
  95 fb1a36c0 2022-01-09 op read_line(const char *prompt)
  96 fb1a36c0 2022-01-09 op {
  97 fb1a36c0 2022-01-09 op 	char *line;
  98 fb1a36c0 2022-01-09 op 
  99 fb1a36c0 2022-01-09 op again:
 100 fb1a36c0 2022-01-09 op 	if ((line = readline(prompt)) == NULL)
 101 fb1a36c0 2022-01-09 op 		return NULL;
 102 fb1a36c0 2022-01-09 op 	/* XXX: trim spaces? */
 103 fb1a36c0 2022-01-09 op 	if (*line == '\0') {
 104 fb1a36c0 2022-01-09 op 		free(line);
 105 fb1a36c0 2022-01-09 op 		goto again;
 106 fb1a36c0 2022-01-09 op 	}
 107 fb1a36c0 2022-01-09 op 
 108 fb1a36c0 2022-01-09 op 	add_history(line);
 109 fb1a36c0 2022-01-09 op 	return line;
 110 fb1a36c0 2022-01-09 op }
 111 fb1a36c0 2022-01-09 op #else
 112 fb1a36c0 2022-01-09 op static char *
 113 fb1a36c0 2022-01-09 op read_line(const char *prompt)
 114 fb1a36c0 2022-01-09 op {
 115 fb1a36c0 2022-01-09 op 	char *ch, *line = NULL;
 116 fb1a36c0 2022-01-09 op 	size_t linesize = 0;
 117 fb1a36c0 2022-01-09 op 	ssize_t linelen;
 118 fb1a36c0 2022-01-09 op 
 119 fb1a36c0 2022-01-09 op 	printf("%s", prompt);
 120 fb1a36c0 2022-01-09 op 	fflush(stdout);
 121 fb1a36c0 2022-01-09 op 
 122 fb1a36c0 2022-01-09 op 	linelen = getline(&line, &linesize, stdin);
 123 fb1a36c0 2022-01-09 op 	if (linelen == -1)
 124 fb1a36c0 2022-01-09 op 		return NULL;
 125 fb1a36c0 2022-01-09 op 
 126 fb1a36c0 2022-01-09 op 	if ((ch = strchr(line, '\n')) != NULL)
 127 fb1a36c0 2022-01-09 op 		*ch = '\0';
 128 fb1a36c0 2022-01-09 op 	return line;
 129 fb1a36c0 2022-01-09 op }
 130 fb1a36c0 2022-01-09 op #endif
 131 fb1a36c0 2022-01-09 op 
 132 fb1a36c0 2022-01-09 op static void
 133 fb1a36c0 2022-01-09 op tty_resized(int signo)
 134 fb1a36c0 2022-01-09 op {
 135 fb1a36c0 2022-01-09 op 	resized = 1;
 136 fb1a36c0 2022-01-09 op }
 137 fb1a36c0 2022-01-09 op 
 138 fb1a36c0 2022-01-09 op static void __dead
 139 fb1a36c0 2022-01-09 op usage(int ret)
 140 fb1a36c0 2022-01-09 op {
 141 fb1a36c0 2022-01-09 op 	fprintf(stderr, "usage: %s [-c] host[:port] [path]\n",
 142 fb1a36c0 2022-01-09 op 	    getprogname());
 143 fb1a36c0 2022-01-09 op 	fprintf(stderr, "kamid suite version " KAMID_VERSION "\n");
 144 fb1a36c0 2022-01-09 op 	exit(ret);
 145 fb1a36c0 2022-01-09 op }
 146 fb1a36c0 2022-01-09 op 
 147 fb1a36c0 2022-01-09 op static void
 148 fb1a36c0 2022-01-09 op do_send(void)
 149 fb1a36c0 2022-01-09 op {
 150 fb1a36c0 2022-01-09 op 	const void	*buf;
 151 fb1a36c0 2022-01-09 op 	size_t		 nbytes;
 152 fb1a36c0 2022-01-09 op 	ssize_t		 r;
 153 fb1a36c0 2022-01-09 op 
 154 fb1a36c0 2022-01-09 op 	while (EVBUFFER_LENGTH(evb) != 0) {
 155 fb1a36c0 2022-01-09 op 		buf = EVBUFFER_DATA(evb);
 156 fb1a36c0 2022-01-09 op 		nbytes = EVBUFFER_LENGTH(evb);
 157 fb1a36c0 2022-01-09 op 
 158 fb1a36c0 2022-01-09 op 		if (ctx == NULL) {
 159 fb1a36c0 2022-01-09 op 			r = write(sock, buf, nbytes);
 160 fb1a36c0 2022-01-09 op 			if (r == 0 || r == -1)
 161 fb1a36c0 2022-01-09 op 				errx(1, "EOF");
 162 fb1a36c0 2022-01-09 op 		} else {
 163 fb1a36c0 2022-01-09 op 			r = tls_write(ctx, buf, nbytes);
 164 fb1a36c0 2022-01-09 op 			if (r == TLS_WANT_POLLIN || r == TLS_WANT_POLLOUT)
 165 fb1a36c0 2022-01-09 op 				continue;
 166 fb1a36c0 2022-01-09 op 			if (r == -1)
 167 fb1a36c0 2022-01-09 op 				errx(1, "tls: %s", tls_error(ctx));
 168 fb1a36c0 2022-01-09 op 		}
 169 fb1a36c0 2022-01-09 op 
 170 fb1a36c0 2022-01-09 op 		evbuffer_drain(evb, r);
 171 fb1a36c0 2022-01-09 op 	}
 172 fb1a36c0 2022-01-09 op }
 173 fb1a36c0 2022-01-09 op 
 174 fb1a36c0 2022-01-09 op static void
 175 fb1a36c0 2022-01-09 op mustread(void *d, size_t len)
 176 fb1a36c0 2022-01-09 op {
 177 fb1a36c0 2022-01-09 op 	ssize_t r;
 178 fb1a36c0 2022-01-09 op 
 179 fb1a36c0 2022-01-09 op 	while (len != 0) {
 180 fb1a36c0 2022-01-09 op 		if (ctx == NULL) {
 181 fb1a36c0 2022-01-09 op 			r = read(sock, d, len);
 182 fb1a36c0 2022-01-09 op 			if (r == 0 || r == -1)
 183 fb1a36c0 2022-01-09 op 				errx(1, "EOF");
 184 fb1a36c0 2022-01-09 op 		} else {
 185 fb1a36c0 2022-01-09 op 			r = tls_read(ctx, d, len);
 186 fb1a36c0 2022-01-09 op 			if (r == TLS_WANT_POLLIN || r == TLS_WANT_POLLOUT)
 187 fb1a36c0 2022-01-09 op 				continue;
 188 fb1a36c0 2022-01-09 op 			if (r == -1)
 189 fb1a36c0 2022-01-09 op 				errx(1, "tls: %s", tls_error(ctx));
 190 fb1a36c0 2022-01-09 op 		}
 191 fb1a36c0 2022-01-09 op 
 192 fb1a36c0 2022-01-09 op 		d += r;
 193 fb1a36c0 2022-01-09 op 		len -= r;
 194 fb1a36c0 2022-01-09 op 	}
 195 fb1a36c0 2022-01-09 op }
 196 fb1a36c0 2022-01-09 op 
 197 fb1a36c0 2022-01-09 op static void
 198 fb1a36c0 2022-01-09 op recv_msg(void)
 199 fb1a36c0 2022-01-09 op {
 200 fb1a36c0 2022-01-09 op 	uint32_t	len, l;
 201 fb1a36c0 2022-01-09 op 	char		tmp[BUFSIZ];
 202 fb1a36c0 2022-01-09 op 
 203 fb1a36c0 2022-01-09 op 	mustread(&len, sizeof(len));
 204 fb1a36c0 2022-01-09 op 	len = le32toh(len);
 205 fb1a36c0 2022-01-09 op 	if (len < HEADERSIZE)
 206 fb1a36c0 2022-01-09 op 		errx(1, "read message of invalid length %d", len);
 207 fb1a36c0 2022-01-09 op 
 208 fb1a36c0 2022-01-09 op 	len -= 4; /* skip the length just read */
 209 fb1a36c0 2022-01-09 op 
 210 fb1a36c0 2022-01-09 op 	while (len != 0) {
 211 fb1a36c0 2022-01-09 op 		l = MIN(len, sizeof(tmp));
 212 fb1a36c0 2022-01-09 op 		mustread(tmp, l);
 213 fb1a36c0 2022-01-09 op 		len -= l;
 214 fb1a36c0 2022-01-09 op 		evbuffer_add(buf, tmp, l);
 215 fb1a36c0 2022-01-09 op 	}
 216 fb1a36c0 2022-01-09 op }
 217 fb1a36c0 2022-01-09 op 
 218 fb1a36c0 2022-01-09 op static uint64_t
 219 fb1a36c0 2022-01-09 op np_read64(struct evbuffer *buf)
 220 fb1a36c0 2022-01-09 op {
 221 fb1a36c0 2022-01-09 op 	uint64_t n;
 222 fb1a36c0 2022-01-09 op 
 223 fb1a36c0 2022-01-09 op 	evbuffer_remove(buf, &n, sizeof(n));
 224 fb1a36c0 2022-01-09 op 	return le64toh(n);
 225 fb1a36c0 2022-01-09 op }
 226 fb1a36c0 2022-01-09 op 
 227 fb1a36c0 2022-01-09 op static uint32_t
 228 fb1a36c0 2022-01-09 op np_read32(struct evbuffer *buf)
 229 fb1a36c0 2022-01-09 op {
 230 fb1a36c0 2022-01-09 op 	uint32_t n;
 231 fb1a36c0 2022-01-09 op 
 232 fb1a36c0 2022-01-09 op 	evbuffer_remove(buf, &n, sizeof(n));
 233 fb1a36c0 2022-01-09 op 	return le32toh(n);
 234 fb1a36c0 2022-01-09 op }
 235 fb1a36c0 2022-01-09 op 
 236 fb1a36c0 2022-01-09 op static uint16_t
 237 fb1a36c0 2022-01-09 op np_read16(struct evbuffer *buf)
 238 fb1a36c0 2022-01-09 op {
 239 fb1a36c0 2022-01-09 op 	uint16_t n;
 240 fb1a36c0 2022-01-09 op 
 241 fb1a36c0 2022-01-09 op 	evbuffer_remove(buf, &n, sizeof(n));
 242 fb1a36c0 2022-01-09 op 	return le16toh(n);
 243 fb1a36c0 2022-01-09 op }
 244 fb1a36c0 2022-01-09 op 
 245 fb1a36c0 2022-01-09 op static uint16_t
 246 fb1a36c0 2022-01-09 op np_read8(struct evbuffer *buf)
 247 fb1a36c0 2022-01-09 op {
 248 fb1a36c0 2022-01-09 op 	uint8_t n;
 249 fb1a36c0 2022-01-09 op 
 250 fb1a36c0 2022-01-09 op 	evbuffer_remove(buf, &n, sizeof(n));
 251 fb1a36c0 2022-01-09 op 	return n;
 252 fb1a36c0 2022-01-09 op }
 253 fb1a36c0 2022-01-09 op 
 254 fb1a36c0 2022-01-09 op static char *
 255 fb1a36c0 2022-01-09 op np_readstr(struct evbuffer *buf)
 256 fb1a36c0 2022-01-09 op {
 257 fb1a36c0 2022-01-09 op 	uint16_t	 len;
 258 fb1a36c0 2022-01-09 op 	char		*str;
 259 fb1a36c0 2022-01-09 op 
 260 fb1a36c0 2022-01-09 op 	len = np_read16(buf);
 261 fb1a36c0 2022-01-09 op 	assert(EVBUFFER_LENGTH(buf) >= len);
 262 fb1a36c0 2022-01-09 op 
 263 fb1a36c0 2022-01-09 op 	if ((str = calloc(1, len+1)) == NULL)
 264 fb1a36c0 2022-01-09 op 		err(1, "calloc");
 265 fb1a36c0 2022-01-09 op 	evbuffer_remove(buf, str, len);
 266 fb1a36c0 2022-01-09 op 	return str;
 267 fb1a36c0 2022-01-09 op }
 268 fb1a36c0 2022-01-09 op 
 269 fb1a36c0 2022-01-09 op static void
 270 fb1a36c0 2022-01-09 op np_read_qid(struct evbuffer *buf, struct qid *qid)
 271 fb1a36c0 2022-01-09 op {
 272 fb1a36c0 2022-01-09 op 	assert(EVBUFFER_LENGTH(buf) >= QIDSIZE);
 273 fb1a36c0 2022-01-09 op 
 274 fb1a36c0 2022-01-09 op 	qid->type = np_read8(buf);
 275 fb1a36c0 2022-01-09 op 	qid->vers = np_read32(buf);
 276 fb1a36c0 2022-01-09 op 	qid->path = np_read64(buf);
 277 fb1a36c0 2022-01-09 op }
 278 fb1a36c0 2022-01-09 op 
 279 fb1a36c0 2022-01-09 op static int
 280 fb1a36c0 2022-01-09 op np_read_stat(struct evbuffer *buf, struct np_stat *st)
 281 fb1a36c0 2022-01-09 op {
 282 fb1a36c0 2022-01-09 op 	uint16_t size;
 283 fb1a36c0 2022-01-09 op 
 284 fb1a36c0 2022-01-09 op 	memset(st, 0, sizeof(*st));
 285 fb1a36c0 2022-01-09 op 
 286 fb1a36c0 2022-01-09 op 	size = np_read16(buf);
 287 fb1a36c0 2022-01-09 op 	if (size > EVBUFFER_LENGTH(buf))
 288 fb1a36c0 2022-01-09 op 		return -1;
 289 fb1a36c0 2022-01-09 op 
 290 fb1a36c0 2022-01-09 op 	st->type = np_read16(buf);
 291 fb1a36c0 2022-01-09 op 	st->dev = np_read32(buf);
 292 fb1a36c0 2022-01-09 op 	np_read_qid(buf, &st->qid);
 293 fb1a36c0 2022-01-09 op 	st->mode = np_read32(buf);
 294 fb1a36c0 2022-01-09 op 	st->atime = np_read32(buf);
 295 fb1a36c0 2022-01-09 op 	st->mtime = np_read32(buf);
 296 fb1a36c0 2022-01-09 op 	st->length = np_read64(buf);
 297 fb1a36c0 2022-01-09 op 	st->name = np_readstr(buf);
 298 fb1a36c0 2022-01-09 op 	st->uid = np_readstr(buf);
 299 fb1a36c0 2022-01-09 op 	st->gid = np_readstr(buf);
 300 fb1a36c0 2022-01-09 op 	st->muid = np_readstr(buf);
 301 fb1a36c0 2022-01-09 op 
 302 fb1a36c0 2022-01-09 op 	return 0;
 303 fb1a36c0 2022-01-09 op }
 304 fb1a36c0 2022-01-09 op 
 305 fb1a36c0 2022-01-09 op static void
 306 fb1a36c0 2022-01-09 op expect(uint8_t type)
 307 fb1a36c0 2022-01-09 op {
 308 fb1a36c0 2022-01-09 op 	uint8_t t;
 309 fb1a36c0 2022-01-09 op 
 310 fb1a36c0 2022-01-09 op 	t = np_read8(buf);
 311 fb1a36c0 2022-01-09 op 	if (t == type)
 312 fb1a36c0 2022-01-09 op 		return;
 313 fb1a36c0 2022-01-09 op 
 314 fb1a36c0 2022-01-09 op 	if (t == Rerror) {
 315 fb1a36c0 2022-01-09 op 		char *err;
 316 fb1a36c0 2022-01-09 op 
 317 fb1a36c0 2022-01-09 op 		/* skip tag */
 318 fb1a36c0 2022-01-09 op 		np_read16(buf);
 319 fb1a36c0 2022-01-09 op 
 320 fb1a36c0 2022-01-09 op 		err = np_readstr(buf);
 321 fb1a36c0 2022-01-09 op 		errx(1, "expected %s, got error %s",
 322 fb1a36c0 2022-01-09 op 		    pp_msg_type(type), err);
 323 fb1a36c0 2022-01-09 op 	}
 324 fb1a36c0 2022-01-09 op 
 325 fb1a36c0 2022-01-09 op 	errx(1, "expected %s, got msg type %s",
 326 fb1a36c0 2022-01-09 op 	    pp_msg_type(type), pp_msg_type(t));
 327 fb1a36c0 2022-01-09 op }
 328 fb1a36c0 2022-01-09 op 
 329 fb1a36c0 2022-01-09 op static void
 330 fb1a36c0 2022-01-09 op expect2(uint8_t type, uint16_t tag)
 331 fb1a36c0 2022-01-09 op {
 332 fb1a36c0 2022-01-09 op 	uint16_t t;
 333 fb1a36c0 2022-01-09 op 
 334 fb1a36c0 2022-01-09 op 	expect(type);
 335 fb1a36c0 2022-01-09 op 
 336 fb1a36c0 2022-01-09 op 	t = np_read16(buf);
 337 fb1a36c0 2022-01-09 op 	if (t == tag)
 338 fb1a36c0 2022-01-09 op 		return;
 339 fb1a36c0 2022-01-09 op 
 340 fb1a36c0 2022-01-09 op 	errx(1, "expected tag 0x%x, got 0x%x", tag, t);
 341 fb1a36c0 2022-01-09 op }
 342 fb1a36c0 2022-01-09 op 
 343 fb1a36c0 2022-01-09 op static void
 344 fb1a36c0 2022-01-09 op do_version(void)
 345 fb1a36c0 2022-01-09 op {
 346 fb1a36c0 2022-01-09 op 	char		*version;
 347 fb1a36c0 2022-01-09 op 
 348 fb1a36c0 2022-01-09 op 	tversion(VERSION9P, MSIZE9P);
 349 fb1a36c0 2022-01-09 op 	do_send();
 350 fb1a36c0 2022-01-09 op 	recv_msg();
 351 fb1a36c0 2022-01-09 op 	expect2(Rversion, NOTAG);
 352 fb1a36c0 2022-01-09 op 
 353 fb1a36c0 2022-01-09 op 	msize = np_read32(buf);
 354 fb1a36c0 2022-01-09 op 	version = np_readstr(buf);
 355 fb1a36c0 2022-01-09 op 
 356 fb1a36c0 2022-01-09 op 	if (msize > MSIZE9P)
 357 fb1a36c0 2022-01-09 op 		errx(1, "got unexpected msize: %d", msize);
 358 fb1a36c0 2022-01-09 op 	if (strcmp(version, VERSION9P))
 359 fb1a36c0 2022-01-09 op 		errx(1, "unexpected 9p version: %s", version);
 360 fb1a36c0 2022-01-09 op 
 361 fb1a36c0 2022-01-09 op 	free(version);
 362 fb1a36c0 2022-01-09 op 	ASSERT_EMPTYBUF();
 363 fb1a36c0 2022-01-09 op }
 364 fb1a36c0 2022-01-09 op 
 365 fb1a36c0 2022-01-09 op static void
 366 fb1a36c0 2022-01-09 op do_attach(const char *path)
 367 fb1a36c0 2022-01-09 op {
 368 fb1a36c0 2022-01-09 op 	const char *user;
 369 fb1a36c0 2022-01-09 op 	struct qid qid;
 370 fb1a36c0 2022-01-09 op 
 371 fb1a36c0 2022-01-09 op 	if (path == NULL)
 372 fb1a36c0 2022-01-09 op 		path = "/";
 373 fb1a36c0 2022-01-09 op 	if ((user = getenv("USER")) == NULL)
 374 fb1a36c0 2022-01-09 op 		user = "flan";
 375 fb1a36c0 2022-01-09 op 
 376 fb1a36c0 2022-01-09 op 	tattach(pwdfid, NOFID, user, path);
 377 fb1a36c0 2022-01-09 op 	do_send();
 378 fb1a36c0 2022-01-09 op 	recv_msg();
 379 fb1a36c0 2022-01-09 op 	expect2(Rattach, iota_tag);
 380 fb1a36c0 2022-01-09 op 	np_read_qid(buf, &qid);
 381 fb1a36c0 2022-01-09 op 
 382 fb1a36c0 2022-01-09 op 	ASSERT_EMPTYBUF();
 383 fb1a36c0 2022-01-09 op }
 384 fb1a36c0 2022-01-09 op 
 385 fb1a36c0 2022-01-09 op static uint32_t
 386 fb1a36c0 2022-01-09 op do_open(uint32_t fid, uint8_t mode)
 387 fb1a36c0 2022-01-09 op {
 388 fb1a36c0 2022-01-09 op 	struct qid qid;
 389 fb1a36c0 2022-01-09 op 	uint32_t iounit;
 390 fb1a36c0 2022-01-09 op 
 391 fb1a36c0 2022-01-09 op 	topen(fid, mode);
 392 fb1a36c0 2022-01-09 op 	do_send();
 393 fb1a36c0 2022-01-09 op 	recv_msg();
 394 fb1a36c0 2022-01-09 op 	expect2(Ropen, iota_tag);
 395 fb1a36c0 2022-01-09 op 
 396 fb1a36c0 2022-01-09 op 	np_read_qid(buf, &qid);
 397 fb1a36c0 2022-01-09 op 	iounit = np_read32(buf);
 398 fb1a36c0 2022-01-09 op 
 399 fb1a36c0 2022-01-09 op 	ASSERT_EMPTYBUF();
 400 fb1a36c0 2022-01-09 op 
 401 fb1a36c0 2022-01-09 op 	return iounit;
 402 fb1a36c0 2022-01-09 op }
 403 fb1a36c0 2022-01-09 op 
 404 fb1a36c0 2022-01-09 op static void
 405 fb1a36c0 2022-01-09 op do_clunk(uint32_t fid)
 406 fb1a36c0 2022-01-09 op {
 407 fb1a36c0 2022-01-09 op 	tclunk(fid);
 408 fb1a36c0 2022-01-09 op 	do_send();
 409 fb1a36c0 2022-01-09 op 	recv_msg();
 410 fb1a36c0 2022-01-09 op 	expect2(Rclunk, iota_tag);
 411 fb1a36c0 2022-01-09 op 
 412 fb1a36c0 2022-01-09 op 	ASSERT_EMPTYBUF();
 413 fb1a36c0 2022-01-09 op }
 414 fb1a36c0 2022-01-09 op 
 415 fb1a36c0 2022-01-09 op static void
 416 fb1a36c0 2022-01-09 op dup_fid(int fid, int nfid)
 417 fb1a36c0 2022-01-09 op {
 418 fb1a36c0 2022-01-09 op 	uint16_t nwqid;
 419 fb1a36c0 2022-01-09 op 
 420 fb1a36c0 2022-01-09 op 	twalk(fid, nfid, NULL, 0);
 421 fb1a36c0 2022-01-09 op 	do_send();
 422 fb1a36c0 2022-01-09 op 	recv_msg();
 423 fb1a36c0 2022-01-09 op 	expect2(Rwalk, iota_tag);
 424 fb1a36c0 2022-01-09 op 
 425 fb1a36c0 2022-01-09 op 	nwqid = np_read16(buf);
 426 fb1a36c0 2022-01-09 op 	assert(nwqid == 0);
 427 fb1a36c0 2022-01-09 op 
 428 fb1a36c0 2022-01-09 op 	ASSERT_EMPTYBUF();
 429 fb1a36c0 2022-01-09 op }
 430 fb1a36c0 2022-01-09 op 
 431 fb1a36c0 2022-01-09 op static int
 432 fb1a36c0 2022-01-09 op walk_path(int fid, int newfid, const char *path, struct qid *qid)
 433 fb1a36c0 2022-01-09 op {
 434 fb1a36c0 2022-01-09 op 	char *wnames[MAXWELEM], *p, *t;
 435 fb1a36c0 2022-01-09 op 	size_t nwname, i;
 436 fb1a36c0 2022-01-09 op 	uint16_t nwqid;
 437 fb1a36c0 2022-01-09 op 
 438 fb1a36c0 2022-01-09 op 	if ((p = strdup(path)) == NULL)
 439 fb1a36c0 2022-01-09 op 		err(1, "strdup");
 440 fb1a36c0 2022-01-09 op 	t = p;
 441 fb1a36c0 2022-01-09 op 
 442 fb1a36c0 2022-01-09 op 	/* strip initial ./ */
 443 fb1a36c0 2022-01-09 op 	if (t[0] == '.' && t[1] == '/')
 444 fb1a36c0 2022-01-09 op 		t += 2;
 445 fb1a36c0 2022-01-09 op 
 446 fb1a36c0 2022-01-09 op 	for (nwname = 0; nwname < nitems(wnames) &&
 447 fb1a36c0 2022-01-09 op 	    (wnames[nwname] = strsep(&t, "/")) != NULL;) {
 448 fb1a36c0 2022-01-09 op 		if (*wnames[nwname] != '\0')
 449 fb1a36c0 2022-01-09 op 			nwname++;
 450 fb1a36c0 2022-01-09 op 	}
 451 fb1a36c0 2022-01-09 op 
 452 fb1a36c0 2022-01-09 op 	twalk(fid, newfid, (const char **)wnames, nwname);
 453 fb1a36c0 2022-01-09 op 	do_send();
 454 fb1a36c0 2022-01-09 op 	recv_msg();
 455 fb1a36c0 2022-01-09 op 	expect2(Rwalk, iota_tag);
 456 fb1a36c0 2022-01-09 op 
 457 fb1a36c0 2022-01-09 op 	nwqid = np_read16(buf);
 458 fb1a36c0 2022-01-09 op 	assert(nwqid <= nwname);
 459 fb1a36c0 2022-01-09 op 
 460 fb1a36c0 2022-01-09 op 	/* consume all qids */
 461 fb1a36c0 2022-01-09 op 	for (i = 0; i < nwname; ++i)
 462 fb1a36c0 2022-01-09 op 		np_read_qid(buf, qid);
 463 fb1a36c0 2022-01-09 op 
 464 fb1a36c0 2022-01-09 op 	free(p);
 465 fb1a36c0 2022-01-09 op 
 466 fb1a36c0 2022-01-09 op 	return nwqid == nwname;
 467 fb1a36c0 2022-01-09 op }
 468 fb1a36c0 2022-01-09 op 
 469 fb1a36c0 2022-01-09 op static void
 470 fb1a36c0 2022-01-09 op do_stat(int fid, struct np_stat *st)
 471 fb1a36c0 2022-01-09 op {
 472 fb1a36c0 2022-01-09 op 	tstat(fid);
 473 fb1a36c0 2022-01-09 op 	do_send();
 474 fb1a36c0 2022-01-09 op 	recv_msg();
 475 fb1a36c0 2022-01-09 op 	expect2(Rstat, iota_tag);
 476 fb1a36c0 2022-01-09 op 
 477 fb1a36c0 2022-01-09 op 	if (np_read_stat(buf, st) == -1)
 478 fb1a36c0 2022-01-09 op 		errx(1, "invalid stat struct read");
 479 fb1a36c0 2022-01-09 op 
 480 fb1a36c0 2022-01-09 op 	ASSERT_EMPTYBUF();
 481 fb1a36c0 2022-01-09 op }
 482 fb1a36c0 2022-01-09 op 
 483 fb1a36c0 2022-01-09 op static size_t
 484 fb1a36c0 2022-01-09 op do_read(int fid, uint64_t off, uint32_t count, void *data)
 485 fb1a36c0 2022-01-09 op {
 486 fb1a36c0 2022-01-09 op 	uint32_t r;
 487 fb1a36c0 2022-01-09 op 
 488 fb1a36c0 2022-01-09 op 	tread(fid, off, count);
 489 fb1a36c0 2022-01-09 op 	do_send();
 490 fb1a36c0 2022-01-09 op 	recv_msg();
 491 fb1a36c0 2022-01-09 op 	expect2(Rread, iota_tag);
 492 fb1a36c0 2022-01-09 op 
 493 fb1a36c0 2022-01-09 op 	r = np_read32(buf);
 494 fb1a36c0 2022-01-09 op 	assert(r == EVBUFFER_LENGTH(buf));
 495 fb1a36c0 2022-01-09 op 	assert(r <= count);
 496 fb1a36c0 2022-01-09 op 	evbuffer_remove(buf, data, r);
 497 fb1a36c0 2022-01-09 op 
 498 fb1a36c0 2022-01-09 op 	ASSERT_EMPTYBUF();
 499 fb1a36c0 2022-01-09 op 
 500 fb1a36c0 2022-01-09 op 	return r;
 501 fb1a36c0 2022-01-09 op }
 502 fb1a36c0 2022-01-09 op 
 503 fb1a36c0 2022-01-09 op static void
 504 fb1a36c0 2022-01-09 op draw_progress(const char *pre, const struct progress *p)
 505 fb1a36c0 2022-01-09 op {
 506 fb1a36c0 2022-01-09 op 	struct winsize ws;
 507 fb1a36c0 2022-01-09 op 	int i, l, w;
 508 fb1a36c0 2022-01-09 op 	double perc;
 509 fb1a36c0 2022-01-09 op 
 510 fb1a36c0 2022-01-09 op 	perc = 100.0 * p->done / p->max;
 511 fb1a36c0 2022-01-09 op 	if (!tty_p) {
 512 fb1a36c0 2022-01-09 op 		fprintf(stderr, "%s: %d%%\n", pre, (int)perc);
 513 fb1a36c0 2022-01-09 op 		return;
 514 fb1a36c0 2022-01-09 op 	}
 515 fb1a36c0 2022-01-09 op 
 516 fb1a36c0 2022-01-09 op 	if (resized) {
 517 fb1a36c0 2022-01-09 op 		resized = 0;
 518 fb1a36c0 2022-01-09 op 
 519 fb1a36c0 2022-01-09 op 		if (ioctl(0, TIOCGWINSZ, &ws) == -1)
 520 fb1a36c0 2022-01-09 op 			return;
 521 fb1a36c0 2022-01-09 op 		tty_width = ws.ws_col;
 522 fb1a36c0 2022-01-09 op 	}
 523 fb1a36c0 2022-01-09 op 	w = tty_width;
 524 fb1a36c0 2022-01-09 op 
 525 fb1a36c0 2022-01-09 op 	if (pre == NULL ||
 526 fb1a36c0 2022-01-09 op 	    ((l = printf("\r%s ", pre)) == -1 || l >= w))
 527 fb1a36c0 2022-01-09 op 		return;
 528 fb1a36c0 2022-01-09 op 
 529 fb1a36c0 2022-01-09 op 	w -= l + 2 + 5; /* 2 for |, 5 for percentage + \n */
 530 fb1a36c0 2022-01-09 op 	if (w < 0) {
 531 fb1a36c0 2022-01-09 op 		printf("%4d%%\n", (int)perc);
 532 fb1a36c0 2022-01-09 op 		return;
 533 fb1a36c0 2022-01-09 op 	}
 534 fb1a36c0 2022-01-09 op 
 535 fb1a36c0 2022-01-09 op 	printf("|");
 536 fb1a36c0 2022-01-09 op 
 537 fb1a36c0 2022-01-09 op 	l = w * MIN(100.0, perc) / 100.0;
 538 fb1a36c0 2022-01-09 op 	for (i = 0; i < l; i++)
 539 fb1a36c0 2022-01-09 op 		printf("*");
 540 fb1a36c0 2022-01-09 op 	for (; i < w; i++)
 541 fb1a36c0 2022-01-09 op 		printf(" ");
 542 fb1a36c0 2022-01-09 op 	printf("|%4d%%", (int)perc);
 543 fb1a36c0 2022-01-09 op 
 544 fb1a36c0 2022-01-09 op 	fflush(stdout);
 545 fb1a36c0 2022-01-09 op }
 546 fb1a36c0 2022-01-09 op 
 547 fb1a36c0 2022-01-09 op static int
 548 fb1a36c0 2022-01-09 op fetch_fid(int fid, const char *path)
 549 fb1a36c0 2022-01-09 op {
 550 fb1a36c0 2022-01-09 op 	struct progress p = {0};
 551 fb1a36c0 2022-01-09 op 	struct np_stat st;
 552 fb1a36c0 2022-01-09 op 	size_t r;
 553 fb1a36c0 2022-01-09 op 	int fd;
 554 fb1a36c0 2022-01-09 op 	char buf[BUFSIZ];
 555 fb1a36c0 2022-01-09 op 
 556 fb1a36c0 2022-01-09 op 	do_stat(fid, &st);
 557 fb1a36c0 2022-01-09 op 	do_open(fid, KOREAD);
 558 fb1a36c0 2022-01-09 op 
 559 fb1a36c0 2022-01-09 op 	if ((fd = open(path, O_WRONLY|O_CREAT|O_TRUNC, 0644)) == -1) {
 560 fb1a36c0 2022-01-09 op 		warn("can't open %s", path);
 561 fb1a36c0 2022-01-09 op 		return -1;
 562 fb1a36c0 2022-01-09 op 	}
 563 fb1a36c0 2022-01-09 op 
 564 fb1a36c0 2022-01-09 op 	p.max = st.length;
 565 fb1a36c0 2022-01-09 op 	for (;;) {
 566 fb1a36c0 2022-01-09 op 		size_t siz, off;
 567 fb1a36c0 2022-01-09 op 		ssize_t nw;
 568 fb1a36c0 2022-01-09 op 
 569 fb1a36c0 2022-01-09 op 		r = do_read(fid, p.done, sizeof(buf), buf);
 570 fb1a36c0 2022-01-09 op 		if (r == 0)
 571 fb1a36c0 2022-01-09 op 			break;
 572 fb1a36c0 2022-01-09 op 
 573 fb1a36c0 2022-01-09 op 		siz = sizeof(buf);
 574 fb1a36c0 2022-01-09 op 		for (off = 0; off < siz; off += nw)
 575 fb1a36c0 2022-01-09 op 			if ((nw = write(fd, buf + off, siz - off)) == 0 ||
 576 fb1a36c0 2022-01-09 op 			    nw == -1)
 577 fb1a36c0 2022-01-09 op 				err(1, "write");
 578 fb1a36c0 2022-01-09 op 
 579 fb1a36c0 2022-01-09 op 		p.done += r;
 580 fb1a36c0 2022-01-09 op 		draw_progress(path, &p);
 581 fb1a36c0 2022-01-09 op 
 582 fb1a36c0 2022-01-09 op #if 0
 583 fb1a36c0 2022-01-09 op 		/* throttle, for debugging purpose */
 584 fb1a36c0 2022-01-09 op 		{
 585 fb1a36c0 2022-01-09 op 			struct timespec ts = { 0, 500000000 };
 586 fb1a36c0 2022-01-09 op 			nanosleep(&ts, NULL);
 587 fb1a36c0 2022-01-09 op 		}
 588 fb1a36c0 2022-01-09 op #endif
 589 fb1a36c0 2022-01-09 op 	}
 590 fb1a36c0 2022-01-09 op 
 591 fb1a36c0 2022-01-09 op 	putchar('\n');
 592 fb1a36c0 2022-01-09 op 
 593 fb1a36c0 2022-01-09 op 	close(fd);
 594 fb1a36c0 2022-01-09 op 	do_clunk(fid);
 595 fb1a36c0 2022-01-09 op 	return 0;
 596 fb1a36c0 2022-01-09 op }
 597 fb1a36c0 2022-01-09 op 
 598 fb1a36c0 2022-01-09 op static void
 599 fb1a36c0 2022-01-09 op do_tls_connect(const char *host, const char *port)
 600 fb1a36c0 2022-01-09 op {
 601 fb1a36c0 2022-01-09 op 	int handshake;
 602 fb1a36c0 2022-01-09 op 
 603 fb1a36c0 2022-01-09 op 	if ((tlsconf = tls_config_new()) == NULL)
 604 fb1a36c0 2022-01-09 op 		fatalx("tls_config_new");
 605 fb1a36c0 2022-01-09 op 	tls_config_insecure_noverifycert(tlsconf);
 606 fb1a36c0 2022-01-09 op 	tls_config_insecure_noverifyname(tlsconf);
 607 fb1a36c0 2022-01-09 op 	if (tls_config_set_keypair_file(tlsconf, crtpath, keypath) == -1)
 608 fb1a36c0 2022-01-09 op 		fatalx("can't load certs (%s, %s)", crtpath, keypath);
 609 fb1a36c0 2022-01-09 op 
 610 fb1a36c0 2022-01-09 op 	if ((ctx = tls_client()) == NULL)
 611 fb1a36c0 2022-01-09 op 		fatal("tls_client");
 612 fb1a36c0 2022-01-09 op 	if (tls_configure(ctx, tlsconf) == -1)
 613 fb1a36c0 2022-01-09 op 		fatalx("tls_configure: %s", tls_error(ctx));
 614 fb1a36c0 2022-01-09 op 
 615 fb1a36c0 2022-01-09 op 	if (tls_connect(ctx, host, port) == -1)
 616 fb1a36c0 2022-01-09 op 		fatalx("can't connect to %s:%s: %s", host, port,
 617 fb1a36c0 2022-01-09 op 		    tls_error(ctx));
 618 fb1a36c0 2022-01-09 op 
 619 fb1a36c0 2022-01-09 op 	for (handshake = 0; !handshake;) {
 620 fb1a36c0 2022-01-09 op 		switch (tls_handshake(ctx)) {
 621 fb1a36c0 2022-01-09 op 		case -1:
 622 fb1a36c0 2022-01-09 op 			fatalx("tls_handshake: %s", tls_error(ctx));
 623 fb1a36c0 2022-01-09 op 		case 0:
 624 fb1a36c0 2022-01-09 op 			handshake = 1;
 625 fb1a36c0 2022-01-09 op 			break;
 626 fb1a36c0 2022-01-09 op 		}
 627 fb1a36c0 2022-01-09 op 	}
 628 fb1a36c0 2022-01-09 op }
 629 fb1a36c0 2022-01-09 op 
 630 fb1a36c0 2022-01-09 op static void
 631 fb1a36c0 2022-01-09 op do_ctxt_connect(const char *host, const char *port)
 632 fb1a36c0 2022-01-09 op {
 633 fb1a36c0 2022-01-09 op 	struct addrinfo hints, *res, *res0;
 634 fb1a36c0 2022-01-09 op 	int error, saved_errno;
 635 fb1a36c0 2022-01-09 op 	const char *cause = NULL;
 636 fb1a36c0 2022-01-09 op 
 637 fb1a36c0 2022-01-09 op 	memset(&hints, 0, sizeof(hints));
 638 fb1a36c0 2022-01-09 op 	hints.ai_family = AF_UNSPEC;
 639 fb1a36c0 2022-01-09 op 	hints.ai_socktype = SOCK_STREAM;
 640 fb1a36c0 2022-01-09 op 	error = getaddrinfo(host, port, &hints, &res0);
 641 fb1a36c0 2022-01-09 op 	if (error)
 642 fb1a36c0 2022-01-09 op 		errx(1, "%s", gai_strerror(error));
 643 fb1a36c0 2022-01-09 op 
 644 fb1a36c0 2022-01-09 op 	sock = -1;
 645 fb1a36c0 2022-01-09 op 	for (res = res0; res != NULL; res = res->ai_next) {
 646 fb1a36c0 2022-01-09 op 		sock = socket(res->ai_family, res->ai_socktype,
 647 fb1a36c0 2022-01-09 op 		    res->ai_protocol);
 648 fb1a36c0 2022-01-09 op 		if (sock == -1) {
 649 fb1a36c0 2022-01-09 op 			cause = "socket";
 650 fb1a36c0 2022-01-09 op 			continue;
 651 fb1a36c0 2022-01-09 op 		}
 652 fb1a36c0 2022-01-09 op 
 653 fb1a36c0 2022-01-09 op 		if (connect(sock, res->ai_addr, res->ai_addrlen) == -1) {
 654 fb1a36c0 2022-01-09 op 			cause = "connect";
 655 fb1a36c0 2022-01-09 op 			saved_errno = errno;
 656 fb1a36c0 2022-01-09 op 			close(sock);
 657 fb1a36c0 2022-01-09 op 			errno = saved_errno;
 658 fb1a36c0 2022-01-09 op 			sock = -1;
 659 fb1a36c0 2022-01-09 op 			continue;
 660 fb1a36c0 2022-01-09 op 		}
 661 fb1a36c0 2022-01-09 op 
 662 fb1a36c0 2022-01-09 op 		break;
 663 fb1a36c0 2022-01-09 op 	}
 664 fb1a36c0 2022-01-09 op 
 665 fb1a36c0 2022-01-09 op 	if (sock == -1)
 666 fb1a36c0 2022-01-09 op 		err(1, "%s", cause);
 667 fb1a36c0 2022-01-09 op 	freeaddrinfo(res0);
 668 fb1a36c0 2022-01-09 op }
 669 fb1a36c0 2022-01-09 op 
 670 fb1a36c0 2022-01-09 op static void
 671 fb1a36c0 2022-01-09 op do_connect(const char *connspec, const char *path)
 672 fb1a36c0 2022-01-09 op {
 673 fb1a36c0 2022-01-09 op 	char *host, *colon;
 674 fb1a36c0 2022-01-09 op 	const char *port;
 675 fb1a36c0 2022-01-09 op 
 676 fb1a36c0 2022-01-09 op 	host = xstrdup(connspec);
 677 fb1a36c0 2022-01-09 op 	if ((colon = strchr(host, ':')) != NULL) {
 678 fb1a36c0 2022-01-09 op 		*colon = '\0';
 679 fb1a36c0 2022-01-09 op 		port = ++colon;
 680 fb1a36c0 2022-01-09 op 	} else
 681 fb1a36c0 2022-01-09 op 		port = "1337";
 682 fb1a36c0 2022-01-09 op 
 683 fb1a36c0 2022-01-09 op 	printf("connecting to %s:%s...", host, port);
 684 fb1a36c0 2022-01-09 op 	fflush(stdout);
 685 fb1a36c0 2022-01-09 op 
 686 fb1a36c0 2022-01-09 op 	if (tls)
 687 fb1a36c0 2022-01-09 op 		do_tls_connect(host, port);
 688 fb1a36c0 2022-01-09 op 	else
 689 fb1a36c0 2022-01-09 op 		do_ctxt_connect(host, port);
 690 fb1a36c0 2022-01-09 op 
 691 fb1a36c0 2022-01-09 op 	printf(" done!\n");
 692 fb1a36c0 2022-01-09 op 
 693 fb1a36c0 2022-01-09 op 	do_version();
 694 fb1a36c0 2022-01-09 op 	do_attach(path);
 695 fb1a36c0 2022-01-09 op 
 696 fb1a36c0 2022-01-09 op 	free(host);
 697 fb1a36c0 2022-01-09 op }
 698 fb1a36c0 2022-01-09 op 
 699 fb1a36c0 2022-01-09 op static void
 700 fb1a36c0 2022-01-09 op cmd_bell(int argc, const char **argv)
 701 fb1a36c0 2022-01-09 op {
 702 fb1a36c0 2022-01-09 op 	if (argc == 0) {
 703 fb1a36c0 2022-01-09 op 		bell = !bell;
 704 fb1a36c0 2022-01-09 op 		if (bell)
 705 fb1a36c0 2022-01-09 op 			puts("bell mode enabled");
 706 fb1a36c0 2022-01-09 op 		else
 707 fb1a36c0 2022-01-09 op 			puts("bell mode disabled");
 708 fb1a36c0 2022-01-09 op 		return;
 709 fb1a36c0 2022-01-09 op 	}
 710 fb1a36c0 2022-01-09 op 
 711 fb1a36c0 2022-01-09 op 	if (argc != 1)
 712 fb1a36c0 2022-01-09 op 		goto usage;
 713 fb1a36c0 2022-01-09 op 
 714 fb1a36c0 2022-01-09 op 	if (!strcmp(*argv, "on")) {
 715 fb1a36c0 2022-01-09 op 		bell = 1;
 716 fb1a36c0 2022-01-09 op 		puts("bell mode enabled");
 717 fb1a36c0 2022-01-09 op 		return;
 718 fb1a36c0 2022-01-09 op 	}
 719 fb1a36c0 2022-01-09 op 
 720 fb1a36c0 2022-01-09 op 	if (!strcmp(*argv, "off")) {
 721 fb1a36c0 2022-01-09 op 		bell = 0;
 722 fb1a36c0 2022-01-09 op 		puts("bell mode disabled");
 723 fb1a36c0 2022-01-09 op 		return;
 724 fb1a36c0 2022-01-09 op 	}
 725 fb1a36c0 2022-01-09 op 
 726 fb1a36c0 2022-01-09 op usage:
 727 fb1a36c0 2022-01-09 op 	printf("bell [on | off]\n");
 728 fb1a36c0 2022-01-09 op }
 729 fb1a36c0 2022-01-09 op 
 730 fb1a36c0 2022-01-09 op static void
 731 fb1a36c0 2022-01-09 op cmd_bye(int argc, const char **argv)
 732 fb1a36c0 2022-01-09 op {
 733 fb1a36c0 2022-01-09 op 	log_warnx("bye\n");
 734 fb1a36c0 2022-01-09 op 	exit(0);
 735 fb1a36c0 2022-01-09 op }
 736 fb1a36c0 2022-01-09 op 
 737 fb1a36c0 2022-01-09 op static void
 738 fb1a36c0 2022-01-09 op cmd_cd(int argc, const char **argv)
 739 fb1a36c0 2022-01-09 op {
 740 fb1a36c0 2022-01-09 op 	struct qid qid;
 741 fb1a36c0 2022-01-09 op 	int nfid;
 742 fb1a36c0 2022-01-09 op 
 743 fb1a36c0 2022-01-09 op 	if (argc != 1) {
 744 fb1a36c0 2022-01-09 op 		printf("usage: cd remote-path\n");
 745 fb1a36c0 2022-01-09 op 		return;
 746 fb1a36c0 2022-01-09 op 	}
 747 fb1a36c0 2022-01-09 op 
 748 fb1a36c0 2022-01-09 op 	nfid = pwdfid+1;
 749 fb1a36c0 2022-01-09 op 	if (walk_path(pwdfid, nfid, argv[0], &qid) == -1 ||
 750 fb1a36c0 2022-01-09 op 	    !(qid.type & QTDIR)) {
 751 fb1a36c0 2022-01-09 op 		printf("can't cd %s\n", argv[0]);
 752 fb1a36c0 2022-01-09 op 		do_clunk(nfid);
 753 fb1a36c0 2022-01-09 op 	} else {
 754 fb1a36c0 2022-01-09 op 		do_clunk(pwdfid);
 755 fb1a36c0 2022-01-09 op 		pwdfid = nfid;
 756 fb1a36c0 2022-01-09 op 	}
 757 fb1a36c0 2022-01-09 op }
 758 fb1a36c0 2022-01-09 op 
 759 fb1a36c0 2022-01-09 op static void
 760 fb1a36c0 2022-01-09 op cmd_get(int argc, const char **argv)
 761 fb1a36c0 2022-01-09 op {
 762 fb1a36c0 2022-01-09 op 	struct qid qid;
 763 fb1a36c0 2022-01-09 op 	const char *l;
 764 fb1a36c0 2022-01-09 op 	int nfid;
 765 fb1a36c0 2022-01-09 op 
 766 fb1a36c0 2022-01-09 op 	if (argc != 1 && argc != 2) {
 767 fb1a36c0 2022-01-09 op 		printf("usage: get remote-file [local-file]\n");
 768 fb1a36c0 2022-01-09 op 		return;
 769 fb1a36c0 2022-01-09 op 	}
 770 fb1a36c0 2022-01-09 op 
 771 fb1a36c0 2022-01-09 op 	if (argc == 2)
 772 fb1a36c0 2022-01-09 op 		l = argv[1];
 773 fb1a36c0 2022-01-09 op 	else if ((l = strrchr(argv[0], '/')) != NULL)
 774 fb1a36c0 2022-01-09 op 		l++; /* skip / */
 775 fb1a36c0 2022-01-09 op 	else
 776 fb1a36c0 2022-01-09 op 		l = argv[1];
 777 fb1a36c0 2022-01-09 op 
 778 fb1a36c0 2022-01-09 op 	nfid = pwdfid+1;
 779 fb1a36c0 2022-01-09 op 	if (walk_path(pwdfid, nfid, argv[0], &qid) == -1) {
 780 fb1a36c0 2022-01-09 op 		printf("can't fetch %s\n", argv[0]);
 781 fb1a36c0 2022-01-09 op 		return;
 782 fb1a36c0 2022-01-09 op 	}
 783 fb1a36c0 2022-01-09 op 
 784 fb1a36c0 2022-01-09 op 	if (qid.type != 0) {
 785 fb1a36c0 2022-01-09 op 		printf("can't fetch %s\n", argv[0]);
 786 fb1a36c0 2022-01-09 op 		do_clunk(nfid);
 787 fb1a36c0 2022-01-09 op 		return;
 788 fb1a36c0 2022-01-09 op 	}
 789 fb1a36c0 2022-01-09 op 
 790 fb1a36c0 2022-01-09 op 	fetch_fid(nfid, l);
 791 fb1a36c0 2022-01-09 op }
 792 fb1a36c0 2022-01-09 op 
 793 fb1a36c0 2022-01-09 op static void
 794 fb1a36c0 2022-01-09 op cmd_lcd(int argc, const char **argv)
 795 fb1a36c0 2022-01-09 op {
 796 fb1a36c0 2022-01-09 op 	const char *dir;
 797 fb1a36c0 2022-01-09 op 
 798 fb1a36c0 2022-01-09 op 	if (argc > 1) {
 799 fb1a36c0 2022-01-09 op 		printf("lcd takes only one argument\n");
 800 fb1a36c0 2022-01-09 op 		return;
 801 fb1a36c0 2022-01-09 op 	}
 802 fb1a36c0 2022-01-09 op 
 803 fb1a36c0 2022-01-09 op 	if (argc == 1)
 804 fb1a36c0 2022-01-09 op 		dir = *argv;
 805 fb1a36c0 2022-01-09 op 
 806 fb1a36c0 2022-01-09 op 	if (argc == 0 && (dir = getenv("HOME")) == NULL) {
 807 fb1a36c0 2022-01-09 op 		printf("HOME is not defined\n");
 808 fb1a36c0 2022-01-09 op 		return;
 809 fb1a36c0 2022-01-09 op 	}
 810 fb1a36c0 2022-01-09 op 
 811 fb1a36c0 2022-01-09 op 	if (chdir(dir) == -1)
 812 fb1a36c0 2022-01-09 op 		printf("cd: %s: %s\n", dir, strerror(errno));
 813 fb1a36c0 2022-01-09 op }
 814 fb1a36c0 2022-01-09 op 
 815 fb1a36c0 2022-01-09 op static void
 816 fb1a36c0 2022-01-09 op cmd_lpwd(int argc, const char **argv)
 817 fb1a36c0 2022-01-09 op {
 818 fb1a36c0 2022-01-09 op 	char path[PATH_MAX];
 819 fb1a36c0 2022-01-09 op 
 820 fb1a36c0 2022-01-09 op 	if (getcwd(path, sizeof(path)) == NULL) {
 821 fb1a36c0 2022-01-09 op 		printf("lpwd: %s\n", strerror(errno));
 822 fb1a36c0 2022-01-09 op 		return;
 823 fb1a36c0 2022-01-09 op 	}
 824 fb1a36c0 2022-01-09 op 
 825 fb1a36c0 2022-01-09 op 	printf("%s\n", path);
 826 fb1a36c0 2022-01-09 op }
 827 fb1a36c0 2022-01-09 op 
 828 fb1a36c0 2022-01-09 op static void
 829 fb1a36c0 2022-01-09 op cmd_ls(int argc, const char **argv)
 830 fb1a36c0 2022-01-09 op {
 831 fb1a36c0 2022-01-09 op 	struct np_stat st;
 832 fb1a36c0 2022-01-09 op 	uint64_t off = 0;
 833 fb1a36c0 2022-01-09 op 	uint32_t len;
 834 fb1a36c0 2022-01-09 op 	char fmt[FMT_SCALED_STRSIZE];
 835 fb1a36c0 2022-01-09 op 
 836 fb1a36c0 2022-01-09 op 	if (argc != 0) {
 837 fb1a36c0 2022-01-09 op 		printf("ls don't take arguments (yet)\n");
 838 fb1a36c0 2022-01-09 op 		return;
 839 fb1a36c0 2022-01-09 op 	}
 840 fb1a36c0 2022-01-09 op 
 841 fb1a36c0 2022-01-09 op 	dup_fid(pwdfid, 1);
 842 fb1a36c0 2022-01-09 op 	do_open(1, KOREAD);
 843 fb1a36c0 2022-01-09 op 
 844 fb1a36c0 2022-01-09 op 	evbuffer_drain(dirbuf, EVBUFFER_LENGTH(dirbuf));
 845 fb1a36c0 2022-01-09 op 
 846 fb1a36c0 2022-01-09 op 	for (;;) {
 847 fb1a36c0 2022-01-09 op 		tread(1, off, BUFSIZ);
 848 fb1a36c0 2022-01-09 op 		do_send();
 849 fb1a36c0 2022-01-09 op 		recv_msg();
 850 fb1a36c0 2022-01-09 op 		expect2(Rread, iota_tag);
 851 fb1a36c0 2022-01-09 op 
 852 fb1a36c0 2022-01-09 op 		len = np_read32(buf);
 853 fb1a36c0 2022-01-09 op 		if (len == 0)
 854 fb1a36c0 2022-01-09 op 			break;
 855 fb1a36c0 2022-01-09 op 
 856 fb1a36c0 2022-01-09 op 		evbuffer_add_buffer(dirbuf, buf);
 857 fb1a36c0 2022-01-09 op 		off += len;
 858 fb1a36c0 2022-01-09 op 
 859 fb1a36c0 2022-01-09 op 		ASSERT_EMPTYBUF();
 860 fb1a36c0 2022-01-09 op 	}
 861 fb1a36c0 2022-01-09 op 
 862 fb1a36c0 2022-01-09 op 	while (EVBUFFER_LENGTH(dirbuf) != 0) {
 863 fb1a36c0 2022-01-09 op 		if (np_read_stat(dirbuf, &st) == -1)
 864 fb1a36c0 2022-01-09 op 			errx(1, "invalid stat struct read");
 865 fb1a36c0 2022-01-09 op 
 866 fb1a36c0 2022-01-09 op 		if (fmt_scaled(st.length, fmt) == -1)
 867 fb1a36c0 2022-01-09 op 			strlcpy(fmt, "xxx", sizeof(fmt));
 868 fb1a36c0 2022-01-09 op 
 869 fb1a36c0 2022-01-09 op 		printf("%4s %8s %s\n", pp_qid_type(st.qid.type), fmt, st.name);
 870 fb1a36c0 2022-01-09 op 
 871 fb1a36c0 2022-01-09 op 		free(st.name);
 872 fb1a36c0 2022-01-09 op 		free(st.uid);
 873 fb1a36c0 2022-01-09 op 		free(st.gid);
 874 fb1a36c0 2022-01-09 op 		free(st.muid);
 875 fb1a36c0 2022-01-09 op 	}
 876 fb1a36c0 2022-01-09 op 
 877 fb1a36c0 2022-01-09 op 	do_clunk(1);
 878 fb1a36c0 2022-01-09 op }
 879 fb1a36c0 2022-01-09 op 
 880 fb1a36c0 2022-01-09 op static void
 881 fb1a36c0 2022-01-09 op cmd_verbose(int argc, const char **argv)
 882 fb1a36c0 2022-01-09 op {
 883 fb1a36c0 2022-01-09 op 	if (argc == 0) {
 884 fb1a36c0 2022-01-09 op 		log_setverbose(!log_getverbose());
 885 fb1a36c0 2022-01-09 op 		if (log_getverbose())
 886 fb1a36c0 2022-01-09 op 			puts("verbose mode enabled");
 887 fb1a36c0 2022-01-09 op 		else
 888 fb1a36c0 2022-01-09 op 			puts("verbose mode disabled");
 889 fb1a36c0 2022-01-09 op 		return;
 890 fb1a36c0 2022-01-09 op 	}
 891 fb1a36c0 2022-01-09 op 
 892 fb1a36c0 2022-01-09 op 	if (argc != 1)
 893 fb1a36c0 2022-01-09 op 		goto usage;
 894 fb1a36c0 2022-01-09 op 
 895 fb1a36c0 2022-01-09 op 	if (!strcmp(*argv, "on")) {
 896 fb1a36c0 2022-01-09 op 		log_setverbose(1);
 897 fb1a36c0 2022-01-09 op 		puts("verbose mode enabled");
 898 fb1a36c0 2022-01-09 op 		return;
 899 fb1a36c0 2022-01-09 op 	}
 900 fb1a36c0 2022-01-09 op 
 901 fb1a36c0 2022-01-09 op 	if (!strcmp(*argv, "off")) {
 902 fb1a36c0 2022-01-09 op 		log_setverbose(0);
 903 fb1a36c0 2022-01-09 op 		puts("verbose mode disabled");
 904 fb1a36c0 2022-01-09 op 		return;
 905 fb1a36c0 2022-01-09 op 	}
 906 fb1a36c0 2022-01-09 op 
 907 fb1a36c0 2022-01-09 op usage:
 908 fb1a36c0 2022-01-09 op 	printf("verbose [on | off]\n");
 909 fb1a36c0 2022-01-09 op }
 910 fb1a36c0 2022-01-09 op 
 911 fb1a36c0 2022-01-09 op static void
 912 fb1a36c0 2022-01-09 op excmd(int argc, const char **argv)
 913 fb1a36c0 2022-01-09 op {
 914 fb1a36c0 2022-01-09 op 	struct cmd {
 915 fb1a36c0 2022-01-09 op 		const char	*name;
 916 fb1a36c0 2022-01-09 op 		void		(*fn)(int, const char **);
 917 fb1a36c0 2022-01-09 op 	} cmds[] = {
 918 fb1a36c0 2022-01-09 op 		{"bell",	cmd_bell},
 919 fb1a36c0 2022-01-09 op 		{"bye",		cmd_bye},
 920 fb1a36c0 2022-01-09 op 		{"cd",		cmd_cd},
 921 fb1a36c0 2022-01-09 op 		{"get",		cmd_get},
 922 fb1a36c0 2022-01-09 op 		{"lcd",		cmd_lcd},
 923 fb1a36c0 2022-01-09 op 		{"lpwd",	cmd_lpwd},
 924 fb1a36c0 2022-01-09 op 		{"ls",		cmd_ls},
 925 fb1a36c0 2022-01-09 op 		{"quit",	cmd_bye},
 926 fb1a36c0 2022-01-09 op 		{"verbose",	cmd_verbose},
 927 fb1a36c0 2022-01-09 op 	};
 928 fb1a36c0 2022-01-09 op 	size_t i;
 929 fb1a36c0 2022-01-09 op 
 930 fb1a36c0 2022-01-09 op 	if (argc == 0)
 931 fb1a36c0 2022-01-09 op 		return;
 932 fb1a36c0 2022-01-09 op 	for (i = 0; i < nitems(cmds); ++i) {
 933 fb1a36c0 2022-01-09 op 		if (!strcmp(cmds[i].name, *argv)) {
 934 fb1a36c0 2022-01-09 op 			cmds[i].fn(argc-1, argv+1);
 935 fb1a36c0 2022-01-09 op 			return;
 936 fb1a36c0 2022-01-09 op 		}
 937 fb1a36c0 2022-01-09 op 	}
 938 fb1a36c0 2022-01-09 op 
 939 fb1a36c0 2022-01-09 op 	log_warnx("unknown command %s", *argv);
 940 fb1a36c0 2022-01-09 op }
 941 fb1a36c0 2022-01-09 op 
 942 fb1a36c0 2022-01-09 op int
 943 fb1a36c0 2022-01-09 op main(int argc, char **argv)
 944 fb1a36c0 2022-01-09 op {
 945 fb1a36c0 2022-01-09 op 	int	ch;
 946 fb1a36c0 2022-01-09 op 
 947 fb1a36c0 2022-01-09 op 	log_init(1, LOG_DAEMON);
 948 fb1a36c0 2022-01-09 op 	log_setverbose(0);
 949 fb1a36c0 2022-01-09 op 	log_procinit(getprogname());
 950 fb1a36c0 2022-01-09 op 
 951 fb1a36c0 2022-01-09 op 	while ((ch = getopt(argc, argv, "C:cK:")) != -1) {
 952 fb1a36c0 2022-01-09 op 		switch (ch) {
 953 fb1a36c0 2022-01-09 op 		case 'C':
 954 fb1a36c0 2022-01-09 op 			crtpath = optarg;
 955 fb1a36c0 2022-01-09 op 			break;
 956 fb1a36c0 2022-01-09 op 		case 'c':
 957 fb1a36c0 2022-01-09 op 			tls = 1;
 958 fb1a36c0 2022-01-09 op 			break;
 959 fb1a36c0 2022-01-09 op 		case 'K':
 960 fb1a36c0 2022-01-09 op 			keypath = optarg;
 961 fb1a36c0 2022-01-09 op 			break;
 962 fb1a36c0 2022-01-09 op 		default:
 963 fb1a36c0 2022-01-09 op 			usage(1);
 964 fb1a36c0 2022-01-09 op 		}
 965 fb1a36c0 2022-01-09 op 	}
 966 fb1a36c0 2022-01-09 op 	argc -= optind;
 967 fb1a36c0 2022-01-09 op 	argv += optind;
 968 fb1a36c0 2022-01-09 op 
 969 fb1a36c0 2022-01-09 op 	if (argc == 0)
 970 fb1a36c0 2022-01-09 op 		usage(1);
 971 fb1a36c0 2022-01-09 op 
 972 fb1a36c0 2022-01-09 op 	if (isatty(1)) {
 973 fb1a36c0 2022-01-09 op 		tty_p = 1;
 974 fb1a36c0 2022-01-09 op 		resized = 1;
 975 fb1a36c0 2022-01-09 op 		signal(SIGWINCH, tty_resized);
 976 fb1a36c0 2022-01-09 op 	}
 977 fb1a36c0 2022-01-09 op 
 978 fb1a36c0 2022-01-09 op 	if ((evb = evbuffer_new()) == NULL)
 979 fb1a36c0 2022-01-09 op 		fatal("evbuffer_new");
 980 fb1a36c0 2022-01-09 op 
 981 fb1a36c0 2022-01-09 op 	if ((buf = evbuffer_new()) == NULL)
 982 fb1a36c0 2022-01-09 op 		fatal("evbuffer_new");
 983 fb1a36c0 2022-01-09 op 
 984 fb1a36c0 2022-01-09 op 	if ((dirbuf = evbuffer_new()) == NULL)
 985 fb1a36c0 2022-01-09 op 		fatal("evbuferr_new");
 986 fb1a36c0 2022-01-09 op 
 987 fb1a36c0 2022-01-09 op 	do_connect(argv[0], argv[1]);
 988 fb1a36c0 2022-01-09 op 
 989 fb1a36c0 2022-01-09 op 	for (;;) {
 990 fb1a36c0 2022-01-09 op 		int argc = 0;
 991 fb1a36c0 2022-01-09 op 		char *line, *argv[16] = {0}, **ap;
 992 fb1a36c0 2022-01-09 op 
 993 fb1a36c0 2022-01-09 op 		if ((line = read_line("kamiftp> ")) == NULL)
 994 fb1a36c0 2022-01-09 op 			break;
 995 fb1a36c0 2022-01-09 op 
 996 fb1a36c0 2022-01-09 op 		for (argc = 0, ap = argv; ap < &argv[15] &&
 997 fb1a36c0 2022-01-09 op 		    (*ap = strsep(&line, " \t")) != NULL;) {
 998 fb1a36c0 2022-01-09 op 			if (**ap != '\0')
 999 fb1a36c0 2022-01-09 op 				ap++, argc++;
1000 fb1a36c0 2022-01-09 op 		}
1001 fb1a36c0 2022-01-09 op 		excmd(argc, (const char **)argv);
1002 fb1a36c0 2022-01-09 op 
1003 fb1a36c0 2022-01-09 op 		if (bell) {
1004 fb1a36c0 2022-01-09 op 			printf("\a");
1005 fb1a36c0 2022-01-09 op 			fflush(stdout);
1006 fb1a36c0 2022-01-09 op 		}
1007 fb1a36c0 2022-01-09 op 
1008 fb1a36c0 2022-01-09 op 		free(line);
1009 fb1a36c0 2022-01-09 op 	}
1010 fb1a36c0 2022-01-09 op 
1011 fb1a36c0 2022-01-09 op 	printf("\n");
1012 fb1a36c0 2022-01-09 op }