blob: 28b1d3b501193e7d7437e22cb43b45404faf441a [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,
47 FPT->getCallConv(),
48 FPT->getNoReturnAttr()), Callee,
Anders Carlsson3b5ad222010-01-01 20:29:01 +000049 ReturnValue, Args, MD);
50}
51
52/// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given
53/// expr can be devirtualized.
54static bool canDevirtualizeMemberFunctionCalls(const Expr *Base) {
55 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) {
56 if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
57 // This is a record decl. We know the type and can devirtualize it.
58 return VD->getType()->isRecordType();
59 }
60
61 return false;
62 }
63
64 // We can always devirtualize calls on temporary object expressions.
Eli Friedman6997aae2010-01-31 20:58:15 +000065 if (isa<CXXConstructExpr>(Base))
Anders Carlsson3b5ad222010-01-01 20:29:01 +000066 return true;
67
68 // And calls on bound temporaries.
69 if (isa<CXXBindTemporaryExpr>(Base))
70 return true;
71
72 // Check if this is a call expr that returns a record type.
73 if (const CallExpr *CE = dyn_cast<CallExpr>(Base))
74 return CE->getCallReturnType()->isRecordType();
75
76 // We can't devirtualize the call.
77 return false;
78}
79
80RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
81 ReturnValueSlot ReturnValue) {
82 if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens()))
83 return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
84
85 const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens());
86 const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
87
88 if (MD->isStatic()) {
89 // The method is static, emit it as we would a regular call.
90 llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
91 return EmitCall(getContext().getPointerType(MD->getType()), Callee,
92 ReturnValue, CE->arg_begin(), CE->arg_end());
93 }
94
95 const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
96
97 const llvm::Type *Ty =
98 CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
99 FPT->isVariadic());
100 llvm::Value *This;
101
102 if (ME->isArrow())
103 This = EmitScalarExpr(ME->getBase());
104 else {
105 LValue BaseLV = EmitLValue(ME->getBase());
106 This = BaseLV.getAddress();
107 }
108
109 if (MD->isCopyAssignment() && MD->isTrivial()) {
110 // We don't like to generate the trivial copy assignment operator when
111 // it isn't necessary; just produce the proper effect here.
112 llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
113 EmitAggregateCopy(This, RHS, CE->getType());
114 return RValue::get(This);
115 }
116
117 // C++ [class.virtual]p12:
118 // Explicit qualification with the scope operator (5.1) suppresses the
119 // virtual call mechanism.
120 //
121 // We also don't emit a virtual call if the base expression has a record type
122 // because then we know what the type is.
123 llvm::Value *Callee;
124 if (const CXXDestructorDecl *Destructor
125 = dyn_cast<CXXDestructorDecl>(MD)) {
126 if (Destructor->isTrivial())
127 return RValue::get(0);
128 if (MD->isVirtual() && !ME->hasQualifier() &&
129 !canDevirtualizeMemberFunctionCalls(ME->getBase())) {
130 Callee = BuildVirtualCall(Destructor, Dtor_Complete, This, Ty);
131 } else {
132 Callee = CGM.GetAddrOfFunction(GlobalDecl(Destructor, Dtor_Complete), Ty);
133 }
134 } else if (MD->isVirtual() && !ME->hasQualifier() &&
135 !canDevirtualizeMemberFunctionCalls(ME->getBase())) {
136 Callee = BuildVirtualCall(MD, This, Ty);
137 } else {
138 Callee = CGM.GetAddrOfFunction(MD, Ty);
139 }
140
Anders Carlssonc997d422010-01-02 01:01:18 +0000141 return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000142 CE->arg_begin(), CE->arg_end());
143}
144
145RValue
146CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
147 ReturnValueSlot ReturnValue) {
148 const BinaryOperator *BO =
149 cast<BinaryOperator>(E->getCallee()->IgnoreParens());
150 const Expr *BaseExpr = BO->getLHS();
151 const Expr *MemFnExpr = BO->getRHS();
152
153 const MemberPointerType *MPT =
154 MemFnExpr->getType()->getAs<MemberPointerType>();
155 const FunctionProtoType *FPT =
156 MPT->getPointeeType()->getAs<FunctionProtoType>();
157 const CXXRecordDecl *RD =
158 cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
159
160 const llvm::FunctionType *FTy =
161 CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(RD, FPT),
162 FPT->isVariadic());
163
Anders Carlsson51591be2010-02-04 17:08:48 +0000164 const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(VMContext);
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000165
166 // Get the member function pointer.
Daniel Dunbar195337d2010-02-09 02:48:28 +0000167 llvm::Value *MemFnPtr = CreateMemTemp(MemFnExpr->getType(), "mem.fn");
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000168 EmitAggExpr(MemFnExpr, MemFnPtr, /*VolatileDest=*/false);
169
170 // Emit the 'this' pointer.
171 llvm::Value *This;
172
173 if (BO->getOpcode() == BinaryOperator::PtrMemI)
174 This = EmitScalarExpr(BaseExpr);
175 else
176 This = EmitLValue(BaseExpr).getAddress();
177
178 // Adjust it.
179 llvm::Value *Adj = Builder.CreateStructGEP(MemFnPtr, 1);
180 Adj = Builder.CreateLoad(Adj, "mem.fn.adj");
181
182 llvm::Value *Ptr = Builder.CreateBitCast(This, Int8PtrTy, "ptr");
183 Ptr = Builder.CreateGEP(Ptr, Adj, "adj");
184
185 This = Builder.CreateBitCast(Ptr, This->getType(), "this");
186
187 llvm::Value *FnPtr = Builder.CreateStructGEP(MemFnPtr, 0, "mem.fn.ptr");
188
189 const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
190
191 llvm::Value *FnAsInt = Builder.CreateLoad(FnPtr, "fn");
192
193 // If the LSB in the function pointer is 1, the function pointer points to
194 // a virtual function.
195 llvm::Value *IsVirtual
196 = Builder.CreateAnd(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1),
197 "and");
198
199 IsVirtual = Builder.CreateTrunc(IsVirtual,
200 llvm::Type::getInt1Ty(VMContext));
201
202 llvm::BasicBlock *FnVirtual = createBasicBlock("fn.virtual");
203 llvm::BasicBlock *FnNonVirtual = createBasicBlock("fn.nonvirtual");
204 llvm::BasicBlock *FnEnd = createBasicBlock("fn.end");
205
206 Builder.CreateCondBr(IsVirtual, FnVirtual, FnNonVirtual);
207 EmitBlock(FnVirtual);
208
Anders Carlsson51591be2010-02-04 17:08:48 +0000209 const llvm::Type *VtableTy =
210 FTy->getPointerTo()->getPointerTo();
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000211
Anders Carlsson51591be2010-02-04 17:08:48 +0000212 llvm::Value *Vtable = Builder.CreateBitCast(This, VtableTy->getPointerTo());
213 Vtable = Builder.CreateLoad(Vtable);
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000214
Anders Carlsson51591be2010-02-04 17:08:48 +0000215 Vtable = Builder.CreateBitCast(Vtable, Int8PtrTy);
216 llvm::Value *VtableOffset =
217 Builder.CreateSub(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1));
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000218
Anders Carlsson51591be2010-02-04 17:08:48 +0000219 Vtable = Builder.CreateGEP(Vtable, VtableOffset, "fn");
220 Vtable = Builder.CreateBitCast(Vtable, VtableTy);
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000221
Anders Carlsson51591be2010-02-04 17:08:48 +0000222 llvm::Value *VirtualFn = Builder.CreateLoad(Vtable, "virtualfn");
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000223
224 EmitBranch(FnEnd);
225 EmitBlock(FnNonVirtual);
226
227 // If the function is not virtual, just load the pointer.
228 llvm::Value *NonVirtualFn = Builder.CreateLoad(FnPtr, "fn");
229 NonVirtualFn = Builder.CreateIntToPtr(NonVirtualFn, FTy->getPointerTo());
230
231 EmitBlock(FnEnd);
232
233 llvm::PHINode *Callee = Builder.CreatePHI(FTy->getPointerTo());
234 Callee->reserveOperandSpace(2);
235 Callee->addIncoming(VirtualFn, FnVirtual);
236 Callee->addIncoming(NonVirtualFn, FnNonVirtual);
237
238 CallArgList Args;
239
240 QualType ThisType =
241 getContext().getPointerType(getContext().getTagDeclType(RD));
242
243 // Push the this ptr.
244 Args.push_back(std::make_pair(RValue::get(This), ThisType));
245
246 // And the rest of the call args
247 EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
John McCall04a67a62010-02-05 21:31:56 +0000248 const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>();
249 return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee,
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000250 ReturnValue, Args);
251}
252
253RValue
254CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
255 const CXXMethodDecl *MD,
256 ReturnValueSlot ReturnValue) {
257 assert(MD->isInstance() &&
258 "Trying to emit a member call expr on a static method!");
259
260 if (MD->isCopyAssignment()) {
261 const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext());
262 if (ClassDecl->hasTrivialCopyAssignment()) {
263 assert(!ClassDecl->hasUserDeclaredCopyAssignment() &&
264 "EmitCXXOperatorMemberCallExpr - user declared copy assignment");
265 llvm::Value *This = EmitLValue(E->getArg(0)).getAddress();
266 llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
267 QualType Ty = E->getType();
268 EmitAggregateCopy(This, Src, Ty);
269 return RValue::get(This);
270 }
271 }
272
273 const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
274 const llvm::Type *Ty =
275 CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
276 FPT->isVariadic());
277
278 llvm::Value *This = EmitLValue(E->getArg(0)).getAddress();
279
280 llvm::Value *Callee;
281 if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0)))
282 Callee = BuildVirtualCall(MD, This, Ty);
283 else
284 Callee = CGM.GetAddrOfFunction(MD, Ty);
285
Anders Carlssonc997d422010-01-02 01:01:18 +0000286 return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000287 E->arg_begin() + 1, E->arg_end());
288}
289
290void
291CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest,
292 const CXXConstructExpr *E) {
293 assert(Dest && "Must have a destination!");
294 const CXXConstructorDecl *CD = E->getConstructor();
295 const ConstantArrayType *Array =
296 getContext().getAsConstantArrayType(E->getType());
297 // For a copy constructor, even if it is trivial, must fall thru so
298 // its argument is code-gen'ed.
299 if (!CD->isCopyConstructor()) {
300 QualType InitType = E->getType();
301 if (Array)
302 InitType = getContext().getBaseElementType(Array);
303 const CXXRecordDecl *RD =
304 cast<CXXRecordDecl>(InitType->getAs<RecordType>()->getDecl());
305 if (RD->hasTrivialConstructor())
306 return;
307 }
308 // Code gen optimization to eliminate copy constructor and return
309 // its first argument instead.
310 if (getContext().getLangOptions().ElideConstructors && E->isElidable()) {
311 const Expr *Arg = E->getArg(0);
312
Douglas Gregor321725d2010-02-03 03:01:57 +0000313 if (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) {
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000314 assert((ICE->getCastKind() == CastExpr::CK_NoOp ||
315 ICE->getCastKind() == CastExpr::CK_ConstructorConversion ||
316 ICE->getCastKind() == CastExpr::CK_UserDefinedConversion) &&
317 "Unknown implicit cast kind in constructor elision");
318 Arg = ICE->getSubExpr();
319 }
320
321 if (const CXXFunctionalCastExpr *FCE = dyn_cast<CXXFunctionalCastExpr>(Arg))
322 Arg = FCE->getSubExpr();
323
324 if (const CXXBindTemporaryExpr *BindExpr =
325 dyn_cast<CXXBindTemporaryExpr>(Arg))
326 Arg = BindExpr->getSubExpr();
327
328 EmitAggExpr(Arg, Dest, false);
329 return;
330 }
331 if (Array) {
332 QualType BaseElementTy = getContext().getBaseElementType(Array);
333 const llvm::Type *BasePtr = ConvertType(BaseElementTy);
334 BasePtr = llvm::PointerType::getUnqual(BasePtr);
335 llvm::Value *BaseAddrPtr =
336 Builder.CreateBitCast(Dest, BasePtr);
337
338 EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr,
339 E->arg_begin(), E->arg_end());
340 }
341 else
342 // Call the constructor.
Douglas Gregor9db7dbb2010-01-31 09:12:51 +0000343 EmitCXXConstructorCall(CD,
344 E->isBaseInitialization()? Ctor_Base : Ctor_Complete,
345 Dest,
Anders Carlsson3b5ad222010-01-01 20:29:01 +0000346 E->arg_begin(), E->arg_end());
347}
348
Ken Dyckcaf647c2010-01-26 19:44:24 +0000349static CharUnits CalculateCookiePadding(ASTContext &Ctx, QualType ElementType) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000350 const RecordType *RT = ElementType->getAs<RecordType>();
Anders Carlssona4d4c012009-09-23 16:07:23 +0000351 if (!RT)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000352 return CharUnits::Zero();
Anders Carlssona4d4c012009-09-23 16:07:23 +0000353
354 const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl());
355 if (!RD)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000356 return CharUnits::Zero();
Anders Carlssona4d4c012009-09-23 16:07:23 +0000357
358 // Check if the class has a trivial destructor.
359 if (RD->hasTrivialDestructor()) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000360 // Check if the usual deallocation function takes two arguments.
Anders Carlsson710f7052009-12-13 20:10:12 +0000361 const CXXMethodDecl *UsualDeallocationFunction = 0;
362
Anders Carlsson871d0782009-12-13 20:04:38 +0000363 DeclarationName OpName =
364 Ctx.DeclarationNames.getCXXOperatorName(OO_Array_Delete);
Anders Carlsson871d0782009-12-13 20:04:38 +0000365 DeclContext::lookup_const_iterator Op, OpEnd;
366 for (llvm::tie(Op, OpEnd) = RD->lookup(OpName);
367 Op != OpEnd; ++Op) {
Anders Carlsson710f7052009-12-13 20:10:12 +0000368 const CXXMethodDecl *Delete = cast<CXXMethodDecl>(*Op);
Anders Carlsson871d0782009-12-13 20:04:38 +0000369
370 if (Delete->isUsualDeallocationFunction()) {
Anders Carlsson710f7052009-12-13 20:10:12 +0000371 UsualDeallocationFunction = Delete;
Anders Carlsson871d0782009-12-13 20:04:38 +0000372 break;
373 }
374 }
Anders Carlsson710f7052009-12-13 20:10:12 +0000375
376 // No usual deallocation function, we don't need a cookie.
377 if (!UsualDeallocationFunction)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000378 return CharUnits::Zero();
Anders Carlsson710f7052009-12-13 20:10:12 +0000379
380 // The usual deallocation function doesn't take a size_t argument, so we
381 // don't need a cookie.
382 if (UsualDeallocationFunction->getNumParams() == 1)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000383 return CharUnits::Zero();
Anders Carlsson710f7052009-12-13 20:10:12 +0000384
385 assert(UsualDeallocationFunction->getNumParams() == 2 &&
386 "Unexpected deallocation function type!");
387 }
Anders Carlssona4d4c012009-09-23 16:07:23 +0000388
Anders Carlsson871d0782009-12-13 20:04:38 +0000389 // Padding is the maximum of sizeof(size_t) and alignof(ElementType)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000390 return std::max(Ctx.getTypeSizeInChars(Ctx.getSizeType()),
391 Ctx.getTypeAlignInChars(ElementType));
Anders Carlsson871d0782009-12-13 20:04:38 +0000392}
393
Ken Dyckcaf647c2010-01-26 19:44:24 +0000394static CharUnits CalculateCookiePadding(ASTContext &Ctx, const CXXNewExpr *E) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000395 if (!E->isArray())
Ken Dyckcaf647c2010-01-26 19:44:24 +0000396 return CharUnits::Zero();
Anders Carlsson871d0782009-12-13 20:04:38 +0000397
Anders Carlssondd937552009-12-13 20:34:34 +0000398 // No cookie is required if the new operator being used is
399 // ::operator new[](size_t, void*).
400 const FunctionDecl *OperatorNew = E->getOperatorNew();
401 if (OperatorNew->getDeclContext()->getLookupContext()->isFileContext()) {
402 if (OperatorNew->getNumParams() == 2) {
403 CanQualType ParamType =
404 Ctx.getCanonicalType(OperatorNew->getParamDecl(1)->getType());
405
406 if (ParamType == Ctx.VoidPtrTy)
Ken Dyckcaf647c2010-01-26 19:44:24 +0000407 return CharUnits::Zero();
Anders Carlssondd937552009-12-13 20:34:34 +0000408 }
409 }
410
Anders Carlsson871d0782009-12-13 20:04:38 +0000411 return CalculateCookiePadding(Ctx, E->getAllocatedType());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000412}
413
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000414static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context,
415 CodeGenFunction &CGF,
Anders Carlssona4d4c012009-09-23 16:07:23 +0000416 const CXXNewExpr *E,
417 llvm::Value *& NumElements) {
418 QualType Type = E->getAllocatedType();
Ken Dyckcaf647c2010-01-26 19:44:24 +0000419 CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(Type);
Anders Carlssona4d4c012009-09-23 16:07:23 +0000420 const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType());
421
422 if (!E->isArray())
Ken Dyckcaf647c2010-01-26 19:44:24 +0000423 return llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000424
Ken Dyckcaf647c2010-01-26 19:44:24 +0000425 CharUnits CookiePadding = CalculateCookiePadding(CGF.getContext(), E);
Anders Carlssona4d4c012009-09-23 16:07:23 +0000426
427 Expr::EvalResult Result;
428 if (E->getArraySize()->Evaluate(Result, CGF.getContext()) &&
429 !Result.HasSideEffects && Result.Val.isInt()) {
430
Ken Dyckcaf647c2010-01-26 19:44:24 +0000431 CharUnits AllocSize =
432 Result.Val.getInt().getZExtValue() * TypeSize + CookiePadding;
Anders Carlssona4d4c012009-09-23 16:07:23 +0000433
434 NumElements =
435 llvm::ConstantInt::get(SizeTy, Result.Val.getInt().getZExtValue());
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000436 while (const ArrayType *AType = Context.getAsArrayType(Type)) {
437 const llvm::ArrayType *llvmAType =
438 cast<llvm::ArrayType>(CGF.ConvertType(Type));
439 NumElements =
440 CGF.Builder.CreateMul(NumElements,
441 llvm::ConstantInt::get(
442 SizeTy, llvmAType->getNumElements()));
443 Type = AType->getElementType();
444 }
Anders Carlssona4d4c012009-09-23 16:07:23 +0000445
Ken Dyckcaf647c2010-01-26 19:44:24 +0000446 return llvm::ConstantInt::get(SizeTy, AllocSize.getQuantity());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000447 }
448
449 // Emit the array size expression.
450 NumElements = CGF.EmitScalarExpr(E->getArraySize());
451
452 // Multiply with the type size.
453 llvm::Value *V =
454 CGF.Builder.CreateMul(NumElements,
Ken Dyckcaf647c2010-01-26 19:44:24 +0000455 llvm::ConstantInt::get(SizeTy,
456 TypeSize.getQuantity()));
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000457
458 while (const ArrayType *AType = Context.getAsArrayType(Type)) {
459 const llvm::ArrayType *llvmAType =
460 cast<llvm::ArrayType>(CGF.ConvertType(Type));
461 NumElements =
462 CGF.Builder.CreateMul(NumElements,
463 llvm::ConstantInt::get(
464 SizeTy, llvmAType->getNumElements()));
465 Type = AType->getElementType();
466 }
Anders Carlssona4d4c012009-09-23 16:07:23 +0000467
468 // And add the cookie padding if necessary.
Ken Dyckcaf647c2010-01-26 19:44:24 +0000469 if (!CookiePadding.isZero())
470 V = CGF.Builder.CreateAdd(V,
471 llvm::ConstantInt::get(SizeTy, CookiePadding.getQuantity()));
Anders Carlssona4d4c012009-09-23 16:07:23 +0000472
473 return V;
474}
475
476static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
477 llvm::Value *NewPtr,
478 llvm::Value *NumElements) {
Anders Carlsson5d4d9462009-11-24 18:43:52 +0000479 if (E->isArray()) {
480 if (CXXConstructorDecl *Ctor = E->getConstructor())
481 CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr,
482 E->constructor_arg_begin(),
483 E->constructor_arg_end());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000484 return;
485 }
486
Anders Carlsson5d4d9462009-11-24 18:43:52 +0000487 QualType AllocType = E->getAllocatedType();
488
489 if (CXXConstructorDecl *Ctor = E->getConstructor()) {
490 CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, NewPtr,
491 E->constructor_arg_begin(),
492 E->constructor_arg_end());
493
494 return;
495 }
496
497 // We have a POD type.
498 if (E->getNumConstructorArgs() == 0)
499 return;
500
501 assert(E->getNumConstructorArgs() == 1 &&
502 "Can only have one argument to initializer of POD type.");
503
504 const Expr *Init = E->getConstructorArg(0);
505
506 if (!CGF.hasAggregateLLVMType(AllocType))
507 CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr,
508 AllocType.isVolatileQualified(), AllocType);
509 else if (AllocType->isAnyComplexType())
510 CGF.EmitComplexExprIntoAddr(Init, NewPtr,
511 AllocType.isVolatileQualified());
512 else
513 CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified());
Anders Carlssona4d4c012009-09-23 16:07:23 +0000514}
515
Anders Carlsson16d81b82009-09-22 22:53:17 +0000516llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
Anders Carlsson16d81b82009-09-22 22:53:17 +0000517 QualType AllocType = E->getAllocatedType();
518 FunctionDecl *NewFD = E->getOperatorNew();
519 const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>();
520
521 CallArgList NewArgs;
522
523 // The allocation size is the first argument.
524 QualType SizeTy = getContext().getSizeType();
Anders Carlsson16d81b82009-09-22 22:53:17 +0000525
Anders Carlssona4d4c012009-09-23 16:07:23 +0000526 llvm::Value *NumElements = 0;
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000527 llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(),
528 *this, E, NumElements);
Anders Carlssona4d4c012009-09-23 16:07:23 +0000529
Anders Carlsson16d81b82009-09-22 22:53:17 +0000530 NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy));
531
532 // Emit the rest of the arguments.
533 // FIXME: Ideally, this should just use EmitCallArgs.
534 CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin();
535
536 // First, use the types from the function type.
537 // We start at 1 here because the first argument (the allocation size)
538 // has already been emitted.
539 for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) {
540 QualType ArgType = NewFTy->getArgType(i);
541
542 assert(getContext().getCanonicalType(ArgType.getNonReferenceType()).
543 getTypePtr() ==
544 getContext().getCanonicalType(NewArg->getType()).getTypePtr() &&
545 "type mismatch in call argument!");
546
547 NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
548 ArgType));
549
550 }
551
552 // Either we've emitted all the call args, or we have a call to a
553 // variadic function.
554 assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) &&
555 "Extra arguments in non-variadic function!");
556
557 // If we still have any arguments, emit them using the type of the argument.
558 for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end();
559 NewArg != NewArgEnd; ++NewArg) {
560 QualType ArgType = NewArg->getType();
561 NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
562 ArgType));
563 }
564
565 // Emit the call to new.
566 RValue RV =
John McCall04a67a62010-02-05 21:31:56 +0000567 EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy),
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000568 CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD);
Anders Carlsson16d81b82009-09-22 22:53:17 +0000569
570 // If an allocation function is declared with an empty exception specification
571 // it returns null to indicate failure to allocate storage. [expr.new]p13.
572 // (We don't need to check for null when there's no new initializer and
573 // we're allocating a POD type).
574 bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() &&
575 !(AllocType->isPODType() && !E->hasInitializer());
576
577 llvm::BasicBlock *NewNull = 0;
578 llvm::BasicBlock *NewNotNull = 0;
579 llvm::BasicBlock *NewEnd = 0;
580
581 llvm::Value *NewPtr = RV.getScalarVal();
582
583 if (NullCheckResult) {
584 NewNull = createBasicBlock("new.null");
585 NewNotNull = createBasicBlock("new.notnull");
586 NewEnd = createBasicBlock("new.end");
587
588 llvm::Value *IsNull =
589 Builder.CreateICmpEQ(NewPtr,
590 llvm::Constant::getNullValue(NewPtr->getType()),
591 "isnull");
592
593 Builder.CreateCondBr(IsNull, NewNull, NewNotNull);
594 EmitBlock(NewNotNull);
595 }
Ken Dyckcaf647c2010-01-26 19:44:24 +0000596
597 CharUnits CookiePadding = CalculateCookiePadding(getContext(), E);
598 if (!CookiePadding.isZero()) {
599 CharUnits CookieOffset =
600 CookiePadding - getContext().getTypeSizeInChars(SizeTy);
Anders Carlsson6ac5fc42009-09-23 18:59:48 +0000601
602 llvm::Value *NumElementsPtr =
Ken Dyckcaf647c2010-01-26 19:44:24 +0000603 Builder.CreateConstInBoundsGEP1_64(NewPtr, CookieOffset.getQuantity());
Anders Carlsson6ac5fc42009-09-23 18:59:48 +0000604
605 NumElementsPtr = Builder.CreateBitCast(NumElementsPtr,
606 ConvertType(SizeTy)->getPointerTo());
607 Builder.CreateStore(NumElements, NumElementsPtr);
608
609 // Now add the padding to the new ptr.
Ken Dyckcaf647c2010-01-26 19:44:24 +0000610 NewPtr = Builder.CreateConstInBoundsGEP1_64(NewPtr,
611 CookiePadding.getQuantity());
Anders Carlsson6ac5fc42009-09-23 18:59:48 +0000612 }
613
Fariborz Jahanianceb43b62010-03-24 16:57:01 +0000614 if (AllocType->isArrayType()) {
615 while (const ArrayType *AType = getContext().getAsArrayType(AllocType))
616 AllocType = AType->getElementType();
617 NewPtr =
618 Builder.CreateBitCast(NewPtr,
619 ConvertType(getContext().getPointerType(AllocType)));
620 EmitNewInitializer(*this, E, NewPtr, NumElements);
621 NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType()));
622 }
623 else {
624 NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType()));
625 EmitNewInitializer(*this, E, NewPtr, NumElements);
626 }
627
Anders Carlsson16d81b82009-09-22 22:53:17 +0000628 if (NullCheckResult) {
629 Builder.CreateBr(NewEnd);
Eli Friedman7f1de452009-11-10 22:39:09 +0000630 NewNotNull = Builder.GetInsertBlock();
Anders Carlsson16d81b82009-09-22 22:53:17 +0000631 EmitBlock(NewNull);
632 Builder.CreateBr(NewEnd);
633 EmitBlock(NewEnd);
634
635 llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType());
636 PHI->reserveOperandSpace(2);
637 PHI->addIncoming(NewPtr, NewNotNull);
638 PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), NewNull);
639
640 NewPtr = PHI;
641 }
642
643 return NewPtr;
644}
645
Anders Carlsson871d0782009-12-13 20:04:38 +0000646static std::pair<llvm::Value *, llvm::Value *>
647GetAllocatedObjectPtrAndNumElements(CodeGenFunction &CGF,
648 llvm::Value *Ptr, QualType DeleteTy) {
649 QualType SizeTy = CGF.getContext().getSizeType();
650 const llvm::Type *SizeLTy = CGF.ConvertType(SizeTy);
651
Ken Dyckcaf647c2010-01-26 19:44:24 +0000652 CharUnits DeleteTypeAlign = CGF.getContext().getTypeAlignInChars(DeleteTy);
653 CharUnits CookiePadding =
654 std::max(CGF.getContext().getTypeSizeInChars(SizeTy),
655 DeleteTypeAlign);
656 assert(!CookiePadding.isZero() && "CookiePadding should not be 0.");
Anders Carlsson871d0782009-12-13 20:04:38 +0000657
658 const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext());
Ken Dyckcaf647c2010-01-26 19:44:24 +0000659 CharUnits CookieOffset =
660 CookiePadding - CGF.getContext().getTypeSizeInChars(SizeTy);
Anders Carlsson871d0782009-12-13 20:04:38 +0000661
662 llvm::Value *AllocatedObjectPtr = CGF.Builder.CreateBitCast(Ptr, Int8PtrTy);
663 AllocatedObjectPtr =
664 CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr,
Ken Dyckcaf647c2010-01-26 19:44:24 +0000665 -CookiePadding.getQuantity());
Anders Carlsson871d0782009-12-13 20:04:38 +0000666
667 llvm::Value *NumElementsPtr =
668 CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr,
Ken Dyckcaf647c2010-01-26 19:44:24 +0000669 CookieOffset.getQuantity());
Anders Carlsson871d0782009-12-13 20:04:38 +0000670 NumElementsPtr =
671 CGF.Builder.CreateBitCast(NumElementsPtr, SizeLTy->getPointerTo());
672
673 llvm::Value *NumElements = CGF.Builder.CreateLoad(NumElementsPtr);
674 NumElements =
675 CGF.Builder.CreateIntCast(NumElements, SizeLTy, /*isSigned=*/false);
676
677 return std::make_pair(AllocatedObjectPtr, NumElements);
678}
679
Eli Friedman5fe05982009-11-18 00:50:08 +0000680void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
681 llvm::Value *Ptr,
682 QualType DeleteTy) {
683 const FunctionProtoType *DeleteFTy =
684 DeleteFD->getType()->getAs<FunctionProtoType>();
685
686 CallArgList DeleteArgs;
687
Anders Carlsson871d0782009-12-13 20:04:38 +0000688 // Check if we need to pass the size to the delete operator.
689 llvm::Value *Size = 0;
690 QualType SizeTy;
691 if (DeleteFTy->getNumArgs() == 2) {
692 SizeTy = DeleteFTy->getArgType(1);
Ken Dyck4f122ef2010-01-26 19:59:28 +0000693 CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
694 Size = llvm::ConstantInt::get(ConvertType(SizeTy),
695 DeleteTypeSize.getQuantity());
Anders Carlsson871d0782009-12-13 20:04:38 +0000696 }
697
698 if (DeleteFD->getOverloadedOperator() == OO_Array_Delete &&
Ken Dyckcaf647c2010-01-26 19:44:24 +0000699 !CalculateCookiePadding(getContext(), DeleteTy).isZero()) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000700 // We need to get the number of elements in the array from the cookie.
701 llvm::Value *AllocatedObjectPtr;
702 llvm::Value *NumElements;
703 llvm::tie(AllocatedObjectPtr, NumElements) =
704 GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy);
705
706 // Multiply the size with the number of elements.
707 if (Size)
708 Size = Builder.CreateMul(NumElements, Size);
709
710 Ptr = AllocatedObjectPtr;
711 }
712
Eli Friedman5fe05982009-11-18 00:50:08 +0000713 QualType ArgTy = DeleteFTy->getArgType(0);
714 llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
715 DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy));
716
Anders Carlsson871d0782009-12-13 20:04:38 +0000717 if (Size)
Eli Friedman5fe05982009-11-18 00:50:08 +0000718 DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy));
Eli Friedman5fe05982009-11-18 00:50:08 +0000719
720 // Emit the call to delete.
John McCall04a67a62010-02-05 21:31:56 +0000721 EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy),
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000722 CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(),
Eli Friedman5fe05982009-11-18 00:50:08 +0000723 DeleteArgs, DeleteFD);
724}
725
Anders Carlsson16d81b82009-09-22 22:53:17 +0000726void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
Fariborz Jahanian72c21532009-11-13 19:27:47 +0000727
Douglas Gregor90916562009-09-29 18:16:17 +0000728 // Get at the argument before we performed the implicit conversion
729 // to void*.
730 const Expr *Arg = E->getArgument();
731 while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) {
732 if (ICE->getCastKind() != CastExpr::CK_UserDefinedConversion &&
733 ICE->getType()->isVoidPointerType())
734 Arg = ICE->getSubExpr();
Douglas Gregord69dd782009-10-01 05:49:51 +0000735 else
736 break;
Douglas Gregor90916562009-09-29 18:16:17 +0000737 }
738
739 QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
Anders Carlsson16d81b82009-09-22 22:53:17 +0000740
Douglas Gregor90916562009-09-29 18:16:17 +0000741 llvm::Value *Ptr = EmitScalarExpr(Arg);
Anders Carlsson16d81b82009-09-22 22:53:17 +0000742
743 // Null check the pointer.
744 llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
745 llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
746
747 llvm::Value *IsNull =
748 Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()),
749 "isnull");
750
751 Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
752 EmitBlock(DeleteNotNull);
Anders Carlsson566abee2009-11-13 04:45:41 +0000753
754 bool ShouldCallDelete = true;
755
Anders Carlsson16d81b82009-09-22 22:53:17 +0000756 // Call the destructor if necessary.
757 if (const RecordType *RT = DeleteTy->getAs<RecordType>()) {
758 if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl())) {
759 if (!RD->hasTrivialDestructor()) {
760 const CXXDestructorDecl *Dtor = RD->getDestructor(getContext());
Fariborz Jahanian72c21532009-11-13 19:27:47 +0000761 if (E->isArrayForm()) {
Anders Carlsson871d0782009-12-13 20:04:38 +0000762 llvm::Value *AllocatedObjectPtr;
763 llvm::Value *NumElements;
764 llvm::tie(AllocatedObjectPtr, NumElements) =
765 GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy);
766
Anders Carlsson61537102009-12-13 18:48:07 +0000767 EmitCXXAggrDestructorCall(Dtor, NumElements, Ptr);
Anders Carlsson61537102009-12-13 18:48:07 +0000768 } else if (Dtor->isVirtual()) {
Anders Carlsson16d81b82009-09-22 22:53:17 +0000769 const llvm::Type *Ty =
770 CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(Dtor),
771 /*isVariadic=*/false);
772
Anders Carlsson566abee2009-11-13 04:45:41 +0000773 llvm::Value *Callee = BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty);
Anders Carlssonc997d422010-01-02 01:01:18 +0000774 EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0,
775 0, 0);
Anders Carlsson566abee2009-11-13 04:45:41 +0000776
777 // The dtor took care of deleting the object.
778 ShouldCallDelete = false;
Anders Carlsson16d81b82009-09-22 22:53:17 +0000779 } else
780 EmitCXXDestructorCall(Dtor, Dtor_Complete, Ptr);
781 }
782 }
783 }
784
Eli Friedman5fe05982009-11-18 00:50:08 +0000785 if (ShouldCallDelete)
786 EmitDeleteCall(E->getOperatorDelete(), Ptr, DeleteTy);
Anders Carlsson16d81b82009-09-22 22:53:17 +0000787
Anders Carlsson16d81b82009-09-22 22:53:17 +0000788 EmitBlock(DeleteEnd);
789}
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000790
791llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
792 QualType Ty = E->getType();
793 const llvm::Type *LTy = ConvertType(Ty)->getPointerTo();
Anders Carlsson31b7f522009-12-11 02:46:30 +0000794
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000795 if (E->isTypeOperand()) {
796 llvm::Constant *TypeInfo =
797 CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand());
798 return Builder.CreateBitCast(TypeInfo, LTy);
799 }
800
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000801 Expr *subE = E->getExprOperand();
Mike Stump5fae8562009-11-17 22:33:00 +0000802 Ty = subE->getType();
803 CanQualType CanTy = CGM.getContext().getCanonicalType(Ty);
804 Ty = CanTy.getUnqualifiedType().getNonReferenceType();
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000805 if (const RecordType *RT = Ty->getAs<RecordType>()) {
806 const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
807 if (RD->isPolymorphic()) {
808 // FIXME: if subE is an lvalue do
809 LValue Obj = EmitLValue(subE);
810 llvm::Value *This = Obj.getAddress();
Mike Stumpf549e892009-11-15 16:52:53 +0000811 LTy = LTy->getPointerTo()->getPointerTo();
812 llvm::Value *V = Builder.CreateBitCast(This, LTy);
813 // We need to do a zero check for *p, unless it has NonNullAttr.
814 // FIXME: PointerType->hasAttr<NonNullAttr>()
815 bool CanBeZero = false;
Mike Stumpdb519a42009-11-17 00:45:21 +0000816 if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens()))
Mike Stumpf549e892009-11-15 16:52:53 +0000817 if (UO->getOpcode() == UnaryOperator::Deref)
818 CanBeZero = true;
819 if (CanBeZero) {
820 llvm::BasicBlock *NonZeroBlock = createBasicBlock();
821 llvm::BasicBlock *ZeroBlock = createBasicBlock();
822
823 llvm::Value *Zero = llvm::Constant::getNullValue(LTy);
824 Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero),
825 NonZeroBlock, ZeroBlock);
826 EmitBlock(ZeroBlock);
827 /// Call __cxa_bad_typeid
828 const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext);
829 const llvm::FunctionType *FTy;
830 FTy = llvm::FunctionType::get(ResultType, false);
831 llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid");
Mike Stumpc849c052009-11-16 06:50:58 +0000832 Builder.CreateCall(F)->setDoesNotReturn();
Mike Stumpf549e892009-11-15 16:52:53 +0000833 Builder.CreateUnreachable();
834 EmitBlock(NonZeroBlock);
835 }
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000836 V = Builder.CreateLoad(V, "vtable");
837 V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL);
838 V = Builder.CreateLoad(V);
839 return V;
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000840 }
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000841 }
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000842 return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy);
Mike Stumpc2e84ae2009-11-15 08:09:41 +0000843}
Mike Stumpc849c052009-11-16 06:50:58 +0000844
845llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V,
846 const CXXDynamicCastExpr *DCE) {
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000847 QualType SrcTy = DCE->getSubExpr()->getType();
848 QualType DestTy = DCE->getTypeAsWritten();
849 QualType InnerType = DestTy->getPointeeType();
850
Mike Stumpc849c052009-11-16 06:50:58 +0000851 const llvm::Type *LTy = ConvertType(DCE->getType());
Mike Stump2b35baf2009-11-16 22:52:20 +0000852
Mike Stumpc849c052009-11-16 06:50:58 +0000853 bool CanBeZero = false;
Mike Stumpc849c052009-11-16 06:50:58 +0000854 bool ToVoid = false;
Mike Stump2b35baf2009-11-16 22:52:20 +0000855 bool ThrowOnBad = false;
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000856 if (DestTy->isPointerType()) {
Mike Stumpc849c052009-11-16 06:50:58 +0000857 // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this
858 CanBeZero = true;
859 if (InnerType->isVoidType())
860 ToVoid = true;
861 } else {
862 LTy = LTy->getPointerTo();
863 ThrowOnBad = true;
864 }
865
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000866 if (SrcTy->isPointerType() || SrcTy->isReferenceType())
867 SrcTy = SrcTy->getPointeeType();
868 SrcTy = SrcTy.getUnqualifiedType();
869
Anders Carlsson6f0e4852009-12-18 14:55:04 +0000870 if (DestTy->isPointerType() || DestTy->isReferenceType())
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000871 DestTy = DestTy->getPointeeType();
872 DestTy = DestTy.getUnqualifiedType();
Mike Stumpc849c052009-11-16 06:50:58 +0000873
Mike Stumpc849c052009-11-16 06:50:58 +0000874 llvm::BasicBlock *ContBlock = createBasicBlock();
875 llvm::BasicBlock *NullBlock = 0;
876 llvm::BasicBlock *NonZeroBlock = 0;
877 if (CanBeZero) {
878 NonZeroBlock = createBasicBlock();
879 NullBlock = createBasicBlock();
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000880 Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock);
Mike Stumpc849c052009-11-16 06:50:58 +0000881 EmitBlock(NonZeroBlock);
882 }
883
Mike Stumpc849c052009-11-16 06:50:58 +0000884 llvm::BasicBlock *BadCastBlock = 0;
Mike Stumpc849c052009-11-16 06:50:58 +0000885
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000886 const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
Mike Stump2b35baf2009-11-16 22:52:20 +0000887
888 // See if this is a dynamic_cast(void*)
889 if (ToVoid) {
890 llvm::Value *This = V;
891 V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo());
892 V = Builder.CreateLoad(V, "vtable");
893 V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL);
894 V = Builder.CreateLoad(V, "offset to top");
895 This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext));
896 V = Builder.CreateInBoundsGEP(This, V);
897 V = Builder.CreateBitCast(V, LTy);
898 } else {
899 /// Call __dynamic_cast
900 const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext);
901 const llvm::FunctionType *FTy;
902 std::vector<const llvm::Type*> ArgTys;
903 const llvm::Type *PtrToInt8Ty
904 = llvm::Type::getInt8Ty(VMContext)->getPointerTo();
905 ArgTys.push_back(PtrToInt8Ty);
906 ArgTys.push_back(PtrToInt8Ty);
907 ArgTys.push_back(PtrToInt8Ty);
908 ArgTys.push_back(PtrDiffTy);
909 FTy = llvm::FunctionType::get(ResultType, ArgTys, false);
Mike Stump2b35baf2009-11-16 22:52:20 +0000910
911 // FIXME: Calculate better hint.
912 llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL);
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000913
914 assert(SrcTy->isRecordType() && "Src type must be record type!");
915 assert(DestTy->isRecordType() && "Dest type must be record type!");
916
Douglas Gregor154fe982009-12-23 22:04:40 +0000917 llvm::Value *SrcArg
918 = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType());
919 llvm::Value *DestArg
920 = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType());
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000921
Mike Stump2b35baf2009-11-16 22:52:20 +0000922 V = Builder.CreateBitCast(V, PtrToInt8Ty);
923 V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"),
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000924 V, SrcArg, DestArg, hint);
Mike Stump2b35baf2009-11-16 22:52:20 +0000925 V = Builder.CreateBitCast(V, LTy);
926
927 if (ThrowOnBad) {
928 BadCastBlock = createBasicBlock();
929
Anders Carlsson1d7088d2009-12-17 07:09:17 +0000930 Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock);
Mike Stump2b35baf2009-11-16 22:52:20 +0000931 EmitBlock(BadCastBlock);
932 /// Call __cxa_bad_cast
933 ResultType = llvm::Type::getVoidTy(VMContext);
934 const llvm::FunctionType *FBadTy;
Mike Stumpfde17be2009-11-17 03:01:03 +0000935 FBadTy = llvm::FunctionType::get(ResultType, false);
Mike Stump2b35baf2009-11-16 22:52:20 +0000936 llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast");
937 Builder.CreateCall(F)->setDoesNotReturn();
Mike Stump8b152b82009-11-17 00:08:50 +0000938 Builder.CreateUnreachable();
Mike Stump2b35baf2009-11-16 22:52:20 +0000939 }
Mike Stumpc849c052009-11-16 06:50:58 +0000940 }
941
942 if (CanBeZero) {
943 Builder.CreateBr(ContBlock);
944 EmitBlock(NullBlock);
945 Builder.CreateBr(ContBlock);
946 }
947 EmitBlock(ContBlock);
948 if (CanBeZero) {
949 llvm::PHINode *PHI = Builder.CreatePHI(LTy);
Mike Stump14431c12009-11-17 00:10:05 +0000950 PHI->reserveOperandSpace(2);
Mike Stumpc849c052009-11-16 06:50:58 +0000951 PHI->addIncoming(V, NonZeroBlock);
952 PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock);
Mike Stumpc849c052009-11-16 06:50:58 +0000953 V = PHI;
954 }
955
956 return V;
957}