CreateInvocationFromCommandLine.cpp revision 5f9e272e632e951b1efe824cd16acb4d96077930
1ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis//===--- CreateInvocationFromCommandLine.cpp - CompilerInvocation from Args ==// 2ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// 3ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// The LLVM Compiler Infrastructure 4ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// 5ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// This file is distributed under the University of Illinois Open Source 6ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// License. See LICENSE.TXT for details. 7ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// 8ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis//===----------------------------------------------------------------------===// 9ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// 10ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// Construct a compiler invocation object for command line driver arguments 11ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis// 12ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis//===----------------------------------------------------------------------===// 13ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 14ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Frontend/Utils.h" 15ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Frontend/CompilerInstance.h" 16ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Frontend/DiagnosticOptions.h" 17ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Frontend/FrontendDiagnostic.h" 18ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Driver/Compilation.h" 19ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Driver/Driver.h" 20ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Driver/ArgList.h" 21ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Driver/Options.h" 22ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "clang/Driver/Tool.h" 23ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis#include "llvm/Support/Host.h" 24ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidisusing namespace clang; 25ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 26ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis/// createInvocationFromCommandLine - Construct a compiler invocation object for 27ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis/// a command line argument vector. 28ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis/// 29ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis/// \return A CompilerInvocation, or 0 if none was built for the given 30ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis/// argument vector. 31ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios KyrtzidisCompilerInvocation * 32ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidisclang::createInvocationFromCommandLine(llvm::ArrayRef<const char *> ArgList, 33ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis llvm::IntrusiveRefCntPtr<Diagnostic> Diags) { 34ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis if (!Diags.getPtr()) { 35ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // No diagnostics engine was provided, so create our own diagnostics object 36ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // with the default options. 37ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis DiagnosticOptions DiagOpts; 38ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis Diags = CompilerInstance::createDiagnostics(DiagOpts, ArgList.size(), 39ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis ArgList.begin()); 40ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis } 41ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 425f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner SmallVector<const char *, 16> Args; 43ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis Args.push_back("<clang>"); // FIXME: Remove dummy argument. 44ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis Args.insert(Args.end(), ArgList.begin(), ArgList.end()); 45ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 46ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // FIXME: Find a cleaner way to force the driver into restricted modes. We 47ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // also want to force it to use clang. 48ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis Args.push_back("-fsyntax-only"); 49ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 50ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // FIXME: We shouldn't have to pass in the path info. 51ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis driver::Driver TheDriver("clang", llvm::sys::getHostTriple(), 52ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis "a.out", false, false, *Diags); 53ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 54ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // Don't check that inputs exist, they may have been remapped. 55ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis TheDriver.setCheckInputsExist(false); 56ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 57ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis llvm::OwningPtr<driver::Compilation> C(TheDriver.BuildCompilation(Args)); 58ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 59ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // Just print the cc1 options if -### was present. 60ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis if (C->getArgs().hasArg(driver::options::OPT__HASH_HASH_HASH)) { 61ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis C->PrintJob(llvm::errs(), C->getJobs(), "\n", true); 62ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis return 0; 63ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis } 64ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 65ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // We expect to get back exactly one command job, if we didn't something 66ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis // failed. 67ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis const driver::JobList &Jobs = C->getJobs(); 684d509341bd5db06a517daa311379f52bb540bc34Eli Friedman if (Jobs.size() != 1 || !isa<driver::Command>(*Jobs.begin())) { 69ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis llvm::SmallString<256> Msg; 70ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis llvm::raw_svector_ostream OS(Msg); 71ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis C->PrintJob(OS, C->getJobs(), "; ", true); 72ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis Diags->Report(diag::err_fe_expected_compiler_job) << OS.str(); 73ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis return 0; 74ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis } 75ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 76ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis const driver::Command *Cmd = cast<driver::Command>(*Jobs.begin()); 775f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner if (StringRef(Cmd->getCreator().getName()) != "clang") { 78ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis Diags->Report(diag::err_fe_expected_clang_command); 79ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis return 0; 80ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis } 81ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis 82ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis const driver::ArgStringList &CCArgs = Cmd->getArguments(); 83ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis CompilerInvocation *CI = new CompilerInvocation(); 84ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis CompilerInvocation::CreateFromArgs(*CI, 85ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis const_cast<const char **>(CCArgs.data()), 86ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis const_cast<const char **>(CCArgs.data()) + 87ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis CCArgs.size(), 88ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis *Diags); 89ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis return CI; 90ea383c0dffb78aea0c427a6ccf4490a5083fcb6aArgyrios Kyrtzidis} 91