Give more descriptive error for non-WSH(OP_TRUE) block generation

ElementsProject · Oct 8, 2019 · 3a2888c · 3a2888c
1 parent ee14d88
commit 3a2888c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/rpc/mining.cpp b/src/rpc/mining.cpp
@@ -142,11 +142,13 @@ UniValue generateBlocks(std::shared_ptr<CReserveScript> coinbaseScript, int nGen
             }
         }
 
-        // Fill out block witness if dynamic federation is enabled
-        // since we are assuming WSH(OP_TRUE)
-        if (!pblock->m_dynafed_params.IsNull()) {
-            CScript op_true(OP_TRUE);
+        // Handle OP_TRUE m_signblockscript case
+        CScript op_true(OP_TRUE);
+        if (pblock->m_dynafed_params.m_current.m_signblockscript ==
+                GetScriptForDestination(WitnessV0ScriptHash(op_true))) {
             pblock->m_signblock_witness.stack.push_back(std::vector<unsigned char>(op_true.begin(), op_true.end()));
+        } else if (!pblock->m_dynafed_params.IsNull()) {
+            throw JSONRPCError(RPC_MISC_ERROR, "Unable to fill out dynamic federation signblockscript witness, are you sure it's WSH(OP_TRUE)?");
         }
 
         std::shared_ptr<const CBlock> shared_pblock = std::make_shared<const CBlock>(*pblock);