1 //===--- Driver.cpp - Clang GCC Compatible Driver -------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "clang/Driver/Driver.h"
11 #include "InputInfo.h"
12 #include "ToolChains.h"
13 #include "clang/Basic/Version.h"
14 #include "clang/Config/config.h"
15 #include "clang/Driver/Action.h"
16 #include "clang/Driver/Compilation.h"
17 #include "clang/Driver/DriverDiagnostic.h"
18 #include "clang/Driver/Job.h"
19 #include "clang/Driver/Options.h"
20 #include "clang/Driver/SanitizerArgs.h"
21 #include "clang/Driver/Tool.h"
22 #include "clang/Driver/ToolChain.h"
23 #include "llvm/ADT/ArrayRef.h"
24 #include "llvm/ADT/STLExtras.h"
25 #include "llvm/ADT/StringExtras.h"
26 #include "llvm/ADT/StringSet.h"
27 #include "llvm/ADT/StringSwitch.h"
28 #include "llvm/Option/Arg.h"
29 #include "llvm/Option/ArgList.h"
30 #include "llvm/Option/OptSpecifier.h"
31 #include "llvm/Option/OptTable.h"
32 #include "llvm/Option/Option.h"
33 #include "llvm/Support/Debug.h"
34 #include "llvm/Support/ErrorHandling.h"
35 #include "llvm/Support/FileSystem.h"
36 #include "llvm/Support/Path.h"
37 #include "llvm/Support/PrettyStackTrace.h"
38 #include "llvm/Support/Process.h"
39 #include "llvm/Support/Program.h"
40 #include "llvm/Support/raw_ostream.h"
41 #include <map>
42 #include <memory>
43 
44 using namespace clang::driver;
45 using namespace clang;
46 using namespace llvm::opt;
47 
48 Driver::Driver(StringRef ClangExecutable, StringRef DefaultTargetTriple,
49                DiagnosticsEngine &Diags)
50     : Opts(createDriverOptTable()), Diags(Diags), Mode(GCCMode),
51       SaveTemps(SaveTempsNone), ClangExecutable(ClangExecutable),
52       SysRoot(DEFAULT_SYSROOT), UseStdLib(true),
53       DefaultTargetTriple(DefaultTargetTriple),
54       DriverTitle("clang LLVM compiler"), CCPrintOptionsFilename(nullptr),
55       CCPrintHeadersFilename(nullptr), CCLogDiagnosticsFilename(nullptr),
56       CCCPrintBindings(false), CCPrintHeaders(false), CCLogDiagnostics(false),
57       CCGenDiagnostics(false), CCCGenericGCCName(""), CheckInputsExist(true),
58       CCCUsePCH(true), SuppressMissingInputWarning(false) {
59 
60   Name = llvm::sys::path::filename(ClangExecutable);
61   Dir = llvm::sys::path::parent_path(ClangExecutable);
62 
63   // Compute the path to the resource directory.
64   StringRef ClangResourceDir(CLANG_RESOURCE_DIR);
65   SmallString<128> P(Dir);
66   if (ClangResourceDir != "") {
67     llvm::sys::path::append(P, ClangResourceDir);
68   } else {
69     StringRef ClangLibdirSuffix(CLANG_LIBDIR_SUFFIX);
70     llvm::sys::path::append(P, "..", Twine("lib") + ClangLibdirSuffix, "clang",
71                             CLANG_VERSION_STRING);
72   }
73   ResourceDir = P.str();
74 }
75 
76 Driver::~Driver() {
77   delete Opts;
78 
79   llvm::DeleteContainerSeconds(ToolChains);
80 }
81 
82 void Driver::ParseDriverMode(ArrayRef<const char *> Args) {
83   const std::string OptName =
84       getOpts().getOption(options::OPT_driver_mode).getPrefixedName();
85 
86   for (const char *ArgPtr : Args) {
87     // Ingore nullptrs, they are response file's EOL markers
88     if (ArgPtr == nullptr)
89       continue;
90     const StringRef Arg = ArgPtr;
91     if (!Arg.startswith(OptName))
92       continue;
93 
94     const StringRef Value = Arg.drop_front(OptName.size());
95     const unsigned M = llvm::StringSwitch<unsigned>(Value)
96                            .Case("gcc", GCCMode)
97                            .Case("g++", GXXMode)
98                            .Case("cpp", CPPMode)
99                            .Case("cl", CLMode)
100                            .Default(~0U);
101 
102     if (M != ~0U)
103       Mode = static_cast<DriverMode>(M);
104     else
105       Diag(diag::err_drv_unsupported_option_argument) << OptName << Value;
106   }
107 }
108 
109 InputArgList Driver::ParseArgStrings(ArrayRef<const char *> ArgStrings) {
110   llvm::PrettyStackTraceString CrashInfo("Command line argument parsing");
111 
112   unsigned IncludedFlagsBitmask;
113   unsigned ExcludedFlagsBitmask;
114   std::tie(IncludedFlagsBitmask, ExcludedFlagsBitmask) =
115       getIncludeExcludeOptionFlagMasks();
116 
117   unsigned MissingArgIndex, MissingArgCount;
118   InputArgList Args =
119       getOpts().ParseArgs(ArgStrings, MissingArgIndex, MissingArgCount,
120                           IncludedFlagsBitmask, ExcludedFlagsBitmask);
121 
122   // Check for missing argument error.
123   if (MissingArgCount)
124     Diag(clang::diag::err_drv_missing_argument)
125         << Args.getArgString(MissingArgIndex) << MissingArgCount;
126 
127   // Check for unsupported options.
128   for (const Arg *A : Args) {
129     if (A->getOption().hasFlag(options::Unsupported)) {
130       Diag(clang::diag::err_drv_unsupported_opt) << A->getAsString(Args);
131       continue;
132     }
133 
134     // Warn about -mcpu= without an argument.
135     if (A->getOption().matches(options::OPT_mcpu_EQ) && A->containsValue("")) {
136       Diag(clang::diag::warn_drv_empty_joined_argument) << A->getAsString(Args);
137     }
138   }
139 
140   for (const Arg *A : Args.filtered(options::OPT_UNKNOWN))
141     Diags.Report(diag::err_drv_unknown_argument) << A->getAsString(Args);
142 
143   return Args;
144 }
145 
146 // Determine which compilation mode we are in. We look for options which
147 // affect the phase, starting with the earliest phases, and record which
148 // option we used to determine the final phase.
149 phases::ID Driver::getFinalPhase(const DerivedArgList &DAL,
150                                  Arg **FinalPhaseArg) const {
151   Arg *PhaseArg = nullptr;
152   phases::ID FinalPhase;
153 
154   // -{E,EP,P,M,MM} only run the preprocessor.
155   if (CCCIsCPP() || (PhaseArg = DAL.getLastArg(options::OPT_E)) ||
156       (PhaseArg = DAL.getLastArg(options::OPT__SLASH_EP)) ||
157       (PhaseArg = DAL.getLastArg(options::OPT_M, options::OPT_MM)) ||
158       (PhaseArg = DAL.getLastArg(options::OPT__SLASH_P))) {
159     FinalPhase = phases::Preprocess;
160 
161     // -{fsyntax-only,-analyze,emit-ast} only run up to the compiler.
162   } else if ((PhaseArg = DAL.getLastArg(options::OPT_fsyntax_only)) ||
163              (PhaseArg = DAL.getLastArg(options::OPT_module_file_info)) ||
164              (PhaseArg = DAL.getLastArg(options::OPT_verify_pch)) ||
165              (PhaseArg = DAL.getLastArg(options::OPT_rewrite_objc)) ||
166              (PhaseArg = DAL.getLastArg(options::OPT_rewrite_legacy_objc)) ||
167              (PhaseArg = DAL.getLastArg(options::OPT__migrate)) ||
168              (PhaseArg = DAL.getLastArg(options::OPT__analyze,
169                                         options::OPT__analyze_auto)) ||
170              (PhaseArg = DAL.getLastArg(options::OPT_emit_ast))) {
171     FinalPhase = phases::Compile;
172 
173     // -S only runs up to the backend.
174   } else if ((PhaseArg = DAL.getLastArg(options::OPT_S))) {
175     FinalPhase = phases::Backend;
176 
177     // -c compilation only runs up to the assembler.
178   } else if ((PhaseArg = DAL.getLastArg(options::OPT_c))) {
179     FinalPhase = phases::Assemble;
180 
181     // Otherwise do everything.
182   } else
183     FinalPhase = phases::Link;
184 
185   if (FinalPhaseArg)
186     *FinalPhaseArg = PhaseArg;
187 
188   return FinalPhase;
189 }
190 
191 static Arg *MakeInputArg(DerivedArgList &Args, OptTable *Opts,
192                          StringRef Value) {
193   Arg *A = new Arg(Opts->getOption(options::OPT_INPUT), Value,
194                    Args.getBaseArgs().MakeIndex(Value), Value.data());
195   Args.AddSynthesizedArg(A);
196   A->claim();
197   return A;
198 }
199 
200 DerivedArgList *Driver::TranslateInputArgs(const InputArgList &Args) const {
201   DerivedArgList *DAL = new DerivedArgList(Args);
202 
203   bool HasNostdlib = Args.hasArg(options::OPT_nostdlib);
204   for (Arg *A : Args) {
205     // Unfortunately, we have to parse some forwarding options (-Xassembler,
206     // -Xlinker, -Xpreprocessor) because we either integrate their functionality
207     // (assembler and preprocessor), or bypass a previous driver ('collect2').
208 
209     // Rewrite linker options, to replace --no-demangle with a custom internal
210     // option.
211     if ((A->getOption().matches(options::OPT_Wl_COMMA) ||
212          A->getOption().matches(options::OPT_Xlinker)) &&
213         A->containsValue("--no-demangle")) {
214       // Add the rewritten no-demangle argument.
215       DAL->AddFlagArg(A, Opts->getOption(options::OPT_Z_Xlinker__no_demangle));
216 
217       // Add the remaining values as Xlinker arguments.
218       for (StringRef Val : A->getValues())
219         if (Val != "--no-demangle")
220           DAL->AddSeparateArg(A, Opts->getOption(options::OPT_Xlinker), Val);
221 
222       continue;
223     }
224 
225     // Rewrite preprocessor options, to replace -Wp,-MD,FOO which is used by
226     // some build systems. We don't try to be complete here because we don't
227     // care to encourage this usage model.
228     if (A->getOption().matches(options::OPT_Wp_COMMA) &&
229         (A->getValue(0) == StringRef("-MD") ||
230          A->getValue(0) == StringRef("-MMD"))) {
231       // Rewrite to -MD/-MMD along with -MF.
232       if (A->getValue(0) == StringRef("-MD"))
233         DAL->AddFlagArg(A, Opts->getOption(options::OPT_MD));
234       else
235         DAL->AddFlagArg(A, Opts->getOption(options::OPT_MMD));
236       if (A->getNumValues() == 2)
237         DAL->AddSeparateArg(A, Opts->getOption(options::OPT_MF),
238                             A->getValue(1));
239       continue;
240     }
241 
242     // Rewrite reserved library names.
243     if (A->getOption().matches(options::OPT_l)) {
244       StringRef Value = A->getValue();
245 
246       // Rewrite unless -nostdlib is present.
247       if (!HasNostdlib && Value == "stdc++") {
248         DAL->AddFlagArg(A, Opts->getOption(options::OPT_Z_reserved_lib_stdcxx));
249         continue;
250       }
251 
252       // Rewrite unconditionally.
253       if (Value == "cc_kext") {
254         DAL->AddFlagArg(A, Opts->getOption(options::OPT_Z_reserved_lib_cckext));
255         continue;
256       }
257     }
258 
259     // Pick up inputs via the -- option.
260     if (A->getOption().matches(options::OPT__DASH_DASH)) {
261       A->claim();
262       for (StringRef Val : A->getValues())
263         DAL->append(MakeInputArg(*DAL, Opts, Val));
264       continue;
265     }
266 
267     DAL->append(A);
268   }
269 
270 // Add a default value of -mlinker-version=, if one was given and the user
271 // didn't specify one.
272 #if defined(HOST_LINK_VERSION)
273   if (!Args.hasArg(options::OPT_mlinker_version_EQ) &&
274       strlen(HOST_LINK_VERSION) > 0) {
275     DAL->AddJoinedArg(0, Opts->getOption(options::OPT_mlinker_version_EQ),
276                       HOST_LINK_VERSION);
277     DAL->getLastArg(options::OPT_mlinker_version_EQ)->claim();
278   }
279 #endif
280 
281   return DAL;
282 }
283 
284 /// \brief Compute target triple from args.
285 ///
286 /// This routine provides the logic to compute a target triple from various
287 /// args passed to the driver and the default triple string.
288 static llvm::Triple computeTargetTriple(StringRef DefaultTargetTriple,
289                                         const ArgList &Args,
290                                         StringRef DarwinArchName = "") {
291   // FIXME: Already done in Compilation *Driver::BuildCompilation
292   if (const Arg *A = Args.getLastArg(options::OPT_target))
293     DefaultTargetTriple = A->getValue();
294 
295   llvm::Triple Target(llvm::Triple::normalize(DefaultTargetTriple));
296 
297   // Handle Apple-specific options available here.
298   if (Target.isOSBinFormatMachO()) {
299     // If an explict Darwin arch name is given, that trumps all.
300     if (!DarwinArchName.empty()) {
301       tools::darwin::setTripleTypeForMachOArchName(Target, DarwinArchName);
302       return Target;
303     }
304 
305     // Handle the Darwin '-arch' flag.
306     if (Arg *A = Args.getLastArg(options::OPT_arch)) {
307       StringRef ArchName = A->getValue();
308       tools::darwin::setTripleTypeForMachOArchName(Target, ArchName);
309     }
310   }
311 
312   // Handle pseudo-target flags '-mlittle-endian'/'-EL' and
313   // '-mbig-endian'/'-EB'.
314   if (Arg *A = Args.getLastArg(options::OPT_mlittle_endian,
315                                options::OPT_mbig_endian)) {
316     if (A->getOption().matches(options::OPT_mlittle_endian)) {
317       llvm::Triple LE = Target.getLittleEndianArchVariant();
318       if (LE.getArch() != llvm::Triple::UnknownArch)
319         Target = std::move(LE);
320     } else {
321       llvm::Triple BE = Target.getBigEndianArchVariant();
322       if (BE.getArch() != llvm::Triple::UnknownArch)
323         Target = std::move(BE);
324     }
325   }
326 
327   // Skip further flag support on OSes which don't support '-m32' or '-m64'.
328   if (Target.getArch() == llvm::Triple::tce ||
329       Target.getOS() == llvm::Triple::Minix)
330     return Target;
331 
332   // Handle pseudo-target flags '-m64', '-mx32', '-m32' and '-m16'.
333   if (Arg *A = Args.getLastArg(options::OPT_m64, options::OPT_mx32,
334                                options::OPT_m32, options::OPT_m16)) {
335     llvm::Triple::ArchType AT = llvm::Triple::UnknownArch;
336 
337     if (A->getOption().matches(options::OPT_m64)) {
338       AT = Target.get64BitArchVariant().getArch();
339       if (Target.getEnvironment() == llvm::Triple::GNUX32)
340         Target.setEnvironment(llvm::Triple::GNU);
341     } else if (A->getOption().matches(options::OPT_mx32) &&
342                Target.get64BitArchVariant().getArch() == llvm::Triple::x86_64) {
343       AT = llvm::Triple::x86_64;
344       Target.setEnvironment(llvm::Triple::GNUX32);
345     } else if (A->getOption().matches(options::OPT_m32)) {
346       AT = Target.get32BitArchVariant().getArch();
347       if (Target.getEnvironment() == llvm::Triple::GNUX32)
348         Target.setEnvironment(llvm::Triple::GNU);
349     } else if (A->getOption().matches(options::OPT_m16) &&
350                Target.get32BitArchVariant().getArch() == llvm::Triple::x86) {
351       AT = llvm::Triple::x86;
352       Target.setEnvironment(llvm::Triple::CODE16);
353     }
354 
355     if (AT != llvm::Triple::UnknownArch && AT != Target.getArch())
356       Target.setArch(AT);
357   }
358 
359   return Target;
360 }
361 
362 Compilation *Driver::BuildCompilation(ArrayRef<const char *> ArgList) {
363   llvm::PrettyStackTraceString CrashInfo("Compilation construction");
364 
365   // FIXME: Handle environment options which affect driver behavior, somewhere
366   // (client?). GCC_EXEC_PREFIX, LPATH, CC_PRINT_OPTIONS.
367 
368   if (char *env = ::getenv("COMPILER_PATH")) {
369     StringRef CompilerPath = env;
370     while (!CompilerPath.empty()) {
371       std::pair<StringRef, StringRef> Split =
372           CompilerPath.split(llvm::sys::EnvPathSeparator);
373       PrefixDirs.push_back(Split.first);
374       CompilerPath = Split.second;
375     }
376   }
377 
378   // We look for the driver mode option early, because the mode can affect
379   // how other options are parsed.
380   ParseDriverMode(ArgList.slice(1));
381 
382   // FIXME: What are we going to do with -V and -b?
383 
384   // FIXME: This stuff needs to go into the Compilation, not the driver.
385   bool CCCPrintPhases;
386 
387   InputArgList Args = ParseArgStrings(ArgList.slice(1));
388 
389   // Silence driver warnings if requested
390   Diags.setIgnoreAllWarnings(Args.hasArg(options::OPT_w));
391 
392   // -no-canonical-prefixes is used very early in main.
393   Args.ClaimAllArgs(options::OPT_no_canonical_prefixes);
394 
395   // Ignore -pipe.
396   Args.ClaimAllArgs(options::OPT_pipe);
397 
398   // Extract -ccc args.
399   //
400   // FIXME: We need to figure out where this behavior should live. Most of it
401   // should be outside in the client; the parts that aren't should have proper
402   // options, either by introducing new ones or by overloading gcc ones like -V
403   // or -b.
404   CCCPrintPhases = Args.hasArg(options::OPT_ccc_print_phases);
405   CCCPrintBindings = Args.hasArg(options::OPT_ccc_print_bindings);
406   if (const Arg *A = Args.getLastArg(options::OPT_ccc_gcc_name))
407     CCCGenericGCCName = A->getValue();
408   CCCUsePCH =
409       Args.hasFlag(options::OPT_ccc_pch_is_pch, options::OPT_ccc_pch_is_pth);
410   // FIXME: DefaultTargetTriple is used by the target-prefixed calls to as/ld
411   // and getToolChain is const.
412   if (IsCLMode()) {
413     // clang-cl targets MSVC-style Win32.
414     llvm::Triple T(DefaultTargetTriple);
415     T.setOS(llvm::Triple::Win32);
416     T.setVendor(llvm::Triple::PC);
417     T.setEnvironment(llvm::Triple::MSVC);
418     DefaultTargetTriple = T.str();
419   }
420   if (const Arg *A = Args.getLastArg(options::OPT_target))
421     DefaultTargetTriple = A->getValue();
422   if (const Arg *A = Args.getLastArg(options::OPT_ccc_install_dir))
423     Dir = InstalledDir = A->getValue();
424   for (const Arg *A : Args.filtered(options::OPT_B)) {
425     A->claim();
426     PrefixDirs.push_back(A->getValue(0));
427   }
428   if (const Arg *A = Args.getLastArg(options::OPT__sysroot_EQ))
429     SysRoot = A->getValue();
430   if (const Arg *A = Args.getLastArg(options::OPT__dyld_prefix_EQ))
431     DyldPrefix = A->getValue();
432   if (Args.hasArg(options::OPT_nostdlib))
433     UseStdLib = false;
434 
435   if (const Arg *A = Args.getLastArg(options::OPT_resource_dir))
436     ResourceDir = A->getValue();
437 
438   if (const Arg *A = Args.getLastArg(options::OPT_save_temps_EQ)) {
439     SaveTemps = llvm::StringSwitch<SaveTempsMode>(A->getValue())
440                     .Case("cwd", SaveTempsCwd)
441                     .Case("obj", SaveTempsObj)
442                     .Default(SaveTempsCwd);
443   }
444 
445   std::unique_ptr<llvm::opt::InputArgList> UArgs =
446       llvm::make_unique<InputArgList>(std::move(Args));
447 
448   // Perform the default argument translations.
449   DerivedArgList *TranslatedArgs = TranslateInputArgs(*UArgs);
450 
451   // Owned by the host.
452   const ToolChain &TC =
453       getToolChain(*UArgs, computeTargetTriple(DefaultTargetTriple, *UArgs));
454 
455   // The compilation takes ownership of Args.
456   Compilation *C = new Compilation(*this, TC, UArgs.release(), TranslatedArgs);
457 
458   if (!HandleImmediateArgs(*C))
459     return C;
460 
461   // Construct the list of inputs.
462   InputList Inputs;
463   BuildInputs(C->getDefaultToolChain(), *TranslatedArgs, Inputs);
464 
465   // Construct the list of abstract actions to perform for this compilation. On
466   // MachO targets this uses the driver-driver and universal actions.
467   if (TC.getTriple().isOSBinFormatMachO())
468     BuildUniversalActions(C->getDefaultToolChain(), C->getArgs(), Inputs,
469                           C->getActions());
470   else
471     BuildActions(C->getDefaultToolChain(), C->getArgs(), Inputs,
472                  C->getActions());
473 
474   if (CCCPrintPhases) {
475     PrintActions(*C);
476     return C;
477   }
478 
479   BuildJobs(*C);
480 
481   return C;
482 }
483 
484 static void printArgList(raw_ostream &OS, const llvm::opt::ArgList &Args) {
485   llvm::opt::ArgStringList ASL;
486   for (const auto *A : Args)
487     A->render(Args, ASL);
488 
489   for (auto I = ASL.begin(), E = ASL.end(); I != E; ++I) {
490     if (I != ASL.begin())
491       OS << ' ';
492     Command::printArg(OS, *I, true);
493   }
494   OS << '\n';
495 }
496 
497 // When clang crashes, produce diagnostic information including the fully
498 // preprocessed source file(s).  Request that the developer attach the
499 // diagnostic information to a bug report.
500 void Driver::generateCompilationDiagnostics(Compilation &C,
501                                             const Command &FailingCommand) {
502   if (C.getArgs().hasArg(options::OPT_fno_crash_diagnostics))
503     return;
504 
505   // Don't try to generate diagnostics for link or dsymutil jobs.
506   if (FailingCommand.getCreator().isLinkJob() ||
507       FailingCommand.getCreator().isDsymutilJob())
508     return;
509 
510   // Print the version of the compiler.
511   PrintVersion(C, llvm::errs());
512 
513   Diag(clang::diag::note_drv_command_failed_diag_msg)
514       << "PLEASE submit a bug report to " BUG_REPORT_URL " and include the "
515          "crash backtrace, preprocessed source, and associated run script.";
516 
517   // Suppress driver output and emit preprocessor output to temp file.
518   Mode = CPPMode;
519   CCGenDiagnostics = true;
520 
521   // Save the original job command(s).
522   Command Cmd = FailingCommand;
523 
524   // Keep track of whether we produce any errors while trying to produce
525   // preprocessed sources.
526   DiagnosticErrorTrap Trap(Diags);
527 
528   // Suppress tool output.
529   C.initCompilationForDiagnostics();
530 
531   // Construct the list of inputs.
532   InputList Inputs;
533   BuildInputs(C.getDefaultToolChain(), C.getArgs(), Inputs);
534 
535   for (InputList::iterator it = Inputs.begin(), ie = Inputs.end(); it != ie;) {
536     bool IgnoreInput = false;
537 
538     // Ignore input from stdin or any inputs that cannot be preprocessed.
539     // Check type first as not all linker inputs have a value.
540     if (types::getPreprocessedType(it->first) == types::TY_INVALID) {
541       IgnoreInput = true;
542     } else if (!strcmp(it->second->getValue(), "-")) {
543       Diag(clang::diag::note_drv_command_failed_diag_msg)
544           << "Error generating preprocessed source(s) - "
545              "ignoring input from stdin.";
546       IgnoreInput = true;
547     }
548 
549     if (IgnoreInput) {
550       it = Inputs.erase(it);
551       ie = Inputs.end();
552     } else {
553       ++it;
554     }
555   }
556 
557   if (Inputs.empty()) {
558     Diag(clang::diag::note_drv_command_failed_diag_msg)
559         << "Error generating preprocessed source(s) - "
560            "no preprocessable inputs.";
561     return;
562   }
563 
564   // Don't attempt to generate preprocessed files if multiple -arch options are
565   // used, unless they're all duplicates.
566   llvm::StringSet<> ArchNames;
567   for (const Arg *A : C.getArgs()) {
568     if (A->getOption().matches(options::OPT_arch)) {
569       StringRef ArchName = A->getValue();
570       ArchNames.insert(ArchName);
571     }
572   }
573   if (ArchNames.size() > 1) {
574     Diag(clang::diag::note_drv_command_failed_diag_msg)
575         << "Error generating preprocessed source(s) - cannot generate "
576            "preprocessed source with multiple -arch options.";
577     return;
578   }
579 
580   // Construct the list of abstract actions to perform for this compilation. On
581   // Darwin OSes this uses the driver-driver and builds universal actions.
582   const ToolChain &TC = C.getDefaultToolChain();
583   if (TC.getTriple().isOSBinFormatMachO())
584     BuildUniversalActions(TC, C.getArgs(), Inputs, C.getActions());
585   else
586     BuildActions(TC, C.getArgs(), Inputs, C.getActions());
587 
588   BuildJobs(C);
589 
590   // If there were errors building the compilation, quit now.
591   if (Trap.hasErrorOccurred()) {
592     Diag(clang::diag::note_drv_command_failed_diag_msg)
593         << "Error generating preprocessed source(s).";
594     return;
595   }
596 
597   // Generate preprocessed output.
598   SmallVector<std::pair<int, const Command *>, 4> FailingCommands;
599   C.ExecuteJobs(C.getJobs(), FailingCommands);
600 
601   // If any of the preprocessing commands failed, clean up and exit.
602   if (!FailingCommands.empty()) {
603     if (!isSaveTempsEnabled())
604       C.CleanupFileList(C.getTempFiles(), true);
605 
606     Diag(clang::diag::note_drv_command_failed_diag_msg)
607         << "Error generating preprocessed source(s).";
608     return;
609   }
610 
611   const ArgStringList &TempFiles = C.getTempFiles();
612   if (TempFiles.empty()) {
613     Diag(clang::diag::note_drv_command_failed_diag_msg)
614         << "Error generating preprocessed source(s).";
615     return;
616   }
617 
618   Diag(clang::diag::note_drv_command_failed_diag_msg)
619       << "\n********************\n\n"
620          "PLEASE ATTACH THE FOLLOWING FILES TO THE BUG REPORT:\n"
621          "Preprocessed source(s) and associated run script(s) are located at:";
622 
623   SmallString<128> VFS;
624   for (const char *TempFile : TempFiles) {
625     Diag(clang::diag::note_drv_command_failed_diag_msg) << TempFile;
626     if (StringRef(TempFile).endswith(".cache")) {
627       // In some cases (modules) we'll dump extra data to help with reproducing
628       // the crash into a directory next to the output.
629       VFS = llvm::sys::path::filename(TempFile);
630       llvm::sys::path::append(VFS, "vfs", "vfs.yaml");
631     }
632   }
633 
634   // Assume associated files are based off of the first temporary file.
635   CrashReportInfo CrashInfo(TempFiles[0], VFS);
636 
637   std::string Script = CrashInfo.Filename.rsplit('.').first.str() + ".sh";
638   std::error_code EC;
639   llvm::raw_fd_ostream ScriptOS(Script, EC, llvm::sys::fs::F_Excl);
640   if (EC) {
641     Diag(clang::diag::note_drv_command_failed_diag_msg)
642         << "Error generating run script: " + Script + " " + EC.message();
643   } else {
644     ScriptOS << "# Crash reproducer for " << getClangFullVersion() << "\n"
645              << "# Driver args: ";
646     printArgList(ScriptOS, C.getInputArgs());
647     ScriptOS << "# Original command: ";
648     Cmd.Print(ScriptOS, "\n", /*Quote=*/true);
649     Cmd.Print(ScriptOS, "\n", /*Quote=*/true, &CrashInfo);
650     Diag(clang::diag::note_drv_command_failed_diag_msg) << Script;
651   }
652 
653   for (const auto &A : C.getArgs().filtered(options::OPT_frewrite_map_file,
654                                             options::OPT_frewrite_map_file_EQ))
655     Diag(clang::diag::note_drv_command_failed_diag_msg) << A->getValue();
656 
657   Diag(clang::diag::note_drv_command_failed_diag_msg)
658       << "\n\n********************";
659 }
660 
661 void Driver::setUpResponseFiles(Compilation &C, Command &Cmd) {
662   // Since argumentsFitWithinSystemLimits() may underestimate system's capacity
663   // if the tool does not support response files, there is a chance/ that things
664   // will just work without a response file, so we silently just skip it.
665   if (Cmd.getCreator().getResponseFilesSupport() == Tool::RF_None ||
666       llvm::sys::argumentsFitWithinSystemLimits(Cmd.getArguments()))
667     return;
668 
669   std::string TmpName = GetTemporaryPath("response", "txt");
670   Cmd.setResponseFile(
671       C.addTempFile(C.getArgs().MakeArgString(TmpName.c_str())));
672 }
673 
674 int Driver::ExecuteCompilation(
675     Compilation &C,
676     SmallVectorImpl<std::pair<int, const Command *>> &FailingCommands) {
677   // Just print if -### was present.
678   if (C.getArgs().hasArg(options::OPT__HASH_HASH_HASH)) {
679     C.getJobs().Print(llvm::errs(), "\n", true);
680     return 0;
681   }
682 
683   // If there were errors building the compilation, quit now.
684   if (Diags.hasErrorOccurred())
685     return 1;
686 
687   // Set up response file names for each command, if necessary
688   for (auto &Job : C.getJobs())
689     setUpResponseFiles(C, Job);
690 
691   C.ExecuteJobs(C.getJobs(), FailingCommands);
692 
693   // Remove temp files.
694   C.CleanupFileList(C.getTempFiles());
695 
696   // If the command succeeded, we are done.
697   if (FailingCommands.empty())
698     return 0;
699 
700   // Otherwise, remove result files and print extra information about abnormal
701   // failures.
702   for (const auto &CmdPair : FailingCommands) {
703     int Res = CmdPair.first;
704     const Command *FailingCommand = CmdPair.second;
705 
706     // Remove result files if we're not saving temps.
707     if (!isSaveTempsEnabled()) {
708       const JobAction *JA = cast<JobAction>(&FailingCommand->getSource());
709       C.CleanupFileMap(C.getResultFiles(), JA, true);
710 
711       // Failure result files are valid unless we crashed.
712       if (Res < 0)
713         C.CleanupFileMap(C.getFailureResultFiles(), JA, true);
714     }
715 
716     // Print extra information about abnormal failures, if possible.
717     //
718     // This is ad-hoc, but we don't want to be excessively noisy. If the result
719     // status was 1, assume the command failed normally. In particular, if it
720     // was the compiler then assume it gave a reasonable error code. Failures
721     // in other tools are less common, and they generally have worse
722     // diagnostics, so always print the diagnostic there.
723     const Tool &FailingTool = FailingCommand->getCreator();
724 
725     if (!FailingCommand->getCreator().hasGoodDiagnostics() || Res != 1) {
726       // FIXME: See FIXME above regarding result code interpretation.
727       if (Res < 0)
728         Diag(clang::diag::err_drv_command_signalled)
729             << FailingTool.getShortName();
730       else
731         Diag(clang::diag::err_drv_command_failed) << FailingTool.getShortName()
732                                                   << Res;
733     }
734   }
735   return 0;
736 }
737 
738 void Driver::PrintHelp(bool ShowHidden) const {
739   unsigned IncludedFlagsBitmask;
740   unsigned ExcludedFlagsBitmask;
741   std::tie(IncludedFlagsBitmask, ExcludedFlagsBitmask) =
742       getIncludeExcludeOptionFlagMasks();
743 
744   ExcludedFlagsBitmask |= options::NoDriverOption;
745   if (!ShowHidden)
746     ExcludedFlagsBitmask |= HelpHidden;
747 
748   getOpts().PrintHelp(llvm::outs(), Name.c_str(), DriverTitle.c_str(),
749                       IncludedFlagsBitmask, ExcludedFlagsBitmask);
750 }
751 
752 void Driver::PrintVersion(const Compilation &C, raw_ostream &OS) const {
753   // FIXME: The following handlers should use a callback mechanism, we don't
754   // know what the client would like to do.
755   OS << getClangFullVersion() << '\n';
756   const ToolChain &TC = C.getDefaultToolChain();
757   OS << "Target: " << TC.getTripleString() << '\n';
758 
759   // Print the threading model.
760   if (Arg *A = C.getArgs().getLastArg(options::OPT_mthread_model)) {
761     // Don't print if the ToolChain would have barfed on it already
762     if (TC.isThreadModelSupported(A->getValue()))
763       OS << "Thread model: " << A->getValue();
764   } else
765     OS << "Thread model: " << TC.getThreadModel();
766   OS << '\n';
767 
768   // Print out the install directory.
769   OS << "InstalledDir: " << InstalledDir << '\n';
770 }
771 
772 /// PrintDiagnosticCategories - Implement the --print-diagnostic-categories
773 /// option.
774 static void PrintDiagnosticCategories(raw_ostream &OS) {
775   // Skip the empty category.
776   for (unsigned i = 1, max = DiagnosticIDs::getNumberOfCategories(); i != max;
777        ++i)
778     OS << i << ',' << DiagnosticIDs::getCategoryNameFromID(i) << '\n';
779 }
780 
781 bool Driver::HandleImmediateArgs(const Compilation &C) {
782   // The order these options are handled in gcc is all over the place, but we
783   // don't expect inconsistencies w.r.t. that to matter in practice.
784 
785   if (C.getArgs().hasArg(options::OPT_dumpmachine)) {
786     llvm::outs() << C.getDefaultToolChain().getTripleString() << '\n';
787     return false;
788   }
789 
790   if (C.getArgs().hasArg(options::OPT_dumpversion)) {
791     // Since -dumpversion is only implemented for pedantic GCC compatibility, we
792     // return an answer which matches our definition of __VERSION__.
793     //
794     // If we want to return a more correct answer some day, then we should
795     // introduce a non-pedantically GCC compatible mode to Clang in which we
796     // provide sensible definitions for -dumpversion, __VERSION__, etc.
797     llvm::outs() << "4.2.1\n";
798     return false;
799   }
800 
801   if (C.getArgs().hasArg(options::OPT__print_diagnostic_categories)) {
802     PrintDiagnosticCategories(llvm::outs());
803     return false;
804   }
805 
806   if (C.getArgs().hasArg(options::OPT_help) ||
807       C.getArgs().hasArg(options::OPT__help_hidden)) {
808     PrintHelp(C.getArgs().hasArg(options::OPT__help_hidden));
809     return false;
810   }
811 
812   if (C.getArgs().hasArg(options::OPT__version)) {
813     // Follow gcc behavior and use stdout for --version and stderr for -v.
814     PrintVersion(C, llvm::outs());
815     return false;
816   }
817 
818   if (C.getArgs().hasArg(options::OPT_v) ||
819       C.getArgs().hasArg(options::OPT__HASH_HASH_HASH)) {
820     PrintVersion(C, llvm::errs());
821     SuppressMissingInputWarning = true;
822   }
823 
824   const ToolChain &TC = C.getDefaultToolChain();
825 
826   if (C.getArgs().hasArg(options::OPT_v))
827     TC.printVerboseInfo(llvm::errs());
828 
829   if (C.getArgs().hasArg(options::OPT_print_search_dirs)) {
830     llvm::outs() << "programs: =";
831     bool separator = false;
832     for (const std::string &Path : TC.getProgramPaths()) {
833       if (separator)
834         llvm::outs() << ':';
835       llvm::outs() << Path;
836       separator = true;
837     }
838     llvm::outs() << "\n";
839     llvm::outs() << "libraries: =" << ResourceDir;
840 
841     StringRef sysroot = C.getSysRoot();
842 
843     for (const std::string &Path : TC.getFilePaths()) {
844       // Always print a separator. ResourceDir was the first item shown.
845       llvm::outs() << ':';
846       // Interpretation of leading '=' is needed only for NetBSD.
847       if (Path[0] == '=')
848         llvm::outs() << sysroot << Path.substr(1);
849       else
850         llvm::outs() << Path;
851     }
852     llvm::outs() << "\n";
853     return false;
854   }
855 
856   // FIXME: The following handlers should use a callback mechanism, we don't
857   // know what the client would like to do.
858   if (Arg *A = C.getArgs().getLastArg(options::OPT_print_file_name_EQ)) {
859     llvm::outs() << GetFilePath(A->getValue(), TC) << "\n";
860     return false;
861   }
862 
863   if (Arg *A = C.getArgs().getLastArg(options::OPT_print_prog_name_EQ)) {
864     llvm::outs() << GetProgramPath(A->getValue(), TC) << "\n";
865     return false;
866   }
867 
868   if (C.getArgs().hasArg(options::OPT_print_libgcc_file_name)) {
869     llvm::outs() << GetFilePath("libgcc.a", TC) << "\n";
870     return false;
871   }
872 
873   if (C.getArgs().hasArg(options::OPT_print_multi_lib)) {
874     for (const Multilib &Multilib : TC.getMultilibs())
875       llvm::outs() << Multilib << "\n";
876     return false;
877   }
878 
879   if (C.getArgs().hasArg(options::OPT_print_multi_directory)) {
880     for (const Multilib &Multilib : TC.getMultilibs()) {
881       if (Multilib.gccSuffix().empty())
882         llvm::outs() << ".\n";
883       else {
884         StringRef Suffix(Multilib.gccSuffix());
885         assert(Suffix.front() == '/');
886         llvm::outs() << Suffix.substr(1) << "\n";
887       }
888     }
889     return false;
890   }
891   return true;
892 }
893 
894 // Display an action graph human-readably.  Action A is the "sink" node
895 // and latest-occuring action. Traversal is in pre-order, visiting the
896 // inputs to each action before printing the action itself.
897 static unsigned PrintActions1(const Compilation &C, Action *A,
898                               std::map<Action *, unsigned> &Ids) {
899   if (Ids.count(A)) // A was already visited.
900     return Ids[A];
901 
902   std::string str;
903   llvm::raw_string_ostream os(str);
904 
905   os << Action::getClassName(A->getKind()) << ", ";
906   if (InputAction *IA = dyn_cast<InputAction>(A)) {
907     os << "\"" << IA->getInputArg().getValue() << "\"";
908   } else if (BindArchAction *BIA = dyn_cast<BindArchAction>(A)) {
909     os << '"' << BIA->getArchName() << '"' << ", {"
910        << PrintActions1(C, *BIA->begin(), Ids) << "}";
911   } else if (CudaDeviceAction *CDA = dyn_cast<CudaDeviceAction>(A)) {
912     os << '"' << CDA->getGpuArchName() << '"' << ", {"
913        << PrintActions1(C, *CDA->begin(), Ids) << "}";
914   } else {
915     ActionList *AL;
916     if (CudaHostAction *CHA = dyn_cast<CudaHostAction>(A)) {
917       os << "{" << PrintActions1(C, *CHA->begin(), Ids) << "}"
918          << ", gpu binaries ";
919       AL = &CHA->getDeviceActions();
920     } else
921       AL = &A->getInputs();
922 
923     if (AL->size()) {
924       const char *Prefix = "{";
925       for (Action *PreRequisite : *AL) {
926         os << Prefix << PrintActions1(C, PreRequisite, Ids);
927         Prefix = ", ";
928       }
929       os << "}";
930     } else
931       os << "{}";
932   }
933 
934   unsigned Id = Ids.size();
935   Ids[A] = Id;
936   llvm::errs() << Id << ": " << os.str() << ", "
937                << types::getTypeName(A->getType()) << "\n";
938 
939   return Id;
940 }
941 
942 // Print the action graphs in a compilation C.
943 // For example "clang -c file1.c file2.c" is composed of two subgraphs.
944 void Driver::PrintActions(const Compilation &C) const {
945   std::map<Action *, unsigned> Ids;
946   for (Action *A : C.getActions())
947     PrintActions1(C, A, Ids);
948 }
949 
950 /// \brief Check whether the given input tree contains any compilation or
951 /// assembly actions.
952 static bool ContainsCompileOrAssembleAction(const Action *A) {
953   if (isa<CompileJobAction>(A) || isa<BackendJobAction>(A) ||
954       isa<AssembleJobAction>(A))
955     return true;
956 
957   for (const Action *Input : *A)
958     if (ContainsCompileOrAssembleAction(Input))
959       return true;
960 
961   return false;
962 }
963 
964 void Driver::BuildUniversalActions(const ToolChain &TC, DerivedArgList &Args,
965                                    const InputList &BAInputs,
966                                    ActionList &Actions) const {
967   llvm::PrettyStackTraceString CrashInfo("Building universal build actions");
968   // Collect the list of architectures. Duplicates are allowed, but should only
969   // be handled once (in the order seen).
970   llvm::StringSet<> ArchNames;
971   SmallVector<const char *, 4> Archs;
972   for (Arg *A : Args) {
973     if (A->getOption().matches(options::OPT_arch)) {
974       // Validate the option here; we don't save the type here because its
975       // particular spelling may participate in other driver choices.
976       llvm::Triple::ArchType Arch =
977           tools::darwin::getArchTypeForMachOArchName(A->getValue());
978       if (Arch == llvm::Triple::UnknownArch) {
979         Diag(clang::diag::err_drv_invalid_arch_name) << A->getAsString(Args);
980         continue;
981       }
982 
983       A->claim();
984       if (ArchNames.insert(A->getValue()).second)
985         Archs.push_back(A->getValue());
986     }
987   }
988 
989   // When there is no explicit arch for this platform, make sure we still bind
990   // the architecture (to the default) so that -Xarch_ is handled correctly.
991   if (!Archs.size())
992     Archs.push_back(Args.MakeArgString(TC.getDefaultUniversalArchName()));
993 
994   ActionList SingleActions;
995   BuildActions(TC, Args, BAInputs, SingleActions);
996 
997   // Add in arch bindings for every top level action, as well as lipo and
998   // dsymutil steps if needed.
999   for (Action* Act : SingleActions) {
1000     // Make sure we can lipo this kind of output. If not (and it is an actual
1001     // output) then we disallow, since we can't create an output file with the
1002     // right name without overwriting it. We could remove this oddity by just
1003     // changing the output names to include the arch, which would also fix
1004     // -save-temps. Compatibility wins for now.
1005 
1006     if (Archs.size() > 1 && !types::canLipoType(Act->getType()))
1007       Diag(clang::diag::err_drv_invalid_output_with_multiple_archs)
1008           << types::getTypeName(Act->getType());
1009 
1010     ActionList Inputs;
1011     for (unsigned i = 0, e = Archs.size(); i != e; ++i) {
1012       Inputs.push_back(
1013           new BindArchAction(std::unique_ptr<Action>(Act), Archs[i]));
1014       if (i != 0)
1015         Inputs.back()->setOwnsInputs(false);
1016     }
1017 
1018     // Lipo if necessary, we do it this way because we need to set the arch flag
1019     // so that -Xarch_ gets overwritten.
1020     if (Inputs.size() == 1 || Act->getType() == types::TY_Nothing)
1021       Actions.append(Inputs.begin(), Inputs.end());
1022     else
1023       Actions.push_back(new LipoJobAction(Inputs, Act->getType()));
1024 
1025     // Handle debug info queries.
1026     Arg *A = Args.getLastArg(options::OPT_g_Group);
1027     if (A && !A->getOption().matches(options::OPT_g0) &&
1028         !A->getOption().matches(options::OPT_gstabs) &&
1029         ContainsCompileOrAssembleAction(Actions.back())) {
1030 
1031       // Add a 'dsymutil' step if necessary, when debug info is enabled and we
1032       // have a compile input. We need to run 'dsymutil' ourselves in such cases
1033       // because the debug info will refer to a temporary object file which
1034       // will be removed at the end of the compilation process.
1035       if (Act->getType() == types::TY_Image) {
1036         ActionList Inputs;
1037         Inputs.push_back(Actions.back());
1038         Actions.pop_back();
1039         Actions.push_back(new DsymutilJobAction(Inputs, types::TY_dSYM));
1040       }
1041 
1042       // Verify the debug info output.
1043       if (Args.hasArg(options::OPT_verify_debug_info)) {
1044         std::unique_ptr<Action> VerifyInput(Actions.back());
1045         Actions.pop_back();
1046         Actions.push_back(new VerifyDebugInfoJobAction(std::move(VerifyInput),
1047                                                        types::TY_Nothing));
1048       }
1049     }
1050   }
1051 }
1052 
1053 /// \brief Check that the file referenced by Value exists. If it doesn't,
1054 /// issue a diagnostic and return false.
1055 static bool DiagnoseInputExistence(const Driver &D, const DerivedArgList &Args,
1056                                    StringRef Value) {
1057   if (!D.getCheckInputsExist())
1058     return true;
1059 
1060   // stdin always exists.
1061   if (Value == "-")
1062     return true;
1063 
1064   SmallString<64> Path(Value);
1065   if (Arg *WorkDir = Args.getLastArg(options::OPT_working_directory)) {
1066     if (!llvm::sys::path::is_absolute(Path)) {
1067       SmallString<64> Directory(WorkDir->getValue());
1068       llvm::sys::path::append(Directory, Value);
1069       Path.assign(Directory);
1070     }
1071   }
1072 
1073   if (llvm::sys::fs::exists(Twine(Path)))
1074     return true;
1075 
1076   if (D.IsCLMode() && !llvm::sys::path::is_absolute(Twine(Path)) &&
1077       llvm::sys::Process::FindInEnvPath("LIB", Value))
1078     return true;
1079 
1080   D.Diag(clang::diag::err_drv_no_such_file) << Path;
1081   return false;
1082 }
1083 
1084 // Construct a the list of inputs and their types.
1085 void Driver::BuildInputs(const ToolChain &TC, DerivedArgList &Args,
1086                          InputList &Inputs) const {
1087   // Track the current user specified (-x) input. We also explicitly track the
1088   // argument used to set the type; we only want to claim the type when we
1089   // actually use it, so we warn about unused -x arguments.
1090   types::ID InputType = types::TY_Nothing;
1091   Arg *InputTypeArg = nullptr;
1092 
1093   // The last /TC or /TP option sets the input type to C or C++ globally.
1094   if (Arg *TCTP = Args.getLastArgNoClaim(options::OPT__SLASH_TC,
1095                                          options::OPT__SLASH_TP)) {
1096     InputTypeArg = TCTP;
1097     InputType = TCTP->getOption().matches(options::OPT__SLASH_TC)
1098                     ? types::TY_C
1099                     : types::TY_CXX;
1100 
1101     arg_iterator it =
1102         Args.filtered_begin(options::OPT__SLASH_TC, options::OPT__SLASH_TP);
1103     const arg_iterator ie = Args.filtered_end();
1104     Arg *Previous = *it++;
1105     bool ShowNote = false;
1106     while (it != ie) {
1107       Diag(clang::diag::warn_drv_overriding_flag_option)
1108           << Previous->getSpelling() << (*it)->getSpelling();
1109       Previous = *it++;
1110       ShowNote = true;
1111     }
1112     if (ShowNote)
1113       Diag(clang::diag::note_drv_t_option_is_global);
1114 
1115     // No driver mode exposes -x and /TC or /TP; we don't support mixing them.
1116     assert(!Args.hasArg(options::OPT_x) && "-x and /TC or /TP is not allowed");
1117   }
1118 
1119   for (Arg *A : Args) {
1120     if (A->getOption().getKind() == Option::InputClass) {
1121       const char *Value = A->getValue();
1122       types::ID Ty = types::TY_INVALID;
1123 
1124       // Infer the input type if necessary.
1125       if (InputType == types::TY_Nothing) {
1126         // If there was an explicit arg for this, claim it.
1127         if (InputTypeArg)
1128           InputTypeArg->claim();
1129 
1130         // stdin must be handled specially.
1131         if (memcmp(Value, "-", 2) == 0) {
1132           // If running with -E, treat as a C input (this changes the builtin
1133           // macros, for example). This may be overridden by -ObjC below.
1134           //
1135           // Otherwise emit an error but still use a valid type to avoid
1136           // spurious errors (e.g., no inputs).
1137           if (!Args.hasArgNoClaim(options::OPT_E) && !CCCIsCPP())
1138             Diag(IsCLMode() ? clang::diag::err_drv_unknown_stdin_type_clang_cl
1139                             : clang::diag::err_drv_unknown_stdin_type);
1140           Ty = types::TY_C;
1141         } else {
1142           // Otherwise lookup by extension.
1143           // Fallback is C if invoked as C preprocessor or Object otherwise.
1144           // We use a host hook here because Darwin at least has its own
1145           // idea of what .s is.
1146           if (const char *Ext = strrchr(Value, '.'))
1147             Ty = TC.LookupTypeForExtension(Ext + 1);
1148 
1149           if (Ty == types::TY_INVALID) {
1150             if (CCCIsCPP())
1151               Ty = types::TY_C;
1152             else
1153               Ty = types::TY_Object;
1154           }
1155 
1156           // If the driver is invoked as C++ compiler (like clang++ or c++) it
1157           // should autodetect some input files as C++ for g++ compatibility.
1158           if (CCCIsCXX()) {
1159             types::ID OldTy = Ty;
1160             Ty = types::lookupCXXTypeForCType(Ty);
1161 
1162             if (Ty != OldTy)
1163               Diag(clang::diag::warn_drv_treating_input_as_cxx)
1164                   << getTypeName(OldTy) << getTypeName(Ty);
1165           }
1166         }
1167 
1168         // -ObjC and -ObjC++ override the default language, but only for "source
1169         // files". We just treat everything that isn't a linker input as a
1170         // source file.
1171         //
1172         // FIXME: Clean this up if we move the phase sequence into the type.
1173         if (Ty != types::TY_Object) {
1174           if (Args.hasArg(options::OPT_ObjC))
1175             Ty = types::TY_ObjC;
1176           else if (Args.hasArg(options::OPT_ObjCXX))
1177             Ty = types::TY_ObjCXX;
1178         }
1179       } else {
1180         assert(InputTypeArg && "InputType set w/o InputTypeArg");
1181         if (!InputTypeArg->getOption().matches(options::OPT_x)) {
1182           // If emulating cl.exe, make sure that /TC and /TP don't affect input
1183           // object files.
1184           const char *Ext = strrchr(Value, '.');
1185           if (Ext && TC.LookupTypeForExtension(Ext + 1) == types::TY_Object)
1186             Ty = types::TY_Object;
1187         }
1188         if (Ty == types::TY_INVALID) {
1189           Ty = InputType;
1190           InputTypeArg->claim();
1191         }
1192       }
1193 
1194       if (DiagnoseInputExistence(*this, Args, Value))
1195         Inputs.push_back(std::make_pair(Ty, A));
1196 
1197     } else if (A->getOption().matches(options::OPT__SLASH_Tc)) {
1198       StringRef Value = A->getValue();
1199       if (DiagnoseInputExistence(*this, Args, Value)) {
1200         Arg *InputArg = MakeInputArg(Args, Opts, A->getValue());
1201         Inputs.push_back(std::make_pair(types::TY_C, InputArg));
1202       }
1203       A->claim();
1204     } else if (A->getOption().matches(options::OPT__SLASH_Tp)) {
1205       StringRef Value = A->getValue();
1206       if (DiagnoseInputExistence(*this, Args, Value)) {
1207         Arg *InputArg = MakeInputArg(Args, Opts, A->getValue());
1208         Inputs.push_back(std::make_pair(types::TY_CXX, InputArg));
1209       }
1210       A->claim();
1211     } else if (A->getOption().hasFlag(options::LinkerInput)) {
1212       // Just treat as object type, we could make a special type for this if
1213       // necessary.
1214       Inputs.push_back(std::make_pair(types::TY_Object, A));
1215 
1216     } else if (A->getOption().matches(options::OPT_x)) {
1217       InputTypeArg = A;
1218       InputType = types::lookupTypeForTypeSpecifier(A->getValue());
1219       A->claim();
1220 
1221       // Follow gcc behavior and treat as linker input for invalid -x
1222       // options. Its not clear why we shouldn't just revert to unknown; but
1223       // this isn't very important, we might as well be bug compatible.
1224       if (!InputType) {
1225         Diag(clang::diag::err_drv_unknown_language) << A->getValue();
1226         InputType = types::TY_Object;
1227       }
1228     }
1229   }
1230   if (CCCIsCPP() && Inputs.empty()) {
1231     // If called as standalone preprocessor, stdin is processed
1232     // if no other input is present.
1233     Arg *A = MakeInputArg(Args, Opts, "-");
1234     Inputs.push_back(std::make_pair(types::TY_C, A));
1235   }
1236 }
1237 
1238 // For each unique --cuda-gpu-arch= argument creates a TY_CUDA_DEVICE
1239 // input action and then wraps each in CudaDeviceAction paired with
1240 // appropriate GPU arch name. In case of partial (i.e preprocessing
1241 // only) or device-only compilation, each device action is added to /p
1242 // Actions and /p Current is released. Otherwise the function creates
1243 // and returns a new CudaHostAction which wraps /p Current and device
1244 // side actions.
1245 static std::unique_ptr<Action>
1246 buildCudaActions(const Driver &D, const ToolChain &TC, DerivedArgList &Args,
1247                  const Arg *InputArg, std::unique_ptr<Action> HostAction,
1248                  ActionList &Actions) {
1249   // Figure out which NVPTX triple to use for device-side compilation based on
1250   // whether host is 64-bit.
1251   const char *DeviceTriple = TC.getTriple().isArch64Bit()
1252                                  ? "nvptx64-nvidia-cuda"
1253                                  : "nvptx-nvidia-cuda";
1254   Arg *PartialCompilationArg = Args.getLastArg(options::OPT_cuda_host_only,
1255                                                options::OPT_cuda_device_only);
1256   // Host-only compilation case.
1257   if (PartialCompilationArg &&
1258       PartialCompilationArg->getOption().matches(options::OPT_cuda_host_only))
1259     return std::unique_ptr<Action>(
1260         new CudaHostAction(std::move(HostAction), {}, DeviceTriple));
1261 
1262   // Collect all cuda_gpu_arch parameters, removing duplicates.
1263   SmallVector<const char *, 4> GpuArchList;
1264   llvm::StringSet<> GpuArchNames;
1265   for (Arg *A : Args) {
1266     if (A->getOption().matches(options::OPT_cuda_gpu_arch_EQ)) {
1267       A->claim();
1268       if (GpuArchNames.insert(A->getValue()).second)
1269         GpuArchList.push_back(A->getValue());
1270     }
1271   }
1272 
1273   // Default to sm_20 which is the lowest common denominator for supported GPUs.
1274   // sm_20 code should work correctly, if suboptimally, on all newer GPUs.
1275   if (GpuArchList.empty())
1276     GpuArchList.push_back("sm_20");
1277 
1278   // Replicate inputs for each GPU architecture.
1279   Driver::InputList CudaDeviceInputs;
1280   for (unsigned I = 0, E = GpuArchList.size(); I != E; ++I)
1281     CudaDeviceInputs.push_back(std::make_pair(types::TY_CUDA_DEVICE, InputArg));
1282 
1283   // Build actions for all device inputs.
1284   ActionList CudaDeviceActions;
1285   D.BuildActions(TC, Args, CudaDeviceInputs, CudaDeviceActions);
1286   assert(GpuArchList.size() == CudaDeviceActions.size() &&
1287          "Failed to create actions for all devices");
1288 
1289   // Check whether any of device actions stopped before they could generate PTX.
1290   bool PartialCompilation = false;
1291   for (unsigned I = 0, E = GpuArchList.size(); I != E; ++I) {
1292     if (CudaDeviceActions[I]->getKind() != Action::BackendJobClass) {
1293       PartialCompilation = true;
1294       break;
1295     }
1296   }
1297 
1298   // Figure out what to do with device actions -- pass them as inputs to the
1299   // host action or run each of them independently.
1300   bool DeviceOnlyCompilation = PartialCompilationArg != nullptr;
1301   if (PartialCompilation || DeviceOnlyCompilation) {
1302     // In case of partial or device-only compilation results of device actions
1303     // are not consumed by the host action device actions have to be added to
1304     // top-level actions list with AtTopLevel=true and run independently.
1305 
1306     // -o is ambiguous if we have more than one top-level action.
1307     if (Args.hasArg(options::OPT_o) &&
1308         (!DeviceOnlyCompilation || GpuArchList.size() > 1)) {
1309       D.Diag(clang::diag::err_drv_output_argument_with_multiple_files);
1310       return nullptr;
1311     }
1312 
1313     for (unsigned I = 0, E = GpuArchList.size(); I != E; ++I)
1314       Actions.push_back(new CudaDeviceAction(
1315           std::unique_ptr<Action>(CudaDeviceActions[I]), GpuArchList[I],
1316           DeviceTriple, /* AtTopLevel */ true));
1317     // Kill host action in case of device-only compilation.
1318     if (DeviceOnlyCompilation)
1319       HostAction.reset(nullptr);
1320     return HostAction;
1321   }
1322 
1323   // Outputs of device actions during complete CUDA compilation get created
1324   // with AtTopLevel=false and become inputs for the host action.
1325   ActionList DeviceActions;
1326   for (unsigned I = 0, E = GpuArchList.size(); I != E; ++I)
1327     DeviceActions.push_back(new CudaDeviceAction(
1328         std::unique_ptr<Action>(CudaDeviceActions[I]), GpuArchList[I],
1329         DeviceTriple, /* AtTopLevel */ false));
1330   // Return a new host action that incorporates original host action and all
1331   // device actions.
1332   return std::unique_ptr<Action>(
1333       new CudaHostAction(std::move(HostAction), DeviceActions, DeviceTriple));
1334 }
1335 
1336 void Driver::BuildActions(const ToolChain &TC, DerivedArgList &Args,
1337                           const InputList &Inputs, ActionList &Actions) const {
1338   llvm::PrettyStackTraceString CrashInfo("Building compilation actions");
1339 
1340   if (!SuppressMissingInputWarning && Inputs.empty()) {
1341     Diag(clang::diag::err_drv_no_input_files);
1342     return;
1343   }
1344 
1345   Arg *FinalPhaseArg;
1346   phases::ID FinalPhase = getFinalPhase(Args, &FinalPhaseArg);
1347 
1348   if (FinalPhase == phases::Link && Args.hasArg(options::OPT_emit_llvm)) {
1349     Diag(clang::diag::err_drv_emit_llvm_link);
1350   }
1351 
1352   // Reject -Z* at the top level, these options should never have been exposed
1353   // by gcc.
1354   if (Arg *A = Args.getLastArg(options::OPT_Z_Joined))
1355     Diag(clang::diag::err_drv_use_of_Z_option) << A->getAsString(Args);
1356 
1357   // Diagnose misuse of /Fo.
1358   if (Arg *A = Args.getLastArg(options::OPT__SLASH_Fo)) {
1359     StringRef V = A->getValue();
1360     if (Inputs.size() > 1 && !V.empty() &&
1361         !llvm::sys::path::is_separator(V.back())) {
1362       // Check whether /Fo tries to name an output file for multiple inputs.
1363       Diag(clang::diag::err_drv_out_file_argument_with_multiple_sources)
1364           << A->getSpelling() << V;
1365       Args.eraseArg(options::OPT__SLASH_Fo);
1366     }
1367   }
1368 
1369   // Diagnose misuse of /Fa.
1370   if (Arg *A = Args.getLastArg(options::OPT__SLASH_Fa)) {
1371     StringRef V = A->getValue();
1372     if (Inputs.size() > 1 && !V.empty() &&
1373         !llvm::sys::path::is_separator(V.back())) {
1374       // Check whether /Fa tries to name an asm file for multiple inputs.
1375       Diag(clang::diag::err_drv_out_file_argument_with_multiple_sources)
1376           << A->getSpelling() << V;
1377       Args.eraseArg(options::OPT__SLASH_Fa);
1378     }
1379   }
1380 
1381   // Diagnose misuse of /o.
1382   if (Arg *A = Args.getLastArg(options::OPT__SLASH_o)) {
1383     if (A->getValue()[0] == '\0') {
1384       // It has to have a value.
1385       Diag(clang::diag::err_drv_missing_argument) << A->getSpelling() << 1;
1386       Args.eraseArg(options::OPT__SLASH_o);
1387     }
1388   }
1389 
1390   // Construct the actions to perform.
1391   ActionList LinkerInputs;
1392 
1393   llvm::SmallVector<phases::ID, phases::MaxNumberOfPhases> PL;
1394   for (auto &I : Inputs) {
1395     types::ID InputType = I.first;
1396     const Arg *InputArg = I.second;
1397 
1398     PL.clear();
1399     types::getCompilationPhases(InputType, PL);
1400 
1401     // If the first step comes after the final phase we are doing as part of
1402     // this compilation, warn the user about it.
1403     phases::ID InitialPhase = PL[0];
1404     if (InitialPhase > FinalPhase) {
1405       // Claim here to avoid the more general unused warning.
1406       InputArg->claim();
1407 
1408       // Suppress all unused style warnings with -Qunused-arguments
1409       if (Args.hasArg(options::OPT_Qunused_arguments))
1410         continue;
1411 
1412       // Special case when final phase determined by binary name, rather than
1413       // by a command-line argument with a corresponding Arg.
1414       if (CCCIsCPP())
1415         Diag(clang::diag::warn_drv_input_file_unused_by_cpp)
1416             << InputArg->getAsString(Args) << getPhaseName(InitialPhase);
1417       // Special case '-E' warning on a previously preprocessed file to make
1418       // more sense.
1419       else if (InitialPhase == phases::Compile &&
1420                FinalPhase == phases::Preprocess &&
1421                getPreprocessedType(InputType) == types::TY_INVALID)
1422         Diag(clang::diag::warn_drv_preprocessed_input_file_unused)
1423             << InputArg->getAsString(Args) << !!FinalPhaseArg
1424             << (FinalPhaseArg ? FinalPhaseArg->getOption().getName() : "");
1425       else
1426         Diag(clang::diag::warn_drv_input_file_unused)
1427             << InputArg->getAsString(Args) << getPhaseName(InitialPhase)
1428             << !!FinalPhaseArg
1429             << (FinalPhaseArg ? FinalPhaseArg->getOption().getName() : "");
1430       continue;
1431     }
1432 
1433     phases::ID CudaInjectionPhase = FinalPhase;
1434     for (const auto &Phase : PL)
1435       if (Phase <= FinalPhase && Phase == phases::Compile) {
1436         CudaInjectionPhase = Phase;
1437         break;
1438       }
1439 
1440     // Build the pipeline for this file.
1441     std::unique_ptr<Action> Current(new InputAction(*InputArg, InputType));
1442     for (SmallVectorImpl<phases::ID>::iterator i = PL.begin(), e = PL.end();
1443          i != e; ++i) {
1444       phases::ID Phase = *i;
1445 
1446       // We are done if this step is past what the user requested.
1447       if (Phase > FinalPhase)
1448         break;
1449 
1450       // Queue linker inputs.
1451       if (Phase == phases::Link) {
1452         assert((i + 1) == e && "linking must be final compilation step.");
1453         LinkerInputs.push_back(Current.release());
1454         break;
1455       }
1456 
1457       // Some types skip the assembler phase (e.g., llvm-bc), but we can't
1458       // encode this in the steps because the intermediate type depends on
1459       // arguments. Just special case here.
1460       if (Phase == phases::Assemble && Current->getType() != types::TY_PP_Asm)
1461         continue;
1462 
1463       // Otherwise construct the appropriate action.
1464       Current = ConstructPhaseAction(TC, Args, Phase, std::move(Current));
1465 
1466       if (InputType == types::TY_CUDA && Phase == CudaInjectionPhase) {
1467         Current = buildCudaActions(*this, TC, Args, InputArg,
1468                                    std::move(Current), Actions);
1469         if (!Current)
1470           break;
1471       }
1472 
1473       if (Current->getType() == types::TY_Nothing)
1474         break;
1475     }
1476 
1477     // If we ended with something, add to the output list.
1478     if (Current)
1479       Actions.push_back(Current.release());
1480   }
1481 
1482   // Add a link action if necessary.
1483   if (!LinkerInputs.empty())
1484     Actions.push_back(new LinkJobAction(LinkerInputs, types::TY_Image));
1485 
1486   // If we are linking, claim any options which are obviously only used for
1487   // compilation.
1488   if (FinalPhase == phases::Link && PL.size() == 1) {
1489     Args.ClaimAllArgs(options::OPT_CompileOnly_Group);
1490     Args.ClaimAllArgs(options::OPT_cl_compile_Group);
1491   }
1492 
1493   // Claim ignored clang-cl options.
1494   Args.ClaimAllArgs(options::OPT_cl_ignored_Group);
1495 
1496   // Claim --cuda-host-only arg which may be passed to non-CUDA
1497   // compilations and should not trigger warnings there.
1498   Args.ClaimAllArgs(options::OPT_cuda_host_only);
1499 }
1500 
1501 std::unique_ptr<Action>
1502 Driver::ConstructPhaseAction(const ToolChain &TC, const ArgList &Args,
1503                              phases::ID Phase,
1504                              std::unique_ptr<Action> Input) const {
1505   llvm::PrettyStackTraceString CrashInfo("Constructing phase actions");
1506   // Build the appropriate action.
1507   switch (Phase) {
1508   case phases::Link:
1509     llvm_unreachable("link action invalid here.");
1510   case phases::Preprocess: {
1511     types::ID OutputTy;
1512     // -{M, MM} alter the output type.
1513     if (Args.hasArg(options::OPT_M, options::OPT_MM)) {
1514       OutputTy = types::TY_Dependencies;
1515     } else {
1516       OutputTy = Input->getType();
1517       if (!Args.hasFlag(options::OPT_frewrite_includes,
1518                         options::OPT_fno_rewrite_includes, false) &&
1519           !CCGenDiagnostics)
1520         OutputTy = types::getPreprocessedType(OutputTy);
1521       assert(OutputTy != types::TY_INVALID &&
1522              "Cannot preprocess this input type!");
1523     }
1524     return llvm::make_unique<PreprocessJobAction>(std::move(Input), OutputTy);
1525   }
1526   case phases::Precompile: {
1527     types::ID OutputTy = types::TY_PCH;
1528     if (Args.hasArg(options::OPT_fsyntax_only)) {
1529       // Syntax checks should not emit a PCH file
1530       OutputTy = types::TY_Nothing;
1531     }
1532     return llvm::make_unique<PrecompileJobAction>(std::move(Input), OutputTy);
1533   }
1534   case phases::Compile: {
1535     if (Args.hasArg(options::OPT_fsyntax_only))
1536       return llvm::make_unique<CompileJobAction>(std::move(Input),
1537                                                  types::TY_Nothing);
1538     if (Args.hasArg(options::OPT_rewrite_objc))
1539       return llvm::make_unique<CompileJobAction>(std::move(Input),
1540                                                  types::TY_RewrittenObjC);
1541     if (Args.hasArg(options::OPT_rewrite_legacy_objc))
1542       return llvm::make_unique<CompileJobAction>(std::move(Input),
1543                                                  types::TY_RewrittenLegacyObjC);
1544     if (Args.hasArg(options::OPT__analyze, options::OPT__analyze_auto))
1545       return llvm::make_unique<AnalyzeJobAction>(std::move(Input),
1546                                                  types::TY_Plist);
1547     if (Args.hasArg(options::OPT__migrate))
1548       return llvm::make_unique<MigrateJobAction>(std::move(Input),
1549                                                  types::TY_Remap);
1550     if (Args.hasArg(options::OPT_emit_ast))
1551       return llvm::make_unique<CompileJobAction>(std::move(Input),
1552                                                  types::TY_AST);
1553     if (Args.hasArg(options::OPT_module_file_info))
1554       return llvm::make_unique<CompileJobAction>(std::move(Input),
1555                                                  types::TY_ModuleFile);
1556     if (Args.hasArg(options::OPT_verify_pch))
1557       return llvm::make_unique<VerifyPCHJobAction>(std::move(Input),
1558                                                    types::TY_Nothing);
1559     return llvm::make_unique<CompileJobAction>(std::move(Input),
1560                                                types::TY_LLVM_BC);
1561   }
1562   case phases::Backend: {
1563     if (IsUsingLTO(Args)) {
1564       types::ID Output =
1565           Args.hasArg(options::OPT_S) ? types::TY_LTO_IR : types::TY_LTO_BC;
1566       return llvm::make_unique<BackendJobAction>(std::move(Input), Output);
1567     }
1568     if (Args.hasArg(options::OPT_emit_llvm)) {
1569       types::ID Output =
1570           Args.hasArg(options::OPT_S) ? types::TY_LLVM_IR : types::TY_LLVM_BC;
1571       return llvm::make_unique<BackendJobAction>(std::move(Input), Output);
1572     }
1573     return llvm::make_unique<BackendJobAction>(std::move(Input),
1574                                                types::TY_PP_Asm);
1575   }
1576   case phases::Assemble:
1577     return llvm::make_unique<AssembleJobAction>(std::move(Input),
1578                                                 types::TY_Object);
1579   }
1580 
1581   llvm_unreachable("invalid phase in ConstructPhaseAction");
1582 }
1583 
1584 bool Driver::IsUsingLTO(const ArgList &Args) const {
1585   return Args.hasFlag(options::OPT_flto, options::OPT_fno_lto, false);
1586 }
1587 
1588 void Driver::BuildJobs(Compilation &C) const {
1589   llvm::PrettyStackTraceString CrashInfo("Building compilation jobs");
1590 
1591   Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o);
1592 
1593   // It is an error to provide a -o option if we are making multiple output
1594   // files.
1595   if (FinalOutput) {
1596     unsigned NumOutputs = 0;
1597     for (const Action *A : C.getActions())
1598       if (A->getType() != types::TY_Nothing)
1599         ++NumOutputs;
1600 
1601     if (NumOutputs > 1) {
1602       Diag(clang::diag::err_drv_output_argument_with_multiple_files);
1603       FinalOutput = nullptr;
1604     }
1605   }
1606 
1607   // Collect the list of architectures.
1608   llvm::StringSet<> ArchNames;
1609   if (C.getDefaultToolChain().getTriple().isOSBinFormatMachO())
1610     for (const Arg *A : C.getArgs())
1611       if (A->getOption().matches(options::OPT_arch))
1612         ArchNames.insert(A->getValue());
1613 
1614   for (Action *A : C.getActions()) {
1615     // If we are linking an image for multiple archs then the linker wants
1616     // -arch_multiple and -final_output <final image name>. Unfortunately, this
1617     // doesn't fit in cleanly because we have to pass this information down.
1618     //
1619     // FIXME: This is a hack; find a cleaner way to integrate this into the
1620     // process.
1621     const char *LinkingOutput = nullptr;
1622     if (isa<LipoJobAction>(A)) {
1623       if (FinalOutput)
1624         LinkingOutput = FinalOutput->getValue();
1625       else
1626         LinkingOutput = getDefaultImageName();
1627     }
1628 
1629     InputInfo II;
1630     BuildJobsForAction(C, A, &C.getDefaultToolChain(),
1631                        /*BoundArch*/ nullptr,
1632                        /*AtTopLevel*/ true,
1633                        /*MultipleArchs*/ ArchNames.size() > 1,
1634                        /*LinkingOutput*/ LinkingOutput, II);
1635   }
1636 
1637   // If the user passed -Qunused-arguments or there were errors, don't warn
1638   // about any unused arguments.
1639   if (Diags.hasErrorOccurred() ||
1640       C.getArgs().hasArg(options::OPT_Qunused_arguments))
1641     return;
1642 
1643   // Claim -### here.
1644   (void)C.getArgs().hasArg(options::OPT__HASH_HASH_HASH);
1645 
1646   // Claim --driver-mode, it was handled earlier.
1647   (void)C.getArgs().hasArg(options::OPT_driver_mode);
1648 
1649   for (Arg *A : C.getArgs()) {
1650     // FIXME: It would be nice to be able to send the argument to the
1651     // DiagnosticsEngine, so that extra values, position, and so on could be
1652     // printed.
1653     if (!A->isClaimed()) {
1654       if (A->getOption().hasFlag(options::NoArgumentUnused))
1655         continue;
1656 
1657       // Suppress the warning automatically if this is just a flag, and it is an
1658       // instance of an argument we already claimed.
1659       const Option &Opt = A->getOption();
1660       if (Opt.getKind() == Option::FlagClass) {
1661         bool DuplicateClaimed = false;
1662 
1663         for (const Arg *AA : C.getArgs().filtered(&Opt)) {
1664           if (AA->isClaimed()) {
1665             DuplicateClaimed = true;
1666             break;
1667           }
1668         }
1669 
1670         if (DuplicateClaimed)
1671           continue;
1672       }
1673 
1674       Diag(clang::diag::warn_drv_unused_argument)
1675           << A->getAsString(C.getArgs());
1676     }
1677   }
1678 }
1679 
1680 // Returns a Tool for a given JobAction.  In case the action and its
1681 // predecessors can be combined, updates Inputs with the inputs of the
1682 // first combined action. If one of the collapsed actions is a
1683 // CudaHostAction, updates CollapsedCHA with the pointer to it so the
1684 // caller can deal with extra handling such action requires.
1685 static const Tool *selectToolForJob(Compilation &C, bool SaveTemps,
1686                                     const ToolChain *TC, const JobAction *JA,
1687                                     const ActionList *&Inputs,
1688                                     const CudaHostAction *&CollapsedCHA) {
1689   const Tool *ToolForJob = nullptr;
1690   CollapsedCHA = nullptr;
1691 
1692   // See if we should look for a compiler with an integrated assembler. We match
1693   // bottom up, so what we are actually looking for is an assembler job with a
1694   // compiler input.
1695 
1696   if (TC->useIntegratedAs() && !SaveTemps &&
1697       !C.getArgs().hasArg(options::OPT_via_file_asm) &&
1698       !C.getArgs().hasArg(options::OPT__SLASH_FA) &&
1699       !C.getArgs().hasArg(options::OPT__SLASH_Fa) &&
1700       isa<AssembleJobAction>(JA) && Inputs->size() == 1 &&
1701       isa<BackendJobAction>(*Inputs->begin())) {
1702     // A BackendJob is always preceded by a CompileJob, and without
1703     // -save-temps they will always get combined together, so instead of
1704     // checking the backend tool, check if the tool for the CompileJob
1705     // has an integrated assembler.
1706     const ActionList *BackendInputs = &(*Inputs)[0]->getInputs();
1707     // Compile job may be wrapped in CudaHostAction, extract it if
1708     // that's the case and update CollapsedCHA if we combine phases.
1709     CudaHostAction *CHA = dyn_cast<CudaHostAction>(*BackendInputs->begin());
1710     JobAction *CompileJA =
1711         cast<CompileJobAction>(CHA ? *CHA->begin() : *BackendInputs->begin());
1712     assert(CompileJA && "Backend job is not preceeded by compile job.");
1713     const Tool *Compiler = TC->SelectTool(*CompileJA);
1714     if (!Compiler)
1715       return nullptr;
1716     if (Compiler->hasIntegratedAssembler()) {
1717       Inputs = &CompileJA->getInputs();
1718       ToolForJob = Compiler;
1719       CollapsedCHA = CHA;
1720     }
1721   }
1722 
1723   // A backend job should always be combined with the preceding compile job
1724   // unless OPT_save_temps is enabled and the compiler is capable of emitting
1725   // LLVM IR as an intermediate output.
1726   if (isa<BackendJobAction>(JA)) {
1727     // Check if the compiler supports emitting LLVM IR.
1728     assert(Inputs->size() == 1);
1729     // Compile job may be wrapped in CudaHostAction, extract it if
1730     // that's the case and update CollapsedCHA if we combine phases.
1731     CudaHostAction *CHA = dyn_cast<CudaHostAction>(*Inputs->begin());
1732     JobAction *CompileJA =
1733         cast<CompileJobAction>(CHA ? *CHA->begin() : *Inputs->begin());
1734     assert(CompileJA && "Backend job is not preceeded by compile job.");
1735     const Tool *Compiler = TC->SelectTool(*CompileJA);
1736     if (!Compiler)
1737       return nullptr;
1738     if (!Compiler->canEmitIR() || !SaveTemps) {
1739       Inputs = &CompileJA->getInputs();
1740       ToolForJob = Compiler;
1741       CollapsedCHA = CHA;
1742     }
1743   }
1744 
1745   // Otherwise use the tool for the current job.
1746   if (!ToolForJob)
1747     ToolForJob = TC->SelectTool(*JA);
1748 
1749   // See if we should use an integrated preprocessor. We do so when we have
1750   // exactly one input, since this is the only use case we care about
1751   // (irrelevant since we don't support combine yet).
1752   if (Inputs->size() == 1 && isa<PreprocessJobAction>(*Inputs->begin()) &&
1753       !C.getArgs().hasArg(options::OPT_no_integrated_cpp) &&
1754       !C.getArgs().hasArg(options::OPT_traditional_cpp) && !SaveTemps &&
1755       !C.getArgs().hasArg(options::OPT_rewrite_objc) &&
1756       ToolForJob->hasIntegratedCPP())
1757     Inputs = &(*Inputs)[0]->getInputs();
1758 
1759   return ToolForJob;
1760 }
1761 
1762 void Driver::BuildJobsForAction(Compilation &C, const Action *A,
1763                                 const ToolChain *TC, const char *BoundArch,
1764                                 bool AtTopLevel, bool MultipleArchs,
1765                                 const char *LinkingOutput,
1766                                 InputInfo &Result) const {
1767   llvm::PrettyStackTraceString CrashInfo("Building compilation jobs");
1768 
1769   InputInfoList CudaDeviceInputInfos;
1770   if (const CudaHostAction *CHA = dyn_cast<CudaHostAction>(A)) {
1771     InputInfo II;
1772     // Append outputs of device jobs to the input list.
1773     for (const Action *DA : CHA->getDeviceActions()) {
1774       BuildJobsForAction(C, DA, TC, "", AtTopLevel,
1775                          /*MultipleArchs*/ false, LinkingOutput, II);
1776       CudaDeviceInputInfos.push_back(II);
1777     }
1778     // Override current action with a real host compile action and continue
1779     // processing it.
1780     A = *CHA->begin();
1781   }
1782 
1783   if (const InputAction *IA = dyn_cast<InputAction>(A)) {
1784     // FIXME: It would be nice to not claim this here; maybe the old scheme of
1785     // just using Args was better?
1786     const Arg &Input = IA->getInputArg();
1787     Input.claim();
1788     if (Input.getOption().matches(options::OPT_INPUT)) {
1789       const char *Name = Input.getValue();
1790       Result = InputInfo(Name, A->getType(), Name);
1791     } else {
1792       Result = InputInfo(&Input, A->getType(), "");
1793     }
1794     return;
1795   }
1796 
1797   if (const BindArchAction *BAA = dyn_cast<BindArchAction>(A)) {
1798     const ToolChain *TC;
1799     const char *ArchName = BAA->getArchName();
1800 
1801     if (ArchName)
1802       TC = &getToolChain(
1803           C.getArgs(),
1804           computeTargetTriple(DefaultTargetTriple, C.getArgs(), ArchName));
1805     else
1806       TC = &C.getDefaultToolChain();
1807 
1808     BuildJobsForAction(C, *BAA->begin(), TC, ArchName, AtTopLevel,
1809                        MultipleArchs, LinkingOutput, Result);
1810     return;
1811   }
1812 
1813   if (const CudaDeviceAction *CDA = dyn_cast<CudaDeviceAction>(A)) {
1814     BuildJobsForAction(
1815         C, *CDA->begin(),
1816         &getToolChain(C.getArgs(), llvm::Triple(CDA->getDeviceTriple())),
1817         CDA->getGpuArchName(), CDA->isAtTopLevel(),
1818         /*MultipleArchs*/ true, LinkingOutput, Result);
1819     return;
1820   }
1821 
1822   const ActionList *Inputs = &A->getInputs();
1823 
1824   const JobAction *JA = cast<JobAction>(A);
1825   const CudaHostAction *CollapsedCHA = nullptr;
1826   const Tool *T =
1827       selectToolForJob(C, isSaveTempsEnabled(), TC, JA, Inputs, CollapsedCHA);
1828   if (!T)
1829     return;
1830 
1831   // If we've collapsed action list that contained CudaHostAction we
1832   // need to build jobs for device-side inputs it may have held.
1833   if (CollapsedCHA) {
1834     InputInfo II;
1835     for (const Action *DA : CollapsedCHA->getDeviceActions()) {
1836       BuildJobsForAction(C, DA, TC, "", AtTopLevel,
1837                          /*MultipleArchs*/ false, LinkingOutput, II);
1838       CudaDeviceInputInfos.push_back(II);
1839     }
1840   }
1841 
1842   // Only use pipes when there is exactly one input.
1843   InputInfoList InputInfos;
1844   for (const Action *Input : *Inputs) {
1845     // Treat dsymutil and verify sub-jobs as being at the top-level too, they
1846     // shouldn't get temporary output names.
1847     // FIXME: Clean this up.
1848     bool SubJobAtTopLevel = false;
1849     if (AtTopLevel && (isa<DsymutilJobAction>(A) || isa<VerifyJobAction>(A)))
1850       SubJobAtTopLevel = true;
1851 
1852     InputInfo II;
1853     BuildJobsForAction(C, Input, TC, BoundArch, SubJobAtTopLevel, MultipleArchs,
1854                        LinkingOutput, II);
1855     InputInfos.push_back(II);
1856   }
1857 
1858   // Always use the first input as the base input.
1859   const char *BaseInput = InputInfos[0].getBaseInput();
1860 
1861   // ... except dsymutil actions, which use their actual input as the base
1862   // input.
1863   if (JA->getType() == types::TY_dSYM)
1864     BaseInput = InputInfos[0].getFilename();
1865 
1866   // Append outputs of cuda device jobs to the input list
1867   if (CudaDeviceInputInfos.size())
1868     InputInfos.append(CudaDeviceInputInfos.begin(), CudaDeviceInputInfos.end());
1869 
1870   // Determine the place to write output to, if any.
1871   if (JA->getType() == types::TY_Nothing)
1872     Result = InputInfo(A->getType(), BaseInput);
1873   else
1874     Result = InputInfo(GetNamedOutputPath(C, *JA, BaseInput, BoundArch,
1875                                           AtTopLevel, MultipleArchs),
1876                        A->getType(), BaseInput);
1877 
1878   if (CCCPrintBindings && !CCGenDiagnostics) {
1879     llvm::errs() << "# \"" << T->getToolChain().getTripleString() << '"'
1880                  << " - \"" << T->getName() << "\", inputs: [";
1881     for (unsigned i = 0, e = InputInfos.size(); i != e; ++i) {
1882       llvm::errs() << InputInfos[i].getAsString();
1883       if (i + 1 != e)
1884         llvm::errs() << ", ";
1885     }
1886     llvm::errs() << "], output: " << Result.getAsString() << "\n";
1887   } else {
1888     T->ConstructJob(C, *JA, Result, InputInfos,
1889                     C.getArgsForToolChain(TC, BoundArch), LinkingOutput);
1890   }
1891 }
1892 
1893 const char *Driver::getDefaultImageName() const {
1894   llvm::Triple Target(llvm::Triple::normalize(DefaultTargetTriple));
1895   return Target.isOSWindows() ? "a.exe" : "a.out";
1896 }
1897 
1898 /// \brief Create output filename based on ArgValue, which could either be a
1899 /// full filename, filename without extension, or a directory. If ArgValue
1900 /// does not provide a filename, then use BaseName, and use the extension
1901 /// suitable for FileType.
1902 static const char *MakeCLOutputFilename(const ArgList &Args, StringRef ArgValue,
1903                                         StringRef BaseName,
1904                                         types::ID FileType) {
1905   SmallString<128> Filename = ArgValue;
1906 
1907   if (ArgValue.empty()) {
1908     // If the argument is empty, output to BaseName in the current dir.
1909     Filename = BaseName;
1910   } else if (llvm::sys::path::is_separator(Filename.back())) {
1911     // If the argument is a directory, output to BaseName in that dir.
1912     llvm::sys::path::append(Filename, BaseName);
1913   }
1914 
1915   if (!llvm::sys::path::has_extension(ArgValue)) {
1916     // If the argument didn't provide an extension, then set it.
1917     const char *Extension = types::getTypeTempSuffix(FileType, true);
1918 
1919     if (FileType == types::TY_Image &&
1920         Args.hasArg(options::OPT__SLASH_LD, options::OPT__SLASH_LDd)) {
1921       // The output file is a dll.
1922       Extension = "dll";
1923     }
1924 
1925     llvm::sys::path::replace_extension(Filename, Extension);
1926   }
1927 
1928   return Args.MakeArgString(Filename.c_str());
1929 }
1930 
1931 const char *Driver::GetNamedOutputPath(Compilation &C, const JobAction &JA,
1932                                        const char *BaseInput,
1933                                        const char *BoundArch, bool AtTopLevel,
1934                                        bool MultipleArchs) const {
1935   llvm::PrettyStackTraceString CrashInfo("Computing output path");
1936   // Output to a user requested destination?
1937   if (AtTopLevel && !isa<DsymutilJobAction>(JA) && !isa<VerifyJobAction>(JA)) {
1938     if (Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o))
1939       return C.addResultFile(FinalOutput->getValue(), &JA);
1940   }
1941 
1942   // For /P, preprocess to file named after BaseInput.
1943   if (C.getArgs().hasArg(options::OPT__SLASH_P)) {
1944     assert(AtTopLevel && isa<PreprocessJobAction>(JA));
1945     StringRef BaseName = llvm::sys::path::filename(BaseInput);
1946     StringRef NameArg;
1947     if (Arg *A = C.getArgs().getLastArg(options::OPT__SLASH_Fi))
1948       NameArg = A->getValue();
1949     return C.addResultFile(
1950         MakeCLOutputFilename(C.getArgs(), NameArg, BaseName, types::TY_PP_C),
1951         &JA);
1952   }
1953 
1954   // Default to writing to stdout?
1955   if (AtTopLevel && !CCGenDiagnostics &&
1956       (isa<PreprocessJobAction>(JA) || JA.getType() == types::TY_ModuleFile))
1957     return "-";
1958 
1959   // Is this the assembly listing for /FA?
1960   if (JA.getType() == types::TY_PP_Asm &&
1961       (C.getArgs().hasArg(options::OPT__SLASH_FA) ||
1962        C.getArgs().hasArg(options::OPT__SLASH_Fa))) {
1963     // Use /Fa and the input filename to determine the asm file name.
1964     StringRef BaseName = llvm::sys::path::filename(BaseInput);
1965     StringRef FaValue = C.getArgs().getLastArgValue(options::OPT__SLASH_Fa);
1966     return C.addResultFile(
1967         MakeCLOutputFilename(C.getArgs(), FaValue, BaseName, JA.getType()),
1968         &JA);
1969   }
1970 
1971   // Output to a temporary file?
1972   if ((!AtTopLevel && !isSaveTempsEnabled() &&
1973        !C.getArgs().hasArg(options::OPT__SLASH_Fo)) ||
1974       CCGenDiagnostics) {
1975     StringRef Name = llvm::sys::path::filename(BaseInput);
1976     std::pair<StringRef, StringRef> Split = Name.split('.');
1977     std::string TmpName = GetTemporaryPath(
1978         Split.first, types::getTypeTempSuffix(JA.getType(), IsCLMode()));
1979     return C.addTempFile(C.getArgs().MakeArgString(TmpName.c_str()));
1980   }
1981 
1982   SmallString<128> BasePath(BaseInput);
1983   StringRef BaseName;
1984 
1985   // Dsymutil actions should use the full path.
1986   if (isa<DsymutilJobAction>(JA) || isa<VerifyJobAction>(JA))
1987     BaseName = BasePath;
1988   else
1989     BaseName = llvm::sys::path::filename(BasePath);
1990 
1991   // Determine what the derived output name should be.
1992   const char *NamedOutput;
1993 
1994   if (JA.getType() == types::TY_Object &&
1995       C.getArgs().hasArg(options::OPT__SLASH_Fo, options::OPT__SLASH_o)) {
1996     // The /Fo or /o flag decides the object filename.
1997     StringRef Val =
1998         C.getArgs()
1999             .getLastArg(options::OPT__SLASH_Fo, options::OPT__SLASH_o)
2000             ->getValue();
2001     NamedOutput =
2002         MakeCLOutputFilename(C.getArgs(), Val, BaseName, types::TY_Object);
2003   } else if (JA.getType() == types::TY_Image &&
2004              C.getArgs().hasArg(options::OPT__SLASH_Fe,
2005                                 options::OPT__SLASH_o)) {
2006     // The /Fe or /o flag names the linked file.
2007     StringRef Val =
2008         C.getArgs()
2009             .getLastArg(options::OPT__SLASH_Fe, options::OPT__SLASH_o)
2010             ->getValue();
2011     NamedOutput =
2012         MakeCLOutputFilename(C.getArgs(), Val, BaseName, types::TY_Image);
2013   } else if (JA.getType() == types::TY_Image) {
2014     if (IsCLMode()) {
2015       // clang-cl uses BaseName for the executable name.
2016       NamedOutput =
2017           MakeCLOutputFilename(C.getArgs(), "", BaseName, types::TY_Image);
2018     } else if (MultipleArchs && BoundArch) {
2019       SmallString<128> Output(getDefaultImageName());
2020       Output += "-";
2021       Output.append(BoundArch);
2022       NamedOutput = C.getArgs().MakeArgString(Output.c_str());
2023     } else
2024       NamedOutput = getDefaultImageName();
2025   } else {
2026     const char *Suffix = types::getTypeTempSuffix(JA.getType(), IsCLMode());
2027     assert(Suffix && "All types used for output should have a suffix.");
2028 
2029     std::string::size_type End = std::string::npos;
2030     if (!types::appendSuffixForType(JA.getType()))
2031       End = BaseName.rfind('.');
2032     SmallString<128> Suffixed(BaseName.substr(0, End));
2033     if (MultipleArchs && BoundArch) {
2034       Suffixed += "-";
2035       Suffixed.append(BoundArch);
2036     }
2037     // When using both -save-temps and -emit-llvm, use a ".tmp.bc" suffix for
2038     // the unoptimized bitcode so that it does not get overwritten by the ".bc"
2039     // optimized bitcode output.
2040     if (!AtTopLevel && C.getArgs().hasArg(options::OPT_emit_llvm) &&
2041         JA.getType() == types::TY_LLVM_BC)
2042       Suffixed += ".tmp";
2043     Suffixed += '.';
2044     Suffixed += Suffix;
2045     NamedOutput = C.getArgs().MakeArgString(Suffixed.c_str());
2046   }
2047 
2048   // Prepend object file path if -save-temps=obj
2049   if (!AtTopLevel && isSaveTempsObj() && C.getArgs().hasArg(options::OPT_o) &&
2050       JA.getType() != types::TY_PCH) {
2051     Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o);
2052     SmallString<128> TempPath(FinalOutput->getValue());
2053     llvm::sys::path::remove_filename(TempPath);
2054     StringRef OutputFileName = llvm::sys::path::filename(NamedOutput);
2055     llvm::sys::path::append(TempPath, OutputFileName);
2056     NamedOutput = C.getArgs().MakeArgString(TempPath.c_str());
2057   }
2058 
2059   // If we're saving temps and the temp file conflicts with the input file,
2060   // then avoid overwriting input file.
2061   if (!AtTopLevel && isSaveTempsEnabled() && NamedOutput == BaseName) {
2062     bool SameFile = false;
2063     SmallString<256> Result;
2064     llvm::sys::fs::current_path(Result);
2065     llvm::sys::path::append(Result, BaseName);
2066     llvm::sys::fs::equivalent(BaseInput, Result.c_str(), SameFile);
2067     // Must share the same path to conflict.
2068     if (SameFile) {
2069       StringRef Name = llvm::sys::path::filename(BaseInput);
2070       std::pair<StringRef, StringRef> Split = Name.split('.');
2071       std::string TmpName = GetTemporaryPath(
2072           Split.first, types::getTypeTempSuffix(JA.getType(), IsCLMode()));
2073       return C.addTempFile(C.getArgs().MakeArgString(TmpName.c_str()));
2074     }
2075   }
2076 
2077   // As an annoying special case, PCH generation doesn't strip the pathname.
2078   if (JA.getType() == types::TY_PCH) {
2079     llvm::sys::path::remove_filename(BasePath);
2080     if (BasePath.empty())
2081       BasePath = NamedOutput;
2082     else
2083       llvm::sys::path::append(BasePath, NamedOutput);
2084     return C.addResultFile(C.getArgs().MakeArgString(BasePath.c_str()), &JA);
2085   } else {
2086     return C.addResultFile(NamedOutput, &JA);
2087   }
2088 }
2089 
2090 std::string Driver::GetFilePath(const char *Name, const ToolChain &TC) const {
2091   // Respect a limited subset of the '-Bprefix' functionality in GCC by
2092   // attempting to use this prefix when looking for file paths.
2093   for (const std::string &Dir : PrefixDirs) {
2094     if (Dir.empty())
2095       continue;
2096     SmallString<128> P(Dir[0] == '=' ? SysRoot + Dir.substr(1) : Dir);
2097     llvm::sys::path::append(P, Name);
2098     if (llvm::sys::fs::exists(Twine(P)))
2099       return P.str();
2100   }
2101 
2102   SmallString<128> P(ResourceDir);
2103   llvm::sys::path::append(P, Name);
2104   if (llvm::sys::fs::exists(Twine(P)))
2105     return P.str();
2106 
2107   for (const std::string &Dir : TC.getFilePaths()) {
2108     if (Dir.empty())
2109       continue;
2110     SmallString<128> P(Dir[0] == '=' ? SysRoot + Dir.substr(1) : Dir);
2111     llvm::sys::path::append(P, Name);
2112     if (llvm::sys::fs::exists(Twine(P)))
2113       return P.str();
2114   }
2115 
2116   return Name;
2117 }
2118 
2119 void Driver::generatePrefixedToolNames(
2120     const char *Tool, const ToolChain &TC,
2121     SmallVectorImpl<std::string> &Names) const {
2122   // FIXME: Needs a better variable than DefaultTargetTriple
2123   Names.emplace_back(DefaultTargetTriple + "-" + Tool);
2124   Names.emplace_back(Tool);
2125 }
2126 
2127 static bool ScanDirForExecutable(SmallString<128> &Dir,
2128                                  ArrayRef<std::string> Names) {
2129   for (const auto &Name : Names) {
2130     llvm::sys::path::append(Dir, Name);
2131     if (llvm::sys::fs::can_execute(Twine(Dir)))
2132       return true;
2133     llvm::sys::path::remove_filename(Dir);
2134   }
2135   return false;
2136 }
2137 
2138 std::string Driver::GetProgramPath(const char *Name,
2139                                    const ToolChain &TC) const {
2140   SmallVector<std::string, 2> TargetSpecificExecutables;
2141   generatePrefixedToolNames(Name, TC, TargetSpecificExecutables);
2142 
2143   // Respect a limited subset of the '-Bprefix' functionality in GCC by
2144   // attempting to use this prefix when looking for program paths.
2145   for (const auto &PrefixDir : PrefixDirs) {
2146     if (llvm::sys::fs::is_directory(PrefixDir)) {
2147       SmallString<128> P(PrefixDir);
2148       if (ScanDirForExecutable(P, TargetSpecificExecutables))
2149         return P.str();
2150     } else {
2151       SmallString<128> P(PrefixDir + Name);
2152       if (llvm::sys::fs::can_execute(Twine(P)))
2153         return P.str();
2154     }
2155   }
2156 
2157   const ToolChain::path_list &List = TC.getProgramPaths();
2158   for (const auto &Path : List) {
2159     SmallString<128> P(Path);
2160     if (ScanDirForExecutable(P, TargetSpecificExecutables))
2161       return P.str();
2162   }
2163 
2164   // If all else failed, search the path.
2165   for (const auto &TargetSpecificExecutable : TargetSpecificExecutables)
2166     if (llvm::ErrorOr<std::string> P =
2167             llvm::sys::findProgramByName(TargetSpecificExecutable))
2168       return *P;
2169 
2170   return Name;
2171 }
2172 
2173 std::string Driver::GetTemporaryPath(StringRef Prefix,
2174                                      const char *Suffix) const {
2175   SmallString<128> Path;
2176   std::error_code EC = llvm::sys::fs::createTemporaryFile(Prefix, Suffix, Path);
2177   if (EC) {
2178     Diag(clang::diag::err_unable_to_make_temp) << EC.message();
2179     return "";
2180   }
2181 
2182   return Path.str();
2183 }
2184 
2185 const ToolChain &Driver::getToolChain(const ArgList &Args,
2186                                       const llvm::Triple &Target) const {
2187 
2188   ToolChain *&TC = ToolChains[Target.str()];
2189   if (!TC) {
2190     switch (Target.getOS()) {
2191     case llvm::Triple::CloudABI:
2192       TC = new toolchains::CloudABI(*this, Target, Args);
2193       break;
2194     case llvm::Triple::Darwin:
2195     case llvm::Triple::MacOSX:
2196     case llvm::Triple::IOS:
2197       TC = new toolchains::DarwinClang(*this, Target, Args);
2198       break;
2199     case llvm::Triple::DragonFly:
2200       TC = new toolchains::DragonFly(*this, Target, Args);
2201       break;
2202     case llvm::Triple::OpenBSD:
2203       TC = new toolchains::OpenBSD(*this, Target, Args);
2204       break;
2205     case llvm::Triple::Bitrig:
2206       TC = new toolchains::Bitrig(*this, Target, Args);
2207       break;
2208     case llvm::Triple::NetBSD:
2209       TC = new toolchains::NetBSD(*this, Target, Args);
2210       break;
2211     case llvm::Triple::FreeBSD:
2212       TC = new toolchains::FreeBSD(*this, Target, Args);
2213       break;
2214     case llvm::Triple::Minix:
2215       TC = new toolchains::Minix(*this, Target, Args);
2216       break;
2217     case llvm::Triple::Linux:
2218       if (Target.getArch() == llvm::Triple::hexagon)
2219         TC = new toolchains::HexagonToolChain(*this, Target, Args);
2220       else
2221         TC = new toolchains::Linux(*this, Target, Args);
2222       break;
2223     case llvm::Triple::NaCl:
2224       TC = new toolchains::NaClToolChain(*this, Target, Args);
2225       break;
2226     case llvm::Triple::Solaris:
2227       TC = new toolchains::Solaris(*this, Target, Args);
2228       break;
2229     case llvm::Triple::AMDHSA:
2230       TC = new toolchains::AMDGPUToolChain(*this, Target, Args);
2231       break;
2232     case llvm::Triple::Win32:
2233       switch (Target.getEnvironment()) {
2234       default:
2235         if (Target.isOSBinFormatELF())
2236           TC = new toolchains::Generic_ELF(*this, Target, Args);
2237         else if (Target.isOSBinFormatMachO())
2238           TC = new toolchains::MachO(*this, Target, Args);
2239         else
2240           TC = new toolchains::Generic_GCC(*this, Target, Args);
2241         break;
2242       case llvm::Triple::GNU:
2243         TC = new toolchains::MinGW(*this, Target, Args);
2244         break;
2245       case llvm::Triple::Itanium:
2246         TC = new toolchains::CrossWindowsToolChain(*this, Target, Args);
2247         break;
2248       case llvm::Triple::MSVC:
2249       case llvm::Triple::UnknownEnvironment:
2250         TC = new toolchains::MSVCToolChain(*this, Target, Args);
2251         break;
2252       }
2253       break;
2254     case llvm::Triple::CUDA:
2255       TC = new toolchains::CudaToolChain(*this, Target, Args);
2256       break;
2257     default:
2258       // Of these targets, Hexagon is the only one that might have
2259       // an OS of Linux, in which case it got handled above already.
2260       switch (Target.getArch()) {
2261       case llvm::Triple::tce:
2262         TC = new toolchains::TCEToolChain(*this, Target, Args);
2263         break;
2264       case llvm::Triple::hexagon:
2265         TC = new toolchains::HexagonToolChain(*this, Target, Args);
2266         break;
2267       case llvm::Triple::xcore:
2268         TC = new toolchains::XCoreToolChain(*this, Target, Args);
2269         break;
2270       case llvm::Triple::wasm32:
2271       case llvm::Triple::wasm64:
2272         TC = new toolchains::WebAssembly(*this, Target, Args);
2273         break;
2274       default:
2275         if (Target.getVendor() == llvm::Triple::Myriad)
2276           TC = new toolchains::MyriadToolChain(*this, Target, Args);
2277         else if (Target.isOSBinFormatELF())
2278           TC = new toolchains::Generic_ELF(*this, Target, Args);
2279         else if (Target.isOSBinFormatMachO())
2280           TC = new toolchains::MachO(*this, Target, Args);
2281         else
2282           TC = new toolchains::Generic_GCC(*this, Target, Args);
2283       }
2284     }
2285   }
2286   return *TC;
2287 }
2288 
2289 bool Driver::ShouldUseClangCompiler(const JobAction &JA) const {
2290   // Say "no" if there is not exactly one input of a type clang understands.
2291   if (JA.size() != 1 || !types::isAcceptedByClang((*JA.begin())->getType()))
2292     return false;
2293 
2294   // And say "no" if this is not a kind of action clang understands.
2295   if (!isa<PreprocessJobAction>(JA) && !isa<PrecompileJobAction>(JA) &&
2296       !isa<CompileJobAction>(JA) && !isa<BackendJobAction>(JA))
2297     return false;
2298 
2299   return true;
2300 }
2301 
2302 /// GetReleaseVersion - Parse (([0-9]+)(.([0-9]+)(.([0-9]+)?))?)? and return the
2303 /// grouped values as integers. Numbers which are not provided are set to 0.
2304 ///
2305 /// \return True if the entire string was parsed (9.2), or all groups were
2306 /// parsed (10.3.5extrastuff).
2307 bool Driver::GetReleaseVersion(const char *Str, unsigned &Major,
2308                                unsigned &Minor, unsigned &Micro,
2309                                bool &HadExtra) {
2310   HadExtra = false;
2311 
2312   Major = Minor = Micro = 0;
2313   if (*Str == '\0')
2314     return false;
2315 
2316   char *End;
2317   Major = (unsigned)strtol(Str, &End, 10);
2318   if (*Str != '\0' && *End == '\0')
2319     return true;
2320   if (*End != '.')
2321     return false;
2322 
2323   Str = End + 1;
2324   Minor = (unsigned)strtol(Str, &End, 10);
2325   if (*Str != '\0' && *End == '\0')
2326     return true;
2327   if (*End != '.')
2328     return false;
2329 
2330   Str = End + 1;
2331   Micro = (unsigned)strtol(Str, &End, 10);
2332   if (*Str != '\0' && *End == '\0')
2333     return true;
2334   if (Str == End)
2335     return false;
2336   HadExtra = true;
2337   return true;
2338 }
2339 
2340 std::pair<unsigned, unsigned> Driver::getIncludeExcludeOptionFlagMasks() const {
2341   unsigned IncludedFlagsBitmask = 0;
2342   unsigned ExcludedFlagsBitmask = options::NoDriverOption;
2343 
2344   if (Mode == CLMode) {
2345     // Include CL and Core options.
2346     IncludedFlagsBitmask |= options::CLOption;
2347     IncludedFlagsBitmask |= options::CoreOption;
2348   } else {
2349     ExcludedFlagsBitmask |= options::CLOption;
2350   }
2351 
2352   return std::make_pair(IncludedFlagsBitmask, ExcludedFlagsBitmask);
2353 }
2354 
2355 bool clang::driver::isOptimizationLevelFast(const ArgList &Args) {
2356   return Args.hasFlag(options::OPT_Ofast, options::OPT_O_Group, false);
2357 }
2358