159486a2dSAnders Carlsson //===--- CGExprCXX.cpp - Emit LLVM Code for C++ expressions ---------------===//
259486a2dSAnders Carlsson //
359486a2dSAnders Carlsson //                     The LLVM Compiler Infrastructure
459486a2dSAnders Carlsson //
559486a2dSAnders Carlsson // This file is distributed under the University of Illinois Open Source
659486a2dSAnders Carlsson // License. See LICENSE.TXT for details.
759486a2dSAnders Carlsson //
859486a2dSAnders Carlsson //===----------------------------------------------------------------------===//
959486a2dSAnders Carlsson //
1059486a2dSAnders Carlsson // This contains code dealing with code generation of C++ expressions
1159486a2dSAnders Carlsson //
1259486a2dSAnders Carlsson //===----------------------------------------------------------------------===//
1359486a2dSAnders Carlsson 
1459486a2dSAnders Carlsson #include "CodeGenFunction.h"
155d865c32SJohn McCall #include "CGCXXABI.h"
1660d215b6SFariborz Jahanian #include "CGObjCRuntime.h"
1726008e07SChris Lattner #include "llvm/Intrinsics.h"
1859486a2dSAnders Carlsson using namespace clang;
1959486a2dSAnders Carlsson using namespace CodeGen;
2059486a2dSAnders Carlsson 
2127da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD,
2227da15baSAnders Carlsson                                           llvm::Value *Callee,
2327da15baSAnders Carlsson                                           ReturnValueSlot ReturnValue,
2427da15baSAnders Carlsson                                           llvm::Value *This,
25e36a6b3eSAnders Carlsson                                           llvm::Value *VTT,
2627da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgBeg,
2727da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgEnd) {
2827da15baSAnders Carlsson   assert(MD->isInstance() &&
2927da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
3027da15baSAnders Carlsson 
3127da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
3227da15baSAnders Carlsson 
3327da15baSAnders Carlsson   CallArgList Args;
3427da15baSAnders Carlsson 
3527da15baSAnders Carlsson   // Push the this ptr.
3627da15baSAnders Carlsson   Args.push_back(std::make_pair(RValue::get(This),
3727da15baSAnders Carlsson                                 MD->getThisType(getContext())));
3827da15baSAnders Carlsson 
39e36a6b3eSAnders Carlsson   // If there is a VTT parameter, emit it.
40e36a6b3eSAnders Carlsson   if (VTT) {
41e36a6b3eSAnders Carlsson     QualType T = getContext().getPointerType(getContext().VoidPtrTy);
42e36a6b3eSAnders Carlsson     Args.push_back(std::make_pair(RValue::get(VTT), T));
43e36a6b3eSAnders Carlsson   }
44e36a6b3eSAnders Carlsson 
4527da15baSAnders Carlsson   // And the rest of the call args
4627da15baSAnders Carlsson   EmitCallArgs(Args, FPT, ArgBeg, ArgEnd);
4727da15baSAnders Carlsson 
48ab26cfa5SJohn McCall   QualType ResultType = FPT->getResultType();
49ab26cfa5SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args,
50c50c27ccSRafael Espindola                                                  FPT->getExtInfo()),
51c50c27ccSRafael Espindola                   Callee, ReturnValue, Args, MD);
5227da15baSAnders Carlsson }
5327da15baSAnders Carlsson 
5427da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given
5527da15baSAnders Carlsson /// expr can be devirtualized.
5627da15baSAnders Carlsson static bool canDevirtualizeMemberFunctionCalls(const Expr *Base) {
5727da15baSAnders Carlsson   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) {
5827da15baSAnders Carlsson     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
5927da15baSAnders Carlsson       // This is a record decl. We know the type and can devirtualize it.
6027da15baSAnders Carlsson       return VD->getType()->isRecordType();
6127da15baSAnders Carlsson     }
6227da15baSAnders Carlsson 
6327da15baSAnders Carlsson     return false;
6427da15baSAnders Carlsson   }
6527da15baSAnders Carlsson 
6627da15baSAnders Carlsson   // We can always devirtualize calls on temporary object expressions.
67a682427eSEli Friedman   if (isa<CXXConstructExpr>(Base))
6827da15baSAnders Carlsson     return true;
6927da15baSAnders Carlsson 
7027da15baSAnders Carlsson   // And calls on bound temporaries.
7127da15baSAnders Carlsson   if (isa<CXXBindTemporaryExpr>(Base))
7227da15baSAnders Carlsson     return true;
7327da15baSAnders Carlsson 
7427da15baSAnders Carlsson   // Check if this is a call expr that returns a record type.
7527da15baSAnders Carlsson   if (const CallExpr *CE = dyn_cast<CallExpr>(Base))
7627da15baSAnders Carlsson     return CE->getCallReturnType()->isRecordType();
7727da15baSAnders Carlsson 
7827da15baSAnders Carlsson   // We can't devirtualize the call.
7927da15baSAnders Carlsson   return false;
8027da15baSAnders Carlsson }
8127da15baSAnders Carlsson 
8227da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
8327da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
8427da15baSAnders Carlsson   if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens()))
8527da15baSAnders Carlsson     return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
8627da15baSAnders Carlsson 
8727da15baSAnders Carlsson   const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens());
8827da15baSAnders Carlsson   const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
8927da15baSAnders Carlsson 
9027da15baSAnders Carlsson   if (MD->isStatic()) {
9127da15baSAnders Carlsson     // The method is static, emit it as we would a regular call.
9227da15baSAnders Carlsson     llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
9327da15baSAnders Carlsson     return EmitCall(getContext().getPointerType(MD->getType()), Callee,
9427da15baSAnders Carlsson                     ReturnValue, CE->arg_begin(), CE->arg_end());
9527da15baSAnders Carlsson   }
9627da15baSAnders Carlsson 
97*0d635f53SJohn McCall   // Compute the object pointer.
9827da15baSAnders Carlsson   llvm::Value *This;
9927da15baSAnders Carlsson   if (ME->isArrow())
10027da15baSAnders Carlsson     This = EmitScalarExpr(ME->getBase());
10127da15baSAnders Carlsson   else {
10227da15baSAnders Carlsson     LValue BaseLV = EmitLValue(ME->getBase());
10327da15baSAnders Carlsson     This = BaseLV.getAddress();
10427da15baSAnders Carlsson   }
10527da15baSAnders Carlsson 
106*0d635f53SJohn McCall   if (MD->isTrivial()) {
107*0d635f53SJohn McCall     if (isa<CXXDestructorDecl>(MD)) return RValue::get(0);
108*0d635f53SJohn McCall 
109*0d635f53SJohn McCall     assert(MD->isCopyAssignment() && "unknown trivial member function");
11027da15baSAnders Carlsson     // We don't like to generate the trivial copy assignment operator when
11127da15baSAnders Carlsson     // it isn't necessary; just produce the proper effect here.
11227da15baSAnders Carlsson     llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
11327da15baSAnders Carlsson     EmitAggregateCopy(This, RHS, CE->getType());
11427da15baSAnders Carlsson     return RValue::get(This);
11527da15baSAnders Carlsson   }
11627da15baSAnders Carlsson 
117*0d635f53SJohn McCall   // Compute the function type we're calling.
118*0d635f53SJohn McCall   const CGFunctionInfo &FInfo =
119*0d635f53SJohn McCall     (isa<CXXDestructorDecl>(MD)
120*0d635f53SJohn McCall      ? CGM.getTypes().getFunctionInfo(cast<CXXDestructorDecl>(MD),
121*0d635f53SJohn McCall                                       Dtor_Complete)
122*0d635f53SJohn McCall      : CGM.getTypes().getFunctionInfo(MD));
123*0d635f53SJohn McCall 
124*0d635f53SJohn McCall   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
125*0d635f53SJohn McCall   const llvm::Type *Ty
126*0d635f53SJohn McCall     = CGM.getTypes().GetFunctionType(FInfo, FPT->isVariadic());
127*0d635f53SJohn McCall 
12827da15baSAnders Carlsson   // C++ [class.virtual]p12:
12927da15baSAnders Carlsson   //   Explicit qualification with the scope operator (5.1) suppresses the
13027da15baSAnders Carlsson   //   virtual call mechanism.
13127da15baSAnders Carlsson   //
13227da15baSAnders Carlsson   // We also don't emit a virtual call if the base expression has a record type
13327da15baSAnders Carlsson   // because then we know what the type is.
134*0d635f53SJohn McCall   bool UseVirtualCall = MD->isVirtual() && !ME->hasQualifier()
135*0d635f53SJohn McCall                      && !canDevirtualizeMemberFunctionCalls(ME->getBase());
136*0d635f53SJohn McCall 
13727da15baSAnders Carlsson   llvm::Value *Callee;
138*0d635f53SJohn McCall   if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(MD)) {
139*0d635f53SJohn McCall     if (UseVirtualCall) {
140*0d635f53SJohn McCall       Callee = BuildVirtualCall(Dtor, Dtor_Complete, This, Ty);
14127da15baSAnders Carlsson     } else {
142*0d635f53SJohn McCall       Callee = CGM.GetAddrOfFunction(GlobalDecl(Dtor, Dtor_Complete), Ty);
14327da15baSAnders Carlsson     }
144*0d635f53SJohn McCall   } else if (UseVirtualCall) {
14527da15baSAnders Carlsson     Callee = BuildVirtualCall(MD, This, Ty);
14627da15baSAnders Carlsson   } else {
14727da15baSAnders Carlsson     Callee = CGM.GetAddrOfFunction(MD, Ty);
14827da15baSAnders Carlsson   }
14927da15baSAnders Carlsson 
150e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
15127da15baSAnders Carlsson                            CE->arg_begin(), CE->arg_end());
15227da15baSAnders Carlsson }
15327da15baSAnders Carlsson 
15427da15baSAnders Carlsson RValue
15527da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
15627da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
15727da15baSAnders Carlsson   const BinaryOperator *BO =
15827da15baSAnders Carlsson       cast<BinaryOperator>(E->getCallee()->IgnoreParens());
15927da15baSAnders Carlsson   const Expr *BaseExpr = BO->getLHS();
16027da15baSAnders Carlsson   const Expr *MemFnExpr = BO->getRHS();
16127da15baSAnders Carlsson 
16227da15baSAnders Carlsson   const MemberPointerType *MPT =
16327da15baSAnders Carlsson     MemFnExpr->getType()->getAs<MemberPointerType>();
164475999dcSJohn McCall 
16527da15baSAnders Carlsson   const FunctionProtoType *FPT =
16627da15baSAnders Carlsson     MPT->getPointeeType()->getAs<FunctionProtoType>();
16727da15baSAnders Carlsson   const CXXRecordDecl *RD =
16827da15baSAnders Carlsson     cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
16927da15baSAnders Carlsson 
17027da15baSAnders Carlsson   // Get the member function pointer.
171a1dee530SJohn McCall   llvm::Value *MemFnPtr = EmitScalarExpr(MemFnExpr);
17227da15baSAnders Carlsson 
17327da15baSAnders Carlsson   // Emit the 'this' pointer.
17427da15baSAnders Carlsson   llvm::Value *This;
17527da15baSAnders Carlsson 
176e302792bSJohn McCall   if (BO->getOpcode() == BO_PtrMemI)
17727da15baSAnders Carlsson     This = EmitScalarExpr(BaseExpr);
17827da15baSAnders Carlsson   else
17927da15baSAnders Carlsson     This = EmitLValue(BaseExpr).getAddress();
18027da15baSAnders Carlsson 
181475999dcSJohn McCall   // Ask the ABI to load the callee.  Note that This is modified.
182475999dcSJohn McCall   llvm::Value *Callee =
183475999dcSJohn McCall     CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(CGF, This, MemFnPtr, MPT);
18427da15baSAnders Carlsson 
18527da15baSAnders Carlsson   CallArgList Args;
18627da15baSAnders Carlsson 
18727da15baSAnders Carlsson   QualType ThisType =
18827da15baSAnders Carlsson     getContext().getPointerType(getContext().getTagDeclType(RD));
18927da15baSAnders Carlsson 
19027da15baSAnders Carlsson   // Push the this ptr.
19127da15baSAnders Carlsson   Args.push_back(std::make_pair(RValue::get(This), ThisType));
19227da15baSAnders Carlsson 
19327da15baSAnders Carlsson   // And the rest of the call args
19427da15baSAnders Carlsson   EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
195ab26cfa5SJohn McCall   const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>();
196ab26cfa5SJohn McCall   return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee,
19727da15baSAnders Carlsson                   ReturnValue, Args);
19827da15baSAnders Carlsson }
19927da15baSAnders Carlsson 
20027da15baSAnders Carlsson RValue
20127da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
20227da15baSAnders Carlsson                                                const CXXMethodDecl *MD,
20327da15baSAnders Carlsson                                                ReturnValueSlot ReturnValue) {
20427da15baSAnders Carlsson   assert(MD->isInstance() &&
20527da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
20627da15baSAnders Carlsson   if (MD->isCopyAssignment()) {
20727da15baSAnders Carlsson     const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext());
20827da15baSAnders Carlsson     if (ClassDecl->hasTrivialCopyAssignment()) {
20927da15baSAnders Carlsson       assert(!ClassDecl->hasUserDeclaredCopyAssignment() &&
21027da15baSAnders Carlsson              "EmitCXXOperatorMemberCallExpr - user declared copy assignment");
21143a40f93SFariborz Jahanian       LValue LV = EmitLValue(E->getArg(0));
21243a40f93SFariborz Jahanian       llvm::Value *This;
21361a31241SFariborz Jahanian       if (LV.isPropertyRef() || LV.isKVCRef()) {
2141b8b8bf2SFariborz Jahanian         llvm::Value *AggLoc  = CreateMemTemp(E->getArg(1)->getType());
215e1b45a5eSFariborz Jahanian         EmitAggExpr(E->getArg(1), AggLoc, false /*VolatileDest*/);
21661a31241SFariborz Jahanian         if (LV.isPropertyRef())
217e1b45a5eSFariborz Jahanian           EmitObjCPropertySet(LV.getPropertyRefExpr(),
21861a31241SFariborz Jahanian                               RValue::getAggregate(AggLoc,
21961a31241SFariborz Jahanian                                                    false /*VolatileDest*/));
22061a31241SFariborz Jahanian         else
22161a31241SFariborz Jahanian           EmitObjCPropertySet(LV.getKVCRefExpr(),
22261a31241SFariborz Jahanian                               RValue::getAggregate(AggLoc,
22361a31241SFariborz Jahanian                                                    false /*VolatileDest*/));
224e1b45a5eSFariborz Jahanian         return RValue::getAggregate(0, false);
22543a40f93SFariborz Jahanian       }
22643a40f93SFariborz Jahanian       else
22743a40f93SFariborz Jahanian         This = LV.getAddress();
22843a40f93SFariborz Jahanian 
22927da15baSAnders Carlsson       llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
23027da15baSAnders Carlsson       QualType Ty = E->getType();
23127da15baSAnders Carlsson       EmitAggregateCopy(This, Src, Ty);
23227da15baSAnders Carlsson       return RValue::get(This);
23327da15baSAnders Carlsson     }
23427da15baSAnders Carlsson   }
23527da15baSAnders Carlsson 
23627da15baSAnders Carlsson   const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
23727da15baSAnders Carlsson   const llvm::Type *Ty =
23827da15baSAnders Carlsson     CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD),
23927da15baSAnders Carlsson                                    FPT->isVariadic());
240fdf474b0SFariborz Jahanian   LValue LV = EmitLValue(E->getArg(0));
241fdf474b0SFariborz Jahanian   llvm::Value *This;
24261a31241SFariborz Jahanian   if (LV.isPropertyRef() || LV.isKVCRef()) {
24361a31241SFariborz Jahanian     QualType QT = E->getArg(0)->getType();
24461a31241SFariborz Jahanian     RValue RV =
24561a31241SFariborz Jahanian       LV.isPropertyRef() ? EmitLoadOfPropertyRefLValue(LV, QT)
24661a31241SFariborz Jahanian                          : EmitLoadOfKVCRefLValue(LV, QT);
2476855ba2cSFariborz Jahanian     assert (!RV.isScalar() && "EmitCXXOperatorMemberCallExpr");
2486855ba2cSFariborz Jahanian     This = RV.getAggregateAddr();
249fdf474b0SFariborz Jahanian   }
250fdf474b0SFariborz Jahanian   else
251fdf474b0SFariborz Jahanian     This = LV.getAddress();
25227da15baSAnders Carlsson 
25327da15baSAnders Carlsson   llvm::Value *Callee;
25427da15baSAnders Carlsson   if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0)))
25527da15baSAnders Carlsson     Callee = BuildVirtualCall(MD, This, Ty);
25627da15baSAnders Carlsson   else
25727da15baSAnders Carlsson     Callee = CGM.GetAddrOfFunction(MD, Ty);
25827da15baSAnders Carlsson 
259e36a6b3eSAnders Carlsson   return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0,
26027da15baSAnders Carlsson                            E->arg_begin() + 1, E->arg_end());
26127da15baSAnders Carlsson }
26227da15baSAnders Carlsson 
26327da15baSAnders Carlsson void
26427da15baSAnders Carlsson CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest,
26527da15baSAnders Carlsson                                       const CXXConstructExpr *E) {
26627da15baSAnders Carlsson   assert(Dest && "Must have a destination!");
26727da15baSAnders Carlsson   const CXXConstructorDecl *CD = E->getConstructor();
268630c76efSDouglas Gregor 
269630c76efSDouglas Gregor   // If we require zero initialization before (or instead of) calling the
270630c76efSDouglas Gregor   // constructor, as can be the case with a non-user-provided default
271630c76efSDouglas Gregor   // constructor, emit the zero initialization now.
272e3b3464dSDouglas Gregor   if (E->requiresZeroInitialization())
273e3b3464dSDouglas Gregor     EmitNullInitialization(Dest, E->getType());
274e3b3464dSDouglas Gregor 
275630c76efSDouglas Gregor 
276630c76efSDouglas Gregor   // If this is a call to a trivial default constructor, do nothing.
277630c76efSDouglas Gregor   if (CD->isTrivial() && CD->isDefaultConstructor())
27827da15baSAnders Carlsson     return;
279630c76efSDouglas Gregor 
28027da15baSAnders Carlsson   // Code gen optimization to eliminate copy constructor and return
281222cf0efSDouglas Gregor   // its first argument instead, if in fact that argument is a temporary
282222cf0efSDouglas Gregor   // object.
28327da15baSAnders Carlsson   if (getContext().getLangOptions().ElideConstructors && E->isElidable()) {
284222cf0efSDouglas Gregor     if (const Expr *Arg = E->getArg(0)->getTemporaryObject()) {
28527da15baSAnders Carlsson       EmitAggExpr(Arg, Dest, false);
28627da15baSAnders Carlsson       return;
28727da15baSAnders Carlsson     }
288222cf0efSDouglas Gregor   }
289630c76efSDouglas Gregor 
290630c76efSDouglas Gregor   const ConstantArrayType *Array
291630c76efSDouglas Gregor     = getContext().getAsConstantArrayType(E->getType());
29227da15baSAnders Carlsson   if (Array) {
29327da15baSAnders Carlsson     QualType BaseElementTy = getContext().getBaseElementType(Array);
29427da15baSAnders Carlsson     const llvm::Type *BasePtr = ConvertType(BaseElementTy);
29527da15baSAnders Carlsson     BasePtr = llvm::PointerType::getUnqual(BasePtr);
29627da15baSAnders Carlsson     llvm::Value *BaseAddrPtr =
29727da15baSAnders Carlsson       Builder.CreateBitCast(Dest, BasePtr);
29827da15baSAnders Carlsson 
29927da15baSAnders Carlsson     EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr,
30027da15baSAnders Carlsson                                E->arg_begin(), E->arg_end());
30127da15baSAnders Carlsson   }
302e11f9ce9SAnders Carlsson   else {
303e11f9ce9SAnders Carlsson     CXXCtorType Type =
304e11f9ce9SAnders Carlsson       (E->getConstructionKind() == CXXConstructExpr::CK_Complete)
305e11f9ce9SAnders Carlsson       ? Ctor_Complete : Ctor_Base;
306e11f9ce9SAnders Carlsson     bool ForVirtualBase =
307e11f9ce9SAnders Carlsson       E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase;
308e11f9ce9SAnders Carlsson 
30927da15baSAnders Carlsson     // Call the constructor.
310e11f9ce9SAnders Carlsson     EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest,
31127da15baSAnders Carlsson                            E->arg_begin(), E->arg_end());
31227da15baSAnders Carlsson   }
313e11f9ce9SAnders Carlsson }
31427da15baSAnders Carlsson 
315aa4149a2SJohn McCall /// Check whether the given operator new[] is the global placement
316aa4149a2SJohn McCall /// operator new[].
317aa4149a2SJohn McCall static bool IsPlacementOperatorNewArray(ASTContext &Ctx,
318aa4149a2SJohn McCall                                         const FunctionDecl *Fn) {
319aa4149a2SJohn McCall   // Must be in global scope.  Note that allocation functions can't be
320aa4149a2SJohn McCall   // declared in namespaces.
32150c68258SSebastian Redl   if (!Fn->getDeclContext()->getRedeclContext()->isFileContext())
322aa4149a2SJohn McCall     return false;
323aa4149a2SJohn McCall 
324aa4149a2SJohn McCall   // Signature must be void *operator new[](size_t, void*).
325aa4149a2SJohn McCall   // The size_t is common to all operator new[]s.
326aa4149a2SJohn McCall   if (Fn->getNumParams() != 2)
327aa4149a2SJohn McCall     return false;
328aa4149a2SJohn McCall 
329aa4149a2SJohn McCall   CanQualType ParamType = Ctx.getCanonicalType(Fn->getParamDecl(1)->getType());
330aa4149a2SJohn McCall   return (ParamType == Ctx.VoidPtrTy);
331aa4149a2SJohn McCall }
332aa4149a2SJohn McCall 
3338ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF,
3348ed55a54SJohn McCall                                         const CXXNewExpr *E) {
33521122cf6SAnders Carlsson   if (!E->isArray())
3363eb55cfeSKen Dyck     return CharUnits::Zero();
33721122cf6SAnders Carlsson 
338399f499fSAnders Carlsson   // No cookie is required if the new operator being used is
339399f499fSAnders Carlsson   // ::operator new[](size_t, void*).
340399f499fSAnders Carlsson   const FunctionDecl *OperatorNew = E->getOperatorNew();
3418ed55a54SJohn McCall   if (IsPlacementOperatorNewArray(CGF.getContext(), OperatorNew))
3423eb55cfeSKen Dyck     return CharUnits::Zero();
343399f499fSAnders Carlsson 
3448ed55a54SJohn McCall   return CGF.CGM.getCXXABI().GetArrayCookieSize(E->getAllocatedType());
34559486a2dSAnders Carlsson }
34659486a2dSAnders Carlsson 
34747b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context,
34847b4629bSFariborz Jahanian                                         CodeGenFunction &CGF,
34959486a2dSAnders Carlsson                                         const CXXNewExpr *E,
35005fc5be3SDouglas Gregor                                         llvm::Value *&NumElements,
35105fc5be3SDouglas Gregor                                         llvm::Value *&SizeWithoutCookie) {
3527648fb46SArgyrios Kyrtzidis   QualType ElemType = E->getAllocatedType();
35359486a2dSAnders Carlsson 
3548ed55a54SJohn McCall   const llvm::IntegerType *SizeTy =
3558ed55a54SJohn McCall     cast<llvm::IntegerType>(CGF.ConvertType(CGF.getContext().getSizeType()));
3568ed55a54SJohn McCall 
3577648fb46SArgyrios Kyrtzidis   CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(ElemType);
3588ed55a54SJohn McCall 
3598ed55a54SJohn McCall   if (!E->isArray()) {
36005fc5be3SDouglas Gregor     SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
36105fc5be3SDouglas Gregor     return SizeWithoutCookie;
36205fc5be3SDouglas Gregor   }
36359486a2dSAnders Carlsson 
3648ed55a54SJohn McCall   // Figure out the cookie size.
3658ed55a54SJohn McCall   CharUnits CookieSize = CalculateCookiePadding(CGF, E);
3668ed55a54SJohn McCall 
36759486a2dSAnders Carlsson   // Emit the array size expression.
3687648fb46SArgyrios Kyrtzidis   // We multiply the size of all dimensions for NumElements.
3697648fb46SArgyrios Kyrtzidis   // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6.
37059486a2dSAnders Carlsson   NumElements = CGF.EmitScalarExpr(E->getArraySize());
3718ed55a54SJohn McCall   assert(NumElements->getType() == SizeTy && "element count not a size_t");
3728ed55a54SJohn McCall 
3738ed55a54SJohn McCall   uint64_t ArraySizeMultiplier = 1;
3747648fb46SArgyrios Kyrtzidis   while (const ConstantArrayType *CAT
3757648fb46SArgyrios Kyrtzidis              = CGF.getContext().getAsConstantArrayType(ElemType)) {
3767648fb46SArgyrios Kyrtzidis     ElemType = CAT->getElementType();
3778ed55a54SJohn McCall     ArraySizeMultiplier *= CAT->getSize().getZExtValue();
3787648fb46SArgyrios Kyrtzidis   }
37959486a2dSAnders Carlsson 
3808ed55a54SJohn McCall   llvm::Value *Size;
38132ac583dSChris Lattner 
38232ac583dSChris Lattner   // If someone is doing 'new int[42]' there is no need to do a dynamic check.
38332ac583dSChris Lattner   // Don't bloat the -O0 code.
38432ac583dSChris Lattner   if (llvm::ConstantInt *NumElementsC =
38532ac583dSChris Lattner         dyn_cast<llvm::ConstantInt>(NumElements)) {
38632ac583dSChris Lattner     llvm::APInt NEC = NumElementsC->getValue();
3878ed55a54SJohn McCall     unsigned SizeWidth = NEC.getBitWidth();
38832ac583dSChris Lattner 
3898ed55a54SJohn McCall     // Determine if there is an overflow here by doing an extended multiply.
3908ed55a54SJohn McCall     NEC.zext(SizeWidth*2);
3918ed55a54SJohn McCall     llvm::APInt SC(SizeWidth*2, TypeSize.getQuantity());
39232ac583dSChris Lattner     SC *= NEC;
39332ac583dSChris Lattner 
3948ed55a54SJohn McCall     if (!CookieSize.isZero()) {
3958ed55a54SJohn McCall       // Save the current size without a cookie.  We don't care if an
3968ed55a54SJohn McCall       // overflow's already happened because SizeWithoutCookie isn't
3978ed55a54SJohn McCall       // used if the allocator returns null or throws, as it should
3988ed55a54SJohn McCall       // always do on an overflow.
3998ed55a54SJohn McCall       llvm::APInt SWC = SC;
4008ed55a54SJohn McCall       SWC.trunc(SizeWidth);
4018ed55a54SJohn McCall       SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, SWC);
4028ed55a54SJohn McCall 
4038ed55a54SJohn McCall       // Add the cookie size.
4048ed55a54SJohn McCall       SC += llvm::APInt(SizeWidth*2, CookieSize.getQuantity());
4058ed55a54SJohn McCall     }
4068ed55a54SJohn McCall 
4078ed55a54SJohn McCall     if (SC.countLeadingZeros() >= SizeWidth) {
4088ed55a54SJohn McCall       SC.trunc(SizeWidth);
4098ed55a54SJohn McCall       Size = llvm::ConstantInt::get(SizeTy, SC);
41032ac583dSChris Lattner     } else {
41132ac583dSChris Lattner       // On overflow, produce a -1 so operator new throws.
4128ed55a54SJohn McCall       Size = llvm::Constant::getAllOnesValue(SizeTy);
41332ac583dSChris Lattner     }
41432ac583dSChris Lattner 
4158ed55a54SJohn McCall     // Scale NumElements while we're at it.
4168ed55a54SJohn McCall     uint64_t N = NEC.getZExtValue() * ArraySizeMultiplier;
4178ed55a54SJohn McCall     NumElements = llvm::ConstantInt::get(SizeTy, N);
41847b4629bSFariborz Jahanian 
4198ed55a54SJohn McCall   // Otherwise, we don't need to do an overflow-checked multiplication if
4208ed55a54SJohn McCall   // we're multiplying by one.
4218ed55a54SJohn McCall   } else if (TypeSize.isOne()) {
4228ed55a54SJohn McCall     assert(ArraySizeMultiplier == 1);
423f2f38701SChris Lattner 
4248ed55a54SJohn McCall     Size = NumElements;
425f2f38701SChris Lattner 
4268ed55a54SJohn McCall     // If we need a cookie, add its size in with an overflow check.
4278ed55a54SJohn McCall     // This is maybe a little paranoid.
4288ed55a54SJohn McCall     if (!CookieSize.isZero()) {
42905fc5be3SDouglas Gregor       SizeWithoutCookie = Size;
430f2f38701SChris Lattner 
4318ed55a54SJohn McCall       llvm::Value *CookieSizeV
4328ed55a54SJohn McCall         = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
4338ed55a54SJohn McCall 
4348ed55a54SJohn McCall       const llvm::Type *Types[] = { SizeTy };
4358ed55a54SJohn McCall       llvm::Value *UAddF
4368ed55a54SJohn McCall         = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1);
4378ed55a54SJohn McCall       llvm::Value *AddRes
4388ed55a54SJohn McCall         = CGF.Builder.CreateCall2(UAddF, Size, CookieSizeV);
4398ed55a54SJohn McCall 
4408ed55a54SJohn McCall       Size = CGF.Builder.CreateExtractValue(AddRes, 0);
4418ed55a54SJohn McCall       llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1);
4428ed55a54SJohn McCall       Size = CGF.Builder.CreateSelect(DidOverflow,
4438ed55a54SJohn McCall                                       llvm::ConstantInt::get(SizeTy, -1),
4448ed55a54SJohn McCall                                       Size);
4458ed55a54SJohn McCall     }
4468ed55a54SJohn McCall 
4478ed55a54SJohn McCall   // Otherwise use the int.umul.with.overflow intrinsic.
4488ed55a54SJohn McCall   } else {
4498ed55a54SJohn McCall     llvm::Value *OutermostElementSize
4508ed55a54SJohn McCall       = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity());
4518ed55a54SJohn McCall 
4528ed55a54SJohn McCall     llvm::Value *NumOutermostElements = NumElements;
4538ed55a54SJohn McCall 
4548ed55a54SJohn McCall     // Scale NumElements by the array size multiplier.  This might
4558ed55a54SJohn McCall     // overflow, but only if the multiplication below also overflows,
4568ed55a54SJohn McCall     // in which case this multiplication isn't used.
4578ed55a54SJohn McCall     if (ArraySizeMultiplier != 1)
4588ed55a54SJohn McCall       NumElements = CGF.Builder.CreateMul(NumElements,
4598ed55a54SJohn McCall                          llvm::ConstantInt::get(SizeTy, ArraySizeMultiplier));
4608ed55a54SJohn McCall 
4618ed55a54SJohn McCall     // The requested size of the outermost array is non-constant.
4628ed55a54SJohn McCall     // Multiply that by the static size of the elements of that array;
4638ed55a54SJohn McCall     // on unsigned overflow, set the size to -1 to trigger an
4648ed55a54SJohn McCall     // exception from the allocation routine.  This is sufficient to
4658ed55a54SJohn McCall     // prevent buffer overruns from the allocator returning a
4668ed55a54SJohn McCall     // seemingly valid pointer to insufficient space.  This idea comes
4678ed55a54SJohn McCall     // originally from MSVC, and GCC has an open bug requesting
4688ed55a54SJohn McCall     // similar behavior:
4698ed55a54SJohn McCall     //   http://gcc.gnu.org/bugzilla/show_bug.cgi?id=19351
4708ed55a54SJohn McCall     //
4718ed55a54SJohn McCall     // This will not be sufficient for C++0x, which requires a
4728ed55a54SJohn McCall     // specific exception class (std::bad_array_new_length).
4738ed55a54SJohn McCall     // That will require ABI support that has not yet been specified.
4748ed55a54SJohn McCall     const llvm::Type *Types[] = { SizeTy };
4758ed55a54SJohn McCall     llvm::Value *UMulF
4768ed55a54SJohn McCall       = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, Types, 1);
4778ed55a54SJohn McCall     llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumOutermostElements,
4788ed55a54SJohn McCall                                                   OutermostElementSize);
4798ed55a54SJohn McCall 
4808ed55a54SJohn McCall     // The overflow bit.
4818ed55a54SJohn McCall     llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1);
4828ed55a54SJohn McCall 
4838ed55a54SJohn McCall     // The result of the multiplication.
4848ed55a54SJohn McCall     Size = CGF.Builder.CreateExtractValue(MulRes, 0);
4858ed55a54SJohn McCall 
4868ed55a54SJohn McCall     // If we have a cookie, we need to add that size in, too.
4878ed55a54SJohn McCall     if (!CookieSize.isZero()) {
4888ed55a54SJohn McCall       SizeWithoutCookie = Size;
4898ed55a54SJohn McCall 
4908ed55a54SJohn McCall       llvm::Value *CookieSizeV
4918ed55a54SJohn McCall         = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
4928ed55a54SJohn McCall       llvm::Value *UAddF
4938ed55a54SJohn McCall         = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1);
4948ed55a54SJohn McCall       llvm::Value *AddRes
4958ed55a54SJohn McCall         = CGF.Builder.CreateCall2(UAddF, SizeWithoutCookie, CookieSizeV);
4968ed55a54SJohn McCall 
4978ed55a54SJohn McCall       Size = CGF.Builder.CreateExtractValue(AddRes, 0);
4988ed55a54SJohn McCall 
4998ed55a54SJohn McCall       llvm::Value *AddDidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1);
5008ed55a54SJohn McCall       DidOverflow = CGF.Builder.CreateAnd(DidOverflow, AddDidOverflow);
5018ed55a54SJohn McCall     }
5028ed55a54SJohn McCall 
5038ed55a54SJohn McCall     Size = CGF.Builder.CreateSelect(DidOverflow,
5048ed55a54SJohn McCall                                     llvm::ConstantInt::get(SizeTy, -1),
5058ed55a54SJohn McCall                                     Size);
5068ed55a54SJohn McCall   }
5078ed55a54SJohn McCall 
5088ed55a54SJohn McCall   if (CookieSize.isZero())
5098ed55a54SJohn McCall     SizeWithoutCookie = Size;
5108ed55a54SJohn McCall   else
5118ed55a54SJohn McCall     assert(SizeWithoutCookie && "didn't set SizeWithoutCookie?");
51259486a2dSAnders Carlsson 
51332ac583dSChris Lattner   return Size;
51459486a2dSAnders Carlsson }
51559486a2dSAnders Carlsson 
516d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E,
517d5202e09SFariborz Jahanian                                     llvm::Value *NewPtr) {
518d5202e09SFariborz Jahanian 
519d5202e09SFariborz Jahanian   assert(E->getNumConstructorArgs() == 1 &&
520d5202e09SFariborz Jahanian          "Can only have one argument to initializer of POD type.");
521d5202e09SFariborz Jahanian 
522d5202e09SFariborz Jahanian   const Expr *Init = E->getConstructorArg(0);
523d5202e09SFariborz Jahanian   QualType AllocType = E->getAllocatedType();
524d5202e09SFariborz Jahanian 
5250381634aSDaniel Dunbar   unsigned Alignment =
5260381634aSDaniel Dunbar     CGF.getContext().getTypeAlignInChars(AllocType).getQuantity();
527d5202e09SFariborz Jahanian   if (!CGF.hasAggregateLLVMType(AllocType))
528d5202e09SFariborz Jahanian     CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr,
5290381634aSDaniel Dunbar                           AllocType.isVolatileQualified(), Alignment,
5300381634aSDaniel Dunbar                           AllocType);
531d5202e09SFariborz Jahanian   else if (AllocType->isAnyComplexType())
532d5202e09SFariborz Jahanian     CGF.EmitComplexExprIntoAddr(Init, NewPtr,
533d5202e09SFariborz Jahanian                                 AllocType.isVolatileQualified());
534d5202e09SFariborz Jahanian   else
535d5202e09SFariborz Jahanian     CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified());
536d5202e09SFariborz Jahanian }
537d5202e09SFariborz Jahanian 
538d5202e09SFariborz Jahanian void
539d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E,
540d5202e09SFariborz Jahanian                                          llvm::Value *NewPtr,
541d5202e09SFariborz Jahanian                                          llvm::Value *NumElements) {
542b66b08efSFariborz Jahanian   // We have a POD type.
543b66b08efSFariborz Jahanian   if (E->getNumConstructorArgs() == 0)
544b66b08efSFariborz Jahanian     return;
545b66b08efSFariborz Jahanian 
546d5202e09SFariborz Jahanian   const llvm::Type *SizeTy = ConvertType(getContext().getSizeType());
547d5202e09SFariborz Jahanian 
548d5202e09SFariborz Jahanian   // Create a temporary for the loop index and initialize it with 0.
549d5202e09SFariborz Jahanian   llvm::Value *IndexPtr = CreateTempAlloca(SizeTy, "loop.index");
550d5202e09SFariborz Jahanian   llvm::Value *Zero = llvm::Constant::getNullValue(SizeTy);
551d5202e09SFariborz Jahanian   Builder.CreateStore(Zero, IndexPtr);
552d5202e09SFariborz Jahanian 
553d5202e09SFariborz Jahanian   // Start the loop with a block that tests the condition.
554d5202e09SFariborz Jahanian   llvm::BasicBlock *CondBlock = createBasicBlock("for.cond");
555d5202e09SFariborz Jahanian   llvm::BasicBlock *AfterFor = createBasicBlock("for.end");
556d5202e09SFariborz Jahanian 
557d5202e09SFariborz Jahanian   EmitBlock(CondBlock);
558d5202e09SFariborz Jahanian 
559d5202e09SFariborz Jahanian   llvm::BasicBlock *ForBody = createBasicBlock("for.body");
560d5202e09SFariborz Jahanian 
561d5202e09SFariborz Jahanian   // Generate: if (loop-index < number-of-elements fall to the loop body,
562d5202e09SFariborz Jahanian   // otherwise, go to the block after the for-loop.
563d5202e09SFariborz Jahanian   llvm::Value *Counter = Builder.CreateLoad(IndexPtr);
564d5202e09SFariborz Jahanian   llvm::Value *IsLess = Builder.CreateICmpULT(Counter, NumElements, "isless");
565d5202e09SFariborz Jahanian   // If the condition is true, execute the body.
566d5202e09SFariborz Jahanian   Builder.CreateCondBr(IsLess, ForBody, AfterFor);
567d5202e09SFariborz Jahanian 
568d5202e09SFariborz Jahanian   EmitBlock(ForBody);
569d5202e09SFariborz Jahanian 
570d5202e09SFariborz Jahanian   llvm::BasicBlock *ContinueBlock = createBasicBlock("for.inc");
571d5202e09SFariborz Jahanian   // Inside the loop body, emit the constructor call on the array element.
572d5202e09SFariborz Jahanian   Counter = Builder.CreateLoad(IndexPtr);
573d5202e09SFariborz Jahanian   llvm::Value *Address = Builder.CreateInBoundsGEP(NewPtr, Counter,
574d5202e09SFariborz Jahanian                                                    "arrayidx");
575d5202e09SFariborz Jahanian   StoreAnyExprIntoOneUnit(*this, E, Address);
576d5202e09SFariborz Jahanian 
577d5202e09SFariborz Jahanian   EmitBlock(ContinueBlock);
578d5202e09SFariborz Jahanian 
579d5202e09SFariborz Jahanian   // Emit the increment of the loop counter.
580d5202e09SFariborz Jahanian   llvm::Value *NextVal = llvm::ConstantInt::get(SizeTy, 1);
581d5202e09SFariborz Jahanian   Counter = Builder.CreateLoad(IndexPtr);
582d5202e09SFariborz Jahanian   NextVal = Builder.CreateAdd(Counter, NextVal, "inc");
583d5202e09SFariborz Jahanian   Builder.CreateStore(NextVal, IndexPtr);
584d5202e09SFariborz Jahanian 
585d5202e09SFariborz Jahanian   // Finally, branch back up to the condition for the next iteration.
586d5202e09SFariborz Jahanian   EmitBranch(CondBlock);
587d5202e09SFariborz Jahanian 
588d5202e09SFariborz Jahanian   // Emit the fall-through block.
589d5202e09SFariborz Jahanian   EmitBlock(AfterFor, true);
590d5202e09SFariborz Jahanian }
591d5202e09SFariborz Jahanian 
59205fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T,
59305fc5be3SDouglas Gregor                            llvm::Value *NewPtr, llvm::Value *Size) {
59405fc5be3SDouglas Gregor   llvm::LLVMContext &VMContext = CGF.CGM.getLLVMContext();
59505fc5be3SDouglas Gregor   const llvm::Type *BP = llvm::Type::getInt8PtrTy(VMContext);
59605fc5be3SDouglas Gregor   if (NewPtr->getType() != BP)
59705fc5be3SDouglas Gregor     NewPtr = CGF.Builder.CreateBitCast(NewPtr, BP, "tmp");
59805fc5be3SDouglas Gregor 
59905fc5be3SDouglas Gregor   CGF.Builder.CreateCall5(CGF.CGM.getMemSetFn(BP, CGF.IntPtrTy), NewPtr,
60005fc5be3SDouglas Gregor                 llvm::Constant::getNullValue(llvm::Type::getInt8Ty(VMContext)),
60105fc5be3SDouglas Gregor                           Size,
60205fc5be3SDouglas Gregor                     llvm::ConstantInt::get(CGF.Int32Ty,
60305fc5be3SDouglas Gregor                                            CGF.getContext().getTypeAlign(T)/8),
60405fc5be3SDouglas Gregor                           llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext),
60505fc5be3SDouglas Gregor                                                  0));
60605fc5be3SDouglas Gregor }
60705fc5be3SDouglas Gregor 
60859486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
60959486a2dSAnders Carlsson                                llvm::Value *NewPtr,
61005fc5be3SDouglas Gregor                                llvm::Value *NumElements,
61105fc5be3SDouglas Gregor                                llvm::Value *AllocSizeWithoutCookie) {
6123a202f60SAnders Carlsson   if (E->isArray()) {
613d040e6b2SAnders Carlsson     if (CXXConstructorDecl *Ctor = E->getConstructor()) {
61405fc5be3SDouglas Gregor       bool RequiresZeroInitialization = false;
61505fc5be3SDouglas Gregor       if (Ctor->getParent()->hasTrivialConstructor()) {
61605fc5be3SDouglas Gregor         // If new expression did not specify value-initialization, then there
61705fc5be3SDouglas Gregor         // is no initialization.
61805fc5be3SDouglas Gregor         if (!E->hasInitializer() || Ctor->getParent()->isEmpty())
61905fc5be3SDouglas Gregor           return;
62005fc5be3SDouglas Gregor 
621614dbdcdSJohn McCall         if (CGF.CGM.getTypes().isZeroInitializable(E->getAllocatedType())) {
62205fc5be3SDouglas Gregor           // Optimization: since zero initialization will just set the memory
62305fc5be3SDouglas Gregor           // to all zeroes, generate a single memset to do it in one shot.
62405fc5be3SDouglas Gregor           EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr,
62505fc5be3SDouglas Gregor                          AllocSizeWithoutCookie);
6263a202f60SAnders Carlsson           return;
6273a202f60SAnders Carlsson         }
62805fc5be3SDouglas Gregor 
62905fc5be3SDouglas Gregor         RequiresZeroInitialization = true;
63005fc5be3SDouglas Gregor       }
63105fc5be3SDouglas Gregor 
63205fc5be3SDouglas Gregor       CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr,
63305fc5be3SDouglas Gregor                                      E->constructor_arg_begin(),
63405fc5be3SDouglas Gregor                                      E->constructor_arg_end(),
63505fc5be3SDouglas Gregor                                      RequiresZeroInitialization);
63605fc5be3SDouglas Gregor       return;
63705fc5be3SDouglas Gregor     } else if (E->getNumConstructorArgs() == 1 &&
63805fc5be3SDouglas Gregor                isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) {
63905fc5be3SDouglas Gregor       // Optimization: since zero initialization will just set the memory
64005fc5be3SDouglas Gregor       // to all zeroes, generate a single memset to do it in one shot.
64105fc5be3SDouglas Gregor       EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr,
64205fc5be3SDouglas Gregor                      AllocSizeWithoutCookie);
64305fc5be3SDouglas Gregor       return;
64405fc5be3SDouglas Gregor     } else {
645d5202e09SFariborz Jahanian       CGF.EmitNewArrayInitializer(E, NewPtr, NumElements);
646d5202e09SFariborz Jahanian       return;
647d040e6b2SAnders Carlsson     }
648d5202e09SFariborz Jahanian   }
64959486a2dSAnders Carlsson 
65059486a2dSAnders Carlsson   if (CXXConstructorDecl *Ctor = E->getConstructor()) {
651747eb784SDouglas Gregor     // Per C++ [expr.new]p15, if we have an initializer, then we're performing
652747eb784SDouglas Gregor     // direct initialization. C++ [dcl.init]p5 requires that we
653747eb784SDouglas Gregor     // zero-initialize storage if there are no user-declared constructors.
654747eb784SDouglas Gregor     if (E->hasInitializer() &&
655747eb784SDouglas Gregor         !Ctor->getParent()->hasUserDeclaredConstructor() &&
656747eb784SDouglas Gregor         !Ctor->getParent()->isEmpty())
657747eb784SDouglas Gregor       CGF.EmitNullInitialization(NewPtr, E->getAllocatedType());
658747eb784SDouglas Gregor 
659e11f9ce9SAnders Carlsson     CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false,
660e11f9ce9SAnders Carlsson                                NewPtr, E->constructor_arg_begin(),
66159486a2dSAnders Carlsson                                E->constructor_arg_end());
66259486a2dSAnders Carlsson 
66359486a2dSAnders Carlsson     return;
66459486a2dSAnders Carlsson   }
665b66b08efSFariborz Jahanian   // We have a POD type.
666b66b08efSFariborz Jahanian   if (E->getNumConstructorArgs() == 0)
667b66b08efSFariborz Jahanian     return;
66859486a2dSAnders Carlsson 
669d5202e09SFariborz Jahanian   StoreAnyExprIntoOneUnit(CGF, E, NewPtr);
67059486a2dSAnders Carlsson }
67159486a2dSAnders Carlsson 
67259486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
67359486a2dSAnders Carlsson   QualType AllocType = E->getAllocatedType();
6748ed55a54SJohn McCall   if (AllocType->isArrayType())
6758ed55a54SJohn McCall     while (const ArrayType *AType = getContext().getAsArrayType(AllocType))
6768ed55a54SJohn McCall       AllocType = AType->getElementType();
6778ed55a54SJohn McCall 
67859486a2dSAnders Carlsson   FunctionDecl *NewFD = E->getOperatorNew();
67959486a2dSAnders Carlsson   const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>();
68059486a2dSAnders Carlsson 
68159486a2dSAnders Carlsson   CallArgList NewArgs;
68259486a2dSAnders Carlsson 
68359486a2dSAnders Carlsson   // The allocation size is the first argument.
68459486a2dSAnders Carlsson   QualType SizeTy = getContext().getSizeType();
68559486a2dSAnders Carlsson 
68659486a2dSAnders Carlsson   llvm::Value *NumElements = 0;
68705fc5be3SDouglas Gregor   llvm::Value *AllocSizeWithoutCookie = 0;
68847b4629bSFariborz Jahanian   llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(),
68905fc5be3SDouglas Gregor                                                *this, E, NumElements,
69005fc5be3SDouglas Gregor                                                AllocSizeWithoutCookie);
69159486a2dSAnders Carlsson 
69259486a2dSAnders Carlsson   NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy));
69359486a2dSAnders Carlsson 
69459486a2dSAnders Carlsson   // Emit the rest of the arguments.
69559486a2dSAnders Carlsson   // FIXME: Ideally, this should just use EmitCallArgs.
69659486a2dSAnders Carlsson   CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin();
69759486a2dSAnders Carlsson 
69859486a2dSAnders Carlsson   // First, use the types from the function type.
69959486a2dSAnders Carlsson   // We start at 1 here because the first argument (the allocation size)
70059486a2dSAnders Carlsson   // has already been emitted.
70159486a2dSAnders Carlsson   for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) {
70259486a2dSAnders Carlsson     QualType ArgType = NewFTy->getArgType(i);
70359486a2dSAnders Carlsson 
70459486a2dSAnders Carlsson     assert(getContext().getCanonicalType(ArgType.getNonReferenceType()).
70559486a2dSAnders Carlsson            getTypePtr() ==
70659486a2dSAnders Carlsson            getContext().getCanonicalType(NewArg->getType()).getTypePtr() &&
70759486a2dSAnders Carlsson            "type mismatch in call argument!");
70859486a2dSAnders Carlsson 
70959486a2dSAnders Carlsson     NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
71059486a2dSAnders Carlsson                                      ArgType));
71159486a2dSAnders Carlsson 
71259486a2dSAnders Carlsson   }
71359486a2dSAnders Carlsson 
71459486a2dSAnders Carlsson   // Either we've emitted all the call args, or we have a call to a
71559486a2dSAnders Carlsson   // variadic function.
71659486a2dSAnders Carlsson   assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) &&
71759486a2dSAnders Carlsson          "Extra arguments in non-variadic function!");
71859486a2dSAnders Carlsson 
71959486a2dSAnders Carlsson   // If we still have any arguments, emit them using the type of the argument.
72059486a2dSAnders Carlsson   for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end();
72159486a2dSAnders Carlsson        NewArg != NewArgEnd; ++NewArg) {
72259486a2dSAnders Carlsson     QualType ArgType = NewArg->getType();
72359486a2dSAnders Carlsson     NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType),
72459486a2dSAnders Carlsson                                      ArgType));
72559486a2dSAnders Carlsson   }
72659486a2dSAnders Carlsson 
72759486a2dSAnders Carlsson   // Emit the call to new.
72859486a2dSAnders Carlsson   RValue RV =
729ab26cfa5SJohn McCall     EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy),
73061a401caSAnders Carlsson              CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD);
73159486a2dSAnders Carlsson 
73259486a2dSAnders Carlsson   // If an allocation function is declared with an empty exception specification
73359486a2dSAnders Carlsson   // it returns null to indicate failure to allocate storage. [expr.new]p13.
73459486a2dSAnders Carlsson   // (We don't need to check for null when there's no new initializer and
73559486a2dSAnders Carlsson   // we're allocating a POD type).
73659486a2dSAnders Carlsson   bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() &&
73759486a2dSAnders Carlsson     !(AllocType->isPODType() && !E->hasInitializer());
73859486a2dSAnders Carlsson 
7398ed55a54SJohn McCall   llvm::BasicBlock *NullCheckSource = 0;
74059486a2dSAnders Carlsson   llvm::BasicBlock *NewNotNull = 0;
74159486a2dSAnders Carlsson   llvm::BasicBlock *NewEnd = 0;
74259486a2dSAnders Carlsson 
74359486a2dSAnders Carlsson   llvm::Value *NewPtr = RV.getScalarVal();
7448ed55a54SJohn McCall   unsigned AS = cast<llvm::PointerType>(NewPtr->getType())->getAddressSpace();
74559486a2dSAnders Carlsson 
74659486a2dSAnders Carlsson   if (NullCheckResult) {
7478ed55a54SJohn McCall     NullCheckSource = Builder.GetInsertBlock();
74859486a2dSAnders Carlsson     NewNotNull = createBasicBlock("new.notnull");
74959486a2dSAnders Carlsson     NewEnd = createBasicBlock("new.end");
75059486a2dSAnders Carlsson 
7518ed55a54SJohn McCall     llvm::Value *IsNull = Builder.CreateIsNull(NewPtr, "new.isnull");
7528ed55a54SJohn McCall     Builder.CreateCondBr(IsNull, NewEnd, NewNotNull);
75359486a2dSAnders Carlsson     EmitBlock(NewNotNull);
75459486a2dSAnders Carlsson   }
75559486a2dSAnders Carlsson 
7568ed55a54SJohn McCall   assert((AllocSize == AllocSizeWithoutCookie) ==
7578ed55a54SJohn McCall          CalculateCookiePadding(*this, E).isZero());
7588ed55a54SJohn McCall   if (AllocSize != AllocSizeWithoutCookie) {
7598ed55a54SJohn McCall     assert(E->isArray());
7608ed55a54SJohn McCall     NewPtr = CGM.getCXXABI().InitializeArrayCookie(CGF, NewPtr, NumElements,
7618ed55a54SJohn McCall                                                    AllocType);
76259486a2dSAnders Carlsson   }
76359486a2dSAnders Carlsson 
764040ad500SDouglas Gregor   const llvm::Type *ElementPtrTy
765040ad500SDouglas Gregor     = ConvertTypeForMem(AllocType)->getPointerTo(AS);
7668ed55a54SJohn McCall   NewPtr = Builder.CreateBitCast(NewPtr, ElementPtrTy);
7678ed55a54SJohn McCall   if (E->isArray()) {
76805fc5be3SDouglas Gregor     EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie);
7698ed55a54SJohn McCall 
7708ed55a54SJohn McCall     // NewPtr is a pointer to the base element type.  If we're
7718ed55a54SJohn McCall     // allocating an array of arrays, we'll need to cast back to the
7728ed55a54SJohn McCall     // array pointer type.
773040ad500SDouglas Gregor     const llvm::Type *ResultTy = ConvertTypeForMem(E->getType());
7748ed55a54SJohn McCall     if (NewPtr->getType() != ResultTy)
7758ed55a54SJohn McCall       NewPtr = Builder.CreateBitCast(NewPtr, ResultTy);
7768ed55a54SJohn McCall   } else {
77705fc5be3SDouglas Gregor     EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie);
77847b4629bSFariborz Jahanian   }
77959486a2dSAnders Carlsson 
78059486a2dSAnders Carlsson   if (NullCheckResult) {
78159486a2dSAnders Carlsson     Builder.CreateBr(NewEnd);
7828ed55a54SJohn McCall     llvm::BasicBlock *NotNullSource = Builder.GetInsertBlock();
78359486a2dSAnders Carlsson     EmitBlock(NewEnd);
78459486a2dSAnders Carlsson 
78559486a2dSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType());
78659486a2dSAnders Carlsson     PHI->reserveOperandSpace(2);
7878ed55a54SJohn McCall     PHI->addIncoming(NewPtr, NotNullSource);
7888ed55a54SJohn McCall     PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()),
7898ed55a54SJohn McCall                      NullCheckSource);
79059486a2dSAnders Carlsson 
79159486a2dSAnders Carlsson     NewPtr = PHI;
79259486a2dSAnders Carlsson   }
79359486a2dSAnders Carlsson 
79459486a2dSAnders Carlsson   return NewPtr;
79559486a2dSAnders Carlsson }
79659486a2dSAnders Carlsson 
79759486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
79859486a2dSAnders Carlsson                                      llvm::Value *Ptr,
79959486a2dSAnders Carlsson                                      QualType DeleteTy) {
8008ed55a54SJohn McCall   assert(DeleteFD->getOverloadedOperator() == OO_Delete);
8018ed55a54SJohn McCall 
80259486a2dSAnders Carlsson   const FunctionProtoType *DeleteFTy =
80359486a2dSAnders Carlsson     DeleteFD->getType()->getAs<FunctionProtoType>();
80459486a2dSAnders Carlsson 
80559486a2dSAnders Carlsson   CallArgList DeleteArgs;
80659486a2dSAnders Carlsson 
80721122cf6SAnders Carlsson   // Check if we need to pass the size to the delete operator.
80821122cf6SAnders Carlsson   llvm::Value *Size = 0;
80921122cf6SAnders Carlsson   QualType SizeTy;
81021122cf6SAnders Carlsson   if (DeleteFTy->getNumArgs() == 2) {
81121122cf6SAnders Carlsson     SizeTy = DeleteFTy->getArgType(1);
8127df3cbebSKen Dyck     CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
8137df3cbebSKen Dyck     Size = llvm::ConstantInt::get(ConvertType(SizeTy),
8147df3cbebSKen Dyck                                   DeleteTypeSize.getQuantity());
81521122cf6SAnders Carlsson   }
81621122cf6SAnders Carlsson 
81759486a2dSAnders Carlsson   QualType ArgTy = DeleteFTy->getArgType(0);
81859486a2dSAnders Carlsson   llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
81959486a2dSAnders Carlsson   DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy));
82059486a2dSAnders Carlsson 
82121122cf6SAnders Carlsson   if (Size)
82259486a2dSAnders Carlsson     DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy));
82359486a2dSAnders Carlsson 
82459486a2dSAnders Carlsson   // Emit the call to delete.
825ab26cfa5SJohn McCall   EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy),
82661a401caSAnders Carlsson            CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(),
82759486a2dSAnders Carlsson            DeleteArgs, DeleteFD);
82859486a2dSAnders Carlsson }
82959486a2dSAnders Carlsson 
8308ed55a54SJohn McCall namespace {
8318ed55a54SJohn McCall   /// Calls the given 'operator delete' on a single object.
8328ed55a54SJohn McCall   struct CallObjectDelete : EHScopeStack::Cleanup {
8338ed55a54SJohn McCall     llvm::Value *Ptr;
8348ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
8358ed55a54SJohn McCall     QualType ElementType;
8368ed55a54SJohn McCall 
8378ed55a54SJohn McCall     CallObjectDelete(llvm::Value *Ptr,
8388ed55a54SJohn McCall                      const FunctionDecl *OperatorDelete,
8398ed55a54SJohn McCall                      QualType ElementType)
8408ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {}
8418ed55a54SJohn McCall 
8428ed55a54SJohn McCall     void Emit(CodeGenFunction &CGF, bool IsForEH) {
8438ed55a54SJohn McCall       CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType);
8448ed55a54SJohn McCall     }
8458ed55a54SJohn McCall   };
8468ed55a54SJohn McCall }
8478ed55a54SJohn McCall 
8488ed55a54SJohn McCall /// Emit the code for deleting a single object.
8498ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF,
8508ed55a54SJohn McCall                              const FunctionDecl *OperatorDelete,
8518ed55a54SJohn McCall                              llvm::Value *Ptr,
8528ed55a54SJohn McCall                              QualType ElementType) {
8538ed55a54SJohn McCall   // Find the destructor for the type, if applicable.  If the
8548ed55a54SJohn McCall   // destructor is virtual, we'll just emit the vcall and return.
8558ed55a54SJohn McCall   const CXXDestructorDecl *Dtor = 0;
8568ed55a54SJohn McCall   if (const RecordType *RT = ElementType->getAs<RecordType>()) {
8578ed55a54SJohn McCall     CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
8588ed55a54SJohn McCall     if (!RD->hasTrivialDestructor()) {
8598ed55a54SJohn McCall       Dtor = RD->getDestructor();
8608ed55a54SJohn McCall 
8618ed55a54SJohn McCall       if (Dtor->isVirtual()) {
8628ed55a54SJohn McCall         const llvm::Type *Ty =
863*0d635f53SJohn McCall           CGF.getTypes().GetFunctionType(CGF.getTypes().getFunctionInfo(Dtor,
864*0d635f53SJohn McCall                                                                Dtor_Complete),
8658ed55a54SJohn McCall                                          /*isVariadic=*/false);
8668ed55a54SJohn McCall 
8678ed55a54SJohn McCall         llvm::Value *Callee
8688ed55a54SJohn McCall           = CGF.BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty);
8698ed55a54SJohn McCall         CGF.EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0,
8708ed55a54SJohn McCall                               0, 0);
8718ed55a54SJohn McCall 
8728ed55a54SJohn McCall         // The dtor took care of deleting the object.
8738ed55a54SJohn McCall         return;
8748ed55a54SJohn McCall       }
8758ed55a54SJohn McCall     }
8768ed55a54SJohn McCall   }
8778ed55a54SJohn McCall 
8788ed55a54SJohn McCall   // Make sure that we call delete even if the dtor throws.
8798ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
8808ed55a54SJohn McCall                                             Ptr, OperatorDelete, ElementType);
8818ed55a54SJohn McCall 
8828ed55a54SJohn McCall   if (Dtor)
8838ed55a54SJohn McCall     CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete,
8848ed55a54SJohn McCall                               /*ForVirtualBase=*/false, Ptr);
8858ed55a54SJohn McCall 
8868ed55a54SJohn McCall   CGF.PopCleanupBlock();
8878ed55a54SJohn McCall }
8888ed55a54SJohn McCall 
8898ed55a54SJohn McCall namespace {
8908ed55a54SJohn McCall   /// Calls the given 'operator delete' on an array of objects.
8918ed55a54SJohn McCall   struct CallArrayDelete : EHScopeStack::Cleanup {
8928ed55a54SJohn McCall     llvm::Value *Ptr;
8938ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
8948ed55a54SJohn McCall     llvm::Value *NumElements;
8958ed55a54SJohn McCall     QualType ElementType;
8968ed55a54SJohn McCall     CharUnits CookieSize;
8978ed55a54SJohn McCall 
8988ed55a54SJohn McCall     CallArrayDelete(llvm::Value *Ptr,
8998ed55a54SJohn McCall                     const FunctionDecl *OperatorDelete,
9008ed55a54SJohn McCall                     llvm::Value *NumElements,
9018ed55a54SJohn McCall                     QualType ElementType,
9028ed55a54SJohn McCall                     CharUnits CookieSize)
9038ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements),
9048ed55a54SJohn McCall         ElementType(ElementType), CookieSize(CookieSize) {}
9058ed55a54SJohn McCall 
9068ed55a54SJohn McCall     void Emit(CodeGenFunction &CGF, bool IsForEH) {
9078ed55a54SJohn McCall       const FunctionProtoType *DeleteFTy =
9088ed55a54SJohn McCall         OperatorDelete->getType()->getAs<FunctionProtoType>();
9098ed55a54SJohn McCall       assert(DeleteFTy->getNumArgs() == 1 || DeleteFTy->getNumArgs() == 2);
9108ed55a54SJohn McCall 
9118ed55a54SJohn McCall       CallArgList Args;
9128ed55a54SJohn McCall 
9138ed55a54SJohn McCall       // Pass the pointer as the first argument.
9148ed55a54SJohn McCall       QualType VoidPtrTy = DeleteFTy->getArgType(0);
9158ed55a54SJohn McCall       llvm::Value *DeletePtr
9168ed55a54SJohn McCall         = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy));
9178ed55a54SJohn McCall       Args.push_back(std::make_pair(RValue::get(DeletePtr), VoidPtrTy));
9188ed55a54SJohn McCall 
9198ed55a54SJohn McCall       // Pass the original requested size as the second argument.
9208ed55a54SJohn McCall       if (DeleteFTy->getNumArgs() == 2) {
9218ed55a54SJohn McCall         QualType size_t = DeleteFTy->getArgType(1);
9228ed55a54SJohn McCall         const llvm::IntegerType *SizeTy
9238ed55a54SJohn McCall           = cast<llvm::IntegerType>(CGF.ConvertType(size_t));
9248ed55a54SJohn McCall 
9258ed55a54SJohn McCall         CharUnits ElementTypeSize =
9268ed55a54SJohn McCall           CGF.CGM.getContext().getTypeSizeInChars(ElementType);
9278ed55a54SJohn McCall 
9288ed55a54SJohn McCall         // The size of an element, multiplied by the number of elements.
9298ed55a54SJohn McCall         llvm::Value *Size
9308ed55a54SJohn McCall           = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity());
9318ed55a54SJohn McCall         Size = CGF.Builder.CreateMul(Size, NumElements);
9328ed55a54SJohn McCall 
9338ed55a54SJohn McCall         // Plus the size of the cookie if applicable.
9348ed55a54SJohn McCall         if (!CookieSize.isZero()) {
9358ed55a54SJohn McCall           llvm::Value *CookieSizeV
9368ed55a54SJohn McCall             = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
9378ed55a54SJohn McCall           Size = CGF.Builder.CreateAdd(Size, CookieSizeV);
9388ed55a54SJohn McCall         }
9398ed55a54SJohn McCall 
9408ed55a54SJohn McCall         Args.push_back(std::make_pair(RValue::get(Size), size_t));
9418ed55a54SJohn McCall       }
9428ed55a54SJohn McCall 
9438ed55a54SJohn McCall       // Emit the call to delete.
9448ed55a54SJohn McCall       CGF.EmitCall(CGF.getTypes().getFunctionInfo(Args, DeleteFTy),
9458ed55a54SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
9468ed55a54SJohn McCall                    ReturnValueSlot(), Args, OperatorDelete);
9478ed55a54SJohn McCall     }
9488ed55a54SJohn McCall   };
9498ed55a54SJohn McCall }
9508ed55a54SJohn McCall 
9518ed55a54SJohn McCall /// Emit the code for deleting an array of objects.
9528ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF,
9538ed55a54SJohn McCall                             const FunctionDecl *OperatorDelete,
9548ed55a54SJohn McCall                             llvm::Value *Ptr,
9558ed55a54SJohn McCall                             QualType ElementType) {
9568ed55a54SJohn McCall   llvm::Value *NumElements = 0;
9578ed55a54SJohn McCall   llvm::Value *AllocatedPtr = 0;
9588ed55a54SJohn McCall   CharUnits CookieSize;
9598ed55a54SJohn McCall   CGF.CGM.getCXXABI().ReadArrayCookie(CGF, Ptr, ElementType,
9608ed55a54SJohn McCall                                       NumElements, AllocatedPtr, CookieSize);
9618ed55a54SJohn McCall 
9628ed55a54SJohn McCall   assert(AllocatedPtr && "ReadArrayCookie didn't set AllocatedPtr");
9638ed55a54SJohn McCall 
9648ed55a54SJohn McCall   // Make sure that we call delete even if one of the dtors throws.
9658ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup,
9668ed55a54SJohn McCall                                            AllocatedPtr, OperatorDelete,
9678ed55a54SJohn McCall                                            NumElements, ElementType,
9688ed55a54SJohn McCall                                            CookieSize);
9698ed55a54SJohn McCall 
9708ed55a54SJohn McCall   if (const CXXRecordDecl *RD = ElementType->getAsCXXRecordDecl()) {
9718ed55a54SJohn McCall     if (!RD->hasTrivialDestructor()) {
9728ed55a54SJohn McCall       assert(NumElements && "ReadArrayCookie didn't find element count"
9738ed55a54SJohn McCall                             " for a class with destructor");
9748ed55a54SJohn McCall       CGF.EmitCXXAggrDestructorCall(RD->getDestructor(), NumElements, Ptr);
9758ed55a54SJohn McCall     }
9768ed55a54SJohn McCall   }
9778ed55a54SJohn McCall 
9788ed55a54SJohn McCall   CGF.PopCleanupBlock();
9798ed55a54SJohn McCall }
9808ed55a54SJohn McCall 
98159486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
98259486a2dSAnders Carlsson 
98359486a2dSAnders Carlsson   // Get at the argument before we performed the implicit conversion
98459486a2dSAnders Carlsson   // to void*.
98559486a2dSAnders Carlsson   const Expr *Arg = E->getArgument();
98659486a2dSAnders Carlsson   while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) {
987e302792bSJohn McCall     if (ICE->getCastKind() != CK_UserDefinedConversion &&
98859486a2dSAnders Carlsson         ICE->getType()->isVoidPointerType())
98959486a2dSAnders Carlsson       Arg = ICE->getSubExpr();
99059486a2dSAnders Carlsson     else
99159486a2dSAnders Carlsson       break;
99259486a2dSAnders Carlsson   }
99359486a2dSAnders Carlsson 
99459486a2dSAnders Carlsson   llvm::Value *Ptr = EmitScalarExpr(Arg);
99559486a2dSAnders Carlsson 
99659486a2dSAnders Carlsson   // Null check the pointer.
99759486a2dSAnders Carlsson   llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
99859486a2dSAnders Carlsson   llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
99959486a2dSAnders Carlsson 
100059486a2dSAnders Carlsson   llvm::Value *IsNull =
100159486a2dSAnders Carlsson     Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()),
100259486a2dSAnders Carlsson                          "isnull");
100359486a2dSAnders Carlsson 
100459486a2dSAnders Carlsson   Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
100559486a2dSAnders Carlsson   EmitBlock(DeleteNotNull);
100659486a2dSAnders Carlsson 
10078ed55a54SJohn McCall   // We might be deleting a pointer to array.  If so, GEP down to the
10088ed55a54SJohn McCall   // first non-array element.
10098ed55a54SJohn McCall   // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*)
10108ed55a54SJohn McCall   QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
10118ed55a54SJohn McCall   if (DeleteTy->isConstantArrayType()) {
10128ed55a54SJohn McCall     llvm::Value *Zero = Builder.getInt32(0);
10138ed55a54SJohn McCall     llvm::SmallVector<llvm::Value*,8> GEP;
101459486a2dSAnders Carlsson 
10158ed55a54SJohn McCall     GEP.push_back(Zero); // point at the outermost array
10168ed55a54SJohn McCall 
10178ed55a54SJohn McCall     // For each layer of array type we're pointing at:
10188ed55a54SJohn McCall     while (const ConstantArrayType *Arr
10198ed55a54SJohn McCall              = getContext().getAsConstantArrayType(DeleteTy)) {
10208ed55a54SJohn McCall       // 1. Unpeel the array type.
10218ed55a54SJohn McCall       DeleteTy = Arr->getElementType();
10228ed55a54SJohn McCall 
10238ed55a54SJohn McCall       // 2. GEP to the first element of the array.
10248ed55a54SJohn McCall       GEP.push_back(Zero);
10258ed55a54SJohn McCall     }
10268ed55a54SJohn McCall 
10278ed55a54SJohn McCall     Ptr = Builder.CreateInBoundsGEP(Ptr, GEP.begin(), GEP.end(), "del.first");
10288ed55a54SJohn McCall   }
10298ed55a54SJohn McCall 
103004f36218SDouglas Gregor   assert(ConvertTypeForMem(DeleteTy) ==
103104f36218SDouglas Gregor          cast<llvm::PointerType>(Ptr->getType())->getElementType());
10328ed55a54SJohn McCall 
103359486a2dSAnders Carlsson   if (E->isArrayForm()) {
10348ed55a54SJohn McCall     EmitArrayDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy);
10358ed55a54SJohn McCall   } else {
10368ed55a54SJohn McCall     EmitObjectDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy);
103759486a2dSAnders Carlsson   }
103859486a2dSAnders Carlsson 
103959486a2dSAnders Carlsson   EmitBlock(DeleteEnd);
104059486a2dSAnders Carlsson }
104159486a2dSAnders Carlsson 
104259486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
104359486a2dSAnders Carlsson   QualType Ty = E->getType();
104459486a2dSAnders Carlsson   const llvm::Type *LTy = ConvertType(Ty)->getPointerTo();
1045fd7dfeb7SAnders Carlsson 
10463f4336cbSAnders Carlsson   if (E->isTypeOperand()) {
10473f4336cbSAnders Carlsson     llvm::Constant *TypeInfo =
10483f4336cbSAnders Carlsson       CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand());
10493f4336cbSAnders Carlsson     return Builder.CreateBitCast(TypeInfo, LTy);
10503f4336cbSAnders Carlsson   }
1051fd7dfeb7SAnders Carlsson 
105259486a2dSAnders Carlsson   Expr *subE = E->getExprOperand();
105359486a2dSAnders Carlsson   Ty = subE->getType();
105459486a2dSAnders Carlsson   CanQualType CanTy = CGM.getContext().getCanonicalType(Ty);
105559486a2dSAnders Carlsson   Ty = CanTy.getUnqualifiedType().getNonReferenceType();
105659486a2dSAnders Carlsson   if (const RecordType *RT = Ty->getAs<RecordType>()) {
105759486a2dSAnders Carlsson     const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
105859486a2dSAnders Carlsson     if (RD->isPolymorphic()) {
105959486a2dSAnders Carlsson       // FIXME: if subE is an lvalue do
106059486a2dSAnders Carlsson       LValue Obj = EmitLValue(subE);
106159486a2dSAnders Carlsson       llvm::Value *This = Obj.getAddress();
106259486a2dSAnders Carlsson       LTy = LTy->getPointerTo()->getPointerTo();
106359486a2dSAnders Carlsson       llvm::Value *V = Builder.CreateBitCast(This, LTy);
106459486a2dSAnders Carlsson       // We need to do a zero check for *p, unless it has NonNullAttr.
106559486a2dSAnders Carlsson       // FIXME: PointerType->hasAttr<NonNullAttr>()
106659486a2dSAnders Carlsson       bool CanBeZero = false;
106759486a2dSAnders Carlsson       if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens()))
1068e302792bSJohn McCall         if (UO->getOpcode() == UO_Deref)
106959486a2dSAnders Carlsson           CanBeZero = true;
107059486a2dSAnders Carlsson       if (CanBeZero) {
107159486a2dSAnders Carlsson         llvm::BasicBlock *NonZeroBlock = createBasicBlock();
107259486a2dSAnders Carlsson         llvm::BasicBlock *ZeroBlock = createBasicBlock();
107359486a2dSAnders Carlsson 
107459486a2dSAnders Carlsson         llvm::Value *Zero = llvm::Constant::getNullValue(LTy);
107559486a2dSAnders Carlsson         Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero),
107659486a2dSAnders Carlsson                              NonZeroBlock, ZeroBlock);
107759486a2dSAnders Carlsson         EmitBlock(ZeroBlock);
107859486a2dSAnders Carlsson         /// Call __cxa_bad_typeid
107959486a2dSAnders Carlsson         const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext);
108059486a2dSAnders Carlsson         const llvm::FunctionType *FTy;
108159486a2dSAnders Carlsson         FTy = llvm::FunctionType::get(ResultType, false);
108259486a2dSAnders Carlsson         llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid");
108359486a2dSAnders Carlsson         Builder.CreateCall(F)->setDoesNotReturn();
108459486a2dSAnders Carlsson         Builder.CreateUnreachable();
108559486a2dSAnders Carlsson         EmitBlock(NonZeroBlock);
108659486a2dSAnders Carlsson       }
108759486a2dSAnders Carlsson       V = Builder.CreateLoad(V, "vtable");
108859486a2dSAnders Carlsson       V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL);
108959486a2dSAnders Carlsson       V = Builder.CreateLoad(V);
109059486a2dSAnders Carlsson       return V;
109159486a2dSAnders Carlsson     }
109259486a2dSAnders Carlsson   }
10933f4336cbSAnders Carlsson   return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy);
109459486a2dSAnders Carlsson }
109559486a2dSAnders Carlsson 
109659486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V,
109759486a2dSAnders Carlsson                                               const CXXDynamicCastExpr *DCE) {
10983f4336cbSAnders Carlsson   QualType SrcTy = DCE->getSubExpr()->getType();
10993f4336cbSAnders Carlsson   QualType DestTy = DCE->getTypeAsWritten();
11003f4336cbSAnders Carlsson   QualType InnerType = DestTy->getPointeeType();
11013f4336cbSAnders Carlsson 
110259486a2dSAnders Carlsson   const llvm::Type *LTy = ConvertType(DCE->getType());
110359486a2dSAnders Carlsson 
110459486a2dSAnders Carlsson   bool CanBeZero = false;
110559486a2dSAnders Carlsson   bool ToVoid = false;
110659486a2dSAnders Carlsson   bool ThrowOnBad = false;
11073f4336cbSAnders Carlsson   if (DestTy->isPointerType()) {
110859486a2dSAnders Carlsson     // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this
110959486a2dSAnders Carlsson     CanBeZero = true;
111059486a2dSAnders Carlsson     if (InnerType->isVoidType())
111159486a2dSAnders Carlsson       ToVoid = true;
111259486a2dSAnders Carlsson   } else {
111359486a2dSAnders Carlsson     LTy = LTy->getPointerTo();
1114fa8b4955SDouglas Gregor 
1115fa8b4955SDouglas Gregor     // FIXME: What if exceptions are disabled?
111659486a2dSAnders Carlsson     ThrowOnBad = true;
111759486a2dSAnders Carlsson   }
111859486a2dSAnders Carlsson 
11193f4336cbSAnders Carlsson   if (SrcTy->isPointerType() || SrcTy->isReferenceType())
11203f4336cbSAnders Carlsson     SrcTy = SrcTy->getPointeeType();
11213f4336cbSAnders Carlsson   SrcTy = SrcTy.getUnqualifiedType();
11223f4336cbSAnders Carlsson 
11230087bc85SAnders Carlsson   if (DestTy->isPointerType() || DestTy->isReferenceType())
11243f4336cbSAnders Carlsson     DestTy = DestTy->getPointeeType();
11253f4336cbSAnders Carlsson   DestTy = DestTy.getUnqualifiedType();
112659486a2dSAnders Carlsson 
112759486a2dSAnders Carlsson   llvm::BasicBlock *ContBlock = createBasicBlock();
112859486a2dSAnders Carlsson   llvm::BasicBlock *NullBlock = 0;
112959486a2dSAnders Carlsson   llvm::BasicBlock *NonZeroBlock = 0;
113059486a2dSAnders Carlsson   if (CanBeZero) {
113159486a2dSAnders Carlsson     NonZeroBlock = createBasicBlock();
113259486a2dSAnders Carlsson     NullBlock = createBasicBlock();
11333f4336cbSAnders Carlsson     Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock);
113459486a2dSAnders Carlsson     EmitBlock(NonZeroBlock);
113559486a2dSAnders Carlsson   }
113659486a2dSAnders Carlsson 
113759486a2dSAnders Carlsson   llvm::BasicBlock *BadCastBlock = 0;
113859486a2dSAnders Carlsson 
11393f4336cbSAnders Carlsson   const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType());
114059486a2dSAnders Carlsson 
114159486a2dSAnders Carlsson   // See if this is a dynamic_cast(void*)
114259486a2dSAnders Carlsson   if (ToVoid) {
114359486a2dSAnders Carlsson     llvm::Value *This = V;
114459486a2dSAnders Carlsson     V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo());
114559486a2dSAnders Carlsson     V = Builder.CreateLoad(V, "vtable");
114659486a2dSAnders Carlsson     V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL);
114759486a2dSAnders Carlsson     V = Builder.CreateLoad(V, "offset to top");
114859486a2dSAnders Carlsson     This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext));
114959486a2dSAnders Carlsson     V = Builder.CreateInBoundsGEP(This, V);
115059486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, LTy);
115159486a2dSAnders Carlsson   } else {
115259486a2dSAnders Carlsson     /// Call __dynamic_cast
115359486a2dSAnders Carlsson     const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext);
115459486a2dSAnders Carlsson     const llvm::FunctionType *FTy;
115559486a2dSAnders Carlsson     std::vector<const llvm::Type*> ArgTys;
115659486a2dSAnders Carlsson     const llvm::Type *PtrToInt8Ty
115759486a2dSAnders Carlsson       = llvm::Type::getInt8Ty(VMContext)->getPointerTo();
115859486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
115959486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
116059486a2dSAnders Carlsson     ArgTys.push_back(PtrToInt8Ty);
116159486a2dSAnders Carlsson     ArgTys.push_back(PtrDiffTy);
116259486a2dSAnders Carlsson     FTy = llvm::FunctionType::get(ResultType, ArgTys, false);
116359486a2dSAnders Carlsson 
116459486a2dSAnders Carlsson     // FIXME: Calculate better hint.
116559486a2dSAnders Carlsson     llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL);
11663f4336cbSAnders Carlsson 
11673f4336cbSAnders Carlsson     assert(SrcTy->isRecordType() && "Src type must be record type!");
11683f4336cbSAnders Carlsson     assert(DestTy->isRecordType() && "Dest type must be record type!");
11693f4336cbSAnders Carlsson 
1170247894b3SDouglas Gregor     llvm::Value *SrcArg
1171247894b3SDouglas Gregor       = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType());
1172247894b3SDouglas Gregor     llvm::Value *DestArg
1173247894b3SDouglas Gregor       = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType());
11743f4336cbSAnders Carlsson 
117559486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, PtrToInt8Ty);
117659486a2dSAnders Carlsson     V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"),
11773f4336cbSAnders Carlsson                             V, SrcArg, DestArg, hint);
117859486a2dSAnders Carlsson     V = Builder.CreateBitCast(V, LTy);
117959486a2dSAnders Carlsson 
118059486a2dSAnders Carlsson     if (ThrowOnBad) {
118159486a2dSAnders Carlsson       BadCastBlock = createBasicBlock();
11823f4336cbSAnders Carlsson       Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock);
118359486a2dSAnders Carlsson       EmitBlock(BadCastBlock);
1184fa8b4955SDouglas Gregor       /// Invoke __cxa_bad_cast
118559486a2dSAnders Carlsson       ResultType = llvm::Type::getVoidTy(VMContext);
118659486a2dSAnders Carlsson       const llvm::FunctionType *FBadTy;
118759486a2dSAnders Carlsson       FBadTy = llvm::FunctionType::get(ResultType, false);
118859486a2dSAnders Carlsson       llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast");
1189fa8b4955SDouglas Gregor       if (llvm::BasicBlock *InvokeDest = getInvokeDest()) {
1190fa8b4955SDouglas Gregor         llvm::BasicBlock *Cont = createBasicBlock("invoke.cont");
1191fa8b4955SDouglas Gregor         Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn();
1192fa8b4955SDouglas Gregor         EmitBlock(Cont);
1193fa8b4955SDouglas Gregor       } else {
1194fa8b4955SDouglas Gregor         // FIXME: Does this ever make sense?
119559486a2dSAnders Carlsson         Builder.CreateCall(F)->setDoesNotReturn();
1196fa8b4955SDouglas Gregor       }
119759486a2dSAnders Carlsson       Builder.CreateUnreachable();
119859486a2dSAnders Carlsson     }
119959486a2dSAnders Carlsson   }
120059486a2dSAnders Carlsson 
120159486a2dSAnders Carlsson   if (CanBeZero) {
120259486a2dSAnders Carlsson     Builder.CreateBr(ContBlock);
120359486a2dSAnders Carlsson     EmitBlock(NullBlock);
120459486a2dSAnders Carlsson     Builder.CreateBr(ContBlock);
120559486a2dSAnders Carlsson   }
120659486a2dSAnders Carlsson   EmitBlock(ContBlock);
120759486a2dSAnders Carlsson   if (CanBeZero) {
120859486a2dSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(LTy);
120959486a2dSAnders Carlsson     PHI->reserveOperandSpace(2);
121059486a2dSAnders Carlsson     PHI->addIncoming(V, NonZeroBlock);
121159486a2dSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock);
121259486a2dSAnders Carlsson     V = PHI;
121359486a2dSAnders Carlsson   }
121459486a2dSAnders Carlsson 
121559486a2dSAnders Carlsson   return V;
121659486a2dSAnders Carlsson }
1217