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