X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=src%2Fwrbuf.c;h=0b92aea2095dbaffea17dfb1bade1ec058c4b170;hp=d4fc08a3cf7fcdd1498407402d83f9027a0491a9;hb=8dc67a9430644549801c18ec7e3a725e5a4ef782;hpb=c6e47cbbff56f39f6d81b079ebaeac41d793d4d9

diff --git a/src/wrbuf.c b/src/wrbuf.c
index d4fc08a..0b92aea 100644
--- a/src/wrbuf.c
+++ b/src/wrbuf.c
@@ -1,8 +1,8 @@
 /*
- * Copyright (c) 1995-2003, Index Data.
+ * Copyright (c) 1995-2004, Index Data.
  * See the file LICENSE for details.
  *
- * $Id: wrbuf.c,v 1.1 2003-10-27 12:21:36 adam Exp $
+ * $Id: wrbuf.c,v 1.4 2004-03-20 07:02:23 adam Exp $
  */
 
 /*
@@ -19,6 +19,7 @@
 #include <stdarg.h>
 
 #include <yaz/wrbuf.h>
+#include <yaz/yaz-iconv.h>
 
 WRBUF wrbuf_alloc(void)
 {
@@ -81,7 +82,12 @@ int wrbuf_puts(WRBUF b, const char *buf)
 
 int wrbuf_xmlputs(WRBUF b, const char *cp)
 {
-    while (*cp)
+    return wrbuf_xmlputs_n(b, cp, strlen(cp));
+}
+
+int wrbuf_xmlputs_n(WRBUF b, const char *cp, int size)
+{
+    while (--size >= 0)
     {
 	switch(*cp)
 	{
@@ -131,3 +137,49 @@ void wrbuf_printf(WRBUF b, const char *fmt, ...)
     va_end(ap);
 }
 
+static int wrbuf_iconv_write_x(WRBUF b, yaz_iconv_t cd, const char *buf,
+			       int size, int cdata)
+{
+    if (cd)
+    {
+	char outbuf[12];
+	size_t inbytesleft = size;
+	const char *inp = buf;
+	while (inbytesleft)
+	{
+	    size_t outbytesleft = sizeof(outbuf);
+	    char *outp = outbuf;
+	    size_t r = yaz_iconv(cd, (char**) &inp,  &inbytesleft,
+				 &outp, &outbytesleft);
+	    if (r == (size_t) (-1))
+	    {
+		int e = yaz_iconv_error(cd);
+		if (e != YAZ_ICONV_E2BIG)
+		    break;
+	    }
+	    if (cdata)
+		wrbuf_xmlputs_n(b, outbuf, outp - outbuf);
+	    else
+		wrbuf_write(b, outbuf, outp - outbuf);
+	}
+    }
+    else
+    {
+	if (cdata)
+	    wrbuf_xmlputs_n(b, buf, size);
+	else
+	    wrbuf_write(b, buf, size);
+    }
+    return wrbuf_len(b);
+}
+
+int wrbuf_iconv_write(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
+{
+    return wrbuf_iconv_write_x(b, cd, buf, size, 0);
+}
+
+int wrbuf_iconv_write_cdata(WRBUF b, yaz_iconv_t cd, const char *buf, int size)
+{
+    return wrbuf_iconv_write_x(b, cd, buf, size, 1);
+}
+