X-Git-Url: https://git.distorted.org.uk/~mdw/disorder/blobdiff_plain/902b9f3ff9d45ff4dccc35b1acaa3311865656ad..96f6312a0a827d2f90e7ab88d3a203c56eb82a9f:/lib/sink.c

diff --git a/lib/sink.c b/lib/sink.c
index 84b9c25..0a3f701 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,119 @@ 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;
+}
+
+/* stdio source *************************************************************/
+
+/** @brief Source that reads from a socket handle */
+struct stdio_source {
+  /** @brief Base member */
+  struct source s;
+
+  FILE *fp;
+};
+
+static int source_stdio_getc(struct source *s) {
+  return getc(((struct stdio_source *)s)->fp);
+}
+
+static int source_stdio_error(struct source *s) {
+  FILE *fp = ((struct stdio_source *)s)->fp;
+  if(ferror(fp)) {
+#if _WIN32
+    return GetLastError();
+#else
+    return errno;
+#endif
+  }
+  return 0;
+}
+
+static int source_stdio_eof(struct source *s) {
+  FILE *fp = ((struct stdio_source *)s)->fp;
+  return feof(fp);
+}
+
+struct source *source_stdio(FILE *fp) {
+  struct stdio_source *ss = xmalloc(sizeof *ss);
+  ss->s.getch = source_stdio_getc;
+  ss->s.error = source_stdio_error;
+  ss->s.eof = source_stdio_eof;
+  ss->s.eclass = ec_errno;
+  ss->fp = fp;
+  return (struct source *)ss;
+}
+
+/* socket source ***********************************************************/
+
+/** @brief Source that reads from a socket handle */
+struct socket_source {
+  /** @brief Base member */
+  struct source s;
+
+  struct socketio *sio;
+};
+
+static int source_socketio_getc(struct source *s) {
+  return socketio_getc(((struct socket_source *)s)->sio);
+}
+
+static int source_socketio_error(struct source *s) {
+  return socketio_error(((struct socket_source *)s)->sio);
+}
+static int source_socketio_eof(struct source *s) {
+  return socketio_eof(((struct socket_source *)s)->sio);
+}
+
+struct source *source_socketio(struct socketio *sio) {
+  struct socket_source *ss = xmalloc(sizeof *ss);
+  ss->s.getch = source_socketio_getc;
+  ss->s.error = source_socketio_error;
+  ss->s.eof = source_socketio_eof;
+  ss->s.eclass = ec_native;
+  ss->sio = sio;
+  return (struct source *)ss;
+}
+
 /*
 Local Variables:
 c-basic-offset:2