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" 19da1f9cb8ce0e89d2848390aef985bad9e32e1ddbRafael Espindola#include "llvm/Support/FileSystem.h" 2003013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Program.h" 2155fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include "llvm/Support/raw_ostream.h" 22e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar#include <errno.h> 2355fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include <sys/stat.h> 241c3199afb8aeab5f4dfcef60999d13d9c6877a67Francois Pichet 251b3bb6efc59a21f794b534078f9ae7e95393f510Daniel Dunbarusing namespace clang::driver; 261c3199afb8aeab5f4dfcef60999d13d9c6877a67Francois Pichetusing namespace clang; 27b1e25a1bc03292dc538d336573e0be1490223171Reid Klecknerusing namespace llvm::opt; 283ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 29279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel DunbarCompilation::Compilation(const Driver &D, const ToolChain &_DefaultToolChain, 30279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar InputArgList *_Args, DerivedArgList *_TranslatedArgs) 31279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar : TheDriver(D), DefaultToolChain(_DefaultToolChain), Args(_Args), 322b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TranslatedArgs(_TranslatedArgs), Redirects(0) { 33586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar} 34586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 351eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpCompilation::~Compilation() { 36279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar delete TranslatedArgs; 37586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar delete Args; 381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 39586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar // Free any derived arg lists. 404954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar for (llvm::DenseMap<std::pair<const ToolChain*, const char*>, 414954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar DerivedArgList*>::iterator it = TCArgs.begin(), 424954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar ie = TCArgs.end(); it != ie; ++it) 43279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (it->second != TranslatedArgs) 44279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar delete it->second; 4521549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar 4621549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar // Free the actions, if built. 471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (ActionList::iterator it = Actions.begin(), ie = Actions.end(); 4821549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar it != ie; ++it) 4921549237f14505cfc2a18a06416372a36229d0ceDaniel Dunbar delete *it; 502b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 512b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Free redirections of stdout/stderr. 522b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier if (Redirects) { 532b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete Redirects[1]; 542b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete Redirects[2]; 552b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier delete [] Redirects; 562b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier } 573ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar} 583ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 594954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbarconst DerivedArgList &Compilation::getArgsForToolChain(const ToolChain *TC, 604954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar const char *BoundArch) { 61586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar if (!TC) 62586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar TC = &DefaultToolChain; 63586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 644954018954bbc97363eef02d0c83bea19ce9b329Daniel Dunbar DerivedArgList *&Entry = TCArgs[std::make_pair(TC, BoundArch)]; 65279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (!Entry) { 66279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar Entry = TC->TranslateArgs(*TranslatedArgs, BoundArch); 67279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar if (!Entry) 68279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar Entry = TranslatedArgs; 69279c1dbebf37cd128f3c73c70741a6b8c35ad025Daniel Dunbar } 70586dc233bb88f2920c9f3638f69cef0ccd55dcedDaniel Dunbar 71aa3e0d292065fd50713b4c70647063c32d99a36aDaniel Dunbar return *Entry; 723ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar} 733ede8d0a7d1813f678ccc6011a99a0834b1b6116Daniel Dunbar 745f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnervoid Compilation::PrintJob(raw_ostream &OS, const Job &J, 7549b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const char *Terminator, bool Quote) const { 7649b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar if (const Command *C = dyn_cast<Command>(&J)) { 7724b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar OS << " \"" << C->getExecutable() << '"'; 7824b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar for (ArgStringList::const_iterator it = C->getArguments().begin(), 7949b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar ie = C->getArguments().end(); it != ie; ++it) { 80b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << ' '; 815f22614327065a4ae78588eda8cb62f8b50502aaBenjamin Kramer if (!Quote && !std::strpbrk(*it, " \"\\$")) { 82b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << *it; 83b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar continue; 84b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar } 85b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar 86b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar // Quote the argument and escape shell special characters; this isn't 87b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar // really complete but is good enough. 88b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '"'; 89b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar for (const char *s = *it; *s; ++s) { 90b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar if (*s == '"' || *s == '\\' || *s == '$') 91b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '\\'; 92b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << *s; 93b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar } 94b86c5fc7393f61221686fc56e992ca409dee2a50Daniel Dunbar OS << '"'; 9549b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } 9624b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar OS << Terminator; 9724b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar } else { 9849b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const JobList *Jobs = cast<JobList>(&J); 991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (JobList::const_iterator 10024b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar it = Jobs->begin(), ie = Jobs->end(); it != ie; ++it) 10149b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar PrintJob(OS, **it, Terminator, Quote); 10224b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar } 10324b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar} 10424b5560b6ea51b8a260194710468fa060775fc01Daniel Dunbar 105c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosierstatic bool skipArg(const char *Flag, bool &SkipNextArg) { 106c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier StringRef FlagRef(Flag); 107c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 108c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Assume we're going to see -Flag <Arg>. 109c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier SkipNextArg = true; 110c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 111c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are all of the form -Flag <Arg> and are treated as two 112c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // arguments. Therefore, we need to skip the flag and the next argument. 113c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier bool Res = llvm::StringSwitch<bool>(Flag) 114c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-I", "-MF", "-MT", "-MQ", true) 115c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-o", "-coverage-file", "-dependency-file", true) 116250172a851a886c0763b5fd61c20bf21791c21e9Douglas Gregor .Cases("-fdebug-compilation-dir", "-idirafter", true) 117c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-include", "-include-pch", "-internal-isystem", true) 1181cd46de4840057198607afc875fe9518a78640e1Chad Rosier .Cases("-internal-externc-isystem", "-iprefix", "-iwithprefix", true) 119c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-iwithprefixbefore", "-isysroot", "-isystem", "-iquote", true) 120c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-resource-dir", "-serialize-diagnostic-file", true) 121c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-dwarf-debug-flags", true) 122c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 123c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 124c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Match found. 125c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (Res) 126c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return Res; 127c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 128c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // The remaining flags are treated as a single argument. 129c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier SkipNextArg = false; 130c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 131c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are all of the form -Flag and have no second argument. 132c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier Res = llvm::StringSwitch<bool>(Flag) 133c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Cases("-M", "-MM", "-MG", "-MP", "-MD", true) 134c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-MMD", true) 135c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 136c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 137c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Match found. 138c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (Res) 139c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return Res; 140c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 141c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // These flags are treated as a single argument (e.g., -F<Dir>). 142c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (FlagRef.startswith("-F") || FlagRef.startswith("-I")) 143c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return true; 144c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 145c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return false; 146c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 147c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 148c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosierstatic bool quoteNextArg(const char *flag) { 149c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier return llvm::StringSwitch<bool>(flag) 150c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Case("-D", true) 151c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier .Default(false); 152c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 153c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 154c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosiervoid Compilation::PrintDiagnosticJob(raw_ostream &OS, const Job &J) const { 155c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (const Command *C = dyn_cast<Command>(&J)) { 156c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << C->getExecutable(); 157c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier unsigned QuoteNextArg = 0; 158c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (ArgStringList::const_iterator it = C->getArguments().begin(), 159c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier ie = C->getArguments().end(); it != ie; ++it) { 160c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 161c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier bool SkipNext; 162c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (skipArg(*it, SkipNext)) { 163c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (SkipNext) ++it; 164c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier continue; 165c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 166c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 167c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (!QuoteNextArg) 168c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier QuoteNextArg = quoteNextArg(*it) ? 2 : 0; 169c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 170c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << ' '; 171c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 172c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg == 1) 173c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 174c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 175c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (!std::strpbrk(*it, " \"\\$")) { 176c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << *it; 177c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } else { 178c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // Quote the argument and escape shell special characters; this isn't 179c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier // really complete but is good enough. 180c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 181c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (const char *s = *it; *s; ++s) { 182c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (*s == '"' || *s == '\\' || *s == '$') 183c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '\\'; 184c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << *s; 185c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 186c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 187c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 188c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 189c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg) { 190c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier if (QuoteNextArg == 1) 191c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '"'; 192c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier --QuoteNextArg; 193c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 194c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 195c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier OS << '\n'; 196c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } else { 197c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier const JobList *Jobs = cast<JobList>(&J); 198c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier for (JobList::const_iterator 199c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier it = Jobs->begin(), ie = Jobs->end(); it != ie; ++it) 200c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier PrintDiagnosticJob(OS, **it); 201c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier } 202c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier} 203c91b41a64ccce6d1097d85c8c7354ea63a5566a0Chad Rosier 2049d718635fa805674aaba5d938f3dc6b35b8632baChad Rosierbool Compilation::CleanupFile(const char *File, bool IssueErrors) const { 205de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola std::string P(File); 206de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola 207de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // FIXME: Why are we trying to remove files that we have not created? For 208de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // example we should only try to remove a temporary assembly file if 209de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // "clang -cc1" succeed in writing it. Was this a workaround for when 210de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // clang was writing directly to a .s file and sometimes leaving it behind 211de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // during a failure? 212de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola 213de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // FIXME: If this is necessary, we can still try to split 214de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // llvm::sys::fs::remove into a removeFile and a removeDir and avoid the 215de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // duplicated stat from is_regular_file. 2169d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2179d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // Don't try to remove files which we don't have write access to (but may be 2189d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // able to remove), or non-regular files. Underlying tools may have 2199d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // intentionally not overwritten them. 220de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola if (!llvm::sys::fs::can_write(File) || !llvm::sys::fs::is_regular_file(File)) 2219d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return true; 2229d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 223de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola if (llvm::error_code EC = llvm::sys::fs::remove(File)) { 224de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // Failure is only failure if the file exists and is "regular". We checked 225de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // for it being regular before, and llvm::sys::fs::remove ignores ENOENT, 226de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola // so we don't need to check again. 2279d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 228de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola if (IssueErrors) 229de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola getDriver().Diag(clang::diag::err_drv_unable_to_remove_file) 230de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola << EC.message(); 231de2b523b9a2a32ff27e0689413c078c2cf87e666Rafael Espindola return false; 2329d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier } 2339d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return true; 2349d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier} 2359d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier 2361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpbool Compilation::CleanupFileList(const ArgStringList &Files, 237e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar bool IssueErrors) const { 238e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar bool Success = true; 2391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump for (ArgStringList::const_iterator 2409d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier it = Files.begin(), ie = Files.end(); it != ie; ++it) 2419d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier Success &= CleanupFile(*it, IssueErrors); 2429d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier return Success; 2439d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier} 24456eec2bc2188574459a514f3650d581e00683126Edward O'Callaghan 2459d718635fa805674aaba5d938f3dc6b35b8632baChad Rosierbool Compilation::CleanupFileMap(const ArgStringMap &Files, 2469d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier const JobAction *JA, 2479d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier bool IssueErrors) const { 2489d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier bool Success = true; 2499d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier for (ArgStringMap::const_iterator 2509d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier it = Files.begin(), ie = Files.end(); it != ie; ++it) { 251e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar 2529d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // If specified, only delete the files associated with the JobAction. 2539d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier // Otherwise, delete all files in the map. 2549d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier if (JA && it->first != JA) 2558ac127a09ab36f84860c176fe6b199c4973be984Daniel Dunbar continue; 2569d718635fa805674aaba5d938f3dc6b35b8632baChad Rosier Success &= CleanupFile(it->second, IssueErrors); 257e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar } 258e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar return Success; 259e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar} 260e530ad407af4a8904377592bfdb236acd320c6c2Daniel Dunbar 26131c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbarint Compilation::ExecuteCommand(const Command &C, 26231c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar const Command *&FailingCommand) const { 2638ce9054eb9051008b6a9850714701099d2a016b1Rafael Espindola std::string Prog(C.getExecutable()); 264ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar const char **Argv = new const char*[C.getArguments().size() + 2]; 265ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar Argv[0] = C.getExecutable(); 266ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar std::copy(C.getArguments().begin(), C.getArguments().end(), Argv+1); 267ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar Argv[C.getArguments().size() + 1] = 0; 2681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2697d3240d3c949b8d0697a23a202cceba4d1a2d21fRafael Espindola if ((getDriver().CCPrintOptions || 2702b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier getArgs().hasArg(options::OPT_v)) && !getDriver().CCGenDiagnostics) { 2715f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner raw_ostream *OS = &llvm::errs(); 2724c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2734c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar // Follow gcc implementation of CC_PRINT_OPTIONS; we could also cache the 2744c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar // output stream. 2754c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (getDriver().CCPrintOptions && getDriver().CCPrintOptionsFilename) { 2764c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar std::string Error; 277d965f95daa97097c8ddc5e1165ceae585a888719Rafael Espindola OS = new llvm::raw_fd_ostream(getDriver().CCPrintOptionsFilename, Error, 278d965f95daa97097c8ddc5e1165ceae585a888719Rafael Espindola llvm::sys::fs::F_Append); 2794c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (!Error.empty()) { 2804c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar getDriver().Diag(clang::diag::err_drv_cc_print_options_failure) 2814c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar << Error; 2824c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar FailingCommand = &C; 2834c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar delete OS; 2844c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar return 1; 2854c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2864c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2874c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2884c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (getDriver().CCPrintOptions) 2894c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar *OS << "[Logging clang options]"; 2904c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2914c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar PrintJob(*OS, C, "\n", /*Quote=*/getDriver().CCPrintOptions); 2924c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar 2934c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar if (OS != &llvm::errs()) 2944c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar delete OS; 2954c00fcdf98d3d7c4cb47b64f8b770f8f4bff1357Daniel Dunbar } 2961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 297ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar std::string Error; 298c48d575f2e41bec767ccdea6591404c907f1bc91Chad Rosier bool ExecutionFailed; 2998ce9054eb9051008b6a9850714701099d2a016b1Rafael Espindola int Res = llvm::sys::ExecuteAndWait(Prog, Argv, /*env*/ 0, Redirects, 300a603569515eea06e54e6e041b1c690d33086f375Rafael Espindola /*secondsToWait*/ 0, /*memoryLimit*/ 0, 301a603569515eea06e54e6e041b1c690d33086f375Rafael Espindola &Error, &ExecutionFailed); 302ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar if (!Error.empty()) { 303ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar assert(Res && "Error string set with 0 result code!"); 304ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar getDriver().Diag(clang::diag::err_drv_command_failure) << Error; 305ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar } 3061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 30731c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar if (Res) 30831c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar FailingCommand = &C; 30931c11ebdaab0acfa10e231730ae95e32c0e39e1fDaniel Dunbar 310ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar delete[] Argv; 311c48d575f2e41bec767ccdea6591404c907f1bc91Chad Rosier return ExecutionFailed ? 1 : Res; 312ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar} 313ceafbc8f55e00345a85d5e6674d3339a45cbbf76Daniel Dunbar 3144c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosiertypedef SmallVectorImpl< std::pair<int, const Command *> > FailingCommandList; 3154c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3164c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosierstatic bool ActionFailed(const Action *A, 3174c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier const FailingCommandList &FailingCommands) { 3184c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3194c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (FailingCommands.empty()) 3204c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return false; 3214c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3224c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier for (FailingCommandList::const_iterator CI = FailingCommands.begin(), 3234c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier CE = FailingCommands.end(); CI != CE; ++CI) 3244c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (A == &(CI->second->getSource())) 3254c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return true; 3264c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3274c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier for (Action::const_iterator AI = A->begin(), AE = A->end(); AI != AE; ++AI) 3284c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (ActionFailed(*AI, FailingCommands)) 3294c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return true; 3304c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3314c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return false; 3324c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier} 3334c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 3344c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosierstatic bool InputsOk(const Command &C, 3354c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier const FailingCommandList &FailingCommands) { 3364c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return !ActionFailed(&C.getSource(), FailingCommands); 3374c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier} 3384c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier 339a16355c31878403443f99077cc8df8318457faf5Chad Rosiervoid Compilation::ExecuteJob(const Job &J, 3404c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier FailingCommandList &FailingCommands) const { 34149b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar if (const Command *C = dyn_cast<Command>(&J)) { 3424c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier if (!InputsOk(*C, FailingCommands)) 3434c4df4520df20ec17d84aa2261f098ec278e1dabChad Rosier return; 344a16355c31878403443f99077cc8df8318457faf5Chad Rosier const Command *FailingCommand = 0; 345a16355c31878403443f99077cc8df8318457faf5Chad Rosier if (int Res = ExecuteCommand(*C, FailingCommand)) 346a16355c31878403443f99077cc8df8318457faf5Chad Rosier FailingCommands.push_back(std::make_pair(Res, FailingCommand)); 34749b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } else { 34849b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar const JobList *Jobs = cast<JobList>(&J); 349a16355c31878403443f99077cc8df8318457faf5Chad Rosier for (JobList::const_iterator it = Jobs->begin(), ie = Jobs->end(); 350a16355c31878403443f99077cc8df8318457faf5Chad Rosier it != ie; ++it) 351a16355c31878403443f99077cc8df8318457faf5Chad Rosier ExecuteJob(**it, FailingCommands); 35249b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar } 35349b98e700e300b8a61b2f7fbb0bb22264e8ec27aDaniel Dunbar} 3542b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 355c4a77906c259cba58c147d8468c406a430ecdcbbDmitri Gribenkovoid Compilation::initCompilationForDiagnostics() { 3562b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Free actions and jobs. 3572b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier DeleteContainerPointers(Actions); 3582b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier Jobs.clear(); 3592b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3602b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Clear temporary/results file lists. 3612b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TempFiles.clear(); 3622b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier ResultFiles.clear(); 3638425a5413b437a0b6ac04c475e7cf54cc9977880Chad Rosier FailureResultFiles.clear(); 3642b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3652b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Remove any user specified output. Claim any unclaimed arguments, so as 3662b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // to avoid emitting warnings about unused args. 36754db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne OptSpecifier OutputOpts[] = { options::OPT_o, options::OPT_MD, 36854db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne options::OPT_MMD }; 369f9e156c774e08409f235f0e9c9e3133ec796e877Chad Rosier for (unsigned i = 0, e = llvm::array_lengthof(OutputOpts); i != e; ++i) { 37054db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne if (TranslatedArgs->hasArg(OutputOpts[i])) 37154db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne TranslatedArgs->eraseArg(OutputOpts[i]); 37254db68bf0e389b8aabe1a434f825a81a9c1f7db8Peter Collingbourne } 3732b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier TranslatedArgs->ClaimAllArgs(); 3742b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier 3752b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier // Redirect stdout/stderr to /dev/null. 37657a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects = new const StringRef*[3](); 37757a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[0] = 0; 37857a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[1] = new const StringRef(); 37957a3bbfa9a97f174b0c80b0309e32f4329c4ae1bRafael Espindola Redirects[2] = new const StringRef(); 3802b81910618f63e4ce2373c926a26e76b4b91373fChad Rosier} 3814762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop 382c4a77906c259cba58c147d8468c406a430ecdcbbDmitri GribenkoStringRef Compilation::getSysRoot() const { 3834762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop return getDriver().SysRoot; 3844762a2da74875d2ae94e0d77d38ed964816cce36Sebastian Pop} 385