From 408d2f5170f3f87903a28f90f68c0324cffb2b1c Mon Sep 17 00:00:00 2001
From: Christian Decker <decker.christian@gmail.com>
Date: Fri, 10 Mar 2017 13:37:20 +0100
Subject: [PATCH] Async connection abstraction

This is an attempt at unifying all the async connection handling into
a reusable module.
---
 lightningd/Makefile     |  1 +
 lightningd/connection.c | 69 +++++++++++++++++++++++++++++++++++++++++
 lightningd/connection.h | 48 ++++++++++++++++++++++++++++
 3 files changed, 118 insertions(+)
 create mode 100644 lightningd/connection.c
 create mode 100644 lightningd/connection.h

diff --git a/lightningd/Makefile b/lightningd/Makefile
index 92f5aa40e..b5cbe6fde 100644
--- a/lightningd/Makefile
+++ b/lightningd/Makefile
@@ -39,6 +39,7 @@ LIGHTNINGD_LIB_SRC :=				\
 	lightningd/channel.c			\
 	lightningd/channel_config.c		\
 	lightningd/commit_tx.c			\
+	lightningd/connection.c			\
 	lightningd/cryptomsg.c			\
 	lightningd/crypto_sync.c		\
 	lightningd/debug.c			\
diff --git a/lightningd/connection.c b/lightningd/connection.c
new file mode 100644
index 000000000..d527a0d0e
--- /dev/null
+++ b/lightningd/connection.c
@@ -0,0 +1,69 @@
+#include "connection.h"
+#include <ccan/take/take.h>
+#include <wire/wire_io.h>
+
+static void daemon_conn_enqueue(struct daemon_conn *dc, u8 *msg)
+{
+	size_t n = tal_count(dc->msg_out);
+	tal_resize(&dc->msg_out, n + 1);
+	dc->msg_out[n] = tal_dup_arr(dc->ctx, u8, msg, tal_count(msg), 0);
+}
+
+static const u8 *daemon_conn_dequeue(struct daemon_conn *dc)
+{
+	const u8 *msg;
+	size_t n = tal_count(dc->msg_out);
+
+	if (n == 0)
+		return NULL;
+	msg = dc->msg_out[0];
+	memmove(dc->msg_out, dc->msg_out + 1, sizeof(dc->msg_in[0]) * (n-1));
+	tal_resize(&dc->msg_out, n-1);
+	return msg;
+}
+
+struct io_plan *daemon_conn_read_next(struct io_conn *conn,
+				      struct daemon_conn *dc)
+{
+	dc->msg_in = tal_free(dc->msg_in);
+	return io_read_wire(conn, dc->ctx, &dc->msg_in, dc->daemon_conn_recv,
+			    dc);
+}
+
+static struct io_plan *daemon_conn_write_next(struct io_conn *conn,
+				       struct daemon_conn *dc)
+{
+	const u8 *msg = daemon_conn_dequeue(dc);
+	if (msg) {
+		return io_write_wire(conn, take(msg), daemon_conn_write_next, dc);
+	} else {
+		return io_out_wait(conn, dc, daemon_conn_write_next, dc);
+	}
+}
+
+static struct io_plan *daemon_conn_start(struct io_conn *conn,
+					 struct daemon_conn *dc)
+{
+	dc->conn = conn;
+	return io_duplex(conn, daemon_conn_read_next(conn, dc),
+			 daemon_conn_write_next(conn, dc));
+}
+
+void daemon_conn_init(tal_t *ctx, struct daemon_conn *dc, int fd,
+		      struct io_plan *(*daemon_conn_recv)(struct io_conn *,
+							  struct daemon_conn *))
+{
+	dc->daemon_conn_recv = daemon_conn_recv;
+
+	dc->ctx = ctx;
+	dc->msg_in = NULL;
+	dc->msg_out = tal_arr(ctx, u8 *, 0);
+	dc->conn_fd = fd;
+	io_new_conn(ctx, fd, daemon_conn_start, dc);
+}
+
+void daemon_conn_send(struct daemon_conn *dc, u8 *msg)
+{
+	daemon_conn_enqueue(dc, msg);
+	io_wake(dc);
+}
diff --git a/lightningd/connection.h b/lightningd/connection.h
new file mode 100644
index 000000000..c79cea32e
--- /dev/null
+++ b/lightningd/connection.h
@@ -0,0 +1,48 @@
+#ifndef LIGHTNING_LIGHTNINGD_CONNECTION_H
+#define LIGHTNING_LIGHTNINGD_CONNECTION_H
+
+#include "config.h"
+#include <ccan/io/io.h>
+#include <ccan/short_types/short_types.h>
+
+struct daemon_conn {
+	/* Context to tallocate all things from, possibly the
+	 * container of this connection. */
+	tal_t *ctx;
+
+	/* Last message we received */
+	u8 *msg_in;
+
+	/* Array of queued outgoing messages */
+	u8 **msg_out;
+
+	int conn_fd;
+	struct io_conn *conn;
+
+	/* Callback for incoming messages */
+	struct io_plan *(*daemon_conn_recv)(struct io_conn *conn, struct daemon_conn *);
+};
+
+/**
+ * daemon_conn_init - Initialize a new daemon connection
+ *
+ * @ctx: context to allocate from
+ * @dc: daemon_conn to initialize
+ * @fd: socket file descriptor to wrap
+ * @daemon_conn_recv: callback function to be called upon receiving a message
+ */
+void daemon_conn_init(tal_t *ctx, struct daemon_conn *dc, int fd,
+		      struct io_plan *(*daemon_conn_recv)(struct io_conn *,
+							  struct daemon_conn *));
+/**
+ * daemon_conn_send - Enqueue an outgoing message to be sent
+ */
+void daemon_conn_send(struct daemon_conn *dc, u8 *msg);
+
+/**
+ * daemon_conn_read_next - Read the next message
+ */
+struct io_plan *daemon_conn_read_next(struct io_conn *conn,
+				      struct daemon_conn *dc);
+
+#endif /* LIGHTNING_LIGHTNINGD_CONNECTION_H */