src/wrbuf.c

   1 /*
   2  * Copyright (c) 1995-2004, Index Data.
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: wrbuf.c,v 1.5 2004-10-01 11:44:46 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     return wrbuf_xmlputs_n(b, cp, strlen(cp));
  86 }
  87
  88 int wrbuf_xmlputs_n(WRBUF b, const char *cp, int size)
  89 {
  90     while (--size >= 0)
  91     {
  92         /* only TAB,CR,LF of ASCII CTRL are allowed in XML 1.0! */
  93         if (*cp >= 0 && *cp <= 31)
  94             if (*cp != 9 && *cp != 10 && *cp != 13)
  95             {
  96                 cp++;  /* we silently ignore (delete) these.. */
  97                 continue;
  98             }
  99         switch(*cp)
 100         {
 101         case '<':
 102             wrbuf_puts(b, "&lt;");
 103             break;
 104         case '>':
 105             wrbuf_puts(b, "&gt;");
 106             break;
 107         case '&':
 108             wrbuf_puts(b, "&amp;");
 109             break;
 110         case '"':
 111             wrbuf_puts(b, "&quot;");
 112             break;
 113         case '\'':
 114             wrbuf_puts(b, "&apos;");
 115             break;
 116         default:
 117             wrbuf_putc(b, *cp);
 118         }
 119         cp++;
 120     }
 121     wrbuf_putc(b, 0);
 122     (b->pos)--;
 123     return 0;
 124 }
 125
 126 void wrbuf_printf(WRBUF b, const char *fmt, ...)
 127 {
 128     va_list ap;
 129     char buf[4096];
 130
 131     va_start(ap, fmt);
 132 #ifdef WIN32
 133     _vsnprintf(buf, sizeof(buf)-1, fmt, ap);
 134 #else
 135 /* !WIN32 */
 136 #if HAVE_VSNPRINTF
 137     vsnprintf(buf, sizeof(buf)-1, fmt, ap);
 138 #else
 139     vsprintf(buf, fmt, ap);
 140 #endif
 141 #endif
 142     wrbuf_puts (b, buf);
 143
 144     va_end(ap);
 145 }
 146
 147 static int wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
 148                                int size, int cdata)
 149 {
 150     if (cd)
 151     {
 152         char outbuf[12];
 153         size_t inbytesleft = size;
 154         const char *inp = buf;
 155         while (inbytesleft)
 156         {
 157             size_t outbytesleft = sizeof(outbuf);
 158             char *outp = outbuf;
 159             size_t r = yaz_iconv(cd, (char**) &inp,  &inbytesleft,
 160                                  &outp, &outbytesleft);
 161             if (r == (size_t) (-1))
 162             {
 163                 int e = yaz_iconv_error(cd);
 164                 if (e != YAZ_ICONV_E2BIG)
 165                     break;
 166             }
 167             if (cdata)
 168                 wrbuf_xmlputs_n(b, outbuf, outp - outbuf);
 169             else
 170                 wrbuf_write(b, outbuf, outp - outbuf);
 171         }
 172     }
 173     else
 174     {
 175         if (cdata)
 176             wrbuf_xmlputs_n(b, buf, size);
 177         else
 178             wrbuf_write(b, buf, size);
 179     }
 180     return wrbuf_len(b);
 181 }
 182
 183 int wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
 184 {
 185     return wrbuf_iconv_write_x(b, cd, buf, size, 0);
 186 }
 187
 188 int wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
 189 {
 190     return wrbuf_iconv_write_x(b, cd, buf, size, 1);
 191 }
 192