From 24c2a4a7bf098873b297903e547051d1ff316ad5 Mon Sep 17 00:00:00 2001
From: Con Kolivas <kernel@kolivas.org>
Date: Mon, 20 Apr 2015 16:03:54 +1000
Subject: [PATCH] Convert the generator loop to use unix receive queues

---
 src/generator.c | 63 ++++++++++++++++++++++++-------------------------
 1 file changed, 31 insertions(+), 32 deletions(-)

diff --git a/src/generator.c b/src/generator.c
index 74e3e784..1a06206c 100644
--- a/src/generator.c
+++ b/src/generator.c
@@ -267,12 +267,20 @@ static void kill_server(server_instance_t *si)
 	dealloc(si->data);
 }
 
+static void clear_unix_msg(unix_msg_t **umsg)
+{
+	if (*umsg) {
+		free((*umsg)->buf);
+		free(*umsg);
+		*umsg = NULL;
+	}
+}
+
 static int gen_loop(proc_instance_t *pi)
 {
-	int sockd = -1, ret = 0, selret;
 	server_instance_t *si = NULL;
 	bool reconnecting = false;
-	unixsock_t *us = &pi->us;
+	unix_msg_t *umsg = NULL;
 	ckpool_t *ckp = pi->ckp;
 	gdata_t *gdata = ckp->data;
 	bool started = false;
@@ -280,9 +288,10 @@ static int gen_loop(proc_instance_t *pi)
 	connsock_t *cs;
 	gbtbase_t *gbt;
 	char hash[68];
+	int ret = 0;
 
 reconnect:
-	Close(sockd);
+	clear_unix_msg(&umsg);
 	if (si) {
 		kill_server(si);
 		reconnecting = true;
@@ -299,36 +308,24 @@ reconnect:
 	}
 
 retry:
-	Close(sockd);
+	clear_unix_msg(&umsg);
 	ckmsgq_add(gdata->srvchk, si);
 
 	do {
-		selret = wait_read_select(us->sockd, 5);
-		if (!selret && !ping_main(ckp)) {
+		umsg = get_unix_msg(pi);
+		if (unlikely(!umsg &&!ping_main(ckp))) {
 			LOGEMERG("Generator failed to ping main process, exiting");
 			ret = 1;
 			goto out;
 		}
-	} while (selret < 1);
+	} while (!umsg);
 
 	if (unlikely(cs->fd < 0)) {
 		LOGWARNING("Bitcoind socket invalidated, will attempt failover");
 		goto reconnect;
 	}
 
-	sockd = accept(us->sockd, NULL, NULL);
-	if (sockd < 0) {
-		LOGEMERG("Failed to accept on generator socket");
-		ret = 1;
-		goto out;
-	}
-
-	dealloc(buf);
-	buf = recv_unix_msg(sockd);
-	if (!buf) {
-		LOGWARNING("Failed to get message in gen_loop");
-		goto retry;
-	}
+	buf = umsg->buf;
 	LOGDEBUG("Generator received request: %s", buf);
 	if (cmdmatch(buf, "shutdown")) {
 		ret = 0;
@@ -338,41 +335,41 @@ retry:
 		if (!gen_gbtbase(cs, gbt)) {
 			LOGWARNING("Failed to get block template from %s:%s",
 				   cs->url, cs->port);
-			send_unix_msg(sockd, "Failed");
+			send_unix_msg(umsg->sockd, "Failed");
 			goto reconnect;
 		} else {
 			char *s = json_dumps(gbt->json, JSON_NO_UTF8);
 
-			send_unix_msg(sockd, s);
+			send_unix_msg(umsg->sockd, s);
 			free(s);
 			clear_gbtbase(gbt);
 		}
 	} else if (cmdmatch(buf, "getbest")) {
 		if (si->notify)
-			send_unix_msg(sockd, "notify");
+			send_unix_msg(umsg->sockd, "notify");
 		else if (!get_bestblockhash(cs, hash)) {
 			LOGINFO("No best block hash support from %s:%s",
 				cs->url, cs->port);
-			send_unix_msg(sockd, "failed");
+			send_unix_msg(umsg->sockd, "failed");
 		} else {
 			if (unlikely(!started)) {
 				started = true;
 				LOGWARNING("%s generator ready", ckp->name);
 			}
-			send_unix_msg(sockd, hash);
+			send_unix_msg(umsg->sockd, hash);
 		}
 	} else if (cmdmatch(buf, "getlast")) {
 		int height;
 
 		if (si->notify)
-			send_unix_msg(sockd, "notify");
+			send_unix_msg(umsg->sockd, "notify");
 		else if ((height = get_blockcount(cs)) == -1) {
-			send_unix_msg(sockd,  "failed");
+			send_unix_msg(umsg->sockd,  "failed");
 			goto reconnect;
 		} else {
 			LOGDEBUG("Height: %d", height);
 			if (!get_blockhash(cs, height, hash)) {
-				send_unix_msg(sockd, "failed");
+				send_unix_msg(umsg->sockd, "failed");
 				goto reconnect;
 			} else {
 				if (unlikely(!started)) {
@@ -380,7 +377,7 @@ retry:
 					LOGWARNING("%s generator ready", ckp->name);
 				}
 
-				send_unix_msg(sockd, hash);
+				send_unix_msg(umsg->sockd, hash);
 				LOGDEBUG("Hash: %s", hash);
 			}
 		}
@@ -395,16 +392,16 @@ retry:
 		send_proc(ckp->stratifier, blockmsg);
 	} else if (cmdmatch(buf, "checkaddr:")) {
 		if (validate_address(cs, buf + 10))
-			send_unix_msg(sockd, "true");
+			send_unix_msg(umsg->sockd, "true");
 		else
-			send_unix_msg(sockd, "false");
+			send_unix_msg(umsg->sockd, "false");
 	} else if (cmdmatch(buf, "reconnect")) {
 		goto reconnect;
 	} else if (cmdmatch(buf, "loglevel")) {
 		sscanf(buf, "loglevel=%d", &ckp->loglevel);
 	} else if (cmdmatch(buf, "ping")) {
 		LOGDEBUG("Generator received ping request");
-		send_unix_msg(sockd, "pong");
+		send_unix_msg(umsg->sockd, "pong");
 	}
 	goto retry;
 
@@ -2734,6 +2731,8 @@ int generator(proc_instance_t *pi)
 	gdata = ckzalloc(sizeof(gdata_t));
 	ckp->data = gdata;
 	gdata->ckp = ckp;
+	create_unix_receiver(pi);
+
 	if (ckp->proxy) {
 		char *buf = NULL;