1 //===--- HeaderSearch.h - Resolve Header File Locations ---------*- C++ -*-===//
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 // This file defines the HeaderSearch interface.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_CLANG_LEX_HEADERSEARCH_H
15 #define LLVM_CLANG_LEX_HEADERSEARCH_H
16 
17 #include "clang/Lex/DirectoryLookup.h"
18 #include "clang/Lex/ModuleMap.h"
19 #include "llvm/ADT/ArrayRef.h"
20 #include "llvm/ADT/IntrusiveRefCntPtr.h"
21 #include "llvm/ADT/StringMap.h"
22 #include "llvm/ADT/StringSet.h"
23 #include "llvm/Support/Allocator.h"
24 #include <memory>
25 #include <vector>
26 
27 namespace clang {
28 
29 class DiagnosticsEngine;
30 class ExternalPreprocessorSource;
31 class FileEntry;
32 class FileManager;
33 class HeaderSearchOptions;
34 class IdentifierInfo;
35 class Preprocessor;
36 
37 /// \brief The preprocessor keeps track of this information for each
38 /// file that is \#included.
39 struct HeaderFileInfo {
40   /// \brief True if this is a \#import'd or \#pragma once file.
41   unsigned isImport : 1;
42 
43   /// \brief True if this is a \#pragma once file.
44   unsigned isPragmaOnce : 1;
45 
46   /// DirInfo - Keep track of whether this is a system header, and if so,
47   /// whether it is C++ clean or not.  This can be set by the include paths or
48   /// by \#pragma gcc system_header.  This is an instance of
49   /// SrcMgr::CharacteristicKind.
50   unsigned DirInfo : 2;
51 
52   /// \brief Whether this header file info was supplied by an external source,
53   /// and has not changed since.
54   unsigned External : 1;
55 
56   /// \brief Whether this header is part of a module.
57   unsigned isModuleHeader : 1;
58 
59   /// \brief Whether this header is part of the module that we are building.
60   unsigned isCompilingModuleHeader : 1;
61 
62   /// \brief Whether this structure is considered to already have been
63   /// "resolved", meaning that it was loaded from the external source.
64   unsigned Resolved : 1;
65 
66   /// \brief Whether this is a header inside a framework that is currently
67   /// being built.
68   ///
69   /// When a framework is being built, the headers have not yet been placed
70   /// into the appropriate framework subdirectories, and therefore are
71   /// provided via a header map. This bit indicates when this is one of
72   /// those framework headers.
73   unsigned IndexHeaderMapHeader : 1;
74 
75   /// \brief Whether this file has been looked up as a header.
76   unsigned IsValid : 1;
77 
78   /// \brief The number of times the file has been included already.
79   unsigned short NumIncludes;
80 
81   /// \brief The ID number of the controlling macro.
82   ///
83   /// This ID number will be non-zero when there is a controlling
84   /// macro whose IdentifierInfo may not yet have been loaded from
85   /// external storage.
86   unsigned ControllingMacroID;
87 
88   /// If this file has a \#ifndef XXX (or equivalent) guard that
89   /// protects the entire contents of the file, this is the identifier
90   /// for the macro that controls whether or not it has any effect.
91   ///
92   /// Note: Most clients should use getControllingMacro() to access
93   /// the controlling macro of this header, since
94   /// getControllingMacro() is able to load a controlling macro from
95   /// external storage.
96   const IdentifierInfo *ControllingMacro;
97 
98   /// \brief If this header came from a framework include, this is the name
99   /// of the framework.
100   StringRef Framework;
101 
HeaderFileInfoHeaderFileInfo102   HeaderFileInfo()
103     : isImport(false), isPragmaOnce(false), DirInfo(SrcMgr::C_User),
104       External(false), isModuleHeader(false), isCompilingModuleHeader(false),
105       Resolved(false), IndexHeaderMapHeader(false), IsValid(0),
106       NumIncludes(0), ControllingMacroID(0), ControllingMacro(nullptr)  {}
107 
108   /// \brief Retrieve the controlling macro for this header file, if
109   /// any.
110   const IdentifierInfo *
111   getControllingMacro(ExternalPreprocessorSource *External);
112 
113   /// \brief Determine whether this is a non-default header file info, e.g.,
114   /// it corresponds to an actual header we've included or tried to include.
isNonDefaultHeaderFileInfo115   bool isNonDefault() const {
116     return isImport || isPragmaOnce || NumIncludes || ControllingMacro ||
117       ControllingMacroID;
118   }
119 };
120 
121 /// \brief An external source of header file information, which may supply
122 /// information about header files already included.
123 class ExternalHeaderFileInfoSource {
124 public:
125   virtual ~ExternalHeaderFileInfoSource();
126 
127   /// \brief Retrieve the header file information for the given file entry.
128   ///
129   /// \returns Header file information for the given file entry, with the
130   /// \c External bit set. If the file entry is not known, return a
131   /// default-constructed \c HeaderFileInfo.
132   virtual HeaderFileInfo GetHeaderFileInfo(const FileEntry *FE) = 0;
133 };
134 
135 /// \brief Encapsulates the information needed to find the file referenced
136 /// by a \#include or \#include_next, (sub-)framework lookup, etc.
137 class HeaderSearch {
138   /// This structure is used to record entries in our framework cache.
139   struct FrameworkCacheEntry {
140     /// The directory entry which should be used for the cached framework.
141     const DirectoryEntry *Directory;
142 
143     /// Whether this framework has been "user-specified" to be treated as if it
144     /// were a system framework (even if it was found outside a system framework
145     /// directory).
146     bool IsUserSpecifiedSystemFramework;
147   };
148 
149   /// \brief Header-search options used to initialize this header search.
150   IntrusiveRefCntPtr<HeaderSearchOptions> HSOpts;
151 
152   DiagnosticsEngine &Diags;
153   FileManager &FileMgr;
154   /// \#include search path information.  Requests for \#include "x" search the
155   /// directory of the \#including file first, then each directory in SearchDirs
156   /// consecutively. Requests for <x> search the current dir first, then each
157   /// directory in SearchDirs, starting at AngledDirIdx, consecutively.  If
158   /// NoCurDirSearch is true, then the check for the file in the current
159   /// directory is suppressed.
160   std::vector<DirectoryLookup> SearchDirs;
161   unsigned AngledDirIdx;
162   unsigned SystemDirIdx;
163   bool NoCurDirSearch;
164 
165   /// \brief \#include prefixes for which the 'system header' property is
166   /// overridden.
167   ///
168   /// For a \#include "x" or \#include \<x> directive, the last string in this
169   /// list which is a prefix of 'x' determines whether the file is treated as
170   /// a system header.
171   std::vector<std::pair<std::string, bool> > SystemHeaderPrefixes;
172 
173   /// \brief The path to the module cache.
174   std::string ModuleCachePath;
175 
176   /// \brief All of the preprocessor-specific data about files that are
177   /// included, indexed by the FileEntry's UID.
178   mutable std::vector<HeaderFileInfo> FileInfo;
179 
180   /// Keeps track of each lookup performed by LookupFile.
181   struct LookupFileCacheInfo {
182     /// Starting index in SearchDirs that the cached search was performed from.
183     /// If there is a hit and this value doesn't match the current query, the
184     /// cache has to be ignored.
185     unsigned StartIdx;
186     /// The entry in SearchDirs that satisfied the query.
187     unsigned HitIdx;
188     /// This is non-null if the original filename was mapped to a framework
189     /// include via a headermap.
190     const char *MappedName;
191 
192     /// Default constructor -- Initialize all members with zero.
LookupFileCacheInfoLookupFileCacheInfo193     LookupFileCacheInfo(): StartIdx(0), HitIdx(0), MappedName(nullptr) {}
194 
resetLookupFileCacheInfo195     void reset(unsigned StartIdx) {
196       this->StartIdx = StartIdx;
197       this->MappedName = nullptr;
198     }
199   };
200   llvm::StringMap<LookupFileCacheInfo, llvm::BumpPtrAllocator> LookupFileCache;
201 
202   /// \brief Collection mapping a framework or subframework
203   /// name like "Carbon" to the Carbon.framework directory.
204   llvm::StringMap<FrameworkCacheEntry, llvm::BumpPtrAllocator> FrameworkMap;
205 
206   /// IncludeAliases - maps include file names (including the quotes or
207   /// angle brackets) to other include file names.  This is used to support the
208   /// include_alias pragma for Microsoft compatibility.
209   typedef llvm::StringMap<std::string, llvm::BumpPtrAllocator>
210     IncludeAliasMap;
211   std::unique_ptr<IncludeAliasMap> IncludeAliases;
212 
213   /// HeaderMaps - This is a mapping from FileEntry -> HeaderMap, uniquing
214   /// headermaps.  This vector owns the headermap.
215   std::vector<std::pair<const FileEntry*, const HeaderMap*> > HeaderMaps;
216 
217   /// \brief The mapping between modules and headers.
218   mutable ModuleMap ModMap;
219 
220   /// \brief Describes whether a given directory has a module map in it.
221   llvm::DenseMap<const DirectoryEntry *, bool> DirectoryHasModuleMap;
222 
223   /// \brief Set of module map files we've already loaded, and a flag indicating
224   /// whether they were valid or not.
225   llvm::DenseMap<const FileEntry *, bool> LoadedModuleMaps;
226 
227   /// \brief Uniqued set of framework names, which is used to track which
228   /// headers were included as framework headers.
229   llvm::StringSet<llvm::BumpPtrAllocator> FrameworkNames;
230 
231   /// \brief Entity used to resolve the identifier IDs of controlling
232   /// macros into IdentifierInfo pointers, and keep the identifire up to date,
233   /// as needed.
234   ExternalPreprocessorSource *ExternalLookup;
235 
236   /// \brief Entity used to look up stored header file information.
237   ExternalHeaderFileInfoSource *ExternalSource;
238 
239   // Various statistics we track for performance analysis.
240   unsigned NumIncluded;
241   unsigned NumMultiIncludeFileOptzn;
242   unsigned NumFrameworkLookups, NumSubFrameworkLookups;
243 
244   const LangOptions &LangOpts;
245 
246   // HeaderSearch doesn't support default or copy construction.
247   HeaderSearch(const HeaderSearch&) = delete;
248   void operator=(const HeaderSearch&) = delete;
249 
250   friend class DirectoryLookup;
251 
252 public:
253   HeaderSearch(IntrusiveRefCntPtr<HeaderSearchOptions> HSOpts,
254                SourceManager &SourceMgr, DiagnosticsEngine &Diags,
255                const LangOptions &LangOpts, const TargetInfo *Target);
256   ~HeaderSearch();
257 
258   /// \brief Retrieve the header-search options with which this header search
259   /// was initialized.
getHeaderSearchOpts()260   HeaderSearchOptions &getHeaderSearchOpts() const { return *HSOpts; }
261 
getFileMgr()262   FileManager &getFileMgr() const { return FileMgr; }
263 
264   /// \brief Interface for setting the file search paths.
SetSearchPaths(const std::vector<DirectoryLookup> & dirs,unsigned angledDirIdx,unsigned systemDirIdx,bool noCurDirSearch)265   void SetSearchPaths(const std::vector<DirectoryLookup> &dirs,
266                       unsigned angledDirIdx, unsigned systemDirIdx,
267                       bool noCurDirSearch) {
268     assert(angledDirIdx <= systemDirIdx && systemDirIdx <= dirs.size() &&
269         "Directory indicies are unordered");
270     SearchDirs = dirs;
271     AngledDirIdx = angledDirIdx;
272     SystemDirIdx = systemDirIdx;
273     NoCurDirSearch = noCurDirSearch;
274     //LookupFileCache.clear();
275   }
276 
277   /// \brief Add an additional search path.
AddSearchPath(const DirectoryLookup & dir,bool isAngled)278   void AddSearchPath(const DirectoryLookup &dir, bool isAngled) {
279     unsigned idx = isAngled ? SystemDirIdx : AngledDirIdx;
280     SearchDirs.insert(SearchDirs.begin() + idx, dir);
281     if (!isAngled)
282       AngledDirIdx++;
283     SystemDirIdx++;
284   }
285 
286   /// \brief Set the list of system header prefixes.
SetSystemHeaderPrefixes(ArrayRef<std::pair<std::string,bool>> P)287   void SetSystemHeaderPrefixes(ArrayRef<std::pair<std::string, bool> > P) {
288     SystemHeaderPrefixes.assign(P.begin(), P.end());
289   }
290 
291   /// \brief Checks whether the map exists or not.
HasIncludeAliasMap()292   bool HasIncludeAliasMap() const { return (bool)IncludeAliases; }
293 
294   /// \brief Map the source include name to the dest include name.
295   ///
296   /// The Source should include the angle brackets or quotes, the dest
297   /// should not.  This allows for distinction between <> and "" headers.
AddIncludeAlias(StringRef Source,StringRef Dest)298   void AddIncludeAlias(StringRef Source, StringRef Dest) {
299     if (!IncludeAliases)
300       IncludeAliases.reset(new IncludeAliasMap);
301     (*IncludeAliases)[Source] = Dest;
302   }
303 
304   /// MapHeaderToIncludeAlias - Maps one header file name to a different header
305   /// file name, for use with the include_alias pragma.  Note that the source
306   /// file name should include the angle brackets or quotes.  Returns StringRef
307   /// as null if the header cannot be mapped.
MapHeaderToIncludeAlias(StringRef Source)308   StringRef MapHeaderToIncludeAlias(StringRef Source) {
309     assert(IncludeAliases && "Trying to map headers when there's no map");
310 
311     // Do any filename replacements before anything else
312     IncludeAliasMap::const_iterator Iter = IncludeAliases->find(Source);
313     if (Iter != IncludeAliases->end())
314       return Iter->second;
315     return StringRef();
316   }
317 
318   /// \brief Set the path to the module cache.
setModuleCachePath(StringRef CachePath)319   void setModuleCachePath(StringRef CachePath) {
320     ModuleCachePath = CachePath;
321   }
322 
323   /// \brief Retrieve the path to the module cache.
getModuleCachePath()324   StringRef getModuleCachePath() const { return ModuleCachePath; }
325 
326   /// \brief Consider modules when including files from this directory.
setDirectoryHasModuleMap(const DirectoryEntry * Dir)327   void setDirectoryHasModuleMap(const DirectoryEntry* Dir) {
328     DirectoryHasModuleMap[Dir] = true;
329   }
330 
331   /// \brief Forget everything we know about headers so far.
ClearFileInfo()332   void ClearFileInfo() {
333     FileInfo.clear();
334   }
335 
SetExternalLookup(ExternalPreprocessorSource * EPS)336   void SetExternalLookup(ExternalPreprocessorSource *EPS) {
337     ExternalLookup = EPS;
338   }
339 
getExternalLookup()340   ExternalPreprocessorSource *getExternalLookup() const {
341     return ExternalLookup;
342   }
343 
344   /// \brief Set the external source of header information.
SetExternalSource(ExternalHeaderFileInfoSource * ES)345   void SetExternalSource(ExternalHeaderFileInfoSource *ES) {
346     ExternalSource = ES;
347   }
348 
349   /// \brief Set the target information for the header search, if not
350   /// already known.
351   void setTarget(const TargetInfo &Target);
352 
353   /// \brief Given a "foo" or \<foo> reference, look up the indicated file,
354   /// return null on failure.
355   ///
356   /// \returns If successful, this returns 'UsedDir', the DirectoryLookup member
357   /// the file was found in, or null if not applicable.
358   ///
359   /// \param IncludeLoc Used for diagnostics if valid.
360   ///
361   /// \param isAngled indicates whether the file reference is a <> reference.
362   ///
363   /// \param CurDir If non-null, the file was found in the specified directory
364   /// search location.  This is used to implement \#include_next.
365   ///
366   /// \param Includers Indicates where the \#including file(s) are, in case
367   /// relative searches are needed. In reverse order of inclusion.
368   ///
369   /// \param SearchPath If non-null, will be set to the search path relative
370   /// to which the file was found. If the include path is absolute, SearchPath
371   /// will be set to an empty string.
372   ///
373   /// \param RelativePath If non-null, will be set to the path relative to
374   /// SearchPath at which the file was found. This only differs from the
375   /// Filename for framework includes.
376   ///
377   /// \param SuggestedModule If non-null, and the file found is semantically
378   /// part of a known module, this will be set to the module that should
379   /// be imported instead of preprocessing/parsing the file found.
380   const FileEntry *LookupFile(
381       StringRef Filename, SourceLocation IncludeLoc, bool isAngled,
382       const DirectoryLookup *FromDir, const DirectoryLookup *&CurDir,
383       ArrayRef<std::pair<const FileEntry *, const DirectoryEntry *>> Includers,
384       SmallVectorImpl<char> *SearchPath, SmallVectorImpl<char> *RelativePath,
385       Module *RequestingModule, ModuleMap::KnownHeader *SuggestedModule,
386       bool SkipCache = false);
387 
388   /// \brief Look up a subframework for the specified \#include file.
389   ///
390   /// For example, if \#include'ing <HIToolbox/HIToolbox.h> from
391   /// within ".../Carbon.framework/Headers/Carbon.h", check to see if
392   /// HIToolbox is a subframework within Carbon.framework.  If so, return
393   /// the FileEntry for the designated file, otherwise return null.
394   const FileEntry *LookupSubframeworkHeader(
395       StringRef Filename, const FileEntry *RelativeFileEnt,
396       SmallVectorImpl<char> *SearchPath, SmallVectorImpl<char> *RelativePath,
397       Module *RequestingModule, ModuleMap::KnownHeader *SuggestedModule);
398 
399   /// \brief Look up the specified framework name in our framework cache.
400   /// \returns The DirectoryEntry it is in if we know, null otherwise.
LookupFrameworkCache(StringRef FWName)401   FrameworkCacheEntry &LookupFrameworkCache(StringRef FWName) {
402     return FrameworkMap[FWName];
403   }
404 
405   /// \brief Mark the specified file as a target of of a \#include,
406   /// \#include_next, or \#import directive.
407   ///
408   /// \return false if \#including the file will have no effect or true
409   /// if we should include it.
410   bool ShouldEnterIncludeFile(Preprocessor &PP, const FileEntry *File,
411                               bool isImport, Module *CorrespondingModule);
412 
413   /// \brief Return whether the specified file is a normal header,
414   /// a system header, or a C++ friendly system header.
getFileDirFlavor(const FileEntry * File)415   SrcMgr::CharacteristicKind getFileDirFlavor(const FileEntry *File) {
416     return (SrcMgr::CharacteristicKind)getFileInfo(File).DirInfo;
417   }
418 
419   /// \brief Mark the specified file as a "once only" file, e.g. due to
420   /// \#pragma once.
MarkFileIncludeOnce(const FileEntry * File)421   void MarkFileIncludeOnce(const FileEntry *File) {
422     HeaderFileInfo &FI = getFileInfo(File);
423     FI.isImport = true;
424     FI.isPragmaOnce = true;
425   }
426 
427   /// \brief Mark the specified file as a system header, e.g. due to
428   /// \#pragma GCC system_header.
MarkFileSystemHeader(const FileEntry * File)429   void MarkFileSystemHeader(const FileEntry *File) {
430     getFileInfo(File).DirInfo = SrcMgr::C_System;
431   }
432 
433   /// \brief Mark the specified file as part of a module.
434   void MarkFileModuleHeader(const FileEntry *File,
435                             ModuleMap::ModuleHeaderRole Role,
436                             bool IsCompiledModuleHeader);
437 
438   /// \brief Increment the count for the number of times the specified
439   /// FileEntry has been entered.
IncrementIncludeCount(const FileEntry * File)440   void IncrementIncludeCount(const FileEntry *File) {
441     ++getFileInfo(File).NumIncludes;
442   }
443 
444   /// \brief Mark the specified file as having a controlling macro.
445   ///
446   /// This is used by the multiple-include optimization to eliminate
447   /// no-op \#includes.
SetFileControllingMacro(const FileEntry * File,const IdentifierInfo * ControllingMacro)448   void SetFileControllingMacro(const FileEntry *File,
449                                const IdentifierInfo *ControllingMacro) {
450     getFileInfo(File).ControllingMacro = ControllingMacro;
451   }
452 
453   /// \brief Return true if this is the first time encountering this header.
FirstTimeLexingFile(const FileEntry * File)454   bool FirstTimeLexingFile(const FileEntry *File) {
455     return getFileInfo(File).NumIncludes == 1;
456   }
457 
458   /// \brief Determine whether this file is intended to be safe from
459   /// multiple inclusions, e.g., it has \#pragma once or a controlling
460   /// macro.
461   ///
462   /// This routine does not consider the effect of \#import
463   bool isFileMultipleIncludeGuarded(const FileEntry *File);
464 
465   /// CreateHeaderMap - This method returns a HeaderMap for the specified
466   /// FileEntry, uniquing them through the 'HeaderMaps' datastructure.
467   const HeaderMap *CreateHeaderMap(const FileEntry *FE);
468 
469   /// \brief Retrieve the name of the module file that should be used to
470   /// load the given module.
471   ///
472   /// \param Module The module whose module file name will be returned.
473   ///
474   /// \returns The name of the module file that corresponds to this module,
475   /// or an empty string if this module does not correspond to any module file.
476   std::string getModuleFileName(Module *Module);
477 
478   /// \brief Retrieve the name of the module file that should be used to
479   /// load a module with the given name.
480   ///
481   /// \param ModuleName The module whose module file name will be returned.
482   ///
483   /// \param ModuleMapPath A path that when combined with \c ModuleName
484   /// uniquely identifies this module. See Module::ModuleMap.
485   ///
486   /// \returns The name of the module file that corresponds to this module,
487   /// or an empty string if this module does not correspond to any module file.
488   std::string getModuleFileName(StringRef ModuleName, StringRef ModuleMapPath);
489 
490   /// \brief Lookup a module Search for a module with the given name.
491   ///
492   /// \param ModuleName The name of the module we're looking for.
493   ///
494   /// \param AllowSearch Whether we are allowed to search in the various
495   /// search directories to produce a module definition. If not, this lookup
496   /// will only return an already-known module.
497   ///
498   /// \returns The module with the given name.
499   Module *lookupModule(StringRef ModuleName, bool AllowSearch = true);
500 
501   /// \brief Try to find a module map file in the given directory, returning
502   /// \c nullptr if none is found.
503   const FileEntry *lookupModuleMapFile(const DirectoryEntry *Dir,
504                                        bool IsFramework);
505 
IncrementFrameworkLookupCount()506   void IncrementFrameworkLookupCount() { ++NumFrameworkLookups; }
507 
508   /// \brief Determine whether there is a module map that may map the header
509   /// with the given file name to a (sub)module.
510   /// Always returns false if modules are disabled.
511   ///
512   /// \param Filename The name of the file.
513   ///
514   /// \param Root The "root" directory, at which we should stop looking for
515   /// module maps.
516   ///
517   /// \param IsSystem Whether the directories we're looking at are system
518   /// header directories.
519   bool hasModuleMap(StringRef Filename, const DirectoryEntry *Root,
520                     bool IsSystem);
521 
522   /// \brief Retrieve the module that corresponds to the given file, if any.
523   ///
524   /// \param File The header that we wish to map to a module.
525   ModuleMap::KnownHeader findModuleForHeader(const FileEntry *File) const;
526 
527   /// \brief Read the contents of the given module map file.
528   ///
529   /// \param File The module map file.
530   /// \param IsSystem Whether this file is in a system header directory.
531   ///
532   /// \returns true if an error occurred, false otherwise.
533   bool loadModuleMapFile(const FileEntry *File, bool IsSystem);
534 
535   /// \brief Collect the set of all known, top-level modules.
536   ///
537   /// \param Modules Will be filled with the set of known, top-level modules.
538   void collectAllModules(SmallVectorImpl<Module *> &Modules);
539 
540   /// \brief Load all known, top-level system modules.
541   void loadTopLevelSystemModules();
542 
543 private:
544   /// \brief Retrieve a module with the given name, which may be part of the
545   /// given framework.
546   ///
547   /// \param Name The name of the module to retrieve.
548   ///
549   /// \param Dir The framework directory (e.g., ModuleName.framework).
550   ///
551   /// \param IsSystem Whether the framework directory is part of the system
552   /// frameworks.
553   ///
554   /// \returns The module, if found; otherwise, null.
555   Module *loadFrameworkModule(StringRef Name,
556                               const DirectoryEntry *Dir,
557                               bool IsSystem);
558 
559   /// \brief Load all of the module maps within the immediate subdirectories
560   /// of the given search directory.
561   void loadSubdirectoryModuleMaps(DirectoryLookup &SearchDir);
562 
563   /// \brief Find and suggest a usable module for the given file.
564   ///
565   /// \return \c true if the file can be used, \c false if we are not permitted to
566   ///         find this file due to requirements from \p RequestingModule.
567   bool findUsableModuleForHeader(const FileEntry *File,
568                                  const DirectoryEntry *Root,
569                                  Module *RequestingModule,
570                                  ModuleMap::KnownHeader *SuggestedModule,
571                                  bool IsSystemHeaderDir);
572 
573   /// \brief Find and suggest a usable module for the given file, which is part of
574   /// the specified framework.
575   ///
576   /// \return \c true if the file can be used, \c false if we are not permitted to
577   ///         find this file due to requirements from \p RequestingModule.
578   bool findUsableModuleForFrameworkHeader(
579       const FileEntry *File, StringRef FrameworkDir, Module *RequestingModule,
580       ModuleMap::KnownHeader *SuggestedModule, bool IsSystemFramework);
581 
582   /// \brief Look up the file with the specified name and determine its owning
583   /// module.
584   const FileEntry *
585   getFileAndSuggestModule(StringRef FileName, const DirectoryEntry *Dir,
586                           bool IsSystemHeaderDir, Module *RequestingModule,
587                           ModuleMap::KnownHeader *SuggestedModule);
588 
589 public:
590   /// \brief Retrieve the module map.
getModuleMap()591   ModuleMap &getModuleMap() { return ModMap; }
592 
593   /// \brief Retrieve the module map.
getModuleMap()594   const ModuleMap &getModuleMap() const { return ModMap; }
595 
header_file_size()596   unsigned header_file_size() const { return FileInfo.size(); }
597 
598   /// \brief Return the HeaderFileInfo structure for the specified FileEntry,
599   /// in preparation for updating it in some way.
600   HeaderFileInfo &getFileInfo(const FileEntry *FE);
601 
602   /// \brief Return the HeaderFileInfo structure for the specified FileEntry,
603   /// if it has ever been filled in.
604   /// \param WantExternal Whether the caller wants purely-external header file
605   ///        info (where \p External is true).
606   const HeaderFileInfo *getExistingFileInfo(const FileEntry *FE,
607                                             bool WantExternal = true) const;
608 
609   // Used by external tools
610   typedef std::vector<DirectoryLookup>::const_iterator search_dir_iterator;
search_dir_begin()611   search_dir_iterator search_dir_begin() const { return SearchDirs.begin(); }
search_dir_end()612   search_dir_iterator search_dir_end() const { return SearchDirs.end(); }
search_dir_size()613   unsigned search_dir_size() const { return SearchDirs.size(); }
614 
quoted_dir_begin()615   search_dir_iterator quoted_dir_begin() const {
616     return SearchDirs.begin();
617   }
quoted_dir_end()618   search_dir_iterator quoted_dir_end() const {
619     return SearchDirs.begin() + AngledDirIdx;
620   }
621 
angled_dir_begin()622   search_dir_iterator angled_dir_begin() const {
623     return SearchDirs.begin() + AngledDirIdx;
624   }
angled_dir_end()625   search_dir_iterator angled_dir_end() const {
626     return SearchDirs.begin() + SystemDirIdx;
627   }
628 
system_dir_begin()629   search_dir_iterator system_dir_begin() const {
630     return SearchDirs.begin() + SystemDirIdx;
631   }
system_dir_end()632   search_dir_iterator system_dir_end() const { return SearchDirs.end(); }
633 
634   /// \brief Retrieve a uniqued framework name.
635   StringRef getUniqueFrameworkName(StringRef Framework);
636 
637   void PrintStats();
638 
639   size_t getTotalMemory() const;
640 
641   static std::string NormalizeDashIncludePath(StringRef File,
642                                               FileManager &FileMgr);
643 
644 private:
645   /// \brief Describes what happened when we tried to load a module map file.
646   enum LoadModuleMapResult {
647     /// \brief The module map file had already been loaded.
648     LMM_AlreadyLoaded,
649     /// \brief The module map file was loaded by this invocation.
650     LMM_NewlyLoaded,
651     /// \brief There is was directory with the given name.
652     LMM_NoDirectory,
653     /// \brief There was either no module map file or the module map file was
654     /// invalid.
655     LMM_InvalidModuleMap
656   };
657 
658   LoadModuleMapResult loadModuleMapFileImpl(const FileEntry *File,
659                                             bool IsSystem,
660                                             const DirectoryEntry *Dir);
661 
662   /// \brief Try to load the module map file in the given directory.
663   ///
664   /// \param DirName The name of the directory where we will look for a module
665   /// map file.
666   /// \param IsSystem Whether this is a system header directory.
667   /// \param IsFramework Whether this is a framework directory.
668   ///
669   /// \returns The result of attempting to load the module map file from the
670   /// named directory.
671   LoadModuleMapResult loadModuleMapFile(StringRef DirName, bool IsSystem,
672                                         bool IsFramework);
673 
674   /// \brief Try to load the module map file in the given directory.
675   ///
676   /// \param Dir The directory where we will look for a module map file.
677   /// \param IsSystem Whether this is a system header directory.
678   /// \param IsFramework Whether this is a framework directory.
679   ///
680   /// \returns The result of attempting to load the module map file from the
681   /// named directory.
682   LoadModuleMapResult loadModuleMapFile(const DirectoryEntry *Dir,
683                                         bool IsSystem, bool IsFramework);
684 };
685 
686 }  // end namespace clang
687 
688 #endif
689