blob: 15b9a924df633626ead8f35963b4e748990644b0 [file] [log] [blame]
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001//===--- CGVtable.cpp - Emit LLVM Code for C++ vtables --------------------===//
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 contains code dealing with C++ code generation of virtual tables.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CodeGenModule.h"
15#include "CodeGenFunction.h"
Anders Carlssonf942ee02009-11-27 20:47:55 +000016#include "clang/AST/CXXInheritance.h"
Anders Carlsson2bb27f52009-10-11 22:13:54 +000017#include "clang/AST/RecordLayout.h"
Anders Carlssond420a312009-11-26 19:32:45 +000018#include "llvm/ADT/DenseSet.h"
Chandler Carruth94eab4a2010-02-13 10:38:52 +000019#include "llvm/Support/Compiler.h"
Anders Carlsson5d40c6f2010-02-11 08:02:13 +000020#include "llvm/Support/Format.h"
Zhongxing Xu1721ef72009-11-13 05:46:16 +000021#include <cstdio>
Anders Carlsson2bb27f52009-10-11 22:13:54 +000022
23using namespace clang;
24using namespace CodeGen;
25
Anders Carlssond59885022009-11-27 22:21:51 +000026namespace {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +000027
Anders Carlssone8ac42a2010-02-11 21:24:32 +000028/// FinalOverriders - Contains the final overrider member functions for all
29/// member functions in the base subobjects of a class.
30class FinalOverriders {
Anders Carlsson136bd192010-02-12 16:55:34 +000031public:
Anders Carlssoneee53d92010-02-13 02:02:03 +000032 /// BaseOffset - Represents an offset from a derived class to a direct or
33 /// indirect base class.
34 struct BaseOffset {
Anders Carlssoncf5a8822010-02-13 21:07:32 +000035 /// DerivedClass - The derived class.
36 const CXXRecordDecl *DerivedClass;
37
Anders Carlssoneee53d92010-02-13 02:02:03 +000038 /// VirtualBase - If the path from the derived class to the base class
39 /// involves a virtual base class, this holds its declaration.
40 const CXXRecordDecl *VirtualBase;
41
42 /// NonVirtualOffset - The offset from the derived class to the base class.
43 /// Or the offset from the virtual base class to the base class, if the path
44 /// from the derived class to the base class involves a virtual base class.
Anders Carlssoncce9ee82010-02-13 23:17:31 +000045 int64_t NonVirtualOffset;
Anders Carlssoneee53d92010-02-13 02:02:03 +000046
Anders Carlssoncf5a8822010-02-13 21:07:32 +000047 BaseOffset() : DerivedClass(0), VirtualBase(0), NonVirtualOffset(0) { }
48 BaseOffset(const CXXRecordDecl *DerivedClass,
Anders Carlssoncce9ee82010-02-13 23:17:31 +000049 const CXXRecordDecl *VirtualBase, int64_t NonVirtualOffset)
Anders Carlssoncf5a8822010-02-13 21:07:32 +000050 : DerivedClass(DerivedClass), VirtualBase(VirtualBase),
51 NonVirtualOffset(NonVirtualOffset) { }
Anders Carlssoneee53d92010-02-13 02:02:03 +000052
Benjamin Kramere047e112010-02-13 09:15:07 +000053 bool isEmpty() const { return !NonVirtualOffset && !VirtualBase; }
Anders Carlssoneee53d92010-02-13 02:02:03 +000054 };
55
Anders Carlsson136bd192010-02-12 16:55:34 +000056 /// OverriderInfo - Information about a final overrider.
57 struct OverriderInfo {
58 /// Method - The method decl of the overrider.
59 const CXXMethodDecl *Method;
60
Anders Carlssoneee53d92010-02-13 02:02:03 +000061 OverriderInfo() : Method(0) { }
Anders Carlsson136bd192010-02-12 16:55:34 +000062 };
63
64private:
Anders Carlssone8ac42a2010-02-11 21:24:32 +000065 /// MostDerivedClass - The most derived class for which the final overriders
66 /// are stored.
67 const CXXRecordDecl *MostDerivedClass;
68
69 ASTContext &Context;
70
71 /// MostDerivedClassLayout - the AST record layout of the most derived class.
72 const ASTRecordLayout &MostDerivedClassLayout;
73
Anders Carlssoneee53d92010-02-13 02:02:03 +000074 /// BaseSubobjectMethodPairTy - Uniquely identifies a member function
75 /// in a base subobject.
76 typedef std::pair<BaseSubobject, const CXXMethodDecl *>
77 BaseSubobjectMethodPairTy;
78
79 typedef llvm::DenseMap<BaseSubobjectMethodPairTy,
Anders Carlsson136bd192010-02-12 16:55:34 +000080 OverriderInfo> OverridersMapTy;
Anders Carlssone8ac42a2010-02-11 21:24:32 +000081
82 /// OverridersMap - The final overriders for all virtual member functions of
83 /// all the base subobjects of the most derived class.
84 OverridersMapTy OverridersMap;
85
Anders Carlssoneee53d92010-02-13 02:02:03 +000086 typedef llvm::DenseMap<BaseSubobjectMethodPairTy, BaseOffset>
87 AdjustmentOffsetsMapTy;
88
89 /// ReturnAdjustments - Holds return adjustments for all the overriders that
90 /// need to perform return value adjustments.
91 AdjustmentOffsetsMapTy ReturnAdjustments;
92
Anders Carlssoncce9ee82010-02-13 23:17:31 +000093 /// ThisAdjustments - Holds 'this' adjustments for all the overriders that
94 /// need them.
95 AdjustmentOffsetsMapTy ThisAdjustments;
96
Anders Carlsson9fbf6d92010-02-12 01:40:03 +000097 typedef llvm::SmallVector<uint64_t, 1> OffsetVectorTy;
98
99 /// SubobjectOffsetsMapTy - This map is used for keeping track of all the
100 /// base subobject offsets that a single class declaration might refer to.
101 ///
102 /// For example, in:
103 ///
104 /// struct A { virtual void f(); };
105 /// struct B1 : A { };
106 /// struct B2 : A { };
107 /// struct C : B1, B2 { virtual void f(); };
108 ///
109 /// when we determine that C::f() overrides A::f(), we need to update the
110 /// overriders map for both A-in-B1 and A-in-B2 and the subobject offsets map
111 /// will have the subobject offsets for both A copies.
112 typedef llvm::DenseMap<const CXXRecordDecl *, OffsetVectorTy>
113 SubobjectOffsetsMapTy;
114
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000115 /// ComputeFinalOverriders - Compute the final overriders for a given base
116 /// subobject (and all its direct and indirect bases).
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000117 void ComputeFinalOverriders(BaseSubobject Base,
118 SubobjectOffsetsMapTy &Offsets);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000119
120 /// AddOverriders - Add the final overriders for this base subobject to the
121 /// map of final overriders.
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000122 void AddOverriders(BaseSubobject Base, SubobjectOffsetsMapTy &Offsets);
123
124 /// PropagateOverrider - Propagate the NewMD overrider to all the functions
125 /// that OldMD overrides. For example, if we have:
126 ///
127 /// struct A { virtual void f(); };
128 /// struct B : A { virtual void f(); };
129 /// struct C : B { virtual void f(); };
130 ///
131 /// and we want to override B::f with C::f, we also need to override A::f with
132 /// C::f.
133 void PropagateOverrider(const CXXMethodDecl *OldMD,
Anders Carlssona678aa12010-02-13 22:23:31 +0000134 BaseSubobject NewBase,
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000135 const CXXMethodDecl *NewMD,
136 SubobjectOffsetsMapTy &Offsets);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000137
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000138 /// ComputeThisAdjustmentBaseOffset - Compute the base offset for adjusting
139 /// the 'this' pointer from BaseRD to DerivedRD.
140 BaseOffset ComputeThisAdjustmentBaseOffset(const CXXRecordDecl *BaseRD,
141 const CXXRecordDecl *DerivedRD);
142
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000143 static void MergeSubobjectOffsets(const SubobjectOffsetsMapTy &NewOffsets,
144 SubobjectOffsetsMapTy &Offsets);
145
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000146public:
147 explicit FinalOverriders(const CXXRecordDecl *MostDerivedClass);
148
149 /// getOverrider - Get the final overrider for the given method declaration in
150 /// the given base subobject.
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000151 OverriderInfo getOverrider(BaseSubobject Base,
152 const CXXMethodDecl *MD) const {
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000153 assert(OverridersMap.count(std::make_pair(Base, MD)) &&
154 "Did not find overrider!");
155
156 return OverridersMap.lookup(std::make_pair(Base, MD));
157 }
158
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000159 /// getReturnAdjustmentOffset - Get the return adjustment offset for the
160 /// method decl in the given base subobject. Returns an empty base offset if
161 /// no adjustment is needed.
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000162 BaseOffset getReturnAdjustmentOffset(BaseSubobject Base,
163 const CXXMethodDecl *MD) const {
164 return ReturnAdjustments.lookup(std::make_pair(Base, MD));
165 }
166
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000167 /// getThisAdjustmentOffset - Get the 'this' pointer adjustment offset for the
168 /// method decl in the given base subobject. Returns an empty base offset if
169 /// no adjustment is needed.
170 BaseOffset getThisAdjustmentOffset(BaseSubobject Base,
171 const CXXMethodDecl *MD) const {
172 return ThisAdjustments.lookup(std::make_pair(Base, MD));
173 }
174
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000175 /// dump - dump the final overriders.
176 void dump() const {
177 dump(llvm::errs(), BaseSubobject(MostDerivedClass, 0));
178 }
179
180 /// dump - dump the final overriders for a base subobject, and all its direct
181 /// and indirect base subobjects.
182 void dump(llvm::raw_ostream &Out, BaseSubobject Base) const;
183};
Anders Carlssoneee53d92010-02-13 02:02:03 +0000184
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000185FinalOverriders::FinalOverriders(const CXXRecordDecl *MostDerivedClass)
186 : MostDerivedClass(MostDerivedClass),
187 Context(MostDerivedClass->getASTContext()),
188 MostDerivedClassLayout(Context.getASTRecordLayout(MostDerivedClass)) {
189
190 // Compute the final overriders.
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000191 SubobjectOffsetsMapTy Offsets;
192 ComputeFinalOverriders(BaseSubobject(MostDerivedClass, 0), Offsets);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000193
194 // And dump them (for now).
195 dump();
Anders Carlsson3f9917d2010-02-13 21:33:22 +0000196
197 // Also dump the base offsets (for now).
198 for (SubobjectOffsetsMapTy::const_iterator I = Offsets.begin(),
199 E = Offsets.end(); I != E; ++I) {
200 const OffsetVectorTy& OffsetVector = I->second;
201
202 llvm::errs() << "Base offsets for ";
203 llvm::errs() << I->first->getQualifiedNameAsString() << '\n';
204
205 for (unsigned I = 0, E = OffsetVector.size(); I != E; ++I)
206 llvm::errs() << " " << I << " - " << OffsetVector[I] << '\n';
207 }
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000208}
209
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000210void FinalOverriders::AddOverriders(BaseSubobject Base,
211 SubobjectOffsetsMapTy &Offsets) {
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000212 const CXXRecordDecl *RD = Base.getBase();
213
214 for (CXXRecordDecl::method_iterator I = RD->method_begin(),
215 E = RD->method_end(); I != E; ++I) {
216 const CXXMethodDecl *MD = *I;
217
218 if (!MD->isVirtual())
219 continue;
220
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000221 // First, propagate the overrider.
Anders Carlssona678aa12010-02-13 22:23:31 +0000222 PropagateOverrider(MD, Base, MD, Offsets);
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000223
224 // Add the overrider as the final overrider of itself.
Anders Carlsson136bd192010-02-12 16:55:34 +0000225 OverriderInfo& Overrider = OverridersMap[std::make_pair(Base, MD)];
226 assert(!Overrider.Method && "Overrider should not exist yet!");
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000227
Anders Carlsson136bd192010-02-12 16:55:34 +0000228 Overrider.Method = MD;
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000229 }
230}
231
Anders Carlssoneee53d92010-02-13 02:02:03 +0000232static FinalOverriders::BaseOffset
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000233ComputeBaseOffset(ASTContext &Context, const CXXRecordDecl *BaseRD,
234 const CXXRecordDecl *DerivedRD) {
Anders Carlssoneee53d92010-02-13 02:02:03 +0000235 CXXBasePaths Paths(/*FindAmbiguities=*/false,
Anders Carlssonf3f67842010-02-13 20:41:15 +0000236 /*RecordPaths=*/true, /*DetectVirtual=*/false);
Anders Carlssoneee53d92010-02-13 02:02:03 +0000237
238 if (!const_cast<CXXRecordDecl *>(DerivedRD)->
239 isDerivedFrom(const_cast<CXXRecordDecl *>(BaseRD), Paths)) {
240 assert(false && "Class must be derived from the passed in base class!");
241 return FinalOverriders::BaseOffset();
242 }
243
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000244 int64_t NonVirtualOffset = 0;
Anders Carlssoneee53d92010-02-13 02:02:03 +0000245
246 const CXXBasePath &Path = Paths.front();
247
Anders Carlssonf3f67842010-02-13 20:41:15 +0000248 unsigned NonVirtualStart = 0;
249 const CXXRecordDecl *VirtualBase = 0;
250
251 // First, look for the virtual base class.
252 for (unsigned I = 0, E = Path.size(); I != E; ++I) {
253 const CXXBasePathElement &Element = Path[I];
254
255 if (Element.Base->isVirtual()) {
256 // FIXME: Can we break when we find the first virtual base?
257 // (If we can't, can't we just iterate over the path in reverse order?)
258 NonVirtualStart = I + 1;
259 QualType VBaseType = Element.Base->getType();
260 VirtualBase =
261 cast<CXXRecordDecl>(VBaseType->getAs<RecordType>()->getDecl());
262 }
263 }
264
265 // Now compute the non-virtual offset.
266 for (unsigned I = NonVirtualStart, E = Path.size(); I != E; ++I) {
267 const CXXBasePathElement &Element = Path[I];
Anders Carlssoneee53d92010-02-13 02:02:03 +0000268
269 // Check the base class offset.
270 const ASTRecordLayout &Layout = Context.getASTRecordLayout(Element.Class);
271
272 const RecordType *BaseType = Element.Base->getType()->getAs<RecordType>();
273 const CXXRecordDecl *Base = cast<CXXRecordDecl>(BaseType->getDecl());
274
275 NonVirtualOffset += Layout.getBaseClassOffset(Base);
276 }
277
278 // FIXME: This should probably use CharUnits or something. Maybe we should
279 // even change the base offsets in ASTRecordLayout to be specified in
280 // CharUnits.
Anders Carlssoncf5a8822010-02-13 21:07:32 +0000281 return FinalOverriders::BaseOffset(DerivedRD, VirtualBase,
282 NonVirtualOffset / 8);
Anders Carlssoneee53d92010-02-13 02:02:03 +0000283}
284
285static FinalOverriders::BaseOffset
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000286ComputeReturnAdjustmentBaseOffset(ASTContext &Context,
287 const CXXMethodDecl *DerivedMD,
288 const CXXMethodDecl *BaseMD) {
Anders Carlssoneee53d92010-02-13 02:02:03 +0000289 const FunctionType *BaseFT = BaseMD->getType()->getAs<FunctionType>();
290 const FunctionType *DerivedFT = DerivedMD->getType()->getAs<FunctionType>();
291
292 // Canonicalize the return types.
293 CanQualType CanDerivedReturnType =
294 Context.getCanonicalType(DerivedFT->getResultType());
295 CanQualType CanBaseReturnType =
296 Context.getCanonicalType(BaseFT->getResultType());
297
298 assert(CanDerivedReturnType->getTypeClass() ==
299 CanBaseReturnType->getTypeClass() &&
300 "Types must have same type class!");
301
302 if (CanDerivedReturnType == CanBaseReturnType) {
303 // No adjustment needed.
304 return FinalOverriders::BaseOffset();
305 }
306
307 if (isa<ReferenceType>(CanDerivedReturnType)) {
308 CanDerivedReturnType =
309 CanDerivedReturnType->getAs<ReferenceType>()->getPointeeType();
310 CanBaseReturnType =
311 CanBaseReturnType->getAs<ReferenceType>()->getPointeeType();
312 } else if (isa<PointerType>(CanDerivedReturnType)) {
313 CanDerivedReturnType =
314 CanDerivedReturnType->getAs<PointerType>()->getPointeeType();
315 CanBaseReturnType =
316 CanBaseReturnType->getAs<PointerType>()->getPointeeType();
317 } else {
318 assert(false && "Unexpected return type!");
319 }
320
321 // We need to compare unqualified types here; consider
322 // const T *Base::foo();
323 // T *Derived::foo();
324 if (CanDerivedReturnType.getUnqualifiedType() ==
325 CanBaseReturnType.getUnqualifiedType()) {
326 // No adjustment needed.
327 return FinalOverriders::BaseOffset();
328 }
329
330 const CXXRecordDecl *DerivedRD =
331 cast<CXXRecordDecl>(cast<RecordType>(CanDerivedReturnType)->getDecl());
332
333 const CXXRecordDecl *BaseRD =
334 cast<CXXRecordDecl>(cast<RecordType>(CanBaseReturnType)->getDecl());
335
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000336 return ComputeBaseOffset(Context, BaseRD, DerivedRD);
Anders Carlssoneee53d92010-02-13 02:02:03 +0000337}
338
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000339FinalOverriders::BaseOffset
340FinalOverriders::ComputeThisAdjustmentBaseOffset(const CXXRecordDecl *BaseRD,
341 const CXXRecordDecl *DerivedRD) {
342 CXXBasePaths Paths(/*FindAmbiguities=*/false,
343 /*RecordPaths=*/true, /*DetectVirtual=*/true);
344
345 if (!const_cast<CXXRecordDecl *>(DerivedRD)->
346 isDerivedFrom(const_cast<CXXRecordDecl *>(BaseRD), Paths)) {
347 assert(false && "Class must be derived from the passed in base class!");
348 return FinalOverriders::BaseOffset();
349 }
350
351 assert(!Paths.getDetectedVirtual() && "FIXME: Handle virtual bases!");
352
353 // FIXME: We need to go through all paths here, just not the first one.
354 BaseOffset Offset = ComputeBaseOffset(Context, BaseRD, DerivedRD);
355
356 Offset.NonVirtualOffset = -Offset.NonVirtualOffset;
357 return Offset;
358}
359
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000360void FinalOverriders::PropagateOverrider(const CXXMethodDecl *OldMD,
Anders Carlssona678aa12010-02-13 22:23:31 +0000361 BaseSubobject NewBase,
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000362 const CXXMethodDecl *NewMD,
363 SubobjectOffsetsMapTy &Offsets) {
364 for (CXXMethodDecl::method_iterator I = OldMD->begin_overridden_methods(),
365 E = OldMD->end_overridden_methods(); I != E; ++I) {
366 const CXXMethodDecl *OverriddenMD = *I;
367 const CXXRecordDecl *OverriddenRD = OverriddenMD->getParent();
368
369 // We want to override OverriddenMD in all subobjects, for example:
370 //
371 /// struct A { virtual void f(); };
372 /// struct B1 : A { };
373 /// struct B2 : A { };
374 /// struct C : B1, B2 { virtual void f(); };
375 ///
376 /// When overriding A::f with C::f we need to do so in both A subobjects.
377 const OffsetVectorTy &OffsetVector = Offsets[OverriddenRD];
378
379 // Go through all the subobjects.
380 for (unsigned I = 0, E = OffsetVector.size(); I != E; ++I) {
381 uint64_t Offset = OffsetVector[I];
382
Anders Carlssoneee53d92010-02-13 02:02:03 +0000383 BaseSubobjectMethodPairTy SubobjectAndMethod =
384 std::make_pair(BaseSubobject(OverriddenRD, Offset), OverriddenMD);
385
386 OverriderInfo &Overrider = OverridersMap[SubobjectAndMethod];
387
Anders Carlsson136bd192010-02-12 16:55:34 +0000388 assert(Overrider.Method && "Did not find existing overrider!");
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000389
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000390 // Check if we need return adjustments or base adjustments.
Anders Carlssonc1290ad2010-02-13 21:16:54 +0000391 // (We don't want to do this for pure virtual member functions).
392 if (!NewMD->isPure()) {
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000393 // Get the return adjustment base offset.
Anders Carlssonc1290ad2010-02-13 21:16:54 +0000394 BaseOffset ReturnBaseOffset =
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000395 ComputeReturnAdjustmentBaseOffset(Context, NewMD, OverriddenMD);
396
Anders Carlssonc1290ad2010-02-13 21:16:54 +0000397 if (!ReturnBaseOffset.isEmpty()) {
398 // Store the return adjustment base offset.
399 ReturnAdjustments[SubobjectAndMethod] = ReturnBaseOffset;
400 }
Anders Carlssona678aa12010-02-13 22:23:31 +0000401
402 // Check if we need a 'this' adjustment base offset as well.
403 if (Offset != NewBase.getBaseOffset()) {
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000404 BaseOffset ThisBaseOffset =
405 ComputeThisAdjustmentBaseOffset(OverriddenMD->getParent(),
406 NewMD->getParent());
407 assert(!ThisBaseOffset.isEmpty() &&
408 "Should not get an empty 'this' adjustment!");
409
410 ThisAdjustments[SubobjectAndMethod] = ThisBaseOffset;
Anders Carlssona678aa12010-02-13 22:23:31 +0000411 }
Anders Carlsson8e661e12010-02-12 17:13:23 +0000412 }
413
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000414 // Set the new overrider.
Anders Carlsson136bd192010-02-12 16:55:34 +0000415 Overrider.Method = NewMD;
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000416
417 // And propagate it further.
Anders Carlssona678aa12010-02-13 22:23:31 +0000418 PropagateOverrider(OverriddenMD, NewBase, NewMD, Offsets);
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000419 }
420 }
421}
422
423void
424FinalOverriders::MergeSubobjectOffsets(const SubobjectOffsetsMapTy &NewOffsets,
425 SubobjectOffsetsMapTy &Offsets) {
426 // Iterate over the new offsets.
427 for (SubobjectOffsetsMapTy::const_iterator I = NewOffsets.begin(),
428 E = NewOffsets.end(); I != E; ++I) {
429 const CXXRecordDecl *NewRD = I->first;
Anders Carlsson3f9917d2010-02-13 21:33:22 +0000430 const OffsetVectorTy& NewOffsetVector = I->second;
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000431
Anders Carlsson3f9917d2010-02-13 21:33:22 +0000432 OffsetVectorTy &OffsetVector = Offsets[NewRD];
433 if (OffsetVector.empty()) {
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000434 // There were no previous offsets in this vector, just insert all entries
Anders Carlsson3f9917d2010-02-13 21:33:22 +0000435 // from the new offset vector.
436 OffsetVector.append(NewOffsetVector.begin(), NewOffsetVector.end());
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000437 continue;
438 }
439
Anders Carlsson3f9917d2010-02-13 21:33:22 +0000440 // We need to merge the new offsets vector into the old, but we don't want
441 // to have duplicate entries. Do this by inserting the old offsets in a set
442 // so they'll be unique. After this, we iterate over the new offset vector
443 // and only append elements that aren't in the set.
444
445 // First, add the existing offsets to the set.
446 llvm::SmallSet<uint64_t, 4> OffsetSet;
447 for (unsigned I = 0, E = OffsetVector.size(); I != E; ++I) {
448 bool Inserted = OffsetSet.insert(OffsetVector[I]);
449 if (!Inserted)
450 assert(false && "Set of offsets should be unique!");
451 }
452
453 // Next, only add the new offsets if they are not already in the set.
454 for (unsigned I = 0, E = NewOffsetVector.size(); I != E; ++I) {
455 uint64_t Offset = NewOffsetVector[I];
456
457 if (OffsetSet.count(Offset)) {
458 // Ignore the offset.
459 continue;
460 }
461
462 // Otherwise, add it to the offsets vector.
463 OffsetVector.push_back(Offset);
464 }
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000465 }
466}
467
468void FinalOverriders::ComputeFinalOverriders(BaseSubobject Base,
469 SubobjectOffsetsMapTy &Offsets) {
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000470 const CXXRecordDecl *RD = Base.getBase();
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000471 const ASTRecordLayout &Layout = Context.getASTRecordLayout(RD);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000472
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000473 SubobjectOffsetsMapTy NewOffsets;
474
475 for (CXXRecordDecl::base_class_const_iterator I = RD->bases_begin(),
476 E = RD->bases_end(); I != E; ++I) {
477 const CXXRecordDecl *BaseDecl =
478 cast<CXXRecordDecl>(I->getType()->getAs<RecordType>()->getDecl());
479
480 assert(!I->isVirtual() && "FIXME: Handle virtual bases!");
481
482 uint64_t BaseOffset = Layout.getBaseClassOffset(BaseDecl) +
483 Base.getBaseOffset();
484
485 // Compute the final overriders for this base.
486 ComputeFinalOverriders(BaseSubobject(BaseDecl, BaseOffset), NewOffsets);
487 }
488
489 /// Now add the overriders for this particular subobject.
490 AddOverriders(Base, NewOffsets);
491
492 // And merge the newly discovered subobject offsets.
493 MergeSubobjectOffsets(NewOffsets, Offsets);
494
495 /// Finally, add the offset for our own subobject.
496 Offsets[RD].push_back(Base.getBaseOffset());
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000497}
498
499void FinalOverriders::dump(llvm::raw_ostream &Out, BaseSubobject Base) const {
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000500 const CXXRecordDecl *RD = Base.getBase();
501 const ASTRecordLayout &Layout = Context.getASTRecordLayout(RD);
502
503 for (CXXRecordDecl::base_class_const_iterator I = RD->bases_begin(),
504 E = RD->bases_end(); I != E; ++I) {
505 assert(!I->isVirtual() && "FIXME: Handle virtual bases!");
506
507 const CXXRecordDecl *BaseDecl =
508 cast<CXXRecordDecl>(I->getType()->getAs<RecordType>()->getDecl());
509
510 uint64_t BaseOffset = Layout.getBaseClassOffset(BaseDecl) +
511 Base.getBaseOffset();
512
513 dump(Out, BaseSubobject(BaseDecl, BaseOffset));
514 }
515
516 Out << "Final overriders for (" << RD->getQualifiedNameAsString() << ", ";
517 Out << Base.getBaseOffset() << ")\n";
518
519 // Now dump the overriders for this base subobject.
520 for (CXXRecordDecl::method_iterator I = RD->method_begin(),
521 E = RD->method_end(); I != E; ++I) {
522 const CXXMethodDecl *MD = *I;
523
524 if (!MD->isVirtual())
525 continue;
526
Anders Carlsson136bd192010-02-12 16:55:34 +0000527 OverriderInfo Overrider = getOverrider(Base, MD);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000528
529 Out << " " << MD->getQualifiedNameAsString() << " - ";
Anders Carlsson8e661e12010-02-12 17:13:23 +0000530 Out << Overrider.Method->getQualifiedNameAsString();
Anders Carlssoneee53d92010-02-13 02:02:03 +0000531
Benjamin Kramer02c67372010-02-13 09:11:28 +0000532 AdjustmentOffsetsMapTy::const_iterator AI =
Anders Carlssoneee53d92010-02-13 02:02:03 +0000533 ReturnAdjustments.find(std::make_pair(Base, MD));
Benjamin Kramer02c67372010-02-13 09:11:28 +0000534 if (AI != ReturnAdjustments.end()) {
535 const BaseOffset &Offset = AI->second;
Anders Carlssonf3f67842010-02-13 20:41:15 +0000536
537 Out << " [ret-adj: ";
538 if (Offset.VirtualBase)
539 Out << Offset.VirtualBase->getQualifiedNameAsString() << " vbase, ";
Anders Carlssoneee53d92010-02-13 02:02:03 +0000540
Anders Carlssonf3f67842010-02-13 20:41:15 +0000541 Out << Offset.NonVirtualOffset << " nv]";
Anders Carlssoneee53d92010-02-13 02:02:03 +0000542 }
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000543
544 AI = ThisAdjustments.find(std::make_pair(Base, MD));
545 if (AI != ThisAdjustments.end()) {
546 const BaseOffset &Offset = AI->second;
547
548 Out << " [this-adj: ";
549 if (Offset.VirtualBase)
550 Out << Offset.VirtualBase->getQualifiedNameAsString() << " vbase, ";
551
552 Out << Offset.NonVirtualOffset << " nv]";
553 }
554
Anders Carlsson8e661e12010-02-12 17:13:23 +0000555 Out << "\n";
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000556 }
557}
558
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000559/// VtableComponent - Represents a single component in a vtable.
560class VtableComponent {
561public:
562 enum Kind {
563 CK_VCallOffset,
564 CK_VBaseOffset,
565 CK_OffsetToTop,
566 CK_RTTI,
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000567 CK_FunctionPointer,
568
569 /// CK_CompleteDtorPointer - A pointer to the complete destructor.
570 CK_CompleteDtorPointer,
571
572 /// CK_DeletingDtorPointer - A pointer to the deleting destructor.
573 CK_DeletingDtorPointer
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000574 };
575
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000576 static VtableComponent MakeOffsetToTop(int64_t Offset) {
577 return VtableComponent(CK_OffsetToTop, Offset);
578 }
579
580 static VtableComponent MakeRTTI(const CXXRecordDecl *RD) {
581 return VtableComponent(CK_RTTI, reinterpret_cast<uintptr_t>(RD));
582 }
583
584 static VtableComponent MakeFunction(const CXXMethodDecl *MD) {
585 assert(!isa<CXXDestructorDecl>(MD) &&
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000586 "Don't use MakeFunction with destructors!");
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000587
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000588 return VtableComponent(CK_FunctionPointer,
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000589 reinterpret_cast<uintptr_t>(MD));
590 }
591
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000592 static VtableComponent MakeCompleteDtor(const CXXDestructorDecl *DD) {
593 return VtableComponent(CK_CompleteDtorPointer,
594 reinterpret_cast<uintptr_t>(DD));
595 }
596
597 static VtableComponent MakeDeletingDtor(const CXXDestructorDecl *DD) {
598 return VtableComponent(CK_DeletingDtorPointer,
599 reinterpret_cast<uintptr_t>(DD));
600 }
601
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000602 /// getKind - Get the kind of this vtable component.
603 Kind getKind() const {
604 return (Kind)(Value & 0x7);
605 }
606
607 int64_t getOffsetToTop() const {
608 assert(getKind() == CK_OffsetToTop && "Invalid component kind!");
609
610 return getOffset();
611 }
612
613 const CXXRecordDecl *getRTTIDecl() const {
614 assert(getKind() == CK_RTTI && "Invalid component kind!");
615
616 return reinterpret_cast<CXXRecordDecl *>(getPointer());
617 }
618
619 const CXXMethodDecl *getFunctionDecl() const {
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000620 assert(getKind() == CK_FunctionPointer);
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000621
622 return reinterpret_cast<CXXMethodDecl *>(getPointer());
623 }
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000624
625 const CXXDestructorDecl *getDestructorDecl() const {
626 assert((getKind() == CK_CompleteDtorPointer ||
627 getKind() == CK_DeletingDtorPointer) && "Invalid component kind!");
628
629 return reinterpret_cast<CXXDestructorDecl *>(getPointer());
630 }
631
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000632private:
633 VtableComponent(Kind ComponentKind, int64_t Offset) {
634 assert((ComponentKind == CK_VCallOffset ||
635 ComponentKind == CK_VBaseOffset ||
636 ComponentKind == CK_OffsetToTop) && "Invalid component kind!");
637 assert(Offset <= ((1LL << 56) - 1) && "Offset is too big!");
638
639 Value = ((Offset << 3) | ComponentKind);
640 }
641
642 VtableComponent(Kind ComponentKind, uintptr_t Ptr) {
643 assert((ComponentKind == CK_RTTI ||
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000644 ComponentKind == CK_FunctionPointer ||
645 ComponentKind == CK_CompleteDtorPointer ||
646 ComponentKind == CK_DeletingDtorPointer) &&
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000647 "Invalid component kind!");
648
649 assert((Ptr & 7) == 0 && "Pointer not sufficiently aligned!");
650
651 Value = Ptr | ComponentKind;
652 }
653
654 int64_t getOffset() const {
655 assert((getKind() == CK_VCallOffset || getKind() == CK_VBaseOffset ||
656 getKind() == CK_OffsetToTop) && "Invalid component kind!");
657
658 return Value >> 3;
659 }
660
661 uintptr_t getPointer() const {
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000662 assert((getKind() == CK_RTTI ||
663 getKind() == CK_FunctionPointer ||
664 getKind() == CK_CompleteDtorPointer ||
665 getKind() == CK_DeletingDtorPointer) &&
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000666 "Invalid component kind!");
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000667
668 return static_cast<uintptr_t>(Value & ~7ULL);
669 }
670
671 /// The kind is stored in the lower 3 bits of the value. For offsets, we
672 /// make use of the facts that classes can't be larger than 2^55 bytes,
673 /// so we store the offset in the lower part of the 61 bytes that remain.
674 /// (The reason that we're not simply using a PointerIntPair here is that we
675 /// need the offsets to be 64-bit, even when on a 32-bit machine).
676 int64_t Value;
677};
678
679/// VtableBuilder - Class for building vtable layout information.
Benjamin Kramer337e3a52009-11-28 19:45:26 +0000680class VtableBuilder {
Anders Carlsson65b49782010-02-12 05:25:12 +0000681public:
682 /// PrimaryBasesSetTy - A set of direct and indirect primary bases.
683 typedef llvm::SmallPtrSet<const CXXRecordDecl *, 8> PrimaryBasesSetTy;
684
685private:
Anders Carlssoncf5a8822010-02-13 21:07:32 +0000686 /// VtableInfo - Global vtable information.
687 CGVtableInfo &VtableInfo;
688
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000689 /// MostDerivedClass - The most derived class for which we're building this
690 /// vtable.
691 const CXXRecordDecl *MostDerivedClass;
692
693 /// Context - The ASTContext which we will use for layout information.
Anders Carlsson65b49782010-02-12 05:25:12 +0000694 ASTContext &Context;
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000695
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000696 /// FinalOverriders - The final overriders of the most derived class.
697 FinalOverriders Overriders;
698
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000699 /// Components - The components of the vtable being built.
700 llvm::SmallVector<VtableComponent, 64> Components;
701
Anders Carlsson932c2f22010-02-11 17:18:51 +0000702 /// AddressPoints - Address points for the vtable being built.
703 CGVtableInfo::AddressPointsMapTy AddressPoints;
704
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000705 /// ReturnAdjustment - A return adjustment.
Anders Carlsson5272c252010-02-12 17:37:14 +0000706 struct ReturnAdjustment {
707 /// NonVirtual - The non-virtual adjustment from the derived object to its
708 /// nearest virtual base.
709 int64_t NonVirtual;
710
Anders Carlssoncf5a8822010-02-13 21:07:32 +0000711 /// VBaseOffsetOffset - The offset, in bytes, relative to the address point
712 /// of the virtual base class offset.
713 int64_t VBaseOffsetOffset;
Anders Carlsson5272c252010-02-12 17:37:14 +0000714
Anders Carlssoncf5a8822010-02-13 21:07:32 +0000715 ReturnAdjustment() : NonVirtual(0), VBaseOffsetOffset(0) { }
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000716
Anders Carlssoncf5a8822010-02-13 21:07:32 +0000717 bool isEmpty() const { return !NonVirtual && !VBaseOffsetOffset; }
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000718 };
719
720 /// ReturnAdjustments - The return adjustments needed in this vtable.
721 llvm::SmallVector<std::pair<uint64_t, ReturnAdjustment>, 16>
722 ReturnAdjustments;
723
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000724 /// ThisAdjustment - A 'this' pointer adjustment thunk.
725 struct ThisAdjustment {
726 /// NonVirtual - The non-virtual adjustment from the derived object to its
727 /// nearest virtual base.
728 int64_t NonVirtual;
729
730 /// FIXME: Add VCallOffsetOffset here.
731
732 ThisAdjustment() : NonVirtual(0) { }
733
734 bool isEmpty() const { return !NonVirtual; }
735 };
736
737 /// ThisAdjustments - The 'this' pointer adjustments needed in this vtable.
738 llvm::SmallVector<std::pair<uint64_t, ThisAdjustment>, 16>
739 ThisAdjustments;
740
Anders Carlssoncf5a8822010-02-13 21:07:32 +0000741 /// ComputeReturnAdjustment - Compute the return adjustment given a return
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000742 /// adjustment base offset.
743 ReturnAdjustment ComputeReturnAdjustment(FinalOverriders::BaseOffset Offset);
744
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000745 /// ComputeThisAdjustment - Compute the 'this' pointer adjustment given a
746 /// 'this' pointer adjustment base offset.
747 ThisAdjustment ComputeThisAdjustment(FinalOverriders::BaseOffset Offset);
748
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000749 /// AddMethod - Add a single virtual member function to the vtable
750 /// components vector.
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000751 void AddMethod(const CXXMethodDecl *MD, ReturnAdjustment ReturnAdjustment,
752 ThisAdjustment ThisAdjustment);
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000753
754 /// AddMethods - Add the methods of this base subobject and all its
755 /// primary bases to the vtable components vector.
756 void AddMethods(BaseSubobject Base, PrimaryBasesSetTy &PrimaryBases);
Anders Carlsson65b49782010-02-12 05:25:12 +0000757
Anders Carlssone75aaa92010-02-13 22:05:23 +0000758 /// layoutVtable - Layout a vtable and all its secondary vtables.
759 void layoutVtable(BaseSubobject Base);
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000760
761public:
Anders Carlssoncf5a8822010-02-13 21:07:32 +0000762 VtableBuilder(CGVtableInfo &VtableInfo, const CXXRecordDecl *MostDerivedClass)
763 : VtableInfo(VtableInfo), MostDerivedClass(MostDerivedClass),
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000764 Context(MostDerivedClass->getASTContext()), Overriders(MostDerivedClass) {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000765
Anders Carlssone75aaa92010-02-13 22:05:23 +0000766 layoutVtable(BaseSubobject(MostDerivedClass, 0));
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000767 }
768
769 /// dumpLayout - Dump the vtable layout.
770 void dumpLayout(llvm::raw_ostream&);
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000771};
772
Anders Carlsson65b49782010-02-12 05:25:12 +0000773/// OverridesMethodInPrimaryBase - Checks whether whether this virtual member
774/// function overrides a member function in a direct or indirect primary base.
775/// Returns the overridden member function, or null if none was found.
776static const CXXMethodDecl *
777OverridesMethodInPrimaryBase(const CXXMethodDecl *MD,
778 VtableBuilder::PrimaryBasesSetTy &PrimaryBases) {
779 for (CXXMethodDecl::method_iterator I = MD->begin_overridden_methods(),
780 E = MD->end_overridden_methods(); I != E; ++I) {
781 const CXXMethodDecl *OverriddenMD = *I;
782 const CXXRecordDecl *OverriddenRD = OverriddenMD->getParent();
783 assert(OverriddenMD->isCanonicalDecl() &&
784 "Should have the canonical decl of the overridden RD!");
785
786 if (PrimaryBases.count(OverriddenRD))
787 return OverriddenMD;
788 }
789
790 return 0;
791}
792
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000793VtableBuilder::ReturnAdjustment
794VtableBuilder::ComputeReturnAdjustment(FinalOverriders::BaseOffset Offset) {
795 ReturnAdjustment Adjustment;
796
797 if (!Offset.isEmpty()) {
Anders Carlssoncf5a8822010-02-13 21:07:32 +0000798 if (Offset.VirtualBase) {
799 // Get the virtual base offset offset.
800 Adjustment.VBaseOffsetOffset =
801 VtableInfo.getVirtualBaseOffsetIndex(Offset.DerivedClass,
802 Offset.VirtualBase);
803 // FIXME: Once the assert in getVirtualBaseOffsetIndex is back again,
804 // we can get rid of this assert.
805 assert(Adjustment.VBaseOffsetOffset != 0 &&
806 "Invalid base offset offset!");
807 }
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000808
809 Adjustment.NonVirtual = Offset.NonVirtualOffset;
810 }
811
812 return Adjustment;
813}
814
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000815VtableBuilder::ThisAdjustment
816VtableBuilder::ComputeThisAdjustment(FinalOverriders::BaseOffset Offset) {
817 ThisAdjustment Adjustment;
818
819 if (!Offset.isEmpty()) {
820 assert(!Offset.VirtualBase && "FIXME: Handle virtual bases!");
821 Adjustment.NonVirtual = Offset.NonVirtualOffset;
822 }
823
824 return Adjustment;
825}
826
Anders Carlsson65b49782010-02-12 05:25:12 +0000827void
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000828VtableBuilder::AddMethod(const CXXMethodDecl *MD,
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000829 ReturnAdjustment ReturnAdjustment,
830 ThisAdjustment ThisAdjustment) {
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000831 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
832 assert(ReturnAdjustment.isEmpty() &&
833 "Destructor can't have return adjustment!");
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000834 // Add the 'this' pointer adjustments if necessary.
835 if (!ThisAdjustment.isEmpty()) {
836 ThisAdjustments.push_back(std::make_pair(Components.size(),
837 ThisAdjustment));
838 ThisAdjustments.push_back(std::make_pair(Components.size() + 1,
839 ThisAdjustment));
840 }
841
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000842 // Add both the complete destructor and the deleting destructor.
843 Components.push_back(VtableComponent::MakeCompleteDtor(DD));
844 Components.push_back(VtableComponent::MakeDeletingDtor(DD));
845 } else {
846 // Add the return adjustment if necessary.
847 if (!ReturnAdjustment.isEmpty())
848 ReturnAdjustments.push_back(std::make_pair(Components.size(),
849 ReturnAdjustment));
850
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000851 // Add the 'this' pointer adjustment if necessary.
852 if (!ThisAdjustment.isEmpty())
853 ThisAdjustments.push_back(std::make_pair(Components.size(),
854 ThisAdjustment));
855
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000856 // Add the function.
857 Components.push_back(VtableComponent::MakeFunction(MD));
858 }
859}
860
861void
862VtableBuilder::AddMethods(BaseSubobject Base, PrimaryBasesSetTy &PrimaryBases) {
Anders Carlsson65b49782010-02-12 05:25:12 +0000863 const CXXRecordDecl *RD = Base.getBase();
864
865 const ASTRecordLayout &Layout = Context.getASTRecordLayout(RD);
866
867 if (const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase()) {
868 if (Layout.getPrimaryBaseWasVirtual())
869 assert(false && "FIXME: Handle vbases here.");
870 else
871 assert(Layout.getBaseClassOffset(PrimaryBase) == 0 &&
872 "Primary base should have a zero offset!");
873
Anders Carlssone75aaa92010-02-13 22:05:23 +0000874 AddMethods(BaseSubobject(PrimaryBase, Base.getBaseOffset()), PrimaryBases);
Anders Carlsson65b49782010-02-12 05:25:12 +0000875
876 if (!PrimaryBases.insert(PrimaryBase))
877 assert(false && "Found a duplicate primary base!");
878 }
Anders Carlsson932c2f22010-02-11 17:18:51 +0000879
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000880 // Now go through all virtual member functions and add them.
881 for (CXXRecordDecl::method_iterator I = RD->method_begin(),
882 E = RD->method_end(); I != E; ++I) {
883 const CXXMethodDecl *MD = *I;
Anders Carlsson65b49782010-02-12 05:25:12 +0000884
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000885 if (!MD->isVirtual())
886 continue;
Anders Carlsson65b49782010-02-12 05:25:12 +0000887
888 // Get the final overrider.
Anders Carlsson136bd192010-02-12 16:55:34 +0000889 FinalOverriders::OverriderInfo Overrider =
890 Overriders.getOverrider(Base, MD);
Anders Carlsson65b49782010-02-12 05:25:12 +0000891
892 // Check if this virtual member function overrides a method in a primary
893 // base. If this is the case, and the return type doesn't require adjustment
894 // then we can just use the member function from the primary base.
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000895 if (const CXXMethodDecl *OverriddenMD =
896 OverridesMethodInPrimaryBase(MD, PrimaryBases)) {
Anders Carlssoncce9ee82010-02-13 23:17:31 +0000897 if (ComputeReturnAdjustmentBaseOffset(Context, MD,
898 OverriddenMD).isEmpty())
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000899 continue;
900 }
Anders Carlssoneee53d92010-02-13 02:02:03 +0000901
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000902 // Check if this overrider needs a return adjustment.
903 FinalOverriders::BaseOffset ReturnAdjustmentOffset =
904 Overriders.getReturnAdjustmentOffset(Base, MD);
905
906 ReturnAdjustment ReturnAdjustment =
907 ComputeReturnAdjustment(ReturnAdjustmentOffset);
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000908
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000909 // Check if this overrider needs a 'this' pointer adjustment.
910 FinalOverriders::BaseOffset ThisAdjustmentOffset =
911 Overriders.getThisAdjustmentOffset(Base, MD);
912
913 ThisAdjustment ThisAdjustment = ComputeThisAdjustment(ThisAdjustmentOffset);
914
915 AddMethod(Overrider.Method, ReturnAdjustment, ThisAdjustment);
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000916 }
917}
918
Anders Carlssone75aaa92010-02-13 22:05:23 +0000919void VtableBuilder::layoutVtable(BaseSubobject Base) {
Anders Carlsson65b49782010-02-12 05:25:12 +0000920 const CXXRecordDecl *RD = Base.getBase();
921
922 // First, add the offset to top.
Anders Carlssone75aaa92010-02-13 22:05:23 +0000923 // FIXME: This is not going to be right for construction vtables.
924 // FIXME: We should not use -8 here.
925 int64_t OffsetToTop = -(int64_t)Base.getBaseOffset() / 8;
926 Components.push_back(VtableComponent::MakeOffsetToTop(OffsetToTop));
Anders Carlsson65b49782010-02-12 05:25:12 +0000927
928 // Next, add the RTTI.
Anders Carlssone75aaa92010-02-13 22:05:23 +0000929 Components.push_back(VtableComponent::MakeRTTI(MostDerivedClass));
Anders Carlsson65b49782010-02-12 05:25:12 +0000930
Anders Carlssone8a97562010-02-12 07:43:48 +0000931 uint64_t AddressPoint = Components.size();
Anders Carlsson65b49782010-02-12 05:25:12 +0000932
933 // Now go through all virtual member functions and add them.
934 PrimaryBasesSetTy PrimaryBases;
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000935 AddMethods(Base, PrimaryBases);
Anders Carlsson65b49782010-02-12 05:25:12 +0000936
Anders Carlssone8a97562010-02-12 07:43:48 +0000937 // Record the address point.
938 AddressPoints.insert(std::make_pair(Base, AddressPoint));
939
940 // Record the address points for all primary bases.
941 for (PrimaryBasesSetTy::const_iterator I = PrimaryBases.begin(),
942 E = PrimaryBases.end(); I != E; ++I) {
943 const CXXRecordDecl *BaseDecl = *I;
944
945 // We know that all the primary bases have the same offset as the base
946 // subobject.
947 BaseSubobject PrimaryBase(BaseDecl, Base.getBaseOffset());
948 AddressPoints.insert(std::make_pair(PrimaryBase, AddressPoint));
949 }
950
Anders Carlsson65b49782010-02-12 05:25:12 +0000951 const ASTRecordLayout &Layout = Context.getASTRecordLayout(RD);
952 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
953
Anders Carlssone75aaa92010-02-13 22:05:23 +0000954 // Layout secondary vtables.
Anders Carlsson65b49782010-02-12 05:25:12 +0000955 for (CXXRecordDecl::base_class_const_iterator I = RD->bases_begin(),
956 E = RD->bases_end(); I != E; ++I) {
957 const CXXRecordDecl *BaseDecl =
958 cast<CXXRecordDecl>(I->getType()->getAs<RecordType>()->getDecl());
959
960 // Ignore the primary base.
961 if (BaseDecl == PrimaryBase)
962 continue;
963
964 assert(!I->isVirtual() && "FIXME: Handle virtual bases");
965
Anders Carlssone75aaa92010-02-13 22:05:23 +0000966 // Get the base offset of this base.
967 uint64_t BaseOffset = Base.getBaseOffset() +
968 Layout.getBaseClassOffset(BaseDecl);
969
970 // Layout this secondary vtable.
971 layoutVtable(BaseSubobject(BaseDecl, BaseOffset));
Anders Carlsson65b49782010-02-12 05:25:12 +0000972 }
973}
974
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000975/// dumpLayout - Dump the vtable layout.
976void VtableBuilder::dumpLayout(llvm::raw_ostream& Out) {
977
978 Out << "Vtable for '" << MostDerivedClass->getQualifiedNameAsString();
979 Out << "' (" << Components.size() << " entries).\n";
980
Anders Carlsson932c2f22010-02-11 17:18:51 +0000981 // Iterate through the address points and insert them into a new map where
982 // they are keyed by the index and not the base object.
983 // Since an address point can be shared by multiple subobjects, we use an
984 // STL multimap.
985 std::multimap<uint64_t, BaseSubobject> AddressPointsByIndex;
986 for (CGVtableInfo::AddressPointsMapTy::const_iterator I =
987 AddressPoints.begin(), E = AddressPoints.end(); I != E; ++I) {
988 const BaseSubobject& Base = I->first;
989 uint64_t Index = I->second;
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000990
Anders Carlsson932c2f22010-02-11 17:18:51 +0000991 AddressPointsByIndex.insert(std::make_pair(Index, Base));
992 }
993
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +0000994 unsigned NextReturnAdjustmentIndex = 0;
Anders Carlssond3adb0c2010-02-13 23:40:17 +0000995 unsigned NextThisAdjustmentIndex = 0;
Anders Carlsson932c2f22010-02-11 17:18:51 +0000996 for (unsigned I = 0, E = Components.size(); I != E; ++I) {
Anders Carlssone8a97562010-02-12 07:43:48 +0000997 uint64_t Index = I;
998
Anders Carlsson932c2f22010-02-11 17:18:51 +0000999 if (AddressPointsByIndex.count(I)) {
Anders Carlssone8a97562010-02-12 07:43:48 +00001000 if (AddressPointsByIndex.count(Index) == 1) {
1001 const BaseSubobject &Base = AddressPointsByIndex.find(Index)->second;
1002
1003 // FIXME: Instead of dividing by 8, we should be using CharUnits.
1004 Out << " -- (" << Base.getBase()->getQualifiedNameAsString();
1005 Out << ", " << Base.getBaseOffset() / 8 << ") vtable address --\n";
1006 } else {
1007 uint64_t BaseOffset =
1008 AddressPointsByIndex.lower_bound(Index)->second.getBaseOffset();
1009
1010 // We store the class names in a set to get a stable order.
1011 std::set<std::string> ClassNames;
1012 for (std::multimap<uint64_t, BaseSubobject>::const_iterator I =
1013 AddressPointsByIndex.lower_bound(Index), E =
1014 AddressPointsByIndex.upper_bound(Index); I != E; ++I) {
1015 assert(I->second.getBaseOffset() == BaseOffset &&
1016 "Invalid base offset!");
1017 const CXXRecordDecl *RD = I->second.getBase();
1018 ClassNames.insert(RD->getQualifiedNameAsString());
1019 }
1020
1021 for (std::set<std::string>::const_iterator I = ClassNames.begin(),
1022 E = ClassNames.end(); I != E; ++I) {
1023 // FIXME: Instead of dividing by 8, we should be using CharUnits.
1024 Out << " -- (" << *I;
1025 Out << ", " << BaseOffset / 8 << ") vtable address --\n";
1026 }
1027 }
Anders Carlsson932c2f22010-02-11 17:18:51 +00001028 }
1029
1030 Out << llvm::format("%4d | ", I);
1031
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001032 const VtableComponent &Component = Components[I];
1033
1034 // Dump the component.
1035 switch (Component.getKind()) {
1036 // FIXME: Remove this default case.
1037 default:
1038 assert(false && "Unhandled component kind!");
1039 break;
1040
1041 case VtableComponent::CK_OffsetToTop:
1042 Out << "offset_to_top (" << Component.getOffsetToTop() << ")";
1043 break;
1044
1045 case VtableComponent::CK_RTTI:
1046 Out << Component.getRTTIDecl()->getQualifiedNameAsString() << " RTTI";
1047 break;
1048
Anders Carlsson5bd8d192010-02-11 18:20:28 +00001049 case VtableComponent::CK_FunctionPointer: {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001050 const CXXMethodDecl *MD = Component.getFunctionDecl();
1051
Anders Carlsson5bd8d192010-02-11 18:20:28 +00001052 std::string Str =
1053 PredefinedExpr::ComputeName(PredefinedExpr::PrettyFunctionNoVirtual,
1054 MD);
1055 Out << Str;
Anders Carlsson09da3372010-02-12 02:38:13 +00001056 if (MD->isPure())
1057 Out << " [pure]";
1058
Anders Carlssond3adb0c2010-02-13 23:40:17 +00001059 // If this function pointer has a return adjustment, dump it.
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +00001060 if (NextReturnAdjustmentIndex < ReturnAdjustments.size() &&
1061 ReturnAdjustments[NextReturnAdjustmentIndex].first == I) {
1062 const ReturnAdjustment Adjustment =
1063 ReturnAdjustments[NextReturnAdjustmentIndex].second;
1064
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +00001065 Out << "\n [return adjustment: ";
Anders Carlssoncf5a8822010-02-13 21:07:32 +00001066 Out << Adjustment.NonVirtual << " non-virtual";
1067
1068 if (Adjustment.VBaseOffsetOffset)
1069 Out << ", " << Adjustment.VBaseOffsetOffset << " vbase offset offset";
1070 Out << ']';
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +00001071
1072 NextReturnAdjustmentIndex++;
1073 }
Anders Carlssond3adb0c2010-02-13 23:40:17 +00001074
1075 // If this function pointer has a 'this' pointer adjustment, dump it.
1076 if (NextThisAdjustmentIndex < ThisAdjustments.size() &&
1077 ThisAdjustments[NextThisAdjustmentIndex].first == I) {
1078 const ThisAdjustment Adjustment =
1079 ThisAdjustments[NextThisAdjustmentIndex].second;
1080
1081 Out << "\n [this adjustment: ";
1082 Out << Adjustment.NonVirtual << " non-virtual";
1083
1084 Out << ']';
1085
1086 NextThisAdjustmentIndex++;
1087 }
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +00001088
Anders Carlsson5bd8d192010-02-11 18:20:28 +00001089 break;
1090 }
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001091
Anders Carlsson5bd8d192010-02-11 18:20:28 +00001092 case VtableComponent::CK_CompleteDtorPointer: {
1093 const CXXDestructorDecl *DD = Component.getDestructorDecl();
1094
1095 Out << DD->getQualifiedNameAsString() << "() [complete]";
Anders Carlsson09da3372010-02-12 02:38:13 +00001096 if (DD->isPure())
1097 Out << " [pure]";
1098
Anders Carlsson5bd8d192010-02-11 18:20:28 +00001099 break;
1100 }
1101
1102 case VtableComponent::CK_DeletingDtorPointer: {
1103 const CXXDestructorDecl *DD = Component.getDestructorDecl();
1104
1105 Out << DD->getQualifiedNameAsString() << "() [deleting]";
Anders Carlsson09da3372010-02-12 02:38:13 +00001106 if (DD->isPure())
1107 Out << " [pure]";
1108
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001109 break;
1110 }
1111
1112 }
Anders Carlsson932c2f22010-02-11 17:18:51 +00001113
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001114 Out << '\n';
1115 }
1116
1117}
1118
1119}
1120
1121namespace {
1122class OldVtableBuilder {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001123public:
1124 /// Index_t - Vtable index type.
1125 typedef uint64_t Index_t;
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001126 typedef std::vector<std::pair<GlobalDecl,
1127 std::pair<GlobalDecl, ThunkAdjustment> > >
1128 SavedAdjustmentsVectorTy;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001129private:
Anders Carlsson472404f2009-12-04 16:19:30 +00001130
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001131 // VtableComponents - The components of the vtable being built.
1132 typedef llvm::SmallVector<llvm::Constant *, 64> VtableComponentsVectorTy;
1133 VtableComponentsVectorTy VtableComponents;
1134
Eli Friedman6c08ce72009-12-05 01:05:03 +00001135 const bool BuildVtable;
1136
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001137 llvm::Type *Ptr8Ty;
Anders Carlssonbad80eb2009-12-04 18:36:22 +00001138
1139 /// MostDerivedClass - The most derived class that this vtable is being
1140 /// built for.
1141 const CXXRecordDecl *MostDerivedClass;
1142
Mike Stump83066c82009-11-13 01:54:23 +00001143 /// LayoutClass - The most derived class used for virtual base layout
1144 /// information.
1145 const CXXRecordDecl *LayoutClass;
Mike Stump653d0b92009-11-13 02:13:54 +00001146 /// LayoutOffset - The offset for Class in LayoutClass.
1147 uint64_t LayoutOffset;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001148 /// BLayout - Layout for the most derived class that this vtable is being
1149 /// built for.
1150 const ASTRecordLayout &BLayout;
1151 llvm::SmallSet<const CXXRecordDecl *, 32> IndirectPrimary;
1152 llvm::SmallSet<const CXXRecordDecl *, 32> SeenVBase;
1153 llvm::Constant *rtti;
1154 llvm::LLVMContext &VMContext;
1155 CodeGenModule &CGM; // Per-module state.
Anders Carlssonf2f31f42009-12-04 03:46:21 +00001156
Mike Stump90181eb2010-01-26 00:05:04 +00001157 llvm::DenseMap<const CXXMethodDecl *, Index_t> VCall;
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001158 llvm::DenseMap<GlobalDecl, Index_t> VCallOffset;
Mike Stump77537b12010-01-26 03:42:22 +00001159 llvm::DenseMap<GlobalDecl, Index_t> VCallOffsetForVCall;
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001160 // This is the offset to the nearest virtual base
Mike Stump90181eb2010-01-26 00:05:04 +00001161 llvm::DenseMap<const CXXMethodDecl *, Index_t> NonVirtualOffset;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001162 llvm::DenseMap<const CXXRecordDecl *, Index_t> VBIndex;
Mike Stumpbb9ff052009-10-27 23:46:47 +00001163
Anders Carlsson323bb042009-11-26 19:54:33 +00001164 /// PureVirtualFunction - Points to __cxa_pure_virtual.
1165 llvm::Constant *PureVirtualFn;
1166
Anders Carlssona84b6e82009-12-04 02:01:07 +00001167 /// VtableMethods - A data structure for keeping track of methods in a vtable.
1168 /// Can add methods, override methods and iterate in vtable order.
1169 class VtableMethods {
1170 // MethodToIndexMap - Maps from a global decl to the index it has in the
1171 // Methods vector.
1172 llvm::DenseMap<GlobalDecl, uint64_t> MethodToIndexMap;
1173
1174 /// Methods - The methods, in vtable order.
1175 typedef llvm::SmallVector<GlobalDecl, 16> MethodsVectorTy;
1176 MethodsVectorTy Methods;
Eli Friedman8174f2c2009-12-06 22:01:30 +00001177 MethodsVectorTy OrigMethods;
Anders Carlssona84b6e82009-12-04 02:01:07 +00001178
1179 public:
1180 /// AddMethod - Add a method to the vtable methods.
1181 void AddMethod(GlobalDecl GD) {
1182 assert(!MethodToIndexMap.count(GD) &&
1183 "Method has already been added!");
1184
1185 MethodToIndexMap[GD] = Methods.size();
1186 Methods.push_back(GD);
Eli Friedman8174f2c2009-12-06 22:01:30 +00001187 OrigMethods.push_back(GD);
Anders Carlssona84b6e82009-12-04 02:01:07 +00001188 }
1189
1190 /// OverrideMethod - Replace a method with another.
1191 void OverrideMethod(GlobalDecl OverriddenGD, GlobalDecl GD) {
1192 llvm::DenseMap<GlobalDecl, uint64_t>::iterator i
1193 = MethodToIndexMap.find(OverriddenGD);
1194 assert(i != MethodToIndexMap.end() && "Did not find entry!");
1195
1196 // Get the index of the old decl.
1197 uint64_t Index = i->second;
1198
1199 // Replace the old decl with the new decl.
1200 Methods[Index] = GD;
1201
Anders Carlssona84b6e82009-12-04 02:01:07 +00001202 // And add the new.
1203 MethodToIndexMap[GD] = Index;
1204 }
1205
Anders Carlsson7bb70762009-12-04 15:49:02 +00001206 /// getIndex - Gives the index of a passed in GlobalDecl. Returns false if
1207 /// the index couldn't be found.
Benjamin Kramer62ab6162009-12-04 22:45:27 +00001208 bool getIndex(GlobalDecl GD, uint64_t &Index) const {
Anders Carlsson7bb70762009-12-04 15:49:02 +00001209 llvm::DenseMap<GlobalDecl, uint64_t>::const_iterator i
1210 = MethodToIndexMap.find(GD);
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001211
Anders Carlsson7bb70762009-12-04 15:49:02 +00001212 if (i == MethodToIndexMap.end())
1213 return false;
Anders Carlssone6096362009-12-04 03:41:37 +00001214
Anders Carlsson7bb70762009-12-04 15:49:02 +00001215 Index = i->second;
1216 return true;
Anders Carlssone6096362009-12-04 03:41:37 +00001217 }
1218
Eli Friedman8174f2c2009-12-06 22:01:30 +00001219 GlobalDecl getOrigMethod(uint64_t Index) const {
1220 return OrigMethods[Index];
1221 }
1222
Anders Carlssona84b6e82009-12-04 02:01:07 +00001223 MethodsVectorTy::size_type size() const {
1224 return Methods.size();
1225 }
1226
1227 void clear() {
1228 MethodToIndexMap.clear();
1229 Methods.clear();
Eli Friedman8174f2c2009-12-06 22:01:30 +00001230 OrigMethods.clear();
Anders Carlssona84b6e82009-12-04 02:01:07 +00001231 }
1232
Anders Carlssone6096362009-12-04 03:41:37 +00001233 GlobalDecl operator[](uint64_t Index) const {
Anders Carlssona84b6e82009-12-04 02:01:07 +00001234 return Methods[Index];
1235 }
1236 };
1237
1238 /// Methods - The vtable methods we're currently building.
1239 VtableMethods Methods;
1240
Anders Carlsson4c837d22009-12-04 02:26:15 +00001241 /// ThisAdjustments - For a given index in the vtable, contains the 'this'
1242 /// pointer adjustment needed for a method.
1243 typedef llvm::DenseMap<uint64_t, ThunkAdjustment> ThisAdjustmentsMapTy;
1244 ThisAdjustmentsMapTy ThisAdjustments;
Anders Carlssond420a312009-11-26 19:32:45 +00001245
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001246 SavedAdjustmentsVectorTy SavedAdjustments;
Eli Friedman8174f2c2009-12-06 22:01:30 +00001247
Anders Carlssonc521f952009-12-04 02:22:02 +00001248 /// BaseReturnTypes - Contains the base return types of methods who have been
1249 /// overridden with methods whose return types require adjustment. Used for
1250 /// generating covariant thunk information.
1251 typedef llvm::DenseMap<uint64_t, CanQualType> BaseReturnTypesMapTy;
1252 BaseReturnTypesMapTy BaseReturnTypes;
Anders Carlssond420a312009-11-26 19:32:45 +00001253
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001254 std::vector<Index_t> VCalls;
Mike Stump2cefe382009-11-12 20:47:57 +00001255
1256 typedef std::pair<const CXXRecordDecl *, uint64_t> CtorVtable_t;
Mike Stumpcd2b8212009-11-19 20:52:19 +00001257 // subAddressPoints - Used to hold the AddressPoints (offsets) into the built
1258 // vtable for use in computing the initializers for the VTT.
1259 llvm::DenseMap<CtorVtable_t, int64_t> &subAddressPoints;
Mike Stump2cefe382009-11-12 20:47:57 +00001260
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001261 /// AddressPoints - Address points for this vtable.
1262 CGVtableInfo::AddressPointsMapTy& AddressPoints;
1263
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001264 typedef CXXRecordDecl::method_iterator method_iter;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001265 const uint32_t LLVMPointerWidth;
1266 Index_t extra;
Mike Stump37dbe962009-10-15 02:04:03 +00001267 typedef std::vector<std::pair<const CXXRecordDecl *, int64_t> > Path_t;
Mike Stumpcd2b8212009-11-19 20:52:19 +00001268 static llvm::DenseMap<CtorVtable_t, int64_t>&
1269 AllocAddressPoint(CodeGenModule &cgm, const CXXRecordDecl *l,
1270 const CXXRecordDecl *c) {
Anders Carlsson93a18842010-01-02 18:02:32 +00001271 CGVtableInfo::AddrMap_t *&oref = cgm.getVtableInfo().AddressPoints[l];
Mike Stumpcd2b8212009-11-19 20:52:19 +00001272 if (oref == 0)
Anders Carlsson93a18842010-01-02 18:02:32 +00001273 oref = new CGVtableInfo::AddrMap_t;
Mike Stumpcd2b8212009-11-19 20:52:19 +00001274
1275 llvm::DenseMap<CtorVtable_t, int64_t> *&ref = (*oref)[c];
1276 if (ref == 0)
1277 ref = new llvm::DenseMap<CtorVtable_t, int64_t>;
1278 return *ref;
1279 }
Anders Carlsson323bb042009-11-26 19:54:33 +00001280
Mike Stump90181eb2010-01-26 00:05:04 +00001281 bool DclIsSame(const FunctionDecl *New, const FunctionDecl *Old) {
1282 FunctionTemplateDecl *OldTemplate = Old->getDescribedFunctionTemplate();
1283 FunctionTemplateDecl *NewTemplate = New->getDescribedFunctionTemplate();
1284
1285 // C++ [temp.fct]p2:
1286 // A function template can be overloaded with other function templates
1287 // and with normal (non-template) functions.
1288 if ((OldTemplate == 0) != (NewTemplate == 0))
1289 return false;
1290
1291 // Is the function New an overload of the function Old?
1292 QualType OldQType = CGM.getContext().getCanonicalType(Old->getType());
1293 QualType NewQType = CGM.getContext().getCanonicalType(New->getType());
1294
1295 // Compare the signatures (C++ 1.3.10) of the two functions to
1296 // determine whether they are overloads. If we find any mismatch
1297 // in the signature, they are overloads.
1298
1299 // If either of these functions is a K&R-style function (no
1300 // prototype), then we consider them to have matching signatures.
1301 if (isa<FunctionNoProtoType>(OldQType.getTypePtr()) ||
1302 isa<FunctionNoProtoType>(NewQType.getTypePtr()))
1303 return true;
1304
1305 FunctionProtoType* OldType = cast<FunctionProtoType>(OldQType);
1306 FunctionProtoType* NewType = cast<FunctionProtoType>(NewQType);
1307
1308 // The signature of a function includes the types of its
1309 // parameters (C++ 1.3.10), which includes the presence or absence
1310 // of the ellipsis; see C++ DR 357).
1311 if (OldQType != NewQType &&
1312 (OldType->getNumArgs() != NewType->getNumArgs() ||
1313 OldType->isVariadic() != NewType->isVariadic() ||
1314 !std::equal(OldType->arg_type_begin(), OldType->arg_type_end(),
1315 NewType->arg_type_begin())))
1316 return false;
1317
1318#if 0
1319 // C++ [temp.over.link]p4:
1320 // The signature of a function template consists of its function
1321 // signature, its return type and its template parameter list. The names
1322 // of the template parameters are significant only for establishing the
1323 // relationship between the template parameters and the rest of the
1324 // signature.
1325 //
1326 // We check the return type and template parameter lists for function
1327 // templates first; the remaining checks follow.
1328 if (NewTemplate &&
1329 (!TemplateParameterListsAreEqual(NewTemplate->getTemplateParameters(),
1330 OldTemplate->getTemplateParameters(),
1331 TPL_TemplateMatch) ||
1332 OldType->getResultType() != NewType->getResultType()))
1333 return false;
1334#endif
1335
1336 // If the function is a class member, its signature includes the
1337 // cv-qualifiers (if any) on the function itself.
1338 //
1339 // As part of this, also check whether one of the member functions
1340 // is static, in which case they are not overloads (C++
1341 // 13.1p2). While not part of the definition of the signature,
1342 // this check is important to determine whether these functions
1343 // can be overloaded.
1344 const CXXMethodDecl* OldMethod = dyn_cast<CXXMethodDecl>(Old);
1345 const CXXMethodDecl* NewMethod = dyn_cast<CXXMethodDecl>(New);
1346 if (OldMethod && NewMethod &&
1347 !OldMethod->isStatic() && !NewMethod->isStatic() &&
1348 OldMethod->getTypeQualifiers() != NewMethod->getTypeQualifiers())
1349 return false;
1350
1351 // The signatures match; this is not an overload.
1352 return true;
1353 }
1354
1355 typedef llvm::DenseMap<const CXXMethodDecl *, const CXXMethodDecl*>
1356 ForwardUnique_t;
1357 ForwardUnique_t ForwardUnique;
1358 llvm::DenseMap<const CXXMethodDecl*, const CXXMethodDecl*> UniqueOverrider;
1359
1360 void BuildUniqueOverrider(const CXXMethodDecl *U, const CXXMethodDecl *MD) {
1361 const CXXMethodDecl *PrevU = UniqueOverrider[MD];
1362 assert(U && "no unique overrider");
1363 if (PrevU == U)
1364 return;
1365 if (PrevU != U && PrevU != 0) {
1366 // If already set, note the two sets as the same
1367 if (0)
1368 printf("%s::%s same as %s::%s\n",
1369 PrevU->getParent()->getNameAsCString(),
1370 PrevU->getNameAsCString(),
1371 U->getParent()->getNameAsCString(),
1372 U->getNameAsCString());
1373 ForwardUnique[PrevU] = U;
1374 return;
1375 }
1376
1377 // Not set, set it now
1378 if (0)
1379 printf("marking %s::%s %p override as %s::%s\n",
1380 MD->getParent()->getNameAsCString(),
1381 MD->getNameAsCString(),
1382 (void*)MD,
1383 U->getParent()->getNameAsCString(),
1384 U->getNameAsCString());
1385 UniqueOverrider[MD] = U;
1386
1387 for (CXXMethodDecl::method_iterator mi = MD->begin_overridden_methods(),
1388 me = MD->end_overridden_methods(); mi != me; ++mi) {
1389 BuildUniqueOverrider(U, *mi);
1390 }
1391 }
1392
1393 void BuildUniqueOverriders(const CXXRecordDecl *RD) {
1394 if (0) printf("walking %s\n", RD->getNameAsCString());
1395 for (CXXRecordDecl::method_iterator i = RD->method_begin(),
1396 e = RD->method_end(); i != e; ++i) {
1397 const CXXMethodDecl *MD = *i;
1398 if (!MD->isVirtual())
1399 continue;
1400
1401 if (UniqueOverrider[MD] == 0) {
1402 // Only set this, if it hasn't been set yet.
1403 BuildUniqueOverrider(MD, MD);
1404 if (0)
1405 printf("top set is %s::%s %p\n",
1406 MD->getParent()->getNameAsCString(),
1407 MD->getNameAsCString(),
1408 (void*)MD);
1409 ForwardUnique[MD] = MD;
1410 }
1411 }
1412 for (CXXRecordDecl::base_class_const_iterator i = RD->bases_begin(),
1413 e = RD->bases_end(); i != e; ++i) {
1414 const CXXRecordDecl *Base =
1415 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
1416 BuildUniqueOverriders(Base);
1417 }
1418 }
1419
1420 static int DclCmp(const void *p1, const void *p2) {
John McCallef3057c2010-02-13 01:04:05 +00001421 const CXXMethodDecl *MD1 = *(const CXXMethodDecl *const *)p1;
1422 const CXXMethodDecl *MD2 = *(const CXXMethodDecl *const *)p2;
1423
1424 return (DeclarationName::compare(MD1->getDeclName(), MD2->getDeclName()));
Mike Stump90181eb2010-01-26 00:05:04 +00001425 }
1426
1427 void MergeForwarding() {
1428 typedef llvm::SmallVector<const CXXMethodDecl *, 100> A_t;
1429 A_t A;
1430 for (ForwardUnique_t::iterator I = ForwardUnique.begin(),
1431 E = ForwardUnique.end(); I != E; ++I) {
1432 if (I->first == I->second)
1433 // Only add the roots of all trees
1434 A.push_back(I->first);
1435 }
1436 llvm::array_pod_sort(A.begin(), A.end(), DclCmp);
1437 for (A_t::iterator I = A.begin(),
1438 E = A.end(); I != E; ++I) {
1439 A_t::iterator J = I;
John McCallef3057c2010-02-13 01:04:05 +00001440 while (++J != E && DclCmp(I, J) == 0)
Mike Stump90181eb2010-01-26 00:05:04 +00001441 if (DclIsSame(*I, *J)) {
Eli Friedman18c28f62010-02-13 00:03:21 +00001442 if (0) printf("connecting %s\n", (*I)->getNameAsCString());
Mike Stump90181eb2010-01-26 00:05:04 +00001443 ForwardUnique[*J] = *I;
1444 }
1445 }
1446 }
1447
1448 const CXXMethodDecl *getUnique(const CXXMethodDecl *MD) {
1449 const CXXMethodDecl *U = UniqueOverrider[MD];
1450 assert(U && "unique overrider not found");
1451 while (ForwardUnique.count(U)) {
1452 const CXXMethodDecl *NU = ForwardUnique[U];
1453 if (NU == U) break;
1454 U = NU;
1455 }
1456 return U;
1457 }
Anders Carlsson72281172010-01-26 17:36:47 +00001458
1459 GlobalDecl getUnique(GlobalDecl GD) {
1460 const CXXMethodDecl *Unique = getUnique(cast<CXXMethodDecl>(GD.getDecl()));
1461
1462 if (const CXXConstructorDecl *CD = dyn_cast<CXXConstructorDecl>(Unique))
1463 return GlobalDecl(CD, GD.getCtorType());
1464
1465 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(Unique))
1466 return GlobalDecl(DD, GD.getDtorType());
1467
1468 return Unique;
Mike Stump90181eb2010-01-26 00:05:04 +00001469 }
1470
Anders Carlsson323bb042009-11-26 19:54:33 +00001471 /// getPureVirtualFn - Return the __cxa_pure_virtual function.
1472 llvm::Constant* getPureVirtualFn() {
1473 if (!PureVirtualFn) {
1474 const llvm::FunctionType *Ty =
1475 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
1476 /*isVarArg=*/false);
1477 PureVirtualFn = wrap(CGM.CreateRuntimeFunction(Ty, "__cxa_pure_virtual"));
1478 }
1479
1480 return PureVirtualFn;
1481 }
1482
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001483public:
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001484 OldVtableBuilder(const CXXRecordDecl *MostDerivedClass,
Eli Friedman6c08ce72009-12-05 01:05:03 +00001485 const CXXRecordDecl *l, uint64_t lo, CodeGenModule &cgm,
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001486 bool build, CGVtableInfo::AddressPointsMapTy& AddressPoints)
Eli Friedman6c08ce72009-12-05 01:05:03 +00001487 : BuildVtable(build), MostDerivedClass(MostDerivedClass), LayoutClass(l),
1488 LayoutOffset(lo), BLayout(cgm.getContext().getASTRecordLayout(l)),
1489 rtti(0), VMContext(cgm.getModule().getContext()),CGM(cgm),
1490 PureVirtualFn(0),
Anders Carlssonbad80eb2009-12-04 18:36:22 +00001491 subAddressPoints(AllocAddressPoint(cgm, l, MostDerivedClass)),
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001492 AddressPoints(AddressPoints),
1493 LLVMPointerWidth(cgm.getContext().Target.getPointerWidth(0))
1494 {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001495 Ptr8Ty = llvm::PointerType::get(llvm::Type::getInt8Ty(VMContext), 0);
Anders Carlsson3f4336c2009-12-17 07:09:17 +00001496 if (BuildVtable) {
1497 QualType ClassType = CGM.getContext().getTagDeclType(MostDerivedClass);
1498 rtti = CGM.GetAddrOfRTTIDescriptor(ClassType);
1499 }
Mike Stump90181eb2010-01-26 00:05:04 +00001500 BuildUniqueOverriders(MostDerivedClass);
1501 MergeForwarding();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001502 }
1503
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001504 // getVtableComponents - Returns a reference to the vtable components.
1505 const VtableComponentsVectorTy &getVtableComponents() const {
1506 return VtableComponents;
Anders Carlsson472404f2009-12-04 16:19:30 +00001507 }
1508
Anders Carlssone36a6b32010-01-02 01:01:18 +00001509 llvm::DenseMap<const CXXRecordDecl *, uint64_t> &getVBIndex()
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001510 { return VBIndex; }
1511
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001512 SavedAdjustmentsVectorTy &getSavedAdjustments()
1513 { return SavedAdjustments; }
Eli Friedman8174f2c2009-12-06 22:01:30 +00001514
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001515 llvm::Constant *wrap(Index_t i) {
1516 llvm::Constant *m;
1517 m = llvm::ConstantInt::get(llvm::Type::getInt64Ty(VMContext), i);
1518 return llvm::ConstantExpr::getIntToPtr(m, Ptr8Ty);
1519 }
1520
1521 llvm::Constant *wrap(llvm::Constant *m) {
1522 return llvm::ConstantExpr::getBitCast(m, Ptr8Ty);
1523 }
1524
Mike Stumpd2808442010-01-22 02:51:26 +00001525//#define D1(x)
1526#define D1(X) do { if (getenv("DEBUG")) { X; } } while (0)
Mike Stump75ce5732009-10-31 20:06:59 +00001527
1528 void GenerateVBaseOffsets(const CXXRecordDecl *RD, uint64_t Offset,
Mike Stump28431212009-10-13 22:54:56 +00001529 bool updateVBIndex, Index_t current_vbindex) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001530 for (CXXRecordDecl::base_class_const_iterator i = RD->bases_begin(),
1531 e = RD->bases_end(); i != e; ++i) {
1532 const CXXRecordDecl *Base =
1533 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
Mike Stump28431212009-10-13 22:54:56 +00001534 Index_t next_vbindex = current_vbindex;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001535 if (i->isVirtual() && !SeenVBase.count(Base)) {
1536 SeenVBase.insert(Base);
Mike Stump28431212009-10-13 22:54:56 +00001537 if (updateVBIndex) {
Mike Stump75ce5732009-10-31 20:06:59 +00001538 next_vbindex = (ssize_t)(-(VCalls.size()*LLVMPointerWidth/8)
Mike Stump28431212009-10-13 22:54:56 +00001539 - 3*LLVMPointerWidth/8);
1540 VBIndex[Base] = next_vbindex;
1541 }
Mike Stump75ce5732009-10-31 20:06:59 +00001542 int64_t BaseOffset = -(Offset/8) + BLayout.getVBaseClassOffset(Base)/8;
1543 VCalls.push_back((0?700:0) + BaseOffset);
1544 D1(printf(" vbase for %s at %d delta %d most derived %s\n",
1545 Base->getNameAsCString(),
1546 (int)-VCalls.size()-3, (int)BaseOffset,
Mike Stumpd2808442010-01-22 02:51:26 +00001547 MostDerivedClass->getNameAsCString()));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001548 }
Mike Stump28431212009-10-13 22:54:56 +00001549 // We also record offsets for non-virtual bases to closest enclosing
1550 // virtual base. We do this so that we don't have to search
1551 // for the nearst virtual base class when generating thunks.
1552 if (updateVBIndex && VBIndex.count(Base) == 0)
1553 VBIndex[Base] = next_vbindex;
Mike Stump75ce5732009-10-31 20:06:59 +00001554 GenerateVBaseOffsets(Base, Offset, updateVBIndex, next_vbindex);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001555 }
1556 }
1557
1558 void StartNewTable() {
1559 SeenVBase.clear();
1560 }
1561
Mike Stump8bccbfd2009-10-15 09:30:16 +00001562 Index_t getNVOffset_1(const CXXRecordDecl *D, const CXXRecordDecl *B,
1563 Index_t Offset = 0) {
1564
1565 if (B == D)
1566 return Offset;
1567
1568 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(D);
1569 for (CXXRecordDecl::base_class_const_iterator i = D->bases_begin(),
1570 e = D->bases_end(); i != e; ++i) {
1571 const CXXRecordDecl *Base =
1572 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
1573 int64_t BaseOffset = 0;
1574 if (!i->isVirtual())
1575 BaseOffset = Offset + Layout.getBaseClassOffset(Base);
1576 int64_t o = getNVOffset_1(Base, B, BaseOffset);
1577 if (o >= 0)
1578 return o;
1579 }
1580
1581 return -1;
1582 }
1583
1584 /// getNVOffset - Returns the non-virtual offset for the given (B) base of the
1585 /// derived class D.
1586 Index_t getNVOffset(QualType qB, QualType qD) {
Mike Stump46271322009-11-03 19:03:17 +00001587 qD = qD->getPointeeType();
1588 qB = qB->getPointeeType();
Mike Stump8bccbfd2009-10-15 09:30:16 +00001589 CXXRecordDecl *D = cast<CXXRecordDecl>(qD->getAs<RecordType>()->getDecl());
1590 CXXRecordDecl *B = cast<CXXRecordDecl>(qB->getAs<RecordType>()->getDecl());
1591 int64_t o = getNVOffset_1(D, B);
1592 if (o >= 0)
1593 return o;
1594
1595 assert(false && "FIXME: non-virtual base not found");
1596 return 0;
1597 }
1598
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001599 /// getVbaseOffset - Returns the index into the vtable for the virtual base
1600 /// offset for the given (B) virtual base of the derived class D.
1601 Index_t getVbaseOffset(QualType qB, QualType qD) {
Mike Stump46271322009-11-03 19:03:17 +00001602 qD = qD->getPointeeType();
1603 qB = qB->getPointeeType();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001604 CXXRecordDecl *D = cast<CXXRecordDecl>(qD->getAs<RecordType>()->getDecl());
1605 CXXRecordDecl *B = cast<CXXRecordDecl>(qB->getAs<RecordType>()->getDecl());
Anders Carlssonbad80eb2009-12-04 18:36:22 +00001606 if (D != MostDerivedClass)
Eli Friedman03aa2f12009-11-30 01:19:33 +00001607 return CGM.getVtableInfo().getVirtualBaseOffsetIndex(D, B);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001608 llvm::DenseMap<const CXXRecordDecl *, Index_t>::iterator i;
1609 i = VBIndex.find(B);
1610 if (i != VBIndex.end())
1611 return i->second;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001612
Mike Stump28431212009-10-13 22:54:56 +00001613 assert(false && "FIXME: Base not found");
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001614 return 0;
1615 }
1616
Eli Friedman81fb0d22009-12-04 08:40:51 +00001617 bool OverrideMethod(GlobalDecl GD, bool MorallyVirtual,
1618 Index_t OverrideOffset, Index_t Offset,
1619 int64_t CurrentVBaseOffset);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001620
Anders Carlsson495634e2009-12-04 02:39:04 +00001621 /// AppendMethods - Append the current methods to the vtable.
Anders Carlssonddf42c82009-12-04 02:56:03 +00001622 void AppendMethodsToVtable();
Anders Carlsson495634e2009-12-04 02:39:04 +00001623
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001624 llvm::Constant *WrapAddrOf(GlobalDecl GD) {
1625 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
1626
Anders Carlsson64457732009-11-24 05:08:52 +00001627 const llvm::Type *Ty = CGM.getTypes().GetFunctionTypeForVtable(MD);
Mike Stump18e8b472009-10-27 23:36:26 +00001628
Mike Stumpcdeb8002009-12-03 16:55:20 +00001629 return wrap(CGM.GetAddrOfFunction(GD, Ty));
Mike Stump18e8b472009-10-27 23:36:26 +00001630 }
1631
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001632 void OverrideMethods(Path_t *Path, bool MorallyVirtual, int64_t Offset,
1633 int64_t CurrentVBaseOffset) {
Mike Stump37dbe962009-10-15 02:04:03 +00001634 for (Path_t::reverse_iterator i = Path->rbegin(),
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001635 e = Path->rend(); i != e; ++i) {
1636 const CXXRecordDecl *RD = i->first;
Mike Stump8bccbfd2009-10-15 09:30:16 +00001637 int64_t OverrideOffset = i->second;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001638 for (method_iter mi = RD->method_begin(), me = RD->method_end(); mi != me;
1639 ++mi) {
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001640 const CXXMethodDecl *MD = *mi;
1641
1642 if (!MD->isVirtual())
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001643 continue;
1644
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001645 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
1646 // Override both the complete and the deleting destructor.
1647 GlobalDecl CompDtor(DD, Dtor_Complete);
Eli Friedman81fb0d22009-12-04 08:40:51 +00001648 OverrideMethod(CompDtor, MorallyVirtual, OverrideOffset, Offset,
1649 CurrentVBaseOffset);
1650
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001651 GlobalDecl DeletingDtor(DD, Dtor_Deleting);
Eli Friedman81fb0d22009-12-04 08:40:51 +00001652 OverrideMethod(DeletingDtor, MorallyVirtual, OverrideOffset, Offset,
1653 CurrentVBaseOffset);
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001654 } else {
Eli Friedman81fb0d22009-12-04 08:40:51 +00001655 OverrideMethod(MD, MorallyVirtual, OverrideOffset, Offset,
1656 CurrentVBaseOffset);
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001657 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001658 }
1659 }
1660 }
1661
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001662 void AddMethod(const GlobalDecl GD, bool MorallyVirtual, Index_t Offset,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001663 int64_t CurrentVBaseOffset) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001664 // If we can find a previously allocated slot for this, reuse it.
Eli Friedman81fb0d22009-12-04 08:40:51 +00001665 if (OverrideMethod(GD, MorallyVirtual, Offset, Offset,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001666 CurrentVBaseOffset))
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001667 return;
1668
Mike Stump1f49d652010-01-22 18:48:47 +00001669 D1(printf(" vfn for %s at %d\n",
1670 dyn_cast<CXXMethodDecl>(GD.getDecl())->getNameAsCString(),
1671 (int)Methods.size()));
1672
Anders Carlssoncdf18982009-12-04 02:08:24 +00001673 // We didn't find an entry in the vtable that we could use, add a new
1674 // entry.
1675 Methods.AddMethod(GD);
1676
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001677 VCallOffset[GD] = Offset/8 - CurrentVBaseOffset/8;
1678
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001679 if (MorallyVirtual) {
Anders Carlsson72281172010-01-26 17:36:47 +00001680 GlobalDecl UGD = getUnique(GD);
1681 const CXXMethodDecl *UMD = cast<CXXMethodDecl>(UGD.getDecl());
1682
Mike Stump90181eb2010-01-26 00:05:04 +00001683 assert(UMD && "final overrider not found");
1684
1685 Index_t &idx = VCall[UMD];
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001686 // Allocate the first one, after that, we reuse the previous one.
1687 if (idx == 0) {
Anders Carlsson72281172010-01-26 17:36:47 +00001688 VCallOffsetForVCall[UGD] = Offset/8;
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001689 NonVirtualOffset[UMD] = Offset/8 - CurrentVBaseOffset/8;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001690 idx = VCalls.size()+1;
Mike Stump90181eb2010-01-26 00:05:04 +00001691 VCalls.push_back(Offset/8 - CurrentVBaseOffset/8);
Mike Stump75ce5732009-10-31 20:06:59 +00001692 D1(printf(" vcall for %s at %d with delta %d\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001693 dyn_cast<CXXMethodDecl>(GD.getDecl())->getNameAsCString(),
Mike Stump90181eb2010-01-26 00:05:04 +00001694 (int)-VCalls.size()-3, (int)VCalls[idx-1]));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001695 }
1696 }
1697 }
1698
1699 void AddMethods(const CXXRecordDecl *RD, bool MorallyVirtual,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001700 Index_t Offset, int64_t CurrentVBaseOffset) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001701 for (method_iter mi = RD->method_begin(), me = RD->method_end(); mi != me;
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001702 ++mi) {
1703 const CXXMethodDecl *MD = *mi;
1704 if (!MD->isVirtual())
1705 continue;
1706
1707 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
1708 // For destructors, add both the complete and the deleting destructor
1709 // to the vtable.
1710 AddMethod(GlobalDecl(DD, Dtor_Complete), MorallyVirtual, Offset,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001711 CurrentVBaseOffset);
Eli Friedman03aa2f12009-11-30 01:19:33 +00001712 AddMethod(GlobalDecl(DD, Dtor_Deleting), MorallyVirtual, Offset,
1713 CurrentVBaseOffset);
1714 } else
1715 AddMethod(MD, MorallyVirtual, Offset, CurrentVBaseOffset);
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001716 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001717 }
1718
1719 void NonVirtualBases(const CXXRecordDecl *RD, const ASTRecordLayout &Layout,
1720 const CXXRecordDecl *PrimaryBase,
1721 bool PrimaryBaseWasVirtual, bool MorallyVirtual,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001722 int64_t Offset, int64_t CurrentVBaseOffset,
1723 Path_t *Path) {
Mike Stump37dbe962009-10-15 02:04:03 +00001724 Path->push_back(std::make_pair(RD, Offset));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001725 for (CXXRecordDecl::base_class_const_iterator i = RD->bases_begin(),
1726 e = RD->bases_end(); i != e; ++i) {
1727 if (i->isVirtual())
1728 continue;
1729 const CXXRecordDecl *Base =
1730 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
Mike Stump9eb76d42010-01-22 06:45:05 +00001731 uint64_t o = Offset + Layout.getBaseClassOffset(Base);
1732 StartNewTable();
1733 GenerateVtableForBase(Base, o, MorallyVirtual, false,
1734 true, Base == PrimaryBase && !PrimaryBaseWasVirtual,
1735 CurrentVBaseOffset, Path);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001736 }
Mike Stump37dbe962009-10-15 02:04:03 +00001737 Path->pop_back();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001738 }
1739
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001740// #define D(X) do { X; } while (0)
1741#define D(X)
1742
1743 void insertVCalls(int InsertionPoint) {
Mike Stump75ce5732009-10-31 20:06:59 +00001744 D1(printf("============= combining vbase/vcall\n"));
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001745 D(VCalls.insert(VCalls.begin(), 673));
1746 D(VCalls.push_back(672));
Eli Friedman6c08ce72009-12-05 01:05:03 +00001747
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001748 VtableComponents.insert(VtableComponents.begin() + InsertionPoint,
1749 VCalls.size(), 0);
Eli Friedman6c08ce72009-12-05 01:05:03 +00001750 if (BuildVtable) {
1751 // The vcalls come first...
1752 for (std::vector<Index_t>::reverse_iterator i = VCalls.rbegin(),
1753 e = VCalls.rend();
1754 i != e; ++i)
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001755 VtableComponents[InsertionPoint++] = wrap((0?600:0) + *i);
Eli Friedman6c08ce72009-12-05 01:05:03 +00001756 }
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001757 VCalls.clear();
Mike Stump9f23a142009-11-10 02:30:51 +00001758 VCall.clear();
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001759 VCallOffsetForVCall.clear();
1760 VCallOffset.clear();
1761 NonVirtualOffset.clear();
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001762 }
1763
Mike Stump559387f2009-11-13 23:13:20 +00001764 void AddAddressPoints(const CXXRecordDecl *RD, uint64_t Offset,
1765 Index_t AddressPoint) {
1766 D1(printf("XXX address point for %s in %s layout %s at offset %d is %d\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001767 RD->getNameAsCString(), MostDerivedClass->getNameAsCString(),
Mike Stump559387f2009-11-13 23:13:20 +00001768 LayoutClass->getNameAsCString(), (int)Offset, (int)AddressPoint));
Mike Stumpcd2b8212009-11-19 20:52:19 +00001769 subAddressPoints[std::make_pair(RD, Offset)] = AddressPoint;
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001770 AddressPoints[BaseSubobject(RD, Offset)] = AddressPoint;
Mike Stump559387f2009-11-13 23:13:20 +00001771
1772 // Now also add the address point for all our primary bases.
1773 while (1) {
1774 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1775 RD = Layout.getPrimaryBase();
1776 const bool PrimaryBaseWasVirtual = Layout.getPrimaryBaseWasVirtual();
1777 // FIXME: Double check this.
1778 if (RD == 0)
1779 break;
1780 if (PrimaryBaseWasVirtual &&
1781 BLayout.getVBaseClassOffset(RD) != Offset)
1782 break;
1783 D1(printf("XXX address point for %s in %s layout %s at offset %d is %d\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001784 RD->getNameAsCString(), MostDerivedClass->getNameAsCString(),
Mike Stump559387f2009-11-13 23:13:20 +00001785 LayoutClass->getNameAsCString(), (int)Offset, (int)AddressPoint));
Mike Stumpcd2b8212009-11-19 20:52:19 +00001786 subAddressPoints[std::make_pair(RD, Offset)] = AddressPoint;
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001787 AddressPoints[BaseSubobject(RD, Offset)] = AddressPoint;
Mike Stump559387f2009-11-13 23:13:20 +00001788 }
1789 }
1790
1791
Mike Stumpd538a6d2009-12-24 07:29:41 +00001792 void FinishGenerateVtable(const CXXRecordDecl *RD,
1793 const ASTRecordLayout &Layout,
1794 const CXXRecordDecl *PrimaryBase,
Mike Stump9eb76d42010-01-22 06:45:05 +00001795 bool ForNPNVBases, bool WasPrimaryBase,
Mike Stumpd538a6d2009-12-24 07:29:41 +00001796 bool PrimaryBaseWasVirtual,
1797 bool MorallyVirtual, int64_t Offset,
1798 bool ForVirtualBase, int64_t CurrentVBaseOffset,
1799 Path_t *Path) {
Mike Stump37dbe962009-10-15 02:04:03 +00001800 bool alloc = false;
1801 if (Path == 0) {
1802 alloc = true;
1803 Path = new Path_t;
1804 }
1805
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001806 StartNewTable();
1807 extra = 0;
Mike Stump9eb76d42010-01-22 06:45:05 +00001808 Index_t AddressPoint = 0;
1809 int VCallInsertionPoint = 0;
1810 if (!ForNPNVBases || !WasPrimaryBase) {
1811 bool DeferVCalls = MorallyVirtual || ForVirtualBase;
1812 VCallInsertionPoint = VtableComponents.size();
1813 if (!DeferVCalls) {
1814 insertVCalls(VCallInsertionPoint);
1815 } else
1816 // FIXME: just for extra, or for all uses of VCalls.size post this?
1817 extra = -VCalls.size();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001818
Mike Stump9eb76d42010-01-22 06:45:05 +00001819 // Add the offset to top.
1820 VtableComponents.push_back(BuildVtable ? wrap(-((Offset-LayoutOffset)/8)) : 0);
Anders Carlsson472404f2009-12-04 16:19:30 +00001821
Mike Stump9eb76d42010-01-22 06:45:05 +00001822 // Add the RTTI information.
1823 VtableComponents.push_back(rtti);
Anders Carlsson472404f2009-12-04 16:19:30 +00001824
Mike Stump9eb76d42010-01-22 06:45:05 +00001825 AddressPoint = VtableComponents.size();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001826
Mike Stump9eb76d42010-01-22 06:45:05 +00001827 AppendMethodsToVtable();
1828 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001829
1830 // and then the non-virtual bases.
1831 NonVirtualBases(RD, Layout, PrimaryBase, PrimaryBaseWasVirtual,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001832 MorallyVirtual, Offset, CurrentVBaseOffset, Path);
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001833
1834 if (ForVirtualBase) {
Mike Stump2cefe382009-11-12 20:47:57 +00001835 // FIXME: We're adding to VCalls in callers, we need to do the overrides
1836 // in the inner part, so that we know the complete set of vcalls during
1837 // the build and don't have to insert into methods. Saving out the
1838 // AddressPoint here, would need to be fixed, if we didn't do that. Also
1839 // retroactively adding vcalls for overrides later wind up in the wrong
1840 // place, the vcall slot has to be alloted during the walk of the base
1841 // when the function is first introduces.
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001842 AddressPoint += VCalls.size();
Mike Stump2cefe382009-11-12 20:47:57 +00001843 insertVCalls(VCallInsertionPoint);
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001844 }
1845
Mike Stump9eb76d42010-01-22 06:45:05 +00001846 if (!ForNPNVBases || !WasPrimaryBase)
1847 AddAddressPoints(RD, Offset, AddressPoint);
Mike Stump2cefe382009-11-12 20:47:57 +00001848
Mike Stump37dbe962009-10-15 02:04:03 +00001849 if (alloc) {
1850 delete Path;
1851 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001852 }
1853
Mike Stump75ce5732009-10-31 20:06:59 +00001854 void Primaries(const CXXRecordDecl *RD, bool MorallyVirtual, int64_t Offset,
1855 bool updateVBIndex, Index_t current_vbindex,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001856 int64_t CurrentVBaseOffset) {
Mike Stump75ce5732009-10-31 20:06:59 +00001857 if (!RD->isDynamicClass())
1858 return;
1859
1860 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1861 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
1862 const bool PrimaryBaseWasVirtual = Layout.getPrimaryBaseWasVirtual();
1863
1864 // vtables are composed from the chain of primaries.
Eli Friedman65d87222009-12-04 08:52:11 +00001865 if (PrimaryBase && !PrimaryBaseWasVirtual) {
Mike Stump75ce5732009-10-31 20:06:59 +00001866 D1(printf(" doing primaries for %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001867 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Eli Friedman65d87222009-12-04 08:52:11 +00001868 Primaries(PrimaryBase, PrimaryBaseWasVirtual|MorallyVirtual, Offset,
1869 updateVBIndex, current_vbindex, CurrentVBaseOffset);
Mike Stump75ce5732009-10-31 20:06:59 +00001870 }
1871
1872 D1(printf(" doing vcall entries for %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001873 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Mike Stump75ce5732009-10-31 20:06:59 +00001874
1875 // And add the virtuals for the class to the primary vtable.
Eli Friedman03aa2f12009-11-30 01:19:33 +00001876 AddMethods(RD, MorallyVirtual, Offset, CurrentVBaseOffset);
Mike Stump75ce5732009-10-31 20:06:59 +00001877 }
1878
1879 void VBPrimaries(const CXXRecordDecl *RD, bool MorallyVirtual, int64_t Offset,
1880 bool updateVBIndex, Index_t current_vbindex,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001881 bool RDisVirtualBase, int64_t CurrentVBaseOffset,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001882 bool bottom) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001883 if (!RD->isDynamicClass())
1884 return;
1885
1886 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1887 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
1888 const bool PrimaryBaseWasVirtual = Layout.getPrimaryBaseWasVirtual();
1889
1890 // vtables are composed from the chain of primaries.
1891 if (PrimaryBase) {
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001892 int BaseCurrentVBaseOffset = CurrentVBaseOffset;
1893 if (PrimaryBaseWasVirtual) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001894 IndirectPrimary.insert(PrimaryBase);
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001895 BaseCurrentVBaseOffset = BLayout.getVBaseClassOffset(PrimaryBase);
1896 }
Mike Stump75ce5732009-10-31 20:06:59 +00001897
1898 D1(printf(" doing primaries for %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001899 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Mike Stump75ce5732009-10-31 20:06:59 +00001900
1901 VBPrimaries(PrimaryBase, PrimaryBaseWasVirtual|MorallyVirtual, Offset,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001902 updateVBIndex, current_vbindex, PrimaryBaseWasVirtual,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001903 BaseCurrentVBaseOffset, false);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001904 }
1905
Mike Stump75ce5732009-10-31 20:06:59 +00001906 D1(printf(" doing vbase entries for %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001907 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Mike Stump75ce5732009-10-31 20:06:59 +00001908 GenerateVBaseOffsets(RD, Offset, updateVBIndex, current_vbindex);
1909
1910 if (RDisVirtualBase || bottom) {
1911 Primaries(RD, MorallyVirtual, Offset, updateVBIndex, current_vbindex,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001912 CurrentVBaseOffset);
Mike Stump75ce5732009-10-31 20:06:59 +00001913 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001914 }
1915
Mike Stumpd538a6d2009-12-24 07:29:41 +00001916 void GenerateVtableForBase(const CXXRecordDecl *RD, int64_t Offset = 0,
1917 bool MorallyVirtual = false,
1918 bool ForVirtualBase = false,
Mike Stump9eb76d42010-01-22 06:45:05 +00001919 bool ForNPNVBases = false,
1920 bool WasPrimaryBase = true,
Mike Stumpd538a6d2009-12-24 07:29:41 +00001921 int CurrentVBaseOffset = 0,
1922 Path_t *Path = 0) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001923 if (!RD->isDynamicClass())
Mike Stumpd538a6d2009-12-24 07:29:41 +00001924 return;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001925
Mike Stumpfa818082009-11-13 02:35:38 +00001926 // Construction vtable don't need parts that have no virtual bases and
1927 // aren't morally virtual.
Anders Carlssonbad80eb2009-12-04 18:36:22 +00001928 if ((LayoutClass != MostDerivedClass) &&
1929 RD->getNumVBases() == 0 && !MorallyVirtual)
Mike Stumpd538a6d2009-12-24 07:29:41 +00001930 return;
Mike Stumpfa818082009-11-13 02:35:38 +00001931
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001932 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1933 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
1934 const bool PrimaryBaseWasVirtual = Layout.getPrimaryBaseWasVirtual();
1935
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001936 extra = 0;
Mike Stump75ce5732009-10-31 20:06:59 +00001937 D1(printf("building entries for base %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001938 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001939
Mike Stump75ce5732009-10-31 20:06:59 +00001940 if (ForVirtualBase)
1941 extra = VCalls.size();
1942
Mike Stump9eb76d42010-01-22 06:45:05 +00001943 if (!ForNPNVBases || !WasPrimaryBase) {
1944 VBPrimaries(RD, MorallyVirtual, Offset, !ForVirtualBase, 0,
1945 ForVirtualBase, CurrentVBaseOffset, true);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001946
Mike Stump9eb76d42010-01-22 06:45:05 +00001947 if (Path)
1948 OverrideMethods(Path, MorallyVirtual, Offset, CurrentVBaseOffset);
1949 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001950
Mike Stump9eb76d42010-01-22 06:45:05 +00001951 FinishGenerateVtable(RD, Layout, PrimaryBase, ForNPNVBases, WasPrimaryBase,
1952 PrimaryBaseWasVirtual, MorallyVirtual, Offset,
1953 ForVirtualBase, CurrentVBaseOffset, Path);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001954 }
1955
1956 void GenerateVtableForVBases(const CXXRecordDecl *RD,
1957 int64_t Offset = 0,
Mike Stump37dbe962009-10-15 02:04:03 +00001958 Path_t *Path = 0) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001959 bool alloc = false;
1960 if (Path == 0) {
1961 alloc = true;
Mike Stump37dbe962009-10-15 02:04:03 +00001962 Path = new Path_t;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001963 }
1964 // FIXME: We also need to override using all paths to a virtual base,
1965 // right now, we just process the first path
1966 Path->push_back(std::make_pair(RD, Offset));
1967 for (CXXRecordDecl::base_class_const_iterator i = RD->bases_begin(),
1968 e = RD->bases_end(); i != e; ++i) {
1969 const CXXRecordDecl *Base =
1970 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
1971 if (i->isVirtual() && !IndirectPrimary.count(Base)) {
1972 // Mark it so we don't output it twice.
1973 IndirectPrimary.insert(Base);
1974 StartNewTable();
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001975 VCall.clear();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001976 int64_t BaseOffset = BLayout.getVBaseClassOffset(Base);
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001977 int64_t CurrentVBaseOffset = BaseOffset;
Mike Stump75ce5732009-10-31 20:06:59 +00001978 D1(printf("vtable %s virtual base %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001979 MostDerivedClass->getNameAsCString(), Base->getNameAsCString()));
Mike Stump9eb76d42010-01-22 06:45:05 +00001980 GenerateVtableForBase(Base, BaseOffset, true, true, false,
1981 true, CurrentVBaseOffset, Path);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001982 }
Mike Stump2cefe382009-11-12 20:47:57 +00001983 int64_t BaseOffset;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001984 if (i->isVirtual())
1985 BaseOffset = BLayout.getVBaseClassOffset(Base);
Mike Stump2cefe382009-11-12 20:47:57 +00001986 else {
1987 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1988 BaseOffset = Offset + Layout.getBaseClassOffset(Base);
1989 }
1990
Mike Stump37dbe962009-10-15 02:04:03 +00001991 if (Base->getNumVBases()) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001992 GenerateVtableForVBases(Base, BaseOffset, Path);
Mike Stump37dbe962009-10-15 02:04:03 +00001993 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001994 }
1995 Path->pop_back();
1996 if (alloc)
1997 delete Path;
1998 }
1999};
Anders Carlssonca1bf682009-12-03 01:54:02 +00002000} // end anonymous namespace
2001
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00002002bool OldVtableBuilder::OverrideMethod(GlobalDecl GD, bool MorallyVirtual,
Eli Friedman81fb0d22009-12-04 08:40:51 +00002003 Index_t OverrideOffset, Index_t Offset,
2004 int64_t CurrentVBaseOffset) {
Anders Carlssonca1bf682009-12-03 01:54:02 +00002005 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
2006
2007 const bool isPure = MD->isPure();
Anders Carlsson21bbc1e2009-12-05 17:04:47 +00002008
Anders Carlssonca1bf682009-12-03 01:54:02 +00002009 // FIXME: Should OverrideOffset's be Offset?
2010
Anders Carlsson21bbc1e2009-12-05 17:04:47 +00002011 for (CXXMethodDecl::method_iterator mi = MD->begin_overridden_methods(),
2012 e = MD->end_overridden_methods(); mi != e; ++mi) {
Anders Carlssonca1bf682009-12-03 01:54:02 +00002013 GlobalDecl OGD;
Mike Stumpa04ecfb2010-01-26 21:35:27 +00002014 GlobalDecl OGD2;
Anders Carlssonca1bf682009-12-03 01:54:02 +00002015
Anders Carlssonf3935b42009-12-04 05:51:56 +00002016 const CXXMethodDecl *OMD = *mi;
Anders Carlssonca1bf682009-12-03 01:54:02 +00002017 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(OMD))
2018 OGD = GlobalDecl(DD, GD.getDtorType());
2019 else
2020 OGD = OMD;
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002021
Eli Friedman8174f2c2009-12-06 22:01:30 +00002022 // Check whether this is the method being overridden in this section of
2023 // the vtable.
Anders Carlsson7bb70762009-12-04 15:49:02 +00002024 uint64_t Index;
2025 if (!Methods.getIndex(OGD, Index))
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002026 continue;
2027
Mike Stumpa04ecfb2010-01-26 21:35:27 +00002028 OGD2 = OGD;
2029
Eli Friedman8174f2c2009-12-06 22:01:30 +00002030 // Get the original method, which we should be computing thunks, etc,
2031 // against.
2032 OGD = Methods.getOrigMethod(Index);
2033 OMD = cast<CXXMethodDecl>(OGD.getDecl());
2034
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002035 QualType ReturnType =
2036 MD->getType()->getAs<FunctionType>()->getResultType();
2037 QualType OverriddenReturnType =
2038 OMD->getType()->getAs<FunctionType>()->getResultType();
Anders Carlssonca1bf682009-12-03 01:54:02 +00002039
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002040 // Check if we need a return type adjustment.
Anders Carlssoncce9ee82010-02-13 23:17:31 +00002041 if (!ComputeReturnAdjustmentBaseOffset(CGM.getContext(), MD,
2042 OMD).isEmpty()) {
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002043 CanQualType &BaseReturnType = BaseReturnTypes[Index];
Anders Carlssonca1bf682009-12-03 01:54:02 +00002044
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002045 // Insert the base return type.
2046 if (BaseReturnType.isNull())
2047 BaseReturnType =
2048 CGM.getContext().getCanonicalType(OverriddenReturnType);
2049 }
Anders Carlssona93e9802009-12-04 03:52:52 +00002050
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002051 Methods.OverrideMethod(OGD, GD);
2052
Anders Carlsson72281172010-01-26 17:36:47 +00002053 GlobalDecl UGD = getUnique(GD);
2054 const CXXMethodDecl *UMD = cast<CXXMethodDecl>(UGD.getDecl());
2055 assert(UGD.getDecl() && "unique overrider not found");
2056 assert(UGD == getUnique(OGD) && "unique overrider not unique");
Mike Stump90181eb2010-01-26 00:05:04 +00002057
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002058 ThisAdjustments.erase(Index);
Mike Stump90181eb2010-01-26 00:05:04 +00002059 if (MorallyVirtual || VCall.count(UMD)) {
Mike Stumpa04ecfb2010-01-26 21:35:27 +00002060
Mike Stump90181eb2010-01-26 00:05:04 +00002061 Index_t &idx = VCall[UMD];
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002062 if (idx == 0) {
Mike Stumpa04ecfb2010-01-26 21:35:27 +00002063 VCallOffset[GD] = VCallOffset[OGD];
2064 // NonVirtualOffset[UMD] = CurrentVBaseOffset/8 - OverrideOffset/8;
2065 NonVirtualOffset[UMD] = VCallOffset[OGD];
Mike Stump77537b12010-01-26 03:42:22 +00002066 VCallOffsetForVCall[UMD] = OverrideOffset/8;
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002067 idx = VCalls.size()+1;
Mike Stump77537b12010-01-26 03:42:22 +00002068 VCalls.push_back(OverrideOffset/8 - CurrentVBaseOffset/8);
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002069 D1(printf(" vcall for %s at %d with delta %d most derived %s\n",
2070 MD->getNameAsString().c_str(), (int)-idx-3,
Mike Stumpd2808442010-01-22 02:51:26 +00002071 (int)VCalls[idx-1], MostDerivedClass->getNameAsCString()));
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002072 } else {
Mike Stumpa04ecfb2010-01-26 21:35:27 +00002073 VCallOffset[GD] = NonVirtualOffset[UMD];
Anders Carlsson72281172010-01-26 17:36:47 +00002074 VCalls[idx-1] = -VCallOffsetForVCall[UGD] + OverrideOffset/8;
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002075 D1(printf(" vcall patch for %s at %d with delta %d most derived %s\n",
2076 MD->getNameAsString().c_str(), (int)-idx-3,
Mike Stumpd2808442010-01-22 02:51:26 +00002077 (int)VCalls[idx-1], MostDerivedClass->getNameAsCString()));
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002078 }
Mike Stumpa04ecfb2010-01-26 21:35:27 +00002079 int64_t NonVirtualAdjustment = -VCallOffset[OGD];
Mike Stumpded0a402010-01-26 22:44:01 +00002080 QualType DerivedType = MD->getThisType(CGM.getContext());
2081 QualType BaseType = cast<const CXXMethodDecl>(OGD.getDecl())->getThisType(CGM.getContext());
2082 int64_t NonVirtualAdjustment2 = -(getNVOffset(BaseType, DerivedType)/8);
2083 if (NonVirtualAdjustment2 != NonVirtualAdjustment) {
2084 NonVirtualAdjustment = NonVirtualAdjustment2;
2085 }
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002086 int64_t VirtualAdjustment =
2087 -((idx + extra + 2) * LLVMPointerWidth / 8);
Anders Carlsson657f1392009-12-03 02:32:59 +00002088
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002089 // Optimize out virtual adjustments of 0.
2090 if (VCalls[idx-1] == 0)
2091 VirtualAdjustment = 0;
Anders Carlsson657f1392009-12-03 02:32:59 +00002092
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002093 ThunkAdjustment ThisAdjustment(NonVirtualAdjustment,
2094 VirtualAdjustment);
Anders Carlsson2ca285f2009-12-03 02:22:59 +00002095
Eli Friedman8174f2c2009-12-06 22:01:30 +00002096 if (!isPure && !ThisAdjustment.isEmpty()) {
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002097 ThisAdjustments[Index] = ThisAdjustment;
Eli Friedman31bc3ad2009-12-07 23:56:34 +00002098 SavedAdjustments.push_back(
2099 std::make_pair(GD, std::make_pair(OGD, ThisAdjustment)));
Eli Friedman8174f2c2009-12-06 22:01:30 +00002100 }
Anders Carlssonca1bf682009-12-03 01:54:02 +00002101 return true;
2102 }
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002103
Mike Stumpa04ecfb2010-01-26 21:35:27 +00002104 VCallOffset[GD] = VCallOffset[OGD2] - OverrideOffset/8;
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002105
Mike Stumpa04ecfb2010-01-26 21:35:27 +00002106 int64_t NonVirtualAdjustment = -VCallOffset[GD];
2107 QualType DerivedType = MD->getThisType(CGM.getContext());
2108 QualType BaseType = cast<const CXXMethodDecl>(OGD.getDecl())->getThisType(CGM.getContext());
2109 int64_t NonVirtualAdjustment2 = -(getNVOffset(BaseType, DerivedType)/8);
2110 if (NonVirtualAdjustment2 != NonVirtualAdjustment) {
2111 NonVirtualAdjustment = NonVirtualAdjustment2;
2112 }
2113
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002114 if (NonVirtualAdjustment) {
2115 ThunkAdjustment ThisAdjustment(NonVirtualAdjustment, 0);
2116
Eli Friedmanf2eda5e2009-12-06 22:33:51 +00002117 if (!isPure) {
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002118 ThisAdjustments[Index] = ThisAdjustment;
Eli Friedman31bc3ad2009-12-07 23:56:34 +00002119 SavedAdjustments.push_back(
2120 std::make_pair(GD, std::make_pair(OGD, ThisAdjustment)));
Eli Friedmanf2eda5e2009-12-06 22:33:51 +00002121 }
Eli Friedman3d2e9de2009-12-04 08:34:14 +00002122 }
2123 return true;
Anders Carlssonca1bf682009-12-03 01:54:02 +00002124 }
2125
2126 return false;
Anders Carlssond59885022009-11-27 22:21:51 +00002127}
2128
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00002129void OldVtableBuilder::AppendMethodsToVtable() {
Eli Friedman6c08ce72009-12-05 01:05:03 +00002130 if (!BuildVtable) {
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00002131 VtableComponents.insert(VtableComponents.end(), Methods.size(),
2132 (llvm::Constant *)0);
Eli Friedman6c08ce72009-12-05 01:05:03 +00002133 ThisAdjustments.clear();
2134 BaseReturnTypes.clear();
2135 Methods.clear();
2136 return;
2137 }
2138
Anders Carlsson472404f2009-12-04 16:19:30 +00002139 // Reserve room in the vtable for our new methods.
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00002140 VtableComponents.reserve(VtableComponents.size() + Methods.size());
Anders Carlssonb07567c2009-12-04 03:07:26 +00002141
Anders Carlsson86809cd2009-12-04 02:43:50 +00002142 for (unsigned i = 0, e = Methods.size(); i != e; ++i) {
2143 GlobalDecl GD = Methods[i];
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00002144 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
2145
2146 // Get the 'this' pointer adjustment.
Anders Carlsson86809cd2009-12-04 02:43:50 +00002147 ThunkAdjustment ThisAdjustment = ThisAdjustments.lookup(i);
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00002148
2149 // Construct the return type adjustment.
2150 ThunkAdjustment ReturnAdjustment;
2151
2152 QualType BaseReturnType = BaseReturnTypes.lookup(i);
2153 if (!BaseReturnType.isNull() && !MD->isPure()) {
2154 QualType DerivedType =
2155 MD->getType()->getAs<FunctionType>()->getResultType();
2156
2157 int64_t NonVirtualAdjustment =
2158 getNVOffset(BaseReturnType, DerivedType) / 8;
2159
2160 int64_t VirtualAdjustment =
2161 getVbaseOffset(BaseReturnType, DerivedType);
2162
2163 ReturnAdjustment = ThunkAdjustment(NonVirtualAdjustment,
2164 VirtualAdjustment);
2165 }
2166
Anders Carlsson50f14742009-12-04 03:06:03 +00002167 llvm::Constant *Method = 0;
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00002168 if (!ReturnAdjustment.isEmpty()) {
2169 // Build a covariant thunk.
2170 CovariantThunkAdjustment Adjustment(ThisAdjustment, ReturnAdjustment);
Eli Friedman8174f2c2009-12-06 22:01:30 +00002171 Method = wrap(CGM.GetAddrOfCovariantThunk(GD, Adjustment));
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00002172 } else if (!ThisAdjustment.isEmpty()) {
2173 // Build a "regular" thunk.
Eli Friedman8174f2c2009-12-06 22:01:30 +00002174 Method = wrap(CGM.GetAddrOfThunk(GD, ThisAdjustment));
Anders Carlssonddf42c82009-12-04 02:56:03 +00002175 } else if (MD->isPure()) {
2176 // We have a pure virtual method.
Anders Carlsson50f14742009-12-04 03:06:03 +00002177 Method = getPureVirtualFn();
2178 } else {
2179 // We have a good old regular method.
2180 Method = WrapAddrOf(GD);
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00002181 }
Anders Carlsson50f14742009-12-04 03:06:03 +00002182
2183 // Add the method to the vtable.
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00002184 VtableComponents.push_back(Method);
Anders Carlsson86809cd2009-12-04 02:43:50 +00002185 }
2186
Anders Carlsson50f14742009-12-04 03:06:03 +00002187
Anders Carlsson86809cd2009-12-04 02:43:50 +00002188 ThisAdjustments.clear();
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00002189 BaseReturnTypes.clear();
Anders Carlsson86809cd2009-12-04 02:43:50 +00002190
Anders Carlsson495634e2009-12-04 02:39:04 +00002191 Methods.clear();
Anders Carlsson495634e2009-12-04 02:39:04 +00002192}
2193
Anders Carlssonf942ee02009-11-27 20:47:55 +00002194void CGVtableInfo::ComputeMethodVtableIndices(const CXXRecordDecl *RD) {
2195
2196 // Itanium C++ ABI 2.5.2:
Anders Carlsson259688c2010-02-02 03:37:46 +00002197 // The order of the virtual function pointers in a virtual table is the
2198 // order of declaration of the corresponding member functions in the class.
Anders Carlssonf942ee02009-11-27 20:47:55 +00002199 //
Anders Carlsson259688c2010-02-02 03:37:46 +00002200 // There is an entry for any virtual function declared in a class,
2201 // whether it is a new function or overrides a base class function,
2202 // unless it overrides a function from the primary base, and conversion
2203 // between their return types does not require an adjustment.
Anders Carlssonf942ee02009-11-27 20:47:55 +00002204
2205 int64_t CurrentIndex = 0;
2206
2207 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
2208 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
2209
2210 if (PrimaryBase) {
Anders Carlssonc920fa22009-11-30 19:43:26 +00002211 assert(PrimaryBase->isDefinition() &&
2212 "Should have the definition decl of the primary base!");
Anders Carlssonf942ee02009-11-27 20:47:55 +00002213
2214 // Since the record decl shares its vtable pointer with the primary base
2215 // we need to start counting at the end of the primary base's vtable.
2216 CurrentIndex = getNumVirtualFunctionPointers(PrimaryBase);
2217 }
Eli Friedman21517252009-12-15 03:31:17 +00002218
2219 // Collect all the primary bases, so we can check whether methods override
2220 // a method from the base.
Anders Carlsson65b49782010-02-12 05:25:12 +00002221 VtableBuilder::PrimaryBasesSetTy PrimaryBases;
Eli Friedman21517252009-12-15 03:31:17 +00002222 for (ASTRecordLayout::primary_base_info_iterator
2223 I = Layout.primary_base_begin(), E = Layout.primary_base_end();
2224 I != E; ++I)
2225 PrimaryBases.insert((*I).getBase());
2226
Anders Carlssonf942ee02009-11-27 20:47:55 +00002227 const CXXDestructorDecl *ImplicitVirtualDtor = 0;
2228
2229 for (CXXRecordDecl::method_iterator i = RD->method_begin(),
2230 e = RD->method_end(); i != e; ++i) {
2231 const CXXMethodDecl *MD = *i;
2232
2233 // We only want virtual methods.
2234 if (!MD->isVirtual())
2235 continue;
2236
Anders Carlssonf942ee02009-11-27 20:47:55 +00002237 // Check if this method overrides a method in the primary base.
Anders Carlsson65b49782010-02-12 05:25:12 +00002238 if (const CXXMethodDecl *OverriddenMD =
2239 OverridesMethodInPrimaryBase(MD, PrimaryBases)) {
2240 // Check if converting from the return type of the method to the
2241 // return type of the overridden method requires conversion.
Anders Carlssoncce9ee82010-02-13 23:17:31 +00002242 if (ComputeReturnAdjustmentBaseOffset(CGM.getContext(), MD,
2243 OverriddenMD).isEmpty()) {
Anders Carlsson65b49782010-02-12 05:25:12 +00002244 // This index is shared between the index in the vtable of the primary
2245 // base class.
2246 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
2247 const CXXDestructorDecl *OverriddenDD =
2248 cast<CXXDestructorDecl>(OverriddenMD);
Anders Carlssonf942ee02009-11-27 20:47:55 +00002249
Anders Carlsson65b49782010-02-12 05:25:12 +00002250 // Add both the complete and deleting entries.
2251 MethodVtableIndices[GlobalDecl(DD, Dtor_Complete)] =
2252 getMethodVtableIndex(GlobalDecl(OverriddenDD, Dtor_Complete));
2253 MethodVtableIndices[GlobalDecl(DD, Dtor_Deleting)] =
2254 getMethodVtableIndex(GlobalDecl(OverriddenDD, Dtor_Deleting));
2255 } else {
2256 MethodVtableIndices[MD] = getMethodVtableIndex(OverriddenMD);
2257 }
2258
2259 // We don't need to add an entry for this method.
Anders Carlsson07604812010-02-12 18:14:46 +00002260 continue;
Anders Carlssonf942ee02009-11-27 20:47:55 +00002261 }
2262 }
2263
Anders Carlssonf942ee02009-11-27 20:47:55 +00002264 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
2265 if (MD->isImplicit()) {
2266 assert(!ImplicitVirtualDtor &&
2267 "Did already see an implicit virtual dtor!");
2268 ImplicitVirtualDtor = DD;
2269 continue;
2270 }
2271
2272 // Add the complete dtor.
2273 MethodVtableIndices[GlobalDecl(DD, Dtor_Complete)] = CurrentIndex++;
2274
2275 // Add the deleting dtor.
2276 MethodVtableIndices[GlobalDecl(DD, Dtor_Deleting)] = CurrentIndex++;
2277 } else {
2278 // Add the entry.
2279 MethodVtableIndices[MD] = CurrentIndex++;
2280 }
2281 }
2282
2283 if (ImplicitVirtualDtor) {
2284 // Itanium C++ ABI 2.5.2:
2285 // If a class has an implicitly-defined virtual destructor,
2286 // its entries come after the declared virtual function pointers.
2287
2288 // Add the complete dtor.
2289 MethodVtableIndices[GlobalDecl(ImplicitVirtualDtor, Dtor_Complete)] =
2290 CurrentIndex++;
2291
2292 // Add the deleting dtor.
2293 MethodVtableIndices[GlobalDecl(ImplicitVirtualDtor, Dtor_Deleting)] =
2294 CurrentIndex++;
2295 }
2296
2297 NumVirtualFunctionPointers[RD] = CurrentIndex;
2298}
2299
2300uint64_t CGVtableInfo::getNumVirtualFunctionPointers(const CXXRecordDecl *RD) {
2301 llvm::DenseMap<const CXXRecordDecl *, uint64_t>::iterator I =
2302 NumVirtualFunctionPointers.find(RD);
2303 if (I != NumVirtualFunctionPointers.end())
2304 return I->second;
2305
2306 ComputeMethodVtableIndices(RD);
2307
2308 I = NumVirtualFunctionPointers.find(RD);
2309 assert(I != NumVirtualFunctionPointers.end() && "Did not find entry!");
2310 return I->second;
2311}
2312
2313uint64_t CGVtableInfo::getMethodVtableIndex(GlobalDecl GD) {
Anders Carlssonfb4dda42009-11-13 17:08:56 +00002314 MethodVtableIndicesTy::iterator I = MethodVtableIndices.find(GD);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00002315 if (I != MethodVtableIndices.end())
2316 return I->second;
2317
Anders Carlssonfb4dda42009-11-13 17:08:56 +00002318 const CXXRecordDecl *RD = cast<CXXMethodDecl>(GD.getDecl())->getParent();
Anders Carlssonf942ee02009-11-27 20:47:55 +00002319
2320 ComputeMethodVtableIndices(RD);
2321
Anders Carlssonfb4dda42009-11-13 17:08:56 +00002322 I = MethodVtableIndices.find(GD);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00002323 assert(I != MethodVtableIndices.end() && "Did not find index!");
2324 return I->second;
2325}
2326
Eli Friedman31bc3ad2009-12-07 23:56:34 +00002327CGVtableInfo::AdjustmentVectorTy*
2328CGVtableInfo::getAdjustments(GlobalDecl GD) {
2329 SavedAdjustmentsTy::iterator I = SavedAdjustments.find(GD);
2330 if (I != SavedAdjustments.end())
2331 return &I->second;
Eli Friedman8174f2c2009-12-06 22:01:30 +00002332
2333 const CXXRecordDecl *RD = cast<CXXRecordDecl>(GD.getDecl()->getDeclContext());
Eli Friedman31bc3ad2009-12-07 23:56:34 +00002334 if (!SavedAdjustmentRecords.insert(RD).second)
2335 return 0;
Eli Friedman8174f2c2009-12-06 22:01:30 +00002336
Anders Carlsson5f9a8812010-01-14 02:29:07 +00002337 AddressPointsMapTy AddressPoints;
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00002338 OldVtableBuilder b(RD, RD, 0, CGM, false, AddressPoints);
Eli Friedman8174f2c2009-12-06 22:01:30 +00002339 D1(printf("vtable %s\n", RD->getNameAsCString()));
2340 b.GenerateVtableForBase(RD);
2341 b.GenerateVtableForVBases(RD);
Eli Friedman8174f2c2009-12-06 22:01:30 +00002342
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00002343 for (OldVtableBuilder::SavedAdjustmentsVectorTy::iterator
Eli Friedman31bc3ad2009-12-07 23:56:34 +00002344 i = b.getSavedAdjustments().begin(),
2345 e = b.getSavedAdjustments().end(); i != e; i++)
2346 SavedAdjustments[i->first].push_back(i->second);
Eli Friedman8174f2c2009-12-06 22:01:30 +00002347
Eli Friedman31bc3ad2009-12-07 23:56:34 +00002348 I = SavedAdjustments.find(GD);
2349 if (I != SavedAdjustments.end())
2350 return &I->second;
2351
2352 return 0;
Eli Friedman8174f2c2009-12-06 22:01:30 +00002353}
2354
Anders Carlsson2bb27f52009-10-11 22:13:54 +00002355int64_t CGVtableInfo::getVirtualBaseOffsetIndex(const CXXRecordDecl *RD,
2356 const CXXRecordDecl *VBase) {
2357 ClassPairTy ClassPair(RD, VBase);
2358
2359 VirtualBaseClassIndiciesTy::iterator I =
2360 VirtualBaseClassIndicies.find(ClassPair);
2361 if (I != VirtualBaseClassIndicies.end())
2362 return I->second;
2363
Anders Carlsson2bb27f52009-10-11 22:13:54 +00002364 // FIXME: This seems expensive. Can we do a partial job to get
2365 // just this data.
Anders Carlsson5f9a8812010-01-14 02:29:07 +00002366 AddressPointsMapTy AddressPoints;
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00002367 OldVtableBuilder b(RD, RD, 0, CGM, false, AddressPoints);
Mike Stump75ce5732009-10-31 20:06:59 +00002368 D1(printf("vtable %s\n", RD->getNameAsCString()));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00002369 b.GenerateVtableForBase(RD);
2370 b.GenerateVtableForVBases(RD);
2371
2372 for (llvm::DenseMap<const CXXRecordDecl *, uint64_t>::iterator I =
2373 b.getVBIndex().begin(), E = b.getVBIndex().end(); I != E; ++I) {
2374 // Insert all types.
2375 ClassPairTy ClassPair(RD, I->first);
2376
2377 VirtualBaseClassIndicies.insert(std::make_pair(ClassPair, I->second));
2378 }
2379
2380 I = VirtualBaseClassIndicies.find(ClassPair);
Anders Carlsson6a6cbfc2010-02-13 20:11:51 +00002381 // FIXME: The assertion below assertion currently fails with the old vtable
2382 /// layout code if there is a non-virtual thunk adjustment in a vtable.
2383 // Once the new layout is in place, this return should be removed.
2384 if (I == VirtualBaseClassIndicies.end())
2385 return 0;
2386
Anders Carlsson2bb27f52009-10-11 22:13:54 +00002387 assert(I != VirtualBaseClassIndicies.end() && "Did not find index!");
2388
2389 return I->second;
2390}
2391
Anders Carlssonc8e39ec2009-12-05 21:03:56 +00002392uint64_t CGVtableInfo::getVtableAddressPoint(const CXXRecordDecl *RD) {
2393 uint64_t AddressPoint =
Anders Carlsson93a18842010-01-02 18:02:32 +00002394 (*(*(CGM.getVtableInfo().AddressPoints[RD]))[RD])[std::make_pair(RD, 0)];
Anders Carlssonc8e39ec2009-12-05 21:03:56 +00002395
2396 return AddressPoint;
2397}
2398
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00002399llvm::GlobalVariable *
Anders Carlsson0911ae82009-12-06 00:23:49 +00002400CGVtableInfo::GenerateVtable(llvm::GlobalVariable::LinkageTypes Linkage,
Anders Carlsson232324c2009-12-06 00:53:22 +00002401 bool GenerateDefinition,
Anders Carlsson0911ae82009-12-06 00:23:49 +00002402 const CXXRecordDecl *LayoutClass,
Anders Carlsson5f9a8812010-01-14 02:29:07 +00002403 const CXXRecordDecl *RD, uint64_t Offset,
2404 AddressPointsMapTy& AddressPoints) {
Anders Carlssoneee53d92010-02-13 02:02:03 +00002405 if (GenerateDefinition && CGM.getLangOptions().DumpVtableLayouts) {
Anders Carlssoncf5a8822010-02-13 21:07:32 +00002406 VtableBuilder Builder(*this, RD);
Anders Carlssoneee53d92010-02-13 02:02:03 +00002407
2408 Builder.dumpLayout(llvm::errs());
2409 }
2410
Anders Carlsson2bb27f52009-10-11 22:13:54 +00002411 llvm::SmallString<256> OutName;
Mike Stump2cefe382009-11-12 20:47:57 +00002412 if (LayoutClass != RD)
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00002413 CGM.getMangleContext().mangleCXXCtorVtable(LayoutClass, Offset / 8,
2414 RD, OutName);
Mike Stumpeac45592009-11-11 20:26:26 +00002415 else
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00002416 CGM.getMangleContext().mangleCXXVtable(RD, OutName);
Daniel Dunbare128dd12009-11-21 09:06:22 +00002417 llvm::StringRef Name = OutName.str();
Benjamin Kramerbb0a07b2009-10-11 22:57:54 +00002418
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00002419 llvm::GlobalVariable *GV = CGM.getModule().getGlobalVariable(Name);
Anders Carlsson93a18842010-01-02 18:02:32 +00002420 if (GV == 0 || CGM.getVtableInfo().AddressPoints[LayoutClass] == 0 ||
2421 GV->isDeclaration()) {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00002422 OldVtableBuilder b(RD, LayoutClass, Offset, CGM, GenerateDefinition,
2423 AddressPoints);
Eli Friedman6c08ce72009-12-05 01:05:03 +00002424
2425 D1(printf("vtable %s\n", RD->getNameAsCString()));
2426 // First comes the vtables for all the non-virtual bases...
Mike Stumpd538a6d2009-12-24 07:29:41 +00002427 b.GenerateVtableForBase(RD, Offset);
Eli Friedman6c08ce72009-12-05 01:05:03 +00002428
2429 // then the vtables for all the virtual bases.
2430 b.GenerateVtableForVBases(RD, Offset);
2431
Anders Carlsson58b271d2009-12-05 22:19:10 +00002432 llvm::Constant *Init = 0;
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00002433 const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGM.getLLVMContext());
Anders Carlsson58b271d2009-12-05 22:19:10 +00002434 llvm::ArrayType *ArrayType =
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00002435 llvm::ArrayType::get(Int8PtrTy, b.getVtableComponents().size());
Anders Carlssona95d4c52009-12-05 21:09:05 +00002436
Anders Carlsson232324c2009-12-06 00:53:22 +00002437 if (GenerateDefinition)
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00002438 Init = llvm::ConstantArray::get(ArrayType, &b.getVtableComponents()[0],
2439 b.getVtableComponents().size());
Anders Carlsson232324c2009-12-06 00:53:22 +00002440
Mike Stumpaa51ad62009-11-19 04:04:36 +00002441 llvm::GlobalVariable *OGV = GV;
Anders Carlsson232324c2009-12-06 00:53:22 +00002442
2443 GV = new llvm::GlobalVariable(CGM.getModule(), ArrayType,
2444 /*isConstant=*/true, Linkage, Init, Name);
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00002445 CGM.setGlobalVisibility(GV, RD);
2446
Mike Stumpaa51ad62009-11-19 04:04:36 +00002447 if (OGV) {
2448 GV->takeName(OGV);
Anders Carlsson58b271d2009-12-05 22:19:10 +00002449 llvm::Constant *NewPtr =
2450 llvm::ConstantExpr::getBitCast(GV, OGV->getType());
Mike Stumpaa51ad62009-11-19 04:04:36 +00002451 OGV->replaceAllUsesWith(NewPtr);
2452 OGV->eraseFromParent();
2453 }
Mike Stumpaa51ad62009-11-19 04:04:36 +00002454 }
Anders Carlssonb3f54b72009-12-05 21:28:12 +00002455
2456 return GV;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00002457}
Mike Stump9f23a142009-11-10 02:30:51 +00002458
Anders Carlsson232324c2009-12-06 00:53:22 +00002459void CGVtableInfo::GenerateClassData(llvm::GlobalVariable::LinkageTypes Linkage,
2460 const CXXRecordDecl *RD) {
Anders Carlssone1b3e622009-12-07 07:59:52 +00002461 llvm::GlobalVariable *&Vtable = Vtables[RD];
2462 if (Vtable) {
2463 assert(Vtable->getInitializer() && "Vtable doesn't have a definition!");
2464 return;
2465 }
2466
Anders Carlsson5f9a8812010-01-14 02:29:07 +00002467 AddressPointsMapTy AddressPoints;
2468 Vtable = GenerateVtable(Linkage, /*GenerateDefinition=*/true, RD, RD, 0,
2469 AddressPoints);
Anders Carlssone36a6b32010-01-02 01:01:18 +00002470 GenerateVTT(Linkage, /*GenerateDefinition=*/true, RD);
Mike Stump1a139f82009-11-19 01:08:19 +00002471}
2472
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00002473llvm::GlobalVariable *CGVtableInfo::getVtable(const CXXRecordDecl *RD) {
Anders Carlsson232324c2009-12-06 00:53:22 +00002474 llvm::GlobalVariable *Vtable = Vtables.lookup(RD);
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00002475
Anders Carlsson5f9a8812010-01-14 02:29:07 +00002476 if (!Vtable) {
2477 AddressPointsMapTy AddressPoints;
Anders Carlsson232324c2009-12-06 00:53:22 +00002478 Vtable = GenerateVtable(llvm::GlobalValue::ExternalLinkage,
Anders Carlsson5f9a8812010-01-14 02:29:07 +00002479 /*GenerateDefinition=*/false, RD, RD, 0,
2480 AddressPoints);
2481 }
Mike Stumpaa51ad62009-11-19 04:04:36 +00002482
Anders Carlsson4ed44eb2009-12-05 22:42:54 +00002483 return Vtable;
Mike Stumpd846d082009-11-10 07:44:33 +00002484}
Mike Stumpeac45592009-11-11 20:26:26 +00002485
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002486void CGVtableInfo::MaybeEmitVtable(GlobalDecl GD) {
2487 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
2488 const CXXRecordDecl *RD = MD->getParent();
2489
Anders Carlsson4ed44eb2009-12-05 22:42:54 +00002490 // If the class doesn't have a vtable we don't need to emit one.
2491 if (!RD->isDynamicClass())
2492 return;
2493
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002494 // Get the key function.
Anders Carlsson5ebf8b42009-12-07 04:35:11 +00002495 const CXXMethodDecl *KeyFunction = CGM.getContext().getKeyFunction(RD);
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002496
Anders Carlsson4ed44eb2009-12-05 22:42:54 +00002497 if (KeyFunction) {
2498 // We don't have the right key function.
2499 if (KeyFunction->getCanonicalDecl() != MD->getCanonicalDecl())
2500 return;
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002501 }
2502
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002503 // Emit the data.
Douglas Gregorccecc1b2010-01-06 20:27:16 +00002504 GenerateClassData(CGM.getVtableLinkage(RD), RD);
Eli Friedman8174f2c2009-12-06 22:01:30 +00002505
2506 for (CXXRecordDecl::method_iterator i = RD->method_begin(),
2507 e = RD->method_end(); i != e; ++i) {
Eli Friedman31bc3ad2009-12-07 23:56:34 +00002508 if ((*i)->isVirtual() && ((*i)->hasInlineBody() || (*i)->isImplicit())) {
Eli Friedman8174f2c2009-12-06 22:01:30 +00002509 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(*i)) {
2510 CGM.BuildThunksForVirtual(GlobalDecl(DD, Dtor_Complete));
2511 CGM.BuildThunksForVirtual(GlobalDecl(DD, Dtor_Deleting));
2512 } else {
2513 CGM.BuildThunksForVirtual(GlobalDecl(*i));
2514 }
2515 }
2516 }
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002517}
2518