]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/llvm/tools/clang/include/clang/Lex/HeaderSearch.h
Update clang to trunk r256633.
[FreeBSD/FreeBSD.git] / contrib / llvm / tools / clang / include / clang / Lex / HeaderSearch.h
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   
102   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.
115   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.
193     LookupFileCacheInfo(): StartIdx(0), HitIdx(0), MappedName(nullptr) {}
194
195     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.
260   HeaderSearchOptions &getHeaderSearchOpts() const { return *HSOpts; }
261   
262   FileManager &getFileMgr() const { return FileMgr; }
263
264   /// \brief Interface for setting the file search paths.
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.
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.
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.
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.
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.
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.
319   void setModuleCachePath(StringRef CachePath) {
320     ModuleCachePath = CachePath;
321   }
322   
323   /// \brief Retrieve the path to the module cache.
324   StringRef getModuleCachePath() const { return ModuleCachePath; }
325
326   /// \brief Consider modules when including files from this directory.
327   void setDirectoryHasModuleMap(const DirectoryEntry* Dir) {
328     DirectoryHasModuleMap[Dir] = true;
329   }
330   
331   /// \brief Forget everything we know about headers so far.
332   void ClearFileInfo() {
333     FileInfo.clear();
334   }
335
336   void SetExternalLookup(ExternalPreprocessorSource *EPS) {
337     ExternalLookup = EPS;
338   }
339
340   ExternalPreprocessorSource *getExternalLookup() const {
341     return ExternalLookup;
342   }
343   
344   /// \brief Set the external source of header information.
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.
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.
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.
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.
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.
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.
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.
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   
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.
591   ModuleMap &getModuleMap() { return ModMap; }
592   
593   /// \brief Retrieve the module map.
594   const ModuleMap &getModuleMap() const { return ModMap; }
595   
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;
611   search_dir_iterator search_dir_begin() const { return SearchDirs.begin(); }
612   search_dir_iterator search_dir_end() const { return SearchDirs.end(); }
613   unsigned search_dir_size() const { return SearchDirs.size(); }
614
615   search_dir_iterator quoted_dir_begin() const {
616     return SearchDirs.begin();
617   }
618   search_dir_iterator quoted_dir_end() const {
619     return SearchDirs.begin() + AngledDirIdx;
620   }
621
622   search_dir_iterator angled_dir_begin() const {
623     return SearchDirs.begin() + AngledDirIdx;
624   }
625   search_dir_iterator angled_dir_end() const {
626     return SearchDirs.begin() + SystemDirIdx;
627   }
628
629   search_dir_iterator system_dir_begin() const {
630     return SearchDirs.begin() + SystemDirIdx;
631   }
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