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 
1491bbb554SDevang Patel #include "clang/Frontend/CodeGenOptions.h"
1559486a2dSAnders Carlsson #include "CodeGenFunction.h"
165d865c32SJohn McCall #include "CGCXXABI.h"
1760d215b6SFariborz Jahanian #include "CGObjCRuntime.h"
1891bbb554SDevang Patel #include "CGDebugInfo.h"
1926008e07SChris Lattner #include "llvm/Intrinsics.h"
20bbe277c4SAnders Carlsson #include "llvm/Support/CallSite.h"
21bbe277c4SAnders Carlsson 
2259486a2dSAnders Carlsson using namespace clang;
2359486a2dSAnders Carlsson using namespace CodeGen;
2459486a2dSAnders Carlsson 
2527da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD,
2627da15baSAnders Carlsson                                           llvm::Value *Callee,
2727da15baSAnders Carlsson                                           ReturnValueSlot ReturnValue,
2827da15baSAnders Carlsson                                           llvm::Value *This,
29e36a6b3eSAnders Carlsson                                           llvm::Value *VTT,
3027da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgBeg,
3127da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgEnd) {
3227da15baSAnders Carlsson   assert(MD->isInstance() &&
3327da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
3427da15baSAnders Carlsson 
3527da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
3627da15baSAnders Carlsson 
3727da15baSAnders Carlsson   CallArgList Args;
3827da15baSAnders Carlsson 
3927da15baSAnders Carlsson   // Push the this ptr.
4043dca6a8SEli Friedman   Args.add(RValue::get(This), MD->getThisType(getContext()));
4127da15baSAnders Carlsson 
42e36a6b3eSAnders Carlsson   // If there is a VTT parameter, emit it.
43e36a6b3eSAnders Carlsson   if (VTT) {
44e36a6b3eSAnders Carlsson     QualType T = getContext().getPointerType(getContext().VoidPtrTy);
4543dca6a8SEli Friedman     Args.add(RValue::get(VTT), T);
46e36a6b3eSAnders Carlsson   }
47e36a6b3eSAnders Carlsson 
4827da15baSAnders Carlsson   // And the rest of the call args
4927da15baSAnders Carlsson   EmitCallArgs(Args, FPT, ArgBeg, ArgEnd);
5027da15baSAnders Carlsson 
51ab26cfa5SJohn McCall   QualType ResultType = FPT->getResultType();
5299cc30c3STilmann Scheller   return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args,
5399cc30c3STilmann Scheller                                                  FPT->getExtInfo()),
54c50c27ccSRafael Espindola                   Callee, ReturnValue, Args, MD);
5527da15baSAnders Carlsson }
5627da15baSAnders Carlsson 
571ae64c5aSAnders Carlsson static const CXXRecordDecl *getMostDerivedClassDecl(const Expr *Base) {
586b3afd7dSAnders Carlsson   const Expr *E = Base;
596b3afd7dSAnders Carlsson 
606b3afd7dSAnders Carlsson   while (true) {
616b3afd7dSAnders Carlsson     E = E->IgnoreParens();
626b3afd7dSAnders Carlsson     if (const CastExpr *CE = dyn_cast<CastExpr>(E)) {
636b3afd7dSAnders Carlsson       if (CE->getCastKind() == CK_DerivedToBase ||
646b3afd7dSAnders Carlsson           CE->getCastKind() == CK_UncheckedDerivedToBase ||
656b3afd7dSAnders Carlsson           CE->getCastKind() == CK_NoOp) {
666b3afd7dSAnders Carlsson         E = CE->getSubExpr();
676b3afd7dSAnders Carlsson         continue;
686b3afd7dSAnders Carlsson       }
696b3afd7dSAnders Carlsson     }
706b3afd7dSAnders Carlsson 
716b3afd7dSAnders Carlsson     break;
726b3afd7dSAnders Carlsson   }
736b3afd7dSAnders Carlsson 
746b3afd7dSAnders Carlsson   QualType DerivedType = E->getType();
751ae64c5aSAnders Carlsson   if (const PointerType *PTy = DerivedType->getAs<PointerType>())
761ae64c5aSAnders Carlsson     DerivedType = PTy->getPointeeType();
771ae64c5aSAnders Carlsson 
781ae64c5aSAnders Carlsson   return cast<CXXRecordDecl>(DerivedType->castAs<RecordType>()->getDecl());
791ae64c5aSAnders Carlsson }
801ae64c5aSAnders Carlsson 
81c53d9e83SAnders Carlsson // FIXME: Ideally Expr::IgnoreParenNoopCasts should do this, but it doesn't do
82c53d9e83SAnders Carlsson // quite what we want.
83c53d9e83SAnders Carlsson static const Expr *skipNoOpCastsAndParens(const Expr *E) {
84c53d9e83SAnders Carlsson   while (true) {
85c53d9e83SAnders Carlsson     if (const ParenExpr *PE = dyn_cast<ParenExpr>(E)) {
86c53d9e83SAnders Carlsson       E = PE->getSubExpr();
87c53d9e83SAnders Carlsson       continue;
88c53d9e83SAnders Carlsson     }
89c53d9e83SAnders Carlsson 
90c53d9e83SAnders Carlsson     if (const CastExpr *CE = dyn_cast<CastExpr>(E)) {
91c53d9e83SAnders Carlsson       if (CE->getCastKind() == CK_NoOp) {
92c53d9e83SAnders Carlsson         E = CE->getSubExpr();
93c53d9e83SAnders Carlsson         continue;
94c53d9e83SAnders Carlsson       }
95c53d9e83SAnders Carlsson     }
96c53d9e83SAnders Carlsson     if (const UnaryOperator *UO = dyn_cast<UnaryOperator>(E)) {
97c53d9e83SAnders Carlsson       if (UO->getOpcode() == UO_Extension) {
98c53d9e83SAnders Carlsson         E = UO->getSubExpr();
99c53d9e83SAnders Carlsson         continue;
100c53d9e83SAnders Carlsson       }
101c53d9e83SAnders Carlsson     }
102c53d9e83SAnders Carlsson     return E;
103c53d9e83SAnders Carlsson   }
104c53d9e83SAnders Carlsson }
105c53d9e83SAnders Carlsson 
10627da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given
10727da15baSAnders Carlsson /// expr can be devirtualized.
108252a47f6SFariborz Jahanian static bool canDevirtualizeMemberFunctionCalls(ASTContext &Context,
109252a47f6SFariborz Jahanian                                                const Expr *Base,
110a7911fa3SAnders Carlsson                                                const CXXMethodDecl *MD) {
111a7911fa3SAnders Carlsson 
1121ae64c5aSAnders Carlsson   // When building with -fapple-kext, all calls must go through the vtable since
1131ae64c5aSAnders Carlsson   // the kernel linker can do runtime patching of vtables.
114252a47f6SFariborz Jahanian   if (Context.getLangOptions().AppleKext)
115252a47f6SFariborz Jahanian     return false;
116252a47f6SFariborz Jahanian 
1171ae64c5aSAnders Carlsson   // If the most derived class is marked final, we know that no subclass can
1181ae64c5aSAnders Carlsson   // override this member function and so we can devirtualize it. For example:
1191ae64c5aSAnders Carlsson   //
1201ae64c5aSAnders Carlsson   // struct A { virtual void f(); }
1211ae64c5aSAnders Carlsson   // struct B final : A { };
1221ae64c5aSAnders Carlsson   //
1231ae64c5aSAnders Carlsson   // void f(B *b) {
1241ae64c5aSAnders Carlsson   //   b->f();
1251ae64c5aSAnders Carlsson   // }
1261ae64c5aSAnders Carlsson   //
1271ae64c5aSAnders Carlsson   const CXXRecordDecl *MostDerivedClassDecl = getMostDerivedClassDecl(Base);
1281ae64c5aSAnders Carlsson   if (MostDerivedClassDecl->hasAttr<FinalAttr>())
1291ae64c5aSAnders Carlsson     return true;
1301ae64c5aSAnders Carlsson 
13119588aa4SAnders Carlsson   // If the member function is marked 'final', we know that it can't be
132b00c2144SAnders Carlsson   // overridden and can therefore devirtualize it.
1331eb95961SAnders Carlsson   if (MD->hasAttr<FinalAttr>())
134a7911fa3SAnders Carlsson     return true;
135a7911fa3SAnders Carlsson 
13619588aa4SAnders Carlsson   // Similarly, if the class itself is marked 'final' it can't be overridden
13719588aa4SAnders Carlsson   // and we can therefore devirtualize the member function call.
1381eb95961SAnders Carlsson   if (MD->getParent()->hasAttr<FinalAttr>())
139b00c2144SAnders Carlsson     return true;
140b00c2144SAnders Carlsson 
141c53d9e83SAnders Carlsson   Base = skipNoOpCastsAndParens(Base);
14227da15baSAnders Carlsson   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) {
14327da15baSAnders Carlsson     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
14427da15baSAnders Carlsson       // This is a record decl. We know the type and can devirtualize it.
14527da15baSAnders Carlsson       return VD->getType()->isRecordType();
14627da15baSAnders Carlsson     }
14727da15baSAnders Carlsson 
14827da15baSAnders Carlsson     return false;
14927da15baSAnders Carlsson   }
15027da15baSAnders Carlsson 
15127da15baSAnders Carlsson   // We can always devirtualize calls on temporary object expressions.
152a682427eSEli Friedman   if (isa<CXXConstructExpr>(Base))
15327da15baSAnders Carlsson     return true;
15427da15baSAnders Carlsson 
15527da15baSAnders Carlsson   // And calls on bound temporaries.
15627da15baSAnders Carlsson   if (isa<CXXBindTemporaryExpr>(Base))
15727da15baSAnders Carlsson     return true;
15827da15baSAnders Carlsson 
15927da15baSAnders Carlsson   // Check if this is a call expr that returns a record type.
16027da15baSAnders Carlsson   if (const CallExpr *CE = dyn_cast<CallExpr>(Base))
16127da15baSAnders Carlsson     return CE->getCallReturnType()->isRecordType();
16227da15baSAnders Carlsson 
16327da15baSAnders Carlsson   // We can't devirtualize the call.
16427da15baSAnders Carlsson   return false;
16527da15baSAnders Carlsson }
16627da15baSAnders Carlsson 
16764225794SFrancois Pichet // Note: This function also emit constructor calls to support a MSVC
16864225794SFrancois Pichet // extensions allowing explicit constructor function call.
16927da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
17027da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
1712d2e8707SJohn McCall   const Expr *callee = CE->getCallee()->IgnoreParens();
1722d2e8707SJohn McCall 
1732d2e8707SJohn McCall   if (isa<BinaryOperator>(callee))
17427da15baSAnders Carlsson     return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
17527da15baSAnders Carlsson 
1762d2e8707SJohn McCall   const MemberExpr *ME = cast<MemberExpr>(callee);
17727da15baSAnders Carlsson   const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
17827da15baSAnders Carlsson 
17991bbb554SDevang Patel   CGDebugInfo *DI = getDebugInfo();
180401c916cSDevang Patel   if (DI && CGM.getCodeGenOpts().LimitDebugInfo
181401c916cSDevang Patel       && !isa<CallExpr>(ME->getBase())) {
18291bbb554SDevang Patel     QualType PQTy = ME->getBase()->IgnoreParenImpCasts()->getType();
18391bbb554SDevang Patel     if (const PointerType * PTy = dyn_cast<PointerType>(PQTy)) {
18491bbb554SDevang Patel       DI->getOrCreateRecordType(PTy->getPointeeType(),
18591bbb554SDevang Patel                                 MD->getParent()->getLocation());
18691bbb554SDevang Patel     }
18791bbb554SDevang Patel   }
18891bbb554SDevang Patel 
18927da15baSAnders Carlsson   if (MD->isStatic()) {
19027da15baSAnders Carlsson     // The method is static, emit it as we would a regular call.
19127da15baSAnders Carlsson     llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
19227da15baSAnders Carlsson     return EmitCall(getContext().getPointerType(MD->getType()), Callee,
19327da15baSAnders Carlsson                     ReturnValue, CE->arg_begin(), CE->arg_end());
19427da15baSAnders Carlsson   }
19527da15baSAnders Carlsson 
1960d635f53SJohn McCall   // Compute the object pointer.
19727da15baSAnders Carlsson   llvm::Value *This;
19827da15baSAnders Carlsson   if (ME->isArrow())
19927da15baSAnders Carlsson     This = EmitScalarExpr(ME->getBase());
200f93ac894SFariborz Jahanian   else
201e26a872bSJohn McCall     This = EmitLValue(ME->getBase()).getAddress();
20227da15baSAnders Carlsson 
2030d635f53SJohn McCall   if (MD->isTrivial()) {
2040d635f53SJohn McCall     if (isa<CXXDestructorDecl>(MD)) return RValue::get(0);
20564225794SFrancois Pichet     if (isa<CXXConstructorDecl>(MD) &&
20664225794SFrancois Pichet         cast<CXXConstructorDecl>(MD)->isDefaultConstructor())
20764225794SFrancois Pichet       return RValue::get(0);
2080d635f53SJohn McCall 
20922653bacSSebastian Redl     if (MD->isCopyAssignmentOperator() || MD->isMoveAssignmentOperator()) {
21022653bacSSebastian Redl       // We don't like to generate the trivial copy/move assignment operator
21122653bacSSebastian Redl       // when it isn't necessary; just produce the proper effect here.
21227da15baSAnders Carlsson       llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
21327da15baSAnders Carlsson       EmitAggregateCopy(This, RHS, CE->getType());
21427da15baSAnders Carlsson       return RValue::get(This);
21527da15baSAnders Carlsson     }
21627da15baSAnders Carlsson 
21764225794SFrancois Pichet     if (isa<CXXConstructorDecl>(MD) &&
21822653bacSSebastian Redl         cast<CXXConstructorDecl>(MD)->isCopyOrMoveConstructor()) {
21922653bacSSebastian Redl       // Trivial move and copy ctor are the same.
22064225794SFrancois Pichet       llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
22164225794SFrancois Pichet       EmitSynthesizedCXXCopyCtorCall(cast<CXXConstructorDecl>(MD), This, RHS,
22264225794SFrancois Pichet                                      CE->arg_begin(), CE->arg_end());
22364225794SFrancois Pichet       return RValue::get(This);
22464225794SFrancois Pichet     }
22564225794SFrancois Pichet     llvm_unreachable("unknown trivial member function");
22664225794SFrancois Pichet   }
22764225794SFrancois Pichet 
2280d635f53SJohn McCall   // Compute the function type we're calling.
22964225794SFrancois Pichet   const CGFunctionInfo *FInfo = 0;
23064225794SFrancois Pichet   if (isa<CXXDestructorDecl>(MD))
23164225794SFrancois Pichet     FInfo = &CGM.getTypes().getFunctionInfo(cast<CXXDestructorDecl>(MD),
23264225794SFrancois Pichet                                            Dtor_Complete);
23364225794SFrancois Pichet   else if (isa<CXXConstructorDecl>(MD))
23464225794SFrancois Pichet     FInfo = &CGM.getTypes().getFunctionInfo(cast<CXXConstructorDecl>(MD),
23564225794SFrancois Pichet                                             Ctor_Complete);
23664225794SFrancois Pichet   else
23764225794SFrancois Pichet     FInfo = &CGM.getTypes().getFunctionInfo(MD);
2380d635f53SJohn McCall 
2390d635f53SJohn McCall   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
2402192fe50SChris Lattner   llvm::Type *Ty
24164225794SFrancois Pichet     = CGM.getTypes().GetFunctionType(*FInfo, FPT->isVariadic());
2420d635f53SJohn McCall 
24327da15baSAnders Carlsson   // C++ [class.virtual]p12:
24427da15baSAnders Carlsson   //   Explicit qualification with the scope operator (5.1) suppresses the
24527da15baSAnders Carlsson   //   virtual call mechanism.
24627da15baSAnders Carlsson   //
24727da15baSAnders Carlsson   // We also don't emit a virtual call if the base expression has a record type
24827da15baSAnders Carlsson   // because then we know what the type is.
24947609b08SFariborz Jahanian   bool UseVirtualCall;
25047609b08SFariborz Jahanian   UseVirtualCall = MD->isVirtual() && !ME->hasQualifier()
251252a47f6SFariborz Jahanian                    && !canDevirtualizeMemberFunctionCalls(getContext(),
252252a47f6SFariborz Jahanian                                                           ME->getBase(), MD);
25327da15baSAnders Carlsson   llvm::Value *Callee;
2540d635f53SJohn McCall   if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(MD)) {
2550d635f53SJohn McCall     if (UseVirtualCall) {
2560d635f53SJohn McCall       Callee = BuildVirtualCall(Dtor, Dtor_Complete, This, Ty);
25727da15baSAnders Carlsson     } else {
258265c325eSFariborz Jahanian       if (getContext().getLangOptions().AppleKext &&
259265c325eSFariborz Jahanian           MD->isVirtual() &&
260265c325eSFariborz Jahanian           ME->hasQualifier())
2617f6f81baSFariborz Jahanian         Callee = BuildAppleKextVirtualCall(MD, ME->getQualifier(), Ty);
262265c325eSFariborz Jahanian       else
2630d635f53SJohn McCall         Callee = CGM.GetAddrOfFunction(GlobalDecl(Dtor, Dtor_Complete), Ty);
26427da15baSAnders Carlsson     }
26564225794SFrancois Pichet   } else if (const CXXConstructorDecl *Ctor =
26664225794SFrancois Pichet                dyn_cast<CXXConstructorDecl>(MD)) {
26764225794SFrancois Pichet     Callee = CGM.GetAddrOfFunction(GlobalDecl(Ctor, Ctor_Complete), Ty);
2680d635f53SJohn McCall   } else if (UseVirtualCall) {
26927da15baSAnders Carlsson       Callee = BuildVirtualCall(MD, This, Ty);
27027da15baSAnders Carlsson   } else {
271252a47f6SFariborz Jahanian     if (getContext().getLangOptions().AppleKext &&
2729f9438b3SFariborz Jahanian         MD->isVirtual() &&
273252a47f6SFariborz Jahanian         ME->hasQualifier())
2747f6f81baSFariborz Jahanian       Callee = BuildAppleKextVirtualCall(MD, ME->getQualifier(), Ty);
275252a47f6SFariborz Jahanian     else
27627da15baSAnders Carlsson       Callee = CGM.GetAddrOfFunction(MD, Ty);
27727da15baSAnders Carlsson   }
27827da15baSAnders Carlsson 
279e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
28027da15baSAnders Carlsson                            CE->arg_begin(), CE->arg_end());
28127da15baSAnders Carlsson }
28227da15baSAnders Carlsson 
28327da15baSAnders Carlsson RValue
28427da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
28527da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
28627da15baSAnders Carlsson   const BinaryOperator *BO =
28727da15baSAnders Carlsson       cast<BinaryOperator>(E->getCallee()->IgnoreParens());
28827da15baSAnders Carlsson   const Expr *BaseExpr = BO->getLHS();
28927da15baSAnders Carlsson   const Expr *MemFnExpr = BO->getRHS();
29027da15baSAnders Carlsson 
29127da15baSAnders Carlsson   const MemberPointerType *MPT =
2920009fcc3SJohn McCall     MemFnExpr->getType()->castAs<MemberPointerType>();
293475999dcSJohn McCall 
29427da15baSAnders Carlsson   const FunctionProtoType *FPT =
2950009fcc3SJohn McCall     MPT->getPointeeType()->castAs<FunctionProtoType>();
29627da15baSAnders Carlsson   const CXXRecordDecl *RD =
29727da15baSAnders Carlsson     cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
29827da15baSAnders Carlsson 
29927da15baSAnders Carlsson   // Get the member function pointer.
300a1dee530SJohn McCall   llvm::Value *MemFnPtr = EmitScalarExpr(MemFnExpr);
30127da15baSAnders Carlsson 
30227da15baSAnders Carlsson   // Emit the 'this' pointer.
30327da15baSAnders Carlsson   llvm::Value *This;
30427da15baSAnders Carlsson 
305e302792bSJohn McCall   if (BO->getOpcode() == BO_PtrMemI)
30627da15baSAnders Carlsson     This = EmitScalarExpr(BaseExpr);
30727da15baSAnders Carlsson   else
30827da15baSAnders Carlsson     This = EmitLValue(BaseExpr).getAddress();
30927da15baSAnders Carlsson 
310475999dcSJohn McCall   // Ask the ABI to load the callee.  Note that This is modified.
311475999dcSJohn McCall   llvm::Value *Callee =
312ad7c5c16SJohn McCall     CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(*this, This, MemFnPtr, MPT);
31327da15baSAnders Carlsson 
31427da15baSAnders Carlsson   CallArgList Args;
31527da15baSAnders Carlsson 
31627da15baSAnders Carlsson   QualType ThisType =
31727da15baSAnders Carlsson     getContext().getPointerType(getContext().getTagDeclType(RD));
31827da15baSAnders Carlsson 
31927da15baSAnders Carlsson   // Push the this ptr.
32043dca6a8SEli Friedman   Args.add(RValue::get(This), ThisType);
32127da15baSAnders Carlsson 
32227da15baSAnders Carlsson   // And the rest of the call args
32327da15baSAnders Carlsson   EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
3240009fcc3SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(Args, FPT), Callee,
32599cc30c3STilmann Scheller                   ReturnValue, Args);
32627da15baSAnders Carlsson }
32727da15baSAnders Carlsson 
32827da15baSAnders Carlsson RValue
32927da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
33027da15baSAnders Carlsson                                                const CXXMethodDecl *MD,
33127da15baSAnders Carlsson                                                ReturnValueSlot ReturnValue) {
33227da15baSAnders Carlsson   assert(MD->isInstance() &&
33327da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
334e26a872bSJohn McCall   LValue LV = EmitLValue(E->getArg(0));
335e26a872bSJohn McCall   llvm::Value *This = LV.getAddress();
336e26a872bSJohn McCall 
337146b8e9aSDouglas Gregor   if ((MD->isCopyAssignmentOperator() || MD->isMoveAssignmentOperator()) &&
338146b8e9aSDouglas Gregor       MD->isTrivial()) {
33927da15baSAnders Carlsson     llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
34027da15baSAnders Carlsson     QualType Ty = E->getType();
34127da15baSAnders Carlsson     EmitAggregateCopy(This, Src, Ty);
34227da15baSAnders Carlsson     return RValue::get(This);
34327da15baSAnders Carlsson   }
34427da15baSAnders Carlsson 
345c36783e8SAnders Carlsson   llvm::Value *Callee = EmitCXXOperatorMemberCallee(E, MD, This);
346e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
34727da15baSAnders Carlsson                            E->arg_begin() + 1, E->arg_end());
34827da15baSAnders Carlsson }
34927da15baSAnders Carlsson 
35027da15baSAnders Carlsson void
3517a626f63SJohn McCall CodeGenFunction::EmitCXXConstructExpr(const CXXConstructExpr *E,
3527a626f63SJohn McCall                                       AggValueSlot Dest) {
3537a626f63SJohn McCall   assert(!Dest.isIgnored() && "Must have a destination!");
35427da15baSAnders Carlsson   const CXXConstructorDecl *CD = E->getConstructor();
355630c76efSDouglas Gregor 
356630c76efSDouglas Gregor   // If we require zero initialization before (or instead of) calling the
357630c76efSDouglas Gregor   // constructor, as can be the case with a non-user-provided default
35803535265SArgyrios Kyrtzidis   // constructor, emit the zero initialization now, unless destination is
35903535265SArgyrios Kyrtzidis   // already zeroed.
36003535265SArgyrios Kyrtzidis   if (E->requiresZeroInitialization() && !Dest.isZeroed())
3617a626f63SJohn McCall     EmitNullInitialization(Dest.getAddr(), E->getType());
362630c76efSDouglas Gregor 
363630c76efSDouglas Gregor   // If this is a call to a trivial default constructor, do nothing.
364630c76efSDouglas Gregor   if (CD->isTrivial() && CD->isDefaultConstructor())
36527da15baSAnders Carlsson     return;
366630c76efSDouglas Gregor 
3678ea46b66SJohn McCall   // Elide the constructor if we're constructing from a temporary.
3688ea46b66SJohn McCall   // The temporary check is required because Sema sets this on NRVO
3698ea46b66SJohn McCall   // returns.
37027da15baSAnders Carlsson   if (getContext().getLangOptions().ElideConstructors && E->isElidable()) {
3718ea46b66SJohn McCall     assert(getContext().hasSameUnqualifiedType(E->getType(),
3728ea46b66SJohn McCall                                                E->getArg(0)->getType()));
3737a626f63SJohn McCall     if (E->getArg(0)->isTemporaryObject(getContext(), CD->getParent())) {
3747a626f63SJohn McCall       EmitAggExpr(E->getArg(0), Dest);
37527da15baSAnders Carlsson       return;
37627da15baSAnders Carlsson     }
377222cf0efSDouglas Gregor   }
378630c76efSDouglas Gregor 
379f677a8e9SJohn McCall   if (const ConstantArrayType *arrayType
380f677a8e9SJohn McCall         = getContext().getAsConstantArrayType(E->getType())) {
381f677a8e9SJohn McCall     EmitCXXAggrConstructorCall(CD, arrayType, Dest.getAddr(),
38227da15baSAnders Carlsson                                E->arg_begin(), E->arg_end());
383f677a8e9SJohn McCall   } else {
384bceca20aSCameron Esfahani     CXXCtorType Type = Ctor_Complete;
385271c3681SAlexis Hunt     bool ForVirtualBase = false;
386271c3681SAlexis Hunt 
387271c3681SAlexis Hunt     switch (E->getConstructionKind()) {
388271c3681SAlexis Hunt      case CXXConstructExpr::CK_Delegating:
38961bc1737SAlexis Hunt       // We should be emitting a constructor; GlobalDecl will assert this
39061bc1737SAlexis Hunt       Type = CurGD.getCtorType();
391271c3681SAlexis Hunt       break;
39261bc1737SAlexis Hunt 
393271c3681SAlexis Hunt      case CXXConstructExpr::CK_Complete:
394271c3681SAlexis Hunt       Type = Ctor_Complete;
395271c3681SAlexis Hunt       break;
396271c3681SAlexis Hunt 
397271c3681SAlexis Hunt      case CXXConstructExpr::CK_VirtualBase:
398271c3681SAlexis Hunt       ForVirtualBase = true;
399271c3681SAlexis Hunt       // fall-through
400271c3681SAlexis Hunt 
401271c3681SAlexis Hunt      case CXXConstructExpr::CK_NonVirtualBase:
402271c3681SAlexis Hunt       Type = Ctor_Base;
403271c3681SAlexis Hunt     }
404e11f9ce9SAnders Carlsson 
40527da15baSAnders Carlsson     // Call the constructor.
4067a626f63SJohn McCall     EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest.getAddr(),
40727da15baSAnders Carlsson                            E->arg_begin(), E->arg_end());
40827da15baSAnders Carlsson   }
409e11f9ce9SAnders Carlsson }
41027da15baSAnders Carlsson 
411e988bdacSFariborz Jahanian void
412e988bdacSFariborz Jahanian CodeGenFunction::EmitSynthesizedCXXCopyCtor(llvm::Value *Dest,
413e988bdacSFariborz Jahanian                                             llvm::Value *Src,
41450198098SFariborz Jahanian                                             const Expr *Exp) {
4155d413781SJohn McCall   if (const ExprWithCleanups *E = dyn_cast<ExprWithCleanups>(Exp))
416e988bdacSFariborz Jahanian     Exp = E->getSubExpr();
417e988bdacSFariborz Jahanian   assert(isa<CXXConstructExpr>(Exp) &&
418e988bdacSFariborz Jahanian          "EmitSynthesizedCXXCopyCtor - unknown copy ctor expr");
419e988bdacSFariborz Jahanian   const CXXConstructExpr* E = cast<CXXConstructExpr>(Exp);
420e988bdacSFariborz Jahanian   const CXXConstructorDecl *CD = E->getConstructor();
421e988bdacSFariborz Jahanian   RunCleanupsScope Scope(*this);
422e988bdacSFariborz Jahanian 
423e988bdacSFariborz Jahanian   // If we require zero initialization before (or instead of) calling the
424e988bdacSFariborz Jahanian   // constructor, as can be the case with a non-user-provided default
425e988bdacSFariborz Jahanian   // constructor, emit the zero initialization now.
426e988bdacSFariborz Jahanian   // FIXME. Do I still need this for a copy ctor synthesis?
427e988bdacSFariborz Jahanian   if (E->requiresZeroInitialization())
428e988bdacSFariborz Jahanian     EmitNullInitialization(Dest, E->getType());
429e988bdacSFariborz Jahanian 
43099da11cfSChandler Carruth   assert(!getContext().getAsConstantArrayType(E->getType())
43199da11cfSChandler Carruth          && "EmitSynthesizedCXXCopyCtor - Copied-in Array");
432e988bdacSFariborz Jahanian   EmitSynthesizedCXXCopyCtorCall(CD, Dest, Src,
433e988bdacSFariborz Jahanian                                  E->arg_begin(), E->arg_end());
434e988bdacSFariborz Jahanian }
435e988bdacSFariborz Jahanian 
4368ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF,
4378ed55a54SJohn McCall                                         const CXXNewExpr *E) {
43821122cf6SAnders Carlsson   if (!E->isArray())
4393eb55cfeSKen Dyck     return CharUnits::Zero();
44021122cf6SAnders Carlsson 
4417ec4b434SJohn McCall   // No cookie is required if the operator new[] being used is the
4427ec4b434SJohn McCall   // reserved placement operator new[].
4437ec4b434SJohn McCall   if (E->getOperatorNew()->isReservedGlobalPlacementOperator())
4443eb55cfeSKen Dyck     return CharUnits::Zero();
445399f499fSAnders Carlsson 
446284c48ffSJohn McCall   return CGF.CGM.getCXXABI().GetArrayCookieSize(E);
44759486a2dSAnders Carlsson }
44859486a2dSAnders Carlsson 
449036f2f6bSJohn McCall static llvm::Value *EmitCXXNewAllocSize(CodeGenFunction &CGF,
450036f2f6bSJohn McCall                                         const CXXNewExpr *e,
451036f2f6bSJohn McCall                                         llvm::Value *&numElements,
452036f2f6bSJohn McCall                                         llvm::Value *&sizeWithoutCookie) {
453036f2f6bSJohn McCall   QualType type = e->getAllocatedType();
45459486a2dSAnders Carlsson 
455036f2f6bSJohn McCall   if (!e->isArray()) {
456036f2f6bSJohn McCall     CharUnits typeSize = CGF.getContext().getTypeSizeInChars(type);
457036f2f6bSJohn McCall     sizeWithoutCookie
458036f2f6bSJohn McCall       = llvm::ConstantInt::get(CGF.SizeTy, typeSize.getQuantity());
459036f2f6bSJohn McCall     return sizeWithoutCookie;
46005fc5be3SDouglas Gregor   }
46159486a2dSAnders Carlsson 
462036f2f6bSJohn McCall   // The width of size_t.
463036f2f6bSJohn McCall   unsigned sizeWidth = CGF.SizeTy->getBitWidth();
464036f2f6bSJohn McCall 
4658ed55a54SJohn McCall   // Figure out the cookie size.
466036f2f6bSJohn McCall   llvm::APInt cookieSize(sizeWidth,
467036f2f6bSJohn McCall                          CalculateCookiePadding(CGF, e).getQuantity());
4688ed55a54SJohn McCall 
46959486a2dSAnders Carlsson   // Emit the array size expression.
4707648fb46SArgyrios Kyrtzidis   // We multiply the size of all dimensions for NumElements.
4717648fb46SArgyrios Kyrtzidis   // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6.
472036f2f6bSJohn McCall   numElements = CGF.EmitScalarExpr(e->getArraySize());
473036f2f6bSJohn McCall   assert(isa<llvm::IntegerType>(numElements->getType()));
4748ed55a54SJohn McCall 
475036f2f6bSJohn McCall   // The number of elements can be have an arbitrary integer type;
476036f2f6bSJohn McCall   // essentially, we need to multiply it by a constant factor, add a
477036f2f6bSJohn McCall   // cookie size, and verify that the result is representable as a
478036f2f6bSJohn McCall   // size_t.  That's just a gloss, though, and it's wrong in one
479036f2f6bSJohn McCall   // important way: if the count is negative, it's an error even if
480036f2f6bSJohn McCall   // the cookie size would bring the total size >= 0.
4816ab2fa8fSDouglas Gregor   bool isSigned
4826ab2fa8fSDouglas Gregor     = e->getArraySize()->getType()->isSignedIntegerOrEnumerationType();
4832192fe50SChris Lattner   llvm::IntegerType *numElementsType
484036f2f6bSJohn McCall     = cast<llvm::IntegerType>(numElements->getType());
485036f2f6bSJohn McCall   unsigned numElementsWidth = numElementsType->getBitWidth();
486036f2f6bSJohn McCall 
487036f2f6bSJohn McCall   // Compute the constant factor.
488036f2f6bSJohn McCall   llvm::APInt arraySizeMultiplier(sizeWidth, 1);
4897648fb46SArgyrios Kyrtzidis   while (const ConstantArrayType *CAT
490036f2f6bSJohn McCall              = CGF.getContext().getAsConstantArrayType(type)) {
491036f2f6bSJohn McCall     type = CAT->getElementType();
492036f2f6bSJohn McCall     arraySizeMultiplier *= CAT->getSize();
4937648fb46SArgyrios Kyrtzidis   }
49459486a2dSAnders Carlsson 
495036f2f6bSJohn McCall   CharUnits typeSize = CGF.getContext().getTypeSizeInChars(type);
496036f2f6bSJohn McCall   llvm::APInt typeSizeMultiplier(sizeWidth, typeSize.getQuantity());
497036f2f6bSJohn McCall   typeSizeMultiplier *= arraySizeMultiplier;
498036f2f6bSJohn McCall 
499036f2f6bSJohn McCall   // This will be a size_t.
500036f2f6bSJohn McCall   llvm::Value *size;
50132ac583dSChris Lattner 
50232ac583dSChris Lattner   // If someone is doing 'new int[42]' there is no need to do a dynamic check.
50332ac583dSChris Lattner   // Don't bloat the -O0 code.
504036f2f6bSJohn McCall   if (llvm::ConstantInt *numElementsC =
505036f2f6bSJohn McCall         dyn_cast<llvm::ConstantInt>(numElements)) {
506036f2f6bSJohn McCall     const llvm::APInt &count = numElementsC->getValue();
50732ac583dSChris Lattner 
508036f2f6bSJohn McCall     bool hasAnyOverflow = false;
50932ac583dSChris Lattner 
510036f2f6bSJohn McCall     // If 'count' was a negative number, it's an overflow.
511036f2f6bSJohn McCall     if (isSigned && count.isNegative())
512036f2f6bSJohn McCall       hasAnyOverflow = true;
5138ed55a54SJohn McCall 
514036f2f6bSJohn McCall     // We want to do all this arithmetic in size_t.  If numElements is
515036f2f6bSJohn McCall     // wider than that, check whether it's already too big, and if so,
516036f2f6bSJohn McCall     // overflow.
517036f2f6bSJohn McCall     else if (numElementsWidth > sizeWidth &&
518036f2f6bSJohn McCall              numElementsWidth - sizeWidth > count.countLeadingZeros())
519036f2f6bSJohn McCall       hasAnyOverflow = true;
520036f2f6bSJohn McCall 
521036f2f6bSJohn McCall     // Okay, compute a count at the right width.
522036f2f6bSJohn McCall     llvm::APInt adjustedCount = count.zextOrTrunc(sizeWidth);
523036f2f6bSJohn McCall 
524036f2f6bSJohn McCall     // Scale numElements by that.  This might overflow, but we don't
525036f2f6bSJohn McCall     // care because it only overflows if allocationSize does, too, and
526036f2f6bSJohn McCall     // if that overflows then we shouldn't use this.
527036f2f6bSJohn McCall     numElements = llvm::ConstantInt::get(CGF.SizeTy,
528036f2f6bSJohn McCall                                          adjustedCount * arraySizeMultiplier);
529036f2f6bSJohn McCall 
530036f2f6bSJohn McCall     // Compute the size before cookie, and track whether it overflowed.
531036f2f6bSJohn McCall     bool overflow;
532036f2f6bSJohn McCall     llvm::APInt allocationSize
533036f2f6bSJohn McCall       = adjustedCount.umul_ov(typeSizeMultiplier, overflow);
534036f2f6bSJohn McCall     hasAnyOverflow |= overflow;
535036f2f6bSJohn McCall 
536036f2f6bSJohn McCall     // Add in the cookie, and check whether it's overflowed.
537036f2f6bSJohn McCall     if (cookieSize != 0) {
538036f2f6bSJohn McCall       // Save the current size without a cookie.  This shouldn't be
539036f2f6bSJohn McCall       // used if there was overflow.
540036f2f6bSJohn McCall       sizeWithoutCookie = llvm::ConstantInt::get(CGF.SizeTy, allocationSize);
541036f2f6bSJohn McCall 
542036f2f6bSJohn McCall       allocationSize = allocationSize.uadd_ov(cookieSize, overflow);
543036f2f6bSJohn McCall       hasAnyOverflow |= overflow;
5448ed55a54SJohn McCall     }
5458ed55a54SJohn McCall 
546036f2f6bSJohn McCall     // On overflow, produce a -1 so operator new will fail.
547036f2f6bSJohn McCall     if (hasAnyOverflow) {
548036f2f6bSJohn McCall       size = llvm::Constant::getAllOnesValue(CGF.SizeTy);
54932ac583dSChris Lattner     } else {
550036f2f6bSJohn McCall       size = llvm::ConstantInt::get(CGF.SizeTy, allocationSize);
55132ac583dSChris Lattner     }
55232ac583dSChris Lattner 
553036f2f6bSJohn McCall   // Otherwise, we might need to use the overflow intrinsics.
5548ed55a54SJohn McCall   } else {
555036f2f6bSJohn McCall     // There are up to four conditions we need to test for:
556036f2f6bSJohn McCall     // 1) if isSigned, we need to check whether numElements is negative;
557036f2f6bSJohn McCall     // 2) if numElementsWidth > sizeWidth, we need to check whether
558036f2f6bSJohn McCall     //   numElements is larger than something representable in size_t;
559036f2f6bSJohn McCall     // 3) we need to compute
560036f2f6bSJohn McCall     //      sizeWithoutCookie := numElements * typeSizeMultiplier
561036f2f6bSJohn McCall     //    and check whether it overflows; and
562036f2f6bSJohn McCall     // 4) if we need a cookie, we need to compute
563036f2f6bSJohn McCall     //      size := sizeWithoutCookie + cookieSize
564036f2f6bSJohn McCall     //    and check whether it overflows.
5658ed55a54SJohn McCall 
566036f2f6bSJohn McCall     llvm::Value *hasOverflow = 0;
5678ed55a54SJohn McCall 
568036f2f6bSJohn McCall     // If numElementsWidth > sizeWidth, then one way or another, we're
569036f2f6bSJohn McCall     // going to have to do a comparison for (2), and this happens to
570036f2f6bSJohn McCall     // take care of (1), too.
571036f2f6bSJohn McCall     if (numElementsWidth > sizeWidth) {
572036f2f6bSJohn McCall       llvm::APInt threshold(numElementsWidth, 1);
573036f2f6bSJohn McCall       threshold <<= sizeWidth;
5748ed55a54SJohn McCall 
575036f2f6bSJohn McCall       llvm::Value *thresholdV
576036f2f6bSJohn McCall         = llvm::ConstantInt::get(numElementsType, threshold);
577036f2f6bSJohn McCall 
578036f2f6bSJohn McCall       hasOverflow = CGF.Builder.CreateICmpUGE(numElements, thresholdV);
579036f2f6bSJohn McCall       numElements = CGF.Builder.CreateTrunc(numElements, CGF.SizeTy);
580036f2f6bSJohn McCall 
581036f2f6bSJohn McCall     // Otherwise, if we're signed, we want to sext up to size_t.
582036f2f6bSJohn McCall     } else if (isSigned) {
583036f2f6bSJohn McCall       if (numElementsWidth < sizeWidth)
584036f2f6bSJohn McCall         numElements = CGF.Builder.CreateSExt(numElements, CGF.SizeTy);
585036f2f6bSJohn McCall 
586036f2f6bSJohn McCall       // If there's a non-1 type size multiplier, then we can do the
587036f2f6bSJohn McCall       // signedness check at the same time as we do the multiply
588036f2f6bSJohn McCall       // because a negative number times anything will cause an
589036f2f6bSJohn McCall       // unsigned overflow.  Otherwise, we have to do it here.
590036f2f6bSJohn McCall       if (typeSizeMultiplier == 1)
591036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateICmpSLT(numElements,
592036f2f6bSJohn McCall                                       llvm::ConstantInt::get(CGF.SizeTy, 0));
593036f2f6bSJohn McCall 
594036f2f6bSJohn McCall     // Otherwise, zext up to size_t if necessary.
595036f2f6bSJohn McCall     } else if (numElementsWidth < sizeWidth) {
596036f2f6bSJohn McCall       numElements = CGF.Builder.CreateZExt(numElements, CGF.SizeTy);
597036f2f6bSJohn McCall     }
598036f2f6bSJohn McCall 
599036f2f6bSJohn McCall     assert(numElements->getType() == CGF.SizeTy);
600036f2f6bSJohn McCall 
601036f2f6bSJohn McCall     size = numElements;
602036f2f6bSJohn McCall 
603036f2f6bSJohn McCall     // Multiply by the type size if necessary.  This multiplier
604036f2f6bSJohn McCall     // includes all the factors for nested arrays.
6058ed55a54SJohn McCall     //
606036f2f6bSJohn McCall     // This step also causes numElements to be scaled up by the
607036f2f6bSJohn McCall     // nested-array factor if necessary.  Overflow on this computation
608036f2f6bSJohn McCall     // can be ignored because the result shouldn't be used if
609036f2f6bSJohn McCall     // allocation fails.
610036f2f6bSJohn McCall     if (typeSizeMultiplier != 1) {
611036f2f6bSJohn McCall       llvm::Value *umul_with_overflow
6128d375cefSBenjamin Kramer         = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, CGF.SizeTy);
6138ed55a54SJohn McCall 
614036f2f6bSJohn McCall       llvm::Value *tsmV =
615036f2f6bSJohn McCall         llvm::ConstantInt::get(CGF.SizeTy, typeSizeMultiplier);
616036f2f6bSJohn McCall       llvm::Value *result =
617036f2f6bSJohn McCall         CGF.Builder.CreateCall2(umul_with_overflow, size, tsmV);
6188ed55a54SJohn McCall 
619036f2f6bSJohn McCall       llvm::Value *overflowed = CGF.Builder.CreateExtractValue(result, 1);
620036f2f6bSJohn McCall       if (hasOverflow)
621036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateOr(hasOverflow, overflowed);
6228ed55a54SJohn McCall       else
623036f2f6bSJohn McCall         hasOverflow = overflowed;
62459486a2dSAnders Carlsson 
625036f2f6bSJohn McCall       size = CGF.Builder.CreateExtractValue(result, 0);
626036f2f6bSJohn McCall 
627036f2f6bSJohn McCall       // Also scale up numElements by the array size multiplier.
628036f2f6bSJohn McCall       if (arraySizeMultiplier != 1) {
629036f2f6bSJohn McCall         // If the base element type size is 1, then we can re-use the
630036f2f6bSJohn McCall         // multiply we just did.
631036f2f6bSJohn McCall         if (typeSize.isOne()) {
632036f2f6bSJohn McCall           assert(arraySizeMultiplier == typeSizeMultiplier);
633036f2f6bSJohn McCall           numElements = size;
634036f2f6bSJohn McCall 
635036f2f6bSJohn McCall         // Otherwise we need a separate multiply.
636036f2f6bSJohn McCall         } else {
637036f2f6bSJohn McCall           llvm::Value *asmV =
638036f2f6bSJohn McCall             llvm::ConstantInt::get(CGF.SizeTy, arraySizeMultiplier);
639036f2f6bSJohn McCall           numElements = CGF.Builder.CreateMul(numElements, asmV);
640036f2f6bSJohn McCall         }
641036f2f6bSJohn McCall       }
642036f2f6bSJohn McCall     } else {
643036f2f6bSJohn McCall       // numElements doesn't need to be scaled.
644036f2f6bSJohn McCall       assert(arraySizeMultiplier == 1);
645036f2f6bSJohn McCall     }
646036f2f6bSJohn McCall 
647036f2f6bSJohn McCall     // Add in the cookie size if necessary.
648036f2f6bSJohn McCall     if (cookieSize != 0) {
649036f2f6bSJohn McCall       sizeWithoutCookie = size;
650036f2f6bSJohn McCall 
651036f2f6bSJohn McCall       llvm::Value *uadd_with_overflow
6528d375cefSBenjamin Kramer         = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, CGF.SizeTy);
653036f2f6bSJohn McCall 
654036f2f6bSJohn McCall       llvm::Value *cookieSizeV = llvm::ConstantInt::get(CGF.SizeTy, cookieSize);
655036f2f6bSJohn McCall       llvm::Value *result =
656036f2f6bSJohn McCall         CGF.Builder.CreateCall2(uadd_with_overflow, size, cookieSizeV);
657036f2f6bSJohn McCall 
658036f2f6bSJohn McCall       llvm::Value *overflowed = CGF.Builder.CreateExtractValue(result, 1);
659036f2f6bSJohn McCall       if (hasOverflow)
660036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateOr(hasOverflow, overflowed);
661036f2f6bSJohn McCall       else
662036f2f6bSJohn McCall         hasOverflow = overflowed;
663036f2f6bSJohn McCall 
664036f2f6bSJohn McCall       size = CGF.Builder.CreateExtractValue(result, 0);
665036f2f6bSJohn McCall     }
666036f2f6bSJohn McCall 
667036f2f6bSJohn McCall     // If we had any possibility of dynamic overflow, make a select to
668036f2f6bSJohn McCall     // overwrite 'size' with an all-ones value, which should cause
669036f2f6bSJohn McCall     // operator new to throw.
670036f2f6bSJohn McCall     if (hasOverflow)
671036f2f6bSJohn McCall       size = CGF.Builder.CreateSelect(hasOverflow,
672036f2f6bSJohn McCall                                  llvm::Constant::getAllOnesValue(CGF.SizeTy),
673036f2f6bSJohn McCall                                       size);
674036f2f6bSJohn McCall   }
675036f2f6bSJohn McCall 
676036f2f6bSJohn McCall   if (cookieSize == 0)
677036f2f6bSJohn McCall     sizeWithoutCookie = size;
678036f2f6bSJohn McCall   else
679036f2f6bSJohn McCall     assert(sizeWithoutCookie && "didn't set sizeWithoutCookie?");
680036f2f6bSJohn McCall 
681036f2f6bSJohn McCall   return size;
68259486a2dSAnders Carlsson }
68359486a2dSAnders Carlsson 
684d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E,
685d5202e09SFariborz Jahanian                                     llvm::Value *NewPtr) {
686d5202e09SFariborz Jahanian 
687d5202e09SFariborz Jahanian   assert(E->getNumConstructorArgs() == 1 &&
688d5202e09SFariborz Jahanian          "Can only have one argument to initializer of POD type.");
689d5202e09SFariborz Jahanian 
690d5202e09SFariborz Jahanian   const Expr *Init = E->getConstructorArg(0);
691d5202e09SFariborz Jahanian   QualType AllocType = E->getAllocatedType();
692d5202e09SFariborz Jahanian 
6930381634aSDaniel Dunbar   unsigned Alignment =
6940381634aSDaniel Dunbar     CGF.getContext().getTypeAlignInChars(AllocType).getQuantity();
695d5202e09SFariborz Jahanian   if (!CGF.hasAggregateLLVMType(AllocType))
6961553b190SJohn McCall     CGF.EmitScalarInit(Init, 0, CGF.MakeAddrLValue(NewPtr, AllocType, Alignment),
6971553b190SJohn McCall                        false);
698d5202e09SFariborz Jahanian   else if (AllocType->isAnyComplexType())
699d5202e09SFariborz Jahanian     CGF.EmitComplexExprIntoAddr(Init, NewPtr,
700d5202e09SFariborz Jahanian                                 AllocType.isVolatileQualified());
7017a626f63SJohn McCall   else {
7027a626f63SJohn McCall     AggValueSlot Slot
7038d6fc958SJohn McCall       = AggValueSlot::forAddr(NewPtr, AllocType.getQualifiers(),
7048d6fc958SJohn McCall                               AggValueSlot::IsDestructed,
70546759f4fSJohn McCall                               AggValueSlot::DoesNotNeedGCBarriers,
70646759f4fSJohn McCall                               AggValueSlot::IsNotAliased);
7077a626f63SJohn McCall     CGF.EmitAggExpr(Init, Slot);
7087a626f63SJohn McCall   }
709d5202e09SFariborz Jahanian }
710d5202e09SFariborz Jahanian 
711d5202e09SFariborz Jahanian void
712d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E,
713*99210dc9SJohn McCall                                          QualType elementType,
714*99210dc9SJohn McCall                                          llvm::Value *beginPtr,
715*99210dc9SJohn McCall                                          llvm::Value *numElements) {
716b66b08efSFariborz Jahanian   // We have a POD type.
717b66b08efSFariborz Jahanian   if (E->getNumConstructorArgs() == 0)
718b66b08efSFariborz Jahanian     return;
719b66b08efSFariborz Jahanian 
720*99210dc9SJohn McCall   // Check if the number of elements is constant.
721*99210dc9SJohn McCall   bool checkZero = true;
722*99210dc9SJohn McCall   if (llvm::ConstantInt *constNum = dyn_cast<llvm::ConstantInt>(numElements)) {
723*99210dc9SJohn McCall     // If it's constant zero, skip the whole loop.
724*99210dc9SJohn McCall     if (constNum->isZero()) return;
725d5202e09SFariborz Jahanian 
726*99210dc9SJohn McCall     checkZero = false;
727*99210dc9SJohn McCall   }
728d5202e09SFariborz Jahanian 
729*99210dc9SJohn McCall   // Find the end of the array, hoisted out of the loop.
730*99210dc9SJohn McCall   llvm::Value *endPtr =
731*99210dc9SJohn McCall     Builder.CreateInBoundsGEP(beginPtr, numElements, "array.end");
732d5202e09SFariborz Jahanian 
733*99210dc9SJohn McCall   // Create the continuation block.
734*99210dc9SJohn McCall   llvm::BasicBlock *contBB = createBasicBlock("new.loop.end");
735d5202e09SFariborz Jahanian 
736*99210dc9SJohn McCall   // If we need to check for zero, do so now.
737*99210dc9SJohn McCall   if (checkZero) {
738*99210dc9SJohn McCall     llvm::BasicBlock *nonEmptyBB = createBasicBlock("new.loop.nonempty");
739*99210dc9SJohn McCall     llvm::Value *isEmpty = Builder.CreateICmpEQ(beginPtr, endPtr,
740*99210dc9SJohn McCall                                                 "array.isempty");
741*99210dc9SJohn McCall     Builder.CreateCondBr(isEmpty, contBB, nonEmptyBB);
742*99210dc9SJohn McCall     EmitBlock(nonEmptyBB);
743*99210dc9SJohn McCall   }
744d5202e09SFariborz Jahanian 
745*99210dc9SJohn McCall   // Enter the loop.
746*99210dc9SJohn McCall   llvm::BasicBlock *entryBB = Builder.GetInsertBlock();
747*99210dc9SJohn McCall   llvm::BasicBlock *loopBB = createBasicBlock("new.loop");
748d5202e09SFariborz Jahanian 
749*99210dc9SJohn McCall   EmitBlock(loopBB);
750d5202e09SFariborz Jahanian 
751*99210dc9SJohn McCall   // Set up the current-element phi.
752*99210dc9SJohn McCall   llvm::PHINode *curPtr =
753*99210dc9SJohn McCall     Builder.CreatePHI(beginPtr->getType(), 2, "array.cur");
754*99210dc9SJohn McCall   curPtr->addIncoming(beginPtr, entryBB);
755d5202e09SFariborz Jahanian 
756*99210dc9SJohn McCall   // Enter a partial-destruction cleanup if necessary.
757*99210dc9SJohn McCall   QualType::DestructionKind dtorKind = elementType.isDestructedType();
758*99210dc9SJohn McCall   EHScopeStack::stable_iterator cleanup;
759*99210dc9SJohn McCall   if (needsEHCleanup(dtorKind)) {
760*99210dc9SJohn McCall     pushRegularPartialArrayCleanup(beginPtr, curPtr, elementType,
761*99210dc9SJohn McCall                                    getDestroyer(dtorKind));
762*99210dc9SJohn McCall     cleanup = EHStack.stable_begin();
763*99210dc9SJohn McCall   }
764d5202e09SFariborz Jahanian 
765*99210dc9SJohn McCall   // Emit the initializer into this element.
766*99210dc9SJohn McCall   StoreAnyExprIntoOneUnit(*this, E, curPtr);
767d5202e09SFariborz Jahanian 
768*99210dc9SJohn McCall   // Leave the cleanup if we entered one.
769*99210dc9SJohn McCall   if (cleanup != EHStack.stable_end())
770*99210dc9SJohn McCall     DeactivateCleanupBlock(cleanup);
771d5202e09SFariborz Jahanian 
772*99210dc9SJohn McCall   // Advance to the next element.
773*99210dc9SJohn McCall   llvm::Value *nextPtr = Builder.CreateConstGEP1_32(curPtr, 1, "array.next");
774*99210dc9SJohn McCall 
775*99210dc9SJohn McCall   // Check whether we've gotten to the end of the array and, if so,
776*99210dc9SJohn McCall   // exit the loop.
777*99210dc9SJohn McCall   llvm::Value *isEnd = Builder.CreateICmpEQ(nextPtr, endPtr, "array.atend");
778*99210dc9SJohn McCall   Builder.CreateCondBr(isEnd, contBB, loopBB);
779*99210dc9SJohn McCall   curPtr->addIncoming(nextPtr, Builder.GetInsertBlock());
780*99210dc9SJohn McCall 
781*99210dc9SJohn McCall   EmitBlock(contBB);
782d5202e09SFariborz Jahanian }
783d5202e09SFariborz Jahanian 
78405fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T,
78505fc5be3SDouglas Gregor                            llvm::Value *NewPtr, llvm::Value *Size) {
786ad7c5c16SJohn McCall   CGF.EmitCastToVoidPtr(NewPtr);
787705ba07eSKen Dyck   CharUnits Alignment = CGF.getContext().getTypeAlignInChars(T);
788acc6b4e2SBenjamin Kramer   CGF.Builder.CreateMemSet(NewPtr, CGF.Builder.getInt8(0), Size,
789705ba07eSKen Dyck                            Alignment.getQuantity(), false);
79005fc5be3SDouglas Gregor }
79105fc5be3SDouglas Gregor 
79259486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
793*99210dc9SJohn McCall                                QualType ElementType,
79459486a2dSAnders Carlsson                                llvm::Value *NewPtr,
79505fc5be3SDouglas Gregor                                llvm::Value *NumElements,
79605fc5be3SDouglas Gregor                                llvm::Value *AllocSizeWithoutCookie) {
7973a202f60SAnders Carlsson   if (E->isArray()) {
798d040e6b2SAnders Carlsson     if (CXXConstructorDecl *Ctor = E->getConstructor()) {
79905fc5be3SDouglas Gregor       bool RequiresZeroInitialization = false;
800f479f1b7SAlexis Hunt       if (Ctor->getParent()->hasTrivialDefaultConstructor()) {
80105fc5be3SDouglas Gregor         // If new expression did not specify value-initialization, then there
80205fc5be3SDouglas Gregor         // is no initialization.
80305fc5be3SDouglas Gregor         if (!E->hasInitializer() || Ctor->getParent()->isEmpty())
80405fc5be3SDouglas Gregor           return;
80505fc5be3SDouglas Gregor 
806*99210dc9SJohn McCall         if (CGF.CGM.getTypes().isZeroInitializable(ElementType)) {
80705fc5be3SDouglas Gregor           // Optimization: since zero initialization will just set the memory
80805fc5be3SDouglas Gregor           // to all zeroes, generate a single memset to do it in one shot.
809*99210dc9SJohn McCall           EmitZeroMemSet(CGF, ElementType, NewPtr, AllocSizeWithoutCookie);
8103a202f60SAnders Carlsson           return;
8113a202f60SAnders Carlsson         }
81205fc5be3SDouglas Gregor 
81305fc5be3SDouglas Gregor         RequiresZeroInitialization = true;
81405fc5be3SDouglas Gregor       }
81505fc5be3SDouglas Gregor 
81605fc5be3SDouglas Gregor       CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr,
81705fc5be3SDouglas Gregor                                      E->constructor_arg_begin(),
81805fc5be3SDouglas Gregor                                      E->constructor_arg_end(),
81905fc5be3SDouglas Gregor                                      RequiresZeroInitialization);
82005fc5be3SDouglas Gregor       return;
82105fc5be3SDouglas Gregor     } else if (E->getNumConstructorArgs() == 1 &&
82205fc5be3SDouglas Gregor                isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) {
82305fc5be3SDouglas Gregor       // Optimization: since zero initialization will just set the memory
82405fc5be3SDouglas Gregor       // to all zeroes, generate a single memset to do it in one shot.
825*99210dc9SJohn McCall       EmitZeroMemSet(CGF, ElementType, NewPtr, AllocSizeWithoutCookie);
82605fc5be3SDouglas Gregor       return;
82705fc5be3SDouglas Gregor     } else {
828*99210dc9SJohn McCall       CGF.EmitNewArrayInitializer(E, ElementType, NewPtr, NumElements);
829d5202e09SFariborz Jahanian       return;
830d040e6b2SAnders Carlsson     }
831d5202e09SFariborz Jahanian   }
83259486a2dSAnders Carlsson 
83359486a2dSAnders Carlsson   if (CXXConstructorDecl *Ctor = E->getConstructor()) {
834747eb784SDouglas Gregor     // Per C++ [expr.new]p15, if we have an initializer, then we're performing
835747eb784SDouglas Gregor     // direct initialization. C++ [dcl.init]p5 requires that we
836747eb784SDouglas Gregor     // zero-initialize storage if there are no user-declared constructors.
837747eb784SDouglas Gregor     if (E->hasInitializer() &&
838747eb784SDouglas Gregor         !Ctor->getParent()->hasUserDeclaredConstructor() &&
839747eb784SDouglas Gregor         !Ctor->getParent()->isEmpty())
840*99210dc9SJohn McCall       CGF.EmitNullInitialization(NewPtr, ElementType);
841747eb784SDouglas Gregor 
842e11f9ce9SAnders Carlsson     CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false,
843e11f9ce9SAnders Carlsson                                NewPtr, E->constructor_arg_begin(),
84459486a2dSAnders Carlsson                                E->constructor_arg_end());
84559486a2dSAnders Carlsson 
84659486a2dSAnders Carlsson     return;
84759486a2dSAnders Carlsson   }
848b66b08efSFariborz Jahanian   // We have a POD type.
849b66b08efSFariborz Jahanian   if (E->getNumConstructorArgs() == 0)
850b66b08efSFariborz Jahanian     return;
85159486a2dSAnders Carlsson 
852d5202e09SFariborz Jahanian   StoreAnyExprIntoOneUnit(CGF, E, NewPtr);
85359486a2dSAnders Carlsson }
85459486a2dSAnders Carlsson 
855824c2f53SJohn McCall namespace {
856824c2f53SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
857824c2f53SJohn McCall   /// abnormal exit from a new expression.
858824c2f53SJohn McCall   class CallDeleteDuringNew : public EHScopeStack::Cleanup {
859824c2f53SJohn McCall     size_t NumPlacementArgs;
860824c2f53SJohn McCall     const FunctionDecl *OperatorDelete;
861824c2f53SJohn McCall     llvm::Value *Ptr;
862824c2f53SJohn McCall     llvm::Value *AllocSize;
863824c2f53SJohn McCall 
864824c2f53SJohn McCall     RValue *getPlacementArgs() { return reinterpret_cast<RValue*>(this+1); }
865824c2f53SJohn McCall 
866824c2f53SJohn McCall   public:
867824c2f53SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
868824c2f53SJohn McCall       return NumPlacementArgs * sizeof(RValue);
869824c2f53SJohn McCall     }
870824c2f53SJohn McCall 
871824c2f53SJohn McCall     CallDeleteDuringNew(size_t NumPlacementArgs,
872824c2f53SJohn McCall                         const FunctionDecl *OperatorDelete,
873824c2f53SJohn McCall                         llvm::Value *Ptr,
874824c2f53SJohn McCall                         llvm::Value *AllocSize)
875824c2f53SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
876824c2f53SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
877824c2f53SJohn McCall 
878824c2f53SJohn McCall     void setPlacementArg(unsigned I, RValue Arg) {
879824c2f53SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
880824c2f53SJohn McCall       getPlacementArgs()[I] = Arg;
881824c2f53SJohn McCall     }
882824c2f53SJohn McCall 
88330317fdaSJohn McCall     void Emit(CodeGenFunction &CGF, Flags flags) {
884824c2f53SJohn McCall       const FunctionProtoType *FPT
885824c2f53SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
886824c2f53SJohn McCall       assert(FPT->getNumArgs() == NumPlacementArgs + 1 ||
887d441b1e6SJohn McCall              (FPT->getNumArgs() == 2 && NumPlacementArgs == 0));
888824c2f53SJohn McCall 
889824c2f53SJohn McCall       CallArgList DeleteArgs;
890824c2f53SJohn McCall 
891824c2f53SJohn McCall       // The first argument is always a void*.
892824c2f53SJohn McCall       FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin();
89343dca6a8SEli Friedman       DeleteArgs.add(RValue::get(Ptr), *AI++);
894824c2f53SJohn McCall 
895824c2f53SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
896824c2f53SJohn McCall       if (FPT->getNumArgs() == NumPlacementArgs + 2)
89743dca6a8SEli Friedman         DeleteArgs.add(RValue::get(AllocSize), *AI++);
898824c2f53SJohn McCall 
899824c2f53SJohn McCall       // Pass the rest of the arguments, which must match exactly.
900824c2f53SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I)
90143dca6a8SEli Friedman         DeleteArgs.add(getPlacementArgs()[I], *AI++);
902824c2f53SJohn McCall 
903824c2f53SJohn McCall       // Call 'operator delete'.
90499cc30c3STilmann Scheller       CGF.EmitCall(CGF.CGM.getTypes().getFunctionInfo(DeleteArgs, FPT),
905824c2f53SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
906824c2f53SJohn McCall                    ReturnValueSlot(), DeleteArgs, OperatorDelete);
907824c2f53SJohn McCall     }
908824c2f53SJohn McCall   };
9097f9c92a9SJohn McCall 
9107f9c92a9SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
9117f9c92a9SJohn McCall   /// abnormal exit from a new expression when the new expression is
9127f9c92a9SJohn McCall   /// conditional.
9137f9c92a9SJohn McCall   class CallDeleteDuringConditionalNew : public EHScopeStack::Cleanup {
9147f9c92a9SJohn McCall     size_t NumPlacementArgs;
9157f9c92a9SJohn McCall     const FunctionDecl *OperatorDelete;
916cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type Ptr;
917cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type AllocSize;
9187f9c92a9SJohn McCall 
919cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type *getPlacementArgs() {
920cb5f77f0SJohn McCall       return reinterpret_cast<DominatingValue<RValue>::saved_type*>(this+1);
9217f9c92a9SJohn McCall     }
9227f9c92a9SJohn McCall 
9237f9c92a9SJohn McCall   public:
9247f9c92a9SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
925cb5f77f0SJohn McCall       return NumPlacementArgs * sizeof(DominatingValue<RValue>::saved_type);
9267f9c92a9SJohn McCall     }
9277f9c92a9SJohn McCall 
9287f9c92a9SJohn McCall     CallDeleteDuringConditionalNew(size_t NumPlacementArgs,
9297f9c92a9SJohn McCall                                    const FunctionDecl *OperatorDelete,
930cb5f77f0SJohn McCall                                    DominatingValue<RValue>::saved_type Ptr,
931cb5f77f0SJohn McCall                               DominatingValue<RValue>::saved_type AllocSize)
9327f9c92a9SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
9337f9c92a9SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
9347f9c92a9SJohn McCall 
935cb5f77f0SJohn McCall     void setPlacementArg(unsigned I, DominatingValue<RValue>::saved_type Arg) {
9367f9c92a9SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
9377f9c92a9SJohn McCall       getPlacementArgs()[I] = Arg;
9387f9c92a9SJohn McCall     }
9397f9c92a9SJohn McCall 
94030317fdaSJohn McCall     void Emit(CodeGenFunction &CGF, Flags flags) {
9417f9c92a9SJohn McCall       const FunctionProtoType *FPT
9427f9c92a9SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
9437f9c92a9SJohn McCall       assert(FPT->getNumArgs() == NumPlacementArgs + 1 ||
9447f9c92a9SJohn McCall              (FPT->getNumArgs() == 2 && NumPlacementArgs == 0));
9457f9c92a9SJohn McCall 
9467f9c92a9SJohn McCall       CallArgList DeleteArgs;
9477f9c92a9SJohn McCall 
9487f9c92a9SJohn McCall       // The first argument is always a void*.
9497f9c92a9SJohn McCall       FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin();
95043dca6a8SEli Friedman       DeleteArgs.add(Ptr.restore(CGF), *AI++);
9517f9c92a9SJohn McCall 
9527f9c92a9SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
9537f9c92a9SJohn McCall       if (FPT->getNumArgs() == NumPlacementArgs + 2) {
954cb5f77f0SJohn McCall         RValue RV = AllocSize.restore(CGF);
95543dca6a8SEli Friedman         DeleteArgs.add(RV, *AI++);
9567f9c92a9SJohn McCall       }
9577f9c92a9SJohn McCall 
9587f9c92a9SJohn McCall       // Pass the rest of the arguments, which must match exactly.
9597f9c92a9SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I) {
960cb5f77f0SJohn McCall         RValue RV = getPlacementArgs()[I].restore(CGF);
96143dca6a8SEli Friedman         DeleteArgs.add(RV, *AI++);
9627f9c92a9SJohn McCall       }
9637f9c92a9SJohn McCall 
9647f9c92a9SJohn McCall       // Call 'operator delete'.
96599cc30c3STilmann Scheller       CGF.EmitCall(CGF.CGM.getTypes().getFunctionInfo(DeleteArgs, FPT),
9667f9c92a9SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
9677f9c92a9SJohn McCall                    ReturnValueSlot(), DeleteArgs, OperatorDelete);
9687f9c92a9SJohn McCall     }
9697f9c92a9SJohn McCall   };
9707f9c92a9SJohn McCall }
9717f9c92a9SJohn McCall 
9727f9c92a9SJohn McCall /// Enter a cleanup to call 'operator delete' if the initializer in a
9737f9c92a9SJohn McCall /// new-expression throws.
9747f9c92a9SJohn McCall static void EnterNewDeleteCleanup(CodeGenFunction &CGF,
9757f9c92a9SJohn McCall                                   const CXXNewExpr *E,
9767f9c92a9SJohn McCall                                   llvm::Value *NewPtr,
9777f9c92a9SJohn McCall                                   llvm::Value *AllocSize,
9787f9c92a9SJohn McCall                                   const CallArgList &NewArgs) {
9797f9c92a9SJohn McCall   // If we're not inside a conditional branch, then the cleanup will
9807f9c92a9SJohn McCall   // dominate and we can do the easier (and more efficient) thing.
9817f9c92a9SJohn McCall   if (!CGF.isInConditionalBranch()) {
9827f9c92a9SJohn McCall     CallDeleteDuringNew *Cleanup = CGF.EHStack
9837f9c92a9SJohn McCall       .pushCleanupWithExtra<CallDeleteDuringNew>(EHCleanup,
9847f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
9857f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
9867f9c92a9SJohn McCall                                                  NewPtr, AllocSize);
9877f9c92a9SJohn McCall     for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
988f4258eb4SEli Friedman       Cleanup->setPlacementArg(I, NewArgs[I+1].RV);
9897f9c92a9SJohn McCall 
9907f9c92a9SJohn McCall     return;
9917f9c92a9SJohn McCall   }
9927f9c92a9SJohn McCall 
9937f9c92a9SJohn McCall   // Otherwise, we need to save all this stuff.
994cb5f77f0SJohn McCall   DominatingValue<RValue>::saved_type SavedNewPtr =
995cb5f77f0SJohn McCall     DominatingValue<RValue>::save(CGF, RValue::get(NewPtr));
996cb5f77f0SJohn McCall   DominatingValue<RValue>::saved_type SavedAllocSize =
997cb5f77f0SJohn McCall     DominatingValue<RValue>::save(CGF, RValue::get(AllocSize));
9987f9c92a9SJohn McCall 
9997f9c92a9SJohn McCall   CallDeleteDuringConditionalNew *Cleanup = CGF.EHStack
10007f9c92a9SJohn McCall     .pushCleanupWithExtra<CallDeleteDuringConditionalNew>(InactiveEHCleanup,
10017f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
10027f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
10037f9c92a9SJohn McCall                                                  SavedNewPtr,
10047f9c92a9SJohn McCall                                                  SavedAllocSize);
10057f9c92a9SJohn McCall   for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
1006cb5f77f0SJohn McCall     Cleanup->setPlacementArg(I,
1007f4258eb4SEli Friedman                      DominatingValue<RValue>::save(CGF, NewArgs[I+1].RV));
10087f9c92a9SJohn McCall 
10097f9c92a9SJohn McCall   CGF.ActivateCleanupBlock(CGF.EHStack.stable_begin());
1010824c2f53SJohn McCall }
1011824c2f53SJohn McCall 
101259486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
101375f9498aSJohn McCall   // The element type being allocated.
101475f9498aSJohn McCall   QualType allocType = getContext().getBaseElementType(E->getAllocatedType());
10158ed55a54SJohn McCall 
101675f9498aSJohn McCall   // 1. Build a call to the allocation function.
101775f9498aSJohn McCall   FunctionDecl *allocator = E->getOperatorNew();
101875f9498aSJohn McCall   const FunctionProtoType *allocatorType =
101975f9498aSJohn McCall     allocator->getType()->castAs<FunctionProtoType>();
102059486a2dSAnders Carlsson 
102175f9498aSJohn McCall   CallArgList allocatorArgs;
102259486a2dSAnders Carlsson 
102359486a2dSAnders Carlsson   // The allocation size is the first argument.
102475f9498aSJohn McCall   QualType sizeType = getContext().getSizeType();
102559486a2dSAnders Carlsson 
102675f9498aSJohn McCall   llvm::Value *numElements = 0;
102775f9498aSJohn McCall   llvm::Value *allocSizeWithoutCookie = 0;
102875f9498aSJohn McCall   llvm::Value *allocSize =
1029036f2f6bSJohn McCall     EmitCXXNewAllocSize(*this, E, numElements, allocSizeWithoutCookie);
103059486a2dSAnders Carlsson 
103143dca6a8SEli Friedman   allocatorArgs.add(RValue::get(allocSize), sizeType);
103259486a2dSAnders Carlsson 
103359486a2dSAnders Carlsson   // Emit the rest of the arguments.
103459486a2dSAnders Carlsson   // FIXME: Ideally, this should just use EmitCallArgs.
103575f9498aSJohn McCall   CXXNewExpr::const_arg_iterator placementArg = E->placement_arg_begin();
103659486a2dSAnders Carlsson 
103759486a2dSAnders Carlsson   // First, use the types from the function type.
103859486a2dSAnders Carlsson   // We start at 1 here because the first argument (the allocation size)
103959486a2dSAnders Carlsson   // has already been emitted.
104075f9498aSJohn McCall   for (unsigned i = 1, e = allocatorType->getNumArgs(); i != e;
104175f9498aSJohn McCall        ++i, ++placementArg) {
104275f9498aSJohn McCall     QualType argType = allocatorType->getArgType(i);
104359486a2dSAnders Carlsson 
104475f9498aSJohn McCall     assert(getContext().hasSameUnqualifiedType(argType.getNonReferenceType(),
104575f9498aSJohn McCall                                                placementArg->getType()) &&
104659486a2dSAnders Carlsson            "type mismatch in call argument!");
104759486a2dSAnders Carlsson 
104832ea9694SJohn McCall     EmitCallArg(allocatorArgs, *placementArg, argType);
104959486a2dSAnders Carlsson   }
105059486a2dSAnders Carlsson 
105159486a2dSAnders Carlsson   // Either we've emitted all the call args, or we have a call to a
105259486a2dSAnders Carlsson   // variadic function.
105375f9498aSJohn McCall   assert((placementArg == E->placement_arg_end() ||
105475f9498aSJohn McCall           allocatorType->isVariadic()) &&
105575f9498aSJohn McCall          "Extra arguments to non-variadic function!");
105659486a2dSAnders Carlsson 
105759486a2dSAnders Carlsson   // If we still have any arguments, emit them using the type of the argument.
105875f9498aSJohn McCall   for (CXXNewExpr::const_arg_iterator placementArgsEnd = E->placement_arg_end();
105975f9498aSJohn McCall        placementArg != placementArgsEnd; ++placementArg) {
106032ea9694SJohn McCall     EmitCallArg(allocatorArgs, *placementArg, placementArg->getType());
106159486a2dSAnders Carlsson   }
106259486a2dSAnders Carlsson 
10637ec4b434SJohn McCall   // Emit the allocation call.  If the allocator is a global placement
10647ec4b434SJohn McCall   // operator, just "inline" it directly.
10657ec4b434SJohn McCall   RValue RV;
10667ec4b434SJohn McCall   if (allocator->isReservedGlobalPlacementOperator()) {
10677ec4b434SJohn McCall     assert(allocatorArgs.size() == 2);
10687ec4b434SJohn McCall     RV = allocatorArgs[1].RV;
10697ec4b434SJohn McCall     // TODO: kill any unnecessary computations done for the size
10707ec4b434SJohn McCall     // argument.
10717ec4b434SJohn McCall   } else {
10727ec4b434SJohn McCall     RV = EmitCall(CGM.getTypes().getFunctionInfo(allocatorArgs, allocatorType),
107375f9498aSJohn McCall                   CGM.GetAddrOfFunction(allocator), ReturnValueSlot(),
107475f9498aSJohn McCall                   allocatorArgs, allocator);
10757ec4b434SJohn McCall   }
107659486a2dSAnders Carlsson 
107775f9498aSJohn McCall   // Emit a null check on the allocation result if the allocation
107875f9498aSJohn McCall   // function is allowed to return null (because it has a non-throwing
107975f9498aSJohn McCall   // exception spec; for this part, we inline
108075f9498aSJohn McCall   // CXXNewExpr::shouldNullCheckAllocation()) and we have an
108175f9498aSJohn McCall   // interesting initializer.
108231ad754cSSebastian Redl   bool nullCheck = allocatorType->isNothrow(getContext()) &&
108331168b07SJohn McCall     !(allocType.isPODType(getContext()) && !E->hasInitializer());
108459486a2dSAnders Carlsson 
108575f9498aSJohn McCall   llvm::BasicBlock *nullCheckBB = 0;
108675f9498aSJohn McCall   llvm::BasicBlock *contBB = 0;
108759486a2dSAnders Carlsson 
108875f9498aSJohn McCall   llvm::Value *allocation = RV.getScalarVal();
108975f9498aSJohn McCall   unsigned AS =
109075f9498aSJohn McCall     cast<llvm::PointerType>(allocation->getType())->getAddressSpace();
109159486a2dSAnders Carlsson 
1092f7dcf320SJohn McCall   // The null-check means that the initializer is conditionally
1093f7dcf320SJohn McCall   // evaluated.
1094f7dcf320SJohn McCall   ConditionalEvaluation conditional(*this);
1095f7dcf320SJohn McCall 
109675f9498aSJohn McCall   if (nullCheck) {
1097f7dcf320SJohn McCall     conditional.begin(*this);
109875f9498aSJohn McCall 
109975f9498aSJohn McCall     nullCheckBB = Builder.GetInsertBlock();
110075f9498aSJohn McCall     llvm::BasicBlock *notNullBB = createBasicBlock("new.notnull");
110175f9498aSJohn McCall     contBB = createBasicBlock("new.cont");
110275f9498aSJohn McCall 
110375f9498aSJohn McCall     llvm::Value *isNull = Builder.CreateIsNull(allocation, "new.isnull");
110475f9498aSJohn McCall     Builder.CreateCondBr(isNull, contBB, notNullBB);
110575f9498aSJohn McCall     EmitBlock(notNullBB);
110659486a2dSAnders Carlsson   }
110759486a2dSAnders Carlsson 
1108824c2f53SJohn McCall   // If there's an operator delete, enter a cleanup to call it if an
1109824c2f53SJohn McCall   // exception is thrown.
111075f9498aSJohn McCall   EHScopeStack::stable_iterator operatorDeleteCleanup;
11117ec4b434SJohn McCall   if (E->getOperatorDelete() &&
11127ec4b434SJohn McCall       !E->getOperatorDelete()->isReservedGlobalPlacementOperator()) {
111375f9498aSJohn McCall     EnterNewDeleteCleanup(*this, E, allocation, allocSize, allocatorArgs);
111475f9498aSJohn McCall     operatorDeleteCleanup = EHStack.stable_begin();
1115824c2f53SJohn McCall   }
1116824c2f53SJohn McCall 
1117cf9b1f65SEli Friedman   assert((allocSize == allocSizeWithoutCookie) ==
1118cf9b1f65SEli Friedman          CalculateCookiePadding(*this, E).isZero());
1119cf9b1f65SEli Friedman   if (allocSize != allocSizeWithoutCookie) {
1120cf9b1f65SEli Friedman     assert(E->isArray());
1121cf9b1f65SEli Friedman     allocation = CGM.getCXXABI().InitializeArrayCookie(*this, allocation,
1122cf9b1f65SEli Friedman                                                        numElements,
1123cf9b1f65SEli Friedman                                                        E, allocType);
1124cf9b1f65SEli Friedman   }
1125cf9b1f65SEli Friedman 
11262192fe50SChris Lattner   llvm::Type *elementPtrTy
112775f9498aSJohn McCall     = ConvertTypeForMem(allocType)->getPointerTo(AS);
112875f9498aSJohn McCall   llvm::Value *result = Builder.CreateBitCast(allocation, elementPtrTy);
1129824c2f53SJohn McCall 
1130*99210dc9SJohn McCall   EmitNewInitializer(*this, E, allocType, result, numElements,
1131*99210dc9SJohn McCall                      allocSizeWithoutCookie);
11328ed55a54SJohn McCall   if (E->isArray()) {
11338ed55a54SJohn McCall     // NewPtr is a pointer to the base element type.  If we're
11348ed55a54SJohn McCall     // allocating an array of arrays, we'll need to cast back to the
11358ed55a54SJohn McCall     // array pointer type.
11362192fe50SChris Lattner     llvm::Type *resultType = ConvertTypeForMem(E->getType());
113775f9498aSJohn McCall     if (result->getType() != resultType)
113875f9498aSJohn McCall       result = Builder.CreateBitCast(result, resultType);
113947b4629bSFariborz Jahanian   }
114059486a2dSAnders Carlsson 
1141824c2f53SJohn McCall   // Deactivate the 'operator delete' cleanup if we finished
1142824c2f53SJohn McCall   // initialization.
114375f9498aSJohn McCall   if (operatorDeleteCleanup.isValid())
114475f9498aSJohn McCall     DeactivateCleanupBlock(operatorDeleteCleanup);
1145824c2f53SJohn McCall 
114675f9498aSJohn McCall   if (nullCheck) {
1147f7dcf320SJohn McCall     conditional.end(*this);
1148f7dcf320SJohn McCall 
114975f9498aSJohn McCall     llvm::BasicBlock *notNullBB = Builder.GetInsertBlock();
115075f9498aSJohn McCall     EmitBlock(contBB);
115159486a2dSAnders Carlsson 
115220c0f02cSJay Foad     llvm::PHINode *PHI = Builder.CreatePHI(result->getType(), 2);
115375f9498aSJohn McCall     PHI->addIncoming(result, notNullBB);
115475f9498aSJohn McCall     PHI->addIncoming(llvm::Constant::getNullValue(result->getType()),
115575f9498aSJohn McCall                      nullCheckBB);
115659486a2dSAnders Carlsson 
115775f9498aSJohn McCall     result = PHI;
115859486a2dSAnders Carlsson   }
115959486a2dSAnders Carlsson 
116075f9498aSJohn McCall   return result;
116159486a2dSAnders Carlsson }
116259486a2dSAnders Carlsson 
116359486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
116459486a2dSAnders Carlsson                                      llvm::Value *Ptr,
116559486a2dSAnders Carlsson                                      QualType DeleteTy) {
11668ed55a54SJohn McCall   assert(DeleteFD->getOverloadedOperator() == OO_Delete);
11678ed55a54SJohn McCall 
116859486a2dSAnders Carlsson   const FunctionProtoType *DeleteFTy =
116959486a2dSAnders Carlsson     DeleteFD->getType()->getAs<FunctionProtoType>();
117059486a2dSAnders Carlsson 
117159486a2dSAnders Carlsson   CallArgList DeleteArgs;
117259486a2dSAnders Carlsson 
117321122cf6SAnders Carlsson   // Check if we need to pass the size to the delete operator.
117421122cf6SAnders Carlsson   llvm::Value *Size = 0;
117521122cf6SAnders Carlsson   QualType SizeTy;
117621122cf6SAnders Carlsson   if (DeleteFTy->getNumArgs() == 2) {
117721122cf6SAnders Carlsson     SizeTy = DeleteFTy->getArgType(1);
11787df3cbebSKen Dyck     CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
11797df3cbebSKen Dyck     Size = llvm::ConstantInt::get(ConvertType(SizeTy),
11807df3cbebSKen Dyck                                   DeleteTypeSize.getQuantity());
118121122cf6SAnders Carlsson   }
118221122cf6SAnders Carlsson 
118359486a2dSAnders Carlsson   QualType ArgTy = DeleteFTy->getArgType(0);
118459486a2dSAnders Carlsson   llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
118543dca6a8SEli Friedman   DeleteArgs.add(RValue::get(DeletePtr), ArgTy);
118659486a2dSAnders Carlsson 
118721122cf6SAnders Carlsson   if (Size)
118843dca6a8SEli Friedman     DeleteArgs.add(RValue::get(Size), SizeTy);
118959486a2dSAnders Carlsson 
119059486a2dSAnders Carlsson   // Emit the call to delete.
119199cc30c3STilmann Scheller   EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy),
119261a401caSAnders Carlsson            CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(),
119359486a2dSAnders Carlsson            DeleteArgs, DeleteFD);
119459486a2dSAnders Carlsson }
119559486a2dSAnders Carlsson 
11968ed55a54SJohn McCall namespace {
11978ed55a54SJohn McCall   /// Calls the given 'operator delete' on a single object.
11988ed55a54SJohn McCall   struct CallObjectDelete : EHScopeStack::Cleanup {
11998ed55a54SJohn McCall     llvm::Value *Ptr;
12008ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
12018ed55a54SJohn McCall     QualType ElementType;
12028ed55a54SJohn McCall 
12038ed55a54SJohn McCall     CallObjectDelete(llvm::Value *Ptr,
12048ed55a54SJohn McCall                      const FunctionDecl *OperatorDelete,
12058ed55a54SJohn McCall                      QualType ElementType)
12068ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {}
12078ed55a54SJohn McCall 
120830317fdaSJohn McCall     void Emit(CodeGenFunction &CGF, Flags flags) {
12098ed55a54SJohn McCall       CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType);
12108ed55a54SJohn McCall     }
12118ed55a54SJohn McCall   };
12128ed55a54SJohn McCall }
12138ed55a54SJohn McCall 
12148ed55a54SJohn McCall /// Emit the code for deleting a single object.
12158ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF,
12168ed55a54SJohn McCall                              const FunctionDecl *OperatorDelete,
12178ed55a54SJohn McCall                              llvm::Value *Ptr,
12181c2e20d7SDouglas Gregor                              QualType ElementType,
12191c2e20d7SDouglas Gregor                              bool UseGlobalDelete) {
12208ed55a54SJohn McCall   // Find the destructor for the type, if applicable.  If the
12218ed55a54SJohn McCall   // destructor is virtual, we'll just emit the vcall and return.
12228ed55a54SJohn McCall   const CXXDestructorDecl *Dtor = 0;
12238ed55a54SJohn McCall   if (const RecordType *RT = ElementType->getAs<RecordType>()) {
12248ed55a54SJohn McCall     CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
1225b23533dbSEli Friedman     if (RD->hasDefinition() && !RD->hasTrivialDestructor()) {
12268ed55a54SJohn McCall       Dtor = RD->getDestructor();
12278ed55a54SJohn McCall 
12288ed55a54SJohn McCall       if (Dtor->isVirtual()) {
12291c2e20d7SDouglas Gregor         if (UseGlobalDelete) {
12301c2e20d7SDouglas Gregor           // If we're supposed to call the global delete, make sure we do so
12311c2e20d7SDouglas Gregor           // even if the destructor throws.
12321c2e20d7SDouglas Gregor           CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
12331c2e20d7SDouglas Gregor                                                     Ptr, OperatorDelete,
12341c2e20d7SDouglas Gregor                                                     ElementType);
12351c2e20d7SDouglas Gregor         }
12361c2e20d7SDouglas Gregor 
12372192fe50SChris Lattner         llvm::Type *Ty =
12380d635f53SJohn McCall           CGF.getTypes().GetFunctionType(CGF.getTypes().getFunctionInfo(Dtor,
12390d635f53SJohn McCall                                                                Dtor_Complete),
12408ed55a54SJohn McCall                                          /*isVariadic=*/false);
12418ed55a54SJohn McCall 
12428ed55a54SJohn McCall         llvm::Value *Callee
12431c2e20d7SDouglas Gregor           = CGF.BuildVirtualCall(Dtor,
12441c2e20d7SDouglas Gregor                                  UseGlobalDelete? Dtor_Complete : Dtor_Deleting,
12451c2e20d7SDouglas Gregor                                  Ptr, Ty);
12468ed55a54SJohn McCall         CGF.EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0,
12478ed55a54SJohn McCall                               0, 0);
12488ed55a54SJohn McCall 
12491c2e20d7SDouglas Gregor         if (UseGlobalDelete) {
12501c2e20d7SDouglas Gregor           CGF.PopCleanupBlock();
12511c2e20d7SDouglas Gregor         }
12521c2e20d7SDouglas Gregor 
12538ed55a54SJohn McCall         return;
12548ed55a54SJohn McCall       }
12558ed55a54SJohn McCall     }
12568ed55a54SJohn McCall   }
12578ed55a54SJohn McCall 
12588ed55a54SJohn McCall   // Make sure that we call delete even if the dtor throws.
1259e4df6c8dSJohn McCall   // This doesn't have to a conditional cleanup because we're going
1260e4df6c8dSJohn McCall   // to pop it off in a second.
12618ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
12628ed55a54SJohn McCall                                             Ptr, OperatorDelete, ElementType);
12638ed55a54SJohn McCall 
12648ed55a54SJohn McCall   if (Dtor)
12658ed55a54SJohn McCall     CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete,
12668ed55a54SJohn McCall                               /*ForVirtualBase=*/false, Ptr);
126731168b07SJohn McCall   else if (CGF.getLangOptions().ObjCAutoRefCount &&
126831168b07SJohn McCall            ElementType->isObjCLifetimeType()) {
126931168b07SJohn McCall     switch (ElementType.getObjCLifetime()) {
127031168b07SJohn McCall     case Qualifiers::OCL_None:
127131168b07SJohn McCall     case Qualifiers::OCL_ExplicitNone:
127231168b07SJohn McCall     case Qualifiers::OCL_Autoreleasing:
127331168b07SJohn McCall       break;
127431168b07SJohn McCall 
127531168b07SJohn McCall     case Qualifiers::OCL_Strong: {
127631168b07SJohn McCall       // Load the pointer value.
127731168b07SJohn McCall       llvm::Value *PtrValue = CGF.Builder.CreateLoad(Ptr,
127831168b07SJohn McCall                                              ElementType.isVolatileQualified());
127931168b07SJohn McCall 
128031168b07SJohn McCall       CGF.EmitARCRelease(PtrValue, /*precise*/ true);
128131168b07SJohn McCall       break;
128231168b07SJohn McCall     }
128331168b07SJohn McCall 
128431168b07SJohn McCall     case Qualifiers::OCL_Weak:
128531168b07SJohn McCall       CGF.EmitARCDestroyWeak(Ptr);
128631168b07SJohn McCall       break;
128731168b07SJohn McCall     }
128831168b07SJohn McCall   }
12898ed55a54SJohn McCall 
12908ed55a54SJohn McCall   CGF.PopCleanupBlock();
12918ed55a54SJohn McCall }
12928ed55a54SJohn McCall 
12938ed55a54SJohn McCall namespace {
12948ed55a54SJohn McCall   /// Calls the given 'operator delete' on an array of objects.
12958ed55a54SJohn McCall   struct CallArrayDelete : EHScopeStack::Cleanup {
12968ed55a54SJohn McCall     llvm::Value *Ptr;
12978ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
12988ed55a54SJohn McCall     llvm::Value *NumElements;
12998ed55a54SJohn McCall     QualType ElementType;
13008ed55a54SJohn McCall     CharUnits CookieSize;
13018ed55a54SJohn McCall 
13028ed55a54SJohn McCall     CallArrayDelete(llvm::Value *Ptr,
13038ed55a54SJohn McCall                     const FunctionDecl *OperatorDelete,
13048ed55a54SJohn McCall                     llvm::Value *NumElements,
13058ed55a54SJohn McCall                     QualType ElementType,
13068ed55a54SJohn McCall                     CharUnits CookieSize)
13078ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements),
13088ed55a54SJohn McCall         ElementType(ElementType), CookieSize(CookieSize) {}
13098ed55a54SJohn McCall 
131030317fdaSJohn McCall     void Emit(CodeGenFunction &CGF, Flags flags) {
13118ed55a54SJohn McCall       const FunctionProtoType *DeleteFTy =
13128ed55a54SJohn McCall         OperatorDelete->getType()->getAs<FunctionProtoType>();
13138ed55a54SJohn McCall       assert(DeleteFTy->getNumArgs() == 1 || DeleteFTy->getNumArgs() == 2);
13148ed55a54SJohn McCall 
13158ed55a54SJohn McCall       CallArgList Args;
13168ed55a54SJohn McCall 
13178ed55a54SJohn McCall       // Pass the pointer as the first argument.
13188ed55a54SJohn McCall       QualType VoidPtrTy = DeleteFTy->getArgType(0);
13198ed55a54SJohn McCall       llvm::Value *DeletePtr
13208ed55a54SJohn McCall         = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy));
132143dca6a8SEli Friedman       Args.add(RValue::get(DeletePtr), VoidPtrTy);
13228ed55a54SJohn McCall 
13238ed55a54SJohn McCall       // Pass the original requested size as the second argument.
13248ed55a54SJohn McCall       if (DeleteFTy->getNumArgs() == 2) {
13258ed55a54SJohn McCall         QualType size_t = DeleteFTy->getArgType(1);
13262192fe50SChris Lattner         llvm::IntegerType *SizeTy
13278ed55a54SJohn McCall           = cast<llvm::IntegerType>(CGF.ConvertType(size_t));
13288ed55a54SJohn McCall 
13298ed55a54SJohn McCall         CharUnits ElementTypeSize =
13308ed55a54SJohn McCall           CGF.CGM.getContext().getTypeSizeInChars(ElementType);
13318ed55a54SJohn McCall 
13328ed55a54SJohn McCall         // The size of an element, multiplied by the number of elements.
13338ed55a54SJohn McCall         llvm::Value *Size
13348ed55a54SJohn McCall           = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity());
13358ed55a54SJohn McCall         Size = CGF.Builder.CreateMul(Size, NumElements);
13368ed55a54SJohn McCall 
13378ed55a54SJohn McCall         // Plus the size of the cookie if applicable.
13388ed55a54SJohn McCall         if (!CookieSize.isZero()) {
13398ed55a54SJohn McCall           llvm::Value *CookieSizeV
13408ed55a54SJohn McCall             = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
13418ed55a54SJohn McCall           Size = CGF.Builder.CreateAdd(Size, CookieSizeV);
13428ed55a54SJohn McCall         }
13438ed55a54SJohn McCall 
134443dca6a8SEli Friedman         Args.add(RValue::get(Size), size_t);
13458ed55a54SJohn McCall       }
13468ed55a54SJohn McCall 
13478ed55a54SJohn McCall       // Emit the call to delete.
134899cc30c3STilmann Scheller       CGF.EmitCall(CGF.getTypes().getFunctionInfo(Args, DeleteFTy),
13498ed55a54SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
13508ed55a54SJohn McCall                    ReturnValueSlot(), Args, OperatorDelete);
13518ed55a54SJohn McCall     }
13528ed55a54SJohn McCall   };
13538ed55a54SJohn McCall }
13548ed55a54SJohn McCall 
13558ed55a54SJohn McCall /// Emit the code for deleting an array of objects.
13568ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF,
1357284c48ffSJohn McCall                             const CXXDeleteExpr *E,
1358ca2c56f2SJohn McCall                             llvm::Value *deletedPtr,
1359ca2c56f2SJohn McCall                             QualType elementType) {
1360ca2c56f2SJohn McCall   llvm::Value *numElements = 0;
1361ca2c56f2SJohn McCall   llvm::Value *allocatedPtr = 0;
1362ca2c56f2SJohn McCall   CharUnits cookieSize;
1363ca2c56f2SJohn McCall   CGF.CGM.getCXXABI().ReadArrayCookie(CGF, deletedPtr, E, elementType,
1364ca2c56f2SJohn McCall                                       numElements, allocatedPtr, cookieSize);
13658ed55a54SJohn McCall 
1366ca2c56f2SJohn McCall   assert(allocatedPtr && "ReadArrayCookie didn't set allocated pointer");
13678ed55a54SJohn McCall 
13688ed55a54SJohn McCall   // Make sure that we call delete even if one of the dtors throws.
1369ca2c56f2SJohn McCall   const FunctionDecl *operatorDelete = E->getOperatorDelete();
13708ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup,
1371ca2c56f2SJohn McCall                                            allocatedPtr, operatorDelete,
1372ca2c56f2SJohn McCall                                            numElements, elementType,
1373ca2c56f2SJohn McCall                                            cookieSize);
13748ed55a54SJohn McCall 
1375ca2c56f2SJohn McCall   // Destroy the elements.
1376ca2c56f2SJohn McCall   if (QualType::DestructionKind dtorKind = elementType.isDestructedType()) {
1377ca2c56f2SJohn McCall     assert(numElements && "no element count for a type with a destructor!");
137831168b07SJohn McCall 
1379ca2c56f2SJohn McCall     llvm::Value *arrayEnd =
1380ca2c56f2SJohn McCall       CGF.Builder.CreateInBoundsGEP(deletedPtr, numElements, "delete.end");
138197eab0a2SJohn McCall 
138297eab0a2SJohn McCall     // Note that it is legal to allocate a zero-length array, and we
138397eab0a2SJohn McCall     // can never fold the check away because the length should always
138497eab0a2SJohn McCall     // come from a cookie.
1385ca2c56f2SJohn McCall     CGF.emitArrayDestroy(deletedPtr, arrayEnd, elementType,
1386ca2c56f2SJohn McCall                          CGF.getDestroyer(dtorKind),
138797eab0a2SJohn McCall                          /*checkZeroLength*/ true,
1388ca2c56f2SJohn McCall                          CGF.needsEHCleanup(dtorKind));
13898ed55a54SJohn McCall   }
13908ed55a54SJohn McCall 
1391ca2c56f2SJohn McCall   // Pop the cleanup block.
13928ed55a54SJohn McCall   CGF.PopCleanupBlock();
13938ed55a54SJohn McCall }
13948ed55a54SJohn McCall 
139559486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
139659486a2dSAnders Carlsson 
139759486a2dSAnders Carlsson   // Get at the argument before we performed the implicit conversion
139859486a2dSAnders Carlsson   // to void*.
139959486a2dSAnders Carlsson   const Expr *Arg = E->getArgument();
140059486a2dSAnders Carlsson   while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) {
1401e302792bSJohn McCall     if (ICE->getCastKind() != CK_UserDefinedConversion &&
140259486a2dSAnders Carlsson         ICE->getType()->isVoidPointerType())
140359486a2dSAnders Carlsson       Arg = ICE->getSubExpr();
140459486a2dSAnders Carlsson     else
140559486a2dSAnders Carlsson       break;
140659486a2dSAnders Carlsson   }
140759486a2dSAnders Carlsson 
140859486a2dSAnders Carlsson   llvm::Value *Ptr = EmitScalarExpr(Arg);
140959486a2dSAnders Carlsson 
141059486a2dSAnders Carlsson   // Null check the pointer.
141159486a2dSAnders Carlsson   llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
141259486a2dSAnders Carlsson   llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
141359486a2dSAnders Carlsson 
141498981b10SAnders Carlsson   llvm::Value *IsNull = Builder.CreateIsNull(Ptr, "isnull");
141559486a2dSAnders Carlsson 
141659486a2dSAnders Carlsson   Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
141759486a2dSAnders Carlsson   EmitBlock(DeleteNotNull);
141859486a2dSAnders Carlsson 
14198ed55a54SJohn McCall   // We might be deleting a pointer to array.  If so, GEP down to the
14208ed55a54SJohn McCall   // first non-array element.
14218ed55a54SJohn McCall   // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*)
14228ed55a54SJohn McCall   QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
14238ed55a54SJohn McCall   if (DeleteTy->isConstantArrayType()) {
14248ed55a54SJohn McCall     llvm::Value *Zero = Builder.getInt32(0);
14250e62c1ccSChris Lattner     SmallVector<llvm::Value*,8> GEP;
142659486a2dSAnders Carlsson 
14278ed55a54SJohn McCall     GEP.push_back(Zero); // point at the outermost array
14288ed55a54SJohn McCall 
14298ed55a54SJohn McCall     // For each layer of array type we're pointing at:
14308ed55a54SJohn McCall     while (const ConstantArrayType *Arr
14318ed55a54SJohn McCall              = getContext().getAsConstantArrayType(DeleteTy)) {
14328ed55a54SJohn McCall       // 1. Unpeel the array type.
14338ed55a54SJohn McCall       DeleteTy = Arr->getElementType();
14348ed55a54SJohn McCall 
14358ed55a54SJohn McCall       // 2. GEP to the first element of the array.
14368ed55a54SJohn McCall       GEP.push_back(Zero);
14378ed55a54SJohn McCall     }
14388ed55a54SJohn McCall 
1439040dd82fSJay Foad     Ptr = Builder.CreateInBoundsGEP(Ptr, GEP, "del.first");
14408ed55a54SJohn McCall   }
14418ed55a54SJohn McCall 
144204f36218SDouglas Gregor   assert(ConvertTypeForMem(DeleteTy) ==
144304f36218SDouglas Gregor          cast<llvm::PointerType>(Ptr->getType())->getElementType());
14448ed55a54SJohn McCall 
144559486a2dSAnders Carlsson   if (E->isArrayForm()) {
1446284c48ffSJohn McCall     EmitArrayDelete(*this, E, Ptr, DeleteTy);
14478ed55a54SJohn McCall   } else {
14481c2e20d7SDouglas Gregor     EmitObjectDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy,
14491c2e20d7SDouglas Gregor                      E->isGlobalDelete());
145059486a2dSAnders Carlsson   }
145159486a2dSAnders Carlsson 
145259486a2dSAnders Carlsson   EmitBlock(DeleteEnd);
145359486a2dSAnders Carlsson }
145459486a2dSAnders Carlsson 
14550c63350bSAnders Carlsson static llvm::Constant *getBadTypeidFn(CodeGenFunction &CGF) {
14560c63350bSAnders Carlsson   // void __cxa_bad_typeid();
14570c63350bSAnders Carlsson 
14582192fe50SChris Lattner   llvm::Type *VoidTy = llvm::Type::getVoidTy(CGF.getLLVMContext());
14592192fe50SChris Lattner   llvm::FunctionType *FTy =
14600c63350bSAnders Carlsson   llvm::FunctionType::get(VoidTy, false);
14610c63350bSAnders Carlsson 
14620c63350bSAnders Carlsson   return CGF.CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid");
14630c63350bSAnders Carlsson }
14640c63350bSAnders Carlsson 
14650c63350bSAnders Carlsson static void EmitBadTypeidCall(CodeGenFunction &CGF) {
1466bbe277c4SAnders Carlsson   llvm::Value *Fn = getBadTypeidFn(CGF);
14675bd375a6SJay Foad   CGF.EmitCallOrInvoke(Fn).setDoesNotReturn();
14680c63350bSAnders Carlsson   CGF.Builder.CreateUnreachable();
14690c63350bSAnders Carlsson }
14700c63350bSAnders Carlsson 
1471940f02d2SAnders Carlsson static llvm::Value *EmitTypeidFromVTable(CodeGenFunction &CGF,
1472940f02d2SAnders Carlsson                                          const Expr *E,
14732192fe50SChris Lattner                                          llvm::Type *StdTypeInfoPtrTy) {
1474940f02d2SAnders Carlsson   // Get the vtable pointer.
1475940f02d2SAnders Carlsson   llvm::Value *ThisPtr = CGF.EmitLValue(E).getAddress();
1476940f02d2SAnders Carlsson 
1477940f02d2SAnders Carlsson   // C++ [expr.typeid]p2:
1478940f02d2SAnders Carlsson   //   If the glvalue expression is obtained by applying the unary * operator to
1479940f02d2SAnders Carlsson   //   a pointer and the pointer is a null pointer value, the typeid expression
1480940f02d2SAnders Carlsson   //   throws the std::bad_typeid exception.
1481940f02d2SAnders Carlsson   if (const UnaryOperator *UO = dyn_cast<UnaryOperator>(E->IgnoreParens())) {
1482940f02d2SAnders Carlsson     if (UO->getOpcode() == UO_Deref) {
1483940f02d2SAnders Carlsson       llvm::BasicBlock *BadTypeidBlock =
1484940f02d2SAnders Carlsson         CGF.createBasicBlock("typeid.bad_typeid");
1485940f02d2SAnders Carlsson       llvm::BasicBlock *EndBlock =
1486940f02d2SAnders Carlsson         CGF.createBasicBlock("typeid.end");
1487940f02d2SAnders Carlsson 
1488940f02d2SAnders Carlsson       llvm::Value *IsNull = CGF.Builder.CreateIsNull(ThisPtr);
1489940f02d2SAnders Carlsson       CGF.Builder.CreateCondBr(IsNull, BadTypeidBlock, EndBlock);
1490940f02d2SAnders Carlsson 
1491940f02d2SAnders Carlsson       CGF.EmitBlock(BadTypeidBlock);
1492940f02d2SAnders Carlsson       EmitBadTypeidCall(CGF);
1493940f02d2SAnders Carlsson       CGF.EmitBlock(EndBlock);
1494940f02d2SAnders Carlsson     }
1495940f02d2SAnders Carlsson   }
1496940f02d2SAnders Carlsson 
1497940f02d2SAnders Carlsson   llvm::Value *Value = CGF.GetVTablePtr(ThisPtr,
1498940f02d2SAnders Carlsson                                         StdTypeInfoPtrTy->getPointerTo());
1499940f02d2SAnders Carlsson 
1500940f02d2SAnders Carlsson   // Load the type info.
1501940f02d2SAnders Carlsson   Value = CGF.Builder.CreateConstInBoundsGEP1_64(Value, -1ULL);
1502940f02d2SAnders Carlsson   return CGF.Builder.CreateLoad(Value);
1503940f02d2SAnders Carlsson }
1504940f02d2SAnders Carlsson 
150559486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
15062192fe50SChris Lattner   llvm::Type *StdTypeInfoPtrTy =
1507940f02d2SAnders Carlsson     ConvertType(E->getType())->getPointerTo();
1508fd7dfeb7SAnders Carlsson 
15093f4336cbSAnders Carlsson   if (E->isTypeOperand()) {
15103f4336cbSAnders Carlsson     llvm::Constant *TypeInfo =
15113f4336cbSAnders Carlsson       CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand());
1512940f02d2SAnders Carlsson     return Builder.CreateBitCast(TypeInfo, StdTypeInfoPtrTy);
15133f4336cbSAnders Carlsson   }
1514fd7dfeb7SAnders Carlsson 
1515940f02d2SAnders Carlsson   // C++ [expr.typeid]p2:
1516940f02d2SAnders Carlsson   //   When typeid is applied to a glvalue expression whose type is a
1517940f02d2SAnders Carlsson   //   polymorphic class type, the result refers to a std::type_info object
1518940f02d2SAnders Carlsson   //   representing the type of the most derived object (that is, the dynamic
1519940f02d2SAnders Carlsson   //   type) to which the glvalue refers.
1520940f02d2SAnders Carlsson   if (E->getExprOperand()->isGLValue()) {
1521940f02d2SAnders Carlsson     if (const RecordType *RT =
1522940f02d2SAnders Carlsson           E->getExprOperand()->getType()->getAs<RecordType>()) {
152359486a2dSAnders Carlsson       const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
1524940f02d2SAnders Carlsson       if (RD->isPolymorphic())
1525940f02d2SAnders Carlsson         return EmitTypeidFromVTable(*this, E->getExprOperand(),
1526940f02d2SAnders Carlsson                                     StdTypeInfoPtrTy);
152759486a2dSAnders Carlsson     }
152859486a2dSAnders Carlsson   }
1529940f02d2SAnders Carlsson 
1530940f02d2SAnders Carlsson   QualType OperandTy = E->getExprOperand()->getType();
1531940f02d2SAnders Carlsson   return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(OperandTy),
1532940f02d2SAnders Carlsson                                StdTypeInfoPtrTy);
153359486a2dSAnders Carlsson }
153459486a2dSAnders Carlsson 
1535882d790fSAnders Carlsson static llvm::Constant *getDynamicCastFn(CodeGenFunction &CGF) {
1536882d790fSAnders Carlsson   // void *__dynamic_cast(const void *sub,
1537882d790fSAnders Carlsson   //                      const abi::__class_type_info *src,
1538882d790fSAnders Carlsson   //                      const abi::__class_type_info *dst,
1539882d790fSAnders Carlsson   //                      std::ptrdiff_t src2dst_offset);
1540882d790fSAnders Carlsson 
1541a5f58b05SChris Lattner   llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext());
1542a5f58b05SChris Lattner   llvm::Type *PtrDiffTy =
1543882d790fSAnders Carlsson     CGF.ConvertType(CGF.getContext().getPointerDiffType());
1544882d790fSAnders Carlsson 
1545a5f58b05SChris Lattner   llvm::Type *Args[4] = { Int8PtrTy, Int8PtrTy, Int8PtrTy, PtrDiffTy };
1546882d790fSAnders Carlsson 
15472192fe50SChris Lattner   llvm::FunctionType *FTy =
1548882d790fSAnders Carlsson     llvm::FunctionType::get(Int8PtrTy, Args, false);
1549882d790fSAnders Carlsson 
1550882d790fSAnders Carlsson   return CGF.CGM.CreateRuntimeFunction(FTy, "__dynamic_cast");
1551882d790fSAnders Carlsson }
1552882d790fSAnders Carlsson 
1553882d790fSAnders Carlsson static llvm::Constant *getBadCastFn(CodeGenFunction &CGF) {
1554882d790fSAnders Carlsson   // void __cxa_bad_cast();
1555882d790fSAnders Carlsson 
15562192fe50SChris Lattner   llvm::Type *VoidTy = llvm::Type::getVoidTy(CGF.getLLVMContext());
15572192fe50SChris Lattner   llvm::FunctionType *FTy =
1558882d790fSAnders Carlsson     llvm::FunctionType::get(VoidTy, false);
1559882d790fSAnders Carlsson 
1560882d790fSAnders Carlsson   return CGF.CGM.CreateRuntimeFunction(FTy, "__cxa_bad_cast");
1561882d790fSAnders Carlsson }
1562882d790fSAnders Carlsson 
1563c1c9971cSAnders Carlsson static void EmitBadCastCall(CodeGenFunction &CGF) {
1564bbe277c4SAnders Carlsson   llvm::Value *Fn = getBadCastFn(CGF);
15655bd375a6SJay Foad   CGF.EmitCallOrInvoke(Fn).setDoesNotReturn();
1566c1c9971cSAnders Carlsson   CGF.Builder.CreateUnreachable();
1567c1c9971cSAnders Carlsson }
1568c1c9971cSAnders Carlsson 
1569882d790fSAnders Carlsson static llvm::Value *
1570882d790fSAnders Carlsson EmitDynamicCastCall(CodeGenFunction &CGF, llvm::Value *Value,
1571882d790fSAnders Carlsson                     QualType SrcTy, QualType DestTy,
1572882d790fSAnders Carlsson                     llvm::BasicBlock *CastEnd) {
15732192fe50SChris Lattner   llvm::Type *PtrDiffLTy =
1574882d790fSAnders Carlsson     CGF.ConvertType(CGF.getContext().getPointerDiffType());
15752192fe50SChris Lattner   llvm::Type *DestLTy = CGF.ConvertType(DestTy);
1576882d790fSAnders Carlsson 
1577882d790fSAnders Carlsson   if (const PointerType *PTy = DestTy->getAs<PointerType>()) {
1578882d790fSAnders Carlsson     if (PTy->getPointeeType()->isVoidType()) {
1579882d790fSAnders Carlsson       // C++ [expr.dynamic.cast]p7:
1580882d790fSAnders Carlsson       //   If T is "pointer to cv void," then the result is a pointer to the
1581882d790fSAnders Carlsson       //   most derived object pointed to by v.
1582882d790fSAnders Carlsson 
1583882d790fSAnders Carlsson       // Get the vtable pointer.
1584882d790fSAnders Carlsson       llvm::Value *VTable = CGF.GetVTablePtr(Value, PtrDiffLTy->getPointerTo());
1585882d790fSAnders Carlsson 
1586882d790fSAnders Carlsson       // Get the offset-to-top from the vtable.
1587882d790fSAnders Carlsson       llvm::Value *OffsetToTop =
1588882d790fSAnders Carlsson         CGF.Builder.CreateConstInBoundsGEP1_64(VTable, -2ULL);
1589882d790fSAnders Carlsson       OffsetToTop = CGF.Builder.CreateLoad(OffsetToTop, "offset.to.top");
1590882d790fSAnders Carlsson 
1591882d790fSAnders Carlsson       // Finally, add the offset to the pointer.
1592882d790fSAnders Carlsson       Value = CGF.EmitCastToVoidPtr(Value);
1593882d790fSAnders Carlsson       Value = CGF.Builder.CreateInBoundsGEP(Value, OffsetToTop);
1594882d790fSAnders Carlsson 
1595882d790fSAnders Carlsson       return CGF.Builder.CreateBitCast(Value, DestLTy);
1596882d790fSAnders Carlsson     }
1597882d790fSAnders Carlsson   }
1598882d790fSAnders Carlsson 
1599882d790fSAnders Carlsson   QualType SrcRecordTy;
1600882d790fSAnders Carlsson   QualType DestRecordTy;
1601882d790fSAnders Carlsson 
1602882d790fSAnders Carlsson   if (const PointerType *DestPTy = DestTy->getAs<PointerType>()) {
1603882d790fSAnders Carlsson     SrcRecordTy = SrcTy->castAs<PointerType>()->getPointeeType();
1604882d790fSAnders Carlsson     DestRecordTy = DestPTy->getPointeeType();
1605882d790fSAnders Carlsson   } else {
1606882d790fSAnders Carlsson     SrcRecordTy = SrcTy;
1607882d790fSAnders Carlsson     DestRecordTy = DestTy->castAs<ReferenceType>()->getPointeeType();
1608882d790fSAnders Carlsson   }
1609882d790fSAnders Carlsson 
1610882d790fSAnders Carlsson   assert(SrcRecordTy->isRecordType() && "source type must be a record type!");
1611882d790fSAnders Carlsson   assert(DestRecordTy->isRecordType() && "dest type must be a record type!");
1612882d790fSAnders Carlsson 
1613882d790fSAnders Carlsson   llvm::Value *SrcRTTI =
1614882d790fSAnders Carlsson     CGF.CGM.GetAddrOfRTTIDescriptor(SrcRecordTy.getUnqualifiedType());
1615882d790fSAnders Carlsson   llvm::Value *DestRTTI =
1616882d790fSAnders Carlsson     CGF.CGM.GetAddrOfRTTIDescriptor(DestRecordTy.getUnqualifiedType());
1617882d790fSAnders Carlsson 
1618882d790fSAnders Carlsson   // FIXME: Actually compute a hint here.
1619882d790fSAnders Carlsson   llvm::Value *OffsetHint = llvm::ConstantInt::get(PtrDiffLTy, -1ULL);
1620882d790fSAnders Carlsson 
1621882d790fSAnders Carlsson   // Emit the call to __dynamic_cast.
1622882d790fSAnders Carlsson   Value = CGF.EmitCastToVoidPtr(Value);
1623882d790fSAnders Carlsson   Value = CGF.Builder.CreateCall4(getDynamicCastFn(CGF), Value,
1624882d790fSAnders Carlsson                                   SrcRTTI, DestRTTI, OffsetHint);
1625882d790fSAnders Carlsson   Value = CGF.Builder.CreateBitCast(Value, DestLTy);
1626882d790fSAnders Carlsson 
1627882d790fSAnders Carlsson   /// C++ [expr.dynamic.cast]p9:
1628882d790fSAnders Carlsson   ///   A failed cast to reference type throws std::bad_cast
1629882d790fSAnders Carlsson   if (DestTy->isReferenceType()) {
1630882d790fSAnders Carlsson     llvm::BasicBlock *BadCastBlock =
1631882d790fSAnders Carlsson       CGF.createBasicBlock("dynamic_cast.bad_cast");
1632882d790fSAnders Carlsson 
1633882d790fSAnders Carlsson     llvm::Value *IsNull = CGF.Builder.CreateIsNull(Value);
1634882d790fSAnders Carlsson     CGF.Builder.CreateCondBr(IsNull, BadCastBlock, CastEnd);
1635882d790fSAnders Carlsson 
1636882d790fSAnders Carlsson     CGF.EmitBlock(BadCastBlock);
1637c1c9971cSAnders Carlsson     EmitBadCastCall(CGF);
1638882d790fSAnders Carlsson   }
1639882d790fSAnders Carlsson 
1640882d790fSAnders Carlsson   return Value;
1641882d790fSAnders Carlsson }
1642882d790fSAnders Carlsson 
1643c1c9971cSAnders Carlsson static llvm::Value *EmitDynamicCastToNull(CodeGenFunction &CGF,
1644c1c9971cSAnders Carlsson                                           QualType DestTy) {
16452192fe50SChris Lattner   llvm::Type *DestLTy = CGF.ConvertType(DestTy);
1646c1c9971cSAnders Carlsson   if (DestTy->isPointerType())
1647c1c9971cSAnders Carlsson     return llvm::Constant::getNullValue(DestLTy);
1648c1c9971cSAnders Carlsson 
1649c1c9971cSAnders Carlsson   /// C++ [expr.dynamic.cast]p9:
1650c1c9971cSAnders Carlsson   ///   A failed cast to reference type throws std::bad_cast
1651c1c9971cSAnders Carlsson   EmitBadCastCall(CGF);
1652c1c9971cSAnders Carlsson 
1653c1c9971cSAnders Carlsson   CGF.EmitBlock(CGF.createBasicBlock("dynamic_cast.end"));
1654c1c9971cSAnders Carlsson   return llvm::UndefValue::get(DestLTy);
1655c1c9971cSAnders Carlsson }
1656c1c9971cSAnders Carlsson 
1657882d790fSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *Value,
165859486a2dSAnders Carlsson                                               const CXXDynamicCastExpr *DCE) {
16593f4336cbSAnders Carlsson   QualType DestTy = DCE->getTypeAsWritten();
16603f4336cbSAnders Carlsson 
1661c1c9971cSAnders Carlsson   if (DCE->isAlwaysNull())
1662c1c9971cSAnders Carlsson     return EmitDynamicCastToNull(*this, DestTy);
1663c1c9971cSAnders Carlsson 
1664c1c9971cSAnders Carlsson   QualType SrcTy = DCE->getSubExpr()->getType();
1665c1c9971cSAnders Carlsson 
1666882d790fSAnders Carlsson   // C++ [expr.dynamic.cast]p4:
1667882d790fSAnders Carlsson   //   If the value of v is a null pointer value in the pointer case, the result
1668882d790fSAnders Carlsson   //   is the null pointer value of type T.
1669882d790fSAnders Carlsson   bool ShouldNullCheckSrcValue = SrcTy->isPointerType();
167059486a2dSAnders Carlsson 
1671882d790fSAnders Carlsson   llvm::BasicBlock *CastNull = 0;
1672882d790fSAnders Carlsson   llvm::BasicBlock *CastNotNull = 0;
1673882d790fSAnders Carlsson   llvm::BasicBlock *CastEnd = createBasicBlock("dynamic_cast.end");
1674fa8b4955SDouglas Gregor 
1675882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1676882d790fSAnders Carlsson     CastNull = createBasicBlock("dynamic_cast.null");
1677882d790fSAnders Carlsson     CastNotNull = createBasicBlock("dynamic_cast.notnull");
1678882d790fSAnders Carlsson 
1679882d790fSAnders Carlsson     llvm::Value *IsNull = Builder.CreateIsNull(Value);
1680882d790fSAnders Carlsson     Builder.CreateCondBr(IsNull, CastNull, CastNotNull);
1681882d790fSAnders Carlsson     EmitBlock(CastNotNull);
168259486a2dSAnders Carlsson   }
168359486a2dSAnders Carlsson 
1684882d790fSAnders Carlsson   Value = EmitDynamicCastCall(*this, Value, SrcTy, DestTy, CastEnd);
16853f4336cbSAnders Carlsson 
1686882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1687882d790fSAnders Carlsson     EmitBranch(CastEnd);
168859486a2dSAnders Carlsson 
1689882d790fSAnders Carlsson     EmitBlock(CastNull);
1690882d790fSAnders Carlsson     EmitBranch(CastEnd);
169159486a2dSAnders Carlsson   }
169259486a2dSAnders Carlsson 
1693882d790fSAnders Carlsson   EmitBlock(CastEnd);
169459486a2dSAnders Carlsson 
1695882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1696882d790fSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(Value->getType(), 2);
1697882d790fSAnders Carlsson     PHI->addIncoming(Value, CastNotNull);
1698882d790fSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(Value->getType()), CastNull);
169959486a2dSAnders Carlsson 
1700882d790fSAnders Carlsson     Value = PHI;
170159486a2dSAnders Carlsson   }
170259486a2dSAnders Carlsson 
1703882d790fSAnders Carlsson   return Value;
170459486a2dSAnders Carlsson }
1705