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   fb_tie(b);
  96   return b;
  97 }
  98
  99 struct fastbuf *fbgrow_create(unsigned basic_size)
 100 {
 101   return fbgrow_create_mp(NULL, basic_size);
 102 }
 103
 104 void fbgrow_reset(struct fastbuf *b)
 105 {
 106   FB_GBUF(b)->end = b->bptr = b->bstop = b->buffer;
 107   b->pos = 0;
 108 }
 109
 110 void fbgrow_rewind(struct fastbuf *b)
 111 {
 112   brewind(b);
 113 }
 114
 115 #ifdef TEST
 116
 117 int main(void)
 118 {
 119   struct fastbuf *f;
 120   uns t;
 121
 122   f = fbgrow_create(3);
 123   for (uns i=0; i<5; i++)
 124     {
 125       fbgrow_reset(f);
 126       bwrite(f, "12345", 5);
 127       bwrite(f, "12345", 5);
 128       printf("<%d>", (int)btell(f));
 129       bflush(f);
 130       printf("<%d>", (int)btell(f));
 131       fbgrow_rewind(f);
 132       printf("<%d>", (int)btell(f));
 133       while ((t = bgetc(f)) != ~0U)
 134         putchar(t);
 135       printf("<%d>", (int)btell(f));
 136       fbgrow_rewind(f);
 137       bseek(f, -1, SEEK_END);
 138       printf("<%d>", (int)btell(f));
 139       while ((t = bgetc(f)) != ~0U)
 140         putchar(t);
 141       printf("<%d>\n", (int)btell(f));
 142     }
 143   bclose(f);
 144   return 0;
 145 }
 146
 147 #endif