159486a2dSAnders Carlsson //===--- CGExprCXX.cpp - Emit LLVM Code for C++ expressions ---------------===//
259486a2dSAnders Carlsson //
359486a2dSAnders Carlsson //                     The LLVM Compiler Infrastructure
459486a2dSAnders Carlsson //
559486a2dSAnders Carlsson // This file is distributed under the University of Illinois Open Source
659486a2dSAnders Carlsson // License. See LICENSE.TXT for details.
759486a2dSAnders Carlsson //
859486a2dSAnders Carlsson //===----------------------------------------------------------------------===//
959486a2dSAnders Carlsson //
1059486a2dSAnders Carlsson // This contains code dealing with code generation of C++ expressions
1159486a2dSAnders Carlsson //
1259486a2dSAnders Carlsson //===----------------------------------------------------------------------===//
1359486a2dSAnders Carlsson 
1459486a2dSAnders Carlsson #include "CodeGenFunction.h"
1560d215b6SFariborz Jahanian #include "CGObjCRuntime.h"
1626008e07SChris Lattner #include "llvm/Intrinsics.h"
1759486a2dSAnders Carlsson using namespace clang;
1859486a2dSAnders Carlsson using namespace CodeGen;
1959486a2dSAnders Carlsson 
2027da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD,
2127da15baSAnders Carlsson                                           llvm::Value *Callee,
2227da15baSAnders Carlsson                                           ReturnValueSlot ReturnValue,
2327da15baSAnders Carlsson                                           llvm::Value *This,
24e36a6b3eSAnders Carlsson                                           llvm::Value *VTT,
2527da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgBeg,
2627da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgEnd) {
2727da15baSAnders Carlsson   assert(MD->isInstance() &&
2827da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
2927da15baSAnders Carlsson 
3027da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
3127da15baSAnders Carlsson 
3227da15baSAnders Carlsson   CallArgList Args;
3327da15baSAnders Carlsson 
3427da15baSAnders Carlsson   // Push the this ptr.
3527da15baSAnders Carlsson   Args.push_back(std::make_pair(RValue::get(This),
3627da15baSAnders Carlsson                                 MD->getThisType(getContext())));
3727da15baSAnders Carlsson 
38e36a6b3eSAnders Carlsson   // If there is a VTT parameter, emit it.
39e36a6b3eSAnders Carlsson   if (VTT) {
40e36a6b3eSAnders Carlsson     QualType T = getContext().getPointerType(getContext().VoidPtrTy);
41e36a6b3eSAnders Carlsson     Args.push_back(std::make_pair(RValue::get(VTT), T));
42e36a6b3eSAnders Carlsson   }
43e36a6b3eSAnders Carlsson 
4427da15baSAnders Carlsson   // And the rest of the call args
4527da15baSAnders Carlsson   EmitCallArgs(Args, FPT, ArgBeg, ArgEnd);
4627da15baSAnders Carlsson 
47ab26cfa5SJohn McCall   QualType ResultType = FPT->getResultType();
48ab26cfa5SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args,
49c50c27ccSRafael Espindola                                                  FPT->getExtInfo()),
50c50c27ccSRafael Espindola                   Callee, ReturnValue, Args, MD);
5127da15baSAnders Carlsson }
5227da15baSAnders Carlsson 
5327da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given
5427da15baSAnders Carlsson /// expr can be devirtualized.
5527da15baSAnders Carlsson static bool canDevirtualizeMemberFunctionCalls(const Expr *Base) {
5627da15baSAnders Carlsson   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) {
5727da15baSAnders Carlsson     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
5827da15baSAnders Carlsson       // This is a record decl. We know the type and can devirtualize it.
5927da15baSAnders Carlsson       return VD->getType()->isRecordType();
6027da15baSAnders Carlsson     }
6127da15baSAnders Carlsson 
6227da15baSAnders Carlsson     return false;
6327da15baSAnders Carlsson   }
6427da15baSAnders Carlsson 
6527da15baSAnders Carlsson   // We can always devirtualize calls on temporary object expressions.
66a682427eSEli Friedman   if (isa<CXXConstructExpr>(Base))
6727da15baSAnders Carlsson     return true;
6827da15baSAnders Carlsson 
6927da15baSAnders Carlsson   // And calls on bound temporaries.
7027da15baSAnders Carlsson   if (isa<CXXBindTemporaryExpr>(Base))
7127da15baSAnders Carlsson     return true;
7227da15baSAnders Carlsson 
7327da15baSAnders Carlsson   // Check if this is a call expr that returns a record type.
7427da15baSAnders Carlsson   if (const CallExpr *CE = dyn_cast<CallExpr>(Base))
7527da15baSAnders Carlsson     return CE->getCallReturnType()->isRecordType();
7627da15baSAnders Carlsson 
7727da15baSAnders Carlsson   // We can't devirtualize the call.
7827da15baSAnders Carlsson   return false;
7927da15baSAnders Carlsson }
8027da15baSAnders Carlsson 
8127da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
8227da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
8327da15baSAnders Carlsson   if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens()))
8427da15baSAnders Carlsson     return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
8527da15baSAnders Carlsson 
8627da15baSAnders Carlsson   const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens());
8727da15baSAnders Carlsson   const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
8827da15baSAnders Carlsson 
8927da15baSAnders Carlsson   if (MD->isStatic()) {
9027da15baSAnders Carlsson     // The method is static, emit it as we would a regular call.
9127da15baSAnders Carlsson     llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
9227da15baSAnders Carlsson     return EmitCall(getContext().getPointerType(MD->getType()), Callee,
9327da15baSAnders Carlsson                     ReturnValue, CE->arg_begin(), CE->arg_end());
9427da15baSAnders Carlsson   }
9527da15baSAnders Carlsson 
9627da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
9727da15baSAnders Carlsson 
9827da15baSAnders Carlsson   const llvm::Type *Ty =
9927da15baSAnders Carlsson     CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
10027da15baSAnders Carlsson                                    FPT->isVariadic());
10127da15baSAnders Carlsson   llvm::Value *This;
10227da15baSAnders Carlsson 
10327da15baSAnders Carlsson   if (ME->isArrow())
10427da15baSAnders Carlsson     This = EmitScalarExpr(ME->getBase());
10527da15baSAnders Carlsson   else {
10627da15baSAnders Carlsson     LValue BaseLV = EmitLValue(ME->getBase());
10727da15baSAnders Carlsson     This = BaseLV.getAddress();
10827da15baSAnders Carlsson   }
10927da15baSAnders Carlsson 
11027da15baSAnders Carlsson   if (MD->isCopyAssignment() && MD->isTrivial()) {
11127da15baSAnders Carlsson     // We don't like to generate the trivial copy assignment operator when
11227da15baSAnders Carlsson     // it isn't necessary; just produce the proper effect here.
11327da15baSAnders Carlsson     llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
11427da15baSAnders Carlsson     EmitAggregateCopy(This, RHS, CE->getType());
11527da15baSAnders Carlsson     return RValue::get(This);
11627da15baSAnders Carlsson   }
11727da15baSAnders Carlsson 
11827da15baSAnders Carlsson   // C++ [class.virtual]p12:
11927da15baSAnders Carlsson   //   Explicit qualification with the scope operator (5.1) suppresses the
12027da15baSAnders Carlsson   //   virtual call mechanism.
12127da15baSAnders Carlsson   //
12227da15baSAnders Carlsson   // We also don't emit a virtual call if the base expression has a record type
12327da15baSAnders Carlsson   // because then we know what the type is.
12427da15baSAnders Carlsson   llvm::Value *Callee;
12527da15baSAnders Carlsson   if (const CXXDestructorDecl *Destructor
12627da15baSAnders Carlsson              = dyn_cast<CXXDestructorDecl>(MD)) {
12727da15baSAnders Carlsson     if (Destructor->isTrivial())
12827da15baSAnders Carlsson       return RValue::get(0);
12927da15baSAnders Carlsson     if (MD->isVirtual() && !ME->hasQualifier() &&
13027da15baSAnders Carlsson         !canDevirtualizeMemberFunctionCalls(ME->getBase())) {
13127da15baSAnders Carlsson       Callee = BuildVirtualCall(Destructor, Dtor_Complete, This, Ty);
13227da15baSAnders Carlsson     } else {
13327da15baSAnders Carlsson       Callee = CGM.GetAddrOfFunction(GlobalDecl(Destructor, Dtor_Complete), Ty);
13427da15baSAnders Carlsson     }
13527da15baSAnders Carlsson   } else if (MD->isVirtual() && !ME->hasQualifier() &&
13627da15baSAnders Carlsson              !canDevirtualizeMemberFunctionCalls(ME->getBase())) {
13727da15baSAnders Carlsson     Callee = BuildVirtualCall(MD, This, Ty);
13827da15baSAnders Carlsson   } else {
13927da15baSAnders Carlsson     Callee = CGM.GetAddrOfFunction(MD, Ty);
14027da15baSAnders Carlsson   }
14127da15baSAnders Carlsson 
142e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
14327da15baSAnders Carlsson                            CE->arg_begin(), CE->arg_end());
14427da15baSAnders Carlsson }
14527da15baSAnders Carlsson 
14627da15baSAnders Carlsson RValue
14727da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
14827da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
14927da15baSAnders Carlsson   const BinaryOperator *BO =
15027da15baSAnders Carlsson       cast<BinaryOperator>(E->getCallee()->IgnoreParens());
15127da15baSAnders Carlsson   const Expr *BaseExpr = BO->getLHS();
15227da15baSAnders Carlsson   const Expr *MemFnExpr = BO->getRHS();
15327da15baSAnders Carlsson 
15427da15baSAnders Carlsson   const MemberPointerType *MPT =
15527da15baSAnders Carlsson     MemFnExpr->getType()->getAs<MemberPointerType>();
15627da15baSAnders Carlsson   const FunctionProtoType *FPT =
15727da15baSAnders Carlsson     MPT->getPointeeType()->getAs<FunctionProtoType>();
15827da15baSAnders Carlsson   const CXXRecordDecl *RD =
15927da15baSAnders Carlsson     cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
16027da15baSAnders Carlsson 
16127da15baSAnders Carlsson   const llvm::FunctionType *FTy =
16227da15baSAnders Carlsson     CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(RD, FPT),
16327da15baSAnders Carlsson                                    FPT->isVariadic());
16427da15baSAnders Carlsson 
165c4325486SAnders Carlsson   const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(VMContext);
16627da15baSAnders Carlsson 
16727da15baSAnders Carlsson   // Get the member function pointer.
168a7566f16SDaniel Dunbar   llvm::Value *MemFnPtr = CreateMemTemp(MemFnExpr->getType(), "mem.fn");
16927da15baSAnders Carlsson   EmitAggExpr(MemFnExpr, MemFnPtr, /*VolatileDest=*/false);
17027da15baSAnders Carlsson 
17127da15baSAnders Carlsson   // Emit the 'this' pointer.
17227da15baSAnders Carlsson   llvm::Value *This;
17327da15baSAnders Carlsson 
17427da15baSAnders Carlsson   if (BO->getOpcode() == BinaryOperator::PtrMemI)
17527da15baSAnders Carlsson     This = EmitScalarExpr(BaseExpr);
17627da15baSAnders Carlsson   else
17727da15baSAnders Carlsson     This = EmitLValue(BaseExpr).getAddress();
17827da15baSAnders Carlsson 
17927da15baSAnders Carlsson   // Adjust it.
18027da15baSAnders Carlsson   llvm::Value *Adj = Builder.CreateStructGEP(MemFnPtr, 1);
18127da15baSAnders Carlsson   Adj = Builder.CreateLoad(Adj, "mem.fn.adj");
18227da15baSAnders Carlsson 
18327da15baSAnders Carlsson   llvm::Value *Ptr = Builder.CreateBitCast(This, Int8PtrTy, "ptr");
18427da15baSAnders Carlsson   Ptr = Builder.CreateGEP(Ptr, Adj, "adj");
18527da15baSAnders Carlsson 
18627da15baSAnders Carlsson   This = Builder.CreateBitCast(Ptr, This->getType(), "this");
18727da15baSAnders Carlsson 
18827da15baSAnders Carlsson   llvm::Value *FnPtr = Builder.CreateStructGEP(MemFnPtr, 0, "mem.fn.ptr");
18927da15baSAnders Carlsson 
19027da15baSAnders Carlsson   const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
19127da15baSAnders Carlsson 
19227da15baSAnders Carlsson   llvm::Value *FnAsInt = Builder.CreateLoad(FnPtr, "fn");
19327da15baSAnders Carlsson 
19427da15baSAnders Carlsson   // If the LSB in the function pointer is 1, the function pointer points to
19527da15baSAnders Carlsson   // a virtual function.
19627da15baSAnders Carlsson   llvm::Value *IsVirtual
19727da15baSAnders Carlsson     = Builder.CreateAnd(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1),
19827da15baSAnders Carlsson                         "and");
19927da15baSAnders Carlsson 
20027da15baSAnders Carlsson   IsVirtual = Builder.CreateTrunc(IsVirtual,
20127da15baSAnders Carlsson                                   llvm::Type::getInt1Ty(VMContext));
20227da15baSAnders Carlsson 
20327da15baSAnders Carlsson   llvm::BasicBlock *FnVirtual = createBasicBlock("fn.virtual");
20427da15baSAnders Carlsson   llvm::BasicBlock *FnNonVirtual = createBasicBlock("fn.nonvirtual");
20527da15baSAnders Carlsson   llvm::BasicBlock *FnEnd = createBasicBlock("fn.end");
20627da15baSAnders Carlsson 
20727da15baSAnders Carlsson   Builder.CreateCondBr(IsVirtual, FnVirtual, FnNonVirtual);
20827da15baSAnders Carlsson   EmitBlock(FnVirtual);
20927da15baSAnders Carlsson 
21011e5140dSAnders Carlsson   const llvm::Type *VTableTy =
211c4325486SAnders Carlsson     FTy->getPointerTo()->getPointerTo();
21227da15baSAnders Carlsson 
21311e5140dSAnders Carlsson   llvm::Value *VTable = Builder.CreateBitCast(This, VTableTy->getPointerTo());
21411e5140dSAnders Carlsson   VTable = Builder.CreateLoad(VTable);
21527da15baSAnders Carlsson 
21611e5140dSAnders Carlsson   VTable = Builder.CreateBitCast(VTable, Int8PtrTy);
21711e5140dSAnders Carlsson   llvm::Value *VTableOffset =
218c4325486SAnders Carlsson     Builder.CreateSub(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1));
21927da15baSAnders Carlsson 
22011e5140dSAnders Carlsson   VTable = Builder.CreateGEP(VTable, VTableOffset, "fn");
22111e5140dSAnders Carlsson   VTable = Builder.CreateBitCast(VTable, VTableTy);
22227da15baSAnders Carlsson 
22311e5140dSAnders Carlsson   llvm::Value *VirtualFn = Builder.CreateLoad(VTable, "virtualfn");
22427da15baSAnders Carlsson 
22527da15baSAnders Carlsson   EmitBranch(FnEnd);
22627da15baSAnders Carlsson   EmitBlock(FnNonVirtual);
22727da15baSAnders Carlsson 
22827da15baSAnders Carlsson   // If the function is not virtual, just load the pointer.
22927da15baSAnders Carlsson   llvm::Value *NonVirtualFn = Builder.CreateLoad(FnPtr, "fn");
23027da15baSAnders Carlsson   NonVirtualFn = Builder.CreateIntToPtr(NonVirtualFn, FTy->getPointerTo());
23127da15baSAnders Carlsson 
23227da15baSAnders Carlsson   EmitBlock(FnEnd);
23327da15baSAnders Carlsson 
23427da15baSAnders Carlsson   llvm::PHINode *Callee = Builder.CreatePHI(FTy->getPointerTo());
23527da15baSAnders Carlsson   Callee->reserveOperandSpace(2);
23627da15baSAnders Carlsson   Callee->addIncoming(VirtualFn, FnVirtual);
23727da15baSAnders Carlsson   Callee->addIncoming(NonVirtualFn, FnNonVirtual);
23827da15baSAnders Carlsson 
23927da15baSAnders Carlsson   CallArgList Args;
24027da15baSAnders Carlsson 
24127da15baSAnders Carlsson   QualType ThisType =
24227da15baSAnders Carlsson     getContext().getPointerType(getContext().getTagDeclType(RD));
24327da15baSAnders Carlsson 
24427da15baSAnders Carlsson   // Push the this ptr.
24527da15baSAnders Carlsson   Args.push_back(std::make_pair(RValue::get(This), ThisType));
24627da15baSAnders Carlsson 
24727da15baSAnders Carlsson   // And the rest of the call args
24827da15baSAnders Carlsson   EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
249ab26cfa5SJohn McCall   const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>();
250ab26cfa5SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee,
25127da15baSAnders Carlsson                   ReturnValue, Args);
25227da15baSAnders Carlsson }
25327da15baSAnders Carlsson 
25427da15baSAnders Carlsson RValue
25527da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
25627da15baSAnders Carlsson                                                const CXXMethodDecl *MD,
25727da15baSAnders Carlsson                                                ReturnValueSlot ReturnValue) {
25827da15baSAnders Carlsson   assert(MD->isInstance() &&
25927da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
26027da15baSAnders Carlsson   if (MD->isCopyAssignment()) {
26127da15baSAnders Carlsson     const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext());
26227da15baSAnders Carlsson     if (ClassDecl->hasTrivialCopyAssignment()) {
26327da15baSAnders Carlsson       assert(!ClassDecl->hasUserDeclaredCopyAssignment() &&
26427da15baSAnders Carlsson              "EmitCXXOperatorMemberCallExpr - user declared copy assignment");
26543a40f93SFariborz Jahanian       LValue LV = EmitLValue(E->getArg(0));
26643a40f93SFariborz Jahanian       llvm::Value *This;
26743a40f93SFariborz Jahanian       if (LV.isPropertyRef()) {
2681b8b8bf2SFariborz Jahanian         llvm::Value *AggLoc  = CreateMemTemp(E->getArg(1)->getType());
269e1b45a5eSFariborz Jahanian         EmitAggExpr(E->getArg(1), AggLoc, false /*VolatileDest*/);
270e1b45a5eSFariborz Jahanian         EmitObjCPropertySet(LV.getPropertyRefExpr(),
271e1b45a5eSFariborz Jahanian                             RValue::getAggregate(AggLoc, false /*VolatileDest*/));
272e1b45a5eSFariborz Jahanian         return RValue::getAggregate(0, false);
27343a40f93SFariborz Jahanian       }
27443a40f93SFariborz Jahanian       else
27543a40f93SFariborz Jahanian         This = LV.getAddress();
27643a40f93SFariborz Jahanian 
27727da15baSAnders Carlsson       llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
27827da15baSAnders Carlsson       QualType Ty = E->getType();
27927da15baSAnders Carlsson       EmitAggregateCopy(This, Src, Ty);
28027da15baSAnders Carlsson       return RValue::get(This);
28127da15baSAnders Carlsson     }
28227da15baSAnders Carlsson   }
28327da15baSAnders Carlsson 
28427da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
28527da15baSAnders Carlsson   const llvm::Type *Ty =
28627da15baSAnders Carlsson     CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
28727da15baSAnders Carlsson                                    FPT->isVariadic());
288fdf474b0SFariborz Jahanian   LValue LV = EmitLValue(E->getArg(0));
289fdf474b0SFariborz Jahanian   llvm::Value *This;
290fdf474b0SFariborz Jahanian   if (LV.isPropertyRef()) {
2916855ba2cSFariborz Jahanian     RValue RV = EmitLoadOfPropertyRefLValue(LV, E->getArg(0)->getType());
2926855ba2cSFariborz Jahanian     assert (!RV.isScalar() && "EmitCXXOperatorMemberCallExpr");
2936855ba2cSFariborz Jahanian     This = RV.getAggregateAddr();
294fdf474b0SFariborz Jahanian   }
295fdf474b0SFariborz Jahanian   else
296fdf474b0SFariborz Jahanian     This = LV.getAddress();
29727da15baSAnders Carlsson 
29827da15baSAnders Carlsson   llvm::Value *Callee;
29927da15baSAnders Carlsson   if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0)))
30027da15baSAnders Carlsson     Callee = BuildVirtualCall(MD, This, Ty);
30127da15baSAnders Carlsson   else
30227da15baSAnders Carlsson     Callee = CGM.GetAddrOfFunction(MD, Ty);
30327da15baSAnders Carlsson 
304e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
30527da15baSAnders Carlsson                            E->arg_begin() + 1, E->arg_end());
30627da15baSAnders Carlsson }
30727da15baSAnders Carlsson 
30827da15baSAnders Carlsson void
30927da15baSAnders Carlsson CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest,
31027da15baSAnders Carlsson                                       const CXXConstructExpr *E) {
31127da15baSAnders Carlsson   assert(Dest && "Must have a destination!");
31227da15baSAnders Carlsson   const CXXConstructorDecl *CD = E->getConstructor();
31327da15baSAnders Carlsson   const ConstantArrayType *Array =
31427da15baSAnders Carlsson   getContext().getAsConstantArrayType(E->getType());
31527da15baSAnders Carlsson   // For a copy constructor, even if it is trivial, must fall thru so
31627da15baSAnders Carlsson   // its argument is code-gen'ed.
31727da15baSAnders Carlsson   if (!CD->isCopyConstructor()) {
31827da15baSAnders Carlsson     QualType InitType = E->getType();
31927da15baSAnders Carlsson     if (Array)
32027da15baSAnders Carlsson       InitType = getContext().getBaseElementType(Array);
32127da15baSAnders Carlsson     const CXXRecordDecl *RD =
32227da15baSAnders Carlsson     cast<CXXRecordDecl>(InitType->getAs<RecordType>()->getDecl());
323*e3b3464dSDouglas Gregor     if (RD->hasTrivialConstructor()) {
324*e3b3464dSDouglas Gregor       // The constructor is trivial, but we may still need to zero-initialize
325*e3b3464dSDouglas Gregor       // the class.
326*e3b3464dSDouglas Gregor       if (E->requiresZeroInitialization())
327*e3b3464dSDouglas Gregor         EmitNullInitialization(Dest, E->getType());
328*e3b3464dSDouglas Gregor 
32927da15baSAnders Carlsson       return;
33027da15baSAnders Carlsson     }
331*e3b3464dSDouglas Gregor   }
33227da15baSAnders Carlsson   // Code gen optimization to eliminate copy constructor and return
333222cf0efSDouglas Gregor   // its first argument instead, if in fact that argument is a temporary
334222cf0efSDouglas Gregor   // object.
33527da15baSAnders Carlsson   if (getContext().getLangOptions().ElideConstructors && E->isElidable()) {
336222cf0efSDouglas Gregor     if (const Expr *Arg = E->getArg(0)->getTemporaryObject()) {
33727da15baSAnders Carlsson       EmitAggExpr(Arg, Dest, false);
33827da15baSAnders Carlsson       return;
33927da15baSAnders Carlsson     }
340222cf0efSDouglas Gregor   }
34127da15baSAnders Carlsson   if (Array) {
34227da15baSAnders Carlsson     QualType BaseElementTy = getContext().getBaseElementType(Array);
34327da15baSAnders Carlsson     const llvm::Type *BasePtr = ConvertType(BaseElementTy);
34427da15baSAnders Carlsson     BasePtr = llvm::PointerType::getUnqual(BasePtr);
34527da15baSAnders Carlsson     llvm::Value *BaseAddrPtr =
34627da15baSAnders Carlsson       Builder.CreateBitCast(Dest, BasePtr);
34727da15baSAnders Carlsson 
34827da15baSAnders Carlsson     EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr,
34927da15baSAnders Carlsson                                E->arg_begin(), E->arg_end());
35027da15baSAnders Carlsson   }
351e11f9ce9SAnders Carlsson   else {
352e11f9ce9SAnders Carlsson     CXXCtorType Type =
353e11f9ce9SAnders Carlsson       (E->getConstructionKind() == CXXConstructExpr::CK_Complete)
354e11f9ce9SAnders Carlsson       ? Ctor_Complete : Ctor_Base;
355e11f9ce9SAnders Carlsson     bool ForVirtualBase =
356e11f9ce9SAnders Carlsson       E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase;
357e11f9ce9SAnders Carlsson 
35827da15baSAnders Carlsson     // Call the constructor.
359e11f9ce9SAnders Carlsson     EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest,
36027da15baSAnders Carlsson                            E->arg_begin(), E->arg_end());
36127da15baSAnders Carlsson   }
362e11f9ce9SAnders Carlsson }
36327da15baSAnders Carlsson 
3643eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, QualType ElementType) {
36521122cf6SAnders Carlsson   const RecordType *RT = ElementType->getAs<RecordType>();
36659486a2dSAnders Carlsson   if (!RT)
3673eb55cfeSKen Dyck     return CharUnits::Zero();
36859486a2dSAnders Carlsson 
36959486a2dSAnders Carlsson   const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl());
37059486a2dSAnders Carlsson   if (!RD)
3713eb55cfeSKen Dyck     return CharUnits::Zero();
37259486a2dSAnders Carlsson 
37359486a2dSAnders Carlsson   // Check if the class has a trivial destructor.
37459486a2dSAnders Carlsson   if (RD->hasTrivialDestructor()) {
37521122cf6SAnders Carlsson     // Check if the usual deallocation function takes two arguments.
376adbe4249SAnders Carlsson     const CXXMethodDecl *UsualDeallocationFunction = 0;
377adbe4249SAnders Carlsson 
37821122cf6SAnders Carlsson     DeclarationName OpName =
37921122cf6SAnders Carlsson       Ctx.DeclarationNames.getCXXOperatorName(OO_Array_Delete);
38021122cf6SAnders Carlsson     DeclContext::lookup_const_iterator Op, OpEnd;
38121122cf6SAnders Carlsson     for (llvm::tie(Op, OpEnd) = RD->lookup(OpName);
38221122cf6SAnders Carlsson          Op != OpEnd; ++Op) {
383adbe4249SAnders Carlsson       const CXXMethodDecl *Delete = cast<CXXMethodDecl>(*Op);
38421122cf6SAnders Carlsson 
38521122cf6SAnders Carlsson       if (Delete->isUsualDeallocationFunction()) {
386adbe4249SAnders Carlsson         UsualDeallocationFunction = Delete;
38721122cf6SAnders Carlsson         break;
38821122cf6SAnders Carlsson       }
38921122cf6SAnders Carlsson     }
390adbe4249SAnders Carlsson 
391adbe4249SAnders Carlsson     // No usual deallocation function, we don't need a cookie.
392adbe4249SAnders Carlsson     if (!UsualDeallocationFunction)
3933eb55cfeSKen Dyck       return CharUnits::Zero();
394adbe4249SAnders Carlsson 
395adbe4249SAnders Carlsson     // The usual deallocation function doesn't take a size_t argument, so we
396adbe4249SAnders Carlsson     // don't need a cookie.
397adbe4249SAnders Carlsson     if (UsualDeallocationFunction->getNumParams() == 1)
3983eb55cfeSKen Dyck       return CharUnits::Zero();
399adbe4249SAnders Carlsson 
400adbe4249SAnders Carlsson     assert(UsualDeallocationFunction->getNumParams() == 2 &&
401adbe4249SAnders Carlsson            "Unexpected deallocation function type!");
40259486a2dSAnders Carlsson   }
40359486a2dSAnders Carlsson 
40421122cf6SAnders Carlsson   // Padding is the maximum of sizeof(size_t) and alignof(ElementType)
4053eb55cfeSKen Dyck   return std::max(Ctx.getTypeSizeInChars(Ctx.getSizeType()),
4063eb55cfeSKen Dyck                   Ctx.getTypeAlignInChars(ElementType));
40721122cf6SAnders Carlsson }
40821122cf6SAnders Carlsson 
4093eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, const CXXNewExpr *E) {
41021122cf6SAnders Carlsson   if (!E->isArray())
4113eb55cfeSKen Dyck     return CharUnits::Zero();
41221122cf6SAnders Carlsson 
413399f499fSAnders Carlsson   // No cookie is required if the new operator being used is
414399f499fSAnders Carlsson   // ::operator new[](size_t, void*).
415399f499fSAnders Carlsson   const FunctionDecl *OperatorNew = E->getOperatorNew();
416399f499fSAnders Carlsson   if (OperatorNew->getDeclContext()->getLookupContext()->isFileContext()) {
417399f499fSAnders Carlsson     if (OperatorNew->getNumParams() == 2) {
418399f499fSAnders Carlsson       CanQualType ParamType =
419399f499fSAnders Carlsson         Ctx.getCanonicalType(OperatorNew->getParamDecl(1)->getType());
420399f499fSAnders Carlsson 
421399f499fSAnders Carlsson       if (ParamType == Ctx.VoidPtrTy)
4223eb55cfeSKen Dyck         return CharUnits::Zero();
423399f499fSAnders Carlsson     }
424399f499fSAnders Carlsson   }
425399f499fSAnders Carlsson 
42621122cf6SAnders Carlsson   return CalculateCookiePadding(Ctx, E->getAllocatedType());
42759486a2dSAnders Carlsson }
42859486a2dSAnders Carlsson 
42947b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context,
43047b4629bSFariborz Jahanian                                         CodeGenFunction &CGF,
43159486a2dSAnders Carlsson                                         const CXXNewExpr *E,
43205fc5be3SDouglas Gregor                                         llvm::Value *&NumElements,
43305fc5be3SDouglas Gregor                                         llvm::Value *&SizeWithoutCookie) {
43459486a2dSAnders Carlsson   QualType Type = E->getAllocatedType();
4353eb55cfeSKen Dyck   CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(Type);
43659486a2dSAnders Carlsson   const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType());
43759486a2dSAnders Carlsson 
43805fc5be3SDouglas Gregor   if (!E->isArray()) {
43905fc5be3SDouglas Gregor     SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
44005fc5be3SDouglas Gregor     return SizeWithoutCookie;
44105fc5be3SDouglas Gregor   }
44259486a2dSAnders Carlsson 
44359486a2dSAnders Carlsson   // Emit the array size expression.
44459486a2dSAnders Carlsson   NumElements = CGF.EmitScalarExpr(E->getArraySize());
44559486a2dSAnders Carlsson 
44632ac583dSChris Lattner   llvm::Value *Size = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
44732ac583dSChris Lattner 
44832ac583dSChris Lattner   // If someone is doing 'new int[42]' there is no need to do a dynamic check.
44932ac583dSChris Lattner   // Don't bloat the -O0 code.
45032ac583dSChris Lattner   if (llvm::ConstantInt *NumElementsC =
45132ac583dSChris Lattner         dyn_cast<llvm::ConstantInt>(NumElements)) {
45232ac583dSChris Lattner     // Determine if there is an overflow here by doing an extended multiply.
45332ac583dSChris Lattner     llvm::APInt NEC = NumElementsC->getValue();
45432ac583dSChris Lattner     NEC.zext(NEC.getBitWidth()*2);
45532ac583dSChris Lattner 
45632ac583dSChris Lattner     llvm::APInt SC = cast<llvm::ConstantInt>(Size)->getValue();
45732ac583dSChris Lattner     SC.zext(SC.getBitWidth()*2);
45832ac583dSChris Lattner     SC *= NEC;
45932ac583dSChris Lattner 
46032ac583dSChris Lattner     if (SC.countLeadingZeros() >= NumElementsC->getValue().getBitWidth()) {
46132ac583dSChris Lattner       SC.trunc(NumElementsC->getValue().getBitWidth());
46232ac583dSChris Lattner       Size = llvm::ConstantInt::get(Size->getContext(), SC);
46332ac583dSChris Lattner     } else {
46432ac583dSChris Lattner       // On overflow, produce a -1 so operator new throws.
46532ac583dSChris Lattner       Size = llvm::Constant::getAllOnesValue(Size->getType());
46632ac583dSChris Lattner     }
46732ac583dSChris Lattner 
46832ac583dSChris Lattner   } else {
46926008e07SChris Lattner     // Multiply with the type size.  This multiply can overflow, e.g. in:
47026008e07SChris Lattner     //   new double[n]
47126008e07SChris Lattner     // where n is 2^30 on a 32-bit machine or 2^62 on a 64-bit machine.  Because
47226008e07SChris Lattner     // of this, we need to detect the overflow and ensure that an exception is
473f2f38701SChris Lattner     // called by forcing the size to -1 on overflow.
47432ac583dSChris Lattner     llvm::Value *UMulF =
47532ac583dSChris Lattner       CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, &SizeTy, 1);
47632ac583dSChris Lattner     llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumElements, Size);
47732ac583dSChris Lattner     // Branch on the overflow bit to the overflow block, which is lazily
47832ac583dSChris Lattner     // created.
47926008e07SChris Lattner     llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1);
48026008e07SChris Lattner     // Get the normal result of the multiplication.
48126008e07SChris Lattner     llvm::Value *V = CGF.Builder.CreateExtractValue(MulRes, 0);
48247b4629bSFariborz Jahanian 
483f2f38701SChris Lattner     llvm::BasicBlock *NormalBB = CGF.createBasicBlock("no_overflow");
484f2f38701SChris Lattner     llvm::BasicBlock *OverflowBB = CGF.createBasicBlock("overflow");
485f2f38701SChris Lattner 
486f2f38701SChris Lattner     CGF.Builder.CreateCondBr(DidOverflow, OverflowBB, NormalBB);
487f2f38701SChris Lattner 
488f2f38701SChris Lattner     llvm::BasicBlock *PrevBB = CGF.Builder.GetInsertBlock();
489f2f38701SChris Lattner 
490f2f38701SChris Lattner     // We just need the overflow block to build a PHI node.
491f2f38701SChris Lattner     CGF.EmitBlock(OverflowBB);
492f2f38701SChris Lattner     CGF.EmitBlock(NormalBB);
493f2f38701SChris Lattner 
494f2f38701SChris Lattner     llvm::PHINode *PN = CGF.Builder.CreatePHI(V->getType());
495f2f38701SChris Lattner 
496f2f38701SChris Lattner     PN->addIncoming(V, PrevBB);
497f2f38701SChris Lattner     PN->addIncoming(llvm::Constant::getAllOnesValue(V->getType()), OverflowBB);
49832ac583dSChris Lattner     Size = PN;
49932ac583dSChris Lattner   }
50005fc5be3SDouglas Gregor   SizeWithoutCookie = Size;
501f2f38701SChris Lattner 
50232ac583dSChris Lattner   // Add the cookie padding if necessary.
50332ac583dSChris Lattner   CharUnits CookiePadding = CalculateCookiePadding(CGF.getContext(), E);
5043eb55cfeSKen Dyck   if (!CookiePadding.isZero())
50532ac583dSChris Lattner     Size = CGF.Builder.CreateAdd(Size,
5063eb55cfeSKen Dyck         llvm::ConstantInt::get(SizeTy, CookiePadding.getQuantity()));
50759486a2dSAnders Carlsson 
50832ac583dSChris Lattner   return Size;
50959486a2dSAnders Carlsson }
51059486a2dSAnders Carlsson 
511d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E,
512d5202e09SFariborz Jahanian                                     llvm::Value *NewPtr) {
513d5202e09SFariborz Jahanian 
514d5202e09SFariborz Jahanian   assert(E->getNumConstructorArgs() == 1 &&
515d5202e09SFariborz Jahanian          "Can only have one argument to initializer of POD type.");
516d5202e09SFariborz Jahanian 
517d5202e09SFariborz Jahanian   const Expr *Init = E->getConstructorArg(0);
518d5202e09SFariborz Jahanian   QualType AllocType = E->getAllocatedType();
519d5202e09SFariborz Jahanian 
520d5202e09SFariborz Jahanian   if (!CGF.hasAggregateLLVMType(AllocType))
521d5202e09SFariborz Jahanian     CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr,
522d5202e09SFariborz Jahanian                           AllocType.isVolatileQualified(), AllocType);
523d5202e09SFariborz Jahanian   else if (AllocType->isAnyComplexType())
524d5202e09SFariborz Jahanian     CGF.EmitComplexExprIntoAddr(Init, NewPtr,
525d5202e09SFariborz Jahanian                                 AllocType.isVolatileQualified());
526d5202e09SFariborz Jahanian   else
527d5202e09SFariborz Jahanian     CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified());
528d5202e09SFariborz Jahanian }
529d5202e09SFariborz Jahanian 
530d5202e09SFariborz Jahanian void
531d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E,
532d5202e09SFariborz Jahanian                                          llvm::Value *NewPtr,
533d5202e09SFariborz Jahanian                                          llvm::Value *NumElements) {
534b66b08efSFariborz Jahanian   // We have a POD type.
535b66b08efSFariborz Jahanian   if (E->getNumConstructorArgs() == 0)
536b66b08efSFariborz Jahanian     return;
537b66b08efSFariborz Jahanian 
538d5202e09SFariborz Jahanian   const llvm::Type *SizeTy = ConvertType(getContext().getSizeType());
539d5202e09SFariborz Jahanian 
540d5202e09SFariborz Jahanian   // Create a temporary for the loop index and initialize it with 0.
541d5202e09SFariborz Jahanian   llvm::Value *IndexPtr = CreateTempAlloca(SizeTy, "loop.index");
542d5202e09SFariborz Jahanian   llvm::Value *Zero = llvm::Constant::getNullValue(SizeTy);
543d5202e09SFariborz Jahanian   Builder.CreateStore(Zero, IndexPtr);
544d5202e09SFariborz Jahanian 
545d5202e09SFariborz Jahanian   // Start the loop with a block that tests the condition.
546d5202e09SFariborz Jahanian   llvm::BasicBlock *CondBlock = createBasicBlock("for.cond");
547d5202e09SFariborz Jahanian   llvm::BasicBlock *AfterFor = createBasicBlock("for.end");
548d5202e09SFariborz Jahanian 
549d5202e09SFariborz Jahanian   EmitBlock(CondBlock);
550d5202e09SFariborz Jahanian 
551d5202e09SFariborz Jahanian   llvm::BasicBlock *ForBody = createBasicBlock("for.body");
552d5202e09SFariborz Jahanian 
553d5202e09SFariborz Jahanian   // Generate: if (loop-index < number-of-elements fall to the loop body,
554d5202e09SFariborz Jahanian   // otherwise, go to the block after the for-loop.
555d5202e09SFariborz Jahanian   llvm::Value *Counter = Builder.CreateLoad(IndexPtr);
556d5202e09SFariborz Jahanian   llvm::Value *IsLess = Builder.CreateICmpULT(Counter, NumElements, "isless");
557d5202e09SFariborz Jahanian   // If the condition is true, execute the body.
558d5202e09SFariborz Jahanian   Builder.CreateCondBr(IsLess, ForBody, AfterFor);
559d5202e09SFariborz Jahanian 
560d5202e09SFariborz Jahanian   EmitBlock(ForBody);
561d5202e09SFariborz Jahanian 
562d5202e09SFariborz Jahanian   llvm::BasicBlock *ContinueBlock = createBasicBlock("for.inc");
563d5202e09SFariborz Jahanian   // Inside the loop body, emit the constructor call on the array element.
564d5202e09SFariborz Jahanian   Counter = Builder.CreateLoad(IndexPtr);
565d5202e09SFariborz Jahanian   llvm::Value *Address = Builder.CreateInBoundsGEP(NewPtr, Counter,
566d5202e09SFariborz Jahanian                                                    "arrayidx");
567d5202e09SFariborz Jahanian   StoreAnyExprIntoOneUnit(*this, E, Address);
568d5202e09SFariborz Jahanian 
569d5202e09SFariborz Jahanian   EmitBlock(ContinueBlock);
570d5202e09SFariborz Jahanian 
571d5202e09SFariborz Jahanian   // Emit the increment of the loop counter.
572d5202e09SFariborz Jahanian   llvm::Value *NextVal = llvm::ConstantInt::get(SizeTy, 1);
573d5202e09SFariborz Jahanian   Counter = Builder.CreateLoad(IndexPtr);
574d5202e09SFariborz Jahanian   NextVal = Builder.CreateAdd(Counter, NextVal, "inc");
575d5202e09SFariborz Jahanian   Builder.CreateStore(NextVal, IndexPtr);
576d5202e09SFariborz Jahanian 
577d5202e09SFariborz Jahanian   // Finally, branch back up to the condition for the next iteration.
578d5202e09SFariborz Jahanian   EmitBranch(CondBlock);
579d5202e09SFariborz Jahanian 
580d5202e09SFariborz Jahanian   // Emit the fall-through block.
581d5202e09SFariborz Jahanian   EmitBlock(AfterFor, true);
582d5202e09SFariborz Jahanian }
583d5202e09SFariborz Jahanian 
58405fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T,
58505fc5be3SDouglas Gregor                            llvm::Value *NewPtr, llvm::Value *Size) {
58605fc5be3SDouglas Gregor   llvm::LLVMContext &VMContext = CGF.CGM.getLLVMContext();
58705fc5be3SDouglas Gregor   const llvm::Type *BP = llvm::Type::getInt8PtrTy(VMContext);
58805fc5be3SDouglas Gregor   if (NewPtr->getType() != BP)
58905fc5be3SDouglas Gregor     NewPtr = CGF.Builder.CreateBitCast(NewPtr, BP, "tmp");
59005fc5be3SDouglas Gregor 
59105fc5be3SDouglas Gregor   CGF.Builder.CreateCall5(CGF.CGM.getMemSetFn(BP, CGF.IntPtrTy), NewPtr,
59205fc5be3SDouglas Gregor                 llvm::Constant::getNullValue(llvm::Type::getInt8Ty(VMContext)),
59305fc5be3SDouglas Gregor                           Size,
59405fc5be3SDouglas Gregor                     llvm::ConstantInt::get(CGF.Int32Ty,
59505fc5be3SDouglas Gregor                                            CGF.getContext().getTypeAlign(T)/8),
59605fc5be3SDouglas Gregor                           llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext),
59705fc5be3SDouglas Gregor                                                  0));
59805fc5be3SDouglas Gregor }
59905fc5be3SDouglas Gregor 
60059486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
60159486a2dSAnders Carlsson                                llvm::Value *NewPtr,
60205fc5be3SDouglas Gregor                                llvm::Value *NumElements,
60305fc5be3SDouglas Gregor                                llvm::Value *AllocSizeWithoutCookie) {
6043a202f60SAnders Carlsson   if (E->isArray()) {
605d040e6b2SAnders Carlsson     if (CXXConstructorDecl *Ctor = E->getConstructor()) {
60605fc5be3SDouglas Gregor       bool RequiresZeroInitialization = false;
60705fc5be3SDouglas Gregor       if (Ctor->getParent()->hasTrivialConstructor()) {
60805fc5be3SDouglas Gregor         // If new expression did not specify value-initialization, then there
60905fc5be3SDouglas Gregor         // is no initialization.
61005fc5be3SDouglas Gregor         if (!E->hasInitializer() || Ctor->getParent()->isEmpty())
61105fc5be3SDouglas Gregor           return;
61205fc5be3SDouglas Gregor 
61305fc5be3SDouglas Gregor         if (!CGF.CGM.getTypes().ContainsPointerToDataMember(
61405fc5be3SDouglas Gregor                                                        E->getAllocatedType())) {
61505fc5be3SDouglas Gregor           // Optimization: since zero initialization will just set the memory
61605fc5be3SDouglas Gregor           // to all zeroes, generate a single memset to do it in one shot.
61705fc5be3SDouglas Gregor           EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr,
61805fc5be3SDouglas Gregor                          AllocSizeWithoutCookie);
6193a202f60SAnders Carlsson           return;
6203a202f60SAnders Carlsson         }
62105fc5be3SDouglas Gregor 
62205fc5be3SDouglas Gregor         RequiresZeroInitialization = true;
62305fc5be3SDouglas Gregor       }
62405fc5be3SDouglas Gregor 
62505fc5be3SDouglas Gregor       CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr,
62605fc5be3SDouglas Gregor                                      E->constructor_arg_begin(),
62705fc5be3SDouglas Gregor                                      E->constructor_arg_end(),
62805fc5be3SDouglas Gregor                                      RequiresZeroInitialization);
62905fc5be3SDouglas Gregor       return;
63005fc5be3SDouglas Gregor     } else if (E->getNumConstructorArgs() == 1 &&
63105fc5be3SDouglas Gregor                isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) {
63205fc5be3SDouglas Gregor       // Optimization: since zero initialization will just set the memory
63305fc5be3SDouglas Gregor       // to all zeroes, generate a single memset to do it in one shot.
63405fc5be3SDouglas Gregor       EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr,
63505fc5be3SDouglas Gregor                      AllocSizeWithoutCookie);
63605fc5be3SDouglas Gregor       return;
63705fc5be3SDouglas Gregor     } else {
638d5202e09SFariborz Jahanian       CGF.EmitNewArrayInitializer(E, NewPtr, NumElements);
639d5202e09SFariborz Jahanian       return;
640d040e6b2SAnders Carlsson     }
641d5202e09SFariborz Jahanian   }
64259486a2dSAnders Carlsson 
64359486a2dSAnders Carlsson   if (CXXConstructorDecl *Ctor = E->getConstructor()) {
644747eb784SDouglas Gregor     // Per C++ [expr.new]p15, if we have an initializer, then we're performing
645747eb784SDouglas Gregor     // direct initialization. C++ [dcl.init]p5 requires that we
646747eb784SDouglas Gregor     // zero-initialize storage if there are no user-declared constructors.
647747eb784SDouglas Gregor     if (E->hasInitializer() &&
648747eb784SDouglas Gregor         !Ctor->getParent()->hasUserDeclaredConstructor() &&
649747eb784SDouglas Gregor         !Ctor->getParent()->isEmpty())
650747eb784SDouglas Gregor       CGF.EmitNullInitialization(NewPtr, E->getAllocatedType());
651747eb784SDouglas Gregor 
652e11f9ce9SAnders Carlsson     CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false,
653e11f9ce9SAnders Carlsson                                NewPtr, E->constructor_arg_begin(),
65459486a2dSAnders Carlsson                                E->constructor_arg_end());
65559486a2dSAnders Carlsson 
65659486a2dSAnders Carlsson     return;
65759486a2dSAnders Carlsson   }
658b66b08efSFariborz Jahanian   // We have a POD type.
659b66b08efSFariborz Jahanian   if (E->getNumConstructorArgs() == 0)
660b66b08efSFariborz Jahanian     return;
66159486a2dSAnders Carlsson 
662d5202e09SFariborz Jahanian   StoreAnyExprIntoOneUnit(CGF, E, NewPtr);
66359486a2dSAnders Carlsson }
66459486a2dSAnders Carlsson 
66559486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
66659486a2dSAnders Carlsson   QualType AllocType = E->getAllocatedType();
66759486a2dSAnders Carlsson   FunctionDecl *NewFD = E->getOperatorNew();
66859486a2dSAnders Carlsson   const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>();
66959486a2dSAnders Carlsson 
67059486a2dSAnders Carlsson   CallArgList NewArgs;
67159486a2dSAnders Carlsson 
67259486a2dSAnders Carlsson   // The allocation size is the first argument.
67359486a2dSAnders Carlsson   QualType SizeTy = getContext().getSizeType();
67459486a2dSAnders Carlsson 
67559486a2dSAnders Carlsson   llvm::Value *NumElements = 0;
67605fc5be3SDouglas Gregor   llvm::Value *AllocSizeWithoutCookie = 0;
67747b4629bSFariborz Jahanian   llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(),
67805fc5be3SDouglas Gregor                                                *this, E, NumElements,
67905fc5be3SDouglas Gregor                                                AllocSizeWithoutCookie);
68059486a2dSAnders Carlsson 
68159486a2dSAnders Carlsson   NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy));
68259486a2dSAnders Carlsson 
68359486a2dSAnders Carlsson   // Emit the rest of the arguments.
68459486a2dSAnders Carlsson   // FIXME: Ideally, this should just use EmitCallArgs.
68559486a2dSAnders Carlsson   CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin();
68659486a2dSAnders Carlsson 
68759486a2dSAnders Carlsson   // First, use the types from the function type.
68859486a2dSAnders Carlsson   // We start at 1 here because the first argument (the allocation size)
68959486a2dSAnders Carlsson   // has already been emitted.
69059486a2dSAnders Carlsson   for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) {
69159486a2dSAnders Carlsson     QualType ArgType = NewFTy->getArgType(i);
69259486a2dSAnders Carlsson 
69359486a2dSAnders Carlsson     assert(getContext().getCanonicalType(ArgType.getNonReferenceType()).
69459486a2dSAnders Carlsson            getTypePtr() ==
69559486a2dSAnders Carlsson            getContext().getCanonicalType(NewArg->getType()).getTypePtr() &&
69659486a2dSAnders Carlsson            "type mismatch in call argument!");
69759486a2dSAnders Carlsson 
69859486a2dSAnders Carlsson     NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
69959486a2dSAnders Carlsson                                      ArgType));
70059486a2dSAnders Carlsson 
70159486a2dSAnders Carlsson   }
70259486a2dSAnders Carlsson 
70359486a2dSAnders Carlsson   // Either we've emitted all the call args, or we have a call to a
70459486a2dSAnders Carlsson   // variadic function.
70559486a2dSAnders Carlsson   assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) &&
70659486a2dSAnders Carlsson          "Extra arguments in non-variadic function!");
70759486a2dSAnders Carlsson 
70859486a2dSAnders Carlsson   // If we still have any arguments, emit them using the type of the argument.
70959486a2dSAnders Carlsson   for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end();
71059486a2dSAnders Carlsson        NewArg != NewArgEnd; ++NewArg) {
71159486a2dSAnders Carlsson     QualType ArgType = NewArg->getType();
71259486a2dSAnders Carlsson     NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
71359486a2dSAnders Carlsson                                      ArgType));
71459486a2dSAnders Carlsson   }
71559486a2dSAnders Carlsson 
71659486a2dSAnders Carlsson   // Emit the call to new.
71759486a2dSAnders Carlsson   RValue RV =
718ab26cfa5SJohn McCall     EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy),
71961a401caSAnders Carlsson              CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD);
72059486a2dSAnders Carlsson 
72159486a2dSAnders Carlsson   // If an allocation function is declared with an empty exception specification
72259486a2dSAnders Carlsson   // it returns null to indicate failure to allocate storage. [expr.new]p13.
72359486a2dSAnders Carlsson   // (We don't need to check for null when there's no new initializer and
72459486a2dSAnders Carlsson   // we're allocating a POD type).
72559486a2dSAnders Carlsson   bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() &&
72659486a2dSAnders Carlsson     !(AllocType->isPODType() && !E->hasInitializer());
72759486a2dSAnders Carlsson 
72859486a2dSAnders Carlsson   llvm::BasicBlock *NewNull = 0;
72959486a2dSAnders Carlsson   llvm::BasicBlock *NewNotNull = 0;
73059486a2dSAnders Carlsson   llvm::BasicBlock *NewEnd = 0;
73159486a2dSAnders Carlsson 
73259486a2dSAnders Carlsson   llvm::Value *NewPtr = RV.getScalarVal();
73359486a2dSAnders Carlsson 
73459486a2dSAnders Carlsson   if (NullCheckResult) {
73559486a2dSAnders Carlsson     NewNull = createBasicBlock("new.null");
73659486a2dSAnders Carlsson     NewNotNull = createBasicBlock("new.notnull");
73759486a2dSAnders Carlsson     NewEnd = createBasicBlock("new.end");
73859486a2dSAnders Carlsson 
73959486a2dSAnders Carlsson     llvm::Value *IsNull =
74059486a2dSAnders Carlsson       Builder.CreateICmpEQ(NewPtr,
74159486a2dSAnders Carlsson                            llvm::Constant::getNullValue(NewPtr->getType()),
74259486a2dSAnders Carlsson                            "isnull");
74359486a2dSAnders Carlsson 
74459486a2dSAnders Carlsson     Builder.CreateCondBr(IsNull, NewNull, NewNotNull);
74559486a2dSAnders Carlsson     EmitBlock(NewNotNull);
74659486a2dSAnders Carlsson   }
74759486a2dSAnders Carlsson 
7483eb55cfeSKen Dyck   CharUnits CookiePadding = CalculateCookiePadding(getContext(), E);
7493eb55cfeSKen Dyck   if (!CookiePadding.isZero()) {
7503eb55cfeSKen Dyck     CharUnits CookieOffset =
7513eb55cfeSKen Dyck       CookiePadding - getContext().getTypeSizeInChars(SizeTy);
75259486a2dSAnders Carlsson 
75359486a2dSAnders Carlsson     llvm::Value *NumElementsPtr =
7543eb55cfeSKen Dyck       Builder.CreateConstInBoundsGEP1_64(NewPtr, CookieOffset.getQuantity());
75559486a2dSAnders Carlsson 
75659486a2dSAnders Carlsson     NumElementsPtr = Builder.CreateBitCast(NumElementsPtr,
75759486a2dSAnders Carlsson                                            ConvertType(SizeTy)->getPointerTo());
75859486a2dSAnders Carlsson     Builder.CreateStore(NumElements, NumElementsPtr);
75959486a2dSAnders Carlsson 
76059486a2dSAnders Carlsson     // Now add the padding to the new ptr.
7613eb55cfeSKen Dyck     NewPtr = Builder.CreateConstInBoundsGEP1_64(NewPtr,
7623eb55cfeSKen Dyck                                                 CookiePadding.getQuantity());
76359486a2dSAnders Carlsson   }
76459486a2dSAnders Carlsson 
76547b4629bSFariborz Jahanian   if (AllocType->isArrayType()) {
76647b4629bSFariborz Jahanian     while (const ArrayType *AType = getContext().getAsArrayType(AllocType))
76747b4629bSFariborz Jahanian       AllocType = AType->getElementType();
76847b4629bSFariborz Jahanian     NewPtr =
76947b4629bSFariborz Jahanian       Builder.CreateBitCast(NewPtr,
77047b4629bSFariborz Jahanian                           ConvertType(getContext().getPointerType(AllocType)));
77105fc5be3SDouglas Gregor     EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie);
77247b4629bSFariborz Jahanian     NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType()));
77347b4629bSFariborz Jahanian   }
77447b4629bSFariborz Jahanian   else {
77547b4629bSFariborz Jahanian     NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType()));
77605fc5be3SDouglas Gregor     EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie);
77747b4629bSFariborz Jahanian   }
77859486a2dSAnders Carlsson 
77959486a2dSAnders Carlsson   if (NullCheckResult) {
78059486a2dSAnders Carlsson     Builder.CreateBr(NewEnd);
78159486a2dSAnders Carlsson     NewNotNull = Builder.GetInsertBlock();
78259486a2dSAnders Carlsson     EmitBlock(NewNull);
78359486a2dSAnders Carlsson     Builder.CreateBr(NewEnd);
78459486a2dSAnders Carlsson     EmitBlock(NewEnd);
78559486a2dSAnders Carlsson 
78659486a2dSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType());
78759486a2dSAnders Carlsson     PHI->reserveOperandSpace(2);
78859486a2dSAnders Carlsson     PHI->addIncoming(NewPtr, NewNotNull);
78959486a2dSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), NewNull);
79059486a2dSAnders Carlsson 
79159486a2dSAnders Carlsson     NewPtr = PHI;
79259486a2dSAnders Carlsson   }
79359486a2dSAnders Carlsson 
79459486a2dSAnders Carlsson   return NewPtr;
79559486a2dSAnders Carlsson }
79659486a2dSAnders Carlsson 
79721122cf6SAnders Carlsson static std::pair<llvm::Value *, llvm::Value *>
79821122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(CodeGenFunction &CGF,
79921122cf6SAnders Carlsson                                     llvm::Value *Ptr, QualType DeleteTy) {
80021122cf6SAnders Carlsson   QualType SizeTy = CGF.getContext().getSizeType();
80121122cf6SAnders Carlsson   const llvm::Type *SizeLTy = CGF.ConvertType(SizeTy);
80221122cf6SAnders Carlsson 
8033eb55cfeSKen Dyck   CharUnits DeleteTypeAlign = CGF.getContext().getTypeAlignInChars(DeleteTy);
8043eb55cfeSKen Dyck   CharUnits CookiePadding =
8053eb55cfeSKen Dyck     std::max(CGF.getContext().getTypeSizeInChars(SizeTy),
8063eb55cfeSKen Dyck              DeleteTypeAlign);
8073eb55cfeSKen Dyck   assert(!CookiePadding.isZero() && "CookiePadding should not be 0.");
80821122cf6SAnders Carlsson 
80921122cf6SAnders Carlsson   const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext());
8103eb55cfeSKen Dyck   CharUnits CookieOffset =
8113eb55cfeSKen Dyck     CookiePadding - CGF.getContext().getTypeSizeInChars(SizeTy);
81221122cf6SAnders Carlsson 
81321122cf6SAnders Carlsson   llvm::Value *AllocatedObjectPtr = CGF.Builder.CreateBitCast(Ptr, Int8PtrTy);
81421122cf6SAnders Carlsson   AllocatedObjectPtr =
81521122cf6SAnders Carlsson     CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr,
8163eb55cfeSKen Dyck                                            -CookiePadding.getQuantity());
81721122cf6SAnders Carlsson 
81821122cf6SAnders Carlsson   llvm::Value *NumElementsPtr =
81921122cf6SAnders Carlsson     CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr,
8203eb55cfeSKen Dyck                                            CookieOffset.getQuantity());
82121122cf6SAnders Carlsson   NumElementsPtr =
82221122cf6SAnders Carlsson     CGF.Builder.CreateBitCast(NumElementsPtr, SizeLTy->getPointerTo());
82321122cf6SAnders Carlsson 
82421122cf6SAnders Carlsson   llvm::Value *NumElements = CGF.Builder.CreateLoad(NumElementsPtr);
82521122cf6SAnders Carlsson   NumElements =
82621122cf6SAnders Carlsson     CGF.Builder.CreateIntCast(NumElements, SizeLTy, /*isSigned=*/false);
82721122cf6SAnders Carlsson 
82821122cf6SAnders Carlsson   return std::make_pair(AllocatedObjectPtr, NumElements);
82921122cf6SAnders Carlsson }
83021122cf6SAnders Carlsson 
83159486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
83259486a2dSAnders Carlsson                                      llvm::Value *Ptr,
83359486a2dSAnders Carlsson                                      QualType DeleteTy) {
83459486a2dSAnders Carlsson   const FunctionProtoType *DeleteFTy =
83559486a2dSAnders Carlsson     DeleteFD->getType()->getAs<FunctionProtoType>();
83659486a2dSAnders Carlsson 
83759486a2dSAnders Carlsson   CallArgList DeleteArgs;
83859486a2dSAnders Carlsson 
83921122cf6SAnders Carlsson   // Check if we need to pass the size to the delete operator.
84021122cf6SAnders Carlsson   llvm::Value *Size = 0;
84121122cf6SAnders Carlsson   QualType SizeTy;
84221122cf6SAnders Carlsson   if (DeleteFTy->getNumArgs() == 2) {
84321122cf6SAnders Carlsson     SizeTy = DeleteFTy->getArgType(1);
8447df3cbebSKen Dyck     CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
8457df3cbebSKen Dyck     Size = llvm::ConstantInt::get(ConvertType(SizeTy),
8467df3cbebSKen Dyck                                   DeleteTypeSize.getQuantity());
84721122cf6SAnders Carlsson   }
84821122cf6SAnders Carlsson 
84921122cf6SAnders Carlsson   if (DeleteFD->getOverloadedOperator() == OO_Array_Delete &&
8503eb55cfeSKen Dyck       !CalculateCookiePadding(getContext(), DeleteTy).isZero()) {
85121122cf6SAnders Carlsson     // We need to get the number of elements in the array from the cookie.
85221122cf6SAnders Carlsson     llvm::Value *AllocatedObjectPtr;
85321122cf6SAnders Carlsson     llvm::Value *NumElements;
85421122cf6SAnders Carlsson     llvm::tie(AllocatedObjectPtr, NumElements) =
85521122cf6SAnders Carlsson       GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy);
85621122cf6SAnders Carlsson 
85721122cf6SAnders Carlsson     // Multiply the size with the number of elements.
85821122cf6SAnders Carlsson     if (Size)
85921122cf6SAnders Carlsson       Size = Builder.CreateMul(NumElements, Size);
86021122cf6SAnders Carlsson 
86121122cf6SAnders Carlsson     Ptr = AllocatedObjectPtr;
86221122cf6SAnders Carlsson   }
86321122cf6SAnders Carlsson 
86459486a2dSAnders Carlsson   QualType ArgTy = DeleteFTy->getArgType(0);
86559486a2dSAnders Carlsson   llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
86659486a2dSAnders Carlsson   DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy));
86759486a2dSAnders Carlsson 
86821122cf6SAnders Carlsson   if (Size)
86959486a2dSAnders Carlsson     DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy));
87059486a2dSAnders Carlsson 
87159486a2dSAnders Carlsson   // Emit the call to delete.
872ab26cfa5SJohn McCall   EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy),
87361a401caSAnders Carlsson            CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(),
87459486a2dSAnders Carlsson            DeleteArgs, DeleteFD);
87559486a2dSAnders Carlsson }
87659486a2dSAnders Carlsson 
87759486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
87859486a2dSAnders Carlsson 
87959486a2dSAnders Carlsson   // Get at the argument before we performed the implicit conversion
88059486a2dSAnders Carlsson   // to void*.
88159486a2dSAnders Carlsson   const Expr *Arg = E->getArgument();
88259486a2dSAnders Carlsson   while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) {
88359486a2dSAnders Carlsson     if (ICE->getCastKind() != CastExpr::CK_UserDefinedConversion &&
88459486a2dSAnders Carlsson         ICE->getType()->isVoidPointerType())
88559486a2dSAnders Carlsson       Arg = ICE->getSubExpr();
88659486a2dSAnders Carlsson     else
88759486a2dSAnders Carlsson       break;
88859486a2dSAnders Carlsson   }
88959486a2dSAnders Carlsson 
89059486a2dSAnders Carlsson   QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
89159486a2dSAnders Carlsson 
89259486a2dSAnders Carlsson   llvm::Value *Ptr = EmitScalarExpr(Arg);
89359486a2dSAnders Carlsson 
89459486a2dSAnders Carlsson   // Null check the pointer.
89559486a2dSAnders Carlsson   llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
89659486a2dSAnders Carlsson   llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
89759486a2dSAnders Carlsson 
89859486a2dSAnders Carlsson   llvm::Value *IsNull =
89959486a2dSAnders Carlsson     Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()),
90059486a2dSAnders Carlsson                          "isnull");
90159486a2dSAnders Carlsson 
90259486a2dSAnders Carlsson   Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
90359486a2dSAnders Carlsson   EmitBlock(DeleteNotNull);
90459486a2dSAnders Carlsson 
90559486a2dSAnders Carlsson   bool ShouldCallDelete = true;
90659486a2dSAnders Carlsson 
90759486a2dSAnders Carlsson   // Call the destructor if necessary.
90859486a2dSAnders Carlsson   if (const RecordType *RT = DeleteTy->getAs<RecordType>()) {
90959486a2dSAnders Carlsson     if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl())) {
91059486a2dSAnders Carlsson       if (!RD->hasTrivialDestructor()) {
911bac7490fSDouglas Gregor         const CXXDestructorDecl *Dtor = RD->getDestructor();
91259486a2dSAnders Carlsson         if (E->isArrayForm()) {
91321122cf6SAnders Carlsson           llvm::Value *AllocatedObjectPtr;
91421122cf6SAnders Carlsson           llvm::Value *NumElements;
91521122cf6SAnders Carlsson           llvm::tie(AllocatedObjectPtr, NumElements) =
91621122cf6SAnders Carlsson             GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy);
917f5af8deaSAnders Carlsson 
91859486a2dSAnders Carlsson           EmitCXXAggrDestructorCall(Dtor, NumElements, Ptr);
919f5af8deaSAnders Carlsson         } else if (Dtor->isVirtual()) {
92059486a2dSAnders Carlsson           const llvm::Type *Ty =
92159486a2dSAnders Carlsson             CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(Dtor),
92259486a2dSAnders Carlsson                                            /*isVariadic=*/false);
92359486a2dSAnders Carlsson 
92459486a2dSAnders Carlsson           llvm::Value *Callee = BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty);
925e36a6b3eSAnders Carlsson           EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0,
926e36a6b3eSAnders Carlsson                             0, 0);
92759486a2dSAnders Carlsson 
92859486a2dSAnders Carlsson           // The dtor took care of deleting the object.
92959486a2dSAnders Carlsson           ShouldCallDelete = false;
93059486a2dSAnders Carlsson         } else
931f8a71f08SAnders Carlsson           EmitCXXDestructorCall(Dtor, Dtor_Complete, /*ForVirtualBase=*/false,
932f8a71f08SAnders Carlsson                                 Ptr);
93359486a2dSAnders Carlsson       }
93459486a2dSAnders Carlsson     }
93559486a2dSAnders Carlsson   }
93659486a2dSAnders Carlsson 
93759486a2dSAnders Carlsson   if (ShouldCallDelete)
93859486a2dSAnders Carlsson     EmitDeleteCall(E->getOperatorDelete(), Ptr, DeleteTy);
93959486a2dSAnders Carlsson 
94059486a2dSAnders Carlsson   EmitBlock(DeleteEnd);
94159486a2dSAnders Carlsson }
94259486a2dSAnders Carlsson 
94359486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
94459486a2dSAnders Carlsson   QualType Ty = E->getType();
94559486a2dSAnders Carlsson   const llvm::Type *LTy = ConvertType(Ty)->getPointerTo();
946fd7dfeb7SAnders Carlsson 
9473f4336cbSAnders Carlsson   if (E->isTypeOperand()) {
9483f4336cbSAnders Carlsson     llvm::Constant *TypeInfo =
9493f4336cbSAnders Carlsson       CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand());
9503f4336cbSAnders Carlsson     return Builder.CreateBitCast(TypeInfo, LTy);
9513f4336cbSAnders Carlsson   }
952fd7dfeb7SAnders Carlsson 
95359486a2dSAnders Carlsson   Expr *subE = E->getExprOperand();
95459486a2dSAnders Carlsson   Ty = subE->getType();
95559486a2dSAnders Carlsson   CanQualType CanTy = CGM.getContext().getCanonicalType(Ty);
95659486a2dSAnders Carlsson   Ty = CanTy.getUnqualifiedType().getNonReferenceType();
95759486a2dSAnders Carlsson   if (const RecordType *RT = Ty->getAs<RecordType>()) {
95859486a2dSAnders Carlsson     const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
95959486a2dSAnders Carlsson     if (RD->isPolymorphic()) {
96059486a2dSAnders Carlsson       // FIXME: if subE is an lvalue do
96159486a2dSAnders Carlsson       LValue Obj = EmitLValue(subE);
96259486a2dSAnders Carlsson       llvm::Value *This = Obj.getAddress();
96359486a2dSAnders Carlsson       LTy = LTy->getPointerTo()->getPointerTo();
96459486a2dSAnders Carlsson       llvm::Value *V = Builder.CreateBitCast(This, LTy);
96559486a2dSAnders Carlsson       // We need to do a zero check for *p, unless it has NonNullAttr.
96659486a2dSAnders Carlsson       // FIXME: PointerType->hasAttr<NonNullAttr>()
96759486a2dSAnders Carlsson       bool CanBeZero = false;
96859486a2dSAnders Carlsson       if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens()))
96959486a2dSAnders Carlsson         if (UO->getOpcode() == UnaryOperator::Deref)
97059486a2dSAnders Carlsson           CanBeZero = true;
97159486a2dSAnders Carlsson       if (CanBeZero) {
97259486a2dSAnders Carlsson         llvm::BasicBlock *NonZeroBlock = createBasicBlock();
97359486a2dSAnders Carlsson         llvm::BasicBlock *ZeroBlock = createBasicBlock();
97459486a2dSAnders Carlsson 
97559486a2dSAnders Carlsson         llvm::Value *Zero = llvm::Constant::getNullValue(LTy);
97659486a2dSAnders Carlsson         Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero),
97759486a2dSAnders Carlsson                              NonZeroBlock, ZeroBlock);
97859486a2dSAnders Carlsson         EmitBlock(ZeroBlock);
97959486a2dSAnders Carlsson         /// Call __cxa_bad_typeid
98059486a2dSAnders Carlsson         const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext);
98159486a2dSAnders Carlsson         const llvm::FunctionType *FTy;
98259486a2dSAnders Carlsson         FTy = llvm::FunctionType::get(ResultType, false);
98359486a2dSAnders Carlsson         llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid");
98459486a2dSAnders Carlsson         Builder.CreateCall(F)->setDoesNotReturn();
98559486a2dSAnders Carlsson         Builder.CreateUnreachable();
98659486a2dSAnders Carlsson         EmitBlock(NonZeroBlock);
98759486a2dSAnders Carlsson       }
98859486a2dSAnders Carlsson       V = Builder.CreateLoad(V, "vtable");
98959486a2dSAnders Carlsson       V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL);
99059486a2dSAnders Carlsson       V = Builder.CreateLoad(V);
99159486a2dSAnders Carlsson       return V;
99259486a2dSAnders Carlsson     }
99359486a2dSAnders Carlsson   }
9943f4336cbSAnders Carlsson   return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy);
99559486a2dSAnders Carlsson }
99659486a2dSAnders Carlsson 
99759486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V,
99859486a2dSAnders Carlsson                                               const CXXDynamicCastExpr *DCE) {
9993f4336cbSAnders Carlsson   QualType SrcTy = DCE->getSubExpr()->getType();
10003f4336cbSAnders Carlsson   QualType DestTy = DCE->getTypeAsWritten();
10013f4336cbSAnders Carlsson   QualType InnerType = DestTy->getPointeeType();
10023f4336cbSAnders Carlsson 
100359486a2dSAnders Carlsson   const llvm::Type *LTy = ConvertType(DCE->getType());
100459486a2dSAnders Carlsson 
100559486a2dSAnders Carlsson   bool CanBeZero = false;
100659486a2dSAnders Carlsson   bool ToVoid = false;
100759486a2dSAnders Carlsson   bool ThrowOnBad = false;
10083f4336cbSAnders Carlsson   if (DestTy->isPointerType()) {
100959486a2dSAnders Carlsson     // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this
101059486a2dSAnders Carlsson     CanBeZero = true;
101159486a2dSAnders Carlsson     if (InnerType->isVoidType())
101259486a2dSAnders Carlsson       ToVoid = true;
101359486a2dSAnders Carlsson   } else {
101459486a2dSAnders Carlsson     LTy = LTy->getPointerTo();
1015fa8b4955SDouglas Gregor 
1016fa8b4955SDouglas Gregor     // FIXME: What if exceptions are disabled?
101759486a2dSAnders Carlsson     ThrowOnBad = true;
101859486a2dSAnders Carlsson   }
101959486a2dSAnders Carlsson 
10203f4336cbSAnders Carlsson   if (SrcTy->isPointerType() || SrcTy->isReferenceType())
10213f4336cbSAnders Carlsson     SrcTy = SrcTy->getPointeeType();
10223f4336cbSAnders Carlsson   SrcTy = SrcTy.getUnqualifiedType();
10233f4336cbSAnders Carlsson 
10240087bc85SAnders Carlsson   if (DestTy->isPointerType() || DestTy->isReferenceType())
10253f4336cbSAnders Carlsson     DestTy = DestTy->getPointeeType();
10263f4336cbSAnders Carlsson   DestTy = DestTy.getUnqualifiedType();
102759486a2dSAnders Carlsson 
102859486a2dSAnders Carlsson   llvm::BasicBlock *ContBlock = createBasicBlock();
102959486a2dSAnders Carlsson   llvm::BasicBlock *NullBlock = 0;
103059486a2dSAnders Carlsson   llvm::BasicBlock *NonZeroBlock = 0;
103159486a2dSAnders Carlsson   if (CanBeZero) {
103259486a2dSAnders Carlsson     NonZeroBlock = createBasicBlock();
103359486a2dSAnders Carlsson     NullBlock = createBasicBlock();
10343f4336cbSAnders Carlsson     Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock);
103559486a2dSAnders Carlsson     EmitBlock(NonZeroBlock);
103659486a2dSAnders Carlsson   }
103759486a2dSAnders Carlsson 
103859486a2dSAnders Carlsson   llvm::BasicBlock *BadCastBlock = 0;
103959486a2dSAnders Carlsson 
10403f4336cbSAnders Carlsson   const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
104159486a2dSAnders Carlsson 
104259486a2dSAnders Carlsson   // See if this is a dynamic_cast(void*)
104359486a2dSAnders Carlsson   if (ToVoid) {
104459486a2dSAnders Carlsson     llvm::Value *This = V;
104559486a2dSAnders Carlsson     V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo());
104659486a2dSAnders Carlsson     V = Builder.CreateLoad(V, "vtable");
104759486a2dSAnders Carlsson     V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL);
104859486a2dSAnders Carlsson     V = Builder.CreateLoad(V, "offset to top");
104959486a2dSAnders Carlsson     This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext));
105059486a2dSAnders Carlsson     V = Builder.CreateInBoundsGEP(This, V);
105159486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, LTy);
105259486a2dSAnders Carlsson   } else {
105359486a2dSAnders Carlsson     /// Call __dynamic_cast
105459486a2dSAnders Carlsson     const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext);
105559486a2dSAnders Carlsson     const llvm::FunctionType *FTy;
105659486a2dSAnders Carlsson     std::vector<const llvm::Type*> ArgTys;
105759486a2dSAnders Carlsson     const llvm::Type *PtrToInt8Ty
105859486a2dSAnders Carlsson       = llvm::Type::getInt8Ty(VMContext)->getPointerTo();
105959486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
106059486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
106159486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
106259486a2dSAnders Carlsson     ArgTys.push_back(PtrDiffTy);
106359486a2dSAnders Carlsson     FTy = llvm::FunctionType::get(ResultType, ArgTys, false);
106459486a2dSAnders Carlsson 
106559486a2dSAnders Carlsson     // FIXME: Calculate better hint.
106659486a2dSAnders Carlsson     llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL);
10673f4336cbSAnders Carlsson 
10683f4336cbSAnders Carlsson     assert(SrcTy->isRecordType() && "Src type must be record type!");
10693f4336cbSAnders Carlsson     assert(DestTy->isRecordType() && "Dest type must be record type!");
10703f4336cbSAnders Carlsson 
1071247894b3SDouglas Gregor     llvm::Value *SrcArg
1072247894b3SDouglas Gregor       = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType());
1073247894b3SDouglas Gregor     llvm::Value *DestArg
1074247894b3SDouglas Gregor       = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType());
10753f4336cbSAnders Carlsson 
107659486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, PtrToInt8Ty);
107759486a2dSAnders Carlsson     V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"),
10783f4336cbSAnders Carlsson                             V, SrcArg, DestArg, hint);
107959486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, LTy);
108059486a2dSAnders Carlsson 
108159486a2dSAnders Carlsson     if (ThrowOnBad) {
108259486a2dSAnders Carlsson       BadCastBlock = createBasicBlock();
10833f4336cbSAnders Carlsson       Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock);
108459486a2dSAnders Carlsson       EmitBlock(BadCastBlock);
1085fa8b4955SDouglas Gregor       /// Invoke __cxa_bad_cast
108659486a2dSAnders Carlsson       ResultType = llvm::Type::getVoidTy(VMContext);
108759486a2dSAnders Carlsson       const llvm::FunctionType *FBadTy;
108859486a2dSAnders Carlsson       FBadTy = llvm::FunctionType::get(ResultType, false);
108959486a2dSAnders Carlsson       llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast");
1090fa8b4955SDouglas Gregor       if (llvm::BasicBlock *InvokeDest = getInvokeDest()) {
1091fa8b4955SDouglas Gregor         llvm::BasicBlock *Cont = createBasicBlock("invoke.cont");
1092fa8b4955SDouglas Gregor         Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn();
1093fa8b4955SDouglas Gregor         EmitBlock(Cont);
1094fa8b4955SDouglas Gregor       } else {
1095fa8b4955SDouglas Gregor         // FIXME: Does this ever make sense?
109659486a2dSAnders Carlsson         Builder.CreateCall(F)->setDoesNotReturn();
1097fa8b4955SDouglas Gregor       }
109859486a2dSAnders Carlsson       Builder.CreateUnreachable();
109959486a2dSAnders Carlsson     }
110059486a2dSAnders Carlsson   }
110159486a2dSAnders Carlsson 
110259486a2dSAnders Carlsson   if (CanBeZero) {
110359486a2dSAnders Carlsson     Builder.CreateBr(ContBlock);
110459486a2dSAnders Carlsson     EmitBlock(NullBlock);
110559486a2dSAnders Carlsson     Builder.CreateBr(ContBlock);
110659486a2dSAnders Carlsson   }
110759486a2dSAnders Carlsson   EmitBlock(ContBlock);
110859486a2dSAnders Carlsson   if (CanBeZero) {
110959486a2dSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(LTy);
111059486a2dSAnders Carlsson     PHI->reserveOperandSpace(2);
111159486a2dSAnders Carlsson     PHI->addIncoming(V, NonZeroBlock);
111259486a2dSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock);
111359486a2dSAnders Carlsson     V = PHI;
111459486a2dSAnders Carlsson   }
111559486a2dSAnders Carlsson 
111659486a2dSAnders Carlsson   return V;
111759486a2dSAnders Carlsson }
1118