1 //===------ CXXInheritance.cpp - C++ Inheritance ----------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file provides routines that help analyzing C++ inheritance hierarchies.
11 //
12 //===----------------------------------------------------------------------===//
13 #include "clang/AST/CXXInheritance.h"
14 #include "clang/AST/ASTContext.h"
15 #include "clang/AST/DeclCXX.h"
16 #include "clang/AST/DeclTemplate.h"
17 #include "clang/AST/RecordLayout.h"
18 #include "llvm/ADT/SetVector.h"
19 #include <algorithm>
20 
21 using namespace clang;
22 
23 /// \brief Computes the set of declarations referenced by these base
24 /// paths.
25 void CXXBasePaths::ComputeDeclsFound() {
26   assert(NumDeclsFound == 0 && !DeclsFound &&
27          "Already computed the set of declarations");
28 
29   llvm::SetVector<NamedDecl *, SmallVector<NamedDecl *, 8> > Decls;
30   for (paths_iterator Path = begin(), PathEnd = end(); Path != PathEnd; ++Path)
31     Decls.insert(Path->Decls.front());
32 
33   NumDeclsFound = Decls.size();
34   DeclsFound = llvm::make_unique<NamedDecl *[]>(NumDeclsFound);
35   std::copy(Decls.begin(), Decls.end(), DeclsFound.get());
36 }
37 
38 CXXBasePaths::decl_range CXXBasePaths::found_decls() {
39   if (NumDeclsFound == 0)
40     ComputeDeclsFound();
41 
42   return decl_range(decl_iterator(DeclsFound.get()),
43                     decl_iterator(DeclsFound.get() + NumDeclsFound));
44 }
45 
46 /// isAmbiguous - Determines whether the set of paths provided is
47 /// ambiguous, i.e., there are two or more paths that refer to
48 /// different base class subobjects of the same type. BaseType must be
49 /// an unqualified, canonical class type.
50 bool CXXBasePaths::isAmbiguous(CanQualType BaseType) {
51   BaseType = BaseType.getUnqualifiedType();
52   std::pair<bool, unsigned>& Subobjects = ClassSubobjects[BaseType];
53   return Subobjects.second + (Subobjects.first? 1 : 0) > 1;
54 }
55 
56 /// clear - Clear out all prior path information.
57 void CXXBasePaths::clear() {
58   Paths.clear();
59   ClassSubobjects.clear();
60   ScratchPath.clear();
61   DetectedVirtual = nullptr;
62 }
63 
64 /// @brief Swaps the contents of this CXXBasePaths structure with the
65 /// contents of Other.
66 void CXXBasePaths::swap(CXXBasePaths &Other) {
67   std::swap(Origin, Other.Origin);
68   Paths.swap(Other.Paths);
69   ClassSubobjects.swap(Other.ClassSubobjects);
70   std::swap(FindAmbiguities, Other.FindAmbiguities);
71   std::swap(RecordPaths, Other.RecordPaths);
72   std::swap(DetectVirtual, Other.DetectVirtual);
73   std::swap(DetectedVirtual, Other.DetectedVirtual);
74 }
75 
76 bool CXXRecordDecl::isDerivedFrom(const CXXRecordDecl *Base) const {
77   CXXBasePaths Paths(/*FindAmbiguities=*/false, /*RecordPaths=*/false,
78                      /*DetectVirtual=*/false);
79   return isDerivedFrom(Base, Paths);
80 }
81 
82 bool CXXRecordDecl::isDerivedFrom(const CXXRecordDecl *Base,
83                                   CXXBasePaths &Paths) const {
84   if (getCanonicalDecl() == Base->getCanonicalDecl())
85     return false;
86 
87   Paths.setOrigin(const_cast<CXXRecordDecl*>(this));
88 
89   const CXXRecordDecl *BaseDecl = Base->getCanonicalDecl();
90   // FIXME: Capturing 'this' is a workaround for name lookup bugs in GCC 4.7.
91   return lookupInBases(
92       [BaseDecl](const CXXBaseSpecifier *Specifier, CXXBasePath &Path) {
93         return FindBaseClass(Specifier, Path, BaseDecl);
94       },
95       Paths);
96 }
97 
98 bool CXXRecordDecl::isVirtuallyDerivedFrom(const CXXRecordDecl *Base) const {
99   if (!getNumVBases())
100     return false;
101 
102   CXXBasePaths Paths(/*FindAmbiguities=*/false, /*RecordPaths=*/false,
103                      /*DetectVirtual=*/false);
104 
105   if (getCanonicalDecl() == Base->getCanonicalDecl())
106     return false;
107 
108   Paths.setOrigin(const_cast<CXXRecordDecl*>(this));
109 
110   const CXXRecordDecl *BaseDecl = Base->getCanonicalDecl();
111   // FIXME: Capturing 'this' is a workaround for name lookup bugs in GCC 4.7.
112   return lookupInBases(
113       [BaseDecl](const CXXBaseSpecifier *Specifier, CXXBasePath &Path) {
114         return FindVirtualBaseClass(Specifier, Path, BaseDecl);
115       },
116       Paths);
117 }
118 
119 bool CXXRecordDecl::isProvablyNotDerivedFrom(const CXXRecordDecl *Base) const {
120   const CXXRecordDecl *TargetDecl = Base->getCanonicalDecl();
121   return forallBases([TargetDecl](const CXXRecordDecl *Base) {
122     return Base->getCanonicalDecl() != TargetDecl;
123   });
124 }
125 
126 bool
127 CXXRecordDecl::isCurrentInstantiation(const DeclContext *CurContext) const {
128   assert(isDependentContext());
129 
130   for (; !CurContext->isFileContext(); CurContext = CurContext->getParent())
131     if (CurContext->Equals(this))
132       return true;
133 
134   return false;
135 }
136 
137 bool CXXRecordDecl::forallBases(ForallBasesCallback BaseMatches,
138                                 bool AllowShortCircuit) const {
139   SmallVector<const CXXRecordDecl*, 8> Queue;
140 
141   const CXXRecordDecl *Record = this;
142   bool AllMatches = true;
143   while (true) {
144     for (const auto &I : Record->bases()) {
145       const RecordType *Ty = I.getType()->getAs<RecordType>();
146       if (!Ty) {
147         if (AllowShortCircuit) return false;
148         AllMatches = false;
149         continue;
150       }
151 
152       CXXRecordDecl *Base =
153             cast_or_null<CXXRecordDecl>(Ty->getDecl()->getDefinition());
154       if (!Base ||
155           (Base->isDependentContext() &&
156            !Base->isCurrentInstantiation(Record))) {
157         if (AllowShortCircuit) return false;
158         AllMatches = false;
159         continue;
160       }
161 
162       Queue.push_back(Base);
163       if (!BaseMatches(Base)) {
164         if (AllowShortCircuit) return false;
165         AllMatches = false;
166         continue;
167       }
168     }
169 
170     if (Queue.empty())
171       break;
172     Record = Queue.pop_back_val(); // not actually a queue.
173   }
174 
175   return AllMatches;
176 }
177 
178 bool CXXBasePaths::lookupInBases(ASTContext &Context,
179                                  const CXXRecordDecl *Record,
180                                  CXXRecordDecl::BaseMatchesCallback BaseMatches,
181                                  bool LookupInDependent) {
182   bool FoundPath = false;
183 
184   // The access of the path down to this record.
185   AccessSpecifier AccessToHere = ScratchPath.Access;
186   bool IsFirstStep = ScratchPath.empty();
187 
188   for (const auto &BaseSpec : Record->bases()) {
189     // Find the record of the base class subobjects for this type.
190     QualType BaseType =
191         Context.getCanonicalType(BaseSpec.getType()).getUnqualifiedType();
192 
193     // C++ [temp.dep]p3:
194     //   In the definition of a class template or a member of a class template,
195     //   if a base class of the class template depends on a template-parameter,
196     //   the base class scope is not examined during unqualified name lookup
197     //   either at the point of definition of the class template or member or
198     //   during an instantiation of the class tem- plate or member.
199     if (!LookupInDependent && BaseType->isDependentType())
200       continue;
201 
202     // Determine whether we need to visit this base class at all,
203     // updating the count of subobjects appropriately.
204     std::pair<bool, unsigned>& Subobjects = ClassSubobjects[BaseType];
205     bool VisitBase = true;
206     bool SetVirtual = false;
207     if (BaseSpec.isVirtual()) {
208       VisitBase = !Subobjects.first;
209       Subobjects.first = true;
210       if (isDetectingVirtual() && DetectedVirtual == nullptr) {
211         // If this is the first virtual we find, remember it. If it turns out
212         // there is no base path here, we'll reset it later.
213         DetectedVirtual = BaseType->getAs<RecordType>();
214         SetVirtual = true;
215       }
216     } else
217       ++Subobjects.second;
218 
219     if (isRecordingPaths()) {
220       // Add this base specifier to the current path.
221       CXXBasePathElement Element;
222       Element.Base = &BaseSpec;
223       Element.Class = Record;
224       if (BaseSpec.isVirtual())
225         Element.SubobjectNumber = 0;
226       else
227         Element.SubobjectNumber = Subobjects.second;
228       ScratchPath.push_back(Element);
229 
230       // Calculate the "top-down" access to this base class.
231       // The spec actually describes this bottom-up, but top-down is
232       // equivalent because the definition works out as follows:
233       // 1. Write down the access along each step in the inheritance
234       //    chain, followed by the access of the decl itself.
235       //    For example, in
236       //      class A { public: int foo; };
237       //      class B : protected A {};
238       //      class C : public B {};
239       //      class D : private C {};
240       //    we would write:
241       //      private public protected public
242       // 2. If 'private' appears anywhere except far-left, access is denied.
243       // 3. Otherwise, overall access is determined by the most restrictive
244       //    access in the sequence.
245       if (IsFirstStep)
246         ScratchPath.Access = BaseSpec.getAccessSpecifier();
247       else
248         ScratchPath.Access = CXXRecordDecl::MergeAccess(AccessToHere,
249                                                  BaseSpec.getAccessSpecifier());
250     }
251 
252     // Track whether there's a path involving this specific base.
253     bool FoundPathThroughBase = false;
254 
255     if (BaseMatches(&BaseSpec, ScratchPath)) {
256       // We've found a path that terminates at this base.
257       FoundPath = FoundPathThroughBase = true;
258       if (isRecordingPaths()) {
259         // We have a path. Make a copy of it before moving on.
260         Paths.push_back(ScratchPath);
261       } else if (!isFindingAmbiguities()) {
262         // We found a path and we don't care about ambiguities;
263         // return immediately.
264         return FoundPath;
265       }
266     } else if (VisitBase) {
267       CXXRecordDecl *BaseRecord;
268       if (LookupInDependent) {
269         BaseRecord = nullptr;
270         const TemplateSpecializationType *TST =
271             BaseSpec.getType()->getAs<TemplateSpecializationType>();
272         if (!TST) {
273           if (auto *RT = BaseSpec.getType()->getAs<RecordType>())
274             BaseRecord = cast<CXXRecordDecl>(RT->getDecl());
275         } else {
276           TemplateName TN = TST->getTemplateName();
277           if (auto *TD =
278                   dyn_cast_or_null<ClassTemplateDecl>(TN.getAsTemplateDecl()))
279             BaseRecord = TD->getTemplatedDecl();
280         }
281       } else {
282         BaseRecord = cast<CXXRecordDecl>(
283             BaseSpec.getType()->castAs<RecordType>()->getDecl());
284       }
285       if (BaseRecord &&
286           lookupInBases(Context, BaseRecord, BaseMatches, LookupInDependent)) {
287         // C++ [class.member.lookup]p2:
288         //   A member name f in one sub-object B hides a member name f in
289         //   a sub-object A if A is a base class sub-object of B. Any
290         //   declarations that are so hidden are eliminated from
291         //   consideration.
292 
293         // There is a path to a base class that meets the criteria. If we're
294         // not collecting paths or finding ambiguities, we're done.
295         FoundPath = FoundPathThroughBase = true;
296         if (!isFindingAmbiguities())
297           return FoundPath;
298       }
299     }
300 
301     // Pop this base specifier off the current path (if we're
302     // collecting paths).
303     if (isRecordingPaths()) {
304       ScratchPath.pop_back();
305     }
306 
307     // If we set a virtual earlier, and this isn't a path, forget it again.
308     if (SetVirtual && !FoundPathThroughBase) {
309       DetectedVirtual = nullptr;
310     }
311   }
312 
313   // Reset the scratch path access.
314   ScratchPath.Access = AccessToHere;
315 
316   return FoundPath;
317 }
318 
319 bool CXXRecordDecl::lookupInBases(BaseMatchesCallback BaseMatches,
320                                   CXXBasePaths &Paths,
321                                   bool LookupInDependent) const {
322   // If we didn't find anything, report that.
323   if (!Paths.lookupInBases(getASTContext(), this, BaseMatches,
324                            LookupInDependent))
325     return false;
326 
327   // If we're not recording paths or we won't ever find ambiguities,
328   // we're done.
329   if (!Paths.isRecordingPaths() || !Paths.isFindingAmbiguities())
330     return true;
331 
332   // C++ [class.member.lookup]p6:
333   //   When virtual base classes are used, a hidden declaration can be
334   //   reached along a path through the sub-object lattice that does
335   //   not pass through the hiding declaration. This is not an
336   //   ambiguity. The identical use with nonvirtual base classes is an
337   //   ambiguity; in that case there is no unique instance of the name
338   //   that hides all the others.
339   //
340   // FIXME: This is an O(N^2) algorithm, but DPG doesn't see an easy
341   // way to make it any faster.
342   Paths.Paths.remove_if([&Paths](const CXXBasePath &Path) {
343     for (const CXXBasePathElement &PE : Path) {
344       if (!PE.Base->isVirtual())
345         continue;
346 
347       CXXRecordDecl *VBase = nullptr;
348       if (const RecordType *Record = PE.Base->getType()->getAs<RecordType>())
349         VBase = cast<CXXRecordDecl>(Record->getDecl());
350       if (!VBase)
351         break;
352 
353       // The declaration(s) we found along this path were found in a
354       // subobject of a virtual base. Check whether this virtual
355       // base is a subobject of any other path; if so, then the
356       // declaration in this path are hidden by that patch.
357       for (const CXXBasePath &HidingP : Paths) {
358         CXXRecordDecl *HidingClass = nullptr;
359         if (const RecordType *Record =
360                 HidingP.back().Base->getType()->getAs<RecordType>())
361           HidingClass = cast<CXXRecordDecl>(Record->getDecl());
362         if (!HidingClass)
363           break;
364 
365         if (HidingClass->isVirtuallyDerivedFrom(VBase))
366           return true;
367       }
368     }
369     return false;
370   });
371 
372   return true;
373 }
374 
375 bool CXXRecordDecl::FindBaseClass(const CXXBaseSpecifier *Specifier,
376                                   CXXBasePath &Path,
377                                   const CXXRecordDecl *BaseRecord) {
378   assert(BaseRecord->getCanonicalDecl() == BaseRecord &&
379          "User data for FindBaseClass is not canonical!");
380   return Specifier->getType()->castAs<RecordType>()->getDecl()
381             ->getCanonicalDecl() == BaseRecord;
382 }
383 
384 bool CXXRecordDecl::FindVirtualBaseClass(const CXXBaseSpecifier *Specifier,
385                                          CXXBasePath &Path,
386                                          const CXXRecordDecl *BaseRecord) {
387   assert(BaseRecord->getCanonicalDecl() == BaseRecord &&
388          "User data for FindBaseClass is not canonical!");
389   return Specifier->isVirtual() &&
390          Specifier->getType()->castAs<RecordType>()->getDecl()
391             ->getCanonicalDecl() == BaseRecord;
392 }
393 
394 bool CXXRecordDecl::FindTagMember(const CXXBaseSpecifier *Specifier,
395                                   CXXBasePath &Path,
396                                   DeclarationName Name) {
397   RecordDecl *BaseRecord =
398     Specifier->getType()->castAs<RecordType>()->getDecl();
399 
400   for (Path.Decls = BaseRecord->lookup(Name);
401        !Path.Decls.empty();
402        Path.Decls = Path.Decls.slice(1)) {
403     if (Path.Decls.front()->isInIdentifierNamespace(IDNS_Tag))
404       return true;
405   }
406 
407   return false;
408 }
409 
410 static bool findOrdinaryMember(RecordDecl *BaseRecord, CXXBasePath &Path,
411                                DeclarationName Name) {
412   const unsigned IDNS = clang::Decl::IDNS_Ordinary | clang::Decl::IDNS_Tag |
413                         clang::Decl::IDNS_Member;
414   for (Path.Decls = BaseRecord->lookup(Name);
415        !Path.Decls.empty();
416        Path.Decls = Path.Decls.slice(1)) {
417     if (Path.Decls.front()->isInIdentifierNamespace(IDNS))
418       return true;
419   }
420 
421   return false;
422 }
423 
424 bool CXXRecordDecl::FindOrdinaryMember(const CXXBaseSpecifier *Specifier,
425                                        CXXBasePath &Path,
426                                        DeclarationName Name) {
427   RecordDecl *BaseRecord =
428       Specifier->getType()->castAs<RecordType>()->getDecl();
429   return findOrdinaryMember(BaseRecord, Path, Name);
430 }
431 
432 bool CXXRecordDecl::FindOrdinaryMemberInDependentClasses(
433     const CXXBaseSpecifier *Specifier, CXXBasePath &Path,
434     DeclarationName Name) {
435   const TemplateSpecializationType *TST =
436       Specifier->getType()->getAs<TemplateSpecializationType>();
437   if (!TST) {
438     auto *RT = Specifier->getType()->getAs<RecordType>();
439     if (!RT)
440       return false;
441     return findOrdinaryMember(RT->getDecl(), Path, Name);
442   }
443   TemplateName TN = TST->getTemplateName();
444   const auto *TD = dyn_cast_or_null<ClassTemplateDecl>(TN.getAsTemplateDecl());
445   if (!TD)
446     return false;
447   CXXRecordDecl *RD = TD->getTemplatedDecl();
448   if (!RD)
449     return false;
450   return findOrdinaryMember(RD, Path, Name);
451 }
452 
453 bool CXXRecordDecl::FindOMPReductionMember(const CXXBaseSpecifier *Specifier,
454                                            CXXBasePath &Path,
455                                            DeclarationName Name) {
456   RecordDecl *BaseRecord =
457       Specifier->getType()->castAs<RecordType>()->getDecl();
458 
459   for (Path.Decls = BaseRecord->lookup(Name); !Path.Decls.empty();
460        Path.Decls = Path.Decls.slice(1)) {
461     if (Path.Decls.front()->isInIdentifierNamespace(IDNS_OMPReduction))
462       return true;
463   }
464 
465   return false;
466 }
467 
468 bool CXXRecordDecl::
469 FindNestedNameSpecifierMember(const CXXBaseSpecifier *Specifier,
470                               CXXBasePath &Path,
471                               DeclarationName Name) {
472   RecordDecl *BaseRecord =
473     Specifier->getType()->castAs<RecordType>()->getDecl();
474 
475   for (Path.Decls = BaseRecord->lookup(Name);
476        !Path.Decls.empty();
477        Path.Decls = Path.Decls.slice(1)) {
478     // FIXME: Refactor the "is it a nested-name-specifier?" check
479     if (isa<TypedefNameDecl>(Path.Decls.front()) ||
480         Path.Decls.front()->isInIdentifierNamespace(IDNS_Tag))
481       return true;
482   }
483 
484   return false;
485 }
486 
487 std::vector<const NamedDecl *> CXXRecordDecl::lookupDependentName(
488     const DeclarationName &Name,
489     llvm::function_ref<bool(const NamedDecl *ND)> Filter) {
490   std::vector<const NamedDecl *> Results;
491   // Lookup in the class.
492   DeclContext::lookup_result DirectResult = lookup(Name);
493   if (!DirectResult.empty()) {
494     for (const NamedDecl *ND : DirectResult) {
495       if (Filter(ND))
496         Results.push_back(ND);
497     }
498     return Results;
499   }
500   // Perform lookup into our base classes.
501   CXXBasePaths Paths;
502   Paths.setOrigin(this);
503   if (!lookupInBases(
504           [&](const CXXBaseSpecifier *Specifier, CXXBasePath &Path) {
505             return CXXRecordDecl::FindOrdinaryMemberInDependentClasses(
506                 Specifier, Path, Name);
507           },
508           Paths, /*LookupInDependent=*/true))
509     return Results;
510   for (const NamedDecl *ND : Paths.front().Decls) {
511     if (Filter(ND))
512       Results.push_back(ND);
513   }
514   return Results;
515 }
516 
517 void OverridingMethods::add(unsigned OverriddenSubobject,
518                             UniqueVirtualMethod Overriding) {
519   SmallVectorImpl<UniqueVirtualMethod> &SubobjectOverrides
520     = Overrides[OverriddenSubobject];
521   if (std::find(SubobjectOverrides.begin(), SubobjectOverrides.end(),
522                 Overriding) == SubobjectOverrides.end())
523     SubobjectOverrides.push_back(Overriding);
524 }
525 
526 void OverridingMethods::add(const OverridingMethods &Other) {
527   for (const_iterator I = Other.begin(), IE = Other.end(); I != IE; ++I) {
528     for (overriding_const_iterator M = I->second.begin(),
529                                 MEnd = I->second.end();
530          M != MEnd;
531          ++M)
532       add(I->first, *M);
533   }
534 }
535 
536 void OverridingMethods::replaceAll(UniqueVirtualMethod Overriding) {
537   for (iterator I = begin(), IEnd = end(); I != IEnd; ++I) {
538     I->second.clear();
539     I->second.push_back(Overriding);
540   }
541 }
542 
543 
544 namespace {
545   class FinalOverriderCollector {
546     /// \brief The number of subobjects of a given class type that
547     /// occur within the class hierarchy.
548     llvm::DenseMap<const CXXRecordDecl *, unsigned> SubobjectCount;
549 
550     /// \brief Overriders for each virtual base subobject.
551     llvm::DenseMap<const CXXRecordDecl *, CXXFinalOverriderMap *> VirtualOverriders;
552 
553     CXXFinalOverriderMap FinalOverriders;
554 
555   public:
556     ~FinalOverriderCollector();
557 
558     void Collect(const CXXRecordDecl *RD, bool VirtualBase,
559                  const CXXRecordDecl *InVirtualSubobject,
560                  CXXFinalOverriderMap &Overriders);
561   };
562 }
563 
564 void FinalOverriderCollector::Collect(const CXXRecordDecl *RD,
565                                       bool VirtualBase,
566                                       const CXXRecordDecl *InVirtualSubobject,
567                                       CXXFinalOverriderMap &Overriders) {
568   unsigned SubobjectNumber = 0;
569   if (!VirtualBase)
570     SubobjectNumber
571       = ++SubobjectCount[cast<CXXRecordDecl>(RD->getCanonicalDecl())];
572 
573   for (const auto &Base : RD->bases()) {
574     if (const RecordType *RT = Base.getType()->getAs<RecordType>()) {
575       const CXXRecordDecl *BaseDecl = cast<CXXRecordDecl>(RT->getDecl());
576       if (!BaseDecl->isPolymorphic())
577         continue;
578 
579       if (Overriders.empty() && !Base.isVirtual()) {
580         // There are no other overriders of virtual member functions,
581         // so let the base class fill in our overriders for us.
582         Collect(BaseDecl, false, InVirtualSubobject, Overriders);
583         continue;
584       }
585 
586       // Collect all of the overridders from the base class subobject
587       // and merge them into the set of overridders for this class.
588       // For virtual base classes, populate or use the cached virtual
589       // overrides so that we do not walk the virtual base class (and
590       // its base classes) more than once.
591       CXXFinalOverriderMap ComputedBaseOverriders;
592       CXXFinalOverriderMap *BaseOverriders = &ComputedBaseOverriders;
593       if (Base.isVirtual()) {
594         CXXFinalOverriderMap *&MyVirtualOverriders = VirtualOverriders[BaseDecl];
595         BaseOverriders = MyVirtualOverriders;
596         if (!MyVirtualOverriders) {
597           MyVirtualOverriders = new CXXFinalOverriderMap;
598 
599           // Collect may cause VirtualOverriders to reallocate, invalidating the
600           // MyVirtualOverriders reference. Set BaseOverriders to the right
601           // value now.
602           BaseOverriders = MyVirtualOverriders;
603 
604           Collect(BaseDecl, true, BaseDecl, *MyVirtualOverriders);
605         }
606       } else
607         Collect(BaseDecl, false, InVirtualSubobject, ComputedBaseOverriders);
608 
609       // Merge the overriders from this base class into our own set of
610       // overriders.
611       for (CXXFinalOverriderMap::iterator OM = BaseOverriders->begin(),
612                                OMEnd = BaseOverriders->end();
613            OM != OMEnd;
614            ++OM) {
615         const CXXMethodDecl *CanonOM
616           = cast<CXXMethodDecl>(OM->first->getCanonicalDecl());
617         Overriders[CanonOM].add(OM->second);
618       }
619     }
620   }
621 
622   for (auto *M : RD->methods()) {
623     // We only care about virtual methods.
624     if (!M->isVirtual())
625       continue;
626 
627     CXXMethodDecl *CanonM = cast<CXXMethodDecl>(M->getCanonicalDecl());
628 
629     if (CanonM->begin_overridden_methods()
630                                        == CanonM->end_overridden_methods()) {
631       // This is a new virtual function that does not override any
632       // other virtual function. Add it to the map of virtual
633       // functions for which we are tracking overridders.
634 
635       // C++ [class.virtual]p2:
636       //   For convenience we say that any virtual function overrides itself.
637       Overriders[CanonM].add(SubobjectNumber,
638                              UniqueVirtualMethod(CanonM, SubobjectNumber,
639                                                  InVirtualSubobject));
640       continue;
641     }
642 
643     // This virtual method overrides other virtual methods, so it does
644     // not add any new slots into the set of overriders. Instead, we
645     // replace entries in the set of overriders with the new
646     // overrider. To do so, we dig down to the original virtual
647     // functions using data recursion and update all of the methods it
648     // overrides.
649     typedef llvm::iterator_range<CXXMethodDecl::method_iterator>
650         OverriddenMethods;
651     SmallVector<OverriddenMethods, 4> Stack;
652     Stack.push_back(llvm::make_range(CanonM->begin_overridden_methods(),
653                                      CanonM->end_overridden_methods()));
654     while (!Stack.empty()) {
655       for (const CXXMethodDecl *OM : Stack.pop_back_val()) {
656         const CXXMethodDecl *CanonOM = OM->getCanonicalDecl();
657 
658         // C++ [class.virtual]p2:
659         //   A virtual member function C::vf of a class object S is
660         //   a final overrider unless the most derived class (1.8)
661         //   of which S is a base class subobject (if any) declares
662         //   or inherits another member function that overrides vf.
663         //
664         // Treating this object like the most derived class, we
665         // replace any overrides from base classes with this
666         // overriding virtual function.
667         Overriders[CanonOM].replaceAll(
668                                UniqueVirtualMethod(CanonM, SubobjectNumber,
669                                                    InVirtualSubobject));
670 
671         if (CanonOM->begin_overridden_methods()
672                                        == CanonOM->end_overridden_methods())
673           continue;
674 
675         // Continue recursion to the methods that this virtual method
676         // overrides.
677         Stack.push_back(llvm::make_range(CanonOM->begin_overridden_methods(),
678                                          CanonOM->end_overridden_methods()));
679       }
680     }
681 
682     // C++ [class.virtual]p2:
683     //   For convenience we say that any virtual function overrides itself.
684     Overriders[CanonM].add(SubobjectNumber,
685                            UniqueVirtualMethod(CanonM, SubobjectNumber,
686                                                InVirtualSubobject));
687   }
688 }
689 
690 FinalOverriderCollector::~FinalOverriderCollector() {
691   for (llvm::DenseMap<const CXXRecordDecl *, CXXFinalOverriderMap *>::iterator
692          VO = VirtualOverriders.begin(), VOEnd = VirtualOverriders.end();
693        VO != VOEnd;
694        ++VO)
695     delete VO->second;
696 }
697 
698 void
699 CXXRecordDecl::getFinalOverriders(CXXFinalOverriderMap &FinalOverriders) const {
700   FinalOverriderCollector Collector;
701   Collector.Collect(this, false, nullptr, FinalOverriders);
702 
703   // Weed out any final overriders that come from virtual base class
704   // subobjects that were hidden by other subobjects along any path.
705   // This is the final-overrider variant of C++ [class.member.lookup]p10.
706   for (auto &OM : FinalOverriders) {
707     for (auto &SO : OM.second) {
708       SmallVectorImpl<UniqueVirtualMethod> &Overriding = SO.second;
709       if (Overriding.size() < 2)
710         continue;
711 
712       auto IsHidden = [&Overriding](const UniqueVirtualMethod &M) {
713         if (!M.InVirtualSubobject)
714           return false;
715 
716         // We have an overriding method in a virtual base class
717         // subobject (or non-virtual base class subobject thereof);
718         // determine whether there exists an other overriding method
719         // in a base class subobject that hides the virtual base class
720         // subobject.
721         for (const UniqueVirtualMethod &OP : Overriding)
722           if (&M != &OP &&
723               OP.Method->getParent()->isVirtuallyDerivedFrom(
724                   M.InVirtualSubobject))
725             return true;
726         return false;
727       };
728 
729       Overriding.erase(
730           std::remove_if(Overriding.begin(), Overriding.end(), IsHidden),
731           Overriding.end());
732     }
733   }
734 }
735 
736 static void
737 AddIndirectPrimaryBases(const CXXRecordDecl *RD, ASTContext &Context,
738                         CXXIndirectPrimaryBaseSet& Bases) {
739   // If the record has a virtual primary base class, add it to our set.
740   const ASTRecordLayout &Layout = Context.getASTRecordLayout(RD);
741   if (Layout.isPrimaryBaseVirtual())
742     Bases.insert(Layout.getPrimaryBase());
743 
744   for (const auto &I : RD->bases()) {
745     assert(!I.getType()->isDependentType() &&
746            "Cannot get indirect primary bases for class with dependent bases.");
747 
748     const CXXRecordDecl *BaseDecl =
749       cast<CXXRecordDecl>(I.getType()->castAs<RecordType>()->getDecl());
750 
751     // Only bases with virtual bases participate in computing the
752     // indirect primary virtual base classes.
753     if (BaseDecl->getNumVBases())
754       AddIndirectPrimaryBases(BaseDecl, Context, Bases);
755   }
756 
757 }
758 
759 void
760 CXXRecordDecl::getIndirectPrimaryBases(CXXIndirectPrimaryBaseSet& Bases) const {
761   ASTContext &Context = getASTContext();
762 
763   if (!getNumVBases())
764     return;
765 
766   for (const auto &I : bases()) {
767     assert(!I.getType()->isDependentType() &&
768            "Cannot get indirect primary bases for class with dependent bases.");
769 
770     const CXXRecordDecl *BaseDecl =
771       cast<CXXRecordDecl>(I.getType()->castAs<RecordType>()->getDecl());
772 
773     // Only bases with virtual bases participate in computing the
774     // indirect primary virtual base classes.
775     if (BaseDecl->getNumVBases())
776       AddIndirectPrimaryBases(BaseDecl, Context, Bases);
777   }
778 }
779