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