1 //===--- CompilationDatabase.cpp - ----------------------------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file contains implementations of the CompilationDatabase base class
11 // and the FixedCompilationDatabase.
13 //===----------------------------------------------------------------------===//
15 #include "clang/Tooling/CompilationDatabase.h"
16 #include "clang/Tooling/CompilationDatabasePluginRegistry.h"
17 #include "clang/Tooling/Tooling.h"
18 #include "llvm/ADT/SmallString.h"
19 #include "llvm/Support/Path.h"
20 #include "llvm/Support/system_error.h"
23 #include "clang/Basic/Diagnostic.h"
24 #include "clang/Driver/Action.h"
25 #include "clang/Driver/Driver.h"
26 #include "clang/Driver/DriverDiagnostic.h"
27 #include "clang/Driver/Job.h"
28 #include "clang/Driver/Compilation.h"
29 #include "clang/Frontend/TextDiagnosticPrinter.h"
30 #include "llvm/Support/Host.h"
31 #include "llvm/Option/Arg.h"
36 CompilationDatabase::~CompilationDatabase() {}
39 CompilationDatabase::loadFromDirectory(StringRef BuildDirectory,
40 std::string &ErrorMessage) {
41 std::stringstream ErrorStream;
42 for (CompilationDatabasePluginRegistry::iterator
43 It = CompilationDatabasePluginRegistry::begin(),
44 Ie = CompilationDatabasePluginRegistry::end();
46 std::string DatabaseErrorMessage;
47 OwningPtr<CompilationDatabasePlugin> Plugin(It->instantiate());
48 if (CompilationDatabase *DB =
49 Plugin->loadFromDirectory(BuildDirectory, DatabaseErrorMessage))
52 ErrorStream << It->getName() << ": " << DatabaseErrorMessage << "\n";
54 ErrorMessage = ErrorStream.str();
58 static CompilationDatabase *
59 findCompilationDatabaseFromDirectory(StringRef Directory,
60 std::string &ErrorMessage) {
61 std::stringstream ErrorStream;
62 bool HasErrorMessage = false;
63 while (!Directory.empty()) {
64 std::string LoadErrorMessage;
66 if (CompilationDatabase *DB =
67 CompilationDatabase::loadFromDirectory(Directory, LoadErrorMessage))
70 if (!HasErrorMessage) {
71 ErrorStream << "No compilation database found in " << Directory.str()
72 << " or any parent directory\n" << LoadErrorMessage;
73 HasErrorMessage = true;
76 Directory = llvm::sys::path::parent_path(Directory);
78 ErrorMessage = ErrorStream.str();
83 CompilationDatabase::autoDetectFromSource(StringRef SourceFile,
84 std::string &ErrorMessage) {
85 SmallString<1024> AbsolutePath(getAbsolutePath(SourceFile));
86 StringRef Directory = llvm::sys::path::parent_path(AbsolutePath);
88 CompilationDatabase *DB = findCompilationDatabaseFromDirectory(Directory,
92 ErrorMessage = ("Could not auto-detect compilation database for file \"" +
93 SourceFile + "\"\n" + ErrorMessage).str();
98 CompilationDatabase::autoDetectFromDirectory(StringRef SourceDir,
99 std::string &ErrorMessage) {
100 SmallString<1024> AbsolutePath(getAbsolutePath(SourceDir));
102 CompilationDatabase *DB = findCompilationDatabaseFromDirectory(AbsolutePath,
106 ErrorMessage = ("Could not auto-detect compilation database from directory \"" +
107 SourceDir + "\"\n" + ErrorMessage).str();
111 CompilationDatabasePlugin::~CompilationDatabasePlugin() {}
113 // Helper for recursively searching through a chain of actions and collecting
114 // all inputs, direct and indirect, of compile jobs.
115 struct CompileJobAnalyzer {
116 void run(const driver::Action *A) {
120 SmallVector<std::string, 2> Inputs;
124 void runImpl(const driver::Action *A, bool Collect) {
125 bool CollectChildren = Collect;
126 switch (A->getKind()) {
127 case driver::Action::CompileJobClass:
128 CollectChildren = true;
131 case driver::Action::InputClass: {
133 const driver::InputAction *IA = cast<driver::InputAction>(A);
134 Inputs.push_back(IA->getInputArg().getSpelling());
139 // Don't care about others
143 for (driver::ActionList::const_iterator I = A->begin(), E = A->end();
145 runImpl(*I, CollectChildren);
149 // Special DiagnosticConsumer that looks for warn_drv_input_file_unused
150 // diagnostics from the driver and collects the option strings for those unused
152 class UnusedInputDiagConsumer : public DiagnosticConsumer {
154 UnusedInputDiagConsumer() : Other(0) {}
156 // Useful for debugging, chain diagnostics to another consumer after
157 // recording for our own purposes.
158 UnusedInputDiagConsumer(DiagnosticConsumer *Other) : Other(Other) {}
160 virtual void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
161 const Diagnostic &Info) LLVM_OVERRIDE {
162 if (Info.getID() == clang::diag::warn_drv_input_file_unused) {
163 // Arg 1 for this diagnostic is the option that didn't get used.
164 UnusedInputs.push_back(Info.getArgStdStr(0));
167 Other->HandleDiagnostic(DiagLevel, Info);
170 DiagnosticConsumer *Other;
171 SmallVector<std::string, 2> UnusedInputs;
174 // Unary functor for asking "Given a StringRef S1, does there exist a string
175 // S2 in Arr where S1 == S2?"
177 MatchesAny(ArrayRef<std::string> Arr) : Arr(Arr) {}
178 bool operator() (StringRef S) {
179 for (const std::string *I = Arr.begin(), *E = Arr.end(); I != E; ++I)
185 ArrayRef<std::string> Arr;
188 /// \brief Strips any positional args and possible argv[0] from a command-line
189 /// provided by the user to construct a FixedCompilationDatabase.
191 /// FixedCompilationDatabase requires a command line to be in this format as it
192 /// constructs the command line for each file by appending the name of the file
193 /// to be compiled. FixedCompilationDatabase also adds its own argv[0] to the
194 /// start of the command line although its value is not important as it's just
195 /// ignored by the Driver invoked by the ClangTool using the
196 /// FixedCompilationDatabase.
198 /// FIXME: This functionality should probably be made available by
199 /// clang::driver::Driver although what the interface should look like is not
202 /// \param[in] Args Args as provided by the user.
203 /// \return Resulting stripped command line.
204 /// \li true if successful.
205 /// \li false if \c Args cannot be used for compilation jobs (e.g.
206 /// contains an option like -E or -version).
207 bool stripPositionalArgs(std::vector<const char *> Args,
208 std::vector<std::string> &Result) {
209 IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
210 UnusedInputDiagConsumer DiagClient;
211 DiagnosticsEngine Diagnostics(
212 IntrusiveRefCntPtr<clang::DiagnosticIDs>(new DiagnosticIDs()),
213 &*DiagOpts, &DiagClient, false);
215 // Neither clang executable nor default image name are required since the
216 // jobs the driver builds will not be executed.
217 OwningPtr<driver::Driver> NewDriver(new driver::Driver(
218 /* ClangExecutable= */ "", llvm::sys::getDefaultTargetTriple(),
219 /* DefaultImageName= */ "", Diagnostics));
220 NewDriver->setCheckInputsExist(false);
222 // This becomes the new argv[0]. The value is actually not important as it
223 // isn't used for invoking Tools.
224 Args.insert(Args.begin(), "clang-tool");
226 // By adding -c, we force the driver to treat compilation as the last phase.
227 // It will then issue warnings via Diagnostics about un-used options that
228 // would have been used for linking. If the user provided a compiler name as
229 // the original argv[0], this will be treated as a linker input thanks to
230 // insertng a new argv[0] above. All un-used options get collected by
231 // UnusedInputdiagConsumer and get stripped out later.
232 Args.push_back("-c");
234 // Put a dummy C++ file on to ensure there's at least one compile job for the
235 // driver to construct. If the user specified some other argument that
236 // prevents compilation, e.g. -E or something like -version, we may still end
237 // up with no jobs but then this is the user's fault.
238 Args.push_back("placeholder.cpp");
240 const OwningPtr<driver::Compilation> Compilation(
241 NewDriver->BuildCompilation(Args));
243 const driver::JobList &Jobs = Compilation->getJobs();
245 CompileJobAnalyzer CompileAnalyzer;
247 for (driver::JobList::const_iterator I = Jobs.begin(), E = Jobs.end(); I != E;
249 if ((*I)->getKind() == driver::Job::CommandClass) {
250 const driver::Command *Cmd = cast<driver::Command>(*I);
251 // Collect only for Assemble jobs. If we do all jobs we get duplicates
252 // since Link jobs point to Assemble jobs as inputs.
253 if (Cmd->getSource().getKind() == driver::Action::AssembleJobClass)
254 CompileAnalyzer.run(&Cmd->getSource());
258 if (CompileAnalyzer.Inputs.empty()) {
259 // No compile jobs found.
260 // FIXME: Emit a warning of some kind?
264 // Remove all compilation input files from the command line. This is
265 // necessary so that getCompileCommands() can construct a command line for
267 std::vector<const char *>::iterator End = std::remove_if(
268 Args.begin(), Args.end(), MatchesAny(CompileAnalyzer.Inputs));
270 // Remove all inputs deemed unused for compilation.
271 End = std::remove_if(Args.begin(), End, MatchesAny(DiagClient.UnusedInputs));
273 // Remove the -c add above as well. It will be at the end right now.
276 Result = std::vector<std::string>(Args.begin() + 1, End);
280 FixedCompilationDatabase *
281 FixedCompilationDatabase::loadFromCommandLine(int &Argc,
284 const char **DoubleDash = std::find(Argv, Argv + Argc, StringRef("--"));
285 if (DoubleDash == Argv + Argc)
287 std::vector<const char *> CommandLine(DoubleDash + 1, Argv + Argc);
288 Argc = DoubleDash - Argv;
290 std::vector<std::string> StrippedArgs;
291 if (!stripPositionalArgs(CommandLine, StrippedArgs))
293 return new FixedCompilationDatabase(Directory, StrippedArgs);
296 FixedCompilationDatabase::
297 FixedCompilationDatabase(Twine Directory, ArrayRef<std::string> CommandLine) {
298 std::vector<std::string> ToolCommandLine(1, "clang-tool");
299 ToolCommandLine.insert(ToolCommandLine.end(),
300 CommandLine.begin(), CommandLine.end());
301 CompileCommands.push_back(CompileCommand(Directory, ToolCommandLine));
304 std::vector<CompileCommand>
305 FixedCompilationDatabase::getCompileCommands(StringRef FilePath) const {
306 std::vector<CompileCommand> Result(CompileCommands);
307 Result[0].CommandLine.push_back(FilePath);
311 std::vector<std::string>
312 FixedCompilationDatabase::getAllFiles() const {
313 return std::vector<std::string>();
316 std::vector<CompileCommand>
317 FixedCompilationDatabase::getAllCompileCommands() const {
318 return std::vector<CompileCommand>();
321 // This anchor is used to force the linker to link in the generated object file
322 // and thus register the JSONCompilationDatabasePlugin.
323 extern volatile int JSONAnchorSource;
324 static int JSONAnchorDest = JSONAnchorSource;
326 } // end namespace tooling
327 } // end namespace clang