]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/llvm/tools/clang/lib/AST/DeclBase.cpp
Merge clang 7.0.1 and several follow-up changes
[FreeBSD/FreeBSD.git] / contrib / llvm / tools / clang / lib / AST / DeclBase.cpp
1 //===- DeclBase.cpp - Declaration AST Node Implementation -----------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the Decl and DeclContext classes.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "clang/AST/DeclBase.h"
15 #include "clang/AST/ASTContext.h"
16 #include "clang/AST/ASTMutationListener.h"
17 #include "clang/AST/Attr.h"
18 #include "clang/AST/AttrIterator.h"
19 #include "clang/AST/Decl.h"
20 #include "clang/AST/DeclCXX.h"
21 #include "clang/AST/DeclContextInternals.h"
22 #include "clang/AST/DeclFriend.h"
23 #include "clang/AST/DeclObjC.h"
24 #include "clang/AST/DeclOpenMP.h"
25 #include "clang/AST/DeclTemplate.h"
26 #include "clang/AST/DependentDiagnostic.h"
27 #include "clang/AST/ExternalASTSource.h"
28 #include "clang/AST/Stmt.h"
29 #include "clang/AST/Type.h"
30 #include "clang/Basic/IdentifierTable.h"
31 #include "clang/Basic/LLVM.h"
32 #include "clang/Basic/LangOptions.h"
33 #include "clang/Basic/ObjCRuntime.h"
34 #include "clang/Basic/PartialDiagnostic.h"
35 #include "clang/Basic/SourceLocation.h"
36 #include "clang/Basic/TargetInfo.h"
37 #include "llvm/ADT/ArrayRef.h"
38 #include "llvm/ADT/PointerIntPair.h"
39 #include "llvm/ADT/SmallVector.h"
40 #include "llvm/ADT/StringRef.h"
41 #include "llvm/Support/Casting.h"
42 #include "llvm/Support/ErrorHandling.h"
43 #include "llvm/Support/MathExtras.h"
44 #include "llvm/Support/VersionTuple.h"
45 #include "llvm/Support/raw_ostream.h"
46 #include <algorithm>
47 #include <cassert>
48 #include <cstddef>
49 #include <string>
50 #include <tuple>
51 #include <utility>
52
53 using namespace clang;
54
55 //===----------------------------------------------------------------------===//
56 //  Statistics
57 //===----------------------------------------------------------------------===//
58
59 #define DECL(DERIVED, BASE) static int n##DERIVED##s = 0;
60 #define ABSTRACT_DECL(DECL)
61 #include "clang/AST/DeclNodes.inc"
62
63 void Decl::updateOutOfDate(IdentifierInfo &II) const {
64   getASTContext().getExternalSource()->updateOutOfDateIdentifier(II);
65 }
66
67 #define DECL(DERIVED, BASE)                                                    \
68   static_assert(alignof(Decl) >= alignof(DERIVED##Decl),                       \
69                 "Alignment sufficient after objects prepended to " #DERIVED);
70 #define ABSTRACT_DECL(DECL)
71 #include "clang/AST/DeclNodes.inc"
72
73 void *Decl::operator new(std::size_t Size, const ASTContext &Context,
74                          unsigned ID, std::size_t Extra) {
75   // Allocate an extra 8 bytes worth of storage, which ensures that the
76   // resulting pointer will still be 8-byte aligned.
77   static_assert(sizeof(unsigned) * 2 >= alignof(Decl),
78                 "Decl won't be misaligned");
79   void *Start = Context.Allocate(Size + Extra + 8);
80   void *Result = (char*)Start + 8;
81
82   unsigned *PrefixPtr = (unsigned *)Result - 2;
83
84   // Zero out the first 4 bytes; this is used to store the owning module ID.
85   PrefixPtr[0] = 0;
86
87   // Store the global declaration ID in the second 4 bytes.
88   PrefixPtr[1] = ID;
89
90   return Result;
91 }
92
93 void *Decl::operator new(std::size_t Size, const ASTContext &Ctx,
94                          DeclContext *Parent, std::size_t Extra) {
95   assert(!Parent || &Parent->getParentASTContext() == &Ctx);
96   // With local visibility enabled, we track the owning module even for local
97   // declarations. We create the TU decl early and may not yet know what the
98   // LangOpts are, so conservatively allocate the storage.
99   if (Ctx.getLangOpts().trackLocalOwningModule() || !Parent) {
100     // Ensure required alignment of the resulting object by adding extra
101     // padding at the start if required.
102     size_t ExtraAlign =
103         llvm::OffsetToAlignment(sizeof(Module *), alignof(Decl));
104     auto *Buffer = reinterpret_cast<char *>(
105         ::operator new(ExtraAlign + sizeof(Module *) + Size + Extra, Ctx));
106     Buffer += ExtraAlign;
107     auto *ParentModule =
108         Parent ? cast<Decl>(Parent)->getOwningModule() : nullptr;
109     return new (Buffer) Module*(ParentModule) + 1;
110   }
111   return ::operator new(Size + Extra, Ctx);
112 }
113
114 Module *Decl::getOwningModuleSlow() const {
115   assert(isFromASTFile() && "Not from AST file?");
116   return getASTContext().getExternalSource()->getModule(getOwningModuleID());
117 }
118
119 bool Decl::hasLocalOwningModuleStorage() const {
120   return getASTContext().getLangOpts().trackLocalOwningModule();
121 }
122
123 const char *Decl::getDeclKindName() const {
124   switch (DeclKind) {
125   default: llvm_unreachable("Declaration not in DeclNodes.inc!");
126 #define DECL(DERIVED, BASE) case DERIVED: return #DERIVED;
127 #define ABSTRACT_DECL(DECL)
128 #include "clang/AST/DeclNodes.inc"
129   }
130 }
131
132 void Decl::setInvalidDecl(bool Invalid) {
133   InvalidDecl = Invalid;
134   assert(!isa<TagDecl>(this) || !cast<TagDecl>(this)->isCompleteDefinition());
135   if (!Invalid) {
136     return;
137   }
138
139   if (!isa<ParmVarDecl>(this)) {
140     // Defensive maneuver for ill-formed code: we're likely not to make it to
141     // a point where we set the access specifier, so default it to "public"
142     // to avoid triggering asserts elsewhere in the front end.
143     setAccess(AS_public);
144   }
145
146   // Marking a DecompositionDecl as invalid implies all the child BindingDecl's
147   // are invalid too.
148   if (auto *DD = dyn_cast<DecompositionDecl>(this)) {
149     for (auto *Binding : DD->bindings()) {
150       Binding->setInvalidDecl();
151     }
152   }
153 }
154
155 const char *DeclContext::getDeclKindName() const {
156   switch (DeclKind) {
157   default: llvm_unreachable("Declaration context not in DeclNodes.inc!");
158 #define DECL(DERIVED, BASE) case Decl::DERIVED: return #DERIVED;
159 #define ABSTRACT_DECL(DECL)
160 #include "clang/AST/DeclNodes.inc"
161   }
162 }
163
164 bool Decl::StatisticsEnabled = false;
165 void Decl::EnableStatistics() {
166   StatisticsEnabled = true;
167 }
168
169 void Decl::PrintStats() {
170   llvm::errs() << "\n*** Decl Stats:\n";
171
172   int totalDecls = 0;
173 #define DECL(DERIVED, BASE) totalDecls += n##DERIVED##s;
174 #define ABSTRACT_DECL(DECL)
175 #include "clang/AST/DeclNodes.inc"
176   llvm::errs() << "  " << totalDecls << " decls total.\n";
177
178   int totalBytes = 0;
179 #define DECL(DERIVED, BASE)                                             \
180   if (n##DERIVED##s > 0) {                                              \
181     totalBytes += (int)(n##DERIVED##s * sizeof(DERIVED##Decl));         \
182     llvm::errs() << "    " << n##DERIVED##s << " " #DERIVED " decls, "  \
183                  << sizeof(DERIVED##Decl) << " each ("                  \
184                  << n##DERIVED##s * sizeof(DERIVED##Decl)               \
185                  << " bytes)\n";                                        \
186   }
187 #define ABSTRACT_DECL(DECL)
188 #include "clang/AST/DeclNodes.inc"
189
190   llvm::errs() << "Total bytes = " << totalBytes << "\n";
191 }
192
193 void Decl::add(Kind k) {
194   switch (k) {
195 #define DECL(DERIVED, BASE) case DERIVED: ++n##DERIVED##s; break;
196 #define ABSTRACT_DECL(DECL)
197 #include "clang/AST/DeclNodes.inc"
198   }
199 }
200
201 bool Decl::isTemplateParameterPack() const {
202   if (const auto *TTP = dyn_cast<TemplateTypeParmDecl>(this))
203     return TTP->isParameterPack();
204   if (const auto *NTTP = dyn_cast<NonTypeTemplateParmDecl>(this))
205     return NTTP->isParameterPack();
206   if (const auto *TTP = dyn_cast<TemplateTemplateParmDecl>(this))
207     return TTP->isParameterPack();
208   return false;
209 }
210
211 bool Decl::isParameterPack() const {
212   if (const auto *Parm = dyn_cast<ParmVarDecl>(this))
213     return Parm->isParameterPack();
214
215   return isTemplateParameterPack();
216 }
217
218 FunctionDecl *Decl::getAsFunction() {
219   if (auto *FD = dyn_cast<FunctionDecl>(this))
220     return FD;
221   if (const auto *FTD = dyn_cast<FunctionTemplateDecl>(this))
222     return FTD->getTemplatedDecl();
223   return nullptr;
224 }
225
226 bool Decl::isTemplateDecl() const {
227   return isa<TemplateDecl>(this);
228 }
229
230 TemplateDecl *Decl::getDescribedTemplate() const {
231   if (auto *FD = dyn_cast<FunctionDecl>(this))
232     return FD->getDescribedFunctionTemplate();
233   else if (auto *RD = dyn_cast<CXXRecordDecl>(this))
234     return RD->getDescribedClassTemplate();
235   else if (auto *VD = dyn_cast<VarDecl>(this))
236     return VD->getDescribedVarTemplate();
237   else if (auto *AD = dyn_cast<TypeAliasDecl>(this))
238     return AD->getDescribedAliasTemplate();
239
240   return nullptr;
241 }
242
243 bool Decl::isTemplated() const {
244   // A declaration is dependent if it is a template or a template pattern, or
245   // is within (lexcially for a friend, semantically otherwise) a dependent
246   // context.
247   // FIXME: Should local extern declarations be treated like friends?
248   if (auto *AsDC = dyn_cast<DeclContext>(this))
249     return AsDC->isDependentContext();
250   auto *DC = getFriendObjectKind() ? getLexicalDeclContext() : getDeclContext();
251   return DC->isDependentContext() || isTemplateDecl() || getDescribedTemplate();
252 }
253
254 const DeclContext *Decl::getParentFunctionOrMethod() const {
255   for (const DeclContext *DC = getDeclContext();
256        DC && !DC->isTranslationUnit() && !DC->isNamespace();
257        DC = DC->getParent())
258     if (DC->isFunctionOrMethod())
259       return DC;
260
261   return nullptr;
262 }
263
264 //===----------------------------------------------------------------------===//
265 // PrettyStackTraceDecl Implementation
266 //===----------------------------------------------------------------------===//
267
268 void PrettyStackTraceDecl::print(raw_ostream &OS) const {
269   SourceLocation TheLoc = Loc;
270   if (TheLoc.isInvalid() && TheDecl)
271     TheLoc = TheDecl->getLocation();
272
273   if (TheLoc.isValid()) {
274     TheLoc.print(OS, SM);
275     OS << ": ";
276   }
277
278   OS << Message;
279
280   if (const auto *DN = dyn_cast_or_null<NamedDecl>(TheDecl)) {
281     OS << " '";
282     DN->printQualifiedName(OS);
283     OS << '\'';
284   }
285   OS << '\n';
286 }
287
288 //===----------------------------------------------------------------------===//
289 // Decl Implementation
290 //===----------------------------------------------------------------------===//
291
292 // Out-of-line virtual method providing a home for Decl.
293 Decl::~Decl() = default;
294
295 void Decl::setDeclContext(DeclContext *DC) {
296   DeclCtx = DC;
297 }
298
299 void Decl::setLexicalDeclContext(DeclContext *DC) {
300   if (DC == getLexicalDeclContext())
301     return;
302
303   if (isInSemaDC()) {
304     setDeclContextsImpl(getDeclContext(), DC, getASTContext());
305   } else {
306     getMultipleDC()->LexicalDC = DC;
307   }
308
309   // FIXME: We shouldn't be changing the lexical context of declarations
310   // imported from AST files.
311   if (!isFromASTFile()) {
312     setModuleOwnershipKind(getModuleOwnershipKindForChildOf(DC));
313     if (hasOwningModule())
314       setLocalOwningModule(cast<Decl>(DC)->getOwningModule());
315   }
316
317   assert(
318       (getModuleOwnershipKind() != ModuleOwnershipKind::VisibleWhenImported ||
319        getOwningModule()) &&
320       "hidden declaration has no owning module");
321 }
322
323 void Decl::setDeclContextsImpl(DeclContext *SemaDC, DeclContext *LexicalDC,
324                                ASTContext &Ctx) {
325   if (SemaDC == LexicalDC) {
326     DeclCtx = SemaDC;
327   } else {
328     auto *MDC = new (Ctx) Decl::MultipleDC();
329     MDC->SemanticDC = SemaDC;
330     MDC->LexicalDC = LexicalDC;
331     DeclCtx = MDC;
332   }
333 }
334
335 bool Decl::isLexicallyWithinFunctionOrMethod() const {
336   const DeclContext *LDC = getLexicalDeclContext();
337   while (true) {
338     if (LDC->isFunctionOrMethod())
339       return true;
340     if (!isa<TagDecl>(LDC))
341       return false;
342     LDC = LDC->getLexicalParent();
343   }
344   return false;
345 }
346
347 bool Decl::isInAnonymousNamespace() const {
348   for (const DeclContext *DC = getDeclContext(); DC; DC = DC->getParent()) {
349     if (const auto *ND = dyn_cast<NamespaceDecl>(DC))
350       if (ND->isAnonymousNamespace())
351         return true;
352   }
353
354   return false;
355 }
356
357 bool Decl::isInStdNamespace() const {
358   return getDeclContext()->isStdNamespace();
359 }
360
361 TranslationUnitDecl *Decl::getTranslationUnitDecl() {
362   if (auto *TUD = dyn_cast<TranslationUnitDecl>(this))
363     return TUD;
364
365   DeclContext *DC = getDeclContext();
366   assert(DC && "This decl is not contained in a translation unit!");
367
368   while (!DC->isTranslationUnit()) {
369     DC = DC->getParent();
370     assert(DC && "This decl is not contained in a translation unit!");
371   }
372
373   return cast<TranslationUnitDecl>(DC);
374 }
375
376 ASTContext &Decl::getASTContext() const {
377   return getTranslationUnitDecl()->getASTContext();
378 }
379
380 ASTMutationListener *Decl::getASTMutationListener() const {
381   return getASTContext().getASTMutationListener();
382 }
383
384 unsigned Decl::getMaxAlignment() const {
385   if (!hasAttrs())
386     return 0;
387
388   unsigned Align = 0;
389   const AttrVec &V = getAttrs();
390   ASTContext &Ctx = getASTContext();
391   specific_attr_iterator<AlignedAttr> I(V.begin()), E(V.end());
392   for (; I != E; ++I)
393     Align = std::max(Align, I->getAlignment(Ctx));
394   return Align;
395 }
396
397 bool Decl::isUsed(bool CheckUsedAttr) const {
398   const Decl *CanonD = getCanonicalDecl();
399   if (CanonD->Used)
400     return true;
401
402   // Check for used attribute.
403   // Ask the most recent decl, since attributes accumulate in the redecl chain.
404   if (CheckUsedAttr && getMostRecentDecl()->hasAttr<UsedAttr>())
405     return true;
406
407   // The information may have not been deserialized yet. Force deserialization
408   // to complete the needed information.
409   return getMostRecentDecl()->getCanonicalDecl()->Used;
410 }
411
412 void Decl::markUsed(ASTContext &C) {
413   if (isUsed(false))
414     return;
415
416   if (C.getASTMutationListener())
417     C.getASTMutationListener()->DeclarationMarkedUsed(this);
418
419   setIsUsed();
420 }
421
422 bool Decl::isReferenced() const {
423   if (Referenced)
424     return true;
425
426   // Check redeclarations.
427   for (const auto *I : redecls())
428     if (I->Referenced)
429       return true;
430
431   return false;
432 }
433
434 bool Decl::isExported() const {
435   if (isModulePrivate())
436     return false;
437   // Namespaces are always exported.
438   if (isa<TranslationUnitDecl>(this) || isa<NamespaceDecl>(this))
439     return true;
440   // Otherwise, this is a strictly lexical check.
441   for (auto *DC = getLexicalDeclContext(); DC; DC = DC->getLexicalParent()) {
442     if (cast<Decl>(DC)->isModulePrivate())
443       return false;
444     if (isa<ExportDecl>(DC))
445       return true;
446   }
447   return false;
448 }
449
450 ExternalSourceSymbolAttr *Decl::getExternalSourceSymbolAttr() const {
451   const Decl *Definition = nullptr;
452   if (auto *ID = dyn_cast<ObjCInterfaceDecl>(this)) {
453     Definition = ID->getDefinition();
454   } else if (auto *PD = dyn_cast<ObjCProtocolDecl>(this)) {
455     Definition = PD->getDefinition();
456   } else if (auto *TD = dyn_cast<TagDecl>(this)) {
457     Definition = TD->getDefinition();
458   }
459   if (!Definition)
460     Definition = this;
461
462   if (auto *attr = Definition->getAttr<ExternalSourceSymbolAttr>())
463     return attr;
464   if (auto *dcd = dyn_cast<Decl>(getDeclContext())) {
465     return dcd->getAttr<ExternalSourceSymbolAttr>();
466   }
467
468   return nullptr;
469 }
470
471 bool Decl::hasDefiningAttr() const {
472   return hasAttr<AliasAttr>() || hasAttr<IFuncAttr>();
473 }
474
475 const Attr *Decl::getDefiningAttr() const {
476   if (auto *AA = getAttr<AliasAttr>())
477     return AA;
478   if (auto *IFA = getAttr<IFuncAttr>())
479     return IFA;
480   return nullptr;
481 }
482
483 static StringRef getRealizedPlatform(const AvailabilityAttr *A,
484                                      const ASTContext &Context) {
485   // Check if this is an App Extension "platform", and if so chop off
486   // the suffix for matching with the actual platform.
487   StringRef RealizedPlatform = A->getPlatform()->getName();
488   if (!Context.getLangOpts().AppExt)
489     return RealizedPlatform;
490   size_t suffix = RealizedPlatform.rfind("_app_extension");
491   if (suffix != StringRef::npos)
492     return RealizedPlatform.slice(0, suffix);
493   return RealizedPlatform;
494 }
495
496 /// Determine the availability of the given declaration based on
497 /// the target platform.
498 ///
499 /// When it returns an availability result other than \c AR_Available,
500 /// if the \p Message parameter is non-NULL, it will be set to a
501 /// string describing why the entity is unavailable.
502 ///
503 /// FIXME: Make these strings localizable, since they end up in
504 /// diagnostics.
505 static AvailabilityResult CheckAvailability(ASTContext &Context,
506                                             const AvailabilityAttr *A,
507                                             std::string *Message,
508                                             VersionTuple EnclosingVersion) {
509   if (EnclosingVersion.empty())
510     EnclosingVersion = Context.getTargetInfo().getPlatformMinVersion();
511
512   if (EnclosingVersion.empty())
513     return AR_Available;
514
515   StringRef ActualPlatform = A->getPlatform()->getName();
516   StringRef TargetPlatform = Context.getTargetInfo().getPlatformName();
517
518   // Match the platform name.
519   if (getRealizedPlatform(A, Context) != TargetPlatform)
520     return AR_Available;
521
522   StringRef PrettyPlatformName
523     = AvailabilityAttr::getPrettyPlatformName(ActualPlatform);
524
525   if (PrettyPlatformName.empty())
526     PrettyPlatformName = ActualPlatform;
527
528   std::string HintMessage;
529   if (!A->getMessage().empty()) {
530     HintMessage = " - ";
531     HintMessage += A->getMessage();
532   }
533
534   // Make sure that this declaration has not been marked 'unavailable'.
535   if (A->getUnavailable()) {
536     if (Message) {
537       Message->clear();
538       llvm::raw_string_ostream Out(*Message);
539       Out << "not available on " << PrettyPlatformName
540           << HintMessage;
541     }
542
543     return AR_Unavailable;
544   }
545
546   // Make sure that this declaration has already been introduced.
547   if (!A->getIntroduced().empty() &&
548       EnclosingVersion < A->getIntroduced()) {
549     if (Message) {
550       Message->clear();
551       llvm::raw_string_ostream Out(*Message);
552       VersionTuple VTI(A->getIntroduced());
553       Out << "introduced in " << PrettyPlatformName << ' '
554           << VTI << HintMessage;
555     }
556
557     return A->getStrict() ? AR_Unavailable : AR_NotYetIntroduced;
558   }
559
560   // Make sure that this declaration hasn't been obsoleted.
561   if (!A->getObsoleted().empty() && EnclosingVersion >= A->getObsoleted()) {
562     if (Message) {
563       Message->clear();
564       llvm::raw_string_ostream Out(*Message);
565       VersionTuple VTO(A->getObsoleted());
566       Out << "obsoleted in " << PrettyPlatformName << ' '
567           << VTO << HintMessage;
568     }
569
570     return AR_Unavailable;
571   }
572
573   // Make sure that this declaration hasn't been deprecated.
574   if (!A->getDeprecated().empty() && EnclosingVersion >= A->getDeprecated()) {
575     if (Message) {
576       Message->clear();
577       llvm::raw_string_ostream Out(*Message);
578       VersionTuple VTD(A->getDeprecated());
579       Out << "first deprecated in " << PrettyPlatformName << ' '
580           << VTD << HintMessage;
581     }
582
583     return AR_Deprecated;
584   }
585
586   return AR_Available;
587 }
588
589 AvailabilityResult Decl::getAvailability(std::string *Message,
590                                          VersionTuple EnclosingVersion,
591                                          StringRef *RealizedPlatform) const {
592   if (auto *FTD = dyn_cast<FunctionTemplateDecl>(this))
593     return FTD->getTemplatedDecl()->getAvailability(Message, EnclosingVersion,
594                                                     RealizedPlatform);
595
596   AvailabilityResult Result = AR_Available;
597   std::string ResultMessage;
598
599   for (const auto *A : attrs()) {
600     if (const auto *Deprecated = dyn_cast<DeprecatedAttr>(A)) {
601       if (Result >= AR_Deprecated)
602         continue;
603
604       if (Message)
605         ResultMessage = Deprecated->getMessage();
606
607       Result = AR_Deprecated;
608       continue;
609     }
610
611     if (const auto *Unavailable = dyn_cast<UnavailableAttr>(A)) {
612       if (Message)
613         *Message = Unavailable->getMessage();
614       return AR_Unavailable;
615     }
616
617     if (const auto *Availability = dyn_cast<AvailabilityAttr>(A)) {
618       AvailabilityResult AR = CheckAvailability(getASTContext(), Availability,
619                                                 Message, EnclosingVersion);
620
621       if (AR == AR_Unavailable) {
622         if (RealizedPlatform)
623           *RealizedPlatform = Availability->getPlatform()->getName();
624         return AR_Unavailable;
625       }
626
627       if (AR > Result) {
628         Result = AR;
629         if (Message)
630           ResultMessage.swap(*Message);
631       }
632       continue;
633     }
634   }
635
636   if (Message)
637     Message->swap(ResultMessage);
638   return Result;
639 }
640
641 VersionTuple Decl::getVersionIntroduced() const {
642   const ASTContext &Context = getASTContext();
643   StringRef TargetPlatform = Context.getTargetInfo().getPlatformName();
644   for (const auto *A : attrs()) {
645     if (const auto *Availability = dyn_cast<AvailabilityAttr>(A)) {
646       if (getRealizedPlatform(Availability, Context) != TargetPlatform)
647         continue;
648       if (!Availability->getIntroduced().empty())
649         return Availability->getIntroduced();
650     }
651   }
652   return {};
653 }
654
655 bool Decl::canBeWeakImported(bool &IsDefinition) const {
656   IsDefinition = false;
657
658   // Variables, if they aren't definitions.
659   if (const auto *Var = dyn_cast<VarDecl>(this)) {
660     if (Var->isThisDeclarationADefinition()) {
661       IsDefinition = true;
662       return false;
663     }
664     return true;
665
666   // Functions, if they aren't definitions.
667   } else if (const auto *FD = dyn_cast<FunctionDecl>(this)) {
668     if (FD->hasBody()) {
669       IsDefinition = true;
670       return false;
671     }
672     return true;
673
674   // Objective-C classes, if this is the non-fragile runtime.
675   } else if (isa<ObjCInterfaceDecl>(this) &&
676              getASTContext().getLangOpts().ObjCRuntime.hasWeakClassImport()) {
677     return true;
678
679   // Nothing else.
680   } else {
681     return false;
682   }
683 }
684
685 bool Decl::isWeakImported() const {
686   bool IsDefinition;
687   if (!canBeWeakImported(IsDefinition))
688     return false;
689
690   for (const auto *A : attrs()) {
691     if (isa<WeakImportAttr>(A))
692       return true;
693
694     if (const auto *Availability = dyn_cast<AvailabilityAttr>(A)) {
695       if (CheckAvailability(getASTContext(), Availability, nullptr,
696                             VersionTuple()) == AR_NotYetIntroduced)
697         return true;
698     }
699   }
700
701   return false;
702 }
703
704 unsigned Decl::getIdentifierNamespaceForKind(Kind DeclKind) {
705   switch (DeclKind) {
706     case Function:
707     case CXXDeductionGuide:
708     case CXXMethod:
709     case CXXConstructor:
710     case ConstructorUsingShadow:
711     case CXXDestructor:
712     case CXXConversion:
713     case EnumConstant:
714     case Var:
715     case ImplicitParam:
716     case ParmVar:
717     case ObjCMethod:
718     case ObjCProperty:
719     case MSProperty:
720       return IDNS_Ordinary;
721     case Label:
722       return IDNS_Label;
723     case IndirectField:
724       return IDNS_Ordinary | IDNS_Member;
725
726     case Binding:
727     case NonTypeTemplateParm:
728     case VarTemplate:
729       // These (C++-only) declarations are found by redeclaration lookup for
730       // tag types, so we include them in the tag namespace.
731       return IDNS_Ordinary | IDNS_Tag;
732
733     case ObjCCompatibleAlias:
734     case ObjCInterface:
735       return IDNS_Ordinary | IDNS_Type;
736
737     case Typedef:
738     case TypeAlias:
739     case TemplateTypeParm:
740     case ObjCTypeParam:
741       return IDNS_Ordinary | IDNS_Type;
742
743     case UnresolvedUsingTypename:
744       return IDNS_Ordinary | IDNS_Type | IDNS_Using;
745
746     case UsingShadow:
747       return 0; // we'll actually overwrite this later
748
749     case UnresolvedUsingValue:
750       return IDNS_Ordinary | IDNS_Using;
751
752     case Using:
753     case UsingPack:
754       return IDNS_Using;
755
756     case ObjCProtocol:
757       return IDNS_ObjCProtocol;
758
759     case Field:
760     case ObjCAtDefsField:
761     case ObjCIvar:
762       return IDNS_Member;
763
764     case Record:
765     case CXXRecord:
766     case Enum:
767       return IDNS_Tag | IDNS_Type;
768
769     case Namespace:
770     case NamespaceAlias:
771       return IDNS_Namespace;
772
773     case FunctionTemplate:
774       return IDNS_Ordinary;
775
776     case ClassTemplate:
777     case TemplateTemplateParm:
778     case TypeAliasTemplate:
779       return IDNS_Ordinary | IDNS_Tag | IDNS_Type;
780
781     case OMPDeclareReduction:
782       return IDNS_OMPReduction;
783
784     // Never have names.
785     case Friend:
786     case FriendTemplate:
787     case AccessSpec:
788     case LinkageSpec:
789     case Export:
790     case FileScopeAsm:
791     case StaticAssert:
792     case ObjCPropertyImpl:
793     case PragmaComment:
794     case PragmaDetectMismatch:
795     case Block:
796     case Captured:
797     case TranslationUnit:
798     case ExternCContext:
799     case Decomposition:
800
801     case UsingDirective:
802     case BuiltinTemplate:
803     case ClassTemplateSpecialization:
804     case ClassTemplatePartialSpecialization:
805     case ClassScopeFunctionSpecialization:
806     case VarTemplateSpecialization:
807     case VarTemplatePartialSpecialization:
808     case ObjCImplementation:
809     case ObjCCategory:
810     case ObjCCategoryImpl:
811     case Import:
812     case OMPThreadPrivate:
813     case OMPCapturedExpr:
814     case Empty:
815       // Never looked up by name.
816       return 0;
817   }
818
819   llvm_unreachable("Invalid DeclKind!");
820 }
821
822 void Decl::setAttrsImpl(const AttrVec &attrs, ASTContext &Ctx) {
823   assert(!HasAttrs && "Decl already contains attrs.");
824
825   AttrVec &AttrBlank = Ctx.getDeclAttrs(this);
826   assert(AttrBlank.empty() && "HasAttrs was wrong?");
827
828   AttrBlank = attrs;
829   HasAttrs = true;
830 }
831
832 void Decl::dropAttrs() {
833   if (!HasAttrs) return;
834
835   HasAttrs = false;
836   getASTContext().eraseDeclAttrs(this);
837 }
838
839 const AttrVec &Decl::getAttrs() const {
840   assert(HasAttrs && "No attrs to get!");
841   return getASTContext().getDeclAttrs(this);
842 }
843
844 Decl *Decl::castFromDeclContext (const DeclContext *D) {
845   Decl::Kind DK = D->getDeclKind();
846   switch(DK) {
847 #define DECL(NAME, BASE)
848 #define DECL_CONTEXT(NAME) \
849     case Decl::NAME:       \
850       return static_cast<NAME##Decl *>(const_cast<DeclContext *>(D));
851 #define DECL_CONTEXT_BASE(NAME)
852 #include "clang/AST/DeclNodes.inc"
853     default:
854 #define DECL(NAME, BASE)
855 #define DECL_CONTEXT_BASE(NAME)                  \
856       if (DK >= first##NAME && DK <= last##NAME) \
857         return static_cast<NAME##Decl *>(const_cast<DeclContext *>(D));
858 #include "clang/AST/DeclNodes.inc"
859       llvm_unreachable("a decl that inherits DeclContext isn't handled");
860   }
861 }
862
863 DeclContext *Decl::castToDeclContext(const Decl *D) {
864   Decl::Kind DK = D->getKind();
865   switch(DK) {
866 #define DECL(NAME, BASE)
867 #define DECL_CONTEXT(NAME) \
868     case Decl::NAME:       \
869       return static_cast<NAME##Decl *>(const_cast<Decl *>(D));
870 #define DECL_CONTEXT_BASE(NAME)
871 #include "clang/AST/DeclNodes.inc"
872     default:
873 #define DECL(NAME, BASE)
874 #define DECL_CONTEXT_BASE(NAME)                                   \
875       if (DK >= first##NAME && DK <= last##NAME)                  \
876         return static_cast<NAME##Decl *>(const_cast<Decl *>(D));
877 #include "clang/AST/DeclNodes.inc"
878       llvm_unreachable("a decl that inherits DeclContext isn't handled");
879   }
880 }
881
882 SourceLocation Decl::getBodyRBrace() const {
883   // Special handling of FunctionDecl to avoid de-serializing the body from PCH.
884   // FunctionDecl stores EndRangeLoc for this purpose.
885   if (const auto *FD = dyn_cast<FunctionDecl>(this)) {
886     const FunctionDecl *Definition;
887     if (FD->hasBody(Definition))
888       return Definition->getSourceRange().getEnd();
889     return {};
890   }
891
892   if (Stmt *Body = getBody())
893     return Body->getSourceRange().getEnd();
894
895   return {};
896 }
897
898 bool Decl::AccessDeclContextSanity() const {
899 #ifndef NDEBUG
900   // Suppress this check if any of the following hold:
901   // 1. this is the translation unit (and thus has no parent)
902   // 2. this is a template parameter (and thus doesn't belong to its context)
903   // 3. this is a non-type template parameter
904   // 4. the context is not a record
905   // 5. it's invalid
906   // 6. it's a C++0x static_assert.
907   // 7. it's a block literal declaration
908   if (isa<TranslationUnitDecl>(this) ||
909       isa<TemplateTypeParmDecl>(this) ||
910       isa<NonTypeTemplateParmDecl>(this) ||
911       !isa<CXXRecordDecl>(getDeclContext()) ||
912       isInvalidDecl() ||
913       isa<StaticAssertDecl>(this) ||
914       isa<BlockDecl>(this) ||
915       // FIXME: a ParmVarDecl can have ClassTemplateSpecialization
916       // as DeclContext (?).
917       isa<ParmVarDecl>(this) ||
918       // FIXME: a ClassTemplateSpecialization or CXXRecordDecl can have
919       // AS_none as access specifier.
920       isa<CXXRecordDecl>(this) ||
921       isa<ClassScopeFunctionSpecializationDecl>(this))
922     return true;
923
924   assert(Access != AS_none &&
925          "Access specifier is AS_none inside a record decl");
926 #endif
927   return true;
928 }
929
930 static Decl::Kind getKind(const Decl *D) { return D->getKind(); }
931 static Decl::Kind getKind(const DeclContext *DC) { return DC->getDeclKind(); }
932
933 const FunctionType *Decl::getFunctionType(bool BlocksToo) const {
934   QualType Ty;
935   if (const auto *D = dyn_cast<ValueDecl>(this))
936     Ty = D->getType();
937   else if (const auto *D = dyn_cast<TypedefNameDecl>(this))
938     Ty = D->getUnderlyingType();
939   else
940     return nullptr;
941
942   if (Ty->isFunctionPointerType())
943     Ty = Ty->getAs<PointerType>()->getPointeeType();
944   else if (BlocksToo && Ty->isBlockPointerType())
945     Ty = Ty->getAs<BlockPointerType>()->getPointeeType();
946
947   return Ty->getAs<FunctionType>();
948 }
949
950 /// Starting at a given context (a Decl or DeclContext), look for a
951 /// code context that is not a closure (a lambda, block, etc.).
952 template <class T> static Decl *getNonClosureContext(T *D) {
953   if (getKind(D) == Decl::CXXMethod) {
954     auto *MD = cast<CXXMethodDecl>(D);
955     if (MD->getOverloadedOperator() == OO_Call &&
956         MD->getParent()->isLambda())
957       return getNonClosureContext(MD->getParent()->getParent());
958     return MD;
959   } else if (auto *FD = dyn_cast<FunctionDecl>(D))
960     return FD;
961   else if (auto *MD = dyn_cast<ObjCMethodDecl>(D))
962     return MD;
963   else if (auto *BD = dyn_cast<BlockDecl>(D))
964     return getNonClosureContext(BD->getParent());
965   else if (auto *CD = dyn_cast<CapturedDecl>(D))
966     return getNonClosureContext(CD->getParent());
967   else
968     return nullptr;
969 }
970
971 Decl *Decl::getNonClosureContext() {
972   return ::getNonClosureContext(this);
973 }
974
975 Decl *DeclContext::getNonClosureAncestor() {
976   return ::getNonClosureContext(this);
977 }
978
979 //===----------------------------------------------------------------------===//
980 // DeclContext Implementation
981 //===----------------------------------------------------------------------===//
982
983 bool DeclContext::classof(const Decl *D) {
984   switch (D->getKind()) {
985 #define DECL(NAME, BASE)
986 #define DECL_CONTEXT(NAME) case Decl::NAME:
987 #define DECL_CONTEXT_BASE(NAME)
988 #include "clang/AST/DeclNodes.inc"
989       return true;
990     default:
991 #define DECL(NAME, BASE)
992 #define DECL_CONTEXT_BASE(NAME)                 \
993       if (D->getKind() >= Decl::first##NAME &&  \
994           D->getKind() <= Decl::last##NAME)     \
995         return true;
996 #include "clang/AST/DeclNodes.inc"
997       return false;
998   }
999 }
1000
1001 DeclContext::~DeclContext() = default;
1002
1003 /// Find the parent context of this context that will be
1004 /// used for unqualified name lookup.
1005 ///
1006 /// Generally, the parent lookup context is the semantic context. However, for
1007 /// a friend function the parent lookup context is the lexical context, which
1008 /// is the class in which the friend is declared.
1009 DeclContext *DeclContext::getLookupParent() {
1010   // FIXME: Find a better way to identify friends
1011   if (isa<FunctionDecl>(this))
1012     if (getParent()->getRedeclContext()->isFileContext() &&
1013         getLexicalParent()->getRedeclContext()->isRecord())
1014       return getLexicalParent();
1015
1016   return getParent();
1017 }
1018
1019 bool DeclContext::isInlineNamespace() const {
1020   return isNamespace() &&
1021          cast<NamespaceDecl>(this)->isInline();
1022 }
1023
1024 bool DeclContext::isStdNamespace() const {
1025   if (!isNamespace())
1026     return false;
1027
1028   const auto *ND = cast<NamespaceDecl>(this);
1029   if (ND->isInline()) {
1030     return ND->getParent()->isStdNamespace();
1031   }
1032
1033   if (!getParent()->getRedeclContext()->isTranslationUnit())
1034     return false;
1035
1036   const IdentifierInfo *II = ND->getIdentifier();
1037   return II && II->isStr("std");
1038 }
1039
1040 bool DeclContext::isDependentContext() const {
1041   if (isFileContext())
1042     return false;
1043
1044   if (isa<ClassTemplatePartialSpecializationDecl>(this))
1045     return true;
1046
1047   if (const auto *Record = dyn_cast<CXXRecordDecl>(this)) {
1048     if (Record->getDescribedClassTemplate())
1049       return true;
1050
1051     if (Record->isDependentLambda())
1052       return true;
1053   }
1054
1055   if (const auto *Function = dyn_cast<FunctionDecl>(this)) {
1056     if (Function->getDescribedFunctionTemplate())
1057       return true;
1058
1059     // Friend function declarations are dependent if their *lexical*
1060     // context is dependent.
1061     if (cast<Decl>(this)->getFriendObjectKind())
1062       return getLexicalParent()->isDependentContext();
1063   }
1064
1065   // FIXME: A variable template is a dependent context, but is not a
1066   // DeclContext. A context within it (such as a lambda-expression)
1067   // should be considered dependent.
1068
1069   return getParent() && getParent()->isDependentContext();
1070 }
1071
1072 bool DeclContext::isTransparentContext() const {
1073   if (DeclKind == Decl::Enum)
1074     return !cast<EnumDecl>(this)->isScoped();
1075   else if (DeclKind == Decl::LinkageSpec || DeclKind == Decl::Export)
1076     return true;
1077
1078   return false;
1079 }
1080
1081 static bool isLinkageSpecContext(const DeclContext *DC,
1082                                  LinkageSpecDecl::LanguageIDs ID) {
1083   while (DC->getDeclKind() != Decl::TranslationUnit) {
1084     if (DC->getDeclKind() == Decl::LinkageSpec)
1085       return cast<LinkageSpecDecl>(DC)->getLanguage() == ID;
1086     DC = DC->getLexicalParent();
1087   }
1088   return false;
1089 }
1090
1091 bool DeclContext::isExternCContext() const {
1092   return isLinkageSpecContext(this, LinkageSpecDecl::lang_c);
1093 }
1094
1095 const LinkageSpecDecl *DeclContext::getExternCContext() const {
1096   const DeclContext *DC = this;
1097   while (DC->getDeclKind() != Decl::TranslationUnit) {
1098     if (DC->getDeclKind() == Decl::LinkageSpec &&
1099         cast<LinkageSpecDecl>(DC)->getLanguage() == LinkageSpecDecl::lang_c)
1100       return cast<LinkageSpecDecl>(DC);
1101     DC = DC->getLexicalParent();
1102   }
1103   return nullptr;
1104 }
1105
1106 bool DeclContext::isExternCXXContext() const {
1107   return isLinkageSpecContext(this, LinkageSpecDecl::lang_cxx);
1108 }
1109
1110 bool DeclContext::Encloses(const DeclContext *DC) const {
1111   if (getPrimaryContext() != this)
1112     return getPrimaryContext()->Encloses(DC);
1113
1114   for (; DC; DC = DC->getParent())
1115     if (DC->getPrimaryContext() == this)
1116       return true;
1117   return false;
1118 }
1119
1120 DeclContext *DeclContext::getPrimaryContext() {
1121   switch (DeclKind) {
1122   case Decl::TranslationUnit:
1123   case Decl::ExternCContext:
1124   case Decl::LinkageSpec:
1125   case Decl::Export:
1126   case Decl::Block:
1127   case Decl::Captured:
1128   case Decl::OMPDeclareReduction:
1129     // There is only one DeclContext for these entities.
1130     return this;
1131
1132   case Decl::Namespace:
1133     // The original namespace is our primary context.
1134     return static_cast<NamespaceDecl *>(this)->getOriginalNamespace();
1135
1136   case Decl::ObjCMethod:
1137     return this;
1138
1139   case Decl::ObjCInterface:
1140     if (auto *Def = cast<ObjCInterfaceDecl>(this)->getDefinition())
1141       return Def;
1142     return this;
1143
1144   case Decl::ObjCProtocol:
1145     if (auto *Def = cast<ObjCProtocolDecl>(this)->getDefinition())
1146       return Def;
1147     return this;
1148
1149   case Decl::ObjCCategory:
1150     return this;
1151
1152   case Decl::ObjCImplementation:
1153   case Decl::ObjCCategoryImpl:
1154     return this;
1155
1156   default:
1157     if (DeclKind >= Decl::firstTag && DeclKind <= Decl::lastTag) {
1158       // If this is a tag type that has a definition or is currently
1159       // being defined, that definition is our primary context.
1160       auto *Tag = cast<TagDecl>(this);
1161
1162       if (TagDecl *Def = Tag->getDefinition())
1163         return Def;
1164
1165       if (const auto *TagTy = dyn_cast<TagType>(Tag->getTypeForDecl())) {
1166         // Note, TagType::getDecl returns the (partial) definition one exists.
1167         TagDecl *PossiblePartialDef = TagTy->getDecl();
1168         if (PossiblePartialDef->isBeingDefined())
1169           return PossiblePartialDef;
1170       } else {
1171         assert(isa<InjectedClassNameType>(Tag->getTypeForDecl()));
1172       }
1173
1174       return Tag;
1175     }
1176
1177     assert(DeclKind >= Decl::firstFunction && DeclKind <= Decl::lastFunction &&
1178           "Unknown DeclContext kind");
1179     return this;
1180   }
1181 }
1182
1183 void
1184 DeclContext::collectAllContexts(SmallVectorImpl<DeclContext *> &Contexts){
1185   Contexts.clear();
1186
1187   if (DeclKind != Decl::Namespace) {
1188     Contexts.push_back(this);
1189     return;
1190   }
1191
1192   auto *Self = static_cast<NamespaceDecl *>(this);
1193   for (NamespaceDecl *N = Self->getMostRecentDecl(); N;
1194        N = N->getPreviousDecl())
1195     Contexts.push_back(N);
1196
1197   std::reverse(Contexts.begin(), Contexts.end());
1198 }
1199
1200 std::pair<Decl *, Decl *>
1201 DeclContext::BuildDeclChain(ArrayRef<Decl *> Decls,
1202                             bool FieldsAlreadyLoaded) {
1203   // Build up a chain of declarations via the Decl::NextInContextAndBits field.
1204   Decl *FirstNewDecl = nullptr;
1205   Decl *PrevDecl = nullptr;
1206   for (auto *D : Decls) {
1207     if (FieldsAlreadyLoaded && isa<FieldDecl>(D))
1208       continue;
1209
1210     if (PrevDecl)
1211       PrevDecl->NextInContextAndBits.setPointer(D);
1212     else
1213       FirstNewDecl = D;
1214
1215     PrevDecl = D;
1216   }
1217
1218   return std::make_pair(FirstNewDecl, PrevDecl);
1219 }
1220
1221 /// We have just acquired external visible storage, and we already have
1222 /// built a lookup map. For every name in the map, pull in the new names from
1223 /// the external storage.
1224 void DeclContext::reconcileExternalVisibleStorage() const {
1225   assert(NeedToReconcileExternalVisibleStorage && LookupPtr);
1226   NeedToReconcileExternalVisibleStorage = false;
1227
1228   for (auto &Lookup : *LookupPtr)
1229     Lookup.second.setHasExternalDecls();
1230 }
1231
1232 /// Load the declarations within this lexical storage from an
1233 /// external source.
1234 /// \return \c true if any declarations were added.
1235 bool
1236 DeclContext::LoadLexicalDeclsFromExternalStorage() const {
1237   ExternalASTSource *Source = getParentASTContext().getExternalSource();
1238   assert(hasExternalLexicalStorage() && Source && "No external storage?");
1239
1240   // Notify that we have a DeclContext that is initializing.
1241   ExternalASTSource::Deserializing ADeclContext(Source);
1242
1243   // Load the external declarations, if any.
1244   SmallVector<Decl*, 64> Decls;
1245   ExternalLexicalStorage = false;
1246   Source->FindExternalLexicalDecls(this, Decls);
1247
1248   if (Decls.empty())
1249     return false;
1250
1251   // We may have already loaded just the fields of this record, in which case
1252   // we need to ignore them.
1253   bool FieldsAlreadyLoaded = false;
1254   if (const auto *RD = dyn_cast<RecordDecl>(this))
1255     FieldsAlreadyLoaded = RD->LoadedFieldsFromExternalStorage;
1256
1257   // Splice the newly-read declarations into the beginning of the list
1258   // of declarations.
1259   Decl *ExternalFirst, *ExternalLast;
1260   std::tie(ExternalFirst, ExternalLast) =
1261       BuildDeclChain(Decls, FieldsAlreadyLoaded);
1262   ExternalLast->NextInContextAndBits.setPointer(FirstDecl);
1263   FirstDecl = ExternalFirst;
1264   if (!LastDecl)
1265     LastDecl = ExternalLast;
1266   return true;
1267 }
1268
1269 DeclContext::lookup_result
1270 ExternalASTSource::SetNoExternalVisibleDeclsForName(const DeclContext *DC,
1271                                                     DeclarationName Name) {
1272   ASTContext &Context = DC->getParentASTContext();
1273   StoredDeclsMap *Map;
1274   if (!(Map = DC->LookupPtr))
1275     Map = DC->CreateStoredDeclsMap(Context);
1276   if (DC->NeedToReconcileExternalVisibleStorage)
1277     DC->reconcileExternalVisibleStorage();
1278
1279   (*Map)[Name].removeExternalDecls();
1280
1281   return DeclContext::lookup_result();
1282 }
1283
1284 DeclContext::lookup_result
1285 ExternalASTSource::SetExternalVisibleDeclsForName(const DeclContext *DC,
1286                                                   DeclarationName Name,
1287                                                   ArrayRef<NamedDecl*> Decls) {
1288   ASTContext &Context = DC->getParentASTContext();
1289   StoredDeclsMap *Map;
1290   if (!(Map = DC->LookupPtr))
1291     Map = DC->CreateStoredDeclsMap(Context);
1292   if (DC->NeedToReconcileExternalVisibleStorage)
1293     DC->reconcileExternalVisibleStorage();
1294
1295   StoredDeclsList &List = (*Map)[Name];
1296
1297   // Clear out any old external visible declarations, to avoid quadratic
1298   // performance in the redeclaration checks below.
1299   List.removeExternalDecls();
1300
1301   if (!List.isNull()) {
1302     // We have both existing declarations and new declarations for this name.
1303     // Some of the declarations may simply replace existing ones. Handle those
1304     // first.
1305     llvm::SmallVector<unsigned, 8> Skip;
1306     for (unsigned I = 0, N = Decls.size(); I != N; ++I)
1307       if (List.HandleRedeclaration(Decls[I], /*IsKnownNewer*/false))
1308         Skip.push_back(I);
1309     Skip.push_back(Decls.size());
1310
1311     // Add in any new declarations.
1312     unsigned SkipPos = 0;
1313     for (unsigned I = 0, N = Decls.size(); I != N; ++I) {
1314       if (I == Skip[SkipPos])
1315         ++SkipPos;
1316       else
1317         List.AddSubsequentDecl(Decls[I]);
1318     }
1319   } else {
1320     // Convert the array to a StoredDeclsList.
1321     for (auto *D : Decls) {
1322       if (List.isNull())
1323         List.setOnlyValue(D);
1324       else
1325         List.AddSubsequentDecl(D);
1326     }
1327   }
1328
1329   return List.getLookupResult();
1330 }
1331
1332 DeclContext::decl_iterator DeclContext::decls_begin() const {
1333   if (hasExternalLexicalStorage())
1334     LoadLexicalDeclsFromExternalStorage();
1335   return decl_iterator(FirstDecl);
1336 }
1337
1338 bool DeclContext::decls_empty() const {
1339   if (hasExternalLexicalStorage())
1340     LoadLexicalDeclsFromExternalStorage();
1341
1342   return !FirstDecl;
1343 }
1344
1345 bool DeclContext::containsDecl(Decl *D) const {
1346   return (D->getLexicalDeclContext() == this &&
1347           (D->NextInContextAndBits.getPointer() || D == LastDecl));
1348 }
1349
1350 bool DeclContext::containsDeclAndLoad(Decl *D) const {
1351   if (hasExternalLexicalStorage())
1352     LoadLexicalDeclsFromExternalStorage();
1353   return containsDecl(D);
1354 }
1355
1356 /// shouldBeHidden - Determine whether a declaration which was declared
1357 /// within its semantic context should be invisible to qualified name lookup.
1358 static bool shouldBeHidden(NamedDecl *D) {
1359   // Skip unnamed declarations.
1360   if (!D->getDeclName())
1361     return true;
1362
1363   // Skip entities that can't be found by name lookup into a particular
1364   // context.
1365   if ((D->getIdentifierNamespace() == 0 && !isa<UsingDirectiveDecl>(D)) ||
1366       D->isTemplateParameter())
1367     return true;
1368
1369   // Skip template specializations.
1370   // FIXME: This feels like a hack. Should DeclarationName support
1371   // template-ids, or is there a better way to keep specializations
1372   // from being visible?
1373   if (isa<ClassTemplateSpecializationDecl>(D))
1374     return true;
1375   if (auto *FD = dyn_cast<FunctionDecl>(D))
1376     if (FD->isFunctionTemplateSpecialization())
1377       return true;
1378
1379   return false;
1380 }
1381
1382 void DeclContext::removeDecl(Decl *D) {
1383   assert(D->getLexicalDeclContext() == this &&
1384          "decl being removed from non-lexical context");
1385   assert((D->NextInContextAndBits.getPointer() || D == LastDecl) &&
1386          "decl is not in decls list");
1387
1388   // Remove D from the decl chain.  This is O(n) but hopefully rare.
1389   if (D == FirstDecl) {
1390     if (D == LastDecl)
1391       FirstDecl = LastDecl = nullptr;
1392     else
1393       FirstDecl = D->NextInContextAndBits.getPointer();
1394   } else {
1395     for (Decl *I = FirstDecl; true; I = I->NextInContextAndBits.getPointer()) {
1396       assert(I && "decl not found in linked list");
1397       if (I->NextInContextAndBits.getPointer() == D) {
1398         I->NextInContextAndBits.setPointer(D->NextInContextAndBits.getPointer());
1399         if (D == LastDecl) LastDecl = I;
1400         break;
1401       }
1402     }
1403   }
1404
1405   // Mark that D is no longer in the decl chain.
1406   D->NextInContextAndBits.setPointer(nullptr);
1407
1408   // Remove D from the lookup table if necessary.
1409   if (isa<NamedDecl>(D)) {
1410     auto *ND = cast<NamedDecl>(D);
1411
1412     // Do not try to remove the declaration if that is invisible to qualified
1413     // lookup.  E.g. template specializations are skipped.
1414     if (shouldBeHidden(ND))
1415       return;
1416
1417     // Remove only decls that have a name
1418     if (!ND->getDeclName())
1419       return;
1420
1421     auto *DC = D->getDeclContext();
1422     do {
1423       StoredDeclsMap *Map = DC->getPrimaryContext()->LookupPtr;
1424       if (Map) {
1425         StoredDeclsMap::iterator Pos = Map->find(ND->getDeclName());
1426         assert(Pos != Map->end() && "no lookup entry for decl");
1427         if (Pos->second.getAsVector() || Pos->second.getAsDecl() == ND)
1428           Pos->second.remove(ND);
1429       }
1430     } while (DC->isTransparentContext() && (DC = DC->getParent()));
1431   }
1432 }
1433
1434 void DeclContext::addHiddenDecl(Decl *D) {
1435   assert(D->getLexicalDeclContext() == this &&
1436          "Decl inserted into wrong lexical context");
1437   assert(!D->getNextDeclInContext() && D != LastDecl &&
1438          "Decl already inserted into a DeclContext");
1439
1440   if (FirstDecl) {
1441     LastDecl->NextInContextAndBits.setPointer(D);
1442     LastDecl = D;
1443   } else {
1444     FirstDecl = LastDecl = D;
1445   }
1446
1447   // Notify a C++ record declaration that we've added a member, so it can
1448   // update its class-specific state.
1449   if (auto *Record = dyn_cast<CXXRecordDecl>(this))
1450     Record->addedMember(D);
1451
1452   // If this is a newly-created (not de-serialized) import declaration, wire
1453   // it in to the list of local import declarations.
1454   if (!D->isFromASTFile()) {
1455     if (auto *Import = dyn_cast<ImportDecl>(D))
1456       D->getASTContext().addedLocalImportDecl(Import);
1457   }
1458 }
1459
1460 void DeclContext::addDecl(Decl *D) {
1461   addHiddenDecl(D);
1462
1463   if (auto *ND = dyn_cast<NamedDecl>(D))
1464     ND->getDeclContext()->getPrimaryContext()->
1465         makeDeclVisibleInContextWithFlags(ND, false, true);
1466 }
1467
1468 void DeclContext::addDeclInternal(Decl *D) {
1469   addHiddenDecl(D);
1470
1471   if (auto *ND = dyn_cast<NamedDecl>(D))
1472     ND->getDeclContext()->getPrimaryContext()->
1473         makeDeclVisibleInContextWithFlags(ND, true, true);
1474 }
1475
1476 /// buildLookup - Build the lookup data structure with all of the
1477 /// declarations in this DeclContext (and any other contexts linked
1478 /// to it or transparent contexts nested within it) and return it.
1479 ///
1480 /// Note that the produced map may miss out declarations from an
1481 /// external source. If it does, those entries will be marked with
1482 /// the 'hasExternalDecls' flag.
1483 StoredDeclsMap *DeclContext::buildLookup() {
1484   assert(this == getPrimaryContext() && "buildLookup called on non-primary DC");
1485
1486   if (!HasLazyLocalLexicalLookups && !HasLazyExternalLexicalLookups)
1487     return LookupPtr;
1488
1489   SmallVector<DeclContext *, 2> Contexts;
1490   collectAllContexts(Contexts);
1491
1492   if (HasLazyExternalLexicalLookups) {
1493     HasLazyExternalLexicalLookups = false;
1494     for (auto *DC : Contexts) {
1495       if (DC->hasExternalLexicalStorage())
1496         HasLazyLocalLexicalLookups |=
1497             DC->LoadLexicalDeclsFromExternalStorage();
1498     }
1499
1500     if (!HasLazyLocalLexicalLookups)
1501       return LookupPtr;
1502   }
1503
1504   for (auto *DC : Contexts)
1505     buildLookupImpl(DC, hasExternalVisibleStorage());
1506
1507   // We no longer have any lazy decls.
1508   HasLazyLocalLexicalLookups = false;
1509   return LookupPtr;
1510 }
1511
1512 /// buildLookupImpl - Build part of the lookup data structure for the
1513 /// declarations contained within DCtx, which will either be this
1514 /// DeclContext, a DeclContext linked to it, or a transparent context
1515 /// nested within it.
1516 void DeclContext::buildLookupImpl(DeclContext *DCtx, bool Internal) {
1517   for (auto *D : DCtx->noload_decls()) {
1518     // Insert this declaration into the lookup structure, but only if
1519     // it's semantically within its decl context. Any other decls which
1520     // should be found in this context are added eagerly.
1521     //
1522     // If it's from an AST file, don't add it now. It'll get handled by
1523     // FindExternalVisibleDeclsByName if needed. Exception: if we're not
1524     // in C++, we do not track external visible decls for the TU, so in
1525     // that case we need to collect them all here.
1526     if (auto *ND = dyn_cast<NamedDecl>(D))
1527       if (ND->getDeclContext() == DCtx && !shouldBeHidden(ND) &&
1528           (!ND->isFromASTFile() ||
1529            (isTranslationUnit() &&
1530             !getParentASTContext().getLangOpts().CPlusPlus)))
1531         makeDeclVisibleInContextImpl(ND, Internal);
1532
1533     // If this declaration is itself a transparent declaration context
1534     // or inline namespace, add the members of this declaration of that
1535     // context (recursively).
1536     if (auto *InnerCtx = dyn_cast<DeclContext>(D))
1537       if (InnerCtx->isTransparentContext() || InnerCtx->isInlineNamespace())
1538         buildLookupImpl(InnerCtx, Internal);
1539   }
1540 }
1541
1542 NamedDecl *const DeclContextLookupResult::SingleElementDummyList = nullptr;
1543
1544 DeclContext::lookup_result
1545 DeclContext::lookup(DeclarationName Name) const {
1546   assert(DeclKind != Decl::LinkageSpec && DeclKind != Decl::Export &&
1547          "should not perform lookups into transparent contexts");
1548
1549   const DeclContext *PrimaryContext = getPrimaryContext();
1550   if (PrimaryContext != this)
1551     return PrimaryContext->lookup(Name);
1552
1553   // If we have an external source, ensure that any later redeclarations of this
1554   // context have been loaded, since they may add names to the result of this
1555   // lookup (or add external visible storage).
1556   ExternalASTSource *Source = getParentASTContext().getExternalSource();
1557   if (Source)
1558     (void)cast<Decl>(this)->getMostRecentDecl();
1559
1560   if (hasExternalVisibleStorage()) {
1561     assert(Source && "external visible storage but no external source?");
1562
1563     if (NeedToReconcileExternalVisibleStorage)
1564       reconcileExternalVisibleStorage();
1565
1566     StoredDeclsMap *Map = LookupPtr;
1567
1568     if (HasLazyLocalLexicalLookups || HasLazyExternalLexicalLookups)
1569       // FIXME: Make buildLookup const?
1570       Map = const_cast<DeclContext*>(this)->buildLookup();
1571
1572     if (!Map)
1573       Map = CreateStoredDeclsMap(getParentASTContext());
1574
1575     // If we have a lookup result with no external decls, we are done.
1576     std::pair<StoredDeclsMap::iterator, bool> R =
1577         Map->insert(std::make_pair(Name, StoredDeclsList()));
1578     if (!R.second && !R.first->second.hasExternalDecls())
1579       return R.first->second.getLookupResult();
1580
1581     if (Source->FindExternalVisibleDeclsByName(this, Name) || !R.second) {
1582       if (StoredDeclsMap *Map = LookupPtr) {
1583         StoredDeclsMap::iterator I = Map->find(Name);
1584         if (I != Map->end())
1585           return I->second.getLookupResult();
1586       }
1587     }
1588
1589     return {};
1590   }
1591
1592   StoredDeclsMap *Map = LookupPtr;
1593   if (HasLazyLocalLexicalLookups || HasLazyExternalLexicalLookups)
1594     Map = const_cast<DeclContext*>(this)->buildLookup();
1595
1596   if (!Map)
1597     return {};
1598
1599   StoredDeclsMap::iterator I = Map->find(Name);
1600   if (I == Map->end())
1601     return {};
1602
1603   return I->second.getLookupResult();
1604 }
1605
1606 DeclContext::lookup_result
1607 DeclContext::noload_lookup(DeclarationName Name) {
1608   assert(DeclKind != Decl::LinkageSpec && DeclKind != Decl::Export &&
1609          "should not perform lookups into transparent contexts");
1610
1611   DeclContext *PrimaryContext = getPrimaryContext();
1612   if (PrimaryContext != this)
1613     return PrimaryContext->noload_lookup(Name);
1614
1615   loadLazyLocalLexicalLookups();
1616   StoredDeclsMap *Map = LookupPtr;
1617   if (!Map)
1618     return {};
1619
1620   StoredDeclsMap::iterator I = Map->find(Name);
1621   return I != Map->end() ? I->second.getLookupResult()
1622                          : lookup_result();
1623 }
1624
1625 // If we have any lazy lexical declarations not in our lookup map, add them
1626 // now. Don't import any external declarations, not even if we know we have
1627 // some missing from the external visible lookups.
1628 void DeclContext::loadLazyLocalLexicalLookups() {
1629   if (HasLazyLocalLexicalLookups) {
1630     SmallVector<DeclContext *, 2> Contexts;
1631     collectAllContexts(Contexts);
1632     for (auto *Context : Contexts)
1633       buildLookupImpl(Context, hasExternalVisibleStorage());
1634     HasLazyLocalLexicalLookups = false;
1635   }
1636 }
1637
1638 void DeclContext::localUncachedLookup(DeclarationName Name,
1639                                       SmallVectorImpl<NamedDecl *> &Results) {
1640   Results.clear();
1641
1642   // If there's no external storage, just perform a normal lookup and copy
1643   // the results.
1644   if (!hasExternalVisibleStorage() && !hasExternalLexicalStorage() && Name) {
1645     lookup_result LookupResults = lookup(Name);
1646     Results.insert(Results.end(), LookupResults.begin(), LookupResults.end());
1647     return;
1648   }
1649
1650   // If we have a lookup table, check there first. Maybe we'll get lucky.
1651   // FIXME: Should we be checking these flags on the primary context?
1652   if (Name && !HasLazyLocalLexicalLookups && !HasLazyExternalLexicalLookups) {
1653     if (StoredDeclsMap *Map = LookupPtr) {
1654       StoredDeclsMap::iterator Pos = Map->find(Name);
1655       if (Pos != Map->end()) {
1656         Results.insert(Results.end(),
1657                        Pos->second.getLookupResult().begin(),
1658                        Pos->second.getLookupResult().end());
1659         return;
1660       }
1661     }
1662   }
1663
1664   // Slow case: grovel through the declarations in our chain looking for
1665   // matches.
1666   // FIXME: If we have lazy external declarations, this will not find them!
1667   // FIXME: Should we CollectAllContexts and walk them all here?
1668   for (Decl *D = FirstDecl; D; D = D->getNextDeclInContext()) {
1669     if (auto *ND = dyn_cast<NamedDecl>(D))
1670       if (ND->getDeclName() == Name)
1671         Results.push_back(ND);
1672   }
1673 }
1674
1675 DeclContext *DeclContext::getRedeclContext() {
1676   DeclContext *Ctx = this;
1677   // Skip through transparent contexts.
1678   while (Ctx->isTransparentContext())
1679     Ctx = Ctx->getParent();
1680   return Ctx;
1681 }
1682
1683 DeclContext *DeclContext::getEnclosingNamespaceContext() {
1684   DeclContext *Ctx = this;
1685   // Skip through non-namespace, non-translation-unit contexts.
1686   while (!Ctx->isFileContext())
1687     Ctx = Ctx->getParent();
1688   return Ctx->getPrimaryContext();
1689 }
1690
1691 RecordDecl *DeclContext::getOuterLexicalRecordContext() {
1692   // Loop until we find a non-record context.
1693   RecordDecl *OutermostRD = nullptr;
1694   DeclContext *DC = this;
1695   while (DC->isRecord()) {
1696     OutermostRD = cast<RecordDecl>(DC);
1697     DC = DC->getLexicalParent();
1698   }
1699   return OutermostRD;
1700 }
1701
1702 bool DeclContext::InEnclosingNamespaceSetOf(const DeclContext *O) const {
1703   // For non-file contexts, this is equivalent to Equals.
1704   if (!isFileContext())
1705     return O->Equals(this);
1706
1707   do {
1708     if (O->Equals(this))
1709       return true;
1710
1711     const auto *NS = dyn_cast<NamespaceDecl>(O);
1712     if (!NS || !NS->isInline())
1713       break;
1714     O = NS->getParent();
1715   } while (O);
1716
1717   return false;
1718 }
1719
1720 void DeclContext::makeDeclVisibleInContext(NamedDecl *D) {
1721   DeclContext *PrimaryDC = this->getPrimaryContext();
1722   DeclContext *DeclDC = D->getDeclContext()->getPrimaryContext();
1723   // If the decl is being added outside of its semantic decl context, we
1724   // need to ensure that we eagerly build the lookup information for it.
1725   PrimaryDC->makeDeclVisibleInContextWithFlags(D, false, PrimaryDC == DeclDC);
1726 }
1727
1728 void DeclContext::makeDeclVisibleInContextWithFlags(NamedDecl *D, bool Internal,
1729                                                     bool Recoverable) {
1730   assert(this == getPrimaryContext() && "expected a primary DC");
1731
1732   if (!isLookupContext()) {
1733     if (isTransparentContext())
1734       getParent()->getPrimaryContext()
1735         ->makeDeclVisibleInContextWithFlags(D, Internal, Recoverable);
1736     return;
1737   }
1738
1739   // Skip declarations which should be invisible to name lookup.
1740   if (shouldBeHidden(D))
1741     return;
1742
1743   // If we already have a lookup data structure, perform the insertion into
1744   // it. If we might have externally-stored decls with this name, look them
1745   // up and perform the insertion. If this decl was declared outside its
1746   // semantic context, buildLookup won't add it, so add it now.
1747   //
1748   // FIXME: As a performance hack, don't add such decls into the translation
1749   // unit unless we're in C++, since qualified lookup into the TU is never
1750   // performed.
1751   if (LookupPtr || hasExternalVisibleStorage() ||
1752       ((!Recoverable || D->getDeclContext() != D->getLexicalDeclContext()) &&
1753        (getParentASTContext().getLangOpts().CPlusPlus ||
1754         !isTranslationUnit()))) {
1755     // If we have lazily omitted any decls, they might have the same name as
1756     // the decl which we are adding, so build a full lookup table before adding
1757     // this decl.
1758     buildLookup();
1759     makeDeclVisibleInContextImpl(D, Internal);
1760   } else {
1761     HasLazyLocalLexicalLookups = true;
1762   }
1763
1764   // If we are a transparent context or inline namespace, insert into our
1765   // parent context, too. This operation is recursive.
1766   if (isTransparentContext() || isInlineNamespace())
1767     getParent()->getPrimaryContext()->
1768         makeDeclVisibleInContextWithFlags(D, Internal, Recoverable);
1769
1770   auto *DCAsDecl = cast<Decl>(this);
1771   // Notify that a decl was made visible unless we are a Tag being defined.
1772   if (!(isa<TagDecl>(DCAsDecl) && cast<TagDecl>(DCAsDecl)->isBeingDefined()))
1773     if (ASTMutationListener *L = DCAsDecl->getASTMutationListener())
1774       L->AddedVisibleDecl(this, D);
1775 }
1776
1777 void DeclContext::makeDeclVisibleInContextImpl(NamedDecl *D, bool Internal) {
1778   // Find or create the stored declaration map.
1779   StoredDeclsMap *Map = LookupPtr;
1780   if (!Map) {
1781     ASTContext *C = &getParentASTContext();
1782     Map = CreateStoredDeclsMap(*C);
1783   }
1784
1785   // If there is an external AST source, load any declarations it knows about
1786   // with this declaration's name.
1787   // If the lookup table contains an entry about this name it means that we
1788   // have already checked the external source.
1789   if (!Internal)
1790     if (ExternalASTSource *Source = getParentASTContext().getExternalSource())
1791       if (hasExternalVisibleStorage() &&
1792           Map->find(D->getDeclName()) == Map->end())
1793         Source->FindExternalVisibleDeclsByName(this, D->getDeclName());
1794
1795   // Insert this declaration into the map.
1796   StoredDeclsList &DeclNameEntries = (*Map)[D->getDeclName()];
1797
1798   if (Internal) {
1799     // If this is being added as part of loading an external declaration,
1800     // this may not be the only external declaration with this name.
1801     // In this case, we never try to replace an existing declaration; we'll
1802     // handle that when we finalize the list of declarations for this name.
1803     DeclNameEntries.setHasExternalDecls();
1804     DeclNameEntries.AddSubsequentDecl(D);
1805     return;
1806   }
1807
1808   if (DeclNameEntries.isNull()) {
1809     DeclNameEntries.setOnlyValue(D);
1810     return;
1811   }
1812
1813   if (DeclNameEntries.HandleRedeclaration(D, /*IsKnownNewer*/!Internal)) {
1814     // This declaration has replaced an existing one for which
1815     // declarationReplaces returns true.
1816     return;
1817   }
1818
1819   // Put this declaration into the appropriate slot.
1820   DeclNameEntries.AddSubsequentDecl(D);
1821 }
1822
1823 UsingDirectiveDecl *DeclContext::udir_iterator::operator*() const {
1824   return cast<UsingDirectiveDecl>(*I);
1825 }
1826
1827 /// Returns iterator range [First, Last) of UsingDirectiveDecls stored within
1828 /// this context.
1829 DeclContext::udir_range DeclContext::using_directives() const {
1830   // FIXME: Use something more efficient than normal lookup for using
1831   // directives. In C++, using directives are looked up more than anything else.
1832   lookup_result Result = lookup(UsingDirectiveDecl::getName());
1833   return udir_range(Result.begin(), Result.end());
1834 }
1835
1836 //===----------------------------------------------------------------------===//
1837 // Creation and Destruction of StoredDeclsMaps.                               //
1838 //===----------------------------------------------------------------------===//
1839
1840 StoredDeclsMap *DeclContext::CreateStoredDeclsMap(ASTContext &C) const {
1841   assert(!LookupPtr && "context already has a decls map");
1842   assert(getPrimaryContext() == this &&
1843          "creating decls map on non-primary context");
1844
1845   StoredDeclsMap *M;
1846   bool Dependent = isDependentContext();
1847   if (Dependent)
1848     M = new DependentStoredDeclsMap();
1849   else
1850     M = new StoredDeclsMap();
1851   M->Previous = C.LastSDM;
1852   C.LastSDM = llvm::PointerIntPair<StoredDeclsMap*,1>(M, Dependent);
1853   LookupPtr = M;
1854   return M;
1855 }
1856
1857 void ASTContext::ReleaseDeclContextMaps() {
1858   // It's okay to delete DependentStoredDeclsMaps via a StoredDeclsMap
1859   // pointer because the subclass doesn't add anything that needs to
1860   // be deleted.
1861   StoredDeclsMap::DestroyAll(LastSDM.getPointer(), LastSDM.getInt());
1862 }
1863
1864 void StoredDeclsMap::DestroyAll(StoredDeclsMap *Map, bool Dependent) {
1865   while (Map) {
1866     // Advance the iteration before we invalidate memory.
1867     llvm::PointerIntPair<StoredDeclsMap*,1> Next = Map->Previous;
1868
1869     if (Dependent)
1870       delete static_cast<DependentStoredDeclsMap*>(Map);
1871     else
1872       delete Map;
1873
1874     Map = Next.getPointer();
1875     Dependent = Next.getInt();
1876   }
1877 }
1878
1879 DependentDiagnostic *DependentDiagnostic::Create(ASTContext &C,
1880                                                  DeclContext *Parent,
1881                                            const PartialDiagnostic &PDiag) {
1882   assert(Parent->isDependentContext()
1883          && "cannot iterate dependent diagnostics of non-dependent context");
1884   Parent = Parent->getPrimaryContext();
1885   if (!Parent->LookupPtr)
1886     Parent->CreateStoredDeclsMap(C);
1887
1888   auto *Map = static_cast<DependentStoredDeclsMap *>(Parent->LookupPtr);
1889
1890   // Allocate the copy of the PartialDiagnostic via the ASTContext's
1891   // BumpPtrAllocator, rather than the ASTContext itself.
1892   PartialDiagnostic::Storage *DiagStorage = nullptr;
1893   if (PDiag.hasStorage())
1894     DiagStorage = new (C) PartialDiagnostic::Storage;
1895
1896   auto *DD = new (C) DependentDiagnostic(PDiag, DiagStorage);
1897
1898   // TODO: Maybe we shouldn't reverse the order during insertion.
1899   DD->NextDiagnostic = Map->FirstDiagnostic;
1900   Map->FirstDiagnostic = DD;
1901
1902   return DD;
1903 }