fuzz: sv2 template provider

Co-authored-by: Christopher Coverdale <[email protected]>

src/Makefile.test.include

@@ -348,6 +348,7 @@ test_fuzz_fuzz_SOURCES = \
  test/fuzz/strprintf.cpp \
  test/fuzz/system.cpp \
  test/fuzz/sv2_messages.cpp \
+ test/fuzz/sv2_template_provider.cpp \
  test/fuzz/timedata.cpp \
  test/fuzz/torcontrol.cpp \
  test/fuzz/transaction.cpp \

src/test/fuzz/sv2_template_provider.cpp

@@ -0,0 +1,59 @@
+// Copyright (c) 2019-2021 The Bitcoin Core developers
+// Distributed under the MIT software license, see the accompanying
+// file COPYING or http://www.opensource.org/licenses/mit-license.php.
+
+#include <node/sv2_messages.h>
+#include <sv2_template_provider.h>
+#include <test/fuzz/fuzz.h>
+#include <streams.h>
+#include <test/util/net.h>
+#include <test/util/setup_common.h>
+
+FUZZ_TARGET(sv2_template_provider)
+{
+    const auto testing_setup = MakeNoLogFileContext<const TestingSetup>();
+    const TestingSetup* setup = testing_setup.get();
+
+    std::vector<uint8_t> random_bytes{buffer.begin(), buffer.end()};
+
+    Sv2TemplateProvider template_provider{*setup->m_node.chainman, *setup->m_node.mempool};
+
+    node::Sv2NewTemplateMsg best_new_template;
+    node::Sv2SetNewPrevHashMsg best_prev_hash;
+    std::map<uint64_t, std::unique_ptr<node::CBlockTemplate>> block_cache;
+    uint64_t template_id;
+
+    auto sock = std::make_shared<StaticContentsSock>(std::string(1, 'a'));
+    Sv2Client client{sock};
+
+    client.m_setup_connection_confirmed = false;
+    auto header = node::Sv2NetHeader{node::Sv2MsgType::SETUP_CONNECTION, static_cast<uint32_t>(random_bytes.size())};
+    auto setup_conn = node::Sv2NetMsg(std::move(header), std::move(random_bytes));
+
+    template_provider.ProcessSv2Message(setup_conn,
+                                        client,
+                                        best_new_template,
+                                        best_prev_hash,
+                                        block_cache,
+                                        template_id);
+
+    client.m_setup_connection_confirmed = true;
+    header = node::Sv2NetHeader{node::Sv2MsgType::COINBASE_OUTPUT_DATA_SIZE, static_cast<uint32_t>(random_bytes.size())};
+    auto coinbase_output_size = node::Sv2NetMsg(std::move(header), std::move(random_bytes));
+    template_provider.ProcessSv2Message(coinbase_output_size,
+                                        client,
+                                        best_new_template,
+                                        best_prev_hash,
+                                        block_cache,
+                                        template_id);
+
+    client.m_coinbase_output_data_size_recv = true;
+    header = node::Sv2NetHeader{node::Sv2MsgType::SUBMIT_SOLUTION, static_cast<uint32_t>(random_bytes.size())};
+    auto submit_solution = node::Sv2NetMsg(std::move(header), std::move(random_bytes));
+    template_provider.ProcessSv2Message(submit_solution,
+                                        client,
+                                        best_new_template,
+                                        best_prev_hash,
+                                        block_cache,
+                                        template_id);
+};