summaryrefslogtreecommitdiffstats
path: root/src/ioline.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/ioline.c')
-rw-r--r--src/ioline.c190
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;
+}