puffs.c revision 1.116 1 1.116 manu /* $NetBSD: puffs.c,v 1.116 2011/05/03 13:16:47 manu 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 manu __RCSID("$NetBSD: puffs.c,v 1.116 2011/05/03 13:16:47 manu 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.90 pooka if (puffs__cc_savemain(pu) == -1) {
997 1.90 pooka goto out;
998 1.90 pooka }
999 1.90 pooka if ((pu->pu_state & PU_MAINRESTORE) == 0)
1000 1.90 pooka puffs_cc_continue(pcc);
1001 1.89 pooka
1002 1.79 pooka finalpush(pu);
1003 1.44 pooka errno = 0;
1004 1.18 alc
1005 1.19 pooka out:
1006 1.44 pooka /* store the real error for a while */
1007 1.44 pooka sverrno = errno;
1008 1.44 pooka
1009 1.44 pooka errno = sverrno;
1010 1.44 pooka if (errno)
1011 1.44 pooka return -1;
1012 1.44 pooka else
1013 1.44 pooka return 0;
1014 1.1 pooka }
1015