1 //===---- ClangQuery.cpp - clang-query tool -------------------------------===// 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 // 10 // This tool is for interactive exploration of the Clang AST using AST matchers. 11 // It currently allows the user to enter a matcher at an interactive prompt and 12 // view the resulting bindings as diagnostics, AST pretty prints or AST dumps. 13 // Example session: 14 // 15 // $ cat foo.c 16 // void foo(void) {} 17 // $ clang-query foo.c -- 18 // clang-query> match functionDecl() 19 // 20 // Match #1: 21 // 22 // foo.c:1:1: note: "root" binds here 23 // void foo(void) {} 24 // ^~~~~~~~~~~~~~~~~ 25 // 1 match. 26 // 27 //===----------------------------------------------------------------------===// 28 29 #include "Query.h" 30 #include "QueryParser.h" 31 #include "QuerySession.h" 32 #include "clang/Frontend/ASTUnit.h" 33 #include "clang/Tooling/CommonOptionsParser.h" 34 #include "clang/Tooling/Tooling.h" 35 #include "llvm/LineEditor/LineEditor.h" 36 #include "llvm/Support/CommandLine.h" 37 #include "llvm/Support/MemoryBuffer.h" 38 #include "llvm/Support/Signals.h" 39 #include <fstream> 40 #include <string> 41 42 using namespace clang; 43 using namespace clang::ast_matchers; 44 using namespace clang::ast_matchers::dynamic; 45 using namespace clang::query; 46 using namespace clang::tooling; 47 using namespace llvm; 48 49 static cl::extrahelp CommonHelp(CommonOptionsParser::HelpMessage); 50 static cl::OptionCategory ClangQueryCategory("clang-query options"); 51 52 static cl::list<std::string> Commands("c", cl::desc("Specify command to run"), 53 cl::value_desc("command"), 54 cl::cat(ClangQueryCategory)); 55 56 static cl::list<std::string> CommandFiles("f", 57 cl::desc("Read commands from file"), 58 cl::value_desc("file"), 59 cl::cat(ClangQueryCategory)); 60 61 static cl::opt<std::string> PreloadFile( 62 "preload", 63 cl::desc("Preload commands from file and start interactive mode"), 64 cl::value_desc("file"), cl::cat(ClangQueryCategory)); 65 66 bool runCommandsInFile(const char *ExeName, std::string const &FileName, 67 QuerySession &QS) { 68 std::ifstream Input(FileName.c_str()); 69 if (!Input.is_open()) { 70 llvm::errs() << ExeName << ": cannot open " << FileName << "\n"; 71 return 1; 72 } 73 while (Input.good()) { 74 std::string Line; 75 std::getline(Input, Line); 76 77 QueryRef Q = QueryParser::parse(Line, QS); 78 if (!Q->run(llvm::outs(), QS)) 79 return true; 80 } 81 return false; 82 } 83 84 int main(int argc, const char **argv) { 85 llvm::sys::PrintStackTraceOnErrorSignal(argv[0]); 86 87 CommonOptionsParser OptionsParser(argc, argv, ClangQueryCategory); 88 89 if (!Commands.empty() && !CommandFiles.empty()) { 90 llvm::errs() << argv[0] << ": cannot specify both -c and -f\n"; 91 return 1; 92 } 93 94 if ((!Commands.empty() || !CommandFiles.empty()) && !PreloadFile.empty()) { 95 llvm::errs() << argv[0] 96 << ": cannot specify both -c or -f with --preload\n"; 97 return 1; 98 } 99 100 ClangTool Tool(OptionsParser.getCompilations(), 101 OptionsParser.getSourcePathList()); 102 std::vector<std::unique_ptr<ASTUnit>> ASTs; 103 if (Tool.buildASTs(ASTs) != 0) 104 return 1; 105 106 QuerySession QS(ASTs); 107 108 if (!Commands.empty()) { 109 for (auto I = Commands.begin(), E = Commands.end(); I != E; ++I) { 110 QueryRef Q = QueryParser::parse(*I, QS); 111 if (!Q->run(llvm::outs(), QS)) 112 return 1; 113 } 114 } else if (!CommandFiles.empty()) { 115 for (auto I = CommandFiles.begin(), E = CommandFiles.end(); I != E; ++I) { 116 if (runCommandsInFile(argv[0], *I, QS)) 117 return 1; 118 } 119 } else { 120 if (!PreloadFile.empty()) { 121 if (runCommandsInFile(argv[0], PreloadFile, QS)) 122 return 1; 123 } 124 LineEditor LE("clang-query"); 125 LE.setListCompleter([&QS](StringRef Line, size_t Pos) { 126 return QueryParser::complete(Line, Pos, QS); 127 }); 128 while (llvm::Optional<std::string> Line = LE.readLine()) { 129 QueryRef Q = QueryParser::parse(*Line, QS); 130 Q->run(llvm::outs(), QS); 131 llvm::outs().flush(); 132 if (QS.Terminate) 133 break; 134 } 135 } 136 137 return 0; 138 } 139