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"
2059486a2dSAnders Carlsson using namespace clang;
2159486a2dSAnders Carlsson using namespace CodeGen;
2259486a2dSAnders Carlsson 
2327da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD,
2427da15baSAnders Carlsson                                           llvm::Value *Callee,
2527da15baSAnders Carlsson                                           ReturnValueSlot ReturnValue,
2627da15baSAnders Carlsson                                           llvm::Value *This,
27e36a6b3eSAnders Carlsson                                           llvm::Value *VTT,
2827da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgBeg,
2927da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgEnd) {
3027da15baSAnders Carlsson   assert(MD->isInstance() &&
3127da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
3227da15baSAnders Carlsson 
3327da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
3427da15baSAnders Carlsson 
3527da15baSAnders Carlsson   CallArgList Args;
3627da15baSAnders Carlsson 
3727da15baSAnders Carlsson   // Push the this ptr.
3827da15baSAnders Carlsson   Args.push_back(std::make_pair(RValue::get(This),
3927da15baSAnders Carlsson                                 MD->getThisType(getContext())));
4027da15baSAnders Carlsson 
41e36a6b3eSAnders Carlsson   // If there is a VTT parameter, emit it.
42e36a6b3eSAnders Carlsson   if (VTT) {
43e36a6b3eSAnders Carlsson     QualType T = getContext().getPointerType(getContext().VoidPtrTy);
44e36a6b3eSAnders Carlsson     Args.push_back(std::make_pair(RValue::get(VTT), T));
45e36a6b3eSAnders Carlsson   }
46e36a6b3eSAnders Carlsson 
4727da15baSAnders Carlsson   // And the rest of the call args
4827da15baSAnders Carlsson   EmitCallArgs(Args, FPT, ArgBeg, ArgEnd);
4927da15baSAnders Carlsson 
50ab26cfa5SJohn McCall   QualType ResultType = FPT->getResultType();
51ab26cfa5SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args,
52c50c27ccSRafael Espindola                                                  FPT->getExtInfo()),
53c50c27ccSRafael Espindola                   Callee, ReturnValue, Args, MD);
5427da15baSAnders Carlsson }
5527da15baSAnders Carlsson 
5627da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given
5727da15baSAnders Carlsson /// expr can be devirtualized.
58a7911fa3SAnders Carlsson static bool canDevirtualizeMemberFunctionCalls(const Expr *Base,
59a7911fa3SAnders Carlsson                                                const CXXMethodDecl *MD) {
60a7911fa3SAnders Carlsson 
61a7911fa3SAnders Carlsson   // If the member function has the "final" attribute, we know that it can't be
62b00c2144SAnders Carlsson   // overridden and can therefore devirtualize it.
63a7911fa3SAnders Carlsson   if (MD->hasAttr<FinalAttr>())
64a7911fa3SAnders Carlsson     return true;
65a7911fa3SAnders Carlsson 
66b00c2144SAnders Carlsson   // Similarly, if the class itself has the "final" attribute it can't be
67b00c2144SAnders Carlsson   // overridden and we can therefore devirtualize the member function call.
68b00c2144SAnders Carlsson   if (MD->getParent()->hasAttr<FinalAttr>())
69b00c2144SAnders Carlsson     return true;
70b00c2144SAnders Carlsson 
7127da15baSAnders Carlsson   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) {
7227da15baSAnders Carlsson     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
7327da15baSAnders Carlsson       // This is a record decl. We know the type and can devirtualize it.
7427da15baSAnders Carlsson       return VD->getType()->isRecordType();
7527da15baSAnders Carlsson     }
7627da15baSAnders Carlsson 
7727da15baSAnders Carlsson     return false;
7827da15baSAnders Carlsson   }
7927da15baSAnders Carlsson 
8027da15baSAnders Carlsson   // We can always devirtualize calls on temporary object expressions.
81a682427eSEli Friedman   if (isa<CXXConstructExpr>(Base))
8227da15baSAnders Carlsson     return true;
8327da15baSAnders Carlsson 
8427da15baSAnders Carlsson   // And calls on bound temporaries.
8527da15baSAnders Carlsson   if (isa<CXXBindTemporaryExpr>(Base))
8627da15baSAnders Carlsson     return true;
8727da15baSAnders Carlsson 
8827da15baSAnders Carlsson   // Check if this is a call expr that returns a record type.
8927da15baSAnders Carlsson   if (const CallExpr *CE = dyn_cast<CallExpr>(Base))
9027da15baSAnders Carlsson     return CE->getCallReturnType()->isRecordType();
9127da15baSAnders Carlsson 
9227da15baSAnders Carlsson   // We can't devirtualize the call.
9327da15baSAnders Carlsson   return false;
9427da15baSAnders Carlsson }
9527da15baSAnders Carlsson 
9627da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
9727da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
9827da15baSAnders Carlsson   if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens()))
9927da15baSAnders Carlsson     return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
10027da15baSAnders Carlsson 
10127da15baSAnders Carlsson   const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens());
10227da15baSAnders Carlsson   const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
10327da15baSAnders Carlsson 
10491bbb554SDevang Patel   CGDebugInfo *DI = getDebugInfo();
105401c916cSDevang Patel   if (DI && CGM.getCodeGenOpts().LimitDebugInfo
106401c916cSDevang Patel       && !isa<CallExpr>(ME->getBase())) {
10791bbb554SDevang Patel     QualType PQTy = ME->getBase()->IgnoreParenImpCasts()->getType();
10891bbb554SDevang Patel     if (const PointerType * PTy = dyn_cast<PointerType>(PQTy)) {
10991bbb554SDevang Patel       DI->getOrCreateRecordType(PTy->getPointeeType(),
11091bbb554SDevang Patel                                 MD->getParent()->getLocation());
11191bbb554SDevang Patel     }
11291bbb554SDevang Patel   }
11391bbb554SDevang Patel 
11427da15baSAnders Carlsson   if (MD->isStatic()) {
11527da15baSAnders Carlsson     // The method is static, emit it as we would a regular call.
11627da15baSAnders Carlsson     llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
11727da15baSAnders Carlsson     return EmitCall(getContext().getPointerType(MD->getType()), Callee,
11827da15baSAnders Carlsson                     ReturnValue, CE->arg_begin(), CE->arg_end());
11927da15baSAnders Carlsson   }
12027da15baSAnders Carlsson 
1210d635f53SJohn McCall   // Compute the object pointer.
12227da15baSAnders Carlsson   llvm::Value *This;
12327da15baSAnders Carlsson   if (ME->isArrow())
12427da15baSAnders Carlsson     This = EmitScalarExpr(ME->getBase());
12527da15baSAnders Carlsson   else {
12627da15baSAnders Carlsson     LValue BaseLV = EmitLValue(ME->getBase());
127f93ac894SFariborz Jahanian     if (BaseLV.isPropertyRef() || BaseLV.isKVCRef()) {
128f93ac894SFariborz Jahanian       QualType QT = ME->getBase()->getType();
129f93ac894SFariborz Jahanian       RValue RV =
130f93ac894SFariborz Jahanian         BaseLV.isPropertyRef() ? EmitLoadOfPropertyRefLValue(BaseLV, QT)
131f93ac894SFariborz Jahanian           : EmitLoadOfKVCRefLValue(BaseLV, QT);
132f93ac894SFariborz Jahanian       This = RV.isScalar() ? RV.getScalarVal() : RV.getAggregateAddr();
133f93ac894SFariborz Jahanian     }
134f93ac894SFariborz Jahanian     else
13527da15baSAnders Carlsson       This = BaseLV.getAddress();
13627da15baSAnders Carlsson   }
13727da15baSAnders Carlsson 
1380d635f53SJohn McCall   if (MD->isTrivial()) {
1390d635f53SJohn McCall     if (isa<CXXDestructorDecl>(MD)) return RValue::get(0);
1400d635f53SJohn McCall 
141ec3bec0cSDouglas Gregor     assert(MD->isCopyAssignmentOperator() && "unknown trivial member function");
14227da15baSAnders Carlsson     // We don't like to generate the trivial copy assignment operator when
14327da15baSAnders Carlsson     // it isn't necessary; just produce the proper effect here.
14427da15baSAnders Carlsson     llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
14527da15baSAnders Carlsson     EmitAggregateCopy(This, RHS, CE->getType());
14627da15baSAnders Carlsson     return RValue::get(This);
14727da15baSAnders Carlsson   }
14827da15baSAnders Carlsson 
1490d635f53SJohn McCall   // Compute the function type we're calling.
1500d635f53SJohn McCall   const CGFunctionInfo &FInfo =
1510d635f53SJohn McCall     (isa<CXXDestructorDecl>(MD)
1520d635f53SJohn McCall      ? CGM.getTypes().getFunctionInfo(cast<CXXDestructorDecl>(MD),
1530d635f53SJohn McCall                                       Dtor_Complete)
1540d635f53SJohn McCall      : CGM.getTypes().getFunctionInfo(MD));
1550d635f53SJohn McCall 
1560d635f53SJohn McCall   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
1570d635f53SJohn McCall   const llvm::Type *Ty
1580d635f53SJohn McCall     = CGM.getTypes().GetFunctionType(FInfo, FPT->isVariadic());
1590d635f53SJohn McCall 
16027da15baSAnders Carlsson   // C++ [class.virtual]p12:
16127da15baSAnders Carlsson   //   Explicit qualification with the scope operator (5.1) suppresses the
16227da15baSAnders Carlsson   //   virtual call mechanism.
16327da15baSAnders Carlsson   //
16427da15baSAnders Carlsson   // We also don't emit a virtual call if the base expression has a record type
16527da15baSAnders Carlsson   // because then we know what the type is.
1660d635f53SJohn McCall   bool UseVirtualCall = MD->isVirtual() && !ME->hasQualifier()
167a7911fa3SAnders Carlsson                      && !canDevirtualizeMemberFunctionCalls(ME->getBase(), MD);
1680d635f53SJohn McCall 
16927da15baSAnders Carlsson   llvm::Value *Callee;
1700d635f53SJohn McCall   if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(MD)) {
1710d635f53SJohn McCall     if (UseVirtualCall) {
1720d635f53SJohn McCall       Callee = BuildVirtualCall(Dtor, Dtor_Complete, This, Ty);
17327da15baSAnders Carlsson     } else {
1740d635f53SJohn McCall       Callee = CGM.GetAddrOfFunction(GlobalDecl(Dtor, Dtor_Complete), Ty);
17527da15baSAnders Carlsson     }
1760d635f53SJohn McCall   } else if (UseVirtualCall) {
17727da15baSAnders Carlsson     Callee = BuildVirtualCall(MD, This, Ty);
17827da15baSAnders Carlsson   } else {
17927da15baSAnders Carlsson     Callee = CGM.GetAddrOfFunction(MD, Ty);
18027da15baSAnders Carlsson   }
18127da15baSAnders Carlsson 
182e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
18327da15baSAnders Carlsson                            CE->arg_begin(), CE->arg_end());
18427da15baSAnders Carlsson }
18527da15baSAnders Carlsson 
18627da15baSAnders Carlsson RValue
18727da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
18827da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
18927da15baSAnders Carlsson   const BinaryOperator *BO =
19027da15baSAnders Carlsson       cast<BinaryOperator>(E->getCallee()->IgnoreParens());
19127da15baSAnders Carlsson   const Expr *BaseExpr = BO->getLHS();
19227da15baSAnders Carlsson   const Expr *MemFnExpr = BO->getRHS();
19327da15baSAnders Carlsson 
19427da15baSAnders Carlsson   const MemberPointerType *MPT =
19527da15baSAnders Carlsson     MemFnExpr->getType()->getAs<MemberPointerType>();
196475999dcSJohn McCall 
19727da15baSAnders Carlsson   const FunctionProtoType *FPT =
19827da15baSAnders Carlsson     MPT->getPointeeType()->getAs<FunctionProtoType>();
19927da15baSAnders Carlsson   const CXXRecordDecl *RD =
20027da15baSAnders Carlsson     cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
20127da15baSAnders Carlsson 
20227da15baSAnders Carlsson   // Get the member function pointer.
203a1dee530SJohn McCall   llvm::Value *MemFnPtr = EmitScalarExpr(MemFnExpr);
20427da15baSAnders Carlsson 
20527da15baSAnders Carlsson   // Emit the 'this' pointer.
20627da15baSAnders Carlsson   llvm::Value *This;
20727da15baSAnders Carlsson 
208e302792bSJohn McCall   if (BO->getOpcode() == BO_PtrMemI)
20927da15baSAnders Carlsson     This = EmitScalarExpr(BaseExpr);
21027da15baSAnders Carlsson   else
21127da15baSAnders Carlsson     This = EmitLValue(BaseExpr).getAddress();
21227da15baSAnders Carlsson 
213475999dcSJohn McCall   // Ask the ABI to load the callee.  Note that This is modified.
214475999dcSJohn McCall   llvm::Value *Callee =
215475999dcSJohn McCall     CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(CGF, This, MemFnPtr, MPT);
21627da15baSAnders Carlsson 
21727da15baSAnders Carlsson   CallArgList Args;
21827da15baSAnders Carlsson 
21927da15baSAnders Carlsson   QualType ThisType =
22027da15baSAnders Carlsson     getContext().getPointerType(getContext().getTagDeclType(RD));
22127da15baSAnders Carlsson 
22227da15baSAnders Carlsson   // Push the this ptr.
22327da15baSAnders Carlsson   Args.push_back(std::make_pair(RValue::get(This), ThisType));
22427da15baSAnders Carlsson 
22527da15baSAnders Carlsson   // And the rest of the call args
22627da15baSAnders Carlsson   EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
227ab26cfa5SJohn McCall   const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>();
228ab26cfa5SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee,
22927da15baSAnders Carlsson                   ReturnValue, Args);
23027da15baSAnders Carlsson }
23127da15baSAnders Carlsson 
23227da15baSAnders Carlsson RValue
23327da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
23427da15baSAnders Carlsson                                                const CXXMethodDecl *MD,
23527da15baSAnders Carlsson                                                ReturnValueSlot ReturnValue) {
23627da15baSAnders Carlsson   assert(MD->isInstance() &&
23727da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
238ec3bec0cSDouglas Gregor   if (MD->isCopyAssignmentOperator()) {
23927da15baSAnders Carlsson     const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext());
24027da15baSAnders Carlsson     if (ClassDecl->hasTrivialCopyAssignment()) {
24127da15baSAnders Carlsson       assert(!ClassDecl->hasUserDeclaredCopyAssignment() &&
24227da15baSAnders Carlsson              "EmitCXXOperatorMemberCallExpr - user declared copy assignment");
24343a40f93SFariborz Jahanian       LValue LV = EmitLValue(E->getArg(0));
24443a40f93SFariborz Jahanian       llvm::Value *This;
24561a31241SFariborz Jahanian       if (LV.isPropertyRef() || LV.isKVCRef()) {
2467a626f63SJohn McCall         AggValueSlot Slot = CreateAggTemp(E->getArg(1)->getType());
2477a626f63SJohn McCall         EmitAggExpr(E->getArg(1), Slot);
24861a31241SFariborz Jahanian         if (LV.isPropertyRef())
2497a626f63SJohn McCall           EmitObjCPropertySet(LV.getPropertyRefExpr(), Slot.asRValue());
25061a31241SFariborz Jahanian         else
2517a626f63SJohn McCall           EmitObjCPropertySet(LV.getKVCRefExpr(), Slot.asRValue());
252e1b45a5eSFariborz Jahanian         return RValue::getAggregate(0, false);
25343a40f93SFariborz Jahanian       }
25443a40f93SFariborz Jahanian       else
25543a40f93SFariborz Jahanian         This = LV.getAddress();
25643a40f93SFariborz Jahanian 
25727da15baSAnders Carlsson       llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
25827da15baSAnders Carlsson       QualType Ty = E->getType();
25927da15baSAnders Carlsson       EmitAggregateCopy(This, Src, Ty);
26027da15baSAnders Carlsson       return RValue::get(This);
26127da15baSAnders Carlsson     }
26227da15baSAnders Carlsson   }
26327da15baSAnders Carlsson 
26427da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
26527da15baSAnders Carlsson   const llvm::Type *Ty =
26627da15baSAnders Carlsson     CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
26727da15baSAnders Carlsson                                    FPT->isVariadic());
268fdf474b0SFariborz Jahanian   LValue LV = EmitLValue(E->getArg(0));
269fdf474b0SFariborz Jahanian   llvm::Value *This;
27061a31241SFariborz Jahanian   if (LV.isPropertyRef() || LV.isKVCRef()) {
27161a31241SFariborz Jahanian     QualType QT = E->getArg(0)->getType();
27261a31241SFariborz Jahanian     RValue RV =
27361a31241SFariborz Jahanian       LV.isPropertyRef() ? EmitLoadOfPropertyRefLValue(LV, QT)
27461a31241SFariborz Jahanian                          : EmitLoadOfKVCRefLValue(LV, QT);
2756855ba2cSFariborz Jahanian     assert (!RV.isScalar() && "EmitCXXOperatorMemberCallExpr");
2766855ba2cSFariborz Jahanian     This = RV.getAggregateAddr();
277fdf474b0SFariborz Jahanian   }
278fdf474b0SFariborz Jahanian   else
279fdf474b0SFariborz Jahanian     This = LV.getAddress();
28027da15baSAnders Carlsson 
28127da15baSAnders Carlsson   llvm::Value *Callee;
282a7911fa3SAnders Carlsson   if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0), MD))
28327da15baSAnders Carlsson     Callee = BuildVirtualCall(MD, This, Ty);
28427da15baSAnders Carlsson   else
28527da15baSAnders Carlsson     Callee = CGM.GetAddrOfFunction(MD, Ty);
28627da15baSAnders Carlsson 
287e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
28827da15baSAnders Carlsson                            E->arg_begin() + 1, E->arg_end());
28927da15baSAnders Carlsson }
29027da15baSAnders Carlsson 
29127da15baSAnders Carlsson void
2927a626f63SJohn McCall CodeGenFunction::EmitCXXConstructExpr(const CXXConstructExpr *E,
2937a626f63SJohn McCall                                       AggValueSlot Dest) {
2947a626f63SJohn McCall   assert(!Dest.isIgnored() && "Must have a destination!");
29527da15baSAnders Carlsson   const CXXConstructorDecl *CD = E->getConstructor();
296630c76efSDouglas Gregor 
297630c76efSDouglas Gregor   // If we require zero initialization before (or instead of) calling the
298630c76efSDouglas Gregor   // constructor, as can be the case with a non-user-provided default
299630c76efSDouglas Gregor   // constructor, emit the zero initialization now.
300e3b3464dSDouglas Gregor   if (E->requiresZeroInitialization())
3017a626f63SJohn McCall     EmitNullInitialization(Dest.getAddr(), E->getType());
302630c76efSDouglas Gregor 
303630c76efSDouglas Gregor   // If this is a call to a trivial default constructor, do nothing.
304630c76efSDouglas Gregor   if (CD->isTrivial() && CD->isDefaultConstructor())
30527da15baSAnders Carlsson     return;
306630c76efSDouglas Gregor 
3078ea46b66SJohn McCall   // Elide the constructor if we're constructing from a temporary.
3088ea46b66SJohn McCall   // The temporary check is required because Sema sets this on NRVO
3098ea46b66SJohn McCall   // returns.
31027da15baSAnders Carlsson   if (getContext().getLangOptions().ElideConstructors && E->isElidable()) {
3118ea46b66SJohn McCall     assert(getContext().hasSameUnqualifiedType(E->getType(),
3128ea46b66SJohn McCall                                                E->getArg(0)->getType()));
3137a626f63SJohn McCall     if (E->getArg(0)->isTemporaryObject(getContext(), CD->getParent())) {
3147a626f63SJohn McCall       EmitAggExpr(E->getArg(0), Dest);
31527da15baSAnders Carlsson       return;
31627da15baSAnders Carlsson     }
317222cf0efSDouglas Gregor   }
318630c76efSDouglas Gregor 
319630c76efSDouglas Gregor   const ConstantArrayType *Array
320630c76efSDouglas Gregor     = getContext().getAsConstantArrayType(E->getType());
32127da15baSAnders Carlsson   if (Array) {
32227da15baSAnders Carlsson     QualType BaseElementTy = getContext().getBaseElementType(Array);
32327da15baSAnders Carlsson     const llvm::Type *BasePtr = ConvertType(BaseElementTy);
32427da15baSAnders Carlsson     BasePtr = llvm::PointerType::getUnqual(BasePtr);
32527da15baSAnders Carlsson     llvm::Value *BaseAddrPtr =
3267a626f63SJohn McCall       Builder.CreateBitCast(Dest.getAddr(), BasePtr);
32727da15baSAnders Carlsson 
32827da15baSAnders Carlsson     EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr,
32927da15baSAnders Carlsson                                E->arg_begin(), E->arg_end());
33027da15baSAnders Carlsson   }
331e11f9ce9SAnders Carlsson   else {
332e11f9ce9SAnders Carlsson     CXXCtorType Type =
333e11f9ce9SAnders Carlsson       (E->getConstructionKind() == CXXConstructExpr::CK_Complete)
334e11f9ce9SAnders Carlsson       ? Ctor_Complete : Ctor_Base;
335e11f9ce9SAnders Carlsson     bool ForVirtualBase =
336e11f9ce9SAnders Carlsson       E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase;
337e11f9ce9SAnders Carlsson 
33827da15baSAnders Carlsson     // Call the constructor.
3397a626f63SJohn McCall     EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest.getAddr(),
34027da15baSAnders Carlsson                            E->arg_begin(), E->arg_end());
34127da15baSAnders Carlsson   }
342e11f9ce9SAnders Carlsson }
34327da15baSAnders Carlsson 
344e988bdacSFariborz Jahanian void
345e988bdacSFariborz Jahanian CodeGenFunction::EmitSynthesizedCXXCopyCtor(llvm::Value *Dest,
346e988bdacSFariborz Jahanian                                             llvm::Value *Src,
347e988bdacSFariborz Jahanian                                             const BlockDeclRefExpr *BDRE) {
348e988bdacSFariborz Jahanian   const Expr *Exp = BDRE->getCopyConstructorExpr();
349e988bdacSFariborz Jahanian   if (const CXXExprWithTemporaries *E = dyn_cast<CXXExprWithTemporaries>(Exp))
350e988bdacSFariborz Jahanian     Exp = E->getSubExpr();
351e988bdacSFariborz Jahanian   assert(isa<CXXConstructExpr>(Exp) &&
352e988bdacSFariborz Jahanian          "EmitSynthesizedCXXCopyCtor - unknown copy ctor expr");
353e988bdacSFariborz Jahanian   const CXXConstructExpr* E = cast<CXXConstructExpr>(Exp);
354e988bdacSFariborz Jahanian   const CXXConstructorDecl *CD = E->getConstructor();
355e988bdacSFariborz Jahanian   RunCleanupsScope Scope(*this);
356e988bdacSFariborz Jahanian 
357e988bdacSFariborz Jahanian   // If we require zero initialization before (or instead of) calling the
358e988bdacSFariborz Jahanian   // constructor, as can be the case with a non-user-provided default
359e988bdacSFariborz Jahanian   // constructor, emit the zero initialization now.
360e988bdacSFariborz Jahanian   // FIXME. Do I still need this for a copy ctor synthesis?
361e988bdacSFariborz Jahanian   if (E->requiresZeroInitialization())
362e988bdacSFariborz Jahanian     EmitNullInitialization(Dest, E->getType());
363e988bdacSFariborz Jahanian 
364*99da11cfSChandler Carruth   assert(!getContext().getAsConstantArrayType(E->getType())
365*99da11cfSChandler Carruth          && "EmitSynthesizedCXXCopyCtor - Copied-in Array");
366e988bdacSFariborz Jahanian   EmitSynthesizedCXXCopyCtorCall(CD, Dest, Src,
367e988bdacSFariborz Jahanian                                  E->arg_begin(), E->arg_end());
368e988bdacSFariborz Jahanian }
369e988bdacSFariborz Jahanian 
370aa4149a2SJohn McCall /// Check whether the given operator new[] is the global placement
371aa4149a2SJohn McCall /// operator new[].
372aa4149a2SJohn McCall static bool IsPlacementOperatorNewArray(ASTContext &Ctx,
373aa4149a2SJohn McCall                                         const FunctionDecl *Fn) {
374aa4149a2SJohn McCall   // Must be in global scope.  Note that allocation functions can't be
375aa4149a2SJohn McCall   // declared in namespaces.
37650c68258SSebastian Redl   if (!Fn->getDeclContext()->getRedeclContext()->isFileContext())
377aa4149a2SJohn McCall     return false;
378aa4149a2SJohn McCall 
379aa4149a2SJohn McCall   // Signature must be void *operator new[](size_t, void*).
380aa4149a2SJohn McCall   // The size_t is common to all operator new[]s.
381aa4149a2SJohn McCall   if (Fn->getNumParams() != 2)
382aa4149a2SJohn McCall     return false;
383aa4149a2SJohn McCall 
384aa4149a2SJohn McCall   CanQualType ParamType = Ctx.getCanonicalType(Fn->getParamDecl(1)->getType());
385aa4149a2SJohn McCall   return (ParamType == Ctx.VoidPtrTy);
386aa4149a2SJohn McCall }
387aa4149a2SJohn McCall 
3888ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF,
3898ed55a54SJohn McCall                                         const CXXNewExpr *E) {
39021122cf6SAnders Carlsson   if (!E->isArray())
3913eb55cfeSKen Dyck     return CharUnits::Zero();
39221122cf6SAnders Carlsson 
393399f499fSAnders Carlsson   // No cookie is required if the new operator being used is
394399f499fSAnders Carlsson   // ::operator new[](size_t, void*).
395399f499fSAnders Carlsson   const FunctionDecl *OperatorNew = E->getOperatorNew();
3968ed55a54SJohn McCall   if (IsPlacementOperatorNewArray(CGF.getContext(), OperatorNew))
3973eb55cfeSKen Dyck     return CharUnits::Zero();
398399f499fSAnders Carlsson 
3998ed55a54SJohn McCall   return CGF.CGM.getCXXABI().GetArrayCookieSize(E->getAllocatedType());
40059486a2dSAnders Carlsson }
40159486a2dSAnders Carlsson 
40247b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context,
40347b4629bSFariborz Jahanian                                         CodeGenFunction &CGF,
40459486a2dSAnders Carlsson                                         const CXXNewExpr *E,
40505fc5be3SDouglas Gregor                                         llvm::Value *&NumElements,
40605fc5be3SDouglas Gregor                                         llvm::Value *&SizeWithoutCookie) {
4077648fb46SArgyrios Kyrtzidis   QualType ElemType = E->getAllocatedType();
40859486a2dSAnders Carlsson 
4098ed55a54SJohn McCall   const llvm::IntegerType *SizeTy =
4108ed55a54SJohn McCall     cast<llvm::IntegerType>(CGF.ConvertType(CGF.getContext().getSizeType()));
4118ed55a54SJohn McCall 
4127648fb46SArgyrios Kyrtzidis   CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(ElemType);
4138ed55a54SJohn McCall 
4148ed55a54SJohn McCall   if (!E->isArray()) {
41505fc5be3SDouglas Gregor     SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
41605fc5be3SDouglas Gregor     return SizeWithoutCookie;
41705fc5be3SDouglas Gregor   }
41859486a2dSAnders Carlsson 
4198ed55a54SJohn McCall   // Figure out the cookie size.
4208ed55a54SJohn McCall   CharUnits CookieSize = CalculateCookiePadding(CGF, E);
4218ed55a54SJohn McCall 
42259486a2dSAnders Carlsson   // Emit the array size expression.
4237648fb46SArgyrios Kyrtzidis   // We multiply the size of all dimensions for NumElements.
4247648fb46SArgyrios Kyrtzidis   // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6.
42559486a2dSAnders Carlsson   NumElements = CGF.EmitScalarExpr(E->getArraySize());
4268ed55a54SJohn McCall   assert(NumElements->getType() == SizeTy && "element count not a size_t");
4278ed55a54SJohn McCall 
4288ed55a54SJohn McCall   uint64_t ArraySizeMultiplier = 1;
4297648fb46SArgyrios Kyrtzidis   while (const ConstantArrayType *CAT
4307648fb46SArgyrios Kyrtzidis              = CGF.getContext().getAsConstantArrayType(ElemType)) {
4317648fb46SArgyrios Kyrtzidis     ElemType = CAT->getElementType();
4328ed55a54SJohn McCall     ArraySizeMultiplier *= CAT->getSize().getZExtValue();
4337648fb46SArgyrios Kyrtzidis   }
43459486a2dSAnders Carlsson 
4358ed55a54SJohn McCall   llvm::Value *Size;
43632ac583dSChris Lattner 
43732ac583dSChris Lattner   // If someone is doing 'new int[42]' there is no need to do a dynamic check.
43832ac583dSChris Lattner   // Don't bloat the -O0 code.
43932ac583dSChris Lattner   if (llvm::ConstantInt *NumElementsC =
44032ac583dSChris Lattner         dyn_cast<llvm::ConstantInt>(NumElements)) {
44132ac583dSChris Lattner     llvm::APInt NEC = NumElementsC->getValue();
4428ed55a54SJohn McCall     unsigned SizeWidth = NEC.getBitWidth();
44332ac583dSChris Lattner 
4448ed55a54SJohn McCall     // Determine if there is an overflow here by doing an extended multiply.
4458ed55a54SJohn McCall     NEC.zext(SizeWidth*2);
4468ed55a54SJohn McCall     llvm::APInt SC(SizeWidth*2, TypeSize.getQuantity());
44732ac583dSChris Lattner     SC *= NEC;
44832ac583dSChris Lattner 
4498ed55a54SJohn McCall     if (!CookieSize.isZero()) {
4508ed55a54SJohn McCall       // Save the current size without a cookie.  We don't care if an
4518ed55a54SJohn McCall       // overflow's already happened because SizeWithoutCookie isn't
4528ed55a54SJohn McCall       // used if the allocator returns null or throws, as it should
4538ed55a54SJohn McCall       // always do on an overflow.
4548ed55a54SJohn McCall       llvm::APInt SWC = SC;
4558ed55a54SJohn McCall       SWC.trunc(SizeWidth);
4568ed55a54SJohn McCall       SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, SWC);
4578ed55a54SJohn McCall 
4588ed55a54SJohn McCall       // Add the cookie size.
4598ed55a54SJohn McCall       SC += llvm::APInt(SizeWidth*2, CookieSize.getQuantity());
4608ed55a54SJohn McCall     }
4618ed55a54SJohn McCall 
4628ed55a54SJohn McCall     if (SC.countLeadingZeros() >= SizeWidth) {
4638ed55a54SJohn McCall       SC.trunc(SizeWidth);
4648ed55a54SJohn McCall       Size = llvm::ConstantInt::get(SizeTy, SC);
46532ac583dSChris Lattner     } else {
46632ac583dSChris Lattner       // On overflow, produce a -1 so operator new throws.
4678ed55a54SJohn McCall       Size = llvm::Constant::getAllOnesValue(SizeTy);
46832ac583dSChris Lattner     }
46932ac583dSChris Lattner 
4708ed55a54SJohn McCall     // Scale NumElements while we're at it.
4718ed55a54SJohn McCall     uint64_t N = NEC.getZExtValue() * ArraySizeMultiplier;
4728ed55a54SJohn McCall     NumElements = llvm::ConstantInt::get(SizeTy, N);
47347b4629bSFariborz Jahanian 
4748ed55a54SJohn McCall   // Otherwise, we don't need to do an overflow-checked multiplication if
4758ed55a54SJohn McCall   // we're multiplying by one.
4768ed55a54SJohn McCall   } else if (TypeSize.isOne()) {
4778ed55a54SJohn McCall     assert(ArraySizeMultiplier == 1);
478f2f38701SChris Lattner 
4798ed55a54SJohn McCall     Size = NumElements;
480f2f38701SChris Lattner 
4818ed55a54SJohn McCall     // If we need a cookie, add its size in with an overflow check.
4828ed55a54SJohn McCall     // This is maybe a little paranoid.
4838ed55a54SJohn McCall     if (!CookieSize.isZero()) {
48405fc5be3SDouglas Gregor       SizeWithoutCookie = Size;
485f2f38701SChris Lattner 
4868ed55a54SJohn McCall       llvm::Value *CookieSizeV
4878ed55a54SJohn McCall         = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
4888ed55a54SJohn McCall 
4898ed55a54SJohn McCall       const llvm::Type *Types[] = { SizeTy };
4908ed55a54SJohn McCall       llvm::Value *UAddF
4918ed55a54SJohn McCall         = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1);
4928ed55a54SJohn McCall       llvm::Value *AddRes
4938ed55a54SJohn McCall         = CGF.Builder.CreateCall2(UAddF, Size, CookieSizeV);
4948ed55a54SJohn McCall 
4958ed55a54SJohn McCall       Size = CGF.Builder.CreateExtractValue(AddRes, 0);
4968ed55a54SJohn McCall       llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1);
4978ed55a54SJohn McCall       Size = CGF.Builder.CreateSelect(DidOverflow,
4988ed55a54SJohn McCall                                       llvm::ConstantInt::get(SizeTy, -1),
4998ed55a54SJohn McCall                                       Size);
5008ed55a54SJohn McCall     }
5018ed55a54SJohn McCall 
5028ed55a54SJohn McCall   // Otherwise use the int.umul.with.overflow intrinsic.
5038ed55a54SJohn McCall   } else {
5048ed55a54SJohn McCall     llvm::Value *OutermostElementSize
5058ed55a54SJohn McCall       = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
5068ed55a54SJohn McCall 
5078ed55a54SJohn McCall     llvm::Value *NumOutermostElements = NumElements;
5088ed55a54SJohn McCall 
5098ed55a54SJohn McCall     // Scale NumElements by the array size multiplier.  This might
5108ed55a54SJohn McCall     // overflow, but only if the multiplication below also overflows,
5118ed55a54SJohn McCall     // in which case this multiplication isn't used.
5128ed55a54SJohn McCall     if (ArraySizeMultiplier != 1)
5138ed55a54SJohn McCall       NumElements = CGF.Builder.CreateMul(NumElements,
5148ed55a54SJohn McCall                          llvm::ConstantInt::get(SizeTy, ArraySizeMultiplier));
5158ed55a54SJohn McCall 
5168ed55a54SJohn McCall     // The requested size of the outermost array is non-constant.
5178ed55a54SJohn McCall     // Multiply that by the static size of the elements of that array;
5188ed55a54SJohn McCall     // on unsigned overflow, set the size to -1 to trigger an
5198ed55a54SJohn McCall     // exception from the allocation routine.  This is sufficient to
5208ed55a54SJohn McCall     // prevent buffer overruns from the allocator returning a
5218ed55a54SJohn McCall     // seemingly valid pointer to insufficient space.  This idea comes
5228ed55a54SJohn McCall     // originally from MSVC, and GCC has an open bug requesting
5238ed55a54SJohn McCall     // similar behavior:
5248ed55a54SJohn McCall     //   http://gcc.gnu.org/bugzilla/show_bug.cgi?id=19351
5258ed55a54SJohn McCall     //
5268ed55a54SJohn McCall     // This will not be sufficient for C++0x, which requires a
5278ed55a54SJohn McCall     // specific exception class (std::bad_array_new_length).
5288ed55a54SJohn McCall     // That will require ABI support that has not yet been specified.
5298ed55a54SJohn McCall     const llvm::Type *Types[] = { SizeTy };
5308ed55a54SJohn McCall     llvm::Value *UMulF
5318ed55a54SJohn McCall       = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, Types, 1);
5328ed55a54SJohn McCall     llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumOutermostElements,
5338ed55a54SJohn McCall                                                   OutermostElementSize);
5348ed55a54SJohn McCall 
5358ed55a54SJohn McCall     // The overflow bit.
5368ed55a54SJohn McCall     llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1);
5378ed55a54SJohn McCall 
5388ed55a54SJohn McCall     // The result of the multiplication.
5398ed55a54SJohn McCall     Size = CGF.Builder.CreateExtractValue(MulRes, 0);
5408ed55a54SJohn McCall 
5418ed55a54SJohn McCall     // If we have a cookie, we need to add that size in, too.
5428ed55a54SJohn McCall     if (!CookieSize.isZero()) {
5438ed55a54SJohn McCall       SizeWithoutCookie = Size;
5448ed55a54SJohn McCall 
5458ed55a54SJohn McCall       llvm::Value *CookieSizeV
5468ed55a54SJohn McCall         = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
5478ed55a54SJohn McCall       llvm::Value *UAddF
5488ed55a54SJohn McCall         = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1);
5498ed55a54SJohn McCall       llvm::Value *AddRes
5508ed55a54SJohn McCall         = CGF.Builder.CreateCall2(UAddF, SizeWithoutCookie, CookieSizeV);
5518ed55a54SJohn McCall 
5528ed55a54SJohn McCall       Size = CGF.Builder.CreateExtractValue(AddRes, 0);
5538ed55a54SJohn McCall 
5548ed55a54SJohn McCall       llvm::Value *AddDidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1);
5558ed55a54SJohn McCall       DidOverflow = CGF.Builder.CreateAnd(DidOverflow, AddDidOverflow);
5568ed55a54SJohn McCall     }
5578ed55a54SJohn McCall 
5588ed55a54SJohn McCall     Size = CGF.Builder.CreateSelect(DidOverflow,
5598ed55a54SJohn McCall                                     llvm::ConstantInt::get(SizeTy, -1),
5608ed55a54SJohn McCall                                     Size);
5618ed55a54SJohn McCall   }
5628ed55a54SJohn McCall 
5638ed55a54SJohn McCall   if (CookieSize.isZero())
5648ed55a54SJohn McCall     SizeWithoutCookie = Size;
5658ed55a54SJohn McCall   else
5668ed55a54SJohn McCall     assert(SizeWithoutCookie && "didn't set SizeWithoutCookie?");
56759486a2dSAnders Carlsson 
56832ac583dSChris Lattner   return Size;
56959486a2dSAnders Carlsson }
57059486a2dSAnders Carlsson 
571d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E,
572d5202e09SFariborz Jahanian                                     llvm::Value *NewPtr) {
573d5202e09SFariborz Jahanian 
574d5202e09SFariborz Jahanian   assert(E->getNumConstructorArgs() == 1 &&
575d5202e09SFariborz Jahanian          "Can only have one argument to initializer of POD type.");
576d5202e09SFariborz Jahanian 
577d5202e09SFariborz Jahanian   const Expr *Init = E->getConstructorArg(0);
578d5202e09SFariborz Jahanian   QualType AllocType = E->getAllocatedType();
579d5202e09SFariborz Jahanian 
5800381634aSDaniel Dunbar   unsigned Alignment =
5810381634aSDaniel Dunbar     CGF.getContext().getTypeAlignInChars(AllocType).getQuantity();
582d5202e09SFariborz Jahanian   if (!CGF.hasAggregateLLVMType(AllocType))
583d5202e09SFariborz Jahanian     CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr,
5840381634aSDaniel Dunbar                           AllocType.isVolatileQualified(), Alignment,
5850381634aSDaniel Dunbar                           AllocType);
586d5202e09SFariborz Jahanian   else if (AllocType->isAnyComplexType())
587d5202e09SFariborz Jahanian     CGF.EmitComplexExprIntoAddr(Init, NewPtr,
588d5202e09SFariborz Jahanian                                 AllocType.isVolatileQualified());
5897a626f63SJohn McCall   else {
5907a626f63SJohn McCall     AggValueSlot Slot
5917a626f63SJohn McCall       = AggValueSlot::forAddr(NewPtr, AllocType.isVolatileQualified(), true);
5927a626f63SJohn McCall     CGF.EmitAggExpr(Init, Slot);
5937a626f63SJohn McCall   }
594d5202e09SFariborz Jahanian }
595d5202e09SFariborz Jahanian 
596d5202e09SFariborz Jahanian void
597d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E,
598d5202e09SFariborz Jahanian                                          llvm::Value *NewPtr,
599d5202e09SFariborz Jahanian                                          llvm::Value *NumElements) {
600b66b08efSFariborz Jahanian   // We have a POD type.
601b66b08efSFariborz Jahanian   if (E->getNumConstructorArgs() == 0)
602b66b08efSFariborz Jahanian     return;
603b66b08efSFariborz Jahanian 
604d5202e09SFariborz Jahanian   const llvm::Type *SizeTy = ConvertType(getContext().getSizeType());
605d5202e09SFariborz Jahanian 
606d5202e09SFariborz Jahanian   // Create a temporary for the loop index and initialize it with 0.
607d5202e09SFariborz Jahanian   llvm::Value *IndexPtr = CreateTempAlloca(SizeTy, "loop.index");
608d5202e09SFariborz Jahanian   llvm::Value *Zero = llvm::Constant::getNullValue(SizeTy);
609d5202e09SFariborz Jahanian   Builder.CreateStore(Zero, IndexPtr);
610d5202e09SFariborz Jahanian 
611d5202e09SFariborz Jahanian   // Start the loop with a block that tests the condition.
612d5202e09SFariborz Jahanian   llvm::BasicBlock *CondBlock = createBasicBlock("for.cond");
613d5202e09SFariborz Jahanian   llvm::BasicBlock *AfterFor = createBasicBlock("for.end");
614d5202e09SFariborz Jahanian 
615d5202e09SFariborz Jahanian   EmitBlock(CondBlock);
616d5202e09SFariborz Jahanian 
617d5202e09SFariborz Jahanian   llvm::BasicBlock *ForBody = createBasicBlock("for.body");
618d5202e09SFariborz Jahanian 
619d5202e09SFariborz Jahanian   // Generate: if (loop-index < number-of-elements fall to the loop body,
620d5202e09SFariborz Jahanian   // otherwise, go to the block after the for-loop.
621d5202e09SFariborz Jahanian   llvm::Value *Counter = Builder.CreateLoad(IndexPtr);
622d5202e09SFariborz Jahanian   llvm::Value *IsLess = Builder.CreateICmpULT(Counter, NumElements, "isless");
623d5202e09SFariborz Jahanian   // If the condition is true, execute the body.
624d5202e09SFariborz Jahanian   Builder.CreateCondBr(IsLess, ForBody, AfterFor);
625d5202e09SFariborz Jahanian 
626d5202e09SFariborz Jahanian   EmitBlock(ForBody);
627d5202e09SFariborz Jahanian 
628d5202e09SFariborz Jahanian   llvm::BasicBlock *ContinueBlock = createBasicBlock("for.inc");
629d5202e09SFariborz Jahanian   // Inside the loop body, emit the constructor call on the array element.
630d5202e09SFariborz Jahanian   Counter = Builder.CreateLoad(IndexPtr);
631d5202e09SFariborz Jahanian   llvm::Value *Address = Builder.CreateInBoundsGEP(NewPtr, Counter,
632d5202e09SFariborz Jahanian                                                    "arrayidx");
633d5202e09SFariborz Jahanian   StoreAnyExprIntoOneUnit(*this, E, Address);
634d5202e09SFariborz Jahanian 
635d5202e09SFariborz Jahanian   EmitBlock(ContinueBlock);
636d5202e09SFariborz Jahanian 
637d5202e09SFariborz Jahanian   // Emit the increment of the loop counter.
638d5202e09SFariborz Jahanian   llvm::Value *NextVal = llvm::ConstantInt::get(SizeTy, 1);
639d5202e09SFariborz Jahanian   Counter = Builder.CreateLoad(IndexPtr);
640d5202e09SFariborz Jahanian   NextVal = Builder.CreateAdd(Counter, NextVal, "inc");
641d5202e09SFariborz Jahanian   Builder.CreateStore(NextVal, IndexPtr);
642d5202e09SFariborz Jahanian 
643d5202e09SFariborz Jahanian   // Finally, branch back up to the condition for the next iteration.
644d5202e09SFariborz Jahanian   EmitBranch(CondBlock);
645d5202e09SFariborz Jahanian 
646d5202e09SFariborz Jahanian   // Emit the fall-through block.
647d5202e09SFariborz Jahanian   EmitBlock(AfterFor, true);
648d5202e09SFariborz Jahanian }
649d5202e09SFariborz Jahanian 
65005fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T,
65105fc5be3SDouglas Gregor                            llvm::Value *NewPtr, llvm::Value *Size) {
65205fc5be3SDouglas Gregor   llvm::LLVMContext &VMContext = CGF.CGM.getLLVMContext();
65305fc5be3SDouglas Gregor   const llvm::Type *BP = llvm::Type::getInt8PtrTy(VMContext);
65405fc5be3SDouglas Gregor   if (NewPtr->getType() != BP)
65505fc5be3SDouglas Gregor     NewPtr = CGF.Builder.CreateBitCast(NewPtr, BP, "tmp");
65605fc5be3SDouglas Gregor 
65705fc5be3SDouglas Gregor   CGF.Builder.CreateCall5(CGF.CGM.getMemSetFn(BP, CGF.IntPtrTy), NewPtr,
65805fc5be3SDouglas Gregor                 llvm::Constant::getNullValue(llvm::Type::getInt8Ty(VMContext)),
65905fc5be3SDouglas Gregor                           Size,
66005fc5be3SDouglas Gregor                     llvm::ConstantInt::get(CGF.Int32Ty,
66105fc5be3SDouglas Gregor                                            CGF.getContext().getTypeAlign(T)/8),
66205fc5be3SDouglas Gregor                           llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext),
66305fc5be3SDouglas Gregor                                                  0));
66405fc5be3SDouglas Gregor }
66505fc5be3SDouglas Gregor 
66659486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
66759486a2dSAnders Carlsson                                llvm::Value *NewPtr,
66805fc5be3SDouglas Gregor                                llvm::Value *NumElements,
66905fc5be3SDouglas Gregor                                llvm::Value *AllocSizeWithoutCookie) {
6703a202f60SAnders Carlsson   if (E->isArray()) {
671d040e6b2SAnders Carlsson     if (CXXConstructorDecl *Ctor = E->getConstructor()) {
67205fc5be3SDouglas Gregor       bool RequiresZeroInitialization = false;
67305fc5be3SDouglas Gregor       if (Ctor->getParent()->hasTrivialConstructor()) {
67405fc5be3SDouglas Gregor         // If new expression did not specify value-initialization, then there
67505fc5be3SDouglas Gregor         // is no initialization.
67605fc5be3SDouglas Gregor         if (!E->hasInitializer() || Ctor->getParent()->isEmpty())
67705fc5be3SDouglas Gregor           return;
67805fc5be3SDouglas Gregor 
679614dbdcdSJohn McCall         if (CGF.CGM.getTypes().isZeroInitializable(E->getAllocatedType())) {
68005fc5be3SDouglas Gregor           // Optimization: since zero initialization will just set the memory
68105fc5be3SDouglas Gregor           // to all zeroes, generate a single memset to do it in one shot.
68205fc5be3SDouglas Gregor           EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr,
68305fc5be3SDouglas Gregor                          AllocSizeWithoutCookie);
6843a202f60SAnders Carlsson           return;
6853a202f60SAnders Carlsson         }
68605fc5be3SDouglas Gregor 
68705fc5be3SDouglas Gregor         RequiresZeroInitialization = true;
68805fc5be3SDouglas Gregor       }
68905fc5be3SDouglas Gregor 
69005fc5be3SDouglas Gregor       CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr,
69105fc5be3SDouglas Gregor                                      E->constructor_arg_begin(),
69205fc5be3SDouglas Gregor                                      E->constructor_arg_end(),
69305fc5be3SDouglas Gregor                                      RequiresZeroInitialization);
69405fc5be3SDouglas Gregor       return;
69505fc5be3SDouglas Gregor     } else if (E->getNumConstructorArgs() == 1 &&
69605fc5be3SDouglas Gregor                isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) {
69705fc5be3SDouglas Gregor       // Optimization: since zero initialization will just set the memory
69805fc5be3SDouglas Gregor       // to all zeroes, generate a single memset to do it in one shot.
69905fc5be3SDouglas Gregor       EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr,
70005fc5be3SDouglas Gregor                      AllocSizeWithoutCookie);
70105fc5be3SDouglas Gregor       return;
70205fc5be3SDouglas Gregor     } else {
703d5202e09SFariborz Jahanian       CGF.EmitNewArrayInitializer(E, NewPtr, NumElements);
704d5202e09SFariborz Jahanian       return;
705d040e6b2SAnders Carlsson     }
706d5202e09SFariborz Jahanian   }
70759486a2dSAnders Carlsson 
70859486a2dSAnders Carlsson   if (CXXConstructorDecl *Ctor = E->getConstructor()) {
709747eb784SDouglas Gregor     // Per C++ [expr.new]p15, if we have an initializer, then we're performing
710747eb784SDouglas Gregor     // direct initialization. C++ [dcl.init]p5 requires that we
711747eb784SDouglas Gregor     // zero-initialize storage if there are no user-declared constructors.
712747eb784SDouglas Gregor     if (E->hasInitializer() &&
713747eb784SDouglas Gregor         !Ctor->getParent()->hasUserDeclaredConstructor() &&
714747eb784SDouglas Gregor         !Ctor->getParent()->isEmpty())
715747eb784SDouglas Gregor       CGF.EmitNullInitialization(NewPtr, E->getAllocatedType());
716747eb784SDouglas Gregor 
717e11f9ce9SAnders Carlsson     CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false,
718e11f9ce9SAnders Carlsson                                NewPtr, E->constructor_arg_begin(),
71959486a2dSAnders Carlsson                                E->constructor_arg_end());
72059486a2dSAnders Carlsson 
72159486a2dSAnders Carlsson     return;
72259486a2dSAnders Carlsson   }
723b66b08efSFariborz Jahanian   // We have a POD type.
724b66b08efSFariborz Jahanian   if (E->getNumConstructorArgs() == 0)
725b66b08efSFariborz Jahanian     return;
72659486a2dSAnders Carlsson 
727d5202e09SFariborz Jahanian   StoreAnyExprIntoOneUnit(CGF, E, NewPtr);
72859486a2dSAnders Carlsson }
72959486a2dSAnders Carlsson 
730fb5e5841SBenjamin Kramer namespace {
7317f9c92a9SJohn McCall /// A utility class for saving an rvalue.
7327f9c92a9SJohn McCall class SavedRValue {
7337f9c92a9SJohn McCall public:
7347f9c92a9SJohn McCall   enum Kind { ScalarLiteral, ScalarAddress,
7357f9c92a9SJohn McCall               AggregateLiteral, AggregateAddress,
7367f9c92a9SJohn McCall               Complex };
7377f9c92a9SJohn McCall 
7387f9c92a9SJohn McCall private:
7397f9c92a9SJohn McCall   llvm::Value *Value;
7407f9c92a9SJohn McCall   Kind K;
7417f9c92a9SJohn McCall 
7427f9c92a9SJohn McCall   SavedRValue(llvm::Value *V, Kind K) : Value(V), K(K) {}
7437f9c92a9SJohn McCall 
7447f9c92a9SJohn McCall public:
7457f9c92a9SJohn McCall   SavedRValue() {}
7467f9c92a9SJohn McCall 
7477f9c92a9SJohn McCall   static SavedRValue forScalarLiteral(llvm::Value *V) {
7487f9c92a9SJohn McCall     return SavedRValue(V, ScalarLiteral);
7497f9c92a9SJohn McCall   }
7507f9c92a9SJohn McCall 
7517f9c92a9SJohn McCall   static SavedRValue forScalarAddress(llvm::Value *Addr) {
7527f9c92a9SJohn McCall     return SavedRValue(Addr, ScalarAddress);
7537f9c92a9SJohn McCall   }
7547f9c92a9SJohn McCall 
7557f9c92a9SJohn McCall   static SavedRValue forAggregateLiteral(llvm::Value *V) {
7567f9c92a9SJohn McCall     return SavedRValue(V, AggregateLiteral);
7577f9c92a9SJohn McCall   }
7587f9c92a9SJohn McCall 
7597f9c92a9SJohn McCall   static SavedRValue forAggregateAddress(llvm::Value *Addr) {
7607f9c92a9SJohn McCall     return SavedRValue(Addr, AggregateAddress);
7617f9c92a9SJohn McCall   }
7627f9c92a9SJohn McCall 
7637f9c92a9SJohn McCall   static SavedRValue forComplexAddress(llvm::Value *Addr) {
7647f9c92a9SJohn McCall     return SavedRValue(Addr, Complex);
7657f9c92a9SJohn McCall   }
7667f9c92a9SJohn McCall 
7677f9c92a9SJohn McCall   Kind getKind() const { return K; }
7687f9c92a9SJohn McCall   llvm::Value *getValue() const { return Value; }
7697f9c92a9SJohn McCall };
770fb5e5841SBenjamin Kramer } // end anonymous namespace
7717f9c92a9SJohn McCall 
7727f9c92a9SJohn McCall /// Given an r-value, perform the code necessary to make sure that a
7737f9c92a9SJohn McCall /// future RestoreRValue will be able to load the value without
7747f9c92a9SJohn McCall /// domination concerns.
7757f9c92a9SJohn McCall static SavedRValue SaveRValue(CodeGenFunction &CGF, RValue RV) {
7767f9c92a9SJohn McCall   if (RV.isScalar()) {
7777f9c92a9SJohn McCall     llvm::Value *V = RV.getScalarVal();
7787f9c92a9SJohn McCall 
7797f9c92a9SJohn McCall     // These automatically dominate and don't need to be saved.
7807f9c92a9SJohn McCall     if (isa<llvm::Constant>(V) || isa<llvm::AllocaInst>(V))
7817f9c92a9SJohn McCall       return SavedRValue::forScalarLiteral(V);
7827f9c92a9SJohn McCall 
7837f9c92a9SJohn McCall     // Everything else needs an alloca.
7847f9c92a9SJohn McCall     llvm::Value *Addr = CGF.CreateTempAlloca(V->getType(), "saved-rvalue");
7857f9c92a9SJohn McCall     CGF.Builder.CreateStore(V, Addr);
7867f9c92a9SJohn McCall     return SavedRValue::forScalarAddress(Addr);
7877f9c92a9SJohn McCall   }
7887f9c92a9SJohn McCall 
7897f9c92a9SJohn McCall   if (RV.isComplex()) {
7907f9c92a9SJohn McCall     CodeGenFunction::ComplexPairTy V = RV.getComplexVal();
7917f9c92a9SJohn McCall     const llvm::Type *ComplexTy =
7927f9c92a9SJohn McCall       llvm::StructType::get(CGF.getLLVMContext(),
7937f9c92a9SJohn McCall                             V.first->getType(), V.second->getType(),
7947f9c92a9SJohn McCall                             (void*) 0);
7957f9c92a9SJohn McCall     llvm::Value *Addr = CGF.CreateTempAlloca(ComplexTy, "saved-complex");
7967f9c92a9SJohn McCall     CGF.StoreComplexToAddr(V, Addr, /*volatile*/ false);
7977f9c92a9SJohn McCall     return SavedRValue::forComplexAddress(Addr);
7987f9c92a9SJohn McCall   }
7997f9c92a9SJohn McCall 
8007f9c92a9SJohn McCall   assert(RV.isAggregate());
8017f9c92a9SJohn McCall   llvm::Value *V = RV.getAggregateAddr(); // TODO: volatile?
8027f9c92a9SJohn McCall   if (isa<llvm::Constant>(V) || isa<llvm::AllocaInst>(V))
8037f9c92a9SJohn McCall     return SavedRValue::forAggregateLiteral(V);
8047f9c92a9SJohn McCall 
8057f9c92a9SJohn McCall   llvm::Value *Addr = CGF.CreateTempAlloca(V->getType(), "saved-rvalue");
8067f9c92a9SJohn McCall   CGF.Builder.CreateStore(V, Addr);
8077f9c92a9SJohn McCall   return SavedRValue::forAggregateAddress(Addr);
8087f9c92a9SJohn McCall }
8097f9c92a9SJohn McCall 
8107f9c92a9SJohn McCall /// Given a saved r-value produced by SaveRValue, perform the code
8117f9c92a9SJohn McCall /// necessary to restore it to usability at the current insertion
8127f9c92a9SJohn McCall /// point.
8137f9c92a9SJohn McCall static RValue RestoreRValue(CodeGenFunction &CGF, SavedRValue RV) {
8147f9c92a9SJohn McCall   switch (RV.getKind()) {
8157f9c92a9SJohn McCall   case SavedRValue::ScalarLiteral:
8167f9c92a9SJohn McCall     return RValue::get(RV.getValue());
8177f9c92a9SJohn McCall   case SavedRValue::ScalarAddress:
8187f9c92a9SJohn McCall     return RValue::get(CGF.Builder.CreateLoad(RV.getValue()));
8197f9c92a9SJohn McCall   case SavedRValue::AggregateLiteral:
8207f9c92a9SJohn McCall     return RValue::getAggregate(RV.getValue());
8217f9c92a9SJohn McCall   case SavedRValue::AggregateAddress:
8227f9c92a9SJohn McCall     return RValue::getAggregate(CGF.Builder.CreateLoad(RV.getValue()));
8237f9c92a9SJohn McCall   case SavedRValue::Complex:
8247f9c92a9SJohn McCall     return RValue::getComplex(CGF.LoadComplexFromAddr(RV.getValue(), false));
8257f9c92a9SJohn McCall   }
8267f9c92a9SJohn McCall 
8277f9c92a9SJohn McCall   llvm_unreachable("bad saved r-value kind");
8287f9c92a9SJohn McCall   return RValue();
8297f9c92a9SJohn McCall }
8307f9c92a9SJohn McCall 
831824c2f53SJohn McCall namespace {
832824c2f53SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
833824c2f53SJohn McCall   /// abnormal exit from a new expression.
834824c2f53SJohn McCall   class CallDeleteDuringNew : public EHScopeStack::Cleanup {
835824c2f53SJohn McCall     size_t NumPlacementArgs;
836824c2f53SJohn McCall     const FunctionDecl *OperatorDelete;
837824c2f53SJohn McCall     llvm::Value *Ptr;
838824c2f53SJohn McCall     llvm::Value *AllocSize;
839824c2f53SJohn McCall 
840824c2f53SJohn McCall     RValue *getPlacementArgs() { return reinterpret_cast<RValue*>(this+1); }
841824c2f53SJohn McCall 
842824c2f53SJohn McCall   public:
843824c2f53SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
844824c2f53SJohn McCall       return NumPlacementArgs * sizeof(RValue);
845824c2f53SJohn McCall     }
846824c2f53SJohn McCall 
847824c2f53SJohn McCall     CallDeleteDuringNew(size_t NumPlacementArgs,
848824c2f53SJohn McCall                         const FunctionDecl *OperatorDelete,
849824c2f53SJohn McCall                         llvm::Value *Ptr,
850824c2f53SJohn McCall                         llvm::Value *AllocSize)
851824c2f53SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
852824c2f53SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
853824c2f53SJohn McCall 
854824c2f53SJohn McCall     void setPlacementArg(unsigned I, RValue Arg) {
855824c2f53SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
856824c2f53SJohn McCall       getPlacementArgs()[I] = Arg;
857824c2f53SJohn McCall     }
858824c2f53SJohn McCall 
859824c2f53SJohn McCall     void Emit(CodeGenFunction &CGF, bool IsForEH) {
860824c2f53SJohn McCall       const FunctionProtoType *FPT
861824c2f53SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
862824c2f53SJohn McCall       assert(FPT->getNumArgs() == NumPlacementArgs + 1 ||
863d441b1e6SJohn McCall              (FPT->getNumArgs() == 2 && NumPlacementArgs == 0));
864824c2f53SJohn McCall 
865824c2f53SJohn McCall       CallArgList DeleteArgs;
866824c2f53SJohn McCall 
867824c2f53SJohn McCall       // The first argument is always a void*.
868824c2f53SJohn McCall       FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin();
869824c2f53SJohn McCall       DeleteArgs.push_back(std::make_pair(RValue::get(Ptr), *AI++));
870824c2f53SJohn McCall 
871824c2f53SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
872824c2f53SJohn McCall       if (FPT->getNumArgs() == NumPlacementArgs + 2)
873824c2f53SJohn McCall         DeleteArgs.push_back(std::make_pair(RValue::get(AllocSize), *AI++));
874824c2f53SJohn McCall 
875824c2f53SJohn McCall       // Pass the rest of the arguments, which must match exactly.
876824c2f53SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I)
877824c2f53SJohn McCall         DeleteArgs.push_back(std::make_pair(getPlacementArgs()[I], *AI++));
878824c2f53SJohn McCall 
879824c2f53SJohn McCall       // Call 'operator delete'.
880824c2f53SJohn McCall       CGF.EmitCall(CGF.CGM.getTypes().getFunctionInfo(DeleteArgs, FPT),
881824c2f53SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
882824c2f53SJohn McCall                    ReturnValueSlot(), DeleteArgs, OperatorDelete);
883824c2f53SJohn McCall     }
884824c2f53SJohn McCall   };
8857f9c92a9SJohn McCall 
8867f9c92a9SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
8877f9c92a9SJohn McCall   /// abnormal exit from a new expression when the new expression is
8887f9c92a9SJohn McCall   /// conditional.
8897f9c92a9SJohn McCall   class CallDeleteDuringConditionalNew : public EHScopeStack::Cleanup {
8907f9c92a9SJohn McCall     size_t NumPlacementArgs;
8917f9c92a9SJohn McCall     const FunctionDecl *OperatorDelete;
8927f9c92a9SJohn McCall     SavedRValue Ptr;
8937f9c92a9SJohn McCall     SavedRValue AllocSize;
8947f9c92a9SJohn McCall 
8957f9c92a9SJohn McCall     SavedRValue *getPlacementArgs() {
8967f9c92a9SJohn McCall       return reinterpret_cast<SavedRValue*>(this+1);
8977f9c92a9SJohn McCall     }
8987f9c92a9SJohn McCall 
8997f9c92a9SJohn McCall   public:
9007f9c92a9SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
9017f9c92a9SJohn McCall       return NumPlacementArgs * sizeof(SavedRValue);
9027f9c92a9SJohn McCall     }
9037f9c92a9SJohn McCall 
9047f9c92a9SJohn McCall     CallDeleteDuringConditionalNew(size_t NumPlacementArgs,
9057f9c92a9SJohn McCall                                    const FunctionDecl *OperatorDelete,
9067f9c92a9SJohn McCall                                    SavedRValue Ptr,
9077f9c92a9SJohn McCall                                    SavedRValue AllocSize)
9087f9c92a9SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
9097f9c92a9SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
9107f9c92a9SJohn McCall 
9117f9c92a9SJohn McCall     void setPlacementArg(unsigned I, SavedRValue Arg) {
9127f9c92a9SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
9137f9c92a9SJohn McCall       getPlacementArgs()[I] = Arg;
9147f9c92a9SJohn McCall     }
9157f9c92a9SJohn McCall 
9167f9c92a9SJohn McCall     void Emit(CodeGenFunction &CGF, bool IsForEH) {
9177f9c92a9SJohn McCall       const FunctionProtoType *FPT
9187f9c92a9SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
9197f9c92a9SJohn McCall       assert(FPT->getNumArgs() == NumPlacementArgs + 1 ||
9207f9c92a9SJohn McCall              (FPT->getNumArgs() == 2 && NumPlacementArgs == 0));
9217f9c92a9SJohn McCall 
9227f9c92a9SJohn McCall       CallArgList DeleteArgs;
9237f9c92a9SJohn McCall 
9247f9c92a9SJohn McCall       // The first argument is always a void*.
9257f9c92a9SJohn McCall       FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin();
9267f9c92a9SJohn McCall       DeleteArgs.push_back(std::make_pair(RestoreRValue(CGF, Ptr), *AI++));
9277f9c92a9SJohn McCall 
9287f9c92a9SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
9297f9c92a9SJohn McCall       if (FPT->getNumArgs() == NumPlacementArgs + 2) {
9307f9c92a9SJohn McCall         RValue RV = RestoreRValue(CGF, AllocSize);
9317f9c92a9SJohn McCall         DeleteArgs.push_back(std::make_pair(RV, *AI++));
9327f9c92a9SJohn McCall       }
9337f9c92a9SJohn McCall 
9347f9c92a9SJohn McCall       // Pass the rest of the arguments, which must match exactly.
9357f9c92a9SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I) {
9367f9c92a9SJohn McCall         RValue RV = RestoreRValue(CGF, getPlacementArgs()[I]);
9377f9c92a9SJohn McCall         DeleteArgs.push_back(std::make_pair(RV, *AI++));
9387f9c92a9SJohn McCall       }
9397f9c92a9SJohn McCall 
9407f9c92a9SJohn McCall       // Call 'operator delete'.
9417f9c92a9SJohn McCall       CGF.EmitCall(CGF.CGM.getTypes().getFunctionInfo(DeleteArgs, FPT),
9427f9c92a9SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
9437f9c92a9SJohn McCall                    ReturnValueSlot(), DeleteArgs, OperatorDelete);
9447f9c92a9SJohn McCall     }
9457f9c92a9SJohn McCall   };
9467f9c92a9SJohn McCall }
9477f9c92a9SJohn McCall 
9487f9c92a9SJohn McCall /// Enter a cleanup to call 'operator delete' if the initializer in a
9497f9c92a9SJohn McCall /// new-expression throws.
9507f9c92a9SJohn McCall static void EnterNewDeleteCleanup(CodeGenFunction &CGF,
9517f9c92a9SJohn McCall                                   const CXXNewExpr *E,
9527f9c92a9SJohn McCall                                   llvm::Value *NewPtr,
9537f9c92a9SJohn McCall                                   llvm::Value *AllocSize,
9547f9c92a9SJohn McCall                                   const CallArgList &NewArgs) {
9557f9c92a9SJohn McCall   // If we're not inside a conditional branch, then the cleanup will
9567f9c92a9SJohn McCall   // dominate and we can do the easier (and more efficient) thing.
9577f9c92a9SJohn McCall   if (!CGF.isInConditionalBranch()) {
9587f9c92a9SJohn McCall     CallDeleteDuringNew *Cleanup = CGF.EHStack
9597f9c92a9SJohn McCall       .pushCleanupWithExtra<CallDeleteDuringNew>(EHCleanup,
9607f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
9617f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
9627f9c92a9SJohn McCall                                                  NewPtr, AllocSize);
9637f9c92a9SJohn McCall     for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
9647f9c92a9SJohn McCall       Cleanup->setPlacementArg(I, NewArgs[I+1].first);
9657f9c92a9SJohn McCall 
9667f9c92a9SJohn McCall     return;
9677f9c92a9SJohn McCall   }
9687f9c92a9SJohn McCall 
9697f9c92a9SJohn McCall   // Otherwise, we need to save all this stuff.
9707f9c92a9SJohn McCall   SavedRValue SavedNewPtr = SaveRValue(CGF, RValue::get(NewPtr));
9717f9c92a9SJohn McCall   SavedRValue SavedAllocSize = SaveRValue(CGF, RValue::get(AllocSize));
9727f9c92a9SJohn McCall 
9737f9c92a9SJohn McCall   CallDeleteDuringConditionalNew *Cleanup = CGF.EHStack
9747f9c92a9SJohn McCall     .pushCleanupWithExtra<CallDeleteDuringConditionalNew>(InactiveEHCleanup,
9757f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
9767f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
9777f9c92a9SJohn McCall                                                  SavedNewPtr,
9787f9c92a9SJohn McCall                                                  SavedAllocSize);
9797f9c92a9SJohn McCall   for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
9807f9c92a9SJohn McCall     Cleanup->setPlacementArg(I, SaveRValue(CGF, NewArgs[I+1].first));
9817f9c92a9SJohn McCall 
9827f9c92a9SJohn McCall   CGF.ActivateCleanupBlock(CGF.EHStack.stable_begin());
983824c2f53SJohn McCall }
984824c2f53SJohn McCall 
98559486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
98659486a2dSAnders Carlsson   QualType AllocType = E->getAllocatedType();
9878ed55a54SJohn McCall   if (AllocType->isArrayType())
9888ed55a54SJohn McCall     while (const ArrayType *AType = getContext().getAsArrayType(AllocType))
9898ed55a54SJohn McCall       AllocType = AType->getElementType();
9908ed55a54SJohn McCall 
99159486a2dSAnders Carlsson   FunctionDecl *NewFD = E->getOperatorNew();
99259486a2dSAnders Carlsson   const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>();
99359486a2dSAnders Carlsson 
99459486a2dSAnders Carlsson   CallArgList NewArgs;
99559486a2dSAnders Carlsson 
99659486a2dSAnders Carlsson   // The allocation size is the first argument.
99759486a2dSAnders Carlsson   QualType SizeTy = getContext().getSizeType();
99859486a2dSAnders Carlsson 
99959486a2dSAnders Carlsson   llvm::Value *NumElements = 0;
100005fc5be3SDouglas Gregor   llvm::Value *AllocSizeWithoutCookie = 0;
100147b4629bSFariborz Jahanian   llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(),
100205fc5be3SDouglas Gregor                                                *this, E, NumElements,
100305fc5be3SDouglas Gregor                                                AllocSizeWithoutCookie);
100459486a2dSAnders Carlsson 
100559486a2dSAnders Carlsson   NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy));
100659486a2dSAnders Carlsson 
100759486a2dSAnders Carlsson   // Emit the rest of the arguments.
100859486a2dSAnders Carlsson   // FIXME: Ideally, this should just use EmitCallArgs.
100959486a2dSAnders Carlsson   CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin();
101059486a2dSAnders Carlsson 
101159486a2dSAnders Carlsson   // First, use the types from the function type.
101259486a2dSAnders Carlsson   // We start at 1 here because the first argument (the allocation size)
101359486a2dSAnders Carlsson   // has already been emitted.
101459486a2dSAnders Carlsson   for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) {
101559486a2dSAnders Carlsson     QualType ArgType = NewFTy->getArgType(i);
101659486a2dSAnders Carlsson 
101759486a2dSAnders Carlsson     assert(getContext().getCanonicalType(ArgType.getNonReferenceType()).
101859486a2dSAnders Carlsson            getTypePtr() ==
101959486a2dSAnders Carlsson            getContext().getCanonicalType(NewArg->getType()).getTypePtr() &&
102059486a2dSAnders Carlsson            "type mismatch in call argument!");
102159486a2dSAnders Carlsson 
102259486a2dSAnders Carlsson     NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
102359486a2dSAnders Carlsson                                      ArgType));
102459486a2dSAnders Carlsson 
102559486a2dSAnders Carlsson   }
102659486a2dSAnders Carlsson 
102759486a2dSAnders Carlsson   // Either we've emitted all the call args, or we have a call to a
102859486a2dSAnders Carlsson   // variadic function.
102959486a2dSAnders Carlsson   assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) &&
103059486a2dSAnders Carlsson          "Extra arguments in non-variadic function!");
103159486a2dSAnders Carlsson 
103259486a2dSAnders Carlsson   // If we still have any arguments, emit them using the type of the argument.
103359486a2dSAnders Carlsson   for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end();
103459486a2dSAnders Carlsson        NewArg != NewArgEnd; ++NewArg) {
103559486a2dSAnders Carlsson     QualType ArgType = NewArg->getType();
103659486a2dSAnders Carlsson     NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
103759486a2dSAnders Carlsson                                      ArgType));
103859486a2dSAnders Carlsson   }
103959486a2dSAnders Carlsson 
104059486a2dSAnders Carlsson   // Emit the call to new.
104159486a2dSAnders Carlsson   RValue RV =
1042ab26cfa5SJohn McCall     EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy),
104361a401caSAnders Carlsson              CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD);
104459486a2dSAnders Carlsson 
104559486a2dSAnders Carlsson   // If an allocation function is declared with an empty exception specification
104659486a2dSAnders Carlsson   // it returns null to indicate failure to allocate storage. [expr.new]p13.
104759486a2dSAnders Carlsson   // (We don't need to check for null when there's no new initializer and
104859486a2dSAnders Carlsson   // we're allocating a POD type).
104959486a2dSAnders Carlsson   bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() &&
105059486a2dSAnders Carlsson     !(AllocType->isPODType() && !E->hasInitializer());
105159486a2dSAnders Carlsson 
10528ed55a54SJohn McCall   llvm::BasicBlock *NullCheckSource = 0;
105359486a2dSAnders Carlsson   llvm::BasicBlock *NewNotNull = 0;
105459486a2dSAnders Carlsson   llvm::BasicBlock *NewEnd = 0;
105559486a2dSAnders Carlsson 
105659486a2dSAnders Carlsson   llvm::Value *NewPtr = RV.getScalarVal();
10578ed55a54SJohn McCall   unsigned AS = cast<llvm::PointerType>(NewPtr->getType())->getAddressSpace();
105859486a2dSAnders Carlsson 
105959486a2dSAnders Carlsson   if (NullCheckResult) {
10608ed55a54SJohn McCall     NullCheckSource = Builder.GetInsertBlock();
106159486a2dSAnders Carlsson     NewNotNull = createBasicBlock("new.notnull");
106259486a2dSAnders Carlsson     NewEnd = createBasicBlock("new.end");
106359486a2dSAnders Carlsson 
10648ed55a54SJohn McCall     llvm::Value *IsNull = Builder.CreateIsNull(NewPtr, "new.isnull");
10658ed55a54SJohn McCall     Builder.CreateCondBr(IsNull, NewEnd, NewNotNull);
106659486a2dSAnders Carlsson     EmitBlock(NewNotNull);
106759486a2dSAnders Carlsson   }
106859486a2dSAnders Carlsson 
10698ed55a54SJohn McCall   assert((AllocSize == AllocSizeWithoutCookie) ==
10708ed55a54SJohn McCall          CalculateCookiePadding(*this, E).isZero());
10718ed55a54SJohn McCall   if (AllocSize != AllocSizeWithoutCookie) {
10728ed55a54SJohn McCall     assert(E->isArray());
10738ed55a54SJohn McCall     NewPtr = CGM.getCXXABI().InitializeArrayCookie(CGF, NewPtr, NumElements,
10748ed55a54SJohn McCall                                                    AllocType);
107559486a2dSAnders Carlsson   }
107659486a2dSAnders Carlsson 
1077824c2f53SJohn McCall   // If there's an operator delete, enter a cleanup to call it if an
1078824c2f53SJohn McCall   // exception is thrown.
1079824c2f53SJohn McCall   EHScopeStack::stable_iterator CallOperatorDelete;
1080824c2f53SJohn McCall   if (E->getOperatorDelete()) {
10817f9c92a9SJohn McCall     EnterNewDeleteCleanup(*this, E, NewPtr, AllocSize, NewArgs);
1082824c2f53SJohn McCall     CallOperatorDelete = EHStack.stable_begin();
1083824c2f53SJohn McCall   }
1084824c2f53SJohn McCall 
1085040ad500SDouglas Gregor   const llvm::Type *ElementPtrTy
1086040ad500SDouglas Gregor     = ConvertTypeForMem(AllocType)->getPointerTo(AS);
10878ed55a54SJohn McCall   NewPtr = Builder.CreateBitCast(NewPtr, ElementPtrTy);
1088824c2f53SJohn McCall 
10898ed55a54SJohn McCall   if (E->isArray()) {
109005fc5be3SDouglas Gregor     EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie);
10918ed55a54SJohn McCall 
10928ed55a54SJohn McCall     // NewPtr is a pointer to the base element type.  If we're
10938ed55a54SJohn McCall     // allocating an array of arrays, we'll need to cast back to the
10948ed55a54SJohn McCall     // array pointer type.
1095040ad500SDouglas Gregor     const llvm::Type *ResultTy = ConvertTypeForMem(E->getType());
10968ed55a54SJohn McCall     if (NewPtr->getType() != ResultTy)
10978ed55a54SJohn McCall       NewPtr = Builder.CreateBitCast(NewPtr, ResultTy);
10988ed55a54SJohn McCall   } else {
109905fc5be3SDouglas Gregor     EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie);
110047b4629bSFariborz Jahanian   }
110159486a2dSAnders Carlsson 
1102824c2f53SJohn McCall   // Deactivate the 'operator delete' cleanup if we finished
1103824c2f53SJohn McCall   // initialization.
1104824c2f53SJohn McCall   if (CallOperatorDelete.isValid())
1105824c2f53SJohn McCall     DeactivateCleanupBlock(CallOperatorDelete);
1106824c2f53SJohn McCall 
110759486a2dSAnders Carlsson   if (NullCheckResult) {
110859486a2dSAnders Carlsson     Builder.CreateBr(NewEnd);
11098ed55a54SJohn McCall     llvm::BasicBlock *NotNullSource = Builder.GetInsertBlock();
111059486a2dSAnders Carlsson     EmitBlock(NewEnd);
111159486a2dSAnders Carlsson 
111259486a2dSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType());
111359486a2dSAnders Carlsson     PHI->reserveOperandSpace(2);
11148ed55a54SJohn McCall     PHI->addIncoming(NewPtr, NotNullSource);
11158ed55a54SJohn McCall     PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()),
11168ed55a54SJohn McCall                      NullCheckSource);
111759486a2dSAnders Carlsson 
111859486a2dSAnders Carlsson     NewPtr = PHI;
111959486a2dSAnders Carlsson   }
112059486a2dSAnders Carlsson 
112159486a2dSAnders Carlsson   return NewPtr;
112259486a2dSAnders Carlsson }
112359486a2dSAnders Carlsson 
112459486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
112559486a2dSAnders Carlsson                                      llvm::Value *Ptr,
112659486a2dSAnders Carlsson                                      QualType DeleteTy) {
11278ed55a54SJohn McCall   assert(DeleteFD->getOverloadedOperator() == OO_Delete);
11288ed55a54SJohn McCall 
112959486a2dSAnders Carlsson   const FunctionProtoType *DeleteFTy =
113059486a2dSAnders Carlsson     DeleteFD->getType()->getAs<FunctionProtoType>();
113159486a2dSAnders Carlsson 
113259486a2dSAnders Carlsson   CallArgList DeleteArgs;
113359486a2dSAnders Carlsson 
113421122cf6SAnders Carlsson   // Check if we need to pass the size to the delete operator.
113521122cf6SAnders Carlsson   llvm::Value *Size = 0;
113621122cf6SAnders Carlsson   QualType SizeTy;
113721122cf6SAnders Carlsson   if (DeleteFTy->getNumArgs() == 2) {
113821122cf6SAnders Carlsson     SizeTy = DeleteFTy->getArgType(1);
11397df3cbebSKen Dyck     CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
11407df3cbebSKen Dyck     Size = llvm::ConstantInt::get(ConvertType(SizeTy),
11417df3cbebSKen Dyck                                   DeleteTypeSize.getQuantity());
114221122cf6SAnders Carlsson   }
114321122cf6SAnders Carlsson 
114459486a2dSAnders Carlsson   QualType ArgTy = DeleteFTy->getArgType(0);
114559486a2dSAnders Carlsson   llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
114659486a2dSAnders Carlsson   DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy));
114759486a2dSAnders Carlsson 
114821122cf6SAnders Carlsson   if (Size)
114959486a2dSAnders Carlsson     DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy));
115059486a2dSAnders Carlsson 
115159486a2dSAnders Carlsson   // Emit the call to delete.
1152ab26cfa5SJohn McCall   EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy),
115361a401caSAnders Carlsson            CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(),
115459486a2dSAnders Carlsson            DeleteArgs, DeleteFD);
115559486a2dSAnders Carlsson }
115659486a2dSAnders Carlsson 
11578ed55a54SJohn McCall namespace {
11588ed55a54SJohn McCall   /// Calls the given 'operator delete' on a single object.
11598ed55a54SJohn McCall   struct CallObjectDelete : EHScopeStack::Cleanup {
11608ed55a54SJohn McCall     llvm::Value *Ptr;
11618ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
11628ed55a54SJohn McCall     QualType ElementType;
11638ed55a54SJohn McCall 
11648ed55a54SJohn McCall     CallObjectDelete(llvm::Value *Ptr,
11658ed55a54SJohn McCall                      const FunctionDecl *OperatorDelete,
11668ed55a54SJohn McCall                      QualType ElementType)
11678ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {}
11688ed55a54SJohn McCall 
11698ed55a54SJohn McCall     void Emit(CodeGenFunction &CGF, bool IsForEH) {
11708ed55a54SJohn McCall       CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType);
11718ed55a54SJohn McCall     }
11728ed55a54SJohn McCall   };
11738ed55a54SJohn McCall }
11748ed55a54SJohn McCall 
11758ed55a54SJohn McCall /// Emit the code for deleting a single object.
11768ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF,
11778ed55a54SJohn McCall                              const FunctionDecl *OperatorDelete,
11788ed55a54SJohn McCall                              llvm::Value *Ptr,
11798ed55a54SJohn McCall                              QualType ElementType) {
11808ed55a54SJohn McCall   // Find the destructor for the type, if applicable.  If the
11818ed55a54SJohn McCall   // destructor is virtual, we'll just emit the vcall and return.
11828ed55a54SJohn McCall   const CXXDestructorDecl *Dtor = 0;
11838ed55a54SJohn McCall   if (const RecordType *RT = ElementType->getAs<RecordType>()) {
11848ed55a54SJohn McCall     CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
11858ed55a54SJohn McCall     if (!RD->hasTrivialDestructor()) {
11868ed55a54SJohn McCall       Dtor = RD->getDestructor();
11878ed55a54SJohn McCall 
11888ed55a54SJohn McCall       if (Dtor->isVirtual()) {
11898ed55a54SJohn McCall         const llvm::Type *Ty =
11900d635f53SJohn McCall           CGF.getTypes().GetFunctionType(CGF.getTypes().getFunctionInfo(Dtor,
11910d635f53SJohn McCall                                                                Dtor_Complete),
11928ed55a54SJohn McCall                                          /*isVariadic=*/false);
11938ed55a54SJohn McCall 
11948ed55a54SJohn McCall         llvm::Value *Callee
11958ed55a54SJohn McCall           = CGF.BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty);
11968ed55a54SJohn McCall         CGF.EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0,
11978ed55a54SJohn McCall                               0, 0);
11988ed55a54SJohn McCall 
11998ed55a54SJohn McCall         // The dtor took care of deleting the object.
12008ed55a54SJohn McCall         return;
12018ed55a54SJohn McCall       }
12028ed55a54SJohn McCall     }
12038ed55a54SJohn McCall   }
12048ed55a54SJohn McCall 
12058ed55a54SJohn McCall   // Make sure that we call delete even if the dtor throws.
12068ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
12078ed55a54SJohn McCall                                             Ptr, OperatorDelete, ElementType);
12088ed55a54SJohn McCall 
12098ed55a54SJohn McCall   if (Dtor)
12108ed55a54SJohn McCall     CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete,
12118ed55a54SJohn McCall                               /*ForVirtualBase=*/false, Ptr);
12128ed55a54SJohn McCall 
12138ed55a54SJohn McCall   CGF.PopCleanupBlock();
12148ed55a54SJohn McCall }
12158ed55a54SJohn McCall 
12168ed55a54SJohn McCall namespace {
12178ed55a54SJohn McCall   /// Calls the given 'operator delete' on an array of objects.
12188ed55a54SJohn McCall   struct CallArrayDelete : EHScopeStack::Cleanup {
12198ed55a54SJohn McCall     llvm::Value *Ptr;
12208ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
12218ed55a54SJohn McCall     llvm::Value *NumElements;
12228ed55a54SJohn McCall     QualType ElementType;
12238ed55a54SJohn McCall     CharUnits CookieSize;
12248ed55a54SJohn McCall 
12258ed55a54SJohn McCall     CallArrayDelete(llvm::Value *Ptr,
12268ed55a54SJohn McCall                     const FunctionDecl *OperatorDelete,
12278ed55a54SJohn McCall                     llvm::Value *NumElements,
12288ed55a54SJohn McCall                     QualType ElementType,
12298ed55a54SJohn McCall                     CharUnits CookieSize)
12308ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements),
12318ed55a54SJohn McCall         ElementType(ElementType), CookieSize(CookieSize) {}
12328ed55a54SJohn McCall 
12338ed55a54SJohn McCall     void Emit(CodeGenFunction &CGF, bool IsForEH) {
12348ed55a54SJohn McCall       const FunctionProtoType *DeleteFTy =
12358ed55a54SJohn McCall         OperatorDelete->getType()->getAs<FunctionProtoType>();
12368ed55a54SJohn McCall       assert(DeleteFTy->getNumArgs() == 1 || DeleteFTy->getNumArgs() == 2);
12378ed55a54SJohn McCall 
12388ed55a54SJohn McCall       CallArgList Args;
12398ed55a54SJohn McCall 
12408ed55a54SJohn McCall       // Pass the pointer as the first argument.
12418ed55a54SJohn McCall       QualType VoidPtrTy = DeleteFTy->getArgType(0);
12428ed55a54SJohn McCall       llvm::Value *DeletePtr
12438ed55a54SJohn McCall         = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy));
12448ed55a54SJohn McCall       Args.push_back(std::make_pair(RValue::get(DeletePtr), VoidPtrTy));
12458ed55a54SJohn McCall 
12468ed55a54SJohn McCall       // Pass the original requested size as the second argument.
12478ed55a54SJohn McCall       if (DeleteFTy->getNumArgs() == 2) {
12488ed55a54SJohn McCall         QualType size_t = DeleteFTy->getArgType(1);
12498ed55a54SJohn McCall         const llvm::IntegerType *SizeTy
12508ed55a54SJohn McCall           = cast<llvm::IntegerType>(CGF.ConvertType(size_t));
12518ed55a54SJohn McCall 
12528ed55a54SJohn McCall         CharUnits ElementTypeSize =
12538ed55a54SJohn McCall           CGF.CGM.getContext().getTypeSizeInChars(ElementType);
12548ed55a54SJohn McCall 
12558ed55a54SJohn McCall         // The size of an element, multiplied by the number of elements.
12568ed55a54SJohn McCall         llvm::Value *Size
12578ed55a54SJohn McCall           = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity());
12588ed55a54SJohn McCall         Size = CGF.Builder.CreateMul(Size, NumElements);
12598ed55a54SJohn McCall 
12608ed55a54SJohn McCall         // Plus the size of the cookie if applicable.
12618ed55a54SJohn McCall         if (!CookieSize.isZero()) {
12628ed55a54SJohn McCall           llvm::Value *CookieSizeV
12638ed55a54SJohn McCall             = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
12648ed55a54SJohn McCall           Size = CGF.Builder.CreateAdd(Size, CookieSizeV);
12658ed55a54SJohn McCall         }
12668ed55a54SJohn McCall 
12678ed55a54SJohn McCall         Args.push_back(std::make_pair(RValue::get(Size), size_t));
12688ed55a54SJohn McCall       }
12698ed55a54SJohn McCall 
12708ed55a54SJohn McCall       // Emit the call to delete.
12718ed55a54SJohn McCall       CGF.EmitCall(CGF.getTypes().getFunctionInfo(Args, DeleteFTy),
12728ed55a54SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
12738ed55a54SJohn McCall                    ReturnValueSlot(), Args, OperatorDelete);
12748ed55a54SJohn McCall     }
12758ed55a54SJohn McCall   };
12768ed55a54SJohn McCall }
12778ed55a54SJohn McCall 
12788ed55a54SJohn McCall /// Emit the code for deleting an array of objects.
12798ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF,
12808ed55a54SJohn McCall                             const FunctionDecl *OperatorDelete,
12818ed55a54SJohn McCall                             llvm::Value *Ptr,
12828ed55a54SJohn McCall                             QualType ElementType) {
12838ed55a54SJohn McCall   llvm::Value *NumElements = 0;
12848ed55a54SJohn McCall   llvm::Value *AllocatedPtr = 0;
12858ed55a54SJohn McCall   CharUnits CookieSize;
12868ed55a54SJohn McCall   CGF.CGM.getCXXABI().ReadArrayCookie(CGF, Ptr, ElementType,
12878ed55a54SJohn McCall                                       NumElements, AllocatedPtr, CookieSize);
12888ed55a54SJohn McCall 
12898ed55a54SJohn McCall   assert(AllocatedPtr && "ReadArrayCookie didn't set AllocatedPtr");
12908ed55a54SJohn McCall 
12918ed55a54SJohn McCall   // Make sure that we call delete even if one of the dtors throws.
12928ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup,
12938ed55a54SJohn McCall                                            AllocatedPtr, OperatorDelete,
12948ed55a54SJohn McCall                                            NumElements, ElementType,
12958ed55a54SJohn McCall                                            CookieSize);
12968ed55a54SJohn McCall 
12978ed55a54SJohn McCall   if (const CXXRecordDecl *RD = ElementType->getAsCXXRecordDecl()) {
12988ed55a54SJohn McCall     if (!RD->hasTrivialDestructor()) {
12998ed55a54SJohn McCall       assert(NumElements && "ReadArrayCookie didn't find element count"
13008ed55a54SJohn McCall                             " for a class with destructor");
13018ed55a54SJohn McCall       CGF.EmitCXXAggrDestructorCall(RD->getDestructor(), NumElements, Ptr);
13028ed55a54SJohn McCall     }
13038ed55a54SJohn McCall   }
13048ed55a54SJohn McCall 
13058ed55a54SJohn McCall   CGF.PopCleanupBlock();
13068ed55a54SJohn McCall }
13078ed55a54SJohn McCall 
130859486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
130959486a2dSAnders Carlsson 
131059486a2dSAnders Carlsson   // Get at the argument before we performed the implicit conversion
131159486a2dSAnders Carlsson   // to void*.
131259486a2dSAnders Carlsson   const Expr *Arg = E->getArgument();
131359486a2dSAnders Carlsson   while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) {
1314e302792bSJohn McCall     if (ICE->getCastKind() != CK_UserDefinedConversion &&
131559486a2dSAnders Carlsson         ICE->getType()->isVoidPointerType())
131659486a2dSAnders Carlsson       Arg = ICE->getSubExpr();
131759486a2dSAnders Carlsson     else
131859486a2dSAnders Carlsson       break;
131959486a2dSAnders Carlsson   }
132059486a2dSAnders Carlsson 
132159486a2dSAnders Carlsson   llvm::Value *Ptr = EmitScalarExpr(Arg);
132259486a2dSAnders Carlsson 
132359486a2dSAnders Carlsson   // Null check the pointer.
132459486a2dSAnders Carlsson   llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
132559486a2dSAnders Carlsson   llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
132659486a2dSAnders Carlsson 
132759486a2dSAnders Carlsson   llvm::Value *IsNull =
132859486a2dSAnders Carlsson     Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()),
132959486a2dSAnders Carlsson                          "isnull");
133059486a2dSAnders Carlsson 
133159486a2dSAnders Carlsson   Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
133259486a2dSAnders Carlsson   EmitBlock(DeleteNotNull);
133359486a2dSAnders Carlsson 
13348ed55a54SJohn McCall   // We might be deleting a pointer to array.  If so, GEP down to the
13358ed55a54SJohn McCall   // first non-array element.
13368ed55a54SJohn McCall   // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*)
13378ed55a54SJohn McCall   QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
13388ed55a54SJohn McCall   if (DeleteTy->isConstantArrayType()) {
13398ed55a54SJohn McCall     llvm::Value *Zero = Builder.getInt32(0);
13408ed55a54SJohn McCall     llvm::SmallVector<llvm::Value*,8> GEP;
134159486a2dSAnders Carlsson 
13428ed55a54SJohn McCall     GEP.push_back(Zero); // point at the outermost array
13438ed55a54SJohn McCall 
13448ed55a54SJohn McCall     // For each layer of array type we're pointing at:
13458ed55a54SJohn McCall     while (const ConstantArrayType *Arr
13468ed55a54SJohn McCall              = getContext().getAsConstantArrayType(DeleteTy)) {
13478ed55a54SJohn McCall       // 1. Unpeel the array type.
13488ed55a54SJohn McCall       DeleteTy = Arr->getElementType();
13498ed55a54SJohn McCall 
13508ed55a54SJohn McCall       // 2. GEP to the first element of the array.
13518ed55a54SJohn McCall       GEP.push_back(Zero);
13528ed55a54SJohn McCall     }
13538ed55a54SJohn McCall 
13548ed55a54SJohn McCall     Ptr = Builder.CreateInBoundsGEP(Ptr, GEP.begin(), GEP.end(), "del.first");
13558ed55a54SJohn McCall   }
13568ed55a54SJohn McCall 
135704f36218SDouglas Gregor   assert(ConvertTypeForMem(DeleteTy) ==
135804f36218SDouglas Gregor          cast<llvm::PointerType>(Ptr->getType())->getElementType());
13598ed55a54SJohn McCall 
136059486a2dSAnders Carlsson   if (E->isArrayForm()) {
13618ed55a54SJohn McCall     EmitArrayDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy);
13628ed55a54SJohn McCall   } else {
13638ed55a54SJohn McCall     EmitObjectDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy);
136459486a2dSAnders Carlsson   }
136559486a2dSAnders Carlsson 
136659486a2dSAnders Carlsson   EmitBlock(DeleteEnd);
136759486a2dSAnders Carlsson }
136859486a2dSAnders Carlsson 
136959486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
137059486a2dSAnders Carlsson   QualType Ty = E->getType();
137159486a2dSAnders Carlsson   const llvm::Type *LTy = ConvertType(Ty)->getPointerTo();
1372fd7dfeb7SAnders Carlsson 
13733f4336cbSAnders Carlsson   if (E->isTypeOperand()) {
13743f4336cbSAnders Carlsson     llvm::Constant *TypeInfo =
13753f4336cbSAnders Carlsson       CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand());
13763f4336cbSAnders Carlsson     return Builder.CreateBitCast(TypeInfo, LTy);
13773f4336cbSAnders Carlsson   }
1378fd7dfeb7SAnders Carlsson 
137959486a2dSAnders Carlsson   Expr *subE = E->getExprOperand();
138059486a2dSAnders Carlsson   Ty = subE->getType();
138159486a2dSAnders Carlsson   CanQualType CanTy = CGM.getContext().getCanonicalType(Ty);
138259486a2dSAnders Carlsson   Ty = CanTy.getUnqualifiedType().getNonReferenceType();
138359486a2dSAnders Carlsson   if (const RecordType *RT = Ty->getAs<RecordType>()) {
138459486a2dSAnders Carlsson     const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
138559486a2dSAnders Carlsson     if (RD->isPolymorphic()) {
138659486a2dSAnders Carlsson       // FIXME: if subE is an lvalue do
138759486a2dSAnders Carlsson       LValue Obj = EmitLValue(subE);
138859486a2dSAnders Carlsson       llvm::Value *This = Obj.getAddress();
138959486a2dSAnders Carlsson       // We need to do a zero check for *p, unless it has NonNullAttr.
139059486a2dSAnders Carlsson       // FIXME: PointerType->hasAttr<NonNullAttr>()
139159486a2dSAnders Carlsson       bool CanBeZero = false;
139259486a2dSAnders Carlsson       if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens()))
1393e302792bSJohn McCall         if (UO->getOpcode() == UO_Deref)
139459486a2dSAnders Carlsson           CanBeZero = true;
139559486a2dSAnders Carlsson       if (CanBeZero) {
139659486a2dSAnders Carlsson         llvm::BasicBlock *NonZeroBlock = createBasicBlock();
139759486a2dSAnders Carlsson         llvm::BasicBlock *ZeroBlock = createBasicBlock();
139859486a2dSAnders Carlsson 
13998fc50c29SDan Gohman         llvm::Value *Zero = llvm::Constant::getNullValue(This->getType());
14008fc50c29SDan Gohman         Builder.CreateCondBr(Builder.CreateICmpNE(This, Zero),
140159486a2dSAnders Carlsson                              NonZeroBlock, ZeroBlock);
140259486a2dSAnders Carlsson         EmitBlock(ZeroBlock);
140359486a2dSAnders Carlsson         /// Call __cxa_bad_typeid
140459486a2dSAnders Carlsson         const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext);
140559486a2dSAnders Carlsson         const llvm::FunctionType *FTy;
140659486a2dSAnders Carlsson         FTy = llvm::FunctionType::get(ResultType, false);
140759486a2dSAnders Carlsson         llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid");
140859486a2dSAnders Carlsson         Builder.CreateCall(F)->setDoesNotReturn();
140959486a2dSAnders Carlsson         Builder.CreateUnreachable();
141059486a2dSAnders Carlsson         EmitBlock(NonZeroBlock);
141159486a2dSAnders Carlsson       }
14128fc50c29SDan Gohman       llvm::Value *V = GetVTablePtr(This, LTy->getPointerTo());
141359486a2dSAnders Carlsson       V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL);
141459486a2dSAnders Carlsson       V = Builder.CreateLoad(V);
141559486a2dSAnders Carlsson       return V;
141659486a2dSAnders Carlsson     }
141759486a2dSAnders Carlsson   }
14183f4336cbSAnders Carlsson   return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy);
141959486a2dSAnders Carlsson }
142059486a2dSAnders Carlsson 
142159486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V,
142259486a2dSAnders Carlsson                                               const CXXDynamicCastExpr *DCE) {
14233f4336cbSAnders Carlsson   QualType SrcTy = DCE->getSubExpr()->getType();
14243f4336cbSAnders Carlsson   QualType DestTy = DCE->getTypeAsWritten();
14253f4336cbSAnders Carlsson   QualType InnerType = DestTy->getPointeeType();
14263f4336cbSAnders Carlsson 
142759486a2dSAnders Carlsson   const llvm::Type *LTy = ConvertType(DCE->getType());
142859486a2dSAnders Carlsson 
142959486a2dSAnders Carlsson   bool CanBeZero = false;
143059486a2dSAnders Carlsson   bool ToVoid = false;
143159486a2dSAnders Carlsson   bool ThrowOnBad = false;
14323f4336cbSAnders Carlsson   if (DestTy->isPointerType()) {
143359486a2dSAnders Carlsson     // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this
143459486a2dSAnders Carlsson     CanBeZero = true;
143559486a2dSAnders Carlsson     if (InnerType->isVoidType())
143659486a2dSAnders Carlsson       ToVoid = true;
143759486a2dSAnders Carlsson   } else {
143859486a2dSAnders Carlsson     LTy = LTy->getPointerTo();
1439fa8b4955SDouglas Gregor 
1440fa8b4955SDouglas Gregor     // FIXME: What if exceptions are disabled?
144159486a2dSAnders Carlsson     ThrowOnBad = true;
144259486a2dSAnders Carlsson   }
144359486a2dSAnders Carlsson 
14443f4336cbSAnders Carlsson   if (SrcTy->isPointerType() || SrcTy->isReferenceType())
14453f4336cbSAnders Carlsson     SrcTy = SrcTy->getPointeeType();
14463f4336cbSAnders Carlsson   SrcTy = SrcTy.getUnqualifiedType();
14473f4336cbSAnders Carlsson 
14480087bc85SAnders Carlsson   if (DestTy->isPointerType() || DestTy->isReferenceType())
14493f4336cbSAnders Carlsson     DestTy = DestTy->getPointeeType();
14503f4336cbSAnders Carlsson   DestTy = DestTy.getUnqualifiedType();
145159486a2dSAnders Carlsson 
145259486a2dSAnders Carlsson   llvm::BasicBlock *ContBlock = createBasicBlock();
145359486a2dSAnders Carlsson   llvm::BasicBlock *NullBlock = 0;
145459486a2dSAnders Carlsson   llvm::BasicBlock *NonZeroBlock = 0;
145559486a2dSAnders Carlsson   if (CanBeZero) {
145659486a2dSAnders Carlsson     NonZeroBlock = createBasicBlock();
145759486a2dSAnders Carlsson     NullBlock = createBasicBlock();
14583f4336cbSAnders Carlsson     Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock);
145959486a2dSAnders Carlsson     EmitBlock(NonZeroBlock);
146059486a2dSAnders Carlsson   }
146159486a2dSAnders Carlsson 
146259486a2dSAnders Carlsson   llvm::BasicBlock *BadCastBlock = 0;
146359486a2dSAnders Carlsson 
14643f4336cbSAnders Carlsson   const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
146559486a2dSAnders Carlsson 
146659486a2dSAnders Carlsson   // See if this is a dynamic_cast(void*)
146759486a2dSAnders Carlsson   if (ToVoid) {
146859486a2dSAnders Carlsson     llvm::Value *This = V;
14698fc50c29SDan Gohman     V = GetVTablePtr(This, PtrDiffTy->getPointerTo());
147059486a2dSAnders Carlsson     V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL);
147159486a2dSAnders Carlsson     V = Builder.CreateLoad(V, "offset to top");
147259486a2dSAnders Carlsson     This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext));
147359486a2dSAnders Carlsson     V = Builder.CreateInBoundsGEP(This, V);
147459486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, LTy);
147559486a2dSAnders Carlsson   } else {
147659486a2dSAnders Carlsson     /// Call __dynamic_cast
147759486a2dSAnders Carlsson     const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext);
147859486a2dSAnders Carlsson     const llvm::FunctionType *FTy;
147959486a2dSAnders Carlsson     std::vector<const llvm::Type*> ArgTys;
148059486a2dSAnders Carlsson     const llvm::Type *PtrToInt8Ty
148159486a2dSAnders Carlsson       = llvm::Type::getInt8Ty(VMContext)->getPointerTo();
148259486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
148359486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
148459486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
148559486a2dSAnders Carlsson     ArgTys.push_back(PtrDiffTy);
148659486a2dSAnders Carlsson     FTy = llvm::FunctionType::get(ResultType, ArgTys, false);
148759486a2dSAnders Carlsson 
148859486a2dSAnders Carlsson     // FIXME: Calculate better hint.
148959486a2dSAnders Carlsson     llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL);
14903f4336cbSAnders Carlsson 
14913f4336cbSAnders Carlsson     assert(SrcTy->isRecordType() && "Src type must be record type!");
14923f4336cbSAnders Carlsson     assert(DestTy->isRecordType() && "Dest type must be record type!");
14933f4336cbSAnders Carlsson 
1494247894b3SDouglas Gregor     llvm::Value *SrcArg
1495247894b3SDouglas Gregor       = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType());
1496247894b3SDouglas Gregor     llvm::Value *DestArg
1497247894b3SDouglas Gregor       = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType());
14983f4336cbSAnders Carlsson 
149959486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, PtrToInt8Ty);
150059486a2dSAnders Carlsson     V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"),
15013f4336cbSAnders Carlsson                             V, SrcArg, DestArg, hint);
150259486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, LTy);
150359486a2dSAnders Carlsson 
150459486a2dSAnders Carlsson     if (ThrowOnBad) {
150559486a2dSAnders Carlsson       BadCastBlock = createBasicBlock();
15063f4336cbSAnders Carlsson       Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock);
150759486a2dSAnders Carlsson       EmitBlock(BadCastBlock);
1508fa8b4955SDouglas Gregor       /// Invoke __cxa_bad_cast
150959486a2dSAnders Carlsson       ResultType = llvm::Type::getVoidTy(VMContext);
151059486a2dSAnders Carlsson       const llvm::FunctionType *FBadTy;
151159486a2dSAnders Carlsson       FBadTy = llvm::FunctionType::get(ResultType, false);
151259486a2dSAnders Carlsson       llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast");
1513fa8b4955SDouglas Gregor       if (llvm::BasicBlock *InvokeDest = getInvokeDest()) {
1514fa8b4955SDouglas Gregor         llvm::BasicBlock *Cont = createBasicBlock("invoke.cont");
1515fa8b4955SDouglas Gregor         Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn();
1516fa8b4955SDouglas Gregor         EmitBlock(Cont);
1517fa8b4955SDouglas Gregor       } else {
1518fa8b4955SDouglas Gregor         // FIXME: Does this ever make sense?
151959486a2dSAnders Carlsson         Builder.CreateCall(F)->setDoesNotReturn();
1520fa8b4955SDouglas Gregor       }
152159486a2dSAnders Carlsson       Builder.CreateUnreachable();
152259486a2dSAnders Carlsson     }
152359486a2dSAnders Carlsson   }
152459486a2dSAnders Carlsson 
152559486a2dSAnders Carlsson   if (CanBeZero) {
152659486a2dSAnders Carlsson     Builder.CreateBr(ContBlock);
152759486a2dSAnders Carlsson     EmitBlock(NullBlock);
152859486a2dSAnders Carlsson     Builder.CreateBr(ContBlock);
152959486a2dSAnders Carlsson   }
153059486a2dSAnders Carlsson   EmitBlock(ContBlock);
153159486a2dSAnders Carlsson   if (CanBeZero) {
153259486a2dSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(LTy);
153359486a2dSAnders Carlsson     PHI->reserveOperandSpace(2);
153459486a2dSAnders Carlsson     PHI->addIncoming(V, NonZeroBlock);
153559486a2dSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock);
153659486a2dSAnders Carlsson     V = PHI;
153759486a2dSAnders Carlsson   }
153859486a2dSAnders Carlsson 
153959486a2dSAnders Carlsson   return V;
154059486a2dSAnders Carlsson }
1541