1 //===--- CGVTables.h - Emit LLVM Code for C++ vtables -----------*- C++ -*-===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This contains code dealing with C++ code generation of virtual tables. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef CLANG_CODEGEN_CGVTABLE_H 15 #define CLANG_CODEGEN_CGVTABLE_H 16 17 #include "llvm/ADT/DenseMap.h" 18 #include "llvm/GlobalVariable.h" 19 #include "clang/Basic/ABI.h" 20 #include "clang/AST/BaseSubobject.h" 21 #include "clang/AST/CharUnits.h" 22 #include "clang/AST/GlobalDecl.h" 23 #include "clang/AST/VTableBuilder.h" 24 25 namespace clang { 26 class CXXRecordDecl; 27 28 namespace CodeGen { 29 class CodeGenModule; 30 31 class CodeGenVTables { 32 CodeGenModule &CGM; 33 34 VTableContext VTContext; 35 36 /// VTables - All the vtables which have been defined. 37 llvm::DenseMap<const CXXRecordDecl *, llvm::GlobalVariable *> VTables; 38 39 /// VTableAddressPointsMapTy - Address points for a single vtable. 40 typedef llvm::DenseMap<BaseSubobject, uint64_t> VTableAddressPointsMapTy; 41 42 typedef std::pair<const CXXRecordDecl *, BaseSubobject> BaseSubobjectPairTy; 43 typedef llvm::DenseMap<BaseSubobjectPairTy, uint64_t> SubVTTIndiciesMapTy; 44 45 /// SubVTTIndicies - Contains indices into the various sub-VTTs. 46 SubVTTIndiciesMapTy SubVTTIndicies; 47 48 typedef llvm::DenseMap<BaseSubobjectPairTy, uint64_t> 49 SecondaryVirtualPointerIndicesMapTy; 50 51 /// SecondaryVirtualPointerIndices - Contains the secondary virtual pointer 52 /// indices. 53 SecondaryVirtualPointerIndicesMapTy SecondaryVirtualPointerIndices; 54 55 /// EmitThunk - Emit a single thunk. 56 void EmitThunk(GlobalDecl GD, const ThunkInfo &Thunk, 57 bool UseAvailableExternallyLinkage); 58 59 /// MaybeEmitThunkAvailableExternally - Try to emit the given thunk with 60 /// available_externally linkage to allow for inlining of thunks. 61 /// This will be done iff optimizations are enabled and the member function 62 /// doesn't contain any incomplete types. 63 void MaybeEmitThunkAvailableExternally(GlobalDecl GD, const ThunkInfo &Thunk); 64 65 /// CreateVTableInitializer - Create a vtable initializer for the given record 66 /// decl. 67 /// \param Components - The vtable components; this is really an array of 68 /// VTableComponents. 69 llvm::Constant *CreateVTableInitializer(const CXXRecordDecl *RD, 70 const VTableComponent *Components, 71 unsigned NumComponents, 72 const VTableLayout::VTableThunkTy *VTableThunks, 73 unsigned NumVTableThunks); 74 75 public: 76 CodeGenVTables(CodeGenModule &CGM); 77 getVTableContext()78 VTableContext &getVTableContext() { return VTContext; } 79 80 /// \brief True if the VTable of this record must be emitted in the 81 /// translation unit. 82 bool ShouldEmitVTableInThisTU(const CXXRecordDecl *RD); 83 84 /// needsVTTParameter - Return whether the given global decl needs a VTT 85 /// parameter, which it does if it's a base constructor or destructor with 86 /// virtual bases. 87 static bool needsVTTParameter(GlobalDecl GD); 88 89 /// getSubVTTIndex - Return the index of the sub-VTT for the base class of the 90 /// given record decl. 91 uint64_t getSubVTTIndex(const CXXRecordDecl *RD, BaseSubobject Base); 92 93 /// getSecondaryVirtualPointerIndex - Return the index in the VTT where the 94 /// virtual pointer for the given subobject is located. 95 uint64_t getSecondaryVirtualPointerIndex(const CXXRecordDecl *RD, 96 BaseSubobject Base); 97 98 /// getAddressPoint - Get the address point of the given subobject in the 99 /// class decl. 100 uint64_t getAddressPoint(BaseSubobject Base, const CXXRecordDecl *RD); 101 102 /// GetAddrOfVTable - Get the address of the vtable for the given record decl. 103 llvm::GlobalVariable *GetAddrOfVTable(const CXXRecordDecl *RD); 104 105 /// EmitVTableDefinition - Emit the definition of the given vtable. 106 void EmitVTableDefinition(llvm::GlobalVariable *VTable, 107 llvm::GlobalVariable::LinkageTypes Linkage, 108 const CXXRecordDecl *RD); 109 110 /// GenerateConstructionVTable - Generate a construction vtable for the given 111 /// base subobject. 112 llvm::GlobalVariable * 113 GenerateConstructionVTable(const CXXRecordDecl *RD, const BaseSubobject &Base, 114 bool BaseIsVirtual, 115 llvm::GlobalVariable::LinkageTypes Linkage, 116 VTableAddressPointsMapTy& AddressPoints); 117 118 119 /// GetAddrOfVTable - Get the address of the VTT for the given record decl. 120 llvm::GlobalVariable *GetAddrOfVTT(const CXXRecordDecl *RD); 121 122 /// EmitVTTDefinition - Emit the definition of the given vtable. 123 void EmitVTTDefinition(llvm::GlobalVariable *VTT, 124 llvm::GlobalVariable::LinkageTypes Linkage, 125 const CXXRecordDecl *RD); 126 127 /// EmitThunks - Emit the associated thunks for the given global decl. 128 void EmitThunks(GlobalDecl GD); 129 130 /// GenerateClassData - Generate all the class data required to be generated 131 /// upon definition of a KeyFunction. This includes the vtable, the 132 /// rtti data structure and the VTT. 133 /// 134 /// \param Linkage - The desired linkage of the vtable, the RTTI and the VTT. 135 void GenerateClassData(llvm::GlobalVariable::LinkageTypes Linkage, 136 const CXXRecordDecl *RD); 137 }; 138 139 } // end namespace CodeGen 140 } // end namespace clang 141 #endif 142