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 
1402d32aecSArgyrios Kyrtzidis #include "clang/Frontend/Utils.h"
1502d32aecSArgyrios Kyrtzidis #include "clang/Frontend/CompilerInstance.h"
1602d32aecSArgyrios Kyrtzidis #include "clang/Frontend/DiagnosticOptions.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,
339c902b55SDavid Blaikie                             llvm::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.
3702d32aecSArgyrios Kyrtzidis     DiagnosticOptions DiagOpts;
3802d32aecSArgyrios Kyrtzidis     Diags = CompilerInstance::createDiagnostics(DiagOpts, 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 
4602d32aecSArgyrios Kyrtzidis   // FIXME: Find a cleaner way to force the driver into restricted modes. We
4702d32aecSArgyrios Kyrtzidis   // also want to force it to use clang.
4802d32aecSArgyrios Kyrtzidis   Args.push_back("-fsyntax-only");
4902d32aecSArgyrios Kyrtzidis 
5002d32aecSArgyrios Kyrtzidis   // FIXME: We shouldn't have to pass in the path info.
51*8188c8a1SSebastian Pop   driver::Driver TheDriver("clang", llvm::sys::getDefaultTargetTriple(),
5205de0298SBob Wilson                            "a.out", false, *Diags);
5302d32aecSArgyrios Kyrtzidis 
5402d32aecSArgyrios Kyrtzidis   // Don't check that inputs exist, they may have been remapped.
5502d32aecSArgyrios Kyrtzidis   TheDriver.setCheckInputsExist(false);
5602d32aecSArgyrios Kyrtzidis 
5702d32aecSArgyrios Kyrtzidis   llvm::OwningPtr<driver::Compilation> C(TheDriver.BuildCompilation(Args));
5802d32aecSArgyrios Kyrtzidis 
5902d32aecSArgyrios Kyrtzidis   // Just print the cc1 options if -### was present.
6002d32aecSArgyrios Kyrtzidis   if (C->getArgs().hasArg(driver::options::OPT__HASH_HASH_HASH)) {
6102d32aecSArgyrios Kyrtzidis     C->PrintJob(llvm::errs(), C->getJobs(), "\n", true);
6202d32aecSArgyrios Kyrtzidis     return 0;
6302d32aecSArgyrios Kyrtzidis   }
6402d32aecSArgyrios Kyrtzidis 
6502d32aecSArgyrios Kyrtzidis   // We expect to get back exactly one command job, if we didn't something
6602d32aecSArgyrios Kyrtzidis   // failed.
6702d32aecSArgyrios Kyrtzidis   const driver::JobList &Jobs = C->getJobs();
68e08bca9eSEli Friedman   if (Jobs.size() != 1 || !isa<driver::Command>(*Jobs.begin())) {
6902d32aecSArgyrios Kyrtzidis     llvm::SmallString<256> Msg;
7002d32aecSArgyrios Kyrtzidis     llvm::raw_svector_ostream OS(Msg);
7102d32aecSArgyrios Kyrtzidis     C->PrintJob(OS, C->getJobs(), "; ", true);
7202d32aecSArgyrios Kyrtzidis     Diags->Report(diag::err_fe_expected_compiler_job) << OS.str();
7302d32aecSArgyrios Kyrtzidis     return 0;
7402d32aecSArgyrios Kyrtzidis   }
7502d32aecSArgyrios Kyrtzidis 
7602d32aecSArgyrios Kyrtzidis   const driver::Command *Cmd = cast<driver::Command>(*Jobs.begin());
770e62c1ccSChris Lattner   if (StringRef(Cmd->getCreator().getName()) != "clang") {
7802d32aecSArgyrios Kyrtzidis     Diags->Report(diag::err_fe_expected_clang_command);
7902d32aecSArgyrios Kyrtzidis     return 0;
8002d32aecSArgyrios Kyrtzidis   }
8102d32aecSArgyrios Kyrtzidis 
8202d32aecSArgyrios Kyrtzidis   const driver::ArgStringList &CCArgs = Cmd->getArguments();
8302d32aecSArgyrios Kyrtzidis   CompilerInvocation *CI = new CompilerInvocation();
8402d32aecSArgyrios Kyrtzidis   CompilerInvocation::CreateFromArgs(*CI,
8502d32aecSArgyrios Kyrtzidis                                      const_cast<const char **>(CCArgs.data()),
8602d32aecSArgyrios Kyrtzidis                                      const_cast<const char **>(CCArgs.data()) +
8702d32aecSArgyrios Kyrtzidis                                      CCArgs.size(),
8802d32aecSArgyrios Kyrtzidis                                      *Diags);
8902d32aecSArgyrios Kyrtzidis   return CI;
9002d32aecSArgyrios Kyrtzidis }
91