1 //===--- ASTReaderInternals.h - AST Reader Internals ------------*- 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 provides internal definitions used in the AST reader.
11 //
12 //===----------------------------------------------------------------------===//
13 #ifndef LLVM_CLANG_LIB_SERIALIZATION_ASTREADERINTERNALS_H
14 #define LLVM_CLANG_LIB_SERIALIZATION_ASTREADERINTERNALS_H
15 
16 #include "clang/AST/DeclarationName.h"
17 #include "clang/Serialization/ASTBitCodes.h"
18 #include "llvm/Support/Endian.h"
19 #include "llvm/Support/OnDiskHashTable.h"
20 #include <utility>
21 
22 namespace clang {
23 
24 class ASTReader;
25 class HeaderSearch;
26 struct HeaderFileInfo;
27 class FileEntry;
28 
29 namespace serialization {
30 
31 class ModuleFile;
32 
33 namespace reader {
34 
35 /// \brief Class that performs name lookup into a DeclContext stored
36 /// in an AST file.
37 class ASTDeclContextNameLookupTrait {
38   ASTReader &Reader;
39   ModuleFile &F;
40 
41 public:
42   /// \brief Pair of begin/end iterators for DeclIDs.
43   ///
44   /// Note that these declaration IDs are local to the module that contains this
45   /// particular lookup t
46   typedef llvm::support::ulittle32_t LE32DeclID;
47   typedef std::pair<LE32DeclID *, LE32DeclID *> data_type;
48   typedef unsigned hash_value_type;
49   typedef unsigned offset_type;
50 
51   /// \brief Special internal key for declaration names.
52   /// The hash table creates keys for comparison; we do not create
53   /// a DeclarationName for the internal key to avoid deserializing types.
54   struct DeclNameKey {
55     DeclarationName::NameKind Kind;
56     uint64_t Data;
DeclNameKeyDeclNameKey57     DeclNameKey() : Kind((DeclarationName::NameKind)0), Data(0) { }
58   };
59 
60   typedef DeclarationName external_key_type;
61   typedef DeclNameKey internal_key_type;
62 
ASTDeclContextNameLookupTrait(ASTReader & Reader,ModuleFile & F)63   explicit ASTDeclContextNameLookupTrait(ASTReader &Reader, ModuleFile &F)
64     : Reader(Reader), F(F) { }
65 
EqualKey(const internal_key_type & a,const internal_key_type & b)66   static bool EqualKey(const internal_key_type& a,
67                        const internal_key_type& b) {
68     return a.Kind == b.Kind && a.Data == b.Data;
69   }
70 
71   hash_value_type ComputeHash(const DeclNameKey &Key) const;
72   internal_key_type GetInternalKey(const external_key_type& Name) const;
73 
74   static std::pair<unsigned, unsigned>
75   ReadKeyDataLength(const unsigned char*& d);
76 
77   internal_key_type ReadKey(const unsigned char* d, unsigned);
78 
79   data_type ReadData(internal_key_type, const unsigned char* d,
80                      unsigned DataLen);
81 };
82 
83 /// \brief Base class for the trait describing the on-disk hash table for the
84 /// identifiers in an AST file.
85 ///
86 /// This class is not useful by itself; rather, it provides common
87 /// functionality for accessing the on-disk hash table of identifiers
88 /// in an AST file. Different subclasses customize that functionality
89 /// based on what information they are interested in. Those subclasses
90 /// must provide the \c data_type typedef and the ReadData operation,
91 /// only.
92 class ASTIdentifierLookupTraitBase {
93 public:
94   typedef StringRef external_key_type;
95   typedef StringRef internal_key_type;
96   typedef unsigned hash_value_type;
97   typedef unsigned offset_type;
98 
EqualKey(const internal_key_type & a,const internal_key_type & b)99   static bool EqualKey(const internal_key_type& a, const internal_key_type& b) {
100     return a == b;
101   }
102 
103   static hash_value_type ComputeHash(const internal_key_type& a);
104 
105   static std::pair<unsigned, unsigned>
106   ReadKeyDataLength(const unsigned char*& d);
107 
108   // This hopefully will just get inlined and removed by the optimizer.
109   static const internal_key_type&
GetInternalKey(const external_key_type & x)110   GetInternalKey(const external_key_type& x) { return x; }
111 
112   // This hopefully will just get inlined and removed by the optimizer.
113   static const external_key_type&
GetExternalKey(const internal_key_type & x)114   GetExternalKey(const internal_key_type& x) { return x; }
115 
116   static internal_key_type ReadKey(const unsigned char* d, unsigned n);
117 };
118 
119 /// \brief Class that performs lookup for an identifier stored in an AST file.
120 class ASTIdentifierLookupTrait : public ASTIdentifierLookupTraitBase {
121   ASTReader &Reader;
122   ModuleFile &F;
123 
124   // If we know the IdentifierInfo in advance, it is here and we will
125   // not build a new one. Used when deserializing information about an
126   // identifier that was constructed before the AST file was read.
127   IdentifierInfo *KnownII;
128 
129 public:
130   typedef IdentifierInfo * data_type;
131 
132   ASTIdentifierLookupTrait(ASTReader &Reader, ModuleFile &F,
133                            IdentifierInfo *II = nullptr)
Reader(Reader)134     : Reader(Reader), F(F), KnownII(II) { }
135 
136   data_type ReadData(const internal_key_type& k,
137                      const unsigned char* d,
138                      unsigned DataLen);
139 
getReader()140   ASTReader &getReader() const { return Reader; }
141 };
142 
143 /// \brief The on-disk hash table used to contain information about
144 /// all of the identifiers in the program.
145 typedef llvm::OnDiskIterableChainedHashTable<ASTIdentifierLookupTrait>
146   ASTIdentifierLookupTable;
147 
148 /// \brief Class that performs lookup for a selector's entries in the global
149 /// method pool stored in an AST file.
150 class ASTSelectorLookupTrait {
151   ASTReader &Reader;
152   ModuleFile &F;
153 
154 public:
155   struct data_type {
156     SelectorID ID;
157     unsigned InstanceBits;
158     unsigned FactoryBits;
159     bool InstanceHasMoreThanOneDecl;
160     bool FactoryHasMoreThanOneDecl;
161     SmallVector<ObjCMethodDecl *, 2> Instance;
162     SmallVector<ObjCMethodDecl *, 2> Factory;
163   };
164 
165   typedef Selector external_key_type;
166   typedef external_key_type internal_key_type;
167   typedef unsigned hash_value_type;
168   typedef unsigned offset_type;
169 
ASTSelectorLookupTrait(ASTReader & Reader,ModuleFile & F)170   ASTSelectorLookupTrait(ASTReader &Reader, ModuleFile &F)
171     : Reader(Reader), F(F) { }
172 
EqualKey(const internal_key_type & a,const internal_key_type & b)173   static bool EqualKey(const internal_key_type& a,
174                        const internal_key_type& b) {
175     return a == b;
176   }
177 
178   static hash_value_type ComputeHash(Selector Sel);
179 
180   static const internal_key_type&
GetInternalKey(const external_key_type & x)181   GetInternalKey(const external_key_type& x) { return x; }
182 
183   static std::pair<unsigned, unsigned>
184   ReadKeyDataLength(const unsigned char*& d);
185 
186   internal_key_type ReadKey(const unsigned char* d, unsigned);
187   data_type ReadData(Selector, const unsigned char* d, unsigned DataLen);
188 };
189 
190 /// \brief The on-disk hash table used for the global method pool.
191 typedef llvm::OnDiskChainedHashTable<ASTSelectorLookupTrait>
192   ASTSelectorLookupTable;
193 
194 /// \brief Trait class used to search the on-disk hash table containing all of
195 /// the header search information.
196 ///
197 /// The on-disk hash table contains a mapping from each header path to
198 /// information about that header (how many times it has been included, its
199 /// controlling macro, etc.). Note that we actually hash based on the size
200 /// and mtime, and support "deep" comparisons of file names based on current
201 /// inode numbers, so that the search can cope with non-normalized path names
202 /// and symlinks.
203 class HeaderFileInfoTrait {
204   ASTReader &Reader;
205   ModuleFile &M;
206   HeaderSearch *HS;
207   const char *FrameworkStrings;
208 
209 public:
210   typedef const FileEntry *external_key_type;
211 
212   struct internal_key_type {
213     off_t Size;
214     time_t ModTime;
215     const char *Filename;
216     bool Imported;
217   };
218   typedef const internal_key_type &internal_key_ref;
219 
220   typedef HeaderFileInfo data_type;
221   typedef unsigned hash_value_type;
222   typedef unsigned offset_type;
223 
HeaderFileInfoTrait(ASTReader & Reader,ModuleFile & M,HeaderSearch * HS,const char * FrameworkStrings)224   HeaderFileInfoTrait(ASTReader &Reader, ModuleFile &M, HeaderSearch *HS,
225                       const char *FrameworkStrings)
226   : Reader(Reader), M(M), HS(HS), FrameworkStrings(FrameworkStrings) { }
227 
228   static hash_value_type ComputeHash(internal_key_ref ikey);
229   static internal_key_type GetInternalKey(const FileEntry *FE);
230   bool EqualKey(internal_key_ref a, internal_key_ref b);
231 
232   static std::pair<unsigned, unsigned>
233   ReadKeyDataLength(const unsigned char*& d);
234 
235   static internal_key_type ReadKey(const unsigned char *d, unsigned);
236 
237   data_type ReadData(internal_key_ref,const unsigned char *d, unsigned DataLen);
238 };
239 
240 /// \brief The on-disk hash table used for known header files.
241 typedef llvm::OnDiskChainedHashTable<HeaderFileInfoTrait>
242   HeaderFileInfoLookupTable;
243 
244 } // end namespace clang::serialization::reader
245 } // end namespace clang::serialization
246 } // end namespace clang
247 
248 
249 #endif
250