CompilerInstance.cpp revision 998b3d3e8528ebd9d2c5d78d3a82edd90a8953a4
12a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar//===--- CompilerInstance.cpp ---------------------------------------------===//
22a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar//
32a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar//                     The LLVM Compiler Infrastructure
42a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar//
52a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar// This file is distributed under the University of Illinois Open Source
62a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar// License. See LICENSE.TXT for details.
72a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar//
82a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar//===----------------------------------------------------------------------===//
92a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar
102a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar#include "clang/Frontend/CompilerInstance.h"
11f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregor#include "clang/Sema/Sema.h"
1212ce6943aae499225708ecf364c5a8b0a3269c87Daniel Dunbar#include "clang/AST/ASTConsumer.h"
135eb810024dc8a1d12d5f066c02c978f07c4fcb00Daniel Dunbar#include "clang/AST/ASTContext.h"
142a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar#include "clang/Basic/Diagnostic.h"
1516b7449d86b843d0926b04f87104cf3fff7149feDaniel Dunbar#include "clang/Basic/FileManager.h"
1616b7449d86b843d0926b04f87104cf3fff7149feDaniel Dunbar#include "clang/Basic/SourceManager.h"
172a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar#include "clang/Basic/TargetInfo.h"
180397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar#include "clang/Basic/Version.h"
1922dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar#include "clang/Lex/HeaderSearch.h"
2022dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar#include "clang/Lex/Preprocessor.h"
2122dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar#include "clang/Lex/PTHManager.h"
220fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar#include "clang/Frontend/ChainedDiagnosticClient.h"
230397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar#include "clang/Frontend/FrontendAction.h"
24c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar#include "clang/Frontend/FrontendDiagnostic.h"
259df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar#include "clang/Frontend/LogDiagnosticPrinter.h"
260fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar#include "clang/Frontend/TextDiagnosticPrinter.h"
27f79dced038c63572965c016b969cfa870670d16eDaniel Dunbar#include "clang/Frontend/VerifyDiagnosticsClient.h"
2822dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar#include "clang/Frontend/Utils.h"
296ab7cd853e9c15cf986a8a7c3db1f8d20e275409Sebastian Redl#include "clang/Serialization/ASTReader.h"
30c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar#include "clang/Sema/CodeCompleteConsumer.h"
3132bef4edba854303800b3b01cb49a282e5da4f69Michael J. Spencer#include "llvm/Support/FileSystem.h"
32ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar#include "llvm/Support/MemoryBuffer.h"
330fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar#include "llvm/Support/raw_ostream.h"
3495dd5583e3900443a1d7970d557d8c54aa320601Douglas Gregor#include "llvm/ADT/Statistic.h"
35f79bafa608a5d7c49ec40ad199af5e32f3038b47Kovarththanan Rajaratnam#include "llvm/Support/Timer.h"
3603013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Host.h"
3703013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Path.h"
3803013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Program.h"
3903013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Signals.h"
403a321e23f66128dbb986343927456ff6702af617Michael J. Spencer#include "llvm/Support/system_error.h"
411626601b00a1feccd6b7a3f54e2991f56b659192Dylan Noblesmith#include "llvm/Config/config.h"
422a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbarusing namespace clang;
432a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar
4442e9f8e4983d50f896ec716207817b9d96e7e79cDaniel DunbarCompilerInstance::CompilerInstance()
45f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor  : Invocation(new CompilerInvocation()), ModuleManager(0) {
466228ca00121669ec06a19df4fad87d5049c097cfDaniel Dunbar}
472a79e162a3fde25c1941151a67966830d873f2dbDaniel Dunbar
482a79e162a3fde25c1941151a67966830d873f2dbDaniel DunbarCompilerInstance::~CompilerInstance() {
4942e9f8e4983d50f896ec716207817b9d96e7e79cDaniel Dunbar}
5042e9f8e4983d50f896ec716207817b9d96e7e79cDaniel Dunbar
516228ca00121669ec06a19df4fad87d5049c097cfDaniel Dunbarvoid CompilerInstance::setInvocation(CompilerInvocation *Value) {
524f32786ac45210143654390177105eb749b614e9Ted Kremenek  Invocation = Value;
536228ca00121669ec06a19df4fad87d5049c097cfDaniel Dunbar}
546228ca00121669ec06a19df4fad87d5049c097cfDaniel Dunbar
558a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbarvoid CompilerInstance::setDiagnostics(Diagnostic *Value) {
5628019772db70d4547be05a042eb950bc910f134fDouglas Gregor  Diagnostics = Value;
578a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar}
588a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar
598a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbarvoid CompilerInstance::setTarget(TargetInfo *Value) {
604f32786ac45210143654390177105eb749b614e9Ted Kremenek  Target = Value;
618a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar}
628a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar
638a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbarvoid CompilerInstance::setFileManager(FileManager *Value) {
644f32786ac45210143654390177105eb749b614e9Ted Kremenek  FileMgr = Value;
658a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar}
668a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar
674f32786ac45210143654390177105eb749b614e9Ted Kremenekvoid CompilerInstance::setSourceManager(SourceManager *Value) {
684f32786ac45210143654390177105eb749b614e9Ted Kremenek  SourceMgr = Value;
698a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar}
708a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar
714f32786ac45210143654390177105eb749b614e9Ted Kremenekvoid CompilerInstance::setPreprocessor(Preprocessor *Value) { PP = Value; }
728a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar
734f32786ac45210143654390177105eb749b614e9Ted Kremenekvoid CompilerInstance::setASTContext(ASTContext *Value) { Context = Value; }
7412ce6943aae499225708ecf364c5a8b0a3269c87Daniel Dunbar
75f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregorvoid CompilerInstance::setSema(Sema *S) {
76f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregor  TheSema.reset(S);
77f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregor}
78f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregor
7912ce6943aae499225708ecf364c5a8b0a3269c87Daniel Dunbarvoid CompilerInstance::setASTConsumer(ASTConsumer *Value) {
8012ce6943aae499225708ecf364c5a8b0a3269c87Daniel Dunbar  Consumer.reset(Value);
818a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar}
828a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar
838a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbarvoid CompilerInstance::setCodeCompletionConsumer(CodeCompleteConsumer *Value) {
848a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar  CompletionConsumer.reset(Value);
858a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar}
868a9f569262860b8d03203327afd6047be2a9b5a6Daniel Dunbar
870fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar// Diagnostics
880fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbarstatic void SetUpBuildDumpLog(const DiagnosticOptions &DiagOpts,
897d0c4ccd65b4549283c55e4923602e234f3811c5Axel Naumann                              unsigned argc, const char* const *argv,
903d67b1e25847319a5a271f9d5a8d607ef18d804aKovarththanan Rajaratnam                              Diagnostic &Diags) {
910fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  std::string ErrorInfo;
925f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  llvm::OwningPtr<raw_ostream> OS(
9369247139f5ce3edf46c48344129792b3a05d090aKovarththanan Rajaratnam    new llvm::raw_fd_ostream(DiagOpts.DumpBuildInformation.c_str(), ErrorInfo));
940fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  if (!ErrorInfo.empty()) {
953d67b1e25847319a5a271f9d5a8d607ef18d804aKovarththanan Rajaratnam    Diags.Report(diag::err_fe_unable_to_open_logfile)
963d67b1e25847319a5a271f9d5a8d607ef18d804aKovarththanan Rajaratnam                 << DiagOpts.DumpBuildInformation << ErrorInfo;
970fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar    return;
980fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  }
990fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
100dd63b28107f21692b5065588f0e90b4534946f93Daniel Dunbar  (*OS) << "clang -cc1 command line arguments: ";
1010fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  for (unsigned i = 0; i != argc; ++i)
1020fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar    (*OS) << argv[i] << ' ';
1030fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  (*OS) << '\n';
1040fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
1050fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  // Chain in a diagnostic client which will log the diagnostics.
1060fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  DiagnosticClient *Logger =
10769247139f5ce3edf46c48344129792b3a05d090aKovarththanan Rajaratnam    new TextDiagnosticPrinter(*OS.take(), DiagOpts, /*OwnsOutputStream=*/true);
108bdbb004f38978da0c4a75af3294d1c7b5ff84af1Douglas Gregor  Diags.setClient(new ChainedDiagnosticClient(Diags.takeClient(), Logger));
1090fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar}
1100fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
1119df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbarstatic void SetUpDiagnosticLog(const DiagnosticOptions &DiagOpts,
112b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar                               const CodeGenOptions *CodeGenOpts,
1139df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar                               Diagnostic &Diags) {
1149df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar  std::string ErrorInfo;
1159df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar  bool OwnsStream = false;
1165f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  raw_ostream *OS = &llvm::errs();
1179df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar  if (DiagOpts.DiagnosticLogFile != "-") {
1189df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar    // Create the output stream.
1199df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar    llvm::raw_fd_ostream *FileOS(
1209df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar      new llvm::raw_fd_ostream(DiagOpts.DiagnosticLogFile.c_str(),
121e01eceb54c9235a41675bfd9f32c50bd005fb8d0Daniel Dunbar                               ErrorInfo, llvm::raw_fd_ostream::F_Append));
1229df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar    if (!ErrorInfo.empty()) {
1239df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar      Diags.Report(diag::warn_fe_cc_log_diagnostics_failure)
1249df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar        << DiagOpts.DumpBuildInformation << ErrorInfo;
1259df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar    } else {
1269df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar      FileOS->SetUnbuffered();
1279df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar      FileOS->SetUseAtomicWrites(true);
1289df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar      OS = FileOS;
1299df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar      OwnsStream = true;
1309df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar    }
1319df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar  }
1329df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar
1339df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar  // Chain in the diagnostic client which will log the diagnostics.
134b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar  LogDiagnosticPrinter *Logger = new LogDiagnosticPrinter(*OS, DiagOpts,
135b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar                                                          OwnsStream);
136b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar  if (CodeGenOpts)
137b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar    Logger->setDwarfDebugFlags(CodeGenOpts->DwarfDebugFlags);
1389df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar  Diags.setClient(new ChainedDiagnosticClient(Diags.takeClient(), Logger));
1399df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar}
1409df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar
141e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregorvoid CompilerInstance::createDiagnostics(int Argc, const char* const *Argv,
142e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                         DiagnosticClient *Client) {
143b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar  Diagnostics = createDiagnostics(getDiagnosticOpts(), Argc, Argv, Client,
144b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar                                  &getCodeGenOpts());
1450fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar}
1460fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
14728019772db70d4547be05a042eb950bc910f134fDouglas Gregorllvm::IntrusiveRefCntPtr<Diagnostic>
14828019772db70d4547be05a042eb950bc910f134fDouglas GregorCompilerInstance::createDiagnostics(const DiagnosticOptions &Opts,
149e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                    int Argc, const char* const *Argv,
150b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar                                    DiagnosticClient *Client,
151b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar                                    const CodeGenOptions *CodeGenOpts) {
15233e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  llvm::IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
15333e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  llvm::IntrusiveRefCntPtr<Diagnostic> Diags(new Diagnostic(DiagID));
154221c7211c507482a91e97ede1bf6cf65a456ff67Daniel Dunbar
1550fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  // Create the diagnostic client for reporting errors or for
1560fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  // implementing -verify.
157e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor  if (Client)
158e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor    Diags->setClient(Client);
159e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor  else
160e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor    Diags->setClient(new TextDiagnosticPrinter(llvm::errs(), Opts));
161f79dced038c63572965c016b969cfa870670d16eDaniel Dunbar
162f79dced038c63572965c016b969cfa870670d16eDaniel Dunbar  // Chain in -verify checker, if requested.
163f79dced038c63572965c016b969cfa870670d16eDaniel Dunbar  if (Opts.VerifyDiagnostics)
164bdbb004f38978da0c4a75af3294d1c7b5ff84af1Douglas Gregor    Diags->setClient(new VerifyDiagnosticsClient(*Diags, Diags->takeClient()));
1650fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
1669df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar  // Chain in -diagnostic-log-file dumper, if requested.
1679df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar  if (!Opts.DiagnosticLogFile.empty())
168b6534bbee90bf73f364072051d10b60352d43c3eDaniel Dunbar    SetUpDiagnosticLog(Opts, CodeGenOpts, *Diags);
1699df23493f5b8a223dfbc491e4b7de3850797c2e7Daniel Dunbar
1700fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  if (!Opts.DumpBuildInformation.empty())
1713d67b1e25847319a5a271f9d5a8d607ef18d804aKovarththanan Rajaratnam    SetUpBuildDumpLog(Opts, Argc, Argv, *Diags);
1720fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
1730fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar  // Configure our handling of diagnostics.
1745bf932bd0d5db98273938675dbf81cbb2f5ffff7Kovarththanan Rajaratnam  ProcessWarningOptions(*Diags, Opts);
1750fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
17628019772db70d4547be05a042eb950bc910f134fDouglas Gregor  return Diags;
1770fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar}
1780fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
1790fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar// File Manager
1800fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
18116b7449d86b843d0926b04f87104cf3fff7149feDaniel Dunbarvoid CompilerInstance::createFileManager() {
1824f32786ac45210143654390177105eb749b614e9Ted Kremenek  FileMgr = new FileManager(getFileSystemOpts());
18316b7449d86b843d0926b04f87104cf3fff7149feDaniel Dunbar}
18416b7449d86b843d0926b04f87104cf3fff7149feDaniel Dunbar
1850fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar// Source Manager
1860fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
18739b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattnervoid CompilerInstance::createSourceManager(FileManager &FileMgr) {
1884f32786ac45210143654390177105eb749b614e9Ted Kremenek  SourceMgr = new SourceManager(getDiagnostics(), FileMgr);
18916b7449d86b843d0926b04f87104cf3fff7149feDaniel Dunbar}
19022dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar
1910fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar// Preprocessor
1920fbb3d9a9cdd2201848be9eb017c54cd78538122Daniel Dunbar
19322dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbarvoid CompilerInstance::createPreprocessor() {
1946aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  const PreprocessorOptions &PPOpts = getPreprocessorOpts();
1956aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
19622dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  // Create a PTH manager if we are using some form of a token cache.
19722dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  PTHManager *PTHMgr = 0;
198049d3a06ea9f8fc03582488a2b7f24512565a335Daniel Dunbar  if (!PPOpts.TokenCache.empty())
1996aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    PTHMgr = PTHManager::Create(PPOpts.TokenCache, getDiagnostics());
2006aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
20122dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  // Create the Preprocessor.
2026aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  HeaderSearch *HeaderInfo = new HeaderSearch(getFileManager());
203998b3d3e8528ebd9d2c5d78d3a82edd90a8953a4Douglas Gregor  PP = new Preprocessor(getDiagnostics(), getLangOpts(), &getTarget(),
2046aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                        getSourceManager(), *HeaderInfo, *this, PTHMgr,
2056aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                        /*OwnsHeaderSearch=*/true);
2066aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
20722dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  // Note that this is different then passing PTHMgr to Preprocessor's ctor.
20822dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  // That argument is used as the IdentifierInfoLookup argument to
20922dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  // IdentifierTable's ctor.
21022dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  if (PTHMgr) {
2116aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    PTHMgr->setPreprocessor(&*PP);
21222dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar    PP->setPTHManager(PTHMgr);
21322dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  }
2146aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
21594dc8f640ebea52241412512ed48601626edbc58Douglas Gregor  if (PPOpts.DetailedRecord)
216dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor    PP->createPreprocessingRecord(
2176aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                  PPOpts.DetailedRecordIncludesNestedMacroExpansions);
2186aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
2196aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  InitializePreprocessor(*PP, PPOpts, getHeaderSearchOpts(), getFrontendOpts());
22094dc8f640ebea52241412512ed48601626edbc58Douglas Gregor
22122dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  // Handle generating dependencies, if requested.
2226aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  const DependencyOutputOptions &DepOpts = getDependencyOutputOpts();
22322dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar  if (!DepOpts.OutputFile.empty())
22422dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar    AttachDependencyFileGen(*PP, DepOpts);
2256aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
226eef63e0997e0f6d6436736ea919b851cfe34955aDaniel Dunbar  // Handle generating header include information, if requested.
227eef63e0997e0f6d6436736ea919b851cfe34955aDaniel Dunbar  if (DepOpts.ShowHeaderIncludes)
228eef63e0997e0f6d6436736ea919b851cfe34955aDaniel Dunbar    AttachHeaderIncludeGen(*PP);
229b34d69b9292534c1c574f168f0ac10aea652adcaDaniel Dunbar  if (!DepOpts.HeaderIncludeOutputFile.empty()) {
2305f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef OutputPath = DepOpts.HeaderIncludeOutputFile;
231b34d69b9292534c1c574f168f0ac10aea652adcaDaniel Dunbar    if (OutputPath == "-")
232b34d69b9292534c1c574f168f0ac10aea652adcaDaniel Dunbar      OutputPath = "";
233da60885cb4ee85d167cf704061f10b758067cf5aDaniel Dunbar    AttachHeaderIncludeGen(*PP, /*ShowAllHeaders=*/true, OutputPath,
234da60885cb4ee85d167cf704061f10b758067cf5aDaniel Dunbar                           /*ShowDepth=*/false);
235b34d69b9292534c1c574f168f0ac10aea652adcaDaniel Dunbar  }
23622dacfacacf5559028550ba6ddfbaa4ea6cb3944Daniel Dunbar}
2375eb810024dc8a1d12d5f066c02c978f07c4fcb00Daniel Dunbar
2385eb810024dc8a1d12d5f066c02c978f07c4fcb00Daniel Dunbar// ASTContext
2395eb810024dc8a1d12d5f066c02c978f07c4fcb00Daniel Dunbar
2405eb810024dc8a1d12d5f066c02c978f07c4fcb00Daniel Dunbarvoid CompilerInstance::createASTContext() {
2415eb810024dc8a1d12d5f066c02c978f07c4fcb00Daniel Dunbar  Preprocessor &PP = getPreprocessor();
2424f32786ac45210143654390177105eb749b614e9Ted Kremenek  Context = new ASTContext(getLangOpts(), PP.getSourceManager(),
2434f32786ac45210143654390177105eb749b614e9Ted Kremenek                           getTarget(), PP.getIdentifierTable(),
2444f32786ac45210143654390177105eb749b614e9Ted Kremenek                           PP.getSelectorTable(), PP.getBuiltinInfo(),
2454f32786ac45210143654390177105eb749b614e9Ted Kremenek                           /*size_reserve=*/ 0);
2465eb810024dc8a1d12d5f066c02c978f07c4fcb00Daniel Dunbar}
2470f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar
2480f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar// ExternalASTSource
2490f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar
2505f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnervoid CompilerInstance::createPCHExternalASTSource(StringRef Path,
251ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl                                                  bool DisablePCHValidation,
2528ef6c8cb6c5627240e2339fd7062c9873f821d7eDouglas Gregor                                                  bool DisableStatCache,
253ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl                                                 void *DeserializationListener){
2540f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar  llvm::OwningPtr<ExternalASTSource> Source;
2551d9f1fe7173e3084325f43c78af812a36d8a2a7cSebastian Redl  bool Preamble = getPreprocessorOpts().PrecompiledPreambleBytes.first != 0;
2560f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar  Source.reset(createPCHExternalASTSource(Path, getHeaderSearchOpts().Sysroot,
2578ef6c8cb6c5627240e2339fd7062c9873f821d7eDouglas Gregor                                          DisablePCHValidation,
2588ef6c8cb6c5627240e2339fd7062c9873f821d7eDouglas Gregor                                          DisableStatCache,
259ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl                                          getPreprocessor(), getASTContext(),
2601d9f1fe7173e3084325f43c78af812a36d8a2a7cSebastian Redl                                          DeserializationListener,
2611d9f1fe7173e3084325f43c78af812a36d8a2a7cSebastian Redl                                          Preamble));
262f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor  ModuleManager = static_cast<ASTReader*>(Source.get());
2630f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar  getASTContext().setExternalSource(Source);
2640f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar}
2650f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar
2660f800391ffbfe3820e1c60246a09a97e5f065179Daniel DunbarExternalASTSource *
2675f9e272e632e951b1efe824cd16acb4d96077930Chris LattnerCompilerInstance::createPCHExternalASTSource(StringRef Path,
2680f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar                                             const std::string &Sysroot,
269fae3b2f4743dad616623c4df2fdb0f5128bd36d9Douglas Gregor                                             bool DisablePCHValidation,
2708ef6c8cb6c5627240e2339fd7062c9873f821d7eDouglas Gregor                                             bool DisableStatCache,
2710f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar                                             Preprocessor &PP,
272ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl                                             ASTContext &Context,
2731d9f1fe7173e3084325f43c78af812a36d8a2a7cSebastian Redl                                             void *DeserializationListener,
2741d9f1fe7173e3084325f43c78af812a36d8a2a7cSebastian Redl                                             bool Preamble) {
275c43b54cbc10654ed59de797898042e1a05265246Sebastian Redl  llvm::OwningPtr<ASTReader> Reader;
276c43b54cbc10654ed59de797898042e1a05265246Sebastian Redl  Reader.reset(new ASTReader(PP, &Context,
277832d620b4ae0fc5fe28561b885b4cfc65cf5c9abDouglas Gregor                             Sysroot.empty() ? "" : Sysroot.c_str(),
2788ef6c8cb6c5627240e2339fd7062c9873f821d7eDouglas Gregor                             DisablePCHValidation, DisableStatCache));
2790f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar
280ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl  Reader->setDeserializationListener(
281571db7f0cb31789737be92fce1c1b738e6dbe795Sebastian Redl            static_cast<ASTDeserializationListener *>(DeserializationListener));
28272a9ae18553bf8b6bdad84d2c54f73741a47e275Douglas Gregor  switch (Reader->ReadAST(Path,
28372a9ae18553bf8b6bdad84d2c54f73741a47e275Douglas Gregor                          Preamble ? serialization::MK_Preamble
28472a9ae18553bf8b6bdad84d2c54f73741a47e275Douglas Gregor                                   : serialization::MK_PCH)) {
285c43b54cbc10654ed59de797898042e1a05265246Sebastian Redl  case ASTReader::Success:
2860f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar    // Set the predefines buffer as suggested by the PCH reader. Typically, the
2870f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar    // predefines buffer will be empty.
2880f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar    PP.setPredefines(Reader->getSuggestedPredefines());
2890f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar    return Reader.take();
2900f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar
291c43b54cbc10654ed59de797898042e1a05265246Sebastian Redl  case ASTReader::Failure:
2920f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar    // Unrecoverable failure: don't even try to process the input file.
2930f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar    break;
2940f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar
295c43b54cbc10654ed59de797898042e1a05265246Sebastian Redl  case ASTReader::IgnorePCH:
2960f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar    // No suitable PCH file could be found. Return an error.
2970f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar    break;
2980f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar  }
2990f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar
3000f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar  return 0;
3010f800391ffbfe3820e1c60246a09a97e5f065179Daniel Dunbar}
302c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar
303c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar// Code Completion
304c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar
3051abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregorstatic bool EnableCodeCompletion(Preprocessor &PP,
3061abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                                 const std::string &Filename,
3071abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                                 unsigned Line,
3081abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                                 unsigned Column) {
3091abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  // Tell the source manager to chop off the given file at a specific
3101abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  // line and column.
31139b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  const FileEntry *Entry = PP.getFileManager().getFile(Filename);
3121abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  if (!Entry) {
3131abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor    PP.getDiagnostics().Report(diag::err_fe_invalid_code_complete_file)
3141abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      << Filename;
3151abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor    return true;
3161abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  }
3171abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor
3181abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  // Truncate the named file at the given line/column.
3191abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  PP.SetCodeCompletionPoint(Entry, Line, Column);
3201abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  return false;
3211abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor}
3221abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor
323c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbarvoid CompilerInstance::createCodeCompletionConsumer() {
324c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar  const ParsedSourceLocation &Loc = getFrontendOpts().CodeCompletionAt;
3251abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  if (!CompletionConsumer) {
3261abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor    CompletionConsumer.reset(
3271abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      createCodeCompletionConsumer(getPreprocessor(),
3281abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                                   Loc.FileName, Loc.Line, Loc.Column,
3291abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                                   getFrontendOpts().ShowMacrosInCodeCompletion,
330d8e8a58ee35ab334ab9d0c2154dca029c1822e8aDouglas Gregor                             getFrontendOpts().ShowCodePatternsInCodeCompletion,
3318071e4212ae08f8014e0c3ae6d18b7388003a5ccDouglas Gregor                           getFrontendOpts().ShowGlobalSymbolsInCodeCompletion,
3321abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                                   llvm::outs()));
3331abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor    if (!CompletionConsumer)
3341abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      return;
3351abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  } else if (EnableCodeCompletion(getPreprocessor(), Loc.FileName,
3361abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                                  Loc.Line, Loc.Column)) {
3371abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor    CompletionConsumer.reset();
338c3d43b783dfb1a1502aa8b31ab1985cf237b1f77Douglas Gregor    return;
3391abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  }
3402b4074f1d3919e77cb33ca49c960521dea27afabDouglas Gregor
3412b4074f1d3919e77cb33ca49c960521dea27afabDouglas Gregor  if (CompletionConsumer->isOutputBinary() &&
3422b4074f1d3919e77cb33ca49c960521dea27afabDouglas Gregor      llvm::sys::Program::ChangeStdoutToBinary()) {
3432b4074f1d3919e77cb33ca49c960521dea27afabDouglas Gregor    getPreprocessor().getDiagnostics().Report(diag::err_fe_stdout_binary);
3442b4074f1d3919e77cb33ca49c960521dea27afabDouglas Gregor    CompletionConsumer.reset();
3452b4074f1d3919e77cb33ca49c960521dea27afabDouglas Gregor  }
346c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar}
347c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar
348f79bafa608a5d7c49ec40ad199af5e32f3038b47Kovarththanan Rajaratnamvoid CompilerInstance::createFrontendTimer() {
349f79bafa608a5d7c49ec40ad199af5e32f3038b47Kovarththanan Rajaratnam  FrontendTimer.reset(new llvm::Timer("Clang front-end timer"));
350f79bafa608a5d7c49ec40ad199af5e32f3038b47Kovarththanan Rajaratnam}
351f79bafa608a5d7c49ec40ad199af5e32f3038b47Kovarththanan Rajaratnam
352c2f484f1f05216a9a427ac84b5773789a4661111Daniel DunbarCodeCompleteConsumer *
353c2f484f1f05216a9a427ac84b5773789a4661111Daniel DunbarCompilerInstance::createCodeCompletionConsumer(Preprocessor &PP,
354c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar                                               const std::string &Filename,
355c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar                                               unsigned Line,
356c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar                                               unsigned Column,
357c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar                                               bool ShowMacros,
358d8e8a58ee35ab334ab9d0c2154dca029c1822e8aDouglas Gregor                                               bool ShowCodePatterns,
3598071e4212ae08f8014e0c3ae6d18b7388003a5ccDouglas Gregor                                               bool ShowGlobals,
3605f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner                                               raw_ostream &OS) {
3611abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  if (EnableCodeCompletion(PP, Filename, Line, Column))
362c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar    return 0;
363c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar
364c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar  // Set up the creation routine for code-completion.
365a9f4f620daf073805b89e893afcdc5eb7a9bdc50Douglas Gregor  return new PrintingCodeCompleteConsumer(ShowMacros, ShowCodePatterns,
3668071e4212ae08f8014e0c3ae6d18b7388003a5ccDouglas Gregor                                          ShowGlobals, OS);
367c2f484f1f05216a9a427ac84b5773789a4661111Daniel Dunbar}
368a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar
369467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregorvoid CompilerInstance::createSema(TranslationUnitKind TUKind,
370f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregor                                  CodeCompleteConsumer *CompletionConsumer) {
371f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregor  TheSema.reset(new Sema(getPreprocessor(), getASTContext(), getASTConsumer(),
372467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor                         TUKind, CompletionConsumer));
373f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregor}
374f18d0d8b39e891460d50f8a8b85029885b264986Douglas Gregor
375a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar// Output Files
376a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar
377dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidisvoid CompilerInstance::addOutputFile(const OutputFile &OutFile) {
378dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  assert(OutFile.OS && "Attempt to add empty stream to output list!");
379dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  OutputFiles.push_back(OutFile);
380a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar}
381a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar
382e51dd7be67808d52c80c09b832d875e9655ce6e0Kovarththanan Rajaratnamvoid CompilerInstance::clearOutputFiles(bool EraseFiles) {
383dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  for (std::list<OutputFile>::iterator
384a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar         it = OutputFiles.begin(), ie = OutputFiles.end(); it != ie; ++it) {
385dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis    delete it->OS;
386dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis    if (!it->TempFilename.empty()) {
387af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson      if (EraseFiles) {
388af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson        bool existed;
389af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson        llvm::sys::fs::remove(it->TempFilename, existed);
390af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson      } else {
391af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson        llvm::SmallString<128> NewOutFile(it->Filename);
392af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson
393389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis        // If '-working-directory' was passed, the output filename should be
394389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis        // relative to that.
3952e2468e2d2ccbb9a38fe3b6b754009af7e5d39a9Anders Carlsson        FileMgr->FixupRelativePath(NewOutFile);
396af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson        if (llvm::error_code ec = llvm::sys::fs::rename(it->TempFilename,
397af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson                                                        NewOutFile.str())) {
398dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis          getDiagnostics().Report(diag::err_fe_unable_to_rename_temp)
399af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson            << it->TempFilename << it->Filename << ec.message();
400af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson
401af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson          bool existed;
402af036a6c25f540a9daf51e721485ae6d0e645bd8Anders Carlsson          llvm::sys::fs::remove(it->TempFilename, existed);
403dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis        }
404dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis      }
405dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis    } else if (!it->Filename.empty() && EraseFiles)
406dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis      llvm::sys::Path(it->Filename).eraseFromDisk();
407dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis
408a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar  }
409a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar  OutputFiles.clear();
410a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar}
411a9204831639e31474b927681b97c46781b758a1aDaniel Dunbar
412f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbarllvm::raw_fd_ostream *
413f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel DunbarCompilerInstance::createDefaultOutputFile(bool Binary,
4145f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner                                          StringRef InFile,
4155f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner                                          StringRef Extension) {
416f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  return createOutputFile(getFrontendOpts().OutputFile, Binary,
417ff9cd968cd5b623e3ec7e5f862b598cd22f7ec79Daniel Dunbar                          /*RemoveFileOnSignal=*/true, InFile, Extension);
418f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar}
419f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar
420f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbarllvm::raw_fd_ostream *
4215f9e272e632e951b1efe824cd16acb4d96077930Chris LattnerCompilerInstance::createOutputFile(StringRef OutputPath,
422ff9cd968cd5b623e3ec7e5f862b598cd22f7ec79Daniel Dunbar                                   bool Binary, bool RemoveFileOnSignal,
4235f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner                                   StringRef InFile,
4247e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis                                   StringRef Extension,
4257e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis                                   bool UseTemporary) {
426dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  std::string Error, OutputPathName, TempPathName;
427f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  llvm::raw_fd_ostream *OS = createOutputFile(OutputPath, Error, Binary,
428ff9cd968cd5b623e3ec7e5f862b598cd22f7ec79Daniel Dunbar                                              RemoveFileOnSignal,
429f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar                                              InFile, Extension,
4307e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis                                              UseTemporary,
431dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis                                              &OutputPathName,
432dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis                                              &TempPathName);
433f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  if (!OS) {
434360435908c9b90429cfe192fab22854af1d4497cDaniel Dunbar    getDiagnostics().Report(diag::err_fe_unable_to_open_output)
435360435908c9b90429cfe192fab22854af1d4497cDaniel Dunbar      << OutputPath << Error;
436360435908c9b90429cfe192fab22854af1d4497cDaniel Dunbar    return 0;
437f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  }
438f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar
439f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  // Add the output file -- but don't try to remove "-", since this means we are
440f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  // using stdin.
441dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  addOutputFile(OutputFile((OutputPathName != "-") ? OutputPathName : "",
442dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis                TempPathName, OS));
443f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar
444f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  return OS;
445f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar}
446f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar
447f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbarllvm::raw_fd_ostream *
4485f9e272e632e951b1efe824cd16acb4d96077930Chris LattnerCompilerInstance::createOutputFile(StringRef OutputPath,
449f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar                                   std::string &Error,
450f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar                                   bool Binary,
451ff9cd968cd5b623e3ec7e5f862b598cd22f7ec79Daniel Dunbar                                   bool RemoveFileOnSignal,
4525f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner                                   StringRef InFile,
4535f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner                                   StringRef Extension,
4547e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis                                   bool UseTemporary,
455dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis                                   std::string *ResultPathName,
456dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis                                   std::string *TempPathName) {
457dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  std::string OutFile, TempFile;
458f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  if (!OutputPath.empty()) {
459f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar    OutFile = OutputPath;
460f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  } else if (InFile == "-") {
461f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar    OutFile = "-";
462f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  } else if (!Extension.empty()) {
463f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar    llvm::sys::Path Path(InFile);
464f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar    Path.eraseSuffix();
465f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar    Path.appendSuffix(Extension);
466f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar    OutFile = Path.str();
467f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  } else {
468f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar    OutFile = "-";
469f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  }
4707e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis
4717e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis  llvm::OwningPtr<llvm::raw_fd_ostream> OS;
4727e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis  std::string OSFile;
4737e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis
4747e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis  if (UseTemporary && OutFile != "-") {
475dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis    llvm::sys::Path OutPath(OutFile);
476dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis    // Only create the temporary if we can actually write to OutPath, otherwise
477dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis    // we want to fail early.
47832bef4edba854303800b3b01cb49a282e5da4f69Michael J. Spencer    bool Exists;
47932bef4edba854303800b3b01cb49a282e5da4f69Michael J. Spencer    if ((llvm::sys::fs::exists(OutPath.str(), Exists) || !Exists) ||
480dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis        (OutPath.isRegularFile() && OutPath.canWrite())) {
481dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis      // Create a temporary file.
4827e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis      llvm::SmallString<128> TempPath;
4837e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis      TempPath = OutFile;
4847e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis      TempPath += "-%%%%%%%%";
4857e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis      int fd;
4867e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis      if (llvm::sys::fs::unique_file(TempPath.str(), fd, TempPath,
4877e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis                               /*makeAbsolute=*/false) == llvm::errc::success) {
4887e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis        OS.reset(new llvm::raw_fd_ostream(fd, /*shouldClose=*/true));
4897e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis        OSFile = TempFile = TempPath.str();
4907e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis      }
491dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis    }
492dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  }
493dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis
4947e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis  if (!OS) {
4957e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis    OSFile = OutFile;
4967e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis    OS.reset(
4977e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis      new llvm::raw_fd_ostream(OSFile.c_str(), Error,
4987e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis                               (Binary ? llvm::raw_fd_ostream::F_Binary : 0)));
4997e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis    if (!Error.empty())
5007e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis      return 0;
5017e90985df09855dc309ed888a5b16a0ae684f8e3Argyrios Kyrtzidis  }
502f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar
503dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  // Make sure the out stream file gets removed if we crash.
504ff9cd968cd5b623e3ec7e5f862b598cd22f7ec79Daniel Dunbar  if (RemoveFileOnSignal)
505ff9cd968cd5b623e3ec7e5f862b598cd22f7ec79Daniel Dunbar    llvm::sys::RemoveFileOnSignal(llvm::sys::Path(OSFile));
506dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis
507f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar  if (ResultPathName)
508f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar    *ResultPathName = OutFile;
509dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis  if (TempPathName)
510dc24572a44575e07a5d8bb6de52641a69f1bab27Argyrios Kyrtzidis    *TempPathName = TempFile;
511f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar
512fc97102a80bfe0afaa25883a2aa6b5e1d7307d0aDaniel Dunbar  return OS.take();
513f482d59386dbc70716f7a5f65adb37ff86b501e6Daniel Dunbar}
514ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar
515ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar// Initialization Utilities
516ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar
5175f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnerbool CompilerInstance::InitializeSourceManager(StringRef InputFile) {
518ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar  return InitializeSourceManager(InputFile, getDiagnostics(), getFileManager(),
519ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar                                 getSourceManager(), getFrontendOpts());
520ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar}
521ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar
5225f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnerbool CompilerInstance::InitializeSourceManager(StringRef InputFile,
523ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar                                               Diagnostic &Diags,
524ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar                                               FileManager &FileMgr,
525ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar                                               SourceManager &SourceMgr,
526ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar                                               const FrontendOptions &Opts) {
527414cb64f09ce48a36377458ce5e5a90c3ad41d00Douglas Gregor  // Figure out where to get and map in the main file, unless it's already
528414cb64f09ce48a36377458ce5e5a90c3ad41d00Douglas Gregor  // been created (e.g., by a precompiled preamble).
529414cb64f09ce48a36377458ce5e5a90c3ad41d00Douglas Gregor  if (!SourceMgr.getMainFileID().isInvalid()) {
530414cb64f09ce48a36377458ce5e5a90c3ad41d00Douglas Gregor    // Do nothing: the main file has already been set.
531414cb64f09ce48a36377458ce5e5a90c3ad41d00Douglas Gregor  } else if (InputFile != "-") {
53239b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner    const FileEntry *File = FileMgr.getFile(InputFile);
533694137c54c79a33c9ac6c07e68327750dcd5adf7Dan Gohman    if (!File) {
534ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar      Diags.Report(diag::err_fe_error_reading) << InputFile;
535ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar      return false;
536ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar    }
537694137c54c79a33c9ac6c07e68327750dcd5adf7Dan Gohman    SourceMgr.createMainFileID(File);
538ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar  } else {
5394eeebc464e1f968d9968a4786c82558f18ac2ed8Michael J. Spencer    llvm::OwningPtr<llvm::MemoryBuffer> SB;
5404eeebc464e1f968d9968a4786c82558f18ac2ed8Michael J. Spencer    if (llvm::MemoryBuffer::getSTDIN(SB)) {
5413a321e23f66128dbb986343927456ff6702af617Michael J. Spencer      // FIXME: Give ec.message() in this diag.
542ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar      Diags.Report(diag::err_fe_error_reading_stdin);
543ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar      return false;
544ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar    }
54590d9081cacb4b0163f2c7527f666d6515257067cDan Gohman    const FileEntry *File = FileMgr.getVirtualFile(SB->getBufferIdentifier(),
54639b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner                                                   SB->getBufferSize(), 0);
54790d9081cacb4b0163f2c7527f666d6515257067cDan Gohman    SourceMgr.createMainFileID(File);
5484eeebc464e1f968d9968a4786c82558f18ac2ed8Michael J. Spencer    SourceMgr.overrideFileContents(File, SB.take());
549ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar  }
550ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar
551694137c54c79a33c9ac6c07e68327750dcd5adf7Dan Gohman  assert(!SourceMgr.getMainFileID().isInvalid() &&
552694137c54c79a33c9ac6c07e68327750dcd5adf7Dan Gohman         "Couldn't establish MainFileID!");
553ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar  return true;
554ccb6cb6fd9e48697564d536b07397b95dfc28d5bDaniel Dunbar}
5550397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
5560397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar// High-Level Operations
5570397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
5580397af277e3bba16da1fd125ddba07415686b429Daniel Dunbarbool CompilerInstance::ExecuteAction(FrontendAction &Act) {
5590397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  assert(hasDiagnostics() && "Diagnostics engine is not initialized!");
5600397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  assert(!getFrontendOpts().ShowHelp && "Client must handle '-help'!");
5610397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  assert(!getFrontendOpts().ShowVersion && "Client must handle '-version'!");
5620397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
5630397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  // FIXME: Take this as an argument, once all the APIs we used have moved to
5640397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  // taking it as an input instead of hard-coding llvm::errs.
5655f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  raw_ostream &OS = llvm::errs();
5660397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
5670397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  // Create the target instance.
5680397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  setTarget(TargetInfo::CreateTargetInfo(getDiagnostics(), getTargetOpts()));
5690397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  if (!hasTarget())
5700397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar    return false;
5710397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
5720397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  // Inform the target of the language options.
5730397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  //
5740397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  // FIXME: We shouldn't need to do this, the target should be immutable once
5750397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  // created. This complexity should be lifted elsewhere.
5760397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  getTarget().setForcedLangOptions(getLangOpts());
5770397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
5780397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  // Validate/process some options.
5790397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  if (getHeaderSearchOpts().Verbose)
5800397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar    OS << "clang -cc1 version " CLANG_VERSION_STRING
5810397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar       << " based upon " << PACKAGE_STRING
5820397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar       << " hosted on " << llvm::sys::getHostTriple() << "\n";
5830397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
5840397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  if (getFrontendOpts().ShowTimers)
5850397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar    createFrontendTimer();
5860397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
58795dd5583e3900443a1d7970d557d8c54aa320601Douglas Gregor  if (getFrontendOpts().ShowStats)
58895dd5583e3900443a1d7970d557d8c54aa320601Douglas Gregor    llvm::EnableStatistics();
58995dd5583e3900443a1d7970d557d8c54aa320601Douglas Gregor
5900397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  for (unsigned i = 0, e = getFrontendOpts().Inputs.size(); i != e; ++i) {
5910397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar    const std::string &InFile = getFrontendOpts().Inputs[i].second;
5920397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
5932056048f0f619adadc9a5416a2c4cdf95c58eef7Daniel Dunbar    // Reset the ID tables if we are reusing the SourceManager.
5942056048f0f619adadc9a5416a2c4cdf95c58eef7Daniel Dunbar    if (hasSourceManager())
5952056048f0f619adadc9a5416a2c4cdf95c58eef7Daniel Dunbar      getSourceManager().clearIDTables();
5960397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
597d3598a65716e120aef45aa2841d730e03f7101feDaniel Dunbar    if (Act.BeginSourceFile(*this, InFile, getFrontendOpts().Inputs[i].first)) {
5980397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar      Act.Execute();
5990397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar      Act.EndSourceFile();
6000397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar    }
6010397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  }
6020397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
60353eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner  if (getDiagnosticOpts().ShowCarets) {
604f2224d89a6ae65a3839529e26d0f6d025d83d6bbArgyrios Kyrtzidis    // We can have multiple diagnostics sharing one diagnostic client.
605f2224d89a6ae65a3839529e26d0f6d025d83d6bbArgyrios Kyrtzidis    // Get the total number of warnings/errors from the client.
606f2224d89a6ae65a3839529e26d0f6d025d83d6bbArgyrios Kyrtzidis    unsigned NumWarnings = getDiagnostics().getClient()->getNumWarnings();
607f2224d89a6ae65a3839529e26d0f6d025d83d6bbArgyrios Kyrtzidis    unsigned NumErrors = getDiagnostics().getClient()->getNumErrors();
60853eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner
60953eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner    if (NumWarnings)
61053eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner      OS << NumWarnings << " warning" << (NumWarnings == 1 ? "" : "s");
61153eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner    if (NumWarnings && NumErrors)
61253eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner      OS << " and ";
61353eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner    if (NumErrors)
61453eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner      OS << NumErrors << " error" << (NumErrors == 1 ? "" : "s");
61553eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner    if (NumWarnings || NumErrors)
61653eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner      OS << " generated.\n";
61753eee7ba970d21ff15bbd4334164037a3b4cc4b8Chris Lattner  }
6180397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
6192056048f0f619adadc9a5416a2c4cdf95c58eef7Daniel Dunbar  if (getFrontendOpts().ShowStats && hasFileManager()) {
6200397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar    getFileManager().PrintStats();
6210397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar    OS << "\n";
6220397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar  }
6230397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
624ab41b97c3cddf1bcdd8bf82ab09ed3ceafcd05b1Argyrios Kyrtzidis  return !getDiagnostics().getClient()->getNumErrors();
6250397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar}
6260397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
6276aa52ec6b969faabf3764baf79d89810b8249a7eDouglas GregorModuleKey CompilerInstance::loadModule(SourceLocation ImportLoc,
6286aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                       IdentifierInfo &ModuleName,
6296aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                       SourceLocation ModuleNameLoc) {
6306aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  // Determine what file we're searching from.
6316aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  SourceManager &SourceMgr = getSourceManager();
6326aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  SourceLocation ExpandedImportLoc = SourceMgr.getExpansionLoc(ImportLoc);
6336aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  const FileEntry *CurFile
6346aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    = SourceMgr.getFileEntryForID(SourceMgr.getFileID(ExpandedImportLoc));
6356aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  if (!CurFile)
6366aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    CurFile = SourceMgr.getFileEntryForID(SourceMgr.getMainFileID());
6376aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
6386aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  // Search for a module with the given name.
6396aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  std::string Filename = ModuleName.getName().str();
6406aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  Filename += ".pcm";
6416aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  const DirectoryLookup *CurDir = 0;
6426aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  const FileEntry *ModuleFile
6436aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    = PP->getHeaderSearchInfo().LookupFile(Filename, /*isAngled=*/false,
6446aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                           /*FromDir=*/0, CurDir, CurFile,
6456aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                           /*SearchPath=*/0,
6466aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                           /*RelativePath=*/0);
6476aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  if (!ModuleFile) {
648e082af17d4b425a49f7f8bccc2a99810f0072828Douglas Gregor    getDiagnostics().Report(ModuleNameLoc, diag::err_module_not_found)
6496aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor      << ModuleName.getName()
6506aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor      << SourceRange(ImportLoc, ModuleNameLoc);
6516aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    return 0;
6526aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  }
6536aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
6546aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  // If we don't already have an ASTReader, create one now.
6556aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  if (!ModuleManager) {
6566aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    std::string Sysroot = getHeaderSearchOpts().Sysroot;
6576aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    const PreprocessorOptions &PPOpts = getPreprocessorOpts();
6586aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    ModuleManager = new ASTReader(getPreprocessor(), &*Context,
6596aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                  Sysroot.empty() ? "" : Sysroot.c_str(),
6606aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                  PPOpts.DisablePCHValidation,
6616aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                  PPOpts.DisableStatCache);
6626aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    ModuleManager->setDeserializationListener(
6636aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor      getASTConsumer().GetASTDeserializationListener());
6646aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    getASTContext().setASTMutationListener(
6656aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor      getASTConsumer().GetASTMutationListener());
6666aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    llvm::OwningPtr<ExternalASTSource> Source;
6676aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    Source.reset(ModuleManager);
6686aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    getASTContext().setExternalSource(Source);
6696aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    ModuleManager->InitializeSema(getSema());
6706aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  }
6716aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
6726aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  // Try to load the module we found.
6736aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  switch (ModuleManager->ReadAST(ModuleFile->getName(),
6746aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor                                 serialization::MK_Module)) {
6756aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  case ASTReader::Success:
6766aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    break;
6776aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
6786aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  case ASTReader::IgnorePCH:
6796aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    // FIXME: The ASTReader will already have complained, but can we showhorn
6806aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    // that diagnostic information into a more useful form?
6816aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    return 0;
6826aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
6836aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  case ASTReader::Failure:
6846aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    // Already complained.
6856aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor    return 0;
6866aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  }
6876aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor
6886aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  // FIXME: The module file's FileEntry makes a poor key indeed!
6896aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor  return (ModuleKey)ModuleFile;
6906aa52ec6b969faabf3764baf79d89810b8249a7eDouglas Gregor}
6910397af277e3bba16da1fd125ddba07415686b429Daniel Dunbar
692