Blame


1 fb1a36c0 2022-01-09 op /*
2 1e84c7b7 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/stat.h>
20 fb1a36c0 2022-01-09 op #include <sys/types.h>
21 fb1a36c0 2022-01-09 op #include <sys/uio.h>
22 fb1a36c0 2022-01-09 op
23 fb1a36c0 2022-01-09 op #include <dirent.h>
24 fb1a36c0 2022-01-09 op #include <endian.h>
25 fb1a36c0 2022-01-09 op #include <errno.h>
26 fb1a36c0 2022-01-09 op #include <fcntl.h>
27 755d75bf 2022-01-17 op #include <inttypes.h>
28 755d75bf 2022-01-17 op #include <libgen.h>
29 fb1a36c0 2022-01-09 op #include <pwd.h>
30 fb1a36c0 2022-01-09 op #include <signal.h>
31 fb1a36c0 2022-01-09 op #include <stdint.h>
32 755d75bf 2022-01-17 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 <unistd.h>
37 fb1a36c0 2022-01-09 op
38 fb1a36c0 2022-01-09 op #include "client.h"
39 fb1a36c0 2022-01-09 op #include "kami.h"
40 fb1a36c0 2022-01-09 op #include "kamid.h"
41 fb1a36c0 2022-01-09 op #include "log.h"
42 fb1a36c0 2022-01-09 op #include "sandbox.h"
43 fb1a36c0 2022-01-09 op #include "utils.h"
44 fb1a36c0 2022-01-09 op
45 fb1a36c0 2022-01-09 op /*
46 6be59daa 2022-05-22 op * The minimum value allowed for the msize.
47 6be59daa 2022-05-22 op */
48 6be59daa 2022-05-22 op #define MIN_MSIZE 256
49 fb1a36c0 2022-01-09 op
50 fb1a36c0 2022-01-09 op #define DEBUG_PACKETS 0
51 fb1a36c0 2022-01-09 op
52 fb1a36c0 2022-01-09 op /* straight outta /src/usr.bin/ssh/scp.c */
53 fb1a36c0 2022-01-09 op #define TYPE_OVERFLOW(type, val) \
54 fb1a36c0 2022-01-09 op ((sizeof(type) == 4 && (val) > INT32_MAX) || \
55 fb1a36c0 2022-01-09 op (sizeof(type) == 8 && (val) > INT64_MAX) || \
56 fb1a36c0 2022-01-09 op (sizeof(type) != 4 && sizeof(type) != 8))
57 fb1a36c0 2022-01-09 op
58 fb1a36c0 2022-01-09 op STAILQ_HEAD(dirhead, dir) dirs;
59 fb1a36c0 2022-01-09 op struct dir {
60 fb1a36c0 2022-01-09 op int refcount;
61 fb1a36c0 2022-01-09 op int fd;
62 fb1a36c0 2022-01-09 op STAILQ_ENTRY(dir) entries;
63 fb1a36c0 2022-01-09 op };
64 fb1a36c0 2022-01-09 op
65 fb1a36c0 2022-01-09 op STAILQ_HEAD(fidhead, fid) fids;
66 fb1a36c0 2022-01-09 op struct fid {
67 fb1a36c0 2022-01-09 op uint32_t fid;
68 fb1a36c0 2022-01-09 op
69 46e4e74b 2022-01-17 op char fname[NAME_MAX];
70 fb1a36c0 2022-01-09 op
71 fb1a36c0 2022-01-09 op /*
72 fb1a36c0 2022-01-09 op * the flags passed to open(2). O_CLOEXEC means ORCLOSE, that
73 fb1a36c0 2022-01-09 op * is to unlink the file upon Tclunk.
74 fb1a36c0 2022-01-09 op */
75 fb1a36c0 2022-01-09 op int iomode;
76 fb1a36c0 2022-01-09 op
77 fb1a36c0 2022-01-09 op /*
78 fb1a36c0 2022-01-09 op * if fd is not -1 this fid was opened, fd represents its
79 fb1a36c0 2022-01-09 op * file descriptor and iomode the flags passed to open(2).
80 fb1a36c0 2022-01-09 op */
81 fb1a36c0 2022-01-09 op int fd;
82 fb1a36c0 2022-01-09 op DIR *d;
83 fb1a36c0 2022-01-09 op struct evbuffer *evb;
84 fb1a36c0 2022-01-09 op
85 fb1a36c0 2022-01-09 op /*
86 fb1a36c0 2022-01-09 op * expected offset for Tread against a directory.
87 fb1a36c0 2022-01-09 op */
88 fb1a36c0 2022-01-09 op uint64_t offset;
89 fb1a36c0 2022-01-09 op
90 fb1a36c0 2022-01-09 op struct qid qid;
91 fb1a36c0 2022-01-09 op struct dir *dir;
92 fb1a36c0 2022-01-09 op STAILQ_ENTRY(fid) entries;
93 fb1a36c0 2022-01-09 op };
94 fb1a36c0 2022-01-09 op
95 fb1a36c0 2022-01-09 op static struct imsgev *iev_listener;
96 fb1a36c0 2022-01-09 op static struct evbuffer *evb;
97 fb1a36c0 2022-01-09 op static uint32_t peerid;
98 fb1a36c0 2022-01-09 op
99 fb1a36c0 2022-01-09 op static int handshaked;
100 fb1a36c0 2022-01-09 op uint32_t msize;
101 fb1a36c0 2022-01-09 op
102 fb1a36c0 2022-01-09 op static __dead void client_shutdown(void);
103 fb1a36c0 2022-01-09 op static void client_sig_handler(int, short, void *);
104 fb1a36c0 2022-01-09 op static void client_dispatch_listener(int, short, void *);
105 fb1a36c0 2022-01-09 op static void client_privdrop(const char *, const char *);
106 fb1a36c0 2022-01-09 op
107 5c69b561 2022-01-29 op static int client_send_listenerp(int, uint32_t, const void *, uint16_t);
108 fb1a36c0 2022-01-09 op static int client_send_listener(int, const void *, uint16_t);
109 fb1a36c0 2022-01-09 op
110 fb1a36c0 2022-01-09 op static void qid_update_from_sb(struct qid *, struct stat *);
111 fb1a36c0 2022-01-09 op
112 fb1a36c0 2022-01-09 op static struct dir *new_dir(int);
113 fb1a36c0 2022-01-09 op static struct dir *dir_incref(struct dir *);
114 fb1a36c0 2022-01-09 op static void dir_decref(struct dir *);
115 fb1a36c0 2022-01-09 op
116 fb1a36c0 2022-01-09 op static struct fid *new_fid(struct dir *, uint32_t, const char *, struct qid *);
117 fb1a36c0 2022-01-09 op static struct fid *fid_by_id(uint32_t);
118 fb1a36c0 2022-01-09 op static void free_fid(struct fid *);
119 fb1a36c0 2022-01-09 op
120 fb1a36c0 2022-01-09 op static void parse_message(const uint8_t *, size_t,
121 fb1a36c0 2022-01-09 op struct np_msg_header *, uint8_t **);
122 fb1a36c0 2022-01-09 op
123 fb1a36c0 2022-01-09 op static void np_write16(struct evbuffer *, uint16_t);
124 fb1a36c0 2022-01-09 op static void np_write32(struct evbuffer *, uint32_t);
125 fb1a36c0 2022-01-09 op static void np_write64(struct evbuffer *, uint64_t);
126 fb1a36c0 2022-01-09 op static void np_header(uint32_t, uint8_t, uint16_t);
127 fb1a36c0 2022-01-09 op static void np_string(struct evbuffer *, uint16_t, const char *);
128 fb1a36c0 2022-01-09 op static void np_qid(struct evbuffer *, struct qid *);
129 fb1a36c0 2022-01-09 op static void do_send(void);
130 fb1a36c0 2022-01-09 op
131 fb1a36c0 2022-01-09 op static void np_version(uint16_t, uint32_t, const char *);
132 fb1a36c0 2022-01-09 op static void np_attach(uint16_t, struct qid *);
133 fb1a36c0 2022-01-09 op static void np_clunk(uint16_t);
134 fb1a36c0 2022-01-09 op static void np_flush(uint16_t);
135 fb1a36c0 2022-01-09 op static void np_walk(uint16_t, int, struct qid *);
136 fb1a36c0 2022-01-09 op static void np_open(uint16_t, struct qid *, uint32_t);
137 fb1a36c0 2022-01-09 op static void np_create(uint16_t, struct qid *, uint32_t);
138 fb1a36c0 2022-01-09 op static void np_read(uint16_t, uint32_t, void *);
139 fb1a36c0 2022-01-09 op static void np_write(uint16_t, uint32_t);
140 70540bb5 2022-02-02 op static void np_stat(uint16_t, uint16_t, void *);
141 755d75bf 2022-01-17 op static void np_wstat(uint16_t);
142 fb1a36c0 2022-01-09 op static void np_remove(uint16_t);
143 fb1a36c0 2022-01-09 op static void np_error(uint16_t, const char *);
144 fb1a36c0 2022-01-09 op static void np_errno(uint16_t);
145 fb1a36c0 2022-01-09 op
146 fb1a36c0 2022-01-09 op static int np_read8(const char *, const char *, uint8_t *,
147 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
148 fb1a36c0 2022-01-09 op static int np_read16(const char *, const char *, uint16_t *,
149 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
150 fb1a36c0 2022-01-09 op static int np_read32(const char *, const char *, uint32_t *,
151 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
152 fb1a36c0 2022-01-09 op static int np_read64(const char *, const char *, uint64_t *,
153 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
154 fb1a36c0 2022-01-09 op
155 fb1a36c0 2022-01-09 op #define READSTRERR -1
156 fb1a36c0 2022-01-09 op #define READSTRTRUNC -2
157 fb1a36c0 2022-01-09 op static int np_readstr(const char *, const char *, char *, size_t,
158 fb1a36c0 2022-01-09 op const uint8_t **, size_t *);
159 fb1a36c0 2022-01-09 op
160 755d75bf 2022-01-17 op static int np_readst(const char *, const char *, struct np_stat *,
161 755d75bf 2022-01-17 op char *, size_t, const uint8_t **, size_t *);
162 755d75bf 2022-01-17 op
163 fb1a36c0 2022-01-09 op #define NPREAD8(f, dst, src, len) np_read8(__func__, f, dst, src, len)
164 fb1a36c0 2022-01-09 op #define NPREAD16(f, dst, src, len) np_read16(__func__, f, dst, src, len)
165 fb1a36c0 2022-01-09 op #define NPREAD32(f, dst, src, len) np_read32(__func__, f, dst, src, len)
166 fb1a36c0 2022-01-09 op #define NPREAD64(f, dst, src, len) np_read64(__func__, f, dst, src, len)
167 fb1a36c0 2022-01-09 op
168 fb1a36c0 2022-01-09 op #define NPREADSTR(f, b, bl, src, len) np_readstr(__func__, f, b, bl, src, len)
169 fb1a36c0 2022-01-09 op
170 755d75bf 2022-01-17 op #define NPREADST(f, dst, n, nl, src, len) np_readst(__func__, f, dst, n, nl, src, len)
171 755d75bf 2022-01-17 op
172 fb1a36c0 2022-01-09 op static void tversion(struct np_msg_header *, const uint8_t *, size_t);
173 fb1a36c0 2022-01-09 op static void tattach(struct np_msg_header *, const uint8_t *, size_t);
174 fb1a36c0 2022-01-09 op static void tclunk(struct np_msg_header *, const uint8_t *, size_t);
175 fb1a36c0 2022-01-09 op static void tflush(struct np_msg_header *, const uint8_t *, size_t);
176 fb1a36c0 2022-01-09 op static void twalk(struct np_msg_header *, const uint8_t *, size_t);
177 fb1a36c0 2022-01-09 op static void topen(struct np_msg_header *, const uint8_t *, size_t);
178 fb1a36c0 2022-01-09 op static void tcreate(struct np_msg_header *, const uint8_t *, size_t);
179 fb1a36c0 2022-01-09 op static void tread(struct np_msg_header *, const uint8_t *, size_t);
180 8b0a5062 2022-05-22 op static void twrite(struct np_msg_header *, const uint8_t *, size_t, struct fid **, off_t *, size_t *, int *);
181 8b0a5062 2022-05-22 op static void twrite_cont(struct fid *, off_t *, size_t *, int *, uint16_t, const uint8_t *, size_t);
182 fb1a36c0 2022-01-09 op static void tstat(struct np_msg_header *, const uint8_t *, size_t);
183 755d75bf 2022-01-17 op static void twstat(struct np_msg_header *, const uint8_t *, size_t);
184 fb1a36c0 2022-01-09 op static void tremove(struct np_msg_header *, const uint8_t *, size_t);
185 8b0a5062 2022-05-22 op static void handle_message(struct imsg *, size_t, int);
186 fb1a36c0 2022-01-09 op
187 fb1a36c0 2022-01-09 op __dead void
188 fb1a36c0 2022-01-09 op client(int debug, int verbose)
189 fb1a36c0 2022-01-09 op {
190 fb1a36c0 2022-01-09 op struct event ev_sigint, ev_sigterm;
191 fb1a36c0 2022-01-09 op
192 fb1a36c0 2022-01-09 op log_init(debug, LOG_DAEMON);
193 fb1a36c0 2022-01-09 op log_setverbose(verbose);
194 fb1a36c0 2022-01-09 op
195 fb1a36c0 2022-01-09 op setproctitle("client");
196 fb1a36c0 2022-01-09 op log_procinit("client");
197 fb1a36c0 2022-01-09 op
198 fb1a36c0 2022-01-09 op log_debug("warming up");
199 fb1a36c0 2022-01-09 op
200 fb1a36c0 2022-01-09 op event_init();
201 fb1a36c0 2022-01-09 op
202 fb1a36c0 2022-01-09 op /* Setup signal handlers */
203 fb1a36c0 2022-01-09 op signal_set(&ev_sigint, SIGINT, client_sig_handler, NULL);
204 fb1a36c0 2022-01-09 op signal_set(&ev_sigterm, SIGTERM, client_sig_handler, NULL);
205 fb1a36c0 2022-01-09 op
206 fb1a36c0 2022-01-09 op signal_add(&ev_sigint, NULL);
207 fb1a36c0 2022-01-09 op signal_add(&ev_sigterm, NULL);
208 fb1a36c0 2022-01-09 op
209 fb1a36c0 2022-01-09 op signal(SIGPIPE, SIG_IGN);
210 fb1a36c0 2022-01-09 op signal(SIGHUP, SIG_IGN);
211 fb1a36c0 2022-01-09 op
212 fb1a36c0 2022-01-09 op /* Setup pipe and event handler to the listener process */
213 fb1a36c0 2022-01-09 op if ((iev_listener = malloc(sizeof(*iev_listener))) == NULL)
214 fb1a36c0 2022-01-09 op fatal(NULL);
215 fb1a36c0 2022-01-09 op
216 fb1a36c0 2022-01-09 op imsg_init(&iev_listener->ibuf, 3);
217 fb1a36c0 2022-01-09 op iev_listener->handler = client_dispatch_listener;
218 fb1a36c0 2022-01-09 op
219 fb1a36c0 2022-01-09 op /* Setup event handlers. */
220 fb1a36c0 2022-01-09 op iev_listener->events = EV_READ;
221 fb1a36c0 2022-01-09 op event_set(&iev_listener->ev, iev_listener->ibuf.fd,
222 fb1a36c0 2022-01-09 op iev_listener->events, iev_listener->handler, iev_listener);
223 fb1a36c0 2022-01-09 op event_add(&iev_listener->ev, NULL);
224 fb1a36c0 2022-01-09 op
225 fb1a36c0 2022-01-09 op event_dispatch();
226 fb1a36c0 2022-01-09 op client_shutdown();
227 fb1a36c0 2022-01-09 op }
228 fb1a36c0 2022-01-09 op
229 fb1a36c0 2022-01-09 op static __dead void
230 fb1a36c0 2022-01-09 op client_shutdown(void)
231 fb1a36c0 2022-01-09 op {
232 fb1a36c0 2022-01-09 op if (evb != NULL)
233 fb1a36c0 2022-01-09 op evbuffer_free(evb);
234 fb1a36c0 2022-01-09 op
235 fb1a36c0 2022-01-09 op msgbuf_clear(&iev_listener->ibuf.w);
236 fb1a36c0 2022-01-09 op close(iev_listener->ibuf.fd);
237 fb1a36c0 2022-01-09 op
238 fb1a36c0 2022-01-09 op free(iev_listener);
239 fb1a36c0 2022-01-09 op
240 fb1a36c0 2022-01-09 op log_debug("client exiting");
241 fb1a36c0 2022-01-09 op exit(0);
242 fb1a36c0 2022-01-09 op }
243 fb1a36c0 2022-01-09 op
244 fb1a36c0 2022-01-09 op static void
245 fb1a36c0 2022-01-09 op client_sig_handler(int sig, short event, void *d)
246 fb1a36c0 2022-01-09 op {
247 fb1a36c0 2022-01-09 op /*
248 fb1a36c0 2022-01-09 op * Normal signal handler rules don't apply because libevent
249 fb1a36c0 2022-01-09 op * decouples for us.
250 fb1a36c0 2022-01-09 op */
251 fb1a36c0 2022-01-09 op
252 fb1a36c0 2022-01-09 op switch (sig) {
253 fb1a36c0 2022-01-09 op case SIGINT:
254 fb1a36c0 2022-01-09 op case SIGTERM:
255 fb1a36c0 2022-01-09 op client_shutdown();
256 fb1a36c0 2022-01-09 op default:
257 fb1a36c0 2022-01-09 op fatalx("unexpected signal %d", sig);
258 fb1a36c0 2022-01-09 op }
259 fb1a36c0 2022-01-09 op }
260 fb1a36c0 2022-01-09 op
261 fb1a36c0 2022-01-09 op static void
262 fb1a36c0 2022-01-09 op client_dispatch_listener(int fd, short event, void *d)
263 fb1a36c0 2022-01-09 op {
264 1e84c7b7 2022-01-10 op static int auth = 0;
265 1e84c7b7 2022-01-10 op struct kd_auth_proc rauth;
266 5c69b561 2022-01-29 op struct kd_debug_info debug;
267 5c69b561 2022-01-29 op struct fid *f;
268 fb1a36c0 2022-01-09 op struct imsg imsg;
269 fb1a36c0 2022-01-09 op struct imsgev *iev = d;
270 fb1a36c0 2022-01-09 op struct imsgbuf *ibuf;
271 fb1a36c0 2022-01-09 op ssize_t n;
272 fb1a36c0 2022-01-09 op int shut = 0;
273 fb1a36c0 2022-01-09 op
274 fb1a36c0 2022-01-09 op ibuf = &iev->ibuf;
275 fb1a36c0 2022-01-09 op
276 fb1a36c0 2022-01-09 op if (event & EV_READ) {
277 fb1a36c0 2022-01-09 op if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
278 fb1a36c0 2022-01-09 op fatal("imsg_read error");
279 fb1a36c0 2022-01-09 op if (n == 0) /* Connection closed */
280 fb1a36c0 2022-01-09 op shut = 1;
281 fb1a36c0 2022-01-09 op }
282 fb1a36c0 2022-01-09 op if (event & EV_WRITE) {
283 fb1a36c0 2022-01-09 op if ((n = msgbuf_write(&ibuf->w)) == -1 && errno != EAGAIN)
284 fb1a36c0 2022-01-09 op fatal("msgbuf_write");
285 fb1a36c0 2022-01-09 op if (n == 0) /* Connection closed */
286 fb1a36c0 2022-01-09 op shut = 1;
287 fb1a36c0 2022-01-09 op }
288 fb1a36c0 2022-01-09 op
289 fb1a36c0 2022-01-09 op for (;;) {
290 fb1a36c0 2022-01-09 op if ((n = imsg_get(ibuf, &imsg)) == -1)
291 fb1a36c0 2022-01-09 op fatal("%s: imsg_get error", __func__);
292 fb1a36c0 2022-01-09 op if (n == 0) /* No more messages. */
293 fb1a36c0 2022-01-09 op break;
294 fb1a36c0 2022-01-09 op
295 fb1a36c0 2022-01-09 op switch (imsg.hdr.type) {
296 982f38dc 2022-01-23 op case IMSG_CTL_LOG_VERBOSE:
297 982f38dc 2022-01-23 op if (IMSG_DATA_SIZE(imsg) != sizeof(verbose))
298 982f38dc 2022-01-23 op fatalx("%s: IMSG_CTL_LOG_VERBOSE wrong size",
299 982f38dc 2022-01-23 op __func__);
300 982f38dc 2022-01-23 op memcpy(&verbose, imsg.data, sizeof(verbose));
301 982f38dc 2022-01-23 op log_setverbose(verbose);
302 982f38dc 2022-01-23 op break;
303 5c69b561 2022-01-29 op case IMSG_CTL_DEBUG:
304 5c69b561 2022-01-29 op STAILQ_FOREACH(f, &fids, entries) {
305 5c69b561 2022-01-29 op memset(&debug, 0, sizeof(debug));
306 5c69b561 2022-01-29 op debug.fid = f->fid;
307 5c69b561 2022-01-29 op strlcpy(debug.path, f->fname,
308 5c69b561 2022-01-29 op sizeof(debug.path));
309 5c69b561 2022-01-29 op client_send_listenerp(IMSG_CTL_DEBUG_BACK,
310 5c69b561 2022-01-29 op imsg.hdr.peerid, &debug, sizeof(debug));
311 5c69b561 2022-01-29 op }
312 5c69b561 2022-01-29 op client_send_listenerp(IMSG_CTL_DEBUG_END,
313 5c69b561 2022-01-29 op imsg.hdr.peerid, NULL, 0);
314 5c69b561 2022-01-29 op break;
315 fb1a36c0 2022-01-09 op case IMSG_AUTH:
316 fb1a36c0 2022-01-09 op peerid = imsg.hdr.peerid;
317 fb1a36c0 2022-01-09 op if (auth)
318 fb1a36c0 2022-01-09 op fatalx("%s: IMSG_AUTH already done", __func__);
319 1e84c7b7 2022-01-10 op auth = 1;
320 1e84c7b7 2022-01-10 op
321 1e84c7b7 2022-01-10 op if (IMSG_DATA_SIZE(imsg) != sizeof(rauth))
322 1e84c7b7 2022-01-10 op fatalx("mismatching size for IMSG_AUTH");
323 1e84c7b7 2022-01-10 op memcpy(&rauth, imsg.data, sizeof(rauth));
324 1e84c7b7 2022-01-10 op if (rauth.uname[sizeof(rauth.uname)-1] != '\0' ||
325 1e84c7b7 2022-01-10 op rauth.dir[sizeof(rauth.dir)-1] != '\0')
326 1e84c7b7 2022-01-10 op fatalx("IMSG_AUTH strings not NUL-terminated");
327 1e84c7b7 2022-01-10 op
328 1e84c7b7 2022-01-10 op client_privdrop(rauth.uname, rauth.dir);
329 1e84c7b7 2022-01-10 op explicit_bzero(&rauth, sizeof(rauth));
330 fb1a36c0 2022-01-09 op break;
331 fb1a36c0 2022-01-09 op case IMSG_BUF:
332 8b0a5062 2022-05-22 op case IMSG_BUF_CONT:
333 fb1a36c0 2022-01-09 op if (!auth)
334 fb1a36c0 2022-01-09 op fatalx("%s: can't handle messages before"
335 fb1a36c0 2022-01-09 op " doing the auth", __func__);
336 8b0a5062 2022-05-22 op handle_message(&imsg, IMSG_DATA_SIZE(imsg),
337 8b0a5062 2022-05-22 op imsg.hdr.type == IMSG_BUF_CONT);
338 fb1a36c0 2022-01-09 op break;
339 fb1a36c0 2022-01-09 op case IMSG_CONN_GONE:
340 fb1a36c0 2022-01-09 op log_debug("closing");
341 fb1a36c0 2022-01-09 op shut = 1;
342 fb1a36c0 2022-01-09 op break;
343 fb1a36c0 2022-01-09 op default:
344 fb1a36c0 2022-01-09 op log_debug("%s: unexpected imsg %d",
345 fb1a36c0 2022-01-09 op __func__, imsg.hdr.type);
346 fb1a36c0 2022-01-09 op break;
347 fb1a36c0 2022-01-09 op }
348 fb1a36c0 2022-01-09 op imsg_free(&imsg);
349 fb1a36c0 2022-01-09 op }
350 fb1a36c0 2022-01-09 op
351 fb1a36c0 2022-01-09 op if (!shut)
352 fb1a36c0 2022-01-09 op imsg_event_add(iev);
353 fb1a36c0 2022-01-09 op else {
354 fb1a36c0 2022-01-09 op /* This pipe is dead. Remove its event handler. */
355 fb1a36c0 2022-01-09 op event_del(&iev->ev);
356 fb1a36c0 2022-01-09 op log_debug("pipe closed, shutting down...");
357 fb1a36c0 2022-01-09 op event_loopexit(NULL);
358 fb1a36c0 2022-01-09 op }
359 fb1a36c0 2022-01-09 op }
360 fb1a36c0 2022-01-09 op
361 fb1a36c0 2022-01-09 op static void
362 fb1a36c0 2022-01-09 op client_privdrop(const char *username, const char *dir)
363 fb1a36c0 2022-01-09 op {
364 fb1a36c0 2022-01-09 op struct passwd *pw;
365 fb1a36c0 2022-01-09 op
366 fb1a36c0 2022-01-09 op setproctitle("client %s", username);
367 fb1a36c0 2022-01-09 op
368 fb1a36c0 2022-01-09 op if ((pw = getpwnam(username)) == NULL)
369 fb1a36c0 2022-01-09 op fatalx("getpwnam(%s) failed", username);
370 fb1a36c0 2022-01-09 op
371 fb1a36c0 2022-01-09 op if (chroot(dir) == -1)
372 fb1a36c0 2022-01-09 op fatal("chroot");
373 fb1a36c0 2022-01-09 op if (chdir("/") == -1)
374 fb1a36c0 2022-01-09 op fatal("chdir(\"/\")");
375 fb1a36c0 2022-01-09 op
376 fb1a36c0 2022-01-09 op if (setgroups(1, &pw->pw_gid) ||
377 fb1a36c0 2022-01-09 op setresgid(pw->pw_gid, pw->pw_gid, pw->pw_gid) ||
378 fb1a36c0 2022-01-09 op setresuid(pw->pw_uid, pw->pw_uid, pw->pw_uid))
379 fb1a36c0 2022-01-09 op fatal("can't drop privileges");
380 fb1a36c0 2022-01-09 op
381 fb1a36c0 2022-01-09 op sandbox_client();
382 fb1a36c0 2022-01-09 op log_debug("client ready; user=%s dir=%s", username, dir);
383 fb1a36c0 2022-01-09 op
384 fb1a36c0 2022-01-09 op if ((evb = evbuffer_new()) == NULL)
385 fb1a36c0 2022-01-09 op fatal("evbuffer_new");
386 fb1a36c0 2022-01-09 op }
387 fb1a36c0 2022-01-09 op
388 fb1a36c0 2022-01-09 op static int
389 5c69b561 2022-01-29 op client_send_listenerp(int type, uint32_t peerid, const void *data, uint16_t len)
390 fb1a36c0 2022-01-09 op {
391 fb1a36c0 2022-01-09 op int ret;
392 fb1a36c0 2022-01-09 op
393 fb1a36c0 2022-01-09 op if ((ret = imsg_compose(&iev_listener->ibuf, type, peerid, 0, -1,
394 fb1a36c0 2022-01-09 op data, len)) != -1)
395 fb1a36c0 2022-01-09 op imsg_event_add(iev_listener);
396 fb1a36c0 2022-01-09 op
397 fb1a36c0 2022-01-09 op return ret;
398 fb1a36c0 2022-01-09 op }
399 fb1a36c0 2022-01-09 op
400 5c69b561 2022-01-29 op static int
401 5c69b561 2022-01-29 op client_send_listener(int type, const void *data, uint16_t len)
402 5c69b561 2022-01-29 op {
403 865dd323 2022-01-29 op return client_send_listenerp(type, peerid, data, len);
404 5c69b561 2022-01-29 op }
405 5c69b561 2022-01-29 op
406 fb1a36c0 2022-01-09 op /* set qid fields from sb */
407 fb1a36c0 2022-01-09 op static void
408 fb1a36c0 2022-01-09 op qid_update_from_sb(struct qid *qid, struct stat *sb)
409 fb1a36c0 2022-01-09 op {
410 fb1a36c0 2022-01-09 op qid->path = sb->st_ino;
411 fb1a36c0 2022-01-09 op
412 fb1a36c0 2022-01-09 op /*
413 fb1a36c0 2022-01-09 op * Theoretically (and hopefully!) this should be a 64 bit
414 fb1a36c0 2022-01-09 op * number. Unfortunately, 9P uses 32 bit timestamps.
415 fb1a36c0 2022-01-09 op */
416 fb1a36c0 2022-01-09 op qid->vers = sb->st_mtim.tv_sec;
417 fb1a36c0 2022-01-09 op
418 fb1a36c0 2022-01-09 op if (S_ISREG(sb->st_mode))
419 fb1a36c0 2022-01-09 op qid->type = QTFILE;
420 fb1a36c0 2022-01-09 op else if (S_ISDIR(sb->st_mode))
421 fb1a36c0 2022-01-09 op qid->type = QTDIR;
422 fb1a36c0 2022-01-09 op else if (S_ISLNK(sb->st_mode))
423 fb1a36c0 2022-01-09 op qid->type = QTSYMLINK;
424 fb1a36c0 2022-01-09 op }
425 fb1a36c0 2022-01-09 op
426 fb1a36c0 2022-01-09 op /* creates a qid given a fd */
427 fb1a36c0 2022-01-09 op static struct dir *
428 fb1a36c0 2022-01-09 op new_dir(int fd)
429 fb1a36c0 2022-01-09 op {
430 fb1a36c0 2022-01-09 op struct dir *dir;
431 fb1a36c0 2022-01-09 op
432 fb1a36c0 2022-01-09 op if ((dir = calloc(1, sizeof(*dir))) == NULL)
433 fb1a36c0 2022-01-09 op return NULL;
434 fb1a36c0 2022-01-09 op
435 fb1a36c0 2022-01-09 op dir->fd = fd;
436 fb1a36c0 2022-01-09 op STAILQ_INSERT_HEAD(&dirs, dir, entries);
437 fb1a36c0 2022-01-09 op return dir;
438 fb1a36c0 2022-01-09 op }
439 fb1a36c0 2022-01-09 op
440 fb1a36c0 2022-01-09 op static struct dir *
441 fb1a36c0 2022-01-09 op dir_incref(struct dir *dir)
442 fb1a36c0 2022-01-09 op {
443 fb1a36c0 2022-01-09 op dir->refcount++;
444 fb1a36c0 2022-01-09 op return dir;
445 fb1a36c0 2022-01-09 op }
446 fb1a36c0 2022-01-09 op
447 fb1a36c0 2022-01-09 op static void
448 fb1a36c0 2022-01-09 op dir_decref(struct dir *dir)
449 fb1a36c0 2022-01-09 op {
450 fb1a36c0 2022-01-09 op if (--dir->refcount > 0)
451 fb1a36c0 2022-01-09 op return;
452 fb1a36c0 2022-01-09 op
453 fb1a36c0 2022-01-09 op STAILQ_REMOVE(&dirs, dir, dir, entries);
454 fb1a36c0 2022-01-09 op
455 fb1a36c0 2022-01-09 op close(dir->fd);
456 fb1a36c0 2022-01-09 op free(dir);
457 fb1a36c0 2022-01-09 op }
458 fb1a36c0 2022-01-09 op
459 fb1a36c0 2022-01-09 op static struct fid *
460 fb1a36c0 2022-01-09 op new_fid(struct dir *dir, uint32_t fid, const char *path, struct qid *qid)
461 fb1a36c0 2022-01-09 op {
462 fb1a36c0 2022-01-09 op struct fid *f;
463 fb1a36c0 2022-01-09 op struct qid q;
464 fb1a36c0 2022-01-09 op struct stat sb;
465 fb1a36c0 2022-01-09 op
466 fb1a36c0 2022-01-09 op if (qid == NULL) {
467 fb1a36c0 2022-01-09 op if (fstatat(dir->fd, path, &sb, 0)) {
468 fb1a36c0 2022-01-09 op log_warn("fstatat(%s)", path);
469 fb1a36c0 2022-01-09 op return NULL;
470 fb1a36c0 2022-01-09 op }
471 fb1a36c0 2022-01-09 op qid_update_from_sb(&q, &sb);
472 fb1a36c0 2022-01-09 op qid = &q;
473 fb1a36c0 2022-01-09 op }
474 fb1a36c0 2022-01-09 op
475 fb1a36c0 2022-01-09 op if ((f = calloc(1, sizeof(*f))) == NULL)
476 fb1a36c0 2022-01-09 op return NULL;
477 fb1a36c0 2022-01-09 op
478 fb1a36c0 2022-01-09 op f->dir = dir_incref(dir);
479 fb1a36c0 2022-01-09 op f->fid = fid;
480 fb1a36c0 2022-01-09 op f->fd = -1;
481 fb1a36c0 2022-01-09 op
482 543fc013 2022-01-17 op strlcpy(f->fname, path, sizeof(f->fname));
483 fb1a36c0 2022-01-09 op
484 fb1a36c0 2022-01-09 op memcpy(&f->qid, qid, sizeof(f->qid));
485 fb1a36c0 2022-01-09 op
486 fb1a36c0 2022-01-09 op STAILQ_INSERT_HEAD(&fids, f, entries);
487 fb1a36c0 2022-01-09 op
488 fb1a36c0 2022-01-09 op return f;
489 fb1a36c0 2022-01-09 op }
490 fb1a36c0 2022-01-09 op
491 fb1a36c0 2022-01-09 op static struct fid *
492 fb1a36c0 2022-01-09 op fid_by_id(uint32_t fid)
493 fb1a36c0 2022-01-09 op {
494 fb1a36c0 2022-01-09 op struct fid *f;
495 fb1a36c0 2022-01-09 op
496 fb1a36c0 2022-01-09 op STAILQ_FOREACH(f, &fids, entries) {
497 fb1a36c0 2022-01-09 op if (f->fid == fid)
498 fb1a36c0 2022-01-09 op return f;
499 fb1a36c0 2022-01-09 op }
500 fb1a36c0 2022-01-09 op
501 fb1a36c0 2022-01-09 op return NULL;
502 fb1a36c0 2022-01-09 op }
503 fb1a36c0 2022-01-09 op
504 fb1a36c0 2022-01-09 op static void
505 fb1a36c0 2022-01-09 op free_fid(struct fid *f)
506 fb1a36c0 2022-01-09 op {
507 fb1a36c0 2022-01-09 op int r;
508 fb1a36c0 2022-01-09 op
509 fb1a36c0 2022-01-09 op if (f->fd != -1) {
510 fb1a36c0 2022-01-09 op if (f->d != NULL)
511 fb1a36c0 2022-01-09 op r = closedir(f->d);
512 fb1a36c0 2022-01-09 op else
513 fb1a36c0 2022-01-09 op r = close(f->fd);
514 fb1a36c0 2022-01-09 op
515 fb1a36c0 2022-01-09 op if (r == -1)
516 fb1a36c0 2022-01-09 op fatal("can't close fid %d", f->fid);
517 fb1a36c0 2022-01-09 op
518 fb1a36c0 2022-01-09 op if (f->evb != NULL)
519 fb1a36c0 2022-01-09 op evbuffer_free(f->evb);
520 fb1a36c0 2022-01-09 op
521 fb1a36c0 2022-01-09 op /* try to honour ORCLOSE if requested */
522 fb1a36c0 2022-01-09 op if (f->iomode & O_CLOEXEC)
523 543fc013 2022-01-17 op unlinkat(f->dir->fd, f->fname, 0);
524 fb1a36c0 2022-01-09 op }
525 fb1a36c0 2022-01-09 op
526 fb1a36c0 2022-01-09 op dir_decref(f->dir);
527 fb1a36c0 2022-01-09 op
528 fb1a36c0 2022-01-09 op STAILQ_REMOVE(&fids, f, fid, entries);
529 fb1a36c0 2022-01-09 op free(f);
530 fb1a36c0 2022-01-09 op }
531 fb1a36c0 2022-01-09 op
532 fb1a36c0 2022-01-09 op static void
533 fb1a36c0 2022-01-09 op parse_message(const uint8_t *data, size_t len, struct np_msg_header *hdr,
534 fb1a36c0 2022-01-09 op uint8_t **cnt)
535 fb1a36c0 2022-01-09 op {
536 fb1a36c0 2022-01-09 op size_t olen = len;
537 fb1a36c0 2022-01-09 op
538 fb1a36c0 2022-01-09 op if (!NPREAD32("len", &hdr->len, &data, &len) ||
539 fb1a36c0 2022-01-09 op !NPREAD8("type", &hdr->type, &data, &len) ||
540 fb1a36c0 2022-01-09 op !NPREAD16("tag", &hdr->tag, &data, &len))
541 fb1a36c0 2022-01-09 op goto err;
542 fb1a36c0 2022-01-09 op
543 fb1a36c0 2022-01-09 op if (olen != hdr->len)
544 fb1a36c0 2022-01-09 op goto err;
545 fb1a36c0 2022-01-09 op
546 fb1a36c0 2022-01-09 op if (hdr->type < Tversion ||
547 fb1a36c0 2022-01-09 op hdr->type >= Tmax ||
548 fb1a36c0 2022-01-09 op hdr->type == Terror ||
549 fb1a36c0 2022-01-09 op (hdr->type & 0x1) != 0) /* cannot recv a R* */
550 fb1a36c0 2022-01-09 op goto err;
551 fb1a36c0 2022-01-09 op
552 fb1a36c0 2022-01-09 op hdr->tag = le32toh(hdr->tag);
553 fb1a36c0 2022-01-09 op
554 fb1a36c0 2022-01-09 op *cnt = (uint8_t *)data;
555 fb1a36c0 2022-01-09 op return;
556 fb1a36c0 2022-01-09 op
557 fb1a36c0 2022-01-09 op err:
558 182b48dc 2022-05-22 op log_warnx("got invalid message: (%d, %d, %d)",
559 182b48dc 2022-05-22 op hdr->len, hdr->type, hdr->tag);
560 182b48dc 2022-05-22 op client_send_listener(IMSG_CLOSE, NULL, 0);
561 182b48dc 2022-05-22 op client_shutdown();
562 fb1a36c0 2022-01-09 op }
563 fb1a36c0 2022-01-09 op
564 fb1a36c0 2022-01-09 op static void
565 fb1a36c0 2022-01-09 op np_write16(struct evbuffer *e, uint16_t x)
566 fb1a36c0 2022-01-09 op {
567 fb1a36c0 2022-01-09 op x = htole16(x);
568 fb1a36c0 2022-01-09 op evbuffer_add(e, &x, sizeof(x));
569 fb1a36c0 2022-01-09 op }
570 fb1a36c0 2022-01-09 op
571 fb1a36c0 2022-01-09 op static void
572 fb1a36c0 2022-01-09 op np_write32(struct evbuffer *e, uint32_t x)
573 fb1a36c0 2022-01-09 op {
574 fb1a36c0 2022-01-09 op x = htole32(x);
575 fb1a36c0 2022-01-09 op evbuffer_add(e, &x, sizeof(x));
576 fb1a36c0 2022-01-09 op }
577 fb1a36c0 2022-01-09 op
578 fb1a36c0 2022-01-09 op static void
579 fb1a36c0 2022-01-09 op np_write64(struct evbuffer *e, uint64_t x)
580 fb1a36c0 2022-01-09 op {
581 fb1a36c0 2022-01-09 op x = htole64(x);
582 fb1a36c0 2022-01-09 op evbuffer_add(e, &x, sizeof(x));
583 fb1a36c0 2022-01-09 op }
584 fb1a36c0 2022-01-09 op
585 fb1a36c0 2022-01-09 op static void
586 fb1a36c0 2022-01-09 op np_writebuf(struct evbuffer *e, size_t len, void *data)
587 fb1a36c0 2022-01-09 op {
588 fb1a36c0 2022-01-09 op evbuffer_add(e, data, len);
589 fb1a36c0 2022-01-09 op }
590 fb1a36c0 2022-01-09 op
591 fb1a36c0 2022-01-09 op static void
592 fb1a36c0 2022-01-09 op np_header(uint32_t len, uint8_t type, uint16_t tag)
593 fb1a36c0 2022-01-09 op {
594 fb1a36c0 2022-01-09 op len += HEADERSIZE;
595 fb1a36c0 2022-01-09 op
596 fb1a36c0 2022-01-09 op len = htole32(len);
597 fb1a36c0 2022-01-09 op tag = htole16(tag);
598 fb1a36c0 2022-01-09 op
599 fb1a36c0 2022-01-09 op evbuffer_add(evb, &len, sizeof(len));
600 fb1a36c0 2022-01-09 op evbuffer_add(evb, &type, sizeof(type));
601 fb1a36c0 2022-01-09 op evbuffer_add(evb, &tag, sizeof(tag));
602 fb1a36c0 2022-01-09 op }
603 fb1a36c0 2022-01-09 op
604 fb1a36c0 2022-01-09 op static void
605 fb1a36c0 2022-01-09 op np_string(struct evbuffer *e, uint16_t len, const char *str)
606 fb1a36c0 2022-01-09 op {
607 fb1a36c0 2022-01-09 op uint16_t l = len;
608 fb1a36c0 2022-01-09 op
609 fb1a36c0 2022-01-09 op len = htole16(len);
610 fb1a36c0 2022-01-09 op evbuffer_add(e, &len, sizeof(len));
611 fb1a36c0 2022-01-09 op evbuffer_add(e, str, l);
612 fb1a36c0 2022-01-09 op }
613 fb1a36c0 2022-01-09 op
614 fb1a36c0 2022-01-09 op static void
615 fb1a36c0 2022-01-09 op np_qid(struct evbuffer *e, struct qid *qid)
616 fb1a36c0 2022-01-09 op {
617 fb1a36c0 2022-01-09 op uint64_t path;
618 fb1a36c0 2022-01-09 op uint32_t vers;
619 fb1a36c0 2022-01-09 op
620 fb1a36c0 2022-01-09 op path = htole64(qid->path);
621 fb1a36c0 2022-01-09 op vers = htole32(qid->vers);
622 fb1a36c0 2022-01-09 op
623 fb1a36c0 2022-01-09 op evbuffer_add(e, &qid->type, sizeof(qid->type));
624 fb1a36c0 2022-01-09 op evbuffer_add(e, &vers, sizeof(vers));
625 fb1a36c0 2022-01-09 op evbuffer_add(e, &path, sizeof(path));
626 fb1a36c0 2022-01-09 op }
627 fb1a36c0 2022-01-09 op
628 fb1a36c0 2022-01-09 op static void
629 fb1a36c0 2022-01-09 op do_send(void)
630 fb1a36c0 2022-01-09 op {
631 fb1a36c0 2022-01-09 op size_t len;
632 fb1a36c0 2022-01-09 op void *data;
633 fb1a36c0 2022-01-09 op
634 fb1a36c0 2022-01-09 op len = EVBUFFER_LENGTH(evb);
635 fb1a36c0 2022-01-09 op data = EVBUFFER_DATA(evb);
636 fb1a36c0 2022-01-09 op
637 fb1a36c0 2022-01-09 op #if DEBUG_PACKETS
638 fb1a36c0 2022-01-09 op hexdump("outgoing packet", data, len);
639 fb1a36c0 2022-01-09 op #endif
640 fb1a36c0 2022-01-09 op client_send_listener(IMSG_BUF, data, len);
641 fb1a36c0 2022-01-09 op evbuffer_drain(evb, len);
642 fb1a36c0 2022-01-09 op }
643 fb1a36c0 2022-01-09 op
644 fb1a36c0 2022-01-09 op static void
645 fb1a36c0 2022-01-09 op np_version(uint16_t tag, uint32_t msize, const char *version)
646 fb1a36c0 2022-01-09 op {
647 fb1a36c0 2022-01-09 op uint16_t l;
648 fb1a36c0 2022-01-09 op
649 fb1a36c0 2022-01-09 op l = strlen(version);
650 fb1a36c0 2022-01-09 op
651 fb1a36c0 2022-01-09 op msize = htole32(msize);
652 fb1a36c0 2022-01-09 op
653 fb1a36c0 2022-01-09 op np_header(sizeof(msize) + sizeof(l) + l, Rversion, tag);
654 fb1a36c0 2022-01-09 op evbuffer_add(evb, &msize, sizeof(msize));
655 fb1a36c0 2022-01-09 op np_string(evb, l, version);
656 fb1a36c0 2022-01-09 op do_send();
657 fb1a36c0 2022-01-09 op }
658 fb1a36c0 2022-01-09 op
659 fb1a36c0 2022-01-09 op static void
660 fb1a36c0 2022-01-09 op np_attach(uint16_t tag, struct qid *qid)
661 fb1a36c0 2022-01-09 op {
662 fb1a36c0 2022-01-09 op np_header(QIDSIZE, Rattach, tag);
663 fb1a36c0 2022-01-09 op np_qid(evb, qid);
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_clunk(uint16_t tag)
669 fb1a36c0 2022-01-09 op {
670 fb1a36c0 2022-01-09 op np_header(0, Rclunk, tag);
671 fb1a36c0 2022-01-09 op do_send();
672 fb1a36c0 2022-01-09 op }
673 fb1a36c0 2022-01-09 op
674 fb1a36c0 2022-01-09 op static void
675 fb1a36c0 2022-01-09 op np_flush(uint16_t tag)
676 fb1a36c0 2022-01-09 op {
677 fb1a36c0 2022-01-09 op np_header(0, Rflush, tag);
678 fb1a36c0 2022-01-09 op do_send();
679 fb1a36c0 2022-01-09 op }
680 fb1a36c0 2022-01-09 op
681 fb1a36c0 2022-01-09 op static void
682 fb1a36c0 2022-01-09 op np_walk(uint16_t tag, int nwqid, struct qid *wqid)
683 fb1a36c0 2022-01-09 op {
684 fb1a36c0 2022-01-09 op int i;
685 fb1a36c0 2022-01-09 op
686 fb1a36c0 2022-01-09 op /* two bytes for the counter */
687 fb1a36c0 2022-01-09 op np_header(2 + QIDSIZE * nwqid, Rwalk, tag);
688 fb1a36c0 2022-01-09 op np_write16(evb, nwqid);
689 fb1a36c0 2022-01-09 op for (i = 0; i < nwqid; ++i)
690 fb1a36c0 2022-01-09 op np_qid(evb, wqid + i);
691 fb1a36c0 2022-01-09 op
692 fb1a36c0 2022-01-09 op do_send();
693 fb1a36c0 2022-01-09 op }
694 fb1a36c0 2022-01-09 op
695 fb1a36c0 2022-01-09 op static void
696 fb1a36c0 2022-01-09 op np_open(uint16_t tag, struct qid *qid, uint32_t iounit)
697 fb1a36c0 2022-01-09 op {
698 fb1a36c0 2022-01-09 op np_header(QIDSIZE + sizeof(iounit), Ropen, tag);
699 fb1a36c0 2022-01-09 op np_qid(evb, qid);
700 fb1a36c0 2022-01-09 op np_write32(evb, iounit);
701 fb1a36c0 2022-01-09 op do_send();
702 fb1a36c0 2022-01-09 op }
703 fb1a36c0 2022-01-09 op
704 fb1a36c0 2022-01-09 op static void
705 fb1a36c0 2022-01-09 op np_create(uint16_t tag, struct qid *qid, uint32_t iounit)
706 fb1a36c0 2022-01-09 op {
707 fb1a36c0 2022-01-09 op np_header(QIDSIZE + sizeof(iounit), Rcreate, tag);
708 fb1a36c0 2022-01-09 op np_qid(evb, qid);
709 fb1a36c0 2022-01-09 op np_write32(evb, iounit);
710 fb1a36c0 2022-01-09 op do_send();
711 fb1a36c0 2022-01-09 op }
712 fb1a36c0 2022-01-09 op
713 fb1a36c0 2022-01-09 op static void
714 fb1a36c0 2022-01-09 op np_read(uint16_t tag, uint32_t count, void *data)
715 fb1a36c0 2022-01-09 op {
716 fb1a36c0 2022-01-09 op if (sizeof(count) + count + HEADERSIZE >= msize) {
717 fb1a36c0 2022-01-09 op np_error(tag, "Rread would overflow");
718 fb1a36c0 2022-01-09 op return;
719 fb1a36c0 2022-01-09 op }
720 fb1a36c0 2022-01-09 op
721 fb1a36c0 2022-01-09 op np_header(sizeof(count) + count, Rread, tag);
722 fb1a36c0 2022-01-09 op np_write32(evb, count);
723 fb1a36c0 2022-01-09 op np_writebuf(evb, count, data);
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_write(uint16_t tag, uint32_t count)
729 fb1a36c0 2022-01-09 op {
730 fb1a36c0 2022-01-09 op np_header(sizeof(count), Rwrite, tag);
731 fb1a36c0 2022-01-09 op np_write32(evb, count);
732 fb1a36c0 2022-01-09 op do_send();
733 fb1a36c0 2022-01-09 op }
734 fb1a36c0 2022-01-09 op
735 fb1a36c0 2022-01-09 op static void
736 70540bb5 2022-02-02 op np_stat(uint16_t tag, uint16_t count, void *data)
737 fb1a36c0 2022-01-09 op {
738 fb1a36c0 2022-01-09 op if (sizeof(count) + count + HEADERSIZE >= msize) {
739 fb1a36c0 2022-01-09 op np_error(tag, "Rstat would overflow");
740 fb1a36c0 2022-01-09 op return;
741 fb1a36c0 2022-01-09 op }
742 fb1a36c0 2022-01-09 op
743 70540bb5 2022-02-02 op np_header(sizeof(count) + count, Rstat, tag);
744 70540bb5 2022-02-02 op np_write16(evb, count);
745 fb1a36c0 2022-01-09 op np_writebuf(evb, count, data);
746 755d75bf 2022-01-17 op do_send();
747 755d75bf 2022-01-17 op }
748 755d75bf 2022-01-17 op
749 755d75bf 2022-01-17 op static void
750 755d75bf 2022-01-17 op np_wstat(uint16_t tag)
751 755d75bf 2022-01-17 op {
752 755d75bf 2022-01-17 op np_header(0, Rwstat, tag);
753 fb1a36c0 2022-01-09 op do_send();
754 fb1a36c0 2022-01-09 op }
755 fb1a36c0 2022-01-09 op
756 fb1a36c0 2022-01-09 op static void
757 fb1a36c0 2022-01-09 op np_remove(uint16_t tag)
758 fb1a36c0 2022-01-09 op {
759 fb1a36c0 2022-01-09 op np_header(0, Rremove, tag);
760 fb1a36c0 2022-01-09 op do_send();
761 fb1a36c0 2022-01-09 op }
762 fb1a36c0 2022-01-09 op
763 fb1a36c0 2022-01-09 op static void
764 fb1a36c0 2022-01-09 op np_error(uint16_t tag, const char *errstr)
765 fb1a36c0 2022-01-09 op {
766 fb1a36c0 2022-01-09 op uint16_t l;
767 fb1a36c0 2022-01-09 op
768 fb1a36c0 2022-01-09 op l = strlen(errstr);
769 fb1a36c0 2022-01-09 op
770 fb1a36c0 2022-01-09 op np_header(sizeof(l) + l, Rerror, tag);
771 fb1a36c0 2022-01-09 op np_string(evb, l, errstr);
772 fb1a36c0 2022-01-09 op do_send();
773 fb1a36c0 2022-01-09 op }
774 fb1a36c0 2022-01-09 op
775 fb1a36c0 2022-01-09 op static void
776 fb1a36c0 2022-01-09 op np_errno(uint16_t tag)
777 fb1a36c0 2022-01-09 op {
778 fb1a36c0 2022-01-09 op int saved_errno;
779 fb1a36c0 2022-01-09 op char buf[NL_TEXTMAX] = {0};
780 fb1a36c0 2022-01-09 op
781 fb1a36c0 2022-01-09 op saved_errno = errno;
782 fb1a36c0 2022-01-09 op
783 fb1a36c0 2022-01-09 op strerror_r(errno, buf, sizeof(buf));
784 fb1a36c0 2022-01-09 op np_error(tag, buf);
785 fb1a36c0 2022-01-09 op
786 fb1a36c0 2022-01-09 op errno = saved_errno;
787 fb1a36c0 2022-01-09 op }
788 fb1a36c0 2022-01-09 op
789 fb1a36c0 2022-01-09 op static int
790 fb1a36c0 2022-01-09 op np_read8(const char *t, const char *f, uint8_t *dst, const uint8_t **src,
791 fb1a36c0 2022-01-09 op size_t *len)
792 fb1a36c0 2022-01-09 op {
793 fb1a36c0 2022-01-09 op if (*len < sizeof(*dst)) {
794 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %zu bytes for the %s field but only "
795 fb1a36c0 2022-01-09 op "%zu are available.", t, sizeof(*dst), f, *len);
796 ae6f257a 2022-01-19 op return 0;
797 fb1a36c0 2022-01-09 op }
798 fb1a36c0 2022-01-09 op
799 fb1a36c0 2022-01-09 op memcpy(dst, *src, sizeof(*dst));
800 fb1a36c0 2022-01-09 op *src += sizeof(*dst);
801 fb1a36c0 2022-01-09 op *len -= sizeof(*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_read16(const char *t, const char *f, uint16_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 ae6f257a 2022-01-19 op return 0;
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 = le16toh(*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_read32(const char *t, const char *f, uint32_t *dst, const uint8_t **src,
826 fb1a36c0 2022-01-09 op size_t *len)
827 fb1a36c0 2022-01-09 op {
828 fb1a36c0 2022-01-09 op if (*len < sizeof(*dst)) {
829 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %zu bytes for the %s field but only "
830 fb1a36c0 2022-01-09 op "%zu are available.", t, sizeof(*dst), f, *len);
831 ae6f257a 2022-01-19 op return 0;
832 fb1a36c0 2022-01-09 op }
833 fb1a36c0 2022-01-09 op
834 fb1a36c0 2022-01-09 op memcpy(dst, *src, sizeof(*dst));
835 fb1a36c0 2022-01-09 op *src += sizeof(*dst);
836 fb1a36c0 2022-01-09 op *len -= sizeof(*dst);
837 fb1a36c0 2022-01-09 op *dst = le32toh(*dst);
838 fb1a36c0 2022-01-09 op
839 fb1a36c0 2022-01-09 op return 1;
840 fb1a36c0 2022-01-09 op }
841 fb1a36c0 2022-01-09 op
842 fb1a36c0 2022-01-09 op static int
843 fb1a36c0 2022-01-09 op np_read64(const char *t, const char *f, uint64_t *dst, const uint8_t **src,
844 fb1a36c0 2022-01-09 op size_t *len)
845 fb1a36c0 2022-01-09 op {
846 fb1a36c0 2022-01-09 op if (*len < sizeof(*dst)) {
847 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %zu bytes for the %s field but only "
848 fb1a36c0 2022-01-09 op "%zu are available.", t, sizeof(*dst), f, *len);
849 ae6f257a 2022-01-19 op return 0;
850 fb1a36c0 2022-01-09 op }
851 fb1a36c0 2022-01-09 op
852 fb1a36c0 2022-01-09 op memcpy(dst, *src, sizeof(*dst));
853 fb1a36c0 2022-01-09 op *src += sizeof(*dst);
854 fb1a36c0 2022-01-09 op *len -= sizeof(*dst);
855 fb1a36c0 2022-01-09 op *dst = le64toh(*dst);
856 fb1a36c0 2022-01-09 op
857 fb1a36c0 2022-01-09 op return 1;
858 fb1a36c0 2022-01-09 op }
859 fb1a36c0 2022-01-09 op
860 fb1a36c0 2022-01-09 op static int
861 fb1a36c0 2022-01-09 op np_readstr(const char *t, const char *f, char *res, size_t reslen,
862 fb1a36c0 2022-01-09 op const uint8_t **src, size_t *len)
863 fb1a36c0 2022-01-09 op {
864 fb1a36c0 2022-01-09 op uint16_t sl;
865 fb1a36c0 2022-01-09 op char buf[32];
866 fb1a36c0 2022-01-09 op
867 fb1a36c0 2022-01-09 op strlcpy(buf, f, sizeof(buf));
868 fb1a36c0 2022-01-09 op strlcat(buf, "-len", sizeof(buf));
869 fb1a36c0 2022-01-09 op
870 fb1a36c0 2022-01-09 op if (!np_read16(t, buf, &sl, src, len))
871 fb1a36c0 2022-01-09 op return READSTRERR;
872 fb1a36c0 2022-01-09 op
873 fb1a36c0 2022-01-09 op if (*len < sl) {
874 fb1a36c0 2022-01-09 op log_warnx("%s: wanted %d bytes for the %s field but only "
875 fb1a36c0 2022-01-09 op "%zu are available.", t, sl, f, *len);
876 fb1a36c0 2022-01-09 op return READSTRERR;
877 fb1a36c0 2022-01-09 op }
878 fb1a36c0 2022-01-09 op
879 fb1a36c0 2022-01-09 op if (*len > reslen-1)
880 fb1a36c0 2022-01-09 op return READSTRTRUNC;
881 fb1a36c0 2022-01-09 op
882 fb1a36c0 2022-01-09 op memcpy(res, *src, sl);
883 fb1a36c0 2022-01-09 op res[sl] = '\0';
884 fb1a36c0 2022-01-09 op *src += sl;
885 fb1a36c0 2022-01-09 op *len -= sl;
886 fb1a36c0 2022-01-09 op
887 fb1a36c0 2022-01-09 op return 0;
888 755d75bf 2022-01-17 op }
889 755d75bf 2022-01-17 op
890 755d75bf 2022-01-17 op static int
891 755d75bf 2022-01-17 op np_readst(const char *t, const char *f, struct np_stat *st,
892 755d75bf 2022-01-17 op char *name, size_t namelen, const uint8_t **src, size_t *len)
893 755d75bf 2022-01-17 op {
894 755d75bf 2022-01-17 op memset(st, 0, sizeof(*st));
895 755d75bf 2022-01-17 op
896 4e7aba8f 2022-02-02 op /* len is sent twice! */
897 4e7aba8f 2022-02-02 op if (!np_read16(t, "stat len", &st->size, src, len) ||
898 4e7aba8f 2022-02-02 op !np_read16(t, "stat.size", &st->size, src, len) ||
899 9029ac6f 2022-01-19 op !np_read16(t, "stat.type", &st->type, src, len) ||
900 755d75bf 2022-01-17 op !np_read32(t, "stat.dev", &st->dev, src, len) ||
901 755d75bf 2022-01-17 op !np_read64(t, "stat.qid.path", &st->qid.path, src, len) ||
902 755d75bf 2022-01-17 op !np_read32(t, "stat.qid.vers", &st->qid.vers, src, len) ||
903 755d75bf 2022-01-17 op !np_read8(t, "stat.qid.type", &st->qid.type, src, len) ||
904 755d75bf 2022-01-17 op !np_read32(t, "stat.mode", &st->mode, src, len) ||
905 755d75bf 2022-01-17 op !np_read32(t, "stat.atime", &st->atime, src, len) ||
906 755d75bf 2022-01-17 op !np_read32(t, "stat.mtime", &st->mtime, src, len) ||
907 755d75bf 2022-01-17 op !np_read64(t, "stat.length", &st->length, src, len))
908 755d75bf 2022-01-17 op return READSTRERR;
909 755d75bf 2022-01-17 op
910 755d75bf 2022-01-17 op /*
911 755d75bf 2022-01-17 op * ignore everything but the name, we don't support
912 755d75bf 2022-01-17 op * changing those fields anyway
913 755d75bf 2022-01-17 op */
914 755d75bf 2022-01-17 op st->name = name;
915 755d75bf 2022-01-17 op return np_readstr(t, "stat.name", name, namelen, src, len);
916 fb1a36c0 2022-01-09 op }
917 fb1a36c0 2022-01-09 op
918 fb1a36c0 2022-01-09 op static void
919 fb1a36c0 2022-01-09 op tversion(struct np_msg_header *hdr, const uint8_t *data, size_t len)
920 fb1a36c0 2022-01-09 op {
921 fb1a36c0 2022-01-09 op char *dot, version[32];
922 fb1a36c0 2022-01-09 op
923 fb1a36c0 2022-01-09 op if (handshaked)
924 fb1a36c0 2022-01-09 op goto err;
925 fb1a36c0 2022-01-09 op
926 fb1a36c0 2022-01-09 op /* msize[4] version[s] */
927 fb1a36c0 2022-01-09 op if (!NPREAD32("msize", &msize, &data, &len))
928 fb1a36c0 2022-01-09 op goto err;
929 fb1a36c0 2022-01-09 op
930 fb1a36c0 2022-01-09 op switch (NPREADSTR("version", version, sizeof(version), &data, &len)) {
931 fb1a36c0 2022-01-09 op case READSTRERR:
932 fb1a36c0 2022-01-09 op goto err;
933 fb1a36c0 2022-01-09 op case READSTRTRUNC:
934 fb1a36c0 2022-01-09 op log_warnx("9P version string too long, truncated");
935 6be59daa 2022-05-22 op np_version(hdr->tag, MSIZE9P, "unknown");
936 6be59daa 2022-05-22 op return;
937 fb1a36c0 2022-01-09 op }
938 fb1a36c0 2022-01-09 op
939 bcdb46a7 2022-01-19 op if (len != 0)
940 bcdb46a7 2022-01-19 op goto err;
941 bcdb46a7 2022-01-19 op
942 fb1a36c0 2022-01-09 op if ((dot = strchr(version, '.')) != NULL)
943 fb1a36c0 2022-01-09 op *dot = '\0';
944 fb1a36c0 2022-01-09 op
945 6be59daa 2022-05-22 op if (strcmp(version, VERSION9P) != 0) {
946 6be59daa 2022-05-22 op log_warnx("unknown 9P version \"%s\"; want "VERSION9P,
947 6be59daa 2022-05-22 op version);
948 6be59daa 2022-05-22 op np_version(hdr->tag, MSIZE9P, "unknown");
949 6be59daa 2022-05-22 op return;
950 6be59daa 2022-05-22 op }
951 6be59daa 2022-05-22 op
952 6be59daa 2022-05-22 op if (msize < MIN_MSIZE) {
953 6be59daa 2022-05-22 op log_warnx("msize too small: %"PRIu32"; want %d at least",
954 6be59daa 2022-05-22 op msize, MIN_MSIZE);
955 6be59daa 2022-05-22 op np_version(hdr->tag, MSIZE9P, "unknown");
956 6be59daa 2022-05-22 op return;
957 6be59daa 2022-05-22 op }
958 fb1a36c0 2022-01-09 op
959 fb1a36c0 2022-01-09 op /* version matched */
960 fb1a36c0 2022-01-09 op handshaked = 1;
961 8b0a5062 2022-05-22 op msize = MIN(msize, MSIZE9P);
962 fb1a36c0 2022-01-09 op client_send_listener(IMSG_MSIZE, &msize, sizeof(msize));
963 fb1a36c0 2022-01-09 op np_version(hdr->tag, msize, VERSION9P);
964 fb1a36c0 2022-01-09 op return;
965 fb1a36c0 2022-01-09 op
966 fb1a36c0 2022-01-09 op err:
967 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
968 fb1a36c0 2022-01-09 op client_shutdown();
969 fb1a36c0 2022-01-09 op }
970 fb1a36c0 2022-01-09 op
971 fb1a36c0 2022-01-09 op static void
972 fb1a36c0 2022-01-09 op tattach(struct np_msg_header *hdr, const uint8_t *data, size_t len)
973 fb1a36c0 2022-01-09 op {
974 fb1a36c0 2022-01-09 op struct dir *dir;
975 fb1a36c0 2022-01-09 op struct fid *f;
976 fb1a36c0 2022-01-09 op uint32_t fid, afid;
977 fb1a36c0 2022-01-09 op int fd;
978 fb1a36c0 2022-01-09 op char aname[PATH_MAX];
979 fb1a36c0 2022-01-09 op
980 fb1a36c0 2022-01-09 op /* fid[4] afid[4] uname[s] aname[s] */
981 fb1a36c0 2022-01-09 op
982 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
983 fb1a36c0 2022-01-09 op !NPREAD32("afid", &afid, &data, &len))
984 fb1a36c0 2022-01-09 op goto err;
985 fb1a36c0 2022-01-09 op
986 fb1a36c0 2022-01-09 op /* read the uname but don't actually use it */
987 fb1a36c0 2022-01-09 op switch (NPREADSTR("uname", aname, sizeof(aname), &data, &len)) {
988 fb1a36c0 2022-01-09 op case READSTRERR:
989 fb1a36c0 2022-01-09 op goto err;
990 fb1a36c0 2022-01-09 op case READSTRTRUNC:
991 fb1a36c0 2022-01-09 op np_error(hdr->tag, "name too long");
992 fb1a36c0 2022-01-09 op return;
993 fb1a36c0 2022-01-09 op }
994 fb1a36c0 2022-01-09 op
995 fb1a36c0 2022-01-09 op switch (NPREADSTR("aname", aname, sizeof(aname), &data, &len)) {
996 fb1a36c0 2022-01-09 op case READSTRERR:
997 fb1a36c0 2022-01-09 op goto err;
998 fb1a36c0 2022-01-09 op case READSTRTRUNC:
999 fb1a36c0 2022-01-09 op np_error(hdr->tag, "name too long");
1000 fb1a36c0 2022-01-09 op return;
1001 fb1a36c0 2022-01-09 op }
1002 fb1a36c0 2022-01-09 op
1003 73df5ca2 2022-02-02 op if (*aname == '\0')
1004 73df5ca2 2022-02-02 op strlcpy(aname, "/", sizeof(aname));
1005 73df5ca2 2022-02-02 op
1006 bcdb46a7 2022-01-19 op if (len != 0)
1007 bcdb46a7 2022-01-19 op goto err;
1008 bcdb46a7 2022-01-19 op
1009 fb1a36c0 2022-01-09 op if (fid_by_id(fid) != NULL || afid != NOFID) {
1010 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid or afid");
1011 fb1a36c0 2022-01-09 op return;
1012 fb1a36c0 2022-01-09 op }
1013 fb1a36c0 2022-01-09 op
1014 fb1a36c0 2022-01-09 op if ((fd = open(aname, O_RDONLY|O_DIRECTORY)) == -1)
1015 fb1a36c0 2022-01-09 op goto fail;
1016 fb1a36c0 2022-01-09 op
1017 fb1a36c0 2022-01-09 op if ((dir = new_dir(fd)) == NULL)
1018 fb1a36c0 2022-01-09 op goto fail;
1019 fb1a36c0 2022-01-09 op
1020 fb1a36c0 2022-01-09 op log_debug("attached %s to %d", aname, fid);
1021 fb1a36c0 2022-01-09 op
1022 fb1a36c0 2022-01-09 op if ((f = new_fid(dir, fid, aname, NULL)) == NULL) {
1023 fb1a36c0 2022-01-09 op dir_decref(dir);
1024 fb1a36c0 2022-01-09 op goto fail;
1025 fb1a36c0 2022-01-09 op }
1026 fb1a36c0 2022-01-09 op
1027 fb1a36c0 2022-01-09 op np_attach(hdr->tag, &f->qid);
1028 fb1a36c0 2022-01-09 op return;
1029 fb1a36c0 2022-01-09 op
1030 fb1a36c0 2022-01-09 op fail:
1031 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1032 fb1a36c0 2022-01-09 op log_warn("failed to attach %s", aname);
1033 fb1a36c0 2022-01-09 op return;
1034 fb1a36c0 2022-01-09 op
1035 fb1a36c0 2022-01-09 op err:
1036 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1037 fb1a36c0 2022-01-09 op client_shutdown();
1038 fb1a36c0 2022-01-09 op }
1039 fb1a36c0 2022-01-09 op
1040 fb1a36c0 2022-01-09 op static void
1041 fb1a36c0 2022-01-09 op tclunk(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1042 fb1a36c0 2022-01-09 op {
1043 fb1a36c0 2022-01-09 op struct fid *f;
1044 fb1a36c0 2022-01-09 op uint32_t fid;
1045 fb1a36c0 2022-01-09 op
1046 fb1a36c0 2022-01-09 op /* fid[4] */
1047 bcdb46a7 2022-01-19 op if (!NPREAD32("fid", &fid, &data, &len) ||
1048 bcdb46a7 2022-01-19 op len != 0) {
1049 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1050 fb1a36c0 2022-01-09 op client_shutdown();
1051 fb1a36c0 2022-01-09 op return;
1052 fb1a36c0 2022-01-09 op }
1053 fb1a36c0 2022-01-09 op
1054 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL) {
1055 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1056 fb1a36c0 2022-01-09 op return;
1057 fb1a36c0 2022-01-09 op }
1058 fb1a36c0 2022-01-09 op
1059 fb1a36c0 2022-01-09 op free_fid(f);
1060 fb1a36c0 2022-01-09 op np_clunk(hdr->tag);
1061 fb1a36c0 2022-01-09 op }
1062 fb1a36c0 2022-01-09 op
1063 fb1a36c0 2022-01-09 op static void
1064 fb1a36c0 2022-01-09 op tflush(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1065 fb1a36c0 2022-01-09 op {
1066 fb1a36c0 2022-01-09 op uint16_t oldtag;
1067 fb1a36c0 2022-01-09 op
1068 fb1a36c0 2022-01-09 op /*
1069 fb1a36c0 2022-01-09 op * We're doing only synchronous I/O. Tflush is implemented
1070 fb1a36c0 2022-01-09 op * only because it's illegal to reply with a Rerror.
1071 fb1a36c0 2022-01-09 op */
1072 fb1a36c0 2022-01-09 op
1073 fb1a36c0 2022-01-09 op /* oldtag[2] */
1074 fb1a36c0 2022-01-09 op if (len != sizeof(oldtag)) {
1075 fb1a36c0 2022-01-09 op log_warnx("Tflush with the wrong size: got %zu want %zu",
1076 fb1a36c0 2022-01-09 op len, sizeof(oldtag));
1077 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1078 fb1a36c0 2022-01-09 op client_shutdown();
1079 fb1a36c0 2022-01-09 op return;
1080 fb1a36c0 2022-01-09 op }
1081 fb1a36c0 2022-01-09 op
1082 fb1a36c0 2022-01-09 op np_flush(hdr->tag);
1083 fb1a36c0 2022-01-09 op }
1084 fb1a36c0 2022-01-09 op
1085 fb1a36c0 2022-01-09 op static void
1086 fb1a36c0 2022-01-09 op twalk(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1087 fb1a36c0 2022-01-09 op {
1088 fb1a36c0 2022-01-09 op struct stat sb;
1089 fb1a36c0 2022-01-09 op struct dir *dir;
1090 fb1a36c0 2022-01-09 op struct qid wqid[MAXWELEM] = {0};
1091 fb1a36c0 2022-01-09 op struct fid *f, *nf;
1092 fb1a36c0 2022-01-09 op uint32_t fid, newfid;
1093 fb1a36c0 2022-01-09 op uint16_t nwname;
1094 fb1a36c0 2022-01-09 op int fd, oldfd, no, nwqid = 0;
1095 fb1a36c0 2022-01-09 op char wnam[PATH_MAX];
1096 fb1a36c0 2022-01-09 op
1097 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
1098 fb1a36c0 2022-01-09 op !NPREAD32("newfid", &newfid, &data, &len) ||
1099 fb1a36c0 2022-01-09 op !NPREAD16("nwname", &nwname, &data, &len))
1100 fb1a36c0 2022-01-09 op goto err;
1101 fb1a36c0 2022-01-09 op
1102 fb1a36c0 2022-01-09 op if (nwname > MAXWELEM) {
1103 fb1a36c0 2022-01-09 op log_warnx("Twalk: more than %d path elements: %d",
1104 fb1a36c0 2022-01-09 op MAXWELEM, nwname);
1105 fb1a36c0 2022-01-09 op goto err;
1106 fb1a36c0 2022-01-09 op }
1107 fb1a36c0 2022-01-09 op
1108 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL) {
1109 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1110 fb1a36c0 2022-01-09 op return;
1111 fb1a36c0 2022-01-09 op }
1112 fb1a36c0 2022-01-09 op
1113 fb1a36c0 2022-01-09 op if (f->fd != -1) {
1114 fb1a36c0 2022-01-09 op np_error(hdr->tag, "fid already opened for I/O");
1115 fb1a36c0 2022-01-09 op return;
1116 fb1a36c0 2022-01-09 op }
1117 fb1a36c0 2022-01-09 op
1118 fb1a36c0 2022-01-09 op if (fid == newfid)
1119 fb1a36c0 2022-01-09 op nf = f;
1120 fb1a36c0 2022-01-09 op else if ((nf = fid_by_id(newfid)) != NULL) {
1121 fb1a36c0 2022-01-09 op np_error(hdr->tag, "newfid already in use");
1122 fb1a36c0 2022-01-09 op return;
1123 fb1a36c0 2022-01-09 op } else
1124 fb1a36c0 2022-01-09 op nf = NULL;
1125 fb1a36c0 2022-01-09 op
1126 fb1a36c0 2022-01-09 op /* special case: fid duplication */
1127 fb1a36c0 2022-01-09 op if (nwname == 0) {
1128 fb1a36c0 2022-01-09 op /*
1129 fb1a36c0 2022-01-09 op * TODO: should we forbid fids duplication when fid ==
1130 fb1a36c0 2022-01-09 op * newfid?
1131 fb1a36c0 2022-01-09 op */
1132 fb1a36c0 2022-01-09 op if (nf == NULL &&
1133 543fc013 2022-01-17 op (nf = new_fid(f->dir, newfid, f->fname, &f->qid)) == NULL)
1134 fb1a36c0 2022-01-09 op fatal("new_fid duplication");
1135 fb1a36c0 2022-01-09 op
1136 fb1a36c0 2022-01-09 op np_walk(hdr->tag, 0, NULL);
1137 fb1a36c0 2022-01-09 op return;
1138 fb1a36c0 2022-01-09 op }
1139 fb1a36c0 2022-01-09 op
1140 fb1a36c0 2022-01-09 op if (!(f->qid.type & QTDIR)) {
1141 fb1a36c0 2022-01-09 op np_error(hdr->tag, "fid doesn't represent a directory");
1142 fb1a36c0 2022-01-09 op return;
1143 fb1a36c0 2022-01-09 op }
1144 fb1a36c0 2022-01-09 op
1145 fb1a36c0 2022-01-09 op oldfd = f->dir->fd;
1146 fb1a36c0 2022-01-09 op
1147 fb1a36c0 2022-01-09 op for (nwqid = 0; nwqid < nwname; nwqid++) {
1148 fb1a36c0 2022-01-09 op switch (NPREADSTR("wname", wnam, sizeof(wnam), &data, &len)) {
1149 fb1a36c0 2022-01-09 op case READSTRERR:
1150 fb1a36c0 2022-01-09 op goto err;
1151 fb1a36c0 2022-01-09 op case READSTRTRUNC:
1152 fb1a36c0 2022-01-09 op np_error(hdr->tag, "wname too long");
1153 fb1a36c0 2022-01-09 op return;
1154 fb1a36c0 2022-01-09 op }
1155 fb1a36c0 2022-01-09 op
1156 fb1a36c0 2022-01-09 op if (*wnam == '\0' ||
1157 fb1a36c0 2022-01-09 op strchr(wnam, '/') != NULL ||
1158 fb1a36c0 2022-01-09 op !strcmp(wnam, ".")) {
1159 fb1a36c0 2022-01-09 op errno = EINVAL;
1160 fb1a36c0 2022-01-09 op goto cantopen;
1161 fb1a36c0 2022-01-09 op }
1162 fb1a36c0 2022-01-09 op
1163 fb1a36c0 2022-01-09 op if ((fd = openat(oldfd, wnam, O_RDONLY|O_DIRECTORY)) == -1 &&
1164 fb1a36c0 2022-01-09 op errno != ENOTDIR)
1165 fb1a36c0 2022-01-09 op goto cantopen;
1166 fb1a36c0 2022-01-09 op
1167 fb1a36c0 2022-01-09 op if ((fd == -1 && fstatat(oldfd, wnam, &sb, 0) == -1) ||
1168 fb1a36c0 2022-01-09 op (fd != -1 && fstat(fd, &sb) == -1))
1169 fb1a36c0 2022-01-09 op goto cantopen;
1170 fb1a36c0 2022-01-09 op
1171 fb1a36c0 2022-01-09 op qid_update_from_sb(&wqid[nwqid], &sb);
1172 fb1a36c0 2022-01-09 op
1173 fb1a36c0 2022-01-09 op /* reached a file but we still have other components */
1174 fb1a36c0 2022-01-09 op if (fd == -1 && nwqid+1 < nwname)
1175 fb1a36c0 2022-01-09 op goto cantopen;
1176 fb1a36c0 2022-01-09 op
1177 fb1a36c0 2022-01-09 op /* reached the end and found a file */
1178 fb1a36c0 2022-01-09 op if (fd == -1 && nwqid+1 == nwname)
1179 fb1a36c0 2022-01-09 op continue;
1180 fb1a36c0 2022-01-09 op
1181 fb1a36c0 2022-01-09 op if (oldfd != f->dir->fd)
1182 fb1a36c0 2022-01-09 op close(oldfd);
1183 fb1a36c0 2022-01-09 op oldfd = fd;
1184 fb1a36c0 2022-01-09 op }
1185 fb1a36c0 2022-01-09 op
1186 bcdb46a7 2022-01-19 op if (len != 0)
1187 bcdb46a7 2022-01-19 op goto err;
1188 bcdb46a7 2022-01-19 op
1189 fb1a36c0 2022-01-09 op /*
1190 fb1a36c0 2022-01-09 op * If fd is -1 we've reached a file, otherwise we've just
1191 fb1a36c0 2022-01-09 op * reached another directory. We must pay attention to what
1192 fb1a36c0 2022-01-09 op * file descriptor we use to create the dir, because if we've
1193 fb1a36c0 2022-01-09 op * reached a file and oldfd is f->dir->fd then we *must* share
1194 fb1a36c0 2022-01-09 op * the same dir (it was a walk of one path from a directory to a
1195 fb1a36c0 2022-01-09 op * file, otherwise fun is bound to happen as soon as the client
1196 fb1a36c0 2022-01-09 op * closes the fid for the directory but keeps the one for the
1197 fb1a36c0 2022-01-09 op * file.
1198 fb1a36c0 2022-01-09 op */
1199 fb1a36c0 2022-01-09 op if (fd == -1 && oldfd == f->dir->fd)
1200 fb1a36c0 2022-01-09 op dir = f->dir;
1201 fb1a36c0 2022-01-09 op else if (fd == -1)
1202 fb1a36c0 2022-01-09 op dir = new_dir(oldfd);
1203 fb1a36c0 2022-01-09 op else
1204 fb1a36c0 2022-01-09 op dir = new_dir(fd);
1205 fb1a36c0 2022-01-09 op
1206 fb1a36c0 2022-01-09 op if (dir == NULL)
1207 fb1a36c0 2022-01-09 op fatal("new_dir");
1208 fb1a36c0 2022-01-09 op
1209 fb1a36c0 2022-01-09 op if (nf == NULL) {
1210 fb1a36c0 2022-01-09 op if ((nf = new_fid(dir, newfid, wnam, &wqid[nwqid-1])) == NULL)
1211 fb1a36c0 2022-01-09 op fatal("new fid");
1212 fb1a36c0 2022-01-09 op } else {
1213 fb1a36c0 2022-01-09 op /* update the dir */
1214 fb1a36c0 2022-01-09 op dir_decref(nf->dir);
1215 fb1a36c0 2022-01-09 op nf->dir = dir_incref(dir);
1216 fb1a36c0 2022-01-09 op }
1217 fb1a36c0 2022-01-09 op
1218 fb1a36c0 2022-01-09 op np_walk(hdr->tag, nwqid, wqid);
1219 fb1a36c0 2022-01-09 op return;
1220 fb1a36c0 2022-01-09 op
1221 fb1a36c0 2022-01-09 op cantopen:
1222 fb1a36c0 2022-01-09 op if (oldfd != f->dir->fd)
1223 fb1a36c0 2022-01-09 op close(oldfd);
1224 fb1a36c0 2022-01-09 op no = errno;
1225 fb1a36c0 2022-01-09 op if (nwqid == 0)
1226 fb1a36c0 2022-01-09 op np_error(hdr->tag, strerror(no));
1227 fb1a36c0 2022-01-09 op else
1228 fb1a36c0 2022-01-09 op np_walk(hdr->tag, nwqid, wqid);
1229 fb1a36c0 2022-01-09 op return;
1230 fb1a36c0 2022-01-09 op
1231 fb1a36c0 2022-01-09 op err:
1232 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1233 fb1a36c0 2022-01-09 op client_shutdown();
1234 fb1a36c0 2022-01-09 op }
1235 fb1a36c0 2022-01-09 op
1236 fb1a36c0 2022-01-09 op static inline int
1237 fb1a36c0 2022-01-09 op npmode_to_unix(uint8_t mode, int *flags)
1238 fb1a36c0 2022-01-09 op {
1239 fb1a36c0 2022-01-09 op switch (mode & 0x0F) {
1240 fb1a36c0 2022-01-09 op case KOREAD:
1241 fb1a36c0 2022-01-09 op *flags = O_RDONLY;
1242 fb1a36c0 2022-01-09 op break;
1243 fb1a36c0 2022-01-09 op case KOWRITE:
1244 fb1a36c0 2022-01-09 op *flags = O_WRONLY;
1245 fb1a36c0 2022-01-09 op break;
1246 fb1a36c0 2022-01-09 op case KORDWR:
1247 fb1a36c0 2022-01-09 op *flags = O_RDWR;
1248 fb1a36c0 2022-01-09 op break;
1249 fb1a36c0 2022-01-09 op case KOEXEC:
1250 fb1a36c0 2022-01-09 op log_warnx("tried to open something with KOEXEC");
1251 fb1a36c0 2022-01-09 op /* fallthrough */
1252 fb1a36c0 2022-01-09 op default:
1253 fb1a36c0 2022-01-09 op return -1;
1254 fb1a36c0 2022-01-09 op }
1255 fb1a36c0 2022-01-09 op
1256 fb1a36c0 2022-01-09 op if (mode & KOTRUNC)
1257 fb1a36c0 2022-01-09 op *flags |= O_TRUNC;
1258 fb1a36c0 2022-01-09 op if (mode & KORCLOSE)
1259 fb1a36c0 2022-01-09 op *flags |= O_CLOEXEC;
1260 fb1a36c0 2022-01-09 op
1261 fb1a36c0 2022-01-09 op return 0;
1262 fb1a36c0 2022-01-09 op }
1263 fb1a36c0 2022-01-09 op
1264 fb1a36c0 2022-01-09 op static void
1265 fb1a36c0 2022-01-09 op topen(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1266 fb1a36c0 2022-01-09 op {
1267 fb1a36c0 2022-01-09 op struct stat sb;
1268 fb1a36c0 2022-01-09 op struct qid qid;
1269 fb1a36c0 2022-01-09 op struct fid *f;
1270 fb1a36c0 2022-01-09 op uint32_t fid;
1271 fb1a36c0 2022-01-09 op uint8_t mode;
1272 fb1a36c0 2022-01-09 op const char *path;
1273 fb1a36c0 2022-01-09 op
1274 fb1a36c0 2022-01-09 op /* fid[4] mode[1] */
1275 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
1276 bcdb46a7 2022-01-19 op !NPREAD8("mode", &mode, &data, &len) ||
1277 bcdb46a7 2022-01-19 op len != 0) {
1278 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1279 fb1a36c0 2022-01-09 op client_shutdown();
1280 fb1a36c0 2022-01-09 op return;
1281 fb1a36c0 2022-01-09 op }
1282 fb1a36c0 2022-01-09 op
1283 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL || f->fd != -1) {
1284 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1285 fb1a36c0 2022-01-09 op return;
1286 fb1a36c0 2022-01-09 op }
1287 fb1a36c0 2022-01-09 op
1288 fb1a36c0 2022-01-09 op if (npmode_to_unix(mode, &f->iomode) == -1) {
1289 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid mode");
1290 fb1a36c0 2022-01-09 op return;
1291 fb1a36c0 2022-01-09 op }
1292 fb1a36c0 2022-01-09 op
1293 543fc013 2022-01-17 op path = f->fname;
1294 fb1a36c0 2022-01-09 op if (f->qid.type & QTDIR)
1295 fb1a36c0 2022-01-09 op path = ".";
1296 fb1a36c0 2022-01-09 op
1297 fb1a36c0 2022-01-09 op if ((f->fd = openat(f->dir->fd, path, f->iomode)) == -1) {
1298 fb1a36c0 2022-01-09 op np_error(hdr->tag, strerror(errno));
1299 fb1a36c0 2022-01-09 op return;
1300 fb1a36c0 2022-01-09 op }
1301 fb1a36c0 2022-01-09 op
1302 fb1a36c0 2022-01-09 op if (fstat(f->fd, &sb) == -1)
1303 fb1a36c0 2022-01-09 op fatal("fstat");
1304 fb1a36c0 2022-01-09 op
1305 fb1a36c0 2022-01-09 op if (S_ISDIR(sb.st_mode)) {
1306 fb1a36c0 2022-01-09 op if ((f->d = fdopendir(f->fd)) == NULL) {
1307 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1308 fb1a36c0 2022-01-09 op close(f->fd);
1309 fb1a36c0 2022-01-09 op f->fd = -1;
1310 fb1a36c0 2022-01-09 op return;
1311 fb1a36c0 2022-01-09 op }
1312 fb1a36c0 2022-01-09 op
1313 fb1a36c0 2022-01-09 op if ((f->evb = evbuffer_new()) == NULL) {
1314 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1315 fb1a36c0 2022-01-09 op closedir(f->d);
1316 fb1a36c0 2022-01-09 op f->d = NULL;
1317 fb1a36c0 2022-01-09 op f->fd = -1;
1318 fb1a36c0 2022-01-09 op }
1319 fb1a36c0 2022-01-09 op }
1320 fb1a36c0 2022-01-09 op
1321 fb1a36c0 2022-01-09 op f->offset = 0;
1322 fb1a36c0 2022-01-09 op
1323 fb1a36c0 2022-01-09 op qid_update_from_sb(&qid, &sb);
1324 fb1a36c0 2022-01-09 op np_open(hdr->tag, &qid, sb.st_blksize);
1325 fb1a36c0 2022-01-09 op }
1326 fb1a36c0 2022-01-09 op
1327 fb1a36c0 2022-01-09 op static void
1328 fb1a36c0 2022-01-09 op tcreate(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1329 fb1a36c0 2022-01-09 op {
1330 fb1a36c0 2022-01-09 op struct stat sb;
1331 fb1a36c0 2022-01-09 op struct qid qid;
1332 fb1a36c0 2022-01-09 op struct fid *f;
1333 fb1a36c0 2022-01-09 op uint32_t fid, perm;
1334 fb1a36c0 2022-01-09 op uint8_t mode;
1335 fb1a36c0 2022-01-09 op char name[PATH_MAX];
1336 fb1a36c0 2022-01-09 op
1337 fb1a36c0 2022-01-09 op /* fid[4] name[s] perm[4] mode[1] */
1338 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len))
1339 fb1a36c0 2022-01-09 op goto err;
1340 fb1a36c0 2022-01-09 op switch (NPREADSTR("name", name, sizeof(name), &data, &len)) {
1341 fb1a36c0 2022-01-09 op case READSTRERR:
1342 fb1a36c0 2022-01-09 op goto err;
1343 fb1a36c0 2022-01-09 op case READSTRTRUNC:
1344 fb1a36c0 2022-01-09 op np_error(hdr->tag, "name too long");
1345 fb1a36c0 2022-01-09 op return;
1346 fb1a36c0 2022-01-09 op }
1347 fb1a36c0 2022-01-09 op if (!NPREAD32("perm", &perm, &data, &len) ||
1348 bcdb46a7 2022-01-19 op !NPREAD8("mode", &mode, &data, &len) ||
1349 bcdb46a7 2022-01-19 op len != 0)
1350 fb1a36c0 2022-01-09 op goto err;
1351 fb1a36c0 2022-01-09 op
1352 fb1a36c0 2022-01-09 op if (!strcmp(name, ".") || !strcmp(name, "..") ||
1353 fb1a36c0 2022-01-09 op strchr(name, '/') != NULL) {
1354 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid name");
1355 fb1a36c0 2022-01-09 op return;
1356 fb1a36c0 2022-01-09 op }
1357 fb1a36c0 2022-01-09 op
1358 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL || f->fd != -1) {
1359 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1360 fb1a36c0 2022-01-09 op return;
1361 fb1a36c0 2022-01-09 op }
1362 fb1a36c0 2022-01-09 op
1363 fb1a36c0 2022-01-09 op if (!(f->qid.type & QTDIR)) {
1364 fb1a36c0 2022-01-09 op np_error(hdr->tag, "fid doesn't identify a directory");
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 if (npmode_to_unix(mode, &f->iomode) == -1) {
1369 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid mode");
1370 fb1a36c0 2022-01-09 op return;
1371 fb1a36c0 2022-01-09 op }
1372 fb1a36c0 2022-01-09 op
1373 fb1a36c0 2022-01-09 op if (f->iomode & O_RDONLY) {
1374 fb1a36c0 2022-01-09 op np_error(hdr->tag, "can't create a read-only file");
1375 fb1a36c0 2022-01-09 op return;
1376 fb1a36c0 2022-01-09 op }
1377 fb1a36c0 2022-01-09 op
1378 fb1a36c0 2022-01-09 op /* TODO: parse the mode */
1379 fb1a36c0 2022-01-09 op
1380 fb1a36c0 2022-01-09 op if (perm & 0x80000000) {
1381 fb1a36c0 2022-01-09 op /* create a directory */
1382 fb1a36c0 2022-01-09 op f->fd = mkdirat(f->dir->fd, name, 0755);
1383 fb1a36c0 2022-01-09 op } else {
1384 fb1a36c0 2022-01-09 op /* create a file */
1385 fb1a36c0 2022-01-09 op f->fd = openat(f->dir->fd, name, f->iomode | O_CREAT | O_TRUNC,
1386 fb1a36c0 2022-01-09 op 0644);
1387 fb1a36c0 2022-01-09 op }
1388 fb1a36c0 2022-01-09 op
1389 fb1a36c0 2022-01-09 op if (f->fd == -1) {
1390 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1391 fb1a36c0 2022-01-09 op return;
1392 fb1a36c0 2022-01-09 op }
1393 fb1a36c0 2022-01-09 op
1394 fb1a36c0 2022-01-09 op if (fstat(f->fd, &sb) == -1)
1395 fb1a36c0 2022-01-09 op fatal("fstat");
1396 fb1a36c0 2022-01-09 op
1397 fb1a36c0 2022-01-09 op if (S_ISDIR(sb.st_mode)) {
1398 fb1a36c0 2022-01-09 op if ((f->d = fdopendir(f->fd)) == NULL) {
1399 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1400 fb1a36c0 2022-01-09 op close(f->fd);
1401 fb1a36c0 2022-01-09 op f->fd = -1;
1402 fb1a36c0 2022-01-09 op return;
1403 fb1a36c0 2022-01-09 op }
1404 fb1a36c0 2022-01-09 op
1405 fb1a36c0 2022-01-09 op if ((f->evb = evbuffer_new()) == NULL) {
1406 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1407 fb1a36c0 2022-01-09 op closedir(f->d);
1408 fb1a36c0 2022-01-09 op f->d = NULL;
1409 fb1a36c0 2022-01-09 op f->fd = -1;
1410 fb1a36c0 2022-01-09 op }
1411 fb1a36c0 2022-01-09 op }
1412 fb1a36c0 2022-01-09 op
1413 fb1a36c0 2022-01-09 op f->offset = 0;
1414 fb1a36c0 2022-01-09 op
1415 fb1a36c0 2022-01-09 op qid_update_from_sb(&qid, &sb);
1416 fb1a36c0 2022-01-09 op np_create(hdr->tag, &qid, sb.st_blksize);
1417 fb1a36c0 2022-01-09 op
1418 fb1a36c0 2022-01-09 op return;
1419 fb1a36c0 2022-01-09 op
1420 fb1a36c0 2022-01-09 op err:
1421 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1422 fb1a36c0 2022-01-09 op client_shutdown();
1423 fb1a36c0 2022-01-09 op }
1424 fb1a36c0 2022-01-09 op
1425 91f48e25 2022-02-02 op static inline int
1426 fb1a36c0 2022-01-09 op serialize_stat(const char *fname, struct stat *sb, struct evbuffer *evb)
1427 fb1a36c0 2022-01-09 op {
1428 fb1a36c0 2022-01-09 op struct qid qid;
1429 fb1a36c0 2022-01-09 op const char *uid, *gid, *muid;
1430 fb1a36c0 2022-01-09 op size_t tot;
1431 bee0a9e3 2022-02-02 op uint32_t mode;
1432 fb1a36c0 2022-01-09 op uint16_t namlen, uidlen, gidlen, ulen;
1433 fb1a36c0 2022-01-09 op
1434 fb1a36c0 2022-01-09 op qid_update_from_sb(&qid, sb);
1435 fb1a36c0 2022-01-09 op
1436 fb1a36c0 2022-01-09 op /* TODO: fill these fields */
1437 fb1a36c0 2022-01-09 op uid = "";
1438 fb1a36c0 2022-01-09 op gid = "";
1439 fb1a36c0 2022-01-09 op muid = "";
1440 fb1a36c0 2022-01-09 op
1441 fb1a36c0 2022-01-09 op namlen = strlen(fname);
1442 fb1a36c0 2022-01-09 op uidlen = strlen(uid);
1443 fb1a36c0 2022-01-09 op gidlen = strlen(gid);
1444 fb1a36c0 2022-01-09 op ulen = strlen(muid);
1445 fb1a36c0 2022-01-09 op
1446 fb1a36c0 2022-01-09 op tot = NPSTATSIZ(namlen, uidlen, gidlen, ulen);
1447 91f48e25 2022-02-02 op if (tot > UINT16_MAX) {
1448 fb1a36c0 2022-01-09 op log_warnx("stat info for dir entry %s would overflow",
1449 fb1a36c0 2022-01-09 op fname);
1450 91f48e25 2022-02-02 op return -1;
1451 fb1a36c0 2022-01-09 op }
1452 bee0a9e3 2022-02-02 op
1453 bee0a9e3 2022-02-02 op mode = sb->st_mode & 0xFFFF;
1454 bee0a9e3 2022-02-02 op if (qid.type & QTDIR)
1455 bee0a9e3 2022-02-02 op mode |= 0x80000000;
1456 fb1a36c0 2022-01-09 op
1457 fb1a36c0 2022-01-09 op np_write16(evb, tot); /* size[2] */
1458 fb1a36c0 2022-01-09 op np_write16(evb, sb->st_rdev); /* type[2] */
1459 fb1a36c0 2022-01-09 op np_write32(evb, sb->st_dev); /* dev[4] */
1460 fb1a36c0 2022-01-09 op np_qid(evb, &qid); /* qid[13] */
1461 bee0a9e3 2022-02-02 op np_write32(evb, mode); /* mode[4] */
1462 fb1a36c0 2022-01-09 op np_write32(evb, sb->st_atim.tv_sec); /* atime[4] */
1463 fb1a36c0 2022-01-09 op np_write32(evb, sb->st_mtim.tv_sec); /* mtime[4] */
1464 68a7ef68 2022-01-16 op
1465 68a7ef68 2022-01-16 op /* special case: directories have size 0 */
1466 68a7ef68 2022-01-16 op if (qid.type & QTDIR)
1467 68a7ef68 2022-01-16 op np_write64(evb, 0);
1468 68a7ef68 2022-01-16 op else
1469 68a7ef68 2022-01-16 op np_write64(evb, sb->st_size); /* length[8] */
1470 68a7ef68 2022-01-16 op
1471 fb1a36c0 2022-01-09 op np_string(evb, namlen, fname); /* name[s] */
1472 fb1a36c0 2022-01-09 op np_string(evb, uidlen, uid); /* uid[s] */
1473 fb1a36c0 2022-01-09 op np_string(evb, gidlen, gid); /* gid[s] */
1474 fb1a36c0 2022-01-09 op np_string(evb, ulen, muid); /* muid[s] */
1475 91f48e25 2022-02-02 op
1476 91f48e25 2022-02-02 op return 0;
1477 fb1a36c0 2022-01-09 op }
1478 fb1a36c0 2022-01-09 op
1479 fb1a36c0 2022-01-09 op static void
1480 fb1a36c0 2022-01-09 op tread(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1481 fb1a36c0 2022-01-09 op {
1482 fb1a36c0 2022-01-09 op struct fid *f;
1483 fb1a36c0 2022-01-09 op ssize_t r;
1484 fb1a36c0 2022-01-09 op size_t howmuch;
1485 fb1a36c0 2022-01-09 op uint64_t off;
1486 fb1a36c0 2022-01-09 op uint32_t fid, count;
1487 fb1a36c0 2022-01-09 op char buf[2048];
1488 fb1a36c0 2022-01-09 op
1489 fb1a36c0 2022-01-09 op /* fid[4] offset[8] count[4] */
1490 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
1491 fb1a36c0 2022-01-09 op !NPREAD64("offset", &off, &data, &len) ||
1492 bcdb46a7 2022-01-19 op !NPREAD32("count", &count, &data, &len) ||
1493 bcdb46a7 2022-01-19 op len != 0) {
1494 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1495 fb1a36c0 2022-01-09 op client_shutdown();
1496 fb1a36c0 2022-01-09 op return;
1497 fb1a36c0 2022-01-09 op }
1498 fb1a36c0 2022-01-09 op
1499 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL || f->fd == -1) {
1500 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1501 fb1a36c0 2022-01-09 op return;
1502 fb1a36c0 2022-01-09 op }
1503 fb1a36c0 2022-01-09 op
1504 fb1a36c0 2022-01-09 op if (TYPE_OVERFLOW(off_t, off)) {
1505 fb1a36c0 2022-01-09 op log_warnx("unexpected off_t size");
1506 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid offset");
1507 fb1a36c0 2022-01-09 op return;
1508 fb1a36c0 2022-01-09 op }
1509 fb1a36c0 2022-01-09 op
1510 fb1a36c0 2022-01-09 op if (f->d == NULL) {
1511 fb1a36c0 2022-01-09 op /* read a file */
1512 fb1a36c0 2022-01-09 op howmuch = MIN(sizeof(buf), count);
1513 fb1a36c0 2022-01-09 op r = pread(f->fd, buf, howmuch, (off_t)off);
1514 fb1a36c0 2022-01-09 op if (r == -1)
1515 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1516 fb1a36c0 2022-01-09 op else
1517 fb1a36c0 2022-01-09 op np_read(hdr->tag, r, buf);
1518 fb1a36c0 2022-01-09 op } else {
1519 fb1a36c0 2022-01-09 op if (off == 0 && f->offset != 0) {
1520 fb1a36c0 2022-01-09 op rewinddir(f->d);
1521 fb1a36c0 2022-01-09 op f->offset = 0;
1522 fb1a36c0 2022-01-09 op evbuffer_drain(f->evb, EVBUFFER_LENGTH(f->evb));
1523 fb1a36c0 2022-01-09 op }
1524 fb1a36c0 2022-01-09 op
1525 fb1a36c0 2022-01-09 op if (off != f->offset) {
1526 fb1a36c0 2022-01-09 op np_error(hdr->tag, "can't seek in directories");
1527 fb1a36c0 2022-01-09 op return;
1528 fb1a36c0 2022-01-09 op }
1529 fb1a36c0 2022-01-09 op
1530 fb1a36c0 2022-01-09 op while (EVBUFFER_LENGTH(f->evb) < count) {
1531 fb1a36c0 2022-01-09 op struct dirent *d;
1532 fb1a36c0 2022-01-09 op struct stat sb;
1533 fb1a36c0 2022-01-09 op
1534 fb1a36c0 2022-01-09 op if ((d = readdir(f->d)) == NULL)
1535 fb1a36c0 2022-01-09 op break;
1536 ff6f19b1 2022-02-02 op if (!strcmp(d->d_name, ".") || !strcmp(d->d_name, ".."))
1537 ff6f19b1 2022-02-02 op continue;
1538 fb1a36c0 2022-01-09 op if (fstatat(f->fd, d->d_name, &sb, 0) == -1) {
1539 fb1a36c0 2022-01-09 op warn("fstatat");
1540 fb1a36c0 2022-01-09 op continue;
1541 fb1a36c0 2022-01-09 op }
1542 fb1a36c0 2022-01-09 op serialize_stat(d->d_name, &sb, f->evb);
1543 fb1a36c0 2022-01-09 op }
1544 fb1a36c0 2022-01-09 op
1545 fb1a36c0 2022-01-09 op count = MIN(count, EVBUFFER_LENGTH(f->evb));
1546 fb1a36c0 2022-01-09 op np_read(hdr->tag, count, EVBUFFER_DATA(f->evb));
1547 fb1a36c0 2022-01-09 op evbuffer_drain(f->evb, count);
1548 fb1a36c0 2022-01-09 op
1549 fb1a36c0 2022-01-09 op f->offset += count;
1550 fb1a36c0 2022-01-09 op }
1551 fb1a36c0 2022-01-09 op }
1552 fb1a36c0 2022-01-09 op
1553 fb1a36c0 2022-01-09 op static void
1554 8b0a5062 2022-05-22 op twrite(struct np_msg_header *hdr, const uint8_t *data, size_t len,
1555 8b0a5062 2022-05-22 op struct fid **writefid, off_t *writepos, size_t *writeleft, int *writeskip)
1556 fb1a36c0 2022-01-09 op {
1557 fb1a36c0 2022-01-09 op struct fid *f;
1558 fb1a36c0 2022-01-09 op ssize_t r;
1559 fb1a36c0 2022-01-09 op uint64_t off;
1560 fb1a36c0 2022-01-09 op uint32_t fid, count;
1561 fb1a36c0 2022-01-09 op
1562 fb1a36c0 2022-01-09 op /* fid[4] offset[8] count[4] data[count] */
1563 fb1a36c0 2022-01-09 op if (!NPREAD32("fid", &fid, &data, &len) ||
1564 fb1a36c0 2022-01-09 op !NPREAD64("off", &off, &data, &len) ||
1565 fb1a36c0 2022-01-09 op !NPREAD32("count", &count, &data, &len) ||
1566 8b0a5062 2022-05-22 op count < len) {
1567 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1568 fb1a36c0 2022-01-09 op client_shutdown();
1569 fb1a36c0 2022-01-09 op return;
1570 fb1a36c0 2022-01-09 op }
1571 fb1a36c0 2022-01-09 op
1572 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL || f->fd == -1) {
1573 8b0a5062 2022-05-22 op *writeskip = 1;
1574 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1575 fb1a36c0 2022-01-09 op return;
1576 fb1a36c0 2022-01-09 op }
1577 fb1a36c0 2022-01-09 op
1578 fb1a36c0 2022-01-09 op if (!(f->iomode & O_WRONLY) &&
1579 fb1a36c0 2022-01-09 op !(f->iomode & O_RDWR)) {
1580 8b0a5062 2022-05-22 op *writeskip = 1;
1581 fb1a36c0 2022-01-09 op np_error(hdr->tag, "fid not opened for writing");
1582 fb1a36c0 2022-01-09 op return;
1583 fb1a36c0 2022-01-09 op }
1584 fb1a36c0 2022-01-09 op
1585 fb1a36c0 2022-01-09 op if (TYPE_OVERFLOW(off_t, off)) {
1586 8b0a5062 2022-05-22 op *writeskip = 1;
1587 fb1a36c0 2022-01-09 op log_warnx("unexpected off_t size");
1588 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid offset");
1589 fb1a36c0 2022-01-09 op return;
1590 fb1a36c0 2022-01-09 op }
1591 fb1a36c0 2022-01-09 op
1592 8b0a5062 2022-05-22 op if ((r = pwrite(f->fd, data, len, off)) == -1) {
1593 8b0a5062 2022-05-22 op *writeskip = 1;
1594 8b0a5062 2022-05-22 op np_errno(hdr->tag);
1595 8b0a5062 2022-05-22 op } else if (count == len)
1596 fb1a36c0 2022-01-09 op np_write(hdr->tag, r);
1597 8b0a5062 2022-05-22 op
1598 8b0a5062 2022-05-22 op /* account for a continuated write */
1599 8b0a5062 2022-05-22 op if (count > len) {
1600 8b0a5062 2022-05-22 op *writefid = f;
1601 8b0a5062 2022-05-22 op *writepos = off + len;
1602 8b0a5062 2022-05-22 op *writeleft = count - len;
1603 8b0a5062 2022-05-22 op *writeskip = 0;
1604 8b0a5062 2022-05-22 op }
1605 fb1a36c0 2022-01-09 op }
1606 fb1a36c0 2022-01-09 op
1607 fb1a36c0 2022-01-09 op static void
1608 8b0a5062 2022-05-22 op twrite_cont(struct fid *f, off_t *writepos, size_t *writeleft, int *writeskip,
1609 8b0a5062 2022-05-22 op uint16_t tag, const uint8_t *data, size_t len)
1610 8b0a5062 2022-05-22 op {
1611 8b0a5062 2022-05-22 op ssize_t r;
1612 8b0a5062 2022-05-22 op
1613 8b0a5062 2022-05-22 op if (len > *writeleft) {
1614 8b0a5062 2022-05-22 op client_send_listener(IMSG_CLOSE, NULL, 0);
1615 8b0a5062 2022-05-22 op client_shutdown();
1616 8b0a5062 2022-05-22 op return;
1617 8b0a5062 2022-05-22 op }
1618 8b0a5062 2022-05-22 op
1619 8b0a5062 2022-05-22 op if ((r = pwrite(f->fd, data, len, *writepos)) == -1) {
1620 8b0a5062 2022-05-22 op *writeskip = 1;
1621 8b0a5062 2022-05-22 op np_errno(tag);
1622 8b0a5062 2022-05-22 op return;
1623 8b0a5062 2022-05-22 op }
1624 8b0a5062 2022-05-22 op
1625 8b0a5062 2022-05-22 op *writeleft -= len;
1626 8b0a5062 2022-05-22 op *writepos += len;
1627 8b0a5062 2022-05-22 op
1628 8b0a5062 2022-05-22 op if (*writeleft == 0)
1629 8b0a5062 2022-05-22 op np_write(tag, r);
1630 8b0a5062 2022-05-22 op }
1631 8b0a5062 2022-05-22 op
1632 8b0a5062 2022-05-22 op static void
1633 fb1a36c0 2022-01-09 op tstat(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1634 fb1a36c0 2022-01-09 op {
1635 fb1a36c0 2022-01-09 op struct evbuffer *evb;
1636 fb1a36c0 2022-01-09 op struct stat sb;
1637 fb1a36c0 2022-01-09 op struct fid *f;
1638 fb1a36c0 2022-01-09 op int r;
1639 fb1a36c0 2022-01-09 op uint32_t fid;
1640 fb1a36c0 2022-01-09 op
1641 fb1a36c0 2022-01-09 op /* fid[4] */
1642 bcdb46a7 2022-01-19 op if (!NPREAD32("fid", &fid, &data, &len) ||
1643 bcdb46a7 2022-01-19 op len != 0) {
1644 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1645 fb1a36c0 2022-01-09 op client_shutdown();
1646 fb1a36c0 2022-01-09 op return;
1647 fb1a36c0 2022-01-09 op }
1648 fb1a36c0 2022-01-09 op
1649 fb1a36c0 2022-01-09 op /*
1650 fb1a36c0 2022-01-09 op * plan9' stat(9P) is not clear on whether the stat is allowed
1651 fb1a36c0 2022-01-09 op * on opened fids or not. We're allowing stat regardless of the
1652 fb1a36c0 2022-01-09 op * status of the fid.
1653 fb1a36c0 2022-01-09 op */
1654 fb1a36c0 2022-01-09 op
1655 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL) {
1656 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1657 fb1a36c0 2022-01-09 op return;
1658 fb1a36c0 2022-01-09 op }
1659 fb1a36c0 2022-01-09 op
1660 fb1a36c0 2022-01-09 op if ((evb = evbuffer_new()) == NULL)
1661 fb1a36c0 2022-01-09 op fatal("evbuffer_new");
1662 fb1a36c0 2022-01-09 op
1663 fb1a36c0 2022-01-09 op if (f->fd != -1)
1664 fb1a36c0 2022-01-09 op r = fstat(f->fd, &sb);
1665 fb1a36c0 2022-01-09 op else if (f->qid.type & QTDIR)
1666 fb1a36c0 2022-01-09 op r = fstat(f->dir->fd, &sb);
1667 fb1a36c0 2022-01-09 op else
1668 543fc013 2022-01-17 op r = fstatat(f->dir->fd, f->fname, &sb, 0);
1669 fb1a36c0 2022-01-09 op
1670 fb1a36c0 2022-01-09 op if (r == -1) {
1671 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1672 fb1a36c0 2022-01-09 op evbuffer_free(evb);
1673 fb1a36c0 2022-01-09 op return;
1674 fb1a36c0 2022-01-09 op }
1675 fb1a36c0 2022-01-09 op
1676 91f48e25 2022-02-02 op if (serialize_stat(f->fname, &sb, evb) == -1)
1677 91f48e25 2022-02-02 op np_error(hdr->tag, "stat would overflow");
1678 91f48e25 2022-02-02 op else
1679 91f48e25 2022-02-02 op np_stat(hdr->tag, EVBUFFER_LENGTH(evb), EVBUFFER_DATA(evb));
1680 fb1a36c0 2022-01-09 op evbuffer_free(evb);
1681 755d75bf 2022-01-17 op }
1682 755d75bf 2022-01-17 op
1683 755d75bf 2022-01-17 op static void
1684 755d75bf 2022-01-17 op twstat(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1685 755d75bf 2022-01-17 op {
1686 755d75bf 2022-01-17 op struct timespec times[2];
1687 755d75bf 2022-01-17 op struct np_stat st;
1688 755d75bf 2022-01-17 op struct fid *f;
1689 755d75bf 2022-01-17 op int r;
1690 755d75bf 2022-01-17 op uint32_t fid;
1691 755d75bf 2022-01-17 op char name[PATH_MAX];
1692 755d75bf 2022-01-17 op
1693 755d75bf 2022-01-17 op /* fid[4] stat[n] */
1694 755d75bf 2022-01-17 op if (!NPREAD32("fid", &fid, &data, &len))
1695 755d75bf 2022-01-17 op goto err;
1696 755d75bf 2022-01-17 op
1697 755d75bf 2022-01-17 op switch (NPREADST("stat", &st, name, sizeof(name), &data, &len)) {
1698 755d75bf 2022-01-17 op case READSTRERR:
1699 755d75bf 2022-01-17 op goto err;
1700 755d75bf 2022-01-17 op case READSTRTRUNC:
1701 755d75bf 2022-01-17 op log_warnx("wstat new name would overflow");
1702 755d75bf 2022-01-17 op np_error(hdr->tag, "new name too long");
1703 755d75bf 2022-01-17 op return;
1704 755d75bf 2022-01-17 op }
1705 755d75bf 2022-01-17 op
1706 f4ad0146 2022-01-19 op /*
1707 f4ad0146 2022-01-19 op * We skip the reading of some fields voluntarily because we
1708 f4ad0146 2022-01-19 op * don't support chown, so len will always be > 0!
1709 f4ad0146 2022-01-19 op */
1710 f4ad0146 2022-01-19 op #ifdef notyet
1711 bcdb46a7 2022-01-19 op if (len != 0)
1712 bcdb46a7 2022-01-19 op goto err;
1713 f4ad0146 2022-01-19 op #endif
1714 bcdb46a7 2022-01-19 op
1715 755d75bf 2022-01-17 op if ((f = fid_by_id(fid)) == NULL) {
1716 755d75bf 2022-01-17 op np_error(hdr->tag, "invalid fid");
1717 755d75bf 2022-01-17 op return;
1718 755d75bf 2022-01-17 op }
1719 755d75bf 2022-01-17 op
1720 755d75bf 2022-01-17 op /*
1721 755d75bf 2022-01-17 op * 9P wants these edits to be done in an atomic fashion,
1722 755d75bf 2022-01-17 op * something that I don't think it's possible on UNIX without
1723 755d75bf 2022-01-17 op * some special kernel help. Changing atime or mtime,
1724 755d75bf 2022-01-17 op * permissions, or rename the file are different syscalls.
1725 755d75bf 2022-01-17 op *
1726 755d75bf 2022-01-17 op * Also, silently ignore stuff we can't/don't want to modify.
1727 755d75bf 2022-01-17 op */
1728 755d75bf 2022-01-17 op
1729 755d75bf 2022-01-17 op /* change the permissions */
1730 755d75bf 2022-01-17 op if (st.mode != (uint32_t)-1) {
1731 b7c2d377 2022-02-13 op mode_t m = st.mode & 0x7FF; /* silently truncate higer bits */
1732 755d75bf 2022-01-17 op if (f->fd != -1)
1733 755d75bf 2022-01-17 op r = fchmod(f->fd, m);
1734 755d75bf 2022-01-17 op else
1735 543fc013 2022-01-17 op r = fchmodat(f->dir->fd, f->fname, m, 0);
1736 755d75bf 2022-01-17 op
1737 755d75bf 2022-01-17 op if (r == -1) {
1738 755d75bf 2022-01-17 op np_errno(hdr->tag);
1739 755d75bf 2022-01-17 op return;
1740 755d75bf 2022-01-17 op }
1741 755d75bf 2022-01-17 op }
1742 755d75bf 2022-01-17 op
1743 755d75bf 2022-01-17 op /* change the atime/mtime of the fid, opened or not */
1744 755d75bf 2022-01-17 op if (st.atime != (uint32_t)-1 || st.mtime != (uint32_t)-1) {
1745 755d75bf 2022-01-17 op times[0].tv_nsec = UTIME_OMIT;
1746 755d75bf 2022-01-17 op times[1].tv_nsec = UTIME_OMIT;
1747 755d75bf 2022-01-17 op
1748 755d75bf 2022-01-17 op if (st.atime != (uint32_t)-1) {
1749 755d75bf 2022-01-17 op times[0].tv_sec = st.atime;
1750 755d75bf 2022-01-17 op times[0].tv_nsec = 0;
1751 755d75bf 2022-01-17 op }
1752 755d75bf 2022-01-17 op if (st.mtime != (uint32_t)-1) {
1753 755d75bf 2022-01-17 op times[1].tv_sec = st.mtime;
1754 755d75bf 2022-01-17 op times[1].tv_nsec = 0;
1755 755d75bf 2022-01-17 op }
1756 755d75bf 2022-01-17 op
1757 755d75bf 2022-01-17 op if (f->fd != -1)
1758 755d75bf 2022-01-17 op r = futimens(f->fd, times);
1759 755d75bf 2022-01-17 op else
1760 543fc013 2022-01-17 op r = utimensat(f->dir->fd, f->fname, times, 0);
1761 755d75bf 2022-01-17 op
1762 755d75bf 2022-01-17 op if (r == -1) {
1763 755d75bf 2022-01-17 op np_errno(hdr->tag);
1764 755d75bf 2022-01-17 op return;
1765 755d75bf 2022-01-17 op }
1766 755d75bf 2022-01-17 op }
1767 755d75bf 2022-01-17 op
1768 755d75bf 2022-01-17 op /* truncate */
1769 755d75bf 2022-01-17 op if (st.length != (uint64_t)-1) {
1770 755d75bf 2022-01-17 op if (f->fd == -1) {
1771 755d75bf 2022-01-17 op np_error(hdr->tag, "can't truncate a non-opened fid");
1772 755d75bf 2022-01-17 op return;
1773 755d75bf 2022-01-17 op }
1774 755d75bf 2022-01-17 op
1775 755d75bf 2022-01-17 op if (f->qid.type & QTDIR && st.length != 0) {
1776 755d75bf 2022-01-17 op np_error(hdr->tag, "can't truncate directories");
1777 755d75bf 2022-01-17 op return;
1778 755d75bf 2022-01-17 op }
1779 755d75bf 2022-01-17 op
1780 755d75bf 2022-01-17 op if (TYPE_OVERFLOW(off_t, st.length)) {
1781 755d75bf 2022-01-17 op log_warnx("truncate request overflows off_t: %"PRIu64,
1782 755d75bf 2022-01-17 op st.length);
1783 755d75bf 2022-01-17 op np_error(hdr->tag, "length overflows");
1784 755d75bf 2022-01-17 op return;
1785 755d75bf 2022-01-17 op }
1786 755d75bf 2022-01-17 op
1787 6d3d71a9 2022-01-19 op if (f->qid.type == 0 &&
1788 755d75bf 2022-01-17 op ftruncate(f->fd, st.length) == -1) {
1789 755d75bf 2022-01-17 op np_errno(hdr->tag);
1790 755d75bf 2022-01-17 op return;
1791 755d75bf 2022-01-17 op }
1792 755d75bf 2022-01-17 op }
1793 755d75bf 2022-01-17 op
1794 755d75bf 2022-01-17 op /* rename (change the name entry) */
1795 755d75bf 2022-01-17 op if (*st.name != '\0') {
1796 755d75bf 2022-01-17 op struct stat sb;
1797 755d75bf 2022-01-17 op const char *newname;
1798 755d75bf 2022-01-17 op
1799 755d75bf 2022-01-17 op /*
1800 755d75bf 2022-01-17 op * XXX: 9P requires that we disallow changing the name
1801 755d75bf 2022-01-17 op * to that of an existing file. We can't do that
1802 755d75bf 2022-01-17 op * atomically (rename is happy about stealing names)
1803 755d75bf 2022-01-17 op * so this is just a best effort.
1804 755d75bf 2022-01-17 op */
1805 755d75bf 2022-01-17 op if (fstatat(f->dir->fd, st.name, &sb, 0) == -1 &&
1806 755d75bf 2022-01-17 op errno != ENOENT) {
1807 755d75bf 2022-01-17 op np_error(hdr->tag, "target file exists");
1808 755d75bf 2022-01-17 op return;
1809 755d75bf 2022-01-17 op }
1810 755d75bf 2022-01-17 op
1811 543fc013 2022-01-17 op r = renameat(f->dir->fd, f->fname, f->dir->fd, st.name);
1812 755d75bf 2022-01-17 op if (r == -1) {
1813 755d75bf 2022-01-17 op np_errno(hdr->tag);
1814 755d75bf 2022-01-17 op return;
1815 755d75bf 2022-01-17 op }
1816 755d75bf 2022-01-17 op
1817 755d75bf 2022-01-17 op /* fix the fid so it points to the new file */
1818 755d75bf 2022-01-17 op
1819 755d75bf 2022-01-17 op if ((newname = strchr(st.name, '/')) != NULL)
1820 755d75bf 2022-01-17 op newname++; /* skip / */
1821 755d75bf 2022-01-17 op else
1822 755d75bf 2022-01-17 op newname = st.name;
1823 543fc013 2022-01-17 op strlcpy(f->fname, newname, sizeof(f->fname));
1824 755d75bf 2022-01-17 op
1825 755d75bf 2022-01-17 op if (strchr(st.name, '/') != NULL) {
1826 755d75bf 2022-01-17 op struct dir *dir;
1827 755d75bf 2022-01-17 op char *dname;
1828 755d75bf 2022-01-17 op int fd;
1829 755d75bf 2022-01-17 op
1830 755d75bf 2022-01-17 op dname = dirname(st.name);
1831 755d75bf 2022-01-17 op fd = openat(f->dir->fd, dname, O_RDONLY|O_DIRECTORY);
1832 755d75bf 2022-01-17 op if (fd == -1) {
1833 755d75bf 2022-01-17 op np_errno(hdr->tag);
1834 755d75bf 2022-01-17 op free_fid(f);
1835 755d75bf 2022-01-17 op return;
1836 755d75bf 2022-01-17 op }
1837 755d75bf 2022-01-17 op
1838 755d75bf 2022-01-17 op if ((dir = new_dir(fd)) == NULL) {
1839 755d75bf 2022-01-17 op np_errno(hdr->tag);
1840 755d75bf 2022-01-17 op free_fid(f);
1841 755d75bf 2022-01-17 op return;
1842 755d75bf 2022-01-17 op }
1843 755d75bf 2022-01-17 op
1844 755d75bf 2022-01-17 op dir_decref(f->dir);
1845 755d75bf 2022-01-17 op f->dir = dir_incref(dir);
1846 755d75bf 2022-01-17 op }
1847 755d75bf 2022-01-17 op }
1848 755d75bf 2022-01-17 op
1849 755d75bf 2022-01-17 op np_wstat(hdr->tag);
1850 755d75bf 2022-01-17 op return;
1851 755d75bf 2022-01-17 op
1852 755d75bf 2022-01-17 op err:
1853 755d75bf 2022-01-17 op client_send_listener(IMSG_CLOSE, NULL, 0);
1854 755d75bf 2022-01-17 op client_shutdown();
1855 fb1a36c0 2022-01-09 op }
1856 fb1a36c0 2022-01-09 op
1857 fb1a36c0 2022-01-09 op static void
1858 fb1a36c0 2022-01-09 op tremove(struct np_msg_header *hdr, const uint8_t *data, size_t len)
1859 fb1a36c0 2022-01-09 op {
1860 fb1a36c0 2022-01-09 op struct fid *f;
1861 fb1a36c0 2022-01-09 op uint32_t fid;
1862 fb1a36c0 2022-01-09 op int r;
1863 fb1a36c0 2022-01-09 op char dirpath[PATH_MAX + 3];
1864 fb1a36c0 2022-01-09 op
1865 fb1a36c0 2022-01-09 op /* fid[4] */
1866 bcdb46a7 2022-01-19 op if (!NPREAD32("fid", &fid, &data, &len) ||
1867 bcdb46a7 2022-01-19 op len != 0) {
1868 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1869 fb1a36c0 2022-01-09 op client_shutdown();
1870 fb1a36c0 2022-01-09 op return;
1871 fb1a36c0 2022-01-09 op }
1872 fb1a36c0 2022-01-09 op
1873 fb1a36c0 2022-01-09 op if ((f = fid_by_id(fid)) == NULL) {
1874 fb1a36c0 2022-01-09 op np_error(hdr->tag, "invalid fid");
1875 fb1a36c0 2022-01-09 op return;
1876 fb1a36c0 2022-01-09 op }
1877 fb1a36c0 2022-01-09 op
1878 fb1a36c0 2022-01-09 op if (f->qid.type & QTDIR) { /* directory */
1879 fb1a36c0 2022-01-09 op strlcpy(dirpath, "../", sizeof(dirpath));
1880 543fc013 2022-01-17 op strlcat(dirpath, f->fname, sizeof(dirpath));
1881 fb1a36c0 2022-01-09 op r = unlinkat(f->dir->fd, dirpath, AT_REMOVEDIR);
1882 fb1a36c0 2022-01-09 op } else /* file */
1883 543fc013 2022-01-17 op r = unlinkat(f->dir->fd, f->fname, 0);
1884 fb1a36c0 2022-01-09 op
1885 fb1a36c0 2022-01-09 op if (r == -1)
1886 fb1a36c0 2022-01-09 op np_errno(hdr->tag);
1887 fb1a36c0 2022-01-09 op else
1888 fb1a36c0 2022-01-09 op np_remove(hdr->tag);
1889 fb1a36c0 2022-01-09 op
1890 fb1a36c0 2022-01-09 op free_fid(f);
1891 fb1a36c0 2022-01-09 op }
1892 fb1a36c0 2022-01-09 op
1893 fb1a36c0 2022-01-09 op static void
1894 8b0a5062 2022-05-22 op handle_message(struct imsg *imsg, size_t len, int cont)
1895 fb1a36c0 2022-01-09 op {
1896 8b0a5062 2022-05-22 op static struct fid *writefid;
1897 8b0a5062 2022-05-22 op static off_t writepos;
1898 8b0a5062 2022-05-22 op static size_t writeleft;
1899 8b0a5062 2022-05-22 op static int writeskip;
1900 8b0a5062 2022-05-22 op static uint16_t writetag;
1901 fb1a36c0 2022-01-09 op struct msg {
1902 fb1a36c0 2022-01-09 op uint8_t type;
1903 fb1a36c0 2022-01-09 op void (*fn)(struct np_msg_header *, const uint8_t *, size_t);
1904 fb1a36c0 2022-01-09 op } msgs[] = {
1905 fb1a36c0 2022-01-09 op {Tversion, tversion},
1906 fb1a36c0 2022-01-09 op {Tattach, tattach},
1907 fb1a36c0 2022-01-09 op {Tclunk, tclunk},
1908 fb1a36c0 2022-01-09 op {Tflush, tflush},
1909 fb1a36c0 2022-01-09 op {Twalk, twalk},
1910 fb1a36c0 2022-01-09 op {Topen, topen},
1911 fb1a36c0 2022-01-09 op {Tcreate, tcreate},
1912 fb1a36c0 2022-01-09 op {Tread, tread},
1913 8b0a5062 2022-05-22 op /* {Twrite, twrite}, */
1914 fb1a36c0 2022-01-09 op {Tstat, tstat},
1915 755d75bf 2022-01-17 op {Twstat, twstat},
1916 fb1a36c0 2022-01-09 op {Tremove, tremove},
1917 fb1a36c0 2022-01-09 op };
1918 fb1a36c0 2022-01-09 op struct np_msg_header hdr;
1919 fb1a36c0 2022-01-09 op size_t i;
1920 fb1a36c0 2022-01-09 op uint8_t *data;
1921 fb1a36c0 2022-01-09 op
1922 fb1a36c0 2022-01-09 op #if DEBUG_PACKETS
1923 fb1a36c0 2022-01-09 op hexdump("incoming packet", imsg->data, len);
1924 fb1a36c0 2022-01-09 op #endif
1925 fb1a36c0 2022-01-09 op
1926 8b0a5062 2022-05-22 op /*
1927 8b0a5062 2022-05-22 op * Twrite is special and can be "continued" to allow writing
1928 8b0a5062 2022-05-22 op * more than what the imsg framework would allow us to.
1929 8b0a5062 2022-05-22 op */
1930 8b0a5062 2022-05-22 op if (writeleft > 0 && !cont) {
1931 8b0a5062 2022-05-22 op log_warnx("received a non continuation message when still "
1932 8b0a5062 2022-05-22 op "missed %zu bytes to write", writeleft);
1933 8b0a5062 2022-05-22 op client_send_listener(IMSG_CLOSE, NULL, 0);
1934 8b0a5062 2022-05-22 op client_shutdown();
1935 8b0a5062 2022-05-22 op return;
1936 8b0a5062 2022-05-22 op }
1937 8b0a5062 2022-05-22 op
1938 8b0a5062 2022-05-22 op if (cont) {
1939 8b0a5062 2022-05-22 op if (writeskip)
1940 8b0a5062 2022-05-22 op return;
1941 8b0a5062 2022-05-22 op
1942 8b0a5062 2022-05-22 op if (writefid == NULL) {
1943 8b0a5062 2022-05-22 op log_warnx("received a continuation message without "
1944 8b0a5062 2022-05-22 op "seeing a Twrite");
1945 8b0a5062 2022-05-22 op client_send_listener(IMSG_CLOSE, NULL, 0);
1946 8b0a5062 2022-05-22 op client_shutdown();
1947 8b0a5062 2022-05-22 op return;
1948 8b0a5062 2022-05-22 op }
1949 8b0a5062 2022-05-22 op
1950 8b0a5062 2022-05-22 op log_warnx("continuing...");
1951 8b0a5062 2022-05-22 op twrite_cont(writefid, &writepos, &writeleft, &writeskip,
1952 8b0a5062 2022-05-22 op writetag, imsg->data, len);
1953 8b0a5062 2022-05-22 op return;
1954 8b0a5062 2022-05-22 op }
1955 8b0a5062 2022-05-22 op
1956 8b0a5062 2022-05-22 op writefid = NULL;
1957 8b0a5062 2022-05-22 op writepos = -1;
1958 8b0a5062 2022-05-22 op writeleft = 0;
1959 8b0a5062 2022-05-22 op writeskip = 0;
1960 8b0a5062 2022-05-22 op
1961 fb1a36c0 2022-01-09 op parse_message(imsg->data, len, &hdr, &data);
1962 fb1a36c0 2022-01-09 op len -= HEADERSIZE;
1963 fb1a36c0 2022-01-09 op
1964 fb1a36c0 2022-01-09 op log_debug("got request: len=%d type=%d[%s] tag=%d",
1965 fb1a36c0 2022-01-09 op hdr.len, hdr.type, pp_msg_type(hdr.type), hdr.tag);
1966 fb1a36c0 2022-01-09 op
1967 fb1a36c0 2022-01-09 op if (!handshaked && hdr.type != Tversion) {
1968 fb1a36c0 2022-01-09 op client_send_listener(IMSG_CLOSE, NULL, 0);
1969 fb1a36c0 2022-01-09 op client_shutdown();
1970 fb1a36c0 2022-01-09 op return;
1971 fb1a36c0 2022-01-09 op }
1972 fb1a36c0 2022-01-09 op
1973 8b0a5062 2022-05-22 op if (hdr.type == Twrite) {
1974 8b0a5062 2022-05-22 op writetag = hdr.tag;
1975 8b0a5062 2022-05-22 op twrite(&hdr, data, len, &writefid, &writepos, &writeleft,
1976 8b0a5062 2022-05-22 op &writeskip);
1977 8b0a5062 2022-05-22 op return;
1978 8b0a5062 2022-05-22 op }
1979 8b0a5062 2022-05-22 op
1980 fb1a36c0 2022-01-09 op for (i = 0; i < sizeof(msgs)/sizeof(msgs[0]); ++i) {
1981 fb1a36c0 2022-01-09 op if (msgs[i].type != hdr.type)
1982 fb1a36c0 2022-01-09 op continue;
1983 fb1a36c0 2022-01-09 op
1984 fb1a36c0 2022-01-09 op msgs[i].fn(&hdr, data, len);
1985 fb1a36c0 2022-01-09 op return;
1986 fb1a36c0 2022-01-09 op }
1987 fb1a36c0 2022-01-09 op
1988 fb1a36c0 2022-01-09 op np_error(hdr.tag, "Not supported.");
1989 fb1a36c0 2022-01-09 op }