contrib/llvm/tools/clang/lib/Driver/Driver.cpp

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