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.3 2004-03-15 21:39:06 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_write_cdata(b, cp, strlen(cp));
  86 }
  87
  88 int wrbuf_write_cdata(WRBUF b, const char *cp, int size)
  89 {
  90     while (--size >= 0)
  91     {
  92         switch(*cp)
  93         {
  94         case '<':
  95             wrbuf_puts(b, "&lt;");
  96             break;
  97         case '>':
  98             wrbuf_puts(b, "&gt;");
  99             break;
 100         case '&':
 101             wrbuf_puts(b, "&amp;");
 102             break;
 103         case '"':
 104             wrbuf_puts(b, "&quot;");
 105             break;
 106         case '\'':
 107             wrbuf_puts(b, "&apos;");
 108             break;
 109         default:
 110             wrbuf_putc(b, *cp);
 111         }
 112         cp++;
 113     }
 114     wrbuf_putc(b, 0);
 115     (b->pos)--;
 116     return 0;
 117 }
 118
 119 void wrbuf_printf(WRBUF b, const char *fmt, ...)
 120 {
 121     va_list ap;
 122     char buf[4096];
 123
 124     va_start(ap, fmt);
 125 #ifdef WIN32
 126     _vsnprintf(buf, sizeof(buf)-1, fmt, ap);
 127 #else
 128 /* !WIN32 */
 129 #if HAVE_VSNPRINTF
 130     vsnprintf(buf, sizeof(buf)-1, fmt, ap);
 131 #else
 132     vsprintf(buf, fmt, ap);
 133 #endif
 134 #endif
 135     wrbuf_puts (b, buf);
 136
 137     va_end(ap);
 138 }
 139
 140 static int wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
 141                                int size, int cdata)
 142 {
 143     if (cd)
 144     {
 145         char outbuf[12];
 146         size_t inbytesleft = size;
 147         const char *inp = buf;
 148         while (inbytesleft)
 149         {
 150             size_t outbytesleft = sizeof(outbuf);
 151             char *outp = outbuf;
 152             size_t r = yaz_iconv(cd, (char**) &inp,  &inbytesleft,
 153                                  &outp, &outbytesleft);
 154             if (r == (size_t) (-1))
 155             {
 156                 int e = yaz_iconv_error(cd);
 157                 if (e != YAZ_ICONV_E2BIG)
 158                     break;
 159             }
 160             if (cdata)
 161                 wrbuf_write_cdata(b, outbuf, outp - outbuf);
 162             else
 163                 wrbuf_write(b, outbuf, outp - outbuf);
 164         }
 165     }
 166     else
 167     {
 168         if (cdata)
 169             wrbuf_write_cdata(b, buf, size);
 170         else
 171             wrbuf_write(b, buf, size);
 172     }
 173     return wrbuf_len(b);
 174 }
 175
 176 int wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
 177 {
 178     return wrbuf_iconv_write_x(b, cd, buf, size, 0);
 179 }
 180
 181 int wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
 182 {
 183     return wrbuf_iconv_write_x(b, cd, buf, size, 1);
 184 }
 185