1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
//
// immer: immutable data structures for C++
// Copyright (C) 2016, 2017, 2018 Juan Pedro Bolivar Puente
//
// This software is distributed under the Boost Software License, Version 1.0.
// See accompanying file LICENSE or copy at http://boost.org/LICENSE_1_0.txt
//
#pragma once
#include <cstdint>
#include <memory>
#include <stdexcept>
#if defined(__GNUC__) && (__GNUC__ == 9 || __GNUC__ == 8 || __GNUC__ == 10)
#define IMMER_DISABLE_FUZZER_DUE_TO_GCC_BUG 1
#endif
struct no_more_input : std::exception
{};
constexpr auto fuzzer_input_max_size = 1 << 16;
struct fuzzer_input
{
const std::uint8_t* data_;
std::size_t size_;
const std::uint8_t* next(std::size_t size)
{
if (size_ < size)
throw no_more_input{};
auto r = data_;
data_ += size;
size_ -= size;
return r;
}
const std::uint8_t* next(std::size_t size, std::size_t align)
{
auto& p = const_cast<void*&>(reinterpret_cast<const void*&>(data_));
auto r = std::align(align, size, p, size_);
if (r == nullptr)
throw no_more_input{};
return next(size);
}
template <typename Fn>
int run(Fn step)
{
if (size_ > fuzzer_input_max_size)
return 0;
try {
while (step(*this))
continue;
} catch (const no_more_input&) {
};
return 0;
}
};
template <typename T>
const T& read(fuzzer_input& fz)
{
return *reinterpret_cast<const T*>(fz.next(sizeof(T), alignof(T)));
}
template <typename T, typename Cond>
T read(fuzzer_input& fz, Cond cond)
{
auto x = read<T>(fz);
while (!cond(x))
x = read<T>(fz);
return x;
}
|
