1 //===--- Driver.cpp - Clang GCC Compatible Driver -------------------------===//
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 
9 #include "clang/Driver/Driver.h"
10 #include "InputInfo.h"
11 #include "ToolChains/AIX.h"
12 #include "ToolChains/AMDGPU.h"
13 #include "ToolChains/AVR.h"
14 #include "ToolChains/Ananas.h"
15 #include "ToolChains/BareMetal.h"
16 #include "ToolChains/Clang.h"
17 #include "ToolChains/CloudABI.h"
18 #include "ToolChains/Contiki.h"
19 #include "ToolChains/CrossWindows.h"
20 #include "ToolChains/Cuda.h"
21 #include "ToolChains/Darwin.h"
22 #include "ToolChains/DragonFly.h"
23 #include "ToolChains/FreeBSD.h"
24 #include "ToolChains/Fuchsia.h"
25 #include "ToolChains/Gnu.h"
26 #include "ToolChains/HIP.h"
27 #include "ToolChains/Haiku.h"
28 #include "ToolChains/Hexagon.h"
29 #include "ToolChains/Hurd.h"
30 #include "ToolChains/Lanai.h"
31 #include "ToolChains/Linux.h"
32 #include "ToolChains/MSP430.h"
33 #include "ToolChains/MSVC.h"
34 #include "ToolChains/MinGW.h"
35 #include "ToolChains/Minix.h"
36 #include "ToolChains/MipsLinux.h"
37 #include "ToolChains/Myriad.h"
38 #include "ToolChains/NaCl.h"
39 #include "ToolChains/NetBSD.h"
40 #include "ToolChains/OpenBSD.h"
41 #include "ToolChains/PPCLinux.h"
42 #include "ToolChains/PS4CPU.h"
43 #include "ToolChains/RISCVToolchain.h"
44 #include "ToolChains/Solaris.h"
45 #include "ToolChains/TCE.h"
46 #include "ToolChains/WebAssembly.h"
47 #include "ToolChains/XCore.h"
48 #include "clang/Basic/Version.h"
49 #include "clang/Config/config.h"
50 #include "clang/Driver/Action.h"
51 #include "clang/Driver/Compilation.h"
52 #include "clang/Driver/DriverDiagnostic.h"
53 #include "clang/Driver/Job.h"
54 #include "clang/Driver/Options.h"
55 #include "clang/Driver/SanitizerArgs.h"
56 #include "clang/Driver/Tool.h"
57 #include "clang/Driver/ToolChain.h"
58 #include "llvm/ADT/ArrayRef.h"
59 #include "llvm/ADT/STLExtras.h"
60 #include "llvm/ADT/SmallSet.h"
61 #include "llvm/ADT/StringExtras.h"
62 #include "llvm/ADT/StringSet.h"
63 #include "llvm/ADT/StringSwitch.h"
64 #include "llvm/Config/llvm-config.h"
65 #include "llvm/Option/Arg.h"
66 #include "llvm/Option/ArgList.h"
67 #include "llvm/Option/OptSpecifier.h"
68 #include "llvm/Option/OptTable.h"
69 #include "llvm/Option/Option.h"
70 #include "llvm/Support/CommandLine.h"
71 #include "llvm/Support/ErrorHandling.h"
72 #include "llvm/Support/FileSystem.h"
73 #include "llvm/Support/FormatVariadic.h"
74 #include "llvm/Support/Host.h"
75 #include "llvm/Support/Path.h"
76 #include "llvm/Support/PrettyStackTrace.h"
77 #include "llvm/Support/Process.h"
78 #include "llvm/Support/Program.h"
79 #include "llvm/Support/StringSaver.h"
80 #include "llvm/Support/TargetRegistry.h"
81 #include "llvm/Support/VirtualFileSystem.h"
82 #include "llvm/Support/raw_ostream.h"
83 #include <map>
84 #include <memory>
85 #include <utility>
86 #if LLVM_ON_UNIX
87 #include <unistd.h> // getpid
88 #include <sysexits.h> // EX_IOERR
89 #endif
90 
91 using namespace clang::driver;
92 using namespace clang;
93 using namespace llvm::opt;
94 
95 // static
96 std::string Driver::GetResourcesPath(StringRef BinaryPath,
97                                      StringRef CustomResourceDir) {
98   // Since the resource directory is embedded in the module hash, it's important
99   // that all places that need it call this function, so that they get the
100   // exact same string ("a/../b/" and "b/" get different hashes, for example).
101 
102   // Dir is bin/ or lib/, depending on where BinaryPath is.
103   std::string Dir = std::string(llvm::sys::path::parent_path(BinaryPath));
104 
105   SmallString<128> P(Dir);
106   if (CustomResourceDir != "") {
107     llvm::sys::path::append(P, CustomResourceDir);
108   } else {
109     // On Windows, libclang.dll is in bin/.
110     // On non-Windows, libclang.so/.dylib is in lib/.
111     // With a static-library build of libclang, LibClangPath will contain the
112     // path of the embedding binary, which for LLVM binaries will be in bin/.
113     // ../lib gets us to lib/ in both cases.
114     P = llvm::sys::path::parent_path(Dir);
115     llvm::sys::path::append(P, Twine("lib") + CLANG_LIBDIR_SUFFIX, "clang",
116                             CLANG_VERSION_STRING);
117   }
118 
119   return std::string(P.str());
120 }
121 
122 Driver::Driver(StringRef ClangExecutable, StringRef TargetTriple,
123                DiagnosticsEngine &Diags,
124                IntrusiveRefCntPtr<llvm::vfs::FileSystem> VFS)
125     : Diags(Diags), VFS(std::move(VFS)), Mode(GCCMode),
126       SaveTemps(SaveTempsNone), BitcodeEmbed(EmbedNone), LTOMode(LTOK_None),
127       ClangExecutable(ClangExecutable), SysRoot(DEFAULT_SYSROOT),
128       DriverTitle("clang LLVM compiler"), CCPrintOptionsFilename(nullptr),
129       CCPrintHeadersFilename(nullptr), CCLogDiagnosticsFilename(nullptr),
130       CCCPrintBindings(false), CCPrintOptions(false), CCPrintHeaders(false),
131       CCLogDiagnostics(false), CCGenDiagnostics(false),
132       TargetTriple(TargetTriple), CCCGenericGCCName(""), Saver(Alloc),
133       CheckInputsExist(true), GenReproducer(false),
134       SuppressMissingInputWarning(false) {
135   // Provide a sane fallback if no VFS is specified.
136   if (!this->VFS)
137     this->VFS = llvm::vfs::getRealFileSystem();
138 
139   Name = std::string(llvm::sys::path::filename(ClangExecutable));
140   Dir = std::string(llvm::sys::path::parent_path(ClangExecutable));
141   InstalledDir = Dir; // Provide a sensible default installed dir.
142 
143   if ((!SysRoot.empty()) && llvm::sys::path::is_relative(SysRoot)) {
144     // Prepend InstalledDir if SysRoot is relative
145     SmallString<128> P(InstalledDir);
146     llvm::sys::path::append(P, SysRoot);
147     SysRoot = std::string(P);
148   }
149 
150 #if defined(CLANG_CONFIG_FILE_SYSTEM_DIR)
151   SystemConfigDir = CLANG_CONFIG_FILE_SYSTEM_DIR;
152 #endif
153 #if defined(CLANG_CONFIG_FILE_USER_DIR)
154   UserConfigDir = CLANG_CONFIG_FILE_USER_DIR;
155 #endif
156 
157   // Compute the path to the resource directory.
158   ResourceDir = GetResourcesPath(ClangExecutable, CLANG_RESOURCE_DIR);
159 }
160 
161 void Driver::ParseDriverMode(StringRef ProgramName,
162                              ArrayRef<const char *> Args) {
163   if (ClangNameParts.isEmpty())
164     ClangNameParts = ToolChain::getTargetAndModeFromProgramName(ProgramName);
165   setDriverModeFromOption(ClangNameParts.DriverMode);
166 
167   for (const char *ArgPtr : Args) {
168     // Ignore nullptrs, they are the response file's EOL markers.
169     if (ArgPtr == nullptr)
170       continue;
171     const StringRef Arg = ArgPtr;
172     setDriverModeFromOption(Arg);
173   }
174 }
175 
176 void Driver::setDriverModeFromOption(StringRef Opt) {
177   const std::string OptName =
178       getOpts().getOption(options::OPT_driver_mode).getPrefixedName();
179   if (!Opt.startswith(OptName))
180     return;
181   StringRef Value = Opt.drop_front(OptName.size());
182 
183   if (auto M = llvm::StringSwitch<llvm::Optional<DriverMode>>(Value)
184                    .Case("gcc", GCCMode)
185                    .Case("g++", GXXMode)
186                    .Case("cpp", CPPMode)
187                    .Case("cl", CLMode)
188                    .Case("flang", FlangMode)
189                    .Default(None))
190     Mode = *M;
191   else
192     Diag(diag::err_drv_unsupported_option_argument) << OptName << Value;
193 }
194 
195 InputArgList Driver::ParseArgStrings(ArrayRef<const char *> ArgStrings,
196                                      bool IsClCompatMode,
197                                      bool &ContainsError) {
198   llvm::PrettyStackTraceString CrashInfo("Command line argument parsing");
199   ContainsError = false;
200 
201   unsigned IncludedFlagsBitmask;
202   unsigned ExcludedFlagsBitmask;
203   std::tie(IncludedFlagsBitmask, ExcludedFlagsBitmask) =
204       getIncludeExcludeOptionFlagMasks(IsClCompatMode);
205 
206   unsigned MissingArgIndex, MissingArgCount;
207   InputArgList Args =
208       getOpts().ParseArgs(ArgStrings, MissingArgIndex, MissingArgCount,
209                           IncludedFlagsBitmask, ExcludedFlagsBitmask);
210 
211   // Check for missing argument error.
212   if (MissingArgCount) {
213     Diag(diag::err_drv_missing_argument)
214         << Args.getArgString(MissingArgIndex) << MissingArgCount;
215     ContainsError |=
216         Diags.getDiagnosticLevel(diag::err_drv_missing_argument,
217                                  SourceLocation()) > DiagnosticsEngine::Warning;
218   }
219 
220   // Check for unsupported options.
221   for (const Arg *A : Args) {
222     if (A->getOption().hasFlag(options::Unsupported)) {
223       unsigned DiagID;
224       auto ArgString = A->getAsString(Args);
225       std::string Nearest;
226       if (getOpts().findNearest(
227             ArgString, Nearest, IncludedFlagsBitmask,
228             ExcludedFlagsBitmask | options::Unsupported) > 1) {
229         DiagID = diag::err_drv_unsupported_opt;
230         Diag(DiagID) << ArgString;
231       } else {
232         DiagID = diag::err_drv_unsupported_opt_with_suggestion;
233         Diag(DiagID) << ArgString << Nearest;
234       }
235       ContainsError |= Diags.getDiagnosticLevel(DiagID, SourceLocation()) >
236                        DiagnosticsEngine::Warning;
237       continue;
238     }
239 
240     // Warn about -mcpu= without an argument.
241     if (A->getOption().matches(options::OPT_mcpu_EQ) && A->containsValue("")) {
242       Diag(diag::warn_drv_empty_joined_argument) << A->getAsString(Args);
243       ContainsError |= Diags.getDiagnosticLevel(
244                            diag::warn_drv_empty_joined_argument,
245                            SourceLocation()) > DiagnosticsEngine::Warning;
246     }
247   }
248 
249   for (const Arg *A : Args.filtered(options::OPT_UNKNOWN)) {
250     unsigned DiagID;
251     auto ArgString = A->getAsString(Args);
252     std::string Nearest;
253     if (getOpts().findNearest(
254           ArgString, Nearest, IncludedFlagsBitmask, ExcludedFlagsBitmask) > 1) {
255       DiagID = IsCLMode() ? diag::warn_drv_unknown_argument_clang_cl
256                           : diag::err_drv_unknown_argument;
257       Diags.Report(DiagID) << ArgString;
258     } else {
259       DiagID = IsCLMode()
260                    ? diag::warn_drv_unknown_argument_clang_cl_with_suggestion
261                    : diag::err_drv_unknown_argument_with_suggestion;
262       Diags.Report(DiagID) << ArgString << Nearest;
263     }
264     ContainsError |= Diags.getDiagnosticLevel(DiagID, SourceLocation()) >
265                      DiagnosticsEngine::Warning;
266   }
267 
268   return Args;
269 }
270 
271 // Determine which compilation mode we are in. We look for options which
272 // affect the phase, starting with the earliest phases, and record which
273 // option we used to determine the final phase.
274 phases::ID Driver::getFinalPhase(const DerivedArgList &DAL,
275                                  Arg **FinalPhaseArg) const {
276   Arg *PhaseArg = nullptr;
277   phases::ID FinalPhase;
278 
279   // -{E,EP,P,M,MM} only run the preprocessor.
280   if (CCCIsCPP() || (PhaseArg = DAL.getLastArg(options::OPT_E)) ||
281       (PhaseArg = DAL.getLastArg(options::OPT__SLASH_EP)) ||
282       (PhaseArg = DAL.getLastArg(options::OPT_M, options::OPT_MM)) ||
283       (PhaseArg = DAL.getLastArg(options::OPT__SLASH_P))) {
284     FinalPhase = phases::Preprocess;
285 
286   // --precompile only runs up to precompilation.
287   } else if ((PhaseArg = DAL.getLastArg(options::OPT__precompile))) {
288     FinalPhase = phases::Precompile;
289 
290   // -{fsyntax-only,-analyze,emit-ast} only run up to the compiler.
291   } else if ((PhaseArg = DAL.getLastArg(options::OPT_fsyntax_only)) ||
292              (PhaseArg = DAL.getLastArg(options::OPT_print_supported_cpus)) ||
293              (PhaseArg = DAL.getLastArg(options::OPT_module_file_info)) ||
294              (PhaseArg = DAL.getLastArg(options::OPT_verify_pch)) ||
295              (PhaseArg = DAL.getLastArg(options::OPT_rewrite_objc)) ||
296              (PhaseArg = DAL.getLastArg(options::OPT_rewrite_legacy_objc)) ||
297              (PhaseArg = DAL.getLastArg(options::OPT__migrate)) ||
298              (PhaseArg = DAL.getLastArg(options::OPT__analyze)) ||
299              (PhaseArg = DAL.getLastArg(options::OPT_emit_ast))) {
300     FinalPhase = phases::Compile;
301 
302   // -S only runs up to the backend.
303   } else if ((PhaseArg = DAL.getLastArg(options::OPT_S))) {
304     FinalPhase = phases::Backend;
305 
306   // -c compilation only runs up to the assembler.
307   } else if ((PhaseArg = DAL.getLastArg(options::OPT_c))) {
308     FinalPhase = phases::Assemble;
309 
310   // Otherwise do everything.
311   } else
312     FinalPhase = phases::Link;
313 
314   if (FinalPhaseArg)
315     *FinalPhaseArg = PhaseArg;
316 
317   return FinalPhase;
318 }
319 
320 static Arg *MakeInputArg(DerivedArgList &Args, const OptTable &Opts,
321                          StringRef Value, bool Claim = true) {
322   Arg *A = new Arg(Opts.getOption(options::OPT_INPUT), Value,
323                    Args.getBaseArgs().MakeIndex(Value), Value.data());
324   Args.AddSynthesizedArg(A);
325   if (Claim)
326     A->claim();
327   return A;
328 }
329 
330 DerivedArgList *Driver::TranslateInputArgs(const InputArgList &Args) const {
331   const llvm::opt::OptTable &Opts = getOpts();
332   DerivedArgList *DAL = new DerivedArgList(Args);
333 
334   bool HasNostdlib = Args.hasArg(options::OPT_nostdlib);
335   bool HasNostdlibxx = Args.hasArg(options::OPT_nostdlibxx);
336   bool HasNodefaultlib = Args.hasArg(options::OPT_nodefaultlibs);
337   for (Arg *A : Args) {
338     // Unfortunately, we have to parse some forwarding options (-Xassembler,
339     // -Xlinker, -Xpreprocessor) because we either integrate their functionality
340     // (assembler and preprocessor), or bypass a previous driver ('collect2').
341 
342     // Rewrite linker options, to replace --no-demangle with a custom internal
343     // option.
344     if ((A->getOption().matches(options::OPT_Wl_COMMA) ||
345          A->getOption().matches(options::OPT_Xlinker)) &&
346         A->containsValue("--no-demangle")) {
347       // Add the rewritten no-demangle argument.
348       DAL->AddFlagArg(A, Opts.getOption(options::OPT_Z_Xlinker__no_demangle));
349 
350       // Add the remaining values as Xlinker arguments.
351       for (StringRef Val : A->getValues())
352         if (Val != "--no-demangle")
353           DAL->AddSeparateArg(A, Opts.getOption(options::OPT_Xlinker), Val);
354 
355       continue;
356     }
357 
358     // Rewrite preprocessor options, to replace -Wp,-MD,FOO which is used by
359     // some build systems. We don't try to be complete here because we don't
360     // care to encourage this usage model.
361     if (A->getOption().matches(options::OPT_Wp_COMMA) &&
362         (A->getValue(0) == StringRef("-MD") ||
363          A->getValue(0) == StringRef("-MMD"))) {
364       // Rewrite to -MD/-MMD along with -MF.
365       if (A->getValue(0) == StringRef("-MD"))
366         DAL->AddFlagArg(A, Opts.getOption(options::OPT_MD));
367       else
368         DAL->AddFlagArg(A, Opts.getOption(options::OPT_MMD));
369       if (A->getNumValues() == 2)
370         DAL->AddSeparateArg(A, Opts.getOption(options::OPT_MF), A->getValue(1));
371       continue;
372     }
373 
374     // Rewrite reserved library names.
375     if (A->getOption().matches(options::OPT_l)) {
376       StringRef Value = A->getValue();
377 
378       // Rewrite unless -nostdlib is present.
379       if (!HasNostdlib && !HasNodefaultlib && !HasNostdlibxx &&
380           Value == "stdc++") {
381         DAL->AddFlagArg(A, Opts.getOption(options::OPT_Z_reserved_lib_stdcxx));
382         continue;
383       }
384 
385       // Rewrite unconditionally.
386       if (Value == "cc_kext") {
387         DAL->AddFlagArg(A, Opts.getOption(options::OPT_Z_reserved_lib_cckext));
388         continue;
389       }
390     }
391 
392     // Pick up inputs via the -- option.
393     if (A->getOption().matches(options::OPT__DASH_DASH)) {
394       A->claim();
395       for (StringRef Val : A->getValues())
396         DAL->append(MakeInputArg(*DAL, Opts, Val, false));
397       continue;
398     }
399 
400     DAL->append(A);
401   }
402 
403   // Enforce -static if -miamcu is present.
404   if (Args.hasFlag(options::OPT_miamcu, options::OPT_mno_iamcu, false))
405     DAL->AddFlagArg(0, Opts.getOption(options::OPT_static));
406 
407 // Add a default value of -mlinker-version=, if one was given and the user
408 // didn't specify one.
409 #if defined(HOST_LINK_VERSION)
410   if (!Args.hasArg(options::OPT_mlinker_version_EQ) &&
411       strlen(HOST_LINK_VERSION) > 0) {
412     DAL->AddJoinedArg(0, Opts.getOption(options::OPT_mlinker_version_EQ),
413                       HOST_LINK_VERSION);
414     DAL->getLastArg(options::OPT_mlinker_version_EQ)->claim();
415   }
416 #endif
417 
418   return DAL;
419 }
420 
421 /// Compute target triple from args.
422 ///
423 /// This routine provides the logic to compute a target triple from various
424 /// args passed to the driver and the default triple string.
425 static llvm::Triple computeTargetTriple(const Driver &D,
426                                         StringRef TargetTriple,
427                                         const ArgList &Args,
428                                         StringRef DarwinArchName = "") {
429   // FIXME: Already done in Compilation *Driver::BuildCompilation
430   if (const Arg *A = Args.getLastArg(options::OPT_target))
431     TargetTriple = A->getValue();
432 
433   llvm::Triple Target(llvm::Triple::normalize(TargetTriple));
434 
435   // GNU/Hurd's triples should have been -hurd-gnu*, but were historically made
436   // -gnu* only, and we can not change this, so we have to detect that case as
437   // being the Hurd OS.
438   if (TargetTriple.find("-unknown-gnu") != StringRef::npos ||
439       TargetTriple.find("-pc-gnu") != StringRef::npos)
440     Target.setOSName("hurd");
441 
442   // Handle Apple-specific options available here.
443   if (Target.isOSBinFormatMachO()) {
444     // If an explicit Darwin arch name is given, that trumps all.
445     if (!DarwinArchName.empty()) {
446       tools::darwin::setTripleTypeForMachOArchName(Target, DarwinArchName);
447       return Target;
448     }
449 
450     // Handle the Darwin '-arch' flag.
451     if (Arg *A = Args.getLastArg(options::OPT_arch)) {
452       StringRef ArchName = A->getValue();
453       tools::darwin::setTripleTypeForMachOArchName(Target, ArchName);
454     }
455   }
456 
457   // Handle pseudo-target flags '-mlittle-endian'/'-EL' and
458   // '-mbig-endian'/'-EB'.
459   if (Arg *A = Args.getLastArg(options::OPT_mlittle_endian,
460                                options::OPT_mbig_endian)) {
461     if (A->getOption().matches(options::OPT_mlittle_endian)) {
462       llvm::Triple LE = Target.getLittleEndianArchVariant();
463       if (LE.getArch() != llvm::Triple::UnknownArch)
464         Target = std::move(LE);
465     } else {
466       llvm::Triple BE = Target.getBigEndianArchVariant();
467       if (BE.getArch() != llvm::Triple::UnknownArch)
468         Target = std::move(BE);
469     }
470   }
471 
472   // Skip further flag support on OSes which don't support '-m32' or '-m64'.
473   if (Target.getArch() == llvm::Triple::tce ||
474       Target.getOS() == llvm::Triple::Minix)
475     return Target;
476 
477   // Handle pseudo-target flags '-m64', '-mx32', '-m32' and '-m16'.
478   Arg *A = Args.getLastArg(options::OPT_m64, options::OPT_mx32,
479                            options::OPT_m32, options::OPT_m16);
480   if (A) {
481     llvm::Triple::ArchType AT = llvm::Triple::UnknownArch;
482 
483     if (A->getOption().matches(options::OPT_m64)) {
484       AT = Target.get64BitArchVariant().getArch();
485       if (Target.getEnvironment() == llvm::Triple::GNUX32)
486         Target.setEnvironment(llvm::Triple::GNU);
487     } else if (A->getOption().matches(options::OPT_mx32) &&
488                Target.get64BitArchVariant().getArch() == llvm::Triple::x86_64) {
489       AT = llvm::Triple::x86_64;
490       Target.setEnvironment(llvm::Triple::GNUX32);
491     } else if (A->getOption().matches(options::OPT_m32)) {
492       AT = Target.get32BitArchVariant().getArch();
493       if (Target.getEnvironment() == llvm::Triple::GNUX32)
494         Target.setEnvironment(llvm::Triple::GNU);
495     } else if (A->getOption().matches(options::OPT_m16) &&
496                Target.get32BitArchVariant().getArch() == llvm::Triple::x86) {
497       AT = llvm::Triple::x86;
498       Target.setEnvironment(llvm::Triple::CODE16);
499     }
500 
501     if (AT != llvm::Triple::UnknownArch && AT != Target.getArch())
502       Target.setArch(AT);
503   }
504 
505   // Handle -miamcu flag.
506   if (Args.hasFlag(options::OPT_miamcu, options::OPT_mno_iamcu, false)) {
507     if (Target.get32BitArchVariant().getArch() != llvm::Triple::x86)
508       D.Diag(diag::err_drv_unsupported_opt_for_target) << "-miamcu"
509                                                        << Target.str();
510 
511     if (A && !A->getOption().matches(options::OPT_m32))
512       D.Diag(diag::err_drv_argument_not_allowed_with)
513           << "-miamcu" << A->getBaseArg().getAsString(Args);
514 
515     Target.setArch(llvm::Triple::x86);
516     Target.setArchName("i586");
517     Target.setEnvironment(llvm::Triple::UnknownEnvironment);
518     Target.setEnvironmentName("");
519     Target.setOS(llvm::Triple::ELFIAMCU);
520     Target.setVendor(llvm::Triple::UnknownVendor);
521     Target.setVendorName("intel");
522   }
523 
524   // If target is MIPS adjust the target triple
525   // accordingly to provided ABI name.
526   A = Args.getLastArg(options::OPT_mabi_EQ);
527   if (A && Target.isMIPS()) {
528     StringRef ABIName = A->getValue();
529     if (ABIName == "32") {
530       Target = Target.get32BitArchVariant();
531       if (Target.getEnvironment() == llvm::Triple::GNUABI64 ||
532           Target.getEnvironment() == llvm::Triple::GNUABIN32)
533         Target.setEnvironment(llvm::Triple::GNU);
534     } else if (ABIName == "n32") {
535       Target = Target.get64BitArchVariant();
536       if (Target.getEnvironment() == llvm::Triple::GNU ||
537           Target.getEnvironment() == llvm::Triple::GNUABI64)
538         Target.setEnvironment(llvm::Triple::GNUABIN32);
539     } else if (ABIName == "64") {
540       Target = Target.get64BitArchVariant();
541       if (Target.getEnvironment() == llvm::Triple::GNU ||
542           Target.getEnvironment() == llvm::Triple::GNUABIN32)
543         Target.setEnvironment(llvm::Triple::GNUABI64);
544     }
545   }
546 
547   // If target is RISC-V adjust the target triple according to
548   // provided architecture name
549   A = Args.getLastArg(options::OPT_march_EQ);
550   if (A && Target.isRISCV()) {
551     StringRef ArchName = A->getValue();
552     if (ArchName.startswith_lower("rv32"))
553       Target.setArch(llvm::Triple::riscv32);
554     else if (ArchName.startswith_lower("rv64"))
555       Target.setArch(llvm::Triple::riscv64);
556   }
557 
558   return Target;
559 }
560 
561 // Parse the LTO options and record the type of LTO compilation
562 // based on which -f(no-)?lto(=.*)? option occurs last.
563 void Driver::setLTOMode(const llvm::opt::ArgList &Args) {
564   LTOMode = LTOK_None;
565   if (!Args.hasFlag(options::OPT_flto, options::OPT_flto_EQ,
566                     options::OPT_fno_lto, false))
567     return;
568 
569   StringRef LTOName("full");
570 
571   const Arg *A = Args.getLastArg(options::OPT_flto_EQ);
572   if (A)
573     LTOName = A->getValue();
574 
575   LTOMode = llvm::StringSwitch<LTOKind>(LTOName)
576                 .Case("full", LTOK_Full)
577                 .Case("thin", LTOK_Thin)
578                 .Default(LTOK_Unknown);
579 
580   if (LTOMode == LTOK_Unknown) {
581     assert(A);
582     Diag(diag::err_drv_unsupported_option_argument) << A->getOption().getName()
583                                                     << A->getValue();
584   }
585 }
586 
587 /// Compute the desired OpenMP runtime from the flags provided.
588 Driver::OpenMPRuntimeKind Driver::getOpenMPRuntime(const ArgList &Args) const {
589   StringRef RuntimeName(CLANG_DEFAULT_OPENMP_RUNTIME);
590 
591   const Arg *A = Args.getLastArg(options::OPT_fopenmp_EQ);
592   if (A)
593     RuntimeName = A->getValue();
594 
595   auto RT = llvm::StringSwitch<OpenMPRuntimeKind>(RuntimeName)
596                 .Case("libomp", OMPRT_OMP)
597                 .Case("libgomp", OMPRT_GOMP)
598                 .Case("libiomp5", OMPRT_IOMP5)
599                 .Default(OMPRT_Unknown);
600 
601   if (RT == OMPRT_Unknown) {
602     if (A)
603       Diag(diag::err_drv_unsupported_option_argument)
604           << A->getOption().getName() << A->getValue();
605     else
606       // FIXME: We could use a nicer diagnostic here.
607       Diag(diag::err_drv_unsupported_opt) << "-fopenmp";
608   }
609 
610   return RT;
611 }
612 
613 void Driver::CreateOffloadingDeviceToolChains(Compilation &C,
614                                               InputList &Inputs) {
615 
616   //
617   // CUDA/HIP
618   //
619   // We need to generate a CUDA/HIP toolchain if any of the inputs has a CUDA
620   // or HIP type. However, mixed CUDA/HIP compilation is not supported.
621   bool IsCuda =
622       llvm::any_of(Inputs, [](std::pair<types::ID, const llvm::opt::Arg *> &I) {
623         return types::isCuda(I.first);
624       });
625   bool IsHIP =
626       llvm::any_of(Inputs,
627                    [](std::pair<types::ID, const llvm::opt::Arg *> &I) {
628                      return types::isHIP(I.first);
629                    }) ||
630       C.getInputArgs().hasArg(options::OPT_hip_link);
631   if (IsCuda && IsHIP) {
632     Diag(clang::diag::err_drv_mix_cuda_hip);
633     return;
634   }
635   if (IsCuda) {
636     const ToolChain *HostTC = C.getSingleOffloadToolChain<Action::OFK_Host>();
637     const llvm::Triple &HostTriple = HostTC->getTriple();
638     StringRef DeviceTripleStr;
639     auto OFK = Action::OFK_Cuda;
640     DeviceTripleStr =
641         HostTriple.isArch64Bit() ? "nvptx64-nvidia-cuda" : "nvptx-nvidia-cuda";
642     llvm::Triple CudaTriple(DeviceTripleStr);
643     // Use the CUDA and host triples as the key into the ToolChains map,
644     // because the device toolchain we create depends on both.
645     auto &CudaTC = ToolChains[CudaTriple.str() + "/" + HostTriple.str()];
646     if (!CudaTC) {
647       CudaTC = std::make_unique<toolchains::CudaToolChain>(
648           *this, CudaTriple, *HostTC, C.getInputArgs(), OFK);
649     }
650     C.addOffloadDeviceToolChain(CudaTC.get(), OFK);
651   } else if (IsHIP) {
652     const ToolChain *HostTC = C.getSingleOffloadToolChain<Action::OFK_Host>();
653     const llvm::Triple &HostTriple = HostTC->getTriple();
654     StringRef DeviceTripleStr;
655     auto OFK = Action::OFK_HIP;
656     DeviceTripleStr = "amdgcn-amd-amdhsa";
657     llvm::Triple HIPTriple(DeviceTripleStr);
658     // Use the HIP and host triples as the key into the ToolChains map,
659     // because the device toolchain we create depends on both.
660     auto &HIPTC = ToolChains[HIPTriple.str() + "/" + HostTriple.str()];
661     if (!HIPTC) {
662       HIPTC = std::make_unique<toolchains::HIPToolChain>(
663           *this, HIPTriple, *HostTC, C.getInputArgs());
664     }
665     C.addOffloadDeviceToolChain(HIPTC.get(), OFK);
666   }
667 
668   //
669   // OpenMP
670   //
671   // We need to generate an OpenMP toolchain if the user specified targets with
672   // the -fopenmp-targets option.
673   if (Arg *OpenMPTargets =
674           C.getInputArgs().getLastArg(options::OPT_fopenmp_targets_EQ)) {
675     if (OpenMPTargets->getNumValues()) {
676       // We expect that -fopenmp-targets is always used in conjunction with the
677       // option -fopenmp specifying a valid runtime with offloading support,
678       // i.e. libomp or libiomp.
679       bool HasValidOpenMPRuntime = C.getInputArgs().hasFlag(
680           options::OPT_fopenmp, options::OPT_fopenmp_EQ,
681           options::OPT_fno_openmp, false);
682       if (HasValidOpenMPRuntime) {
683         OpenMPRuntimeKind OpenMPKind = getOpenMPRuntime(C.getInputArgs());
684         HasValidOpenMPRuntime =
685             OpenMPKind == OMPRT_OMP || OpenMPKind == OMPRT_IOMP5;
686       }
687 
688       if (HasValidOpenMPRuntime) {
689         llvm::StringMap<const char *> FoundNormalizedTriples;
690         for (const char *Val : OpenMPTargets->getValues()) {
691           llvm::Triple TT(Val);
692           std::string NormalizedName = TT.normalize();
693 
694           // Make sure we don't have a duplicate triple.
695           auto Duplicate = FoundNormalizedTriples.find(NormalizedName);
696           if (Duplicate != FoundNormalizedTriples.end()) {
697             Diag(clang::diag::warn_drv_omp_offload_target_duplicate)
698                 << Val << Duplicate->second;
699             continue;
700           }
701 
702           // Store the current triple so that we can check for duplicates in the
703           // following iterations.
704           FoundNormalizedTriples[NormalizedName] = Val;
705 
706           // If the specified target is invalid, emit a diagnostic.
707           if (TT.getArch() == llvm::Triple::UnknownArch)
708             Diag(clang::diag::err_drv_invalid_omp_target) << Val;
709           else {
710             const ToolChain *TC;
711             // CUDA toolchains have to be selected differently. They pair host
712             // and device in their implementation.
713             if (TT.isNVPTX()) {
714               const ToolChain *HostTC =
715                   C.getSingleOffloadToolChain<Action::OFK_Host>();
716               assert(HostTC && "Host toolchain should be always defined.");
717               auto &CudaTC =
718                   ToolChains[TT.str() + "/" + HostTC->getTriple().normalize()];
719               if (!CudaTC)
720                 CudaTC = std::make_unique<toolchains::CudaToolChain>(
721                     *this, TT, *HostTC, C.getInputArgs(), Action::OFK_OpenMP);
722               TC = CudaTC.get();
723             } else
724               TC = &getToolChain(C.getInputArgs(), TT);
725             C.addOffloadDeviceToolChain(TC, Action::OFK_OpenMP);
726           }
727         }
728       } else
729         Diag(clang::diag::err_drv_expecting_fopenmp_with_fopenmp_targets);
730     } else
731       Diag(clang::diag::warn_drv_empty_joined_argument)
732           << OpenMPTargets->getAsString(C.getInputArgs());
733   }
734 
735   //
736   // TODO: Add support for other offloading programming models here.
737   //
738 }
739 
740 /// Looks the given directories for the specified file.
741 ///
742 /// \param[out] FilePath File path, if the file was found.
743 /// \param[in]  Dirs Directories used for the search.
744 /// \param[in]  FileName Name of the file to search for.
745 /// \return True if file was found.
746 ///
747 /// Looks for file specified by FileName sequentially in directories specified
748 /// by Dirs.
749 ///
750 static bool searchForFile(SmallVectorImpl<char> &FilePath,
751                           ArrayRef<std::string> Dirs,
752                           StringRef FileName) {
753   SmallString<128> WPath;
754   for (const std::string &Dir : Dirs) {
755     if (Dir.empty())
756       continue;
757     WPath.clear();
758     llvm::sys::path::append(WPath, Dir, FileName);
759     llvm::sys::path::native(WPath);
760     if (llvm::sys::fs::is_regular_file(WPath)) {
761       FilePath = std::move(WPath);
762       return true;
763     }
764   }
765   return false;
766 }
767 
768 bool Driver::readConfigFile(StringRef FileName) {
769   // Try reading the given file.
770   SmallVector<const char *, 32> NewCfgArgs;
771   if (!llvm::cl::readConfigFile(FileName, Saver, NewCfgArgs)) {
772     Diag(diag::err_drv_cannot_read_config_file) << FileName;
773     return true;
774   }
775 
776   // Read options from config file.
777   llvm::SmallString<128> CfgFileName(FileName);
778   llvm::sys::path::native(CfgFileName);
779   ConfigFile = std::string(CfgFileName.str());
780   bool ContainErrors;
781   CfgOptions = std::make_unique<InputArgList>(
782       ParseArgStrings(NewCfgArgs, IsCLMode(), ContainErrors));
783   if (ContainErrors) {
784     CfgOptions.reset();
785     return true;
786   }
787 
788   if (CfgOptions->hasArg(options::OPT_config)) {
789     CfgOptions.reset();
790     Diag(diag::err_drv_nested_config_file);
791     return true;
792   }
793 
794   // Claim all arguments that come from a configuration file so that the driver
795   // does not warn on any that is unused.
796   for (Arg *A : *CfgOptions)
797     A->claim();
798   return false;
799 }
800 
801 bool Driver::loadConfigFile() {
802   std::string CfgFileName;
803   bool FileSpecifiedExplicitly = false;
804 
805   // Process options that change search path for config files.
806   if (CLOptions) {
807     if (CLOptions->hasArg(options::OPT_config_system_dir_EQ)) {
808       SmallString<128> CfgDir;
809       CfgDir.append(
810           CLOptions->getLastArgValue(options::OPT_config_system_dir_EQ));
811       if (!CfgDir.empty()) {
812         if (llvm::sys::fs::make_absolute(CfgDir).value() != 0)
813           SystemConfigDir.clear();
814         else
815           SystemConfigDir = std::string(CfgDir.begin(), CfgDir.end());
816       }
817     }
818     if (CLOptions->hasArg(options::OPT_config_user_dir_EQ)) {
819       SmallString<128> CfgDir;
820       CfgDir.append(
821           CLOptions->getLastArgValue(options::OPT_config_user_dir_EQ));
822       if (!CfgDir.empty()) {
823         if (llvm::sys::fs::make_absolute(CfgDir).value() != 0)
824           UserConfigDir.clear();
825         else
826           UserConfigDir = std::string(CfgDir.begin(), CfgDir.end());
827       }
828     }
829   }
830 
831   // First try to find config file specified in command line.
832   if (CLOptions) {
833     std::vector<std::string> ConfigFiles =
834         CLOptions->getAllArgValues(options::OPT_config);
835     if (ConfigFiles.size() > 1) {
836       Diag(diag::err_drv_duplicate_config);
837       return true;
838     }
839 
840     if (!ConfigFiles.empty()) {
841       CfgFileName = ConfigFiles.front();
842       assert(!CfgFileName.empty());
843 
844       // If argument contains directory separator, treat it as a path to
845       // configuration file.
846       if (llvm::sys::path::has_parent_path(CfgFileName)) {
847         SmallString<128> CfgFilePath;
848         if (llvm::sys::path::is_relative(CfgFileName))
849           llvm::sys::fs::current_path(CfgFilePath);
850         llvm::sys::path::append(CfgFilePath, CfgFileName);
851         if (!llvm::sys::fs::is_regular_file(CfgFilePath)) {
852           Diag(diag::err_drv_config_file_not_exist) << CfgFilePath;
853           return true;
854         }
855         return readConfigFile(CfgFilePath);
856       }
857 
858       FileSpecifiedExplicitly = true;
859     }
860   }
861 
862   // If config file is not specified explicitly, try to deduce configuration
863   // from executable name. For instance, an executable 'armv7l-clang' will
864   // search for config file 'armv7l-clang.cfg'.
865   if (CfgFileName.empty() && !ClangNameParts.TargetPrefix.empty())
866     CfgFileName = ClangNameParts.TargetPrefix + '-' + ClangNameParts.ModeSuffix;
867 
868   if (CfgFileName.empty())
869     return false;
870 
871   // Determine architecture part of the file name, if it is present.
872   StringRef CfgFileArch = CfgFileName;
873   size_t ArchPrefixLen = CfgFileArch.find('-');
874   if (ArchPrefixLen == StringRef::npos)
875     ArchPrefixLen = CfgFileArch.size();
876   llvm::Triple CfgTriple;
877   CfgFileArch = CfgFileArch.take_front(ArchPrefixLen);
878   CfgTriple = llvm::Triple(llvm::Triple::normalize(CfgFileArch));
879   if (CfgTriple.getArch() == llvm::Triple::ArchType::UnknownArch)
880     ArchPrefixLen = 0;
881 
882   if (!StringRef(CfgFileName).endswith(".cfg"))
883     CfgFileName += ".cfg";
884 
885   // If config file starts with architecture name and command line options
886   // redefine architecture (with options like -m32 -LE etc), try finding new
887   // config file with that architecture.
888   SmallString<128> FixedConfigFile;
889   size_t FixedArchPrefixLen = 0;
890   if (ArchPrefixLen) {
891     // Get architecture name from config file name like 'i386.cfg' or
892     // 'armv7l-clang.cfg'.
893     // Check if command line options changes effective triple.
894     llvm::Triple EffectiveTriple = computeTargetTriple(*this,
895                                              CfgTriple.getTriple(), *CLOptions);
896     if (CfgTriple.getArch() != EffectiveTriple.getArch()) {
897       FixedConfigFile = EffectiveTriple.getArchName();
898       FixedArchPrefixLen = FixedConfigFile.size();
899       // Append the rest of original file name so that file name transforms
900       // like: i386-clang.cfg -> x86_64-clang.cfg.
901       if (ArchPrefixLen < CfgFileName.size())
902         FixedConfigFile += CfgFileName.substr(ArchPrefixLen);
903     }
904   }
905 
906   // Prepare list of directories where config file is searched for.
907   SmallVector<std::string, 3> CfgFileSearchDirs;
908   CfgFileSearchDirs.push_back(UserConfigDir);
909   CfgFileSearchDirs.push_back(SystemConfigDir);
910   CfgFileSearchDirs.push_back(Dir);
911 
912   // Try to find config file. First try file with corrected architecture.
913   llvm::SmallString<128> CfgFilePath;
914   if (!FixedConfigFile.empty()) {
915     if (searchForFile(CfgFilePath, CfgFileSearchDirs, FixedConfigFile))
916       return readConfigFile(CfgFilePath);
917     // If 'x86_64-clang.cfg' was not found, try 'x86_64.cfg'.
918     FixedConfigFile.resize(FixedArchPrefixLen);
919     FixedConfigFile.append(".cfg");
920     if (searchForFile(CfgFilePath, CfgFileSearchDirs, FixedConfigFile))
921       return readConfigFile(CfgFilePath);
922   }
923 
924   // Then try original file name.
925   if (searchForFile(CfgFilePath, CfgFileSearchDirs, CfgFileName))
926     return readConfigFile(CfgFilePath);
927 
928   // Finally try removing driver mode part: 'x86_64-clang.cfg' -> 'x86_64.cfg'.
929   if (!ClangNameParts.ModeSuffix.empty() &&
930       !ClangNameParts.TargetPrefix.empty()) {
931     CfgFileName.assign(ClangNameParts.TargetPrefix);
932     CfgFileName.append(".cfg");
933     if (searchForFile(CfgFilePath, CfgFileSearchDirs, CfgFileName))
934       return readConfigFile(CfgFilePath);
935   }
936 
937   // Report error but only if config file was specified explicitly, by option
938   // --config. If it was deduced from executable name, it is not an error.
939   if (FileSpecifiedExplicitly) {
940     Diag(diag::err_drv_config_file_not_found) << CfgFileName;
941     for (const std::string &SearchDir : CfgFileSearchDirs)
942       if (!SearchDir.empty())
943         Diag(diag::note_drv_config_file_searched_in) << SearchDir;
944     return true;
945   }
946 
947   return false;
948 }
949 
950 Compilation *Driver::BuildCompilation(ArrayRef<const char *> ArgList) {
951   llvm::PrettyStackTraceString CrashInfo("Compilation construction");
952 
953   // FIXME: Handle environment options which affect driver behavior, somewhere
954   // (client?). GCC_EXEC_PREFIX, LPATH, CC_PRINT_OPTIONS.
955 
956   if (Optional<std::string> CompilerPathValue =
957           llvm::sys::Process::GetEnv("COMPILER_PATH")) {
958     StringRef CompilerPath = *CompilerPathValue;
959     while (!CompilerPath.empty()) {
960       std::pair<StringRef, StringRef> Split =
961           CompilerPath.split(llvm::sys::EnvPathSeparator);
962       PrefixDirs.push_back(std::string(Split.first));
963       CompilerPath = Split.second;
964     }
965   }
966 
967   // We look for the driver mode option early, because the mode can affect
968   // how other options are parsed.
969   ParseDriverMode(ClangExecutable, ArgList.slice(1));
970 
971   // FIXME: What are we going to do with -V and -b?
972 
973   // Arguments specified in command line.
974   bool ContainsError;
975   CLOptions = std::make_unique<InputArgList>(
976       ParseArgStrings(ArgList.slice(1), IsCLMode(), ContainsError));
977 
978   // Try parsing configuration file.
979   if (!ContainsError)
980     ContainsError = loadConfigFile();
981   bool HasConfigFile = !ContainsError && (CfgOptions.get() != nullptr);
982 
983   // All arguments, from both config file and command line.
984   InputArgList Args = std::move(HasConfigFile ? std::move(*CfgOptions)
985                                               : std::move(*CLOptions));
986 
987   // The args for config files or /clang: flags belong to different InputArgList
988   // objects than Args. This copies an Arg from one of those other InputArgLists
989   // to the ownership of Args.
990   auto appendOneArg = [&Args](const Arg *Opt, const Arg *BaseArg) {
991       unsigned Index = Args.MakeIndex(Opt->getSpelling());
992       Arg *Copy = new llvm::opt::Arg(Opt->getOption(), Opt->getSpelling(),
993                                      Index, BaseArg);
994       Copy->getValues() = Opt->getValues();
995       if (Opt->isClaimed())
996         Copy->claim();
997       Args.append(Copy);
998   };
999 
1000   if (HasConfigFile)
1001     for (auto *Opt : *CLOptions) {
1002       if (Opt->getOption().matches(options::OPT_config))
1003         continue;
1004       const Arg *BaseArg = &Opt->getBaseArg();
1005       if (BaseArg == Opt)
1006         BaseArg = nullptr;
1007       appendOneArg(Opt, BaseArg);
1008     }
1009 
1010   // In CL mode, look for any pass-through arguments
1011   if (IsCLMode() && !ContainsError) {
1012     SmallVector<const char *, 16> CLModePassThroughArgList;
1013     for (const auto *A : Args.filtered(options::OPT__SLASH_clang)) {
1014       A->claim();
1015       CLModePassThroughArgList.push_back(A->getValue());
1016     }
1017 
1018     if (!CLModePassThroughArgList.empty()) {
1019       // Parse any pass through args using default clang processing rather
1020       // than clang-cl processing.
1021       auto CLModePassThroughOptions = std::make_unique<InputArgList>(
1022           ParseArgStrings(CLModePassThroughArgList, false, ContainsError));
1023 
1024       if (!ContainsError)
1025         for (auto *Opt : *CLModePassThroughOptions) {
1026           appendOneArg(Opt, nullptr);
1027         }
1028     }
1029   }
1030 
1031   // Check for working directory option before accessing any files
1032   if (Arg *WD = Args.getLastArg(options::OPT_working_directory))
1033     if (VFS->setCurrentWorkingDirectory(WD->getValue()))
1034       Diag(diag::err_drv_unable_to_set_working_directory) << WD->getValue();
1035 
1036   // FIXME: This stuff needs to go into the Compilation, not the driver.
1037   bool CCCPrintPhases;
1038 
1039   // Silence driver warnings if requested
1040   Diags.setIgnoreAllWarnings(Args.hasArg(options::OPT_w));
1041 
1042   // -no-canonical-prefixes is used very early in main.
1043   Args.ClaimAllArgs(options::OPT_no_canonical_prefixes);
1044 
1045   // f(no-)integated-cc1 is also used very early in main.
1046   Args.ClaimAllArgs(options::OPT_fintegrated_cc1);
1047   Args.ClaimAllArgs(options::OPT_fno_integrated_cc1);
1048 
1049   // Ignore -pipe.
1050   Args.ClaimAllArgs(options::OPT_pipe);
1051 
1052   // Extract -ccc args.
1053   //
1054   // FIXME: We need to figure out where this behavior should live. Most of it
1055   // should be outside in the client; the parts that aren't should have proper
1056   // options, either by introducing new ones or by overloading gcc ones like -V
1057   // or -b.
1058   CCCPrintPhases = Args.hasArg(options::OPT_ccc_print_phases);
1059   CCCPrintBindings = Args.hasArg(options::OPT_ccc_print_bindings);
1060   if (const Arg *A = Args.getLastArg(options::OPT_ccc_gcc_name))
1061     CCCGenericGCCName = A->getValue();
1062   GenReproducer = Args.hasFlag(options::OPT_gen_reproducer,
1063                                options::OPT_fno_crash_diagnostics,
1064                                !!::getenv("FORCE_CLANG_DIAGNOSTICS_CRASH"));
1065   // FIXME: TargetTriple is used by the target-prefixed calls to as/ld
1066   // and getToolChain is const.
1067   if (IsCLMode()) {
1068     // clang-cl targets MSVC-style Win32.
1069     llvm::Triple T(TargetTriple);
1070     T.setOS(llvm::Triple::Win32);
1071     T.setVendor(llvm::Triple::PC);
1072     T.setEnvironment(llvm::Triple::MSVC);
1073     T.setObjectFormat(llvm::Triple::COFF);
1074     TargetTriple = T.str();
1075   }
1076   if (const Arg *A = Args.getLastArg(options::OPT_target))
1077     TargetTriple = A->getValue();
1078   if (const Arg *A = Args.getLastArg(options::OPT_ccc_install_dir))
1079     Dir = InstalledDir = A->getValue();
1080   for (const Arg *A : Args.filtered(options::OPT_B)) {
1081     A->claim();
1082     PrefixDirs.push_back(A->getValue(0));
1083   }
1084   if (const Arg *A = Args.getLastArg(options::OPT__sysroot_EQ))
1085     SysRoot = A->getValue();
1086   if (const Arg *A = Args.getLastArg(options::OPT__dyld_prefix_EQ))
1087     DyldPrefix = A->getValue();
1088 
1089   if (const Arg *A = Args.getLastArg(options::OPT_resource_dir))
1090     ResourceDir = A->getValue();
1091 
1092   if (const Arg *A = Args.getLastArg(options::OPT_save_temps_EQ)) {
1093     SaveTemps = llvm::StringSwitch<SaveTempsMode>(A->getValue())
1094                     .Case("cwd", SaveTempsCwd)
1095                     .Case("obj", SaveTempsObj)
1096                     .Default(SaveTempsCwd);
1097   }
1098 
1099   setLTOMode(Args);
1100 
1101   // Process -fembed-bitcode= flags.
1102   if (Arg *A = Args.getLastArg(options::OPT_fembed_bitcode_EQ)) {
1103     StringRef Name = A->getValue();
1104     unsigned Model = llvm::StringSwitch<unsigned>(Name)
1105         .Case("off", EmbedNone)
1106         .Case("all", EmbedBitcode)
1107         .Case("bitcode", EmbedBitcode)
1108         .Case("marker", EmbedMarker)
1109         .Default(~0U);
1110     if (Model == ~0U) {
1111       Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args)
1112                                                 << Name;
1113     } else
1114       BitcodeEmbed = static_cast<BitcodeEmbedMode>(Model);
1115   }
1116 
1117   std::unique_ptr<llvm::opt::InputArgList> UArgs =
1118       std::make_unique<InputArgList>(std::move(Args));
1119 
1120   // Perform the default argument translations.
1121   DerivedArgList *TranslatedArgs = TranslateInputArgs(*UArgs);
1122 
1123   // Owned by the host.
1124   const ToolChain &TC = getToolChain(
1125       *UArgs, computeTargetTriple(*this, TargetTriple, *UArgs));
1126 
1127   // The compilation takes ownership of Args.
1128   Compilation *C = new Compilation(*this, TC, UArgs.release(), TranslatedArgs,
1129                                    ContainsError);
1130 
1131   if (!HandleImmediateArgs(*C))
1132     return C;
1133 
1134   // Construct the list of inputs.
1135   InputList Inputs;
1136   BuildInputs(C->getDefaultToolChain(), *TranslatedArgs, Inputs);
1137 
1138   // Populate the tool chains for the offloading devices, if any.
1139   CreateOffloadingDeviceToolChains(*C, Inputs);
1140 
1141   // Construct the list of abstract actions to perform for this compilation. On
1142   // MachO targets this uses the driver-driver and universal actions.
1143   if (TC.getTriple().isOSBinFormatMachO())
1144     BuildUniversalActions(*C, C->getDefaultToolChain(), Inputs);
1145   else
1146     BuildActions(*C, C->getArgs(), Inputs, C->getActions());
1147 
1148   if (CCCPrintPhases) {
1149     PrintActions(*C);
1150     return C;
1151   }
1152 
1153   BuildJobs(*C);
1154 
1155   return C;
1156 }
1157 
1158 static void printArgList(raw_ostream &OS, const llvm::opt::ArgList &Args) {
1159   llvm::opt::ArgStringList ASL;
1160   for (const auto *A : Args)
1161     A->render(Args, ASL);
1162 
1163   for (auto I = ASL.begin(), E = ASL.end(); I != E; ++I) {
1164     if (I != ASL.begin())
1165       OS << ' ';
1166     Command::printArg(OS, *I, true);
1167   }
1168   OS << '\n';
1169 }
1170 
1171 bool Driver::getCrashDiagnosticFile(StringRef ReproCrashFilename,
1172                                     SmallString<128> &CrashDiagDir) {
1173   using namespace llvm::sys;
1174   assert(llvm::Triple(llvm::sys::getProcessTriple()).isOSDarwin() &&
1175          "Only knows about .crash files on Darwin");
1176 
1177   // The .crash file can be found on at ~/Library/Logs/DiagnosticReports/
1178   // (or /Library/Logs/DiagnosticReports for root) and has the filename pattern
1179   // clang-<VERSION>_<YYYY-MM-DD-HHMMSS>_<hostname>.crash.
1180   path::home_directory(CrashDiagDir);
1181   if (CrashDiagDir.startswith("/var/root"))
1182     CrashDiagDir = "/";
1183   path::append(CrashDiagDir, "Library/Logs/DiagnosticReports");
1184   int PID =
1185 #if LLVM_ON_UNIX
1186       getpid();
1187 #else
1188       0;
1189 #endif
1190   std::error_code EC;
1191   fs::file_status FileStatus;
1192   TimePoint<> LastAccessTime;
1193   SmallString<128> CrashFilePath;
1194   // Lookup the .crash files and get the one generated by a subprocess spawned
1195   // by this driver invocation.
1196   for (fs::directory_iterator File(CrashDiagDir, EC), FileEnd;
1197        File != FileEnd && !EC; File.increment(EC)) {
1198     StringRef FileName = path::filename(File->path());
1199     if (!FileName.startswith(Name))
1200       continue;
1201     if (fs::status(File->path(), FileStatus))
1202       continue;
1203     llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> CrashFile =
1204         llvm::MemoryBuffer::getFile(File->path());
1205     if (!CrashFile)
1206       continue;
1207     // The first line should start with "Process:", otherwise this isn't a real
1208     // .crash file.
1209     StringRef Data = CrashFile.get()->getBuffer();
1210     if (!Data.startswith("Process:"))
1211       continue;
1212     // Parse parent process pid line, e.g: "Parent Process: clang-4.0 [79141]"
1213     size_t ParentProcPos = Data.find("Parent Process:");
1214     if (ParentProcPos == StringRef::npos)
1215       continue;
1216     size_t LineEnd = Data.find_first_of("\n", ParentProcPos);
1217     if (LineEnd == StringRef::npos)
1218       continue;
1219     StringRef ParentProcess = Data.slice(ParentProcPos+15, LineEnd).trim();
1220     int OpenBracket = -1, CloseBracket = -1;
1221     for (size_t i = 0, e = ParentProcess.size(); i < e; ++i) {
1222       if (ParentProcess[i] == '[')
1223         OpenBracket = i;
1224       if (ParentProcess[i] == ']')
1225         CloseBracket = i;
1226     }
1227     // Extract the parent process PID from the .crash file and check whether
1228     // it matches this driver invocation pid.
1229     int CrashPID;
1230     if (OpenBracket < 0 || CloseBracket < 0 ||
1231         ParentProcess.slice(OpenBracket + 1, CloseBracket)
1232             .getAsInteger(10, CrashPID) || CrashPID != PID) {
1233       continue;
1234     }
1235 
1236     // Found a .crash file matching the driver pid. To avoid getting an older
1237     // and misleading crash file, continue looking for the most recent.
1238     // FIXME: the driver can dispatch multiple cc1 invocations, leading to
1239     // multiple crashes poiting to the same parent process. Since the driver
1240     // does not collect pid information for the dispatched invocation there's
1241     // currently no way to distinguish among them.
1242     const auto FileAccessTime = FileStatus.getLastModificationTime();
1243     if (FileAccessTime > LastAccessTime) {
1244       CrashFilePath.assign(File->path());
1245       LastAccessTime = FileAccessTime;
1246     }
1247   }
1248 
1249   // If found, copy it over to the location of other reproducer files.
1250   if (!CrashFilePath.empty()) {
1251     EC = fs::copy_file(CrashFilePath, ReproCrashFilename);
1252     if (EC)
1253       return false;
1254     return true;
1255   }
1256 
1257   return false;
1258 }
1259 
1260 // When clang crashes, produce diagnostic information including the fully
1261 // preprocessed source file(s).  Request that the developer attach the
1262 // diagnostic information to a bug report.
1263 void Driver::generateCompilationDiagnostics(
1264     Compilation &C, const Command &FailingCommand,
1265     StringRef AdditionalInformation, CompilationDiagnosticReport *Report) {
1266   if (C.getArgs().hasArg(options::OPT_fno_crash_diagnostics))
1267     return;
1268 
1269   // Don't try to generate diagnostics for link or dsymutil jobs.
1270   if (FailingCommand.getCreator().isLinkJob() ||
1271       FailingCommand.getCreator().isDsymutilJob())
1272     return;
1273 
1274   // Print the version of the compiler.
1275   PrintVersion(C, llvm::errs());
1276 
1277   // Suppress driver output and emit preprocessor output to temp file.
1278   Mode = CPPMode;
1279   CCGenDiagnostics = true;
1280 
1281   // Save the original job command(s).
1282   Command Cmd = FailingCommand;
1283 
1284   // Keep track of whether we produce any errors while trying to produce
1285   // preprocessed sources.
1286   DiagnosticErrorTrap Trap(Diags);
1287 
1288   // Suppress tool output.
1289   C.initCompilationForDiagnostics();
1290 
1291   // Construct the list of inputs.
1292   InputList Inputs;
1293   BuildInputs(C.getDefaultToolChain(), C.getArgs(), Inputs);
1294 
1295   for (InputList::iterator it = Inputs.begin(), ie = Inputs.end(); it != ie;) {
1296     bool IgnoreInput = false;
1297 
1298     // Ignore input from stdin or any inputs that cannot be preprocessed.
1299     // Check type first as not all linker inputs have a value.
1300     if (types::getPreprocessedType(it->first) == types::TY_INVALID) {
1301       IgnoreInput = true;
1302     } else if (!strcmp(it->second->getValue(), "-")) {
1303       Diag(clang::diag::note_drv_command_failed_diag_msg)
1304           << "Error generating preprocessed source(s) - "
1305              "ignoring input from stdin.";
1306       IgnoreInput = true;
1307     }
1308 
1309     if (IgnoreInput) {
1310       it = Inputs.erase(it);
1311       ie = Inputs.end();
1312     } else {
1313       ++it;
1314     }
1315   }
1316 
1317   if (Inputs.empty()) {
1318     Diag(clang::diag::note_drv_command_failed_diag_msg)
1319         << "Error generating preprocessed source(s) - "
1320            "no preprocessable inputs.";
1321     return;
1322   }
1323 
1324   // Don't attempt to generate preprocessed files if multiple -arch options are
1325   // used, unless they're all duplicates.
1326   llvm::StringSet<> ArchNames;
1327   for (const Arg *A : C.getArgs()) {
1328     if (A->getOption().matches(options::OPT_arch)) {
1329       StringRef ArchName = A->getValue();
1330       ArchNames.insert(ArchName);
1331     }
1332   }
1333   if (ArchNames.size() > 1) {
1334     Diag(clang::diag::note_drv_command_failed_diag_msg)
1335         << "Error generating preprocessed source(s) - cannot generate "
1336            "preprocessed source with multiple -arch options.";
1337     return;
1338   }
1339 
1340   // Construct the list of abstract actions to perform for this compilation. On
1341   // Darwin OSes this uses the driver-driver and builds universal actions.
1342   const ToolChain &TC = C.getDefaultToolChain();
1343   if (TC.getTriple().isOSBinFormatMachO())
1344     BuildUniversalActions(C, TC, Inputs);
1345   else
1346     BuildActions(C, C.getArgs(), Inputs, C.getActions());
1347 
1348   BuildJobs(C);
1349 
1350   // If there were errors building the compilation, quit now.
1351   if (Trap.hasErrorOccurred()) {
1352     Diag(clang::diag::note_drv_command_failed_diag_msg)
1353         << "Error generating preprocessed source(s).";
1354     return;
1355   }
1356 
1357   // Generate preprocessed output.
1358   SmallVector<std::pair<int, const Command *>, 4> FailingCommands;
1359   C.ExecuteJobs(C.getJobs(), FailingCommands);
1360 
1361   // If any of the preprocessing commands failed, clean up and exit.
1362   if (!FailingCommands.empty()) {
1363     Diag(clang::diag::note_drv_command_failed_diag_msg)
1364         << "Error generating preprocessed source(s).";
1365     return;
1366   }
1367 
1368   const ArgStringList &TempFiles = C.getTempFiles();
1369   if (TempFiles.empty()) {
1370     Diag(clang::diag::note_drv_command_failed_diag_msg)
1371         << "Error generating preprocessed source(s).";
1372     return;
1373   }
1374 
1375   Diag(clang::diag::note_drv_command_failed_diag_msg)
1376       << "\n********************\n\n"
1377          "PLEASE ATTACH THE FOLLOWING FILES TO THE BUG REPORT:\n"
1378          "Preprocessed source(s) and associated run script(s) are located at:";
1379 
1380   SmallString<128> VFS;
1381   SmallString<128> ReproCrashFilename;
1382   for (const char *TempFile : TempFiles) {
1383     Diag(clang::diag::note_drv_command_failed_diag_msg) << TempFile;
1384     if (Report)
1385       Report->TemporaryFiles.push_back(TempFile);
1386     if (ReproCrashFilename.empty()) {
1387       ReproCrashFilename = TempFile;
1388       llvm::sys::path::replace_extension(ReproCrashFilename, ".crash");
1389     }
1390     if (StringRef(TempFile).endswith(".cache")) {
1391       // In some cases (modules) we'll dump extra data to help with reproducing
1392       // the crash into a directory next to the output.
1393       VFS = llvm::sys::path::filename(TempFile);
1394       llvm::sys::path::append(VFS, "vfs", "vfs.yaml");
1395     }
1396   }
1397 
1398   // Assume associated files are based off of the first temporary file.
1399   CrashReportInfo CrashInfo(TempFiles[0], VFS);
1400 
1401   llvm::SmallString<128> Script(CrashInfo.Filename);
1402   llvm::sys::path::replace_extension(Script, "sh");
1403   std::error_code EC;
1404   llvm::raw_fd_ostream ScriptOS(Script, EC, llvm::sys::fs::CD_CreateNew);
1405   if (EC) {
1406     Diag(clang::diag::note_drv_command_failed_diag_msg)
1407         << "Error generating run script: " << Script << " " << EC.message();
1408   } else {
1409     ScriptOS << "# Crash reproducer for " << getClangFullVersion() << "\n"
1410              << "# Driver args: ";
1411     printArgList(ScriptOS, C.getInputArgs());
1412     ScriptOS << "# Original command: ";
1413     Cmd.Print(ScriptOS, "\n", /*Quote=*/true);
1414     Cmd.Print(ScriptOS, "\n", /*Quote=*/true, &CrashInfo);
1415     if (!AdditionalInformation.empty())
1416       ScriptOS << "\n# Additional information: " << AdditionalInformation
1417                << "\n";
1418     if (Report)
1419       Report->TemporaryFiles.push_back(std::string(Script.str()));
1420     Diag(clang::diag::note_drv_command_failed_diag_msg) << Script;
1421   }
1422 
1423   // On darwin, provide information about the .crash diagnostic report.
1424   if (llvm::Triple(llvm::sys::getProcessTriple()).isOSDarwin()) {
1425     SmallString<128> CrashDiagDir;
1426     if (getCrashDiagnosticFile(ReproCrashFilename, CrashDiagDir)) {
1427       Diag(clang::diag::note_drv_command_failed_diag_msg)
1428           << ReproCrashFilename.str();
1429     } else { // Suggest a directory for the user to look for .crash files.
1430       llvm::sys::path::append(CrashDiagDir, Name);
1431       CrashDiagDir += "_<YYYY-MM-DD-HHMMSS>_<hostname>.crash";
1432       Diag(clang::diag::note_drv_command_failed_diag_msg)
1433           << "Crash backtrace is located in";
1434       Diag(clang::diag::note_drv_command_failed_diag_msg)
1435           << CrashDiagDir.str();
1436       Diag(clang::diag::note_drv_command_failed_diag_msg)
1437           << "(choose the .crash file that corresponds to your crash)";
1438     }
1439   }
1440 
1441   for (const auto &A : C.getArgs().filtered(options::OPT_frewrite_map_file,
1442                                             options::OPT_frewrite_map_file_EQ))
1443     Diag(clang::diag::note_drv_command_failed_diag_msg) << A->getValue();
1444 
1445   Diag(clang::diag::note_drv_command_failed_diag_msg)
1446       << "\n\n********************";
1447 }
1448 
1449 void Driver::setUpResponseFiles(Compilation &C, Command &Cmd) {
1450   // Since commandLineFitsWithinSystemLimits() may underestimate system's
1451   // capacity if the tool does not support response files, there is a chance/
1452   // that things will just work without a response file, so we silently just
1453   // skip it.
1454   if (Cmd.getCreator().getResponseFilesSupport() == Tool::RF_None ||
1455       llvm::sys::commandLineFitsWithinSystemLimits(Cmd.getExecutable(),
1456                                                    Cmd.getArguments()))
1457     return;
1458 
1459   std::string TmpName = GetTemporaryPath("response", "txt");
1460   Cmd.setResponseFile(C.addTempFile(C.getArgs().MakeArgString(TmpName)));
1461 }
1462 
1463 int Driver::ExecuteCompilation(
1464     Compilation &C,
1465     SmallVectorImpl<std::pair<int, const Command *>> &FailingCommands) {
1466   // Just print if -### was present.
1467   if (C.getArgs().hasArg(options::OPT__HASH_HASH_HASH)) {
1468     C.getJobs().Print(llvm::errs(), "\n", true);
1469     return 0;
1470   }
1471 
1472   // If there were errors building the compilation, quit now.
1473   if (Diags.hasErrorOccurred())
1474     return 1;
1475 
1476   // Set up response file names for each command, if necessary
1477   for (auto &Job : C.getJobs())
1478     setUpResponseFiles(C, Job);
1479 
1480   C.ExecuteJobs(C.getJobs(), FailingCommands);
1481 
1482   // If the command succeeded, we are done.
1483   if (FailingCommands.empty())
1484     return 0;
1485 
1486   // Otherwise, remove result files and print extra information about abnormal
1487   // failures.
1488   int Res = 0;
1489   for (const auto &CmdPair : FailingCommands) {
1490     int CommandRes = CmdPair.first;
1491     const Command *FailingCommand = CmdPair.second;
1492 
1493     // Remove result files if we're not saving temps.
1494     if (!isSaveTempsEnabled()) {
1495       const JobAction *JA = cast<JobAction>(&FailingCommand->getSource());
1496       C.CleanupFileMap(C.getResultFiles(), JA, true);
1497 
1498       // Failure result files are valid unless we crashed.
1499       if (CommandRes < 0)
1500         C.CleanupFileMap(C.getFailureResultFiles(), JA, true);
1501     }
1502 
1503 #if LLVM_ON_UNIX
1504     // llvm/lib/Support/Unix/Signals.inc will exit with a special return code
1505     // for SIGPIPE. Do not print diagnostics for this case.
1506     if (CommandRes == EX_IOERR) {
1507       Res = CommandRes;
1508       continue;
1509     }
1510 #endif
1511 
1512     // Print extra information about abnormal failures, if possible.
1513     //
1514     // This is ad-hoc, but we don't want to be excessively noisy. If the result
1515     // status was 1, assume the command failed normally. In particular, if it
1516     // was the compiler then assume it gave a reasonable error code. Failures
1517     // in other tools are less common, and they generally have worse
1518     // diagnostics, so always print the diagnostic there.
1519     const Tool &FailingTool = FailingCommand->getCreator();
1520 
1521     if (!FailingCommand->getCreator().hasGoodDiagnostics() || CommandRes != 1) {
1522       // FIXME: See FIXME above regarding result code interpretation.
1523       if (CommandRes < 0)
1524         Diag(clang::diag::err_drv_command_signalled)
1525             << FailingTool.getShortName();
1526       else
1527         Diag(clang::diag::err_drv_command_failed)
1528             << FailingTool.getShortName() << CommandRes;
1529     }
1530   }
1531   return Res;
1532 }
1533 
1534 void Driver::PrintHelp(bool ShowHidden) const {
1535   unsigned IncludedFlagsBitmask;
1536   unsigned ExcludedFlagsBitmask;
1537   std::tie(IncludedFlagsBitmask, ExcludedFlagsBitmask) =
1538       getIncludeExcludeOptionFlagMasks(IsCLMode());
1539 
1540   ExcludedFlagsBitmask |= options::NoDriverOption;
1541   if (!ShowHidden)
1542     ExcludedFlagsBitmask |= HelpHidden;
1543 
1544   std::string Usage = llvm::formatv("{0} [options] file...", Name).str();
1545   getOpts().PrintHelp(llvm::outs(), Usage.c_str(), DriverTitle.c_str(),
1546                       IncludedFlagsBitmask, ExcludedFlagsBitmask,
1547                       /*ShowAllAliases=*/false);
1548 }
1549 
1550 void Driver::PrintVersion(const Compilation &C, raw_ostream &OS) const {
1551   // FIXME: The following handlers should use a callback mechanism, we don't
1552   // know what the client would like to do.
1553   OS << getClangFullVersion() << '\n';
1554   const ToolChain &TC = C.getDefaultToolChain();
1555   OS << "Target: " << TC.getTripleString() << '\n';
1556 
1557   // Print the threading model.
1558   if (Arg *A = C.getArgs().getLastArg(options::OPT_mthread_model)) {
1559     // Don't print if the ToolChain would have barfed on it already
1560     if (TC.isThreadModelSupported(A->getValue()))
1561       OS << "Thread model: " << A->getValue();
1562   } else
1563     OS << "Thread model: " << TC.getThreadModel();
1564   OS << '\n';
1565 
1566   // Print out the install directory.
1567   OS << "InstalledDir: " << InstalledDir << '\n';
1568 
1569   // If configuration file was used, print its path.
1570   if (!ConfigFile.empty())
1571     OS << "Configuration file: " << ConfigFile << '\n';
1572 }
1573 
1574 /// PrintDiagnosticCategories - Implement the --print-diagnostic-categories
1575 /// option.
1576 static void PrintDiagnosticCategories(raw_ostream &OS) {
1577   // Skip the empty category.
1578   for (unsigned i = 1, max = DiagnosticIDs::getNumberOfCategories(); i != max;
1579        ++i)
1580     OS << i << ',' << DiagnosticIDs::getCategoryNameFromID(i) << '\n';
1581 }
1582 
1583 void Driver::HandleAutocompletions(StringRef PassedFlags) const {
1584   if (PassedFlags == "")
1585     return;
1586   // Print out all options that start with a given argument. This is used for
1587   // shell autocompletion.
1588   std::vector<std::string> SuggestedCompletions;
1589   std::vector<std::string> Flags;
1590 
1591   unsigned short DisableFlags =
1592       options::NoDriverOption | options::Unsupported | options::Ignored;
1593 
1594   // Distinguish "--autocomplete=-someflag" and "--autocomplete=-someflag,"
1595   // because the latter indicates that the user put space before pushing tab
1596   // which should end up in a file completion.
1597   const bool HasSpace = PassedFlags.endswith(",");
1598 
1599   // Parse PassedFlags by "," as all the command-line flags are passed to this
1600   // function separated by ","
1601   StringRef TargetFlags = PassedFlags;
1602   while (TargetFlags != "") {
1603     StringRef CurFlag;
1604     std::tie(CurFlag, TargetFlags) = TargetFlags.split(",");
1605     Flags.push_back(std::string(CurFlag));
1606   }
1607 
1608   // We want to show cc1-only options only when clang is invoked with -cc1 or
1609   // -Xclang.
1610   if (llvm::is_contained(Flags, "-Xclang") || llvm::is_contained(Flags, "-cc1"))
1611     DisableFlags &= ~options::NoDriverOption;
1612 
1613   const llvm::opt::OptTable &Opts = getOpts();
1614   StringRef Cur;
1615   Cur = Flags.at(Flags.size() - 1);
1616   StringRef Prev;
1617   if (Flags.size() >= 2) {
1618     Prev = Flags.at(Flags.size() - 2);
1619     SuggestedCompletions = Opts.suggestValueCompletions(Prev, Cur);
1620   }
1621 
1622   if (SuggestedCompletions.empty())
1623     SuggestedCompletions = Opts.suggestValueCompletions(Cur, "");
1624 
1625   // If Flags were empty, it means the user typed `clang [tab]` where we should
1626   // list all possible flags. If there was no value completion and the user
1627   // pressed tab after a space, we should fall back to a file completion.
1628   // We're printing a newline to be consistent with what we print at the end of
1629   // this function.
1630   if (SuggestedCompletions.empty() && HasSpace && !Flags.empty()) {
1631     llvm::outs() << '\n';
1632     return;
1633   }
1634 
1635   // When flag ends with '=' and there was no value completion, return empty
1636   // string and fall back to the file autocompletion.
1637   if (SuggestedCompletions.empty() && !Cur.endswith("=")) {
1638     // If the flag is in the form of "--autocomplete=-foo",
1639     // we were requested to print out all option names that start with "-foo".
1640     // For example, "--autocomplete=-fsyn" is expanded to "-fsyntax-only".
1641     SuggestedCompletions = Opts.findByPrefix(Cur, DisableFlags);
1642 
1643     // We have to query the -W flags manually as they're not in the OptTable.
1644     // TODO: Find a good way to add them to OptTable instead and them remove
1645     // this code.
1646     for (StringRef S : DiagnosticIDs::getDiagnosticFlags())
1647       if (S.startswith(Cur))
1648         SuggestedCompletions.push_back(std::string(S));
1649   }
1650 
1651   // Sort the autocomplete candidates so that shells print them out in a
1652   // deterministic order. We could sort in any way, but we chose
1653   // case-insensitive sorting for consistency with the -help option
1654   // which prints out options in the case-insensitive alphabetical order.
1655   llvm::sort(SuggestedCompletions, [](StringRef A, StringRef B) {
1656     if (int X = A.compare_lower(B))
1657       return X < 0;
1658     return A.compare(B) > 0;
1659   });
1660 
1661   llvm::outs() << llvm::join(SuggestedCompletions, "\n") << '\n';
1662 }
1663 
1664 bool Driver::HandleImmediateArgs(const Compilation &C) {
1665   // The order these options are handled in gcc is all over the place, but we
1666   // don't expect inconsistencies w.r.t. that to matter in practice.
1667 
1668   if (C.getArgs().hasArg(options::OPT_dumpmachine)) {
1669     llvm::outs() << C.getDefaultToolChain().getTripleString() << '\n';
1670     return false;
1671   }
1672 
1673   if (C.getArgs().hasArg(options::OPT_dumpversion)) {
1674     // Since -dumpversion is only implemented for pedantic GCC compatibility, we
1675     // return an answer which matches our definition of __VERSION__.
1676     llvm::outs() << CLANG_VERSION_STRING << "\n";
1677     return false;
1678   }
1679 
1680   if (C.getArgs().hasArg(options::OPT__print_diagnostic_categories)) {
1681     PrintDiagnosticCategories(llvm::outs());
1682     return false;
1683   }
1684 
1685   if (C.getArgs().hasArg(options::OPT_help) ||
1686       C.getArgs().hasArg(options::OPT__help_hidden)) {
1687     PrintHelp(C.getArgs().hasArg(options::OPT__help_hidden));
1688     return false;
1689   }
1690 
1691   if (C.getArgs().hasArg(options::OPT__version)) {
1692     // Follow gcc behavior and use stdout for --version and stderr for -v.
1693     PrintVersion(C, llvm::outs());
1694     return false;
1695   }
1696 
1697   if (C.getArgs().hasArg(options::OPT_v) ||
1698       C.getArgs().hasArg(options::OPT__HASH_HASH_HASH) ||
1699       C.getArgs().hasArg(options::OPT_print_supported_cpus)) {
1700     PrintVersion(C, llvm::errs());
1701     SuppressMissingInputWarning = true;
1702   }
1703 
1704   if (C.getArgs().hasArg(options::OPT_v)) {
1705     if (!SystemConfigDir.empty())
1706       llvm::errs() << "System configuration file directory: "
1707                    << SystemConfigDir << "\n";
1708     if (!UserConfigDir.empty())
1709       llvm::errs() << "User configuration file directory: "
1710                    << UserConfigDir << "\n";
1711   }
1712 
1713   const ToolChain &TC = C.getDefaultToolChain();
1714 
1715   if (C.getArgs().hasArg(options::OPT_v))
1716     TC.printVerboseInfo(llvm::errs());
1717 
1718   if (C.getArgs().hasArg(options::OPT_print_resource_dir)) {
1719     llvm::outs() << ResourceDir << '\n';
1720     return false;
1721   }
1722 
1723   if (C.getArgs().hasArg(options::OPT_print_search_dirs)) {
1724     llvm::outs() << "programs: =";
1725     bool separator = false;
1726     for (const std::string &Path : TC.getProgramPaths()) {
1727       if (separator)
1728         llvm::outs() << llvm::sys::EnvPathSeparator;
1729       llvm::outs() << Path;
1730       separator = true;
1731     }
1732     llvm::outs() << "\n";
1733     llvm::outs() << "libraries: =" << ResourceDir;
1734 
1735     StringRef sysroot = C.getSysRoot();
1736 
1737     for (const std::string &Path : TC.getFilePaths()) {
1738       // Always print a separator. ResourceDir was the first item shown.
1739       llvm::outs() << llvm::sys::EnvPathSeparator;
1740       // Interpretation of leading '=' is needed only for NetBSD.
1741       if (Path[0] == '=')
1742         llvm::outs() << sysroot << Path.substr(1);
1743       else
1744         llvm::outs() << Path;
1745     }
1746     llvm::outs() << "\n";
1747     return false;
1748   }
1749 
1750   // FIXME: The following handlers should use a callback mechanism, we don't
1751   // know what the client would like to do.
1752   if (Arg *A = C.getArgs().getLastArg(options::OPT_print_file_name_EQ)) {
1753     llvm::outs() << GetFilePath(A->getValue(), TC) << "\n";
1754     return false;
1755   }
1756 
1757   if (Arg *A = C.getArgs().getLastArg(options::OPT_print_prog_name_EQ)) {
1758     StringRef ProgName = A->getValue();
1759 
1760     // Null program name cannot have a path.
1761     if (! ProgName.empty())
1762       llvm::outs() << GetProgramPath(ProgName, TC);
1763 
1764     llvm::outs() << "\n";
1765     return false;
1766   }
1767 
1768   if (Arg *A = C.getArgs().getLastArg(options::OPT_autocomplete)) {
1769     StringRef PassedFlags = A->getValue();
1770     HandleAutocompletions(PassedFlags);
1771     return false;
1772   }
1773 
1774   if (C.getArgs().hasArg(options::OPT_print_libgcc_file_name)) {
1775     ToolChain::RuntimeLibType RLT = TC.GetRuntimeLibType(C.getArgs());
1776     const llvm::Triple Triple(TC.ComputeEffectiveClangTriple(C.getArgs()));
1777     RegisterEffectiveTriple TripleRAII(TC, Triple);
1778     switch (RLT) {
1779     case ToolChain::RLT_CompilerRT:
1780       llvm::outs() << TC.getCompilerRT(C.getArgs(), "builtins") << "\n";
1781       break;
1782     case ToolChain::RLT_Libgcc:
1783       llvm::outs() << GetFilePath("libgcc.a", TC) << "\n";
1784       break;
1785     }
1786     return false;
1787   }
1788 
1789   if (C.getArgs().hasArg(options::OPT_print_multi_lib)) {
1790     for (const Multilib &Multilib : TC.getMultilibs())
1791       llvm::outs() << Multilib << "\n";
1792     return false;
1793   }
1794 
1795   if (C.getArgs().hasArg(options::OPT_print_multi_directory)) {
1796     const Multilib &Multilib = TC.getMultilib();
1797     if (Multilib.gccSuffix().empty())
1798       llvm::outs() << ".\n";
1799     else {
1800       StringRef Suffix(Multilib.gccSuffix());
1801       assert(Suffix.front() == '/');
1802       llvm::outs() << Suffix.substr(1) << "\n";
1803     }
1804     return false;
1805   }
1806 
1807   if (C.getArgs().hasArg(options::OPT_print_target_triple)) {
1808     llvm::outs() << TC.getTripleString() << "\n";
1809     return false;
1810   }
1811 
1812   if (C.getArgs().hasArg(options::OPT_print_effective_triple)) {
1813     const llvm::Triple Triple(TC.ComputeEffectiveClangTriple(C.getArgs()));
1814     llvm::outs() << Triple.getTriple() << "\n";
1815     return false;
1816   }
1817 
1818   if (C.getArgs().hasArg(options::OPT_print_targets)) {
1819     llvm::TargetRegistry::printRegisteredTargetsForVersion(llvm::outs());
1820     return false;
1821   }
1822 
1823   return true;
1824 }
1825 
1826 enum {
1827   TopLevelAction = 0,
1828   HeadSibAction = 1,
1829   OtherSibAction = 2,
1830 };
1831 
1832 // Display an action graph human-readably.  Action A is the "sink" node
1833 // and latest-occuring action. Traversal is in pre-order, visiting the
1834 // inputs to each action before printing the action itself.
1835 static unsigned PrintActions1(const Compilation &C, Action *A,
1836                               std::map<Action *, unsigned> &Ids,
1837                               Twine Indent = {}, int Kind = TopLevelAction) {
1838   if (Ids.count(A)) // A was already visited.
1839     return Ids[A];
1840 
1841   std::string str;
1842   llvm::raw_string_ostream os(str);
1843 
1844   auto getSibIndent = [](int K) -> Twine {
1845     return (K == HeadSibAction) ? "   " : (K == OtherSibAction) ? "|  " : "";
1846   };
1847 
1848   Twine SibIndent = Indent + getSibIndent(Kind);
1849   int SibKind = HeadSibAction;
1850   os << Action::getClassName(A->getKind()) << ", ";
1851   if (InputAction *IA = dyn_cast<InputAction>(A)) {
1852     os << "\"" << IA->getInputArg().getValue() << "\"";
1853   } else if (BindArchAction *BIA = dyn_cast<BindArchAction>(A)) {
1854     os << '"' << BIA->getArchName() << '"' << ", {"
1855        << PrintActions1(C, *BIA->input_begin(), Ids, SibIndent, SibKind) << "}";
1856   } else if (OffloadAction *OA = dyn_cast<OffloadAction>(A)) {
1857     bool IsFirst = true;
1858     OA->doOnEachDependence(
1859         [&](Action *A, const ToolChain *TC, const char *BoundArch) {
1860           assert(TC && "Unknown host toolchain");
1861           // E.g. for two CUDA device dependences whose bound arch is sm_20 and
1862           // sm_35 this will generate:
1863           // "cuda-device" (nvptx64-nvidia-cuda:sm_20) {#ID}, "cuda-device"
1864           // (nvptx64-nvidia-cuda:sm_35) {#ID}
1865           if (!IsFirst)
1866             os << ", ";
1867           os << '"';
1868           os << A->getOffloadingKindPrefix();
1869           os << " (";
1870           os << TC->getTriple().normalize();
1871           if (BoundArch)
1872             os << ":" << BoundArch;
1873           os << ")";
1874           os << '"';
1875           os << " {" << PrintActions1(C, A, Ids, SibIndent, SibKind) << "}";
1876           IsFirst = false;
1877           SibKind = OtherSibAction;
1878         });
1879   } else {
1880     const ActionList *AL = &A->getInputs();
1881 
1882     if (AL->size()) {
1883       const char *Prefix = "{";
1884       for (Action *PreRequisite : *AL) {
1885         os << Prefix << PrintActions1(C, PreRequisite, Ids, SibIndent, SibKind);
1886         Prefix = ", ";
1887         SibKind = OtherSibAction;
1888       }
1889       os << "}";
1890     } else
1891       os << "{}";
1892   }
1893 
1894   // Append offload info for all options other than the offloading action
1895   // itself (e.g. (cuda-device, sm_20) or (cuda-host)).
1896   std::string offload_str;
1897   llvm::raw_string_ostream offload_os(offload_str);
1898   if (!isa<OffloadAction>(A)) {
1899     auto S = A->getOffloadingKindPrefix();
1900     if (!S.empty()) {
1901       offload_os << ", (" << S;
1902       if (A->getOffloadingArch())
1903         offload_os << ", " << A->getOffloadingArch();
1904       offload_os << ")";
1905     }
1906   }
1907 
1908   auto getSelfIndent = [](int K) -> Twine {
1909     return (K == HeadSibAction) ? "+- " : (K == OtherSibAction) ? "|- " : "";
1910   };
1911 
1912   unsigned Id = Ids.size();
1913   Ids[A] = Id;
1914   llvm::errs() << Indent + getSelfIndent(Kind) << Id << ": " << os.str() << ", "
1915                << types::getTypeName(A->getType()) << offload_os.str() << "\n";
1916 
1917   return Id;
1918 }
1919 
1920 // Print the action graphs in a compilation C.
1921 // For example "clang -c file1.c file2.c" is composed of two subgraphs.
1922 void Driver::PrintActions(const Compilation &C) const {
1923   std::map<Action *, unsigned> Ids;
1924   for (Action *A : C.getActions())
1925     PrintActions1(C, A, Ids);
1926 }
1927 
1928 /// Check whether the given input tree contains any compilation or
1929 /// assembly actions.
1930 static bool ContainsCompileOrAssembleAction(const Action *A) {
1931   if (isa<CompileJobAction>(A) || isa<BackendJobAction>(A) ||
1932       isa<AssembleJobAction>(A))
1933     return true;
1934 
1935   for (const Action *Input : A->inputs())
1936     if (ContainsCompileOrAssembleAction(Input))
1937       return true;
1938 
1939   return false;
1940 }
1941 
1942 void Driver::BuildUniversalActions(Compilation &C, const ToolChain &TC,
1943                                    const InputList &BAInputs) const {
1944   DerivedArgList &Args = C.getArgs();
1945   ActionList &Actions = C.getActions();
1946   llvm::PrettyStackTraceString CrashInfo("Building universal build actions");
1947   // Collect the list of architectures. Duplicates are allowed, but should only
1948   // be handled once (in the order seen).
1949   llvm::StringSet<> ArchNames;
1950   SmallVector<const char *, 4> Archs;
1951   for (Arg *A : Args) {
1952     if (A->getOption().matches(options::OPT_arch)) {
1953       // Validate the option here; we don't save the type here because its
1954       // particular spelling may participate in other driver choices.
1955       llvm::Triple::ArchType Arch =
1956           tools::darwin::getArchTypeForMachOArchName(A->getValue());
1957       if (Arch == llvm::Triple::UnknownArch) {
1958         Diag(clang::diag::err_drv_invalid_arch_name) << A->getAsString(Args);
1959         continue;
1960       }
1961 
1962       A->claim();
1963       if (ArchNames.insert(A->getValue()).second)
1964         Archs.push_back(A->getValue());
1965     }
1966   }
1967 
1968   // When there is no explicit arch for this platform, make sure we still bind
1969   // the architecture (to the default) so that -Xarch_ is handled correctly.
1970   if (!Archs.size())
1971     Archs.push_back(Args.MakeArgString(TC.getDefaultUniversalArchName()));
1972 
1973   ActionList SingleActions;
1974   BuildActions(C, Args, BAInputs, SingleActions);
1975 
1976   // Add in arch bindings for every top level action, as well as lipo and
1977   // dsymutil steps if needed.
1978   for (Action* Act : SingleActions) {
1979     // Make sure we can lipo this kind of output. If not (and it is an actual
1980     // output) then we disallow, since we can't create an output file with the
1981     // right name without overwriting it. We could remove this oddity by just
1982     // changing the output names to include the arch, which would also fix
1983     // -save-temps. Compatibility wins for now.
1984 
1985     if (Archs.size() > 1 && !types::canLipoType(Act->getType()))
1986       Diag(clang::diag::err_drv_invalid_output_with_multiple_archs)
1987           << types::getTypeName(Act->getType());
1988 
1989     ActionList Inputs;
1990     for (unsigned i = 0, e = Archs.size(); i != e; ++i)
1991       Inputs.push_back(C.MakeAction<BindArchAction>(Act, Archs[i]));
1992 
1993     // Lipo if necessary, we do it this way because we need to set the arch flag
1994     // so that -Xarch_ gets overwritten.
1995     if (Inputs.size() == 1 || Act->getType() == types::TY_Nothing)
1996       Actions.append(Inputs.begin(), Inputs.end());
1997     else
1998       Actions.push_back(C.MakeAction<LipoJobAction>(Inputs, Act->getType()));
1999 
2000     // Handle debug info queries.
2001     Arg *A = Args.getLastArg(options::OPT_g_Group);
2002     bool enablesDebugInfo = A && !A->getOption().matches(options::OPT_g0) &&
2003                             !A->getOption().matches(options::OPT_gstabs);
2004     if ((enablesDebugInfo || willEmitRemarks(Args)) &&
2005         ContainsCompileOrAssembleAction(Actions.back())) {
2006 
2007       // Add a 'dsymutil' step if necessary, when debug info is enabled and we
2008       // have a compile input. We need to run 'dsymutil' ourselves in such cases
2009       // because the debug info will refer to a temporary object file which
2010       // will be removed at the end of the compilation process.
2011       if (Act->getType() == types::TY_Image) {
2012         ActionList Inputs;
2013         Inputs.push_back(Actions.back());
2014         Actions.pop_back();
2015         Actions.push_back(
2016             C.MakeAction<DsymutilJobAction>(Inputs, types::TY_dSYM));
2017       }
2018 
2019       // Verify the debug info output.
2020       if (Args.hasArg(options::OPT_verify_debug_info)) {
2021         Action* LastAction = Actions.back();
2022         Actions.pop_back();
2023         Actions.push_back(C.MakeAction<VerifyDebugInfoJobAction>(
2024             LastAction, types::TY_Nothing));
2025       }
2026     }
2027   }
2028 }
2029 
2030 bool Driver::DiagnoseInputExistence(const DerivedArgList &Args, StringRef Value,
2031                                     types::ID Ty, bool TypoCorrect) const {
2032   if (!getCheckInputsExist())
2033     return true;
2034 
2035   // stdin always exists.
2036   if (Value == "-")
2037     return true;
2038 
2039   if (getVFS().exists(Value))
2040     return true;
2041 
2042   if (IsCLMode()) {
2043     if (!llvm::sys::path::is_absolute(Twine(Value)) &&
2044         llvm::sys::Process::FindInEnvPath("LIB", Value))
2045       return true;
2046 
2047     if (Args.hasArg(options::OPT__SLASH_link) && Ty == types::TY_Object) {
2048       // Arguments to the /link flag might cause the linker to search for object
2049       // and library files in paths we don't know about. Don't error in such
2050       // cases.
2051       return true;
2052     }
2053   }
2054 
2055   if (TypoCorrect) {
2056     // Check if the filename is a typo for an option flag. OptTable thinks
2057     // that all args that are not known options and that start with / are
2058     // filenames, but e.g. `/diagnostic:caret` is more likely a typo for
2059     // the option `/diagnostics:caret` than a reference to a file in the root
2060     // directory.
2061     unsigned IncludedFlagsBitmask;
2062     unsigned ExcludedFlagsBitmask;
2063     std::tie(IncludedFlagsBitmask, ExcludedFlagsBitmask) =
2064         getIncludeExcludeOptionFlagMasks(IsCLMode());
2065     std::string Nearest;
2066     if (getOpts().findNearest(Value, Nearest, IncludedFlagsBitmask,
2067                               ExcludedFlagsBitmask) <= 1) {
2068       Diag(clang::diag::err_drv_no_such_file_with_suggestion)
2069           << Value << Nearest;
2070       return false;
2071     }
2072   }
2073 
2074   Diag(clang::diag::err_drv_no_such_file) << Value;
2075   return false;
2076 }
2077 
2078 // Construct a the list of inputs and their types.
2079 void Driver::BuildInputs(const ToolChain &TC, DerivedArgList &Args,
2080                          InputList &Inputs) const {
2081   const llvm::opt::OptTable &Opts = getOpts();
2082   // Track the current user specified (-x) input. We also explicitly track the
2083   // argument used to set the type; we only want to claim the type when we
2084   // actually use it, so we warn about unused -x arguments.
2085   types::ID InputType = types::TY_Nothing;
2086   Arg *InputTypeArg = nullptr;
2087 
2088   // The last /TC or /TP option sets the input type to C or C++ globally.
2089   if (Arg *TCTP = Args.getLastArgNoClaim(options::OPT__SLASH_TC,
2090                                          options::OPT__SLASH_TP)) {
2091     InputTypeArg = TCTP;
2092     InputType = TCTP->getOption().matches(options::OPT__SLASH_TC)
2093                     ? types::TY_C
2094                     : types::TY_CXX;
2095 
2096     Arg *Previous = nullptr;
2097     bool ShowNote = false;
2098     for (Arg *A :
2099          Args.filtered(options::OPT__SLASH_TC, options::OPT__SLASH_TP)) {
2100       if (Previous) {
2101         Diag(clang::diag::warn_drv_overriding_flag_option)
2102           << Previous->getSpelling() << A->getSpelling();
2103         ShowNote = true;
2104       }
2105       Previous = A;
2106     }
2107     if (ShowNote)
2108       Diag(clang::diag::note_drv_t_option_is_global);
2109 
2110     // No driver mode exposes -x and /TC or /TP; we don't support mixing them.
2111     assert(!Args.hasArg(options::OPT_x) && "-x and /TC or /TP is not allowed");
2112   }
2113 
2114   for (Arg *A : Args) {
2115     if (A->getOption().getKind() == Option::InputClass) {
2116       const char *Value = A->getValue();
2117       types::ID Ty = types::TY_INVALID;
2118 
2119       // Infer the input type if necessary.
2120       if (InputType == types::TY_Nothing) {
2121         // If there was an explicit arg for this, claim it.
2122         if (InputTypeArg)
2123           InputTypeArg->claim();
2124 
2125         // stdin must be handled specially.
2126         if (memcmp(Value, "-", 2) == 0) {
2127           // If running with -E, treat as a C input (this changes the builtin
2128           // macros, for example). This may be overridden by -ObjC below.
2129           //
2130           // Otherwise emit an error but still use a valid type to avoid
2131           // spurious errors (e.g., no inputs).
2132           if (!Args.hasArgNoClaim(options::OPT_E) && !CCCIsCPP())
2133             Diag(IsCLMode() ? clang::diag::err_drv_unknown_stdin_type_clang_cl
2134                             : clang::diag::err_drv_unknown_stdin_type);
2135           Ty = types::TY_C;
2136         } else {
2137           // Otherwise lookup by extension.
2138           // Fallback is C if invoked as C preprocessor, C++ if invoked with
2139           // clang-cl /E, or Object otherwise.
2140           // We use a host hook here because Darwin at least has its own
2141           // idea of what .s is.
2142           if (const char *Ext = strrchr(Value, '.'))
2143             Ty = TC.LookupTypeForExtension(Ext + 1);
2144 
2145           if (Ty == types::TY_INVALID) {
2146             if (CCCIsCPP())
2147               Ty = types::TY_C;
2148             else if (IsCLMode() && Args.hasArgNoClaim(options::OPT_E))
2149               Ty = types::TY_CXX;
2150             else
2151               Ty = types::TY_Object;
2152           }
2153 
2154           // If the driver is invoked as C++ compiler (like clang++ or c++) it
2155           // should autodetect some input files as C++ for g++ compatibility.
2156           if (CCCIsCXX()) {
2157             types::ID OldTy = Ty;
2158             Ty = types::lookupCXXTypeForCType(Ty);
2159 
2160             if (Ty != OldTy)
2161               Diag(clang::diag::warn_drv_treating_input_as_cxx)
2162                   << getTypeName(OldTy) << getTypeName(Ty);
2163           }
2164 
2165           // If running with -fthinlto-index=, extensions that normally identify
2166           // native object files actually identify LLVM bitcode files.
2167           if (Args.hasArgNoClaim(options::OPT_fthinlto_index_EQ) &&
2168               Ty == types::TY_Object)
2169             Ty = types::TY_LLVM_BC;
2170         }
2171 
2172         // -ObjC and -ObjC++ override the default language, but only for "source
2173         // files". We just treat everything that isn't a linker input as a
2174         // source file.
2175         //
2176         // FIXME: Clean this up if we move the phase sequence into the type.
2177         if (Ty != types::TY_Object) {
2178           if (Args.hasArg(options::OPT_ObjC))
2179             Ty = types::TY_ObjC;
2180           else if (Args.hasArg(options::OPT_ObjCXX))
2181             Ty = types::TY_ObjCXX;
2182         }
2183       } else {
2184         assert(InputTypeArg && "InputType set w/o InputTypeArg");
2185         if (!InputTypeArg->getOption().matches(options::OPT_x)) {
2186           // If emulating cl.exe, make sure that /TC and /TP don't affect input
2187           // object files.
2188           const char *Ext = strrchr(Value, '.');
2189           if (Ext && TC.LookupTypeForExtension(Ext + 1) == types::TY_Object)
2190             Ty = types::TY_Object;
2191         }
2192         if (Ty == types::TY_INVALID) {
2193           Ty = InputType;
2194           InputTypeArg->claim();
2195         }
2196       }
2197 
2198       if (DiagnoseInputExistence(Args, Value, Ty, /*TypoCorrect=*/true))
2199         Inputs.push_back(std::make_pair(Ty, A));
2200 
2201     } else if (A->getOption().matches(options::OPT__SLASH_Tc)) {
2202       StringRef Value = A->getValue();
2203       if (DiagnoseInputExistence(Args, Value, types::TY_C,
2204                                  /*TypoCorrect=*/false)) {
2205         Arg *InputArg = MakeInputArg(Args, Opts, A->getValue());
2206         Inputs.push_back(std::make_pair(types::TY_C, InputArg));
2207       }
2208       A->claim();
2209     } else if (A->getOption().matches(options::OPT__SLASH_Tp)) {
2210       StringRef Value = A->getValue();
2211       if (DiagnoseInputExistence(Args, Value, types::TY_CXX,
2212                                  /*TypoCorrect=*/false)) {
2213         Arg *InputArg = MakeInputArg(Args, Opts, A->getValue());
2214         Inputs.push_back(std::make_pair(types::TY_CXX, InputArg));
2215       }
2216       A->claim();
2217     } else if (A->getOption().hasFlag(options::LinkerInput)) {
2218       // Just treat as object type, we could make a special type for this if
2219       // necessary.
2220       Inputs.push_back(std::make_pair(types::TY_Object, A));
2221 
2222     } else if (A->getOption().matches(options::OPT_x)) {
2223       InputTypeArg = A;
2224       InputType = types::lookupTypeForTypeSpecifier(A->getValue());
2225       A->claim();
2226 
2227       // Follow gcc behavior and treat as linker input for invalid -x
2228       // options. Its not clear why we shouldn't just revert to unknown; but
2229       // this isn't very important, we might as well be bug compatible.
2230       if (!InputType) {
2231         Diag(clang::diag::err_drv_unknown_language) << A->getValue();
2232         InputType = types::TY_Object;
2233       }
2234     } else if (A->getOption().getID() == options::OPT_U) {
2235       assert(A->getNumValues() == 1 && "The /U option has one value.");
2236       StringRef Val = A->getValue(0);
2237       if (Val.find_first_of("/\\") != StringRef::npos) {
2238         // Warn about e.g. "/Users/me/myfile.c".
2239         Diag(diag::warn_slash_u_filename) << Val;
2240         Diag(diag::note_use_dashdash);
2241       }
2242     }
2243   }
2244   if (CCCIsCPP() && Inputs.empty()) {
2245     // If called as standalone preprocessor, stdin is processed
2246     // if no other input is present.
2247     Arg *A = MakeInputArg(Args, Opts, "-");
2248     Inputs.push_back(std::make_pair(types::TY_C, A));
2249   }
2250 }
2251 
2252 namespace {
2253 /// Provides a convenient interface for different programming models to generate
2254 /// the required device actions.
2255 class OffloadingActionBuilder final {
2256   /// Flag used to trace errors in the builder.
2257   bool IsValid = false;
2258 
2259   /// The compilation that is using this builder.
2260   Compilation &C;
2261 
2262   /// Map between an input argument and the offload kinds used to process it.
2263   std::map<const Arg *, unsigned> InputArgToOffloadKindMap;
2264 
2265   /// Builder interface. It doesn't build anything or keep any state.
2266   class DeviceActionBuilder {
2267   public:
2268     typedef const llvm::SmallVectorImpl<phases::ID> PhasesTy;
2269 
2270     enum ActionBuilderReturnCode {
2271       // The builder acted successfully on the current action.
2272       ABRT_Success,
2273       // The builder didn't have to act on the current action.
2274       ABRT_Inactive,
2275       // The builder was successful and requested the host action to not be
2276       // generated.
2277       ABRT_Ignore_Host,
2278     };
2279 
2280   protected:
2281     /// Compilation associated with this builder.
2282     Compilation &C;
2283 
2284     /// Tool chains associated with this builder. The same programming
2285     /// model may have associated one or more tool chains.
2286     SmallVector<const ToolChain *, 2> ToolChains;
2287 
2288     /// The derived arguments associated with this builder.
2289     DerivedArgList &Args;
2290 
2291     /// The inputs associated with this builder.
2292     const Driver::InputList &Inputs;
2293 
2294     /// The associated offload kind.
2295     Action::OffloadKind AssociatedOffloadKind = Action::OFK_None;
2296 
2297   public:
2298     DeviceActionBuilder(Compilation &C, DerivedArgList &Args,
2299                         const Driver::InputList &Inputs,
2300                         Action::OffloadKind AssociatedOffloadKind)
2301         : C(C), Args(Args), Inputs(Inputs),
2302           AssociatedOffloadKind(AssociatedOffloadKind) {}
2303     virtual ~DeviceActionBuilder() {}
2304 
2305     /// Fill up the array \a DA with all the device dependences that should be
2306     /// added to the provided host action \a HostAction. By default it is
2307     /// inactive.
2308     virtual ActionBuilderReturnCode
2309     getDeviceDependences(OffloadAction::DeviceDependences &DA,
2310                          phases::ID CurPhase, phases::ID FinalPhase,
2311                          PhasesTy &Phases) {
2312       return ABRT_Inactive;
2313     }
2314 
2315     /// Update the state to include the provided host action \a HostAction as a
2316     /// dependency of the current device action. By default it is inactive.
2317     virtual ActionBuilderReturnCode addDeviceDepences(Action *HostAction) {
2318       return ABRT_Inactive;
2319     }
2320 
2321     /// Append top level actions generated by the builder.
2322     virtual void appendTopLevelActions(ActionList &AL) {}
2323 
2324     /// Append linker actions generated by the builder.
2325     virtual void appendLinkActions(ActionList &AL) {}
2326 
2327     /// Append linker actions generated by the builder.
2328     virtual void appendLinkDependences(OffloadAction::DeviceDependences &DA) {}
2329 
2330     /// Initialize the builder. Return true if any initialization errors are
2331     /// found.
2332     virtual bool initialize() { return false; }
2333 
2334     /// Return true if the builder can use bundling/unbundling.
2335     virtual bool canUseBundlerUnbundler() const { return false; }
2336 
2337     /// Return true if this builder is valid. We have a valid builder if we have
2338     /// associated device tool chains.
2339     bool isValid() { return !ToolChains.empty(); }
2340 
2341     /// Return the associated offload kind.
2342     Action::OffloadKind getAssociatedOffloadKind() {
2343       return AssociatedOffloadKind;
2344     }
2345   };
2346 
2347   /// Base class for CUDA/HIP action builder. It injects device code in
2348   /// the host backend action.
2349   class CudaActionBuilderBase : public DeviceActionBuilder {
2350   protected:
2351     /// Flags to signal if the user requested host-only or device-only
2352     /// compilation.
2353     bool CompileHostOnly = false;
2354     bool CompileDeviceOnly = false;
2355     bool EmitLLVM = false;
2356     bool EmitAsm = false;
2357 
2358     /// List of GPU architectures to use in this compilation.
2359     SmallVector<CudaArch, 4> GpuArchList;
2360 
2361     /// The CUDA actions for the current input.
2362     ActionList CudaDeviceActions;
2363 
2364     /// The CUDA fat binary if it was generated for the current input.
2365     Action *CudaFatBinary = nullptr;
2366 
2367     /// Flag that is set to true if this builder acted on the current input.
2368     bool IsActive = false;
2369 
2370     /// Flag for -fgpu-rdc.
2371     bool Relocatable = false;
2372 
2373     /// Default GPU architecture if there's no one specified.
2374     CudaArch DefaultCudaArch = CudaArch::UNKNOWN;
2375 
2376   public:
2377     CudaActionBuilderBase(Compilation &C, DerivedArgList &Args,
2378                           const Driver::InputList &Inputs,
2379                           Action::OffloadKind OFKind)
2380         : DeviceActionBuilder(C, Args, Inputs, OFKind) {}
2381 
2382     ActionBuilderReturnCode addDeviceDepences(Action *HostAction) override {
2383       // While generating code for CUDA, we only depend on the host input action
2384       // to trigger the creation of all the CUDA device actions.
2385 
2386       // If we are dealing with an input action, replicate it for each GPU
2387       // architecture. If we are in host-only mode we return 'success' so that
2388       // the host uses the CUDA offload kind.
2389       if (auto *IA = dyn_cast<InputAction>(HostAction)) {
2390         assert(!GpuArchList.empty() &&
2391                "We should have at least one GPU architecture.");
2392 
2393         // If the host input is not CUDA or HIP, we don't need to bother about
2394         // this input.
2395         if (IA->getType() != types::TY_CUDA &&
2396             IA->getType() != types::TY_HIP) {
2397           // The builder will ignore this input.
2398           IsActive = false;
2399           return ABRT_Inactive;
2400         }
2401 
2402         // Set the flag to true, so that the builder acts on the current input.
2403         IsActive = true;
2404 
2405         if (CompileHostOnly)
2406           return ABRT_Success;
2407 
2408         // Replicate inputs for each GPU architecture.
2409         auto Ty = IA->getType() == types::TY_HIP ? types::TY_HIP_DEVICE
2410                                                  : types::TY_CUDA_DEVICE;
2411         for (unsigned I = 0, E = GpuArchList.size(); I != E; ++I) {
2412           CudaDeviceActions.push_back(
2413               C.MakeAction<InputAction>(IA->getInputArg(), Ty));
2414         }
2415 
2416         return ABRT_Success;
2417       }
2418 
2419       // If this is an unbundling action use it as is for each CUDA toolchain.
2420       if (auto *UA = dyn_cast<OffloadUnbundlingJobAction>(HostAction)) {
2421 
2422         // If -fgpu-rdc is disabled, should not unbundle since there is no
2423         // device code to link.
2424         if (!Relocatable)
2425           return ABRT_Inactive;
2426 
2427         CudaDeviceActions.clear();
2428         auto *IA = cast<InputAction>(UA->getInputs().back());
2429         std::string FileName = IA->getInputArg().getAsString(Args);
2430         // Check if the type of the file is the same as the action. Do not
2431         // unbundle it if it is not. Do not unbundle .so files, for example,
2432         // which are not object files.
2433         if (IA->getType() == types::TY_Object &&
2434             (!llvm::sys::path::has_extension(FileName) ||
2435              types::lookupTypeForExtension(
2436                  llvm::sys::path::extension(FileName).drop_front()) !=
2437                  types::TY_Object))
2438           return ABRT_Inactive;
2439 
2440         for (auto Arch : GpuArchList) {
2441           CudaDeviceActions.push_back(UA);
2442           UA->registerDependentActionInfo(ToolChains[0], CudaArchToString(Arch),
2443                                           AssociatedOffloadKind);
2444         }
2445         return ABRT_Success;
2446       }
2447 
2448       return IsActive ? ABRT_Success : ABRT_Inactive;
2449     }
2450 
2451     void appendTopLevelActions(ActionList &AL) override {
2452       // Utility to append actions to the top level list.
2453       auto AddTopLevel = [&](Action *A, CudaArch BoundArch) {
2454         OffloadAction::DeviceDependences Dep;
2455         Dep.add(*A, *ToolChains.front(), CudaArchToString(BoundArch),
2456                 AssociatedOffloadKind);
2457         AL.push_back(C.MakeAction<OffloadAction>(Dep, A->getType()));
2458       };
2459 
2460       // If we have a fat binary, add it to the list.
2461       if (CudaFatBinary) {
2462         AddTopLevel(CudaFatBinary, CudaArch::UNKNOWN);
2463         CudaDeviceActions.clear();
2464         CudaFatBinary = nullptr;
2465         return;
2466       }
2467 
2468       if (CudaDeviceActions.empty())
2469         return;
2470 
2471       // If we have CUDA actions at this point, that's because we have a have
2472       // partial compilation, so we should have an action for each GPU
2473       // architecture.
2474       assert(CudaDeviceActions.size() == GpuArchList.size() &&
2475              "Expecting one action per GPU architecture.");
2476       assert(ToolChains.size() == 1 &&
2477              "Expecting to have a sing CUDA toolchain.");
2478       for (unsigned I = 0, E = GpuArchList.size(); I != E; ++I)
2479         AddTopLevel(CudaDeviceActions[I], GpuArchList[I]);
2480 
2481       CudaDeviceActions.clear();
2482     }
2483 
2484     bool initialize() override {
2485       assert(AssociatedOffloadKind == Action::OFK_Cuda ||
2486              AssociatedOffloadKind == Action::OFK_HIP);
2487 
2488       // We don't need to support CUDA.
2489       if (AssociatedOffloadKind == Action::OFK_Cuda &&
2490           !C.hasOffloadToolChain<Action::OFK_Cuda>())
2491         return false;
2492 
2493       // We don't need to support HIP.
2494       if (AssociatedOffloadKind == Action::OFK_HIP &&
2495           !C.hasOffloadToolChain<Action::OFK_HIP>())
2496         return false;
2497 
2498       Relocatable = Args.hasFlag(options::OPT_fgpu_rdc,
2499           options::OPT_fno_gpu_rdc, /*Default=*/false);
2500 
2501       const ToolChain *HostTC = C.getSingleOffloadToolChain<Action::OFK_Host>();
2502       assert(HostTC && "No toolchain for host compilation.");
2503       if (HostTC->getTriple().isNVPTX() ||
2504           HostTC->getTriple().getArch() == llvm::Triple::amdgcn) {
2505         // We do not support targeting NVPTX/AMDGCN for host compilation. Throw
2506         // an error and abort pipeline construction early so we don't trip
2507         // asserts that assume device-side compilation.
2508         C.getDriver().Diag(diag::err_drv_cuda_host_arch)
2509             << HostTC->getTriple().getArchName();
2510         return true;
2511       }
2512 
2513       ToolChains.push_back(
2514           AssociatedOffloadKind == Action::OFK_Cuda
2515               ? C.getSingleOffloadToolChain<Action::OFK_Cuda>()
2516               : C.getSingleOffloadToolChain<Action::OFK_HIP>());
2517 
2518       Arg *PartialCompilationArg = Args.getLastArg(
2519           options::OPT_cuda_host_only, options::OPT_cuda_device_only,
2520           options::OPT_cuda_compile_host_device);
2521       CompileHostOnly = PartialCompilationArg &&
2522                         PartialCompilationArg->getOption().matches(
2523                             options::OPT_cuda_host_only);
2524       CompileDeviceOnly = PartialCompilationArg &&
2525                           PartialCompilationArg->getOption().matches(
2526                               options::OPT_cuda_device_only);
2527       EmitLLVM = Args.getLastArg(options::OPT_emit_llvm);
2528       EmitAsm = Args.getLastArg(options::OPT_S);
2529 
2530       // Collect all cuda_gpu_arch parameters, removing duplicates.
2531       std::set<CudaArch> GpuArchs;
2532       bool Error = false;
2533       for (Arg *A : Args) {
2534         if (!(A->getOption().matches(options::OPT_offload_arch_EQ) ||
2535               A->getOption().matches(options::OPT_no_offload_arch_EQ)))
2536           continue;
2537         A->claim();
2538 
2539         const StringRef ArchStr = A->getValue();
2540         if (A->getOption().matches(options::OPT_no_offload_arch_EQ) &&
2541             ArchStr == "all") {
2542           GpuArchs.clear();
2543           continue;
2544         }
2545         CudaArch Arch = StringToCudaArch(ArchStr);
2546         if (Arch == CudaArch::UNKNOWN) {
2547           C.getDriver().Diag(clang::diag::err_drv_cuda_bad_gpu_arch) << ArchStr;
2548           Error = true;
2549         } else if (A->getOption().matches(options::OPT_offload_arch_EQ))
2550           GpuArchs.insert(Arch);
2551         else if (A->getOption().matches(options::OPT_no_offload_arch_EQ))
2552           GpuArchs.erase(Arch);
2553         else
2554           llvm_unreachable("Unexpected option.");
2555       }
2556 
2557       // Collect list of GPUs remaining in the set.
2558       for (CudaArch Arch : GpuArchs)
2559         GpuArchList.push_back(Arch);
2560 
2561       // Default to sm_20 which is the lowest common denominator for
2562       // supported GPUs.  sm_20 code should work correctly, if
2563       // suboptimally, on all newer GPUs.
2564       if (GpuArchList.empty())
2565         GpuArchList.push_back(DefaultCudaArch);
2566 
2567       return Error;
2568     }
2569   };
2570 
2571   /// \brief CUDA action builder. It injects device code in the host backend
2572   /// action.
2573   class CudaActionBuilder final : public CudaActionBuilderBase {
2574   public:
2575     CudaActionBuilder(Compilation &C, DerivedArgList &Args,
2576                       const Driver::InputList &Inputs)
2577         : CudaActionBuilderBase(C, Args, Inputs, Action::OFK_Cuda) {
2578       DefaultCudaArch = CudaArch::SM_20;
2579     }
2580 
2581     ActionBuilderReturnCode
2582     getDeviceDependences(OffloadAction::DeviceDependences &DA,
2583                          phases::ID CurPhase, phases::ID FinalPhase,
2584                          PhasesTy &Phases) override {
2585       if (!IsActive)
2586         return ABRT_Inactive;
2587 
2588       // If we don't have more CUDA actions, we don't have any dependences to
2589       // create for the host.
2590       if (CudaDeviceActions.empty())
2591         return ABRT_Success;
2592 
2593       assert(CudaDeviceActions.size() == GpuArchList.size() &&
2594              "Expecting one action per GPU architecture.");
2595       assert(!CompileHostOnly &&
2596              "Not expecting CUDA actions in host-only compilation.");
2597 
2598       // If we are generating code for the device or we are in a backend phase,
2599       // we attempt to generate the fat binary. We compile each arch to ptx and
2600       // assemble to cubin, then feed the cubin *and* the ptx into a device
2601       // "link" action, which uses fatbinary to combine these cubins into one
2602       // fatbin.  The fatbin is then an input to the host action if not in
2603       // device-only mode.
2604       if (CompileDeviceOnly || CurPhase == phases::Backend) {
2605         ActionList DeviceActions;
2606         for (unsigned I = 0, E = GpuArchList.size(); I != E; ++I) {
2607           // Produce the device action from the current phase up to the assemble
2608           // phase.
2609           for (auto Ph : Phases) {
2610             // Skip the phases that were already dealt with.
2611             if (Ph < CurPhase)
2612               continue;
2613             // We have to be consistent with the host final phase.
2614             if (Ph > FinalPhase)
2615               break;
2616 
2617             CudaDeviceActions[I] = C.getDriver().ConstructPhaseAction(
2618                 C, Args, Ph, CudaDeviceActions[I], Action::OFK_Cuda);
2619 
2620             if (Ph == phases::Assemble)
2621               break;
2622           }
2623 
2624           // If we didn't reach the assemble phase, we can't generate the fat
2625           // binary. We don't need to generate the fat binary if we are not in
2626           // device-only mode.
2627           if (!isa<AssembleJobAction>(CudaDeviceActions[I]) ||
2628               CompileDeviceOnly)
2629             continue;
2630 
2631           Action *AssembleAction = CudaDeviceActions[I];
2632           assert(AssembleAction->getType() == types::TY_Object);
2633           assert(AssembleAction->getInputs().size() == 1);
2634 
2635           Action *BackendAction = AssembleAction->getInputs()[0];
2636           assert(BackendAction->getType() == types::TY_PP_Asm);
2637 
2638           for (auto &A : {AssembleAction, BackendAction}) {
2639             OffloadAction::DeviceDependences DDep;
2640             DDep.add(*A, *ToolChains.front(), CudaArchToString(GpuArchList[I]),
2641                      Action::OFK_Cuda);
2642             DeviceActions.push_back(
2643                 C.MakeAction<OffloadAction>(DDep, A->getType()));
2644           }
2645         }
2646 
2647         // We generate the fat binary if we have device input actions.
2648         if (!DeviceActions.empty()) {
2649           CudaFatBinary =
2650               C.MakeAction<LinkJobAction>(DeviceActions, types::TY_CUDA_FATBIN);
2651 
2652           if (!CompileDeviceOnly) {
2653             DA.add(*CudaFatBinary, *ToolChains.front(), /*BoundArch=*/nullptr,
2654                    Action::OFK_Cuda);
2655             // Clear the fat binary, it is already a dependence to an host
2656             // action.
2657             CudaFatBinary = nullptr;
2658           }
2659 
2660           // Remove the CUDA actions as they are already connected to an host
2661           // action or fat binary.
2662           CudaDeviceActions.clear();
2663         }
2664 
2665         // We avoid creating host action in device-only mode.
2666         return CompileDeviceOnly ? ABRT_Ignore_Host : ABRT_Success;
2667       } else if (CurPhase > phases::Backend) {
2668         // If we are past the backend phase and still have a device action, we
2669         // don't have to do anything as this action is already a device
2670         // top-level action.
2671         return ABRT_Success;
2672       }
2673 
2674       assert(CurPhase < phases::Backend && "Generating single CUDA "
2675                                            "instructions should only occur "
2676                                            "before the backend phase!");
2677 
2678       // By default, we produce an action for each device arch.
2679       for (Action *&A : CudaDeviceActions)
2680         A = C.getDriver().ConstructPhaseAction(C, Args, CurPhase, A);
2681 
2682       return ABRT_Success;
2683     }
2684   };
2685   /// \brief HIP action builder. It injects device code in the host backend
2686   /// action.
2687   class HIPActionBuilder final : public CudaActionBuilderBase {
2688     /// The linker inputs obtained for each device arch.
2689     SmallVector<ActionList, 8> DeviceLinkerInputs;
2690 
2691   public:
2692     HIPActionBuilder(Compilation &C, DerivedArgList &Args,
2693                      const Driver::InputList &Inputs)
2694         : CudaActionBuilderBase(C, Args, Inputs, Action::OFK_HIP) {
2695       DefaultCudaArch = CudaArch::GFX803;
2696     }
2697 
2698     bool canUseBundlerUnbundler() const override { return true; }
2699 
2700     ActionBuilderReturnCode
2701     getDeviceDependences(OffloadAction::DeviceDependences &DA,
2702                          phases::ID CurPhase, phases::ID FinalPhase,
2703                          PhasesTy &Phases) override {
2704       // amdgcn does not support linking of object files, therefore we skip
2705       // backend and assemble phases to output LLVM IR. Except for generating
2706       // non-relocatable device coee, where we generate fat binary for device
2707       // code and pass to host in Backend phase.
2708       if (CudaDeviceActions.empty() ||
2709           (CurPhase == phases::Backend && Relocatable) ||
2710           CurPhase == phases::Assemble)
2711         return ABRT_Success;
2712 
2713       assert(((CurPhase == phases::Link && Relocatable) ||
2714               CudaDeviceActions.size() == GpuArchList.size()) &&
2715              "Expecting one action per GPU architecture.");
2716       assert(!CompileHostOnly &&
2717              "Not expecting CUDA actions in host-only compilation.");
2718 
2719       if (!Relocatable && CurPhase == phases::Backend && !EmitLLVM &&
2720           !EmitAsm) {
2721         // If we are in backend phase, we attempt to generate the fat binary.
2722         // We compile each arch to IR and use a link action to generate code
2723         // object containing ISA. Then we use a special "link" action to create
2724         // a fat binary containing all the code objects for different GPU's.
2725         // The fat binary is then an input to the host action.
2726         for (unsigned I = 0, E = GpuArchList.size(); I != E; ++I) {
2727           // Create a link action to link device IR with device library
2728           // and generate ISA.
2729           ActionList AL;
2730           AL.push_back(CudaDeviceActions[I]);
2731           CudaDeviceActions[I] =
2732               C.MakeAction<LinkJobAction>(AL, types::TY_Image);
2733 
2734           // OffloadingActionBuilder propagates device arch until an offload
2735           // action. Since the next action for creating fatbin does
2736           // not have device arch, whereas the above link action and its input
2737           // have device arch, an offload action is needed to stop the null
2738           // device arch of the next action being propagated to the above link
2739           // action.
2740           OffloadAction::DeviceDependences DDep;
2741           DDep.add(*CudaDeviceActions[I], *ToolChains.front(),
2742                    CudaArchToString(GpuArchList[I]), AssociatedOffloadKind);
2743           CudaDeviceActions[I] = C.MakeAction<OffloadAction>(
2744               DDep, CudaDeviceActions[I]->getType());
2745         }
2746         // Create HIP fat binary with a special "link" action.
2747         CudaFatBinary =
2748             C.MakeAction<LinkJobAction>(CudaDeviceActions,
2749                 types::TY_HIP_FATBIN);
2750 
2751         if (!CompileDeviceOnly) {
2752           DA.add(*CudaFatBinary, *ToolChains.front(), /*BoundArch=*/nullptr,
2753                  AssociatedOffloadKind);
2754           // Clear the fat binary, it is already a dependence to an host
2755           // action.
2756           CudaFatBinary = nullptr;
2757         }
2758 
2759         // Remove the CUDA actions as they are already connected to an host
2760         // action or fat binary.
2761         CudaDeviceActions.clear();
2762 
2763         return CompileDeviceOnly ? ABRT_Ignore_Host : ABRT_Success;
2764       } else if (CurPhase == phases::Link) {
2765         // Save CudaDeviceActions to DeviceLinkerInputs for each GPU subarch.
2766         // This happens to each device action originated from each input file.
2767         // Later on, device actions in DeviceLinkerInputs are used to create
2768         // device link actions in appendLinkDependences and the created device
2769         // link actions are passed to the offload action as device dependence.
2770         DeviceLinkerInputs.resize(CudaDeviceActions.size());
2771         auto LI = DeviceLinkerInputs.begin();
2772         for (auto *A : CudaDeviceActions) {
2773           LI->push_back(A);
2774           ++LI;
2775         }
2776 
2777         // We will pass the device action as a host dependence, so we don't
2778         // need to do anything else with them.
2779         CudaDeviceActions.clear();
2780         return ABRT_Success;
2781       }
2782 
2783       // By default, we produce an action for each device arch.
2784       for (Action *&A : CudaDeviceActions)
2785         A = C.getDriver().ConstructPhaseAction(C, Args, CurPhase, A,
2786                                                AssociatedOffloadKind);
2787 
2788       return (CompileDeviceOnly && CurPhase == FinalPhase) ? ABRT_Ignore_Host
2789                                                            : ABRT_Success;
2790     }
2791 
2792     void appendLinkDependences(OffloadAction::DeviceDependences &DA) override {
2793       // Append a new link action for each device.
2794       unsigned I = 0;
2795       for (auto &LI : DeviceLinkerInputs) {
2796         auto *DeviceLinkAction =
2797             C.MakeAction<LinkJobAction>(LI, types::TY_Image);
2798         DA.add(*DeviceLinkAction, *ToolChains[0],
2799                CudaArchToString(GpuArchList[I]), AssociatedOffloadKind);
2800         ++I;
2801       }
2802     }
2803   };
2804 
2805   /// OpenMP action builder. The host bitcode is passed to the device frontend
2806   /// and all the device linked images are passed to the host link phase.
2807   class OpenMPActionBuilder final : public DeviceActionBuilder {
2808     /// The OpenMP actions for the current input.
2809     ActionList OpenMPDeviceActions;
2810 
2811     /// The linker inputs obtained for each toolchain.
2812     SmallVector<ActionList, 8> DeviceLinkerInputs;
2813 
2814   public:
2815     OpenMPActionBuilder(Compilation &C, DerivedArgList &Args,
2816                         const Driver::InputList &Inputs)
2817         : DeviceActionBuilder(C, Args, Inputs, Action::OFK_OpenMP) {}
2818 
2819     ActionBuilderReturnCode
2820     getDeviceDependences(OffloadAction::DeviceDependences &DA,
2821                          phases::ID CurPhase, phases::ID FinalPhase,
2822                          PhasesTy &Phases) override {
2823       if (OpenMPDeviceActions.empty())
2824         return ABRT_Inactive;
2825 
2826       // We should always have an action for each input.
2827       assert(OpenMPDeviceActions.size() == ToolChains.size() &&
2828              "Number of OpenMP actions and toolchains do not match.");
2829 
2830       // The host only depends on device action in the linking phase, when all
2831       // the device images have to be embedded in the host image.
2832       if (CurPhase == phases::Link) {
2833         assert(ToolChains.size() == DeviceLinkerInputs.size() &&
2834                "Toolchains and linker inputs sizes do not match.");
2835         auto LI = DeviceLinkerInputs.begin();
2836         for (auto *A : OpenMPDeviceActions) {
2837           LI->push_back(A);
2838           ++LI;
2839         }
2840 
2841         // We passed the device action as a host dependence, so we don't need to
2842         // do anything else with them.
2843         OpenMPDeviceActions.clear();
2844         return ABRT_Success;
2845       }
2846 
2847       // By default, we produce an action for each device arch.
2848       for (Action *&A : OpenMPDeviceActions)
2849         A = C.getDriver().ConstructPhaseAction(C, Args, CurPhase, A);
2850 
2851       return ABRT_Success;
2852     }
2853 
2854     ActionBuilderReturnCode addDeviceDepences(Action *HostAction) override {
2855 
2856       // If this is an input action replicate it for each OpenMP toolchain.
2857       if (auto *IA = dyn_cast<InputAction>(HostAction)) {
2858         OpenMPDeviceActions.clear();
2859         for (unsigned I = 0; I < ToolChains.size(); ++I)
2860           OpenMPDeviceActions.push_back(
2861               C.MakeAction<InputAction>(IA->getInputArg(), IA->getType()));
2862         return ABRT_Success;
2863       }
2864 
2865       // If this is an unbundling action use it as is for each OpenMP toolchain.
2866       if (auto *UA = dyn_cast<OffloadUnbundlingJobAction>(HostAction)) {
2867         OpenMPDeviceActions.clear();
2868         auto *IA = cast<InputAction>(UA->getInputs().back());
2869         std::string FileName = IA->getInputArg().getAsString(Args);
2870         // Check if the type of the file is the same as the action. Do not
2871         // unbundle it if it is not. Do not unbundle .so files, for example,
2872         // which are not object files.
2873         if (IA->getType() == types::TY_Object &&
2874             (!llvm::sys::path::has_extension(FileName) ||
2875              types::lookupTypeForExtension(
2876                  llvm::sys::path::extension(FileName).drop_front()) !=
2877                  types::TY_Object))
2878           return ABRT_Inactive;
2879         for (unsigned I = 0; I < ToolChains.size(); ++I) {
2880           OpenMPDeviceActions.push_back(UA);
2881           UA->registerDependentActionInfo(
2882               ToolChains[I], /*BoundArch=*/StringRef(), Action::OFK_OpenMP);
2883         }
2884         return ABRT_Success;
2885       }
2886 
2887       // When generating code for OpenMP we use the host compile phase result as
2888       // a dependence to the device compile phase so that it can learn what
2889       // declarations should be emitted. However, this is not the only use for
2890       // the host action, so we prevent it from being collapsed.
2891       if (isa<CompileJobAction>(HostAction)) {
2892         HostAction->setCannotBeCollapsedWithNextDependentAction();
2893         assert(ToolChains.size() == OpenMPDeviceActions.size() &&
2894                "Toolchains and device action sizes do not match.");
2895         OffloadAction::HostDependence HDep(
2896             *HostAction, *C.getSingleOffloadToolChain<Action::OFK_Host>(),
2897             /*BoundArch=*/nullptr, Action::OFK_OpenMP);
2898         auto TC = ToolChains.begin();
2899         for (Action *&A : OpenMPDeviceActions) {
2900           assert(isa<CompileJobAction>(A));
2901           OffloadAction::DeviceDependences DDep;
2902           DDep.add(*A, **TC, /*BoundArch=*/nullptr, Action::OFK_OpenMP);
2903           A = C.MakeAction<OffloadAction>(HDep, DDep);
2904           ++TC;
2905         }
2906       }
2907       return ABRT_Success;
2908     }
2909 
2910     void appendTopLevelActions(ActionList &AL) override {
2911       if (OpenMPDeviceActions.empty())
2912         return;
2913 
2914       // We should always have an action for each input.
2915       assert(OpenMPDeviceActions.size() == ToolChains.size() &&
2916              "Number of OpenMP actions and toolchains do not match.");
2917 
2918       // Append all device actions followed by the proper offload action.
2919       auto TI = ToolChains.begin();
2920       for (auto *A : OpenMPDeviceActions) {
2921         OffloadAction::DeviceDependences Dep;
2922         Dep.add(*A, **TI, /*BoundArch=*/nullptr, Action::OFK_OpenMP);
2923         AL.push_back(C.MakeAction<OffloadAction>(Dep, A->getType()));
2924         ++TI;
2925       }
2926       // We no longer need the action stored in this builder.
2927       OpenMPDeviceActions.clear();
2928     }
2929 
2930     void appendLinkActions(ActionList &AL) override {
2931       assert(ToolChains.size() == DeviceLinkerInputs.size() &&
2932              "Toolchains and linker inputs sizes do not match.");
2933 
2934       // Append a new link action for each device.
2935       auto TC = ToolChains.begin();
2936       for (auto &LI : DeviceLinkerInputs) {
2937         auto *DeviceLinkAction =
2938             C.MakeAction<LinkJobAction>(LI, types::TY_Image);
2939         OffloadAction::DeviceDependences DeviceLinkDeps;
2940         DeviceLinkDeps.add(*DeviceLinkAction, **TC, /*BoundArch=*/nullptr,
2941 		        Action::OFK_OpenMP);
2942         AL.push_back(C.MakeAction<OffloadAction>(DeviceLinkDeps,
2943             DeviceLinkAction->getType()));
2944         ++TC;
2945       }
2946       DeviceLinkerInputs.clear();
2947     }
2948 
2949     void appendLinkDependences(OffloadAction::DeviceDependences &DA) override {}
2950 
2951     bool initialize() override {
2952       // Get the OpenMP toolchains. If we don't get any, the action builder will
2953       // know there is nothing to do related to OpenMP offloading.
2954       auto OpenMPTCRange = C.getOffloadToolChains<Action::OFK_OpenMP>();
2955       for (auto TI = OpenMPTCRange.first, TE = OpenMPTCRange.second; TI != TE;
2956            ++TI)
2957         ToolChains.push_back(TI->second);
2958 
2959       DeviceLinkerInputs.resize(ToolChains.size());
2960       return false;
2961     }
2962 
2963     bool canUseBundlerUnbundler() const override {
2964       // OpenMP should use bundled files whenever possible.
2965       return true;
2966     }
2967   };
2968 
2969   ///
2970   /// TODO: Add the implementation for other specialized builders here.
2971   ///
2972 
2973   /// Specialized builders being used by this offloading action builder.
2974   SmallVector<DeviceActionBuilder *, 4> SpecializedBuilders;
2975 
2976   /// Flag set to true if all valid builders allow file bundling/unbundling.
2977   bool CanUseBundler;
2978 
2979 public:
2980   OffloadingActionBuilder(Compilation &C, DerivedArgList &Args,
2981                           const Driver::InputList &Inputs)
2982       : C(C) {
2983     // Create a specialized builder for each device toolchain.
2984 
2985     IsValid = true;
2986 
2987     // Create a specialized builder for CUDA.
2988     SpecializedBuilders.push_back(new CudaActionBuilder(C, Args, Inputs));
2989 
2990     // Create a specialized builder for HIP.
2991     SpecializedBuilders.push_back(new HIPActionBuilder(C, Args, Inputs));
2992 
2993     // Create a specialized builder for OpenMP.
2994     SpecializedBuilders.push_back(new OpenMPActionBuilder(C, Args, Inputs));
2995 
2996     //
2997     // TODO: Build other specialized builders here.
2998     //
2999 
3000     // Initialize all the builders, keeping track of errors. If all valid
3001     // builders agree that we can use bundling, set the flag to true.
3002     unsigned ValidBuilders = 0u;
3003     unsigned ValidBuildersSupportingBundling = 0u;
3004     for (auto *SB : SpecializedBuilders) {
3005       IsValid = IsValid && !SB->initialize();
3006 
3007       // Update the counters if the builder is valid.
3008       if (SB->isValid()) {
3009         ++ValidBuilders;
3010         if (SB->canUseBundlerUnbundler())
3011           ++ValidBuildersSupportingBundling;
3012       }
3013     }
3014     CanUseBundler =
3015         ValidBuilders && ValidBuilders == ValidBuildersSupportingBundling;
3016   }
3017 
3018   ~OffloadingActionBuilder() {
3019     for (auto *SB : SpecializedBuilders)
3020       delete SB;
3021   }
3022 
3023   /// Generate an action that adds device dependences (if any) to a host action.
3024   /// If no device dependence actions exist, just return the host action \a
3025   /// HostAction. If an error is found or if no builder requires the host action
3026   /// to be generated, return nullptr.
3027   Action *
3028   addDeviceDependencesToHostAction(Action *HostAction, const Arg *InputArg,
3029                                    phases::ID CurPhase, phases::ID FinalPhase,
3030                                    DeviceActionBuilder::PhasesTy &Phases) {
3031     if (!IsValid)
3032       return nullptr;
3033 
3034     if (SpecializedBuilders.empty())
3035       return HostAction;
3036 
3037     assert(HostAction && "Invalid host action!");
3038 
3039     OffloadAction::DeviceDependences DDeps;
3040     // Check if all the programming models agree we should not emit the host
3041     // action. Also, keep track of the offloading kinds employed.
3042     auto &OffloadKind = InputArgToOffloadKindMap[InputArg];
3043     unsigned InactiveBuilders = 0u;
3044     unsigned IgnoringBuilders = 0u;
3045     for (auto *SB : SpecializedBuilders) {
3046       if (!SB->isValid()) {
3047         ++InactiveBuilders;
3048         continue;
3049       }
3050 
3051       auto RetCode =
3052           SB->getDeviceDependences(DDeps, CurPhase, FinalPhase, Phases);
3053 
3054       // If the builder explicitly says the host action should be ignored,
3055       // we need to increment the variable that tracks the builders that request
3056       // the host object to be ignored.
3057       if (RetCode == DeviceActionBuilder::ABRT_Ignore_Host)
3058         ++IgnoringBuilders;
3059 
3060       // Unless the builder was inactive for this action, we have to record the
3061       // offload kind because the host will have to use it.
3062       if (RetCode != DeviceActionBuilder::ABRT_Inactive)
3063         OffloadKind |= SB->getAssociatedOffloadKind();
3064     }
3065 
3066     // If all builders agree that the host object should be ignored, just return
3067     // nullptr.
3068     if (IgnoringBuilders &&
3069         SpecializedBuilders.size() == (InactiveBuilders + IgnoringBuilders))
3070       return nullptr;
3071 
3072     if (DDeps.getActions().empty())
3073       return HostAction;
3074 
3075     // We have dependences we need to bundle together. We use an offload action
3076     // for that.
3077     OffloadAction::HostDependence HDep(
3078         *HostAction, *C.getSingleOffloadToolChain<Action::OFK_Host>(),
3079         /*BoundArch=*/nullptr, DDeps);
3080     return C.MakeAction<OffloadAction>(HDep, DDeps);
3081   }
3082 
3083   /// Generate an action that adds a host dependence to a device action. The
3084   /// results will be kept in this action builder. Return true if an error was
3085   /// found.
3086   bool addHostDependenceToDeviceActions(Action *&HostAction,
3087                                         const Arg *InputArg) {
3088     if (!IsValid)
3089       return true;
3090 
3091     // If we are supporting bundling/unbundling and the current action is an
3092     // input action of non-source file, we replace the host action by the
3093     // unbundling action. The bundler tool has the logic to detect if an input
3094     // is a bundle or not and if the input is not a bundle it assumes it is a
3095     // host file. Therefore it is safe to create an unbundling action even if
3096     // the input is not a bundle.
3097     if (CanUseBundler && isa<InputAction>(HostAction) &&
3098         InputArg->getOption().getKind() == llvm::opt::Option::InputClass &&
3099         !types::isSrcFile(HostAction->getType())) {
3100       auto UnbundlingHostAction =
3101           C.MakeAction<OffloadUnbundlingJobAction>(HostAction);
3102       UnbundlingHostAction->registerDependentActionInfo(
3103           C.getSingleOffloadToolChain<Action::OFK_Host>(),
3104           /*BoundArch=*/StringRef(), Action::OFK_Host);
3105       HostAction = UnbundlingHostAction;
3106     }
3107 
3108     assert(HostAction && "Invalid host action!");
3109 
3110     // Register the offload kinds that are used.
3111     auto &OffloadKind = InputArgToOffloadKindMap[InputArg];
3112     for (auto *SB : SpecializedBuilders) {
3113       if (!SB->isValid())
3114         continue;
3115 
3116       auto RetCode = SB->addDeviceDepences(HostAction);
3117 
3118       // Host dependences for device actions are not compatible with that same
3119       // action being ignored.
3120       assert(RetCode != DeviceActionBuilder::ABRT_Ignore_Host &&
3121              "Host dependence not expected to be ignored.!");
3122 
3123       // Unless the builder was inactive for this action, we have to record the
3124       // offload kind because the host will have to use it.
3125       if (RetCode != DeviceActionBuilder::ABRT_Inactive)
3126         OffloadKind |= SB->getAssociatedOffloadKind();
3127     }
3128 
3129     // Do not use unbundler if the Host does not depend on device action.
3130     if (OffloadKind == Action::OFK_None && CanUseBundler)
3131       if (auto *UA = dyn_cast<OffloadUnbundlingJobAction>(HostAction))
3132         HostAction = UA->getInputs().back();
3133 
3134     return false;
3135   }
3136 
3137   /// Add the offloading top level actions to the provided action list. This
3138   /// function can replace the host action by a bundling action if the
3139   /// programming models allow it.
3140   bool appendTopLevelActions(ActionList &AL, Action *HostAction,
3141                              const Arg *InputArg) {
3142     // Get the device actions to be appended.
3143     ActionList OffloadAL;
3144     for (auto *SB : SpecializedBuilders) {
3145       if (!SB->isValid())
3146         continue;
3147       SB->appendTopLevelActions(OffloadAL);
3148     }
3149 
3150     // If we can use the bundler, replace the host action by the bundling one in
3151     // the resulting list. Otherwise, just append the device actions. For
3152     // device only compilation, HostAction is a null pointer, therefore only do
3153     // this when HostAction is not a null pointer.
3154     if (CanUseBundler && HostAction &&
3155         HostAction->getType() != types::TY_Nothing && !OffloadAL.empty()) {
3156       // Add the host action to the list in order to create the bundling action.
3157       OffloadAL.push_back(HostAction);
3158 
3159       // We expect that the host action was just appended to the action list
3160       // before this method was called.
3161       assert(HostAction == AL.back() && "Host action not in the list??");
3162       HostAction = C.MakeAction<OffloadBundlingJobAction>(OffloadAL);
3163       AL.back() = HostAction;
3164     } else
3165       AL.append(OffloadAL.begin(), OffloadAL.end());
3166 
3167     // Propagate to the current host action (if any) the offload information
3168     // associated with the current input.
3169     if (HostAction)
3170       HostAction->propagateHostOffloadInfo(InputArgToOffloadKindMap[InputArg],
3171                                            /*BoundArch=*/nullptr);
3172     return false;
3173   }
3174 
3175   Action* makeHostLinkAction() {
3176     // Build a list of device linking actions.
3177     ActionList DeviceAL;
3178     for (DeviceActionBuilder *SB : SpecializedBuilders) {
3179       if (!SB->isValid())
3180         continue;
3181       SB->appendLinkActions(DeviceAL);
3182     }
3183 
3184     if (DeviceAL.empty())
3185       return nullptr;
3186 
3187     // Create wrapper bitcode from the result of device link actions and compile
3188     // it to an object which will be added to the host link command.
3189     auto *BC = C.MakeAction<OffloadWrapperJobAction>(DeviceAL, types::TY_LLVM_BC);
3190     auto *ASM = C.MakeAction<BackendJobAction>(BC, types::TY_PP_Asm);
3191     return C.MakeAction<AssembleJobAction>(ASM, types::TY_Object);
3192   }
3193 
3194   /// Processes the host linker action. This currently consists of replacing it
3195   /// with an offload action if there are device link objects and propagate to
3196   /// the host action all the offload kinds used in the current compilation. The
3197   /// resulting action is returned.
3198   Action *processHostLinkAction(Action *HostAction) {
3199     // Add all the dependences from the device linking actions.
3200     OffloadAction::DeviceDependences DDeps;
3201     for (auto *SB : SpecializedBuilders) {
3202       if (!SB->isValid())
3203         continue;
3204 
3205       SB->appendLinkDependences(DDeps);
3206     }
3207 
3208     // Calculate all the offload kinds used in the current compilation.
3209     unsigned ActiveOffloadKinds = 0u;
3210     for (auto &I : InputArgToOffloadKindMap)
3211       ActiveOffloadKinds |= I.second;
3212 
3213     // If we don't have device dependencies, we don't have to create an offload
3214     // action.
3215     if (DDeps.getActions().empty()) {
3216       // Propagate all the active kinds to host action. Given that it is a link
3217       // action it is assumed to depend on all actions generated so far.
3218       HostAction->propagateHostOffloadInfo(ActiveOffloadKinds,
3219                                            /*BoundArch=*/nullptr);
3220       return HostAction;
3221     }
3222 
3223     // Create the offload action with all dependences. When an offload action
3224     // is created the kinds are propagated to the host action, so we don't have
3225     // to do that explicitly here.
3226     OffloadAction::HostDependence HDep(
3227         *HostAction, *C.getSingleOffloadToolChain<Action::OFK_Host>(),
3228         /*BoundArch*/ nullptr, ActiveOffloadKinds);
3229     return C.MakeAction<OffloadAction>(HDep, DDeps);
3230   }
3231 };
3232 } // anonymous namespace.
3233 
3234 void Driver::handleArguments(Compilation &C, DerivedArgList &Args,
3235                              const InputList &Inputs,
3236                              ActionList &Actions) const {
3237 
3238   // Ignore /Yc/Yu if both /Yc and /Yu passed but with different filenames.
3239   Arg *YcArg = Args.getLastArg(options::OPT__SLASH_Yc);
3240   Arg *YuArg = Args.getLastArg(options::OPT__SLASH_Yu);
3241   if (YcArg && YuArg && strcmp(YcArg->getValue(), YuArg->getValue()) != 0) {
3242     Diag(clang::diag::warn_drv_ycyu_different_arg_clang_cl);
3243     Args.eraseArg(options::OPT__SLASH_Yc);
3244     Args.eraseArg(options::OPT__SLASH_Yu);
3245     YcArg = YuArg = nullptr;
3246   }
3247   if (YcArg && Inputs.size() > 1) {
3248     Diag(clang::diag::warn_drv_yc_multiple_inputs_clang_cl);
3249     Args.eraseArg(options::OPT__SLASH_Yc);
3250     YcArg = nullptr;
3251   }
3252 
3253   Arg *FinalPhaseArg;
3254   phases::ID FinalPhase = getFinalPhase(Args, &FinalPhaseArg);
3255 
3256   if (FinalPhase == phases::Link) {
3257     if (Args.hasArg(options::OPT_emit_llvm))
3258       Diag(clang::diag::err_drv_emit_llvm_link);
3259     if (IsCLMode() && LTOMode != LTOK_None &&
3260         !Args.getLastArgValue(options::OPT_fuse_ld_EQ).equals_lower("lld"))
3261       Diag(clang::diag::err_drv_lto_without_lld);
3262   }
3263 
3264   if (FinalPhase == phases::Preprocess || Args.hasArg(options::OPT__SLASH_Y_)) {
3265     // If only preprocessing or /Y- is used, all pch handling is disabled.
3266     // Rather than check for it everywhere, just remove clang-cl pch-related
3267     // flags here.
3268     Args.eraseArg(options::OPT__SLASH_Fp);
3269     Args.eraseArg(options::OPT__SLASH_Yc);
3270     Args.eraseArg(options::OPT__SLASH_Yu);
3271     YcArg = YuArg = nullptr;
3272   }
3273 
3274   unsigned LastPLSize = 0;
3275   for (auto &I : Inputs) {
3276     types::ID InputType = I.first;
3277     const Arg *InputArg = I.second;
3278 
3279     llvm::SmallVector<phases::ID, phases::MaxNumberOfPhases> PL;
3280     types::getCompilationPhases(InputType, PL);
3281     LastPLSize = PL.size();
3282 
3283     // If the first step comes after the final phase we are doing as part of
3284     // this compilation, warn the user about it.
3285     phases::ID InitialPhase = PL[0];
3286     if (InitialPhase > FinalPhase) {
3287       if (InputArg->isClaimed())
3288         continue;
3289 
3290       // Claim here to avoid the more general unused warning.
3291       InputArg->claim();
3292 
3293       // Suppress all unused style warnings with -Qunused-arguments
3294       if (Args.hasArg(options::OPT_Qunused_arguments))
3295         continue;
3296 
3297       // Special case when final phase determined by binary name, rather than
3298       // by a command-line argument with a corresponding Arg.
3299       if (CCCIsCPP())
3300         Diag(clang::diag::warn_drv_input_file_unused_by_cpp)
3301             << InputArg->getAsString(Args) << getPhaseName(InitialPhase);
3302       // Special case '-E' warning on a previously preprocessed file to make
3303       // more sense.
3304       else if (InitialPhase == phases::Compile &&
3305                (Args.getLastArg(options::OPT__SLASH_EP,
3306                                 options::OPT__SLASH_P) ||
3307                 Args.getLastArg(options::OPT_E) ||
3308                 Args.getLastArg(options::OPT_M, options::OPT_MM)) &&
3309                getPreprocessedType(InputType) == types::TY_INVALID)
3310         Diag(clang::diag::warn_drv_preprocessed_input_file_unused)
3311             << InputArg->getAsString(Args) << !!FinalPhaseArg
3312             << (FinalPhaseArg ? FinalPhaseArg->getOption().getName() : "");
3313       else
3314         Diag(clang::diag::warn_drv_input_file_unused)
3315             << InputArg->getAsString(Args) << getPhaseName(InitialPhase)
3316             << !!FinalPhaseArg
3317             << (FinalPhaseArg ? FinalPhaseArg->getOption().getName() : "");
3318       continue;
3319     }
3320 
3321     if (YcArg) {
3322       // Add a separate precompile phase for the compile phase.
3323       if (FinalPhase >= phases::Compile) {
3324         const types::ID HeaderType = lookupHeaderTypeForSourceType(InputType);
3325         llvm::SmallVector<phases::ID, phases::MaxNumberOfPhases> PCHPL;
3326         types::getCompilationPhases(HeaderType, PCHPL);
3327         // Build the pipeline for the pch file.
3328         Action *ClangClPch = C.MakeAction<InputAction>(*InputArg, HeaderType);
3329         for (phases::ID Phase : PCHPL)
3330           ClangClPch = ConstructPhaseAction(C, Args, Phase, ClangClPch);
3331         assert(ClangClPch);
3332         Actions.push_back(ClangClPch);
3333         // The driver currently exits after the first failed command.  This
3334         // relies on that behavior, to make sure if the pch generation fails,
3335         // the main compilation won't run.
3336         // FIXME: If the main compilation fails, the PCH generation should
3337         // probably not be considered successful either.
3338       }
3339     }
3340   }
3341 
3342   // If we are linking, claim any options which are obviously only used for
3343   // compilation.
3344   // FIXME: Understand why the last Phase List length is used here.
3345   if (FinalPhase == phases::Link && LastPLSize == 1) {
3346     Args.ClaimAllArgs(options::OPT_CompileOnly_Group);
3347     Args.ClaimAllArgs(options::OPT_cl_compile_Group);
3348   }
3349 }
3350 
3351 void Driver::BuildActions(Compilation &C, DerivedArgList &Args,
3352                           const InputList &Inputs, ActionList &Actions) const {
3353   llvm::PrettyStackTraceString CrashInfo("Building compilation actions");
3354 
3355   if (!SuppressMissingInputWarning && Inputs.empty()) {
3356     Diag(clang::diag::err_drv_no_input_files);
3357     return;
3358   }
3359 
3360   // Reject -Z* at the top level, these options should never have been exposed
3361   // by gcc.
3362   if (Arg *A = Args.getLastArg(options::OPT_Z_Joined))
3363     Diag(clang::diag::err_drv_use_of_Z_option) << A->getAsString(Args);
3364 
3365   // Diagnose misuse of /Fo.
3366   if (Arg *A = Args.getLastArg(options::OPT__SLASH_Fo)) {
3367     StringRef V = A->getValue();
3368     if (Inputs.size() > 1 && !V.empty() &&
3369         !llvm::sys::path::is_separator(V.back())) {
3370       // Check whether /Fo tries to name an output file for multiple inputs.
3371       Diag(clang::diag::err_drv_out_file_argument_with_multiple_sources)
3372           << A->getSpelling() << V;
3373       Args.eraseArg(options::OPT__SLASH_Fo);
3374     }
3375   }
3376 
3377   // Diagnose misuse of /Fa.
3378   if (Arg *A = Args.getLastArg(options::OPT__SLASH_Fa)) {
3379     StringRef V = A->getValue();
3380     if (Inputs.size() > 1 && !V.empty() &&
3381         !llvm::sys::path::is_separator(V.back())) {
3382       // Check whether /Fa tries to name an asm file for multiple inputs.
3383       Diag(clang::diag::err_drv_out_file_argument_with_multiple_sources)
3384           << A->getSpelling() << V;
3385       Args.eraseArg(options::OPT__SLASH_Fa);
3386     }
3387   }
3388 
3389   // Diagnose misuse of /o.
3390   if (Arg *A = Args.getLastArg(options::OPT__SLASH_o)) {
3391     if (A->getValue()[0] == '\0') {
3392       // It has to have a value.
3393       Diag(clang::diag::err_drv_missing_argument) << A->getSpelling() << 1;
3394       Args.eraseArg(options::OPT__SLASH_o);
3395     }
3396   }
3397 
3398   handleArguments(C, Args, Inputs, Actions);
3399 
3400   // Builder to be used to build offloading actions.
3401   OffloadingActionBuilder OffloadBuilder(C, Args, Inputs);
3402 
3403   // Construct the actions to perform.
3404   HeaderModulePrecompileJobAction *HeaderModuleAction = nullptr;
3405   ActionList LinkerInputs;
3406   ActionList MergerInputs;
3407 
3408   for (auto &I : Inputs) {
3409     types::ID InputType = I.first;
3410     const Arg *InputArg = I.second;
3411 
3412     llvm::SmallVector<phases::ID, phases::MaxNumberOfPhases> PL;
3413     types::getCompilationPhases(*this, Args, InputType, PL);
3414     if (PL.empty())
3415       continue;
3416 
3417     llvm::SmallVector<phases::ID, phases::MaxNumberOfPhases> FullPL;
3418     types::getCompilationPhases(InputType, FullPL);
3419 
3420     // Build the pipeline for this file.
3421     Action *Current = C.MakeAction<InputAction>(*InputArg, InputType);
3422 
3423     // Use the current host action in any of the offloading actions, if
3424     // required.
3425     if (OffloadBuilder.addHostDependenceToDeviceActions(Current, InputArg))
3426       break;
3427 
3428     for (phases::ID Phase : PL) {
3429 
3430       // Add any offload action the host action depends on.
3431       Current = OffloadBuilder.addDeviceDependencesToHostAction(
3432           Current, InputArg, Phase, PL.back(), FullPL);
3433       if (!Current)
3434         break;
3435 
3436       // Queue linker inputs.
3437       if (Phase == phases::Link) {
3438         assert(Phase == PL.back() && "linking must be final compilation step.");
3439         LinkerInputs.push_back(Current);
3440         Current = nullptr;
3441         break;
3442       }
3443 
3444       // TODO: Consider removing this because the merged may not end up being
3445       // the final Phase in the pipeline. Perhaps the merged could just merge
3446       // and then pass an artifact of some sort to the Link Phase.
3447       // Queue merger inputs.
3448       if (Phase == phases::IfsMerge) {
3449         assert(Phase == PL.back() && "merging must be final compilation step.");
3450         MergerInputs.push_back(Current);
3451         Current = nullptr;
3452         break;
3453       }
3454 
3455       // Each precompiled header file after a module file action is a module
3456       // header of that same module file, rather than being compiled to a
3457       // separate PCH.
3458       if (Phase == phases::Precompile && HeaderModuleAction &&
3459           getPrecompiledType(InputType) == types::TY_PCH) {
3460         HeaderModuleAction->addModuleHeaderInput(Current);
3461         Current = nullptr;
3462         break;
3463       }
3464 
3465       // FIXME: Should we include any prior module file outputs as inputs of
3466       // later actions in the same command line?
3467 
3468       // Otherwise construct the appropriate action.
3469       Action *NewCurrent = ConstructPhaseAction(C, Args, Phase, Current);
3470 
3471       // We didn't create a new action, so we will just move to the next phase.
3472       if (NewCurrent == Current)
3473         continue;
3474 
3475       if (auto *HMA = dyn_cast<HeaderModulePrecompileJobAction>(NewCurrent))
3476         HeaderModuleAction = HMA;
3477 
3478       Current = NewCurrent;
3479 
3480       // Use the current host action in any of the offloading actions, if
3481       // required.
3482       if (OffloadBuilder.addHostDependenceToDeviceActions(Current, InputArg))
3483         break;
3484 
3485       if (Current->getType() == types::TY_Nothing)
3486         break;
3487     }
3488 
3489     // If we ended with something, add to the output list.
3490     if (Current)
3491       Actions.push_back(Current);
3492 
3493     // Add any top level actions generated for offloading.
3494     OffloadBuilder.appendTopLevelActions(Actions, Current, InputArg);
3495   }
3496 
3497   // Add a link action if necessary.
3498   if (!LinkerInputs.empty()) {
3499     if (Action *Wrapper = OffloadBuilder.makeHostLinkAction())
3500       LinkerInputs.push_back(Wrapper);
3501     Action *LA = C.MakeAction<LinkJobAction>(LinkerInputs, types::TY_Image);
3502     LA = OffloadBuilder.processHostLinkAction(LA);
3503     Actions.push_back(LA);
3504   }
3505 
3506   // Add an interface stubs merge action if necessary.
3507   if (!MergerInputs.empty())
3508     Actions.push_back(
3509         C.MakeAction<IfsMergeJobAction>(MergerInputs, types::TY_Image));
3510 
3511   if (Args.hasArg(options::OPT_emit_interface_stubs)) {
3512     llvm::SmallVector<phases::ID, phases::MaxNumberOfPhases> PhaseList;
3513     if (Args.hasArg(options::OPT_c)) {
3514       llvm::SmallVector<phases::ID, phases::MaxNumberOfPhases> CompilePhaseList;
3515       types::getCompilationPhases(types::TY_IFS_CPP, CompilePhaseList);
3516       llvm::copy_if(CompilePhaseList, std::back_inserter(PhaseList),
3517                     [&](phases::ID Phase) { return Phase <= phases::Compile; });
3518     } else {
3519       types::getCompilationPhases(types::TY_IFS_CPP, PhaseList);
3520     }
3521 
3522     ActionList MergerInputs;
3523 
3524     for (auto &I : Inputs) {
3525       types::ID InputType = I.first;
3526       const Arg *InputArg = I.second;
3527 
3528       // Currently clang and the llvm assembler do not support generating symbol
3529       // stubs from assembly, so we skip the input on asm files. For ifs files
3530       // we rely on the normal pipeline setup in the pipeline setup code above.
3531       if (InputType == types::TY_IFS || InputType == types::TY_PP_Asm ||
3532           InputType == types::TY_Asm)
3533         continue;
3534 
3535       Action *Current = C.MakeAction<InputAction>(*InputArg, InputType);
3536 
3537       for (auto Phase : PhaseList) {
3538         switch (Phase) {
3539         default:
3540           llvm_unreachable(
3541               "IFS Pipeline can only consist of Compile followed by IfsMerge.");
3542         case phases::Compile: {
3543           // Only IfsMerge (llvm-ifs) can handle .o files by looking for ifs
3544           // files where the .o file is located. The compile action can not
3545           // handle this.
3546           if (InputType == types::TY_Object)
3547             break;
3548 
3549           Current = C.MakeAction<CompileJobAction>(Current, types::TY_IFS_CPP);
3550           break;
3551         }
3552         case phases::IfsMerge: {
3553           assert(Phase == PhaseList.back() &&
3554                  "merging must be final compilation step.");
3555           MergerInputs.push_back(Current);
3556           Current = nullptr;
3557           break;
3558         }
3559         }
3560       }
3561 
3562       // If we ended with something, add to the output list.
3563       if (Current)
3564         Actions.push_back(Current);
3565     }
3566 
3567     // Add an interface stubs merge action if necessary.
3568     if (!MergerInputs.empty())
3569       Actions.push_back(
3570           C.MakeAction<IfsMergeJobAction>(MergerInputs, types::TY_Image));
3571   }
3572 
3573   // If --print-supported-cpus, -mcpu=? or -mtune=? is specified, build a custom
3574   // Compile phase that prints out supported cpu models and quits.
3575   if (Arg *A = Args.getLastArg(options::OPT_print_supported_cpus)) {
3576     // Use the -mcpu=? flag as the dummy input to cc1.
3577     Actions.clear();
3578     Action *InputAc = C.MakeAction<InputAction>(*A, types::TY_C);
3579     Actions.push_back(
3580         C.MakeAction<PrecompileJobAction>(InputAc, types::TY_Nothing));
3581     for (auto &I : Inputs)
3582       I.second->claim();
3583   }
3584 
3585   // Claim ignored clang-cl options.
3586   Args.ClaimAllArgs(options::OPT_cl_ignored_Group);
3587 
3588   // Claim --cuda-host-only and --cuda-compile-host-device, which may be passed
3589   // to non-CUDA compilations and should not trigger warnings there.
3590   Args.ClaimAllArgs(options::OPT_cuda_host_only);
3591   Args.ClaimAllArgs(options::OPT_cuda_compile_host_device);
3592 }
3593 
3594 Action *Driver::ConstructPhaseAction(
3595     Compilation &C, const ArgList &Args, phases::ID Phase, Action *Input,
3596     Action::OffloadKind TargetDeviceOffloadKind) const {
3597   llvm::PrettyStackTraceString CrashInfo("Constructing phase actions");
3598 
3599   // Some types skip the assembler phase (e.g., llvm-bc), but we can't
3600   // encode this in the steps because the intermediate type depends on
3601   // arguments. Just special case here.
3602   if (Phase == phases::Assemble && Input->getType() != types::TY_PP_Asm)
3603     return Input;
3604 
3605   // Build the appropriate action.
3606   switch (Phase) {
3607   case phases::Link:
3608     llvm_unreachable("link action invalid here.");
3609   case phases::IfsMerge:
3610     llvm_unreachable("ifsmerge action invalid here.");
3611   case phases::Preprocess: {
3612     types::ID OutputTy;
3613     // -M and -MM specify the dependency file name by altering the output type,
3614     // -if -MD and -MMD are not specified.
3615     if (Args.hasArg(options::OPT_M, options::OPT_MM) &&
3616         !Args.hasArg(options::OPT_MD, options::OPT_MMD)) {
3617       OutputTy = types::TY_Dependencies;
3618     } else {
3619       OutputTy = Input->getType();
3620       if (!Args.hasFlag(options::OPT_frewrite_includes,
3621                         options::OPT_fno_rewrite_includes, false) &&
3622           !Args.hasFlag(options::OPT_frewrite_imports,
3623                         options::OPT_fno_rewrite_imports, false) &&
3624           !CCGenDiagnostics)
3625         OutputTy = types::getPreprocessedType(OutputTy);
3626       assert(OutputTy != types::TY_INVALID &&
3627              "Cannot preprocess this input type!");
3628     }
3629     return C.MakeAction<PreprocessJobAction>(Input, OutputTy);
3630   }
3631   case phases::Precompile: {
3632     types::ID OutputTy = getPrecompiledType(Input->getType());
3633     assert(OutputTy != types::TY_INVALID &&
3634            "Cannot precompile this input type!");
3635 
3636     // If we're given a module name, precompile header file inputs as a
3637     // module, not as a precompiled header.
3638     const char *ModName = nullptr;
3639     if (OutputTy == types::TY_PCH) {
3640       if (Arg *A = Args.getLastArg(options::OPT_fmodule_name_EQ))
3641         ModName = A->getValue();
3642       if (ModName)
3643         OutputTy = types::TY_ModuleFile;
3644     }
3645 
3646     if (Args.hasArg(options::OPT_fsyntax_only)) {
3647       // Syntax checks should not emit a PCH file
3648       OutputTy = types::TY_Nothing;
3649     }
3650 
3651     if (ModName)
3652       return C.MakeAction<HeaderModulePrecompileJobAction>(Input, OutputTy,
3653                                                            ModName);
3654     return C.MakeAction<PrecompileJobAction>(Input, OutputTy);
3655   }
3656   case phases::Compile: {
3657     if (Args.hasArg(options::OPT_fsyntax_only))
3658       return C.MakeAction<CompileJobAction>(Input, types::TY_Nothing);
3659     if (Args.hasArg(options::OPT_rewrite_objc))
3660       return C.MakeAction<CompileJobAction>(Input, types::TY_RewrittenObjC);
3661     if (Args.hasArg(options::OPT_rewrite_legacy_objc))
3662       return C.MakeAction<CompileJobAction>(Input,
3663                                             types::TY_RewrittenLegacyObjC);
3664     if (Args.hasArg(options::OPT__analyze))
3665       return C.MakeAction<AnalyzeJobAction>(Input, types::TY_Plist);
3666     if (Args.hasArg(options::OPT__migrate))
3667       return C.MakeAction<MigrateJobAction>(Input, types::TY_Remap);
3668     if (Args.hasArg(options::OPT_emit_ast))
3669       return C.MakeAction<CompileJobAction>(Input, types::TY_AST);
3670     if (Args.hasArg(options::OPT_module_file_info))
3671       return C.MakeAction<CompileJobAction>(Input, types::TY_ModuleFile);
3672     if (Args.hasArg(options::OPT_verify_pch))
3673       return C.MakeAction<VerifyPCHJobAction>(Input, types::TY_Nothing);
3674     return C.MakeAction<CompileJobAction>(Input, types::TY_LLVM_BC);
3675   }
3676   case phases::Backend: {
3677     if (isUsingLTO() && TargetDeviceOffloadKind == Action::OFK_None) {
3678       types::ID Output =
3679           Args.hasArg(options::OPT_S) ? types::TY_LTO_IR : types::TY_LTO_BC;
3680       return C.MakeAction<BackendJobAction>(Input, Output);
3681     }
3682     if (Args.hasArg(options::OPT_emit_llvm)) {
3683       types::ID Output =
3684           Args.hasArg(options::OPT_S) ? types::TY_LLVM_IR : types::TY_LLVM_BC;
3685       return C.MakeAction<BackendJobAction>(Input, Output);
3686     }
3687     return C.MakeAction<BackendJobAction>(Input, types::TY_PP_Asm);
3688   }
3689   case phases::Assemble:
3690     return C.MakeAction<AssembleJobAction>(std::move(Input), types::TY_Object);
3691   }
3692 
3693   llvm_unreachable("invalid phase in ConstructPhaseAction");
3694 }
3695 
3696 void Driver::BuildJobs(Compilation &C) const {
3697   llvm::PrettyStackTraceString CrashInfo("Building compilation jobs");
3698 
3699   Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o);
3700 
3701   // It is an error to provide a -o option if we are making multiple output
3702   // files. There are exceptions:
3703   //
3704   // IfsMergeJob: when generating interface stubs enabled we want to be able to
3705   // generate the stub file at the same time that we generate the real
3706   // library/a.out. So when a .o, .so, etc are the output, with clang interface
3707   // stubs there will also be a .ifs and .ifso at the same location.
3708   //
3709   // CompileJob of type TY_IFS_CPP: when generating interface stubs is enabled
3710   // and -c is passed, we still want to be able to generate a .ifs file while
3711   // we are also generating .o files. So we allow more than one output file in
3712   // this case as well.
3713   //
3714   if (FinalOutput) {
3715     unsigned NumOutputs = 0;
3716     unsigned NumIfsOutputs = 0;
3717     for (const Action *A : C.getActions())
3718       if (A->getType() != types::TY_Nothing &&
3719           !(A->getKind() == Action::IfsMergeJobClass ||
3720             (A->getType() == clang::driver::types::TY_IFS_CPP &&
3721              A->getKind() == clang::driver::Action::CompileJobClass &&
3722              0 == NumIfsOutputs++) ||
3723             (A->getKind() == Action::BindArchClass && A->getInputs().size() &&
3724              A->getInputs().front()->getKind() == Action::IfsMergeJobClass)))
3725         ++NumOutputs;
3726 
3727     if (NumOutputs > 1) {
3728       Diag(clang::diag::err_drv_output_argument_with_multiple_files);
3729       FinalOutput = nullptr;
3730     }
3731   }
3732 
3733   // Collect the list of architectures.
3734   llvm::StringSet<> ArchNames;
3735   if (C.getDefaultToolChain().getTriple().isOSBinFormatMachO())
3736     for (const Arg *A : C.getArgs())
3737       if (A->getOption().matches(options::OPT_arch))
3738         ArchNames.insert(A->getValue());
3739 
3740   // Set of (Action, canonical ToolChain triple) pairs we've built jobs for.
3741   std::map<std::pair<const Action *, std::string>, InputInfo> CachedResults;
3742   for (Action *A : C.getActions()) {
3743     // If we are linking an image for multiple archs then the linker wants
3744     // -arch_multiple and -final_output <final image name>. Unfortunately, this
3745     // doesn't fit in cleanly because we have to pass this information down.
3746     //
3747     // FIXME: This is a hack; find a cleaner way to integrate this into the
3748     // process.
3749     const char *LinkingOutput = nullptr;
3750     if (isa<LipoJobAction>(A)) {
3751       if (FinalOutput)
3752         LinkingOutput = FinalOutput->getValue();
3753       else
3754         LinkingOutput = getDefaultImageName();
3755     }
3756 
3757     BuildJobsForAction(C, A, &C.getDefaultToolChain(),
3758                        /*BoundArch*/ StringRef(),
3759                        /*AtTopLevel*/ true,
3760                        /*MultipleArchs*/ ArchNames.size() > 1,
3761                        /*LinkingOutput*/ LinkingOutput, CachedResults,
3762                        /*TargetDeviceOffloadKind*/ Action::OFK_None);
3763   }
3764 
3765   // If we have more than one job, then disable integrated-cc1 for now.
3766   if (C.getJobs().size() > 1)
3767     for (auto &J : C.getJobs())
3768       J.InProcess = false;
3769 
3770   // If the user passed -Qunused-arguments or there were errors, don't warn
3771   // about any unused arguments.
3772   if (Diags.hasErrorOccurred() ||
3773       C.getArgs().hasArg(options::OPT_Qunused_arguments))
3774     return;
3775 
3776   // Claim -### here.
3777   (void)C.getArgs().hasArg(options::OPT__HASH_HASH_HASH);
3778 
3779   // Claim --driver-mode, --rsp-quoting, it was handled earlier.
3780   (void)C.getArgs().hasArg(options::OPT_driver_mode);
3781   (void)C.getArgs().hasArg(options::OPT_rsp_quoting);
3782 
3783   for (Arg *A : C.getArgs()) {
3784     // FIXME: It would be nice to be able to send the argument to the
3785     // DiagnosticsEngine, so that extra values, position, and so on could be
3786     // printed.
3787     if (!A->isClaimed()) {
3788       if (A->getOption().hasFlag(options::NoArgumentUnused))
3789         continue;
3790 
3791       // Suppress the warning automatically if this is just a flag, and it is an
3792       // instance of an argument we already claimed.
3793       const Option &Opt = A->getOption();
3794       if (Opt.getKind() == Option::FlagClass) {
3795         bool DuplicateClaimed = false;
3796 
3797         for (const Arg *AA : C.getArgs().filtered(&Opt)) {
3798           if (AA->isClaimed()) {
3799             DuplicateClaimed = true;
3800             break;
3801           }
3802         }
3803 
3804         if (DuplicateClaimed)
3805           continue;
3806       }
3807 
3808       // In clang-cl, don't mention unknown arguments here since they have
3809       // already been warned about.
3810       if (!IsCLMode() || !A->getOption().matches(options::OPT_UNKNOWN))
3811         Diag(clang::diag::warn_drv_unused_argument)
3812             << A->getAsString(C.getArgs());
3813     }
3814   }
3815 }
3816 
3817 namespace {
3818 /// Utility class to control the collapse of dependent actions and select the
3819 /// tools accordingly.
3820 class ToolSelector final {
3821   /// The tool chain this selector refers to.
3822   const ToolChain &TC;
3823 
3824   /// The compilation this selector refers to.
3825   const Compilation &C;
3826 
3827   /// The base action this selector refers to.
3828   const JobAction *BaseAction;
3829 
3830   /// Set to true if the current toolchain refers to host actions.
3831   bool IsHostSelector;
3832 
3833   /// Set to true if save-temps and embed-bitcode functionalities are active.
3834   bool SaveTemps;
3835   bool EmbedBitcode;
3836 
3837   /// Get previous dependent action or null if that does not exist. If
3838   /// \a CanBeCollapsed is false, that action must be legal to collapse or
3839   /// null will be returned.
3840   const JobAction *getPrevDependentAction(const ActionList &Inputs,
3841                                           ActionList &SavedOffloadAction,
3842                                           bool CanBeCollapsed = true) {
3843     // An option can be collapsed only if it has a single input.
3844     if (Inputs.size() != 1)
3845       return nullptr;
3846 
3847     Action *CurAction = *Inputs.begin();
3848     if (CanBeCollapsed &&
3849         !CurAction->isCollapsingWithNextDependentActionLegal())
3850       return nullptr;
3851 
3852     // If the input action is an offload action. Look through it and save any
3853     // offload action that can be dropped in the event of a collapse.
3854     if (auto *OA = dyn_cast<OffloadAction>(CurAction)) {
3855       // If the dependent action is a device action, we will attempt to collapse
3856       // only with other device actions. Otherwise, we would do the same but
3857       // with host actions only.
3858       if (!IsHostSelector) {
3859         if (OA->hasSingleDeviceDependence(/*DoNotConsiderHostActions=*/true)) {
3860           CurAction =
3861               OA->getSingleDeviceDependence(/*DoNotConsiderHostActions=*/true);
3862           if (CanBeCollapsed &&
3863               !CurAction->isCollapsingWithNextDependentActionLegal())
3864             return nullptr;
3865           SavedOffloadAction.push_back(OA);
3866           return dyn_cast<JobAction>(CurAction);
3867         }
3868       } else if (OA->hasHostDependence()) {
3869         CurAction = OA->getHostDependence();
3870         if (CanBeCollapsed &&
3871             !CurAction->isCollapsingWithNextDependentActionLegal())
3872           return nullptr;
3873         SavedOffloadAction.push_back(OA);
3874         return dyn_cast<JobAction>(CurAction);
3875       }
3876       return nullptr;
3877     }
3878 
3879     return dyn_cast<JobAction>(CurAction);
3880   }
3881 
3882   /// Return true if an assemble action can be collapsed.
3883   bool canCollapseAssembleAction() const {
3884     return TC.useIntegratedAs() && !SaveTemps &&
3885            !C.getArgs().hasArg(options::OPT_via_file_asm) &&
3886            !C.getArgs().hasArg(options::OPT__SLASH_FA) &&
3887            !C.getArgs().hasArg(options::OPT__SLASH_Fa);
3888   }
3889 
3890   /// Return true if a preprocessor action can be collapsed.
3891   bool canCollapsePreprocessorAction() const {
3892     return !C.getArgs().hasArg(options::OPT_no_integrated_cpp) &&
3893            !C.getArgs().hasArg(options::OPT_traditional_cpp) && !SaveTemps &&
3894            !C.getArgs().hasArg(options::OPT_rewrite_objc);
3895   }
3896 
3897   /// Struct that relates an action with the offload actions that would be
3898   /// collapsed with it.
3899   struct JobActionInfo final {
3900     /// The action this info refers to.
3901     const JobAction *JA = nullptr;
3902     /// The offload actions we need to take care off if this action is
3903     /// collapsed.
3904     ActionList SavedOffloadAction;
3905   };
3906 
3907   /// Append collapsed offload actions from the give nnumber of elements in the
3908   /// action info array.
3909   static void AppendCollapsedOffloadAction(ActionList &CollapsedOffloadAction,
3910                                            ArrayRef<JobActionInfo> &ActionInfo,
3911                                            unsigned ElementNum) {
3912     assert(ElementNum <= ActionInfo.size() && "Invalid number of elements.");
3913     for (unsigned I = 0; I < ElementNum; ++I)
3914       CollapsedOffloadAction.append(ActionInfo[I].SavedOffloadAction.begin(),
3915                                     ActionInfo[I].SavedOffloadAction.end());
3916   }
3917 
3918   /// Functions that attempt to perform the combining. They detect if that is
3919   /// legal, and if so they update the inputs \a Inputs and the offload action
3920   /// that were collapsed in \a CollapsedOffloadAction. A tool that deals with
3921   /// the combined action is returned. If the combining is not legal or if the
3922   /// tool does not exist, null is returned.
3923   /// Currently three kinds of collapsing are supported:
3924   ///  - Assemble + Backend + Compile;
3925   ///  - Assemble + Backend ;
3926   ///  - Backend + Compile.
3927   const Tool *
3928   combineAssembleBackendCompile(ArrayRef<JobActionInfo> ActionInfo,
3929                                 ActionList &Inputs,
3930                                 ActionList &CollapsedOffloadAction) {
3931     if (ActionInfo.size() < 3 || !canCollapseAssembleAction())
3932       return nullptr;
3933     auto *AJ = dyn_cast<AssembleJobAction>(ActionInfo[0].JA);
3934     auto *BJ = dyn_cast<BackendJobAction>(ActionInfo[1].JA);
3935     auto *CJ = dyn_cast<CompileJobAction>(ActionInfo[2].JA);
3936     if (!AJ || !BJ || !CJ)
3937       return nullptr;
3938 
3939     // Get compiler tool.
3940     const Tool *T = TC.SelectTool(*CJ);
3941     if (!T)
3942       return nullptr;
3943 
3944     // When using -fembed-bitcode, it is required to have the same tool (clang)
3945     // for both CompilerJA and BackendJA. Otherwise, combine two stages.
3946     if (EmbedBitcode) {
3947       const Tool *BT = TC.SelectTool(*BJ);
3948       if (BT == T)
3949         return nullptr;
3950     }
3951 
3952     if (!T->hasIntegratedAssembler())
3953       return nullptr;
3954 
3955     Inputs = CJ->getInputs();
3956     AppendCollapsedOffloadAction(CollapsedOffloadAction, ActionInfo,
3957                                  /*NumElements=*/3);
3958     return T;
3959   }
3960   const Tool *combineAssembleBackend(ArrayRef<JobActionInfo> ActionInfo,
3961                                      ActionList &Inputs,
3962                                      ActionList &CollapsedOffloadAction) {
3963     if (ActionInfo.size() < 2 || !canCollapseAssembleAction())
3964       return nullptr;
3965     auto *AJ = dyn_cast<AssembleJobAction>(ActionInfo[0].JA);
3966     auto *BJ = dyn_cast<BackendJobAction>(ActionInfo[1].JA);
3967     if (!AJ || !BJ)
3968       return nullptr;
3969 
3970     // Get backend tool.
3971     const Tool *T = TC.SelectTool(*BJ);
3972     if (!T)
3973       return nullptr;
3974 
3975     if (!T->hasIntegratedAssembler())
3976       return nullptr;
3977 
3978     Inputs = BJ->getInputs();
3979     AppendCollapsedOffloadAction(CollapsedOffloadAction, ActionInfo,
3980                                  /*NumElements=*/2);
3981     return T;
3982   }
3983   const Tool *combineBackendCompile(ArrayRef<JobActionInfo> ActionInfo,
3984                                     ActionList &Inputs,
3985                                     ActionList &CollapsedOffloadAction) {
3986     if (ActionInfo.size() < 2)
3987       return nullptr;
3988     auto *BJ = dyn_cast<BackendJobAction>(ActionInfo[0].JA);
3989     auto *CJ = dyn_cast<CompileJobAction>(ActionInfo[1].JA);
3990     if (!BJ || !CJ)
3991       return nullptr;
3992 
3993     // Check if the initial input (to the compile job or its predessor if one
3994     // exists) is LLVM bitcode. In that case, no preprocessor step is required
3995     // and we can still collapse the compile and backend jobs when we have
3996     // -save-temps. I.e. there is no need for a separate compile job just to
3997     // emit unoptimized bitcode.
3998     bool InputIsBitcode = true;
3999     for (size_t i = 1; i < ActionInfo.size(); i++)
4000       if (ActionInfo[i].JA->getType() != types::TY_LLVM_BC &&
4001           ActionInfo[i].JA->getType() != types::TY_LTO_BC) {
4002         InputIsBitcode = false;
4003         break;
4004       }
4005     if (!InputIsBitcode && !canCollapsePreprocessorAction())
4006       return nullptr;
4007 
4008     // Get compiler tool.
4009     const Tool *T = TC.SelectTool(*CJ);
4010     if (!T)
4011       return nullptr;
4012 
4013     if (T->canEmitIR() && ((SaveTemps && !InputIsBitcode) || EmbedBitcode))
4014       return nullptr;
4015 
4016     Inputs = CJ->getInputs();
4017     AppendCollapsedOffloadAction(CollapsedOffloadAction, ActionInfo,
4018                                  /*NumElements=*/2);
4019     return T;
4020   }
4021 
4022   /// Updates the inputs if the obtained tool supports combining with
4023   /// preprocessor action, and the current input is indeed a preprocessor
4024   /// action. If combining results in the collapse of offloading actions, those
4025   /// are appended to \a CollapsedOffloadAction.
4026   void combineWithPreprocessor(const Tool *T, ActionList &Inputs,
4027                                ActionList &CollapsedOffloadAction) {
4028     if (!T || !canCollapsePreprocessorAction() || !T->hasIntegratedCPP())
4029       return;
4030 
4031     // Attempt to get a preprocessor action dependence.
4032     ActionList PreprocessJobOffloadActions;
4033     ActionList NewInputs;
4034     for (Action *A : Inputs) {
4035       auto *PJ = getPrevDependentAction({A}, PreprocessJobOffloadActions);
4036       if (!PJ || !isa<PreprocessJobAction>(PJ)) {
4037         NewInputs.push_back(A);
4038         continue;
4039       }
4040 
4041       // This is legal to combine. Append any offload action we found and add the
4042       // current input to preprocessor inputs.
4043       CollapsedOffloadAction.append(PreprocessJobOffloadActions.begin(),
4044                                     PreprocessJobOffloadActions.end());
4045       NewInputs.append(PJ->input_begin(), PJ->input_end());
4046     }
4047     Inputs = NewInputs;
4048   }
4049 
4050 public:
4051   ToolSelector(const JobAction *BaseAction, const ToolChain &TC,
4052                const Compilation &C, bool SaveTemps, bool EmbedBitcode)
4053       : TC(TC), C(C), BaseAction(BaseAction), SaveTemps(SaveTemps),
4054         EmbedBitcode(EmbedBitcode) {
4055     assert(BaseAction && "Invalid base action.");
4056     IsHostSelector = BaseAction->getOffloadingDeviceKind() == Action::OFK_None;
4057   }
4058 
4059   /// Check if a chain of actions can be combined and return the tool that can
4060   /// handle the combination of actions. The pointer to the current inputs \a
4061   /// Inputs and the list of offload actions \a CollapsedOffloadActions
4062   /// connected to collapsed actions are updated accordingly. The latter enables
4063   /// the caller of the selector to process them afterwards instead of just
4064   /// dropping them. If no suitable tool is found, null will be returned.
4065   const Tool *getTool(ActionList &Inputs,
4066                       ActionList &CollapsedOffloadAction) {
4067     //
4068     // Get the largest chain of actions that we could combine.
4069     //
4070 
4071     SmallVector<JobActionInfo, 5> ActionChain(1);
4072     ActionChain.back().JA = BaseAction;
4073     while (ActionChain.back().JA) {
4074       const Action *CurAction = ActionChain.back().JA;
4075 
4076       // Grow the chain by one element.
4077       ActionChain.resize(ActionChain.size() + 1);
4078       JobActionInfo &AI = ActionChain.back();
4079 
4080       // Attempt to fill it with the
4081       AI.JA =
4082           getPrevDependentAction(CurAction->getInputs(), AI.SavedOffloadAction);
4083     }
4084 
4085     // Pop the last action info as it could not be filled.
4086     ActionChain.pop_back();
4087 
4088     //
4089     // Attempt to combine actions. If all combining attempts failed, just return
4090     // the tool of the provided action. At the end we attempt to combine the
4091     // action with any preprocessor action it may depend on.
4092     //
4093 
4094     const Tool *T = combineAssembleBackendCompile(ActionChain, Inputs,
4095                                                   CollapsedOffloadAction);
4096     if (!T)
4097       T = combineAssembleBackend(ActionChain, Inputs, CollapsedOffloadAction);
4098     if (!T)
4099       T = combineBackendCompile(ActionChain, Inputs, CollapsedOffloadAction);
4100     if (!T) {
4101       Inputs = BaseAction->getInputs();
4102       T = TC.SelectTool(*BaseAction);
4103     }
4104 
4105     combineWithPreprocessor(T, Inputs, CollapsedOffloadAction);
4106     return T;
4107   }
4108 };
4109 }
4110 
4111 /// Return a string that uniquely identifies the result of a job. The bound arch
4112 /// is not necessarily represented in the toolchain's triple -- for example,
4113 /// armv7 and armv7s both map to the same triple -- so we need both in our map.
4114 /// Also, we need to add the offloading device kind, as the same tool chain can
4115 /// be used for host and device for some programming models, e.g. OpenMP.
4116 static std::string GetTriplePlusArchString(const ToolChain *TC,
4117                                            StringRef BoundArch,
4118                                            Action::OffloadKind OffloadKind) {
4119   std::string TriplePlusArch = TC->getTriple().normalize();
4120   if (!BoundArch.empty()) {
4121     TriplePlusArch += "-";
4122     TriplePlusArch += BoundArch;
4123   }
4124   TriplePlusArch += "-";
4125   TriplePlusArch += Action::GetOffloadKindName(OffloadKind);
4126   return TriplePlusArch;
4127 }
4128 
4129 InputInfo Driver::BuildJobsForAction(
4130     Compilation &C, const Action *A, const ToolChain *TC, StringRef BoundArch,
4131     bool AtTopLevel, bool MultipleArchs, const char *LinkingOutput,
4132     std::map<std::pair<const Action *, std::string>, InputInfo> &CachedResults,
4133     Action::OffloadKind TargetDeviceOffloadKind) const {
4134   std::pair<const Action *, std::string> ActionTC = {
4135       A, GetTriplePlusArchString(TC, BoundArch, TargetDeviceOffloadKind)};
4136   auto CachedResult = CachedResults.find(ActionTC);
4137   if (CachedResult != CachedResults.end()) {
4138     return CachedResult->second;
4139   }
4140   InputInfo Result = BuildJobsForActionNoCache(
4141       C, A, TC, BoundArch, AtTopLevel, MultipleArchs, LinkingOutput,
4142       CachedResults, TargetDeviceOffloadKind);
4143   CachedResults[ActionTC] = Result;
4144   return Result;
4145 }
4146 
4147 InputInfo Driver::BuildJobsForActionNoCache(
4148     Compilation &C, const Action *A, const ToolChain *TC, StringRef BoundArch,
4149     bool AtTopLevel, bool MultipleArchs, const char *LinkingOutput,
4150     std::map<std::pair<const Action *, std::string>, InputInfo> &CachedResults,
4151     Action::OffloadKind TargetDeviceOffloadKind) const {
4152   llvm::PrettyStackTraceString CrashInfo("Building compilation jobs");
4153 
4154   InputInfoList OffloadDependencesInputInfo;
4155   bool BuildingForOffloadDevice = TargetDeviceOffloadKind != Action::OFK_None;
4156   if (const OffloadAction *OA = dyn_cast<OffloadAction>(A)) {
4157     // The 'Darwin' toolchain is initialized only when its arguments are
4158     // computed. Get the default arguments for OFK_None to ensure that
4159     // initialization is performed before processing the offload action.
4160     // FIXME: Remove when darwin's toolchain is initialized during construction.
4161     C.getArgsForToolChain(TC, BoundArch, Action::OFK_None);
4162 
4163     // The offload action is expected to be used in four different situations.
4164     //
4165     // a) Set a toolchain/architecture/kind for a host action:
4166     //    Host Action 1 -> OffloadAction -> Host Action 2
4167     //
4168     // b) Set a toolchain/architecture/kind for a device action;
4169     //    Device Action 1 -> OffloadAction -> Device Action 2
4170     //
4171     // c) Specify a device dependence to a host action;
4172     //    Device Action 1  _
4173     //                      \
4174     //      Host Action 1  ---> OffloadAction -> Host Action 2
4175     //
4176     // d) Specify a host dependence to a device action.
4177     //      Host Action 1  _
4178     //                      \
4179     //    Device Action 1  ---> OffloadAction -> Device Action 2
4180     //
4181     // For a) and b), we just return the job generated for the dependence. For
4182     // c) and d) we override the current action with the host/device dependence
4183     // if the current toolchain is host/device and set the offload dependences
4184     // info with the jobs obtained from the device/host dependence(s).
4185 
4186     // If there is a single device option, just generate the job for it.
4187     if (OA->hasSingleDeviceDependence()) {
4188       InputInfo DevA;
4189       OA->doOnEachDeviceDependence([&](Action *DepA, const ToolChain *DepTC,
4190                                        const char *DepBoundArch) {
4191         DevA =
4192             BuildJobsForAction(C, DepA, DepTC, DepBoundArch, AtTopLevel,
4193                                /*MultipleArchs*/ !!DepBoundArch, LinkingOutput,
4194                                CachedResults, DepA->getOffloadingDeviceKind());
4195       });
4196       return DevA;
4197     }
4198 
4199     // If 'Action 2' is host, we generate jobs for the device dependences and
4200     // override the current action with the host dependence. Otherwise, we
4201     // generate the host dependences and override the action with the device
4202     // dependence. The dependences can't therefore be a top-level action.
4203     OA->doOnEachDependence(
4204         /*IsHostDependence=*/BuildingForOffloadDevice,
4205         [&](Action *DepA, const ToolChain *DepTC, const char *DepBoundArch) {
4206           OffloadDependencesInputInfo.push_back(BuildJobsForAction(
4207               C, DepA, DepTC, DepBoundArch, /*AtTopLevel=*/false,
4208               /*MultipleArchs*/ !!DepBoundArch, LinkingOutput, CachedResults,
4209               DepA->getOffloadingDeviceKind()));
4210         });
4211 
4212     A = BuildingForOffloadDevice
4213             ? OA->getSingleDeviceDependence(/*DoNotConsiderHostActions=*/true)
4214             : OA->getHostDependence();
4215   }
4216 
4217   if (const InputAction *IA = dyn_cast<InputAction>(A)) {
4218     // FIXME: It would be nice to not claim this here; maybe the old scheme of
4219     // just using Args was better?
4220     const Arg &Input = IA->getInputArg();
4221     Input.claim();
4222     if (Input.getOption().matches(options::OPT_INPUT)) {
4223       const char *Name = Input.getValue();
4224       return InputInfo(A, Name, /* _BaseInput = */ Name);
4225     }
4226     return InputInfo(A, &Input, /* _BaseInput = */ "");
4227   }
4228 
4229   if (const BindArchAction *BAA = dyn_cast<BindArchAction>(A)) {
4230     const ToolChain *TC;
4231     StringRef ArchName = BAA->getArchName();
4232 
4233     if (!ArchName.empty())
4234       TC = &getToolChain(C.getArgs(),
4235                          computeTargetTriple(*this, TargetTriple,
4236                                              C.getArgs(), ArchName));
4237     else
4238       TC = &C.getDefaultToolChain();
4239 
4240     return BuildJobsForAction(C, *BAA->input_begin(), TC, ArchName, AtTopLevel,
4241                               MultipleArchs, LinkingOutput, CachedResults,
4242                               TargetDeviceOffloadKind);
4243   }
4244 
4245 
4246   ActionList Inputs = A->getInputs();
4247 
4248   const JobAction *JA = cast<JobAction>(A);
4249   ActionList CollapsedOffloadActions;
4250 
4251   ToolSelector TS(JA, *TC, C, isSaveTempsEnabled(),
4252                   embedBitcodeInObject() && !isUsingLTO());
4253   const Tool *T = TS.getTool(Inputs, CollapsedOffloadActions);
4254 
4255   if (!T)
4256     return InputInfo();
4257 
4258   // If we've collapsed action list that contained OffloadAction we
4259   // need to build jobs for host/device-side inputs it may have held.
4260   for (const auto *OA : CollapsedOffloadActions)
4261     cast<OffloadAction>(OA)->doOnEachDependence(
4262         /*IsHostDependence=*/BuildingForOffloadDevice,
4263         [&](Action *DepA, const ToolChain *DepTC, const char *DepBoundArch) {
4264           OffloadDependencesInputInfo.push_back(BuildJobsForAction(
4265               C, DepA, DepTC, DepBoundArch, /* AtTopLevel */ false,
4266               /*MultipleArchs=*/!!DepBoundArch, LinkingOutput, CachedResults,
4267               DepA->getOffloadingDeviceKind()));
4268         });
4269 
4270   // Only use pipes when there is exactly one input.
4271   InputInfoList InputInfos;
4272   for (const Action *Input : Inputs) {
4273     // Treat dsymutil and verify sub-jobs as being at the top-level too, they
4274     // shouldn't get temporary output names.
4275     // FIXME: Clean this up.
4276     bool SubJobAtTopLevel =
4277         AtTopLevel && (isa<DsymutilJobAction>(A) || isa<VerifyJobAction>(A));
4278     InputInfos.push_back(BuildJobsForAction(
4279         C, Input, TC, BoundArch, SubJobAtTopLevel, MultipleArchs, LinkingOutput,
4280         CachedResults, A->getOffloadingDeviceKind()));
4281   }
4282 
4283   // Always use the first input as the base input.
4284   const char *BaseInput = InputInfos[0].getBaseInput();
4285 
4286   // ... except dsymutil actions, which use their actual input as the base
4287   // input.
4288   if (JA->getType() == types::TY_dSYM)
4289     BaseInput = InputInfos[0].getFilename();
4290 
4291   // ... and in header module compilations, which use the module name.
4292   if (auto *ModuleJA = dyn_cast<HeaderModulePrecompileJobAction>(JA))
4293     BaseInput = ModuleJA->getModuleName();
4294 
4295   // Append outputs of offload device jobs to the input list
4296   if (!OffloadDependencesInputInfo.empty())
4297     InputInfos.append(OffloadDependencesInputInfo.begin(),
4298                       OffloadDependencesInputInfo.end());
4299 
4300   // Set the effective triple of the toolchain for the duration of this job.
4301   llvm::Triple EffectiveTriple;
4302   const ToolChain &ToolTC = T->getToolChain();
4303   const ArgList &Args =
4304       C.getArgsForToolChain(TC, BoundArch, A->getOffloadingDeviceKind());
4305   if (InputInfos.size() != 1) {
4306     EffectiveTriple = llvm::Triple(ToolTC.ComputeEffectiveClangTriple(Args));
4307   } else {
4308     // Pass along the input type if it can be unambiguously determined.
4309     EffectiveTriple = llvm::Triple(
4310         ToolTC.ComputeEffectiveClangTriple(Args, InputInfos[0].getType()));
4311   }
4312   RegisterEffectiveTriple TripleRAII(ToolTC, EffectiveTriple);
4313 
4314   // Determine the place to write output to, if any.
4315   InputInfo Result;
4316   InputInfoList UnbundlingResults;
4317   if (auto *UA = dyn_cast<OffloadUnbundlingJobAction>(JA)) {
4318     // If we have an unbundling job, we need to create results for all the
4319     // outputs. We also update the results cache so that other actions using
4320     // this unbundling action can get the right results.
4321     for (auto &UI : UA->getDependentActionsInfo()) {
4322       assert(UI.DependentOffloadKind != Action::OFK_None &&
4323              "Unbundling with no offloading??");
4324 
4325       // Unbundling actions are never at the top level. When we generate the
4326       // offloading prefix, we also do that for the host file because the
4327       // unbundling action does not change the type of the output which can
4328       // cause a overwrite.
4329       std::string OffloadingPrefix = Action::GetOffloadingFileNamePrefix(
4330           UI.DependentOffloadKind,
4331           UI.DependentToolChain->getTriple().normalize(),
4332           /*CreatePrefixForHost=*/true);
4333       auto CurI = InputInfo(
4334           UA,
4335           GetNamedOutputPath(C, *UA, BaseInput, UI.DependentBoundArch,
4336                              /*AtTopLevel=*/false,
4337                              MultipleArchs ||
4338                                  UI.DependentOffloadKind == Action::OFK_HIP,
4339                              OffloadingPrefix),
4340           BaseInput);
4341       // Save the unbundling result.
4342       UnbundlingResults.push_back(CurI);
4343 
4344       // Get the unique string identifier for this dependence and cache the
4345       // result.
4346       StringRef Arch;
4347       if (TargetDeviceOffloadKind == Action::OFK_HIP) {
4348         if (UI.DependentOffloadKind == Action::OFK_Host)
4349           Arch = StringRef();
4350         else
4351           Arch = UI.DependentBoundArch;
4352       } else
4353         Arch = BoundArch;
4354 
4355       CachedResults[{A, GetTriplePlusArchString(UI.DependentToolChain, Arch,
4356                                                 UI.DependentOffloadKind)}] =
4357           CurI;
4358     }
4359 
4360     // Now that we have all the results generated, select the one that should be
4361     // returned for the current depending action.
4362     std::pair<const Action *, std::string> ActionTC = {
4363         A, GetTriplePlusArchString(TC, BoundArch, TargetDeviceOffloadKind)};
4364     assert(CachedResults.find(ActionTC) != CachedResults.end() &&
4365            "Result does not exist??");
4366     Result = CachedResults[ActionTC];
4367   } else if (JA->getType() == types::TY_Nothing)
4368     Result = InputInfo(A, BaseInput);
4369   else {
4370     // We only have to generate a prefix for the host if this is not a top-level
4371     // action.
4372     std::string OffloadingPrefix = Action::GetOffloadingFileNamePrefix(
4373         A->getOffloadingDeviceKind(), TC->getTriple().normalize(),
4374         /*CreatePrefixForHost=*/!!A->getOffloadingHostActiveKinds() &&
4375             !AtTopLevel);
4376     if (isa<OffloadWrapperJobAction>(JA)) {
4377       OffloadingPrefix += "-wrapper";
4378       if (Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o))
4379         BaseInput = FinalOutput->getValue();
4380       else
4381         BaseInput = getDefaultImageName();
4382     }
4383     Result = InputInfo(A, GetNamedOutputPath(C, *JA, BaseInput, BoundArch,
4384                                              AtTopLevel, MultipleArchs,
4385                                              OffloadingPrefix),
4386                        BaseInput);
4387   }
4388 
4389   if (CCCPrintBindings && !CCGenDiagnostics) {
4390     llvm::errs() << "# \"" << T->getToolChain().getTripleString() << '"'
4391                  << " - \"" << T->getName() << "\", inputs: [";
4392     for (unsigned i = 0, e = InputInfos.size(); i != e; ++i) {
4393       llvm::errs() << InputInfos[i].getAsString();
4394       if (i + 1 != e)
4395         llvm::errs() << ", ";
4396     }
4397     if (UnbundlingResults.empty())
4398       llvm::errs() << "], output: " << Result.getAsString() << "\n";
4399     else {
4400       llvm::errs() << "], outputs: [";
4401       for (unsigned i = 0, e = UnbundlingResults.size(); i != e; ++i) {
4402         llvm::errs() << UnbundlingResults[i].getAsString();
4403         if (i + 1 != e)
4404           llvm::errs() << ", ";
4405       }
4406       llvm::errs() << "] \n";
4407     }
4408   } else {
4409     if (UnbundlingResults.empty())
4410       T->ConstructJob(
4411           C, *JA, Result, InputInfos,
4412           C.getArgsForToolChain(TC, BoundArch, JA->getOffloadingDeviceKind()),
4413           LinkingOutput);
4414     else
4415       T->ConstructJobMultipleOutputs(
4416           C, *JA, UnbundlingResults, InputInfos,
4417           C.getArgsForToolChain(TC, BoundArch, JA->getOffloadingDeviceKind()),
4418           LinkingOutput);
4419   }
4420   return Result;
4421 }
4422 
4423 const char *Driver::getDefaultImageName() const {
4424   llvm::Triple Target(llvm::Triple::normalize(TargetTriple));
4425   return Target.isOSWindows() ? "a.exe" : "a.out";
4426 }
4427 
4428 /// Create output filename based on ArgValue, which could either be a
4429 /// full filename, filename without extension, or a directory. If ArgValue
4430 /// does not provide a filename, then use BaseName, and use the extension
4431 /// suitable for FileType.
4432 static const char *MakeCLOutputFilename(const ArgList &Args, StringRef ArgValue,
4433                                         StringRef BaseName,
4434                                         types::ID FileType) {
4435   SmallString<128> Filename = ArgValue;
4436 
4437   if (ArgValue.empty()) {
4438     // If the argument is empty, output to BaseName in the current dir.
4439     Filename = BaseName;
4440   } else if (llvm::sys::path::is_separator(Filename.back())) {
4441     // If the argument is a directory, output to BaseName in that dir.
4442     llvm::sys::path::append(Filename, BaseName);
4443   }
4444 
4445   if (!llvm::sys::path::has_extension(ArgValue)) {
4446     // If the argument didn't provide an extension, then set it.
4447     const char *Extension = types::getTypeTempSuffix(FileType, true);
4448 
4449     if (FileType == types::TY_Image &&
4450         Args.hasArg(options::OPT__SLASH_LD, options::OPT__SLASH_LDd)) {
4451       // The output file is a dll.
4452       Extension = "dll";
4453     }
4454 
4455     llvm::sys::path::replace_extension(Filename, Extension);
4456   }
4457 
4458   return Args.MakeArgString(Filename.c_str());
4459 }
4460 
4461 const char *Driver::GetNamedOutputPath(Compilation &C, const JobAction &JA,
4462                                        const char *BaseInput,
4463                                        StringRef BoundArch, bool AtTopLevel,
4464                                        bool MultipleArchs,
4465                                        StringRef OffloadingPrefix) const {
4466   llvm::PrettyStackTraceString CrashInfo("Computing output path");
4467   // Output to a user requested destination?
4468   if (AtTopLevel && !isa<DsymutilJobAction>(JA) && !isa<VerifyJobAction>(JA)) {
4469     if (Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o))
4470       return C.addResultFile(FinalOutput->getValue(), &JA);
4471   }
4472 
4473   // For /P, preprocess to file named after BaseInput.
4474   if (C.getArgs().hasArg(options::OPT__SLASH_P)) {
4475     assert(AtTopLevel && isa<PreprocessJobAction>(JA));
4476     StringRef BaseName = llvm::sys::path::filename(BaseInput);
4477     StringRef NameArg;
4478     if (Arg *A = C.getArgs().getLastArg(options::OPT__SLASH_Fi))
4479       NameArg = A->getValue();
4480     return C.addResultFile(
4481         MakeCLOutputFilename(C.getArgs(), NameArg, BaseName, types::TY_PP_C),
4482         &JA);
4483   }
4484 
4485   // Default to writing to stdout?
4486   if (AtTopLevel && !CCGenDiagnostics && isa<PreprocessJobAction>(JA))
4487     return "-";
4488 
4489   // Is this the assembly listing for /FA?
4490   if (JA.getType() == types::TY_PP_Asm &&
4491       (C.getArgs().hasArg(options::OPT__SLASH_FA) ||
4492        C.getArgs().hasArg(options::OPT__SLASH_Fa))) {
4493     // Use /Fa and the input filename to determine the asm file name.
4494     StringRef BaseName = llvm::sys::path::filename(BaseInput);
4495     StringRef FaValue = C.getArgs().getLastArgValue(options::OPT__SLASH_Fa);
4496     return C.addResultFile(
4497         MakeCLOutputFilename(C.getArgs(), FaValue, BaseName, JA.getType()),
4498         &JA);
4499   }
4500 
4501   // Output to a temporary file?
4502   if ((!AtTopLevel && !isSaveTempsEnabled() &&
4503        !C.getArgs().hasArg(options::OPT__SLASH_Fo)) ||
4504       CCGenDiagnostics) {
4505     StringRef Name = llvm::sys::path::filename(BaseInput);
4506     std::pair<StringRef, StringRef> Split = Name.split('.');
4507     SmallString<128> TmpName;
4508     const char *Suffix = types::getTypeTempSuffix(JA.getType(), IsCLMode());
4509     Arg *A = C.getArgs().getLastArg(options::OPT_fcrash_diagnostics_dir);
4510     if (CCGenDiagnostics && A) {
4511       SmallString<128> CrashDirectory(A->getValue());
4512       if (!getVFS().exists(CrashDirectory))
4513         llvm::sys::fs::create_directories(CrashDirectory);
4514       llvm::sys::path::append(CrashDirectory, Split.first);
4515       const char *Middle = Suffix ? "-%%%%%%." : "-%%%%%%";
4516       std::error_code EC = llvm::sys::fs::createUniqueFile(
4517           CrashDirectory + Middle + Suffix, TmpName);
4518       if (EC) {
4519         Diag(clang::diag::err_unable_to_make_temp) << EC.message();
4520         return "";
4521       }
4522     } else {
4523       TmpName = GetTemporaryPath(Split.first, Suffix);
4524     }
4525     return C.addTempFile(C.getArgs().MakeArgString(TmpName));
4526   }
4527 
4528   SmallString<128> BasePath(BaseInput);
4529   StringRef BaseName;
4530 
4531   // Dsymutil actions should use the full path.
4532   if (isa<DsymutilJobAction>(JA) || isa<VerifyJobAction>(JA))
4533     BaseName = BasePath;
4534   else
4535     BaseName = llvm::sys::path::filename(BasePath);
4536 
4537   // Determine what the derived output name should be.
4538   const char *NamedOutput;
4539 
4540   if ((JA.getType() == types::TY_Object || JA.getType() == types::TY_LTO_BC) &&
4541       C.getArgs().hasArg(options::OPT__SLASH_Fo, options::OPT__SLASH_o)) {
4542     // The /Fo or /o flag decides the object filename.
4543     StringRef Val =
4544         C.getArgs()
4545             .getLastArg(options::OPT__SLASH_Fo, options::OPT__SLASH_o)
4546             ->getValue();
4547     NamedOutput =
4548         MakeCLOutputFilename(C.getArgs(), Val, BaseName, types::TY_Object);
4549   } else if (JA.getType() == types::TY_Image &&
4550              C.getArgs().hasArg(options::OPT__SLASH_Fe,
4551                                 options::OPT__SLASH_o)) {
4552     // The /Fe or /o flag names the linked file.
4553     StringRef Val =
4554         C.getArgs()
4555             .getLastArg(options::OPT__SLASH_Fe, options::OPT__SLASH_o)
4556             ->getValue();
4557     NamedOutput =
4558         MakeCLOutputFilename(C.getArgs(), Val, BaseName, types::TY_Image);
4559   } else if (JA.getType() == types::TY_Image) {
4560     if (IsCLMode()) {
4561       // clang-cl uses BaseName for the executable name.
4562       NamedOutput =
4563           MakeCLOutputFilename(C.getArgs(), "", BaseName, types::TY_Image);
4564     } else {
4565       SmallString<128> Output(getDefaultImageName());
4566       // HIP image for device compilation with -fno-gpu-rdc is per compilation
4567       // unit.
4568       bool IsHIPNoRDC = JA.getOffloadingDeviceKind() == Action::OFK_HIP &&
4569                         !C.getArgs().hasFlag(options::OPT_fgpu_rdc,
4570                                              options::OPT_fno_gpu_rdc, false);
4571       if (IsHIPNoRDC) {
4572         Output = BaseName;
4573         llvm::sys::path::replace_extension(Output, "");
4574       }
4575       Output += OffloadingPrefix;
4576       if (MultipleArchs && !BoundArch.empty()) {
4577         Output += "-";
4578         Output.append(BoundArch);
4579       }
4580       if (IsHIPNoRDC)
4581         Output += ".out";
4582       NamedOutput = C.getArgs().MakeArgString(Output.c_str());
4583     }
4584   } else if (JA.getType() == types::TY_PCH && IsCLMode()) {
4585     NamedOutput = C.getArgs().MakeArgString(GetClPchPath(C, BaseName));
4586   } else {
4587     const char *Suffix = types::getTypeTempSuffix(JA.getType(), IsCLMode());
4588     assert(Suffix && "All types used for output should have a suffix.");
4589 
4590     std::string::size_type End = std::string::npos;
4591     if (!types::appendSuffixForType(JA.getType()))
4592       End = BaseName.rfind('.');
4593     SmallString<128> Suffixed(BaseName.substr(0, End));
4594     Suffixed += OffloadingPrefix;
4595     if (MultipleArchs && !BoundArch.empty()) {
4596       Suffixed += "-";
4597       Suffixed.append(BoundArch);
4598     }
4599     // When using both -save-temps and -emit-llvm, use a ".tmp.bc" suffix for
4600     // the unoptimized bitcode so that it does not get overwritten by the ".bc"
4601     // optimized bitcode output.
4602     if (!AtTopLevel && C.getArgs().hasArg(options::OPT_emit_llvm) &&
4603         JA.getType() == types::TY_LLVM_BC)
4604       Suffixed += ".tmp";
4605     Suffixed += '.';
4606     Suffixed += Suffix;
4607     NamedOutput = C.getArgs().MakeArgString(Suffixed.c_str());
4608   }
4609 
4610   // Prepend object file path if -save-temps=obj
4611   if (!AtTopLevel && isSaveTempsObj() && C.getArgs().hasArg(options::OPT_o) &&
4612       JA.getType() != types::TY_PCH) {
4613     Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o);
4614     SmallString<128> TempPath(FinalOutput->getValue());
4615     llvm::sys::path::remove_filename(TempPath);
4616     StringRef OutputFileName = llvm::sys::path::filename(NamedOutput);
4617     llvm::sys::path::append(TempPath, OutputFileName);
4618     NamedOutput = C.getArgs().MakeArgString(TempPath.c_str());
4619   }
4620 
4621   // If we're saving temps and the temp file conflicts with the input file,
4622   // then avoid overwriting input file.
4623   if (!AtTopLevel && isSaveTempsEnabled() && NamedOutput == BaseName) {
4624     bool SameFile = false;
4625     SmallString<256> Result;
4626     llvm::sys::fs::current_path(Result);
4627     llvm::sys::path::append(Result, BaseName);
4628     llvm::sys::fs::equivalent(BaseInput, Result.c_str(), SameFile);
4629     // Must share the same path to conflict.
4630     if (SameFile) {
4631       StringRef Name = llvm::sys::path::filename(BaseInput);
4632       std::pair<StringRef, StringRef> Split = Name.split('.');
4633       std::string TmpName = GetTemporaryPath(
4634           Split.first, types::getTypeTempSuffix(JA.getType(), IsCLMode()));
4635       return C.addTempFile(C.getArgs().MakeArgString(TmpName));
4636     }
4637   }
4638 
4639   // As an annoying special case, PCH generation doesn't strip the pathname.
4640   if (JA.getType() == types::TY_PCH && !IsCLMode()) {
4641     llvm::sys::path::remove_filename(BasePath);
4642     if (BasePath.empty())
4643       BasePath = NamedOutput;
4644     else
4645       llvm::sys::path::append(BasePath, NamedOutput);
4646     return C.addResultFile(C.getArgs().MakeArgString(BasePath.c_str()), &JA);
4647   } else {
4648     return C.addResultFile(NamedOutput, &JA);
4649   }
4650 }
4651 
4652 std::string Driver::GetFilePath(StringRef Name, const ToolChain &TC) const {
4653   // Search for Name in a list of paths.
4654   auto SearchPaths = [&](const llvm::SmallVectorImpl<std::string> &P)
4655       -> llvm::Optional<std::string> {
4656     // Respect a limited subset of the '-Bprefix' functionality in GCC by
4657     // attempting to use this prefix when looking for file paths.
4658     for (const auto &Dir : P) {
4659       if (Dir.empty())
4660         continue;
4661       SmallString<128> P(Dir[0] == '=' ? SysRoot + Dir.substr(1) : Dir);
4662       llvm::sys::path::append(P, Name);
4663       if (llvm::sys::fs::exists(Twine(P)))
4664         return std::string(P);
4665     }
4666     return None;
4667   };
4668 
4669   if (auto P = SearchPaths(PrefixDirs))
4670     return *P;
4671 
4672   SmallString<128> R(ResourceDir);
4673   llvm::sys::path::append(R, Name);
4674   if (llvm::sys::fs::exists(Twine(R)))
4675     return std::string(R.str());
4676 
4677   SmallString<128> P(TC.getCompilerRTPath());
4678   llvm::sys::path::append(P, Name);
4679   if (llvm::sys::fs::exists(Twine(P)))
4680     return std::string(P.str());
4681 
4682   SmallString<128> D(Dir);
4683   llvm::sys::path::append(D, "..", Name);
4684   if (llvm::sys::fs::exists(Twine(D)))
4685     return std::string(D.str());
4686 
4687   if (auto P = SearchPaths(TC.getLibraryPaths()))
4688     return *P;
4689 
4690   if (auto P = SearchPaths(TC.getFilePaths()))
4691     return *P;
4692 
4693   return std::string(Name);
4694 }
4695 
4696 void Driver::generatePrefixedToolNames(
4697     StringRef Tool, const ToolChain &TC,
4698     SmallVectorImpl<std::string> &Names) const {
4699   // FIXME: Needs a better variable than TargetTriple
4700   Names.emplace_back((TargetTriple + "-" + Tool).str());
4701   Names.emplace_back(Tool);
4702 
4703   // Allow the discovery of tools prefixed with LLVM's default target triple.
4704   std::string DefaultTargetTriple = llvm::sys::getDefaultTargetTriple();
4705   if (DefaultTargetTriple != TargetTriple)
4706     Names.emplace_back((DefaultTargetTriple + "-" + Tool).str());
4707 }
4708 
4709 static bool ScanDirForExecutable(SmallString<128> &Dir,
4710                                  ArrayRef<std::string> Names) {
4711   for (const auto &Name : Names) {
4712     llvm::sys::path::append(Dir, Name);
4713     if (llvm::sys::fs::can_execute(Twine(Dir)))
4714       return true;
4715     llvm::sys::path::remove_filename(Dir);
4716   }
4717   return false;
4718 }
4719 
4720 std::string Driver::GetProgramPath(StringRef Name, const ToolChain &TC) const {
4721   SmallVector<std::string, 2> TargetSpecificExecutables;
4722   generatePrefixedToolNames(Name, TC, TargetSpecificExecutables);
4723 
4724   // Respect a limited subset of the '-Bprefix' functionality in GCC by
4725   // attempting to use this prefix when looking for program paths.
4726   for (const auto &PrefixDir : PrefixDirs) {
4727     if (llvm::sys::fs::is_directory(PrefixDir)) {
4728       SmallString<128> P(PrefixDir);
4729       if (ScanDirForExecutable(P, TargetSpecificExecutables))
4730         return std::string(P.str());
4731     } else {
4732       SmallString<128> P((PrefixDir + Name).str());
4733       if (llvm::sys::fs::can_execute(Twine(P)))
4734         return std::string(P.str());
4735     }
4736   }
4737 
4738   const ToolChain::path_list &List = TC.getProgramPaths();
4739   for (const auto &Path : List) {
4740     SmallString<128> P(Path);
4741     if (ScanDirForExecutable(P, TargetSpecificExecutables))
4742       return std::string(P.str());
4743   }
4744 
4745   // If all else failed, search the path.
4746   for (const auto &TargetSpecificExecutable : TargetSpecificExecutables)
4747     if (llvm::ErrorOr<std::string> P =
4748             llvm::sys::findProgramByName(TargetSpecificExecutable))
4749       return *P;
4750 
4751   return std::string(Name);
4752 }
4753 
4754 std::string Driver::GetTemporaryPath(StringRef Prefix, StringRef Suffix) const {
4755   SmallString<128> Path;
4756   std::error_code EC = llvm::sys::fs::createTemporaryFile(Prefix, Suffix, Path);
4757   if (EC) {
4758     Diag(clang::diag::err_unable_to_make_temp) << EC.message();
4759     return "";
4760   }
4761 
4762   return std::string(Path.str());
4763 }
4764 
4765 std::string Driver::GetTemporaryDirectory(StringRef Prefix) const {
4766   SmallString<128> Path;
4767   std::error_code EC = llvm::sys::fs::createUniqueDirectory(Prefix, Path);
4768   if (EC) {
4769     Diag(clang::diag::err_unable_to_make_temp) << EC.message();
4770     return "";
4771   }
4772 
4773   return std::string(Path.str());
4774 }
4775 
4776 std::string Driver::GetClPchPath(Compilation &C, StringRef BaseName) const {
4777   SmallString<128> Output;
4778   if (Arg *FpArg = C.getArgs().getLastArg(options::OPT__SLASH_Fp)) {
4779     // FIXME: If anybody needs it, implement this obscure rule:
4780     // "If you specify a directory without a file name, the default file name
4781     // is VCx0.pch., where x is the major version of Visual C++ in use."
4782     Output = FpArg->getValue();
4783 
4784     // "If you do not specify an extension as part of the path name, an
4785     // extension of .pch is assumed. "
4786     if (!llvm::sys::path::has_extension(Output))
4787       Output += ".pch";
4788   } else {
4789     if (Arg *YcArg = C.getArgs().getLastArg(options::OPT__SLASH_Yc))
4790       Output = YcArg->getValue();
4791     if (Output.empty())
4792       Output = BaseName;
4793     llvm::sys::path::replace_extension(Output, ".pch");
4794   }
4795   return std::string(Output.str());
4796 }
4797 
4798 const ToolChain &Driver::getToolChain(const ArgList &Args,
4799                                       const llvm::Triple &Target) const {
4800 
4801   auto &TC = ToolChains[Target.str()];
4802   if (!TC) {
4803     switch (Target.getOS()) {
4804     case llvm::Triple::AIX:
4805       TC = std::make_unique<toolchains::AIX>(*this, Target, Args);
4806       break;
4807     case llvm::Triple::Haiku:
4808       TC = std::make_unique<toolchains::Haiku>(*this, Target, Args);
4809       break;
4810     case llvm::Triple::Ananas:
4811       TC = std::make_unique<toolchains::Ananas>(*this, Target, Args);
4812       break;
4813     case llvm::Triple::CloudABI:
4814       TC = std::make_unique<toolchains::CloudABI>(*this, Target, Args);
4815       break;
4816     case llvm::Triple::Darwin:
4817     case llvm::Triple::MacOSX:
4818     case llvm::Triple::IOS:
4819     case llvm::Triple::TvOS:
4820     case llvm::Triple::WatchOS:
4821       TC = std::make_unique<toolchains::DarwinClang>(*this, Target, Args);
4822       break;
4823     case llvm::Triple::DragonFly:
4824       TC = std::make_unique<toolchains::DragonFly>(*this, Target, Args);
4825       break;
4826     case llvm::Triple::OpenBSD:
4827       TC = std::make_unique<toolchains::OpenBSD>(*this, Target, Args);
4828       break;
4829     case llvm::Triple::NetBSD:
4830       TC = std::make_unique<toolchains::NetBSD>(*this, Target, Args);
4831       break;
4832     case llvm::Triple::FreeBSD:
4833       TC = std::make_unique<toolchains::FreeBSD>(*this, Target, Args);
4834       break;
4835     case llvm::Triple::Minix:
4836       TC = std::make_unique<toolchains::Minix>(*this, Target, Args);
4837       break;
4838     case llvm::Triple::Linux:
4839     case llvm::Triple::ELFIAMCU:
4840       if (Target.getArch() == llvm::Triple::hexagon)
4841         TC = std::make_unique<toolchains::HexagonToolChain>(*this, Target,
4842                                                              Args);
4843       else if ((Target.getVendor() == llvm::Triple::MipsTechnologies) &&
4844                !Target.hasEnvironment())
4845         TC = std::make_unique<toolchains::MipsLLVMToolChain>(*this, Target,
4846                                                               Args);
4847       else if (Target.getArch() == llvm::Triple::ppc ||
4848                Target.getArch() == llvm::Triple::ppc64 ||
4849                Target.getArch() == llvm::Triple::ppc64le)
4850         TC = std::make_unique<toolchains::PPCLinuxToolChain>(*this, Target,
4851                                                               Args);
4852       else
4853         TC = std::make_unique<toolchains::Linux>(*this, Target, Args);
4854       break;
4855     case llvm::Triple::NaCl:
4856       TC = std::make_unique<toolchains::NaClToolChain>(*this, Target, Args);
4857       break;
4858     case llvm::Triple::Fuchsia:
4859       TC = std::make_unique<toolchains::Fuchsia>(*this, Target, Args);
4860       break;
4861     case llvm::Triple::Solaris:
4862       TC = std::make_unique<toolchains::Solaris>(*this, Target, Args);
4863       break;
4864     case llvm::Triple::AMDHSA:
4865       TC = std::make_unique<toolchains::ROCMToolChain>(*this, Target, Args);
4866       break;
4867     case llvm::Triple::AMDPAL:
4868     case llvm::Triple::Mesa3D:
4869       TC = std::make_unique<toolchains::AMDGPUToolChain>(*this, Target, Args);
4870       break;
4871     case llvm::Triple::Win32:
4872       switch (Target.getEnvironment()) {
4873       default:
4874         if (Target.isOSBinFormatELF())
4875           TC = std::make_unique<toolchains::Generic_ELF>(*this, Target, Args);
4876         else if (Target.isOSBinFormatMachO())
4877           TC = std::make_unique<toolchains::MachO>(*this, Target, Args);
4878         else
4879           TC = std::make_unique<toolchains::Generic_GCC>(*this, Target, Args);
4880         break;
4881       case llvm::Triple::GNU:
4882         TC = std::make_unique<toolchains::MinGW>(*this, Target, Args);
4883         break;
4884       case llvm::Triple::Itanium:
4885         TC = std::make_unique<toolchains::CrossWindowsToolChain>(*this, Target,
4886                                                                   Args);
4887         break;
4888       case llvm::Triple::MSVC:
4889       case llvm::Triple::UnknownEnvironment:
4890         if (Args.getLastArgValue(options::OPT_fuse_ld_EQ)
4891                 .startswith_lower("bfd"))
4892           TC = std::make_unique<toolchains::CrossWindowsToolChain>(
4893               *this, Target, Args);
4894         else
4895           TC =
4896               std::make_unique<toolchains::MSVCToolChain>(*this, Target, Args);
4897         break;
4898       }
4899       break;
4900     case llvm::Triple::PS4:
4901       TC = std::make_unique<toolchains::PS4CPU>(*this, Target, Args);
4902       break;
4903     case llvm::Triple::Contiki:
4904       TC = std::make_unique<toolchains::Contiki>(*this, Target, Args);
4905       break;
4906     case llvm::Triple::Hurd:
4907       TC = std::make_unique<toolchains::Hurd>(*this, Target, Args);
4908       break;
4909     default:
4910       // Of these targets, Hexagon is the only one that might have
4911       // an OS of Linux, in which case it got handled above already.
4912       switch (Target.getArch()) {
4913       case llvm::Triple::tce:
4914         TC = std::make_unique<toolchains::TCEToolChain>(*this, Target, Args);
4915         break;
4916       case llvm::Triple::tcele:
4917         TC = std::make_unique<toolchains::TCELEToolChain>(*this, Target, Args);
4918         break;
4919       case llvm::Triple::hexagon:
4920         TC = std::make_unique<toolchains::HexagonToolChain>(*this, Target,
4921                                                              Args);
4922         break;
4923       case llvm::Triple::lanai:
4924         TC = std::make_unique<toolchains::LanaiToolChain>(*this, Target, Args);
4925         break;
4926       case llvm::Triple::xcore:
4927         TC = std::make_unique<toolchains::XCoreToolChain>(*this, Target, Args);
4928         break;
4929       case llvm::Triple::wasm32:
4930       case llvm::Triple::wasm64:
4931         TC = std::make_unique<toolchains::WebAssembly>(*this, Target, Args);
4932         break;
4933       case llvm::Triple::avr:
4934         TC = std::make_unique<toolchains::AVRToolChain>(*this, Target, Args);
4935         break;
4936       case llvm::Triple::msp430:
4937         TC =
4938             std::make_unique<toolchains::MSP430ToolChain>(*this, Target, Args);
4939         break;
4940       case llvm::Triple::riscv32:
4941       case llvm::Triple::riscv64:
4942         TC = std::make_unique<toolchains::RISCVToolChain>(*this, Target, Args);
4943         break;
4944       default:
4945         if (Target.getVendor() == llvm::Triple::Myriad)
4946           TC = std::make_unique<toolchains::MyriadToolChain>(*this, Target,
4947                                                               Args);
4948         else if (toolchains::BareMetal::handlesTarget(Target))
4949           TC = std::make_unique<toolchains::BareMetal>(*this, Target, Args);
4950         else if (Target.isOSBinFormatELF())
4951           TC = std::make_unique<toolchains::Generic_ELF>(*this, Target, Args);
4952         else if (Target.isOSBinFormatMachO())
4953           TC = std::make_unique<toolchains::MachO>(*this, Target, Args);
4954         else
4955           TC = std::make_unique<toolchains::Generic_GCC>(*this, Target, Args);
4956       }
4957     }
4958   }
4959 
4960   // Intentionally omitted from the switch above: llvm::Triple::CUDA.  CUDA
4961   // compiles always need two toolchains, the CUDA toolchain and the host
4962   // toolchain.  So the only valid way to create a CUDA toolchain is via
4963   // CreateOffloadingDeviceToolChains.
4964 
4965   return *TC;
4966 }
4967 
4968 bool Driver::ShouldUseClangCompiler(const JobAction &JA) const {
4969   // Say "no" if there is not exactly one input of a type clang understands.
4970   if (JA.size() != 1 ||
4971       !types::isAcceptedByClang((*JA.input_begin())->getType()))
4972     return false;
4973 
4974   // And say "no" if this is not a kind of action clang understands.
4975   if (!isa<PreprocessJobAction>(JA) && !isa<PrecompileJobAction>(JA) &&
4976       !isa<CompileJobAction>(JA) && !isa<BackendJobAction>(JA))
4977     return false;
4978 
4979   return true;
4980 }
4981 
4982 bool Driver::ShouldUseFlangCompiler(const JobAction &JA) const {
4983   // Say "no" if there is not exactly one input of a type flang understands.
4984   if (JA.size() != 1 ||
4985       !types::isFortran((*JA.input_begin())->getType()))
4986     return false;
4987 
4988   // And say "no" if this is not a kind of action flang understands.
4989   if (!isa<PreprocessJobAction>(JA) && !isa<CompileJobAction>(JA) && !isa<BackendJobAction>(JA))
4990     return false;
4991 
4992   return true;
4993 }
4994 
4995 /// GetReleaseVersion - Parse (([0-9]+)(.([0-9]+)(.([0-9]+)?))?)? and return the
4996 /// grouped values as integers. Numbers which are not provided are set to 0.
4997 ///
4998 /// \return True if the entire string was parsed (9.2), or all groups were
4999 /// parsed (10.3.5extrastuff).
5000 bool Driver::GetReleaseVersion(StringRef Str, unsigned &Major, unsigned &Minor,
5001                                unsigned &Micro, bool &HadExtra) {
5002   HadExtra = false;
5003 
5004   Major = Minor = Micro = 0;
5005   if (Str.empty())
5006     return false;
5007 
5008   if (Str.consumeInteger(10, Major))
5009     return false;
5010   if (Str.empty())
5011     return true;
5012   if (Str[0] != '.')
5013     return false;
5014 
5015   Str = Str.drop_front(1);
5016 
5017   if (Str.consumeInteger(10, Minor))
5018     return false;
5019   if (Str.empty())
5020     return true;
5021   if (Str[0] != '.')
5022     return false;
5023   Str = Str.drop_front(1);
5024 
5025   if (Str.consumeInteger(10, Micro))
5026     return false;
5027   if (!Str.empty())
5028     HadExtra = true;
5029   return true;
5030 }
5031 
5032 /// Parse digits from a string \p Str and fulfill \p Digits with
5033 /// the parsed numbers. This method assumes that the max number of
5034 /// digits to look for is equal to Digits.size().
5035 ///
5036 /// \return True if the entire string was parsed and there are
5037 /// no extra characters remaining at the end.
5038 bool Driver::GetReleaseVersion(StringRef Str,
5039                                MutableArrayRef<unsigned> Digits) {
5040   if (Str.empty())
5041     return false;
5042 
5043   unsigned CurDigit = 0;
5044   while (CurDigit < Digits.size()) {
5045     unsigned Digit;
5046     if (Str.consumeInteger(10, Digit))
5047       return false;
5048     Digits[CurDigit] = Digit;
5049     if (Str.empty())
5050       return true;
5051     if (Str[0] != '.')
5052       return false;
5053     Str = Str.drop_front(1);
5054     CurDigit++;
5055   }
5056 
5057   // More digits than requested, bail out...
5058   return false;
5059 }
5060 
5061 std::pair<unsigned, unsigned>
5062 Driver::getIncludeExcludeOptionFlagMasks(bool IsClCompatMode) const {
5063   unsigned IncludedFlagsBitmask = 0;
5064   unsigned ExcludedFlagsBitmask = options::NoDriverOption;
5065 
5066   if (IsClCompatMode) {
5067     // Include CL and Core options.
5068     IncludedFlagsBitmask |= options::CLOption;
5069     IncludedFlagsBitmask |= options::CoreOption;
5070   } else {
5071     ExcludedFlagsBitmask |= options::CLOption;
5072   }
5073 
5074   return std::make_pair(IncludedFlagsBitmask, ExcludedFlagsBitmask);
5075 }
5076 
5077 bool clang::driver::isOptimizationLevelFast(const ArgList &Args) {
5078   return Args.hasFlag(options::OPT_Ofast, options::OPT_O_Group, false);
5079 }
5080 
5081 bool clang::driver::willEmitRemarks(const ArgList &Args) {
5082   // -fsave-optimization-record enables it.
5083   if (Args.hasFlag(options::OPT_fsave_optimization_record,
5084                    options::OPT_fno_save_optimization_record, false))
5085     return true;
5086 
5087   // -fsave-optimization-record=<format> enables it as well.
5088   if (Args.hasFlag(options::OPT_fsave_optimization_record_EQ,
5089                    options::OPT_fno_save_optimization_record, false))
5090     return true;
5091 
5092   // -foptimization-record-file alone enables it too.
5093   if (Args.hasFlag(options::OPT_foptimization_record_file_EQ,
5094                    options::OPT_fno_save_optimization_record, false))
5095     return true;
5096 
5097   // -foptimization-record-passes alone enables it too.
5098   if (Args.hasFlag(options::OPT_foptimization_record_passes_EQ,
5099                    options::OPT_fno_save_optimization_record, false))
5100     return true;
5101   return false;
5102 }
5103