mirror of
https://github.com/bitcoin/bitcoin.git
synced 2026-03-21 21:20:07 +01:00
0802398e74
* Make the methods of `CThreadInterrupt` virtual and store a pointer to it in `CConnman`, thus making it possible to override with a mocked instance. * Initialize `CConnman::m_interrupt_net` from the constructor, making it possible for callers to supply mocked version. * Introduce `FuzzedThreadInterrupt` and `ConsumeThreadInterrupt()` and use them in `src/test/fuzz/connman.cpp` and `src/test/fuzz/i2p.cpp`. This improves the CPU utilization of the `connman` fuzz test. As a nice side effect, the `std::shared_ptr` used for `CConnman::m_interrupt_net` resolves the possible lifetime issues with it (see the removed comment for that variable).
34 lines
1.0 KiB
C++
34 lines
1.0 KiB
C++
// Copyright (c) 2024-present The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#ifndef BITCOIN_TEST_FUZZ_UTIL_THREADINTERRUPT_H
|
|
#define BITCOIN_TEST_FUZZ_UTIL_THREADINTERRUPT_H
|
|
|
|
#include <test/fuzz/FuzzedDataProvider.h>
|
|
#include <util/threadinterrupt.h>
|
|
|
|
#include <memory>
|
|
|
|
/**
|
|
* Mocked CThreadInterrupt that returns "randomly" whether it is interrupted and never sleeps.
|
|
*/
|
|
class FuzzedThreadInterrupt : public CThreadInterrupt
|
|
{
|
|
public:
|
|
explicit FuzzedThreadInterrupt(FuzzedDataProvider& fuzzed_data_provider);
|
|
|
|
virtual bool interrupted() const override;
|
|
virtual bool sleep_for(Clock::duration) override;
|
|
|
|
private:
|
|
FuzzedDataProvider& m_fuzzed_data_provider;
|
|
};
|
|
|
|
[[nodiscard]] inline std::shared_ptr<CThreadInterrupt> ConsumeThreadInterrupt(FuzzedDataProvider& fuzzed_data_provider)
|
|
{
|
|
return std::make_shared<FuzzedThreadInterrupt>(fuzzed_data_provider);
|
|
}
|
|
|
|
#endif // BITCOIN_TEST_FUZZ_UTIL_THREADINTERRUPT_H
|