Compilation.cpp revision b1e25a1bc03292dc538d336573e0be1490223171
13fdcc6fb12d7cf38d2a3111736f80f0dd55447b4Nick Lewycky//===--- Compilation.cpp - Compilation Task Implementation ----------------===// 23ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar// 33ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar// The LLVM Compiler Infrastructure 43ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar// 53ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar// This file is distributed under the University of Illinois Open Source 63ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar// License. See LICENSE.TXT for details. 73ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar// 83ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar//===----------------------------------------------------------------------===// 93ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 103ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar#include "clang/Driver/Compilation.h" 1121549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar#include "clang/Driver/Action.h" 12e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar#include "clang/Driver/Driver.h" 13e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar#include "clang/Driver/DriverDiagnostic.h" 14265e9ef9f3ef30a97790e6e7bbc3c17d97981ca7Daniel Dunbar#include "clang/Driver/Options.h" 15586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar#include "clang/Driver/ToolChain.h" 162b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier#include "llvm/ADT/STLExtras.h" 17c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier#include "llvm/ADT/StringSwitch.h" 18b1e25a1bc03292dc538d336573e0be1490223171Reid Kleckner#include "llvm/Option/ArgList.h" 1903013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Program.h" 2055fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include "llvm/Support/raw_ostream.h" 21e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar#include <errno.h> 2255fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include <sys/stat.h> 231c3199afb8aeab5f4dfcef60999d13d9c6877a67Francois Pichet 241b3bb6efc59a21f794b534078f9ae7e95393f510Daniel Dunbarusing namespace clang::driver; 251c3199afb8aeab5f4dfcef60999d13d9c6877a67Francois Pichetusing namespace clang; 26b1e25a1bc03292dc538d336573e0be1490223171Reid Klecknerusing namespace llvm::opt; 273ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 28279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel DunbarCompilation::Compilation(const Driver &D, const ToolChain &_DefaultToolChain, 29279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar InputArgList *_Args, DerivedArgList *_TranslatedArgs) 30279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar : TheDriver(D), DefaultToolChain(_DefaultToolChain), Args(_Args), 312b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TranslatedArgs(_TranslatedArgs), Redirects(0) { 32586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar} 33586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 341eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpCompilation::~Compilation() { 35279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar delete TranslatedArgs; 36586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar delete Args; 371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 38586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar // Free any derived arg lists. 394954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar for (llvm::DenseMap<std::pair<const ToolChain*, const char*>, 404954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar DerivedArgList*>::iterator it = TCArgs.begin(), 414954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar ie = TCArgs.end(); it != ie; ++it) 42279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (it->second != TranslatedArgs) 43279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar delete it->second; 4421549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar 4521549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar // Free the actions, if built. 461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (ActionList::iterator it = Actions.begin(), ie = Actions.end(); 4721549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar it != ie; ++it) 4821549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar delete *it; 492b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 502b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Free redirections of stdout/stderr. 512b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier if (Redirects) { 522b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete Redirects[1]; 532b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete Redirects[2]; 542b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete [] Redirects; 552b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier } 563ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar} 573ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 584954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbarconst DerivedArgList &Compilation::getArgsForToolChain(const ToolChain *TC, 594954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar const char *BoundArch) { 60586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar if (!TC) 61586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar TC = &DefaultToolChain; 62586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 634954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar DerivedArgList *&Entry = TCArgs[std::make_pair(TC, BoundArch)]; 64279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (!Entry) { 65279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar Entry = TC->TranslateArgs(*TranslatedArgs, BoundArch); 66279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (!Entry) 67279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar Entry = TranslatedArgs; 68279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar } 69586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 70aa3e0d292065fd50713b4c70647063c32d99a36aDaniel Dunbar return *Entry; 713ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar} 723ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 735f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnervoid Compilation::PrintJob(raw_ostream &OS, const Job &J, 7449b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const char *Terminator, bool Quote) const { 7549b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar if (const Command *C = dyn_cast<Command>(&J)) { 7624b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar OS << " \"" << C->getExecutable() << '"'; 7724b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar for (ArgStringList::const_iterator it = C->getArguments().begin(), 7849b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar ie = C->getArguments().end(); it != ie; ++it) { 79b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << ' '; 805f22614327065a4ae78588eda8cb62f8b50502aaBenjamin Kramer if (!Quote && !std::strpbrk(*it, " \"\\$")) { 81b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << *it; 82b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar continue; 83b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar } 84b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar 85b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar // Quote the argument and escape shell special characters; this isn't 86b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar // really complete but is good enough. 87b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '"'; 88b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar for (const char *s = *it; *s; ++s) { 89b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar if (*s == '"' || *s == '\\' || *s == '$') 90b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '\\'; 91b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << *s; 92b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar } 93b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '"'; 9449b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } 9524b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar OS << Terminator; 9624b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar } else { 9749b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const JobList *Jobs = cast<JobList>(&J); 981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (JobList::const_iterator 9924b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar it = Jobs->begin(), ie = Jobs->end(); it != ie; ++it) 10049b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar PrintJob(OS, **it, Terminator, Quote); 10124b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar } 10224b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar} 10324b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar 104c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosierstatic bool skipArg(const char *Flag, bool &SkipNextArg) { 105c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier StringRef FlagRef(Flag); 106c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 107c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Assume we're going to see -Flag <Arg>. 108c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier SkipNextArg = true; 109c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 110c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are all of the form -Flag <Arg> and are treated as two 111c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // arguments. Therefore, we need to skip the flag and the next argument. 112c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier bool Res = llvm::StringSwitch<bool>(Flag) 113c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-I", "-MF", "-MT", "-MQ", true) 114c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-o", "-coverage-file", "-dependency-file", true) 115250172a851a886c0763b5fd61c20bf21791c21e9Douglas Gregor .Cases("-fdebug-compilation-dir", "-idirafter", true) 116c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-include", "-include-pch", "-internal-isystem", true) 1171cd46de4840057198607afc875fe9518a78640e1Chad Rosier .Cases("-internal-externc-isystem", "-iprefix", "-iwithprefix", true) 118c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-iwithprefixbefore", "-isysroot", "-isystem", "-iquote", true) 119c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-resource-dir", "-serialize-diagnostic-file", true) 120c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-dwarf-debug-flags", true) 121c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 122c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 123c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Match found. 124c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (Res) 125c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return Res; 126c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 127c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // The remaining flags are treated as a single argument. 128c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier SkipNextArg = false; 129c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 130c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are all of the form -Flag and have no second argument. 131c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier Res = llvm::StringSwitch<bool>(Flag) 132c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-M", "-MM", "-MG", "-MP", "-MD", true) 133c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-MMD", true) 134c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 135c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 136c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Match found. 137c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (Res) 138c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return Res; 139c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 140c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are treated as a single argument (e.g., -F<Dir>). 141c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (FlagRef.startswith("-F") || FlagRef.startswith("-I")) 142c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return true; 143c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 144c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return false; 145c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 146c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 147c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosierstatic bool quoteNextArg(const char *flag) { 148c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return llvm::StringSwitch<bool>(flag) 149c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-D", true) 150c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 151c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 152c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 153c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosiervoid Compilation::PrintDiagnosticJob(raw_ostream &OS, const Job &J) const { 154c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (const Command *C = dyn_cast<Command>(&J)) { 155c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << C->getExecutable(); 156c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier unsigned QuoteNextArg = 0; 157c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (ArgStringList::const_iterator it = C->getArguments().begin(), 158c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier ie = C->getArguments().end(); it != ie; ++it) { 159c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 160c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier bool SkipNext; 161c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (skipArg(*it, SkipNext)) { 162c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (SkipNext) ++it; 163c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier continue; 164c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 165c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 166c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (!QuoteNextArg) 167c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier QuoteNextArg = quoteNextArg(*it) ? 2 : 0; 168c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 169c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << ' '; 170c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 171c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg == 1) 172c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 173c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 174c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (!std::strpbrk(*it, " \"\\$")) { 175c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << *it; 176c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } else { 177c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Quote the argument and escape shell special characters; this isn't 178c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // really complete but is good enough. 179c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 180c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (const char *s = *it; *s; ++s) { 181c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (*s == '"' || *s == '\\' || *s == '$') 182c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '\\'; 183c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << *s; 184c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 185c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 186c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 187c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 188c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg) { 189c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg == 1) 190c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 191c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier --QuoteNextArg; 192c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 193c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 194c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '\n'; 195c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } else { 196c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier const JobList *Jobs = cast<JobList>(&J); 197c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (JobList::const_iterator 198c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier it = Jobs->begin(), ie = Jobs->end(); it != ie; ++it) 199c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier PrintDiagnosticJob(OS, **it); 200c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 201c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 202c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 2039d718635fa805674aaba5d938f3dc6b35b8632baChad Rosierbool Compilation::CleanupFile(const char *File, bool IssueErrors) const { 2049d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier llvm::sys::Path P(File); 2059d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier std::string Error; 2069d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2079d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // Don't try to remove files which we don't have write access to (but may be 2089d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // able to remove), or non-regular files. Underlying tools may have 2099d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // intentionally not overwritten them. 2109d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (!P.canWrite() || !P.isRegularFile()) 2119d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return true; 2129d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2139d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (P.eraseFromDisk(false, &Error)) { 2149d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // Failure is only failure if the file exists and is "regular". There is 2159d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // a race condition here due to the limited interface of 2169d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // llvm::sys::Path, we want to know if the removal gave ENOENT. 2179d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2189d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // FIXME: Grumble, P.exists() is broken. PR3837. 2199d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier struct stat buf; 2209d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (::stat(P.c_str(), &buf) == 0 ? (buf.st_mode & S_IFMT) == S_IFREG : 2219d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier (errno != ENOENT)) { 2229d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (IssueErrors) 2239d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier getDriver().Diag(clang::diag::err_drv_unable_to_remove_file) 2249d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier << Error; 2259d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return false; 2269d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier } 2279d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier } 2289d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return true; 2299d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier} 2309d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpbool Compilation::CleanupFileList(const ArgStringList &Files, 232e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar bool IssueErrors) const { 233e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar bool Success = true; 2341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (ArgStringList::const_iterator 2359d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier it = Files.begin(), ie = Files.end(); it != ie; ++it) 2369d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier Success &= CleanupFile(*it, IssueErrors); 2379d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return Success; 2389d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier} 23956eec2bc2188574459a514f3650d581e00683126Edward O'Callaghan 2409d718635fa805674aaba5d938f3dc6b35b8632baChad Rosierbool Compilation::CleanupFileMap(const ArgStringMap &Files, 2419d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier const JobAction *JA, 2429d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier bool IssueErrors) const { 2439d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier bool Success = true; 2449d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier for (ArgStringMap::const_iterator 2459d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier it = Files.begin(), ie = Files.end(); it != ie; ++it) { 246e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar 2479d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // If specified, only delete the files associated with the JobAction. 2489d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // Otherwise, delete all files in the map. 2499d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (JA && it->first != JA) 2508ac127a09ab36f84860c176fe6b199c4973be984Daniel Dunbar continue; 2519d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier Success &= CleanupFile(it->second, IssueErrors); 252e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar } 253e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar return Success; 254e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar} 255e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar 25631c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbarint Compilation::ExecuteCommand(const Command &C, 25731c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar const Command *&FailingCommand) const { 258ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar llvm::sys::Path Prog(C.getExecutable()); 259ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar const char **Argv = new const char*[C.getArguments().size() + 2]; 260ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar Argv[0] = C.getExecutable(); 261ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar std::copy(C.getArguments().begin(), C.getArguments().end(), Argv+1); 262ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar Argv[C.getArguments().size() + 1] = 0; 2631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2642b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier if ((getDriver().CCCEcho || getDriver().CCPrintOptions || 2652b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier getArgs().hasArg(options::OPT_v)) && !getDriver().CCGenDiagnostics) { 2665f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner raw_ostream *OS = &llvm::errs(); 2674c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2684c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar // Follow gcc implementation of CC_PRINT_OPTIONS; we could also cache the 2694c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar // output stream. 2704c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (getDriver().CCPrintOptions && getDriver().CCPrintOptionsFilename) { 2714c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar std::string Error; 2724c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar OS = new llvm::raw_fd_ostream(getDriver().CCPrintOptionsFilename, 2734c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar Error, 2744c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar llvm::raw_fd_ostream::F_Append); 2754c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (!Error.empty()) { 2764c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar getDriver().Diag(clang::diag::err_drv_cc_print_options_failure) 2774c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar << Error; 2784c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar FailingCommand = &C; 2794c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar delete OS; 2804c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar return 1; 2814c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2824c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2834c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2844c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (getDriver().CCPrintOptions) 2854c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar *OS << "[Logging clang options]"; 2864c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2874c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar PrintJob(*OS, C, "\n", /*Quote=*/getDriver().CCPrintOptions); 2884c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2894c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (OS != &llvm::errs()) 2904c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar delete OS; 2914c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 293ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar std::string Error; 294c48d575f2e41bec767ccdea6591404c907f1bc91Chad Rosier bool ExecutionFailed; 29557a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola int Res = llvm::sys::ExecuteAndWait(Prog.str(), Argv, /*env*/ 0, Redirects, 296a603569515eea06e54e6e041b1c690d33086f375Rafael Espindola /*secondsToWait*/ 0, /*memoryLimit*/ 0, 297a603569515eea06e54e6e041b1c690d33086f375Rafael Espindola &Error, &ExecutionFailed); 298ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar if (!Error.empty()) { 299ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar assert(Res && "Error string set with 0 result code!"); 300ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar getDriver().Diag(clang::diag::err_drv_command_failure) << Error; 301ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar } 3021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 30331c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar if (Res) 30431c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar FailingCommand = &C; 30531c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar 306ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar delete[] Argv; 307c48d575f2e41bec767ccdea6591404c907f1bc91Chad Rosier return ExecutionFailed ? 1 : Res; 308ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar} 309ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar 3104c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosiertypedef SmallVectorImpl< std::pair<int, const Command *> > FailingCommandList; 3114c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3124c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosierstatic bool ActionFailed(const Action *A, 3134c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier const FailingCommandList &FailingCommands) { 3144c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3154c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (FailingCommands.empty()) 3164c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return false; 3174c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3184c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier for (FailingCommandList::const_iterator CI = FailingCommands.begin(), 3194c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier CE = FailingCommands.end(); CI != CE; ++CI) 3204c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (A == &(CI->second->getSource())) 3214c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return true; 3224c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3234c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier for (Action::const_iterator AI = A->begin(), AE = A->end(); AI != AE; ++AI) 3244c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (ActionFailed(*AI, FailingCommands)) 3254c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return true; 3264c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3274c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return false; 3284c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier} 3294c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3304c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosierstatic bool InputsOk(const Command &C, 3314c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier const FailingCommandList &FailingCommands) { 3324c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return !ActionFailed(&C.getSource(), FailingCommands); 3334c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier} 3344c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 335a16355c31878403443f99077cc8df8318457faf5Chad Rosiervoid Compilation::ExecuteJob(const Job &J, 3364c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier FailingCommandList &FailingCommands) const { 33749b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar if (const Command *C = dyn_cast<Command>(&J)) { 3384c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (!InputsOk(*C, FailingCommands)) 3394c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return; 340a16355c31878403443f99077cc8df8318457faf5Chad Rosier const Command *FailingCommand = 0; 341a16355c31878403443f99077cc8df8318457faf5Chad Rosier if (int Res = ExecuteCommand(*C, FailingCommand)) 342a16355c31878403443f99077cc8df8318457faf5Chad Rosier FailingCommands.push_back(std::make_pair(Res, FailingCommand)); 34349b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } else { 34449b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const JobList *Jobs = cast<JobList>(&J); 345a16355c31878403443f99077cc8df8318457faf5Chad Rosier for (JobList::const_iterator it = Jobs->begin(), ie = Jobs->end(); 346a16355c31878403443f99077cc8df8318457faf5Chad Rosier it != ie; ++it) 347a16355c31878403443f99077cc8df8318457faf5Chad Rosier ExecuteJob(**it, FailingCommands); 34849b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } 34949b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar} 3502b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 351c4a77906c259cba58c147d8468c406a430ecdcbbDmitri Gribenkovoid Compilation::initCompilationForDiagnostics() { 3522b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Free actions and jobs. 3532b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier DeleteContainerPointers(Actions); 3542b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier Jobs.clear(); 3552b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3562b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Clear temporary/results file lists. 3572b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TempFiles.clear(); 3582b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier ResultFiles.clear(); 3598425a5413b437a0b6ac04c475e7cf54cc9977880Chad Rosier FailureResultFiles.clear(); 3602b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3612b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Remove any user specified output. Claim any unclaimed arguments, so as 3622b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // to avoid emitting warnings about unused args. 36354db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne OptSpecifier OutputOpts[] = { options::OPT_o, options::OPT_MD, 36454db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne options::OPT_MMD }; 365f9e156c774e08409f235f0e9c9e3133ec796e877Chad Rosier for (unsigned i = 0, e = llvm::array_lengthof(OutputOpts); i != e; ++i) { 36654db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne if (TranslatedArgs->hasArg(OutputOpts[i])) 36754db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne TranslatedArgs->eraseArg(OutputOpts[i]); 36854db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne } 3692b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TranslatedArgs->ClaimAllArgs(); 3702b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3712b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Redirect stdout/stderr to /dev/null. 37257a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects = new const StringRef*[3](); 37357a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[0] = 0; 37457a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[1] = new const StringRef(); 37557a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[2] = new const StringRef(); 3762b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier} 3774762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop 378c4a77906c259cba58c147d8468c406a430ecdcbbDmitri GribenkoStringRef Compilation::getSysRoot() const { 3794762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop return getDriver().SysRoot; 3804762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop} 381