1 /* This file is part of the YAZ toolkit.
2 * Copyright (C) 1995-2010 Index Data
3 * See the file LICENSE for details.
8 * \brief Implements WRBUF (growing buffer)
20 #include <yaz/wrbuf.h>
21 #include <yaz/snprintf.h>
22 #include <yaz/yaz-iconv.h>
24 WRBUF wrbuf_alloc(void)
28 if (!(n = (WRBUF)xmalloc(sizeof(*n))))
36 void wrbuf_destroy(WRBUF b)
42 void wrbuf_rewind(WRBUF b)
47 int wrbuf_grow(WRBUF b, size_t minsize)
57 if (b->size && !(b->buf =(char *)xrealloc(b->buf, b->size += togrow)))
59 else if (!b->size && !(b->buf = (char *)xmalloc(b->size = togrow)))
64 void wrbuf_write(WRBUF b, const char *buf, size_t size)
68 if (b->pos + size >= b->size)
70 memcpy(b->buf + b->pos, buf, size);
74 void wrbuf_puts(WRBUF b, const char *buf)
76 wrbuf_write(b, buf, strlen(buf));
79 void wrbuf_vputs(const char *buf, void *client_data)
81 wrbuf_write((WRBUF) client_data, buf, strlen(buf));
84 void wrbuf_puts_replace_char(WRBUF b, const char *buf,
85 const char from, const char to)
97 void wrbuf_chop_right(WRBUF b)
99 while (b->pos && b->buf[b->pos-1] == ' ')
105 void wrbuf_xmlputs(WRBUF b, const char *cp)
107 wrbuf_xmlputs_n(b, cp, strlen(cp));
110 void wrbuf_xmlputs_n(WRBUF b, const char *cp, size_t size)
114 /* only TAB,CR,LF of ASCII CTRL are allowed in XML 1.0! */
115 if (*cp >= 0 && *cp <= 31)
116 if (*cp != 9 && *cp != 10 && *cp != 13)
118 cp++; /* we silently ignore (delete) these.. */
124 wrbuf_puts(b, "<");
127 wrbuf_puts(b, ">");
130 wrbuf_puts(b, "&");
133 wrbuf_puts(b, """);
136 wrbuf_puts(b, "'");
145 void wrbuf_printf(WRBUF b, const char *fmt, ...)
151 yaz_vsnprintf(buf, sizeof(buf)-1, fmt, ap);
157 static void wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
158 size_t size, int cdata)
163 size_t inbytesleft = size;
164 const char *inp = buf;
167 size_t outbytesleft = sizeof(outbuf);
169 size_t r = yaz_iconv(cd, (char**) &inp, &inbytesleft,
170 &outp, &outbytesleft);
171 if (r == (size_t) (-1))
173 int e = yaz_iconv_error(cd);
174 if (e != YAZ_ICONV_E2BIG)
178 wrbuf_xmlputs_n(b, outbuf, outp - outbuf);
180 wrbuf_write(b, outbuf, outp - outbuf);
186 wrbuf_xmlputs_n(b, buf, size);
188 wrbuf_write(b, buf, size);
192 void wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, size_t size)
194 wrbuf_iconv_write_x(b, cd, buf, size, 0);
197 void wrbuf_iconv_puts(WRBUF b, yaz_iconv_t cd, const char *strz)
199 wrbuf_iconv_write(b, cd, strz, strlen(strz));
202 void wrbuf_iconv_putchar(WRBUF b, yaz_iconv_t cd, int ch)
206 wrbuf_iconv_write(b, cd, buf, 1);
209 void wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, size_t size)
211 wrbuf_iconv_write_x(b, cd, buf, size, 1);
214 void wrbuf_iconv_puts_cdata(WRBUF b, yaz_iconv_t cd, const char *strz)
216 wrbuf_iconv_write_x(b, cd, strz, strlen(strz), 1);
219 void wrbuf_iconv_reset(WRBUF b, yaz_iconv_t cd)
224 size_t outbytesleft = sizeof(outbuf);
226 size_t r = yaz_iconv(cd, 0, 0, &outp, &outbytesleft);
227 if (r != (size_t) (-1))
228 wrbuf_write(b, outbuf, outp - outbuf);
232 const char *wrbuf_cstr(WRBUF b)
234 wrbuf_putc(b, '\0'); /* add '\0' */
235 (b->pos)--; /* don't include '\0' in count */
239 void wrbuf_cut_right(WRBUF b, size_t no_to_remove)
241 if (no_to_remove > b->pos)
242 no_to_remove = b->pos;
243 b->pos = b->pos - no_to_remove;
246 void wrbuf_puts_escaped(WRBUF b, const char *str)
248 wrbuf_write_escaped(b, str, strlen(str));
251 void wrbuf_write_escaped(WRBUF b, const char *str, size_t len)
254 for (i = 0; i < len; i++)
255 if (str[i] < ' ' || str[i] > 126)
256 wrbuf_printf(b, "\\x%02X", str[i] & 0xff);
258 wrbuf_putc(b, str[i]);
264 * c-file-style: "Stroustrup"
265 * indent-tabs-mode: nil
267 * vim: shiftwidth=4 tabstop=8 expandtab