1 //===--- CreateInvocationFromCommandLine.cpp - CompilerInvocation from Args ==//
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 // Construct a compiler invocation object for command line driver arguments
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "clang/Frontend/Utils.h"
15 #include "clang/Basic/DiagnosticOptions.h"
16 #include "clang/Driver/Compilation.h"
17 #include "clang/Driver/Driver.h"
18 #include "clang/Driver/Action.h"
19 #include "clang/Driver/Options.h"
20 #include "clang/Driver/Tool.h"
21 #include "clang/Frontend/CompilerInstance.h"
22 #include "clang/Frontend/FrontendDiagnostic.h"
23 #include "llvm/Option/ArgList.h"
24 #include "llvm/Support/Host.h"
25 using namespace clang;
26 using namespace llvm::opt;
27
28 /// createInvocationFromCommandLine - Construct a compiler invocation object for
29 /// a command line argument vector.
30 ///
31 /// \return A CompilerInvocation, or 0 if none was built for the given
32 /// argument vector.
33 CompilerInvocation *
createInvocationFromCommandLine(ArrayRef<const char * > ArgList,IntrusiveRefCntPtr<DiagnosticsEngine> Diags)34 clang::createInvocationFromCommandLine(ArrayRef<const char *> ArgList,
35 IntrusiveRefCntPtr<DiagnosticsEngine> Diags) {
36 if (!Diags.get()) {
37 // No diagnostics engine was provided, so create our own diagnostics object
38 // with the default options.
39 Diags = CompilerInstance::createDiagnostics(new DiagnosticOptions);
40 }
41
42 SmallVector<const char *, 16> Args(ArgList.begin(), ArgList.end());
43
44 // FIXME: Find a cleaner way to force the driver into restricted modes.
45 Args.push_back("-fsyntax-only");
46
47 // FIXME: We shouldn't have to pass in the path info.
48 driver::Driver TheDriver(Args[0], llvm::sys::getDefaultTargetTriple(),
49 *Diags);
50
51 // Don't check that inputs exist, they may have been remapped.
52 TheDriver.setCheckInputsExist(false);
53
54 std::unique_ptr<driver::Compilation> C(TheDriver.BuildCompilation(Args));
55
56 // Just print the cc1 options if -### was present.
57 if (C->getArgs().hasArg(driver::options::OPT__HASH_HASH_HASH)) {
58 C->getJobs().Print(llvm::errs(), "\n", true);
59 return nullptr;
60 }
61
62 // We expect to get back exactly one command job, if we didn't something
63 // failed. CUDA compilation is an exception as it creates multiple jobs. If
64 // that's the case, we proceed with the first job. If caller needs particular
65 // CUDA job, it should be controlled via --cuda-{host|device}-only option
66 // passed to the driver.
67 const driver::JobList &Jobs = C->getJobs();
68 bool CudaCompilation = false;
69 if (Jobs.size() > 1) {
70 for (auto &A : C->getActions()){
71 // On MacOSX real actions may end up being wrapped in BindArchAction
72 if (isa<driver::BindArchAction>(A))
73 A = *A->begin();
74 if (isa<driver::CudaDeviceAction>(A)) {
75 CudaCompilation = true;
76 break;
77 }
78 }
79 }
80 if (Jobs.size() == 0 || !isa<driver::Command>(*Jobs.begin()) ||
81 (Jobs.size() > 1 && !CudaCompilation)) {
82 SmallString<256> Msg;
83 llvm::raw_svector_ostream OS(Msg);
84 Jobs.Print(OS, "; ", true);
85 Diags->Report(diag::err_fe_expected_compiler_job) << OS.str();
86 return nullptr;
87 }
88
89 const driver::Command &Cmd = cast<driver::Command>(*Jobs.begin());
90 if (StringRef(Cmd.getCreator().getName()) != "clang") {
91 Diags->Report(diag::err_fe_expected_clang_command);
92 return nullptr;
93 }
94
95 const ArgStringList &CCArgs = Cmd.getArguments();
96 std::unique_ptr<CompilerInvocation> CI(new CompilerInvocation());
97 if (!CompilerInvocation::CreateFromArgs(*CI,
98 const_cast<const char **>(CCArgs.data()),
99 const_cast<const char **>(CCArgs.data()) +
100 CCArgs.size(),
101 *Diags))
102 return nullptr;
103 return CI.release();
104 }
105