blob: b60e5f14a6769783b472664e5536de53bf37b04a [file] [log] [blame]
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001//===- ASTReader.cpp - AST File Reader ------------------------------------===//
Guy Benyei11169dd2012-12-18 14:30:41 +00002//
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 ASTReader class, which reads AST files.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/Serialization/ASTReader.h"
15#include "ASTCommon.h"
16#include "ASTReaderInternals.h"
17#include "clang/AST/ASTConsumer.h"
18#include "clang/AST/ASTContext.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000019#include "clang/AST/ASTMutationListener.h"
20#include "clang/AST/ASTUnresolvedSet.h"
21#include "clang/AST/Decl.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000022#include "clang/AST/DeclBase.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000023#include "clang/AST/DeclCXX.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000024#include "clang/AST/DeclFriend.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000025#include "clang/AST/DeclGroup.h"
26#include "clang/AST/DeclObjC.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000027#include "clang/AST/DeclTemplate.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000028#include "clang/AST/DeclarationName.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000029#include "clang/AST/Expr.h"
30#include "clang/AST/ExprCXX.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000031#include "clang/AST/ExternalASTSource.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000032#include "clang/AST/NestedNameSpecifier.h"
Richard Trieue7f7ed22017-02-22 01:11:25 +000033#include "clang/AST/ODRHash.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000034#include "clang/AST/RawCommentList.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000035#include "clang/AST/TemplateBase.h"
36#include "clang/AST/TemplateName.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000037#include "clang/AST/Type.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000038#include "clang/AST/TypeLoc.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000039#include "clang/AST/TypeLocVisitor.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000040#include "clang/AST/UnresolvedSet.h"
41#include "clang/Basic/CommentOptions.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000042#include "clang/Basic/Diagnostic.h"
Benjamin Kramerf3ca26982014-05-10 16:31:55 +000043#include "clang/Basic/DiagnosticOptions.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000044#include "clang/Basic/ExceptionSpecificationType.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000045#include "clang/Basic/FileManager.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000046#include "clang/Basic/FileSystemOptions.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000047#include "clang/Basic/IdentifierTable.h"
48#include "clang/Basic/LLVM.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000049#include "clang/Basic/LangOptions.h"
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +000050#include "clang/Basic/MemoryBufferCache.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000051#include "clang/Basic/Module.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000052#include "clang/Basic/ObjCRuntime.h"
53#include "clang/Basic/OperatorKinds.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000054#include "clang/Basic/PragmaKinds.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000055#include "clang/Basic/Sanitizers.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000056#include "clang/Basic/SourceLocation.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000057#include "clang/Basic/SourceManager.h"
58#include "clang/Basic/SourceManagerInternals.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000059#include "clang/Basic/Specifiers.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000060#include "clang/Basic/TargetInfo.h"
61#include "clang/Basic/TargetOptions.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000062#include "clang/Basic/TokenKinds.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000063#include "clang/Basic/Version.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000064#include "clang/Frontend/PCHContainerOperations.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000065#include "clang/Lex/HeaderSearch.h"
66#include "clang/Lex/HeaderSearchOptions.h"
67#include "clang/Lex/MacroInfo.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000068#include "clang/Lex/ModuleMap.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000069#include "clang/Lex/PreprocessingRecord.h"
70#include "clang/Lex/Preprocessor.h"
71#include "clang/Lex/PreprocessorOptions.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000072#include "clang/Lex/Token.h"
73#include "clang/Sema/ObjCMethodList.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000074#include "clang/Sema/Scope.h"
75#include "clang/Sema/Sema.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000076#include "clang/Sema/Weak.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000077#include "clang/Serialization/ASTBitCodes.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000078#include "clang/Serialization/ASTDeserializationListener.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000079#include "clang/Serialization/ContinuousRangeMap.h"
Douglas Gregore060e572013-01-25 01:03:03 +000080#include "clang/Serialization/GlobalModuleIndex.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000081#include "clang/Serialization/Module.h"
82#include "clang/Serialization/ModuleFileExtension.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000083#include "clang/Serialization/ModuleManager.h"
84#include "clang/Serialization/SerializationDiagnostic.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000085#include "llvm/ADT/APFloat.h"
86#include "llvm/ADT/APInt.h"
87#include "llvm/ADT/APSInt.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000088#include "llvm/ADT/ArrayRef.h"
89#include "llvm/ADT/DenseMap.h"
90#include "llvm/ADT/FoldingSet.h"
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +000091#include "llvm/ADT/Hashing.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000092#include "llvm/ADT/IntrusiveRefCntPtr.h"
93#include "llvm/ADT/None.h"
94#include "llvm/ADT/Optional.h"
95#include "llvm/ADT/STLExtras.h"
96#include "llvm/ADT/SmallPtrSet.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +000097#include "llvm/ADT/SmallString.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000098#include "llvm/ADT/SmallVector.h"
Vedant Kumar48b4f762018-04-14 01:40:48 +000099#include "llvm/ADT/StringExtras.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000100#include "llvm/ADT/StringMap.h"
101#include "llvm/ADT/StringRef.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000102#include "llvm/ADT/Triple.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000103#include "llvm/ADT/iterator_range.h"
Guy Benyei11169dd2012-12-18 14:30:41 +0000104#include "llvm/Bitcode/BitstreamReader.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000105#include "llvm/Support/Casting.h"
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000106#include "llvm/Support/Compiler.h"
Pavel Labathd8c62902018-06-11 10:28:04 +0000107#include "llvm/Support/Compression.h"
Jonas Devlieghere560ce2c2018-02-26 15:16:42 +0000108#include "llvm/Support/DJB.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000109#include "llvm/Support/Endian.h"
George Rimarc39f5492017-01-17 15:45:31 +0000110#include "llvm/Support/Error.h"
Guy Benyei11169dd2012-12-18 14:30:41 +0000111#include "llvm/Support/ErrorHandling.h"
112#include "llvm/Support/FileSystem.h"
113#include "llvm/Support/MemoryBuffer.h"
114#include "llvm/Support/Path.h"
115#include "llvm/Support/SaveAndRestore.h"
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000116#include "llvm/Support/Timer.h"
Pavel Labathd8c62902018-06-11 10:28:04 +0000117#include "llvm/Support/VersionTuple.h"
Dmitri Gribenkof430da42014-02-12 10:33:14 +0000118#include "llvm/Support/raw_ostream.h"
Guy Benyei11169dd2012-12-18 14:30:41 +0000119#include <algorithm>
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000120#include <cassert>
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000121#include <cstddef>
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000122#include <cstdint>
Chris Lattner91f373e2013-01-20 00:57:52 +0000123#include <cstdio>
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000124#include <ctime>
Guy Benyei11169dd2012-12-18 14:30:41 +0000125#include <iterator>
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000126#include <limits>
127#include <map>
128#include <memory>
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000129#include <string>
Rafael Espindola8a8e5542014-06-12 17:19:42 +0000130#include <system_error>
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000131#include <tuple>
132#include <utility>
133#include <vector>
Guy Benyei11169dd2012-12-18 14:30:41 +0000134
135using namespace clang;
Vedant Kumar48b4f762018-04-14 01:40:48 +0000136using namespace clang::serialization;
137using namespace clang::serialization::reader;
Chris Lattner7fb3bef2013-01-20 00:56:42 +0000138using llvm::BitstreamCursor;
Guy Benyei11169dd2012-12-18 14:30:41 +0000139
Ben Langmuircb69b572014-03-07 06:40:32 +0000140//===----------------------------------------------------------------------===//
141// ChainedASTReaderListener implementation
142//===----------------------------------------------------------------------===//
143
144bool
145ChainedASTReaderListener::ReadFullVersionInformation(StringRef FullVersion) {
146 return First->ReadFullVersionInformation(FullVersion) ||
147 Second->ReadFullVersionInformation(FullVersion);
148}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000149
Ben Langmuir4f5212a2014-04-14 22:12:44 +0000150void ChainedASTReaderListener::ReadModuleName(StringRef ModuleName) {
151 First->ReadModuleName(ModuleName);
152 Second->ReadModuleName(ModuleName);
153}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000154
Ben Langmuir4f5212a2014-04-14 22:12:44 +0000155void ChainedASTReaderListener::ReadModuleMapFile(StringRef ModuleMapPath) {
156 First->ReadModuleMapFile(ModuleMapPath);
157 Second->ReadModuleMapFile(ModuleMapPath);
158}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000159
Richard Smith1e2cf0d2014-10-31 02:28:58 +0000160bool
161ChainedASTReaderListener::ReadLanguageOptions(const LangOptions &LangOpts,
162 bool Complain,
163 bool AllowCompatibleDifferences) {
164 return First->ReadLanguageOptions(LangOpts, Complain,
165 AllowCompatibleDifferences) ||
166 Second->ReadLanguageOptions(LangOpts, Complain,
167 AllowCompatibleDifferences);
Ben Langmuircb69b572014-03-07 06:40:32 +0000168}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000169
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000170bool ChainedASTReaderListener::ReadTargetOptions(
171 const TargetOptions &TargetOpts, bool Complain,
172 bool AllowCompatibleDifferences) {
173 return First->ReadTargetOptions(TargetOpts, Complain,
174 AllowCompatibleDifferences) ||
175 Second->ReadTargetOptions(TargetOpts, Complain,
176 AllowCompatibleDifferences);
Ben Langmuircb69b572014-03-07 06:40:32 +0000177}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000178
Ben Langmuircb69b572014-03-07 06:40:32 +0000179bool ChainedASTReaderListener::ReadDiagnosticOptions(
Ben Langmuirb92de022014-04-29 16:25:26 +0000180 IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts, bool Complain) {
Ben Langmuircb69b572014-03-07 06:40:32 +0000181 return First->ReadDiagnosticOptions(DiagOpts, Complain) ||
182 Second->ReadDiagnosticOptions(DiagOpts, Complain);
183}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000184
Ben Langmuircb69b572014-03-07 06:40:32 +0000185bool
186ChainedASTReaderListener::ReadFileSystemOptions(const FileSystemOptions &FSOpts,
187 bool Complain) {
188 return First->ReadFileSystemOptions(FSOpts, Complain) ||
189 Second->ReadFileSystemOptions(FSOpts, Complain);
190}
191
192bool ChainedASTReaderListener::ReadHeaderSearchOptions(
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +0000193 const HeaderSearchOptions &HSOpts, StringRef SpecificModuleCachePath,
194 bool Complain) {
195 return First->ReadHeaderSearchOptions(HSOpts, SpecificModuleCachePath,
196 Complain) ||
197 Second->ReadHeaderSearchOptions(HSOpts, SpecificModuleCachePath,
198 Complain);
Ben Langmuircb69b572014-03-07 06:40:32 +0000199}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000200
Ben Langmuircb69b572014-03-07 06:40:32 +0000201bool ChainedASTReaderListener::ReadPreprocessorOptions(
202 const PreprocessorOptions &PPOpts, bool Complain,
203 std::string &SuggestedPredefines) {
204 return First->ReadPreprocessorOptions(PPOpts, Complain,
205 SuggestedPredefines) ||
206 Second->ReadPreprocessorOptions(PPOpts, Complain, SuggestedPredefines);
207}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000208
Ben Langmuircb69b572014-03-07 06:40:32 +0000209void ChainedASTReaderListener::ReadCounter(const serialization::ModuleFile &M,
210 unsigned Value) {
211 First->ReadCounter(M, Value);
212 Second->ReadCounter(M, Value);
213}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000214
Ben Langmuircb69b572014-03-07 06:40:32 +0000215bool ChainedASTReaderListener::needsInputFileVisitation() {
216 return First->needsInputFileVisitation() ||
217 Second->needsInputFileVisitation();
218}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000219
Ben Langmuircb69b572014-03-07 06:40:32 +0000220bool ChainedASTReaderListener::needsSystemInputFileVisitation() {
221 return First->needsSystemInputFileVisitation() ||
222 Second->needsSystemInputFileVisitation();
223}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000224
Richard Smith216a3bd2015-08-13 17:57:10 +0000225void ChainedASTReaderListener::visitModuleFile(StringRef Filename,
226 ModuleKind Kind) {
227 First->visitModuleFile(Filename, Kind);
228 Second->visitModuleFile(Filename, Kind);
Argyrios Kyrtzidis6d0753d2014-03-14 03:07:38 +0000229}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000230
Ben Langmuircb69b572014-03-07 06:40:32 +0000231bool ChainedASTReaderListener::visitInputFile(StringRef Filename,
Argyrios Kyrtzidis68ccbe02014-03-14 02:26:31 +0000232 bool isSystem,
Richard Smith216a3bd2015-08-13 17:57:10 +0000233 bool isOverridden,
234 bool isExplicitModule) {
Justin Bognerc65a66d2014-05-22 06:04:59 +0000235 bool Continue = false;
236 if (First->needsInputFileVisitation() &&
237 (!isSystem || First->needsSystemInputFileVisitation()))
Richard Smith216a3bd2015-08-13 17:57:10 +0000238 Continue |= First->visitInputFile(Filename, isSystem, isOverridden,
239 isExplicitModule);
Justin Bognerc65a66d2014-05-22 06:04:59 +0000240 if (Second->needsInputFileVisitation() &&
241 (!isSystem || Second->needsSystemInputFileVisitation()))
Richard Smith216a3bd2015-08-13 17:57:10 +0000242 Continue |= Second->visitInputFile(Filename, isSystem, isOverridden,
243 isExplicitModule);
Justin Bognerc65a66d2014-05-22 06:04:59 +0000244 return Continue;
Ben Langmuircb69b572014-03-07 06:40:32 +0000245}
246
Douglas Gregor6623e1f2015-11-03 18:33:07 +0000247void ChainedASTReaderListener::readModuleFileExtension(
248 const ModuleFileExtensionMetadata &Metadata) {
249 First->readModuleFileExtension(Metadata);
250 Second->readModuleFileExtension(Metadata);
251}
252
Guy Benyei11169dd2012-12-18 14:30:41 +0000253//===----------------------------------------------------------------------===//
254// PCH validator implementation
255//===----------------------------------------------------------------------===//
256
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000257ASTReaderListener::~ASTReaderListener() = default;
Guy Benyei11169dd2012-12-18 14:30:41 +0000258
Adrian Prantl9fc8faf2018-05-09 01:00:01 +0000259/// Compare the given set of language options against an existing set of
Guy Benyei11169dd2012-12-18 14:30:41 +0000260/// language options.
261///
262/// \param Diags If non-NULL, diagnostics will be emitted via this engine.
Richard Smith1e2cf0d2014-10-31 02:28:58 +0000263/// \param AllowCompatibleDifferences If true, differences between compatible
264/// language options will be permitted.
Guy Benyei11169dd2012-12-18 14:30:41 +0000265///
266/// \returns true if the languagae options mis-match, false otherwise.
267static bool checkLanguageOptions(const LangOptions &LangOpts,
268 const LangOptions &ExistingLangOpts,
Richard Smith1e2cf0d2014-10-31 02:28:58 +0000269 DiagnosticsEngine *Diags,
270 bool AllowCompatibleDifferences = true) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000271#define LANGOPT(Name, Bits, Default, Description) \
272 if (ExistingLangOpts.Name != LangOpts.Name) { \
273 if (Diags) \
274 Diags->Report(diag::err_pch_langopt_mismatch) \
275 << Description << LangOpts.Name << ExistingLangOpts.Name; \
276 return true; \
277 }
278
279#define VALUE_LANGOPT(Name, Bits, Default, Description) \
280 if (ExistingLangOpts.Name != LangOpts.Name) { \
281 if (Diags) \
282 Diags->Report(diag::err_pch_langopt_value_mismatch) \
283 << Description; \
284 return true; \
285 }
286
287#define ENUM_LANGOPT(Name, Type, Bits, Default, Description) \
288 if (ExistingLangOpts.get##Name() != LangOpts.get##Name()) { \
289 if (Diags) \
290 Diags->Report(diag::err_pch_langopt_value_mismatch) \
291 << Description; \
292 return true; \
293 }
294
Richard Smith1e2cf0d2014-10-31 02:28:58 +0000295#define COMPATIBLE_LANGOPT(Name, Bits, Default, Description) \
296 if (!AllowCompatibleDifferences) \
297 LANGOPT(Name, Bits, Default, Description)
298
299#define COMPATIBLE_ENUM_LANGOPT(Name, Bits, Default, Description) \
300 if (!AllowCompatibleDifferences) \
301 ENUM_LANGOPT(Name, Bits, Default, Description)
302
Richard Smitha1ddf5e2016-04-07 20:47:37 +0000303#define COMPATIBLE_VALUE_LANGOPT(Name, Bits, Default, Description) \
304 if (!AllowCompatibleDifferences) \
305 VALUE_LANGOPT(Name, Bits, Default, Description)
306
Guy Benyei11169dd2012-12-18 14:30:41 +0000307#define BENIGN_LANGOPT(Name, Bits, Default, Description)
308#define BENIGN_ENUM_LANGOPT(Name, Type, Bits, Default, Description)
Richard Smitha1ddf5e2016-04-07 20:47:37 +0000309#define BENIGN_VALUE_LANGOPT(Name, Type, Bits, Default, Description)
Guy Benyei11169dd2012-12-18 14:30:41 +0000310#include "clang/Basic/LangOptions.def"
311
Ben Langmuircd98cb72015-06-23 18:20:18 +0000312 if (ExistingLangOpts.ModuleFeatures != LangOpts.ModuleFeatures) {
313 if (Diags)
314 Diags->Report(diag::err_pch_langopt_value_mismatch) << "module features";
315 return true;
316 }
317
Guy Benyei11169dd2012-12-18 14:30:41 +0000318 if (ExistingLangOpts.ObjCRuntime != LangOpts.ObjCRuntime) {
319 if (Diags)
320 Diags->Report(diag::err_pch_langopt_value_mismatch)
321 << "target Objective-C runtime";
322 return true;
323 }
324
Dmitri Gribenkoacf2e782013-02-22 14:21:27 +0000325 if (ExistingLangOpts.CommentOpts.BlockCommandNames !=
326 LangOpts.CommentOpts.BlockCommandNames) {
327 if (Diags)
328 Diags->Report(diag::err_pch_langopt_value_mismatch)
329 << "block command names";
330 return true;
331 }
332
Vedant Kumar85a83c22017-06-01 20:01:01 +0000333 // Sanitizer feature mismatches are treated as compatible differences. If
334 // compatible differences aren't allowed, we still only want to check for
335 // mismatches of non-modular sanitizers (the only ones which can affect AST
336 // generation).
337 if (!AllowCompatibleDifferences) {
338 SanitizerMask ModularSanitizers = getPPTransparentSanitizers();
339 SanitizerSet ExistingSanitizers = ExistingLangOpts.Sanitize;
340 SanitizerSet ImportedSanitizers = LangOpts.Sanitize;
341 ExistingSanitizers.clear(ModularSanitizers);
342 ImportedSanitizers.clear(ModularSanitizers);
343 if (ExistingSanitizers.Mask != ImportedSanitizers.Mask) {
344 const std::string Flag = "-fsanitize=";
345 if (Diags) {
346#define SANITIZER(NAME, ID) \
347 { \
348 bool InExistingModule = ExistingSanitizers.has(SanitizerKind::ID); \
349 bool InImportedModule = ImportedSanitizers.has(SanitizerKind::ID); \
350 if (InExistingModule != InImportedModule) \
351 Diags->Report(diag::err_pch_targetopt_feature_mismatch) \
352 << InExistingModule << (Flag + NAME); \
353 }
354#include "clang/Basic/Sanitizers.def"
355 }
356 return true;
357 }
358 }
359
Guy Benyei11169dd2012-12-18 14:30:41 +0000360 return false;
361}
362
Adrian Prantl9fc8faf2018-05-09 01:00:01 +0000363/// Compare the given set of target options against an existing set of
Guy Benyei11169dd2012-12-18 14:30:41 +0000364/// target options.
365///
366/// \param Diags If non-NULL, diagnostics will be emitted via this engine.
367///
368/// \returns true if the target options mis-match, false otherwise.
369static bool checkTargetOptions(const TargetOptions &TargetOpts,
370 const TargetOptions &ExistingTargetOpts,
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000371 DiagnosticsEngine *Diags,
372 bool AllowCompatibleDifferences = true) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000373#define CHECK_TARGET_OPT(Field, Name) \
374 if (TargetOpts.Field != ExistingTargetOpts.Field) { \
375 if (Diags) \
376 Diags->Report(diag::err_pch_targetopt_mismatch) \
377 << Name << TargetOpts.Field << ExistingTargetOpts.Field; \
378 return true; \
379 }
380
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000381 // The triple and ABI must match exactly.
Guy Benyei11169dd2012-12-18 14:30:41 +0000382 CHECK_TARGET_OPT(Triple, "target");
Guy Benyei11169dd2012-12-18 14:30:41 +0000383 CHECK_TARGET_OPT(ABI, "target ABI");
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000384
385 // We can tolerate different CPUs in many cases, notably when one CPU
386 // supports a strict superset of another. When allowing compatible
387 // differences skip this check.
388 if (!AllowCompatibleDifferences)
389 CHECK_TARGET_OPT(CPU, "target CPU");
390
Guy Benyei11169dd2012-12-18 14:30:41 +0000391#undef CHECK_TARGET_OPT
392
393 // Compare feature sets.
394 SmallVector<StringRef, 4> ExistingFeatures(
395 ExistingTargetOpts.FeaturesAsWritten.begin(),
396 ExistingTargetOpts.FeaturesAsWritten.end());
397 SmallVector<StringRef, 4> ReadFeatures(TargetOpts.FeaturesAsWritten.begin(),
398 TargetOpts.FeaturesAsWritten.end());
Mandeep Singh Grangc205d8c2018-03-27 16:50:00 +0000399 llvm::sort(ExistingFeatures.begin(), ExistingFeatures.end());
400 llvm::sort(ReadFeatures.begin(), ReadFeatures.end());
Guy Benyei11169dd2012-12-18 14:30:41 +0000401
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000402 // We compute the set difference in both directions explicitly so that we can
403 // diagnose the differences differently.
404 SmallVector<StringRef, 4> UnmatchedExistingFeatures, UnmatchedReadFeatures;
405 std::set_difference(
406 ExistingFeatures.begin(), ExistingFeatures.end(), ReadFeatures.begin(),
407 ReadFeatures.end(), std::back_inserter(UnmatchedExistingFeatures));
408 std::set_difference(ReadFeatures.begin(), ReadFeatures.end(),
409 ExistingFeatures.begin(), ExistingFeatures.end(),
410 std::back_inserter(UnmatchedReadFeatures));
Guy Benyei11169dd2012-12-18 14:30:41 +0000411
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000412 // If we are allowing compatible differences and the read feature set is
413 // a strict subset of the existing feature set, there is nothing to diagnose.
414 if (AllowCompatibleDifferences && UnmatchedReadFeatures.empty())
415 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000416
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000417 if (Diags) {
418 for (StringRef Feature : UnmatchedReadFeatures)
Guy Benyei11169dd2012-12-18 14:30:41 +0000419 Diags->Report(diag::err_pch_targetopt_feature_mismatch)
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000420 << /* is-existing-feature */ false << Feature;
421 for (StringRef Feature : UnmatchedExistingFeatures)
422 Diags->Report(diag::err_pch_targetopt_feature_mismatch)
423 << /* is-existing-feature */ true << Feature;
Guy Benyei11169dd2012-12-18 14:30:41 +0000424 }
425
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000426 return !UnmatchedReadFeatures.empty() || !UnmatchedExistingFeatures.empty();
Guy Benyei11169dd2012-12-18 14:30:41 +0000427}
428
429bool
430PCHValidator::ReadLanguageOptions(const LangOptions &LangOpts,
Richard Smith1e2cf0d2014-10-31 02:28:58 +0000431 bool Complain,
432 bool AllowCompatibleDifferences) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000433 const LangOptions &ExistingLangOpts = PP.getLangOpts();
434 return checkLanguageOptions(LangOpts, ExistingLangOpts,
Richard Smith1e2cf0d2014-10-31 02:28:58 +0000435 Complain ? &Reader.Diags : nullptr,
436 AllowCompatibleDifferences);
Guy Benyei11169dd2012-12-18 14:30:41 +0000437}
438
439bool PCHValidator::ReadTargetOptions(const TargetOptions &TargetOpts,
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000440 bool Complain,
441 bool AllowCompatibleDifferences) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000442 const TargetOptions &ExistingTargetOpts = PP.getTargetInfo().getTargetOpts();
443 return checkTargetOptions(TargetOpts, ExistingTargetOpts,
Chandler Carruth0d745bc2015-03-14 04:47:43 +0000444 Complain ? &Reader.Diags : nullptr,
445 AllowCompatibleDifferences);
Guy Benyei11169dd2012-12-18 14:30:41 +0000446}
447
448namespace {
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000449
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000450using MacroDefinitionsMap =
451 llvm::StringMap<std::pair<StringRef, bool /*IsUndef*/>>;
452using DeclsMap = llvm::DenseMap<DeclarationName, SmallVector<NamedDecl *, 8>>;
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +0000453
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000454} // namespace
Guy Benyei11169dd2012-12-18 14:30:41 +0000455
Ben Langmuirb92de022014-04-29 16:25:26 +0000456static bool checkDiagnosticGroupMappings(DiagnosticsEngine &StoredDiags,
457 DiagnosticsEngine &Diags,
458 bool Complain) {
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000459 using Level = DiagnosticsEngine::Level;
Ben Langmuirb92de022014-04-29 16:25:26 +0000460
461 // Check current mappings for new -Werror mappings, and the stored mappings
462 // for cases that were explicitly mapped to *not* be errors that are now
463 // errors because of options like -Werror.
464 DiagnosticsEngine *MappingSources[] = { &Diags, &StoredDiags };
465
Vedant Kumar48b4f762018-04-14 01:40:48 +0000466 for (DiagnosticsEngine *MappingSource : MappingSources) {
Ben Langmuirb92de022014-04-29 16:25:26 +0000467 for (auto DiagIDMappingPair : MappingSource->getDiagnosticMappings()) {
468 diag::kind DiagID = DiagIDMappingPair.first;
469 Level CurLevel = Diags.getDiagnosticLevel(DiagID, SourceLocation());
470 if (CurLevel < DiagnosticsEngine::Error)
471 continue; // not significant
472 Level StoredLevel =
473 StoredDiags.getDiagnosticLevel(DiagID, SourceLocation());
474 if (StoredLevel < DiagnosticsEngine::Error) {
475 if (Complain)
476 Diags.Report(diag::err_pch_diagopt_mismatch) << "-Werror=" +
477 Diags.getDiagnosticIDs()->getWarningOptionForDiag(DiagID).str();
478 return true;
479 }
480 }
481 }
482
483 return false;
484}
485
Alp Tokerac4e8e52014-06-22 21:58:33 +0000486static bool isExtHandlingFromDiagsError(DiagnosticsEngine &Diags) {
487 diag::Severity Ext = Diags.getExtensionHandlingBehavior();
488 if (Ext == diag::Severity::Warning && Diags.getWarningsAsErrors())
489 return true;
490 return Ext >= diag::Severity::Error;
Ben Langmuirb92de022014-04-29 16:25:26 +0000491}
492
493static bool checkDiagnosticMappings(DiagnosticsEngine &StoredDiags,
494 DiagnosticsEngine &Diags,
495 bool IsSystem, bool Complain) {
496 // Top-level options
497 if (IsSystem) {
498 if (Diags.getSuppressSystemWarnings())
499 return false;
500 // If -Wsystem-headers was not enabled before, be conservative
501 if (StoredDiags.getSuppressSystemWarnings()) {
502 if (Complain)
503 Diags.Report(diag::err_pch_diagopt_mismatch) << "-Wsystem-headers";
504 return true;
505 }
506 }
507
508 if (Diags.getWarningsAsErrors() && !StoredDiags.getWarningsAsErrors()) {
509 if (Complain)
510 Diags.Report(diag::err_pch_diagopt_mismatch) << "-Werror";
511 return true;
512 }
513
514 if (Diags.getWarningsAsErrors() && Diags.getEnableAllWarnings() &&
515 !StoredDiags.getEnableAllWarnings()) {
516 if (Complain)
517 Diags.Report(diag::err_pch_diagopt_mismatch) << "-Weverything -Werror";
518 return true;
519 }
520
521 if (isExtHandlingFromDiagsError(Diags) &&
522 !isExtHandlingFromDiagsError(StoredDiags)) {
523 if (Complain)
524 Diags.Report(diag::err_pch_diagopt_mismatch) << "-pedantic-errors";
525 return true;
526 }
527
528 return checkDiagnosticGroupMappings(StoredDiags, Diags, Complain);
529}
530
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +0000531/// Return the top import module if it is implicit, nullptr otherwise.
532static Module *getTopImportImplicitModule(ModuleManager &ModuleMgr,
533 Preprocessor &PP) {
534 // If the original import came from a file explicitly generated by the user,
535 // don't check the diagnostic mappings.
536 // FIXME: currently this is approximated by checking whether this is not a
537 // module import of an implicitly-loaded module file.
538 // Note: ModuleMgr.rbegin() may not be the current module, but it must be in
539 // the transitive closure of its imports, since unrelated modules cannot be
540 // imported until after this module finishes validation.
541 ModuleFile *TopImport = &*ModuleMgr.rbegin();
542 while (!TopImport->ImportedBy.empty())
543 TopImport = TopImport->ImportedBy[0];
544 if (TopImport->Kind != MK_ImplicitModule)
545 return nullptr;
546
547 StringRef ModuleName = TopImport->ModuleName;
548 assert(!ModuleName.empty() && "diagnostic options read before module name");
549
550 Module *M = PP.getHeaderSearchInfo().lookupModule(ModuleName);
551 assert(M && "missing module");
552 return M;
553}
554
Ben Langmuirb92de022014-04-29 16:25:26 +0000555bool PCHValidator::ReadDiagnosticOptions(
556 IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts, bool Complain) {
557 DiagnosticsEngine &ExistingDiags = PP.getDiagnostics();
558 IntrusiveRefCntPtr<DiagnosticIDs> DiagIDs(ExistingDiags.getDiagnosticIDs());
559 IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
Alp Tokerf994cef2014-07-05 03:08:06 +0000560 new DiagnosticsEngine(DiagIDs, DiagOpts.get()));
Ben Langmuirb92de022014-04-29 16:25:26 +0000561 // This should never fail, because we would have processed these options
562 // before writing them to an ASTFile.
563 ProcessWarningOptions(*Diags, *DiagOpts, /*Report*/false);
564
565 ModuleManager &ModuleMgr = Reader.getModuleManager();
566 assert(ModuleMgr.size() >= 1 && "what ASTFile is this then");
567
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +0000568 Module *TopM = getTopImportImplicitModule(ModuleMgr, PP);
569 if (!TopM)
Ben Langmuirb92de022014-04-29 16:25:26 +0000570 return false;
571
Ben Langmuirb92de022014-04-29 16:25:26 +0000572 // FIXME: if the diagnostics are incompatible, save a DiagnosticOptions that
573 // contains the union of their flags.
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +0000574 return checkDiagnosticMappings(*Diags, ExistingDiags, TopM->IsSystem,
575 Complain);
Ben Langmuirb92de022014-04-29 16:25:26 +0000576}
577
Adrian Prantl9fc8faf2018-05-09 01:00:01 +0000578/// Collect the macro definitions provided by the given preprocessor
Guy Benyei11169dd2012-12-18 14:30:41 +0000579/// options.
Craig Toppera13603a2014-05-22 05:54:18 +0000580static void
581collectMacroDefinitions(const PreprocessorOptions &PPOpts,
582 MacroDefinitionsMap &Macros,
583 SmallVectorImpl<StringRef> *MacroNames = nullptr) {
Vedant Kumar48b4f762018-04-14 01:40:48 +0000584 for (unsigned I = 0, N = PPOpts.Macros.size(); I != N; ++I) {
585 StringRef Macro = PPOpts.Macros[I].first;
586 bool IsUndef = PPOpts.Macros[I].second;
Guy Benyei11169dd2012-12-18 14:30:41 +0000587
588 std::pair<StringRef, StringRef> MacroPair = Macro.split('=');
589 StringRef MacroName = MacroPair.first;
590 StringRef MacroBody = MacroPair.second;
591
592 // For an #undef'd macro, we only care about the name.
593 if (IsUndef) {
594 if (MacroNames && !Macros.count(MacroName))
595 MacroNames->push_back(MacroName);
596
597 Macros[MacroName] = std::make_pair("", true);
598 continue;
599 }
600
601 // For a #define'd macro, figure out the actual definition.
602 if (MacroName.size() == Macro.size())
603 MacroBody = "1";
604 else {
605 // Note: GCC drops anything following an end-of-line character.
606 StringRef::size_type End = MacroBody.find_first_of("\n\r");
607 MacroBody = MacroBody.substr(0, End);
608 }
609
610 if (MacroNames && !Macros.count(MacroName))
611 MacroNames->push_back(MacroName);
612 Macros[MacroName] = std::make_pair(MacroBody, false);
613 }
614}
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +0000615
Adrian Prantl9fc8faf2018-05-09 01:00:01 +0000616/// Check the preprocessor options deserialized from the control block
Guy Benyei11169dd2012-12-18 14:30:41 +0000617/// against the preprocessor options in an existing preprocessor.
618///
619/// \param Diags If non-null, produce diagnostics for any mismatches incurred.
Yaxun Liu43712e02016-09-07 18:40:20 +0000620/// \param Validate If true, validate preprocessor options. If false, allow
621/// macros defined by \p ExistingPPOpts to override those defined by
622/// \p PPOpts in SuggestedPredefines.
Guy Benyei11169dd2012-12-18 14:30:41 +0000623static bool checkPreprocessorOptions(const PreprocessorOptions &PPOpts,
624 const PreprocessorOptions &ExistingPPOpts,
625 DiagnosticsEngine *Diags,
626 FileManager &FileMgr,
Argyrios Kyrtzidisd3afa0c2013-04-26 21:33:40 +0000627 std::string &SuggestedPredefines,
Yaxun Liu43712e02016-09-07 18:40:20 +0000628 const LangOptions &LangOpts,
629 bool Validate = true) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000630 // Check macro definitions.
631 MacroDefinitionsMap ASTFileMacros;
632 collectMacroDefinitions(PPOpts, ASTFileMacros);
633 MacroDefinitionsMap ExistingMacros;
634 SmallVector<StringRef, 4> ExistingMacroNames;
635 collectMacroDefinitions(ExistingPPOpts, ExistingMacros, &ExistingMacroNames);
636
Vedant Kumar48b4f762018-04-14 01:40:48 +0000637 for (unsigned I = 0, N = ExistingMacroNames.size(); I != N; ++I) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000638 // Dig out the macro definition in the existing preprocessor options.
Vedant Kumar48b4f762018-04-14 01:40:48 +0000639 StringRef MacroName = ExistingMacroNames[I];
Guy Benyei11169dd2012-12-18 14:30:41 +0000640 std::pair<StringRef, bool> Existing = ExistingMacros[MacroName];
641
642 // Check whether we know anything about this macro name or not.
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000643 llvm::StringMap<std::pair<StringRef, bool /*IsUndef*/>>::iterator Known =
644 ASTFileMacros.find(MacroName);
Yaxun Liu43712e02016-09-07 18:40:20 +0000645 if (!Validate || Known == ASTFileMacros.end()) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000646 // FIXME: Check whether this identifier was referenced anywhere in the
647 // AST file. If so, we should reject the AST file. Unfortunately, this
648 // information isn't in the control block. What shall we do about it?
649
650 if (Existing.second) {
651 SuggestedPredefines += "#undef ";
652 SuggestedPredefines += MacroName.str();
653 SuggestedPredefines += '\n';
654 } else {
655 SuggestedPredefines += "#define ";
656 SuggestedPredefines += MacroName.str();
657 SuggestedPredefines += ' ';
658 SuggestedPredefines += Existing.first.str();
659 SuggestedPredefines += '\n';
660 }
661 continue;
662 }
663
664 // If the macro was defined in one but undef'd in the other, we have a
665 // conflict.
666 if (Existing.second != Known->second.second) {
667 if (Diags) {
668 Diags->Report(diag::err_pch_macro_def_undef)
669 << MacroName << Known->second.second;
670 }
671 return true;
672 }
673
674 // If the macro was #undef'd in both, or if the macro bodies are identical,
675 // it's fine.
676 if (Existing.second || Existing.first == Known->second.first)
677 continue;
678
679 // The macro bodies differ; complain.
680 if (Diags) {
681 Diags->Report(diag::err_pch_macro_def_conflict)
682 << MacroName << Known->second.first << Existing.first;
683 }
684 return true;
685 }
686
687 // Check whether we're using predefines.
Yaxun Liu43712e02016-09-07 18:40:20 +0000688 if (PPOpts.UsePredefines != ExistingPPOpts.UsePredefines && Validate) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000689 if (Diags) {
690 Diags->Report(diag::err_pch_undef) << ExistingPPOpts.UsePredefines;
691 }
692 return true;
693 }
694
Argyrios Kyrtzidisd3afa0c2013-04-26 21:33:40 +0000695 // Detailed record is important since it is used for the module cache hash.
696 if (LangOpts.Modules &&
Yaxun Liu43712e02016-09-07 18:40:20 +0000697 PPOpts.DetailedRecord != ExistingPPOpts.DetailedRecord && Validate) {
Argyrios Kyrtzidisd3afa0c2013-04-26 21:33:40 +0000698 if (Diags) {
699 Diags->Report(diag::err_pch_pp_detailed_record) << PPOpts.DetailedRecord;
700 }
701 return true;
702 }
703
Guy Benyei11169dd2012-12-18 14:30:41 +0000704 // Compute the #include and #include_macros lines we need.
Vedant Kumar48b4f762018-04-14 01:40:48 +0000705 for (unsigned I = 0, N = ExistingPPOpts.Includes.size(); I != N; ++I) {
706 StringRef File = ExistingPPOpts.Includes[I];
Erich Keane76675de2018-07-05 17:22:13 +0000707
708 if (!ExistingPPOpts.ImplicitPCHInclude.empty() &&
709 !ExistingPPOpts.PCHThroughHeader.empty()) {
710 // In case the through header is an include, we must add all the includes
711 // to the predefines so the start point can be determined.
712 SuggestedPredefines += "#include \"";
713 SuggestedPredefines += File;
714 SuggestedPredefines += "\"\n";
715 continue;
716 }
717
Guy Benyei11169dd2012-12-18 14:30:41 +0000718 if (File == ExistingPPOpts.ImplicitPCHInclude)
719 continue;
720
721 if (std::find(PPOpts.Includes.begin(), PPOpts.Includes.end(), File)
722 != PPOpts.Includes.end())
723 continue;
724
725 SuggestedPredefines += "#include \"";
Manuel Klimek9af34ae2014-08-12 08:25:57 +0000726 SuggestedPredefines += File;
Guy Benyei11169dd2012-12-18 14:30:41 +0000727 SuggestedPredefines += "\"\n";
728 }
729
Vedant Kumar48b4f762018-04-14 01:40:48 +0000730 for (unsigned I = 0, N = ExistingPPOpts.MacroIncludes.size(); I != N; ++I) {
731 StringRef File = ExistingPPOpts.MacroIncludes[I];
Guy Benyei11169dd2012-12-18 14:30:41 +0000732 if (std::find(PPOpts.MacroIncludes.begin(), PPOpts.MacroIncludes.end(),
733 File)
734 != PPOpts.MacroIncludes.end())
735 continue;
736
737 SuggestedPredefines += "#__include_macros \"";
Manuel Klimek9af34ae2014-08-12 08:25:57 +0000738 SuggestedPredefines += File;
Guy Benyei11169dd2012-12-18 14:30:41 +0000739 SuggestedPredefines += "\"\n##\n";
740 }
741
742 return false;
743}
744
745bool PCHValidator::ReadPreprocessorOptions(const PreprocessorOptions &PPOpts,
746 bool Complain,
747 std::string &SuggestedPredefines) {
748 const PreprocessorOptions &ExistingPPOpts = PP.getPreprocessorOpts();
749
750 return checkPreprocessorOptions(PPOpts, ExistingPPOpts,
Craig Toppera13603a2014-05-22 05:54:18 +0000751 Complain? &Reader.Diags : nullptr,
Guy Benyei11169dd2012-12-18 14:30:41 +0000752 PP.getFileManager(),
Argyrios Kyrtzidisd3afa0c2013-04-26 21:33:40 +0000753 SuggestedPredefines,
754 PP.getLangOpts());
Guy Benyei11169dd2012-12-18 14:30:41 +0000755}
756
Yaxun Liu43712e02016-09-07 18:40:20 +0000757bool SimpleASTReaderListener::ReadPreprocessorOptions(
758 const PreprocessorOptions &PPOpts,
759 bool Complain,
760 std::string &SuggestedPredefines) {
761 return checkPreprocessorOptions(PPOpts,
762 PP.getPreprocessorOpts(),
763 nullptr,
764 PP.getFileManager(),
765 SuggestedPredefines,
766 PP.getLangOpts(),
767 false);
768}
769
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +0000770/// Check the header search options deserialized from the control block
771/// against the header search options in an existing preprocessor.
772///
773/// \param Diags If non-null, produce diagnostics for any mismatches incurred.
774static bool checkHeaderSearchOptions(const HeaderSearchOptions &HSOpts,
775 StringRef SpecificModuleCachePath,
776 StringRef ExistingModuleCachePath,
777 DiagnosticsEngine *Diags,
778 const LangOptions &LangOpts) {
779 if (LangOpts.Modules) {
780 if (SpecificModuleCachePath != ExistingModuleCachePath) {
781 if (Diags)
782 Diags->Report(diag::err_pch_modulecache_mismatch)
783 << SpecificModuleCachePath << ExistingModuleCachePath;
784 return true;
785 }
786 }
787
788 return false;
789}
790
791bool PCHValidator::ReadHeaderSearchOptions(const HeaderSearchOptions &HSOpts,
792 StringRef SpecificModuleCachePath,
793 bool Complain) {
794 return checkHeaderSearchOptions(HSOpts, SpecificModuleCachePath,
795 PP.getHeaderSearchInfo().getModuleCachePath(),
796 Complain ? &Reader.Diags : nullptr,
797 PP.getLangOpts());
798}
799
Guy Benyei11169dd2012-12-18 14:30:41 +0000800void PCHValidator::ReadCounter(const ModuleFile &M, unsigned Value) {
801 PP.setCounterValue(Value);
802}
803
804//===----------------------------------------------------------------------===//
805// AST reader implementation
806//===----------------------------------------------------------------------===//
807
Nico Weber824285e2014-05-08 04:26:47 +0000808void ASTReader::setDeserializationListener(ASTDeserializationListener *Listener,
809 bool TakeOwnership) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000810 DeserializationListener = Listener;
Nico Weber824285e2014-05-08 04:26:47 +0000811 OwnsDeserializationListener = TakeOwnership;
Guy Benyei11169dd2012-12-18 14:30:41 +0000812}
813
Guy Benyei11169dd2012-12-18 14:30:41 +0000814unsigned ASTSelectorLookupTrait::ComputeHash(Selector Sel) {
815 return serialization::ComputeHash(Sel);
816}
817
Guy Benyei11169dd2012-12-18 14:30:41 +0000818std::pair<unsigned, unsigned>
819ASTSelectorLookupTrait::ReadKeyDataLength(const unsigned char*& d) {
Justin Bogner57ba0b22014-03-28 22:03:24 +0000820 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000821
Justin Bogner57ba0b22014-03-28 22:03:24 +0000822 unsigned KeyLen = endian::readNext<uint16_t, little, unaligned>(d);
823 unsigned DataLen = endian::readNext<uint16_t, little, unaligned>(d);
Guy Benyei11169dd2012-12-18 14:30:41 +0000824 return std::make_pair(KeyLen, DataLen);
825}
826
David L. Jonesc4808b9e2016-12-15 20:53:26 +0000827ASTSelectorLookupTrait::internal_key_type
Guy Benyei11169dd2012-12-18 14:30:41 +0000828ASTSelectorLookupTrait::ReadKey(const unsigned char* d, unsigned) {
Justin Bogner57ba0b22014-03-28 22:03:24 +0000829 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000830
Guy Benyei11169dd2012-12-18 14:30:41 +0000831 SelectorTable &SelTable = Reader.getContext().Selectors;
Justin Bogner57ba0b22014-03-28 22:03:24 +0000832 unsigned N = endian::readNext<uint16_t, little, unaligned>(d);
833 IdentifierInfo *FirstII = Reader.getLocalIdentifier(
834 F, endian::readNext<uint32_t, little, unaligned>(d));
Guy Benyei11169dd2012-12-18 14:30:41 +0000835 if (N == 0)
836 return SelTable.getNullarySelector(FirstII);
837 else if (N == 1)
838 return SelTable.getUnarySelector(FirstII);
839
840 SmallVector<IdentifierInfo *, 16> Args;
841 Args.push_back(FirstII);
842 for (unsigned I = 1; I != N; ++I)
Justin Bogner57ba0b22014-03-28 22:03:24 +0000843 Args.push_back(Reader.getLocalIdentifier(
844 F, endian::readNext<uint32_t, little, unaligned>(d)));
Guy Benyei11169dd2012-12-18 14:30:41 +0000845
846 return SelTable.getSelector(N, Args.data());
847}
848
David L. Jonesc4808b9e2016-12-15 20:53:26 +0000849ASTSelectorLookupTrait::data_type
850ASTSelectorLookupTrait::ReadData(Selector, const unsigned char* d,
Guy Benyei11169dd2012-12-18 14:30:41 +0000851 unsigned DataLen) {
Justin Bogner57ba0b22014-03-28 22:03:24 +0000852 using namespace llvm::support;
Guy Benyei11169dd2012-12-18 14:30:41 +0000853
854 data_type Result;
855
Justin Bogner57ba0b22014-03-28 22:03:24 +0000856 Result.ID = Reader.getGlobalSelectorID(
857 F, endian::readNext<uint32_t, little, unaligned>(d));
Nico Weberff4b35e2014-12-27 22:14:15 +0000858 unsigned FullInstanceBits = endian::readNext<uint16_t, little, unaligned>(d);
859 unsigned FullFactoryBits = endian::readNext<uint16_t, little, unaligned>(d);
860 Result.InstanceBits = FullInstanceBits & 0x3;
861 Result.InstanceHasMoreThanOneDecl = (FullInstanceBits >> 2) & 0x1;
862 Result.FactoryBits = FullFactoryBits & 0x3;
863 Result.FactoryHasMoreThanOneDecl = (FullFactoryBits >> 2) & 0x1;
864 unsigned NumInstanceMethods = FullInstanceBits >> 3;
865 unsigned NumFactoryMethods = FullFactoryBits >> 3;
Guy Benyei11169dd2012-12-18 14:30:41 +0000866
867 // Load instance methods
868 for (unsigned I = 0; I != NumInstanceMethods; ++I) {
Vedant Kumar48b4f762018-04-14 01:40:48 +0000869 if (ObjCMethodDecl *Method = Reader.GetLocalDeclAs<ObjCMethodDecl>(
Justin Bogner57ba0b22014-03-28 22:03:24 +0000870 F, endian::readNext<uint32_t, little, unaligned>(d)))
Guy Benyei11169dd2012-12-18 14:30:41 +0000871 Result.Instance.push_back(Method);
872 }
873
874 // Load factory methods
875 for (unsigned I = 0; I != NumFactoryMethods; ++I) {
Vedant Kumar48b4f762018-04-14 01:40:48 +0000876 if (ObjCMethodDecl *Method = Reader.GetLocalDeclAs<ObjCMethodDecl>(
Justin Bogner57ba0b22014-03-28 22:03:24 +0000877 F, endian::readNext<uint32_t, little, unaligned>(d)))
Guy Benyei11169dd2012-12-18 14:30:41 +0000878 Result.Factory.push_back(Method);
879 }
880
881 return Result;
882}
883
Douglas Gregorbfd73d72013-01-23 18:53:14 +0000884unsigned ASTIdentifierLookupTraitBase::ComputeHash(const internal_key_type& a) {
Jonas Devlieghere560ce2c2018-02-26 15:16:42 +0000885 return llvm::djbHash(a);
Guy Benyei11169dd2012-12-18 14:30:41 +0000886}
887
888std::pair<unsigned, unsigned>
Douglas Gregorbfd73d72013-01-23 18:53:14 +0000889ASTIdentifierLookupTraitBase::ReadKeyDataLength(const unsigned char*& d) {
Justin Bogner57ba0b22014-03-28 22:03:24 +0000890 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000891
Justin Bogner57ba0b22014-03-28 22:03:24 +0000892 unsigned DataLen = endian::readNext<uint16_t, little, unaligned>(d);
893 unsigned KeyLen = endian::readNext<uint16_t, little, unaligned>(d);
Guy Benyei11169dd2012-12-18 14:30:41 +0000894 return std::make_pair(KeyLen, DataLen);
895}
896
Douglas Gregorbfd73d72013-01-23 18:53:14 +0000897ASTIdentifierLookupTraitBase::internal_key_type
898ASTIdentifierLookupTraitBase::ReadKey(const unsigned char* d, unsigned n) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000899 assert(n >= 2 && d[n-1] == '\0');
Douglas Gregorbfd73d72013-01-23 18:53:14 +0000900 return StringRef((const char*) d, n-1);
Guy Benyei11169dd2012-12-18 14:30:41 +0000901}
902
Adrian Prantl9fc8faf2018-05-09 01:00:01 +0000903/// Whether the given identifier is "interesting".
Richard Smitha534a312015-07-21 23:54:07 +0000904static bool isInterestingIdentifier(ASTReader &Reader, IdentifierInfo &II,
905 bool IsModule) {
Richard Smithcab89802015-07-17 20:19:56 +0000906 return II.hadMacroDefinition() ||
907 II.isPoisoned() ||
Richard Smith9c254182015-07-19 21:41:12 +0000908 (IsModule ? II.hasRevertedBuiltin() : II.getObjCOrBuiltinID()) ||
Douglas Gregordcf25082013-02-11 18:16:18 +0000909 II.hasRevertedTokenIDToIdentifier() ||
Richard Smithdbafb6c2017-06-29 23:23:46 +0000910 (!(IsModule && Reader.getPreprocessor().getLangOpts().CPlusPlus) &&
Richard Smitha534a312015-07-21 23:54:07 +0000911 II.getFETokenInfo<void>());
Douglas Gregordcf25082013-02-11 18:16:18 +0000912}
913
Richard Smith76c2f2c2015-07-17 20:09:43 +0000914static bool readBit(unsigned &Bits) {
915 bool Value = Bits & 0x1;
916 Bits >>= 1;
917 return Value;
918}
919
Richard Smith79bf9202015-08-24 03:33:22 +0000920IdentID ASTIdentifierLookupTrait::ReadIdentifierID(const unsigned char *d) {
921 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000922
Richard Smith79bf9202015-08-24 03:33:22 +0000923 unsigned RawID = endian::readNext<uint32_t, little, unaligned>(d);
924 return Reader.getGlobalIdentifierID(F, RawID >> 1);
925}
926
Richard Smitheb4b58f62016-02-05 01:40:54 +0000927static void markIdentifierFromAST(ASTReader &Reader, IdentifierInfo &II) {
928 if (!II.isFromAST()) {
929 II.setIsFromAST();
930 bool IsModule = Reader.getPreprocessor().getCurrentModule() != nullptr;
931 if (isInterestingIdentifier(Reader, II, IsModule))
932 II.setChangedSinceDeserialization();
933 }
934}
935
Guy Benyei11169dd2012-12-18 14:30:41 +0000936IdentifierInfo *ASTIdentifierLookupTrait::ReadData(const internal_key_type& k,
937 const unsigned char* d,
938 unsigned DataLen) {
Justin Bogner57ba0b22014-03-28 22:03:24 +0000939 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +0000940
Justin Bogner57ba0b22014-03-28 22:03:24 +0000941 unsigned RawID = endian::readNext<uint32_t, little, unaligned>(d);
Guy Benyei11169dd2012-12-18 14:30:41 +0000942 bool IsInteresting = RawID & 0x01;
943
944 // Wipe out the "is interesting" bit.
945 RawID = RawID >> 1;
946
Richard Smith76c2f2c2015-07-17 20:09:43 +0000947 // Build the IdentifierInfo and link the identifier ID with it.
948 IdentifierInfo *II = KnownII;
949 if (!II) {
950 II = &Reader.getIdentifierTable().getOwn(k);
951 KnownII = II;
952 }
Richard Smitheb4b58f62016-02-05 01:40:54 +0000953 markIdentifierFromAST(Reader, *II);
Richard Smith76c2f2c2015-07-17 20:09:43 +0000954 Reader.markIdentifierUpToDate(II);
955
Guy Benyei11169dd2012-12-18 14:30:41 +0000956 IdentID ID = Reader.getGlobalIdentifierID(F, RawID);
957 if (!IsInteresting) {
Richard Smith76c2f2c2015-07-17 20:09:43 +0000958 // For uninteresting identifiers, there's nothing else to do. Just notify
959 // the reader that we've finished loading this identifier.
Guy Benyei11169dd2012-12-18 14:30:41 +0000960 Reader.SetIdentifierInfo(ID, II);
Guy Benyei11169dd2012-12-18 14:30:41 +0000961 return II;
962 }
963
Justin Bogner57ba0b22014-03-28 22:03:24 +0000964 unsigned ObjCOrBuiltinID = endian::readNext<uint16_t, little, unaligned>(d);
965 unsigned Bits = endian::readNext<uint16_t, little, unaligned>(d);
Richard Smith76c2f2c2015-07-17 20:09:43 +0000966 bool CPlusPlusOperatorKeyword = readBit(Bits);
967 bool HasRevertedTokenIDToIdentifier = readBit(Bits);
Richard Smith9c254182015-07-19 21:41:12 +0000968 bool HasRevertedBuiltin = readBit(Bits);
Richard Smith76c2f2c2015-07-17 20:09:43 +0000969 bool Poisoned = readBit(Bits);
970 bool ExtensionToken = readBit(Bits);
971 bool HadMacroDefinition = readBit(Bits);
Guy Benyei11169dd2012-12-18 14:30:41 +0000972
973 assert(Bits == 0 && "Extra bits in the identifier?");
974 DataLen -= 8;
975
Guy Benyei11169dd2012-12-18 14:30:41 +0000976 // Set or check the various bits in the IdentifierInfo structure.
977 // Token IDs are read-only.
Argyrios Kyrtzidisddee8c92013-02-27 01:13:51 +0000978 if (HasRevertedTokenIDToIdentifier && II->getTokenID() != tok::identifier)
Richard Smith9c254182015-07-19 21:41:12 +0000979 II->revertTokenIDToIdentifier();
980 if (!F.isModule())
981 II->setObjCOrBuiltinID(ObjCOrBuiltinID);
982 else if (HasRevertedBuiltin && II->getBuiltinID()) {
983 II->revertBuiltin();
984 assert((II->hasRevertedBuiltin() ||
985 II->getObjCOrBuiltinID() == ObjCOrBuiltinID) &&
986 "Incorrect ObjC keyword or builtin ID");
987 }
Guy Benyei11169dd2012-12-18 14:30:41 +0000988 assert(II->isExtensionToken() == ExtensionToken &&
989 "Incorrect extension token flag");
990 (void)ExtensionToken;
991 if (Poisoned)
992 II->setIsPoisoned(true);
993 assert(II->isCPlusPlusOperatorKeyword() == CPlusPlusOperatorKeyword &&
994 "Incorrect C++ operator keyword flag");
995 (void)CPlusPlusOperatorKeyword;
996
997 // If this identifier is a macro, deserialize the macro
998 // definition.
Richard Smith76c2f2c2015-07-17 20:09:43 +0000999 if (HadMacroDefinition) {
Justin Bogner57ba0b22014-03-28 22:03:24 +00001000 uint32_t MacroDirectivesOffset =
1001 endian::readNext<uint32_t, little, unaligned>(d);
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001002 DataLen -= 4;
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001003
Richard Smithd7329392015-04-21 21:46:32 +00001004 Reader.addPendingMacro(II, &F, MacroDirectivesOffset);
Guy Benyei11169dd2012-12-18 14:30:41 +00001005 }
1006
1007 Reader.SetIdentifierInfo(ID, II);
1008
1009 // Read all of the declarations visible at global scope with this
1010 // name.
1011 if (DataLen > 0) {
1012 SmallVector<uint32_t, 4> DeclIDs;
1013 for (; DataLen > 0; DataLen -= 4)
Justin Bogner57ba0b22014-03-28 22:03:24 +00001014 DeclIDs.push_back(Reader.getGlobalDeclID(
1015 F, endian::readNext<uint32_t, little, unaligned>(d)));
Guy Benyei11169dd2012-12-18 14:30:41 +00001016 Reader.SetGloballyVisibleDecls(II, DeclIDs);
1017 }
1018
1019 return II;
1020}
1021
Richard Smitha06c7e62015-08-26 23:55:49 +00001022DeclarationNameKey::DeclarationNameKey(DeclarationName Name)
1023 : Kind(Name.getNameKind()) {
1024 switch (Kind) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001025 case DeclarationName::Identifier:
Richard Smitha06c7e62015-08-26 23:55:49 +00001026 Data = (uint64_t)Name.getAsIdentifierInfo();
Guy Benyei11169dd2012-12-18 14:30:41 +00001027 break;
1028 case DeclarationName::ObjCZeroArgSelector:
1029 case DeclarationName::ObjCOneArgSelector:
1030 case DeclarationName::ObjCMultiArgSelector:
Richard Smitha06c7e62015-08-26 23:55:49 +00001031 Data = (uint64_t)Name.getObjCSelector().getAsOpaquePtr();
Guy Benyei11169dd2012-12-18 14:30:41 +00001032 break;
1033 case DeclarationName::CXXOperatorName:
Richard Smitha06c7e62015-08-26 23:55:49 +00001034 Data = Name.getCXXOverloadedOperator();
1035 break;
1036 case DeclarationName::CXXLiteralOperatorName:
1037 Data = (uint64_t)Name.getCXXLiteralIdentifier();
1038 break;
Richard Smith35845152017-02-07 01:37:30 +00001039 case DeclarationName::CXXDeductionGuideName:
1040 Data = (uint64_t)Name.getCXXDeductionGuideTemplate()
1041 ->getDeclName().getAsIdentifierInfo();
1042 break;
Richard Smitha06c7e62015-08-26 23:55:49 +00001043 case DeclarationName::CXXConstructorName:
1044 case DeclarationName::CXXDestructorName:
1045 case DeclarationName::CXXConversionFunctionName:
1046 case DeclarationName::CXXUsingDirective:
1047 Data = 0;
1048 break;
1049 }
1050}
1051
1052unsigned DeclarationNameKey::getHash() const {
1053 llvm::FoldingSetNodeID ID;
1054 ID.AddInteger(Kind);
1055
1056 switch (Kind) {
1057 case DeclarationName::Identifier:
1058 case DeclarationName::CXXLiteralOperatorName:
Richard Smith35845152017-02-07 01:37:30 +00001059 case DeclarationName::CXXDeductionGuideName:
Richard Smitha06c7e62015-08-26 23:55:49 +00001060 ID.AddString(((IdentifierInfo*)Data)->getName());
1061 break;
1062 case DeclarationName::ObjCZeroArgSelector:
1063 case DeclarationName::ObjCOneArgSelector:
1064 case DeclarationName::ObjCMultiArgSelector:
1065 ID.AddInteger(serialization::ComputeHash(Selector(Data)));
1066 break;
1067 case DeclarationName::CXXOperatorName:
1068 ID.AddInteger((OverloadedOperatorKind)Data);
Guy Benyei11169dd2012-12-18 14:30:41 +00001069 break;
1070 case DeclarationName::CXXConstructorName:
1071 case DeclarationName::CXXDestructorName:
1072 case DeclarationName::CXXConversionFunctionName:
1073 case DeclarationName::CXXUsingDirective:
1074 break;
1075 }
1076
1077 return ID.ComputeHash();
1078}
1079
Richard Smithd88a7f12015-09-01 20:35:42 +00001080ModuleFile *
1081ASTDeclContextNameLookupTrait::ReadFileRef(const unsigned char *&d) {
1082 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001083
Richard Smithd88a7f12015-09-01 20:35:42 +00001084 uint32_t ModuleFileID = endian::readNext<uint32_t, little, unaligned>(d);
1085 return Reader.getLocalModuleFile(F, ModuleFileID);
1086}
1087
Guy Benyei11169dd2012-12-18 14:30:41 +00001088std::pair<unsigned, unsigned>
Richard Smitha06c7e62015-08-26 23:55:49 +00001089ASTDeclContextNameLookupTrait::ReadKeyDataLength(const unsigned char *&d) {
Justin Bogner57ba0b22014-03-28 22:03:24 +00001090 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001091
Justin Bogner57ba0b22014-03-28 22:03:24 +00001092 unsigned KeyLen = endian::readNext<uint16_t, little, unaligned>(d);
1093 unsigned DataLen = endian::readNext<uint16_t, little, unaligned>(d);
Guy Benyei11169dd2012-12-18 14:30:41 +00001094 return std::make_pair(KeyLen, DataLen);
1095}
1096
Richard Smitha06c7e62015-08-26 23:55:49 +00001097ASTDeclContextNameLookupTrait::internal_key_type
1098ASTDeclContextNameLookupTrait::ReadKey(const unsigned char *d, unsigned) {
Justin Bogner57ba0b22014-03-28 22:03:24 +00001099 using namespace llvm::support;
Guy Benyei11169dd2012-12-18 14:30:41 +00001100
Vedant Kumar48b4f762018-04-14 01:40:48 +00001101 auto Kind = (DeclarationName::NameKind)*d++;
Richard Smitha06c7e62015-08-26 23:55:49 +00001102 uint64_t Data;
1103 switch (Kind) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001104 case DeclarationName::Identifier:
Richard Smith35845152017-02-07 01:37:30 +00001105 case DeclarationName::CXXLiteralOperatorName:
1106 case DeclarationName::CXXDeductionGuideName:
Richard Smitha06c7e62015-08-26 23:55:49 +00001107 Data = (uint64_t)Reader.getLocalIdentifier(
Justin Bogner57ba0b22014-03-28 22:03:24 +00001108 F, endian::readNext<uint32_t, little, unaligned>(d));
Guy Benyei11169dd2012-12-18 14:30:41 +00001109 break;
1110 case DeclarationName::ObjCZeroArgSelector:
1111 case DeclarationName::ObjCOneArgSelector:
1112 case DeclarationName::ObjCMultiArgSelector:
Richard Smitha06c7e62015-08-26 23:55:49 +00001113 Data =
Justin Bogner57ba0b22014-03-28 22:03:24 +00001114 (uint64_t)Reader.getLocalSelector(
1115 F, endian::readNext<uint32_t, little, unaligned>(
1116 d)).getAsOpaquePtr();
Guy Benyei11169dd2012-12-18 14:30:41 +00001117 break;
1118 case DeclarationName::CXXOperatorName:
Richard Smitha06c7e62015-08-26 23:55:49 +00001119 Data = *d++; // OverloadedOperatorKind
Guy Benyei11169dd2012-12-18 14:30:41 +00001120 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00001121 case DeclarationName::CXXConstructorName:
1122 case DeclarationName::CXXDestructorName:
1123 case DeclarationName::CXXConversionFunctionName:
1124 case DeclarationName::CXXUsingDirective:
Richard Smitha06c7e62015-08-26 23:55:49 +00001125 Data = 0;
Guy Benyei11169dd2012-12-18 14:30:41 +00001126 break;
1127 }
1128
Richard Smitha06c7e62015-08-26 23:55:49 +00001129 return DeclarationNameKey(Kind, Data);
Guy Benyei11169dd2012-12-18 14:30:41 +00001130}
1131
Richard Smithd88a7f12015-09-01 20:35:42 +00001132void ASTDeclContextNameLookupTrait::ReadDataInto(internal_key_type,
1133 const unsigned char *d,
1134 unsigned DataLen,
1135 data_type_builder &Val) {
Justin Bogner57ba0b22014-03-28 22:03:24 +00001136 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001137
Richard Smithd88a7f12015-09-01 20:35:42 +00001138 for (unsigned NumDecls = DataLen / 4; NumDecls; --NumDecls) {
1139 uint32_t LocalID = endian::readNext<uint32_t, little, unaligned>(d);
1140 Val.insert(Reader.getGlobalDeclID(F, LocalID));
1141 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001142}
1143
Richard Smith0f4e2c42015-08-06 04:23:48 +00001144bool ASTReader::ReadLexicalDeclContextStorage(ModuleFile &M,
1145 BitstreamCursor &Cursor,
1146 uint64_t Offset,
1147 DeclContext *DC) {
1148 assert(Offset != 0);
1149
Guy Benyei11169dd2012-12-18 14:30:41 +00001150 SavedStreamPosition SavedPosition(Cursor);
Richard Smith0f4e2c42015-08-06 04:23:48 +00001151 Cursor.JumpToBit(Offset);
Guy Benyei11169dd2012-12-18 14:30:41 +00001152
Richard Smith0f4e2c42015-08-06 04:23:48 +00001153 RecordData Record;
1154 StringRef Blob;
1155 unsigned Code = Cursor.ReadCode();
1156 unsigned RecCode = Cursor.readRecord(Code, Record, &Blob);
1157 if (RecCode != DECL_CONTEXT_LEXICAL) {
1158 Error("Expected lexical block");
1159 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +00001160 }
1161
Richard Smith82f8fcd2015-08-06 22:07:25 +00001162 assert(!isa<TranslationUnitDecl>(DC) &&
1163 "expected a TU_UPDATE_LEXICAL record for TU");
Richard Smith9c9173d2015-08-11 22:00:24 +00001164 // If we are handling a C++ class template instantiation, we can see multiple
1165 // lexical updates for the same record. It's important that we select only one
1166 // of them, so that field numbering works properly. Just pick the first one we
1167 // see.
1168 auto &Lex = LexicalDecls[DC];
1169 if (!Lex.first) {
1170 Lex = std::make_pair(
1171 &M, llvm::makeArrayRef(
1172 reinterpret_cast<const llvm::support::unaligned_uint32_t *>(
1173 Blob.data()),
1174 Blob.size() / 4));
1175 }
Richard Smith0f4e2c42015-08-06 04:23:48 +00001176 DC->setHasExternalLexicalStorage(true);
1177 return false;
1178}
Guy Benyei11169dd2012-12-18 14:30:41 +00001179
Richard Smith0f4e2c42015-08-06 04:23:48 +00001180bool ASTReader::ReadVisibleDeclContextStorage(ModuleFile &M,
1181 BitstreamCursor &Cursor,
1182 uint64_t Offset,
1183 DeclID ID) {
1184 assert(Offset != 0);
1185
1186 SavedStreamPosition SavedPosition(Cursor);
1187 Cursor.JumpToBit(Offset);
1188
1189 RecordData Record;
1190 StringRef Blob;
1191 unsigned Code = Cursor.ReadCode();
1192 unsigned RecCode = Cursor.readRecord(Code, Record, &Blob);
1193 if (RecCode != DECL_CONTEXT_VISIBLE) {
1194 Error("Expected visible lookup table block");
1195 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +00001196 }
1197
Richard Smith0f4e2c42015-08-06 04:23:48 +00001198 // We can't safely determine the primary context yet, so delay attaching the
1199 // lookup table until we're done with recursive deserialization.
Richard Smithd88a7f12015-09-01 20:35:42 +00001200 auto *Data = (const unsigned char*)Blob.data();
1201 PendingVisibleUpdates[ID].push_back(PendingVisibleUpdate{&M, Data});
Guy Benyei11169dd2012-12-18 14:30:41 +00001202 return false;
1203}
1204
Richard Smith37a93df2017-02-18 00:32:02 +00001205void ASTReader::Error(StringRef Msg) const {
Guy Benyei11169dd2012-12-18 14:30:41 +00001206 Error(diag::err_fe_pch_malformed, Msg);
Richard Smithdbafb6c2017-06-29 23:23:46 +00001207 if (PP.getLangOpts().Modules && !Diags.isDiagnosticInFlight() &&
Richard Smithfb1e7f72015-08-14 05:02:58 +00001208 !PP.getHeaderSearchInfo().getModuleCachePath().empty()) {
Douglas Gregor940e8052013-05-10 22:15:13 +00001209 Diag(diag::note_module_cache_path)
1210 << PP.getHeaderSearchInfo().getModuleCachePath();
1211 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001212}
1213
1214void ASTReader::Error(unsigned DiagID,
Richard Smith37a93df2017-02-18 00:32:02 +00001215 StringRef Arg1, StringRef Arg2) const {
Guy Benyei11169dd2012-12-18 14:30:41 +00001216 if (Diags.isDiagnosticInFlight())
1217 Diags.SetDelayedDiagnostic(DiagID, Arg1, Arg2);
1218 else
1219 Diag(DiagID) << Arg1 << Arg2;
1220}
1221
1222//===----------------------------------------------------------------------===//
1223// Source Manager Deserialization
1224//===----------------------------------------------------------------------===//
1225
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00001226/// Read the line table in the source manager block.
Guy Benyei11169dd2012-12-18 14:30:41 +00001227/// \returns true if there was an error.
1228bool ASTReader::ParseLineTable(ModuleFile &F,
Richard Smith7ed1bc92014-12-05 22:42:13 +00001229 const RecordData &Record) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001230 unsigned Idx = 0;
1231 LineTableInfo &LineTable = SourceMgr.getLineTable();
1232
1233 // Parse the file names
1234 std::map<int, int> FileIDs;
Hans Wennborg14487362017-12-04 22:28:45 +00001235 FileIDs[-1] = -1; // For unspecified filenames.
Richard Smith63078492015-09-01 07:41:55 +00001236 for (unsigned I = 0; Record[Idx]; ++I) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001237 // Extract the file name
Richard Smith7ed1bc92014-12-05 22:42:13 +00001238 auto Filename = ReadPath(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00001239 FileIDs[I] = LineTable.getLineTableFilenameID(Filename);
1240 }
Richard Smith63078492015-09-01 07:41:55 +00001241 ++Idx;
Guy Benyei11169dd2012-12-18 14:30:41 +00001242
1243 // Parse the line entries
1244 std::vector<LineEntry> Entries;
1245 while (Idx < Record.size()) {
1246 int FID = Record[Idx++];
1247 assert(FID >= 0 && "Serialized line entries for non-local file.");
1248 // Remap FileID from 1-based old view.
1249 FID += F.SLocEntryBaseID - 1;
1250
1251 // Extract the line entries
1252 unsigned NumEntries = Record[Idx++];
Richard Smith63078492015-09-01 07:41:55 +00001253 assert(NumEntries && "no line entries for file ID");
Guy Benyei11169dd2012-12-18 14:30:41 +00001254 Entries.clear();
1255 Entries.reserve(NumEntries);
1256 for (unsigned I = 0; I != NumEntries; ++I) {
1257 unsigned FileOffset = Record[Idx++];
1258 unsigned LineNo = Record[Idx++];
1259 int FilenameID = FileIDs[Record[Idx++]];
Vedant Kumar48b4f762018-04-14 01:40:48 +00001260 SrcMgr::CharacteristicKind FileKind
1261 = (SrcMgr::CharacteristicKind)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00001262 unsigned IncludeOffset = Record[Idx++];
1263 Entries.push_back(LineEntry::get(FileOffset, LineNo, FilenameID,
1264 FileKind, IncludeOffset));
1265 }
1266 LineTable.AddEntry(FileID::get(FID), Entries);
1267 }
1268
1269 return false;
1270}
1271
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00001272/// Read a source manager block
Guy Benyei11169dd2012-12-18 14:30:41 +00001273bool ASTReader::ReadSourceManagerBlock(ModuleFile &F) {
1274 using namespace SrcMgr;
1275
Chris Lattner7fb3bef2013-01-20 00:56:42 +00001276 BitstreamCursor &SLocEntryCursor = F.SLocEntryCursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00001277
1278 // Set the source-location entry cursor to the current position in
1279 // the stream. This cursor will be used to read the contents of the
1280 // source manager block initially, and then lazily read
1281 // source-location entries as needed.
1282 SLocEntryCursor = F.Stream;
1283
1284 // The stream itself is going to skip over the source manager block.
1285 if (F.Stream.SkipBlock()) {
1286 Error("malformed block record in AST file");
1287 return true;
1288 }
1289
1290 // Enter the source manager block.
1291 if (SLocEntryCursor.EnterSubBlock(SOURCE_MANAGER_BLOCK_ID)) {
1292 Error("malformed source manager block record in AST file");
1293 return true;
1294 }
1295
1296 RecordData Record;
1297 while (true) {
Chris Lattnere7b154b2013-01-19 21:39:22 +00001298 llvm::BitstreamEntry E = SLocEntryCursor.advanceSkippingSubblocks();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001299
Chris Lattnere7b154b2013-01-19 21:39:22 +00001300 switch (E.Kind) {
1301 case llvm::BitstreamEntry::SubBlock: // Handled for us already.
1302 case llvm::BitstreamEntry::Error:
1303 Error("malformed block record in AST file");
1304 return true;
1305 case llvm::BitstreamEntry::EndBlock:
Guy Benyei11169dd2012-12-18 14:30:41 +00001306 return false;
Chris Lattnere7b154b2013-01-19 21:39:22 +00001307 case llvm::BitstreamEntry::Record:
1308 // The interesting case.
1309 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00001310 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001311
Guy Benyei11169dd2012-12-18 14:30:41 +00001312 // Read a record.
Guy Benyei11169dd2012-12-18 14:30:41 +00001313 Record.clear();
Chris Lattner15c3e7d2013-01-21 18:28:26 +00001314 StringRef Blob;
1315 switch (SLocEntryCursor.readRecord(E.ID, Record, &Blob)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001316 default: // Default behavior: ignore.
1317 break;
1318
1319 case SM_SLOC_FILE_ENTRY:
1320 case SM_SLOC_BUFFER_ENTRY:
1321 case SM_SLOC_EXPANSION_ENTRY:
1322 // Once we hit one of the source location entries, we're done.
1323 return false;
1324 }
1325 }
1326}
1327
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00001328/// If a header file is not found at the path that we expect it to be
Guy Benyei11169dd2012-12-18 14:30:41 +00001329/// and the PCH file was moved from its original location, try to resolve the
1330/// file by assuming that header+PCH were moved together and the header is in
1331/// the same place relative to the PCH.
1332static std::string
1333resolveFileRelativeToOriginalDir(const std::string &Filename,
1334 const std::string &OriginalDir,
1335 const std::string &CurrDir) {
1336 assert(OriginalDir != CurrDir &&
1337 "No point trying to resolve the file if the PCH dir didn't change");
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001338
Guy Benyei11169dd2012-12-18 14:30:41 +00001339 using namespace llvm::sys;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001340
Guy Benyei11169dd2012-12-18 14:30:41 +00001341 SmallString<128> filePath(Filename);
1342 fs::make_absolute(filePath);
1343 assert(path::is_absolute(OriginalDir));
1344 SmallString<128> currPCHPath(CurrDir);
1345
1346 path::const_iterator fileDirI = path::begin(path::parent_path(filePath)),
1347 fileDirE = path::end(path::parent_path(filePath));
1348 path::const_iterator origDirI = path::begin(OriginalDir),
1349 origDirE = path::end(OriginalDir);
1350 // Skip the common path components from filePath and OriginalDir.
1351 while (fileDirI != fileDirE && origDirI != origDirE &&
1352 *fileDirI == *origDirI) {
1353 ++fileDirI;
1354 ++origDirI;
1355 }
1356 for (; origDirI != origDirE; ++origDirI)
1357 path::append(currPCHPath, "..");
1358 path::append(currPCHPath, fileDirI, fileDirE);
1359 path::append(currPCHPath, path::filename(Filename));
1360 return currPCHPath.str();
1361}
1362
1363bool ASTReader::ReadSLocEntry(int ID) {
1364 if (ID == 0)
1365 return false;
1366
1367 if (unsigned(-ID) - 2 >= getTotalNumSLocs() || ID > 0) {
1368 Error("source location entry ID out-of-range for AST file");
1369 return true;
1370 }
1371
Richard Smithaada85c2016-02-06 02:06:43 +00001372 // Local helper to read the (possibly-compressed) buffer data following the
1373 // entry record.
1374 auto ReadBuffer = [this](
1375 BitstreamCursor &SLocEntryCursor,
1376 StringRef Name) -> std::unique_ptr<llvm::MemoryBuffer> {
1377 RecordData Record;
1378 StringRef Blob;
1379 unsigned Code = SLocEntryCursor.ReadCode();
1380 unsigned RecCode = SLocEntryCursor.readRecord(Code, Record, &Blob);
1381
1382 if (RecCode == SM_SLOC_BUFFER_BLOB_COMPRESSED) {
George Rimarc39f5492017-01-17 15:45:31 +00001383 if (!llvm::zlib::isAvailable()) {
1384 Error("zlib is not available");
1385 return nullptr;
1386 }
Richard Smithaada85c2016-02-06 02:06:43 +00001387 SmallString<0> Uncompressed;
George Rimarc39f5492017-01-17 15:45:31 +00001388 if (llvm::Error E =
1389 llvm::zlib::uncompress(Blob, Uncompressed, Record[0])) {
1390 Error("could not decompress embedded file contents: " +
1391 llvm::toString(std::move(E)));
Richard Smithaada85c2016-02-06 02:06:43 +00001392 return nullptr;
1393 }
1394 return llvm::MemoryBuffer::getMemBufferCopy(Uncompressed, Name);
1395 } else if (RecCode == SM_SLOC_BUFFER_BLOB) {
1396 return llvm::MemoryBuffer::getMemBuffer(Blob.drop_back(1), Name, true);
1397 } else {
1398 Error("AST record has invalid code");
1399 return nullptr;
1400 }
1401 };
1402
Guy Benyei11169dd2012-12-18 14:30:41 +00001403 ModuleFile *F = GlobalSLocEntryMap.find(-ID)->second;
1404 F->SLocEntryCursor.JumpToBit(F->SLocEntryOffsets[ID - F->SLocEntryBaseID]);
Chris Lattner7fb3bef2013-01-20 00:56:42 +00001405 BitstreamCursor &SLocEntryCursor = F->SLocEntryCursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00001406 unsigned BaseOffset = F->SLocEntryBaseOffset;
1407
1408 ++NumSLocEntriesRead;
Chris Lattnere7b154b2013-01-19 21:39:22 +00001409 llvm::BitstreamEntry Entry = SLocEntryCursor.advance();
1410 if (Entry.Kind != llvm::BitstreamEntry::Record) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001411 Error("incorrectly-formatted source location entry in AST file");
1412 return true;
1413 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001414
Guy Benyei11169dd2012-12-18 14:30:41 +00001415 RecordData Record;
Chris Lattner0e6c9402013-01-20 02:38:54 +00001416 StringRef Blob;
1417 switch (SLocEntryCursor.readRecord(Entry.ID, Record, &Blob)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001418 default:
1419 Error("incorrectly-formatted source location entry in AST file");
1420 return true;
1421
1422 case SM_SLOC_FILE_ENTRY: {
1423 // We will detect whether a file changed and return 'Failure' for it, but
1424 // we will also try to fail gracefully by setting up the SLocEntry.
1425 unsigned InputID = Record[4];
1426 InputFile IF = getInputFile(*F, InputID);
Argyrios Kyrtzidis61c3d872013-03-01 03:26:04 +00001427 const FileEntry *File = IF.getFile();
1428 bool OverriddenBuffer = IF.isOverridden();
Guy Benyei11169dd2012-12-18 14:30:41 +00001429
Argyrios Kyrtzidis61c3d872013-03-01 03:26:04 +00001430 // Note that we only check if a File was returned. If it was out-of-date
1431 // we have complained but we will continue creating a FileID to recover
1432 // gracefully.
1433 if (!File)
Guy Benyei11169dd2012-12-18 14:30:41 +00001434 return true;
1435
1436 SourceLocation IncludeLoc = ReadSourceLocation(*F, Record[1]);
1437 if (IncludeLoc.isInvalid() && F->Kind != MK_MainFile) {
1438 // This is the module's main file.
1439 IncludeLoc = getImportLocation(F);
1440 }
Vedant Kumar48b4f762018-04-14 01:40:48 +00001441 SrcMgr::CharacteristicKind
1442 FileCharacter = (SrcMgr::CharacteristicKind)Record[2];
Guy Benyei11169dd2012-12-18 14:30:41 +00001443 FileID FID = SourceMgr.createFileID(File, IncludeLoc, FileCharacter,
1444 ID, BaseOffset + Record[0]);
Vedant Kumar48b4f762018-04-14 01:40:48 +00001445 SrcMgr::FileInfo &FileInfo =
1446 const_cast<SrcMgr::FileInfo&>(SourceMgr.getSLocEntry(FID).getFile());
Guy Benyei11169dd2012-12-18 14:30:41 +00001447 FileInfo.NumCreatedFIDs = Record[5];
1448 if (Record[3])
1449 FileInfo.setHasLineDirectives();
1450
1451 const DeclID *FirstDecl = F->FileSortedDecls + Record[6];
1452 unsigned NumFileDecls = Record[7];
Richard Smithdbafb6c2017-06-29 23:23:46 +00001453 if (NumFileDecls && ContextObj) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001454 assert(F->FileSortedDecls && "FILE_SORTED_DECLS not encountered yet ?");
1455 FileDeclIDs[FID] = FileDeclsInfo(F, llvm::makeArrayRef(FirstDecl,
1456 NumFileDecls));
1457 }
Richard Smithaada85c2016-02-06 02:06:43 +00001458
Guy Benyei11169dd2012-12-18 14:30:41 +00001459 const SrcMgr::ContentCache *ContentCache
Richard Smithf3f84612017-06-29 02:19:42 +00001460 = SourceMgr.getOrCreateContentCache(File, isSystem(FileCharacter));
Guy Benyei11169dd2012-12-18 14:30:41 +00001461 if (OverriddenBuffer && !ContentCache->BufferOverridden &&
Richard Smitha8cfffa2015-11-26 02:04:16 +00001462 ContentCache->ContentsEntry == ContentCache->OrigEntry &&
1463 !ContentCache->getRawBuffer()) {
Richard Smithaada85c2016-02-06 02:06:43 +00001464 auto Buffer = ReadBuffer(SLocEntryCursor, File->getName());
1465 if (!Buffer)
Guy Benyei11169dd2012-12-18 14:30:41 +00001466 return true;
David Blaikie49cc3182014-08-27 20:54:45 +00001467 SourceMgr.overrideFileContents(File, std::move(Buffer));
Guy Benyei11169dd2012-12-18 14:30:41 +00001468 }
1469
1470 break;
1471 }
1472
1473 case SM_SLOC_BUFFER_ENTRY: {
Chris Lattner0e6c9402013-01-20 02:38:54 +00001474 const char *Name = Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00001475 unsigned Offset = Record[0];
Vedant Kumar48b4f762018-04-14 01:40:48 +00001476 SrcMgr::CharacteristicKind
1477 FileCharacter = (SrcMgr::CharacteristicKind)Record[2];
Guy Benyei11169dd2012-12-18 14:30:41 +00001478 SourceLocation IncludeLoc = ReadSourceLocation(*F, Record[1]);
Manman Ren11f2a472016-08-18 17:42:15 +00001479 if (IncludeLoc.isInvalid() && F->isModule()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001480 IncludeLoc = getImportLocation(F);
1481 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001482
Richard Smithaada85c2016-02-06 02:06:43 +00001483 auto Buffer = ReadBuffer(SLocEntryCursor, Name);
1484 if (!Buffer)
Guy Benyei11169dd2012-12-18 14:30:41 +00001485 return true;
David Blaikie50a5f972014-08-29 07:59:55 +00001486 SourceMgr.createFileID(std::move(Buffer), FileCharacter, ID,
Rafael Espindolad87f8d72014-08-27 20:03:29 +00001487 BaseOffset + Offset, IncludeLoc);
Guy Benyei11169dd2012-12-18 14:30:41 +00001488 break;
1489 }
1490
1491 case SM_SLOC_EXPANSION_ENTRY: {
1492 SourceLocation SpellingLoc = ReadSourceLocation(*F, Record[1]);
1493 SourceMgr.createExpansionLoc(SpellingLoc,
1494 ReadSourceLocation(*F, Record[2]),
1495 ReadSourceLocation(*F, Record[3]),
Richard Smithb5f81712018-04-30 05:25:48 +00001496 Record[5],
Guy Benyei11169dd2012-12-18 14:30:41 +00001497 Record[4],
1498 ID,
1499 BaseOffset + Record[0]);
1500 break;
1501 }
1502 }
1503
1504 return false;
1505}
1506
1507std::pair<SourceLocation, StringRef> ASTReader::getModuleImportLoc(int ID) {
1508 if (ID == 0)
1509 return std::make_pair(SourceLocation(), "");
1510
1511 if (unsigned(-ID) - 2 >= getTotalNumSLocs() || ID > 0) {
1512 Error("source location entry ID out-of-range for AST file");
1513 return std::make_pair(SourceLocation(), "");
1514 }
1515
1516 // Find which module file this entry lands in.
1517 ModuleFile *M = GlobalSLocEntryMap.find(-ID)->second;
Manman Ren11f2a472016-08-18 17:42:15 +00001518 if (!M->isModule())
Guy Benyei11169dd2012-12-18 14:30:41 +00001519 return std::make_pair(SourceLocation(), "");
1520
1521 // FIXME: Can we map this down to a particular submodule? That would be
1522 // ideal.
Ben Langmuirbeee15e2014-04-14 18:00:01 +00001523 return std::make_pair(M->ImportLoc, StringRef(M->ModuleName));
Guy Benyei11169dd2012-12-18 14:30:41 +00001524}
1525
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00001526/// Find the location where the module F is imported.
Guy Benyei11169dd2012-12-18 14:30:41 +00001527SourceLocation ASTReader::getImportLocation(ModuleFile *F) {
1528 if (F->ImportLoc.isValid())
1529 return F->ImportLoc;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001530
Guy Benyei11169dd2012-12-18 14:30:41 +00001531 // Otherwise we have a PCH. It's considered to be "imported" at the first
1532 // location of its includer.
1533 if (F->ImportedBy.empty() || !F->ImportedBy[0]) {
Ben Langmuirbeee15e2014-04-14 18:00:01 +00001534 // Main file is the importer.
Yaron Keren8b563662015-10-03 10:46:20 +00001535 assert(SourceMgr.getMainFileID().isValid() && "missing main file");
Ben Langmuirbeee15e2014-04-14 18:00:01 +00001536 return SourceMgr.getLocForStartOfFile(SourceMgr.getMainFileID());
Guy Benyei11169dd2012-12-18 14:30:41 +00001537 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001538 return F->ImportedBy[0]->FirstLoc;
1539}
1540
1541/// ReadBlockAbbrevs - Enter a subblock of the specified BlockID with the
1542/// specified cursor. Read the abbreviations that are at the top of the block
1543/// and then leave the cursor pointing into the block.
Chris Lattner7fb3bef2013-01-20 00:56:42 +00001544bool ASTReader::ReadBlockAbbrevs(BitstreamCursor &Cursor, unsigned BlockID) {
Richard Smith0516b182015-09-08 19:40:14 +00001545 if (Cursor.EnterSubBlock(BlockID))
1546 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +00001547
1548 while (true) {
1549 uint64_t Offset = Cursor.GetCurrentBitNo();
1550 unsigned Code = Cursor.ReadCode();
1551
1552 // We expect all abbrevs to be at the start of the block.
1553 if (Code != llvm::bitc::DEFINE_ABBREV) {
1554 Cursor.JumpToBit(Offset);
1555 return false;
1556 }
1557 Cursor.ReadAbbrevRecord();
1558 }
1559}
1560
Richard Smithe40f2ba2013-08-07 21:41:30 +00001561Token ASTReader::ReadToken(ModuleFile &F, const RecordDataImpl &Record,
John McCallf413f5e2013-05-03 00:10:13 +00001562 unsigned &Idx) {
1563 Token Tok;
1564 Tok.startToken();
1565 Tok.setLocation(ReadSourceLocation(F, Record, Idx));
1566 Tok.setLength(Record[Idx++]);
1567 if (IdentifierInfo *II = getLocalIdentifier(F, Record[Idx++]))
1568 Tok.setIdentifierInfo(II);
1569 Tok.setKind((tok::TokenKind)Record[Idx++]);
1570 Tok.setFlag((Token::TokenFlags)Record[Idx++]);
1571 return Tok;
1572}
1573
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001574MacroInfo *ASTReader::ReadMacroRecord(ModuleFile &F, uint64_t Offset) {
Chris Lattner7fb3bef2013-01-20 00:56:42 +00001575 BitstreamCursor &Stream = F.MacroCursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00001576
1577 // Keep track of where we are in the stream, then jump back there
1578 // after reading this macro.
1579 SavedStreamPosition SavedPosition(Stream);
1580
1581 Stream.JumpToBit(Offset);
1582 RecordData Record;
Faisal Valiac506d72017-07-17 17:18:43 +00001583 SmallVector<IdentifierInfo*, 16> MacroParams;
Craig Toppera13603a2014-05-22 05:54:18 +00001584 MacroInfo *Macro = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00001585
Guy Benyei11169dd2012-12-18 14:30:41 +00001586 while (true) {
Chris Lattnerefa77172013-01-20 00:00:22 +00001587 // Advance to the next record, but if we get to the end of the block, don't
1588 // pop it (removing all the abbreviations from the cursor) since we want to
1589 // be able to reseek within the block and read entries.
Chris Lattner7fb3bef2013-01-20 00:56:42 +00001590 unsigned Flags = BitstreamCursor::AF_DontPopBlockAtEnd;
Chris Lattnerefa77172013-01-20 00:00:22 +00001591 llvm::BitstreamEntry Entry = Stream.advanceSkippingSubblocks(Flags);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001592
Chris Lattnerefa77172013-01-20 00:00:22 +00001593 switch (Entry.Kind) {
1594 case llvm::BitstreamEntry::SubBlock: // Handled for us already.
1595 case llvm::BitstreamEntry::Error:
1596 Error("malformed block record in AST file");
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001597 return Macro;
Chris Lattnerefa77172013-01-20 00:00:22 +00001598 case llvm::BitstreamEntry::EndBlock:
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001599 return Macro;
Chris Lattnerefa77172013-01-20 00:00:22 +00001600 case llvm::BitstreamEntry::Record:
1601 // The interesting case.
1602 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00001603 }
1604
1605 // Read a record.
Guy Benyei11169dd2012-12-18 14:30:41 +00001606 Record.clear();
Vedant Kumar48b4f762018-04-14 01:40:48 +00001607 PreprocessorRecordTypes RecType =
1608 (PreprocessorRecordTypes)Stream.readRecord(Entry.ID, Record);
Guy Benyei11169dd2012-12-18 14:30:41 +00001609 switch (RecType) {
Richard Smithd7329392015-04-21 21:46:32 +00001610 case PP_MODULE_MACRO:
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001611 case PP_MACRO_DIRECTIVE_HISTORY:
1612 return Macro;
1613
Guy Benyei11169dd2012-12-18 14:30:41 +00001614 case PP_MACRO_OBJECT_LIKE:
1615 case PP_MACRO_FUNCTION_LIKE: {
1616 // If we already have a macro, that means that we've hit the end
1617 // of the definition of the macro we were looking for. We're
1618 // done.
1619 if (Macro)
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001620 return Macro;
Guy Benyei11169dd2012-12-18 14:30:41 +00001621
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001622 unsigned NextIndex = 1; // Skip identifier ID.
Guy Benyei11169dd2012-12-18 14:30:41 +00001623 SourceLocation Loc = ReadSourceLocation(F, Record, NextIndex);
Richard Smith3f6dd7a2017-05-12 23:40:52 +00001624 MacroInfo *MI = PP.AllocateMacroInfo(Loc);
Argyrios Kyrtzidis7572be22013-01-07 19:16:23 +00001625 MI->setDefinitionEndLoc(ReadSourceLocation(F, Record, NextIndex));
Guy Benyei11169dd2012-12-18 14:30:41 +00001626 MI->setIsUsed(Record[NextIndex++]);
Argyrios Kyrtzidis9ef53ce2014-04-09 18:21:23 +00001627 MI->setUsedForHeaderGuard(Record[NextIndex++]);
Guy Benyei11169dd2012-12-18 14:30:41 +00001628
Guy Benyei11169dd2012-12-18 14:30:41 +00001629 if (RecType == PP_MACRO_FUNCTION_LIKE) {
1630 // Decode function-like macro info.
1631 bool isC99VarArgs = Record[NextIndex++];
1632 bool isGNUVarArgs = Record[NextIndex++];
1633 bool hasCommaPasting = Record[NextIndex++];
Faisal Valiac506d72017-07-17 17:18:43 +00001634 MacroParams.clear();
Guy Benyei11169dd2012-12-18 14:30:41 +00001635 unsigned NumArgs = Record[NextIndex++];
1636 for (unsigned i = 0; i != NumArgs; ++i)
Faisal Valiac506d72017-07-17 17:18:43 +00001637 MacroParams.push_back(getLocalIdentifier(F, Record[NextIndex++]));
Guy Benyei11169dd2012-12-18 14:30:41 +00001638
1639 // Install function-like macro info.
1640 MI->setIsFunctionLike();
1641 if (isC99VarArgs) MI->setIsC99Varargs();
1642 if (isGNUVarArgs) MI->setIsGNUVarargs();
1643 if (hasCommaPasting) MI->setHasCommaPasting();
Faisal Valiac506d72017-07-17 17:18:43 +00001644 MI->setParameterList(MacroParams, PP.getPreprocessorAllocator());
Guy Benyei11169dd2012-12-18 14:30:41 +00001645 }
1646
Guy Benyei11169dd2012-12-18 14:30:41 +00001647 // Remember that we saw this macro last so that we add the tokens that
1648 // form its body to it.
1649 Macro = MI;
1650
1651 if (NextIndex + 1 == Record.size() && PP.getPreprocessingRecord() &&
1652 Record[NextIndex]) {
1653 // We have a macro definition. Register the association
1654 PreprocessedEntityID
1655 GlobalID = getGlobalPreprocessedEntityID(F, Record[NextIndex]);
1656 PreprocessingRecord &PPRec = *PP.getPreprocessingRecord();
Richard Smith66a81862015-05-04 02:25:31 +00001657 PreprocessingRecord::PPEntityID PPID =
1658 PPRec.getPPEntityID(GlobalID - 1, /*isLoaded=*/true);
Vedant Kumar48b4f762018-04-14 01:40:48 +00001659 MacroDefinitionRecord *PPDef = cast_or_null<MacroDefinitionRecord>(
Richard Smith66a81862015-05-04 02:25:31 +00001660 PPRec.getPreprocessedEntity(PPID));
Argyrios Kyrtzidis832de9f2013-02-22 18:35:59 +00001661 if (PPDef)
1662 PPRec.RegisterMacroDefinition(Macro, PPDef);
Guy Benyei11169dd2012-12-18 14:30:41 +00001663 }
1664
1665 ++NumMacrosRead;
1666 break;
1667 }
1668
1669 case PP_TOKEN: {
1670 // If we see a TOKEN before a PP_MACRO_*, then the file is
1671 // erroneous, just pretend we didn't see this.
Craig Toppera13603a2014-05-22 05:54:18 +00001672 if (!Macro) break;
Guy Benyei11169dd2012-12-18 14:30:41 +00001673
John McCallf413f5e2013-05-03 00:10:13 +00001674 unsigned Idx = 0;
1675 Token Tok = ReadToken(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00001676 Macro->AddTokenToBody(Tok);
1677 break;
1678 }
1679 }
1680 }
1681}
1682
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001683PreprocessedEntityID
Richard Smith37a93df2017-02-18 00:32:02 +00001684ASTReader::getGlobalPreprocessedEntityID(ModuleFile &M,
1685 unsigned LocalID) const {
1686 if (!M.ModuleOffsetMap.empty())
1687 ReadModuleOffsetMap(M);
1688
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001689 ContinuousRangeMap<uint32_t, int, 2>::const_iterator
Guy Benyei11169dd2012-12-18 14:30:41 +00001690 I = M.PreprocessedEntityRemap.find(LocalID - NUM_PREDEF_PP_ENTITY_IDS);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001691 assert(I != M.PreprocessedEntityRemap.end()
Guy Benyei11169dd2012-12-18 14:30:41 +00001692 && "Invalid index into preprocessed entity index remap");
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001693
Guy Benyei11169dd2012-12-18 14:30:41 +00001694 return LocalID + I->second;
1695}
1696
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001697unsigned HeaderFileInfoTrait::ComputeHash(internal_key_ref ikey) {
1698 return llvm::hash_combine(ikey.Size, ikey.ModTime);
Guy Benyei11169dd2012-12-18 14:30:41 +00001699}
Richard Smith7ed1bc92014-12-05 22:42:13 +00001700
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001701HeaderFileInfoTrait::internal_key_type
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001702HeaderFileInfoTrait::GetInternalKey(const FileEntry *FE) {
Richard Smithe75ee0f2015-08-17 07:13:32 +00001703 internal_key_type ikey = {FE->getSize(),
1704 M.HasTimestamps ? FE->getModificationTime() : 0,
1705 FE->getName(), /*Imported*/ false};
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001706 return ikey;
1707}
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001708
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001709bool HeaderFileInfoTrait::EqualKey(internal_key_ref a, internal_key_ref b) {
Richard Smithe75ee0f2015-08-17 07:13:32 +00001710 if (a.Size != b.Size || (a.ModTime && b.ModTime && a.ModTime != b.ModTime))
Guy Benyei11169dd2012-12-18 14:30:41 +00001711 return false;
1712
Mehdi Amini004b9c72016-10-10 22:52:47 +00001713 if (llvm::sys::path::is_absolute(a.Filename) && a.Filename == b.Filename)
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001714 return true;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001715
Guy Benyei11169dd2012-12-18 14:30:41 +00001716 // Determine whether the actual files are equivalent.
Argyrios Kyrtzidis2a513e82013-03-04 20:33:40 +00001717 FileManager &FileMgr = Reader.getFileManager();
Richard Smith7ed1bc92014-12-05 22:42:13 +00001718 auto GetFile = [&](const internal_key_type &Key) -> const FileEntry* {
1719 if (!Key.Imported)
1720 return FileMgr.getFile(Key.Filename);
1721
1722 std::string Resolved = Key.Filename;
1723 Reader.ResolveImportedPath(M, Resolved);
1724 return FileMgr.getFile(Resolved);
1725 };
1726
1727 const FileEntry *FEA = GetFile(a);
1728 const FileEntry *FEB = GetFile(b);
1729 return FEA && FEA == FEB;
Guy Benyei11169dd2012-12-18 14:30:41 +00001730}
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001731
Guy Benyei11169dd2012-12-18 14:30:41 +00001732std::pair<unsigned, unsigned>
1733HeaderFileInfoTrait::ReadKeyDataLength(const unsigned char*& d) {
Justin Bogner57ba0b22014-03-28 22:03:24 +00001734 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001735
Vedant Kumar48b4f762018-04-14 01:40:48 +00001736 unsigned KeyLen = (unsigned) endian::readNext<uint16_t, little, unaligned>(d);
1737 unsigned DataLen = (unsigned) *d++;
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001738 return std::make_pair(KeyLen, DataLen);
Guy Benyei11169dd2012-12-18 14:30:41 +00001739}
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001740
1741HeaderFileInfoTrait::internal_key_type
1742HeaderFileInfoTrait::ReadKey(const unsigned char *d, unsigned) {
Justin Bogner57ba0b22014-03-28 22:03:24 +00001743 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001744
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001745 internal_key_type ikey;
Justin Bogner57ba0b22014-03-28 22:03:24 +00001746 ikey.Size = off_t(endian::readNext<uint64_t, little, unaligned>(d));
1747 ikey.ModTime = time_t(endian::readNext<uint64_t, little, unaligned>(d));
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001748 ikey.Filename = (const char *)d;
Richard Smith7ed1bc92014-12-05 22:42:13 +00001749 ikey.Imported = true;
Argyrios Kyrtzidis5c2a3452013-03-06 18:12:47 +00001750 return ikey;
1751}
1752
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001753HeaderFileInfoTrait::data_type
Argyrios Kyrtzidisb146baa2013-03-13 21:13:51 +00001754HeaderFileInfoTrait::ReadData(internal_key_ref key, const unsigned char *d,
Guy Benyei11169dd2012-12-18 14:30:41 +00001755 unsigned DataLen) {
Justin Bogner57ba0b22014-03-28 22:03:24 +00001756 using namespace llvm::support;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001757
1758 const unsigned char *End = d + DataLen;
Guy Benyei11169dd2012-12-18 14:30:41 +00001759 HeaderFileInfo HFI;
1760 unsigned Flags = *d++;
Richard Smith386bb072015-08-18 23:42:23 +00001761 // FIXME: Refactor with mergeHeaderFileInfo in HeaderSearch.cpp.
Richard Smithf3f84612017-06-29 02:19:42 +00001762 HFI.isImport |= (Flags >> 5) & 0x01;
1763 HFI.isPragmaOnce |= (Flags >> 4) & 0x01;
1764 HFI.DirInfo = (Flags >> 1) & 0x07;
Guy Benyei11169dd2012-12-18 14:30:41 +00001765 HFI.IndexHeaderMapHeader = Flags & 0x01;
Richard Smith386bb072015-08-18 23:42:23 +00001766 // FIXME: Find a better way to handle this. Maybe just store a
1767 // "has been included" flag?
1768 HFI.NumIncludes = std::max(endian::readNext<uint16_t, little, unaligned>(d),
1769 HFI.NumIncludes);
Justin Bogner57ba0b22014-03-28 22:03:24 +00001770 HFI.ControllingMacroID = Reader.getGlobalIdentifierID(
1771 M, endian::readNext<uint32_t, little, unaligned>(d));
1772 if (unsigned FrameworkOffset =
1773 endian::readNext<uint32_t, little, unaligned>(d)) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001774 // The framework offset is 1 greater than the actual offset,
Guy Benyei11169dd2012-12-18 14:30:41 +00001775 // since 0 is used as an indicator for "no framework name".
1776 StringRef FrameworkName(FrameworkStrings + FrameworkOffset - 1);
1777 HFI.Framework = HS->getUniqueFrameworkName(FrameworkName);
1778 }
Richard Smith386bb072015-08-18 23:42:23 +00001779
1780 assert((End - d) % 4 == 0 &&
1781 "Wrong data length in HeaderFileInfo deserialization");
1782 while (d != End) {
Justin Bogner57ba0b22014-03-28 22:03:24 +00001783 uint32_t LocalSMID = endian::readNext<uint32_t, little, unaligned>(d);
Richard Smith386bb072015-08-18 23:42:23 +00001784 auto HeaderRole = static_cast<ModuleMap::ModuleHeaderRole>(LocalSMID & 3);
1785 LocalSMID >>= 2;
1786
1787 // This header is part of a module. Associate it with the module to enable
1788 // implicit module import.
1789 SubmoduleID GlobalSMID = Reader.getGlobalSubmoduleID(M, LocalSMID);
1790 Module *Mod = Reader.getSubmodule(GlobalSMID);
1791 FileManager &FileMgr = Reader.getFileManager();
1792 ModuleMap &ModMap =
1793 Reader.getPreprocessor().getHeaderSearchInfo().getModuleMap();
1794
1795 std::string Filename = key.Filename;
1796 if (key.Imported)
1797 Reader.ResolveImportedPath(M, Filename);
1798 // FIXME: This is not always the right filename-as-written, but we're not
1799 // going to use this information to rebuild the module, so it doesn't make
1800 // a lot of difference.
1801 Module::Header H = { key.Filename, FileMgr.getFile(Filename) };
Richard Smithd8879c82015-08-24 21:59:32 +00001802 ModMap.addHeader(Mod, H, HeaderRole, /*Imported*/true);
1803 HFI.isModuleHeader |= !(HeaderRole & ModuleMap::TextualHeader);
Argyrios Kyrtzidisb146baa2013-03-13 21:13:51 +00001804 }
1805
Guy Benyei11169dd2012-12-18 14:30:41 +00001806 // This HeaderFileInfo was externally loaded.
1807 HFI.External = true;
Richard Smithd8879c82015-08-24 21:59:32 +00001808 HFI.IsValid = true;
Guy Benyei11169dd2012-12-18 14:30:41 +00001809 return HFI;
1810}
1811
Richard Smithd7329392015-04-21 21:46:32 +00001812void ASTReader::addPendingMacro(IdentifierInfo *II,
1813 ModuleFile *M,
1814 uint64_t MacroDirectivesOffset) {
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001815 assert(NumCurrentElementsDeserializing > 0 &&"Missing deserialization guard");
1816 PendingMacroIDs[II].push_back(PendingMacroInfo(M, MacroDirectivesOffset));
Guy Benyei11169dd2012-12-18 14:30:41 +00001817}
1818
1819void ASTReader::ReadDefinedMacros() {
1820 // Note that we are loading defined macros.
1821 Deserializing Macros(this);
1822
Vedant Kumar48b4f762018-04-14 01:40:48 +00001823 for (ModuleFile &I : llvm::reverse(ModuleMgr)) {
Duncan P. N. Exon Smith96a06e02017-01-28 22:15:22 +00001824 BitstreamCursor &MacroCursor = I.MacroCursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00001825
1826 // If there was no preprocessor block, skip this file.
Peter Collingbourne77c89b62016-11-08 04:17:11 +00001827 if (MacroCursor.getBitcodeBytes().empty())
Guy Benyei11169dd2012-12-18 14:30:41 +00001828 continue;
1829
Chris Lattner7fb3bef2013-01-20 00:56:42 +00001830 BitstreamCursor Cursor = MacroCursor;
Duncan P. N. Exon Smith96a06e02017-01-28 22:15:22 +00001831 Cursor.JumpToBit(I.MacroStartOffset);
Guy Benyei11169dd2012-12-18 14:30:41 +00001832
1833 RecordData Record;
1834 while (true) {
Chris Lattnere7b154b2013-01-19 21:39:22 +00001835 llvm::BitstreamEntry E = Cursor.advanceSkippingSubblocks();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001836
Chris Lattnere7b154b2013-01-19 21:39:22 +00001837 switch (E.Kind) {
1838 case llvm::BitstreamEntry::SubBlock: // Handled for us already.
1839 case llvm::BitstreamEntry::Error:
1840 Error("malformed block record in AST file");
1841 return;
1842 case llvm::BitstreamEntry::EndBlock:
1843 goto NextCursor;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001844
Chris Lattnere7b154b2013-01-19 21:39:22 +00001845 case llvm::BitstreamEntry::Record:
Chris Lattnere7b154b2013-01-19 21:39:22 +00001846 Record.clear();
Chris Lattner0e6c9402013-01-20 02:38:54 +00001847 switch (Cursor.readRecord(E.ID, Record)) {
Chris Lattnere7b154b2013-01-19 21:39:22 +00001848 default: // Default behavior: ignore.
1849 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001850
Chris Lattnere7b154b2013-01-19 21:39:22 +00001851 case PP_MACRO_OBJECT_LIKE:
Sean Callananf3682a72016-05-14 06:24:14 +00001852 case PP_MACRO_FUNCTION_LIKE: {
Duncan P. N. Exon Smith96a06e02017-01-28 22:15:22 +00001853 IdentifierInfo *II = getLocalIdentifier(I, Record[0]);
Sean Callananf3682a72016-05-14 06:24:14 +00001854 if (II->isOutOfDate())
1855 updateOutOfDateIdentifier(*II);
Chris Lattnere7b154b2013-01-19 21:39:22 +00001856 break;
Sean Callananf3682a72016-05-14 06:24:14 +00001857 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001858
Chris Lattnere7b154b2013-01-19 21:39:22 +00001859 case PP_TOKEN:
1860 // Ignore tokens.
1861 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00001862 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001863 break;
1864 }
1865 }
Chris Lattnere7b154b2013-01-19 21:39:22 +00001866 NextCursor: ;
Guy Benyei11169dd2012-12-18 14:30:41 +00001867 }
1868}
1869
1870namespace {
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00001871
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00001872 /// Visitor class used to look up identifirs in an AST file.
Guy Benyei11169dd2012-12-18 14:30:41 +00001873 class IdentifierLookupVisitor {
1874 StringRef Name;
Richard Smith3b637412015-07-14 18:42:41 +00001875 unsigned NameHash;
Guy Benyei11169dd2012-12-18 14:30:41 +00001876 unsigned PriorGeneration;
Douglas Gregor00a50f72013-01-25 00:38:33 +00001877 unsigned &NumIdentifierLookups;
1878 unsigned &NumIdentifierLookupHits;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001879 IdentifierInfo *Found = nullptr;
Douglas Gregor00a50f72013-01-25 00:38:33 +00001880
Guy Benyei11169dd2012-12-18 14:30:41 +00001881 public:
Douglas Gregor00a50f72013-01-25 00:38:33 +00001882 IdentifierLookupVisitor(StringRef Name, unsigned PriorGeneration,
1883 unsigned &NumIdentifierLookups,
1884 unsigned &NumIdentifierLookupHits)
Richard Smith3b637412015-07-14 18:42:41 +00001885 : Name(Name), NameHash(ASTIdentifierLookupTrait::ComputeHash(Name)),
1886 PriorGeneration(PriorGeneration),
Douglas Gregor00a50f72013-01-25 00:38:33 +00001887 NumIdentifierLookups(NumIdentifierLookups),
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001888 NumIdentifierLookupHits(NumIdentifierLookupHits) {}
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00001889
1890 bool operator()(ModuleFile &M) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001891 // If we've already searched this module file, skip it now.
Richard Smithbdf2d932015-07-30 03:37:16 +00001892 if (M.Generation <= PriorGeneration)
Guy Benyei11169dd2012-12-18 14:30:41 +00001893 return true;
Douglas Gregore060e572013-01-25 01:03:03 +00001894
Vedant Kumar48b4f762018-04-14 01:40:48 +00001895 ASTIdentifierLookupTable *IdTable
1896 = (ASTIdentifierLookupTable *)M.IdentifierLookupTable;
Guy Benyei11169dd2012-12-18 14:30:41 +00001897 if (!IdTable)
1898 return false;
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00001899
1900 ASTIdentifierLookupTrait Trait(IdTable->getInfoObj().getReader(), M,
Richard Smithbdf2d932015-07-30 03:37:16 +00001901 Found);
1902 ++NumIdentifierLookups;
Richard Smith3b637412015-07-14 18:42:41 +00001903 ASTIdentifierLookupTable::iterator Pos =
Richard Smithbdf2d932015-07-30 03:37:16 +00001904 IdTable->find_hashed(Name, NameHash, &Trait);
Guy Benyei11169dd2012-12-18 14:30:41 +00001905 if (Pos == IdTable->end())
1906 return false;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001907
Guy Benyei11169dd2012-12-18 14:30:41 +00001908 // Dereferencing the iterator has the effect of building the
1909 // IdentifierInfo node and populating it with the various
1910 // declarations it needs.
Richard Smithbdf2d932015-07-30 03:37:16 +00001911 ++NumIdentifierLookupHits;
1912 Found = *Pos;
Guy Benyei11169dd2012-12-18 14:30:41 +00001913 return true;
1914 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001915
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00001916 // Retrieve the identifier info found within the module
Guy Benyei11169dd2012-12-18 14:30:41 +00001917 // files.
1918 IdentifierInfo *getIdentifierInfo() const { return Found; }
1919 };
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00001920
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00001921} // namespace
Guy Benyei11169dd2012-12-18 14:30:41 +00001922
1923void ASTReader::updateOutOfDateIdentifier(IdentifierInfo &II) {
1924 // Note that we are loading an identifier.
1925 Deserializing AnIdentifier(this);
1926
1927 unsigned PriorGeneration = 0;
1928 if (getContext().getLangOpts().Modules)
1929 PriorGeneration = IdentifierGeneration[&II];
Douglas Gregore060e572013-01-25 01:03:03 +00001930
1931 // If there is a global index, look there first to determine which modules
1932 // provably do not have any results for this identifier.
Douglas Gregor7211ac12013-01-25 23:32:03 +00001933 GlobalModuleIndex::HitSet Hits;
Craig Toppera13603a2014-05-22 05:54:18 +00001934 GlobalModuleIndex::HitSet *HitsPtr = nullptr;
Douglas Gregore060e572013-01-25 01:03:03 +00001935 if (!loadGlobalIndex()) {
Douglas Gregor7211ac12013-01-25 23:32:03 +00001936 if (GlobalIndex->lookupIdentifier(II.getName(), Hits)) {
1937 HitsPtr = &Hits;
Douglas Gregore060e572013-01-25 01:03:03 +00001938 }
1939 }
1940
Douglas Gregor7211ac12013-01-25 23:32:03 +00001941 IdentifierLookupVisitor Visitor(II.getName(), PriorGeneration,
Douglas Gregor00a50f72013-01-25 00:38:33 +00001942 NumIdentifierLookups,
1943 NumIdentifierLookupHits);
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00001944 ModuleMgr.visit(Visitor, HitsPtr);
Guy Benyei11169dd2012-12-18 14:30:41 +00001945 markIdentifierUpToDate(&II);
1946}
1947
1948void ASTReader::markIdentifierUpToDate(IdentifierInfo *II) {
1949 if (!II)
1950 return;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00001951
Guy Benyei11169dd2012-12-18 14:30:41 +00001952 II->setOutOfDate(false);
1953
1954 // Update the generation for this identifier.
1955 if (getContext().getLangOpts().Modules)
Richard Smith053f6c62014-05-16 23:01:30 +00001956 IdentifierGeneration[II] = getGeneration();
Guy Benyei11169dd2012-12-18 14:30:41 +00001957}
1958
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001959void ASTReader::resolvePendingMacro(IdentifierInfo *II,
1960 const PendingMacroInfo &PMInfo) {
Richard Smithd7329392015-04-21 21:46:32 +00001961 ModuleFile &M = *PMInfo.M;
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001962
1963 BitstreamCursor &Cursor = M.MacroCursor;
1964 SavedStreamPosition SavedPosition(Cursor);
Richard Smithd7329392015-04-21 21:46:32 +00001965 Cursor.JumpToBit(PMInfo.MacroDirectivesOffset);
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001966
Richard Smith713369b2015-04-23 20:40:50 +00001967 struct ModuleMacroRecord {
1968 SubmoduleID SubModID;
1969 MacroInfo *MI;
1970 SmallVector<SubmoduleID, 8> Overrides;
1971 };
1972 llvm::SmallVector<ModuleMacroRecord, 8> ModuleMacros;
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001973
Richard Smithd7329392015-04-21 21:46:32 +00001974 // We expect to see a sequence of PP_MODULE_MACRO records listing exported
1975 // macros, followed by a PP_MACRO_DIRECTIVE_HISTORY record with the complete
1976 // macro histroy.
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00001977 RecordData Record;
Richard Smithd7329392015-04-21 21:46:32 +00001978 while (true) {
1979 llvm::BitstreamEntry Entry =
1980 Cursor.advance(BitstreamCursor::AF_DontPopBlockAtEnd);
1981 if (Entry.Kind != llvm::BitstreamEntry::Record) {
1982 Error("malformed block record in AST file");
1983 return;
1984 }
1985
1986 Record.clear();
Aaron Ballmanc75a1922015-04-22 15:25:05 +00001987 switch ((PreprocessorRecordTypes)Cursor.readRecord(Entry.ID, Record)) {
Richard Smithd7329392015-04-21 21:46:32 +00001988 case PP_MACRO_DIRECTIVE_HISTORY:
1989 break;
1990
1991 case PP_MODULE_MACRO: {
Richard Smith713369b2015-04-23 20:40:50 +00001992 ModuleMacros.push_back(ModuleMacroRecord());
1993 auto &Info = ModuleMacros.back();
Richard Smithe56c8bc2015-04-22 00:26:11 +00001994 Info.SubModID = getGlobalSubmoduleID(M, Record[0]);
1995 Info.MI = getMacro(getGlobalMacroID(M, Record[1]));
Richard Smith713369b2015-04-23 20:40:50 +00001996 for (int I = 2, N = Record.size(); I != N; ++I)
1997 Info.Overrides.push_back(getGlobalSubmoduleID(M, Record[I]));
Richard Smithd7329392015-04-21 21:46:32 +00001998 continue;
1999 }
2000
2001 default:
2002 Error("malformed block record in AST file");
2003 return;
2004 }
2005
2006 // We found the macro directive history; that's the last record
2007 // for this macro.
2008 break;
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00002009 }
2010
Richard Smithd7329392015-04-21 21:46:32 +00002011 // Module macros are listed in reverse dependency order.
Richard Smithe56c8bc2015-04-22 00:26:11 +00002012 {
2013 std::reverse(ModuleMacros.begin(), ModuleMacros.end());
Richard Smithe56c8bc2015-04-22 00:26:11 +00002014 llvm::SmallVector<ModuleMacro*, 8> Overrides;
Richard Smith713369b2015-04-23 20:40:50 +00002015 for (auto &MMR : ModuleMacros) {
Richard Smithe56c8bc2015-04-22 00:26:11 +00002016 Overrides.clear();
Vedant Kumar48b4f762018-04-14 01:40:48 +00002017 for (unsigned ModID : MMR.Overrides) {
Richard Smithb8b2ed62015-04-23 18:18:26 +00002018 Module *Mod = getSubmodule(ModID);
2019 auto *Macro = PP.getModuleMacro(Mod, II);
Richard Smithe56c8bc2015-04-22 00:26:11 +00002020 assert(Macro && "missing definition for overridden macro");
Richard Smith5dbef922015-04-22 02:09:43 +00002021 Overrides.push_back(Macro);
Richard Smithe56c8bc2015-04-22 00:26:11 +00002022 }
2023
2024 bool Inserted = false;
Richard Smith713369b2015-04-23 20:40:50 +00002025 Module *Owner = getSubmodule(MMR.SubModID);
Richard Smith20e883e2015-04-29 23:20:19 +00002026 PP.addModuleMacro(Owner, II, MMR.MI, Overrides, Inserted);
Richard Smithd7329392015-04-21 21:46:32 +00002027 }
2028 }
2029
2030 // Don't read the directive history for a module; we don't have anywhere
2031 // to put it.
Manman Ren11f2a472016-08-18 17:42:15 +00002032 if (M.isModule())
Richard Smithd7329392015-04-21 21:46:32 +00002033 return;
2034
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00002035 // Deserialize the macro directives history in reverse source-order.
Craig Toppera13603a2014-05-22 05:54:18 +00002036 MacroDirective *Latest = nullptr, *Earliest = nullptr;
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00002037 unsigned Idx = 0, N = Record.size();
2038 while (Idx < N) {
Craig Toppera13603a2014-05-22 05:54:18 +00002039 MacroDirective *MD = nullptr;
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00002040 SourceLocation Loc = ReadSourceLocation(M, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00002041 MacroDirective::Kind K = (MacroDirective::Kind)Record[Idx++];
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002042 switch (K) {
2043 case MacroDirective::MD_Define: {
Richard Smith713369b2015-04-23 20:40:50 +00002044 MacroInfo *MI = getMacro(getGlobalMacroID(M, Record[Idx++]));
Richard Smith3981b172015-04-30 02:16:23 +00002045 MD = PP.AllocateDefMacroDirective(MI, Loc);
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002046 break;
2047 }
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00002048 case MacroDirective::MD_Undefine:
Richard Smith3981b172015-04-30 02:16:23 +00002049 MD = PP.AllocateUndefMacroDirective(Loc);
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002050 break;
Richard Smithdaa69e02014-07-25 04:40:03 +00002051 case MacroDirective::MD_Visibility:
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002052 bool isPublic = Record[Idx++];
2053 MD = PP.AllocateVisibilityMacroDirective(Loc, isPublic);
2054 break;
2055 }
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00002056
2057 if (!Latest)
2058 Latest = MD;
2059 if (Earliest)
2060 Earliest->setPrevious(MD);
2061 Earliest = MD;
2062 }
2063
Richard Smithd6e8c0d2015-05-04 19:58:00 +00002064 if (Latest)
Nico Weberfd870702016-12-09 17:32:52 +00002065 PP.setLoadedMacroDirective(II, Earliest, Latest);
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00002066}
2067
Argyrios Kyrtzidisce9b49e2014-03-14 02:26:27 +00002068ASTReader::InputFileInfo
2069ASTReader::readInputFileInfo(ModuleFile &F, unsigned ID) {
Ben Langmuir198c1682014-03-07 07:27:49 +00002070 // Go find this input file.
2071 BitstreamCursor &Cursor = F.InputFilesCursor;
2072 SavedStreamPosition SavedPosition(Cursor);
2073 Cursor.JumpToBit(F.InputFileOffsets[ID-1]);
2074
2075 unsigned Code = Cursor.ReadCode();
2076 RecordData Record;
2077 StringRef Blob;
2078
2079 unsigned Result = Cursor.readRecord(Code, Record, &Blob);
2080 assert(static_cast<InputFileRecordTypes>(Result) == INPUT_FILE &&
2081 "invalid record type for input file");
2082 (void)Result;
2083
2084 assert(Record[0] == ID && "Bogus stored ID or offset");
Richard Smitha8cfffa2015-11-26 02:04:16 +00002085 InputFileInfo R;
2086 R.StoredSize = static_cast<off_t>(Record[1]);
2087 R.StoredTime = static_cast<time_t>(Record[2]);
2088 R.Overridden = static_cast<bool>(Record[3]);
2089 R.Transient = static_cast<bool>(Record[4]);
Richard Smithf3f84612017-06-29 02:19:42 +00002090 R.TopLevelModuleMap = static_cast<bool>(Record[5]);
Richard Smitha8cfffa2015-11-26 02:04:16 +00002091 R.Filename = Blob;
2092 ResolveImportedPath(F, R.Filename);
Hans Wennborg73945142014-03-14 17:45:06 +00002093 return R;
Ben Langmuir198c1682014-03-07 07:27:49 +00002094}
2095
Manman Renc8c94152016-10-21 23:35:03 +00002096static unsigned moduleKindForDiagnostic(ModuleKind Kind);
Argyrios Kyrtzidis61c3d872013-03-01 03:26:04 +00002097InputFile ASTReader::getInputFile(ModuleFile &F, unsigned ID, bool Complain) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002098 // If this ID is bogus, just return an empty input file.
2099 if (ID == 0 || ID > F.InputFilesLoaded.size())
2100 return InputFile();
2101
2102 // If we've already loaded this input file, return it.
Argyrios Kyrtzidis61c3d872013-03-01 03:26:04 +00002103 if (F.InputFilesLoaded[ID-1].getFile())
Guy Benyei11169dd2012-12-18 14:30:41 +00002104 return F.InputFilesLoaded[ID-1];
2105
Argyrios Kyrtzidis9308f0a2014-01-08 19:13:34 +00002106 if (F.InputFilesLoaded[ID-1].isNotFound())
2107 return InputFile();
2108
Guy Benyei11169dd2012-12-18 14:30:41 +00002109 // Go find this input file.
Chris Lattner7fb3bef2013-01-20 00:56:42 +00002110 BitstreamCursor &Cursor = F.InputFilesCursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00002111 SavedStreamPosition SavedPosition(Cursor);
2112 Cursor.JumpToBit(F.InputFileOffsets[ID-1]);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002113
Argyrios Kyrtzidisce9b49e2014-03-14 02:26:27 +00002114 InputFileInfo FI = readInputFileInfo(F, ID);
2115 off_t StoredSize = FI.StoredSize;
2116 time_t StoredTime = FI.StoredTime;
2117 bool Overridden = FI.Overridden;
Richard Smitha8cfffa2015-11-26 02:04:16 +00002118 bool Transient = FI.Transient;
Argyrios Kyrtzidisce9b49e2014-03-14 02:26:27 +00002119 StringRef Filename = FI.Filename;
Argyrios Kyrtzidis61c3d872013-03-01 03:26:04 +00002120
Richard Smitha8cfffa2015-11-26 02:04:16 +00002121 const FileEntry *File = FileMgr.getFile(Filename, /*OpenFile=*/false);
Ben Langmuir198c1682014-03-07 07:27:49 +00002122 // If we didn't find the file, resolve it relative to the
2123 // original directory from which this AST file was created.
Manuel Klimek1b29b4f2017-07-25 10:22:06 +00002124 if (File == nullptr && !F.OriginalDir.empty() && !F.BaseDirectory.empty() &&
2125 F.OriginalDir != F.BaseDirectory) {
2126 std::string Resolved = resolveFileRelativeToOriginalDir(
2127 Filename, F.OriginalDir, F.BaseDirectory);
Ben Langmuir198c1682014-03-07 07:27:49 +00002128 if (!Resolved.empty())
2129 File = FileMgr.getFile(Resolved);
2130 }
2131
2132 // For an overridden file, create a virtual file with the stored
2133 // size/timestamp.
Richard Smitha8cfffa2015-11-26 02:04:16 +00002134 if ((Overridden || Transient) && File == nullptr)
Ben Langmuir198c1682014-03-07 07:27:49 +00002135 File = FileMgr.getVirtualFile(Filename, StoredSize, StoredTime);
Ben Langmuir198c1682014-03-07 07:27:49 +00002136
Craig Toppera13603a2014-05-22 05:54:18 +00002137 if (File == nullptr) {
Ben Langmuir198c1682014-03-07 07:27:49 +00002138 if (Complain) {
2139 std::string ErrorStr = "could not find file '";
2140 ErrorStr += Filename;
Richard Smith68142212015-10-13 01:26:26 +00002141 ErrorStr += "' referenced by AST file '";
2142 ErrorStr += F.FileName;
2143 ErrorStr += "'";
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00002144 Error(ErrorStr);
Guy Benyei11169dd2012-12-18 14:30:41 +00002145 }
Ben Langmuir198c1682014-03-07 07:27:49 +00002146 // Record that we didn't find the file.
2147 F.InputFilesLoaded[ID-1] = InputFile::getNotFound();
2148 return InputFile();
2149 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002150
Ben Langmuir198c1682014-03-07 07:27:49 +00002151 // Check if there was a request to override the contents of the file
Alexander Kornienko2a8c18d2018-04-06 15:14:32 +00002152 // that was part of the precompiled header. Overriding such a file
Ben Langmuir198c1682014-03-07 07:27:49 +00002153 // can lead to problems when lexing using the source locations from the
2154 // PCH.
2155 SourceManager &SM = getSourceManager();
Richard Smith64daf7b2015-12-01 03:32:49 +00002156 // FIXME: Reject if the overrides are different.
2157 if ((!Overridden && !Transient) && SM.isFileOverridden(File)) {
Ben Langmuir198c1682014-03-07 07:27:49 +00002158 if (Complain)
2159 Error(diag::err_fe_pch_file_overridden, Filename);
2160 // After emitting the diagnostic, recover by disabling the override so
2161 // that the original file will be used.
Richard Smitha8cfffa2015-11-26 02:04:16 +00002162 //
2163 // FIXME: This recovery is just as broken as the original state; there may
2164 // be another precompiled module that's using the overridden contents, or
2165 // we might be half way through parsing it. Instead, we should treat the
2166 // overridden contents as belonging to a separate FileEntry.
Ben Langmuir198c1682014-03-07 07:27:49 +00002167 SM.disableFileContentsOverride(File);
2168 // The FileEntry is a virtual file entry with the size of the contents
2169 // that would override the original contents. Set it to the original's
2170 // size/time.
2171 FileMgr.modifyFileEntry(const_cast<FileEntry*>(File),
2172 StoredSize, StoredTime);
2173 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002174
Ben Langmuir198c1682014-03-07 07:27:49 +00002175 bool IsOutOfDate = false;
2176
2177 // For an overridden file, there is nothing to validate.
Richard Smith96fdab62014-10-28 16:24:08 +00002178 if (!Overridden && //
2179 (StoredSize != File->getSize() ||
Richard Smithe75ee0f2015-08-17 07:13:32 +00002180 (StoredTime && StoredTime != File->getModificationTime() &&
2181 !DisableValidation)
Ben Langmuir198c1682014-03-07 07:27:49 +00002182 )) {
2183 if (Complain) {
2184 // Build a list of the PCH imports that got us here (in reverse).
2185 SmallVector<ModuleFile *, 4> ImportStack(1, &F);
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00002186 while (!ImportStack.back()->ImportedBy.empty())
Ben Langmuir198c1682014-03-07 07:27:49 +00002187 ImportStack.push_back(ImportStack.back()->ImportedBy[0]);
Ben Langmuire82630d2014-01-17 00:19:09 +00002188
Ben Langmuir198c1682014-03-07 07:27:49 +00002189 // The top-level PCH is stale.
2190 StringRef TopLevelPCHName(ImportStack.back()->FileName);
Manman Renc8c94152016-10-21 23:35:03 +00002191 unsigned DiagnosticKind = moduleKindForDiagnostic(ImportStack.back()->Kind);
2192 if (DiagnosticKind == 0)
2193 Error(diag::err_fe_pch_file_modified, Filename, TopLevelPCHName);
2194 else if (DiagnosticKind == 1)
2195 Error(diag::err_fe_module_file_modified, Filename, TopLevelPCHName);
2196 else
2197 Error(diag::err_fe_ast_file_modified, Filename, TopLevelPCHName);
Ben Langmuire82630d2014-01-17 00:19:09 +00002198
Ben Langmuir198c1682014-03-07 07:27:49 +00002199 // Print the import stack.
2200 if (ImportStack.size() > 1 && !Diags.isDiagnosticInFlight()) {
2201 Diag(diag::note_pch_required_by)
2202 << Filename << ImportStack[0]->FileName;
2203 for (unsigned I = 1; I < ImportStack.size(); ++I)
Ben Langmuire82630d2014-01-17 00:19:09 +00002204 Diag(diag::note_pch_required_by)
Ben Langmuir198c1682014-03-07 07:27:49 +00002205 << ImportStack[I-1]->FileName << ImportStack[I]->FileName;
Douglas Gregor7029ce12013-03-19 00:28:20 +00002206 }
2207
Ben Langmuir198c1682014-03-07 07:27:49 +00002208 if (!Diags.isDiagnosticInFlight())
2209 Diag(diag::note_pch_rebuild_required) << TopLevelPCHName;
Guy Benyei11169dd2012-12-18 14:30:41 +00002210 }
2211
Ben Langmuir198c1682014-03-07 07:27:49 +00002212 IsOutOfDate = true;
Guy Benyei11169dd2012-12-18 14:30:41 +00002213 }
Richard Smitha8cfffa2015-11-26 02:04:16 +00002214 // FIXME: If the file is overridden and we've already opened it,
2215 // issue an error (or split it into a separate FileEntry).
Guy Benyei11169dd2012-12-18 14:30:41 +00002216
Richard Smitha8cfffa2015-11-26 02:04:16 +00002217 InputFile IF = InputFile(File, Overridden || Transient, IsOutOfDate);
Ben Langmuir198c1682014-03-07 07:27:49 +00002218
2219 // Note that we've loaded this input file.
2220 F.InputFilesLoaded[ID-1] = IF;
2221 return IF;
Guy Benyei11169dd2012-12-18 14:30:41 +00002222}
2223
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00002224/// If we are loading a relocatable PCH or module file, and the filename
Richard Smith7ed1bc92014-12-05 22:42:13 +00002225/// is not an absolute path, add the system or module root to the beginning of
2226/// the file name.
2227void ASTReader::ResolveImportedPath(ModuleFile &M, std::string &Filename) {
2228 // Resolve relative to the base directory, if we have one.
2229 if (!M.BaseDirectory.empty())
2230 return ResolveImportedPath(Filename, M.BaseDirectory);
Guy Benyei11169dd2012-12-18 14:30:41 +00002231}
2232
Richard Smith7ed1bc92014-12-05 22:42:13 +00002233void ASTReader::ResolveImportedPath(std::string &Filename, StringRef Prefix) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002234 if (Filename.empty() || llvm::sys::path::is_absolute(Filename))
2235 return;
2236
Richard Smith7ed1bc92014-12-05 22:42:13 +00002237 SmallString<128> Buffer;
2238 llvm::sys::path::append(Buffer, Prefix, Filename);
2239 Filename.assign(Buffer.begin(), Buffer.end());
Guy Benyei11169dd2012-12-18 14:30:41 +00002240}
2241
Richard Smith0f99d6a2015-08-09 08:48:41 +00002242static bool isDiagnosedResult(ASTReader::ASTReadResult ARR, unsigned Caps) {
2243 switch (ARR) {
2244 case ASTReader::Failure: return true;
2245 case ASTReader::Missing: return !(Caps & ASTReader::ARR_Missing);
2246 case ASTReader::OutOfDate: return !(Caps & ASTReader::ARR_OutOfDate);
2247 case ASTReader::VersionMismatch: return !(Caps & ASTReader::ARR_VersionMismatch);
2248 case ASTReader::ConfigurationMismatch:
2249 return !(Caps & ASTReader::ARR_ConfigurationMismatch);
2250 case ASTReader::HadErrors: return true;
2251 case ASTReader::Success: return false;
2252 }
2253
2254 llvm_unreachable("unknown ASTReadResult");
2255}
2256
Richard Smith0516b182015-09-08 19:40:14 +00002257ASTReader::ASTReadResult ASTReader::ReadOptionsBlock(
2258 BitstreamCursor &Stream, unsigned ClientLoadCapabilities,
2259 bool AllowCompatibleConfigurationMismatch, ASTReaderListener &Listener,
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00002260 std::string &SuggestedPredefines) {
Richard Smith0516b182015-09-08 19:40:14 +00002261 if (Stream.EnterSubBlock(OPTIONS_BLOCK_ID))
2262 return Failure;
2263
2264 // Read all of the records in the options block.
2265 RecordData Record;
2266 ASTReadResult Result = Success;
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00002267 while (true) {
Richard Smith0516b182015-09-08 19:40:14 +00002268 llvm::BitstreamEntry Entry = Stream.advance();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002269
Richard Smith0516b182015-09-08 19:40:14 +00002270 switch (Entry.Kind) {
2271 case llvm::BitstreamEntry::Error:
2272 case llvm::BitstreamEntry::SubBlock:
2273 return Failure;
2274
2275 case llvm::BitstreamEntry::EndBlock:
2276 return Result;
2277
2278 case llvm::BitstreamEntry::Record:
2279 // The interesting case.
2280 break;
2281 }
2282
2283 // Read and process a record.
2284 Record.clear();
2285 switch ((OptionsRecordTypes)Stream.readRecord(Entry.ID, Record)) {
2286 case LANGUAGE_OPTIONS: {
2287 bool Complain = (ClientLoadCapabilities & ARR_ConfigurationMismatch) == 0;
2288 if (ParseLanguageOptions(Record, Complain, Listener,
2289 AllowCompatibleConfigurationMismatch))
2290 Result = ConfigurationMismatch;
2291 break;
2292 }
2293
2294 case TARGET_OPTIONS: {
2295 bool Complain = (ClientLoadCapabilities & ARR_ConfigurationMismatch) == 0;
2296 if (ParseTargetOptions(Record, Complain, Listener,
2297 AllowCompatibleConfigurationMismatch))
2298 Result = ConfigurationMismatch;
2299 break;
2300 }
2301
Richard Smith0516b182015-09-08 19:40:14 +00002302 case FILE_SYSTEM_OPTIONS: {
2303 bool Complain = (ClientLoadCapabilities & ARR_ConfigurationMismatch) == 0;
2304 if (!AllowCompatibleConfigurationMismatch &&
2305 ParseFileSystemOptions(Record, Complain, Listener))
2306 Result = ConfigurationMismatch;
2307 break;
2308 }
2309
2310 case HEADER_SEARCH_OPTIONS: {
2311 bool Complain = (ClientLoadCapabilities & ARR_ConfigurationMismatch) == 0;
2312 if (!AllowCompatibleConfigurationMismatch &&
2313 ParseHeaderSearchOptions(Record, Complain, Listener))
2314 Result = ConfigurationMismatch;
2315 break;
2316 }
2317
2318 case PREPROCESSOR_OPTIONS:
2319 bool Complain = (ClientLoadCapabilities & ARR_ConfigurationMismatch) == 0;
2320 if (!AllowCompatibleConfigurationMismatch &&
2321 ParsePreprocessorOptions(Record, Complain, Listener,
2322 SuggestedPredefines))
2323 Result = ConfigurationMismatch;
2324 break;
2325 }
2326 }
2327}
2328
Guy Benyei11169dd2012-12-18 14:30:41 +00002329ASTReader::ASTReadResult
2330ASTReader::ReadControlBlock(ModuleFile &F,
Dmitri Gribenkof8579502013-01-12 19:30:44 +00002331 SmallVectorImpl<ImportedModule> &Loaded,
Ben Langmuirbeee15e2014-04-14 18:00:01 +00002332 const ModuleFile *ImportedBy,
Guy Benyei11169dd2012-12-18 14:30:41 +00002333 unsigned ClientLoadCapabilities) {
Chris Lattner7fb3bef2013-01-20 00:56:42 +00002334 BitstreamCursor &Stream = F.Stream;
Richard Smith8a308ec2015-11-05 00:54:55 +00002335 ASTReadResult Result = Success;
Guy Benyei11169dd2012-12-18 14:30:41 +00002336
2337 if (Stream.EnterSubBlock(CONTROL_BLOCK_ID)) {
2338 Error("malformed block record in AST file");
2339 return Failure;
2340 }
2341
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00002342 // Lambda to read the unhashed control block the first time it's called.
2343 //
2344 // For PCM files, the unhashed control block cannot be read until after the
2345 // MODULE_NAME record. However, PCH files have no MODULE_NAME, and yet still
2346 // need to look ahead before reading the IMPORTS record. For consistency,
2347 // this block is always read somehow (see BitstreamEntry::EndBlock).
2348 bool HasReadUnhashedControlBlock = false;
2349 auto readUnhashedControlBlockOnce = [&]() {
2350 if (!HasReadUnhashedControlBlock) {
2351 HasReadUnhashedControlBlock = true;
2352 if (ASTReadResult Result =
2353 readUnhashedControlBlock(F, ImportedBy, ClientLoadCapabilities))
2354 return Result;
2355 }
2356 return Success;
2357 };
2358
Guy Benyei11169dd2012-12-18 14:30:41 +00002359 // Read all of the records and blocks in the control block.
2360 RecordData Record;
Richard Smitha1825302014-10-23 22:18:29 +00002361 unsigned NumInputs = 0;
2362 unsigned NumUserInputs = 0;
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00002363 while (true) {
Chris Lattnere7b154b2013-01-19 21:39:22 +00002364 llvm::BitstreamEntry Entry = Stream.advance();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002365
Chris Lattnere7b154b2013-01-19 21:39:22 +00002366 switch (Entry.Kind) {
2367 case llvm::BitstreamEntry::Error:
2368 Error("malformed block record in AST file");
2369 return Failure;
Dmitri Gribenkof430da42014-02-12 10:33:14 +00002370 case llvm::BitstreamEntry::EndBlock: {
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00002371 // Validate the module before returning. This call catches an AST with
2372 // no module name and no imports.
2373 if (ASTReadResult Result = readUnhashedControlBlockOnce())
2374 return Result;
2375
Dmitri Gribenkof430da42014-02-12 10:33:14 +00002376 // Validate input files.
2377 const HeaderSearchOptions &HSOpts =
2378 PP.getHeaderSearchInfo().getHeaderSearchOpts();
Ben Langmuircb69b572014-03-07 06:40:32 +00002379
Richard Smitha1825302014-10-23 22:18:29 +00002380 // All user input files reside at the index range [0, NumUserInputs), and
Richard Smith0f99d6a2015-08-09 08:48:41 +00002381 // system input files reside at [NumUserInputs, NumInputs). For explicitly
2382 // loaded module files, ignore missing inputs.
Manman Ren11f2a472016-08-18 17:42:15 +00002383 if (!DisableValidation && F.Kind != MK_ExplicitModule &&
2384 F.Kind != MK_PrebuiltModule) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002385 bool Complain = (ClientLoadCapabilities & ARR_OutOfDate) == 0;
Ben Langmuircb69b572014-03-07 06:40:32 +00002386
Dmitri Gribenkof430da42014-02-12 10:33:14 +00002387 // If we are reading a module, we will create a verification timestamp,
2388 // so we verify all input files. Otherwise, verify only user input
2389 // files.
Ben Langmuircb69b572014-03-07 06:40:32 +00002390
2391 unsigned N = NumUserInputs;
2392 if (ValidateSystemInputs ||
Richard Smithe842a472014-10-22 02:05:46 +00002393 (HSOpts.ModulesValidateOncePerBuildSession &&
Ben Langmuiracb803e2014-11-10 22:13:10 +00002394 F.InputFilesValidationTimestamp <= HSOpts.BuildSessionTimestamp &&
Richard Smithe842a472014-10-22 02:05:46 +00002395 F.Kind == MK_ImplicitModule))
Ben Langmuircb69b572014-03-07 06:40:32 +00002396 N = NumInputs;
2397
Ben Langmuir3d4417c2014-02-07 17:31:11 +00002398 for (unsigned I = 0; I < N; ++I) {
Argyrios Kyrtzidis61c3d872013-03-01 03:26:04 +00002399 InputFile IF = getInputFile(F, I+1, Complain);
2400 if (!IF.getFile() || IF.isOutOfDate())
Guy Benyei11169dd2012-12-18 14:30:41 +00002401 return OutOfDate;
Argyrios Kyrtzidis61c3d872013-03-01 03:26:04 +00002402 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002403 }
Ben Langmuircb69b572014-03-07 06:40:32 +00002404
Argyrios Kyrtzidis6d0753d2014-03-14 03:07:38 +00002405 if (Listener)
Richard Smith216a3bd2015-08-13 17:57:10 +00002406 Listener->visitModuleFile(F.FileName, F.Kind);
Argyrios Kyrtzidis6d0753d2014-03-14 03:07:38 +00002407
Ben Langmuircb69b572014-03-07 06:40:32 +00002408 if (Listener && Listener->needsInputFileVisitation()) {
2409 unsigned N = Listener->needsSystemInputFileVisitation() ? NumInputs
2410 : NumUserInputs;
Argyrios Kyrtzidis68ccbe02014-03-14 02:26:31 +00002411 for (unsigned I = 0; I < N; ++I) {
2412 bool IsSystem = I >= NumUserInputs;
2413 InputFileInfo FI = readInputFileInfo(F, I+1);
Richard Smith216a3bd2015-08-13 17:57:10 +00002414 Listener->visitInputFile(FI.Filename, IsSystem, FI.Overridden,
Manman Ren11f2a472016-08-18 17:42:15 +00002415 F.Kind == MK_ExplicitModule ||
2416 F.Kind == MK_PrebuiltModule);
Argyrios Kyrtzidis68ccbe02014-03-14 02:26:31 +00002417 }
Ben Langmuircb69b572014-03-07 06:40:32 +00002418 }
2419
Richard Smith8a308ec2015-11-05 00:54:55 +00002420 return Result;
Dmitri Gribenkof430da42014-02-12 10:33:14 +00002421 }
2422
Chris Lattnere7b154b2013-01-19 21:39:22 +00002423 case llvm::BitstreamEntry::SubBlock:
2424 switch (Entry.ID) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002425 case INPUT_FILES_BLOCK_ID:
2426 F.InputFilesCursor = Stream;
2427 if (Stream.SkipBlock() || // Skip with the main cursor
2428 // Read the abbreviations
2429 ReadBlockAbbrevs(F.InputFilesCursor, INPUT_FILES_BLOCK_ID)) {
2430 Error("malformed block record in AST file");
2431 return Failure;
2432 }
2433 continue;
Richard Smith0516b182015-09-08 19:40:14 +00002434
2435 case OPTIONS_BLOCK_ID:
2436 // If we're reading the first module for this group, check its options
2437 // are compatible with ours. For modules it imports, no further checking
2438 // is required, because we checked them when we built it.
2439 if (Listener && !ImportedBy) {
2440 // Should we allow the configuration of the module file to differ from
2441 // the configuration of the current translation unit in a compatible
2442 // way?
2443 //
2444 // FIXME: Allow this for files explicitly specified with -include-pch.
2445 bool AllowCompatibleConfigurationMismatch =
Manman Ren11f2a472016-08-18 17:42:15 +00002446 F.Kind == MK_ExplicitModule || F.Kind == MK_PrebuiltModule;
Richard Smith0516b182015-09-08 19:40:14 +00002447
Richard Smith8a308ec2015-11-05 00:54:55 +00002448 Result = ReadOptionsBlock(Stream, ClientLoadCapabilities,
2449 AllowCompatibleConfigurationMismatch,
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00002450 *Listener, SuggestedPredefines);
Richard Smith0516b182015-09-08 19:40:14 +00002451 if (Result == Failure) {
2452 Error("malformed block record in AST file");
2453 return Result;
2454 }
2455
Richard Smith8a308ec2015-11-05 00:54:55 +00002456 if (DisableValidation ||
2457 (AllowConfigurationMismatch && Result == ConfigurationMismatch))
2458 Result = Success;
2459
Ben Langmuir9b1e442e2016-02-11 18:54:02 +00002460 // If we can't load the module, exit early since we likely
2461 // will rebuild the module anyway. The stream may be in the
2462 // middle of a block.
2463 if (Result != Success)
Richard Smith0516b182015-09-08 19:40:14 +00002464 return Result;
2465 } else if (Stream.SkipBlock()) {
2466 Error("malformed block record in AST file");
2467 return Failure;
2468 }
2469 continue;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002470
Guy Benyei11169dd2012-12-18 14:30:41 +00002471 default:
Chris Lattnere7b154b2013-01-19 21:39:22 +00002472 if (Stream.SkipBlock()) {
2473 Error("malformed block record in AST file");
2474 return Failure;
2475 }
2476 continue;
Guy Benyei11169dd2012-12-18 14:30:41 +00002477 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002478
Chris Lattnere7b154b2013-01-19 21:39:22 +00002479 case llvm::BitstreamEntry::Record:
2480 // The interesting case.
2481 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00002482 }
2483
2484 // Read and process a record.
2485 Record.clear();
Chris Lattner0e6c9402013-01-20 02:38:54 +00002486 StringRef Blob;
2487 switch ((ControlRecordTypes)Stream.readRecord(Entry.ID, Record, &Blob)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002488 case METADATA: {
2489 if (Record[0] != VERSION_MAJOR && !DisableValidation) {
2490 if ((ClientLoadCapabilities & ARR_VersionMismatch) == 0)
Dmitri Gribenko2228cd32014-02-11 15:40:09 +00002491 Diag(Record[0] < VERSION_MAJOR? diag::err_pch_version_too_old
2492 : diag::err_pch_version_too_new);
Guy Benyei11169dd2012-12-18 14:30:41 +00002493 return VersionMismatch;
2494 }
2495
Hans Wennborg08c5a7b2018-06-25 13:23:49 +00002496 bool hasErrors = Record[7];
Guy Benyei11169dd2012-12-18 14:30:41 +00002497 if (hasErrors && !DisableValidation && !AllowASTWithCompilerErrors) {
2498 Diag(diag::err_pch_with_compiler_errors);
2499 return HadErrors;
2500 }
Argyrios Kyrtzidis70ec1c72016-07-13 20:35:26 +00002501 if (hasErrors) {
2502 Diags.ErrorOccurred = true;
2503 Diags.UncompilableErrorOccurred = true;
2504 Diags.UnrecoverableErrorOccurred = true;
2505 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002506
2507 F.RelocatablePCH = Record[4];
Richard Smith7ed1bc92014-12-05 22:42:13 +00002508 // Relative paths in a relocatable PCH are relative to our sysroot.
2509 if (F.RelocatablePCH)
2510 F.BaseDirectory = isysroot.empty() ? "/" : isysroot;
Guy Benyei11169dd2012-12-18 14:30:41 +00002511
Richard Smithe75ee0f2015-08-17 07:13:32 +00002512 F.HasTimestamps = Record[5];
2513
Hans Wennborg08c5a7b2018-06-25 13:23:49 +00002514 F.PCHHasObjectFile = Record[6];
2515
Guy Benyei11169dd2012-12-18 14:30:41 +00002516 const std::string &CurBranch = getClangFullRepositoryVersion();
Chris Lattner0e6c9402013-01-20 02:38:54 +00002517 StringRef ASTBranch = Blob;
Guy Benyei11169dd2012-12-18 14:30:41 +00002518 if (StringRef(CurBranch) != ASTBranch && !DisableValidation) {
2519 if ((ClientLoadCapabilities & ARR_VersionMismatch) == 0)
Dmitri Gribenko2228cd32014-02-11 15:40:09 +00002520 Diag(diag::err_pch_different_branch) << ASTBranch << CurBranch;
Guy Benyei11169dd2012-12-18 14:30:41 +00002521 return VersionMismatch;
2522 }
2523 break;
2524 }
2525
2526 case IMPORTS: {
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00002527 // Validate the AST before processing any imports (otherwise, untangling
2528 // them can be error-prone and expensive). A module will have a name and
2529 // will already have been validated, but this catches the PCH case.
2530 if (ASTReadResult Result = readUnhashedControlBlockOnce())
2531 return Result;
2532
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002533 // Load each of the imported PCH files.
Guy Benyei11169dd2012-12-18 14:30:41 +00002534 unsigned Idx = 0, N = Record.size();
2535 while (Idx < N) {
2536 // Read information about the AST file.
Vedant Kumar48b4f762018-04-14 01:40:48 +00002537 ModuleKind ImportedKind = (ModuleKind)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00002538 // The import location will be the local one for now; we will adjust
2539 // all import locations of module imports after the global source
Richard Smithb22a1d12016-03-27 20:13:24 +00002540 // location info are setup, in ReadAST.
Guy Benyei11169dd2012-12-18 14:30:41 +00002541 SourceLocation ImportLoc =
Richard Smithb22a1d12016-03-27 20:13:24 +00002542 ReadUntranslatedSourceLocation(Record[Idx++]);
Vedant Kumar48b4f762018-04-14 01:40:48 +00002543 off_t StoredSize = (off_t)Record[Idx++];
2544 time_t StoredModTime = (time_t)Record[Idx++];
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00002545 ASTFileSignature StoredSignature = {
2546 {{(uint32_t)Record[Idx++], (uint32_t)Record[Idx++],
2547 (uint32_t)Record[Idx++], (uint32_t)Record[Idx++],
2548 (uint32_t)Record[Idx++]}}};
Boris Kolpackovd30446f2017-08-31 06:26:43 +00002549
2550 std::string ImportedName = ReadString(Record, Idx);
2551 std::string ImportedFile;
2552
2553 // For prebuilt and explicit modules first consult the file map for
2554 // an override. Note that here we don't search prebuilt module
2555 // directories, only the explicit name to file mappings. Also, we will
2556 // still verify the size/signature making sure it is essentially the
2557 // same file but perhaps in a different location.
2558 if (ImportedKind == MK_PrebuiltModule || ImportedKind == MK_ExplicitModule)
2559 ImportedFile = PP.getHeaderSearchInfo().getPrebuiltModuleFileName(
2560 ImportedName, /*FileMapOnly*/ true);
2561
2562 if (ImportedFile.empty())
2563 ImportedFile = ReadPath(F, Record, Idx);
2564 else
2565 SkipPath(Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00002566
Richard Smith0f99d6a2015-08-09 08:48:41 +00002567 // If our client can't cope with us being out of date, we can't cope with
2568 // our dependency being missing.
2569 unsigned Capabilities = ClientLoadCapabilities;
2570 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0)
2571 Capabilities &= ~ARR_Missing;
2572
Guy Benyei11169dd2012-12-18 14:30:41 +00002573 // Load the AST file.
Richard Smith0f99d6a2015-08-09 08:48:41 +00002574 auto Result = ReadASTCore(ImportedFile, ImportedKind, ImportLoc, &F,
2575 Loaded, StoredSize, StoredModTime,
2576 StoredSignature, Capabilities);
2577
2578 // If we diagnosed a problem, produce a backtrace.
2579 if (isDiagnosedResult(Result, Capabilities))
2580 Diag(diag::note_module_file_imported_by)
2581 << F.FileName << !F.ModuleName.empty() << F.ModuleName;
2582
2583 switch (Result) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002584 case Failure: return Failure;
2585 // If we have to ignore the dependency, we'll have to ignore this too.
Douglas Gregor2f1806e2013-03-19 00:38:50 +00002586 case Missing:
Guy Benyei11169dd2012-12-18 14:30:41 +00002587 case OutOfDate: return OutOfDate;
2588 case VersionMismatch: return VersionMismatch;
2589 case ConfigurationMismatch: return ConfigurationMismatch;
2590 case HadErrors: return HadErrors;
2591 case Success: break;
2592 }
2593 }
2594 break;
2595 }
2596
Guy Benyei11169dd2012-12-18 14:30:41 +00002597 case ORIGINAL_FILE:
2598 F.OriginalSourceFileID = FileID::get(Record[0]);
Chris Lattner0e6c9402013-01-20 02:38:54 +00002599 F.ActualOriginalSourceFileName = Blob;
Guy Benyei11169dd2012-12-18 14:30:41 +00002600 F.OriginalSourceFileName = F.ActualOriginalSourceFileName;
Richard Smith7ed1bc92014-12-05 22:42:13 +00002601 ResolveImportedPath(F, F.OriginalSourceFileName);
Guy Benyei11169dd2012-12-18 14:30:41 +00002602 break;
2603
2604 case ORIGINAL_FILE_ID:
2605 F.OriginalSourceFileID = FileID::get(Record[0]);
2606 break;
2607
2608 case ORIGINAL_PCH_DIR:
Chris Lattner0e6c9402013-01-20 02:38:54 +00002609 F.OriginalDir = Blob;
Guy Benyei11169dd2012-12-18 14:30:41 +00002610 break;
2611
Ben Langmuirbeee15e2014-04-14 18:00:01 +00002612 case MODULE_NAME:
2613 F.ModuleName = Blob;
Ben Langmuir4f5212a2014-04-14 22:12:44 +00002614 if (Listener)
2615 Listener->ReadModuleName(F.ModuleName);
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00002616
2617 // Validate the AST as soon as we have a name so we can exit early on
2618 // failure.
2619 if (ASTReadResult Result = readUnhashedControlBlockOnce())
2620 return Result;
Vedant Kumar48b4f762018-04-14 01:40:48 +00002621
Ben Langmuirbeee15e2014-04-14 18:00:01 +00002622 break;
2623
Richard Smith223d3f22014-12-06 03:21:08 +00002624 case MODULE_DIRECTORY: {
2625 assert(!F.ModuleName.empty() &&
2626 "MODULE_DIRECTORY found before MODULE_NAME");
2627 // If we've already loaded a module map file covering this module, we may
2628 // have a better path for it (relative to the current build).
Bruno Cardoso Lopes52431f32018-07-18 23:21:19 +00002629 Module *M = PP.getHeaderSearchInfo().lookupModule(
2630 F.ModuleName, /*AllowSearch*/ true,
2631 /*AllowExtraModuleMapSearch*/ true);
Richard Smith223d3f22014-12-06 03:21:08 +00002632 if (M && M->Directory) {
2633 // If we're implicitly loading a module, the base directory can't
2634 // change between the build and use.
Yuka Takahashid8baec22018-08-01 09:50:02 +00002635 // Don't emit module relocation error if we have -fno-validate-pch
2636 if (!PP.getPreprocessorOpts().DisablePCHValidation &&
2637 F.Kind != MK_ExplicitModule && F.Kind != MK_PrebuiltModule) {
Richard Smith223d3f22014-12-06 03:21:08 +00002638 const DirectoryEntry *BuildDir =
2639 PP.getFileManager().getDirectory(Blob);
2640 if (!BuildDir || BuildDir != M->Directory) {
2641 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0)
2642 Diag(diag::err_imported_module_relocated)
2643 << F.ModuleName << Blob << M->Directory->getName();
2644 return OutOfDate;
2645 }
2646 }
2647 F.BaseDirectory = M->Directory->getName();
2648 } else {
2649 F.BaseDirectory = Blob;
2650 }
Richard Smith7ed1bc92014-12-05 22:42:13 +00002651 break;
Richard Smith223d3f22014-12-06 03:21:08 +00002652 }
Richard Smith7ed1bc92014-12-05 22:42:13 +00002653
Ben Langmuirbeee15e2014-04-14 18:00:01 +00002654 case MODULE_MAP_FILE:
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00002655 if (ASTReadResult Result =
2656 ReadModuleMapFileBlock(Record, F, ImportedBy, ClientLoadCapabilities))
2657 return Result;
Ben Langmuir264ea152014-11-08 00:06:39 +00002658 break;
2659
Justin Bognerca9c0cc2015-06-21 20:32:36 +00002660 case INPUT_FILE_OFFSETS:
Richard Smitha1825302014-10-23 22:18:29 +00002661 NumInputs = Record[0];
2662 NumUserInputs = Record[1];
Justin Bogner4c183242015-06-21 20:32:40 +00002663 F.InputFileOffsets =
2664 (const llvm::support::unaligned_uint64_t *)Blob.data();
Richard Smitha1825302014-10-23 22:18:29 +00002665 F.InputFilesLoaded.resize(NumInputs);
Argyrios Kyrtzidisa38cb202017-01-30 06:05:58 +00002666 F.NumUserInputFiles = NumUserInputs;
Guy Benyei11169dd2012-12-18 14:30:41 +00002667 break;
2668 }
2669 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002670}
2671
Ben Langmuir2c9af442014-04-10 17:57:43 +00002672ASTReader::ASTReadResult
2673ASTReader::ReadASTBlock(ModuleFile &F, unsigned ClientLoadCapabilities) {
Chris Lattner7fb3bef2013-01-20 00:56:42 +00002674 BitstreamCursor &Stream = F.Stream;
Guy Benyei11169dd2012-12-18 14:30:41 +00002675
2676 if (Stream.EnterSubBlock(AST_BLOCK_ID)) {
2677 Error("malformed block record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002678 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002679 }
2680
2681 // Read all of the records and blocks for the AST file.
2682 RecordData Record;
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00002683 while (true) {
Chris Lattnere7b154b2013-01-19 21:39:22 +00002684 llvm::BitstreamEntry Entry = Stream.advance();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002685
Chris Lattnere7b154b2013-01-19 21:39:22 +00002686 switch (Entry.Kind) {
2687 case llvm::BitstreamEntry::Error:
2688 Error("error at end of module block in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002689 return Failure;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00002690 case llvm::BitstreamEntry::EndBlock:
Richard Smithc0fbba72013-04-03 22:49:41 +00002691 // Outside of C++, we do not store a lookup map for the translation unit.
2692 // Instead, mark it as needing a lookup map to be built if this module
2693 // contains any declarations lexically within it (which it always does!).
2694 // This usually has no cost, since we very rarely need the lookup map for
2695 // the translation unit outside C++.
Richard Smithdbafb6c2017-06-29 23:23:46 +00002696 if (ASTContext *Ctx = ContextObj) {
2697 DeclContext *DC = Ctx->getTranslationUnitDecl();
2698 if (DC->hasExternalLexicalStorage() && !Ctx->getLangOpts().CPlusPlus)
2699 DC->setMustBuildLookupTable();
2700 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002701
Ben Langmuir2c9af442014-04-10 17:57:43 +00002702 return Success;
Chris Lattnere7b154b2013-01-19 21:39:22 +00002703 case llvm::BitstreamEntry::SubBlock:
2704 switch (Entry.ID) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002705 case DECLTYPES_BLOCK_ID:
2706 // We lazily load the decls block, but we want to set up the
2707 // DeclsCursor cursor to point into it. Clone our current bitcode
2708 // cursor to it, enter the block and read the abbrevs in that block.
2709 // With the main cursor, we just skip over it.
2710 F.DeclsCursor = Stream;
2711 if (Stream.SkipBlock() || // Skip with the main cursor.
2712 // Read the abbrevs.
2713 ReadBlockAbbrevs(F.DeclsCursor, DECLTYPES_BLOCK_ID)) {
2714 Error("malformed block record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002715 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002716 }
2717 break;
Richard Smithb9eab6d2014-03-20 19:44:17 +00002718
Guy Benyei11169dd2012-12-18 14:30:41 +00002719 case PREPROCESSOR_BLOCK_ID:
2720 F.MacroCursor = Stream;
2721 if (!PP.getExternalSource())
2722 PP.setExternalSource(this);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002723
Guy Benyei11169dd2012-12-18 14:30:41 +00002724 if (Stream.SkipBlock() ||
2725 ReadBlockAbbrevs(F.MacroCursor, PREPROCESSOR_BLOCK_ID)) {
2726 Error("malformed block record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002727 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002728 }
2729 F.MacroStartOffset = F.MacroCursor.GetCurrentBitNo();
2730 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002731
Guy Benyei11169dd2012-12-18 14:30:41 +00002732 case PREPROCESSOR_DETAIL_BLOCK_ID:
2733 F.PreprocessorDetailCursor = Stream;
2734 if (Stream.SkipBlock() ||
Chris Lattnere7b154b2013-01-19 21:39:22 +00002735 ReadBlockAbbrevs(F.PreprocessorDetailCursor,
Guy Benyei11169dd2012-12-18 14:30:41 +00002736 PREPROCESSOR_DETAIL_BLOCK_ID)) {
Chris Lattnere7b154b2013-01-19 21:39:22 +00002737 Error("malformed preprocessor detail record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002738 return Failure;
Chris Lattnere7b154b2013-01-19 21:39:22 +00002739 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002740 F.PreprocessorDetailStartOffset
Chris Lattnere7b154b2013-01-19 21:39:22 +00002741 = F.PreprocessorDetailCursor.GetCurrentBitNo();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002742
Guy Benyei11169dd2012-12-18 14:30:41 +00002743 if (!PP.getPreprocessingRecord())
2744 PP.createPreprocessingRecord();
2745 if (!PP.getPreprocessingRecord()->getExternalSource())
2746 PP.getPreprocessingRecord()->SetExternalSource(*this);
2747 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002748
Guy Benyei11169dd2012-12-18 14:30:41 +00002749 case SOURCE_MANAGER_BLOCK_ID:
2750 if (ReadSourceManagerBlock(F))
Ben Langmuir2c9af442014-04-10 17:57:43 +00002751 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002752 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002753
Guy Benyei11169dd2012-12-18 14:30:41 +00002754 case SUBMODULE_BLOCK_ID:
David Blaikie9ffe5a32017-01-30 05:00:26 +00002755 if (ASTReadResult Result =
2756 ReadSubmoduleBlock(F, ClientLoadCapabilities))
Ben Langmuir2c9af442014-04-10 17:57:43 +00002757 return Result;
Guy Benyei11169dd2012-12-18 14:30:41 +00002758 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002759
Guy Benyei11169dd2012-12-18 14:30:41 +00002760 case COMMENTS_BLOCK_ID: {
Chris Lattner7fb3bef2013-01-20 00:56:42 +00002761 BitstreamCursor C = Stream;
Guy Benyei11169dd2012-12-18 14:30:41 +00002762 if (Stream.SkipBlock() ||
2763 ReadBlockAbbrevs(C, COMMENTS_BLOCK_ID)) {
2764 Error("malformed comments block in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002765 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002766 }
2767 CommentsCursors.push_back(std::make_pair(C, &F));
2768 break;
2769 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002770
Guy Benyei11169dd2012-12-18 14:30:41 +00002771 default:
Chris Lattnere7b154b2013-01-19 21:39:22 +00002772 if (Stream.SkipBlock()) {
2773 Error("malformed block record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002774 return Failure;
Chris Lattnere7b154b2013-01-19 21:39:22 +00002775 }
2776 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00002777 }
2778 continue;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002779
Chris Lattnere7b154b2013-01-19 21:39:22 +00002780 case llvm::BitstreamEntry::Record:
2781 // The interesting case.
2782 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00002783 }
2784
2785 // Read and process a record.
2786 Record.clear();
Chris Lattner0e6c9402013-01-20 02:38:54 +00002787 StringRef Blob;
Richard Smithdbafb6c2017-06-29 23:23:46 +00002788 auto RecordType =
2789 (ASTRecordTypes)Stream.readRecord(Entry.ID, Record, &Blob);
2790
2791 // If we're not loading an AST context, we don't care about most records.
2792 if (!ContextObj) {
2793 switch (RecordType) {
2794 case IDENTIFIER_TABLE:
2795 case IDENTIFIER_OFFSET:
2796 case INTERESTING_IDENTIFIERS:
2797 case STATISTICS:
2798 case PP_CONDITIONAL_STACK:
2799 case PP_COUNTER_VALUE:
2800 case SOURCE_LOCATION_OFFSETS:
2801 case MODULE_OFFSET_MAP:
2802 case SOURCE_MANAGER_LINE_TABLE:
2803 case SOURCE_LOCATION_PRELOADS:
2804 case PPD_ENTITIES_OFFSETS:
2805 case HEADER_SEARCH_TABLE:
2806 case IMPORTED_MODULES:
2807 case MACRO_OFFSET:
2808 break;
2809 default:
2810 continue;
2811 }
2812 }
2813
2814 switch (RecordType) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002815 default: // Default behavior: ignore.
2816 break;
2817
2818 case TYPE_OFFSET: {
2819 if (F.LocalNumTypes != 0) {
2820 Error("duplicate TYPE_OFFSET record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002821 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002822 }
Chris Lattner0e6c9402013-01-20 02:38:54 +00002823 F.TypeOffsets = (const uint32_t *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00002824 F.LocalNumTypes = Record[0];
2825 unsigned LocalBaseTypeIndex = Record[1];
2826 F.BaseTypeIndex = getTotalNumTypes();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002827
Guy Benyei11169dd2012-12-18 14:30:41 +00002828 if (F.LocalNumTypes > 0) {
2829 // Introduce the global -> local mapping for types within this module.
2830 GlobalTypeMap.insert(std::make_pair(getTotalNumTypes(), &F));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002831
Guy Benyei11169dd2012-12-18 14:30:41 +00002832 // Introduce the local -> global mapping for types within this module.
2833 F.TypeRemap.insertOrReplace(
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002834 std::make_pair(LocalBaseTypeIndex,
Guy Benyei11169dd2012-12-18 14:30:41 +00002835 F.BaseTypeIndex - LocalBaseTypeIndex));
Ben Langmuir52ca6782014-10-20 16:27:32 +00002836
2837 TypesLoaded.resize(TypesLoaded.size() + F.LocalNumTypes);
Guy Benyei11169dd2012-12-18 14:30:41 +00002838 }
2839 break;
2840 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002841
Guy Benyei11169dd2012-12-18 14:30:41 +00002842 case DECL_OFFSET: {
2843 if (F.LocalNumDecls != 0) {
2844 Error("duplicate DECL_OFFSET record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002845 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002846 }
Chris Lattner0e6c9402013-01-20 02:38:54 +00002847 F.DeclOffsets = (const DeclOffset *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00002848 F.LocalNumDecls = Record[0];
2849 unsigned LocalBaseDeclID = Record[1];
2850 F.BaseDeclID = getTotalNumDecls();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002851
Guy Benyei11169dd2012-12-18 14:30:41 +00002852 if (F.LocalNumDecls > 0) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002853 // Introduce the global -> local mapping for declarations within this
Guy Benyei11169dd2012-12-18 14:30:41 +00002854 // module.
2855 GlobalDeclMap.insert(
2856 std::make_pair(getTotalNumDecls() + NUM_PREDEF_DECL_IDS, &F));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002857
Guy Benyei11169dd2012-12-18 14:30:41 +00002858 // Introduce the local -> global mapping for declarations within this
2859 // module.
2860 F.DeclRemap.insertOrReplace(
2861 std::make_pair(LocalBaseDeclID, F.BaseDeclID - LocalBaseDeclID));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002862
Guy Benyei11169dd2012-12-18 14:30:41 +00002863 // Introduce the global -> local mapping for declarations within this
2864 // module.
2865 F.GlobalToLocalDeclIDs[&F] = LocalBaseDeclID;
Ben Langmuirfe971d92014-08-16 04:54:18 +00002866
Ben Langmuir52ca6782014-10-20 16:27:32 +00002867 DeclsLoaded.resize(DeclsLoaded.size() + F.LocalNumDecls);
2868 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002869 break;
2870 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002871
Guy Benyei11169dd2012-12-18 14:30:41 +00002872 case TU_UPDATE_LEXICAL: {
Richard Smithdbafb6c2017-06-29 23:23:46 +00002873 DeclContext *TU = ContextObj->getTranslationUnitDecl();
Richard Smith82f8fcd2015-08-06 22:07:25 +00002874 LexicalContents Contents(
2875 reinterpret_cast<const llvm::support::unaligned_uint32_t *>(
2876 Blob.data()),
2877 static_cast<unsigned int>(Blob.size() / 4));
2878 TULexicalDecls.push_back(std::make_pair(&F, Contents));
Guy Benyei11169dd2012-12-18 14:30:41 +00002879 TU->setHasExternalLexicalStorage(true);
2880 break;
2881 }
2882
2883 case UPDATE_VISIBLE: {
2884 unsigned Idx = 0;
2885 serialization::DeclID ID = ReadDeclID(F, Record, Idx);
Richard Smith0f4e2c42015-08-06 04:23:48 +00002886 auto *Data = (const unsigned char*)Blob.data();
Richard Smithd88a7f12015-09-01 20:35:42 +00002887 PendingVisibleUpdates[ID].push_back(PendingVisibleUpdate{&F, Data});
Richard Smith0f4e2c42015-08-06 04:23:48 +00002888 // If we've already loaded the decl, perform the updates when we finish
2889 // loading this block.
2890 if (Decl *D = GetExistingDecl(ID))
Vassil Vassilev74c3e8c2017-05-19 16:46:06 +00002891 PendingUpdateRecords.push_back(
2892 PendingUpdateRecord(ID, D, /*JustLoaded=*/false));
Guy Benyei11169dd2012-12-18 14:30:41 +00002893 break;
2894 }
2895
2896 case IDENTIFIER_TABLE:
Chris Lattner0e6c9402013-01-20 02:38:54 +00002897 F.IdentifierTableData = Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00002898 if (Record[0]) {
Justin Bognerda4e6502014-04-14 16:34:29 +00002899 F.IdentifierLookupTable = ASTIdentifierLookupTable::Create(
2900 (const unsigned char *)F.IdentifierTableData + Record[0],
2901 (const unsigned char *)F.IdentifierTableData + sizeof(uint32_t),
2902 (const unsigned char *)F.IdentifierTableData,
2903 ASTIdentifierLookupTrait(*this, F));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002904
Guy Benyei11169dd2012-12-18 14:30:41 +00002905 PP.getIdentifierTable().setExternalIdentifierLookup(this);
2906 }
2907 break;
2908
2909 case IDENTIFIER_OFFSET: {
2910 if (F.LocalNumIdentifiers != 0) {
2911 Error("duplicate IDENTIFIER_OFFSET record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002912 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002913 }
Chris Lattner0e6c9402013-01-20 02:38:54 +00002914 F.IdentifierOffsets = (const uint32_t *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00002915 F.LocalNumIdentifiers = Record[0];
2916 unsigned LocalBaseIdentifierID = Record[1];
2917 F.BaseIdentifierID = getTotalNumIdentifiers();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002918
Guy Benyei11169dd2012-12-18 14:30:41 +00002919 if (F.LocalNumIdentifiers > 0) {
2920 // Introduce the global -> local mapping for identifiers within this
2921 // module.
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002922 GlobalIdentifierMap.insert(std::make_pair(getTotalNumIdentifiers() + 1,
Guy Benyei11169dd2012-12-18 14:30:41 +00002923 &F));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002924
Guy Benyei11169dd2012-12-18 14:30:41 +00002925 // Introduce the local -> global mapping for identifiers within this
2926 // module.
2927 F.IdentifierRemap.insertOrReplace(
2928 std::make_pair(LocalBaseIdentifierID,
2929 F.BaseIdentifierID - LocalBaseIdentifierID));
Ben Langmuirfe971d92014-08-16 04:54:18 +00002930
Ben Langmuir52ca6782014-10-20 16:27:32 +00002931 IdentifiersLoaded.resize(IdentifiersLoaded.size()
2932 + F.LocalNumIdentifiers);
2933 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002934 break;
2935 }
2936
Richard Smith33e0f7e2015-07-22 02:08:40 +00002937 case INTERESTING_IDENTIFIERS:
2938 F.PreloadIdentifierOffsets.assign(Record.begin(), Record.end());
2939 break;
2940
Ben Langmuir332aafe2014-01-31 01:06:56 +00002941 case EAGERLY_DESERIALIZED_DECLS:
Richard Smith9e2341d2015-03-23 03:25:59 +00002942 // FIXME: Skip reading this record if our ASTConsumer doesn't care
2943 // about "interesting" decls (for instance, if we're building a module).
Guy Benyei11169dd2012-12-18 14:30:41 +00002944 for (unsigned I = 0, N = Record.size(); I != N; ++I)
Ben Langmuir332aafe2014-01-31 01:06:56 +00002945 EagerlyDeserializedDecls.push_back(getGlobalDeclID(F, Record[I]));
Guy Benyei11169dd2012-12-18 14:30:41 +00002946 break;
2947
David Blaikie9ffe5a32017-01-30 05:00:26 +00002948 case MODULAR_CODEGEN_DECLS:
2949 // FIXME: Skip reading this record if our ASTConsumer doesn't care about
2950 // them (ie: if we're not codegenerating this module).
2951 if (F.Kind == MK_MainFile)
2952 for (unsigned I = 0, N = Record.size(); I != N; ++I)
2953 EagerlyDeserializedDecls.push_back(getGlobalDeclID(F, Record[I]));
2954 break;
2955
Guy Benyei11169dd2012-12-18 14:30:41 +00002956 case SPECIAL_TYPES:
Douglas Gregor44180f82013-02-01 23:45:03 +00002957 if (SpecialTypes.empty()) {
2958 for (unsigned I = 0, N = Record.size(); I != N; ++I)
2959 SpecialTypes.push_back(getGlobalTypeID(F, Record[I]));
2960 break;
2961 }
2962
2963 if (SpecialTypes.size() != Record.size()) {
2964 Error("invalid special-types record");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002965 return Failure;
Douglas Gregor44180f82013-02-01 23:45:03 +00002966 }
2967
2968 for (unsigned I = 0, N = Record.size(); I != N; ++I) {
2969 serialization::TypeID ID = getGlobalTypeID(F, Record[I]);
2970 if (!SpecialTypes[I])
2971 SpecialTypes[I] = ID;
2972 // FIXME: If ID && SpecialTypes[I] != ID, do we need a separate
2973 // merge step?
2974 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002975 break;
2976
2977 case STATISTICS:
2978 TotalNumStatements += Record[0];
2979 TotalNumMacros += Record[1];
2980 TotalLexicalDeclContexts += Record[2];
2981 TotalVisibleDeclContexts += Record[3];
2982 break;
2983
2984 case UNUSED_FILESCOPED_DECLS:
2985 for (unsigned I = 0, N = Record.size(); I != N; ++I)
2986 UnusedFileScopedDecls.push_back(getGlobalDeclID(F, Record[I]));
2987 break;
2988
2989 case DELEGATING_CTORS:
2990 for (unsigned I = 0, N = Record.size(); I != N; ++I)
2991 DelegatingCtorDecls.push_back(getGlobalDeclID(F, Record[I]));
2992 break;
2993
2994 case WEAK_UNDECLARED_IDENTIFIERS:
2995 if (Record.size() % 4 != 0) {
2996 Error("invalid weak identifiers record");
Ben Langmuir2c9af442014-04-10 17:57:43 +00002997 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00002998 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00002999
3000 // FIXME: Ignore weak undeclared identifiers from non-original PCH
Guy Benyei11169dd2012-12-18 14:30:41 +00003001 // files. This isn't the way to do it :)
3002 WeakUndeclaredIdentifiers.clear();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003003
Guy Benyei11169dd2012-12-18 14:30:41 +00003004 // Translate the weak, undeclared identifiers into global IDs.
3005 for (unsigned I = 0, N = Record.size(); I < N; /* in loop */) {
3006 WeakUndeclaredIdentifiers.push_back(
3007 getGlobalIdentifierID(F, Record[I++]));
3008 WeakUndeclaredIdentifiers.push_back(
3009 getGlobalIdentifierID(F, Record[I++]));
3010 WeakUndeclaredIdentifiers.push_back(
3011 ReadSourceLocation(F, Record, I).getRawEncoding());
3012 WeakUndeclaredIdentifiers.push_back(Record[I++]);
3013 }
3014 break;
3015
Guy Benyei11169dd2012-12-18 14:30:41 +00003016 case SELECTOR_OFFSETS: {
Chris Lattner0e6c9402013-01-20 02:38:54 +00003017 F.SelectorOffsets = (const uint32_t *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00003018 F.LocalNumSelectors = Record[0];
3019 unsigned LocalBaseSelectorID = Record[1];
3020 F.BaseSelectorID = getTotalNumSelectors();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003021
Guy Benyei11169dd2012-12-18 14:30:41 +00003022 if (F.LocalNumSelectors > 0) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003023 // Introduce the global -> local mapping for selectors within this
Guy Benyei11169dd2012-12-18 14:30:41 +00003024 // module.
3025 GlobalSelectorMap.insert(std::make_pair(getTotalNumSelectors()+1, &F));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003026
3027 // Introduce the local -> global mapping for selectors within this
Guy Benyei11169dd2012-12-18 14:30:41 +00003028 // module.
3029 F.SelectorRemap.insertOrReplace(
3030 std::make_pair(LocalBaseSelectorID,
3031 F.BaseSelectorID - LocalBaseSelectorID));
Ben Langmuir52ca6782014-10-20 16:27:32 +00003032
3033 SelectorsLoaded.resize(SelectorsLoaded.size() + F.LocalNumSelectors);
Guy Benyei11169dd2012-12-18 14:30:41 +00003034 }
3035 break;
3036 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003037
Guy Benyei11169dd2012-12-18 14:30:41 +00003038 case METHOD_POOL:
Chris Lattner0e6c9402013-01-20 02:38:54 +00003039 F.SelectorLookupTableData = (const unsigned char *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00003040 if (Record[0])
3041 F.SelectorLookupTable
3042 = ASTSelectorLookupTable::Create(
3043 F.SelectorLookupTableData + Record[0],
3044 F.SelectorLookupTableData,
3045 ASTSelectorLookupTrait(*this, F));
3046 TotalNumMethodPoolEntries += Record[1];
3047 break;
3048
3049 case REFERENCED_SELECTOR_POOL:
3050 if (!Record.empty()) {
3051 for (unsigned Idx = 0, N = Record.size() - 1; Idx < N; /* in loop */) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003052 ReferencedSelectorsData.push_back(getGlobalSelectorID(F,
Guy Benyei11169dd2012-12-18 14:30:41 +00003053 Record[Idx++]));
3054 ReferencedSelectorsData.push_back(ReadSourceLocation(F, Record, Idx).
3055 getRawEncoding());
3056 }
3057 }
3058 break;
3059
Erik Verbruggenb34c79f2017-05-30 11:54:55 +00003060 case PP_CONDITIONAL_STACK:
3061 if (!Record.empty()) {
Erik Verbruggen4d1eb2d2017-11-03 09:40:07 +00003062 unsigned Idx = 0, End = Record.size() - 1;
3063 bool ReachedEOFWhileSkipping = Record[Idx++];
3064 llvm::Optional<Preprocessor::PreambleSkipInfo> SkipInfo;
3065 if (ReachedEOFWhileSkipping) {
3066 SourceLocation HashToken = ReadSourceLocation(F, Record, Idx);
3067 SourceLocation IfTokenLoc = ReadSourceLocation(F, Record, Idx);
3068 bool FoundNonSkipPortion = Record[Idx++];
3069 bool FoundElse = Record[Idx++];
3070 SourceLocation ElseLoc = ReadSourceLocation(F, Record, Idx);
3071 SkipInfo.emplace(HashToken, IfTokenLoc, FoundNonSkipPortion,
3072 FoundElse, ElseLoc);
3073 }
Erik Verbruggenb34c79f2017-05-30 11:54:55 +00003074 SmallVector<PPConditionalInfo, 4> ConditionalStack;
Erik Verbruggen4d1eb2d2017-11-03 09:40:07 +00003075 while (Idx < End) {
Erik Verbruggenb34c79f2017-05-30 11:54:55 +00003076 auto Loc = ReadSourceLocation(F, Record, Idx);
3077 bool WasSkipping = Record[Idx++];
3078 bool FoundNonSkip = Record[Idx++];
3079 bool FoundElse = Record[Idx++];
3080 ConditionalStack.push_back(
3081 {Loc, WasSkipping, FoundNonSkip, FoundElse});
3082 }
Erik Verbruggen4d1eb2d2017-11-03 09:40:07 +00003083 PP.setReplayablePreambleConditionalStack(ConditionalStack, SkipInfo);
Erik Verbruggenb34c79f2017-05-30 11:54:55 +00003084 }
3085 break;
3086
Guy Benyei11169dd2012-12-18 14:30:41 +00003087 case PP_COUNTER_VALUE:
3088 if (!Record.empty() && Listener)
3089 Listener->ReadCounter(F, Record[0]);
3090 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003091
Guy Benyei11169dd2012-12-18 14:30:41 +00003092 case FILE_SORTED_DECLS:
Chris Lattner0e6c9402013-01-20 02:38:54 +00003093 F.FileSortedDecls = (const DeclID *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00003094 F.NumFileSortedDecls = Record[0];
3095 break;
3096
3097 case SOURCE_LOCATION_OFFSETS: {
Chris Lattner0e6c9402013-01-20 02:38:54 +00003098 F.SLocEntryOffsets = (const uint32_t *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00003099 F.LocalNumSLocEntries = Record[0];
3100 unsigned SLocSpaceSize = Record[1];
Benjamin Kramer867ea1d2014-03-02 13:01:17 +00003101 std::tie(F.SLocEntryBaseID, F.SLocEntryBaseOffset) =
Ben Langmuir52ca6782014-10-20 16:27:32 +00003102 SourceMgr.AllocateLoadedSLocEntries(F.LocalNumSLocEntries,
Guy Benyei11169dd2012-12-18 14:30:41 +00003103 SLocSpaceSize);
Richard Smith78d81ec2015-08-12 22:25:24 +00003104 if (!F.SLocEntryBaseID) {
3105 Error("ran out of source locations");
3106 break;
3107 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003108 // Make our entry in the range map. BaseID is negative and growing, so
3109 // we invert it. Because we invert it, though, we need the other end of
3110 // the range.
3111 unsigned RangeStart =
3112 unsigned(-F.SLocEntryBaseID) - F.LocalNumSLocEntries + 1;
3113 GlobalSLocEntryMap.insert(std::make_pair(RangeStart, &F));
3114 F.FirstLoc = SourceLocation::getFromRawEncoding(F.SLocEntryBaseOffset);
3115
3116 // SLocEntryBaseOffset is lower than MaxLoadedOffset and decreasing.
3117 assert((F.SLocEntryBaseOffset & (1U << 31U)) == 0);
3118 GlobalSLocOffsetMap.insert(
3119 std::make_pair(SourceManager::MaxLoadedOffset - F.SLocEntryBaseOffset
3120 - SLocSpaceSize,&F));
3121
3122 // Initialize the remapping table.
3123 // Invalid stays invalid.
Richard Smithb9eab6d2014-03-20 19:44:17 +00003124 F.SLocRemap.insertOrReplace(std::make_pair(0U, 0));
Guy Benyei11169dd2012-12-18 14:30:41 +00003125 // This module. Base was 2 when being compiled.
Richard Smithb9eab6d2014-03-20 19:44:17 +00003126 F.SLocRemap.insertOrReplace(std::make_pair(2U,
Guy Benyei11169dd2012-12-18 14:30:41 +00003127 static_cast<int>(F.SLocEntryBaseOffset - 2)));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003128
Guy Benyei11169dd2012-12-18 14:30:41 +00003129 TotalNumSLocEntries += F.LocalNumSLocEntries;
3130 break;
3131 }
3132
Richard Smith37a93df2017-02-18 00:32:02 +00003133 case MODULE_OFFSET_MAP:
3134 F.ModuleOffsetMap = Blob;
Guy Benyei11169dd2012-12-18 14:30:41 +00003135 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00003136
3137 case SOURCE_MANAGER_LINE_TABLE:
3138 if (ParseLineTable(F, Record))
Ben Langmuir2c9af442014-04-10 17:57:43 +00003139 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003140 break;
3141
3142 case SOURCE_LOCATION_PRELOADS: {
3143 // Need to transform from the local view (1-based IDs) to the global view,
3144 // which is based off F.SLocEntryBaseID.
3145 if (!F.PreloadSLocEntries.empty()) {
3146 Error("Multiple SOURCE_LOCATION_PRELOADS records in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003147 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003148 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003149
Guy Benyei11169dd2012-12-18 14:30:41 +00003150 F.PreloadSLocEntries.swap(Record);
3151 break;
3152 }
3153
3154 case EXT_VECTOR_DECLS:
3155 for (unsigned I = 0, N = Record.size(); I != N; ++I)
3156 ExtVectorDecls.push_back(getGlobalDeclID(F, Record[I]));
3157 break;
3158
3159 case VTABLE_USES:
3160 if (Record.size() % 3 != 0) {
3161 Error("Invalid VTABLE_USES record");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003162 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003163 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003164
Guy Benyei11169dd2012-12-18 14:30:41 +00003165 // Later tables overwrite earlier ones.
3166 // FIXME: Modules will have some trouble with this. This is clearly not
3167 // the right way to do this.
3168 VTableUses.clear();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003169
Guy Benyei11169dd2012-12-18 14:30:41 +00003170 for (unsigned Idx = 0, N = Record.size(); Idx != N; /* In loop */) {
3171 VTableUses.push_back(getGlobalDeclID(F, Record[Idx++]));
3172 VTableUses.push_back(
3173 ReadSourceLocation(F, Record, Idx).getRawEncoding());
3174 VTableUses.push_back(Record[Idx++]);
3175 }
3176 break;
3177
Guy Benyei11169dd2012-12-18 14:30:41 +00003178 case PENDING_IMPLICIT_INSTANTIATIONS:
3179 if (PendingInstantiations.size() % 2 != 0) {
3180 Error("Invalid existing PendingInstantiations");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003181 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003182 }
3183
3184 if (Record.size() % 2 != 0) {
3185 Error("Invalid PENDING_IMPLICIT_INSTANTIATIONS block");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003186 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003187 }
3188
3189 for (unsigned I = 0, N = Record.size(); I != N; /* in loop */) {
3190 PendingInstantiations.push_back(getGlobalDeclID(F, Record[I++]));
3191 PendingInstantiations.push_back(
3192 ReadSourceLocation(F, Record, I).getRawEncoding());
3193 }
3194 break;
3195
3196 case SEMA_DECL_REFS:
Richard Smith96269c52016-09-29 22:49:46 +00003197 if (Record.size() != 3) {
Richard Smith3d8e97e2013-10-18 06:54:39 +00003198 Error("Invalid SEMA_DECL_REFS block");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003199 return Failure;
Richard Smith3d8e97e2013-10-18 06:54:39 +00003200 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003201 for (unsigned I = 0, N = Record.size(); I != N; ++I)
3202 SemaDeclRefs.push_back(getGlobalDeclID(F, Record[I]));
3203 break;
3204
3205 case PPD_ENTITIES_OFFSETS: {
Chris Lattner0e6c9402013-01-20 02:38:54 +00003206 F.PreprocessedEntityOffsets = (const PPEntityOffset *)Blob.data();
3207 assert(Blob.size() % sizeof(PPEntityOffset) == 0);
3208 F.NumPreprocessedEntities = Blob.size() / sizeof(PPEntityOffset);
Guy Benyei11169dd2012-12-18 14:30:41 +00003209
3210 unsigned LocalBasePreprocessedEntityID = Record[0];
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003211
Guy Benyei11169dd2012-12-18 14:30:41 +00003212 unsigned StartingID;
3213 if (!PP.getPreprocessingRecord())
3214 PP.createPreprocessingRecord();
3215 if (!PP.getPreprocessingRecord()->getExternalSource())
3216 PP.getPreprocessingRecord()->SetExternalSource(*this);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003217 StartingID
Guy Benyei11169dd2012-12-18 14:30:41 +00003218 = PP.getPreprocessingRecord()
Ben Langmuir52ca6782014-10-20 16:27:32 +00003219 ->allocateLoadedEntities(F.NumPreprocessedEntities);
Guy Benyei11169dd2012-12-18 14:30:41 +00003220 F.BasePreprocessedEntityID = StartingID;
3221
3222 if (F.NumPreprocessedEntities > 0) {
3223 // Introduce the global -> local mapping for preprocessed entities in
3224 // this module.
3225 GlobalPreprocessedEntityMap.insert(std::make_pair(StartingID, &F));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003226
Guy Benyei11169dd2012-12-18 14:30:41 +00003227 // Introduce the local -> global mapping for preprocessed entities in
3228 // this module.
3229 F.PreprocessedEntityRemap.insertOrReplace(
3230 std::make_pair(LocalBasePreprocessedEntityID,
3231 F.BasePreprocessedEntityID - LocalBasePreprocessedEntityID));
3232 }
3233
3234 break;
3235 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003236
Cameron Desrochersb60f1b62018-01-15 19:14:16 +00003237 case PPD_SKIPPED_RANGES: {
3238 F.PreprocessedSkippedRangeOffsets = (const PPSkippedRange*)Blob.data();
3239 assert(Blob.size() % sizeof(PPSkippedRange) == 0);
3240 F.NumPreprocessedSkippedRanges = Blob.size() / sizeof(PPSkippedRange);
3241
3242 if (!PP.getPreprocessingRecord())
3243 PP.createPreprocessingRecord();
3244 if (!PP.getPreprocessingRecord()->getExternalSource())
3245 PP.getPreprocessingRecord()->SetExternalSource(*this);
3246 F.BasePreprocessedSkippedRangeID = PP.getPreprocessingRecord()
3247 ->allocateSkippedRanges(F.NumPreprocessedSkippedRanges);
Jonas Devlieghere560ce2c2018-02-26 15:16:42 +00003248
Cameron Desrochersb60f1b62018-01-15 19:14:16 +00003249 if (F.NumPreprocessedSkippedRanges > 0)
3250 GlobalSkippedRangeMap.insert(
3251 std::make_pair(F.BasePreprocessedSkippedRangeID, &F));
3252 break;
3253 }
3254
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00003255 case DECL_UPDATE_OFFSETS:
Guy Benyei11169dd2012-12-18 14:30:41 +00003256 if (Record.size() % 2 != 0) {
3257 Error("invalid DECL_UPDATE_OFFSETS block in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003258 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003259 }
Richard Smithcd45dbc2014-04-19 03:48:30 +00003260 for (unsigned I = 0, N = Record.size(); I != N; I += 2) {
3261 GlobalDeclID ID = getGlobalDeclID(F, Record[I]);
3262 DeclUpdateOffsets[ID].push_back(std::make_pair(&F, Record[I + 1]));
3263
3264 // If we've already loaded the decl, perform the updates when we finish
3265 // loading this block.
3266 if (Decl *D = GetExistingDecl(ID))
Vassil Vassilev74c3e8c2017-05-19 16:46:06 +00003267 PendingUpdateRecords.push_back(
3268 PendingUpdateRecord(ID, D, /*JustLoaded=*/false));
Richard Smithcd45dbc2014-04-19 03:48:30 +00003269 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003270 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00003271
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00003272 case OBJC_CATEGORIES_MAP:
Guy Benyei11169dd2012-12-18 14:30:41 +00003273 if (F.LocalNumObjCCategoriesInMap != 0) {
3274 Error("duplicate OBJC_CATEGORIES_MAP record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003275 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003276 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003277
Guy Benyei11169dd2012-12-18 14:30:41 +00003278 F.LocalNumObjCCategoriesInMap = Record[0];
Chris Lattner0e6c9402013-01-20 02:38:54 +00003279 F.ObjCCategoriesMap = (const ObjCCategoriesInfo *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00003280 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003281
Guy Benyei11169dd2012-12-18 14:30:41 +00003282 case OBJC_CATEGORIES:
3283 F.ObjCCategories.swap(Record);
3284 break;
Richard Smithc2bb8182015-03-24 06:36:48 +00003285
Guy Benyei11169dd2012-12-18 14:30:41 +00003286 case CUDA_SPECIAL_DECL_REFS:
3287 // Later tables overwrite earlier ones.
3288 // FIXME: Modules will have trouble with this.
3289 CUDASpecialDeclRefs.clear();
3290 for (unsigned I = 0, N = Record.size(); I != N; ++I)
3291 CUDASpecialDeclRefs.push_back(getGlobalDeclID(F, Record[I]));
3292 break;
3293
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00003294 case HEADER_SEARCH_TABLE:
Chris Lattner0e6c9402013-01-20 02:38:54 +00003295 F.HeaderFileInfoTableData = Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00003296 F.LocalNumHeaderFileInfos = Record[1];
Guy Benyei11169dd2012-12-18 14:30:41 +00003297 if (Record[0]) {
3298 F.HeaderFileInfoTable
3299 = HeaderFileInfoLookupTable::Create(
3300 (const unsigned char *)F.HeaderFileInfoTableData + Record[0],
3301 (const unsigned char *)F.HeaderFileInfoTableData,
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003302 HeaderFileInfoTrait(*this, F,
Guy Benyei11169dd2012-12-18 14:30:41 +00003303 &PP.getHeaderSearchInfo(),
Chris Lattner0e6c9402013-01-20 02:38:54 +00003304 Blob.data() + Record[2]));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003305
Guy Benyei11169dd2012-12-18 14:30:41 +00003306 PP.getHeaderSearchInfo().SetExternalSource(this);
3307 if (!PP.getHeaderSearchInfo().getExternalLookup())
3308 PP.getHeaderSearchInfo().SetExternalLookup(this);
3309 }
3310 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003311
Guy Benyei11169dd2012-12-18 14:30:41 +00003312 case FP_PRAGMA_OPTIONS:
3313 // Later tables overwrite earlier ones.
3314 FPPragmaOptions.swap(Record);
3315 break;
3316
3317 case OPENCL_EXTENSIONS:
Yaxun Liu5b746652016-12-18 05:18:55 +00003318 for (unsigned I = 0, E = Record.size(); I != E; ) {
3319 auto Name = ReadString(Record, I);
3320 auto &Opt = OpenCLExtensions.OptMap[Name];
Yaxun Liucc2741c2016-12-18 06:35:06 +00003321 Opt.Supported = Record[I++] != 0;
3322 Opt.Enabled = Record[I++] != 0;
Yaxun Liu5b746652016-12-18 05:18:55 +00003323 Opt.Avail = Record[I++];
3324 Opt.Core = Record[I++];
3325 }
3326 break;
3327
3328 case OPENCL_EXTENSION_TYPES:
3329 for (unsigned I = 0, E = Record.size(); I != E;) {
3330 auto TypeID = static_cast<::TypeID>(Record[I++]);
3331 auto *Type = GetType(TypeID).getTypePtr();
3332 auto NumExt = static_cast<unsigned>(Record[I++]);
3333 for (unsigned II = 0; II != NumExt; ++II) {
3334 auto Ext = ReadString(Record, I);
3335 OpenCLTypeExtMap[Type].insert(Ext);
3336 }
3337 }
3338 break;
3339
3340 case OPENCL_EXTENSION_DECLS:
3341 for (unsigned I = 0, E = Record.size(); I != E;) {
3342 auto DeclID = static_cast<::DeclID>(Record[I++]);
3343 auto *Decl = GetDecl(DeclID);
3344 auto NumExt = static_cast<unsigned>(Record[I++]);
3345 for (unsigned II = 0; II != NumExt; ++II) {
3346 auto Ext = ReadString(Record, I);
3347 OpenCLDeclExtMap[Decl].insert(Ext);
3348 }
3349 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003350 break;
3351
3352 case TENTATIVE_DEFINITIONS:
3353 for (unsigned I = 0, N = Record.size(); I != N; ++I)
3354 TentativeDefinitions.push_back(getGlobalDeclID(F, Record[I]));
3355 break;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003356
Guy Benyei11169dd2012-12-18 14:30:41 +00003357 case KNOWN_NAMESPACES:
3358 for (unsigned I = 0, N = Record.size(); I != N; ++I)
3359 KnownNamespaces.push_back(getGlobalDeclID(F, Record[I]));
3360 break;
Nick Lewycky8334af82013-01-26 00:35:08 +00003361
Nick Lewycky9c7eb1d2013-02-01 08:13:20 +00003362 case UNDEFINED_BUT_USED:
3363 if (UndefinedButUsed.size() % 2 != 0) {
3364 Error("Invalid existing UndefinedButUsed");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003365 return Failure;
Nick Lewycky8334af82013-01-26 00:35:08 +00003366 }
3367
3368 if (Record.size() % 2 != 0) {
Nick Lewycky9c7eb1d2013-02-01 08:13:20 +00003369 Error("invalid undefined-but-used record");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003370 return Failure;
Nick Lewycky8334af82013-01-26 00:35:08 +00003371 }
3372 for (unsigned I = 0, N = Record.size(); I != N; /* in loop */) {
Nick Lewycky9c7eb1d2013-02-01 08:13:20 +00003373 UndefinedButUsed.push_back(getGlobalDeclID(F, Record[I++]));
3374 UndefinedButUsed.push_back(
Nick Lewycky8334af82013-01-26 00:35:08 +00003375 ReadSourceLocation(F, Record, I).getRawEncoding());
3376 }
3377 break;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00003378
Ismail Pazarbasie5768d12015-05-18 19:59:11 +00003379 case DELETE_EXPRS_TO_ANALYZE:
3380 for (unsigned I = 0, N = Record.size(); I != N;) {
3381 DelayedDeleteExprs.push_back(getGlobalDeclID(F, Record[I++]));
3382 const uint64_t Count = Record[I++];
3383 DelayedDeleteExprs.push_back(Count);
3384 for (uint64_t C = 0; C < Count; ++C) {
3385 DelayedDeleteExprs.push_back(ReadSourceLocation(F, Record, I).getRawEncoding());
3386 bool IsArrayForm = Record[I++] == 1;
3387 DelayedDeleteExprs.push_back(IsArrayForm);
3388 }
3389 }
3390 break;
Nick Lewycky8334af82013-01-26 00:35:08 +00003391
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00003392 case IMPORTED_MODULES:
Manman Ren11f2a472016-08-18 17:42:15 +00003393 if (!F.isModule()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003394 // If we aren't loading a module (which has its own exports), make
3395 // all of the imported modules visible.
3396 // FIXME: Deal with macros-only imports.
Richard Smith56be7542014-03-21 00:33:59 +00003397 for (unsigned I = 0, N = Record.size(); I != N; /**/) {
3398 unsigned GlobalID = getGlobalSubmoduleID(F, Record[I++]);
3399 SourceLocation Loc = ReadSourceLocation(F, Record, I);
Graydon Hoare9c982442017-01-18 20:36:59 +00003400 if (GlobalID) {
Aaron Ballman4f45b712014-03-21 15:22:56 +00003401 ImportedModules.push_back(ImportedSubmodule(GlobalID, Loc));
Graydon Hoare9c982442017-01-18 20:36:59 +00003402 if (DeserializationListener)
3403 DeserializationListener->ModuleImportRead(GlobalID, Loc);
3404 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003405 }
3406 }
3407 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00003408
Guy Benyei11169dd2012-12-18 14:30:41 +00003409 case MACRO_OFFSET: {
3410 if (F.LocalNumMacros != 0) {
3411 Error("duplicate MACRO_OFFSET record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00003412 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003413 }
Chris Lattner0e6c9402013-01-20 02:38:54 +00003414 F.MacroOffsets = (const uint32_t *)Blob.data();
Guy Benyei11169dd2012-12-18 14:30:41 +00003415 F.LocalNumMacros = Record[0];
3416 unsigned LocalBaseMacroID = Record[1];
3417 F.BaseMacroID = getTotalNumMacros();
3418
3419 if (F.LocalNumMacros > 0) {
3420 // Introduce the global -> local mapping for macros within this module.
3421 GlobalMacroMap.insert(std::make_pair(getTotalNumMacros() + 1, &F));
3422
3423 // Introduce the local -> global mapping for macros within this module.
3424 F.MacroRemap.insertOrReplace(
3425 std::make_pair(LocalBaseMacroID,
3426 F.BaseMacroID - LocalBaseMacroID));
Ben Langmuir52ca6782014-10-20 16:27:32 +00003427
3428 MacrosLoaded.resize(MacrosLoaded.size() + F.LocalNumMacros);
Guy Benyei11169dd2012-12-18 14:30:41 +00003429 }
3430 break;
3431 }
3432
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00003433 case LATE_PARSED_TEMPLATE:
Richard Smithe40f2ba2013-08-07 21:41:30 +00003434 LateParsedTemplates.append(Record.begin(), Record.end());
3435 break;
Dario Domizioli13a0a382014-05-23 12:13:25 +00003436
3437 case OPTIMIZE_PRAGMA_OPTIONS:
3438 if (Record.size() != 1) {
3439 Error("invalid pragma optimize record");
3440 return Failure;
3441 }
3442 OptimizeOffPragmaLocation = ReadSourceLocation(F, Record[0]);
3443 break;
Nico Weber72889432014-09-06 01:25:55 +00003444
Nico Weber779355f2016-03-02 23:22:00 +00003445 case MSSTRUCT_PRAGMA_OPTIONS:
3446 if (Record.size() != 1) {
3447 Error("invalid pragma ms_struct record");
3448 return Failure;
3449 }
3450 PragmaMSStructState = Record[0];
3451 break;
3452
Nico Weber42932312016-03-03 00:17:35 +00003453 case POINTERS_TO_MEMBERS_PRAGMA_OPTIONS:
3454 if (Record.size() != 2) {
3455 Error("invalid pragma ms_struct record");
3456 return Failure;
3457 }
3458 PragmaMSPointersToMembersState = Record[0];
3459 PointersToMembersPragmaLocation = ReadSourceLocation(F, Record[1]);
3460 break;
3461
Nico Weber72889432014-09-06 01:25:55 +00003462 case UNUSED_LOCAL_TYPEDEF_NAME_CANDIDATES:
3463 for (unsigned I = 0, N = Record.size(); I != N; ++I)
3464 UnusedLocalTypedefNameCandidates.push_back(
3465 getGlobalDeclID(F, Record[I]));
3466 break;
Justin Lebar67a78a62016-10-08 22:15:58 +00003467
3468 case CUDA_PRAGMA_FORCE_HOST_DEVICE_DEPTH:
3469 if (Record.size() != 1) {
3470 Error("invalid cuda pragma options record");
3471 return Failure;
3472 }
3473 ForceCUDAHostDeviceDepth = Record[0];
3474 break;
Alex Lorenz7d7e1e02017-03-31 15:36:21 +00003475
3476 case PACK_PRAGMA_OPTIONS: {
3477 if (Record.size() < 3) {
3478 Error("invalid pragma pack record");
3479 return Failure;
3480 }
3481 PragmaPackCurrentValue = Record[0];
3482 PragmaPackCurrentLocation = ReadSourceLocation(F, Record[1]);
3483 unsigned NumStackEntries = Record[2];
3484 unsigned Idx = 3;
3485 // Reset the stack when importing a new module.
3486 PragmaPackStack.clear();
3487 for (unsigned I = 0; I < NumStackEntries; ++I) {
3488 PragmaPackStackEntry Entry;
3489 Entry.Value = Record[Idx++];
3490 Entry.Location = ReadSourceLocation(F, Record[Idx++]);
Alex Lorenz45b40142017-07-28 14:41:21 +00003491 Entry.PushLocation = ReadSourceLocation(F, Record[Idx++]);
Alex Lorenz7d7e1e02017-03-31 15:36:21 +00003492 PragmaPackStrings.push_back(ReadString(Record, Idx));
3493 Entry.SlotLabel = PragmaPackStrings.back();
3494 PragmaPackStack.push_back(Entry);
3495 }
3496 break;
3497 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003498 }
3499 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003500}
3501
Richard Smith37a93df2017-02-18 00:32:02 +00003502void ASTReader::ReadModuleOffsetMap(ModuleFile &F) const {
3503 assert(!F.ModuleOffsetMap.empty() && "no module offset map to read");
3504
3505 // Additional remapping information.
Vedant Kumar48b4f762018-04-14 01:40:48 +00003506 const unsigned char *Data = (const unsigned char*)F.ModuleOffsetMap.data();
Richard Smith37a93df2017-02-18 00:32:02 +00003507 const unsigned char *DataEnd = Data + F.ModuleOffsetMap.size();
3508 F.ModuleOffsetMap = StringRef();
3509
3510 // If we see this entry before SOURCE_LOCATION_OFFSETS, add placeholders.
3511 if (F.SLocRemap.find(0) == F.SLocRemap.end()) {
3512 F.SLocRemap.insert(std::make_pair(0U, 0));
3513 F.SLocRemap.insert(std::make_pair(2U, 1));
3514 }
3515
3516 // Continuous range maps we may be updating in our module.
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00003517 using RemapBuilder = ContinuousRangeMap<uint32_t, int, 2>::Builder;
Richard Smith37a93df2017-02-18 00:32:02 +00003518 RemapBuilder SLocRemap(F.SLocRemap);
3519 RemapBuilder IdentifierRemap(F.IdentifierRemap);
3520 RemapBuilder MacroRemap(F.MacroRemap);
3521 RemapBuilder PreprocessedEntityRemap(F.PreprocessedEntityRemap);
3522 RemapBuilder SubmoduleRemap(F.SubmoduleRemap);
3523 RemapBuilder SelectorRemap(F.SelectorRemap);
3524 RemapBuilder DeclRemap(F.DeclRemap);
3525 RemapBuilder TypeRemap(F.TypeRemap);
3526
3527 while (Data < DataEnd) {
Boris Kolpackovd30446f2017-08-31 06:26:43 +00003528 // FIXME: Looking up dependency modules by filename is horrible. Let's
3529 // start fixing this with prebuilt and explicit modules and see how it
3530 // goes...
Richard Smith37a93df2017-02-18 00:32:02 +00003531 using namespace llvm::support;
Vedant Kumar48b4f762018-04-14 01:40:48 +00003532 ModuleKind Kind = static_cast<ModuleKind>(
3533 endian::readNext<uint8_t, little, unaligned>(Data));
Richard Smith37a93df2017-02-18 00:32:02 +00003534 uint16_t Len = endian::readNext<uint16_t, little, unaligned>(Data);
3535 StringRef Name = StringRef((const char*)Data, Len);
3536 Data += Len;
Boris Kolpackovd30446f2017-08-31 06:26:43 +00003537 ModuleFile *OM = (Kind == MK_PrebuiltModule || Kind == MK_ExplicitModule
3538 ? ModuleMgr.lookupByModuleName(Name)
3539 : ModuleMgr.lookupByFileName(Name));
Richard Smith37a93df2017-02-18 00:32:02 +00003540 if (!OM) {
3541 std::string Msg =
3542 "SourceLocation remap refers to unknown module, cannot find ";
3543 Msg.append(Name);
3544 Error(Msg);
3545 return;
3546 }
3547
3548 uint32_t SLocOffset =
3549 endian::readNext<uint32_t, little, unaligned>(Data);
3550 uint32_t IdentifierIDOffset =
3551 endian::readNext<uint32_t, little, unaligned>(Data);
3552 uint32_t MacroIDOffset =
3553 endian::readNext<uint32_t, little, unaligned>(Data);
3554 uint32_t PreprocessedEntityIDOffset =
3555 endian::readNext<uint32_t, little, unaligned>(Data);
3556 uint32_t SubmoduleIDOffset =
3557 endian::readNext<uint32_t, little, unaligned>(Data);
3558 uint32_t SelectorIDOffset =
3559 endian::readNext<uint32_t, little, unaligned>(Data);
3560 uint32_t DeclIDOffset =
3561 endian::readNext<uint32_t, little, unaligned>(Data);
3562 uint32_t TypeIndexOffset =
3563 endian::readNext<uint32_t, little, unaligned>(Data);
3564
3565 uint32_t None = std::numeric_limits<uint32_t>::max();
3566
3567 auto mapOffset = [&](uint32_t Offset, uint32_t BaseOffset,
3568 RemapBuilder &Remap) {
3569 if (Offset != None)
3570 Remap.insert(std::make_pair(Offset,
3571 static_cast<int>(BaseOffset - Offset)));
3572 };
3573 mapOffset(SLocOffset, OM->SLocEntryBaseOffset, SLocRemap);
3574 mapOffset(IdentifierIDOffset, OM->BaseIdentifierID, IdentifierRemap);
3575 mapOffset(MacroIDOffset, OM->BaseMacroID, MacroRemap);
3576 mapOffset(PreprocessedEntityIDOffset, OM->BasePreprocessedEntityID,
3577 PreprocessedEntityRemap);
3578 mapOffset(SubmoduleIDOffset, OM->BaseSubmoduleID, SubmoduleRemap);
3579 mapOffset(SelectorIDOffset, OM->BaseSelectorID, SelectorRemap);
3580 mapOffset(DeclIDOffset, OM->BaseDeclID, DeclRemap);
3581 mapOffset(TypeIndexOffset, OM->BaseTypeIndex, TypeRemap);
3582
3583 // Global -> local mappings.
3584 F.GlobalToLocalDeclIDs[OM] = DeclIDOffset;
3585 }
3586}
3587
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003588ASTReader::ASTReadResult
3589ASTReader::ReadModuleMapFileBlock(RecordData &Record, ModuleFile &F,
3590 const ModuleFile *ImportedBy,
3591 unsigned ClientLoadCapabilities) {
3592 unsigned Idx = 0;
Richard Smith7ed1bc92014-12-05 22:42:13 +00003593 F.ModuleMapPath = ReadPath(F, Record, Idx);
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003594
3595 // Try to resolve ModuleName in the current header search context and
3596 // verify that it is found in the same module map file as we saved. If the
3597 // top-level AST file is a main file, skip this check because there is no
3598 // usable header search context.
3599 assert(!F.ModuleName.empty() &&
Richard Smithe842a472014-10-22 02:05:46 +00003600 "MODULE_NAME should come before MODULE_MAP_FILE");
Duncan P. N. Exon Smith96a06e02017-01-28 22:15:22 +00003601 if (F.Kind == MK_ImplicitModule && ModuleMgr.begin()->Kind != MK_MainFile) {
Richard Smithe842a472014-10-22 02:05:46 +00003602 // An implicitly-loaded module file should have its module listed in some
3603 // module map file that we've already loaded.
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003604 Module *M = PP.getHeaderSearchInfo().lookupModule(F.ModuleName);
Richard Smithe842a472014-10-22 02:05:46 +00003605 auto &Map = PP.getHeaderSearchInfo().getModuleMap();
3606 const FileEntry *ModMap = M ? Map.getModuleMapFileForUniquing(M) : nullptr;
Yuka Takahashid8baec22018-08-01 09:50:02 +00003607 // Don't emit module relocation error if we have -fno-validate-pch
3608 if (!PP.getPreprocessorOpts().DisablePCHValidation && !ModMap) {
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003609 assert(ImportedBy && "top-level import should be verified");
Richard Smith0f99d6a2015-08-09 08:48:41 +00003610 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0) {
Bruno Cardoso Lopesa66a3252017-11-17 03:24:11 +00003611 if (auto *ASTFE = M ? M->getASTFile() : nullptr) {
Richard Smith0f99d6a2015-08-09 08:48:41 +00003612 // This module was defined by an imported (explicit) module.
3613 Diag(diag::err_module_file_conflict) << F.ModuleName << F.FileName
3614 << ASTFE->getName();
Bruno Cardoso Lopesa66a3252017-11-17 03:24:11 +00003615 } else {
Richard Smith0f99d6a2015-08-09 08:48:41 +00003616 // This module was built with a different module map.
3617 Diag(diag::err_imported_module_not_found)
3618 << F.ModuleName << F.FileName << ImportedBy->FileName
3619 << F.ModuleMapPath;
Bruno Cardoso Lopesa66a3252017-11-17 03:24:11 +00003620 // In case it was imported by a PCH, there's a chance the user is
3621 // just missing to include the search path to the directory containing
3622 // the modulemap.
3623 if (ImportedBy->Kind == MK_PCH)
3624 Diag(diag::note_imported_by_pch_module_not_found)
3625 << llvm::sys::path::parent_path(F.ModuleMapPath);
3626 }
Richard Smith0f99d6a2015-08-09 08:48:41 +00003627 }
3628 return OutOfDate;
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003629 }
3630
Richard Smithe842a472014-10-22 02:05:46 +00003631 assert(M->Name == F.ModuleName && "found module with different name");
3632
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003633 // Check the primary module map file.
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003634 const FileEntry *StoredModMap = FileMgr.getFile(F.ModuleMapPath);
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003635 if (StoredModMap == nullptr || StoredModMap != ModMap) {
3636 assert(ModMap && "found module is missing module map file");
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003637 assert(ImportedBy && "top-level import should be verified");
3638 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0)
3639 Diag(diag::err_imported_module_modmap_changed)
3640 << F.ModuleName << ImportedBy->FileName
3641 << ModMap->getName() << F.ModuleMapPath;
3642 return OutOfDate;
3643 }
3644
3645 llvm::SmallPtrSet<const FileEntry *, 1> AdditionalStoredMaps;
3646 for (unsigned I = 0, N = Record[Idx++]; I < N; ++I) {
3647 // FIXME: we should use input files rather than storing names.
Richard Smith7ed1bc92014-12-05 22:42:13 +00003648 std::string Filename = ReadPath(F, Record, Idx);
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003649 const FileEntry *F =
3650 FileMgr.getFile(Filename, false, false);
3651 if (F == nullptr) {
3652 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0)
3653 Error("could not find file '" + Filename +"' referenced by AST file");
3654 return OutOfDate;
3655 }
3656 AdditionalStoredMaps.insert(F);
3657 }
3658
3659 // Check any additional module map files (e.g. module.private.modulemap)
3660 // that are not in the pcm.
3661 if (auto *AdditionalModuleMaps = Map.getAdditionalModuleMapFiles(M)) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00003662 for (const FileEntry *ModMap : *AdditionalModuleMaps) {
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003663 // Remove files that match
3664 // Note: SmallPtrSet::erase is really remove
3665 if (!AdditionalStoredMaps.erase(ModMap)) {
3666 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0)
3667 Diag(diag::err_module_different_modmap)
3668 << F.ModuleName << /*new*/0 << ModMap->getName();
3669 return OutOfDate;
3670 }
3671 }
3672 }
3673
3674 // Check any additional module map files that are in the pcm, but not
3675 // found in header search. Cases that match are already removed.
Vedant Kumar48b4f762018-04-14 01:40:48 +00003676 for (const FileEntry *ModMap : AdditionalStoredMaps) {
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00003677 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0)
3678 Diag(diag::err_module_different_modmap)
3679 << F.ModuleName << /*not new*/1 << ModMap->getName();
3680 return OutOfDate;
3681 }
3682 }
3683
3684 if (Listener)
3685 Listener->ReadModuleMapFile(F.ModuleMapPath);
3686 return Success;
3687}
3688
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00003689/// Move the given method to the back of the global list of methods.
Douglas Gregorc1489562013-02-12 23:36:21 +00003690static void moveMethodToBackOfGlobalList(Sema &S, ObjCMethodDecl *Method) {
3691 // Find the entry for this selector in the method pool.
3692 Sema::GlobalMethodPool::iterator Known
3693 = S.MethodPool.find(Method->getSelector());
3694 if (Known == S.MethodPool.end())
3695 return;
3696
3697 // Retrieve the appropriate method list.
3698 ObjCMethodList &Start = Method->isInstanceMethod()? Known->second.first
3699 : Known->second.second;
3700 bool Found = false;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00003701 for (ObjCMethodList *List = &Start; List; List = List->getNext()) {
Douglas Gregorc1489562013-02-12 23:36:21 +00003702 if (!Found) {
Nico Weber2e0c8f72014-12-27 03:58:08 +00003703 if (List->getMethod() == Method) {
Douglas Gregorc1489562013-02-12 23:36:21 +00003704 Found = true;
3705 } else {
3706 // Keep searching.
3707 continue;
3708 }
3709 }
3710
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00003711 if (List->getNext())
Nico Weber2e0c8f72014-12-27 03:58:08 +00003712 List->setMethod(List->getNext()->getMethod());
Douglas Gregorc1489562013-02-12 23:36:21 +00003713 else
Nico Weber2e0c8f72014-12-27 03:58:08 +00003714 List->setMethod(Method);
Douglas Gregorc1489562013-02-12 23:36:21 +00003715 }
3716}
3717
Richard Smithde711422015-04-23 21:20:19 +00003718void ASTReader::makeNamesVisible(const HiddenNames &Names, Module *Owner) {
Richard Smith10434f32015-05-02 02:08:26 +00003719 assert(Owner->NameVisibility != Module::Hidden && "nothing to make visible?");
Vedant Kumar48b4f762018-04-14 01:40:48 +00003720 for (Decl *D : Names) {
Richard Smith90dc5252017-06-23 01:04:34 +00003721 bool wasHidden = D->isHidden();
3722 D->setVisibleDespiteOwningModule();
Guy Benyei11169dd2012-12-18 14:30:41 +00003723
Vedant Kumar48b4f762018-04-14 01:40:48 +00003724 if (wasHidden && SemaObj) {
3725 if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(D)) {
Richard Smith49f906a2014-03-01 00:08:04 +00003726 moveMethodToBackOfGlobalList(*SemaObj, Method);
Vedant Kumar48b4f762018-04-14 01:40:48 +00003727 }
3728 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003729 }
3730}
3731
Richard Smith49f906a2014-03-01 00:08:04 +00003732void ASTReader::makeModuleVisible(Module *Mod,
Argyrios Kyrtzidis125df052013-02-01 16:36:12 +00003733 Module::NameVisibilityKind NameVisibility,
Richard Smitha7e2cc62015-05-01 01:53:09 +00003734 SourceLocation ImportLoc) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003735 llvm::SmallPtrSet<Module *, 4> Visited;
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003736 SmallVector<Module *, 4> Stack;
Robert Wilhelm25284cc2013-08-23 16:11:15 +00003737 Stack.push_back(Mod);
Guy Benyei11169dd2012-12-18 14:30:41 +00003738 while (!Stack.empty()) {
Robert Wilhelm25284cc2013-08-23 16:11:15 +00003739 Mod = Stack.pop_back_val();
Guy Benyei11169dd2012-12-18 14:30:41 +00003740
3741 if (NameVisibility <= Mod->NameVisibility) {
Robert Wilhelm25284cc2013-08-23 16:11:15 +00003742 // This module already has this level of visibility (or greater), so
Guy Benyei11169dd2012-12-18 14:30:41 +00003743 // there is nothing more to do.
3744 continue;
3745 }
Richard Smith49f906a2014-03-01 00:08:04 +00003746
Guy Benyei11169dd2012-12-18 14:30:41 +00003747 if (!Mod->isAvailable()) {
3748 // Modules that aren't available cannot be made visible.
3749 continue;
3750 }
3751
3752 // Update the module's name visibility.
3753 Mod->NameVisibility = NameVisibility;
Richard Smith49f906a2014-03-01 00:08:04 +00003754
Guy Benyei11169dd2012-12-18 14:30:41 +00003755 // If we've already deserialized any names from this module,
3756 // mark them as visible.
3757 HiddenNamesMapType::iterator Hidden = HiddenNamesMap.find(Mod);
3758 if (Hidden != HiddenNamesMap.end()) {
Richard Smith57721ac2014-07-21 04:10:40 +00003759 auto HiddenNames = std::move(*Hidden);
Guy Benyei11169dd2012-12-18 14:30:41 +00003760 HiddenNamesMap.erase(Hidden);
Richard Smithde711422015-04-23 21:20:19 +00003761 makeNamesVisible(HiddenNames.second, HiddenNames.first);
Richard Smith57721ac2014-07-21 04:10:40 +00003762 assert(HiddenNamesMap.find(Mod) == HiddenNamesMap.end() &&
3763 "making names visible added hidden names");
Guy Benyei11169dd2012-12-18 14:30:41 +00003764 }
Dmitri Gribenkoe9bcf5b2013-11-04 21:51:33 +00003765
Guy Benyei11169dd2012-12-18 14:30:41 +00003766 // Push any exported modules onto the stack to be marked as visible.
Argyrios Kyrtzidis8739f7b2013-02-19 19:34:40 +00003767 SmallVector<Module *, 16> Exports;
3768 Mod->getExportedModules(Exports);
Vedant Kumar48b4f762018-04-14 01:40:48 +00003769 for (SmallVectorImpl<Module *>::iterator
3770 I = Exports.begin(), E = Exports.end(); I != E; ++I) {
3771 Module *Exported = *I;
David Blaikie82e95a32014-11-19 07:49:47 +00003772 if (Visited.insert(Exported).second)
Argyrios Kyrtzidis8739f7b2013-02-19 19:34:40 +00003773 Stack.push_back(Exported);
Vedant Kumar48b4f762018-04-14 01:40:48 +00003774 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003775 }
3776}
3777
Richard Smith6561f922016-09-12 21:06:40 +00003778/// We've merged the definition \p MergedDef into the existing definition
3779/// \p Def. Ensure that \p Def is made visible whenever \p MergedDef is made
3780/// visible.
3781void ASTReader::mergeDefinitionVisibility(NamedDecl *Def,
3782 NamedDecl *MergedDef) {
Richard Smith40344612018-09-12 02:28:14 +00003783 // FIXME: This doesn't correctly handle the case where MergedDef is visible
3784 // in modules other than its owning module. We should instead give the
3785 // ASTContext a list of merged definitions for Def.
Richard Smith6561f922016-09-12 21:06:40 +00003786 if (Def->isHidden()) {
3787 // If MergedDef is visible or becomes visible, make the definition visible.
Benjamin Kramera72a70a2016-10-17 13:00:44 +00003788 if (!MergedDef->isHidden())
Richard Smith90dc5252017-06-23 01:04:34 +00003789 Def->setVisibleDespiteOwningModule();
Richard Smith40344612018-09-12 02:28:14 +00003790 else if (getContext().getLangOpts().ModulesLocalVisibility) {
Benjamin Kramera72a70a2016-10-17 13:00:44 +00003791 getContext().mergeDefinitionIntoModule(
3792 Def, MergedDef->getImportedOwningModule(),
3793 /*NotifyListeners*/ false);
3794 PendingMergedDefinitionsToDeduplicate.insert(Def);
Richard Smith40344612018-09-12 02:28:14 +00003795 } else {
3796 auto SubmoduleID = MergedDef->getOwningModuleID();
3797 assert(SubmoduleID && "hidden definition in no module");
3798 HiddenNamesMap[getSubmodule(SubmoduleID)].push_back(Def);
Benjamin Kramera72a70a2016-10-17 13:00:44 +00003799 }
Richard Smith6561f922016-09-12 21:06:40 +00003800 }
3801}
3802
Douglas Gregore060e572013-01-25 01:03:03 +00003803bool ASTReader::loadGlobalIndex() {
3804 if (GlobalIndex)
3805 return false;
3806
3807 if (TriedLoadingGlobalIndex || !UseGlobalIndex ||
Richard Smithdbafb6c2017-06-29 23:23:46 +00003808 !PP.getLangOpts().Modules)
Douglas Gregore060e572013-01-25 01:03:03 +00003809 return true;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003810
Douglas Gregore060e572013-01-25 01:03:03 +00003811 // Try to load the global index.
3812 TriedLoadingGlobalIndex = true;
3813 StringRef ModuleCachePath
3814 = getPreprocessor().getHeaderSearchInfo().getModuleCachePath();
3815 std::pair<GlobalModuleIndex *, GlobalModuleIndex::ErrorCode> Result
Douglas Gregor7029ce12013-03-19 00:28:20 +00003816 = GlobalModuleIndex::readIndex(ModuleCachePath);
Douglas Gregore060e572013-01-25 01:03:03 +00003817 if (!Result.first)
3818 return true;
3819
3820 GlobalIndex.reset(Result.first);
Douglas Gregor7211ac12013-01-25 23:32:03 +00003821 ModuleMgr.setGlobalIndex(GlobalIndex.get());
Douglas Gregore060e572013-01-25 01:03:03 +00003822 return false;
3823}
3824
3825bool ASTReader::isGlobalIndexUnavailable() const {
Richard Smithdbafb6c2017-06-29 23:23:46 +00003826 return PP.getLangOpts().Modules && UseGlobalIndex &&
Douglas Gregore060e572013-01-25 01:03:03 +00003827 !hasGlobalIndex() && TriedLoadingGlobalIndex;
3828}
3829
Dmitri Gribenkof430da42014-02-12 10:33:14 +00003830static void updateModuleTimestamp(ModuleFile &MF) {
3831 // Overwrite the timestamp file contents so that file's mtime changes.
3832 std::string TimestampFilename = MF.getTimestampFilename();
Rafael Espindoladae941a2014-08-25 18:17:04 +00003833 std::error_code EC;
3834 llvm::raw_fd_ostream OS(TimestampFilename, EC, llvm::sys::fs::F_Text);
3835 if (EC)
Dmitri Gribenkof430da42014-02-12 10:33:14 +00003836 return;
3837 OS << "Timestamp file\n";
Alex Lorenz0bafa022017-06-02 10:36:56 +00003838 OS.close();
3839 OS.clear_error(); // Avoid triggering a fatal error.
Dmitri Gribenkof430da42014-02-12 10:33:14 +00003840}
3841
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00003842/// Given a cursor at the start of an AST file, scan ahead and drop the
Douglas Gregor6623e1f2015-11-03 18:33:07 +00003843/// cursor into the start of the given block ID, returning false on success and
3844/// true on failure.
3845static bool SkipCursorToBlock(BitstreamCursor &Cursor, unsigned BlockID) {
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00003846 while (true) {
Douglas Gregor6623e1f2015-11-03 18:33:07 +00003847 llvm::BitstreamEntry Entry = Cursor.advance();
3848 switch (Entry.Kind) {
3849 case llvm::BitstreamEntry::Error:
3850 case llvm::BitstreamEntry::EndBlock:
3851 return true;
3852
3853 case llvm::BitstreamEntry::Record:
3854 // Ignore top-level records.
3855 Cursor.skipRecord(Entry.ID);
3856 break;
3857
3858 case llvm::BitstreamEntry::SubBlock:
3859 if (Entry.ID == BlockID) {
3860 if (Cursor.EnterSubBlock(BlockID))
3861 return true;
3862 // Found it!
3863 return false;
3864 }
3865
3866 if (Cursor.SkipBlock())
3867 return true;
3868 }
3869 }
3870}
3871
Benjamin Kramer0772c422016-02-13 13:42:54 +00003872ASTReader::ASTReadResult ASTReader::ReadAST(StringRef FileName,
Guy Benyei11169dd2012-12-18 14:30:41 +00003873 ModuleKind Type,
3874 SourceLocation ImportLoc,
Graydon Hoaree7196af2016-12-09 21:45:49 +00003875 unsigned ClientLoadCapabilities,
3876 SmallVectorImpl<ImportedSubmodule> *Imported) {
Argyrios Kyrtzidisdc9fdaf2013-05-24 05:44:08 +00003877 llvm::SaveAndRestore<SourceLocation>
3878 SetCurImportLocRAII(CurrentImportLoc, ImportLoc);
3879
Richard Smithd1c46742014-04-30 02:24:17 +00003880 // Defer any pending actions until we get to the end of reading the AST file.
3881 Deserializing AnASTFile(this);
3882
Guy Benyei11169dd2012-12-18 14:30:41 +00003883 // Bump the generation number.
Richard Smithdbafb6c2017-06-29 23:23:46 +00003884 unsigned PreviousGeneration = 0;
3885 if (ContextObj)
3886 PreviousGeneration = incrementGeneration(*ContextObj);
Guy Benyei11169dd2012-12-18 14:30:41 +00003887
3888 unsigned NumModules = ModuleMgr.size();
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003889 SmallVector<ImportedModule, 4> Loaded;
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00003890 switch (ASTReadResult ReadResult =
3891 ReadASTCore(FileName, Type, ImportLoc,
3892 /*ImportedBy=*/nullptr, Loaded, 0, 0,
3893 ASTFileSignature(), ClientLoadCapabilities)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003894 case Failure:
Douglas Gregor7029ce12013-03-19 00:28:20 +00003895 case Missing:
Guy Benyei11169dd2012-12-18 14:30:41 +00003896 case OutOfDate:
3897 case VersionMismatch:
3898 case ConfigurationMismatch:
Ben Langmuir9801b252014-06-20 00:24:56 +00003899 case HadErrors: {
3900 llvm::SmallPtrSet<ModuleFile *, 4> LoadedSet;
Vedant Kumar48b4f762018-04-14 01:40:48 +00003901 for (const ImportedModule &IM : Loaded)
Ben Langmuir9801b252014-06-20 00:24:56 +00003902 LoadedSet.insert(IM.Mod);
3903
Duncan P. N. Exon Smith8e6bc1972017-01-28 23:02:12 +00003904 ModuleMgr.removeModules(ModuleMgr.begin() + NumModules, LoadedSet,
Richard Smithdbafb6c2017-06-29 23:23:46 +00003905 PP.getLangOpts().Modules
Duncan P. N. Exon Smith8e6bc1972017-01-28 23:02:12 +00003906 ? &PP.getHeaderSearchInfo().getModuleMap()
3907 : nullptr);
Douglas Gregore060e572013-01-25 01:03:03 +00003908
3909 // If we find that any modules are unusable, the global index is going
3910 // to be out-of-date. Just remove it.
3911 GlobalIndex.reset();
Craig Toppera13603a2014-05-22 05:54:18 +00003912 ModuleMgr.setGlobalIndex(nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00003913 return ReadResult;
Ben Langmuir9801b252014-06-20 00:24:56 +00003914 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003915 case Success:
3916 break;
3917 }
3918
3919 // Here comes stuff that we only do once the entire chain is loaded.
3920
3921 // Load the AST blocks of all of the modules that we loaded.
Vedant Kumar48b4f762018-04-14 01:40:48 +00003922 for (SmallVectorImpl<ImportedModule>::iterator M = Loaded.begin(),
3923 MEnd = Loaded.end();
3924 M != MEnd; ++M) {
3925 ModuleFile &F = *M->Mod;
Guy Benyei11169dd2012-12-18 14:30:41 +00003926
3927 // Read the AST block.
Ben Langmuir2c9af442014-04-10 17:57:43 +00003928 if (ASTReadResult Result = ReadASTBlock(F, ClientLoadCapabilities))
3929 return Result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003930
Douglas Gregor6623e1f2015-11-03 18:33:07 +00003931 // Read the extension blocks.
3932 while (!SkipCursorToBlock(F.Stream, EXTENSION_BLOCK_ID)) {
3933 if (ASTReadResult Result = ReadExtensionBlock(F))
3934 return Result;
3935 }
3936
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003937 // Once read, set the ModuleFile bit base offset and update the size in
Guy Benyei11169dd2012-12-18 14:30:41 +00003938 // bits of all files we've seen.
3939 F.GlobalBitOffset = TotalModulesSizeInBits;
3940 TotalModulesSizeInBits += F.SizeInBits;
3941 GlobalBitOffsetsMap.insert(std::make_pair(F.GlobalBitOffset, &F));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00003942
Guy Benyei11169dd2012-12-18 14:30:41 +00003943 // Preload SLocEntries.
Vedant Kumar48b4f762018-04-14 01:40:48 +00003944 for (unsigned I = 0, N = F.PreloadSLocEntries.size(); I != N; ++I) {
3945 int Index = int(F.PreloadSLocEntries[I] - 1) + F.SLocEntryBaseID;
Guy Benyei11169dd2012-12-18 14:30:41 +00003946 // Load it through the SourceManager and don't call ReadSLocEntry()
3947 // directly because the entry may have already been loaded in which case
3948 // calling ReadSLocEntry() directly would trigger an assertion in
3949 // SourceManager.
3950 SourceMgr.getLoadedSLocEntryByID(Index);
3951 }
Richard Smith33e0f7e2015-07-22 02:08:40 +00003952
Richard Smithea741482017-05-01 22:10:47 +00003953 // Map the original source file ID into the ID space of the current
3954 // compilation.
3955 if (F.OriginalSourceFileID.isValid()) {
3956 F.OriginalSourceFileID = FileID::get(
3957 F.SLocEntryBaseID + F.OriginalSourceFileID.getOpaqueValue() - 1);
3958 }
3959
Richard Smith33e0f7e2015-07-22 02:08:40 +00003960 // Preload all the pending interesting identifiers by marking them out of
3961 // date.
3962 for (auto Offset : F.PreloadIdentifierOffsets) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00003963 const unsigned char *Data = reinterpret_cast<const unsigned char *>(
Richard Smith33e0f7e2015-07-22 02:08:40 +00003964 F.IdentifierTableData + Offset);
3965
3966 ASTIdentifierLookupTrait Trait(*this, F);
3967 auto KeyDataLen = Trait.ReadKeyDataLength(Data);
3968 auto Key = Trait.ReadKey(Data, KeyDataLen.first);
Richard Smith79bf9202015-08-24 03:33:22 +00003969 auto &II = PP.getIdentifierTable().getOwn(Key);
3970 II.setOutOfDate(true);
3971
3972 // Mark this identifier as being from an AST file so that we can track
3973 // whether we need to serialize it.
Richard Smitheb4b58f62016-02-05 01:40:54 +00003974 markIdentifierFromAST(*this, II);
Richard Smith79bf9202015-08-24 03:33:22 +00003975
3976 // Associate the ID with the identifier so that the writer can reuse it.
3977 auto ID = Trait.ReadIdentifierID(Data + KeyDataLen.first);
3978 SetIdentifierInfo(ID, &II);
Richard Smith33e0f7e2015-07-22 02:08:40 +00003979 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003980 }
3981
Douglas Gregor603cd862013-03-22 18:50:14 +00003982 // Setup the import locations and notify the module manager that we've
3983 // committed to these module files.
Vedant Kumar48b4f762018-04-14 01:40:48 +00003984 for (SmallVectorImpl<ImportedModule>::iterator M = Loaded.begin(),
3985 MEnd = Loaded.end();
3986 M != MEnd; ++M) {
3987 ModuleFile &F = *M->Mod;
Douglas Gregor603cd862013-03-22 18:50:14 +00003988
3989 ModuleMgr.moduleFileAccepted(&F);
3990
3991 // Set the import location.
Argyrios Kyrtzidis71c1af82013-02-01 16:36:14 +00003992 F.DirectImportLoc = ImportLoc;
Richard Smithb22a1d12016-03-27 20:13:24 +00003993 // FIXME: We assume that locations from PCH / preamble do not need
3994 // any translation.
Vedant Kumar48b4f762018-04-14 01:40:48 +00003995 if (!M->ImportedBy)
3996 F.ImportLoc = M->ImportLoc;
Guy Benyei11169dd2012-12-18 14:30:41 +00003997 else
Vedant Kumar48b4f762018-04-14 01:40:48 +00003998 F.ImportLoc = TranslateSourceLocation(*M->ImportedBy, M->ImportLoc);
Guy Benyei11169dd2012-12-18 14:30:41 +00003999 }
4000
Richard Smithdbafb6c2017-06-29 23:23:46 +00004001 if (!PP.getLangOpts().CPlusPlus ||
Manman Ren11f2a472016-08-18 17:42:15 +00004002 (Type != MK_ImplicitModule && Type != MK_ExplicitModule &&
4003 Type != MK_PrebuiltModule)) {
Richard Smith33e0f7e2015-07-22 02:08:40 +00004004 // Mark all of the identifiers in the identifier table as being out of date,
4005 // so that various accessors know to check the loaded modules when the
4006 // identifier is used.
4007 //
4008 // For C++ modules, we don't need information on many identifiers (just
4009 // those that provide macros or are poisoned), so we mark all of
4010 // the interesting ones via PreloadIdentifierOffsets.
Vedant Kumar48b4f762018-04-14 01:40:48 +00004011 for (IdentifierTable::iterator Id = PP.getIdentifierTable().begin(),
4012 IdEnd = PP.getIdentifierTable().end();
4013 Id != IdEnd; ++Id)
4014 Id->second->setOutOfDate(true);
Richard Smith33e0f7e2015-07-22 02:08:40 +00004015 }
Manman Rena0f31a02016-04-29 19:04:05 +00004016 // Mark selectors as out of date.
4017 for (auto Sel : SelectorGeneration)
4018 SelectorOutOfDate[Sel.first] = true;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004019
Guy Benyei11169dd2012-12-18 14:30:41 +00004020 // Resolve any unresolved module exports.
Vedant Kumar48b4f762018-04-14 01:40:48 +00004021 for (unsigned I = 0, N = UnresolvedModuleRefs.size(); I != N; ++I) {
4022 UnresolvedModuleRef &Unresolved = UnresolvedModuleRefs[I];
Guy Benyei11169dd2012-12-18 14:30:41 +00004023 SubmoduleID GlobalID = getGlobalSubmoduleID(*Unresolved.File,Unresolved.ID);
4024 Module *ResolvedMod = getSubmodule(GlobalID);
Douglas Gregorfb912652013-03-20 21:10:35 +00004025
4026 switch (Unresolved.Kind) {
4027 case UnresolvedModuleRef::Conflict:
4028 if (ResolvedMod) {
4029 Module::Conflict Conflict;
4030 Conflict.Other = ResolvedMod;
4031 Conflict.Message = Unresolved.String.str();
4032 Unresolved.Mod->Conflicts.push_back(Conflict);
4033 }
4034 continue;
4035
4036 case UnresolvedModuleRef::Import:
Guy Benyei11169dd2012-12-18 14:30:41 +00004037 if (ResolvedMod)
Richard Smith38477db2015-05-02 00:45:56 +00004038 Unresolved.Mod->Imports.insert(ResolvedMod);
Guy Benyei11169dd2012-12-18 14:30:41 +00004039 continue;
Guy Benyei11169dd2012-12-18 14:30:41 +00004040
Douglas Gregorfb912652013-03-20 21:10:35 +00004041 case UnresolvedModuleRef::Export:
4042 if (ResolvedMod || Unresolved.IsWildcard)
4043 Unresolved.Mod->Exports.push_back(
4044 Module::ExportDecl(ResolvedMod, Unresolved.IsWildcard));
4045 continue;
4046 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004047 }
Douglas Gregorfb912652013-03-20 21:10:35 +00004048 UnresolvedModuleRefs.clear();
Daniel Jasperba7f2f72013-09-24 09:14:14 +00004049
Graydon Hoaree7196af2016-12-09 21:45:49 +00004050 if (Imported)
4051 Imported->append(ImportedModules.begin(),
4052 ImportedModules.end());
4053
Daniel Jasperba7f2f72013-09-24 09:14:14 +00004054 // FIXME: How do we load the 'use'd modules? They may not be submodules.
4055 // Might be unnecessary as use declarations are only used to build the
4056 // module itself.
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004057
Richard Smithdbafb6c2017-06-29 23:23:46 +00004058 if (ContextObj)
4059 InitializeContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00004060
Richard Smith3d8e97e2013-10-18 06:54:39 +00004061 if (SemaObj)
4062 UpdateSema();
4063
Guy Benyei11169dd2012-12-18 14:30:41 +00004064 if (DeserializationListener)
4065 DeserializationListener->ReaderInitialized(this);
4066
4067 ModuleFile &PrimaryModule = ModuleMgr.getPrimaryModule();
Yaron Keren8b563662015-10-03 10:46:20 +00004068 if (PrimaryModule.OriginalSourceFileID.isValid()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004069 // If this AST file is a precompiled preamble, then set the
4070 // preamble file ID of the source manager to the file source file
4071 // from which the preamble was built.
4072 if (Type == MK_Preamble) {
4073 SourceMgr.setPreambleFileID(PrimaryModule.OriginalSourceFileID);
4074 } else if (Type == MK_MainFile) {
4075 SourceMgr.setMainFileID(PrimaryModule.OriginalSourceFileID);
4076 }
4077 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004078
Guy Benyei11169dd2012-12-18 14:30:41 +00004079 // For any Objective-C class definitions we have already loaded, make sure
4080 // that we load any additional categories.
Vedant Kumar48b4f762018-04-14 01:40:48 +00004081 if (ContextObj) {
4082 for (unsigned I = 0, N = ObjCClassesLoaded.size(); I != N; ++I) {
4083 loadObjCCategories(ObjCClassesLoaded[I]->getGlobalID(),
4084 ObjCClassesLoaded[I],
Richard Smithdbafb6c2017-06-29 23:23:46 +00004085 PreviousGeneration);
Vedant Kumar48b4f762018-04-14 01:40:48 +00004086 }
4087 }
Douglas Gregore060e572013-01-25 01:03:03 +00004088
Dmitri Gribenkof430da42014-02-12 10:33:14 +00004089 if (PP.getHeaderSearchInfo()
4090 .getHeaderSearchOpts()
4091 .ModulesValidateOncePerBuildSession) {
4092 // Now we are certain that the module and all modules it depends on are
4093 // up to date. Create or update timestamp files for modules that are
4094 // located in the module cache (not for PCH files that could be anywhere
4095 // in the filesystem).
Vedant Kumar48b4f762018-04-14 01:40:48 +00004096 for (unsigned I = 0, N = Loaded.size(); I != N; ++I) {
4097 ImportedModule &M = Loaded[I];
4098 if (M.Mod->Kind == MK_ImplicitModule) {
Dmitri Gribenkof430da42014-02-12 10:33:14 +00004099 updateModuleTimestamp(*M.Mod);
Vedant Kumar48b4f762018-04-14 01:40:48 +00004100 }
4101 }
Dmitri Gribenkof430da42014-02-12 10:33:14 +00004102 }
4103
Guy Benyei11169dd2012-12-18 14:30:41 +00004104 return Success;
4105}
4106
Peter Collingbourne77c89b62016-11-08 04:17:11 +00004107static ASTFileSignature readASTFileSignature(StringRef PCH);
Ben Langmuir487ea142014-10-23 18:05:36 +00004108
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00004109/// Whether \p Stream starts with the AST/PCH file magic number 'CPCH'.
Ben Langmuir70a1b812015-03-24 04:43:52 +00004110static bool startsWithASTFileMagic(BitstreamCursor &Stream) {
Peter Collingbourne028eb5a2016-11-02 00:08:19 +00004111 return Stream.canSkipToPos(4) &&
4112 Stream.Read(8) == 'C' &&
Ben Langmuir70a1b812015-03-24 04:43:52 +00004113 Stream.Read(8) == 'P' &&
4114 Stream.Read(8) == 'C' &&
4115 Stream.Read(8) == 'H';
4116}
4117
Richard Smith0f99d6a2015-08-09 08:48:41 +00004118static unsigned moduleKindForDiagnostic(ModuleKind Kind) {
4119 switch (Kind) {
4120 case MK_PCH:
4121 return 0; // PCH
4122 case MK_ImplicitModule:
4123 case MK_ExplicitModule:
Manman Ren11f2a472016-08-18 17:42:15 +00004124 case MK_PrebuiltModule:
Richard Smith0f99d6a2015-08-09 08:48:41 +00004125 return 1; // module
4126 case MK_MainFile:
4127 case MK_Preamble:
4128 return 2; // main source file
4129 }
4130 llvm_unreachable("unknown module kind");
4131}
4132
Guy Benyei11169dd2012-12-18 14:30:41 +00004133ASTReader::ASTReadResult
4134ASTReader::ReadASTCore(StringRef FileName,
4135 ModuleKind Type,
4136 SourceLocation ImportLoc,
4137 ModuleFile *ImportedBy,
Dmitri Gribenkof8579502013-01-12 19:30:44 +00004138 SmallVectorImpl<ImportedModule> &Loaded,
Douglas Gregor7029ce12013-03-19 00:28:20 +00004139 off_t ExpectedSize, time_t ExpectedModTime,
Ben Langmuir487ea142014-10-23 18:05:36 +00004140 ASTFileSignature ExpectedSignature,
Guy Benyei11169dd2012-12-18 14:30:41 +00004141 unsigned ClientLoadCapabilities) {
4142 ModuleFile *M;
Guy Benyei11169dd2012-12-18 14:30:41 +00004143 std::string ErrorStr;
Douglas Gregor7029ce12013-03-19 00:28:20 +00004144 ModuleManager::AddModuleResult AddResult
4145 = ModuleMgr.addModule(FileName, Type, ImportLoc, ImportedBy,
Richard Smith053f6c62014-05-16 23:01:30 +00004146 getGeneration(), ExpectedSize, ExpectedModTime,
Ben Langmuir487ea142014-10-23 18:05:36 +00004147 ExpectedSignature, readASTFileSignature,
Douglas Gregor7029ce12013-03-19 00:28:20 +00004148 M, ErrorStr);
Guy Benyei11169dd2012-12-18 14:30:41 +00004149
Douglas Gregor7029ce12013-03-19 00:28:20 +00004150 switch (AddResult) {
4151 case ModuleManager::AlreadyLoaded:
4152 return Success;
4153
4154 case ModuleManager::NewlyLoaded:
4155 // Load module file below.
4156 break;
4157
4158 case ModuleManager::Missing:
Richard Smithe842a472014-10-22 02:05:46 +00004159 // The module file was missing; if the client can handle that, return
Douglas Gregor7029ce12013-03-19 00:28:20 +00004160 // it.
4161 if (ClientLoadCapabilities & ARR_Missing)
4162 return Missing;
4163
4164 // Otherwise, return an error.
Richard Smith0f99d6a2015-08-09 08:48:41 +00004165 Diag(diag::err_module_file_not_found) << moduleKindForDiagnostic(Type)
Adrian Prantlb3b5a732016-08-29 20:46:59 +00004166 << FileName << !ErrorStr.empty()
Richard Smith0f99d6a2015-08-09 08:48:41 +00004167 << ErrorStr;
Douglas Gregor7029ce12013-03-19 00:28:20 +00004168 return Failure;
4169
4170 case ModuleManager::OutOfDate:
4171 // We couldn't load the module file because it is out-of-date. If the
4172 // client can handle out-of-date, return it.
4173 if (ClientLoadCapabilities & ARR_OutOfDate)
4174 return OutOfDate;
4175
4176 // Otherwise, return an error.
Richard Smith0f99d6a2015-08-09 08:48:41 +00004177 Diag(diag::err_module_file_out_of_date) << moduleKindForDiagnostic(Type)
Adrian Prantl9a06a882016-08-29 20:46:56 +00004178 << FileName << !ErrorStr.empty()
Richard Smith0f99d6a2015-08-09 08:48:41 +00004179 << ErrorStr;
Guy Benyei11169dd2012-12-18 14:30:41 +00004180 return Failure;
4181 }
4182
Douglas Gregor7029ce12013-03-19 00:28:20 +00004183 assert(M && "Missing module file");
Guy Benyei11169dd2012-12-18 14:30:41 +00004184
Guy Benyei11169dd2012-12-18 14:30:41 +00004185 ModuleFile &F = *M;
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004186 BitstreamCursor &Stream = F.Stream;
Peter Collingbourne77c89b62016-11-08 04:17:11 +00004187 Stream = BitstreamCursor(PCHContainerRdr.ExtractPCH(*F.Buffer));
Adrian Prantlcbc368c2015-02-25 02:44:04 +00004188 F.SizeInBits = F.Buffer->getBufferSize() * 8;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004189
Guy Benyei11169dd2012-12-18 14:30:41 +00004190 // Sniff for the signature.
Ben Langmuir70a1b812015-03-24 04:43:52 +00004191 if (!startsWithASTFileMagic(Stream)) {
Richard Smith0f99d6a2015-08-09 08:48:41 +00004192 Diag(diag::err_module_file_invalid) << moduleKindForDiagnostic(Type)
4193 << FileName;
Guy Benyei11169dd2012-12-18 14:30:41 +00004194 return Failure;
4195 }
4196
4197 // This is used for compatibility with older PCH formats.
4198 bool HaveReadControlBlock = false;
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00004199 while (true) {
Chris Lattnerefa77172013-01-20 00:00:22 +00004200 llvm::BitstreamEntry Entry = Stream.advance();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004201
Chris Lattnerefa77172013-01-20 00:00:22 +00004202 switch (Entry.Kind) {
4203 case llvm::BitstreamEntry::Error:
Chris Lattnerefa77172013-01-20 00:00:22 +00004204 case llvm::BitstreamEntry::Record:
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004205 case llvm::BitstreamEntry::EndBlock:
Guy Benyei11169dd2012-12-18 14:30:41 +00004206 Error("invalid record at top-level of AST file");
4207 return Failure;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004208
Chris Lattnerefa77172013-01-20 00:00:22 +00004209 case llvm::BitstreamEntry::SubBlock:
4210 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00004211 }
4212
Chris Lattnerefa77172013-01-20 00:00:22 +00004213 switch (Entry.ID) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004214 case CONTROL_BLOCK_ID:
4215 HaveReadControlBlock = true;
Ben Langmuirbeee15e2014-04-14 18:00:01 +00004216 switch (ReadControlBlock(F, Loaded, ImportedBy, ClientLoadCapabilities)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004217 case Success:
Richard Smith0f99d6a2015-08-09 08:48:41 +00004218 // Check that we didn't try to load a non-module AST file as a module.
4219 //
4220 // FIXME: Should we also perform the converse check? Loading a module as
4221 // a PCH file sort of works, but it's a bit wonky.
Manman Ren11f2a472016-08-18 17:42:15 +00004222 if ((Type == MK_ImplicitModule || Type == MK_ExplicitModule ||
4223 Type == MK_PrebuiltModule) &&
Richard Smith0f99d6a2015-08-09 08:48:41 +00004224 F.ModuleName.empty()) {
4225 auto Result = (Type == MK_ImplicitModule) ? OutOfDate : Failure;
4226 if (Result != OutOfDate ||
4227 (ClientLoadCapabilities & ARR_OutOfDate) == 0)
4228 Diag(diag::err_module_file_not_module) << FileName;
4229 return Result;
4230 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004231 break;
4232
4233 case Failure: return Failure;
Douglas Gregor7029ce12013-03-19 00:28:20 +00004234 case Missing: return Missing;
Guy Benyei11169dd2012-12-18 14:30:41 +00004235 case OutOfDate: return OutOfDate;
4236 case VersionMismatch: return VersionMismatch;
4237 case ConfigurationMismatch: return ConfigurationMismatch;
4238 case HadErrors: return HadErrors;
4239 }
4240 break;
Richard Smithf8c32552015-09-02 17:45:54 +00004241
Guy Benyei11169dd2012-12-18 14:30:41 +00004242 case AST_BLOCK_ID:
4243 if (!HaveReadControlBlock) {
4244 if ((ClientLoadCapabilities & ARR_VersionMismatch) == 0)
Dmitri Gribenko2228cd32014-02-11 15:40:09 +00004245 Diag(diag::err_pch_version_too_old);
Guy Benyei11169dd2012-12-18 14:30:41 +00004246 return VersionMismatch;
4247 }
4248
4249 // Record that we've loaded this module.
4250 Loaded.push_back(ImportedModule(M, ImportedBy, ImportLoc));
4251 return Success;
4252
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004253 case UNHASHED_CONTROL_BLOCK_ID:
4254 // This block is handled using look-ahead during ReadControlBlock. We
4255 // shouldn't get here!
4256 Error("malformed block record in AST file");
4257 return Failure;
4258
Guy Benyei11169dd2012-12-18 14:30:41 +00004259 default:
4260 if (Stream.SkipBlock()) {
4261 Error("malformed block record in AST file");
4262 return Failure;
4263 }
4264 break;
4265 }
4266 }
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004267
4268 return Success;
4269}
4270
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004271ASTReader::ASTReadResult
4272ASTReader::readUnhashedControlBlock(ModuleFile &F, bool WasImportedBy,
4273 unsigned ClientLoadCapabilities) {
4274 const HeaderSearchOptions &HSOpts =
4275 PP.getHeaderSearchInfo().getHeaderSearchOpts();
4276 bool AllowCompatibleConfigurationMismatch =
4277 F.Kind == MK_ExplicitModule || F.Kind == MK_PrebuiltModule;
4278
4279 ASTReadResult Result = readUnhashedControlBlockImpl(
4280 &F, F.Data, ClientLoadCapabilities, AllowCompatibleConfigurationMismatch,
4281 Listener.get(),
4282 WasImportedBy ? false : HSOpts.ModulesValidateDiagnosticOptions);
4283
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +00004284 // If F was directly imported by another module, it's implicitly validated by
4285 // the importing module.
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004286 if (DisableValidation || WasImportedBy ||
4287 (AllowConfigurationMismatch && Result == ConfigurationMismatch))
4288 return Success;
4289
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +00004290 if (Result == Failure) {
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004291 Error("malformed block record in AST file");
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +00004292 return Failure;
4293 }
4294
4295 if (Result == OutOfDate && F.Kind == MK_ImplicitModule) {
4296 // If this module has already been finalized in the PCMCache, we're stuck
4297 // with it; we can only load a single version of each module.
4298 //
4299 // This can happen when a module is imported in two contexts: in one, as a
4300 // user module; in another, as a system module (due to an import from
4301 // another module marked with the [system] flag). It usually indicates a
4302 // bug in the module map: this module should also be marked with [system].
4303 //
4304 // If -Wno-system-headers (the default), and the first import is as a
4305 // system module, then validation will fail during the as-user import,
4306 // since -Werror flags won't have been validated. However, it's reasonable
4307 // to treat this consistently as a system module.
4308 //
4309 // If -Wsystem-headers, the PCM on disk was built with
4310 // -Wno-system-headers, and the first import is as a user module, then
4311 // validation will fail during the as-system import since the PCM on disk
4312 // doesn't guarantee that -Werror was respected. However, the -Werror
4313 // flags were checked during the initial as-user import.
4314 if (PCMCache.isBufferFinal(F.FileName)) {
4315 Diag(diag::warn_module_system_bit_conflict) << F.FileName;
4316 return Success;
4317 }
4318 }
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004319
4320 return Result;
4321}
4322
4323ASTReader::ASTReadResult ASTReader::readUnhashedControlBlockImpl(
4324 ModuleFile *F, llvm::StringRef StreamData, unsigned ClientLoadCapabilities,
4325 bool AllowCompatibleConfigurationMismatch, ASTReaderListener *Listener,
4326 bool ValidateDiagnosticOptions) {
4327 // Initialize a stream.
4328 BitstreamCursor Stream(StreamData);
4329
4330 // Sniff for the signature.
4331 if (!startsWithASTFileMagic(Stream))
4332 return Failure;
4333
4334 // Scan for the UNHASHED_CONTROL_BLOCK_ID block.
4335 if (SkipCursorToBlock(Stream, UNHASHED_CONTROL_BLOCK_ID))
4336 return Failure;
4337
4338 // Read all of the records in the options block.
4339 RecordData Record;
4340 ASTReadResult Result = Success;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00004341 while (true) {
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004342 llvm::BitstreamEntry Entry = Stream.advance();
4343
4344 switch (Entry.Kind) {
4345 case llvm::BitstreamEntry::Error:
4346 case llvm::BitstreamEntry::SubBlock:
4347 return Failure;
4348
4349 case llvm::BitstreamEntry::EndBlock:
4350 return Result;
4351
4352 case llvm::BitstreamEntry::Record:
4353 // The interesting case.
4354 break;
4355 }
4356
4357 // Read and process a record.
4358 Record.clear();
4359 switch (
4360 (UnhashedControlBlockRecordTypes)Stream.readRecord(Entry.ID, Record)) {
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00004361 case SIGNATURE:
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004362 if (F)
4363 std::copy(Record.begin(), Record.end(), F->Signature.data());
4364 break;
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004365 case DIAGNOSTIC_OPTIONS: {
4366 bool Complain = (ClientLoadCapabilities & ARR_OutOfDate) == 0;
4367 if (Listener && ValidateDiagnosticOptions &&
4368 !AllowCompatibleConfigurationMismatch &&
4369 ParseDiagnosticOptions(Record, Complain, *Listener))
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +00004370 Result = OutOfDate; // Don't return early. Read the signature.
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004371 break;
4372 }
4373 case DIAG_PRAGMA_MAPPINGS:
4374 if (!F)
4375 break;
4376 if (F->PragmaDiagMappings.empty())
4377 F->PragmaDiagMappings.swap(Record);
4378 else
4379 F->PragmaDiagMappings.insert(F->PragmaDiagMappings.end(),
4380 Record.begin(), Record.end());
4381 break;
4382 }
4383 }
4384}
4385
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004386/// Parse a record and blob containing module file extension metadata.
4387static bool parseModuleFileExtensionMetadata(
4388 const SmallVectorImpl<uint64_t> &Record,
4389 StringRef Blob,
4390 ModuleFileExtensionMetadata &Metadata) {
4391 if (Record.size() < 4) return true;
4392
4393 Metadata.MajorVersion = Record[0];
4394 Metadata.MinorVersion = Record[1];
4395
4396 unsigned BlockNameLen = Record[2];
4397 unsigned UserInfoLen = Record[3];
4398
4399 if (BlockNameLen + UserInfoLen > Blob.size()) return true;
4400
4401 Metadata.BlockName = std::string(Blob.data(), Blob.data() + BlockNameLen);
4402 Metadata.UserInfo = std::string(Blob.data() + BlockNameLen,
4403 Blob.data() + BlockNameLen + UserInfoLen);
4404 return false;
4405}
4406
4407ASTReader::ASTReadResult ASTReader::ReadExtensionBlock(ModuleFile &F) {
4408 BitstreamCursor &Stream = F.Stream;
4409
4410 RecordData Record;
4411 while (true) {
4412 llvm::BitstreamEntry Entry = Stream.advance();
4413 switch (Entry.Kind) {
4414 case llvm::BitstreamEntry::SubBlock:
4415 if (Stream.SkipBlock())
4416 return Failure;
4417
4418 continue;
4419
4420 case llvm::BitstreamEntry::EndBlock:
4421 return Success;
4422
4423 case llvm::BitstreamEntry::Error:
4424 return HadErrors;
4425
4426 case llvm::BitstreamEntry::Record:
4427 break;
4428 }
4429
4430 Record.clear();
4431 StringRef Blob;
4432 unsigned RecCode = Stream.readRecord(Entry.ID, Record, &Blob);
4433 switch (RecCode) {
4434 case EXTENSION_METADATA: {
4435 ModuleFileExtensionMetadata Metadata;
4436 if (parseModuleFileExtensionMetadata(Record, Blob, Metadata))
4437 return Failure;
4438
4439 // Find a module file extension with this block name.
4440 auto Known = ModuleFileExtensions.find(Metadata.BlockName);
4441 if (Known == ModuleFileExtensions.end()) break;
4442
4443 // Form a reader.
4444 if (auto Reader = Known->second->createExtensionReader(Metadata, *this,
4445 F, Stream)) {
4446 F.ExtensionReaders.push_back(std::move(Reader));
4447 }
4448
4449 break;
4450 }
4451 }
4452 }
4453
4454 return Success;
Guy Benyei11169dd2012-12-18 14:30:41 +00004455}
4456
Richard Smitha7e2cc62015-05-01 01:53:09 +00004457void ASTReader::InitializeContext() {
Richard Smithdbafb6c2017-06-29 23:23:46 +00004458 assert(ContextObj && "no context to initialize");
4459 ASTContext &Context = *ContextObj;
4460
Guy Benyei11169dd2012-12-18 14:30:41 +00004461 // If there's a listener, notify them that we "read" the translation unit.
4462 if (DeserializationListener)
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004463 DeserializationListener->DeclRead(PREDEF_DECL_TRANSLATION_UNIT_ID,
Guy Benyei11169dd2012-12-18 14:30:41 +00004464 Context.getTranslationUnitDecl());
4465
Guy Benyei11169dd2012-12-18 14:30:41 +00004466 // FIXME: Find a better way to deal with collisions between these
4467 // built-in types. Right now, we just ignore the problem.
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004468
Guy Benyei11169dd2012-12-18 14:30:41 +00004469 // Load the special types.
4470 if (SpecialTypes.size() >= NumSpecialTypeIDs) {
4471 if (unsigned String = SpecialTypes[SPECIAL_TYPE_CF_CONSTANT_STRING]) {
4472 if (!Context.CFConstantStringTypeDecl)
4473 Context.setCFConstantStringType(GetType(String));
4474 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004475
Guy Benyei11169dd2012-12-18 14:30:41 +00004476 if (unsigned File = SpecialTypes[SPECIAL_TYPE_FILE]) {
4477 QualType FileType = GetType(File);
4478 if (FileType.isNull()) {
4479 Error("FILE type is NULL");
4480 return;
4481 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004482
Guy Benyei11169dd2012-12-18 14:30:41 +00004483 if (!Context.FILEDecl) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00004484 if (const TypedefType *Typedef = FileType->getAs<TypedefType>())
Guy Benyei11169dd2012-12-18 14:30:41 +00004485 Context.setFILEDecl(Typedef->getDecl());
4486 else {
Vedant Kumar48b4f762018-04-14 01:40:48 +00004487 const TagType *Tag = FileType->getAs<TagType>();
Guy Benyei11169dd2012-12-18 14:30:41 +00004488 if (!Tag) {
4489 Error("Invalid FILE type in AST file");
4490 return;
4491 }
4492 Context.setFILEDecl(Tag->getDecl());
4493 }
4494 }
4495 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004496
Guy Benyei11169dd2012-12-18 14:30:41 +00004497 if (unsigned Jmp_buf = SpecialTypes[SPECIAL_TYPE_JMP_BUF]) {
4498 QualType Jmp_bufType = GetType(Jmp_buf);
4499 if (Jmp_bufType.isNull()) {
4500 Error("jmp_buf type is NULL");
4501 return;
4502 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004503
Guy Benyei11169dd2012-12-18 14:30:41 +00004504 if (!Context.jmp_bufDecl) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00004505 if (const TypedefType *Typedef = Jmp_bufType->getAs<TypedefType>())
Guy Benyei11169dd2012-12-18 14:30:41 +00004506 Context.setjmp_bufDecl(Typedef->getDecl());
4507 else {
Vedant Kumar48b4f762018-04-14 01:40:48 +00004508 const TagType *Tag = Jmp_bufType->getAs<TagType>();
Guy Benyei11169dd2012-12-18 14:30:41 +00004509 if (!Tag) {
4510 Error("Invalid jmp_buf type in AST file");
4511 return;
4512 }
4513 Context.setjmp_bufDecl(Tag->getDecl());
4514 }
4515 }
4516 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004517
Guy Benyei11169dd2012-12-18 14:30:41 +00004518 if (unsigned Sigjmp_buf = SpecialTypes[SPECIAL_TYPE_SIGJMP_BUF]) {
4519 QualType Sigjmp_bufType = GetType(Sigjmp_buf);
4520 if (Sigjmp_bufType.isNull()) {
4521 Error("sigjmp_buf type is NULL");
4522 return;
4523 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004524
Guy Benyei11169dd2012-12-18 14:30:41 +00004525 if (!Context.sigjmp_bufDecl) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00004526 if (const TypedefType *Typedef = Sigjmp_bufType->getAs<TypedefType>())
Guy Benyei11169dd2012-12-18 14:30:41 +00004527 Context.setsigjmp_bufDecl(Typedef->getDecl());
4528 else {
Vedant Kumar48b4f762018-04-14 01:40:48 +00004529 const TagType *Tag = Sigjmp_bufType->getAs<TagType>();
Guy Benyei11169dd2012-12-18 14:30:41 +00004530 assert(Tag && "Invalid sigjmp_buf type in AST file");
4531 Context.setsigjmp_bufDecl(Tag->getDecl());
4532 }
4533 }
4534 }
4535
4536 if (unsigned ObjCIdRedef
4537 = SpecialTypes[SPECIAL_TYPE_OBJC_ID_REDEFINITION]) {
4538 if (Context.ObjCIdRedefinitionType.isNull())
4539 Context.ObjCIdRedefinitionType = GetType(ObjCIdRedef);
4540 }
4541
4542 if (unsigned ObjCClassRedef
4543 = SpecialTypes[SPECIAL_TYPE_OBJC_CLASS_REDEFINITION]) {
4544 if (Context.ObjCClassRedefinitionType.isNull())
4545 Context.ObjCClassRedefinitionType = GetType(ObjCClassRedef);
4546 }
4547
4548 if (unsigned ObjCSelRedef
4549 = SpecialTypes[SPECIAL_TYPE_OBJC_SEL_REDEFINITION]) {
4550 if (Context.ObjCSelRedefinitionType.isNull())
4551 Context.ObjCSelRedefinitionType = GetType(ObjCSelRedef);
4552 }
4553
4554 if (unsigned Ucontext_t = SpecialTypes[SPECIAL_TYPE_UCONTEXT_T]) {
4555 QualType Ucontext_tType = GetType(Ucontext_t);
4556 if (Ucontext_tType.isNull()) {
4557 Error("ucontext_t type is NULL");
4558 return;
4559 }
4560
4561 if (!Context.ucontext_tDecl) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00004562 if (const TypedefType *Typedef = Ucontext_tType->getAs<TypedefType>())
Guy Benyei11169dd2012-12-18 14:30:41 +00004563 Context.setucontext_tDecl(Typedef->getDecl());
4564 else {
Vedant Kumar48b4f762018-04-14 01:40:48 +00004565 const TagType *Tag = Ucontext_tType->getAs<TagType>();
Guy Benyei11169dd2012-12-18 14:30:41 +00004566 assert(Tag && "Invalid ucontext_t type in AST file");
4567 Context.setucontext_tDecl(Tag->getDecl());
4568 }
4569 }
4570 }
4571 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004572
Guy Benyei11169dd2012-12-18 14:30:41 +00004573 ReadPragmaDiagnosticMappings(Context.getDiagnostics());
4574
4575 // If there were any CUDA special declarations, deserialize them.
4576 if (!CUDASpecialDeclRefs.empty()) {
4577 assert(CUDASpecialDeclRefs.size() == 1 && "More decl refs than expected!");
4578 Context.setcudaConfigureCallDecl(
4579 cast<FunctionDecl>(GetDecl(CUDASpecialDeclRefs[0])));
4580 }
Richard Smith56be7542014-03-21 00:33:59 +00004581
Guy Benyei11169dd2012-12-18 14:30:41 +00004582 // Re-export any modules that were imported by a non-module AST file.
Richard Smitha7e2cc62015-05-01 01:53:09 +00004583 // FIXME: This does not make macro-only imports visible again.
Richard Smith56be7542014-03-21 00:33:59 +00004584 for (auto &Import : ImportedModules) {
Richard Smitha7e2cc62015-05-01 01:53:09 +00004585 if (Module *Imported = getSubmodule(Import.ID)) {
Argyrios Kyrtzidis125df052013-02-01 16:36:12 +00004586 makeModuleVisible(Imported, Module::AllVisible,
Richard Smitha7e2cc62015-05-01 01:53:09 +00004587 /*ImportLoc=*/Import.ImportLoc);
Ben Langmuir6d25fdc2016-02-11 17:04:42 +00004588 if (Import.ImportLoc.isValid())
4589 PP.makeModuleVisible(Imported, Import.ImportLoc);
4590 // FIXME: should we tell Sema to make the module visible too?
Richard Smitha7e2cc62015-05-01 01:53:09 +00004591 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004592 }
4593 ImportedModules.clear();
4594}
4595
4596void ASTReader::finalizeForWriting() {
Richard Smithde711422015-04-23 21:20:19 +00004597 // Nothing to do for now.
Guy Benyei11169dd2012-12-18 14:30:41 +00004598}
4599
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00004600/// Reads and return the signature record from \p PCH's control block, or
Peter Collingbourne77c89b62016-11-08 04:17:11 +00004601/// else returns 0.
4602static ASTFileSignature readASTFileSignature(StringRef PCH) {
4603 BitstreamCursor Stream(PCH);
Ben Langmuir70a1b812015-03-24 04:43:52 +00004604 if (!startsWithASTFileMagic(Stream))
Vedant Kumar48b4f762018-04-14 01:40:48 +00004605 return ASTFileSignature();
Ben Langmuir487ea142014-10-23 18:05:36 +00004606
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004607 // Scan for the UNHASHED_CONTROL_BLOCK_ID block.
4608 if (SkipCursorToBlock(Stream, UNHASHED_CONTROL_BLOCK_ID))
Vedant Kumar48b4f762018-04-14 01:40:48 +00004609 return ASTFileSignature();
Ben Langmuir487ea142014-10-23 18:05:36 +00004610
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004611 // Scan for SIGNATURE inside the diagnostic options block.
Ben Langmuir487ea142014-10-23 18:05:36 +00004612 ASTReader::RecordData Record;
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00004613 while (true) {
Ben Langmuir487ea142014-10-23 18:05:36 +00004614 llvm::BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
Simon Pilgrim0b33f112016-11-16 16:11:08 +00004615 if (Entry.Kind != llvm::BitstreamEntry::Record)
Vedant Kumar48b4f762018-04-14 01:40:48 +00004616 return ASTFileSignature();
Ben Langmuir487ea142014-10-23 18:05:36 +00004617
4618 Record.clear();
4619 StringRef Blob;
4620 if (SIGNATURE == Stream.readRecord(Entry.ID, Record, &Blob))
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004621 return {{{(uint32_t)Record[0], (uint32_t)Record[1], (uint32_t)Record[2],
4622 (uint32_t)Record[3], (uint32_t)Record[4]}}};
Ben Langmuir487ea142014-10-23 18:05:36 +00004623 }
4624}
4625
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00004626/// Retrieve the name of the original source file name
Guy Benyei11169dd2012-12-18 14:30:41 +00004627/// directly from the AST file, without actually loading the AST
4628/// file.
Adrian Prantlbb165fb2015-06-20 18:53:08 +00004629std::string ASTReader::getOriginalSourceFile(
4630 const std::string &ASTFileName, FileManager &FileMgr,
Adrian Prantlfb2398d2015-07-17 01:19:54 +00004631 const PCHContainerReader &PCHContainerRdr, DiagnosticsEngine &Diags) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004632 // Open the AST file.
Benjamin Kramera8857962014-10-26 22:44:13 +00004633 auto Buffer = FileMgr.getBufferForFile(ASTFileName);
Guy Benyei11169dd2012-12-18 14:30:41 +00004634 if (!Buffer) {
Benjamin Kramera8857962014-10-26 22:44:13 +00004635 Diags.Report(diag::err_fe_unable_to_read_pch_file)
4636 << ASTFileName << Buffer.getError().message();
Vedant Kumar48b4f762018-04-14 01:40:48 +00004637 return std::string();
Guy Benyei11169dd2012-12-18 14:30:41 +00004638 }
4639
4640 // Initialize the stream
Peter Collingbourne77c89b62016-11-08 04:17:11 +00004641 BitstreamCursor Stream(PCHContainerRdr.ExtractPCH(**Buffer));
Guy Benyei11169dd2012-12-18 14:30:41 +00004642
4643 // Sniff for the signature.
Ben Langmuir70a1b812015-03-24 04:43:52 +00004644 if (!startsWithASTFileMagic(Stream)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004645 Diags.Report(diag::err_fe_not_a_pch_file) << ASTFileName;
Vedant Kumar48b4f762018-04-14 01:40:48 +00004646 return std::string();
Guy Benyei11169dd2012-12-18 14:30:41 +00004647 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004648
Chris Lattnere7b154b2013-01-19 21:39:22 +00004649 // Scan for the CONTROL_BLOCK_ID block.
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004650 if (SkipCursorToBlock(Stream, CONTROL_BLOCK_ID)) {
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004651 Diags.Report(diag::err_fe_pch_malformed_block) << ASTFileName;
Vedant Kumar48b4f762018-04-14 01:40:48 +00004652 return std::string();
Chris Lattnere7b154b2013-01-19 21:39:22 +00004653 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004654
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004655 // Scan for ORIGINAL_FILE inside the control block.
4656 RecordData Record;
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00004657 while (true) {
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004658 llvm::BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
Chris Lattnere7b154b2013-01-19 21:39:22 +00004659 if (Entry.Kind == llvm::BitstreamEntry::EndBlock)
Vedant Kumar48b4f762018-04-14 01:40:48 +00004660 return std::string();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004661
Chris Lattnere7b154b2013-01-19 21:39:22 +00004662 if (Entry.Kind != llvm::BitstreamEntry::Record) {
4663 Diags.Report(diag::err_fe_pch_malformed_block) << ASTFileName;
Vedant Kumar48b4f762018-04-14 01:40:48 +00004664 return std::string();
Guy Benyei11169dd2012-12-18 14:30:41 +00004665 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004666
Guy Benyei11169dd2012-12-18 14:30:41 +00004667 Record.clear();
Chris Lattner0e6c9402013-01-20 02:38:54 +00004668 StringRef Blob;
4669 if (Stream.readRecord(Entry.ID, Record, &Blob) == ORIGINAL_FILE)
4670 return Blob.str();
Guy Benyei11169dd2012-12-18 14:30:41 +00004671 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004672}
4673
4674namespace {
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00004675
Guy Benyei11169dd2012-12-18 14:30:41 +00004676 class SimplePCHValidator : public ASTReaderListener {
4677 const LangOptions &ExistingLangOpts;
4678 const TargetOptions &ExistingTargetOpts;
4679 const PreprocessorOptions &ExistingPPOpts;
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +00004680 std::string ExistingModuleCachePath;
Guy Benyei11169dd2012-12-18 14:30:41 +00004681 FileManager &FileMgr;
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +00004682
Guy Benyei11169dd2012-12-18 14:30:41 +00004683 public:
4684 SimplePCHValidator(const LangOptions &ExistingLangOpts,
4685 const TargetOptions &ExistingTargetOpts,
4686 const PreprocessorOptions &ExistingPPOpts,
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +00004687 StringRef ExistingModuleCachePath,
Guy Benyei11169dd2012-12-18 14:30:41 +00004688 FileManager &FileMgr)
4689 : ExistingLangOpts(ExistingLangOpts),
4690 ExistingTargetOpts(ExistingTargetOpts),
4691 ExistingPPOpts(ExistingPPOpts),
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +00004692 ExistingModuleCachePath(ExistingModuleCachePath),
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00004693 FileMgr(FileMgr) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00004694
Richard Smith1e2cf0d2014-10-31 02:28:58 +00004695 bool ReadLanguageOptions(const LangOptions &LangOpts, bool Complain,
4696 bool AllowCompatibleDifferences) override {
4697 return checkLanguageOptions(ExistingLangOpts, LangOpts, nullptr,
4698 AllowCompatibleDifferences);
Guy Benyei11169dd2012-12-18 14:30:41 +00004699 }
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00004700
Chandler Carruth0d745bc2015-03-14 04:47:43 +00004701 bool ReadTargetOptions(const TargetOptions &TargetOpts, bool Complain,
4702 bool AllowCompatibleDifferences) override {
4703 return checkTargetOptions(ExistingTargetOpts, TargetOpts, nullptr,
4704 AllowCompatibleDifferences);
Guy Benyei11169dd2012-12-18 14:30:41 +00004705 }
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00004706
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +00004707 bool ReadHeaderSearchOptions(const HeaderSearchOptions &HSOpts,
4708 StringRef SpecificModuleCachePath,
4709 bool Complain) override {
4710 return checkHeaderSearchOptions(HSOpts, SpecificModuleCachePath,
4711 ExistingModuleCachePath,
4712 nullptr, ExistingLangOpts);
4713 }
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00004714
Craig Topper3e89dfe2014-03-13 02:13:41 +00004715 bool ReadPreprocessorOptions(const PreprocessorOptions &PPOpts,
4716 bool Complain,
4717 std::string &SuggestedPredefines) override {
Craig Toppera13603a2014-05-22 05:54:18 +00004718 return checkPreprocessorOptions(ExistingPPOpts, PPOpts, nullptr, FileMgr,
Argyrios Kyrtzidisd3afa0c2013-04-26 21:33:40 +00004719 SuggestedPredefines, ExistingLangOpts);
Guy Benyei11169dd2012-12-18 14:30:41 +00004720 }
4721 };
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00004722
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00004723} // namespace
Guy Benyei11169dd2012-12-18 14:30:41 +00004724
Adrian Prantlbb165fb2015-06-20 18:53:08 +00004725bool ASTReader::readASTFileControlBlock(
4726 StringRef Filename, FileManager &FileMgr,
Adrian Prantlfb2398d2015-07-17 01:19:54 +00004727 const PCHContainerReader &PCHContainerRdr,
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004728 bool FindModuleFileExtensions,
Manman Ren47a44452016-07-26 17:12:17 +00004729 ASTReaderListener &Listener, bool ValidateDiagnosticOptions) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004730 // Open the AST file.
Richard Smith7f330cd2015-03-18 01:42:29 +00004731 // FIXME: This allows use of the VFS; we do not allow use of the
4732 // VFS when actually loading a module.
Benjamin Kramera8857962014-10-26 22:44:13 +00004733 auto Buffer = FileMgr.getBufferForFile(Filename);
Guy Benyei11169dd2012-12-18 14:30:41 +00004734 if (!Buffer) {
4735 return true;
4736 }
4737
4738 // Initialize the stream
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004739 StringRef Bytes = PCHContainerRdr.ExtractPCH(**Buffer);
4740 BitstreamCursor Stream(Bytes);
Guy Benyei11169dd2012-12-18 14:30:41 +00004741
4742 // Sniff for the signature.
Ben Langmuir70a1b812015-03-24 04:43:52 +00004743 if (!startsWithASTFileMagic(Stream))
Guy Benyei11169dd2012-12-18 14:30:41 +00004744 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +00004745
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004746 // Scan for the CONTROL_BLOCK_ID block.
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004747 if (SkipCursorToBlock(Stream, CONTROL_BLOCK_ID))
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004748 return true;
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004749
4750 bool NeedsInputFiles = Listener.needsInputFileVisitation();
Ben Langmuircb69b572014-03-07 06:40:32 +00004751 bool NeedsSystemInputFiles = Listener.needsSystemInputFileVisitation();
Richard Smithd4b230b2014-10-27 23:01:16 +00004752 bool NeedsImports = Listener.needsImportVisitation();
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004753 BitstreamCursor InputFilesCursor;
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004754
Guy Benyei11169dd2012-12-18 14:30:41 +00004755 RecordData Record;
Richard Smith7ed1bc92014-12-05 22:42:13 +00004756 std::string ModuleDir;
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004757 bool DoneWithControlBlock = false;
4758 while (!DoneWithControlBlock) {
Richard Smith0516b182015-09-08 19:40:14 +00004759 llvm::BitstreamEntry Entry = Stream.advance();
4760
4761 switch (Entry.Kind) {
4762 case llvm::BitstreamEntry::SubBlock: {
4763 switch (Entry.ID) {
4764 case OPTIONS_BLOCK_ID: {
4765 std::string IgnoredSuggestedPredefines;
4766 if (ReadOptionsBlock(Stream, ARR_ConfigurationMismatch | ARR_OutOfDate,
4767 /*AllowCompatibleConfigurationMismatch*/ false,
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004768 Listener, IgnoredSuggestedPredefines) != Success)
Richard Smith0516b182015-09-08 19:40:14 +00004769 return true;
4770 break;
4771 }
4772
4773 case INPUT_FILES_BLOCK_ID:
4774 InputFilesCursor = Stream;
4775 if (Stream.SkipBlock() ||
4776 (NeedsInputFiles &&
4777 ReadBlockAbbrevs(InputFilesCursor, INPUT_FILES_BLOCK_ID)))
4778 return true;
4779 break;
4780
4781 default:
4782 if (Stream.SkipBlock())
4783 return true;
4784 break;
4785 }
4786
4787 continue;
4788 }
4789
4790 case llvm::BitstreamEntry::EndBlock:
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004791 DoneWithControlBlock = true;
4792 break;
Richard Smith0516b182015-09-08 19:40:14 +00004793
4794 case llvm::BitstreamEntry::Error:
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004795 return true;
Richard Smith0516b182015-09-08 19:40:14 +00004796
4797 case llvm::BitstreamEntry::Record:
4798 break;
4799 }
4800
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004801 if (DoneWithControlBlock) break;
4802
Guy Benyei11169dd2012-12-18 14:30:41 +00004803 Record.clear();
Chris Lattner0e6c9402013-01-20 02:38:54 +00004804 StringRef Blob;
4805 unsigned RecCode = Stream.readRecord(Entry.ID, Record, &Blob);
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004806 switch ((ControlRecordTypes)RecCode) {
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00004807 case METADATA:
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004808 if (Record[0] != VERSION_MAJOR)
4809 return true;
Douglas Gregorbf7fc9c2013-03-27 16:47:18 +00004810 if (Listener.ReadFullVersionInformation(Blob))
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004811 return true;
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004812 break;
Ben Langmuir4f5212a2014-04-14 22:12:44 +00004813 case MODULE_NAME:
4814 Listener.ReadModuleName(Blob);
4815 break;
Richard Smith7ed1bc92014-12-05 22:42:13 +00004816 case MODULE_DIRECTORY:
4817 ModuleDir = Blob;
4818 break;
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00004819 case MODULE_MAP_FILE: {
4820 unsigned Idx = 0;
Richard Smith7ed1bc92014-12-05 22:42:13 +00004821 auto Path = ReadString(Record, Idx);
4822 ResolveImportedPath(Path, ModuleDir);
4823 Listener.ReadModuleMapFile(Path);
Ben Langmuir4f5212a2014-04-14 22:12:44 +00004824 break;
Ben Langmuir4b8a9e92014-08-12 16:42:33 +00004825 }
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004826 case INPUT_FILE_OFFSETS: {
4827 if (!NeedsInputFiles)
4828 break;
4829
4830 unsigned NumInputFiles = Record[0];
4831 unsigned NumUserFiles = Record[1];
Raphael Isemanneb13d3d2018-05-23 09:02:40 +00004832 const llvm::support::unaligned_uint64_t *InputFileOffs =
4833 (const llvm::support::unaligned_uint64_t *)Blob.data();
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004834 for (unsigned I = 0; I != NumInputFiles; ++I) {
4835 // Go find this input file.
4836 bool isSystemFile = I >= NumUserFiles;
Ben Langmuircb69b572014-03-07 06:40:32 +00004837
4838 if (isSystemFile && !NeedsSystemInputFiles)
4839 break; // the rest are system input files
4840
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004841 BitstreamCursor &Cursor = InputFilesCursor;
4842 SavedStreamPosition SavedPosition(Cursor);
4843 Cursor.JumpToBit(InputFileOffs[I]);
4844
4845 unsigned Code = Cursor.ReadCode();
4846 RecordData Record;
4847 StringRef Blob;
4848 bool shouldContinue = false;
4849 switch ((InputFileRecordTypes)Cursor.readRecord(Code, Record, &Blob)) {
4850 case INPUT_FILE:
Vedant Kumar48b4f762018-04-14 01:40:48 +00004851 bool Overridden = static_cast<bool>(Record[3]);
Richard Smith7ed1bc92014-12-05 22:42:13 +00004852 std::string Filename = Blob;
4853 ResolveImportedPath(Filename, ModuleDir);
Richard Smith216a3bd2015-08-13 17:57:10 +00004854 shouldContinue = Listener.visitInputFile(
4855 Filename, isSystemFile, Overridden, /*IsExplicitModule*/false);
Argyrios Kyrtzidisc4cd2c42013-05-06 19:23:40 +00004856 break;
4857 }
4858 if (!shouldContinue)
4859 break;
4860 }
4861 break;
4862 }
4863
Richard Smithd4b230b2014-10-27 23:01:16 +00004864 case IMPORTS: {
4865 if (!NeedsImports)
4866 break;
4867
4868 unsigned Idx = 0, N = Record.size();
4869 while (Idx < N) {
4870 // Read information about the AST file.
Bruno Cardoso Lopes6fc8a562018-09-11 05:17:13 +00004871 Idx += 1+1+1+1+5; // Kind, ImportLoc, Size, ModTime, Signature
4872 std::string ModuleName = ReadString(Record, Idx);
Richard Smith7ed1bc92014-12-05 22:42:13 +00004873 std::string Filename = ReadString(Record, Idx);
4874 ResolveImportedPath(Filename, ModuleDir);
Bruno Cardoso Lopes6fc8a562018-09-11 05:17:13 +00004875 Listener.visitImport(ModuleName, Filename);
Richard Smithd4b230b2014-10-27 23:01:16 +00004876 }
4877 break;
4878 }
4879
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004880 default:
4881 // No other validation to perform.
4882 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00004883 }
4884 }
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004885
4886 // Look for module file extension blocks, if requested.
4887 if (FindModuleFileExtensions) {
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004888 BitstreamCursor SavedStream = Stream;
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004889 while (!SkipCursorToBlock(Stream, EXTENSION_BLOCK_ID)) {
4890 bool DoneWithExtensionBlock = false;
4891 while (!DoneWithExtensionBlock) {
4892 llvm::BitstreamEntry Entry = Stream.advance();
4893
4894 switch (Entry.Kind) {
4895 case llvm::BitstreamEntry::SubBlock:
4896 if (Stream.SkipBlock())
4897 return true;
4898
4899 continue;
4900
4901 case llvm::BitstreamEntry::EndBlock:
4902 DoneWithExtensionBlock = true;
4903 continue;
4904
4905 case llvm::BitstreamEntry::Error:
4906 return true;
4907
4908 case llvm::BitstreamEntry::Record:
4909 break;
4910 }
4911
4912 Record.clear();
4913 StringRef Blob;
4914 unsigned RecCode = Stream.readRecord(Entry.ID, Record, &Blob);
4915 switch (RecCode) {
4916 case EXTENSION_METADATA: {
4917 ModuleFileExtensionMetadata Metadata;
4918 if (parseModuleFileExtensionMetadata(Record, Blob, Metadata))
4919 return true;
4920
4921 Listener.readModuleFileExtension(Metadata);
4922 break;
4923 }
4924 }
4925 }
4926 }
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004927 Stream = SavedStream;
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004928 }
4929
Duncan P. N. Exon Smith60fa2882017-03-13 18:45:08 +00004930 // Scan for the UNHASHED_CONTROL_BLOCK_ID block.
4931 if (readUnhashedControlBlockImpl(
4932 nullptr, Bytes, ARR_ConfigurationMismatch | ARR_OutOfDate,
4933 /*AllowCompatibleConfigurationMismatch*/ false, &Listener,
4934 ValidateDiagnosticOptions) != Success)
4935 return true;
4936
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004937 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +00004938}
4939
Benjamin Kramerf6021ec2017-03-21 21:35:04 +00004940bool ASTReader::isAcceptableASTFile(StringRef Filename, FileManager &FileMgr,
4941 const PCHContainerReader &PCHContainerRdr,
4942 const LangOptions &LangOpts,
4943 const TargetOptions &TargetOpts,
4944 const PreprocessorOptions &PPOpts,
4945 StringRef ExistingModuleCachePath) {
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +00004946 SimplePCHValidator validator(LangOpts, TargetOpts, PPOpts,
4947 ExistingModuleCachePath, FileMgr);
Adrian Prantlfb2398d2015-07-17 01:19:54 +00004948 return !readASTFileControlBlock(Filename, FileMgr, PCHContainerRdr,
Douglas Gregor6623e1f2015-11-03 18:33:07 +00004949 /*FindModuleFileExtensions=*/false,
Manman Ren47a44452016-07-26 17:12:17 +00004950 validator,
4951 /*ValidateDiagnosticOptions=*/true);
Guy Benyei11169dd2012-12-18 14:30:41 +00004952}
4953
Ben Langmuir2c9af442014-04-10 17:57:43 +00004954ASTReader::ASTReadResult
4955ASTReader::ReadSubmoduleBlock(ModuleFile &F, unsigned ClientLoadCapabilities) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004956 // Enter the submodule block.
4957 if (F.Stream.EnterSubBlock(SUBMODULE_BLOCK_ID)) {
4958 Error("malformed submodule block record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00004959 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00004960 }
4961
4962 ModuleMap &ModMap = PP.getHeaderSearchInfo().getModuleMap();
4963 bool First = true;
Craig Toppera13603a2014-05-22 05:54:18 +00004964 Module *CurrentModule = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00004965 RecordData Record;
4966 while (true) {
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004967 llvm::BitstreamEntry Entry = F.Stream.advanceSkippingSubblocks();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00004968
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004969 switch (Entry.Kind) {
4970 case llvm::BitstreamEntry::SubBlock: // Handled for us already.
4971 case llvm::BitstreamEntry::Error:
4972 Error("malformed block record in AST file");
Ben Langmuir2c9af442014-04-10 17:57:43 +00004973 return Failure;
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004974 case llvm::BitstreamEntry::EndBlock:
Ben Langmuir2c9af442014-04-10 17:57:43 +00004975 return Success;
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004976 case llvm::BitstreamEntry::Record:
4977 // The interesting case.
4978 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00004979 }
Chris Lattner7fb3bef2013-01-20 00:56:42 +00004980
Guy Benyei11169dd2012-12-18 14:30:41 +00004981 // Read a record.
Chris Lattner0e6c9402013-01-20 02:38:54 +00004982 StringRef Blob;
Guy Benyei11169dd2012-12-18 14:30:41 +00004983 Record.clear();
Richard Smith03478d92014-10-23 22:12:14 +00004984 auto Kind = F.Stream.readRecord(Entry.ID, Record, &Blob);
4985
4986 if ((Kind == SUBMODULE_METADATA) != First) {
4987 Error("submodule metadata record should be at beginning of block");
4988 return Failure;
4989 }
4990 First = false;
4991
4992 // Submodule information is only valid if we have a current module.
4993 // FIXME: Should we error on these cases?
4994 if (!CurrentModule && Kind != SUBMODULE_METADATA &&
4995 Kind != SUBMODULE_DEFINITION)
4996 continue;
4997
4998 switch (Kind) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004999 default: // Default behavior: ignore.
5000 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00005001
Richard Smith03478d92014-10-23 22:12:14 +00005002 case SUBMODULE_DEFINITION: {
Jordan Rose90b0a1f2018-04-20 17:16:04 +00005003 if (Record.size() < 12) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005004 Error("malformed module definition");
Ben Langmuir2c9af442014-04-10 17:57:43 +00005005 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00005006 }
Richard Smith03478d92014-10-23 22:12:14 +00005007
Chris Lattner0e6c9402013-01-20 02:38:54 +00005008 StringRef Name = Blob;
Richard Smith9bca2982014-03-08 00:03:56 +00005009 unsigned Idx = 0;
5010 SubmoduleID GlobalID = getGlobalSubmoduleID(F, Record[Idx++]);
5011 SubmoduleID Parent = getGlobalSubmoduleID(F, Record[Idx++]);
Vedant Kumar48b4f762018-04-14 01:40:48 +00005012 Module::ModuleKind Kind = (Module::ModuleKind)Record[Idx++];
Richard Smith9bca2982014-03-08 00:03:56 +00005013 bool IsFramework = Record[Idx++];
5014 bool IsExplicit = Record[Idx++];
5015 bool IsSystem = Record[Idx++];
5016 bool IsExternC = Record[Idx++];
5017 bool InferSubmodules = Record[Idx++];
5018 bool InferExplicitSubmodules = Record[Idx++];
5019 bool InferExportWildcard = Record[Idx++];
5020 bool ConfigMacrosExhaustive = Record[Idx++];
Jordan Rose90b0a1f2018-04-20 17:16:04 +00005021 bool ModuleMapIsPrivate = Record[Idx++];
Douglas Gregor8d932422013-03-20 03:59:18 +00005022
Ben Langmuirbeee15e2014-04-14 18:00:01 +00005023 Module *ParentModule = nullptr;
Ben Langmuir9d6448b2014-08-09 00:57:23 +00005024 if (Parent)
Guy Benyei11169dd2012-12-18 14:30:41 +00005025 ParentModule = getSubmodule(Parent);
Ben Langmuirbeee15e2014-04-14 18:00:01 +00005026
Guy Benyei11169dd2012-12-18 14:30:41 +00005027 // Retrieve this (sub)module from the module map, creating it if
5028 // necessary.
David Blaikie9ffe5a32017-01-30 05:00:26 +00005029 CurrentModule =
5030 ModMap.findOrCreateModule(Name, ParentModule, IsFramework, IsExplicit)
5031 .first;
Ben Langmuir9d6448b2014-08-09 00:57:23 +00005032
5033 // FIXME: set the definition loc for CurrentModule, or call
5034 // ModMap.setInferredModuleAllowedBy()
5035
Guy Benyei11169dd2012-12-18 14:30:41 +00005036 SubmoduleID GlobalIndex = GlobalID - NUM_PREDEF_SUBMODULE_IDS;
5037 if (GlobalIndex >= SubmodulesLoaded.size() ||
5038 SubmodulesLoaded[GlobalIndex]) {
5039 Error("too many submodules");
Ben Langmuir2c9af442014-04-10 17:57:43 +00005040 return Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00005041 }
Douglas Gregor8a114ab2013-02-06 22:40:31 +00005042
Douglas Gregor7029ce12013-03-19 00:28:20 +00005043 if (!ParentModule) {
5044 if (const FileEntry *CurFile = CurrentModule->getASTFile()) {
Yuka Takahashid8baec22018-08-01 09:50:02 +00005045 // Don't emit module relocation error if we have -fno-validate-pch
5046 if (!PP.getPreprocessorOpts().DisablePCHValidation &&
5047 CurFile != F.File) {
Douglas Gregor7029ce12013-03-19 00:28:20 +00005048 if (!Diags.isDiagnosticInFlight()) {
5049 Diag(diag::err_module_file_conflict)
5050 << CurrentModule->getTopLevelModuleName()
5051 << CurFile->getName()
5052 << F.File->getName();
5053 }
Ben Langmuir2c9af442014-04-10 17:57:43 +00005054 return Failure;
Douglas Gregor8a114ab2013-02-06 22:40:31 +00005055 }
Douglas Gregor8a114ab2013-02-06 22:40:31 +00005056 }
Douglas Gregor7029ce12013-03-19 00:28:20 +00005057
5058 CurrentModule->setASTFile(F.File);
Richard Smithab755972017-06-05 18:10:11 +00005059 CurrentModule->PresumedModuleMapFile = F.ModuleMapPath;
Douglas Gregor8a114ab2013-02-06 22:40:31 +00005060 }
Ben Langmuirbeee15e2014-04-14 18:00:01 +00005061
Richard Smithdd8b5332017-09-04 05:37:53 +00005062 CurrentModule->Kind = Kind;
Adrian Prantl15bcf702015-06-30 17:39:43 +00005063 CurrentModule->Signature = F.Signature;
Guy Benyei11169dd2012-12-18 14:30:41 +00005064 CurrentModule->IsFromModuleFile = true;
5065 CurrentModule->IsSystem = IsSystem || CurrentModule->IsSystem;
Richard Smith9bca2982014-03-08 00:03:56 +00005066 CurrentModule->IsExternC = IsExternC;
Guy Benyei11169dd2012-12-18 14:30:41 +00005067 CurrentModule->InferSubmodules = InferSubmodules;
5068 CurrentModule->InferExplicitSubmodules = InferExplicitSubmodules;
5069 CurrentModule->InferExportWildcard = InferExportWildcard;
Douglas Gregor8d932422013-03-20 03:59:18 +00005070 CurrentModule->ConfigMacrosExhaustive = ConfigMacrosExhaustive;
Jordan Rose90b0a1f2018-04-20 17:16:04 +00005071 CurrentModule->ModuleMapIsPrivate = ModuleMapIsPrivate;
Guy Benyei11169dd2012-12-18 14:30:41 +00005072 if (DeserializationListener)
5073 DeserializationListener->ModuleRead(GlobalID, CurrentModule);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005074
Guy Benyei11169dd2012-12-18 14:30:41 +00005075 SubmodulesLoaded[GlobalIndex] = CurrentModule;
Douglas Gregor6ddfca92013-01-14 17:21:00 +00005076
Richard Smith8a3e39a2016-03-28 21:31:09 +00005077 // Clear out data that will be replaced by what is in the module file.
Douglas Gregor6ddfca92013-01-14 17:21:00 +00005078 CurrentModule->LinkLibraries.clear();
Douglas Gregor8d932422013-03-20 03:59:18 +00005079 CurrentModule->ConfigMacros.clear();
Douglas Gregorfb912652013-03-20 21:10:35 +00005080 CurrentModule->UnresolvedConflicts.clear();
5081 CurrentModule->Conflicts.clear();
Richard Smith8a3e39a2016-03-28 21:31:09 +00005082
5083 // The module is available unless it's missing a requirement; relevant
5084 // requirements will be (re-)added by SUBMODULE_REQUIRES records.
5085 // Missing headers that were present when the module was built do not
5086 // make it unavailable -- if we got this far, this must be an explicitly
5087 // imported module file.
5088 CurrentModule->Requirements.clear();
5089 CurrentModule->MissingHeaders.clear();
5090 CurrentModule->IsMissingRequirement =
5091 ParentModule && ParentModule->IsMissingRequirement;
5092 CurrentModule->IsAvailable = !CurrentModule->IsMissingRequirement;
Guy Benyei11169dd2012-12-18 14:30:41 +00005093 break;
5094 }
Richard Smith8a3e39a2016-03-28 21:31:09 +00005095
Guy Benyei11169dd2012-12-18 14:30:41 +00005096 case SUBMODULE_UMBRELLA_HEADER: {
Richard Smith2b63d152015-05-16 02:28:53 +00005097 std::string Filename = Blob;
5098 ResolveImportedPath(F, Filename);
5099 if (auto *Umbrella = PP.getFileManager().getFile(Filename)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005100 if (!CurrentModule->getUmbrellaHeader())
Richard Smith2b63d152015-05-16 02:28:53 +00005101 ModMap.setUmbrellaHeader(CurrentModule, Umbrella, Blob);
5102 else if (CurrentModule->getUmbrellaHeader().Entry != Umbrella) {
Bruno Cardoso Lopes573b13f2017-03-22 00:11:21 +00005103 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0)
5104 Error("mismatched umbrella headers in submodule");
5105 return OutOfDate;
Guy Benyei11169dd2012-12-18 14:30:41 +00005106 }
5107 }
5108 break;
5109 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005110
Richard Smith202210b2014-10-24 20:23:01 +00005111 case SUBMODULE_HEADER:
5112 case SUBMODULE_EXCLUDED_HEADER:
5113 case SUBMODULE_PRIVATE_HEADER:
5114 // We lazily associate headers with their modules via the HeaderInfo table.
Argyrios Kyrtzidisb146baa2013-03-13 21:13:51 +00005115 // FIXME: Re-evaluate this section; maybe only store InputFile IDs instead
5116 // of complete filenames or remove it entirely.
Richard Smith202210b2014-10-24 20:23:01 +00005117 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00005118
Richard Smith202210b2014-10-24 20:23:01 +00005119 case SUBMODULE_TEXTUAL_HEADER:
5120 case SUBMODULE_PRIVATE_TEXTUAL_HEADER:
5121 // FIXME: Textual headers are not marked in the HeaderInfo table. Load
5122 // them here.
5123 break;
Lawrence Crowlb53e5482013-06-20 21:14:14 +00005124
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005125 case SUBMODULE_TOPHEADER:
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00005126 CurrentModule->addTopHeaderFilename(Blob);
Guy Benyei11169dd2012-12-18 14:30:41 +00005127 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00005128
5129 case SUBMODULE_UMBRELLA_DIR: {
Richard Smith2b63d152015-05-16 02:28:53 +00005130 std::string Dirname = Blob;
5131 ResolveImportedPath(F, Dirname);
5132 if (auto *Umbrella = PP.getFileManager().getDirectory(Dirname)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005133 if (!CurrentModule->getUmbrellaDir())
Richard Smith2b63d152015-05-16 02:28:53 +00005134 ModMap.setUmbrellaDir(CurrentModule, Umbrella, Blob);
5135 else if (CurrentModule->getUmbrellaDir().Entry != Umbrella) {
Ben Langmuir2c9af442014-04-10 17:57:43 +00005136 if ((ClientLoadCapabilities & ARR_OutOfDate) == 0)
5137 Error("mismatched umbrella directories in submodule");
5138 return OutOfDate;
Guy Benyei11169dd2012-12-18 14:30:41 +00005139 }
5140 }
5141 break;
5142 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005143
Guy Benyei11169dd2012-12-18 14:30:41 +00005144 case SUBMODULE_METADATA: {
Guy Benyei11169dd2012-12-18 14:30:41 +00005145 F.BaseSubmoduleID = getTotalNumSubmodules();
5146 F.LocalNumSubmodules = Record[0];
5147 unsigned LocalBaseSubmoduleID = Record[1];
5148 if (F.LocalNumSubmodules > 0) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005149 // Introduce the global -> local mapping for submodules within this
Guy Benyei11169dd2012-12-18 14:30:41 +00005150 // module.
5151 GlobalSubmoduleMap.insert(std::make_pair(getTotalNumSubmodules()+1,&F));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005152
5153 // Introduce the local -> global mapping for submodules within this
Guy Benyei11169dd2012-12-18 14:30:41 +00005154 // module.
5155 F.SubmoduleRemap.insertOrReplace(
5156 std::make_pair(LocalBaseSubmoduleID,
5157 F.BaseSubmoduleID - LocalBaseSubmoduleID));
Ben Langmuirfe971d92014-08-16 04:54:18 +00005158
Ben Langmuir52ca6782014-10-20 16:27:32 +00005159 SubmodulesLoaded.resize(SubmodulesLoaded.size() + F.LocalNumSubmodules);
5160 }
Guy Benyei11169dd2012-12-18 14:30:41 +00005161 break;
5162 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005163
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005164 case SUBMODULE_IMPORTS:
Guy Benyei11169dd2012-12-18 14:30:41 +00005165 for (unsigned Idx = 0; Idx != Record.size(); ++Idx) {
Douglas Gregorfb912652013-03-20 21:10:35 +00005166 UnresolvedModuleRef Unresolved;
Guy Benyei11169dd2012-12-18 14:30:41 +00005167 Unresolved.File = &F;
5168 Unresolved.Mod = CurrentModule;
5169 Unresolved.ID = Record[Idx];
Douglas Gregorfb912652013-03-20 21:10:35 +00005170 Unresolved.Kind = UnresolvedModuleRef::Import;
Guy Benyei11169dd2012-12-18 14:30:41 +00005171 Unresolved.IsWildcard = false;
Douglas Gregorfb912652013-03-20 21:10:35 +00005172 UnresolvedModuleRefs.push_back(Unresolved);
Guy Benyei11169dd2012-12-18 14:30:41 +00005173 }
5174 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00005175
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005176 case SUBMODULE_EXPORTS:
Guy Benyei11169dd2012-12-18 14:30:41 +00005177 for (unsigned Idx = 0; Idx + 1 < Record.size(); Idx += 2) {
Douglas Gregorfb912652013-03-20 21:10:35 +00005178 UnresolvedModuleRef Unresolved;
Guy Benyei11169dd2012-12-18 14:30:41 +00005179 Unresolved.File = &F;
5180 Unresolved.Mod = CurrentModule;
5181 Unresolved.ID = Record[Idx];
Douglas Gregorfb912652013-03-20 21:10:35 +00005182 Unresolved.Kind = UnresolvedModuleRef::Export;
Guy Benyei11169dd2012-12-18 14:30:41 +00005183 Unresolved.IsWildcard = Record[Idx + 1];
Douglas Gregorfb912652013-03-20 21:10:35 +00005184 UnresolvedModuleRefs.push_back(Unresolved);
Guy Benyei11169dd2012-12-18 14:30:41 +00005185 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005186
5187 // Once we've loaded the set of exports, there's no reason to keep
Guy Benyei11169dd2012-12-18 14:30:41 +00005188 // the parsed, unresolved exports around.
5189 CurrentModule->UnresolvedExports.clear();
5190 break;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005191
5192 case SUBMODULE_REQUIRES:
Richard Smithdbafb6c2017-06-29 23:23:46 +00005193 CurrentModule->addRequirement(Blob, Record[0], PP.getLangOpts(),
5194 PP.getTargetInfo());
Guy Benyei11169dd2012-12-18 14:30:41 +00005195 break;
Douglas Gregor6ddfca92013-01-14 17:21:00 +00005196
5197 case SUBMODULE_LINK_LIBRARY:
Bruno Cardoso Lopesa3b5f712018-04-16 19:42:32 +00005198 ModMap.resolveLinkAsDependencies(CurrentModule);
Douglas Gregor6ddfca92013-01-14 17:21:00 +00005199 CurrentModule->LinkLibraries.push_back(
Chris Lattner0e6c9402013-01-20 02:38:54 +00005200 Module::LinkLibrary(Blob, Record[0]));
Douglas Gregor6ddfca92013-01-14 17:21:00 +00005201 break;
Douglas Gregor35b13ec2013-03-20 00:22:05 +00005202
5203 case SUBMODULE_CONFIG_MACRO:
Douglas Gregor35b13ec2013-03-20 00:22:05 +00005204 CurrentModule->ConfigMacros.push_back(Blob.str());
5205 break;
Douglas Gregorfb912652013-03-20 21:10:35 +00005206
5207 case SUBMODULE_CONFLICT: {
Douglas Gregorfb912652013-03-20 21:10:35 +00005208 UnresolvedModuleRef Unresolved;
5209 Unresolved.File = &F;
5210 Unresolved.Mod = CurrentModule;
5211 Unresolved.ID = Record[0];
5212 Unresolved.Kind = UnresolvedModuleRef::Conflict;
5213 Unresolved.IsWildcard = false;
5214 Unresolved.String = Blob;
5215 UnresolvedModuleRefs.push_back(Unresolved);
5216 break;
5217 }
Richard Smithdc1f0422016-07-20 19:10:16 +00005218
Douglas Gregorf0b11de2017-09-14 23:38:44 +00005219 case SUBMODULE_INITIALIZERS: {
Richard Smithdbafb6c2017-06-29 23:23:46 +00005220 if (!ContextObj)
5221 break;
Richard Smithdc1f0422016-07-20 19:10:16 +00005222 SmallVector<uint32_t, 16> Inits;
5223 for (auto &ID : Record)
5224 Inits.push_back(getGlobalDeclID(F, ID));
Richard Smithdbafb6c2017-06-29 23:23:46 +00005225 ContextObj->addLazyModuleInitializers(CurrentModule, Inits);
Richard Smithdc1f0422016-07-20 19:10:16 +00005226 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00005227 }
Douglas Gregorf0b11de2017-09-14 23:38:44 +00005228
5229 case SUBMODULE_EXPORT_AS:
5230 CurrentModule->ExportAsModule = Blob.str();
Bruno Cardoso Lopesa3b5f712018-04-16 19:42:32 +00005231 ModMap.addLinkAsDependency(CurrentModule);
Douglas Gregorf0b11de2017-09-14 23:38:44 +00005232 break;
5233 }
Guy Benyei11169dd2012-12-18 14:30:41 +00005234 }
5235}
5236
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00005237/// Parse the record that corresponds to a LangOptions data
Guy Benyei11169dd2012-12-18 14:30:41 +00005238/// structure.
5239///
5240/// This routine parses the language options from the AST file and then gives
5241/// them to the AST listener if one is set.
5242///
5243/// \returns true if the listener deems the file unacceptable, false otherwise.
5244bool ASTReader::ParseLanguageOptions(const RecordData &Record,
5245 bool Complain,
Richard Smith1e2cf0d2014-10-31 02:28:58 +00005246 ASTReaderListener &Listener,
5247 bool AllowCompatibleDifferences) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005248 LangOptions LangOpts;
5249 unsigned Idx = 0;
5250#define LANGOPT(Name, Bits, Default, Description) \
5251 LangOpts.Name = Record[Idx++];
5252#define ENUM_LANGOPT(Name, Type, Bits, Default, Description) \
5253 LangOpts.set##Name(static_cast<LangOptions::Type>(Record[Idx++]));
5254#include "clang/Basic/LangOptions.def"
Alexey Samsonovedf99a92014-11-07 22:29:38 +00005255#define SANITIZER(NAME, ID) \
5256 LangOpts.Sanitize.set(SanitizerKind::ID, Record[Idx++]);
Will Dietzf54319c2013-01-18 11:30:38 +00005257#include "clang/Basic/Sanitizers.def"
Guy Benyei11169dd2012-12-18 14:30:41 +00005258
Ben Langmuircd98cb72015-06-23 18:20:18 +00005259 for (unsigned N = Record[Idx++]; N; --N)
5260 LangOpts.ModuleFeatures.push_back(ReadString(Record, Idx));
5261
Vedant Kumar48b4f762018-04-14 01:40:48 +00005262 ObjCRuntime::Kind runtimeKind = (ObjCRuntime::Kind) Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00005263 VersionTuple runtimeVersion = ReadVersionTuple(Record, Idx);
5264 LangOpts.ObjCRuntime = ObjCRuntime(runtimeKind, runtimeVersion);
Dmitri Gribenkoacf2e782013-02-22 14:21:27 +00005265
Ben Langmuird4a667a2015-06-23 18:20:23 +00005266 LangOpts.CurrentModule = ReadString(Record, Idx);
Dmitri Gribenkoacf2e782013-02-22 14:21:27 +00005267
5268 // Comment options.
5269 for (unsigned N = Record[Idx++]; N; --N) {
5270 LangOpts.CommentOpts.BlockCommandNames.push_back(
5271 ReadString(Record, Idx));
5272 }
Dmitri Gribenkoa7d16ce2013-04-10 15:35:17 +00005273 LangOpts.CommentOpts.ParseAllComments = Record[Idx++];
Dmitri Gribenkoacf2e782013-02-22 14:21:27 +00005274
Samuel Antaoee8fb302016-01-06 13:42:12 +00005275 // OpenMP offloading options.
5276 for (unsigned N = Record[Idx++]; N; --N) {
5277 LangOpts.OMPTargetTriples.push_back(llvm::Triple(ReadString(Record, Idx)));
5278 }
5279
5280 LangOpts.OMPHostIRFile = ReadString(Record, Idx);
5281
Richard Smith1e2cf0d2014-10-31 02:28:58 +00005282 return Listener.ReadLanguageOptions(LangOpts, Complain,
5283 AllowCompatibleDifferences);
Guy Benyei11169dd2012-12-18 14:30:41 +00005284}
5285
Chandler Carruth0d745bc2015-03-14 04:47:43 +00005286bool ASTReader::ParseTargetOptions(const RecordData &Record, bool Complain,
5287 ASTReaderListener &Listener,
5288 bool AllowCompatibleDifferences) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005289 unsigned Idx = 0;
5290 TargetOptions TargetOpts;
5291 TargetOpts.Triple = ReadString(Record, Idx);
5292 TargetOpts.CPU = ReadString(Record, Idx);
5293 TargetOpts.ABI = ReadString(Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00005294 for (unsigned N = Record[Idx++]; N; --N) {
5295 TargetOpts.FeaturesAsWritten.push_back(ReadString(Record, Idx));
5296 }
5297 for (unsigned N = Record[Idx++]; N; --N) {
5298 TargetOpts.Features.push_back(ReadString(Record, Idx));
5299 }
5300
Chandler Carruth0d745bc2015-03-14 04:47:43 +00005301 return Listener.ReadTargetOptions(TargetOpts, Complain,
5302 AllowCompatibleDifferences);
Guy Benyei11169dd2012-12-18 14:30:41 +00005303}
5304
5305bool ASTReader::ParseDiagnosticOptions(const RecordData &Record, bool Complain,
5306 ASTReaderListener &Listener) {
Ben Langmuirb92de022014-04-29 16:25:26 +00005307 IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts(new DiagnosticOptions);
Guy Benyei11169dd2012-12-18 14:30:41 +00005308 unsigned Idx = 0;
Ben Langmuirb92de022014-04-29 16:25:26 +00005309#define DIAGOPT(Name, Bits, Default) DiagOpts->Name = Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00005310#define ENUM_DIAGOPT(Name, Type, Bits, Default) \
Ben Langmuirb92de022014-04-29 16:25:26 +00005311 DiagOpts->set##Name(static_cast<Type>(Record[Idx++]));
Guy Benyei11169dd2012-12-18 14:30:41 +00005312#include "clang/Basic/DiagnosticOptions.def"
5313
Richard Smith3be1cb22014-08-07 00:24:21 +00005314 for (unsigned N = Record[Idx++]; N; --N)
Ben Langmuirb92de022014-04-29 16:25:26 +00005315 DiagOpts->Warnings.push_back(ReadString(Record, Idx));
Richard Smith3be1cb22014-08-07 00:24:21 +00005316 for (unsigned N = Record[Idx++]; N; --N)
5317 DiagOpts->Remarks.push_back(ReadString(Record, Idx));
Guy Benyei11169dd2012-12-18 14:30:41 +00005318
5319 return Listener.ReadDiagnosticOptions(DiagOpts, Complain);
5320}
5321
5322bool ASTReader::ParseFileSystemOptions(const RecordData &Record, bool Complain,
5323 ASTReaderListener &Listener) {
5324 FileSystemOptions FSOpts;
5325 unsigned Idx = 0;
5326 FSOpts.WorkingDir = ReadString(Record, Idx);
5327 return Listener.ReadFileSystemOptions(FSOpts, Complain);
5328}
5329
5330bool ASTReader::ParseHeaderSearchOptions(const RecordData &Record,
5331 bool Complain,
5332 ASTReaderListener &Listener) {
5333 HeaderSearchOptions HSOpts;
5334 unsigned Idx = 0;
5335 HSOpts.Sysroot = ReadString(Record, Idx);
5336
5337 // Include entries.
5338 for (unsigned N = Record[Idx++]; N; --N) {
5339 std::string Path = ReadString(Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00005340 frontend::IncludeDirGroup Group
5341 = static_cast<frontend::IncludeDirGroup>(Record[Idx++]);
Guy Benyei11169dd2012-12-18 14:30:41 +00005342 bool IsFramework = Record[Idx++];
5343 bool IgnoreSysRoot = Record[Idx++];
Benjamin Kramer3204b152015-05-29 19:42:19 +00005344 HSOpts.UserEntries.emplace_back(std::move(Path), Group, IsFramework,
5345 IgnoreSysRoot);
Guy Benyei11169dd2012-12-18 14:30:41 +00005346 }
5347
5348 // System header prefixes.
5349 for (unsigned N = Record[Idx++]; N; --N) {
5350 std::string Prefix = ReadString(Record, Idx);
5351 bool IsSystemHeader = Record[Idx++];
Benjamin Kramer3204b152015-05-29 19:42:19 +00005352 HSOpts.SystemHeaderPrefixes.emplace_back(std::move(Prefix), IsSystemHeader);
Guy Benyei11169dd2012-12-18 14:30:41 +00005353 }
5354
5355 HSOpts.ResourceDir = ReadString(Record, Idx);
5356 HSOpts.ModuleCachePath = ReadString(Record, Idx);
Argyrios Kyrtzidis1594c152014-03-03 08:12:05 +00005357 HSOpts.ModuleUserBuildPath = ReadString(Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00005358 HSOpts.DisableModuleHash = Record[Idx++];
Richard Smith18934752017-06-06 00:32:01 +00005359 HSOpts.ImplicitModuleMaps = Record[Idx++];
5360 HSOpts.ModuleMapFileHomeIsCwd = Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00005361 HSOpts.UseBuiltinIncludes = Record[Idx++];
5362 HSOpts.UseStandardSystemIncludes = Record[Idx++];
5363 HSOpts.UseStandardCXXIncludes = Record[Idx++];
5364 HSOpts.UseLibcxx = Record[Idx++];
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +00005365 std::string SpecificModuleCachePath = ReadString(Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00005366
Argyrios Kyrtzidisbd0b6512015-02-19 20:12:20 +00005367 return Listener.ReadHeaderSearchOptions(HSOpts, SpecificModuleCachePath,
5368 Complain);
Guy Benyei11169dd2012-12-18 14:30:41 +00005369}
5370
5371bool ASTReader::ParsePreprocessorOptions(const RecordData &Record,
5372 bool Complain,
5373 ASTReaderListener &Listener,
5374 std::string &SuggestedPredefines) {
5375 PreprocessorOptions PPOpts;
5376 unsigned Idx = 0;
5377
5378 // Macro definitions/undefs
5379 for (unsigned N = Record[Idx++]; N; --N) {
5380 std::string Macro = ReadString(Record, Idx);
5381 bool IsUndef = Record[Idx++];
5382 PPOpts.Macros.push_back(std::make_pair(Macro, IsUndef));
5383 }
5384
5385 // Includes
5386 for (unsigned N = Record[Idx++]; N; --N) {
5387 PPOpts.Includes.push_back(ReadString(Record, Idx));
5388 }
5389
5390 // Macro Includes
5391 for (unsigned N = Record[Idx++]; N; --N) {
5392 PPOpts.MacroIncludes.push_back(ReadString(Record, Idx));
5393 }
5394
5395 PPOpts.UsePredefines = Record[Idx++];
Argyrios Kyrtzidisd3afa0c2013-04-26 21:33:40 +00005396 PPOpts.DetailedRecord = Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00005397 PPOpts.ImplicitPCHInclude = ReadString(Record, Idx);
5398 PPOpts.ImplicitPTHInclude = ReadString(Record, Idx);
5399 PPOpts.ObjCXXARCStandardLibrary =
5400 static_cast<ObjCXXARCStandardLibraryKind>(Record[Idx++]);
5401 SuggestedPredefines.clear();
5402 return Listener.ReadPreprocessorOptions(PPOpts, Complain,
5403 SuggestedPredefines);
5404}
5405
5406std::pair<ModuleFile *, unsigned>
5407ASTReader::getModulePreprocessedEntity(unsigned GlobalIndex) {
5408 GlobalPreprocessedEntityMapType::iterator
5409 I = GlobalPreprocessedEntityMap.find(GlobalIndex);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005410 assert(I != GlobalPreprocessedEntityMap.end() &&
Guy Benyei11169dd2012-12-18 14:30:41 +00005411 "Corrupted global preprocessed entity map");
5412 ModuleFile *M = I->second;
5413 unsigned LocalIndex = GlobalIndex - M->BasePreprocessedEntityID;
5414 return std::make_pair(M, LocalIndex);
5415}
5416
Benjamin Kramerb4ef6682015-02-06 17:25:10 +00005417llvm::iterator_range<PreprocessingRecord::iterator>
Guy Benyei11169dd2012-12-18 14:30:41 +00005418ASTReader::getModulePreprocessedEntities(ModuleFile &Mod) const {
5419 if (PreprocessingRecord *PPRec = PP.getPreprocessingRecord())
5420 return PPRec->getIteratorsForLoadedRange(Mod.BasePreprocessedEntityID,
5421 Mod.NumPreprocessedEntities);
5422
Benjamin Kramerb4ef6682015-02-06 17:25:10 +00005423 return llvm::make_range(PreprocessingRecord::iterator(),
5424 PreprocessingRecord::iterator());
Guy Benyei11169dd2012-12-18 14:30:41 +00005425}
5426
Benjamin Kramerb4ef6682015-02-06 17:25:10 +00005427llvm::iterator_range<ASTReader::ModuleDeclIterator>
Guy Benyei11169dd2012-12-18 14:30:41 +00005428ASTReader::getModuleFileLevelDecls(ModuleFile &Mod) {
Benjamin Kramerb4ef6682015-02-06 17:25:10 +00005429 return llvm::make_range(
5430 ModuleDeclIterator(this, &Mod, Mod.FileSortedDecls),
5431 ModuleDeclIterator(this, &Mod,
5432 Mod.FileSortedDecls + Mod.NumFileSortedDecls));
Guy Benyei11169dd2012-12-18 14:30:41 +00005433}
5434
Cameron Desrochersb60f1b62018-01-15 19:14:16 +00005435SourceRange ASTReader::ReadSkippedRange(unsigned GlobalIndex) {
5436 auto I = GlobalSkippedRangeMap.find(GlobalIndex);
5437 assert(I != GlobalSkippedRangeMap.end() &&
5438 "Corrupted global skipped range map");
5439 ModuleFile *M = I->second;
5440 unsigned LocalIndex = GlobalIndex - M->BasePreprocessedSkippedRangeID;
5441 assert(LocalIndex < M->NumPreprocessedSkippedRanges);
5442 PPSkippedRange RawRange = M->PreprocessedSkippedRangeOffsets[LocalIndex];
5443 SourceRange Range(TranslateSourceLocation(*M, RawRange.getBegin()),
5444 TranslateSourceLocation(*M, RawRange.getEnd()));
5445 assert(Range.isValid());
5446 return Range;
5447}
5448
Guy Benyei11169dd2012-12-18 14:30:41 +00005449PreprocessedEntity *ASTReader::ReadPreprocessedEntity(unsigned Index) {
5450 PreprocessedEntityID PPID = Index+1;
5451 std::pair<ModuleFile *, unsigned> PPInfo = getModulePreprocessedEntity(Index);
5452 ModuleFile &M = *PPInfo.first;
5453 unsigned LocalIndex = PPInfo.second;
5454 const PPEntityOffset &PPOffs = M.PreprocessedEntityOffsets[LocalIndex];
5455
Guy Benyei11169dd2012-12-18 14:30:41 +00005456 if (!PP.getPreprocessingRecord()) {
5457 Error("no preprocessing record");
Craig Toppera13603a2014-05-22 05:54:18 +00005458 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005459 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005460
5461 SavedStreamPosition SavedPosition(M.PreprocessorDetailCursor);
Chris Lattner7fb3bef2013-01-20 00:56:42 +00005462 M.PreprocessorDetailCursor.JumpToBit(PPOffs.BitOffset);
5463
5464 llvm::BitstreamEntry Entry =
5465 M.PreprocessorDetailCursor.advance(BitstreamCursor::AF_DontPopBlockAtEnd);
5466 if (Entry.Kind != llvm::BitstreamEntry::Record)
Craig Toppera13603a2014-05-22 05:54:18 +00005467 return nullptr;
Chris Lattner7fb3bef2013-01-20 00:56:42 +00005468
Guy Benyei11169dd2012-12-18 14:30:41 +00005469 // Read the record.
Richard Smithcb34bd32016-03-27 07:28:06 +00005470 SourceRange Range(TranslateSourceLocation(M, PPOffs.getBegin()),
5471 TranslateSourceLocation(M, PPOffs.getEnd()));
Guy Benyei11169dd2012-12-18 14:30:41 +00005472 PreprocessingRecord &PPRec = *PP.getPreprocessingRecord();
Chris Lattner0e6c9402013-01-20 02:38:54 +00005473 StringRef Blob;
Guy Benyei11169dd2012-12-18 14:30:41 +00005474 RecordData Record;
Vedant Kumar48b4f762018-04-14 01:40:48 +00005475 PreprocessorDetailRecordTypes RecType =
5476 (PreprocessorDetailRecordTypes)M.PreprocessorDetailCursor.readRecord(
5477 Entry.ID, Record, &Blob);
Guy Benyei11169dd2012-12-18 14:30:41 +00005478 switch (RecType) {
5479 case PPD_MACRO_EXPANSION: {
5480 bool isBuiltin = Record[0];
Craig Toppera13603a2014-05-22 05:54:18 +00005481 IdentifierInfo *Name = nullptr;
Richard Smith66a81862015-05-04 02:25:31 +00005482 MacroDefinitionRecord *Def = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005483 if (isBuiltin)
5484 Name = getLocalIdentifier(M, Record[1]);
5485 else {
Richard Smith66a81862015-05-04 02:25:31 +00005486 PreprocessedEntityID GlobalID =
5487 getGlobalPreprocessedEntityID(M, Record[1]);
5488 Def = cast<MacroDefinitionRecord>(
5489 PPRec.getLoadedPreprocessedEntity(GlobalID - 1));
Guy Benyei11169dd2012-12-18 14:30:41 +00005490 }
5491
5492 MacroExpansion *ME;
5493 if (isBuiltin)
5494 ME = new (PPRec) MacroExpansion(Name, Range);
5495 else
5496 ME = new (PPRec) MacroExpansion(Def, Range);
5497
5498 return ME;
5499 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005500
Guy Benyei11169dd2012-12-18 14:30:41 +00005501 case PPD_MACRO_DEFINITION: {
5502 // Decode the identifier info and then check again; if the macro is
5503 // still defined and associated with the identifier,
5504 IdentifierInfo *II = getLocalIdentifier(M, Record[0]);
Vedant Kumar48b4f762018-04-14 01:40:48 +00005505 MacroDefinitionRecord *MD = new (PPRec) MacroDefinitionRecord(II, Range);
Guy Benyei11169dd2012-12-18 14:30:41 +00005506
5507 if (DeserializationListener)
5508 DeserializationListener->MacroDefinitionRead(PPID, MD);
5509
5510 return MD;
5511 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005512
Guy Benyei11169dd2012-12-18 14:30:41 +00005513 case PPD_INCLUSION_DIRECTIVE: {
Chris Lattner0e6c9402013-01-20 02:38:54 +00005514 const char *FullFileNameStart = Blob.data() + Record[0];
5515 StringRef FullFileName(FullFileNameStart, Blob.size() - Record[0]);
Craig Toppera13603a2014-05-22 05:54:18 +00005516 const FileEntry *File = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005517 if (!FullFileName.empty())
5518 File = PP.getFileManager().getFile(FullFileName);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005519
Guy Benyei11169dd2012-12-18 14:30:41 +00005520 // FIXME: Stable encoding
Vedant Kumar48b4f762018-04-14 01:40:48 +00005521 InclusionDirective::InclusionKind Kind
5522 = static_cast<InclusionDirective::InclusionKind>(Record[2]);
5523 InclusionDirective *ID
5524 = new (PPRec) InclusionDirective(PPRec, Kind,
Chris Lattner0e6c9402013-01-20 02:38:54 +00005525 StringRef(Blob.data(), Record[0]),
Guy Benyei11169dd2012-12-18 14:30:41 +00005526 Record[1], Record[3],
5527 File,
5528 Range);
5529 return ID;
5530 }
5531 }
5532
5533 llvm_unreachable("Invalid PreprocessorDetailRecordTypes");
5534}
5535
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00005536/// Find the next module that contains entities and return the ID
Guy Benyei11169dd2012-12-18 14:30:41 +00005537/// of the first entry.
NAKAMURA Takumi12ab07e2017-10-12 09:42:14 +00005538///
5539/// \param SLocMapI points at a chunk of a module that contains no
5540/// preprocessed entities or the entities it contains are not the ones we are
5541/// looking for.
Guy Benyei11169dd2012-12-18 14:30:41 +00005542PreprocessedEntityID ASTReader::findNextPreprocessedEntity(
5543 GlobalSLocOffsetMapType::const_iterator SLocMapI) const {
5544 ++SLocMapI;
5545 for (GlobalSLocOffsetMapType::const_iterator
5546 EndI = GlobalSLocOffsetMap.end(); SLocMapI != EndI; ++SLocMapI) {
5547 ModuleFile &M = *SLocMapI->second;
5548 if (M.NumPreprocessedEntities)
5549 return M.BasePreprocessedEntityID;
5550 }
5551
5552 return getTotalNumPreprocessedEntities();
5553}
5554
5555namespace {
5556
Guy Benyei11169dd2012-12-18 14:30:41 +00005557struct PPEntityComp {
5558 const ASTReader &Reader;
5559 ModuleFile &M;
5560
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005561 PPEntityComp(const ASTReader &Reader, ModuleFile &M) : Reader(Reader), M(M) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00005562
5563 bool operator()(const PPEntityOffset &L, const PPEntityOffset &R) const {
5564 SourceLocation LHS = getLoc(L);
5565 SourceLocation RHS = getLoc(R);
5566 return Reader.getSourceManager().isBeforeInTranslationUnit(LHS, RHS);
5567 }
5568
5569 bool operator()(const PPEntityOffset &L, SourceLocation RHS) const {
5570 SourceLocation LHS = getLoc(L);
5571 return Reader.getSourceManager().isBeforeInTranslationUnit(LHS, RHS);
5572 }
5573
5574 bool operator()(SourceLocation LHS, const PPEntityOffset &R) const {
5575 SourceLocation RHS = getLoc(R);
5576 return Reader.getSourceManager().isBeforeInTranslationUnit(LHS, RHS);
5577 }
5578
5579 SourceLocation getLoc(const PPEntityOffset &PPE) const {
Richard Smithb22a1d12016-03-27 20:13:24 +00005580 return Reader.TranslateSourceLocation(M, PPE.getBegin());
Guy Benyei11169dd2012-12-18 14:30:41 +00005581 }
5582};
5583
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005584} // namespace
Guy Benyei11169dd2012-12-18 14:30:41 +00005585
Alp Toker2e9ce4c2014-05-16 18:59:21 +00005586PreprocessedEntityID ASTReader::findPreprocessedEntity(SourceLocation Loc,
5587 bool EndsAfter) const {
5588 if (SourceMgr.isLocalSourceLocation(Loc))
Guy Benyei11169dd2012-12-18 14:30:41 +00005589 return getTotalNumPreprocessedEntities();
5590
Alp Toker2e9ce4c2014-05-16 18:59:21 +00005591 GlobalSLocOffsetMapType::const_iterator SLocMapI = GlobalSLocOffsetMap.find(
5592 SourceManager::MaxLoadedOffset - Loc.getOffset() - 1);
Guy Benyei11169dd2012-12-18 14:30:41 +00005593 assert(SLocMapI != GlobalSLocOffsetMap.end() &&
5594 "Corrupted global sloc offset map");
5595
5596 if (SLocMapI->second->NumPreprocessedEntities == 0)
5597 return findNextPreprocessedEntity(SLocMapI);
5598
5599 ModuleFile &M = *SLocMapI->second;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005600
5601 using pp_iterator = const PPEntityOffset *;
5602
Guy Benyei11169dd2012-12-18 14:30:41 +00005603 pp_iterator pp_begin = M.PreprocessedEntityOffsets;
5604 pp_iterator pp_end = pp_begin + M.NumPreprocessedEntities;
5605
5606 size_t Count = M.NumPreprocessedEntities;
5607 size_t Half;
5608 pp_iterator First = pp_begin;
5609 pp_iterator PPI;
5610
Alp Toker2e9ce4c2014-05-16 18:59:21 +00005611 if (EndsAfter) {
5612 PPI = std::upper_bound(pp_begin, pp_end, Loc,
Richard Smithb22a1d12016-03-27 20:13:24 +00005613 PPEntityComp(*this, M));
Alp Toker2e9ce4c2014-05-16 18:59:21 +00005614 } else {
5615 // Do a binary search manually instead of using std::lower_bound because
5616 // The end locations of entities may be unordered (when a macro expansion
5617 // is inside another macro argument), but for this case it is not important
5618 // whether we get the first macro expansion or its containing macro.
5619 while (Count > 0) {
5620 Half = Count / 2;
5621 PPI = First;
5622 std::advance(PPI, Half);
Richard Smithb22a1d12016-03-27 20:13:24 +00005623 if (SourceMgr.isBeforeInTranslationUnit(
5624 TranslateSourceLocation(M, PPI->getEnd()), Loc)) {
Alp Toker2e9ce4c2014-05-16 18:59:21 +00005625 First = PPI;
5626 ++First;
5627 Count = Count - Half - 1;
5628 } else
5629 Count = Half;
5630 }
Guy Benyei11169dd2012-12-18 14:30:41 +00005631 }
5632
5633 if (PPI == pp_end)
5634 return findNextPreprocessedEntity(SLocMapI);
5635
5636 return M.BasePreprocessedEntityID + (PPI - pp_begin);
5637}
5638
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00005639/// Returns a pair of [Begin, End) indices of preallocated
Guy Benyei11169dd2012-12-18 14:30:41 +00005640/// preprocessed entities that \arg Range encompasses.
5641std::pair<unsigned, unsigned>
5642 ASTReader::findPreprocessedEntitiesInRange(SourceRange Range) {
5643 if (Range.isInvalid())
5644 return std::make_pair(0,0);
5645 assert(!SourceMgr.isBeforeInTranslationUnit(Range.getEnd(),Range.getBegin()));
5646
Alp Toker2e9ce4c2014-05-16 18:59:21 +00005647 PreprocessedEntityID BeginID =
5648 findPreprocessedEntity(Range.getBegin(), false);
5649 PreprocessedEntityID EndID = findPreprocessedEntity(Range.getEnd(), true);
Guy Benyei11169dd2012-12-18 14:30:41 +00005650 return std::make_pair(BeginID, EndID);
5651}
5652
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00005653/// Optionally returns true or false if the preallocated preprocessed
Guy Benyei11169dd2012-12-18 14:30:41 +00005654/// entity with index \arg Index came from file \arg FID.
David Blaikie05785d12013-02-20 22:23:23 +00005655Optional<bool> ASTReader::isPreprocessedEntityInFileID(unsigned Index,
Guy Benyei11169dd2012-12-18 14:30:41 +00005656 FileID FID) {
5657 if (FID.isInvalid())
5658 return false;
5659
5660 std::pair<ModuleFile *, unsigned> PPInfo = getModulePreprocessedEntity(Index);
5661 ModuleFile &M = *PPInfo.first;
5662 unsigned LocalIndex = PPInfo.second;
5663 const PPEntityOffset &PPOffs = M.PreprocessedEntityOffsets[LocalIndex];
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005664
Richard Smithcb34bd32016-03-27 07:28:06 +00005665 SourceLocation Loc = TranslateSourceLocation(M, PPOffs.getBegin());
Guy Benyei11169dd2012-12-18 14:30:41 +00005666 if (Loc.isInvalid())
5667 return false;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005668
Guy Benyei11169dd2012-12-18 14:30:41 +00005669 if (SourceMgr.isInFileID(SourceMgr.getFileLoc(Loc), FID))
5670 return true;
5671 else
5672 return false;
5673}
5674
5675namespace {
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00005676
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00005677 /// Visitor used to search for information about a header file.
Guy Benyei11169dd2012-12-18 14:30:41 +00005678 class HeaderFileInfoVisitor {
Guy Benyei11169dd2012-12-18 14:30:41 +00005679 const FileEntry *FE;
David Blaikie05785d12013-02-20 22:23:23 +00005680 Optional<HeaderFileInfo> HFI;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005681
Guy Benyei11169dd2012-12-18 14:30:41 +00005682 public:
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005683 explicit HeaderFileInfoVisitor(const FileEntry *FE) : FE(FE) {}
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00005684
5685 bool operator()(ModuleFile &M) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00005686 HeaderFileInfoLookupTable *Table
5687 = static_cast<HeaderFileInfoLookupTable *>(M.HeaderFileInfoTable);
Guy Benyei11169dd2012-12-18 14:30:41 +00005688 if (!Table)
5689 return false;
5690
5691 // Look in the on-disk hash table for an entry for this file name.
Richard Smithbdf2d932015-07-30 03:37:16 +00005692 HeaderFileInfoLookupTable::iterator Pos = Table->find(FE);
Guy Benyei11169dd2012-12-18 14:30:41 +00005693 if (Pos == Table->end())
5694 return false;
5695
Richard Smithbdf2d932015-07-30 03:37:16 +00005696 HFI = *Pos;
Guy Benyei11169dd2012-12-18 14:30:41 +00005697 return true;
5698 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005699
David Blaikie05785d12013-02-20 22:23:23 +00005700 Optional<HeaderFileInfo> getHeaderFileInfo() const { return HFI; }
Guy Benyei11169dd2012-12-18 14:30:41 +00005701 };
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00005702
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00005703} // namespace
Guy Benyei11169dd2012-12-18 14:30:41 +00005704
5705HeaderFileInfo ASTReader::GetHeaderFileInfo(const FileEntry *FE) {
Argyrios Kyrtzidis61a38962013-03-06 18:12:44 +00005706 HeaderFileInfoVisitor Visitor(FE);
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00005707 ModuleMgr.visit(Visitor);
Argyrios Kyrtzidis1054bbf2013-05-08 23:46:55 +00005708 if (Optional<HeaderFileInfo> HFI = Visitor.getHeaderFileInfo())
Guy Benyei11169dd2012-12-18 14:30:41 +00005709 return *HFI;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005710
Guy Benyei11169dd2012-12-18 14:30:41 +00005711 return HeaderFileInfo();
5712}
5713
5714void ASTReader::ReadPragmaDiagnosticMappings(DiagnosticsEngine &Diag) {
Richard Smithd230de22017-01-26 01:01:01 +00005715 using DiagState = DiagnosticsEngine::DiagState;
5716 SmallVector<DiagState *, 32> DiagStates;
5717
Vedant Kumar48b4f762018-04-14 01:40:48 +00005718 for (ModuleFile &F : ModuleMgr) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005719 unsigned Idx = 0;
Richard Smithd230de22017-01-26 01:01:01 +00005720 auto &Record = F.PragmaDiagMappings;
5721 if (Record.empty())
5722 continue;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005723
Richard Smithd230de22017-01-26 01:01:01 +00005724 DiagStates.clear();
5725
5726 auto ReadDiagState =
5727 [&](const DiagState &BasedOn, SourceLocation Loc,
5728 bool IncludeNonPragmaStates) -> DiagnosticsEngine::DiagState * {
5729 unsigned BackrefID = Record[Idx++];
5730 if (BackrefID != 0)
5731 return DiagStates[BackrefID - 1];
5732
Guy Benyei11169dd2012-12-18 14:30:41 +00005733 // A new DiagState was created here.
Richard Smithd230de22017-01-26 01:01:01 +00005734 Diag.DiagStates.push_back(BasedOn);
5735 DiagState *NewState = &Diag.DiagStates.back();
Guy Benyei11169dd2012-12-18 14:30:41 +00005736 DiagStates.push_back(NewState);
Duncan P. N. Exon Smith3cb183b2017-03-14 19:31:27 +00005737 unsigned Size = Record[Idx++];
5738 assert(Idx + Size * 2 <= Record.size() &&
5739 "Invalid data, not enough diag/map pairs");
5740 while (Size--) {
Richard Smithd230de22017-01-26 01:01:01 +00005741 unsigned DiagID = Record[Idx++];
Duncan P. N. Exon Smith900f8172017-04-12 03:58:58 +00005742 DiagnosticMapping NewMapping =
Richard Smithe37391c2017-05-03 00:28:49 +00005743 DiagnosticMapping::deserialize(Record[Idx++]);
Duncan P. N. Exon Smith900f8172017-04-12 03:58:58 +00005744 if (!NewMapping.isPragma() && !IncludeNonPragmaStates)
5745 continue;
5746
5747 DiagnosticMapping &Mapping = NewState->getOrAddMapping(DiagID);
5748
5749 // If this mapping was specified as a warning but the severity was
5750 // upgraded due to diagnostic settings, simulate the current diagnostic
5751 // settings (and use a warning).
Richard Smithe37391c2017-05-03 00:28:49 +00005752 if (NewMapping.wasUpgradedFromWarning() && !Mapping.isErrorOrFatal()) {
5753 NewMapping.setSeverity(diag::Severity::Warning);
5754 NewMapping.setUpgradedFromWarning(false);
Duncan P. N. Exon Smith900f8172017-04-12 03:58:58 +00005755 }
5756
Duncan P. N. Exon Smith900f8172017-04-12 03:58:58 +00005757 Mapping = NewMapping;
Richard Smithd230de22017-01-26 01:01:01 +00005758 }
Richard Smithd230de22017-01-26 01:01:01 +00005759 return NewState;
5760 };
5761
Duncan P. N. Exon Smitha351c102017-04-12 03:45:32 +00005762 // Read the first state.
Duncan P. N. Exon Smith900f8172017-04-12 03:58:58 +00005763 DiagState *FirstState;
5764 if (F.Kind == MK_ImplicitModule) {
5765 // Implicitly-built modules are reused with different diagnostic
5766 // settings. Use the initial diagnostic state from Diag to simulate this
5767 // compilation's diagnostic settings.
5768 FirstState = Diag.DiagStatesByLoc.FirstDiagState;
5769 DiagStates.push_back(FirstState);
5770
5771 // Skip the initial diagnostic state from the serialized module.
Richard Smithe37391c2017-05-03 00:28:49 +00005772 assert(Record[1] == 0 &&
Duncan P. N. Exon Smith900f8172017-04-12 03:58:58 +00005773 "Invalid data, unexpected backref in initial state");
Richard Smithe37391c2017-05-03 00:28:49 +00005774 Idx = 3 + Record[2] * 2;
Duncan P. N. Exon Smith900f8172017-04-12 03:58:58 +00005775 assert(Idx < Record.size() &&
5776 "Invalid data, not enough state change pairs in initial state");
Richard Smithe37391c2017-05-03 00:28:49 +00005777 } else if (F.isModule()) {
5778 // For an explicit module, preserve the flags from the module build
5779 // command line (-w, -Weverything, -Werror, ...) along with any explicit
5780 // -Wblah flags.
5781 unsigned Flags = Record[Idx++];
5782 DiagState Initial;
5783 Initial.SuppressSystemWarnings = Flags & 1; Flags >>= 1;
5784 Initial.ErrorsAsFatal = Flags & 1; Flags >>= 1;
5785 Initial.WarningsAsErrors = Flags & 1; Flags >>= 1;
5786 Initial.EnableAllWarnings = Flags & 1; Flags >>= 1;
5787 Initial.IgnoreAllWarnings = Flags & 1; Flags >>= 1;
5788 Initial.ExtBehavior = (diag::Severity)Flags;
5789 FirstState = ReadDiagState(Initial, SourceLocation(), true);
Richard Smithea741482017-05-01 22:10:47 +00005790
Richard Smith6c2b5a82018-02-09 01:15:13 +00005791 assert(F.OriginalSourceFileID.isValid());
5792
Richard Smithe37391c2017-05-03 00:28:49 +00005793 // Set up the root buffer of the module to start with the initial
5794 // diagnostic state of the module itself, to cover files that contain no
5795 // explicit transitions (for which we did not serialize anything).
5796 Diag.DiagStatesByLoc.Files[F.OriginalSourceFileID]
5797 .StateTransitions.push_back({FirstState, 0});
5798 } else {
5799 // For prefix ASTs, start with whatever the user configured on the
5800 // command line.
5801 Idx++; // Skip flags.
5802 FirstState = ReadDiagState(*Diag.DiagStatesByLoc.CurDiagState,
5803 SourceLocation(), false);
Duncan P. N. Exon Smith900f8172017-04-12 03:58:58 +00005804 }
Richard Smithd230de22017-01-26 01:01:01 +00005805
Duncan P. N. Exon Smitha351c102017-04-12 03:45:32 +00005806 // Read the state transitions.
5807 unsigned NumLocations = Record[Idx++];
5808 while (NumLocations--) {
5809 assert(Idx < Record.size() &&
5810 "Invalid data, missing pragma diagnostic states");
Richard Smithd230de22017-01-26 01:01:01 +00005811 SourceLocation Loc = ReadSourceLocation(F, Record[Idx++]);
5812 auto IDAndOffset = SourceMgr.getDecomposedLoc(Loc);
Richard Smith6c2b5a82018-02-09 01:15:13 +00005813 assert(IDAndOffset.first.isValid() && "invalid FileID for transition");
Richard Smithd230de22017-01-26 01:01:01 +00005814 assert(IDAndOffset.second == 0 && "not a start location for a FileID");
5815 unsigned Transitions = Record[Idx++];
5816
5817 // Note that we don't need to set up Parent/ParentOffset here, because
5818 // we won't be changing the diagnostic state within imported FileIDs
5819 // (other than perhaps appending to the main source file, which has no
5820 // parent).
5821 auto &F = Diag.DiagStatesByLoc.Files[IDAndOffset.first];
5822 F.StateTransitions.reserve(F.StateTransitions.size() + Transitions);
5823 for (unsigned I = 0; I != Transitions; ++I) {
5824 unsigned Offset = Record[Idx++];
5825 auto *State =
5826 ReadDiagState(*FirstState, Loc.getLocWithOffset(Offset), false);
5827 F.StateTransitions.push_back({State, Offset});
Guy Benyei11169dd2012-12-18 14:30:41 +00005828 }
5829 }
Richard Smithd230de22017-01-26 01:01:01 +00005830
Duncan P. N. Exon Smitha351c102017-04-12 03:45:32 +00005831 // Read the final state.
5832 assert(Idx < Record.size() &&
5833 "Invalid data, missing final pragma diagnostic state");
5834 SourceLocation CurStateLoc =
5835 ReadSourceLocation(F, F.PragmaDiagMappings[Idx++]);
5836 auto *CurState = ReadDiagState(*FirstState, CurStateLoc, false);
5837
5838 if (!F.isModule()) {
5839 Diag.DiagStatesByLoc.CurDiagState = CurState;
5840 Diag.DiagStatesByLoc.CurDiagStateLoc = CurStateLoc;
5841
5842 // Preserve the property that the imaginary root file describes the
5843 // current state.
Simon Pilgrim22518632017-10-10 13:56:17 +00005844 FileID NullFile;
5845 auto &T = Diag.DiagStatesByLoc.Files[NullFile].StateTransitions;
Duncan P. N. Exon Smitha351c102017-04-12 03:45:32 +00005846 if (T.empty())
5847 T.push_back({CurState, 0});
5848 else
5849 T[0].State = CurState;
5850 }
5851
Richard Smithd230de22017-01-26 01:01:01 +00005852 // Don't try to read these mappings again.
5853 Record.clear();
Guy Benyei11169dd2012-12-18 14:30:41 +00005854 }
5855}
5856
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00005857/// Get the correct cursor and offset for loading a type.
Guy Benyei11169dd2012-12-18 14:30:41 +00005858ASTReader::RecordLocation ASTReader::TypeCursorForIndex(unsigned Index) {
5859 GlobalTypeMapType::iterator I = GlobalTypeMap.find(Index);
5860 assert(I != GlobalTypeMap.end() && "Corrupted global type map");
5861 ModuleFile *M = I->second;
5862 return RecordLocation(M, M->TypeOffsets[Index - M->BaseTypeIndex]);
5863}
5864
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00005865/// Read and return the type with the given index..
Guy Benyei11169dd2012-12-18 14:30:41 +00005866///
5867/// The index is the type ID, shifted and minus the number of predefs. This
5868/// routine actually reads the record corresponding to the type at the given
5869/// location. It is a helper routine for GetType, which deals with reading type
5870/// IDs.
5871QualType ASTReader::readTypeRecord(unsigned Index) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00005872 assert(ContextObj && "reading type with no AST context");
5873 ASTContext &Context = *ContextObj;
Guy Benyei11169dd2012-12-18 14:30:41 +00005874 RecordLocation Loc = TypeCursorForIndex(Index);
Chris Lattner7fb3bef2013-01-20 00:56:42 +00005875 BitstreamCursor &DeclsCursor = Loc.F->DeclsCursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00005876
5877 // Keep track of where we are in the stream, then jump back there
5878 // after reading this type.
5879 SavedStreamPosition SavedPosition(DeclsCursor);
5880
5881 ReadingKindTracker ReadingKind(Read_Type, *this);
5882
5883 // Note that we are loading a type record.
5884 Deserializing AType(this);
5885
5886 unsigned Idx = 0;
5887 DeclsCursor.JumpToBit(Loc.Offset);
5888 RecordData Record;
5889 unsigned Code = DeclsCursor.ReadCode();
Chris Lattner0e6c9402013-01-20 02:38:54 +00005890 switch ((TypeCode)DeclsCursor.readRecord(Code, Record)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005891 case TYPE_EXT_QUAL: {
5892 if (Record.size() != 2) {
5893 Error("Incorrect encoding of extended qualifier type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005894 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00005895 }
5896 QualType Base = readType(*Loc.F, Record, Idx);
5897 Qualifiers Quals = Qualifiers::fromOpaqueValue(Record[Idx++]);
5898 return Context.getQualifiedType(Base, Quals);
5899 }
5900
5901 case TYPE_COMPLEX: {
5902 if (Record.size() != 1) {
5903 Error("Incorrect encoding of complex type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005904 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00005905 }
5906 QualType ElemType = readType(*Loc.F, Record, Idx);
5907 return Context.getComplexType(ElemType);
5908 }
5909
5910 case TYPE_POINTER: {
5911 if (Record.size() != 1) {
5912 Error("Incorrect encoding of pointer type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005913 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00005914 }
5915 QualType PointeeType = readType(*Loc.F, Record, Idx);
5916 return Context.getPointerType(PointeeType);
5917 }
5918
Reid Kleckner8a365022013-06-24 17:51:48 +00005919 case TYPE_DECAYED: {
5920 if (Record.size() != 1) {
5921 Error("Incorrect encoding of decayed type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005922 return QualType();
Reid Kleckner8a365022013-06-24 17:51:48 +00005923 }
5924 QualType OriginalType = readType(*Loc.F, Record, Idx);
5925 QualType DT = Context.getAdjustedParameterType(OriginalType);
5926 if (!isa<DecayedType>(DT))
5927 Error("Decayed type does not decay");
5928 return DT;
5929 }
5930
Reid Kleckner0503a872013-12-05 01:23:43 +00005931 case TYPE_ADJUSTED: {
5932 if (Record.size() != 2) {
5933 Error("Incorrect encoding of adjusted type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005934 return QualType();
Reid Kleckner0503a872013-12-05 01:23:43 +00005935 }
5936 QualType OriginalTy = readType(*Loc.F, Record, Idx);
5937 QualType AdjustedTy = readType(*Loc.F, Record, Idx);
5938 return Context.getAdjustedType(OriginalTy, AdjustedTy);
5939 }
5940
Guy Benyei11169dd2012-12-18 14:30:41 +00005941 case TYPE_BLOCK_POINTER: {
5942 if (Record.size() != 1) {
5943 Error("Incorrect encoding of block pointer type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005944 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00005945 }
5946 QualType PointeeType = readType(*Loc.F, Record, Idx);
5947 return Context.getBlockPointerType(PointeeType);
5948 }
5949
5950 case TYPE_LVALUE_REFERENCE: {
5951 if (Record.size() != 2) {
5952 Error("Incorrect encoding of lvalue reference type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005953 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00005954 }
5955 QualType PointeeType = readType(*Loc.F, Record, Idx);
5956 return Context.getLValueReferenceType(PointeeType, Record[1]);
5957 }
5958
5959 case TYPE_RVALUE_REFERENCE: {
5960 if (Record.size() != 1) {
5961 Error("Incorrect encoding of rvalue reference type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005962 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00005963 }
5964 QualType PointeeType = readType(*Loc.F, Record, Idx);
5965 return Context.getRValueReferenceType(PointeeType);
5966 }
5967
5968 case TYPE_MEMBER_POINTER: {
5969 if (Record.size() != 2) {
5970 Error("Incorrect encoding of member pointer type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00005971 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00005972 }
5973 QualType PointeeType = readType(*Loc.F, Record, Idx);
5974 QualType ClassType = readType(*Loc.F, Record, Idx);
5975 if (PointeeType.isNull() || ClassType.isNull())
Vedant Kumar48b4f762018-04-14 01:40:48 +00005976 return QualType();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00005977
Guy Benyei11169dd2012-12-18 14:30:41 +00005978 return Context.getMemberPointerType(PointeeType, ClassType.getTypePtr());
5979 }
5980
5981 case TYPE_CONSTANT_ARRAY: {
5982 QualType ElementType = readType(*Loc.F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00005983 ArrayType::ArraySizeModifier ASM = (ArrayType::ArraySizeModifier)Record[1];
Guy Benyei11169dd2012-12-18 14:30:41 +00005984 unsigned IndexTypeQuals = Record[2];
5985 unsigned Idx = 3;
5986 llvm::APInt Size = ReadAPInt(Record, Idx);
5987 return Context.getConstantArrayType(ElementType, Size,
5988 ASM, IndexTypeQuals);
5989 }
5990
5991 case TYPE_INCOMPLETE_ARRAY: {
5992 QualType ElementType = readType(*Loc.F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00005993 ArrayType::ArraySizeModifier ASM = (ArrayType::ArraySizeModifier)Record[1];
Guy Benyei11169dd2012-12-18 14:30:41 +00005994 unsigned IndexTypeQuals = Record[2];
5995 return Context.getIncompleteArrayType(ElementType, ASM, IndexTypeQuals);
5996 }
5997
5998 case TYPE_VARIABLE_ARRAY: {
5999 QualType ElementType = readType(*Loc.F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00006000 ArrayType::ArraySizeModifier ASM = (ArrayType::ArraySizeModifier)Record[1];
Guy Benyei11169dd2012-12-18 14:30:41 +00006001 unsigned IndexTypeQuals = Record[2];
6002 SourceLocation LBLoc = ReadSourceLocation(*Loc.F, Record[3]);
6003 SourceLocation RBLoc = ReadSourceLocation(*Loc.F, Record[4]);
6004 return Context.getVariableArrayType(ElementType, ReadExpr(*Loc.F),
6005 ASM, IndexTypeQuals,
6006 SourceRange(LBLoc, RBLoc));
6007 }
6008
6009 case TYPE_VECTOR: {
6010 if (Record.size() != 3) {
6011 Error("incorrect encoding of vector type in AST file");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006012 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006013 }
6014
6015 QualType ElementType = readType(*Loc.F, Record, Idx);
6016 unsigned NumElements = Record[1];
6017 unsigned VecKind = Record[2];
6018 return Context.getVectorType(ElementType, NumElements,
6019 (VectorType::VectorKind)VecKind);
6020 }
6021
6022 case TYPE_EXT_VECTOR: {
6023 if (Record.size() != 3) {
6024 Error("incorrect encoding of extended vector type in AST file");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006025 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006026 }
6027
6028 QualType ElementType = readType(*Loc.F, Record, Idx);
6029 unsigned NumElements = Record[1];
6030 return Context.getExtVectorType(ElementType, NumElements);
6031 }
6032
6033 case TYPE_FUNCTION_NO_PROTO: {
Oren Ben Simhon220671a2018-03-17 13:31:35 +00006034 if (Record.size() != 8) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006035 Error("incorrect encoding of no-proto function type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006036 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006037 }
6038 QualType ResultType = readType(*Loc.F, Record, Idx);
6039 FunctionType::ExtInfo Info(Record[1], Record[2], Record[3],
Fangrui Song6907ce22018-07-30 19:24:48 +00006040 (CallingConv)Record[4], Record[5], Record[6],
Oren Ben Simhon220671a2018-03-17 13:31:35 +00006041 Record[7]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006042 return Context.getFunctionNoProtoType(ResultType, Info);
6043 }
6044
6045 case TYPE_FUNCTION_PROTO: {
6046 QualType ResultType = readType(*Loc.F, Record, Idx);
6047
6048 FunctionProtoType::ExtProtoInfo EPI;
6049 EPI.ExtInfo = FunctionType::ExtInfo(/*noreturn*/ Record[1],
6050 /*hasregparm*/ Record[2],
6051 /*regparm*/ Record[3],
6052 static_cast<CallingConv>(Record[4]),
Oren Ben Simhon318a6ea2017-04-27 12:01:00 +00006053 /*produces*/ Record[5],
Oren Ben Simhon220671a2018-03-17 13:31:35 +00006054 /*nocallersavedregs*/ Record[6],
6055 /*nocfcheck*/ Record[7]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006056
Oren Ben Simhon220671a2018-03-17 13:31:35 +00006057 unsigned Idx = 8;
Guy Benyei11169dd2012-12-18 14:30:41 +00006058
6059 EPI.Variadic = Record[Idx++];
6060 EPI.HasTrailingReturn = Record[Idx++];
6061 EPI.TypeQuals = Record[Idx++];
6062 EPI.RefQualifier = static_cast<RefQualifierKind>(Record[Idx++]);
Richard Smith564417a2014-03-20 21:47:22 +00006063 SmallVector<QualType, 8> ExceptionStorage;
Richard Smith8acb4282014-07-31 21:57:55 +00006064 readExceptionSpec(*Loc.F, ExceptionStorage, EPI.ExceptionSpec, Record, Idx);
Richard Smith01b2cb42014-07-26 06:37:51 +00006065
6066 unsigned NumParams = Record[Idx++];
6067 SmallVector<QualType, 16> ParamTypes;
6068 for (unsigned I = 0; I != NumParams; ++I)
6069 ParamTypes.push_back(readType(*Loc.F, Record, Idx));
6070
John McCall18afab72016-03-01 00:49:02 +00006071 SmallVector<FunctionProtoType::ExtParameterInfo, 4> ExtParameterInfos;
6072 if (Idx != Record.size()) {
6073 for (unsigned I = 0; I != NumParams; ++I)
6074 ExtParameterInfos.push_back(
6075 FunctionProtoType::ExtParameterInfo
6076 ::getFromOpaqueValue(Record[Idx++]));
6077 EPI.ExtParameterInfos = ExtParameterInfos.data();
6078 }
6079
6080 assert(Idx == Record.size());
6081
Jordan Rose5c382722013-03-08 21:51:21 +00006082 return Context.getFunctionType(ResultType, ParamTypes, EPI);
Guy Benyei11169dd2012-12-18 14:30:41 +00006083 }
6084
6085 case TYPE_UNRESOLVED_USING: {
6086 unsigned Idx = 0;
6087 return Context.getTypeDeclType(
6088 ReadDeclAs<UnresolvedUsingTypenameDecl>(*Loc.F, Record, Idx));
6089 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006090
Guy Benyei11169dd2012-12-18 14:30:41 +00006091 case TYPE_TYPEDEF: {
6092 if (Record.size() != 2) {
6093 Error("incorrect encoding of typedef type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006094 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006095 }
6096 unsigned Idx = 0;
Vedant Kumar48b4f762018-04-14 01:40:48 +00006097 TypedefNameDecl *Decl = ReadDeclAs<TypedefNameDecl>(*Loc.F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00006098 QualType Canonical = readType(*Loc.F, Record, Idx);
6099 if (!Canonical.isNull())
6100 Canonical = Context.getCanonicalType(Canonical);
6101 return Context.getTypedefType(Decl, Canonical);
6102 }
6103
6104 case TYPE_TYPEOF_EXPR:
6105 return Context.getTypeOfExprType(ReadExpr(*Loc.F));
6106
6107 case TYPE_TYPEOF: {
6108 if (Record.size() != 1) {
6109 Error("incorrect encoding of typeof(type) in AST file");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006110 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006111 }
6112 QualType UnderlyingType = readType(*Loc.F, Record, Idx);
6113 return Context.getTypeOfType(UnderlyingType);
6114 }
6115
6116 case TYPE_DECLTYPE: {
6117 QualType UnderlyingType = readType(*Loc.F, Record, Idx);
6118 return Context.getDecltypeType(ReadExpr(*Loc.F), UnderlyingType);
6119 }
6120
6121 case TYPE_UNARY_TRANSFORM: {
6122 QualType BaseType = readType(*Loc.F, Record, Idx);
6123 QualType UnderlyingType = readType(*Loc.F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00006124 UnaryTransformType::UTTKind UKind = (UnaryTransformType::UTTKind)Record[2];
Guy Benyei11169dd2012-12-18 14:30:41 +00006125 return Context.getUnaryTransformType(BaseType, UnderlyingType, UKind);
6126 }
6127
Richard Smith74aeef52013-04-26 16:15:35 +00006128 case TYPE_AUTO: {
6129 QualType Deduced = readType(*Loc.F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00006130 AutoTypeKeyword Keyword = (AutoTypeKeyword)Record[Idx++];
Richard Smith27d807c2013-04-30 13:56:41 +00006131 bool IsDependent = Deduced.isNull() ? Record[Idx++] : false;
Richard Smithe301ba22015-11-11 02:02:15 +00006132 return Context.getAutoType(Deduced, Keyword, IsDependent);
Richard Smith74aeef52013-04-26 16:15:35 +00006133 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006134
Richard Smith600b5262017-01-26 20:40:47 +00006135 case TYPE_DEDUCED_TEMPLATE_SPECIALIZATION: {
6136 TemplateName Name = ReadTemplateName(*Loc.F, Record, Idx);
6137 QualType Deduced = readType(*Loc.F, Record, Idx);
6138 bool IsDependent = Deduced.isNull() ? Record[Idx++] : false;
6139 return Context.getDeducedTemplateSpecializationType(Name, Deduced,
6140 IsDependent);
6141 }
6142
Guy Benyei11169dd2012-12-18 14:30:41 +00006143 case TYPE_RECORD: {
6144 if (Record.size() != 2) {
6145 Error("incorrect encoding of record type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006146 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006147 }
6148 unsigned Idx = 0;
6149 bool IsDependent = Record[Idx++];
Vedant Kumar48b4f762018-04-14 01:40:48 +00006150 RecordDecl *RD = ReadDeclAs<RecordDecl>(*Loc.F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00006151 RD = cast_or_null<RecordDecl>(RD->getCanonicalDecl());
6152 QualType T = Context.getRecordType(RD);
6153 const_cast<Type*>(T.getTypePtr())->setDependent(IsDependent);
6154 return T;
6155 }
6156
6157 case TYPE_ENUM: {
6158 if (Record.size() != 2) {
6159 Error("incorrect encoding of enum type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006160 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006161 }
6162 unsigned Idx = 0;
6163 bool IsDependent = Record[Idx++];
6164 QualType T
6165 = Context.getEnumType(ReadDeclAs<EnumDecl>(*Loc.F, Record, Idx));
6166 const_cast<Type*>(T.getTypePtr())->setDependent(IsDependent);
6167 return T;
6168 }
6169
6170 case TYPE_ATTRIBUTED: {
6171 if (Record.size() != 3) {
6172 Error("incorrect encoding of attributed type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006173 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006174 }
6175 QualType modifiedType = readType(*Loc.F, Record, Idx);
6176 QualType equivalentType = readType(*Loc.F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00006177 AttributedType::Kind kind = static_cast<AttributedType::Kind>(Record[2]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006178 return Context.getAttributedType(kind, modifiedType, equivalentType);
6179 }
6180
6181 case TYPE_PAREN: {
6182 if (Record.size() != 1) {
6183 Error("incorrect encoding of paren type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006184 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006185 }
6186 QualType InnerType = readType(*Loc.F, Record, Idx);
6187 return Context.getParenType(InnerType);
6188 }
6189
6190 case TYPE_PACK_EXPANSION: {
6191 if (Record.size() != 2) {
6192 Error("incorrect encoding of pack expansion type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006193 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006194 }
6195 QualType Pattern = readType(*Loc.F, Record, Idx);
6196 if (Pattern.isNull())
Vedant Kumar48b4f762018-04-14 01:40:48 +00006197 return QualType();
David Blaikie05785d12013-02-20 22:23:23 +00006198 Optional<unsigned> NumExpansions;
Guy Benyei11169dd2012-12-18 14:30:41 +00006199 if (Record[1])
6200 NumExpansions = Record[1] - 1;
6201 return Context.getPackExpansionType(Pattern, NumExpansions);
6202 }
6203
6204 case TYPE_ELABORATED: {
6205 unsigned Idx = 0;
Vedant Kumar48b4f762018-04-14 01:40:48 +00006206 ElaboratedTypeKeyword Keyword = (ElaboratedTypeKeyword)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00006207 NestedNameSpecifier *NNS = ReadNestedNameSpecifier(*Loc.F, Record, Idx);
6208 QualType NamedType = readType(*Loc.F, Record, Idx);
Joel E. Denny7509a2f2018-05-14 19:36:45 +00006209 TagDecl *OwnedTagDecl = ReadDeclAs<TagDecl>(*Loc.F, Record, Idx);
6210 return Context.getElaboratedType(Keyword, NNS, NamedType, OwnedTagDecl);
Guy Benyei11169dd2012-12-18 14:30:41 +00006211 }
6212
6213 case TYPE_OBJC_INTERFACE: {
6214 unsigned Idx = 0;
Vedant Kumar48b4f762018-04-14 01:40:48 +00006215 ObjCInterfaceDecl *ItfD
6216 = ReadDeclAs<ObjCInterfaceDecl>(*Loc.F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00006217 return Context.getObjCInterfaceType(ItfD->getCanonicalDecl());
6218 }
6219
Manman Rene6be26c2016-09-13 17:25:08 +00006220 case TYPE_OBJC_TYPE_PARAM: {
6221 unsigned Idx = 0;
Vedant Kumar48b4f762018-04-14 01:40:48 +00006222 ObjCTypeParamDecl *Decl
6223 = ReadDeclAs<ObjCTypeParamDecl>(*Loc.F, Record, Idx);
Manman Rene6be26c2016-09-13 17:25:08 +00006224 unsigned NumProtos = Record[Idx++];
6225 SmallVector<ObjCProtocolDecl*, 4> Protos;
6226 for (unsigned I = 0; I != NumProtos; ++I)
6227 Protos.push_back(ReadDeclAs<ObjCProtocolDecl>(*Loc.F, Record, Idx));
6228 return Context.getObjCTypeParamType(Decl, Protos);
6229 }
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006230
Guy Benyei11169dd2012-12-18 14:30:41 +00006231 case TYPE_OBJC_OBJECT: {
6232 unsigned Idx = 0;
6233 QualType Base = readType(*Loc.F, Record, Idx);
Douglas Gregore9d95f12015-07-07 03:57:35 +00006234 unsigned NumTypeArgs = Record[Idx++];
6235 SmallVector<QualType, 4> TypeArgs;
6236 for (unsigned I = 0; I != NumTypeArgs; ++I)
6237 TypeArgs.push_back(readType(*Loc.F, Record, Idx));
Guy Benyei11169dd2012-12-18 14:30:41 +00006238 unsigned NumProtos = Record[Idx++];
6239 SmallVector<ObjCProtocolDecl*, 4> Protos;
6240 for (unsigned I = 0; I != NumProtos; ++I)
6241 Protos.push_back(ReadDeclAs<ObjCProtocolDecl>(*Loc.F, Record, Idx));
Douglas Gregorab209d82015-07-07 03:58:42 +00006242 bool IsKindOf = Record[Idx++];
6243 return Context.getObjCObjectType(Base, TypeArgs, Protos, IsKindOf);
Guy Benyei11169dd2012-12-18 14:30:41 +00006244 }
6245
6246 case TYPE_OBJC_OBJECT_POINTER: {
6247 unsigned Idx = 0;
6248 QualType Pointee = readType(*Loc.F, Record, Idx);
6249 return Context.getObjCObjectPointerType(Pointee);
6250 }
6251
6252 case TYPE_SUBST_TEMPLATE_TYPE_PARM: {
6253 unsigned Idx = 0;
6254 QualType Parm = readType(*Loc.F, Record, Idx);
6255 QualType Replacement = readType(*Loc.F, Record, Idx);
Stephan Tolksdorfe96f8b32014-03-15 10:23:27 +00006256 return Context.getSubstTemplateTypeParmType(
6257 cast<TemplateTypeParmType>(Parm),
6258 Context.getCanonicalType(Replacement));
Guy Benyei11169dd2012-12-18 14:30:41 +00006259 }
6260
6261 case TYPE_SUBST_TEMPLATE_TYPE_PARM_PACK: {
6262 unsigned Idx = 0;
6263 QualType Parm = readType(*Loc.F, Record, Idx);
6264 TemplateArgument ArgPack = ReadTemplateArgument(*Loc.F, Record, Idx);
6265 return Context.getSubstTemplateTypeParmPackType(
6266 cast<TemplateTypeParmType>(Parm),
6267 ArgPack);
6268 }
6269
6270 case TYPE_INJECTED_CLASS_NAME: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00006271 CXXRecordDecl *D = ReadDeclAs<CXXRecordDecl>(*Loc.F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00006272 QualType TST = readType(*Loc.F, Record, Idx); // probably derivable
6273 // FIXME: ASTContext::getInjectedClassNameType is not currently suitable
6274 // for AST reading, too much interdependencies.
Richard Smith6377f8f2014-10-21 21:15:18 +00006275 const Type *T = nullptr;
6276 for (auto *DI = D; DI; DI = DI->getPreviousDecl()) {
6277 if (const Type *Existing = DI->getTypeForDecl()) {
6278 T = Existing;
6279 break;
6280 }
6281 }
6282 if (!T) {
Richard Smithf17fdbd2014-04-24 02:25:27 +00006283 T = new (Context, TypeAlignment) InjectedClassNameType(D, TST);
Richard Smith6377f8f2014-10-21 21:15:18 +00006284 for (auto *DI = D; DI; DI = DI->getPreviousDecl())
6285 DI->setTypeForDecl(T);
6286 }
Richard Smithf17fdbd2014-04-24 02:25:27 +00006287 return QualType(T, 0);
Guy Benyei11169dd2012-12-18 14:30:41 +00006288 }
6289
6290 case TYPE_TEMPLATE_TYPE_PARM: {
6291 unsigned Idx = 0;
6292 unsigned Depth = Record[Idx++];
6293 unsigned Index = Record[Idx++];
6294 bool Pack = Record[Idx++];
Vedant Kumar48b4f762018-04-14 01:40:48 +00006295 TemplateTypeParmDecl *D
6296 = ReadDeclAs<TemplateTypeParmDecl>(*Loc.F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00006297 return Context.getTemplateTypeParmType(Depth, Index, Pack, D);
6298 }
6299
6300 case TYPE_DEPENDENT_NAME: {
6301 unsigned Idx = 0;
Vedant Kumar48b4f762018-04-14 01:40:48 +00006302 ElaboratedTypeKeyword Keyword = (ElaboratedTypeKeyword)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00006303 NestedNameSpecifier *NNS = ReadNestedNameSpecifier(*Loc.F, Record, Idx);
Richard Smithbdf2d932015-07-30 03:37:16 +00006304 const IdentifierInfo *Name = GetIdentifierInfo(*Loc.F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00006305 QualType Canon = readType(*Loc.F, Record, Idx);
6306 if (!Canon.isNull())
6307 Canon = Context.getCanonicalType(Canon);
6308 return Context.getDependentNameType(Keyword, NNS, Name, Canon);
6309 }
6310
6311 case TYPE_DEPENDENT_TEMPLATE_SPECIALIZATION: {
6312 unsigned Idx = 0;
Vedant Kumar48b4f762018-04-14 01:40:48 +00006313 ElaboratedTypeKeyword Keyword = (ElaboratedTypeKeyword)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00006314 NestedNameSpecifier *NNS = ReadNestedNameSpecifier(*Loc.F, Record, Idx);
Richard Smithbdf2d932015-07-30 03:37:16 +00006315 const IdentifierInfo *Name = GetIdentifierInfo(*Loc.F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00006316 unsigned NumArgs = Record[Idx++];
6317 SmallVector<TemplateArgument, 8> Args;
6318 Args.reserve(NumArgs);
6319 while (NumArgs--)
6320 Args.push_back(ReadTemplateArgument(*Loc.F, Record, Idx));
6321 return Context.getDependentTemplateSpecializationType(Keyword, NNS, Name,
David Majnemer6fbeee32016-07-07 04:43:07 +00006322 Args);
Guy Benyei11169dd2012-12-18 14:30:41 +00006323 }
6324
6325 case TYPE_DEPENDENT_SIZED_ARRAY: {
6326 unsigned Idx = 0;
6327
6328 // ArrayType
6329 QualType ElementType = readType(*Loc.F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00006330 ArrayType::ArraySizeModifier ASM
6331 = (ArrayType::ArraySizeModifier)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00006332 unsigned IndexTypeQuals = Record[Idx++];
6333
6334 // DependentSizedArrayType
6335 Expr *NumElts = ReadExpr(*Loc.F);
6336 SourceRange Brackets = ReadSourceRange(*Loc.F, Record, Idx);
6337
6338 return Context.getDependentSizedArrayType(ElementType, NumElts, ASM,
6339 IndexTypeQuals, Brackets);
6340 }
6341
6342 case TYPE_TEMPLATE_SPECIALIZATION: {
6343 unsigned Idx = 0;
6344 bool IsDependent = Record[Idx++];
6345 TemplateName Name = ReadTemplateName(*Loc.F, Record, Idx);
6346 SmallVector<TemplateArgument, 8> Args;
6347 ReadTemplateArgumentList(Args, *Loc.F, Record, Idx);
6348 QualType Underlying = readType(*Loc.F, Record, Idx);
6349 QualType T;
6350 if (Underlying.isNull())
David Majnemer6fbeee32016-07-07 04:43:07 +00006351 T = Context.getCanonicalTemplateSpecializationType(Name, Args);
Guy Benyei11169dd2012-12-18 14:30:41 +00006352 else
David Majnemer6fbeee32016-07-07 04:43:07 +00006353 T = Context.getTemplateSpecializationType(Name, Args, Underlying);
Guy Benyei11169dd2012-12-18 14:30:41 +00006354 const_cast<Type*>(T.getTypePtr())->setDependent(IsDependent);
6355 return T;
6356 }
6357
6358 case TYPE_ATOMIC: {
6359 if (Record.size() != 1) {
6360 Error("Incorrect encoding of atomic type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006361 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006362 }
6363 QualType ValueType = readType(*Loc.F, Record, Idx);
6364 return Context.getAtomicType(ValueType);
6365 }
Xiuli Pan9c14e282016-01-09 12:53:17 +00006366
Joey Goulye3c85de2016-12-01 11:30:49 +00006367 case TYPE_PIPE: {
6368 if (Record.size() != 2) {
Xiuli Pan9c14e282016-01-09 12:53:17 +00006369 Error("Incorrect encoding of pipe type");
Vedant Kumar48b4f762018-04-14 01:40:48 +00006370 return QualType();
Xiuli Pan9c14e282016-01-09 12:53:17 +00006371 }
6372
6373 // Reading the pipe element type.
6374 QualType ElementType = readType(*Loc.F, Record, Idx);
Joey Goulye3c85de2016-12-01 11:30:49 +00006375 unsigned ReadOnly = Record[1];
6376 return Context.getPipeType(ElementType, ReadOnly);
Joey Gouly5788b782016-11-18 14:10:54 +00006377 }
6378
Erich Keanef702b022018-07-13 19:46:04 +00006379 case TYPE_DEPENDENT_SIZED_VECTOR: {
6380 unsigned Idx = 0;
6381 QualType ElementType = readType(*Loc.F, Record, Idx);
6382 Expr *SizeExpr = ReadExpr(*Loc.F);
6383 SourceLocation AttrLoc = ReadSourceLocation(*Loc.F, Record, Idx);
6384 unsigned VecKind = Record[Idx];
6385
6386 return Context.getDependentVectorType(ElementType, SizeExpr, AttrLoc,
6387 (VectorType::VectorKind)VecKind);
6388 }
6389
Alex Lorenz00aee432017-03-22 10:04:48 +00006390 case TYPE_DEPENDENT_SIZED_EXT_VECTOR: {
6391 unsigned Idx = 0;
6392
6393 // DependentSizedExtVectorType
6394 QualType ElementType = readType(*Loc.F, Record, Idx);
6395 Expr *SizeExpr = ReadExpr(*Loc.F);
6396 SourceLocation AttrLoc = ReadSourceLocation(*Loc.F, Record, Idx);
6397
6398 return Context.getDependentSizedExtVectorType(ElementType, SizeExpr,
6399 AttrLoc);
6400 }
Andrew Gozillon572bbb02017-10-02 06:25:51 +00006401
6402 case TYPE_DEPENDENT_ADDRESS_SPACE: {
6403 unsigned Idx = 0;
6404
6405 // DependentAddressSpaceType
6406 QualType PointeeType = readType(*Loc.F, Record, Idx);
6407 Expr *AddrSpaceExpr = ReadExpr(*Loc.F);
6408 SourceLocation AttrLoc = ReadSourceLocation(*Loc.F, Record, Idx);
6409
6410 return Context.getDependentAddressSpaceType(PointeeType, AddrSpaceExpr,
6411 AttrLoc);
6412 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006413 }
6414 llvm_unreachable("Invalid TypeCode!");
6415}
6416
Richard Smith564417a2014-03-20 21:47:22 +00006417void ASTReader::readExceptionSpec(ModuleFile &ModuleFile,
6418 SmallVectorImpl<QualType> &Exceptions,
Richard Smith8acb4282014-07-31 21:57:55 +00006419 FunctionProtoType::ExceptionSpecInfo &ESI,
Richard Smith564417a2014-03-20 21:47:22 +00006420 const RecordData &Record, unsigned &Idx) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00006421 ExceptionSpecificationType EST =
6422 static_cast<ExceptionSpecificationType>(Record[Idx++]);
Richard Smith8acb4282014-07-31 21:57:55 +00006423 ESI.Type = EST;
Richard Smith564417a2014-03-20 21:47:22 +00006424 if (EST == EST_Dynamic) {
Richard Smith8acb4282014-07-31 21:57:55 +00006425 for (unsigned I = 0, N = Record[Idx++]; I != N; ++I)
Richard Smith564417a2014-03-20 21:47:22 +00006426 Exceptions.push_back(readType(ModuleFile, Record, Idx));
Richard Smith8acb4282014-07-31 21:57:55 +00006427 ESI.Exceptions = Exceptions;
Richard Smitheaf11ad2018-05-03 03:58:32 +00006428 } else if (isComputedNoexcept(EST)) {
Richard Smith8acb4282014-07-31 21:57:55 +00006429 ESI.NoexceptExpr = ReadExpr(ModuleFile);
Richard Smith564417a2014-03-20 21:47:22 +00006430 } else if (EST == EST_Uninstantiated) {
Richard Smith8acb4282014-07-31 21:57:55 +00006431 ESI.SourceDecl = ReadDeclAs<FunctionDecl>(ModuleFile, Record, Idx);
6432 ESI.SourceTemplate = ReadDeclAs<FunctionDecl>(ModuleFile, Record, Idx);
Richard Smith564417a2014-03-20 21:47:22 +00006433 } else if (EST == EST_Unevaluated) {
Richard Smith8acb4282014-07-31 21:57:55 +00006434 ESI.SourceDecl = ReadDeclAs<FunctionDecl>(ModuleFile, Record, Idx);
Richard Smith564417a2014-03-20 21:47:22 +00006435 }
6436}
6437
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006438namespace clang {
6439
6440class TypeLocReader : public TypeLocVisitor<TypeLocReader> {
David L. Jonesbe1557a2016-12-21 00:17:49 +00006441 ModuleFile *F;
6442 ASTReader *Reader;
6443 const ASTReader::RecordData &Record;
Guy Benyei11169dd2012-12-18 14:30:41 +00006444 unsigned &Idx;
6445
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006446 SourceLocation ReadSourceLocation() {
David L. Jonesbe1557a2016-12-21 00:17:49 +00006447 return Reader->ReadSourceLocation(*F, Record, Idx);
6448 }
6449
6450 TypeSourceInfo *GetTypeSourceInfo() {
6451 return Reader->GetTypeSourceInfo(*F, Record, Idx);
6452 }
6453
6454 NestedNameSpecifierLoc ReadNestedNameSpecifierLoc() {
6455 return Reader->ReadNestedNameSpecifierLoc(*F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00006456 }
6457
Richard Smithe43e2b32018-08-20 21:47:29 +00006458 Attr *ReadAttr() {
6459 return Reader->ReadAttr(*F, Record, Idx);
6460 }
6461
Guy Benyei11169dd2012-12-18 14:30:41 +00006462public:
David L. Jonesbe1557a2016-12-21 00:17:49 +00006463 TypeLocReader(ModuleFile &F, ASTReader &Reader,
Guy Benyei11169dd2012-12-18 14:30:41 +00006464 const ASTReader::RecordData &Record, unsigned &Idx)
David L. Jonesbe1557a2016-12-21 00:17:49 +00006465 : F(&F), Reader(&Reader), Record(Record), Idx(Idx) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00006466
6467 // We want compile-time assurance that we've enumerated all of
6468 // these, so unfortunately we have to declare them first, then
6469 // define them out-of-line.
6470#define ABSTRACT_TYPELOC(CLASS, PARENT)
6471#define TYPELOC(CLASS, PARENT) \
6472 void Visit##CLASS##TypeLoc(CLASS##TypeLoc TyLoc);
6473#include "clang/AST/TypeLocNodes.def"
6474
6475 void VisitFunctionTypeLoc(FunctionTypeLoc);
6476 void VisitArrayTypeLoc(ArrayTypeLoc);
6477};
6478
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006479} // namespace clang
6480
Guy Benyei11169dd2012-12-18 14:30:41 +00006481void TypeLocReader::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
6482 // nothing to do
6483}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006484
Guy Benyei11169dd2012-12-18 14:30:41 +00006485void TypeLocReader::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006486 TL.setBuiltinLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006487 if (TL.needsExtraLocalData()) {
David L. Jonesbe1557a2016-12-21 00:17:49 +00006488 TL.setWrittenTypeSpec(static_cast<DeclSpec::TST>(Record[Idx++]));
6489 TL.setWrittenSignSpec(static_cast<DeclSpec::TSS>(Record[Idx++]));
6490 TL.setWrittenWidthSpec(static_cast<DeclSpec::TSW>(Record[Idx++]));
6491 TL.setModeAttr(Record[Idx++]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006492 }
6493}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006494
Guy Benyei11169dd2012-12-18 14:30:41 +00006495void TypeLocReader::VisitComplexTypeLoc(ComplexTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006496 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006497}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006498
Guy Benyei11169dd2012-12-18 14:30:41 +00006499void TypeLocReader::VisitPointerTypeLoc(PointerTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006500 TL.setStarLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006501}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006502
Reid Kleckner8a365022013-06-24 17:51:48 +00006503void TypeLocReader::VisitDecayedTypeLoc(DecayedTypeLoc TL) {
6504 // nothing to do
6505}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006506
Reid Kleckner0503a872013-12-05 01:23:43 +00006507void TypeLocReader::VisitAdjustedTypeLoc(AdjustedTypeLoc TL) {
6508 // nothing to do
6509}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006510
Guy Benyei11169dd2012-12-18 14:30:41 +00006511void TypeLocReader::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006512 TL.setCaretLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006513}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006514
Guy Benyei11169dd2012-12-18 14:30:41 +00006515void TypeLocReader::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006516 TL.setAmpLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006517}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006518
Guy Benyei11169dd2012-12-18 14:30:41 +00006519void TypeLocReader::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006520 TL.setAmpAmpLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006521}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006522
Guy Benyei11169dd2012-12-18 14:30:41 +00006523void TypeLocReader::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006524 TL.setStarLoc(ReadSourceLocation());
David L. Jonesbe1557a2016-12-21 00:17:49 +00006525 TL.setClassTInfo(GetTypeSourceInfo());
Guy Benyei11169dd2012-12-18 14:30:41 +00006526}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006527
Guy Benyei11169dd2012-12-18 14:30:41 +00006528void TypeLocReader::VisitArrayTypeLoc(ArrayTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006529 TL.setLBracketLoc(ReadSourceLocation());
6530 TL.setRBracketLoc(ReadSourceLocation());
David L. Jonesbe1557a2016-12-21 00:17:49 +00006531 if (Record[Idx++])
6532 TL.setSizeExpr(Reader->ReadExpr(*F));
Guy Benyei11169dd2012-12-18 14:30:41 +00006533 else
Craig Toppera13603a2014-05-22 05:54:18 +00006534 TL.setSizeExpr(nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00006535}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006536
Guy Benyei11169dd2012-12-18 14:30:41 +00006537void TypeLocReader::VisitConstantArrayTypeLoc(ConstantArrayTypeLoc TL) {
6538 VisitArrayTypeLoc(TL);
6539}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006540
Guy Benyei11169dd2012-12-18 14:30:41 +00006541void TypeLocReader::VisitIncompleteArrayTypeLoc(IncompleteArrayTypeLoc TL) {
6542 VisitArrayTypeLoc(TL);
6543}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006544
Guy Benyei11169dd2012-12-18 14:30:41 +00006545void TypeLocReader::VisitVariableArrayTypeLoc(VariableArrayTypeLoc TL) {
6546 VisitArrayTypeLoc(TL);
6547}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006548
Guy Benyei11169dd2012-12-18 14:30:41 +00006549void TypeLocReader::VisitDependentSizedArrayTypeLoc(
6550 DependentSizedArrayTypeLoc TL) {
6551 VisitArrayTypeLoc(TL);
6552}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006553
Andrew Gozillon572bbb02017-10-02 06:25:51 +00006554void TypeLocReader::VisitDependentAddressSpaceTypeLoc(
6555 DependentAddressSpaceTypeLoc TL) {
6556
6557 TL.setAttrNameLoc(ReadSourceLocation());
6558 SourceRange range;
6559 range.setBegin(ReadSourceLocation());
6560 range.setEnd(ReadSourceLocation());
6561 TL.setAttrOperandParensRange(range);
6562 TL.setAttrExprOperand(Reader->ReadExpr(*F));
6563}
6564
Guy Benyei11169dd2012-12-18 14:30:41 +00006565void TypeLocReader::VisitDependentSizedExtVectorTypeLoc(
6566 DependentSizedExtVectorTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006567 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006568}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006569
Guy Benyei11169dd2012-12-18 14:30:41 +00006570void TypeLocReader::VisitVectorTypeLoc(VectorTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006571 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006572}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006573
Erich Keanef702b022018-07-13 19:46:04 +00006574void TypeLocReader::VisitDependentVectorTypeLoc(
6575 DependentVectorTypeLoc TL) {
6576 TL.setNameLoc(ReadSourceLocation());
6577}
6578
Guy Benyei11169dd2012-12-18 14:30:41 +00006579void TypeLocReader::VisitExtVectorTypeLoc(ExtVectorTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006580 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006581}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006582
Guy Benyei11169dd2012-12-18 14:30:41 +00006583void TypeLocReader::VisitFunctionTypeLoc(FunctionTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006584 TL.setLocalRangeBegin(ReadSourceLocation());
6585 TL.setLParenLoc(ReadSourceLocation());
6586 TL.setRParenLoc(ReadSourceLocation());
Malcolm Parsonsa3220ce2017-01-12 16:11:28 +00006587 TL.setExceptionSpecRange(SourceRange(Reader->ReadSourceLocation(*F, Record, Idx),
6588 Reader->ReadSourceLocation(*F, Record, Idx)));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006589 TL.setLocalRangeEnd(ReadSourceLocation());
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00006590 for (unsigned i = 0, e = TL.getNumParams(); i != e; ++i) {
David L. Jonesbe1557a2016-12-21 00:17:49 +00006591 TL.setParam(i, Reader->ReadDeclAs<ParmVarDecl>(*F, Record, Idx));
Guy Benyei11169dd2012-12-18 14:30:41 +00006592 }
6593}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006594
Guy Benyei11169dd2012-12-18 14:30:41 +00006595void TypeLocReader::VisitFunctionProtoTypeLoc(FunctionProtoTypeLoc TL) {
6596 VisitFunctionTypeLoc(TL);
6597}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006598
Guy Benyei11169dd2012-12-18 14:30:41 +00006599void TypeLocReader::VisitFunctionNoProtoTypeLoc(FunctionNoProtoTypeLoc TL) {
6600 VisitFunctionTypeLoc(TL);
6601}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006602
Guy Benyei11169dd2012-12-18 14:30:41 +00006603void TypeLocReader::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006604 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006605}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006606
Guy Benyei11169dd2012-12-18 14:30:41 +00006607void TypeLocReader::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006608 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006609}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006610
Guy Benyei11169dd2012-12-18 14:30:41 +00006611void TypeLocReader::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006612 TL.setTypeofLoc(ReadSourceLocation());
6613 TL.setLParenLoc(ReadSourceLocation());
6614 TL.setRParenLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006615}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006616
Guy Benyei11169dd2012-12-18 14:30:41 +00006617void TypeLocReader::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006618 TL.setTypeofLoc(ReadSourceLocation());
6619 TL.setLParenLoc(ReadSourceLocation());
6620 TL.setRParenLoc(ReadSourceLocation());
David L. Jonesbe1557a2016-12-21 00:17:49 +00006621 TL.setUnderlyingTInfo(GetTypeSourceInfo());
Guy Benyei11169dd2012-12-18 14:30:41 +00006622}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006623
Guy Benyei11169dd2012-12-18 14:30:41 +00006624void TypeLocReader::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006625 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006626}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006627
Guy Benyei11169dd2012-12-18 14:30:41 +00006628void TypeLocReader::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006629 TL.setKWLoc(ReadSourceLocation());
6630 TL.setLParenLoc(ReadSourceLocation());
6631 TL.setRParenLoc(ReadSourceLocation());
David L. Jonesbe1557a2016-12-21 00:17:49 +00006632 TL.setUnderlyingTInfo(GetTypeSourceInfo());
Guy Benyei11169dd2012-12-18 14:30:41 +00006633}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006634
Guy Benyei11169dd2012-12-18 14:30:41 +00006635void TypeLocReader::VisitAutoTypeLoc(AutoTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006636 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006637}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006638
Richard Smith600b5262017-01-26 20:40:47 +00006639void TypeLocReader::VisitDeducedTemplateSpecializationTypeLoc(
6640 DeducedTemplateSpecializationTypeLoc TL) {
6641 TL.setTemplateNameLoc(ReadSourceLocation());
6642}
6643
Guy Benyei11169dd2012-12-18 14:30:41 +00006644void TypeLocReader::VisitRecordTypeLoc(RecordTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006645 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006646}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006647
Guy Benyei11169dd2012-12-18 14:30:41 +00006648void TypeLocReader::VisitEnumTypeLoc(EnumTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006649 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006650}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006651
Guy Benyei11169dd2012-12-18 14:30:41 +00006652void TypeLocReader::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
Richard Smithe43e2b32018-08-20 21:47:29 +00006653 TL.setAttr(ReadAttr());
Guy Benyei11169dd2012-12-18 14:30:41 +00006654}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006655
Guy Benyei11169dd2012-12-18 14:30:41 +00006656void TypeLocReader::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006657 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006658}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006659
Guy Benyei11169dd2012-12-18 14:30:41 +00006660void TypeLocReader::VisitSubstTemplateTypeParmTypeLoc(
6661 SubstTemplateTypeParmTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006662 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006663}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006664
Guy Benyei11169dd2012-12-18 14:30:41 +00006665void TypeLocReader::VisitSubstTemplateTypeParmPackTypeLoc(
6666 SubstTemplateTypeParmPackTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006667 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006668}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006669
Guy Benyei11169dd2012-12-18 14:30:41 +00006670void TypeLocReader::VisitTemplateSpecializationTypeLoc(
6671 TemplateSpecializationTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006672 TL.setTemplateKeywordLoc(ReadSourceLocation());
6673 TL.setTemplateNameLoc(ReadSourceLocation());
6674 TL.setLAngleLoc(ReadSourceLocation());
6675 TL.setRAngleLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006676 for (unsigned i = 0, e = TL.getNumArgs(); i != e; ++i)
David L. Jonesbe1557a2016-12-21 00:17:49 +00006677 TL.setArgLocInfo(
6678 i,
6679 Reader->GetTemplateArgumentLocInfo(
6680 *F, TL.getTypePtr()->getArg(i).getKind(), Record, Idx));
Guy Benyei11169dd2012-12-18 14:30:41 +00006681}
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00006682
Guy Benyei11169dd2012-12-18 14:30:41 +00006683void TypeLocReader::VisitParenTypeLoc(ParenTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006684 TL.setLParenLoc(ReadSourceLocation());
6685 TL.setRParenLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006686}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006687
Guy Benyei11169dd2012-12-18 14:30:41 +00006688void TypeLocReader::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006689 TL.setElaboratedKeywordLoc(ReadSourceLocation());
David L. Jonesbe1557a2016-12-21 00:17:49 +00006690 TL.setQualifierLoc(ReadNestedNameSpecifierLoc());
Guy Benyei11169dd2012-12-18 14:30:41 +00006691}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006692
Guy Benyei11169dd2012-12-18 14:30:41 +00006693void TypeLocReader::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006694 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006695}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006696
Guy Benyei11169dd2012-12-18 14:30:41 +00006697void TypeLocReader::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006698 TL.setElaboratedKeywordLoc(ReadSourceLocation());
David L. Jonesbe1557a2016-12-21 00:17:49 +00006699 TL.setQualifierLoc(ReadNestedNameSpecifierLoc());
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006700 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006701}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006702
Guy Benyei11169dd2012-12-18 14:30:41 +00006703void TypeLocReader::VisitDependentTemplateSpecializationTypeLoc(
6704 DependentTemplateSpecializationTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006705 TL.setElaboratedKeywordLoc(ReadSourceLocation());
David L. Jonesbe1557a2016-12-21 00:17:49 +00006706 TL.setQualifierLoc(ReadNestedNameSpecifierLoc());
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006707 TL.setTemplateKeywordLoc(ReadSourceLocation());
6708 TL.setTemplateNameLoc(ReadSourceLocation());
6709 TL.setLAngleLoc(ReadSourceLocation());
6710 TL.setRAngleLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006711 for (unsigned I = 0, E = TL.getNumArgs(); I != E; ++I)
David L. Jonesbe1557a2016-12-21 00:17:49 +00006712 TL.setArgLocInfo(
6713 I,
6714 Reader->GetTemplateArgumentLocInfo(
6715 *F, TL.getTypePtr()->getArg(I).getKind(), Record, Idx));
Guy Benyei11169dd2012-12-18 14:30:41 +00006716}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006717
Guy Benyei11169dd2012-12-18 14:30:41 +00006718void TypeLocReader::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006719 TL.setEllipsisLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006720}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006721
Guy Benyei11169dd2012-12-18 14:30:41 +00006722void TypeLocReader::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006723 TL.setNameLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006724}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006725
Manman Rene6be26c2016-09-13 17:25:08 +00006726void TypeLocReader::VisitObjCTypeParamTypeLoc(ObjCTypeParamTypeLoc TL) {
6727 if (TL.getNumProtocols()) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006728 TL.setProtocolLAngleLoc(ReadSourceLocation());
6729 TL.setProtocolRAngleLoc(ReadSourceLocation());
Manman Rene6be26c2016-09-13 17:25:08 +00006730 }
6731 for (unsigned i = 0, e = TL.getNumProtocols(); i != e; ++i)
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006732 TL.setProtocolLoc(i, ReadSourceLocation());
Manman Rene6be26c2016-09-13 17:25:08 +00006733}
6734
Guy Benyei11169dd2012-12-18 14:30:41 +00006735void TypeLocReader::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
David L. Jonesbe1557a2016-12-21 00:17:49 +00006736 TL.setHasBaseTypeAsWritten(Record[Idx++]);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006737 TL.setTypeArgsLAngleLoc(ReadSourceLocation());
6738 TL.setTypeArgsRAngleLoc(ReadSourceLocation());
Douglas Gregore9d95f12015-07-07 03:57:35 +00006739 for (unsigned i = 0, e = TL.getNumTypeArgs(); i != e; ++i)
David L. Jonesbe1557a2016-12-21 00:17:49 +00006740 TL.setTypeArgTInfo(i, GetTypeSourceInfo());
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006741 TL.setProtocolLAngleLoc(ReadSourceLocation());
6742 TL.setProtocolRAngleLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006743 for (unsigned i = 0, e = TL.getNumProtocols(); i != e; ++i)
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006744 TL.setProtocolLoc(i, ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006745}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006746
Guy Benyei11169dd2012-12-18 14:30:41 +00006747void TypeLocReader::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006748 TL.setStarLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006749}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006750
Guy Benyei11169dd2012-12-18 14:30:41 +00006751void TypeLocReader::VisitAtomicTypeLoc(AtomicTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006752 TL.setKWLoc(ReadSourceLocation());
6753 TL.setLParenLoc(ReadSourceLocation());
6754 TL.setRParenLoc(ReadSourceLocation());
Guy Benyei11169dd2012-12-18 14:30:41 +00006755}
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00006756
Xiuli Pan9c14e282016-01-09 12:53:17 +00006757void TypeLocReader::VisitPipeTypeLoc(PipeTypeLoc TL) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00006758 TL.setKWLoc(ReadSourceLocation());
Xiuli Pan9c14e282016-01-09 12:53:17 +00006759}
Guy Benyei11169dd2012-12-18 14:30:41 +00006760
Richard Smithc23d7342018-06-29 20:46:25 +00006761void ASTReader::ReadTypeLoc(ModuleFile &F, const ASTReader::RecordData &Record,
6762 unsigned &Idx, TypeLoc TL) {
6763 TypeLocReader TLR(F, *this, Record, Idx);
6764 for (; !TL.isNull(); TL = TL.getNextTypeLoc())
6765 TLR.Visit(TL);
6766}
6767
David L. Jonesbe1557a2016-12-21 00:17:49 +00006768TypeSourceInfo *
6769ASTReader::GetTypeSourceInfo(ModuleFile &F, const ASTReader::RecordData &Record,
6770 unsigned &Idx) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006771 QualType InfoTy = readType(F, Record, Idx);
6772 if (InfoTy.isNull())
Craig Toppera13603a2014-05-22 05:54:18 +00006773 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006774
6775 TypeSourceInfo *TInfo = getContext().CreateTypeSourceInfo(InfoTy);
Richard Smithc23d7342018-06-29 20:46:25 +00006776 ReadTypeLoc(F, Record, Idx, TInfo->getTypeLoc());
Guy Benyei11169dd2012-12-18 14:30:41 +00006777 return TInfo;
6778}
6779
6780QualType ASTReader::GetType(TypeID ID) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00006781 assert(ContextObj && "reading type with no AST context");
6782 ASTContext &Context = *ContextObj;
6783
Guy Benyei11169dd2012-12-18 14:30:41 +00006784 unsigned FastQuals = ID & Qualifiers::FastMask;
6785 unsigned Index = ID >> Qualifiers::FastWidth;
6786
6787 if (Index < NUM_PREDEF_TYPE_IDS) {
6788 QualType T;
6789 switch ((PredefinedTypeIDs)Index) {
Alexey Baderbdf7c842015-09-15 12:18:29 +00006790 case PREDEF_TYPE_NULL_ID:
Vedant Kumar48b4f762018-04-14 01:40:48 +00006791 return QualType();
Alexey Baderbdf7c842015-09-15 12:18:29 +00006792 case PREDEF_TYPE_VOID_ID:
6793 T = Context.VoidTy;
6794 break;
6795 case PREDEF_TYPE_BOOL_ID:
6796 T = Context.BoolTy;
6797 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00006798 case PREDEF_TYPE_CHAR_U_ID:
6799 case PREDEF_TYPE_CHAR_S_ID:
6800 // FIXME: Check that the signedness of CharTy is correct!
6801 T = Context.CharTy;
6802 break;
Alexey Baderbdf7c842015-09-15 12:18:29 +00006803 case PREDEF_TYPE_UCHAR_ID:
6804 T = Context.UnsignedCharTy;
6805 break;
6806 case PREDEF_TYPE_USHORT_ID:
6807 T = Context.UnsignedShortTy;
6808 break;
6809 case PREDEF_TYPE_UINT_ID:
6810 T = Context.UnsignedIntTy;
6811 break;
6812 case PREDEF_TYPE_ULONG_ID:
6813 T = Context.UnsignedLongTy;
6814 break;
6815 case PREDEF_TYPE_ULONGLONG_ID:
6816 T = Context.UnsignedLongLongTy;
6817 break;
6818 case PREDEF_TYPE_UINT128_ID:
6819 T = Context.UnsignedInt128Ty;
6820 break;
6821 case PREDEF_TYPE_SCHAR_ID:
6822 T = Context.SignedCharTy;
6823 break;
6824 case PREDEF_TYPE_WCHAR_ID:
6825 T = Context.WCharTy;
6826 break;
6827 case PREDEF_TYPE_SHORT_ID:
6828 T = Context.ShortTy;
6829 break;
6830 case PREDEF_TYPE_INT_ID:
6831 T = Context.IntTy;
6832 break;
6833 case PREDEF_TYPE_LONG_ID:
6834 T = Context.LongTy;
6835 break;
6836 case PREDEF_TYPE_LONGLONG_ID:
6837 T = Context.LongLongTy;
6838 break;
6839 case PREDEF_TYPE_INT128_ID:
6840 T = Context.Int128Ty;
6841 break;
6842 case PREDEF_TYPE_HALF_ID:
6843 T = Context.HalfTy;
6844 break;
6845 case PREDEF_TYPE_FLOAT_ID:
6846 T = Context.FloatTy;
6847 break;
6848 case PREDEF_TYPE_DOUBLE_ID:
6849 T = Context.DoubleTy;
6850 break;
6851 case PREDEF_TYPE_LONGDOUBLE_ID:
6852 T = Context.LongDoubleTy;
6853 break;
Leonard Chanf921d852018-06-04 16:07:52 +00006854 case PREDEF_TYPE_SHORT_ACCUM_ID:
6855 T = Context.ShortAccumTy;
6856 break;
6857 case PREDEF_TYPE_ACCUM_ID:
6858 T = Context.AccumTy;
6859 break;
6860 case PREDEF_TYPE_LONG_ACCUM_ID:
6861 T = Context.LongAccumTy;
6862 break;
6863 case PREDEF_TYPE_USHORT_ACCUM_ID:
6864 T = Context.UnsignedShortAccumTy;
6865 break;
6866 case PREDEF_TYPE_UACCUM_ID:
6867 T = Context.UnsignedAccumTy;
6868 break;
6869 case PREDEF_TYPE_ULONG_ACCUM_ID:
6870 T = Context.UnsignedLongAccumTy;
6871 break;
Leonard Chanab80f3c2018-06-14 14:53:51 +00006872 case PREDEF_TYPE_SHORT_FRACT_ID:
6873 T = Context.ShortFractTy;
6874 break;
6875 case PREDEF_TYPE_FRACT_ID:
6876 T = Context.FractTy;
6877 break;
6878 case PREDEF_TYPE_LONG_FRACT_ID:
6879 T = Context.LongFractTy;
6880 break;
6881 case PREDEF_TYPE_USHORT_FRACT_ID:
6882 T = Context.UnsignedShortFractTy;
6883 break;
6884 case PREDEF_TYPE_UFRACT_ID:
6885 T = Context.UnsignedFractTy;
6886 break;
6887 case PREDEF_TYPE_ULONG_FRACT_ID:
6888 T = Context.UnsignedLongFractTy;
6889 break;
6890 case PREDEF_TYPE_SAT_SHORT_ACCUM_ID:
6891 T = Context.SatShortAccumTy;
6892 break;
6893 case PREDEF_TYPE_SAT_ACCUM_ID:
6894 T = Context.SatAccumTy;
6895 break;
6896 case PREDEF_TYPE_SAT_LONG_ACCUM_ID:
6897 T = Context.SatLongAccumTy;
6898 break;
6899 case PREDEF_TYPE_SAT_USHORT_ACCUM_ID:
6900 T = Context.SatUnsignedShortAccumTy;
6901 break;
6902 case PREDEF_TYPE_SAT_UACCUM_ID:
6903 T = Context.SatUnsignedAccumTy;
6904 break;
6905 case PREDEF_TYPE_SAT_ULONG_ACCUM_ID:
6906 T = Context.SatUnsignedLongAccumTy;
6907 break;
6908 case PREDEF_TYPE_SAT_SHORT_FRACT_ID:
6909 T = Context.SatShortFractTy;
6910 break;
6911 case PREDEF_TYPE_SAT_FRACT_ID:
6912 T = Context.SatFractTy;
6913 break;
6914 case PREDEF_TYPE_SAT_LONG_FRACT_ID:
6915 T = Context.SatLongFractTy;
6916 break;
6917 case PREDEF_TYPE_SAT_USHORT_FRACT_ID:
6918 T = Context.SatUnsignedShortFractTy;
6919 break;
6920 case PREDEF_TYPE_SAT_UFRACT_ID:
6921 T = Context.SatUnsignedFractTy;
6922 break;
6923 case PREDEF_TYPE_SAT_ULONG_FRACT_ID:
6924 T = Context.SatUnsignedLongFractTy;
6925 break;
Sjoerd Meijercc623ad2017-09-08 15:15:00 +00006926 case PREDEF_TYPE_FLOAT16_ID:
6927 T = Context.Float16Ty;
6928 break;
Nemanja Ivanovicbb1ea2d2016-05-09 08:52:33 +00006929 case PREDEF_TYPE_FLOAT128_ID:
6930 T = Context.Float128Ty;
6931 break;
Alexey Baderbdf7c842015-09-15 12:18:29 +00006932 case PREDEF_TYPE_OVERLOAD_ID:
6933 T = Context.OverloadTy;
6934 break;
6935 case PREDEF_TYPE_BOUND_MEMBER:
6936 T = Context.BoundMemberTy;
6937 break;
6938 case PREDEF_TYPE_PSEUDO_OBJECT:
6939 T = Context.PseudoObjectTy;
6940 break;
6941 case PREDEF_TYPE_DEPENDENT_ID:
6942 T = Context.DependentTy;
6943 break;
6944 case PREDEF_TYPE_UNKNOWN_ANY:
6945 T = Context.UnknownAnyTy;
6946 break;
6947 case PREDEF_TYPE_NULLPTR_ID:
6948 T = Context.NullPtrTy;
6949 break;
Richard Smith3a8244d2018-05-01 05:02:45 +00006950 case PREDEF_TYPE_CHAR8_ID:
6951 T = Context.Char8Ty;
6952 break;
Alexey Baderbdf7c842015-09-15 12:18:29 +00006953 case PREDEF_TYPE_CHAR16_ID:
6954 T = Context.Char16Ty;
6955 break;
6956 case PREDEF_TYPE_CHAR32_ID:
6957 T = Context.Char32Ty;
6958 break;
6959 case PREDEF_TYPE_OBJC_ID:
6960 T = Context.ObjCBuiltinIdTy;
6961 break;
6962 case PREDEF_TYPE_OBJC_CLASS:
6963 T = Context.ObjCBuiltinClassTy;
6964 break;
6965 case PREDEF_TYPE_OBJC_SEL:
6966 T = Context.ObjCBuiltinSelTy;
6967 break;
Alexey Bader954ba212016-04-08 13:40:33 +00006968#define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
6969 case PREDEF_TYPE_##Id##_ID: \
6970 T = Context.SingletonId; \
Alexey Baderbdf7c842015-09-15 12:18:29 +00006971 break;
Alexey Baderb62f1442016-04-13 08:33:41 +00006972#include "clang/Basic/OpenCLImageTypes.def"
Alexey Baderbdf7c842015-09-15 12:18:29 +00006973 case PREDEF_TYPE_SAMPLER_ID:
6974 T = Context.OCLSamplerTy;
6975 break;
6976 case PREDEF_TYPE_EVENT_ID:
6977 T = Context.OCLEventTy;
6978 break;
Alexey Bader9c8453f2015-09-15 11:18:52 +00006979 case PREDEF_TYPE_CLK_EVENT_ID:
6980 T = Context.OCLClkEventTy;
6981 break;
6982 case PREDEF_TYPE_QUEUE_ID:
6983 T = Context.OCLQueueTy;
6984 break;
Alexey Bader9c8453f2015-09-15 11:18:52 +00006985 case PREDEF_TYPE_RESERVE_ID_ID:
6986 T = Context.OCLReserveIDTy;
6987 break;
Alexey Baderbdf7c842015-09-15 12:18:29 +00006988 case PREDEF_TYPE_AUTO_DEDUCT:
6989 T = Context.getAutoDeductType();
6990 break;
Alexey Baderbdf7c842015-09-15 12:18:29 +00006991 case PREDEF_TYPE_AUTO_RREF_DEDUCT:
6992 T = Context.getAutoRRefDeductType();
Guy Benyei11169dd2012-12-18 14:30:41 +00006993 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00006994 case PREDEF_TYPE_ARC_UNBRIDGED_CAST:
6995 T = Context.ARCUnbridgedCastTy;
6996 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00006997 case PREDEF_TYPE_BUILTIN_FN:
6998 T = Context.BuiltinFnTy;
6999 break;
Alexey Bataev1a3320e2015-08-25 14:24:04 +00007000 case PREDEF_TYPE_OMP_ARRAY_SECTION:
7001 T = Context.OMPArraySectionTy;
7002 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00007003 }
7004
7005 assert(!T.isNull() && "Unknown predefined type");
7006 return T.withFastQualifiers(FastQuals);
7007 }
7008
7009 Index -= NUM_PREDEF_TYPE_IDS;
7010 assert(Index < TypesLoaded.size() && "Type index out-of-range");
7011 if (TypesLoaded[Index].isNull()) {
7012 TypesLoaded[Index] = readTypeRecord(Index);
7013 if (TypesLoaded[Index].isNull())
Vedant Kumar48b4f762018-04-14 01:40:48 +00007014 return QualType();
Guy Benyei11169dd2012-12-18 14:30:41 +00007015
7016 TypesLoaded[Index]->setFromAST();
7017 if (DeserializationListener)
7018 DeserializationListener->TypeRead(TypeIdx::fromTypeID(ID),
7019 TypesLoaded[Index]);
7020 }
7021
7022 return TypesLoaded[Index].withFastQualifiers(FastQuals);
7023}
7024
7025QualType ASTReader::getLocalType(ModuleFile &F, unsigned LocalID) {
7026 return GetType(getGlobalTypeID(F, LocalID));
7027}
7028
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007029serialization::TypeID
Guy Benyei11169dd2012-12-18 14:30:41 +00007030ASTReader::getGlobalTypeID(ModuleFile &F, unsigned LocalID) const {
7031 unsigned FastQuals = LocalID & Qualifiers::FastMask;
7032 unsigned LocalIndex = LocalID >> Qualifiers::FastWidth;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007033
Guy Benyei11169dd2012-12-18 14:30:41 +00007034 if (LocalIndex < NUM_PREDEF_TYPE_IDS)
7035 return LocalID;
7036
Richard Smith37a93df2017-02-18 00:32:02 +00007037 if (!F.ModuleOffsetMap.empty())
7038 ReadModuleOffsetMap(F);
7039
Guy Benyei11169dd2012-12-18 14:30:41 +00007040 ContinuousRangeMap<uint32_t, int, 2>::iterator I
7041 = F.TypeRemap.find(LocalIndex - NUM_PREDEF_TYPE_IDS);
7042 assert(I != F.TypeRemap.end() && "Invalid index into type index remap");
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007043
Guy Benyei11169dd2012-12-18 14:30:41 +00007044 unsigned GlobalIndex = LocalIndex + I->second;
7045 return (GlobalIndex << Qualifiers::FastWidth) | FastQuals;
7046}
7047
7048TemplateArgumentLocInfo
7049ASTReader::GetTemplateArgumentLocInfo(ModuleFile &F,
7050 TemplateArgument::ArgKind Kind,
7051 const RecordData &Record,
7052 unsigned &Index) {
7053 switch (Kind) {
7054 case TemplateArgument::Expression:
7055 return ReadExpr(F);
7056 case TemplateArgument::Type:
7057 return GetTypeSourceInfo(F, Record, Index);
7058 case TemplateArgument::Template: {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007059 NestedNameSpecifierLoc QualifierLoc = ReadNestedNameSpecifierLoc(F, Record,
Guy Benyei11169dd2012-12-18 14:30:41 +00007060 Index);
7061 SourceLocation TemplateNameLoc = ReadSourceLocation(F, Record, Index);
7062 return TemplateArgumentLocInfo(QualifierLoc, TemplateNameLoc,
7063 SourceLocation());
7064 }
7065 case TemplateArgument::TemplateExpansion: {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007066 NestedNameSpecifierLoc QualifierLoc = ReadNestedNameSpecifierLoc(F, Record,
Guy Benyei11169dd2012-12-18 14:30:41 +00007067 Index);
7068 SourceLocation TemplateNameLoc = ReadSourceLocation(F, Record, Index);
7069 SourceLocation EllipsisLoc = ReadSourceLocation(F, Record, Index);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007070 return TemplateArgumentLocInfo(QualifierLoc, TemplateNameLoc,
Guy Benyei11169dd2012-12-18 14:30:41 +00007071 EllipsisLoc);
7072 }
7073 case TemplateArgument::Null:
7074 case TemplateArgument::Integral:
7075 case TemplateArgument::Declaration:
7076 case TemplateArgument::NullPtr:
7077 case TemplateArgument::Pack:
7078 // FIXME: Is this right?
Vedant Kumar48b4f762018-04-14 01:40:48 +00007079 return TemplateArgumentLocInfo();
Guy Benyei11169dd2012-12-18 14:30:41 +00007080 }
7081 llvm_unreachable("unexpected template argument loc");
7082}
7083
7084TemplateArgumentLoc
7085ASTReader::ReadTemplateArgumentLoc(ModuleFile &F,
7086 const RecordData &Record, unsigned &Index) {
7087 TemplateArgument Arg = ReadTemplateArgument(F, Record, Index);
7088
7089 if (Arg.getKind() == TemplateArgument::Expression) {
7090 if (Record[Index++]) // bool InfoHasSameExpr.
7091 return TemplateArgumentLoc(Arg, TemplateArgumentLocInfo(Arg.getAsExpr()));
7092 }
7093 return TemplateArgumentLoc(Arg, GetTemplateArgumentLocInfo(F, Arg.getKind(),
7094 Record, Index));
7095}
7096
Enea Zaffanella6dbe1872013-08-10 07:24:53 +00007097const ASTTemplateArgumentListInfo*
7098ASTReader::ReadASTTemplateArgumentListInfo(ModuleFile &F,
7099 const RecordData &Record,
7100 unsigned &Index) {
7101 SourceLocation LAngleLoc = ReadSourceLocation(F, Record, Index);
7102 SourceLocation RAngleLoc = ReadSourceLocation(F, Record, Index);
7103 unsigned NumArgsAsWritten = Record[Index++];
7104 TemplateArgumentListInfo TemplArgsInfo(LAngleLoc, RAngleLoc);
7105 for (unsigned i = 0; i != NumArgsAsWritten; ++i)
7106 TemplArgsInfo.addArgument(ReadTemplateArgumentLoc(F, Record, Index));
7107 return ASTTemplateArgumentListInfo::Create(getContext(), TemplArgsInfo);
7108}
7109
Guy Benyei11169dd2012-12-18 14:30:41 +00007110Decl *ASTReader::GetExternalDecl(uint32_t ID) {
7111 return GetDecl(ID);
7112}
7113
Richard Smith053f6c62014-05-16 23:01:30 +00007114void ASTReader::CompleteRedeclChain(const Decl *D) {
Richard Smith851072e2014-05-19 20:59:20 +00007115 if (NumCurrentElementsDeserializing) {
7116 // We arrange to not care about the complete redeclaration chain while we're
7117 // deserializing. Just remember that the AST has marked this one as complete
7118 // but that it's not actually complete yet, so we know we still need to
7119 // complete it later.
7120 PendingIncompleteDeclChains.push_back(const_cast<Decl*>(D));
7121 return;
7122 }
7123
Richard Smith053f6c62014-05-16 23:01:30 +00007124 const DeclContext *DC = D->getDeclContext()->getRedeclContext();
7125
Richard Smith053f6c62014-05-16 23:01:30 +00007126 // If this is a named declaration, complete it by looking it up
7127 // within its context.
7128 //
Richard Smith01bdb7a2014-08-28 05:44:07 +00007129 // FIXME: Merging a function definition should merge
Richard Smith053f6c62014-05-16 23:01:30 +00007130 // all mergeable entities within it.
7131 if (isa<TranslationUnitDecl>(DC) || isa<NamespaceDecl>(DC) ||
7132 isa<CXXRecordDecl>(DC) || isa<EnumDecl>(DC)) {
7133 if (DeclarationName Name = cast<NamedDecl>(D)->getDeclName()) {
Richard Smitha534a312015-07-21 23:54:07 +00007134 if (!getContext().getLangOpts().CPlusPlus &&
7135 isa<TranslationUnitDecl>(DC)) {
Richard Smith053f6c62014-05-16 23:01:30 +00007136 // Outside of C++, we don't have a lookup table for the TU, so update
Richard Smitha534a312015-07-21 23:54:07 +00007137 // the identifier instead. (For C++ modules, we don't store decls
7138 // in the serialized identifier table, so we do the lookup in the TU.)
7139 auto *II = Name.getAsIdentifierInfo();
7140 assert(II && "non-identifier name in C?");
Richard Smith053f6c62014-05-16 23:01:30 +00007141 if (II->isOutOfDate())
7142 updateOutOfDateIdentifier(*II);
7143 } else
7144 DC->lookup(Name);
Richard Smith01bdb7a2014-08-28 05:44:07 +00007145 } else if (needsAnonymousDeclarationNumber(cast<NamedDecl>(D))) {
Richard Smith3cb15722015-08-05 22:41:45 +00007146 // Find all declarations of this kind from the relevant context.
7147 for (auto *DCDecl : cast<Decl>(D->getLexicalDeclContext())->redecls()) {
7148 auto *DC = cast<DeclContext>(DCDecl);
7149 SmallVector<Decl*, 8> Decls;
7150 FindExternalLexicalDecls(
7151 DC, [&](Decl::Kind K) { return K == D->getKind(); }, Decls);
7152 }
Richard Smith053f6c62014-05-16 23:01:30 +00007153 }
7154 }
Richard Smith50895422015-01-31 03:04:55 +00007155
7156 if (auto *CTSD = dyn_cast<ClassTemplateSpecializationDecl>(D))
7157 CTSD->getSpecializedTemplate()->LoadLazySpecializations();
7158 if (auto *VTSD = dyn_cast<VarTemplateSpecializationDecl>(D))
7159 VTSD->getSpecializedTemplate()->LoadLazySpecializations();
7160 if (auto *FD = dyn_cast<FunctionDecl>(D)) {
7161 if (auto *Template = FD->getPrimaryTemplate())
7162 Template->LoadLazySpecializations();
7163 }
Richard Smith053f6c62014-05-16 23:01:30 +00007164}
7165
Richard Smithc2bb8182015-03-24 06:36:48 +00007166CXXCtorInitializer **
7167ASTReader::GetExternalCXXCtorInitializers(uint64_t Offset) {
7168 RecordLocation Loc = getLocalBitOffset(Offset);
7169 BitstreamCursor &Cursor = Loc.F->DeclsCursor;
7170 SavedStreamPosition SavedPosition(Cursor);
7171 Cursor.JumpToBit(Loc.Offset);
7172 ReadingKindTracker ReadingKind(Read_Decl, *this);
7173
7174 RecordData Record;
7175 unsigned Code = Cursor.ReadCode();
7176 unsigned RecCode = Cursor.readRecord(Code, Record);
7177 if (RecCode != DECL_CXX_CTOR_INITIALIZERS) {
7178 Error("malformed AST file: missing C++ ctor initializers");
7179 return nullptr;
7180 }
7181
7182 unsigned Idx = 0;
7183 return ReadCXXCtorInitializers(*Loc.F, Record, Idx);
7184}
7185
Guy Benyei11169dd2012-12-18 14:30:41 +00007186CXXBaseSpecifier *ASTReader::GetExternalCXXBaseSpecifiers(uint64_t Offset) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00007187 assert(ContextObj && "reading base specifiers with no AST context");
7188 ASTContext &Context = *ContextObj;
7189
Guy Benyei11169dd2012-12-18 14:30:41 +00007190 RecordLocation Loc = getLocalBitOffset(Offset);
Chris Lattner7fb3bef2013-01-20 00:56:42 +00007191 BitstreamCursor &Cursor = Loc.F->DeclsCursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00007192 SavedStreamPosition SavedPosition(Cursor);
7193 Cursor.JumpToBit(Loc.Offset);
7194 ReadingKindTracker ReadingKind(Read_Decl, *this);
7195 RecordData Record;
7196 unsigned Code = Cursor.ReadCode();
Chris Lattner0e6c9402013-01-20 02:38:54 +00007197 unsigned RecCode = Cursor.readRecord(Code, Record);
Guy Benyei11169dd2012-12-18 14:30:41 +00007198 if (RecCode != DECL_CXX_BASE_SPECIFIERS) {
Richard Smithcd45dbc2014-04-19 03:48:30 +00007199 Error("malformed AST file: missing C++ base specifiers");
Craig Toppera13603a2014-05-22 05:54:18 +00007200 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007201 }
7202
7203 unsigned Idx = 0;
7204 unsigned NumBases = Record[Idx++];
7205 void *Mem = Context.Allocate(sizeof(CXXBaseSpecifier) * NumBases);
Vedant Kumar48b4f762018-04-14 01:40:48 +00007206 CXXBaseSpecifier *Bases = new (Mem) CXXBaseSpecifier [NumBases];
Guy Benyei11169dd2012-12-18 14:30:41 +00007207 for (unsigned I = 0; I != NumBases; ++I)
7208 Bases[I] = ReadCXXBaseSpecifier(*Loc.F, Record, Idx);
7209 return Bases;
7210}
7211
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007212serialization::DeclID
Guy Benyei11169dd2012-12-18 14:30:41 +00007213ASTReader::getGlobalDeclID(ModuleFile &F, LocalDeclID LocalID) const {
7214 if (LocalID < NUM_PREDEF_DECL_IDS)
7215 return LocalID;
7216
Richard Smith37a93df2017-02-18 00:32:02 +00007217 if (!F.ModuleOffsetMap.empty())
7218 ReadModuleOffsetMap(F);
7219
Guy Benyei11169dd2012-12-18 14:30:41 +00007220 ContinuousRangeMap<uint32_t, int, 2>::iterator I
7221 = F.DeclRemap.find(LocalID - NUM_PREDEF_DECL_IDS);
7222 assert(I != F.DeclRemap.end() && "Invalid index into decl index remap");
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007223
Guy Benyei11169dd2012-12-18 14:30:41 +00007224 return LocalID + I->second;
7225}
7226
7227bool ASTReader::isDeclIDFromModule(serialization::GlobalDeclID ID,
7228 ModuleFile &M) const {
Richard Smithfe620d22015-03-05 23:24:12 +00007229 // Predefined decls aren't from any module.
7230 if (ID < NUM_PREDEF_DECL_IDS)
7231 return false;
7232
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007233 return ID - NUM_PREDEF_DECL_IDS >= M.BaseDeclID &&
Richard Smithbcda1a92015-07-12 23:51:20 +00007234 ID - NUM_PREDEF_DECL_IDS < M.BaseDeclID + M.LocalNumDecls;
Guy Benyei11169dd2012-12-18 14:30:41 +00007235}
7236
Douglas Gregor9f782892013-01-21 15:25:38 +00007237ModuleFile *ASTReader::getOwningModuleFile(const Decl *D) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007238 if (!D->isFromASTFile())
Craig Toppera13603a2014-05-22 05:54:18 +00007239 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007240 GlobalDeclMapType::const_iterator I = GlobalDeclMap.find(D->getGlobalID());
7241 assert(I != GlobalDeclMap.end() && "Corrupted global declaration map");
7242 return I->second;
7243}
7244
7245SourceLocation ASTReader::getSourceLocationForDeclID(GlobalDeclID ID) {
7246 if (ID < NUM_PREDEF_DECL_IDS)
Vedant Kumar48b4f762018-04-14 01:40:48 +00007247 return SourceLocation();
Richard Smithcd45dbc2014-04-19 03:48:30 +00007248
Guy Benyei11169dd2012-12-18 14:30:41 +00007249 unsigned Index = ID - NUM_PREDEF_DECL_IDS;
7250
7251 if (Index > DeclsLoaded.size()) {
7252 Error("declaration ID out-of-range for AST file");
Vedant Kumar48b4f762018-04-14 01:40:48 +00007253 return SourceLocation();
Guy Benyei11169dd2012-12-18 14:30:41 +00007254 }
Richard Smithcd45dbc2014-04-19 03:48:30 +00007255
Guy Benyei11169dd2012-12-18 14:30:41 +00007256 if (Decl *D = DeclsLoaded[Index])
7257 return D->getLocation();
7258
Richard Smithcb34bd32016-03-27 07:28:06 +00007259 SourceLocation Loc;
7260 DeclCursorForID(ID, Loc);
7261 return Loc;
Guy Benyei11169dd2012-12-18 14:30:41 +00007262}
7263
Richard Smithfe620d22015-03-05 23:24:12 +00007264static Decl *getPredefinedDecl(ASTContext &Context, PredefinedDeclIDs ID) {
7265 switch (ID) {
7266 case PREDEF_DECL_NULL_ID:
7267 return nullptr;
7268
7269 case PREDEF_DECL_TRANSLATION_UNIT_ID:
7270 return Context.getTranslationUnitDecl();
7271
7272 case PREDEF_DECL_OBJC_ID_ID:
7273 return Context.getObjCIdDecl();
7274
7275 case PREDEF_DECL_OBJC_SEL_ID:
7276 return Context.getObjCSelDecl();
7277
7278 case PREDEF_DECL_OBJC_CLASS_ID:
7279 return Context.getObjCClassDecl();
7280
7281 case PREDEF_DECL_OBJC_PROTOCOL_ID:
7282 return Context.getObjCProtocolDecl();
7283
7284 case PREDEF_DECL_INT_128_ID:
7285 return Context.getInt128Decl();
7286
7287 case PREDEF_DECL_UNSIGNED_INT_128_ID:
7288 return Context.getUInt128Decl();
7289
7290 case PREDEF_DECL_OBJC_INSTANCETYPE_ID:
7291 return Context.getObjCInstanceTypeDecl();
7292
7293 case PREDEF_DECL_BUILTIN_VA_LIST_ID:
7294 return Context.getBuiltinVaListDecl();
Richard Smithf19e1272015-03-07 00:04:49 +00007295
Richard Smith9b88a4c2015-07-27 05:40:23 +00007296 case PREDEF_DECL_VA_LIST_TAG:
7297 return Context.getVaListTagDecl();
7298
Charles Davisc7d5c942015-09-17 20:55:33 +00007299 case PREDEF_DECL_BUILTIN_MS_VA_LIST_ID:
7300 return Context.getBuiltinMSVaListDecl();
7301
Richard Smithf19e1272015-03-07 00:04:49 +00007302 case PREDEF_DECL_EXTERN_C_CONTEXT_ID:
7303 return Context.getExternCContextDecl();
David Majnemerd9b1a4f2015-11-04 03:40:30 +00007304
7305 case PREDEF_DECL_MAKE_INTEGER_SEQ_ID:
7306 return Context.getMakeIntegerSeqDecl();
Quentin Colombet043406b2016-02-03 22:41:00 +00007307
7308 case PREDEF_DECL_CF_CONSTANT_STRING_ID:
7309 return Context.getCFConstantStringDecl();
Ben Langmuirf5416742016-02-04 00:55:24 +00007310
7311 case PREDEF_DECL_CF_CONSTANT_STRING_TAG_ID:
7312 return Context.getCFConstantStringTagDecl();
Eric Fiselier6ad68552016-07-01 01:24:09 +00007313
7314 case PREDEF_DECL_TYPE_PACK_ELEMENT_ID:
7315 return Context.getTypePackElementDecl();
Richard Smithfe620d22015-03-05 23:24:12 +00007316 }
Yaron Keren322bdad2015-03-06 07:49:14 +00007317 llvm_unreachable("PredefinedDeclIDs unknown enum value");
Richard Smithfe620d22015-03-05 23:24:12 +00007318}
7319
Richard Smithcd45dbc2014-04-19 03:48:30 +00007320Decl *ASTReader::GetExistingDecl(DeclID ID) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00007321 assert(ContextObj && "reading decl with no AST context");
Richard Smithcd45dbc2014-04-19 03:48:30 +00007322 if (ID < NUM_PREDEF_DECL_IDS) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00007323 Decl *D = getPredefinedDecl(*ContextObj, (PredefinedDeclIDs)ID);
Richard Smithfe620d22015-03-05 23:24:12 +00007324 if (D) {
7325 // Track that we have merged the declaration with ID \p ID into the
7326 // pre-existing predefined declaration \p D.
Richard Smith5fc18a92015-07-12 23:43:21 +00007327 auto &Merged = KeyDecls[D->getCanonicalDecl()];
Richard Smithfe620d22015-03-05 23:24:12 +00007328 if (Merged.empty())
7329 Merged.push_back(ID);
Guy Benyei11169dd2012-12-18 14:30:41 +00007330 }
Richard Smithfe620d22015-03-05 23:24:12 +00007331 return D;
Guy Benyei11169dd2012-12-18 14:30:41 +00007332 }
Richard Smithcd45dbc2014-04-19 03:48:30 +00007333
Guy Benyei11169dd2012-12-18 14:30:41 +00007334 unsigned Index = ID - NUM_PREDEF_DECL_IDS;
7335
7336 if (Index >= DeclsLoaded.size()) {
7337 assert(0 && "declaration ID out-of-range for AST file");
7338 Error("declaration ID out-of-range for AST file");
Craig Toppera13603a2014-05-22 05:54:18 +00007339 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007340 }
Richard Smithcd45dbc2014-04-19 03:48:30 +00007341
7342 return DeclsLoaded[Index];
7343}
7344
7345Decl *ASTReader::GetDecl(DeclID ID) {
7346 if (ID < NUM_PREDEF_DECL_IDS)
7347 return GetExistingDecl(ID);
7348
7349 unsigned Index = ID - NUM_PREDEF_DECL_IDS;
7350
7351 if (Index >= DeclsLoaded.size()) {
7352 assert(0 && "declaration ID out-of-range for AST file");
7353 Error("declaration ID out-of-range for AST file");
Craig Toppera13603a2014-05-22 05:54:18 +00007354 return nullptr;
Richard Smithcd45dbc2014-04-19 03:48:30 +00007355 }
7356
Guy Benyei11169dd2012-12-18 14:30:41 +00007357 if (!DeclsLoaded[Index]) {
7358 ReadDeclRecord(ID);
7359 if (DeserializationListener)
7360 DeserializationListener->DeclRead(ID, DeclsLoaded[Index]);
7361 }
7362
7363 return DeclsLoaded[Index];
7364}
7365
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007366DeclID ASTReader::mapGlobalIDToModuleFileGlobalID(ModuleFile &M,
Guy Benyei11169dd2012-12-18 14:30:41 +00007367 DeclID GlobalID) {
7368 if (GlobalID < NUM_PREDEF_DECL_IDS)
7369 return GlobalID;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007370
Guy Benyei11169dd2012-12-18 14:30:41 +00007371 GlobalDeclMapType::const_iterator I = GlobalDeclMap.find(GlobalID);
7372 assert(I != GlobalDeclMap.end() && "Corrupted global declaration map");
7373 ModuleFile *Owner = I->second;
7374
7375 llvm::DenseMap<ModuleFile *, serialization::DeclID>::iterator Pos
7376 = M.GlobalToLocalDeclIDs.find(Owner);
7377 if (Pos == M.GlobalToLocalDeclIDs.end())
7378 return 0;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007379
Guy Benyei11169dd2012-12-18 14:30:41 +00007380 return GlobalID - Owner->BaseDeclID + Pos->second;
7381}
7382
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007383serialization::DeclID ASTReader::ReadDeclID(ModuleFile &F,
Guy Benyei11169dd2012-12-18 14:30:41 +00007384 const RecordData &Record,
7385 unsigned &Idx) {
7386 if (Idx >= Record.size()) {
7387 Error("Corrupted AST file");
7388 return 0;
7389 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007390
Guy Benyei11169dd2012-12-18 14:30:41 +00007391 return getGlobalDeclID(F, Record[Idx++]);
7392}
7393
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00007394/// Resolve the offset of a statement into a statement.
Guy Benyei11169dd2012-12-18 14:30:41 +00007395///
7396/// This operation will read a new statement from the external
7397/// source each time it is called, and is meant to be used via a
7398/// LazyOffsetPtr (which is used by Decls for the body of functions, etc).
7399Stmt *ASTReader::GetExternalDeclStmt(uint64_t Offset) {
7400 // Switch case IDs are per Decl.
7401 ClearSwitchCaseIDs();
7402
7403 // Offset here is a global offset across the entire chain.
7404 RecordLocation Loc = getLocalBitOffset(Offset);
7405 Loc.F->DeclsCursor.JumpToBit(Loc.Offset);
David Blaikie9fd16f82017-03-08 23:57:08 +00007406 assert(NumCurrentElementsDeserializing == 0 &&
7407 "should not be called while already deserializing");
7408 Deserializing D(this);
Guy Benyei11169dd2012-12-18 14:30:41 +00007409 return ReadStmtFromStream(*Loc.F);
7410}
7411
Richard Smith3cb15722015-08-05 22:41:45 +00007412void ASTReader::FindExternalLexicalDecls(
7413 const DeclContext *DC, llvm::function_ref<bool(Decl::Kind)> IsKindWeWant,
7414 SmallVectorImpl<Decl *> &Decls) {
Richard Smith82f8fcd2015-08-06 22:07:25 +00007415 bool PredefsVisited[NUM_PREDEF_DECL_IDS] = {};
7416
Richard Smith9ccdd932015-08-06 22:14:12 +00007417 auto Visit = [&] (ModuleFile *M, LexicalContents LexicalDecls) {
Richard Smith82f8fcd2015-08-06 22:07:25 +00007418 assert(LexicalDecls.size() % 2 == 0 && "expected an even number of entries");
7419 for (int I = 0, N = LexicalDecls.size(); I != N; I += 2) {
7420 auto K = (Decl::Kind)+LexicalDecls[I];
7421 if (!IsKindWeWant(K))
7422 continue;
7423
7424 auto ID = (serialization::DeclID)+LexicalDecls[I + 1];
7425
7426 // Don't add predefined declarations to the lexical context more
7427 // than once.
7428 if (ID < NUM_PREDEF_DECL_IDS) {
7429 if (PredefsVisited[ID])
7430 continue;
7431
7432 PredefsVisited[ID] = true;
7433 }
7434
7435 if (Decl *D = GetLocalDecl(*M, ID)) {
Richard Smith2317a3e2015-08-11 21:21:20 +00007436 assert(D->getKind() == K && "wrong kind for lexical decl");
Richard Smith82f8fcd2015-08-06 22:07:25 +00007437 if (!DC->isDeclInLexicalTraversal(D))
7438 Decls.push_back(D);
7439 }
7440 }
7441 };
7442
7443 if (isa<TranslationUnitDecl>(DC)) {
7444 for (auto Lexical : TULexicalDecls)
7445 Visit(Lexical.first, Lexical.second);
7446 } else {
7447 auto I = LexicalDecls.find(DC);
7448 if (I != LexicalDecls.end())
Richard Smith9c9173d2015-08-11 22:00:24 +00007449 Visit(I->second.first, I->second.second);
Richard Smith82f8fcd2015-08-06 22:07:25 +00007450 }
7451
Guy Benyei11169dd2012-12-18 14:30:41 +00007452 ++NumLexicalDeclContextsRead;
Guy Benyei11169dd2012-12-18 14:30:41 +00007453}
7454
7455namespace {
7456
7457class DeclIDComp {
7458 ASTReader &Reader;
7459 ModuleFile &Mod;
7460
7461public:
7462 DeclIDComp(ASTReader &Reader, ModuleFile &M) : Reader(Reader), Mod(M) {}
7463
7464 bool operator()(LocalDeclID L, LocalDeclID R) const {
7465 SourceLocation LHS = getLocation(L);
7466 SourceLocation RHS = getLocation(R);
7467 return Reader.getSourceManager().isBeforeInTranslationUnit(LHS, RHS);
7468 }
7469
7470 bool operator()(SourceLocation LHS, LocalDeclID R) const {
7471 SourceLocation RHS = getLocation(R);
7472 return Reader.getSourceManager().isBeforeInTranslationUnit(LHS, RHS);
7473 }
7474
7475 bool operator()(LocalDeclID L, SourceLocation RHS) const {
7476 SourceLocation LHS = getLocation(L);
7477 return Reader.getSourceManager().isBeforeInTranslationUnit(LHS, RHS);
7478 }
7479
7480 SourceLocation getLocation(LocalDeclID ID) const {
7481 return Reader.getSourceManager().getFileLoc(
7482 Reader.getSourceLocationForDeclID(Reader.getGlobalDeclID(Mod, ID)));
7483 }
7484};
7485
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00007486} // namespace
Guy Benyei11169dd2012-12-18 14:30:41 +00007487
7488void ASTReader::FindFileRegionDecls(FileID File,
7489 unsigned Offset, unsigned Length,
7490 SmallVectorImpl<Decl *> &Decls) {
7491 SourceManager &SM = getSourceManager();
7492
7493 llvm::DenseMap<FileID, FileDeclsInfo>::iterator I = FileDeclIDs.find(File);
7494 if (I == FileDeclIDs.end())
7495 return;
7496
7497 FileDeclsInfo &DInfo = I->second;
7498 if (DInfo.Decls.empty())
7499 return;
7500
7501 SourceLocation
7502 BeginLoc = SM.getLocForStartOfFile(File).getLocWithOffset(Offset);
7503 SourceLocation EndLoc = BeginLoc.getLocWithOffset(Length);
7504
7505 DeclIDComp DIDComp(*this, *DInfo.Mod);
7506 ArrayRef<serialization::LocalDeclID>::iterator
7507 BeginIt = std::lower_bound(DInfo.Decls.begin(), DInfo.Decls.end(),
7508 BeginLoc, DIDComp);
7509 if (BeginIt != DInfo.Decls.begin())
7510 --BeginIt;
7511
7512 // If we are pointing at a top-level decl inside an objc container, we need
7513 // to backtrack until we find it otherwise we will fail to report that the
7514 // region overlaps with an objc container.
7515 while (BeginIt != DInfo.Decls.begin() &&
7516 GetDecl(getGlobalDeclID(*DInfo.Mod, *BeginIt))
7517 ->isTopLevelDeclInObjCContainer())
7518 --BeginIt;
7519
7520 ArrayRef<serialization::LocalDeclID>::iterator
7521 EndIt = std::upper_bound(DInfo.Decls.begin(), DInfo.Decls.end(),
7522 EndLoc, DIDComp);
7523 if (EndIt != DInfo.Decls.end())
7524 ++EndIt;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007525
Guy Benyei11169dd2012-12-18 14:30:41 +00007526 for (ArrayRef<serialization::LocalDeclID>::iterator
7527 DIt = BeginIt; DIt != EndIt; ++DIt)
7528 Decls.push_back(GetDecl(getGlobalDeclID(*DInfo.Mod, *DIt)));
7529}
7530
Richard Smith9ce12e32013-02-07 03:30:24 +00007531bool
Guy Benyei11169dd2012-12-18 14:30:41 +00007532ASTReader::FindExternalVisibleDeclsByName(const DeclContext *DC,
7533 DeclarationName Name) {
Richard Smithd88a7f12015-09-01 20:35:42 +00007534 assert(DC->hasExternalVisibleStorage() && DC == DC->getPrimaryContext() &&
Guy Benyei11169dd2012-12-18 14:30:41 +00007535 "DeclContext has no visible decls in storage");
7536 if (!Name)
Richard Smith9ce12e32013-02-07 03:30:24 +00007537 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +00007538
Richard Smithd88a7f12015-09-01 20:35:42 +00007539 auto It = Lookups.find(DC);
7540 if (It == Lookups.end())
7541 return false;
7542
Richard Smith8c913ec2014-08-14 02:21:01 +00007543 Deserializing LookupResults(this);
7544
Richard Smithd88a7f12015-09-01 20:35:42 +00007545 // Load the list of declarations.
Guy Benyei11169dd2012-12-18 14:30:41 +00007546 SmallVector<NamedDecl *, 64> Decls;
Vedant Kumar48b4f762018-04-14 01:40:48 +00007547 for (DeclID ID : It->second.Table.find(Name)) {
7548 NamedDecl *ND = cast<NamedDecl>(GetDecl(ID));
Richard Smithd88a7f12015-09-01 20:35:42 +00007549 if (ND->getDeclName() == Name)
7550 Decls.push_back(ND);
7551 }
Richard Smithcd45dbc2014-04-19 03:48:30 +00007552
Guy Benyei11169dd2012-12-18 14:30:41 +00007553 ++NumVisibleDeclContextsRead;
7554 SetExternalVisibleDeclsForName(DC, Name, Decls);
Richard Smith9ce12e32013-02-07 03:30:24 +00007555 return !Decls.empty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007556}
7557
Guy Benyei11169dd2012-12-18 14:30:41 +00007558void ASTReader::completeVisibleDeclsMap(const DeclContext *DC) {
7559 if (!DC->hasExternalVisibleStorage())
7560 return;
Richard Smithd88a7f12015-09-01 20:35:42 +00007561
7562 auto It = Lookups.find(DC);
7563 assert(It != Lookups.end() &&
7564 "have external visible storage but no lookup tables");
7565
Craig Topper79be4cd2013-07-05 04:33:53 +00007566 DeclsMap Decls;
Guy Benyei11169dd2012-12-18 14:30:41 +00007567
Vedant Kumar48b4f762018-04-14 01:40:48 +00007568 for (DeclID ID : It->second.Table.findAll()) {
7569 NamedDecl *ND = cast<NamedDecl>(GetDecl(ID));
Richard Smithd88a7f12015-09-01 20:35:42 +00007570 Decls[ND->getDeclName()].push_back(ND);
Guy Benyei11169dd2012-12-18 14:30:41 +00007571 }
7572
Guy Benyei11169dd2012-12-18 14:30:41 +00007573 ++NumVisibleDeclContextsRead;
7574
Vedant Kumar48b4f762018-04-14 01:40:48 +00007575 for (DeclsMap::iterator I = Decls.begin(), E = Decls.end(); I != E; ++I) {
7576 SetExternalVisibleDeclsForName(DC, I->first, I->second);
7577 }
Guy Benyei11169dd2012-12-18 14:30:41 +00007578 const_cast<DeclContext *>(DC)->setHasExternalVisibleStorage(false);
7579}
7580
Richard Smithd88a7f12015-09-01 20:35:42 +00007581const serialization::reader::DeclContextLookupTable *
7582ASTReader::getLoadedLookupTables(DeclContext *Primary) const {
7583 auto I = Lookups.find(Primary);
7584 return I == Lookups.end() ? nullptr : &I->second;
7585}
7586
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00007587/// Under non-PCH compilation the consumer receives the objc methods
Guy Benyei11169dd2012-12-18 14:30:41 +00007588/// before receiving the implementation, and codegen depends on this.
7589/// We simulate this by deserializing and passing to consumer the methods of the
7590/// implementation before passing the deserialized implementation decl.
7591static void PassObjCImplDeclToConsumer(ObjCImplDecl *ImplD,
7592 ASTConsumer *Consumer) {
7593 assert(ImplD && Consumer);
7594
Aaron Ballmanaff18c02014-03-13 19:03:34 +00007595 for (auto *I : ImplD->methods())
7596 Consumer->HandleInterestingDecl(DeclGroupRef(I));
Guy Benyei11169dd2012-12-18 14:30:41 +00007597
7598 Consumer->HandleInterestingDecl(DeclGroupRef(ImplD));
7599}
7600
Guy Benyei11169dd2012-12-18 14:30:41 +00007601void ASTReader::PassInterestingDeclToConsumer(Decl *D) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00007602 if (ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00007603 PassObjCImplDeclToConsumer(ImplD, Consumer);
7604 else
7605 Consumer->HandleInterestingDecl(DeclGroupRef(D));
7606}
7607
7608void ASTReader::StartTranslationUnit(ASTConsumer *Consumer) {
7609 this->Consumer = Consumer;
7610
Richard Smith9e2341d2015-03-23 03:25:59 +00007611 if (Consumer)
7612 PassInterestingDeclsToConsumer();
Richard Smith7f330cd2015-03-18 01:42:29 +00007613
7614 if (DeserializationListener)
7615 DeserializationListener->ReaderInitialized(this);
Guy Benyei11169dd2012-12-18 14:30:41 +00007616}
7617
7618void ASTReader::PrintStats() {
7619 std::fprintf(stderr, "*** AST File Statistics:\n");
7620
7621 unsigned NumTypesLoaded
7622 = TypesLoaded.size() - std::count(TypesLoaded.begin(), TypesLoaded.end(),
7623 QualType());
7624 unsigned NumDeclsLoaded
7625 = DeclsLoaded.size() - std::count(DeclsLoaded.begin(), DeclsLoaded.end(),
Craig Toppera13603a2014-05-22 05:54:18 +00007626 (Decl *)nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00007627 unsigned NumIdentifiersLoaded
7628 = IdentifiersLoaded.size() - std::count(IdentifiersLoaded.begin(),
7629 IdentifiersLoaded.end(),
Craig Toppera13603a2014-05-22 05:54:18 +00007630 (IdentifierInfo *)nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00007631 unsigned NumMacrosLoaded
7632 = MacrosLoaded.size() - std::count(MacrosLoaded.begin(),
7633 MacrosLoaded.end(),
Craig Toppera13603a2014-05-22 05:54:18 +00007634 (MacroInfo *)nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00007635 unsigned NumSelectorsLoaded
7636 = SelectorsLoaded.size() - std::count(SelectorsLoaded.begin(),
7637 SelectorsLoaded.end(),
7638 Selector());
7639
7640 if (unsigned TotalNumSLocEntries = getTotalNumSLocs())
7641 std::fprintf(stderr, " %u/%u source location entries read (%f%%)\n",
7642 NumSLocEntriesRead, TotalNumSLocEntries,
7643 ((float)NumSLocEntriesRead/TotalNumSLocEntries * 100));
7644 if (!TypesLoaded.empty())
7645 std::fprintf(stderr, " %u/%u types read (%f%%)\n",
7646 NumTypesLoaded, (unsigned)TypesLoaded.size(),
7647 ((float)NumTypesLoaded/TypesLoaded.size() * 100));
7648 if (!DeclsLoaded.empty())
7649 std::fprintf(stderr, " %u/%u declarations read (%f%%)\n",
7650 NumDeclsLoaded, (unsigned)DeclsLoaded.size(),
7651 ((float)NumDeclsLoaded/DeclsLoaded.size() * 100));
7652 if (!IdentifiersLoaded.empty())
7653 std::fprintf(stderr, " %u/%u identifiers read (%f%%)\n",
7654 NumIdentifiersLoaded, (unsigned)IdentifiersLoaded.size(),
7655 ((float)NumIdentifiersLoaded/IdentifiersLoaded.size() * 100));
7656 if (!MacrosLoaded.empty())
7657 std::fprintf(stderr, " %u/%u macros read (%f%%)\n",
7658 NumMacrosLoaded, (unsigned)MacrosLoaded.size(),
7659 ((float)NumMacrosLoaded/MacrosLoaded.size() * 100));
7660 if (!SelectorsLoaded.empty())
7661 std::fprintf(stderr, " %u/%u selectors read (%f%%)\n",
7662 NumSelectorsLoaded, (unsigned)SelectorsLoaded.size(),
7663 ((float)NumSelectorsLoaded/SelectorsLoaded.size() * 100));
7664 if (TotalNumStatements)
7665 std::fprintf(stderr, " %u/%u statements read (%f%%)\n",
7666 NumStatementsRead, TotalNumStatements,
7667 ((float)NumStatementsRead/TotalNumStatements * 100));
7668 if (TotalNumMacros)
7669 std::fprintf(stderr, " %u/%u macros read (%f%%)\n",
7670 NumMacrosRead, TotalNumMacros,
7671 ((float)NumMacrosRead/TotalNumMacros * 100));
7672 if (TotalLexicalDeclContexts)
7673 std::fprintf(stderr, " %u/%u lexical declcontexts read (%f%%)\n",
7674 NumLexicalDeclContextsRead, TotalLexicalDeclContexts,
7675 ((float)NumLexicalDeclContextsRead/TotalLexicalDeclContexts
7676 * 100));
7677 if (TotalVisibleDeclContexts)
7678 std::fprintf(stderr, " %u/%u visible declcontexts read (%f%%)\n",
7679 NumVisibleDeclContextsRead, TotalVisibleDeclContexts,
7680 ((float)NumVisibleDeclContextsRead/TotalVisibleDeclContexts
7681 * 100));
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00007682 if (TotalNumMethodPoolEntries)
Guy Benyei11169dd2012-12-18 14:30:41 +00007683 std::fprintf(stderr, " %u/%u method pool entries read (%f%%)\n",
7684 NumMethodPoolEntriesRead, TotalNumMethodPoolEntries,
7685 ((float)NumMethodPoolEntriesRead/TotalNumMethodPoolEntries
7686 * 100));
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00007687 if (NumMethodPoolLookups)
Douglas Gregorad2f7a52013-01-28 17:54:36 +00007688 std::fprintf(stderr, " %u/%u method pool lookups succeeded (%f%%)\n",
7689 NumMethodPoolHits, NumMethodPoolLookups,
7690 ((float)NumMethodPoolHits/NumMethodPoolLookups * 100.0));
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00007691 if (NumMethodPoolTableLookups)
Douglas Gregorad2f7a52013-01-28 17:54:36 +00007692 std::fprintf(stderr, " %u/%u method pool table lookups succeeded (%f%%)\n",
7693 NumMethodPoolTableHits, NumMethodPoolTableLookups,
7694 ((float)NumMethodPoolTableHits/NumMethodPoolTableLookups
7695 * 100.0));
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00007696 if (NumIdentifierLookupHits)
Douglas Gregor00a50f72013-01-25 00:38:33 +00007697 std::fprintf(stderr,
7698 " %u / %u identifier table lookups succeeded (%f%%)\n",
7699 NumIdentifierLookupHits, NumIdentifierLookups,
7700 (double)NumIdentifierLookupHits*100.0/NumIdentifierLookups);
Douglas Gregor00a50f72013-01-25 00:38:33 +00007701
Douglas Gregore060e572013-01-25 01:03:03 +00007702 if (GlobalIndex) {
7703 std::fprintf(stderr, "\n");
7704 GlobalIndex->printStats();
7705 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007706
Guy Benyei11169dd2012-12-18 14:30:41 +00007707 std::fprintf(stderr, "\n");
7708 dump();
7709 std::fprintf(stderr, "\n");
7710}
7711
7712template<typename Key, typename ModuleFile, unsigned InitialCapacity>
Vassil Vassilevb2710682017-03-02 18:13:19 +00007713LLVM_DUMP_METHOD static void
Guy Benyei11169dd2012-12-18 14:30:41 +00007714dumpModuleIDMap(StringRef Name,
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007715 const ContinuousRangeMap<Key, ModuleFile *,
Guy Benyei11169dd2012-12-18 14:30:41 +00007716 InitialCapacity> &Map) {
7717 if (Map.begin() == Map.end())
7718 return;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007719
Vedant Kumar48b4f762018-04-14 01:40:48 +00007720 using MapType = ContinuousRangeMap<Key, ModuleFile *, InitialCapacity>;
7721
Guy Benyei11169dd2012-12-18 14:30:41 +00007722 llvm::errs() << Name << ":\n";
Vedant Kumar48b4f762018-04-14 01:40:48 +00007723 for (typename MapType::const_iterator I = Map.begin(), IEnd = Map.end();
7724 I != IEnd; ++I) {
7725 llvm::errs() << " " << I->first << " -> " << I->second->FileName
7726 << "\n";
7727 }
Guy Benyei11169dd2012-12-18 14:30:41 +00007728}
7729
Yaron Kerencdae9412016-01-29 19:38:18 +00007730LLVM_DUMP_METHOD void ASTReader::dump() {
Guy Benyei11169dd2012-12-18 14:30:41 +00007731 llvm::errs() << "*** PCH/ModuleFile Remappings:\n";
7732 dumpModuleIDMap("Global bit offset map", GlobalBitOffsetsMap);
7733 dumpModuleIDMap("Global source location entry map", GlobalSLocEntryMap);
7734 dumpModuleIDMap("Global type map", GlobalTypeMap);
7735 dumpModuleIDMap("Global declaration map", GlobalDeclMap);
7736 dumpModuleIDMap("Global identifier map", GlobalIdentifierMap);
7737 dumpModuleIDMap("Global macro map", GlobalMacroMap);
7738 dumpModuleIDMap("Global submodule map", GlobalSubmoduleMap);
7739 dumpModuleIDMap("Global selector map", GlobalSelectorMap);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007740 dumpModuleIDMap("Global preprocessed entity map",
Guy Benyei11169dd2012-12-18 14:30:41 +00007741 GlobalPreprocessedEntityMap);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00007742
Guy Benyei11169dd2012-12-18 14:30:41 +00007743 llvm::errs() << "\n*** PCH/Modules Loaded:";
Vedant Kumar48b4f762018-04-14 01:40:48 +00007744 for (ModuleFile &M : ModuleMgr)
Duncan P. N. Exon Smith96a06e02017-01-28 22:15:22 +00007745 M.dump();
Guy Benyei11169dd2012-12-18 14:30:41 +00007746}
7747
7748/// Return the amount of memory used by memory buffers, breaking down
7749/// by heap-backed versus mmap'ed memory.
7750void ASTReader::getMemoryBufferSizes(MemoryBufferSizes &sizes) const {
Vedant Kumar48b4f762018-04-14 01:40:48 +00007751 for (ModuleFile &I : ModuleMgr) {
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +00007752 if (llvm::MemoryBuffer *buf = I.Buffer) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007753 size_t bytes = buf->getBufferSize();
7754 switch (buf->getBufferKind()) {
7755 case llvm::MemoryBuffer::MemoryBuffer_Malloc:
7756 sizes.malloc_bytes += bytes;
7757 break;
7758 case llvm::MemoryBuffer::MemoryBuffer_MMap:
7759 sizes.mmap_bytes += bytes;
7760 break;
7761 }
7762 }
7763 }
7764}
7765
7766void ASTReader::InitializeSema(Sema &S) {
7767 SemaObj = &S;
7768 S.addExternalSource(this);
7769
7770 // Makes sure any declarations that were deserialized "too early"
7771 // still get added to the identifier's declaration chains.
Vedant Kumar48b4f762018-04-14 01:40:48 +00007772 for (uint64_t ID : PreloadedDeclIDs) {
7773 NamedDecl *D = cast<NamedDecl>(GetDecl(ID));
Ben Langmuir5418f402014-09-10 21:29:41 +00007774 pushExternalDeclIntoScope(D, D->getDeclName());
Guy Benyei11169dd2012-12-18 14:30:41 +00007775 }
Ben Langmuir5418f402014-09-10 21:29:41 +00007776 PreloadedDeclIDs.clear();
Guy Benyei11169dd2012-12-18 14:30:41 +00007777
Richard Smith3d8e97e2013-10-18 06:54:39 +00007778 // FIXME: What happens if these are changed by a module import?
Guy Benyei11169dd2012-12-18 14:30:41 +00007779 if (!FPPragmaOptions.empty()) {
7780 assert(FPPragmaOptions.size() == 1 && "Wrong number of FP_PRAGMA_OPTIONS");
Adam Nemet484aa452017-03-27 19:17:25 +00007781 SemaObj->FPFeatures = FPOptions(FPPragmaOptions[0]);
Guy Benyei11169dd2012-12-18 14:30:41 +00007782 }
7783
Yaxun Liu5b746652016-12-18 05:18:55 +00007784 SemaObj->OpenCLFeatures.copy(OpenCLExtensions);
7785 SemaObj->OpenCLTypeExtMap = OpenCLTypeExtMap;
7786 SemaObj->OpenCLDeclExtMap = OpenCLDeclExtMap;
Richard Smith3d8e97e2013-10-18 06:54:39 +00007787
7788 UpdateSema();
7789}
7790
7791void ASTReader::UpdateSema() {
7792 assert(SemaObj && "no Sema to update");
7793
7794 // Load the offsets of the declarations that Sema references.
7795 // They will be lazily deserialized when needed.
7796 if (!SemaDeclRefs.empty()) {
Richard Smith96269c52016-09-29 22:49:46 +00007797 assert(SemaDeclRefs.size() % 3 == 0);
7798 for (unsigned I = 0; I != SemaDeclRefs.size(); I += 3) {
Richard Smith3d8e97e2013-10-18 06:54:39 +00007799 if (!SemaObj->StdNamespace)
7800 SemaObj->StdNamespace = SemaDeclRefs[I];
7801 if (!SemaObj->StdBadAlloc)
7802 SemaObj->StdBadAlloc = SemaDeclRefs[I+1];
Richard Smith96269c52016-09-29 22:49:46 +00007803 if (!SemaObj->StdAlignValT)
7804 SemaObj->StdAlignValT = SemaDeclRefs[I+2];
Richard Smith3d8e97e2013-10-18 06:54:39 +00007805 }
7806 SemaDeclRefs.clear();
7807 }
Dario Domizioli13a0a382014-05-23 12:13:25 +00007808
Nico Weber779355f2016-03-02 23:22:00 +00007809 // Update the state of pragmas. Use the same API as if we had encountered the
7810 // pragma in the source.
Dario Domizioli13a0a382014-05-23 12:13:25 +00007811 if(OptimizeOffPragmaLocation.isValid())
7812 SemaObj->ActOnPragmaOptimize(/* IsOn = */ false, OptimizeOffPragmaLocation);
Nico Weber779355f2016-03-02 23:22:00 +00007813 if (PragmaMSStructState != -1)
7814 SemaObj->ActOnPragmaMSStruct((PragmaMSStructKind)PragmaMSStructState);
Nico Weber42932312016-03-03 00:17:35 +00007815 if (PointersToMembersPragmaLocation.isValid()) {
7816 SemaObj->ActOnPragmaMSPointersToMembers(
7817 (LangOptions::PragmaMSPointersToMembersKind)
7818 PragmaMSPointersToMembersState,
7819 PointersToMembersPragmaLocation);
7820 }
Justin Lebar67a78a62016-10-08 22:15:58 +00007821 SemaObj->ForceCUDAHostDeviceDepth = ForceCUDAHostDeviceDepth;
Alex Lorenz7d7e1e02017-03-31 15:36:21 +00007822
7823 if (PragmaPackCurrentValue) {
7824 // The bottom of the stack might have a default value. It must be adjusted
7825 // to the current value to ensure that the packing state is preserved after
7826 // popping entries that were included/imported from a PCH/module.
7827 bool DropFirst = false;
7828 if (!PragmaPackStack.empty() &&
7829 PragmaPackStack.front().Location.isInvalid()) {
7830 assert(PragmaPackStack.front().Value == SemaObj->PackStack.DefaultValue &&
7831 "Expected a default alignment value");
7832 SemaObj->PackStack.Stack.emplace_back(
7833 PragmaPackStack.front().SlotLabel, SemaObj->PackStack.CurrentValue,
Alex Lorenz45b40142017-07-28 14:41:21 +00007834 SemaObj->PackStack.CurrentPragmaLocation,
7835 PragmaPackStack.front().PushLocation);
Alex Lorenz7d7e1e02017-03-31 15:36:21 +00007836 DropFirst = true;
7837 }
7838 for (const auto &Entry :
7839 llvm::makeArrayRef(PragmaPackStack).drop_front(DropFirst ? 1 : 0))
7840 SemaObj->PackStack.Stack.emplace_back(Entry.SlotLabel, Entry.Value,
Alex Lorenz45b40142017-07-28 14:41:21 +00007841 Entry.Location, Entry.PushLocation);
Alex Lorenz7d7e1e02017-03-31 15:36:21 +00007842 if (PragmaPackCurrentLocation.isInvalid()) {
7843 assert(*PragmaPackCurrentValue == SemaObj->PackStack.DefaultValue &&
7844 "Expected a default alignment value");
7845 // Keep the current values.
7846 } else {
7847 SemaObj->PackStack.CurrentValue = *PragmaPackCurrentValue;
7848 SemaObj->PackStack.CurrentPragmaLocation = PragmaPackCurrentLocation;
7849 }
7850 }
Guy Benyei11169dd2012-12-18 14:30:41 +00007851}
7852
Richard Smitha8d5b6a2015-07-17 19:51:03 +00007853IdentifierInfo *ASTReader::get(StringRef Name) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007854 // Note that we are loading an identifier.
7855 Deserializing AnIdentifier(this);
Douglas Gregore060e572013-01-25 01:03:03 +00007856
Douglas Gregor7211ac12013-01-25 23:32:03 +00007857 IdentifierLookupVisitor Visitor(Name, /*PriorGeneration=*/0,
Douglas Gregor00a50f72013-01-25 00:38:33 +00007858 NumIdentifierLookups,
7859 NumIdentifierLookupHits);
Richard Smith33e0f7e2015-07-22 02:08:40 +00007860
7861 // We don't need to do identifier table lookups in C++ modules (we preload
7862 // all interesting declarations, and don't need to use the scope for name
7863 // lookups). Perform the lookup in PCH files, though, since we don't build
7864 // a complete initial identifier table if we're carrying on from a PCH.
Richard Smithdbafb6c2017-06-29 23:23:46 +00007865 if (PP.getLangOpts().CPlusPlus) {
Richard Smith33e0f7e2015-07-22 02:08:40 +00007866 for (auto F : ModuleMgr.pch_modules())
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00007867 if (Visitor(*F))
Richard Smith33e0f7e2015-07-22 02:08:40 +00007868 break;
7869 } else {
7870 // If there is a global index, look there first to determine which modules
7871 // provably do not have any results for this identifier.
7872 GlobalModuleIndex::HitSet Hits;
7873 GlobalModuleIndex::HitSet *HitsPtr = nullptr;
7874 if (!loadGlobalIndex()) {
7875 if (GlobalIndex->lookupIdentifier(Name, Hits)) {
7876 HitsPtr = &Hits;
7877 }
7878 }
7879
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00007880 ModuleMgr.visit(Visitor, HitsPtr);
Richard Smith33e0f7e2015-07-22 02:08:40 +00007881 }
7882
Guy Benyei11169dd2012-12-18 14:30:41 +00007883 IdentifierInfo *II = Visitor.getIdentifierInfo();
7884 markIdentifierUpToDate(II);
7885 return II;
7886}
7887
7888namespace clang {
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00007889
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00007890 /// An identifier-lookup iterator that enumerates all of the
Guy Benyei11169dd2012-12-18 14:30:41 +00007891 /// identifiers stored within a set of AST files.
7892 class ASTIdentifierIterator : public IdentifierIterator {
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00007893 /// The AST reader whose identifiers are being enumerated.
Guy Benyei11169dd2012-12-18 14:30:41 +00007894 const ASTReader &Reader;
7895
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00007896 /// The current index into the chain of AST files stored in
Guy Benyei11169dd2012-12-18 14:30:41 +00007897 /// the AST reader.
7898 unsigned Index;
7899
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00007900 /// The current position within the identifier lookup table
Guy Benyei11169dd2012-12-18 14:30:41 +00007901 /// of the current AST file.
7902 ASTIdentifierLookupTable::key_iterator Current;
7903
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00007904 /// The end position within the identifier lookup table of
Guy Benyei11169dd2012-12-18 14:30:41 +00007905 /// the current AST file.
7906 ASTIdentifierLookupTable::key_iterator End;
7907
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00007908 /// Whether to skip any modules in the ASTReader.
Ben Langmuir537c5b52016-05-04 00:53:13 +00007909 bool SkipModules;
7910
Guy Benyei11169dd2012-12-18 14:30:41 +00007911 public:
Ben Langmuir537c5b52016-05-04 00:53:13 +00007912 explicit ASTIdentifierIterator(const ASTReader &Reader,
7913 bool SkipModules = false);
Guy Benyei11169dd2012-12-18 14:30:41 +00007914
Craig Topper3e89dfe2014-03-13 02:13:41 +00007915 StringRef Next() override;
Guy Benyei11169dd2012-12-18 14:30:41 +00007916 };
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00007917
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00007918} // namespace clang
Guy Benyei11169dd2012-12-18 14:30:41 +00007919
Ben Langmuir537c5b52016-05-04 00:53:13 +00007920ASTIdentifierIterator::ASTIdentifierIterator(const ASTReader &Reader,
7921 bool SkipModules)
7922 : Reader(Reader), Index(Reader.ModuleMgr.size()), SkipModules(SkipModules) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007923}
7924
7925StringRef ASTIdentifierIterator::Next() {
7926 while (Current == End) {
7927 // If we have exhausted all of our AST files, we're done.
7928 if (Index == 0)
Vedant Kumar48b4f762018-04-14 01:40:48 +00007929 return StringRef();
Guy Benyei11169dd2012-12-18 14:30:41 +00007930
7931 --Index;
Ben Langmuir537c5b52016-05-04 00:53:13 +00007932 ModuleFile &F = Reader.ModuleMgr[Index];
7933 if (SkipModules && F.isModule())
7934 continue;
7935
Vedant Kumar48b4f762018-04-14 01:40:48 +00007936 ASTIdentifierLookupTable *IdTable =
7937 (ASTIdentifierLookupTable *)F.IdentifierLookupTable;
Guy Benyei11169dd2012-12-18 14:30:41 +00007938 Current = IdTable->key_begin();
7939 End = IdTable->key_end();
7940 }
7941
7942 // We have any identifiers remaining in the current AST file; return
7943 // the next one.
Douglas Gregorbfd73d72013-01-23 18:53:14 +00007944 StringRef Result = *Current;
Guy Benyei11169dd2012-12-18 14:30:41 +00007945 ++Current;
Douglas Gregorbfd73d72013-01-23 18:53:14 +00007946 return Result;
Guy Benyei11169dd2012-12-18 14:30:41 +00007947}
7948
Ben Langmuir537c5b52016-05-04 00:53:13 +00007949namespace {
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00007950
Ben Langmuir537c5b52016-05-04 00:53:13 +00007951/// A utility for appending two IdentifierIterators.
7952class ChainedIdentifierIterator : public IdentifierIterator {
7953 std::unique_ptr<IdentifierIterator> Current;
7954 std::unique_ptr<IdentifierIterator> Queued;
7955
7956public:
7957 ChainedIdentifierIterator(std::unique_ptr<IdentifierIterator> First,
7958 std::unique_ptr<IdentifierIterator> Second)
7959 : Current(std::move(First)), Queued(std::move(Second)) {}
7960
7961 StringRef Next() override {
7962 if (!Current)
Vedant Kumar48b4f762018-04-14 01:40:48 +00007963 return StringRef();
Ben Langmuir537c5b52016-05-04 00:53:13 +00007964
7965 StringRef result = Current->Next();
7966 if (!result.empty())
7967 return result;
7968
7969 // Try the queued iterator, which may itself be empty.
7970 Current.reset();
7971 std::swap(Current, Queued);
7972 return Next();
7973 }
7974};
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00007975
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00007976} // namespace
Ben Langmuir537c5b52016-05-04 00:53:13 +00007977
Argyrios Kyrtzidis9aca3c62013-04-17 22:10:55 +00007978IdentifierIterator *ASTReader::getIdentifiers() {
Ben Langmuir537c5b52016-05-04 00:53:13 +00007979 if (!loadGlobalIndex()) {
7980 std::unique_ptr<IdentifierIterator> ReaderIter(
7981 new ASTIdentifierIterator(*this, /*SkipModules=*/true));
7982 std::unique_ptr<IdentifierIterator> ModulesIter(
7983 GlobalIndex->createIdentifierIterator());
7984 return new ChainedIdentifierIterator(std::move(ReaderIter),
7985 std::move(ModulesIter));
7986 }
Argyrios Kyrtzidis9aca3c62013-04-17 22:10:55 +00007987
Guy Benyei11169dd2012-12-18 14:30:41 +00007988 return new ASTIdentifierIterator(*this);
7989}
7990
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00007991namespace clang {
7992namespace serialization {
7993
Guy Benyei11169dd2012-12-18 14:30:41 +00007994 class ReadMethodPoolVisitor {
7995 ASTReader &Reader;
7996 Selector Sel;
7997 unsigned PriorGeneration;
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00007998 unsigned InstanceBits = 0;
7999 unsigned FactoryBits = 0;
8000 bool InstanceHasMoreThanOneDecl = false;
8001 bool FactoryHasMoreThanOneDecl = false;
Dmitri Gribenkof8579502013-01-12 19:30:44 +00008002 SmallVector<ObjCMethodDecl *, 4> InstanceMethods;
8003 SmallVector<ObjCMethodDecl *, 4> FactoryMethods;
Guy Benyei11169dd2012-12-18 14:30:41 +00008004
8005 public:
Nico Weber2e0c8f72014-12-27 03:58:08 +00008006 ReadMethodPoolVisitor(ASTReader &Reader, Selector Sel,
Guy Benyei11169dd2012-12-18 14:30:41 +00008007 unsigned PriorGeneration)
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00008008 : Reader(Reader), Sel(Sel), PriorGeneration(PriorGeneration) {}
Nico Weber2e0c8f72014-12-27 03:58:08 +00008009
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00008010 bool operator()(ModuleFile &M) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008011 if (!M.SelectorLookupTable)
8012 return false;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008013
Guy Benyei11169dd2012-12-18 14:30:41 +00008014 // If we've already searched this module file, skip it now.
Richard Smithbdf2d932015-07-30 03:37:16 +00008015 if (M.Generation <= PriorGeneration)
Guy Benyei11169dd2012-12-18 14:30:41 +00008016 return true;
8017
Richard Smithbdf2d932015-07-30 03:37:16 +00008018 ++Reader.NumMethodPoolTableLookups;
Vedant Kumar48b4f762018-04-14 01:40:48 +00008019 ASTSelectorLookupTable *PoolTable
8020 = (ASTSelectorLookupTable*)M.SelectorLookupTable;
Richard Smithbdf2d932015-07-30 03:37:16 +00008021 ASTSelectorLookupTable::iterator Pos = PoolTable->find(Sel);
Guy Benyei11169dd2012-12-18 14:30:41 +00008022 if (Pos == PoolTable->end())
8023 return false;
Douglas Gregorad2f7a52013-01-28 17:54:36 +00008024
Richard Smithbdf2d932015-07-30 03:37:16 +00008025 ++Reader.NumMethodPoolTableHits;
8026 ++Reader.NumSelectorsRead;
Guy Benyei11169dd2012-12-18 14:30:41 +00008027 // FIXME: Not quite happy with the statistics here. We probably should
8028 // disable this tracking when called via LoadSelector.
8029 // Also, should entries without methods count as misses?
Richard Smithbdf2d932015-07-30 03:37:16 +00008030 ++Reader.NumMethodPoolEntriesRead;
Guy Benyei11169dd2012-12-18 14:30:41 +00008031 ASTSelectorLookupTrait::data_type Data = *Pos;
Richard Smithbdf2d932015-07-30 03:37:16 +00008032 if (Reader.DeserializationListener)
8033 Reader.DeserializationListener->SelectorRead(Data.ID, Sel);
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00008034
Richard Smithbdf2d932015-07-30 03:37:16 +00008035 InstanceMethods.append(Data.Instance.begin(), Data.Instance.end());
8036 FactoryMethods.append(Data.Factory.begin(), Data.Factory.end());
8037 InstanceBits = Data.InstanceBits;
8038 FactoryBits = Data.FactoryBits;
8039 InstanceHasMoreThanOneDecl = Data.InstanceHasMoreThanOneDecl;
8040 FactoryHasMoreThanOneDecl = Data.FactoryHasMoreThanOneDecl;
Guy Benyei11169dd2012-12-18 14:30:41 +00008041 return true;
8042 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008043
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00008044 /// Retrieve the instance methods found by this visitor.
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008045 ArrayRef<ObjCMethodDecl *> getInstanceMethods() const {
8046 return InstanceMethods;
Guy Benyei11169dd2012-12-18 14:30:41 +00008047 }
8048
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00008049 /// Retrieve the instance methods found by this visitor.
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008050 ArrayRef<ObjCMethodDecl *> getFactoryMethods() const {
Guy Benyei11169dd2012-12-18 14:30:41 +00008051 return FactoryMethods;
8052 }
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00008053
8054 unsigned getInstanceBits() const { return InstanceBits; }
8055 unsigned getFactoryBits() const { return FactoryBits; }
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00008056
Nico Weberff4b35e2014-12-27 22:14:15 +00008057 bool instanceHasMoreThanOneDecl() const {
8058 return InstanceHasMoreThanOneDecl;
8059 }
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00008060
Nico Weberff4b35e2014-12-27 22:14:15 +00008061 bool factoryHasMoreThanOneDecl() const { return FactoryHasMoreThanOneDecl; }
Guy Benyei11169dd2012-12-18 14:30:41 +00008062 };
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00008063
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00008064} // namespace serialization
8065} // namespace clang
Guy Benyei11169dd2012-12-18 14:30:41 +00008066
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00008067/// Add the given set of methods to the method list.
Guy Benyei11169dd2012-12-18 14:30:41 +00008068static void addMethodsToPool(Sema &S, ArrayRef<ObjCMethodDecl *> Methods,
8069 ObjCMethodList &List) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008070 for (unsigned I = 0, N = Methods.size(); I != N; ++I) {
8071 S.addMethodToGlobalList(&List, Methods[I]);
8072 }
Guy Benyei11169dd2012-12-18 14:30:41 +00008073}
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008074
Guy Benyei11169dd2012-12-18 14:30:41 +00008075void ASTReader::ReadMethodPool(Selector Sel) {
8076 // Get the selector generation and update it to the current generation.
8077 unsigned &Generation = SelectorGeneration[Sel];
8078 unsigned PriorGeneration = Generation;
Richard Smith053f6c62014-05-16 23:01:30 +00008079 Generation = getGeneration();
Manman Rena0f31a02016-04-29 19:04:05 +00008080 SelectorOutOfDate[Sel] = false;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008081
Guy Benyei11169dd2012-12-18 14:30:41 +00008082 // Search for methods defined with this selector.
Douglas Gregorad2f7a52013-01-28 17:54:36 +00008083 ++NumMethodPoolLookups;
Guy Benyei11169dd2012-12-18 14:30:41 +00008084 ReadMethodPoolVisitor Visitor(*this, Sel, PriorGeneration);
Benjamin Kramer9a9efba2015-07-25 12:14:04 +00008085 ModuleMgr.visit(Visitor);
8086
Guy Benyei11169dd2012-12-18 14:30:41 +00008087 if (Visitor.getInstanceMethods().empty() &&
Douglas Gregorad2f7a52013-01-28 17:54:36 +00008088 Visitor.getFactoryMethods().empty())
Guy Benyei11169dd2012-12-18 14:30:41 +00008089 return;
Douglas Gregorad2f7a52013-01-28 17:54:36 +00008090
8091 ++NumMethodPoolHits;
8092
Guy Benyei11169dd2012-12-18 14:30:41 +00008093 if (!getSema())
8094 return;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008095
Guy Benyei11169dd2012-12-18 14:30:41 +00008096 Sema &S = *getSema();
8097 Sema::GlobalMethodPool::iterator Pos
8098 = S.MethodPool.insert(std::make_pair(Sel, Sema::GlobalMethods())).first;
Ben Langmuira0c32e92015-01-12 19:27:00 +00008099
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00008100 Pos->second.first.setBits(Visitor.getInstanceBits());
Nico Weberff4b35e2014-12-27 22:14:15 +00008101 Pos->second.first.setHasMoreThanOneDecl(Visitor.instanceHasMoreThanOneDecl());
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00008102 Pos->second.second.setBits(Visitor.getFactoryBits());
Nico Weberff4b35e2014-12-27 22:14:15 +00008103 Pos->second.second.setHasMoreThanOneDecl(Visitor.factoryHasMoreThanOneDecl());
Ben Langmuira0c32e92015-01-12 19:27:00 +00008104
8105 // Add methods to the global pool *after* setting hasMoreThanOneDecl, since
8106 // when building a module we keep every method individually and may need to
8107 // update hasMoreThanOneDecl as we add the methods.
8108 addMethodsToPool(S, Visitor.getInstanceMethods(), Pos->second.first);
8109 addMethodsToPool(S, Visitor.getFactoryMethods(), Pos->second.second);
Guy Benyei11169dd2012-12-18 14:30:41 +00008110}
8111
Manman Rena0f31a02016-04-29 19:04:05 +00008112void ASTReader::updateOutOfDateSelector(Selector Sel) {
8113 if (SelectorOutOfDate[Sel])
8114 ReadMethodPool(Sel);
8115}
8116
Guy Benyei11169dd2012-12-18 14:30:41 +00008117void ASTReader::ReadKnownNamespaces(
8118 SmallVectorImpl<NamespaceDecl *> &Namespaces) {
8119 Namespaces.clear();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008120
Vedant Kumar48b4f762018-04-14 01:40:48 +00008121 for (unsigned I = 0, N = KnownNamespaces.size(); I != N; ++I) {
8122 if (NamespaceDecl *Namespace
8123 = dyn_cast_or_null<NamespaceDecl>(GetDecl(KnownNamespaces[I])))
Guy Benyei11169dd2012-12-18 14:30:41 +00008124 Namespaces.push_back(Namespace);
Vedant Kumar48b4f762018-04-14 01:40:48 +00008125 }
Guy Benyei11169dd2012-12-18 14:30:41 +00008126}
8127
Nick Lewycky9c7eb1d2013-02-01 08:13:20 +00008128void ASTReader::ReadUndefinedButUsed(
Richard Smithd6a04d72016-03-25 21:49:43 +00008129 llvm::MapVector<NamedDecl *, SourceLocation> &Undefined) {
Nick Lewycky9c7eb1d2013-02-01 08:13:20 +00008130 for (unsigned Idx = 0, N = UndefinedButUsed.size(); Idx != N;) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008131 NamedDecl *D = cast<NamedDecl>(GetDecl(UndefinedButUsed[Idx++]));
Nick Lewycky8334af82013-01-26 00:35:08 +00008132 SourceLocation Loc =
Nick Lewycky9c7eb1d2013-02-01 08:13:20 +00008133 SourceLocation::getFromRawEncoding(UndefinedButUsed[Idx++]);
Nick Lewycky8334af82013-01-26 00:35:08 +00008134 Undefined.insert(std::make_pair(D, Loc));
8135 }
8136}
Nick Lewycky8334af82013-01-26 00:35:08 +00008137
Ismail Pazarbasie5768d12015-05-18 19:59:11 +00008138void ASTReader::ReadMismatchingDeleteExpressions(llvm::MapVector<
8139 FieldDecl *, llvm::SmallVector<std::pair<SourceLocation, bool>, 4>> &
8140 Exprs) {
8141 for (unsigned Idx = 0, N = DelayedDeleteExprs.size(); Idx != N;) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008142 FieldDecl *FD = cast<FieldDecl>(GetDecl(DelayedDeleteExprs[Idx++]));
Ismail Pazarbasie5768d12015-05-18 19:59:11 +00008143 uint64_t Count = DelayedDeleteExprs[Idx++];
8144 for (uint64_t C = 0; C < Count; ++C) {
8145 SourceLocation DeleteLoc =
8146 SourceLocation::getFromRawEncoding(DelayedDeleteExprs[Idx++]);
8147 const bool IsArrayForm = DelayedDeleteExprs[Idx++];
8148 Exprs[FD].push_back(std::make_pair(DeleteLoc, IsArrayForm));
8149 }
8150 }
8151}
8152
Guy Benyei11169dd2012-12-18 14:30:41 +00008153void ASTReader::ReadTentativeDefinitions(
8154 SmallVectorImpl<VarDecl *> &TentativeDefs) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008155 for (unsigned I = 0, N = TentativeDefinitions.size(); I != N; ++I) {
8156 VarDecl *Var = dyn_cast_or_null<VarDecl>(GetDecl(TentativeDefinitions[I]));
Guy Benyei11169dd2012-12-18 14:30:41 +00008157 if (Var)
8158 TentativeDefs.push_back(Var);
8159 }
8160 TentativeDefinitions.clear();
8161}
8162
8163void ASTReader::ReadUnusedFileScopedDecls(
8164 SmallVectorImpl<const DeclaratorDecl *> &Decls) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008165 for (unsigned I = 0, N = UnusedFileScopedDecls.size(); I != N; ++I) {
8166 DeclaratorDecl *D
8167 = dyn_cast_or_null<DeclaratorDecl>(GetDecl(UnusedFileScopedDecls[I]));
Guy Benyei11169dd2012-12-18 14:30:41 +00008168 if (D)
8169 Decls.push_back(D);
8170 }
8171 UnusedFileScopedDecls.clear();
8172}
8173
8174void ASTReader::ReadDelegatingConstructors(
8175 SmallVectorImpl<CXXConstructorDecl *> &Decls) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008176 for (unsigned I = 0, N = DelegatingCtorDecls.size(); I != N; ++I) {
8177 CXXConstructorDecl *D
8178 = dyn_cast_or_null<CXXConstructorDecl>(GetDecl(DelegatingCtorDecls[I]));
Guy Benyei11169dd2012-12-18 14:30:41 +00008179 if (D)
8180 Decls.push_back(D);
8181 }
8182 DelegatingCtorDecls.clear();
8183}
8184
8185void ASTReader::ReadExtVectorDecls(SmallVectorImpl<TypedefNameDecl *> &Decls) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008186 for (unsigned I = 0, N = ExtVectorDecls.size(); I != N; ++I) {
8187 TypedefNameDecl *D
8188 = dyn_cast_or_null<TypedefNameDecl>(GetDecl(ExtVectorDecls[I]));
Guy Benyei11169dd2012-12-18 14:30:41 +00008189 if (D)
8190 Decls.push_back(D);
8191 }
8192 ExtVectorDecls.clear();
8193}
8194
Nico Weber72889432014-09-06 01:25:55 +00008195void ASTReader::ReadUnusedLocalTypedefNameCandidates(
8196 llvm::SmallSetVector<const TypedefNameDecl *, 4> &Decls) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008197 for (unsigned I = 0, N = UnusedLocalTypedefNameCandidates.size(); I != N;
8198 ++I) {
8199 TypedefNameDecl *D = dyn_cast_or_null<TypedefNameDecl>(
8200 GetDecl(UnusedLocalTypedefNameCandidates[I]));
Nico Weber72889432014-09-06 01:25:55 +00008201 if (D)
8202 Decls.insert(D);
8203 }
8204 UnusedLocalTypedefNameCandidates.clear();
8205}
8206
Guy Benyei11169dd2012-12-18 14:30:41 +00008207void ASTReader::ReadReferencedSelectors(
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00008208 SmallVectorImpl<std::pair<Selector, SourceLocation>> &Sels) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008209 if (ReferencedSelectorsData.empty())
8210 return;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008211
Guy Benyei11169dd2012-12-18 14:30:41 +00008212 // If there are @selector references added them to its pool. This is for
8213 // implementation of -Wselector.
8214 unsigned int DataSize = ReferencedSelectorsData.size()-1;
8215 unsigned I = 0;
8216 while (I < DataSize) {
8217 Selector Sel = DecodeSelector(ReferencedSelectorsData[I++]);
8218 SourceLocation SelLoc
8219 = SourceLocation::getFromRawEncoding(ReferencedSelectorsData[I++]);
8220 Sels.push_back(std::make_pair(Sel, SelLoc));
8221 }
8222 ReferencedSelectorsData.clear();
8223}
8224
8225void ASTReader::ReadWeakUndeclaredIdentifiers(
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00008226 SmallVectorImpl<std::pair<IdentifierInfo *, WeakInfo>> &WeakIDs) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008227 if (WeakUndeclaredIdentifiers.empty())
8228 return;
8229
8230 for (unsigned I = 0, N = WeakUndeclaredIdentifiers.size(); I < N; /*none*/) {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008231 IdentifierInfo *WeakId
Guy Benyei11169dd2012-12-18 14:30:41 +00008232 = DecodeIdentifierInfo(WeakUndeclaredIdentifiers[I++]);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008233 IdentifierInfo *AliasId
Guy Benyei11169dd2012-12-18 14:30:41 +00008234 = DecodeIdentifierInfo(WeakUndeclaredIdentifiers[I++]);
8235 SourceLocation Loc
8236 = SourceLocation::getFromRawEncoding(WeakUndeclaredIdentifiers[I++]);
8237 bool Used = WeakUndeclaredIdentifiers[I++];
8238 WeakInfo WI(AliasId, Loc);
8239 WI.setUsed(Used);
8240 WeakIDs.push_back(std::make_pair(WeakId, WI));
8241 }
8242 WeakUndeclaredIdentifiers.clear();
8243}
8244
8245void ASTReader::ReadUsedVTables(SmallVectorImpl<ExternalVTableUse> &VTables) {
8246 for (unsigned Idx = 0, N = VTableUses.size(); Idx < N; /* In loop */) {
8247 ExternalVTableUse VT;
8248 VT.Record = dyn_cast_or_null<CXXRecordDecl>(GetDecl(VTableUses[Idx++]));
8249 VT.Location = SourceLocation::getFromRawEncoding(VTableUses[Idx++]);
8250 VT.DefinitionRequired = VTableUses[Idx++];
8251 VTables.push_back(VT);
8252 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008253
Guy Benyei11169dd2012-12-18 14:30:41 +00008254 VTableUses.clear();
8255}
8256
8257void ASTReader::ReadPendingInstantiations(
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00008258 SmallVectorImpl<std::pair<ValueDecl *, SourceLocation>> &Pending) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008259 for (unsigned Idx = 0, N = PendingInstantiations.size(); Idx < N;) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008260 ValueDecl *D = cast<ValueDecl>(GetDecl(PendingInstantiations[Idx++]));
Guy Benyei11169dd2012-12-18 14:30:41 +00008261 SourceLocation Loc
8262 = SourceLocation::getFromRawEncoding(PendingInstantiations[Idx++]);
8263
8264 Pending.push_back(std::make_pair(D, Loc));
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008265 }
Guy Benyei11169dd2012-12-18 14:30:41 +00008266 PendingInstantiations.clear();
8267}
8268
Richard Smithe40f2ba2013-08-07 21:41:30 +00008269void ASTReader::ReadLateParsedTemplates(
Justin Lebar28f09c52016-10-10 16:26:08 +00008270 llvm::MapVector<const FunctionDecl *, std::unique_ptr<LateParsedTemplate>>
8271 &LPTMap) {
Richard Smithe40f2ba2013-08-07 21:41:30 +00008272 for (unsigned Idx = 0, N = LateParsedTemplates.size(); Idx < N;
8273 /* In loop */) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008274 FunctionDecl *FD = cast<FunctionDecl>(GetDecl(LateParsedTemplates[Idx++]));
Richard Smithe40f2ba2013-08-07 21:41:30 +00008275
Justin Lebar28f09c52016-10-10 16:26:08 +00008276 auto LT = llvm::make_unique<LateParsedTemplate>();
Richard Smithe40f2ba2013-08-07 21:41:30 +00008277 LT->D = GetDecl(LateParsedTemplates[Idx++]);
8278
8279 ModuleFile *F = getOwningModuleFile(LT->D);
8280 assert(F && "No module");
8281
8282 unsigned TokN = LateParsedTemplates[Idx++];
8283 LT->Toks.reserve(TokN);
8284 for (unsigned T = 0; T < TokN; ++T)
8285 LT->Toks.push_back(ReadToken(*F, LateParsedTemplates, Idx));
8286
Justin Lebar28f09c52016-10-10 16:26:08 +00008287 LPTMap.insert(std::make_pair(FD, std::move(LT)));
Richard Smithe40f2ba2013-08-07 21:41:30 +00008288 }
8289
8290 LateParsedTemplates.clear();
8291}
8292
Guy Benyei11169dd2012-12-18 14:30:41 +00008293void ASTReader::LoadSelector(Selector Sel) {
8294 // It would be complicated to avoid reading the methods anyway. So don't.
8295 ReadMethodPool(Sel);
8296}
8297
8298void ASTReader::SetIdentifierInfo(IdentifierID ID, IdentifierInfo *II) {
8299 assert(ID && "Non-zero identifier ID required");
8300 assert(ID <= IdentifiersLoaded.size() && "identifier ID out of range");
8301 IdentifiersLoaded[ID - 1] = II;
8302 if (DeserializationListener)
8303 DeserializationListener->IdentifierRead(ID, II);
8304}
8305
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00008306/// Set the globally-visible declarations associated with the given
Guy Benyei11169dd2012-12-18 14:30:41 +00008307/// identifier.
8308///
8309/// If the AST reader is currently in a state where the given declaration IDs
8310/// cannot safely be resolved, they are queued until it is safe to resolve
8311/// them.
8312///
8313/// \param II an IdentifierInfo that refers to one or more globally-visible
8314/// declarations.
8315///
8316/// \param DeclIDs the set of declaration IDs with the name @p II that are
8317/// visible at global scope.
8318///
Douglas Gregor6168bd22013-02-18 15:53:43 +00008319/// \param Decls if non-null, this vector will be populated with the set of
8320/// deserialized declarations. These declarations will not be pushed into
8321/// scope.
Guy Benyei11169dd2012-12-18 14:30:41 +00008322void
8323ASTReader::SetGloballyVisibleDecls(IdentifierInfo *II,
8324 const SmallVectorImpl<uint32_t> &DeclIDs,
Douglas Gregor6168bd22013-02-18 15:53:43 +00008325 SmallVectorImpl<Decl *> *Decls) {
8326 if (NumCurrentElementsDeserializing && !Decls) {
8327 PendingIdentifierInfos[II].append(DeclIDs.begin(), DeclIDs.end());
Guy Benyei11169dd2012-12-18 14:30:41 +00008328 return;
8329 }
8330
Vedant Kumar48b4f762018-04-14 01:40:48 +00008331 for (unsigned I = 0, N = DeclIDs.size(); I != N; ++I) {
Ben Langmuir5418f402014-09-10 21:29:41 +00008332 if (!SemaObj) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008333 // Queue this declaration so that it will be added to the
8334 // translation unit scope and identifier's declaration chain
8335 // once a Sema object is known.
Vedant Kumar48b4f762018-04-14 01:40:48 +00008336 PreloadedDeclIDs.push_back(DeclIDs[I]);
Ben Langmuir5418f402014-09-10 21:29:41 +00008337 continue;
Guy Benyei11169dd2012-12-18 14:30:41 +00008338 }
Ben Langmuir5418f402014-09-10 21:29:41 +00008339
Vedant Kumar48b4f762018-04-14 01:40:48 +00008340 NamedDecl *D = cast<NamedDecl>(GetDecl(DeclIDs[I]));
Ben Langmuir5418f402014-09-10 21:29:41 +00008341
8342 // If we're simply supposed to record the declarations, do so now.
8343 if (Decls) {
8344 Decls->push_back(D);
8345 continue;
8346 }
8347
8348 // Introduce this declaration into the translation-unit scope
8349 // and add it to the declaration chain for this identifier, so
8350 // that (unqualified) name lookup will find it.
8351 pushExternalDeclIntoScope(D, II);
Guy Benyei11169dd2012-12-18 14:30:41 +00008352 }
8353}
8354
Douglas Gregorc8a992f2013-01-21 16:52:34 +00008355IdentifierInfo *ASTReader::DecodeIdentifierInfo(IdentifierID ID) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008356 if (ID == 0)
Craig Toppera13603a2014-05-22 05:54:18 +00008357 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00008358
8359 if (IdentifiersLoaded.empty()) {
8360 Error("no identifier table in AST file");
Craig Toppera13603a2014-05-22 05:54:18 +00008361 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00008362 }
8363
8364 ID -= 1;
8365 if (!IdentifiersLoaded[ID]) {
8366 GlobalIdentifierMapType::iterator I = GlobalIdentifierMap.find(ID + 1);
8367 assert(I != GlobalIdentifierMap.end() && "Corrupted global identifier map");
8368 ModuleFile *M = I->second;
8369 unsigned Index = ID - M->BaseIdentifierID;
8370 const char *Str = M->IdentifierTableData + M->IdentifierOffsets[Index];
8371
8372 // All of the strings in the AST file are preceded by a 16-bit length.
8373 // Extract that 16-bit length to avoid having to execute strlen().
8374 // NOTE: 'StrLenPtr' is an 'unsigned char*' so that we load bytes as
8375 // unsigned integers. This is important to avoid integer overflow when
8376 // we cast them to 'unsigned'.
8377 const unsigned char *StrLenPtr = (const unsigned char*) Str - 2;
8378 unsigned StrLen = (((unsigned) StrLenPtr[0])
8379 | (((unsigned) StrLenPtr[1]) << 8)) - 1;
Richard Smitheb4b58f62016-02-05 01:40:54 +00008380 auto &II = PP.getIdentifierTable().get(StringRef(Str, StrLen));
8381 IdentifiersLoaded[ID] = &II;
8382 markIdentifierFromAST(*this, II);
Guy Benyei11169dd2012-12-18 14:30:41 +00008383 if (DeserializationListener)
Richard Smitheb4b58f62016-02-05 01:40:54 +00008384 DeserializationListener->IdentifierRead(ID + 1, &II);
Guy Benyei11169dd2012-12-18 14:30:41 +00008385 }
8386
8387 return IdentifiersLoaded[ID];
8388}
8389
Douglas Gregorc8a992f2013-01-21 16:52:34 +00008390IdentifierInfo *ASTReader::getLocalIdentifier(ModuleFile &M, unsigned LocalID) {
8391 return DecodeIdentifierInfo(getGlobalIdentifierID(M, LocalID));
Guy Benyei11169dd2012-12-18 14:30:41 +00008392}
8393
8394IdentifierID ASTReader::getGlobalIdentifierID(ModuleFile &M, unsigned LocalID) {
8395 if (LocalID < NUM_PREDEF_IDENT_IDS)
8396 return LocalID;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008397
Richard Smith37a93df2017-02-18 00:32:02 +00008398 if (!M.ModuleOffsetMap.empty())
8399 ReadModuleOffsetMap(M);
8400
Guy Benyei11169dd2012-12-18 14:30:41 +00008401 ContinuousRangeMap<uint32_t, int, 2>::iterator I
8402 = M.IdentifierRemap.find(LocalID - NUM_PREDEF_IDENT_IDS);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008403 assert(I != M.IdentifierRemap.end()
Guy Benyei11169dd2012-12-18 14:30:41 +00008404 && "Invalid index into identifier index remap");
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008405
Guy Benyei11169dd2012-12-18 14:30:41 +00008406 return LocalID + I->second;
8407}
8408
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00008409MacroInfo *ASTReader::getMacro(MacroID ID) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008410 if (ID == 0)
Craig Toppera13603a2014-05-22 05:54:18 +00008411 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00008412
8413 if (MacrosLoaded.empty()) {
8414 Error("no macro table in AST file");
Craig Toppera13603a2014-05-22 05:54:18 +00008415 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00008416 }
8417
8418 ID -= NUM_PREDEF_MACRO_IDS;
8419 if (!MacrosLoaded[ID]) {
8420 GlobalMacroMapType::iterator I
8421 = GlobalMacroMap.find(ID + NUM_PREDEF_MACRO_IDS);
8422 assert(I != GlobalMacroMap.end() && "Corrupted global macro map");
8423 ModuleFile *M = I->second;
8424 unsigned Index = ID - M->BaseMacroID;
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00008425 MacrosLoaded[ID] = ReadMacroRecord(*M, M->MacroOffsets[Index]);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008426
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00008427 if (DeserializationListener)
8428 DeserializationListener->MacroRead(ID + NUM_PREDEF_MACRO_IDS,
8429 MacrosLoaded[ID]);
Guy Benyei11169dd2012-12-18 14:30:41 +00008430 }
8431
8432 return MacrosLoaded[ID];
8433}
8434
8435MacroID ASTReader::getGlobalMacroID(ModuleFile &M, unsigned LocalID) {
8436 if (LocalID < NUM_PREDEF_MACRO_IDS)
8437 return LocalID;
8438
Richard Smith37a93df2017-02-18 00:32:02 +00008439 if (!M.ModuleOffsetMap.empty())
8440 ReadModuleOffsetMap(M);
8441
Guy Benyei11169dd2012-12-18 14:30:41 +00008442 ContinuousRangeMap<uint32_t, int, 2>::iterator I
8443 = M.MacroRemap.find(LocalID - NUM_PREDEF_MACRO_IDS);
8444 assert(I != M.MacroRemap.end() && "Invalid index into macro index remap");
8445
8446 return LocalID + I->second;
8447}
8448
8449serialization::SubmoduleID
8450ASTReader::getGlobalSubmoduleID(ModuleFile &M, unsigned LocalID) {
8451 if (LocalID < NUM_PREDEF_SUBMODULE_IDS)
8452 return LocalID;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008453
Richard Smith37a93df2017-02-18 00:32:02 +00008454 if (!M.ModuleOffsetMap.empty())
8455 ReadModuleOffsetMap(M);
8456
Guy Benyei11169dd2012-12-18 14:30:41 +00008457 ContinuousRangeMap<uint32_t, int, 2>::iterator I
8458 = M.SubmoduleRemap.find(LocalID - NUM_PREDEF_SUBMODULE_IDS);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008459 assert(I != M.SubmoduleRemap.end()
Guy Benyei11169dd2012-12-18 14:30:41 +00008460 && "Invalid index into submodule index remap");
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008461
Guy Benyei11169dd2012-12-18 14:30:41 +00008462 return LocalID + I->second;
8463}
8464
8465Module *ASTReader::getSubmodule(SubmoduleID GlobalID) {
8466 if (GlobalID < NUM_PREDEF_SUBMODULE_IDS) {
8467 assert(GlobalID == 0 && "Unhandled global submodule ID");
Craig Toppera13603a2014-05-22 05:54:18 +00008468 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00008469 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008470
Guy Benyei11169dd2012-12-18 14:30:41 +00008471 if (GlobalID > SubmodulesLoaded.size()) {
8472 Error("submodule ID out of range in AST file");
Craig Toppera13603a2014-05-22 05:54:18 +00008473 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00008474 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008475
Guy Benyei11169dd2012-12-18 14:30:41 +00008476 return SubmodulesLoaded[GlobalID - NUM_PREDEF_SUBMODULE_IDS];
8477}
Douglas Gregorc147b0b2013-01-12 01:29:50 +00008478
8479Module *ASTReader::getModule(unsigned ID) {
8480 return getSubmodule(ID);
8481}
8482
Hans Wennborg08c5a7b2018-06-25 13:23:49 +00008483bool ASTReader::DeclIsFromPCHWithObjectFile(const Decl *D) {
8484 ModuleFile *MF = getOwningModuleFile(D);
8485 return MF && MF->PCHHasObjectFile;
8486}
8487
Richard Smithd88a7f12015-09-01 20:35:42 +00008488ModuleFile *ASTReader::getLocalModuleFile(ModuleFile &F, unsigned ID) {
8489 if (ID & 1) {
8490 // It's a module, look it up by submodule ID.
8491 auto I = GlobalSubmoduleMap.find(getGlobalSubmoduleID(F, ID >> 1));
8492 return I == GlobalSubmoduleMap.end() ? nullptr : I->second;
8493 } else {
8494 // It's a prefix (preamble, PCH, ...). Look it up by index.
8495 unsigned IndexFromEnd = ID >> 1;
8496 assert(IndexFromEnd && "got reference to unknown module file");
8497 return getModuleManager().pch_modules().end()[-IndexFromEnd];
8498 }
8499}
8500
8501unsigned ASTReader::getModuleFileID(ModuleFile *F) {
8502 if (!F)
8503 return 1;
8504
8505 // For a file representing a module, use the submodule ID of the top-level
8506 // module as the file ID. For any other kind of file, the number of such
8507 // files loaded beforehand will be the same on reload.
8508 // FIXME: Is this true even if we have an explicit module file and a PCH?
8509 if (F->isModule())
8510 return ((F->BaseSubmoduleID + NUM_PREDEF_SUBMODULE_IDS) << 1) | 1;
8511
8512 auto PCHModules = getModuleManager().pch_modules();
8513 auto I = std::find(PCHModules.begin(), PCHModules.end(), F);
8514 assert(I != PCHModules.end() && "emitting reference to unknown file");
8515 return (I - PCHModules.end()) << 1;
8516}
8517
Adrian Prantl15bcf702015-06-30 17:39:43 +00008518llvm::Optional<ExternalASTSource::ASTSourceDescriptor>
8519ASTReader::getSourceDescriptor(unsigned ID) {
8520 if (const Module *M = getSubmodule(ID))
Adrian Prantlc6458d62015-09-19 00:10:32 +00008521 return ExternalASTSource::ASTSourceDescriptor(*M);
Adrian Prantl15bcf702015-06-30 17:39:43 +00008522
8523 // If there is only a single PCH, return it instead.
Hiroshi Inoue3170de02017-07-01 08:46:43 +00008524 // Chained PCH are not supported.
Saleem Abdulrasool97d25552017-03-02 17:37:11 +00008525 const auto &PCHChain = ModuleMgr.pch_modules();
8526 if (std::distance(std::begin(PCHChain), std::end(PCHChain))) {
Adrian Prantl15bcf702015-06-30 17:39:43 +00008527 ModuleFile &MF = ModuleMgr.getPrimaryModule();
Adrian Prantl3a2d4942016-01-22 23:30:56 +00008528 StringRef ModuleName = llvm::sys::path::filename(MF.OriginalSourceFileName);
Adrian Prantl9bc3c4f2016-04-27 17:06:22 +00008529 StringRef FileName = llvm::sys::path::filename(MF.FileName);
8530 return ASTReader::ASTSourceDescriptor(ModuleName, MF.OriginalDir, FileName,
8531 MF.Signature);
Adrian Prantl15bcf702015-06-30 17:39:43 +00008532 }
8533 return None;
8534}
8535
David Blaikie1ac9c982017-04-11 21:13:37 +00008536ExternalASTSource::ExtKind ASTReader::hasExternalDefinitions(const Decl *FD) {
Richard Smitha4653622017-09-06 20:01:14 +00008537 auto I = DefinitionSource.find(FD);
8538 if (I == DefinitionSource.end())
David Blaikie9ffe5a32017-01-30 05:00:26 +00008539 return EK_ReplyHazy;
David Blaikiee6b7c282017-04-11 20:46:34 +00008540 return I->second ? EK_Never : EK_Always;
David Blaikie9ffe5a32017-01-30 05:00:26 +00008541}
8542
Guy Benyei11169dd2012-12-18 14:30:41 +00008543Selector ASTReader::getLocalSelector(ModuleFile &M, unsigned LocalID) {
8544 return DecodeSelector(getGlobalSelectorID(M, LocalID));
8545}
8546
8547Selector ASTReader::DecodeSelector(serialization::SelectorID ID) {
8548 if (ID == 0)
Vedant Kumar48b4f762018-04-14 01:40:48 +00008549 return Selector();
Guy Benyei11169dd2012-12-18 14:30:41 +00008550
8551 if (ID > SelectorsLoaded.size()) {
8552 Error("selector ID out of range in AST file");
Vedant Kumar48b4f762018-04-14 01:40:48 +00008553 return Selector();
Guy Benyei11169dd2012-12-18 14:30:41 +00008554 }
8555
Craig Toppera13603a2014-05-22 05:54:18 +00008556 if (SelectorsLoaded[ID - 1].getAsOpaquePtr() == nullptr) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008557 // Load this selector from the selector table.
8558 GlobalSelectorMapType::iterator I = GlobalSelectorMap.find(ID);
8559 assert(I != GlobalSelectorMap.end() && "Corrupted global selector map");
8560 ModuleFile &M = *I->second;
8561 ASTSelectorLookupTrait Trait(*this, M);
8562 unsigned Idx = ID - M.BaseSelectorID - NUM_PREDEF_SELECTOR_IDS;
8563 SelectorsLoaded[ID - 1] =
8564 Trait.ReadKey(M.SelectorLookupTableData + M.SelectorOffsets[Idx], 0);
8565 if (DeserializationListener)
8566 DeserializationListener->SelectorRead(ID, SelectorsLoaded[ID - 1]);
8567 }
8568
8569 return SelectorsLoaded[ID - 1];
8570}
8571
8572Selector ASTReader::GetExternalSelector(serialization::SelectorID ID) {
8573 return DecodeSelector(ID);
8574}
8575
8576uint32_t ASTReader::GetNumExternalSelectors() {
8577 // ID 0 (the null selector) is considered an external selector.
8578 return getTotalNumSelectors() + 1;
8579}
8580
8581serialization::SelectorID
8582ASTReader::getGlobalSelectorID(ModuleFile &M, unsigned LocalID) const {
8583 if (LocalID < NUM_PREDEF_SELECTOR_IDS)
8584 return LocalID;
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008585
Richard Smith37a93df2017-02-18 00:32:02 +00008586 if (!M.ModuleOffsetMap.empty())
8587 ReadModuleOffsetMap(M);
8588
Guy Benyei11169dd2012-12-18 14:30:41 +00008589 ContinuousRangeMap<uint32_t, int, 2>::iterator I
8590 = M.SelectorRemap.find(LocalID - NUM_PREDEF_SELECTOR_IDS);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008591 assert(I != M.SelectorRemap.end()
Guy Benyei11169dd2012-12-18 14:30:41 +00008592 && "Invalid index into selector index remap");
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008593
Guy Benyei11169dd2012-12-18 14:30:41 +00008594 return LocalID + I->second;
8595}
8596
8597DeclarationName
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008598ASTReader::ReadDeclarationName(ModuleFile &F,
Guy Benyei11169dd2012-12-18 14:30:41 +00008599 const RecordData &Record, unsigned &Idx) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00008600 ASTContext &Context = getContext();
Vedant Kumar48b4f762018-04-14 01:40:48 +00008601 DeclarationName::NameKind Kind = (DeclarationName::NameKind)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00008602 switch (Kind) {
8603 case DeclarationName::Identifier:
Douglas Gregorc8a992f2013-01-21 16:52:34 +00008604 return DeclarationName(GetIdentifierInfo(F, Record, Idx));
Guy Benyei11169dd2012-12-18 14:30:41 +00008605
8606 case DeclarationName::ObjCZeroArgSelector:
8607 case DeclarationName::ObjCOneArgSelector:
8608 case DeclarationName::ObjCMultiArgSelector:
8609 return DeclarationName(ReadSelector(F, Record, Idx));
8610
8611 case DeclarationName::CXXConstructorName:
8612 return Context.DeclarationNames.getCXXConstructorName(
8613 Context.getCanonicalType(readType(F, Record, Idx)));
8614
8615 case DeclarationName::CXXDestructorName:
8616 return Context.DeclarationNames.getCXXDestructorName(
8617 Context.getCanonicalType(readType(F, Record, Idx)));
8618
Richard Smith35845152017-02-07 01:37:30 +00008619 case DeclarationName::CXXDeductionGuideName:
8620 return Context.DeclarationNames.getCXXDeductionGuideName(
8621 ReadDeclAs<TemplateDecl>(F, Record, Idx));
8622
Guy Benyei11169dd2012-12-18 14:30:41 +00008623 case DeclarationName::CXXConversionFunctionName:
8624 return Context.DeclarationNames.getCXXConversionFunctionName(
8625 Context.getCanonicalType(readType(F, Record, Idx)));
8626
8627 case DeclarationName::CXXOperatorName:
8628 return Context.DeclarationNames.getCXXOperatorName(
8629 (OverloadedOperatorKind)Record[Idx++]);
8630
8631 case DeclarationName::CXXLiteralOperatorName:
8632 return Context.DeclarationNames.getCXXLiteralOperatorName(
8633 GetIdentifierInfo(F, Record, Idx));
8634
8635 case DeclarationName::CXXUsingDirective:
8636 return DeclarationName::getUsingDirectiveName();
8637 }
8638
8639 llvm_unreachable("Invalid NameKind!");
8640}
8641
8642void ASTReader::ReadDeclarationNameLoc(ModuleFile &F,
8643 DeclarationNameLoc &DNLoc,
8644 DeclarationName Name,
8645 const RecordData &Record, unsigned &Idx) {
8646 switch (Name.getNameKind()) {
8647 case DeclarationName::CXXConstructorName:
8648 case DeclarationName::CXXDestructorName:
8649 case DeclarationName::CXXConversionFunctionName:
8650 DNLoc.NamedType.TInfo = GetTypeSourceInfo(F, Record, Idx);
8651 break;
8652
8653 case DeclarationName::CXXOperatorName:
8654 DNLoc.CXXOperatorName.BeginOpNameLoc
8655 = ReadSourceLocation(F, Record, Idx).getRawEncoding();
8656 DNLoc.CXXOperatorName.EndOpNameLoc
8657 = ReadSourceLocation(F, Record, Idx).getRawEncoding();
8658 break;
8659
8660 case DeclarationName::CXXLiteralOperatorName:
8661 DNLoc.CXXLiteralOperatorName.OpNameLoc
8662 = ReadSourceLocation(F, Record, Idx).getRawEncoding();
8663 break;
8664
8665 case DeclarationName::Identifier:
8666 case DeclarationName::ObjCZeroArgSelector:
8667 case DeclarationName::ObjCOneArgSelector:
8668 case DeclarationName::ObjCMultiArgSelector:
8669 case DeclarationName::CXXUsingDirective:
Richard Smith35845152017-02-07 01:37:30 +00008670 case DeclarationName::CXXDeductionGuideName:
Guy Benyei11169dd2012-12-18 14:30:41 +00008671 break;
8672 }
8673}
8674
8675void ASTReader::ReadDeclarationNameInfo(ModuleFile &F,
8676 DeclarationNameInfo &NameInfo,
8677 const RecordData &Record, unsigned &Idx) {
8678 NameInfo.setName(ReadDeclarationName(F, Record, Idx));
8679 NameInfo.setLoc(ReadSourceLocation(F, Record, Idx));
8680 DeclarationNameLoc DNLoc;
8681 ReadDeclarationNameLoc(F, DNLoc, NameInfo.getName(), Record, Idx);
8682 NameInfo.setInfo(DNLoc);
8683}
8684
8685void ASTReader::ReadQualifierInfo(ModuleFile &F, QualifierInfo &Info,
8686 const RecordData &Record, unsigned &Idx) {
8687 Info.QualifierLoc = ReadNestedNameSpecifierLoc(F, Record, Idx);
8688 unsigned NumTPLists = Record[Idx++];
8689 Info.NumTemplParamLists = NumTPLists;
8690 if (NumTPLists) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00008691 Info.TemplParamLists =
8692 new (getContext()) TemplateParameterList *[NumTPLists];
Eugene Zelenkoe95e7d52016-09-07 21:53:17 +00008693 for (unsigned i = 0; i != NumTPLists; ++i)
Guy Benyei11169dd2012-12-18 14:30:41 +00008694 Info.TemplParamLists[i] = ReadTemplateParameterList(F, Record, Idx);
8695 }
8696}
8697
8698TemplateName
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008699ASTReader::ReadTemplateName(ModuleFile &F, const RecordData &Record,
Guy Benyei11169dd2012-12-18 14:30:41 +00008700 unsigned &Idx) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00008701 ASTContext &Context = getContext();
Vedant Kumar48b4f762018-04-14 01:40:48 +00008702 TemplateName::NameKind Kind = (TemplateName::NameKind)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00008703 switch (Kind) {
8704 case TemplateName::Template:
8705 return TemplateName(ReadDeclAs<TemplateDecl>(F, Record, Idx));
8706
8707 case TemplateName::OverloadedTemplate: {
8708 unsigned size = Record[Idx++];
8709 UnresolvedSet<8> Decls;
8710 while (size--)
8711 Decls.addDecl(ReadDeclAs<NamedDecl>(F, Record, Idx));
8712
8713 return Context.getOverloadedTemplateName(Decls.begin(), Decls.end());
8714 }
8715
8716 case TemplateName::QualifiedTemplate: {
8717 NestedNameSpecifier *NNS = ReadNestedNameSpecifier(F, Record, Idx);
8718 bool hasTemplKeyword = Record[Idx++];
Vedant Kumar48b4f762018-04-14 01:40:48 +00008719 TemplateDecl *Template = ReadDeclAs<TemplateDecl>(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00008720 return Context.getQualifiedTemplateName(NNS, hasTemplKeyword, Template);
8721 }
8722
8723 case TemplateName::DependentTemplate: {
8724 NestedNameSpecifier *NNS = ReadNestedNameSpecifier(F, Record, Idx);
8725 if (Record[Idx++]) // isIdentifier
8726 return Context.getDependentTemplateName(NNS,
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008727 GetIdentifierInfo(F, Record,
Guy Benyei11169dd2012-12-18 14:30:41 +00008728 Idx));
8729 return Context.getDependentTemplateName(NNS,
8730 (OverloadedOperatorKind)Record[Idx++]);
8731 }
8732
8733 case TemplateName::SubstTemplateTemplateParm: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008734 TemplateTemplateParmDecl *param
8735 = ReadDeclAs<TemplateTemplateParmDecl>(F, Record, Idx);
8736 if (!param) return TemplateName();
8737 TemplateName replacement = ReadTemplateName(F, Record, Idx);
8738 return Context.getSubstTemplateTemplateParm(param, replacement);
Guy Benyei11169dd2012-12-18 14:30:41 +00008739 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008740
Guy Benyei11169dd2012-12-18 14:30:41 +00008741 case TemplateName::SubstTemplateTemplateParmPack: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008742 TemplateTemplateParmDecl *Param
8743 = ReadDeclAs<TemplateTemplateParmDecl>(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00008744 if (!Param)
Vedant Kumar48b4f762018-04-14 01:40:48 +00008745 return TemplateName();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008746
Guy Benyei11169dd2012-12-18 14:30:41 +00008747 TemplateArgument ArgPack = ReadTemplateArgument(F, Record, Idx);
8748 if (ArgPack.getKind() != TemplateArgument::Pack)
Vedant Kumar48b4f762018-04-14 01:40:48 +00008749 return TemplateName();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008750
Guy Benyei11169dd2012-12-18 14:30:41 +00008751 return Context.getSubstTemplateTemplateParmPack(Param, ArgPack);
8752 }
8753 }
8754
8755 llvm_unreachable("Unhandled template name kind!");
8756}
8757
Richard Smith2bb3c342015-08-09 01:05:31 +00008758TemplateArgument ASTReader::ReadTemplateArgument(ModuleFile &F,
8759 const RecordData &Record,
8760 unsigned &Idx,
8761 bool Canonicalize) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00008762 ASTContext &Context = getContext();
Richard Smith2bb3c342015-08-09 01:05:31 +00008763 if (Canonicalize) {
8764 // The caller wants a canonical template argument. Sometimes the AST only
8765 // wants template arguments in canonical form (particularly as the template
8766 // argument lists of template specializations) so ensure we preserve that
8767 // canonical form across serialization.
8768 TemplateArgument Arg = ReadTemplateArgument(F, Record, Idx, false);
8769 return Context.getCanonicalTemplateArgument(Arg);
8770 }
8771
Vedant Kumar48b4f762018-04-14 01:40:48 +00008772 TemplateArgument::ArgKind Kind = (TemplateArgument::ArgKind)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00008773 switch (Kind) {
8774 case TemplateArgument::Null:
Vedant Kumar48b4f762018-04-14 01:40:48 +00008775 return TemplateArgument();
Guy Benyei11169dd2012-12-18 14:30:41 +00008776 case TemplateArgument::Type:
8777 return TemplateArgument(readType(F, Record, Idx));
8778 case TemplateArgument::Declaration: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008779 ValueDecl *D = ReadDeclAs<ValueDecl>(F, Record, Idx);
David Blaikie0f62c8d2014-10-16 04:21:25 +00008780 return TemplateArgument(D, readType(F, Record, Idx));
Guy Benyei11169dd2012-12-18 14:30:41 +00008781 }
8782 case TemplateArgument::NullPtr:
8783 return TemplateArgument(readType(F, Record, Idx), /*isNullPtr*/true);
8784 case TemplateArgument::Integral: {
8785 llvm::APSInt Value = ReadAPSInt(Record, Idx);
8786 QualType T = readType(F, Record, Idx);
8787 return TemplateArgument(Context, Value, T);
8788 }
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008789 case TemplateArgument::Template:
Guy Benyei11169dd2012-12-18 14:30:41 +00008790 return TemplateArgument(ReadTemplateName(F, Record, Idx));
8791 case TemplateArgument::TemplateExpansion: {
8792 TemplateName Name = ReadTemplateName(F, Record, Idx);
David Blaikie05785d12013-02-20 22:23:23 +00008793 Optional<unsigned> NumTemplateExpansions;
Guy Benyei11169dd2012-12-18 14:30:41 +00008794 if (unsigned NumExpansions = Record[Idx++])
8795 NumTemplateExpansions = NumExpansions - 1;
8796 return TemplateArgument(Name, NumTemplateExpansions);
8797 }
8798 case TemplateArgument::Expression:
8799 return TemplateArgument(ReadExpr(F));
8800 case TemplateArgument::Pack: {
8801 unsigned NumArgs = Record[Idx++];
Vedant Kumar48b4f762018-04-14 01:40:48 +00008802 TemplateArgument *Args = new (Context) TemplateArgument[NumArgs];
Guy Benyei11169dd2012-12-18 14:30:41 +00008803 for (unsigned I = 0; I != NumArgs; ++I)
8804 Args[I] = ReadTemplateArgument(F, Record, Idx);
Benjamin Kramercce63472015-08-05 09:40:22 +00008805 return TemplateArgument(llvm::makeArrayRef(Args, NumArgs));
Guy Benyei11169dd2012-12-18 14:30:41 +00008806 }
8807 }
8808
8809 llvm_unreachable("Unhandled template argument kind!");
8810}
8811
8812TemplateParameterList *
8813ASTReader::ReadTemplateParameterList(ModuleFile &F,
8814 const RecordData &Record, unsigned &Idx) {
8815 SourceLocation TemplateLoc = ReadSourceLocation(F, Record, Idx);
8816 SourceLocation LAngleLoc = ReadSourceLocation(F, Record, Idx);
8817 SourceLocation RAngleLoc = ReadSourceLocation(F, Record, Idx);
8818
8819 unsigned NumParams = Record[Idx++];
8820 SmallVector<NamedDecl *, 16> Params;
8821 Params.reserve(NumParams);
8822 while (NumParams--)
8823 Params.push_back(ReadDeclAs<NamedDecl>(F, Record, Idx));
8824
Hubert Tonge4a0c0e2016-07-30 22:33:34 +00008825 // TODO: Concepts
Richard Smithdbafb6c2017-06-29 23:23:46 +00008826 TemplateParameterList *TemplateParams = TemplateParameterList::Create(
8827 getContext(), TemplateLoc, LAngleLoc, Params, RAngleLoc, nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00008828 return TemplateParams;
8829}
8830
8831void
8832ASTReader::
Craig Topper5603df42013-07-05 19:34:19 +00008833ReadTemplateArgumentList(SmallVectorImpl<TemplateArgument> &TemplArgs,
Guy Benyei11169dd2012-12-18 14:30:41 +00008834 ModuleFile &F, const RecordData &Record,
Richard Smith2bb3c342015-08-09 01:05:31 +00008835 unsigned &Idx, bool Canonicalize) {
Guy Benyei11169dd2012-12-18 14:30:41 +00008836 unsigned NumTemplateArgs = Record[Idx++];
8837 TemplArgs.reserve(NumTemplateArgs);
8838 while (NumTemplateArgs--)
Richard Smith2bb3c342015-08-09 01:05:31 +00008839 TemplArgs.push_back(ReadTemplateArgument(F, Record, Idx, Canonicalize));
Guy Benyei11169dd2012-12-18 14:30:41 +00008840}
8841
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00008842/// Read a UnresolvedSet structure.
Richard Smitha4ba74c2013-08-30 04:46:40 +00008843void ASTReader::ReadUnresolvedSet(ModuleFile &F, LazyASTUnresolvedSet &Set,
Guy Benyei11169dd2012-12-18 14:30:41 +00008844 const RecordData &Record, unsigned &Idx) {
8845 unsigned NumDecls = Record[Idx++];
Richard Smithdbafb6c2017-06-29 23:23:46 +00008846 Set.reserve(getContext(), NumDecls);
Guy Benyei11169dd2012-12-18 14:30:41 +00008847 while (NumDecls--) {
Richard Smitha4ba74c2013-08-30 04:46:40 +00008848 DeclID ID = ReadDeclID(F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00008849 AccessSpecifier AS = (AccessSpecifier)Record[Idx++];
Richard Smithdbafb6c2017-06-29 23:23:46 +00008850 Set.addLazyDecl(getContext(), ID, AS);
Guy Benyei11169dd2012-12-18 14:30:41 +00008851 }
8852}
8853
8854CXXBaseSpecifier
8855ASTReader::ReadCXXBaseSpecifier(ModuleFile &F,
8856 const RecordData &Record, unsigned &Idx) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008857 bool isVirtual = static_cast<bool>(Record[Idx++]);
8858 bool isBaseOfClass = static_cast<bool>(Record[Idx++]);
8859 AccessSpecifier AS = static_cast<AccessSpecifier>(Record[Idx++]);
8860 bool inheritConstructors = static_cast<bool>(Record[Idx++]);
Guy Benyei11169dd2012-12-18 14:30:41 +00008861 TypeSourceInfo *TInfo = GetTypeSourceInfo(F, Record, Idx);
8862 SourceRange Range = ReadSourceRange(F, Record, Idx);
8863 SourceLocation EllipsisLoc = ReadSourceLocation(F, Record, Idx);
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008864 CXXBaseSpecifier Result(Range, isVirtual, isBaseOfClass, AS, TInfo,
Guy Benyei11169dd2012-12-18 14:30:41 +00008865 EllipsisLoc);
8866 Result.setInheritConstructors(inheritConstructors);
8867 return Result;
8868}
8869
Richard Smithc2bb8182015-03-24 06:36:48 +00008870CXXCtorInitializer **
Guy Benyei11169dd2012-12-18 14:30:41 +00008871ASTReader::ReadCXXCtorInitializers(ModuleFile &F, const RecordData &Record,
8872 unsigned &Idx) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00008873 ASTContext &Context = getContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00008874 unsigned NumInitializers = Record[Idx++];
Richard Smithc2bb8182015-03-24 06:36:48 +00008875 assert(NumInitializers && "wrote ctor initializers but have no inits");
Vedant Kumar48b4f762018-04-14 01:40:48 +00008876 auto **CtorInitializers = new (Context) CXXCtorInitializer*[NumInitializers];
Richard Smithc2bb8182015-03-24 06:36:48 +00008877 for (unsigned i = 0; i != NumInitializers; ++i) {
8878 TypeSourceInfo *TInfo = nullptr;
8879 bool IsBaseVirtual = false;
8880 FieldDecl *Member = nullptr;
8881 IndirectFieldDecl *IndirectMember = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00008882
Vedant Kumar48b4f762018-04-14 01:40:48 +00008883 CtorInitializerType Type = (CtorInitializerType)Record[Idx++];
Richard Smithc2bb8182015-03-24 06:36:48 +00008884 switch (Type) {
8885 case CTOR_INITIALIZER_BASE:
8886 TInfo = GetTypeSourceInfo(F, Record, Idx);
8887 IsBaseVirtual = Record[Idx++];
8888 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00008889
Richard Smithc2bb8182015-03-24 06:36:48 +00008890 case CTOR_INITIALIZER_DELEGATING:
8891 TInfo = GetTypeSourceInfo(F, Record, Idx);
8892 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00008893
Richard Smithc2bb8182015-03-24 06:36:48 +00008894 case CTOR_INITIALIZER_MEMBER:
8895 Member = ReadDeclAs<FieldDecl>(F, Record, Idx);
8896 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00008897
Richard Smithc2bb8182015-03-24 06:36:48 +00008898 case CTOR_INITIALIZER_INDIRECT_MEMBER:
8899 IndirectMember = ReadDeclAs<IndirectFieldDecl>(F, Record, Idx);
8900 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00008901 }
Richard Smithc2bb8182015-03-24 06:36:48 +00008902
8903 SourceLocation MemberOrEllipsisLoc = ReadSourceLocation(F, Record, Idx);
8904 Expr *Init = ReadExpr(F);
8905 SourceLocation LParenLoc = ReadSourceLocation(F, Record, Idx);
8906 SourceLocation RParenLoc = ReadSourceLocation(F, Record, Idx);
Richard Smithc2bb8182015-03-24 06:36:48 +00008907
8908 CXXCtorInitializer *BOMInit;
Richard Smith30e304e2016-12-14 00:03:17 +00008909 if (Type == CTOR_INITIALIZER_BASE)
Richard Smithc2bb8182015-03-24 06:36:48 +00008910 BOMInit = new (Context)
8911 CXXCtorInitializer(Context, TInfo, IsBaseVirtual, LParenLoc, Init,
8912 RParenLoc, MemberOrEllipsisLoc);
Richard Smith30e304e2016-12-14 00:03:17 +00008913 else if (Type == CTOR_INITIALIZER_DELEGATING)
Richard Smithc2bb8182015-03-24 06:36:48 +00008914 BOMInit = new (Context)
8915 CXXCtorInitializer(Context, TInfo, LParenLoc, Init, RParenLoc);
Richard Smith30e304e2016-12-14 00:03:17 +00008916 else if (Member)
8917 BOMInit = new (Context)
8918 CXXCtorInitializer(Context, Member, MemberOrEllipsisLoc, LParenLoc,
8919 Init, RParenLoc);
8920 else
8921 BOMInit = new (Context)
8922 CXXCtorInitializer(Context, IndirectMember, MemberOrEllipsisLoc,
8923 LParenLoc, Init, RParenLoc);
8924
Richard Smith418ed822016-12-14 19:45:03 +00008925 if (/*IsWritten*/Record[Idx++]) {
Richard Smith30e304e2016-12-14 00:03:17 +00008926 unsigned SourceOrder = Record[Idx++];
8927 BOMInit->setSourceOrder(SourceOrder);
Richard Smithc2bb8182015-03-24 06:36:48 +00008928 }
8929
Richard Smithc2bb8182015-03-24 06:36:48 +00008930 CtorInitializers[i] = BOMInit;
Guy Benyei11169dd2012-12-18 14:30:41 +00008931 }
8932
Richard Smithc2bb8182015-03-24 06:36:48 +00008933 return CtorInitializers;
Guy Benyei11169dd2012-12-18 14:30:41 +00008934}
8935
8936NestedNameSpecifier *
8937ASTReader::ReadNestedNameSpecifier(ModuleFile &F,
8938 const RecordData &Record, unsigned &Idx) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00008939 ASTContext &Context = getContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00008940 unsigned N = Record[Idx++];
Craig Toppera13603a2014-05-22 05:54:18 +00008941 NestedNameSpecifier *NNS = nullptr, *Prev = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00008942 for (unsigned I = 0; I != N; ++I) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008943 NestedNameSpecifier::SpecifierKind Kind
8944 = (NestedNameSpecifier::SpecifierKind)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00008945 switch (Kind) {
8946 case NestedNameSpecifier::Identifier: {
8947 IdentifierInfo *II = GetIdentifierInfo(F, Record, Idx);
8948 NNS = NestedNameSpecifier::Create(Context, Prev, II);
8949 break;
8950 }
8951
8952 case NestedNameSpecifier::Namespace: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008953 NamespaceDecl *NS = ReadDeclAs<NamespaceDecl>(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00008954 NNS = NestedNameSpecifier::Create(Context, Prev, NS);
8955 break;
8956 }
8957
8958 case NestedNameSpecifier::NamespaceAlias: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008959 NamespaceAliasDecl *Alias =ReadDeclAs<NamespaceAliasDecl>(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00008960 NNS = NestedNameSpecifier::Create(Context, Prev, Alias);
8961 break;
8962 }
8963
8964 case NestedNameSpecifier::TypeSpec:
8965 case NestedNameSpecifier::TypeSpecWithTemplate: {
8966 const Type *T = readType(F, Record, Idx).getTypePtrOrNull();
8967 if (!T)
Craig Toppera13603a2014-05-22 05:54:18 +00008968 return nullptr;
8969
Guy Benyei11169dd2012-12-18 14:30:41 +00008970 bool Template = Record[Idx++];
8971 NNS = NestedNameSpecifier::Create(Context, Prev, Template, T);
8972 break;
8973 }
8974
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00008975 case NestedNameSpecifier::Global:
Guy Benyei11169dd2012-12-18 14:30:41 +00008976 NNS = NestedNameSpecifier::GlobalSpecifier(Context);
8977 // No associated value, and there can't be a prefix.
8978 break;
Nikola Smiljanic67860242014-09-26 00:28:20 +00008979
8980 case NestedNameSpecifier::Super: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008981 CXXRecordDecl *RD = ReadDeclAs<CXXRecordDecl>(F, Record, Idx);
Nikola Smiljanic67860242014-09-26 00:28:20 +00008982 NNS = NestedNameSpecifier::SuperSpecifier(Context, RD);
8983 break;
8984 }
Guy Benyei11169dd2012-12-18 14:30:41 +00008985 }
8986 Prev = NNS;
8987 }
8988 return NNS;
8989}
8990
8991NestedNameSpecifierLoc
David L. Jonesc4808b9e2016-12-15 20:53:26 +00008992ASTReader::ReadNestedNameSpecifierLoc(ModuleFile &F, const RecordData &Record,
Guy Benyei11169dd2012-12-18 14:30:41 +00008993 unsigned &Idx) {
Richard Smithdbafb6c2017-06-29 23:23:46 +00008994 ASTContext &Context = getContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00008995 unsigned N = Record[Idx++];
8996 NestedNameSpecifierLocBuilder Builder;
8997 for (unsigned I = 0; I != N; ++I) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00008998 NestedNameSpecifier::SpecifierKind Kind
8999 = (NestedNameSpecifier::SpecifierKind)Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00009000 switch (Kind) {
9001 case NestedNameSpecifier::Identifier: {
David L. Jonesc4808b9e2016-12-15 20:53:26 +00009002 IdentifierInfo *II = GetIdentifierInfo(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00009003 SourceRange Range = ReadSourceRange(F, Record, Idx);
9004 Builder.Extend(Context, II, Range.getBegin(), Range.getEnd());
9005 break;
9006 }
9007
9008 case NestedNameSpecifier::Namespace: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00009009 NamespaceDecl *NS = ReadDeclAs<NamespaceDecl>(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00009010 SourceRange Range = ReadSourceRange(F, Record, Idx);
9011 Builder.Extend(Context, NS, Range.getBegin(), Range.getEnd());
9012 break;
9013 }
9014
9015 case NestedNameSpecifier::NamespaceAlias: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00009016 NamespaceAliasDecl *Alias =ReadDeclAs<NamespaceAliasDecl>(F, Record, Idx);
Guy Benyei11169dd2012-12-18 14:30:41 +00009017 SourceRange Range = ReadSourceRange(F, Record, Idx);
9018 Builder.Extend(Context, Alias, Range.getBegin(), Range.getEnd());
9019 break;
9020 }
9021
9022 case NestedNameSpecifier::TypeSpec:
9023 case NestedNameSpecifier::TypeSpecWithTemplate: {
9024 bool Template = Record[Idx++];
9025 TypeSourceInfo *T = GetTypeSourceInfo(F, Record, Idx);
9026 if (!T)
Vedant Kumar48b4f762018-04-14 01:40:48 +00009027 return NestedNameSpecifierLoc();
Guy Benyei11169dd2012-12-18 14:30:41 +00009028 SourceLocation ColonColonLoc = ReadSourceLocation(F, Record, Idx);
9029
9030 // FIXME: 'template' keyword location not saved anywhere, so we fake it.
David L. Jonesc4808b9e2016-12-15 20:53:26 +00009031 Builder.Extend(Context,
Guy Benyei11169dd2012-12-18 14:30:41 +00009032 Template? T->getTypeLoc().getBeginLoc() : SourceLocation(),
9033 T->getTypeLoc(), ColonColonLoc);
9034 break;
9035 }
9036
9037 case NestedNameSpecifier::Global: {
9038 SourceLocation ColonColonLoc = ReadSourceLocation(F, Record, Idx);
9039 Builder.MakeGlobal(Context, ColonColonLoc);
9040 break;
9041 }
Nikola Smiljanic67860242014-09-26 00:28:20 +00009042
9043 case NestedNameSpecifier::Super: {
Vedant Kumar48b4f762018-04-14 01:40:48 +00009044 CXXRecordDecl *RD = ReadDeclAs<CXXRecordDecl>(F, Record, Idx);
Nikola Smiljanic67860242014-09-26 00:28:20 +00009045 SourceRange Range = ReadSourceRange(F, Record, Idx);
9046 Builder.MakeSuper(Context, RD, Range.getBegin(), Range.getEnd());
9047 break;
9048 }
Guy Benyei11169dd2012-12-18 14:30:41 +00009049 }
9050 }
Nikola Smiljanic67860242014-09-26 00:28:20 +00009051
Guy Benyei11169dd2012-12-18 14:30:41 +00009052 return Builder.getWithLocInContext(Context);
9053}
9054
9055SourceRange
9056ASTReader::ReadSourceRange(ModuleFile &F, const RecordData &Record,
9057 unsigned &Idx) {
9058 SourceLocation beg = ReadSourceLocation(F, Record, Idx);
9059 SourceLocation end = ReadSourceLocation(F, Record, Idx);
9060 return SourceRange(beg, end);
9061}
9062
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00009063/// Read an integral value
Guy Benyei11169dd2012-12-18 14:30:41 +00009064llvm::APInt ASTReader::ReadAPInt(const RecordData &Record, unsigned &Idx) {
9065 unsigned BitWidth = Record[Idx++];
9066 unsigned NumWords = llvm::APInt::getNumWords(BitWidth);
9067 llvm::APInt Result(BitWidth, NumWords, &Record[Idx]);
9068 Idx += NumWords;
9069 return Result;
9070}
9071
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00009072/// Read a signed integral value
Guy Benyei11169dd2012-12-18 14:30:41 +00009073llvm::APSInt ASTReader::ReadAPSInt(const RecordData &Record, unsigned &Idx) {
9074 bool isUnsigned = Record[Idx++];
9075 return llvm::APSInt(ReadAPInt(Record, Idx), isUnsigned);
9076}
9077
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00009078/// Read a floating-point value
Tim Northover178723a2013-01-22 09:46:51 +00009079llvm::APFloat ASTReader::ReadAPFloat(const RecordData &Record,
9080 const llvm::fltSemantics &Sem,
9081 unsigned &Idx) {
9082 return llvm::APFloat(Sem, ReadAPInt(Record, Idx));
Guy Benyei11169dd2012-12-18 14:30:41 +00009083}
9084
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00009085// Read a string
Guy Benyei11169dd2012-12-18 14:30:41 +00009086std::string ASTReader::ReadString(const RecordData &Record, unsigned &Idx) {
9087 unsigned Len = Record[Idx++];
9088 std::string Result(Record.data() + Idx, Record.data() + Idx + Len);
9089 Idx += Len;
9090 return Result;
9091}
9092
Richard Smith7ed1bc92014-12-05 22:42:13 +00009093std::string ASTReader::ReadPath(ModuleFile &F, const RecordData &Record,
9094 unsigned &Idx) {
9095 std::string Filename = ReadString(Record, Idx);
9096 ResolveImportedPath(F, Filename);
9097 return Filename;
9098}
9099
David L. Jonesc4808b9e2016-12-15 20:53:26 +00009100VersionTuple ASTReader::ReadVersionTuple(const RecordData &Record,
Guy Benyei11169dd2012-12-18 14:30:41 +00009101 unsigned &Idx) {
9102 unsigned Major = Record[Idx++];
9103 unsigned Minor = Record[Idx++];
9104 unsigned Subminor = Record[Idx++];
9105 if (Minor == 0)
9106 return VersionTuple(Major);
9107 if (Subminor == 0)
9108 return VersionTuple(Major, Minor - 1);
9109 return VersionTuple(Major, Minor - 1, Subminor - 1);
9110}
9111
David L. Jonesc4808b9e2016-12-15 20:53:26 +00009112CXXTemporary *ASTReader::ReadCXXTemporary(ModuleFile &F,
Guy Benyei11169dd2012-12-18 14:30:41 +00009113 const RecordData &Record,
9114 unsigned &Idx) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00009115 CXXDestructorDecl *Decl = ReadDeclAs<CXXDestructorDecl>(F, Record, Idx);
Richard Smithdbafb6c2017-06-29 23:23:46 +00009116 return CXXTemporary::Create(getContext(), Decl);
Guy Benyei11169dd2012-12-18 14:30:41 +00009117}
9118
Richard Smith37a93df2017-02-18 00:32:02 +00009119DiagnosticBuilder ASTReader::Diag(unsigned DiagID) const {
Argyrios Kyrtzidisdc9fdaf2013-05-24 05:44:08 +00009120 return Diag(CurrentImportLoc, DiagID);
Guy Benyei11169dd2012-12-18 14:30:41 +00009121}
9122
Richard Smith37a93df2017-02-18 00:32:02 +00009123DiagnosticBuilder ASTReader::Diag(SourceLocation Loc, unsigned DiagID) const {
Guy Benyei11169dd2012-12-18 14:30:41 +00009124 return Diags.Report(Loc, DiagID);
9125}
9126
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00009127/// Retrieve the identifier table associated with the
Guy Benyei11169dd2012-12-18 14:30:41 +00009128/// preprocessor.
9129IdentifierTable &ASTReader::getIdentifierTable() {
9130 return PP.getIdentifierTable();
9131}
9132
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00009133/// Record that the given ID maps to the given switch-case
Guy Benyei11169dd2012-12-18 14:30:41 +00009134/// statement.
9135void ASTReader::RecordSwitchCaseID(SwitchCase *SC, unsigned ID) {
Craig Toppera13603a2014-05-22 05:54:18 +00009136 assert((*CurrSwitchCaseStmts)[ID] == nullptr &&
Guy Benyei11169dd2012-12-18 14:30:41 +00009137 "Already have a SwitchCase with this ID");
9138 (*CurrSwitchCaseStmts)[ID] = SC;
9139}
9140
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00009141/// Retrieve the switch-case statement with the given ID.
Guy Benyei11169dd2012-12-18 14:30:41 +00009142SwitchCase *ASTReader::getSwitchCaseWithID(unsigned ID) {
Craig Toppera13603a2014-05-22 05:54:18 +00009143 assert((*CurrSwitchCaseStmts)[ID] != nullptr && "No SwitchCase with this ID");
Guy Benyei11169dd2012-12-18 14:30:41 +00009144 return (*CurrSwitchCaseStmts)[ID];
9145}
9146
9147void ASTReader::ClearSwitchCaseIDs() {
9148 CurrSwitchCaseStmts->clear();
9149}
9150
9151void ASTReader::ReadComments() {
Richard Smithdbafb6c2017-06-29 23:23:46 +00009152 ASTContext &Context = getContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00009153 std::vector<RawComment *> Comments;
Vedant Kumar48b4f762018-04-14 01:40:48 +00009154 for (SmallVectorImpl<std::pair<BitstreamCursor,
9155 serialization::ModuleFile *>>::iterator
9156 I = CommentsCursors.begin(),
9157 E = CommentsCursors.end();
9158 I != E; ++I) {
Dmitri Gribenko9ee0e302014-03-27 15:40:39 +00009159 Comments.clear();
Vedant Kumar48b4f762018-04-14 01:40:48 +00009160 BitstreamCursor &Cursor = I->first;
9161 serialization::ModuleFile &F = *I->second;
Guy Benyei11169dd2012-12-18 14:30:41 +00009162 SavedStreamPosition SavedPosition(Cursor);
9163
9164 RecordData Record;
9165 while (true) {
Chris Lattner7fb3bef2013-01-20 00:56:42 +00009166 llvm::BitstreamEntry Entry =
9167 Cursor.advanceSkippingSubblocks(BitstreamCursor::AF_DontPopBlockAtEnd);
Dmitri Gribenko9ee0e302014-03-27 15:40:39 +00009168
Chris Lattner7fb3bef2013-01-20 00:56:42 +00009169 switch (Entry.Kind) {
9170 case llvm::BitstreamEntry::SubBlock: // Handled for us already.
9171 case llvm::BitstreamEntry::Error:
9172 Error("malformed block record in AST file");
9173 return;
9174 case llvm::BitstreamEntry::EndBlock:
9175 goto NextCursor;
9176 case llvm::BitstreamEntry::Record:
9177 // The interesting case.
Guy Benyei11169dd2012-12-18 14:30:41 +00009178 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00009179 }
9180
9181 // Read a record.
9182 Record.clear();
Chris Lattner0e6c9402013-01-20 02:38:54 +00009183 switch ((CommentRecordTypes)Cursor.readRecord(Entry.ID, Record)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00009184 case COMMENTS_RAW_COMMENT: {
9185 unsigned Idx = 0;
9186 SourceRange SR = ReadSourceRange(F, Record, Idx);
Vedant Kumar48b4f762018-04-14 01:40:48 +00009187 RawComment::CommentKind Kind =
9188 (RawComment::CommentKind) Record[Idx++];
Guy Benyei11169dd2012-12-18 14:30:41 +00009189 bool IsTrailingComment = Record[Idx++];
9190 bool IsAlmostTrailingComment = Record[Idx++];
Dmitri Gribenkoa7d16ce2013-04-10 15:35:17 +00009191 Comments.push_back(new (Context) RawComment(
David L. Jones13d5a872018-03-02 00:07:45 +00009192 SR, Kind, IsTrailingComment, IsAlmostTrailingComment));
Guy Benyei11169dd2012-12-18 14:30:41 +00009193 break;
9194 }
9195 }
9196 }
Dmitri Gribenko9ee0e302014-03-27 15:40:39 +00009197 NextCursor:
Bruno Cardoso Lopesc23af572016-12-19 21:06:06 +00009198 // De-serialized SourceLocations get negative FileIDs for other modules,
9199 // potentially invalidating the original order. Sort it again.
Mandeep Singh Grangc205d8c2018-03-27 16:50:00 +00009200 llvm::sort(Comments.begin(), Comments.end(),
9201 BeforeThanCompare<RawComment>(SourceMgr));
Dmitri Gribenko9ee0e302014-03-27 15:40:39 +00009202 Context.Comments.addDeserializedComments(Comments);
Guy Benyei11169dd2012-12-18 14:30:41 +00009203 }
Guy Benyei11169dd2012-12-18 14:30:41 +00009204}
9205
Argyrios Kyrtzidisa38cb202017-01-30 06:05:58 +00009206void ASTReader::visitInputFiles(serialization::ModuleFile &MF,
9207 bool IncludeSystem, bool Complain,
9208 llvm::function_ref<void(const serialization::InputFile &IF,
9209 bool isSystem)> Visitor) {
9210 unsigned NumUserInputs = MF.NumUserInputFiles;
9211 unsigned NumInputs = MF.InputFilesLoaded.size();
9212 assert(NumUserInputs <= NumInputs);
9213 unsigned N = IncludeSystem ? NumInputs : NumUserInputs;
9214 for (unsigned I = 0; I < N; ++I) {
9215 bool IsSystem = I >= NumUserInputs;
9216 InputFile IF = getInputFile(MF, I+1, Complain);
9217 Visitor(IF, IsSystem);
9218 }
9219}
9220
Richard Smithf3f84612017-06-29 02:19:42 +00009221void ASTReader::visitTopLevelModuleMaps(
9222 serialization::ModuleFile &MF,
9223 llvm::function_ref<void(const FileEntry *FE)> Visitor) {
9224 unsigned NumInputs = MF.InputFilesLoaded.size();
9225 for (unsigned I = 0; I < NumInputs; ++I) {
9226 InputFileInfo IFI = readInputFileInfo(MF, I + 1);
9227 if (IFI.TopLevelModuleMap)
9228 // FIXME: This unnecessarily re-reads the InputFileInfo.
9229 if (auto *FE = getInputFile(MF, I + 1).getFile())
9230 Visitor(FE);
9231 }
9232}
9233
Richard Smithcd45dbc2014-04-19 03:48:30 +00009234std::string ASTReader::getOwningModuleNameForDiagnostic(const Decl *D) {
9235 // If we know the owning module, use it.
Richard Smith42413142015-05-15 20:05:43 +00009236 if (Module *M = D->getImportedOwningModule())
Richard Smithcd45dbc2014-04-19 03:48:30 +00009237 return M->getFullModuleName();
9238
9239 // Otherwise, use the name of the top-level module the decl is within.
9240 if (ModuleFile *M = getOwningModuleFile(D))
9241 return M->ModuleName;
9242
9243 // Not from a module.
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00009244 return {};
Richard Smithcd45dbc2014-04-19 03:48:30 +00009245}
9246
Guy Benyei11169dd2012-12-18 14:30:41 +00009247void ASTReader::finishPendingActions() {
Richard Smitha62d1982018-08-03 01:00:01 +00009248 while (!PendingIdentifierInfos.empty() || !PendingFunctionTypes.empty() ||
Richard Smith851072e2014-05-19 20:59:20 +00009249 !PendingIncompleteDeclChains.empty() || !PendingDeclChains.empty() ||
Richard Smith2b9e3e32013-10-18 06:05:18 +00009250 !PendingMacroIDs.empty() || !PendingDeclContextInfos.empty() ||
Richard Smitha0ce9c42014-07-29 23:23:27 +00009251 !PendingUpdateRecords.empty()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00009252 // If any identifiers with corresponding top-level declarations have
9253 // been loaded, load those declarations now.
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +00009254 using TopLevelDeclsMap =
9255 llvm::DenseMap<IdentifierInfo *, SmallVector<Decl *, 2>>;
Craig Topper79be4cd2013-07-05 04:33:53 +00009256 TopLevelDeclsMap TopLevelDecls;
9257
Guy Benyei11169dd2012-12-18 14:30:41 +00009258 while (!PendingIdentifierInfos.empty()) {
Douglas Gregor6168bd22013-02-18 15:53:43 +00009259 IdentifierInfo *II = PendingIdentifierInfos.back().first;
Richard Smithf0ae3c2d2014-03-28 17:31:23 +00009260 SmallVector<uint32_t, 4> DeclIDs =
9261 std::move(PendingIdentifierInfos.back().second);
Douglas Gregorcb15f082013-02-19 18:26:28 +00009262 PendingIdentifierInfos.pop_back();
Douglas Gregor6168bd22013-02-18 15:53:43 +00009263
9264 SetGloballyVisibleDecls(II, DeclIDs, &TopLevelDecls[II]);
Guy Benyei11169dd2012-12-18 14:30:41 +00009265 }
Richard Smithf0ae3c2d2014-03-28 17:31:23 +00009266
Richard Smitha62d1982018-08-03 01:00:01 +00009267 // Load each function type that we deferred loading because it was a
9268 // deduced type that might refer to a local type declared within itself.
9269 for (unsigned I = 0; I != PendingFunctionTypes.size(); ++I) {
9270 auto *FD = PendingFunctionTypes[I].first;
9271 FD->setType(GetType(PendingFunctionTypes[I].second));
9272
9273 // If we gave a function a deduced return type, remember that we need to
9274 // propagate that along the redeclaration chain.
9275 auto *DT = FD->getReturnType()->getContainedDeducedType();
9276 if (DT && DT->isDeduced())
9277 PendingDeducedTypeUpdates.insert(
9278 {FD->getCanonicalDecl(), FD->getReturnType()});
9279 }
9280 PendingFunctionTypes.clear();
9281
Richard Smith851072e2014-05-19 20:59:20 +00009282 // For each decl chain that we wanted to complete while deserializing, mark
9283 // it as "still needs to be completed".
Vedant Kumar48b4f762018-04-14 01:40:48 +00009284 for (unsigned I = 0; I != PendingIncompleteDeclChains.size(); ++I) {
9285 markIncompleteDeclChain(PendingIncompleteDeclChains[I]);
9286 }
Richard Smith851072e2014-05-19 20:59:20 +00009287 PendingIncompleteDeclChains.clear();
9288
Guy Benyei11169dd2012-12-18 14:30:41 +00009289 // Load pending declaration chains.
Vedant Kumar48b4f762018-04-14 01:40:48 +00009290 for (unsigned I = 0; I != PendingDeclChains.size(); ++I)
Richard Smitha62d1982018-08-03 01:00:01 +00009291 loadPendingDeclChain(PendingDeclChains[I].first,
9292 PendingDeclChains[I].second);
Guy Benyei11169dd2012-12-18 14:30:41 +00009293 PendingDeclChains.clear();
9294
Douglas Gregor6168bd22013-02-18 15:53:43 +00009295 // Make the most recent of the top-level declarations visible.
Vedant Kumar48b4f762018-04-14 01:40:48 +00009296 for (TopLevelDeclsMap::iterator TLD = TopLevelDecls.begin(),
9297 TLDEnd = TopLevelDecls.end(); TLD != TLDEnd; ++TLD) {
9298 IdentifierInfo *II = TLD->first;
9299 for (unsigned I = 0, N = TLD->second.size(); I != N; ++I) {
9300 pushExternalDeclIntoScope(cast<NamedDecl>(TLD->second[I]), II);
Douglas Gregor6168bd22013-02-18 15:53:43 +00009301 }
9302 }
9303
Guy Benyei11169dd2012-12-18 14:30:41 +00009304 // Load any pending macro definitions.
9305 for (unsigned I = 0; I != PendingMacroIDs.size(); ++I) {
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00009306 IdentifierInfo *II = PendingMacroIDs.begin()[I].first;
9307 SmallVector<PendingMacroInfo, 2> GlobalIDs;
9308 GlobalIDs.swap(PendingMacroIDs.begin()[I].second);
9309 // Initialize the macro history from chained-PCHs ahead of module imports.
Richard Smith49f906a2014-03-01 00:08:04 +00009310 for (unsigned IDIdx = 0, NumIDs = GlobalIDs.size(); IDIdx != NumIDs;
Argyrios Kyrtzidis719736c2013-01-19 03:14:56 +00009311 ++IDIdx) {
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00009312 const PendingMacroInfo &Info = GlobalIDs[IDIdx];
Manman Ren11f2a472016-08-18 17:42:15 +00009313 if (!Info.M->isModule())
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00009314 resolvePendingMacro(II, Info);
9315 }
9316 // Handle module imports.
Richard Smith49f906a2014-03-01 00:08:04 +00009317 for (unsigned IDIdx = 0, NumIDs = GlobalIDs.size(); IDIdx != NumIDs;
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00009318 ++IDIdx) {
9319 const PendingMacroInfo &Info = GlobalIDs[IDIdx];
Manman Ren11f2a472016-08-18 17:42:15 +00009320 if (Info.M->isModule())
Argyrios Kyrtzidiseb663da2013-03-22 21:12:57 +00009321 resolvePendingMacro(II, Info);
Guy Benyei11169dd2012-12-18 14:30:41 +00009322 }
9323 }
9324 PendingMacroIDs.clear();
Argyrios Kyrtzidis83a6e3b2013-02-16 00:48:59 +00009325
9326 // Wire up the DeclContexts for Decls that we delayed setting until
9327 // recursive loading is completed.
9328 while (!PendingDeclContextInfos.empty()) {
9329 PendingDeclContextInfo Info = PendingDeclContextInfos.front();
9330 PendingDeclContextInfos.pop_front();
Vedant Kumar48b4f762018-04-14 01:40:48 +00009331 DeclContext *SemaDC = cast<DeclContext>(GetDecl(Info.SemaDC));
9332 DeclContext *LexicalDC = cast<DeclContext>(GetDecl(Info.LexicalDC));
Argyrios Kyrtzidis83a6e3b2013-02-16 00:48:59 +00009333 Info.D->setDeclContextsImpl(SemaDC, LexicalDC, getContext());
9334 }
Richard Smith2b9e3e32013-10-18 06:05:18 +00009335
Richard Smithd1c46742014-04-30 02:24:17 +00009336 // Perform any pending declaration updates.
Richard Smithd6db68c2014-08-07 20:58:41 +00009337 while (!PendingUpdateRecords.empty()) {
Richard Smithd1c46742014-04-30 02:24:17 +00009338 auto Update = PendingUpdateRecords.pop_back_val();
9339 ReadingKindTracker ReadingKind(Read_Decl, *this);
Vassil Vassilev74c3e8c2017-05-19 16:46:06 +00009340 loadDeclUpdateRecords(Update);
Richard Smithd1c46742014-04-30 02:24:17 +00009341 }
Guy Benyei11169dd2012-12-18 14:30:41 +00009342 }
Richard Smith8a639892015-01-24 01:07:20 +00009343
9344 // At this point, all update records for loaded decls are in place, so any
9345 // fake class definitions should have become real.
9346 assert(PendingFakeDefinitionData.empty() &&
9347 "faked up a class definition but never saw the real one");
9348
Guy Benyei11169dd2012-12-18 14:30:41 +00009349 // If we deserialized any C++ or Objective-C class definitions, any
9350 // Objective-C protocol definitions, or any redeclarable templates, make sure
David L. Jonesc4808b9e2016-12-15 20:53:26 +00009351 // that all redeclarations point to the definitions. Note that this can only
Guy Benyei11169dd2012-12-18 14:30:41 +00009352 // happen now, after the redeclaration chains have been fully wired.
Vedant Kumar48b4f762018-04-14 01:40:48 +00009353 for (Decl *D : PendingDefinitions) {
9354 if (TagDecl *TD = dyn_cast<TagDecl>(D)) {
9355 if (const TagType *TagT = dyn_cast<TagType>(TD->getTypeForDecl())) {
Guy Benyei11169dd2012-12-18 14:30:41 +00009356 // Make sure that the TagType points at the definition.
9357 const_cast<TagType*>(TagT)->decl = TD;
9358 }
Richard Smith8ce51082015-03-11 01:44:51 +00009359
Vedant Kumar48b4f762018-04-14 01:40:48 +00009360 if (auto RD = dyn_cast<CXXRecordDecl>(D)) {
Richard Smith8ce51082015-03-11 01:44:51 +00009361 for (auto *R = getMostRecentExistingDecl(RD); R;
9362 R = R->getPreviousDecl()) {
9363 assert((R == D) ==
9364 cast<CXXRecordDecl>(R)->isThisDeclarationADefinition() &&
Richard Smith2c381642014-08-27 23:11:59 +00009365 "declaration thinks it's the definition but it isn't");
Aaron Ballman86c93902014-03-06 23:45:36 +00009366 cast<CXXRecordDecl>(R)->DefinitionData = RD->DefinitionData;
Richard Smith2c381642014-08-27 23:11:59 +00009367 }
Guy Benyei11169dd2012-12-18 14:30:41 +00009368 }
9369
9370 continue;
9371 }
Richard Smith8ce51082015-03-11 01:44:51 +00009372
Vedant Kumar48b4f762018-04-14 01:40:48 +00009373 if (auto ID = dyn_cast<ObjCInterfaceDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00009374 // Make sure that the ObjCInterfaceType points at the definition.
9375 const_cast<ObjCInterfaceType *>(cast<ObjCInterfaceType>(ID->TypeForDecl))
9376 ->Decl = ID;
Richard Smith8ce51082015-03-11 01:44:51 +00009377
9378 for (auto *R = getMostRecentExistingDecl(ID); R; R = R->getPreviousDecl())
9379 cast<ObjCInterfaceDecl>(R)->Data = ID->Data;
9380
Guy Benyei11169dd2012-12-18 14:30:41 +00009381 continue;
9382 }
Richard Smith8ce51082015-03-11 01:44:51 +00009383
Vedant Kumar48b4f762018-04-14 01:40:48 +00009384 if (auto PD = dyn_cast<ObjCProtocolDecl>(D)) {
Richard Smith8ce51082015-03-11 01:44:51 +00009385 for (auto *R = getMostRecentExistingDecl(PD); R; R = R->getPreviousDecl())
9386 cast<ObjCProtocolDecl>(R)->Data = PD->Data;
9387
Guy Benyei11169dd2012-12-18 14:30:41 +00009388 continue;
9389 }
Richard Smith8ce51082015-03-11 01:44:51 +00009390
Vedant Kumar48b4f762018-04-14 01:40:48 +00009391 auto RTD = cast<RedeclarableTemplateDecl>(D)->getCanonicalDecl();
Richard Smith8ce51082015-03-11 01:44:51 +00009392 for (auto *R = getMostRecentExistingDecl(RTD); R; R = R->getPreviousDecl())
9393 cast<RedeclarableTemplateDecl>(R)->Common = RTD->Common;
Guy Benyei11169dd2012-12-18 14:30:41 +00009394 }
9395 PendingDefinitions.clear();
9396
Daniel Jasper4a6d5b72017-10-11 07:47:54 +00009397 // Load the bodies of any functions or methods we've encountered. We do
9398 // this now (delayed) so that we can be sure that the declaration chains
9399 // have been fully wired up (hasBody relies on this).
9400 // FIXME: We shouldn't require complete redeclaration chains here.
Vedant Kumar48b4f762018-04-14 01:40:48 +00009401 for (PendingBodiesMap::iterator PB = PendingBodies.begin(),
9402 PBEnd = PendingBodies.end();
9403 PB != PBEnd; ++PB) {
9404 if (FunctionDecl *FD = dyn_cast<FunctionDecl>(PB->first)) {
Richard Smith0b70de12018-06-28 01:07:28 +00009405 // For a function defined inline within a class template, force the
9406 // canonical definition to be the one inside the canonical definition of
9407 // the template. This ensures that we instantiate from a correct view
9408 // of the template.
9409 //
9410 // Sadly we can't do this more generally: we can't be sure that all
9411 // copies of an arbitrary class definition will have the same members
9412 // defined (eg, some member functions may not be instantiated, and some
9413 // special members may or may not have been implicitly defined).
9414 if (auto *RD = dyn_cast<CXXRecordDecl>(FD->getLexicalParent()))
9415 if (RD->isDependentContext() && !RD->isThisDeclarationADefinition())
9416 continue;
9417
Daniel Jasper4a6d5b72017-10-11 07:47:54 +00009418 // FIXME: Check for =delete/=default?
9419 // FIXME: Complain about ODR violations here?
9420 const FunctionDecl *Defn = nullptr;
9421 if (!getContext().getLangOpts().Modules || !FD->hasBody(Defn)) {
Vedant Kumar48b4f762018-04-14 01:40:48 +00009422 FD->setLazyBody(PB->second);
Richard Trieue6caa262017-12-23 00:41:01 +00009423 } else {
Vedant Kumar48b4f762018-04-14 01:40:48 +00009424 auto *NonConstDefn = const_cast<FunctionDecl*>(Defn);
Richard Trieue6caa262017-12-23 00:41:01 +00009425 mergeDefinitionVisibility(NonConstDefn, FD);
9426
9427 if (!FD->isLateTemplateParsed() &&
9428 !NonConstDefn->isLateTemplateParsed() &&
9429 FD->getODRHash() != NonConstDefn->getODRHash()) {
Richard Trieu27c1b1a2018-07-10 01:40:50 +00009430 if (!isa<CXXMethodDecl>(FD)) {
9431 PendingFunctionOdrMergeFailures[FD].push_back(NonConstDefn);
9432 } else if (FD->getLexicalParent()->isFileContext() &&
9433 NonConstDefn->getLexicalParent()->isFileContext()) {
9434 // Only diagnose out-of-line method definitions. If they are
9435 // in class definitions, then an error will be generated when
9436 // processing the class bodies.
9437 PendingFunctionOdrMergeFailures[FD].push_back(NonConstDefn);
9438 }
Richard Trieue6caa262017-12-23 00:41:01 +00009439 }
9440 }
Daniel Jasper4a6d5b72017-10-11 07:47:54 +00009441 continue;
9442 }
9443
Vedant Kumar48b4f762018-04-14 01:40:48 +00009444 ObjCMethodDecl *MD = cast<ObjCMethodDecl>(PB->first);
Daniel Jasper4a6d5b72017-10-11 07:47:54 +00009445 if (!getContext().getLangOpts().Modules || !MD->hasBody())
Vedant Kumar48b4f762018-04-14 01:40:48 +00009446 MD->setLazyBody(PB->second);
Daniel Jasper4a6d5b72017-10-11 07:47:54 +00009447 }
9448 PendingBodies.clear();
9449
Richard Smith42413142015-05-15 20:05:43 +00009450 // Do some cleanup.
9451 for (auto *ND : PendingMergedDefinitionsToDeduplicate)
9452 getContext().deduplicateMergedDefinitonsFor(ND);
9453 PendingMergedDefinitionsToDeduplicate.clear();
Richard Smitha0ce9c42014-07-29 23:23:27 +00009454}
9455
9456void ASTReader::diagnoseOdrViolations() {
Richard Trieue6caa262017-12-23 00:41:01 +00009457 if (PendingOdrMergeFailures.empty() && PendingOdrMergeChecks.empty() &&
Richard Trieuab4d7302018-07-25 22:52:05 +00009458 PendingFunctionOdrMergeFailures.empty() &&
9459 PendingEnumOdrMergeFailures.empty())
Richard Smithbb853c72014-08-13 01:23:33 +00009460 return;
9461
Richard Smitha0ce9c42014-07-29 23:23:27 +00009462 // Trigger the import of the full definition of each class that had any
9463 // odr-merging problems, so we can produce better diagnostics for them.
Richard Smithbb853c72014-08-13 01:23:33 +00009464 // These updates may in turn find and diagnose some ODR failures, so take
9465 // ownership of the set first.
9466 auto OdrMergeFailures = std::move(PendingOdrMergeFailures);
9467 PendingOdrMergeFailures.clear();
9468 for (auto &Merge : OdrMergeFailures) {
Richard Smitha0ce9c42014-07-29 23:23:27 +00009469 Merge.first->buildLookup();
9470 Merge.first->decls_begin();
9471 Merge.first->bases_begin();
9472 Merge.first->vbases_begin();
Richard Trieue13eabe2017-09-30 02:19:17 +00009473 for (auto &RecordPair : Merge.second) {
9474 auto *RD = RecordPair.first;
Richard Smitha0ce9c42014-07-29 23:23:27 +00009475 RD->decls_begin();
9476 RD->bases_begin();
9477 RD->vbases_begin();
9478 }
9479 }
9480
Richard Trieue6caa262017-12-23 00:41:01 +00009481 // Trigger the import of functions.
9482 auto FunctionOdrMergeFailures = std::move(PendingFunctionOdrMergeFailures);
9483 PendingFunctionOdrMergeFailures.clear();
9484 for (auto &Merge : FunctionOdrMergeFailures) {
9485 Merge.first->buildLookup();
9486 Merge.first->decls_begin();
9487 Merge.first->getBody();
9488 for (auto &FD : Merge.second) {
9489 FD->buildLookup();
9490 FD->decls_begin();
9491 FD->getBody();
9492 }
9493 }
9494
Richard Trieuab4d7302018-07-25 22:52:05 +00009495 // Trigger the import of enums.
9496 auto EnumOdrMergeFailures = std::move(PendingEnumOdrMergeFailures);
9497 PendingEnumOdrMergeFailures.clear();
9498 for (auto &Merge : EnumOdrMergeFailures) {
9499 Merge.first->decls_begin();
9500 for (auto &Enum : Merge.second) {
9501 Enum->decls_begin();
9502 }
9503 }
9504
Richard Smitha0ce9c42014-07-29 23:23:27 +00009505 // For each declaration from a merged context, check that the canonical
9506 // definition of that context also contains a declaration of the same
9507 // entity.
9508 //
9509 // Caution: this loop does things that might invalidate iterators into
9510 // PendingOdrMergeChecks. Don't turn this into a range-based for loop!
9511 while (!PendingOdrMergeChecks.empty()) {
9512 NamedDecl *D = PendingOdrMergeChecks.pop_back_val();
9513
9514 // FIXME: Skip over implicit declarations for now. This matters for things
9515 // like implicitly-declared special member functions. This isn't entirely
9516 // correct; we can end up with multiple unmerged declarations of the same
9517 // implicit entity.
9518 if (D->isImplicit())
9519 continue;
9520
9521 DeclContext *CanonDef = D->getDeclContext();
Richard Smitha0ce9c42014-07-29 23:23:27 +00009522
9523 bool Found = false;
9524 const Decl *DCanon = D->getCanonicalDecl();
9525
Richard Smith01bdb7a2014-08-28 05:44:07 +00009526 for (auto RI : D->redecls()) {
9527 if (RI->getLexicalDeclContext() == CanonDef) {
9528 Found = true;
9529 break;
9530 }
9531 }
9532 if (Found)
9533 continue;
9534
Richard Smith0f4e2c42015-08-06 04:23:48 +00009535 // Quick check failed, time to do the slow thing. Note, we can't just
9536 // look up the name of D in CanonDef here, because the member that is
9537 // in CanonDef might not be found by name lookup (it might have been
9538 // replaced by a more recent declaration in the lookup table), and we
9539 // can't necessarily find it in the redeclaration chain because it might
9540 // be merely mergeable, not redeclarable.
Richard Smitha0ce9c42014-07-29 23:23:27 +00009541 llvm::SmallVector<const NamedDecl*, 4> Candidates;
Richard Smith0f4e2c42015-08-06 04:23:48 +00009542 for (auto *CanonMember : CanonDef->decls()) {
9543 if (CanonMember->getCanonicalDecl() == DCanon) {
9544 // This can happen if the declaration is merely mergeable and not
9545 // actually redeclarable (we looked for redeclarations earlier).
9546 //
9547 // FIXME: We should be able to detect this more efficiently, without
9548 // pulling in all of the members of CanonDef.
9549 Found = true;
9550 break;
Richard Smitha0ce9c42014-07-29 23:23:27 +00009551 }
Richard Smith0f4e2c42015-08-06 04:23:48 +00009552 if (auto *ND = dyn_cast<NamedDecl>(CanonMember))
9553 if (ND->getDeclName() == D->getDeclName())
9554 Candidates.push_back(ND);
Richard Smitha0ce9c42014-07-29 23:23:27 +00009555 }
9556
9557 if (!Found) {
Richard Smithd08aeb62014-08-28 01:33:39 +00009558 // The AST doesn't like TagDecls becoming invalid after they've been
9559 // completed. We only really need to mark FieldDecls as invalid here.
9560 if (!isa<TagDecl>(D))
9561 D->setInvalidDecl();
David L. Jonesc4808b9e2016-12-15 20:53:26 +00009562
Richard Smith4ab3dbd2015-02-13 22:43:51 +00009563 // Ensure we don't accidentally recursively enter deserialization while
9564 // we're producing our diagnostic.
9565 Deserializing RecursionGuard(this);
Richard Smitha0ce9c42014-07-29 23:23:27 +00009566
9567 std::string CanonDefModule =
9568 getOwningModuleNameForDiagnostic(cast<Decl>(CanonDef));
9569 Diag(D->getLocation(), diag::err_module_odr_violation_missing_decl)
9570 << D << getOwningModuleNameForDiagnostic(D)
9571 << CanonDef << CanonDefModule.empty() << CanonDefModule;
9572
9573 if (Candidates.empty())
9574 Diag(cast<Decl>(CanonDef)->getLocation(),
9575 diag::note_module_odr_violation_no_possible_decls) << D;
9576 else {
Vedant Kumar48b4f762018-04-14 01:40:48 +00009577 for (unsigned I = 0, N = Candidates.size(); I != N; ++I)
9578 Diag(Candidates[I]->getLocation(),
Richard Smitha0ce9c42014-07-29 23:23:27 +00009579 diag::note_module_odr_violation_possible_decl)
Vedant Kumar48b4f762018-04-14 01:40:48 +00009580 << Candidates[I];
Richard Smitha0ce9c42014-07-29 23:23:27 +00009581 }
9582
9583 DiagnosedOdrMergeFailures.insert(CanonDef);
9584 }
9585 }
Richard Smithcd45dbc2014-04-19 03:48:30 +00009586
Richard Trieuab4d7302018-07-25 22:52:05 +00009587 if (OdrMergeFailures.empty() && FunctionOdrMergeFailures.empty() &&
9588 EnumOdrMergeFailures.empty())
Richard Smith4ab3dbd2015-02-13 22:43:51 +00009589 return;
9590
9591 // Ensure we don't accidentally recursively enter deserialization while
9592 // we're producing our diagnostics.
9593 Deserializing RecursionGuard(this);
9594
Richard Trieue6caa262017-12-23 00:41:01 +00009595 // Common code for hashing helpers.
9596 ODRHash Hash;
9597 auto ComputeQualTypeODRHash = [&Hash](QualType Ty) {
9598 Hash.clear();
9599 Hash.AddQualType(Ty);
9600 return Hash.CalculateHash();
9601 };
9602
9603 auto ComputeODRHash = [&Hash](const Stmt *S) {
9604 assert(S);
9605 Hash.clear();
9606 Hash.AddStmt(S);
9607 return Hash.CalculateHash();
9608 };
9609
9610 auto ComputeSubDeclODRHash = [&Hash](const Decl *D) {
9611 assert(D);
9612 Hash.clear();
9613 Hash.AddSubDecl(D);
9614 return Hash.CalculateHash();
9615 };
9616
Richard Trieu7282d322018-04-25 00:31:15 +00009617 auto ComputeTemplateArgumentODRHash = [&Hash](const TemplateArgument &TA) {
9618 Hash.clear();
9619 Hash.AddTemplateArgument(TA);
9620 return Hash.CalculateHash();
9621 };
9622
Richard Trieu9359e8f2018-05-30 01:12:26 +00009623 auto ComputeTemplateParameterListODRHash =
9624 [&Hash](const TemplateParameterList *TPL) {
9625 assert(TPL);
9626 Hash.clear();
9627 Hash.AddTemplateParameterList(TPL);
9628 return Hash.CalculateHash();
9629 };
9630
Richard Smithcd45dbc2014-04-19 03:48:30 +00009631 // Issue any pending ODR-failure diagnostics.
Richard Smithbb853c72014-08-13 01:23:33 +00009632 for (auto &Merge : OdrMergeFailures) {
Richard Smitha0ce9c42014-07-29 23:23:27 +00009633 // If we've already pointed out a specific problem with this class, don't
9634 // bother issuing a general "something's different" diagnostic.
David Blaikie82e95a32014-11-19 07:49:47 +00009635 if (!DiagnosedOdrMergeFailures.insert(Merge.first).second)
Richard Smithcd45dbc2014-04-19 03:48:30 +00009636 continue;
9637
9638 bool Diagnosed = false;
Richard Trieue7f7ed22017-02-22 01:11:25 +00009639 CXXRecordDecl *FirstRecord = Merge.first;
9640 std::string FirstModule = getOwningModuleNameForDiagnostic(FirstRecord);
Richard Trieue13eabe2017-09-30 02:19:17 +00009641 for (auto &RecordPair : Merge.second) {
9642 CXXRecordDecl *SecondRecord = RecordPair.first;
Richard Smithcd45dbc2014-04-19 03:48:30 +00009643 // Multiple different declarations got merged together; tell the user
9644 // where they came from.
Richard Trieue7f7ed22017-02-22 01:11:25 +00009645 if (FirstRecord == SecondRecord)
9646 continue;
9647
9648 std::string SecondModule = getOwningModuleNameForDiagnostic(SecondRecord);
Richard Trieue13eabe2017-09-30 02:19:17 +00009649
9650 auto *FirstDD = FirstRecord->DefinitionData;
9651 auto *SecondDD = RecordPair.second;
9652
9653 assert(FirstDD && SecondDD && "Definitions without DefinitionData");
9654
9655 // Diagnostics from DefinitionData are emitted here.
9656 if (FirstDD != SecondDD) {
9657 enum ODRDefinitionDataDifference {
9658 NumBases,
9659 NumVBases,
9660 BaseType,
9661 BaseVirtual,
9662 BaseAccess,
9663 };
9664 auto ODRDiagError = [FirstRecord, &FirstModule,
9665 this](SourceLocation Loc, SourceRange Range,
9666 ODRDefinitionDataDifference DiffType) {
9667 return Diag(Loc, diag::err_module_odr_violation_definition_data)
9668 << FirstRecord << FirstModule.empty() << FirstModule << Range
9669 << DiffType;
9670 };
9671 auto ODRDiagNote = [&SecondModule,
9672 this](SourceLocation Loc, SourceRange Range,
9673 ODRDefinitionDataDifference DiffType) {
9674 return Diag(Loc, diag::note_module_odr_violation_definition_data)
9675 << SecondModule << Range << DiffType;
9676 };
9677
Richard Trieue13eabe2017-09-30 02:19:17 +00009678 unsigned FirstNumBases = FirstDD->NumBases;
9679 unsigned FirstNumVBases = FirstDD->NumVBases;
9680 unsigned SecondNumBases = SecondDD->NumBases;
9681 unsigned SecondNumVBases = SecondDD->NumVBases;
9682
9683 auto GetSourceRange = [](struct CXXRecordDecl::DefinitionData *DD) {
9684 unsigned NumBases = DD->NumBases;
9685 if (NumBases == 0) return SourceRange();
9686 auto bases = DD->bases();
Stephen Kellyf2ceec42018-08-09 21:08:08 +00009687 return SourceRange(bases[0].getBeginLoc(),
Stephen Kelly1c301dc2018-08-09 21:09:38 +00009688 bases[NumBases - 1].getEndLoc());
Richard Trieue13eabe2017-09-30 02:19:17 +00009689 };
9690
9691 if (FirstNumBases != SecondNumBases) {
9692 ODRDiagError(FirstRecord->getLocation(), GetSourceRange(FirstDD),
9693 NumBases)
9694 << FirstNumBases;
9695 ODRDiagNote(SecondRecord->getLocation(), GetSourceRange(SecondDD),
9696 NumBases)
9697 << SecondNumBases;
9698 Diagnosed = true;
9699 break;
9700 }
9701
9702 if (FirstNumVBases != SecondNumVBases) {
9703 ODRDiagError(FirstRecord->getLocation(), GetSourceRange(FirstDD),
9704 NumVBases)
9705 << FirstNumVBases;
9706 ODRDiagNote(SecondRecord->getLocation(), GetSourceRange(SecondDD),
9707 NumVBases)
9708 << SecondNumVBases;
9709 Diagnosed = true;
9710 break;
9711 }
9712
9713 auto FirstBases = FirstDD->bases();
9714 auto SecondBases = SecondDD->bases();
9715 unsigned i = 0;
9716 for (i = 0; i < FirstNumBases; ++i) {
9717 auto FirstBase = FirstBases[i];
9718 auto SecondBase = SecondBases[i];
9719 if (ComputeQualTypeODRHash(FirstBase.getType()) !=
9720 ComputeQualTypeODRHash(SecondBase.getType())) {
9721 ODRDiagError(FirstRecord->getLocation(), FirstBase.getSourceRange(),
9722 BaseType)
9723 << (i + 1) << FirstBase.getType();
9724 ODRDiagNote(SecondRecord->getLocation(),
9725 SecondBase.getSourceRange(), BaseType)
9726 << (i + 1) << SecondBase.getType();
9727 break;
9728 }
9729
9730 if (FirstBase.isVirtual() != SecondBase.isVirtual()) {
9731 ODRDiagError(FirstRecord->getLocation(), FirstBase.getSourceRange(),
9732 BaseVirtual)
9733 << (i + 1) << FirstBase.isVirtual() << FirstBase.getType();
9734 ODRDiagNote(SecondRecord->getLocation(),
9735 SecondBase.getSourceRange(), BaseVirtual)
9736 << (i + 1) << SecondBase.isVirtual() << SecondBase.getType();
9737 break;
9738 }
9739
9740 if (FirstBase.getAccessSpecifierAsWritten() !=
9741 SecondBase.getAccessSpecifierAsWritten()) {
9742 ODRDiagError(FirstRecord->getLocation(), FirstBase.getSourceRange(),
9743 BaseAccess)
9744 << (i + 1) << FirstBase.getType()
9745 << (int)FirstBase.getAccessSpecifierAsWritten();
9746 ODRDiagNote(SecondRecord->getLocation(),
9747 SecondBase.getSourceRange(), BaseAccess)
9748 << (i + 1) << SecondBase.getType()
9749 << (int)SecondBase.getAccessSpecifierAsWritten();
9750 break;
9751 }
9752 }
9753
9754 if (i != FirstNumBases) {
9755 Diagnosed = true;
9756 break;
9757 }
9758 }
9759
Richard Trieue7f7ed22017-02-22 01:11:25 +00009760 using DeclHashes = llvm::SmallVector<std::pair<Decl *, unsigned>, 4>;
Richard Trieu498117b2017-08-23 02:43:59 +00009761
9762 const ClassTemplateDecl *FirstTemplate =
9763 FirstRecord->getDescribedClassTemplate();
9764 const ClassTemplateDecl *SecondTemplate =
9765 SecondRecord->getDescribedClassTemplate();
9766
9767 assert(!FirstTemplate == !SecondTemplate &&
9768 "Both pointers should be null or non-null");
9769
9770 enum ODRTemplateDifference {
9771 ParamEmptyName,
9772 ParamName,
9773 ParamSingleDefaultArgument,
9774 ParamDifferentDefaultArgument,
9775 };
9776
9777 if (FirstTemplate && SecondTemplate) {
9778 DeclHashes FirstTemplateHashes;
9779 DeclHashes SecondTemplateHashes;
Richard Trieu498117b2017-08-23 02:43:59 +00009780
9781 auto PopulateTemplateParameterHashs =
Richard Trieue6caa262017-12-23 00:41:01 +00009782 [&ComputeSubDeclODRHash](DeclHashes &Hashes,
9783 const ClassTemplateDecl *TD) {
Richard Trieu498117b2017-08-23 02:43:59 +00009784 for (auto *D : TD->getTemplateParameters()->asArray()) {
Richard Trieue6caa262017-12-23 00:41:01 +00009785 Hashes.emplace_back(D, ComputeSubDeclODRHash(D));
Richard Trieu498117b2017-08-23 02:43:59 +00009786 }
9787 };
9788
9789 PopulateTemplateParameterHashs(FirstTemplateHashes, FirstTemplate);
9790 PopulateTemplateParameterHashs(SecondTemplateHashes, SecondTemplate);
9791
9792 assert(FirstTemplateHashes.size() == SecondTemplateHashes.size() &&
9793 "Number of template parameters should be equal.");
9794
9795 auto FirstIt = FirstTemplateHashes.begin();
9796 auto FirstEnd = FirstTemplateHashes.end();
9797 auto SecondIt = SecondTemplateHashes.begin();
9798 for (; FirstIt != FirstEnd; ++FirstIt, ++SecondIt) {
9799 if (FirstIt->second == SecondIt->second)
9800 continue;
9801
9802 auto ODRDiagError = [FirstRecord, &FirstModule,
9803 this](SourceLocation Loc, SourceRange Range,
9804 ODRTemplateDifference DiffType) {
9805 return Diag(Loc, diag::err_module_odr_violation_template_parameter)
9806 << FirstRecord << FirstModule.empty() << FirstModule << Range
9807 << DiffType;
9808 };
9809 auto ODRDiagNote = [&SecondModule,
9810 this](SourceLocation Loc, SourceRange Range,
9811 ODRTemplateDifference DiffType) {
9812 return Diag(Loc, diag::note_module_odr_violation_template_parameter)
9813 << SecondModule << Range << DiffType;
9814 };
9815
Vedant Kumar48b4f762018-04-14 01:40:48 +00009816 const NamedDecl* FirstDecl = cast<NamedDecl>(FirstIt->first);
9817 const NamedDecl* SecondDecl = cast<NamedDecl>(SecondIt->first);
Richard Trieu498117b2017-08-23 02:43:59 +00009818
9819 assert(FirstDecl->getKind() == SecondDecl->getKind() &&
9820 "Parameter Decl's should be the same kind.");
9821
9822 DeclarationName FirstName = FirstDecl->getDeclName();
9823 DeclarationName SecondName = SecondDecl->getDeclName();
9824
9825 if (FirstName != SecondName) {
9826 const bool FirstNameEmpty =
9827 FirstName.isIdentifier() && !FirstName.getAsIdentifierInfo();
9828 const bool SecondNameEmpty =
9829 SecondName.isIdentifier() && !SecondName.getAsIdentifierInfo();
9830 assert((!FirstNameEmpty || !SecondNameEmpty) &&
9831 "Both template parameters cannot be unnamed.");
9832 ODRDiagError(FirstDecl->getLocation(), FirstDecl->getSourceRange(),
9833 FirstNameEmpty ? ParamEmptyName : ParamName)
9834 << FirstName;
9835 ODRDiagNote(SecondDecl->getLocation(), SecondDecl->getSourceRange(),
9836 SecondNameEmpty ? ParamEmptyName : ParamName)
9837 << SecondName;
9838 break;
9839 }
9840
9841 switch (FirstDecl->getKind()) {
9842 default:
9843 llvm_unreachable("Invalid template parameter type.");
9844 case Decl::TemplateTypeParm: {
9845 const auto *FirstParam = cast<TemplateTypeParmDecl>(FirstDecl);
9846 const auto *SecondParam = cast<TemplateTypeParmDecl>(SecondDecl);
9847 const bool HasFirstDefaultArgument =
9848 FirstParam->hasDefaultArgument() &&
9849 !FirstParam->defaultArgumentWasInherited();
9850 const bool HasSecondDefaultArgument =
9851 SecondParam->hasDefaultArgument() &&
9852 !SecondParam->defaultArgumentWasInherited();
9853
9854 if (HasFirstDefaultArgument != HasSecondDefaultArgument) {
9855 ODRDiagError(FirstDecl->getLocation(),
9856 FirstDecl->getSourceRange(),
9857 ParamSingleDefaultArgument)
9858 << HasFirstDefaultArgument;
9859 ODRDiagNote(SecondDecl->getLocation(),
9860 SecondDecl->getSourceRange(),
9861 ParamSingleDefaultArgument)
9862 << HasSecondDefaultArgument;
9863 break;
9864 }
9865
9866 assert(HasFirstDefaultArgument && HasSecondDefaultArgument &&
9867 "Expecting default arguments.");
9868
9869 ODRDiagError(FirstDecl->getLocation(), FirstDecl->getSourceRange(),
9870 ParamDifferentDefaultArgument);
9871 ODRDiagNote(SecondDecl->getLocation(), SecondDecl->getSourceRange(),
9872 ParamDifferentDefaultArgument);
9873
9874 break;
9875 }
9876 case Decl::NonTypeTemplateParm: {
9877 const auto *FirstParam = cast<NonTypeTemplateParmDecl>(FirstDecl);
9878 const auto *SecondParam = cast<NonTypeTemplateParmDecl>(SecondDecl);
9879 const bool HasFirstDefaultArgument =
9880 FirstParam->hasDefaultArgument() &&
9881 !FirstParam->defaultArgumentWasInherited();
9882 const bool HasSecondDefaultArgument =
9883 SecondParam->hasDefaultArgument() &&
9884 !SecondParam->defaultArgumentWasInherited();
9885
9886 if (HasFirstDefaultArgument != HasSecondDefaultArgument) {
9887 ODRDiagError(FirstDecl->getLocation(),
9888 FirstDecl->getSourceRange(),
9889 ParamSingleDefaultArgument)
9890 << HasFirstDefaultArgument;
9891 ODRDiagNote(SecondDecl->getLocation(),
9892 SecondDecl->getSourceRange(),
9893 ParamSingleDefaultArgument)
9894 << HasSecondDefaultArgument;
9895 break;
9896 }
9897
9898 assert(HasFirstDefaultArgument && HasSecondDefaultArgument &&
9899 "Expecting default arguments.");
9900
9901 ODRDiagError(FirstDecl->getLocation(), FirstDecl->getSourceRange(),
9902 ParamDifferentDefaultArgument);
9903 ODRDiagNote(SecondDecl->getLocation(), SecondDecl->getSourceRange(),
9904 ParamDifferentDefaultArgument);
9905
9906 break;
9907 }
9908 case Decl::TemplateTemplateParm: {
9909 const auto *FirstParam = cast<TemplateTemplateParmDecl>(FirstDecl);
9910 const auto *SecondParam =
9911 cast<TemplateTemplateParmDecl>(SecondDecl);
9912 const bool HasFirstDefaultArgument =
9913 FirstParam->hasDefaultArgument() &&
9914 !FirstParam->defaultArgumentWasInherited();
9915 const bool HasSecondDefaultArgument =
9916 SecondParam->hasDefaultArgument() &&
9917 !SecondParam->defaultArgumentWasInherited();
9918
9919 if (HasFirstDefaultArgument != HasSecondDefaultArgument) {
9920 ODRDiagError(FirstDecl->getLocation(),
9921 FirstDecl->getSourceRange(),
9922 ParamSingleDefaultArgument)
9923 << HasFirstDefaultArgument;
9924 ODRDiagNote(SecondDecl->getLocation(),
9925 SecondDecl->getSourceRange(),
9926 ParamSingleDefaultArgument)
9927 << HasSecondDefaultArgument;
9928 break;
9929 }
9930
9931 assert(HasFirstDefaultArgument && HasSecondDefaultArgument &&
9932 "Expecting default arguments.");
9933
9934 ODRDiagError(FirstDecl->getLocation(), FirstDecl->getSourceRange(),
9935 ParamDifferentDefaultArgument);
9936 ODRDiagNote(SecondDecl->getLocation(), SecondDecl->getSourceRange(),
9937 ParamDifferentDefaultArgument);
9938
9939 break;
9940 }
9941 }
9942
9943 break;
9944 }
9945
9946 if (FirstIt != FirstEnd) {
9947 Diagnosed = true;
9948 break;
9949 }
9950 }
9951
Richard Trieue7f7ed22017-02-22 01:11:25 +00009952 DeclHashes FirstHashes;
9953 DeclHashes SecondHashes;
Richard Trieue7f7ed22017-02-22 01:11:25 +00009954
Richard Trieue6caa262017-12-23 00:41:01 +00009955 auto PopulateHashes = [&ComputeSubDeclODRHash, FirstRecord](
9956 DeclHashes &Hashes, CXXRecordDecl *Record) {
Richard Trieue7f7ed22017-02-22 01:11:25 +00009957 for (auto *D : Record->decls()) {
9958 // Due to decl merging, the first CXXRecordDecl is the parent of
9959 // Decls in both records.
9960 if (!ODRHash::isWhitelistedDecl(D, FirstRecord))
9961 continue;
Richard Trieue6caa262017-12-23 00:41:01 +00009962 Hashes.emplace_back(D, ComputeSubDeclODRHash(D));
Richard Trieue7f7ed22017-02-22 01:11:25 +00009963 }
9964 };
9965 PopulateHashes(FirstHashes, FirstRecord);
9966 PopulateHashes(SecondHashes, SecondRecord);
9967
9968 // Used with err_module_odr_violation_mismatch_decl and
9969 // note_module_odr_violation_mismatch_decl
Richard Trieu11d566a2017-06-12 21:58:22 +00009970 // This list should be the same Decl's as in ODRHash::isWhiteListedDecl
Richard Trieue7f7ed22017-02-22 01:11:25 +00009971 enum {
9972 EndOfClass,
9973 PublicSpecifer,
9974 PrivateSpecifer,
9975 ProtectedSpecifer,
Richard Trieu639d7b62017-02-22 22:22:42 +00009976 StaticAssert,
Richard Trieud0786092017-02-23 00:23:01 +00009977 Field,
Richard Trieu48143742017-02-28 21:24:38 +00009978 CXXMethod,
Richard Trieu11d566a2017-06-12 21:58:22 +00009979 TypeAlias,
9980 TypeDef,
Richard Trieu6e13ff32017-06-16 02:44:29 +00009981 Var,
Richard Trieuac6a1b62017-07-08 02:04:42 +00009982 Friend,
Richard Trieu9359e8f2018-05-30 01:12:26 +00009983 FunctionTemplate,
Richard Trieue7f7ed22017-02-22 01:11:25 +00009984 Other
9985 } FirstDiffType = Other,
9986 SecondDiffType = Other;
9987
9988 auto DifferenceSelector = [](Decl *D) {
9989 assert(D && "valid Decl required");
9990 switch (D->getKind()) {
9991 default:
9992 return Other;
9993 case Decl::AccessSpec:
9994 switch (D->getAccess()) {
9995 case AS_public:
9996 return PublicSpecifer;
9997 case AS_private:
9998 return PrivateSpecifer;
9999 case AS_protected:
10000 return ProtectedSpecifer;
10001 case AS_none:
Simon Pilgrimeeb1b302017-02-22 13:21:24 +000010002 break;
Richard Trieue7f7ed22017-02-22 01:11:25 +000010003 }
Simon Pilgrimeeb1b302017-02-22 13:21:24 +000010004 llvm_unreachable("Invalid access specifier");
Richard Trieu639d7b62017-02-22 22:22:42 +000010005 case Decl::StaticAssert:
10006 return StaticAssert;
Richard Trieud0786092017-02-23 00:23:01 +000010007 case Decl::Field:
10008 return Field;
Richard Trieu48143742017-02-28 21:24:38 +000010009 case Decl::CXXMethod:
Richard Trieu1c71d512017-07-15 02:55:13 +000010010 case Decl::CXXConstructor:
10011 case Decl::CXXDestructor:
Richard Trieu48143742017-02-28 21:24:38 +000010012 return CXXMethod;
Richard Trieu11d566a2017-06-12 21:58:22 +000010013 case Decl::TypeAlias:
10014 return TypeAlias;
10015 case Decl::Typedef:
10016 return TypeDef;
Richard Trieu6e13ff32017-06-16 02:44:29 +000010017 case Decl::Var:
10018 return Var;
Richard Trieuac6a1b62017-07-08 02:04:42 +000010019 case Decl::Friend:
10020 return Friend;
Richard Trieu9359e8f2018-05-30 01:12:26 +000010021 case Decl::FunctionTemplate:
10022 return FunctionTemplate;
Richard Trieue7f7ed22017-02-22 01:11:25 +000010023 }
10024 };
10025
10026 Decl *FirstDecl = nullptr;
10027 Decl *SecondDecl = nullptr;
10028 auto FirstIt = FirstHashes.begin();
10029 auto SecondIt = SecondHashes.begin();
10030
10031 // If there is a diagnoseable difference, FirstDiffType and
10032 // SecondDiffType will not be Other and FirstDecl and SecondDecl will be
10033 // filled in if not EndOfClass.
10034 while (FirstIt != FirstHashes.end() || SecondIt != SecondHashes.end()) {
Benjamin Kramer6f224d22017-02-22 10:19:45 +000010035 if (FirstIt != FirstHashes.end() && SecondIt != SecondHashes.end() &&
10036 FirstIt->second == SecondIt->second) {
Richard Trieue7f7ed22017-02-22 01:11:25 +000010037 ++FirstIt;
10038 ++SecondIt;
10039 continue;
Richard Trieudc4cb022017-02-17 07:19:24 +000010040 }
Richard Smithcd45dbc2014-04-19 03:48:30 +000010041
Richard Trieue7f7ed22017-02-22 01:11:25 +000010042 FirstDecl = FirstIt == FirstHashes.end() ? nullptr : FirstIt->first;
10043 SecondDecl = SecondIt == SecondHashes.end() ? nullptr : SecondIt->first;
10044
10045 FirstDiffType = FirstDecl ? DifferenceSelector(FirstDecl) : EndOfClass;
10046 SecondDiffType =
10047 SecondDecl ? DifferenceSelector(SecondDecl) : EndOfClass;
10048
10049 break;
Richard Smithcd45dbc2014-04-19 03:48:30 +000010050 }
Richard Trieue7f7ed22017-02-22 01:11:25 +000010051
10052 if (FirstDiffType == Other || SecondDiffType == Other) {
10053 // Reaching this point means an unexpected Decl was encountered
10054 // or no difference was detected. This causes a generic error
10055 // message to be emitted.
10056 Diag(FirstRecord->getLocation(),
10057 diag::err_module_odr_violation_different_definitions)
10058 << FirstRecord << FirstModule.empty() << FirstModule;
10059
Richard Trieuca48d362017-06-21 01:43:13 +000010060 if (FirstDecl) {
10061 Diag(FirstDecl->getLocation(), diag::note_first_module_difference)
10062 << FirstRecord << FirstDecl->getSourceRange();
10063 }
10064
Richard Trieue7f7ed22017-02-22 01:11:25 +000010065 Diag(SecondRecord->getLocation(),
10066 diag::note_module_odr_violation_different_definitions)
10067 << SecondModule;
Richard Trieuca48d362017-06-21 01:43:13 +000010068
10069 if (SecondDecl) {
10070 Diag(SecondDecl->getLocation(), diag::note_second_module_difference)
10071 << SecondDecl->getSourceRange();
10072 }
10073
Richard Trieue7f7ed22017-02-22 01:11:25 +000010074 Diagnosed = true;
10075 break;
10076 }
10077
10078 if (FirstDiffType != SecondDiffType) {
10079 SourceLocation FirstLoc;
10080 SourceRange FirstRange;
10081 if (FirstDiffType == EndOfClass) {
10082 FirstLoc = FirstRecord->getBraceRange().getEnd();
10083 } else {
10084 FirstLoc = FirstIt->first->getLocation();
10085 FirstRange = FirstIt->first->getSourceRange();
10086 }
10087 Diag(FirstLoc, diag::err_module_odr_violation_mismatch_decl)
10088 << FirstRecord << FirstModule.empty() << FirstModule << FirstRange
10089 << FirstDiffType;
10090
10091 SourceLocation SecondLoc;
10092 SourceRange SecondRange;
10093 if (SecondDiffType == EndOfClass) {
10094 SecondLoc = SecondRecord->getBraceRange().getEnd();
10095 } else {
10096 SecondLoc = SecondDecl->getLocation();
10097 SecondRange = SecondDecl->getSourceRange();
10098 }
10099 Diag(SecondLoc, diag::note_module_odr_violation_mismatch_decl)
10100 << SecondModule << SecondRange << SecondDiffType;
10101 Diagnosed = true;
10102 break;
10103 }
10104
Richard Trieu639d7b62017-02-22 22:22:42 +000010105 assert(FirstDiffType == SecondDiffType);
10106
10107 // Used with err_module_odr_violation_mismatch_decl_diff and
10108 // note_module_odr_violation_mismatch_decl_diff
Richard Trieu9359e8f2018-05-30 01:12:26 +000010109 enum ODRDeclDifference {
Richard Trieu639d7b62017-02-22 22:22:42 +000010110 StaticAssertCondition,
10111 StaticAssertMessage,
10112 StaticAssertOnlyMessage,
Richard Trieud0786092017-02-23 00:23:01 +000010113 FieldName,
Richard Trieu8459ddf2017-02-24 02:59:12 +000010114 FieldTypeName,
Richard Trieu93772fc2017-02-24 20:59:28 +000010115 FieldSingleBitField,
Richard Trieu8d543e22017-02-24 23:35:37 +000010116 FieldDifferentWidthBitField,
10117 FieldSingleMutable,
10118 FieldSingleInitializer,
10119 FieldDifferentInitializers,
Richard Trieu48143742017-02-28 21:24:38 +000010120 MethodName,
Richard Trieu583e2c12017-03-04 00:08:58 +000010121 MethodDeleted,
Richard Trieu27c1b1a2018-07-10 01:40:50 +000010122 MethodDefaulted,
Richard Trieu583e2c12017-03-04 00:08:58 +000010123 MethodVirtual,
10124 MethodStatic,
10125 MethodVolatile,
10126 MethodConst,
10127 MethodInline,
Richard Trieu02552272017-05-02 23:58:52 +000010128 MethodNumberParameters,
10129 MethodParameterType,
10130 MethodParameterName,
Richard Trieu6e13ff32017-06-16 02:44:29 +000010131 MethodParameterSingleDefaultArgument,
10132 MethodParameterDifferentDefaultArgument,
Richard Trieu7282d322018-04-25 00:31:15 +000010133 MethodNoTemplateArguments,
10134 MethodDifferentNumberTemplateArguments,
10135 MethodDifferentTemplateArgument,
Richard Trieu27c1b1a2018-07-10 01:40:50 +000010136 MethodSingleBody,
10137 MethodDifferentBody,
Richard Trieu11d566a2017-06-12 21:58:22 +000010138 TypedefName,
10139 TypedefType,
Richard Trieu6e13ff32017-06-16 02:44:29 +000010140 VarName,
10141 VarType,
10142 VarSingleInitializer,
10143 VarDifferentInitializer,
10144 VarConstexpr,
Richard Trieuac6a1b62017-07-08 02:04:42 +000010145 FriendTypeFunction,
10146 FriendType,
10147 FriendFunction,
Richard Trieu9359e8f2018-05-30 01:12:26 +000010148 FunctionTemplateDifferentNumberParameters,
10149 FunctionTemplateParameterDifferentKind,
10150 FunctionTemplateParameterName,
10151 FunctionTemplateParameterSingleDefaultArgument,
10152 FunctionTemplateParameterDifferentDefaultArgument,
10153 FunctionTemplateParameterDifferentType,
10154 FunctionTemplatePackParameter,
Richard Trieu639d7b62017-02-22 22:22:42 +000010155 };
10156
10157 // These lambdas have the common portions of the ODR diagnostics. This
10158 // has the same return as Diag(), so addition parameters can be passed
10159 // in with operator<<
10160 auto ODRDiagError = [FirstRecord, &FirstModule, this](
10161 SourceLocation Loc, SourceRange Range, ODRDeclDifference DiffType) {
10162 return Diag(Loc, diag::err_module_odr_violation_mismatch_decl_diff)
10163 << FirstRecord << FirstModule.empty() << FirstModule << Range
10164 << DiffType;
10165 };
10166 auto ODRDiagNote = [&SecondModule, this](
10167 SourceLocation Loc, SourceRange Range, ODRDeclDifference DiffType) {
10168 return Diag(Loc, diag::note_module_odr_violation_mismatch_decl_diff)
10169 << SecondModule << Range << DiffType;
10170 };
10171
Richard Trieu639d7b62017-02-22 22:22:42 +000010172 switch (FirstDiffType) {
10173 case Other:
10174 case EndOfClass:
10175 case PublicSpecifer:
10176 case PrivateSpecifer:
10177 case ProtectedSpecifer:
10178 llvm_unreachable("Invalid diff type");
10179
10180 case StaticAssert: {
Vedant Kumar48b4f762018-04-14 01:40:48 +000010181 StaticAssertDecl *FirstSA = cast<StaticAssertDecl>(FirstDecl);
10182 StaticAssertDecl *SecondSA = cast<StaticAssertDecl>(SecondDecl);
Richard Trieu639d7b62017-02-22 22:22:42 +000010183
10184 Expr *FirstExpr = FirstSA->getAssertExpr();
10185 Expr *SecondExpr = SecondSA->getAssertExpr();
10186 unsigned FirstODRHash = ComputeODRHash(FirstExpr);
10187 unsigned SecondODRHash = ComputeODRHash(SecondExpr);
10188 if (FirstODRHash != SecondODRHash) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +000010189 ODRDiagError(FirstExpr->getBeginLoc(), FirstExpr->getSourceRange(),
Richard Trieu639d7b62017-02-22 22:22:42 +000010190 StaticAssertCondition);
Stephen Kellyf2ceec42018-08-09 21:08:08 +000010191 ODRDiagNote(SecondExpr->getBeginLoc(), SecondExpr->getSourceRange(),
10192 StaticAssertCondition);
Richard Trieu639d7b62017-02-22 22:22:42 +000010193 Diagnosed = true;
10194 break;
10195 }
10196
10197 StringLiteral *FirstStr = FirstSA->getMessage();
10198 StringLiteral *SecondStr = SecondSA->getMessage();
10199 assert((FirstStr || SecondStr) && "Both messages cannot be empty");
10200 if ((FirstStr && !SecondStr) || (!FirstStr && SecondStr)) {
10201 SourceLocation FirstLoc, SecondLoc;
10202 SourceRange FirstRange, SecondRange;
10203 if (FirstStr) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +000010204 FirstLoc = FirstStr->getBeginLoc();
Richard Trieu639d7b62017-02-22 22:22:42 +000010205 FirstRange = FirstStr->getSourceRange();
10206 } else {
Stephen Kellyf2ceec42018-08-09 21:08:08 +000010207 FirstLoc = FirstSA->getBeginLoc();
Richard Trieu639d7b62017-02-22 22:22:42 +000010208 FirstRange = FirstSA->getSourceRange();
10209 }
10210 if (SecondStr) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +000010211 SecondLoc = SecondStr->getBeginLoc();
Richard Trieu639d7b62017-02-22 22:22:42 +000010212 SecondRange = SecondStr->getSourceRange();
10213 } else {
Stephen Kellyf2ceec42018-08-09 21:08:08 +000010214 SecondLoc = SecondSA->getBeginLoc();
Richard Trieu639d7b62017-02-22 22:22:42 +000010215 SecondRange = SecondSA->getSourceRange();
10216 }
10217 ODRDiagError(FirstLoc, FirstRange, StaticAssertOnlyMessage)
10218 << (FirstStr == nullptr);
10219 ODRDiagNote(SecondLoc, SecondRange, StaticAssertOnlyMessage)
10220 << (SecondStr == nullptr);
10221 Diagnosed = true;
10222 break;
10223 }
10224
10225 if (FirstStr && SecondStr &&
10226 FirstStr->getString() != SecondStr->getString()) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +000010227 ODRDiagError(FirstStr->getBeginLoc(), FirstStr->getSourceRange(),
Richard Trieu639d7b62017-02-22 22:22:42 +000010228 StaticAssertMessage);
Stephen Kellyf2ceec42018-08-09 21:08:08 +000010229 ODRDiagNote(SecondStr->getBeginLoc(), SecondStr->getSourceRange(),
Richard Trieu639d7b62017-02-22 22:22:42 +000010230 StaticAssertMessage);
10231 Diagnosed = true;
10232 break;
10233 }
10234 break;
10235 }
Richard Trieud0786092017-02-23 00:23:01 +000010236 case Field: {
Vedant Kumar48b4f762018-04-14 01:40:48 +000010237 FieldDecl *FirstField = cast<FieldDecl>(FirstDecl);
10238 FieldDecl *SecondField = cast<FieldDecl>(SecondDecl);
Richard Trieud0786092017-02-23 00:23:01 +000010239 IdentifierInfo *FirstII = FirstField->getIdentifier();
10240 IdentifierInfo *SecondII = SecondField->getIdentifier();
10241 if (FirstII->getName() != SecondII->getName()) {
10242 ODRDiagError(FirstField->getLocation(), FirstField->getSourceRange(),
10243 FieldName)
10244 << FirstII;
10245 ODRDiagNote(SecondField->getLocation(), SecondField->getSourceRange(),
10246 FieldName)
10247 << SecondII;
10248
10249 Diagnosed = true;
10250 break;
10251 }
Richard Trieu8459ddf2017-02-24 02:59:12 +000010252
Richard Smithdbafb6c2017-06-29 23:23:46 +000010253 assert(getContext().hasSameType(FirstField->getType(),
10254 SecondField->getType()));
Richard Trieu8459ddf2017-02-24 02:59:12 +000010255
10256 QualType FirstType = FirstField->getType();
10257 QualType SecondType = SecondField->getType();
Richard Trieuce81b192017-05-17 03:23:35 +000010258 if (ComputeQualTypeODRHash(FirstType) !=
10259 ComputeQualTypeODRHash(SecondType)) {
Richard Trieu8459ddf2017-02-24 02:59:12 +000010260 ODRDiagError(FirstField->getLocation(), FirstField->getSourceRange(),
10261 FieldTypeName)
10262 << FirstII << FirstType;
10263 ODRDiagNote(SecondField->getLocation(), SecondField->getSourceRange(),
10264 FieldTypeName)
10265 << SecondII << SecondType;
10266
10267 Diagnosed = true;
10268 break;
10269 }
10270
Richard Trieu93772fc2017-02-24 20:59:28 +000010271 const bool IsFirstBitField = FirstField->isBitField();
10272 const bool IsSecondBitField = SecondField->isBitField();
10273 if (IsFirstBitField != IsSecondBitField) {
10274 ODRDiagError(FirstField->getLocation(), FirstField->getSourceRange(),
10275 FieldSingleBitField)
10276 << FirstII << IsFirstBitField;
10277 ODRDiagNote(SecondField->getLocation(), SecondField->getSourceRange(),
10278 FieldSingleBitField)
10279 << SecondII << IsSecondBitField;
10280 Diagnosed = true;
10281 break;
10282 }
10283
10284 if (IsFirstBitField && IsSecondBitField) {
10285 ODRDiagError(FirstField->getLocation(), FirstField->getSourceRange(),
10286 FieldDifferentWidthBitField)
10287 << FirstII << FirstField->getBitWidth()->getSourceRange();
10288 ODRDiagNote(SecondField->getLocation(), SecondField->getSourceRange(),
10289 FieldDifferentWidthBitField)
10290 << SecondII << SecondField->getBitWidth()->getSourceRange();
10291 Diagnosed = true;
10292 break;
10293 }
10294
Richard Trieu8d543e22017-02-24 23:35:37 +000010295 const bool IsFirstMutable = FirstField->isMutable();
10296 const bool IsSecondMutable = SecondField->isMutable();
10297 if (IsFirstMutable != IsSecondMutable) {
10298 ODRDiagError(FirstField->getLocation(), FirstField->getSourceRange(),
10299 FieldSingleMutable)
10300 << FirstII << IsFirstMutable;
10301 ODRDiagNote(SecondField->getLocation(), SecondField->getSourceRange(),
10302 FieldSingleMutable)
10303 << SecondII << IsSecondMutable;
10304 Diagnosed = true;
10305 break;
10306 }
10307
10308 const Expr *FirstInitializer = FirstField->getInClassInitializer();
10309 const Expr *SecondInitializer = SecondField->getInClassInitializer();
10310 if ((!FirstInitializer && SecondInitializer) ||
10311 (FirstInitializer && !SecondInitializer)) {
10312 ODRDiagError(FirstField->getLocation(), FirstField->getSourceRange(),
10313 FieldSingleInitializer)
10314 << FirstII << (FirstInitializer != nullptr);
10315 ODRDiagNote(SecondField->getLocation(), SecondField->getSourceRange(),
10316 FieldSingleInitializer)
10317 << SecondII << (SecondInitializer != nullptr);
10318 Diagnosed = true;
10319 break;
10320 }
10321
10322 if (FirstInitializer && SecondInitializer) {
10323 unsigned FirstInitHash = ComputeODRHash(FirstInitializer);
10324 unsigned SecondInitHash = ComputeODRHash(SecondInitializer);
10325 if (FirstInitHash != SecondInitHash) {
10326 ODRDiagError(FirstField->getLocation(),
10327 FirstField->getSourceRange(),
10328 FieldDifferentInitializers)
10329 << FirstII << FirstInitializer->getSourceRange();
10330 ODRDiagNote(SecondField->getLocation(),
10331 SecondField->getSourceRange(),
10332 FieldDifferentInitializers)
10333 << SecondII << SecondInitializer->getSourceRange();
10334 Diagnosed = true;
10335 break;
10336 }
10337 }
10338
Richard Trieud0786092017-02-23 00:23:01 +000010339 break;
10340 }
Richard Trieu48143742017-02-28 21:24:38 +000010341 case CXXMethod: {
Richard Trieu1c71d512017-07-15 02:55:13 +000010342 enum {
10343 DiagMethod,
10344 DiagConstructor,
10345 DiagDestructor,
10346 } FirstMethodType,
10347 SecondMethodType;
10348 auto GetMethodTypeForDiagnostics = [](const CXXMethodDecl* D) {
10349 if (isa<CXXConstructorDecl>(D)) return DiagConstructor;
10350 if (isa<CXXDestructorDecl>(D)) return DiagDestructor;
10351 return DiagMethod;
10352 };
Vedant Kumar48b4f762018-04-14 01:40:48 +000010353 const CXXMethodDecl *FirstMethod = cast<CXXMethodDecl>(FirstDecl);
10354 const CXXMethodDecl *SecondMethod = cast<CXXMethodDecl>(SecondDecl);
Richard Trieu1c71d512017-07-15 02:55:13 +000010355 FirstMethodType = GetMethodTypeForDiagnostics(FirstMethod);
10356 SecondMethodType = GetMethodTypeForDiagnostics(SecondMethod);
Richard Trieu583e2c12017-03-04 00:08:58 +000010357 auto FirstName = FirstMethod->getDeclName();
10358 auto SecondName = SecondMethod->getDeclName();
Richard Trieu1c71d512017-07-15 02:55:13 +000010359 if (FirstMethodType != SecondMethodType || FirstName != SecondName) {
Richard Trieu48143742017-02-28 21:24:38 +000010360 ODRDiagError(FirstMethod->getLocation(),
10361 FirstMethod->getSourceRange(), MethodName)
Richard Trieu1c71d512017-07-15 02:55:13 +000010362 << FirstMethodType << FirstName;
Richard Trieu48143742017-02-28 21:24:38 +000010363 ODRDiagNote(SecondMethod->getLocation(),
10364 SecondMethod->getSourceRange(), MethodName)
Richard Trieu1c71d512017-07-15 02:55:13 +000010365 << SecondMethodType << SecondName;
Richard Trieu48143742017-02-28 21:24:38 +000010366
10367 Diagnosed = true;
10368 break;
10369 }
10370
Richard Trieu27c1b1a2018-07-10 01:40:50 +000010371 const bool FirstDeleted = FirstMethod->isDeletedAsWritten();
10372 const bool SecondDeleted = SecondMethod->isDeletedAsWritten();
Richard Trieu583e2c12017-03-04 00:08:58 +000010373 if (FirstDeleted != SecondDeleted) {
10374 ODRDiagError(FirstMethod->getLocation(),
10375 FirstMethod->getSourceRange(), MethodDeleted)
Richard Trieu1c71d512017-07-15 02:55:13 +000010376 << FirstMethodType << FirstName << FirstDeleted;
Richard Trieu583e2c12017-03-04 00:08:58 +000010377
10378 ODRDiagNote(SecondMethod->getLocation(),
10379 SecondMethod->getSourceRange(), MethodDeleted)
Richard Trieu1c71d512017-07-15 02:55:13 +000010380 << SecondMethodType << SecondName << SecondDeleted;
Richard Trieu583e2c12017-03-04 00:08:58 +000010381 Diagnosed = true;
10382 break;
10383 }
10384
Richard Trieu27c1b1a2018-07-10 01:40:50 +000010385 const bool FirstDefaulted = FirstMethod->isExplicitlyDefaulted();
10386 const bool SecondDefaulted = SecondMethod->isExplicitlyDefaulted();
10387 if (FirstDefaulted != SecondDefaulted) {
10388 ODRDiagError(FirstMethod->getLocation(),
10389 FirstMethod->getSourceRange(), MethodDefaulted)
10390 << FirstMethodType << FirstName << FirstDefaulted;
10391
10392 ODRDiagNote(SecondMethod->getLocation(),
10393 SecondMethod->getSourceRange(), MethodDefaulted)
10394 << SecondMethodType << SecondName << SecondDefaulted;
10395 Diagnosed = true;
10396 break;
10397 }
10398
Richard Trieu583e2c12017-03-04 00:08:58 +000010399 const bool FirstVirtual = FirstMethod->isVirtualAsWritten();
10400 const bool SecondVirtual = SecondMethod->isVirtualAsWritten();
10401 const bool FirstPure = FirstMethod->isPure();
10402 const bool SecondPure = SecondMethod->isPure();
10403 if ((FirstVirtual || SecondVirtual) &&
10404 (FirstVirtual != SecondVirtual || FirstPure != SecondPure)) {
10405 ODRDiagError(FirstMethod->getLocation(),
10406 FirstMethod->getSourceRange(), MethodVirtual)
Richard Trieu1c71d512017-07-15 02:55:13 +000010407 << FirstMethodType << FirstName << FirstPure << FirstVirtual;
Richard Trieu583e2c12017-03-04 00:08:58 +000010408 ODRDiagNote(SecondMethod->getLocation(),
10409 SecondMethod->getSourceRange(), MethodVirtual)
Richard Trieu1c71d512017-07-15 02:55:13 +000010410 << SecondMethodType << SecondName << SecondPure << SecondVirtual;
Richard Trieu583e2c12017-03-04 00:08:58 +000010411 Diagnosed = true;
10412 break;
10413 }
10414
10415 // CXXMethodDecl::isStatic uses the canonical Decl. With Decl merging,
10416 // FirstDecl is the canonical Decl of SecondDecl, so the storage
10417 // class needs to be checked instead.
10418 const auto FirstStorage = FirstMethod->getStorageClass();
10419 const auto SecondStorage = SecondMethod->getStorageClass();
10420 const bool FirstStatic = FirstStorage == SC_Static;
10421 const bool SecondStatic = SecondStorage == SC_Static;
10422 if (FirstStatic != SecondStatic) {
10423 ODRDiagError(FirstMethod->getLocation(),
10424 FirstMethod->getSourceRange(), MethodStatic)
Richard Trieu1c71d512017-07-15 02:55:13 +000010425 << FirstMethodType << FirstName << FirstStatic;
Richard Trieu583e2c12017-03-04 00:08:58 +000010426 ODRDiagNote(SecondMethod->getLocation(),
10427 SecondMethod->getSourceRange(), MethodStatic)
Richard Trieu1c71d512017-07-15 02:55:13 +000010428 << SecondMethodType << SecondName << SecondStatic;
Richard Trieu583e2c12017-03-04 00:08:58 +000010429 Diagnosed = true;
10430 break;
10431 }
10432
10433 const bool FirstVolatile = FirstMethod->isVolatile();
10434 const bool SecondVolatile = SecondMethod->isVolatile();
10435 if (FirstVolatile != SecondVolatile) {
10436 ODRDiagError(FirstMethod->getLocation(),
10437 FirstMethod->getSourceRange(), MethodVolatile)
Richard Trieu1c71d512017-07-15 02:55:13 +000010438 << FirstMethodType << FirstName << FirstVolatile;
Richard Trieu583e2c12017-03-04 00:08:58 +000010439 ODRDiagNote(SecondMethod->getLocation(),
10440 SecondMethod->getSourceRange(), MethodVolatile)
Richard Trieu1c71d512017-07-15 02:55:13 +000010441 << SecondMethodType << SecondName << SecondVolatile;
Richard Trieu583e2c12017-03-04 00:08:58 +000010442 Diagnosed = true;
10443 break;
10444 }
10445
10446 const bool FirstConst = FirstMethod->isConst();
10447 const bool SecondConst = SecondMethod->isConst();
10448 if (FirstConst != SecondConst) {
10449 ODRDiagError(FirstMethod->getLocation(),
10450 FirstMethod->getSourceRange(), MethodConst)
Richard Trieu1c71d512017-07-15 02:55:13 +000010451 << FirstMethodType << FirstName << FirstConst;
Richard Trieu583e2c12017-03-04 00:08:58 +000010452 ODRDiagNote(SecondMethod->getLocation(),
10453 SecondMethod->getSourceRange(), MethodConst)
Richard Trieu1c71d512017-07-15 02:55:13 +000010454 << SecondMethodType << SecondName << SecondConst;
Richard Trieu583e2c12017-03-04 00:08:58 +000010455 Diagnosed = true;
10456 break;
10457 }
10458
10459 const bool FirstInline = FirstMethod->isInlineSpecified();
10460 const bool SecondInline = SecondMethod->isInlineSpecified();
10461 if (FirstInline != SecondInline) {
10462 ODRDiagError(FirstMethod->getLocation(),
10463 FirstMethod->getSourceRange(), MethodInline)
Richard Trieu1c71d512017-07-15 02:55:13 +000010464 << FirstMethodType << FirstName << FirstInline;
Richard Trieu583e2c12017-03-04 00:08:58 +000010465 ODRDiagNote(SecondMethod->getLocation(),
10466 SecondMethod->getSourceRange(), MethodInline)
Richard Trieu1c71d512017-07-15 02:55:13 +000010467 << SecondMethodType << SecondName << SecondInline;
Richard Trieu583e2c12017-03-04 00:08:58 +000010468 Diagnosed = true;
10469 break;
10470 }
10471
Richard Trieu02552272017-05-02 23:58:52 +000010472 const unsigned FirstNumParameters = FirstMethod->param_size();
10473 const unsigned SecondNumParameters = SecondMethod->param_size();
10474 if (FirstNumParameters != SecondNumParameters) {
10475 ODRDiagError(FirstMethod->getLocation(),
10476 FirstMethod->getSourceRange(), MethodNumberParameters)
Richard Trieu1c71d512017-07-15 02:55:13 +000010477 << FirstMethodType << FirstName << FirstNumParameters;
Richard Trieu02552272017-05-02 23:58:52 +000010478 ODRDiagNote(SecondMethod->getLocation(),
10479 SecondMethod->getSourceRange(), MethodNumberParameters)
Richard Trieu1c71d512017-07-15 02:55:13 +000010480 << SecondMethodType << SecondName << SecondNumParameters;
Richard Trieu02552272017-05-02 23:58:52 +000010481 Diagnosed = true;
10482 break;
10483 }
10484
10485 // Need this status boolean to know when break out of the switch.
10486 bool ParameterMismatch = false;
10487 for (unsigned I = 0; I < FirstNumParameters; ++I) {
10488 const ParmVarDecl *FirstParam = FirstMethod->getParamDecl(I);
10489 const ParmVarDecl *SecondParam = SecondMethod->getParamDecl(I);
10490
10491 QualType FirstParamType = FirstParam->getType();
10492 QualType SecondParamType = SecondParam->getType();
10493 if (FirstParamType != SecondParamType &&
10494 ComputeQualTypeODRHash(FirstParamType) !=
10495 ComputeQualTypeODRHash(SecondParamType)) {
Vedant Kumar48b4f762018-04-14 01:40:48 +000010496 if (const DecayedType *ParamDecayedType =
Richard Trieu02552272017-05-02 23:58:52 +000010497 FirstParamType->getAs<DecayedType>()) {
10498 ODRDiagError(FirstMethod->getLocation(),
10499 FirstMethod->getSourceRange(), MethodParameterType)
Richard Trieu1c71d512017-07-15 02:55:13 +000010500 << FirstMethodType << FirstName << (I + 1) << FirstParamType
10501 << true << ParamDecayedType->getOriginalType();
Richard Trieu02552272017-05-02 23:58:52 +000010502 } else {
10503 ODRDiagError(FirstMethod->getLocation(),
10504 FirstMethod->getSourceRange(), MethodParameterType)
Richard Trieu1c71d512017-07-15 02:55:13 +000010505 << FirstMethodType << FirstName << (I + 1) << FirstParamType
10506 << false;
Richard Trieu02552272017-05-02 23:58:52 +000010507 }
10508
Vedant Kumar48b4f762018-04-14 01:40:48 +000010509 if (const DecayedType *ParamDecayedType =
Richard Trieu02552272017-05-02 23:58:52 +000010510 SecondParamType->getAs<DecayedType>()) {
10511 ODRDiagNote(SecondMethod->getLocation(),
10512 SecondMethod->getSourceRange(), MethodParameterType)
Richard Trieu1c71d512017-07-15 02:55:13 +000010513 << SecondMethodType << SecondName << (I + 1)
10514 << SecondParamType << true
Richard Trieu02552272017-05-02 23:58:52 +000010515 << ParamDecayedType->getOriginalType();
10516 } else {
10517 ODRDiagNote(SecondMethod->getLocation(),
10518 SecondMethod->getSourceRange(), MethodParameterType)
Richard Trieu1c71d512017-07-15 02:55:13 +000010519 << SecondMethodType << SecondName << (I + 1)
10520 << SecondParamType << false;
Richard Trieu02552272017-05-02 23:58:52 +000010521 }
10522 ParameterMismatch = true;
10523 break;
10524 }
10525
10526 DeclarationName FirstParamName = FirstParam->getDeclName();
10527 DeclarationName SecondParamName = SecondParam->getDeclName();
10528 if (FirstParamName != SecondParamName) {
10529 ODRDiagError(FirstMethod->getLocation(),
10530 FirstMethod->getSourceRange(), MethodParameterName)
Richard Trieu1c71d512017-07-15 02:55:13 +000010531 << FirstMethodType << FirstName << (I + 1) << FirstParamName;
Richard Trieu02552272017-05-02 23:58:52 +000010532 ODRDiagNote(SecondMethod->getLocation(),
10533 SecondMethod->getSourceRange(), MethodParameterName)
Richard Trieu1c71d512017-07-15 02:55:13 +000010534 << SecondMethodType << SecondName << (I + 1) << SecondParamName;
Richard Trieu02552272017-05-02 23:58:52 +000010535 ParameterMismatch = true;
10536 break;
10537 }
Richard Trieu6e13ff32017-06-16 02:44:29 +000010538
10539 const Expr *FirstInit = FirstParam->getInit();
10540 const Expr *SecondInit = SecondParam->getInit();
10541 if ((FirstInit == nullptr) != (SecondInit == nullptr)) {
10542 ODRDiagError(FirstMethod->getLocation(),
10543 FirstMethod->getSourceRange(),
10544 MethodParameterSingleDefaultArgument)
Richard Trieu1c71d512017-07-15 02:55:13 +000010545 << FirstMethodType << FirstName << (I + 1)
10546 << (FirstInit == nullptr)
Richard Trieu6e13ff32017-06-16 02:44:29 +000010547 << (FirstInit ? FirstInit->getSourceRange() : SourceRange());
10548 ODRDiagNote(SecondMethod->getLocation(),
10549 SecondMethod->getSourceRange(),
10550 MethodParameterSingleDefaultArgument)
Richard Trieu1c71d512017-07-15 02:55:13 +000010551 << SecondMethodType << SecondName << (I + 1)
10552 << (SecondInit == nullptr)
Richard Trieu6e13ff32017-06-16 02:44:29 +000010553 << (SecondInit ? SecondInit->getSourceRange() : SourceRange());
10554 ParameterMismatch = true;
10555 break;
10556 }
10557
10558 if (FirstInit && SecondInit &&
10559 ComputeODRHash(FirstInit) != ComputeODRHash(SecondInit)) {
10560 ODRDiagError(FirstMethod->getLocation(),
10561 FirstMethod->getSourceRange(),
10562 MethodParameterDifferentDefaultArgument)
Richard Trieu1c71d512017-07-15 02:55:13 +000010563 << FirstMethodType << FirstName << (I + 1)
10564 << FirstInit->getSourceRange();
Richard Trieu6e13ff32017-06-16 02:44:29 +000010565 ODRDiagNote(SecondMethod->getLocation(),
10566 SecondMethod->getSourceRange(),
10567 MethodParameterDifferentDefaultArgument)
Richard Trieu1c71d512017-07-15 02:55:13 +000010568 << SecondMethodType << SecondName << (I + 1)
10569 << SecondInit->getSourceRange();
Richard Trieu6e13ff32017-06-16 02:44:29 +000010570 ParameterMismatch = true;
10571 break;
10572
10573 }
Richard Trieu02552272017-05-02 23:58:52 +000010574 }
10575
10576 if (ParameterMismatch) {
10577 Diagnosed = true;
10578 break;
10579 }
10580
Richard Trieu7282d322018-04-25 00:31:15 +000010581 const auto *FirstTemplateArgs =
10582 FirstMethod->getTemplateSpecializationArgs();
10583 const auto *SecondTemplateArgs =
10584 SecondMethod->getTemplateSpecializationArgs();
10585
10586 if ((FirstTemplateArgs && !SecondTemplateArgs) ||
10587 (!FirstTemplateArgs && SecondTemplateArgs)) {
10588 ODRDiagError(FirstMethod->getLocation(),
10589 FirstMethod->getSourceRange(), MethodNoTemplateArguments)
10590 << FirstMethodType << FirstName << (FirstTemplateArgs != nullptr);
10591 ODRDiagNote(SecondMethod->getLocation(),
10592 SecondMethod->getSourceRange(), MethodNoTemplateArguments)
10593 << SecondMethodType << SecondName
10594 << (SecondTemplateArgs != nullptr);
10595
10596 Diagnosed = true;
10597 break;
10598 }
10599
10600 if (FirstTemplateArgs && SecondTemplateArgs) {
10601 // Remove pack expansions from argument list.
10602 auto ExpandTemplateArgumentList =
10603 [](const TemplateArgumentList *TAL) {
10604 llvm::SmallVector<const TemplateArgument *, 8> ExpandedList;
10605 for (const TemplateArgument &TA : TAL->asArray()) {
10606 if (TA.getKind() != TemplateArgument::Pack) {
10607 ExpandedList.push_back(&TA);
10608 continue;
10609 }
10610 for (const TemplateArgument &PackTA : TA.getPackAsArray()) {
10611 ExpandedList.push_back(&PackTA);
10612 }
10613 }
10614 return ExpandedList;
10615 };
10616 llvm::SmallVector<const TemplateArgument *, 8> FirstExpandedList =
10617 ExpandTemplateArgumentList(FirstTemplateArgs);
10618 llvm::SmallVector<const TemplateArgument *, 8> SecondExpandedList =
10619 ExpandTemplateArgumentList(SecondTemplateArgs);
10620
10621 if (FirstExpandedList.size() != SecondExpandedList.size()) {
10622 ODRDiagError(FirstMethod->getLocation(),
10623 FirstMethod->getSourceRange(),
10624 MethodDifferentNumberTemplateArguments)
10625 << FirstMethodType << FirstName
10626 << (unsigned)FirstExpandedList.size();
10627 ODRDiagNote(SecondMethod->getLocation(),
10628 SecondMethod->getSourceRange(),
10629 MethodDifferentNumberTemplateArguments)
10630 << SecondMethodType << SecondName
10631 << (unsigned)SecondExpandedList.size();
10632
10633 Diagnosed = true;
10634 break;
10635 }
10636
10637 bool TemplateArgumentMismatch = false;
10638 for (unsigned i = 0, e = FirstExpandedList.size(); i != e; ++i) {
10639 const TemplateArgument &FirstTA = *FirstExpandedList[i],
10640 &SecondTA = *SecondExpandedList[i];
10641 if (ComputeTemplateArgumentODRHash(FirstTA) ==
10642 ComputeTemplateArgumentODRHash(SecondTA)) {
10643 continue;
10644 }
10645
10646 ODRDiagError(FirstMethod->getLocation(),
10647 FirstMethod->getSourceRange(),
10648 MethodDifferentTemplateArgument)
10649 << FirstMethodType << FirstName << FirstTA << i + 1;
10650 ODRDiagNote(SecondMethod->getLocation(),
10651 SecondMethod->getSourceRange(),
10652 MethodDifferentTemplateArgument)
10653 << SecondMethodType << SecondName << SecondTA << i + 1;
10654
10655 TemplateArgumentMismatch = true;
10656 break;
10657 }
10658
10659 if (TemplateArgumentMismatch) {
10660 Diagnosed = true;
10661 break;
10662 }
10663 }
Richard Trieu27c1b1a2018-07-10 01:40:50 +000010664
10665 // Compute the hash of the method as if it has no body.
10666 auto ComputeCXXMethodODRHash = [&Hash](const CXXMethodDecl *D) {
10667 Hash.clear();
10668 Hash.AddFunctionDecl(D, true /*SkipBody*/);
10669 return Hash.CalculateHash();
10670 };
10671
10672 // Compare the hash generated to the hash stored. A difference means
10673 // that a body was present in the original source. Due to merging,
10674 // the stardard way of detecting a body will not work.
10675 const bool HasFirstBody =
10676 ComputeCXXMethodODRHash(FirstMethod) != FirstMethod->getODRHash();
10677 const bool HasSecondBody =
10678 ComputeCXXMethodODRHash(SecondMethod) != SecondMethod->getODRHash();
10679
10680 if (HasFirstBody != HasSecondBody) {
10681 ODRDiagError(FirstMethod->getLocation(),
10682 FirstMethod->getSourceRange(), MethodSingleBody)
10683 << FirstMethodType << FirstName << HasFirstBody;
10684 ODRDiagNote(SecondMethod->getLocation(),
10685 SecondMethod->getSourceRange(), MethodSingleBody)
10686 << SecondMethodType << SecondName << HasSecondBody;
10687 Diagnosed = true;
10688 break;
10689 }
10690
10691 if (HasFirstBody && HasSecondBody) {
10692 ODRDiagError(FirstMethod->getLocation(),
10693 FirstMethod->getSourceRange(), MethodDifferentBody)
10694 << FirstMethodType << FirstName;
10695 ODRDiagNote(SecondMethod->getLocation(),
10696 SecondMethod->getSourceRange(), MethodDifferentBody)
10697 << SecondMethodType << SecondName;
10698 Diagnosed = true;
10699 break;
10700 }
10701
Richard Trieu48143742017-02-28 21:24:38 +000010702 break;
10703 }
Richard Trieu11d566a2017-06-12 21:58:22 +000010704 case TypeAlias:
10705 case TypeDef: {
Vedant Kumar48b4f762018-04-14 01:40:48 +000010706 TypedefNameDecl *FirstTD = cast<TypedefNameDecl>(FirstDecl);
10707 TypedefNameDecl *SecondTD = cast<TypedefNameDecl>(SecondDecl);
Richard Trieu11d566a2017-06-12 21:58:22 +000010708 auto FirstName = FirstTD->getDeclName();
10709 auto SecondName = SecondTD->getDeclName();
10710 if (FirstName != SecondName) {
10711 ODRDiagError(FirstTD->getLocation(), FirstTD->getSourceRange(),
10712 TypedefName)
10713 << (FirstDiffType == TypeAlias) << FirstName;
10714 ODRDiagNote(SecondTD->getLocation(), SecondTD->getSourceRange(),
10715 TypedefName)
10716 << (FirstDiffType == TypeAlias) << SecondName;
10717 Diagnosed = true;
10718 break;
10719 }
10720
10721 QualType FirstType = FirstTD->getUnderlyingType();
10722 QualType SecondType = SecondTD->getUnderlyingType();
10723 if (ComputeQualTypeODRHash(FirstType) !=
10724 ComputeQualTypeODRHash(SecondType)) {
10725 ODRDiagError(FirstTD->getLocation(), FirstTD->getSourceRange(),
10726 TypedefType)
10727 << (FirstDiffType == TypeAlias) << FirstName << FirstType;
10728 ODRDiagNote(SecondTD->getLocation(), SecondTD->getSourceRange(),
10729 TypedefType)
10730 << (FirstDiffType == TypeAlias) << SecondName << SecondType;
10731 Diagnosed = true;
10732 break;
10733 }
10734 break;
10735 }
Richard Trieu6e13ff32017-06-16 02:44:29 +000010736 case Var: {
Vedant Kumar48b4f762018-04-14 01:40:48 +000010737 VarDecl *FirstVD = cast<VarDecl>(FirstDecl);
10738 VarDecl *SecondVD = cast<VarDecl>(SecondDecl);
Richard Trieu6e13ff32017-06-16 02:44:29 +000010739 auto FirstName = FirstVD->getDeclName();
10740 auto SecondName = SecondVD->getDeclName();
10741 if (FirstName != SecondName) {
10742 ODRDiagError(FirstVD->getLocation(), FirstVD->getSourceRange(),
10743 VarName)
10744 << FirstName;
10745 ODRDiagNote(SecondVD->getLocation(), SecondVD->getSourceRange(),
10746 VarName)
10747 << SecondName;
10748 Diagnosed = true;
10749 break;
10750 }
10751
10752 QualType FirstType = FirstVD->getType();
10753 QualType SecondType = SecondVD->getType();
10754 if (ComputeQualTypeODRHash(FirstType) !=
10755 ComputeQualTypeODRHash(SecondType)) {
10756 ODRDiagError(FirstVD->getLocation(), FirstVD->getSourceRange(),
10757 VarType)
10758 << FirstName << FirstType;
10759 ODRDiagNote(SecondVD->getLocation(), SecondVD->getSourceRange(),
10760 VarType)
10761 << SecondName << SecondType;
10762 Diagnosed = true;
10763 break;
10764 }
10765
10766 const Expr *FirstInit = FirstVD->getInit();
10767 const Expr *SecondInit = SecondVD->getInit();
10768 if ((FirstInit == nullptr) != (SecondInit == nullptr)) {
10769 ODRDiagError(FirstVD->getLocation(), FirstVD->getSourceRange(),
10770 VarSingleInitializer)
10771 << FirstName << (FirstInit == nullptr)
10772 << (FirstInit ? FirstInit->getSourceRange(): SourceRange());
10773 ODRDiagNote(SecondVD->getLocation(), SecondVD->getSourceRange(),
10774 VarSingleInitializer)
10775 << SecondName << (SecondInit == nullptr)
10776 << (SecondInit ? SecondInit->getSourceRange() : SourceRange());
10777 Diagnosed = true;
10778 break;
10779 }
10780
10781 if (FirstInit && SecondInit &&
10782 ComputeODRHash(FirstInit) != ComputeODRHash(SecondInit)) {
10783 ODRDiagError(FirstVD->getLocation(), FirstVD->getSourceRange(),
10784 VarDifferentInitializer)
10785 << FirstName << FirstInit->getSourceRange();
10786 ODRDiagNote(SecondVD->getLocation(), SecondVD->getSourceRange(),
10787 VarDifferentInitializer)
10788 << SecondName << SecondInit->getSourceRange();
10789 Diagnosed = true;
10790 break;
10791 }
10792
10793 const bool FirstIsConstexpr = FirstVD->isConstexpr();
10794 const bool SecondIsConstexpr = SecondVD->isConstexpr();
10795 if (FirstIsConstexpr != SecondIsConstexpr) {
10796 ODRDiagError(FirstVD->getLocation(), FirstVD->getSourceRange(),
10797 VarConstexpr)
10798 << FirstName << FirstIsConstexpr;
10799 ODRDiagNote(SecondVD->getLocation(), SecondVD->getSourceRange(),
10800 VarConstexpr)
10801 << SecondName << SecondIsConstexpr;
10802 Diagnosed = true;
10803 break;
10804 }
10805 break;
10806 }
Richard Trieuac6a1b62017-07-08 02:04:42 +000010807 case Friend: {
Vedant Kumar48b4f762018-04-14 01:40:48 +000010808 FriendDecl *FirstFriend = cast<FriendDecl>(FirstDecl);
10809 FriendDecl *SecondFriend = cast<FriendDecl>(SecondDecl);
Richard Trieuac6a1b62017-07-08 02:04:42 +000010810
10811 NamedDecl *FirstND = FirstFriend->getFriendDecl();
10812 NamedDecl *SecondND = SecondFriend->getFriendDecl();
10813
10814 TypeSourceInfo *FirstTSI = FirstFriend->getFriendType();
10815 TypeSourceInfo *SecondTSI = SecondFriend->getFriendType();
10816
10817 if (FirstND && SecondND) {
10818 ODRDiagError(FirstFriend->getFriendLoc(),
10819 FirstFriend->getSourceRange(), FriendFunction)
10820 << FirstND;
10821 ODRDiagNote(SecondFriend->getFriendLoc(),
10822 SecondFriend->getSourceRange(), FriendFunction)
10823 << SecondND;
10824
10825 Diagnosed = true;
10826 break;
10827 }
10828
10829 if (FirstTSI && SecondTSI) {
10830 QualType FirstFriendType = FirstTSI->getType();
10831 QualType SecondFriendType = SecondTSI->getType();
10832 assert(ComputeQualTypeODRHash(FirstFriendType) !=
10833 ComputeQualTypeODRHash(SecondFriendType));
10834 ODRDiagError(FirstFriend->getFriendLoc(),
10835 FirstFriend->getSourceRange(), FriendType)
10836 << FirstFriendType;
10837 ODRDiagNote(SecondFriend->getFriendLoc(),
10838 SecondFriend->getSourceRange(), FriendType)
10839 << SecondFriendType;
10840 Diagnosed = true;
10841 break;
10842 }
10843
10844 ODRDiagError(FirstFriend->getFriendLoc(), FirstFriend->getSourceRange(),
10845 FriendTypeFunction)
10846 << (FirstTSI == nullptr);
10847 ODRDiagNote(SecondFriend->getFriendLoc(),
10848 SecondFriend->getSourceRange(), FriendTypeFunction)
10849 << (SecondTSI == nullptr);
10850
10851 Diagnosed = true;
10852 break;
10853 }
Richard Trieu9359e8f2018-05-30 01:12:26 +000010854 case FunctionTemplate: {
10855 FunctionTemplateDecl *FirstTemplate =
10856 cast<FunctionTemplateDecl>(FirstDecl);
10857 FunctionTemplateDecl *SecondTemplate =
10858 cast<FunctionTemplateDecl>(SecondDecl);
10859
10860 TemplateParameterList *FirstTPL =
10861 FirstTemplate->getTemplateParameters();
10862 TemplateParameterList *SecondTPL =
10863 SecondTemplate->getTemplateParameters();
10864
10865 if (FirstTPL->size() != SecondTPL->size()) {
10866 ODRDiagError(FirstTemplate->getLocation(),
10867 FirstTemplate->getSourceRange(),
10868 FunctionTemplateDifferentNumberParameters)
10869 << FirstTemplate << FirstTPL->size();
10870 ODRDiagNote(SecondTemplate->getLocation(),
10871 SecondTemplate->getSourceRange(),
10872 FunctionTemplateDifferentNumberParameters)
10873 << SecondTemplate << SecondTPL->size();
10874
10875 Diagnosed = true;
10876 break;
10877 }
10878
10879 bool ParameterMismatch = false;
10880 for (unsigned i = 0, e = FirstTPL->size(); i != e; ++i) {
10881 NamedDecl *FirstParam = FirstTPL->getParam(i);
10882 NamedDecl *SecondParam = SecondTPL->getParam(i);
10883
10884 if (FirstParam->getKind() != SecondParam->getKind()) {
10885 enum {
10886 TemplateTypeParameter,
10887 NonTypeTemplateParameter,
10888 TemplateTemplateParameter,
10889 };
10890 auto GetParamType = [](NamedDecl *D) {
10891 switch (D->getKind()) {
10892 default:
10893 llvm_unreachable("Unexpected template parameter type");
10894 case Decl::TemplateTypeParm:
10895 return TemplateTypeParameter;
10896 case Decl::NonTypeTemplateParm:
10897 return NonTypeTemplateParameter;
10898 case Decl::TemplateTemplateParm:
10899 return TemplateTemplateParameter;
10900 }
10901 };
10902
10903 ODRDiagError(FirstTemplate->getLocation(),
10904 FirstTemplate->getSourceRange(),
10905 FunctionTemplateParameterDifferentKind)
10906 << FirstTemplate << (i + 1) << GetParamType(FirstParam);
10907 ODRDiagNote(SecondTemplate->getLocation(),
10908 SecondTemplate->getSourceRange(),
10909 FunctionTemplateParameterDifferentKind)
10910 << SecondTemplate << (i + 1) << GetParamType(SecondParam);
10911
10912 ParameterMismatch = true;
10913 break;
10914 }
10915
10916 if (FirstParam->getName() != SecondParam->getName()) {
10917 ODRDiagError(FirstTemplate->getLocation(),
10918 FirstTemplate->getSourceRange(),
10919 FunctionTemplateParameterName)
10920 << FirstTemplate << (i + 1) << (bool)FirstParam->getIdentifier()
10921 << FirstParam;
10922 ODRDiagNote(SecondTemplate->getLocation(),
10923 SecondTemplate->getSourceRange(),
10924 FunctionTemplateParameterName)
10925 << SecondTemplate << (i + 1)
10926 << (bool)SecondParam->getIdentifier() << SecondParam;
10927 ParameterMismatch = true;
10928 break;
10929 }
10930
10931 if (isa<TemplateTypeParmDecl>(FirstParam) &&
10932 isa<TemplateTypeParmDecl>(SecondParam)) {
10933 TemplateTypeParmDecl *FirstTTPD =
10934 cast<TemplateTypeParmDecl>(FirstParam);
10935 TemplateTypeParmDecl *SecondTTPD =
10936 cast<TemplateTypeParmDecl>(SecondParam);
10937 bool HasFirstDefaultArgument =
10938 FirstTTPD->hasDefaultArgument() &&
10939 !FirstTTPD->defaultArgumentWasInherited();
10940 bool HasSecondDefaultArgument =
10941 SecondTTPD->hasDefaultArgument() &&
10942 !SecondTTPD->defaultArgumentWasInherited();
10943 if (HasFirstDefaultArgument != HasSecondDefaultArgument) {
10944 ODRDiagError(FirstTemplate->getLocation(),
10945 FirstTemplate->getSourceRange(),
10946 FunctionTemplateParameterSingleDefaultArgument)
10947 << FirstTemplate << (i + 1) << HasFirstDefaultArgument;
10948 ODRDiagNote(SecondTemplate->getLocation(),
10949 SecondTemplate->getSourceRange(),
10950 FunctionTemplateParameterSingleDefaultArgument)
10951 << SecondTemplate << (i + 1) << HasSecondDefaultArgument;
10952 ParameterMismatch = true;
10953 break;
10954 }
10955
10956 if (HasFirstDefaultArgument && HasSecondDefaultArgument) {
10957 QualType FirstType = FirstTTPD->getDefaultArgument();
10958 QualType SecondType = SecondTTPD->getDefaultArgument();
10959 if (ComputeQualTypeODRHash(FirstType) !=
10960 ComputeQualTypeODRHash(SecondType)) {
10961 ODRDiagError(FirstTemplate->getLocation(),
10962 FirstTemplate->getSourceRange(),
10963 FunctionTemplateParameterDifferentDefaultArgument)
10964 << FirstTemplate << (i + 1) << FirstType;
10965 ODRDiagNote(SecondTemplate->getLocation(),
10966 SecondTemplate->getSourceRange(),
10967 FunctionTemplateParameterDifferentDefaultArgument)
10968 << SecondTemplate << (i + 1) << SecondType;
10969 ParameterMismatch = true;
10970 break;
10971 }
10972 }
10973
10974 if (FirstTTPD->isParameterPack() !=
10975 SecondTTPD->isParameterPack()) {
10976 ODRDiagError(FirstTemplate->getLocation(),
10977 FirstTemplate->getSourceRange(),
10978 FunctionTemplatePackParameter)
10979 << FirstTemplate << (i + 1) << FirstTTPD->isParameterPack();
10980 ODRDiagNote(SecondTemplate->getLocation(),
10981 SecondTemplate->getSourceRange(),
10982 FunctionTemplatePackParameter)
10983 << SecondTemplate << (i + 1) << SecondTTPD->isParameterPack();
10984 ParameterMismatch = true;
10985 break;
10986 }
10987 }
10988
10989 if (isa<TemplateTemplateParmDecl>(FirstParam) &&
10990 isa<TemplateTemplateParmDecl>(SecondParam)) {
10991 TemplateTemplateParmDecl *FirstTTPD =
10992 cast<TemplateTemplateParmDecl>(FirstParam);
10993 TemplateTemplateParmDecl *SecondTTPD =
10994 cast<TemplateTemplateParmDecl>(SecondParam);
10995
10996 TemplateParameterList *FirstTPL =
10997 FirstTTPD->getTemplateParameters();
10998 TemplateParameterList *SecondTPL =
10999 SecondTTPD->getTemplateParameters();
11000
11001 if (ComputeTemplateParameterListODRHash(FirstTPL) !=
11002 ComputeTemplateParameterListODRHash(SecondTPL)) {
11003 ODRDiagError(FirstTemplate->getLocation(),
11004 FirstTemplate->getSourceRange(),
11005 FunctionTemplateParameterDifferentType)
11006 << FirstTemplate << (i + 1);
11007 ODRDiagNote(SecondTemplate->getLocation(),
11008 SecondTemplate->getSourceRange(),
11009 FunctionTemplateParameterDifferentType)
11010 << SecondTemplate << (i + 1);
11011 ParameterMismatch = true;
11012 break;
11013 }
11014
11015 bool HasFirstDefaultArgument =
11016 FirstTTPD->hasDefaultArgument() &&
11017 !FirstTTPD->defaultArgumentWasInherited();
11018 bool HasSecondDefaultArgument =
11019 SecondTTPD->hasDefaultArgument() &&
11020 !SecondTTPD->defaultArgumentWasInherited();
11021 if (HasFirstDefaultArgument != HasSecondDefaultArgument) {
11022 ODRDiagError(FirstTemplate->getLocation(),
11023 FirstTemplate->getSourceRange(),
11024 FunctionTemplateParameterSingleDefaultArgument)
11025 << FirstTemplate << (i + 1) << HasFirstDefaultArgument;
11026 ODRDiagNote(SecondTemplate->getLocation(),
11027 SecondTemplate->getSourceRange(),
11028 FunctionTemplateParameterSingleDefaultArgument)
11029 << SecondTemplate << (i + 1) << HasSecondDefaultArgument;
11030 ParameterMismatch = true;
11031 break;
11032 }
11033
11034 if (HasFirstDefaultArgument && HasSecondDefaultArgument) {
11035 TemplateArgument FirstTA =
11036 FirstTTPD->getDefaultArgument().getArgument();
11037 TemplateArgument SecondTA =
11038 SecondTTPD->getDefaultArgument().getArgument();
11039 if (ComputeTemplateArgumentODRHash(FirstTA) !=
11040 ComputeTemplateArgumentODRHash(SecondTA)) {
11041 ODRDiagError(FirstTemplate->getLocation(),
11042 FirstTemplate->getSourceRange(),
11043 FunctionTemplateParameterDifferentDefaultArgument)
11044 << FirstTemplate << (i + 1) << FirstTA;
11045 ODRDiagNote(SecondTemplate->getLocation(),
11046 SecondTemplate->getSourceRange(),
11047 FunctionTemplateParameterDifferentDefaultArgument)
11048 << SecondTemplate << (i + 1) << SecondTA;
11049 ParameterMismatch = true;
11050 break;
11051 }
11052 }
11053
11054 if (FirstTTPD->isParameterPack() !=
11055 SecondTTPD->isParameterPack()) {
11056 ODRDiagError(FirstTemplate->getLocation(),
11057 FirstTemplate->getSourceRange(),
11058 FunctionTemplatePackParameter)
11059 << FirstTemplate << (i + 1) << FirstTTPD->isParameterPack();
11060 ODRDiagNote(SecondTemplate->getLocation(),
11061 SecondTemplate->getSourceRange(),
11062 FunctionTemplatePackParameter)
11063 << SecondTemplate << (i + 1) << SecondTTPD->isParameterPack();
11064 ParameterMismatch = true;
11065 break;
11066 }
11067 }
11068
11069 if (isa<NonTypeTemplateParmDecl>(FirstParam) &&
11070 isa<NonTypeTemplateParmDecl>(SecondParam)) {
11071 NonTypeTemplateParmDecl *FirstNTTPD =
11072 cast<NonTypeTemplateParmDecl>(FirstParam);
11073 NonTypeTemplateParmDecl *SecondNTTPD =
11074 cast<NonTypeTemplateParmDecl>(SecondParam);
11075
11076 QualType FirstType = FirstNTTPD->getType();
11077 QualType SecondType = SecondNTTPD->getType();
11078 if (ComputeQualTypeODRHash(FirstType) !=
11079 ComputeQualTypeODRHash(SecondType)) {
11080 ODRDiagError(FirstTemplate->getLocation(),
11081 FirstTemplate->getSourceRange(),
11082 FunctionTemplateParameterDifferentType)
11083 << FirstTemplate << (i + 1);
11084 ODRDiagNote(SecondTemplate->getLocation(),
11085 SecondTemplate->getSourceRange(),
11086 FunctionTemplateParameterDifferentType)
11087 << SecondTemplate << (i + 1);
11088 ParameterMismatch = true;
11089 break;
11090 }
11091
11092 bool HasFirstDefaultArgument =
11093 FirstNTTPD->hasDefaultArgument() &&
11094 !FirstNTTPD->defaultArgumentWasInherited();
11095 bool HasSecondDefaultArgument =
11096 SecondNTTPD->hasDefaultArgument() &&
11097 !SecondNTTPD->defaultArgumentWasInherited();
11098 if (HasFirstDefaultArgument != HasSecondDefaultArgument) {
11099 ODRDiagError(FirstTemplate->getLocation(),
11100 FirstTemplate->getSourceRange(),
11101 FunctionTemplateParameterSingleDefaultArgument)
11102 << FirstTemplate << (i + 1) << HasFirstDefaultArgument;
11103 ODRDiagNote(SecondTemplate->getLocation(),
11104 SecondTemplate->getSourceRange(),
11105 FunctionTemplateParameterSingleDefaultArgument)
11106 << SecondTemplate << (i + 1) << HasSecondDefaultArgument;
11107 ParameterMismatch = true;
11108 break;
11109 }
11110
11111 if (HasFirstDefaultArgument && HasSecondDefaultArgument) {
11112 Expr *FirstDefaultArgument = FirstNTTPD->getDefaultArgument();
11113 Expr *SecondDefaultArgument = SecondNTTPD->getDefaultArgument();
11114 if (ComputeODRHash(FirstDefaultArgument) !=
11115 ComputeODRHash(SecondDefaultArgument)) {
11116 ODRDiagError(FirstTemplate->getLocation(),
11117 FirstTemplate->getSourceRange(),
11118 FunctionTemplateParameterDifferentDefaultArgument)
11119 << FirstTemplate << (i + 1) << FirstDefaultArgument;
11120 ODRDiagNote(SecondTemplate->getLocation(),
11121 SecondTemplate->getSourceRange(),
11122 FunctionTemplateParameterDifferentDefaultArgument)
11123 << SecondTemplate << (i + 1) << SecondDefaultArgument;
11124 ParameterMismatch = true;
11125 break;
11126 }
11127 }
11128
11129 if (FirstNTTPD->isParameterPack() !=
11130 SecondNTTPD->isParameterPack()) {
11131 ODRDiagError(FirstTemplate->getLocation(),
11132 FirstTemplate->getSourceRange(),
11133 FunctionTemplatePackParameter)
11134 << FirstTemplate << (i + 1) << FirstNTTPD->isParameterPack();
11135 ODRDiagNote(SecondTemplate->getLocation(),
11136 SecondTemplate->getSourceRange(),
11137 FunctionTemplatePackParameter)
11138 << SecondTemplate << (i + 1)
11139 << SecondNTTPD->isParameterPack();
11140 ParameterMismatch = true;
11141 break;
11142 }
11143 }
11144 }
11145
11146 if (ParameterMismatch) {
11147 Diagnosed = true;
11148 break;
11149 }
11150
11151 break;
11152 }
Richard Trieu639d7b62017-02-22 22:22:42 +000011153 }
11154
Eugene Zelenkob8c9e2a2017-11-08 01:03:16 +000011155 if (Diagnosed)
Richard Trieue7f7ed22017-02-22 01:11:25 +000011156 continue;
11157
Richard Trieu708859a2017-06-08 00:56:21 +000011158 Diag(FirstDecl->getLocation(),
11159 diag::err_module_odr_violation_mismatch_decl_unknown)
11160 << FirstRecord << FirstModule.empty() << FirstModule << FirstDiffType
11161 << FirstDecl->getSourceRange();
11162 Diag(SecondDecl->getLocation(),
11163 diag::note_module_odr_violation_mismatch_decl_unknown)
11164 << SecondModule << FirstDiffType << SecondDecl->getSourceRange();
Richard Trieue7f7ed22017-02-22 01:11:25 +000011165 Diagnosed = true;
Richard Smithcd45dbc2014-04-19 03:48:30 +000011166 }
11167
11168 if (!Diagnosed) {
11169 // All definitions are updates to the same declaration. This happens if a
11170 // module instantiates the declaration of a class template specialization
11171 // and two or more other modules instantiate its definition.
11172 //
11173 // FIXME: Indicate which modules had instantiations of this definition.
11174 // FIXME: How can this even happen?
11175 Diag(Merge.first->getLocation(),
11176 diag::err_module_odr_violation_different_instantiations)
11177 << Merge.first;
11178 }
11179 }
Richard Trieue6caa262017-12-23 00:41:01 +000011180
11181 // Issue ODR failures diagnostics for functions.
11182 for (auto &Merge : FunctionOdrMergeFailures) {
11183 enum ODRFunctionDifference {
11184 ReturnType,
11185 ParameterName,
11186 ParameterType,
11187 ParameterSingleDefaultArgument,
11188 ParameterDifferentDefaultArgument,
11189 FunctionBody,
11190 };
11191
11192 FunctionDecl *FirstFunction = Merge.first;
11193 std::string FirstModule = getOwningModuleNameForDiagnostic(FirstFunction);
11194
11195 bool Diagnosed = false;
11196 for (auto &SecondFunction : Merge.second) {
Vedant Kumar48b4f762018-04-14 01:40:48 +000011197
Richard Trieue6caa262017-12-23 00:41:01 +000011198 if (FirstFunction == SecondFunction)
11199 continue;
11200
11201 std::string SecondModule =
11202 getOwningModuleNameForDiagnostic(SecondFunction);
11203
11204 auto ODRDiagError = [FirstFunction, &FirstModule,
11205 this](SourceLocation Loc, SourceRange Range,
11206 ODRFunctionDifference DiffType) {
11207 return Diag(Loc, diag::err_module_odr_violation_function)
11208 << FirstFunction << FirstModule.empty() << FirstModule << Range
11209 << DiffType;
11210 };
11211 auto ODRDiagNote = [&SecondModule, this](SourceLocation Loc,
11212 SourceRange Range,
11213 ODRFunctionDifference DiffType) {
11214 return Diag(Loc, diag::note_module_odr_violation_function)
11215 << SecondModule << Range << DiffType;
11216 };
11217
11218 if (ComputeQualTypeODRHash(FirstFunction->getReturnType()) !=
11219 ComputeQualTypeODRHash(SecondFunction->getReturnType())) {
11220 ODRDiagError(FirstFunction->getReturnTypeSourceRange().getBegin(),
11221 FirstFunction->getReturnTypeSourceRange(), ReturnType)
11222 << FirstFunction->getReturnType();
11223 ODRDiagNote(SecondFunction->getReturnTypeSourceRange().getBegin(),
11224 SecondFunction->getReturnTypeSourceRange(), ReturnType)
11225 << SecondFunction->getReturnType();
11226 Diagnosed = true;
11227 break;
11228 }
11229
11230 assert(FirstFunction->param_size() == SecondFunction->param_size() &&
11231 "Merged functions with different number of parameters");
11232
11233 auto ParamSize = FirstFunction->param_size();
11234 bool ParameterMismatch = false;
11235 for (unsigned I = 0; I < ParamSize; ++I) {
11236 auto *FirstParam = FirstFunction->getParamDecl(I);
11237 auto *SecondParam = SecondFunction->getParamDecl(I);
11238
11239 assert(getContext().hasSameType(FirstParam->getType(),
11240 SecondParam->getType()) &&
11241 "Merged function has different parameter types.");
11242
11243 if (FirstParam->getDeclName() != SecondParam->getDeclName()) {
11244 ODRDiagError(FirstParam->getLocation(), FirstParam->getSourceRange(),
11245 ParameterName)
11246 << I + 1 << FirstParam->getDeclName();
11247 ODRDiagNote(SecondParam->getLocation(), SecondParam->getSourceRange(),
11248 ParameterName)
11249 << I + 1 << SecondParam->getDeclName();
11250 ParameterMismatch = true;
11251 break;
11252 };
11253
11254 QualType FirstParamType = FirstParam->getType();
11255 QualType SecondParamType = SecondParam->getType();
11256 if (FirstParamType != SecondParamType &&
11257 ComputeQualTypeODRHash(FirstParamType) !=
11258 ComputeQualTypeODRHash(SecondParamType)) {
Vedant Kumar48b4f762018-04-14 01:40:48 +000011259 if (const DecayedType *ParamDecayedType =
Richard Trieue6caa262017-12-23 00:41:01 +000011260 FirstParamType->getAs<DecayedType>()) {
11261 ODRDiagError(FirstParam->getLocation(),
11262 FirstParam->getSourceRange(), ParameterType)
11263 << (I + 1) << FirstParamType << true
11264 << ParamDecayedType->getOriginalType();
11265 } else {
11266 ODRDiagError(FirstParam->getLocation(),
11267 FirstParam->getSourceRange(), ParameterType)
11268 << (I + 1) << FirstParamType << false;
11269 }
11270
Vedant Kumar48b4f762018-04-14 01:40:48 +000011271 if (const DecayedType *ParamDecayedType =
Richard Trieue6caa262017-12-23 00:41:01 +000011272 SecondParamType->getAs<DecayedType>()) {
11273 ODRDiagNote(SecondParam->getLocation(),
11274 SecondParam->getSourceRange(), ParameterType)
11275 << (I + 1) << SecondParamType << true
11276 << ParamDecayedType->getOriginalType();
11277 } else {
11278 ODRDiagNote(SecondParam->getLocation(),
11279 SecondParam->getSourceRange(), ParameterType)
11280 << (I + 1) << SecondParamType << false;
11281 }
11282 ParameterMismatch = true;
11283 break;
11284 }
11285
11286 const Expr *FirstInit = FirstParam->getInit();
11287 const Expr *SecondInit = SecondParam->getInit();
11288 if ((FirstInit == nullptr) != (SecondInit == nullptr)) {
11289 ODRDiagError(FirstParam->getLocation(), FirstParam->getSourceRange(),
11290 ParameterSingleDefaultArgument)
11291 << (I + 1) << (FirstInit == nullptr)
11292 << (FirstInit ? FirstInit->getSourceRange() : SourceRange());
11293 ODRDiagNote(SecondParam->getLocation(), SecondParam->getSourceRange(),
11294 ParameterSingleDefaultArgument)
11295 << (I + 1) << (SecondInit == nullptr)
11296 << (SecondInit ? SecondInit->getSourceRange() : SourceRange());
11297 ParameterMismatch = true;
11298 break;
11299 }
11300
11301 if (FirstInit && SecondInit &&
11302 ComputeODRHash(FirstInit) != ComputeODRHash(SecondInit)) {
11303 ODRDiagError(FirstParam->getLocation(), FirstParam->getSourceRange(),
11304 ParameterDifferentDefaultArgument)
11305 << (I + 1) << FirstInit->getSourceRange();
11306 ODRDiagNote(SecondParam->getLocation(), SecondParam->getSourceRange(),
11307 ParameterDifferentDefaultArgument)
11308 << (I + 1) << SecondInit->getSourceRange();
11309 ParameterMismatch = true;
11310 break;
11311 }
11312
11313 assert(ComputeSubDeclODRHash(FirstParam) ==
11314 ComputeSubDeclODRHash(SecondParam) &&
11315 "Undiagnosed parameter difference.");
11316 }
11317
11318 if (ParameterMismatch) {
11319 Diagnosed = true;
11320 break;
11321 }
11322
11323 // If no error has been generated before now, assume the problem is in
11324 // the body and generate a message.
11325 ODRDiagError(FirstFunction->getLocation(),
11326 FirstFunction->getSourceRange(), FunctionBody);
11327 ODRDiagNote(SecondFunction->getLocation(),
11328 SecondFunction->getSourceRange(), FunctionBody);
11329 Diagnosed = true;
11330 break;
11331 }
Evgeny Stupachenkobf25d672018-01-05 02:22:52 +000011332 (void)Diagnosed;
Richard Trieue6caa262017-12-23 00:41:01 +000011333 assert(Diagnosed && "Unable to emit ODR diagnostic.");
11334 }
Richard Trieuab4d7302018-07-25 22:52:05 +000011335
11336 // Issue ODR failures diagnostics for enums.
11337 for (auto &Merge : EnumOdrMergeFailures) {
11338 enum ODREnumDifference {
11339 SingleScopedEnum,
11340 EnumTagKeywordMismatch,
11341 SingleSpecifiedType,
11342 DifferentSpecifiedTypes,
11343 DifferentNumberEnumConstants,
11344 EnumConstantName,
11345 EnumConstantSingleInitilizer,
11346 EnumConstantDifferentInitilizer,
11347 };
11348
11349 // If we've already pointed out a specific problem with this enum, don't
11350 // bother issuing a general "something's different" diagnostic.
11351 if (!DiagnosedOdrMergeFailures.insert(Merge.first).second)
11352 continue;
11353
11354 EnumDecl *FirstEnum = Merge.first;
11355 std::string FirstModule = getOwningModuleNameForDiagnostic(FirstEnum);
11356
11357 using DeclHashes =
11358 llvm::SmallVector<std::pair<EnumConstantDecl *, unsigned>, 4>;
11359 auto PopulateHashes = [&ComputeSubDeclODRHash, FirstEnum](
11360 DeclHashes &Hashes, EnumDecl *Enum) {
11361 for (auto *D : Enum->decls()) {
11362 // Due to decl merging, the first EnumDecl is the parent of
11363 // Decls in both records.
11364 if (!ODRHash::isWhitelistedDecl(D, FirstEnum))
11365 continue;
11366 assert(isa<EnumConstantDecl>(D) && "Unexpected Decl kind");
11367 Hashes.emplace_back(cast<EnumConstantDecl>(D),
11368 ComputeSubDeclODRHash(D));
11369 }
11370 };
11371 DeclHashes FirstHashes;
11372 PopulateHashes(FirstHashes, FirstEnum);
11373 bool Diagnosed = false;
11374 for (auto &SecondEnum : Merge.second) {
11375
11376 if (FirstEnum == SecondEnum)
11377 continue;
11378
11379 std::string SecondModule =
11380 getOwningModuleNameForDiagnostic(SecondEnum);
11381
11382 auto ODRDiagError = [FirstEnum, &FirstModule,
11383 this](SourceLocation Loc, SourceRange Range,
11384 ODREnumDifference DiffType) {
11385 return Diag(Loc, diag::err_module_odr_violation_enum)
11386 << FirstEnum << FirstModule.empty() << FirstModule << Range
11387 << DiffType;
11388 };
11389 auto ODRDiagNote = [&SecondModule, this](SourceLocation Loc,
11390 SourceRange Range,
11391 ODREnumDifference DiffType) {
11392 return Diag(Loc, diag::note_module_odr_violation_enum)
11393 << SecondModule << Range << DiffType;
11394 };
11395
11396 if (FirstEnum->isScoped() != SecondEnum->isScoped()) {
11397 ODRDiagError(FirstEnum->getLocation(), FirstEnum->getSourceRange(),
11398 SingleScopedEnum)
11399 << FirstEnum->isScoped();
11400 ODRDiagNote(SecondEnum->getLocation(), SecondEnum->getSourceRange(),
11401 SingleScopedEnum)
11402 << SecondEnum->isScoped();
11403 Diagnosed = true;
11404 continue;
11405 }
11406
11407 if (FirstEnum->isScoped() && SecondEnum->isScoped()) {
11408 if (FirstEnum->isScopedUsingClassTag() !=
11409 SecondEnum->isScopedUsingClassTag()) {
11410 ODRDiagError(FirstEnum->getLocation(), FirstEnum->getSourceRange(),
11411 EnumTagKeywordMismatch)
11412 << FirstEnum->isScopedUsingClassTag();
11413 ODRDiagNote(SecondEnum->getLocation(), SecondEnum->getSourceRange(),
11414 EnumTagKeywordMismatch)
11415 << SecondEnum->isScopedUsingClassTag();
11416 Diagnosed = true;
11417 continue;
11418 }
11419 }
11420
11421 QualType FirstUnderlyingType =
11422 FirstEnum->getIntegerTypeSourceInfo()
11423 ? FirstEnum->getIntegerTypeSourceInfo()->getType()
11424 : QualType();
11425 QualType SecondUnderlyingType =
11426 SecondEnum->getIntegerTypeSourceInfo()
11427 ? SecondEnum->getIntegerTypeSourceInfo()->getType()
11428 : QualType();
11429 if (FirstUnderlyingType.isNull() != SecondUnderlyingType.isNull()) {
11430 ODRDiagError(FirstEnum->getLocation(), FirstEnum->getSourceRange(),
11431 SingleSpecifiedType)
11432 << !FirstUnderlyingType.isNull();
11433 ODRDiagNote(SecondEnum->getLocation(), SecondEnum->getSourceRange(),
11434 SingleSpecifiedType)
11435 << !SecondUnderlyingType.isNull();
11436 Diagnosed = true;
11437 continue;
11438 }
11439
11440 if (!FirstUnderlyingType.isNull() && !SecondUnderlyingType.isNull()) {
11441 if (ComputeQualTypeODRHash(FirstUnderlyingType) !=
11442 ComputeQualTypeODRHash(SecondUnderlyingType)) {
11443 ODRDiagError(FirstEnum->getLocation(), FirstEnum->getSourceRange(),
11444 DifferentSpecifiedTypes)
11445 << FirstUnderlyingType;
11446 ODRDiagNote(SecondEnum->getLocation(), SecondEnum->getSourceRange(),
11447 DifferentSpecifiedTypes)
11448 << SecondUnderlyingType;
11449 Diagnosed = true;
11450 continue;
11451 }
11452 }
11453
11454 DeclHashes SecondHashes;
11455 PopulateHashes(SecondHashes, SecondEnum);
11456
11457 if (FirstHashes.size() != SecondHashes.size()) {
11458 ODRDiagError(FirstEnum->getLocation(), FirstEnum->getSourceRange(),
11459 DifferentNumberEnumConstants)
11460 << (int)FirstHashes.size();
11461 ODRDiagNote(SecondEnum->getLocation(), SecondEnum->getSourceRange(),
11462 DifferentNumberEnumConstants)
11463 << (int)SecondHashes.size();
11464 Diagnosed = true;
11465 continue;
11466 }
11467
11468 for (unsigned I = 0; I < FirstHashes.size(); ++I) {
11469 if (FirstHashes[I].second == SecondHashes[I].second)
11470 continue;
11471 const EnumConstantDecl *FirstEnumConstant = FirstHashes[I].first;
11472 const EnumConstantDecl *SecondEnumConstant = SecondHashes[I].first;
11473
11474 if (FirstEnumConstant->getDeclName() !=
11475 SecondEnumConstant->getDeclName()) {
11476
11477 ODRDiagError(FirstEnumConstant->getLocation(),
11478 FirstEnumConstant->getSourceRange(), EnumConstantName)
11479 << I + 1 << FirstEnumConstant;
11480 ODRDiagNote(SecondEnumConstant->getLocation(),
11481 SecondEnumConstant->getSourceRange(), EnumConstantName)
11482 << I + 1 << SecondEnumConstant;
11483 Diagnosed = true;
11484 break;
11485 }
11486
11487 const Expr *FirstInit = FirstEnumConstant->getInitExpr();
11488 const Expr *SecondInit = SecondEnumConstant->getInitExpr();
11489 if (!FirstInit && !SecondInit)
11490 continue;
11491
11492 if (!FirstInit || !SecondInit) {
11493 ODRDiagError(FirstEnumConstant->getLocation(),
11494 FirstEnumConstant->getSourceRange(),
11495 EnumConstantSingleInitilizer)
11496 << I + 1 << FirstEnumConstant << (FirstInit != nullptr);
11497 ODRDiagNote(SecondEnumConstant->getLocation(),
11498 SecondEnumConstant->getSourceRange(),
11499 EnumConstantSingleInitilizer)
11500 << I + 1 << SecondEnumConstant << (SecondInit != nullptr);
11501 Diagnosed = true;
11502 break;
11503 }
11504
11505 if (ComputeODRHash(FirstInit) != ComputeODRHash(SecondInit)) {
11506 ODRDiagError(FirstEnumConstant->getLocation(),
11507 FirstEnumConstant->getSourceRange(),
11508 EnumConstantDifferentInitilizer)
11509 << I + 1 << FirstEnumConstant;
11510 ODRDiagNote(SecondEnumConstant->getLocation(),
11511 SecondEnumConstant->getSourceRange(),
11512 EnumConstantDifferentInitilizer)
11513 << I + 1 << SecondEnumConstant;
11514 Diagnosed = true;
11515 break;
11516 }
11517 }
11518 }
11519
11520 (void)Diagnosed;
11521 assert(Diagnosed && "Unable to emit ODR diagnostic.");
11522 }
Guy Benyei11169dd2012-12-18 14:30:41 +000011523}
11524
Richard Smithce18a182015-07-14 00:26:00 +000011525void ASTReader::StartedDeserializing() {
David L. Jonesc4808b9e2016-12-15 20:53:26 +000011526 if (++NumCurrentElementsDeserializing == 1 && ReadTimer.get())
Richard Smithce18a182015-07-14 00:26:00 +000011527 ReadTimer->startTimer();
11528}
11529
Guy Benyei11169dd2012-12-18 14:30:41 +000011530void ASTReader::FinishedDeserializing() {
11531 assert(NumCurrentElementsDeserializing &&
11532 "FinishedDeserializing not paired with StartedDeserializing");
11533 if (NumCurrentElementsDeserializing == 1) {
11534 // We decrease NumCurrentElementsDeserializing only after pending actions
11535 // are finished, to avoid recursively re-calling finishPendingActions().
11536 finishPendingActions();
11537 }
11538 --NumCurrentElementsDeserializing;
11539
Richard Smitha0ce9c42014-07-29 23:23:27 +000011540 if (NumCurrentElementsDeserializing == 0) {
Richard Smitha62d1982018-08-03 01:00:01 +000011541 // Propagate exception specification and deduced type updates along
11542 // redeclaration chains.
11543 //
11544 // We do this now rather than in finishPendingActions because we want to
11545 // be able to walk the complete redeclaration chains of the updated decls.
11546 while (!PendingExceptionSpecUpdates.empty() ||
11547 !PendingDeducedTypeUpdates.empty()) {
11548 auto ESUpdates = std::move(PendingExceptionSpecUpdates);
Richard Smith7226f2a2015-03-23 19:54:56 +000011549 PendingExceptionSpecUpdates.clear();
Richard Smitha62d1982018-08-03 01:00:01 +000011550 for (auto Update : ESUpdates) {
Vassil Vassilev19765fb2016-07-22 21:08:24 +000011551 ProcessingUpdatesRAIIObj ProcessingUpdates(*this);
Richard Smith7226f2a2015-03-23 19:54:56 +000011552 auto *FPT = Update.second->getType()->castAs<FunctionProtoType>();
Richard Smith1d0f1992015-08-19 21:09:32 +000011553 auto ESI = FPT->getExtProtoInfo().ExceptionSpec;
Richard Smithdbafb6c2017-06-29 23:23:46 +000011554 if (auto *Listener = getContext().getASTMutationListener())
Richard Smithd88a7f12015-09-01 20:35:42 +000011555 Listener->ResolvedExceptionSpec(cast<FunctionDecl>(Update.second));
Richard Smith1d0f1992015-08-19 21:09:32 +000011556 for (auto *Redecl : Update.second->redecls())
Richard Smithdbafb6c2017-06-29 23:23:46 +000011557 getContext().adjustExceptionSpec(cast<FunctionDecl>(Redecl), ESI);
Richard Smith7226f2a2015-03-23 19:54:56 +000011558 }
Richard Smitha62d1982018-08-03 01:00:01 +000011559
11560 auto DTUpdates = std::move(PendingDeducedTypeUpdates);
11561 PendingDeducedTypeUpdates.clear();
11562 for (auto Update : DTUpdates) {
11563 ProcessingUpdatesRAIIObj ProcessingUpdates(*this);
11564 // FIXME: If the return type is already deduced, check that it matches.
11565 getContext().adjustDeducedFunctionResultType(Update.first,
11566 Update.second);
11567 }
Richard Smith9e2341d2015-03-23 03:25:59 +000011568 }
11569
Richard Smithce18a182015-07-14 00:26:00 +000011570 if (ReadTimer)
11571 ReadTimer->stopTimer();
11572
Richard Smith0f4e2c42015-08-06 04:23:48 +000011573 diagnoseOdrViolations();
11574
Richard Smith04d05b52014-03-23 00:27:18 +000011575 // We are not in recursive loading, so it's safe to pass the "interesting"
11576 // decls to the consumer.
Richard Smitha0ce9c42014-07-29 23:23:27 +000011577 if (Consumer)
11578 PassInterestingDeclsToConsumer();
Guy Benyei11169dd2012-12-18 14:30:41 +000011579 }
11580}
11581
Argyrios Kyrtzidise5edbf92013-04-26 21:33:35 +000011582void ASTReader::pushExternalDeclIntoScope(NamedDecl *D, DeclarationName Name) {
Richard Smith9e2341d2015-03-23 03:25:59 +000011583 if (IdentifierInfo *II = Name.getAsIdentifierInfo()) {
11584 // Remove any fake results before adding any real ones.
11585 auto It = PendingFakeLookupResults.find(II);
11586 if (It != PendingFakeLookupResults.end()) {
Richard Smitha534a312015-07-21 23:54:07 +000011587 for (auto *ND : It->second)
Richard Smith9e2341d2015-03-23 03:25:59 +000011588 SemaObj->IdResolver.RemoveDecl(ND);
Ben Langmuireb8bd2d2015-04-10 22:25:42 +000011589 // FIXME: this works around module+PCH performance issue.
11590 // Rather than erase the result from the map, which is O(n), just clear
11591 // the vector of NamedDecls.
11592 It->second.clear();
Richard Smith9e2341d2015-03-23 03:25:59 +000011593 }
11594 }
Argyrios Kyrtzidise5edbf92013-04-26 21:33:35 +000011595
11596 if (SemaObj->IdResolver.tryAddTopLevelDecl(D, Name) && SemaObj->TUScope) {
11597 SemaObj->TUScope->AddDecl(D);
11598 } else if (SemaObj->TUScope) {
11599 // Adding the decl to IdResolver may have failed because it was already in
11600 // (even though it was not added in scope). If it is already in, make sure
11601 // it gets in the scope as well.
11602 if (std::find(SemaObj->IdResolver.begin(Name),
11603 SemaObj->IdResolver.end(), D) != SemaObj->IdResolver.end())
11604 SemaObj->TUScope->AddDecl(D);
11605 }
11606}
11607
Richard Smithdbafb6c2017-06-29 23:23:46 +000011608ASTReader::ASTReader(Preprocessor &PP, ASTContext *Context,
David Blaikie61137e12017-01-05 18:23:18 +000011609 const PCHContainerReader &PCHContainerRdr,
11610 ArrayRef<std::shared_ptr<ModuleFileExtension>> Extensions,
11611 StringRef isysroot, bool DisableValidation,
11612 bool AllowASTWithCompilerErrors,
11613 bool AllowConfigurationMismatch, bool ValidateSystemInputs,
11614 bool UseGlobalIndex,
11615 std::unique_ptr<llvm::Timer> ReadTimer)
11616 : Listener(DisableValidation
11617 ? cast<ASTReaderListener>(new SimpleASTReaderListener(PP))
11618 : cast<ASTReaderListener>(new PCHValidator(PP, *this))),
David Blaikie61137e12017-01-05 18:23:18 +000011619 SourceMgr(PP.getSourceManager()), FileMgr(PP.getFileManager()),
David Blaikie9d7c1ba2017-01-05 18:45:43 +000011620 PCHContainerRdr(PCHContainerRdr), Diags(PP.getDiagnostics()), PP(PP),
Richard Smithdbafb6c2017-06-29 23:23:46 +000011621 ContextObj(Context),
Boris Kolpackovd30446f2017-08-31 06:26:43 +000011622 ModuleMgr(PP.getFileManager(), PP.getPCMCache(), PCHContainerRdr,
11623 PP.getHeaderSearchInfo()),
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +000011624 PCMCache(PP.getPCMCache()), DummyIdResolver(PP),
11625 ReadTimer(std::move(ReadTimer)), isysroot(isysroot),
David Blaikie61137e12017-01-05 18:23:18 +000011626 DisableValidation(DisableValidation),
Nico Weber824285e2014-05-08 04:26:47 +000011627 AllowASTWithCompilerErrors(AllowASTWithCompilerErrors),
11628 AllowConfigurationMismatch(AllowConfigurationMismatch),
11629 ValidateSystemInputs(ValidateSystemInputs),
David Blaikie9d7c1ba2017-01-05 18:45:43 +000011630 UseGlobalIndex(UseGlobalIndex), CurrSwitchCaseStmts(&SwitchCaseStmts) {
Guy Benyei11169dd2012-12-18 14:30:41 +000011631 SourceMgr.setExternalSLocEntrySource(this);
Douglas Gregor6623e1f2015-11-03 18:33:07 +000011632
11633 for (const auto &Ext : Extensions) {
11634 auto BlockName = Ext->getExtensionMetadata().BlockName;
11635 auto Known = ModuleFileExtensions.find(BlockName);
11636 if (Known != ModuleFileExtensions.end()) {
11637 Diags.Report(diag::warn_duplicate_module_file_extension)
11638 << BlockName;
11639 continue;
11640 }
11641
11642 ModuleFileExtensions.insert({BlockName, Ext});
11643 }
Guy Benyei11169dd2012-12-18 14:30:41 +000011644}
11645
11646ASTReader::~ASTReader() {
Nico Weber824285e2014-05-08 04:26:47 +000011647 if (OwnsDeserializationListener)
11648 delete DeserializationListener;
Guy Benyei11169dd2012-12-18 14:30:41 +000011649}
Richard Smith10379092016-05-06 23:14:07 +000011650
11651IdentifierResolver &ASTReader::getIdResolver() {
11652 return SemaObj ? SemaObj->IdResolver : DummyIdResolver;
11653}
David L. Jonesbe1557a2016-12-21 00:17:49 +000011654
11655unsigned ASTRecordReader::readRecord(llvm::BitstreamCursor &Cursor,
11656 unsigned AbbrevID) {
11657 Idx = 0;
11658 Record.clear();
11659 return Cursor.readRecord(AbbrevID, Record);
11660}