framebuf.c revision 1.14 1 1.14 pooka /* $NetBSD: framebuf.c,v 1.14 2007/07/07 21:13:41 pooka Exp $ */
2 1.1 pooka
3 1.1 pooka /*
4 1.1 pooka * Copyright (c) 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 * Finnish Cultural Foundation.
8 1.1 pooka *
9 1.1 pooka * Redistribution and use in source and binary forms, with or without
10 1.1 pooka * modification, are permitted provided that the following conditions
11 1.1 pooka * are met:
12 1.1 pooka * 1. Redistributions of source code must retain the above copyright
13 1.1 pooka * notice, this list of conditions and the following disclaimer.
14 1.1 pooka * 2. Redistributions in binary form must reproduce the above copyright
15 1.1 pooka * notice, this list of conditions and the following disclaimer in the
16 1.1 pooka * documentation and/or other materials provided with the distribution.
17 1.1 pooka *
18 1.1 pooka * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
19 1.1 pooka * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20 1.1 pooka * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21 1.1 pooka * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22 1.1 pooka * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 1.1 pooka * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
24 1.1 pooka * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 1.1 pooka * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 1.1 pooka * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 1.1 pooka * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 1.1 pooka * SUCH DAMAGE.
29 1.1 pooka */
30 1.1 pooka
31 1.1 pooka #include <sys/cdefs.h>
32 1.1 pooka #if !defined(lint)
33 1.14 pooka __RCSID("$NetBSD: framebuf.c,v 1.14 2007/07/07 21:13:41 pooka Exp $");
34 1.1 pooka #endif /* !lint */
35 1.1 pooka
36 1.1 pooka #include <sys/types.h>
37 1.1 pooka #include <sys/queue.h>
38 1.1 pooka
39 1.1 pooka #include <assert.h>
40 1.1 pooka #include <errno.h>
41 1.1 pooka #include <poll.h>
42 1.1 pooka #include <puffs.h>
43 1.1 pooka #include <stdlib.h>
44 1.5 pooka #include <unistd.h>
45 1.1 pooka
46 1.1 pooka #include "puffs_priv.h"
47 1.1 pooka
48 1.1 pooka struct puffs_framebuf {
49 1.1 pooka struct puffs_cc *pcc; /* pcc to continue with */
50 1.1 pooka /* OR */
51 1.8 pooka puffs_framev_cb fcb; /* non-blocking callback */
52 1.1 pooka void *fcb_arg; /* argument for previous */
53 1.1 pooka
54 1.1 pooka uint8_t *buf; /* buffer base */
55 1.1 pooka size_t len; /* total length */
56 1.1 pooka
57 1.1 pooka size_t offset; /* cursor, telloff() */
58 1.1 pooka size_t maxoff; /* maximum offset for data, tellsize() */
59 1.1 pooka
60 1.8 pooka volatile int rv; /* errno value */
61 1.1 pooka
62 1.1 pooka int istat;
63 1.1 pooka
64 1.1 pooka TAILQ_ENTRY(puffs_framebuf) pfb_entries;
65 1.1 pooka };
66 1.1 pooka #define ISTAT_NODESTROY 0x01 /* indestructible by framebuf_destroy() */
67 1.1 pooka #define ISTAT_INTERNAL 0x02 /* never leaves library */
68 1.1 pooka #define ISTAT_NOREPLY 0x04 /* nuke after sending */
69 1.14 pooka #define ISTAT_DIRECT 0x08 /* receive directly, no moveinfo */
70 1.1 pooka
71 1.13 pooka #define ISTAT_ONQUEUE ISTAT_NODESTROY /* alias */
72 1.13 pooka
73 1.11 pooka #define PUFBUF_INCRALLOC 4096
74 1.1 pooka #define PUFBUF_REMAIN(p) (p->len - p->offset)
75 1.1 pooka
76 1.5 pooka static struct puffs_fctrl_io *
77 1.5 pooka getfiobyfd(struct puffs_usermount *pu, int fd)
78 1.5 pooka {
79 1.6 pooka struct puffs_framectrl *pfctrl = &pu->pu_framectrl;
80 1.5 pooka struct puffs_fctrl_io *fio;
81 1.5 pooka
82 1.6 pooka LIST_FOREACH(fio, &pfctrl->fb_ios, fio_entries)
83 1.5 pooka if (fio->io_fd == fd)
84 1.5 pooka return fio;
85 1.5 pooka return NULL;
86 1.5 pooka }
87 1.5 pooka
88 1.1 pooka struct puffs_framebuf *
89 1.1 pooka puffs_framebuf_make()
90 1.1 pooka {
91 1.1 pooka struct puffs_framebuf *pufbuf;
92 1.1 pooka
93 1.1 pooka pufbuf = malloc(sizeof(struct puffs_framebuf));
94 1.1 pooka if (pufbuf == NULL)
95 1.1 pooka return NULL;
96 1.1 pooka memset(pufbuf, 0, sizeof(struct puffs_framebuf));
97 1.1 pooka
98 1.1 pooka pufbuf->buf = malloc(PUFBUF_INCRALLOC);
99 1.1 pooka if (pufbuf->buf == NULL) {
100 1.1 pooka free(pufbuf);
101 1.1 pooka return NULL;
102 1.1 pooka }
103 1.11 pooka pufbuf->len = PUFBUF_INCRALLOC;
104 1.1 pooka
105 1.1 pooka puffs_framebuf_recycle(pufbuf);
106 1.1 pooka return pufbuf;
107 1.1 pooka }
108 1.1 pooka
109 1.1 pooka void
110 1.1 pooka puffs_framebuf_destroy(struct puffs_framebuf *pufbuf)
111 1.1 pooka {
112 1.1 pooka
113 1.1 pooka assert((pufbuf->istat & ISTAT_NODESTROY) == 0);
114 1.1 pooka
115 1.1 pooka free(pufbuf->buf);
116 1.1 pooka free(pufbuf);
117 1.1 pooka }
118 1.1 pooka
119 1.1 pooka void
120 1.1 pooka puffs_framebuf_recycle(struct puffs_framebuf *pufbuf)
121 1.1 pooka {
122 1.1 pooka
123 1.1 pooka assert((pufbuf->istat & ISTAT_NODESTROY) == 0);
124 1.1 pooka
125 1.1 pooka pufbuf->offset = 0;
126 1.1 pooka pufbuf->maxoff = 0;
127 1.1 pooka pufbuf->istat = 0;
128 1.1 pooka }
129 1.1 pooka
130 1.1 pooka static int
131 1.1 pooka reservespace(struct puffs_framebuf *pufbuf, size_t off, size_t wantsize)
132 1.1 pooka {
133 1.1 pooka size_t incr;
134 1.1 pooka void *nd;
135 1.1 pooka
136 1.1 pooka if (off <= pufbuf->len && pufbuf->len - off >= wantsize)
137 1.1 pooka return 0;
138 1.1 pooka
139 1.1 pooka for (incr = PUFBUF_INCRALLOC;
140 1.1 pooka pufbuf->len + incr < off + wantsize;
141 1.1 pooka incr += PUFBUF_INCRALLOC)
142 1.1 pooka continue;
143 1.1 pooka
144 1.1 pooka nd = realloc(pufbuf->buf, pufbuf->offset + incr);
145 1.1 pooka if (nd == NULL)
146 1.1 pooka return -1;
147 1.1 pooka
148 1.1 pooka pufbuf->buf = nd;
149 1.1 pooka pufbuf->len += incr;
150 1.1 pooka
151 1.1 pooka return 0;
152 1.1 pooka }
153 1.1 pooka
154 1.1 pooka int
155 1.1 pooka puffs_framebuf_reserve_space(struct puffs_framebuf *pufbuf, size_t wantsize)
156 1.1 pooka {
157 1.1 pooka
158 1.1 pooka return reservespace(pufbuf, pufbuf->offset, wantsize);
159 1.1 pooka }
160 1.1 pooka
161 1.1 pooka int
162 1.1 pooka puffs_framebuf_putdata(struct puffs_framebuf *pufbuf,
163 1.1 pooka const void *data, size_t dlen)
164 1.1 pooka {
165 1.1 pooka
166 1.1 pooka if (PUFBUF_REMAIN(pufbuf) < dlen)
167 1.1 pooka if (puffs_framebuf_reserve_space(pufbuf, dlen) == -1)
168 1.1 pooka return -1;
169 1.1 pooka
170 1.1 pooka memcpy(pufbuf->buf + pufbuf->offset, data, dlen);
171 1.1 pooka pufbuf->offset += dlen;
172 1.1 pooka
173 1.1 pooka if (pufbuf->offset > pufbuf->maxoff)
174 1.1 pooka pufbuf->maxoff = pufbuf->offset;
175 1.1 pooka
176 1.1 pooka return 0;
177 1.1 pooka }
178 1.1 pooka
179 1.1 pooka int
180 1.1 pooka puffs_framebuf_putdata_atoff(struct puffs_framebuf *pufbuf, size_t offset,
181 1.1 pooka const void *data, size_t dlen)
182 1.1 pooka {
183 1.1 pooka
184 1.1 pooka if (reservespace(pufbuf, offset, dlen) == -1)
185 1.1 pooka return -1;
186 1.1 pooka
187 1.1 pooka memcpy(pufbuf->buf + offset, data, dlen);
188 1.1 pooka
189 1.1 pooka if (offset + dlen > pufbuf->maxoff)
190 1.1 pooka pufbuf->maxoff = offset + dlen;
191 1.1 pooka
192 1.1 pooka return 0;
193 1.1 pooka }
194 1.1 pooka
195 1.1 pooka int
196 1.1 pooka puffs_framebuf_getdata(struct puffs_framebuf *pufbuf, void *data, size_t dlen)
197 1.1 pooka {
198 1.1 pooka
199 1.1 pooka if (pufbuf->maxoff < pufbuf->offset + dlen) {
200 1.1 pooka errno = ENOBUFS;
201 1.1 pooka return -1;
202 1.1 pooka }
203 1.1 pooka
204 1.1 pooka memcpy(data, pufbuf->buf + pufbuf->offset, dlen);
205 1.1 pooka pufbuf->offset += dlen;
206 1.1 pooka
207 1.1 pooka return 0;
208 1.1 pooka }
209 1.1 pooka
210 1.1 pooka int
211 1.1 pooka puffs_framebuf_getdata_atoff(struct puffs_framebuf *pufbuf, size_t offset,
212 1.1 pooka void *data, size_t dlen)
213 1.1 pooka {
214 1.1 pooka
215 1.1 pooka if (pufbuf->maxoff < offset + dlen) {
216 1.1 pooka errno = ENOBUFS;
217 1.1 pooka return -1;
218 1.1 pooka }
219 1.1 pooka
220 1.1 pooka memcpy(data, pufbuf->buf + offset, dlen);
221 1.1 pooka return 0;
222 1.1 pooka }
223 1.1 pooka
224 1.1 pooka size_t
225 1.1 pooka puffs_framebuf_telloff(struct puffs_framebuf *pufbuf)
226 1.1 pooka {
227 1.1 pooka
228 1.1 pooka return pufbuf->offset;
229 1.1 pooka }
230 1.1 pooka
231 1.1 pooka size_t
232 1.1 pooka puffs_framebuf_tellsize(struct puffs_framebuf *pufbuf)
233 1.1 pooka {
234 1.1 pooka
235 1.1 pooka return pufbuf->maxoff;
236 1.1 pooka }
237 1.1 pooka
238 1.3 pooka size_t
239 1.3 pooka puffs_framebuf_remaining(struct puffs_framebuf *pufbuf)
240 1.3 pooka {
241 1.3 pooka
242 1.3 pooka return puffs_framebuf_tellsize(pufbuf) - puffs_framebuf_telloff(pufbuf);
243 1.3 pooka }
244 1.3 pooka
245 1.1 pooka int
246 1.1 pooka puffs_framebuf_seekset(struct puffs_framebuf *pufbuf, size_t newoff)
247 1.1 pooka {
248 1.1 pooka
249 1.1 pooka if (reservespace(pufbuf, newoff, 0) == -1)
250 1.1 pooka return -1;
251 1.1 pooka
252 1.1 pooka pufbuf->offset = newoff;
253 1.1 pooka return 0;
254 1.1 pooka }
255 1.1 pooka
256 1.1 pooka int
257 1.1 pooka puffs_framebuf_getwindow(struct puffs_framebuf *pufbuf, size_t winoff,
258 1.1 pooka void **data, size_t *dlen)
259 1.1 pooka {
260 1.1 pooka size_t winlen;
261 1.1 pooka
262 1.1 pooka #ifdef WINTESTING
263 1.1 pooka winlen = MIN(*dlen, 32);
264 1.1 pooka #else
265 1.1 pooka winlen = *dlen;
266 1.1 pooka #endif
267 1.1 pooka
268 1.1 pooka if (reservespace(pufbuf, winoff, winlen) == -1)
269 1.1 pooka return -1;
270 1.1 pooka
271 1.1 pooka *data = pufbuf->buf + winoff;
272 1.1 pooka if (pufbuf->maxoff < winoff + winlen)
273 1.1 pooka pufbuf->maxoff = winoff + winlen;
274 1.1 pooka
275 1.1 pooka return 0;
276 1.1 pooka }
277 1.1 pooka
278 1.8 pooka static void
279 1.8 pooka errnotify(struct puffs_framebuf *pufbuf, int error)
280 1.8 pooka {
281 1.8 pooka
282 1.8 pooka pufbuf->rv = error;
283 1.8 pooka if (pufbuf->pcc) {
284 1.8 pooka puffs_goto(pufbuf->pcc);
285 1.8 pooka } else if (pufbuf->fcb) {
286 1.8 pooka pufbuf->istat &= ~ISTAT_NODESTROY;
287 1.8 pooka pufbuf->fcb(puffs_cc_getusermount(pufbuf->pcc),
288 1.10 pooka pufbuf, pufbuf->fcb_arg, error);
289 1.8 pooka } else {
290 1.8 pooka pufbuf->istat &= ~ISTAT_NODESTROY;
291 1.8 pooka puffs_framebuf_destroy(pufbuf);
292 1.8 pooka }
293 1.8 pooka }
294 1.8 pooka
295 1.8 pooka #define GETFIO(fd) \
296 1.8 pooka do { \
297 1.8 pooka fio = getfiobyfd(pu, fd); \
298 1.8 pooka if (fio == NULL) { \
299 1.8 pooka errno = EINVAL; \
300 1.8 pooka return -1; \
301 1.8 pooka } \
302 1.8 pooka if (fio->stat & FIO_WRGONE) { \
303 1.8 pooka errno = ESHUTDOWN; \
304 1.8 pooka return -1; \
305 1.8 pooka } \
306 1.8 pooka } while (/*CONSTCOND*/0)
307 1.8 pooka
308 1.1 pooka int
309 1.8 pooka puffs_framev_enqueue_cc(struct puffs_cc *pcc, int fd,
310 1.14 pooka struct puffs_framebuf *pufbuf, int flags)
311 1.1 pooka {
312 1.1 pooka struct puffs_usermount *pu = puffs_cc_getusermount(pcc);
313 1.5 pooka struct puffs_fctrl_io *fio;
314 1.5 pooka
315 1.8 pooka /*
316 1.8 pooka * Technically we shouldn't allow this is RDGONE, but it's
317 1.8 pooka * difficult to trap write close without allowing writes.
318 1.8 pooka * And besides, there's probably a disconnect sequence in
319 1.8 pooka * the protocol, so unexpectedly getting a closed fd is
320 1.8 pooka * most likely an error condition.
321 1.8 pooka */
322 1.8 pooka GETFIO(fd);
323 1.1 pooka
324 1.1 pooka pufbuf->pcc = pcc;
325 1.1 pooka pufbuf->fcb = NULL;
326 1.1 pooka pufbuf->fcb_arg = NULL;
327 1.1 pooka
328 1.1 pooka pufbuf->offset = 0;
329 1.1 pooka pufbuf->istat |= ISTAT_NODESTROY;
330 1.1 pooka
331 1.14 pooka if (flags & PUFFS_FBQUEUE_URGENT)
332 1.14 pooka TAILQ_INSERT_HEAD(&fio->snd_qing, pufbuf, pfb_entries);
333 1.14 pooka else
334 1.14 pooka TAILQ_INSERT_TAIL(&fio->snd_qing, pufbuf, pfb_entries);
335 1.1 pooka
336 1.1 pooka puffs_cc_yield(pcc);
337 1.1 pooka if (pufbuf->rv) {
338 1.8 pooka pufbuf->istat &= ~ISTAT_NODESTROY;
339 1.1 pooka errno = pufbuf->rv;
340 1.1 pooka return -1;
341 1.1 pooka }
342 1.1 pooka
343 1.1 pooka return 0;
344 1.1 pooka }
345 1.1 pooka
346 1.5 pooka int
347 1.8 pooka puffs_framev_enqueue_cb(struct puffs_usermount *pu, int fd,
348 1.14 pooka struct puffs_framebuf *pufbuf, puffs_framev_cb fcb, void *arg,
349 1.14 pooka int flags)
350 1.1 pooka {
351 1.5 pooka struct puffs_fctrl_io *fio;
352 1.5 pooka
353 1.8 pooka /* see enqueue_cc */
354 1.8 pooka GETFIO(fd);
355 1.1 pooka
356 1.1 pooka pufbuf->pcc = NULL;
357 1.1 pooka pufbuf->fcb = fcb;
358 1.1 pooka pufbuf->fcb_arg = arg;
359 1.1 pooka
360 1.1 pooka pufbuf->offset = 0;
361 1.1 pooka pufbuf->istat |= ISTAT_NODESTROY;
362 1.1 pooka
363 1.14 pooka if (flags & PUFFS_FBQUEUE_URGENT)
364 1.14 pooka TAILQ_INSERT_HEAD(&fio->snd_qing, pufbuf, pfb_entries);
365 1.14 pooka else
366 1.14 pooka TAILQ_INSERT_TAIL(&fio->snd_qing, pufbuf, pfb_entries);
367 1.5 pooka
368 1.5 pooka return 0;
369 1.1 pooka }
370 1.1 pooka
371 1.5 pooka int
372 1.8 pooka puffs_framev_enqueue_justsend(struct puffs_usermount *pu, int fd,
373 1.14 pooka struct puffs_framebuf *pufbuf, int reply, int flags)
374 1.1 pooka {
375 1.5 pooka struct puffs_fctrl_io *fio;
376 1.5 pooka
377 1.8 pooka GETFIO(fd);
378 1.1 pooka
379 1.1 pooka pufbuf->pcc = NULL;
380 1.1 pooka pufbuf->fcb = NULL;
381 1.1 pooka pufbuf->fcb_arg = NULL;
382 1.1 pooka
383 1.1 pooka pufbuf->offset = 0;
384 1.1 pooka pufbuf->istat |= ISTAT_NODESTROY;
385 1.1 pooka if (!reply)
386 1.1 pooka pufbuf->istat |= ISTAT_NOREPLY;
387 1.1 pooka
388 1.14 pooka if (flags & PUFFS_FBQUEUE_URGENT)
389 1.14 pooka TAILQ_INSERT_HEAD(&fio->snd_qing, pufbuf, pfb_entries);
390 1.14 pooka else
391 1.14 pooka TAILQ_INSERT_TAIL(&fio->snd_qing, pufbuf, pfb_entries);
392 1.14 pooka
393 1.14 pooka return 0;
394 1.14 pooka }
395 1.14 pooka
396 1.14 pooka /* ARGSUSED */
397 1.14 pooka int
398 1.14 pooka puffs_framev_enqueue_directreceive(struct puffs_cc *pcc, int fd,
399 1.14 pooka struct puffs_framebuf *pufbuf, int flags /* used in the future */)
400 1.14 pooka {
401 1.14 pooka struct puffs_usermount *pu = puffs_cc_getusermount(pcc);
402 1.14 pooka struct puffs_fctrl_io *fio;
403 1.14 pooka
404 1.14 pooka fio = getfiobyfd(pu, fd);
405 1.14 pooka if (fio == NULL) {
406 1.14 pooka errno = EINVAL;
407 1.14 pooka return -1;
408 1.14 pooka }
409 1.14 pooka
410 1.14 pooka /* XXX: should have cur_in queue */
411 1.14 pooka assert(fio->cur_in == NULL);
412 1.14 pooka fio->cur_in = pufbuf;
413 1.14 pooka
414 1.14 pooka pufbuf->pcc = pcc;
415 1.14 pooka pufbuf->fcb = NULL;
416 1.14 pooka pufbuf->fcb_arg = NULL;
417 1.14 pooka
418 1.14 pooka pufbuf->offset = 0;
419 1.14 pooka pufbuf->istat |= ISTAT_NODESTROY | ISTAT_DIRECT;
420 1.14 pooka
421 1.14 pooka puffs_cc_yield(pcc);
422 1.14 pooka pufbuf->istat &= ~ISTAT_NODESTROY; /* XXX: not the right place */
423 1.14 pooka if (pufbuf->rv) {
424 1.14 pooka errno = pufbuf->rv;
425 1.14 pooka return -1;
426 1.14 pooka }
427 1.14 pooka
428 1.14 pooka return 0;
429 1.14 pooka }
430 1.14 pooka
431 1.14 pooka int
432 1.14 pooka puffs_framev_enqueue_directsend(struct puffs_cc *pcc, int fd,
433 1.14 pooka struct puffs_framebuf *pufbuf, int flags)
434 1.14 pooka {
435 1.14 pooka struct puffs_usermount *pu = puffs_cc_getusermount(pcc);
436 1.14 pooka struct puffs_fctrl_io *fio;
437 1.14 pooka
438 1.14 pooka if (flags & PUFFS_FBQUEUE_URGENT)
439 1.14 pooka abort(); /* EOPNOTSUPP for now */
440 1.14 pooka
441 1.14 pooka GETFIO(fd);
442 1.14 pooka
443 1.14 pooka pufbuf->pcc = pcc;
444 1.14 pooka pufbuf->fcb = NULL;
445 1.14 pooka pufbuf->fcb_arg = NULL;
446 1.14 pooka
447 1.14 pooka pufbuf->offset = 0;
448 1.14 pooka pufbuf->istat |= ISTAT_NODESTROY | ISTAT_DIRECT;
449 1.14 pooka
450 1.5 pooka TAILQ_INSERT_TAIL(&fio->snd_qing, pufbuf, pfb_entries);
451 1.5 pooka
452 1.14 pooka puffs_cc_yield(pcc);
453 1.14 pooka if (pufbuf->rv) {
454 1.14 pooka pufbuf->istat &= ~ISTAT_NODESTROY;
455 1.14 pooka errno = pufbuf->rv;
456 1.14 pooka return -1;
457 1.14 pooka }
458 1.14 pooka
459 1.5 pooka return 0;
460 1.1 pooka }
461 1.1 pooka
462 1.13 pooka /*
463 1.13 pooka * this beauty shall remain undocumented for now
464 1.13 pooka */
465 1.13 pooka int
466 1.13 pooka puffs_framev_framebuf_ccpromote(struct puffs_framebuf *pufbuf,
467 1.13 pooka struct puffs_cc *pcc)
468 1.13 pooka {
469 1.13 pooka
470 1.13 pooka if ((pufbuf->istat & ISTAT_ONQUEUE) == 0) {
471 1.13 pooka errno = EBUSY;
472 1.13 pooka return -1;
473 1.13 pooka }
474 1.13 pooka
475 1.13 pooka pufbuf->pcc = pcc;
476 1.13 pooka pufbuf->fcb = NULL;
477 1.13 pooka pufbuf->fcb_arg = NULL;
478 1.13 pooka pufbuf->istat &= ~ISTAT_NOREPLY;
479 1.13 pooka
480 1.13 pooka puffs_cc_yield(pcc);
481 1.13 pooka
482 1.13 pooka return 0;
483 1.13 pooka }
484 1.13 pooka
485 1.1 pooka static struct puffs_framebuf *
486 1.1 pooka findbuf(struct puffs_usermount *pu, struct puffs_framectrl *fctrl,
487 1.5 pooka struct puffs_fctrl_io *fio, struct puffs_framebuf *findme)
488 1.1 pooka {
489 1.1 pooka struct puffs_framebuf *cand;
490 1.1 pooka
491 1.5 pooka TAILQ_FOREACH(cand, &fio->res_qing, pfb_entries)
492 1.12 pooka if (fctrl->cmpfb(pu, findme, cand) == 0)
493 1.1 pooka break;
494 1.1 pooka
495 1.1 pooka if (cand == NULL)
496 1.1 pooka return NULL;
497 1.1 pooka
498 1.5 pooka TAILQ_REMOVE(&fio->res_qing, cand, pfb_entries);
499 1.1 pooka return cand;
500 1.1 pooka }
501 1.1 pooka
502 1.1 pooka static void
503 1.1 pooka moveinfo(struct puffs_framebuf *from, struct puffs_framebuf *to)
504 1.1 pooka {
505 1.1 pooka
506 1.1 pooka assert(from->istat & ISTAT_INTERNAL);
507 1.1 pooka
508 1.1 pooka /* migrate buffer */
509 1.1 pooka free(to->buf);
510 1.1 pooka to->buf = from->buf;
511 1.1 pooka from->buf = NULL;
512 1.1 pooka
513 1.1 pooka /* migrate buffer info */
514 1.1 pooka to->len = from->len;
515 1.1 pooka to->offset = from->offset;
516 1.1 pooka to->maxoff = from->maxoff;
517 1.1 pooka }
518 1.1 pooka
519 1.8 pooka void
520 1.8 pooka puffs_framev_input(struct puffs_usermount *pu, struct puffs_framectrl *fctrl,
521 1.5 pooka struct puffs_fctrl_io *fio, struct puffs_putreq *ppr)
522 1.1 pooka {
523 1.1 pooka struct puffs_framebuf *pufbuf, *appbuf;
524 1.1 pooka int rv, complete;
525 1.1 pooka
526 1.14 pooka while ((fio->stat & FIO_DEAD) == 0 && (fio->stat & FIO_ENABLE_R)) {
527 1.5 pooka if ((pufbuf = fio->cur_in) == NULL) {
528 1.1 pooka pufbuf = puffs_framebuf_make();
529 1.1 pooka if (pufbuf == NULL)
530 1.8 pooka return;
531 1.1 pooka pufbuf->istat |= ISTAT_INTERNAL;
532 1.5 pooka fio->cur_in = pufbuf;
533 1.1 pooka }
534 1.1 pooka
535 1.1 pooka complete = 0;
536 1.5 pooka rv = fctrl->rfb(pu, pufbuf, fio->io_fd, &complete);
537 1.1 pooka
538 1.1 pooka /* error */
539 1.1 pooka if (rv) {
540 1.8 pooka puffs_framev_readclose(pu, fio, rv);
541 1.8 pooka return;
542 1.1 pooka }
543 1.1 pooka
544 1.1 pooka /* partial read, come back to fight another day */
545 1.1 pooka if (complete == 0)
546 1.1 pooka break;
547 1.1 pooka
548 1.1 pooka /* else: full read, process */
549 1.14 pooka if ((pufbuf->istat & ISTAT_DIRECT) == 0) {
550 1.14 pooka appbuf = findbuf(pu, fctrl, fio, pufbuf);
551 1.1 pooka
552 1.14 pooka /* XXX: error delivery? */
553 1.14 pooka if (appbuf == NULL) {
554 1.14 pooka /* errno = ENOMSG; */
555 1.14 pooka return;
556 1.14 pooka }
557 1.14 pooka
558 1.14 pooka moveinfo(pufbuf, appbuf);
559 1.14 pooka puffs_framebuf_destroy(pufbuf);
560 1.14 pooka } else {
561 1.14 pooka appbuf = pufbuf;
562 1.1 pooka }
563 1.1 pooka appbuf->istat &= ~ISTAT_NODESTROY;
564 1.14 pooka fio->cur_in = NULL;
565 1.14 pooka
566 1.1 pooka if (appbuf->pcc) {
567 1.1 pooka puffs_docc(appbuf->pcc, ppr);
568 1.1 pooka } else if (appbuf->fcb) {
569 1.10 pooka appbuf->fcb(pu, appbuf, appbuf->fcb_arg, 0);
570 1.1 pooka } else {
571 1.1 pooka puffs_framebuf_destroy(appbuf);
572 1.1 pooka }
573 1.1 pooka
574 1.1 pooka /* hopeless romantics, here we go again */
575 1.1 pooka }
576 1.1 pooka }
577 1.1 pooka
578 1.6 pooka int
579 1.8 pooka puffs_framev_output(struct puffs_usermount *pu, struct puffs_framectrl *fctrl,
580 1.14 pooka struct puffs_fctrl_io *fio, struct puffs_putreq *ppr)
581 1.1 pooka {
582 1.1 pooka struct puffs_framebuf *pufbuf, *pufbuf_next;
583 1.14 pooka int rv, complete, done;
584 1.8 pooka
585 1.8 pooka if (fio->stat & FIO_DEAD)
586 1.8 pooka return 0;
587 1.1 pooka
588 1.14 pooka for (pufbuf = TAILQ_FIRST(&fio->snd_qing), done = 0;
589 1.14 pooka pufbuf && (fio->stat & FIO_DEAD) == 0 && fio->stat & FIO_ENABLE_W;
590 1.1 pooka pufbuf = pufbuf_next) {
591 1.1 pooka complete = 0;
592 1.1 pooka pufbuf_next = TAILQ_NEXT(pufbuf, pfb_entries);
593 1.5 pooka rv = fctrl->wfb(pu, pufbuf, fio->io_fd, &complete);
594 1.1 pooka
595 1.1 pooka if (rv) {
596 1.8 pooka puffs_framev_writeclose(pu, fio, rv);
597 1.14 pooka done = 1;
598 1.8 pooka break;
599 1.1 pooka }
600 1.1 pooka
601 1.1 pooka /* partial write */
602 1.1 pooka if (complete == 0)
603 1.14 pooka return done;
604 1.1 pooka
605 1.1 pooka /* else, complete write */
606 1.5 pooka TAILQ_REMOVE(&fio->snd_qing, pufbuf, pfb_entries);
607 1.1 pooka
608 1.8 pooka /* can't wait for result if we can't read */
609 1.8 pooka if (fio->stat & FIO_RDGONE) {
610 1.8 pooka errnotify(pufbuf, ENXIO);
611 1.14 pooka done = 1;
612 1.14 pooka } else if ((pufbuf->istat & ISTAT_DIRECT)) {
613 1.14 pooka pufbuf->istat &= ~ISTAT_NODESTROY;
614 1.14 pooka puffs_docc(pufbuf->pcc, ppr);
615 1.14 pooka done = 1;
616 1.14 pooka } else if ((pufbuf->istat & ISTAT_NOREPLY) == 0) {
617 1.5 pooka TAILQ_INSERT_TAIL(&fio->res_qing, pufbuf,
618 1.1 pooka pfb_entries);
619 1.1 pooka } else {
620 1.1 pooka pufbuf->istat &= ~ISTAT_NODESTROY;
621 1.1 pooka puffs_framebuf_destroy(pufbuf);
622 1.1 pooka }
623 1.8 pooka
624 1.8 pooka /* omstart! */
625 1.1 pooka }
626 1.1 pooka
627 1.14 pooka return done;
628 1.1 pooka }
629 1.1 pooka
630 1.1 pooka int
631 1.14 pooka puffs_framev_addfd(struct puffs_usermount *pu, int fd, int what)
632 1.5 pooka {
633 1.6 pooka struct puffs_framectrl *pfctrl = &pu->pu_framectrl;
634 1.5 pooka struct puffs_fctrl_io *fio;
635 1.14 pooka struct kevent *newevs;
636 1.5 pooka struct kevent kev[2];
637 1.7 pooka size_t nfds;
638 1.14 pooka int rv, readenable;
639 1.5 pooka
640 1.6 pooka nfds = pfctrl->nfds+1;
641 1.6 pooka newevs = realloc(pfctrl->evs, (2*nfds+1) * sizeof(struct kevent));
642 1.5 pooka if (newevs == NULL)
643 1.5 pooka return -1;
644 1.5 pooka pfctrl->evs = newevs;
645 1.5 pooka
646 1.5 pooka fio = malloc(sizeof(struct puffs_fctrl_io));
647 1.5 pooka if (fio == NULL)
648 1.5 pooka return -1;
649 1.14 pooka fio->io_fd = fd;
650 1.14 pooka fio->cur_in = NULL;
651 1.14 pooka fio->stat = 0;
652 1.14 pooka TAILQ_INIT(&fio->snd_qing);
653 1.14 pooka TAILQ_INIT(&fio->res_qing);
654 1.14 pooka
655 1.14 pooka readenable = 0;
656 1.14 pooka if ((what & PUFFS_FBIO_READ) == 0)
657 1.14 pooka readenable = EV_DISABLE;
658 1.5 pooka
659 1.6 pooka if (pu->pu_state & PU_INLOOP) {
660 1.14 pooka EV_SET(&kev[0], fd, EVFILT_READ,
661 1.14 pooka EV_ADD|readenable, 0, 0, (intptr_t)fio);
662 1.14 pooka EV_SET(&kev[1], fd, EVFILT_WRITE,
663 1.14 pooka EV_ADD|EV_DISABLE, 0, 0, (intptr_t)fio);
664 1.6 pooka rv = kevent(pu->pu_kq, kev, 2, NULL, 0, NULL);
665 1.6 pooka if (rv == -1) {
666 1.6 pooka free(fio);
667 1.6 pooka return -1;
668 1.6 pooka }
669 1.5 pooka }
670 1.14 pooka if (what & PUFFS_FBIO_READ)
671 1.14 pooka fio->stat |= FIO_ENABLE_R;
672 1.14 pooka if (what & PUFFS_FBIO_WRITE)
673 1.14 pooka fio->stat |= FIO_ENABLE_W;
674 1.5 pooka
675 1.5 pooka LIST_INSERT_HEAD(&pfctrl->fb_ios, fio, fio_entries);
676 1.6 pooka pfctrl->nfds = nfds;
677 1.5 pooka
678 1.5 pooka return 0;
679 1.5 pooka }
680 1.5 pooka
681 1.14 pooka /*
682 1.14 pooka * XXX: the following en/disable should be coalesced and executed
683 1.14 pooka * only during the actual keven call. So feel free to fix if
684 1.14 pooka * threatened by mindblowing boredom.
685 1.14 pooka */
686 1.14 pooka
687 1.14 pooka int
688 1.14 pooka puffs_framev_enablefd(struct puffs_usermount *pu, int fd, int what)
689 1.14 pooka {
690 1.14 pooka struct kevent kev;
691 1.14 pooka struct puffs_fctrl_io *fio;
692 1.14 pooka int rv;
693 1.14 pooka
694 1.14 pooka assert((what & (PUFFS_FBIO_READ | PUFFS_FBIO_WRITE)) != 0);
695 1.14 pooka
696 1.14 pooka fio = getfiobyfd(pu, fd);
697 1.14 pooka if (fio == NULL) {
698 1.14 pooka errno = ENXIO;
699 1.14 pooka return -1;
700 1.14 pooka }
701 1.14 pooka
702 1.14 pooka /* write is enabled in the event loop if there is output */
703 1.14 pooka EV_SET(&kev, fd, EVFILT_READ, EV_ENABLE, 0, 0, (uintptr_t)fio);
704 1.14 pooka rv = kevent(pu->pu_kq, &kev, 1, NULL, 0, NULL);
705 1.14 pooka
706 1.14 pooka if (rv == 0) {
707 1.14 pooka if (what & PUFFS_FBIO_READ)
708 1.14 pooka fio->stat |= FIO_ENABLE_R;
709 1.14 pooka if (what & PUFFS_FBIO_WRITE)
710 1.14 pooka fio->stat |= FIO_ENABLE_W;
711 1.14 pooka }
712 1.14 pooka
713 1.14 pooka return rv;
714 1.14 pooka }
715 1.14 pooka
716 1.14 pooka int
717 1.14 pooka puffs_framev_disablefd(struct puffs_usermount *pu, int fd, int what)
718 1.14 pooka {
719 1.14 pooka struct kevent kev[2];
720 1.14 pooka struct puffs_fctrl_io *fio;
721 1.14 pooka size_t i;
722 1.14 pooka int rv;
723 1.14 pooka
724 1.14 pooka assert((what & (PUFFS_FBIO_READ | PUFFS_FBIO_WRITE)) != 0);
725 1.14 pooka
726 1.14 pooka fio = getfiobyfd(pu, fd);
727 1.14 pooka if (fio == NULL) {
728 1.14 pooka errno = ENXIO;
729 1.14 pooka return -1;
730 1.14 pooka }
731 1.14 pooka
732 1.14 pooka i = 0;
733 1.14 pooka
734 1.14 pooka if (what & PUFFS_FBIO_READ) {
735 1.14 pooka EV_SET(&kev[0], fd,
736 1.14 pooka EVFILT_READ, EV_DISABLE, 0, 0, (uintptr_t)fio);
737 1.14 pooka i++;
738 1.14 pooka }
739 1.14 pooka if (what & PUFFS_FBIO_WRITE && fio->stat & FIO_WR) {
740 1.14 pooka EV_SET(&kev[1], fd,
741 1.14 pooka EVFILT_WRITE, EV_DISABLE, 0, 0, (uintptr_t)fio);
742 1.14 pooka i++;
743 1.14 pooka }
744 1.14 pooka rv = kevent(pu->pu_kq, kev, i, NULL, 0, NULL);
745 1.14 pooka
746 1.14 pooka if (rv == 0) {
747 1.14 pooka if (what & PUFFS_FBIO_READ)
748 1.14 pooka fio->stat &= ~FIO_ENABLE_R;
749 1.14 pooka if (what & PUFFS_FBIO_WRITE)
750 1.14 pooka fio->stat &= ~FIO_ENABLE_W;
751 1.14 pooka }
752 1.14 pooka
753 1.14 pooka return rv;
754 1.14 pooka }
755 1.14 pooka
756 1.8 pooka void
757 1.8 pooka puffs_framev_readclose(struct puffs_usermount *pu,
758 1.8 pooka struct puffs_fctrl_io *fio, int error)
759 1.5 pooka {
760 1.5 pooka struct puffs_framebuf *pufbuf;
761 1.8 pooka struct kevent kev;
762 1.8 pooka int notflag;
763 1.8 pooka
764 1.8 pooka if (fio->stat & FIO_RDGONE || fio->stat & FIO_DEAD)
765 1.8 pooka return;
766 1.8 pooka fio->stat |= FIO_RDGONE;
767 1.8 pooka
768 1.8 pooka if (fio->cur_in) {
769 1.14 pooka if ((fio->cur_in->istat & ISTAT_DIRECT) == 0) {
770 1.14 pooka puffs_framebuf_destroy(fio->cur_in);
771 1.14 pooka fio->cur_in = NULL;
772 1.14 pooka } else {
773 1.14 pooka errnotify(fio->cur_in, error);
774 1.14 pooka }
775 1.8 pooka }
776 1.5 pooka
777 1.8 pooka while ((pufbuf = TAILQ_FIRST(&fio->res_qing)) != NULL) {
778 1.8 pooka TAILQ_REMOVE(&fio->res_qing, pufbuf, pfb_entries);
779 1.8 pooka errnotify(pufbuf, error);
780 1.6 pooka }
781 1.5 pooka
782 1.8 pooka EV_SET(&kev, fio->io_fd, EVFILT_READ, EV_DELETE, 0, 0, 0);
783 1.8 pooka (void) kevent(pu->pu_kq, &kev, 1, NULL, 0, NULL);
784 1.8 pooka
785 1.14 pooka notflag = PUFFS_FBIO_READ;
786 1.8 pooka if (fio->stat & FIO_WRGONE)
787 1.14 pooka notflag |= PUFFS_FBIO_WRITE;
788 1.8 pooka
789 1.8 pooka pu->pu_framectrl.fdnotfn(pu, fio->io_fd, notflag);
790 1.8 pooka }
791 1.8 pooka
792 1.8 pooka void
793 1.8 pooka puffs_framev_writeclose(struct puffs_usermount *pu,
794 1.8 pooka struct puffs_fctrl_io *fio, int error)
795 1.8 pooka {
796 1.8 pooka struct puffs_framebuf *pufbuf;
797 1.8 pooka struct kevent kev;
798 1.8 pooka int notflag;
799 1.8 pooka
800 1.8 pooka if (fio->stat & FIO_WRGONE || fio->stat & FIO_DEAD)
801 1.8 pooka return;
802 1.8 pooka fio->stat |= FIO_WRGONE;
803 1.5 pooka
804 1.5 pooka while ((pufbuf = TAILQ_FIRST(&fio->snd_qing)) != NULL) {
805 1.5 pooka TAILQ_REMOVE(&fio->snd_qing, pufbuf, pfb_entries);
806 1.8 pooka errnotify(pufbuf, error);
807 1.5 pooka }
808 1.8 pooka
809 1.8 pooka EV_SET(&kev, fio->io_fd, EVFILT_WRITE, EV_DELETE, 0, 0, 0);
810 1.8 pooka (void) kevent(pu->pu_kq, &kev, 1, NULL, 0, NULL);
811 1.8 pooka
812 1.14 pooka notflag = PUFFS_FBIO_WRITE;
813 1.8 pooka if (fio->stat & FIO_RDGONE)
814 1.14 pooka notflag |= PUFFS_FBIO_READ;
815 1.8 pooka
816 1.8 pooka pu->pu_framectrl.fdnotfn(pu, fio->io_fd, notflag);
817 1.8 pooka }
818 1.8 pooka
819 1.8 pooka static int
820 1.8 pooka removefio(struct puffs_usermount *pu, struct puffs_fctrl_io *fio, int error)
821 1.8 pooka {
822 1.8 pooka struct puffs_framectrl *pfctrl = &pu->pu_framectrl;
823 1.8 pooka
824 1.8 pooka LIST_REMOVE(fio, fio_entries);
825 1.8 pooka if (pu->pu_state & PU_INLOOP) {
826 1.8 pooka puffs_framev_readclose(pu, fio, error);
827 1.8 pooka puffs_framev_writeclose(pu, fio, error);
828 1.5 pooka }
829 1.5 pooka
830 1.5 pooka /* don't bother with realloc */
831 1.6 pooka pfctrl->nfds--;
832 1.5 pooka
833 1.8 pooka /* don't free us yet, might have some references in event arrays */
834 1.8 pooka fio->stat |= FIO_DEAD;
835 1.8 pooka LIST_INSERT_HEAD(&pfctrl->fb_ios_rmlist, fio, fio_entries);
836 1.8 pooka
837 1.5 pooka return 0;
838 1.5 pooka
839 1.5 pooka }
840 1.5 pooka
841 1.5 pooka int
842 1.8 pooka puffs_framev_removefd(struct puffs_usermount *pu, int fd, int error)
843 1.5 pooka {
844 1.5 pooka struct puffs_fctrl_io *fio;
845 1.5 pooka
846 1.5 pooka fio = getfiobyfd(pu, fd);
847 1.5 pooka if (fio == NULL) {
848 1.5 pooka errno = ENXIO;
849 1.5 pooka return -1;
850 1.5 pooka }
851 1.5 pooka
852 1.9 pooka return removefio(pu, fio, error ? error : ECONNRESET);
853 1.8 pooka }
854 1.8 pooka
855 1.8 pooka static void
856 1.8 pooka defaultnot(struct puffs_usermount *pu, int fd, int what)
857 1.8 pooka {
858 1.8 pooka
859 1.8 pooka if (PUFFS_FBGONE_BOTH(what))
860 1.9 pooka (void) puffs_framev_removefd(pu, fd, ECONNRESET);
861 1.5 pooka }
862 1.5 pooka
863 1.8 pooka void
864 1.8 pooka puffs_framev_unmountonclose(struct puffs_usermount *pu, int fd, int what)
865 1.8 pooka {
866 1.8 pooka
867 1.8 pooka /* XXX & X: unmount is non-sensible */
868 1.8 pooka defaultnot(pu, fd, what);
869 1.8 pooka if (PUFFS_FBGONE_BOTH(what))
870 1.8 pooka PU_SETSTATE(pu, PUFFS_STATE_UNMOUNTED);
871 1.8 pooka }
872 1.8 pooka
873 1.8 pooka void
874 1.8 pooka puffs_framev_init(struct puffs_usermount *pu,
875 1.8 pooka puffs_framev_readframe_fn rfb, puffs_framev_writeframe_fn wfb,
876 1.12 pooka puffs_framev_cmpframe_fn cmpfb, puffs_framev_fdnotify_fn fdnotfn)
877 1.1 pooka {
878 1.6 pooka struct puffs_framectrl *pfctrl;
879 1.6 pooka
880 1.6 pooka pfctrl = &pu->pu_framectrl;
881 1.1 pooka pfctrl->rfb = rfb;
882 1.1 pooka pfctrl->wfb = wfb;
883 1.1 pooka pfctrl->cmpfb = cmpfb;
884 1.8 pooka if (fdnotfn)
885 1.8 pooka pfctrl->fdnotfn = fdnotfn;
886 1.8 pooka else
887 1.8 pooka pfctrl->fdnotfn = defaultnot;
888 1.6 pooka }
889 1.1 pooka
890 1.6 pooka void
891 1.8 pooka puffs_framev_exit(struct puffs_usermount *pu)
892 1.6 pooka {
893 1.6 pooka struct puffs_framectrl *pfctrl = &pu->pu_framectrl;
894 1.6 pooka struct puffs_fctrl_io *fio;
895 1.1 pooka
896 1.6 pooka while ((fio = LIST_FIRST(&pfctrl->fb_ios)) != NULL)
897 1.8 pooka removefio(pu, fio, ENXIO);
898 1.6 pooka free(pfctrl->evs);
899 1.1 pooka
900 1.6 pooka /* closing pu->pu_kq takes care of puffsfd */
901 1.1 pooka }
902