| //===- 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 |
