From cab3e94ada175533d7da54efdbda89effcb31b66 Mon Sep 17 00:00:00 2001
From: ckolivas <kernel@kolivas.org>
Date: Sun, 26 Oct 2014 20:57:03 +1100
Subject: [PATCH] Allow reconnect to take a :url,port extension to specify
 where to reconnect to

---
 src/ckpool.c     |  2 +-
 src/stratifier.c | 17 ++++++++++++++---
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/src/ckpool.c b/src/ckpool.c
index 688d9731..8c522d46 100644
--- a/src/ckpool.c
+++ b/src/ckpool.c
@@ -296,7 +296,7 @@ retry:
 		send_unix_msg(sockd, "rejecting");
 	} else if (cmdmatch(buf, "reconnect")) {
 		LOGWARNING("Listener received request to send reconnect to clients");
-		send_procmsg(ckp->stratifier, "reconnect");
+		send_procmsg(ckp->stratifier, buf);
 		send_unix_msg(sockd, "reconnecting");
 	} else if (cmdmatch(buf, "restart")) {
 		LOGWARNING("Listener received restart message, attempting handover");
diff --git a/src/stratifier.c b/src/stratifier.c
index e5832d65..52073773 100644
--- a/src/stratifier.c
+++ b/src/stratifier.c
@@ -1145,11 +1145,22 @@ static void stratum_broadcast_message(const char *msg)
 
 /* Send a generic reconnect to all clients without parameters to make them
  * reconnect to the same server. */
-static void reconnect_clients(void)
+static void reconnect_clients(const char *cmd)
 {
+	char *port = strdupa(cmd), *url = NULL;
 	json_t *json_msg;
 
-	JSON_CPACK(json_msg, "{sosss[]}", "id", json_null(), "method", "client.reconnect",
+	strsep(&port, ":");
+	if (port)
+		url = strsep(&port, ",");
+	if (url && port) {
+		int port_no;
+
+		port_no = strtol(port, NULL, 10);
+		JSON_CPACK(json_msg, "{sosss[sii]}", "id", json_null(), "method", "client.reconnect",
+			"params", url, port_no, 0);
+	} else
+		JSON_CPACK(json_msg, "{sosss[]}", "id", json_null(), "method", "client.reconnect",
 		   "params");
 	stratum_broadcast(json_msg);
 }
@@ -1347,7 +1358,7 @@ retry:
 	} else if (cmdmatch(buf, "noblock")) {
 		block_reject(buf + 8);
 	} else if (cmdmatch(buf, "reconnect")) {
-		reconnect_clients();
+		reconnect_clients(buf);
 	} else if (cmdmatch(buf, "loglevel")) {
 		sscanf(buf, "loglevel=%d", &ckp->loglevel);
 	} else {