src/wrbuf.c

   1 /*
   2  * Copyright (c) 1995-2003, Index Data.
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: wrbuf.c,v 1.2 2003-12-11 00:37:22 adam Exp $
   6  */
   7
   8 /*
   9  * Growing buffer for writing various stuff.
  10  */
  11
  12 #if HAVE_CONFIG_H
  13 #include <config.h>
  14 #endif
  15
  16 #include <stdio.h>
  17 #include <stdlib.h>
  18 #include <string.h>
  19 #include <stdarg.h>
  20
  21 #include <yaz/wrbuf.h>
  22 #include <yaz/yaz-iconv.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 int wrbuf_xmlputs(WRBUF b, const char *cp)
  84 {
  85     while (*cp)
  86     {
  87         switch(*cp)
  88         {
  89         case '<':
  90             wrbuf_puts(b, "&lt;");
  91             break;
  92         case '>':
  93             wrbuf_puts(b, "&gt;");
  94             break;
  95         case '&':
  96             wrbuf_puts(b, "&amp;");
  97             break;
  98         case '"':
  99             wrbuf_puts(b, "&quot;");
 100             break;
 101         case '\'':
 102             wrbuf_puts(b, "&apos;");
 103             break;
 104         default:
 105             wrbuf_putc(b, *cp);
 106         }
 107         cp++;
 108     }
 109     wrbuf_putc(b, 0);
 110     (b->pos)--;
 111     return 0;
 112 }
 113
 114 void wrbuf_printf(WRBUF b, const char *fmt, ...)
 115 {
 116     va_list ap;
 117     char buf[4096];
 118
 119     va_start(ap, fmt);
 120 #ifdef WIN32
 121     _vsnprintf(buf, sizeof(buf)-1, fmt, ap);
 122 #else
 123 /* !WIN32 */
 124 #if HAVE_VSNPRINTF
 125     vsnprintf(buf, sizeof(buf)-1, fmt, ap);
 126 #else
 127     vsprintf(buf, fmt, ap);
 128 #endif
 129 #endif
 130     wrbuf_puts (b, buf);
 131
 132     va_end(ap);
 133 }
 134
 135 int wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
 136 {
 137     if (cd)
 138     {
 139         char outbuf[12];
 140         size_t inbytesleft = size;
 141         const char *inp = buf;
 142         while (inbytesleft)
 143         {
 144             size_t outbytesleft = sizeof(outbuf);
 145             char *outp = outbuf;
 146             size_t r = yaz_iconv(cd, (char**) &inp,  &inbytesleft,
 147                                  &outp, &outbytesleft);
 148             if (r == (size_t) (-1))
 149             {
 150                 int e = yaz_iconv_error(cd);
 151                 if (e != YAZ_ICONV_E2BIG)
 152                     break;
 153             }
 154             wrbuf_write(b, outbuf, outp - outbuf);
 155         }
 156     }
 157     else
 158         wrbuf_write(b, buf, size);
 159     return wrbuf_len(b);
 160 }