Skip to content

Commit fa4be8e

Browse files
author
MacroFake
committed
move-only: InitializeNode to handshake helper
1 parent fa70989 commit fa4be8e

File tree

5 files changed

+6
-6
lines changed

5 files changed

+6
-6
lines changed

src/test/fuzz/process_message.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,7 @@ void fuzz_target(FuzzBufferType buffer, const std::string& LIMIT_TO_MESSAGE_TYPE
8080
CNode& p2p_node = *ConsumeNodeAsUniquePtr(fuzzed_data_provider).release();
8181

8282
connman.AddTestNode(p2p_node);
83-
g_setup->m_node.peerman->InitializeNode(&p2p_node);
84-
FillNode(fuzzed_data_provider, connman, *g_setup->m_node.peerman, p2p_node);
83+
FillNode(fuzzed_data_provider, connman, p2p_node);
8584

8685
const auto mock_time = ConsumeTime(fuzzed_data_provider);
8786
SetMockTime(mock_time);

src/test/fuzz/process_messages.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,7 @@ FUZZ_TARGET_INIT(process_messages, initialize_process_messages)
4646
peers.push_back(ConsumeNodeAsUniquePtr(fuzzed_data_provider, i).release());
4747
CNode& p2p_node = *peers.back();
4848

49-
g_setup->m_node.peerman->InitializeNode(&p2p_node);
50-
FillNode(fuzzed_data_provider, connman, *g_setup->m_node.peerman, p2p_node);
49+
FillNode(fuzzed_data_provider, connman, p2p_node);
5150

5251
connman.AddTestNode(p2p_node);
5352
}

src/test/fuzz/util.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,7 @@ bool FuzzedSock::IsConnected(std::string& errmsg) const
289289
return false;
290290
}
291291

292-
void FillNode(FuzzedDataProvider& fuzzed_data_provider, ConnmanTestMsg& connman, PeerManager& peerman, CNode& node) noexcept
292+
void FillNode(FuzzedDataProvider& fuzzed_data_provider, ConnmanTestMsg& connman, CNode& node) noexcept
293293
{
294294
connman.Handshake(node,
295295
/*successfully_connected=*/fuzzed_data_provider.ConsumeBool(),

src/test/fuzz/util.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,7 @@ auto ConsumeNode(FuzzedDataProvider& fuzzed_data_provider, const std::optional<N
331331
}
332332
inline std::unique_ptr<CNode> ConsumeNodeAsUniquePtr(FuzzedDataProvider& fdp, const std::optional<NodeId>& node_id_in = std::nullopt) { return ConsumeNode<true>(fdp, node_id_in); }
333333

334-
void FillNode(FuzzedDataProvider& fuzzed_data_provider, ConnmanTestMsg& connman, PeerManager& peerman, CNode& node) noexcept;
334+
void FillNode(FuzzedDataProvider& fuzzed_data_provider, ConnmanTestMsg& connman, CNode& node) noexcept;
335335

336336
class FuzzedFileProvider
337337
{

src/test/util/net.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@ void ConnmanTestMsg::Handshake(CNode& node,
2424
auto& connman{*this};
2525
const CNetMsgMaker mm{0};
2626

27+
peerman.InitializeNode(&node);
28+
2729
CSerializedNetMsg msg_version{
2830
mm.Make(NetMsgType::VERSION,
2931
version, //

0 commit comments

Comments
 (0)