69 lines
2.4 KiB
C++
69 lines
2.4 KiB
C++
// Declares clang::SyntaxOnlyAction.
|
|
#include "clang/Frontend/FrontendActions.h"
|
|
#include "clang/Tooling/CommonOptionsParser.h"
|
|
#include "clang/Tooling/Tooling.h"
|
|
// Declares llvm::cl::extrahelp.
|
|
#include "llvm/Support/CommandLine.h"
|
|
|
|
#include "OperationAstMatcher.hpp"
|
|
#include "OperationFinder.hpp"
|
|
#include "OperationStorage.hpp"
|
|
#include "OperationFinderAstAction.hpp"
|
|
|
|
using namespace clang::tooling;
|
|
using namespace clang::ast_matchers;
|
|
using namespace llvm;
|
|
|
|
// Apply a custom category to all command-line options so that they are the
|
|
// only ones displayed.
|
|
static cl::OptionCategory MyToolCategory("op-finder options");
|
|
|
|
static cl::opt<std::string> OutputFile("o", cl::desc("File to output the JSON to."),
|
|
cl::cat(MyToolCategory));
|
|
|
|
static cl::opt<std::string> RootDirectory("r", cl::desc("The root directory of the source files."),
|
|
cl::cat(MyToolCategory));
|
|
|
|
// CommonOptionsParser declares HelpMessage with a description of the common
|
|
// command-line options related to the compilation database and input files.
|
|
// It's nice to have this help message in all tools.
|
|
static cl::extrahelp CommonHelp(CommonOptionsParser::HelpMessage);
|
|
|
|
// A help message for this specific tool can be added afterwards.
|
|
static cl::extrahelp MoreHelp("\nThe program takes the input <source0> ... files, parses their\n"
|
|
"AST and outputs a singular file containing a list of all noteworthy operations\n"
|
|
"for later analysis.\n");
|
|
|
|
int main(int argc, const char** argv)
|
|
{
|
|
auto ExpectedParser = CommonOptionsParser::create(argc, argv, MyToolCategory);
|
|
if (!ExpectedParser)
|
|
{
|
|
// Fail gracefully for unsupported options.
|
|
llvm::errs() << ExpectedParser.takeError();
|
|
return 1;
|
|
}
|
|
|
|
CommonOptionsParser& OptionsParser = ExpectedParser.get();
|
|
ClangTool Tool(OptionsParser.getCompilations(),
|
|
OptionsParser.getSourcePathList());
|
|
|
|
OperationStorage storage(OutputFile.getValue());
|
|
|
|
OperationFinder op_finder(&storage);
|
|
|
|
#if 0
|
|
MatchFinder matcher;
|
|
OperationASTMatcher finder(&op_finder);
|
|
|
|
finder.addToFinder(matcher);
|
|
|
|
//Tool.run(newFrontendActionFactory<DebuggeringASTAction>().get());
|
|
|
|
return Tool.run(newFrontendActionFactory(&finder).get());
|
|
#endif
|
|
|
|
OperationFinderAstAction action(&op_finder);
|
|
|
|
return Tool.run(newFrontendActionFactory(&action).get());
|
|
} |