Home | History | Annotate | Line # | Download | only in libpuffs
puffs.c revision 1.116.4.1
      1  1.116.4.1   yamt /*	$NetBSD: puffs.c,v 1.116.4.1 2012/04/17 00:05:31 yamt 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.116.4.1   yamt __RCSID("$NetBSD: puffs.c,v 1.116.4.1 2012/04/17 00:05:31 yamt 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.115  pooka #include <pthread.h>
     47        1.1  pooka #include <puffs.h>
     48        1.1  pooka #include <stdio.h>
     49        1.1  pooka #include <stdlib.h>
     50        1.1  pooka #include <string.h>
     51        1.1  pooka #include <syslog.h>
     52        1.1  pooka #include <unistd.h>
     53        1.1  pooka 
     54       1.19  pooka #include "puffs_priv.h"
     55       1.19  pooka 
     56       1.21  pooka /* Most file systems want this for opts, so just give it to them */
     57       1.21  pooka const struct mntopt puffsmopts[] = {
     58       1.21  pooka 	MOPT_STDOPTS,
     59       1.21  pooka 	PUFFSMOPT_STD,
     60       1.21  pooka 	MOPT_NULL,
     61       1.21  pooka };
     62       1.21  pooka 
     63       1.72  pooka pthread_mutex_t pu_lock = PTHREAD_MUTEX_INITIALIZER;
     64       1.72  pooka 
     65       1.10  pooka #define FILLOP(lower, upper)						\
     66       1.10  pooka do {									\
     67       1.13  pooka 	if (pops->puffs_node_##lower)					\
     68       1.10  pooka 		opmask[PUFFS_VN_##upper] = 1;				\
     69       1.10  pooka } while (/*CONSTCOND*/0)
     70       1.10  pooka static void
     71      1.107  pooka fillvnopmask(struct puffs_ops *pops, struct puffs_kargs *pa)
     72       1.10  pooka {
     73      1.107  pooka 	uint8_t *opmask = pa->pa_vnopmask;
     74       1.10  pooka 
     75      1.107  pooka 	memset(opmask, 0, sizeof(pa->pa_vnopmask));
     76       1.10  pooka 
     77       1.10  pooka 	FILLOP(create,   CREATE);
     78       1.10  pooka 	FILLOP(mknod,    MKNOD);
     79       1.10  pooka 	FILLOP(open,     OPEN);
     80       1.10  pooka 	FILLOP(close,    CLOSE);
     81       1.10  pooka 	FILLOP(access,   ACCESS);
     82       1.10  pooka 	FILLOP(getattr,  GETATTR);
     83       1.10  pooka 	FILLOP(setattr,  SETATTR);
     84      1.106  pooka 	FILLOP(poll,     POLL);
     85       1.16  pooka 	FILLOP(mmap,     MMAP);
     86       1.10  pooka 	FILLOP(fsync,    FSYNC);
     87       1.10  pooka 	FILLOP(seek,     SEEK);
     88       1.10  pooka 	FILLOP(remove,   REMOVE);
     89       1.10  pooka 	FILLOP(link,     LINK);
     90       1.10  pooka 	FILLOP(rename,   RENAME);
     91       1.10  pooka 	FILLOP(mkdir,    MKDIR);
     92       1.10  pooka 	FILLOP(rmdir,    RMDIR);
     93       1.10  pooka 	FILLOP(symlink,  SYMLINK);
     94       1.10  pooka 	FILLOP(readdir,  READDIR);
     95       1.10  pooka 	FILLOP(readlink, READLINK);
     96       1.10  pooka 	FILLOP(reclaim,  RECLAIM);
     97       1.10  pooka 	FILLOP(inactive, INACTIVE);
     98       1.10  pooka 	FILLOP(print,    PRINT);
     99       1.10  pooka 	FILLOP(read,     READ);
    100       1.10  pooka 	FILLOP(write,    WRITE);
    101      1.116   manu 	FILLOP(advlock,  ADVLOCK);
    102       1.99  pooka 	FILLOP(abortop,  ABORTOP);
    103      1.108  pooka 	FILLOP(pathconf, PATHCONF);
    104      1.107  pooka 
    105      1.107  pooka 	FILLOP(getextattr,  GETEXTATTR);
    106      1.107  pooka 	FILLOP(setextattr,  SETEXTATTR);
    107      1.107  pooka 	FILLOP(listextattr, LISTEXTATTR);
    108      1.107  pooka 	FILLOP(deleteextattr, DELETEEXTATTR);
    109       1.19  pooka }
    110       1.19  pooka #undef FILLOP
    111       1.19  pooka 
    112       1.79  pooka /*
    113       1.79  pooka  * Go over all framev entries and write everything we can.  This is
    114       1.79  pooka  * mostly for the benefit of delivering "unmount" to the kernel.
    115       1.79  pooka  */
    116       1.79  pooka static void
    117       1.79  pooka finalpush(struct puffs_usermount *pu)
    118       1.79  pooka {
    119       1.79  pooka 	struct puffs_fctrl_io *fio;
    120       1.79  pooka 
    121       1.79  pooka 	LIST_FOREACH(fio, &pu->pu_ios, fio_entries) {
    122       1.79  pooka 		if (fio->stat & FIO_WRGONE)
    123       1.79  pooka 			continue;
    124       1.79  pooka 
    125       1.89  pooka 		puffs__framev_output(pu, fio->fctrl, fio);
    126       1.79  pooka 	}
    127       1.79  pooka }
    128       1.79  pooka 
    129       1.64  pooka /*ARGSUSED*/
    130      1.101  pooka void
    131      1.101  pooka puffs_kernerr_abort(struct puffs_usermount *pu, uint8_t type,
    132       1.92  pooka 	int error, const char *str, puffs_cookie_t cookie)
    133       1.63  pooka {
    134       1.63  pooka 
    135       1.77  pooka 	fprintf(stderr, "abort: type %d, error %d, cookie %p (%s)\n",
    136       1.77  pooka 	    type, error, cookie, str);
    137       1.63  pooka 	abort();
    138       1.63  pooka }
    139       1.63  pooka 
    140      1.101  pooka /*ARGSUSED*/
    141      1.101  pooka void
    142      1.101  pooka puffs_kernerr_log(struct puffs_usermount *pu, uint8_t type,
    143      1.101  pooka 	int error, const char *str, puffs_cookie_t cookie)
    144      1.101  pooka {
    145      1.101  pooka 
    146      1.101  pooka 	syslog(LOG_WARNING, "kernel: type %d, error %d, cookie %p (%s)\n",
    147      1.101  pooka 	    type, error, cookie, str);
    148      1.101  pooka }
    149      1.101  pooka 
    150       1.19  pooka int
    151       1.19  pooka puffs_getselectable(struct puffs_usermount *pu)
    152       1.19  pooka {
    153       1.10  pooka 
    154       1.60  pooka 	return pu->pu_fd;
    155       1.10  pooka }
    156       1.19  pooka 
    157       1.80  pooka uint64_t
    158       1.80  pooka puffs__nextreq(struct puffs_usermount *pu)
    159       1.80  pooka {
    160       1.80  pooka 	uint64_t rv;
    161       1.80  pooka 
    162       1.80  pooka 	PU_LOCK();
    163      1.104  pooka 	rv = pu->pu_nextreq++ | (uint64_t)1<<63;
    164       1.80  pooka 	PU_UNLOCK();
    165       1.80  pooka 
    166       1.80  pooka 	return rv;
    167       1.80  pooka }
    168       1.80  pooka 
    169       1.19  pooka int
    170       1.19  pooka puffs_setblockingmode(struct puffs_usermount *pu, int mode)
    171       1.19  pooka {
    172       1.46  pooka 	int rv, x;
    173       1.46  pooka 
    174       1.74  pooka 	assert(puffs_getstate(pu) == PUFFS_STATE_RUNNING);
    175       1.74  pooka 
    176       1.46  pooka 	if (mode != PUFFSDEV_BLOCK && mode != PUFFSDEV_NONBLOCK) {
    177       1.46  pooka 		errno = EINVAL;
    178       1.46  pooka 		return -1;
    179       1.46  pooka 	}
    180       1.19  pooka 
    181       1.19  pooka 	x = mode;
    182       1.60  pooka 	rv = ioctl(pu->pu_fd, FIONBIO, &x);
    183       1.46  pooka 
    184       1.46  pooka 	if (rv == 0) {
    185       1.46  pooka 		if (mode == PUFFSDEV_BLOCK)
    186       1.46  pooka 			pu->pu_state &= ~PU_ASYNCFD;
    187       1.46  pooka 		else
    188       1.46  pooka 			pu->pu_state |= PU_ASYNCFD;
    189       1.46  pooka 	}
    190       1.46  pooka 
    191       1.46  pooka 	return rv;
    192       1.19  pooka }
    193       1.19  pooka 
    194       1.19  pooka int
    195       1.19  pooka puffs_getstate(struct puffs_usermount *pu)
    196       1.19  pooka {
    197       1.19  pooka 
    198       1.44  pooka 	return pu->pu_state & PU_STATEMASK;
    199       1.19  pooka }
    200       1.19  pooka 
    201       1.19  pooka void
    202       1.19  pooka puffs_setstacksize(struct puffs_usermount *pu, size_t ss)
    203       1.19  pooka {
    204       1.86  pooka 	long psize, minsize;
    205       1.69  pooka 	int stackshift;
    206       1.87  pooka 	int bonus;
    207       1.78  pooka 
    208       1.86  pooka 	assert(puffs_getstate(pu) == PUFFS_STATE_BEFOREMOUNT);
    209       1.86  pooka 
    210       1.78  pooka 	psize = sysconf(_SC_PAGESIZE);
    211       1.87  pooka 	minsize = 4*psize;
    212       1.98  lukem 	if (ss < (size_t)minsize || ss == PUFFS_STACKSIZE_MIN) {
    213       1.86  pooka 		if (ss != PUFFS_STACKSIZE_MIN)
    214       1.86  pooka 			fprintf(stderr, "puffs_setstacksize: adjusting "
    215       1.86  pooka 			    "stacksize to minimum %ld\n", minsize);
    216       1.87  pooka 		ss = 4*psize;
    217       1.78  pooka 	}
    218       1.69  pooka 
    219       1.69  pooka 	stackshift = -1;
    220       1.87  pooka 	bonus = 0;
    221       1.69  pooka 	while (ss) {
    222       1.87  pooka 		if (ss & 0x1)
    223       1.87  pooka 			bonus++;
    224       1.69  pooka 		ss >>= 1;
    225       1.69  pooka 		stackshift++;
    226       1.69  pooka 	}
    227       1.87  pooka 	if (bonus > 1) {
    228       1.87  pooka 		stackshift++;
    229       1.87  pooka 		fprintf(stderr, "puffs_setstacksize: using next power of two: "
    230       1.88  pooka 		    "%d\n", 1<<stackshift);
    231       1.87  pooka 	}
    232       1.87  pooka 
    233       1.69  pooka 	pu->pu_cc_stackshift = stackshift;
    234       1.19  pooka }
    235       1.19  pooka 
    236       1.24  pooka struct puffs_pathobj *
    237       1.24  pooka puffs_getrootpathobj(struct puffs_usermount *pu)
    238       1.19  pooka {
    239       1.19  pooka 	struct puffs_node *pnr;
    240       1.19  pooka 
    241       1.19  pooka 	pnr = pu->pu_pn_root;
    242       1.19  pooka 	if (pnr == NULL) {
    243       1.19  pooka 		errno = ENOENT;
    244       1.24  pooka 		return NULL;
    245       1.19  pooka 	}
    246       1.19  pooka 
    247       1.24  pooka 	return &pnr->pn_po;
    248       1.24  pooka }
    249       1.24  pooka 
    250       1.35  pooka void
    251       1.35  pooka puffs_setroot(struct puffs_usermount *pu, struct puffs_node *pn)
    252       1.35  pooka {
    253       1.35  pooka 
    254       1.35  pooka 	pu->pu_pn_root = pn;
    255       1.35  pooka }
    256       1.35  pooka 
    257       1.35  pooka struct puffs_node *
    258       1.35  pooka puffs_getroot(struct puffs_usermount *pu)
    259       1.35  pooka {
    260       1.35  pooka 
    261       1.35  pooka 	return pu->pu_pn_root;
    262       1.35  pooka }
    263       1.35  pooka 
    264       1.49  pooka void
    265       1.49  pooka puffs_setrootinfo(struct puffs_usermount *pu, enum vtype vt,
    266       1.49  pooka 	vsize_t vsize, dev_t rdev)
    267       1.49  pooka {
    268       1.60  pooka 	struct puffs_kargs *pargs = pu->pu_kargp;
    269       1.49  pooka 
    270       1.60  pooka 	if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT) {
    271       1.49  pooka 		warnx("puffs_setrootinfo: call has effect only "
    272       1.49  pooka 		    "before mount\n");
    273       1.60  pooka 		return;
    274       1.60  pooka 	}
    275       1.49  pooka 
    276       1.49  pooka 	pargs->pa_root_vtype = vt;
    277       1.49  pooka 	pargs->pa_root_vsize = vsize;
    278       1.49  pooka 	pargs->pa_root_rdev = rdev;
    279       1.49  pooka }
    280       1.49  pooka 
    281       1.35  pooka void *
    282       1.35  pooka puffs_getspecific(struct puffs_usermount *pu)
    283       1.35  pooka {
    284       1.35  pooka 
    285       1.35  pooka 	return pu->pu_privdata;
    286       1.35  pooka }
    287       1.35  pooka 
    288       1.93  pooka void
    289       1.93  pooka puffs_setspecific(struct puffs_usermount *pu, void *privdata)
    290       1.93  pooka {
    291       1.93  pooka 
    292       1.93  pooka 	pu->pu_privdata = privdata;
    293       1.93  pooka }
    294       1.93  pooka 
    295      1.100  pooka void
    296      1.100  pooka puffs_setmntinfo(struct puffs_usermount *pu,
    297      1.100  pooka 	const char *mntfromname, const char *puffsname)
    298      1.100  pooka {
    299      1.100  pooka 	struct puffs_kargs *pargs = pu->pu_kargp;
    300      1.100  pooka 
    301      1.100  pooka 	(void)strlcpy(pargs->pa_mntfromname, mntfromname,
    302      1.100  pooka 	    sizeof(pargs->pa_mntfromname));
    303      1.100  pooka 	(void)strlcpy(pargs->pa_typename, puffsname,
    304      1.100  pooka 	    sizeof(pargs->pa_typename));
    305      1.100  pooka }
    306      1.100  pooka 
    307       1.35  pooka size_t
    308       1.35  pooka puffs_getmaxreqlen(struct puffs_usermount *pu)
    309       1.35  pooka {
    310       1.35  pooka 
    311       1.60  pooka 	return pu->pu_maxreqlen;
    312       1.35  pooka }
    313       1.24  pooka 
    314       1.24  pooka void
    315       1.37  pooka puffs_setmaxreqlen(struct puffs_usermount *pu, size_t reqlen)
    316       1.37  pooka {
    317       1.37  pooka 
    318       1.44  pooka 	if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
    319       1.37  pooka 		warnx("puffs_setmaxreqlen: call has effect only "
    320       1.37  pooka 		    "before mount\n");
    321       1.37  pooka 
    322       1.66  pooka 	pu->pu_kargp->pa_maxmsglen = reqlen;
    323       1.37  pooka }
    324       1.37  pooka 
    325       1.37  pooka void
    326       1.38  pooka puffs_setfhsize(struct puffs_usermount *pu, size_t fhsize, int flags)
    327       1.37  pooka {
    328       1.37  pooka 
    329       1.44  pooka 	if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
    330       1.37  pooka 		warnx("puffs_setfhsize: call has effect only before mount\n");
    331       1.37  pooka 
    332       1.60  pooka 	pu->pu_kargp->pa_fhsize = fhsize;
    333       1.60  pooka 	pu->pu_kargp->pa_fhflags = flags;
    334       1.37  pooka }
    335       1.37  pooka 
    336       1.37  pooka void
    337       1.37  pooka puffs_setncookiehash(struct puffs_usermount *pu, int nhash)
    338       1.37  pooka {
    339       1.37  pooka 
    340       1.44  pooka 	if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
    341       1.37  pooka 		warnx("puffs_setfhsize: call has effect only before mount\n");
    342       1.37  pooka 
    343       1.60  pooka 	pu->pu_kargp->pa_nhashbuckets = nhash;
    344       1.37  pooka }
    345       1.37  pooka 
    346       1.37  pooka void
    347       1.24  pooka puffs_set_pathbuild(struct puffs_usermount *pu, pu_pathbuild_fn fn)
    348       1.24  pooka {
    349       1.24  pooka 
    350       1.24  pooka 	pu->pu_pathbuild = fn;
    351       1.24  pooka }
    352       1.24  pooka 
    353       1.24  pooka void
    354       1.24  pooka puffs_set_pathtransform(struct puffs_usermount *pu, pu_pathtransform_fn fn)
    355       1.24  pooka {
    356       1.24  pooka 
    357       1.24  pooka 	pu->pu_pathtransform = fn;
    358       1.24  pooka }
    359       1.24  pooka 
    360       1.24  pooka void
    361       1.24  pooka puffs_set_pathcmp(struct puffs_usermount *pu, pu_pathcmp_fn fn)
    362       1.24  pooka {
    363       1.24  pooka 
    364       1.24  pooka 	pu->pu_pathcmp = fn;
    365       1.24  pooka }
    366       1.24  pooka 
    367       1.24  pooka void
    368       1.24  pooka puffs_set_pathfree(struct puffs_usermount *pu, pu_pathfree_fn fn)
    369       1.24  pooka {
    370       1.19  pooka 
    371       1.24  pooka 	pu->pu_pathfree = fn;
    372       1.19  pooka }
    373       1.19  pooka 
    374       1.24  pooka void
    375       1.24  pooka puffs_set_namemod(struct puffs_usermount *pu, pu_namemod_fn fn)
    376       1.24  pooka {
    377       1.24  pooka 
    378       1.24  pooka 	pu->pu_namemod = fn;
    379       1.24  pooka }
    380       1.19  pooka 
    381       1.40  pooka void
    382       1.63  pooka puffs_set_errnotify(struct puffs_usermount *pu, pu_errnotify_fn fn)
    383       1.63  pooka {
    384       1.63  pooka 
    385       1.63  pooka 	pu->pu_errnotify = fn;
    386       1.63  pooka }
    387       1.63  pooka 
    388       1.63  pooka void
    389       1.81  pooka puffs_set_cmap(struct puffs_usermount *pu, pu_cmap_fn fn)
    390       1.81  pooka {
    391       1.81  pooka 
    392       1.81  pooka 	pu->pu_cmap = fn;
    393       1.81  pooka }
    394       1.81  pooka 
    395       1.81  pooka void
    396       1.46  pooka puffs_ml_setloopfn(struct puffs_usermount *pu, puffs_ml_loop_fn lfn)
    397       1.46  pooka {
    398       1.46  pooka 
    399       1.46  pooka 	pu->pu_ml_lfn = lfn;
    400       1.46  pooka }
    401       1.46  pooka 
    402       1.46  pooka void
    403       1.46  pooka puffs_ml_settimeout(struct puffs_usermount *pu, struct timespec *ts)
    404       1.46  pooka {
    405       1.46  pooka 
    406       1.46  pooka 	if (ts == NULL) {
    407       1.46  pooka 		pu->pu_ml_timep = NULL;
    408       1.46  pooka 	} else {
    409       1.46  pooka 		pu->pu_ml_timeout = *ts;
    410       1.46  pooka 		pu->pu_ml_timep = &pu->pu_ml_timeout;
    411       1.46  pooka 	}
    412       1.46  pooka }
    413       1.46  pooka 
    414       1.46  pooka void
    415       1.71  pooka puffs_set_prepost(struct puffs_usermount *pu,
    416       1.71  pooka 	pu_prepost_fn pre, pu_prepost_fn pst)
    417       1.71  pooka {
    418       1.71  pooka 
    419       1.71  pooka 	pu->pu_oppre = pre;
    420       1.71  pooka 	pu->pu_oppost = pst;
    421       1.71  pooka }
    422       1.71  pooka 
    423       1.71  pooka void
    424       1.40  pooka puffs_setback(struct puffs_cc *pcc, int whatback)
    425       1.40  pooka {
    426       1.79  pooka 	struct puffs_req *preq = puffs__framebuf_getdataptr(pcc->pcc_pb);
    427       1.40  pooka 
    428       1.40  pooka 	assert(PUFFSOP_OPCLASS(preq->preq_opclass) == PUFFSOP_VN && (
    429       1.40  pooka 	    preq->preq_optype == PUFFS_VN_OPEN ||
    430       1.40  pooka 	    preq->preq_optype == PUFFS_VN_MMAP ||
    431       1.40  pooka 	    preq->preq_optype == PUFFS_VN_REMOVE ||
    432       1.54  pooka 	    preq->preq_optype == PUFFS_VN_RMDIR ||
    433       1.54  pooka 	    preq->preq_optype == PUFFS_VN_INACTIVE));
    434       1.40  pooka 
    435       1.40  pooka 	preq->preq_setbacks |= whatback & PUFFS_SETBACK_MASK;
    436       1.40  pooka }
    437       1.40  pooka 
    438       1.36  pooka int
    439       1.75  pooka puffs_daemon(struct puffs_usermount *pu, int nochdir, int noclose)
    440       1.75  pooka {
    441       1.97    dsl 	long int n;
    442       1.75  pooka 	int parent, value, fd;
    443       1.75  pooka 
    444       1.75  pooka 	if (pipe(pu->pu_dpipe) == -1)
    445       1.75  pooka 		return -1;
    446       1.75  pooka 
    447       1.75  pooka 	switch (fork()) {
    448       1.75  pooka 	case -1:
    449       1.75  pooka 		return -1;
    450       1.75  pooka 	case 0:
    451       1.75  pooka 		parent = 0;
    452       1.75  pooka 		break;
    453       1.75  pooka 	default:
    454       1.75  pooka 		parent = 1;
    455       1.75  pooka 		break;
    456       1.75  pooka 	}
    457       1.75  pooka 	pu->pu_state |= PU_PUFFSDAEMON;
    458       1.75  pooka 
    459       1.75  pooka 	if (parent) {
    460       1.95  pooka 		close(pu->pu_dpipe[1]);
    461       1.75  pooka 		n = read(pu->pu_dpipe[0], &value, sizeof(int));
    462       1.75  pooka 		if (n == -1)
    463       1.75  pooka 			err(1, "puffs_daemon");
    464       1.95  pooka 		if (n != sizeof(value))
    465       1.97    dsl 			errx(1, "puffs_daemon got %ld bytes", n);
    466       1.75  pooka 		if (value) {
    467       1.75  pooka 			errno = value;
    468       1.75  pooka 			err(1, "puffs_daemon");
    469       1.75  pooka 		}
    470       1.75  pooka 		exit(0);
    471       1.75  pooka 	} else {
    472       1.75  pooka 		if (setsid() == -1)
    473       1.75  pooka 			goto fail;
    474       1.75  pooka 
    475       1.75  pooka 		if (!nochdir)
    476       1.75  pooka 			chdir("/");
    477       1.75  pooka 
    478       1.75  pooka 		if (!noclose) {
    479       1.75  pooka 			fd = open(_PATH_DEVNULL, O_RDWR, 0);
    480       1.75  pooka 			if (fd == -1)
    481       1.75  pooka 				goto fail;
    482       1.75  pooka 			dup2(fd, STDIN_FILENO);
    483       1.75  pooka 			dup2(fd, STDOUT_FILENO);
    484       1.75  pooka 			dup2(fd, STDERR_FILENO);
    485       1.75  pooka 			if (fd > STDERR_FILENO)
    486       1.75  pooka 				close(fd);
    487       1.75  pooka 		}
    488       1.75  pooka 		return 0;
    489       1.75  pooka 	}
    490       1.75  pooka 
    491       1.75  pooka  fail:
    492       1.75  pooka 	n = write(pu->pu_dpipe[1], &errno, sizeof(int));
    493       1.75  pooka 	assert(n == 4);
    494       1.75  pooka 	return -1;
    495       1.75  pooka }
    496       1.75  pooka 
    497       1.94  pooka static void
    498       1.94  pooka shutdaemon(struct puffs_usermount *pu, int error)
    499       1.94  pooka {
    500       1.94  pooka 	ssize_t n;
    501       1.94  pooka 
    502       1.94  pooka 	n = write(pu->pu_dpipe[1], &error, sizeof(int));
    503       1.94  pooka 	assert(n == 4);
    504       1.94  pooka 	close(pu->pu_dpipe[0]);
    505       1.94  pooka 	close(pu->pu_dpipe[1]);
    506       1.94  pooka 	pu->pu_state &= ~PU_PUFFSDAEMON;
    507       1.94  pooka }
    508       1.94  pooka 
    509       1.75  pooka int
    510       1.49  pooka puffs_mount(struct puffs_usermount *pu, const char *dir, int mntflags,
    511       1.92  pooka 	puffs_cookie_t cookie)
    512       1.36  pooka {
    513       1.75  pooka 	int rv, fd, sverrno;
    514       1.82  pooka 	char *comfd;
    515       1.82  pooka 
    516       1.82  pooka 	pu->pu_kargp->pa_root_cookie = cookie;
    517       1.36  pooka 
    518       1.36  pooka 	/* XXXkludgehere */
    519       1.36  pooka 	/* kauth doesn't provide this service any longer */
    520       1.36  pooka 	if (geteuid() != 0)
    521       1.36  pooka 		mntflags |= MNT_NOSUID | MNT_NODEV;
    522       1.36  pooka 
    523       1.82  pooka 	/*
    524       1.82  pooka 	 * Undocumented...  Well, documented only here.
    525       1.82  pooka 	 *
    526       1.82  pooka 	 * This is used for imaginative purposes.  If the env variable is
    527       1.82  pooka 	 * set, puffs_mount() doesn't do the regular mount procedure.
    528       1.82  pooka 	 * Rather, it crams the mount data down the comfd and sets comfd as
    529       1.82  pooka 	 * the puffs descriptor.
    530       1.82  pooka 	 *
    531       1.82  pooka 	 * This shouldn't be used unless you can read my mind ( ... or write
    532       1.82  pooka 	 * it, not to mention execute it, but that's starting to get silly).
    533       1.82  pooka 	 */
    534       1.82  pooka 	if ((comfd = getenv("PUFFS_COMFD")) != NULL) {
    535       1.82  pooka 		size_t len;
    536       1.82  pooka 
    537       1.82  pooka 		if (sscanf(comfd, "%d", &pu->pu_fd) != 1) {
    538       1.82  pooka 			errno = EINVAL;
    539       1.82  pooka 			rv = -1;
    540       1.82  pooka 			goto out;
    541       1.82  pooka 		}
    542       1.82  pooka 		/* check that what we got at least resembles an fd */
    543       1.82  pooka 		if (fcntl(pu->pu_fd, F_GETFL) == -1) {
    544       1.82  pooka 			rv = -1;
    545       1.82  pooka 			goto out;
    546       1.82  pooka 		}
    547       1.82  pooka 
    548       1.82  pooka #define allwrite(buf, len)						\
    549       1.82  pooka do {									\
    550       1.82  pooka 	ssize_t al_rv;							\
    551       1.82  pooka 	al_rv = write(pu->pu_fd, buf, len);				\
    552       1.98  lukem 	if ((size_t)al_rv != len) {					\
    553       1.82  pooka 		if (al_rv != -1)					\
    554       1.82  pooka 			errno = EIO;					\
    555       1.82  pooka 		rv = -1;						\
    556       1.82  pooka 		goto out;						\
    557       1.82  pooka 	}								\
    558       1.82  pooka } while (/*CONSTCOND*/0)
    559      1.111  pooka 		len = strlen(dir)+1;
    560       1.82  pooka 		allwrite(&len, sizeof(len));
    561       1.82  pooka 		allwrite(dir, len);
    562       1.84  pooka 		len = strlen(pu->pu_kargp->pa_mntfromname)+1;
    563       1.83  pooka 		allwrite(&len, sizeof(len));
    564       1.83  pooka 		allwrite(pu->pu_kargp->pa_mntfromname, len);
    565       1.82  pooka 		allwrite(&mntflags, sizeof(mntflags));
    566      1.111  pooka 		len = sizeof(*pu->pu_kargp);
    567      1.111  pooka 		allwrite(&len, sizeof(len));
    568       1.82  pooka 		allwrite(pu->pu_kargp, sizeof(*pu->pu_kargp));
    569       1.82  pooka 		allwrite(&pu->pu_flags, sizeof(pu->pu_flags));
    570       1.82  pooka #undef allwrite
    571       1.82  pooka 
    572       1.82  pooka 		rv = 0;
    573       1.82  pooka 	} else {
    574      1.113  pooka 		char rp[MAXPATHLEN];
    575      1.113  pooka 
    576      1.113  pooka 		if (realpath(dir, rp) == NULL) {
    577      1.113  pooka 			rv = -1;
    578      1.113  pooka 			goto out;
    579      1.113  pooka 		}
    580      1.113  pooka 
    581      1.113  pooka 		if (strcmp(dir, rp) != 0) {
    582      1.113  pooka 			warnx("puffs_mount: \"%s\" is a relative path.", dir);
    583      1.113  pooka 			warnx("puffs_mount: using \"%s\" instead.", rp);
    584      1.113  pooka 		}
    585      1.113  pooka 
    586       1.82  pooka 		fd = open(_PATH_PUFFS, O_RDWR);
    587       1.82  pooka 		if (fd == -1) {
    588       1.82  pooka 			warnx("puffs_mount: cannot open %s", _PATH_PUFFS);
    589       1.82  pooka 			rv = -1;
    590       1.82  pooka 			goto out;
    591       1.82  pooka 		}
    592       1.82  pooka 		if (fd <= 2)
    593       1.82  pooka 			warnx("puffs_mount: device fd %d (<= 2), sure this is "
    594       1.82  pooka 			    "what you want?", fd);
    595       1.82  pooka 
    596       1.82  pooka 		pu->pu_kargp->pa_fd = pu->pu_fd = fd;
    597       1.82  pooka 		if ((rv = mount(MOUNT_PUFFS, rp, mntflags,
    598       1.82  pooka 		    pu->pu_kargp, sizeof(struct puffs_kargs))) == -1)
    599       1.82  pooka 			goto out;
    600       1.74  pooka 	}
    601       1.74  pooka 
    602       1.49  pooka 	PU_SETSTATE(pu, PUFFS_STATE_RUNNING);
    603       1.36  pooka 
    604       1.60  pooka  out:
    605       1.76  pooka 	if (rv != 0)
    606       1.76  pooka 		sverrno = errno;
    607       1.76  pooka 	else
    608       1.76  pooka 		sverrno = 0;
    609       1.60  pooka 	free(pu->pu_kargp);
    610       1.60  pooka 	pu->pu_kargp = NULL;
    611       1.60  pooka 
    612       1.94  pooka 	if (pu->pu_state & PU_PUFFSDAEMON)
    613       1.94  pooka 		shutdaemon(pu, sverrno);
    614       1.75  pooka 
    615       1.75  pooka 	errno = sverrno;
    616       1.60  pooka 	return rv;
    617       1.36  pooka }
    618       1.10  pooka 
    619        1.1  pooka struct puffs_usermount *
    620      1.107  pooka puffs_init(struct puffs_ops *pops, const char *mntfromname,
    621       1.58  pooka 	const char *puffsname, void *priv, uint32_t pflags)
    622        1.1  pooka {
    623        1.1  pooka 	struct puffs_usermount *pu;
    624       1.36  pooka 	struct puffs_kargs *pargs;
    625       1.74  pooka 	int sverrno;
    626        1.1  pooka 
    627      1.100  pooka 	if (puffsname == PUFFS_DEFER)
    628      1.100  pooka 		puffsname = "n/a";
    629      1.100  pooka 	if (mntfromname == PUFFS_DEFER)
    630      1.100  pooka 		mntfromname = "n/a";
    631      1.100  pooka 	if (priv == PUFFS_DEFER)
    632      1.100  pooka 		priv = NULL;
    633      1.100  pooka 
    634       1.36  pooka 	pu = malloc(sizeof(struct puffs_usermount));
    635       1.36  pooka 	if (pu == NULL)
    636       1.36  pooka 		goto failfree;
    637       1.47  pooka 	memset(pu, 0, sizeof(struct puffs_usermount));
    638        1.1  pooka 
    639       1.60  pooka 	pargs = pu->pu_kargp = malloc(sizeof(struct puffs_kargs));
    640       1.60  pooka 	if (pargs == NULL)
    641       1.60  pooka 		goto failfree;
    642       1.60  pooka 	memset(pargs, 0, sizeof(struct puffs_kargs));
    643       1.60  pooka 
    644      1.107  pooka 	pargs->pa_vers = PUFFSVERSION;
    645       1.36  pooka 	pargs->pa_flags = PUFFS_FLAG_KERN(pflags);
    646      1.107  pooka 	fillvnopmask(pops, pargs);
    647      1.100  pooka 	puffs_setmntinfo(pu, mntfromname, puffsname);
    648        1.1  pooka 
    649       1.49  pooka 	puffs_zerostatvfs(&pargs->pa_svfsb);
    650       1.49  pooka 	pargs->pa_root_cookie = NULL;
    651       1.49  pooka 	pargs->pa_root_vtype = VDIR;
    652       1.49  pooka 	pargs->pa_root_vsize = 0;
    653       1.49  pooka 	pargs->pa_root_rdev = 0;
    654       1.66  pooka 	pargs->pa_maxmsglen = 0;
    655      1.114  pooka 	if (/*CONSTCOND*/ sizeof(time_t) == 4)
    656      1.112  pooka 		pargs->pa_time32 = 1;
    657      1.112  pooka 	else
    658      1.112  pooka 		pargs->pa_time32 = 0;
    659       1.49  pooka 
    660        1.1  pooka 	pu->pu_flags = pflags;
    661       1.13  pooka 	pu->pu_ops = *pops;
    662       1.21  pooka 	free(pops); /* XXX */
    663       1.36  pooka 
    664       1.19  pooka 	pu->pu_privdata = priv;
    665       1.69  pooka 	pu->pu_cc_stackshift = PUFFS_CC_STACKSHIFT_DEFAULT;
    666        1.1  pooka 	LIST_INIT(&pu->pu_pnodelst);
    667       1.79  pooka 	LIST_INIT(&pu->pu_ios);
    668       1.79  pooka 	LIST_INIT(&pu->pu_ios_rmlist);
    669       1.88  pooka 	LIST_INIT(&pu->pu_ccmagazin);
    670       1.67  pooka 	TAILQ_INIT(&pu->pu_sched);
    671        1.1  pooka 
    672       1.89  pooka 	pu->pu_framectrl[PU_FRAMECTRL_FS].rfb = puffs__fsframe_read;
    673       1.89  pooka 	pu->pu_framectrl[PU_FRAMECTRL_FS].wfb = puffs__fsframe_write;
    674       1.89  pooka 	pu->pu_framectrl[PU_FRAMECTRL_FS].cmpfb = puffs__fsframe_cmp;
    675       1.89  pooka 	pu->pu_framectrl[PU_FRAMECTRL_FS].gotfb = puffs__fsframe_gotframe;
    676       1.79  pooka 	pu->pu_framectrl[PU_FRAMECTRL_FS].fdnotfn = puffs_framev_unmountonclose;
    677       1.79  pooka 
    678       1.24  pooka 	/* defaults for some user-settable translation functions */
    679       1.24  pooka 	pu->pu_cmap = NULL; /* identity translation */
    680       1.24  pooka 
    681       1.30  pooka 	pu->pu_pathbuild = puffs_stdpath_buildpath;
    682       1.30  pooka 	pu->pu_pathfree = puffs_stdpath_freepath;
    683       1.30  pooka 	pu->pu_pathcmp = puffs_stdpath_cmppath;
    684       1.24  pooka 	pu->pu_pathtransform = NULL;
    685       1.24  pooka 	pu->pu_namemod = NULL;
    686       1.24  pooka 
    687      1.101  pooka 	pu->pu_errnotify = puffs_kernerr_log;
    688       1.63  pooka 
    689       1.44  pooka 	PU_SETSTATE(pu, PUFFS_STATE_BEFOREMOUNT);
    690        1.1  pooka 
    691        1.1  pooka 	return pu;
    692        1.6  pooka 
    693        1.1  pooka  failfree:
    694        1.6  pooka 	/* can't unmount() from here for obvious reasons */
    695       1.60  pooka 	sverrno = errno;
    696        1.1  pooka 	free(pu);
    697       1.60  pooka 	errno = sverrno;
    698        1.1  pooka 	return NULL;
    699        1.1  pooka }
    700        1.1  pooka 
    701       1.94  pooka void
    702       1.94  pooka puffs_cancel(struct puffs_usermount *pu, int error)
    703       1.94  pooka {
    704       1.94  pooka 
    705       1.94  pooka 	assert(puffs_getstate(pu) < PUFFS_STATE_RUNNING);
    706       1.94  pooka 	shutdaemon(pu, error);
    707       1.94  pooka 	free(pu);
    708       1.94  pooka }
    709       1.94  pooka 
    710       1.36  pooka /*ARGSUSED1*/
    711       1.21  pooka int
    712      1.103  pooka puffs_exit(struct puffs_usermount *pu, int unused /* strict compat */)
    713       1.21  pooka {
    714      1.103  pooka 	struct puffs_framebuf *pb;
    715      1.103  pooka 	struct puffs_req *preq;
    716      1.103  pooka 	void *winp;
    717      1.103  pooka 	size_t winlen;
    718      1.103  pooka 	int sverrno;
    719       1.21  pooka 
    720      1.103  pooka 	pb = puffs_framebuf_make();
    721      1.103  pooka 	if (pb == NULL) {
    722      1.103  pooka 		errno = ENOMEM;
    723      1.103  pooka 		return -1;
    724      1.103  pooka 	}
    725       1.21  pooka 
    726      1.103  pooka 	winlen = sizeof(struct puffs_req);
    727      1.103  pooka 	if (puffs_framebuf_getwindow(pb, 0, &winp, &winlen) == -1) {
    728      1.103  pooka 		sverrno = errno;
    729      1.103  pooka 		puffs_framebuf_destroy(pb);
    730      1.103  pooka 		errno = sverrno;
    731      1.103  pooka 		return -1;
    732      1.103  pooka 	}
    733      1.103  pooka 	preq = winp;
    734       1.21  pooka 
    735      1.103  pooka 	preq->preq_buflen = sizeof(struct puffs_req);
    736      1.103  pooka 	preq->preq_opclass = PUFFSOP_UNMOUNT;
    737      1.103  pooka 	preq->preq_id = puffs__nextreq(pu);
    738       1.44  pooka 
    739      1.103  pooka 	puffs_framev_enqueue_justsend(pu, puffs_getselectable(pu), pb, 1, 0);
    740       1.21  pooka 
    741      1.103  pooka 	return 0;
    742       1.21  pooka }
    743       1.21  pooka 
    744      1.105  pooka /* no sigset_t static intializer */
    745      1.105  pooka static int sigs[NSIG] = { 0, };
    746      1.105  pooka static int sigcatch = 0;
    747      1.105  pooka 
    748      1.105  pooka int
    749      1.105  pooka puffs_unmountonsignal(int sig, bool sigignore)
    750      1.105  pooka {
    751      1.105  pooka 
    752      1.105  pooka 	if (sig < 0 || sig >= (int)NSIG) {
    753      1.105  pooka 		errno = EINVAL;
    754      1.105  pooka 		return -1;
    755      1.105  pooka 	}
    756      1.105  pooka 	if (sigignore)
    757      1.105  pooka 		if (signal(sig, SIG_IGN) == SIG_ERR)
    758      1.105  pooka 			return -1;
    759      1.105  pooka 
    760      1.105  pooka 	if (!sigs[sig])
    761      1.105  pooka 		sigcatch++;
    762      1.105  pooka 	sigs[sig] = 1;
    763      1.105  pooka 
    764      1.105  pooka 	return 0;
    765      1.105  pooka }
    766      1.105  pooka 
    767       1.89  pooka /*
    768       1.89  pooka  * Actual mainloop.  This is called from a context which can block.
    769       1.89  pooka  * It is called either from puffs_mainloop (indirectly, via
    770       1.89  pooka  * puffs_cc_continue() or from puffs_cc_yield()).
    771       1.89  pooka  */
    772       1.89  pooka void
    773       1.89  pooka puffs__theloop(struct puffs_cc *pcc)
    774        1.1  pooka {
    775       1.89  pooka 	struct puffs_usermount *pu = pcc->pcc_pu;
    776       1.79  pooka 	struct puffs_framectrl *pfctrl;
    777       1.44  pooka 	struct puffs_fctrl_io *fio;
    778       1.79  pooka 	struct kevent *curev;
    779       1.44  pooka 	size_t nchanges;
    780       1.89  pooka 	int ndone;
    781       1.44  pooka 
    782       1.89  pooka 	while (puffs_getstate(pu) != PUFFS_STATE_UNMOUNTED) {
    783      1.105  pooka 
    784       1.89  pooka 		/*
    785       1.89  pooka 		 * Schedule existing requests.
    786       1.89  pooka 		 */
    787       1.89  pooka 		while ((pcc = TAILQ_FIRST(&pu->pu_sched)) != NULL) {
    788       1.89  pooka 			TAILQ_REMOVE(&pu->pu_sched, pcc, pcc_schedent);
    789       1.89  pooka 			puffs__goto(pcc);
    790       1.89  pooka 		}
    791       1.19  pooka 
    792       1.46  pooka 		if (pu->pu_ml_lfn)
    793       1.46  pooka 			pu->pu_ml_lfn(pu);
    794       1.46  pooka 
    795       1.79  pooka 		/* XXX: can we still do these optimizations? */
    796       1.79  pooka #if 0
    797       1.55  pooka 		/*
    798       1.55  pooka 		 * Do this here, because:
    799       1.55  pooka 		 *  a) loopfunc might generate some results
    800       1.55  pooka 		 *  b) it's still "after" event handling (except for round 1)
    801       1.55  pooka 		 */
    802       1.55  pooka 		if (puffs_req_putput(ppr) == -1)
    803       1.55  pooka 			goto out;
    804       1.55  pooka 		puffs_req_resetput(ppr);
    805       1.55  pooka 
    806       1.46  pooka 		/* micro optimization: skip kevent syscall if possible */
    807       1.79  pooka 		if (pu->pu_nfds == 1 && pu->pu_ml_timep == NULL
    808       1.46  pooka 		    && (pu->pu_state & PU_ASYNCFD) == 0) {
    809       1.79  pooka 			pfctrl = XXX->fctrl;
    810       1.79  pooka 			puffs_framev_input(pu, pfctrl, XXX);
    811       1.46  pooka 			continue;
    812       1.46  pooka 		}
    813       1.79  pooka #endif
    814       1.55  pooka 
    815       1.46  pooka 		/* else: do full processing */
    816       1.79  pooka 		/* Don't bother worrying about O(n) for now */
    817       1.79  pooka 		LIST_FOREACH(fio, &pu->pu_ios, fio_entries) {
    818       1.46  pooka 			if (fio->stat & FIO_WRGONE)
    819       1.46  pooka 				continue;
    820       1.46  pooka 
    821       1.79  pooka 			pfctrl = fio->fctrl;
    822       1.79  pooka 
    823       1.44  pooka 			/*
    824       1.44  pooka 			 * Try to write out everything to avoid the
    825       1.44  pooka 			 * need for enabling EVFILT_WRITE.  The likely
    826       1.44  pooka 			 * case is that we can fit everything into the
    827       1.44  pooka 			 * socket buffer.
    828       1.44  pooka 			 */
    829       1.89  pooka 			puffs__framev_output(pu, pfctrl, fio);
    830       1.79  pooka 		}
    831       1.79  pooka 
    832       1.79  pooka 		/*
    833       1.79  pooka 		 * Build list of which to enable/disable in writecheck.
    834       1.79  pooka 		 */
    835       1.80  pooka 		nchanges = 0;
    836       1.79  pooka 		LIST_FOREACH(fio, &pu->pu_ios, fio_entries) {
    837       1.79  pooka 			if (fio->stat & FIO_WRGONE)
    838       1.79  pooka 				continue;
    839       1.19  pooka 
    840       1.46  pooka 			/* en/disable write checks for kqueue as needed */
    841       1.44  pooka 			assert((FIO_EN_WRITE(fio) && FIO_RM_WRITE(fio)) == 0);
    842       1.44  pooka 			if (FIO_EN_WRITE(fio)) {
    843       1.79  pooka 				EV_SET(&pu->pu_evs[nchanges], fio->io_fd,
    844       1.44  pooka 				    EVFILT_WRITE, EV_ENABLE, 0, 0,
    845       1.44  pooka 				    (uintptr_t)fio);
    846       1.46  pooka 				fio->stat |= FIO_WR;
    847       1.44  pooka 				nchanges++;
    848       1.44  pooka 			}
    849       1.44  pooka 			if (FIO_RM_WRITE(fio)) {
    850       1.79  pooka 				EV_SET(&pu->pu_evs[nchanges], fio->io_fd,
    851       1.44  pooka 				    EVFILT_WRITE, EV_DISABLE, 0, 0,
    852       1.44  pooka 				    (uintptr_t)fio);
    853       1.46  pooka 				fio->stat &= ~FIO_WR;
    854       1.44  pooka 				nchanges++;
    855       1.44  pooka 			}
    856       1.19  pooka 		}
    857       1.44  pooka 
    858       1.79  pooka 		ndone = kevent(pu->pu_kq, pu->pu_evs, nchanges,
    859      1.105  pooka 		    pu->pu_evs, pu->pu_nevs, pu->pu_ml_timep);
    860       1.50  pooka 
    861       1.50  pooka 		if (ndone == -1) {
    862       1.50  pooka 			if (errno != EINTR)
    863       1.89  pooka 				break;
    864       1.50  pooka 			else
    865       1.50  pooka 				continue;
    866       1.50  pooka 		}
    867       1.50  pooka 
    868       1.50  pooka 		/* uoptimize */
    869       1.46  pooka 		if (ndone == 0)
    870       1.46  pooka 			continue;
    871       1.44  pooka 
    872       1.44  pooka 		/* iterate over the results */
    873       1.79  pooka 		for (curev = pu->pu_evs; ndone--; curev++) {
    874       1.61  pooka 			int what;
    875       1.61  pooka 
    876       1.79  pooka #if 0
    877       1.44  pooka 			/* get & possibly dispatch events from kernel */
    878       1.44  pooka 			if (curev->ident == puffsfd) {
    879       1.44  pooka 				if (puffs_req_handle(pgr, ppr, 0) == -1)
    880       1.44  pooka 					goto out;
    881       1.44  pooka 				continue;
    882       1.44  pooka 			}
    883       1.79  pooka #endif
    884       1.44  pooka 
    885       1.46  pooka 			fio = (void *)curev->udata;
    886      1.105  pooka 			if (__predict_true(fio))
    887      1.105  pooka 				pfctrl = fio->fctrl;
    888      1.105  pooka 			else
    889      1.105  pooka 				pfctrl = NULL;
    890       1.46  pooka 			if (curev->flags & EV_ERROR) {
    891       1.46  pooka 				assert(curev->filter == EVFILT_WRITE);
    892       1.46  pooka 				fio->stat &= ~FIO_WR;
    893       1.46  pooka 
    894       1.46  pooka 				/* XXX: how to know if it's a transient error */
    895       1.89  pooka 				puffs__framev_writeclose(pu, fio,
    896       1.46  pooka 				    (int)curev->data);
    897       1.89  pooka 				puffs__framev_notify(fio, PUFFS_FBIO_ERROR);
    898       1.46  pooka 				continue;
    899       1.44  pooka 			}
    900       1.44  pooka 
    901       1.61  pooka 			what = 0;
    902       1.61  pooka 			if (curev->filter == EVFILT_READ) {
    903       1.89  pooka 				puffs__framev_input(pu, pfctrl, fio);
    904       1.61  pooka 				what |= PUFFS_FBIO_READ;
    905       1.61  pooka 			}
    906       1.56  pooka 
    907       1.61  pooka 			else if (curev->filter == EVFILT_WRITE) {
    908       1.89  pooka 				puffs__framev_output(pu, pfctrl, fio);
    909       1.61  pooka 				what |= PUFFS_FBIO_WRITE;
    910       1.61  pooka 			}
    911      1.105  pooka 
    912      1.105  pooka 			else if (__predict_false(curev->filter==EVFILT_SIGNAL)){
    913      1.105  pooka 				if ((pu->pu_state & PU_DONEXIT) == 0) {
    914      1.105  pooka 					PU_SETSFLAG(pu, PU_DONEXIT);
    915      1.105  pooka 					puffs_exit(pu, 0);
    916      1.105  pooka 				}
    917      1.105  pooka 			}
    918       1.61  pooka 			if (what)
    919       1.89  pooka 				puffs__framev_notify(fio, what);
    920       1.67  pooka 		}
    921       1.67  pooka 
    922       1.67  pooka 		/*
    923       1.46  pooka 		 * Really free fd's now that we don't have references
    924       1.46  pooka 		 * to them.
    925       1.46  pooka 		 */
    926       1.79  pooka 		while ((fio = LIST_FIRST(&pu->pu_ios_rmlist)) != NULL) {
    927       1.46  pooka 			LIST_REMOVE(fio, fio_entries);
    928       1.46  pooka 			free(fio);
    929       1.46  pooka 		}
    930       1.18    alc 	}
    931       1.90  pooka 
    932       1.90  pooka 	if (puffs__cc_restoremain(pu) == -1)
    933       1.90  pooka 		warn("cannot restore main context.  impending doom");
    934       1.89  pooka }
    935       1.89  pooka int
    936       1.89  pooka puffs_mainloop(struct puffs_usermount *pu)
    937       1.89  pooka {
    938       1.89  pooka 	struct puffs_fctrl_io *fio;
    939       1.89  pooka 	struct puffs_cc *pcc;
    940       1.89  pooka 	struct kevent *curev;
    941      1.105  pooka 	size_t nevs;
    942      1.105  pooka 	int sverrno, i;
    943       1.89  pooka 
    944       1.89  pooka 	assert(puffs_getstate(pu) >= PUFFS_STATE_RUNNING);
    945       1.89  pooka 
    946       1.89  pooka 	pu->pu_kq = kqueue();
    947       1.89  pooka 	if (pu->pu_kq == -1)
    948       1.89  pooka 		goto out;
    949       1.89  pooka 	pu->pu_state |= PU_HASKQ;
    950       1.89  pooka 
    951       1.89  pooka 	puffs_setblockingmode(pu, PUFFSDEV_NONBLOCK);
    952       1.89  pooka 	if (puffs__framev_addfd_ctrl(pu, puffs_getselectable(pu),
    953       1.89  pooka 	    PUFFS_FBIO_READ | PUFFS_FBIO_WRITE,
    954       1.89  pooka 	    &pu->pu_framectrl[PU_FRAMECTRL_FS]) == -1)
    955       1.89  pooka 		goto out;
    956       1.89  pooka 
    957      1.105  pooka 	nevs = pu->pu_nevs + sigcatch;
    958      1.105  pooka 	curev = realloc(pu->pu_evs, nevs * sizeof(struct kevent));
    959       1.89  pooka 	if (curev == NULL)
    960       1.89  pooka 		goto out;
    961       1.89  pooka 	pu->pu_evs = curev;
    962      1.105  pooka 	pu->pu_nevs = nevs;
    963       1.89  pooka 
    964       1.89  pooka 	LIST_FOREACH(fio, &pu->pu_ios, fio_entries) {
    965       1.89  pooka 		EV_SET(curev, fio->io_fd, EVFILT_READ, EV_ADD,
    966       1.89  pooka 		    0, 0, (uintptr_t)fio);
    967       1.89  pooka 		curev++;
    968       1.89  pooka 		EV_SET(curev, fio->io_fd, EVFILT_WRITE, EV_ADD | EV_DISABLE,
    969       1.89  pooka 		    0, 0, (uintptr_t)fio);
    970       1.89  pooka 		curev++;
    971       1.89  pooka 	}
    972      1.105  pooka 	for (i = 0; i < NSIG; i++) {
    973      1.105  pooka 		if (sigs[i]) {
    974      1.105  pooka 			EV_SET(curev, i, EVFILT_SIGNAL, EV_ADD | EV_ENABLE,
    975      1.105  pooka 			    0, 0, 0);
    976      1.105  pooka 			curev++;
    977      1.105  pooka 		}
    978      1.105  pooka 	}
    979      1.105  pooka 	assert(curev - pu->pu_evs == (ssize_t)pu->pu_nevs);
    980      1.105  pooka 	if (kevent(pu->pu_kq, pu->pu_evs, pu->pu_nevs, NULL, 0, NULL) == -1)
    981       1.89  pooka 		goto out;
    982       1.89  pooka 
    983       1.89  pooka 	pu->pu_state |= PU_INLOOP;
    984       1.89  pooka 
    985       1.90  pooka 	/*
    986       1.90  pooka 	 * Create alternate execution context and jump to it.  Note
    987       1.90  pooka 	 * that we come "out" of savemain twice.  Where we come out
    988       1.90  pooka 	 * of it depends on the architecture.  If the return address is
    989       1.90  pooka 	 * stored on the stack, we jump out from puffs_cc_continue(),
    990       1.90  pooka 	 * for a register return address from puffs__cc_savemain().
    991       1.90  pooka 	 * PU_MAINRESTORE makes sure we DTRT in both cases.
    992       1.90  pooka 	 */
    993       1.90  pooka 	if (puffs__cc_create(pu, puffs__theloop, &pcc) == -1) {
    994       1.89  pooka 		goto out;
    995       1.90  pooka 	}
    996  1.116.4.1   yamt 
    997  1.116.4.1   yamt #if 0
    998       1.90  pooka 	if (puffs__cc_savemain(pu) == -1) {
    999       1.90  pooka 		goto out;
   1000       1.90  pooka 	}
   1001  1.116.4.1   yamt #else
   1002  1.116.4.1   yamt 	/*
   1003  1.116.4.1   yamt 	 * XXX
   1004  1.116.4.1   yamt 	 * puffs__cc_savemain() uses getcontext() and then returns.
   1005  1.116.4.1   yamt 	 * the caller (this function) may overwrite the stack frame
   1006  1.116.4.1   yamt 	 * of puffs__cc_savemain(), so when we call setcontext() later and
   1007  1.116.4.1   yamt 	 * return from puffs__cc_savemain() again, the return address or
   1008  1.116.4.1   yamt 	 * saved stack pointer can be garbage.
   1009  1.116.4.1   yamt 	 * avoid this by calling getcontext() directly here.
   1010  1.116.4.1   yamt 	 */
   1011  1.116.4.1   yamt 	extern int puffs_fakecc;
   1012  1.116.4.1   yamt 	if (!puffs_fakecc) {
   1013  1.116.4.1   yamt 		PU_CLRSFLAG(pu, PU_MAINRESTORE);
   1014  1.116.4.1   yamt 		if (getcontext(&pu->pu_mainctx) == -1) {
   1015  1.116.4.1   yamt 			goto out;
   1016  1.116.4.1   yamt 		}
   1017  1.116.4.1   yamt 	}
   1018  1.116.4.1   yamt #endif
   1019  1.116.4.1   yamt 
   1020       1.90  pooka 	if ((pu->pu_state & PU_MAINRESTORE) == 0)
   1021       1.90  pooka 		puffs_cc_continue(pcc);
   1022       1.89  pooka 
   1023       1.79  pooka 	finalpush(pu);
   1024       1.44  pooka 	errno = 0;
   1025       1.18    alc 
   1026       1.19  pooka  out:
   1027       1.44  pooka 	/* store the real error for a while */
   1028       1.44  pooka 	sverrno = errno;
   1029       1.44  pooka 
   1030       1.44  pooka 	errno = sverrno;
   1031       1.44  pooka 	if (errno)
   1032       1.44  pooka 		return -1;
   1033       1.44  pooka 	else
   1034       1.44  pooka 		return 0;
   1035        1.1  pooka }
   1036