Blame


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