1//===- FuzzerDataFlowTrace.h - Internal header for the Fuzzer ---*- C++ -* ===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9// fuzzer::DataFlowTrace; reads and handles a data-flow trace.
10//
11// A data flow trace is generated by e.g. dataflow/DataFlow.cpp
12// and is stored on disk in a separate directory.
13//
14// The trace dir contains a file 'functions.txt' which lists function names,
15// oner per line, e.g.
16// ==> functions.txt <==
17// Func2
18// LLVMFuzzerTestOneInput
19// Func1
20//
21// All other files in the dir are the traces, see dataflow/DataFlow.cpp.
22// The name of the file is sha1 of the input used to generate the trace.
23//
24// Current status:
25//   the data is parsed and the summary is printed, but the data is not yet
26//   used in any other way.
27//===----------------------------------------------------------------------===//
28
29#ifndef LLVM_FUZZER_DATA_FLOW_TRACE
30#define LLVM_FUZZER_DATA_FLOW_TRACE
31
32#include "FuzzerDefs.h"
33
34#include <unordered_map>
35#include <vector>
36#include <string>
37
38namespace fuzzer {
39class DataFlowTrace {
40 public:
41  void Init(const std::string &DirPath, const std::string &FocusFunction);
42  void Clear() { Traces.clear(); }
43  const Vector<uint8_t> *Get(const std::string &InputSha1) const {
44    auto It = Traces.find(InputSha1);
45    if (It != Traces.end())
46      return &It->second;
47    return nullptr;
48  }
49
50 private:
51  // Input's sha1 => DFT for the FocusFunction.
52  std::unordered_map<std::string, Vector<uint8_t> > Traces;
53};
54}  // namespace fuzzer
55
56#endif // LLVM_FUZZER_DATA_FLOW_TRACE
57