contrib/llvm/tools/clang/lib/Frontend/CompilerInstance.cpp

   1 //===--- CompilerInstance.cpp ---------------------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "clang/Frontend/CompilerInstance.h"
  11 #include "clang/AST/ASTConsumer.h"
  12 #include "clang/AST/ASTContext.h"
  13 #include "clang/AST/Decl.h"
  14 #include "clang/Basic/Diagnostic.h"
  15 #include "clang/Basic/FileManager.h"
  16 #include "clang/Basic/SourceManager.h"
  17 #include "clang/Basic/TargetInfo.h"
  18 #include "clang/Basic/Version.h"
  19 #include "clang/Config/config.h"
  20 #include "clang/Frontend/ChainedDiagnosticConsumer.h"
  21 #include "clang/Frontend/FrontendAction.h"
  22 #include "clang/Frontend/FrontendActions.h"
  23 #include "clang/Frontend/FrontendDiagnostic.h"
  24 #include "clang/Frontend/LogDiagnosticPrinter.h"
  25 #include "clang/Frontend/SerializedDiagnosticPrinter.h"
  26 #include "clang/Frontend/TextDiagnosticPrinter.h"
  27 #include "clang/Frontend/Utils.h"
  28 #include "clang/Frontend/VerifyDiagnosticConsumer.h"
  29 #include "clang/Lex/HeaderSearch.h"
  30 #include "clang/Lex/PTHManager.h"
  31 #include "clang/Lex/Preprocessor.h"
  32 #include "clang/Sema/CodeCompleteConsumer.h"
  33 #include "clang/Sema/Sema.h"
  34 #include "clang/Serialization/ASTReader.h"
  35 #include "clang/Serialization/GlobalModuleIndex.h"
  36 #include "llvm/ADT/Statistic.h"
  37 #include "llvm/Support/CrashRecoveryContext.h"
  38 #include "llvm/Support/Errc.h"
  39 #include "llvm/Support/FileSystem.h"
  40 #include "llvm/Support/Host.h"
  41 #include "llvm/Support/LockFileManager.h"
  42 #include "llvm/Support/MemoryBuffer.h"
  43 #include "llvm/Support/Path.h"
  44 #include "llvm/Support/Program.h"
  45 #include "llvm/Support/Signals.h"
  46 #include "llvm/Support/Timer.h"
  47 #include "llvm/Support/raw_ostream.h"
  48 #include <sys/stat.h>
  49 #include <system_error>
  50 #include <time.h>
  51
  52 using namespace clang;
  53
  54 CompilerInstance::CompilerInstance(bool BuildingModule)
  55   : ModuleLoader(BuildingModule),
  56     Invocation(new CompilerInvocation()), ModuleManager(nullptr),
  57     BuildGlobalModuleIndex(false), HaveFullGlobalModuleIndex(false),
  58     ModuleBuildFailed(false) {
  59 }
  60
  61 CompilerInstance::~CompilerInstance() {
  62   assert(OutputFiles.empty() && "Still output files in flight?");
  63 }
  64
  65 void CompilerInstance::setInvocation(CompilerInvocation *Value) {
  66   Invocation = Value;
  67 }
  68
  69 bool CompilerInstance::shouldBuildGlobalModuleIndex() const {
  70   return (BuildGlobalModuleIndex ||
  71           (ModuleManager && ModuleManager->isGlobalIndexUnavailable() &&
  72            getFrontendOpts().GenerateGlobalModuleIndex)) &&
  73          !ModuleBuildFailed;
  74 }
  75
  76 void CompilerInstance::setDiagnostics(DiagnosticsEngine *Value) {
  77   Diagnostics = Value;
  78 }
  79
  80 void CompilerInstance::setTarget(TargetInfo *Value) {
  81   Target = Value;
  82 }
  83
  84 void CompilerInstance::setFileManager(FileManager *Value) {
  85   FileMgr = Value;
  86   if (Value)
  87     VirtualFileSystem = Value->getVirtualFileSystem();
  88   else
  89     VirtualFileSystem.reset();
  90 }
  91
  92 void CompilerInstance::setSourceManager(SourceManager *Value) {
  93   SourceMgr = Value;
  94 }
  95
  96 void CompilerInstance::setPreprocessor(Preprocessor *Value) { PP = Value; }
  97
  98 void CompilerInstance::setASTContext(ASTContext *Value) { Context = Value; }
  99
 100 void CompilerInstance::setSema(Sema *S) {
 101   TheSema.reset(S);
 102 }
 103
 104 void CompilerInstance::setASTConsumer(std::unique_ptr<ASTConsumer> Value) {
 105   Consumer = std::move(Value);
 106 }
 107
 108 void CompilerInstance::setCodeCompletionConsumer(CodeCompleteConsumer *Value) {
 109   CompletionConsumer.reset(Value);
 110 }
 111
 112 std::unique_ptr<Sema> CompilerInstance::takeSema() {
 113   return std::move(TheSema);
 114 }
 115
 116 IntrusiveRefCntPtr<ASTReader> CompilerInstance::getModuleManager() const {
 117   return ModuleManager;
 118 }
 119 void CompilerInstance::setModuleManager(IntrusiveRefCntPtr<ASTReader> Reader) {
 120   ModuleManager = Reader;
 121 }
 122
 123 std::shared_ptr<ModuleDependencyCollector>
 124 CompilerInstance::getModuleDepCollector() const {
 125   return ModuleDepCollector;
 126 }
 127
 128 void CompilerInstance::setModuleDepCollector(
 129     std::shared_ptr<ModuleDependencyCollector> Collector) {
 130   ModuleDepCollector = Collector;
 131 }
 132
 133 // Diagnostics
 134 static void SetUpDiagnosticLog(DiagnosticOptions *DiagOpts,
 135                                const CodeGenOptions *CodeGenOpts,
 136                                DiagnosticsEngine &Diags) {
 137   std::error_code EC;
 138   std::unique_ptr<raw_ostream> StreamOwner;
 139   raw_ostream *OS = &llvm::errs();
 140   if (DiagOpts->DiagnosticLogFile != "-") {
 141     // Create the output stream.
 142     auto FileOS = llvm::make_unique<llvm::raw_fd_ostream>(
 143         DiagOpts->DiagnosticLogFile, EC,
 144         llvm::sys::fs::F_Append | llvm::sys::fs::F_Text);
 145     if (EC) {
 146       Diags.Report(diag::warn_fe_cc_log_diagnostics_failure)
 147           << DiagOpts->DiagnosticLogFile << EC.message();
 148     } else {
 149       FileOS->SetUnbuffered();
 150       FileOS->SetUseAtomicWrites(true);
 151       OS = FileOS.get();
 152       StreamOwner = std::move(FileOS);
 153     }
 154   }
 155
 156   // Chain in the diagnostic client which will log the diagnostics.
 157   auto Logger = llvm::make_unique<LogDiagnosticPrinter>(*OS, DiagOpts,
 158                                                         std::move(StreamOwner));
 159   if (CodeGenOpts)
 160     Logger->setDwarfDebugFlags(CodeGenOpts->DwarfDebugFlags);
 161   assert(Diags.ownsClient());
 162   Diags.setClient(
 163       new ChainedDiagnosticConsumer(Diags.takeClient(), std::move(Logger)));
 164 }
 165
 166 static void SetupSerializedDiagnostics(DiagnosticOptions *DiagOpts,
 167                                        DiagnosticsEngine &Diags,
 168                                        StringRef OutputFile) {
 169   auto SerializedConsumer =
 170       clang::serialized_diags::create(OutputFile, DiagOpts);
 171
 172   if (Diags.ownsClient()) {
 173     Diags.setClient(new ChainedDiagnosticConsumer(
 174         Diags.takeClient(), std::move(SerializedConsumer)));
 175   } else {
 176     Diags.setClient(new ChainedDiagnosticConsumer(
 177         Diags.getClient(), std::move(SerializedConsumer)));
 178   }
 179 }
 180
 181 void CompilerInstance::createDiagnostics(DiagnosticConsumer *Client,
 182                                          bool ShouldOwnClient) {
 183   Diagnostics = createDiagnostics(&getDiagnosticOpts(), Client,
 184                                   ShouldOwnClient, &getCodeGenOpts());
 185 }
 186
 187 IntrusiveRefCntPtr<DiagnosticsEngine>
 188 CompilerInstance::createDiagnostics(DiagnosticOptions *Opts,
 189                                     DiagnosticConsumer *Client,
 190                                     bool ShouldOwnClient,
 191                                     const CodeGenOptions *CodeGenOpts) {
 192   IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
 193   IntrusiveRefCntPtr<DiagnosticsEngine>
 194       Diags(new DiagnosticsEngine(DiagID, Opts));
 195
 196   // Create the diagnostic client for reporting errors or for
 197   // implementing -verify.
 198   if (Client) {
 199     Diags->setClient(Client, ShouldOwnClient);
 200   } else
 201     Diags->setClient(new TextDiagnosticPrinter(llvm::errs(), Opts));
 202
 203   // Chain in -verify checker, if requested.
 204   if (Opts->VerifyDiagnostics)
 205     Diags->setClient(new VerifyDiagnosticConsumer(*Diags));
 206
 207   // Chain in -diagnostic-log-file dumper, if requested.
 208   if (!Opts->DiagnosticLogFile.empty())
 209     SetUpDiagnosticLog(Opts, CodeGenOpts, *Diags);
 210
 211   if (!Opts->DiagnosticSerializationFile.empty())
 212     SetupSerializedDiagnostics(Opts, *Diags,
 213                                Opts->DiagnosticSerializationFile);
 214
 215   // Configure our handling of diagnostics.
 216   ProcessWarningOptions(*Diags, *Opts);
 217
 218   return Diags;
 219 }
 220
 221 // File Manager
 222
 223 void CompilerInstance::createFileManager() {
 224   if (!hasVirtualFileSystem()) {
 225     // TODO: choose the virtual file system based on the CompilerInvocation.
 226     setVirtualFileSystem(vfs::getRealFileSystem());
 227   }
 228   FileMgr = new FileManager(getFileSystemOpts(), VirtualFileSystem);
 229 }
 230
 231 // Source Manager
 232
 233 void CompilerInstance::createSourceManager(FileManager &FileMgr) {
 234   SourceMgr = new SourceManager(getDiagnostics(), FileMgr);
 235 }
 236
 237 // Initialize the remapping of files to alternative contents, e.g.,
 238 // those specified through other files.
 239 static void InitializeFileRemapping(DiagnosticsEngine &Diags,
 240                                     SourceManager &SourceMgr,
 241                                     FileManager &FileMgr,
 242                                     const PreprocessorOptions &InitOpts) {
 243   // Remap files in the source manager (with buffers).
 244   for (const auto &RB : InitOpts.RemappedFileBuffers) {
 245     // Create the file entry for the file that we're mapping from.
 246     const FileEntry *FromFile =
 247         FileMgr.getVirtualFile(RB.first, RB.second->getBufferSize(), 0);
 248     if (!FromFile) {
 249       Diags.Report(diag::err_fe_remap_missing_from_file) << RB.first;
 250       if (!InitOpts.RetainRemappedFileBuffers)
 251         delete RB.second;
 252       continue;
 253     }
 254
 255     // Override the contents of the "from" file with the contents of
 256     // the "to" file.
 257     SourceMgr.overrideFileContents(FromFile, RB.second,
 258                                    InitOpts.RetainRemappedFileBuffers);
 259   }
 260
 261   // Remap files in the source manager (with other files).
 262   for (const auto &RF : InitOpts.RemappedFiles) {
 263     // Find the file that we're mapping to.
 264     const FileEntry *ToFile = FileMgr.getFile(RF.second);
 265     if (!ToFile) {
 266       Diags.Report(diag::err_fe_remap_missing_to_file) << RF.first << RF.second;
 267       continue;
 268     }
 269
 270     // Create the file entry for the file that we're mapping from.
 271     const FileEntry *FromFile =
 272         FileMgr.getVirtualFile(RF.first, ToFile->getSize(), 0);
 273     if (!FromFile) {
 274       Diags.Report(diag::err_fe_remap_missing_from_file) << RF.first;
 275       continue;
 276     }
 277
 278     // Override the contents of the "from" file with the contents of
 279     // the "to" file.
 280     SourceMgr.overrideFileContents(FromFile, ToFile);
 281   }
 282
 283   SourceMgr.setOverridenFilesKeepOriginalName(
 284       InitOpts.RemappedFilesKeepOriginalName);
 285 }
 286
 287 // Preprocessor
 288
 289 void CompilerInstance::createPreprocessor(TranslationUnitKind TUKind) {
 290   const PreprocessorOptions &PPOpts = getPreprocessorOpts();
 291
 292   // Create a PTH manager if we are using some form of a token cache.
 293   PTHManager *PTHMgr = nullptr;
 294   if (!PPOpts.TokenCache.empty())
 295     PTHMgr = PTHManager::Create(PPOpts.TokenCache, getDiagnostics());
 296
 297   // Create the Preprocessor.
 298   HeaderSearch *HeaderInfo = new HeaderSearch(&getHeaderSearchOpts(),
 299                                               getSourceManager(),
 300                                               getDiagnostics(),
 301                                               getLangOpts(),
 302                                               &getTarget());
 303   PP = new Preprocessor(&getPreprocessorOpts(), getDiagnostics(), getLangOpts(),
 304                         getSourceManager(), *HeaderInfo, *this, PTHMgr,
 305                         /*OwnsHeaderSearch=*/true, TUKind);
 306   PP->Initialize(getTarget());
 307
 308   // Note that this is different then passing PTHMgr to Preprocessor's ctor.
 309   // That argument is used as the IdentifierInfoLookup argument to
 310   // IdentifierTable's ctor.
 311   if (PTHMgr) {
 312     PTHMgr->setPreprocessor(&*PP);
 313     PP->setPTHManager(PTHMgr);
 314   }
 315
 316   if (PPOpts.DetailedRecord)
 317     PP->createPreprocessingRecord();
 318
 319   // Apply remappings to the source manager.
 320   InitializeFileRemapping(PP->getDiagnostics(), PP->getSourceManager(),
 321                           PP->getFileManager(), PPOpts);
 322
 323   // Predefine macros and configure the preprocessor.
 324   InitializePreprocessor(*PP, PPOpts, getFrontendOpts());
 325
 326   // Initialize the header search object.
 327   ApplyHeaderSearchOptions(PP->getHeaderSearchInfo(), getHeaderSearchOpts(),
 328                            PP->getLangOpts(), PP->getTargetInfo().getTriple());
 329
 330   PP->setPreprocessedOutput(getPreprocessorOutputOpts().ShowCPP);
 331
 332   if (PP->getLangOpts().Modules)
 333     PP->getHeaderSearchInfo().setModuleCachePath(getSpecificModuleCachePath());
 334
 335   // Handle generating dependencies, if requested.
 336   const DependencyOutputOptions &DepOpts = getDependencyOutputOpts();
 337   if (!DepOpts.OutputFile.empty())
 338     TheDependencyFileGenerator.reset(
 339         DependencyFileGenerator::CreateAndAttachToPreprocessor(*PP, DepOpts));
 340   if (!DepOpts.DOTOutputFile.empty())
 341     AttachDependencyGraphGen(*PP, DepOpts.DOTOutputFile,
 342                              getHeaderSearchOpts().Sysroot);
 343
 344   for (auto &Listener : DependencyCollectors)
 345     Listener->attachToPreprocessor(*PP);
 346
 347   // If we don't have a collector, but we are collecting module dependencies,
 348   // then we're the top level compiler instance and need to create one.
 349   if (!ModuleDepCollector && !DepOpts.ModuleDependencyOutputDir.empty())
 350     ModuleDepCollector = std::make_shared<ModuleDependencyCollector>(
 351         DepOpts.ModuleDependencyOutputDir);
 352
 353   // Handle generating header include information, if requested.
 354   if (DepOpts.ShowHeaderIncludes)
 355     AttachHeaderIncludeGen(*PP);
 356   if (!DepOpts.HeaderIncludeOutputFile.empty()) {
 357     StringRef OutputPath = DepOpts.HeaderIncludeOutputFile;
 358     if (OutputPath == "-")
 359       OutputPath = "";
 360     AttachHeaderIncludeGen(*PP, /*ShowAllHeaders=*/true, OutputPath,
 361                            /*ShowDepth=*/false);
 362   }
 363
 364   if (DepOpts.PrintShowIncludes) {
 365     AttachHeaderIncludeGen(*PP, /*ShowAllHeaders=*/false, /*OutputPath=*/"",
 366                            /*ShowDepth=*/true, /*MSStyle=*/true);
 367   }
 368 }
 369
 370 std::string CompilerInstance::getSpecificModuleCachePath() {
 371   // Set up the module path, including the hash for the
 372   // module-creation options.
 373   SmallString<256> SpecificModuleCache(
 374                            getHeaderSearchOpts().ModuleCachePath);
 375   if (!getHeaderSearchOpts().DisableModuleHash)
 376     llvm::sys::path::append(SpecificModuleCache,
 377                             getInvocation().getModuleHash());
 378   return SpecificModuleCache.str();
 379 }
 380
 381 // ASTContext
 382
 383 void CompilerInstance::createASTContext() {
 384   Preprocessor &PP = getPreprocessor();
 385   Context = new ASTContext(getLangOpts(), PP.getSourceManager(),
 386                            PP.getIdentifierTable(), PP.getSelectorTable(),
 387                            PP.getBuiltinInfo());
 388   Context->InitBuiltinTypes(getTarget());
 389 }
 390
 391 // ExternalASTSource
 392
 393 void CompilerInstance::createPCHExternalASTSource(
 394     StringRef Path, bool DisablePCHValidation, bool AllowPCHWithCompilerErrors,
 395     void *DeserializationListener, bool OwnDeserializationListener) {
 396   bool Preamble = getPreprocessorOpts().PrecompiledPreambleBytes.first != 0;
 397   ModuleManager = createPCHExternalASTSource(
 398       Path, getHeaderSearchOpts().Sysroot, DisablePCHValidation,
 399       AllowPCHWithCompilerErrors, getPreprocessor(), getASTContext(),
 400       DeserializationListener, OwnDeserializationListener, Preamble,
 401       getFrontendOpts().UseGlobalModuleIndex);
 402 }
 403
 404 IntrusiveRefCntPtr<ASTReader> CompilerInstance::createPCHExternalASTSource(
 405     StringRef Path, const std::string &Sysroot, bool DisablePCHValidation,
 406     bool AllowPCHWithCompilerErrors, Preprocessor &PP, ASTContext &Context,
 407     void *DeserializationListener, bool OwnDeserializationListener,
 408     bool Preamble, bool UseGlobalModuleIndex) {
 409   HeaderSearchOptions &HSOpts = PP.getHeaderSearchInfo().getHeaderSearchOpts();
 410
 411   IntrusiveRefCntPtr<ASTReader> Reader(
 412       new ASTReader(PP, Context, Sysroot.empty() ? "" : Sysroot.c_str(),
 413                     DisablePCHValidation, AllowPCHWithCompilerErrors,
 414                     /*AllowConfigurationMismatch*/ false,
 415                     HSOpts.ModulesValidateSystemHeaders, UseGlobalModuleIndex));
 416
 417   // We need the external source to be set up before we read the AST, because
 418   // eagerly-deserialized declarations may use it.
 419   Context.setExternalSource(Reader.get());
 420
 421   Reader->setDeserializationListener(
 422       static_cast<ASTDeserializationListener *>(DeserializationListener),
 423       /*TakeOwnership=*/OwnDeserializationListener);
 424   switch (Reader->ReadAST(Path,
 425                           Preamble ? serialization::MK_Preamble
 426                                    : serialization::MK_PCH,
 427                           SourceLocation(),
 428                           ASTReader::ARR_None)) {
 429   case ASTReader::Success:
 430     // Set the predefines buffer as suggested by the PCH reader. Typically, the
 431     // predefines buffer will be empty.
 432     PP.setPredefines(Reader->getSuggestedPredefines());
 433     return Reader;
 434
 435   case ASTReader::Failure:
 436     // Unrecoverable failure: don't even try to process the input file.
 437     break;
 438
 439   case ASTReader::Missing:
 440   case ASTReader::OutOfDate:
 441   case ASTReader::VersionMismatch:
 442   case ASTReader::ConfigurationMismatch:
 443   case ASTReader::HadErrors:
 444     // No suitable PCH file could be found. Return an error.
 445     break;
 446   }
 447
 448   Context.setExternalSource(nullptr);
 449   return nullptr;
 450 }
 451
 452 // Code Completion
 453
 454 static bool EnableCodeCompletion(Preprocessor &PP,
 455                                  const std::string &Filename,
 456                                  unsigned Line,
 457                                  unsigned Column) {
 458   // Tell the source manager to chop off the given file at a specific
 459   // line and column.
 460   const FileEntry *Entry = PP.getFileManager().getFile(Filename);
 461   if (!Entry) {
 462     PP.getDiagnostics().Report(diag::err_fe_invalid_code_complete_file)
 463       << Filename;
 464     return true;
 465   }
 466
 467   // Truncate the named file at the given line/column.
 468   PP.SetCodeCompletionPoint(Entry, Line, Column);
 469   return false;
 470 }
 471
 472 void CompilerInstance::createCodeCompletionConsumer() {
 473   const ParsedSourceLocation &Loc = getFrontendOpts().CodeCompletionAt;
 474   if (!CompletionConsumer) {
 475     setCodeCompletionConsumer(
 476       createCodeCompletionConsumer(getPreprocessor(),
 477                                    Loc.FileName, Loc.Line, Loc.Column,
 478                                    getFrontendOpts().CodeCompleteOpts,
 479                                    llvm::outs()));
 480     if (!CompletionConsumer)
 481       return;
 482   } else if (EnableCodeCompletion(getPreprocessor(), Loc.FileName,
 483                                   Loc.Line, Loc.Column)) {
 484     setCodeCompletionConsumer(nullptr);
 485     return;
 486   }
 487
 488   if (CompletionConsumer->isOutputBinary() &&
 489       llvm::sys::ChangeStdoutToBinary()) {
 490     getPreprocessor().getDiagnostics().Report(diag::err_fe_stdout_binary);
 491     setCodeCompletionConsumer(nullptr);
 492   }
 493 }
 494
 495 void CompilerInstance::createFrontendTimer() {
 496   FrontendTimer.reset(new llvm::Timer("Clang front-end timer"));
 497 }
 498
 499 CodeCompleteConsumer *
 500 CompilerInstance::createCodeCompletionConsumer(Preprocessor &PP,
 501                                                const std::string &Filename,
 502                                                unsigned Line,
 503                                                unsigned Column,
 504                                                const CodeCompleteOptions &Opts,
 505                                                raw_ostream &OS) {
 506   if (EnableCodeCompletion(PP, Filename, Line, Column))
 507     return nullptr;
 508
 509   // Set up the creation routine for code-completion.
 510   return new PrintingCodeCompleteConsumer(Opts, OS);
 511 }
 512
 513 void CompilerInstance::createSema(TranslationUnitKind TUKind,
 514                                   CodeCompleteConsumer *CompletionConsumer) {
 515   TheSema.reset(new Sema(getPreprocessor(), getASTContext(), getASTConsumer(),
 516                          TUKind, CompletionConsumer));
 517 }
 518
 519 // Output Files
 520
 521 void CompilerInstance::addOutputFile(OutputFile &&OutFile) {
 522   assert(OutFile.OS && "Attempt to add empty stream to output list!");
 523   OutputFiles.push_back(std::move(OutFile));
 524 }
 525
 526 void CompilerInstance::clearOutputFiles(bool EraseFiles) {
 527   for (OutputFile &OF : OutputFiles) {
 528     // Manually close the stream before we rename it.
 529     OF.OS.reset();
 530
 531     if (!OF.TempFilename.empty()) {
 532       if (EraseFiles) {
 533         llvm::sys::fs::remove(OF.TempFilename);
 534       } else {
 535         SmallString<128> NewOutFile(OF.Filename);
 536
 537         // If '-working-directory' was passed, the output filename should be
 538         // relative to that.
 539         FileMgr->FixupRelativePath(NewOutFile);
 540         if (std::error_code ec =
 541                 llvm::sys::fs::rename(OF.TempFilename, NewOutFile)) {
 542           getDiagnostics().Report(diag::err_unable_to_rename_temp)
 543             << OF.TempFilename << OF.Filename << ec.message();
 544
 545           llvm::sys::fs::remove(OF.TempFilename);
 546         }
 547       }
 548     } else if (!OF.Filename.empty() && EraseFiles)
 549       llvm::sys::fs::remove(OF.Filename);
 550
 551   }
 552   OutputFiles.clear();
 553   NonSeekStream.reset();
 554 }
 555
 556 raw_pwrite_stream *
 557 CompilerInstance::createDefaultOutputFile(bool Binary, StringRef InFile,
 558                                           StringRef Extension) {
 559   return createOutputFile(getFrontendOpts().OutputFile, Binary,
 560                           /*RemoveFileOnSignal=*/true, InFile, Extension,
 561                           /*UseTemporary=*/true);
 562 }
 563
 564 llvm::raw_null_ostream *CompilerInstance::createNullOutputFile() {
 565   auto OS = llvm::make_unique<llvm::raw_null_ostream>();
 566   llvm::raw_null_ostream *Ret = OS.get();
 567   addOutputFile(OutputFile("", "", std::move(OS)));
 568   return Ret;
 569 }
 570
 571 raw_pwrite_stream *
 572 CompilerInstance::createOutputFile(StringRef OutputPath, bool Binary,
 573                                    bool RemoveFileOnSignal, StringRef InFile,
 574                                    StringRef Extension, bool UseTemporary,
 575                                    bool CreateMissingDirectories) {
 576   std::string OutputPathName, TempPathName;
 577   std::error_code EC;
 578   std::unique_ptr<raw_pwrite_stream> OS = createOutputFile(
 579       OutputPath, EC, Binary, RemoveFileOnSignal, InFile, Extension,
 580       UseTemporary, CreateMissingDirectories, &OutputPathName, &TempPathName);
 581   if (!OS) {
 582     getDiagnostics().Report(diag::err_fe_unable_to_open_output) << OutputPath
 583                                                                 << EC.message();
 584     return nullptr;
 585   }
 586
 587   raw_pwrite_stream *Ret = OS.get();
 588   // Add the output file -- but don't try to remove "-", since this means we are
 589   // using stdin.
 590   addOutputFile(OutputFile((OutputPathName != "-") ? OutputPathName : "",
 591                            TempPathName, std::move(OS)));
 592
 593   return Ret;
 594 }
 595
 596 std::unique_ptr<llvm::raw_pwrite_stream> CompilerInstance::createOutputFile(
 597     StringRef OutputPath, std::error_code &Error, bool Binary,
 598     bool RemoveFileOnSignal, StringRef InFile, StringRef Extension,
 599     bool UseTemporary, bool CreateMissingDirectories,
 600     std::string *ResultPathName, std::string *TempPathName) {
 601   assert((!CreateMissingDirectories || UseTemporary) &&
 602          "CreateMissingDirectories is only allowed when using temporary files");
 603
 604   std::string OutFile, TempFile;
 605   if (!OutputPath.empty()) {
 606     OutFile = OutputPath;
 607   } else if (InFile == "-") {
 608     OutFile = "-";
 609   } else if (!Extension.empty()) {
 610     SmallString<128> Path(InFile);
 611     llvm::sys::path::replace_extension(Path, Extension);
 612     OutFile = Path.str();
 613   } else {
 614     OutFile = "-";
 615   }
 616
 617   std::unique_ptr<llvm::raw_fd_ostream> OS;
 618   std::string OSFile;
 619
 620   if (UseTemporary) {
 621     if (OutFile == "-")
 622       UseTemporary = false;
 623     else {
 624       llvm::sys::fs::file_status Status;
 625       llvm::sys::fs::status(OutputPath, Status);
 626       if (llvm::sys::fs::exists(Status)) {
 627         // Fail early if we can't write to the final destination.
 628         if (!llvm::sys::fs::can_write(OutputPath))
 629           return nullptr;
 630
 631         // Don't use a temporary if the output is a special file. This handles
 632         // things like '-o /dev/null'
 633         if (!llvm::sys::fs::is_regular_file(Status))
 634           UseTemporary = false;
 635       }
 636     }
 637   }
 638
 639   if (UseTemporary) {
 640     // Create a temporary file.
 641     SmallString<128> TempPath;
 642     TempPath = OutFile;
 643     TempPath += "-%%%%%%%%";
 644     int fd;
 645     std::error_code EC =
 646         llvm::sys::fs::createUniqueFile(TempPath, fd, TempPath);
 647
 648     if (CreateMissingDirectories &&
 649         EC == llvm::errc::no_such_file_or_directory) {
 650       StringRef Parent = llvm::sys::path::parent_path(OutputPath);
 651       EC = llvm::sys::fs::create_directories(Parent);
 652       if (!EC) {
 653         EC = llvm::sys::fs::createUniqueFile(TempPath, fd, TempPath);
 654       }
 655     }
 656
 657     if (!EC) {
 658       OS.reset(new llvm::raw_fd_ostream(fd, /*shouldClose=*/true));
 659       OSFile = TempFile = TempPath.str();
 660     }
 661     // If we failed to create the temporary, fallback to writing to the file
 662     // directly. This handles the corner case where we cannot write to the
 663     // directory, but can write to the file.
 664   }
 665
 666   if (!OS) {
 667     OSFile = OutFile;
 668     OS.reset(new llvm::raw_fd_ostream(
 669         OSFile, Error,
 670         (Binary ? llvm::sys::fs::F_None : llvm::sys::fs::F_Text)));
 671     if (Error)
 672       return nullptr;
 673   }
 674
 675   // Make sure the out stream file gets removed if we crash.
 676   if (RemoveFileOnSignal)
 677     llvm::sys::RemoveFileOnSignal(OSFile);
 678
 679   if (ResultPathName)
 680     *ResultPathName = OutFile;
 681   if (TempPathName)
 682     *TempPathName = TempFile;
 683
 684   if (!Binary || OS->supportsSeeking())
 685     return std::move(OS);
 686
 687   auto B = llvm::make_unique<llvm::buffer_ostream>(*OS);
 688   assert(!NonSeekStream);
 689   NonSeekStream = std::move(OS);
 690   return std::move(B);
 691 }
 692
 693 // Initialization Utilities
 694
 695 bool CompilerInstance::InitializeSourceManager(const FrontendInputFile &Input){
 696   return InitializeSourceManager(Input, getDiagnostics(),
 697                                  getFileManager(), getSourceManager(),
 698                                  getFrontendOpts());
 699 }
 700
 701 bool CompilerInstance::InitializeSourceManager(const FrontendInputFile &Input,
 702                                                DiagnosticsEngine &Diags,
 703                                                FileManager &FileMgr,
 704                                                SourceManager &SourceMgr,
 705                                                const FrontendOptions &Opts) {
 706   SrcMgr::CharacteristicKind
 707     Kind = Input.isSystem() ? SrcMgr::C_System : SrcMgr::C_User;
 708
 709   if (Input.isBuffer()) {
 710     SourceMgr.setMainFileID(SourceMgr.createFileID(
 711         std::unique_ptr<llvm::MemoryBuffer>(Input.getBuffer()), Kind));
 712     assert(!SourceMgr.getMainFileID().isInvalid() &&
 713            "Couldn't establish MainFileID!");
 714     return true;
 715   }
 716
 717   StringRef InputFile = Input.getFile();
 718
 719   // Figure out where to get and map in the main file.
 720   if (InputFile != "-") {
 721     const FileEntry *File = FileMgr.getFile(InputFile, /*OpenFile=*/true);
 722     if (!File) {
 723       Diags.Report(diag::err_fe_error_reading) << InputFile;
 724       return false;
 725     }
 726
 727     // The natural SourceManager infrastructure can't currently handle named
 728     // pipes, but we would at least like to accept them for the main
 729     // file. Detect them here, read them with the volatile flag so FileMgr will
 730     // pick up the correct size, and simply override their contents as we do for
 731     // STDIN.
 732     if (File->isNamedPipe()) {
 733       auto MB = FileMgr.getBufferForFile(File, /*isVolatile=*/true);
 734       if (MB) {
 735         // Create a new virtual file that will have the correct size.
 736         File = FileMgr.getVirtualFile(InputFile, (*MB)->getBufferSize(), 0);
 737         SourceMgr.overrideFileContents(File, std::move(*MB));
 738       } else {
 739         Diags.Report(diag::err_cannot_open_file) << InputFile
 740                                                  << MB.getError().message();
 741         return false;
 742       }
 743     }
 744
 745     SourceMgr.setMainFileID(
 746         SourceMgr.createFileID(File, SourceLocation(), Kind));
 747   } else {
 748     llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> SBOrErr =
 749         llvm::MemoryBuffer::getSTDIN();
 750     if (std::error_code EC = SBOrErr.getError()) {
 751       Diags.Report(diag::err_fe_error_reading_stdin) << EC.message();
 752       return false;
 753     }
 754     std::unique_ptr<llvm::MemoryBuffer> SB = std::move(SBOrErr.get());
 755
 756     const FileEntry *File = FileMgr.getVirtualFile(SB->getBufferIdentifier(),
 757                                                    SB->getBufferSize(), 0);
 758     SourceMgr.setMainFileID(
 759         SourceMgr.createFileID(File, SourceLocation(), Kind));
 760     SourceMgr.overrideFileContents(File, std::move(SB));
 761   }
 762
 763   assert(!SourceMgr.getMainFileID().isInvalid() &&
 764          "Couldn't establish MainFileID!");
 765   return true;
 766 }
 767
 768 // High-Level Operations
 769
 770 bool CompilerInstance::ExecuteAction(FrontendAction &Act) {
 771   assert(hasDiagnostics() && "Diagnostics engine is not initialized!");
 772   assert(!getFrontendOpts().ShowHelp && "Client must handle '-help'!");
 773   assert(!getFrontendOpts().ShowVersion && "Client must handle '-version'!");
 774
 775   // FIXME: Take this as an argument, once all the APIs we used have moved to
 776   // taking it as an input instead of hard-coding llvm::errs.
 777   raw_ostream &OS = llvm::errs();
 778
 779   // Create the target instance.
 780   setTarget(TargetInfo::CreateTargetInfo(getDiagnostics(),
 781                                          getInvocation().TargetOpts));
 782   if (!hasTarget())
 783     return false;
 784
 785   // Inform the target of the language options.
 786   //
 787   // FIXME: We shouldn't need to do this, the target should be immutable once
 788   // created. This complexity should be lifted elsewhere.
 789   getTarget().adjust(getLangOpts());
 790
 791   // rewriter project will change target built-in bool type from its default.
 792   if (getFrontendOpts().ProgramAction == frontend::RewriteObjC)
 793     getTarget().noSignedCharForObjCBool();
 794
 795   // Validate/process some options.
 796   if (getHeaderSearchOpts().Verbose)
 797     OS << "clang -cc1 version " CLANG_VERSION_STRING
 798        << " based upon " << BACKEND_PACKAGE_STRING
 799        << " default target " << llvm::sys::getDefaultTargetTriple() << "\n";
 800
 801   if (getFrontendOpts().ShowTimers)
 802     createFrontendTimer();
 803
 804   if (getFrontendOpts().ShowStats)
 805     llvm::EnableStatistics();
 806
 807   for (unsigned i = 0, e = getFrontendOpts().Inputs.size(); i != e; ++i) {
 808     // Reset the ID tables if we are reusing the SourceManager and parsing
 809     // regular files.
 810     if (hasSourceManager() && !Act.isModelParsingAction())
 811       getSourceManager().clearIDTables();
 812
 813     if (Act.BeginSourceFile(*this, getFrontendOpts().Inputs[i])) {
 814       Act.Execute();
 815       Act.EndSourceFile();
 816     }
 817   }
 818
 819   // Notify the diagnostic client that all files were processed.
 820   getDiagnostics().getClient()->finish();
 821
 822   if (getDiagnosticOpts().ShowCarets) {
 823     // We can have multiple diagnostics sharing one diagnostic client.
 824     // Get the total number of warnings/errors from the client.
 825     unsigned NumWarnings = getDiagnostics().getClient()->getNumWarnings();
 826     unsigned NumErrors = getDiagnostics().getClient()->getNumErrors();
 827
 828     if (NumWarnings)
 829       OS << NumWarnings << " warning" << (NumWarnings == 1 ? "" : "s");
 830     if (NumWarnings && NumErrors)
 831       OS << " and ";
 832     if (NumErrors)
 833       OS << NumErrors << " error" << (NumErrors == 1 ? "" : "s");
 834     if (NumWarnings || NumErrors)
 835       OS << " generated.\n";
 836   }
 837
 838   if (getFrontendOpts().ShowStats && hasFileManager()) {
 839     getFileManager().PrintStats();
 840     OS << "\n";
 841   }
 842
 843   return !getDiagnostics().getClient()->getNumErrors();
 844 }
 845
 846 /// \brief Determine the appropriate source input kind based on language
 847 /// options.
 848 static InputKind getSourceInputKindFromOptions(const LangOptions &LangOpts) {
 849   if (LangOpts.OpenCL)
 850     return IK_OpenCL;
 851   if (LangOpts.CUDA)
 852     return IK_CUDA;
 853   if (LangOpts.ObjC1)
 854     return LangOpts.CPlusPlus? IK_ObjCXX : IK_ObjC;
 855   return LangOpts.CPlusPlus? IK_CXX : IK_C;
 856 }
 857
 858 /// \brief Compile a module file for the given module, using the options
 859 /// provided by the importing compiler instance. Returns true if the module
 860 /// was built without errors.
 861 static bool compileModuleImpl(CompilerInstance &ImportingInstance,
 862                               SourceLocation ImportLoc,
 863                               Module *Module,
 864                               StringRef ModuleFileName) {
 865   ModuleMap &ModMap
 866     = ImportingInstance.getPreprocessor().getHeaderSearchInfo().getModuleMap();
 867
 868   // Construct a compiler invocation for creating this module.
 869   IntrusiveRefCntPtr<CompilerInvocation> Invocation
 870     (new CompilerInvocation(ImportingInstance.getInvocation()));
 871
 872   PreprocessorOptions &PPOpts = Invocation->getPreprocessorOpts();
 873
 874   // For any options that aren't intended to affect how a module is built,
 875   // reset them to their default values.
 876   Invocation->getLangOpts()->resetNonModularOptions();
 877   PPOpts.resetNonModularOptions();
 878
 879   // Remove any macro definitions that are explicitly ignored by the module.
 880   // They aren't supposed to affect how the module is built anyway.
 881   const HeaderSearchOptions &HSOpts = Invocation->getHeaderSearchOpts();
 882   PPOpts.Macros.erase(
 883       std::remove_if(PPOpts.Macros.begin(), PPOpts.Macros.end(),
 884                      [&HSOpts](const std::pair<std::string, bool> &def) {
 885         StringRef MacroDef = def.first;
 886         return HSOpts.ModulesIgnoreMacros.count(MacroDef.split('=').first) > 0;
 887       }),
 888       PPOpts.Macros.end());
 889
 890   // Note the name of the module we're building.
 891   Invocation->getLangOpts()->CurrentModule = Module->getTopLevelModuleName();
 892
 893   // Make sure that the failed-module structure has been allocated in
 894   // the importing instance, and propagate the pointer to the newly-created
 895   // instance.
 896   PreprocessorOptions &ImportingPPOpts
 897     = ImportingInstance.getInvocation().getPreprocessorOpts();
 898   if (!ImportingPPOpts.FailedModules)
 899     ImportingPPOpts.FailedModules = new PreprocessorOptions::FailedModulesSet;
 900   PPOpts.FailedModules = ImportingPPOpts.FailedModules;
 901
 902   // If there is a module map file, build the module using the module map.
 903   // Set up the inputs/outputs so that we build the module from its umbrella
 904   // header.
 905   FrontendOptions &FrontendOpts = Invocation->getFrontendOpts();
 906   FrontendOpts.OutputFile = ModuleFileName.str();
 907   FrontendOpts.DisableFree = false;
 908   FrontendOpts.GenerateGlobalModuleIndex = false;
 909   FrontendOpts.Inputs.clear();
 910   InputKind IK = getSourceInputKindFromOptions(*Invocation->getLangOpts());
 911
 912   // Don't free the remapped file buffers; they are owned by our caller.
 913   PPOpts.RetainRemappedFileBuffers = true;
 914
 915   Invocation->getDiagnosticOpts().VerifyDiagnostics = 0;
 916   assert(ImportingInstance.getInvocation().getModuleHash() ==
 917          Invocation->getModuleHash() && "Module hash mismatch!");
 918
 919   // Construct a compiler instance that will be used to actually create the
 920   // module.
 921   CompilerInstance Instance(/*BuildingModule=*/true);
 922   Instance.setInvocation(&*Invocation);
 923
 924   Instance.createDiagnostics(new ForwardingDiagnosticConsumer(
 925                                    ImportingInstance.getDiagnosticClient()),
 926                              /*ShouldOwnClient=*/true);
 927
 928   Instance.setVirtualFileSystem(&ImportingInstance.getVirtualFileSystem());
 929
 930   // Note that this module is part of the module build stack, so that we
 931   // can detect cycles in the module graph.
 932   Instance.setFileManager(&ImportingInstance.getFileManager());
 933   Instance.createSourceManager(Instance.getFileManager());
 934   SourceManager &SourceMgr = Instance.getSourceManager();
 935   SourceMgr.setModuleBuildStack(
 936     ImportingInstance.getSourceManager().getModuleBuildStack());
 937   SourceMgr.pushModuleBuildStack(Module->getTopLevelModuleName(),
 938     FullSourceLoc(ImportLoc, ImportingInstance.getSourceManager()));
 939
 940   // If we're collecting module dependencies, we need to share a collector
 941   // between all of the module CompilerInstances.
 942   Instance.setModuleDepCollector(ImportingInstance.getModuleDepCollector());
 943
 944   // Get or create the module map that we'll use to build this module.
 945   std::string InferredModuleMapContent;
 946   if (const FileEntry *ModuleMapFile =
 947           ModMap.getContainingModuleMapFile(Module)) {
 948     // Use the module map where this module resides.
 949     FrontendOpts.Inputs.emplace_back(ModuleMapFile->getName(), IK);
 950   } else {
 951     SmallString<128> FakeModuleMapFile(Module->Directory->getName());
 952     llvm::sys::path::append(FakeModuleMapFile, "__inferred_module.map");
 953     FrontendOpts.Inputs.emplace_back(FakeModuleMapFile, IK);
 954
 955     llvm::raw_string_ostream OS(InferredModuleMapContent);
 956     Module->print(OS);
 957     OS.flush();
 958
 959     std::unique_ptr<llvm::MemoryBuffer> ModuleMapBuffer =
 960         llvm::MemoryBuffer::getMemBuffer(InferredModuleMapContent);
 961     ModuleMapFile = Instance.getFileManager().getVirtualFile(
 962         FakeModuleMapFile, InferredModuleMapContent.size(), 0);
 963     SourceMgr.overrideFileContents(ModuleMapFile, std::move(ModuleMapBuffer));
 964   }
 965
 966   // Construct a module-generating action. Passing through the module map is
 967   // safe because the FileManager is shared between the compiler instances.
 968   GenerateModuleAction CreateModuleAction(
 969       ModMap.getModuleMapFileForUniquing(Module), Module->IsSystem);
 970
 971   ImportingInstance.getDiagnostics().Report(ImportLoc,
 972                                             diag::remark_module_build)
 973     << Module->Name << ModuleFileName;
 974
 975   // Execute the action to actually build the module in-place. Use a separate
 976   // thread so that we get a stack large enough.
 977   const unsigned ThreadStackSize = 8 << 20;
 978   llvm::CrashRecoveryContext CRC;
 979   CRC.RunSafelyOnThread([&]() { Instance.ExecuteAction(CreateModuleAction); },
 980                         ThreadStackSize);
 981
 982   ImportingInstance.getDiagnostics().Report(ImportLoc,
 983                                             diag::remark_module_build_done)
 984     << Module->Name;
 985
 986   // Delete the temporary module map file.
 987   // FIXME: Even though we're executing under crash protection, it would still
 988   // be nice to do this with RemoveFileOnSignal when we can. However, that
 989   // doesn't make sense for all clients, so clean this up manually.
 990   Instance.clearOutputFiles(/*EraseFiles=*/true);
 991
 992   // We've rebuilt a module. If we're allowed to generate or update the global
 993   // module index, record that fact in the importing compiler instance.
 994   if (ImportingInstance.getFrontendOpts().GenerateGlobalModuleIndex) {
 995     ImportingInstance.setBuildGlobalModuleIndex(true);
 996   }
 997
 998   return !Instance.getDiagnostics().hasErrorOccurred();
 999 }
