240 lines
6.1 KiB
C
240 lines
6.1 KiB
C
|
/* $NetBSD: requests.c,v 1.24 2013/01/23 20:22:34 riastradh Exp $ */
|
||
|
|
||
|
/*
|
||
|
* Copyright (c) 2007 Antti Kantee. All Rights Reserved.
|
||
|
*
|
||
|
* Development of this software was supported by the
|
||
|
* Research Foundation of Helsinki University of Technology
|
||
|
*
|
||
|
* Redistribution and use in source and binary forms, with or without
|
||
|
* modification, are permitted provided that the following conditions
|
||
|
* are met:
|
||
|
* 1. Redistributions of source code must retain the above copyright
|
||
|
* notice, this list of conditions and the following disclaimer.
|
||
|
* 2. Redistributions in binary form must reproduce the above copyright
|
||
|
* notice, this list of conditions and the following disclaimer in the
|
||
|
* documentation and/or other materials provided with the distribution.
|
||
|
*
|
||
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
|
||
|
* OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||
|
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
||
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
||
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
||
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
||
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
||
|
* SUCH DAMAGE.
|
||
|
*/
|
||
|
|
||
|
#include <sys/cdefs.h>
|
||
|
#if !defined(lint)
|
||
|
__RCSID("$NetBSD: requests.c,v 1.24 2013/01/23 20:22:34 riastradh Exp $");
|
||
|
#endif /* !lint */
|
||
|
|
||
|
#include <sys/types.h>
|
||
|
#include <sys/ioctl.h>
|
||
|
#include <sys/queue.h>
|
||
|
#include <sys/socket.h>
|
||
|
|
||
|
#include <dev/putter/putter.h>
|
||
|
|
||
|
#include <assert.h>
|
||
|
#include <errno.h>
|
||
|
#include <puffs.h>
|
||
|
#include <stdio.h>
|
||
|
#include <stdlib.h>
|
||
|
#include <unistd.h>
|
||
|
|
||
|
#include "puffs_priv.h"
|
||
|
|
||
|
/*
|
||
|
* Read a frame from the upstream provider. First read the frame
|
||
|
* length and after this read the actual contents. Yes, optimize
|
||
|
* me some day.
|
||
|
*/
|
||
|
/*ARGSUSED*/
|
||
|
int
|
||
|
puffs__fsframe_read(struct puffs_usermount *pu, struct puffs_framebuf *pb,
|
||
|
int fd, int *done)
|
||
|
{
|
||
|
struct putter_hdr phdr;
|
||
|
void *win;
|
||
|
size_t howmuch, winlen, curoff;
|
||
|
ssize_t n;
|
||
|
int lenstate;
|
||
|
|
||
|
/* How much to read? */
|
||
|
the_next_level:
|
||
|
curoff = puffs_framebuf_telloff(pb);
|
||
|
if (curoff < sizeof(struct putter_hdr)) {
|
||
|
howmuch = sizeof(struct putter_hdr) - curoff;
|
||
|
lenstate = 1;
|
||
|
} else {
|
||
|
puffs_framebuf_getdata_atoff(pb, 0, &phdr, sizeof(phdr));
|
||
|
/*LINTED*/
|
||
|
howmuch = phdr.pth_framelen - curoff;
|
||
|
lenstate = 0;
|
||
|
}
|
||
|
|
||
|
if (puffs_framebuf_reserve_space(pb, howmuch) == -1)
|
||
|
return errno;
|
||
|
|
||
|
/* Read contents */
|
||
|
while (howmuch) {
|
||
|
winlen = howmuch;
|
||
|
curoff = puffs_framebuf_telloff(pb);
|
||
|
if (puffs_framebuf_getwindow(pb, curoff, &win, &winlen) == -1)
|
||
|
return errno;
|
||
|
n = read(fd, win, winlen);
|
||
|
switch (n) {
|
||
|
case 0:
|
||
|
return ECONNRESET;
|
||
|
case -1:
|
||
|
if (errno == EAGAIN)
|
||
|
return 0;
|
||
|
return errno;
|
||
|
default:
|
||
|
howmuch -= n;
|
||
|
puffs_framebuf_seekset(pb, curoff + n);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (lenstate)
|
||
|
goto the_next_level;
|
||
|
|
||
|
puffs_framebuf_seekset(pb, 0);
|
||
|
*done = 1;
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Write a frame upstream
|
||
|
*/
|
||
|
/*ARGSUSED*/
|
||
|
int
|
||
|
puffs__fsframe_write(struct puffs_usermount *pu, struct puffs_framebuf *pb,
|
||
|
int fd, int *done)
|
||
|
{
|
||
|
void *win;
|
||
|
uint64_t flen;
|
||
|
size_t winlen, howmuch, curoff;
|
||
|
ssize_t n;
|
||
|
int rv;
|
||
|
|
||
|
/*
|
||
|
* Finalize it if we haven't written anything yet (or we're still
|
||
|
* attempting to write the first byte)
|
||
|
*
|
||
|
* XXX: this shouldn't be here
|
||
|
*/
|
||
|
if (puffs_framebuf_telloff(pb) == 0) {
|
||
|
struct puffs_req *preq;
|
||
|
|
||
|
winlen = sizeof(struct puffs_req);
|
||
|
rv = puffs_framebuf_getwindow(pb, 0, (void *)&preq, &winlen);
|
||
|
if (rv == -1)
|
||
|
return errno;
|
||
|
preq->preq_pth.pth_framelen = flen = preq->preq_buflen;
|
||
|
} else {
|
||
|
struct putter_hdr phdr;
|
||
|
|
||
|
puffs_framebuf_getdata_atoff(pb, 0, &phdr, sizeof(phdr));
|
||
|
flen = phdr.pth_framelen;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Then write it. Chances are if we are talking to the kernel it'll
|
||
|
* just shlosh in all at once, but if we're e.g. talking to the
|
||
|
* network it might take a few tries.
|
||
|
*/
|
||
|
/*LINTED*/
|
||
|
howmuch = flen - puffs_framebuf_telloff(pb);
|
||
|
|
||
|
while (howmuch) {
|
||
|
winlen = howmuch;
|
||
|
curoff = puffs_framebuf_telloff(pb);
|
||
|
if (puffs_framebuf_getwindow(pb, curoff, &win, &winlen) == -1)
|
||
|
return errno;
|
||
|
|
||
|
/*
|
||
|
* XXX: we know from the framebuf implementation that we
|
||
|
* will always managed to map the entire window. But if
|
||
|
* that changes, this will catch it. Then we can do stuff
|
||
|
* iov stuff instead.
|
||
|
*/
|
||
|
assert(winlen == howmuch);
|
||
|
|
||
|
/* XXX: want NOSIGNAL if writing to a pipe */
|
||
|
#if 0
|
||
|
n = send(fd, win, winlen, MSG_NOSIGNAL);
|
||
|
#else
|
||
|
n = write(fd, win, winlen);
|
||
|
#endif
|
||
|
switch (n) {
|
||
|
case 0:
|
||
|
return ECONNRESET;
|
||
|
case -1:
|
||
|
if (errno == EAGAIN)
|
||
|
return 0;
|
||
|
return errno;
|
||
|
default:
|
||
|
howmuch -= n;
|
||
|
puffs_framebuf_seekset(pb, curoff + n);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
*done = 1;
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Compare if "pb1" is a response to a previously sent frame pb2.
|
||
|
* More often than not "pb1" is not a response to anything but
|
||
|
* rather a fresh request from the kernel.
|
||
|
*/
|
||
|
/*ARGSUSED*/
|
||
|
int
|
||
|
puffs__fsframe_cmp(struct puffs_usermount *pu,
|
||
|
struct puffs_framebuf *pb1, struct puffs_framebuf *pb2, int *notresp)
|
||
|
{
|
||
|
struct puffs_req *preq1, *preq2;
|
||
|
size_t winlen;
|
||
|
int rv;
|
||
|
|
||
|
/* map incoming preq */
|
||
|
winlen = sizeof(struct puffs_req);
|
||
|
rv = puffs_framebuf_getwindow(pb1, 0, (void *)&preq1, &winlen);
|
||
|
assert(rv == 0); /* frames are always at least puffs_req in size */
|
||
|
assert(winlen == sizeof(struct puffs_req));
|
||
|
|
||
|
/*
|
||
|
* Check if this is not a response in this slot. That's the
|
||
|
* likely case.
|
||
|
*/
|
||
|
if ((preq1->preq_opclass & PUFFSOPFLAG_ISRESPONSE) == 0) {
|
||
|
*notresp = 1;
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/* map second preq */
|
||
|
winlen = sizeof(struct puffs_req);
|
||
|
rv = puffs_framebuf_getwindow(pb2, 0, (void *)&preq2, &winlen);
|
||
|
assert(rv == 0); /* frames are always at least puffs_req in size */
|
||
|
assert(winlen == sizeof(struct puffs_req));
|
||
|
|
||
|
/* then compare: resid equal? */
|
||
|
return preq1->preq_id != preq2->preq_id;
|
||
|
}
|
||
|
|
||
|
void
|
||
|
puffs__fsframe_gotframe(struct puffs_usermount *pu, struct puffs_framebuf *pb)
|
||
|
{
|
||
|
|
||
|
puffs_framebuf_seekset(pb, 0);
|
||
|
puffs__ml_dispatch(pu, pb);
|
||
|
}
|