forked from OSchip/llvm-project
57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
//===- FuzzerDataFlowTrace.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.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
// fuzzer::DataFlowTrace; reads and handles a data-flow trace.
|
|
//
|
|
// A data flow trace is generated by e.g. dataflow/DataFlow.cpp
|
|
// and is stored on disk in a separate directory.
|
|
//
|
|
// The trace dir contains a file 'functions.txt' which lists function names,
|
|
// oner per line, e.g.
|
|
// ==> functions.txt <==
|
|
// Func2
|
|
// LLVMFuzzerTestOneInput
|
|
// Func1
|
|
//
|
|
// All other files in the dir are the traces, see dataflow/DataFlow.cpp.
|
|
// The name of the file is sha1 of the input used to generate the trace.
|
|
//
|
|
// Current status:
|
|
// the data is parsed and the summary is printed, but the data is not yet
|
|
// used in any other way.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_FUZZER_DATA_FLOW_TRACE
|
|
#define LLVM_FUZZER_DATA_FLOW_TRACE
|
|
|
|
#include "FuzzerDefs.h"
|
|
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
namespace fuzzer {
|
|
class DataFlowTrace {
|
|
public:
|
|
void Init(const std::string &DirPath, const std::string &FocusFunction);
|
|
void Clear() { Traces.clear(); }
|
|
const Vector<uint8_t> *Get(const std::string &InputSha1) const {
|
|
auto It = Traces.find(InputSha1);
|
|
if (It != Traces.end())
|
|
return &It->second;
|
|
return nullptr;
|
|
}
|
|
|
|
private:
|
|
// Input's sha1 => DFT for the FocusFunction.
|
|
std::unordered_map<std::string, Vector<uint8_t> > Traces;
|
|
};
|
|
} // namespace fuzzer
|
|
|
|
#endif // LLVM_FUZZER_DATA_FLOW_TRACE
|