1 files changed, 190 insertions, 0 deletions
diff --git a/src/ioline.c b/src/ioline.c
new file mode 100644
index 00000000..c37737a6
--- /dev/null
+++ b/src/ioline.c
@@ -0,0 +1,190 @@
+#include <errno.h>
+#include <stdio.h>
+#include <assert.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "ioline.h"
+
+#define BUFFER_LIMIT (64*1024)
+#define READ_SIZE (1024)
+
+struct ioline {
+    struct iochannel *io;
+    int dead;
+
+    char *wbuf;
+    size_t wbuf_length, wbuf_index, wbuf_valid_length;
+
+    char *rbuf;
+    size_t rbuf_length, rbuf_index, rbuf_valid_length;
+
+    void (*callback)(struct ioline*io, const char *s, void *userdata);
+    void *userdata;
+};
+
+static void io_callback(struct iochannel*io, void *userdata);
+static int do_write(struct ioline *l);
+
+struct ioline* ioline_new(struct iochannel *io) {
+    struct ioline *l;
+    assert(io);
+    
+    l = malloc(sizeof(struct ioline));
+    assert(l);
+    l->io = io;
+    l->dead = 0;
+
+    l->wbuf = NULL;
+    l->wbuf_length = l->wbuf_index = l->wbuf_valid_length = 0;
+
+    l->rbuf = NULL;
+    l->rbuf_length = l->rbuf_index = l->rbuf_valid_length = 0;
+
+    l->callback = NULL;
+    l->userdata = NULL;
+
+    iochannel_set_callback(io, io_callback, l);
+    
+    return l;
+}
+
+void ioline_free(struct ioline *l) {
+    assert(l);
+    iochannel_free(l->io);
+    free(l->wbuf);
+    free(l->rbuf);
+    free(l);
+}
+
+void ioline_puts(struct ioline *l, const char *c) {
+    size_t len;
+    assert(l && c);
+    
+    len = strlen(c);
+    if (len > BUFFER_LIMIT - l->wbuf_valid_length)
+        len = BUFFER_LIMIT - l->wbuf_valid_length;
+
+    if (!len)
+        return;
+    
+    if (len > l->wbuf_length - l->wbuf_valid_length) {
+        size_t n = l->wbuf_valid_length+len;
+        char *new = malloc(n);
+        if (l->wbuf) {
+            memcpy(new, l->wbuf+l->wbuf_index, l->wbuf_valid_length);
+            free(l->wbuf);
+        }
+        l->wbuf = new;
+        l->wbuf_length = n;
+        l->wbuf_index = 0;
+    } else if (len > l->wbuf_length - l->wbuf_valid_length - l->wbuf_index) {
+        memmove(l->wbuf, l->wbuf+l->wbuf_index, l->wbuf_valid_length);
+        l->wbuf_index = 0;
+    }
+
+    memcpy(l->wbuf+l->wbuf_index+l->wbuf_valid_length, c, len);
+    l->wbuf_valid_length += len;
+
+    do_write(l);
+}
+
+void ioline_set_callback(struct ioline*l, void (*callback)(struct ioline*io, const char *s, void *userdata), void *userdata) {
+    assert(l && callback);
+    l->callback = callback;
+    l->userdata = userdata;
+}
+
+static int do_read(struct ioline *l) {
+    ssize_t r;
+    size_t m, len;
+    char *p, *e;
+    assert(l);
+
+    if (!iochannel_is_readable(l->io))
+        return 0;
+
+    len = l->rbuf_length - l->rbuf_index - l->rbuf_valid_length;
+
+    if (len < READ_SIZE) {
+        size_t n = l->rbuf_valid_length+READ_SIZE;
+
+        if (n >= BUFFER_LIMIT)
+            n = BUFFER_LIMIT;
+        
+        if (l->rbuf_length >= n) {
+            if (l->rbuf_valid_length)
+                memmove(l->rbuf, l->rbuf+l->rbuf_index, l->rbuf_valid_length);
+        } else {
+            char *new = malloc(n);
+            if (l->rbuf_valid_length)
+                memcpy(new, l->rbuf+l->rbuf_index, l->rbuf_valid_length);
+            free(l->rbuf);
+            l->rbuf = new;
+            l->rbuf_length = n;
+        }
+        
+        l->rbuf_index = 0;
+    }
+
+    len = l->rbuf_length - l->rbuf_index - l->rbuf_valid_length;
+    
+    if ((r = iochannel_read(l->io, l->rbuf+l->rbuf_index+l->rbuf_valid_length, len)) <= 0)
+        return -1;
+
+    e = memchr(l->rbuf+l->rbuf_index+l->rbuf_valid_length, '\n', r);
+    l->rbuf_valid_length += r;
+
+    if (!e && l->rbuf_valid_length >= BUFFER_LIMIT)
+        e = l->rbuf+BUFFER_LIMIT-1;
+        
+    *e = 0;
+    p = l->rbuf+l->rbuf_index;
+    m = strlen(p);
+
+    if (l->callback)
+        l->callback(l, p, l->userdata);
+
+    l->rbuf_index += m+1;
+    l->rbuf_valid_length -= m+1;
+
+    if (l->rbuf_valid_length == 0)
+        l->rbuf_index = 0;
+
+    return 0;
+}
+
+static int do_write(struct ioline *l) {
+    ssize_t r;
+    assert(l);
+
+    if (!l->wbuf_valid_length || !iochannel_is_writable(l->io))
+        return 0;
+    
+    if ((r = iochannel_write(l->io, l->wbuf+l->wbuf_index, l->wbuf_valid_length)) < 0)
+        return -1;
+
+    l->wbuf_valid_length -= r;
+    if (l->wbuf_valid_length == 0)
+        l->wbuf_index = 0;
+
+    return 0;
+}
+
+static void io_callback(struct iochannel*io, void *userdata) {
+    struct ioline *l = userdata;
+    assert(io && l);
+    
+    if (!l->dead && do_read(l) < 0)
+        goto fail;
+    
+    if (!l->dead && do_write(l) < 0)
+        goto fail;
+    
+    return;
+
+fail:
+    if (l->callback)
+        l->callback(l, NULL, l->userdata);
+    l->dead = 1;
+}