You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.5 KiB
76 lines
1.5 KiB
3 months ago
|
/*
|
||
|
|
* UCW Library -- Fast Buffered Input on Limited File Descriptors
|
||
|
|
*
|
||
|
|
* (c) 2003--2004 Martin Mares <mj@ucw.cz>
|
||
|
|
*
|
||
|
|
* This software may be freely distributed and used according to the terms
|
||
|
|
* of the GNU Lesser General Public License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <ucw/lib.h>
|
||
|
|
#include <ucw/fastbuf.h>
|
||
|
|
|
||
|
|
#include <unistd.h>
|
||
|
|
|
||
|
|
struct fb_limfd {
|
||
|
|
struct fastbuf fb;
|
||
|
|
int fd; /* File descriptor */
|
||
|
|
int limit;
|
||
|
|
};
|
||
|
|
#define FB_LIMFD(f) ((struct fb_limfd *)(f))
|
||
|
|
#define FB_BUFFER(f) (byte *)(FB_LIMFD(f) + 1)
|
||
|
|
|
||
|
|
static int
|
||
|
|
bfl_refill(struct fastbuf *f)
|
||
|
|
{
|
||
|
|
f->bptr = f->buffer = FB_BUFFER(f);
|
||
|
|
int max = MIN(FB_LIMFD(f)->limit - f->pos, f->bufend - f->buffer);
|
||
|
|
int l = read(FB_LIMFD(f)->fd, f->buffer, max);
|
||
|
|
if (l < 0)
|
||
|
|
bthrow(f, "read", "Error reading %s: %m", f->name);
|
||
|
|
f->bstop = f->buffer + l;
|
||
|
|
f->pos += l;
|
||
|
|
return l;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void
|
||
|
|
bfl_close(struct fastbuf *f)
|
||
|
|
{
|
||
|
|
xfree(f);
|
||
|
|
}
|
||
|
|
|
||
|
|
struct fastbuf *
|
||
|
|
bopen_limited_fd(int fd, uint buflen, uint limit)
|
||
|
|
{
|
||
|
|
struct fb_limfd *F = xmalloc(sizeof(struct fb_limfd) + buflen);
|
||
|
|
struct fastbuf *f = &F->fb;
|
||
|
|
|
||
|
|
bzero(F, sizeof(*F));
|
||
|
|
f->buffer = (char *)(F+1);
|
||
|
|
f->bptr = f->bstop = f->buffer;
|
||
|
|
f->bufend = f->buffer + buflen;
|
||
|
|
f->name = "limited-fd";
|
||
|
|
F->fd = fd;
|
||
|
|
F->limit = limit;
|
||
|
|
f->refill = bfl_refill;
|
||
|
|
f->close = bfl_close;
|
||
|
|
f->can_overwrite_buffer = 2;
|
||
|
|
return f;
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef TEST
|
||
|
|
|
||
|
|
int main(int UNUSED argc, char UNUSED **argv)
|
||
|
|
{
|
||
|
|
struct fastbuf *f = bopen_limited_fd(0, 3, 13);
|
||
|
|
struct fastbuf *o = bfdopen_shared(1, 16);
|
||
|
|
int c;
|
||
|
|
while ((c = bgetc(f)) >= 0)
|
||
|
|
bputc(o, c);
|
||
|
|
bclose(o);
|
||
|
|
bclose(f);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|