• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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->input_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