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