util/wrbuf.c

   1 /*
   2  * Copyright (c) 1995-2003, Index Data.
   3  * See the file LICENSE for details.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * $Id: wrbuf.c,v 1.12 2003-01-06 08:20:28 adam Exp $
   7  */
   8
   9 /*
  10  * Growing buffer for writing various stuff.
  11  */
  12
  13 #if HAVE_CONFIG_H
  14 #include <config.h>
  15 #endif
  16
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19 #include <string.h>
  20 #include <stdarg.h>
  21
  22 #include <yaz/wrbuf.h>
  23
  24 WRBUF wrbuf_alloc(void)
  25 {
  26     WRBUF n;
  27
  28     if (!(n = (WRBUF)xmalloc(sizeof(*n))))
  29         abort();
  30     n->buf = 0;
  31     n->size = 0;
  32     n->pos = 0;
  33     return n;
  34 }
  35
  36 void wrbuf_free(WRBUF b, int free_buf)
  37 {
  38     if (free_buf && b->buf)
  39         xfree(b->buf);
  40     xfree(b);
  41 }
  42
  43 void wrbuf_rewind(WRBUF b)
  44 {
  45     b->pos = 0;
  46 }
  47
  48 int wrbuf_grow(WRBUF b, int minsize)
  49 {
  50     int togrow;
  51
  52     if (!b->size)
  53         togrow = 1024;
  54     else
  55         togrow = b->size;
  56     if (togrow < minsize)
  57         togrow = minsize;
  58     if (b->size && !(b->buf =(char *)xrealloc(b->buf, b->size += togrow)))
  59         abort();
  60     else if (!b->size && !(b->buf = (char *)xmalloc(b->size = togrow)))
  61         abort();
  62     return 0;
  63 }
  64
  65 int wrbuf_write(WRBUF b, const char *buf, int size)
  66 {
  67     if (size <= 0)
  68         return 0;
  69     if (b->pos + size >= b->size)
  70         wrbuf_grow(b, size);
  71     memcpy(b->buf + b->pos, buf, size);
  72     b->pos += size;
  73     return 0;
  74 }
  75
  76 int wrbuf_puts(WRBUF b, const char *buf)
  77 {
  78     wrbuf_write(b, buf, strlen(buf)+1);  /* '\0'-terminate as well */
  79     (b->pos)--;                          /* don't include '\0' in count */
  80     return 0;
  81 }
  82
  83 void wrbuf_printf(WRBUF b, const char *fmt, ...)
  84 {
  85     va_list ap;
  86     char buf[4096];
  87
  88     va_start(ap, fmt);
  89 #ifdef WIN32
  90     _vsnprintf(buf, sizeof(buf)-1, fmt, ap);
  91 #else
  92 /* !WIN32 */
  93 #if HAVE_VSNPRINTF
  94     vsnprintf(buf, sizeof(buf)-1, fmt, ap);
  95 #else
  96     vsprintf(buf, fmt, ap);
  97 #endif
  98 #endif
  99     wrbuf_puts (b, buf);
 100
 101     va_end(ap);
 102 }
 103