Blame


1 fb1a36c0 2022-01-09 op /*
2 fb1a36c0 2022-01-09 op * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3 fb1a36c0 2022-01-09 op *
4 fb1a36c0 2022-01-09 op * Permission to use, copy, modify, and distribute this software for any
5 fb1a36c0 2022-01-09 op * purpose with or without fee is hereby granted, provided that the above
6 fb1a36c0 2022-01-09 op * copyright notice and this permission notice appear in all copies.
7 fb1a36c0 2022-01-09 op *
8 fb1a36c0 2022-01-09 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 fb1a36c0 2022-01-09 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 fb1a36c0 2022-01-09 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 fb1a36c0 2022-01-09 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 fb1a36c0 2022-01-09 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 fb1a36c0 2022-01-09 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 fb1a36c0 2022-01-09 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 fb1a36c0 2022-01-09 op */
16 fb1a36c0 2022-01-09 op
17 fb1a36c0 2022-01-09 op #include <sys/stat.h>
18 fb1a36c0 2022-01-09 op #include <sys/types.h>
19 fb1a36c0 2022-01-09 op #include <sys/queue.h>
20 fb1a36c0 2022-01-09 op #include <sys/uio.h>
21 fb1a36c0 2022-01-09 op
22 fb1a36c0 2022-01-09 op #include <dirent.h>
23 fb1a36c0 2022-01-09 op #include <endian.h>
24 fb1a36c0 2022-01-09 op #include <err.h>
25 fb1a36c0 2022-01-09 op #include <errno.h>
26 fb1a36c0 2022-01-09 op #include <event.h>
27 fb1a36c0 2022-01-09 op #include <fcntl.h>
28 fb1a36c0 2022-01-09 op #include <pwd.h>
29 fb1a36c0 2022-01-09 op #include <signal.h>
30 fb1a36c0 2022-01-09 op #include <stdint.h>
31 fb1a36c0 2022-01-09 op #include <stdlib.h>
32 fb1a36c0 2022-01-09 op #include <string.h>
33 fb1a36c0 2022-01-09 op #include <syslog.h>
34 fb1a36c0 2022-01-09 op #include <unistd.h>
35 fb1a36c0 2022-01-09 op #include <imsg.h>
36 fb1a36c0 2022-01-09 op
37 fb1a36c0 2022-01-09 op #include "client.h"
38 fb1a36c0 2022-01-09 op #include "kami.h"
39 fb1a36c0 2022-01-09 op #include "kamid.h"
40 fb1a36c0 2022-01-09 op #include "log.h"
41 fb1a36c0 2022-01-09 op #include "sandbox.h"
42 fb1a36c0 2022-01-09 op #include "utils.h"
43 fb1a36c0 2022-01-09 op
44 fb1a36c0 2022-01-09 op /*
45 fb1a36c0 2022-01-09 op * XXX: atm is difficult to accept messages bigger than MAX_IMSGSIZE
46 fb1a36c0 2022-01-09 op * minus IMSG_HEADER_SIZE, we need something to split messages into
47 fb1a36c0 2022-01-09 op * chunks and receive them one by the other.
48 fb1a36c0 2022-01-09 op *
49 fb1a36c0 2022-01-09 op * CLIENT_MSIZE is thus the maximum message size we can handle now.
50 fb1a36c0 2022-01-09 op */
51 fb1a36c0 2022-01-09 op #define CLIENT_MSIZE (MAX_IMSGSIZE - IMSG_HEADER_SIZE)
52 fb1a36c0 2022-01-09 op
53 fb1a36c0 2022-01-09 op #define DEBUG_PACKETS 0
54 fb1a36c0 2022-01-09 op
55 fb1a36c0 2022-01-09 op /* straight outta /src/usr.bin/ssh/scp.c */
56 fb1a36c0 2022-01-09 op #define TYPE_OVERFLOW(type, val) \
57 fb1a36c0 2022-01-09 op ((sizeof(type) == 4 && (val) > INT32_MAX) || \
58 fb1a36c0 2022-01-09 op (sizeof(type) == 8 && (val) > INT64_MAX) || \
59 fb1a36c0 2022-01-09 op (sizeof(type) != 4 && sizeof(type) != 8))
60 fb1a36c0 2022-01-09 op
61 fb1a36c0 2022-01-09 op STAILQ_HEAD(dirhead, dir) dirs;
62 fb1a36c0 2022-01-09 op struct dir {
63 fb1a36c0 2022-01-09 op int refcount;
64 fb1a36c0 2022-01-09 op int fd;
65 fb1a36c0 2022-01-09 op STAILQ_ENTRY(dir) entries;
66 fb1a36c0 2022-01-09 op };
67 fb1a36c0 2022-01-09 op
68 fb1a36c0 2022-01-09 op STAILQ_HEAD(fidhead, fid) fids;
69 fb1a36c0 2022-01-09 op struct fid {
70 fb1a36c0 2022-01-09 op uint32_t fid;
71 fb1a36c0 2022-01-09 op
72 fb1a36c0 2022-01-09 op char fpath[PATH_MAX];
73 fb1a36c0 2022-01-09 op
74 fb1a36c0 2022-01-09 op /*
75 fb1a36c0 2022-01-09 op * the flags passed to open(2). O_CLOEXEC means ORCLOSE, that
76 fb1a36c0 2022-01-09 op * is to unlink the file upon Tclunk.
77 fb1a36c0 2022-01-09 op */
78 fb1a36c0 2022-01-09 op int iomode;
79 fb1a36c0 2022-01-09 op
80 fb1a36c0 2022-01-09 op /*
81 fb1a36c0 2022-01-09 op * if fd is not -1 this fid was opened, fd represents its
82 fb1a36c0 2022-01-09 op * file descriptor and iomode the flags passed to open(2).
83 fb1a36c0 2022-01-09 op */
84 fb1a36c0 2022-01-09 op int fd;
85 fb1a36c0 2022-01-09 op DIR *d;
86 fb1a36c0 2022-01-09 op struct evbuffer *evb;
87 fb1a36c0 2022-01-09 op
88 fb1a36c0 2022-01-09 op /*
89 fb1a36c0 2022-01-09 op * expected offset for Tread against a directory.
90 fb1a36c0 2022-01-09 op */
91 fb1a36c0 2022-01-09 op uint64_t offset;
92 fb1a36c0 2022-01-09 op
93 fb1a36c0 2022-01-09 op struct qid qid;
94 fb1a36c0 2022-01-09 op struct dir *dir;
95 fb1a36c0 2022-01-09 op STAILQ_ENTRY(fid) entries;
96 fb1a36c0 2022-01-09 op };
97 fb1a36c0 2022-01-09 op
98 fb1a36c0 2022-01-09 op static struct imsgev *iev_listener;
99 fb1a36c0 2022-01-09 op static struct evbuffer *evb;
100 fb1a36c0 2022-01-09 op static uint32_t peerid;
101 fb1a36c0 2022-01-09 op
102 fb1a36c0 2022-01-09 op static int handshaked;
103 fb1a36c0 2022-01-09 op uint32_t msize;
104 fb1a36c0 2022-01-09 op
105 fb1a36c0 2022-01-09 op static __dead void client_shutdown(void);
106 fb1a36c0 2022-01-09 op static void client_sig_handler(int, short, void *);
107 fb1a36c0 2022-01-09 op static void client_dispatch_listener(int, short, void *);
108 fb1a36c0 2022-01-09 op static void client_privdrop(const char *, const char *);
109 fb1a36c0 2022-01-09 op
110 fb1a36c0 2022-01-09 op static int client_send_listener(int, const void *, uint16_t);
111 fb1a36c0 2022-01-09 op
112 fb1a36c0 2022-01-09 op static void qid_update_from_sb(struct qid *, struct stat *);
113 fb1a36c0 2022-01-09 op
114 fb1a36c0 2022-01-09 op static struct dir *new_dir(int);
115 fb1a36c0 2022-01-09 op static struct dir *dir_incref(struct dir *);
116 fb1a36c0 2022-01-09 op static void dir_decref(struct dir *);
117 fb1a36c0 2022-01-09 op
118 fb1a36c0 2022-01-09 op static struct fid *new_fid(struct dir *, uint32_t, const char *, struct qid *);
119 fb1a36c0 2022-01-09 op static struct fid *fid_by_id(uint32_t);
120 fb1a36c0 2022-01-09 op static void free_fid(struct fid *);
121 fb1a36c0 2022-01-09 op
122 fb1a36c0 2022-01-09 op static void parse_message(const uint8_t *, size_t,
123 fb1a36c0 2022-01-09 op struct np_msg_header *, uint8_t **);
124 fb1a36c0 2022-01-09 op
125 fb1a36c0 2022-01-09 op static void np_write16(struct evbuffer *, uint16_t);
126 fb1a36c0 2022-01-09 op static void np_write32(struct evbuffer *, uint32_t);
127 fb1a36c0 2022-01-09 op static void np_write64(struct evbuffer *, uint64_t);
128 fb1a36c0 2022-01-09 op static void np_header(uint32_t, uint8_t, uint16_t);
129 fb1a36c0 2022-01-09 op static void np_string(struct evbuffer *, uint16_t, const char *);
130 fb1a36c0 2022-01-09 op static void np_qid(struct evbuffer *, struct qid *);
131 fb1a36c0 2022-01-09 op static void do_send(void);
132 fb1a36c0 2022-01-09 op
133 fb1a36c0 2022-01-09 op static void np_version(uint16_t, uint32_t, const char *);
134 fb1a36c0 2022-01-09 op static void np_attach(uint16_t, struct qid *);
135 fb1a36c0 2022-01-09 op static void np_clunk(uint16_t);
136 fb1a36c0 2022-01-09 op static void np_flush(uint16_t);
137 fb1a36c0 2022-01-09 op static void np_walk(uint16_t, int, struct qid *);
138 fb1a36c0 2022-01-09 op static void np_open(uint16_t, struct qid *, uint32_t);
139 fb1a36c0 2022-01-09 op static void np_create(uint16_t, struct qid *, uint32_t);
140 fb1a36c0 2022-01-09 op static void np_read(uint16_t, uint32_t, void *);
141 fb1a36c0 2022-01-09 op static void np_write(uint16_t, uint32_t);
142 fb1a36c0 2022-01-09 op static void np_stat(uint16_t, uint32_t, void *);
143 fb1a36c0 2022-01-09 op static void np_remove(uint16_t);
144 fb1a36c0 2022-01-09 op static void np_error(uint16_t, const char *);
145 fb1a36c0 2022-01-09 op static void np_errno(uint16_t);
146 fb1a36c0 2022-01-09 op
147 fb1a36c0 2022-01-09 op static int np_read8(const char *, const char *, uint8_t *,
148 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
149 fb1a36c0 2022-01-09 op static int np_read16(const char *, const char *, uint16_t *,
150 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
151 fb1a36c0 2022-01-09 op static int np_read32(const char *, const char *, uint32_t *,
152 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
153 fb1a36c0 2022-01-09 op static int np_read64(const char *, const char *, uint64_t *,
154 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
155 fb1a36c0 2022-01-09 op
156 fb1a36c0 2022-01-09 op #define READSTRERR -1
157 fb1a36c0 2022-01-09 op #define READSTRTRUNC -2
158 fb1a36c0 2022-01-09 op static int np_readstr(const char *, const char *, char *, size_t,
159 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
160 fb1a36c0 2022-01-09 op
161 fb1a36c0 2022-01-09 op #define NPREAD8(f, dst, src, len) np_read8(__func__, f, dst, src, len)
162 fb1a36c0 2022-01-09 op #define NPREAD16(f, dst, src, len) np_read16(__func__, f, dst, src, len)
163 fb1a36c0 2022-01-09 op #define NPREAD32(f, dst, src, len) np_read32(__func__, f, dst, src, len)
164 fb1a36c0 2022-01-09 op #define NPREAD64(f, dst, src, len) np_read64(__func__, f, dst, src, len)
165 fb1a36c0 2022-01-09 op
166 fb1a36c0 2022-01-09 op #define NPREADSTR(f, b, bl, src, len) np_readstr(__func__, f, b, bl, src, len)
167 fb1a36c0 2022-01-09 op
168 fb1a36c0 2022-01-09 op static void tversion(struct np_msg_header *, const uint8_t *, size_t);
169 fb1a36c0 2022-01-09 op static void tattach(struct np_msg_header *, const uint8_t *, size_t);
170 fb1a36c0 2022-01-09 op static void tclunk(struct np_msg_header *, const uint8_t *, size_t);
171 fb1a36c0 2022-01-09 op static void tflush(struct np_msg_header *, const uint8_t *, size_t);
172 fb1a36c0 2022-01-09 op static void twalk(struct np_msg_header *, const uint8_t *, size_t);
173 fb1a36c0 2022-01-09 op static void topen(struct np_msg_header *, const uint8_t *, size_t);
174 fb1a36c0 2022-01-09 op static void tcreate(struct np_msg_header *, const uint8_t *, size_t);
175 fb1a36c0 2022-01-09 op static void tread(struct np_msg_header *, const uint8_t *, size_t);
176 fb1a36c0 2022-01-09 op static void twrite(struct np_msg_header *, const uint8_t *, size_t);
177 fb1a36c0 2022-01-09 op static void tstat(struct np_msg_header *, const uint8_t *, size_t);
178 fb1a36c0 2022-01-09 op static void tremove(struct np_msg_header *, const uint8_t *, size_t);
179 fb1a36c0 2022-01-09 op static void handle_message(struct imsg *, size_t);
180 fb1a36c0 2022-01-09 op
181 fb1a36c0 2022-01-09 op __dead void
182 fb1a36c0 2022-01-09 op client(int debug, int verbose)
183 fb1a36c0 2022-01-09 op {
184 fb1a36c0 2022-01-09 op struct event ev_sigint, ev_sigterm;
185 fb1a36c0 2022-01-09 op
186 fb1a36c0 2022-01-09 op log_init(debug, LOG_DAEMON);
187 fb1a36c0 2022-01-09 op log_setverbose(verbose);
188 fb1a36c0 2022-01-09 op
189 fb1a36c0 2022-01-09 op setproctitle("client");
190 fb1a36c0 2022-01-09 op log_procinit("client");
191 fb1a36c0 2022-01-09 op
192 fb1a36c0 2022-01-09 op log_debug("warming up");
193 fb1a36c0 2022-01-09 op
194 fb1a36c0 2022-01-09 op event_init();
195 fb1a36c0 2022-01-09 op
196 fb1a36c0 2022-01-09 op /* Setup signal handlers */
197 fb1a36c0 2022-01-09 op signal_set(&ev_sigint, SIGINT, client_sig_handler, NULL);
198 fb1a36c0 2022-01-09 op signal_set(&ev_sigterm, SIGTERM, client_sig_handler, NULL);
199 fb1a36c0 2022-01-09 op
200 fb1a36c0 2022-01-09 op signal_add(&ev_sigint, NULL);
201 fb1a36c0 2022-01-09 op signal_add(&ev_sigterm, NULL);
202 fb1a36c0 2022-01-09 op
203 fb1a36c0 2022-01-09 op signal(SIGPIPE, SIG_IGN);
204 fb1a36c0 2022-01-09 op signal(SIGHUP, SIG_IGN);
205 fb1a36c0 2022-01-09 op
206 fb1a36c0 2022-01-09 op /* Setup pipe and event handler to the listener process */
207 fb1a36c0 2022-01-09 op if ((iev_listener = malloc(sizeof(*iev_listener))) == NULL)
208 fb1a36c0 2022-01-09 op fatal(NULL);
209 fb1a36c0 2022-01-09 op
210 fb1a36c0 2022-01-09 op imsg_init(&iev_listener->ibuf, 3);
211 fb1a36c0 2022-01-09 op iev_listener->handler = client_dispatch_listener;
212 fb1a36c0 2022-01-09 op
213 fb1a36c0 2022-01-09 op /* Setup event handlers. */
214 fb1a36c0 2022-01-09 op iev_listener->events = EV_READ;
215 fb1a36c0 2022-01-09 op event_set(&iev_listener->ev, iev_listener->ibuf.fd,
216 fb1a36c0 2022-01-09 op iev_listener->events, iev_listener->handler, iev_listener);
217 fb1a36c0 2022-01-09 op event_add(&iev_listener->ev, NULL);
218 fb1a36c0 2022-01-09 op
219 fb1a36c0 2022-01-09 op event_dispatch();
220 fb1a36c0 2022-01-09 op client_shutdown();
221 fb1a36c0 2022-01-09 op }
222 fb1a36c0 2022-01-09 op
223 fb1a36c0 2022-01-09 op static __dead void
224 fb1a36c0 2022-01-09 op client_shutdown(void)
225 fb1a36c0 2022-01-09 op {
226 fb1a36c0 2022-01-09 op if (evb != NULL)
227 fb1a36c0 2022-01-09 op evbuffer_free(evb);
228 fb1a36c0 2022-01-09 op
229 fb1a36c0 2022-01-09 op msgbuf_clear(&iev_listener->ibuf.w);
230 fb1a36c0 2022-01-09 op close(iev_listener->ibuf.fd);
231 fb1a36c0 2022-01-09 op
232 fb1a36c0 2022-01-09 op free(iev_listener);
233 fb1a36c0 2022-01-09 op
234 fb1a36c0 2022-01-09 op log_debug("client exiting");
235 fb1a36c0 2022-01-09 op exit(0);
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 client_sig_handler(int sig, short event, void *d)
240 fb1a36c0 2022-01-09 op {
241 fb1a36c0 2022-01-09 op /*
242 fb1a36c0 2022-01-09 op * Normal signal handler rules don't apply because libevent
243 fb1a36c0 2022-01-09 op * decouples for us.
244 fb1a36c0 2022-01-09 op */
245 fb1a36c0 2022-01-09 op
246 fb1a36c0 2022-01-09 op switch (sig) {
247 fb1a36c0 2022-01-09 op case SIGINT:
248 fb1a36c0 2022-01-09 op case SIGTERM:
249 fb1a36c0 2022-01-09 op client_shutdown();
250 fb1a36c0 2022-01-09 op default:
251 fb1a36c0 2022-01-09 op fatalx("unexpected signal %d", sig);
252 fb1a36c0 2022-01-09 op }
253 fb1a36c0 2022-01-09 op }
254 fb1a36c0 2022-01-09 op
255 fb1a36c0 2022-01-09 op #define AUTH_NONE 0
256 fb1a36c0 2022-01-09 op #define AUTH_USER 1
257 fb1a36c0 2022-01-09 op #define AUTH_DONE 2
258 fb1a36c0 2022-01-09 op
259 fb1a36c0 2022-01-09 op static void
260 fb1a36c0 2022-01-09 op client_dispatch_listener(int fd, short event, void *d)
261 fb1a36c0 2022-01-09 op {
262 fb1a36c0 2022-01-09 op static int auth = AUTH_NONE;
263 fb1a36c0 2022-01-09 op static char username[64] = {0};
264 fb1a36c0 2022-01-09 op static char dir[PATH_MAX] = {0};
265 fb1a36c0 2022-01-09 op struct imsg imsg;
266 fb1a36c0 2022-01-09 op struct imsgev *iev = d;
267 fb1a36c0 2022-01-09 op struct imsgbuf *ibuf;
268 fb1a36c0 2022-01-09 op ssize_t n;
269 fb1a36c0 2022-01-09 op int shut = 0;
270 fb1a36c0 2022-01-09 op
271 fb1a36c0 2022-01-09 op ibuf = &iev->ibuf;
272 fb1a36c0 2022-01-09 op
273 fb1a36c0 2022-01-09 op if (event & EV_READ) {
274 fb1a36c0 2022-01-09 op if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
275 fb1a36c0 2022-01-09 op fatal("imsg_read error");
276 fb1a36c0 2022-01-09 op if (n == 0) /* Connection closed */
277 fb1a36c0 2022-01-09 op shut = 1;
278 fb1a36c0 2022-01-09 op }
279 fb1a36c0 2022-01-09 op if (event & EV_WRITE) {
280 fb1a36c0 2022-01-09 op if ((n = msgbuf_write(&ibuf->w)) == -1 && errno != EAGAIN)
281 fb1a36c0 2022-01-09 op fatal("msgbuf_write");
282 fb1a36c0 2022-01-09 op if (n == 0) /* Connection closed */
283 fb1a36c0 2022-01-09 op shut = 1;
284 fb1a36c0 2022-01-09 op }
285 fb1a36c0 2022-01-09 op
286 fb1a36c0 2022-01-09 op for (;;) {
287 fb1a36c0 2022-01-09 op if ((n = imsg_get(ibuf, &imsg)) == -1)
288 fb1a36c0 2022-01-09 op fatal("%s: imsg_get error", __func__);
289 fb1a36c0 2022-01-09 op if (n == 0) /* No more messages. */
290 fb1a36c0 2022-01-09 op break;
291 fb1a36c0 2022-01-09 op
292 fb1a36c0 2022-01-09 op switch (imsg.hdr.type) {
293 fb1a36c0 2022-01-09 op case IMSG_AUTH:
294 fb1a36c0 2022-01-09 op peerid = imsg.hdr.peerid;
295 fb1a36c0 2022-01-09 op if (auth)
296 fb1a36c0 2022-01-09 op fatalx("%s: IMSG_AUTH already done", __func__);
297 fb1a36c0 2022-01-09 op auth = AUTH_USER;
298 fb1a36c0 2022-01-09 op ((char *)imsg.data)[IMSG_DATA_SIZE(imsg)-1] = '\0';
299 fb1a36c0 2022-01-09 op strlcpy(username, imsg.data, sizeof(username));
300 fb1a36c0 2022-01-09 op break;
301 fb1a36c0 2022-01-09 op case IMSG_AUTH_DIR:
302 fb1a36c0 2022-01-09 op if (auth != AUTH_USER)
303 fb1a36c0 2022-01-09 op fatalx("%s: IMSG_AUTH_DIR not after IMSG_AUTH",
304 fb1a36c0 2022-01-09 op __func__);
305 fb1a36c0 2022-01-09 op auth = AUTH_DONE;
306 fb1a36c0 2022-01-09 op ((char *)imsg.data)[IMSG_DATA_SIZE(imsg)-1] = '\0';
307 fb1a36c0 2022-01-09 op strlcpy(dir, imsg.data, sizeof(dir));
308 fb1a36c0 2022-01-09 op client_privdrop(username, dir);
309 fb1a36c0 2022-01-09 op memset(username, 0, sizeof(username));
310 fb1a36c0 2022-01-09 op memset(dir, 0, sizeof(username));
311 fb1a36c0 2022-01-09 op break;
312 fb1a36c0 2022-01-09 op case IMSG_BUF:
313 fb1a36c0 2022-01-09 op /* echo! */
314 fb1a36c0 2022-01-09 op if (!auth)
315 fb1a36c0 2022-01-09 op fatalx("%s: can't handle messages before"
316 fb1a36c0 2022-01-09 op " doing the auth", __func__);
317 fb1a36c0 2022-01-09 op handle_message(&imsg, IMSG_DATA_SIZE(imsg));
318 fb1a36c0 2022-01-09 op break;
319 fb1a36c0 2022-01-09 op case IMSG_CONN_GONE:
320 fb1a36c0 2022-01-09 op log_debug("closing");
321 fb1a36c0 2022-01-09 op shut = 1;
322 fb1a36c0 2022-01-09 op break;
323 fb1a36c0 2022-01-09 op default:
324 fb1a36c0 2022-01-09 op log_debug("%s: unexpected imsg %d",
325 fb1a36c0 2022-01-09 op __func__, imsg.hdr.type);
326 fb1a36c0 2022-01-09 op break;
327 fb1a36c0 2022-01-09 op }
328 fb1a36c0 2022-01-09 op imsg_free(&imsg);
329 fb1a36c0 2022-01-09 op }
330 fb1a36c0 2022-01-09 op
331 fb1a36c0 2022-01-09 op if (!shut)
332 fb1a36c0 2022-01-09 op imsg_event_add(iev);
333 fb1a36c0 2022-01-09 op else {
334 fb1a36c0 2022-01-09 op /* This pipe is dead. Remove its event handler. */
335 fb1a36c0 2022-01-09 op event_del(&iev->ev);
336 fb1a36c0 2022-01-09 op log_debug("pipe closed, shutting down...");
337 fb1a36c0 2022-01-09 op event_loopexit(NULL);
338 fb1a36c0 2022-01-09 op }
339 fb1a36c0 2022-01-09 op }
340 fb1a36c0 2022-01-09 op
341 fb1a36c0 2022-01-09 op static void
342 fb1a36c0 2022-01-09 op client_privdrop(const char *username, const char *dir)
343 fb1a36c0 2022-01-09 op {
344 fb1a36c0 2022-01-09 op struct passwd *pw;
345 fb1a36c0 2022-01-09 op
346 fb1a36c0 2022-01-09 op setproctitle("client %s", username);
347 fb1a36c0 2022-01-09 op
348 fb1a36c0 2022-01-09 op if ((pw = getpwnam(username)) == NULL)
349 fb1a36c0 2022-01-09 op fatalx("getpwnam(%s) failed", username);
350 fb1a36c0 2022-01-09 op
351 fb1a36c0 2022-01-09 op if (chroot(dir) == -1)
352 fb1a36c0 2022-01-09 op fatal("chroot");
353 fb1a36c0 2022-01-09 op if (chdir("/") == -1)
354 fb1a36c0 2022-01-09 op fatal("chdir(\"/\")");
355 fb1a36c0 2022-01-09 op
356 fb1a36c0 2022-01-09 op if (setgroups(1, &pw->pw_gid) ||
357 fb1a36c0 2022-01-09 op setresgid(pw->pw_gid, pw->pw_gid, pw->pw_gid) ||
358 fb1a36c0 2022-01-09 op setresuid(pw->pw_uid, pw->pw_uid, pw->pw_uid))
359 fb1a36c0 2022-01-09 op fatal("can't drop privileges");
360 fb1a36c0 2022-01-09 op
361 fb1a36c0 2022-01-09 op sandbox_client();
362 fb1a36c0 2022-01-09 op log_debug("client ready; user=%s dir=%s", username, dir);
363 fb1a36c0 2022-01-09 op
364 fb1a36c0 2022-01-09 op if ((evb = evbuffer_new()) == NULL)
365 fb1a36c0 2022-01-09 op fatal("evbuffer_new");
366 fb1a36c0 2022-01-09 op }
367 fb1a36c0 2022-01-09 op
368 fb1a36c0 2022-01-09 op static int
369 fb1a36c0 2022-01-09 op client_send_listener(int type, const void *data, uint16_t len)
370 fb1a36c0 2022-01-09 op {
371 fb1a36c0 2022-01-09 op int ret;
372 fb1a36c0 2022-01-09 op
373 fb1a36c0 2022-01-09 op if ((ret = imsg_compose(&iev_listener->ibuf, type, peerid, 0, -1,
374 fb1a36c0 2022-01-09 op data, len)) != -1)
375 fb1a36c0 2022-01-09 op imsg_event_add(iev_listener);
376 fb1a36c0 2022-01-09 op
377 fb1a36c0 2022-01-09 op return ret;
378 fb1a36c0 2022-01-09 op }
379 fb1a36c0 2022-01-09 op
380 fb1a36c0 2022-01-09 op /* set qid fields from sb */
381 fb1a36c0 2022-01-09 op static void
382 fb1a36c0 2022-01-09 op qid_update_from_sb(struct qid *qid, struct stat *sb)
383 fb1a36c0 2022-01-09 op {
384 fb1a36c0 2022-01-09 op qid->path = sb->st_ino;
385 fb1a36c0 2022-01-09 op
386 fb1a36c0 2022-01-09 op /*
387 fb1a36c0 2022-01-09 op * Theoretically (and hopefully!) this should be a 64 bit
388 fb1a36c0 2022-01-09 op * number. Unfortunately, 9P uses 32 bit timestamps.
389 fb1a36c0 2022-01-09 op */
390 fb1a36c0 2022-01-09 op qid->vers = sb->st_mtim.tv_sec;
391 fb1a36c0 2022-01-09 op
392 fb1a36c0 2022-01-09 op if (S_ISREG(sb->st_mode))
393 fb1a36c0 2022-01-09 op qid->type = QTFILE;
394 fb1a36c0 2022-01-09 op else if (S_ISDIR(sb->st_mode))
395 fb1a36c0 2022-01-09 op qid->type = QTDIR;
396 fb1a36c0 2022-01-09 op else if (S_ISLNK(sb->st_mode))
397 fb1a36c0 2022-01-09 op qid->type = QTSYMLINK;
398 fb1a36c0 2022-01-09 op }
399 fb1a36c0 2022-01-09 op
400 fb1a36c0 2022-01-09 op /* creates a qid given a fd */
401 fb1a36c0 2022-01-09 op static struct dir *
402 fb1a36c0 2022-01-09 op new_dir(int fd)
403 fb1a36c0 2022-01-09 op {
404 fb1a36c0 2022-01-09 op struct dir *dir;
405 fb1a36c0 2022-01-09 op
406 fb1a36c0 2022-01-09 op if ((dir = calloc(1, sizeof(*dir))) == NULL)
407 fb1a36c0 2022-01-09 op return NULL;
408 fb1a36c0 2022-01-09 op
409 fb1a36c0 2022-01-09 op dir->fd = fd;
410 fb1a36c0 2022-01-09 op STAILQ_INSERT_HEAD(&dirs, dir, entries);
411 fb1a36c0 2022-01-09 op return dir;
412 fb1a36c0 2022-01-09 op }
413 fb1a36c0 2022-01-09 op
414 fb1a36c0 2022-01-09 op static struct dir *
415 fb1a36c0 2022-01-09 op dir_incref(struct dir *dir)
416 fb1a36c0 2022-01-09 op {
417 fb1a36c0 2022-01-09 op dir->refcount++;
418 fb1a36c0 2022-01-09 op return dir;
419 fb1a36c0 2022-01-09 op }
420 fb1a36c0 2022-01-09 op
421 fb1a36c0 2022-01-09 op static void
422 fb1a36c0 2022-01-09 op dir_decref(struct dir *dir)
423 fb1a36c0 2022-01-09 op {
424 fb1a36c0 2022-01-09 op if (--dir->refcount > 0)
425 fb1a36c0 2022-01-09 op return;
426 fb1a36c0 2022-01-09 op
427 fb1a36c0 2022-01-09 op STAILQ_REMOVE(&dirs, dir, dir, entries);
428 fb1a36c0 2022-01-09 op
429 fb1a36c0 2022-01-09 op close(dir->fd);
430 fb1a36c0 2022-01-09 op free(dir);
431 fb1a36c0 2022-01-09 op }
432 fb1a36c0 2022-01-09 op
433 fb1a36c0 2022-01-09 op static struct fid *
434 fb1a36c0 2022-01-09 op new_fid(struct dir *dir, uint32_t fid, const char *path, struct qid *qid)
435 fb1a36c0 2022-01-09 op {
436 fb1a36c0 2022-01-09 op struct fid *f;
437 fb1a36c0 2022-01-09 op struct qid q;
438 fb1a36c0 2022-01-09 op struct stat sb;
439 fb1a36c0 2022-01-09 op
440 fb1a36c0 2022-01-09 op if (qid == NULL) {
441 fb1a36c0 2022-01-09 op if (fstatat(dir->fd, path, &sb, 0)) {
442 fb1a36c0 2022-01-09 op log_warn("fstatat(%s)", path);
443 fb1a36c0 2022-01-09 op return NULL;
444 fb1a36c0 2022-01-09 op }
445 fb1a36c0 2022-01-09 op qid_update_from_sb(&q, &sb);
446 fb1a36c0 2022-01-09 op qid = &q;
447 fb1a36c0 2022-01-09 op }
448 fb1a36c0 2022-01-09 op
449 fb1a36c0 2022-01-09 op if ((f = calloc(1, sizeof(*f))) == NULL)
450 fb1a36c0 2022-01-09 op return NULL;
451 fb1a36c0 2022-01-09 op
452 fb1a36c0 2022-01-09 op f->dir = dir_incref(dir);
453 fb1a36c0 2022-01-09 op f->fid = fid;
454 fb1a36c0 2022-01-09 op f->fd = -1;
455 fb1a36c0 2022-01-09 op
456 fb1a36c0 2022-01-09 op strlcpy(f->fpath, path, sizeof(f->fpath));
457 fb1a36c0 2022-01-09 op
458 fb1a36c0 2022-01-09 op memcpy(&f->qid, qid, sizeof(f->qid));
459 fb1a36c0 2022-01-09 op
460 fb1a36c0 2022-01-09 op STAILQ_INSERT_HEAD(&fids, f, entries);
461 fb1a36c0 2022-01-09 op
462 fb1a36c0 2022-01-09 op return f;
463 fb1a36c0 2022-01-09 op }
464 fb1a36c0 2022-01-09 op
465 fb1a36c0 2022-01-09 op static struct fid *
466 fb1a36c0 2022-01-09 op fid_by_id(uint32_t fid)
467 fb1a36c0 2022-01-09 op {
468 fb1a36c0 2022-01-09 op struct fid *f;
469 fb1a36c0 2022-01-09 op
470 fb1a36c0 2022-01-09 op STAILQ_FOREACH(f, &fids, entries) {
471 fb1a36c0 2022-01-09 op if (f->fid == fid)
472 fb1a36c0 2022-01-09 op return f;
473 fb1a36c0 2022-01-09 op }
474 fb1a36c0 2022-01-09 op
475 fb1a36c0 2022-01-09 op return NULL;
476 fb1a36c0 2022-01-09 op }
477 fb1a36c0 2022-01-09 op
478 fb1a36c0 2022-01-09 op static void
479 fb1a36c0 2022-01-09 op free_fid(struct fid *f)
480 fb1a36c0 2022-01-09 op {
481 fb1a36c0 2022-01-09 op int r;
482 fb1a36c0 2022-01-09 op
483 fb1a36c0 2022-01-09 op if (f->fd != -1) {
484 fb1a36c0 2022-01-09 op if (f->d != NULL)
485 fb1a36c0 2022-01-09 op r = closedir(f->d);
486 fb1a36c0 2022-01-09 op else
487 fb1a36c0 2022-01-09 op r = close(f->fd);
488 fb1a36c0 2022-01-09 op
489 fb1a36c0 2022-01-09 op if (r == -1)
490 fb1a36c0 2022-01-09 op fatal("can't close fid %d", f->fid);
491 fb1a36c0 2022-01-09 op
492 fb1a36c0 2022-01-09 op if (f->evb != NULL)
493 fb1a36c0 2022-01-09 op evbuffer_free(f->evb);
494 fb1a36c0 2022-01-09 op
495 fb1a36c0 2022-01-09 op /* try to honour ORCLOSE if requested */
496 fb1a36c0 2022-01-09 op if (f->iomode & O_CLOEXEC)
497 fb1a36c0 2022-01-09 op unlinkat(f->dir->fd, f->fpath, 0);
498 fb1a36c0 2022-01-09 op }
499 fb1a36c0 2022-01-09 op
500 fb1a36c0 2022-01-09 op dir_decref(f->dir);
501 fb1a36c0 2022-01-09 op
502 fb1a36c0 2022-01-09 op STAILQ_REMOVE(&fids, f, fid, entries);
503 fb1a36c0 2022-01-09 op free(f);
504 fb1a36c0 2022-01-09 op }
505 fb1a36c0 2022-01-09 op
506 fb1a36c0 2022-01-09 op static void
507 fb1a36c0 2022-01-09 op parse_message(const uint8_t *data, size_t len, struct np_msg_header *hdr,
508 fb1a36c0 2022-01-09 op uint8_t **cnt)
509 fb1a36c0 2022-01-09 op {
510 fb1a36c0 2022-01-09 op size_t olen = len;
511 fb1a36c0 2022-01-09 op
512 fb1a36c0 2022-01-09 op if (!NPREAD32("len", &hdr->len, &data, &len) ||
513 fb1a36c0 2022-01-09 op !NPREAD8("type", &hdr->type, &data, &len) ||
514 fb1a36c0 2022-01-09 op !NPREAD16("tag", &hdr->tag, &data, &len))
515 fb1a36c0 2022-01-09 op goto err;
516 fb1a36c0 2022-01-09 op
517 fb1a36c0 2022-01-09 op if (olen != hdr->len)
518 fb1a36c0 2022-01-09 op goto err;
519 fb1a36c0 2022-01-09 op
520 fb1a36c0 2022-01-09 op if (hdr->type < Tversion ||
521 fb1a36c0 2022-01-09 op hdr->type >= Tmax ||
522 fb1a36c0 2022-01-09 op hdr->type == Terror ||
523 fb1a36c0 2022-01-09 op (hdr->type & 0x1) != 0) /* cannot recv a R* */
524 fb1a36c0 2022-01-09 op goto err;
525 fb1a36c0 2022-01-09 op
526 fb1a36c0 2022-01-09 op hdr->tag = le32toh(hdr->tag);
527 fb1a36c0 2022-01-09 op
528 fb1a36c0 2022-01-09 op *cnt = (uint8_t *)data;
529 fb1a36c0 2022-01-09 op return;
530 fb1a36c0 2022-01-09 op
531 fb1a36c0 2022-01-09 op err:
532 fb1a36c0 2022-01-09 op /* TODO: send a proper message to terminate the connection. */
533 fb1a36c0 2022-01-09 op fatalx("got invalid message");
534 fb1a36c0 2022-01-09 op }
535 fb1a36c0 2022-01-09 op
536 fb1a36c0 2022-01-09 op static void
537 fb1a36c0 2022-01-09 op np_write16(struct evbuffer *e, uint16_t x)
538 fb1a36c0 2022-01-09 op {
539 fb1a36c0 2022-01-09 op x = htole16(x);
540 fb1a36c0 2022-01-09 op evbuffer_add(e, &x, sizeof(x));
541 fb1a36c0 2022-01-09 op }
542 fb1a36c0 2022-01-09 op
543 fb1a36c0 2022-01-09 op static void
544 fb1a36c0 2022-01-09 op np_write32(struct evbuffer *e, uint32_t x)
545 fb1a36c0 2022-01-09 op {
546 fb1a36c0 2022-01-09 op x = htole32(x);
547 fb1a36c0 2022-01-09 op evbuffer_add(e, &x, sizeof(x));
548 fb1a36c0 2022-01-09 op }
549 fb1a36c0 2022-01-09 op
550 fb1a36c0 2022-01-09 op static void
551 fb1a36c0 2022-01-09 op np_write64(struct evbuffer *e, uint64_t x)
552 fb1a36c0 2022-01-09 op {
553 fb1a36c0 2022-01-09 op x = htole64(x);
554 fb1a36c0 2022-01-09 op evbuffer_add(e, &x, sizeof(x));
555 fb1a36c0 2022-01-09 op }
556 fb1a36c0 2022-01-09 op
557 fb1a36c0 2022-01-09 op static void
558 fb1a36c0 2022-01-09 op np_writebuf(struct evbuffer *e, size_t len, void *data)
559 fb1a36c0 2022-01-09 op {
560 fb1a36c0 2022-01-09 op evbuffer_add(e, data, len);
561 fb1a36c0 2022-01-09 op }
562 fb1a36c0 2022-01-09 op
563 fb1a36c0 2022-01-09 op static void
564 fb1a36c0 2022-01-09 op np_header(uint32_t len, uint8_t type, uint16_t tag)
565 fb1a36c0 2022-01-09 op {
566 fb1a36c0 2022-01-09 op len += HEADERSIZE;
567 fb1a36c0 2022-01-09 op
568 fb1a36c0 2022-01-09 op len = htole32(len);
569 fb1a36c0 2022-01-09 op tag = htole16(tag);
570 fb1a36c0 2022-01-09 op
571 fb1a36c0 2022-01-09 op evbuffer_add(evb, &len, sizeof(len));
572 fb1a36c0 2022-01-09 op evbuffer_add(evb, &type, sizeof(type));
573 fb1a36c0 2022-01-09 op evbuffer_add(evb, &tag, sizeof(tag));
574 fb1a36c0 2022-01-09 op }
575 fb1a36c0 2022-01-09 op
576 fb1a36c0 2022-01-09 op static void
577 fb1a36c0 2022-01-09 op np_string(struct evbuffer *e, uint16_t len, const char *str)
578 fb1a36c0 2022-01-09 op {
579 fb1a36c0 2022-01-09 op uint16_t l = len;
580 fb1a36c0 2022-01-09 op
581 fb1a36c0 2022-01-09 op len = htole16(len);
582 fb1a36c0 2022-01-09 op evbuffer_add(e, &len, sizeof(len));
583 fb1a36c0 2022-01-09 op evbuffer_add(e, str, l);
584 fb1a36c0 2022-01-09 op }
585 fb1a36c0 2022-01-09 op
586 fb1a36c0 2022-01-09 op static void
587 fb1a36c0 2022-01-09 op np_qid(struct evbuffer *e, struct qid *qid)
588 fb1a36c0 2022-01-09 op {
589 fb1a36c0 2022-01-09 op uint64_t path;
590 fb1a36c0 2022-01-09 op uint32_t vers;
591 fb1a36c0 2022-01-09 op
592 fb1a36c0 2022-01-09 op path = htole64(qid->path);
593 fb1a36c0 2022-01-09 op vers = htole32(qid->vers);
594 fb1a36c0 2022-01-09 op
595 fb1a36c0 2022-01-09 op evbuffer_add(e, &qid->type, sizeof(qid->type));
596 fb1a36c0 2022-01-09 op evbuffer_add(e, &vers, sizeof(vers));
597 fb1a36c0 2022-01-09 op evbuffer_add(e, &path, sizeof(path));
598 fb1a36c0 2022-01-09 op }
599 fb1a36c0 2022-01-09 op
600 fb1a36c0 2022-01-09 op static void
601 fb1a36c0 2022-01-09 op do_send(void)
602 fb1a36c0 2022-01-09 op {
603 fb1a36c0 2022-01-09 op size_t len;
604 fb1a36c0 2022-01-09 op void *data;
605 fb1a36c0 2022-01-09 op
606 fb1a36c0 2022-01-09 op len = EVBUFFER_LENGTH(evb);
607 fb1a36c0 2022-01-09 op data = EVBUFFER_DATA(evb);
608 fb1a36c0 2022-01-09 op
609 fb1a36c0 2022-01-09 op #if DEBUG_PACKETS
610 fb1a36c0 2022-01-09 op hexdump("outgoing packet", data, len);
611 fb1a36c0 2022-01-09 op #endif
612 fb1a36c0 2022-01-09 op client_send_listener(IMSG_BUF, data, len);
613 fb1a36c0 2022-01-09 op evbuffer_drain(evb, len);
614 fb1a36c0 2022-01-09 op }
615 fb1a36c0 2022-01-09 op
616 fb1a36c0 2022-01-09 op static void
617 fb1a36c0 2022-01-09 op np_version(uint16_t tag, uint32_t msize, const char *version)
618 fb1a36c0 2022-01-09 op {
619 fb1a36c0 2022-01-09 op uint16_t l;
620 fb1a36c0 2022-01-09 op
621 fb1a36c0 2022-01-09 op l = strlen(version);
622 fb1a36c0 2022-01-09 op
623 fb1a36c0 2022-01-09 op msize = htole32(msize);
624 fb1a36c0 2022-01-09 op
625 fb1a36c0 2022-01-09 op np_header(sizeof(msize) + sizeof(l) + l, Rversion, tag);
626 fb1a36c0 2022-01-09 op evbuffer_add(evb, &msize, sizeof(msize));
627 fb1a36c0 2022-01-09 op np_string(evb, l, version);
628 fb1a36c0 2022-01-09 op do_send();
629 fb1a36c0 2022-01-09 op }
630 fb1a36c0 2022-01-09 op
631 fb1a36c0 2022-01-09 op static void
632 fb1a36c0 2022-01-09 op np_attach(uint16_t tag, struct qid *qid)
633 fb1a36c0 2022-01-09 op {
634 fb1a36c0 2022-01-09 op np_header(QIDSIZE, Rattach, tag);
635 fb1a36c0 2022-01-09 op np_qid(evb, qid);
636 fb1a36c0 2022-01-09 op do_send();
637 fb1a36c0 2022-01-09 op }
638 fb1a36c0 2022-01-09 op
639 fb1a36c0 2022-01-09 op static void
640 fb1a36c0 2022-01-09 op np_clunk(uint16_t tag)
641 fb1a36c0 2022-01-09 op {
642 fb1a36c0 2022-01-09 op np_header(0, Rclunk, tag);
643 fb1a36c0 2022-01-09 op do_send();
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 np_flush(uint16_t tag)
648 fb1a36c0 2022-01-09 op {
649 fb1a36c0 2022-01-09 op np_header(0, Rflush, tag);
650 fb1a36c0 2022-01-09 op do_send();
651 fb1a36c0 2022-01-09 op }
652 fb1a36c0 2022-01-09 op
653 fb1a36c0 2022-01-09 op static void
654 fb1a36c0 2022-01-09 op np_walk(uint16_t tag, int nwqid, struct qid *wqid)
655 fb1a36c0 2022-01-09 op {
656 fb1a36c0 2022-01-09 op int i;
657 fb1a36c0 2022-01-09 op
658 fb1a36c0 2022-01-09 op /* two bytes for the counter */
659 fb1a36c0 2022-01-09 op np_header(2 + QIDSIZE * nwqid, Rwalk, tag);
660 fb1a36c0 2022-01-09 op np_write16(evb, nwqid);
661 fb1a36c0 2022-01-09 op for (i = 0; i < nwqid; ++i)
662 fb1a36c0 2022-01-09 op np_qid(evb, wqid + i);
663 fb1a36c0 2022-01-09 op
664 fb1a36c0 2022-01-09 op do_send();
665 fb1a36c0 2022-01-09 op }
666 fb1a36c0 2022-01-09 op
667 fb1a36c0 2022-01-09 op static void
668 fb1a36c0 2022-01-09 op np_open(uint16_t tag, struct qid *qid, uint32_t iounit)
669 fb1a36c0 2022-01-09 op {
670 fb1a36c0 2022-01-09 op np_header(QIDSIZE + sizeof(iounit), Ropen, tag);
671 fb1a36c0 2022-01-09 op np_qid(evb, qid);
672 fb1a36c0 2022-01-09 op np_write32(evb, iounit);
673 fb1a36c0 2022-01-09 op do_send();
674 fb1a36c0 2022-01-09 op }
675 fb1a36c0 2022-01-09 op
676 fb1a36c0 2022-01-09 op static void
677 fb1a36c0 2022-01-09 op np_create(uint16_t tag, struct qid *qid, uint32_t iounit)
678 fb1a36c0 2022-01-09 op {
679 fb1a36c0 2022-01-09 op np_header(QIDSIZE + sizeof(iounit), Rcreate, tag);
680 fb1a36c0 2022-01-09 op np_qid(evb, qid);
681 fb1a36c0 2022-01-09 op np_write32(evb, iounit);
682 fb1a36c0 2022-01-09 op do_send();
683 fb1a36c0 2022-01-09 op }
684 fb1a36c0 2022-01-09 op
685 fb1a36c0 2022-01-09 op static void
686 fb1a36c0 2022-01-09 op np_read(uint16_t tag, uint32_t count, void *data)
687 fb1a36c0 2022-01-09 op {
688 fb1a36c0 2022-01-09 op if (sizeof(count) + count + HEADERSIZE >= msize) {
689 fb1a36c0 2022-01-09 op np_error(tag, "Rread would overflow");
690 fb1a36c0 2022-01-09 op return;
691 fb1a36c0 2022-01-09 op }
692 fb1a36c0 2022-01-09 op
693 fb1a36c0 2022-01-09 op np_header(sizeof(count) + count, Rread, tag);
694 fb1a36c0 2022-01-09 op np_write32(evb, count);
695 fb1a36c0 2022-01-09 op np_writebuf(evb, count, data);
696 fb1a36c0 2022-01-09 op do_send();
697 fb1a36c0 2022-01-09 op }
698 fb1a36c0 2022-01-09 op
699 fb1a36c0 2022-01-09 op static void
700 fb1a36c0 2022-01-09 op np_write(uint16_t tag, uint32_t count)
701 fb1a36c0 2022-01-09 op {
702 fb1a36c0 2022-01-09 op np_header(sizeof(count), Rwrite, tag);
703 fb1a36c0 2022-01-09 op np_write32(evb, count);
704 fb1a36c0 2022-01-09 op do_send();
705 fb1a36c0 2022-01-09 op }
706 fb1a36c0 2022-01-09 op
707 fb1a36c0 2022-01-09 op static void
708 fb1a36c0 2022-01-09 op np_stat(uint16_t tag, uint32_t count, void *data)
709 fb1a36c0 2022-01-09 op {
710 fb1a36c0 2022-01-09 op if (sizeof(count) + count + HEADERSIZE >= msize) {
711 fb1a36c0 2022-01-09 op np_error(tag, "Rstat would overflow");
712 fb1a36c0 2022-01-09 op return;
713 fb1a36c0 2022-01-09 op }
714 fb1a36c0 2022-01-09 op
715 fb1a36c0 2022-01-09 op np_header(count, Rstat, tag);
716 fb1a36c0 2022-01-09 op np_writebuf(evb, count, data);
717 fb1a36c0 2022-01-09 op do_send();
718 fb1a36c0 2022-01-09 op }
719 fb1a36c0 2022-01-09 op
720 fb1a36c0 2022-01-09 op static void
721 fb1a36c0 2022-01-09 op np_remove(uint16_t tag)
722 fb1a36c0 2022-01-09 op {
723 fb1a36c0 2022-01-09 op np_header(0, Rremove, tag);
724 fb1a36c0 2022-01-09 op do_send();
725 fb1a36c0 2022-01-09 op }
726 fb1a36c0 2022-01-09 op
727 fb1a36c0 2022-01-09 op static void
728 fb1a36c0 2022-01-09 op np_error(uint16_t tag, const char *errstr)
729 fb1a36c0 2022-01-09 op {
730 fb1a36c0 2022-01-09 op uint16_t l;
731 fb1a36c0 2022-01-09 op
732 fb1a36c0 2022-01-09 op l = strlen(errstr);
733 fb1a36c0 2022-01-09 op
734 fb1a36c0 2022-01-09 op np_header(sizeof(l) + l, Rerror, tag);
735 fb1a36c0 2022-01-09 op np_string(evb, l, errstr);
736 fb1a36c0 2022-01-09 op do_send();
737 fb1a36c0 2022-01-09 op }
738 fb1a36c0 2022-01-09 op
739 fb1a36c0 2022-01-09 op static void
740 fb1a36c0 2022-01-09 op np_errno(uint16_t tag)
741 fb1a36c0 2022-01-09 op {
742 fb1a36c0 2022-01-09 op int saved_errno;
743 fb1a36c0 2022-01-09 op char buf[NL_TEXTMAX] = {0};
744 fb1a36c0 2022-01-09 op
745 fb1a36c0 2022-01-09 op saved_errno = errno;
746 fb1a36c0 2022-01-09 op
747 fb1a36c0 2022-01-09 op strerror_r(errno, buf, sizeof(buf));
748 fb1a36c0 2022-01-09 op np_error(tag, buf);
749 fb1a36c0 2022-01-09 op
750 fb1a36c0 2022-01-09 op errno = saved_errno;
751 fb1a36c0 2022-01-09 op }
752 fb1a36c0 2022-01-09 op
753 fb1a36c0 2022-01-09 op static int
754 fb1a36c0 2022-01-09 op np_read8(const char *t, const char *f, uint8_t *dst, const uint8_t **src,
755 fb1a36c0 2022-01-09 op size_t *len)
756 fb1a36c0 2022-01-09 op {
757 fb1a36c0 2022-01-09 op if (*len < sizeof(*dst)) {
758 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %zu bytes for the %s field but only "
759 fb1a36c0 2022-01-09 op "%zu are available.", t, sizeof(*dst), f, *len);
760 fb1a36c0 2022-01-09 op return -1;
761 fb1a36c0 2022-01-09 op }
762 fb1a36c0 2022-01-09 op
763 fb1a36c0 2022-01-09 op memcpy(dst, *src, sizeof(*dst));
764 fb1a36c0 2022-01-09 op *src += sizeof(*dst);
765 fb1a36c0 2022-01-09 op *len -= sizeof(*dst);
766 fb1a36c0 2022-01-09 op
767 fb1a36c0 2022-01-09 op return 1;
768 fb1a36c0 2022-01-09 op }
769 fb1a36c0 2022-01-09 op
770 fb1a36c0 2022-01-09 op static int
771 fb1a36c0 2022-01-09 op np_read16(const char *t, const char *f, uint16_t *dst, const uint8_t **src,
772 fb1a36c0 2022-01-09 op size_t *len)
773 fb1a36c0 2022-01-09 op {
774 fb1a36c0 2022-01-09 op if (*len < sizeof(*dst)) {
775 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %zu bytes for the %s field but only "
776 fb1a36c0 2022-01-09 op "%zu are available.", t, sizeof(*dst), f, *len);
777 fb1a36c0 2022-01-09 op return -1;
778 fb1a36c0 2022-01-09 op }
779 fb1a36c0 2022-01-09 op
780 fb1a36c0 2022-01-09 op memcpy(dst, *src, sizeof(*dst));
781 fb1a36c0 2022-01-09 op *src += sizeof(*dst);
782 fb1a36c0 2022-01-09 op *len -= sizeof(*dst);
783 fb1a36c0 2022-01-09 op *dst = le16toh(*dst);
784 fb1a36c0 2022-01-09 op
785 fb1a36c0 2022-01-09 op return 1;
786 fb1a36c0 2022-01-09 op }
787 fb1a36c0 2022-01-09 op
788 fb1a36c0 2022-01-09 op static int
789 fb1a36c0 2022-01-09 op np_read32(const char *t, const char *f, uint32_t *dst, const uint8_t **src,
790 fb1a36c0 2022-01-09 op size_t *len)
791 fb1a36c0 2022-01-09 op {
792 fb1a36c0 2022-01-09 op if (*len < sizeof(*dst)) {
793 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %zu bytes for the %s field but only "
794 fb1a36c0 2022-01-09 op "%zu are available.", t, sizeof(*dst), f, *len);
795 fb1a36c0 2022-01-09 op return -1;
796 fb1a36c0 2022-01-09 op }
797 fb1a36c0 2022-01-09 op
798 fb1a36c0 2022-01-09 op memcpy(dst, *src, sizeof(*dst));
799 fb1a36c0 2022-01-09 op *src += sizeof(*dst);
800 fb1a36c0 2022-01-09 op *len -= sizeof(*dst);
801 fb1a36c0 2022-01-09 op *dst = le32toh(*dst);
802 fb1a36c0 2022-01-09 op
803 fb1a36c0 2022-01-09 op return 1;
804 fb1a36c0 2022-01-09 op }
805 fb1a36c0 2022-01-09 op
806 fb1a36c0 2022-01-09 op static int
807 fb1a36c0 2022-01-09 op np_read64(const char *t, const char *f, uint64_t *dst, const uint8_t **src,
808 fb1a36c0 2022-01-09 op size_t *len)
809 fb1a36c0 2022-01-09 op {
810 fb1a36c0 2022-01-09 op if (*len < sizeof(*dst)) {
811 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %zu bytes for the %s field but only "
812 fb1a36c0 2022-01-09 op "%zu are available.", t, sizeof(*dst), f, *len);
813 fb1a36c0 2022-01-09 op return -1;
814 fb1a36c0 2022-01-09 op }
815 fb1a36c0 2022-01-09 op
816 fb1a36c0 2022-01-09 op memcpy(dst, *src, sizeof(*dst));
817 fb1a36c0 2022-01-09 op *src += sizeof(*dst);
818 fb1a36c0 2022-01-09 op *len -= sizeof(*dst);
819 fb1a36c0 2022-01-09 op *dst = le64toh(*dst);
820 fb1a36c0 2022-01-09 op
821 fb1a36c0 2022-01-09 op return 1;
822 fb1a36c0 2022-01-09 op }
823 fb1a36c0 2022-01-09 op
824 fb1a36c0 2022-01-09 op static int
825 fb1a36c0 2022-01-09 op np_readstr(const char *t, const char *f, char *res, size_t reslen,
826 fb1a36c0 2022-01-09 op const uint8_t **src, size_t *len)
827 fb1a36c0 2022-01-09 op {
828 fb1a36c0 2022-01-09 op uint16_t sl;
829 fb1a36c0 2022-01-09 op char buf[32];
830 fb1a36c0 2022-01-09 op
831 fb1a36c0 2022-01-09 op strlcpy(buf, f, sizeof(buf));
832 fb1a36c0 2022-01-09 op strlcat(buf, "-len", sizeof(buf));
833 fb1a36c0 2022-01-09 op
834 fb1a36c0 2022-01-09 op if (!np_read16(t, buf, &sl, src, len))
835 fb1a36c0 2022-01-09 op return READSTRERR;
836 fb1a36c0 2022-01-09 op
837 fb1a36c0 2022-01-09 op if (*len < sl) {
838 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %d bytes for the %s field but only "
839 fb1a36c0 2022-01-09 op "%zu are available.", t, sl, f, *len);
840 fb1a36c0 2022-01-09 op return READSTRERR;
841 fb1a36c0 2022-01-09 op }
842 fb1a36c0 2022-01-09 op
843 fb1a36c0 2022-01-09 op if (*len > reslen-1)
844 fb1a36c0 2022-01-09 op return READSTRTRUNC;
845 fb1a36c0 2022-01-09 op
846 fb1a36c0 2022-01-09 op memcpy(res, *src, sl);
847 fb1a36c0 2022-01-09 op res[sl] = '\0';
848 fb1a36c0 2022-01-09 op *src += sl;
849 fb1a36c0 2022-01-09 op *len -= sl;
850 fb1a36c0 2022-01-09 op
851 fb1a36c0 2022-01-09 op return 0;
852 fb1a36c0 2022-01-09 op }
853 fb1a36c0 2022-01-09 op
854 fb1a36c0 2022-01-09 op static void
855 fb1a36c0 2022-01-09 op tversion(struct np_msg_header *hdr, const uint8_t *data, size_t len)
856 fb1a36c0 2022-01-09 op {
857 fb1a36c0 2022-01-09 op char *dot, version[32];
858 fb1a36c0 2022-01-09 op
859 fb1a36c0 2022-01-09 op if (handshaked)
860 fb1a36c0 2022-01-09 op goto err;
861 fb1a36c0 2022-01-09 op
862 fb1a36c0 2022-01-09 op /* msize[4] version[s] */
863 fb1a36c0 2022-01-09 op if (!NPREAD32("msize", &msize, &data, &len))
864 fb1a36c0 2022-01-09 op goto err;
865 fb1a36c0 2022-01-09 op
866 fb1a36c0 2022-01-09 op switch (NPREADSTR("version", version, sizeof(version), &data, &len)) {
867 fb1a36c0 2022-01-09 op case READSTRERR:
868 fb1a36c0 2022-01-09 op goto err;
869 fb1a36c0 2022-01-09 op case READSTRTRUNC:
870 fb1a36c0 2022-01-09 op log_warnx("9P version string too long, truncated");
871 fb1a36c0 2022-01-09 op goto mismatch;
872 fb1a36c0 2022-01-09 op }
873 fb1a36c0 2022-01-09 op
874 fb1a36c0 2022-01-09 op if ((dot = strchr(version, '.')) != NULL)
875 fb1a36c0 2022-01-09 op *dot = '\0';
876 fb1a36c0 2022-01-09 op
877 fb1a36c0 2022-01-09 op if (strcmp(version, VERSION9P) != 0 ||
878 fb1a36c0 2022-01-09 op msize == 0)
879 fb1a36c0 2022-01-09 op goto mismatch;
880 fb1a36c0 2022-01-09 op
881 fb1a36c0 2022-01-09 op /* version matched */
882 fb1a36c0 2022-01-09 op handshaked = 1;
883 fb1a36c0 2022-01-09 op msize = MIN(msize, CLIENT_MSIZE);
884 fb1a36c0 2022-01-09 op client_send_listener(IMSG_MSIZE, &msize, sizeof(msize));
885 fb1a36c0 2022-01-09 op np_version(hdr->tag, msize, VERSION9P);
886 fb1a36c0 2022-01-09 op return;
887 fb1a36c0 2022-01-09 op
888 fb1a36c0 2022-01-09 op mismatch:
889 fb1a36c0 2022-01-09 op log_warnx("unknown 9P version string: \"%s\", want "VERSION9P,
890 fb1a36c0 2022-01-09 op version);
891 fb1a36c0 2022-01-09 op np_version(hdr->tag, MSIZE9P, "unknown");
892 fb1a36c0 2022-01-09 op return;
893 fb1a36c0 2022-01-09 op
894 fb1a36c0 2022-01-09 op err:
895 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
896 fb1a36c0 2022-01-09 op client_shutdown();
897 fb1a36c0 2022-01-09 op }
898 fb1a36c0 2022-01-09 op
899 fb1a36c0 2022-01-09 op static void
900 fb1a36c0 2022-01-09 op tattach(struct np_msg_header *hdr, const uint8_t *data, size_t len)
901 fb1a36c0 2022-01-09 op {
902 fb1a36c0 2022-01-09 op struct dir *dir;
903 fb1a36c0 2022-01-09 op struct fid *f;
904 fb1a36c0 2022-01-09 op uint32_t fid, afid;
905 fb1a36c0 2022-01-09 op int fd;
906 fb1a36c0 2022-01-09 op char aname[PATH_MAX];
907 fb1a36c0 2022-01-09 op
908 fb1a36c0 2022-01-09 op /* fid[4] afid[4] uname[s] aname[s] */
909 fb1a36c0 2022-01-09 op
910 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
911 fb1a36c0 2022-01-09 op !NPREAD32("afid", &afid, &data, &len))
912 fb1a36c0 2022-01-09 op goto err;
913 fb1a36c0 2022-01-09 op
914 fb1a36c0 2022-01-09 op /* read the uname but don't actually use it */
915 fb1a36c0 2022-01-09 op switch (NPREADSTR("uname", aname, sizeof(aname), &data, &len)) {
916 fb1a36c0 2022-01-09 op case READSTRERR:
917 fb1a36c0 2022-01-09 op goto err;
918 fb1a36c0 2022-01-09 op case READSTRTRUNC:
919 fb1a36c0 2022-01-09 op np_error(hdr->tag, "name too long");
920 fb1a36c0 2022-01-09 op return;
921 fb1a36c0 2022-01-09 op }
922 fb1a36c0 2022-01-09 op
923 fb1a36c0 2022-01-09 op switch (NPREADSTR("aname", aname, sizeof(aname), &data, &len)) {
924 fb1a36c0 2022-01-09 op case READSTRERR:
925 fb1a36c0 2022-01-09 op goto err;
926 fb1a36c0 2022-01-09 op case READSTRTRUNC:
927 fb1a36c0 2022-01-09 op np_error(hdr->tag, "name too long");
928 fb1a36c0 2022-01-09 op return;
929 fb1a36c0 2022-01-09 op }
930 fb1a36c0 2022-01-09 op
931 fb1a36c0 2022-01-09 op if (fid_by_id(fid) != NULL || afid != NOFID) {
932 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid or afid");
933 fb1a36c0 2022-01-09 op return;
934 fb1a36c0 2022-01-09 op }
935 fb1a36c0 2022-01-09 op
936 fb1a36c0 2022-01-09 op if ((fd = open(aname, O_RDONLY|O_DIRECTORY)) == -1)
937 fb1a36c0 2022-01-09 op goto fail;
938 fb1a36c0 2022-01-09 op
939 fb1a36c0 2022-01-09 op if ((dir = new_dir(fd)) == NULL)
940 fb1a36c0 2022-01-09 op goto fail;
941 fb1a36c0 2022-01-09 op
942 fb1a36c0 2022-01-09 op log_debug("attached %s to %d", aname, fid);
943 fb1a36c0 2022-01-09 op
944 fb1a36c0 2022-01-09 op if ((f = new_fid(dir, fid, aname, NULL)) == NULL) {
945 fb1a36c0 2022-01-09 op dir_decref(dir);
946 fb1a36c0 2022-01-09 op goto fail;
947 fb1a36c0 2022-01-09 op }
948 fb1a36c0 2022-01-09 op
949 fb1a36c0 2022-01-09 op np_attach(hdr->tag, &f->qid);
950 fb1a36c0 2022-01-09 op return;
951 fb1a36c0 2022-01-09 op
952 fb1a36c0 2022-01-09 op fail:
953 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
954 fb1a36c0 2022-01-09 op log_warn("failed to attach %s", aname);
955 fb1a36c0 2022-01-09 op return;
956 fb1a36c0 2022-01-09 op
957 fb1a36c0 2022-01-09 op err:
958 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
959 fb1a36c0 2022-01-09 op client_shutdown();
960 fb1a36c0 2022-01-09 op }
961 fb1a36c0 2022-01-09 op
962 fb1a36c0 2022-01-09 op static void
963 fb1a36c0 2022-01-09 op tclunk(struct np_msg_header *hdr, const uint8_t *data, size_t len)
964 fb1a36c0 2022-01-09 op {
965 fb1a36c0 2022-01-09 op struct fid *f;
966 fb1a36c0 2022-01-09 op uint32_t fid;
967 fb1a36c0 2022-01-09 op
968 fb1a36c0 2022-01-09 op /* fid[4] */
969 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len)) {
970 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
971 fb1a36c0 2022-01-09 op client_shutdown();
972 fb1a36c0 2022-01-09 op return;
973 fb1a36c0 2022-01-09 op }
974 fb1a36c0 2022-01-09 op
975 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL) {
976 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
977 fb1a36c0 2022-01-09 op return;
978 fb1a36c0 2022-01-09 op }
979 fb1a36c0 2022-01-09 op
980 fb1a36c0 2022-01-09 op free_fid(f);
981 fb1a36c0 2022-01-09 op np_clunk(hdr->tag);
982 fb1a36c0 2022-01-09 op }
983 fb1a36c0 2022-01-09 op
984 fb1a36c0 2022-01-09 op static void
985 fb1a36c0 2022-01-09 op tflush(struct np_msg_header *hdr, const uint8_t *data, size_t len)
986 fb1a36c0 2022-01-09 op {
987 fb1a36c0 2022-01-09 op uint16_t oldtag;
988 fb1a36c0 2022-01-09 op
989 fb1a36c0 2022-01-09 op /*
990 fb1a36c0 2022-01-09 op * We're doing only synchronous I/O. Tflush is implemented
991 fb1a36c0 2022-01-09 op * only because it's illegal to reply with a Rerror.
992 fb1a36c0 2022-01-09 op */
993 fb1a36c0 2022-01-09 op
994 fb1a36c0 2022-01-09 op /* oldtag[2] */
995 fb1a36c0 2022-01-09 op if (len != sizeof(oldtag)) {
996 fb1a36c0 2022-01-09 op log_warnx("Tflush with the wrong size: got %zu want %zu",
997 fb1a36c0 2022-01-09 op len, sizeof(oldtag));
998 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
999 fb1a36c0 2022-01-09 op client_shutdown();
1000 fb1a36c0 2022-01-09 op return;
1001 fb1a36c0 2022-01-09 op }
1002 fb1a36c0 2022-01-09 op
1003 fb1a36c0 2022-01-09 op np_flush(hdr->tag);
1004 fb1a36c0 2022-01-09 op }
1005 fb1a36c0 2022-01-09 op
1006 fb1a36c0 2022-01-09 op static void
1007 fb1a36c0 2022-01-09 op twalk(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1008 fb1a36c0 2022-01-09 op {
1009 fb1a36c0 2022-01-09 op struct stat sb;
1010 fb1a36c0 2022-01-09 op struct dir *dir;
1011 fb1a36c0 2022-01-09 op struct qid wqid[MAXWELEM] = {0};
1012 fb1a36c0 2022-01-09 op struct fid *f, *nf;
1013 fb1a36c0 2022-01-09 op uint32_t fid, newfid;
1014 fb1a36c0 2022-01-09 op uint16_t nwname;
1015 fb1a36c0 2022-01-09 op int fd, oldfd, no, nwqid = 0;
1016 fb1a36c0 2022-01-09 op char wnam[PATH_MAX];
1017 fb1a36c0 2022-01-09 op
1018 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
1019 fb1a36c0 2022-01-09 op !NPREAD32("newfid", &newfid, &data, &len) ||
1020 fb1a36c0 2022-01-09 op !NPREAD16("nwname", &nwname, &data, &len))
1021 fb1a36c0 2022-01-09 op goto err;
1022 fb1a36c0 2022-01-09 op
1023 fb1a36c0 2022-01-09 op if (nwname > MAXWELEM) {
1024 fb1a36c0 2022-01-09 op log_warnx("Twalk: more than %d path elements: %d",
1025 fb1a36c0 2022-01-09 op MAXWELEM, nwname);
1026 fb1a36c0 2022-01-09 op goto err;
1027 fb1a36c0 2022-01-09 op }
1028 fb1a36c0 2022-01-09 op
1029 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL) {
1030 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1031 fb1a36c0 2022-01-09 op return;
1032 fb1a36c0 2022-01-09 op }
1033 fb1a36c0 2022-01-09 op
1034 fb1a36c0 2022-01-09 op if (f->fd != -1) {
1035 fb1a36c0 2022-01-09 op np_error(hdr->tag, "fid already opened for I/O");
1036 fb1a36c0 2022-01-09 op return;
1037 fb1a36c0 2022-01-09 op }
1038 fb1a36c0 2022-01-09 op
1039 fb1a36c0 2022-01-09 op if (fid == newfid)
1040 fb1a36c0 2022-01-09 op nf = f;
1041 fb1a36c0 2022-01-09 op else if ((nf = fid_by_id(newfid)) != NULL) {
1042 fb1a36c0 2022-01-09 op np_error(hdr->tag, "newfid already in use");
1043 fb1a36c0 2022-01-09 op return;
1044 fb1a36c0 2022-01-09 op } else
1045 fb1a36c0 2022-01-09 op nf = NULL;
1046 fb1a36c0 2022-01-09 op
1047 fb1a36c0 2022-01-09 op /* special case: fid duplication */
1048 fb1a36c0 2022-01-09 op if (nwname == 0) {
1049 fb1a36c0 2022-01-09 op /*
1050 fb1a36c0 2022-01-09 op * TODO: should we forbid fids duplication when fid ==
1051 fb1a36c0 2022-01-09 op * newfid?
1052 fb1a36c0 2022-01-09 op */
1053 fb1a36c0 2022-01-09 op if (nf == NULL &&
1054 fb1a36c0 2022-01-09 op (nf = new_fid(f->dir, newfid, f->fpath, &f->qid)) == NULL)
1055 fb1a36c0 2022-01-09 op fatal("new_fid duplication");
1056 fb1a36c0 2022-01-09 op
1057 fb1a36c0 2022-01-09 op np_walk(hdr->tag, 0, NULL);
1058 fb1a36c0 2022-01-09 op return;
1059 fb1a36c0 2022-01-09 op }
1060 fb1a36c0 2022-01-09 op
1061 fb1a36c0 2022-01-09 op if (!(f->qid.type & QTDIR)) {
1062 fb1a36c0 2022-01-09 op np_error(hdr->tag, "fid doesn't represent a directory");
1063 fb1a36c0 2022-01-09 op return;
1064 fb1a36c0 2022-01-09 op }
1065 fb1a36c0 2022-01-09 op
1066 fb1a36c0 2022-01-09 op oldfd = f->dir->fd;
1067 fb1a36c0 2022-01-09 op
1068 fb1a36c0 2022-01-09 op for (nwqid = 0; nwqid < nwname; nwqid++) {
1069 fb1a36c0 2022-01-09 op switch (NPREADSTR("wname", wnam, sizeof(wnam), &data, &len)) {
1070 fb1a36c0 2022-01-09 op case READSTRERR:
1071 fb1a36c0 2022-01-09 op goto err;
1072 fb1a36c0 2022-01-09 op case READSTRTRUNC:
1073 fb1a36c0 2022-01-09 op np_error(hdr->tag, "wname too long");
1074 fb1a36c0 2022-01-09 op return;
1075 fb1a36c0 2022-01-09 op }
1076 fb1a36c0 2022-01-09 op
1077 fb1a36c0 2022-01-09 op if (*wnam == '\0' ||
1078 fb1a36c0 2022-01-09 op strchr(wnam, '/') != NULL ||
1079 fb1a36c0 2022-01-09 op !strcmp(wnam, ".")) {
1080 fb1a36c0 2022-01-09 op errno = EINVAL;
1081 fb1a36c0 2022-01-09 op goto cantopen;
1082 fb1a36c0 2022-01-09 op }
1083 fb1a36c0 2022-01-09 op
1084 fb1a36c0 2022-01-09 op if ((fd = openat(oldfd, wnam, O_RDONLY|O_DIRECTORY)) == -1 &&
1085 fb1a36c0 2022-01-09 op errno != ENOTDIR)
1086 fb1a36c0 2022-01-09 op goto cantopen;
1087 fb1a36c0 2022-01-09 op
1088 fb1a36c0 2022-01-09 op if ((fd == -1 && fstatat(oldfd, wnam, &sb, 0) == -1) ||
1089 fb1a36c0 2022-01-09 op (fd != -1 && fstat(fd, &sb) == -1))
1090 fb1a36c0 2022-01-09 op goto cantopen;
1091 fb1a36c0 2022-01-09 op
1092 fb1a36c0 2022-01-09 op qid_update_from_sb(&wqid[nwqid], &sb);
1093 fb1a36c0 2022-01-09 op
1094 fb1a36c0 2022-01-09 op /* reached a file but we still have other components */
1095 fb1a36c0 2022-01-09 op if (fd == -1 && nwqid+1 < nwname)
1096 fb1a36c0 2022-01-09 op goto cantopen;
1097 fb1a36c0 2022-01-09 op
1098 fb1a36c0 2022-01-09 op /* reached the end and found a file */
1099 fb1a36c0 2022-01-09 op if (fd == -1 && nwqid+1 == nwname)
1100 fb1a36c0 2022-01-09 op continue;
1101 fb1a36c0 2022-01-09 op
1102 fb1a36c0 2022-01-09 op if (oldfd != f->dir->fd)
1103 fb1a36c0 2022-01-09 op close(oldfd);
1104 fb1a36c0 2022-01-09 op oldfd = fd;
1105 fb1a36c0 2022-01-09 op }
1106 fb1a36c0 2022-01-09 op
1107 fb1a36c0 2022-01-09 op /*
1108 fb1a36c0 2022-01-09 op * If fd is -1 we've reached a file, otherwise we've just
1109 fb1a36c0 2022-01-09 op * reached another directory. We must pay attention to what
1110 fb1a36c0 2022-01-09 op * file descriptor we use to create the dir, because if we've
1111 fb1a36c0 2022-01-09 op * reached a file and oldfd is f->dir->fd then we *must* share
1112 fb1a36c0 2022-01-09 op * the same dir (it was a walk of one path from a directory to a
1113 fb1a36c0 2022-01-09 op * file, otherwise fun is bound to happen as soon as the client
1114 fb1a36c0 2022-01-09 op * closes the fid for the directory but keeps the one for the
1115 fb1a36c0 2022-01-09 op * file.
1116 fb1a36c0 2022-01-09 op */
1117 fb1a36c0 2022-01-09 op if (fd == -1 && oldfd == f->dir->fd)
1118 fb1a36c0 2022-01-09 op dir = f->dir;
1119 fb1a36c0 2022-01-09 op else if (fd == -1)
1120 fb1a36c0 2022-01-09 op dir = new_dir(oldfd);
1121 fb1a36c0 2022-01-09 op else
1122 fb1a36c0 2022-01-09 op dir = new_dir(fd);
1123 fb1a36c0 2022-01-09 op
1124 fb1a36c0 2022-01-09 op if (dir == NULL)
1125 fb1a36c0 2022-01-09 op fatal("new_dir");
1126 fb1a36c0 2022-01-09 op
1127 fb1a36c0 2022-01-09 op if (nf == NULL) {
1128 fb1a36c0 2022-01-09 op if ((nf = new_fid(dir, newfid, wnam, &wqid[nwqid-1])) == NULL)
1129 fb1a36c0 2022-01-09 op fatal("new fid");
1130 fb1a36c0 2022-01-09 op } else {
1131 fb1a36c0 2022-01-09 op /* update the dir */
1132 fb1a36c0 2022-01-09 op dir_decref(nf->dir);
1133 fb1a36c0 2022-01-09 op nf->dir = dir_incref(dir);
1134 fb1a36c0 2022-01-09 op }
1135 fb1a36c0 2022-01-09 op
1136 fb1a36c0 2022-01-09 op np_walk(hdr->tag, nwqid, wqid);
1137 fb1a36c0 2022-01-09 op return;
1138 fb1a36c0 2022-01-09 op
1139 fb1a36c0 2022-01-09 op cantopen:
1140 fb1a36c0 2022-01-09 op if (oldfd != f->dir->fd)
1141 fb1a36c0 2022-01-09 op close(oldfd);
1142 fb1a36c0 2022-01-09 op no = errno;
1143 fb1a36c0 2022-01-09 op if (nwqid == 0)
1144 fb1a36c0 2022-01-09 op np_error(hdr->tag, strerror(no));
1145 fb1a36c0 2022-01-09 op else
1146 fb1a36c0 2022-01-09 op np_walk(hdr->tag, nwqid, wqid);
1147 fb1a36c0 2022-01-09 op return;
1148 fb1a36c0 2022-01-09 op
1149 fb1a36c0 2022-01-09 op err:
1150 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1151 fb1a36c0 2022-01-09 op client_shutdown();
1152 fb1a36c0 2022-01-09 op }
1153 fb1a36c0 2022-01-09 op
1154 fb1a36c0 2022-01-09 op static inline int
1155 fb1a36c0 2022-01-09 op npmode_to_unix(uint8_t mode, int *flags)
1156 fb1a36c0 2022-01-09 op {
1157 fb1a36c0 2022-01-09 op switch (mode & 0x0F) {
1158 fb1a36c0 2022-01-09 op case KOREAD:
1159 fb1a36c0 2022-01-09 op *flags = O_RDONLY;
1160 fb1a36c0 2022-01-09 op break;
1161 fb1a36c0 2022-01-09 op case KOWRITE:
1162 fb1a36c0 2022-01-09 op *flags = O_WRONLY;
1163 fb1a36c0 2022-01-09 op break;
1164 fb1a36c0 2022-01-09 op case KORDWR:
1165 fb1a36c0 2022-01-09 op *flags = O_RDWR;
1166 fb1a36c0 2022-01-09 op break;
1167 fb1a36c0 2022-01-09 op case KOEXEC:
1168 fb1a36c0 2022-01-09 op log_warnx("tried to open something with KOEXEC");
1169 fb1a36c0 2022-01-09 op /* fallthrough */
1170 fb1a36c0 2022-01-09 op default:
1171 fb1a36c0 2022-01-09 op return -1;
1172 fb1a36c0 2022-01-09 op }
1173 fb1a36c0 2022-01-09 op
1174 fb1a36c0 2022-01-09 op if (mode & KOTRUNC)
1175 fb1a36c0 2022-01-09 op *flags |= O_TRUNC;
1176 fb1a36c0 2022-01-09 op if (mode & KORCLOSE)
1177 fb1a36c0 2022-01-09 op *flags |= O_CLOEXEC;
1178 fb1a36c0 2022-01-09 op
1179 fb1a36c0 2022-01-09 op return 0;
1180 fb1a36c0 2022-01-09 op }
1181 fb1a36c0 2022-01-09 op
1182 fb1a36c0 2022-01-09 op static void
1183 fb1a36c0 2022-01-09 op topen(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1184 fb1a36c0 2022-01-09 op {
1185 fb1a36c0 2022-01-09 op struct stat sb;
1186 fb1a36c0 2022-01-09 op struct qid qid;
1187 fb1a36c0 2022-01-09 op struct fid *f;
1188 fb1a36c0 2022-01-09 op uint32_t fid;
1189 fb1a36c0 2022-01-09 op uint8_t mode;
1190 fb1a36c0 2022-01-09 op const char *path;
1191 fb1a36c0 2022-01-09 op
1192 fb1a36c0 2022-01-09 op /* fid[4] mode[1] */
1193 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
1194 fb1a36c0 2022-01-09 op !NPREAD8("mode", &mode, &data, &len)) {
1195 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1196 fb1a36c0 2022-01-09 op client_shutdown();
1197 fb1a36c0 2022-01-09 op return;
1198 fb1a36c0 2022-01-09 op }
1199 fb1a36c0 2022-01-09 op
1200 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL || f->fd != -1) {
1201 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1202 fb1a36c0 2022-01-09 op return;
1203 fb1a36c0 2022-01-09 op }
1204 fb1a36c0 2022-01-09 op
1205 fb1a36c0 2022-01-09 op if (npmode_to_unix(mode, &f->iomode) == -1) {
1206 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid mode");
1207 fb1a36c0 2022-01-09 op return;
1208 fb1a36c0 2022-01-09 op }
1209 fb1a36c0 2022-01-09 op
1210 fb1a36c0 2022-01-09 op path = f->fpath;
1211 fb1a36c0 2022-01-09 op if (f->qid.type & QTDIR)
1212 fb1a36c0 2022-01-09 op path = ".";
1213 fb1a36c0 2022-01-09 op
1214 fb1a36c0 2022-01-09 op if ((f->fd = openat(f->dir->fd, path, f->iomode)) == -1) {
1215 fb1a36c0 2022-01-09 op np_error(hdr->tag, strerror(errno));
1216 fb1a36c0 2022-01-09 op return;
1217 fb1a36c0 2022-01-09 op }
1218 fb1a36c0 2022-01-09 op
1219 fb1a36c0 2022-01-09 op if (fstat(f->fd, &sb) == -1)
1220 fb1a36c0 2022-01-09 op fatal("fstat");
1221 fb1a36c0 2022-01-09 op
1222 fb1a36c0 2022-01-09 op if (S_ISDIR(sb.st_mode)) {
1223 fb1a36c0 2022-01-09 op if ((f->d = fdopendir(f->fd)) == NULL) {
1224 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1225 fb1a36c0 2022-01-09 op close(f->fd);
1226 fb1a36c0 2022-01-09 op f->fd = -1;
1227 fb1a36c0 2022-01-09 op return;
1228 fb1a36c0 2022-01-09 op }
1229 fb1a36c0 2022-01-09 op
1230 fb1a36c0 2022-01-09 op if ((f->evb = evbuffer_new()) == NULL) {
1231 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1232 fb1a36c0 2022-01-09 op closedir(f->d);
1233 fb1a36c0 2022-01-09 op f->d = NULL;
1234 fb1a36c0 2022-01-09 op f->fd = -1;
1235 fb1a36c0 2022-01-09 op }
1236 fb1a36c0 2022-01-09 op }
1237 fb1a36c0 2022-01-09 op
1238 fb1a36c0 2022-01-09 op f->offset = 0;
1239 fb1a36c0 2022-01-09 op
1240 fb1a36c0 2022-01-09 op qid_update_from_sb(&qid, &sb);
1241 fb1a36c0 2022-01-09 op np_open(hdr->tag, &qid, sb.st_blksize);
1242 fb1a36c0 2022-01-09 op }
1243 fb1a36c0 2022-01-09 op
1244 fb1a36c0 2022-01-09 op static void
1245 fb1a36c0 2022-01-09 op tcreate(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1246 fb1a36c0 2022-01-09 op {
1247 fb1a36c0 2022-01-09 op struct stat sb;
1248 fb1a36c0 2022-01-09 op struct qid qid;
1249 fb1a36c0 2022-01-09 op struct fid *f;
1250 fb1a36c0 2022-01-09 op uint32_t fid, perm;
1251 fb1a36c0 2022-01-09 op uint8_t mode;
1252 fb1a36c0 2022-01-09 op char name[PATH_MAX];
1253 fb1a36c0 2022-01-09 op
1254 fb1a36c0 2022-01-09 op /* fid[4] name[s] perm[4] mode[1] */
1255 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len))
1256 fb1a36c0 2022-01-09 op goto err;
1257 fb1a36c0 2022-01-09 op switch (NPREADSTR("name", name, sizeof(name), &data, &len)) {
1258 fb1a36c0 2022-01-09 op case READSTRERR:
1259 fb1a36c0 2022-01-09 op goto err;
1260 fb1a36c0 2022-01-09 op case READSTRTRUNC:
1261 fb1a36c0 2022-01-09 op np_error(hdr->tag, "name too long");
1262 fb1a36c0 2022-01-09 op return;
1263 fb1a36c0 2022-01-09 op }
1264 fb1a36c0 2022-01-09 op if (!NPREAD32("perm", &perm, &data, &len) ||
1265 fb1a36c0 2022-01-09 op !NPREAD8("mode", &mode, &data, &len))
1266 fb1a36c0 2022-01-09 op goto err;
1267 fb1a36c0 2022-01-09 op
1268 fb1a36c0 2022-01-09 op if (!strcmp(name, ".") || !strcmp(name, "..") ||
1269 fb1a36c0 2022-01-09 op strchr(name, '/') != NULL) {
1270 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid name");
1271 fb1a36c0 2022-01-09 op return;
1272 fb1a36c0 2022-01-09 op }
1273 fb1a36c0 2022-01-09 op
1274 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL || f->fd != -1) {
1275 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1276 fb1a36c0 2022-01-09 op return;
1277 fb1a36c0 2022-01-09 op }
1278 fb1a36c0 2022-01-09 op
1279 fb1a36c0 2022-01-09 op if (!(f->qid.type & QTDIR)) {
1280 fb1a36c0 2022-01-09 op np_error(hdr->tag, "fid doesn't identify a directory");
1281 fb1a36c0 2022-01-09 op return;
1282 fb1a36c0 2022-01-09 op }
1283 fb1a36c0 2022-01-09 op
1284 fb1a36c0 2022-01-09 op if (npmode_to_unix(mode, &f->iomode) == -1) {
1285 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid mode");
1286 fb1a36c0 2022-01-09 op return;
1287 fb1a36c0 2022-01-09 op }
1288 fb1a36c0 2022-01-09 op
1289 fb1a36c0 2022-01-09 op if (f->iomode & O_RDONLY) {
1290 fb1a36c0 2022-01-09 op np_error(hdr->tag, "can't create a read-only file");
1291 fb1a36c0 2022-01-09 op return;
1292 fb1a36c0 2022-01-09 op }
1293 fb1a36c0 2022-01-09 op
1294 fb1a36c0 2022-01-09 op /* TODO: parse the mode */
1295 fb1a36c0 2022-01-09 op
1296 fb1a36c0 2022-01-09 op if (perm & 0x80000000) {
1297 fb1a36c0 2022-01-09 op /* create a directory */
1298 fb1a36c0 2022-01-09 op f->fd = mkdirat(f->dir->fd, name, 0755);
1299 fb1a36c0 2022-01-09 op } else {
1300 fb1a36c0 2022-01-09 op /* create a file */
1301 fb1a36c0 2022-01-09 op f->fd = openat(f->dir->fd, name, f->iomode | O_CREAT | O_TRUNC,
1302 fb1a36c0 2022-01-09 op 0644);
1303 fb1a36c0 2022-01-09 op }
1304 fb1a36c0 2022-01-09 op
1305 fb1a36c0 2022-01-09 op if (f->fd == -1) {
1306 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1307 fb1a36c0 2022-01-09 op return;
1308 fb1a36c0 2022-01-09 op }
1309 fb1a36c0 2022-01-09 op
1310 fb1a36c0 2022-01-09 op if (fstat(f->fd, &sb) == -1)
1311 fb1a36c0 2022-01-09 op fatal("fstat");
1312 fb1a36c0 2022-01-09 op
1313 fb1a36c0 2022-01-09 op if (S_ISDIR(sb.st_mode)) {
1314 fb1a36c0 2022-01-09 op if ((f->d = fdopendir(f->fd)) == NULL) {
1315 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1316 fb1a36c0 2022-01-09 op close(f->fd);
1317 fb1a36c0 2022-01-09 op f->fd = -1;
1318 fb1a36c0 2022-01-09 op return;
1319 fb1a36c0 2022-01-09 op }
1320 fb1a36c0 2022-01-09 op
1321 fb1a36c0 2022-01-09 op if ((f->evb = evbuffer_new()) == NULL) {
1322 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1323 fb1a36c0 2022-01-09 op closedir(f->d);
1324 fb1a36c0 2022-01-09 op f->d = NULL;
1325 fb1a36c0 2022-01-09 op f->fd = -1;
1326 fb1a36c0 2022-01-09 op }
1327 fb1a36c0 2022-01-09 op }
1328 fb1a36c0 2022-01-09 op
1329 fb1a36c0 2022-01-09 op f->offset = 0;
1330 fb1a36c0 2022-01-09 op
1331 fb1a36c0 2022-01-09 op qid_update_from_sb(&qid, &sb);
1332 fb1a36c0 2022-01-09 op np_create(hdr->tag, &qid, sb.st_blksize);
1333 fb1a36c0 2022-01-09 op
1334 fb1a36c0 2022-01-09 op return;
1335 fb1a36c0 2022-01-09 op
1336 fb1a36c0 2022-01-09 op err:
1337 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1338 fb1a36c0 2022-01-09 op client_shutdown();
1339 fb1a36c0 2022-01-09 op }
1340 fb1a36c0 2022-01-09 op
1341 fb1a36c0 2022-01-09 op static inline void
1342 fb1a36c0 2022-01-09 op serialize_stat(const char *fname, struct stat *sb, struct evbuffer *evb)
1343 fb1a36c0 2022-01-09 op {
1344 fb1a36c0 2022-01-09 op struct qid qid;
1345 fb1a36c0 2022-01-09 op const char *uid, *gid, *muid;
1346 fb1a36c0 2022-01-09 op size_t tot;
1347 fb1a36c0 2022-01-09 op uint16_t namlen, uidlen, gidlen, ulen;
1348 fb1a36c0 2022-01-09 op
1349 fb1a36c0 2022-01-09 op qid_update_from_sb(&qid, sb);
1350 fb1a36c0 2022-01-09 op
1351 fb1a36c0 2022-01-09 op /* TODO: fill these fields */
1352 fb1a36c0 2022-01-09 op uid = "";
1353 fb1a36c0 2022-01-09 op gid = "";
1354 fb1a36c0 2022-01-09 op muid = "";
1355 fb1a36c0 2022-01-09 op
1356 fb1a36c0 2022-01-09 op namlen = strlen(fname);
1357 fb1a36c0 2022-01-09 op uidlen = strlen(uid);
1358 fb1a36c0 2022-01-09 op gidlen = strlen(gid);
1359 fb1a36c0 2022-01-09 op ulen = strlen(muid);
1360 fb1a36c0 2022-01-09 op
1361 fb1a36c0 2022-01-09 op tot = NPSTATSIZ(namlen, uidlen, gidlen, ulen);
1362 fb1a36c0 2022-01-09 op if (tot > UINT32_MAX) {
1363 fb1a36c0 2022-01-09 op log_warnx("stat info for dir entry %s would overflow",
1364 fb1a36c0 2022-01-09 op fname);
1365 fb1a36c0 2022-01-09 op return;
1366 fb1a36c0 2022-01-09 op }
1367 fb1a36c0 2022-01-09 op
1368 fb1a36c0 2022-01-09 op np_write16(evb, tot); /* size[2] */
1369 fb1a36c0 2022-01-09 op np_write16(evb, sb->st_rdev); /* type[2] */
1370 fb1a36c0 2022-01-09 op np_write32(evb, sb->st_dev); /* dev[4] */
1371 fb1a36c0 2022-01-09 op np_qid(evb, &qid); /* qid[13] */
1372 fb1a36c0 2022-01-09 op
1373 fb1a36c0 2022-01-09 op /* XXX: translate? */
1374 fb1a36c0 2022-01-09 op np_write32(evb, sb->st_mode); /* mode[4] */
1375 fb1a36c0 2022-01-09 op
1376 fb1a36c0 2022-01-09 op np_write32(evb, sb->st_atim.tv_sec); /* atime[4] */
1377 fb1a36c0 2022-01-09 op np_write32(evb, sb->st_mtim.tv_sec); /* mtime[4] */
1378 fb1a36c0 2022-01-09 op np_write64(evb, sb->st_size); /* length[8] */
1379 fb1a36c0 2022-01-09 op np_string(evb, namlen, fname); /* name[s] */
1380 fb1a36c0 2022-01-09 op np_string(evb, uidlen, uid); /* uid[s] */
1381 fb1a36c0 2022-01-09 op np_string(evb, gidlen, gid); /* gid[s] */
1382 fb1a36c0 2022-01-09 op np_string(evb, ulen, muid); /* muid[s] */
1383 fb1a36c0 2022-01-09 op }
1384 fb1a36c0 2022-01-09 op
1385 fb1a36c0 2022-01-09 op static void
1386 fb1a36c0 2022-01-09 op tread(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1387 fb1a36c0 2022-01-09 op {
1388 fb1a36c0 2022-01-09 op struct fid *f;
1389 fb1a36c0 2022-01-09 op ssize_t r;
1390 fb1a36c0 2022-01-09 op size_t howmuch;
1391 fb1a36c0 2022-01-09 op uint64_t off;
1392 fb1a36c0 2022-01-09 op uint32_t fid, count;
1393 fb1a36c0 2022-01-09 op char buf[2048];
1394 fb1a36c0 2022-01-09 op
1395 fb1a36c0 2022-01-09 op /* fid[4] offset[8] count[4] */
1396 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
1397 fb1a36c0 2022-01-09 op !NPREAD64("offset", &off, &data, &len) ||
1398 fb1a36c0 2022-01-09 op !NPREAD32("count", &count, &data, &len)) {
1399 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1400 fb1a36c0 2022-01-09 op client_shutdown();
1401 fb1a36c0 2022-01-09 op return;
1402 fb1a36c0 2022-01-09 op }
1403 fb1a36c0 2022-01-09 op
1404 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL || f->fd == -1) {
1405 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1406 fb1a36c0 2022-01-09 op return;
1407 fb1a36c0 2022-01-09 op }
1408 fb1a36c0 2022-01-09 op
1409 fb1a36c0 2022-01-09 op if (TYPE_OVERFLOW(off_t, off)) {
1410 fb1a36c0 2022-01-09 op log_warnx("unexpected off_t size");
1411 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid offset");
1412 fb1a36c0 2022-01-09 op return;
1413 fb1a36c0 2022-01-09 op }
1414 fb1a36c0 2022-01-09 op
1415 fb1a36c0 2022-01-09 op if (f->d == NULL) {
1416 fb1a36c0 2022-01-09 op /* read a file */
1417 fb1a36c0 2022-01-09 op howmuch = MIN(sizeof(buf), count);
1418 fb1a36c0 2022-01-09 op r = pread(f->fd, buf, howmuch, (off_t)off);
1419 fb1a36c0 2022-01-09 op if (r == -1)
1420 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1421 fb1a36c0 2022-01-09 op else
1422 fb1a36c0 2022-01-09 op np_read(hdr->tag, r, buf);
1423 fb1a36c0 2022-01-09 op } else {
1424 fb1a36c0 2022-01-09 op if (off == 0 && f->offset != 0) {
1425 fb1a36c0 2022-01-09 op rewinddir(f->d);
1426 fb1a36c0 2022-01-09 op f->offset = 0;
1427 fb1a36c0 2022-01-09 op evbuffer_drain(f->evb, EVBUFFER_LENGTH(f->evb));
1428 fb1a36c0 2022-01-09 op }
1429 fb1a36c0 2022-01-09 op
1430 fb1a36c0 2022-01-09 op if (off != f->offset) {
1431 fb1a36c0 2022-01-09 op np_error(hdr->tag, "can't seek in directories");
1432 fb1a36c0 2022-01-09 op return;
1433 fb1a36c0 2022-01-09 op }
1434 fb1a36c0 2022-01-09 op
1435 fb1a36c0 2022-01-09 op while (EVBUFFER_LENGTH(f->evb) < count) {
1436 fb1a36c0 2022-01-09 op struct dirent *d;
1437 fb1a36c0 2022-01-09 op struct stat sb;
1438 fb1a36c0 2022-01-09 op
1439 fb1a36c0 2022-01-09 op if ((d = readdir(f->d)) == NULL)
1440 fb1a36c0 2022-01-09 op break;
1441 fb1a36c0 2022-01-09 op if (fstatat(f->fd, d->d_name, &sb, 0) == -1) {
1442 fb1a36c0 2022-01-09 op warn("fstatat");
1443 fb1a36c0 2022-01-09 op continue;
1444 fb1a36c0 2022-01-09 op }
1445 fb1a36c0 2022-01-09 op serialize_stat(d->d_name, &sb, f->evb);
1446 fb1a36c0 2022-01-09 op }
1447 fb1a36c0 2022-01-09 op
1448 fb1a36c0 2022-01-09 op count = MIN(count, EVBUFFER_LENGTH(f->evb));
1449 fb1a36c0 2022-01-09 op np_read(hdr->tag, count, EVBUFFER_DATA(f->evb));
1450 fb1a36c0 2022-01-09 op evbuffer_drain(f->evb, count);
1451 fb1a36c0 2022-01-09 op
1452 fb1a36c0 2022-01-09 op f->offset += count;
1453 fb1a36c0 2022-01-09 op }
1454 fb1a36c0 2022-01-09 op }
1455 fb1a36c0 2022-01-09 op
1456 fb1a36c0 2022-01-09 op static void
1457 fb1a36c0 2022-01-09 op twrite(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1458 fb1a36c0 2022-01-09 op {
1459 fb1a36c0 2022-01-09 op struct fid *f;
1460 fb1a36c0 2022-01-09 op ssize_t r;
1461 fb1a36c0 2022-01-09 op uint64_t off;
1462 fb1a36c0 2022-01-09 op uint32_t fid, count;
1463 fb1a36c0 2022-01-09 op
1464 fb1a36c0 2022-01-09 op /* fid[4] offset[8] count[4] data[count] */
1465 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
1466 fb1a36c0 2022-01-09 op !NPREAD64("off", &off, &data, &len) ||
1467 fb1a36c0 2022-01-09 op !NPREAD32("count", &count, &data, &len) ||
1468 fb1a36c0 2022-01-09 op len != count) {
1469 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1470 fb1a36c0 2022-01-09 op client_shutdown();
1471 fb1a36c0 2022-01-09 op return;
1472 fb1a36c0 2022-01-09 op }
1473 fb1a36c0 2022-01-09 op
1474 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL || f->fd == -1) {
1475 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1476 fb1a36c0 2022-01-09 op return;
1477 fb1a36c0 2022-01-09 op }
1478 fb1a36c0 2022-01-09 op
1479 fb1a36c0 2022-01-09 op if (!(f->iomode & O_WRONLY) &&
1480 fb1a36c0 2022-01-09 op !(f->iomode & O_RDWR)) {
1481 fb1a36c0 2022-01-09 op np_error(hdr->tag, "fid not opened for writing");
1482 fb1a36c0 2022-01-09 op return;
1483 fb1a36c0 2022-01-09 op }
1484 fb1a36c0 2022-01-09 op
1485 fb1a36c0 2022-01-09 op if (TYPE_OVERFLOW(off_t, off)) {
1486 fb1a36c0 2022-01-09 op log_warnx("unexpected off_t size");
1487 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid offset");
1488 fb1a36c0 2022-01-09 op return;
1489 fb1a36c0 2022-01-09 op }
1490 fb1a36c0 2022-01-09 op
1491 fb1a36c0 2022-01-09 op if ((r = pwrite(f->fd, data, len, off)) == -1)
1492 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1493 fb1a36c0 2022-01-09 op else
1494 fb1a36c0 2022-01-09 op np_write(hdr->tag, r);
1495 fb1a36c0 2022-01-09 op }
1496 fb1a36c0 2022-01-09 op
1497 fb1a36c0 2022-01-09 op static void
1498 fb1a36c0 2022-01-09 op tstat(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1499 fb1a36c0 2022-01-09 op {
1500 fb1a36c0 2022-01-09 op struct evbuffer *evb;
1501 fb1a36c0 2022-01-09 op struct stat sb;
1502 fb1a36c0 2022-01-09 op struct fid *f;
1503 fb1a36c0 2022-01-09 op int r;
1504 fb1a36c0 2022-01-09 op uint32_t fid;
1505 fb1a36c0 2022-01-09 op
1506 fb1a36c0 2022-01-09 op /* fid[4] */
1507 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len)) {
1508 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1509 fb1a36c0 2022-01-09 op client_shutdown();
1510 fb1a36c0 2022-01-09 op return;
1511 fb1a36c0 2022-01-09 op }
1512 fb1a36c0 2022-01-09 op
1513 fb1a36c0 2022-01-09 op /*
1514 fb1a36c0 2022-01-09 op * plan9' stat(9P) is not clear on whether the stat is allowed
1515 fb1a36c0 2022-01-09 op * on opened fids or not. We're allowing stat regardless of the
1516 fb1a36c0 2022-01-09 op * status of the fid.
1517 fb1a36c0 2022-01-09 op */
1518 fb1a36c0 2022-01-09 op
1519 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL) {
1520 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1521 fb1a36c0 2022-01-09 op return;
1522 fb1a36c0 2022-01-09 op }
1523 fb1a36c0 2022-01-09 op
1524 fb1a36c0 2022-01-09 op if ((evb = evbuffer_new()) == NULL)
1525 fb1a36c0 2022-01-09 op fatal("evbuffer_new");
1526 fb1a36c0 2022-01-09 op
1527 fb1a36c0 2022-01-09 op if (f->fd != -1)
1528 fb1a36c0 2022-01-09 op r = fstat(f->fd, &sb);
1529 fb1a36c0 2022-01-09 op else if (f->qid.type & QTDIR)
1530 fb1a36c0 2022-01-09 op r = fstat(f->dir->fd, &sb);
1531 fb1a36c0 2022-01-09 op else
1532 fb1a36c0 2022-01-09 op r = fstatat(f->dir->fd, f->fpath, &sb, 0);
1533 fb1a36c0 2022-01-09 op
1534 fb1a36c0 2022-01-09 op if (r == -1) {
1535 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1536 fb1a36c0 2022-01-09 op evbuffer_free(evb);
1537 fb1a36c0 2022-01-09 op return;
1538 fb1a36c0 2022-01-09 op }
1539 fb1a36c0 2022-01-09 op
1540 fb1a36c0 2022-01-09 op serialize_stat(f->fpath, &sb, evb);
1541 fb1a36c0 2022-01-09 op np_stat(hdr->tag, EVBUFFER_LENGTH(evb), EVBUFFER_DATA(evb));
1542 fb1a36c0 2022-01-09 op evbuffer_free(evb);
1543 fb1a36c0 2022-01-09 op }
1544 fb1a36c0 2022-01-09 op
1545 fb1a36c0 2022-01-09 op static void
1546 fb1a36c0 2022-01-09 op tremove(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1547 fb1a36c0 2022-01-09 op {
1548 fb1a36c0 2022-01-09 op struct fid *f;
1549 fb1a36c0 2022-01-09 op uint32_t fid;
1550 fb1a36c0 2022-01-09 op int r;
1551 fb1a36c0 2022-01-09 op char dirpath[PATH_MAX + 3];
1552 fb1a36c0 2022-01-09 op
1553 fb1a36c0 2022-01-09 op /* fid[4] */
1554 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len)) {
1555 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1556 fb1a36c0 2022-01-09 op client_shutdown();
1557 fb1a36c0 2022-01-09 op return;
1558 fb1a36c0 2022-01-09 op }
1559 fb1a36c0 2022-01-09 op
1560 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL) {
1561 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1562 fb1a36c0 2022-01-09 op return;
1563 fb1a36c0 2022-01-09 op }
1564 fb1a36c0 2022-01-09 op
1565 fb1a36c0 2022-01-09 op if (f->qid.type & QTDIR) { /* directory */
1566 fb1a36c0 2022-01-09 op strlcpy(dirpath, "../", sizeof(dirpath));
1567 fb1a36c0 2022-01-09 op strlcat(dirpath, f->fpath, sizeof(dirpath));
1568 fb1a36c0 2022-01-09 op r = unlinkat(f->dir->fd, dirpath, AT_REMOVEDIR);
1569 fb1a36c0 2022-01-09 op } else /* file */
1570 fb1a36c0 2022-01-09 op r = unlinkat(f->dir->fd, f->fpath, 0);
1571 fb1a36c0 2022-01-09 op
1572 fb1a36c0 2022-01-09 op if (r == -1)
1573 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1574 fb1a36c0 2022-01-09 op else
1575 fb1a36c0 2022-01-09 op np_remove(hdr->tag);
1576 fb1a36c0 2022-01-09 op
1577 fb1a36c0 2022-01-09 op free_fid(f);
1578 fb1a36c0 2022-01-09 op }
1579 fb1a36c0 2022-01-09 op
1580 fb1a36c0 2022-01-09 op static void
1581 fb1a36c0 2022-01-09 op handle_message(struct imsg *imsg, size_t len)
1582 fb1a36c0 2022-01-09 op {
1583 fb1a36c0 2022-01-09 op struct msg {
1584 fb1a36c0 2022-01-09 op uint8_t type;
1585 fb1a36c0 2022-01-09 op void (*fn)(struct np_msg_header *, const uint8_t *, size_t);
1586 fb1a36c0 2022-01-09 op } msgs[] = {
1587 fb1a36c0 2022-01-09 op {Tversion, tversion},
1588 fb1a36c0 2022-01-09 op {Tattach, tattach},
1589 fb1a36c0 2022-01-09 op {Tclunk, tclunk},
1590 fb1a36c0 2022-01-09 op {Tflush, tflush},
1591 fb1a36c0 2022-01-09 op {Twalk, twalk},
1592 fb1a36c0 2022-01-09 op {Topen, topen},
1593 fb1a36c0 2022-01-09 op {Tcreate, tcreate},
1594 fb1a36c0 2022-01-09 op {Tread, tread},
1595 fb1a36c0 2022-01-09 op {Twrite, twrite},
1596 fb1a36c0 2022-01-09 op {Tstat, tstat},
1597 fb1a36c0 2022-01-09 op {Tremove, tremove},
1598 fb1a36c0 2022-01-09 op };
1599 fb1a36c0 2022-01-09 op struct np_msg_header hdr;
1600 fb1a36c0 2022-01-09 op size_t i;
1601 fb1a36c0 2022-01-09 op uint8_t *data;
1602 fb1a36c0 2022-01-09 op
1603 fb1a36c0 2022-01-09 op #if DEBUG_PACKETS
1604 fb1a36c0 2022-01-09 op hexdump("incoming packet", imsg->data, len);
1605 fb1a36c0 2022-01-09 op #endif
1606 fb1a36c0 2022-01-09 op
1607 fb1a36c0 2022-01-09 op parse_message(imsg->data, len, &hdr, &data);
1608 fb1a36c0 2022-01-09 op len -= HEADERSIZE;
1609 fb1a36c0 2022-01-09 op
1610 fb1a36c0 2022-01-09 op log_debug("got request: len=%d type=%d[%s] tag=%d",
1611 fb1a36c0 2022-01-09 op hdr.len, hdr.type, pp_msg_type(hdr.type), hdr.tag);
1612 fb1a36c0 2022-01-09 op
1613 fb1a36c0 2022-01-09 op if (!handshaked && hdr.type != Tversion) {
1614 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1615 fb1a36c0 2022-01-09 op client_shutdown();
1616 fb1a36c0 2022-01-09 op return;
1617 fb1a36c0 2022-01-09 op }
1618 fb1a36c0 2022-01-09 op
1619 fb1a36c0 2022-01-09 op for (i = 0; i < sizeof(msgs)/sizeof(msgs[0]); ++i) {
1620 fb1a36c0 2022-01-09 op if (msgs[i].type != hdr.type)
1621 fb1a36c0 2022-01-09 op continue;
1622 fb1a36c0 2022-01-09 op
1623 fb1a36c0 2022-01-09 op msgs[i].fn(&hdr, data, len);
1624 fb1a36c0 2022-01-09 op return;
1625 fb1a36c0 2022-01-09 op }
1626 fb1a36c0 2022-01-09 op
1627 fb1a36c0 2022-01-09 op np_error(hdr.tag, "Not supported.");
1628 fb1a36c0 2022-01-09 op }