1//===-- tools/llvm-reduce/TestRunner.h ---------------------------*- C++ -*-===/
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#ifndef LLVM_TOOLS_LLVM_REDUCE_TESTRUNNER_H
10#define LLVM_TOOLS_LLVM_REDUCE_TESTRUNNER_H
11
12#include "ReducerWorkItem.h"
13#include "llvm/IR/Module.h"
14#include "llvm/Support/Error.h"
15#include "llvm/Support/FileSystem.h"
16#include "llvm/Support/Path.h"
17#include "llvm/Support/Program.h"
18#include "llvm/Target/TargetMachine.h"
19#include <vector>
20
21namespace llvm {
22
23// This class contains all the info necessary for running the provided
24// interesting-ness test, as well as the most reduced module and its
25// respective filename.
26class TestRunner {
27public:
28 TestRunner(StringRef TestName, ArrayRef<std::string> TestArgs,
29 std::unique_ptr<ReducerWorkItem> Program,
30 std::unique_ptr<TargetMachine> TM, StringRef ToolName,
31 StringRef OutputFilename, bool InputIsBitcode, bool OutputBitcode);
32
33 /// Runs the interesting-ness test for the specified file
34 /// @returns 0 if test was successful, 1 if otherwise
35 int run(StringRef Filename) const;
36
37 /// Returns the most reduced version of the original testcase
38 ReducerWorkItem &getProgram() const { return *Program; }
39
40 void setProgram(std::unique_ptr<ReducerWorkItem> &&P) {
41 assert(P && "Setting null program?");
42 Program = std::move(P);
43 }
44
45 const TargetMachine *getTargetMachine() const { return TM.get(); }
46
47 StringRef getToolName() const { return ToolName; }
48
49 void writeOutput(StringRef Message);
50
51 bool inputIsBitcode() const {
52 return InputIsBitcode;
53 }
54
55private:
56 StringRef TestName;
57 StringRef ToolName;
58 SmallVector<StringRef> TestArgs;
59 std::unique_ptr<ReducerWorkItem> Program;
60 std::unique_ptr<TargetMachine> TM;
61 StringRef OutputFilename;
62 const bool InputIsBitcode;
63 bool EmitBitcode;
64};
65
66} // namespace llvm
67
68#endif
69