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))
  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       uint 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, "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, uint 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 b;
  96 }
  97
  98 struct fastbuf *fbgrow_create(uint 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 uint fbgrow_get_buf(struct fastbuf *b, byte **buf)
 115 {
 116   byte *end = FB_GBUF(b)->end;
 117   end = MAX(end, b->bptr);
 118   if (buf)
 119     *buf = b->buffer;
 120   return end - b->buffer;
 121 }
 122
 123 #ifdef TEST
 124
 125 int main(void)
 126 {
 127   struct fastbuf *f;
 128   uint t;
 129
 130   f = fbgrow_create(3);
 131   for (uint i=0; i<5; i++)
 132     {
 133       fbgrow_reset(f);
 134       bwrite(f, "12345", 5);
 135       bwrite(f, "12345", 5);
 136       printf("<%d>", (int)btell(f));
 137       bflush(f);
 138       printf("<%d>", (int)btell(f));
 139       fbgrow_rewind(f);
 140       printf("<%d>", (int)btell(f));
 141       while ((t = bgetc(f)) != ~0U)
 142         putchar(t);
 143       printf("<%d>", (int)btell(f));
 144       fbgrow_rewind(f);
 145       bseek(f, -1, SEEK_END);
 146       printf("<%d>", (int)btell(f));
 147       while ((t = bgetc(f)) != ~0U)
 148         putchar(t);
 149       printf("<%d>\n", (int)btell(f));
 150     }
 151   bclose(f);
 152   return 0;
 153 }
 154
 155 #endif