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