2 * Copyright (c) 1995-2004, Index Data.
3 * See the file LICENSE for details.
5 * $Id: wrbuf.c,v 1.5 2004-10-01 11:44:46 adam Exp $
9 * Growing buffer for writing various stuff.
21 #include <yaz/wrbuf.h>
22 #include <yaz/yaz-iconv.h>
24 WRBUF wrbuf_alloc(void)
28 if (!(n = (WRBUF)xmalloc(sizeof(*n))))
36 void wrbuf_free(WRBUF b, int free_buf)
38 if (free_buf && b->buf)
43 void wrbuf_rewind(WRBUF b)
48 int wrbuf_grow(WRBUF b, int minsize)
58 if (b->size && !(b->buf =(char *)xrealloc(b->buf, b->size += togrow)))
60 else if (!b->size && !(b->buf = (char *)xmalloc(b->size = togrow)))
65 int wrbuf_write(WRBUF b, const char *buf, int size)
69 if (b->pos + size >= b->size)
71 memcpy(b->buf + b->pos, buf, size);
76 int wrbuf_puts(WRBUF b, const char *buf)
78 wrbuf_write(b, buf, strlen(buf)+1); /* '\0'-terminate as well */
79 (b->pos)--; /* don't include '\0' in count */
83 int wrbuf_xmlputs(WRBUF b, const char *cp)
85 return wrbuf_xmlputs_n(b, cp, strlen(cp));
88 int wrbuf_xmlputs_n(WRBUF b, const char *cp, int size)
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)
96 cp++; /* we silently ignore (delete) these.. */
102 wrbuf_puts(b, "<");
105 wrbuf_puts(b, ">");
108 wrbuf_puts(b, "&");
111 wrbuf_puts(b, """);
114 wrbuf_puts(b, "'");
126 void wrbuf_printf(WRBUF b, const char *fmt, ...)
133 _vsnprintf(buf, sizeof(buf)-1, fmt, ap);
137 vsnprintf(buf, sizeof(buf)-1, fmt, ap);
139 vsprintf(buf, fmt, ap);
147 static int wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
153 size_t inbytesleft = size;
154 const char *inp = buf;
157 size_t outbytesleft = sizeof(outbuf);
159 size_t r = yaz_iconv(cd, (char**) &inp, &inbytesleft,
160 &outp, &outbytesleft);
161 if (r == (size_t) (-1))
163 int e = yaz_iconv_error(cd);
164 if (e != YAZ_ICONV_E2BIG)
168 wrbuf_xmlputs_n(b, outbuf, outp - outbuf);
170 wrbuf_write(b, outbuf, outp - outbuf);
176 wrbuf_xmlputs_n(b, buf, size);
178 wrbuf_write(b, buf, size);
183 int wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
185 return wrbuf_iconv_write_x(b, cd, buf, size, 0);
188 int wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
190 return wrbuf_iconv_write_x(b, cd, buf, size, 1);