clang  12.0.0git
Compilation.cpp
Go to the documentation of this file.
1 //===- Compilation.cpp - Compilation Task Implementation ------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
10 #include "clang/Basic/LLVM.h"
11 #include "clang/Driver/Action.h"
12 #include "clang/Driver/Driver.h"
14 #include "clang/Driver/Job.h"
15 #include "clang/Driver/Options.h"
16 #include "clang/Driver/ToolChain.h"
17 #include "clang/Driver/Util.h"
18 #include "llvm/ADT/None.h"
19 #include "llvm/ADT/STLExtras.h"
20 #include "llvm/ADT/SmallVector.h"
21 #include "llvm/ADT/Triple.h"
22 #include "llvm/Option/ArgList.h"
23 #include "llvm/Option/OptSpecifier.h"
24 #include "llvm/Option/Option.h"
25 #include "llvm/Support/FileSystem.h"
26 #include "llvm/Support/raw_ostream.h"
27 #include <cassert>
28 #include <string>
29 #include <system_error>
30 #include <utility>
31 
32 using namespace clang;
33 using namespace driver;
34 using namespace llvm::opt;
35 
36 Compilation::Compilation(const Driver &D, const ToolChain &_DefaultToolChain,
37  InputArgList *_Args, DerivedArgList *_TranslatedArgs,
38  bool ContainsError)
39  : TheDriver(D), DefaultToolChain(_DefaultToolChain), Args(_Args),
40  TranslatedArgs(_TranslatedArgs), ContainsError(ContainsError) {
41  // The offloading host toolchain is the default toolchain.
42  OrderedOffloadingToolchains.insert(
43  std::make_pair(Action::OFK_Host, &DefaultToolChain));
44 }
45 
47  // Remove temporary files. This must be done before arguments are freed, as
48  // the file names might be derived from the input arguments.
49  if (!TheDriver.isSaveTempsEnabled() && !ForceKeepTempFiles)
50  CleanupFileList(TempFiles);
51 
52  delete TranslatedArgs;
53  delete Args;
54 
55  // Free any derived arg lists.
56  for (auto Arg : TCArgs)
57  if (Arg.second != TranslatedArgs)
58  delete Arg.second;
59 }
60 
61 const DerivedArgList &
62 Compilation::getArgsForToolChain(const ToolChain *TC, StringRef BoundArch,
63  Action::OffloadKind DeviceOffloadKind) {
64  if (!TC)
65  TC = &DefaultToolChain;
66 
67  DerivedArgList *&Entry = TCArgs[{TC, BoundArch, DeviceOffloadKind}];
68  if (!Entry) {
69  SmallVector<Arg *, 4> AllocatedArgs;
70  DerivedArgList *OpenMPArgs = nullptr;
71  // Translate OpenMP toolchain arguments provided via the -Xopenmp-target flags.
72  if (DeviceOffloadKind == Action::OFK_OpenMP) {
73  const ToolChain *HostTC = getSingleOffloadToolChain<Action::OFK_Host>();
74  bool SameTripleAsHost = (TC->getTriple() == HostTC->getTriple());
75  OpenMPArgs = TC->TranslateOpenMPTargetArgs(
76  *TranslatedArgs, SameTripleAsHost, AllocatedArgs);
77  }
78 
79  DerivedArgList *NewDAL = nullptr;
80  if (!OpenMPArgs) {
81  NewDAL = TC->TranslateXarchArgs(*TranslatedArgs, BoundArch,
82  DeviceOffloadKind, &AllocatedArgs);
83  } else {
84  NewDAL = TC->TranslateXarchArgs(*OpenMPArgs, BoundArch, DeviceOffloadKind,
85  &AllocatedArgs);
86  if (!NewDAL)
87  NewDAL = OpenMPArgs;
88  else
89  delete OpenMPArgs;
90  }
91 
92  if (!NewDAL) {
93  Entry = TC->TranslateArgs(*TranslatedArgs, BoundArch, DeviceOffloadKind);
94  if (!Entry)
95  Entry = TranslatedArgs;
96  } else {
97  Entry = TC->TranslateArgs(*NewDAL, BoundArch, DeviceOffloadKind);
98  if (!Entry)
99  Entry = NewDAL;
100  else
101  delete NewDAL;
102  }
103 
104  // Add allocated arguments to the final DAL.
105  for (auto ArgPtr : AllocatedArgs)
106  Entry->AddSynthesizedArg(ArgPtr);
107  }
108 
109  return *Entry;
110 }
111 
112 bool Compilation::CleanupFile(const char *File, bool IssueErrors) const {
113  // FIXME: Why are we trying to remove files that we have not created? For
114  // example we should only try to remove a temporary assembly file if
115  // "clang -cc1" succeed in writing it. Was this a workaround for when
116  // clang was writing directly to a .s file and sometimes leaving it behind
117  // during a failure?
118 
119  // FIXME: If this is necessary, we can still try to split
120  // llvm::sys::fs::remove into a removeFile and a removeDir and avoid the
121  // duplicated stat from is_regular_file.
122 
123  // Don't try to remove files which we don't have write access to (but may be
124  // able to remove), or non-regular files. Underlying tools may have
125  // intentionally not overwritten them.
126  if (!llvm::sys::fs::can_write(File) || !llvm::sys::fs::is_regular_file(File))
127  return true;
128 
129  if (std::error_code EC = llvm::sys::fs::remove(File)) {
130  // Failure is only failure if the file exists and is "regular". We checked
131  // for it being regular before, and llvm::sys::fs::remove ignores ENOENT,
132  // so we don't need to check again.
133 
134  if (IssueErrors)
135  getDriver().Diag(diag::err_drv_unable_to_remove_file)
136  << EC.message();
137  return false;
138  }
139  return true;
140 }
141 
142 bool Compilation::CleanupFileList(const llvm::opt::ArgStringList &Files,
143  bool IssueErrors) const {
144  bool Success = true;
145  for (const auto &File: Files)
146  Success &= CleanupFile(File, IssueErrors);
147  return Success;
148 }
149 
151  const JobAction *JA,
152  bool IssueErrors) const {
153  bool Success = true;
154  for (const auto &File : Files) {
155  // If specified, only delete the files associated with the JobAction.
156  // Otherwise, delete all files in the map.
157  if (JA && File.first != JA)
158  continue;
159  Success &= CleanupFile(File.second, IssueErrors);
160  }
161  return Success;
162 }
163 
165  const Command *&FailingCommand) const {
166  if ((getDriver().CCPrintOptions ||
167  getArgs().hasArg(options::OPT_v)) && !getDriver().CCGenDiagnostics) {
168  raw_ostream *OS = &llvm::errs();
169  std::unique_ptr<llvm::raw_fd_ostream> OwnedStream;
170 
171  // Follow gcc implementation of CC_PRINT_OPTIONS; we could also cache the
172  // output stream.
173  if (getDriver().CCPrintOptions && getDriver().CCPrintOptionsFilename) {
174  std::error_code EC;
175  OwnedStream.reset(new llvm::raw_fd_ostream(
176  getDriver().CCPrintOptionsFilename, EC,
177  llvm::sys::fs::OF_Append | llvm::sys::fs::OF_Text));
178  if (EC) {
179  getDriver().Diag(diag::err_drv_cc_print_options_failure)
180  << EC.message();
181  FailingCommand = &C;
182  return 1;
183  }
184  OS = OwnedStream.get();
185  }
186 
187  if (getDriver().CCPrintOptions)
188  *OS << "[Logging clang options]\n";
189 
190  C.Print(*OS, "\n", /*Quote=*/getDriver().CCPrintOptions);
191  }
192 
193  std::string Error;
194  bool ExecutionFailed;
195  int Res = C.Execute(Redirects, &Error, &ExecutionFailed);
196  if (PostCallback)
197  PostCallback(C, Res);
198  if (!Error.empty()) {
199  assert(Res && "Error string set with 0 result code!");
200  getDriver().Diag(diag::err_drv_command_failure) << Error;
201  }
202 
203  if (Res)
204  FailingCommand = &C;
205 
206  return ExecutionFailed ? 1 : Res;
207 }
208 
209 using FailingCommandList = SmallVectorImpl<std::pair<int, const Command *>>;
210 
211 static bool ActionFailed(const Action *A,
212  const FailingCommandList &FailingCommands) {
213  if (FailingCommands.empty())
214  return false;
215 
216  // CUDA/HIP can have the same input source code compiled multiple times so do
217  // not compiled again if there are already failures. It is OK to abort the
218  // CUDA pipeline on errors.
220  return true;
221 
222  for (const auto &CI : FailingCommands)
223  if (A == &(CI.second->getSource()))
224  return true;
225 
226  for (const auto *AI : A->inputs())
227  if (ActionFailed(AI, FailingCommands))
228  return true;
229 
230  return false;
231 }
232 
233 static bool InputsOk(const Command &C,
234  const FailingCommandList &FailingCommands) {
235  return !ActionFailed(&C.getSource(), FailingCommands);
236 }
237 
239  FailingCommandList &FailingCommands) const {
240  // According to UNIX standard, driver need to continue compiling all the
241  // inputs on the command line even one of them failed.
242  // In all but CLMode, execute all the jobs unless the necessary inputs for the
243  // job is missing due to previous failures.
244  for (const auto &Job : Jobs) {
245  if (!InputsOk(Job, FailingCommands))
246  continue;
247  const Command *FailingCommand = nullptr;
248  if (int Res = ExecuteCommand(Job, FailingCommand)) {
249  FailingCommands.push_back(std::make_pair(Res, FailingCommand));
250  // Bail as soon as one command fails in cl driver mode.
251  if (TheDriver.IsCLMode())
252  return;
253  }
254  }
255 }
256 
258  ForDiagnostics = true;
259 
260  // Free actions and jobs.
261  Actions.clear();
262  AllActions.clear();
263  Jobs.clear();
264 
265  // Remove temporary files.
266  if (!TheDriver.isSaveTempsEnabled() && !ForceKeepTempFiles)
267  CleanupFileList(TempFiles);
268 
269  // Clear temporary/results file lists.
270  TempFiles.clear();
271  ResultFiles.clear();
272  FailureResultFiles.clear();
273 
274  // Remove any user specified output. Claim any unclaimed arguments, so as
275  // to avoid emitting warnings about unused args.
276  OptSpecifier OutputOpts[] = {
277  options::OPT_o, options::OPT_MD, options::OPT_MMD, options::OPT_M,
278  options::OPT_MM, options::OPT_MF, options::OPT_MG, options::OPT_MJ,
279  options::OPT_MQ, options::OPT_MT, options::OPT_MV};
280  for (unsigned i = 0, e = llvm::array_lengthof(OutputOpts); i != e; ++i) {
281  if (TranslatedArgs->hasArg(OutputOpts[i]))
282  TranslatedArgs->eraseArg(OutputOpts[i]);
283  }
284  TranslatedArgs->ClaimAllArgs();
285 
286  // Force re-creation of the toolchain Args, otherwise our modifications just
287  // above will have no effect.
288  for (auto Arg : TCArgs)
289  if (Arg.second != TranslatedArgs)
290  delete Arg.second;
291  TCArgs.clear();
292 
293  // Redirect stdout/stderr to /dev/null.
294  Redirects = {None, {""}, {""}};
295 
296  // Temporary files added by diagnostics should be kept.
297  ForceKeepTempFiles = true;
298 }
299 
300 StringRef Compilation::getSysRoot() const {
301  return getDriver().SysRoot;
302 }
303 
305  this->Redirects = Redirects;
306 }
StringRef getSysRoot() const
Returns the sysroot path.
input_range inputs()
Definition: Action.h:152
DiagnosticBuilder Diag(unsigned DiagID) const
Definition: Driver.h:109
static bool InputsOk(const Command &C, const FailingCommandList &FailingCommands)
const llvm::opt::DerivedArgList & getArgsForToolChain(const ToolChain *TC, StringRef BoundArch, Action::OffloadKind DeviceOffloadKind)
getArgsForToolChain - Return the derived argument list for the tool chain TC (or the default tool cha...
Definition: Compilation.cpp:62
bool CleanupFileList(const llvm::opt::ArgStringList &Files, bool IssueErrors=false) const
CleanupFileList - Remove the files in the given list.
virtual void TranslateXarchArgs(const llvm::opt::DerivedArgList &Args, llvm::opt::Arg *&A, llvm::opt::DerivedArgList *DAL, SmallVectorImpl< llvm::opt::Arg * > *AllocatedArgs=nullptr) const
Append the argument following A to DAL assuming A is an Xarch argument.
Definition: ToolChain.cpp:1224
bool isOffloading(OffloadKind OKind) const
Definition: Action.h:210
Action - Represent an abstract compilation step to perform.
Definition: Action.h:47
Forward-declares and imports various common LLVM datatypes that clang wants to use unqualified.
llvm::Error Error
Driver - Encapsulate logic for constructing compilation processes from a set of gcc-driver-like comma...
Definition: Driver.h:59
bool IsCLMode() const
Whether the driver should follow cl.exe like behavior.
Definition: Driver.h:182
JobList - A sequence of jobs to perform.
Definition: Job.h:282
ASTEdit remove(RangeSelector S)
Removes the source selected by S.
llvm::DenseMap< const JobAction *, const char * > ArgStringMap
ArgStringMap - Type used to map a JobAction to its result file.
Definition: Util.h:20
bool CleanupFile(const char *File, bool IssueErrors=false) const
CleanupFile - Delete a given file.
void clear()
Clear the job list.
Definition: Job.cpp:491
const llvm::opt::DerivedArgList & getArgs() const
Definition: Compilation.h:191
void Redirect(ArrayRef< Optional< StringRef >> Redirects)
Redirect - Redirect output of this compilation.
Command - An executable path/name and argument vector to execute.
Definition: Job.h:105
bool isSaveTempsEnabled() const
Definition: Driver.h:345
int ExecuteCommand(const Command &C, const Command *&FailingCommand) const
ExecuteCommand - Execute an actual command.
Compilation(const Driver &D, const ToolChain &DefaultToolChain, llvm::opt::InputArgList *Args, llvm::opt::DerivedArgList *TranslatedArgs, bool ContainsError)
Definition: Compilation.cpp:36
virtual llvm::opt::DerivedArgList * TranslateOpenMPTargetArgs(const llvm::opt::DerivedArgList &Args, bool SameTripleAsHost, SmallVectorImpl< llvm::opt::Arg * > &AllocatedArgs) const
TranslateOpenMPTargetArgs - Create a new derived argument list for that contains the OpenMP target sp...
Definition: ToolChain.cpp:1154
Dataflow Directional Tag Classes.
static bool ActionFailed(const Action *A, const FailingCommandList &FailingCommands)
std::string SysRoot
sysroot, if present
Definition: Driver.h:148
const Driver & getDriver() const
Definition: Compilation.h:138
const llvm::Triple & getTriple() const
Definition: ToolChain.h:206
bool CleanupFileMap(const ArgStringMap &Files, const JobAction *JA, bool IssueErrors=false) const
CleanupFileMap - Remove the files in the given map.
SmallVectorImpl< std::pair< int, const Command * > > FailingCommandList
void ExecuteJobs(const JobList &Jobs, SmallVectorImpl< std::pair< int, const Command * >> &FailingCommands) const
ExecuteJob - Execute a single job.
virtual llvm::opt::DerivedArgList * TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch, Action::OffloadKind DeviceOffloadKind) const
TranslateArgs - Create a new derived argument list for any argument translations this ToolChain may w...
Definition: ToolChain.h:289
void initCompilationForDiagnostics()
initCompilationForDiagnostics - Remove stale state and suppress output so compilation can be reexecut...
ToolChain - Access to tools for a single platform.
Definition: ToolChain.h:90