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