diff options
author | ed <ed@FreeBSD.org> | 2009-06-02 17:58:47 +0000 |
---|---|---|
committer | ed <ed@FreeBSD.org> | 2009-06-02 17:58:47 +0000 |
commit | f27e5a09a0d815b8a4814152954ff87dadfdefc0 (patch) | |
tree | ce7d964cbb5e39695b71481698f10cb099c23d4a /lib/Driver/Compilation.cpp | |
download | FreeBSD-src-f27e5a09a0d815b8a4814152954ff87dadfdefc0.zip FreeBSD-src-f27e5a09a0d815b8a4814152954ff87dadfdefc0.tar.gz |
Import Clang, at r72732.
Diffstat (limited to 'lib/Driver/Compilation.cpp')
-rw-r--r-- | lib/Driver/Compilation.cpp | 174 |
1 files changed, 174 insertions, 0 deletions
diff --git a/lib/Driver/Compilation.cpp b/lib/Driver/Compilation.cpp new file mode 100644 index 0000000..1e044c6b --- /dev/null +++ b/lib/Driver/Compilation.cpp @@ -0,0 +1,174 @@ +//===--- Compilation.cpp - Compilation Task Implementation --------------*-===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +#include "clang/Driver/Compilation.h" + +#include "clang/Driver/Action.h" +#include "clang/Driver/ArgList.h" +#include "clang/Driver/Driver.h" +#include "clang/Driver/DriverDiagnostic.h" +#include "clang/Driver/ToolChain.h" + +#include "llvm/Support/raw_ostream.h" +#include "llvm/System/Program.h" +#include <sys/stat.h> +#include <errno.h> +using namespace clang::driver; + +Compilation::Compilation(Driver &D, + ToolChain &_DefaultToolChain, + InputArgList *_Args) + : TheDriver(D), DefaultToolChain(_DefaultToolChain), Args(_Args) { +} + +Compilation::~Compilation() { + delete Args; + + // Free any derived arg lists. + for (llvm::DenseMap<const ToolChain*, DerivedArgList*>::iterator + it = TCArgs.begin(), ie = TCArgs.end(); it != ie; ++it) + delete it->second; + + // Free the actions, if built. + for (ActionList::iterator it = Actions.begin(), ie = Actions.end(); + it != ie; ++it) + delete *it; +} + +const DerivedArgList &Compilation::getArgsForToolChain(const ToolChain *TC) { + if (!TC) + TC = &DefaultToolChain; + + DerivedArgList *&Entry = TCArgs[TC]; + if (!Entry) + Entry = TC->TranslateArgs(*Args); + + return *Entry; +} + +void Compilation::PrintJob(llvm::raw_ostream &OS, const Job &J, + const char *Terminator, bool Quote) const { + if (const Command *C = dyn_cast<Command>(&J)) { + OS << " \"" << C->getExecutable() << '"'; + for (ArgStringList::const_iterator it = C->getArguments().begin(), + ie = C->getArguments().end(); it != ie; ++it) { + if (Quote) + OS << " \"" << *it << '"'; + else + OS << ' ' << *it; + } + OS << Terminator; + } else if (const PipedJob *PJ = dyn_cast<PipedJob>(&J)) { + for (PipedJob::const_iterator + it = PJ->begin(), ie = PJ->end(); it != ie; ++it) + PrintJob(OS, **it, (it + 1 != PJ->end()) ? " |\n" : "\n", Quote); + } else { + const JobList *Jobs = cast<JobList>(&J); + for (JobList::const_iterator + it = Jobs->begin(), ie = Jobs->end(); it != ie; ++it) + PrintJob(OS, **it, Terminator, Quote); + } +} + +bool Compilation::CleanupFileList(const ArgStringList &Files, + bool IssueErrors) const { + bool Success = true; + + for (ArgStringList::const_iterator + it = Files.begin(), ie = Files.end(); it != ie; ++it) { + llvm::sys::Path P(*it); + std::string Error; + + if (P.eraseFromDisk(false, &Error)) { + // Failure is only failure if the file doesn't exist. There is a + // race condition here due to the limited interface of + // llvm::sys::Path, we want to know if the removal gave E_NOENT. + + // FIXME: Grumble, P.exists() is broken. PR3837. + struct stat buf; + if (::stat(P.c_str(), &buf) == 0 + || errno != ENOENT) { + if (IssueErrors) + getDriver().Diag(clang::diag::err_drv_unable_to_remove_file) + << Error; + Success = false; + } + } + } + + return Success; +} + +int Compilation::ExecuteCommand(const Command &C) const { + llvm::sys::Path Prog(C.getExecutable()); + const char **Argv = new const char*[C.getArguments().size() + 2]; + Argv[0] = C.getExecutable(); + std::copy(C.getArguments().begin(), C.getArguments().end(), Argv+1); + Argv[C.getArguments().size() + 1] = 0; + + if (getDriver().CCCEcho || getArgs().hasArg(options::OPT_v)) + PrintJob(llvm::errs(), C, "\n", false); + + std::string Error; + int Res = + llvm::sys::Program::ExecuteAndWait(Prog, Argv, + /*env*/0, /*redirects*/0, + /*secondsToWait*/0, /*memoryLimit*/0, + &Error); + if (!Error.empty()) { + assert(Res && "Error string set with 0 result code!"); + getDriver().Diag(clang::diag::err_drv_command_failure) << Error; + } + + delete[] Argv; + return Res; +} + +int Compilation::ExecuteJob(const Job &J) const { + if (const Command *C = dyn_cast<Command>(&J)) { + return ExecuteCommand(*C); + } else if (const PipedJob *PJ = dyn_cast<PipedJob>(&J)) { + // Piped commands with a single job are easy. + if (PJ->size() == 1) + return ExecuteCommand(**PJ->begin()); + + getDriver().Diag(clang::diag::err_drv_unsupported_opt) << "-pipe"; + return 1; + } else { + const JobList *Jobs = cast<JobList>(&J); + for (JobList::const_iterator + it = Jobs->begin(), ie = Jobs->end(); it != ie; ++it) + if (int Res = ExecuteJob(**it)) + return Res; + return 0; + } +} + +int Compilation::Execute() const { + // Just print if -### was present. + if (getArgs().hasArg(options::OPT__HASH_HASH_HASH)) { + PrintJob(llvm::errs(), Jobs, "\n", true); + return 0; + } + + // If there were errors building the compilation, quit now. + if (getDriver().getDiags().getNumErrors()) + return 1; + + int Res = ExecuteJob(Jobs); + + // Remove temp files. + CleanupFileList(TempFiles); + + // If the compilation failed, remove result files as well. + if (Res != 0 && !getArgs().hasArg(options::OPT_save_temps)) + CleanupFileList(ResultFiles, true); + + return Res; +} |