Blob


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 */
17 #include "compat.h"
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>
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>
40 #ifdef HAVE_READLINE
41 #include <readline/readline.h>
42 #include <readline/history.h>
43 #endif
45 #ifndef nitems
46 #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
47 #endif
49 #include "kami.h"
50 #include "utils.h"
51 #include "log.h"
52 #include "9pclib.h"
54 #define TMPFSTR "/tmp/kamiftp.XXXXXXXXXX"
55 #define TMPFSTRLEN sizeof(TMPFSTR)
57 /* flags */
58 int tls;
59 const char *crtpath;
60 const char *keypath;
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;
71 volatile sig_atomic_t resized;
72 int tty_p;
73 int tty_width;
74 int xdump;
76 struct progress {
77 uint64_t max;
78 uint64_t done;
79 };
81 int pwdfid;
83 #define ASSERT_EMPTYBUF() assert(EVBUFFER_LENGTH(buf) == 0)
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;
93 printf("%s", prompt);
94 fflush(stdout);
96 linelen = getline(&line, &linesize, stdin);
97 if (linelen == -1)
98 return NULL;
100 if ((ch = strchr(line, '\n')) != NULL)
101 *ch = '\0';
102 return line;
105 void
106 add_history(const char *line)
108 return;
110 #endif
112 static char *
113 read_line(const char *prompt)
115 char *line;
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;
126 add_history(line);
127 return line;
130 static void
131 spawn(const char *argv0, ...)
133 pid_t pid;
134 size_t i;
135 int status;
136 const char *argv[16], *last;
137 va_list ap;
139 memset(argv, 0, sizeof(argv));
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;
149 va_end(ap);
151 assert(last == NULL);
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);
164 static void
165 tty_resized(int signo)
167 resized = 1;
170 static void __dead
171 usage(int ret)
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);
179 static int
180 nextfid(void)
182 uint32_t i;
184 for (i = 0; ; ++i) {
185 if (i != pwdfid)
186 return i;
190 static void
191 do_send(void)
193 const void *buf;
194 size_t nbytes;
195 ssize_t r;
197 if (xdump)
198 hexdump("outgoing message", EVBUFFER_DATA(evb),
199 EVBUFFER_LENGTH(evb));
201 while (EVBUFFER_LENGTH(evb) != 0) {
202 buf = EVBUFFER_DATA(evb);
203 nbytes = EVBUFFER_LENGTH(evb);
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));
217 evbuffer_drain(evb, r);
221 static void
222 mustread(void *d, size_t len)
224 ssize_t r;
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));
239 d += r;
240 len -= r;
244 static void
245 recv_msg(void)
247 uint32_t len, l;
248 char tmp[BUFSIZ];
250 mustread(&len, sizeof(len));
251 len = le32toh(len);
252 if (len < HEADERSIZE)
253 errx(1, "read message of invalid length %d", len);
255 len -= 4; /* skip the length just read */
257 while (len != 0) {
258 l = MIN(len, sizeof(tmp));
259 mustread(tmp, l);
260 len -= l;
261 evbuffer_add(buf, tmp, l);
264 if (xdump)
265 hexdump("incoming packet", EVBUFFER_DATA(buf),
266 EVBUFFER_LENGTH(buf));
269 static uint64_t
270 np_read64(struct evbuffer *buf)
272 uint64_t n;
274 evbuffer_remove(buf, &n, sizeof(n));
275 return le64toh(n);
278 static uint32_t
279 np_read32(struct evbuffer *buf)
281 uint32_t n;
283 evbuffer_remove(buf, &n, sizeof(n));
284 return le32toh(n);
287 static uint16_t
288 np_read16(struct evbuffer *buf)
290 uint16_t n;
292 evbuffer_remove(buf, &n, sizeof(n));
293 return le16toh(n);
296 static uint16_t
297 np_read8(struct evbuffer *buf)
299 uint8_t n;
301 evbuffer_remove(buf, &n, sizeof(n));
302 return n;
305 static char *
306 np_readstr(struct evbuffer *buf)
308 uint16_t len;
309 char *str;
311 len = np_read16(buf);
312 assert(EVBUFFER_LENGTH(buf) >= len);
314 if ((str = calloc(1, len+1)) == NULL)
315 err(1, "calloc");
316 evbuffer_remove(buf, str, len);
317 return str;
320 static void
321 np_read_qid(struct evbuffer *buf, struct qid *qid)
323 assert(EVBUFFER_LENGTH(buf) >= QIDSIZE);
325 qid->type = np_read8(buf);
326 qid->vers = np_read32(buf);
327 qid->path = np_read64(buf);
330 static int
331 np_read_stat(struct evbuffer *buf, struct np_stat *st)
333 uint16_t size;
335 memset(st, 0, sizeof(*st));
337 size = np_read16(buf);
338 if (size > EVBUFFER_LENGTH(buf))
339 return -1;
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);
353 return 0;
356 static void
357 expect(uint8_t type)
359 uint8_t t;
361 t = np_read8(buf);
362 if (t == type)
363 return;
365 if (t == Rerror) {
366 char *err;
368 /* skip tag */
369 np_read16(buf);
371 err = np_readstr(buf);
372 errx(1, "expected %s, got error %s",
373 pp_msg_type(type), err);
376 errx(1, "expected %s, got msg type %s",
377 pp_msg_type(type), pp_msg_type(t));
380 static void
381 expect2(uint8_t type, uint16_t tag)
383 uint16_t t;
385 expect(type);
387 t = np_read16(buf);
388 if (t == tag)
389 return;
391 errx(1, "expected tag 0x%x, got 0x%x", tag, t);
394 static char *
395 check(uint8_t type, uint16_t tag)
397 uint16_t rtag;
398 uint8_t rtype;
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;
408 if (rtype == Rerror)
409 return np_readstr(buf);
411 errx(1, "expected %s, got msg type %s",
412 pp_msg_type(type), pp_msg_type(rtype));
415 static void
416 do_version(void)
418 char *version;
420 tversion(VERSION9P, MSIZE9P);
421 do_send();
422 recv_msg();
423 expect2(Rversion, NOTAG);
425 msize = np_read32(buf);
426 version = np_readstr(buf);
428 if (msize > MSIZE9P)
429 errx(1, "got unexpected msize: %d", msize);
430 if (strcmp(version, VERSION9P))
431 errx(1, "unexpected 9p version: %s", version);
433 free(version);
434 ASSERT_EMPTYBUF();
437 static void
438 do_attach(const char *path)
440 const char *user;
441 struct qid qid;
443 if (path == NULL)
444 path = "/";
445 if ((user = getenv("USER")) == NULL)
446 user = "flan";
448 tattach(pwdfid, NOFID, user, path);
449 do_send();
450 recv_msg();
451 expect2(Rattach, iota_tag);
452 np_read_qid(buf, &qid);
454 ASSERT_EMPTYBUF();
457 static uint32_t
458 do_open(uint32_t fid, uint8_t mode)
460 struct qid qid;
461 uint32_t iounit;
463 topen(fid, mode);
464 do_send();
465 recv_msg();
466 expect2(Ropen, iota_tag);
468 np_read_qid(buf, &qid);
469 iounit = np_read32(buf);
471 ASSERT_EMPTYBUF();
473 return iounit;
476 static uint32_t
477 do_create(uint32_t fid, const char *name, uint32_t perm, uint8_t mode)
479 struct qid qid;
480 uint32_t iounit;
482 tcreate(fid, name, perm, mode);
483 do_send();
484 recv_msg();
485 expect2(Rcreate, iota_tag);
487 np_read_qid(buf, &qid);
488 iounit = np_read32(buf);
490 ASSERT_EMPTYBUF();
492 return iounit;
495 static void
496 do_clunk(uint32_t fid)
498 tclunk(fid);
499 do_send();
500 recv_msg();
501 expect2(Rclunk, iota_tag);
503 ASSERT_EMPTYBUF();
506 static char *
507 dup_fid(int fid, int nfid)
509 uint16_t nwqid;
510 char *errstr;
512 twalk(fid, nfid, NULL, 0);
513 do_send();
514 recv_msg();
516 if ((errstr = check(Rwalk, iota_tag)) != NULL)
517 return errstr;
519 nwqid = np_read16(buf);
520 assert(nwqid == 0);
522 ASSERT_EMPTYBUF();
524 return NULL;
527 static char *
528 walk_path(int fid, int newfid, const char *path, int *missing,
529 struct qid *qid)
531 char *wnames[MAXWELEM], *p, *t, *errstr;
532 size_t nwname, i;
533 uint16_t nwqid;
535 if ((p = strdup(path)) == NULL)
536 err(1, "strdup");
537 t = p;
539 /* strip initial ./ */
540 if (t[0] == '.' && t[1] == '/')
541 t += 2;
543 for (nwname = 0; nwname < nitems(wnames) &&
544 (wnames[nwname] = strsep(&t, "/")) != NULL;) {
545 if (*wnames[nwname] != '\0')
546 nwname++;
549 twalk(fid, newfid, (const char **)wnames, nwname);
550 do_send();
551 recv_msg();
553 *missing = nwname;
554 if ((errstr = check(Rwalk, iota_tag)) != NULL) {
555 free(p);
556 return errstr;
559 nwqid = np_read16(buf);
560 assert(nwqid <= nwname);
562 /* consume all qids */
563 for (i = 0; i < nwqid; ++i)
564 np_read_qid(buf, qid);
566 free(p);
568 *missing = nwname - nwqid;
569 return NULL;
572 static void
573 do_stat(int fid, struct np_stat *st)
575 tstat(fid);
576 do_send();
577 recv_msg();
578 expect2(Rstat, iota_tag);
580 if (np_read_stat(buf, st) == -1)
581 errx(1, "invalid stat struct read");
583 ASSERT_EMPTYBUF();
586 static char *
587 do_wstat(int fid, const struct np_stat *st)
589 char *errstr;
591 twstat(fid, st);
592 do_send();
593 recv_msg();
595 if ((errstr = check(Rwstat, iota_tag)) != NULL)
596 return errstr;
598 ASSERT_EMPTYBUF();
600 return NULL;
603 static size_t
604 do_read(int fid, uint64_t off, uint32_t count, void *data)
606 uint32_t r;
608 tread(fid, off, count);
609 do_send();
610 recv_msg();
611 expect2(Rread, iota_tag);
613 r = np_read32(buf);
614 assert(r == EVBUFFER_LENGTH(buf));
615 assert(r <= count);
616 evbuffer_remove(buf, data, r);
618 ASSERT_EMPTYBUF();
620 return r;
623 static size_t
624 do_write(int fid, uint64_t off, uint32_t count, void *data)
626 uint32_t r;
628 twrite(fid, off, data, count);
629 do_send();
630 recv_msg();
631 expect2(Rwrite, iota_tag);
633 r = np_read32(buf);
634 assert(r <= count);
636 ASSERT_EMPTYBUF();
638 return r;
641 static void
642 draw_progress(const char *pre, const struct progress *p)
644 struct winsize ws;
645 int i, l, w;
646 double perc;
648 if (xdump)
649 return;
651 perc = 100.0 * p->done / p->max;
652 if (!tty_p) {
653 fprintf(stderr, "%s: %d%%\n", pre, (int)perc);
654 return;
657 if (resized) {
658 resized = 0;
660 if (ioctl(0, TIOCGWINSZ, &ws) == -1)
661 return;
662 tty_width = ws.ws_col;
664 w = tty_width;
666 if (pre == NULL ||
667 ((l = printf("\r%s ", pre)) == -1 || l >= w))
668 return;
670 w -= l + 2 + 5; /* 2 for |, 5 for percentage + \n */
671 if (w < 0) {
672 printf("%4d%%\n", (int)perc);
673 return;
676 printf("|");
678 l = w * MIN(100.0, perc) / 100.0;
679 for (i = 0; i < l; i++)
680 printf("*");
681 for (; i < w; i++)
682 printf(" ");
683 printf("|%4d%%", (int)perc);
685 fflush(stdout);
688 static int
689 fetch_fid(int fid, int fd, const char *name)
691 struct progress p = {0};
692 struct np_stat st;
693 size_t r;
694 int ret = 0;
695 char buf[BUFSIZ];
697 do_stat(fid, &st);
698 do_open(fid, KOREAD);
700 p.max = st.length;
701 for (;;) {
702 size_t off;
703 ssize_t nw;
705 r = do_read(fid, p.done, sizeof(buf), buf);
706 if (r == 0)
707 break;
709 for (off = 0; off < r; off += nw)
710 if ((nw = write(fd, buf + off, r - off)) == 0 ||
711 nw == -1) {
712 ret = -1;
713 goto end;
716 p.done += r;
717 draw_progress(name, &p);
719 #if 0
720 /* throttle, for debugging purpose */
722 struct timespec ts = { 0, 500000000 };
723 nanosleep(&ts, NULL);
725 #endif
728 end:
729 putchar('\n');
731 do_clunk(fid);
732 return ret;
735 static void
736 send_fid(int fid, const char *fnam, int open_flags, int fd, const char *name)
738 struct progress p = {0};
739 struct stat sb;
740 ssize_t r;
741 size_t w;
742 char buf[BUFSIZ];
744 if (fstat(fd, &sb) == -1)
745 err(1, "fstat");
747 if (fnam != NULL)
748 do_create(fid, fnam, 0644, KOWRITE);
749 else
750 do_open(fid, open_flags | KOWRITE);
752 p.max = sb.st_size;
753 for (;;) {
754 r = read(fd, buf, sizeof(buf));
755 if (r == 0)
756 break;
757 if (r == -1)
758 err(1, "read");
760 w = do_write(fid, p.done, r, buf);
761 p.done += w;
763 draw_progress(name, &p);
765 #if 0
766 /* throttle, for debugging purpose */
768 struct timespec ts = { 0, 500000000 };
769 nanosleep(&ts, NULL);
771 #endif
774 putchar('\n');
775 do_clunk(fid);
778 static int
779 woc_file(int fd, const char *prompt, const char *path)
781 struct qid qid;
782 const char *n = NULL;
783 char *errstr;
784 int nfid, miss;
786 nfid = nextfid();
787 errstr = walk_path(pwdfid, nfid, path, &miss, &qid);
788 if (errstr != NULL && miss > 1) {
789 printf("%s: %s\n", path, errstr);
790 free(errstr);
791 return -1;
794 if (errstr != NULL || miss == 1) {
795 char p[PATH_MAX], *dn;
797 /*
798 * If it's only one component missing (the file name), walk
799 * to the parent directory and try to create the file.
800 */
802 if (strlcpy(p, path, sizeof(p)) >= sizeof(p)) {
803 printf("path too long: %s\n", path);
804 return -1;
806 dn = dirname(p);
808 if (!strcmp(dn, ".")) {
809 errstr = dup_fid(pwdfid, nfid);
810 miss = 0;
811 } else
812 errstr = walk_path(pwdfid, nfid, dn, &miss, &qid);
814 if (errstr != NULL) {
815 printf("%s: %s\n", dn, errstr);
816 free(errstr);
817 return -1;
820 if (miss != 0) {
821 printf("%s: not a directory\n", dn);
822 return -1;
825 if ((n = strrchr(path, '/')) != NULL)
826 n++;
827 else
828 n = path;
831 free(errstr);
833 if (miss > 1) {
834 printf("can't create %s: missing %d path component(s)\n",
835 path, miss);
836 return -1;
839 send_fid(nfid, n, KOTRUNC, fd, prompt);
840 return 0;
843 static void
844 do_tls_connect(const char *host, const char *port)
846 int handshake;
848 if ((tlsconf = tls_config_new()) == NULL)
849 fatalx("tls_config_new");
850 tls_config_insecure_noverifycert(tlsconf);
851 tls_config_insecure_noverifyname(tlsconf);
852 if (tls_config_set_keypair_file(tlsconf, crtpath, keypath) == -1)
853 fatalx("can't load certs (%s, %s)", crtpath, keypath);
855 if ((ctx = tls_client()) == NULL)
856 fatal("tls_client");
857 if (tls_configure(ctx, tlsconf) == -1)
858 fatalx("tls_configure: %s", tls_error(ctx));
860 if (tls_connect(ctx, host, port) == -1)
861 fatalx("can't connect to %s:%s: %s", host, port,
862 tls_error(ctx));
864 for (handshake = 0; !handshake;) {
865 switch (tls_handshake(ctx)) {
866 case -1:
867 fatalx("tls_handshake: %s", tls_error(ctx));
868 case 0:
869 handshake = 1;
870 break;
875 static void
876 do_ctxt_connect(const char *host, const char *port)
878 struct addrinfo hints, *res, *res0;
879 int error, saved_errno;
880 const char *cause = NULL;
882 memset(&hints, 0, sizeof(hints));
883 hints.ai_family = AF_UNSPEC;
884 hints.ai_socktype = SOCK_STREAM;
885 error = getaddrinfo(host, port, &hints, &res0);
886 if (error)
887 errx(1, "%s", gai_strerror(error));
889 sock = -1;
890 for (res = res0; res != NULL; res = res->ai_next) {
891 sock = socket(res->ai_family, res->ai_socktype|SOCK_CLOEXEC,
892 res->ai_protocol);
893 if (sock == -1) {
894 cause = "socket";
895 continue;
898 if (connect(sock, res->ai_addr, res->ai_addrlen) == -1) {
899 cause = "connect";
900 saved_errno = errno;
901 close(sock);
902 errno = saved_errno;
903 sock = -1;
904 continue;
907 break;
910 if (sock == -1)
911 err(1, "%s", cause);
912 freeaddrinfo(res0);
915 static void
916 do_connect(const char *connspec, const char *path)
918 char *host, *colon;
919 const char *port;
921 host = xstrdup(connspec);
922 if ((colon = strchr(host, ':')) != NULL) {
923 *colon = '\0';
924 port = ++colon;
925 } else
926 port = "1337";
928 printf("connecting to %s:%s...", host, port);
929 fflush(stdout);
931 if (tls)
932 do_tls_connect(host, port);
933 else
934 do_ctxt_connect(host, port);
936 printf(" done!\n");
938 do_version();
939 do_attach(path);
941 free(host);
944 static int
945 tmp_file(char sfn[TMPFSTRLEN])
947 int tmpfd;
949 strlcpy(sfn, TMPFSTR, TMPFSTRLEN);
950 if ((tmpfd = mkstemp(sfn)) == -1) {
951 warn("mkstemp %s", sfn);
952 return -1;
955 /* set the close-on-exec flag */
956 if (fcntl(tmpfd, F_SETFD, FD_CLOEXEC) == -1) {
957 warn("fcntl");
958 close(tmpfd);
959 return -1;
962 return tmpfd;
965 static inline const char *
966 pp_perm(uint8_t x)
968 switch (x & 0x7) {
969 case 0x0:
970 return "---";
971 case 0x1:
972 return "--x";
973 case 0x2:
974 return "-w-";
975 case 0x3:
976 return "-wx";
977 case 0x4:
978 return "r--";
979 case 0x5:
980 return "r-x";
981 case 0x6:
982 return "rw-";
983 case 0x7:
984 return "rwx";
985 default:
986 /* unreachable, just for the compiler' happiness */
987 return "???";
991 static inline void
992 prepare_wstat(struct np_stat *st)
994 memset(st, 0xFF, sizeof(*st));
995 st->name = NULL;
996 st->uid = NULL;
997 st->gid = NULL;
998 st->muid = NULL;
1001 static void
1002 cmd_bell(int argc, const char **argv)
1004 if (argc == 0) {
1005 bell = !bell;
1006 if (bell)
1007 puts("bell mode enabled");
1008 else
1009 puts("bell mode disabled");
1010 return;
1013 if (argc != 1)
1014 goto usage;
1016 if (!strcmp(*argv, "on")) {
1017 bell = 1;
1018 puts("bell mode enabled");
1019 return;
1022 if (!strcmp(*argv, "off")) {
1023 bell = 0;
1024 puts("bell mode disabled");
1025 return;
1028 usage:
1029 printf("bell [on | off]\n");
1032 static void
1033 cmd_bye(int argc, const char **argv)
1035 log_warnx("bye\n");
1036 exit(0);
1039 static void
1040 cmd_cd(int argc, const char **argv)
1042 struct qid qid;
1043 int nfid, miss;
1044 char *errstr;
1046 if (argc != 1) {
1047 printf("usage: cd remote-path\n");
1048 return;
1051 nfid = nextfid();
1052 errstr = walk_path(pwdfid, nfid, argv[0], &miss, &qid);
1053 if (errstr != NULL) {
1054 printf("%s: %s\n", argv[0], errstr);
1055 free(errstr);
1056 return;
1059 if (miss != 0 || !(qid.type & QTDIR)) {
1060 printf("%s: not a directory\n", argv[0]);
1061 if (miss == 0)
1062 do_clunk(nfid);
1063 return;
1066 do_clunk(pwdfid);
1067 pwdfid = nfid;
1070 static void
1071 cmd_edit(int argc, const char **argv)
1073 struct qid qid;
1074 int nfid, tmpfd, miss;
1075 char sfn[TMPFSTRLEN], p[PATH_MAX], *name, *errstr;
1076 const char *ed;
1078 if (argc != 1) {
1079 puts("usage: edit file");
1080 return;
1083 if ((ed = getenv("VISUAL")) == NULL &&
1084 (ed = getenv("EDITOR")) == NULL)
1085 ed = "ed";
1087 nfid = nextfid();
1088 errstr = walk_path(pwdfid, nfid, *argv, &miss, &qid);
1089 if (errstr != NULL) {
1090 printf("%s: %s\n", *argv, errstr);
1091 free(errstr);
1092 return;
1095 if (miss != 0 || qid.type != 0) {
1096 printf("%s: not a file\n", *argv);
1097 if (miss == 0)
1098 do_clunk(nfid);
1099 return;
1102 if ((tmpfd = tmp_file(sfn)) == -1) {
1103 do_clunk(nfid);
1104 return;
1107 strlcpy(p, *argv, sizeof(p));
1108 name = basename(p);
1110 if (fetch_fid(nfid, tmpfd, name)) {
1111 warn("failed fetch or can't write %s", sfn);
1112 goto end;
1114 close(tmpfd);
1116 spawn(ed, sfn, NULL);
1119 * Re-open the file because it's not guaranteed that the
1120 * file descriptor tmpfd is still associated with the file
1121 * pointed by sfn: it's not uncommon for editor to write
1122 * a backup file and then rename(2) it to the file name.
1124 if ((tmpfd = open(sfn, O_RDONLY)) == -1) {
1125 warn("can't open %s", sfn);
1126 goto end;
1129 woc_file(tmpfd, *argv, name);
1130 close(tmpfd);
1132 end:
1133 unlink(sfn);
1136 static void
1137 cmd_get(int argc, const char **argv)
1139 struct qid qid;
1140 const char *l;
1141 char *errstr;
1142 int nfid, fd, miss;
1144 if (argc != 1 && argc != 2) {
1145 printf("usage: get remote-file [local-file]\n");
1146 return;
1149 if (argc == 2)
1150 l = argv[1];
1151 else if ((l = strrchr(argv[0], '/')) != NULL)
1152 l++; /* skip / */
1153 else
1154 l = argv[0];
1156 nfid = nextfid();
1157 errstr = walk_path(pwdfid, nfid, argv[0], &miss, &qid);
1158 if (errstr != NULL) {
1159 printf("%s: %s\n", argv[0], errstr);
1160 free(errstr);
1161 return;
1164 if (miss != 0 || qid.type != 0) {
1165 printf("%s: not a file\n", argv[0]);
1166 if (miss == 0)
1167 do_clunk(nfid);
1168 return;
1171 if ((fd = open(l, O_WRONLY|O_CREAT|O_TRUNC|O_CLOEXEC, 0644)) == -1) {
1172 warn("can't open %s", l);
1173 do_clunk(nfid);
1174 return;
1177 if (fetch_fid(nfid, fd, l) == -1)
1178 warn("write %s", l);
1179 close(fd);
1182 static void
1183 cmd_hexdump(int argc, const char **argv)
1185 if (argc == 0) {
1186 xdump = !xdump;
1187 if (xdump)
1188 puts("hexdump mode enabled");
1189 else
1190 puts("hexdump mode disabled");
1191 return;
1194 if (argc > 1)
1195 goto usage;
1197 if (!strcmp(*argv, "on")) {
1198 xdump = 1;
1199 puts("hexdump mode enabled");
1200 return;
1203 if (!strcmp(*argv, "off")) {
1204 xdump = 0;
1205 puts("hexdump mode disabled");
1206 return;
1209 usage:
1210 puts("usage: hexdump [on | off]");
1213 static void
1214 cmd_lcd(int argc, const char **argv)
1216 const char *dir;
1218 if (argc > 1) {
1219 printf("usage: lcd [local-directory]\n");
1220 return;
1223 if (argc == 1)
1224 dir = *argv;
1226 if (argc == 0 && (dir = getenv("HOME")) == NULL) {
1227 printf("HOME is not defined\n");
1228 return;
1231 if (chdir(dir) == -1)
1232 printf("cd: %s: %s\n", dir, strerror(errno));
1235 static void
1236 cmd_lpwd(int argc, const char **argv)
1238 char path[PATH_MAX];
1240 if (argc != 0) {
1241 printf("usage: lpwd\n");
1242 return;
1245 if (getcwd(path, sizeof(path)) == NULL) {
1246 printf("lpwd: %s\n", strerror(errno));
1247 return;
1250 printf("%s\n", path);
1253 static void
1254 cmd_ls(int argc, const char **argv)
1256 struct np_stat st;
1257 time_t now, mtime;
1258 struct tm *tm;
1259 uint64_t off = 0;
1260 uint32_t len;
1261 int nfid;
1262 const char *timfmt;
1263 char fmt[FMT_SCALED_STRSIZE], tim[13], *errstr;
1265 if (argc != 0) {
1266 printf("ls don't take arguments (yet)\n");
1267 return;
1270 now = time(NULL);
1272 nfid = nextfid();
1273 if ((errstr = dup_fid(pwdfid, nfid)) != NULL) {
1274 printf(".: %s\n", errstr);
1275 free(errstr);
1276 return;
1279 do_open(nfid, KOREAD);
1281 evbuffer_drain(dirbuf, EVBUFFER_LENGTH(dirbuf));
1283 for (;;) {
1284 tread(nfid, off, BUFSIZ);
1285 do_send();
1286 recv_msg();
1287 expect2(Rread, iota_tag);
1289 len = np_read32(buf);
1290 if (len == 0)
1291 break;
1293 evbuffer_add_buffer(dirbuf, buf);
1294 off += len;
1296 ASSERT_EMPTYBUF();
1299 while (EVBUFFER_LENGTH(dirbuf) != 0) {
1300 if (np_read_stat(dirbuf, &st) == -1)
1301 errx(1, "invalid stat struct read");
1303 if (fmt_scaled(st.length, fmt) == -1)
1304 strlcpy(fmt, "xxx", sizeof(fmt));
1306 mtime = st.mtime;
1308 if (now > mtime && (now - mtime) < 365/2 * 24 * 12 * 60)
1309 timfmt = "%b %e %R";
1310 else
1311 timfmt = "%b %e %Y";
1313 if ((tm = localtime(&mtime)) == NULL ||
1314 strftime(tim, sizeof(tim), timfmt, tm) == 0)
1315 strlcpy(tim, "unknown", sizeof(tim));
1317 if (st.qid.type & QTDIR)
1318 printf("d");
1319 else
1320 printf("-");
1321 printf("%s", pp_perm(st.mode >> 6));
1322 printf("%s", pp_perm(st.mode >> 3));
1323 printf("%s", pp_perm(st.mode));
1324 printf(" %8s %12s %s%s\n", fmt, tim, st.name,
1325 st.qid.type & QTDIR ? "/" : "");
1327 free(st.name);
1328 free(st.uid);
1329 free(st.gid);
1330 free(st.muid);
1333 do_clunk(nfid);
1336 static void
1337 cmd_page(int argc, const char **argv)
1339 struct qid qid;
1340 int nfid, tmpfd, miss, r;
1341 char sfn[TMPFSTRLEN], p[PATH_MAX], *name, *errstr;
1342 const char *pager;
1344 if (argc != 1) {
1345 puts("usage: page file");
1346 return;
1349 if ((pager = getenv("PAGER")) == NULL)
1350 pager = "less";
1352 nfid = nextfid();
1353 errstr = walk_path(pwdfid, nfid, *argv, &miss, &qid);
1354 if (errstr != NULL) {
1355 printf("%s: %s\n", *argv, errstr);
1356 free(errstr);
1357 return;
1360 if (miss != 0 || qid.type != 0) {
1361 printf("%s: not a file\n", *argv);
1362 if (miss == 0)
1363 do_clunk(nfid);
1364 return;
1367 if ((tmpfd = tmp_file(sfn)) == -1) {
1368 do_clunk(nfid);
1369 return;
1372 strlcpy(p, *argv, sizeof(p));
1373 name = basename(p);
1374 if ((r = fetch_fid(nfid, tmpfd, name)) == -1)
1375 warn("write %s", sfn);
1376 close(tmpfd);
1377 if (r != -1)
1378 spawn(pager, sfn, NULL);
1379 unlink(sfn);
1382 static void
1383 cmd_pipe(int argc, const char **argv)
1385 struct qid qid;
1386 pid_t pid;
1387 int nfid, tmpfd, miss, status;
1388 int filedes[2]; /* read end, write end */
1389 char *errstr;
1391 if (argc < 2) {
1392 puts("usage: pipe remote-file cmd [args...]");
1393 return;
1396 nfid = nextfid();
1397 errstr = walk_path(pwdfid, nfid, *argv, &miss, &qid);
1398 if (errstr != NULL) {
1399 printf("%s: %s\n", *argv, errstr);
1400 free(errstr);
1401 return;
1404 if (miss != 0 || qid.type != 0) {
1405 printf("%s: not a file\n", *argv);
1406 if (miss == 0)
1407 do_clunk(nfid);
1408 return;
1411 if (pipe(filedes) == -1)
1412 err(1, "pipe");
1414 switch (pid = vfork()) {
1415 case -1:
1416 err(1, "vfork");
1417 case 0:
1418 close(filedes[1]);
1419 if (dup2(filedes[0], 0) == -1)
1420 err(1, "dup2");
1421 execvp(argv[1], (char *const *)argv + 1);
1422 err(1, "execvp");
1425 close(filedes[0]);
1426 if (fetch_fid(nfid, filedes[1], *argv) == -1)
1427 warnx("failed to fetch all the file");
1428 close(filedes[1]);
1430 waitpid(pid, &status, 0);
1433 static void
1434 cmd_put(int argc, const char **argv)
1436 struct qid qid;
1437 const char *l;
1438 int fd;
1440 if (argc != 1 && argc != 2) {
1441 printf("usage: put local-file [remote-file]\n");
1442 return;
1445 if (argc == 2)
1446 l = argv[1];
1447 else if ((l = strrchr(argv[0], '/')) != NULL)
1448 l++; /* skip / */
1449 else
1450 l = argv[0];
1452 if ((fd = open(argv[0], O_RDONLY)) == -1) {
1453 warn("%s", argv[0]);
1454 return;
1457 woc_file(fd, argv[0], l);
1458 close(fd);
1461 static void
1462 cmd_rename(int argc, const char **argv)
1464 struct np_stat st;
1465 struct qid qid;
1466 char *errstr;
1467 int nfid, miss;
1469 if (argc != 2) {
1470 puts("usage: rename remote-file new-remote-name");
1471 return;
1474 nfid = nextfid();
1475 errstr = walk_path(pwdfid, nfid, argv[0], &miss, &qid);
1476 if (errstr != NULL) {
1477 printf("%s: %s\n", argv[0], errstr);
1478 free(errstr);
1479 return;
1482 if (miss != 0) {
1483 printf("%s: not such file or directory\n", argv[0]);
1484 return;
1487 prepare_wstat(&st);
1488 st.name = (char *)argv[1];
1489 if ((errstr = do_wstat(nfid, &st)) != NULL) {
1490 printf("rename: %s\n", errstr);
1491 free(errstr);
1494 do_clunk(nfid);
1497 static void
1498 cmd_verbose(int argc, const char **argv)
1500 if (argc == 0) {
1501 log_setverbose(!log_getverbose());
1502 if (log_getverbose())
1503 puts("verbose mode enabled");
1504 else
1505 puts("verbose mode disabled");
1506 return;
1509 if (argc != 1)
1510 goto usage;
1512 if (!strcmp(*argv, "on")) {
1513 log_setverbose(1);
1514 puts("verbose mode enabled");
1515 return;
1518 if (!strcmp(*argv, "off")) {
1519 log_setverbose(0);
1520 puts("verbose mode disabled");
1521 return;
1524 usage:
1525 printf("verbose [on | off]\n");
1528 static void
1529 excmd(int argc, const char **argv)
1531 struct cmd {
1532 const char *name;
1533 void (*fn)(int, const char **);
1534 } cmds[] = {
1535 {"bell", cmd_bell},
1536 {"bye", cmd_bye},
1537 {"cd", cmd_cd},
1538 {"edit", cmd_edit},
1539 {"get", cmd_get},
1540 {"hexdump", cmd_hexdump},
1541 {"lcd", cmd_lcd},
1542 {"lpwd", cmd_lpwd},
1543 {"ls", cmd_ls},
1544 {"page", cmd_page},
1545 {"pipe", cmd_pipe},
1546 {"put", cmd_put},
1547 {"quit", cmd_bye},
1548 {"rename", cmd_rename},
1549 {"verbose", cmd_verbose},
1551 size_t i;
1553 if (argc == 0)
1554 return;
1555 for (i = 0; i < nitems(cmds); ++i) {
1556 if (!strcmp(cmds[i].name, *argv)) {
1557 cmds[i].fn(argc-1, argv+1);
1558 return;
1562 log_warnx("unknown command %s", *argv);
1565 int
1566 main(int argc, char **argv)
1568 int ch;
1570 log_init(1, LOG_DAEMON);
1571 log_setverbose(0);
1572 log_procinit(getprogname());
1574 while ((ch = getopt(argc, argv, "C:cK:")) != -1) {
1575 switch (ch) {
1576 case 'C':
1577 crtpath = optarg;
1578 break;
1579 case 'c':
1580 tls = 1;
1581 break;
1582 case 'K':
1583 keypath = optarg;
1584 break;
1585 default:
1586 usage(1);
1589 argc -= optind;
1590 argv += optind;
1592 if (argc == 0)
1593 usage(1);
1595 signal(SIGPIPE, SIG_IGN);
1596 if (isatty(1)) {
1597 tty_p = 1;
1598 resized = 1;
1599 signal(SIGWINCH, tty_resized);
1602 if ((evb = evbuffer_new()) == NULL)
1603 fatal("evbuffer_new");
1605 if ((buf = evbuffer_new()) == NULL)
1606 fatal("evbuffer_new");
1608 if ((dirbuf = evbuffer_new()) == NULL)
1609 fatal("evbuferr_new");
1611 do_connect(argv[0], argv[1]);
1613 for (;;) {
1614 int argc = 0;
1615 char *line, *argv[16] = {0}, **ap;
1617 if ((line = read_line("kamiftp> ")) == NULL)
1618 break;
1620 for (argc = 0, ap = argv; ap < &argv[15] &&
1621 (*ap = strsep(&line, " \t")) != NULL;) {
1622 if (**ap != '\0')
1623 ap++, argc++;
1625 excmd(argc, (const char **)argv);
1627 if (bell) {
1628 printf("\a");
1629 fflush(stdout);
1632 free(line);
1635 printf("\n");