Added fb_wrap_charconv_in().

diff --git a/charset/fb-charconv.c b/charset/fb-charconv.c
index 9c54c883779daa84262fc14b009ba99de4340ff6..9a36590aeb64f2b1746da03bd8fcd0e63a9220b0 100644 (file)
--- a/charset/fb-charconv.c
+++ b/charset/fb-charconv.c
@@ -1,7 +1,7 @@
 /*
  *     Sherlock Library -- Charset Conversion Wrapper for Fast Buffered I/O
  *
- *     (c) 2003 Martin Mares <mj@ucw.cz>
+ *     (c) 2003--2005 Martin Mares <mj@ucw.cz>
  *
  *     This software may be freely distributed and used according to the terms
  *     of the GNU Lesser General Public License.
@@ -18,7 +18,7 @@
 
 struct fb_charconv {
   struct fastbuf fb;
-  struct fastbuf *out;
+  struct fastbuf *orig;
   struct conv_context ctxt;
   byte buf[BUFSIZE];
 };
@@ -36,8 +36,8 @@ fb_cc_spout(struct fastbuf *f)
     {
       flags = conv_run(ct);
       if (ct->dest > ct->dest_start)
-       bdirect_write_commit(FB_CC(f)->out, ct->dest);
-      uns l = bdirect_write_prepare(FB_CC(f)->out, &ct->dest_start);
+       bdirect_write_commit(FB_CC(f)->orig, ct->dest);
+      uns l = bdirect_write_prepare(FB_CC(f)->orig, &ct->dest_start);
       ct->dest = ct->dest_start;
       ct->dest_end = ct->dest + l;
     }
@@ -46,10 +46,35 @@ fb_cc_spout(struct fastbuf *f)
   f->bptr = f->buffer;
 }
 
+static int
+fb_cc_refill(struct fastbuf *f)
+{
+  struct conv_context *ct = &FB_CC(f)->ctxt;
+  int flags;
+
+  f->bptr = f->bstop = f->buffer;
+  do
+    {
+      byte *src;
+      uns len = bdirect_read_prepare(FB_CC(f)->orig, &src);
+      if (!len)
+       break;
+      ct->source = src;
+      ct->source_end = ct->source + len;
+      ct->dest = ct->dest_start = f->bstop;
+      ct->dest_end = f->bufend;
+      flags = conv_run(ct);
+      bdirect_read_commit(FB_CC(f)->orig, (byte*)ct->source);
+      f->bstop = ct->dest;
+    }
+  while (!(flags & CONV_DEST_END));
+  return (f->bstop > f->bptr);
+}
+
 static void
 fb_cc_close(struct fastbuf *f)
 {
-  bflush(FB_CC(f)->out);
+  bflush(FB_CC(f)->orig);
   xfree(f);
 }
 
@@ -60,7 +85,7 @@ fb_wrap_charconv_out(struct fastbuf *f, int cs_from, int cs_to)
     return f;
 
   struct fastbuf *g = xmalloc_zero(sizeof(struct fb_charconv));
-  FB_CC(g)->out = f;
+  FB_CC(g)->orig = f;
   conv_init(&FB_CC(g)->ctxt);
   conv_set_charset(&FB_CC(g)->ctxt, cs_from, cs_to);
   g->name = "<charconv-out>";
@@ -70,3 +95,21 @@ fb_wrap_charconv_out(struct fastbuf *f, int cs_from, int cs_to)
   g->bufend = g->buffer + BUFSIZE;
   return g;
 }
+
+struct fastbuf *
+fb_wrap_charconv_in(struct fastbuf *f, int cs_from, int cs_to)
+{
+  if (cs_from == cs_to)
+    return f;
+
+  struct fastbuf *g = xmalloc_zero(sizeof(struct fb_charconv));
+  FB_CC(g)->orig = f;
+  conv_init(&FB_CC(g)->ctxt);
+  conv_set_charset(&FB_CC(g)->ctxt, cs_from, cs_to);
+  g->name = "<charconv-in>";
+  g->refill = fb_cc_refill;
+  g->close = fb_cc_close;
+  g->buffer = g->bstop = g->bptr = FB_CC(g)->buf;
+  g->bufend = g->buffer + BUFSIZE;
+  return g;
+}

diff --git a/charset/fb-charconv.h b/charset/fb-charconv.h
index 8e8c1486009df371289b8829bbe89be44792aee2..64e3beeaf75e5517f6f9a0125a688e7901916872 100644 (file)
--- a/charset/fb-charconv.h
+++ b/charset/fb-charconv.h
@@ -1,10 +1,11 @@
 /*
  *     Sherlock Library -- Charset Conversion Wrapper for Fast Buffered I/O
  *
- *     (c) 2003 Martin Mares <mj@ucw.cz>
+ *     (c) 2003--2005 Martin Mares <mj@ucw.cz>
  *
  *     This software may be freely distributed and used according to the terms
  *     of the GNU Lesser General Public License.
  */
 
+struct fastbuf *fb_wrap_charconv_in(struct fastbuf *f, int cs_from, int cs_to);
 struct fastbuf *fb_wrap_charconv_out(struct fastbuf *f, int cs_from, int cs_to);