puffs.c revision 1.88 1 1.88 pooka /* $NetBSD: puffs.c,v 1.88 2008/01/16 21:30:00 pooka Exp $ */
2 1.1 pooka
3 1.1 pooka /*
4 1.44 pooka * Copyright (c) 2005, 2006, 2007 Antti Kantee. All Rights Reserved.
5 1.1 pooka *
6 1.1 pooka * Development of this software was supported by the
7 1.1 pooka * Google Summer of Code program and the Ulla Tuominen Foundation.
8 1.1 pooka * The Google SoC project was mentored by Bill Studenmund.
9 1.1 pooka *
10 1.1 pooka * Redistribution and use in source and binary forms, with or without
11 1.1 pooka * modification, are permitted provided that the following conditions
12 1.1 pooka * are met:
13 1.1 pooka * 1. Redistributions of source code must retain the above copyright
14 1.1 pooka * notice, this list of conditions and the following disclaimer.
15 1.1 pooka * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 pooka * notice, this list of conditions and the following disclaimer in the
17 1.1 pooka * documentation and/or other materials provided with the distribution.
18 1.1 pooka *
19 1.1 pooka * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
20 1.1 pooka * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
21 1.1 pooka * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
22 1.1 pooka * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23 1.1 pooka * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 1.1 pooka * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
25 1.1 pooka * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 1.1 pooka * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 1.1 pooka * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 1.1 pooka * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 1.1 pooka * SUCH DAMAGE.
30 1.1 pooka */
31 1.1 pooka
32 1.1 pooka #include <sys/cdefs.h>
33 1.1 pooka #if !defined(lint)
34 1.88 pooka __RCSID("$NetBSD: puffs.c,v 1.88 2008/01/16 21:30:00 pooka Exp $");
35 1.1 pooka #endif /* !lint */
36 1.1 pooka
37 1.1 pooka #include <sys/param.h>
38 1.1 pooka #include <sys/mount.h>
39 1.1 pooka
40 1.9 pooka #include <assert.h>
41 1.12 pooka #include <err.h>
42 1.1 pooka #include <errno.h>
43 1.1 pooka #include <fcntl.h>
44 1.21 pooka #include <mntopts.h>
45 1.58 pooka #include <paths.h>
46 1.1 pooka #include <puffs.h>
47 1.1 pooka #include <stdio.h>
48 1.1 pooka #include <stdlib.h>
49 1.1 pooka #include <string.h>
50 1.1 pooka #include <syslog.h>
51 1.1 pooka #include <unistd.h>
52 1.1 pooka
53 1.19 pooka #include "puffs_priv.h"
54 1.19 pooka
55 1.21 pooka /* Most file systems want this for opts, so just give it to them */
56 1.21 pooka const struct mntopt puffsmopts[] = {
57 1.21 pooka MOPT_STDOPTS,
58 1.21 pooka PUFFSMOPT_STD,
59 1.21 pooka MOPT_NULL,
60 1.21 pooka };
61 1.21 pooka
62 1.72 pooka #ifdef PUFFS_WITH_THREADS
63 1.72 pooka #include <pthread.h>
64 1.72 pooka pthread_mutex_t pu_lock = PTHREAD_MUTEX_INITIALIZER;
65 1.72 pooka #endif
66 1.72 pooka
67 1.10 pooka #define FILLOP(lower, upper) \
68 1.10 pooka do { \
69 1.13 pooka if (pops->puffs_node_##lower) \
70 1.10 pooka opmask[PUFFS_VN_##upper] = 1; \
71 1.10 pooka } while (/*CONSTCOND*/0)
72 1.10 pooka static void
73 1.13 pooka fillvnopmask(struct puffs_ops *pops, uint8_t *opmask)
74 1.10 pooka {
75 1.10 pooka
76 1.10 pooka memset(opmask, 0, PUFFS_VN_MAX);
77 1.10 pooka
78 1.10 pooka FILLOP(create, CREATE);
79 1.10 pooka FILLOP(mknod, MKNOD);
80 1.10 pooka FILLOP(open, OPEN);
81 1.10 pooka FILLOP(close, CLOSE);
82 1.10 pooka FILLOP(access, ACCESS);
83 1.10 pooka FILLOP(getattr, GETATTR);
84 1.10 pooka FILLOP(setattr, SETATTR);
85 1.10 pooka FILLOP(poll, POLL); /* XXX: not ready in kernel */
86 1.16 pooka FILLOP(mmap, MMAP);
87 1.10 pooka FILLOP(fsync, FSYNC);
88 1.10 pooka FILLOP(seek, SEEK);
89 1.10 pooka FILLOP(remove, REMOVE);
90 1.10 pooka FILLOP(link, LINK);
91 1.10 pooka FILLOP(rename, RENAME);
92 1.10 pooka FILLOP(mkdir, MKDIR);
93 1.10 pooka FILLOP(rmdir, RMDIR);
94 1.10 pooka FILLOP(symlink, SYMLINK);
95 1.10 pooka FILLOP(readdir, READDIR);
96 1.10 pooka FILLOP(readlink, READLINK);
97 1.10 pooka FILLOP(reclaim, RECLAIM);
98 1.10 pooka FILLOP(inactive, INACTIVE);
99 1.10 pooka FILLOP(print, PRINT);
100 1.10 pooka FILLOP(read, READ);
101 1.10 pooka FILLOP(write, WRITE);
102 1.19 pooka }
103 1.19 pooka #undef FILLOP
104 1.19 pooka
105 1.79 pooka /*
106 1.79 pooka * Go over all framev entries and write everything we can. This is
107 1.79 pooka * mostly for the benefit of delivering "unmount" to the kernel.
108 1.79 pooka */
109 1.79 pooka static void
110 1.79 pooka finalpush(struct puffs_usermount *pu)
111 1.79 pooka {
112 1.79 pooka struct puffs_fctrl_io *fio;
113 1.79 pooka
114 1.79 pooka LIST_FOREACH(fio, &pu->pu_ios, fio_entries) {
115 1.79 pooka if (fio->stat & FIO_WRGONE)
116 1.79 pooka continue;
117 1.79 pooka
118 1.79 pooka puffs_framev_output(pu, fio->fctrl, fio);
119 1.79 pooka }
120 1.79 pooka }
121 1.79 pooka
122 1.64 pooka /*ARGSUSED*/
123 1.63 pooka static void
124 1.63 pooka puffs_defaulterror(struct puffs_usermount *pu, uint8_t type,
125 1.65 pooka int error, const char *str, void *cookie)
126 1.63 pooka {
127 1.63 pooka
128 1.77 pooka fprintf(stderr, "abort: type %d, error %d, cookie %p (%s)\n",
129 1.77 pooka type, error, cookie, str);
130 1.63 pooka abort();
131 1.63 pooka }
132 1.63 pooka
133 1.19 pooka int
134 1.19 pooka puffs_getselectable(struct puffs_usermount *pu)
135 1.19 pooka {
136 1.10 pooka
137 1.60 pooka return pu->pu_fd;
138 1.10 pooka }
139 1.19 pooka
140 1.80 pooka uint64_t
141 1.80 pooka puffs__nextreq(struct puffs_usermount *pu)
142 1.80 pooka {
143 1.80 pooka uint64_t rv;
144 1.80 pooka
145 1.80 pooka PU_LOCK();
146 1.80 pooka rv = pu->pu_nextreq++;
147 1.80 pooka PU_UNLOCK();
148 1.80 pooka
149 1.80 pooka return rv;
150 1.80 pooka }
151 1.80 pooka
152 1.19 pooka int
153 1.19 pooka puffs_setblockingmode(struct puffs_usermount *pu, int mode)
154 1.19 pooka {
155 1.46 pooka int rv, x;
156 1.46 pooka
157 1.74 pooka assert(puffs_getstate(pu) == PUFFS_STATE_RUNNING);
158 1.74 pooka
159 1.46 pooka if (mode != PUFFSDEV_BLOCK && mode != PUFFSDEV_NONBLOCK) {
160 1.46 pooka errno = EINVAL;
161 1.46 pooka return -1;
162 1.46 pooka }
163 1.19 pooka
164 1.19 pooka x = mode;
165 1.60 pooka rv = ioctl(pu->pu_fd, FIONBIO, &x);
166 1.46 pooka
167 1.46 pooka if (rv == 0) {
168 1.46 pooka if (mode == PUFFSDEV_BLOCK)
169 1.46 pooka pu->pu_state &= ~PU_ASYNCFD;
170 1.46 pooka else
171 1.46 pooka pu->pu_state |= PU_ASYNCFD;
172 1.46 pooka }
173 1.46 pooka
174 1.46 pooka return rv;
175 1.19 pooka }
176 1.19 pooka
177 1.19 pooka int
178 1.19 pooka puffs_getstate(struct puffs_usermount *pu)
179 1.19 pooka {
180 1.19 pooka
181 1.44 pooka return pu->pu_state & PU_STATEMASK;
182 1.19 pooka }
183 1.19 pooka
184 1.19 pooka void
185 1.19 pooka puffs_setstacksize(struct puffs_usermount *pu, size_t ss)
186 1.19 pooka {
187 1.86 pooka long psize, minsize;
188 1.69 pooka int stackshift;
189 1.87 pooka int bonus;
190 1.78 pooka
191 1.86 pooka assert(puffs_getstate(pu) == PUFFS_STATE_BEFOREMOUNT);
192 1.86 pooka
193 1.78 pooka psize = sysconf(_SC_PAGESIZE);
194 1.87 pooka minsize = 4*psize;
195 1.86 pooka if (ss < minsize || ss == PUFFS_STACKSIZE_MIN) {
196 1.86 pooka if (ss != PUFFS_STACKSIZE_MIN)
197 1.86 pooka fprintf(stderr, "puffs_setstacksize: adjusting "
198 1.86 pooka "stacksize to minimum %ld\n", minsize);
199 1.87 pooka ss = 4*psize;
200 1.78 pooka }
201 1.69 pooka
202 1.69 pooka stackshift = -1;
203 1.87 pooka bonus = 0;
204 1.69 pooka while (ss) {
205 1.87 pooka if (ss & 0x1)
206 1.87 pooka bonus++;
207 1.69 pooka ss >>= 1;
208 1.69 pooka stackshift++;
209 1.69 pooka }
210 1.87 pooka if (bonus > 1) {
211 1.87 pooka stackshift++;
212 1.87 pooka fprintf(stderr, "puffs_setstacksize: using next power of two: "
213 1.88 pooka "%d\n", 1<<stackshift);
214 1.87 pooka }
215 1.87 pooka
216 1.69 pooka pu->pu_cc_stackshift = stackshift;
217 1.19 pooka }
218 1.19 pooka
219 1.24 pooka struct puffs_pathobj *
220 1.24 pooka puffs_getrootpathobj(struct puffs_usermount *pu)
221 1.19 pooka {
222 1.19 pooka struct puffs_node *pnr;
223 1.19 pooka
224 1.19 pooka pnr = pu->pu_pn_root;
225 1.19 pooka if (pnr == NULL) {
226 1.19 pooka errno = ENOENT;
227 1.24 pooka return NULL;
228 1.19 pooka }
229 1.19 pooka
230 1.24 pooka return &pnr->pn_po;
231 1.24 pooka }
232 1.24 pooka
233 1.35 pooka void
234 1.35 pooka puffs_setroot(struct puffs_usermount *pu, struct puffs_node *pn)
235 1.35 pooka {
236 1.35 pooka
237 1.35 pooka pu->pu_pn_root = pn;
238 1.35 pooka }
239 1.35 pooka
240 1.35 pooka struct puffs_node *
241 1.35 pooka puffs_getroot(struct puffs_usermount *pu)
242 1.35 pooka {
243 1.35 pooka
244 1.35 pooka return pu->pu_pn_root;
245 1.35 pooka }
246 1.35 pooka
247 1.49 pooka void
248 1.49 pooka puffs_setrootinfo(struct puffs_usermount *pu, enum vtype vt,
249 1.49 pooka vsize_t vsize, dev_t rdev)
250 1.49 pooka {
251 1.60 pooka struct puffs_kargs *pargs = pu->pu_kargp;
252 1.49 pooka
253 1.60 pooka if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT) {
254 1.49 pooka warnx("puffs_setrootinfo: call has effect only "
255 1.49 pooka "before mount\n");
256 1.60 pooka return;
257 1.60 pooka }
258 1.49 pooka
259 1.49 pooka pargs->pa_root_vtype = vt;
260 1.49 pooka pargs->pa_root_vsize = vsize;
261 1.49 pooka pargs->pa_root_rdev = rdev;
262 1.49 pooka }
263 1.49 pooka
264 1.35 pooka void *
265 1.35 pooka puffs_getspecific(struct puffs_usermount *pu)
266 1.35 pooka {
267 1.35 pooka
268 1.35 pooka return pu->pu_privdata;
269 1.35 pooka }
270 1.35 pooka
271 1.35 pooka size_t
272 1.35 pooka puffs_getmaxreqlen(struct puffs_usermount *pu)
273 1.35 pooka {
274 1.35 pooka
275 1.60 pooka return pu->pu_maxreqlen;
276 1.35 pooka }
277 1.24 pooka
278 1.24 pooka void
279 1.37 pooka puffs_setmaxreqlen(struct puffs_usermount *pu, size_t reqlen)
280 1.37 pooka {
281 1.37 pooka
282 1.44 pooka if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
283 1.37 pooka warnx("puffs_setmaxreqlen: call has effect only "
284 1.37 pooka "before mount\n");
285 1.37 pooka
286 1.66 pooka pu->pu_kargp->pa_maxmsglen = reqlen;
287 1.37 pooka }
288 1.37 pooka
289 1.37 pooka void
290 1.38 pooka puffs_setfhsize(struct puffs_usermount *pu, size_t fhsize, int flags)
291 1.37 pooka {
292 1.37 pooka
293 1.44 pooka if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
294 1.37 pooka warnx("puffs_setfhsize: call has effect only before mount\n");
295 1.37 pooka
296 1.60 pooka pu->pu_kargp->pa_fhsize = fhsize;
297 1.60 pooka pu->pu_kargp->pa_fhflags = flags;
298 1.37 pooka }
299 1.37 pooka
300 1.37 pooka void
301 1.37 pooka puffs_setncookiehash(struct puffs_usermount *pu, int nhash)
302 1.37 pooka {
303 1.37 pooka
304 1.44 pooka if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
305 1.37 pooka warnx("puffs_setfhsize: call has effect only before mount\n");
306 1.37 pooka
307 1.60 pooka pu->pu_kargp->pa_nhashbuckets = nhash;
308 1.37 pooka }
309 1.37 pooka
310 1.37 pooka void
311 1.24 pooka puffs_set_pathbuild(struct puffs_usermount *pu, pu_pathbuild_fn fn)
312 1.24 pooka {
313 1.24 pooka
314 1.24 pooka pu->pu_pathbuild = fn;
315 1.24 pooka }
316 1.24 pooka
317 1.24 pooka void
318 1.24 pooka puffs_set_pathtransform(struct puffs_usermount *pu, pu_pathtransform_fn fn)
319 1.24 pooka {
320 1.24 pooka
321 1.24 pooka pu->pu_pathtransform = fn;
322 1.24 pooka }
323 1.24 pooka
324 1.24 pooka void
325 1.24 pooka puffs_set_pathcmp(struct puffs_usermount *pu, pu_pathcmp_fn fn)
326 1.24 pooka {
327 1.24 pooka
328 1.24 pooka pu->pu_pathcmp = fn;
329 1.24 pooka }
330 1.24 pooka
331 1.24 pooka void
332 1.24 pooka puffs_set_pathfree(struct puffs_usermount *pu, pu_pathfree_fn fn)
333 1.24 pooka {
334 1.19 pooka
335 1.24 pooka pu->pu_pathfree = fn;
336 1.19 pooka }
337 1.19 pooka
338 1.24 pooka void
339 1.24 pooka puffs_set_namemod(struct puffs_usermount *pu, pu_namemod_fn fn)
340 1.24 pooka {
341 1.24 pooka
342 1.24 pooka pu->pu_namemod = fn;
343 1.24 pooka }
344 1.19 pooka
345 1.40 pooka void
346 1.63 pooka puffs_set_errnotify(struct puffs_usermount *pu, pu_errnotify_fn fn)
347 1.63 pooka {
348 1.63 pooka
349 1.63 pooka pu->pu_errnotify = fn;
350 1.63 pooka }
351 1.63 pooka
352 1.63 pooka void
353 1.81 pooka puffs_set_cmap(struct puffs_usermount *pu, pu_cmap_fn fn)
354 1.81 pooka {
355 1.81 pooka
356 1.81 pooka pu->pu_cmap = fn;
357 1.81 pooka }
358 1.81 pooka
359 1.81 pooka void
360 1.46 pooka puffs_ml_setloopfn(struct puffs_usermount *pu, puffs_ml_loop_fn lfn)
361 1.46 pooka {
362 1.46 pooka
363 1.46 pooka pu->pu_ml_lfn = lfn;
364 1.46 pooka }
365 1.46 pooka
366 1.46 pooka void
367 1.46 pooka puffs_ml_settimeout(struct puffs_usermount *pu, struct timespec *ts)
368 1.46 pooka {
369 1.46 pooka
370 1.46 pooka if (ts == NULL) {
371 1.46 pooka pu->pu_ml_timep = NULL;
372 1.46 pooka } else {
373 1.46 pooka pu->pu_ml_timeout = *ts;
374 1.46 pooka pu->pu_ml_timep = &pu->pu_ml_timeout;
375 1.46 pooka }
376 1.46 pooka }
377 1.46 pooka
378 1.46 pooka void
379 1.71 pooka puffs_set_prepost(struct puffs_usermount *pu,
380 1.71 pooka pu_prepost_fn pre, pu_prepost_fn pst)
381 1.71 pooka {
382 1.71 pooka
383 1.71 pooka pu->pu_oppre = pre;
384 1.71 pooka pu->pu_oppost = pst;
385 1.71 pooka }
386 1.71 pooka
387 1.71 pooka void
388 1.40 pooka puffs_setback(struct puffs_cc *pcc, int whatback)
389 1.40 pooka {
390 1.79 pooka struct puffs_req *preq = puffs__framebuf_getdataptr(pcc->pcc_pb);
391 1.40 pooka
392 1.40 pooka assert(PUFFSOP_OPCLASS(preq->preq_opclass) == PUFFSOP_VN && (
393 1.40 pooka preq->preq_optype == PUFFS_VN_OPEN ||
394 1.40 pooka preq->preq_optype == PUFFS_VN_MMAP ||
395 1.40 pooka preq->preq_optype == PUFFS_VN_REMOVE ||
396 1.54 pooka preq->preq_optype == PUFFS_VN_RMDIR ||
397 1.54 pooka preq->preq_optype == PUFFS_VN_INACTIVE));
398 1.40 pooka
399 1.40 pooka preq->preq_setbacks |= whatback & PUFFS_SETBACK_MASK;
400 1.40 pooka }
401 1.40 pooka
402 1.36 pooka int
403 1.75 pooka puffs_daemon(struct puffs_usermount *pu, int nochdir, int noclose)
404 1.75 pooka {
405 1.75 pooka ssize_t n;
406 1.75 pooka int parent, value, fd;
407 1.75 pooka
408 1.75 pooka if (pipe(pu->pu_dpipe) == -1)
409 1.75 pooka return -1;
410 1.75 pooka
411 1.75 pooka switch (fork()) {
412 1.75 pooka case -1:
413 1.75 pooka return -1;
414 1.75 pooka case 0:
415 1.75 pooka parent = 0;
416 1.75 pooka break;
417 1.75 pooka default:
418 1.75 pooka parent = 1;
419 1.75 pooka break;
420 1.75 pooka }
421 1.75 pooka pu->pu_state |= PU_PUFFSDAEMON;
422 1.75 pooka
423 1.75 pooka if (parent) {
424 1.75 pooka n = read(pu->pu_dpipe[0], &value, sizeof(int));
425 1.75 pooka if (n == -1)
426 1.75 pooka err(1, "puffs_daemon");
427 1.75 pooka assert(n == sizeof(value));
428 1.75 pooka if (value) {
429 1.75 pooka errno = value;
430 1.75 pooka err(1, "puffs_daemon");
431 1.75 pooka }
432 1.75 pooka exit(0);
433 1.75 pooka } else {
434 1.75 pooka if (setsid() == -1)
435 1.75 pooka goto fail;
436 1.75 pooka
437 1.75 pooka if (!nochdir)
438 1.75 pooka chdir("/");
439 1.75 pooka
440 1.75 pooka if (!noclose) {
441 1.75 pooka fd = open(_PATH_DEVNULL, O_RDWR, 0);
442 1.75 pooka if (fd == -1)
443 1.75 pooka goto fail;
444 1.75 pooka dup2(fd, STDIN_FILENO);
445 1.75 pooka dup2(fd, STDOUT_FILENO);
446 1.75 pooka dup2(fd, STDERR_FILENO);
447 1.75 pooka if (fd > STDERR_FILENO)
448 1.75 pooka close(fd);
449 1.75 pooka }
450 1.75 pooka return 0;
451 1.75 pooka }
452 1.75 pooka
453 1.75 pooka fail:
454 1.75 pooka n = write(pu->pu_dpipe[1], &errno, sizeof(int));
455 1.75 pooka assert(n == 4);
456 1.75 pooka return -1;
457 1.75 pooka }
458 1.75 pooka
459 1.75 pooka int
460 1.49 pooka puffs_mount(struct puffs_usermount *pu, const char *dir, int mntflags,
461 1.49 pooka void *cookie)
462 1.36 pooka {
463 1.62 pooka char rp[MAXPATHLEN];
464 1.75 pooka ssize_t n;
465 1.75 pooka int rv, fd, sverrno;
466 1.82 pooka char *comfd;
467 1.82 pooka
468 1.82 pooka pu->pu_kargp->pa_root_cookie = cookie;
469 1.36 pooka
470 1.36 pooka /* XXXkludgehere */
471 1.36 pooka /* kauth doesn't provide this service any longer */
472 1.36 pooka if (geteuid() != 0)
473 1.36 pooka mntflags |= MNT_NOSUID | MNT_NODEV;
474 1.36 pooka
475 1.76 pooka if (realpath(dir, rp) == NULL) {
476 1.76 pooka rv = -1;
477 1.76 pooka goto out;
478 1.76 pooka }
479 1.62 pooka
480 1.62 pooka if (strcmp(dir, rp) != 0) {
481 1.62 pooka warnx("puffs_mount: \"%s\" is a relative path.", dir);
482 1.62 pooka warnx("puffs_mount: using \"%s\" instead.", rp);
483 1.62 pooka }
484 1.62 pooka
485 1.82 pooka /*
486 1.82 pooka * Undocumented... Well, documented only here.
487 1.82 pooka *
488 1.82 pooka * This is used for imaginative purposes. If the env variable is
489 1.82 pooka * set, puffs_mount() doesn't do the regular mount procedure.
490 1.82 pooka * Rather, it crams the mount data down the comfd and sets comfd as
491 1.82 pooka * the puffs descriptor.
492 1.82 pooka *
493 1.82 pooka * This shouldn't be used unless you can read my mind ( ... or write
494 1.82 pooka * it, not to mention execute it, but that's starting to get silly).
495 1.82 pooka */
496 1.82 pooka if ((comfd = getenv("PUFFS_COMFD")) != NULL) {
497 1.82 pooka size_t len;
498 1.82 pooka
499 1.82 pooka if (sscanf(comfd, "%d", &pu->pu_fd) != 1) {
500 1.82 pooka errno = EINVAL;
501 1.82 pooka rv = -1;
502 1.82 pooka goto out;
503 1.82 pooka }
504 1.82 pooka /* check that what we got at least resembles an fd */
505 1.82 pooka if (fcntl(pu->pu_fd, F_GETFL) == -1) {
506 1.82 pooka rv = -1;
507 1.82 pooka goto out;
508 1.82 pooka }
509 1.82 pooka
510 1.82 pooka len = strlen(dir)+1;
511 1.82 pooka
512 1.82 pooka #define allwrite(buf, len) \
513 1.82 pooka do { \
514 1.82 pooka ssize_t al_rv; \
515 1.82 pooka al_rv = write(pu->pu_fd, buf, len); \
516 1.82 pooka if (al_rv != len) { \
517 1.82 pooka if (al_rv != -1) \
518 1.82 pooka errno = EIO; \
519 1.82 pooka rv = -1; \
520 1.82 pooka abort();\
521 1.82 pooka goto out; \
522 1.82 pooka } \
523 1.82 pooka } while (/*CONSTCOND*/0)
524 1.82 pooka allwrite(&len, sizeof(len));
525 1.82 pooka allwrite(dir, len);
526 1.84 pooka len = strlen(pu->pu_kargp->pa_mntfromname)+1;
527 1.83 pooka allwrite(&len, sizeof(len));
528 1.83 pooka allwrite(pu->pu_kargp->pa_mntfromname, len);
529 1.82 pooka allwrite(&mntflags, sizeof(mntflags));
530 1.82 pooka allwrite(pu->pu_kargp, sizeof(*pu->pu_kargp));
531 1.82 pooka allwrite(&pu->pu_flags, sizeof(pu->pu_flags));
532 1.82 pooka #undef allwrite
533 1.82 pooka
534 1.82 pooka rv = 0;
535 1.82 pooka } else {
536 1.82 pooka fd = open(_PATH_PUFFS, O_RDWR);
537 1.82 pooka if (fd == -1) {
538 1.82 pooka warnx("puffs_mount: cannot open %s", _PATH_PUFFS);
539 1.82 pooka rv = -1;
540 1.82 pooka goto out;
541 1.82 pooka }
542 1.82 pooka if (fd <= 2)
543 1.82 pooka warnx("puffs_mount: device fd %d (<= 2), sure this is "
544 1.82 pooka "what you want?", fd);
545 1.82 pooka
546 1.82 pooka pu->pu_kargp->pa_fd = pu->pu_fd = fd;
547 1.82 pooka if ((rv = mount(MOUNT_PUFFS, rp, mntflags,
548 1.82 pooka pu->pu_kargp, sizeof(struct puffs_kargs))) == -1)
549 1.82 pooka goto out;
550 1.74 pooka }
551 1.74 pooka
552 1.49 pooka PU_SETSTATE(pu, PUFFS_STATE_RUNNING);
553 1.36 pooka
554 1.60 pooka out:
555 1.76 pooka if (rv != 0)
556 1.76 pooka sverrno = errno;
557 1.76 pooka else
558 1.76 pooka sverrno = 0;
559 1.60 pooka free(pu->pu_kargp);
560 1.60 pooka pu->pu_kargp = NULL;
561 1.60 pooka
562 1.75 pooka if (pu->pu_state & PU_PUFFSDAEMON) {
563 1.75 pooka n = write(pu->pu_dpipe[1], &sverrno, sizeof(int));
564 1.75 pooka assert(n == 4);
565 1.75 pooka close(pu->pu_dpipe[0]);
566 1.75 pooka close(pu->pu_dpipe[1]);
567 1.75 pooka }
568 1.75 pooka
569 1.75 pooka errno = sverrno;
570 1.60 pooka return rv;
571 1.36 pooka }
572 1.10 pooka
573 1.1 pooka struct puffs_usermount *
574 1.58 pooka _puffs_init(int develv, struct puffs_ops *pops, const char *mntfromname,
575 1.58 pooka const char *puffsname, void *priv, uint32_t pflags)
576 1.1 pooka {
577 1.1 pooka struct puffs_usermount *pu;
578 1.36 pooka struct puffs_kargs *pargs;
579 1.74 pooka int sverrno;
580 1.1 pooka
581 1.20 pooka if (develv != PUFFS_DEVEL_LIBVERSION) {
582 1.53 pooka warnx("puffs_init: mounting with lib version %d, need %d",
583 1.20 pooka develv, PUFFS_DEVEL_LIBVERSION);
584 1.20 pooka errno = EINVAL;
585 1.20 pooka return NULL;
586 1.20 pooka }
587 1.20 pooka
588 1.36 pooka pu = malloc(sizeof(struct puffs_usermount));
589 1.36 pooka if (pu == NULL)
590 1.36 pooka goto failfree;
591 1.47 pooka memset(pu, 0, sizeof(struct puffs_usermount));
592 1.1 pooka
593 1.60 pooka pargs = pu->pu_kargp = malloc(sizeof(struct puffs_kargs));
594 1.60 pooka if (pargs == NULL)
595 1.60 pooka goto failfree;
596 1.60 pooka memset(pargs, 0, sizeof(struct puffs_kargs));
597 1.60 pooka
598 1.36 pooka pargs->pa_vers = PUFFSDEVELVERS | PUFFSVERSION;
599 1.36 pooka pargs->pa_flags = PUFFS_FLAG_KERN(pflags);
600 1.36 pooka fillvnopmask(pops, pargs->pa_vnopmask);
601 1.58 pooka (void)strlcpy(pargs->pa_typename, puffsname,
602 1.58 pooka sizeof(pargs->pa_typename));
603 1.58 pooka (void)strlcpy(pargs->pa_mntfromname, mntfromname,
604 1.58 pooka sizeof(pargs->pa_mntfromname));
605 1.1 pooka
606 1.49 pooka puffs_zerostatvfs(&pargs->pa_svfsb);
607 1.49 pooka pargs->pa_root_cookie = NULL;
608 1.49 pooka pargs->pa_root_vtype = VDIR;
609 1.49 pooka pargs->pa_root_vsize = 0;
610 1.49 pooka pargs->pa_root_rdev = 0;
611 1.66 pooka pargs->pa_maxmsglen = 0;
612 1.49 pooka
613 1.1 pooka pu->pu_flags = pflags;
614 1.13 pooka pu->pu_ops = *pops;
615 1.21 pooka free(pops); /* XXX */
616 1.36 pooka
617 1.19 pooka pu->pu_privdata = priv;
618 1.69 pooka pu->pu_cc_stackshift = PUFFS_CC_STACKSHIFT_DEFAULT;
619 1.1 pooka LIST_INIT(&pu->pu_pnodelst);
620 1.79 pooka LIST_INIT(&pu->pu_ios);
621 1.79 pooka LIST_INIT(&pu->pu_ios_rmlist);
622 1.88 pooka LIST_INIT(&pu->pu_ccmagazin);
623 1.46 pooka LIST_INIT(&pu->pu_ccnukelst);
624 1.67 pooka TAILQ_INIT(&pu->pu_sched);
625 1.70 pooka TAILQ_INIT(&pu->pu_exq);
626 1.1 pooka
627 1.79 pooka pu->pu_framectrl[PU_FRAMECTRL_FS].rfb = puffs_fsframe_read;
628 1.79 pooka pu->pu_framectrl[PU_FRAMECTRL_FS].wfb = puffs_fsframe_write;
629 1.79 pooka pu->pu_framectrl[PU_FRAMECTRL_FS].cmpfb = puffs_fsframe_cmp;
630 1.79 pooka pu->pu_framectrl[PU_FRAMECTRL_FS].gotfb = puffs_fsframe_gotframe;
631 1.79 pooka pu->pu_framectrl[PU_FRAMECTRL_FS].fdnotfn = puffs_framev_unmountonclose;
632 1.79 pooka
633 1.24 pooka /* defaults for some user-settable translation functions */
634 1.24 pooka pu->pu_cmap = NULL; /* identity translation */
635 1.24 pooka
636 1.30 pooka pu->pu_pathbuild = puffs_stdpath_buildpath;
637 1.30 pooka pu->pu_pathfree = puffs_stdpath_freepath;
638 1.30 pooka pu->pu_pathcmp = puffs_stdpath_cmppath;
639 1.24 pooka pu->pu_pathtransform = NULL;
640 1.24 pooka pu->pu_namemod = NULL;
641 1.24 pooka
642 1.63 pooka pu->pu_errnotify = puffs_defaulterror;
643 1.63 pooka
644 1.44 pooka PU_SETSTATE(pu, PUFFS_STATE_BEFOREMOUNT);
645 1.1 pooka
646 1.1 pooka return pu;
647 1.6 pooka
648 1.1 pooka failfree:
649 1.6 pooka /* can't unmount() from here for obvious reasons */
650 1.60 pooka sverrno = errno;
651 1.1 pooka free(pu);
652 1.60 pooka errno = sverrno;
653 1.1 pooka return NULL;
654 1.1 pooka }
655 1.1 pooka
656 1.21 pooka /*
657 1.21 pooka * XXX: there's currently no clean way to request unmount from
658 1.21 pooka * within the user server, so be very brutal about it.
659 1.21 pooka */
660 1.36 pooka /*ARGSUSED1*/
661 1.21 pooka int
662 1.21 pooka puffs_exit(struct puffs_usermount *pu, int force)
663 1.21 pooka {
664 1.44 pooka struct puffs_node *pn;
665 1.21 pooka
666 1.21 pooka force = 1; /* currently */
667 1.21 pooka
668 1.60 pooka if (pu->pu_fd)
669 1.60 pooka close(pu->pu_fd);
670 1.21 pooka
671 1.44 pooka while ((pn = LIST_FIRST(&pu->pu_pnodelst)) != NULL)
672 1.24 pooka puffs_pn_put(pn);
673 1.44 pooka
674 1.79 pooka finalpush(pu);
675 1.46 pooka puffs_framev_exit(pu);
676 1.68 pooka if (pu->pu_state & PU_HASKQ)
677 1.44 pooka close(pu->pu_kq);
678 1.21 pooka free(pu);
679 1.21 pooka
680 1.21 pooka return 0; /* always succesful for now, WILL CHANGE */
681 1.21 pooka }
682 1.21 pooka
683 1.21 pooka int
684 1.73 pooka puffs_mainloop(struct puffs_usermount *pu)
685 1.1 pooka {
686 1.79 pooka struct puffs_framectrl *pfctrl;
687 1.44 pooka struct puffs_fctrl_io *fio;
688 1.67 pooka struct puffs_cc *pcc;
689 1.79 pooka struct kevent *curev;
690 1.44 pooka size_t nchanges;
691 1.79 pooka int ndone, sverrno;
692 1.44 pooka
693 1.44 pooka assert(puffs_getstate(pu) >= PUFFS_STATE_RUNNING);
694 1.1 pooka
695 1.79 pooka pu->pu_kq = kqueue();
696 1.79 pooka if (pu->pu_kq == -1)
697 1.44 pooka goto out;
698 1.79 pooka pu->pu_state |= PU_HASKQ;
699 1.18 alc
700 1.79 pooka puffs_setblockingmode(pu, PUFFSDEV_NONBLOCK);
701 1.79 pooka if (puffs__framev_addfd_ctrl(pu, puffs_getselectable(pu),
702 1.79 pooka PUFFS_FBIO_READ | PUFFS_FBIO_WRITE,
703 1.79 pooka &pu->pu_framectrl[PU_FRAMECTRL_FS]) == -1)
704 1.44 pooka goto out;
705 1.44 pooka
706 1.79 pooka curev = realloc(pu->pu_evs, (2*pu->pu_nfds)*sizeof(struct kevent));
707 1.79 pooka if (curev == NULL)
708 1.44 pooka goto out;
709 1.79 pooka pu->pu_evs = curev;
710 1.19 pooka
711 1.79 pooka LIST_FOREACH(fio, &pu->pu_ios, fio_entries) {
712 1.44 pooka EV_SET(curev, fio->io_fd, EVFILT_READ, EV_ADD,
713 1.44 pooka 0, 0, (uintptr_t)fio);
714 1.44 pooka curev++;
715 1.44 pooka EV_SET(curev, fio->io_fd, EVFILT_WRITE, EV_ADD | EV_DISABLE,
716 1.44 pooka 0, 0, (uintptr_t)fio);
717 1.44 pooka curev++;
718 1.44 pooka }
719 1.79 pooka if (kevent(pu->pu_kq, pu->pu_evs, 2*pu->pu_nfds, NULL, 0, NULL) == -1)
720 1.44 pooka goto out;
721 1.44 pooka
722 1.79 pooka pu->pu_state |= PU_INLOOP;
723 1.79 pooka
724 1.44 pooka while (puffs_getstate(pu) != PUFFS_STATE_UNMOUNTED) {
725 1.46 pooka if (pu->pu_ml_lfn)
726 1.46 pooka pu->pu_ml_lfn(pu);
727 1.46 pooka
728 1.79 pooka /* XXX: can we still do these optimizations? */
729 1.79 pooka #if 0
730 1.55 pooka /*
731 1.55 pooka * Do this here, because:
732 1.55 pooka * a) loopfunc might generate some results
733 1.55 pooka * b) it's still "after" event handling (except for round 1)
734 1.55 pooka */
735 1.55 pooka if (puffs_req_putput(ppr) == -1)
736 1.55 pooka goto out;
737 1.55 pooka puffs_req_resetput(ppr);
738 1.55 pooka
739 1.46 pooka /* micro optimization: skip kevent syscall if possible */
740 1.79 pooka if (pu->pu_nfds == 1 && pu->pu_ml_timep == NULL
741 1.46 pooka && (pu->pu_state & PU_ASYNCFD) == 0) {
742 1.79 pooka pfctrl = XXX->fctrl;
743 1.79 pooka puffs_framev_input(pu, pfctrl, XXX);
744 1.46 pooka continue;
745 1.46 pooka }
746 1.79 pooka #endif
747 1.55 pooka
748 1.46 pooka /* else: do full processing */
749 1.79 pooka /* Don't bother worrying about O(n) for now */
750 1.79 pooka LIST_FOREACH(fio, &pu->pu_ios, fio_entries) {
751 1.46 pooka if (fio->stat & FIO_WRGONE)
752 1.46 pooka continue;
753 1.46 pooka
754 1.79 pooka pfctrl = fio->fctrl;
755 1.79 pooka
756 1.44 pooka /*
757 1.44 pooka * Try to write out everything to avoid the
758 1.44 pooka * need for enabling EVFILT_WRITE. The likely
759 1.44 pooka * case is that we can fit everything into the
760 1.44 pooka * socket buffer.
761 1.44 pooka */
762 1.79 pooka puffs_framev_output(pu, pfctrl, fio);
763 1.79 pooka }
764 1.79 pooka
765 1.79 pooka /*
766 1.79 pooka * Build list of which to enable/disable in writecheck.
767 1.79 pooka */
768 1.80 pooka nchanges = 0;
769 1.79 pooka LIST_FOREACH(fio, &pu->pu_ios, fio_entries) {
770 1.79 pooka if (fio->stat & FIO_WRGONE)
771 1.79 pooka continue;
772 1.19 pooka
773 1.46 pooka /* en/disable write checks for kqueue as needed */
774 1.44 pooka assert((FIO_EN_WRITE(fio) && FIO_RM_WRITE(fio)) == 0);
775 1.44 pooka if (FIO_EN_WRITE(fio)) {
776 1.79 pooka EV_SET(&pu->pu_evs[nchanges], fio->io_fd,
777 1.44 pooka EVFILT_WRITE, EV_ENABLE, 0, 0,
778 1.44 pooka (uintptr_t)fio);
779 1.46 pooka fio->stat |= FIO_WR;
780 1.44 pooka nchanges++;
781 1.44 pooka }
782 1.44 pooka if (FIO_RM_WRITE(fio)) {
783 1.79 pooka EV_SET(&pu->pu_evs[nchanges], fio->io_fd,
784 1.44 pooka EVFILT_WRITE, EV_DISABLE, 0, 0,
785 1.44 pooka (uintptr_t)fio);
786 1.46 pooka fio->stat &= ~FIO_WR;
787 1.44 pooka nchanges++;
788 1.44 pooka }
789 1.79 pooka assert(nchanges <= pu->pu_nfds);
790 1.19 pooka }
791 1.44 pooka
792 1.79 pooka ndone = kevent(pu->pu_kq, pu->pu_evs, nchanges,
793 1.79 pooka pu->pu_evs, 2*pu->pu_nfds, pu->pu_ml_timep);
794 1.50 pooka
795 1.50 pooka if (ndone == -1) {
796 1.50 pooka if (errno != EINTR)
797 1.50 pooka goto out;
798 1.50 pooka else
799 1.50 pooka continue;
800 1.50 pooka }
801 1.50 pooka
802 1.50 pooka /* uoptimize */
803 1.46 pooka if (ndone == 0)
804 1.46 pooka continue;
805 1.44 pooka
806 1.44 pooka /* iterate over the results */
807 1.79 pooka for (curev = pu->pu_evs; ndone--; curev++) {
808 1.61 pooka int what;
809 1.61 pooka
810 1.79 pooka #if 0
811 1.44 pooka /* get & possibly dispatch events from kernel */
812 1.44 pooka if (curev->ident == puffsfd) {
813 1.44 pooka if (puffs_req_handle(pgr, ppr, 0) == -1)
814 1.44 pooka goto out;
815 1.44 pooka continue;
816 1.44 pooka }
817 1.79 pooka #endif
818 1.44 pooka
819 1.46 pooka fio = (void *)curev->udata;
820 1.79 pooka pfctrl = fio->fctrl;
821 1.46 pooka if (curev->flags & EV_ERROR) {
822 1.46 pooka assert(curev->filter == EVFILT_WRITE);
823 1.46 pooka fio->stat &= ~FIO_WR;
824 1.46 pooka
825 1.46 pooka /* XXX: how to know if it's a transient error */
826 1.46 pooka puffs_framev_writeclose(pu, fio,
827 1.46 pooka (int)curev->data);
828 1.61 pooka puffs_framev_notify(fio, PUFFS_FBIO_ERROR);
829 1.46 pooka continue;
830 1.44 pooka }
831 1.44 pooka
832 1.61 pooka what = 0;
833 1.61 pooka if (curev->filter == EVFILT_READ) {
834 1.79 pooka puffs_framev_input(pu, pfctrl, fio);
835 1.61 pooka what |= PUFFS_FBIO_READ;
836 1.61 pooka }
837 1.56 pooka
838 1.61 pooka else if (curev->filter == EVFILT_WRITE) {
839 1.79 pooka puffs_framev_output(pu, pfctrl, fio);
840 1.61 pooka what |= PUFFS_FBIO_WRITE;
841 1.61 pooka }
842 1.61 pooka if (what)
843 1.61 pooka puffs_framev_notify(fio, what);
844 1.19 pooka }
845 1.44 pooka
846 1.46 pooka /*
847 1.67 pooka * Schedule continuations.
848 1.67 pooka */
849 1.67 pooka while ((pcc = TAILQ_FIRST(&pu->pu_sched)) != NULL) {
850 1.88 pooka TAILQ_REMOVE(&pu->pu_sched, pcc, pcc_schedent);
851 1.67 pooka puffs_goto(pcc);
852 1.67 pooka }
853 1.67 pooka
854 1.67 pooka /*
855 1.46 pooka * Really free fd's now that we don't have references
856 1.46 pooka * to them.
857 1.46 pooka */
858 1.79 pooka while ((fio = LIST_FIRST(&pu->pu_ios_rmlist)) != NULL) {
859 1.46 pooka LIST_REMOVE(fio, fio_entries);
860 1.46 pooka free(fio);
861 1.46 pooka }
862 1.18 alc }
863 1.79 pooka finalpush(pu);
864 1.44 pooka errno = 0;
865 1.18 alc
866 1.19 pooka out:
867 1.44 pooka /* store the real error for a while */
868 1.44 pooka sverrno = errno;
869 1.44 pooka
870 1.44 pooka errno = sverrno;
871 1.44 pooka if (errno)
872 1.44 pooka return -1;
873 1.44 pooka else
874 1.44 pooka return 0;
875 1.1 pooka }
876