Branch data     Line data    Source code

       1                 :            : // Copyright (c) 2009-2021 The Bitcoin Core developers
       2                 :            : // Distributed under the MIT software license, see the accompanying
       3                 :            : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
       4                 :            : 
       5                 :            : #ifndef BITCOIN_TEST_FUZZ_FUZZ_H
       6                 :            : #define BITCOIN_TEST_FUZZ_FUZZ_H
       7                 :            : 
       8                 :            : #include <span.h>
       9                 :            : 
      10                 :            : #include <cstdint>
      11                 :            : #include <functional>
      12                 :            : #include <string_view>
      13                 :            : 
      14                 :            : /**
      15                 :            :  * Can be used to limit a theoretically unbounded loop. This caps the runtime
      16                 :            :  * to avoid timeouts or OOMs.
      17                 :            :  */
      18                 :            : #define LIMITED_WHILE(condition, limit) \
      19                 :            :     for (unsigned _count{limit}; (condition) && _count; --_count)
      20                 :            : 
      21                 :            : using FuzzBufferType = Span<const uint8_t>;
      22                 :            : 
      23                 :            : using TypeTestOneInput = std::function<void(FuzzBufferType)>;
      24                 :            : struct FuzzTargetOptions {
      25                 :          0 :     std::function<void()> init{[] {}};
      26                 :            :     bool hidden{false};
      27                 :            : };
      28                 :            : 
      29                 :            : void FuzzFrameworkRegisterTarget(std::string_view name, TypeTestOneInput target, FuzzTargetOptions opts);
      30                 :            : 
      31                 :            : #if defined(__clang__)
      32                 :            : #define FUZZ_TARGET(...) _Pragma("clang diagnostic push") _Pragma("clang diagnostic ignored \"-Wgnu-zero-variadic-macro-arguments\"") DETAIL_FUZZ(__VA_ARGS__) _Pragma("clang diagnostic pop")
      33                 :            : #else
      34                 :            : #define FUZZ_TARGET(...) DETAIL_FUZZ(__VA_ARGS__)
      35                 :            : #endif
      36                 :            : 
      37                 :            : #define DETAIL_FUZZ(name, ...)                                                        \
      38                 :            :     void name##_fuzz_target(FuzzBufferType);                                          \
      39                 :            :     struct name##_Before_Main {                                                       \
      40                 :            :         name##_Before_Main()                                                          \
      41                 :            :         {                                                                             \
      42                 :            :             FuzzFrameworkRegisterTarget(#name, name##_fuzz_target, {__VA_ARGS__});    \
      43                 :            :         }                                                                             \
      44                 :            :     } const static g_##name##_before_main;                                            \
      45                 :            :     void name##_fuzz_target(FuzzBufferType buffer)
      46                 :            : 
      47                 :            : #endif // BITCOIN_TEST_FUZZ_FUZZ_H