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