blob: 09d6067f46d554739c357f66bcaea7e2be49de4b [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"
Anders Carlsson5d40c6f2010-02-11 08:02:13 +000019#include "llvm/Support/Format.h"
Zhongxing Xu1721ef72009-11-13 05:46:16 +000020#include <cstdio>
Anders Carlsson2bb27f52009-10-11 22:13:54 +000021
22using namespace clang;
23using namespace CodeGen;
24
Anders Carlsson727ffb12010-02-11 19:39:49 +000025/// TypeConversionRequiresAdjustment - Returns whether conversion from a
26/// derived type to a base type requires adjustment.
27static bool
28TypeConversionRequiresAdjustment(ASTContext &Ctx,
29 const CXXRecordDecl *DerivedDecl,
30 const CXXRecordDecl *BaseDecl) {
31 CXXBasePaths Paths(/*FindAmbiguities=*/false,
32 /*RecordPaths=*/true, /*DetectVirtual=*/true);
33 if (!const_cast<CXXRecordDecl *>(DerivedDecl)->
34 isDerivedFrom(const_cast<CXXRecordDecl *>(BaseDecl), Paths)) {
35 assert(false && "Class must be derived from the passed in base class!");
36 return false;
37 }
38
39 // If we found a virtual base we always want to require adjustment.
40 if (Paths.getDetectedVirtual())
41 return true;
42
43 const CXXBasePath &Path = Paths.front();
44
45 for (size_t Start = 0, End = Path.size(); Start != End; ++Start) {
46 const CXXBasePathElement &Element = Path[Start];
47
48 // Check the base class offset.
49 const ASTRecordLayout &Layout = Ctx.getASTRecordLayout(Element.Class);
50
51 const RecordType *BaseType = Element.Base->getType()->getAs<RecordType>();
52 const CXXRecordDecl *Base = cast<CXXRecordDecl>(BaseType->getDecl());
53
54 if (Layout.getBaseClassOffset(Base) != 0) {
55 // This requires an adjustment.
56 return true;
57 }
58 }
59
60 return false;
61}
62
63static bool
64TypeConversionRequiresAdjustment(ASTContext &Ctx,
65 QualType DerivedType, QualType BaseType) {
66 // Canonicalize the types.
67 QualType CanDerivedType = Ctx.getCanonicalType(DerivedType);
68 QualType CanBaseType = Ctx.getCanonicalType(BaseType);
69
70 assert(CanDerivedType->getTypeClass() == CanBaseType->getTypeClass() &&
71 "Types must have same type class!");
72
73 if (CanDerivedType == CanBaseType) {
74 // No adjustment needed.
75 return false;
76 }
77
Anders Carlssondd27b5d2010-02-11 19:45:15 +000078 if (const ReferenceType *RT = CanDerivedType->getAs<ReferenceType>()) {
Anders Carlsson727ffb12010-02-11 19:39:49 +000079 CanDerivedType = RT->getPointeeType();
Anders Carlssondd27b5d2010-02-11 19:45:15 +000080 CanBaseType = CanBaseType->getAs<ReferenceType>()->getPointeeType();
81 } else if (const PointerType *PT = CanDerivedType->getAs<PointerType>()) {
Anders Carlsson727ffb12010-02-11 19:39:49 +000082 CanDerivedType = PT->getPointeeType();
Anders Carlssondd27b5d2010-02-11 19:45:15 +000083 CanBaseType = CanBaseType->getAs<PointerType>()->getPointeeType();
Anders Carlsson727ffb12010-02-11 19:39:49 +000084 } else {
85 assert(false && "Unexpected return type!");
86 }
87
88 if (CanDerivedType == CanBaseType) {
89 // No adjustment needed.
90 return false;
91 }
92
93 const CXXRecordDecl *DerivedDecl =
Anders Carlssondd27b5d2010-02-11 19:45:15 +000094 cast<CXXRecordDecl>(CanDerivedType->getAs<RecordType>()->getDecl());
Anders Carlsson727ffb12010-02-11 19:39:49 +000095
96 const CXXRecordDecl *BaseDecl =
Anders Carlssondd27b5d2010-02-11 19:45:15 +000097 cast<CXXRecordDecl>(CanBaseType->getAs<RecordType>()->getDecl());
Anders Carlsson727ffb12010-02-11 19:39:49 +000098
99 return TypeConversionRequiresAdjustment(Ctx, DerivedDecl, BaseDecl);
100}
101
Anders Carlssond59885022009-11-27 22:21:51 +0000102namespace {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000103
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000104/// FinalOverriders - Contains the final overrider member functions for all
105/// member functions in the base subobjects of a class.
106class FinalOverriders {
107 /// MostDerivedClass - The most derived class for which the final overriders
108 /// are stored.
109 const CXXRecordDecl *MostDerivedClass;
110
111 ASTContext &Context;
112
113 /// MostDerivedClassLayout - the AST record layout of the most derived class.
114 const ASTRecordLayout &MostDerivedClassLayout;
115
116 typedef llvm::DenseMap<std::pair<BaseSubobject, const CXXMethodDecl *>,
117 const CXXMethodDecl *> OverridersMapTy;
118
119 /// OverridersMap - The final overriders for all virtual member functions of
120 /// all the base subobjects of the most derived class.
121 OverridersMapTy OverridersMap;
122
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000123 typedef llvm::SmallVector<uint64_t, 1> OffsetVectorTy;
124
125 /// SubobjectOffsetsMapTy - This map is used for keeping track of all the
126 /// base subobject offsets that a single class declaration might refer to.
127 ///
128 /// For example, in:
129 ///
130 /// struct A { virtual void f(); };
131 /// struct B1 : A { };
132 /// struct B2 : A { };
133 /// struct C : B1, B2 { virtual void f(); };
134 ///
135 /// when we determine that C::f() overrides A::f(), we need to update the
136 /// overriders map for both A-in-B1 and A-in-B2 and the subobject offsets map
137 /// will have the subobject offsets for both A copies.
138 typedef llvm::DenseMap<const CXXRecordDecl *, OffsetVectorTy>
139 SubobjectOffsetsMapTy;
140
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000141 /// ComputeFinalOverriders - Compute the final overriders for a given base
142 /// subobject (and all its direct and indirect bases).
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000143 void ComputeFinalOverriders(BaseSubobject Base,
144 SubobjectOffsetsMapTy &Offsets);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000145
146 /// AddOverriders - Add the final overriders for this base subobject to the
147 /// map of final overriders.
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000148 void AddOverriders(BaseSubobject Base, SubobjectOffsetsMapTy &Offsets);
149
150 /// PropagateOverrider - Propagate the NewMD overrider to all the functions
151 /// that OldMD overrides. For example, if we have:
152 ///
153 /// struct A { virtual void f(); };
154 /// struct B : A { virtual void f(); };
155 /// struct C : B { virtual void f(); };
156 ///
157 /// and we want to override B::f with C::f, we also need to override A::f with
158 /// C::f.
159 void PropagateOverrider(const CXXMethodDecl *OldMD,
160 const CXXMethodDecl *NewMD,
161 SubobjectOffsetsMapTy &Offsets);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000162
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000163 static void MergeSubobjectOffsets(const SubobjectOffsetsMapTy &NewOffsets,
164 SubobjectOffsetsMapTy &Offsets);
165
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000166public:
167 explicit FinalOverriders(const CXXRecordDecl *MostDerivedClass);
168
169 /// getOverrider - Get the final overrider for the given method declaration in
170 /// the given base subobject.
171 const CXXMethodDecl *getOverrider(BaseSubobject Base,
172 const CXXMethodDecl *MD) const {
173 assert(OverridersMap.count(std::make_pair(Base, MD)) &&
174 "Did not find overrider!");
175
176 return OverridersMap.lookup(std::make_pair(Base, MD));
177 }
178
179 /// dump - dump the final overriders.
180 void dump() const {
181 dump(llvm::errs(), BaseSubobject(MostDerivedClass, 0));
182 }
183
184 /// dump - dump the final overriders for a base subobject, and all its direct
185 /// and indirect base subobjects.
186 void dump(llvm::raw_ostream &Out, BaseSubobject Base) const;
187};
188
189FinalOverriders::FinalOverriders(const CXXRecordDecl *MostDerivedClass)
190 : MostDerivedClass(MostDerivedClass),
191 Context(MostDerivedClass->getASTContext()),
192 MostDerivedClassLayout(Context.getASTRecordLayout(MostDerivedClass)) {
193
194 // Compute the final overriders.
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000195 SubobjectOffsetsMapTy Offsets;
196 ComputeFinalOverriders(BaseSubobject(MostDerivedClass, 0), Offsets);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000197
198 // And dump them (for now).
199 dump();
200}
201
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000202void FinalOverriders::AddOverriders(BaseSubobject Base,
203 SubobjectOffsetsMapTy &Offsets) {
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000204 const CXXRecordDecl *RD = Base.getBase();
205
206 for (CXXRecordDecl::method_iterator I = RD->method_begin(),
207 E = RD->method_end(); I != E; ++I) {
208 const CXXMethodDecl *MD = *I;
209
210 if (!MD->isVirtual())
211 continue;
212
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000213 // First, propagate the overrider.
214 PropagateOverrider(MD, MD, Offsets);
215
216 // Add the overrider as the final overrider of itself.
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000217 const CXXMethodDecl *&Overrider = OverridersMap[std::make_pair(Base, MD)];
218 assert(!Overrider && "Overrider should not exist yet!");
219
220 Overrider = MD;
221 }
222}
223
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000224void FinalOverriders::PropagateOverrider(const CXXMethodDecl *OldMD,
225 const CXXMethodDecl *NewMD,
226 SubobjectOffsetsMapTy &Offsets) {
227 for (CXXMethodDecl::method_iterator I = OldMD->begin_overridden_methods(),
228 E = OldMD->end_overridden_methods(); I != E; ++I) {
229 const CXXMethodDecl *OverriddenMD = *I;
230 const CXXRecordDecl *OverriddenRD = OverriddenMD->getParent();
231
232 // We want to override OverriddenMD in all subobjects, for example:
233 //
234 /// struct A { virtual void f(); };
235 /// struct B1 : A { };
236 /// struct B2 : A { };
237 /// struct C : B1, B2 { virtual void f(); };
238 ///
239 /// When overriding A::f with C::f we need to do so in both A subobjects.
240 const OffsetVectorTy &OffsetVector = Offsets[OverriddenRD];
241
242 // Go through all the subobjects.
243 for (unsigned I = 0, E = OffsetVector.size(); I != E; ++I) {
244 uint64_t Offset = OffsetVector[I];
245
246 const CXXMethodDecl *&Overrider =
247 OverridersMap[std::make_pair(BaseSubobject(OverriddenRD, Offset),
248 OverriddenMD)];
249 assert(Overrider && "Did not find existing overrider!");
250
251 // Set the new overrider.
252 Overrider = NewMD;
253
254 // And propagate it further.
255 PropagateOverrider(OverriddenMD, NewMD, Offsets);
256 }
257 }
258}
259
260void
261FinalOverriders::MergeSubobjectOffsets(const SubobjectOffsetsMapTy &NewOffsets,
262 SubobjectOffsetsMapTy &Offsets) {
263 // Iterate over the new offsets.
264 for (SubobjectOffsetsMapTy::const_iterator I = NewOffsets.begin(),
265 E = NewOffsets.end(); I != E; ++I) {
266 const CXXRecordDecl *NewRD = I->first;
267 const OffsetVectorTy& NewOffsetsVector = I->second;
268
269 OffsetVectorTy &OffsetsVector = Offsets[NewRD];
270 if (OffsetsVector.empty()) {
271 // There were no previous offsets in this vector, just insert all entries
272 // from the new offsets vector.
273 OffsetsVector.append(NewOffsetsVector.begin(), NewOffsetsVector.end());
274 continue;
275 }
276
277 assert(false && "FIXME: Handle merging the subobject offsets!");
278 }
279}
280
281void FinalOverriders::ComputeFinalOverriders(BaseSubobject Base,
282 SubobjectOffsetsMapTy &Offsets) {
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000283 const CXXRecordDecl *RD = Base.getBase();
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000284 const ASTRecordLayout &Layout = Context.getASTRecordLayout(RD);
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000285
Anders Carlsson9fbf6d92010-02-12 01:40:03 +0000286 SubobjectOffsetsMapTy NewOffsets;
287
288 for (CXXRecordDecl::base_class_const_iterator I = RD->bases_begin(),
289 E = RD->bases_end(); I != E; ++I) {
290 const CXXRecordDecl *BaseDecl =
291 cast<CXXRecordDecl>(I->getType()->getAs<RecordType>()->getDecl());
292
293 assert(!I->isVirtual() && "FIXME: Handle virtual bases!");
294
295 uint64_t BaseOffset = Layout.getBaseClassOffset(BaseDecl) +
296 Base.getBaseOffset();
297
298 // Compute the final overriders for this base.
299 ComputeFinalOverriders(BaseSubobject(BaseDecl, BaseOffset), NewOffsets);
300 }
301
302 /// Now add the overriders for this particular subobject.
303 AddOverriders(Base, NewOffsets);
304
305 // And merge the newly discovered subobject offsets.
306 MergeSubobjectOffsets(NewOffsets, Offsets);
307
308 /// Finally, add the offset for our own subobject.
309 Offsets[RD].push_back(Base.getBaseOffset());
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000310}
311
312void FinalOverriders::dump(llvm::raw_ostream &Out, BaseSubobject Base) const {
313
314 const CXXRecordDecl *RD = Base.getBase();
315 const ASTRecordLayout &Layout = Context.getASTRecordLayout(RD);
316
317 for (CXXRecordDecl::base_class_const_iterator I = RD->bases_begin(),
318 E = RD->bases_end(); I != E; ++I) {
319 assert(!I->isVirtual() && "FIXME: Handle virtual bases!");
320
321 const CXXRecordDecl *BaseDecl =
322 cast<CXXRecordDecl>(I->getType()->getAs<RecordType>()->getDecl());
323
324 uint64_t BaseOffset = Layout.getBaseClassOffset(BaseDecl) +
325 Base.getBaseOffset();
326
327 dump(Out, BaseSubobject(BaseDecl, BaseOffset));
328 }
329
330 Out << "Final overriders for (" << RD->getQualifiedNameAsString() << ", ";
331 Out << Base.getBaseOffset() << ")\n";
332
333 // Now dump the overriders for this base subobject.
334 for (CXXRecordDecl::method_iterator I = RD->method_begin(),
335 E = RD->method_end(); I != E; ++I) {
336 const CXXMethodDecl *MD = *I;
337
338 if (!MD->isVirtual())
339 continue;
340
341 const CXXMethodDecl *Overrider = getOverrider(Base, MD);
342
343 Out << " " << MD->getQualifiedNameAsString() << " - ";
344 Out << Overrider->getQualifiedNameAsString() << "\n";
345 }
346}
347
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000348/// VtableComponent - Represents a single component in a vtable.
349class VtableComponent {
350public:
351 enum Kind {
352 CK_VCallOffset,
353 CK_VBaseOffset,
354 CK_OffsetToTop,
355 CK_RTTI,
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000356 CK_FunctionPointer,
357
358 /// CK_CompleteDtorPointer - A pointer to the complete destructor.
359 CK_CompleteDtorPointer,
360
361 /// CK_DeletingDtorPointer - A pointer to the deleting destructor.
362 CK_DeletingDtorPointer
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000363 };
364
365 /// dump - Dump the contents of this component to the given stream.
366 void dump(llvm::raw_ostream &Out);
367
368 static VtableComponent MakeOffsetToTop(int64_t Offset) {
369 return VtableComponent(CK_OffsetToTop, Offset);
370 }
371
372 static VtableComponent MakeRTTI(const CXXRecordDecl *RD) {
373 return VtableComponent(CK_RTTI, reinterpret_cast<uintptr_t>(RD));
374 }
375
376 static VtableComponent MakeFunction(const CXXMethodDecl *MD) {
377 assert(!isa<CXXDestructorDecl>(MD) &&
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000378 "Don't use MakeFunction with destructors!");
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000379
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000380 return VtableComponent(CK_FunctionPointer,
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000381 reinterpret_cast<uintptr_t>(MD));
382 }
383
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000384 static VtableComponent MakeCompleteDtor(const CXXDestructorDecl *DD) {
385 return VtableComponent(CK_CompleteDtorPointer,
386 reinterpret_cast<uintptr_t>(DD));
387 }
388
389 static VtableComponent MakeDeletingDtor(const CXXDestructorDecl *DD) {
390 return VtableComponent(CK_DeletingDtorPointer,
391 reinterpret_cast<uintptr_t>(DD));
392 }
393
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000394 /// getKind - Get the kind of this vtable component.
395 Kind getKind() const {
396 return (Kind)(Value & 0x7);
397 }
398
399 int64_t getOffsetToTop() const {
400 assert(getKind() == CK_OffsetToTop && "Invalid component kind!");
401
402 return getOffset();
403 }
404
405 const CXXRecordDecl *getRTTIDecl() const {
406 assert(getKind() == CK_RTTI && "Invalid component kind!");
407
408 return reinterpret_cast<CXXRecordDecl *>(getPointer());
409 }
410
411 const CXXMethodDecl *getFunctionDecl() const {
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000412 assert(getKind() == CK_FunctionPointer);
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000413
414 return reinterpret_cast<CXXMethodDecl *>(getPointer());
415 }
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000416
417 const CXXDestructorDecl *getDestructorDecl() const {
418 assert((getKind() == CK_CompleteDtorPointer ||
419 getKind() == CK_DeletingDtorPointer) && "Invalid component kind!");
420
421 return reinterpret_cast<CXXDestructorDecl *>(getPointer());
422 }
423
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000424private:
425 VtableComponent(Kind ComponentKind, int64_t Offset) {
426 assert((ComponentKind == CK_VCallOffset ||
427 ComponentKind == CK_VBaseOffset ||
428 ComponentKind == CK_OffsetToTop) && "Invalid component kind!");
429 assert(Offset <= ((1LL << 56) - 1) && "Offset is too big!");
430
431 Value = ((Offset << 3) | ComponentKind);
432 }
433
434 VtableComponent(Kind ComponentKind, uintptr_t Ptr) {
435 assert((ComponentKind == CK_RTTI ||
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000436 ComponentKind == CK_FunctionPointer ||
437 ComponentKind == CK_CompleteDtorPointer ||
438 ComponentKind == CK_DeletingDtorPointer) &&
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000439 "Invalid component kind!");
440
441 assert((Ptr & 7) == 0 && "Pointer not sufficiently aligned!");
442
443 Value = Ptr | ComponentKind;
444 }
445
446 int64_t getOffset() const {
447 assert((getKind() == CK_VCallOffset || getKind() == CK_VBaseOffset ||
448 getKind() == CK_OffsetToTop) && "Invalid component kind!");
449
450 return Value >> 3;
451 }
452
453 uintptr_t getPointer() const {
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000454 assert((getKind() == CK_RTTI ||
455 getKind() == CK_FunctionPointer ||
456 getKind() == CK_CompleteDtorPointer ||
457 getKind() == CK_DeletingDtorPointer) &&
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000458 "Invalid component kind!");
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000459
460 return static_cast<uintptr_t>(Value & ~7ULL);
461 }
462
463 /// The kind is stored in the lower 3 bits of the value. For offsets, we
464 /// make use of the facts that classes can't be larger than 2^55 bytes,
465 /// so we store the offset in the lower part of the 61 bytes that remain.
466 /// (The reason that we're not simply using a PointerIntPair here is that we
467 /// need the offsets to be 64-bit, even when on a 32-bit machine).
468 int64_t Value;
469};
470
471/// VtableBuilder - Class for building vtable layout information.
Benjamin Kramer337e3a52009-11-28 19:45:26 +0000472class VtableBuilder {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000473 /// MostDerivedClass - The most derived class for which we're building this
474 /// vtable.
475 const CXXRecordDecl *MostDerivedClass;
476
477 /// Context - The ASTContext which we will use for layout information.
478 const ASTContext &Context;
479
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000480 /// FinalOverriders - The final overriders of the most derived class.
481 FinalOverriders Overriders;
482
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000483 /// Components - The components of the vtable being built.
484 llvm::SmallVector<VtableComponent, 64> Components;
485
Anders Carlsson932c2f22010-02-11 17:18:51 +0000486 /// AddressPoints - Address points for the vtable being built.
487 CGVtableInfo::AddressPointsMapTy AddressPoints;
488
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000489 /// layoutSimpleVtable - A test function that will layout very simple vtables
490 /// without any bases. Just used for testing for now.
491 void layoutSimpleVtable(const CXXRecordDecl *RD);
492
493public:
494 VtableBuilder(const CXXRecordDecl *MostDerivedClass)
495 : MostDerivedClass(MostDerivedClass),
Anders Carlssone8ac42a2010-02-11 21:24:32 +0000496 Context(MostDerivedClass->getASTContext()), Overriders(MostDerivedClass) {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000497
498 layoutSimpleVtable(MostDerivedClass);
499 }
500
501 /// dumpLayout - Dump the vtable layout.
502 void dumpLayout(llvm::raw_ostream&);
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000503};
504
505void VtableBuilder::layoutSimpleVtable(const CXXRecordDecl *RD) {
506 assert(!RD->getNumBases() &&
507 "We don't support layout for vtables with bases right now!");
508
509 // First, add the offset to top.
510 Components.push_back(VtableComponent::MakeOffsetToTop(0));
511
512 // Next, add the RTTI.
513 Components.push_back(VtableComponent::MakeRTTI(RD));
514
Anders Carlsson932c2f22010-02-11 17:18:51 +0000515 // Record the address point.
516 AddressPoints.insert(std::make_pair(BaseSubobject(RD, 0), Components.size()));
517
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000518 // Now go through all virtual member functions and add them.
519 for (CXXRecordDecl::method_iterator I = RD->method_begin(),
520 E = RD->method_end(); I != E; ++I) {
521 const CXXMethodDecl *MD = *I;
522
523 if (!MD->isVirtual())
524 continue;
525
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000526 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
527 // Add both the complete destructor and the deleting destructor.
528 Components.push_back(VtableComponent::MakeCompleteDtor(DD));
529 Components.push_back(VtableComponent::MakeDeletingDtor(DD));
530 } else {
531 // Add the function.
532 Components.push_back(VtableComponent::MakeFunction(MD));
533 }
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000534 }
535}
536
537/// dumpLayout - Dump the vtable layout.
538void VtableBuilder::dumpLayout(llvm::raw_ostream& Out) {
539
540 Out << "Vtable for '" << MostDerivedClass->getQualifiedNameAsString();
541 Out << "' (" << Components.size() << " entries).\n";
542
Anders Carlsson932c2f22010-02-11 17:18:51 +0000543 // Iterate through the address points and insert them into a new map where
544 // they are keyed by the index and not the base object.
545 // Since an address point can be shared by multiple subobjects, we use an
546 // STL multimap.
547 std::multimap<uint64_t, BaseSubobject> AddressPointsByIndex;
548 for (CGVtableInfo::AddressPointsMapTy::const_iterator I =
549 AddressPoints.begin(), E = AddressPoints.end(); I != E; ++I) {
550 const BaseSubobject& Base = I->first;
551 uint64_t Index = I->second;
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000552
Anders Carlsson932c2f22010-02-11 17:18:51 +0000553 AddressPointsByIndex.insert(std::make_pair(Index, Base));
554 }
555
556 for (unsigned I = 0, E = Components.size(); I != E; ++I) {
557 if (AddressPointsByIndex.count(I)) {
558 assert(AddressPointsByIndex.count(I) == 1 &&
559 "FIXME: Handle dumping multiple base subobjects for a single "
560 "address point!");
561
562 const BaseSubobject &Base = AddressPointsByIndex.find(I)->second;
563 Out << " -- (" << Base.getBase()->getQualifiedNameAsString();
564
565 // FIXME: Instead of dividing by 8, we should be using CharUnits.
566 Out << ", " << Base.getBaseOffset() / 8 << ") vtable address --\n";
567 }
568
569 Out << llvm::format("%4d | ", I);
570
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000571 const VtableComponent &Component = Components[I];
572
573 // Dump the component.
574 switch (Component.getKind()) {
575 // FIXME: Remove this default case.
576 default:
577 assert(false && "Unhandled component kind!");
578 break;
579
580 case VtableComponent::CK_OffsetToTop:
581 Out << "offset_to_top (" << Component.getOffsetToTop() << ")";
582 break;
583
584 case VtableComponent::CK_RTTI:
585 Out << Component.getRTTIDecl()->getQualifiedNameAsString() << " RTTI";
586 break;
587
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000588 case VtableComponent::CK_FunctionPointer: {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000589 const CXXMethodDecl *MD = Component.getFunctionDecl();
590
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000591 std::string Str =
592 PredefinedExpr::ComputeName(PredefinedExpr::PrettyFunctionNoVirtual,
593 MD);
594 Out << Str;
Anders Carlsson09da3372010-02-12 02:38:13 +0000595 if (MD->isPure())
596 Out << " [pure]";
597
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000598 break;
599 }
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000600
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000601 case VtableComponent::CK_CompleteDtorPointer: {
602 const CXXDestructorDecl *DD = Component.getDestructorDecl();
603
604 Out << DD->getQualifiedNameAsString() << "() [complete]";
Anders Carlsson09da3372010-02-12 02:38:13 +0000605 if (DD->isPure())
606 Out << " [pure]";
607
Anders Carlsson5bd8d192010-02-11 18:20:28 +0000608 break;
609 }
610
611 case VtableComponent::CK_DeletingDtorPointer: {
612 const CXXDestructorDecl *DD = Component.getDestructorDecl();
613
614 Out << DD->getQualifiedNameAsString() << "() [deleting]";
Anders Carlsson09da3372010-02-12 02:38:13 +0000615 if (DD->isPure())
616 Out << " [pure]";
617
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000618 break;
619 }
620
621 }
Anders Carlsson932c2f22010-02-11 17:18:51 +0000622
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000623 Out << '\n';
624 }
625
626}
627
628}
629
630namespace {
631class OldVtableBuilder {
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000632public:
633 /// Index_t - Vtable index type.
634 typedef uint64_t Index_t;
Eli Friedman31bc3ad2009-12-07 23:56:34 +0000635 typedef std::vector<std::pair<GlobalDecl,
636 std::pair<GlobalDecl, ThunkAdjustment> > >
637 SavedAdjustmentsVectorTy;
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000638private:
Anders Carlsson472404f2009-12-04 16:19:30 +0000639
Anders Carlssonfe5f7d92009-12-06 01:09:21 +0000640 // VtableComponents - The components of the vtable being built.
641 typedef llvm::SmallVector<llvm::Constant *, 64> VtableComponentsVectorTy;
642 VtableComponentsVectorTy VtableComponents;
643
Eli Friedman6c08ce72009-12-05 01:05:03 +0000644 const bool BuildVtable;
645
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000646 llvm::Type *Ptr8Ty;
Anders Carlssonbad80eb2009-12-04 18:36:22 +0000647
648 /// MostDerivedClass - The most derived class that this vtable is being
649 /// built for.
650 const CXXRecordDecl *MostDerivedClass;
651
Mike Stump83066c82009-11-13 01:54:23 +0000652 /// LayoutClass - The most derived class used for virtual base layout
653 /// information.
654 const CXXRecordDecl *LayoutClass;
Mike Stump653d0b92009-11-13 02:13:54 +0000655 /// LayoutOffset - The offset for Class in LayoutClass.
656 uint64_t LayoutOffset;
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000657 /// BLayout - Layout for the most derived class that this vtable is being
658 /// built for.
659 const ASTRecordLayout &BLayout;
660 llvm::SmallSet<const CXXRecordDecl *, 32> IndirectPrimary;
661 llvm::SmallSet<const CXXRecordDecl *, 32> SeenVBase;
662 llvm::Constant *rtti;
663 llvm::LLVMContext &VMContext;
664 CodeGenModule &CGM; // Per-module state.
Anders Carlssonf2f31f42009-12-04 03:46:21 +0000665
Mike Stump90181eb2010-01-26 00:05:04 +0000666 llvm::DenseMap<const CXXMethodDecl *, Index_t> VCall;
Anders Carlssonfb4dda42009-11-13 17:08:56 +0000667 llvm::DenseMap<GlobalDecl, Index_t> VCallOffset;
Mike Stump77537b12010-01-26 03:42:22 +0000668 llvm::DenseMap<GlobalDecl, Index_t> VCallOffsetForVCall;
Mike Stumpcd6f9ed2009-11-06 23:27:42 +0000669 // This is the offset to the nearest virtual base
Mike Stump90181eb2010-01-26 00:05:04 +0000670 llvm::DenseMap<const CXXMethodDecl *, Index_t> NonVirtualOffset;
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000671 llvm::DenseMap<const CXXRecordDecl *, Index_t> VBIndex;
Mike Stumpbb9ff052009-10-27 23:46:47 +0000672
Anders Carlsson323bb042009-11-26 19:54:33 +0000673 /// PureVirtualFunction - Points to __cxa_pure_virtual.
674 llvm::Constant *PureVirtualFn;
675
Anders Carlssona84b6e82009-12-04 02:01:07 +0000676 /// VtableMethods - A data structure for keeping track of methods in a vtable.
677 /// Can add methods, override methods and iterate in vtable order.
678 class VtableMethods {
679 // MethodToIndexMap - Maps from a global decl to the index it has in the
680 // Methods vector.
681 llvm::DenseMap<GlobalDecl, uint64_t> MethodToIndexMap;
682
683 /// Methods - The methods, in vtable order.
684 typedef llvm::SmallVector<GlobalDecl, 16> MethodsVectorTy;
685 MethodsVectorTy Methods;
Eli Friedman8174f2c2009-12-06 22:01:30 +0000686 MethodsVectorTy OrigMethods;
Anders Carlssona84b6e82009-12-04 02:01:07 +0000687
688 public:
689 /// AddMethod - Add a method to the vtable methods.
690 void AddMethod(GlobalDecl GD) {
691 assert(!MethodToIndexMap.count(GD) &&
692 "Method has already been added!");
693
694 MethodToIndexMap[GD] = Methods.size();
695 Methods.push_back(GD);
Eli Friedman8174f2c2009-12-06 22:01:30 +0000696 OrigMethods.push_back(GD);
Anders Carlssona84b6e82009-12-04 02:01:07 +0000697 }
698
699 /// OverrideMethod - Replace a method with another.
700 void OverrideMethod(GlobalDecl OverriddenGD, GlobalDecl GD) {
701 llvm::DenseMap<GlobalDecl, uint64_t>::iterator i
702 = MethodToIndexMap.find(OverriddenGD);
703 assert(i != MethodToIndexMap.end() && "Did not find entry!");
704
705 // Get the index of the old decl.
706 uint64_t Index = i->second;
707
708 // Replace the old decl with the new decl.
709 Methods[Index] = GD;
710
Anders Carlssona84b6e82009-12-04 02:01:07 +0000711 // And add the new.
712 MethodToIndexMap[GD] = Index;
713 }
714
Anders Carlsson7bb70762009-12-04 15:49:02 +0000715 /// getIndex - Gives the index of a passed in GlobalDecl. Returns false if
716 /// the index couldn't be found.
Benjamin Kramer62ab6162009-12-04 22:45:27 +0000717 bool getIndex(GlobalDecl GD, uint64_t &Index) const {
Anders Carlsson7bb70762009-12-04 15:49:02 +0000718 llvm::DenseMap<GlobalDecl, uint64_t>::const_iterator i
719 = MethodToIndexMap.find(GD);
Eli Friedman3d2e9de2009-12-04 08:34:14 +0000720
Anders Carlsson7bb70762009-12-04 15:49:02 +0000721 if (i == MethodToIndexMap.end())
722 return false;
Anders Carlssone6096362009-12-04 03:41:37 +0000723
Anders Carlsson7bb70762009-12-04 15:49:02 +0000724 Index = i->second;
725 return true;
Anders Carlssone6096362009-12-04 03:41:37 +0000726 }
727
Eli Friedman8174f2c2009-12-06 22:01:30 +0000728 GlobalDecl getOrigMethod(uint64_t Index) const {
729 return OrigMethods[Index];
730 }
731
Anders Carlssona84b6e82009-12-04 02:01:07 +0000732 MethodsVectorTy::size_type size() const {
733 return Methods.size();
734 }
735
736 void clear() {
737 MethodToIndexMap.clear();
738 Methods.clear();
Eli Friedman8174f2c2009-12-06 22:01:30 +0000739 OrigMethods.clear();
Anders Carlssona84b6e82009-12-04 02:01:07 +0000740 }
741
Anders Carlssone6096362009-12-04 03:41:37 +0000742 GlobalDecl operator[](uint64_t Index) const {
Anders Carlssona84b6e82009-12-04 02:01:07 +0000743 return Methods[Index];
744 }
745 };
746
747 /// Methods - The vtable methods we're currently building.
748 VtableMethods Methods;
749
Anders Carlsson4c837d22009-12-04 02:26:15 +0000750 /// ThisAdjustments - For a given index in the vtable, contains the 'this'
751 /// pointer adjustment needed for a method.
752 typedef llvm::DenseMap<uint64_t, ThunkAdjustment> ThisAdjustmentsMapTy;
753 ThisAdjustmentsMapTy ThisAdjustments;
Anders Carlssond420a312009-11-26 19:32:45 +0000754
Eli Friedman31bc3ad2009-12-07 23:56:34 +0000755 SavedAdjustmentsVectorTy SavedAdjustments;
Eli Friedman8174f2c2009-12-06 22:01:30 +0000756
Anders Carlssonc521f952009-12-04 02:22:02 +0000757 /// BaseReturnTypes - Contains the base return types of methods who have been
758 /// overridden with methods whose return types require adjustment. Used for
759 /// generating covariant thunk information.
760 typedef llvm::DenseMap<uint64_t, CanQualType> BaseReturnTypesMapTy;
761 BaseReturnTypesMapTy BaseReturnTypes;
Anders Carlssond420a312009-11-26 19:32:45 +0000762
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000763 std::vector<Index_t> VCalls;
Mike Stump2cefe382009-11-12 20:47:57 +0000764
765 typedef std::pair<const CXXRecordDecl *, uint64_t> CtorVtable_t;
Mike Stumpcd2b8212009-11-19 20:52:19 +0000766 // subAddressPoints - Used to hold the AddressPoints (offsets) into the built
767 // vtable for use in computing the initializers for the VTT.
768 llvm::DenseMap<CtorVtable_t, int64_t> &subAddressPoints;
Mike Stump2cefe382009-11-12 20:47:57 +0000769
Anders Carlsson5f9a8812010-01-14 02:29:07 +0000770 /// AddressPoints - Address points for this vtable.
771 CGVtableInfo::AddressPointsMapTy& AddressPoints;
772
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000773 typedef CXXRecordDecl::method_iterator method_iter;
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000774 const uint32_t LLVMPointerWidth;
775 Index_t extra;
Mike Stump37dbe962009-10-15 02:04:03 +0000776 typedef std::vector<std::pair<const CXXRecordDecl *, int64_t> > Path_t;
Mike Stumpcd2b8212009-11-19 20:52:19 +0000777 static llvm::DenseMap<CtorVtable_t, int64_t>&
778 AllocAddressPoint(CodeGenModule &cgm, const CXXRecordDecl *l,
779 const CXXRecordDecl *c) {
Anders Carlsson93a18842010-01-02 18:02:32 +0000780 CGVtableInfo::AddrMap_t *&oref = cgm.getVtableInfo().AddressPoints[l];
Mike Stumpcd2b8212009-11-19 20:52:19 +0000781 if (oref == 0)
Anders Carlsson93a18842010-01-02 18:02:32 +0000782 oref = new CGVtableInfo::AddrMap_t;
Mike Stumpcd2b8212009-11-19 20:52:19 +0000783
784 llvm::DenseMap<CtorVtable_t, int64_t> *&ref = (*oref)[c];
785 if (ref == 0)
786 ref = new llvm::DenseMap<CtorVtable_t, int64_t>;
787 return *ref;
788 }
Anders Carlsson323bb042009-11-26 19:54:33 +0000789
Mike Stump90181eb2010-01-26 00:05:04 +0000790 bool DclIsSame(const FunctionDecl *New, const FunctionDecl *Old) {
791 FunctionTemplateDecl *OldTemplate = Old->getDescribedFunctionTemplate();
792 FunctionTemplateDecl *NewTemplate = New->getDescribedFunctionTemplate();
793
794 // C++ [temp.fct]p2:
795 // A function template can be overloaded with other function templates
796 // and with normal (non-template) functions.
797 if ((OldTemplate == 0) != (NewTemplate == 0))
798 return false;
799
800 // Is the function New an overload of the function Old?
801 QualType OldQType = CGM.getContext().getCanonicalType(Old->getType());
802 QualType NewQType = CGM.getContext().getCanonicalType(New->getType());
803
804 // Compare the signatures (C++ 1.3.10) of the two functions to
805 // determine whether they are overloads. If we find any mismatch
806 // in the signature, they are overloads.
807
808 // If either of these functions is a K&R-style function (no
809 // prototype), then we consider them to have matching signatures.
810 if (isa<FunctionNoProtoType>(OldQType.getTypePtr()) ||
811 isa<FunctionNoProtoType>(NewQType.getTypePtr()))
812 return true;
813
814 FunctionProtoType* OldType = cast<FunctionProtoType>(OldQType);
815 FunctionProtoType* NewType = cast<FunctionProtoType>(NewQType);
816
817 // The signature of a function includes the types of its
818 // parameters (C++ 1.3.10), which includes the presence or absence
819 // of the ellipsis; see C++ DR 357).
820 if (OldQType != NewQType &&
821 (OldType->getNumArgs() != NewType->getNumArgs() ||
822 OldType->isVariadic() != NewType->isVariadic() ||
823 !std::equal(OldType->arg_type_begin(), OldType->arg_type_end(),
824 NewType->arg_type_begin())))
825 return false;
826
827#if 0
828 // C++ [temp.over.link]p4:
829 // The signature of a function template consists of its function
830 // signature, its return type and its template parameter list. The names
831 // of the template parameters are significant only for establishing the
832 // relationship between the template parameters and the rest of the
833 // signature.
834 //
835 // We check the return type and template parameter lists for function
836 // templates first; the remaining checks follow.
837 if (NewTemplate &&
838 (!TemplateParameterListsAreEqual(NewTemplate->getTemplateParameters(),
839 OldTemplate->getTemplateParameters(),
840 TPL_TemplateMatch) ||
841 OldType->getResultType() != NewType->getResultType()))
842 return false;
843#endif
844
845 // If the function is a class member, its signature includes the
846 // cv-qualifiers (if any) on the function itself.
847 //
848 // As part of this, also check whether one of the member functions
849 // is static, in which case they are not overloads (C++
850 // 13.1p2). While not part of the definition of the signature,
851 // this check is important to determine whether these functions
852 // can be overloaded.
853 const CXXMethodDecl* OldMethod = dyn_cast<CXXMethodDecl>(Old);
854 const CXXMethodDecl* NewMethod = dyn_cast<CXXMethodDecl>(New);
855 if (OldMethod && NewMethod &&
856 !OldMethod->isStatic() && !NewMethod->isStatic() &&
857 OldMethod->getTypeQualifiers() != NewMethod->getTypeQualifiers())
858 return false;
859
860 // The signatures match; this is not an overload.
861 return true;
862 }
863
864 typedef llvm::DenseMap<const CXXMethodDecl *, const CXXMethodDecl*>
865 ForwardUnique_t;
866 ForwardUnique_t ForwardUnique;
867 llvm::DenseMap<const CXXMethodDecl*, const CXXMethodDecl*> UniqueOverrider;
868
869 void BuildUniqueOverrider(const CXXMethodDecl *U, const CXXMethodDecl *MD) {
870 const CXXMethodDecl *PrevU = UniqueOverrider[MD];
871 assert(U && "no unique overrider");
872 if (PrevU == U)
873 return;
874 if (PrevU != U && PrevU != 0) {
875 // If already set, note the two sets as the same
876 if (0)
877 printf("%s::%s same as %s::%s\n",
878 PrevU->getParent()->getNameAsCString(),
879 PrevU->getNameAsCString(),
880 U->getParent()->getNameAsCString(),
881 U->getNameAsCString());
882 ForwardUnique[PrevU] = U;
883 return;
884 }
885
886 // Not set, set it now
887 if (0)
888 printf("marking %s::%s %p override as %s::%s\n",
889 MD->getParent()->getNameAsCString(),
890 MD->getNameAsCString(),
891 (void*)MD,
892 U->getParent()->getNameAsCString(),
893 U->getNameAsCString());
894 UniqueOverrider[MD] = U;
895
896 for (CXXMethodDecl::method_iterator mi = MD->begin_overridden_methods(),
897 me = MD->end_overridden_methods(); mi != me; ++mi) {
898 BuildUniqueOverrider(U, *mi);
899 }
900 }
901
902 void BuildUniqueOverriders(const CXXRecordDecl *RD) {
903 if (0) printf("walking %s\n", RD->getNameAsCString());
904 for (CXXRecordDecl::method_iterator i = RD->method_begin(),
905 e = RD->method_end(); i != e; ++i) {
906 const CXXMethodDecl *MD = *i;
907 if (!MD->isVirtual())
908 continue;
909
910 if (UniqueOverrider[MD] == 0) {
911 // Only set this, if it hasn't been set yet.
912 BuildUniqueOverrider(MD, MD);
913 if (0)
914 printf("top set is %s::%s %p\n",
915 MD->getParent()->getNameAsCString(),
916 MD->getNameAsCString(),
917 (void*)MD);
918 ForwardUnique[MD] = MD;
919 }
920 }
921 for (CXXRecordDecl::base_class_const_iterator i = RD->bases_begin(),
922 e = RD->bases_end(); i != e; ++i) {
923 const CXXRecordDecl *Base =
924 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
925 BuildUniqueOverriders(Base);
926 }
927 }
928
929 static int DclCmp(const void *p1, const void *p2) {
930 const CXXMethodDecl *MD1 = (const CXXMethodDecl *)p1;
931 const CXXMethodDecl *MD2 = (const CXXMethodDecl *)p2;
932 return (MD1->getIdentifier() - MD2->getIdentifier());
933 }
934
935 void MergeForwarding() {
936 typedef llvm::SmallVector<const CXXMethodDecl *, 100> A_t;
937 A_t A;
938 for (ForwardUnique_t::iterator I = ForwardUnique.begin(),
939 E = ForwardUnique.end(); I != E; ++I) {
940 if (I->first == I->second)
941 // Only add the roots of all trees
942 A.push_back(I->first);
943 }
944 llvm::array_pod_sort(A.begin(), A.end(), DclCmp);
945 for (A_t::iterator I = A.begin(),
946 E = A.end(); I != E; ++I) {
947 A_t::iterator J = I;
948 while (++J != E && DclCmp(*I, *J) == 0)
949 if (DclIsSame(*I, *J)) {
950 printf("connecting %s\n", (*I)->getNameAsCString());
951 ForwardUnique[*J] = *I;
952 }
953 }
954 }
955
956 const CXXMethodDecl *getUnique(const CXXMethodDecl *MD) {
957 const CXXMethodDecl *U = UniqueOverrider[MD];
958 assert(U && "unique overrider not found");
959 while (ForwardUnique.count(U)) {
960 const CXXMethodDecl *NU = ForwardUnique[U];
961 if (NU == U) break;
962 U = NU;
963 }
964 return U;
965 }
Anders Carlsson72281172010-01-26 17:36:47 +0000966
967 GlobalDecl getUnique(GlobalDecl GD) {
968 const CXXMethodDecl *Unique = getUnique(cast<CXXMethodDecl>(GD.getDecl()));
969
970 if (const CXXConstructorDecl *CD = dyn_cast<CXXConstructorDecl>(Unique))
971 return GlobalDecl(CD, GD.getCtorType());
972
973 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(Unique))
974 return GlobalDecl(DD, GD.getDtorType());
975
976 return Unique;
Mike Stump90181eb2010-01-26 00:05:04 +0000977 }
978
Anders Carlsson323bb042009-11-26 19:54:33 +0000979 /// getPureVirtualFn - Return the __cxa_pure_virtual function.
980 llvm::Constant* getPureVirtualFn() {
981 if (!PureVirtualFn) {
982 const llvm::FunctionType *Ty =
983 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
984 /*isVarArg=*/false);
985 PureVirtualFn = wrap(CGM.CreateRuntimeFunction(Ty, "__cxa_pure_virtual"));
986 }
987
988 return PureVirtualFn;
989 }
990
Anders Carlsson2bb27f52009-10-11 22:13:54 +0000991public:
Anders Carlsson5d40c6f2010-02-11 08:02:13 +0000992 OldVtableBuilder(const CXXRecordDecl *MostDerivedClass,
Eli Friedman6c08ce72009-12-05 01:05:03 +0000993 const CXXRecordDecl *l, uint64_t lo, CodeGenModule &cgm,
Anders Carlsson5f9a8812010-01-14 02:29:07 +0000994 bool build, CGVtableInfo::AddressPointsMapTy& AddressPoints)
Eli Friedman6c08ce72009-12-05 01:05:03 +0000995 : BuildVtable(build), MostDerivedClass(MostDerivedClass), LayoutClass(l),
996 LayoutOffset(lo), BLayout(cgm.getContext().getASTRecordLayout(l)),
997 rtti(0), VMContext(cgm.getModule().getContext()),CGM(cgm),
998 PureVirtualFn(0),
Anders Carlssonbad80eb2009-12-04 18:36:22 +0000999 subAddressPoints(AllocAddressPoint(cgm, l, MostDerivedClass)),
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001000 AddressPoints(AddressPoints),
1001 LLVMPointerWidth(cgm.getContext().Target.getPointerWidth(0))
1002 {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001003 Ptr8Ty = llvm::PointerType::get(llvm::Type::getInt8Ty(VMContext), 0);
Anders Carlsson3f4336c2009-12-17 07:09:17 +00001004 if (BuildVtable) {
1005 QualType ClassType = CGM.getContext().getTagDeclType(MostDerivedClass);
1006 rtti = CGM.GetAddrOfRTTIDescriptor(ClassType);
1007 }
Mike Stump90181eb2010-01-26 00:05:04 +00001008 BuildUniqueOverriders(MostDerivedClass);
1009 MergeForwarding();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001010 }
1011
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001012 // getVtableComponents - Returns a reference to the vtable components.
1013 const VtableComponentsVectorTy &getVtableComponents() const {
1014 return VtableComponents;
Anders Carlsson472404f2009-12-04 16:19:30 +00001015 }
1016
Anders Carlssone36a6b32010-01-02 01:01:18 +00001017 llvm::DenseMap<const CXXRecordDecl *, uint64_t> &getVBIndex()
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001018 { return VBIndex; }
1019
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001020 SavedAdjustmentsVectorTy &getSavedAdjustments()
1021 { return SavedAdjustments; }
Eli Friedman8174f2c2009-12-06 22:01:30 +00001022
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001023 llvm::Constant *wrap(Index_t i) {
1024 llvm::Constant *m;
1025 m = llvm::ConstantInt::get(llvm::Type::getInt64Ty(VMContext), i);
1026 return llvm::ConstantExpr::getIntToPtr(m, Ptr8Ty);
1027 }
1028
1029 llvm::Constant *wrap(llvm::Constant *m) {
1030 return llvm::ConstantExpr::getBitCast(m, Ptr8Ty);
1031 }
1032
Mike Stumpd2808442010-01-22 02:51:26 +00001033//#define D1(x)
1034#define D1(X) do { if (getenv("DEBUG")) { X; } } while (0)
Mike Stump75ce5732009-10-31 20:06:59 +00001035
1036 void GenerateVBaseOffsets(const CXXRecordDecl *RD, uint64_t Offset,
Mike Stump28431212009-10-13 22:54:56 +00001037 bool updateVBIndex, Index_t current_vbindex) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001038 for (CXXRecordDecl::base_class_const_iterator i = RD->bases_begin(),
1039 e = RD->bases_end(); i != e; ++i) {
1040 const CXXRecordDecl *Base =
1041 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
Mike Stump28431212009-10-13 22:54:56 +00001042 Index_t next_vbindex = current_vbindex;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001043 if (i->isVirtual() && !SeenVBase.count(Base)) {
1044 SeenVBase.insert(Base);
Mike Stump28431212009-10-13 22:54:56 +00001045 if (updateVBIndex) {
Mike Stump75ce5732009-10-31 20:06:59 +00001046 next_vbindex = (ssize_t)(-(VCalls.size()*LLVMPointerWidth/8)
Mike Stump28431212009-10-13 22:54:56 +00001047 - 3*LLVMPointerWidth/8);
1048 VBIndex[Base] = next_vbindex;
1049 }
Mike Stump75ce5732009-10-31 20:06:59 +00001050 int64_t BaseOffset = -(Offset/8) + BLayout.getVBaseClassOffset(Base)/8;
1051 VCalls.push_back((0?700:0) + BaseOffset);
1052 D1(printf(" vbase for %s at %d delta %d most derived %s\n",
1053 Base->getNameAsCString(),
1054 (int)-VCalls.size()-3, (int)BaseOffset,
Mike Stumpd2808442010-01-22 02:51:26 +00001055 MostDerivedClass->getNameAsCString()));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001056 }
Mike Stump28431212009-10-13 22:54:56 +00001057 // We also record offsets for non-virtual bases to closest enclosing
1058 // virtual base. We do this so that we don't have to search
1059 // for the nearst virtual base class when generating thunks.
1060 if (updateVBIndex && VBIndex.count(Base) == 0)
1061 VBIndex[Base] = next_vbindex;
Mike Stump75ce5732009-10-31 20:06:59 +00001062 GenerateVBaseOffsets(Base, Offset, updateVBIndex, next_vbindex);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001063 }
1064 }
1065
1066 void StartNewTable() {
1067 SeenVBase.clear();
1068 }
1069
Mike Stump8bccbfd2009-10-15 09:30:16 +00001070 Index_t getNVOffset_1(const CXXRecordDecl *D, const CXXRecordDecl *B,
1071 Index_t Offset = 0) {
1072
1073 if (B == D)
1074 return Offset;
1075
1076 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(D);
1077 for (CXXRecordDecl::base_class_const_iterator i = D->bases_begin(),
1078 e = D->bases_end(); i != e; ++i) {
1079 const CXXRecordDecl *Base =
1080 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
1081 int64_t BaseOffset = 0;
1082 if (!i->isVirtual())
1083 BaseOffset = Offset + Layout.getBaseClassOffset(Base);
1084 int64_t o = getNVOffset_1(Base, B, BaseOffset);
1085 if (o >= 0)
1086 return o;
1087 }
1088
1089 return -1;
1090 }
1091
1092 /// getNVOffset - Returns the non-virtual offset for the given (B) base of the
1093 /// derived class D.
1094 Index_t getNVOffset(QualType qB, QualType qD) {
Mike Stump46271322009-11-03 19:03:17 +00001095 qD = qD->getPointeeType();
1096 qB = qB->getPointeeType();
Mike Stump8bccbfd2009-10-15 09:30:16 +00001097 CXXRecordDecl *D = cast<CXXRecordDecl>(qD->getAs<RecordType>()->getDecl());
1098 CXXRecordDecl *B = cast<CXXRecordDecl>(qB->getAs<RecordType>()->getDecl());
1099 int64_t o = getNVOffset_1(D, B);
1100 if (o >= 0)
1101 return o;
1102
1103 assert(false && "FIXME: non-virtual base not found");
1104 return 0;
1105 }
1106
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001107 /// getVbaseOffset - Returns the index into the vtable for the virtual base
1108 /// offset for the given (B) virtual base of the derived class D.
1109 Index_t getVbaseOffset(QualType qB, QualType qD) {
Mike Stump46271322009-11-03 19:03:17 +00001110 qD = qD->getPointeeType();
1111 qB = qB->getPointeeType();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001112 CXXRecordDecl *D = cast<CXXRecordDecl>(qD->getAs<RecordType>()->getDecl());
1113 CXXRecordDecl *B = cast<CXXRecordDecl>(qB->getAs<RecordType>()->getDecl());
Anders Carlssonbad80eb2009-12-04 18:36:22 +00001114 if (D != MostDerivedClass)
Eli Friedman03aa2f12009-11-30 01:19:33 +00001115 return CGM.getVtableInfo().getVirtualBaseOffsetIndex(D, B);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001116 llvm::DenseMap<const CXXRecordDecl *, Index_t>::iterator i;
1117 i = VBIndex.find(B);
1118 if (i != VBIndex.end())
1119 return i->second;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001120
Mike Stump28431212009-10-13 22:54:56 +00001121 assert(false && "FIXME: Base not found");
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001122 return 0;
1123 }
1124
Eli Friedman81fb0d22009-12-04 08:40:51 +00001125 bool OverrideMethod(GlobalDecl GD, bool MorallyVirtual,
1126 Index_t OverrideOffset, Index_t Offset,
1127 int64_t CurrentVBaseOffset);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001128
Anders Carlsson495634e2009-12-04 02:39:04 +00001129 /// AppendMethods - Append the current methods to the vtable.
Anders Carlssonddf42c82009-12-04 02:56:03 +00001130 void AppendMethodsToVtable();
Anders Carlsson495634e2009-12-04 02:39:04 +00001131
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001132 llvm::Constant *WrapAddrOf(GlobalDecl GD) {
1133 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
1134
Anders Carlsson64457732009-11-24 05:08:52 +00001135 const llvm::Type *Ty = CGM.getTypes().GetFunctionTypeForVtable(MD);
Mike Stump18e8b472009-10-27 23:36:26 +00001136
Mike Stumpcdeb8002009-12-03 16:55:20 +00001137 return wrap(CGM.GetAddrOfFunction(GD, Ty));
Mike Stump18e8b472009-10-27 23:36:26 +00001138 }
1139
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001140 void OverrideMethods(Path_t *Path, bool MorallyVirtual, int64_t Offset,
1141 int64_t CurrentVBaseOffset) {
Mike Stump37dbe962009-10-15 02:04:03 +00001142 for (Path_t::reverse_iterator i = Path->rbegin(),
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001143 e = Path->rend(); i != e; ++i) {
1144 const CXXRecordDecl *RD = i->first;
Mike Stump8bccbfd2009-10-15 09:30:16 +00001145 int64_t OverrideOffset = i->second;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001146 for (method_iter mi = RD->method_begin(), me = RD->method_end(); mi != me;
1147 ++mi) {
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001148 const CXXMethodDecl *MD = *mi;
1149
1150 if (!MD->isVirtual())
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001151 continue;
1152
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001153 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
1154 // Override both the complete and the deleting destructor.
1155 GlobalDecl CompDtor(DD, Dtor_Complete);
Eli Friedman81fb0d22009-12-04 08:40:51 +00001156 OverrideMethod(CompDtor, MorallyVirtual, OverrideOffset, Offset,
1157 CurrentVBaseOffset);
1158
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001159 GlobalDecl DeletingDtor(DD, Dtor_Deleting);
Eli Friedman81fb0d22009-12-04 08:40:51 +00001160 OverrideMethod(DeletingDtor, MorallyVirtual, OverrideOffset, Offset,
1161 CurrentVBaseOffset);
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001162 } else {
Eli Friedman81fb0d22009-12-04 08:40:51 +00001163 OverrideMethod(MD, MorallyVirtual, OverrideOffset, Offset,
1164 CurrentVBaseOffset);
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001165 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001166 }
1167 }
1168 }
1169
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001170 void AddMethod(const GlobalDecl GD, bool MorallyVirtual, Index_t Offset,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001171 int64_t CurrentVBaseOffset) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001172 // If we can find a previously allocated slot for this, reuse it.
Eli Friedman81fb0d22009-12-04 08:40:51 +00001173 if (OverrideMethod(GD, MorallyVirtual, Offset, Offset,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001174 CurrentVBaseOffset))
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001175 return;
1176
Mike Stump1f49d652010-01-22 18:48:47 +00001177 D1(printf(" vfn for %s at %d\n",
1178 dyn_cast<CXXMethodDecl>(GD.getDecl())->getNameAsCString(),
1179 (int)Methods.size()));
1180
Anders Carlssoncdf18982009-12-04 02:08:24 +00001181 // We didn't find an entry in the vtable that we could use, add a new
1182 // entry.
1183 Methods.AddMethod(GD);
1184
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001185 VCallOffset[GD] = Offset/8 - CurrentVBaseOffset/8;
1186
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001187 if (MorallyVirtual) {
Anders Carlsson72281172010-01-26 17:36:47 +00001188 GlobalDecl UGD = getUnique(GD);
1189 const CXXMethodDecl *UMD = cast<CXXMethodDecl>(UGD.getDecl());
1190
Mike Stump90181eb2010-01-26 00:05:04 +00001191 assert(UMD && "final overrider not found");
1192
1193 Index_t &idx = VCall[UMD];
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001194 // Allocate the first one, after that, we reuse the previous one.
1195 if (idx == 0) {
Anders Carlsson72281172010-01-26 17:36:47 +00001196 VCallOffsetForVCall[UGD] = Offset/8;
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001197 NonVirtualOffset[UMD] = Offset/8 - CurrentVBaseOffset/8;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001198 idx = VCalls.size()+1;
Mike Stump90181eb2010-01-26 00:05:04 +00001199 VCalls.push_back(Offset/8 - CurrentVBaseOffset/8);
Mike Stump75ce5732009-10-31 20:06:59 +00001200 D1(printf(" vcall for %s at %d with delta %d\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001201 dyn_cast<CXXMethodDecl>(GD.getDecl())->getNameAsCString(),
Mike Stump90181eb2010-01-26 00:05:04 +00001202 (int)-VCalls.size()-3, (int)VCalls[idx-1]));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001203 }
1204 }
1205 }
1206
1207 void AddMethods(const CXXRecordDecl *RD, bool MorallyVirtual,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001208 Index_t Offset, int64_t CurrentVBaseOffset) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001209 for (method_iter mi = RD->method_begin(), me = RD->method_end(); mi != me;
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001210 ++mi) {
1211 const CXXMethodDecl *MD = *mi;
1212 if (!MD->isVirtual())
1213 continue;
1214
1215 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
1216 // For destructors, add both the complete and the deleting destructor
1217 // to the vtable.
1218 AddMethod(GlobalDecl(DD, Dtor_Complete), MorallyVirtual, Offset,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001219 CurrentVBaseOffset);
Eli Friedman03aa2f12009-11-30 01:19:33 +00001220 AddMethod(GlobalDecl(DD, Dtor_Deleting), MorallyVirtual, Offset,
1221 CurrentVBaseOffset);
1222 } else
1223 AddMethod(MD, MorallyVirtual, Offset, CurrentVBaseOffset);
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001224 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001225 }
1226
1227 void NonVirtualBases(const CXXRecordDecl *RD, const ASTRecordLayout &Layout,
1228 const CXXRecordDecl *PrimaryBase,
1229 bool PrimaryBaseWasVirtual, bool MorallyVirtual,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001230 int64_t Offset, int64_t CurrentVBaseOffset,
1231 Path_t *Path) {
Mike Stump37dbe962009-10-15 02:04:03 +00001232 Path->push_back(std::make_pair(RD, Offset));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001233 for (CXXRecordDecl::base_class_const_iterator i = RD->bases_begin(),
1234 e = RD->bases_end(); i != e; ++i) {
1235 if (i->isVirtual())
1236 continue;
1237 const CXXRecordDecl *Base =
1238 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
Mike Stump9eb76d42010-01-22 06:45:05 +00001239 uint64_t o = Offset + Layout.getBaseClassOffset(Base);
1240 StartNewTable();
1241 GenerateVtableForBase(Base, o, MorallyVirtual, false,
1242 true, Base == PrimaryBase && !PrimaryBaseWasVirtual,
1243 CurrentVBaseOffset, Path);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001244 }
Mike Stump37dbe962009-10-15 02:04:03 +00001245 Path->pop_back();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001246 }
1247
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001248// #define D(X) do { X; } while (0)
1249#define D(X)
1250
1251 void insertVCalls(int InsertionPoint) {
Mike Stump75ce5732009-10-31 20:06:59 +00001252 D1(printf("============= combining vbase/vcall\n"));
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001253 D(VCalls.insert(VCalls.begin(), 673));
1254 D(VCalls.push_back(672));
Eli Friedman6c08ce72009-12-05 01:05:03 +00001255
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001256 VtableComponents.insert(VtableComponents.begin() + InsertionPoint,
1257 VCalls.size(), 0);
Eli Friedman6c08ce72009-12-05 01:05:03 +00001258 if (BuildVtable) {
1259 // The vcalls come first...
1260 for (std::vector<Index_t>::reverse_iterator i = VCalls.rbegin(),
1261 e = VCalls.rend();
1262 i != e; ++i)
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001263 VtableComponents[InsertionPoint++] = wrap((0?600:0) + *i);
Eli Friedman6c08ce72009-12-05 01:05:03 +00001264 }
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001265 VCalls.clear();
Mike Stump9f23a142009-11-10 02:30:51 +00001266 VCall.clear();
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001267 VCallOffsetForVCall.clear();
1268 VCallOffset.clear();
1269 NonVirtualOffset.clear();
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001270 }
1271
Mike Stump559387f2009-11-13 23:13:20 +00001272 void AddAddressPoints(const CXXRecordDecl *RD, uint64_t Offset,
1273 Index_t AddressPoint) {
1274 D1(printf("XXX address point for %s in %s layout %s at offset %d is %d\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001275 RD->getNameAsCString(), MostDerivedClass->getNameAsCString(),
Mike Stump559387f2009-11-13 23:13:20 +00001276 LayoutClass->getNameAsCString(), (int)Offset, (int)AddressPoint));
Mike Stumpcd2b8212009-11-19 20:52:19 +00001277 subAddressPoints[std::make_pair(RD, Offset)] = AddressPoint;
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001278 AddressPoints[BaseSubobject(RD, Offset)] = AddressPoint;
Mike Stump559387f2009-11-13 23:13:20 +00001279
1280 // Now also add the address point for all our primary bases.
1281 while (1) {
1282 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1283 RD = Layout.getPrimaryBase();
1284 const bool PrimaryBaseWasVirtual = Layout.getPrimaryBaseWasVirtual();
1285 // FIXME: Double check this.
1286 if (RD == 0)
1287 break;
1288 if (PrimaryBaseWasVirtual &&
1289 BLayout.getVBaseClassOffset(RD) != Offset)
1290 break;
1291 D1(printf("XXX address point for %s in %s layout %s at offset %d is %d\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001292 RD->getNameAsCString(), MostDerivedClass->getNameAsCString(),
Mike Stump559387f2009-11-13 23:13:20 +00001293 LayoutClass->getNameAsCString(), (int)Offset, (int)AddressPoint));
Mike Stumpcd2b8212009-11-19 20:52:19 +00001294 subAddressPoints[std::make_pair(RD, Offset)] = AddressPoint;
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001295 AddressPoints[BaseSubobject(RD, Offset)] = AddressPoint;
Mike Stump559387f2009-11-13 23:13:20 +00001296 }
1297 }
1298
1299
Mike Stumpd538a6d2009-12-24 07:29:41 +00001300 void FinishGenerateVtable(const CXXRecordDecl *RD,
1301 const ASTRecordLayout &Layout,
1302 const CXXRecordDecl *PrimaryBase,
Mike Stump9eb76d42010-01-22 06:45:05 +00001303 bool ForNPNVBases, bool WasPrimaryBase,
Mike Stumpd538a6d2009-12-24 07:29:41 +00001304 bool PrimaryBaseWasVirtual,
1305 bool MorallyVirtual, int64_t Offset,
1306 bool ForVirtualBase, int64_t CurrentVBaseOffset,
1307 Path_t *Path) {
Mike Stump37dbe962009-10-15 02:04:03 +00001308 bool alloc = false;
1309 if (Path == 0) {
1310 alloc = true;
1311 Path = new Path_t;
1312 }
1313
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001314 StartNewTable();
1315 extra = 0;
Mike Stump9eb76d42010-01-22 06:45:05 +00001316 Index_t AddressPoint = 0;
1317 int VCallInsertionPoint = 0;
1318 if (!ForNPNVBases || !WasPrimaryBase) {
1319 bool DeferVCalls = MorallyVirtual || ForVirtualBase;
1320 VCallInsertionPoint = VtableComponents.size();
1321 if (!DeferVCalls) {
1322 insertVCalls(VCallInsertionPoint);
1323 } else
1324 // FIXME: just for extra, or for all uses of VCalls.size post this?
1325 extra = -VCalls.size();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001326
Mike Stump9eb76d42010-01-22 06:45:05 +00001327 // Add the offset to top.
1328 VtableComponents.push_back(BuildVtable ? wrap(-((Offset-LayoutOffset)/8)) : 0);
Anders Carlsson472404f2009-12-04 16:19:30 +00001329
Mike Stump9eb76d42010-01-22 06:45:05 +00001330 // Add the RTTI information.
1331 VtableComponents.push_back(rtti);
Anders Carlsson472404f2009-12-04 16:19:30 +00001332
Mike Stump9eb76d42010-01-22 06:45:05 +00001333 AddressPoint = VtableComponents.size();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001334
Mike Stump9eb76d42010-01-22 06:45:05 +00001335 AppendMethodsToVtable();
1336 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001337
1338 // and then the non-virtual bases.
1339 NonVirtualBases(RD, Layout, PrimaryBase, PrimaryBaseWasVirtual,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001340 MorallyVirtual, Offset, CurrentVBaseOffset, Path);
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001341
1342 if (ForVirtualBase) {
Mike Stump2cefe382009-11-12 20:47:57 +00001343 // FIXME: We're adding to VCalls in callers, we need to do the overrides
1344 // in the inner part, so that we know the complete set of vcalls during
1345 // the build and don't have to insert into methods. Saving out the
1346 // AddressPoint here, would need to be fixed, if we didn't do that. Also
1347 // retroactively adding vcalls for overrides later wind up in the wrong
1348 // place, the vcall slot has to be alloted during the walk of the base
1349 // when the function is first introduces.
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001350 AddressPoint += VCalls.size();
Mike Stump2cefe382009-11-12 20:47:57 +00001351 insertVCalls(VCallInsertionPoint);
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001352 }
1353
Mike Stump9eb76d42010-01-22 06:45:05 +00001354 if (!ForNPNVBases || !WasPrimaryBase)
1355 AddAddressPoints(RD, Offset, AddressPoint);
Mike Stump2cefe382009-11-12 20:47:57 +00001356
Mike Stump37dbe962009-10-15 02:04:03 +00001357 if (alloc) {
1358 delete Path;
1359 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001360 }
1361
Mike Stump75ce5732009-10-31 20:06:59 +00001362 void Primaries(const CXXRecordDecl *RD, bool MorallyVirtual, int64_t Offset,
1363 bool updateVBIndex, Index_t current_vbindex,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001364 int64_t CurrentVBaseOffset) {
Mike Stump75ce5732009-10-31 20:06:59 +00001365 if (!RD->isDynamicClass())
1366 return;
1367
1368 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1369 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
1370 const bool PrimaryBaseWasVirtual = Layout.getPrimaryBaseWasVirtual();
1371
1372 // vtables are composed from the chain of primaries.
Eli Friedman65d87222009-12-04 08:52:11 +00001373 if (PrimaryBase && !PrimaryBaseWasVirtual) {
Mike Stump75ce5732009-10-31 20:06:59 +00001374 D1(printf(" doing primaries for %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001375 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Eli Friedman65d87222009-12-04 08:52:11 +00001376 Primaries(PrimaryBase, PrimaryBaseWasVirtual|MorallyVirtual, Offset,
1377 updateVBIndex, current_vbindex, CurrentVBaseOffset);
Mike Stump75ce5732009-10-31 20:06:59 +00001378 }
1379
1380 D1(printf(" doing vcall entries for %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001381 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Mike Stump75ce5732009-10-31 20:06:59 +00001382
1383 // And add the virtuals for the class to the primary vtable.
Eli Friedman03aa2f12009-11-30 01:19:33 +00001384 AddMethods(RD, MorallyVirtual, Offset, CurrentVBaseOffset);
Mike Stump75ce5732009-10-31 20:06:59 +00001385 }
1386
1387 void VBPrimaries(const CXXRecordDecl *RD, bool MorallyVirtual, int64_t Offset,
1388 bool updateVBIndex, Index_t current_vbindex,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001389 bool RDisVirtualBase, int64_t CurrentVBaseOffset,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001390 bool bottom) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001391 if (!RD->isDynamicClass())
1392 return;
1393
1394 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1395 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
1396 const bool PrimaryBaseWasVirtual = Layout.getPrimaryBaseWasVirtual();
1397
1398 // vtables are composed from the chain of primaries.
1399 if (PrimaryBase) {
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001400 int BaseCurrentVBaseOffset = CurrentVBaseOffset;
1401 if (PrimaryBaseWasVirtual) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001402 IndirectPrimary.insert(PrimaryBase);
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001403 BaseCurrentVBaseOffset = BLayout.getVBaseClassOffset(PrimaryBase);
1404 }
Mike Stump75ce5732009-10-31 20:06:59 +00001405
1406 D1(printf(" doing primaries for %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001407 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Mike Stump75ce5732009-10-31 20:06:59 +00001408
1409 VBPrimaries(PrimaryBase, PrimaryBaseWasVirtual|MorallyVirtual, Offset,
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001410 updateVBIndex, current_vbindex, PrimaryBaseWasVirtual,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001411 BaseCurrentVBaseOffset, false);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001412 }
1413
Mike Stump75ce5732009-10-31 20:06:59 +00001414 D1(printf(" doing vbase entries for %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001415 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Mike Stump75ce5732009-10-31 20:06:59 +00001416 GenerateVBaseOffsets(RD, Offset, updateVBIndex, current_vbindex);
1417
1418 if (RDisVirtualBase || bottom) {
1419 Primaries(RD, MorallyVirtual, Offset, updateVBIndex, current_vbindex,
Eli Friedman03aa2f12009-11-30 01:19:33 +00001420 CurrentVBaseOffset);
Mike Stump75ce5732009-10-31 20:06:59 +00001421 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001422 }
1423
Mike Stumpd538a6d2009-12-24 07:29:41 +00001424 void GenerateVtableForBase(const CXXRecordDecl *RD, int64_t Offset = 0,
1425 bool MorallyVirtual = false,
1426 bool ForVirtualBase = false,
Mike Stump9eb76d42010-01-22 06:45:05 +00001427 bool ForNPNVBases = false,
1428 bool WasPrimaryBase = true,
Mike Stumpd538a6d2009-12-24 07:29:41 +00001429 int CurrentVBaseOffset = 0,
1430 Path_t *Path = 0) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001431 if (!RD->isDynamicClass())
Mike Stumpd538a6d2009-12-24 07:29:41 +00001432 return;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001433
Mike Stumpfa818082009-11-13 02:35:38 +00001434 // Construction vtable don't need parts that have no virtual bases and
1435 // aren't morally virtual.
Anders Carlssonbad80eb2009-12-04 18:36:22 +00001436 if ((LayoutClass != MostDerivedClass) &&
1437 RD->getNumVBases() == 0 && !MorallyVirtual)
Mike Stumpd538a6d2009-12-24 07:29:41 +00001438 return;
Mike Stumpfa818082009-11-13 02:35:38 +00001439
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001440 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1441 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
1442 const bool PrimaryBaseWasVirtual = Layout.getPrimaryBaseWasVirtual();
1443
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001444 extra = 0;
Mike Stump75ce5732009-10-31 20:06:59 +00001445 D1(printf("building entries for base %s most derived %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001446 RD->getNameAsCString(), MostDerivedClass->getNameAsCString()));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001447
Mike Stump75ce5732009-10-31 20:06:59 +00001448 if (ForVirtualBase)
1449 extra = VCalls.size();
1450
Mike Stump9eb76d42010-01-22 06:45:05 +00001451 if (!ForNPNVBases || !WasPrimaryBase) {
1452 VBPrimaries(RD, MorallyVirtual, Offset, !ForVirtualBase, 0,
1453 ForVirtualBase, CurrentVBaseOffset, true);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001454
Mike Stump9eb76d42010-01-22 06:45:05 +00001455 if (Path)
1456 OverrideMethods(Path, MorallyVirtual, Offset, CurrentVBaseOffset);
1457 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001458
Mike Stump9eb76d42010-01-22 06:45:05 +00001459 FinishGenerateVtable(RD, Layout, PrimaryBase, ForNPNVBases, WasPrimaryBase,
1460 PrimaryBaseWasVirtual, MorallyVirtual, Offset,
1461 ForVirtualBase, CurrentVBaseOffset, Path);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001462 }
1463
1464 void GenerateVtableForVBases(const CXXRecordDecl *RD,
1465 int64_t Offset = 0,
Mike Stump37dbe962009-10-15 02:04:03 +00001466 Path_t *Path = 0) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001467 bool alloc = false;
1468 if (Path == 0) {
1469 alloc = true;
Mike Stump37dbe962009-10-15 02:04:03 +00001470 Path = new Path_t;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001471 }
1472 // FIXME: We also need to override using all paths to a virtual base,
1473 // right now, we just process the first path
1474 Path->push_back(std::make_pair(RD, Offset));
1475 for (CXXRecordDecl::base_class_const_iterator i = RD->bases_begin(),
1476 e = RD->bases_end(); i != e; ++i) {
1477 const CXXRecordDecl *Base =
1478 cast<CXXRecordDecl>(i->getType()->getAs<RecordType>()->getDecl());
1479 if (i->isVirtual() && !IndirectPrimary.count(Base)) {
1480 // Mark it so we don't output it twice.
1481 IndirectPrimary.insert(Base);
1482 StartNewTable();
Mike Stumpb21c4ee2009-10-14 18:14:51 +00001483 VCall.clear();
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001484 int64_t BaseOffset = BLayout.getVBaseClassOffset(Base);
Mike Stumpcd6f9ed2009-11-06 23:27:42 +00001485 int64_t CurrentVBaseOffset = BaseOffset;
Mike Stump75ce5732009-10-31 20:06:59 +00001486 D1(printf("vtable %s virtual base %s\n",
Mike Stumpd2808442010-01-22 02:51:26 +00001487 MostDerivedClass->getNameAsCString(), Base->getNameAsCString()));
Mike Stump9eb76d42010-01-22 06:45:05 +00001488 GenerateVtableForBase(Base, BaseOffset, true, true, false,
1489 true, CurrentVBaseOffset, Path);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001490 }
Mike Stump2cefe382009-11-12 20:47:57 +00001491 int64_t BaseOffset;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001492 if (i->isVirtual())
1493 BaseOffset = BLayout.getVBaseClassOffset(Base);
Mike Stump2cefe382009-11-12 20:47:57 +00001494 else {
1495 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1496 BaseOffset = Offset + Layout.getBaseClassOffset(Base);
1497 }
1498
Mike Stump37dbe962009-10-15 02:04:03 +00001499 if (Base->getNumVBases()) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001500 GenerateVtableForVBases(Base, BaseOffset, Path);
Mike Stump37dbe962009-10-15 02:04:03 +00001501 }
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001502 }
1503 Path->pop_back();
1504 if (alloc)
1505 delete Path;
1506 }
1507};
Anders Carlssonca1bf682009-12-03 01:54:02 +00001508} // end anonymous namespace
1509
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001510bool OldVtableBuilder::OverrideMethod(GlobalDecl GD, bool MorallyVirtual,
Eli Friedman81fb0d22009-12-04 08:40:51 +00001511 Index_t OverrideOffset, Index_t Offset,
1512 int64_t CurrentVBaseOffset) {
Anders Carlssonca1bf682009-12-03 01:54:02 +00001513 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
1514
1515 const bool isPure = MD->isPure();
Anders Carlsson21bbc1e2009-12-05 17:04:47 +00001516
Anders Carlssonca1bf682009-12-03 01:54:02 +00001517 // FIXME: Should OverrideOffset's be Offset?
1518
Anders Carlsson21bbc1e2009-12-05 17:04:47 +00001519 for (CXXMethodDecl::method_iterator mi = MD->begin_overridden_methods(),
1520 e = MD->end_overridden_methods(); mi != e; ++mi) {
Anders Carlssonca1bf682009-12-03 01:54:02 +00001521 GlobalDecl OGD;
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001522 GlobalDecl OGD2;
Anders Carlssonca1bf682009-12-03 01:54:02 +00001523
Anders Carlssonf3935b42009-12-04 05:51:56 +00001524 const CXXMethodDecl *OMD = *mi;
Anders Carlssonca1bf682009-12-03 01:54:02 +00001525 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(OMD))
1526 OGD = GlobalDecl(DD, GD.getDtorType());
1527 else
1528 OGD = OMD;
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001529
Eli Friedman8174f2c2009-12-06 22:01:30 +00001530 // Check whether this is the method being overridden in this section of
1531 // the vtable.
Anders Carlsson7bb70762009-12-04 15:49:02 +00001532 uint64_t Index;
1533 if (!Methods.getIndex(OGD, Index))
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001534 continue;
1535
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001536 OGD2 = OGD;
1537
Eli Friedman8174f2c2009-12-06 22:01:30 +00001538 // Get the original method, which we should be computing thunks, etc,
1539 // against.
1540 OGD = Methods.getOrigMethod(Index);
1541 OMD = cast<CXXMethodDecl>(OGD.getDecl());
1542
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001543 QualType ReturnType =
1544 MD->getType()->getAs<FunctionType>()->getResultType();
1545 QualType OverriddenReturnType =
1546 OMD->getType()->getAs<FunctionType>()->getResultType();
Anders Carlssonca1bf682009-12-03 01:54:02 +00001547
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001548 // Check if we need a return type adjustment.
1549 if (TypeConversionRequiresAdjustment(CGM.getContext(), ReturnType,
1550 OverriddenReturnType)) {
1551 CanQualType &BaseReturnType = BaseReturnTypes[Index];
Anders Carlssonca1bf682009-12-03 01:54:02 +00001552
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001553 // Insert the base return type.
1554 if (BaseReturnType.isNull())
1555 BaseReturnType =
1556 CGM.getContext().getCanonicalType(OverriddenReturnType);
1557 }
Anders Carlssona93e9802009-12-04 03:52:52 +00001558
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001559 Methods.OverrideMethod(OGD, GD);
1560
Anders Carlsson72281172010-01-26 17:36:47 +00001561 GlobalDecl UGD = getUnique(GD);
1562 const CXXMethodDecl *UMD = cast<CXXMethodDecl>(UGD.getDecl());
1563 assert(UGD.getDecl() && "unique overrider not found");
1564 assert(UGD == getUnique(OGD) && "unique overrider not unique");
Mike Stump90181eb2010-01-26 00:05:04 +00001565
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001566 ThisAdjustments.erase(Index);
Mike Stump90181eb2010-01-26 00:05:04 +00001567 if (MorallyVirtual || VCall.count(UMD)) {
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001568
Mike Stump90181eb2010-01-26 00:05:04 +00001569 Index_t &idx = VCall[UMD];
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001570 if (idx == 0) {
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001571 VCallOffset[GD] = VCallOffset[OGD];
1572 // NonVirtualOffset[UMD] = CurrentVBaseOffset/8 - OverrideOffset/8;
1573 NonVirtualOffset[UMD] = VCallOffset[OGD];
Mike Stump77537b12010-01-26 03:42:22 +00001574 VCallOffsetForVCall[UMD] = OverrideOffset/8;
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001575 idx = VCalls.size()+1;
Mike Stump77537b12010-01-26 03:42:22 +00001576 VCalls.push_back(OverrideOffset/8 - CurrentVBaseOffset/8);
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001577 D1(printf(" vcall for %s at %d with delta %d most derived %s\n",
1578 MD->getNameAsString().c_str(), (int)-idx-3,
Mike Stumpd2808442010-01-22 02:51:26 +00001579 (int)VCalls[idx-1], MostDerivedClass->getNameAsCString()));
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001580 } else {
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001581 VCallOffset[GD] = NonVirtualOffset[UMD];
Anders Carlsson72281172010-01-26 17:36:47 +00001582 VCalls[idx-1] = -VCallOffsetForVCall[UGD] + OverrideOffset/8;
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001583 D1(printf(" vcall patch for %s at %d with delta %d most derived %s\n",
1584 MD->getNameAsString().c_str(), (int)-idx-3,
Mike Stumpd2808442010-01-22 02:51:26 +00001585 (int)VCalls[idx-1], MostDerivedClass->getNameAsCString()));
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001586 }
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001587 int64_t NonVirtualAdjustment = -VCallOffset[OGD];
Mike Stumpded0a402010-01-26 22:44:01 +00001588 QualType DerivedType = MD->getThisType(CGM.getContext());
1589 QualType BaseType = cast<const CXXMethodDecl>(OGD.getDecl())->getThisType(CGM.getContext());
1590 int64_t NonVirtualAdjustment2 = -(getNVOffset(BaseType, DerivedType)/8);
1591 if (NonVirtualAdjustment2 != NonVirtualAdjustment) {
1592 NonVirtualAdjustment = NonVirtualAdjustment2;
1593 }
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001594 int64_t VirtualAdjustment =
1595 -((idx + extra + 2) * LLVMPointerWidth / 8);
Anders Carlsson657f1392009-12-03 02:32:59 +00001596
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001597 // Optimize out virtual adjustments of 0.
1598 if (VCalls[idx-1] == 0)
1599 VirtualAdjustment = 0;
Anders Carlsson657f1392009-12-03 02:32:59 +00001600
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001601 ThunkAdjustment ThisAdjustment(NonVirtualAdjustment,
1602 VirtualAdjustment);
Anders Carlsson2ca285f2009-12-03 02:22:59 +00001603
Eli Friedman8174f2c2009-12-06 22:01:30 +00001604 if (!isPure && !ThisAdjustment.isEmpty()) {
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001605 ThisAdjustments[Index] = ThisAdjustment;
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001606 SavedAdjustments.push_back(
1607 std::make_pair(GD, std::make_pair(OGD, ThisAdjustment)));
Eli Friedman8174f2c2009-12-06 22:01:30 +00001608 }
Anders Carlssonca1bf682009-12-03 01:54:02 +00001609 return true;
1610 }
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001611
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001612 VCallOffset[GD] = VCallOffset[OGD2] - OverrideOffset/8;
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001613
Mike Stumpa04ecfb2010-01-26 21:35:27 +00001614 int64_t NonVirtualAdjustment = -VCallOffset[GD];
1615 QualType DerivedType = MD->getThisType(CGM.getContext());
1616 QualType BaseType = cast<const CXXMethodDecl>(OGD.getDecl())->getThisType(CGM.getContext());
1617 int64_t NonVirtualAdjustment2 = -(getNVOffset(BaseType, DerivedType)/8);
1618 if (NonVirtualAdjustment2 != NonVirtualAdjustment) {
1619 NonVirtualAdjustment = NonVirtualAdjustment2;
1620 }
1621
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001622 if (NonVirtualAdjustment) {
1623 ThunkAdjustment ThisAdjustment(NonVirtualAdjustment, 0);
1624
Eli Friedmanf2eda5e2009-12-06 22:33:51 +00001625 if (!isPure) {
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001626 ThisAdjustments[Index] = ThisAdjustment;
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001627 SavedAdjustments.push_back(
1628 std::make_pair(GD, std::make_pair(OGD, ThisAdjustment)));
Eli Friedmanf2eda5e2009-12-06 22:33:51 +00001629 }
Eli Friedman3d2e9de2009-12-04 08:34:14 +00001630 }
1631 return true;
Anders Carlssonca1bf682009-12-03 01:54:02 +00001632 }
1633
1634 return false;
Anders Carlssond59885022009-11-27 22:21:51 +00001635}
1636
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001637void OldVtableBuilder::AppendMethodsToVtable() {
Eli Friedman6c08ce72009-12-05 01:05:03 +00001638 if (!BuildVtable) {
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001639 VtableComponents.insert(VtableComponents.end(), Methods.size(),
1640 (llvm::Constant *)0);
Eli Friedman6c08ce72009-12-05 01:05:03 +00001641 ThisAdjustments.clear();
1642 BaseReturnTypes.clear();
1643 Methods.clear();
1644 return;
1645 }
1646
Anders Carlsson472404f2009-12-04 16:19:30 +00001647 // Reserve room in the vtable for our new methods.
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001648 VtableComponents.reserve(VtableComponents.size() + Methods.size());
Anders Carlssonb07567c2009-12-04 03:07:26 +00001649
Anders Carlsson86809cd2009-12-04 02:43:50 +00001650 for (unsigned i = 0, e = Methods.size(); i != e; ++i) {
1651 GlobalDecl GD = Methods[i];
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00001652 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
1653
1654 // Get the 'this' pointer adjustment.
Anders Carlsson86809cd2009-12-04 02:43:50 +00001655 ThunkAdjustment ThisAdjustment = ThisAdjustments.lookup(i);
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00001656
1657 // Construct the return type adjustment.
1658 ThunkAdjustment ReturnAdjustment;
1659
1660 QualType BaseReturnType = BaseReturnTypes.lookup(i);
1661 if (!BaseReturnType.isNull() && !MD->isPure()) {
1662 QualType DerivedType =
1663 MD->getType()->getAs<FunctionType>()->getResultType();
1664
1665 int64_t NonVirtualAdjustment =
1666 getNVOffset(BaseReturnType, DerivedType) / 8;
1667
1668 int64_t VirtualAdjustment =
1669 getVbaseOffset(BaseReturnType, DerivedType);
1670
1671 ReturnAdjustment = ThunkAdjustment(NonVirtualAdjustment,
1672 VirtualAdjustment);
1673 }
1674
Anders Carlsson50f14742009-12-04 03:06:03 +00001675 llvm::Constant *Method = 0;
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00001676 if (!ReturnAdjustment.isEmpty()) {
1677 // Build a covariant thunk.
1678 CovariantThunkAdjustment Adjustment(ThisAdjustment, ReturnAdjustment);
Eli Friedman8174f2c2009-12-06 22:01:30 +00001679 Method = wrap(CGM.GetAddrOfCovariantThunk(GD, Adjustment));
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00001680 } else if (!ThisAdjustment.isEmpty()) {
1681 // Build a "regular" thunk.
Eli Friedman8174f2c2009-12-06 22:01:30 +00001682 Method = wrap(CGM.GetAddrOfThunk(GD, ThisAdjustment));
Anders Carlssonddf42c82009-12-04 02:56:03 +00001683 } else if (MD->isPure()) {
1684 // We have a pure virtual method.
Anders Carlsson50f14742009-12-04 03:06:03 +00001685 Method = getPureVirtualFn();
1686 } else {
1687 // We have a good old regular method.
1688 Method = WrapAddrOf(GD);
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00001689 }
Anders Carlsson50f14742009-12-04 03:06:03 +00001690
1691 // Add the method to the vtable.
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001692 VtableComponents.push_back(Method);
Anders Carlsson86809cd2009-12-04 02:43:50 +00001693 }
1694
Anders Carlsson50f14742009-12-04 03:06:03 +00001695
Anders Carlsson86809cd2009-12-04 02:43:50 +00001696 ThisAdjustments.clear();
Anders Carlsson5b3ea9b2009-12-04 02:52:22 +00001697 BaseReturnTypes.clear();
Anders Carlsson86809cd2009-12-04 02:43:50 +00001698
Anders Carlsson495634e2009-12-04 02:39:04 +00001699 Methods.clear();
Anders Carlsson495634e2009-12-04 02:39:04 +00001700}
1701
Anders Carlssonf942ee02009-11-27 20:47:55 +00001702void CGVtableInfo::ComputeMethodVtableIndices(const CXXRecordDecl *RD) {
1703
1704 // Itanium C++ ABI 2.5.2:
Anders Carlsson259688c2010-02-02 03:37:46 +00001705 // The order of the virtual function pointers in a virtual table is the
1706 // order of declaration of the corresponding member functions in the class.
Anders Carlssonf942ee02009-11-27 20:47:55 +00001707 //
Anders Carlsson259688c2010-02-02 03:37:46 +00001708 // There is an entry for any virtual function declared in a class,
1709 // whether it is a new function or overrides a base class function,
1710 // unless it overrides a function from the primary base, and conversion
1711 // between their return types does not require an adjustment.
Anders Carlssonf942ee02009-11-27 20:47:55 +00001712
1713 int64_t CurrentIndex = 0;
1714
1715 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1716 const CXXRecordDecl *PrimaryBase = Layout.getPrimaryBase();
1717
1718 if (PrimaryBase) {
Anders Carlssonc920fa22009-11-30 19:43:26 +00001719 assert(PrimaryBase->isDefinition() &&
1720 "Should have the definition decl of the primary base!");
Anders Carlssonf942ee02009-11-27 20:47:55 +00001721
1722 // Since the record decl shares its vtable pointer with the primary base
1723 // we need to start counting at the end of the primary base's vtable.
1724 CurrentIndex = getNumVirtualFunctionPointers(PrimaryBase);
1725 }
Eli Friedman21517252009-12-15 03:31:17 +00001726
1727 // Collect all the primary bases, so we can check whether methods override
1728 // a method from the base.
1729 llvm::SmallPtrSet<const CXXRecordDecl *, 5> PrimaryBases;
1730 for (ASTRecordLayout::primary_base_info_iterator
1731 I = Layout.primary_base_begin(), E = Layout.primary_base_end();
1732 I != E; ++I)
1733 PrimaryBases.insert((*I).getBase());
1734
Anders Carlssonf942ee02009-11-27 20:47:55 +00001735 const CXXDestructorDecl *ImplicitVirtualDtor = 0;
1736
1737 for (CXXRecordDecl::method_iterator i = RD->method_begin(),
1738 e = RD->method_end(); i != e; ++i) {
1739 const CXXMethodDecl *MD = *i;
1740
1741 // We only want virtual methods.
1742 if (!MD->isVirtual())
1743 continue;
1744
1745 bool ShouldAddEntryForMethod = true;
1746
1747 // Check if this method overrides a method in the primary base.
1748 for (CXXMethodDecl::method_iterator i = MD->begin_overridden_methods(),
1749 e = MD->end_overridden_methods(); i != e; ++i) {
Anders Carlssonf3935b42009-12-04 05:51:56 +00001750 const CXXMethodDecl *OverriddenMD = *i;
Anders Carlssonf942ee02009-11-27 20:47:55 +00001751 const CXXRecordDecl *OverriddenRD = OverriddenMD->getParent();
1752 assert(OverriddenMD->isCanonicalDecl() &&
1753 "Should have the canonical decl of the overridden RD!");
1754
Eli Friedman21517252009-12-15 03:31:17 +00001755 if (PrimaryBases.count(OverriddenRD)) {
Anders Carlssonf942ee02009-11-27 20:47:55 +00001756 // Check if converting from the return type of the method to the
1757 // return type of the overridden method requires conversion.
1758 QualType ReturnType =
1759 MD->getType()->getAs<FunctionType>()->getResultType();
1760 QualType OverriddenReturnType =
1761 OverriddenMD->getType()->getAs<FunctionType>()->getResultType();
1762
1763 if (!TypeConversionRequiresAdjustment(CGM.getContext(),
1764 ReturnType, OverriddenReturnType)) {
1765 // This index is shared between the index in the vtable of the primary
1766 // base class.
1767 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
1768 const CXXDestructorDecl *OverriddenDD =
1769 cast<CXXDestructorDecl>(OverriddenMD);
1770
1771 // Add both the complete and deleting entries.
1772 MethodVtableIndices[GlobalDecl(DD, Dtor_Complete)] =
1773 getMethodVtableIndex(GlobalDecl(OverriddenDD, Dtor_Complete));
1774 MethodVtableIndices[GlobalDecl(DD, Dtor_Deleting)] =
1775 getMethodVtableIndex(GlobalDecl(OverriddenDD, Dtor_Deleting));
1776 } else {
1777 MethodVtableIndices[MD] = getMethodVtableIndex(OverriddenMD);
1778 }
1779
1780 // We don't need to add an entry for this method.
1781 ShouldAddEntryForMethod = false;
1782 break;
1783 }
1784 }
1785 }
1786
1787 if (!ShouldAddEntryForMethod)
1788 continue;
1789
1790 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(MD)) {
1791 if (MD->isImplicit()) {
1792 assert(!ImplicitVirtualDtor &&
1793 "Did already see an implicit virtual dtor!");
1794 ImplicitVirtualDtor = DD;
1795 continue;
1796 }
1797
1798 // Add the complete dtor.
1799 MethodVtableIndices[GlobalDecl(DD, Dtor_Complete)] = CurrentIndex++;
1800
1801 // Add the deleting dtor.
1802 MethodVtableIndices[GlobalDecl(DD, Dtor_Deleting)] = CurrentIndex++;
1803 } else {
1804 // Add the entry.
1805 MethodVtableIndices[MD] = CurrentIndex++;
1806 }
1807 }
1808
1809 if (ImplicitVirtualDtor) {
1810 // Itanium C++ ABI 2.5.2:
1811 // If a class has an implicitly-defined virtual destructor,
1812 // its entries come after the declared virtual function pointers.
1813
1814 // Add the complete dtor.
1815 MethodVtableIndices[GlobalDecl(ImplicitVirtualDtor, Dtor_Complete)] =
1816 CurrentIndex++;
1817
1818 // Add the deleting dtor.
1819 MethodVtableIndices[GlobalDecl(ImplicitVirtualDtor, Dtor_Deleting)] =
1820 CurrentIndex++;
1821 }
1822
1823 NumVirtualFunctionPointers[RD] = CurrentIndex;
1824}
1825
1826uint64_t CGVtableInfo::getNumVirtualFunctionPointers(const CXXRecordDecl *RD) {
1827 llvm::DenseMap<const CXXRecordDecl *, uint64_t>::iterator I =
1828 NumVirtualFunctionPointers.find(RD);
1829 if (I != NumVirtualFunctionPointers.end())
1830 return I->second;
1831
1832 ComputeMethodVtableIndices(RD);
1833
1834 I = NumVirtualFunctionPointers.find(RD);
1835 assert(I != NumVirtualFunctionPointers.end() && "Did not find entry!");
1836 return I->second;
1837}
1838
1839uint64_t CGVtableInfo::getMethodVtableIndex(GlobalDecl GD) {
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001840 MethodVtableIndicesTy::iterator I = MethodVtableIndices.find(GD);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001841 if (I != MethodVtableIndices.end())
1842 return I->second;
1843
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001844 const CXXRecordDecl *RD = cast<CXXMethodDecl>(GD.getDecl())->getParent();
Anders Carlssonf942ee02009-11-27 20:47:55 +00001845
1846 ComputeMethodVtableIndices(RD);
1847
Anders Carlssonfb4dda42009-11-13 17:08:56 +00001848 I = MethodVtableIndices.find(GD);
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001849 assert(I != MethodVtableIndices.end() && "Did not find index!");
1850 return I->second;
1851}
1852
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001853CGVtableInfo::AdjustmentVectorTy*
1854CGVtableInfo::getAdjustments(GlobalDecl GD) {
1855 SavedAdjustmentsTy::iterator I = SavedAdjustments.find(GD);
1856 if (I != SavedAdjustments.end())
1857 return &I->second;
Eli Friedman8174f2c2009-12-06 22:01:30 +00001858
1859 const CXXRecordDecl *RD = cast<CXXRecordDecl>(GD.getDecl()->getDeclContext());
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001860 if (!SavedAdjustmentRecords.insert(RD).second)
1861 return 0;
Eli Friedman8174f2c2009-12-06 22:01:30 +00001862
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001863 AddressPointsMapTy AddressPoints;
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001864 OldVtableBuilder b(RD, RD, 0, CGM, false, AddressPoints);
Eli Friedman8174f2c2009-12-06 22:01:30 +00001865 D1(printf("vtable %s\n", RD->getNameAsCString()));
1866 b.GenerateVtableForBase(RD);
1867 b.GenerateVtableForVBases(RD);
Eli Friedman8174f2c2009-12-06 22:01:30 +00001868
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001869 for (OldVtableBuilder::SavedAdjustmentsVectorTy::iterator
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001870 i = b.getSavedAdjustments().begin(),
1871 e = b.getSavedAdjustments().end(); i != e; i++)
1872 SavedAdjustments[i->first].push_back(i->second);
Eli Friedman8174f2c2009-12-06 22:01:30 +00001873
Eli Friedman31bc3ad2009-12-07 23:56:34 +00001874 I = SavedAdjustments.find(GD);
1875 if (I != SavedAdjustments.end())
1876 return &I->second;
1877
1878 return 0;
Eli Friedman8174f2c2009-12-06 22:01:30 +00001879}
1880
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001881int64_t CGVtableInfo::getVirtualBaseOffsetIndex(const CXXRecordDecl *RD,
1882 const CXXRecordDecl *VBase) {
1883 ClassPairTy ClassPair(RD, VBase);
1884
1885 VirtualBaseClassIndiciesTy::iterator I =
1886 VirtualBaseClassIndicies.find(ClassPair);
1887 if (I != VirtualBaseClassIndicies.end())
1888 return I->second;
1889
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001890 // FIXME: This seems expensive. Can we do a partial job to get
1891 // just this data.
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001892 AddressPointsMapTy AddressPoints;
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001893 OldVtableBuilder b(RD, RD, 0, CGM, false, AddressPoints);
Mike Stump75ce5732009-10-31 20:06:59 +00001894 D1(printf("vtable %s\n", RD->getNameAsCString()));
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001895 b.GenerateVtableForBase(RD);
1896 b.GenerateVtableForVBases(RD);
1897
1898 for (llvm::DenseMap<const CXXRecordDecl *, uint64_t>::iterator I =
1899 b.getVBIndex().begin(), E = b.getVBIndex().end(); I != E; ++I) {
1900 // Insert all types.
1901 ClassPairTy ClassPair(RD, I->first);
1902
1903 VirtualBaseClassIndicies.insert(std::make_pair(ClassPair, I->second));
1904 }
1905
1906 I = VirtualBaseClassIndicies.find(ClassPair);
1907 assert(I != VirtualBaseClassIndicies.end() && "Did not find index!");
1908
1909 return I->second;
1910}
1911
Anders Carlssonc8e39ec2009-12-05 21:03:56 +00001912uint64_t CGVtableInfo::getVtableAddressPoint(const CXXRecordDecl *RD) {
1913 uint64_t AddressPoint =
Anders Carlsson93a18842010-01-02 18:02:32 +00001914 (*(*(CGM.getVtableInfo().AddressPoints[RD]))[RD])[std::make_pair(RD, 0)];
Anders Carlssonc8e39ec2009-12-05 21:03:56 +00001915
1916 return AddressPoint;
1917}
1918
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00001919llvm::GlobalVariable *
Anders Carlsson0911ae82009-12-06 00:23:49 +00001920CGVtableInfo::GenerateVtable(llvm::GlobalVariable::LinkageTypes Linkage,
Anders Carlsson232324c2009-12-06 00:53:22 +00001921 bool GenerateDefinition,
Anders Carlsson0911ae82009-12-06 00:23:49 +00001922 const CXXRecordDecl *LayoutClass,
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001923 const CXXRecordDecl *RD, uint64_t Offset,
1924 AddressPointsMapTy& AddressPoints) {
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001925 llvm::SmallString<256> OutName;
Mike Stump2cefe382009-11-12 20:47:57 +00001926 if (LayoutClass != RD)
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00001927 CGM.getMangleContext().mangleCXXCtorVtable(LayoutClass, Offset / 8,
1928 RD, OutName);
Mike Stumpeac45592009-11-11 20:26:26 +00001929 else
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00001930 CGM.getMangleContext().mangleCXXVtable(RD, OutName);
Daniel Dunbare128dd12009-11-21 09:06:22 +00001931 llvm::StringRef Name = OutName.str();
Benjamin Kramerbb0a07b2009-10-11 22:57:54 +00001932
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00001933 llvm::GlobalVariable *GV = CGM.getModule().getGlobalVariable(Name);
Anders Carlsson93a18842010-01-02 18:02:32 +00001934 if (GV == 0 || CGM.getVtableInfo().AddressPoints[LayoutClass] == 0 ||
1935 GV->isDeclaration()) {
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001936 OldVtableBuilder b(RD, LayoutClass, Offset, CGM, GenerateDefinition,
1937 AddressPoints);
Eli Friedman6c08ce72009-12-05 01:05:03 +00001938
1939 D1(printf("vtable %s\n", RD->getNameAsCString()));
1940 // First comes the vtables for all the non-virtual bases...
Mike Stumpd538a6d2009-12-24 07:29:41 +00001941 b.GenerateVtableForBase(RD, Offset);
Eli Friedman6c08ce72009-12-05 01:05:03 +00001942
1943 // then the vtables for all the virtual bases.
1944 b.GenerateVtableForVBases(RD, Offset);
1945
Anders Carlsson58b271d2009-12-05 22:19:10 +00001946 llvm::Constant *Init = 0;
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00001947 const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGM.getLLVMContext());
Anders Carlsson58b271d2009-12-05 22:19:10 +00001948 llvm::ArrayType *ArrayType =
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001949 llvm::ArrayType::get(Int8PtrTy, b.getVtableComponents().size());
Anders Carlssona95d4c52009-12-05 21:09:05 +00001950
Anders Carlsson232324c2009-12-06 00:53:22 +00001951 if (GenerateDefinition)
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001952 Init = llvm::ConstantArray::get(ArrayType, &b.getVtableComponents()[0],
1953 b.getVtableComponents().size());
Anders Carlsson232324c2009-12-06 00:53:22 +00001954
Mike Stumpaa51ad62009-11-19 04:04:36 +00001955 llvm::GlobalVariable *OGV = GV;
Anders Carlsson232324c2009-12-06 00:53:22 +00001956
1957 GV = new llvm::GlobalVariable(CGM.getModule(), ArrayType,
1958 /*isConstant=*/true, Linkage, Init, Name);
Anders Carlssonfe5f7d92009-12-06 01:09:21 +00001959 CGM.setGlobalVisibility(GV, RD);
1960
Mike Stumpaa51ad62009-11-19 04:04:36 +00001961 if (OGV) {
1962 GV->takeName(OGV);
Anders Carlsson58b271d2009-12-05 22:19:10 +00001963 llvm::Constant *NewPtr =
1964 llvm::ConstantExpr::getBitCast(GV, OGV->getType());
Mike Stumpaa51ad62009-11-19 04:04:36 +00001965 OGV->replaceAllUsesWith(NewPtr);
1966 OGV->eraseFromParent();
1967 }
Mike Stumpaa51ad62009-11-19 04:04:36 +00001968 }
Anders Carlssonb3f54b72009-12-05 21:28:12 +00001969
Anders Carlsson5d40c6f2010-02-11 08:02:13 +00001970 if (GenerateDefinition && CGM.getLangOptions().DumpVtableLayouts) {
1971 VtableBuilder Builder(RD);
1972
1973 Builder.dumpLayout(llvm::errs());
1974 }
1975
Anders Carlssonb3f54b72009-12-05 21:28:12 +00001976 return GV;
Anders Carlsson2bb27f52009-10-11 22:13:54 +00001977}
Mike Stump9f23a142009-11-10 02:30:51 +00001978
Anders Carlsson232324c2009-12-06 00:53:22 +00001979void CGVtableInfo::GenerateClassData(llvm::GlobalVariable::LinkageTypes Linkage,
1980 const CXXRecordDecl *RD) {
Anders Carlssone1b3e622009-12-07 07:59:52 +00001981 llvm::GlobalVariable *&Vtable = Vtables[RD];
1982 if (Vtable) {
1983 assert(Vtable->getInitializer() && "Vtable doesn't have a definition!");
1984 return;
1985 }
1986
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001987 AddressPointsMapTy AddressPoints;
1988 Vtable = GenerateVtable(Linkage, /*GenerateDefinition=*/true, RD, RD, 0,
1989 AddressPoints);
Anders Carlssone36a6b32010-01-02 01:01:18 +00001990 GenerateVTT(Linkage, /*GenerateDefinition=*/true, RD);
Mike Stump1a139f82009-11-19 01:08:19 +00001991}
1992
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00001993llvm::GlobalVariable *CGVtableInfo::getVtable(const CXXRecordDecl *RD) {
Anders Carlsson232324c2009-12-06 00:53:22 +00001994 llvm::GlobalVariable *Vtable = Vtables.lookup(RD);
Anders Carlsson7e28c5f2009-12-06 00:01:05 +00001995
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001996 if (!Vtable) {
1997 AddressPointsMapTy AddressPoints;
Anders Carlsson232324c2009-12-06 00:53:22 +00001998 Vtable = GenerateVtable(llvm::GlobalValue::ExternalLinkage,
Anders Carlsson5f9a8812010-01-14 02:29:07 +00001999 /*GenerateDefinition=*/false, RD, RD, 0,
2000 AddressPoints);
2001 }
Mike Stumpaa51ad62009-11-19 04:04:36 +00002002
Anders Carlsson4ed44eb2009-12-05 22:42:54 +00002003 return Vtable;
Mike Stumpd846d082009-11-10 07:44:33 +00002004}
Mike Stumpeac45592009-11-11 20:26:26 +00002005
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002006void CGVtableInfo::MaybeEmitVtable(GlobalDecl GD) {
2007 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
2008 const CXXRecordDecl *RD = MD->getParent();
2009
Anders Carlsson4ed44eb2009-12-05 22:42:54 +00002010 // If the class doesn't have a vtable we don't need to emit one.
2011 if (!RD->isDynamicClass())
2012 return;
2013
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002014 // Get the key function.
Anders Carlsson5ebf8b42009-12-07 04:35:11 +00002015 const CXXMethodDecl *KeyFunction = CGM.getContext().getKeyFunction(RD);
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002016
Anders Carlsson4ed44eb2009-12-05 22:42:54 +00002017 if (KeyFunction) {
2018 // We don't have the right key function.
2019 if (KeyFunction->getCanonicalDecl() != MD->getCanonicalDecl())
2020 return;
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002021 }
2022
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002023 // Emit the data.
Douglas Gregorccecc1b2010-01-06 20:27:16 +00002024 GenerateClassData(CGM.getVtableLinkage(RD), RD);
Eli Friedman8174f2c2009-12-06 22:01:30 +00002025
2026 for (CXXRecordDecl::method_iterator i = RD->method_begin(),
2027 e = RD->method_end(); i != e; ++i) {
Eli Friedman31bc3ad2009-12-07 23:56:34 +00002028 if ((*i)->isVirtual() && ((*i)->hasInlineBody() || (*i)->isImplicit())) {
Eli Friedman8174f2c2009-12-06 22:01:30 +00002029 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(*i)) {
2030 CGM.BuildThunksForVirtual(GlobalDecl(DD, Dtor_Complete));
2031 CGM.BuildThunksForVirtual(GlobalDecl(DD, Dtor_Deleting));
2032 } else {
2033 CGM.BuildThunksForVirtual(GlobalDecl(*i));
2034 }
2035 }
2036 }
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +00002037}
2038