blob: bfe967dc2d7f296723d7caabf5d66bb7d83c4137 [file] [log] [blame]
Anders Carlsson5b955922009-11-24 05:51:11 +00001//===--- CGExprCXX.cpp - Emit LLVM Code for C++ expressions ---------------===//
Anders Carlsson16d81b82009-09-22 22:53:17 +00002//
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 code generation of C++ expressions
11//
12//===----------------------------------------------------------------------===//
13
14#include "CodeGenFunction.h"
15using namespace clang;
16using namespace CodeGen;
17
Anders Carlsson3b5ad222010-01-01 20:29:01 +000018RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD,
19 llvm::Value *Callee,
20 ReturnValueSlot ReturnValue,
21 llvm::Value *This,
Anders Carlssonc997d422010-01-02 01:01:18 +000022 llvm::Value *VTT,
Anders Carlsson3b5ad222010-01-01 20:29:01 +000023 CallExpr::const_arg_iterator ArgBeg,
24 CallExpr::const_arg_iterator ArgEnd) {
25 assert(MD->isInstance() &&
26 "Trying to emit a member call expr on a static method!");
27
28 const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
29
30 CallArgList Args;
31
32 // Push the this ptr.
33 Args.push_back(std::make_pair(RValue::get(This),
34 MD->getThisType(getContext())));
35
Anders Carlssonc997d422010-01-02 01:01:18 +000036 // If there is a VTT parameter, emit it.
37 if (VTT) {
38 QualType T = getContext().getPointerType(getContext().VoidPtrTy);
39 Args.push_back(std::make_pair(RValue::get(VTT), T));
40 }
41
Anders Carlsson3b5ad222010-01-01 20:29:01 +000042 // And the rest of the call args
43 EmitCallArgs(Args, FPT, ArgBeg, ArgEnd);
44
John McCall04a67a62010-02-05 21:31:56 +000045 QualType ResultType = FPT->getResultType();
46 return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args,
Rafael Espindola264ba482010-03-30 20:24:48 +000047 FPT->getExtInfo()),
48 Callee, ReturnValue, Args, MD);
Anders Carlsson3b5ad222010-01-01 20:29:01 +000049}
50
51/// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given
52/// expr can be devirtualized.
53static bool canDevirtualizeMemberFunctionCalls(const Expr *Base) {
54 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) {
55 if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
56 // This is a record decl. We know the type and can devirtualize it.
57 return VD->getType()->isRecordType();
58 }
59
60 return false;
61 }
62
63 // We can always devirtualize calls on temporary object expressions.
Eli Friedman6997aae2010-01-31 20:58:15 +000064 if (isa<CXXConstructExpr>(Base))
Anders Carlsson3b5ad222010-01-01 20:29:01 +000065 return true;
66
67 // And calls on bound temporaries.
68 if (isa<CXXBindTemporaryExpr>(Base))
69 return true;
70
71 // Check if this is a call expr that returns a record type.
72 if (const CallExpr *CE = dyn_cast<CallExpr>(Base))
73 return CE->getCallReturnType()->isRecordType();
74
75 // We can't devirtualize the call.
76 return false;
77}
78
79RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
80 ReturnValueSlot ReturnValue) {
81 if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens()))
82 return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
83
84 const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens());
85 const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
86
87 if (MD->isStatic()) {
88 // The method is static, emit it as we would a regular call.
89 llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
90 return EmitCall(getContext().getPointerType(MD->getType()), Callee,
91 ReturnValue, CE->arg_begin(), CE->arg_end());
92 }
93
94 const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
95
96 const llvm::Type *Ty =
97 CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
98 FPT->isVariadic());
99 llvm::Value *This;
100
101 if (ME->isArrow())
102 This = EmitScalarExpr(ME->getBase());
103 else {
104 LValue BaseLV = EmitLValue(ME->getBase());
105 This = BaseLV.getAddress();
106 }
107
108 if (MD->isCopyAssignment() && MD->isTrivial()) {
109 // We don't like to generate the trivial copy assignment operator when
110 // it isn't necessary; just produce the proper effect here.
111 llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
112 EmitAggregateCopy(This, RHS, CE->getType());
113 return RValue::get(This);
114 }
115
116 // C++ [class.virtual]p12:
117 // Explicit qualification with the scope operator (5.1) suppresses the
118 // virtual call mechanism.
119 //
120 // We also don't emit a virtual call if the base expression has a record type
121 // because then we know what the type is.
122 llvm::Value *Callee;
123 if (const CXXDestructorDecl *Destructor
124 = dyn_cast<CXXDestructorDecl>(MD)) {
125 if (Destructor->isTrivial())
126 return RValue::get(0);
127 if (MD->isVirtual() && !ME->hasQualifier() &&
128 !canDevirtualizeMemberFunctionCalls(ME->getBase())) {
129 Callee = BuildVirtualCall(Destructor, Dtor_Complete, This, Ty);
130 } else {
131 Callee = CGM.GetAddrOfFunction(GlobalDecl(Destructor, Dtor_Complete), Ty);
132 }
133 } else if (MD->isVirtual() && !ME->hasQualifier() &&
134 !canDevirtualizeMemberFunctionCalls(ME->getBase())) {
135 Callee = BuildVirtualCall(MD, This, Ty);
136 } else {
137 Callee = CGM.GetAddrOfFunction(MD, Ty);
138 }
139
Anders Carlssonc997d422010-01-02 01:01:18 +0000140 return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000141 CE->arg_begin(), CE->arg_end());
142}
143
144RValue
145CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
146 ReturnValueSlot ReturnValue) {
147 const BinaryOperator *BO =
148 cast<BinaryOperator>(E->getCallee()->IgnoreParens());
149 const Expr *BaseExpr = BO->getLHS();
150 const Expr *MemFnExpr = BO->getRHS();
151
152 const MemberPointerType *MPT =
153 MemFnExpr->getType()->getAs<MemberPointerType>();
154 const FunctionProtoType *FPT =
155 MPT->getPointeeType()->getAs<FunctionProtoType>();
156 const CXXRecordDecl *RD =
157 cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
158
159 const llvm::FunctionType *FTy =
160 CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(RD, FPT),
161 FPT->isVariadic());
162
Anders Carlsson51591be2010-02-04 17:08:48 +0000163 const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(VMContext);
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000164
165 // Get the member function pointer.
Daniel Dunbar195337d2010-02-09 02:48:28 +0000166 llvm::Value *MemFnPtr = CreateMemTemp(MemFnExpr->getType(), "mem.fn");
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000167 EmitAggExpr(MemFnExpr, MemFnPtr, /*VolatileDest=*/false);
168
169 // Emit the 'this' pointer.
170 llvm::Value *This;
171
172 if (BO->getOpcode() == BinaryOperator::PtrMemI)
173 This = EmitScalarExpr(BaseExpr);
174 else
175 This = EmitLValue(BaseExpr).getAddress();
176
177 // Adjust it.
178 llvm::Value *Adj = Builder.CreateStructGEP(MemFnPtr, 1);
179 Adj = Builder.CreateLoad(Adj, "mem.fn.adj");
180
181 llvm::Value *Ptr = Builder.CreateBitCast(This, Int8PtrTy, "ptr");
182 Ptr = Builder.CreateGEP(Ptr, Adj, "adj");
183
184 This = Builder.CreateBitCast(Ptr, This->getType(), "this");
185
186 llvm::Value *FnPtr = Builder.CreateStructGEP(MemFnPtr, 0, "mem.fn.ptr");
187
188 const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
189
190 llvm::Value *FnAsInt = Builder.CreateLoad(FnPtr, "fn");
191
192 // If the LSB in the function pointer is 1, the function pointer points to
193 // a virtual function.
194 llvm::Value *IsVirtual
195 = Builder.CreateAnd(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1),
196 "and");
197
198 IsVirtual = Builder.CreateTrunc(IsVirtual,
199 llvm::Type::getInt1Ty(VMContext));
200
201 llvm::BasicBlock *FnVirtual = createBasicBlock("fn.virtual");
202 llvm::BasicBlock *FnNonVirtual = createBasicBlock("fn.nonvirtual");
203 llvm::BasicBlock *FnEnd = createBasicBlock("fn.end");
204
205 Builder.CreateCondBr(IsVirtual, FnVirtual, FnNonVirtual);
206 EmitBlock(FnVirtual);
207
Anders Carlsson046c2942010-04-17 20:15:18 +0000208 const llvm::Type *VTableTy =
Anders Carlsson51591be2010-02-04 17:08:48 +0000209 FTy->getPointerTo()->getPointerTo();
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000210
Anders Carlsson046c2942010-04-17 20:15:18 +0000211 llvm::Value *VTable = Builder.CreateBitCast(This, VTableTy->getPointerTo());
212 VTable = Builder.CreateLoad(VTable);
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000213
Anders Carlsson046c2942010-04-17 20:15:18 +0000214 VTable = Builder.CreateBitCast(VTable, Int8PtrTy);
215 llvm::Value *VTableOffset =
Anders Carlsson51591be2010-02-04 17:08:48 +0000216 Builder.CreateSub(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1));
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000217
Anders Carlsson046c2942010-04-17 20:15:18 +0000218 VTable = Builder.CreateGEP(VTable, VTableOffset, "fn");
219 VTable = Builder.CreateBitCast(VTable, VTableTy);
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000220
Anders Carlsson046c2942010-04-17 20:15:18 +0000221 llvm::Value *VirtualFn = Builder.CreateLoad(VTable, "virtualfn");
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000222
223 EmitBranch(FnEnd);
224 EmitBlock(FnNonVirtual);
225
226 // If the function is not virtual, just load the pointer.
227 llvm::Value *NonVirtualFn = Builder.CreateLoad(FnPtr, "fn");
228 NonVirtualFn = Builder.CreateIntToPtr(NonVirtualFn, FTy->getPointerTo());
229
230 EmitBlock(FnEnd);
231
232 llvm::PHINode *Callee = Builder.CreatePHI(FTy->getPointerTo());
233 Callee->reserveOperandSpace(2);
234 Callee->addIncoming(VirtualFn, FnVirtual);
235 Callee->addIncoming(NonVirtualFn, FnNonVirtual);
236
237 CallArgList Args;
238
239 QualType ThisType =
240 getContext().getPointerType(getContext().getTagDeclType(RD));
241
242 // Push the this ptr.
243 Args.push_back(std::make_pair(RValue::get(This), ThisType));
244
245 // And the rest of the call args
246 EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
John McCall04a67a62010-02-05 21:31:56 +0000247 const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>();
248 return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee,
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000249 ReturnValue, Args);
250}
251
252RValue
253CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
254 const CXXMethodDecl *MD,
255 ReturnValueSlot ReturnValue) {
256 assert(MD->isInstance() &&
257 "Trying to emit a member call expr on a static method!");
258
259 if (MD->isCopyAssignment()) {
260 const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext());
261 if (ClassDecl->hasTrivialCopyAssignment()) {
262 assert(!ClassDecl->hasUserDeclaredCopyAssignment() &&
263 "EmitCXXOperatorMemberCallExpr - user declared copy assignment");
Fariborz Jahanianb3ebe942010-05-10 22:57:35 +0000264 LValue LV = EmitLValue(E->getArg(0));
265 llvm::Value *This;
266 if (LV.isPropertyRef()) {
Fariborz Jahanianc9a8fa42010-05-16 00:10:46 +0000267 llvm::Value *AggLoc = CreateMemTemp(E->getArg(1)->getType());
Fariborz Jahanian0ca0b1f2010-05-15 23:05:52 +0000268 EmitAggExpr(E->getArg(1), AggLoc, false /*VolatileDest*/);
269 EmitObjCPropertySet(LV.getPropertyRefExpr(),
270 RValue::getAggregate(AggLoc, false /*VolatileDest*/));
271 return RValue::getAggregate(0, false);
Fariborz Jahanianb3ebe942010-05-10 22:57:35 +0000272 }
273 else
274 This = LV.getAddress();
275
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000276 llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
277 QualType Ty = E->getType();
278 EmitAggregateCopy(This, Src, Ty);
279 return RValue::get(This);
280 }
281 }
282
283 const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
284 const llvm::Type *Ty =
285 CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
286 FPT->isVariadic());
Fariborz Jahanianbbb52242010-05-07 18:56:13 +0000287 LValue LV = EmitLValue(E->getArg(0));
288 llvm::Value *This;
289 if (LV.isPropertyRef()) {
Fariborz Jahanianc9a8fa42010-05-16 00:10:46 +0000290 llvm::Value *AggLoc = CreateMemTemp(E->getArg(1)->getType());
Fariborz Jahanian0ca0b1f2010-05-15 23:05:52 +0000291 EmitAggExpr(E->getArg(1), AggLoc, false /*VolatileDest*/);
292 EmitObjCPropertySet(LV.getPropertyRefExpr(),
293 RValue::getAggregate(AggLoc, false /*VolatileDest*/));
294 return RValue::getAggregate(0, false);
Fariborz Jahanianbbb52242010-05-07 18:56:13 +0000295 }
296 else
297 This = LV.getAddress();
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000298
299 llvm::Value *Callee;
300 if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0)))
301 Callee = BuildVirtualCall(MD, This, Ty);
302 else
303 Callee = CGM.GetAddrOfFunction(MD, Ty);
304
Anders Carlssonc997d422010-01-02 01:01:18 +0000305 return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000306 E->arg_begin() + 1, E->arg_end());
307}
308
309void
310CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest,
311 const CXXConstructExpr *E) {
312 assert(Dest && "Must have a destination!");
313 const CXXConstructorDecl *CD = E->getConstructor();
314 const ConstantArrayType *Array =
315 getContext().getAsConstantArrayType(E->getType());
316 // For a copy constructor, even if it is trivial, must fall thru so
317 // its argument is code-gen'ed.
318 if (!CD->isCopyConstructor()) {
319 QualType InitType = E->getType();
320 if (Array)
321 InitType = getContext().getBaseElementType(Array);
322 const CXXRecordDecl *RD =
323 cast<CXXRecordDecl>(InitType->getAs<RecordType>()->getDecl());
324 if (RD->hasTrivialConstructor())
325 return;
326 }
327 // Code gen optimization to eliminate copy constructor and return
Douglas Gregor3c9034c2010-05-15 00:13:29 +0000328 // its first argument instead, if in fact that argument is a temporary
329 // object.
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000330 if (getContext().getLangOptions().ElideConstructors && E->isElidable()) {
Douglas Gregor3c9034c2010-05-15 00:13:29 +0000331 if (const Expr *Arg = E->getArg(0)->getTemporaryObject()) {
332 EmitAggExpr(Arg, Dest, false);
333 return;
334 }
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000335 }
336 if (Array) {
337 QualType BaseElementTy = getContext().getBaseElementType(Array);
338 const llvm::Type *BasePtr = ConvertType(BaseElementTy);
339 BasePtr = llvm::PointerType::getUnqual(BasePtr);
340 llvm::Value *BaseAddrPtr =
Anders Carlsson43db20e2010-05-01 17:02:18 +0000341 Builder.CreateBitCast(Dest, BasePtr);
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000342
343 EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr,
344 E->arg_begin(), E->arg_end());
345 }
Anders Carlsson155ed4a2010-05-02 23:20:53 +0000346 else {
347 CXXCtorType Type =
348 (E->getConstructionKind() == CXXConstructExpr::CK_Complete)
349 ? Ctor_Complete : Ctor_Base;
350 bool ForVirtualBase =
351 E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase;
352
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000353 // Call the constructor.
Anders Carlsson155ed4a2010-05-02 23:20:53 +0000354 EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest,
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000355 E->arg_begin(), E->arg_end());
Anders Carlsson155ed4a2010-05-02 23:20:53 +0000356 }
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000357}
358
Ken Dyckcaf647c2010-01-26 19:44:24 +0000359static CharUnits CalculateCookiePadding(ASTContext &Ctx, QualType ElementType) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000360 const RecordType *RT = ElementType->getAs<RecordType>();
Anders Carlssona4d4c012009-09-23 16:07:23 +0000361 if (!RT)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000362 return CharUnits::Zero();
Anders Carlssona4d4c012009-09-23 16:07:23 +0000363
364 const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl());
365 if (!RD)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000366 return CharUnits::Zero();
Anders Carlssona4d4c012009-09-23 16:07:23 +0000367
368 // Check if the class has a trivial destructor.
369 if (RD->hasTrivialDestructor()) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000370 // Check if the usual deallocation function takes two arguments.
Anders Carlsson710f7052009-12-13 20:10:12 +0000371 const CXXMethodDecl *UsualDeallocationFunction = 0;
372
Anders Carlsson871d0782009-12-13 20:04:38 +0000373 DeclarationName OpName =
374 Ctx.DeclarationNames.getCXXOperatorName(OO_Array_Delete);
Anders Carlsson871d0782009-12-13 20:04:38 +0000375 DeclContext::lookup_const_iterator Op, OpEnd;
376 for (llvm::tie(Op, OpEnd) = RD->lookup(OpName);
377 Op != OpEnd; ++Op) {
Anders Carlsson710f7052009-12-13 20:10:12 +0000378 const CXXMethodDecl *Delete = cast<CXXMethodDecl>(*Op);
Anders Carlsson871d0782009-12-13 20:04:38 +0000379
380 if (Delete->isUsualDeallocationFunction()) {
Anders Carlsson710f7052009-12-13 20:10:12 +0000381 UsualDeallocationFunction = Delete;
Anders Carlsson871d0782009-12-13 20:04:38 +0000382 break;
383 }
384 }
Anders Carlsson710f7052009-12-13 20:10:12 +0000385
386 // No usual deallocation function, we don't need a cookie.
387 if (!UsualDeallocationFunction)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000388 return CharUnits::Zero();
Anders Carlsson710f7052009-12-13 20:10:12 +0000389
390 // The usual deallocation function doesn't take a size_t argument, so we
391 // don't need a cookie.
392 if (UsualDeallocationFunction->getNumParams() == 1)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000393 return CharUnits::Zero();
Anders Carlsson710f7052009-12-13 20:10:12 +0000394
395 assert(UsualDeallocationFunction->getNumParams() == 2 &&
396 "Unexpected deallocation function type!");
397 }
Anders Carlssona4d4c012009-09-23 16:07:23 +0000398
Anders Carlsson871d0782009-12-13 20:04:38 +0000399 // Padding is the maximum of sizeof(size_t) and alignof(ElementType)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000400 return std::max(Ctx.getTypeSizeInChars(Ctx.getSizeType()),
401 Ctx.getTypeAlignInChars(ElementType));
Anders Carlsson871d0782009-12-13 20:04:38 +0000402}
403
Ken Dyckcaf647c2010-01-26 19:44:24 +0000404static CharUnits CalculateCookiePadding(ASTContext &Ctx, const CXXNewExpr *E) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000405 if (!E->isArray())
Ken Dyckcaf647c2010-01-26 19:44:24 +0000406 return CharUnits::Zero();
Anders Carlsson871d0782009-12-13 20:04:38 +0000407
Anders Carlssondd937552009-12-13 20:34:34 +0000408 // No cookie is required if the new operator being used is
409 // ::operator new[](size_t, void*).
410 const FunctionDecl *OperatorNew = E->getOperatorNew();
411 if (OperatorNew->getDeclContext()->getLookupContext()->isFileContext()) {
412 if (OperatorNew->getNumParams() == 2) {
413 CanQualType ParamType =
414 Ctx.getCanonicalType(OperatorNew->getParamDecl(1)->getType());
415
416 if (ParamType == Ctx.VoidPtrTy)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000417 return CharUnits::Zero();
Anders Carlssondd937552009-12-13 20:34:34 +0000418 }
419 }
420
Anders Carlsson871d0782009-12-13 20:04:38 +0000421 return CalculateCookiePadding(Ctx, E->getAllocatedType());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000422}
423
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000424static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context,
425 CodeGenFunction &CGF,
Anders Carlssona4d4c012009-09-23 16:07:23 +0000426 const CXXNewExpr *E,
427 llvm::Value *& NumElements) {
428 QualType Type = E->getAllocatedType();
Ken Dyckcaf647c2010-01-26 19:44:24 +0000429 CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(Type);
Anders Carlssona4d4c012009-09-23 16:07:23 +0000430 const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType());
431
432 if (!E->isArray())
Ken Dyckcaf647c2010-01-26 19:44:24 +0000433 return llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000434
Ken Dyckcaf647c2010-01-26 19:44:24 +0000435 CharUnits CookiePadding = CalculateCookiePadding(CGF.getContext(), E);
Anders Carlssona4d4c012009-09-23 16:07:23 +0000436
437 Expr::EvalResult Result;
438 if (E->getArraySize()->Evaluate(Result, CGF.getContext()) &&
439 !Result.HasSideEffects && Result.Val.isInt()) {
440
Ken Dyckcaf647c2010-01-26 19:44:24 +0000441 CharUnits AllocSize =
442 Result.Val.getInt().getZExtValue() * TypeSize + CookiePadding;
Anders Carlssona4d4c012009-09-23 16:07:23 +0000443
444 NumElements =
445 llvm::ConstantInt::get(SizeTy, Result.Val.getInt().getZExtValue());
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000446 while (const ArrayType *AType = Context.getAsArrayType(Type)) {
447 const llvm::ArrayType *llvmAType =
448 cast<llvm::ArrayType>(CGF.ConvertType(Type));
449 NumElements =
450 CGF.Builder.CreateMul(NumElements,
451 llvm::ConstantInt::get(
452 SizeTy, llvmAType->getNumElements()));
453 Type = AType->getElementType();
454 }
Anders Carlssona4d4c012009-09-23 16:07:23 +0000455
Ken Dyckcaf647c2010-01-26 19:44:24 +0000456 return llvm::ConstantInt::get(SizeTy, AllocSize.getQuantity());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000457 }
458
459 // Emit the array size expression.
460 NumElements = CGF.EmitScalarExpr(E->getArraySize());
461
462 // Multiply with the type size.
463 llvm::Value *V =
464 CGF.Builder.CreateMul(NumElements,
Ken Dyckcaf647c2010-01-26 19:44:24 +0000465 llvm::ConstantInt::get(SizeTy,
466 TypeSize.getQuantity()));
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000467
468 while (const ArrayType *AType = Context.getAsArrayType(Type)) {
469 const llvm::ArrayType *llvmAType =
470 cast<llvm::ArrayType>(CGF.ConvertType(Type));
471 NumElements =
472 CGF.Builder.CreateMul(NumElements,
473 llvm::ConstantInt::get(
474 SizeTy, llvmAType->getNumElements()));
475 Type = AType->getElementType();
476 }
Anders Carlssona4d4c012009-09-23 16:07:23 +0000477
478 // And add the cookie padding if necessary.
Ken Dyckcaf647c2010-01-26 19:44:24 +0000479 if (!CookiePadding.isZero())
480 V = CGF.Builder.CreateAdd(V,
481 llvm::ConstantInt::get(SizeTy, CookiePadding.getQuantity()));
Anders Carlssona4d4c012009-09-23 16:07:23 +0000482
483 return V;
484}
485
486static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
487 llvm::Value *NewPtr,
488 llvm::Value *NumElements) {
Anders Carlsson5d4d9462009-11-24 18:43:52 +0000489 if (E->isArray()) {
Anders Carlssone99bdb62010-05-03 15:09:17 +0000490 if (CXXConstructorDecl *Ctor = E->getConstructor()) {
491 if (!Ctor->getParent()->hasTrivialConstructor())
492 CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr,
493 E->constructor_arg_begin(),
494 E->constructor_arg_end());
495 return;
496 }
Anders Carlssona4d4c012009-09-23 16:07:23 +0000497 }
498
Anders Carlsson5d4d9462009-11-24 18:43:52 +0000499 QualType AllocType = E->getAllocatedType();
500
501 if (CXXConstructorDecl *Ctor = E->getConstructor()) {
Anders Carlsson155ed4a2010-05-02 23:20:53 +0000502 CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false,
503 NewPtr, E->constructor_arg_begin(),
Anders Carlsson5d4d9462009-11-24 18:43:52 +0000504 E->constructor_arg_end());
505
506 return;
507 }
508
509 // We have a POD type.
510 if (E->getNumConstructorArgs() == 0)
511 return;
512
513 assert(E->getNumConstructorArgs() == 1 &&
514 "Can only have one argument to initializer of POD type.");
515
516 const Expr *Init = E->getConstructorArg(0);
517
518 if (!CGF.hasAggregateLLVMType(AllocType))
519 CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr,
520 AllocType.isVolatileQualified(), AllocType);
521 else if (AllocType->isAnyComplexType())
522 CGF.EmitComplexExprIntoAddr(Init, NewPtr,
523 AllocType.isVolatileQualified());
524 else
525 CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000526}
527
Anders Carlsson16d81b82009-09-22 22:53:17 +0000528llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
Anders Carlsson16d81b82009-09-22 22:53:17 +0000529 QualType AllocType = E->getAllocatedType();
530 FunctionDecl *NewFD = E->getOperatorNew();
531 const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>();
532
533 CallArgList NewArgs;
534
535 // The allocation size is the first argument.
536 QualType SizeTy = getContext().getSizeType();
Anders Carlsson16d81b82009-09-22 22:53:17 +0000537
Anders Carlssona4d4c012009-09-23 16:07:23 +0000538 llvm::Value *NumElements = 0;
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000539 llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(),
540 *this, E, NumElements);
Anders Carlssona4d4c012009-09-23 16:07:23 +0000541
Anders Carlsson16d81b82009-09-22 22:53:17 +0000542 NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy));
543
544 // Emit the rest of the arguments.
545 // FIXME: Ideally, this should just use EmitCallArgs.
546 CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin();
547
548 // First, use the types from the function type.
549 // We start at 1 here because the first argument (the allocation size)
550 // has already been emitted.
551 for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) {
552 QualType ArgType = NewFTy->getArgType(i);
553
554 assert(getContext().getCanonicalType(ArgType.getNonReferenceType()).
555 getTypePtr() ==
556 getContext().getCanonicalType(NewArg->getType()).getTypePtr() &&
557 "type mismatch in call argument!");
558
559 NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
560 ArgType));
561
562 }
563
564 // Either we've emitted all the call args, or we have a call to a
565 // variadic function.
566 assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) &&
567 "Extra arguments in non-variadic function!");
568
569 // If we still have any arguments, emit them using the type of the argument.
570 for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end();
571 NewArg != NewArgEnd; ++NewArg) {
572 QualType ArgType = NewArg->getType();
573 NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
574 ArgType));
575 }
576
577 // Emit the call to new.
578 RValue RV =
John McCall04a67a62010-02-05 21:31:56 +0000579 EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy),
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000580 CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD);
Anders Carlsson16d81b82009-09-22 22:53:17 +0000581
582 // If an allocation function is declared with an empty exception specification
583 // it returns null to indicate failure to allocate storage. [expr.new]p13.
584 // (We don't need to check for null when there's no new initializer and
585 // we're allocating a POD type).
586 bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() &&
587 !(AllocType->isPODType() && !E->hasInitializer());
588
589 llvm::BasicBlock *NewNull = 0;
590 llvm::BasicBlock *NewNotNull = 0;
591 llvm::BasicBlock *NewEnd = 0;
592
593 llvm::Value *NewPtr = RV.getScalarVal();
594
595 if (NullCheckResult) {
596 NewNull = createBasicBlock("new.null");
597 NewNotNull = createBasicBlock("new.notnull");
598 NewEnd = createBasicBlock("new.end");
599
600 llvm::Value *IsNull =
601 Builder.CreateICmpEQ(NewPtr,
602 llvm::Constant::getNullValue(NewPtr->getType()),
603 "isnull");
604
605 Builder.CreateCondBr(IsNull, NewNull, NewNotNull);
606 EmitBlock(NewNotNull);
607 }
Ken Dyckcaf647c2010-01-26 19:44:24 +0000608
609 CharUnits CookiePadding = CalculateCookiePadding(getContext(), E);
610 if (!CookiePadding.isZero()) {
611 CharUnits CookieOffset =
612 CookiePadding - getContext().getTypeSizeInChars(SizeTy);
Anders Carlsson6ac5fc42009-09-23 18:59:48 +0000613
614 llvm::Value *NumElementsPtr =
Ken Dyckcaf647c2010-01-26 19:44:24 +0000615 Builder.CreateConstInBoundsGEP1_64(NewPtr, CookieOffset.getQuantity());
Anders Carlsson6ac5fc42009-09-23 18:59:48 +0000616
617 NumElementsPtr = Builder.CreateBitCast(NumElementsPtr,
618 ConvertType(SizeTy)->getPointerTo());
619 Builder.CreateStore(NumElements, NumElementsPtr);
620
621 // Now add the padding to the new ptr.
Ken Dyckcaf647c2010-01-26 19:44:24 +0000622 NewPtr = Builder.CreateConstInBoundsGEP1_64(NewPtr,
623 CookiePadding.getQuantity());
Anders Carlsson6ac5fc42009-09-23 18:59:48 +0000624 }
625
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000626 if (AllocType->isArrayType()) {
627 while (const ArrayType *AType = getContext().getAsArrayType(AllocType))
628 AllocType = AType->getElementType();
629 NewPtr =
630 Builder.CreateBitCast(NewPtr,
631 ConvertType(getContext().getPointerType(AllocType)));
632 EmitNewInitializer(*this, E, NewPtr, NumElements);
633 NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType()));
634 }
635 else {
636 NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType()));
637 EmitNewInitializer(*this, E, NewPtr, NumElements);
638 }
639
Anders Carlsson16d81b82009-09-22 22:53:17 +0000640 if (NullCheckResult) {
641 Builder.CreateBr(NewEnd);
Eli Friedman7f1de452009-11-10 22:39:09 +0000642 NewNotNull = Builder.GetInsertBlock();
Anders Carlsson16d81b82009-09-22 22:53:17 +0000643 EmitBlock(NewNull);
644 Builder.CreateBr(NewEnd);
645 EmitBlock(NewEnd);
646
647 llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType());
648 PHI->reserveOperandSpace(2);
649 PHI->addIncoming(NewPtr, NewNotNull);
650 PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), NewNull);
651
652 NewPtr = PHI;
653 }
654
655 return NewPtr;
656}
657
Anders Carlsson871d0782009-12-13 20:04:38 +0000658static std::pair<llvm::Value *, llvm::Value *>
659GetAllocatedObjectPtrAndNumElements(CodeGenFunction &CGF,
660 llvm::Value *Ptr, QualType DeleteTy) {
661 QualType SizeTy = CGF.getContext().getSizeType();
662 const llvm::Type *SizeLTy = CGF.ConvertType(SizeTy);
663
Ken Dyckcaf647c2010-01-26 19:44:24 +0000664 CharUnits DeleteTypeAlign = CGF.getContext().getTypeAlignInChars(DeleteTy);
665 CharUnits CookiePadding =
666 std::max(CGF.getContext().getTypeSizeInChars(SizeTy),
667 DeleteTypeAlign);
668 assert(!CookiePadding.isZero() && "CookiePadding should not be 0.");
Anders Carlsson871d0782009-12-13 20:04:38 +0000669
670 const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext());
Ken Dyckcaf647c2010-01-26 19:44:24 +0000671 CharUnits CookieOffset =
672 CookiePadding - CGF.getContext().getTypeSizeInChars(SizeTy);
Anders Carlsson871d0782009-12-13 20:04:38 +0000673
674 llvm::Value *AllocatedObjectPtr = CGF.Builder.CreateBitCast(Ptr, Int8PtrTy);
675 AllocatedObjectPtr =
676 CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr,
Ken Dyckcaf647c2010-01-26 19:44:24 +0000677 -CookiePadding.getQuantity());
Anders Carlsson871d0782009-12-13 20:04:38 +0000678
679 llvm::Value *NumElementsPtr =
680 CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr,
Ken Dyckcaf647c2010-01-26 19:44:24 +0000681 CookieOffset.getQuantity());
Anders Carlsson871d0782009-12-13 20:04:38 +0000682 NumElementsPtr =
683 CGF.Builder.CreateBitCast(NumElementsPtr, SizeLTy->getPointerTo());
684
685 llvm::Value *NumElements = CGF.Builder.CreateLoad(NumElementsPtr);
686 NumElements =
687 CGF.Builder.CreateIntCast(NumElements, SizeLTy, /*isSigned=*/false);
688
689 return std::make_pair(AllocatedObjectPtr, NumElements);
690}
691
Eli Friedman5fe05982009-11-18 00:50:08 +0000692void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
693 llvm::Value *Ptr,
694 QualType DeleteTy) {
695 const FunctionProtoType *DeleteFTy =
696 DeleteFD->getType()->getAs<FunctionProtoType>();
697
698 CallArgList DeleteArgs;
699
Anders Carlsson871d0782009-12-13 20:04:38 +0000700 // Check if we need to pass the size to the delete operator.
701 llvm::Value *Size = 0;
702 QualType SizeTy;
703 if (DeleteFTy->getNumArgs() == 2) {
704 SizeTy = DeleteFTy->getArgType(1);
Ken Dyck4f122ef2010-01-26 19:59:28 +0000705 CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
706 Size = llvm::ConstantInt::get(ConvertType(SizeTy),
707 DeleteTypeSize.getQuantity());
Anders Carlsson871d0782009-12-13 20:04:38 +0000708 }
709
710 if (DeleteFD->getOverloadedOperator() == OO_Array_Delete &&
Ken Dyckcaf647c2010-01-26 19:44:24 +0000711 !CalculateCookiePadding(getContext(), DeleteTy).isZero()) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000712 // We need to get the number of elements in the array from the cookie.
713 llvm::Value *AllocatedObjectPtr;
714 llvm::Value *NumElements;
715 llvm::tie(AllocatedObjectPtr, NumElements) =
716 GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy);
717
718 // Multiply the size with the number of elements.
719 if (Size)
720 Size = Builder.CreateMul(NumElements, Size);
721
722 Ptr = AllocatedObjectPtr;
723 }
724
Eli Friedman5fe05982009-11-18 00:50:08 +0000725 QualType ArgTy = DeleteFTy->getArgType(0);
726 llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
727 DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy));
728
Anders Carlsson871d0782009-12-13 20:04:38 +0000729 if (Size)
Eli Friedman5fe05982009-11-18 00:50:08 +0000730 DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy));
Eli Friedman5fe05982009-11-18 00:50:08 +0000731
732 // Emit the call to delete.
John McCall04a67a62010-02-05 21:31:56 +0000733 EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy),
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000734 CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(),
Eli Friedman5fe05982009-11-18 00:50:08 +0000735 DeleteArgs, DeleteFD);
736}
737
Anders Carlsson16d81b82009-09-22 22:53:17 +0000738void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
Fariborz Jahanian72c21532009-11-13 19:27:47 +0000739
Douglas Gregor90916562009-09-29 18:16:17 +0000740 // Get at the argument before we performed the implicit conversion
741 // to void*.
742 const Expr *Arg = E->getArgument();
743 while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) {
744 if (ICE->getCastKind() != CastExpr::CK_UserDefinedConversion &&
745 ICE->getType()->isVoidPointerType())
746 Arg = ICE->getSubExpr();
Douglas Gregord69dd782009-10-01 05:49:51 +0000747 else
748 break;
Douglas Gregor90916562009-09-29 18:16:17 +0000749 }
750
751 QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
Anders Carlsson16d81b82009-09-22 22:53:17 +0000752
Douglas Gregor90916562009-09-29 18:16:17 +0000753 llvm::Value *Ptr = EmitScalarExpr(Arg);
Anders Carlsson16d81b82009-09-22 22:53:17 +0000754
755 // Null check the pointer.
756 llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
757 llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
758
759 llvm::Value *IsNull =
760 Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()),
761 "isnull");
762
763 Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
764 EmitBlock(DeleteNotNull);
Anders Carlsson566abee2009-11-13 04:45:41 +0000765
766 bool ShouldCallDelete = true;
767
Anders Carlsson16d81b82009-09-22 22:53:17 +0000768 // Call the destructor if necessary.
769 if (const RecordType *RT = DeleteTy->getAs<RecordType>()) {
770 if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl())) {
771 if (!RD->hasTrivialDestructor()) {
772 const CXXDestructorDecl *Dtor = RD->getDestructor(getContext());
Fariborz Jahanian72c21532009-11-13 19:27:47 +0000773 if (E->isArrayForm()) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000774 llvm::Value *AllocatedObjectPtr;
775 llvm::Value *NumElements;
776 llvm::tie(AllocatedObjectPtr, NumElements) =
777 GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy);
778
Anders Carlsson61537102009-12-13 18:48:07 +0000779 EmitCXXAggrDestructorCall(Dtor, NumElements, Ptr);
Anders Carlsson61537102009-12-13 18:48:07 +0000780 } else if (Dtor->isVirtual()) {
Anders Carlsson16d81b82009-09-22 22:53:17 +0000781 const llvm::Type *Ty =
782 CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(Dtor),
783 /*isVariadic=*/false);
784
Anders Carlsson566abee2009-11-13 04:45:41 +0000785 llvm::Value *Callee = BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty);
Anders Carlssonc997d422010-01-02 01:01:18 +0000786 EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0,
787 0, 0);
Anders Carlsson566abee2009-11-13 04:45:41 +0000788
789 // The dtor took care of deleting the object.
790 ShouldCallDelete = false;
Anders Carlsson16d81b82009-09-22 22:53:17 +0000791 } else
Anders Carlsson8e6404c2010-05-02 23:29:11 +0000792 EmitCXXDestructorCall(Dtor, Dtor_Complete, /*ForVirtualBase=*/false,
793 Ptr);
Anders Carlsson16d81b82009-09-22 22:53:17 +0000794 }
795 }
796 }
797
Eli Friedman5fe05982009-11-18 00:50:08 +0000798 if (ShouldCallDelete)
799 EmitDeleteCall(E->getOperatorDelete(), Ptr, DeleteTy);
Anders Carlsson16d81b82009-09-22 22:53:17 +0000800
Anders Carlsson16d81b82009-09-22 22:53:17 +0000801 EmitBlock(DeleteEnd);
802}
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000803
804llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
805 QualType Ty = E->getType();
806 const llvm::Type *LTy = ConvertType(Ty)->getPointerTo();
Anders Carlsson31b7f522009-12-11 02:46:30 +0000807
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000808 if (E->isTypeOperand()) {
809 llvm::Constant *TypeInfo =
810 CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand());
811 return Builder.CreateBitCast(TypeInfo, LTy);
812 }
813
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000814 Expr *subE = E->getExprOperand();
Mike Stump5fae8562009-11-17 22:33:00 +0000815 Ty = subE->getType();
816 CanQualType CanTy = CGM.getContext().getCanonicalType(Ty);
817 Ty = CanTy.getUnqualifiedType().getNonReferenceType();
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000818 if (const RecordType *RT = Ty->getAs<RecordType>()) {
819 const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
820 if (RD->isPolymorphic()) {
821 // FIXME: if subE is an lvalue do
822 LValue Obj = EmitLValue(subE);
823 llvm::Value *This = Obj.getAddress();
Mike Stumpf549e892009-11-15 16:52:53 +0000824 LTy = LTy->getPointerTo()->getPointerTo();
825 llvm::Value *V = Builder.CreateBitCast(This, LTy);
826 // We need to do a zero check for *p, unless it has NonNullAttr.
827 // FIXME: PointerType->hasAttr<NonNullAttr>()
828 bool CanBeZero = false;
Mike Stumpdb519a42009-11-17 00:45:21 +0000829 if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens()))
Mike Stumpf549e892009-11-15 16:52:53 +0000830 if (UO->getOpcode() == UnaryOperator::Deref)
831 CanBeZero = true;
832 if (CanBeZero) {
833 llvm::BasicBlock *NonZeroBlock = createBasicBlock();
834 llvm::BasicBlock *ZeroBlock = createBasicBlock();
835
836 llvm::Value *Zero = llvm::Constant::getNullValue(LTy);
837 Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero),
838 NonZeroBlock, ZeroBlock);
839 EmitBlock(ZeroBlock);
840 /// Call __cxa_bad_typeid
841 const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext);
842 const llvm::FunctionType *FTy;
843 FTy = llvm::FunctionType::get(ResultType, false);
844 llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid");
Mike Stumpc849c052009-11-16 06:50:58 +0000845 Builder.CreateCall(F)->setDoesNotReturn();
Mike Stumpf549e892009-11-15 16:52:53 +0000846 Builder.CreateUnreachable();
847 EmitBlock(NonZeroBlock);
848 }
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000849 V = Builder.CreateLoad(V, "vtable");
850 V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL);
851 V = Builder.CreateLoad(V);
852 return V;
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000853 }
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000854 }
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000855 return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy);
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000856}
Mike Stumpc849c052009-11-16 06:50:58 +0000857
858llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V,
859 const CXXDynamicCastExpr *DCE) {
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000860 QualType SrcTy = DCE->getSubExpr()->getType();
861 QualType DestTy = DCE->getTypeAsWritten();
862 QualType InnerType = DestTy->getPointeeType();
863
Mike Stumpc849c052009-11-16 06:50:58 +0000864 const llvm::Type *LTy = ConvertType(DCE->getType());
Mike Stump2b35baf2009-11-16 22:52:20 +0000865
Mike Stumpc849c052009-11-16 06:50:58 +0000866 bool CanBeZero = false;
Mike Stumpc849c052009-11-16 06:50:58 +0000867 bool ToVoid = false;
Mike Stump2b35baf2009-11-16 22:52:20 +0000868 bool ThrowOnBad = false;
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000869 if (DestTy->isPointerType()) {
Mike Stumpc849c052009-11-16 06:50:58 +0000870 // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this
871 CanBeZero = true;
872 if (InnerType->isVoidType())
873 ToVoid = true;
874 } else {
875 LTy = LTy->getPointerTo();
Douglas Gregor485ee322010-05-14 21:14:41 +0000876
877 // FIXME: What if exceptions are disabled?
Mike Stumpc849c052009-11-16 06:50:58 +0000878 ThrowOnBad = true;
879 }
880
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000881 if (SrcTy->isPointerType() || SrcTy->isReferenceType())
882 SrcTy = SrcTy->getPointeeType();
883 SrcTy = SrcTy.getUnqualifiedType();
884
Anders Carlsson6f0e4852009-12-18 14:55:04 +0000885 if (DestTy->isPointerType() || DestTy->isReferenceType())
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000886 DestTy = DestTy->getPointeeType();
887 DestTy = DestTy.getUnqualifiedType();
Mike Stumpc849c052009-11-16 06:50:58 +0000888
Mike Stumpc849c052009-11-16 06:50:58 +0000889 llvm::BasicBlock *ContBlock = createBasicBlock();
890 llvm::BasicBlock *NullBlock = 0;
891 llvm::BasicBlock *NonZeroBlock = 0;
892 if (CanBeZero) {
893 NonZeroBlock = createBasicBlock();
894 NullBlock = createBasicBlock();
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000895 Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock);
Mike Stumpc849c052009-11-16 06:50:58 +0000896 EmitBlock(NonZeroBlock);
897 }
898
Mike Stumpc849c052009-11-16 06:50:58 +0000899 llvm::BasicBlock *BadCastBlock = 0;
Mike Stumpc849c052009-11-16 06:50:58 +0000900
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000901 const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
Mike Stump2b35baf2009-11-16 22:52:20 +0000902
903 // See if this is a dynamic_cast(void*)
904 if (ToVoid) {
905 llvm::Value *This = V;
906 V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo());
907 V = Builder.CreateLoad(V, "vtable");
908 V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL);
909 V = Builder.CreateLoad(V, "offset to top");
910 This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext));
911 V = Builder.CreateInBoundsGEP(This, V);
912 V = Builder.CreateBitCast(V, LTy);
913 } else {
914 /// Call __dynamic_cast
915 const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext);
916 const llvm::FunctionType *FTy;
917 std::vector<const llvm::Type*> ArgTys;
918 const llvm::Type *PtrToInt8Ty
919 = llvm::Type::getInt8Ty(VMContext)->getPointerTo();
920 ArgTys.push_back(PtrToInt8Ty);
921 ArgTys.push_back(PtrToInt8Ty);
922 ArgTys.push_back(PtrToInt8Ty);
923 ArgTys.push_back(PtrDiffTy);
924 FTy = llvm::FunctionType::get(ResultType, ArgTys, false);
Mike Stump2b35baf2009-11-16 22:52:20 +0000925
926 // FIXME: Calculate better hint.
927 llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL);
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000928
929 assert(SrcTy->isRecordType() && "Src type must be record type!");
930 assert(DestTy->isRecordType() && "Dest type must be record type!");
931
Douglas Gregor154fe982009-12-23 22:04:40 +0000932 llvm::Value *SrcArg
933 = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType());
934 llvm::Value *DestArg
935 = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType());
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000936
Mike Stump2b35baf2009-11-16 22:52:20 +0000937 V = Builder.CreateBitCast(V, PtrToInt8Ty);
938 V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"),
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000939 V, SrcArg, DestArg, hint);
Mike Stump2b35baf2009-11-16 22:52:20 +0000940 V = Builder.CreateBitCast(V, LTy);
941
942 if (ThrowOnBad) {
943 BadCastBlock = createBasicBlock();
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000944 Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock);
Mike Stump2b35baf2009-11-16 22:52:20 +0000945 EmitBlock(BadCastBlock);
Douglas Gregor485ee322010-05-14 21:14:41 +0000946 /// Invoke __cxa_bad_cast
Mike Stump2b35baf2009-11-16 22:52:20 +0000947 ResultType = llvm::Type::getVoidTy(VMContext);
948 const llvm::FunctionType *FBadTy;
Mike Stumpfde17be2009-11-17 03:01:03 +0000949 FBadTy = llvm::FunctionType::get(ResultType, false);
Mike Stump2b35baf2009-11-16 22:52:20 +0000950 llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast");
Douglas Gregor485ee322010-05-14 21:14:41 +0000951 if (llvm::BasicBlock *InvokeDest = getInvokeDest()) {
952 llvm::BasicBlock *Cont = createBasicBlock("invoke.cont");
953 Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn();
954 EmitBlock(Cont);
955 } else {
956 // FIXME: Does this ever make sense?
957 Builder.CreateCall(F)->setDoesNotReturn();
958 }
Mike Stump8b152b82009-11-17 00:08:50 +0000959 Builder.CreateUnreachable();
Mike Stump2b35baf2009-11-16 22:52:20 +0000960 }
Mike Stumpc849c052009-11-16 06:50:58 +0000961 }
962
963 if (CanBeZero) {
964 Builder.CreateBr(ContBlock);
965 EmitBlock(NullBlock);
966 Builder.CreateBr(ContBlock);
967 }
968 EmitBlock(ContBlock);
969 if (CanBeZero) {
970 llvm::PHINode *PHI = Builder.CreatePHI(LTy);
Mike Stump14431c12009-11-17 00:10:05 +0000971 PHI->reserveOperandSpace(2);
Mike Stumpc849c052009-11-16 06:50:58 +0000972 PHI->addIncoming(V, NonZeroBlock);
973 PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock);
Mike Stumpc849c052009-11-16 06:50:58 +0000974 V = PHI;
975 }
976
977 return V;
978}