From 339ecc833ea4d7e135dc770602aaac9b2ba9cc5a Mon Sep 17 00:00:00 2001
From: Con Kolivas <kernel@kolivas.org>
Date: Sun, 10 Jan 2016 08:43:52 +1100
Subject: [PATCH] Whitelist cgminer based clients for receiving stratum
 messages and only send stratum messages to whitelisted clients

---
 src/stratifier.c | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/stratifier.c b/src/stratifier.c
index 365525c8..39f56927 100644
--- a/src/stratifier.c
+++ b/src/stratifier.c
@@ -285,6 +285,7 @@ struct stratum_instance {
 	char *useragent;
 	char *workername;
 	char *password;
+	bool messages; /* Is this a client that understands stratum messages */
 	int user_id;
 	int server; /* Which server is this instance bound to */
 
@@ -2314,6 +2315,11 @@ static void stratum_broadcast(sdata_t *sdata, json_t *val, const int msg_type)
 
 		if (!client_active(client))
 			continue;
+
+		/* Only send messages to whitelisted clients */
+		if (msg_type == SM_MSG && !client->messages)
+			continue;
+
 		client_msg = ckalloc(sizeof(ckmsg_t));
 		msg = ckzalloc(sizeof(smsg_t));
 		if (passthrough_subclient(client->id))
@@ -3736,6 +3742,10 @@ static json_t *parse_subscribe(stratum_instance_t *client, const int64_t client_
 	} else
 		client->useragent = ckzalloc(1);
 
+	/* Whitelist cgminer based clients to receive stratum messages */
+	if (strcasestr(client->useragent, "gminer"))
+		client->messages = true;
+
 	/* We got what we needed */
 	if (ckp->node)
 		return NULL;
@@ -4432,6 +4442,9 @@ static void stratum_send_message(sdata_t *sdata, const stratum_instance_t *clien
 {
 	json_t *json_msg;
 
+	/* Only send messages to whitelisted clients */
+	if (!client->messages)
+		return;
 	JSON_CPACK(json_msg, "{sosss[s]}", "id", json_null(), "method", "client.show_message",
 			     "params", msg);
 	stratum_add_send(sdata, json_msg, client->id, SM_MSG);