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 #ifdef HAVE_CLANG_CONFIG_H
  11 # include "clang/Config/config.h"
  12 #endif
  13
  14 #include "clang/Driver/Driver.h"
  15
  16 #include "clang/Driver/Action.h"
  17 #include "clang/Driver/Arg.h"
  18 #include "clang/Driver/ArgList.h"
  19 #include "clang/Driver/Compilation.h"
  20 #include "clang/Driver/DriverDiagnostic.h"
  21 #include "clang/Driver/HostInfo.h"
  22 #include "clang/Driver/Job.h"
  23 #include "clang/Driver/OptTable.h"
  24 #include "clang/Driver/Option.h"
  25 #include "clang/Driver/Options.h"
  26 #include "clang/Driver/Tool.h"
  27 #include "clang/Driver/ToolChain.h"
  28 #include "clang/Driver/Types.h"
  29
  30 #include "clang/Basic/Version.h"
  31
  32 #include "llvm/Config/config.h"
  33 #include "llvm/ADT/ArrayRef.h"
  34 #include "llvm/ADT/StringSet.h"
  35 #include "llvm/ADT/OwningPtr.h"
  36 #include "llvm/Support/PrettyStackTrace.h"
  37 #include "llvm/Support/raw_ostream.h"
  38 #include "llvm/Support/FileSystem.h"
  39 #include "llvm/Support/Path.h"
  40 #include "llvm/Support/Program.h"
  41
  42 #include "InputInfo.h"
  43
  44 #include <map>
  45
  46 using namespace clang::driver;
  47 using namespace clang;
  48
  49 Driver::Driver(llvm::StringRef ClangExecutable,
  50                llvm::StringRef DefaultHostTriple,
  51                llvm::StringRef DefaultImageName,
  52                bool IsProduction, bool CXXIsProduction,
  53                Diagnostic &Diags)
  54   : Opts(createDriverOptTable()), Diags(Diags),
  55     ClangExecutable(ClangExecutable), UseStdLib(true),
  56     DefaultHostTriple(DefaultHostTriple), DefaultImageName(DefaultImageName),
  57     DriverTitle("clang \"gcc-compatible\" driver"),
  58     Host(0),
  59     CCPrintOptionsFilename(0), CCPrintHeadersFilename(0),
  60     CCLogDiagnosticsFilename(0), CCCIsCXX(false),
  61     CCCIsCPP(false),CCCEcho(false), CCCPrintBindings(false),
  62     CCPrintOptions(false), CCPrintHeaders(false), CCLogDiagnostics(false),
  63     CCCGenericGCCName(""), CheckInputsExist(true), CCCUseClang(true),
  64     CCCUseClangCXX(true), CCCUseClangCPP(true), CCCUsePCH(true),
  65     SuppressMissingInputWarning(false) {
  66   if (IsProduction) {
  67     // In a "production" build, only use clang on architectures we expect to
  68     // work, and don't use clang C++.
  69     //
  70     // During development its more convenient to always have the driver use
  71     // clang, but we don't want users to be confused when things don't work, or
  72     // to file bugs for things we don't support.
  73     CCCClangArchs.insert(llvm::Triple::x86);
  74     CCCClangArchs.insert(llvm::Triple::x86_64);
  75     CCCClangArchs.insert(llvm::Triple::arm);
  76
  77     if (!CXXIsProduction)
  78       CCCUseClangCXX = false;
  79   }
  80
  81   Name = llvm::sys::path::stem(ClangExecutable);
  82   Dir  = llvm::sys::path::parent_path(ClangExecutable);
  83
  84   // Compute the path to the resource directory.
  85   llvm::StringRef ClangResourceDir(CLANG_RESOURCE_DIR);
  86   llvm::SmallString<128> P(Dir);
  87   if (ClangResourceDir != "")
  88     llvm::sys::path::append(P, ClangResourceDir);
  89   else
  90     llvm::sys::path::append(P, "..", "lib", "clang", CLANG_VERSION_STRING);
  91   ResourceDir = P.str();
  92 }
  93
  94 Driver::~Driver() {
  95   delete Opts;
  96   delete Host;
  97 }
  98
  99 InputArgList *Driver::ParseArgStrings(llvm::ArrayRef<const char *> ArgList) {
 100   llvm::PrettyStackTraceString CrashInfo("Command line argument parsing");
 101   unsigned MissingArgIndex, MissingArgCount;
 102   InputArgList *Args = getOpts().ParseArgs(ArgList.begin(), ArgList.end(),
 103                                            MissingArgIndex, MissingArgCount);
 104
 105   // Check for missing argument error.
 106   if (MissingArgCount)
 107     Diag(clang::diag::err_drv_missing_argument)
 108       << Args->getArgString(MissingArgIndex) << MissingArgCount;
 109
 110   // Check for unsupported options.
 111   for (ArgList::const_iterator it = Args->begin(), ie = Args->end();
 112        it != ie; ++it) {
 113     Arg *A = *it;
 114     if (A->getOption().isUnsupported()) {
 115       Diag(clang::diag::err_drv_unsupported_opt) << A->getAsString(*Args);
 116       continue;
 117     }
 118   }
 119
 120   return Args;
 121 }
 122
 123 DerivedArgList *Driver::TranslateInputArgs(const InputArgList &Args) const {
 124   DerivedArgList *DAL = new DerivedArgList(Args);
 125
 126   bool HasNostdlib = Args.hasArg(options::OPT_nostdlib);
 127   for (ArgList::const_iterator it = Args.begin(),
 128          ie = Args.end(); it != ie; ++it) {
 129     const Arg *A = *it;
 130
 131     // Unfortunately, we have to parse some forwarding options (-Xassembler,
 132     // -Xlinker, -Xpreprocessor) because we either integrate their functionality
 133     // (assembler and preprocessor), or bypass a previous driver ('collect2').
 134
 135     // Rewrite linker options, to replace --no-demangle with a custom internal
 136     // option.
 137     if ((A->getOption().matches(options::OPT_Wl_COMMA) ||
 138          A->getOption().matches(options::OPT_Xlinker)) &&
 139         A->containsValue("--no-demangle")) {
 140       // Add the rewritten no-demangle argument.
 141       DAL->AddFlagArg(A, Opts->getOption(options::OPT_Z_Xlinker__no_demangle));
 142
 143       // Add the remaining values as Xlinker arguments.
 144       for (unsigned i = 0, e = A->getNumValues(); i != e; ++i)
 145         if (llvm::StringRef(A->getValue(Args, i)) != "--no-demangle")
 146           DAL->AddSeparateArg(A, Opts->getOption(options::OPT_Xlinker),
 147                               A->getValue(Args, i));
 148
 149       continue;
 150     }
 151
 152     // Rewrite preprocessor options, to replace -Wp,-MD,FOO which is used by
 153     // some build systems. We don't try to be complete here because we don't
 154     // care to encourage this usage model.
 155     if (A->getOption().matches(options::OPT_Wp_COMMA) &&
 156         A->getNumValues() == 2 &&
 157         (A->getValue(Args, 0) == llvm::StringRef("-MD") ||
 158          A->getValue(Args, 0) == llvm::StringRef("-MMD"))) {
 159       // Rewrite to -MD/-MMD along with -MF.
 160       if (A->getValue(Args, 0) == llvm::StringRef("-MD"))
 161         DAL->AddFlagArg(A, Opts->getOption(options::OPT_MD));
 162       else
 163         DAL->AddFlagArg(A, Opts->getOption(options::OPT_MMD));
 164       DAL->AddSeparateArg(A, Opts->getOption(options::OPT_MF),
 165                           A->getValue(Args, 1));
 166       continue;
 167     }
 168
 169     // Rewrite reserved library names.
 170     if (A->getOption().matches(options::OPT_l)) {
 171       llvm::StringRef Value = A->getValue(Args);
 172
 173       // Rewrite unless -nostdlib is present.
 174       if (!HasNostdlib && Value == "stdc++") {
 175         DAL->AddFlagArg(A, Opts->getOption(
 176                               options::OPT_Z_reserved_lib_stdcxx));
 177         continue;
 178       }
 179
 180       // Rewrite unconditionally.
 181       if (Value == "cc_kext") {
 182         DAL->AddFlagArg(A, Opts->getOption(
 183                               options::OPT_Z_reserved_lib_cckext));
 184         continue;
 185       }
 186     }
 187
 188     DAL->append(*it);
 189   }
 190
 191   // Add a default value of -mlinker-version=, if one was given and the user
 192   // didn't specify one.
 193 #if defined(HOST_LINK_VERSION)
 194   if (!Args.hasArg(options::OPT_mlinker_version_EQ)) {
 195     DAL->AddJoinedArg(0, Opts->getOption(options::OPT_mlinker_version_EQ),
 196                       HOST_LINK_VERSION);
 197     DAL->getLastArg(options::OPT_mlinker_version_EQ)->claim();
 198   }
 199 #endif
 200
 201   return DAL;
 202 }
 203
 204 Compilation *Driver::BuildCompilation(llvm::ArrayRef<const char *> ArgList) {
 205   llvm::PrettyStackTraceString CrashInfo("Compilation construction");
 206
 207   // FIXME: Handle environment options which effect driver behavior, somewhere
 208   // (client?). GCC_EXEC_PREFIX, COMPILER_PATH, LIBRARY_PATH, LPATH,
 209   // CC_PRINT_OPTIONS.
 210
 211   // FIXME: What are we going to do with -V and -b?
 212
 213   // FIXME: This stuff needs to go into the Compilation, not the driver.
 214   bool CCCPrintOptions = false, CCCPrintActions = false;
 215
 216   InputArgList *Args = ParseArgStrings(ArgList.slice(1));
 217
 218   // -no-canonical-prefixes is used very early in main.
 219   Args->ClaimAllArgs(options::OPT_no_canonical_prefixes);
 220
 221   // Ignore -pipe.
 222   Args->ClaimAllArgs(options::OPT_pipe);
 223
 224   // Extract -ccc args.
 225   //
 226   // FIXME: We need to figure out where this behavior should live. Most of it
 227   // should be outside in the client; the parts that aren't should have proper
 228   // options, either by introducing new ones or by overloading gcc ones like -V
 229   // or -b.
 230   CCCPrintOptions = Args->hasArg(options::OPT_ccc_print_options);
 231   CCCPrintActions = Args->hasArg(options::OPT_ccc_print_phases);
 232   CCCPrintBindings = Args->hasArg(options::OPT_ccc_print_bindings);
 233   CCCIsCXX = Args->hasArg(options::OPT_ccc_cxx) || CCCIsCXX;
 234   CCCEcho = Args->hasArg(options::OPT_ccc_echo);
 235   if (const Arg *A = Args->getLastArg(options::OPT_ccc_gcc_name))
 236     CCCGenericGCCName = A->getValue(*Args);
 237   CCCUseClangCXX = Args->hasFlag(options::OPT_ccc_clang_cxx,
 238                                  options::OPT_ccc_no_clang_cxx,
 239                                  CCCUseClangCXX);
 240   CCCUsePCH = Args->hasFlag(options::OPT_ccc_pch_is_pch,
 241                             options::OPT_ccc_pch_is_pth);
 242   CCCUseClang = !Args->hasArg(options::OPT_ccc_no_clang);
 243   CCCUseClangCPP = !Args->hasArg(options::OPT_ccc_no_clang_cpp);
 244   if (const Arg *A = Args->getLastArg(options::OPT_ccc_clang_archs)) {
 245     llvm::StringRef Cur = A->getValue(*Args);
 246
 247     CCCClangArchs.clear();
 248     while (!Cur.empty()) {
 249       std::pair<llvm::StringRef, llvm::StringRef> Split = Cur.split(',');
 250
 251       if (!Split.first.empty()) {
 252         llvm::Triple::ArchType Arch =
 253           llvm::Triple(Split.first, "", "").getArch();
 254
 255         if (Arch == llvm::Triple::UnknownArch)
 256           Diag(clang::diag::err_drv_invalid_arch_name) << Split.first;
 257
 258         CCCClangArchs.insert(Arch);
 259       }
 260
 261       Cur = Split.second;
 262     }
 263   }
 264   // FIXME: We shouldn't overwrite the default host triple here, but we have
 265   // nowhere else to put this currently.
 266   if (const Arg *A = Args->getLastArg(options::OPT_ccc_host_triple))
 267     DefaultHostTriple = A->getValue(*Args);
 268   if (const Arg *A = Args->getLastArg(options::OPT_ccc_install_dir))
 269     Dir = InstalledDir = A->getValue(*Args);
 270   for (arg_iterator it = Args->filtered_begin(options::OPT_B),
 271          ie = Args->filtered_end(); it != ie; ++it) {
 272     const Arg *A = *it;
 273     A->claim();
 274     PrefixDirs.push_back(A->getValue(*Args, 0));
 275   }
 276   if (const Arg *A = Args->getLastArg(options::OPT__sysroot_EQ))
 277     SysRoot = A->getValue(*Args);
 278   if (Args->hasArg(options::OPT_nostdlib))
 279     UseStdLib = false;
 280
 281   Host = GetHostInfo(DefaultHostTriple.c_str());
 282
 283   // Perform the default argument translations.
 284   DerivedArgList *TranslatedArgs = TranslateInputArgs(*Args);
 285
 286   // The compilation takes ownership of Args.
 287   Compilation *C = new Compilation(*this, *Host->CreateToolChain(*Args), Args,
 288                                    TranslatedArgs);
 289
 290   // FIXME: This behavior shouldn't be here.
 291   if (CCCPrintOptions) {
 292     PrintOptions(C->getInputArgs());
 293     return C;
 294   }
 295
 296   if (!HandleImmediateArgs(*C))
 297     return C;
 298
 299   // Construct the list of abstract actions to perform for this compilation.
 300   if (Host->useDriverDriver())
 301     BuildUniversalActions(C->getDefaultToolChain(), C->getArgs(),
 302                           C->getActions());
 303   else
 304     BuildActions(C->getDefaultToolChain(), C->getArgs(), C->getActions());
 305
 306   if (CCCPrintActions) {
 307     PrintActions(*C);
 308     return C;
 309   }
 310
 311   BuildJobs(*C);
 312
 313   return C;
 314 }
 315
 316 int Driver::ExecuteCompilation(const Compilation &C) const {
 317   // Just print if -### was present.
 318   if (C.getArgs().hasArg(options::OPT__HASH_HASH_HASH)) {
 319     C.PrintJob(llvm::errs(), C.getJobs(), "\n", true);
 320     return 0;
 321   }
 322
 323   // If there were errors building the compilation, quit now.
 324   if (getDiags().hasErrorOccurred())
 325     return 1;
 326
 327   const Command *FailingCommand = 0;
 328   int Res = C.ExecuteJob(C.getJobs(), FailingCommand);
 329
 330   // Remove temp files.
 331   C.CleanupFileList(C.getTempFiles());
 332
 333   // If the command succeeded, we are done.
 334   if (Res == 0)
 335     return Res;
 336
 337   // Otherwise, remove result files as well.
 338   if (!C.getArgs().hasArg(options::OPT_save_temps))
 339     C.CleanupFileList(C.getResultFiles(), true);
 340
 341   // Print extra information about abnormal failures, if possible.
 342   //
 343   // This is ad-hoc, but we don't want to be excessively noisy. If the result
 344   // status was 1, assume the command failed normally. In particular, if it was
 345   // the compiler then assume it gave a reasonable error code. Failures in other
 346   // tools are less common, and they generally have worse diagnostics, so always
 347   // print the diagnostic there.
 348   const Tool &FailingTool = FailingCommand->getCreator();
 349
 350   if (!FailingCommand->getCreator().hasGoodDiagnostics() || Res != 1) {
 351     // FIXME: See FIXME above regarding result code interpretation.
 352     if (Res < 0)
 353       Diag(clang::diag::err_drv_command_signalled)
 354         << FailingTool.getShortName() << -Res;
 355     else
 356       Diag(clang::diag::err_drv_command_failed)
 357         << FailingTool.getShortName() << Res;
 358   }
 359
 360   return Res;
 361 }
 362
 363 void Driver::PrintOptions(const ArgList &Args) const {
 364   unsigned i = 0;
 365   for (ArgList::const_iterator it = Args.begin(), ie = Args.end();
 366        it != ie; ++it, ++i) {
 367     Arg *A = *it;
 368     llvm::errs() << "Option " << i << " - "
 369                  << "Name: \"" << A->getOption().getName() << "\", "
 370                  << "Values: {";
 371     for (unsigned j = 0; j < A->getNumValues(); ++j) {
 372       if (j)
 373         llvm::errs() << ", ";
 374       llvm::errs() << '"' << A->getValue(Args, j) << '"';
 375     }
 376     llvm::errs() << "}\n";
 377   }
 378 }
 379
 380 void Driver::PrintHelp(bool ShowHidden) const {
 381   getOpts().PrintHelp(llvm::outs(), Name.c_str(), DriverTitle.c_str(),
 382                       ShowHidden);
 383 }
 384
 385 void Driver::PrintVersion(const Compilation &C, llvm::raw_ostream &OS) const {
 386   // FIXME: The following handlers should use a callback mechanism, we don't
 387   // know what the client would like to do.
 388   OS << getClangFullVersion() << '\n';
 389   const ToolChain &TC = C.getDefaultToolChain();
 390   OS << "Target: " << TC.getTripleString() << '\n';
 391
 392   // Print the threading model.
 393   //
 394   // FIXME: Implement correctly.
 395   OS << "Thread model: " << "posix" << '\n';
 396 }
 397
 398 /// PrintDiagnosticCategories - Implement the --print-diagnostic-categories
 399 /// option.
 400 static void PrintDiagnosticCategories(llvm::raw_ostream &OS) {
 401   // Skip the empty category.
 402   for (unsigned i = 1, max = DiagnosticIDs::getNumberOfCategories();
 403        i != max; ++i)
 404     OS << i << ',' << DiagnosticIDs::getCategoryNameFromID(i) << '\n';
 405 }
 406
 407 bool Driver::HandleImmediateArgs(const Compilation &C) {
 408   // The order these options are handled in gcc is all over the place, but we
 409   // don't expect inconsistencies w.r.t. that to matter in practice.
 410
 411   if (C.getArgs().hasArg(options::OPT_dumpmachine)) {
 412     llvm::outs() << C.getDefaultToolChain().getTripleString() << '\n';
 413     return false;
 414   }
 415
 416   if (C.getArgs().hasArg(options::OPT_dumpversion)) {
 417     // Since -dumpversion is only implemented for pedantic GCC compatibility, we
 418     // return an answer which matches our definition of __VERSION__.
 419     //
 420     // If we want to return a more correct answer some day, then we should
 421     // introduce a non-pedantically GCC compatible mode to Clang in which we
 422     // provide sensible definitions for -dumpversion, __VERSION__, etc.
 423     llvm::outs() << "4.2.1\n";
 424     return false;
 425   }
 426
 427   if (C.getArgs().hasArg(options::OPT__print_diagnostic_categories)) {
 428     PrintDiagnosticCategories(llvm::outs());
 429     return false;
 430   }
 431
 432   if (C.getArgs().hasArg(options::OPT__help) ||
 433       C.getArgs().hasArg(options::OPT__help_hidden)) {
 434     PrintHelp(C.getArgs().hasArg(options::OPT__help_hidden));
 435     return false;
 436   }
 437
 438   if (C.getArgs().hasArg(options::OPT__version)) {
 439     // Follow gcc behavior and use stdout for --version and stderr for -v.
 440     PrintVersion(C, llvm::outs());
 441     return false;
 442   }
 443
 444   if (C.getArgs().hasArg(options::OPT_v) ||
 445       C.getArgs().hasArg(options::OPT__HASH_HASH_HASH)) {
 446     PrintVersion(C, llvm::errs());
 447     SuppressMissingInputWarning = true;
 448   }
 449
 450   const ToolChain &TC = C.getDefaultToolChain();
 451   if (C.getArgs().hasArg(options::OPT_print_search_dirs)) {
 452     llvm::outs() << "programs: =";
 453     for (ToolChain::path_list::const_iterator it = TC.getProgramPaths().begin(),
 454            ie = TC.getProgramPaths().end(); it != ie; ++it) {
 455       if (it != TC.getProgramPaths().begin())
 456         llvm::outs() << ':';
 457       llvm::outs() << *it;
 458     }
 459     llvm::outs() << "\n";
 460     llvm::outs() << "libraries: =";
 461
 462     std::string sysroot;
 463     if (Arg *A = C.getArgs().getLastArg(options::OPT__sysroot_EQ))
 464       sysroot = A->getValue(C.getArgs());
 465
 466     for (ToolChain::path_list::const_iterator it = TC.getFilePaths().begin(),
 467            ie = TC.getFilePaths().end(); it != ie; ++it) {
 468       if (it != TC.getFilePaths().begin())
 469         llvm::outs() << ':';
 470       const char *path = it->c_str();
 471       if (path[0] == '=')
 472         llvm::outs() << sysroot << path + 1;
 473       else
 474         llvm::outs() << path;
 475     }
 476     llvm::outs() << "\n";
 477     return false;
 478   }
 479
 480   // FIXME: The following handlers should use a callback mechanism, we don't
 481   // know what the client would like to do.
 482   if (Arg *A = C.getArgs().getLastArg(options::OPT_print_file_name_EQ)) {
 483     llvm::outs() << GetFilePath(A->getValue(C.getArgs()), TC) << "\n";
 484     return false;
 485   }
 486
 487   if (Arg *A = C.getArgs().getLastArg(options::OPT_print_prog_name_EQ)) {
 488     llvm::outs() << GetProgramPath(A->getValue(C.getArgs()), TC) << "\n";
 489     return false;
 490   }
 491
 492   if (C.getArgs().hasArg(options::OPT_print_libgcc_file_name)) {
 493     llvm::outs() << GetFilePath("libgcc.a", TC) << "\n";
 494     return false;
 495   }
 496
 497   if (C.getArgs().hasArg(options::OPT_print_multi_lib)) {
 498     // FIXME: We need tool chain support for this.
 499     llvm::outs() << ".;\n";
 500
 501     switch (C.getDefaultToolChain().getTriple().getArch()) {
 502     default:
 503       break;
 504
 505     case llvm::Triple::x86_64:
 506       llvm::outs() << "x86_64;@m64" << "\n";
 507       break;
 508
 509     case llvm::Triple::ppc64:
 510       llvm::outs() << "ppc64;@m64" << "\n";
 511       break;
 512     }
 513     return false;
 514   }
 515
 516   // FIXME: What is the difference between print-multi-directory and
 517   // print-multi-os-directory?
 518   if (C.getArgs().hasArg(options::OPT_print_multi_directory) ||
 519       C.getArgs().hasArg(options::OPT_print_multi_os_directory)) {
 520     switch (C.getDefaultToolChain().getTriple().getArch()) {
 521     default:
 522     case llvm::Triple::x86:
 523     case llvm::Triple::ppc:
 524       llvm::outs() << "." << "\n";
 525       break;
 526
 527     case llvm::Triple::x86_64:
 528       llvm::outs() << "." << "\n";
 529       break;
 530
 531     case llvm::Triple::ppc64:
 532       llvm::outs() << "ppc64" << "\n";
 533       break;
 534     }
 535     return false;
 536   }
 537
 538   return true;
 539 }
 540
 541 static unsigned PrintActions1(const Compilation &C, Action *A,
 542                               std::map<Action*, unsigned> &Ids) {
 543   if (Ids.count(A))
 544     return Ids[A];
 545
 546   std::string str;
 547   llvm::raw_string_ostream os(str);
 548
 549   os << Action::getClassName(A->getKind()) << ", ";
 550   if (InputAction *IA = dyn_cast<InputAction>(A)) {
 551     os << "\"" << IA->getInputArg().getValue(C.getArgs()) << "\"";
 552   } else if (BindArchAction *BIA = dyn_cast<BindArchAction>(A)) {
 553     os << '"' << (BIA->getArchName() ? BIA->getArchName() :
 554                   C.getDefaultToolChain().getArchName()) << '"'
 555        << ", {" << PrintActions1(C, *BIA->begin(), Ids) << "}";
 556   } else {
 557     os << "{";
 558     for (Action::iterator it = A->begin(), ie = A->end(); it != ie;) {
 559       os << PrintActions1(C, *it, Ids);
 560       ++it;
 561       if (it != ie)
 562         os << ", ";
 563     }
 564     os << "}";
 565   }
 566
 567   unsigned Id = Ids.size();
 568   Ids[A] = Id;
 569   llvm::errs() << Id << ": " << os.str() << ", "
 570                << types::getTypeName(A->getType()) << "\n";
 571
 572   return Id;
 573 }
 574
 575 void Driver::PrintActions(const Compilation &C) const {
 576   std::map<Action*, unsigned> Ids;
 577   for (ActionList::const_iterator it = C.getActions().begin(),
 578          ie = C.getActions().end(); it != ie; ++it)
 579     PrintActions1(C, *it, Ids);
 580 }
 581
 582 /// \brief Check whether the given input tree contains any compilation or
 583 /// assembly actions.
 584 static bool ContainsCompileOrAssembleAction(const Action *A) {
 585   if (isa<CompileJobAction>(A) || isa<AssembleJobAction>(A))
 586     return true;
 587
 588   for (Action::const_iterator it = A->begin(), ie = A->end(); it != ie; ++it)
 589     if (ContainsCompileOrAssembleAction(*it))
 590       return true;
 591
 592   return false;
 593 }
 594
 595 void Driver::BuildUniversalActions(const ToolChain &TC,
 596                                    const DerivedArgList &Args,
 597                                    ActionList &Actions) const {
 598   llvm::PrettyStackTraceString CrashInfo("Building universal build actions");
 599   // Collect the list of architectures. Duplicates are allowed, but should only
 600   // be handled once (in the order seen).
 601   llvm::StringSet<> ArchNames;
 602   llvm::SmallVector<const char *, 4> Archs;
 603   for (ArgList::const_iterator it = Args.begin(), ie = Args.end();
 604        it != ie; ++it) {
 605     Arg *A = *it;
 606
 607     if (A->getOption().matches(options::OPT_arch)) {
 608       // Validate the option here; we don't save the type here because its
 609       // particular spelling may participate in other driver choices.
 610       llvm::Triple::ArchType Arch =
 611         llvm::Triple::getArchTypeForDarwinArchName(A->getValue(Args));
 612       if (Arch == llvm::Triple::UnknownArch) {
 613         Diag(clang::diag::err_drv_invalid_arch_name)
 614           << A->getAsString(Args);
 615         continue;
 616       }
 617
 618       A->claim();
 619       if (ArchNames.insert(A->getValue(Args)))
 620         Archs.push_back(A->getValue(Args));
 621     }
 622   }
 623
 624   // When there is no explicit arch for this platform, make sure we still bind
 625   // the architecture (to the default) so that -Xarch_ is handled correctly.
 626   if (!Archs.size())
 627     Archs.push_back(0);
 628
 629   // FIXME: We killed off some others but these aren't yet detected in a
 630   // functional manner. If we added information to jobs about which "auxiliary"
 631   // files they wrote then we could detect the conflict these cause downstream.
 632   if (Archs.size() > 1) {
 633     // No recovery needed, the point of this is just to prevent
 634     // overwriting the same files.
 635     if (const Arg *A = Args.getLastArg(options::OPT_save_temps))
 636       Diag(clang::diag::err_drv_invalid_opt_with_multiple_archs)
 637         << A->getAsString(Args);
 638   }
 639
 640   ActionList SingleActions;
 641   BuildActions(TC, Args, SingleActions);
 642
 643   // Add in arch bindings for every top level action, as well as lipo and
 644   // dsymutil steps if needed.
 645   for (unsigned i = 0, e = SingleActions.size(); i != e; ++i) {
 646     Action *Act = SingleActions[i];
 647
 648     // Make sure we can lipo this kind of output. If not (and it is an actual
 649     // output) then we disallow, since we can't create an output file with the
 650     // right name without overwriting it. We could remove this oddity by just
 651     // changing the output names to include the arch, which would also fix
 652     // -save-temps. Compatibility wins for now.
 653
 654     if (Archs.size() > 1 && !types::canLipoType(Act->getType()))
 655       Diag(clang::diag::err_drv_invalid_output_with_multiple_archs)
 656         << types::getTypeName(Act->getType());
 657
 658     ActionList Inputs;
 659     for (unsigned i = 0, e = Archs.size(); i != e; ++i) {
 660       Inputs.push_back(new BindArchAction(Act, Archs[i]));
 661       if (i != 0)
 662         Inputs.back()->setOwnsInputs(false);
 663     }
 664
 665     // Lipo if necessary, we do it this way because we need to set the arch flag
 666     // so that -Xarch_ gets overwritten.
 667     if (Inputs.size() == 1 || Act->getType() == types::TY_Nothing)
 668       Actions.append(Inputs.begin(), Inputs.end());
 669     else
 670       Actions.push_back(new LipoJobAction(Inputs, Act->getType()));
 671
 672     // Add a 'dsymutil' step if necessary, when debug info is enabled and we
 673     // have a compile input. We need to run 'dsymutil' ourselves in such cases
 674     // because the debug info will refer to a temporary object file which is
 675     // will be removed at the end of the compilation process.
 676     if (Act->getType() == types::TY_Image) {
 677       Arg *A = Args.getLastArg(options::OPT_g_Group);
 678       if (A && !A->getOption().matches(options::OPT_g0) &&
 679           !A->getOption().matches(options::OPT_gstabs) &&
 680           ContainsCompileOrAssembleAction(Actions.back())) {
 681         ActionList Inputs;
 682         Inputs.push_back(Actions.back());
 683         Actions.pop_back();
 684
 685         Actions.push_back(new DsymutilJobAction(Inputs, types::TY_dSYM));
 686       }
 687     }
 688   }
 689 }
 690
 691 void Driver::BuildActions(const ToolChain &TC, const DerivedArgList &Args,
 692                           ActionList &Actions) const {
 693   llvm::PrettyStackTraceString CrashInfo("Building compilation actions");
 694   // Start by constructing the list of inputs and their types.
 695
 696   // Track the current user specified (-x) input. We also explicitly track the
 697   // argument used to set the type; we only want to claim the type when we
 698   // actually use it, so we warn about unused -x arguments.
 699   types::ID InputType = types::TY_Nothing;
 700   Arg *InputTypeArg = 0;
 701
 702   llvm::SmallVector<std::pair<types::ID, const Arg*>, 16> Inputs;
 703   for (ArgList::const_iterator it = Args.begin(), ie = Args.end();
 704        it != ie; ++it) {
 705     Arg *A = *it;
 706
 707     if (isa<InputOption>(A->getOption())) {
 708       const char *Value = A->getValue(Args);
 709       types::ID Ty = types::TY_INVALID;
 710
 711       // Infer the input type if necessary.
 712       if (InputType == types::TY_Nothing) {
 713         // If there was an explicit arg for this, claim it.
 714         if (InputTypeArg)
 715           InputTypeArg->claim();
 716
 717         // stdin must be handled specially.
 718         if (memcmp(Value, "-", 2) == 0) {
 719           // If running with -E, treat as a C input (this changes the builtin
 720           // macros, for example). This may be overridden by -ObjC below.
 721           //
 722           // Otherwise emit an error but still use a valid type to avoid
 723           // spurious errors (e.g., no inputs).
 724           if (!Args.hasArgNoClaim(options::OPT_E) && !CCCIsCPP)
 725             Diag(clang::diag::err_drv_unknown_stdin_type);
 726           Ty = types::TY_C;
 727         } else {
 728           // Otherwise lookup by extension.
 729           // Fallback is C if invoked as C preprocessor or Object otherwise.
 730           // We use a host hook here because Darwin at least has its own
 731           // idea of what .s is.
 732           if (const char *Ext = strrchr(Value, '.'))
 733             Ty = TC.LookupTypeForExtension(Ext + 1);
 734
 735           if (Ty == types::TY_INVALID) {
 736             if (CCCIsCPP)
 737               Ty = types::TY_C;
 738             else
 739               Ty = types::TY_Object;
 740           }
 741
 742           // If the driver is invoked as C++ compiler (like clang++ or c++) it
 743           // should autodetect some input files as C++ for g++ compatibility.
 744           if (CCCIsCXX) {
 745             types::ID OldTy = Ty;
 746             Ty = types::lookupCXXTypeForCType(Ty);
 747
 748             if (Ty != OldTy)
 749               Diag(clang::diag::warn_drv_treating_input_as_cxx)
 750                 << getTypeName(OldTy) << getTypeName(Ty);
 751           }
 752         }
 753
 754         // -ObjC and -ObjC++ override the default language, but only for "source
 755         // files". We just treat everything that isn't a linker input as a
 756         // source file.
 757         //
 758         // FIXME: Clean this up if we move the phase sequence into the type.
 759         if (Ty != types::TY_Object) {
 760           if (Args.hasArg(options::OPT_ObjC))
 761             Ty = types::TY_ObjC;
 762           else if (Args.hasArg(options::OPT_ObjCXX))
 763             Ty = types::TY_ObjCXX;
 764         }
 765       } else {
 766         assert(InputTypeArg && "InputType set w/o InputTypeArg");
 767         InputTypeArg->claim();
 768         Ty = InputType;
 769       }
 770
 771       // Check that the file exists, if enabled.
 772       if (CheckInputsExist && memcmp(Value, "-", 2) != 0) {
 773         llvm::SmallString<64> Path(Value);
 774         if (Arg *WorkDir = Args.getLastArg(options::OPT_working_directory))
 775           if (llvm::sys::path::is_absolute(Path.str())) {
 776             Path = WorkDir->getValue(Args);
 777             llvm::sys::path::append(Path, Value);
 778           }
 779
 780         bool exists = false;
 781         if (/*error_code ec =*/llvm::sys::fs::exists(Value, exists) || !exists)
 782           Diag(clang::diag::err_drv_no_such_file) << Path.str();
 783         else
 784           Inputs.push_back(std::make_pair(Ty, A));
 785       } else
 786         Inputs.push_back(std::make_pair(Ty, A));
 787
 788     } else if (A->getOption().isLinkerInput()) {
 789       // Just treat as object type, we could make a special type for this if
 790       // necessary.
 791       Inputs.push_back(std::make_pair(types::TY_Object, A));
 792
 793     } else if (A->getOption().matches(options::OPT_x)) {
 794       InputTypeArg = A;
 795       InputType = types::lookupTypeForTypeSpecifier(A->getValue(Args));
 796
 797       // Follow gcc behavior and treat as linker input for invalid -x
 798       // options. Its not clear why we shouldn't just revert to unknown; but
 799       // this isn't very important, we might as well be bug compatible.
 800       if (!InputType) {
 801         Diag(clang::diag::err_drv_unknown_language) << A->getValue(Args);
 802         InputType = types::TY_Object;
 803       }
 804     }
 805   }
 806
 807   if (CCCIsCPP && Inputs.empty()) {
 808     // If called as standalone preprocessor, stdin is processed
 809     // if no other input is present.
 810     unsigned Index = Args.getBaseArgs().MakeIndex("-");
 811     Arg *A = Opts->ParseOneArg(Args, Index);
 812     A->claim();
 813     Inputs.push_back(std::make_pair(types::TY_C, A));
 814   }
 815
 816   if (!SuppressMissingInputWarning && Inputs.empty()) {
 817     Diag(clang::diag::err_drv_no_input_files);
 818     return;
 819   }
 820
 821   // Determine which compilation mode we are in. We look for options which
 822   // affect the phase, starting with the earliest phases, and record which
 823   // option we used to determine the final phase.
 824   Arg *FinalPhaseArg = 0;
 825   phases::ID FinalPhase;
 826
 827   // -{E,M,MM} only run the preprocessor.
 828   if (CCCIsCPP ||
 829       (FinalPhaseArg = Args.getLastArg(options::OPT_E)) ||
 830       (FinalPhaseArg = Args.getLastArg(options::OPT_M, options::OPT_MM))) {
 831     FinalPhase = phases::Preprocess;
 832
 833     // -{fsyntax-only,-analyze,emit-ast,S} only run up to the compiler.
 834   } else if ((FinalPhaseArg = Args.getLastArg(options::OPT_fsyntax_only)) ||
 835              (FinalPhaseArg = Args.getLastArg(options::OPT_rewrite_objc)) ||
 836              (FinalPhaseArg = Args.getLastArg(options::OPT__analyze,
 837                                               options::OPT__analyze_auto)) ||
 838              (FinalPhaseArg = Args.getLastArg(options::OPT_emit_ast)) ||
 839              (FinalPhaseArg = Args.getLastArg(options::OPT_S))) {
 840     FinalPhase = phases::Compile;
 841
 842     // -c only runs up to the assembler.
 843   } else if ((FinalPhaseArg = Args.getLastArg(options::OPT_c))) {
 844     FinalPhase = phases::Assemble;
 845
 846     // Otherwise do everything.
 847   } else
 848     FinalPhase = phases::Link;
 849
 850   // Reject -Z* at the top level, these options should never have been exposed
 851   // by gcc.
 852   if (Arg *A = Args.getLastArg(options::OPT_Z_Joined))
 853     Diag(clang::diag::err_drv_use_of_Z_option) << A->getAsString(Args);
 854
 855   // Construct the actions to perform.
 856   ActionList LinkerInputs;
 857   for (unsigned i = 0, e = Inputs.size(); i != e; ++i) {
 858     types::ID InputType = Inputs[i].first;
 859     const Arg *InputArg = Inputs[i].second;
 860
 861     unsigned NumSteps = types::getNumCompilationPhases(InputType);
 862     assert(NumSteps && "Invalid number of steps!");
 863
 864     // If the first step comes after the final phase we are doing as part of
 865     // this compilation, warn the user about it.
 866     phases::ID InitialPhase = types::getCompilationPhase(InputType, 0);
 867     if (InitialPhase > FinalPhase) {
 868       // Claim here to avoid the more general unused warning.
 869       InputArg->claim();
 870
 871       // Suppress all unused style warnings with -Qunused-arguments
 872       if (Args.hasArg(options::OPT_Qunused_arguments))
 873         continue;
 874
 875       // Special case '-E' warning on a previously preprocessed file to make
 876       // more sense.
 877       if (InitialPhase == phases::Compile && FinalPhase == phases::Preprocess &&
 878           getPreprocessedType(InputType) == types::TY_INVALID)
 879         Diag(clang::diag::warn_drv_preprocessed_input_file_unused)
 880           << InputArg->getAsString(Args)
 881           << FinalPhaseArg->getOption().getName();
 882       else
 883         Diag(clang::diag::warn_drv_input_file_unused)
 884           << InputArg->getAsString(Args)
 885           << getPhaseName(InitialPhase)
 886           << FinalPhaseArg->getOption().getName();
 887       continue;
 888     }
 889
 890     // Build the pipeline for this file.
 891     llvm::OwningPtr<Action> Current(new InputAction(*InputArg, InputType));
 892     for (unsigned i = 0; i != NumSteps; ++i) {
 893       phases::ID Phase = types::getCompilationPhase(InputType, i);
 894
 895       // We are done if this step is past what the user requested.
 896       if (Phase > FinalPhase)
 897         break;
 898
 899       // Queue linker inputs.
 900       if (Phase == phases::Link) {
 901         assert(i + 1 == NumSteps && "linking must be final compilation step.");
 902         LinkerInputs.push_back(Current.take());
 903         break;
 904       }
 905
 906       // Some types skip the assembler phase (e.g., llvm-bc), but we can't
 907       // encode this in the steps because the intermediate type depends on
 908       // arguments. Just special case here.
 909       if (Phase == phases::Assemble && Current->getType() != types::TY_PP_Asm)
 910         continue;
 911
 912       // Otherwise construct the appropriate action.
 913       Current.reset(ConstructPhaseAction(Args, Phase, Current.take()));
 914       if (Current->getType() == types::TY_Nothing)
 915         break;
 916     }
 917
 918     // If we ended with something, add to the output list.
 919     if (Current)
 920       Actions.push_back(Current.take());
 921   }
 922
 923   // Add a link action if necessary.
 924   if (!LinkerInputs.empty())
 925     Actions.push_back(new LinkJobAction(LinkerInputs, types::TY_Image));
 926
 927   // If we are linking, claim any options which are obviously only used for
 928   // compilation.
 929   if (FinalPhase == phases::Link)
 930     Args.ClaimAllArgs(options::OPT_CompileOnly_Group);
 931 }
 932
 933 Action *Driver::ConstructPhaseAction(const ArgList &Args, phases::ID Phase,
 934                                      Action *Input) const {
 935   llvm::PrettyStackTraceString CrashInfo("Constructing phase actions");
 936   // Build the appropriate action.
 937   switch (Phase) {
 938   case phases::Link: assert(0 && "link action invalid here.");
 939   case phases::Preprocess: {
 940     types::ID OutputTy;
 941     // -{M, MM} alter the output type.
 942     if (Args.hasArg(options::OPT_M, options::OPT_MM)) {
 943       OutputTy = types::TY_Dependencies;
 944     } else {
 945       OutputTy = types::getPreprocessedType(Input->getType());
 946       assert(OutputTy != types::TY_INVALID &&
 947              "Cannot preprocess this input type!");
 948     }
 949     return new PreprocessJobAction(Input, OutputTy);
 950   }
 951   case phases::Precompile:
 952     return new PrecompileJobAction(Input, types::TY_PCH);
 953   case phases::Compile: {
 954     if (Args.hasArg(options::OPT_fsyntax_only)) {
 955       return new CompileJobAction(Input, types::TY_Nothing);
 956     } else if (Args.hasArg(options::OPT_rewrite_objc)) {
 957       return new CompileJobAction(Input, types::TY_RewrittenObjC);
 958     } else if (Args.hasArg(options::OPT__analyze, options::OPT__analyze_auto)) {
 959       return new AnalyzeJobAction(Input, types::TY_Plist);
 960     } else if (Args.hasArg(options::OPT_emit_ast)) {
 961       return new CompileJobAction(Input, types::TY_AST);
 962     } else if (IsUsingLTO(Args)) {
 963       types::ID Output =
 964         Args.hasArg(options::OPT_S) ? types::TY_LTO_IR : types::TY_LTO_BC;
 965       return new CompileJobAction(Input, Output);
 966     } else {
 967       return new CompileJobAction(Input, types::TY_PP_Asm);
 968     }
 969   }
 970   case phases::Assemble:
 971     return new AssembleJobAction(Input, types::TY_Object);
 972   }
 973
 974   assert(0 && "invalid phase in ConstructPhaseAction");
 975   return 0;
 976 }
 977
 978 bool Driver::IsUsingLTO(const ArgList &Args) const {
 979   // Check for -emit-llvm or -flto.
 980   if (Args.hasArg(options::OPT_emit_llvm) ||
 981       Args.hasFlag(options::OPT_flto, options::OPT_fno_lto, false))
 982     return true;
 983
 984   // Check for -O4.
 985   if (const Arg *A = Args.getLastArg(options::OPT_O_Group))
 986       return A->getOption().matches(options::OPT_O4);
 987
 988   return false;
 989 }
 990
 991 void Driver::BuildJobs(Compilation &C) const {
 992   llvm::PrettyStackTraceString CrashInfo("Building compilation jobs");
 993
 994   Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o);
 995
 996   // It is an error to provide a -o option if we are making multiple output
 997   // files.
 998   if (FinalOutput) {
 999     unsigned NumOutputs = 0;
1000     for (ActionList::const_iterator it = C.getActions().begin(),
1001            ie = C.getActions().end(); it != ie; ++it)
1002       if ((*it)->getType() != types::TY_Nothing)
1003         ++NumOutputs;
1004
1005     if (NumOutputs > 1) {
1006       Diag(clang::diag::err_drv_output_argument_with_multiple_files);
1007       FinalOutput = 0;
1008     }
1009   }
1010
1011   for (ActionList::const_iterator it = C.getActions().begin(),
1012          ie = C.getActions().end(); it != ie; ++it) {
1013     Action *A = *it;
1014
1015     // If we are linking an image for multiple archs then the linker wants
1016     // -arch_multiple and -final_output <final image name>. Unfortunately, this
1017     // doesn't fit in cleanly because we have to pass this information down.
1018     //
1019     // FIXME: This is a hack; find a cleaner way to integrate this into the
1020     // process.
1021     const char *LinkingOutput = 0;
1022     if (isa<LipoJobAction>(A)) {
1023       if (FinalOutput)
1024         LinkingOutput = FinalOutput->getValue(C.getArgs());
1025       else
1026         LinkingOutput = DefaultImageName.c_str();
1027     }
1028
1029     InputInfo II;
1030     BuildJobsForAction(C, A, &C.getDefaultToolChain(),
1031                        /*BoundArch*/0,
1032                        /*AtTopLevel*/ true,
1033                        /*LinkingOutput*/ LinkingOutput,
1034                        II);
1035   }
1036
1037   // If the user passed -Qunused-arguments or there were errors, don't warn
1038   // about any unused arguments.
1039   if (Diags.hasErrorOccurred() ||
1040       C.getArgs().hasArg(options::OPT_Qunused_arguments))
1041     return;
1042
1043   // Claim -### here.
1044   (void) C.getArgs().hasArg(options::OPT__HASH_HASH_HASH);
1045
1046   for (ArgList::const_iterator it = C.getArgs().begin(), ie = C.getArgs().end();
1047        it != ie; ++it) {
1048     Arg *A = *it;
1049
1050     // FIXME: It would be nice to be able to send the argument to the
1051     // Diagnostic, so that extra values, position, and so on could be printed.
1052     if (!A->isClaimed()) {
1053       if (A->getOption().hasNoArgumentUnused())
1054         continue;
1055
1056       // Suppress the warning automatically if this is just a flag, and it is an
1057       // instance of an argument we already claimed.
1058       const Option &Opt = A->getOption();
1059       if (isa<FlagOption>(Opt)) {
1060         bool DuplicateClaimed = false;
1061
1062         for (arg_iterator it = C.getArgs().filtered_begin(&Opt),
1063                ie = C.getArgs().filtered_end(); it != ie; ++it) {
1064           if ((*it)->isClaimed()) {
1065             DuplicateClaimed = true;
1066             break;
1067           }
1068         }
1069
1070         if (DuplicateClaimed)
1071           continue;
1072       }
1073
1074       Diag(clang::diag::warn_drv_unused_argument)
1075         << A->getAsString(C.getArgs());
1076     }
1077   }
1078 }
1079
1080 static const Tool &SelectToolForJob(Compilation &C, const ToolChain *TC,
1081                                     const JobAction *JA,
1082                                     const ActionList *&Inputs) {
1083   const Tool *ToolForJob = 0;
1084
1085   // See if we should look for a compiler with an integrated assembler. We match
1086   // bottom up, so what we are actually looking for is an assembler job with a
1087   // compiler input.
1088
1089   // FIXME: This doesn't belong here, but ideally we will support static soon
1090   // anyway.
1091   bool HasStatic = (C.getArgs().hasArg(options::OPT_mkernel) ||
1092                     C.getArgs().hasArg(options::OPT_static) ||
1093                     C.getArgs().hasArg(options::OPT_fapple_kext));
1094   bool IsDarwin = TC->getTriple().getOS() == llvm::Triple::Darwin;
1095   bool IsIADefault = TC->IsIntegratedAssemblerDefault() &&
1096     !(HasStatic && IsDarwin);
1097   if (C.getArgs().hasFlag(options::OPT_integrated_as,
1098                          options::OPT_no_integrated_as,
1099                          IsIADefault) &&
1100       !C.getArgs().hasArg(options::OPT_save_temps) &&
1101       isa<AssembleJobAction>(JA) &&
1102       Inputs->size() == 1 && isa<CompileJobAction>(*Inputs->begin())) {
1103     const Tool &Compiler = TC->SelectTool(
1104       C, cast<JobAction>(**Inputs->begin()), (*Inputs)[0]->getInputs());
1105     if (Compiler.hasIntegratedAssembler()) {
1106       Inputs = &(*Inputs)[0]->getInputs();
1107       ToolForJob = &Compiler;
1108     }
1109   }
1110
1111   // Otherwise use the tool for the current job.
1112   if (!ToolForJob)
1113     ToolForJob = &TC->SelectTool(C, *JA, *Inputs);
1114
1115   // See if we should use an integrated preprocessor. We do so when we have
1116   // exactly one input, since this is the only use case we care about
1117   // (irrelevant since we don't support combine yet).
1118   if (Inputs->size() == 1 && isa<PreprocessJobAction>(*Inputs->begin()) &&
1119       !C.getArgs().hasArg(options::OPT_no_integrated_cpp) &&
1120       !C.getArgs().hasArg(options::OPT_traditional_cpp) &&
1121       !C.getArgs().hasArg(options::OPT_save_temps) &&
1122       ToolForJob->hasIntegratedCPP())
1123     Inputs = &(*Inputs)[0]->getInputs();
1124
1125   return *ToolForJob;
1126 }
1127
1128 void Driver::BuildJobsForAction(Compilation &C,
1129                                 const Action *A,
1130                                 const ToolChain *TC,
1131                                 const char *BoundArch,
1132                                 bool AtTopLevel,
1133                                 const char *LinkingOutput,
1134                                 InputInfo &Result) const {
1135   llvm::PrettyStackTraceString CrashInfo("Building compilation jobs");
1136
1137   if (const InputAction *IA = dyn_cast<InputAction>(A)) {
1138     // FIXME: It would be nice to not claim this here; maybe the old scheme of
1139     // just using Args was better?
1140     const Arg &Input = IA->getInputArg();
1141     Input.claim();
1142     if (Input.getOption().matches(options::OPT_INPUT)) {
1143       const char *Name = Input.getValue(C.getArgs());
1144       Result = InputInfo(Name, A->getType(), Name);
1145     } else
1146       Result = InputInfo(&Input, A->getType(), "");
1147     return;
1148   }
1149
1150   if (const BindArchAction *BAA = dyn_cast<BindArchAction>(A)) {
1151     const ToolChain *TC = &C.getDefaultToolChain();
1152
1153     std::string Arch;
1154     if (BAA->getArchName())
1155       TC = Host->CreateToolChain(C.getArgs(), BAA->getArchName());
1156
1157     BuildJobsForAction(C, *BAA->begin(), TC, BAA->getArchName(),
1158                        AtTopLevel, LinkingOutput, Result);
1159     return;
1160   }
1161
1162   const ActionList *Inputs = &A->getInputs();
1163
1164   const JobAction *JA = cast<JobAction>(A);
1165   const Tool &T = SelectToolForJob(C, TC, JA, Inputs);
1166
1167   // Only use pipes when there is exactly one input.
1168   InputInfoList InputInfos;
1169   for (ActionList::const_iterator it = Inputs->begin(), ie = Inputs->end();
1170        it != ie; ++it) {
1171     // Treat dsymutil sub-jobs as being at the top-level too, they shouldn't get
1172     // temporary output names.
1173     //
1174     // FIXME: Clean this up.
1175     bool SubJobAtTopLevel = false;
1176     if (AtTopLevel && isa<DsymutilJobAction>(A))
1177       SubJobAtTopLevel = true;
1178
1179     InputInfo II;
1180     BuildJobsForAction(C, *it, TC, BoundArch,
1181                        SubJobAtTopLevel, LinkingOutput, II);
1182     InputInfos.push_back(II);
1183   }
1184
1185   // Always use the first input as the base input.
1186   const char *BaseInput = InputInfos[0].getBaseInput();
1187
1188   // ... except dsymutil actions, which use their actual input as the base
1189   // input.
1190   if (JA->getType() == types::TY_dSYM)
1191     BaseInput = InputInfos[0].getFilename();
1192
1193   // Determine the place to write output to, if any.
1194   if (JA->getType() == types::TY_Nothing) {
1195     Result = InputInfo(A->getType(), BaseInput);
1196   } else {
1197     Result = InputInfo(GetNamedOutputPath(C, *JA, BaseInput, AtTopLevel),
1198                        A->getType(), BaseInput);
1199   }
1200
1201   if (CCCPrintBindings) {
1202     llvm::errs() << "# \"" << T.getToolChain().getTripleString() << '"'
1203                  << " - \"" << T.getName() << "\", inputs: [";
1204     for (unsigned i = 0, e = InputInfos.size(); i != e; ++i) {
1205       llvm::errs() << InputInfos[i].getAsString();
1206       if (i + 1 != e)
1207         llvm::errs() << ", ";
1208     }
1209     llvm::errs() << "], output: " << Result.getAsString() << "\n";
1210   } else {
1211     T.ConstructJob(C, *JA, Result, InputInfos,
1212                    C.getArgsForToolChain(TC, BoundArch), LinkingOutput);
1213   }
1214 }
1215
1216 const char *Driver::GetNamedOutputPath(Compilation &C,
1217                                        const JobAction &JA,
1218                                        const char *BaseInput,
1219                                        bool AtTopLevel) const {
1220   llvm::PrettyStackTraceString CrashInfo("Computing output path");
1221   // Output to a user requested destination?
1222   if (AtTopLevel && !isa<DsymutilJobAction>(JA)) {
1223     if (Arg *FinalOutput = C.getArgs().getLastArg(options::OPT_o))
1224       return C.addResultFile(FinalOutput->getValue(C.getArgs()));
1225   }
1226
1227   // Default to writing to stdout?
1228   if (AtTopLevel && isa<PreprocessJobAction>(JA))
1229     return "-";
1230
1231   // Output to a temporary file?
1232   if (!AtTopLevel && !C.getArgs().hasArg(options::OPT_save_temps)) {
1233     std::string TmpName =
1234       GetTemporaryPath(types::getTypeTempSuffix(JA.getType()));
1235     return C.addTempFile(C.getArgs().MakeArgString(TmpName.c_str()));
1236   }
1237
1238   llvm::SmallString<128> BasePath(BaseInput);
1239   llvm::StringRef BaseName;
1240
1241   // Dsymutil actions should use the full path.
1242   if (isa<DsymutilJobAction>(JA))
1243     BaseName = BasePath;
1244   else
1245     BaseName = llvm::sys::path::filename(BasePath);
1246
1247   // Determine what the derived output name should be.
1248   const char *NamedOutput;
1249   if (JA.getType() == types::TY_Image) {
1250     NamedOutput = DefaultImageName.c_str();
1251   } else {
1252     const char *Suffix = types::getTypeTempSuffix(JA.getType());
1253     assert(Suffix && "All types used for output should have a suffix.");
1254
1255     std::string::size_type End = std::string::npos;
1256     if (!types::appendSuffixForType(JA.getType()))
1257       End = BaseName.rfind('.');
1258     std::string Suffixed(BaseName.substr(0, End));
1259     Suffixed += '.';
1260     Suffixed += Suffix;
1261     NamedOutput = C.getArgs().MakeArgString(Suffixed.c_str());
1262   }
1263
1264   // If we're saving temps and the temp filename conflicts with the input
1265   // filename, then avoid overwriting input file.
1266   if (!AtTopLevel && C.getArgs().hasArg(options::OPT_save_temps) &&
1267     NamedOutput == BaseName) {
1268     std::string TmpName =
1269       GetTemporaryPath(types::getTypeTempSuffix(JA.getType()));
1270     return C.addTempFile(C.getArgs().MakeArgString(TmpName.c_str()));
1271   }
1272
1273   // As an annoying special case, PCH generation doesn't strip the pathname.
1274   if (JA.getType() == types::TY_PCH) {
1275     llvm::sys::path::remove_filename(BasePath);
1276     if (BasePath.empty())
1277       BasePath = NamedOutput;
1278     else
1279       llvm::sys::path::append(BasePath, NamedOutput);
1280     return C.addResultFile(C.getArgs().MakeArgString(BasePath.c_str()));
1281   } else {
1282     return C.addResultFile(NamedOutput);
1283   }
1284 }
1285
1286 std::string Driver::GetFilePath(const char *Name, const ToolChain &TC) const {
1287   // Respect a limited subset of the '-Bprefix' functionality in GCC by
1288   // attempting to use this prefix when lokup up program paths.
1289   for (Driver::prefix_list::const_iterator it = PrefixDirs.begin(),
1290        ie = PrefixDirs.end(); it != ie; ++it) {
1291     std::string Dir(*it);
1292     if (Dir.empty())
1293       continue;
1294     if (Dir[0] == '=')
1295       Dir = SysRoot + Dir.substr(1);
1296     llvm::sys::Path P(Dir);
1297     P.appendComponent(Name);
1298     bool Exists;
1299     if (!llvm::sys::fs::exists(P.str(), Exists) && Exists)
1300       return P.str();
1301   }
1302
1303   const ToolChain::path_list &List = TC.getFilePaths();
1304   for (ToolChain::path_list::const_iterator
1305          it = List.begin(), ie = List.end(); it != ie; ++it) {
1306     std::string Dir(*it);
1307     if (Dir.empty())
1308       continue;
1309     if (Dir[0] == '=')
1310       Dir = SysRoot + Dir.substr(1);
1311     llvm::sys::Path P(Dir);
1312     P.appendComponent(Name);
1313     bool Exists;
1314     if (!llvm::sys::fs::exists(P.str(), Exists) && Exists)
1315       return P.str();
1316   }
1317
1318   return Name;
1319 }
1320
1321 std::string Driver::GetProgramPath(const char *Name, const ToolChain &TC,
1322                                    bool WantFile) const {
1323   // Respect a limited subset of the '-Bprefix' functionality in GCC by
1324   // attempting to use this prefix when lokup up program paths.
1325   for (Driver::prefix_list::const_iterator it = PrefixDirs.begin(),
1326        ie = PrefixDirs.end(); it != ie; ++it) {
1327     llvm::sys::Path P(*it);
1328     P.appendComponent(Name);
1329     bool Exists;
1330     if (WantFile ? !llvm::sys::fs::exists(P.str(), Exists) && Exists
1331                  : P.canExecute())
1332       return P.str();
1333   }
1334
1335   const ToolChain::path_list &List = TC.getProgramPaths();
1336   for (ToolChain::path_list::const_iterator
1337          it = List.begin(), ie = List.end(); it != ie; ++it) {
1338     llvm::sys::Path P(*it);
1339     P.appendComponent(Name);
1340     bool Exists;
1341     if (WantFile ? !llvm::sys::fs::exists(P.str(), Exists) && Exists
1342                  : P.canExecute())
1343       return P.str();
1344   }
1345
1346   // If all else failed, search the path.
1347   llvm::sys::Path P(llvm::sys::Program::FindProgramByName(Name));
1348   if (!P.empty())
1349     return P.str();
1350
1351   return Name;
1352 }
1353
1354 std::string Driver::GetTemporaryPath(const char *Suffix) const {
1355   // FIXME: This is lame; sys::Path should provide this function (in particular,
1356   // it should know how to find the temporary files dir).
1357   std::string Error;
1358   const char *TmpDir = ::getenv("TMPDIR");
1359   if (!TmpDir)
1360     TmpDir = ::getenv("TEMP");
1361   if (!TmpDir)
1362     TmpDir = ::getenv("TMP");
1363   if (!TmpDir)
1364     TmpDir = "/tmp";
1365   llvm::sys::Path P(TmpDir);
1366   P.appendComponent("cc");
1367   if (P.makeUnique(false, &Error)) {
1368     Diag(clang::diag::err_drv_unable_to_make_temp) << Error;
1369     return "";
1370   }
1371
1372   // FIXME: Grumble, makeUnique sometimes leaves the file around!?  PR3837.
1373   P.eraseFromDisk(false, 0);
1374
1375   P.appendSuffix(Suffix);
1376   return P.str();
1377 }
1378
1379 const HostInfo *Driver::GetHostInfo(const char *TripleStr) const {
1380   llvm::PrettyStackTraceString CrashInfo("Constructing host");
1381   llvm::Triple Triple(llvm::Triple::normalize(TripleStr).c_str());
1382
1383   // TCE is an osless target
1384   if (Triple.getArchName() == "tce")
1385     return createTCEHostInfo(*this, Triple);
1386
1387   switch (Triple.getOS()) {
1388   case llvm::Triple::AuroraUX:
1389     return createAuroraUXHostInfo(*this, Triple);
1390   case llvm::Triple::Darwin:
1391     return createDarwinHostInfo(*this, Triple);
1392   case llvm::Triple::DragonFly:
1393     return createDragonFlyHostInfo(*this, Triple);
1394   case llvm::Triple::OpenBSD:
1395     return createOpenBSDHostInfo(*this, Triple);
1396   case llvm::Triple::NetBSD:
1397     return createNetBSDHostInfo(*this, Triple);
1398   case llvm::Triple::FreeBSD:
1399     return createFreeBSDHostInfo(*this, Triple);
1400   case llvm::Triple::Minix:
1401     return createMinixHostInfo(*this, Triple);
1402   case llvm::Triple::Linux:
1403     return createLinuxHostInfo(*this, Triple);
1404   case llvm::Triple::Win32:
1405     return createWindowsHostInfo(*this, Triple);
1406   case llvm::Triple::MinGW32:
1407     return createMinGWHostInfo(*this, Triple);
1408   default:
1409     return createUnknownHostInfo(*this, Triple);
1410   }
1411 }
1412
1413 bool Driver::ShouldUseClangCompiler(const Compilation &C, const JobAction &JA,
1414                                     const llvm::Triple &Triple) const {
1415   // Check if user requested no clang, or clang doesn't understand this type (we
1416   // only handle single inputs for now).
1417   if (!CCCUseClang || JA.size() != 1 ||
1418       !types::isAcceptedByClang((*JA.begin())->getType()))
1419     return false;
1420
1421   // Otherwise make sure this is an action clang understands.
1422   if (isa<PreprocessJobAction>(JA)) {
1423     if (!CCCUseClangCPP) {
1424       Diag(clang::diag::warn_drv_not_using_clang_cpp);
1425       return false;
1426     }
1427   } else if (!isa<PrecompileJobAction>(JA) && !isa<CompileJobAction>(JA))
1428     return false;
1429
1430   // Use clang for C++?
1431   if (!CCCUseClangCXX && types::isCXX((*JA.begin())->getType())) {
1432     Diag(clang::diag::warn_drv_not_using_clang_cxx);
1433     return false;
1434   }
1435
1436   // Always use clang for precompiling, AST generation, and rewriting,
1437   // regardless of archs.
1438   if (isa<PrecompileJobAction>(JA) ||
1439       types::isOnlyAcceptedByClang(JA.getType()))
1440     return true;
1441
1442   // Finally, don't use clang if this isn't one of the user specified archs to
1443   // build.
1444   if (!CCCClangArchs.empty() && !CCCClangArchs.count(Triple.getArch())) {
1445     Diag(clang::diag::warn_drv_not_using_clang_arch) << Triple.getArchName();
1446     return false;
1447   }
1448
1449   return true;
1450 }
1451
1452 /// GetReleaseVersion - Parse (([0-9]+)(.([0-9]+)(.([0-9]+)?))?)? and return the
1453 /// grouped values as integers. Numbers which are not provided are set to 0.
1454 ///
1455 /// \return True if the entire string was parsed (9.2), or all groups were
1456 /// parsed (10.3.5extrastuff).
1457 bool Driver::GetReleaseVersion(const char *Str, unsigned &Major,
1458                                unsigned &Minor, unsigned &Micro,
1459                                bool &HadExtra) {
1460   HadExtra = false;
1461
1462   Major = Minor = Micro = 0;
1463   if (*Str == '\0')
1464     return true;
1465
1466   char *End;
1467   Major = (unsigned) strtol(Str, &End, 10);
1468   if (*Str != '\0' && *End == '\0')
1469     return true;
1470   if (*End != '.')
1471     return false;
1472
1473   Str = End+1;
1474   Minor = (unsigned) strtol(Str, &End, 10);
1475   if (*Str != '\0' && *End == '\0')
1476     return true;
1477   if (*End != '.')
1478     return false;
1479
1480   Str = End+1;
1481   Micro = (unsigned) strtol(Str, &End, 10);
1482   if (*Str != '\0' && *End == '\0')
1483     return true;
1484   if (Str == End)
1485     return false;
1486   HadExtra = true;
1487   return true;
1488 }