mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-02-10 10:52:31 -05:00
47 lines
1.8 KiB
C++
47 lines
1.8 KiB
C++
// Copyright (c) 2009-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.
|
|
|
|
#ifndef BITCOIN_TEST_FUZZ_FUZZ_H
|
|
#define BITCOIN_TEST_FUZZ_FUZZ_H
|
|
|
|
#include <span.h>
|
|
|
|
#include <cstdint>
|
|
#include <functional>
|
|
#include <string_view>
|
|
|
|
/**
|
|
* Can be used to limit a theoretically unbounded loop. This caps the runtime
|
|
* to avoid timeouts or OOMs.
|
|
*/
|
|
#define LIMITED_WHILE(condition, limit) \
|
|
for (unsigned _count{limit}; (condition) && _count; --_count)
|
|
|
|
using FuzzBufferType = Span<const uint8_t>;
|
|
|
|
using TypeTestOneInput = std::function<void(FuzzBufferType)>;
|
|
using TypeInitialize = std::function<void()>;
|
|
using TypeHidden = bool;
|
|
|
|
void FuzzFrameworkRegisterTarget(std::string_view name, TypeTestOneInput target, TypeInitialize init, TypeHidden hidden);
|
|
|
|
inline void FuzzFrameworkEmptyInitFun() {}
|
|
|
|
#define FUZZ_TARGET(name) \
|
|
FUZZ_TARGET_INIT(name, FuzzFrameworkEmptyInitFun)
|
|
|
|
#define FUZZ_TARGET_INIT(name, init_fun) \
|
|
FUZZ_TARGET_INIT_HIDDEN(name, init_fun, false)
|
|
|
|
#define FUZZ_TARGET_INIT_HIDDEN(name, init_fun, hidden) \
|
|
void name##_fuzz_target(FuzzBufferType); \
|
|
struct name##_Before_Main { \
|
|
name##_Before_Main() \
|
|
{ \
|
|
FuzzFrameworkRegisterTarget(#name, name##_fuzz_target, init_fun, hidden); \
|
|
} \
|
|
} const static g_##name##_before_main; \
|
|
void name##_fuzz_target(FuzzBufferType buffer)
|
|
|
|
#endif // BITCOIN_TEST_FUZZ_FUZZ_H
|