-/*
- * Copyright (C) 1995-2005, Index Data ApS
+/* This file is part of the YAZ toolkit.
+ * Copyright (C) 1995-2013 Index Data
* See the file LICENSE for details.
- *
- * $Id: wrbuf.c,v 1.9 2005-09-27 17:52:46 adam Exp $
*/
/**
#include <stdarg.h>
#include <yaz/wrbuf.h>
+#include <yaz/snprintf.h>
#include <yaz/yaz-iconv.h>
WRBUF wrbuf_alloc(void)
return n;
}
-void wrbuf_free(WRBUF b, int free_buf)
+void wrbuf_destroy(WRBUF b)
{
- if (free_buf && b->buf)
+ if (b)
+ {
xfree(b->buf);
- xfree(b);
+ xfree(b);
+ }
}
void wrbuf_rewind(WRBUF b)
b->pos = 0;
}
-int wrbuf_grow(WRBUF b, int minsize)
+int wrbuf_grow(WRBUF b, size_t minsize)
{
- int togrow;
+ size_t togrow;
if (!b->size)
togrow = 1024;
return 0;
}
-int wrbuf_write(WRBUF b, const char *buf, int size)
+void wrbuf_write(WRBUF b, const char *buf, size_t size)
{
if (size <= 0)
- return 0;
+ return;
if (b->pos + size >= b->size)
wrbuf_grow(b, size);
memcpy(b->buf + b->pos, buf, size);
b->pos += size;
- return 0;
}
-int wrbuf_puts(WRBUF b, const char *buf)
+void wrbuf_insert(WRBUF b, size_t pos, const char *buf, size_t size)
{
- wrbuf_write(b, buf, strlen(buf)+1); /* '\0'-terminate as well */
- (b->pos)--; /* don't include '\0' in count */
- return 0;
+ if (size <= 0 || pos > b->pos)
+ return;
+ if (b->pos + size >= b->size)
+ wrbuf_grow(b, size);
+ memmove(b->buf + pos + size, b->buf + pos, b->pos - pos);
+ memcpy(b->buf + pos, buf, size);
+ b->pos += size;
+}
+
+void wrbuf_puts(WRBUF b, const char *buf)
+{
+ wrbuf_write(b, buf, strlen(buf));
+}
+
+void wrbuf_vp_puts(const char *buf, void *client_data)
+{
+ WRBUF b = (WRBUF) client_data;
+ wrbuf_puts(b, buf);
+}
+
+void wrbuf_puts_replace_char(WRBUF b, const char *buf,
+ const char from, const char to)
+{
+ while(*buf)
+ {
+ if (*buf == from)
+ wrbuf_putc(b, to);
+ else
+ wrbuf_putc(b, *buf);
+ buf++;
+ }
}
void wrbuf_chop_right(WRBUF b)
while (b->pos && b->buf[b->pos-1] == ' ')
{
(b->pos)--;
- b->buf[b->pos] = '\0';
}
}
-int wrbuf_xmlputs(WRBUF b, const char *cp)
+void wrbuf_xmlputs(WRBUF b, const char *cp)
{
- return wrbuf_xmlputs_n(b, cp, strlen(cp));
+ wrbuf_xmlputs_n(b, cp, strlen(cp));
}
-int wrbuf_xmlputs_n(WRBUF b, const char *cp, int size)
+void wrbuf_xmlputs_n(WRBUF b, const char *cp, size_t size)
{
- while (--size >= 0)
+ for (; size; size--)
{
/* only TAB,CR,LF of ASCII CTRL are allowed in XML 1.0! */
if (*cp >= 0 && *cp <= 31)
}
cp++;
}
- wrbuf_putc(b, 0);
- (b->pos)--;
- return 0;
}
void wrbuf_printf(WRBUF b, const char *fmt, ...)
char buf[4096];
va_start(ap, fmt);
-#ifdef WIN32
- _vsnprintf(buf, sizeof(buf)-1, fmt, ap);
-#else
-/* !WIN32 */
-#if HAVE_VSNPRINTF
- vsnprintf(buf, sizeof(buf)-1, fmt, ap);
-#else
- vsprintf(buf, fmt, ap);
-#endif
-#endif
+ yaz_vsnprintf(buf, sizeof(buf)-1, fmt, ap);
wrbuf_puts (b, buf);
va_end(ap);
}
-static int wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
- int size, int cdata)
+int wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
+ size_t size, int cdata)
{
+ int ret = 0;
if (cd)
{
- char outbuf[12];
+ char outbuf[128];
size_t inbytesleft = size;
const char *inp = buf;
while (inbytesleft)
{
int e = yaz_iconv_error(cd);
if (e != YAZ_ICONV_E2BIG)
+ {
+ ret = -1;
break;
+ }
}
if (cdata)
wrbuf_xmlputs_n(b, outbuf, outp - outbuf);
else
wrbuf_write(b, buf, size);
}
- return wrbuf_len(b);
+ return ret;
+}
+
+void wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, size_t size)
+{
+ wrbuf_iconv_write_x(b, cd, buf, size, 0);
}
-int wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
+void wrbuf_iconv_puts(WRBUF b, yaz_iconv_t cd, const char *strz)
{
- return wrbuf_iconv_write_x(b, cd, buf, size, 0);
+ wrbuf_iconv_write(b, cd, strz, strlen(strz));
}
-int wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
+void wrbuf_iconv_putchar(WRBUF b, yaz_iconv_t cd, int ch)
{
- return wrbuf_iconv_write_x(b, cd, buf, size, 1);
+ char buf[1];
+ buf[0] = ch;
+ wrbuf_iconv_write(b, cd, buf, 1);
+}
+
+void wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, size_t size)
+{
+ wrbuf_iconv_write_x(b, cd, buf, size, 1);
+}
+
+void wrbuf_iconv_puts_cdata(WRBUF b, yaz_iconv_t cd, const char *strz)
+{
+ wrbuf_iconv_write_x(b, cd, strz, strlen(strz), 1);
+}
+
+void wrbuf_iconv_reset(WRBUF b, yaz_iconv_t cd)
+{
+ if (cd)
+ {
+ char outbuf[16];
+ size_t outbytesleft = sizeof(outbuf);
+ char *outp = outbuf;
+ size_t r = yaz_iconv(cd, 0, 0, &outp, &outbytesleft);
+ if (r != (size_t) (-1))
+ wrbuf_write(b, outbuf, outp - outbuf);
+ }
+}
+
+const char *wrbuf_cstr(WRBUF b)
+{
+ wrbuf_putc(b, '\0'); /* add '\0' */
+ (b->pos)--; /* don't include '\0' in count */
+ return b->buf;
+}
+
+void wrbuf_cut_right(WRBUF b, size_t no_to_remove)
+{
+ if (no_to_remove > b->pos)
+ no_to_remove = b->pos;
+ b->pos = b->pos - no_to_remove;
+}
+
+void wrbuf_puts_escaped(WRBUF b, const char *str)
+{
+ wrbuf_write_escaped(b, str, strlen(str));
+}
+
+void wrbuf_write_escaped(WRBUF b, const char *str, size_t len)
+{
+ size_t i;
+ for (i = 0; i < len; i++)
+ if (str[i] < ' ' || str[i] > 126)
+ wrbuf_printf(b, "\\x%02X", str[i] & 0xff);
+ else
+ wrbuf_putc(b, str[i]);
}
/*
* Local variables:
* c-basic-offset: 4
+ * c-file-style: "Stroustrup"
* indent-tabs-mode: nil
* End:
* vim: shiftwidth=4 tabstop=8 expandtab