summaryrefslogtreecommitdiffstats
path: root/llvm/lib/Fuzzer/FuzzerInterface.cpp
blob: 149ec66e5d5f70a1f6c07508a856a6c8ea8664bf (plain)
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
//===- FuzzerInterface.cpp - Mutate a test input --------------------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
// Parts of public interface for libFuzzer.
//===----------------------------------------------------------------------===//


#include "FuzzerInterface.h"
#include "FuzzerInternal.h"
#include <random>

namespace fuzzer {

void FuzzerRandomLibc::ResetSeed(unsigned int seed) { srand(seed); }

size_t FuzzerRandomLibc::Rand() { return rand(); }

struct FuzzerRandom_mt19937::Impl : public std::mt19937 {
  Impl(unsigned seed) : std::mt19937(seed) {}
};

void FuzzerRandom_mt19937::ResetSeed(unsigned int seed) {
  delete R;
  R = new Impl(seed);
}

FuzzerRandom_mt19937::~FuzzerRandom_mt19937() { delete R; }

size_t FuzzerRandom_mt19937::Rand() { return (*R)(); }

UserSuppliedFuzzer::UserSuppliedFuzzer(FuzzerRandomBase *Rand)
    : Rand(Rand), MD(new MutationDispatcher(*Rand)) {}

UserSuppliedFuzzer::~UserSuppliedFuzzer() {
  if (OwnRand)
    delete Rand;
  delete MD;
}

size_t UserSuppliedFuzzer::Mutate(uint8_t *Data, size_t Size, size_t MaxSize) {
  return GetMD().Mutate(Data, Size, MaxSize);
}

size_t UserSuppliedFuzzer::CrossOver(const uint8_t *Data1, size_t Size1,
                                     const uint8_t *Data2, size_t Size2,
                                     uint8_t *Out, size_t MaxOutSize) {
  return GetMD().CrossOver(Data1, Size1, Data2, Size2, Out, MaxOutSize);
}

size_t Mutate(uint8_t *Data, size_t Size, size_t MaxSize,
              FuzzerRandomBase &Rand) {
  MutationDispatcher MD(Rand);
  return MD.Mutate(Data, Size, MaxSize);
}



}  // namespace fuzzer.
OpenPOWER on IntegriCloud