puffs.c revision 1.62 1 1.62 pooka /* $NetBSD: puffs.c,v 1.62 2007/08/11 18:04:50 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.62 pooka __RCSID("$NetBSD: puffs.c,v 1.62 2007/08/11 18:04:50 pooka Exp $");
35 1.1 pooka #endif /* !lint */
36 1.1 pooka
37 1.1 pooka #include <sys/param.h>
38 1.1 pooka #include <sys/mount.h>
39 1.1 pooka
40 1.9 pooka #include <assert.h>
41 1.12 pooka #include <err.h>
42 1.1 pooka #include <errno.h>
43 1.1 pooka #include <fcntl.h>
44 1.21 pooka #include <mntopts.h>
45 1.58 pooka #include <paths.h>
46 1.1 pooka #include <puffs.h>
47 1.1 pooka #include <stdio.h>
48 1.1 pooka #include <stdlib.h>
49 1.1 pooka #include <string.h>
50 1.1 pooka #include <syslog.h>
51 1.1 pooka #include <unistd.h>
52 1.1 pooka
53 1.19 pooka #include "puffs_priv.h"
54 1.19 pooka
55 1.21 pooka /* Most file systems want this for opts, so just give it to them */
56 1.21 pooka const struct mntopt puffsmopts[] = {
57 1.21 pooka MOPT_STDOPTS,
58 1.21 pooka PUFFSMOPT_STD,
59 1.21 pooka MOPT_NULL,
60 1.21 pooka };
61 1.21 pooka
62 1.10 pooka #define FILLOP(lower, upper) \
63 1.10 pooka do { \
64 1.13 pooka if (pops->puffs_node_##lower) \
65 1.10 pooka opmask[PUFFS_VN_##upper] = 1; \
66 1.10 pooka } while (/*CONSTCOND*/0)
67 1.10 pooka static void
68 1.13 pooka fillvnopmask(struct puffs_ops *pops, uint8_t *opmask)
69 1.10 pooka {
70 1.10 pooka
71 1.10 pooka memset(opmask, 0, PUFFS_VN_MAX);
72 1.10 pooka
73 1.10 pooka FILLOP(create, CREATE);
74 1.10 pooka FILLOP(mknod, MKNOD);
75 1.10 pooka FILLOP(open, OPEN);
76 1.10 pooka FILLOP(close, CLOSE);
77 1.10 pooka FILLOP(access, ACCESS);
78 1.10 pooka FILLOP(getattr, GETATTR);
79 1.10 pooka FILLOP(setattr, SETATTR);
80 1.10 pooka FILLOP(poll, POLL); /* XXX: not ready in kernel */
81 1.16 pooka FILLOP(mmap, MMAP);
82 1.10 pooka FILLOP(fsync, FSYNC);
83 1.10 pooka FILLOP(seek, SEEK);
84 1.10 pooka FILLOP(remove, REMOVE);
85 1.10 pooka FILLOP(link, LINK);
86 1.10 pooka FILLOP(rename, RENAME);
87 1.10 pooka FILLOP(mkdir, MKDIR);
88 1.10 pooka FILLOP(rmdir, RMDIR);
89 1.10 pooka FILLOP(symlink, SYMLINK);
90 1.10 pooka FILLOP(readdir, READDIR);
91 1.10 pooka FILLOP(readlink, READLINK);
92 1.10 pooka FILLOP(reclaim, RECLAIM);
93 1.10 pooka FILLOP(inactive, INACTIVE);
94 1.10 pooka FILLOP(print, PRINT);
95 1.10 pooka FILLOP(read, READ);
96 1.10 pooka FILLOP(write, WRITE);
97 1.10 pooka
98 1.10 pooka /* XXX: not implemented in the kernel */
99 1.10 pooka FILLOP(getextattr, GETEXTATTR);
100 1.10 pooka FILLOP(setextattr, SETEXTATTR);
101 1.10 pooka FILLOP(listextattr, LISTEXTATTR);
102 1.19 pooka }
103 1.19 pooka #undef FILLOP
104 1.19 pooka
105 1.19 pooka int
106 1.19 pooka puffs_getselectable(struct puffs_usermount *pu)
107 1.19 pooka {
108 1.10 pooka
109 1.60 pooka return pu->pu_fd;
110 1.10 pooka }
111 1.19 pooka
112 1.19 pooka int
113 1.19 pooka puffs_setblockingmode(struct puffs_usermount *pu, int mode)
114 1.19 pooka {
115 1.46 pooka int rv, x;
116 1.46 pooka
117 1.46 pooka if (mode != PUFFSDEV_BLOCK && mode != PUFFSDEV_NONBLOCK) {
118 1.46 pooka errno = EINVAL;
119 1.46 pooka return -1;
120 1.46 pooka }
121 1.19 pooka
122 1.19 pooka x = mode;
123 1.60 pooka rv = ioctl(pu->pu_fd, FIONBIO, &x);
124 1.46 pooka
125 1.46 pooka if (rv == 0) {
126 1.46 pooka if (mode == PUFFSDEV_BLOCK)
127 1.46 pooka pu->pu_state &= ~PU_ASYNCFD;
128 1.46 pooka else
129 1.46 pooka pu->pu_state |= PU_ASYNCFD;
130 1.46 pooka }
131 1.46 pooka
132 1.46 pooka return rv;
133 1.19 pooka }
134 1.19 pooka
135 1.19 pooka int
136 1.19 pooka puffs_getstate(struct puffs_usermount *pu)
137 1.19 pooka {
138 1.19 pooka
139 1.44 pooka return pu->pu_state & PU_STATEMASK;
140 1.19 pooka }
141 1.19 pooka
142 1.19 pooka void
143 1.19 pooka puffs_setstacksize(struct puffs_usermount *pu, size_t ss)
144 1.19 pooka {
145 1.19 pooka
146 1.19 pooka pu->pu_cc_stacksize = ss;
147 1.19 pooka }
148 1.19 pooka
149 1.24 pooka struct puffs_pathobj *
150 1.24 pooka puffs_getrootpathobj(struct puffs_usermount *pu)
151 1.19 pooka {
152 1.19 pooka struct puffs_node *pnr;
153 1.19 pooka
154 1.19 pooka pnr = pu->pu_pn_root;
155 1.19 pooka if (pnr == NULL) {
156 1.19 pooka errno = ENOENT;
157 1.24 pooka return NULL;
158 1.19 pooka }
159 1.19 pooka
160 1.24 pooka return &pnr->pn_po;
161 1.24 pooka }
162 1.24 pooka
163 1.35 pooka void
164 1.35 pooka puffs_setroot(struct puffs_usermount *pu, struct puffs_node *pn)
165 1.35 pooka {
166 1.35 pooka
167 1.35 pooka pu->pu_pn_root = pn;
168 1.35 pooka }
169 1.35 pooka
170 1.35 pooka struct puffs_node *
171 1.35 pooka puffs_getroot(struct puffs_usermount *pu)
172 1.35 pooka {
173 1.35 pooka
174 1.35 pooka return pu->pu_pn_root;
175 1.35 pooka }
176 1.35 pooka
177 1.49 pooka void
178 1.49 pooka puffs_setrootinfo(struct puffs_usermount *pu, enum vtype vt,
179 1.49 pooka vsize_t vsize, dev_t rdev)
180 1.49 pooka {
181 1.60 pooka struct puffs_kargs *pargs = pu->pu_kargp;
182 1.49 pooka
183 1.60 pooka if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT) {
184 1.49 pooka warnx("puffs_setrootinfo: call has effect only "
185 1.49 pooka "before mount\n");
186 1.60 pooka return;
187 1.60 pooka }
188 1.49 pooka
189 1.49 pooka pargs->pa_root_vtype = vt;
190 1.49 pooka pargs->pa_root_vsize = vsize;
191 1.49 pooka pargs->pa_root_rdev = rdev;
192 1.49 pooka }
193 1.49 pooka
194 1.35 pooka void *
195 1.35 pooka puffs_getspecific(struct puffs_usermount *pu)
196 1.35 pooka {
197 1.35 pooka
198 1.35 pooka return pu->pu_privdata;
199 1.35 pooka }
200 1.35 pooka
201 1.35 pooka size_t
202 1.35 pooka puffs_getmaxreqlen(struct puffs_usermount *pu)
203 1.35 pooka {
204 1.35 pooka
205 1.60 pooka return pu->pu_maxreqlen;
206 1.35 pooka }
207 1.24 pooka
208 1.24 pooka void
209 1.37 pooka puffs_setmaxreqlen(struct puffs_usermount *pu, size_t reqlen)
210 1.37 pooka {
211 1.37 pooka
212 1.44 pooka if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
213 1.37 pooka warnx("puffs_setmaxreqlen: call has effect only "
214 1.37 pooka "before mount\n");
215 1.37 pooka
216 1.60 pooka pu->pu_kargp->pa_maxreqlen = reqlen;
217 1.37 pooka }
218 1.37 pooka
219 1.37 pooka void
220 1.38 pooka puffs_setfhsize(struct puffs_usermount *pu, size_t fhsize, int flags)
221 1.37 pooka {
222 1.37 pooka
223 1.44 pooka if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
224 1.37 pooka warnx("puffs_setfhsize: call has effect only before mount\n");
225 1.37 pooka
226 1.60 pooka pu->pu_kargp->pa_fhsize = fhsize;
227 1.60 pooka pu->pu_kargp->pa_fhflags = flags;
228 1.37 pooka }
229 1.37 pooka
230 1.37 pooka void
231 1.37 pooka puffs_setncookiehash(struct puffs_usermount *pu, int nhash)
232 1.37 pooka {
233 1.37 pooka
234 1.44 pooka if (puffs_getstate(pu) != PUFFS_STATE_BEFOREMOUNT)
235 1.37 pooka warnx("puffs_setfhsize: call has effect only before mount\n");
236 1.37 pooka
237 1.60 pooka pu->pu_kargp->pa_nhashbuckets = nhash;
238 1.37 pooka }
239 1.37 pooka
240 1.37 pooka void
241 1.24 pooka puffs_set_pathbuild(struct puffs_usermount *pu, pu_pathbuild_fn fn)
242 1.24 pooka {
243 1.24 pooka
244 1.24 pooka pu->pu_pathbuild = fn;
245 1.24 pooka }
246 1.24 pooka
247 1.24 pooka void
248 1.24 pooka puffs_set_pathtransform(struct puffs_usermount *pu, pu_pathtransform_fn fn)
249 1.24 pooka {
250 1.24 pooka
251 1.24 pooka pu->pu_pathtransform = fn;
252 1.24 pooka }
253 1.24 pooka
254 1.24 pooka void
255 1.24 pooka puffs_set_pathcmp(struct puffs_usermount *pu, pu_pathcmp_fn fn)
256 1.24 pooka {
257 1.24 pooka
258 1.24 pooka pu->pu_pathcmp = fn;
259 1.24 pooka }
260 1.24 pooka
261 1.24 pooka void
262 1.24 pooka puffs_set_pathfree(struct puffs_usermount *pu, pu_pathfree_fn fn)
263 1.24 pooka {
264 1.19 pooka
265 1.24 pooka pu->pu_pathfree = fn;
266 1.19 pooka }
267 1.19 pooka
268 1.24 pooka void
269 1.24 pooka puffs_set_namemod(struct puffs_usermount *pu, pu_namemod_fn fn)
270 1.24 pooka {
271 1.24 pooka
272 1.24 pooka pu->pu_namemod = fn;
273 1.24 pooka }
274 1.19 pooka
275 1.40 pooka void
276 1.46 pooka puffs_ml_setloopfn(struct puffs_usermount *pu, puffs_ml_loop_fn lfn)
277 1.46 pooka {
278 1.46 pooka
279 1.46 pooka pu->pu_ml_lfn = lfn;
280 1.46 pooka }
281 1.46 pooka
282 1.46 pooka void
283 1.46 pooka puffs_ml_settimeout(struct puffs_usermount *pu, struct timespec *ts)
284 1.46 pooka {
285 1.46 pooka
286 1.46 pooka if (ts == NULL) {
287 1.46 pooka pu->pu_ml_timep = NULL;
288 1.46 pooka } else {
289 1.46 pooka pu->pu_ml_timeout = *ts;
290 1.46 pooka pu->pu_ml_timep = &pu->pu_ml_timeout;
291 1.46 pooka }
292 1.46 pooka }
293 1.46 pooka
294 1.46 pooka void
295 1.40 pooka puffs_setback(struct puffs_cc *pcc, int whatback)
296 1.40 pooka {
297 1.40 pooka struct puffs_req *preq = pcc->pcc_preq;
298 1.40 pooka
299 1.40 pooka assert(PUFFSOP_OPCLASS(preq->preq_opclass) == PUFFSOP_VN && (
300 1.40 pooka preq->preq_optype == PUFFS_VN_OPEN ||
301 1.40 pooka preq->preq_optype == PUFFS_VN_MMAP ||
302 1.40 pooka preq->preq_optype == PUFFS_VN_REMOVE ||
303 1.54 pooka preq->preq_optype == PUFFS_VN_RMDIR ||
304 1.54 pooka preq->preq_optype == PUFFS_VN_INACTIVE));
305 1.40 pooka
306 1.40 pooka preq->preq_setbacks |= whatback & PUFFS_SETBACK_MASK;
307 1.40 pooka }
308 1.40 pooka
309 1.36 pooka int
310 1.49 pooka puffs_mount(struct puffs_usermount *pu, const char *dir, int mntflags,
311 1.49 pooka void *cookie)
312 1.36 pooka {
313 1.62 pooka char rp[MAXPATHLEN];
314 1.60 pooka int rv;
315 1.36 pooka
316 1.36 pooka #if 1
317 1.36 pooka /* XXXkludgehere */
318 1.36 pooka /* kauth doesn't provide this service any longer */
319 1.36 pooka if (geteuid() != 0)
320 1.36 pooka mntflags |= MNT_NOSUID | MNT_NODEV;
321 1.36 pooka #endif
322 1.36 pooka
323 1.62 pooka if (realpath(dir, rp) == NULL)
324 1.62 pooka return -1;
325 1.62 pooka
326 1.62 pooka if (strcmp(dir, rp) != 0) {
327 1.62 pooka warnx("puffs_mount: \"%s\" is a relative path.", dir);
328 1.62 pooka warnx("puffs_mount: using \"%s\" instead.", rp);
329 1.62 pooka }
330 1.62 pooka
331 1.60 pooka pu->pu_kargp->pa_root_cookie = cookie;
332 1.62 pooka if ((rv = mount(MOUNT_PUFFS, rp, mntflags,
333 1.60 pooka pu->pu_kargp, sizeof(struct puffs_kargs))) == -1)
334 1.60 pooka goto out;
335 1.60 pooka if ((rv = ioctl(pu->pu_fd, PUFFSREQSIZEOP, &pu->pu_maxreqlen)) == -1)
336 1.60 pooka goto out;
337 1.49 pooka PU_SETSTATE(pu, PUFFS_STATE_RUNNING);
338 1.36 pooka
339 1.60 pooka out:
340 1.60 pooka free(pu->pu_kargp);
341 1.60 pooka pu->pu_kargp = NULL;
342 1.60 pooka
343 1.60 pooka return rv;
344 1.36 pooka }
345 1.10 pooka
346 1.1 pooka struct puffs_usermount *
347 1.58 pooka _puffs_init(int develv, struct puffs_ops *pops, const char *mntfromname,
348 1.58 pooka const char *puffsname, void *priv, uint32_t pflags)
349 1.1 pooka {
350 1.1 pooka struct puffs_usermount *pu;
351 1.36 pooka struct puffs_kargs *pargs;
352 1.60 pooka int sverrno, fd;
353 1.1 pooka
354 1.20 pooka if (develv != PUFFS_DEVEL_LIBVERSION) {
355 1.53 pooka warnx("puffs_init: mounting with lib version %d, need %d",
356 1.20 pooka develv, PUFFS_DEVEL_LIBVERSION);
357 1.20 pooka errno = EINVAL;
358 1.20 pooka return NULL;
359 1.20 pooka }
360 1.20 pooka
361 1.58 pooka fd = open(_PATH_PUFFS, O_RDONLY);
362 1.53 pooka if (fd == -1) {
363 1.58 pooka warnx("puffs_init: cannot open %s", _PATH_PUFFS);
364 1.1 pooka return NULL;
365 1.53 pooka }
366 1.12 pooka if (fd <= 2)
367 1.53 pooka warnx("puffs_init: device fd %d (<= 2), sure this is "
368 1.12 pooka "what you want?", fd);
369 1.1 pooka
370 1.36 pooka pu = malloc(sizeof(struct puffs_usermount));
371 1.36 pooka if (pu == NULL)
372 1.36 pooka goto failfree;
373 1.47 pooka memset(pu, 0, sizeof(struct puffs_usermount));
374 1.1 pooka
375 1.60 pooka pargs = pu->pu_kargp = malloc(sizeof(struct puffs_kargs));
376 1.60 pooka if (pargs == NULL)
377 1.60 pooka goto failfree;
378 1.60 pooka memset(pargs, 0, sizeof(struct puffs_kargs));
379 1.60 pooka
380 1.36 pooka pargs->pa_vers = PUFFSDEVELVERS | PUFFSVERSION;
381 1.36 pooka pargs->pa_flags = PUFFS_FLAG_KERN(pflags);
382 1.60 pooka pargs->pa_fd = pu->pu_fd = fd;
383 1.36 pooka fillvnopmask(pops, pargs->pa_vnopmask);
384 1.58 pooka (void)strlcpy(pargs->pa_typename, puffsname,
385 1.58 pooka sizeof(pargs->pa_typename));
386 1.58 pooka (void)strlcpy(pargs->pa_mntfromname, mntfromname,
387 1.58 pooka sizeof(pargs->pa_mntfromname));
388 1.1 pooka
389 1.49 pooka puffs_zerostatvfs(&pargs->pa_svfsb);
390 1.49 pooka pargs->pa_root_cookie = NULL;
391 1.49 pooka pargs->pa_root_vtype = VDIR;
392 1.49 pooka pargs->pa_root_vsize = 0;
393 1.49 pooka pargs->pa_root_rdev = 0;
394 1.59 pooka pargs->pa_maxreqlen = 0;
395 1.49 pooka
396 1.1 pooka pu->pu_flags = pflags;
397 1.13 pooka pu->pu_ops = *pops;
398 1.21 pooka free(pops); /* XXX */
399 1.36 pooka
400 1.19 pooka pu->pu_privdata = priv;
401 1.19 pooka pu->pu_cc_stacksize = PUFFS_CC_STACKSIZE_DEFAULT;
402 1.1 pooka LIST_INIT(&pu->pu_pnodelst);
403 1.44 pooka LIST_INIT(&pu->pu_framectrl.fb_ios);
404 1.46 pooka LIST_INIT(&pu->pu_ccnukelst);
405 1.1 pooka
406 1.24 pooka /* defaults for some user-settable translation functions */
407 1.24 pooka pu->pu_cmap = NULL; /* identity translation */
408 1.24 pooka
409 1.30 pooka pu->pu_pathbuild = puffs_stdpath_buildpath;
410 1.30 pooka pu->pu_pathfree = puffs_stdpath_freepath;
411 1.30 pooka pu->pu_pathcmp = puffs_stdpath_cmppath;
412 1.24 pooka pu->pu_pathtransform = NULL;
413 1.24 pooka pu->pu_namemod = NULL;
414 1.24 pooka
415 1.44 pooka PU_SETSTATE(pu, PUFFS_STATE_BEFOREMOUNT);
416 1.1 pooka
417 1.1 pooka return pu;
418 1.6 pooka
419 1.1 pooka failfree:
420 1.6 pooka /* can't unmount() from here for obvious reasons */
421 1.60 pooka sverrno = errno;
422 1.36 pooka close(fd);
423 1.1 pooka free(pu);
424 1.60 pooka errno = sverrno;
425 1.1 pooka return NULL;
426 1.1 pooka }
427 1.1 pooka
428 1.21 pooka /*
429 1.21 pooka * XXX: there's currently no clean way to request unmount from
430 1.21 pooka * within the user server, so be very brutal about it.
431 1.21 pooka */
432 1.36 pooka /*ARGSUSED1*/
433 1.21 pooka int
434 1.21 pooka puffs_exit(struct puffs_usermount *pu, int force)
435 1.21 pooka {
436 1.44 pooka struct puffs_node *pn;
437 1.21 pooka
438 1.21 pooka force = 1; /* currently */
439 1.21 pooka
440 1.60 pooka if (pu->pu_fd)
441 1.60 pooka close(pu->pu_fd);
442 1.21 pooka
443 1.44 pooka while ((pn = LIST_FIRST(&pu->pu_pnodelst)) != NULL)
444 1.24 pooka puffs_pn_put(pn);
445 1.44 pooka
446 1.46 pooka puffs_framev_exit(pu);
447 1.44 pooka if (pu->pu_haskq)
448 1.44 pooka close(pu->pu_kq);
449 1.21 pooka free(pu);
450 1.21 pooka
451 1.21 pooka return 0; /* always succesful for now, WILL CHANGE */
452 1.21 pooka }
453 1.21 pooka
454 1.21 pooka int
455 1.9 pooka puffs_mainloop(struct puffs_usermount *pu, int flags)
456 1.1 pooka {
457 1.44 pooka struct puffs_getreq *pgr = NULL;
458 1.44 pooka struct puffs_putreq *ppr = NULL;
459 1.44 pooka struct puffs_framectrl *pfctrl = &pu->pu_framectrl;
460 1.44 pooka struct puffs_fctrl_io *fio;
461 1.44 pooka struct kevent *curev, *newevs;
462 1.44 pooka size_t nchanges;
463 1.44 pooka int puffsfd, sverrno;
464 1.44 pooka int ndone;
465 1.44 pooka
466 1.44 pooka assert(puffs_getstate(pu) >= PUFFS_STATE_RUNNING);
467 1.1 pooka
468 1.36 pooka pgr = puffs_req_makeget(pu, puffs_getmaxreqlen(pu), 0);
469 1.19 pooka if (pgr == NULL)
470 1.44 pooka goto out;
471 1.18 alc
472 1.26 pooka ppr = puffs_req_makeput(pu);
473 1.44 pooka if (ppr == NULL)
474 1.44 pooka goto out;
475 1.44 pooka
476 1.44 pooka newevs = realloc(pfctrl->evs, (2*pfctrl->nfds+1)*sizeof(struct kevent));
477 1.44 pooka if (newevs == NULL)
478 1.44 pooka goto out;
479 1.44 pooka pfctrl->evs = newevs;
480 1.19 pooka
481 1.19 pooka if ((flags & PUFFSLOOP_NODAEMON) == 0)
482 1.22 pooka if (daemon(1, 0) == -1)
483 1.19 pooka goto out;
484 1.44 pooka pu->pu_state |= PU_INLOOP;
485 1.44 pooka
486 1.44 pooka pu->pu_kq = kqueue();
487 1.44 pooka if (pu->pu_kq == -1)
488 1.44 pooka goto out;
489 1.44 pooka pu->pu_haskq = 1;
490 1.44 pooka
491 1.44 pooka curev = pfctrl->evs;
492 1.44 pooka LIST_FOREACH(fio, &pfctrl->fb_ios, fio_entries) {
493 1.44 pooka EV_SET(curev, fio->io_fd, EVFILT_READ, EV_ADD,
494 1.44 pooka 0, 0, (uintptr_t)fio);
495 1.44 pooka curev++;
496 1.44 pooka EV_SET(curev, fio->io_fd, EVFILT_WRITE, EV_ADD | EV_DISABLE,
497 1.44 pooka 0, 0, (uintptr_t)fio);
498 1.44 pooka curev++;
499 1.44 pooka }
500 1.44 pooka puffsfd = puffs_getselectable(pu);
501 1.44 pooka EV_SET(curev, puffsfd, EVFILT_READ, EV_ADD, 0, 0, 0);
502 1.44 pooka if (kevent(pu->pu_kq, pfctrl->evs, 2*pfctrl->nfds+1,
503 1.44 pooka NULL, 0, NULL) == -1)
504 1.44 pooka goto out;
505 1.44 pooka
506 1.44 pooka while (puffs_getstate(pu) != PUFFS_STATE_UNMOUNTED) {
507 1.46 pooka if (pu->pu_ml_lfn)
508 1.46 pooka pu->pu_ml_lfn(pu);
509 1.46 pooka
510 1.55 pooka /*
511 1.55 pooka * Do this here, because:
512 1.55 pooka * a) loopfunc might generate some results
513 1.55 pooka * b) it's still "after" event handling (except for round 1)
514 1.55 pooka */
515 1.55 pooka if (puffs_req_putput(ppr) == -1)
516 1.55 pooka goto out;
517 1.55 pooka puffs_req_resetput(ppr);
518 1.55 pooka
519 1.46 pooka /* micro optimization: skip kevent syscall if possible */
520 1.46 pooka if (pfctrl->nfds == 0 && pu->pu_ml_timep == NULL
521 1.46 pooka && (pu->pu_state & PU_ASYNCFD) == 0) {
522 1.46 pooka if (puffs_req_handle(pgr, ppr, 0) == -1)
523 1.46 pooka goto out;
524 1.46 pooka continue;
525 1.46 pooka }
526 1.55 pooka
527 1.46 pooka /* else: do full processing */
528 1.1 pooka
529 1.44 pooka /*
530 1.44 pooka * Build list of which to enable/disable in writecheck.
531 1.44 pooka * Don't bother worrying about O(n) for now.
532 1.44 pooka */
533 1.44 pooka nchanges = 0;
534 1.44 pooka LIST_FOREACH(fio, &pfctrl->fb_ios, fio_entries) {
535 1.46 pooka if (fio->stat & FIO_WRGONE)
536 1.46 pooka continue;
537 1.46 pooka
538 1.44 pooka /*
539 1.44 pooka * Try to write out everything to avoid the
540 1.44 pooka * need for enabling EVFILT_WRITE. The likely
541 1.44 pooka * case is that we can fit everything into the
542 1.44 pooka * socket buffer.
543 1.44 pooka */
544 1.56 pooka if (puffs_framev_output(pu, pfctrl, fio, ppr)) {
545 1.46 pooka /* need kernel notify? (error condition) */
546 1.46 pooka if (puffs_req_putput(ppr) == -1)
547 1.46 pooka goto out;
548 1.46 pooka puffs_req_resetput(ppr);
549 1.46 pooka }
550 1.19 pooka
551 1.46 pooka /* en/disable write checks for kqueue as needed */
552 1.44 pooka assert((FIO_EN_WRITE(fio) && FIO_RM_WRITE(fio)) == 0);
553 1.44 pooka if (FIO_EN_WRITE(fio)) {
554 1.46 pooka EV_SET(&pfctrl->evs[nchanges], fio->io_fd,
555 1.44 pooka EVFILT_WRITE, EV_ENABLE, 0, 0,
556 1.44 pooka (uintptr_t)fio);
557 1.46 pooka fio->stat |= FIO_WR;
558 1.44 pooka nchanges++;
559 1.44 pooka }
560 1.44 pooka if (FIO_RM_WRITE(fio)) {
561 1.46 pooka EV_SET(&pfctrl->evs[nchanges], fio->io_fd,
562 1.44 pooka EVFILT_WRITE, EV_DISABLE, 0, 0,
563 1.44 pooka (uintptr_t)fio);
564 1.46 pooka fio->stat &= ~FIO_WR;
565 1.44 pooka nchanges++;
566 1.44 pooka }
567 1.46 pooka assert(nchanges <= pfctrl->nfds);
568 1.19 pooka }
569 1.44 pooka
570 1.46 pooka ndone = kevent(pu->pu_kq, pfctrl->evs, nchanges,
571 1.46 pooka pfctrl->evs, pfctrl->nfds+1, pu->pu_ml_timep);
572 1.50 pooka
573 1.50 pooka if (ndone == -1) {
574 1.50 pooka if (errno != EINTR)
575 1.50 pooka goto out;
576 1.50 pooka else
577 1.50 pooka continue;
578 1.50 pooka }
579 1.50 pooka
580 1.50 pooka /* uoptimize */
581 1.46 pooka if (ndone == 0)
582 1.46 pooka continue;
583 1.44 pooka
584 1.44 pooka /* iterate over the results */
585 1.44 pooka for (curev = pfctrl->evs; ndone--; curev++) {
586 1.61 pooka int what;
587 1.61 pooka
588 1.44 pooka /* get & possibly dispatch events from kernel */
589 1.44 pooka if (curev->ident == puffsfd) {
590 1.44 pooka if (puffs_req_handle(pgr, ppr, 0) == -1)
591 1.44 pooka goto out;
592 1.44 pooka continue;
593 1.44 pooka }
594 1.44 pooka
595 1.46 pooka fio = (void *)curev->udata;
596 1.46 pooka if (curev->flags & EV_ERROR) {
597 1.46 pooka assert(curev->filter == EVFILT_WRITE);
598 1.46 pooka fio->stat &= ~FIO_WR;
599 1.46 pooka
600 1.46 pooka /* XXX: how to know if it's a transient error */
601 1.46 pooka puffs_framev_writeclose(pu, fio,
602 1.46 pooka (int)curev->data);
603 1.61 pooka puffs_framev_notify(fio, PUFFS_FBIO_ERROR);
604 1.46 pooka continue;
605 1.44 pooka }
606 1.44 pooka
607 1.61 pooka what = 0;
608 1.61 pooka if (curev->filter == EVFILT_READ) {
609 1.56 pooka puffs_framev_input(pu, pfctrl, fio, ppr);
610 1.61 pooka what |= PUFFS_FBIO_READ;
611 1.61 pooka }
612 1.56 pooka
613 1.61 pooka else if (curev->filter == EVFILT_WRITE) {
614 1.56 pooka puffs_framev_output(pu, pfctrl, fio, ppr);
615 1.61 pooka what |= PUFFS_FBIO_WRITE;
616 1.61 pooka }
617 1.61 pooka if (what)
618 1.61 pooka puffs_framev_notify(fio, what);
619 1.19 pooka }
620 1.44 pooka
621 1.46 pooka /*
622 1.46 pooka * Really free fd's now that we don't have references
623 1.46 pooka * to them.
624 1.46 pooka */
625 1.46 pooka while ((fio = LIST_FIRST(&pfctrl->fb_ios_rmlist)) != NULL) {
626 1.46 pooka LIST_REMOVE(fio, fio_entries);
627 1.46 pooka free(fio);
628 1.46 pooka }
629 1.18 alc }
630 1.44 pooka errno = 0;
631 1.51 pooka puffs_req_putput(ppr);
632 1.18 alc
633 1.19 pooka out:
634 1.44 pooka /* store the real error for a while */
635 1.44 pooka sverrno = errno;
636 1.44 pooka
637 1.44 pooka if (ppr)
638 1.44 pooka puffs_req_destroyput(ppr);
639 1.44 pooka if (pgr)
640 1.44 pooka puffs_req_destroyget(pgr);
641 1.44 pooka
642 1.44 pooka errno = sverrno;
643 1.44 pooka if (errno)
644 1.44 pooka return -1;
645 1.44 pooka else
646 1.44 pooka return 0;
647 1.1 pooka }
648