• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===- DeclBase.cpp - Declaration AST Node Implementation -----------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements the Decl and DeclContext classes.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "clang/AST/DeclBase.h"
14 #include "clang/AST/ASTContext.h"
15 #include "clang/AST/ASTLambda.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 
updateOutOfDate(IdentifierInfo & II) const63 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 
operator new(std::size_t Size,const ASTContext & Context,unsigned ID,std::size_t Extra)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 
operator new(std::size_t Size,const ASTContext & Ctx,DeclContext * Parent,std::size_t Extra)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 *), llvm::Align(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 
getOwningModuleSlow() const114 Module *Decl::getOwningModuleSlow() const {
115   assert(isFromASTFile() && "Not from AST file?");
116   return getASTContext().getExternalSource()->getModule(getOwningModuleID());
117 }
118 
hasLocalOwningModuleStorage() const119 bool Decl::hasLocalOwningModuleStorage() const {
120   return getASTContext().getLangOpts().trackLocalOwningModule();
121 }
122 
getDeclKindName() const123 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 
setInvalidDecl(bool Invalid)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 
getDeclKindName() const155 const char *DeclContext::getDeclKindName() const {
156   switch (getDeclKind()) {
157 #define DECL(DERIVED, BASE) case Decl::DERIVED: return #DERIVED;
158 #define ABSTRACT_DECL(DECL)
159 #include "clang/AST/DeclNodes.inc"
160   }
161   llvm_unreachable("Declaration context not in DeclNodes.inc!");
162 }
163 
164 bool Decl::StatisticsEnabled = false;
EnableStatistics()165 void Decl::EnableStatistics() {
166   StatisticsEnabled = true;
167 }
168 
PrintStats()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 
add(Kind k)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 
isTemplateParameterPack() const201 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 
isParameterPack() const211 bool Decl::isParameterPack() const {
212   if (const auto *Var = dyn_cast<VarDecl>(this))
213     return Var->isParameterPack();
214 
215   return isTemplateParameterPack();
216 }
217 
getAsFunction()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 
isTemplateDecl() const226 bool Decl::isTemplateDecl() const {
227   return isa<TemplateDecl>(this);
228 }
229 
getDescribedTemplate() const230 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 
getDescribedTemplateParams() const243 const TemplateParameterList *Decl::getDescribedTemplateParams() const {
244   if (auto *TD = getDescribedTemplate())
245     return TD->getTemplateParameters();
246   if (auto *CTPSD = dyn_cast<ClassTemplatePartialSpecializationDecl>(this))
247     return CTPSD->getTemplateParameters();
248   if (auto *VTPSD = dyn_cast<VarTemplatePartialSpecializationDecl>(this))
249     return VTPSD->getTemplateParameters();
250   return nullptr;
251 }
252 
isTemplated() const253 bool Decl::isTemplated() const {
254   // A declaration is templated if it is a template or a template pattern, or
255   // is within (lexcially for a friend, semantically otherwise) a dependent
256   // context.
257   // FIXME: Should local extern declarations be treated like friends?
258   if (auto *AsDC = dyn_cast<DeclContext>(this))
259     return AsDC->isDependentContext();
260   auto *DC = getFriendObjectKind() ? getLexicalDeclContext() : getDeclContext();
261   return DC->isDependentContext() || isTemplateDecl() ||
262          getDescribedTemplateParams();
263 }
264 
getTemplateDepth() const265 unsigned Decl::getTemplateDepth() const {
266   if (auto *DC = dyn_cast<DeclContext>(this))
267     if (DC->isFileContext())
268       return 0;
269 
270   if (auto *TPL = getDescribedTemplateParams())
271     return TPL->getDepth() + 1;
272 
273   // If this is a dependent lambda, there might be an enclosing variable
274   // template. In this case, the next step is not the parent DeclContext (or
275   // even a DeclContext at all).
276   auto *RD = dyn_cast<CXXRecordDecl>(this);
277   if (RD && RD->isDependentLambda())
278     if (Decl *Context = RD->getLambdaContextDecl())
279       return Context->getTemplateDepth();
280 
281   const DeclContext *DC =
282       getFriendObjectKind() ? getLexicalDeclContext() : getDeclContext();
283   return cast<Decl>(DC)->getTemplateDepth();
284 }
285 
getParentFunctionOrMethod() const286 const DeclContext *Decl::getParentFunctionOrMethod() const {
287   for (const DeclContext *DC = getDeclContext();
288        DC && !DC->isTranslationUnit() && !DC->isNamespace();
289        DC = DC->getParent())
290     if (DC->isFunctionOrMethod())
291       return DC;
292 
293   return nullptr;
294 }
295 
296 //===----------------------------------------------------------------------===//
297 // PrettyStackTraceDecl Implementation
298 //===----------------------------------------------------------------------===//
299 
print(raw_ostream & OS) const300 void PrettyStackTraceDecl::print(raw_ostream &OS) const {
301   SourceLocation TheLoc = Loc;
302   if (TheLoc.isInvalid() && TheDecl)
303     TheLoc = TheDecl->getLocation();
304 
305   if (TheLoc.isValid()) {
306     TheLoc.print(OS, SM);
307     OS << ": ";
308   }
309 
310   OS << Message;
311 
312   if (const auto *DN = dyn_cast_or_null<NamedDecl>(TheDecl)) {
313     OS << " '";
314     DN->printQualifiedName(OS);
315     OS << '\'';
316   }
317   OS << '\n';
318 }
319 
320 //===----------------------------------------------------------------------===//
321 // Decl Implementation
322 //===----------------------------------------------------------------------===//
323 
324 // Out-of-line virtual method providing a home for Decl.
325 Decl::~Decl() = default;
326 
setDeclContext(DeclContext * DC)327 void Decl::setDeclContext(DeclContext *DC) {
328   DeclCtx = DC;
329 }
330 
setLexicalDeclContext(DeclContext * DC)331 void Decl::setLexicalDeclContext(DeclContext *DC) {
332   if (DC == getLexicalDeclContext())
333     return;
334 
335   if (isInSemaDC()) {
336     setDeclContextsImpl(getDeclContext(), DC, getASTContext());
337   } else {
338     getMultipleDC()->LexicalDC = DC;
339   }
340 
341   // FIXME: We shouldn't be changing the lexical context of declarations
342   // imported from AST files.
343   if (!isFromASTFile()) {
344     setModuleOwnershipKind(getModuleOwnershipKindForChildOf(DC));
345     if (hasOwningModule())
346       setLocalOwningModule(cast<Decl>(DC)->getOwningModule());
347   }
348 
349   assert(
350       (getModuleOwnershipKind() != ModuleOwnershipKind::VisibleWhenImported ||
351        getOwningModule()) &&
352       "hidden declaration has no owning module");
353 }
354 
setDeclContextsImpl(DeclContext * SemaDC,DeclContext * LexicalDC,ASTContext & Ctx)355 void Decl::setDeclContextsImpl(DeclContext *SemaDC, DeclContext *LexicalDC,
356                                ASTContext &Ctx) {
357   if (SemaDC == LexicalDC) {
358     DeclCtx = SemaDC;
359   } else {
360     auto *MDC = new (Ctx) Decl::MultipleDC();
361     MDC->SemanticDC = SemaDC;
362     MDC->LexicalDC = LexicalDC;
363     DeclCtx = MDC;
364   }
365 }
366 
isInLocalScopeForInstantiation() const367 bool Decl::isInLocalScopeForInstantiation() const {
368   const DeclContext *LDC = getLexicalDeclContext();
369   if (!LDC->isDependentContext())
370     return false;
371   while (true) {
372     if (LDC->isFunctionOrMethod())
373       return true;
374     if (!isa<TagDecl>(LDC))
375       return false;
376     if (const auto *CRD = dyn_cast<CXXRecordDecl>(LDC))
377       if (CRD->isLambda())
378         return true;
379     LDC = LDC->getLexicalParent();
380   }
381   return false;
382 }
383 
isInAnonymousNamespace() const384 bool Decl::isInAnonymousNamespace() const {
385   for (const DeclContext *DC = getDeclContext(); DC; DC = DC->getParent()) {
386     if (const auto *ND = dyn_cast<NamespaceDecl>(DC))
387       if (ND->isAnonymousNamespace())
388         return true;
389   }
390 
391   return false;
392 }
393 
isInStdNamespace() const394 bool Decl::isInStdNamespace() const {
395   const DeclContext *DC = getDeclContext();
396   return DC && DC->isStdNamespace();
397 }
398 
getTranslationUnitDecl()399 TranslationUnitDecl *Decl::getTranslationUnitDecl() {
400   if (auto *TUD = dyn_cast<TranslationUnitDecl>(this))
401     return TUD;
402 
403   DeclContext *DC = getDeclContext();
404   assert(DC && "This decl is not contained in a translation unit!");
405 
406   while (!DC->isTranslationUnit()) {
407     DC = DC->getParent();
408     assert(DC && "This decl is not contained in a translation unit!");
409   }
410 
411   return cast<TranslationUnitDecl>(DC);
412 }
413 
getASTContext() const414 ASTContext &Decl::getASTContext() const {
415   return getTranslationUnitDecl()->getASTContext();
416 }
417 
418 /// Helper to get the language options from the ASTContext.
419 /// Defined out of line to avoid depending on ASTContext.h.
getLangOpts() const420 const LangOptions &Decl::getLangOpts() const {
421   return getASTContext().getLangOpts();
422 }
423 
getASTMutationListener() const424 ASTMutationListener *Decl::getASTMutationListener() const {
425   return getASTContext().getASTMutationListener();
426 }
427 
getMaxAlignment() const428 unsigned Decl::getMaxAlignment() const {
429   if (!hasAttrs())
430     return 0;
431 
432   unsigned Align = 0;
433   const AttrVec &V = getAttrs();
434   ASTContext &Ctx = getASTContext();
435   specific_attr_iterator<AlignedAttr> I(V.begin()), E(V.end());
436   for (; I != E; ++I) {
437     if (!I->isAlignmentErrorDependent())
438       Align = std::max(Align, I->getAlignment(Ctx));
439   }
440   return Align;
441 }
442 
isUsed(bool CheckUsedAttr) const443 bool Decl::isUsed(bool CheckUsedAttr) const {
444   const Decl *CanonD = getCanonicalDecl();
445   if (CanonD->Used)
446     return true;
447 
448   // Check for used attribute.
449   // Ask the most recent decl, since attributes accumulate in the redecl chain.
450   if (CheckUsedAttr && getMostRecentDecl()->hasAttr<UsedAttr>())
451     return true;
452 
453   // The information may have not been deserialized yet. Force deserialization
454   // to complete the needed information.
455   return getMostRecentDecl()->getCanonicalDecl()->Used;
456 }
457 
markUsed(ASTContext & C)458 void Decl::markUsed(ASTContext &C) {
459   if (isUsed(false))
460     return;
461 
462   if (C.getASTMutationListener())
463     C.getASTMutationListener()->DeclarationMarkedUsed(this);
464 
465   setIsUsed();
466 }
467 
isReferenced() const468 bool Decl::isReferenced() const {
469   if (Referenced)
470     return true;
471 
472   // Check redeclarations.
473   for (const auto *I : redecls())
474     if (I->Referenced)
475       return true;
476 
477   return false;
478 }
479 
getExternalSourceSymbolAttr() const480 ExternalSourceSymbolAttr *Decl::getExternalSourceSymbolAttr() const {
481   const Decl *Definition = nullptr;
482   if (auto *ID = dyn_cast<ObjCInterfaceDecl>(this)) {
483     Definition = ID->getDefinition();
484   } else if (auto *PD = dyn_cast<ObjCProtocolDecl>(this)) {
485     Definition = PD->getDefinition();
486   } else if (auto *TD = dyn_cast<TagDecl>(this)) {
487     Definition = TD->getDefinition();
488   }
489   if (!Definition)
490     Definition = this;
491 
492   if (auto *attr = Definition->getAttr<ExternalSourceSymbolAttr>())
493     return attr;
494   if (auto *dcd = dyn_cast<Decl>(getDeclContext())) {
495     return dcd->getAttr<ExternalSourceSymbolAttr>();
496   }
497 
498   return nullptr;
499 }
500 
hasDefiningAttr() const501 bool Decl::hasDefiningAttr() const {
502   return hasAttr<AliasAttr>() || hasAttr<IFuncAttr>() ||
503          hasAttr<LoaderUninitializedAttr>();
504 }
505 
getDefiningAttr() const506 const Attr *Decl::getDefiningAttr() const {
507   if (auto *AA = getAttr<AliasAttr>())
508     return AA;
509   if (auto *IFA = getAttr<IFuncAttr>())
510     return IFA;
511   if (auto *NZA = getAttr<LoaderUninitializedAttr>())
512     return NZA;
513   return nullptr;
514 }
515 
getRealizedPlatform(const AvailabilityAttr * A,const ASTContext & Context)516 static StringRef getRealizedPlatform(const AvailabilityAttr *A,
517                                      const ASTContext &Context) {
518   // Check if this is an App Extension "platform", and if so chop off
519   // the suffix for matching with the actual platform.
520   StringRef RealizedPlatform = A->getPlatform()->getName();
521   if (!Context.getLangOpts().AppExt)
522     return RealizedPlatform;
523   size_t suffix = RealizedPlatform.rfind("_app_extension");
524   if (suffix != StringRef::npos)
525     return RealizedPlatform.slice(0, suffix);
526   return RealizedPlatform;
527 }
528 
529 /// Determine the availability of the given declaration based on
530 /// the target platform.
531 ///
532 /// When it returns an availability result other than \c AR_Available,
533 /// if the \p Message parameter is non-NULL, it will be set to a
534 /// string describing why the entity is unavailable.
535 ///
536 /// FIXME: Make these strings localizable, since they end up in
537 /// diagnostics.
CheckAvailability(ASTContext & Context,const AvailabilityAttr * A,std::string * Message,VersionTuple EnclosingVersion)538 static AvailabilityResult CheckAvailability(ASTContext &Context,
539                                             const AvailabilityAttr *A,
540                                             std::string *Message,
541                                             VersionTuple EnclosingVersion) {
542   if (EnclosingVersion.empty())
543     EnclosingVersion = Context.getTargetInfo().getPlatformMinVersion();
544 
545   if (EnclosingVersion.empty())
546     return AR_Available;
547 
548   StringRef ActualPlatform = A->getPlatform()->getName();
549   StringRef TargetPlatform = Context.getTargetInfo().getPlatformName();
550 
551   // Match the platform name.
552   if (getRealizedPlatform(A, Context) != TargetPlatform)
553     return AR_Available;
554 
555   StringRef PrettyPlatformName
556     = AvailabilityAttr::getPrettyPlatformName(ActualPlatform);
557 
558   if (PrettyPlatformName.empty())
559     PrettyPlatformName = ActualPlatform;
560 
561   std::string HintMessage;
562   if (!A->getMessage().empty()) {
563     HintMessage = " - ";
564     HintMessage += A->getMessage();
565   }
566 
567   // Make sure that this declaration has not been marked 'unavailable'.
568   if (A->getUnavailable()) {
569     if (Message) {
570       Message->clear();
571       llvm::raw_string_ostream Out(*Message);
572       Out << "not available on " << PrettyPlatformName
573           << HintMessage;
574     }
575 
576     return AR_Unavailable;
577   }
578 
579   // Make sure that this declaration has already been introduced.
580   if (!A->getIntroduced().empty() &&
581       EnclosingVersion < A->getIntroduced()) {
582     if (Message) {
583       Message->clear();
584       llvm::raw_string_ostream Out(*Message);
585       VersionTuple VTI(A->getIntroduced());
586       Out << "introduced in " << PrettyPlatformName << ' '
587           << VTI << HintMessage;
588     }
589 
590     return A->getStrict() ? AR_Unavailable : AR_NotYetIntroduced;
591   }
592 
593   // Make sure that this declaration hasn't been obsoleted.
594   if (!A->getObsoleted().empty() && EnclosingVersion >= A->getObsoleted()) {
595     if (Message) {
596       Message->clear();
597       llvm::raw_string_ostream Out(*Message);
598       VersionTuple VTO(A->getObsoleted());
599       Out << "obsoleted in " << PrettyPlatformName << ' '
600           << VTO << HintMessage;
601     }
602 
603     return AR_Unavailable;
604   }
605 
606   // Make sure that this declaration hasn't been deprecated.
607   if (!A->getDeprecated().empty() && EnclosingVersion >= A->getDeprecated()) {
608     if (Message) {
609       Message->clear();
610       llvm::raw_string_ostream Out(*Message);
611       VersionTuple VTD(A->getDeprecated());
612       Out << "first deprecated in " << PrettyPlatformName << ' '
613           << VTD << HintMessage;
614     }
615 
616     return AR_Deprecated;
617   }
618 
619   return AR_Available;
620 }
621 
getAvailability(std::string * Message,VersionTuple EnclosingVersion,StringRef * RealizedPlatform) const622 AvailabilityResult Decl::getAvailability(std::string *Message,
623                                          VersionTuple EnclosingVersion,
624                                          StringRef *RealizedPlatform) const {
625   if (auto *FTD = dyn_cast<FunctionTemplateDecl>(this))
626     return FTD->getTemplatedDecl()->getAvailability(Message, EnclosingVersion,
627                                                     RealizedPlatform);
628 
629   AvailabilityResult Result = AR_Available;
630   std::string ResultMessage;
631 
632   for (const auto *A : attrs()) {
633     if (const auto *Deprecated = dyn_cast<DeprecatedAttr>(A)) {
634       if (Result >= AR_Deprecated)
635         continue;
636 
637       if (Message)
638         ResultMessage = std::string(Deprecated->getMessage());
639 
640       Result = AR_Deprecated;
641       continue;
642     }
643 
644     if (const auto *Unavailable = dyn_cast<UnavailableAttr>(A)) {
645       if (Message)
646         *Message = std::string(Unavailable->getMessage());
647       return AR_Unavailable;
648     }
649 
650     if (const auto *Availability = dyn_cast<AvailabilityAttr>(A)) {
651       AvailabilityResult AR = CheckAvailability(getASTContext(), Availability,
652                                                 Message, EnclosingVersion);
653 
654       if (AR == AR_Unavailable) {
655         if (RealizedPlatform)
656           *RealizedPlatform = Availability->getPlatform()->getName();
657         return AR_Unavailable;
658       }
659 
660       if (AR > Result) {
661         Result = AR;
662         if (Message)
663           ResultMessage.swap(*Message);
664       }
665       continue;
666     }
667   }
668 
669   if (Message)
670     Message->swap(ResultMessage);
671   return Result;
672 }
673 
getVersionIntroduced() const674 VersionTuple Decl::getVersionIntroduced() const {
675   const ASTContext &Context = getASTContext();
676   StringRef TargetPlatform = Context.getTargetInfo().getPlatformName();
677   for (const auto *A : attrs()) {
678     if (const auto *Availability = dyn_cast<AvailabilityAttr>(A)) {
679       if (getRealizedPlatform(Availability, Context) != TargetPlatform)
680         continue;
681       if (!Availability->getIntroduced().empty())
682         return Availability->getIntroduced();
683     }
684   }
685   return {};
686 }
687 
canBeWeakImported(bool & IsDefinition) const688 bool Decl::canBeWeakImported(bool &IsDefinition) const {
689   IsDefinition = false;
690 
691   // Variables, if they aren't definitions.
692   if (const auto *Var = dyn_cast<VarDecl>(this)) {
693     if (Var->isThisDeclarationADefinition()) {
694       IsDefinition = true;
695       return false;
696     }
697     return true;
698 
699   // Functions, if they aren't definitions.
700   } else if (const auto *FD = dyn_cast<FunctionDecl>(this)) {
701     if (FD->hasBody()) {
702       IsDefinition = true;
703       return false;
704     }
705     return true;
706 
707   // Objective-C classes, if this is the non-fragile runtime.
708   } else if (isa<ObjCInterfaceDecl>(this) &&
709              getASTContext().getLangOpts().ObjCRuntime.hasWeakClassImport()) {
710     return true;
711 
712   // Nothing else.
713   } else {
714     return false;
715   }
716 }
717 
isWeakImported() const718 bool Decl::isWeakImported() const {
719   bool IsDefinition;
720   if (!canBeWeakImported(IsDefinition))
721     return false;
722 
723   for (const auto *A : getMostRecentDecl()->attrs()) {
724     if (isa<WeakImportAttr>(A))
725       return true;
726 
727     if (const auto *Availability = dyn_cast<AvailabilityAttr>(A)) {
728       if (CheckAvailability(getASTContext(), Availability, nullptr,
729                             VersionTuple()) == AR_NotYetIntroduced)
730         return true;
731     }
732   }
733 
734   return false;
735 }
736 
getIdentifierNamespaceForKind(Kind DeclKind)737 unsigned Decl::getIdentifierNamespaceForKind(Kind DeclKind) {
738   switch (DeclKind) {
739     case Function:
740     case CXXDeductionGuide:
741     case CXXMethod:
742     case CXXConstructor:
743     case ConstructorUsingShadow:
744     case CXXDestructor:
745     case CXXConversion:
746     case EnumConstant:
747     case Var:
748     case ImplicitParam:
749     case ParmVar:
750     case ObjCMethod:
751     case ObjCProperty:
752     case MSProperty:
753       return IDNS_Ordinary;
754     case Label:
755       return IDNS_Label;
756     case IndirectField:
757       return IDNS_Ordinary | IDNS_Member;
758 
759     case Binding:
760     case NonTypeTemplateParm:
761     case VarTemplate:
762     case Concept:
763       // These (C++-only) declarations are found by redeclaration lookup for
764       // tag types, so we include them in the tag namespace.
765       return IDNS_Ordinary | IDNS_Tag;
766 
767     case ObjCCompatibleAlias:
768     case ObjCInterface:
769       return IDNS_Ordinary | IDNS_Type;
770 
771     case Typedef:
772     case TypeAlias:
773     case TemplateTypeParm:
774     case ObjCTypeParam:
775       return IDNS_Ordinary | IDNS_Type;
776 
777     case UnresolvedUsingTypename:
778       return IDNS_Ordinary | IDNS_Type | IDNS_Using;
779 
780     case UsingShadow:
781       return 0; // we'll actually overwrite this later
782 
783     case UnresolvedUsingValue:
784       return IDNS_Ordinary | IDNS_Using;
785 
786     case Using:
787     case UsingPack:
788       return IDNS_Using;
789 
790     case ObjCProtocol:
791       return IDNS_ObjCProtocol;
792 
793     case Field:
794     case ObjCAtDefsField:
795     case ObjCIvar:
796       return IDNS_Member;
797 
798     case Record:
799     case CXXRecord:
800     case Enum:
801       return IDNS_Tag | IDNS_Type;
802 
803     case Namespace:
804     case NamespaceAlias:
805       return IDNS_Namespace;
806 
807     case FunctionTemplate:
808       return IDNS_Ordinary;
809 
810     case ClassTemplate:
811     case TemplateTemplateParm:
812     case TypeAliasTemplate:
813       return IDNS_Ordinary | IDNS_Tag | IDNS_Type;
814 
815     case OMPDeclareReduction:
816       return IDNS_OMPReduction;
817 
818     case OMPDeclareMapper:
819       return IDNS_OMPMapper;
820 
821     // Never have names.
822     case Friend:
823     case FriendTemplate:
824     case AccessSpec:
825     case LinkageSpec:
826     case Export:
827     case FileScopeAsm:
828     case StaticAssert:
829     case ObjCPropertyImpl:
830     case PragmaComment:
831     case PragmaDetectMismatch:
832     case Block:
833     case Captured:
834     case TranslationUnit:
835     case ExternCContext:
836     case Decomposition:
837     case MSGuid:
838     case TemplateParamObject:
839 
840     case UsingDirective:
841     case BuiltinTemplate:
842     case ClassTemplateSpecialization:
843     case ClassTemplatePartialSpecialization:
844     case ClassScopeFunctionSpecialization:
845     case VarTemplateSpecialization:
846     case VarTemplatePartialSpecialization:
847     case ObjCImplementation:
848     case ObjCCategory:
849     case ObjCCategoryImpl:
850     case Import:
851     case OMPThreadPrivate:
852     case OMPAllocate:
853     case OMPRequires:
854     case OMPCapturedExpr:
855     case Empty:
856     case LifetimeExtendedTemporary:
857     case RequiresExprBody:
858       // Never looked up by name.
859       return 0;
860   }
861 
862   llvm_unreachable("Invalid DeclKind!");
863 }
864 
setAttrsImpl(const AttrVec & attrs,ASTContext & Ctx)865 void Decl::setAttrsImpl(const AttrVec &attrs, ASTContext &Ctx) {
866   assert(!HasAttrs && "Decl already contains attrs.");
867 
868   AttrVec &AttrBlank = Ctx.getDeclAttrs(this);
869   assert(AttrBlank.empty() && "HasAttrs was wrong?");
870 
871   AttrBlank = attrs;
872   HasAttrs = true;
873 }
874 
dropAttrs()875 void Decl::dropAttrs() {
876   if (!HasAttrs) return;
877 
878   HasAttrs = false;
879   getASTContext().eraseDeclAttrs(this);
880 }
881 
addAttr(Attr * A)882 void Decl::addAttr(Attr *A) {
883   if (!hasAttrs()) {
884     setAttrs(AttrVec(1, A));
885     return;
886   }
887 
888   AttrVec &Attrs = getAttrs();
889   if (!A->isInherited()) {
890     Attrs.push_back(A);
891     return;
892   }
893 
894   // Attribute inheritance is processed after attribute parsing. To keep the
895   // order as in the source code, add inherited attributes before non-inherited
896   // ones.
897   auto I = Attrs.begin(), E = Attrs.end();
898   for (; I != E; ++I) {
899     if (!(*I)->isInherited())
900       break;
901   }
902   Attrs.insert(I, A);
903 }
904 
getAttrs() const905 const AttrVec &Decl::getAttrs() const {
906   assert(HasAttrs && "No attrs to get!");
907   return getASTContext().getDeclAttrs(this);
908 }
909 
castFromDeclContext(const DeclContext * D)910 Decl *Decl::castFromDeclContext (const DeclContext *D) {
911   Decl::Kind DK = D->getDeclKind();
912   switch(DK) {
913 #define DECL(NAME, BASE)
914 #define DECL_CONTEXT(NAME) \
915     case Decl::NAME:       \
916       return static_cast<NAME##Decl *>(const_cast<DeclContext *>(D));
917 #define DECL_CONTEXT_BASE(NAME)
918 #include "clang/AST/DeclNodes.inc"
919     default:
920 #define DECL(NAME, BASE)
921 #define DECL_CONTEXT_BASE(NAME)                  \
922       if (DK >= first##NAME && DK <= last##NAME) \
923         return static_cast<NAME##Decl *>(const_cast<DeclContext *>(D));
924 #include "clang/AST/DeclNodes.inc"
925       llvm_unreachable("a decl that inherits DeclContext isn't handled");
926   }
927 }
928 
castToDeclContext(const Decl * D)929 DeclContext *Decl::castToDeclContext(const Decl *D) {
930   Decl::Kind DK = D->getKind();
931   switch(DK) {
932 #define DECL(NAME, BASE)
933 #define DECL_CONTEXT(NAME) \
934     case Decl::NAME:       \
935       return static_cast<NAME##Decl *>(const_cast<Decl *>(D));
936 #define DECL_CONTEXT_BASE(NAME)
937 #include "clang/AST/DeclNodes.inc"
938     default:
939 #define DECL(NAME, BASE)
940 #define DECL_CONTEXT_BASE(NAME)                                   \
941       if (DK >= first##NAME && DK <= last##NAME)                  \
942         return static_cast<NAME##Decl *>(const_cast<Decl *>(D));
943 #include "clang/AST/DeclNodes.inc"
944       llvm_unreachable("a decl that inherits DeclContext isn't handled");
945   }
946 }
947 
getBodyRBrace() const948 SourceLocation Decl::getBodyRBrace() const {
949   // Special handling of FunctionDecl to avoid de-serializing the body from PCH.
950   // FunctionDecl stores EndRangeLoc for this purpose.
951   if (const auto *FD = dyn_cast<FunctionDecl>(this)) {
952     const FunctionDecl *Definition;
953     if (FD->hasBody(Definition))
954       return Definition->getSourceRange().getEnd();
955     return {};
956   }
957 
958   if (Stmt *Body = getBody())
959     return Body->getSourceRange().getEnd();
960 
961   return {};
962 }
963 
AccessDeclContextSanity() const964 bool Decl::AccessDeclContextSanity() const {
965 #ifndef NDEBUG
966   // Suppress this check if any of the following hold:
967   // 1. this is the translation unit (and thus has no parent)
968   // 2. this is a template parameter (and thus doesn't belong to its context)
969   // 3. this is a non-type template parameter
970   // 4. the context is not a record
971   // 5. it's invalid
972   // 6. it's a C++0x static_assert.
973   // 7. it's a block literal declaration
974   if (isa<TranslationUnitDecl>(this) ||
975       isa<TemplateTypeParmDecl>(this) ||
976       isa<NonTypeTemplateParmDecl>(this) ||
977       !getDeclContext() ||
978       !isa<CXXRecordDecl>(getDeclContext()) ||
979       isInvalidDecl() ||
980       isa<StaticAssertDecl>(this) ||
981       isa<BlockDecl>(this) ||
982       // FIXME: a ParmVarDecl can have ClassTemplateSpecialization
983       // as DeclContext (?).
984       isa<ParmVarDecl>(this) ||
985       // FIXME: a ClassTemplateSpecialization or CXXRecordDecl can have
986       // AS_none as access specifier.
987       isa<CXXRecordDecl>(this) ||
988       isa<ClassScopeFunctionSpecializationDecl>(this))
989     return true;
990 
991   assert(Access != AS_none &&
992          "Access specifier is AS_none inside a record decl");
993 #endif
994   return true;
995 }
996 
getKind(const Decl * D)997 static Decl::Kind getKind(const Decl *D) { return D->getKind(); }
getKind(const DeclContext * DC)998 static Decl::Kind getKind(const DeclContext *DC) { return DC->getDeclKind(); }
999 
getID() const1000 int64_t Decl::getID() const {
1001   return getASTContext().getAllocator().identifyKnownAlignedObject<Decl>(this);
1002 }
1003 
getFunctionType(bool BlocksToo) const1004 const FunctionType *Decl::getFunctionType(bool BlocksToo) const {
1005   QualType Ty;
1006   if (const auto *D = dyn_cast<ValueDecl>(this))
1007     Ty = D->getType();
1008   else if (const auto *D = dyn_cast<TypedefNameDecl>(this))
1009     Ty = D->getUnderlyingType();
1010   else
1011     return nullptr;
1012 
1013   if (Ty->isFunctionPointerType())
1014     Ty = Ty->castAs<PointerType>()->getPointeeType();
1015   else if (Ty->isFunctionReferenceType())
1016     Ty = Ty->castAs<ReferenceType>()->getPointeeType();
1017   else if (BlocksToo && Ty->isBlockPointerType())
1018     Ty = Ty->castAs<BlockPointerType>()->getPointeeType();
1019 
1020   return Ty->getAs<FunctionType>();
1021 }
1022 
1023 /// Starting at a given context (a Decl or DeclContext), look for a
1024 /// code context that is not a closure (a lambda, block, etc.).
getNonClosureContext(T * D)1025 template <class T> static Decl *getNonClosureContext(T *D) {
1026   if (getKind(D) == Decl::CXXMethod) {
1027     auto *MD = cast<CXXMethodDecl>(D);
1028     if (MD->getOverloadedOperator() == OO_Call &&
1029         MD->getParent()->isLambda())
1030       return getNonClosureContext(MD->getParent()->getParent());
1031     return MD;
1032   } else if (auto *FD = dyn_cast<FunctionDecl>(D))
1033     return FD;
1034   else if (auto *MD = dyn_cast<ObjCMethodDecl>(D))
1035     return MD;
1036   else if (auto *BD = dyn_cast<BlockDecl>(D))
1037     return getNonClosureContext(BD->getParent());
1038   else if (auto *CD = dyn_cast<CapturedDecl>(D))
1039     return getNonClosureContext(CD->getParent());
1040   else
1041     return nullptr;
1042 }
1043 
getNonClosureContext()1044 Decl *Decl::getNonClosureContext() {
1045   return ::getNonClosureContext(this);
1046 }
1047 
getNonClosureAncestor()1048 Decl *DeclContext::getNonClosureAncestor() {
1049   return ::getNonClosureContext(this);
1050 }
1051 
1052 //===----------------------------------------------------------------------===//
1053 // DeclContext Implementation
1054 //===----------------------------------------------------------------------===//
1055 
DeclContext(Decl::Kind K)1056 DeclContext::DeclContext(Decl::Kind K) {
1057   DeclContextBits.DeclKind = K;
1058   setHasExternalLexicalStorage(false);
1059   setHasExternalVisibleStorage(false);
1060   setNeedToReconcileExternalVisibleStorage(false);
1061   setHasLazyLocalLexicalLookups(false);
1062   setHasLazyExternalLexicalLookups(false);
1063   setUseQualifiedLookup(false);
1064 }
1065 
classof(const Decl * D)1066 bool DeclContext::classof(const Decl *D) {
1067   switch (D->getKind()) {
1068 #define DECL(NAME, BASE)
1069 #define DECL_CONTEXT(NAME) case Decl::NAME:
1070 #define DECL_CONTEXT_BASE(NAME)
1071 #include "clang/AST/DeclNodes.inc"
1072       return true;
1073     default:
1074 #define DECL(NAME, BASE)
1075 #define DECL_CONTEXT_BASE(NAME)                 \
1076       if (D->getKind() >= Decl::first##NAME &&  \
1077           D->getKind() <= Decl::last##NAME)     \
1078         return true;
1079 #include "clang/AST/DeclNodes.inc"
1080       return false;
1081   }
1082 }
1083 
1084 DeclContext::~DeclContext() = default;
1085 
1086 /// Find the parent context of this context that will be
1087 /// used for unqualified name lookup.
1088 ///
1089 /// Generally, the parent lookup context is the semantic context. However, for
1090 /// a friend function the parent lookup context is the lexical context, which
1091 /// is the class in which the friend is declared.
getLookupParent()1092 DeclContext *DeclContext::getLookupParent() {
1093   // FIXME: Find a better way to identify friends.
1094   if (isa<FunctionDecl>(this))
1095     if (getParent()->getRedeclContext()->isFileContext() &&
1096         getLexicalParent()->getRedeclContext()->isRecord())
1097       return getLexicalParent();
1098 
1099   // A lookup within the call operator of a lambda never looks in the lambda
1100   // class; instead, skip to the context in which that closure type is
1101   // declared.
1102   if (isLambdaCallOperator(this))
1103     return getParent()->getParent();
1104 
1105   return getParent();
1106 }
1107 
getInnermostBlockDecl() const1108 const BlockDecl *DeclContext::getInnermostBlockDecl() const {
1109   const DeclContext *Ctx = this;
1110 
1111   do {
1112     if (Ctx->isClosure())
1113       return cast<BlockDecl>(Ctx);
1114     Ctx = Ctx->getParent();
1115   } while (Ctx);
1116 
1117   return nullptr;
1118 }
1119 
isInlineNamespace() const1120 bool DeclContext::isInlineNamespace() const {
1121   return isNamespace() &&
1122          cast<NamespaceDecl>(this)->isInline();
1123 }
1124 
isStdNamespace() const1125 bool DeclContext::isStdNamespace() const {
1126   if (!isNamespace())
1127     return false;
1128 
1129   const auto *ND = cast<NamespaceDecl>(this);
1130   if (ND->isInline()) {
1131     return ND->getParent()->isStdNamespace();
1132   }
1133 
1134   if (!getParent()->getRedeclContext()->isTranslationUnit())
1135     return false;
1136 
1137   const IdentifierInfo *II = ND->getIdentifier();
1138   return II && II->isStr("std");
1139 }
1140 
isDependentContext() const1141 bool DeclContext::isDependentContext() const {
1142   if (isFileContext())
1143     return false;
1144 
1145   if (isa<ClassTemplatePartialSpecializationDecl>(this))
1146     return true;
1147 
1148   if (const auto *Record = dyn_cast<CXXRecordDecl>(this)) {
1149     if (Record->getDescribedClassTemplate())
1150       return true;
1151 
1152     if (Record->isDependentLambda())
1153       return true;
1154   }
1155 
1156   if (const auto *Function = dyn_cast<FunctionDecl>(this)) {
1157     if (Function->getDescribedFunctionTemplate())
1158       return true;
1159 
1160     // Friend function declarations are dependent if their *lexical*
1161     // context is dependent.
1162     if (cast<Decl>(this)->getFriendObjectKind())
1163       return getLexicalParent()->isDependentContext();
1164   }
1165 
1166   // FIXME: A variable template is a dependent context, but is not a
1167   // DeclContext. A context within it (such as a lambda-expression)
1168   // should be considered dependent.
1169 
1170   return getParent() && getParent()->isDependentContext();
1171 }
1172 
isTransparentContext() const1173 bool DeclContext::isTransparentContext() const {
1174   if (getDeclKind() == Decl::Enum)
1175     return !cast<EnumDecl>(this)->isScoped();
1176   else if (getDeclKind() == Decl::LinkageSpec || getDeclKind() == Decl::Export)
1177     return true;
1178 
1179   return false;
1180 }
1181 
isLinkageSpecContext(const DeclContext * DC,LinkageSpecDecl::LanguageIDs ID)1182 static bool isLinkageSpecContext(const DeclContext *DC,
1183                                  LinkageSpecDecl::LanguageIDs ID) {
1184   while (DC->getDeclKind() != Decl::TranslationUnit) {
1185     if (DC->getDeclKind() == Decl::LinkageSpec)
1186       return cast<LinkageSpecDecl>(DC)->getLanguage() == ID;
1187     DC = DC->getLexicalParent();
1188   }
1189   return false;
1190 }
1191 
isExternCContext() const1192 bool DeclContext::isExternCContext() const {
1193   return isLinkageSpecContext(this, LinkageSpecDecl::lang_c);
1194 }
1195 
getExternCContext() const1196 const LinkageSpecDecl *DeclContext::getExternCContext() const {
1197   const DeclContext *DC = this;
1198   while (DC->getDeclKind() != Decl::TranslationUnit) {
1199     if (DC->getDeclKind() == Decl::LinkageSpec &&
1200         cast<LinkageSpecDecl>(DC)->getLanguage() == LinkageSpecDecl::lang_c)
1201       return cast<LinkageSpecDecl>(DC);
1202     DC = DC->getLexicalParent();
1203   }
1204   return nullptr;
1205 }
1206 
isExternCXXContext() const1207 bool DeclContext::isExternCXXContext() const {
1208   return isLinkageSpecContext(this, LinkageSpecDecl::lang_cxx);
1209 }
1210 
Encloses(const DeclContext * DC) const1211 bool DeclContext::Encloses(const DeclContext *DC) const {
1212   if (getPrimaryContext() != this)
1213     return getPrimaryContext()->Encloses(DC);
1214 
1215   for (; DC; DC = DC->getParent())
1216     if (DC->getPrimaryContext() == this)
1217       return true;
1218   return false;
1219 }
1220 
getPrimaryContext()1221 DeclContext *DeclContext::getPrimaryContext() {
1222   switch (getDeclKind()) {
1223   case Decl::TranslationUnit:
1224   case Decl::ExternCContext:
1225   case Decl::LinkageSpec:
1226   case Decl::Export:
1227   case Decl::Block:
1228   case Decl::Captured:
1229   case Decl::OMPDeclareReduction:
1230   case Decl::OMPDeclareMapper:
1231   case Decl::RequiresExprBody:
1232     // There is only one DeclContext for these entities.
1233     return this;
1234 
1235   case Decl::Namespace:
1236     // The original namespace is our primary context.
1237     return static_cast<NamespaceDecl *>(this)->getOriginalNamespace();
1238 
1239   case Decl::ObjCMethod:
1240     return this;
1241 
1242   case Decl::ObjCInterface:
1243     if (auto *OID = dyn_cast<ObjCInterfaceDecl>(this))
1244       if (auto *Def = OID->getDefinition())
1245         return Def;
1246     return this;
1247 
1248   case Decl::ObjCProtocol:
1249     if (auto *OPD = dyn_cast<ObjCProtocolDecl>(this))
1250       if (auto *Def = OPD->getDefinition())
1251         return Def;
1252     return this;
1253 
1254   case Decl::ObjCCategory:
1255     return this;
1256 
1257   case Decl::ObjCImplementation:
1258   case Decl::ObjCCategoryImpl:
1259     return this;
1260 
1261   default:
1262     if (getDeclKind() >= Decl::firstTag && getDeclKind() <= Decl::lastTag) {
1263       // If this is a tag type that has a definition or is currently
1264       // being defined, that definition is our primary context.
1265       auto *Tag = cast<TagDecl>(this);
1266 
1267       if (TagDecl *Def = Tag->getDefinition())
1268         return Def;
1269 
1270       if (const auto *TagTy = dyn_cast<TagType>(Tag->getTypeForDecl())) {
1271         // Note, TagType::getDecl returns the (partial) definition one exists.
1272         TagDecl *PossiblePartialDef = TagTy->getDecl();
1273         if (PossiblePartialDef->isBeingDefined())
1274           return PossiblePartialDef;
1275       } else {
1276         assert(isa<InjectedClassNameType>(Tag->getTypeForDecl()));
1277       }
1278 
1279       return Tag;
1280     }
1281 
1282     assert(getDeclKind() >= Decl::firstFunction &&
1283            getDeclKind() <= Decl::lastFunction &&
1284           "Unknown DeclContext kind");
1285     return this;
1286   }
1287 }
1288 
1289 void
collectAllContexts(SmallVectorImpl<DeclContext * > & Contexts)1290 DeclContext::collectAllContexts(SmallVectorImpl<DeclContext *> &Contexts){
1291   Contexts.clear();
1292 
1293   if (getDeclKind() != Decl::Namespace) {
1294     Contexts.push_back(this);
1295     return;
1296   }
1297 
1298   auto *Self = static_cast<NamespaceDecl *>(this);
1299   for (NamespaceDecl *N = Self->getMostRecentDecl(); N;
1300        N = N->getPreviousDecl())
1301     Contexts.push_back(N);
1302 
1303   std::reverse(Contexts.begin(), Contexts.end());
1304 }
1305 
1306 std::pair<Decl *, Decl *>
BuildDeclChain(ArrayRef<Decl * > Decls,bool FieldsAlreadyLoaded)1307 DeclContext::BuildDeclChain(ArrayRef<Decl *> Decls,
1308                             bool FieldsAlreadyLoaded) {
1309   // Build up a chain of declarations via the Decl::NextInContextAndBits field.
1310   Decl *FirstNewDecl = nullptr;
1311   Decl *PrevDecl = nullptr;
1312   for (auto *D : Decls) {
1313     if (FieldsAlreadyLoaded && isa<FieldDecl>(D))
1314       continue;
1315 
1316     if (PrevDecl)
1317       PrevDecl->NextInContextAndBits.setPointer(D);
1318     else
1319       FirstNewDecl = D;
1320 
1321     PrevDecl = D;
1322   }
1323 
1324   return std::make_pair(FirstNewDecl, PrevDecl);
1325 }
1326 
1327 /// We have just acquired external visible storage, and we already have
1328 /// built a lookup map. For every name in the map, pull in the new names from
1329 /// the external storage.
reconcileExternalVisibleStorage() const1330 void DeclContext::reconcileExternalVisibleStorage() const {
1331   assert(hasNeedToReconcileExternalVisibleStorage() && LookupPtr);
1332   setNeedToReconcileExternalVisibleStorage(false);
1333 
1334   for (auto &Lookup : *LookupPtr)
1335     Lookup.second.setHasExternalDecls();
1336 }
1337 
1338 /// Load the declarations within this lexical storage from an
1339 /// external source.
1340 /// \return \c true if any declarations were added.
1341 bool
LoadLexicalDeclsFromExternalStorage() const1342 DeclContext::LoadLexicalDeclsFromExternalStorage() const {
1343   ExternalASTSource *Source = getParentASTContext().getExternalSource();
1344   assert(hasExternalLexicalStorage() && Source && "No external storage?");
1345 
1346   // Notify that we have a DeclContext that is initializing.
1347   ExternalASTSource::Deserializing ADeclContext(Source);
1348 
1349   // Load the external declarations, if any.
1350   SmallVector<Decl*, 64> Decls;
1351   setHasExternalLexicalStorage(false);
1352   Source->FindExternalLexicalDecls(this, Decls);
1353 
1354   if (Decls.empty())
1355     return false;
1356 
1357   // We may have already loaded just the fields of this record, in which case
1358   // we need to ignore them.
1359   bool FieldsAlreadyLoaded = false;
1360   if (const auto *RD = dyn_cast<RecordDecl>(this))
1361     FieldsAlreadyLoaded = RD->hasLoadedFieldsFromExternalStorage();
1362 
1363   // Splice the newly-read declarations into the beginning of the list
1364   // of declarations.
1365   Decl *ExternalFirst, *ExternalLast;
1366   std::tie(ExternalFirst, ExternalLast) =
1367       BuildDeclChain(Decls, FieldsAlreadyLoaded);
1368   ExternalLast->NextInContextAndBits.setPointer(FirstDecl);
1369   FirstDecl = ExternalFirst;
1370   if (!LastDecl)
1371     LastDecl = ExternalLast;
1372   return true;
1373 }
1374 
1375 DeclContext::lookup_result
SetNoExternalVisibleDeclsForName(const DeclContext * DC,DeclarationName Name)1376 ExternalASTSource::SetNoExternalVisibleDeclsForName(const DeclContext *DC,
1377                                                     DeclarationName Name) {
1378   ASTContext &Context = DC->getParentASTContext();
1379   StoredDeclsMap *Map;
1380   if (!(Map = DC->LookupPtr))
1381     Map = DC->CreateStoredDeclsMap(Context);
1382   if (DC->hasNeedToReconcileExternalVisibleStorage())
1383     DC->reconcileExternalVisibleStorage();
1384 
1385   (*Map)[Name].removeExternalDecls();
1386 
1387   return DeclContext::lookup_result();
1388 }
1389 
1390 DeclContext::lookup_result
SetExternalVisibleDeclsForName(const DeclContext * DC,DeclarationName Name,ArrayRef<NamedDecl * > Decls)1391 ExternalASTSource::SetExternalVisibleDeclsForName(const DeclContext *DC,
1392                                                   DeclarationName Name,
1393                                                   ArrayRef<NamedDecl*> Decls) {
1394   ASTContext &Context = DC->getParentASTContext();
1395   StoredDeclsMap *Map;
1396   if (!(Map = DC->LookupPtr))
1397     Map = DC->CreateStoredDeclsMap(Context);
1398   if (DC->hasNeedToReconcileExternalVisibleStorage())
1399     DC->reconcileExternalVisibleStorage();
1400 
1401   StoredDeclsList &List = (*Map)[Name];
1402 
1403   // Clear out any old external visible declarations, to avoid quadratic
1404   // performance in the redeclaration checks below.
1405   List.removeExternalDecls();
1406 
1407   if (!List.isNull()) {
1408     // We have both existing declarations and new declarations for this name.
1409     // Some of the declarations may simply replace existing ones. Handle those
1410     // first.
1411     llvm::SmallVector<unsigned, 8> Skip;
1412     for (unsigned I = 0, N = Decls.size(); I != N; ++I)
1413       if (List.HandleRedeclaration(Decls[I], /*IsKnownNewer*/false))
1414         Skip.push_back(I);
1415     Skip.push_back(Decls.size());
1416 
1417     // Add in any new declarations.
1418     unsigned SkipPos = 0;
1419     for (unsigned I = 0, N = Decls.size(); I != N; ++I) {
1420       if (I == Skip[SkipPos])
1421         ++SkipPos;
1422       else
1423         List.AddSubsequentDecl(Decls[I]);
1424     }
1425   } else {
1426     // Convert the array to a StoredDeclsList.
1427     for (auto *D : Decls) {
1428       if (List.isNull())
1429         List.setOnlyValue(D);
1430       else
1431         List.AddSubsequentDecl(D);
1432     }
1433   }
1434 
1435   return List.getLookupResult();
1436 }
1437 
decls_begin() const1438 DeclContext::decl_iterator DeclContext::decls_begin() const {
1439   if (hasExternalLexicalStorage())
1440     LoadLexicalDeclsFromExternalStorage();
1441   return decl_iterator(FirstDecl);
1442 }
1443 
decls_empty() const1444 bool DeclContext::decls_empty() const {
1445   if (hasExternalLexicalStorage())
1446     LoadLexicalDeclsFromExternalStorage();
1447 
1448   return !FirstDecl;
1449 }
1450 
containsDecl(Decl * D) const1451 bool DeclContext::containsDecl(Decl *D) const {
1452   return (D->getLexicalDeclContext() == this &&
1453           (D->NextInContextAndBits.getPointer() || D == LastDecl));
1454 }
1455 
containsDeclAndLoad(Decl * D) const1456 bool DeclContext::containsDeclAndLoad(Decl *D) const {
1457   if (hasExternalLexicalStorage())
1458     LoadLexicalDeclsFromExternalStorage();
1459   return containsDecl(D);
1460 }
1461 
1462 /// shouldBeHidden - Determine whether a declaration which was declared
1463 /// within its semantic context should be invisible to qualified name lookup.
shouldBeHidden(NamedDecl * D)1464 static bool shouldBeHidden(NamedDecl *D) {
1465   // Skip unnamed declarations.
1466   if (!D->getDeclName())
1467     return true;
1468 
1469   // Skip entities that can't be found by name lookup into a particular
1470   // context.
1471   if ((D->getIdentifierNamespace() == 0 && !isa<UsingDirectiveDecl>(D)) ||
1472       D->isTemplateParameter())
1473     return true;
1474 
1475   // Skip friends and local extern declarations unless they're the first
1476   // declaration of the entity.
1477   if ((D->isLocalExternDecl() || D->getFriendObjectKind()) &&
1478       D != D->getCanonicalDecl())
1479     return true;
1480 
1481   // Skip template specializations.
1482   // FIXME: This feels like a hack. Should DeclarationName support
1483   // template-ids, or is there a better way to keep specializations
1484   // from being visible?
1485   if (isa<ClassTemplateSpecializationDecl>(D))
1486     return true;
1487   if (auto *FD = dyn_cast<FunctionDecl>(D))
1488     if (FD->isFunctionTemplateSpecialization())
1489       return true;
1490 
1491   // Hide destructors that are invalid. There should always be one destructor,
1492   // but if it is an invalid decl, another one is created. We need to hide the
1493   // invalid one from places that expect exactly one destructor, like the
1494   // serialization code.
1495   if (isa<CXXDestructorDecl>(D) && D->isInvalidDecl())
1496     return true;
1497 
1498   return false;
1499 }
1500 
removeDecl(Decl * D)1501 void DeclContext::removeDecl(Decl *D) {
1502   assert(D->getLexicalDeclContext() == this &&
1503          "decl being removed from non-lexical context");
1504   assert((D->NextInContextAndBits.getPointer() || D == LastDecl) &&
1505          "decl is not in decls list");
1506 
1507   // Remove D from the decl chain.  This is O(n) but hopefully rare.
1508   if (D == FirstDecl) {
1509     if (D == LastDecl)
1510       FirstDecl = LastDecl = nullptr;
1511     else
1512       FirstDecl = D->NextInContextAndBits.getPointer();
1513   } else {
1514     for (Decl *I = FirstDecl; true; I = I->NextInContextAndBits.getPointer()) {
1515       assert(I && "decl not found in linked list");
1516       if (I->NextInContextAndBits.getPointer() == D) {
1517         I->NextInContextAndBits.setPointer(D->NextInContextAndBits.getPointer());
1518         if (D == LastDecl) LastDecl = I;
1519         break;
1520       }
1521     }
1522   }
1523 
1524   // Mark that D is no longer in the decl chain.
1525   D->NextInContextAndBits.setPointer(nullptr);
1526 
1527   // Remove D from the lookup table if necessary.
1528   if (isa<NamedDecl>(D)) {
1529     auto *ND = cast<NamedDecl>(D);
1530 
1531     // Do not try to remove the declaration if that is invisible to qualified
1532     // lookup.  E.g. template specializations are skipped.
1533     if (shouldBeHidden(ND))
1534       return;
1535 
1536     // Remove only decls that have a name
1537     if (!ND->getDeclName())
1538       return;
1539 
1540     auto *DC = D->getDeclContext();
1541     do {
1542       StoredDeclsMap *Map = DC->getPrimaryContext()->LookupPtr;
1543       if (Map) {
1544         StoredDeclsMap::iterator Pos = Map->find(ND->getDeclName());
1545         assert(Pos != Map->end() && "no lookup entry for decl");
1546         // Remove the decl only if it is contained.
1547         StoredDeclsList::DeclsTy *Vec = Pos->second.getAsVector();
1548         if ((Vec && is_contained(*Vec, ND)) || Pos->second.getAsDecl() == ND)
1549           Pos->second.remove(ND);
1550       }
1551     } while (DC->isTransparentContext() && (DC = DC->getParent()));
1552   }
1553 }
1554 
addHiddenDecl(Decl * D)1555 void DeclContext::addHiddenDecl(Decl *D) {
1556   assert(D->getLexicalDeclContext() == this &&
1557          "Decl inserted into wrong lexical context");
1558   assert(!D->getNextDeclInContext() && D != LastDecl &&
1559          "Decl already inserted into a DeclContext");
1560 
1561   if (FirstDecl) {
1562     LastDecl->NextInContextAndBits.setPointer(D);
1563     LastDecl = D;
1564   } else {
1565     FirstDecl = LastDecl = D;
1566   }
1567 
1568   // Notify a C++ record declaration that we've added a member, so it can
1569   // update its class-specific state.
1570   if (auto *Record = dyn_cast<CXXRecordDecl>(this))
1571     Record->addedMember(D);
1572 
1573   // If this is a newly-created (not de-serialized) import declaration, wire
1574   // it in to the list of local import declarations.
1575   if (!D->isFromASTFile()) {
1576     if (auto *Import = dyn_cast<ImportDecl>(D))
1577       D->getASTContext().addedLocalImportDecl(Import);
1578   }
1579 }
1580 
addDecl(Decl * D)1581 void DeclContext::addDecl(Decl *D) {
1582   addHiddenDecl(D);
1583 
1584   if (auto *ND = dyn_cast<NamedDecl>(D))
1585     ND->getDeclContext()->getPrimaryContext()->
1586         makeDeclVisibleInContextWithFlags(ND, false, true);
1587 }
1588 
addDeclInternal(Decl * D)1589 void DeclContext::addDeclInternal(Decl *D) {
1590   addHiddenDecl(D);
1591 
1592   if (auto *ND = dyn_cast<NamedDecl>(D))
1593     ND->getDeclContext()->getPrimaryContext()->
1594         makeDeclVisibleInContextWithFlags(ND, true, true);
1595 }
1596 
1597 /// buildLookup - Build the lookup data structure with all of the
1598 /// declarations in this DeclContext (and any other contexts linked
1599 /// to it or transparent contexts nested within it) and return it.
1600 ///
1601 /// Note that the produced map may miss out declarations from an
1602 /// external source. If it does, those entries will be marked with
1603 /// the 'hasExternalDecls' flag.
buildLookup()1604 StoredDeclsMap *DeclContext::buildLookup() {
1605   assert(this == getPrimaryContext() && "buildLookup called on non-primary DC");
1606 
1607   if (!hasLazyLocalLexicalLookups() &&
1608       !hasLazyExternalLexicalLookups())
1609     return LookupPtr;
1610 
1611   SmallVector<DeclContext *, 2> Contexts;
1612   collectAllContexts(Contexts);
1613 
1614   if (hasLazyExternalLexicalLookups()) {
1615     setHasLazyExternalLexicalLookups(false);
1616     for (auto *DC : Contexts) {
1617       if (DC->hasExternalLexicalStorage()) {
1618         bool LoadedDecls = DC->LoadLexicalDeclsFromExternalStorage();
1619         setHasLazyLocalLexicalLookups(
1620             hasLazyLocalLexicalLookups() | LoadedDecls );
1621       }
1622     }
1623 
1624     if (!hasLazyLocalLexicalLookups())
1625       return LookupPtr;
1626   }
1627 
1628   for (auto *DC : Contexts)
1629     buildLookupImpl(DC, hasExternalVisibleStorage());
1630 
1631   // We no longer have any lazy decls.
1632   setHasLazyLocalLexicalLookups(false);
1633   return LookupPtr;
1634 }
1635 
1636 /// buildLookupImpl - Build part of the lookup data structure for the
1637 /// declarations contained within DCtx, which will either be this
1638 /// DeclContext, a DeclContext linked to it, or a transparent context
1639 /// nested within it.
buildLookupImpl(DeclContext * DCtx,bool Internal)1640 void DeclContext::buildLookupImpl(DeclContext *DCtx, bool Internal) {
1641   for (auto *D : DCtx->noload_decls()) {
1642     // Insert this declaration into the lookup structure, but only if
1643     // it's semantically within its decl context. Any other decls which
1644     // should be found in this context are added eagerly.
1645     //
1646     // If it's from an AST file, don't add it now. It'll get handled by
1647     // FindExternalVisibleDeclsByName if needed. Exception: if we're not
1648     // in C++, we do not track external visible decls for the TU, so in
1649     // that case we need to collect them all here.
1650     if (auto *ND = dyn_cast<NamedDecl>(D))
1651       if (ND->getDeclContext() == DCtx && !shouldBeHidden(ND) &&
1652           (!ND->isFromASTFile() ||
1653            (isTranslationUnit() &&
1654             !getParentASTContext().getLangOpts().CPlusPlus)))
1655         makeDeclVisibleInContextImpl(ND, Internal);
1656 
1657     // If this declaration is itself a transparent declaration context
1658     // or inline namespace, add the members of this declaration of that
1659     // context (recursively).
1660     if (auto *InnerCtx = dyn_cast<DeclContext>(D))
1661       if (InnerCtx->isTransparentContext() || InnerCtx->isInlineNamespace())
1662         buildLookupImpl(InnerCtx, Internal);
1663   }
1664 }
1665 
1666 NamedDecl *const DeclContextLookupResult::SingleElementDummyList = nullptr;
1667 
1668 DeclContext::lookup_result
lookup(DeclarationName Name) const1669 DeclContext::lookup(DeclarationName Name) const {
1670   assert(getDeclKind() != Decl::LinkageSpec &&
1671          getDeclKind() != Decl::Export &&
1672          "should not perform lookups into transparent contexts");
1673 
1674   const DeclContext *PrimaryContext = getPrimaryContext();
1675   if (PrimaryContext != this)
1676     return PrimaryContext->lookup(Name);
1677 
1678   // If we have an external source, ensure that any later redeclarations of this
1679   // context have been loaded, since they may add names to the result of this
1680   // lookup (or add external visible storage).
1681   ExternalASTSource *Source = getParentASTContext().getExternalSource();
1682   if (Source)
1683     (void)cast<Decl>(this)->getMostRecentDecl();
1684 
1685   if (hasExternalVisibleStorage()) {
1686     assert(Source && "external visible storage but no external source?");
1687 
1688     if (hasNeedToReconcileExternalVisibleStorage())
1689       reconcileExternalVisibleStorage();
1690 
1691     StoredDeclsMap *Map = LookupPtr;
1692 
1693     if (hasLazyLocalLexicalLookups() ||
1694         hasLazyExternalLexicalLookups())
1695       // FIXME: Make buildLookup const?
1696       Map = const_cast<DeclContext*>(this)->buildLookup();
1697 
1698     if (!Map)
1699       Map = CreateStoredDeclsMap(getParentASTContext());
1700 
1701     // If we have a lookup result with no external decls, we are done.
1702     std::pair<StoredDeclsMap::iterator, bool> R =
1703         Map->insert(std::make_pair(Name, StoredDeclsList()));
1704     if (!R.second && !R.first->second.hasExternalDecls())
1705       return R.first->second.getLookupResult();
1706 
1707     if (Source->FindExternalVisibleDeclsByName(this, Name) || !R.second) {
1708       if (StoredDeclsMap *Map = LookupPtr) {
1709         StoredDeclsMap::iterator I = Map->find(Name);
1710         if (I != Map->end())
1711           return I->second.getLookupResult();
1712       }
1713     }
1714 
1715     return {};
1716   }
1717 
1718   StoredDeclsMap *Map = LookupPtr;
1719   if (hasLazyLocalLexicalLookups() ||
1720       hasLazyExternalLexicalLookups())
1721     Map = const_cast<DeclContext*>(this)->buildLookup();
1722 
1723   if (!Map)
1724     return {};
1725 
1726   StoredDeclsMap::iterator I = Map->find(Name);
1727   if (I == Map->end())
1728     return {};
1729 
1730   return I->second.getLookupResult();
1731 }
1732 
1733 DeclContext::lookup_result
noload_lookup(DeclarationName Name)1734 DeclContext::noload_lookup(DeclarationName Name) {
1735   assert(getDeclKind() != Decl::LinkageSpec &&
1736          getDeclKind() != Decl::Export &&
1737          "should not perform lookups into transparent contexts");
1738 
1739   DeclContext *PrimaryContext = getPrimaryContext();
1740   if (PrimaryContext != this)
1741     return PrimaryContext->noload_lookup(Name);
1742 
1743   loadLazyLocalLexicalLookups();
1744   StoredDeclsMap *Map = LookupPtr;
1745   if (!Map)
1746     return {};
1747 
1748   StoredDeclsMap::iterator I = Map->find(Name);
1749   return I != Map->end() ? I->second.getLookupResult()
1750                          : lookup_result();
1751 }
1752 
1753 // If we have any lazy lexical declarations not in our lookup map, add them
1754 // now. Don't import any external declarations, not even if we know we have
1755 // some missing from the external visible lookups.
loadLazyLocalLexicalLookups()1756 void DeclContext::loadLazyLocalLexicalLookups() {
1757   if (hasLazyLocalLexicalLookups()) {
1758     SmallVector<DeclContext *, 2> Contexts;
1759     collectAllContexts(Contexts);
1760     for (auto *Context : Contexts)
1761       buildLookupImpl(Context, hasExternalVisibleStorage());
1762     setHasLazyLocalLexicalLookups(false);
1763   }
1764 }
1765 
localUncachedLookup(DeclarationName Name,SmallVectorImpl<NamedDecl * > & Results)1766 void DeclContext::localUncachedLookup(DeclarationName Name,
1767                                       SmallVectorImpl<NamedDecl *> &Results) {
1768   Results.clear();
1769 
1770   // If there's no external storage, just perform a normal lookup and copy
1771   // the results.
1772   if (!hasExternalVisibleStorage() && !hasExternalLexicalStorage() && Name) {
1773     lookup_result LookupResults = lookup(Name);
1774     Results.insert(Results.end(), LookupResults.begin(), LookupResults.end());
1775     return;
1776   }
1777 
1778   // If we have a lookup table, check there first. Maybe we'll get lucky.
1779   // FIXME: Should we be checking these flags on the primary context?
1780   if (Name && !hasLazyLocalLexicalLookups() &&
1781       !hasLazyExternalLexicalLookups()) {
1782     if (StoredDeclsMap *Map = LookupPtr) {
1783       StoredDeclsMap::iterator Pos = Map->find(Name);
1784       if (Pos != Map->end()) {
1785         Results.insert(Results.end(),
1786                        Pos->second.getLookupResult().begin(),
1787                        Pos->second.getLookupResult().end());
1788         return;
1789       }
1790     }
1791   }
1792 
1793   // Slow case: grovel through the declarations in our chain looking for
1794   // matches.
1795   // FIXME: If we have lazy external declarations, this will not find them!
1796   // FIXME: Should we CollectAllContexts and walk them all here?
1797   for (Decl *D = FirstDecl; D; D = D->getNextDeclInContext()) {
1798     if (auto *ND = dyn_cast<NamedDecl>(D))
1799       if (ND->getDeclName() == Name)
1800         Results.push_back(ND);
1801   }
1802 }
1803 
getRedeclContext()1804 DeclContext *DeclContext::getRedeclContext() {
1805   DeclContext *Ctx = this;
1806 
1807   // In C, a record type is the redeclaration context for its fields only. If
1808   // we arrive at a record context after skipping anything else, we should skip
1809   // the record as well. Currently, this means skipping enumerations because
1810   // they're the only transparent context that can exist within a struct or
1811   // union.
1812   bool SkipRecords = getDeclKind() == Decl::Kind::Enum &&
1813                      !getParentASTContext().getLangOpts().CPlusPlus;
1814 
1815   // Skip through contexts to get to the redeclaration context. Transparent
1816   // contexts are always skipped.
1817   while ((SkipRecords && Ctx->isRecord()) || Ctx->isTransparentContext())
1818     Ctx = Ctx->getParent();
1819   return Ctx;
1820 }
1821 
getEnclosingNamespaceContext()1822 DeclContext *DeclContext::getEnclosingNamespaceContext() {
1823   DeclContext *Ctx = this;
1824   // Skip through non-namespace, non-translation-unit contexts.
1825   while (!Ctx->isFileContext())
1826     Ctx = Ctx->getParent();
1827   return Ctx->getPrimaryContext();
1828 }
1829 
getOuterLexicalRecordContext()1830 RecordDecl *DeclContext::getOuterLexicalRecordContext() {
1831   // Loop until we find a non-record context.
1832   RecordDecl *OutermostRD = nullptr;
1833   DeclContext *DC = this;
1834   while (DC->isRecord()) {
1835     OutermostRD = cast<RecordDecl>(DC);
1836     DC = DC->getLexicalParent();
1837   }
1838   return OutermostRD;
1839 }
1840 
InEnclosingNamespaceSetOf(const DeclContext * O) const1841 bool DeclContext::InEnclosingNamespaceSetOf(const DeclContext *O) const {
1842   // For non-file contexts, this is equivalent to Equals.
1843   if (!isFileContext())
1844     return O->Equals(this);
1845 
1846   do {
1847     if (O->Equals(this))
1848       return true;
1849 
1850     const auto *NS = dyn_cast<NamespaceDecl>(O);
1851     if (!NS || !NS->isInline())
1852       break;
1853     O = NS->getParent();
1854   } while (O);
1855 
1856   return false;
1857 }
1858 
makeDeclVisibleInContext(NamedDecl * D)1859 void DeclContext::makeDeclVisibleInContext(NamedDecl *D) {
1860   DeclContext *PrimaryDC = this->getPrimaryContext();
1861   DeclContext *DeclDC = D->getDeclContext()->getPrimaryContext();
1862   // If the decl is being added outside of its semantic decl context, we
1863   // need to ensure that we eagerly build the lookup information for it.
1864   PrimaryDC->makeDeclVisibleInContextWithFlags(D, false, PrimaryDC == DeclDC);
1865 }
1866 
makeDeclVisibleInContextWithFlags(NamedDecl * D,bool Internal,bool Recoverable)1867 void DeclContext::makeDeclVisibleInContextWithFlags(NamedDecl *D, bool Internal,
1868                                                     bool Recoverable) {
1869   assert(this == getPrimaryContext() && "expected a primary DC");
1870 
1871   if (!isLookupContext()) {
1872     if (isTransparentContext())
1873       getParent()->getPrimaryContext()
1874         ->makeDeclVisibleInContextWithFlags(D, Internal, Recoverable);
1875     return;
1876   }
1877 
1878   // Skip declarations which should be invisible to name lookup.
1879   if (shouldBeHidden(D))
1880     return;
1881 
1882   // If we already have a lookup data structure, perform the insertion into
1883   // it. If we might have externally-stored decls with this name, look them
1884   // up and perform the insertion. If this decl was declared outside its
1885   // semantic context, buildLookup won't add it, so add it now.
1886   //
1887   // FIXME: As a performance hack, don't add such decls into the translation
1888   // unit unless we're in C++, since qualified lookup into the TU is never
1889   // performed.
1890   if (LookupPtr || hasExternalVisibleStorage() ||
1891       ((!Recoverable || D->getDeclContext() != D->getLexicalDeclContext()) &&
1892        (getParentASTContext().getLangOpts().CPlusPlus ||
1893         !isTranslationUnit()))) {
1894     // If we have lazily omitted any decls, they might have the same name as
1895     // the decl which we are adding, so build a full lookup table before adding
1896     // this decl.
1897     buildLookup();
1898     makeDeclVisibleInContextImpl(D, Internal);
1899   } else {
1900     setHasLazyLocalLexicalLookups(true);
1901   }
1902 
1903   // If we are a transparent context or inline namespace, insert into our
1904   // parent context, too. This operation is recursive.
1905   if (isTransparentContext() || isInlineNamespace())
1906     getParent()->getPrimaryContext()->
1907         makeDeclVisibleInContextWithFlags(D, Internal, Recoverable);
1908 
1909   auto *DCAsDecl = cast<Decl>(this);
1910   // Notify that a decl was made visible unless we are a Tag being defined.
1911   if (!(isa<TagDecl>(DCAsDecl) && cast<TagDecl>(DCAsDecl)->isBeingDefined()))
1912     if (ASTMutationListener *L = DCAsDecl->getASTMutationListener())
1913       L->AddedVisibleDecl(this, D);
1914 }
1915 
makeDeclVisibleInContextImpl(NamedDecl * D,bool Internal)1916 void DeclContext::makeDeclVisibleInContextImpl(NamedDecl *D, bool Internal) {
1917   // Find or create the stored declaration map.
1918   StoredDeclsMap *Map = LookupPtr;
1919   if (!Map) {
1920     ASTContext *C = &getParentASTContext();
1921     Map = CreateStoredDeclsMap(*C);
1922   }
1923 
1924   // If there is an external AST source, load any declarations it knows about
1925   // with this declaration's name.
1926   // If the lookup table contains an entry about this name it means that we
1927   // have already checked the external source.
1928   if (!Internal)
1929     if (ExternalASTSource *Source = getParentASTContext().getExternalSource())
1930       if (hasExternalVisibleStorage() &&
1931           Map->find(D->getDeclName()) == Map->end())
1932         Source->FindExternalVisibleDeclsByName(this, D->getDeclName());
1933 
1934   // Insert this declaration into the map.
1935   StoredDeclsList &DeclNameEntries = (*Map)[D->getDeclName()];
1936 
1937   if (Internal) {
1938     // If this is being added as part of loading an external declaration,
1939     // this may not be the only external declaration with this name.
1940     // In this case, we never try to replace an existing declaration; we'll
1941     // handle that when we finalize the list of declarations for this name.
1942     DeclNameEntries.setHasExternalDecls();
1943     DeclNameEntries.AddSubsequentDecl(D);
1944     return;
1945   }
1946 
1947   if (DeclNameEntries.isNull()) {
1948     DeclNameEntries.setOnlyValue(D);
1949     return;
1950   }
1951 
1952   if (DeclNameEntries.HandleRedeclaration(D, /*IsKnownNewer*/!Internal)) {
1953     // This declaration has replaced an existing one for which
1954     // declarationReplaces returns true.
1955     return;
1956   }
1957 
1958   // Put this declaration into the appropriate slot.
1959   DeclNameEntries.AddSubsequentDecl(D);
1960 }
1961 
operator *() const1962 UsingDirectiveDecl *DeclContext::udir_iterator::operator*() const {
1963   return cast<UsingDirectiveDecl>(*I);
1964 }
1965 
1966 /// Returns iterator range [First, Last) of UsingDirectiveDecls stored within
1967 /// this context.
using_directives() const1968 DeclContext::udir_range DeclContext::using_directives() const {
1969   // FIXME: Use something more efficient than normal lookup for using
1970   // directives. In C++, using directives are looked up more than anything else.
1971   lookup_result Result = lookup(UsingDirectiveDecl::getName());
1972   return udir_range(Result.begin(), Result.end());
1973 }
1974 
1975 //===----------------------------------------------------------------------===//
1976 // Creation and Destruction of StoredDeclsMaps.                               //
1977 //===----------------------------------------------------------------------===//
1978 
CreateStoredDeclsMap(ASTContext & C) const1979 StoredDeclsMap *DeclContext::CreateStoredDeclsMap(ASTContext &C) const {
1980   assert(!LookupPtr && "context already has a decls map");
1981   assert(getPrimaryContext() == this &&
1982          "creating decls map on non-primary context");
1983 
1984   StoredDeclsMap *M;
1985   bool Dependent = isDependentContext();
1986   if (Dependent)
1987     M = new DependentStoredDeclsMap();
1988   else
1989     M = new StoredDeclsMap();
1990   M->Previous = C.LastSDM;
1991   C.LastSDM = llvm::PointerIntPair<StoredDeclsMap*,1>(M, Dependent);
1992   LookupPtr = M;
1993   return M;
1994 }
1995 
ReleaseDeclContextMaps()1996 void ASTContext::ReleaseDeclContextMaps() {
1997   // It's okay to delete DependentStoredDeclsMaps via a StoredDeclsMap
1998   // pointer because the subclass doesn't add anything that needs to
1999   // be deleted.
2000   StoredDeclsMap::DestroyAll(LastSDM.getPointer(), LastSDM.getInt());
2001 }
2002 
DestroyAll(StoredDeclsMap * Map,bool Dependent)2003 void StoredDeclsMap::DestroyAll(StoredDeclsMap *Map, bool Dependent) {
2004   while (Map) {
2005     // Advance the iteration before we invalidate memory.
2006     llvm::PointerIntPair<StoredDeclsMap*,1> Next = Map->Previous;
2007 
2008     if (Dependent)
2009       delete static_cast<DependentStoredDeclsMap*>(Map);
2010     else
2011       delete Map;
2012 
2013     Map = Next.getPointer();
2014     Dependent = Next.getInt();
2015   }
2016 }
2017 
Create(ASTContext & C,DeclContext * Parent,const PartialDiagnostic & PDiag)2018 DependentDiagnostic *DependentDiagnostic::Create(ASTContext &C,
2019                                                  DeclContext *Parent,
2020                                            const PartialDiagnostic &PDiag) {
2021   assert(Parent->isDependentContext()
2022          && "cannot iterate dependent diagnostics of non-dependent context");
2023   Parent = Parent->getPrimaryContext();
2024   if (!Parent->LookupPtr)
2025     Parent->CreateStoredDeclsMap(C);
2026 
2027   auto *Map = static_cast<DependentStoredDeclsMap *>(Parent->LookupPtr);
2028 
2029   // Allocate the copy of the PartialDiagnostic via the ASTContext's
2030   // BumpPtrAllocator, rather than the ASTContext itself.
2031   DiagnosticStorage *DiagStorage = nullptr;
2032   if (PDiag.hasStorage())
2033     DiagStorage = new (C) DiagnosticStorage;
2034 
2035   auto *DD = new (C) DependentDiagnostic(PDiag, DiagStorage);
2036 
2037   // TODO: Maybe we shouldn't reverse the order during insertion.
2038   DD->NextDiagnostic = Map->FirstDiagnostic;
2039   Map->FirstDiagnostic = DD;
2040 
2041   return DD;
2042 }
2043