1000
1001 static bool compileAndLoadModule(CompilerInstance &ImportingInstance,
1002                                  SourceLocation ImportLoc,
1003                                  SourceLocation ModuleNameLoc, Module *Module,
1004                                  StringRef ModuleFileName) {
1005   DiagnosticsEngine &Diags = ImportingInstance.getDiagnostics();
1006
1007   auto diagnoseBuildFailure = [&] {
1008     Diags.Report(ModuleNameLoc, diag::err_module_not_built)
1009         << Module->Name << SourceRange(ImportLoc, ModuleNameLoc);
1010   };
1011
1012   // FIXME: have LockFileManager return an error_code so that we can
1013   // avoid the mkdir when the directory already exists.
1014   StringRef Dir = llvm::sys::path::parent_path(ModuleFileName);
1015   llvm::sys::fs::create_directories(Dir);
1016
1017   while (1) {
1018     unsigned ModuleLoadCapabilities = ASTReader::ARR_Missing;
1019     llvm::LockFileManager Locked(ModuleFileName);
1020     switch (Locked) {
1021     case llvm::LockFileManager::LFS_Error:
1022       Diags.Report(ModuleNameLoc, diag::err_module_lock_failure)
1023           << Module->Name;
1024       return false;
1025
1026     case llvm::LockFileManager::LFS_Owned:
1027       // We're responsible for building the module ourselves.
1028       if (!compileModuleImpl(ImportingInstance, ModuleNameLoc, Module,
1029                              ModuleFileName)) {
1030         diagnoseBuildFailure();
1031         return false;
1032       }
1033       break;
1034
1035     case llvm::LockFileManager::LFS_Shared:
1036       // Someone else is responsible for building the module. Wait for them to
1037       // finish.
1038       switch (Locked.waitForUnlock()) {
1039       case llvm::LockFileManager::Res_Success:
1040         ModuleLoadCapabilities |= ASTReader::ARR_OutOfDate;
1041         break;
1042       case llvm::LockFileManager::Res_OwnerDied:
1043         continue; // try again to get the lock.
1044       case llvm::LockFileManager::Res_Timeout:
1045         Diags.Report(ModuleNameLoc, diag::err_module_lock_timeout)
1046             << Module->Name;
1047         // Clear the lock file so that future invokations can make progress.
1048         Locked.unsafeRemoveLockFile();
1049         return false;
1050       }
1051       break;
1052     }
1053
1054     // Try to read the module file, now that we've compiled it.
1055     ASTReader::ASTReadResult ReadResult =
1056         ImportingInstance.getModuleManager()->ReadAST(
1057             ModuleFileName, serialization::MK_ImplicitModule, ImportLoc,
1058             ModuleLoadCapabilities);
1059
1060     if (ReadResult == ASTReader::OutOfDate &&
1061         Locked == llvm::LockFileManager::LFS_Shared) {
1062       // The module may be out of date in the presence of file system races,
1063       // or if one of its imports depends on header search paths that are not
1064       // consistent with this ImportingInstance.  Try again...
1065       continue;
1066     } else if (ReadResult == ASTReader::Missing) {
1067       diagnoseBuildFailure();
1068     } else if (ReadResult != ASTReader::Success &&
1069                !Diags.hasErrorOccurred()) {
1070       // The ASTReader didn't diagnose the error, so conservatively report it.
1071       diagnoseBuildFailure();
1072     }
1073     return ReadResult == ASTReader::Success;
1074   }
1075 }
1076
1077 /// \brief Diagnose differences between the current definition of the given
1078 /// configuration macro and the definition provided on the command line.
1079 static void checkConfigMacro(Preprocessor &PP, StringRef ConfigMacro,
1080                              Module *Mod, SourceLocation ImportLoc) {
1081   IdentifierInfo *Id = PP.getIdentifierInfo(ConfigMacro);
1082   SourceManager &SourceMgr = PP.getSourceManager();
1083
1084   // If this identifier has never had a macro definition, then it could
1085   // not have changed.
1086   if (!Id->hadMacroDefinition())
1087     return;
1088   auto *LatestLocalMD = PP.getLocalMacroDirectiveHistory(Id);
1089
1090   // Find the macro definition from the command line.
1091   MacroInfo *CmdLineDefinition = nullptr;
1092   for (auto *MD = LatestLocalMD; MD; MD = MD->getPrevious()) {
1093     // We only care about the predefines buffer.
1094     FileID FID = SourceMgr.getFileID(MD->getLocation());
1095     if (FID.isInvalid() || FID != PP.getPredefinesFileID())
1096       continue;
1097     if (auto *DMD = dyn_cast<DefMacroDirective>(MD))
1098       CmdLineDefinition = DMD->getMacroInfo();
1099     break;
1100   }
1101
1102   auto *CurrentDefinition = PP.getMacroInfo(Id);
1103   if (CurrentDefinition == CmdLineDefinition) {
1104     // Macro matches. Nothing to do.
1105   } else if (!CurrentDefinition) {
1106     // This macro was defined on the command line, then #undef'd later.
1107     // Complain.
1108     PP.Diag(ImportLoc, diag::warn_module_config_macro_undef)
1109       << true << ConfigMacro << Mod->getFullModuleName();
1110     auto LatestDef = LatestLocalMD->getDefinition();
1111     assert(LatestDef.isUndefined() &&
1112            "predefined macro went away with no #undef?");
1113     PP.Diag(LatestDef.getUndefLocation(), diag::note_module_def_undef_here)
1114       << true;
1115     return;
1116   } else if (!CmdLineDefinition) {
1117     // There was no definition for this macro in the predefines buffer,
1118     // but there was a local definition. Complain.
1119     PP.Diag(ImportLoc, diag::warn_module_config_macro_undef)
1120       << false << ConfigMacro << Mod->getFullModuleName();
1121     PP.Diag(CurrentDefinition->getDefinitionLoc(),
1122             diag::note_module_def_undef_here)
1123       << false;
1124   } else if (!CurrentDefinition->isIdenticalTo(*CmdLineDefinition, PP,
1125                                                /*Syntactically=*/true)) {
1126     // The macro definitions differ.
1127     PP.Diag(ImportLoc, diag::warn_module_config_macro_undef)
1128       << false << ConfigMacro << Mod->getFullModuleName();
1129     PP.Diag(CurrentDefinition->getDefinitionLoc(),
1130             diag::note_module_def_undef_here)
1131       << false;
1132   }
1133 }
1134
1135 /// \brief Write a new timestamp file with the given path.
1136 static void writeTimestampFile(StringRef TimestampFile) {
1137   std::error_code EC;
1138   llvm::raw_fd_ostream Out(TimestampFile.str(), EC, llvm::sys::fs::F_None);
1139 }
1140
1141 /// \brief Prune the module cache of modules that haven't been accessed in
1142 /// a long time.
1143 static void pruneModuleCache(const HeaderSearchOptions &HSOpts) {
1144   struct stat StatBuf;
1145   llvm::SmallString<128> TimestampFile;
1146   TimestampFile = HSOpts.ModuleCachePath;
1147   llvm::sys::path::append(TimestampFile, "modules.timestamp");
1148
1149   // Try to stat() the timestamp file.
1150   if (::stat(TimestampFile.c_str(), &StatBuf)) {
1151     // If the timestamp file wasn't there, create one now.
1152     if (errno == ENOENT) {
1153       writeTimestampFile(TimestampFile);
1154     }
1155     return;
1156   }
1157
1158   // Check whether the time stamp is older than our pruning interval.
1159   // If not, do nothing.
1160   time_t TimeStampModTime = StatBuf.st_mtime;
1161   time_t CurrentTime = time(nullptr);
1162   if (CurrentTime - TimeStampModTime <= time_t(HSOpts.ModuleCachePruneInterval))
1163     return;
1164
1165   // Write a new timestamp file so that nobody else attempts to prune.
1166   // There is a benign race condition here, if two Clang instances happen to
1167   // notice at the same time that the timestamp is out-of-date.
1168   writeTimestampFile(TimestampFile);
1169
1170   // Walk the entire module cache, looking for unused module files and module
1171   // indices.
1172   std::error_code EC;
1173   SmallString<128> ModuleCachePathNative;
1174   llvm::sys::path::native(HSOpts.ModuleCachePath, ModuleCachePathNative);
1175   for (llvm::sys::fs::directory_iterator Dir(ModuleCachePathNative, EC), DirEnd;
1176        Dir != DirEnd && !EC; Dir.increment(EC)) {
1177     // If we don't have a directory, there's nothing to look into.
1178     if (!llvm::sys::fs::is_directory(Dir->path()))
1179       continue;
1180
1181     // Walk all of the files within this directory.
1182     for (llvm::sys::fs::directory_iterator File(Dir->path(), EC), FileEnd;
1183          File != FileEnd && !EC; File.increment(EC)) {
1184       // We only care about module and global module index files.
1185       StringRef Extension = llvm::sys::path::extension(File->path());
1186       if (Extension != ".pcm" && Extension != ".timestamp" &&
1187           llvm::sys::path::filename(File->path()) != "modules.idx")
1188         continue;
1189
1190       // Look at this file. If we can't stat it, there's nothing interesting
1191       // there.
1192       if (::stat(File->path().c_str(), &StatBuf))
1193         continue;
1194
1195       // If the file has been used recently enough, leave it there.
1196       time_t FileAccessTime = StatBuf.st_atime;
1197       if (CurrentTime - FileAccessTime <=
1198               time_t(HSOpts.ModuleCachePruneAfter)) {
1199         continue;
1200       }
1201
1202       // Remove the file.
1203       llvm::sys::fs::remove(File->path());
1204
1205       // Remove the timestamp file.
1206       std::string TimpestampFilename = File->path() + ".timestamp";
1207       llvm::sys::fs::remove(TimpestampFilename);
1208     }
1209
1210     // If we removed all of the files in the directory, remove the directory
1211     // itself.
1212     if (llvm::sys::fs::directory_iterator(Dir->path(), EC) ==
1213             llvm::sys::fs::directory_iterator() && !EC)
1214       llvm::sys::fs::remove(Dir->path());
1215   }
1216 }
1217
1218 void CompilerInstance::createModuleManager() {
1219   if (!ModuleManager) {
1220     if (!hasASTContext())
1221       createASTContext();
1222
1223     // If we're implicitly building modules but not currently recursively
1224     // building a module, check whether we need to prune the module cache.
1225     if (getLangOpts().ImplicitModules &&
1226         getSourceManager().getModuleBuildStack().empty() &&
1227         getHeaderSearchOpts().ModuleCachePruneInterval > 0 &&
1228         getHeaderSearchOpts().ModuleCachePruneAfter > 0) {
1229       pruneModuleCache(getHeaderSearchOpts());
1230     }
1231
1232     HeaderSearchOptions &HSOpts = getHeaderSearchOpts();
1233     std::string Sysroot = HSOpts.Sysroot;
1234     const PreprocessorOptions &PPOpts = getPreprocessorOpts();
1235     ModuleManager = new ASTReader(getPreprocessor(), *Context,
1236                                   Sysroot.empty() ? "" : Sysroot.c_str(),
1237                                   PPOpts.DisablePCHValidation,
1238                                   /*AllowASTWithCompilerErrors=*/false,
1239                                   /*AllowConfigurationMismatch=*/false,
1240                                   HSOpts.ModulesValidateSystemHeaders,
1241                                   getFrontendOpts().UseGlobalModuleIndex);
1242     if (hasASTConsumer()) {
1243       ModuleManager->setDeserializationListener(
1244         getASTConsumer().GetASTDeserializationListener());
1245       getASTContext().setASTMutationListener(
1246         getASTConsumer().GetASTMutationListener());
1247     }
1248     getASTContext().setExternalSource(ModuleManager);
1249     if (hasSema())
1250       ModuleManager->InitializeSema(getSema());
1251     if (hasASTConsumer())
1252       ModuleManager->StartTranslationUnit(&getASTConsumer());
1253   }
1254 }
1255
1256 bool CompilerInstance::loadModuleFile(StringRef FileName) {
1257   // Helper to recursively read the module names for all modules we're adding.
1258   // We mark these as known and redirect any attempt to load that module to
1259   // the files we were handed.
1260   struct ReadModuleNames : ASTReaderListener {
1261     CompilerInstance &CI;
1262     std::vector<StringRef> ModuleFileStack;
1263     std::vector<StringRef> ModuleNameStack;
1264     bool Failed;
1265     bool TopFileIsModule;
1266
1267     ReadModuleNames(CompilerInstance &CI)
1268         : CI(CI), Failed(false), TopFileIsModule(false) {}
1269
1270     bool needsImportVisitation() const override { return true; }
1271
1272     void visitImport(StringRef FileName) override {
1273       if (!CI.ExplicitlyLoadedModuleFiles.insert(FileName).second) {
1274         if (ModuleFileStack.size() == 0)
1275           TopFileIsModule = true;
1276         return;
1277       }
1278
1279       ModuleFileStack.push_back(FileName);
1280       ModuleNameStack.push_back(StringRef());
1281       if (ASTReader::readASTFileControlBlock(FileName, CI.getFileManager(),
1282                                              *this)) {
1283         CI.getDiagnostics().Report(
1284             SourceLocation(), CI.getFileManager().getBufferForFile(FileName)
1285                                   ? diag::err_module_file_invalid
1286                                   : diag::err_module_file_not_found)
1287             << FileName;
1288         for (int I = ModuleFileStack.size() - 2; I >= 0; --I)
1289           CI.getDiagnostics().Report(SourceLocation(),
1290                                      diag::note_module_file_imported_by)
1291               << ModuleFileStack[I]
1292               << !ModuleNameStack[I].empty() << ModuleNameStack[I];
1293         Failed = true;
1294       }
1295       ModuleNameStack.pop_back();
1296       ModuleFileStack.pop_back();
1297     }
1298
1299     void ReadModuleName(StringRef ModuleName) override {
1300       if (ModuleFileStack.size() == 1)
1301         TopFileIsModule = true;
1302       ModuleNameStack.back() = ModuleName;
1303
1304       auto &ModuleFile = CI.ModuleFileOverrides[ModuleName];
1305       if (!ModuleFile.empty() &&
1306           CI.getFileManager().getFile(ModuleFile) !=
1307               CI.getFileManager().getFile(ModuleFileStack.back()))
1308         CI.getDiagnostics().Report(SourceLocation(),
1309                                    diag::err_conflicting_module_files)
1310             << ModuleName << ModuleFile << ModuleFileStack.back();
1311       ModuleFile = ModuleFileStack.back();
1312     }
1313   } RMN(*this);
1314
1315   // If we don't already have an ASTReader, create one now.
1316   if (!ModuleManager)
1317     createModuleManager();
1318
1319   // Tell the module manager about this module file.
1320   if (getModuleManager()->getModuleManager().addKnownModuleFile(FileName)) {
1321     getDiagnostics().Report(SourceLocation(), diag::err_module_file_not_found)
1322       << FileName;
1323     return false;
1324   }
1325
1326   // Build our mapping of module names to module files from this file
1327   // and its imports.
1328   RMN.visitImport(FileName);
1329
1330   if (RMN.Failed)
1331     return false;
1332
1333   // If we never found a module name for the top file, then it's not a module,
1334   // it's a PCH or preamble or something.
1335   if (!RMN.TopFileIsModule) {
1336     getDiagnostics().Report(SourceLocation(), diag::err_module_file_not_module)
1337       << FileName;
1338     return false;
1339   }
1340
1341   return true;
1342 }
1343
1344 ModuleLoadResult
1345 CompilerInstance::loadModule(SourceLocation ImportLoc,
1346                              ModuleIdPath Path,
1347                              Module::NameVisibilityKind Visibility,
1348                              bool IsInclusionDirective) {
1349   // Determine what file we're searching from.
1350   StringRef ModuleName = Path[0].first->getName();
1351   SourceLocation ModuleNameLoc = Path[0].second;
1352
1353   // If we've already handled this import, just return the cached result.
1354   // This one-element cache is important to eliminate redundant diagnostics
1355   // when both the preprocessor and parser see the same import declaration.
1356   if (!ImportLoc.isInvalid() && LastModuleImportLoc == ImportLoc) {
1357     // Make the named module visible.
1358     if (LastModuleImportResult && ModuleName != getLangOpts().CurrentModule &&
1359         ModuleName != getLangOpts().ImplementationOfModule)
1360       ModuleManager->makeModuleVisible(LastModuleImportResult, Visibility,
1361                                        ImportLoc);
1362     return LastModuleImportResult;
1363   }
1364
1365   clang::Module *Module = nullptr;
1366
1367   // If we don't already have information on this module, load the module now.
1368   llvm::DenseMap<const IdentifierInfo *, clang::Module *>::iterator Known
1369     = KnownModules.find(Path[0].first);
1370   if (Known != KnownModules.end()) {
1371     // Retrieve the cached top-level module.
1372     Module = Known->second;
1373   } else if (ModuleName == getLangOpts().CurrentModule ||
1374              ModuleName == getLangOpts().ImplementationOfModule) {
1375     // This is the module we're building.
1376     Module = PP->getHeaderSearchInfo().lookupModule(ModuleName);
1377     Known = KnownModules.insert(std::make_pair(Path[0].first, Module)).first;
1378   } else {
1379     // Search for a module with the given name.
1380     Module = PP->getHeaderSearchInfo().lookupModule(ModuleName);
1381     if (!Module) {
1382       getDiagnostics().Report(ModuleNameLoc, diag::err_module_not_found)
1383       << ModuleName
1384       << SourceRange(ImportLoc, ModuleNameLoc);
1385       ModuleBuildFailed = true;
1386       return ModuleLoadResult();
1387     }
1388
1389     auto Override = ModuleFileOverrides.find(ModuleName);
1390     bool Explicit = Override != ModuleFileOverrides.end();
1391     if (!Explicit && !getLangOpts().ImplicitModules) {
1392       getDiagnostics().Report(ModuleNameLoc, diag::err_module_build_disabled)
1393           << ModuleName;
1394       ModuleBuildFailed = true;
1395       return ModuleLoadResult();
1396     }
1397
1398     std::string ModuleFileName =
1399         Explicit ? Override->second
1400                  : PP->getHeaderSearchInfo().getModuleFileName(Module);
1401
1402     // If we don't already have an ASTReader, create one now.
1403     if (!ModuleManager)
1404       createModuleManager();
1405
1406     if (TheDependencyFileGenerator)
1407       TheDependencyFileGenerator->AttachToASTReader(*ModuleManager);
1408
1409     if (ModuleDepCollector)
1410       ModuleDepCollector->attachToASTReader(*ModuleManager);
1411
1412     for (auto &Listener : DependencyCollectors)
1413       Listener->attachToASTReader(*ModuleManager);
1414
1415     // Try to load the module file.
1416     unsigned ARRFlags =
1417         Explicit ? 0 : ASTReader::ARR_OutOfDate | ASTReader::ARR_Missing;
1418     switch (ModuleManager->ReadAST(ModuleFileName,
1419                                    Explicit ? serialization::MK_ExplicitModule
1420                                             : serialization::MK_ImplicitModule,
1421                                    ImportLoc, ARRFlags)) {
1422     case ASTReader::Success:
1423       break;
1424
1425     case ASTReader::OutOfDate:
1426     case ASTReader::Missing: {
1427       if (Explicit) {
1428         // ReadAST has already complained for us.
1429         ModuleLoader::HadFatalFailure = true;
1430         KnownModules[Path[0].first] = nullptr;
1431         return ModuleLoadResult();
1432       }
1433
1434       // The module file is missing or out-of-date. Build it.
1435       assert(Module && "missing module file");
1436       // Check whether there is a cycle in the module graph.
1437       ModuleBuildStack ModPath = getSourceManager().getModuleBuildStack();
1438       ModuleBuildStack::iterator Pos = ModPath.begin(), PosEnd = ModPath.end();
1439       for (; Pos != PosEnd; ++Pos) {
1440         if (Pos->first == ModuleName)
1441           break;
1442       }
1443
1444       if (Pos != PosEnd) {
1445         SmallString<256> CyclePath;
1446         for (; Pos != PosEnd; ++Pos) {
1447           CyclePath += Pos->first;
1448           CyclePath += " -> ";
1449         }
1450         CyclePath += ModuleName;
1451
1452         getDiagnostics().Report(ModuleNameLoc, diag::err_module_cycle)
1453           << ModuleName << CyclePath;
1454         return ModuleLoadResult();
1455       }
1456
1457       // Check whether we have already attempted to build this module (but
1458       // failed).
1459       if (getPreprocessorOpts().FailedModules &&
1460           getPreprocessorOpts().FailedModules->hasAlreadyFailed(ModuleName)) {
1461         getDiagnostics().Report(ModuleNameLoc, diag::err_module_not_built)
1462           << ModuleName
1463           << SourceRange(ImportLoc, ModuleNameLoc);
1464         ModuleBuildFailed = true;
1465         return ModuleLoadResult();
1466       }
1467
1468       // Try to compile and then load the module.
1469       if (!compileAndLoadModule(*this, ImportLoc, ModuleNameLoc, Module,
1470                                 ModuleFileName)) {
1471         assert(getDiagnostics().hasErrorOccurred() &&
1472                "undiagnosed error in compileAndLoadModule");
1473         if (getPreprocessorOpts().FailedModules)
1474           getPreprocessorOpts().FailedModules->addFailed(ModuleName);
1475         KnownModules[Path[0].first] = nullptr;
1476         ModuleBuildFailed = true;
1477         return ModuleLoadResult();
1478       }
1479
1480       // Okay, we've rebuilt and now loaded the module.
1481       break;
1482     }
1483
1484     case ASTReader::VersionMismatch:
1485     case ASTReader::ConfigurationMismatch:
1486     case ASTReader::HadErrors:
1487       ModuleLoader::HadFatalFailure = true;
1488       // FIXME: The ASTReader will already have complained, but can we showhorn
1489       // that diagnostic information into a more useful form?
1490       KnownModules[Path[0].first] = nullptr;
1491       return ModuleLoadResult();
1492
1493     case ASTReader::Failure:
1494       ModuleLoader::HadFatalFailure = true;
1495       // Already complained, but note now that we failed.
1496       KnownModules[Path[0].first] = nullptr;
1497       ModuleBuildFailed = true;
1498       return ModuleLoadResult();
1499     }
1500
1501     // Cache the result of this top-level module lookup for later.
1502     Known = KnownModules.insert(std::make_pair(Path[0].first, Module)).first;
1503   }
1504
1505   // If we never found the module, fail.
1506   if (!Module)
1507     return ModuleLoadResult();
1508
1509   // Verify that the rest of the module path actually corresponds to
1510   // a submodule.
1511   if (Path.size() > 1) {
1512     for (unsigned I = 1, N = Path.size(); I != N; ++I) {
1513       StringRef Name = Path[I].first->getName();
1514       clang::Module *Sub = Module->findSubmodule(Name);
1515
1516       if (!Sub) {
1517         // Attempt to perform typo correction to find a module name that works.
1518         SmallVector<StringRef, 2> Best;
1519         unsigned BestEditDistance = (std::numeric_limits<unsigned>::max)();
1520
1521         for (clang::Module::submodule_iterator J = Module->submodule_begin(),
1522                                             JEnd = Module->submodule_end();
1523              J != JEnd; ++J) {
1524           unsigned ED = Name.edit_distance((*J)->Name,
1525                                            /*AllowReplacements=*/true,
1526                                            BestEditDistance);
1527           if (ED <= BestEditDistance) {
1528             if (ED < BestEditDistance) {
1529               Best.clear();
1530               BestEditDistance = ED;
1531             }
1532
1533             Best.push_back((*J)->Name);
1534           }
1535         }
1536
1537         // If there was a clear winner, user it.
1538         if (Best.size() == 1) {
1539           getDiagnostics().Report(Path[I].second,
1540                                   diag::err_no_submodule_suggest)
1541             << Path[I].first << Module->getFullModuleName() << Best[0]
1542             << SourceRange(Path[0].second, Path[I-1].second)
1543             << FixItHint::CreateReplacement(SourceRange(Path[I].second),
1544                                             Best[0]);
1545
1546           Sub = Module->findSubmodule(Best[0]);
1547         }
1548       }
1549
1550       if (!Sub) {
1551         // No submodule by this name. Complain, and don't look for further
1552         // submodules.
1553         getDiagnostics().Report(Path[I].second, diag::err_no_submodule)
1554           << Path[I].first << Module->getFullModuleName()
1555           << SourceRange(Path[0].second, Path[I-1].second);
1556         break;
1557       }
1558
1559       Module = Sub;
1560     }
1561   }
1562
1563   // Don't make the module visible if we are in the implementation.
1564   if (ModuleName == getLangOpts().ImplementationOfModule)
1565     return ModuleLoadResult(Module, false);
1566
1567   // Make the named module visible, if it's not already part of the module
1568   // we are parsing.
1569   if (ModuleName != getLangOpts().CurrentModule) {
1570     if (!Module->IsFromModuleFile) {
1571       // We have an umbrella header or directory that doesn't actually include
1572       // all of the headers within the directory it covers. Complain about
1573       // this missing submodule and recover by forgetting that we ever saw
1574       // this submodule.
1575       // FIXME: Should we detect this at module load time? It seems fairly
1576       // expensive (and rare).
1577       getDiagnostics().Report(ImportLoc, diag::warn_missing_submodule)
1578         << Module->getFullModuleName()
1579         << SourceRange(Path.front().second, Path.back().second);
1580
1581       return ModuleLoadResult(nullptr, true);
1582     }
1583
1584     // Check whether this module is available.
1585     clang::Module::Requirement Requirement;
1586     clang::Module::UnresolvedHeaderDirective MissingHeader;
1587     if (!Module->isAvailable(getLangOpts(), getTarget(), Requirement,
1588                              MissingHeader)) {
1589       if (MissingHeader.FileNameLoc.isValid()) {
1590         getDiagnostics().Report(MissingHeader.FileNameLoc,
1591                                 diag::err_module_header_missing)
1592           << MissingHeader.IsUmbrella << MissingHeader.FileName;
1593       } else {
1594         getDiagnostics().Report(ImportLoc, diag::err_module_unavailable)
1595           << Module->getFullModuleName()
1596           << Requirement.second << Requirement.first
1597           << SourceRange(Path.front().second, Path.back().second);
1598       }
1599       LastModuleImportLoc = ImportLoc;
1600       LastModuleImportResult = ModuleLoadResult();
1601       return ModuleLoadResult();
1602     }
1603
1604     ModuleManager->makeModuleVisible(Module, Visibility, ImportLoc);
1605   }
1606
1607   // Check for any configuration macros that have changed.
1608   clang::Module *TopModule = Module->getTopLevelModule();
1609   for (unsigned I = 0, N = TopModule->ConfigMacros.size(); I != N; ++I) {
1610     checkConfigMacro(getPreprocessor(), TopModule->ConfigMacros[I],
1611                      Module, ImportLoc);
1612   }
1613
1614   LastModuleImportLoc = ImportLoc;
1615   LastModuleImportResult = ModuleLoadResult(Module, false);
1616   return LastModuleImportResult;
1617 }
1618
1619 void CompilerInstance::makeModuleVisible(Module *Mod,
1620                                          Module::NameVisibilityKind Visibility,
1621                                          SourceLocation ImportLoc) {
1622   if (!ModuleManager)
1623     createModuleManager();
1624   if (!ModuleManager)
1625     return;
1626
1627   ModuleManager->makeModuleVisible(Mod, Visibility, ImportLoc);
1628 }
1629
1630 GlobalModuleIndex *CompilerInstance::loadGlobalModuleIndex(
1631     SourceLocation TriggerLoc) {
1632   if (!ModuleManager)
1633     createModuleManager();
1634   // Can't do anything if we don't have the module manager.
1635   if (!ModuleManager)
1636     return nullptr;
1637   // Get an existing global index.  This loads it if not already
1638   // loaded.
1639   ModuleManager->loadGlobalIndex();
1640   GlobalModuleIndex *GlobalIndex = ModuleManager->getGlobalIndex();
1641   // If the global index doesn't exist, create it.
1642   if (!GlobalIndex && shouldBuildGlobalModuleIndex() && hasFileManager() &&
1643       hasPreprocessor()) {
1644     llvm::sys::fs::create_directories(
1645       getPreprocessor().getHeaderSearchInfo().getModuleCachePath());
1646     GlobalModuleIndex::writeIndex(
1647       getFileManager(),
1648       getPreprocessor().getHeaderSearchInfo().getModuleCachePath());
1649     ModuleManager->resetForReload();
1650     ModuleManager->loadGlobalIndex();
1651     GlobalIndex = ModuleManager->getGlobalIndex();
1652   }
1653   // For finding modules needing to be imported for fixit messages,
1654   // we need to make the global index cover all modules, so we do that here.
1655   if (!HaveFullGlobalModuleIndex && GlobalIndex && !buildingModule()) {
1656     ModuleMap &MMap = getPreprocessor().getHeaderSearchInfo().getModuleMap();
1657     bool RecreateIndex = false;
1658     for (ModuleMap::module_iterator I = MMap.module_begin(),
1659         E = MMap.module_end(); I != E; ++I) {
1660       Module *TheModule = I->second;
1661       const FileEntry *Entry = TheModule->getASTFile();
1662       if (!Entry) {
1663         SmallVector<std::pair<IdentifierInfo *, SourceLocation>, 2> Path;
1664         Path.push_back(std::make_pair(
1665                                   getPreprocessor().getIdentifierInfo(TheModule->Name), TriggerLoc));
1666         std::reverse(Path.begin(), Path.end());
1667                     // Load a module as hidden.  This also adds it to the global index.
1668         loadModule(TheModule->DefinitionLoc, Path,
1669                                              Module::Hidden, false);
1670         RecreateIndex = true;
1671       }
1672     }
1673     if (RecreateIndex) {
1674       GlobalModuleIndex::writeIndex(
1675         getFileManager(),
1676         getPreprocessor().getHeaderSearchInfo().getModuleCachePath());
1677       ModuleManager->resetForReload();
1678       ModuleManager->loadGlobalIndex();
1679       GlobalIndex = ModuleManager->getGlobalIndex();
1680     }
1681     HaveFullGlobalModuleIndex = true;
1682   }
1683   return GlobalIndex;
1684 }
1685
1686 // Check global module index for missing imports.
1687 bool
1688 CompilerInstance::lookupMissingImports(StringRef Name,
1689                                        SourceLocation TriggerLoc) {
1690   // Look for the symbol in non-imported modules, but only if an error
1691   // actually occurred.
1692   if (!buildingModule()) {
1693     // Load global module index, or retrieve a previously loaded one.
1694     GlobalModuleIndex *GlobalIndex = loadGlobalModuleIndex(
1695       TriggerLoc);
1696
1697     // Only if we have a global index.
1698     if (GlobalIndex) {
1699       GlobalModuleIndex::HitSet FoundModules;
1700
1701       // Find the modules that reference the identifier.
1702       // Note that this only finds top-level modules.
1703       // We'll let diagnoseTypo find the actual declaration module.
1704       if (GlobalIndex->lookupIdentifier(Name, FoundModules))
1705         return true;
1706     }
1707   }
1708
1709   return false;
1710 }
1711 void CompilerInstance::resetAndLeakSema() { BuryPointer(takeSema()); }