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"
1559486a2dSAnders Carlsson using namespace clang;
1659486a2dSAnders Carlsson using namespace CodeGen;
1759486a2dSAnders Carlsson 
1827da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD,
1927da15baSAnders Carlsson                                           llvm::Value *Callee,
2027da15baSAnders Carlsson                                           ReturnValueSlot ReturnValue,
2127da15baSAnders Carlsson                                           llvm::Value *This,
22e36a6b3eSAnders Carlsson                                           llvm::Value *VTT,
2327da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgBeg,
2427da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgEnd) {
2527da15baSAnders Carlsson   assert(MD->isInstance() &&
2627da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
2727da15baSAnders Carlsson 
2827da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
2927da15baSAnders Carlsson 
3027da15baSAnders Carlsson   CallArgList Args;
3127da15baSAnders Carlsson 
3227da15baSAnders Carlsson   // Push the this ptr.
3327da15baSAnders Carlsson   Args.push_back(std::make_pair(RValue::get(This),
3427da15baSAnders Carlsson                                 MD->getThisType(getContext())));
3527da15baSAnders Carlsson 
36e36a6b3eSAnders Carlsson   // If there is a VTT parameter, emit it.
37e36a6b3eSAnders Carlsson   if (VTT) {
38e36a6b3eSAnders Carlsson     QualType T = getContext().getPointerType(getContext().VoidPtrTy);
39e36a6b3eSAnders Carlsson     Args.push_back(std::make_pair(RValue::get(VTT), T));
40e36a6b3eSAnders Carlsson   }
41e36a6b3eSAnders Carlsson 
4227da15baSAnders Carlsson   // And the rest of the call args
4327da15baSAnders Carlsson   EmitCallArgs(Args, FPT, ArgBeg, ArgEnd);
4427da15baSAnders Carlsson 
45ab26cfa5SJohn McCall   QualType ResultType = FPT->getResultType();
46ab26cfa5SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args,
47c50c27ccSRafael Espindola                                                  FPT->getExtInfo()),
48c50c27ccSRafael Espindola                   Callee, ReturnValue, Args, MD);
4927da15baSAnders Carlsson }
5027da15baSAnders Carlsson 
5127da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given
5227da15baSAnders Carlsson /// expr can be devirtualized.
5327da15baSAnders Carlsson static bool canDevirtualizeMemberFunctionCalls(const Expr *Base) {
5427da15baSAnders Carlsson   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) {
5527da15baSAnders Carlsson     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
5627da15baSAnders Carlsson       // This is a record decl. We know the type and can devirtualize it.
5727da15baSAnders Carlsson       return VD->getType()->isRecordType();
5827da15baSAnders Carlsson     }
5927da15baSAnders Carlsson 
6027da15baSAnders Carlsson     return false;
6127da15baSAnders Carlsson   }
6227da15baSAnders Carlsson 
6327da15baSAnders Carlsson   // We can always devirtualize calls on temporary object expressions.
64a682427eSEli Friedman   if (isa<CXXConstructExpr>(Base))
6527da15baSAnders Carlsson     return true;
6627da15baSAnders Carlsson 
6727da15baSAnders Carlsson   // And calls on bound temporaries.
6827da15baSAnders Carlsson   if (isa<CXXBindTemporaryExpr>(Base))
6927da15baSAnders Carlsson     return true;
7027da15baSAnders Carlsson 
7127da15baSAnders Carlsson   // Check if this is a call expr that returns a record type.
7227da15baSAnders Carlsson   if (const CallExpr *CE = dyn_cast<CallExpr>(Base))
7327da15baSAnders Carlsson     return CE->getCallReturnType()->isRecordType();
7427da15baSAnders Carlsson 
7527da15baSAnders Carlsson   // We can't devirtualize the call.
7627da15baSAnders Carlsson   return false;
7727da15baSAnders Carlsson }
7827da15baSAnders Carlsson 
7927da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
8027da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
8127da15baSAnders Carlsson   if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens()))
8227da15baSAnders Carlsson     return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
8327da15baSAnders Carlsson 
8427da15baSAnders Carlsson   const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens());
8527da15baSAnders Carlsson   const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
8627da15baSAnders Carlsson 
8727da15baSAnders Carlsson   if (MD->isStatic()) {
8827da15baSAnders Carlsson     // The method is static, emit it as we would a regular call.
8927da15baSAnders Carlsson     llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
9027da15baSAnders Carlsson     return EmitCall(getContext().getPointerType(MD->getType()), Callee,
9127da15baSAnders Carlsson                     ReturnValue, CE->arg_begin(), CE->arg_end());
9227da15baSAnders Carlsson   }
9327da15baSAnders Carlsson 
9427da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
9527da15baSAnders Carlsson 
9627da15baSAnders Carlsson   const llvm::Type *Ty =
9727da15baSAnders Carlsson     CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
9827da15baSAnders Carlsson                                    FPT->isVariadic());
9927da15baSAnders Carlsson   llvm::Value *This;
10027da15baSAnders Carlsson 
10127da15baSAnders Carlsson   if (ME->isArrow())
10227da15baSAnders Carlsson     This = EmitScalarExpr(ME->getBase());
10327da15baSAnders Carlsson   else {
10427da15baSAnders Carlsson     LValue BaseLV = EmitLValue(ME->getBase());
10527da15baSAnders Carlsson     This = BaseLV.getAddress();
10627da15baSAnders Carlsson   }
10727da15baSAnders Carlsson 
10827da15baSAnders Carlsson   if (MD->isCopyAssignment() && MD->isTrivial()) {
10927da15baSAnders Carlsson     // We don't like to generate the trivial copy assignment operator when
11027da15baSAnders Carlsson     // it isn't necessary; just produce the proper effect here.
11127da15baSAnders Carlsson     llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
11227da15baSAnders Carlsson     EmitAggregateCopy(This, RHS, CE->getType());
11327da15baSAnders Carlsson     return RValue::get(This);
11427da15baSAnders Carlsson   }
11527da15baSAnders Carlsson 
11627da15baSAnders Carlsson   // C++ [class.virtual]p12:
11727da15baSAnders Carlsson   //   Explicit qualification with the scope operator (5.1) suppresses the
11827da15baSAnders Carlsson   //   virtual call mechanism.
11927da15baSAnders Carlsson   //
12027da15baSAnders Carlsson   // We also don't emit a virtual call if the base expression has a record type
12127da15baSAnders Carlsson   // because then we know what the type is.
12227da15baSAnders Carlsson   llvm::Value *Callee;
12327da15baSAnders Carlsson   if (const CXXDestructorDecl *Destructor
12427da15baSAnders Carlsson              = dyn_cast<CXXDestructorDecl>(MD)) {
12527da15baSAnders Carlsson     if (Destructor->isTrivial())
12627da15baSAnders Carlsson       return RValue::get(0);
12727da15baSAnders Carlsson     if (MD->isVirtual() && !ME->hasQualifier() &&
12827da15baSAnders Carlsson         !canDevirtualizeMemberFunctionCalls(ME->getBase())) {
12927da15baSAnders Carlsson       Callee = BuildVirtualCall(Destructor, Dtor_Complete, This, Ty);
13027da15baSAnders Carlsson     } else {
13127da15baSAnders Carlsson       Callee = CGM.GetAddrOfFunction(GlobalDecl(Destructor, Dtor_Complete), Ty);
13227da15baSAnders Carlsson     }
13327da15baSAnders Carlsson   } else if (MD->isVirtual() && !ME->hasQualifier() &&
13427da15baSAnders Carlsson              !canDevirtualizeMemberFunctionCalls(ME->getBase())) {
13527da15baSAnders Carlsson     Callee = BuildVirtualCall(MD, This, Ty);
13627da15baSAnders Carlsson   } else {
13727da15baSAnders Carlsson     Callee = CGM.GetAddrOfFunction(MD, Ty);
13827da15baSAnders Carlsson   }
13927da15baSAnders Carlsson 
140e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
14127da15baSAnders Carlsson                            CE->arg_begin(), CE->arg_end());
14227da15baSAnders Carlsson }
14327da15baSAnders Carlsson 
14427da15baSAnders Carlsson RValue
14527da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
14627da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
14727da15baSAnders Carlsson   const BinaryOperator *BO =
14827da15baSAnders Carlsson       cast<BinaryOperator>(E->getCallee()->IgnoreParens());
14927da15baSAnders Carlsson   const Expr *BaseExpr = BO->getLHS();
15027da15baSAnders Carlsson   const Expr *MemFnExpr = BO->getRHS();
15127da15baSAnders Carlsson 
15227da15baSAnders Carlsson   const MemberPointerType *MPT =
15327da15baSAnders Carlsson     MemFnExpr->getType()->getAs<MemberPointerType>();
15427da15baSAnders Carlsson   const FunctionProtoType *FPT =
15527da15baSAnders Carlsson     MPT->getPointeeType()->getAs<FunctionProtoType>();
15627da15baSAnders Carlsson   const CXXRecordDecl *RD =
15727da15baSAnders Carlsson     cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
15827da15baSAnders Carlsson 
15927da15baSAnders Carlsson   const llvm::FunctionType *FTy =
16027da15baSAnders Carlsson     CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(RD, FPT),
16127da15baSAnders Carlsson                                    FPT->isVariadic());
16227da15baSAnders Carlsson 
163c4325486SAnders Carlsson   const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(VMContext);
16427da15baSAnders Carlsson 
16527da15baSAnders Carlsson   // Get the member function pointer.
166a7566f16SDaniel Dunbar   llvm::Value *MemFnPtr = CreateMemTemp(MemFnExpr->getType(), "mem.fn");
16727da15baSAnders Carlsson   EmitAggExpr(MemFnExpr, MemFnPtr, /*VolatileDest=*/false);
16827da15baSAnders Carlsson 
16927da15baSAnders Carlsson   // Emit the 'this' pointer.
17027da15baSAnders Carlsson   llvm::Value *This;
17127da15baSAnders Carlsson 
17227da15baSAnders Carlsson   if (BO->getOpcode() == BinaryOperator::PtrMemI)
17327da15baSAnders Carlsson     This = EmitScalarExpr(BaseExpr);
17427da15baSAnders Carlsson   else
17527da15baSAnders Carlsson     This = EmitLValue(BaseExpr).getAddress();
17627da15baSAnders Carlsson 
17727da15baSAnders Carlsson   // Adjust it.
17827da15baSAnders Carlsson   llvm::Value *Adj = Builder.CreateStructGEP(MemFnPtr, 1);
17927da15baSAnders Carlsson   Adj = Builder.CreateLoad(Adj, "mem.fn.adj");
18027da15baSAnders Carlsson 
18127da15baSAnders Carlsson   llvm::Value *Ptr = Builder.CreateBitCast(This, Int8PtrTy, "ptr");
18227da15baSAnders Carlsson   Ptr = Builder.CreateGEP(Ptr, Adj, "adj");
18327da15baSAnders Carlsson 
18427da15baSAnders Carlsson   This = Builder.CreateBitCast(Ptr, This->getType(), "this");
18527da15baSAnders Carlsson 
18627da15baSAnders Carlsson   llvm::Value *FnPtr = Builder.CreateStructGEP(MemFnPtr, 0, "mem.fn.ptr");
18727da15baSAnders Carlsson 
18827da15baSAnders Carlsson   const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
18927da15baSAnders Carlsson 
19027da15baSAnders Carlsson   llvm::Value *FnAsInt = Builder.CreateLoad(FnPtr, "fn");
19127da15baSAnders Carlsson 
19227da15baSAnders Carlsson   // If the LSB in the function pointer is 1, the function pointer points to
19327da15baSAnders Carlsson   // a virtual function.
19427da15baSAnders Carlsson   llvm::Value *IsVirtual
19527da15baSAnders Carlsson     = Builder.CreateAnd(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1),
19627da15baSAnders Carlsson                         "and");
19727da15baSAnders Carlsson 
19827da15baSAnders Carlsson   IsVirtual = Builder.CreateTrunc(IsVirtual,
19927da15baSAnders Carlsson                                   llvm::Type::getInt1Ty(VMContext));
20027da15baSAnders Carlsson 
20127da15baSAnders Carlsson   llvm::BasicBlock *FnVirtual = createBasicBlock("fn.virtual");
20227da15baSAnders Carlsson   llvm::BasicBlock *FnNonVirtual = createBasicBlock("fn.nonvirtual");
20327da15baSAnders Carlsson   llvm::BasicBlock *FnEnd = createBasicBlock("fn.end");
20427da15baSAnders Carlsson 
20527da15baSAnders Carlsson   Builder.CreateCondBr(IsVirtual, FnVirtual, FnNonVirtual);
20627da15baSAnders Carlsson   EmitBlock(FnVirtual);
20727da15baSAnders Carlsson 
208c4325486SAnders Carlsson   const llvm::Type *VtableTy =
209c4325486SAnders Carlsson     FTy->getPointerTo()->getPointerTo();
21027da15baSAnders Carlsson 
211c4325486SAnders Carlsson   llvm::Value *Vtable = Builder.CreateBitCast(This, VtableTy->getPointerTo());
212c4325486SAnders Carlsson   Vtable = Builder.CreateLoad(Vtable);
21327da15baSAnders Carlsson 
214c4325486SAnders Carlsson   Vtable = Builder.CreateBitCast(Vtable, Int8PtrTy);
215c4325486SAnders Carlsson   llvm::Value *VtableOffset =
216c4325486SAnders Carlsson     Builder.CreateSub(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1));
21727da15baSAnders Carlsson 
218c4325486SAnders Carlsson   Vtable = Builder.CreateGEP(Vtable, VtableOffset, "fn");
219c4325486SAnders Carlsson   Vtable = Builder.CreateBitCast(Vtable, VtableTy);
22027da15baSAnders Carlsson 
221c4325486SAnders Carlsson   llvm::Value *VirtualFn = Builder.CreateLoad(Vtable, "virtualfn");
22227da15baSAnders Carlsson 
22327da15baSAnders Carlsson   EmitBranch(FnEnd);
22427da15baSAnders Carlsson   EmitBlock(FnNonVirtual);
22527da15baSAnders Carlsson 
22627da15baSAnders Carlsson   // If the function is not virtual, just load the pointer.
22727da15baSAnders Carlsson   llvm::Value *NonVirtualFn = Builder.CreateLoad(FnPtr, "fn");
22827da15baSAnders Carlsson   NonVirtualFn = Builder.CreateIntToPtr(NonVirtualFn, FTy->getPointerTo());
22927da15baSAnders Carlsson 
23027da15baSAnders Carlsson   EmitBlock(FnEnd);
23127da15baSAnders Carlsson 
23227da15baSAnders Carlsson   llvm::PHINode *Callee = Builder.CreatePHI(FTy->getPointerTo());
23327da15baSAnders Carlsson   Callee->reserveOperandSpace(2);
23427da15baSAnders Carlsson   Callee->addIncoming(VirtualFn, FnVirtual);
23527da15baSAnders Carlsson   Callee->addIncoming(NonVirtualFn, FnNonVirtual);
23627da15baSAnders Carlsson 
23727da15baSAnders Carlsson   CallArgList Args;
23827da15baSAnders Carlsson 
23927da15baSAnders Carlsson   QualType ThisType =
24027da15baSAnders Carlsson     getContext().getPointerType(getContext().getTagDeclType(RD));
24127da15baSAnders Carlsson 
24227da15baSAnders Carlsson   // Push the this ptr.
24327da15baSAnders Carlsson   Args.push_back(std::make_pair(RValue::get(This), ThisType));
24427da15baSAnders Carlsson 
24527da15baSAnders Carlsson   // And the rest of the call args
24627da15baSAnders Carlsson   EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
247ab26cfa5SJohn McCall   const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>();
248ab26cfa5SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee,
24927da15baSAnders Carlsson                   ReturnValue, Args);
25027da15baSAnders Carlsson }
25127da15baSAnders Carlsson 
25227da15baSAnders Carlsson RValue
25327da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
25427da15baSAnders Carlsson                                                const CXXMethodDecl *MD,
25527da15baSAnders Carlsson                                                ReturnValueSlot ReturnValue) {
25627da15baSAnders Carlsson   assert(MD->isInstance() &&
25727da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
25827da15baSAnders Carlsson 
25927da15baSAnders Carlsson   if (MD->isCopyAssignment()) {
26027da15baSAnders Carlsson     const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext());
26127da15baSAnders Carlsson     if (ClassDecl->hasTrivialCopyAssignment()) {
26227da15baSAnders Carlsson       assert(!ClassDecl->hasUserDeclaredCopyAssignment() &&
26327da15baSAnders Carlsson              "EmitCXXOperatorMemberCallExpr - user declared copy assignment");
26427da15baSAnders Carlsson       llvm::Value *This = EmitLValue(E->getArg(0)).getAddress();
26527da15baSAnders Carlsson       llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
26627da15baSAnders Carlsson       QualType Ty = E->getType();
26727da15baSAnders Carlsson       EmitAggregateCopy(This, Src, Ty);
26827da15baSAnders Carlsson       return RValue::get(This);
26927da15baSAnders Carlsson     }
27027da15baSAnders Carlsson   }
27127da15baSAnders Carlsson 
27227da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
27327da15baSAnders Carlsson   const llvm::Type *Ty =
27427da15baSAnders Carlsson     CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
27527da15baSAnders Carlsson                                    FPT->isVariadic());
27627da15baSAnders Carlsson 
27727da15baSAnders Carlsson   llvm::Value *This = EmitLValue(E->getArg(0)).getAddress();
27827da15baSAnders Carlsson 
27927da15baSAnders Carlsson   llvm::Value *Callee;
28027da15baSAnders Carlsson   if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0)))
28127da15baSAnders Carlsson     Callee = BuildVirtualCall(MD, This, Ty);
28227da15baSAnders Carlsson   else
28327da15baSAnders Carlsson     Callee = CGM.GetAddrOfFunction(MD, Ty);
28427da15baSAnders Carlsson 
285e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
28627da15baSAnders Carlsson                            E->arg_begin() + 1, E->arg_end());
28727da15baSAnders Carlsson }
28827da15baSAnders Carlsson 
28927da15baSAnders Carlsson void
29027da15baSAnders Carlsson CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest,
29127da15baSAnders Carlsson                                       const CXXConstructExpr *E) {
29227da15baSAnders Carlsson   assert(Dest && "Must have a destination!");
29327da15baSAnders Carlsson   const CXXConstructorDecl *CD = E->getConstructor();
29427da15baSAnders Carlsson   const ConstantArrayType *Array =
29527da15baSAnders Carlsson   getContext().getAsConstantArrayType(E->getType());
29627da15baSAnders Carlsson   // For a copy constructor, even if it is trivial, must fall thru so
29727da15baSAnders Carlsson   // its argument is code-gen'ed.
29827da15baSAnders Carlsson   if (!CD->isCopyConstructor()) {
29927da15baSAnders Carlsson     QualType InitType = E->getType();
30027da15baSAnders Carlsson     if (Array)
30127da15baSAnders Carlsson       InitType = getContext().getBaseElementType(Array);
30227da15baSAnders Carlsson     const CXXRecordDecl *RD =
30327da15baSAnders Carlsson     cast<CXXRecordDecl>(InitType->getAs<RecordType>()->getDecl());
30427da15baSAnders Carlsson     if (RD->hasTrivialConstructor())
30527da15baSAnders Carlsson       return;
30627da15baSAnders Carlsson   }
30727da15baSAnders Carlsson   // Code gen optimization to eliminate copy constructor and return
30827da15baSAnders Carlsson   // its first argument instead.
30927da15baSAnders Carlsson   if (getContext().getLangOptions().ElideConstructors && E->isElidable()) {
310*45cf7e3dSDouglas Gregor     const Expr *Arg = E->getArg(0)->getTemporaryObject();
31127da15baSAnders Carlsson     EmitAggExpr(Arg, Dest, false);
31227da15baSAnders Carlsson     return;
31327da15baSAnders Carlsson   }
31427da15baSAnders Carlsson   if (Array) {
31527da15baSAnders Carlsson     QualType BaseElementTy = getContext().getBaseElementType(Array);
31627da15baSAnders Carlsson     const llvm::Type *BasePtr = ConvertType(BaseElementTy);
31727da15baSAnders Carlsson     BasePtr = llvm::PointerType::getUnqual(BasePtr);
31827da15baSAnders Carlsson     llvm::Value *BaseAddrPtr =
31927da15baSAnders Carlsson     Builder.CreateBitCast(Dest, BasePtr);
32027da15baSAnders Carlsson 
32127da15baSAnders Carlsson     EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr,
32227da15baSAnders Carlsson                                E->arg_begin(), E->arg_end());
32327da15baSAnders Carlsson   }
32427da15baSAnders Carlsson   else
32527da15baSAnders Carlsson     // Call the constructor.
3267ae2d775SDouglas Gregor     EmitCXXConstructorCall(CD,
3277ae2d775SDouglas Gregor                            E->isBaseInitialization()? Ctor_Base : Ctor_Complete,
3287ae2d775SDouglas Gregor                            Dest,
32927da15baSAnders Carlsson                            E->arg_begin(), E->arg_end());
33027da15baSAnders Carlsson }
33127da15baSAnders Carlsson 
3323eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, QualType ElementType) {
33321122cf6SAnders Carlsson   const RecordType *RT = ElementType->getAs<RecordType>();
33459486a2dSAnders Carlsson   if (!RT)
3353eb55cfeSKen Dyck     return CharUnits::Zero();
33659486a2dSAnders Carlsson 
33759486a2dSAnders Carlsson   const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl());
33859486a2dSAnders Carlsson   if (!RD)
3393eb55cfeSKen Dyck     return CharUnits::Zero();
34059486a2dSAnders Carlsson 
34159486a2dSAnders Carlsson   // Check if the class has a trivial destructor.
34259486a2dSAnders Carlsson   if (RD->hasTrivialDestructor()) {
34321122cf6SAnders Carlsson     // Check if the usual deallocation function takes two arguments.
344adbe4249SAnders Carlsson     const CXXMethodDecl *UsualDeallocationFunction = 0;
345adbe4249SAnders Carlsson 
34621122cf6SAnders Carlsson     DeclarationName OpName =
34721122cf6SAnders Carlsson       Ctx.DeclarationNames.getCXXOperatorName(OO_Array_Delete);
34821122cf6SAnders Carlsson     DeclContext::lookup_const_iterator Op, OpEnd;
34921122cf6SAnders Carlsson     for (llvm::tie(Op, OpEnd) = RD->lookup(OpName);
35021122cf6SAnders Carlsson          Op != OpEnd; ++Op) {
351adbe4249SAnders Carlsson       const CXXMethodDecl *Delete = cast<CXXMethodDecl>(*Op);
35221122cf6SAnders Carlsson 
35321122cf6SAnders Carlsson       if (Delete->isUsualDeallocationFunction()) {
354adbe4249SAnders Carlsson         UsualDeallocationFunction = Delete;
35521122cf6SAnders Carlsson         break;
35621122cf6SAnders Carlsson       }
35721122cf6SAnders Carlsson     }
358adbe4249SAnders Carlsson 
359adbe4249SAnders Carlsson     // No usual deallocation function, we don't need a cookie.
360adbe4249SAnders Carlsson     if (!UsualDeallocationFunction)
3613eb55cfeSKen Dyck       return CharUnits::Zero();
362adbe4249SAnders Carlsson 
363adbe4249SAnders Carlsson     // The usual deallocation function doesn't take a size_t argument, so we
364adbe4249SAnders Carlsson     // don't need a cookie.
365adbe4249SAnders Carlsson     if (UsualDeallocationFunction->getNumParams() == 1)
3663eb55cfeSKen Dyck       return CharUnits::Zero();
367adbe4249SAnders Carlsson 
368adbe4249SAnders Carlsson     assert(UsualDeallocationFunction->getNumParams() == 2 &&
369adbe4249SAnders Carlsson            "Unexpected deallocation function type!");
37059486a2dSAnders Carlsson   }
37159486a2dSAnders Carlsson 
37221122cf6SAnders Carlsson   // Padding is the maximum of sizeof(size_t) and alignof(ElementType)
3733eb55cfeSKen Dyck   return std::max(Ctx.getTypeSizeInChars(Ctx.getSizeType()),
3743eb55cfeSKen Dyck                   Ctx.getTypeAlignInChars(ElementType));
37521122cf6SAnders Carlsson }
37621122cf6SAnders Carlsson 
3773eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, const CXXNewExpr *E) {
37821122cf6SAnders Carlsson   if (!E->isArray())
3793eb55cfeSKen Dyck     return CharUnits::Zero();
38021122cf6SAnders Carlsson 
381399f499fSAnders Carlsson   // No cookie is required if the new operator being used is
382399f499fSAnders Carlsson   // ::operator new[](size_t, void*).
383399f499fSAnders Carlsson   const FunctionDecl *OperatorNew = E->getOperatorNew();
384399f499fSAnders Carlsson   if (OperatorNew->getDeclContext()->getLookupContext()->isFileContext()) {
385399f499fSAnders Carlsson     if (OperatorNew->getNumParams() == 2) {
386399f499fSAnders Carlsson       CanQualType ParamType =
387399f499fSAnders Carlsson         Ctx.getCanonicalType(OperatorNew->getParamDecl(1)->getType());
388399f499fSAnders Carlsson 
389399f499fSAnders Carlsson       if (ParamType == Ctx.VoidPtrTy)
3903eb55cfeSKen Dyck         return CharUnits::Zero();
391399f499fSAnders Carlsson     }
392399f499fSAnders Carlsson   }
393399f499fSAnders Carlsson 
39421122cf6SAnders Carlsson   return CalculateCookiePadding(Ctx, E->getAllocatedType());
39559486a2dSAnders Carlsson }
39659486a2dSAnders Carlsson 
39747b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context,
39847b4629bSFariborz Jahanian                                         CodeGenFunction &CGF,
39959486a2dSAnders Carlsson                                         const CXXNewExpr *E,
40059486a2dSAnders Carlsson                                         llvm::Value *& NumElements) {
40159486a2dSAnders Carlsson   QualType Type = E->getAllocatedType();
4023eb55cfeSKen Dyck   CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(Type);
40359486a2dSAnders Carlsson   const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType());
40459486a2dSAnders Carlsson 
40559486a2dSAnders Carlsson   if (!E->isArray())
4063eb55cfeSKen Dyck     return llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
40759486a2dSAnders Carlsson 
4083eb55cfeSKen Dyck   CharUnits CookiePadding = CalculateCookiePadding(CGF.getContext(), E);
40959486a2dSAnders Carlsson 
41059486a2dSAnders Carlsson   Expr::EvalResult Result;
41159486a2dSAnders Carlsson   if (E->getArraySize()->Evaluate(Result, CGF.getContext()) &&
41259486a2dSAnders Carlsson       !Result.HasSideEffects && Result.Val.isInt()) {
41359486a2dSAnders Carlsson 
4143eb55cfeSKen Dyck     CharUnits AllocSize =
4153eb55cfeSKen Dyck       Result.Val.getInt().getZExtValue() * TypeSize + CookiePadding;
41659486a2dSAnders Carlsson 
41759486a2dSAnders Carlsson     NumElements =
41859486a2dSAnders Carlsson       llvm::ConstantInt::get(SizeTy, Result.Val.getInt().getZExtValue());
41947b4629bSFariborz Jahanian     while (const ArrayType *AType = Context.getAsArrayType(Type)) {
42047b4629bSFariborz Jahanian       const llvm::ArrayType *llvmAType =
42147b4629bSFariborz Jahanian         cast<llvm::ArrayType>(CGF.ConvertType(Type));
42247b4629bSFariborz Jahanian       NumElements =
42347b4629bSFariborz Jahanian         CGF.Builder.CreateMul(NumElements,
42447b4629bSFariborz Jahanian                               llvm::ConstantInt::get(
42547b4629bSFariborz Jahanian                                         SizeTy, llvmAType->getNumElements()));
42647b4629bSFariborz Jahanian       Type = AType->getElementType();
42747b4629bSFariborz Jahanian     }
42859486a2dSAnders Carlsson 
4293eb55cfeSKen Dyck     return llvm::ConstantInt::get(SizeTy, AllocSize.getQuantity());
43059486a2dSAnders Carlsson   }
43159486a2dSAnders Carlsson 
43259486a2dSAnders Carlsson   // Emit the array size expression.
43359486a2dSAnders Carlsson   NumElements = CGF.EmitScalarExpr(E->getArraySize());
43459486a2dSAnders Carlsson 
43559486a2dSAnders Carlsson   // Multiply with the type size.
43659486a2dSAnders Carlsson   llvm::Value *V =
43759486a2dSAnders Carlsson     CGF.Builder.CreateMul(NumElements,
4383eb55cfeSKen Dyck                           llvm::ConstantInt::get(SizeTy,
4393eb55cfeSKen Dyck                                                  TypeSize.getQuantity()));
44059486a2dSAnders Carlsson 
44147b4629bSFariborz Jahanian   while (const ArrayType *AType = Context.getAsArrayType(Type)) {
44247b4629bSFariborz Jahanian     const llvm::ArrayType *llvmAType =
44347b4629bSFariborz Jahanian       cast<llvm::ArrayType>(CGF.ConvertType(Type));
44447b4629bSFariborz Jahanian     NumElements =
44547b4629bSFariborz Jahanian       CGF.Builder.CreateMul(NumElements,
44647b4629bSFariborz Jahanian                             llvm::ConstantInt::get(
44747b4629bSFariborz Jahanian                                           SizeTy, llvmAType->getNumElements()));
44847b4629bSFariborz Jahanian     Type = AType->getElementType();
44947b4629bSFariborz Jahanian   }
45047b4629bSFariborz Jahanian 
45159486a2dSAnders Carlsson   // And add the cookie padding if necessary.
4523eb55cfeSKen Dyck   if (!CookiePadding.isZero())
4533eb55cfeSKen Dyck     V = CGF.Builder.CreateAdd(V,
4543eb55cfeSKen Dyck         llvm::ConstantInt::get(SizeTy, CookiePadding.getQuantity()));
45559486a2dSAnders Carlsson 
45659486a2dSAnders Carlsson   return V;
45759486a2dSAnders Carlsson }
45859486a2dSAnders Carlsson 
45959486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
46059486a2dSAnders Carlsson                                llvm::Value *NewPtr,
46159486a2dSAnders Carlsson                                llvm::Value *NumElements) {
4623a202f60SAnders Carlsson   if (E->isArray()) {
4633a202f60SAnders Carlsson     if (CXXConstructorDecl *Ctor = E->getConstructor())
4643a202f60SAnders Carlsson       CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr,
4653a202f60SAnders Carlsson                                      E->constructor_arg_begin(),
4663a202f60SAnders Carlsson                                      E->constructor_arg_end());
4673a202f60SAnders Carlsson     return;
4683a202f60SAnders Carlsson   }
4693a202f60SAnders Carlsson 
47059486a2dSAnders Carlsson   QualType AllocType = E->getAllocatedType();
47159486a2dSAnders Carlsson 
47259486a2dSAnders Carlsson   if (CXXConstructorDecl *Ctor = E->getConstructor()) {
47359486a2dSAnders Carlsson     CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, NewPtr,
47459486a2dSAnders Carlsson                                E->constructor_arg_begin(),
47559486a2dSAnders Carlsson                                E->constructor_arg_end());
47659486a2dSAnders Carlsson 
47759486a2dSAnders Carlsson     return;
47859486a2dSAnders Carlsson   }
47959486a2dSAnders Carlsson 
48059486a2dSAnders Carlsson   // We have a POD type.
48159486a2dSAnders Carlsson   if (E->getNumConstructorArgs() == 0)
48259486a2dSAnders Carlsson     return;
48359486a2dSAnders Carlsson 
48459486a2dSAnders Carlsson   assert(E->getNumConstructorArgs() == 1 &&
48559486a2dSAnders Carlsson          "Can only have one argument to initializer of POD type.");
48659486a2dSAnders Carlsson 
48759486a2dSAnders Carlsson   const Expr *Init = E->getConstructorArg(0);
48859486a2dSAnders Carlsson 
48959486a2dSAnders Carlsson   if (!CGF.hasAggregateLLVMType(AllocType))
49059486a2dSAnders Carlsson     CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr,
49159486a2dSAnders Carlsson                           AllocType.isVolatileQualified(), AllocType);
49259486a2dSAnders Carlsson   else if (AllocType->isAnyComplexType())
49359486a2dSAnders Carlsson     CGF.EmitComplexExprIntoAddr(Init, NewPtr,
49459486a2dSAnders Carlsson                                 AllocType.isVolatileQualified());
49559486a2dSAnders Carlsson   else
49659486a2dSAnders Carlsson     CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified());
49759486a2dSAnders Carlsson }
49859486a2dSAnders Carlsson 
49959486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
50059486a2dSAnders Carlsson   QualType AllocType = E->getAllocatedType();
50159486a2dSAnders Carlsson   FunctionDecl *NewFD = E->getOperatorNew();
50259486a2dSAnders Carlsson   const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>();
50359486a2dSAnders Carlsson 
50459486a2dSAnders Carlsson   CallArgList NewArgs;
50559486a2dSAnders Carlsson 
50659486a2dSAnders Carlsson   // The allocation size is the first argument.
50759486a2dSAnders Carlsson   QualType SizeTy = getContext().getSizeType();
50859486a2dSAnders Carlsson 
50959486a2dSAnders Carlsson   llvm::Value *NumElements = 0;
51047b4629bSFariborz Jahanian   llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(),
51147b4629bSFariborz Jahanian                                                *this, E, NumElements);
51259486a2dSAnders Carlsson 
51359486a2dSAnders Carlsson   NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy));
51459486a2dSAnders Carlsson 
51559486a2dSAnders Carlsson   // Emit the rest of the arguments.
51659486a2dSAnders Carlsson   // FIXME: Ideally, this should just use EmitCallArgs.
51759486a2dSAnders Carlsson   CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin();
51859486a2dSAnders Carlsson 
51959486a2dSAnders Carlsson   // First, use the types from the function type.
52059486a2dSAnders Carlsson   // We start at 1 here because the first argument (the allocation size)
52159486a2dSAnders Carlsson   // has already been emitted.
52259486a2dSAnders Carlsson   for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) {
52359486a2dSAnders Carlsson     QualType ArgType = NewFTy->getArgType(i);
52459486a2dSAnders Carlsson 
52559486a2dSAnders Carlsson     assert(getContext().getCanonicalType(ArgType.getNonReferenceType()).
52659486a2dSAnders Carlsson            getTypePtr() ==
52759486a2dSAnders Carlsson            getContext().getCanonicalType(NewArg->getType()).getTypePtr() &&
52859486a2dSAnders Carlsson            "type mismatch in call argument!");
52959486a2dSAnders Carlsson 
53059486a2dSAnders Carlsson     NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
53159486a2dSAnders Carlsson                                      ArgType));
53259486a2dSAnders Carlsson 
53359486a2dSAnders Carlsson   }
53459486a2dSAnders Carlsson 
53559486a2dSAnders Carlsson   // Either we've emitted all the call args, or we have a call to a
53659486a2dSAnders Carlsson   // variadic function.
53759486a2dSAnders Carlsson   assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) &&
53859486a2dSAnders Carlsson          "Extra arguments in non-variadic function!");
53959486a2dSAnders Carlsson 
54059486a2dSAnders Carlsson   // If we still have any arguments, emit them using the type of the argument.
54159486a2dSAnders Carlsson   for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end();
54259486a2dSAnders Carlsson        NewArg != NewArgEnd; ++NewArg) {
54359486a2dSAnders Carlsson     QualType ArgType = NewArg->getType();
54459486a2dSAnders Carlsson     NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
54559486a2dSAnders Carlsson                                      ArgType));
54659486a2dSAnders Carlsson   }
54759486a2dSAnders Carlsson 
54859486a2dSAnders Carlsson   // Emit the call to new.
54959486a2dSAnders Carlsson   RValue RV =
550ab26cfa5SJohn McCall     EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy),
55161a401caSAnders Carlsson              CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD);
55259486a2dSAnders Carlsson 
55359486a2dSAnders Carlsson   // If an allocation function is declared with an empty exception specification
55459486a2dSAnders Carlsson   // it returns null to indicate failure to allocate storage. [expr.new]p13.
55559486a2dSAnders Carlsson   // (We don't need to check for null when there's no new initializer and
55659486a2dSAnders Carlsson   // we're allocating a POD type).
55759486a2dSAnders Carlsson   bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() &&
55859486a2dSAnders Carlsson     !(AllocType->isPODType() && !E->hasInitializer());
55959486a2dSAnders Carlsson 
56059486a2dSAnders Carlsson   llvm::BasicBlock *NewNull = 0;
56159486a2dSAnders Carlsson   llvm::BasicBlock *NewNotNull = 0;
56259486a2dSAnders Carlsson   llvm::BasicBlock *NewEnd = 0;
56359486a2dSAnders Carlsson 
56459486a2dSAnders Carlsson   llvm::Value *NewPtr = RV.getScalarVal();
56559486a2dSAnders Carlsson 
56659486a2dSAnders Carlsson   if (NullCheckResult) {
56759486a2dSAnders Carlsson     NewNull = createBasicBlock("new.null");
56859486a2dSAnders Carlsson     NewNotNull = createBasicBlock("new.notnull");
56959486a2dSAnders Carlsson     NewEnd = createBasicBlock("new.end");
57059486a2dSAnders Carlsson 
57159486a2dSAnders Carlsson     llvm::Value *IsNull =
57259486a2dSAnders Carlsson       Builder.CreateICmpEQ(NewPtr,
57359486a2dSAnders Carlsson                            llvm::Constant::getNullValue(NewPtr->getType()),
57459486a2dSAnders Carlsson                            "isnull");
57559486a2dSAnders Carlsson 
57659486a2dSAnders Carlsson     Builder.CreateCondBr(IsNull, NewNull, NewNotNull);
57759486a2dSAnders Carlsson     EmitBlock(NewNotNull);
57859486a2dSAnders Carlsson   }
57959486a2dSAnders Carlsson 
5803eb55cfeSKen Dyck   CharUnits CookiePadding = CalculateCookiePadding(getContext(), E);
5813eb55cfeSKen Dyck   if (!CookiePadding.isZero()) {
5823eb55cfeSKen Dyck     CharUnits CookieOffset =
5833eb55cfeSKen Dyck       CookiePadding - getContext().getTypeSizeInChars(SizeTy);
58459486a2dSAnders Carlsson 
58559486a2dSAnders Carlsson     llvm::Value *NumElementsPtr =
5863eb55cfeSKen Dyck       Builder.CreateConstInBoundsGEP1_64(NewPtr, CookieOffset.getQuantity());
58759486a2dSAnders Carlsson 
58859486a2dSAnders Carlsson     NumElementsPtr = Builder.CreateBitCast(NumElementsPtr,
58959486a2dSAnders Carlsson                                            ConvertType(SizeTy)->getPointerTo());
59059486a2dSAnders Carlsson     Builder.CreateStore(NumElements, NumElementsPtr);
59159486a2dSAnders Carlsson 
59259486a2dSAnders Carlsson     // Now add the padding to the new ptr.
5933eb55cfeSKen Dyck     NewPtr = Builder.CreateConstInBoundsGEP1_64(NewPtr,
5943eb55cfeSKen Dyck                                                 CookiePadding.getQuantity());
59559486a2dSAnders Carlsson   }
59659486a2dSAnders Carlsson 
59747b4629bSFariborz Jahanian   if (AllocType->isArrayType()) {
59847b4629bSFariborz Jahanian     while (const ArrayType *AType = getContext().getAsArrayType(AllocType))
59947b4629bSFariborz Jahanian       AllocType = AType->getElementType();
60047b4629bSFariborz Jahanian     NewPtr =
60147b4629bSFariborz Jahanian       Builder.CreateBitCast(NewPtr,
60247b4629bSFariborz Jahanian                           ConvertType(getContext().getPointerType(AllocType)));
60359486a2dSAnders Carlsson     EmitNewInitializer(*this, E, NewPtr, NumElements);
60447b4629bSFariborz Jahanian     NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType()));
60547b4629bSFariborz Jahanian   }
60647b4629bSFariborz Jahanian   else {
60747b4629bSFariborz Jahanian     NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType()));
60847b4629bSFariborz Jahanian     EmitNewInitializer(*this, E, NewPtr, NumElements);
60947b4629bSFariborz Jahanian   }
61059486a2dSAnders Carlsson 
61159486a2dSAnders Carlsson   if (NullCheckResult) {
61259486a2dSAnders Carlsson     Builder.CreateBr(NewEnd);
61359486a2dSAnders Carlsson     NewNotNull = Builder.GetInsertBlock();
61459486a2dSAnders Carlsson     EmitBlock(NewNull);
61559486a2dSAnders Carlsson     Builder.CreateBr(NewEnd);
61659486a2dSAnders Carlsson     EmitBlock(NewEnd);
61759486a2dSAnders Carlsson 
61859486a2dSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType());
61959486a2dSAnders Carlsson     PHI->reserveOperandSpace(2);
62059486a2dSAnders Carlsson     PHI->addIncoming(NewPtr, NewNotNull);
62159486a2dSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), NewNull);
62259486a2dSAnders Carlsson 
62359486a2dSAnders Carlsson     NewPtr = PHI;
62459486a2dSAnders Carlsson   }
62559486a2dSAnders Carlsson 
62659486a2dSAnders Carlsson   return NewPtr;
62759486a2dSAnders Carlsson }
62859486a2dSAnders Carlsson 
62921122cf6SAnders Carlsson static std::pair<llvm::Value *, llvm::Value *>
63021122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(CodeGenFunction &CGF,
63121122cf6SAnders Carlsson                                     llvm::Value *Ptr, QualType DeleteTy) {
63221122cf6SAnders Carlsson   QualType SizeTy = CGF.getContext().getSizeType();
63321122cf6SAnders Carlsson   const llvm::Type *SizeLTy = CGF.ConvertType(SizeTy);
63421122cf6SAnders Carlsson 
6353eb55cfeSKen Dyck   CharUnits DeleteTypeAlign = CGF.getContext().getTypeAlignInChars(DeleteTy);
6363eb55cfeSKen Dyck   CharUnits CookiePadding =
6373eb55cfeSKen Dyck     std::max(CGF.getContext().getTypeSizeInChars(SizeTy),
6383eb55cfeSKen Dyck              DeleteTypeAlign);
6393eb55cfeSKen Dyck   assert(!CookiePadding.isZero() && "CookiePadding should not be 0.");
64021122cf6SAnders Carlsson 
64121122cf6SAnders Carlsson   const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext());
6423eb55cfeSKen Dyck   CharUnits CookieOffset =
6433eb55cfeSKen Dyck     CookiePadding - CGF.getContext().getTypeSizeInChars(SizeTy);
64421122cf6SAnders Carlsson 
64521122cf6SAnders Carlsson   llvm::Value *AllocatedObjectPtr = CGF.Builder.CreateBitCast(Ptr, Int8PtrTy);
64621122cf6SAnders Carlsson   AllocatedObjectPtr =
64721122cf6SAnders Carlsson     CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr,
6483eb55cfeSKen Dyck                                            -CookiePadding.getQuantity());
64921122cf6SAnders Carlsson 
65021122cf6SAnders Carlsson   llvm::Value *NumElementsPtr =
65121122cf6SAnders Carlsson     CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr,
6523eb55cfeSKen Dyck                                            CookieOffset.getQuantity());
65321122cf6SAnders Carlsson   NumElementsPtr =
65421122cf6SAnders Carlsson     CGF.Builder.CreateBitCast(NumElementsPtr, SizeLTy->getPointerTo());
65521122cf6SAnders Carlsson 
65621122cf6SAnders Carlsson   llvm::Value *NumElements = CGF.Builder.CreateLoad(NumElementsPtr);
65721122cf6SAnders Carlsson   NumElements =
65821122cf6SAnders Carlsson     CGF.Builder.CreateIntCast(NumElements, SizeLTy, /*isSigned=*/false);
65921122cf6SAnders Carlsson 
66021122cf6SAnders Carlsson   return std::make_pair(AllocatedObjectPtr, NumElements);
66121122cf6SAnders Carlsson }
66221122cf6SAnders Carlsson 
66359486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
66459486a2dSAnders Carlsson                                      llvm::Value *Ptr,
66559486a2dSAnders Carlsson                                      QualType DeleteTy) {
66659486a2dSAnders Carlsson   const FunctionProtoType *DeleteFTy =
66759486a2dSAnders Carlsson     DeleteFD->getType()->getAs<FunctionProtoType>();
66859486a2dSAnders Carlsson 
66959486a2dSAnders Carlsson   CallArgList DeleteArgs;
67059486a2dSAnders Carlsson 
67121122cf6SAnders Carlsson   // Check if we need to pass the size to the delete operator.
67221122cf6SAnders Carlsson   llvm::Value *Size = 0;
67321122cf6SAnders Carlsson   QualType SizeTy;
67421122cf6SAnders Carlsson   if (DeleteFTy->getNumArgs() == 2) {
67521122cf6SAnders Carlsson     SizeTy = DeleteFTy->getArgType(1);
6767df3cbebSKen Dyck     CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
6777df3cbebSKen Dyck     Size = llvm::ConstantInt::get(ConvertType(SizeTy),
6787df3cbebSKen Dyck                                   DeleteTypeSize.getQuantity());
67921122cf6SAnders Carlsson   }
68021122cf6SAnders Carlsson 
68121122cf6SAnders Carlsson   if (DeleteFD->getOverloadedOperator() == OO_Array_Delete &&
6823eb55cfeSKen Dyck       !CalculateCookiePadding(getContext(), DeleteTy).isZero()) {
68321122cf6SAnders Carlsson     // We need to get the number of elements in the array from the cookie.
68421122cf6SAnders Carlsson     llvm::Value *AllocatedObjectPtr;
68521122cf6SAnders Carlsson     llvm::Value *NumElements;
68621122cf6SAnders Carlsson     llvm::tie(AllocatedObjectPtr, NumElements) =
68721122cf6SAnders Carlsson       GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy);
68821122cf6SAnders Carlsson 
68921122cf6SAnders Carlsson     // Multiply the size with the number of elements.
69021122cf6SAnders Carlsson     if (Size)
69121122cf6SAnders Carlsson       Size = Builder.CreateMul(NumElements, Size);
69221122cf6SAnders Carlsson 
69321122cf6SAnders Carlsson     Ptr = AllocatedObjectPtr;
69421122cf6SAnders Carlsson   }
69521122cf6SAnders Carlsson 
69659486a2dSAnders Carlsson   QualType ArgTy = DeleteFTy->getArgType(0);
69759486a2dSAnders Carlsson   llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
69859486a2dSAnders Carlsson   DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy));
69959486a2dSAnders Carlsson 
70021122cf6SAnders Carlsson   if (Size)
70159486a2dSAnders Carlsson     DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy));
70259486a2dSAnders Carlsson 
70359486a2dSAnders Carlsson   // Emit the call to delete.
704ab26cfa5SJohn McCall   EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy),
70561a401caSAnders Carlsson            CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(),
70659486a2dSAnders Carlsson            DeleteArgs, DeleteFD);
70759486a2dSAnders Carlsson }
70859486a2dSAnders Carlsson 
70959486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
71059486a2dSAnders Carlsson 
71159486a2dSAnders Carlsson   // Get at the argument before we performed the implicit conversion
71259486a2dSAnders Carlsson   // to void*.
71359486a2dSAnders Carlsson   const Expr *Arg = E->getArgument();
71459486a2dSAnders Carlsson   while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) {
71559486a2dSAnders Carlsson     if (ICE->getCastKind() != CastExpr::CK_UserDefinedConversion &&
71659486a2dSAnders Carlsson         ICE->getType()->isVoidPointerType())
71759486a2dSAnders Carlsson       Arg = ICE->getSubExpr();
71859486a2dSAnders Carlsson     else
71959486a2dSAnders Carlsson       break;
72059486a2dSAnders Carlsson   }
72159486a2dSAnders Carlsson 
72259486a2dSAnders Carlsson   QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
72359486a2dSAnders Carlsson 
72459486a2dSAnders Carlsson   llvm::Value *Ptr = EmitScalarExpr(Arg);
72559486a2dSAnders Carlsson 
72659486a2dSAnders Carlsson   // Null check the pointer.
72759486a2dSAnders Carlsson   llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
72859486a2dSAnders Carlsson   llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
72959486a2dSAnders Carlsson 
73059486a2dSAnders Carlsson   llvm::Value *IsNull =
73159486a2dSAnders Carlsson     Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()),
73259486a2dSAnders Carlsson                          "isnull");
73359486a2dSAnders Carlsson 
73459486a2dSAnders Carlsson   Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
73559486a2dSAnders Carlsson   EmitBlock(DeleteNotNull);
73659486a2dSAnders Carlsson 
73759486a2dSAnders Carlsson   bool ShouldCallDelete = true;
73859486a2dSAnders Carlsson 
73959486a2dSAnders Carlsson   // Call the destructor if necessary.
74059486a2dSAnders Carlsson   if (const RecordType *RT = DeleteTy->getAs<RecordType>()) {
74159486a2dSAnders Carlsson     if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl())) {
74259486a2dSAnders Carlsson       if (!RD->hasTrivialDestructor()) {
74359486a2dSAnders Carlsson         const CXXDestructorDecl *Dtor = RD->getDestructor(getContext());
74459486a2dSAnders Carlsson         if (E->isArrayForm()) {
74521122cf6SAnders Carlsson           llvm::Value *AllocatedObjectPtr;
74621122cf6SAnders Carlsson           llvm::Value *NumElements;
74721122cf6SAnders Carlsson           llvm::tie(AllocatedObjectPtr, NumElements) =
74821122cf6SAnders Carlsson             GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy);
749f5af8deaSAnders Carlsson 
75059486a2dSAnders Carlsson           EmitCXXAggrDestructorCall(Dtor, NumElements, Ptr);
751f5af8deaSAnders Carlsson         } else if (Dtor->isVirtual()) {
75259486a2dSAnders Carlsson           const llvm::Type *Ty =
75359486a2dSAnders Carlsson             CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(Dtor),
75459486a2dSAnders Carlsson                                            /*isVariadic=*/false);
75559486a2dSAnders Carlsson 
75659486a2dSAnders Carlsson           llvm::Value *Callee = BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty);
757e36a6b3eSAnders Carlsson           EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0,
758e36a6b3eSAnders Carlsson                             0, 0);
75959486a2dSAnders Carlsson 
76059486a2dSAnders Carlsson           // The dtor took care of deleting the object.
76159486a2dSAnders Carlsson           ShouldCallDelete = false;
76259486a2dSAnders Carlsson         } else
76359486a2dSAnders Carlsson           EmitCXXDestructorCall(Dtor, Dtor_Complete, Ptr);
76459486a2dSAnders Carlsson       }
76559486a2dSAnders Carlsson     }
76659486a2dSAnders Carlsson   }
76759486a2dSAnders Carlsson 
76859486a2dSAnders Carlsson   if (ShouldCallDelete)
76959486a2dSAnders Carlsson     EmitDeleteCall(E->getOperatorDelete(), Ptr, DeleteTy);
77059486a2dSAnders Carlsson 
77159486a2dSAnders Carlsson   EmitBlock(DeleteEnd);
77259486a2dSAnders Carlsson }
77359486a2dSAnders Carlsson 
77459486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
77559486a2dSAnders Carlsson   QualType Ty = E->getType();
77659486a2dSAnders Carlsson   const llvm::Type *LTy = ConvertType(Ty)->getPointerTo();
777fd7dfeb7SAnders Carlsson 
7783f4336cbSAnders Carlsson   if (E->isTypeOperand()) {
7793f4336cbSAnders Carlsson     llvm::Constant *TypeInfo =
7803f4336cbSAnders Carlsson       CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand());
7813f4336cbSAnders Carlsson     return Builder.CreateBitCast(TypeInfo, LTy);
7823f4336cbSAnders Carlsson   }
783fd7dfeb7SAnders Carlsson 
78459486a2dSAnders Carlsson   Expr *subE = E->getExprOperand();
78559486a2dSAnders Carlsson   Ty = subE->getType();
78659486a2dSAnders Carlsson   CanQualType CanTy = CGM.getContext().getCanonicalType(Ty);
78759486a2dSAnders Carlsson   Ty = CanTy.getUnqualifiedType().getNonReferenceType();
78859486a2dSAnders Carlsson   if (const RecordType *RT = Ty->getAs<RecordType>()) {
78959486a2dSAnders Carlsson     const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
79059486a2dSAnders Carlsson     if (RD->isPolymorphic()) {
79159486a2dSAnders Carlsson       // FIXME: if subE is an lvalue do
79259486a2dSAnders Carlsson       LValue Obj = EmitLValue(subE);
79359486a2dSAnders Carlsson       llvm::Value *This = Obj.getAddress();
79459486a2dSAnders Carlsson       LTy = LTy->getPointerTo()->getPointerTo();
79559486a2dSAnders Carlsson       llvm::Value *V = Builder.CreateBitCast(This, LTy);
79659486a2dSAnders Carlsson       // We need to do a zero check for *p, unless it has NonNullAttr.
79759486a2dSAnders Carlsson       // FIXME: PointerType->hasAttr<NonNullAttr>()
79859486a2dSAnders Carlsson       bool CanBeZero = false;
79959486a2dSAnders Carlsson       if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens()))
80059486a2dSAnders Carlsson         if (UO->getOpcode() == UnaryOperator::Deref)
80159486a2dSAnders Carlsson           CanBeZero = true;
80259486a2dSAnders Carlsson       if (CanBeZero) {
80359486a2dSAnders Carlsson         llvm::BasicBlock *NonZeroBlock = createBasicBlock();
80459486a2dSAnders Carlsson         llvm::BasicBlock *ZeroBlock = createBasicBlock();
80559486a2dSAnders Carlsson 
80659486a2dSAnders Carlsson         llvm::Value *Zero = llvm::Constant::getNullValue(LTy);
80759486a2dSAnders Carlsson         Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero),
80859486a2dSAnders Carlsson                              NonZeroBlock, ZeroBlock);
80959486a2dSAnders Carlsson         EmitBlock(ZeroBlock);
81059486a2dSAnders Carlsson         /// Call __cxa_bad_typeid
81159486a2dSAnders Carlsson         const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext);
81259486a2dSAnders Carlsson         const llvm::FunctionType *FTy;
81359486a2dSAnders Carlsson         FTy = llvm::FunctionType::get(ResultType, false);
81459486a2dSAnders Carlsson         llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid");
81559486a2dSAnders Carlsson         Builder.CreateCall(F)->setDoesNotReturn();
81659486a2dSAnders Carlsson         Builder.CreateUnreachable();
81759486a2dSAnders Carlsson         EmitBlock(NonZeroBlock);
81859486a2dSAnders Carlsson       }
81959486a2dSAnders Carlsson       V = Builder.CreateLoad(V, "vtable");
82059486a2dSAnders Carlsson       V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL);
82159486a2dSAnders Carlsson       V = Builder.CreateLoad(V);
82259486a2dSAnders Carlsson       return V;
82359486a2dSAnders Carlsson     }
82459486a2dSAnders Carlsson   }
8253f4336cbSAnders Carlsson   return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy);
82659486a2dSAnders Carlsson }
82759486a2dSAnders Carlsson 
82859486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V,
82959486a2dSAnders Carlsson                                               const CXXDynamicCastExpr *DCE) {
8303f4336cbSAnders Carlsson   QualType SrcTy = DCE->getSubExpr()->getType();
8313f4336cbSAnders Carlsson   QualType DestTy = DCE->getTypeAsWritten();
8323f4336cbSAnders Carlsson   QualType InnerType = DestTy->getPointeeType();
8333f4336cbSAnders Carlsson 
83459486a2dSAnders Carlsson   const llvm::Type *LTy = ConvertType(DCE->getType());
83559486a2dSAnders Carlsson 
83659486a2dSAnders Carlsson   bool CanBeZero = false;
83759486a2dSAnders Carlsson   bool ToVoid = false;
83859486a2dSAnders Carlsson   bool ThrowOnBad = false;
8393f4336cbSAnders Carlsson   if (DestTy->isPointerType()) {
84059486a2dSAnders Carlsson     // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this
84159486a2dSAnders Carlsson     CanBeZero = true;
84259486a2dSAnders Carlsson     if (InnerType->isVoidType())
84359486a2dSAnders Carlsson       ToVoid = true;
84459486a2dSAnders Carlsson   } else {
84559486a2dSAnders Carlsson     LTy = LTy->getPointerTo();
84659486a2dSAnders Carlsson     ThrowOnBad = true;
84759486a2dSAnders Carlsson   }
84859486a2dSAnders Carlsson 
8493f4336cbSAnders Carlsson   if (SrcTy->isPointerType() || SrcTy->isReferenceType())
8503f4336cbSAnders Carlsson     SrcTy = SrcTy->getPointeeType();
8513f4336cbSAnders Carlsson   SrcTy = SrcTy.getUnqualifiedType();
8523f4336cbSAnders Carlsson 
8530087bc85SAnders Carlsson   if (DestTy->isPointerType() || DestTy->isReferenceType())
8543f4336cbSAnders Carlsson     DestTy = DestTy->getPointeeType();
8553f4336cbSAnders Carlsson   DestTy = DestTy.getUnqualifiedType();
85659486a2dSAnders Carlsson 
85759486a2dSAnders Carlsson   llvm::BasicBlock *ContBlock = createBasicBlock();
85859486a2dSAnders Carlsson   llvm::BasicBlock *NullBlock = 0;
85959486a2dSAnders Carlsson   llvm::BasicBlock *NonZeroBlock = 0;
86059486a2dSAnders Carlsson   if (CanBeZero) {
86159486a2dSAnders Carlsson     NonZeroBlock = createBasicBlock();
86259486a2dSAnders Carlsson     NullBlock = createBasicBlock();
8633f4336cbSAnders Carlsson     Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock);
86459486a2dSAnders Carlsson     EmitBlock(NonZeroBlock);
86559486a2dSAnders Carlsson   }
86659486a2dSAnders Carlsson 
86759486a2dSAnders Carlsson   llvm::BasicBlock *BadCastBlock = 0;
86859486a2dSAnders Carlsson 
8693f4336cbSAnders Carlsson   const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
87059486a2dSAnders Carlsson 
87159486a2dSAnders Carlsson   // See if this is a dynamic_cast(void*)
87259486a2dSAnders Carlsson   if (ToVoid) {
87359486a2dSAnders Carlsson     llvm::Value *This = V;
87459486a2dSAnders Carlsson     V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo());
87559486a2dSAnders Carlsson     V = Builder.CreateLoad(V, "vtable");
87659486a2dSAnders Carlsson     V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL);
87759486a2dSAnders Carlsson     V = Builder.CreateLoad(V, "offset to top");
87859486a2dSAnders Carlsson     This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext));
87959486a2dSAnders Carlsson     V = Builder.CreateInBoundsGEP(This, V);
88059486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, LTy);
88159486a2dSAnders Carlsson   } else {
88259486a2dSAnders Carlsson     /// Call __dynamic_cast
88359486a2dSAnders Carlsson     const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext);
88459486a2dSAnders Carlsson     const llvm::FunctionType *FTy;
88559486a2dSAnders Carlsson     std::vector<const llvm::Type*> ArgTys;
88659486a2dSAnders Carlsson     const llvm::Type *PtrToInt8Ty
88759486a2dSAnders Carlsson       = llvm::Type::getInt8Ty(VMContext)->getPointerTo();
88859486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
88959486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
89059486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
89159486a2dSAnders Carlsson     ArgTys.push_back(PtrDiffTy);
89259486a2dSAnders Carlsson     FTy = llvm::FunctionType::get(ResultType, ArgTys, false);
89359486a2dSAnders Carlsson 
89459486a2dSAnders Carlsson     // FIXME: Calculate better hint.
89559486a2dSAnders Carlsson     llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL);
8963f4336cbSAnders Carlsson 
8973f4336cbSAnders Carlsson     assert(SrcTy->isRecordType() && "Src type must be record type!");
8983f4336cbSAnders Carlsson     assert(DestTy->isRecordType() && "Dest type must be record type!");
8993f4336cbSAnders Carlsson 
900247894b3SDouglas Gregor     llvm::Value *SrcArg
901247894b3SDouglas Gregor       = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType());
902247894b3SDouglas Gregor     llvm::Value *DestArg
903247894b3SDouglas Gregor       = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType());
9043f4336cbSAnders Carlsson 
90559486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, PtrToInt8Ty);
90659486a2dSAnders Carlsson     V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"),
9073f4336cbSAnders Carlsson                             V, SrcArg, DestArg, hint);
90859486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, LTy);
90959486a2dSAnders Carlsson 
91059486a2dSAnders Carlsson     if (ThrowOnBad) {
91159486a2dSAnders Carlsson       BadCastBlock = createBasicBlock();
91259486a2dSAnders Carlsson 
9133f4336cbSAnders Carlsson       Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock);
91459486a2dSAnders Carlsson       EmitBlock(BadCastBlock);
91559486a2dSAnders Carlsson       /// Call __cxa_bad_cast
91659486a2dSAnders Carlsson       ResultType = llvm::Type::getVoidTy(VMContext);
91759486a2dSAnders Carlsson       const llvm::FunctionType *FBadTy;
91859486a2dSAnders Carlsson       FBadTy = llvm::FunctionType::get(ResultType, false);
91959486a2dSAnders Carlsson       llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast");
92059486a2dSAnders Carlsson       Builder.CreateCall(F)->setDoesNotReturn();
92159486a2dSAnders Carlsson       Builder.CreateUnreachable();
92259486a2dSAnders Carlsson     }
92359486a2dSAnders Carlsson   }
92459486a2dSAnders Carlsson 
92559486a2dSAnders Carlsson   if (CanBeZero) {
92659486a2dSAnders Carlsson     Builder.CreateBr(ContBlock);
92759486a2dSAnders Carlsson     EmitBlock(NullBlock);
92859486a2dSAnders Carlsson     Builder.CreateBr(ContBlock);
92959486a2dSAnders Carlsson   }
93059486a2dSAnders Carlsson   EmitBlock(ContBlock);
93159486a2dSAnders Carlsson   if (CanBeZero) {
93259486a2dSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(LTy);
93359486a2dSAnders Carlsson     PHI->reserveOperandSpace(2);
93459486a2dSAnders Carlsson     PHI->addIncoming(V, NonZeroBlock);
93559486a2dSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock);
93659486a2dSAnders Carlsson     V = PHI;
93759486a2dSAnders Carlsson   }
93859486a2dSAnders Carlsson 
93959486a2dSAnders Carlsson   return V;
94059486a2dSAnders Carlsson }
941