highperfocused/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-08-26 20:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

fuzz: Reset chainman state in process_message(s) targets

Browse Source
This commit is contained in:
MarcoFalke
2025-07-01 16:53:44 +02:00
parent fa9a3de09b
commit fa1a14a13a
2 changed files with 38 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/test/fuzz/process_message.cpp
View File

@@ -30,8 +30,20 @@
#include <vector> #include <vector>
namespace { namespace {
const TestingSetup* g_setup; TestingSetup* g_setup;
std::string_view LIMIT_TO_MESSAGE_TYPE{}; std::string_view LIMIT_TO_MESSAGE_TYPE{};
void ResetChainman(TestingSetup& setup)
{
SetMockTime(setup.m_node.chainman->GetParams().GenesisBlock().Time());
setup.m_node.chainman.reset();
setup.m_make_chainman();
setup.LoadVerifyActivateChainstate();
for (int i = 0; i < 2 * COINBASE_MATURITY; i++) {
MineBlock(setup.m_node, {});
}
setup.m_node.validation_signals->SyncWithValidationInterfaceQueue();
}
} // namespace } // namespace
void initialize_process_message() void initialize_process_message()
@@ -47,11 +59,7 @@ void initialize_process_message()
{}), {}),
}; };
g_setup = testing_setup.get(); g_setup = testing_setup.get();
SetMockTime(WITH_LOCK(g_setup->m_node.chainman->GetMutex(), return g_setup->m_node.chainman->ActiveTip()->Time())); ResetChainman(*g_setup);
for (int i = 0; i < 2 * COINBASE_MATURITY; i++) {
MineBlock(g_setup->m_node, {});
}
g_setup->m_node.validation_signals->SyncWithValidationInterfaceQueue();
} }
FUZZ_TARGET(process_message, .init = initialize_process_message) FUZZ_TARGET(process_message, .init = initialize_process_message)
@@ -63,6 +71,7 @@ FUZZ_TARGET(process_message, .init = initialize_process_message)
connman.ResetAddrCache(); connman.ResetAddrCache();
connman.ResetMaxOutboundCycle(); connman.ResetMaxOutboundCycle();
auto& chainman = static_cast<TestChainstateManager&>(*g_setup->m_node.chainman); auto& chainman = static_cast<TestChainstateManager&>(*g_setup->m_node.chainman);
const auto block_index_size{WITH_LOCK(chainman.GetMutex(), return chainman.BlockIndex().size())};
SetMockTime(1610000000); // any time to successfully reset ibd SetMockTime(1610000000); // any time to successfully reset ibd
chainman.ResetIbd(); chainman.ResetIbd();
chainman.DisableNextWrite(); chainman.DisableNextWrite();
@@ -111,4 +120,8 @@ FUZZ_TARGET(process_message, .init = initialize_process_message)
} }
g_setup->m_node.validation_signals->SyncWithValidationInterfaceQueue(); g_setup->m_node.validation_signals->SyncWithValidationInterfaceQueue();
g_setup->m_node.connman->StopNodes(); g_setup->m_node.connman->StopNodes();
if (block_index_size != WITH_LOCK(chainman.GetMutex(), return chainman.BlockIndex().size())) {
// Reuse the global chainman, but reset it when it is dirty
ResetChainman(*g_setup);
}
} }

25
src/test/fuzz/process_messages.cpp
View File

@@ -26,7 +26,19 @@
#include <vector> #include <vector>
namespace { namespace {
const TestingSetup* g_setup; TestingSetup* g_setup;
void ResetChainman(TestingSetup& setup)
{
SetMockTime(setup.m_node.chainman->GetParams().GenesisBlock().Time());
setup.m_node.chainman.reset();
setup.m_make_chainman();
setup.LoadVerifyActivateChainstate();
for (int i = 0; i < 2 * COINBASE_MATURITY; i++) {
MineBlock(setup.m_node, {});
}
setup.m_node.validation_signals->SyncWithValidationInterfaceQueue();
}
} // namespace } // namespace
void initialize_process_messages() void initialize_process_messages()
@@ -37,11 +49,7 @@ void initialize_process_messages()
{}), {}),
}; };
g_setup = testing_setup.get(); g_setup = testing_setup.get();
SetMockTime(WITH_LOCK(g_setup->m_node.chainman->GetMutex(), return g_setup->m_node.chainman->ActiveTip()->Time())); ResetChainman(*g_setup);
for (int i = 0; i < 2 * COINBASE_MATURITY; i++) {
MineBlock(g_setup->m_node, {});
}
g_setup->m_node.validation_signals->SyncWithValidationInterfaceQueue();
} }
FUZZ_TARGET(process_messages, .init = initialize_process_messages) FUZZ_TARGET(process_messages, .init = initialize_process_messages)
@@ -53,6 +61,7 @@ FUZZ_TARGET(process_messages, .init = initialize_process_messages)
connman.ResetAddrCache(); connman.ResetAddrCache();
connman.ResetMaxOutboundCycle(); connman.ResetMaxOutboundCycle();
auto& chainman = static_cast<TestChainstateManager&>(*g_setup->m_node.chainman); auto& chainman = static_cast<TestChainstateManager&>(*g_setup->m_node.chainman);
const auto block_index_size{WITH_LOCK(chainman.GetMutex(), return chainman.BlockIndex().size())};
SetMockTime(1610000000); // any time to successfully reset ibd SetMockTime(1610000000); // any time to successfully reset ibd
chainman.ResetIbd(); chainman.ResetIbd();
chainman.DisableNextWrite(); chainman.DisableNextWrite();
@@ -111,4 +120,8 @@ FUZZ_TARGET(process_messages, .init = initialize_process_messages)
} }
g_setup->m_node.validation_signals->SyncWithValidationInterfaceQueue(); g_setup->m_node.validation_signals->SyncWithValidationInterfaceQueue();
g_setup->m_node.connman->StopNodes(); g_setup->m_node.connman->StopNodes();
if (block_index_size != WITH_LOCK(chainman.GetMutex(), return chainman.BlockIndex().size())) {
// Reuse the global chainman, but reset it when it is dirty
ResetChainman(*g_setup);
}
} }