From 4916591ae5f7c1e2edc6c8e8d7c07a2ee31e10c7 Mon Sep 17 00:00:00 2001
From: Con Kolivas <kernel@kolivas.org>
Date: Fri, 25 Apr 2014 20:27:28 +1000
Subject: [PATCH] Create response to mining.subscribe request and deny all
 other invalid requests

---
 src/stratifier.c | 76 +++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 72 insertions(+), 4 deletions(-)

diff --git a/src/stratifier.c b/src/stratifier.c
index 4e136eb8..60c33f5d 100644
--- a/src/stratifier.c
+++ b/src/stratifier.c
@@ -115,7 +115,7 @@ struct stratum_instance {
 	char enonce1[20];
 	double diff;
 	bool authorised;
-	char *packagever;
+	char *name;
 };
 
 typedef struct stratum_instance stratum_instance_t;
@@ -439,15 +439,83 @@ static void *blockupdate(void *arg)
 	return NULL;
 }
 
-static json_t *gen_json_result(int client_id, json_t *method, json_t *params)
+static json_t *parse_subscribe(int client_id, json_t *params_val)
 {
+	stratum_instance_t *client = NULL;
+	char *enonce1;
+	int arr_size;
+	json_t *ret;
+
+	if (unlikely(!json_is_array(params_val)))
+		return json_string("params not an array");
+
+	ck_rlock(&instance_lock);
+	client = __instance_by_id(client_id);
+	if (unlikely(!client)) {
+		ck_runlock(&instance_lock);
+		LOGERR("Failed to find client id %d in hashtable!", client_id);
+		return NULL;
+	}
+	arr_size = json_array_size(params_val);
+	if (arr_size > 0) {
+		const char *buf;
+
+		buf = json_string_value(json_array_get(params_val, 0));
+		if (buf && strlen(buf))
+			client->name = strdup(buf);
+		if (arr_size > 1) {
+			/* This would be the session id for reconnect */
+			buf = json_string_value(json_array_get(params_val, 1));
+			LOGDEBUG("Found old session id %s", buf);
+			/* Add matching here */
+		}
+	}
+	enonce1 = strdup(client->enonce1);
+	ck_runlock(&instance_lock);
+
+	ck_rlock(&workbase_lock);
+	ret = json_pack("[[s,s],s,i]", "mining.notify", enonce1, enonce1, workbases->enonce2varlen);
+	ck_runlock(&workbase_lock);
+
+	free(enonce1);
+
+	return ret;
+}
+
+/* We should have already determined all the values passed to this are valid
+ * by now. Set update if we should also send the latest stratum parameters */
+static json_t *gen_json_result(int client_id, json_t *method_val, json_t *params_val, bool *update)
+{
+	stratum_instance_t *client = NULL;
+	const char *method;
+
+	method = json_string_value(method_val);
+	if (!strncasecmp(method, "mining.subscribe", 16)) {
+		*update = true;
+		return parse_subscribe(client_id, params_val);
+	}
+
+	/* We should only accept authorised requests from here on */
+	ck_rlock(&instance_lock);
+	client = __instance_by_id(client_id);
+	if (unlikely(!client)) {
+		ck_runlock(&instance_lock);
+		LOGERR("Failed to find client id %d in hashtable!", client_id);
+		return NULL;
+	}
+	ck_runlock(&instance_lock);
+
+	if (!client->authorised)
+		return json_string("Unauthorised");
+
 	return json_string("Empty");
 }
 
 static void parse_instance_msg(int client_id, json_t *msg)
 {
 	json_t *result_val = NULL, *err_val = NULL, *id_val = NULL;
-	json_t *method, *msg_id, *params;
+	json_t *method, *params;
+	bool update = false;
 	json_t *json_msg;
 
 	json_msg = json_object();
@@ -475,7 +543,7 @@ static void parse_instance_msg(int client_id, json_t *msg)
 		goto out;
 	}
 	err_val = json_null();
-	result_val = gen_json_result(client_id, method, params);
+	result_val = gen_json_result(client_id, method, params, &update);
 out:
 	json_object_set_nocheck(json_msg, "id", id_val);
 	json_object_set_nocheck(json_msg, "error", err_val);