2 * Copyright (C) 1995-2007, Index Data ApS
3 * See the file LICENSE for details.
5 * $Id: wrbuf.c,v 1.16 2007-02-23 10:15:01 adam Exp $
10 * \brief Implements WRBUF (growing buffer)
22 #include <yaz/wrbuf.h>
23 #include <yaz/snprintf.h>
24 #include <yaz/yaz-iconv.h>
26 WRBUF wrbuf_alloc(void)
30 if (!(n = (WRBUF)xmalloc(sizeof(*n))))
38 void wrbuf_free(WRBUF b, int free_buf)
40 if (free_buf && b->buf)
45 void wrbuf_destroy(WRBUF b)
50 void wrbuf_rewind(WRBUF b)
55 int wrbuf_grow(WRBUF b, int minsize)
65 if (b->size && !(b->buf =(char *)xrealloc(b->buf, b->size += togrow)))
67 else if (!b->size && !(b->buf = (char *)xmalloc(b->size = togrow)))
72 int wrbuf_write(WRBUF b, const char *buf, int size)
76 if (b->pos + size >= b->size)
78 memcpy(b->buf + b->pos, buf, size);
83 int wrbuf_puts(WRBUF b, const char *buf)
85 wrbuf_write(b, buf, strlen(buf)+1); /* '\0'-terminate as well */
86 (b->pos)--; /* don't include '\0' in count */
90 int wrbuf_puts_replace_char(WRBUF b, const char *buf,
91 const char from, const char to)
101 (b->pos)--; /* don't include '\0' in count */
105 void wrbuf_chop_right(WRBUF b)
107 while (b->pos && b->buf[b->pos-1] == ' ')
110 b->buf[b->pos] = '\0';
114 int wrbuf_xmlputs(WRBUF b, const char *cp)
116 return wrbuf_xmlputs_n(b, cp, strlen(cp));
119 int wrbuf_xmlputs_n(WRBUF b, const char *cp, int size)
123 /* only TAB,CR,LF of ASCII CTRL are allowed in XML 1.0! */
124 if (*cp >= 0 && *cp <= 31)
125 if (*cp != 9 && *cp != 10 && *cp != 13)
127 cp++; /* we silently ignore (delete) these.. */
133 wrbuf_puts(b, "<");
136 wrbuf_puts(b, ">");
139 wrbuf_puts(b, "&");
142 wrbuf_puts(b, """);
145 wrbuf_puts(b, "'");
157 void wrbuf_printf(WRBUF b, const char *fmt, ...)
163 yaz_vsnprintf(buf, sizeof(buf)-1, fmt, ap);
169 static int wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
175 size_t inbytesleft = size;
176 const char *inp = buf;
179 size_t outbytesleft = sizeof(outbuf);
181 size_t r = yaz_iconv(cd, (char**) &inp, &inbytesleft,
182 &outp, &outbytesleft);
183 if (r == (size_t) (-1))
185 int e = yaz_iconv_error(cd);
186 if (e != YAZ_ICONV_E2BIG)
190 wrbuf_xmlputs_n(b, outbuf, outp - outbuf);
192 wrbuf_write(b, outbuf, outp - outbuf);
198 wrbuf_xmlputs_n(b, buf, size);
200 wrbuf_write(b, buf, size);
205 int wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
207 return wrbuf_iconv_write_x(b, cd, buf, size, 0);
210 int wrbuf_iconv_puts(WRBUF b, yaz_iconv_t cd, const char *strz)
212 return wrbuf_iconv_write(b, cd, strz, strlen(strz));
215 int wrbuf_iconv_putchar(WRBUF b, yaz_iconv_t cd, int ch)
219 return wrbuf_iconv_write(b, cd, buf, 1);
222 int wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
224 return wrbuf_iconv_write_x(b, cd, buf, size, 1);
227 const char *wrbuf_cstr(WRBUF b)
229 wrbuf_write(b, "", 1); /* '\0'-terminate as well */
230 (b->pos)--; /* don't include '\0' in count */
237 * indent-tabs-mode: nil
239 * vim: shiftwidth=4 tabstop=8 expandtab