xref: /llvm-project/clang/lib/Frontend/CompilerInstance.cpp (revision 824285ec537f78d2e7143c148edc651b7eb78d99)
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