Check that state proofs aren't empty3924-handle-stateproof-in-rest-api-goal

1 files changed, 11 insertions, 1 deletions

diff --git a/node/node.go b/node/node.go
index c2a379de1..366144c72 100644
--- a/node/node.go
+++ b/node/node.go
@@ -404,7 +404,7 @@ func (node *AlgorandFullNode) Start() {
 // startMonitoringRoutines starts the internal monitoring routines used by the node.
 func (node *AlgorandFullNode) startMonitoringRoutines() {
 	node.monitoringRoutinesWaitGroup.Add(2)
-	go node.txPoolGaugeThread()
+	go node.txPoolGaugeThread(node.ctx.Done())
 	// Delete old participation keys
 	go node.oldKeyDeletionThread(node.ctx.Done())
 
@@ -899,6 +899,16 @@ func (node *AlgorandFullNode) InstallParticipationKey(partKeyBinary []byte) (acc
 	}
 	defer partkey.Close()
 
+	// We want to make sure that the state proof keys are included in the participation key
+
+	if partkey.StateProofSecrets == nil {
+		return account.ParticipationID{}, fmt.Errorf("cannot install partkey with missing state proof keys")
+	}
+
+	if len(partkey.StateProofSecrets.GetAllKeys()) == 0 {
+		return account.ParticipationID{}, fmt.Errorf("cannot install partkey with missing state proof keys")
+	}
+
 	if partkey.Parent == (basics.Address{}) {
 		return account.ParticipationID{}, fmt.Errorf("cannot install partkey with missing (zero) parent address")
 	}