From: Richard Kettlewell <rjk@greenend.org.uk>
Date: Sun, 17 Nov 2013 11:22:33 +0000 (+0000)
Subject: socketio sink
X-Git-Tag: 5.2~104
X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~mdw/git/disorder/commitdiff_plain/59ef69ba619917734a8713db0db842ae3edd6086

socketio sink
---

diff --git a/lib/sink.c b/lib/sink.c
index 84b9c25..152b7dc 100644
--- a/lib/sink.c
+++ b/lib/sink.c
@@ -26,6 +26,7 @@
 
 #include "mem.h"
 #include "vector.h"
+#include "socketio.h"
 #include "log.h"
 #include "sink.h"
 #include "printf.h"
@@ -185,6 +186,47 @@ struct sink *sink_error(void) {
   return s;
 }
 
+/* socket sink *************************************************************/
+
+/** @brief Sink that writes to a socket handle */
+struct socketio_sink {
+  /** @brief Base member */
+  struct sink s;
+
+  struct socketio *sio;
+};
+
+static int sink_socketio_flush(struct sink *s) {
+  struct socketio_sink *ss = (struct socketio_sink *)s;
+  return socketio_flush(ss->sio);
+}
+
+/** @brief Write callback for @ref stdio_sink */
+static int sink_socketio_write(struct sink *s, const void *buffer, int nbytes) {
+  struct socketio_sink *ss = (struct socketio_sink *)s;
+  return socketio_write(ss->sio, buffer, nbytes);
+}
+
+static int sink_socketio_error(struct sink *s) {
+  struct socketio_sink *ss = (struct socketio_sink *)s;
+  return socketio_error(ss->sio);
+}
+
+/** @brief Create a sink that writes to a socket
+ * @param sio Socket IO context
+ * @return Pointer to new sink
+ */
+struct sink *sink_socketio(struct socketio *sio) {
+  struct socketio_sink *s = xmalloc(sizeof *s);
+
+  s->s.write = sink_socketio_write;
+  s->s.flush = sink_socketio_flush;
+  s->s.error = sink_socketio_error;
+  s->s.eclass = ec_native;
+  s->sio = sio;
+  return &s->s;
+}
+
 /*
 Local Variables:
 c-basic-offset:2
diff --git a/lib/sink.h b/lib/sink.h
index d3ab999..ff763a6 100644
--- a/lib/sink.h
+++ b/lib/sink.h
@@ -25,6 +25,7 @@
 #include <stdarg.h>
 
 struct dynstr;
+struct socketio;
 
 /** @brief Sink type
  *
@@ -70,6 +71,9 @@ struct sink *sink_discard(void);
 struct sink *sink_error(void);
 /* return a sink which fails all writes */
 
+struct sink *sink_socketio(struct socketio *sio);
+/* return a sink which writes to a socket */
+
 int sink_vprintf(struct sink *s, const char *fmt, va_list ap);
 int sink_printf(struct sink *s, const char *fmt, ...)
   attribute((format (printf, 2, 3)));