102d32aecSArgyrios Kyrtzidis //===--- CreateInvocationFromCommandLine.cpp - CompilerInvocation from Args ==//
202d32aecSArgyrios Kyrtzidis //
302d32aecSArgyrios Kyrtzidis //                     The LLVM Compiler Infrastructure
402d32aecSArgyrios Kyrtzidis //
502d32aecSArgyrios Kyrtzidis // This file is distributed under the University of Illinois Open Source
602d32aecSArgyrios Kyrtzidis // License. See LICENSE.TXT for details.
702d32aecSArgyrios Kyrtzidis //
802d32aecSArgyrios Kyrtzidis //===----------------------------------------------------------------------===//
902d32aecSArgyrios Kyrtzidis //
1002d32aecSArgyrios Kyrtzidis // Construct a compiler invocation object for command line driver arguments
1102d32aecSArgyrios Kyrtzidis //
1202d32aecSArgyrios Kyrtzidis //===----------------------------------------------------------------------===//
1302d32aecSArgyrios Kyrtzidis 
14*811db4eaSDouglas Gregor #include "clang/Basic/DiagnosticOptions.h"
1502d32aecSArgyrios Kyrtzidis #include "clang/Frontend/Utils.h"
1602d32aecSArgyrios Kyrtzidis #include "clang/Frontend/CompilerInstance.h"
1702d32aecSArgyrios Kyrtzidis #include "clang/Frontend/FrontendDiagnostic.h"
1802d32aecSArgyrios Kyrtzidis #include "clang/Driver/Compilation.h"
1902d32aecSArgyrios Kyrtzidis #include "clang/Driver/Driver.h"
2002d32aecSArgyrios Kyrtzidis #include "clang/Driver/ArgList.h"
2102d32aecSArgyrios Kyrtzidis #include "clang/Driver/Options.h"
2202d32aecSArgyrios Kyrtzidis #include "clang/Driver/Tool.h"
2302d32aecSArgyrios Kyrtzidis #include "llvm/Support/Host.h"
2402d32aecSArgyrios Kyrtzidis using namespace clang;
2502d32aecSArgyrios Kyrtzidis 
2602d32aecSArgyrios Kyrtzidis /// createInvocationFromCommandLine - Construct a compiler invocation object for
2702d32aecSArgyrios Kyrtzidis /// a command line argument vector.
2802d32aecSArgyrios Kyrtzidis ///
2902d32aecSArgyrios Kyrtzidis /// \return A CompilerInvocation, or 0 if none was built for the given
3002d32aecSArgyrios Kyrtzidis /// argument vector.
3102d32aecSArgyrios Kyrtzidis CompilerInvocation *
3254b1677dSChris Lattner clang::createInvocationFromCommandLine(ArrayRef<const char *> ArgList,
33c95d8192SDylan Noblesmith                             IntrusiveRefCntPtr<DiagnosticsEngine> Diags) {
3402d32aecSArgyrios Kyrtzidis   if (!Diags.getPtr()) {
3502d32aecSArgyrios Kyrtzidis     // No diagnostics engine was provided, so create our own diagnostics object
3602d32aecSArgyrios Kyrtzidis     // with the default options.
37*811db4eaSDouglas Gregor     Diags = CompilerInstance::createDiagnostics(new DiagnosticOptions,
38*811db4eaSDouglas Gregor                                                 ArgList.size(),
3902d32aecSArgyrios Kyrtzidis                                                 ArgList.begin());
4002d32aecSArgyrios Kyrtzidis   }
4102d32aecSArgyrios Kyrtzidis 
420e62c1ccSChris Lattner   SmallVector<const char *, 16> Args;
4302d32aecSArgyrios Kyrtzidis   Args.push_back("<clang>"); // FIXME: Remove dummy argument.
4402d32aecSArgyrios Kyrtzidis   Args.insert(Args.end(), ArgList.begin(), ArgList.end());
4502d32aecSArgyrios Kyrtzidis 
463169e806SArgyrios Kyrtzidis   // FIXME: Find a cleaner way to force the driver into restricted modes.
4702d32aecSArgyrios Kyrtzidis   Args.push_back("-fsyntax-only");
4802d32aecSArgyrios Kyrtzidis 
4902d32aecSArgyrios Kyrtzidis   // FIXME: We shouldn't have to pass in the path info.
508188c8a1SSebastian Pop   driver::Driver TheDriver("clang", llvm::sys::getDefaultTargetTriple(),
5105de0298SBob Wilson                            "a.out", false, *Diags);
523169e806SArgyrios Kyrtzidis   // Force driver to use clang.
533169e806SArgyrios Kyrtzidis   // FIXME: This seems like a hack. Maybe the "Clang" tool subclass should be
543169e806SArgyrios Kyrtzidis   // available for using it to get the arguments, thus avoiding the overkill
553169e806SArgyrios Kyrtzidis   // of using the driver.
563169e806SArgyrios Kyrtzidis   TheDriver.setForcedClangUse();
5702d32aecSArgyrios Kyrtzidis 
5802d32aecSArgyrios Kyrtzidis   // Don't check that inputs exist, they may have been remapped.
5902d32aecSArgyrios Kyrtzidis   TheDriver.setCheckInputsExist(false);
6002d32aecSArgyrios Kyrtzidis 
61e2778999SDylan Noblesmith   OwningPtr<driver::Compilation> C(TheDriver.BuildCompilation(Args));
6202d32aecSArgyrios Kyrtzidis 
6302d32aecSArgyrios Kyrtzidis   // Just print the cc1 options if -### was present.
6402d32aecSArgyrios Kyrtzidis   if (C->getArgs().hasArg(driver::options::OPT__HASH_HASH_HASH)) {
6502d32aecSArgyrios Kyrtzidis     C->PrintJob(llvm::errs(), C->getJobs(), "\n", true);
6602d32aecSArgyrios Kyrtzidis     return 0;
6702d32aecSArgyrios Kyrtzidis   }
6802d32aecSArgyrios Kyrtzidis 
6902d32aecSArgyrios Kyrtzidis   // We expect to get back exactly one command job, if we didn't something
7002d32aecSArgyrios Kyrtzidis   // failed.
7102d32aecSArgyrios Kyrtzidis   const driver::JobList &Jobs = C->getJobs();
72e08bca9eSEli Friedman   if (Jobs.size() != 1 || !isa<driver::Command>(*Jobs.begin())) {
732c1dd271SDylan Noblesmith     SmallString<256> Msg;
7402d32aecSArgyrios Kyrtzidis     llvm::raw_svector_ostream OS(Msg);
7502d32aecSArgyrios Kyrtzidis     C->PrintJob(OS, C->getJobs(), "; ", true);
7602d32aecSArgyrios Kyrtzidis     Diags->Report(diag::err_fe_expected_compiler_job) << OS.str();
7702d32aecSArgyrios Kyrtzidis     return 0;
7802d32aecSArgyrios Kyrtzidis   }
7902d32aecSArgyrios Kyrtzidis 
8002d32aecSArgyrios Kyrtzidis   const driver::Command *Cmd = cast<driver::Command>(*Jobs.begin());
810e62c1ccSChris Lattner   if (StringRef(Cmd->getCreator().getName()) != "clang") {
8202d32aecSArgyrios Kyrtzidis     Diags->Report(diag::err_fe_expected_clang_command);
8302d32aecSArgyrios Kyrtzidis     return 0;
8402d32aecSArgyrios Kyrtzidis   }
8502d32aecSArgyrios Kyrtzidis 
8602d32aecSArgyrios Kyrtzidis   const driver::ArgStringList &CCArgs = Cmd->getArguments();
87e2778999SDylan Noblesmith   OwningPtr<CompilerInvocation> CI(new CompilerInvocation());
88e99b27f3SDylan Noblesmith   if (!CompilerInvocation::CreateFromArgs(*CI,
8902d32aecSArgyrios Kyrtzidis                                      const_cast<const char **>(CCArgs.data()),
9002d32aecSArgyrios Kyrtzidis                                      const_cast<const char **>(CCArgs.data()) +
9102d32aecSArgyrios Kyrtzidis                                      CCArgs.size(),
92e99b27f3SDylan Noblesmith                                      *Diags))
93e99b27f3SDylan Noblesmith     return 0;
94e99b27f3SDylan Noblesmith   return CI.take();
9502d32aecSArgyrios Kyrtzidis }
96