1 /* This file is part of the YAZ toolkit.
2 * Copyright (C) Index Data
3 * See the file LICENSE for details.
8 * \brief Implements WRBUF (growing buffer)
21 #include <yaz/wrbuf.h>
22 #include <yaz/snprintf.h>
23 #include <yaz/yaz-iconv.h>
25 WRBUF wrbuf_alloc(void)
29 if (!(n = (WRBUF)xmalloc(sizeof(*n))))
37 void wrbuf_destroy(WRBUF b)
46 void wrbuf_rewind(WRBUF b)
51 int wrbuf_grow(WRBUF b, size_t minsize)
61 b->buf = (char *) xrealloc(b->buf, 1 + (b->size += togrow));
67 void wrbuf_write(WRBUF b, const char *buf, size_t size)
71 if (b->pos + size >= b->size)
73 memcpy(b->buf + b->pos, buf, size);
77 void wrbuf_insert(WRBUF b, size_t pos, const char *buf, size_t size)
79 if (size <= 0 || pos > b->pos)
81 if (b->pos + size >= b->size)
83 memmove(b->buf + pos + size, b->buf + pos, b->pos - pos);
84 memcpy(b->buf + pos, buf, size);
88 void wrbuf_puts(WRBUF b, const char *buf)
90 wrbuf_write(b, buf, strlen(buf));
93 void wrbuf_vp_puts(const char *buf, void *client_data)
95 WRBUF b = (WRBUF) client_data;
99 void wrbuf_puts_replace_char(WRBUF b, const char *buf,
100 const char from, const char to)
112 void wrbuf_chop_right(WRBUF b)
114 while (b->pos && b->buf[b->pos-1] == ' ')
120 void wrbuf_xmlputs(WRBUF b, const char *cp)
122 wrbuf_xmlputs_n(b, cp, strlen(cp));
125 void wrbuf_xmlputs_n(WRBUF b, const char *cp, size_t size)
129 /* only TAB,CR,LF of ASCII CTRL are allowed in XML 1.0! */
130 if (*cp >= 0 && *cp <= 31)
131 if (*cp != 9 && *cp != 10 && *cp != 13)
133 cp++; /* we silently ignore (delete) these.. */
139 wrbuf_puts(b, "<");
142 wrbuf_puts(b, ">");
145 wrbuf_puts(b, "&");
148 wrbuf_puts(b, """);
151 wrbuf_puts(b, "'");
160 void wrbuf_printf(WRBUF b, const char *fmt, ...)
166 yaz_vsnprintf(buf, sizeof(buf)-1, fmt, ap);
172 int wrbuf_iconv_write2(WRBUF b, yaz_iconv_t cd, const char *buf,
174 void (*wfunc)(WRBUF, const char *, size_t))
180 size_t inbytesleft = size;
181 const char *inp = buf;
184 size_t outbytesleft = sizeof(outbuf);
186 size_t r = yaz_iconv(cd, (char**) &inp, &inbytesleft,
187 &outp, &outbytesleft);
188 if (r == (size_t) (-1))
190 int e = yaz_iconv_error(cd);
191 if (e != YAZ_ICONV_E2BIG)
197 (*wfunc)(b, outbuf, outp - outbuf);
201 (*wfunc)(b, buf, size);
205 int wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
206 size_t size, int cdata)
208 return wrbuf_iconv_write2(b, cd, buf, size,
209 cdata ? wrbuf_xmlputs_n : wrbuf_write);
212 void wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, size_t size)
214 wrbuf_iconv_write2(b, cd, buf, size, wrbuf_write);
217 void wrbuf_iconv_puts(WRBUF b, yaz_iconv_t cd, const char *strz)
219 wrbuf_iconv_write(b, cd, strz, strlen(strz));
222 void wrbuf_iconv_putchar(WRBUF b, yaz_iconv_t cd, int ch)
226 wrbuf_iconv_write(b, cd, buf, 1);
229 void wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, size_t size)
231 wrbuf_iconv_write2(b, cd, buf, size, wrbuf_xmlputs_n);
234 void wrbuf_iconv_puts_cdata(WRBUF b, yaz_iconv_t cd, const char *strz)
236 wrbuf_iconv_write2(b, cd, strz, strlen(strz), wrbuf_xmlputs_n);
239 void wrbuf_iconv_json_write(WRBUF b, yaz_iconv_t cd,
240 const char *buf, size_t size)
242 wrbuf_iconv_write2(b, cd, buf, size, wrbuf_json_write);
245 void wrbuf_iconv_json_puts(WRBUF b, yaz_iconv_t cd, const char *strz)
247 wrbuf_iconv_write2(b, cd, strz, strlen(strz), wrbuf_json_write);
250 void wrbuf_iconv_reset(WRBUF b, yaz_iconv_t cd)
255 size_t outbytesleft = sizeof(outbuf);
257 size_t r = yaz_iconv(cd, 0, 0, &outp, &outbytesleft);
258 if (r != (size_t) (-1))
259 wrbuf_write(b, outbuf, outp - outbuf);
263 const char *wrbuf_cstr(WRBUF b)
267 assert(b->pos <= b->size);
268 b->buf[b->pos] = '\0';
272 const char *wrbuf_cstr_null(WRBUF b)
274 if (!b || b->pos == 0)
276 assert(b->pos <= b->size);
277 b->buf[b->pos] = '\0';
281 void wrbuf_cut_right(WRBUF b, size_t no_to_remove)
283 if (no_to_remove > b->pos)
284 no_to_remove = b->pos;
285 b->pos = b->pos - no_to_remove;
288 void wrbuf_puts_escaped(WRBUF b, const char *str)
290 wrbuf_write_escaped(b, str, strlen(str));
293 void wrbuf_write_escaped(WRBUF b, const char *str, size_t len)
296 for (i = 0; i < len; i++)
297 if (str[i] < ' ' || str[i] > 126)
298 wrbuf_printf(b, "\\x%02X", str[i] & 0xff);
300 wrbuf_putc(b, str[i]);
303 void wrbuf_json_write(WRBUF b, const char *cp, size_t sz)
306 for (i = 0; i < sz; i++)
308 if (cp[i] > 0 && cp[i] < 32)
313 case '\b': wrbuf_putc(b, 'b'); break;
314 case '\f': wrbuf_putc(b, 'f'); break;
315 case '\n': wrbuf_putc(b, 'n'); break;
316 case '\r': wrbuf_putc(b, 'r'); break;
317 case '\t': wrbuf_putc(b, 't'); break;
319 wrbuf_printf(b, "u%04x", cp[i]);
322 else if (cp[i] == '"')
324 wrbuf_putc(b, '\\'); wrbuf_putc(b, '"');
326 else if (cp[i] == '\\')
328 wrbuf_putc(b, '\\'); wrbuf_putc(b, '\\');
331 { /* leave encoding as raw UTF-8 */
332 wrbuf_putc(b, cp[i]);
338 void wrbuf_json_puts(WRBUF b, const char *str)
340 wrbuf_json_write(b, str, strlen(str));
346 * c-file-style: "Stroustrup"
347 * indent-tabs-mode: nil
349 * vim: shiftwidth=4 tabstop=8 expandtab