op public repos

Blob

Date:: Mon May 23 13:24:50 2022 UTC
Message:: do optimal (i.e. maximum) reads and writes
Actions:: History | Blame | Raw File
1 /*
2  * Copyright (c) 2021, 2022 Omar Polo <op@omarpolo.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include "compat.h"
18 
19 #include <sys/ioctl.h>
20 #include <sys/types.h>
21 #include <sys/socket.h>
22 #include <sys/stat.h>
23 #include <sys/wait.h>
24 
25 #include <assert.h>
26 #include <errno.h>
27 #include <fcntl.h>
28 #include <inttypes.h>
29 #include <netdb.h>
30 #include <libgen.h>
31 #include <limits.h>
32 #include <signal.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <syslog.h>
37 #include <tls.h>
38 #include <unistd.h>
39 
40 #ifdef HAVE_READLINE
41 #include <readline/readline.h>
42 #include <readline/history.h>
43 #endif
44 
45 #ifndef nitems
46 #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
47 #endif
48 
49 #include "kami.h"
50 #include "utils.h"
51 #include "log.h"
52 #include "9pclib.h"
53 
54 #define TMPFSTR		"/tmp/kamiftp.XXXXXXXXXX"
55 #define TMPFSTRLEN	sizeof(TMPFSTR)
56 
57 /* flags */
58 int		 tls;
59 const char	*crtpath;
60 const char	*keypath;
61 
62 /* state */
63 struct tls_config	*tlsconf;
64 struct tls		*ctx;
65 int			 sock;
66 struct evbuffer		*buf;
67 struct evbuffer		*dirbuf;
68 uint32_t		 msize;
69 int			 bell;
70 
71 volatile sig_atomic_t	 resized;
72 int			 tty_p;
73 int			 tty_width;
74 int			 xdump;
75 
76 struct progress {
77 	uint64_t	max;
78 	uint64_t	done;
79 };
80 
81 int pwdfid;
82 
83 #define ASSERT_EMPTYBUF() assert(EVBUFFER_LENGTH(buf) == 0)
84 
85 #if !HAVE_READLINE
86 char *
87 readline(const char *prompt)
88 {
89 	char *ch, *line = NULL;
90 	size_t linesize = 0;
91 	ssize_t linelen;
92 
93 	printf("%s", prompt);
94 	fflush(stdout);
95 
96 	linelen = getline(&line, &linesize, stdin);
97 	if (linelen == -1)
98 		return NULL;
99 
100 	if ((ch = strchr(line, '\n')) != NULL)
101 		*ch = '\0';
102 	return line;
103 }
104 
105 void
106 add_history(const char *line)
107 {
108 	return;
109 }
110 #endif
111 
112 static char *
113 read_line(const char *prompt)
114 {
115 	char *line;
116 
117 again:
118 	if ((line = readline(prompt)) == NULL)
119 		return NULL;
120 	/* XXX: trim spaces? */
121 	if (*line == '\0') {
122 		free(line);
123 		goto again;
124 	}
125 
126 	add_history(line);
127 	return line;
128 }
129 
130 static void
131 spawn(const char *argv0, ...)
132 {
133 	pid_t pid;
134 	size_t i;
135 	int status;
136 	const char *argv[16], *last;
137 	va_list ap;
138 
139 	memset(argv, 0, sizeof(argv));
140 
141 	va_start(ap, argv0);
142 	argv[0] = argv0;
143 	for (i = 1; i < nitems(argv); ++i) {
144 		last = va_arg(ap, const char *);
145 		if (last == NULL)
146 			break;
147 		argv[i] = last;
148 	}
149 	va_end(ap);
150 
151 	assert(last == NULL);
152 
153 	switch (pid = fork()) {
154 	case -1:
155 		err(1, "fork");
156 	case 0: /* child */
157 		execvp(argv[0], (char *const *)argv);
158 		err(1, "execvp");
159 	default:
160 		waitpid(pid, &status, 0);
161 	}
162 }
163 
164 static void
165 tty_resized(int signo)
166 {
167 	resized = 1;
168 }
169 
170 static void __dead
171 usage(int ret)
172 {
173 	fprintf(stderr, "usage: %s [-c] [-C cert] [-K key] "
174 	    "host[:port] [path]\n", getprogname());
175 	fprintf(stderr, "kamid suite version " KAMID_VERSION "\n");
176 	exit(ret);
177 }
178 
179 static int
180 nextfid(void)
181 {
182 	uint32_t i;
183 
184 	for (i = 0; ; ++i) {
185 		if (i != pwdfid)
186 			return i;
187 	}
188 }
189 
190 static void
191 do_send(void)
192 {
193 	const void	*buf;
194 	size_t		 nbytes;
195 	ssize_t		 r;
196 
197 	if (xdump)
198 		hexdump("outgoing message", EVBUFFER_DATA(evb),
199 		    EVBUFFER_LENGTH(evb));
200 
201 	while (EVBUFFER_LENGTH(evb) != 0) {
202 		buf = EVBUFFER_DATA(evb);
203 		nbytes = EVBUFFER_LENGTH(evb);
204 
205 		if (ctx == NULL) {
206 			r = write(sock, buf, nbytes);
207 			if (r == 0 || r == -1)
208 				errx(1, "EOF");
209 		} else {
210 			r = tls_write(ctx, buf, nbytes);
211 			if (r == TLS_WANT_POLLIN || r == TLS_WANT_POLLOUT)
212 				continue;
213 			if (r == -1)
214 				errx(1, "tls: %s", tls_error(ctx));
215 		}
216 
217 		evbuffer_drain(evb, r);
218 	}
219 }
220 
221 static void
222 mustread(void *d, size_t len)
223 {
224 	ssize_t r;
225 
226 	while (len != 0) {
227 		if (ctx == NULL) {
228 			r = read(sock, d, len);
229 			if (r == 0 || r == -1)
230 				errx(1, "EOF");
231 		} else {
232 			r = tls_read(ctx, d, len);
233 			if (r == TLS_WANT_POLLIN || r == TLS_WANT_POLLOUT)
234 				continue;
235 			if (r == -1)
236 				errx(1, "tls: %s", tls_error(ctx));
237 		}
238 
239 		d += r;
240 		len -= r;
241 	}
242 }
243 
244 static void
245 recv_msg(void)
246 {
247 	uint32_t	len, l;
248 	char		tmp[BUFSIZ];
249 
250 	mustread(&len, sizeof(len));
251 	len = le32toh(len);
252 	if (len < HEADERSIZE)
253 		errx(1, "read message of invalid length %d", len);
254 
255 	len -= 4; /* skip the length just read */
256 
257 	while (len != 0) {
258 		l = MIN(len, sizeof(tmp));
259 		mustread(tmp, l);
260 		len -= l;
261 		evbuffer_add(buf, tmp, l);
262 	}
263 
264 	if (xdump)
265 		hexdump("incoming packet", EVBUFFER_DATA(buf),
266 		    EVBUFFER_LENGTH(buf));
267 }
268 
269 static uint64_t
270 np_read64(struct evbuffer *buf)
271 {
272 	uint64_t n;
273 
274 	evbuffer_remove(buf, &n, sizeof(n));
275 	return le64toh(n);
276 }
277 
278 static uint32_t
279 np_read32(struct evbuffer *buf)
280 {
281 	uint32_t n;
282 
283 	evbuffer_remove(buf, &n, sizeof(n));
284 	return le32toh(n);
285 }
286 
287 static uint16_t
288 np_read16(struct evbuffer *buf)
289 {
290 	uint16_t n;
291 
292 	evbuffer_remove(buf, &n, sizeof(n));
293 	return le16toh(n);
294 }
295 
296 static uint16_t
297 np_read8(struct evbuffer *buf)
298 {
299 	uint8_t n;
300 
301 	evbuffer_remove(buf, &n, sizeof(n));
302 	return n;
303 }
304 
305 static char *
306 np_readstr(struct evbuffer *buf)
307 {
308 	uint16_t	 len;
309 	char		*str;
310 
311 	len = np_read16(buf);
312 	assert(EVBUFFER_LENGTH(buf) >= len);
313 
314 	if ((str = calloc(1, len+1)) == NULL)
315 		err(1, "calloc");
316 	evbuffer_remove(buf, str, len);
317 	return str;
318 }
319 
320 static void
321 np_read_qid(struct evbuffer *buf, struct qid *qid)
322 {
323 	assert(EVBUFFER_LENGTH(buf) >= QIDSIZE);
324 
325 	qid->type = np_read8(buf);
326 	qid->vers = np_read32(buf);
327 	qid->path = np_read64(buf);
328 }
329 
330 static int
331 np_read_stat(struct evbuffer *buf, struct np_stat *st)
332 {
333 	uint16_t size;
334 
335 	memset(st, 0, sizeof(*st));
336 
337 	size = np_read16(buf);
338 	if (size > EVBUFFER_LENGTH(buf))
339 		return -1;
340 
341 	st->type = np_read16(buf);
342 	st->dev = np_read32(buf);
343 	np_read_qid(buf, &st->qid);
344 	st->mode = np_read32(buf);
345 	st->atime = np_read32(buf);
346 	st->mtime = np_read32(buf);
347 	st->length = np_read64(buf);
348 	st->name = np_readstr(buf);
349 	st->uid = np_readstr(buf);
350 	st->gid = np_readstr(buf);
351 	st->muid = np_readstr(buf);
352 
353 	return 0;
354 }
355 
356 static void
357 expect(uint8_t type)
358 {
359 	uint8_t t;
360 
361 	t = np_read8(buf);
362 	if (t == type)
363 		return;
364 
365 	if (t == Rerror) {
366 		char *err;
367 
368 		/* skip tag */
369 		np_read16(buf);
370 
371 		err = np_readstr(buf);
372 		errx(1, "expected %s, got error %s",
373 		    pp_msg_type(type), err);
374 	}
375 
376 	errx(1, "expected %s, got msg type %s",
377 	    pp_msg_type(type), pp_msg_type(t));
378 }
379 
380 static void
381 expect2(uint8_t type, uint16_t tag)
382 {
383 	uint16_t t;
384 
385 	expect(type);
386 
387 	t = np_read16(buf);
388 	if (t == tag)
389 		return;
390 
391 	errx(1, "expected tag 0x%x, got 0x%x", tag, t);
392 }
393 
394 static char *
395 check(uint8_t type, uint16_t tag)
396 {
397 	uint16_t rtag;
398 	uint8_t rtype;
399 
400 	rtype = np_read8(buf);
401 	rtag = np_read16(buf);
402 	if (rtype == type) {
403 		if (rtag != tag)
404 			errx(1, "expected tag 0x%x, got 0x%x", tag, rtag);
405 		return NULL;
406 	}
407 
408 	if (rtype == Rerror)
409 		return np_readstr(buf);
410 
411 	errx(1, "expected %s, got msg type %s",
412 	    pp_msg_type(type), pp_msg_type(rtype));
413 }
414 
415 static void
416 do_version(void)
417 {
418 	char		*version;
419 
420 	tversion(VERSION9P, MSIZE9P);
421 	do_send();
422 	recv_msg();
423 	expect2(Rversion, NOTAG);
424 
425 	msize = np_read32(buf);
426 	version = np_readstr(buf);
427 
428 	if (msize > MSIZE9P || msize < 256)
429 		errx(1, "got unexpected msize: %d", msize);
430 	if (strcmp(version, VERSION9P))
431 		errx(1, "unexpected 9p version: %s", version);
432 
433 	free(version);
434 	ASSERT_EMPTYBUF();
435 }
436 
437 static void
438 do_attach(const char *path)
439 {
440 	const char *user;
441 	struct qid qid;
442 
443 	if (path == NULL)
444 		path = "/";
445 	if ((user = getenv("USER")) == NULL)
446 		user = "flan";
447 
448 	tattach(pwdfid, NOFID, user, path);
449 	do_send();
450 	recv_msg();
451 	expect2(Rattach, iota_tag);
452 	np_read_qid(buf, &qid);
453 
454 	ASSERT_EMPTYBUF();
455 }
456 
457 static uint32_t
458 do_open(uint32_t fid, uint8_t mode)
459 {
460 	struct qid qid;
461 	uint32_t iounit;
462 
463 	topen(fid, mode);
464 	do_send();
465 	recv_msg();
466 	expect2(Ropen, iota_tag);
467 
468 	np_read_qid(buf, &qid);
469 	iounit = np_read32(buf);
470 
471 	ASSERT_EMPTYBUF();
472 
473 	return iounit;
474 }
475 
476 static uint32_t
477 do_create(uint32_t fid, const char *name, uint32_t perm, uint8_t mode)
478 {
479 	struct qid qid;
480 	uint32_t iounit;
481 
482 	tcreate(fid, name, perm, mode);
483 	do_send();
484 	recv_msg();
485 	expect2(Rcreate, iota_tag);
486 
487 	np_read_qid(buf, &qid);
488 	iounit = np_read32(buf);
489 
490 	ASSERT_EMPTYBUF();
491 
492 	return iounit;
493 }
494 
495 static void
496 do_clunk(uint32_t fid)
497 {
498 	tclunk(fid);
499 	do_send();
500 	recv_msg();
501 	expect2(Rclunk, iota_tag);
502 
503 	ASSERT_EMPTYBUF();
504 }
505 
506 static char *
507 dup_fid(int fid, int nfid)
508 {
509 	uint16_t nwqid;
510 	char *errstr;
511 
512 	twalk(fid, nfid, NULL, 0);
513 	do_send();
514 	recv_msg();
515 
516 	if ((errstr = check(Rwalk, iota_tag)) != NULL)
517 		return errstr;
518 
519 	nwqid = np_read16(buf);
520 	assert(nwqid == 0);
521 
522 	ASSERT_EMPTYBUF();
523 
524 	return NULL;
525 }
526 
527 static char *
528 walk_path(int fid, int newfid, const char *path, int *missing,
529     struct qid *qid)
530 {
531 	char *wnames[MAXWELEM], *p, *t, *errstr;
532 	size_t nwname, i;
533 	uint16_t nwqid;
534 
535 	if ((p = strdup(path)) == NULL)
536 		err(1, "strdup");
537 	t = p;
538 
539 	/* strip initial ./ */
540 	if (t[0] == '.' && t[1] == '/')
541 		t += 2;
542 
543 	for (nwname = 0; nwname < nitems(wnames) &&
544 	    (wnames[nwname] = strsep(&t, "/")) != NULL;) {
545 		if (*wnames[nwname] != '\0')
546 			nwname++;
547 	}
548 
549 	twalk(fid, newfid, (const char **)wnames, nwname);
550 	do_send();
551 	recv_msg();
552 
553 	*missing = nwname;
554 	if ((errstr = check(Rwalk, iota_tag)) != NULL) {
555 		free(p);
556 		return errstr;
557 	}
558 
559 	nwqid = np_read16(buf);
560 	assert(nwqid <= nwname);
561 
562 	/* consume all qids */
563 	for (i = 0; i < nwqid; ++i)
564 		np_read_qid(buf, qid);
565 
566 	free(p);
567 
568 	*missing = nwname - nwqid;
569 	return NULL;
570 }
571 
572 static void
573 do_stat(int fid, struct np_stat *st)
574 {
575 	tstat(fid);
576 	do_send();
577 	recv_msg();
578 	expect2(Rstat, iota_tag);
579 
580 	/* eat up the first two byte length */
581 	np_read16(buf);
582 
583 	if (np_read_stat(buf, st) == -1)
584 		errx(1, "invalid stat struct read");
585 
586 	ASSERT_EMPTYBUF();
587 }
588 
589 static char *
590 do_wstat(int fid, const struct np_stat *st)
591 {
592 	char *errstr;
593 
594 	twstat(fid, st);
595 	do_send();
596 	recv_msg();
597 
598 	if ((errstr = check(Rwstat, iota_tag)) != NULL)
599 		return errstr;
600 
601 	ASSERT_EMPTYBUF();
602 
603 	return NULL;
604 }
605 
606 static size_t
607 do_read(int fid, uint64_t off, uint32_t count, void *data)
608 {
609 	uint32_t r;
610 
611 	tread(fid, off, count);
612 	do_send();
613 	recv_msg();
614 	expect2(Rread, iota_tag);
615 
616 	r = np_read32(buf);
617 	assert(r == EVBUFFER_LENGTH(buf));
618 	assert(r <= count);
619 	evbuffer_remove(buf, data, r);
620 
621 	ASSERT_EMPTYBUF();
622 
623 	return r;
624 }
625 
626 static size_t
627 do_write(int fid, uint64_t off, uint32_t count, void *data)
628 {
629 	uint32_t r;
630 
631 	twrite(fid, off, data, count);
632 	do_send();
633 	recv_msg();
634 	expect2(Rwrite, iota_tag);
635 
636 	r = np_read32(buf);
637 	assert(r <= count);
638 
639 	ASSERT_EMPTYBUF();
640 
641 	return r;
642 }
643 
644 static void
645 draw_progress(const char *pre, const struct progress *p)
646 {
647 	struct winsize ws;
648 	int i, l, w;
649 	double perc;
650 
651 	if (xdump)
652 		return;
653 
654 	perc = 100.0 * p->done / p->max;
655 	if (!tty_p) {
656 		fprintf(stderr, "%s: %d%%\n", pre, (int)perc);
657 		return;
658 	}
659 
660 	if (resized) {
661 		resized = 0;
662 
663 		if (ioctl(0, TIOCGWINSZ, &ws) == -1)
664 			return;
665 		tty_width = ws.ws_col;
666 	}
667 	w = tty_width;
668 
669 	if (pre == NULL ||
670 	    ((l = printf("\r%s ", pre)) == -1 || l >= w))
671 		return;
672 
673 	w -= l + 2 + 5; /* 2 for |, 5 for percentage + \n */
674 	if (w < 0) {
675 		printf("%4d%%\n", (int)perc);
676 		return;
677 	}
678 
679 	printf("|");
680 
681 	l = w * MIN(100.0, perc) / 100.0;
682 	for (i = 0; i < l; i++)
683 		printf("*");
684 	for (; i < w; i++)
685 		printf(" ");
686 	printf("|%4d%%", (int)perc);
687 
688 	fflush(stdout);
689 }
690 
691 static int
692 fetch_fid(int fid, int fd, const char *name)
693 {
694 	static char buf[MSIZE9P];
695 	struct progress p = {0};
696 	struct np_stat st;
697 	size_t r;
698 	int ret = 0;
699 
700 	do_stat(fid, &st);
701 	do_open(fid, KOREAD);
702 
703 	p.max = st.length;
704 	for (;;) {
705 		size_t len, off;
706 		ssize_t nw;
707 
708 		len = MIN(sizeof(buf), msize);
709 		len -= HEADERSIZE + 4; /* for the request' fields */
710 
711 		r = do_read(fid, p.done, len, buf);
712 		if (r == 0)
713 			break;
714 
715 		for (off = 0; off < r; off += nw)
716 			if ((nw = write(fd, buf + off, r - off)) == 0 ||
717 			    nw == -1) {
718 				ret = -1;
719 				goto end;
720 			}
721 
722 		p.done += r;
723 		draw_progress(name, &p);
724 
725 #if 0
726 		/* throttle, for debugging purpose */
727 		{
728 			struct timespec ts = { 0, 500000000 };
729 			nanosleep(&ts, NULL);
730 		}
731 #endif
732 	}
733 
734 end:
735 	putchar('\n');
736 
737 	do_clunk(fid);
738 	return ret;
739 }
740 
741 static void
742 send_fid(int fid, const char *fnam, int open_flags, int fd, const char *name)
743 {
744 	static char buf[MSIZE9P];
745 	struct progress p = {0};
746 	struct stat sb;
747 	ssize_t r;
748 	size_t w, len;
749 
750 	if (fstat(fd, &sb) == -1)
751 		err(1, "fstat");
752 
753 	if (fnam != NULL)
754 		do_create(fid, fnam, 0644, KOWRITE);
755 	else
756 		do_open(fid, open_flags | KOWRITE);
757 
758 	p.max = sb.st_size;
759 	for (;;) {
760 		len = MIN(sizeof(buf), msize);
761 		len -= HEADERSIZE + 4 + 4 + 8; /* for the request' fields */
762 
763 		r = read(fd, buf, len);
764 		if (r == 0)
765 			break;
766 		if (r == -1)
767 			err(1, "read");
768 
769 		w = do_write(fid, p.done, r, buf);
770 		p.done += w;
771 
772 		draw_progress(name, &p);
773 
774 #if 0
775 		/* throttle, for debugging purpose */
776 		{
777 			struct timespec ts = { 0, 500000000 };
778 			nanosleep(&ts, NULL);
779 		}
780 #endif
781 	}
782 
783 	putchar('\n');
784 	do_clunk(fid);
785 }
786 
787 static int
788 woc_file(int fd, const char *prompt, const char *path)
789 {
790 	struct qid qid;
791 	const char *n = NULL;
792 	char *errstr;
793 	int nfid, miss;
794 
795 	nfid = nextfid();
796 	errstr = walk_path(pwdfid, nfid, path, &miss, &qid);
797 	if (errstr != NULL && miss > 1) {
798 		printf("%s: %s\n", path, errstr);
799 		free(errstr);
800 		return -1;
801 	}
802 
803 	if (errstr != NULL || miss == 1) {
804 		char p[PATH_MAX], *dn;
805 
806 		/*
807 		 * If it's only one component missing (the file name), walk
808 		 * to the parent directory and try to create the file.
809 		 */
810 
811 		if (strlcpy(p, path, sizeof(p)) >= sizeof(p)) {
812 			printf("path too long: %s\n", path);
813 			return -1;
814 		}
815 		dn = dirname(p);
816 
817 		if (!strcmp(dn, ".")) {
818 			errstr = dup_fid(pwdfid, nfid);
819 			miss = 0;
820 		} else
821 			errstr = walk_path(pwdfid, nfid, dn, &miss, &qid);
822 
823 		if (errstr != NULL) {
824 			printf("%s: %s\n", dn, errstr);
825 			free(errstr);
826 			return -1;
827 		}
828 
829 		if (miss != 0) {
830 			printf("%s: not a directory\n", dn);
831 			return -1;
832 		}
833 
834 		if ((n = strrchr(path, '/')) != NULL)
835 			n++;
836 		else
837 			n = path;
838 	}
839 
840 	free(errstr);
841 
842 	if (miss > 1) {
843 		printf("can't create %s: missing %d path component(s)\n",
844 		    path, miss);
845 		return -1;
846 	}
847 
848 	send_fid(nfid, n, KOTRUNC, fd, prompt);
849 	return 0;
850 }
851 
852 static void
853 do_tls_connect(const char *host, const char *port)
854 {
855 	int handshake;
856 
857 	if ((tlsconf = tls_config_new()) == NULL)
858 		fatalx("tls_config_new");
859 	tls_config_insecure_noverifycert(tlsconf);
860 	tls_config_insecure_noverifyname(tlsconf);
861 
862 	if (keypath == NULL)
863 		keypath = crtpath;
864 
865 	if (tls_config_set_keypair_file(tlsconf, crtpath, keypath) == -1)
866 		fatalx("can't load certs (%s, %s)", crtpath, keypath);
867 
868 	if ((ctx = tls_client()) == NULL)
869 		fatal("tls_client");
870 	if (tls_configure(ctx, tlsconf) == -1)
871 		fatalx("tls_configure: %s", tls_error(ctx));
872 
873 	if (tls_connect(ctx, host, port) == -1)
874 		fatalx("can't connect to %s:%s: %s", host, port,
875 		    tls_error(ctx));
876 
877 	for (handshake = 0; !handshake;) {
878 		switch (tls_handshake(ctx)) {
879 		case -1:
880 			fatalx("tls_handshake: %s", tls_error(ctx));
881 		case 0:
882 			handshake = 1;
883 			break;
884 		}
885 	}
886 }
887 
888 static void
889 do_ctxt_connect(const char *host, const char *port)
890 {
891 	struct addrinfo hints, *res, *res0;
892 	int error, saved_errno;
893 	const char *cause = NULL;
894 
895 	memset(&hints, 0, sizeof(hints));
896 	hints.ai_family = AF_UNSPEC;
897 	hints.ai_socktype = SOCK_STREAM;
898 	error = getaddrinfo(host, port, &hints, &res0);
899 	if (error)
900 		errx(1, "%s", gai_strerror(error));
901 
902 	sock = -1;
903 	for (res = res0; res != NULL; res = res->ai_next) {
904 		sock = socket(res->ai_family, res->ai_socktype|SOCK_CLOEXEC,
905 		    res->ai_protocol);
906 		if (sock == -1) {
907 			cause = "socket";
908 			continue;
909 		}
910 
911 		if (connect(sock, res->ai_addr, res->ai_addrlen) == -1) {
912 			cause = "connect";
913 			saved_errno = errno;
914 			close(sock);
915 			errno = saved_errno;
916 			sock = -1;
917 			continue;
918 		}
919 
920 		break;
921 	}
922 
923 	if (sock == -1)
924 		err(1, "%s", cause);
925 	freeaddrinfo(res0);
926 }
927 
928 static void
929 do_connect(const char *connspec, const char *path)
930 {
931 	char *host, *colon;
932 	const char *port;
933 
934 	host = xstrdup(connspec);
935 	if ((colon = strchr(host, ':')) != NULL) {
936 		*colon = '\0';
937 		port = ++colon;
938 	} else
939 		port = "1337";
940 
941 	printf("connecting to %s:%s...", host, port);
942 	fflush(stdout);
943 
944 	if (tls)
945 		do_tls_connect(host, port);
946 	else
947 		do_ctxt_connect(host, port);
948 
949 	printf(" done!\n");
950 
951 	do_version();
952 	do_attach(path);
953 
954 	free(host);
955 }
956 
957 static int
958 tmp_file(char sfn[TMPFSTRLEN])
959 {
960 	int tmpfd;
961 
962 	strlcpy(sfn, TMPFSTR, TMPFSTRLEN);
963 	if ((tmpfd = mkstemp(sfn)) == -1) {
964 		warn("mkstemp %s", sfn);
965 		return -1;
966 	}
967 
968 	/* set the close-on-exec flag */
969 	if (fcntl(tmpfd, F_SETFD, FD_CLOEXEC) == -1) {
970 		warn("fcntl");
971 		close(tmpfd);
972 		return -1;
973 	}
974 
975 	return tmpfd;
976 }
977 
978 static inline const char *
979 pp_perm(uint8_t x)
980 {
981 	switch (x & 0x7) {
982 	case 0x0:
983 		return "---";
984 	case 0x1:
985 		return "--x";
986 	case 0x2:
987 		return "-w-";
988 	case 0x3:
989 		return "-wx";
990 	case 0x4:
991 		return "r--";
992 	case 0x5:
993 		return "r-x";
994 	case 0x6:
995 		return "rw-";
996 	case 0x7:
997 		return "rwx";
998 	default:
999 		/* unreachable, just for the compiler' happiness */
1000 		return "???";
1001 	}
1002 }
1003 
1004 static inline void
1005 prepare_wstat(struct np_stat *st)
1006 {
1007 	memset(st, 0xFF, sizeof(*st));
1008 	st->name = NULL;
1009 	st->uid = NULL;
1010 	st->gid = NULL;
1011 	st->muid = NULL;
1012 }
1013 
1014 static void
1015 cmd_bell(int argc, const char **argv)
1016 {
1017 	if (argc == 0) {
1018 		bell = !bell;
1019 		if (bell)
1020 			puts("bell mode enabled");
1021 		else
1022 			puts("bell mode disabled");
1023 		return;
1024 	}
1025 
1026 	if (argc != 1)
1027 		goto usage;
1028 
1029 	if (!strcmp(*argv, "on")) {
1030 		bell = 1;
1031 		puts("bell mode enabled");
1032 		return;
1033 	}
1034 
1035 	if (!strcmp(*argv, "off")) {
1036 		bell = 0;
1037 		puts("bell mode disabled");
1038 		return;
1039 	}
1040 
1041 usage:
1042 	printf("bell [on | off]\n");
1043 }
1044 
1045 static void
1046 cmd_bye(int argc, const char **argv)
1047 {
1048 	log_warnx("bye\n");
1049 	exit(0);
1050 }
1051 
1052 static void
1053 cmd_cd(int argc, const char **argv)
1054 {
1055 	struct qid qid;
1056 	int nfid, miss;
1057 	char *errstr;
1058 
1059 	if (argc != 1) {
1060 		printf("usage: cd remote-path\n");
1061 		return;
1062 	}
1063 
1064 	nfid = nextfid();
1065 	errstr = walk_path(pwdfid, nfid, argv[0], &miss, &qid);
1066 	if (errstr != NULL) {
1067 		printf("%s: %s\n", argv[0], errstr);
1068 		free(errstr);
1069 		return;
1070 	}
1071 
1072 	if (miss != 0 || !(qid.type & QTDIR)) {
1073 		printf("%s: not a directory\n", argv[0]);
1074 		if (miss == 0)
1075 			do_clunk(nfid);
1076 		return;
1077 	}
1078 
1079 	do_clunk(pwdfid);
1080 	pwdfid = nfid;
1081 }
1082 
1083 static void
1084 cmd_edit(int argc, const char **argv)
1085 {
1086 	struct qid qid;
1087 	int nfid, tmpfd, miss;
1088 	char sfn[TMPFSTRLEN], p[PATH_MAX], *name, *errstr;
1089 	const char *ed;
1090 
1091 	if (argc != 1) {
1092 		puts("usage: edit file");
1093 		return;
1094 	}
1095 
1096 	if ((ed = getenv("VISUAL")) == NULL &&
1097 	    (ed = getenv("EDITOR")) == NULL)
1098 		ed = "ed";
1099 
1100 	nfid = nextfid();
1101 	errstr = walk_path(pwdfid, nfid, *argv, &miss, &qid);
1102 	if (errstr != NULL) {
1103 		printf("%s: %s\n", *argv, errstr);
1104 		free(errstr);
1105 		return;
1106 	}
1107 
1108 	if (miss != 0 || qid.type != 0) {
1109 		printf("%s: not a file\n", *argv);
1110 		if (miss == 0)
1111 			do_clunk(nfid);
1112 		return;
1113 	}
1114 
1115 	if ((tmpfd = tmp_file(sfn)) == -1) {
1116 		do_clunk(nfid);
1117 		return;
1118 	}
1119 
1120 	strlcpy(p, *argv, sizeof(p));
1121 	name = basename(p);
1122 
1123 	if (fetch_fid(nfid, tmpfd, name)) {
1124 		warn("failed fetch or can't write %s", sfn);
1125 		goto end;
1126 	}
1127 	close(tmpfd);
1128 
1129 	spawn(ed, sfn, NULL);
1130 
1131 	/*
1132 	 * Re-open the file because it's not guaranteed that the
1133 	 * file descriptor tmpfd is still associated with the file
1134 	 * pointed by sfn: it's not uncommon for editor to write
1135 	 * a backup file and then rename(2) it to the file name.
1136 	 */
1137 	if ((tmpfd = open(sfn, O_RDONLY)) == -1) {
1138 		warn("can't open %s", sfn);
1139 		goto end;
1140 	}
1141 
1142 	woc_file(tmpfd, *argv, name);
1143 	close(tmpfd);
1144 
1145 end:
1146 	unlink(sfn);
1147 }
1148 
1149 static void
1150 cmd_get(int argc, const char **argv)
1151 {
1152 	struct qid qid;
1153 	const char *l;
1154 	char *errstr;
1155 	int nfid, fd, miss;
1156 
1157 	if (argc != 1 && argc != 2) {
1158 		printf("usage: get remote-file [local-file]\n");
1159 		return;
1160 	}
1161 
1162 	if (argc == 2)
1163 		l = argv[1];
1164 	else if ((l = strrchr(argv[0], '/')) != NULL)
1165 		l++; /* skip / */
1166 	else
1167 		l = argv[0];
1168 
1169 	nfid = nextfid();
1170 	errstr = walk_path(pwdfid, nfid, argv[0], &miss, &qid);
1171 	if (errstr != NULL) {
1172 		printf("%s: %s\n", argv[0], errstr);
1173 		free(errstr);
1174 		return;
1175 	}
1176 
1177 	if (miss != 0 || qid.type != 0) {
1178 		printf("%s: not a file\n", argv[0]);
1179 		if (miss == 0)
1180 			do_clunk(nfid);
1181 		return;
1182 	}
1183 
1184 	if ((fd = open(l, O_WRONLY|O_CREAT|O_TRUNC|O_CLOEXEC, 0644)) == -1) {
1185 		warn("can't open %s", l);
1186 		do_clunk(nfid);
1187 		return;
1188 	}
1189 
1190 	if (fetch_fid(nfid, fd, l) == -1)
1191 		warn("write %s", l);
1192 	close(fd);
1193 }
1194 
1195 static void
1196 cmd_hexdump(int argc, const char **argv)
1197 {
1198 	if (argc == 0) {
1199 		xdump = !xdump;
1200 		if (xdump)
1201 			puts("hexdump mode enabled");
1202 		else
1203 			puts("hexdump mode disabled");
1204 		return;
1205 	}
1206 
1207 	if (argc > 1)
1208 		goto usage;
1209 
1210 	if (!strcmp(*argv, "on")) {
1211 		xdump = 1;
1212 		puts("hexdump mode enabled");
1213 		return;
1214 	}
1215 
1216 	if (!strcmp(*argv, "off")) {
1217 		xdump = 0;
1218 		puts("hexdump mode disabled");
1219 		return;
1220 	}
1221 
1222 usage:
1223 	puts("usage: hexdump [on | off]");
1224 }
1225 
1226 static void
1227 cmd_lcd(int argc, const char **argv)
1228 {
1229 	const char *dir;
1230 
1231 	if (argc > 1) {
1232 		printf("usage: lcd [local-directory]\n");
1233 		return;
1234 	}
1235 
1236 	if (argc == 1)
1237 		dir = *argv;
1238 
1239 	if (argc == 0 && (dir = getenv("HOME")) == NULL) {
1240 		printf("HOME is not defined\n");
1241 		return;
1242 	}
1243 
1244 	if (chdir(dir) == -1)
1245 		printf("cd: %s: %s\n", dir, strerror(errno));
1246 }
1247 
1248 static void
1249 cmd_lpwd(int argc, const char **argv)
1250 {
1251 	char path[PATH_MAX];
1252 
1253 	if (argc != 0) {
1254 		printf("usage: lpwd\n");
1255 		return;
1256 	}
1257 
1258 	if (getcwd(path, sizeof(path)) == NULL) {
1259 		printf("lpwd: %s\n", strerror(errno));
1260 		return;
1261 	}
1262 
1263 	printf("%s\n", path);
1264 }
1265 
1266 static void
1267 cmd_ls(int argc, const char **argv)
1268 {
1269 	struct np_stat st;
1270 	time_t now, mtime;
1271 	struct tm *tm;
1272 	uint64_t off = 0;
1273 	uint32_t len;
1274 	int nfid;
1275 	const char *timfmt;
1276 	char fmt[FMT_SCALED_STRSIZE], tim[13], *errstr;
1277 
1278 	if (argc != 0) {
1279 		printf("ls don't take arguments (yet)\n");
1280 		return;
1281 	}
1282 
1283 	now = time(NULL);
1284 
1285 	nfid = nextfid();
1286 	if ((errstr = dup_fid(pwdfid, nfid)) != NULL) {
1287 		printf(".: %s\n", errstr);
1288 		free(errstr);
1289 		return;
1290 	}
1291 
1292 	do_open(nfid, KOREAD);
1293 
1294 	evbuffer_drain(dirbuf, EVBUFFER_LENGTH(dirbuf));
1295 
1296 	for (;;) {
1297 		tread(nfid, off, msize - 4);
1298 		do_send();
1299 		recv_msg();
1300 		expect2(Rread, iota_tag);
1301 
1302 		len = np_read32(buf);
1303 		if (len == 0)
1304 			break;
1305 
1306 		evbuffer_add_buffer(dirbuf, buf);
1307 		off += len;
1308 
1309 		ASSERT_EMPTYBUF();
1310 	}
1311 
1312 	while (EVBUFFER_LENGTH(dirbuf) != 0) {
1313 		if (np_read_stat(dirbuf, &st) == -1)
1314 			errx(1, "invalid stat struct read");
1315 
1316 		if (fmt_scaled(st.length, fmt) == -1)
1317 			strlcpy(fmt, "xxx", sizeof(fmt));
1318 
1319 		mtime = st.mtime;
1320 
1321 		if (now > mtime && (now - mtime) < 365/2 * 24 * 12 * 60)
1322 			timfmt = "%b %e %R";
1323 		else
1324 			timfmt = "%b %e  %Y";
1325 
1326 		if ((tm = localtime(&mtime)) == NULL ||
1327 		    strftime(tim, sizeof(tim), timfmt, tm) == 0)
1328 			strlcpy(tim, "unknown", sizeof(tim));
1329 
1330 		if (st.qid.type & QTDIR)
1331 			printf("d");
1332 		else
1333 			printf("-");
1334 		printf("%s", pp_perm(st.mode >> 6));
1335 		printf("%s", pp_perm(st.mode >> 3));
1336 		printf("%s", pp_perm(st.mode));
1337 		printf(" %8s %12s %s%s\n", fmt, tim, st.name,
1338 		    st.qid.type & QTDIR ? "/" : "");
1339 
1340 		free(st.name);
1341 		free(st.uid);
1342 		free(st.gid);
1343 		free(st.muid);
1344 	}
1345 
1346 	do_clunk(nfid);
1347 }
1348 
1349 static void
1350 cmd_page(int argc, const char **argv)
1351 {
1352 	struct qid qid;
1353 	int nfid, tmpfd, miss, r;
1354 	char sfn[TMPFSTRLEN], p[PATH_MAX], *name, *errstr;
1355 	const char *pager;
1356 
1357 	if (argc != 1) {
1358 		puts("usage: page file");
1359 		return;
1360 	}
1361 
1362 	if ((pager = getenv("PAGER")) == NULL)
1363 		pager = "less";
1364 
1365 	nfid = nextfid();
1366 	errstr = walk_path(pwdfid, nfid, *argv, &miss, &qid);
1367 	if (errstr != NULL) {
1368 		printf("%s: %s\n", *argv, errstr);
1369 		free(errstr);
1370 		return;
1371 	}
1372 
1373 	if (miss != 0 || qid.type != 0) {
1374 		printf("%s: not a file\n", *argv);
1375 		if (miss == 0)
1376 			do_clunk(nfid);
1377 		return;
1378 	}
1379 
1380 	if ((tmpfd = tmp_file(sfn)) == -1) {
1381 		do_clunk(nfid);
1382 		return;
1383 	}
1384 
1385 	strlcpy(p, *argv, sizeof(p));
1386 	name = basename(p);
1387 	if ((r = fetch_fid(nfid, tmpfd, name)) == -1)
1388 		warn("write %s", sfn);
1389 	close(tmpfd);
1390 	if (r != -1)
1391 		spawn(pager, sfn, NULL);
1392 	unlink(sfn);
1393 }
1394 
1395 static void
1396 cmd_pipe(int argc, const char **argv)
1397 {
1398 	struct qid qid;
1399 	pid_t pid;
1400 	int nfid, tmpfd, miss, status;
1401 	int filedes[2]; /* read end, write end */
1402 	char *errstr;
1403 
1404 	if (argc < 2) {
1405 		puts("usage: pipe remote-file cmd [args...]");
1406 		return;
1407 	}
1408 
1409 	nfid = nextfid();
1410 	errstr = walk_path(pwdfid, nfid, *argv, &miss, &qid);
1411 	if (errstr != NULL) {
1412 		printf("%s: %s\n", *argv, errstr);
1413 		free(errstr);
1414 		return;
1415 	}
1416 
1417 	if (miss != 0 || qid.type != 0) {
1418 		printf("%s: not a file\n", *argv);
1419 		if (miss == 0)
1420 			do_clunk(nfid);
1421 		return;
1422 	}
1423 
1424 	if (pipe(filedes) == -1)
1425 		err(1, "pipe");
1426 
1427 	switch (pid = vfork()) {
1428 	case -1:
1429 		err(1, "vfork");
1430 	case 0:
1431 		close(filedes[1]);
1432 		if (dup2(filedes[0], 0) == -1)
1433 			err(1, "dup2");
1434 		execvp(argv[1], (char *const *)argv + 1);
1435 		err(1, "execvp");
1436 	}
1437 
1438 	close(filedes[0]);
1439 	if (fetch_fid(nfid, filedes[1], *argv) == -1)
1440 		warnx("failed to fetch all the file");
1441 	close(filedes[1]);
1442 
1443 	waitpid(pid, &status, 0);
1444 }
1445 
1446 static void
1447 cmd_put(int argc, const char **argv)
1448 {
1449 	struct qid qid;
1450 	const char *l;
1451 	int fd;
1452 
1453 	if (argc != 1 && argc != 2) {
1454 		printf("usage: put local-file [remote-file]\n");
1455 		return;
1456 	}
1457 
1458 	if (argc == 2)
1459 		l = argv[1];
1460 	else if ((l = strrchr(argv[0], '/')) != NULL)
1461 		l++; /* skip / */
1462 	else
1463 		l = argv[0];
1464 
1465 	if ((fd = open(argv[0], O_RDONLY)) == -1) {
1466 		warn("%s", argv[0]);
1467 		return;
1468 	}
1469 
1470 	woc_file(fd, argv[0], l);
1471 	close(fd);
1472 }
1473 
1474 static void
1475 cmd_rename(int argc, const char **argv)
1476 {
1477 	struct np_stat st;
1478 	struct qid qid;
1479 	char *errstr;
1480 	int nfid, miss;
1481 
1482 	if (argc != 2) {
1483 		puts("usage: rename remote-file new-remote-name");
1484 		return;
1485 	}
1486 
1487 	nfid = nextfid();
1488 	errstr = walk_path(pwdfid, nfid, argv[0], &miss, &qid);
1489 	if (errstr != NULL) {
1490 		printf("%s: %s\n", argv[0], errstr);
1491 		free(errstr);
1492 		return;
1493 	}
1494 
1495 	if (miss != 0) {
1496 		printf("%s: not such file or directory\n", argv[0]);
1497 		return;
1498 	}
1499 
1500 	prepare_wstat(&st);
1501 	st.name = (char *)argv[1];
1502 	if ((errstr = do_wstat(nfid, &st)) != NULL) {
1503 		printf("rename: %s\n", errstr);
1504 		free(errstr);
1505 	}
1506 
1507 	do_clunk(nfid);
1508 }
1509 
1510 static void
1511 cmd_verbose(int argc, const char **argv)
1512 {
1513 	if (argc == 0) {
1514 		log_setverbose(!log_getverbose());
1515 		if (log_getverbose())
1516 			puts("verbose mode enabled");
1517 		else
1518 			puts("verbose mode disabled");
1519 		return;
1520 	}
1521 
1522 	if (argc != 1)
1523 		goto usage;
1524 
1525 	if (!strcmp(*argv, "on")) {
1526 		log_setverbose(1);
1527 		puts("verbose mode enabled");
1528 		return;
1529 	}
1530 
1531 	if (!strcmp(*argv, "off")) {
1532 		log_setverbose(0);
1533 		puts("verbose mode disabled");
1534 		return;
1535 	}
1536 
1537 usage:
1538 	printf("verbose [on | off]\n");
1539 }
1540 
1541 static void
1542 excmd(int argc, const char **argv)
1543 {
1544 	struct cmd {
1545 		const char	*name;
1546 		void		(*fn)(int, const char **);
1547 	} cmds[] = {
1548 		{"bell",	cmd_bell},
1549 		{"bye",		cmd_bye},
1550 		{"cd",		cmd_cd},
1551 		{"edit",	cmd_edit},
1552 		{"get",		cmd_get},
1553 		{"hexdump",	cmd_hexdump},
1554 		{"lcd",		cmd_lcd},
1555 		{"lpwd",	cmd_lpwd},
1556 		{"ls",		cmd_ls},
1557 		{"page",	cmd_page},
1558 		{"pipe",	cmd_pipe},
1559 		{"put",		cmd_put},
1560 		{"quit",	cmd_bye},
1561 		{"rename",	cmd_rename},
1562 		{"verbose",	cmd_verbose},
1563 	};
1564 	size_t i;
1565 
1566 	if (argc == 0)
1567 		return;
1568 	for (i = 0; i < nitems(cmds); ++i) {
1569 		if (!strcmp(cmds[i].name, *argv)) {
1570 			cmds[i].fn(argc-1, argv+1);
1571 			return;
1572 		}
1573 	}
1574 
1575 	log_warnx("unknown command %s", *argv);
1576 }
1577 
1578 int
1579 main(int argc, char **argv)
1580 {
1581 	int	ch;
1582 
1583 	log_init(1, LOG_DAEMON);
1584 	log_setverbose(0);
1585 	log_procinit(getprogname());
1586 
1587 	while ((ch = getopt(argc, argv, "C:cK:")) != -1) {
1588 		switch (ch) {
1589 		case 'C':
1590 			tls = 1;
1591 			crtpath = optarg;
1592 			break;
1593 		case 'c':
1594 			tls = 1;
1595 			break;
1596 		case 'K':
1597 			tls = 1;
1598 			keypath = optarg;
1599 			break;
1600 		default:
1601 			usage(1);
1602 		}
1603 	}
1604 	argc -= optind;
1605 	argv += optind;
1606 
1607 	if (argc == 0 || (tls && crtpath == NULL))
1608 		usage(1);
1609 
1610 	signal(SIGPIPE, SIG_IGN);
1611 	if (isatty(1)) {
1612 		tty_p = 1;
1613 		resized = 1;
1614 		signal(SIGWINCH, tty_resized);
1615 	}
1616 
1617 	if ((evb = evbuffer_new()) == NULL)
1618 		fatal("evbuffer_new");
1619 
1620 	if ((buf = evbuffer_new()) == NULL)
1621 		fatal("evbuffer_new");
1622 
1623 	if ((dirbuf = evbuffer_new()) == NULL)
1624 		fatal("evbuferr_new");
1625 
1626 	do_connect(argv[0], argv[1]);
1627 
1628 	for (;;) {
1629 		int argc = 0;
1630 		char *line, *argv[16] = {0}, **ap;
1631 
1632 		if ((line = read_line("kamiftp> ")) == NULL)
1633 			break;
1634 
1635 		for (argc = 0, ap = argv; ap < &argv[15] &&
1636 		    (*ap = strsep(&line, " \t")) != NULL;) {
1637 			if (**ap != '\0')
1638 				ap++, argc++;
1639 		}
1640 		excmd(argc, (const char **)argv);
1641 
1642 		if (bell) {
1643 			printf("\a");
1644 			fflush(stdout);
1645 		}
1646 
1647 		free(line);
1648 	}
1649 
1650 	printf("\n");
1651 }