ucw/fb-grow.c

   1 /*
   2  *      UCW Library -- Fast Buffered I/O on Growing Buffers
   3  *
   4  *      (c) 2006 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 "ucw/lib.h"
  11 #include "ucw/fastbuf.h"
  12 #include "ucw/mempool.h"
  13
  14 #include <stdio.h>
  15 #include <stdlib.h>
  16
  17 struct fb_gbuf {
  18   struct fastbuf fb;
  19   struct mempool *mp;
  20   byte *end;
  21 };
  22 #define FB_GBUF(f) ((struct fb_gbuf *)(f)->is_fastbuf)
  23
  24 static int fbgrow_refill(struct fastbuf *b)
  25 {
  26   b->bstop = FB_GBUF(b)->end;
  27   b->pos = b->bstop - b->buffer;
  28   return b->bstop > b->bptr;
  29 }
  30
  31 static void fbgrow_spout(struct fastbuf *b)
  32 {
  33   if (b->bptr == b->bufend)
  34     {
  35       uns len = b->bufend - b->buffer;
  36       if (FB_GBUF(b)->mp)
  37         {
  38           byte *old = b->buffer;
  39           b->buffer = mp_alloc(FB_GBUF(b)->mp, 2 * len);
  40           memcpy(b->buffer, old, len);
  41         }
  42       else
  43         b->buffer = xrealloc(b->buffer, 2 * len);
  44       b->bufend = b->buffer + 2 * len;
  45       FB_GBUF(b)->end = b->bptr = b->buffer + len;
  46     }
  47   else if (FB_GBUF(b)->end < b->bptr)
  48     FB_GBUF(b)->end = b->bptr;
  49   b->bstop = b->buffer;
  50   b->pos = 0;
  51 }
  52
  53 static int fbgrow_seek(struct fastbuf *b, ucw_off_t pos, int whence)
  54 {
  55   ucw_off_t len = FB_GBUF(b)->end - b->buffer;
  56   if (whence == SEEK_END)
  57     pos += len;
  58   if (pos < 0 || pos > len)
  59     bthrow(b, "fb.seek", "Seek out of range");
  60   b->bptr = b->buffer + pos;
  61   b->bstop = b->buffer;
  62   b->pos = 0;
  63   return 1;
  64 }
  65
  66 static void fbgrow_close(struct fastbuf *b)
  67 {
  68   xfree(b->buffer);
  69   xfree(b);
  70 }
  71
  72 struct fastbuf *fbgrow_create_mp(struct mempool *mp, unsigned basic_size)
  73 {
  74   ASSERT(basic_size);
  75   struct fastbuf *b;
  76   if (mp)
  77     {
  78       b = mp_alloc_zero(mp, sizeof(struct fb_gbuf));
  79       b->buffer = mp_alloc(mp, basic_size);
  80       FB_GBUF(b)->mp = mp;
  81     }
  82   else
  83     {
  84       b = xmalloc_zero(sizeof(struct fb_gbuf));
  85       b->buffer = xmalloc(basic_size);
  86       b->close = fbgrow_close;
  87     }
  88   b->bufend = b->buffer + basic_size;
  89   b->bptr = b->bstop = b->buffer;
  90   b->name = "<fbgbuf>";
  91   b->refill = fbgrow_refill;
  92   b->spout = fbgrow_spout;
  93   b->seek = fbgrow_seek;
  94   b->can_overwrite_buffer = 1;
  95   return fb_tie(b);
  96 }
  97
  98 struct fastbuf *fbgrow_create(unsigned basic_size)
  99 {
 100   return fbgrow_create_mp(NULL, basic_size);
 101 }
 102
 103 void fbgrow_reset(struct fastbuf *b)
 104 {
 105   FB_GBUF(b)->end = b->bptr = b->bstop = b->buffer;
 106   b->pos = 0;
 107 }
 108
 109 void fbgrow_rewind(struct fastbuf *b)
 110 {
 111   brewind(b);
 112 }
 113
 114 #ifdef TEST
 115
 116 int main(void)
 117 {
 118   struct fastbuf *f;
 119   uns t;
 120
 121   f = fbgrow_create(3);
 122   for (uns i=0; i<5; i++)
 123     {
 124       fbgrow_reset(f);
 125       bwrite(f, "12345", 5);
 126       bwrite(f, "12345", 5);
 127       printf("<%d>", (int)btell(f));
 128       bflush(f);
 129       printf("<%d>", (int)btell(f));
 130       fbgrow_rewind(f);
 131       printf("<%d>", (int)btell(f));
 132       while ((t = bgetc(f)) != ~0U)
 133         putchar(t);
 134       printf("<%d>", (int)btell(f));
 135       fbgrow_rewind(f);
 136       bseek(f, -1, SEEK_END);
 137       printf("<%d>", (int)btell(f));
 138       while ((t = bgetc(f)) != ~0U)
 139         putchar(t);
 140       printf("<%d>\n", (int)btell(f));
 141     }
 142   bclose(f);
 143   return 0;
 144 }
 145
 146 #endif