Compilation.cpp revision 57a3bbfa9a97f174b0c80b0309e32f4329c4ae1b
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" 12586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar#include "clang/Driver/ArgList.h" 13e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar#include "clang/Driver/Driver.h" 14e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar#include "clang/Driver/DriverDiagnostic.h" 15265e9ef9f3ef30a97790e6e7bbc3c17d97981ca7Daniel Dunbar#include "clang/Driver/Options.h" 16586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar#include "clang/Driver/ToolChain.h" 172b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier#include "llvm/ADT/STLExtras.h" 18c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier#include "llvm/ADT/StringSwitch.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; 263ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 27279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel DunbarCompilation::Compilation(const Driver &D, const ToolChain &_DefaultToolChain, 28279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar InputArgList *_Args, DerivedArgList *_TranslatedArgs) 29279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar : TheDriver(D), DefaultToolChain(_DefaultToolChain), Args(_Args), 302b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TranslatedArgs(_TranslatedArgs), Redirects(0) { 31586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar} 32586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 331eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpCompilation::~Compilation() { 34279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar delete TranslatedArgs; 35586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar delete Args; 361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 37586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar // Free any derived arg lists. 384954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar for (llvm::DenseMap<std::pair<const ToolChain*, const char*>, 394954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar DerivedArgList*>::iterator it = TCArgs.begin(), 404954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar ie = TCArgs.end(); it != ie; ++it) 41279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (it->second != TranslatedArgs) 42279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar delete it->second; 4321549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar 4421549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar // Free the actions, if built. 451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (ActionList::iterator it = Actions.begin(), ie = Actions.end(); 4621549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar it != ie; ++it) 4721549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar delete *it; 482b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 492b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Free redirections of stdout/stderr. 502b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier if (Redirects) { 512b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete Redirects[1]; 522b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete Redirects[2]; 532b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete [] Redirects; 542b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier } 553ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar} 563ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 574954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbarconst DerivedArgList &Compilation::getArgsForToolChain(const ToolChain *TC, 584954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar const char *BoundArch) { 59586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar if (!TC) 60586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar TC = &DefaultToolChain; 61586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 624954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar DerivedArgList *&Entry = TCArgs[std::make_pair(TC, BoundArch)]; 63279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (!Entry) { 64279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar Entry = TC->TranslateArgs(*TranslatedArgs, BoundArch); 65279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (!Entry) 66279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar Entry = TranslatedArgs; 67279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar } 68586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 69aa3e0d292065fd50713b4c70647063c32d99a36aDaniel Dunbar return *Entry; 703ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar} 713ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 725f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnervoid Compilation::PrintJob(raw_ostream &OS, const Job &J, 7349b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const char *Terminator, bool Quote) const { 7449b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar if (const Command *C = dyn_cast<Command>(&J)) { 7524b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar OS << " \"" << C->getExecutable() << '"'; 7624b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar for (ArgStringList::const_iterator it = C->getArguments().begin(), 7749b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar ie = C->getArguments().end(); it != ie; ++it) { 78b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << ' '; 795f22614327065a4ae78588eda8cb62f8b50502aaBenjamin Kramer if (!Quote && !std::strpbrk(*it, " \"\\$")) { 80b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << *it; 81b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar continue; 82b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar } 83b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar 84b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar // Quote the argument and escape shell special characters; this isn't 85b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar // really complete but is good enough. 86b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '"'; 87b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar for (const char *s = *it; *s; ++s) { 88b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar if (*s == '"' || *s == '\\' || *s == '$') 89b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '\\'; 90b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << *s; 91b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar } 92b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '"'; 9349b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } 9424b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar OS << Terminator; 9524b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar } else { 9649b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const JobList *Jobs = cast<JobList>(&J); 971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (JobList::const_iterator 9824b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar it = Jobs->begin(), ie = Jobs->end(); it != ie; ++it) 9949b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar PrintJob(OS, **it, Terminator, Quote); 10024b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar } 10124b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar} 10224b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar 103c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosierstatic bool skipArg(const char *Flag, bool &SkipNextArg) { 104c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier StringRef FlagRef(Flag); 105c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 106c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Assume we're going to see -Flag <Arg>. 107c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier SkipNextArg = true; 108c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 109c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are all of the form -Flag <Arg> and are treated as two 110c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // arguments. Therefore, we need to skip the flag and the next argument. 111c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier bool Res = llvm::StringSwitch<bool>(Flag) 112c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-I", "-MF", "-MT", "-MQ", true) 113c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-o", "-coverage-file", "-dependency-file", true) 114250172a851a886c0763b5fd61c20bf21791c21e9Douglas Gregor .Cases("-fdebug-compilation-dir", "-idirafter", true) 115c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-include", "-include-pch", "-internal-isystem", true) 1161cd46de4840057198607afc875fe9518a78640e1Chad Rosier .Cases("-internal-externc-isystem", "-iprefix", "-iwithprefix", true) 117c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-iwithprefixbefore", "-isysroot", "-isystem", "-iquote", true) 118c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-resource-dir", "-serialize-diagnostic-file", true) 119c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-dwarf-debug-flags", true) 120c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 121c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 122c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Match found. 123c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (Res) 124c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return Res; 125c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 126c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // The remaining flags are treated as a single argument. 127c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier SkipNextArg = false; 128c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 129c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are all of the form -Flag and have no second argument. 130c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier Res = llvm::StringSwitch<bool>(Flag) 131c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-M", "-MM", "-MG", "-MP", "-MD", true) 132c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-MMD", true) 133c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 134c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 135c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Match found. 136c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (Res) 137c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return Res; 138c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 139c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are treated as a single argument (e.g., -F<Dir>). 140c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (FlagRef.startswith("-F") || FlagRef.startswith("-I")) 141c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return true; 142c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 143c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return false; 144c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 145c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 146c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosierstatic bool quoteNextArg(const char *flag) { 147c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return llvm::StringSwitch<bool>(flag) 148c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-D", true) 149c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 150c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 151c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 152c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosiervoid Compilation::PrintDiagnosticJob(raw_ostream &OS, const Job &J) const { 153c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (const Command *C = dyn_cast<Command>(&J)) { 154c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << C->getExecutable(); 155c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier unsigned QuoteNextArg = 0; 156c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (ArgStringList::const_iterator it = C->getArguments().begin(), 157c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier ie = C->getArguments().end(); it != ie; ++it) { 158c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 159c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier bool SkipNext; 160c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (skipArg(*it, SkipNext)) { 161c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (SkipNext) ++it; 162c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier continue; 163c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 164c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 165c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (!QuoteNextArg) 166c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier QuoteNextArg = quoteNextArg(*it) ? 2 : 0; 167c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 168c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << ' '; 169c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 170c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg == 1) 171c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 172c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 173c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (!std::strpbrk(*it, " \"\\$")) { 174c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << *it; 175c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } else { 176c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Quote the argument and escape shell special characters; this isn't 177c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // really complete but is good enough. 178c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 179c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (const char *s = *it; *s; ++s) { 180c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (*s == '"' || *s == '\\' || *s == '$') 181c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '\\'; 182c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << *s; 183c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 184c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 185c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 186c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 187c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg) { 188c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg == 1) 189c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 190c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier --QuoteNextArg; 191c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 192c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 193c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '\n'; 194c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } else { 195c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier const JobList *Jobs = cast<JobList>(&J); 196c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (JobList::const_iterator 197c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier it = Jobs->begin(), ie = Jobs->end(); it != ie; ++it) 198c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier PrintDiagnosticJob(OS, **it); 199c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 200c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 201c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 2029d718635fa805674aaba5d938f3dc6b35b8632baChad Rosierbool Compilation::CleanupFile(const char *File, bool IssueErrors) const { 2039d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier llvm::sys::Path P(File); 2049d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier std::string Error; 2059d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2069d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // Don't try to remove files which we don't have write access to (but may be 2079d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // able to remove), or non-regular files. Underlying tools may have 2089d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // intentionally not overwritten them. 2099d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (!P.canWrite() || !P.isRegularFile()) 2109d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return true; 2119d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2129d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (P.eraseFromDisk(false, &Error)) { 2139d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // Failure is only failure if the file exists and is "regular". There is 2149d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // a race condition here due to the limited interface of 2159d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // llvm::sys::Path, we want to know if the removal gave ENOENT. 2169d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2179d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // FIXME: Grumble, P.exists() is broken. PR3837. 2189d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier struct stat buf; 2199d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (::stat(P.c_str(), &buf) == 0 ? (buf.st_mode & S_IFMT) == S_IFREG : 2209d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier (errno != ENOENT)) { 2219d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (IssueErrors) 2229d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier getDriver().Diag(clang::diag::err_drv_unable_to_remove_file) 2239d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier << Error; 2249d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return false; 2259d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier } 2269d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier } 2279d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return true; 2289d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier} 2299d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpbool Compilation::CleanupFileList(const ArgStringList &Files, 231e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar bool IssueErrors) const { 232e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar bool Success = true; 2331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (ArgStringList::const_iterator 2349d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier it = Files.begin(), ie = Files.end(); it != ie; ++it) 2359d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier Success &= CleanupFile(*it, IssueErrors); 2369d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return Success; 2379d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier} 23856eec2bc2188574459a514f3650d581e00683126Edward O'Callaghan 2399d718635fa805674aaba5d938f3dc6b35b8632baChad Rosierbool Compilation::CleanupFileMap(const ArgStringMap &Files, 2409d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier const JobAction *JA, 2419d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier bool IssueErrors) const { 2429d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier bool Success = true; 2439d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier for (ArgStringMap::const_iterator 2449d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier it = Files.begin(), ie = Files.end(); it != ie; ++it) { 245e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar 2469d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // If specified, only delete the files associated with the JobAction. 2479d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // Otherwise, delete all files in the map. 2489d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (JA && it->first != JA) 2498ac127a09ab36f84860c176fe6b199c4973be984Daniel Dunbar continue; 2509d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier Success &= CleanupFile(it->second, IssueErrors); 251e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar } 252e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar return Success; 253e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar} 254e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar 25531c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbarint Compilation::ExecuteCommand(const Command &C, 25631c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar const Command *&FailingCommand) const { 257ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar llvm::sys::Path Prog(C.getExecutable()); 258ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar const char **Argv = new const char*[C.getArguments().size() + 2]; 259ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar Argv[0] = C.getExecutable(); 260ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar std::copy(C.getArguments().begin(), C.getArguments().end(), Argv+1); 261ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar Argv[C.getArguments().size() + 1] = 0; 2621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2632b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier if ((getDriver().CCCEcho || getDriver().CCPrintOptions || 2642b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier getArgs().hasArg(options::OPT_v)) && !getDriver().CCGenDiagnostics) { 2655f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner raw_ostream *OS = &llvm::errs(); 2664c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2674c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar // Follow gcc implementation of CC_PRINT_OPTIONS; we could also cache the 2684c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar // output stream. 2694c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (getDriver().CCPrintOptions && getDriver().CCPrintOptionsFilename) { 2704c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar std::string Error; 2714c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar OS = new llvm::raw_fd_ostream(getDriver().CCPrintOptionsFilename, 2724c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar Error, 2734c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar llvm::raw_fd_ostream::F_Append); 2744c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (!Error.empty()) { 2754c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar getDriver().Diag(clang::diag::err_drv_cc_print_options_failure) 2764c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar << Error; 2774c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar FailingCommand = &C; 2784c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar delete OS; 2794c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar return 1; 2804c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2814c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2824c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2834c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (getDriver().CCPrintOptions) 2844c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar *OS << "[Logging clang options]"; 2854c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2864c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar PrintJob(*OS, C, "\n", /*Quote=*/getDriver().CCPrintOptions); 2874c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2884c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (OS != &llvm::errs()) 2894c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar delete OS; 2904c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 292ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar std::string Error; 293c48d575f2e41bec767ccdea6591404c907f1bc91Chad Rosier bool ExecutionFailed; 29457a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola int Res = llvm::sys::ExecuteAndWait(Prog.str(), Argv, /*env*/ 0, Redirects, 295a603569515eea06e54e6e041b1c690d33086f375Rafael Espindola /*secondsToWait*/ 0, /*memoryLimit*/ 0, 296a603569515eea06e54e6e041b1c690d33086f375Rafael Espindola &Error, &ExecutionFailed); 297ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar if (!Error.empty()) { 298ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar assert(Res && "Error string set with 0 result code!"); 299ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar getDriver().Diag(clang::diag::err_drv_command_failure) << Error; 300ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar } 3011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 30231c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar if (Res) 30331c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar FailingCommand = &C; 30431c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar 305ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar delete[] Argv; 306c48d575f2e41bec767ccdea6591404c907f1bc91Chad Rosier return ExecutionFailed ? 1 : Res; 307ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar} 308ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar 3094c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosiertypedef SmallVectorImpl< std::pair<int, const Command *> > FailingCommandList; 3104c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3114c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosierstatic bool ActionFailed(const Action *A, 3124c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier const FailingCommandList &FailingCommands) { 3134c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3144c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (FailingCommands.empty()) 3154c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return false; 3164c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3174c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier for (FailingCommandList::const_iterator CI = FailingCommands.begin(), 3184c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier CE = FailingCommands.end(); CI != CE; ++CI) 3194c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (A == &(CI->second->getSource())) 3204c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return true; 3214c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3224c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier for (Action::const_iterator AI = A->begin(), AE = A->end(); AI != AE; ++AI) 3234c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (ActionFailed(*AI, FailingCommands)) 3244c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return true; 3254c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3264c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return false; 3274c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier} 3284c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3294c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosierstatic bool InputsOk(const Command &C, 3304c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier const FailingCommandList &FailingCommands) { 3314c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return !ActionFailed(&C.getSource(), FailingCommands); 3324c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier} 3334c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 334a16355c31878403443f99077cc8df8318457faf5Chad Rosiervoid Compilation::ExecuteJob(const Job &J, 3354c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier FailingCommandList &FailingCommands) const { 33649b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar if (const Command *C = dyn_cast<Command>(&J)) { 3374c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (!InputsOk(*C, FailingCommands)) 3384c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return; 339a16355c31878403443f99077cc8df8318457faf5Chad Rosier const Command *FailingCommand = 0; 340a16355c31878403443f99077cc8df8318457faf5Chad Rosier if (int Res = ExecuteCommand(*C, FailingCommand)) 341a16355c31878403443f99077cc8df8318457faf5Chad Rosier FailingCommands.push_back(std::make_pair(Res, FailingCommand)); 34249b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } else { 34349b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const JobList *Jobs = cast<JobList>(&J); 344a16355c31878403443f99077cc8df8318457faf5Chad Rosier for (JobList::const_iterator it = Jobs->begin(), ie = Jobs->end(); 345a16355c31878403443f99077cc8df8318457faf5Chad Rosier it != ie; ++it) 346a16355c31878403443f99077cc8df8318457faf5Chad Rosier ExecuteJob(**it, FailingCommands); 34749b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } 34849b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar} 3492b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 350c4a77906c259cba58c147d8468c406a430ecdcbbDmitri Gribenkovoid Compilation::initCompilationForDiagnostics() { 3512b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Free actions and jobs. 3522b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier DeleteContainerPointers(Actions); 3532b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier Jobs.clear(); 3542b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3552b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Clear temporary/results file lists. 3562b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TempFiles.clear(); 3572b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier ResultFiles.clear(); 3588425a5413b437a0b6ac04c475e7cf54cc9977880Chad Rosier FailureResultFiles.clear(); 3592b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3602b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Remove any user specified output. Claim any unclaimed arguments, so as 3612b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // to avoid emitting warnings about unused args. 36254db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne OptSpecifier OutputOpts[] = { options::OPT_o, options::OPT_MD, 36354db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne options::OPT_MMD }; 364f9e156c774e08409f235f0e9c9e3133ec796e877Chad Rosier for (unsigned i = 0, e = llvm::array_lengthof(OutputOpts); i != e; ++i) { 36554db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne if (TranslatedArgs->hasArg(OutputOpts[i])) 36654db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne TranslatedArgs->eraseArg(OutputOpts[i]); 36754db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne } 3682b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TranslatedArgs->ClaimAllArgs(); 3692b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3702b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Redirect stdout/stderr to /dev/null. 37157a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects = new const StringRef*[3](); 37257a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[0] = 0; 37357a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[1] = new const StringRef(); 37457a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[2] = new const StringRef(); 3752b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier} 3764762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop 377c4a77906c259cba58c147d8468c406a430ecdcbbDmitri GribenkoStringRef Compilation::getSysRoot() const { 3784762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop return getDriver().SysRoot; 3794762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop} 380