From faf87ceeee08abe8ddb77b0371a0b21e39dbb0b7 Mon Sep 17 00:00:00 2001
From: Con Kolivas <kernel@kolivas.org>
Date: Sat, 17 May 2014 20:36:59 +1000
Subject: [PATCH] Store the proxy base subscribe details in a struct protected
 by the workbase lock

---
 src/generator.c  |  4 ++++
 src/stratifier.c | 37 +++++++++++++++++++++++++++++++++----
 2 files changed, 37 insertions(+), 4 deletions(-)

diff --git a/src/generator.c b/src/generator.c
index 07213570..5a0688ff 100644
--- a/src/generator.c
+++ b/src/generator.c
@@ -353,6 +353,10 @@ static bool parse_subscribe(connsock_t *cs, proxy_instance_t *proxi)
 	}
 	proxi->enonce1 = strdup(string);
 	proxi->nonce1len = strlen(proxi->enonce1) / 2;
+	if (proxi->nonce1len > 15) {
+		LOGWARNING("Nonce1 too long at %d", proxi->nonce1len);
+		goto out;
+	}
 	proxi->enonce1bin = ckalloc(proxi->nonce1len);
 	hex2bin(proxi->enonce1bin, proxi->enonce1, proxi->nonce1len);
 	tmp = json_array_get(res_val, 2);
diff --git a/src/stratifier.c b/src/stratifier.c
index 9882aa6b..af114dd6 100644
--- a/src/stratifier.c
+++ b/src/stratifier.c
@@ -137,6 +137,17 @@ static cklock_t workbase_lock;
 /* For the hashtable of all workbases */
 static workbase_t *workbases;
 static workbase_t *current_workbase;
+
+static struct {
+	double diff;
+
+	char enonce1[32];
+
+	int nonce2len;
+	int subnonce2len;
+	uint32_t subnonce2;
+} proxy_base;
+
 static uint64_t workbase_id;
 static uint64_t blockchange_id;
 static char lasthash[68];
@@ -429,17 +440,34 @@ static void update_base(ckpool_t *ckp)
 	stratum_broadcast_update(new_block);
 }
 
-static void update_subscribe(ckpool_t *ckp)
+static bool update_subscribe(ckpool_t *ckp)
 {
+	json_t *val;
 	char *buf;
 
 	buf = send_recv_proc(ckp->generator, "getsubscribe");
 	if (unlikely(!buf)) {
 		LOGWARNING("Failed to get subscribe from generator in update_notify");
-		return;
+		return false;
 	}
-	LOGWARNING("Subscribe was %s", buf);
+	LOGDEBUG("Update subscribe: %s", buf);
+	val = json_loads(buf, 0, NULL);
 	free(buf);
+
+	ck_wlock(&workbase_lock);
+	if (!proxy_base.diff)
+		proxy_base.diff = 1;
+	/* Length is checked by generator */
+	strcpy(proxy_base.enonce1, json_string_value(json_object_get(val, "enonce1")));
+	proxy_base.nonce2len = json_integer_value(json_object_get(val, "nonce2len"));
+	if (proxy_base.nonce2len > 5)
+		proxy_base.subnonce2len = 4;
+	else
+		proxy_base.subnonce2len = 2;
+	ck_wunlock(&workbase_lock);
+
+	json_decref(val);
+	return true;
 }
 
 static void update_notify(ckpool_t *ckp)
@@ -672,7 +700,8 @@ retry:
 		goto reset;
 	} else if (!strncasecmp(buf, "subscribe", 9)) {
 		/* Proxifier has a new subscription */
-		update_subscribe(ckp);
+		if (!update_subscribe(ckp))
+			goto out;
 		goto reset;
 	} else if (!strncasecmp(buf, "notify", 6)) {
 		/* Proxifier has a new notify ready */