mirror of
https://github.com/nlohmann/json.git
synced 2024-08-25 14:27:02 +02:00
183 lines
5.4 KiB
C++
183 lines
5.4 KiB
C++
//===- FuzzerInternal.h - Internal header for the Fuzzer --------*- C++ -* ===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
// Define the main class fuzzer::Fuzzer and most functions.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_FUZZER_INTERNAL_H
|
|
#define LLVM_FUZZER_INTERNAL_H
|
|
|
|
#include "FuzzerDefs.h"
|
|
#include "FuzzerExtFunctions.h"
|
|
#include "FuzzerInterface.h"
|
|
#include "FuzzerOptions.h"
|
|
#include "FuzzerSHA1.h"
|
|
#include "FuzzerValueBitMap.h"
|
|
#include <algorithm>
|
|
#include <atomic>
|
|
#include <chrono>
|
|
#include <climits>
|
|
#include <cstdlib>
|
|
#include <string.h>
|
|
|
|
namespace fuzzer {
|
|
|
|
using namespace std::chrono;
|
|
|
|
class Fuzzer {
|
|
public:
|
|
|
|
// Aggregates all available coverage measurements.
|
|
struct Coverage {
|
|
Coverage() { Reset(); }
|
|
|
|
void Reset() {
|
|
BlockCoverage = 0;
|
|
CallerCalleeCoverage = 0;
|
|
CounterBitmapBits = 0;
|
|
CounterBitmap.clear();
|
|
VPMap.Reset();
|
|
}
|
|
|
|
size_t BlockCoverage;
|
|
size_t CallerCalleeCoverage;
|
|
// Precalculated number of bits in CounterBitmap.
|
|
size_t CounterBitmapBits;
|
|
std::vector<uint8_t> CounterBitmap;
|
|
ValueBitMap VPMap;
|
|
};
|
|
|
|
Fuzzer(UserCallback CB, InputCorpus &Corpus, MutationDispatcher &MD,
|
|
FuzzingOptions Options);
|
|
~Fuzzer();
|
|
void Loop();
|
|
void MinimizeCrashLoop(const Unit &U);
|
|
void ShuffleAndMinimize(UnitVector *V);
|
|
void InitializeTraceState();
|
|
void RereadOutputCorpus(size_t MaxSize);
|
|
|
|
size_t secondsSinceProcessStartUp() {
|
|
return duration_cast<seconds>(system_clock::now() - ProcessStartTime)
|
|
.count();
|
|
}
|
|
|
|
bool TimedOut() {
|
|
return Options.MaxTotalTimeSec > 0 &&
|
|
secondsSinceProcessStartUp() >
|
|
static_cast<size_t>(Options.MaxTotalTimeSec);
|
|
}
|
|
|
|
size_t execPerSec() {
|
|
size_t Seconds = secondsSinceProcessStartUp();
|
|
return Seconds ? TotalNumberOfRuns / Seconds : 0;
|
|
}
|
|
|
|
size_t getTotalNumberOfRuns() { return TotalNumberOfRuns; }
|
|
|
|
static void StaticAlarmCallback();
|
|
static void StaticCrashSignalCallback();
|
|
static void StaticInterruptCallback();
|
|
|
|
void ExecuteCallback(const uint8_t *Data, size_t Size);
|
|
size_t RunOne(const uint8_t *Data, size_t Size);
|
|
|
|
// Merge Corpora[1:] into Corpora[0].
|
|
void Merge(const std::vector<std::string> &Corpora);
|
|
void CrashResistantMerge(const std::vector<std::string> &Args,
|
|
const std::vector<std::string> &Corpora);
|
|
void CrashResistantMergeInternalStep(const std::string &ControlFilePath);
|
|
// Returns a subset of 'Extra' that adds coverage to 'Initial'.
|
|
UnitVector FindExtraUnits(const UnitVector &Initial, const UnitVector &Extra);
|
|
MutationDispatcher &GetMD() { return MD; }
|
|
void PrintFinalStats();
|
|
void SetMaxInputLen(size_t MaxInputLen);
|
|
void SetMaxMutationLen(size_t MaxMutationLen);
|
|
void RssLimitCallback();
|
|
|
|
// Public for tests.
|
|
void ResetCoverage();
|
|
|
|
bool InFuzzingThread() const { return IsMyThread; }
|
|
size_t GetCurrentUnitInFuzzingThead(const uint8_t **Data) const;
|
|
void TryDetectingAMemoryLeak(const uint8_t *Data, size_t Size,
|
|
bool DuringInitialCorpusExecution);
|
|
|
|
void HandleMalloc(size_t Size);
|
|
|
|
private:
|
|
void AlarmCallback();
|
|
void CrashCallback();
|
|
void InterruptCallback();
|
|
void MutateAndTestOne();
|
|
void ReportNewCoverage(InputInfo *II, const Unit &U);
|
|
size_t RunOne(const Unit &U) { return RunOne(U.data(), U.size()); }
|
|
void WriteToOutputCorpus(const Unit &U);
|
|
void WriteUnitToFileWithPrefix(const Unit &U, const char *Prefix);
|
|
void PrintStats(const char *Where, const char *End = "\n", size_t Units = 0);
|
|
void PrintStatusForNewUnit(const Unit &U);
|
|
void ShuffleCorpus(UnitVector *V);
|
|
void AddToCorpus(const Unit &U);
|
|
void CheckExitOnSrcPosOrItem();
|
|
|
|
// Trace-based fuzzing: we run a unit with some kind of tracing
|
|
// enabled and record potentially useful mutations. Then
|
|
// We apply these mutations one by one to the unit and run it again.
|
|
|
|
// Start tracing; forget all previously proposed mutations.
|
|
void StartTraceRecording();
|
|
// Stop tracing.
|
|
void StopTraceRecording();
|
|
|
|
void SetDeathCallback();
|
|
static void StaticDeathCallback();
|
|
void DumpCurrentUnit(const char *Prefix);
|
|
void DeathCallback();
|
|
|
|
void ResetEdgeCoverage();
|
|
void ResetCounters();
|
|
void PrepareCounters(Fuzzer::Coverage *C);
|
|
bool RecordMaxCoverage(Fuzzer::Coverage *C);
|
|
|
|
void AllocateCurrentUnitData();
|
|
uint8_t *CurrentUnitData = nullptr;
|
|
std::atomic<size_t> CurrentUnitSize;
|
|
uint8_t BaseSha1[kSHA1NumBytes]; // Checksum of the base unit.
|
|
bool RunningCB = false;
|
|
|
|
size_t TotalNumberOfRuns = 0;
|
|
size_t NumberOfNewUnitsAdded = 0;
|
|
|
|
bool HasMoreMallocsThanFrees = false;
|
|
size_t NumberOfLeakDetectionAttempts = 0;
|
|
|
|
UserCallback CB;
|
|
InputCorpus &Corpus;
|
|
MutationDispatcher &MD;
|
|
FuzzingOptions Options;
|
|
|
|
system_clock::time_point ProcessStartTime = system_clock::now();
|
|
system_clock::time_point UnitStartTime, UnitStopTime;
|
|
long TimeOfLongestUnitInSeconds = 0;
|
|
long EpochOfLastReadOfOutputCorpus = 0;
|
|
|
|
// Maximum recorded coverage.
|
|
Coverage MaxCoverage;
|
|
|
|
size_t MaxInputLen = 0;
|
|
size_t MaxMutationLen = 0;
|
|
|
|
// Need to know our own thread.
|
|
static thread_local bool IsMyThread;
|
|
|
|
bool InMergeMode = false;
|
|
};
|
|
|
|
}; // namespace fuzzer
|
|
|
|
#endif // LLVM_FUZZER_INTERNAL_H
|