llmq: Make sure CDKGSessionHandler uses a valid LLMQ type

xdustinface · UdjinM6 · xdustinface · commit ef66d06620d5 · 2020-07-15T00:07:27.000+02:00
Co-Authored-By: UdjinM6 &lt;UdjinM6@users.noreply.github.com&gt;
diff --git a/src/llmq/quorums_dkgsessionhandler.cpp b/src/llmq/quorums_dkgsessionhandler.cpp
@@ -94,6 +94,9 @@ CDKGSessionHandler::CDKGSessionHandler(const Consensus::LLMQParams& _params, CBL
     pendingJustifications((size_t)_params.size * 2, MSG_QUORUM_JUSTIFICATION),
     pendingPrematureCommitments((size_t)_params.size * 2, MSG_QUORUM_PREMATURE_COMMITMENT)
 {
+    if (params.type == Consensus::LLMQ_NONE) {
+        throw std::runtime_error("Can't initialize CDKGSessionHandler with LLMQ_NONE type.");
+    }
 }
 
 CDKGSessionHandler::~CDKGSessionHandler()

Original file line number	Diff line number	Diff line change
`@@ -94,6 +94,9 @@ CDKGSessionHandler::CDKGSessionHandler(const Consensus::LLMQParams& _params, CBL`
`94`	`94`	`pendingJustifications((size_t)_params.size * 2, MSG_QUORUM_JUSTIFICATION),`
`95`	`95`	`pendingPrematureCommitments((size_t)_params.size * 2, MSG_QUORUM_PREMATURE_COMMITMENT)`
`96`	`96`	`{`
	`97`	`+ if (params.type == Consensus::LLMQ_NONE) {`
	`98`	`+ throw std::runtime_error("Can't initialize CDKGSessionHandler with LLMQ_NONE type.");`
	`99`	`+ }`
`97`	`100`	`}`
`98`	`101`
`99`	`102`	`CDKGSessionHandler::~CDKGSessionHandler()`