lib/fb-limfd.c

   1 /*
   2  *      UCW Library -- Fast Buffered Input on Limited File Descriptors
   3  *
   4  *      (c) 2003--2004 Martin Mares <mj@ucw.cz>
   5  *
   6  *      This software may be freely distributed and used according to the terms
   7  *      of the GNU Lesser General Public License.
   8  */
   9
  10 #include "lib/lib.h"
  11 #include "lib/fastbuf.h"
  12
  13 #include <stdlib.h>
  14 #include <unistd.h>
  15
  16 struct fb_limfd {
  17   struct fastbuf fb;
  18   int fd;                               /* File descriptor */
  19   int limit;
  20 };
  21 #define FB_LIMFD(f) ((struct fb_limfd *)(f)->is_fastbuf)
  22 #define FB_BUFFER(f) (byte *)(FB_LIMFD(f) + 1)
  23
  24 static int
  25 bfl_refill(struct fastbuf *f)
  26 {
  27   f->bptr = f->buffer = FB_BUFFER(f);
  28   int max = MIN(FB_LIMFD(f)->limit - f->pos, f->bufend - f->buffer);
  29   int l = read(FB_LIMFD(f)->fd, f->buffer, max);
  30   if (l < 0)
  31     die("Error reading %s: %m", f->name);
  32   f->bstop = f->buffer + l;
  33   f->pos += l;
  34   return l;
  35 }
  36
  37 static void
  38 bfl_close(struct fastbuf *f)
  39 {
  40   xfree(f);
  41 }
  42
  43 struct fastbuf *
  44 bopen_limited_fd(int fd, uns buflen, uns limit)
  45 {
  46   struct fb_limfd *F = xmalloc(sizeof(struct fb_limfd) + buflen);
  47   struct fastbuf *f = &F->fb;
  48
  49   bzero(F, sizeof(*F));
  50   f->buffer = (char *)(F+1);
  51   f->bptr = f->bstop = f->buffer;
  52   f->bufend = f->buffer + buflen;
  53   f->name = "limited-fd";
  54   F->fd = fd;
  55   F->limit = limit;
  56   f->refill = bfl_refill;
  57   f->close = bfl_close;
  58   f->can_overwrite_buffer = 2;
  59   return f;
  60 }
  61
  62 #ifdef TEST
  63
  64 int main(int argc, char **argv)
  65 {
  66   struct fastbuf *f = bopen_limited_fd(0, 3, 13);
  67   struct fastbuf *o = bfdopen_shared(1, 16);
  68   int c;
  69   while ((c = bgetc(f)) >= 0)
  70     bputc(o, c);
  71   bclose(o);
  72   bclose(f);
  73   return 0;
  74 }
  75
  76 #endif