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